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

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

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
// class for running the reconstruction                                      //
// Clusters and tracks are created for all detectors and all events by       //
// typing:                                                                   //
//                                                                           //
//   AliReconstruction rec;                                                  //
//   rec.Run();                                                              //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////


#include <TNamed.h>
#include <TString.h>
#include <TObjArray.h>

class AliReconstructor;
class AliRunLoader;
class AliRawReader;
class AliLoader;
class AliTracker;
class AliVertexer;
class AliESDVertex;
class AliESDEvent;
class TFile;
class TTree;
class AliQADataMaker;

class AliReconstruction: public TNamed {
public:
  AliReconstruction(const char* gAliceFilename = "galice.root",
  		    const char* cdbUri = "local://$ALICE_ROOT",
		    const char* name = "AliReconstruction", 
		    const char* title = "reconstruction");
  AliReconstruction(const AliReconstruction& rec);
  AliReconstruction& operator = (const AliReconstruction& rec);
  virtual ~AliReconstruction();

  void           SetGAliceFile(const char* fileName);
  void           SetInput(const char* input) {fInput = input;};
  void           SetEquipmentIdMap(const char *mapFile) {fEquipIdMap = mapFile;};
  void           SetEventRange(Int_t firstEvent = 0, Int_t lastEvent = -1) 
    {fFirstEvent = firstEvent; fLastEvent = lastEvent;};
  void           SetNumberOfEventsPerFile(UInt_t nEvents)
    {fNumberOfEventsPerFile = nEvents;};
  void           SetOption(const char* detector, const char* option);

  void           SetRunLocalReconstruction(const char* detectors) {
    fRunLocalReconstruction = detectors;};
  void           SetRunTracking(const char* detectors) {
    fRunTracking = detectors;};
  void           SetFillESD(const char* detectors) {fFillESD = detectors;};
  void           SetRunReconstruction(const char* detectors) {
    SetRunLocalReconstruction(detectors); 
    SetRunTracking(detectors);
    SetFillESD(detectors);};
  void           SetUseTrackingErrorsForAlignment(const char* detectors) 
    {fUseTrackingErrorsForAlignment = detectors;};
  void           SetLoadAlignFromCDB(Bool_t load)  {fLoadAlignFromCDB = load;};
  void           SetLoadAlignData(const char* detectors) 
    {fLoadAlignData = detectors;};
  void           SetESDParLocation(const char *c){fESDPar = c;}
  TNamed *CopyFileToTNamed(TString fPath,TString fName);
  void  TNamedToFile(TTree* fTree, TString fName);

  //*** Global reconstruction flag setters
  void SetUniformFieldTracking(Bool_t flag=kTRUE){fUniformField=flag;} 
  void SetRunVertexFinder(Bool_t flag=kTRUE) {fRunVertexFinder=flag;};
  void SetRunHLTTracking(Bool_t flag=kTRUE) {fRunHLTTracking=flag;};
  void SetRunV0Finder(Bool_t flag=kTRUE) {fRunV0Finder=flag;};
  void SetRunCascadeFinder(Bool_t flag=kTRUE) {fRunCascadeFinder=flag;};
  void SetStopOnError(Bool_t flag=kTRUE) {fStopOnError=flag;}
  void SetWriteAlignmentData(Bool_t flag=kTRUE){fWriteAlignmentData=flag;}
  void SetWriteESDfriend(Bool_t flag=kTRUE){fWriteESDfriend=flag;}
  void SetWriteAOD(Bool_t flag=kTRUE){fWriteAOD=flag;}
  void SetFillTriggerESD(Bool_t flag=kTRUE){fFillTriggerESD=flag;}
  void SetDiamondProfile(AliESDVertex *dp) {fDiamondProfile=dp;}
		   
  void SetCleanESD(Bool_t flag=kTRUE){fCleanESD=flag;}
  void SetDmax(Float_t d) {fDmax=d;}
  void SetZmax(Float_t z) {fZmax=z;}
  Float_t GetDmax() const {return fDmax;}
  Float_t GetZmax() const {return fZmax;}

  void           SetCheckPointLevel(Int_t checkPointLevel)
    {fCheckPointLevel = checkPointLevel;}
  // CDB storage activation
  void InitCDBStorage();
  void SetDefaultStorage(const char* uri);
  void SetRemoteStorage(const char* uri);
  void SetSpecificStorage(const char* calibType, const char* uri);

  Bool_t SetRunNumber();

  Bool_t MisalignGeometry(const TString& detectors);

  void           SetAlignObjArray(TObjArray *array)
                   {fAlignObjArray = array;
		   fLoadAlignFromCDB = kFALSE;}

  virtual Bool_t Run(const char* input = NULL);

  // Quality Assurance 
  virtual Bool_t RunQA(const char* detectors, AliESDEvent *& esd);
  void    SetQACycles(const char * detector, const Int_t cycles) { fQACycles[GetDetIndex(detector)] = cycles ; }

private:
  Bool_t         RunLocalReconstruction(const TString& detectors);
  Bool_t         RunLocalEventReconstruction(const TString& detectors);
  Bool_t         RunVertexFinder(AliESDEvent*& esd);
  Bool_t         RunHLTTracking(AliESDEvent*& esd);
  Bool_t         RunMuonTracking(AliESDEvent*& esd);
  Bool_t         RunTracking(AliESDEvent*& esd);
  Bool_t         CleanESD(AliESDEvent *esd);
  Bool_t         FillESD(AliESDEvent*& esd, const TString& detectors);
  Bool_t         FillTriggerESD(AliESDEvent*& esd);
  Bool_t         FillRawEventHeaderESD(AliESDEvent*& esd);

  Bool_t         IsSelected(TString detName, TString& detectors) const;
  Bool_t         InitRunLoader();
  AliReconstructor* GetReconstructor(Int_t iDet);
  Bool_t         CreateVertexer();
  Bool_t         CreateTrackers(const TString& detectors);
  void           CleanUp(TFile* file = NULL, TFile* fileOld = NULL);

  Bool_t         ReadESD(AliESDEvent*& esd, const char* recStep) const;
  void           WriteESD(AliESDEvent* esd, const char* recStep) const;

   //==========================================//
  void           ESDFile2AODFile(TFile* esdFile, TFile* aodFile);

  void           WriteAlignmentData(AliESDEvent* esd);

  void           FillRawDataErrorLog(Int_t iEvent, AliESDEvent* esd);

  //Quality Assurance
  Int_t                GetDetIndex(const char * detector);
  AliQADataMaker*      GetQADataMaker(Int_t iDet);
  const Int_t          GetQACycles(const char * detector) { return fQACycles[GetDetIndex(detector)] ; }
  void                 CheckQA() ;

  //*** Global reconstruction flags *******************
  Bool_t         fUniformField;       // uniform field tracking flag
  Bool_t         fRunVertexFinder;    // run the vertex finder
  Bool_t         fRunHLTTracking;     // run the HLT tracking
  Bool_t         fRunMuonTracking;    // run the HLT tracking
  Bool_t         fRunV0Finder;        // run the ESD V0 finder
  Bool_t         fRunCascadeFinder;   // run the ESD cascade finder
  Bool_t         fStopOnError;        // stop or continue on errors
  Bool_t         fWriteAlignmentData; // write track space-points flag
  Bool_t         fWriteESDfriend;     // write ESD friend flag
  Bool_t         fWriteAOD;           // write AOD flag
  Bool_t         fFillTriggerESD;     // fill trigger info into ESD

  //*** Clean ESD flag and parameters *******************
  Bool_t         fCleanESD;      // clean ESD flag
  Float_t        fDmax;          // max. allowed transverse impact parameter 
  Float_t        fZmax;          // max. allowed longitudinal impact parameter 

  TString        fRunLocalReconstruction; // run the local reconstruction for these detectors
  TString        fRunTracking;        // run the tracking for these detectors
  TString        fFillESD;            // fill ESD for these detectors
  TString        fUseTrackingErrorsForAlignment; // for these detectors
  TString        fGAliceFileName;     // name of the galice file
  TString        fInput;              // name of input file or directory
  TString        fEquipIdMap;         // name of file with equipment id map
  Int_t          fFirstEvent;         // index of first event to be reconstr.
  Int_t          fLastEvent;          // index of last event to be reconstr.
  UInt_t         fNumberOfEventsPerFile; // number of events per file in case of raw-data reconstruction
  Int_t          fCheckPointLevel;    // level of ESD check points
  TObjArray      fOptions;            // options for reconstructor objects
  Bool_t         fLoadAlignFromCDB;   // Load alignment data from CDB and apply it to geometry or not
  TString        fLoadAlignData;      // Load alignment data from CDB for these detectors
  TString        fESDPar;             // String where the esd.par is stored, will be attached to the tree         


  AliRunLoader*  fRunLoader;          //! current run loader object
  AliRawReader*  fRawReader;          //! current raw data reader

  static const Int_t fgkNDetectors = 15;   //! number of detectors
  static const char* fgkDetectorName[fgkNDetectors]; //! names of detectors
  AliReconstructor*  fReconstructor[fgkNDetectors];  //! array of reconstructor objects
  AliLoader*     fLoader[fgkNDetectors];   //! detector loaders
  AliVertexer*   fVertexer;                //! vertexer for ITS
  AliTracker*    fTracker[fgkNDetectors];  //! trackers
  AliESDVertex*  fDiamondProfile;          // (x,y) diamond profile for AliVertexerTracks

  TObjArray* 	 fAlignObjArray;      // array with the alignment objects to be applied to the geometry

  TString	 fCDBUri;	      // Uri of the default CDB storage
  TString	 fRemoteCDBUri;	      // Uri of the remote CDB storage
  TObjArray      fSpecCDBUri;         // Array with detector specific CDB storages

  //Quality Assurance
  AliQADataMaker * fQADataMaker[fgkNDetectors];  //! array of QA data maker objects
  Int_t                 fQACycles[fgkNDetectors] ;         // cycle length (# events) over which QA data are accumulated

  ClassDef(AliReconstruction, 15)      // class for running the reconstruction
};

#endif
Commit	Line	Data
596a855f	1	#ifndef ALIRECONSTRUCTION_H
	2	#define ALIRECONSTRUCTION_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
af7ba10c	8	///////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// class for running the reconstruction //
	11	// Clusters and tracks are created for all detectors and all events by //
	12	// typing: //
	13	// //
	14	// AliReconstruction rec; //
	15	// rec.Run(); //
	16	// //
	17	///////////////////////////////////////////////////////////////////////////////
	18
	19
596a855f	20	#include <TNamed.h>
596a855f	21	#include <TString.h>
59697224	22	#include <TObjArray.h>
596a855f	23
b8cd5251	24	class AliReconstructor;
596a855f	25	class AliRunLoader;
b649205a	26	class AliRawReader;
596a855f	27	class AliLoader;
596a855f	28	class AliTracker;
c84a5e9e	29	class AliVertexer;
9178838a	30	class AliESDVertex;
af885e0f	31	class AliESDEvent;
e583c30d	32	class TFile;
46698ae4	33	class TTree;
2e42b4d4	34	class AliQADataMaker;
596a855f	35
	36	class AliReconstruction: public TNamed {
	37	public:
e583c30d	38	AliReconstruction(const char* gAliceFilename = "galice.root",
024cf675	39	const char* cdbUri = "local://$ALICE_ROOT",
e583c30d	40	const char* name = "AliReconstruction",
596a855f	41	const char* title = "reconstruction");
	42	AliReconstruction(const AliReconstruction& rec);
	43	AliReconstruction& operator = (const AliReconstruction& rec);
	44	virtual ~AliReconstruction();
	45
	46	void SetGAliceFile(const char* fileName);
b649205a	47	void SetInput(const char* input) {fInput = input;};
35042093	48	void SetEquipmentIdMap(const char *mapFile) {fEquipIdMap = mapFile;};
b26c3770	49	void SetEventRange(Int_t firstEvent = 0, Int_t lastEvent = -1)
b26c3770	50	{fFirstEvent = firstEvent; fLastEvent = lastEvent;};
973388c2	51	void SetNumberOfEventsPerFile(UInt_t nEvents)
973388c2	52	{fNumberOfEventsPerFile = nEvents;};
efd2085e	53	void SetOption(const char* detector, const char* option);
596a855f	54
59697224	55	void SetRunLocalReconstruction(const char* detectors) {
59697224	56	fRunLocalReconstruction = detectors;};
b8cd5251	57	void SetRunTracking(const char* detectors) {
b8cd5251	58	fRunTracking = detectors;};
596a855f	59	void SetFillESD(const char* detectors) {fFillESD = detectors;};
b8cd5251	60	void SetRunReconstruction(const char* detectors) {
	61	SetRunLocalReconstruction(detectors);
	62	SetRunTracking(detectors);
	63	SetFillESD(detectors);};
48ce48d1	64	void SetUseTrackingErrorsForAlignment(const char* detectors)
48ce48d1	65	{fUseTrackingErrorsForAlignment = detectors;};
6bae477a	66	void SetLoadAlignFromCDB(Bool_t load) {fLoadAlignFromCDB = load;};
	67	void SetLoadAlignData(const char* detectors)
	68	{fLoadAlignData = detectors;};
46698ae4	69	void SetESDParLocation(const char *c){fESDPar = c;}
	70	TNamed *CopyFileToTNamed(TString fPath,TString fName);
	71	void TNamedToFile(TTree* fTree, TString fName);
c84a5e9e	72
1d99986f	73	//*** Global reconstruction flag setters
	74	void SetUniformFieldTracking(Bool_t flag=kTRUE){fUniformField=flag;}
	75	void SetRunVertexFinder(Bool_t flag=kTRUE) {fRunVertexFinder=flag;};
	76	void SetRunHLTTracking(Bool_t flag=kTRUE) {fRunHLTTracking=flag;};
d1683eef	77	void SetRunV0Finder(Bool_t flag=kTRUE) {fRunV0Finder=flag;};
d1683eef	78	void SetRunCascadeFinder(Bool_t flag=kTRUE) {fRunCascadeFinder=flag;};
1d99986f	79	void SetStopOnError(Bool_t flag=kTRUE) {fStopOnError=flag;}
	80	void SetWriteAlignmentData(Bool_t flag=kTRUE){fWriteAlignmentData=flag;}
	81	void SetWriteESDfriend(Bool_t flag=kTRUE){fWriteESDfriend=flag;}
a7807689	82	void SetWriteAOD(Bool_t flag=kTRUE){fWriteAOD=flag;}
b647652d	83	void SetFillTriggerESD(Bool_t flag=kTRUE){fFillTriggerESD=flag;}
9178838a	84	void SetDiamondProfile(AliESDVertex *dp) {fDiamondProfile=dp;}
1d99986f	85
7f68891d	86	void SetCleanESD(Bool_t flag=kTRUE){fCleanESD=flag;}
	87	void SetDmax(Float_t d) {fDmax=d;}
	88	void SetZmax(Float_t z) {fZmax=z;}
	89	Float_t GetDmax() const {return fDmax;}
	90	Float_t GetZmax() const {return fZmax;}
	91
24f7a148	92	void SetCheckPointLevel(Int_t checkPointLevel)
24f7a148	93	{fCheckPointLevel = checkPointLevel;}
024cf675	94	// CDB storage activation
	95	void InitCDBStorage();
	96	void SetDefaultStorage(const char* uri);
00aa02d5	97	void SetRemoteStorage(const char* uri);
c3a7b59a	98	void SetSpecificStorage(const char* calibType, const char* uri);
24f7a148	99
6bae477a	100	Bool_t SetRunNumber();
6bae477a	101
6bae477a	102	Bool_t MisalignGeometry(const TString& detectors);
	103
	104	void SetAlignObjArray(TObjArray *array)
	105	{fAlignObjArray = array;
	106	fLoadAlignFromCDB = kFALSE;}
6bae477a	107
4a33489c	108	virtual Bool_t Run(const char* input = NULL);
596a855f	109
a5fa6165	110	// Quality Assurance
2e42b4d4	111	virtual Bool_t RunQA(const char* detectors, AliESDEvent *& esd);
5b188f2f	112	void SetQACycles(const char * detector, const Int_t cycles) { fQACycles[GetDetIndex(detector)] = cycles ; }
98937d93	113
596a855f	114	private:
59697224	115	Bool_t RunLocalReconstruction(const TString& detectors);
b26c3770	116	Bool_t RunLocalEventReconstruction(const TString& detectors);
af885e0f	117	Bool_t RunVertexFinder(AliESDEvent*& esd);
	118	Bool_t RunHLTTracking(AliESDEvent*& esd);
	119	Bool_t RunMuonTracking(AliESDEvent*& esd);
	120	Bool_t RunTracking(AliESDEvent*& esd);
d64bd07d	121	Bool_t CleanESD(AliESDEvent *esd);
af885e0f	122	Bool_t FillESD(AliESDEvent*& esd, const TString& detectors);
	123	Bool_t FillTriggerESD(AliESDEvent*& esd);
	124	Bool_t FillRawEventHeaderESD(AliESDEvent*& esd);
596a855f	125
e583c30d	126	Bool_t IsSelected(TString detName, TString& detectors) const;
f08fc9f5	127	Bool_t InitRunLoader();
b8cd5251	128	AliReconstructor* GetReconstructor(Int_t iDet);
2257f27e	129	Bool_t CreateVertexer();
b8cd5251	130	Bool_t CreateTrackers(const TString& detectors);
b26c3770	131	void CleanUp(TFile* file = NULL, TFile* fileOld = NULL);
e583c30d	132
af885e0f	133	Bool_t ReadESD(AliESDEvent& esd, const char recStep) const;
af885e0f	134	void WriteESD(AliESDEvent* esd, const char* recStep) const;
24f7a148	135
593df3b8	136	//==========================================//
f29f1726	137	void ESDFile2AODFile(TFile* esdFile, TFile* aodFile);
f3a97c86	138
af885e0f	139	void WriteAlignmentData(AliESDEvent* esd);
f3a97c86	140
af885e0f	141	void FillRawDataErrorLog(Int_t iEvent, AliESDEvent* esd);
f3a97c86	142
a5fa6165	143	//Quality Assurance
5b188f2f	144	Int_t GetDetIndex(const char * detector);
2e42b4d4	145	AliQADataMaker* GetQADataMaker(Int_t iDet);
5b188f2f	146	const Int_t GetQACycles(const char * detector) { return fQACycles[GetDetIndex(detector)] ; }
8661738e	147	void CheckQA() ;
a5fa6165	148
1d99986f	149	//* Global reconstruction flags *****************
c84a5e9e	150	Bool_t fUniformField; // uniform field tracking flag
2257f27e	151	Bool_t fRunVertexFinder; // run the vertex finder
1f46a9ae	152	Bool_t fRunHLTTracking; // run the HLT tracking
d1683eef	153	Bool_t fRunMuonTracking; // run the HLT tracking
	154	Bool_t fRunV0Finder; // run the ESD V0 finder
	155	Bool_t fRunCascadeFinder; // run the ESD cascade finder
1d99986f	156	Bool_t fStopOnError; // stop or continue on errors
	157	Bool_t fWriteAlignmentData; // write track space-points flag
	158	Bool_t fWriteESDfriend; // write ESD friend flag
a7807689	159	Bool_t fWriteAOD; // write AOD flag
b647652d	160	Bool_t fFillTriggerESD; // fill trigger info into ESD
1d99986f	161
7f68891d	162	//* Clean ESD flag and parameters *****************
	163	Bool_t fCleanESD; // clean ESD flag
	164	Float_t fDmax; // max. allowed transverse impact parameter
	165	Float_t fZmax; // max. allowed longitudinal impact parameter
	166
1d99986f	167	TString fRunLocalReconstruction; // run the local reconstruction for these detectors
b8cd5251	168	TString fRunTracking; // run the tracking for these detectors
596a855f	169	TString fFillESD; // fill ESD for these detectors
48ce48d1	170	TString fUseTrackingErrorsForAlignment; // for these detectors
596a855f	171	TString fGAliceFileName; // name of the galice file
b649205a	172	TString fInput; // name of input file or directory
35042093	173	TString fEquipIdMap; // name of file with equipment id map
b26c3770	174	Int_t fFirstEvent; // index of first event to be reconstr.
b26c3770	175	Int_t fLastEvent; // index of last event to be reconstr.
973388c2	176	UInt_t fNumberOfEventsPerFile; // number of events per file in case of raw-data reconstruction
24f7a148	177	Int_t fCheckPointLevel; // level of ESD check points
b8cd5251	178	TObjArray fOptions; // options for reconstructor objects
6bae477a	179	Bool_t fLoadAlignFromCDB; // Load alignment data from CDB and apply it to geometry or not
6bae477a	180	TString fLoadAlignData; // Load alignment data from CDB for these detectors
46698ae4	181	TString fESDPar; // String where the esd.par is stored, will be attached to the tree
46698ae4	182
596a855f	183
596a855f	184	AliRunLoader* fRunLoader; //! current run loader object
b649205a	185	AliRawReader* fRawReader; //! current raw data reader
596a855f	186
482070f2	187	static const Int_t fgkNDetectors = 15; //! number of detectors
c757bafd	188	static const char* fgkDetectorName[fgkNDetectors]; //! names of detectors
b8cd5251	189	AliReconstructor* fReconstructor[fgkNDetectors]; //! array of reconstructor objects
	190	AliLoader* fLoader[fgkNDetectors]; //! detector loaders
	191	AliVertexer* fVertexer; //! vertexer for ITS
	192	AliTracker* fTracker[fgkNDetectors]; //! trackers
9178838a	193	AliESDVertex* fDiamondProfile; // (x,y) diamond profile for AliVertexerTracks
59697224	194
6bae477a	195	TObjArray* fAlignObjArray; // array with the alignment objects to be applied to the geometry
98937d93	196
024cf675	197	TString fCDBUri; // Uri of the default CDB storage
00aa02d5	198	TString fRemoteCDBUri; // Uri of the remote CDB storage
ec92bee0	199	TObjArray fSpecCDBUri; // Array with detector specific CDB storages
024cf675	200
a5fa6165	201	//Quality Assurance
2e42b4d4	202	AliQADataMaker * fQADataMaker[fgkNDetectors]; //! array of QA data maker objects
5b188f2f	203	Int_t fQACycles[fgkNDetectors] ; // cycle length (# events) over which QA data are accumulated
a5fa6165	204
7f68891d	205	ClassDef(AliReconstruction, 15) // class for running the reconstruction
596a855f	206	};
	207
	208	#endif