{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Performance testing of OBJ loader"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "%load_ext autoreload\n",
    "%autoreload 2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import imageio\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "import timeit\n",
    "import lmfunctest as ft\n",
    "import lmscene\n",
    "import lightmetrica as lm"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "%load_ext lightmetrica_jupyter"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[I|0.000|114@user  ] Lightmetrica -- Version 3.0.0 (rev. fe30e7c) Linux x64\n"
     ]
    }
   ],
   "source": [
    "lm.init('user::default', {})\n",
    "lm.parallel.init('parallel::openmp', {\n",
    "    'numThreads': -1\n",
    "})\n",
    "lm.log.init('logger::jupyter', {})\n",
    "lm.info()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[I|0.020|179@compon] Loading plugin [name='objloader_tinyobjloader']\n",
      "[I|0.020|197@compon]   Successfully loaded\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "True"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "lm.comp.loadPlugin(os.path.join(ft.env.bin_path, 'objloader_tinyobjloader'))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "objloaders = ['objloader::simple', 'objloader::tinyobjloader']\n",
    "scenes = lmscene.scenes_small()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[I|0.068|48@assets ] Loading asset [name='camera_main']\n",
      "[I|0.068|48@assets ] Loading asset [name='model_obj']\n",
      "[I|0.068|29@objload]   Loading OBJ file [path='fireplace_room.obj']\n",
      "[I|0.068|169@objloa]   Loading MTL file [path='fireplace_room.mtl']\n",
      "[I|0.068|44@texture]   Loading texture [path='wood.ppm']\n",
      "[I|0.174|44@texture]   Loading texture [path='leaf.ppm']\n",
      "[I|0.178|44@texture]   Loading texture [path='picture8.ppm']\n",
      "[I|0.219|44@texture]   Loading texture [path='wood5.ppm']\n",
      "[I|0.660|48@assets ] Loading asset [name='camera_main']\n",
      "[I|0.661|48@assets ] Loading asset [name='model_obj']\n",
      "[I|0.922|44@texture]   Loading texture [path='wood.ppm']\n",
      "[I|1.027|44@texture]   Loading texture [path='leaf.ppm']\n",
      "[I|1.030|44@texture]   Loading texture [path='picture8.ppm']\n",
      "[I|1.066|44@texture]   Loading texture [path='wood5.ppm']\n",
      "[I|1.131|48@assets ] Loading asset [name='camera_main']\n",
      "[I|1.132|48@assets ] Loading asset [name='model_obj']\n",
      "[I|1.132|29@objload]   Loading OBJ file [path='CornellBox-Sphere.obj']\n",
      "[I|1.132|169@objloa]   Loading MTL file [path='CornellBox-Sphere.mtl']\n",
      "[I|1.135|48@assets ] Loading asset [name='camera_main']\n",
      "[I|1.135|48@assets ] Loading asset [name='model_obj']\n",
      "[W|1.138|43@objload]   Both `d` and `Tr` parameters defined for \"light\". Use the value of `d` for dissolve (line 76 in .mtl.)\n",
      "\n",
      "[I|1.139|48@assets ] Loading asset [name='camera_main']\n",
      "[I|1.139|48@assets ] Loading asset [name='model_obj']\n",
      "[I|1.139|29@objload]   Loading OBJ file [path='cube.obj']\n",
      "[I|1.139|169@objloa]   Loading MTL file [path='default.mtl']\n",
      "[I|1.139|44@texture]   Loading texture [path='default.png']\n",
      "[I|1.156|48@assets ] Loading asset [name='camera_main']\n",
      "[I|1.156|48@assets ] Loading asset [name='model_obj']\n",
      "[I|1.156|44@texture]   Loading texture [path='default.png']\n"
     ]
    }
   ],
   "source": [
    "loading_time_df = pd.DataFrame(columns=objloaders, index=scenes)\n",
    "for scene in scenes:\n",
    "    # Check consistency with other loaders\n",
    "    for objloader in objloaders:\n",
    "        # Load the scene with selected obj loader\n",
    "        lm.objloader.init(objloader, {})\n",
    "        lm.reset()\n",
    "        def load_scene():\n",
    "            lmscene.load(ft.env.scene_path, scene)\n",
    "        loading_time = timeit.timeit(stmt=load_scene, number=1)\n",
    "        loading_time_df[objloader][scene] = loading_time"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>objloader::simple</th>\n",
       "      <th>objloader::tinyobjloader</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>fireplace_room</th>\n",
       "      <td>0.590972</td>\n",
       "      <td>0.468631</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>cornell_box_sphere</th>\n",
       "      <td>0.00331342</td>\n",
       "      <td>0.00296321</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>cube</th>\n",
       "      <td>0.0166067</td>\n",
       "      <td>0.0149226</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                   objloader::simple objloader::tinyobjloader\n",
       "fireplace_room              0.590972                 0.468631\n",
       "cornell_box_sphere        0.00331342               0.00296321\n",
       "cube                       0.0166067                0.0149226"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "loading_time_df"
   ]
  }
 ],
 "metadata": {
  "jupytext": {
   "formats": "ipynb,py:light"
  },
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}