vtkoutput.h

Go to the documentation of this file.

/*
 *
 *    Copyright (c) 2014-2019
 *      SMASH Team
 *
 *    GNU General Public License (GPLv3 or later)
 *
 */

#ifndef SRC_INCLUDE_VTKOUTPUT_H_
#define SRC_INCLUDE_VTKOUTPUT_H_

#include <string>

#include <boost/filesystem.hpp>

#include "density.h"
#include "forwarddeclarations.h"
#include "outputinterface.h"
#include "outputparameters.h"

namespace smash {

class VtkOutput : public OutputInterface {
 public:
  VtkOutput(const bf::path &path, const std::string &name,
            const OutputParameters &out_par);
  ~VtkOutput();

  void at_eventstart(const Particles &particles,
                     const int event_number) override;

  void at_eventend(const Particles &particles, const int event_number,
                   double impact_parameter, bool empty_event) override;

  void at_intermediate_time(const Particles &particles,
                            const std::unique_ptr<Clock> &clock,
                            const DensityParameters &dens_param) 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;

 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);

  const bf::path base_path_;

  int current_event_ = 0;
  int vtk_output_counter_ = 0;

  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;
  bool is_thermodynamics_output_;
};

}  // namespace smash

#endif  // SRC_INCLUDE_VTKOUTPUT_H_