DataExporter.h

Go to the documentation of this file.

// $Id$
//==============================================================================
//==============================================================================
 
#ifndef _DATA_EXPORTER_H
#define _DATA_EXPORTER_H
 
#include "ControlFIFO.h"
#include <map>
#include <string>
#include <vector>
 
class DataWriter;
class ProcessAdm;
class TimeStep;
namespace utl {
class LogStream;
}
 
 
class DataExporter : public ControlCallback
{
public:
  enum FieldType {
    VECTOR,
    INTVECTOR,
    KNOTSPAN,
    SIM,
    NODALFORCES,
    BASIS
  };
 
  enum Results {
    PRIMARY      = 1, 
    DISPLACEMENT = 2, 
    SECONDARY    = 4, 
    NORMS        = 8, 
    EIGENMODES   = 16, 
    ONCE         = 32, 
    GRID         = 128, 
    REDUNDANT    = 256, 
    L2G_NODE     = 512, 
    ELEMENT_MASK = 1024 
  };
 
  struct FileEntry {
    std::string description; 
    FieldType   field;       
    int         results;     
    const void* data;        
    std::vector<const void*> data2; 
    std::string prefix;      
    bool enabled;            
    int  ncmps;              
  };
 
  DataExporter(bool dynWriters = false, int ndump = 1, int level0 = 0) :
    m_delete(dynWriters), m_ndump(ndump), m_level(level0-1), m_last_step(-1) {}
 
  virtual ~DataExporter();
 
  void registerWriter(DataWriter* writer) { m_writers.push_back(writer); }
 
  bool registerField(const std::string& name,
                     const std::string& description,
                     FieldType field, int results = PRIMARY,
                     const std::string& prefix = "", int ncmps = 0);
 
  bool setFieldValue(const std::string& name,
                     const void* data,
                     const void* data2=nullptr,
                     const void* data3=nullptr,
                     const void* data4=nullptr);
 
  bool dumpTimeLevel(const TimeStep* tp = nullptr,
                     bool geoUpd = false, bool doLog = false);
 
  virtual void OnControl(const tinyxml2::XMLElement* context);
  virtual std::string GetContext() const { return "datawriter"; }
 
  std::string getName() const;
 
  int getTimeLevel() const { return m_level; }
  int getStride() const { return m_ndump; }
 
protected:
  std::map<std::string,FileEntry> m_entry;
  std::vector<DataWriter*>        m_writers;
 
  bool m_delete;    
  int  m_ndump;     
  int  m_level;     
  int  m_last_step; 
};
 
using DataEntry = std::pair<std::string,DataExporter::FileEntry>;
 
 
class DataWriter
{
protected:
  DataWriter(const std::string& name, const ProcessAdm& adm,
             const char* defaultExt = nullptr);
 
public:
  virtual ~DataWriter() {}
 
  virtual int getLastTimeLevel() = 0;
 
  virtual void openFile(int level) = 0;
 
  virtual void closeFile(int level) = 0;
 
  virtual void writeVector(int level, const DataEntry& entry) = 0;
 
  virtual void writeSIM(int level, double time, const DataEntry& entry,
                        bool geometryUpdated, const std::string& prefix) = 0;
 
  virtual void writeNodalForces(int level, const DataEntry& entry) = 0;
 
  virtual void writeKnotspan(int level, const DataEntry& entry,
                             const std::string& prefix) = 0;
 
  virtual void writeBasis(int level, const DataEntry& entry,
                          const std::string& prefix) = 0;
 
  virtual bool writeTimeInfo(int level, double time) = 0;
 
  virtual bool writeLog(const std::string& data, const std::string& name) = 0;
 
  const std::string& getName() const { return m_name; }
 
protected:
  std::string m_name; 
 
  int m_size; 
  int m_rank; 
};
 
#endif