vtkoutput.h

Go to the documentation of this file.

/*
 *
 *    Copyright (c) 2014-2022,2024
 *      SMASH Team
 *
 *    GNU General Public License (GPLv3 or later)
 *
 */
 
#ifndef SRC_INCLUDE_SMASH_VTKOUTPUT_H_
#define SRC_INCLUDE_SMASH_VTKOUTPUT_H_
 
#include <filesystem>
#include <map>
#include <memory>
#include <string>
#include <utility>
 
#include "density.h"
#include "forwarddeclarations.h"
#include "outputinterface.h"
#include "outputparameters.h"
 
namespace smash {
 
class VtkOutput : public OutputInterface {
 public:
  VtkOutput(const std::filesystem::path &path, const std::string &name,
            const OutputParameters &out_par);
  ~VtkOutput();
 
  void at_eventstart(const Particles &particles, const EventLabel &event_label,
                     const EventInfo &event) override;
 
  void at_eventend(const Particles &particles, const EventLabel &event_label,
                   const EventInfo &event) override;
 
  void at_intermediate_time(const Particles &particles,
                            const std::unique_ptr<Clock> &clock,
                            const DensityParameters &dens_param,
                            const EventLabel &event_label,
                            const EventInfo &event) override;
 
  void thermodynamics_output(
      const ThermodynamicQuantity tq, const DensityType dt,
      RectangularLattice<DensityOnLattice> &lattice) override;
 
  void thermodynamics_output(
      const ThermodynamicQuantity tq, const DensityType dt,
      RectangularLattice<EnergyMomentumTensor> &lattice) override;
 
  void thermodynamics_output(const GrandCanThermalizer &gct) override;
 
  void fields_output(
      const std::string name1, const std::string name2,
      RectangularLattice<std::pair<ThreeVector, ThreeVector>> &lat) override;
 
 private:
  void write(const Particles &particles);
 
  std::string make_filename(const std::string &description, int counter);
 
  std::string make_varname(const ThermodynamicQuantity tq,
                           const DensityType dens_type);
 
  template <typename T>
  void write_vtk_header(std::ofstream &file, RectangularLattice<T> &lat,
                        const std::string &description);
 
  template <typename T, typename F>
  void write_vtk_scalar(std::ofstream &file, RectangularLattice<T> &lat,
                        const std::string &varname, F &&function);
 
  template <typename T, typename F>
  void write_vtk_vector(std::ofstream &file, RectangularLattice<T> &lat,
                        const std::string &varname, F &&function);
 
  std::pair<int, int> counter_key() {
    return {current_event_, current_ensemble_};
  }
 
  const std::filesystem::path base_path_;
 
  int current_event_ = 0;
  int current_ensemble_ = 0;
  std::map<std::pair<int, int>, int> vtk_output_counter_{};
 
  int vtk_density_output_counter_ = 0;
  int vtk_tmn_output_counter_ = 0;
  int vtk_tmn_landau_output_counter_ = 0;
  int vtk_v_landau_output_counter_ = 0;
  int vtk_fluidization_counter_ = 0;
  int vtk_fields_output_counter_ = 0;
  bool is_thermodynamics_output_;
  bool is_fields_output_;
};
 
}  // namespace smash
 
#endif  // SRC_INCLUDE_SMASH_VTKOUTPUT_H_