{
 "cells": [
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# How to Read STELLA models with TARDIS"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Iterations:          0/? [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Packets:             0/? [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "from tardis.io.model import read_stella_model"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "stella_model = read_stella_model('mesa.stella.dat')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "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>mass_of_cell</th>\n",
       "      <th>cell_center_m</th>\n",
       "      <th>cell_center_r</th>\n",
       "      <th>cell_center_v</th>\n",
       "      <th>avg_density</th>\n",
       "      <th>radiation_pressure</th>\n",
       "      <th>avg_temperature</th>\n",
       "      <th>radiation_temperature</th>\n",
       "      <th>avg_opacity</th>\n",
       "      <th>tau</th>\n",
       "      <th>...</th>\n",
       "      <th>cr48</th>\n",
       "      <th>cr60</th>\n",
       "      <th>fe52</th>\n",
       "      <th>fe54</th>\n",
       "      <th>fe56</th>\n",
       "      <th>co56</th>\n",
       "      <th>ni56</th>\n",
       "      <th>luminosity</th>\n",
       "      <th>n_bar</th>\n",
       "      <th>n_e</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>6.006769e+29</td>\n",
       "      <td>5.190243e+33</td>\n",
       "      <td>2.517209e+13</td>\n",
       "      <td>3.930633e+06</td>\n",
       "      <td>1.005529e-11</td>\n",
       "      <td>11237.954506</td>\n",
       "      <td>45943.453023</td>\n",
       "      <td>45943.453023</td>\n",
       "      <td>0.162712</td>\n",
       "      <td>5403.504881</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000021</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.079500e-03</td>\n",
       "      <td>7.648868e+38</td>\n",
       "      <td>6.055413e+12</td>\n",
       "      <td>2.412188e+12</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>1.262406e+30</td>\n",
       "      <td>5.191506e+33</td>\n",
       "      <td>2.970626e+13</td>\n",
       "      <td>5.050403e+06</td>\n",
       "      <td>2.928088e-11</td>\n",
       "      <td>11236.544022</td>\n",
       "      <td>45942.182229</td>\n",
       "      <td>45942.182229</td>\n",
       "      <td>0.164412</td>\n",
       "      <td>5381.678444</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000021</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.079500e-03</td>\n",
       "      <td>9.609214e+38</td>\n",
       "      <td>1.762304e+13</td>\n",
       "      <td>6.969790e+12</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1.264624e+30</td>\n",
       "      <td>5.192771e+33</td>\n",
       "      <td>3.198879e+13</td>\n",
       "      <td>5.680288e+06</td>\n",
       "      <td>4.619659e-11</td>\n",
       "      <td>11133.526767</td>\n",
       "      <td>45836.518142</td>\n",
       "      <td>45836.518142</td>\n",
       "      <td>0.165012</td>\n",
       "      <td>5364.280084</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000021</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.079500e-03</td>\n",
       "      <td>1.050921e+39</td>\n",
       "      <td>2.783916e+13</td>\n",
       "      <td>1.091728e+13</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>1.263067e+30</td>\n",
       "      <td>5.194034e+33</td>\n",
       "      <td>3.357130e+13</td>\n",
       "      <td>6.135230e+06</td>\n",
       "      <td>5.896887e-11</td>\n",
       "      <td>11133.526767</td>\n",
       "      <td>45836.518142</td>\n",
       "      <td>45836.518142</td>\n",
       "      <td>0.165310</td>\n",
       "      <td>5348.854704</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000021</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.079500e-03</td>\n",
       "      <td>1.100921e+39</td>\n",
       "      <td>3.545183e+13</td>\n",
       "      <td>1.392188e+13</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>1.259387e+30</td>\n",
       "      <td>5.195293e+33</td>\n",
       "      <td>3.480638e+13</td>\n",
       "      <td>6.495173e+06</td>\n",
       "      <td>6.912717e-11</td>\n",
       "      <td>11032.838720</td>\n",
       "      <td>45732.362031</td>\n",
       "      <td>45732.362031</td>\n",
       "      <td>0.165510</td>\n",
       "      <td>5334.724893</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000021</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.079500e-03</td>\n",
       "      <td>1.140921e+39</td>\n",
       "      <td>4.166219e+13</td>\n",
       "      <td>1.632649e+13</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",
       "      <td>...</td>\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",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>396</th>\n",
       "      <td>1.511731e+31</td>\n",
       "      <td>2.612818e+34</td>\n",
       "      <td>1.533756e+15</td>\n",
       "      <td>3.435050e+08</td>\n",
       "      <td>1.540408e-14</td>\n",
       "      <td>0.136196</td>\n",
       "      <td>3259.329684</td>\n",
       "      <td>2710.845252</td>\n",
       "      <td>0.000430</td>\n",
       "      <td>0.001173</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000017</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.108306e-14</td>\n",
       "      <td>3.729654e+41</td>\n",
       "      <td>9.283015e+09</td>\n",
       "      <td>4.408753e+05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>397</th>\n",
       "      <td>1.511951e+31</td>\n",
       "      <td>2.614332e+34</td>\n",
       "      <td>1.576303e+15</td>\n",
       "      <td>3.545750e+08</td>\n",
       "      <td>1.168519e-14</td>\n",
       "      <td>0.128874</td>\n",
       "      <td>3192.480784</td>\n",
       "      <td>2673.651060</td>\n",
       "      <td>0.000467</td>\n",
       "      <td>0.000941</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000017</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.108306e-14</td>\n",
       "      <td>3.719770e+41</td>\n",
       "      <td>7.040251e+09</td>\n",
       "      <td>3.305759e+05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>398</th>\n",
       "      <td>1.512167e+31</td>\n",
       "      <td>2.615844e+34</td>\n",
       "      <td>1.632356e+15</td>\n",
       "      <td>3.682800e+08</td>\n",
       "      <td>8.348776e-15</td>\n",
       "      <td>0.119719</td>\n",
       "      <td>3126.918024</td>\n",
       "      <td>2624.851429</td>\n",
       "      <td>0.000525</td>\n",
       "      <td>0.000695</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000017</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.108306e-14</td>\n",
       "      <td>3.719654e+41</td>\n",
       "      <td>5.027256e+09</td>\n",
       "      <td>2.333801e+05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>399</th>\n",
       "      <td>1.512377e+31</td>\n",
       "      <td>2.617356e+34</td>\n",
       "      <td>1.698066e+15</td>\n",
       "      <td>4.244850e+08</td>\n",
       "      <td>6.601396e-15</td>\n",
       "      <td>0.110183</td>\n",
       "      <td>3055.657872</td>\n",
       "      <td>2570.946173</td>\n",
       "      <td>0.000569</td>\n",
       "      <td>0.000448</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000017</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.108306e-14</td>\n",
       "      <td>3.709770e+41</td>\n",
       "      <td>3.975759e+09</td>\n",
       "      <td>1.842764e+05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>400</th>\n",
       "      <td>1.512582e+31</td>\n",
       "      <td>2.618867e+34</td>\n",
       "      <td>2.126273e+15</td>\n",
       "      <td>4.732800e+08</td>\n",
       "      <td>7.649548e-16</td>\n",
       "      <td>0.076237</td>\n",
       "      <td>2780.233476</td>\n",
       "      <td>2344.794175</td>\n",
       "      <td>0.001368</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.000017</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.108306e-14</td>\n",
       "      <td>3.679885e+41</td>\n",
       "      <td>4.606795e+08</td>\n",
       "      <td>2.123225e+04</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>400 rows × 36 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "     mass_of_cell  cell_center_m  cell_center_r  cell_center_v   avg_density  \\\n",
       "0                                                                              \n",
       "1    6.006769e+29   5.190243e+33   2.517209e+13   3.930633e+06  1.005529e-11   \n",
       "2    1.262406e+30   5.191506e+33   2.970626e+13   5.050403e+06  2.928088e-11   \n",
       "3    1.264624e+30   5.192771e+33   3.198879e+13   5.680288e+06  4.619659e-11   \n",
       "4    1.263067e+30   5.194034e+33   3.357130e+13   6.135230e+06  5.896887e-11   \n",
       "5    1.259387e+30   5.195293e+33   3.480638e+13   6.495173e+06  6.912717e-11   \n",
       "..            ...            ...            ...            ...           ...   \n",
       "396  1.511731e+31   2.612818e+34   1.533756e+15   3.435050e+08  1.540408e-14   \n",
       "397  1.511951e+31   2.614332e+34   1.576303e+15   3.545750e+08  1.168519e-14   \n",
       "398  1.512167e+31   2.615844e+34   1.632356e+15   3.682800e+08  8.348776e-15   \n",
       "399  1.512377e+31   2.617356e+34   1.698066e+15   4.244850e+08  6.601396e-15   \n",
       "400  1.512582e+31   2.618867e+34   2.126273e+15   4.732800e+08  7.649548e-16   \n",
       "\n",
       "     radiation_pressure  avg_temperature  radiation_temperature  avg_opacity  \\\n",
       "0                                                                              \n",
       "1          11237.954506     45943.453023           45943.453023     0.162712   \n",
       "2          11236.544022     45942.182229           45942.182229     0.164412   \n",
       "3          11133.526767     45836.518142           45836.518142     0.165012   \n",
       "4          11133.526767     45836.518142           45836.518142     0.165310   \n",
       "5          11032.838720     45732.362031           45732.362031     0.165510   \n",
       "..                  ...              ...                    ...          ...   \n",
       "396            0.136196      3259.329684            2710.845252     0.000430   \n",
       "397            0.128874      3192.480784            2673.651060     0.000467   \n",
       "398            0.119719      3126.918024            2624.851429     0.000525   \n",
       "399            0.110183      3055.657872            2570.946173     0.000569   \n",
       "400            0.076237      2780.233476            2344.794175     0.001368   \n",
       "\n",
       "             tau  ...  cr48  cr60  fe52  fe54      fe56  co56          ni56  \\\n",
       "0                 ...                                                         \n",
       "1    5403.504881  ...   0.0   0.0   0.0   0.0  0.000021   0.0  4.079500e-03   \n",
       "2    5381.678444  ...   0.0   0.0   0.0   0.0  0.000021   0.0  4.079500e-03   \n",
       "3    5364.280084  ...   0.0   0.0   0.0   0.0  0.000021   0.0  4.079500e-03   \n",
       "4    5348.854704  ...   0.0   0.0   0.0   0.0  0.000021   0.0  4.079500e-03   \n",
       "5    5334.724893  ...   0.0   0.0   0.0   0.0  0.000021   0.0  4.079500e-03   \n",
       "..           ...  ...   ...   ...   ...   ...       ...   ...           ...   \n",
       "396     0.001173  ...   0.0   0.0   0.0   0.0  0.000017   0.0  1.108306e-14   \n",
       "397     0.000941  ...   0.0   0.0   0.0   0.0  0.000017   0.0  1.108306e-14   \n",
       "398     0.000695  ...   0.0   0.0   0.0   0.0  0.000017   0.0  1.108306e-14   \n",
       "399     0.000448  ...   0.0   0.0   0.0   0.0  0.000017   0.0  1.108306e-14   \n",
       "400     0.000000  ...   0.0   0.0   0.0   0.0  0.000017   0.0  1.108306e-14   \n",
       "\n",
       "       luminosity         n_bar           n_e  \n",
       "0                                              \n",
       "1    7.648868e+38  6.055413e+12  2.412188e+12  \n",
       "2    9.609214e+38  1.762304e+13  6.969790e+12  \n",
       "3    1.050921e+39  2.783916e+13  1.091728e+13  \n",
       "4    1.100921e+39  3.545183e+13  1.392188e+13  \n",
       "5    1.140921e+39  4.166219e+13  1.632649e+13  \n",
       "..            ...           ...           ...  \n",
       "396  3.729654e+41  9.283015e+09  4.408753e+05  \n",
       "397  3.719770e+41  7.040251e+09  3.305759e+05  \n",
       "398  3.719654e+41  5.027256e+09  2.333801e+05  \n",
       "399  3.709770e+41  3.975759e+09  1.842764e+05  \n",
       "400  3.679885e+41  4.606795e+08  2.123225e+04  \n",
       "\n",
       "[400 rows x 36 columns]"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "stella_model.data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'t_max': <Quantity 50. d>,\n",
       " 'zones': 400,\n",
       " 'inner_boundary_mass': <Quantity 5.19024252e+33 g>,\n",
       " 'total_mass': <Quantity 2.61886734e+34 g>,\n",
       " 'mass_of_cell_unit': Unit(\"g\"),\n",
       " 'cell_center_m_unit': Unit(\"g\"),\n",
       " 'cell_center_r_unit': Unit(\"cm\"),\n",
       " 'cell_center_v_unit': Unit(\"cm / s\"),\n",
       " 'outer_edge_m_unit': Unit(\"g\"),\n",
       " 'outer_edge_r_unit': Unit(\"cm\")}"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "stella_model.metadata"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.8.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}