generators/phhepmc/HepMCFlowAfterBurner.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003
0004 #ifndef PHHEPMC_HEPMCFLOWAFTERBURNER_H
0005 #define PHHEPMC_HEPMCFLOWAFTERBURNER_H
0006
0007 #include <fun4all/SubsysReco.h>
0008
0009 #include <string>
0010
0011 class PHCompositeNode;
0012
0013 class HepMCFlowAfterBurner : public SubsysReco
0014 {
0015  public:
0016   HepMCFlowAfterBurner(const std::string &name = "HEPMCFLOWAFTERBURNER");
0017   ~HepMCFlowAfterBurner() override {}
0018
0019   int Init(PHCompositeNode *) override;
0020   int process_event(PHCompositeNode *) override;
0021
0022   void Print(const std::string &what = "ALL") const override;
0023
0024   void setAlgorithmName(const std::string &name);
0025
0026   void setMinEta(const float f)
0027   {
0028     mineta = f;
0029   }
0030   void setMaxEta(const float f)
0031   {
0032     maxeta = f;
0033   }
0034   void setMinPt(const float f)
0035   {
0036     minpt = f;
0037   }
0038   void setMaxPt(const float f)
0039   {
0040     maxpt = f;
0041   }
0042   void setSeed(const long il);
0043
0044   void SaveRandomState(const std::string &savefile = "HepMCFlowAfterBurner.ransave");
0045   void RestoreRandomState(const std::string &savefile = "HepMCFlowAfterBurner.ransave");
0046
0047  protected:
0048   std::string config_filename;
0049   std::string algorithmName = "MINBIAS";
0050
0051   float mineta = -5.;
0052   float maxeta = 5.;
0053
0054   float minpt = 0.;
0055   float maxpt = 100.;
0056
0057   int seedset = 0;
0058   long seed = 0;
0059   long randomSeed = 11793;
0060 };
0061
0062 #endif