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 "../Efield/E_2DS.h"
#include "../print.h"
#include "hdf5_helpers.h"
#include "hdf5_efield.h"
 
#define EPATH 
 
int hdf5_efield_read_2DS(hid_t f, E_2DS_data* data, char* qid);
int hdf5_efield_read_1DS(hid_t f, E_1DS_data* data, char* qid);
int hdf5_efield_read_TC(hid_t f, E_TC_data* data, char* qid);

int hdf5_efield_init(hid_t f, E_field_data* data, 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) ) {
        data->type = E_field_type_TC;
        err = hdf5_efield_read_TC(f, &data->ETC, qid);
    }
    hdf5_gen_path("/efield/E_1DS_XXXXXXXXXX", qid, path);
    if(hdf5_find_group(f, path) == 0) {
        data->type = E_field_type_1DS;
        err = hdf5_efield_read_1DS(f, &data->E1DS, qid);
    }
    hdf5_gen_path("/efield/E_2DS_XXXXXXXXXX", qid, path);
    if(hdf5_find_group(f, path) == 0) {
        data->type = E_field_type_2DS;
        err = hdf5_efield_read_2DS(f, &data->E2DS, qid);
    }
    return err;
}

int hdf5_efield_read_2DS(hid_t f, E_2DS_data* data, char* qid) {
    #undef EPATH
    #define EPATH "/efield/E_2DS_XXXXXXXXXX/"
 
    int nr, nz;
    real rmin, rmax, zmin, zmax;
    if( hdf5_read_int(EPATH "nr", &nr,
                      f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(EPATH "rmin", &rmin,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(EPATH "rmax", &rmax,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_int(EPATH "nz", &nz,
                      f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(EPATH "zmin", &zmin,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(EPATH "zmax", &zmax,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
 
    real* vpot = (real*) malloc( nr*nz*sizeof(real) );
    if( hdf5_read_double(EPATH "vpot", vpot,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    int err = E_2DS_init(data, nr, rmin, rmax, nz, zmin, zmax, vpot);
    free(vpot);
    return err;
}

int hdf5_efield_read_1DS(hid_t f, E_1DS_data* data, char* qid) {
    #undef EPATH
    #define EPATH "/efield/E_1DS_XXXXXXXXXX/"
 
    int nrho;
    real rhomin, rhomax, reff;
    if( hdf5_read_int(EPATH "nrho", &nrho,
                      f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(EPATH "rhomin", &rhomin,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(EPATH "rhomax", &rhomax,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(EPATH "reff", &reff,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
 
    real* dvdrho = (real*) malloc( nrho*sizeof(real) );
    if( hdf5_read_double(EPATH "dvdrho", dvdrho,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    int err = E_1DS_init(data, nrho, rhomin, rhomax, reff, dvdrho);
    free(dvdrho);
    return err;
}

int hdf5_efield_read_TC(hid_t f, E_TC_data* data, char* qid) {
    #undef EPATH
    #define EPATH "/efield/E_TC_XXXXXXXXXX/"
 
    real exyz[3];
    if( hdf5_read_double(EPATH "exyz", exyz,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    int err = E_TC_init(data, exyz);
    return err;
}