def trace_vpacket_within_shell(v_packet, numba_model, numba_plasma):
"""
Trace VPacket within one shell (relatively simple operation)
"""
r_inner = numba_model.r_inner[v_packet.current_shell_id]
r_outer = numba_model.r_outer[v_packet.current_shell_id]
distance_boundary, delta_shell = calculate_distance_boundary(
v_packet.r, v_packet.mu, r_inner, r_outer)
# defining start for line interaction
start_line_id = v_packet.next_line_id
# e scattering initialization
cur_electron_density = numba_plasma.electron_density[
v_packet.current_shell_id]
tau_electron = calculate_tau_electron(cur_electron_density,
distance_boundary)
tau_trace_combined = tau_electron
# Calculating doppler factor
doppler_factor = get_doppler_factor(v_packet.r, v_packet.mu,
numba_model.time_explosion)
comov_nu = v_packet.nu * doppler_factor
cur_line_id = start_line_id
for cur_line_id in range(start_line_id, len(numba_plasma.line_list_nu)):
# if tau_trace_combined > 10: ### FIXME ?????
# break
nu_line = numba_plasma.line_list_nu[cur_line_id]
# TODO: Check if this is what the C code does
tau_trace_line = numba_plasma.tau_sobolev[cur_line_id,
v_packet.current_shell_id]
if cur_line_id == len(numba_plasma.line_list_nu) - 1:
is_last_line = True
else:
is_last_line = False
distance_trace_line = calculate_distance_line(
v_packet,
comov_nu,
is_last_line,
nu_line,
numba_model.time_explosion,
)
if distance_boundary <= distance_trace_line:
break
tau_trace_combined += tau_trace_line
else:
if cur_line_id == (len(numba_plasma.line_list_nu) - 1):
cur_line_id += 1
v_packet.next_line_id = cur_line_id
return tau_trace_combined, distance_boundary, delta_shell
def test_calculate_tau_electron(electron_density, distance):
actual = opacities.calculate_tau_electron(electron_density, distance)
expected = electron_density * SIGMA_THOMSON * distance
assert_almost_equal(actual, expected)
def trace_packet(r_packet, numba_model, numba_plasma, estimators):
"""
Traces the RPacket through the ejecta and stops when an interaction happens (heart of the calculation)
Parameters
----------
r_packet : tardis.montecarlo.montecarlo_numba.r_packet.RPacket
numba_model : tardis.montecarlo.montecarlo_numba.numba_interface.NumbaModel
numba_plasma : tardis.montecarlo.montecarlo_numba.numba_interface.NumbaPlasma
estimators : tardis.montecarlo.montecarlo_numba.numba_interface.Estimators
Returns
-------
"""
r_inner = numba_model.r_inner[r_packet.current_shell_id]
r_outer = numba_model.r_outer[r_packet.current_shell_id]
(
distance_boundary,
delta_shell,
) = calculate_distance_boundary(r_packet.r, r_packet.mu, r_inner, r_outer)
# defining start for line interaction
start_line_id = r_packet.next_line_id
# defining taus
tau_event = -np.log(np.random.random())
tau_trace_line_combined = 0.0
# e scattering initialization
cur_electron_density = numba_plasma.electron_density[
r_packet.current_shell_id
]
distance_electron = calculate_distance_electron(
cur_electron_density, tau_event
)
# Calculating doppler factor
doppler_factor = get_doppler_factor(
r_packet.r, r_packet.mu, numba_model.time_explosion
)
comov_nu = r_packet.nu * doppler_factor
cur_line_id = start_line_id # initializing varibale for Numba
# - do not remove
last_line_id = len(numba_plasma.line_list_nu) - 1
for cur_line_id in range(start_line_id, len(numba_plasma.line_list_nu)):
# Going through the lines
nu_line = numba_plasma.line_list_nu[cur_line_id]
nu_line_last_interaction = numba_plasma.line_list_nu[cur_line_id - 1]
# Getting the tau for the next line
tau_trace_line = numba_plasma.tau_sobolev[
cur_line_id, r_packet.current_shell_id
]
# Adding it to the tau_trace_line_combined
tau_trace_line_combined += tau_trace_line
# Calculating the distance until the current photons co-moving nu
# redshifts to the line frequency
is_last_line = cur_line_id == last_line_id
distance_trace = calculate_distance_line(
r_packet,
comov_nu,
is_last_line,
nu_line,
numba_model.time_explosion,
)
# calculating the tau electron of how far the trace has progressed
tau_trace_electron = calculate_tau_electron(
cur_electron_density, distance_trace
)
# calculating the trace
tau_trace_combined = tau_trace_line_combined + tau_trace_electron
if (
(distance_boundary <= distance_trace)
and (distance_boundary <= distance_electron)
) and distance_trace != 0.0:
interaction_type = InteractionType.BOUNDARY # BOUNDARY
r_packet.next_line_id = cur_line_id
distance = distance_boundary
break
if (
(distance_electron < distance_trace)
and (distance_electron < distance_boundary)
) and distance_trace != 0.0:
interaction_type = InteractionType.ESCATTERING
# print('scattering')
distance = distance_electron
r_packet.next_line_id = cur_line_id
break
# Updating the J_b_lu and E_dot_lu
# This means we are still looking for line interaction and have not
# been kicked out of the path by boundary or electron interaction
update_line_estimators(
estimators,
r_packet,
cur_line_id,
distance_trace,
numba_model.time_explosion,
)
if (
tau_trace_combined > tau_event
and not montecarlo_configuration.disable_line_scattering
):
interaction_type = InteractionType.LINE # Line
r_packet.last_interaction_in_nu = r_packet.nu
r_packet.last_line_interaction_in_id = cur_line_id
r_packet.next_line_id = cur_line_id
distance = distance_trace
break
# Recalculating distance_electron using tau_event -
# tau_trace_line_combined
distance_electron = calculate_distance_electron(
cur_electron_density, tau_event - tau_trace_line_combined
)
else: # Executed when no break occurs in the for loop
# We are beyond the line list now and the only next thing is to see
# if we are interacting with the boundary or electron scattering
if cur_line_id == (len(numba_plasma.line_list_nu) - 1):
# Treatment for last line
cur_line_id += 1
if distance_electron < distance_boundary:
distance = distance_electron
interaction_type = InteractionType.ESCATTERING
# print('scattering')
else:
distance = distance_boundary
interaction_type = InteractionType.BOUNDARY
# r_packet.next_line_id = cur_line_id
return distance, interaction_type, delta_shell
