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 thomson_scatter(r_packet, time_explosion):
"""
Thomson scattering — no longer line scattering
2) get the doppler factor at that position with the old angle
3) convert the current energy and nu into the comoving
frame with the old mu
4) Scatter and draw new mu - update mu
5) Transform the comoving energy and nu back using the new mu
Parameters
----------
r_packet : RPacket
time_explosion: float
time since explosion in seconds
"""
old_doppler_factor = get_doppler_factor(r_packet.r, r_packet.mu,
time_explosion)
comov_nu = r_packet.nu * old_doppler_factor
comov_energy = r_packet.energy * old_doppler_factor
r_packet.mu = get_random_mu()
inverse_new_doppler_factor = get_inverse_doppler_factor(
r_packet.r, r_packet.mu, time_explosion)
r_packet.nu = comov_nu * inverse_new_doppler_factor
r_packet.energy = comov_energy * inverse_new_doppler_factor
if montecarlo_configuration.full_relativity:
r_packet.mu = angle_aberration_CMF_to_LF(r_packet, time_explosion,
r_packet.mu)
def line_scatter(r_packet, time_explosion, line_interaction_type,
numba_plasma):
"""
Line scatter function that handles the scattering itself, including new angle drawn, and calculating nu out using macro atom
r_packet: RPacket
time_explosion: float
line_interaction_type: enum
numba_plasma: NumbaPlasma
"""
old_doppler_factor = get_doppler_factor(r_packet.r, r_packet.mu,
time_explosion)
r_packet.mu = get_random_mu()
inverse_new_doppler_factor = get_inverse_doppler_factor(
r_packet.r, r_packet.mu, time_explosion)
comov_energy = r_packet.energy * old_doppler_factor
r_packet.energy = comov_energy * inverse_new_doppler_factor
if line_interaction_type == LineInteractionType.SCATTER:
line_emission(r_packet, r_packet.next_line_id, time_explosion,
numba_plasma)
else: # includes both macro atom and downbranch - encoded in the transition probabilities
emission_line_id = macro_atom(r_packet, numba_plasma)
line_emission(r_packet, emission_line_id, time_explosion, numba_plasma)
def set_packet_props_partial_relativity(r_packet, numba_model):
inverse_doppler_factor = get_inverse_doppler_factor(
r_packet.r,
r_packet.mu,
numba_model.time_explosion,
)
r_packet.nu *= inverse_doppler_factor
r_packet.energy *= inverse_doppler_factor
def test_unphysical_inverse_doppler_factor(mu, r, inv_t_exp):
# Set the params from test cases here
# TODO: add relativity tests
time_explosion = 1 / inv_t_exp
# Perform any other setups just before this, they can be additional calls
# to other methods or introduction of some temporary variables
with pytest.raises(r_packet.SuperluminalError):
obtained = r_packet.get_inverse_doppler_factor(r, mu, time_explosion)
def set_packet_props_full_relativity(r_packet, numba_model):
beta = (r_packet.r / numba_model.time_explosion) / C_SPEED_OF_LIGHT
inverse_doppler_factor = get_inverse_doppler_factor(
r_packet.r,
r_packet.mu,
numba_model.time_explosion,
)
r_packet.nu *= inverse_doppler_factor
r_packet.energy *= inverse_doppler_factor
r_packet.mu = (r_packet.mu + beta) / (1 + beta * r_packet.mu)
def test_get_inverse_doppler_factor(mu, r, inv_t_exp, expected):
# Set the params from test cases here
# TODO: add relativity tests
time_explosion = 1 / inv_t_exp
# Perform any other setups just before this, they can be additional calls
# to other methods or introduction of some temporary variables
obtained = r_packet.get_inverse_doppler_factor(r, mu, time_explosion)
# Perform required assertions
assert_almost_equal(obtained, expected)
def test_frame_transformations(mu, r, inv_t_exp, full_relativity):
packet = r_packet.RPacket(r=r, mu=mu, energy=0.9, nu=0.4)
mc.full_relativity = bool(full_relativity)
mc.full_relativity = full_relativity
inverse_doppler_factor = r_packet.get_inverse_doppler_factor(
r, mu, 1 / inv_t_exp)
r_packet.angle_aberration_CMF_to_LF(packet, 1 / inv_t_exp, packet.mu)
doppler_factor = get_doppler_factor(r, mu, 1 / inv_t_exp)
mc.full_relativity = False
assert_almost_equal(doppler_factor * inverse_doppler_factor, 1.0)