hdf5orthoinfo.h

Go to the documentation of this file.

#ifndef _MY_HDF5_ORTHO_INFO
#define _MY_HDF5_ORTHO_INFO
#include <hdf5.h>
#include "exception.h"

class HDF5OrthoInfo
{

 public:
  Point3D P0,P1;
  unsigned nElems[3];
  VecWellInfo wells;
  //  VecWellInfo wells;

  HDF5OrthoInfo(){}
 HDF5OrthoInfo(Point3D P0,Point3D P1,unsigned nEX,unsigned nEY,unsigned nEZ,VecWellInfo wells)
   :P0(P0),P1(P1),wells(wells)
  {
    nElems[0] = nEX;
    nElems[1] = nEY;
    nElems[2] = nEZ;
  }
  void writeOrthoMeshInfo(hid_t triaGroup) 
  {
    double data[9];
    data[0]=this->P0(0);
    data[1]=this->P0(1);
    data[2]=this->P0(2);
    data[3]=this->P1(0);
    data[4]=this->P1(1);
    data[5]=this->P1(2);
    data[6]=this->nElems[0];
    data[7]=this->nElems[1];
    data[8]=this->nElems[2];
    //    std::cout << data[0] << data[1] << data[2] << data[3] << data[4] << data[5] << data[6] << std::endl;  
    hsize_t dim[] = {9};
    hid_t dataspace = H5Screate_simple(1,dim,NULL);
    hid_t datatype  = H5Tcopy(H5T_NATIVE_DOUBLE);
    H5Tset_order(datatype,H5T_ORDER_LE);
    hid_t dataset = H5Dcreate1(triaGroup, "dims", datatype, dataspace, H5P_DEFAULT);
    H5Dwrite(dataset,H5T_NATIVE_DOUBLE,H5S_ALL,H5S_ALL,H5P_DEFAULT,data);
    H5Sclose(dataspace);
    H5Tclose(datatype);
    H5Dclose(dataset);
    
    if (!wells.empty())
    {
      hsize_t dim[] = {wells.size(),6};
      hid_t dataspace = H5Screate_simple(2,dim,NULL);
      hid_t datatype  = H5Tcopy(H5T_NATIVE_DOUBLE);
      H5Tset_order(datatype,H5T_ORDER_LE);
      hid_t dataset = H5Dcreate1(triaGroup, "wells", datatype, dataspace, H5P_DEFAULT);
      
      double data[wells.size()][6];
      for (unsigned i=0;i<wells.size();i++)
      {
        data[i][0]=wells[i].P(0);
        data[i][1]=wells[i].P(1);
        data[i][2]=wells[i].P(2);
        data[i][3]=wells[i].Q(0);
        data[i][4]=wells[i].Q(1);
        data[i][5]=wells[i].Q(2);
      }
      H5Dwrite(dataset,H5T_NATIVE_DOUBLE,H5S_ALL,H5S_ALL,H5P_DEFAULT,data);
      H5Sclose(dataspace);
      H5Tclose(datatype);
      H5Dclose(dataset);
    }
    

    
  };
  void readOrthoMeshInfo(hid_t triaGroup) 
  {
    double data[9];
    hid_t dataset = H5Dopen1(triaGroup,"dims");
    if (dataset < 0)
      throw new Exception("DataSet \"dims\" not found\n"); 


    H5Dread(dataset,H5T_NATIVE_DOUBLE,H5S_ALL,H5S_ALL,H5P_DEFAULT,data);
    this->P0(0)=data[0];
    this->P0(1)=data[1];
    this->P0(2)=data[2];
    this->P1(0)=data[3];
    this->P1(1)=data[4];
    this->P1(2)=data[5];
    this->nElems[0]=data[6];
    this->nElems[1]=data[7];
    this->nElems[2]=data[8];
    H5Dclose(dataset);

    if (H5Lexists(triaGroup,"wells",H5P_DEFAULT)) //if data exists
    {
      hid_t datasetWells = H5Dopen1(triaGroup,"wells");

      hid_t dataspace = H5Dget_space(datasetWells);
      assert(H5Sget_simple_extent_ndims(dataspace) == 2);
      hsize_t dims[2];
      H5Sget_simple_extent_dims(dataspace,dims,NULL);
      assert(dims[1] == 6);
      double data[dims[0]][dims[1]];
      H5Dread(datasetWells,H5T_NATIVE_DOUBLE,H5S_ALL,H5S_ALL,H5P_DEFAULT,data);
      for (unsigned i=0;i<dims[0];i++)
      {
        Point3D p1(data[i][0],
                   data[i][1],
                   data[i][2]);
        Point3D p2(data[i][3],
                   data[i][4],
                   data[i][5]);
        WellInfo wl(p1,p2,0,WellInfo::PRESSURE);
        wells.push_back(wl);
      }
      H5Sclose(dataspace);
      H5Dclose(datasetWells);

    } 

  };


};
#endif

---