gmHdf5ModelWriter.h

Go to the documentation of this file.

/************************************************************************
**
** Copyright (C) 2014 by Carlos Augusto Teixera Mendes
** All rights reserved.
**
** This file is part of the "GeMA" software. It's use should respect
** the terms in the license agreement that can be found together 
** with this source code.
** It is provided AS IS, with NO WARRANTY OF ANY KIND, 
** INCLUDING THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR 
** A PARTICULAR PURPOSE.
**
************************************************************************/
 
#ifndef _GEMA_HDF5_MODEL_WRITER_H_
#define _GEMA_HDF5_MODEL_WRITER_H_
 
#include "gmLog.h"
#include "gmHdf5Model.h"
#include "gmPODVector.h"
#include "gmCellType.h"
#include "gmBoundaryCondition.h"
#include "gmDiscontinuitySet.h"
#include "gmNanBoxedValue.h"
 
#include <QStringList>
 
class GmModelData;
class GmMesh;
class GmCellMesh;
class GmValueAccessor;
class GmUserFunction;
class GmNodeSet;
class GmCellBoundary;
 
namespace H5 {
  class H5File;
}
 
 
class GMC_API_EXPORT GmHdf5ModelWriter :  public GmHdf5Model
{
public:
 
  GmHdf5ModelWriter(QString fileName, const GmLogCategory& logger);
  ~GmHdf5ModelWriter();
 
  void setInitialData(const QStringList& nodeAttributes, const QStringList& cellAttributes, 
                      const QStringList& gaussAttributes);
 
  bool saveFromModel(const GmModelData* data, QString modelName = "", QString meshName = "");
 
  bool addNewMesh   (QString meshName, QString modelName = "");
  bool addDictionary(DictionaryType type, const QStringList& names);
 
  bool addGeometryNodeCoordinates  (int n, int d, const double* values, int nghost = 0);
  bool addGeometryCells            (int ncells, int size, const int* data);
  bool addGeometryCellGhostNodes   (int ncells, int size, const int* data);
  bool addGeometryIndexTable       (GroupType group, QString tableName, int n, const int* values);
  bool addGeometryBorderTable      (GroupType group, QString tableName, int n, const int* values);
  bool addInitialData              (GroupType group, QString attributeName, int n, int d, const double* values, bool hasFunctions, int history = -1);
  bool addBoundaryCondition        (QString bcName, GmBoundaryCondition::BcType type, int n, const int* data);
  bool addBoundaryConditionProperty(QString bcName, QString propertyName, int n, int d, const double* values, bool hasFunctions);
  bool addDiscontinuitySet         (QString dsName, GmDiscontinuitySet::DiscontinuityType type, const QStringList& discontinuityNames);
  bool addDiscontinuityPolyline    (QString dsName, int index, int n, int d, const double* coordinates);
  bool addDiscontinuitySurface     (QString dsName, int index, int n, const double* coordinates, int ntri, const int* triangles);
  bool addDiscontinuityEdgeList    (QString dsName, int index, int n, const int* borderData);
  bool addDiscontinuityFaceList    (QString dsName, int index, int n, const int* borderData);
 
 
  bool       cellVectorBuildBegin(int npsets, int expectedSize);
  bool       cellVectorAddCell   (GmCellType type, const int* propertyIndices, const int* nodeIndices);
  const int* cellVectorBuildEnd  (int* size);
 
  bool       cellGhostNodesVectorBuildBegin(int expectedSize);
  bool       cellGhostNodesVectorAddCell   (int cellId, int numGhostNodes, const int* nodeIndices);
  const int* cellGhostNodesVectorBuildEnd  (int* size);
 
  double dindexToNan(unsigned index) const
  {
    assert(index < (unsigned)_dictionaries[GM_HDF5MODEL_FUNCTIONS].size());
    return GmNanBoxedIndex(index).toRawDouble();
  }
 
  double nanToCanonicNan(double v) const { return GmNanBoxedIndex(v, true).toDouble(); }
 
private:
  bool buildAndSaveFunctionsDictionary   (const GmModelData* data);
  bool buildAndSavePropertySetsDictionary(const GmCellMesh*  mesh);
  bool buildAndSaveCellGroupsDictionary  (const GmCellMesh*  mesh);
  bool buildAndSaveNodeSetsDictionary    (const GmMesh*      mesh);
  bool buildAndSaveCellBordersDictionary (const GmCellMesh*  mesh);
 
  bool saveMeshGeometryNodes     (const GmMesh*      mesh);
  bool saveMeshGeometryNodeSets  (const GmMesh*      mesh);
  bool saveMeshGeometryCells     (const GmCellMesh*  mesh);
  bool saveMeshGeometryBoundaries(const GmCellMesh*  mesh);
  bool saveInitialData           (const GmMesh*      mesh);
  bool saveBoundaryConditions    (const GmMesh*      mesh);
  bool saveDiscontinuitySets     (const GmModelData* data, const GmCellMesh* mesh);
 
  int* cellVectorAddCell          (GmCellType type);
  int* cellGhostNodesVectorAddCell(int cellId, int numGhostNodes);
 
  template <class T>
  bool addDataTable(GroupType group, QString subGroup, QString tableName, int n, int d, const T* values);
 
  template <class T, class A>
  bool addDataTableWithAttributes(GroupType group, QString subGroup, QString tableName, int n, int d, const T* values,
                                  const QStringList& attrNames, const QVector<A>& attrValues);
 
 
  QString groupName(GroupType group, QString subGroupName) const { return GmHdf5Model::groupName(_currentMesh, group, subGroupName); }
 
  bool createGroupIfNecessary(QString groupName);
 
  void initSavedAttributeLists(const GmMesh* mesh);
 
  bool initBuffer(const GmModelData* data, const GmMesh* mesh);
  template<class T> T* getBuffer(int n);
 
  double* getFilledDataBuffer(GmValueAccessor* ac, const GmMesh* mesh, bool* bufferHasFunctions = NULL);
 
  QString              _fileName;              
  const GmLogCategory& _logger;                
 
  bool _userDefinedAttributes; 
 
  QVector<QPair<QString, int>> _savedNodeAttributes;   
  QVector<QPair<QString, int>> _savedCellAttributes;   
  QVector<QPair<QString, int>> _savedGaussAttributes;  
 
  GmPODVector<char, size_t, GmPODVectorTightGrow> _saveBuffer; 
 
  H5::H5File*                         _fileh;          
  QString                             _currentMesh;    
  QVector<const GmBoundaryCondition*> _currentMeshBcs; 
 
  QVector<QStringList> _dictionaries;     
  QSet<QString>        _fileGroups;       
  int                  _npsets;           
 
  QMap<QString,               int> _userFunctionMap; 
  QMap<const GmNodeSet*,      int> _nodeSetMap;      
  QMap<const GmCellBoundary*, int> _edgeMap;         
  QMap<const GmCellBoundary*, int> _faceMap;         
};
 
#endif