gmNfFileWriter.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_NF_FILE_WRITER_H_
#define _GEMA_NF_FILE_WRITER_H_
 
#include "gmFileWriter.h"
#include "gmNfFileFormat.h"
#include "gmFileIOUtils.h"
#include "gmCell.h"
 
#include <stdio.h>
 
class QTemporaryFile;
class GmFileFiler;
class GmSimulationData;
class GmTaskManager;
 
class GmNfFileWriter : public GmFileWriter, public GmNfFileFormat
{
  const int maxNodeIdChars = 10; //Greather than 2^32, so we can save up to 10 digit node ids
  struct CellOrder { int id; int cellId; };
 
  //Per property write functions
  using wNFunc =  std::function<int(char*, int)>; //Node write function
  using wCFunc =  std::function<int(char*, CellOrder, const GmCell*)>; //Cell write function
  using wGFunc =  std::function<int(char*, CellOrder, int,  const GmCell*, const GmIntegrationRule*)>; //Gauss write function
  using GetCoord = std::function<const GmVector* (const GmCell*, GmVector*)>; //Eval coordinates for cell data
public:
  GmNfFileWriter(const GmLogCategory& logger);
 
  virtual ~GmNfFileWriter();
 
  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)
  {
    Q_UNUSED(sets); Q_UNUSED(mode); Q_UNUSED(conv); Q_UNUSED(scaleFactor); Q_UNUSED(offset);
    //TODO
    return true;
  }
 
  virtual bool saveData(const QVector<GmNodeAcResultsDataSrc*>&  nodeData,
                        const QVector<GmCellAcResultsDataSrc*>&  cellData,
                        const QVector<GmGaussAcResultsDataSrc*>& gaussData,
                        int extraColumns, const QVector<int>& propertySets);
 
  virtual bool endWrite();
  virtual bool closeFiles(bool deleteFiles = false);
 
  virtual QString tempFile();
 
  virtual void flushFiles();
 
  virtual GmFileFormat* fileFormat() { return this; }
 
  virtual bool dumpSupport() const;
  virtual bool addSavedItemsToMap   (QVariantMap& map);
  virtual bool loadSavedItemsFromMap(QVariantMap& map);
 
private:
  bool copyResultData(FILE* srcFile, FILE* dstFile, qint64 size = 0);
  
  void writeHeader();
  void saveNodeCoordinates(GmValueAccessor* nodeAcc, const QVector<int>& indexer, const double* scaleFactor, const double* offset);
  void saveMeshTopology(const GmCellMesh* mesh);
  int fillExtraNodesIndex(const GmMesh* mesh, const GmValueAccessor* nodeAc);
  void writeResultsHeader();
  void findMaterialGroup   (const GmMesh* mesh, const QVariantMap& options);
  void saveMaterialGroup   (const GmCellMesh* mesh);
  void saveIntegrationRules(const GmElementMesh* mesh);
  bool saveNodeData(const QVector<GmNodeAcResultsDataSrc*>&  nodeData, int extraCol);
  void saveDisplacement(GmValueAccessor* ac, const QVector<int>& indexer);
  bool saveCellGaussData(const QVector<GmCellAcResultsDataSrc*>& cellData, const QVector<GmGaussAcResultsDataSrc*>& gaussData, int extraCol);
 
  void fillNodeCoordBuffer(const GmValueAccessor* cac, int start, int count, const double* scale, const double* offset, const QVector<int>& indexer, GmFileIOUtils::GmWBuffer& buf);
  void fillCellTopoBuffer(const GmCellMesh* mesh, int start, int count, int nnodes, const char* afterMaterial,
   const char* afterNode, const QVector<int>& nodeNumbering, const QVector<CellOrder>& indexer, GmFileIOUtils::GmWBuffer& buf);
 
  int fillCellIndexerPerType(QVector<CellOrder> order[GM_NUM_CELL_TYPES], bool checkIr = false);
 
  //Split properties into scalar (directly write functions) and vector
  int buildNodeFunctions (const QVector<GmNodeAcResultsDataSrc*>& prop, const QVector<int>& indexer, int extraCol, QVector<wNFunc>& func, GmValueAccessor** dispAc, QString& header);
 
  int buildCellFunctions (const QVector<GmCellAcResultsDataSrc*>& prop, int extraCol, const GetCoord& getCoord, QVector<wCFunc>& func, QString& header);
  int buildGaussFunctions(const QVector<GmGaussAcResultsDataSrc*>& prop,int extraCol, QVector<wGFunc>& func, QString& header);
 
 
 
  QString _nfFileName;          
  FILE*   _nfFile;              
  FILE*   _tempFile;            
  QTemporaryFile* _tempFileObj; 
  int     _materialGroup;       
  int     _intRuleSet;          
  int     _elementRule[GM_NUM_CELL_TYPES]; 
  int     _elementNip[GM_NUM_CELL_TYPES];
  int     _nresults;            
  bool    _saveDisplacements;   
  bool    _useModelOrder;       
 
  QVector<double> _extraNodes; // Set of extra nodes created to transform a INT2DL6 into a QUAD8 and INT3DL12 into a HEX20
 
  const GmFileFilter* _filter; 
  const GmSimulationData* _simData; 
  GmTaskManager* _tm;
  bool _saveGaussData;   
  bool _multipleResults; 
  int _dumpMode;
};
 
#endif