Fun4AllDstPileupInputManager.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef G4MAIN_FUN4ALLDSTPILEUPINPUTMANAGER_H
#define G4MAIN_FUN4ALLDSTPILEUPINPUTMANAGER_H

#include <fun4all/Fun4AllInputManager.h>
#include <fun4all/Fun4AllReturnCodes.h>   // for SYNC_NOOBJECT, SYNC_OK

#include <phool/PHCompositeNode.h>  // for PHCompositeNode
#include <phool/PHNodeIOManager.h>  // for PHNodeIOManager

#include <gsl/gsl_rng.h>

#include <map>
#include <memory>
#include <string>

class SyncObject;

class Fun4AllDstPileupInputManager : public Fun4AllInputManager
{
 public:
  Fun4AllDstPileupInputManager(const std::string &name = "DUMMY", const std::string &nodename = "DST", const std::string &topnodename = "TOP");
  int fileopen(const std::string &filenam) override;
  int fileclose() override;
  int run(const int nevents = 0) override;
  int BranchSelect(const std::string &branch, const int iflag) override;
  int setBranches() override;
  void Print(const std::string &what = "ALL") const override;
  int PushBackEvents(const int i) override;

  // Effectivly turn off the synchronization checking (copy from Fun4AllNoSyncDstInputManager)
  int SyncIt(const SyncObject* /*mastersync*/) override { return Fun4AllReturnCodes::SYNC_OK; }
  int GetSyncObject(SyncObject** /*mastersync*/) override { return Fun4AllReturnCodes::SYNC_NOOBJECT; }
  int NoSyncPushBackEvents(const int nevt) override { return PushBackEvents(nevt); }

  void setCollisionRate(double Hz)
  { m_collision_rate = Hz; }

  void setTimeBetweenCrossings(double nsec)
  { m_time_between_crossings = nsec; }

  void setPileupTimeWindow(double tmin, double tmax)
  {
    m_tmin = tmin;
    m_tmax = tmax;
  }

 private:

  int runOne(const int nevents = 0);


  bool m_ReadRunTTree = true;
  int m_ievent_total = 0;
  int m_ievent_thisfile = 0;

  std::string m_fullfilename;
  std::string m_RunNode = "RUN";
  std::map<const std::string, int> m_branchread;

  PHCompositeNode *m_dstNode = nullptr;

  PHCompositeNode *m_runNode = nullptr;

  std::unique_ptr<PHCompositeNode> m_dstNodeInternal;


  std::unique_ptr<PHCompositeNode> m_runNodeCopy;
  std::unique_ptr<PHCompositeNode> m_runNodeSum;


  std::unique_ptr<PHNodeIOManager> m_IManager;

  double m_time_between_crossings = 106;

  double m_collision_rate = 5e4;

  double m_tmin = -13500;

  double m_tmax = 13500;

  class Deleter
  {
    public:
    void operator() (gsl_rng* rng) const { gsl_rng_free(rng); }
  };

  std::unique_ptr<gsl_rng, Deleter> m_rng;

};

#endif /* __Fun4AllDstPileupInputManager_H__ */