writer.h

//  Copyright (c) 2019 Prashant K. Jha
//  Copyright (c) 2019 Patrick Diehl
//
//  Distributed under the Boost Software License, Version 1.0. (See accompanying
//  file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
 
#ifndef RW_WRITER_H
#define RW_WRITER_H
 
#include "util/point.h"           // definition of Point3
#include "util/matrix.h"          // definition of SymMatrix3
#include <string>
#include <vector>
 
// forward declaration
namespace rw {
namespace writer {
class VtkWriter;
class LegacyVtkWriter;
class MshWriter;
}
} // namespace rw
 
namespace rw {
 
namespace writer {
 
class Writer {
 
public:
  Writer();
 
  explicit Writer(const std::string &filename, const std::string &format =
    "vtu", const std::string &compress_type = "");
 
  ~Writer();
 
  void open(const std::string &filename, const std::string &format = "vtu",
            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 appendPointData(const std::string &name,
                       const std::vector<blaze::StaticMatrix<double, 3, 3> > *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:
  rw::writer::VtkWriter *d_vtkWriter_p;
 
  rw::writer::LegacyVtkWriter *d_legacyVtkWriter_p;
 
  rw::writer::MshWriter *d_mshWriter_p;
 
  std::string d_format;
 
  inline void checkLength(const size_t length, const std::string &name);
 
}; // class Writer
 
} // namespace writer
 
} // namespace rw
 
#endif // RW_WRITER_H