]>
Commit | Line | Data |
---|---|---|
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 | ||
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 | ||
20 | #include <TNamed.h> | |
21 | #include <TString.h> | |
22 | #include <TObjArray.h> | |
23 | ||
24 | class AliReconstructor; | |
25 | class AliRunLoader; | |
26 | class AliRawReader; | |
27 | class AliLoader; | |
28 | class AliTracker; | |
29 | class AliVertexer; | |
30 | class AliESD; | |
31 | class TFile; | |
32 | ||
33 | class AliRunTag; | |
34 | class AliLHCTag; | |
35 | class AliDetectorTag; | |
36 | class AliEventTag; | |
37 | ||
38 | ||
39 | class AliReconstruction: public TNamed { | |
40 | public: | |
41 | AliReconstruction(const char* gAliceFilename = "galice.root", | |
42 | const char* cdbUri = "local://$ALICE_ROOT", | |
43 | const char* name = "AliReconstruction", | |
44 | const char* title = "reconstruction"); | |
45 | AliReconstruction(const AliReconstruction& rec); | |
46 | AliReconstruction& operator = (const AliReconstruction& rec); | |
47 | virtual ~AliReconstruction(); | |
48 | ||
49 | void SetGAliceFile(const char* fileName); | |
50 | void SetInput(const char* input) {fInput = input;}; | |
51 | void SetEventRange(Int_t firstEvent = 0, Int_t lastEvent = -1) | |
52 | {fFirstEvent = firstEvent; fLastEvent = lastEvent;}; | |
53 | void SetOption(const char* detector, const char* option); | |
54 | ||
55 | void SetRunLocalReconstruction(const char* detectors) { | |
56 | fRunLocalReconstruction = detectors;}; | |
57 | void SetRunTracking(const char* detectors) { | |
58 | fRunTracking = detectors;}; | |
59 | void SetFillESD(const char* detectors) {fFillESD = detectors;}; | |
60 | void SetRunReconstruction(const char* detectors) { | |
61 | SetRunLocalReconstruction(detectors); | |
62 | SetRunTracking(detectors); | |
63 | SetFillESD(detectors);}; | |
64 | void SetLoadAlignFromCDB(Bool_t load) {fLoadAlignFromCDB = load;}; | |
65 | void SetLoadAlignData(const char* detectors) | |
66 | {fLoadAlignData = detectors;}; | |
67 | ||
68 | ||
69 | //*** Global reconstruction flag setters | |
70 | void SetUniformFieldTracking(Bool_t flag=kTRUE){fUniformField=flag;} | |
71 | void SetRunVertexFinder(Bool_t flag=kTRUE) {fRunVertexFinder=flag;}; | |
72 | void SetRunHLTTracking(Bool_t flag=kTRUE) {fRunHLTTracking=flag;}; | |
73 | void SetStopOnError(Bool_t flag=kTRUE) {fStopOnError=flag;} | |
74 | void SetWriteAlignmentData(Bool_t flag=kTRUE){fWriteAlignmentData=flag;} | |
75 | void SetWriteESDfriend(Bool_t flag=kTRUE){fWriteESDfriend=flag;} | |
76 | void SetFillTriggerESD(Bool_t flag=kTRUE){fFillTriggerESD=flag;} | |
77 | ||
78 | ||
79 | void SetCheckPointLevel(Int_t checkPointLevel) | |
80 | {fCheckPointLevel = checkPointLevel;} | |
81 | // CDB storage activation | |
82 | void InitCDBStorage(); | |
83 | void SetDefaultStorage(const char* uri); | |
84 | void SetSpecificStorage(const char* detName, const char* uri); | |
85 | ||
86 | Bool_t SetRunNumber(); | |
87 | ||
88 | Bool_t SetAlignObjArraySingleDet(const char* detName); | |
89 | Bool_t MisalignGeometry(const TString& detectors); | |
90 | ||
91 | void SetAlignObjArray(TObjArray *array) | |
92 | {fAlignObjArray = array; | |
93 | fLoadAlignFromCDB = kFALSE;} | |
94 | Bool_t ApplyAlignObjsToGeom(TObjArray* alObjArray); | |
95 | ||
96 | virtual Bool_t Run(const char* input, | |
97 | Int_t firstEvent, Int_t lastEvent = -1); | |
98 | Bool_t Run(const char* input = NULL) | |
99 | {return Run(input, fFirstEvent, fLastEvent);}; | |
100 | Bool_t Run(Int_t firstEvent, Int_t lastEvent = -1) | |
101 | {return Run(NULL, firstEvent, lastEvent);}; | |
102 | ||
103 | ||
104 | private: | |
105 | Bool_t RunLocalReconstruction(const TString& detectors); | |
106 | Bool_t RunLocalEventReconstruction(const TString& detectors); | |
107 | Bool_t RunVertexFinder(AliESD*& esd); | |
108 | Bool_t RunHLTTracking(AliESD*& esd); | |
109 | Bool_t RunTracking(AliESD*& esd); | |
110 | Bool_t FillESD(AliESD*& esd, const TString& detectors); | |
111 | Bool_t FillTriggerESD(AliESD*& esd); | |
112 | ||
113 | Bool_t IsSelected(TString detName, TString& detectors) const; | |
114 | Bool_t InitRunLoader(); | |
115 | AliReconstructor* GetReconstructor(Int_t iDet); | |
116 | Bool_t CreateVertexer(); | |
117 | Bool_t CreateTrackers(const TString& detectors); | |
118 | void CleanUp(TFile* file = NULL, TFile* fileOld = NULL); | |
119 | ||
120 | Bool_t ReadESD(AliESD*& esd, const char* recStep) const; | |
121 | void WriteESD(AliESD* esd, const char* recStep) const; | |
122 | ||
123 | ||
124 | //===========================================// | |
125 | void CreateTag(TFile* file); | |
126 | //==========================================// | |
127 | ||
128 | void WriteAlignmentData(AliESD* esd); | |
129 | ||
130 | ||
131 | //*** Global reconstruction flags ******************* | |
132 | Bool_t fUniformField; // uniform field tracking flag | |
133 | Bool_t fRunVertexFinder; // run the vertex finder | |
134 | Bool_t fRunHLTTracking; // run the HLT tracking | |
135 | Bool_t fStopOnError; // stop or continue on errors | |
136 | Bool_t fWriteAlignmentData; // write track space-points flag | |
137 | Bool_t fWriteESDfriend; // write ESD friend flag | |
138 | Bool_t fFillTriggerESD; // fill trigger info into ESD | |
139 | ||
140 | TString fRunLocalReconstruction; // run the local reconstruction for these detectors | |
141 | TString fRunTracking; // run the tracking for these detectors | |
142 | TString fFillESD; // fill ESD for these detectors | |
143 | TString fGAliceFileName; // name of the galice file | |
144 | TString fInput; // name of input file or directory | |
145 | Int_t fFirstEvent; // index of first event to be reconstr. | |
146 | Int_t fLastEvent; // index of last event to be reconstr. | |
147 | Int_t fCheckPointLevel; // level of ESD check points | |
148 | TObjArray fOptions; // options for reconstructor objects | |
149 | Bool_t fLoadAlignFromCDB; // Load alignment data from CDB and apply it to geometry or not | |
150 | TString fLoadAlignData; // Load alignment data from CDB for these detectors | |
151 | ||
152 | AliRunLoader* fRunLoader; //! current run loader object | |
153 | AliRawReader* fRawReader; //! current raw data reader | |
154 | ||
155 | static const Int_t fgkNDetectors = 15; //! number of detectors | |
156 | static const char* fgkDetectorName[fgkNDetectors]; //! names of detectors | |
157 | AliReconstructor* fReconstructor[fgkNDetectors]; //! array of reconstructor objects | |
158 | AliLoader* fLoader[fgkNDetectors]; //! detector loaders | |
159 | AliVertexer* fVertexer; //! vertexer for ITS | |
160 | AliTracker* fTracker[fgkNDetectors]; //! trackers | |
161 | ||
162 | TObjArray* fAlignObjArray; // array with the alignment objects to be applied to the geometry | |
163 | ||
164 | TString fCDBUri; // Uri of the default CDB storage | |
165 | ||
166 | ClassDef(AliReconstruction, 7) // class for running the reconstruction | |
167 | }; | |
168 | ||
169 | #endif |