mshWriter.h

// Copyright (c) 2019    Prashant K. Jha
//
// Distributed under the GNU GENERAL PUBLIC LICENSE, Version 3.0.
// (See accompanying file LICENSE.txt)
 
#ifndef RW_MSHWRITER_H
#define RW_MSHWRITER_H
 
#include "util/matrix.h" // definition of SymMatrix3
#include "util/point.h"  // definition of Point3
#include <fstream>
#include <vector>
 
namespace rw {
 
namespace writer {
 
class MshWriter {
 
public:
  explicit MshWriter(const std::string &filename, const std::string &compress_type = "");
 
  void appendNodes(const std::vector<util::Point3> *nodes,
                   const std::vector<util::Point3> *u = nullptr);
 
  void appendMesh(const std::vector<util::Point3> *nodes,
                  const size_t &element_type,
                  const std::vector<size_t> *en_con,
                  const std::vector<util::Point3> *u = nullptr);
 
  void appendPointData(const std::string &name,
                       const std::vector<uint8_t> *data);
 
  void appendPointData(const std::string &name,
                       const std::vector<size_t> *data);
 
  void appendPointData(const std::string &name, const std::vector<int> *data);
 
  void appendPointData(const std::string &name, const std::vector<float> *data);
 
  void appendPointData(const std::string &name,
                       const std::vector<double> *data);
 
  void appendPointData(const std::string &name,
                       const std::vector<util::Point3> *data);
 
  void appendPointData(const std::string &name,
                       const std::vector<util::SymMatrix3> *data);
 
  void appendCellData(const std::string &name, const std::vector<float> *data);
 
  void appendCellData(const std::string &name,
                       const std::vector<util::SymMatrix3> *data);
 
  void appendFieldData(const std::string &name, const double &data);
 
  void appendFieldData(const std::string &name, const float &data);
 
  void addTimeStep(const double &timestep);
 
  void close();
 
private:
  void writeMshDataHeader(const std::string &name, int field_type,
                          size_t num_data, bool is_node_data = true);
 
  std::string d_filename;
 
  std::string d_compressType;
 
  FILE *d_file;
};
 
} // namespace writer
 
} // namespace rw
 
#endif // RW_MSHWRITER_H