int benchmark(int num_stations, int num_sources, int type,
int jones_type, int loc, int use_extended, int use_time_ave, int niter,
std::vector<double>& times)
{
int status = 0;
oskar_Timer* timer;
timer = oskar_timer_create(loc == OSKAR_GPU ?
OSKAR_TIMER_CUDA : OSKAR_TIMER_OMP);
// Set up a test sky model, telescope model and Jones matrices.
oskar_Telescope* tel = oskar_telescope_create(type, loc,
num_stations, &status);
oskar_Sky* sky = oskar_sky_create(type, loc, num_sources, &status);
oskar_Jones* J = oskar_jones_create(jones_type, loc, num_stations,
num_sources, &status);
oskar_telescope_set_channel_bandwidth(tel, 1e6);
oskar_telescope_set_time_average(tel, (double) use_time_ave);
oskar_sky_set_use_extended(sky, use_extended);
// Memory for visibility coordinates and output visibility slice.
oskar_Mem *vis, *u, *v, *w;
vis = oskar_mem_create(jones_type, loc, oskar_telescope_num_baselines(tel),
&status);
u = oskar_mem_create(type, loc, num_stations, &status);
v = oskar_mem_create(type, loc, num_stations, &status);
w = oskar_mem_create(type, loc, num_stations, &status);
// Run benchmark.
times.resize(niter);
for (int i = 0; i < niter; ++i)
{
oskar_timer_start(timer);
oskar_cross_correlate(vis, oskar_sky_num_sources(sky), J, sky, tel, u, v, w,
0.0, 100e6, &status);
times[i] = oskar_timer_elapsed(timer);
}
// Free memory.
oskar_mem_free(u, &status);
oskar_mem_free(v, &status);
oskar_mem_free(w, &status);
oskar_mem_free(vis, &status);
oskar_jones_free(J, &status);
oskar_telescope_free(tel, &status);
oskar_sky_free(sky, &status);
oskar_timer_free(timer);
return status;
}
void createTestData(int precision, int location, int matrix)
{
int status = 0, type;
// Allocate memory for data structures.
type = precision | OSKAR_COMPLEX;
if (matrix) type |= OSKAR_MATRIX;
jones = oskar_jones_create(type, location, num_stations, num_sources,
&status);
u_ = oskar_mem_create(precision, location, num_stations, &status);
v_ = oskar_mem_create(precision, location, num_stations, &status);
w_ = oskar_mem_create(precision, location, num_stations, &status);
sky = oskar_sky_create(precision, location, num_sources, &status);
tel = oskar_telescope_create(precision, location,
num_stations, &status);
ASSERT_EQ(0, status) << oskar_get_error_string(status);
// Fill data structures with random data in sensible ranges.
srand(2);
oskar_mem_random_range(oskar_jones_mem(jones), 1.0, 5.0, &status);
oskar_mem_random_range(u_, 1.0, 5.0, &status);
oskar_mem_random_range(v_, 1.0, 5.0, &status);
oskar_mem_random_range(w_, 1.0, 5.0, &status);
oskar_mem_random_range(
oskar_telescope_station_true_x_offset_ecef_metres(tel),
0.1, 1000.0, &status);
oskar_mem_random_range(
oskar_telescope_station_true_y_offset_ecef_metres(tel),
0.1, 1000.0, &status);
oskar_mem_random_range(
oskar_telescope_station_true_z_offset_ecef_metres(tel),
0.1, 1000.0, &status);
oskar_mem_random_range(oskar_sky_I(sky), 1.0, 2.0, &status);
oskar_mem_random_range(oskar_sky_Q(sky), 0.1, 1.0, &status);
oskar_mem_random_range(oskar_sky_U(sky), 0.1, 0.5, &status);
oskar_mem_random_range(oskar_sky_V(sky), 0.1, 0.2, &status);
oskar_mem_random_range(oskar_sky_l(sky), 0.1, 0.9, &status);
oskar_mem_random_range(oskar_sky_m(sky), 0.1, 0.9, &status);
oskar_mem_random_range(oskar_sky_n(sky), 0.1, 0.9, &status);
oskar_mem_random_range(oskar_sky_gaussian_a(sky), 0.1e-6, 0.2e-6,
&status);
oskar_mem_random_range(oskar_sky_gaussian_b(sky), 0.1e-6, 0.2e-6,
&status);
oskar_mem_random_range(oskar_sky_gaussian_c(sky), 0.1e-6, 0.2e-6,
&status);
ASSERT_EQ(0, status) << oskar_get_error_string(status);
}
oskar_Telescope* oskar_telescope_create_copy(const oskar_Telescope* src,
int location, int* status)
{
int i = 0;
oskar_Telescope* telescope;
/* Create a new, empty model. */
telescope = oskar_telescope_create(oskar_telescope_precision(src),
location, 0, status);
/* Copy private meta-data. */
telescope->precision = src->precision;
telescope->mem_location = location;
/* Copy the meta-data. */
telescope->pol_mode = src->pol_mode;
telescope->num_stations = src->num_stations;
telescope->max_station_size = src->max_station_size;
telescope->max_station_depth = src->max_station_depth;
telescope->identical_stations = src->identical_stations;
telescope->allow_station_beam_duplication = src->allow_station_beam_duplication;
telescope->enable_numerical_patterns = src->enable_numerical_patterns;
telescope->lon_rad = src->lon_rad;
telescope->lat_rad = src->lat_rad;
telescope->alt_metres = src->alt_metres;
telescope->pm_x_rad = src->pm_x_rad;
telescope->pm_y_rad = src->pm_y_rad;
telescope->phase_centre_coord_type = src->phase_centre_coord_type;
telescope->phase_centre_ra_rad = src->phase_centre_ra_rad;
telescope->phase_centre_dec_rad = src->phase_centre_dec_rad;
telescope->channel_bandwidth_hz = src->channel_bandwidth_hz;
telescope->time_average_sec = src->time_average_sec;
telescope->uv_filter_min = src->uv_filter_min;
telescope->uv_filter_max = src->uv_filter_max;
telescope->uv_filter_units = src->uv_filter_units;
telescope->noise_enabled = src->noise_enabled;
telescope->noise_seed = src->noise_seed;
/* Copy the coordinates. */
oskar_mem_copy(telescope->station_true_x_offset_ecef_metres,
src->station_true_x_offset_ecef_metres, status);
oskar_mem_copy(telescope->station_true_y_offset_ecef_metres,
src->station_true_y_offset_ecef_metres, status);
oskar_mem_copy(telescope->station_true_z_offset_ecef_metres,
src->station_true_z_offset_ecef_metres, status);
oskar_mem_copy(telescope->station_true_x_enu_metres,
src->station_true_x_enu_metres, status);
oskar_mem_copy(telescope->station_true_y_enu_metres,
src->station_true_y_enu_metres, status);
oskar_mem_copy(telescope->station_true_z_enu_metres,
src->station_true_z_enu_metres, status);
oskar_mem_copy(telescope->station_measured_x_offset_ecef_metres,
src->station_measured_x_offset_ecef_metres, status);
oskar_mem_copy(telescope->station_measured_y_offset_ecef_metres,
src->station_measured_y_offset_ecef_metres, status);
oskar_mem_copy(telescope->station_measured_z_offset_ecef_metres,
src->station_measured_z_offset_ecef_metres, status);
oskar_mem_copy(telescope->station_measured_x_enu_metres,
src->station_measured_x_enu_metres, status);
oskar_mem_copy(telescope->station_measured_y_enu_metres,
src->station_measured_y_enu_metres, status);
oskar_mem_copy(telescope->station_measured_z_enu_metres,
src->station_measured_z_enu_metres, status);
/* Copy each station. */
telescope->station = malloc(src->num_stations * sizeof(oskar_Station*));
for (i = 0; i < src->num_stations; ++i)
{
telescope->station[i] = oskar_station_create_copy(
oskar_telescope_station_const(src, i), location, status);
}
/* Return pointer to data structure. */
return telescope;
}
