int64_t montecarlo_one_packet(storage_model_t *storage, rpacket_t *packet, int64_t virtual_mode)
{
int64_t i;
rpacket_t virt_packet;
double mu_bin;
double mu_min;
double doppler_factor_ratio;
double weight;
int64_t virt_id_nu;
if (virtual_mode == 0)
{
montecarlo_one_packet_loop(storage, packet, 0);
}
else
{
for (i = 0; i < rpacket_get_virtual_packet_flag(packet); i++)
{
memcpy((void *)&virt_packet, (void *)packet, sizeof(rpacket_t));
if (virt_packet.r > storage->r_inner[0])
{
mu_min = -1.0 * sqrt(1.0 - (storage->r_inner[0] / virt_packet.r) *
(storage->r_inner[0] / virt_packet.r));
}
else
{
mu_min = 0.0;
}
mu_bin = (1.0 - mu_min) / rpacket_get_virtual_packet_flag(packet);
virt_packet.mu = mu_min + (i + rk_double(&mt_state)) * mu_bin;
switch(virtual_mode)
{
case -2:
weight = 1.0 / rpacket_get_virtual_packet_flag(packet);
break;
case -1:
weight = 2.0 * virt_packet.mu / rpacket_get_virtual_packet_flag(packet);
break;
case 1:
weight = (1.0 - mu_min) / 2.0 / rpacket_get_virtual_packet_flag(packet);
break;
default:
fprintf(stderr, "Something has gone horribly wrong!\n");
}
doppler_factor_ratio =
rpacket_doppler_factor(packet, storage) /
rpacket_doppler_factor(&virt_packet, storage);
virt_packet.energy = rpacket_get_energy(packet) * doppler_factor_ratio;
virt_packet.nu = rpacket_get_nu(packet) * doppler_factor_ratio;
montecarlo_one_packet_loop(storage, &virt_packet, 1);
if ((virt_packet.nu < storage->spectrum_end_nu) &&
(virt_packet.nu > storage->spectrum_start_nu))
{
virt_id_nu = floor((virt_packet.nu - storage->spectrum_start_nu) /
storage->spectrum_delta_nu);
storage->spectrum_virt_nu[virt_id_nu] += virt_packet.energy * weight;
}
}
}
}
double
test_calculate_chi_bf(void){
rpacket_t rp;
storage_model_t sm;
init_rpacket(&rp);
init_storage_model(&sm);
rk_state mt_state;
irandom(&mt_state);
int64_t j_blue_idx = 0;
double D_BOUNDARY = compute_distance2boundary(&rp, &sm);
rpacket_set_d_boundary(&rp, D_BOUNDARY);
double D_LINE;
// FIXME MR: return status of compute_distance2line() is ignored
compute_distance2line(&rp, &sm, &D_LINE);
rpacket_set_d_line(&rp, D_LINE);
move_packet(&rp, &sm, 1e13);
double d_line = rpacket_get_d_line(&rp);
increment_j_blue_estimator(&rp, &sm, d_line, j_blue_idx);
double DISTANCE = 1e13;
montecarlo_line_scatter(&rp, &sm, DISTANCE, &mt_state);
DISTANCE = 0.95e13;
// MR: wrong: move_packet_across_shell_boundary() returns void
move_packet_across_shell_boundary(&rp, &sm, DISTANCE, &mt_state);
montecarlo_one_packet(&sm, &rp, 1, &mt_state);
montecarlo_one_packet_loop(&sm, &rp, 1, &mt_state);
DISTANCE = 1e13;
montecarlo_thomson_scatter(&rp, &sm, DISTANCE, &mt_state);
calculate_chi_bf(&rp, &sm);
double res = rpacket_doppler_factor (&rp, &sm);
dealloc_storage_model(&sm);
return res;
}
int64_t
test_montecarlo_one_packet_loop(void){
rpacket_t rp;
storage_model_t sm;
init_rpacket(&rp);
init_storage_model(&sm);
rk_state mt_state;
irandom(&mt_state);
int64_t res= montecarlo_one_packet_loop(&sm, &rp, 1, &mt_state);
dealloc_storage_model(&sm);
return res;
}
int64_t
montecarlo_one_packet (storage_model_t * storage, rpacket_t * packet,
int64_t virtual_mode, rk_state *mt_state)
{
int64_t reabsorbed=-1;
if (virtual_mode == 0)
{
reabsorbed = montecarlo_one_packet_loop (storage, packet, 0, mt_state);
}
else
{
if ((rpacket_get_nu (packet) > storage->spectrum_virt_start_nu) && (rpacket_get_nu(packet) < storage->spectrum_virt_end_nu))
{
for (int64_t i = 0; i < rpacket_get_virtual_packet_flag (packet); i++)
{
double weight;
rpacket_t virt_packet = *packet;
double mu_min;
if (rpacket_get_r(&virt_packet) > storage->r_inner[0])
{
mu_min =
-1.0 * sqrt (1.0 -
(storage->r_inner[0] / rpacket_get_r(&virt_packet)) *
(storage->r_inner[0] / rpacket_get_r(&virt_packet)));
}
else
{
mu_min = 0.0;
}
double mu_bin = (1.0 - mu_min) / rpacket_get_virtual_packet_flag (packet);
rpacket_set_mu(&virt_packet,mu_min + (i + rk_double (mt_state)) * mu_bin);
switch (virtual_mode)
{
case -2:
weight = 1.0 / rpacket_get_virtual_packet_flag (packet);
break;
case -1:
weight =
2.0 * rpacket_get_mu(&virt_packet) /
rpacket_get_virtual_packet_flag (packet);
break;
case 1:
weight =
(1.0 -
mu_min) / 2.0 / rpacket_get_virtual_packet_flag (packet);
break;
default:
fprintf (stderr, "Something has gone horribly wrong!\n");
// FIXME MR: we need to somehow signal an error here
// I'm adding an exit() here to inform the compiler about the impossible path
exit(1);
}
double doppler_factor_ratio =
rpacket_doppler_factor (packet, storage) /
rpacket_doppler_factor (&virt_packet, storage);
rpacket_set_energy(&virt_packet,
rpacket_get_energy (packet) * doppler_factor_ratio);
rpacket_set_nu(&virt_packet,rpacket_get_nu (packet) * doppler_factor_ratio);
reabsorbed = montecarlo_one_packet_loop (storage, &virt_packet, 1, mt_state);
#ifdef WITH_VPACKET_LOGGING
#ifdef WITHOPENMP
#pragma omp critical
{
#endif // WITHOPENMP
if (storage->virt_packet_count >= storage->virt_array_size)
{
storage->virt_array_size *= 2;
storage->virt_packet_nus = safe_realloc(storage->virt_packet_nus, sizeof(double) * storage->virt_array_size);
storage->virt_packet_energies = safe_realloc(storage->virt_packet_energies, sizeof(double) * storage->virt_array_size);
storage->virt_packet_last_interaction_in_nu = safe_realloc(storage->virt_packet_last_interaction_in_nu, sizeof(double) * storage->virt_array_size);
storage->virt_packet_last_interaction_type = safe_realloc(storage->virt_packet_last_interaction_type, sizeof(int64_t) * storage->virt_array_size);
storage->virt_packet_last_line_interaction_in_id = safe_realloc(storage->virt_packet_last_line_interaction_in_id, sizeof(int64_t) * storage->virt_array_size);
storage->virt_packet_last_line_interaction_out_id = safe_realloc(storage->virt_packet_last_line_interaction_out_id, sizeof(int64_t) * storage->virt_array_size);
}
storage->virt_packet_nus[storage->virt_packet_count] = rpacket_get_nu(&virt_packet);
storage->virt_packet_energies[storage->virt_packet_count] = rpacket_get_energy(&virt_packet) * weight;
storage->virt_packet_last_interaction_in_nu[storage->virt_packet_count] = storage->last_interaction_in_nu[rpacket_get_id (packet)];
storage->virt_packet_last_interaction_type[storage->virt_packet_count] = storage->last_interaction_type[rpacket_get_id (packet)];
storage->virt_packet_last_line_interaction_in_id[storage->virt_packet_count] = storage->last_line_interaction_in_id[rpacket_get_id (packet)];
storage->virt_packet_last_line_interaction_out_id[storage->virt_packet_count] = storage->last_line_interaction_out_id[rpacket_get_id (packet)];
storage->virt_packet_count += 1;
#ifdef WITHOPENMP
}
#endif // WITHOPENMP
#endif // WITH_VPACKET_LOGGING
if ((rpacket_get_nu(&virt_packet) < storage->spectrum_end_nu) &&
(rpacket_get_nu(&virt_packet) > storage->spectrum_start_nu))
{
#ifdef WITHOPENMP
#pragma omp critical
{
#endif // WITHOPENMP
int64_t virt_id_nu =
floor ((rpacket_get_nu(&virt_packet) -
storage->spectrum_start_nu) /
storage->spectrum_delta_nu);
storage->spectrum_virt_nu[virt_id_nu] +=
rpacket_get_energy(&virt_packet) * weight;
#ifdef WITHOPENMP
}
#endif // WITHOPENMP
}
}
}
else
{
return 1;
}
}
return reabsorbed;
}
