1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
18 ///////////////////////////////////////////////////////////////////////////////
20 // Class for TRD reconstruction //
22 ///////////////////////////////////////////////////////////////////////////////
25 #include <TObjString.h>
26 #include <TObjArray.h>
27 #include <TClonesArray.h>
29 #include "AliRunLoader.h"
30 #include "AliRawReader.h"
32 #include "AliESDTrdTrack.h"
33 #include "AliESDEvent.h"
35 #include "AliTRDReconstructor.h"
36 #include "AliTRDclusterizer.h"
37 #include "AliTRDtracker.h"
38 #include "AliTRDpidESD.h"
39 #include "AliTRDgtuTrack.h"
40 #include "AliTRDrawData.h"
41 #include "AliTRDdigitsManager.h"
42 #include "AliTRDtrackerV1.h"
43 #include "AliTRDrecoParam.h"
45 ClassImp(AliTRDReconstructor)
47 TClonesArray *AliTRDReconstructor::fgClusters = 0x0;
48 //_____________________________________________________________________________
49 AliTRDReconstructor::AliTRDReconstructor()
51 ,fSteerParam(0x00000007)
53 memset(fStreamLevel, 0, 5*sizeof(UChar_t));
54 // Xe tail cancellation parameters
55 fTCParams[0] = 1.156; // r1
56 fTCParams[1] = 0.130; // r2
57 fTCParams[2] = 0.114; // c1
58 fTCParams[3] = 0.624; // c2
59 // Ar tail cancellation parameters
60 fTCParams[4] = 6.; // r1
61 fTCParams[5] = 0.62; // r2
62 fTCParams[6] = 0.0087;// c1
63 fTCParams[7] = 0.07; // c2
66 //_____________________________________________________________________________
67 AliTRDReconstructor::AliTRDReconstructor(const AliTRDReconstructor &r)
69 ,fSteerParam(r.fSteerParam)
71 memcpy(fStreamLevel, r.fStreamLevel, 5*sizeof(UChar_t));
72 memcpy(fTCParams, r.fTCParams, 8*sizeof(Double_t));
75 //_____________________________________________________________________________
76 void AliTRDReconstructor::Init(){
80 SetOption(GetOption());
82 AliInfo("TRD reconstruction will use the following settings:");
83 printf("\tWrite Clusters [cw] : %s\n", fSteerParam&kWriteClusters?"yes":"no");
84 printf("\tWrite Online Tracklets [tw] : %s\n", fSteerParam&kWriteTracklets?"yes":"no");
85 printf("\tDrift Gas Argon [ar] : %s\n", fSteerParam&kDriftGas?"yes":"no");
86 printf("\tStand Alone Tracking [sa] : %s\n", fSteerParam&kSeeding?"yes":"no");
87 printf("\tHLT Tracking [hlt] : %s\n", fSteerParam&kHLT?"yes":"no");
88 printf("\tNN PID [nn] : %s\n", fSteerParam&kSteerPID?"yes":"no");
89 printf("\tStreaming Levels : Clusterizer[%d] Tracker[%d] PID[%d]\n", fStreamLevel[kClusterizer], fStreamLevel[kTracker], fStreamLevel[kPID]);
92 //_____________________________________________________________________________
93 void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
94 , TTree *digitsTree) const
97 // Convert raw data digits into digit objects in a root tree
100 AliInfo("Convert raw data digits into digit objects [RawReader -> Digit TTree]");
102 AliTRDrawData rawData;
104 rawReader->Select("TRD");
105 AliTRDdigitsManager *manager = rawData.Raw2Digits(rawReader);
106 manager->MakeBranch(digitsTree);
107 manager->WriteDigits();
112 //_____________________________________________________________________________
113 void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
114 , TTree *clusterTree) const
117 // Reconstruct clusters
120 //AliInfo("Reconstruct TRD clusters from RAW data [RawReader -> Cluster TTree]");
124 rawReader->Select("TRD");
126 // New (fast) cluster finder
127 AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
128 clusterer.SetReconstructor(this);
129 clusterer.OpenOutput(clusterTree);
130 clusterer.SetAddLabels(kFALSE);
131 clusterer.Raw2ClustersChamber(rawReader);
133 if(IsWritingClusters()) return;
135 // take over ownership of clusters
136 fgClusters = clusterer.RecPoints();
137 clusterer.SetClustersOwner(kFALSE);
140 //_____________________________________________________________________________
141 void AliTRDReconstructor::Reconstruct(TTree *digitsTree
142 , TTree *clusterTree) const
145 // Reconstruct clusters
148 //AliInfo("Reconstruct TRD clusters from Digits [Digit TTree -> Cluster TTree]");
150 AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
151 clusterer.SetReconstructor(this);
152 clusterer.OpenOutput(clusterTree);
153 clusterer.ReadDigits(digitsTree);
154 clusterer.MakeClusters();
156 if(IsWritingClusters()) return;
158 // take over ownership of clusters
159 fgClusters = clusterer.RecPoints();
160 clusterer.SetClustersOwner(kFALSE);
163 //_____________________________________________________________________________
164 AliTracker *AliTRDReconstructor::CreateTracker() const
167 // Create a TRD tracker
170 //return new AliTRDtracker(NULL);
171 AliTRDtrackerV1 *tracker = new AliTRDtrackerV1();
172 tracker->SetReconstructor(this);
177 //_____________________________________________________________________________
178 void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
179 , TTree* /*clusterTree*/
180 , AliESDEvent* /*esd*/) const
189 //_____________________________________________________________________________
190 void AliTRDReconstructor::SetOption(Option_t *opt)
192 // Read option string into the steer param.
194 // Default steer param values
196 // write clusters [cw] = true
197 // track seeding (stand alone tracking) [sa] = true
198 // PID method in reconstruction (NN) [nn] = true
199 // write online tracklets [tw] = false
200 // drift gas [ar] = false
201 // HLT tracking [hlt] = false
203 fSteerParam = 0x00000007;
206 TObjArray *opar = s.Tokenize(",");
207 for(Int_t ipar=0; ipar<opar->GetEntriesFast(); ipar++){
208 TString sopt(((TObjString*)(*opar)[ipar])->String());
209 if(sopt.Contains("!cw")){
210 fSteerParam &= ~kWriteClusters;
212 } else if(sopt.Contains("!sa")){
213 fSteerParam &= ~kSeeding;
215 } else if(sopt.Contains("!nn")){
216 fSteerParam &= ~kSteerPID;
218 } else if(sopt.Contains("tw")){
219 if(!sopt.Contains("!")) fSteerParam |= kWriteTracklets;
221 } else if(sopt.Contains("ar")){
222 if(!sopt.Contains("!")) fSteerParam |= kDriftGas;
224 } else if(sopt.Contains("hlt")){
225 if(!sopt.Contains("!")) fSteerParam |= kHLT;
227 } else if(sopt.Contains("sl")){
228 TObjArray *stl = sopt.Tokenize("_");
229 if(stl->GetEntriesFast() < 3) continue;
230 TString taskstr(((TObjString*)(*stl)[1])->String());
231 TString levelstring(((TObjString*)(*stl)[2])->String());
232 // Set the stream Level
233 Int_t level = levelstring.Atoi();
234 AliTRDReconstructorTask task = kTracker;
235 if(taskstr.CompareTo("cl") == 0) task = kClusterizer;
236 else if(taskstr.CompareTo("tr") == 0) task = kTracker;
237 else if(taskstr.CompareTo("pi") == 0) task = kPID;
238 SetStreamLevel(level, task);
244 //_____________________________________________________________________________
245 void AliTRDReconstructor::SetStreamLevel(Int_t level, AliTRDReconstructorTask task){
247 // Set the Stream Level for one of the tasks Clusterizer, Tracker or PID
251 case kClusterizer: taskname = "Clusterizer";
253 case kTracker: taskname = "Tracker";
255 case kPID: taskname = "PID";
258 //AliInfo(Form("Setting Stream Level for Task %s to %d", taskname.Data(),level));
259 fStreamLevel[(Int_t)task] = level;