Classes moved to STEERBase.
[u/mrichter/AliRoot.git] / TRD / AliTRDReconstructor.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 //                                                                           //
20 // Class for TRD reconstruction                                              //
21 //                                                                           //
22 ///////////////////////////////////////////////////////////////////////////////
23
24 #include <TFile.h>
25
26 #include "AliRunLoader.h"
27 #include "AliRawReader.h"
28 #include "AliLog.h"
29 #include "AliESDTrdTrack.h"
30 #include "AliESDEvent.h"
31
32 #include "AliTRDReconstructor.h"
33 #include "AliTRDclusterizerV1.h"
34 #include "AliTRDclusterizerV2.h"
35 //#include "AliTRDclusterizerV2xMP.h"
36 #include "AliTRDtracker.h"
37 #include "AliTRDpidESD.h"
38 #include "AliTRDgtuTrack.h"
39 #include "AliTRDrawData.h"
40 #include "AliTRDdigitsManager.h"
41
42 ClassImp(AliTRDReconstructor)
43
44 Bool_t AliTRDReconstructor::fgkSeedingOn  = kFALSE;
45 Int_t  AliTRDReconstructor::fgStreamLevel = 0;      // Stream (debug) level
46
47 //_____________________________________________________________________________
48 void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
49                                       , TTree *digitsTree) const
50 {
51   //
52   // Convert raw data digits into digit objects in a root tree
53   //
54
55   AliInfo("Convert raw data digits into digit objects [RawReader -> Digit TTree]");
56
57   AliTRDrawData rawData;
58   rawReader->Reset();
59   rawReader->Select("TRD");
60   AliTRDdigitsManager *manager = rawData.Raw2Digits(rawReader);
61   manager->MakeBranch(digitsTree);
62   manager->WriteDigits();
63
64 }
65
66 //_____________________________________________________________________________
67 void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader
68                                     , AliRawReader *rawReader) const
69 {
70   //
71   // Reconstruct clusters
72   //
73
74   AliInfo("Reconstruct TRD clusters from RAW data [RunLoader, RawReader]");
75
76   AliLoader *loader = runLoader->GetLoader("TRDLoader");
77   loader->LoadRecPoints("recreate");
78
79   runLoader->CdGAFile();
80   Int_t nEvents = runLoader->GetNumberOfEvents();
81
82   rawReader->Reset();
83   rawReader->Select("TRD");
84
85   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
86
87     if (!rawReader->NextEvent()) break;
88
89     // Old (slow) cluster finder
90     //AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
91     //clusterer.Open(runLoader->GetFileName(),iEvent);
92     //clusterer.ReadDigits(rawReader);
93     //clusterer.MakeClusters();
94
95     // New (fast) cluster finder
96     AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
97     clusterer.Open(runLoader->GetFileName(),iEvent);
98     clusterer.Raw2ClustersChamber(rawReader);
99
100     clusterer.WriteClusters(-1);
101
102   }
103
104   loader->UnloadRecPoints();
105
106 }
107
108 //_____________________________________________________________________________
109 void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
110                                     , TTree *clusterTree) const
111 {
112   //
113   // Reconstruct clusters
114   //
115
116   AliInfo("Reconstruct TRD clusters from RAW data [RawReader -> Cluster TTree]");
117
118   rawReader->Reset();
119   rawReader->Select("TRD");
120
121   // Old (slow) cluster finder
122   //AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
123   //clusterer.OpenOutput(clusterTree);
124   //clusterer.ReadDigits(rawReader);
125   //clusterer.MakeClusters();
126
127   // New (fast) cluster finder
128   AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
129   clusterer.OpenOutput(clusterTree);
130   clusterer.SetAddLabels(kFALSE);
131   clusterer.Raw2ClustersChamber(rawReader);
132
133 }
134
135 //_____________________________________________________________________________
136 void AliTRDReconstructor::Reconstruct(TTree *digitsTree
137                                     , TTree *clusterTree) const
138 {
139   //
140   // Reconstruct clusters
141   //
142   AliInfo("Reconstruct TRD clusters from Digits [Digit TTree -> Cluster TTree]");
143
144   AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
145   //AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
146   clusterer.OpenOutput(clusterTree);
147   clusterer.ReadDigits(digitsTree);
148   clusterer.MakeClusters();
149
150 }
151
152 //_____________________________________________________________________________
153 void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader) const
154 {
155   //
156   // Reconstruct clusters
157   //
158
159   AliInfo("Reconstruct TRD clusters [AliRunLoader]");
160   AliLoader *loader = runLoader->GetLoader("TRDLoader");
161   loader->LoadRecPoints("recreate");
162
163   runLoader->CdGAFile();
164   Int_t nEvents = runLoader->GetNumberOfEvents();
165
166   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
167     AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
168     //AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
169     clusterer.Open(runLoader->GetFileName(),iEvent);
170     clusterer.ReadDigits();
171     clusterer.MakeClusters();
172     clusterer.WriteClusters(-1);
173   }
174
175   loader->UnloadRecPoints();
176
177 }
178
179 //_____________________________________________________________________________
180 AliTracker *AliTRDReconstructor::CreateTracker(AliRunLoader *runLoader) const
181 {
182   //
183   // Create a TRD tracker
184   //
185
186   runLoader->CdGAFile();
187
188   return new AliTRDtracker(gFile);
189
190 }
191
192 //_____________________________________________________________________________
193 void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/
194                                 , AliRawReader* /*rawReader*/
195                                 , AliESDEvent* /*esd*/) const
196 {
197   //
198   // Make PID
199   //
200
201   //AliTRDpidESD trdPID;
202   //trdPID.MakePID(esd);
203
204 }
205
206 //_____________________________________________________________________________
207 void AliTRDReconstructor::FillESD(AliRawReader* /*rawReader*/
208                                 , TTree* /*clusterTree*/
209                                 , AliESDEvent* /*esd*/) const
210 {
211   //
212   // Make PID
213   //
214
215   //AliTRDpidESD trdPID;
216   //trdPID.MakePID(esd);
217
218 }
219
220 //_____________________________________________________________________________
221 void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
222                                 , TTree* /*clusterTree*/
223                                 , AliESDEvent* /*esd*/) const
224 {
225   //
226   // Make PID
227   //
228
229   //AliTRDpidESD trdPID;
230   //trdPID.MakePID(esd);
231
232 }
233
234 //_____________________________________________________________________________
235 void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/
236                                 , AliESDEvent* /*esd*/) const
237 {
238   //
239   // Make PID
240   //
241
242   //AliTRDpidESD trdPID;
243   //trdPID.MakePID(esd);
244
245 }