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>
28 #include "AliRunLoader.h"
29 #include "AliRawReader.h"
31 #include "AliESDTrdTrack.h"
32 #include "AliESDEvent.h"
34 #include "AliTRDReconstructor.h"
35 #include "AliTRDclusterizer.h"
36 #include "AliTRDtracker.h"
37 #include "AliTRDpidESD.h"
38 #include "AliTRDgtuTrack.h"
39 #include "AliTRDrawData.h"
40 #include "AliTRDdigitsManager.h"
41 #include "AliTRDtrackerV1.h"
42 #include "AliTRDrecoParam.h"
44 ClassImp(AliTRDReconstructor)
47 //_____________________________________________________________________________
48 AliTRDReconstructor::AliTRDReconstructor()
50 ,fSteerParam(0x00000007)
52 memset(fStreamLevel, 0, 5*sizeof(UChar_t));
53 // Xe tail cancellation parameters
54 fTCParams[0] = 1.156; // r1
55 fTCParams[1] = 0.130; // r2
56 fTCParams[2] = 0.114; // c1
57 fTCParams[3] = 0.624; // c2
58 // Ar tail cancellation parameters
59 fTCParams[4] = 6.; // r1
60 fTCParams[5] = 0.62; // r2
61 fTCParams[6] = 0.0087;// c1
62 fTCParams[7] = 0.07; // c2
65 //_____________________________________________________________________________
66 AliTRDReconstructor::AliTRDReconstructor(const AliTRDReconstructor &r)
68 ,fSteerParam(r.fSteerParam)
70 memcpy(fStreamLevel, r.fStreamLevel, 5*sizeof(UChar_t));
71 memcpy(fTCParams, r.fTCParams, 8*sizeof(Double_t));
74 //_____________________________________________________________________________
75 void AliTRDReconstructor::Init(){
79 SetOption(GetOption());
82 //_____________________________________________________________________________
83 void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
84 , TTree *digitsTree) const
87 // Convert raw data digits into digit objects in a root tree
90 AliInfo("Convert raw data digits into digit objects [RawReader -> Digit TTree]");
92 AliTRDrawData rawData;
94 rawReader->Select("TRD");
95 AliTRDdigitsManager *manager = rawData.Raw2Digits(rawReader);
96 manager->MakeBranch(digitsTree);
97 manager->WriteDigits();
102 //_____________________________________________________________________________
103 void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
104 , TTree *clusterTree) const
107 // Reconstruct clusters
110 AliInfo("Reconstruct TRD clusters from RAW data [RawReader -> Cluster TTree]");
114 rawReader->Select("TRD");
116 // New (fast) cluster finder
117 AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
118 clusterer.SetReconstructor(this);
119 clusterer.OpenOutput(clusterTree);
120 clusterer.SetAddLabels(kFALSE);
121 clusterer.Raw2ClustersChamber(rawReader);
125 //_____________________________________________________________________________
126 void AliTRDReconstructor::Reconstruct(TTree *digitsTree
127 , TTree *clusterTree) const
130 // Reconstruct clusters
133 AliInfo("Reconstruct TRD clusters from Digits [Digit TTree -> Cluster TTree]");
135 AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
136 clusterer.SetReconstructor(this);
137 clusterer.OpenOutput(clusterTree);
138 clusterer.ReadDigits(digitsTree);
139 clusterer.MakeClusters();
143 //_____________________________________________________________________________
144 AliTracker *AliTRDReconstructor::CreateTracker() const
147 // Create a TRD tracker
150 //return new AliTRDtracker(NULL);
151 AliTRDtrackerV1 *tracker = new AliTRDtrackerV1();
152 tracker->SetReconstructor(this);
157 //_____________________________________________________________________________
158 void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
159 , TTree* /*clusterTree*/
160 , AliESDEvent* /*esd*/) const
169 //_____________________________________________________________________________
170 void AliTRDReconstructor::SetOption(Option_t *opt)
172 // Read option string into the steer param.
174 // Default steer param values
176 // write clusters [cw] = true
177 // track seeding (stand alone tracking) [sa] = true
178 // PID method in reconstruction (NN) [nn] = true
179 // write online tracklets [tw] = false
180 // drift gas [ar] = false
182 fSteerParam = 0x00000007;
185 TObjArray *opar = s.Tokenize(",");
186 for(Int_t ipar=0; ipar<opar->GetEntriesFast(); ipar++){
187 TString sopt(((TObjString*)(*opar)[ipar])->String());
188 if(sopt.Contains("!cw")){
189 fSteerParam &= ~kWriteClusters;
191 } else if(sopt.Contains("!sa")){
192 fSteerParam &= ~kSeeding;
194 } else if(sopt.Contains("!nn")){
195 fSteerParam &= ~kSteerPID;
197 } else if(sopt.Contains("tw")){
198 fSteerParam |= kWriteTracklets;
200 } else if(sopt.Contains("ar")){
201 fSteerParam |= kDriftGas;
203 } else if(sopt.Contains("sl")){
204 TObjArray *stl = sopt.Tokenize("_");
205 if(stl->GetEntriesFast() < 3) continue;
206 TString taskstr(((TObjString*)(*stl)[1])->String());
207 TString levelstring(((TObjString*)(*stl)[2])->String());
208 // Set the stream Level
209 Int_t level = levelstring.Atoi();
210 AliTRDReconstructorTask task = kTracker;
211 if(taskstr.CompareTo("cl") == 0) task = kClusterizer;
212 else if(taskstr.CompareTo("tr") == 0) task = kTracker;
213 else if(taskstr.CompareTo("pi") == 0) task = kPID;
214 SetStreamLevel(level, task);
220 //_____________________________________________________________________________
221 void AliTRDReconstructor::SetStreamLevel(Int_t level, AliTRDReconstructorTask task){
223 // Set the Stream Level for one of the tasks Clusterizer, Tracker or PID
227 case kClusterizer: taskname = "Clusterizer";
229 case kTracker: taskname = "Tracker";
231 case kPID: taskname = "PID";
234 AliInfo(Form("Setting Stream Level for Task %s to %d", taskname.Data(),level));
235 fStreamLevel[(Int_t)task] = level;