oskar_Imager* oskar_imager_create(int imager_precision, int* status)
{
oskar_Imager* h = 0;
h = (oskar_Imager*) calloc(1, sizeof(oskar_Imager));
/* Create timers. */
h->tmr_grid_finalise = oskar_timer_create(OSKAR_TIMER_NATIVE);
h->tmr_grid_update = oskar_timer_create(OSKAR_TIMER_NATIVE);
h->tmr_init = oskar_timer_create(OSKAR_TIMER_NATIVE);
h->tmr_read = oskar_timer_create(OSKAR_TIMER_NATIVE);
h->tmr_write = oskar_timer_create(OSKAR_TIMER_NATIVE);
h->mutex = oskar_mutex_create();
/* Create scratch arrays. */
h->imager_prec = imager_precision;
h->uu_im = oskar_mem_create(imager_precision, OSKAR_CPU, 0, status);
h->vv_im = oskar_mem_create(imager_precision, OSKAR_CPU, 0, status);
h->ww_im = oskar_mem_create(imager_precision, OSKAR_CPU, 0, status);
h->uu_tmp = oskar_mem_create(imager_precision, OSKAR_CPU, 0, status);
h->vv_tmp = oskar_mem_create(imager_precision, OSKAR_CPU, 0, status);
h->ww_tmp = oskar_mem_create(imager_precision, OSKAR_CPU, 0, status);
h->vis_im = oskar_mem_create(imager_precision | OSKAR_COMPLEX,
OSKAR_CPU, 0, status);
h->weight_im = oskar_mem_create(imager_precision, OSKAR_CPU, 0, status);
h->weight_tmp = oskar_mem_create(imager_precision, OSKAR_CPU, 0, status);
h->time_im = oskar_mem_create(OSKAR_DOUBLE, OSKAR_CPU, 0, status);
/* Check data type. */
if (imager_precision != OSKAR_SINGLE && imager_precision != OSKAR_DOUBLE)
{
*status = OSKAR_ERR_BAD_DATA_TYPE;
return h;
}
/* Get number of devices available, and device location. */
oskar_device_set_require_double_precision(imager_precision == OSKAR_DOUBLE);
h->num_gpus_avail = oskar_device_count(0, &h->dev_loc);
/* Set sensible defaults. */
oskar_imager_set_gpus(h, -1, 0, status);
oskar_imager_set_num_devices(h, -1);
oskar_imager_set_algorithm(h, "FFT", status);
oskar_imager_set_image_type(h, "I", status);
oskar_imager_set_weighting(h, "Natural", status);
oskar_imager_set_ms_column(h, "DATA", status);
oskar_imager_set_default_direction(h);
oskar_imager_set_generate_w_kernels_on_gpu(h, 1);
oskar_imager_set_fov(h, 1.0);
oskar_imager_set_size(h, 256, status);
oskar_imager_set_uv_filter_max(h, DBL_MAX);
return h;
}
void oskar_simulator_set_gpus(oskar_Simulator* h, int num,
int* ids, int* status)
{
int i, num_gpus_avail;
if (*status) return;
free_device_data(h, status);
num_gpus_avail = oskar_device_count(status);
if (*status) return;
if (num < 0)
{
h->num_gpus = num_gpus_avail;
h->gpu_ids = (int*) realloc(h->gpu_ids, h->num_gpus * sizeof(int));
for (i = 0; i < h->num_gpus; ++i)
h->gpu_ids[i] = i;
}
else if (num > 0)
{
if (num > num_gpus_avail)
{
oskar_log_error(h->log, "More GPUs were requested than found.");
*status = OSKAR_ERR_COMPUTE_DEVICES;
return;
}
h->num_gpus = num;
h->gpu_ids = (int*) realloc(h->gpu_ids, h->num_gpus * sizeof(int));
for (i = 0; i < h->num_gpus; ++i)
h->gpu_ids[i] = ids[i];
}
else /* num == 0 */
{
free(h->gpu_ids);
h->gpu_ids = 0;
h->num_gpus = 0;
}
for (i = 0; i < h->num_gpus; ++i)
{
oskar_device_set(h->gpu_ids[i], status);
if (*status) return;
}
}
void oskar_imager_set_gpus(oskar_Imager* h, int num, const int* ids,
int* status)
{
int i, num_gpus_avail;
if (*status) return;
oskar_imager_free_device_data(h, status);
num_gpus_avail = oskar_device_count(status);
if (*status) return;
if (num < 0)
{
h->num_gpus = num_gpus_avail;
h->gpu_ids = (int*) realloc(h->gpu_ids, h->num_gpus * sizeof(int));
for (i = 0; i < h->num_gpus; ++i)
h->gpu_ids[i] = i;
}
else if (num > 0)
{
if (num > num_gpus_avail)
{
*status = OSKAR_ERR_COMPUTE_DEVICES;
return;
}
h->num_gpus = num;
h->gpu_ids = (int*) realloc(h->gpu_ids, h->num_gpus * sizeof(int));
for (i = 0; i < h->num_gpus; ++i)
h->gpu_ids[i] = ids[i];
}
else /* num == 0 */
{
free(h->gpu_ids);
h->gpu_ids = 0;
h->num_gpus = 0;
}
for (i = 0; i < h->num_gpus; ++i)
{
oskar_device_set(h->gpu_ids[i], status);
if (*status) return;
}
}
