STEER/AliSimulation.h

   1 #ifndef ALISIMULATION_H
   2 #define ALISIMULATION_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 //
   9 // class for running generation, simulation and digitization
  10 // Hits, sdigits and digits are created for all detectors by typing:
  11 //   AliSimulation sim;
  12 //   sim.Run();
  13 //
  14
  15 #include <TNamed.h>
  16 #include <TString.h>
  17 #include <TObjArray.h>
  18 #include "AliQAv1.h"
  19 #include "AliQAManager.h"
  20
  21 class AliCDBId;
  22 class AliCDBParam;
  23 class AliRunLoader;
  24 class AliLegoGenerator;
  25 class AliLego;
  26 class AliMagF;
  27 class AliHLTSimulation;
  28
  29 class AliSimulation: public TNamed {
  30 public:
  31   AliSimulation(const char* configFileName = "Config.C",
  32                 const char* name = "AliSimulation",
  33                 const char* title = "generation, simulation and digitization");
  34   virtual ~AliSimulation();
  35
  36   static AliSimulation *Instance() {return fgInstance;}
  37
  38   void           SetNumberOfEvents(Int_t nEvents);
  39   void           SetConfigFile(const char* fileName);
  40   void           SetGAliceFile(const char* fileName);
  41   void           SetEventsPerFile(const char* detector, const char* type,
  42                                   Int_t nEvents);
  43
  44   void           SetRunGeneration(Bool_t run) {fRunGeneration = run;};
  45   void           SetRunSimulation(Bool_t run) {fRunSimulation = run;};
  46   void           SetLoadAlignFromCDB(Bool_t load)  {fLoadAlignFromCDB = load;};
  47   void           SetLoadAlignData(const char* detectors)
  48                    {fLoadAlObjsListOfDets = detectors;};
  49   void           SetMakeSDigits(const char* detectors)
  50                    {fMakeSDigits = detectors;};
  51   void           MergeWith(const char* fileName, Int_t nSignalPerBkgrd = 0);
  52   void           EmbedInto(const char* fileName, Int_t nSignalPerBkgrd = 0);
  53   void           SetUseBkgrdVertex(Bool_t useBkgrdVertex)
  54                    {fUseBkgrdVertex = useBkgrdVertex;};
  55   void           SetRegionOfInterest(Bool_t flag) {fRegionOfInterest = flag;};
  56   void           SetMakeDigits(const char* detectors)
  57                    {fMakeDigits = detectors;};
  58   void           SetMakeTrigger(const char* descriptors)
  59                    {fMakeTrigger = descriptors;};
  60   void           SetMakeDigitsFromHits(const char* detectors)
  61                    {fMakeDigitsFromHits = detectors;};
  62   void           SetWriteRawData(const char* detectors,
  63                                  const char* fileName = NULL,
  64                                  Bool_t deleteIntermediateFiles = kFALSE)
  65                    {fWriteRawData = detectors; fRawDataFileName = fileName;
  66                    fDeleteIntermediateFiles = deleteIntermediateFiles;};
  67   void           SetWriteSelRawData(Bool_t sel = kTRUE)
  68                    {fWriteSelRawData = sel;}
  69   void           SetAlignObjArray(TObjArray *array)
  70                    {fAlignObjArray = array;
  71                    fLoadAlignFromCDB = kFALSE;}
  72
  73   Bool_t         MisalignGeometry(AliRunLoader *runLoader = NULL);
  74
  75   void           SetRunNumber(Int_t run);
  76   void           SetSeed(Int_t seed);
  77
  78   void           ProcessEnvironmentVars();
  79
  80   // CDB storage activation
  81   void SetDefaultStorage(const char* uri);
  82   void SetSpecificStorage(const char* calibType, const char* uri);
  83
  84   virtual Bool_t Run(Int_t nEvents = 0);
  85   virtual Bool_t RunLego(const char *setup="Config.C",Int_t nc1=60,Float_t c1min=2,Float_t c1max=178,
  86                          Int_t nc2=60,Float_t c2min=0,Float_t c2max=360,Float_t rmin=0,
  87                          Float_t rmax=430,Float_t zmax=10000, AliLegoGenerator* gener=NULL, Int_t nev = -1);
  88
  89   virtual Bool_t RunSimulation(Int_t nEvents = 0);
  90   virtual Bool_t RunSDigitization(const char* detectors = "ALL");
  91   virtual Bool_t RunTrigger(const char* descriptors ="", const char* detectors = "ALL");
  92   virtual Bool_t WriteTriggerRawData();
  93   virtual Bool_t RunDigitization(const char* detectors = "ALL",
  94                                  const char* excludeDetectors = "");
  95   virtual Bool_t RunHitsDigitization(const char* detectors = "ALL");
  96   virtual Bool_t WriteRawData(const char* detectors = "ALL",
  97                               const char* fileName = NULL,
  98                               Bool_t deleteIntermediateFiles = kFALSE,
  99                               Bool_t selrawdata = kFALSE);
 100   virtual Bool_t WriteRawFiles(const char* detectors = "ALL");
 101   virtual Bool_t ConvertRawFilesToDate(const char* dateFileName = "raw.date",
 102                                        const char* rootFileName = "");
 103   virtual Bool_t ConvertDateToRoot(const char* dateFileName = "raw.date",
 104                                    const char* rootFileName = "raw.root");
 105   virtual Bool_t ConvertRaw2SDigits(const char* rawDirectory, const char* esdFile = "");
 106
 107   // HLT
 108   void SetRunHLT(const char* options) {fRunHLT=options;}
 109   virtual Bool_t CreateHLT();
 110   virtual Bool_t RunHLT();
 111   virtual  Bool_t IsLegoRun() const {return (fLego!=0);}
 112   AliLego* Lego() const {return fLego;}
 113   virtual  void  FinishRun();
 114
 115   //Quality Assurance
 116   Int_t       GetDetIndex(const char * detector);
 117   void        SetQACycles(AliQAv1::DETECTORINDEX_t det, const Int_t cycles) {  AliQAManager::QAManager()->SetCycleLength(det, cycles) ; }
 118   Bool_t      RunQA() ;
 119   Bool_t      SetRunQA(TString detAndAction="ALL:ALL") ;
 120   void        SetQAWriteExpert(AliQAv1::DETECTORINDEX_t det) { AliQAManager::QAManager()->SetWriteExpert(det) ; }
 121   void        SetQARefDefaultStorage(const char* uri);
 122   void        InitQA();
 123   void        SetEventSpecie(AliRecoParam::EventSpecie_t es) { fEventSpecie = es ; }
 124   void        SetWriteQAExpert() { fWriteQAExpertData = kTRUE ; }
 125
 126   void SetWriteGRPEntry(Bool_t flag = kTRUE) { fWriteGRPEntry = flag; }
 127   void WriteGRPEntry();
 128
 129 private:
 130
 131   AliSimulation(const AliSimulation&); // Not implemented
 132   AliSimulation& operator = (const AliSimulation&); // Not implemented
 133
 134   void           InitCDB();
 135   void           InitRunNumber();
 136   void           SetCDBLock();
 137   Bool_t         SetRunNumberFromData();
 138   AliRunLoader*  LoadRun(const char* mode = "UPDATE") const;
 139   Int_t          GetNSignalPerBkgrd(Int_t nEvents = 0) const;
 140   Bool_t         IsSelected(TString detName, TString& detectors) const;
 141
 142   static AliSimulation *fgInstance;    // Static pointer to object
 143
 144   Bool_t         fRunGeneration;      // generate prim. particles or not
 145   Bool_t         fRunSimulation;      // simulate detectors (hits) or not
 146   Bool_t         fLoadAlignFromCDB;   // Load alignment data from CDB and apply it to geometry or not
 147   TString        fLoadAlObjsListOfDets;   // Load alignment data from CDB for these detectors
 148   TString        fMakeSDigits;        // create sdigits for these detectors
 149   TString        fMakeDigits;         // create digits for these detectors
 150   TString        fMakeTrigger;        // run trigger for these descriptors
 151   TString        fMakeDigitsFromHits; // create digits from hits for these detectors
 152   TString        fWriteRawData;       // write raw data for these detectors
 153   TString        fRawDataFileName;    // file name for the raw data file
 154   Bool_t         fDeleteIntermediateFiles; // delete intermediate raw data files
 155   Bool_t         fWriteSelRawData;    // write detectors raw data in a separate file accoring to the trigger cluster
 156   Bool_t         fStopOnError;        // stop or continue on errors
 157
 158   Int_t          fNEvents;            // number of events
 159   TString        fConfigFileName;     // name of the config file
 160   TString        fGAliceFileName;     // name of the galice file
 161   TObjArray      fEventsPerFile;      // number of events per file for given detectors and data types
 162
 163   TObjArray*     fBkgrdFileNames;     // names of background files for merging
 164   TObjArray*       fAlignObjArray;      // array with the alignment objects to be applied to the geometry
 165   Bool_t         fUseBkgrdVertex;     // use vertex from background in case of merging
 166   Bool_t         fRegionOfInterest;   // digitization in region of interest
 167
 168   TString        fCDBUri;                            //! Uri of the default CDB storage
 169   TString        fQARefUri;                        //! Uri of the default QA reference storage
 170   TObjArray  fSpecCDBUri;                  //! Array with detector specific CDB storages
 171   Int_t            fRun;                                     //! Run number, will be passed to CDB and gAlice!!
 172   Int_t            fSeed;                        //! Seed for random number generator
 173   Bool_t           fInitCDBCalled;               //! flag to check if CDB storages are already initialized
 174   Bool_t           fInitRunNumberCalled;         //! flag to check if run number is already initialized
 175   Bool_t           fSetRunNumberFromDataCalled;  //! flag to check if run number is already loaded from run loader
 176
 177   Bool_t     fEmbeddingFlag;      // Flag for embedding
 178   AliLego       *fLego;              //! Pointer to aliLego object if it exists
 179
 180   //QA stuff
 181   static const Int_t   fgkNDetectors = 15 ;             // number of detectors
 182   static const char *  fgkDetectorName[fgkNDetectors] ; // names of detectors
 183   TString              fQADetectors ;                   // list of detectors to be QA'ed
 184   TString              fQATasks ;                       // list of QA tasks to be performed
 185   Bool_t               fRunQA ;                         // Runs the QA at the end of simulation
 186   AliRecoParam::EventSpecie_t fEventSpecie ;            // type of event (see AliRecoParam::EventSpecie_t)
 187   Bool_t               fWriteQAExpertData ;             //! decides wheter or not to write experts QA data; true by default
 188
 189   //HLT
 190   TString              fRunHLT;       //! HLT options, HLT is disabled if empty, default='default'
 191   AliHLTSimulation*    fpHLT;         //! The instance of HLT simulation
 192
 193   Bool_t         fWriteGRPEntry;      // Write or not GRP entry corresponding to the settings in Config.C
 194
 195   ClassDef(AliSimulation, 10)  // class for running generation, simulation and digitization
 196 };
 197
 198 #endif