PHG4Reco.h

Go to the documentation of this file.

#ifndef PHG4Reco_h
#define PHG4Reco_h
 
#include <fun4all/SubsysReco.h>
 
#include <phfield/PHFieldConfig.h>
 
#include <phool/PHTimeServer.h>
 
#include <list>
 
// Forward declerations
class PHCompositeNode;
class G4RunManager;
class PHG4PrimaryGeneratorAction;
class PHG4PhenixDetector;
class PHG4PhenixEventAction;
class PHG4PhenixSteppingAction;
class PHG4PhenixTrackingAction;
class PHG4Subsystem;
class PHG4EventGenerator;
class G4VModularPhysicsList;
class G4TBMagneticFieldSetup;
class G4VUserPrimaryGeneratorAction;
class PHG4UIsession;
 
// for the G4 cmd interface and the graphics
class G4UImanager;
class G4VisManager;
 
class PHG4Reco : public SubsysReco
{
 public:
  PHG4Reco(const std::string &name = "PHG4RECO");
 
  virtual ~PHG4Reco();
 
  int Init(PHCompositeNode *);
 
  int InitRun(PHCompositeNode *topNode);
 
  int process_event(PHCompositeNode *);
 
  int ResetEvent(PHCompositeNode *);
 
  int End(PHCompositeNode *);
 
  void Print(const std::string &what = std::string()) const;
 
  void registerSubsystem(PHG4Subsystem *subsystem)
  {
    subsystems_.push_back(subsystem);
  }
 
  int ApplyCommand(const std::string &cmd);
 
  int StartGui();
 
  int InitField(PHCompositeNode *topNode);
 
  void set_field(const float tesla)
  {
    magfield = tesla;
  }
 
  void set_field_map(const std::string &fmap, const PHFieldConfig::FieldConfigTypes dim)
  {
    fieldmapfile = fmap;
    mapdim = dim;
  }
 
  void set_field_map();
  
  void set_field_rescale(const float rescale) { magfield_rescale = rescale; }
 
  void set_optical_photon(bool b) { optical_photon_ = b; }
  void set_energy_threshold(double t) { energy_threshold_ = t; }
  void set_zero_field_line(double z) { zero_field_line_ = z; }
 
  void save_DST_geometry(bool b) { save_DST_geometry_ = b; }
  void SetWorldSizeX(const double sx) { WorldSize[0] = sx; }
  void SetWorldSizeY(const double sy) { WorldSize[1] = sy; }
  void SetWorldSizeZ(const double sz) { WorldSize[2] = sz; }
  double GetWorldSizeX() const { return WorldSize[0]; }
  double GetWorldSizeY() const { return WorldSize[1]; }
  double GetWorldSizeZ() const { return WorldSize[2]; }
  void SetWorldShape(const std::string &s) { worldshape = s; }
  void SetWorldMaterial(const std::string &s) { worldmaterial = s; }
  void SetPhysicsList(const std::string &s) { physicslist = s; }
  void set_rapidity_coverage(const double eta);
 
  int setupInputEventNodeReader(PHCompositeNode *);
 
  static void G4Seed(const unsigned int i);
 
  // this is an ugly hack to get Au ions working for CAD
  // our particle generators have pdg build in which doesn't work
  // with ions, so the generator action has to be replaced
  // which is hardcoded in PHG4Reco (it has to be created after
  // the physics lists are instantiated
  void setGeneratorAction(G4VUserPrimaryGeneratorAction *action);
 
  PHG4Subsystem *getSubsystem(const std::string &name);
 
  void Dump_GDML(const std::string &filename);
 
  void G4Verbosity(const int i);
 protected:
  int InitUImanager();
  void DefineMaterials();
  void DefineRegions();
  float magfield;
  float magfield_rescale;
  double WorldSize[3];
 
  G4TBMagneticFieldSetup *field_;
 
  G4RunManager *runManager_;
 
  PHG4UIsession *uisession_;
 
  PHG4PhenixDetector *detector_;
 
  PHG4PhenixEventAction *eventAction_;
 
  PHG4PhenixSteppingAction *steppingAction_;
 
  PHG4PhenixTrackingAction *trackingAction_;
 
  PHG4PrimaryGeneratorAction *generatorAction_;
 
  typedef std::list<PHG4Subsystem *> SubsystemList;
  SubsystemList subsystems_;
 
  // visualization
  G4VisManager *visManager;
 
  double _eta_coverage;
  PHFieldConfig::FieldConfigTypes mapdim;
  std::string fieldmapfile;
  std::string worldshape;
  std::string worldmaterial;
  std::string physicslist;
 
  bool save_DST_geometry_;
 
  PHTimeServer::timer _timer;
 
  bool   optical_photon_;        //< enable process for the optical photon, default disabled
  double energy_threshold_;      //< kill low-energy tracks fast, set to 0. to disable
  double zero_field_line_;       //< z position after which the field manager will be set to nullptr, set to numbers larger than the world size to disable
};
 
#endif