thermodynamiclatticeoutput.h

Go to the documentation of this file.

/*
 *
 *    Copyright (c) 2014-2022
 *      SMASH Team
 *
 *    GNU General Public License (GPLv3 or later)
 *
 */
 
#ifndef SRC_INCLUDE_SMASH_THERMODYNAMICLATTICEOUTPUT_H_
#define SRC_INCLUDE_SMASH_THERMODYNAMICLATTICEOUTPUT_H_
 
#include <filesystem>
#include <map>
#include <memory>
#include <set>
#include <string>
#include <vector>
 
#include "density.h"
#include "experimentparameters.h"
#include "file.h"
#include "forwarddeclarations.h"
#include "logging.h"
#include "outputinterface.h"
#include "outputparameters.h"
#include "threevector.h"
 
namespace smash {
 
class ThermodynamicLatticeOutput : public OutputInterface {
 public:
  static const double_t version;
 
  ThermodynamicLatticeOutput(const std::filesystem::path &path,
                             const std::string &name,
                             const OutputParameters &out_par,
                             const bool enable_ascii, const bool enable_binary);
  ~ThermodynamicLatticeOutput();
  void at_eventstart(
      const int event_number, const ThermodynamicQuantity tq,
      const DensityType dens_type,
      const RectangularLattice<DensityOnLattice> lattice) override;
  void at_eventstart(
      const int event_number, const ThermodynamicQuantity tq,
      const DensityType dens_type,
      const RectangularLattice<EnergyMomentumTensor> lattice) override;
 
  void at_eventend(const ThermodynamicQuantity tq) override;
 
  void thermodynamics_lattice_output(
      RectangularLattice<DensityOnLattice> &lattice,
      double current_time) override;
 
  void thermodynamics_lattice_output(
      RectangularLattice<DensityOnLattice> &lattice, const double current_time,
      const std::vector<Particles> &ensembles,
      const DensityParameters &dens_param) override;
 
  void thermodynamics_lattice_output(
      const ThermodynamicQuantity tq,
      RectangularLattice<EnergyMomentumTensor> &lattice,
      double current_time) override;
 
 private:
  const OutputParameters out_par_;
 
  std::string make_filename(const std::string &description,
                            const int event_number, const char type);
 
  std::string make_varname(const ThermodynamicQuantity tq,
                           const DensityType dens_type);
 
  void write_therm_lattice_ascii_header(std::shared_ptr<std::ofstream> file,
                                        const ThermodynamicQuantity &tq);
 
  void write_therm_lattice_binary_header(std::shared_ptr<std::ofstream> file,
                                         const ThermodynamicQuantity &tq);
 
  int to_int(const ThermodynamicQuantity &tq);
 
  const std::filesystem::path base_path_;
 
  std::map<ThermodynamicQuantity, std::shared_ptr<std::ofstream>>
      output_ascii_files_;
 
  std::map<ThermodynamicQuantity, std::shared_ptr<std::ofstream>>
      output_binary_files_;
 
  std::array<int, 3> nodes_;
 
  std::array<double, 3> sizes_;
 
  std::array<double, 3> origin_;
 
  bool enable_ascii_;
 
  bool enable_binary_;
 
  bool enable_output_;
};
 
}  // namespace smash
 
#endif  // SRC_INCLUDE_SMASH_THERMODYNAMICLATTICEOUTPUT_H_