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