Merge pull request #530 from karandesai-96/test-cmontecarlo

Refactor cmontecarlo testing framework
tardis-sn · Apr 11, 2016 · 5ddca21 · 5ddca21
2 parents 3370d8d + 8488ccb
commit 5ddca21
Show file tree

Hide file tree

Showing 2 changed files with 411 additions and 65 deletions.
diff --git a/tardis/montecarlo/struct.py b/tardis/montecarlo/struct.py
@@ -0,0 +1,105 @@
+from ctypes import Structure, POINTER, c_int, c_int64, c_double, c_ulong
+
+
+class RPacket(Structure):
+    _fields_ = [
+        ('nu', c_double),
+        ('mu', c_double),
+        ('energy', c_double),
+        ('r', c_double),
+        ('tau_event', c_double),
+        ('nu_line', c_double),
+        ('current_shell_id', c_int64),
+        ('next_line_id', c_int64),
+        ('last_line', c_int64),
+        ('close_line', c_int64),
+        ('current_continuum_id', c_int64),
+        ('virtual_packet_flag', c_int64),
+        ('virtual_packet', c_int64),
+        ('d_line', c_double),
+        ('d_electron', c_double),
+        ('d_boundary', c_double),
+        ('d_cont', c_double),
+        ('next_shell_id', c_int64),
+        ('status', c_int),
+        ('id', c_int64),
+        ('chi_th', c_double),
+        ('chi_cont', c_double),
+        ('chi_ff', c_double),
+        ('chi_bf', c_double)
+    ]
+
+
+class StorageModel(Structure):
+    _fields_ = [
+        ('packet_nus', POINTER(c_double)),
+        ('packet_mus', POINTER(c_double)),
+        ('packet_energies', POINTER(c_double)),
+        ('output_nus', POINTER(c_double)),
+        ('output_energies', POINTER(c_double)),
+        ('last_interaction_in_nu', POINTER(c_double)),
+        ('last_line_interaction_in_id', POINTER(c_int64)),
+        ('last_line_interaction_out_id', POINTER(c_int64)),
+        ('last_line_interaction_shell_id', POINTER(c_int64)),
+        ('last_line_interaction_type', POINTER(c_int64)),
+        ('no_of_packets', c_int64),
+        ('no_of_shells', c_int64),
+        ('r_inner', POINTER(c_double)),
+        ('r_outer', POINTER(c_double)),
+        ('v_inner', POINTER(c_double)),
+        ('time_explosion', c_double),
+        ('inverse_time_explosion', c_double),
+        ('electron_densities', POINTER(c_double)),
+        ('inverse_electron_densities', POINTER(c_double)),
+        ('line_list_nu', POINTER(c_double)),
+        ('continuum_list_nu', POINTER(c_double)),
+        ('line_lists_tau_sobolevs', POINTER(c_double)),
+        ('line_lists_tau_sobolevs_nd', c_int64),
+        ('line_lists_j_blues', POINTER(c_double)),
+        ('line_lists_j_blues_nd', c_int64),
+        ('no_of_lines', c_int64),
+        ('no_of_edges', c_int64),
+        ('line_interaction_id', c_int64),
+        ('transition_probabilities', POINTER(c_double)),
+        ('transition_probabilities_nd', c_int64),
+        ('line2macro_level_upper', POINTER(c_int64)),
+        ('macro_block_references', POINTER(c_int64)),
+        ('transition_type', POINTER(c_int64)),
+        ('destination_level_id', POINTER(c_int64)),
+        ('transition_line_id', POINTER(c_int64)),
+        ('js', POINTER(c_double)),
+        ('nubars', POINTER(c_double)),
+        ('spectrum_start_nu', c_double),
+        ('spectrum_delta_nu', c_double),
+        ('spectrum_end_nu', c_double),
+        ('spectrum_virt_start_nu', c_double),
+        ('spectrum_virt_end_nu', c_double),
+        ('spectrum_virt_nu', POINTER(c_double)),
+        ('sigma_thomson', c_double),
+        ('inverse_sigma_thomson', c_double),
+        ('inner_boundary_albedo', c_double),
+        ('reflective_inner_boundary', c_int64),
+        ('current_packet_id', c_int64),
+        ('chi_bf_tmp_partial', POINTER(c_double)),
+        ('t_electrons', POINTER(c_double)),
+        ('l_pop', POINTER(c_double)),
+        ('l_pop_r', POINTER(c_double)),
+        ('cont_status', c_int),
+        ('virt_packet_nus', POINTER(c_double)),
+        ('virt_packet_energies', POINTER(c_double)),
+        ('virt_packet_last_interaction_in_nu', POINTER(c_double)),
+        ('virt_packet_last_interaction_type', POINTER(c_int64)),
+        ('virt_packet_last_line_interaction_in_id', POINTER(c_int64)),
+        ('virt_packet_last_line_interaction_out_id', POINTER(c_int64)),
+        ('virt_packet_count', c_int64),
+        ('virt_array_size', c_int64)
+    ]
+
+
+class RKState(Structure):
+    _fields_ = [
+        ('key', POINTER(c_ulong)),
+        ('pos', c_int),
+        ('has_gauss', c_int),
+        ('gauss', c_double)
+    ]