Restructure/separate transport state

docs/io/output/rpacket_tracking.ipynb

@@ -202,7 +202,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "type(sim.transport.rpacket_tracker)"
+    "type(sim.transport.transport_state.rpacket_tracker)"
    ]
   },
   {
@@ -243,7 +243,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "len(sim.transport.rpacket_tracker)"
+    "len(sim.transport.transport_state.rpacket_tracker)"
    ]
   },
   {
@@ -281,7 +281,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sim.transport.rpacket_tracker[10].index"
+    "sim.transport.transport_state.rpacket_tracker[10].index"
    ]
   },
   {
@@ -300,7 +300,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sim.transport.rpacket_tracker[10].seed"
+    "sim.transport.transport_state.rpacket_tracker[10].seed"
    ]
   },
   {
@@ -319,7 +319,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sim.transport.rpacket_tracker[10].status"
+    "sim.transport.transport_state.rpacket_tracker[10].status"
    ]
   },
   {
@@ -347,7 +347,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "len(sim.transport.rpacket_tracker[10].shell_id)"
+    "len(sim.transport.transport_state.rpacket_tracker[10].shell_id)"
    ]
   },
   {
@@ -403,7 +403,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "type(sim.transport.rpacket_tracker_df)"
+    "type(sim.transport.transport_state.rpacket_tracker_df)"
    ]
   },
   {
@@ -461,7 +461,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sim.transport.rpacket_tracker_df"
+    "sim.transport.transport_state.rpacket_tracker_df"
    ]
   },
   {
@@ -490,7 +490,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sim.transport.rpacket_tracker_df.loc[10]"
+    "sim.transport.transport_state.rpacket_tracker_df.loc[10]"
    ]
   },
   {
@@ -509,7 +509,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sim.transport.rpacket_tracker_df.loc[10][\"energy\"]"
+    "sim.transport.transport_state.rpacket_tracker_df.loc[10][\"energy\"]"
    ]
   },
   {
@@ -528,7 +528,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sim.transport.rpacket_tracker_df.loc[10,5][\"energy\"]"
+    "sim.transport.transport_state.rpacket_tracker_df.loc[10,5][\"energy\"]"
    ]
   },
   {
@@ -538,7 +538,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sim.transport.rpacket_tracker_df.loc[10][\"energy\"][5]"
+    "sim.transport.transport_state.rpacket_tracker_df.loc[10][\"energy\"][5]"
    ]
   },
   {
@@ -567,7 +567,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.13"
+   "version": "3.11.5"
   },
   "vscode": {
    "interpreter": {

docs/io/visualization/montecarlo_packet_visualization.ipynb

@@ -89,7 +89,7 @@
    "outputs": [],
    "source": [
     "# accessing the rpacket_tracker dataframe\n",
-    "sim.transport.rpacket_tracker_df"
+    "sim.transport.transport_state.rpacket_tracker_df"
    ]
   },
   {
@@ -191,11 +191,11 @@
    "source": [
     "# function for getting coordinates of all packets\n",
     "def get_coordinates_multiple_packets_ints(r_packet_tracker,time):\n",
-    "    thetas = np.linspace(0, 2*math.pi,sim.transport.rpacket_tracker_df.index[-1][0]+2)\n",
+    "    thetas = np.linspace(0, 2*math.pi,sim.transport.transport_state.rpacket_tracker_df.index[-1][0]+2)\n",
     "    x = []\n",
     "    y = []\n",
     "    inters = []\n",
-    "    for i in range(sim.transport.rpacket_tracker_df.index[-1][0]+1):\n",
+    "    for i in range(sim.transport.transport_state.rpacket_tracker_df.index[-1][0]+1):\n",
     "        xs,ys,ints = get_x_y_ints_with_theta_init(r_packet_tracker.loc[i][\"r\"],r_packet_tracker.loc[i][\"mu\"],time,r_packet_tracker.loc[i][\"interaction_type\"],'n',thetas[i])\n",
     "        x.append(xs)\n",
     "        y.append(ys)\n",
@@ -236,7 +236,7 @@
     "# animated plot\n",
     "\n",
     "v_shells = sim.simulation_state.radius.value * 1e-5 / sim.simulation_state.time_explosion.value\n",
-    "xs,ys,ints = get_coordinates_multiple_packets_ints(sim.transport.rpacket_tracker_df,sim.simulation_state.time_explosion.value)\n",
+    "xs,ys,ints = get_coordinates_multiple_packets_ints(sim.transport.transport_state.rpacket_tracker_df,sim.simulation_state.time_explosion.value)\n",
     "xs,ys,ints,max_size = get_equal_array_size(xs,ys,ints)\n",
     "\n",
     "fig = go.Figure()\n",
@@ -273,7 +273,7 @@
     "\n",
     "#Add the packet trajectory\n",
     "\n",
-    "df = sim.transport.rpacket_tracker_df\n",
+    "df = sim.transport.transport_state.rpacket_tracker_df\n",
     "interaction_from_num = {0: \"No Interaction\", 1: \"EScattering\", 2: \"Line\"}\n",
     "interaction_color_from_num = {0: \"darkslategrey\", 1: \"#3366FF\", 2: \"#FF3300\"}\n",
     "interaction_opacity_from_num = {0: 0, 1: 1, 2: 1}\n",

tardis/analysis.py

@@ -3,25 +3,26 @@
 """
 
 import re
-import os
 
-from astropy import units as u
-from tardis import constants
 import numpy as np
 import pandas as pd
+from astropy import units as u
+
+from tardis import constants
 
 INVALID_ION_ERROR_MSG = "Atomic number, ion_number pair not present in model"
 
 
-class LastLineInteraction(object):
+class LastLineInteraction:
     @classmethod
     def from_simulation(cls, simulation, packet_filter_mode="packet_out_nu"):
+        transport_state = simulation.transport.transport_state
         return cls(
-            simulation.transport.last_line_interaction_in_id,
-            simulation.transport.last_line_interaction_out_id,
-            simulation.transport.last_line_interaction_shell_id,
-            simulation.transport.transport_state.packet_collection.output_nus,
-            simulation.transport.last_interaction_in_nu,
+            transport_state.last_line_interaction_in_id,
+            transport_state.last_line_interaction_out_id,
+            transport_state.last_line_interaction_shell_id,
+            transport_state.packet_collection.output_nus,
+            transport_state.last_interaction_in_nu,
             simulation.plasma.atomic_data.lines,
             packet_filter_mode,
         )
@@ -260,7 +261,7 @@ def onpress(event):
         fig.canvas.mpl_connect("on_press", onpress)
 
 
-class TARDISHistory(object):
+class TARDISHistory:
     """
     Records the history of the model
     """

tardis/conftest.py

@@ -232,3 +232,14 @@ def simulation_verysimple(config_verysimple, atomic_dataset):
     sim = Simulation.from_config(config_verysimple, atom_data=atomic_data)
     sim.iterate(4000)
     return sim
+
+
+@pytest.fixture(scope="session")
+def simulation_verysimple_vpacket_tracking(config_verysimple, atomic_dataset):
+    atomic_data = deepcopy(atomic_dataset)
+    sim = Simulation.from_config(
+        config_verysimple, atom_data=atomic_data, virtual_packet_logging=True
+    )
+    sim.last_no_of_packets = 4000
+    sim.run_final()
+    return sim

tardis/grid/tests/test_grid.py

@@ -4,8 +4,11 @@
 import pandas as pd
 
 import tardis
+
+from pathlib import Path
 import tardis.grid as grid
 
+
 DATA_PATH = Path(tardis.__path__[0]) / "grid" / "tests" / "data"
 
 

tardis/model/parse_input.py

@@ -598,6 +598,14 @@
     ValueError
         If both t_inner and luminosity_requested are None.
     """
+    if config.montecarlo.enable_full_relativity:
+        packet_source = BlackBodySimpleSourceRelativistic(
+            base_seed=config.montecarlo.seed,
+            time_explosion=config.supernova.time_explosion,
+        )
+    else:
+        packet_source = BlackBodySimpleSource(base_seed=config.montecarlo.seed)
+
     luminosity_requested = config.supernova.luminosity_requested
     if config.plasma.initial_t_inner > 0.0 * u.K:
         packet_source.radius = geometry.r_inner[0]