hdf5_efield.c

Go to the documentation of this file.

 
#include <stdio.h>
#include <stdlib.h>
#include <hdf5.h>
#include <hdf5_hl.h>
#include "../ascot5.h"
#include "../E_field.h"
#include "../Efield/E_TC.h"
#include "../Efield/E_1DS.h"
#include "../print.h"
#include "hdf5_helpers.h"
#include "hdf5_efield.h"
 
#define EPATH 
int hdf5_efield_read_1DS(hid_t f, E_1DS_offload_data* offload_data,
                         real** offload_array, char* qid);
int hdf5_efield_read_TC(hid_t f, E_TC_offload_data* offload_data,
                        real** offload_array, char* qid);
 
int hdf5_efield_init_offload(hid_t f, E_field_offload_data* offload_data,
                             real** offload_array, char* qid) {
    char path[256];
    int err = 1;
 
    /* Read data the QID corresponds to */
 
    hdf5_gen_path("/efield/E_TC_XXXXXXXXXX", qid, path);
    if( !hdf5_find_group(f, path) ) {
        offload_data->type = E_field_type_TC;
        err = hdf5_efield_read_TC(f, &(offload_data->ETC),
                                  offload_array, qid);
    }
 
    hdf5_gen_path("/efield/E_1DS_XXXXXXXXXX", qid, path);
    if(hdf5_find_group(f, path) == 0) {
        offload_data->type = E_field_type_1DS;
        err = hdf5_efield_read_1DS(f, &(offload_data->E1DS),
                                   offload_array, qid);
    }
 
    /* Initialize if data was read succesfully */
    if(!err) {
        err = E_field_init_offload(offload_data, offload_array);
    }
 
    return err;
}
 
int hdf5_efield_read_1DS(hid_t f, E_1DS_offload_data* offload_data,
                         real** offload_array, char* qid) {
    #undef EPATH
    #define EPATH "/efield/E_1DS_XXXXXXXXXX/"
 
    if( hdf5_read_int(EPATH "nrho", &(offload_data->n_rho),
                      f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(EPATH "rhomin", &(offload_data->rho_min),
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(EPATH "rhomax", &(offload_data->rho_max),
                         f, qid, __FILE__, __LINE__) ) {return 1;}
 
    /* Allocate n_rho space for dV/drho */
    *offload_array = (real*) malloc(offload_data->n_rho*sizeof(real));
 
    if( hdf5_read_double(EPATH "dvdrho", *offload_array,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
 
    /* Effective minor radius */
    real r_eff;
    if( hdf5_read_double(EPATH "reff", &(r_eff),
                         f, qid, __FILE__, __LINE__) ) {return 1;}
 
    /* Scale derivatives by effective minor radius */
    for(int i = 0; i < offload_data->n_rho; i++) {
        (*offload_array)[i] = r_eff * (*offload_array)[i];
    }
 
    return 0;
}
 
int hdf5_efield_read_TC(hid_t f, E_TC_offload_data* offload_data,
                        real** offload_array, char* qid) {
    #undef EPATH
    #define EPATH "/efield/E_TC_XXXXXXXXXX/"
 
    *offload_array = NULL;
 
    if( hdf5_read_double(EPATH "exyz", offload_data->Exyz,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
 
    return 0;
}