listmodus.h

Go to the documentation of this file.

/*
 *    Copyright (c) 2015-2023
 *      SMASH Team
 *
 *    GNU General Public License (GPLv3 or later)
 */
#ifndef SRC_INCLUDE_SMASH_LISTMODUS_H_
#define SRC_INCLUDE_SMASH_LISTMODUS_H_
 
#include <cmath>
#include <cstdint>
#include <list>
#include <string>
#include <utility>
 
#include "forwarddeclarations.h"
#include "modusdefault.h"
 
namespace smash {
 
class ListModus : public ModusDefault {
 public:
  explicit ListModus(Configuration modus_config,
                     const ExperimentParameters &parameters);
 
  ListModus() : shift_id_(0) {}
 
  double initial_conditions(Particles *particles,
                            const ExperimentParameters &parameters);
  void backpropagate_to_same_time(Particles &particles);
 
  void try_create_particle(Particles &particles, PdgCode pdgcode, double t,
                           double x, double y, double z, double mass, double E,
                           double px, double py, double pz);
 
  struct LoadFailure : public std::runtime_error {
    using std::runtime_error::runtime_error;
  };
 
  bool is_list() const { return true; }
 
  void set_file_id(const double file_id_inh) { file_id_ = file_id_inh; }
 
  void set_particle_list_file_directory(
      std::string particle_list_file_directory_inh) {
    particle_list_file_directory_ = particle_list_file_directory_inh;
  }
 
  void set_particle_list_file_prefix(
      std::string particle_list_file_prefix_inh) {
    particle_list_file_prefix_ = particle_list_file_prefix_inh;
  }
 
  void set_event_id(int event_id_inh) { event_id_ = event_id_inh; }
 
 protected:
  double start_time_ = 0.;
 
 private:
  bool file_has_events_(std::filesystem::path filepath,
                        std::streampos last_position);
 
  std::filesystem::path file_path_(const int file_id);
 
  std::string next_event_();
 
  std::string particle_list_file_directory_;
 
  std::string particle_list_file_prefix_;
 
  std::string current_particle_list_file_;
 
  const int shift_id_;
 
  int event_id_;
 
  int file_id_;
 
  bool warn_about_mass_discrepancy_ = true;
  bool warn_about_off_shell_particles_ = true;
 
  std::streampos last_read_position_;
 
  friend std::ostream &operator<<(std::ostream &, const ListModus &);
};
 
class ListBoxModus : public ListModus {
 public:
  explicit ListBoxModus(Configuration modus_config,
                        const ExperimentParameters &parameters);
 
  bool is_box() const { return true; }
 
  int impose_boundary_conditions(Particles *particles,
                                 const OutputsList &output_list = {});
 
  Grid<GridOptions::PeriodicBoundaries> create_grid(
      const Particles &particles, double min_cell_length,
      double timestep_duration, CollisionCriterion crit,
      const bool include_unformed_particles,
      CellSizeStrategy strategy = CellSizeStrategy::Optimal) const {
    CellNumberLimitation limit = CellNumberLimitation::ParticleNumber;
    if (crit == CollisionCriterion::Stochastic) {
      limit = CellNumberLimitation::None;
    }
    return {{{0, 0, 0}, {length_, length_, length_}},
            particles,
            min_cell_length,
            timestep_duration,
            limit,
            include_unformed_particles,
            strategy};
  }
 
 private:
  const int shift_id_;
 
  const double length_;
 
  int file_id_;
 
  int event_id_;
 
  std::string particle_list_file_directory_;
 
  std::string particle_list_file_prefix_;
};
 
}  // namespace smash
 
#endif  // SRC_INCLUDE_SMASH_LISTMODUS_H_