]>
Commit | Line | Data |
---|---|---|
f6019cda | 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 | //*-- Yves Schutz (SUBATECH) | |
21 | // Reconstruction class. Redesigned from the old AliReconstructionner class and | |
22 | // derived from STEER/AliReconstructor. | |
23 | // | |
24 | // --- ROOT system --- | |
25 | ||
26 | // --- Standard library --- | |
27 | ||
28 | // --- AliRoot header files --- | |
f6019cda | 29 | #include "AliEMCALReconstructor.h" |
5dee926e | 30 | |
af885e0f | 31 | #include "AliESDEvent.h" |
89ffc0b0 | 32 | #include "AliESDCaloCluster.h" |
6a0cf740 | 33 | #include "AliESDtrack.h" |
5dee926e | 34 | #include "AliEMCALLoader.h" |
98e9578e | 35 | #include "AliEMCALRawUtils.h" |
f6019cda | 36 | #include "AliEMCALClusterizerv1.h" |
5dee926e | 37 | #include "AliEMCALRecPoint.h" |
dc293ae9 | 38 | #include "AliEMCALPID.h" |
0964c2e9 | 39 | #include "AliEMCALTrigger.h" |
1d59832c | 40 | #include "AliRawReader.h" |
c47157cd | 41 | // to be removed - it is here just because of geom |
42 | #include "AliRun.h" | |
43 | #include "AliRunLoader.h" | |
1d59832c | 44 | |
f6019cda | 45 | ClassImp(AliEMCALReconstructor) |
46 | ||
c47157cd | 47 | AliEMCALRecParam* AliEMCALReconstructor::fgkRecParam = 0; // EMCAL rec. parameters |
48 | ||
f6019cda | 49 | //____________________________________________________________________________ |
18a21c7c | 50 | AliEMCALReconstructor::AliEMCALReconstructor() |
51 | : fDebug(kFALSE) | |
f6019cda | 52 | { |
53 | // ctor | |
c47157cd | 54 | if (!fgkRecParam) { |
55 | AliWarning("The Reconstruction parameters for EMCAL nonitialized - Used default one"); | |
56 | fgkRecParam = new AliEMCALRecParam; | |
57 | } | |
f6019cda | 58 | } |
59 | ||
0a4cb131 | 60 | //____________________________________________________________________________ |
18a21c7c | 61 | AliEMCALReconstructor::AliEMCALReconstructor(const AliEMCALReconstructor & rec) |
62 | : AliReconstructor(rec), | |
63 | fDebug(rec.fDebug) | |
0a4cb131 | 64 | { |
65 | //copy ctor | |
0a4cb131 | 66 | } |
f6019cda | 67 | |
68 | //____________________________________________________________________________ | |
69 | AliEMCALReconstructor::~AliEMCALReconstructor() | |
70 | { | |
71 | // dtor | |
72 | } | |
73 | ||
74 | //____________________________________________________________________________ | |
c47157cd | 75 | void AliEMCALReconstructor::Reconstruct(TTree* digitsTree, TTree* clustersTree) const |
f6019cda | 76 | { |
77 | // method called by AliReconstruction; | |
78 | // Only the clusterization is performed,; the rest of the reconstruction is done in FillESD because the track | |
79 | // segment maker needs access to the AliESD object to retrieve the tracks reconstructed by | |
80 | // the global tracking. | |
c47157cd | 81 | // Works on the current event. |
f6019cda | 82 | |
c47157cd | 83 | AliEMCALClusterizerv1 clu; |
84 | clu.SetInput(digitsTree); | |
85 | clu.SetOutput(clustersTree); | |
86 | if ( Debug() ) | |
87 | clu.Digits2Clusters("deb all") ; | |
88 | else | |
89 | clu.Digits2Clusters("pseudo") ; | |
f6019cda | 90 | } |
91 | ||
a68156e6 | 92 | //____________________________________________________________________________ |
c47157cd | 93 | void AliEMCALReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const |
98e9578e | 94 | |
a68156e6 | 95 | { |
c47157cd | 96 | // Conversion from raw data to |
97 | // EMCAL digits. | |
98 | // Works on a single-event basis | |
85c60a8e | 99 | |
98e9578e | 100 | rawReader->Reset() ; |
98e9578e | 101 | |
c47157cd | 102 | TClonesArray *digitsArr = new TClonesArray("AliEMCALDigit",100); |
103 | Int_t bufsize = 32000; | |
104 | digitsTree->Branch("EMCAL", &digitsArr, bufsize); | |
98e9578e | 105 | |
c47157cd | 106 | static AliEMCALRawUtils rawUtils; |
107 | rawUtils.Raw2Digits(rawReader,digitsArr); | |
a68156e6 | 108 | } |
109 | ||
f6019cda | 110 | //____________________________________________________________________________ |
d13093b8 | 111 | void AliEMCALReconstructor::FillESD(TTree* /*digitsTree*/, TTree* clustersTree, |
c47157cd | 112 | AliESDEvent* esd) const |
f6019cda | 113 | { |
98e9578e | 114 | // Called by AliReconstruct after Reconstruct() and global tracking and vertexing |
c47157cd | 115 | // Works on the current event |
98e9578e | 116 | const double timeScale = 1.e+11; // transition constant from sec to 0.01 ns |
92da3372 | 117 | |
85c60a8e | 118 | // Creates AliESDCaloCluster from AliEMCALRecPoints |
c47157cd | 119 | |
120 | TObjArray *clusters = new TObjArray(100); | |
121 | TBranch *branch = clustersTree->GetBranch("EMCALECARP"); | |
122 | branch->SetAddress(&clusters); | |
123 | clustersTree->GetEvent(0); | |
124 | ||
a7a5421e | 125 | Int_t nClusters = clusters->GetEntries(), nClustersNew=0; |
c47157cd | 126 | AliDebug(1,Form("%d clusters",nClusters)); |
5d9e7bd7 | 127 | // Int_t nRP=0, nPC=0; // in input |
a7a5421e | 128 | esd->SetFirstEMCALCluster(esd->GetNumberOfCaloClusters()); // Put after Phos clusters |
92da3372 | 129 | |
6a0cf740 | 130 | //###################################################### |
0964c2e9 | 131 | //#########Calculate trigger and set trigger info########### |
6a0cf740 | 132 | //###################################################### |
0964c2e9 | 133 | |
134 | AliEMCALTrigger tr ; | |
135 | // tr.SetPatchSize(1);//create 4x4 patches | |
136 | tr.Trigger(); | |
137 | ||
138 | Float_t maxAmp2x2 = tr.Get2x2MaxAmplitude(); | |
139 | Float_t maxAmpnxn = tr.GetnxnMaxAmplitude(); | |
140 | Float_t ampOutOfPatch2x2 = tr.Get2x2AmpOutOfPatch() ; | |
141 | Float_t ampOutOfPatchnxn = tr.GetnxnAmpOutOfPatch() ; | |
142 | ||
98e9578e | 143 | AliEMCALGeometry * geom = 0; |
c47157cd | 144 | AliRunLoader *runLoader = AliRunLoader::GetRunLoader(); |
98e9578e | 145 | if (runLoader->GetAliRun() && runLoader->GetAliRun()->GetDetector("EMCAL")) |
146 | geom = dynamic_cast<AliEMCAL*>(runLoader->GetAliRun()->GetDetector("EMCAL"))->GetGeometry(); | |
147 | if (geom == 0) | |
148 | geom = AliEMCALGeometry::GetInstance(AliEMCALGeometry::GetDefaulGeometryName()); | |
149 | ||
0964c2e9 | 150 | Int_t iSM2x2 = tr.Get2x2SuperModule(); |
151 | Int_t iSMnxn = tr.GetnxnSuperModule(); | |
152 | Int_t iCellPhi2x2 = tr.Get2x2CellPhi(); | |
153 | Int_t iCellPhinxn = tr.GetnxnCellPhi(); | |
154 | Int_t iCellEta2x2 = tr.Get2x2CellEta(); | |
155 | Int_t iCellEtanxn = tr.GetnxnCellEta(); | |
156 | ||
157 | AliDebug(2, Form("Trigger 2x2 max amp %f, out amp %f, SM %d, iphi %d ieta %d", maxAmp2x2, ampOutOfPatch2x2, iSM2x2,iCellPhi2x2, iCellEta2x2)); | |
158 | AliDebug(2, Form("Trigger 4x4 max amp %f , out amp %f, SM %d, iphi %d, ieta %d", maxAmpnxn, ampOutOfPatchnxn, iSMnxn,iCellPhinxn, iCellEtanxn)); | |
159 | ||
160 | TVector3 pos2x2(-1,-1,-1); | |
161 | TVector3 posnxn(-1,-1,-1); | |
162 | ||
163 | Int_t iAbsId2x2 = geom->GetAbsCellIdFromCellIndexes( iSM2x2, iCellPhi2x2, iCellEta2x2) ; | |
164 | Int_t iAbsIdnxn = geom->GetAbsCellIdFromCellIndexes( iSMnxn, iCellPhinxn, iCellEtanxn) ; | |
165 | geom->GetGlobal(iAbsId2x2, pos2x2); | |
166 | geom->GetGlobal(iAbsIdnxn, posnxn); | |
167 | ||
168 | TArrayF triggerPosition(6); | |
169 | triggerPosition[0] = pos2x2(0) ; | |
170 | triggerPosition[1] = pos2x2(1) ; | |
171 | triggerPosition[2] = pos2x2(2) ; | |
172 | triggerPosition[3] = posnxn(0) ; | |
173 | triggerPosition[4] = posnxn(1) ; | |
174 | triggerPosition[5] = posnxn(2) ; | |
175 | ||
176 | TArrayF triggerAmplitudes(4); | |
177 | triggerAmplitudes[0] = maxAmp2x2 ; | |
178 | triggerAmplitudes[1] = ampOutOfPatch2x2 ; | |
179 | triggerAmplitudes[2] = maxAmpnxn ; | |
180 | triggerAmplitudes[3] = ampOutOfPatchnxn ; | |
181 | ||
182 | esd->AddEMCALTriggerPosition(triggerPosition); | |
183 | esd->AddEMCALTriggerAmplitudes(triggerAmplitudes); | |
184 | ||
6a0cf740 | 185 | //###################################################### |
186 | //#######################TRACK MATCHING############### | |
187 | //###################################################### | |
188 | //Fill list of integers, each one is index of track to which the cluster belongs. | |
189 | ||
190 | // step 1 - initialize array of matched track indexes | |
191 | Int_t *matchedTrack = new Int_t[nClusters]; | |
192 | for (Int_t iclus = 0; iclus < nClusters; iclus++) | |
193 | matchedTrack[iclus] = -1; // neg. index --> no matched track | |
194 | ||
195 | // step 2, change the flag for all matched clusters found in tracks | |
196 | Int_t iemcalMatch = -1; | |
197 | Int_t endtpc = esd->GetNumberOfTracks(); | |
198 | for (Int_t itrack = 0; itrack < endtpc; itrack++) { | |
199 | AliESDtrack * track = esd->GetTrack(itrack) ; // retrieve track | |
200 | iemcalMatch = track->GetEMCALcluster(); | |
65f4a419 | 201 | if(iemcalMatch >= 0) matchedTrack[iemcalMatch] = itrack; |
6a0cf740 | 202 | } |
203 | ||
204 | //######################################## | |
205 | //##############Fill CaloClusters############# | |
206 | //######################################## | |
0964c2e9 | 207 | |
0964c2e9 | 208 | |
5dee926e | 209 | for (Int_t iClust = 0 ; iClust < nClusters ; iClust++) { |
c47157cd | 210 | const AliEMCALRecPoint * clust = (const AliEMCALRecPoint*)clusters->At(iClust); |
8ada0ffe | 211 | //if(clust->GetClusterType()== AliESDCaloCluster::kEMCALClusterv1) nRP++; else nPC++; |
85c60a8e | 212 | if (Debug()) clust->Print(); |
a7a5421e | 213 | // Get information from EMCAL reconstruction points |
85c60a8e | 214 | Float_t xyz[3]; |
5dee926e | 215 | TVector3 gpos; |
216 | clust->GetGlobalPosition(gpos); | |
f6019cda | 217 | for (Int_t ixyz=0; ixyz<3; ixyz++) |
5dee926e | 218 | xyz[ixyz] = gpos[ixyz]; |
7592dfc4 | 219 | |
85c60a8e | 220 | Int_t digitMult = clust->GetMultiplicity(); |
d13093b8 | 221 | TArrayS amplList(digitMult); |
222 | TArrayS timeList(digitMult); | |
223 | TArrayS digiList(digitMult); | |
85c60a8e | 224 | Float_t *amplFloat = clust->GetEnergiesList(); |
225 | Float_t *timeFloat = clust->GetTimeList(); | |
226 | Int_t *digitInts = clust->GetAbsId(); | |
8a11aef1 | 227 | Float_t elipAxis[2]; |
92da3372 | 228 | clust->GetElipsAxis(elipAxis); |
7592dfc4 | 229 | |
230 | // Convert Float_t* and Int_t* to Short_t* to save memory | |
a7a5421e | 231 | // Problem : we should recalculate a cluster characteristics when discard digit(s) |
f1487f22 | 232 | Int_t newdigitMult = 0; |
92da3372 | 233 | for (Int_t iDigit=0; iDigit<digitMult; iDigit++) { |
98e9578e | 234 | if (amplFloat[iDigit] > 0) { |
a7a5421e | 235 | amplList[newdigitMult] = (UShort_t)(amplFloat[iDigit]*500); |
98e9578e | 236 | // Time in units of 0.01 ns = 10 ps |
237 | if(timeFloat[iDigit] < 65536./timeScale) | |
238 | timeList[newdigitMult] = (UShort_t)(timeFloat[iDigit]*timeScale); | |
239 | else | |
240 | timeList[newdigitMult] = 65535; | |
241 | digiList[newdigitMult] = (UShort_t)(digitInts[iDigit]); | |
242 | newdigitMult++; | |
92da3372 | 243 | } |
8ada0ffe | 244 | else if (clust->GetClusterType() != AliESDCaloCluster::kEMCALPseudoCluster) |
98e9578e | 245 | Warning("FillESD()","Negative or 0 digit amplitude in cluster"); |
92da3372 | 246 | } |
7592dfc4 | 247 | |
a7a5421e | 248 | if(newdigitMult > 0) { // accept cluster if it has some digit |
249 | nClustersNew++; | |
7592dfc4 | 250 | |
d13093b8 | 251 | amplList.Set(newdigitMult); |
252 | timeList.Set(newdigitMult); | |
253 | digiList.Set(newdigitMult); | |
254 | ||
65721814 | 255 | //Primaries |
7592dfc4 | 256 | Int_t parentMult = 0; |
257 | Int_t *parentInts = clust->GetParents(parentMult); | |
d13093b8 | 258 | TArrayS parentList(parentMult); |
7592dfc4 | 259 | for (Int_t ipr=0; ipr<parentMult; ipr++) |
260 | parentList[ipr] = (Short_t)(parentInts[ipr]); | |
261 | ||
17773e2e | 262 | |
a7a5421e | 263 | // fills the ESDCaloCluster |
264 | AliESDCaloCluster * ec = new AliESDCaloCluster() ; | |
265 | ec->SetClusterType(clust->GetClusterType()); | |
7592dfc4 | 266 | ec->SetPosition(xyz); |
267 | ec->SetE(clust->GetEnergy()); | |
d13093b8 | 268 | ec->AddDigitAmplitude(amplList); |
269 | ec->AddDigitTime(timeList); | |
270 | ec->AddDigitIndex(digiList); | |
7592dfc4 | 271 | |
8ada0ffe | 272 | if(clust->GetClusterType()== AliESDCaloCluster::kEMCALClusterv1){ |
7592dfc4 | 273 | |
a7a5421e | 274 | ec->SetClusterDisp(clust->GetDispersion()); |
275 | ec->SetClusterChi2(-1); //not yet implemented | |
276 | ec->SetM02(elipAxis[0]*elipAxis[0]) ; | |
277 | ec->SetM20(elipAxis[1]*elipAxis[1]) ; | |
278 | ec->SetM11(-1) ; //not yet implemented | |
7592dfc4 | 279 | |
280 | TArrayS arrayTrackMatched(1);// Only one track, temporal solution. | |
281 | arrayTrackMatched[0]= (Short_t)(matchedTrack[iClust]); | |
282 | ec->AddTracksMatched(arrayTrackMatched); | |
283 | ||
d13093b8 | 284 | ec->AddLabels(parentList); |
a7a5421e | 285 | } |
7592dfc4 | 286 | |
6a0cf740 | 287 | // add the cluster to the esd object |
a7a5421e | 288 | esd->AddCaloCluster(ec); |
289 | delete ec; | |
a7a5421e | 290 | } |
17773e2e | 291 | |
a7a5421e | 292 | } // cycle on clusters |
17773e2e | 293 | |
294 | delete [] matchedTrack; | |
295 | ||
a7a5421e | 296 | esd->SetNumberOfEMCALClusters(nClustersNew); |
5d9e7bd7 | 297 | //if(nClustersNew != nClusters) |
298 | //printf(" ##### nClusters %i -> new %i ##### \n", nClusters, nClustersNew ); | |
7592dfc4 | 299 | |
dc293ae9 | 300 | //Fill ESDCaloCluster with PID weights |
dc293ae9 | 301 | AliEMCALPID *pid = new AliEMCALPID; |
302 | //pid->SetPrintInfo(kTRUE); | |
303 | pid->SetReconstructor(kTRUE); | |
304 | pid->RunPID(esd); | |
d0708678 | 305 | delete pid; |
f6019cda | 306 | } |
dc293ae9 | 307 | |
98e9578e | 308 |