Changing once more (hopefully we get it correct this time...) the logic to trig the...
[u/mrichter/AliRoot.git] / ITS / AliITSreconstruction.cxx
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  
16 /* $Id$ */
17 /////////////////////////////////////////////////////////////////////////
18 //                                                                     //
19 // Class for ITS RecPoint reconstruction                               //
20 //                                                                     //
21 ////////////////////////////////////////////////////////////////////////
22
23 #include <TString.h>
24 #include "AliRun.h"
25 #include "AliRunLoader.h"
26 #include "AliITSDetTypeRec.h"
27 #include "AliITSLoader.h"
28 #include "AliITSreconstruction.h"
29 #include "AliITSgeom.h"
30
31
32 ClassImp(AliITSreconstruction)
33
34 //______________________________________________________________________
35 AliITSreconstruction::AliITSreconstruction():
36  fInit(kFALSE),
37  fEnt(0),
38  fEnt0(0),
39  fDetTypeRec(0x0),
40  fDfArp(kFALSE),
41  fITSgeom(0x0),
42  fLoader(0x0),
43  fRunLoader(0x0)
44 {
45     // Default constructor.
46     // Inputs:
47     //  none.
48     // Outputs:
49     //   none.
50     // Return:
51     //    A zero-ed constructed AliITSreconstruction class.
52     fDet[0] = fDet[1] = fDet[2] = kTRUE;
53 }
54 //______________________________________________________________________
55
56 AliITSreconstruction::AliITSreconstruction(AliRunLoader *rl):
57  fInit(kFALSE),
58  fEnt(0),
59  fEnt0(0),
60  fDetTypeRec(0x0),
61  fDfArp(kFALSE),
62  fITSgeom(0x0),
63  fLoader(0x0),
64  fRunLoader(rl)
65 {
66   fDet[0] = fDet[1] = fDet[2] = kTRUE;
67 }
68 //______________________________________________________________________
69 AliITSreconstruction::AliITSreconstruction(const char* filename):
70  fInit(kFALSE),
71  fEnt(0),
72  fEnt0(0),
73  fDetTypeRec(0x0),
74  fDfArp(kFALSE),
75  fITSgeom(0x0),
76  fLoader(0x0),
77  fRunLoader(0x0)
78 {
79     // Standard constructor.
80     // Inputs:
81     //  const char* filename    filename containing the digits to be
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.
86     // Outputs:
87     //   none.
88     // Return:
89     //    A standardly constructed AliITSreconstruction class.
90
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      }
99
100 }
101
102 //______________________________________________________________________
103 AliITSreconstruction::AliITSreconstruction(const AliITSreconstruction &rec):TTask(rec),
104 fInit(rec.fInit),
105 fEnt(rec.fEnt),
106 fEnt0(rec.fEnt0),
107 fDetTypeRec(rec.fDetTypeRec),
108 fDfArp(rec.fDfArp),
109 fITSgeom(rec.fITSgeom),
110 fLoader(rec.fLoader),
111 fRunLoader(rec.fRunLoader)
112 {
113     // Copy constructor. 
114
115   
116 }
117
118 //______________________________________________________________________
119 AliITSreconstruction& AliITSreconstruction::operator=(const AliITSreconstruction& source){
120     // Assignment operator. 
121   this->~AliITSreconstruction();
122   new(this) AliITSreconstruction(source);
123   return *this;
124
125 }
126
127 //______________________________________________________________________
128 AliITSreconstruction::~AliITSreconstruction(){
129     //    A destroyed AliITSreconstruction class.
130     
131     //fITS      = 0;
132     delete fRunLoader;
133     
134 }
135 //______________________________________________________________________
136 Bool_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
144     Info("Init","");
145     if (fRunLoader == 0x0)
146      {
147        Error("Init","Run Loader is NULL");
148        return kFALSE;
149      }
150     //  fRunLoader->LoadgAlice();
151     //   fRunLoader->LoadHeader();  
152
153     fLoader = (AliITSLoader*) fRunLoader->GetLoader("ITSLoader");
154     if(!fLoader) {
155       Error("Init","ITS loader not found");
156       fInit = kFALSE;
157     }
158
159     // Now ready to init.
160  
161     //fRunLoader->CdGAFile();
162     fITSgeom = fLoader->GetITSgeom();
163
164     fDetTypeRec = new AliITSDetTypeRec();
165     fDetTypeRec->SetITSgeom(fITSgeom);
166     fDetTypeRec->SetDefaults();
167     fDet[0] = fDet[1] = fDet[2] = kTRUE;
168     fEnt0 = 0;
169
170     fEnt = Int_t(fRunLoader->GetNumberOfEvents());
171
172     fLoader->LoadDigits("read");
173     fLoader->LoadRecPoints("recreate");
174     if (fLoader->TreeR() == 0x0) fLoader->MakeTree("R");
175  
176     fDetTypeRec->SetTreeAddressD(fLoader->TreeD());
177     fDetTypeRec->MakeBranchR(fLoader->TreeR());
178     fDetTypeRec->SetTreeAddressR(fLoader->TreeR());
179
180     fInit = InitRec();
181
182     Info("Init","  Done\n\n\n");
183
184     return fInit;
185 }
186 //______________________________________________________________________
187 Bool_t AliITSreconstruction::InitRec(){
188     // Sets up Reconstruction part of AliITSDetType..
189     // Inputs:
190     //      none.
191     // Outputs:
192     //      none.
193     // Return:
194     //      none.
195
196
197   fDetTypeRec->SetDefaultClusterFindersV2();
198   Info("InitRec","    Done\n");
199   return kTRUE;
200 }
201 //______________________________________________________________________ 
202 void 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;
211     Int_t evnt;
212
213     if(strstr(opt,"All")||strstr(opt,"ALL")||strstr(opt,"ITS")||opt==0){
214       fDet[0] = fDet[1] = fDet[2] = kTRUE;
215       lopt = "All";
216     }else{
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;
223     } // end if strstr(opt,...)
224
225     if(!fInit){
226       cout << "Initilization Failed, Can't run Exec." << endl;
227       return;
228     } // end if !fInit
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");
237       fDetTypeRec->MakeBranchR(0);
238       fDetTypeRec->SetTreeAddressR(fLoader->TreeR());
239       fDetTypeRec->SetTreeAddressD(fLoader->TreeD());
240       fDetTypeRec->DigitsToRecPoints(fLoader->TreeD(),fLoader->TreeR(),0,lopt);
241     } // end for evnt
242 }
243 //______________________________________________________________________ 
244 void AliITSreconstruction::SetOutputFile(TString filename){
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   }
256 }