hdf5writer.cpp

Go to the documentation of this file.

#include "hdf5writer.h"
#include "exception.h"
#include <hdf5_hl.h>

using std::string;

 HDF5Writer::HDF5Writer() 
{
  m_file =-1;
  H5Eset_auto2(H5E_DEFAULT,(herr_t (*)(hid_t,void*))   &H5Eprint,stdout);

}

void HDF5Writer::setOutputFile(string strFileName) 
{
  m_strFileName=strFileName;
  m_file = H5Fcreate(strFileName.c_str(), H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT);
  H5Gclose(H5Gcreate1(m_file,"/DataSets",0));
  H5Gclose(H5Gcreate1(m_file,"/Triangulations",0));
}


void HDF5Writer::close() 
{
  if (m_file != -1)
  {
    H5Fclose(m_file);
  }
}
HDF5Writer::~HDF5Writer() 
{
  close();
}


void HDF5Writer::validateField(string triaId, string fieldName) 
{
  if ( varsHeap.find(fieldName) != varsHeap.end())
  {
    printf("HDF5Writer: Field name %s already defined as a variable\n",fieldName.c_str());
    abort();
  }
  if (triaHeap.find(fieldName) != triaHeap.end())
  {
    printf("HDF5Writer: Field name %s already defined as a triangulation\n",fieldName.c_str());
    abort();
  }
  if (triaHeap.find(triaId) == triaHeap.end())
  {    
    printf("HDF5Writer: Triangulation \"%s\" referenced by  field %s not defined\n",triaId.c_str(),fieldName.c_str());
    abort();
  }

}



void HDF5Writer::setVariable(std::string var, std::string value) 
{
  varsHeap[var]=value;
}

std::string HDF5Writer::getVariableValue(std::string var) 
{
  return varsHeap[var];
}


void HDF5Writer::setVariable(std::string var, double dValue) 
{
  char str[100];
  sprintf(str,"%g",dValue);
  setVariable(var,str);
}


  
  




HDF5Writer::TriaInformation& HDF5Writer::getTriaInformation(string strName) 
{
  if (triaHeap.find(strName) == triaHeap.end())
    throw new Exception("HDF5Writer: Triangulation %s does not exist",strName.c_str());
  return triaHeap.find(strName)->second;
}
bool HDF5Writer::existTriaInformation(string strName) 
{
  if (triaHeap.find(strName) == triaHeap.end())
    return false;
  else
    return true;
}

int HDF5Writer::getFieldCount(string fieldName) 
{
  if (fieldInfo.find(fieldName) == fieldInfo.end())
  {
    fieldInfo[fieldName]=FieldInformation();
    return 0;
  }
  else
  {
    return fieldInfo[fieldName].refCount;
  }
}


void HDF5Writer::incFieldCount(string fieldName) 
{
  if (fieldInfo.find(fieldName) == fieldInfo.end())
  {
    fieldInfo[fieldName]=FieldInformation();
  }
  fieldInfo[fieldName].refCount++;
  
}

HDF5Writer::VariableIterator HDF5Writer::getVariablesIterator() 
{
  return this->varsHeap.begin();
}

HDF5Writer::VariableIterator HDF5Writer::getVariablesEnd() 
{
  return this->varsHeap.end();
}



void HDF5Writer::writeContextVariables(std::string dataSetPath) 
{
  assert(m_file != -1);
  //Para todas as variaveis
  for (HDF5Writer::VariableIterator it = getVariablesIterator();it != getVariablesEnd();it++)
  {
    H5LTset_attribute_string(m_file,dataSetPath.c_str(),it->first.c_str(),it->second.c_str());
  }

}


void HDF5Writer::appendMeshInfo(string triaId,struct TriaInformation info) 
{
 triaHeap[triaId]=info;
 m_lastTriaName = triaId; 

}


void HDF5Writer::writeInChunks(hid_t dataset, hsize_t offset[], hsize_t count[],void *buff) 
{
  if (!count[0])
    return;
  hid_t dataspace = H5Dget_space(dataset);
  H5Sselect_hyperslab(dataspace, H5S_SELECT_SET, offset, NULL,count, NULL);

  hid_t datatype = H5Dget_type(dataset);

  hsize_t dim[]={count[0],count[1]};
  hid_t memspace = H5Screate_simple(2,dim,NULL);

  int error = H5Dwrite(dataset,datatype,memspace,dataspace,H5P_DEFAULT,buff);
  assert(error >=0);

  H5Sclose(memspace);
  H5Sclose(dataspace);
  H5Tclose(datatype);
}

void HDF5Writer::putMeshIntoDictionary(string triaId,unsigned verts,unsigned cells) 
{
  appendMeshInfo(triaId,TriaInformation(verts,cells));
  
}




void HDF5Writer::setAtt(string objPath,string attName,string value) 
{
  herr_t error = H5LTset_attribute_string(getFile(),objPath.c_str(),attName.c_str(),value.c_str());
  if (error < 0)
    throw new Exception("HDF5OrthoWriter::setAtt error trying to set attribute \"%s/%s\"",objPath.c_str(),attName.c_str());
}



void HDF5Writer::setAtt(string objPath,string attName,double dValue)
{
  char value[100];
  sprintf(value,"%g",dValue);
  herr_t error = H5LTset_attribute_string(getFile(),objPath.c_str(),attName.c_str(),value);
  if (error < 0)
  throw new Exception("HDF5OrthoWriter::setAtt error trying to set attribute \"%s/%s\"",objPath.c_str(),attName.c_str());
}

---