Fix frame transformations of continuum opacity (#2347)

tardis-sn · Jul 7, 2023 · d9b2165 · d9b2165
1 parent 0896a43
commit d9b2165
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tardis/montecarlo/montecarlo_numba/single_packet_loop.py b/tardis/montecarlo/montecarlo_numba/single_packet_loop.py
@@ -105,7 +105,10 @@ def single_packet_loop(
                 numba_plasma, comov_nu, r_packet.current_shell_id
             )
             chi_continuum = chi_e + chi_bf_tot + chi_ff
+
             escat_prob = chi_e / chi_continuum  # probability of e-scatter
+            if montecarlo_configuration.full_relativity:
+                chi_continuum *= doppler_factor
             distance, interaction_type, delta_shell = trace_packet(
                 r_packet,
                 numba_radial_1d_geometry,
@@ -129,6 +132,8 @@ def single_packet_loop(
         else:
             escat_prob = 1.0
             chi_continuum = chi_e
+            if montecarlo_configuration.full_relativity:
+                chi_continuum *= doppler_factor
             distance, interaction_type, delta_shell = trace_packet(
                 r_packet,
                 numba_radial_1d_geometry,

diff --git a/tardis/montecarlo/montecarlo_numba/vpacket.py b/tardis/montecarlo/montecarlo_numba/vpacket.py
@@ -115,6 +115,9 @@ def trace_vpacket_within_shell(
     else:
         chi_continuum = chi_e
 
+    if montecarlo_configuration.full_relativity:
+        chi_continuum *= doppler_factor
+
     tau_continuum = chi_continuum * distance_boundary
     tau_trace_combined = tau_continuum