1 change to improve tracking in machine magnetic field + some DQM stuff
[u/mrichter/AliRoot.git] / ITS / AliITSreconstruction.cxx
CommitLineData
e69020a0 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15
88cb7938 16/* $Id$ */
8ace09b6 17/////////////////////////////////////////////////////////////////////////
18// //
7d62fb64 19// Class for ITS RecPoint reconstruction //
8ace09b6 20// //
21////////////////////////////////////////////////////////////////////////
b9d0a01d 22
e69020a0 23#include <TString.h>
e69020a0 24#include "AliRun.h"
88cb7938 25#include "AliRunLoader.h"
7d62fb64 26#include "AliITSDetTypeRec.h"
88cb7938 27#include "AliITSLoader.h"
e69020a0 28#include "AliITSreconstruction.h"
e69020a0 29#include "AliITSgeom.h"
30
7d62fb64 31
e69020a0 32ClassImp(AliITSreconstruction)
33
34//______________________________________________________________________
88cb7938 35AliITSreconstruction::AliITSreconstruction():
36 fInit(kFALSE),
37 fEnt(0),
38 fEnt0(0),
8221b41b 39 fDetTypeRec(0x0),
88cb7938 40 fDfArp(kFALSE),
8221b41b 41 fITSgeom(0x0),
88cb7938 42 fLoader(0x0),
43 fRunLoader(0x0)
44{
e69020a0 45 // Default constructor.
46 // Inputs:
47 // none.
48 // Outputs:
49 // none.
50 // Return:
51 // A zero-ed constructed AliITSreconstruction class.
e69020a0 52 fDet[0] = fDet[1] = fDet[2] = kTRUE;
4d24e763 53}
54//______________________________________________________________________
4d24e763 55
88cb7938 56AliITSreconstruction::AliITSreconstruction(AliRunLoader *rl):
57 fInit(kFALSE),
58 fEnt(0),
59 fEnt0(0),
8221b41b 60 fDetTypeRec(0x0),
88cb7938 61 fDfArp(kFALSE),
8221b41b 62 fITSgeom(0x0),
88cb7938 63 fLoader(0x0),
64 fRunLoader(rl)
65{
66 fDet[0] = fDet[1] = fDet[2] = kTRUE;
e69020a0 67}
68//______________________________________________________________________
88cb7938 69AliITSreconstruction::AliITSreconstruction(const char* filename):
70 fInit(kFALSE),
71 fEnt(0),
72 fEnt0(0),
8221b41b 73 fDetTypeRec(0x0),
88cb7938 74 fDfArp(kFALSE),
8221b41b 75 fITSgeom(0x0),
88cb7938 76 fLoader(0x0),
77 fRunLoader(0x0)
78{
e69020a0 79 // Standard constructor.
80 // Inputs:
81 // const char* filename filename containing the digits to be
3e9ba728 82 // reconstructed. If filename = 0 (nil)
83 // then no file is opened but a file is
84 // assumed to already be opened. This
85 // already opened file will be used.
e69020a0 86 // Outputs:
87 // none.
88 // Return:
89 // A standardly constructed AliITSreconstruction class.
90
88cb7938 91 fDet[0] = fDet[1] = fDet[2] = kTRUE;
92
93 fRunLoader = AliRunLoader::Open(filename);
94 if (fRunLoader == 0x0)
95 {
96 Error("AliITSreconstruction","Can not load the session",filename);
97 return;
98 }
88cb7938 99
e69020a0 100}
8221b41b 101
e69020a0 102//______________________________________________________________________
8221b41b 103AliITSreconstruction::AliITSreconstruction(const AliITSreconstruction &rec):TTask(rec),
104fInit(rec.fInit),
105fEnt(rec.fEnt),
106fEnt0(rec.fEnt0),
107fDetTypeRec(rec.fDetTypeRec),
108fDfArp(rec.fDfArp),
109fITSgeom(rec.fITSgeom),
110fLoader(rec.fLoader),
111fRunLoader(rec.fRunLoader)
112{
7d62fb64 113 // Copy constructor.
114
7d62fb64 115
116}
8221b41b 117
7d62fb64 118//______________________________________________________________________
8221b41b 119AliITSreconstruction& AliITSreconstruction::operator=(const AliITSreconstruction& source){
120 // Assignment operator.
121 this->~AliITSreconstruction();
122 new(this) AliITSreconstruction(source);
123 return *this;
124
7d62fb64 125}
126
127//______________________________________________________________________
e69020a0 128AliITSreconstruction::~AliITSreconstruction(){
e69020a0 129 // A destroyed AliITSreconstruction class.
7d62fb64 130
131 //fITS = 0;
88cb7938 132 delete fRunLoader;
7d62fb64 133
e69020a0 134}
135//______________________________________________________________________
136Bool_t AliITSreconstruction::Init(){
137 // Class Initilizer.
138 // Inputs:
139 // none.
140 // Outputs:
141 // none.
142 // Return:
143 // kTRUE if no errors initilizing this class occurse else kFALSE
88cb7938 144 Info("Init","");
145 if (fRunLoader == 0x0)
146 {
147 Error("Init","Run Loader is NULL");
148 return kFALSE;
149 }
60b9526b 150 // fRunLoader->LoadgAlice();
151 // fRunLoader->LoadHeader();
88cb7938 152
153 fLoader = (AliITSLoader*) fRunLoader->GetLoader("ITSLoader");
154 if(!fLoader) {
155 Error("Init","ITS loader not found");
156 fInit = kFALSE;
157 }
158
e69020a0 159 // Now ready to init.
7d62fb64 160
60b9526b 161 //fRunLoader->CdGAFile();
c7dca079 162 fITSgeom = fLoader->GetITSgeom();
e69020a0 163
6cae184e 164 fDetTypeRec = new AliITSDetTypeRec();
165 fDetTypeRec->SetITSgeom(fITSgeom);
7d62fb64 166 fDetTypeRec->SetDefaults();
e69020a0 167 fDet[0] = fDet[1] = fDet[2] = kTRUE;
168 fEnt0 = 0;
88cb7938 169
60b9526b 170 fEnt = Int_t(fRunLoader->GetNumberOfEvents());
88cb7938 171
172 fLoader->LoadDigits("read");
173 fLoader->LoadRecPoints("recreate");
88cb7938 174 if (fLoader->TreeR() == 0x0) fLoader->MakeTree("R");
7d62fb64 175
6cae184e 176 fDetTypeRec->SetTreeAddressD(fLoader->TreeD());
177 fDetTypeRec->MakeBranchR(fLoader->TreeR());
7d62fb64 178 fDetTypeRec->SetTreeAddressR(fLoader->TreeR());
179
e69020a0 180 fInit = InitRec();
88cb7938 181
182 Info("Init"," Done\n\n\n");
183
e69020a0 184 return fInit;
185}
186//______________________________________________________________________
187Bool_t AliITSreconstruction::InitRec(){
188 // Sets up Reconstruction part of AliITSDetType..
189 // Inputs:
190 // none.
191 // Outputs:
192 // none.
193 // Return:
194 // none.
5d2c2f86 195
196
197 fDetTypeRec->SetDefaultClusterFindersV2();
198 Info("InitRec"," Done\n");
199 return kTRUE;
e69020a0 200}
201//______________________________________________________________________
202void AliITSreconstruction::Exec(const Option_t *opt){
203 // Main reconstruction function.
204 // Inputs:
205 // Option_t * opt list of subdetector to digitize. =0 all.
206 // Outputs:
207 // none.
208 // Return:
209 // none.
210 Option_t *lopt;
88cb7938 211 Int_t evnt;
e69020a0 212
213 if(strstr(opt,"All")||strstr(opt,"ALL")||strstr(opt,"ITS")||opt==0){
88cb7938 214 fDet[0] = fDet[1] = fDet[2] = kTRUE;
215 lopt = "All";
e69020a0 216 }else{
88cb7938 217 fDet[0] = fDet[1] = fDet[2] = kFALSE;
218 if(strstr(opt,"SPD")) fDet[kSPD] = kTRUE;
219 if(strstr(opt,"SDD")) fDet[kSDD] = kTRUE;
220 if(strstr(opt,"SSD")) fDet[kSSD] = kTRUE;
221 if(fDet[kSPD] && fDet[kSDD] && fDet[kSSD]) lopt = "All";
222 else lopt = opt;
e69020a0 223 } // end if strstr(opt,...)
224
225 if(!fInit){
88cb7938 226 cout << "Initilization Failed, Can't run Exec." << endl;
227 return;
e69020a0 228 } // end if !fInit
88cb7938 229 for(evnt=0;evnt<fEnt;evnt++)
230 {
231 Info("Exec","");
232 Info("Exec","Processing Event %d",evnt);
233 Info("Exec","");
234
235 fRunLoader->GetEvent(evnt);
236 if (fLoader->TreeR() == 0x0) fLoader->MakeTree("R");
7d62fb64 237 fDetTypeRec->MakeBranchR(0);
7d62fb64 238 fDetTypeRec->SetTreeAddressR(fLoader->TreeR());
239 fDetTypeRec->SetTreeAddressD(fLoader->TreeD());
6cae184e 240 fDetTypeRec->DigitsToRecPoints(fLoader->TreeD(),fLoader->TreeR(),0,lopt);
e69020a0 241 } // end for evnt
f243fbe6 242}
243//______________________________________________________________________
244void AliITSreconstruction::SetOutputFile(TString filename){
999cc3eb 245 // Set a new file name for recpoints.
246 // It must be called before Init()
247 if(!fLoader)fLoader = (AliITSLoader*) fRunLoader->GetLoader("ITSLoader");
248 if(fLoader){
249 Info("SetOutputFile","name for rec points is %s",filename.Data());
250 fLoader->SetRecPointsFileName(filename);
251 }
252 else {
253 Error("SetOutputFile",
254 "ITS loader not available. Not possible to set name: %s",filename.Data());
255 }
e69020a0 256}