Fun4AllHepMCPileupInputManager.h

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

#ifndef PHHEPMC_FUN4ALLHEPMCPILEUPINPUTMANAGER_H
#define PHHEPMC_FUN4ALLHEPMCPILEUPINPUTMANAGER_H

#include "Fun4AllHepMCInputManager.h"

#include <gsl/gsl_rng.h>

#include <map>
#include <string>

class Fun4AllHepMCPileupInputManager : public Fun4AllHepMCInputManager
{
 public:
  Fun4AllHepMCPileupInputManager(const std::string &name = "DUMMY",
                                 const std::string &nodename = "DST",
                                 const std::string &topnodename = "TOP");
  ~Fun4AllHepMCPileupInputManager() override;

  int run(const int nevents = 0) override { return run(nevents, false); }

  int run(const int nevents, const bool skip);
  int ResetEvent() override;
  void set_time_window(double past_nsec, double future_nsec)
  {
    _min_integration_time = past_nsec;
    _max_integration_time = future_nsec;
  }

  void set_collision_rate(double Hz) { _collision_rate = Hz; }
  void set_time_between_crossings(double nsec) { _time_between_crossings = nsec; }

  int SkipForThisManager(const int nevents) override;
  void SignalInputManager(Fun4AllHepMCInputManager *in) { m_SignalInputManager = in; }
  int PushBackEvents(const int i) override;

 private:
  int InsertEvent(HepMC::GenEvent *evt, const double crossing_time);

  Fun4AllHepMCInputManager *m_SignalInputManager = nullptr;
  gsl_rng *RandomGenerator = nullptr;

  int m_SignalEventNumber = 0;
  double _min_integration_time = -17500.0;
  double _max_integration_time = 17500.0;
  double _collision_rate = 100.0e3;
  double _time_between_crossings = 106.0;

  //derived parameters
  double _ave_coll_per_crossing = 1.;
  int _min_crossing = 0;
  int _max_crossing = 0;

  bool _first_run = true;

  std::map<int, double> m_EventNumberMap;
};

#endif /* PHHEPMC_FUN4ALLHEPMCINPUTMANAGER_H */