gmFileIO.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_FILE_IO_H_
#define _GEMA_FILE_IO_H_

#include "gmCoreConfig.h"

#include "gmStateDumpItem.h"
#include "gmGaussToElementNodeInterpolator.h"

#include <QStringList>
#include <QVariantMap>

class GmFileReader;
class GmFileWriter;
class GmFileFormat;
class GmFileFilter;

class GmLogCategory;
class GmSimulationData;
class GmMesh;
class GmCellMesh;
class GmElementMesh;
class GmCellGroupSet;
class GmBoundaryCondition;
class GmDiscontinuitySet;

class GmResultNodeDataSrc;
class GmResultCellDataSrc;
class GmResultGaussDataSrc;

class GmValueInfo;
class GmValueSetData;
class GmValueAccessor;
class GmSpatialIndex;
class GmNumSolver;

class UnitConverter;

class GMC_API_EXPORT GmFileIO : public GmGroupDumpItem
{
public:
  enum DumpMode
  {
    DISABLED,  
    DUMP,      
    RESTORE,   
  };

  enum DiscontinuityDataMode
  {
    NO_DISC_DATA,            
    ORIGINAL_DISC_DATA,      
    INTERSECTION_DISC_DATA,  
    BOTH_DISC_DATA,          
  };

  enum ExtraColumns {
    NODE_ID_COLUMN       = 0x01, 
    CELL_ID_COLUMN       = 0x02, 
    CELL_TYPE_COLUMN     = 0x04, 
    CELL_ACTIVE_COLUMN   = 0x08, 
    CELL_MATERIAL_COLUMN = 0x10, 
    GAUSS_ID_COLUMN      = 0x20, 
    NODE_BC_COLUMN       = 0x40, 
    // --- Combined options ---
    NODE_COLUMNS         = NODE_ID_COLUMN, 
    CELL_COLUMNS         = CELL_ID_COLUMN | CELL_TYPE_COLUMN | CELL_ACTIVE_COLUMN | CELL_MATERIAL_COLUMN,
    GAUSS_COLUMNS        = GAUSS_ID_COLUMN,
  };

  ~GmFileIO();

  static GmFileIO* createInstance(QString fileType, bool writeMode, const GmLogCategory& logger);

  bool setNodeData (LuaTable& tab, GmMesh*        mesh, bool split);
  bool setCellData (LuaTable& tab, GmCellMesh*    mesh, bool split);
  bool setGaussData(LuaTable& tab, GmElementMesh* mesh, bool split);

  bool saveData(QString fileName, const GmSimulationData* simData, GmMesh* mesh, const GmCellGroupSet* group, const QVariantMap& options);
                
  bool prepareForSaving(QString fileName, const GmSimulationData* simData, GmMesh* mesh, const GmCellGroupSet* group, const QVariantMap& options);

  bool saveCurrentResult(double currentTime, bool meshChanged = false, bool discChanged = false);
  bool closeFile();

  QString tempFile();

  virtual bool addStateItemsToGroup(GmStateDump* state, int groupId);
  virtual bool stateAboutToBeSaved (GmStateDump* state);
  virtual bool stateLoaded         (GmStateDump* state);

  static QString extraColumnName(ExtraColumns col, const GmCellMesh* mesh = NULL, int psIndex = -1);

private:
  GmFileIO(GmFileReader* reader, GmFileWriter* writer, const GmLogCategory& logger);

  Q_DISABLE_COPY(GmFileIO);

  template <class T, class Mesh, class DsF>
  bool setData(LuaTable& tab, Mesh* mesh, const DsF& dsF, QVector<T*>& dataList, const char* entity, bool split);

  bool parseAndCheckDumpMode  (bool multipleTimeSteps);
  bool parseAndCheckStates    (bool multipleTimeSteps, int* state, bool* allStates, QVector<double>& timeList);
  bool parseAndCheckGaussRules(QVector<int>& rules);
  bool parseAndCheckGaussInterpolationOptions(const GmSimulationData* simData);
  bool parseAndCheckCoordAccessor();
  bool parseAndCheckDiscontinuityOptions(const GmSimulationData* simData);
  bool parseAndCheckExtraColumnOptions(const GmSimulationData* simData);
  bool checkDataForSaving(bool multipleTimeSteps);
  bool adjustAcStates    (int state);
  bool adjustAndCheckIds ();

  bool prepareNodeInterpolatedValues();
  bool prepareNodeBcValues();

  bool savePrepare(QString fileName, const GmSimulationData* simData, GmMesh* mesh, const GmCellGroupSet* group, 
                   const QVariantMap& options, bool multipleTimeSteps, QVector<double>& timeList);

  bool doSaveGeometry();
  bool doSaveDiscontinuities();
  bool doSaveData();

  bool cleanup(bool deleteFiles = false);

  const GmLogCategory& _logger;    
  GmFileWriter*        _writer;    
  GmFileReader*        _reader;    
  GmFileFilter*        _filter;    
  QString              _fileName;  
  DumpMode             _dumpMode;  

  GmMesh*                        _mesh;      
  QVector<GmResultNodeDataSrc*>  _nodeData;  
  QVector<GmResultCellDataSrc*>  _cellData;  
  QVector<GmResultGaussDataSrc*> _gaussData; 

  GmValueAccessor* _nodeAc;                  
  bool             _coordTransf;             
  double           _coordScaleFactor[3];     
  double           _coordOffset[3];          

  bool         _openFiles;    
  int          _saveIter;     
  QVariantMap  _options;      
  int          _extraColumns; 
  QVector<int> _propertySets; 

  struct InternalVsData {
    GmValueInfo*         _info; 
    GmValueSetData*      _vs;   
    GmResultNodeDataSrc* _ds;   

    InternalVsData(GmValueInfo* info, GmValueSetData* vs, GmResultNodeDataSrc* ds) : _info(info), _vs(vs), _ds(ds) {}
    ~InternalVsData();
  private:
    Q_DISABLE_COPY(InternalVsData);
  };

  QVector<const GmBoundaryCondition*> _meshBcs;      
  InternalVsData*                     _nodeBcVsData; 

  enum InterpDataMode
  {
    ELEMENT_INT_MODE, 
    NODE_INT_MODE,    
    SPLINE_INT_MODE,  
    CLOUD_INT_MODE,   
  };

  struct InterpData {
    InterpDataMode           _mode;              
    bool                     _alsoSaveAtGauss;   
    GmInterpolatorType       _interpType;        
    QVariant                 _interpParam;       
    GmInterpolatorAveraging  _elemAvgMode;       
    bool                     _nodeClosest;       
    bool                     _searchModeRadius;  
    double                   _searchDomain;      
    const GmSpatialIndex*    _gaussIndex;        
    const GmNumSolver*       _splineNumSolver;   
    GmCellGroupSet*          _meshGroup;         
    QVector<InternalVsData*> _nodeVsData;        

    InterpData();
    ~InterpData();
  private:
    Q_DISABLE_COPY(InterpData);
  };

  InterpData* _interpData; 

  QVector<GmDiscontinuitySet*> _dsList; 
  DiscontinuityDataMode        _dsMode; 
  UnitConverter*               _dsConv; 

  QVariantMap _stateMap;  
};

#endif