PHG4Reco.h

Go to the documentation of this file.

#ifndef PHG4Reco_h
#define PHG4Reco_h

#include <g4decayer/EDecayType.hh>

#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_rescale(const float rescale) { magfield_rescale = rescale; }

  void set_decayer_active(bool b) { active_decayer_ = b; }
  void set_force_decay(EDecayType force_decay_type)
  {
    active_decayer_ = true;
    active_force_decay_ = true;
    force_decay_type_ = force_decay_type;
  }

  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;

  // settings for the external Pythia6 decayer
  bool active_decayer_;          //< turn on/off decayer
  bool active_force_decay_;      //< turn on/off force decay channels
  EDecayType force_decay_type_;  //< forced decay channel setting

  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