dynamic_bar.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import plotly_express as px  # import plotly.express as px\n",
    "import plotly.graph_objects as go\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "import os\n",
    "\n",
    "name = [\"NeRF\", \"PANO\", \"OURS+\", \"OURS\"]\n",
    "foveal = 8.0\n",
    "mid = 2.8\n",
    "far = 2.6\n",
    "blend = 0.1\n",
    "ours_full = 562\n",
    "pano = 1e4\n",
    "nerf = 9e4\n",
    "\n",
    "times = {\n",
    "    \"fovea_l\": 0,\n",
    "    \"mid_l\": 0,\n",
    "    \"far_l\": 0,\n",
    "    \"fovea_r\": 0,\n",
    "    \"mid_r\": 0,\n",
    "    \"far_r\": 0,\n",
    "    \"blend\": 0,\n",
    "    \"ours_full\": 0,\n",
    "    \"pano\": 0,\n",
    "    \"nerf\": 0,\n",
    "}\n",
    "clip = 0\n",
    "frame_id = 0\n",
    "\n",
    "\n",
    "def calc_total():\n",
    "    return [\n",
    "        times[\"nerf\"],\n",
    "        times[\"pano\"],\n",
    "        times[\"ours_full\"],\n",
    "        times[\"fovea_l\"] + times[\"mid_l\"] + times[\"far_l\"] +\n",
    "            times[\"fovea_r\"] + times[\"mid_r\"] + times[\"far_r\"] + times[\"blend\"]\n",
    "    ]\n",
    "\n",
    "\n",
    "def draw_frame(*, xlim=None, **kwargs):\n",
    "    global frame_id\n",
    "    for key in kwargs:\n",
    "        times[key] = kwargs[key]\n",
    "    tot = calc_total()\n",
    "    data = {\n",
    "        \"fovea_l\": [0, 0, 0, times[\"fovea_l\"]],\n",
    "        \"mid_l\": [0, 0, 0, times[\"mid_l\"]],\n",
    "        \"far_l\": [0, 0, 0, times[\"far_l\"]],\n",
    "        \"fovea_r\": [0, 0, 0, times[\"fovea_r\"]],\n",
    "        \"mid_r\": [0, 0, 0, times[\"mid_r\"]],\n",
    "        \"far_r\": [0, 0, 0, times[\"far_r\"]],\n",
    "        \"blend\": [0, 0, 0, times[\"blend\"]],\n",
    "        \"ours_full\": [0, 0, times[\"ours_full\"], 0],\n",
    "        \"pano\": [0, times[\"pano\"], 0, 0],\n",
    "        \"nerf\": [times[\"nerf\"], 0, 0, 0],\n",
    "    }\n",
    "    if xlim is None or xlim < max(tot) * 1.1:\n",
    "        xlim = max(tot) * 1.1\n",
    "    \n",
    "    fig = go.Figure()\n",
    "    times_keys = list(times.keys())\n",
    "    for key in times_keys:\n",
    "        if key == times_keys[-1]:\n",
    "            fig.add_trace(go.Bar(\n",
    "                y=name,\n",
    "                x=data[key],\n",
    "                name=key,\n",
    "                orientation='h',\n",
    "                text=[\"\" if item == 0 else f\"{item:.1f}\" if item < 1000 else f\"{item:.1e}\" for item in tot],\n",
    "                textposition=\"outside\"\n",
    "            ))\n",
    "        else:\n",
    "            fig.add_trace(go.Bar(\n",
    "                y=name,\n",
    "                x=data[key],\n",
    "                name=key,\n",
    "                orientation='h',\n",
    "            ))\n",
    "    fig.update_traces(width=0.5)\n",
    "    fig.update_layout(barmode='stack', showlegend=False,\n",
    "                      yaxis_visible=False, yaxis_showticklabels=False, xaxis_range=[0, xlim])\n",
    "    \n",
    "    # fig.show()\n",
    "    fig.write_image(f\"dynamic_bar/clip_{clip}/{frame_id:04d}.png\", width=1920 // 2, height=1080 // 2, scale=2)\n",
    "    frame_id = frame_id + 1\n",
    "\n",
    "def add_animation(*, frames, xlim=None, **kwargs):\n",
    "    if frames == 1:\n",
    "        draw_frame(**kwargs, xlim=xlim)\n",
    "        return\n",
    "    data = {\n",
    "        key: np.linspace(times[key], kwargs[key], frames)\n",
    "        for key in kwargs\n",
    "    }\n",
    "    for i in range(frames):\n",
    "        draw_frame(**{key: data[key][i] for key in data}, xlim=xlim)\n",
    "\n",
    "def new_clip():\n",
    "    global clip, frame_id\n",
    "    clip += 1\n",
    "    frame_id = 0\n",
    "    os.system(f\"mkdir dynamic_bar/clip_{clip}\")\n",
    "\n",
    "os.system('rm -f -r dynamic_bar')\n",
    "os.system('mkdir dynamic_bar')\n",
    "\n",
    "# ours mono\n",
    "new_clip()\n",
    "add_animation(fovea_l=foveal, frames=48, xlim=30)   # Step 1: grow foveal\n",
    "add_animation(mid_l=mid, frames=16, xlim=30)         # Step 2: grow mid\n",
    "add_animation(far_l=far, frames=16, xlim=30)         # Step 3: grow far\n",
    "add_animation(blend=blend, frames=1, xlim=30)     # Step 4: grow blend\n",
    "\n",
    "# ours stereo\n",
    "new_clip()\n",
    "add_animation(fovea_r=foveal, frames=24, xlim=30)   # Step 1: grow foveal\n",
    "add_animation(mid_r=mid, frames=8, xlim=30)         # Step 2: grow mid\n",
    "add_animation(far_r=far, frames=8, xlim=30)         # Step 3: grow far\n",
    "\n",
    "# ours stereo adapt\n",
    "new_clip()\n",
    "add_animation(mid_r=0, far_r=0, frames=24, xlim=30)\n",
    "\n",
    "# other series\n",
    "new_clip()\n",
    "add_animation(ours_full=ours_full, frames=48, xlim=30)\n",
    "new_clip()\n",
    "add_animation(pano=pano, frames=48, xlim=30)\n",
    "new_clip()\n",
    "add_animation(nerf=nerf, frames=48, xlim=30)\n",
    "\n",
    "#os.system(f'ffmpeg -y -r 24 -i dynamic_bar/%04d.png dynamic_bar.avi')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "fig = px.bar(\n",
    "    df1,  # 绘图数据\n",
    "    x=list(times.keys()),    # y轴\n",
    "    y=\"name\",  # x轴\n",
    "    orientation='h',   # 水平柱状图\n",
    "    #text=[[\"a\", \"tot\"], \"tot1\", \"tot2\", {\"fovea_l\": \"\", \"blend_r\": 13.5}]   # 需要显示的数据\n",
    ")\n",
    "fig.update_traces(textposition=\"outside\", showlegend=False, text=[[\"a\"]*11, [\"b\"]*11, [\"c\"]*11, [\"d\"]*11,[\"e\"]*11])\n",
    "fig.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "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>country</th>\n",
       "      <th>continent</th>\n",
       "      <th>year</th>\n",
       "      <th>lifeExp</th>\n",
       "      <th>pop</th>\n",
       "      <th>gdpPercap</th>\n",
       "      <th>iso_alpha</th>\n",
       "      <th>iso_num</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Afghanistan</td>\n",
       "      <td>Asia</td>\n",
       "      <td>1952</td>\n",
       "      <td>28.801</td>\n",
       "      <td>8425333</td>\n",
       "      <td>779.445314</td>\n",
       "      <td>AFG</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Afghanistan</td>\n",
       "      <td>Asia</td>\n",
       "      <td>1957</td>\n",
       "      <td>30.332</td>\n",
       "      <td>9240934</td>\n",
       "      <td>820.853030</td>\n",
       "      <td>AFG</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Afghanistan</td>\n",
       "      <td>Asia</td>\n",
       "      <td>1962</td>\n",
       "      <td>31.997</td>\n",
       "      <td>10267083</td>\n",
       "      <td>853.100710</td>\n",
       "      <td>AFG</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Afghanistan</td>\n",
       "      <td>Asia</td>\n",
       "      <td>1967</td>\n",
       "      <td>34.020</td>\n",
       "      <td>11537966</td>\n",
       "      <td>836.197138</td>\n",
       "      <td>AFG</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Afghanistan</td>\n",
       "      <td>Asia</td>\n",
       "      <td>1972</td>\n",
       "      <td>36.088</td>\n",
       "      <td>13079460</td>\n",
       "      <td>739.981106</td>\n",
       "      <td>AFG</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1699</th>\n",
       "      <td>Zimbabwe</td>\n",
       "      <td>Africa</td>\n",
       "      <td>1987</td>\n",
       "      <td>62.351</td>\n",
       "      <td>9216418</td>\n",
       "      <td>706.157306</td>\n",
       "      <td>ZWE</td>\n",
       "      <td>716</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1700</th>\n",
       "      <td>Zimbabwe</td>\n",
       "      <td>Africa</td>\n",
       "      <td>1992</td>\n",
       "      <td>60.377</td>\n",
       "      <td>10704340</td>\n",
       "      <td>693.420786</td>\n",
       "      <td>ZWE</td>\n",
       "      <td>716</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1701</th>\n",
       "      <td>Zimbabwe</td>\n",
       "      <td>Africa</td>\n",
       "      <td>1997</td>\n",
       "      <td>46.809</td>\n",
       "      <td>11404948</td>\n",
       "      <td>792.449960</td>\n",
       "      <td>ZWE</td>\n",
       "      <td>716</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1702</th>\n",
       "      <td>Zimbabwe</td>\n",
       "      <td>Africa</td>\n",
       "      <td>2002</td>\n",
       "      <td>39.989</td>\n",
       "      <td>11926563</td>\n",
       "      <td>672.038623</td>\n",
       "      <td>ZWE</td>\n",
       "      <td>716</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1703</th>\n",
       "      <td>Zimbabwe</td>\n",
       "      <td>Africa</td>\n",
       "      <td>2007</td>\n",
       "      <td>43.487</td>\n",
       "      <td>12311143</td>\n",
       "      <td>469.709298</td>\n",
       "      <td>ZWE</td>\n",
       "      <td>716</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>1704 rows × 8 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "          country continent  year  lifeExp       pop   gdpPercap iso_alpha  \\\n",
       "0     Afghanistan      Asia  1952   28.801   8425333  779.445314       AFG   \n",
       "1     Afghanistan      Asia  1957   30.332   9240934  820.853030       AFG   \n",
       "2     Afghanistan      Asia  1962   31.997  10267083  853.100710       AFG   \n",
       "3     Afghanistan      Asia  1967   34.020  11537966  836.197138       AFG   \n",
       "4     Afghanistan      Asia  1972   36.088  13079460  739.981106       AFG   \n",
       "...           ...       ...   ...      ...       ...         ...       ...   \n",
       "1699     Zimbabwe    Africa  1987   62.351   9216418  706.157306       ZWE   \n",
       "1700     Zimbabwe    Africa  1992   60.377  10704340  693.420786       ZWE   \n",
       "1701     Zimbabwe    Africa  1997   46.809  11404948  792.449960       ZWE   \n",
       "1702     Zimbabwe    Africa  2002   39.989  11926563  672.038623       ZWE   \n",
       "1703     Zimbabwe    Africa  2007   43.487  12311143  469.709298       ZWE   \n",
       "\n",
       "      iso_num  \n",
       "0           4  \n",
       "1           4  \n",
       "2           4  \n",
       "3           4  \n",
       "4           4  \n",
       "...       ...  \n",
       "1699      716  \n",
       "1700      716  \n",
       "1701      716  \n",
       "1702      716  \n",
       "1703      716  \n",
       "\n",
       "[1704 rows x 8 columns]"
      ]
     },
     "execution_count": 33,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df = px.data.gapminder()\n",
    "df"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3.10.0 ('dvs')",
   "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.10.0"
  },
  "orig_nbformat": 4,
  "vscode": {
   "interpreter": {
    "hash": "4469b029896260c1221afa6e0e6159922aafd2738570e75b7bc15e28db242604"
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}