event.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file event.h

// TRENTO: Reduced Thickness Event-by-event Nuclear Topology
// Copyright 2015 Jonah E. Bernhard, J. Scott Moreland
// TRENTO3D: Three-dimensional extension of TRENTO by Weiyao Ke
// MIT License

#ifndef EVENT_H
#define EVENT_H

#include <functional>
#include <map>

#ifdef NDEBUG
#define BOOST_DISABLE_ASSERTS
#endif
#include <boost/multi_array.hpp>

#include "fwd_decl.h"
#include "rapidity_profile.h"
#include "nucleon.h"

namespace trento {

class NucleonProfile;


class Event {
 public:
  explicit Event(const VarMap& var_map);

  void compute(const Nucleus& nucleusA, const Nucleus& nucleusB,
               NucleonProfile& profile);

  using Grid = boost::multi_array<double, 2>;

  using Grid3D = boost::multi_array<double, 3>;

  const int& npart() const
  { return npart_; }

  const int& ncoll() const
  { return ncoll_; }

  const double& multiplicity() const
  { return multiplicity_; }

  const std::map<int, double>& eccentricity() const
  { return eccentricity_; }

  const Grid3D& density_grid() const {
    if (is3D())
      return density_;
        else
          return TR_;
  }

  const double& dxy() const
  { return dxy_; }

  const double& deta() const
  { return deta_; }

  const std::map<int, double>& participant_plane() const
  { return psi_; }


  const Grid& TAB_grid() const
  { return TAB_; }

  void clear_TAB(void);
  void accumulate_TAB(Nucleon& A, Nucleon& B, NucleonProfile& profile);

  const bool& with_ncoll() const
  { return with_ncoll_; }

  const std::pair<double, double> mass_center_index() const
{ return std::make_pair(ixcm_, iycm_); }

 private:
  void compute_nuclear_thickness(
      const Nucleus& nucleus, NucleonProfile& profile, Grid& TX);

  template <typename GenMean>
  void compute_reduced_thickness(GenMean gen_mean);

  std::function<void()> compute_reduced_thickness_;

  void compute_observables();

  // Returns true if running in 3D mode, false otherwise.
  bool is3D() const;

  const double norm_;

  const double beam_energy_, exp_ybeam_;

  const double mean_coeff_, std_coeff_, skew_coeff_;
  const int skew_type_;

  const double dxy_, deta_;

  const int nsteps_, neta_;

  const double xymax_, etamax_;

  fast_eta2y eta2y_;

  cumulant_generating cgf_;

  Grid3D TR_, density_;

  Grid TA_, TB_, TAB_;

  double ixcm_, iycm_;

  int npart_;

  int ncoll_;

  double multiplicity_;

  std::map<int, double> eccentricity_;

  std::map<int, double> psi_;

  bool with_ncoll_;
};

}  // namespace trento

#endif  // EVENT_H