hdf5_wall.c

Go to the documentation of this file.

 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <hdf5.h>
#include <hdf5_hl.h>
#include "../wall.h"
#include "../wall/wall_2d.h"
#include "../wall/wall_3d.h"
#include "hdf5_wall.h"
#include "hdf5_helpers.h"
 
#define WPATH 
int hdf5_wall_read_2D(hid_t f, wall_2d_offload_data* offload_data,
                      real** offload_array, char* qid);
int hdf5_wall_read_3D(hid_t f, wall_3d_offload_data* offload_data,
                      real** offload_array, char* qid);
 
int hdf5_wall_init_offload(hid_t f, wall_offload_data* offload_data,
                           real** offload_array, int** int_offload_array,
                           char* qid) {
 
    char path[256];
    int err = 1;
 
    /* Read data the QID corresponds to */
 
    hdf5_gen_path("/wall/wall_2D_XXXXXXXXXX", qid, path);
    if(hdf5_find_group(f, path) == 0) {
        offload_data->type = wall_type_2D;
        err = hdf5_wall_read_2D(f, &(offload_data->w2d),
                                offload_array, qid);
    }
 
    hdf5_gen_path("/wall/wall_3D_XXXXXXXXXX", qid, path);
    if(hdf5_find_group(f, path) == 0) {
        offload_data->type = wall_type_3D;
        err = hdf5_wall_read_3D(f, &(offload_data->w3d),
                                offload_array, qid);
    }
 
    /* Initialize if data was read succesfully */
    if(!err) {
        err = wall_init_offload(offload_data, offload_array, int_offload_array);
    }
 
    return err;
}
 
int hdf5_wall_read_2D(hid_t f, wall_2d_offload_data* offload_data,
                      real** offload_array, char* qid) {
    #undef WPATH
    #define WPATH "/wall/wall_2D_XXXXXXXXXX/"
 
    int nelements;
 
    /* Read number of wall elements and allocate offload array */
    if( hdf5_read_int(WPATH "nelements", &nelements,
                      f, qid, __FILE__, __LINE__) ) {return 1;}
 
    offload_data->n = nelements;
    offload_data->offload_array_length = 2 * offload_data->n;
    *offload_array = (real*) malloc(2 * offload_data->n * sizeof(real));
 
    /* Read the wall polygon */
    if( hdf5_read_double(WPATH "r", &((*offload_array)[0]),
        f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(WPATH "z", &((*offload_array)[nelements]),
        f, qid, __FILE__, __LINE__) ) {return 1;}
 
    return 0;
}
 
int hdf5_wall_read_3D(hid_t f, wall_3d_offload_data* offload_data,
                      real** offload_array, char* qid) {
    #undef WPATH
    #define WPATH "/wall/wall_3D_XXXXXXXXXX/"
 
    int nelements;
 
    /* Read number of wall elements and allocate offload array to
       store n 3D triangles */
    if( hdf5_read_int(WPATH "nelements", &nelements,
                      f, qid, __FILE__, __LINE__) ) {return 1;}
    offload_data->n = nelements;
 
    /* Allocate temporary arrays for x1x2x3, y1y2y3, z1z2z3 for each triangle */
    real* x1x2x3 = (real*)malloc(3 * nelements * sizeof(real));
    real* y1y2y3 = (real*)malloc(3 * nelements * sizeof(real));
    real* z1z2z3 = (real*)malloc(3 * nelements * sizeof(real));
 
    if( hdf5_read_double(WPATH "x1x2x3", x1x2x3,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(WPATH "y1y2y3", y1y2y3,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
    if( hdf5_read_double(WPATH "z1z2z3", z1z2z3,
                         f, qid, __FILE__, __LINE__) ) {return 1;}
 
    /* The data in the offload array is to be in the format
     *  [x1 y1 z1 x2 y2 z2 x3 y3 z3; ... ]
     */
    *offload_array = (real*)malloc(9 * nelements * sizeof(real));
    offload_data->offload_array_length = 9 * nelements;
    for(int i = 0; i < nelements; i++) {
        for(int j = 0; j < 3; j++) {
            (*offload_array)[i*9 + j*3 + 0] = x1x2x3[3*i+j];
            (*offload_array)[i*9 + j*3 + 1] = y1y2y3[3*i+j];
            (*offload_array)[i*9 + j*3 + 2] = z1z2z3[3*i+j];
        }
    }
    free(x1x2x3);
    free(y1y2y3);
    free(z1z2z3);
    return 0;
}