From ff4b045c260963b25b84d22020541f071a2cc472 Mon Sep 17 00:00:00 2001
From: Wolfgang Kerzendorf <wkerzendorf@gmail.com>
Date: Fri, 12 Jul 2024 10:45:28 -0400
Subject: [PATCH] added configuration fodler

Co-authored-by: Joshua Shields <jvshields@users.noreply.github.com>
---
 .../montecarlo/configuration/base.py          | 91 +++++++++++++++++++
 .../montecarlo/configuration/constants.py     | 15 +++
 .../configuration/montecarlo_globals.py       |  9 ++
 3 files changed, 115 insertions(+)
 create mode 100644 tardis/transport/montecarlo/configuration/base.py
 create mode 100644 tardis/transport/montecarlo/configuration/constants.py
 create mode 100644 tardis/transport/montecarlo/configuration/montecarlo_globals.py

diff --git a/tardis/transport/montecarlo/configuration/base.py b/tardis/transport/montecarlo/configuration/base.py
new file mode 100644
index 00000000000..c1cef012f8e
--- /dev/null
+++ b/tardis/transport/montecarlo/configuration/base.py
@@ -0,0 +1,91 @@
+from numba import float64, int64, boolean
+from numba.experimental import jitclass
+import numpy as np
+from astropy import units as u
+
+from tardis.transport.montecarlo.configuration.constants import (
+    LineInteractionType,
+)
+
+from tardis.transport.montecarlo.configuration import montecarlo_globals
+
+montecarlo_config_spec = [
+    ("TEMPORARY_V_PACKET_BINS", int64),
+    ("NUMBER_OF_VPACKETS", int64),
+    ("MONTECARLO_SEED", int64),
+    ("PACKET_SEEDS", int64[:]),
+    ("SURVIVAL_PROBABILITY", float64),
+    ("VPACKET_TAU_RUSSIAN", float64),
+    ("INITIAL_TRACKING_ARRAY_LENGTH", int64),
+    ("VPACKET_SPAWN_START_FREQUENCY", float64),
+    ("VPACKET_SPAWN_END_FREQUENCY", float64),
+]
+
+
+@jitclass(montecarlo_config_spec)
+class MonteCarloConfiguration:
+    def __init__(self):
+        self.TEMPORARY_V_PACKET_BINS = 0
+        self.NUMBER_OF_VPACKETS = 0
+        self.MONTECARLO_SEED = 0
+
+        self.PACKET_SEEDS = np.empty(1, dtype=np.int64)
+        self.SURVIVAL_PROBABILITY = 0.0
+        self.VPACKET_TAU_RUSSIAN = 10.0
+
+        self.INITIAL_TRACKING_ARRAY_LENGTH = 0
+
+        self.VPACKET_SPAWN_START_FREQUENCY = 0
+        self.VPACKET_SPAWN_END_FREQUENCY = 1e200
+
+
+def configuration_initialize(
+    montecarlo_configuration, transport_solver, number_of_vpackets
+):
+    if transport_solver.line_interaction_type == "macroatom":
+        montecarlo_globals.LINE_INTERACTION_TYPE = LineInteractionType.MACROATOM
+    elif transport_solver.line_interaction_type == "downbranch":
+        montecarlo_globals.LINE_INTERACTION_TYPE = (
+            LineInteractionType.DOWNBRANCH
+        )
+    elif transport_solver.line_interaction_type == "scatter":
+        montecarlo_globals.LINE_INTERACTION_TYPE = LineInteractionType.SCATTER
+    else:
+        raise ValueError(
+            f'Line interaction type must be one of "macroatom",'
+            f'"downbranch", or "scatter" but is '
+            f"{transport_solver.line_interaction_type}"
+        )
+    montecarlo_configuration.NUMBER_OF_VPACKETS = number_of_vpackets
+    montecarlo_configuration.TEMPORARY_V_PACKET_BINS = number_of_vpackets
+    montecarlo_globals.ENABLE_FULL_RELATIVITY = (
+        transport_solver.enable_full_relativity
+    )
+    montecarlo_configuration.MONTECARLO_SEED = (
+        transport_solver.packet_source.base_seed
+    )
+    montecarlo_configuration.VPACKET_SPAWN_START_FREQUENCY = (
+        transport_solver.virtual_spectrum_spawn_range.end.to(
+            u.Hz, equivalencies=u.spectral()
+        ).value
+    )
+    montecarlo_configuration.VPACKET_SPAWN_END_FREQUENCY = (
+        transport_solver.virtual_spectrum_spawn_range.start.to(
+            u.Hz, equivalencies=u.spectral()
+        ).value
+    )
+    montecarlo_globals.ENABLE_VPACKET_TRACKING = (
+        transport_solver.enable_vpacket_tracking
+    )
+    montecarlo_globals.ENABLE_RPACKET_TRACKING = (
+        transport_solver.enable_rpacket_tracking
+    )
+
+    montecarlo_globals.LEGACY_MODE_ENABLED = transport_solver.enable_legacy_mode
+
+    montecarlo_globals.DISABLE_ELECTRON_SCATTERING = (
+        transport_solver.disable_electron_scattering
+    )
+    montecarlo_globals.DISABLE_LINE_SCATTERING = (
+        transport_solver.disable_line_scattering
+    )
diff --git a/tardis/transport/montecarlo/configuration/constants.py b/tardis/transport/montecarlo/configuration/constants.py
new file mode 100644
index 00000000000..f853aad91ba
--- /dev/null
+++ b/tardis/transport/montecarlo/configuration/constants.py
@@ -0,0 +1,15 @@
+from enum import IntEnum
+from tardis import constants as const
+
+SIGMA_THOMSON = const.sigma_T.to("cm^2").value
+CLOSE_LINE_THRESHOLD = 1e-14
+C_SPEED_OF_LIGHT = const.c.to("cm/s").value
+MISS_DISTANCE = 1e99
+KB = const.k_B.cgs.value
+H = const.h.cgs.value
+
+
+class LineInteractionType(IntEnum):
+    SCATTER = 0
+    DOWNBRANCH = 1
+    MACROATOM = 2
diff --git a/tardis/transport/montecarlo/configuration/montecarlo_globals.py b/tardis/transport/montecarlo/configuration/montecarlo_globals.py
new file mode 100644
index 00000000000..b7617705ad6
--- /dev/null
+++ b/tardis/transport/montecarlo/configuration/montecarlo_globals.py
@@ -0,0 +1,9 @@
+ENABLE_FULL_RELATIVITY = False
+LINE_INTERACTION_TYPE = 0
+DISABLE_ELECTRON_SCATTERING = False
+DISABLE_LINE_SCATTERING = False
+
+LEGACY_MODE_ENABLED = False
+ENABLE_RPACKET_TRACKING = False
+CONTINUUM_PROCESSES_ENABLED = False
+ENABLE_VPACKET_TRACKING = False