PndMasterRunSim.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#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();

  virtual Bool_t Setup(TString outprefix = "");

  void Finish();

  virtual void CreateGeometry();

  void CreateGeometryDefault();

  virtual 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; }

  virtual void SetupTargetMode() {};

  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()); }

 protected:
  void GetRange(TString par, double &min, double &max);
  void GetCoords(TString par, double &X, double &Y, double &Z);
  virtual void SetOutputFile(TString outprefix);
  virtual void SetRtdb();
  virtual void SetField();
  virtual void DefineParticles();
  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); 

};