gmXdmfFileWriter.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_XDMF_FILE_WRITER_H_
#define _GEMA_XDMF_FILE_WRITER_H_

#include "gmFileWriter.h"
#include "gmXdmfFileFormat.h"
#include "gmValueInfo.h"

#include <stdio.h>

class GmHdf5;

class GmXdmfFileWriter : public GmFileWriter, public GmXdmfFileFormat
{
public:
  GmXdmfFileWriter(const GmLogCategory& logger);

        virtual ~GmXdmfFileWriter();

  virtual bool createFiles(QString fileName, const GmSimulationData* simData, const GmFileFilter* filter,
                           const QVariantMap& options, bool multipleResults, bool gaussValues, bool discValues,
                           GmFileIO::DumpMode dumpMode, QString timeUnit);

  virtual bool setCurrentTime(int iter, double time);

  virtual bool saveGeometry(GmValueAccessor* nodeAc, const double* scaleFactor, const double* offset);

  virtual bool saveDiscontinuities(const QVector<GmDiscontinuitySet*>& sets, GmFileIO::DiscontinuityDataMode mode, 
                                   const UnitConverter* conv, const double* scaleFactor, const double* offset);

  virtual bool saveData(const QVector<GmResultNodeDataSrc*>&  nodeData,
                        const QVector<GmResultCellDataSrc*>&  cellData,
                        const QVector<GmResultGaussDataSrc*>& gaussData,
                        int extraColumns, const QVector<int>& propertySets);

        virtual bool closeFiles(bool deleteFiles = false);

  virtual void flushFiles();

        virtual GmFileFormat* fileFormat() { return this; }

  virtual bool dumpSupport() const;
  virtual bool addSavedItemsToMap   (QVariantMap& map);
  virtual bool loadSavedItemsFromMap(QVariantMap& map);
        
private:
  const char* tab() const { return _ident.constData(); }

  QString meshPath         (int rule = -1) const;
  QString discontinuityPath(QString dsName, bool original) const;
  QString path             (const char* baseName, int rule, int iter) const;

  QString geometryPath(int iter = 0) const { return path("geometry", -1, iter); }

  QString discontinuityGeometryPath(QString dsName, bool original, int iter = 0) const;

  QString nodeDataPath(int iter = 0) const { return path("nodeData", -1, iter); }

  QString cellDataPath(int iter = 0) const { return path("cellData", -1, iter); }

  QString gaussGeometryPath(int rule, int iter=0) const { assert(rule >= 0);  return path("geometry", rule, iter); }

  QString gaussDataPath(int rule, int iter=0) const { assert(rule >= 0);  return path("gaussData", rule, iter); }

  QString adjustSequenceInName(QString path, int seq) const
  {
    if(seq == -1)
      seq = _currSequence;
    assert((!_splitHDF5Files && !seq) || (_splitHDF5Files && seq > 0));
    return seq ? path.arg(seq) : path;
  }

  QString hdf5Name(int seq = -1) const { return adjustSequenceInName(_hdf5Name, seq); }

  QString hdf5Prefix(int seq = -1) const { return adjustSequenceInName(_hdf5Prefix, seq); }

  bool createCurrentHdf5File();
  void closeCurrentHdf5File(bool onError = false);

  void partialFileNames(QString& writingName, QString& partialName) const;
  bool savePartialFile();
  bool removeOldFiles(bool partialOnly);

  void addMeshGeometryEntries(QString basePath, bool pointCloud, int ruleIndex = -1);

  void addMainSection             (bool start, const GmSimulationData* simData);
  void addTimeCollectionSection   (bool start);
  void addSpatialCollectionSection(bool start);
  void addGridCollectionSection   (bool start);
  void addGridGaussSection        (bool start, int rule = -1);
  void addGridDiscontinuitySection(bool start, QString dsName, bool original);

  void addGeometryEntry     (QString tableName, int nlin, int ncol, QString unit, GmStorageType storage, bool forDiscontinuity = false);
  void addTopologyEntry     (QString tableName, int cellType, int numElem, int dataSize, bool forDiscontinuity = false);
  void addNodeTopologyEntry (QString tableName, int numNodes);
  void addDataAttributeEntry(QString tableName, QString name, QString unit, bool node, int nlin, int ncol1, int ncol2, GmStorageType storage);
  void addDataItemEntry     (QString tableName, int d1, int d2, int d3, GmStorageType storage, int seq);

  const GmSimulationData* _simData; 
  const GmFileFilter*     _filter;  

  QString          _fileName;   
  FILE*      _xdmfh;      
  QString    _hdf5Name;   
  QString    _hdf5Prefix; 
        GmHdf5*    _hdf5;       
  GmFileIO::DumpMode _dumpMode;   
  QByteArray _ident;      
  bool       _openGrid;   

  bool       _multiTime;           
  bool       _multiGrid;           
  bool       _gaussMultiGrid;      
  bool       _dsMultiGrid;         
  int        _splitHDF5Files;      
  int        _savePartialXdmfFile; 

  bool _saveCloudPointTopology; 

  double  _startTime;          
  double  _currTime;           
  int     _currIter;           
  QString _timeUnit;           
  int     _currSequence;       
  double  _sequenceStartTime;  
  bool    _newHDF5File;        

  int           _lastGeometryIter;          
  int           _lastGeometrySequence;      
  int           _lastDiscontinuityIter;     
  int           _lastDiscontinuitySequence; 
  int           _lastPartialXdmfOffset;     
  int           _lastPartialSuffix;         
  int           _geometryDim;               
  GmStorageType _geometryStorage;           
  QString       _geometryUnit;              
  bool          _singlePrecision;           

  QStringList                     _dsNames; 
  GmFileIO::DiscontinuityDataMode _dsMode;  

  QVector<int> _dsSizes; 
};

#endif