vtk_writer.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2025 VTT Technical Research Centre of Finland Ltd
// SPDX-License-Identifier: AGPL-3.0-or-later
 
#ifndef PFC_VTK_WRITER_HPP
#define PFC_VTK_WRITER_HPP
 
#include "openpfc/results_writer.hpp"
#include <array>
#include <fstream>
#include <iomanip>
#include <mpi.h>
#include <sstream>
#include <string>
#include <vector>
 
namespace pfc {
 
class VTKWriter : public ResultsWriter {
private:
  std::array<int, 3> m_global_size;
  std::array<int, 3> m_local_size;
  std::array<int, 3> m_offset;
  std::array<double, 3> m_origin = {0.0, 0.0, 0.0};
  std::array<double, 3> m_spacing = {1.0, 1.0, 1.0};
  std::string m_field_name = "Field";
  int m_rank = 0;
  int m_num_ranks = 1;
  MPI_Comm m_comm = MPI_COMM_WORLD;
 
  std::string generate_filename(int increment, int rank = -1) const;
 
  void write_vti_header(std::ofstream &file, int increment) const;
 
  void write_vti_data(std::ofstream &file, const RealField &data) const;
 
  void write_pvti_file(int increment) const;
 
public:
  explicit VTKWriter(const std::string &filename) : ResultsWriter(filename) {
    MPI_Comm_rank(m_comm, &m_rank);
    MPI_Comm_size(m_comm, &m_num_ranks);
  }
 
  void set_domain(const std::array<int, 3> &arr_global,
                  const std::array<int, 3> &arr_local,
                  const std::array<int, 3> &arr_offset) override;
 
  void set_origin(const std::array<double, 3> &origin) { m_origin = origin; }
 
  void set_spacing(const std::array<double, 3> &spacing) { m_spacing = spacing; }
 
  void set_field_name(const std::string &name) { m_field_name = name; }
 
  MPI_Status write(int increment, const RealField &data) override;
 
  MPI_Status write(int increment, const ComplexField &data) override;
};
 
} // namespace pfc
 
#endif // PFC_VTK_WRITER_HPP