def line_emission(r_packet, emission_line_id, time_explosion, numba_plasma):
"""
Sets the frequency of the RPacket properly given the emission channel
Parameters
-----------
r_packet: RPacket
emission_line_id: int
time_explosion: float
numba_plasma: NumbaPlasma
"""
r_packet.last_line_interaction_out_id = emission_line_id
if emission_line_id != r_packet.next_line_id:
pass
inverse_doppler_factor = get_inverse_doppler_factor(
r_packet.r, r_packet.mu, time_explosion)
r_packet.nu = (numba_plasma.line_list_nu[emission_line_id] *
inverse_doppler_factor)
r_packet.next_line_id = emission_line_id + 1
nu_line = numba_plasma.line_list_nu[emission_line_id]
if emission_line_id != (len(numba_plasma.line_list_nu) - 1):
test_for_close_line(r_packet, emission_line_id + 1, nu_line,
numba_plasma)
if montecarlo_configuration.full_relativity:
r_packet.mu = angle_aberration_CMF_to_LF(r_packet, time_explosion,
r_packet.mu)
def test_test_for_close_line(packet, line_id, nu_line, verysimple_numba_plasma,
expected):
r_packet.test_for_close_line(packet, line_id, nu_line,
verysimple_numba_plasma)
assert packet.is_close_line == expected
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 cur_line_id != (len(numba_plasma.line_list_nu) - 1):
test_for_close_line(
v_packet,
cur_line_id,
numba_plasma.line_list_nu[cur_line_id - 1],
numba_plasma,
)
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 trace_vpacket_volley(r_packet, vpacket_collection, numba_model,
numba_plasma):
"""
Shoot a volley of vpackets (the vpacket collection specifies how many)
from the current position of the rpacket.
Parameters
----------
r_packet : [type]
[description]
vpacket_collection : [type]
[description]
numba_model : [type]
[description]
numba_plasma : [type]
[description]
"""
if (r_packet.nu < vpacket_collection.v_packet_spawn_start_frequency) or (
r_packet.nu > vpacket_collection.v_packet_spawn_end_frequency):
return
no_of_vpackets = vpacket_collection.number_of_vpackets
if no_of_vpackets == 0:
return
### TODO theoretical check for r_packet nu within vpackets bins - is done somewhere else I think
if r_packet.r > numba_model.r_inner[0]: # not on inner_boundary
r_inner_over_r = numba_model.r_inner[0] / r_packet.r
mu_min = -math.sqrt(1 - r_inner_over_r * r_inner_over_r)
v_packet_on_inner_boundary = False
if montecarlo_configuration.full_relativity:
mu_min = angle_aberration_LF_to_CMF(r_packet,
numba_model.time_explosion,
mu_min)
else:
v_packet_on_inner_boundary = True
mu_min = 0.0
mu_bin = (1.0 - mu_min) / no_of_vpackets
r_packet_doppler_factor = get_doppler_factor(r_packet.r, r_packet.mu,
numba_model.time_explosion)
for i in range(no_of_vpackets):
v_packet_mu = mu_min + i * mu_bin + np.random.random() * mu_bin
if v_packet_on_inner_boundary: # The weights are described in K&S 2014
weight = 2 * v_packet_mu / no_of_vpackets
else:
weight = (1 - mu_min) / (2 * no_of_vpackets)
# C code: next line, angle_aberration_CMF_to_LF( & virt_packet, storage);
if montecarlo_configuration.full_relativity:
v_packet_mu = angle_aberration_CMF_to_LF(
r_packet, numba_model.time_explosion, v_packet_mu)
v_packet_doppler_factor = get_doppler_factor(
r_packet.r, v_packet_mu, numba_model.time_explosion)
# transform between r_packet mu and v_packet_mu
doppler_factor_ratio = r_packet_doppler_factor / v_packet_doppler_factor
v_packet_nu = r_packet.nu * doppler_factor_ratio
v_packet_energy = r_packet.energy * weight * doppler_factor_ratio
v_packet = VPacket(
r_packet.r,
v_packet_mu,
v_packet_nu,
v_packet_energy,
r_packet.current_shell_id,
r_packet.next_line_id,
i,
r_packet.is_close_line,
)
if r_packet.next_line_id <= (len(numba_plasma.line_list_nu) - 1):
test_for_close_line(
v_packet,
r_packet.next_line_id,
numba_plasma.line_list_nu[r_packet.next_line_id - 1],
numba_plasma,
)
tau_vpacket = trace_vpacket(v_packet, numba_model, numba_plasma)
v_packet.energy *= math.exp(-tau_vpacket)
vpacket_collection.set_properties(
v_packet.nu,
v_packet.energy,
r_packet.last_interaction_in_nu,
r_packet.last_interaction_type,
r_packet.last_line_interaction_in_id,
r_packet.last_line_interaction_out_id,
)
