PndMasterRunSim.h

Go to the documentation of this file.

#pragma once
 
#include "FairRunSim.h"
#include "FairRuntimeDb.h"
#include "FairTask.h"

#include "TStopwatch.h"
#include "TString.h"

class FairFilteredPrimaryGenerator;
class PndFilteredPrimaryGenerator;
class FairGenerator;

class FairBoxGenerator;

class PndBoxGenerator;

class PndMasterRunSim : public FairRunSim {
 public:
  PndMasterRunSim();

  virtual ~PndMasterRunSim();

  Bool_t Setup(TString outprefix = "");

  void Finish();

  void CreateGeometry();

  void CreateGeometryDefault();

  void CreateGeometryPhase1();

  void CreateGeometryDay1();

  void AddSimTasks();

  void SetGenerator();

  void AddGenerator(FairGenerator *aGen);

  void SetGenerator(FairGenerator *aGen);

  void SetDpmFlag(Int_t Mode) { fDpmFlag = Mode; };

  void SetFtfFlag(Int_t Mode) { fFtfFlag = Mode; };

  void UseDpmGenerator();

  void UseFtfGenerator(TString ftfData);

  void UsePythia8Generator(TString py8Data);

  void UseEvtGenGenerator(TString EvtGenFile);

  void UseBoxGenerator(TString BoxConfig);

  void UsePiPiGenerator(TString pipiConfig);

  void UseLepLepGenerator(TString leplepConfig);
  void UseAsciiGenerator(TString AsciiFile);

  void SetInput(TString par) { fInput = par; }

  void SetInputDir(TString par) { fInputDir = par; }

  void SetParamRootFile(TString par) { fParamRootFile = par; }

  void SetParamAsciiFile(TString par) { fParamAsciiFile = par; }

  void SetOptions(TString par)
  {
    fOptions = par;
    fOptions.ToLower();
  }
  TString GetOptions() { return fOptions; }

  void SetNumberOfEvents(Int_t par) { fNEvents = par; }

  void SetEventCounterRate(Int_t par) { fEventCounterRate = par; }

  void SetTargetMode(Short_t par) { fTargetMode = par; }

  FairFilteredPrimaryGenerator *GetFilteredPrimaryGenerator() { return (FairFilteredPrimaryGenerator *)fGen; }
  PndFilteredPrimaryGenerator *GetPndFilteredPrimaryGenerator() { return (PndFilteredPrimaryGenerator *)fGen; }

  void PrintListOfTasks(TList *list, TString prefix = " - ");
  void PrintListOfTasks() { PrintListOfTasks(GetMainTask()->GetListOfTasks()); }

 private:
  void GetRange(TString par, double &min, double &max);
  void GetCoords(TString par, double &X, double &Y, double &Z);
  void CreatePrimGen();

  TString fInput;          
  TString fInputDir;       
  TString fOutFile;        
  TString fParamRootFile;  
  TString fParamAsciiFile; 
  TString fOptions;        

  Int_t fDpmFlag;          
  Int_t fFtfFlag;          
  Int_t fNEvents;          
  Int_t fEventCounterRate; 
  Short_t fTargetMode;     

  FairRuntimeDb *fRtdb; 
  TStopwatch fTimer;    

  ClassDef(PndMasterRunSim, 2); 

};