vtkParticleWriter.h

Go to the documentation of this file.

/*
 * -------------------------------------------
 * Copyright (c) 2021 - 2024 Prashant K. Jha
 * -------------------------------------------
 * PeriDEM https://github.com/prashjha/PeriDEM
 *
 * Distributed under the Boost Software License, Version 1.0. (See accompanying
 * file LICENSE)
 */
 
#ifndef RW_VTK_PARTICLE_WRITER_H
#define RW_VTK_PARTICLE_WRITER_H
 
#include <vtkSmartPointer.h>
#include <vtkUnstructuredGrid.h>
#include <vtkXMLUnstructuredGridWriter.h>
 
// forward declaration
namespace model {
class ModelData;
}
 
namespace rw {
 
namespace writer {
 
class VtkParticleWriter {
 
public:
  explicit VtkParticleWriter(const std::string &filename, const std::string
  &compress_type = "");
 
  void appendNodes(const model::ModelData *model,
                   const std::vector<std::string> &tags);
 
  void appendMesh(const model::ModelData *model,
             const std::vector<std::string> &tags);
 
  void appendContactData(const model::ModelData *model,
                         const std::vector<size_t> *processed_nodes,
                         const std::vector <
                             std::pair<size_t, size_t>> *processed_elems);
 
  void appendStrainStress(const model::ModelData *model);
 
  void addTimeStep(const double &timestep);
 
  void close();
 
private:
  vtkSmartPointer<vtkXMLUnstructuredGridWriter> d_writer_p;
 
  vtkSmartPointer<vtkUnstructuredGrid> d_grid_p;
 
  std::string d_compressType;
};
 
} // namespace writer
 
} // namespace rw
 
#endif // RW_VTK_PARTICLE_WRITER_H