[u/mrichter/AliRoot.git] / STEER / AliSimulation.h

#ifndef ALISIMULATION_H
#define ALISIMULATION_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//
// class for running generation, simulation and digitization
// Hits, sdigits and digits are created for all detectors by typing:
//   AliSimulation sim;
//   sim.Run();
//

#include <TNamed.h>
#include <TString.h>
#include <TObjArray.h>
#include "AliQAv1.h"
#include "AliQAManager.h"

class AliCDBId;
class AliCDBParam;
class AliRunLoader;
class AliLegoGenerator;
class AliLego;
class AliMagF;
class AliHLTSimulation;

class AliSimulation: public TNamed {
public:
  AliSimulation(const char* configFileName = "Config.C",
		const char* name = "AliSimulation", 
		const char* title = "generation, simulation and digitization");
  virtual ~AliSimulation();

  static AliSimulation *Instance() {if(!fgInstance) fgInstance=new AliSimulation(); return fgInstance;}

  void           SetNumberOfEvents(Int_t nEvents);
  void           SetConfigFile(const char* fileName);
  void           SetGAliceFile(const char* fileName);
  void           SetEventsPerFile(const char* detector, const char* type, 
				  Int_t nEvents);

  void           SetRunGeneration(Bool_t run) {fRunGeneration = run;};
  void           SetRunSimulation(Bool_t run) {fRunSimulation = run;};
  void           SetLoadAlignFromCDB(Bool_t load)  {fLoadAlignFromCDB = load;};
  void           SetLoadAlignData(const char* detectors) 
                   {fLoadAlObjsListOfDets = detectors;};
  void           SetMakeSDigits(const char* detectors) 
                   {fMakeSDigits = detectors;};
  void           MergeWith(const char* fileName, Int_t nSignalPerBkgrd = 0);
  void           EmbedInto(const char* fileName, Int_t nSignalPerBkgrd = 0);
  void           SetUseBkgrdVertex(Bool_t useBkgrdVertex)
                   {fUseBkgrdVertex = useBkgrdVertex;};
  void           SetRegionOfInterest(Bool_t flag) {fRegionOfInterest = flag;};
  void           SetMakeDigits(const char* detectors)
                   {fMakeDigits = detectors;};
  void           SetMakeDigitsFromHits(const char* detectors)
                   {fMakeDigitsFromHits = detectors;};
  void           SetWriteRawData(const char* detectors, 
				 const char* fileName = NULL,
				 Bool_t deleteIntermediateFiles = kFALSE)
                   {fWriteRawData = detectors; fRawDataFileName = fileName;
		   fDeleteIntermediateFiles = deleteIntermediateFiles;};
  void           SetWriteSelRawData(Bool_t sel = kTRUE)
                   {fWriteSelRawData = sel;}
  void           SetTriggerConfig(TString conf) {fTriggerConfig=conf;}
  const Char_t*  GetTriggerConfig() const {return fTriggerConfig.Data();}
  void           SetAlignObjArray(TObjArray *array)
                   {fAlignObjArray = array;
		   fLoadAlignFromCDB = kFALSE;}

  Bool_t         MisalignGeometry(AliRunLoader *runLoader = NULL);

  void           SetRunNumber(Int_t run);
  void           SetSeed(Int_t seed);
    
  void 		 ProcessEnvironmentVars();
		   
  // CDB storage activation
  void SetDefaultStorage(const char* uri);
  void SetSpecificStorage(const char* calibType, const char* uri);

  virtual Bool_t Run(Int_t nEvents = 0);
  virtual Bool_t RunLego(const char *setup="Config.C",Int_t nc1=60,Float_t c1min=2,Float_t c1max=178,
			 Int_t nc2=60,Float_t c2min=0,Float_t c2max=360,Float_t rmin=0,
			 Float_t rmax=430,Float_t zmax=10000, AliLegoGenerator* gener=NULL, Int_t nev = -1);

  virtual Bool_t RunSimulation(Int_t nEvents = 0);
  virtual Bool_t RunSDigitization(const char* detectors = "ALL");
  virtual Bool_t RunTrigger(const char* descriptors ="", const char* detectors = "ALL");
  virtual Bool_t WriteTriggerRawData();
  virtual Bool_t RunDigitization(const char* detectors = "ALL",
				 const char* excludeDetectors = "");
  virtual Bool_t RunHitsDigitization(const char* detectors = "ALL");
  virtual Bool_t WriteRawData(const char* detectors = "ALL",
			      const char* fileName = NULL,
			      Bool_t deleteIntermediateFiles = kFALSE,
			      Bool_t selrawdata = kFALSE);
  virtual Bool_t WriteRawFiles(const char* detectors = "ALL");
  virtual Bool_t ConvertRawFilesToDate(const char* dateFileName = "raw.date",
				       const char* rootFileName = "");
  virtual Bool_t ConvertDateToRoot(const char* dateFileName = "raw.date",
				   const char* rootFileName = "raw.root");
  virtual Bool_t ConvertRaw2SDigits(const char* rawDirectory, const char* esdFile = "", Int_t N=-1);

  // Sets the name of the file from which the geometry is loaded
  virtual void SetGeometryFile(const Char_t* filename) {fGeometryFile=filename;}
  virtual const Char_t* GetGeometryFile() const {return fGeometryFile.Data();}
  virtual Bool_t IsGeometryFromFile() const {return !fGeometryFile.IsNull();}


  // HLT
  void SetRunHLT(const char* options) {fRunHLT=options;}
  virtual Bool_t CreateHLT();
  virtual Bool_t RunHLT();
  virtual  Bool_t IsLegoRun() const {return (fLego!=0);}
  AliLego* Lego() const {return fLego;}
  virtual  void  FinishRun();

  //Quality Assurance
  Int_t       GetDetIndex(const char * detector);
  void        SetQACycles(AliQAv1::DETECTORINDEX_t det, const Int_t cycles) {  AliQAManager::QAManager()->SetCycleLength(det, cycles) ; }
  Bool_t      RunQA() ;
  Bool_t      SetRunQA(TString detAndAction="ALL:ALL") ; 
  void        SetQAWriteExpert(AliQAv1::DETECTORINDEX_t det) { AliQAManager::QAManager()->SetWriteExpert(det) ; }  
  void        SetQARefDefaultStorage(const char* uri);
  void        InitQA();
  void        SetEventSpecie(AliRecoParam::EventSpecie_t es) { fEventSpecie = es ; }
  void        SetWriteQAExpert() { fWriteQAExpertData = kTRUE ; }

  void SetWriteGRPEntry(Bool_t flag = kTRUE) { fWriteGRPEntry = flag; }
  void WriteGRPEntry();
  void UseVertexFromCDB()   {fUseVertexFromCDB   = kTRUE;}
  void UseMagFieldFromGRP() {fUseMagFieldFromGRP = kTRUE;} 
  void SetGRPWriteLocation(char* loc) {fGRPWriteLocation = loc;}

private:

  AliSimulation(const AliSimulation&); // Not implemented
  AliSimulation& operator = (const AliSimulation&); // Not implemented

  void 		 InitCDB();
  void 		 InitRunNumber();
  void 		 SetCDBLock();
  Bool_t         SetRunNumberFromData();
  AliRunLoader*  LoadRun(const char* mode = "UPDATE") const;
  Int_t          GetNSignalPerBkgrd(Int_t nEvents = 0) const;
  Bool_t         IsSelected(TString detName, TString& detectors) const;

  static AliSimulation *fgInstance;    // Static pointer to object

  Bool_t         fRunGeneration;      // generate prim. particles or not
  Bool_t         fRunSimulation;      // simulate detectors (hits) or not
  Bool_t         fLoadAlignFromCDB;   // Load alignment data from CDB and apply it to geometry or not
  TString        fLoadAlObjsListOfDets;   // Load alignment data from CDB for these detectors
  TString        fMakeSDigits;        // create sdigits for these detectors
  TString        fMakeDigits;         // create digits for these detectors
  TString        fTriggerConfig;      // run trigger for these descriptors
  TString        fMakeDigitsFromHits; // create digits from hits for these detectors
  TString        fWriteRawData;       // write raw data for these detectors
  TString        fRawDataFileName;    // file name for the raw data file
  Bool_t         fDeleteIntermediateFiles; // delete intermediate raw data files
  Bool_t         fWriteSelRawData;    // write detectors raw data in a separate file accoring to the trigger cluster
  Bool_t         fStopOnError;        // stop or continue on errors

  Int_t          fNEvents;            // number of events
  TString        fConfigFileName;     // name of the config file
  TString        fGAliceFileName;     // name of the galice file
  TObjArray      fEventsPerFile;      // number of events per file for given detectors and data types

  TObjArray*     fBkgrdFileNames;     // names of background files for merging
  TObjArray*     fAlignObjArray;      // array with the alignment objects to be applied to the geometry
  Bool_t         fUseBkgrdVertex;     // use vertex from background in case of merging
  Bool_t         fRegionOfInterest;   // digitization in region of interest

  TString 	 fCDBUri;	                     //! Uri of the default CDB storage
  TString 	 fQARefUri;	                     //! Uri of the default QA reference storage
  TObjArray      fSpecCDBUri;                        //! Array with detector specific CDB storages
  Int_t 	   fRun; 		                     //! Run number, will be passed to CDB and gAlice!!
  Int_t 	   fSeed;                        //! Seed for random number generator 
  Bool_t 	   fInitCDBCalled;               //! flag to check if CDB storages are already initialized
  Bool_t 	   fInitRunNumberCalled;         //! flag to check if run number is already initialized
  Bool_t 	   fSetRunNumberFromDataCalled;  //! flag to check if run number is already loaded from run loader
  
  Bool_t         fEmbeddingFlag;       // Flag for embedding
  AliLego       *fLego;                //! Pointer to aliLego object if it exists
  // OCDB
  ULong_t         fKey;                //! current CDB key
  Bool_t          fUseVertexFromCDB;   // Flag to use Vertex from CDB
  Bool_t          fUseMagFieldFromGRP; // Use magnetic field settings from GRP
  TString         fGRPWriteLocation;   // Location to write the GRP entry from simulation
  
  //QA stuff
  static const Int_t   fgkNDetectors = 15 ;             // number of detectors
  static const char *  fgkDetectorName[fgkNDetectors] ; // names of detectors
  TString              fQADetectors ;                   // list of detectors to be QA'ed 	
  TString              fQATasks ;                       // list of QA tasks to be performed	
  Bool_t               fRunQA ;                         // Runs the QA at the end of simulation
  AliRecoParam::EventSpecie_t fEventSpecie ;            // type of event (see AliRecoParam::EventSpecie_t)
  Bool_t               fWriteQAExpertData ;             //! decides wheter or not to write experts QA data; true by default

  TString              fGeometryFile;                   // Geometry file

  //HLT
  TString              fRunHLT;       //! HLT options, HLT is disabled if empty, default='default'
  AliHLTSimulation*    fpHLT;         //! The instance of HLT simulation

  Bool_t         fWriteGRPEntry;      // Write or not GRP entry corresponding to the settings in Config.C

  ClassDef(AliSimulation, 11)  // class for running generation, simulation and digitization
};

#endif
Commit	Line	Data
	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() {if(!fgInstance) fgInstance=new AliSimulation(); 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 SetMakeDigitsFromHits(const char* detectors)
	59	{fMakeDigitsFromHits = detectors;};
	60	void SetWriteRawData(const char* detectors,
	61	const char* fileName = NULL,
	62	Bool_t deleteIntermediateFiles = kFALSE)
	63	{fWriteRawData = detectors; fRawDataFileName = fileName;
	64	fDeleteIntermediateFiles = deleteIntermediateFiles;};
	65	void SetWriteSelRawData(Bool_t sel = kTRUE)
	66	{fWriteSelRawData = sel;}
	67	void SetTriggerConfig(TString conf) {fTriggerConfig=conf;}
	68	const Char_t* GetTriggerConfig() const {return fTriggerConfig.Data();}
	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 = "", Int_t N=-1);
	106
	107	// Sets the name of the file from which the geometry is loaded
	108	virtual void SetGeometryFile(const Char_t* filename) {fGeometryFile=filename;}
	109	virtual const Char_t* GetGeometryFile() const {return fGeometryFile.Data();}
	110	virtual Bool_t IsGeometryFromFile() const {return !fGeometryFile.IsNull();}
	111
	112
	113	// HLT
	114	void SetRunHLT(const char* options) {fRunHLT=options;}
	115	virtual Bool_t CreateHLT();
	116	virtual Bool_t RunHLT();
	117	virtual Bool_t IsLegoRun() const {return (fLego!=0);}
	118	AliLego* Lego() const {return fLego;}
	119	virtual void FinishRun();
	120
	121	//Quality Assurance
	122	Int_t GetDetIndex(const char * detector);
	123	void SetQACycles(AliQAv1::DETECTORINDEX_t det, const Int_t cycles) { AliQAManager::QAManager()->SetCycleLength(det, cycles) ; }
	124	Bool_t RunQA() ;
	125	Bool_t SetRunQA(TString detAndAction="ALL:ALL") ;
	126	void SetQAWriteExpert(AliQAv1::DETECTORINDEX_t det) { AliQAManager::QAManager()->SetWriteExpert(det) ; }
	127	void SetQARefDefaultStorage(const char* uri);
	128	void InitQA();
	129	void SetEventSpecie(AliRecoParam::EventSpecie_t es) { fEventSpecie = es ; }
	130	void SetWriteQAExpert() { fWriteQAExpertData = kTRUE ; }
	131
	132	void SetWriteGRPEntry(Bool_t flag = kTRUE) { fWriteGRPEntry = flag; }
	133	void WriteGRPEntry();
	134	void UseVertexFromCDB() {fUseVertexFromCDB = kTRUE;}
	135	void UseMagFieldFromGRP() {fUseMagFieldFromGRP = kTRUE;}
	136	void SetGRPWriteLocation(char* loc) {fGRPWriteLocation = loc;}
	137
	138	private:
	139
	140	AliSimulation(const AliSimulation&); // Not implemented
	141	AliSimulation& operator = (const AliSimulation&); // Not implemented
	142
	143	void InitCDB();
	144	void InitRunNumber();
	145	void SetCDBLock();
	146	Bool_t SetRunNumberFromData();
	147	AliRunLoader* LoadRun(const char* mode = "UPDATE") const;
	148	Int_t GetNSignalPerBkgrd(Int_t nEvents = 0) const;
	149	Bool_t IsSelected(TString detName, TString& detectors) const;
	150
	151	static AliSimulation *fgInstance; // Static pointer to object
	152
	153	Bool_t fRunGeneration; // generate prim. particles or not
	154	Bool_t fRunSimulation; // simulate detectors (hits) or not
	155	Bool_t fLoadAlignFromCDB; // Load alignment data from CDB and apply it to geometry or not
	156	TString fLoadAlObjsListOfDets; // Load alignment data from CDB for these detectors
	157	TString fMakeSDigits; // create sdigits for these detectors
	158	TString fMakeDigits; // create digits for these detectors
	159	TString fTriggerConfig; // run trigger for these descriptors
	160	TString fMakeDigitsFromHits; // create digits from hits for these detectors
	161	TString fWriteRawData; // write raw data for these detectors
	162	TString fRawDataFileName; // file name for the raw data file
	163	Bool_t fDeleteIntermediateFiles; // delete intermediate raw data files
	164	Bool_t fWriteSelRawData; // write detectors raw data in a separate file accoring to the trigger cluster
	165	Bool_t fStopOnError; // stop or continue on errors
	166
	167	Int_t fNEvents; // number of events
	168	TString fConfigFileName; // name of the config file
	169	TString fGAliceFileName; // name of the galice file
	170	TObjArray fEventsPerFile; // number of events per file for given detectors and data types
	171
	172	TObjArray* fBkgrdFileNames; // names of background files for merging
	173	TObjArray* fAlignObjArray; // array with the alignment objects to be applied to the geometry
	174	Bool_t fUseBkgrdVertex; // use vertex from background in case of merging
	175	Bool_t fRegionOfInterest; // digitization in region of interest
	176
	177	TString fCDBUri; //! Uri of the default CDB storage
	178	TString fQARefUri; //! Uri of the default QA reference storage
	179	TObjArray fSpecCDBUri; //! Array with detector specific CDB storages
	180	Int_t fRun; //! Run number, will be passed to CDB and gAlice!!
	181	Int_t fSeed; //! Seed for random number generator
	182	Bool_t fInitCDBCalled; //! flag to check if CDB storages are already initialized
	183	Bool_t fInitRunNumberCalled; //! flag to check if run number is already initialized
	184	Bool_t fSetRunNumberFromDataCalled; //! flag to check if run number is already loaded from run loader
	185
	186	Bool_t fEmbeddingFlag; // Flag for embedding
	187	AliLego *fLego; //! Pointer to aliLego object if it exists
	188	// OCDB
	189	ULong_t fKey; //! current CDB key
	190	Bool_t fUseVertexFromCDB; // Flag to use Vertex from CDB
	191	Bool_t fUseMagFieldFromGRP; // Use magnetic field settings from GRP
	192	TString fGRPWriteLocation; // Location to write the GRP entry from simulation
	193
	194	//QA stuff
	195	static const Int_t fgkNDetectors = 15 ; // number of detectors
	196	static const char * fgkDetectorName[fgkNDetectors] ; // names of detectors
	197	TString fQADetectors ; // list of detectors to be QA'ed
	198	TString fQATasks ; // list of QA tasks to be performed
	199	Bool_t fRunQA ; // Runs the QA at the end of simulation
	200	AliRecoParam::EventSpecie_t fEventSpecie ; // type of event (see AliRecoParam::EventSpecie_t)
	201	Bool_t fWriteQAExpertData ; //! decides wheter or not to write experts QA data; true by default
	202
	203	TString fGeometryFile; // Geometry file
	204
	205	//HLT
	206	TString fRunHLT; //! HLT options, HLT is disabled if empty, default='default'
	207	AliHLTSimulation* fpHLT; //! The instance of HLT simulation
	208
	209	Bool_t fWriteGRPEntry; // Write or not GRP entry corresponding to the settings in Config.C
	210
	211	ClassDef(AliSimulation, 11) // class for running generation, simulation and digitization
	212	};
	213
	214	#endif