Correct index mapping, add L1 amp histogram and other
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALReconstructor.cxx
CommitLineData
f6019cda 1/**************************************************************************
d874f457 2#/eliza17/alice/esdfiles/data/2011/LHC11a/000146805/ESDs/pass2_without_SDD/11000146805038.550/root_archive.zip#AliESDs.root
f6019cda 3 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * *
5 * Author: The ALICE Off-line Project. *
6 * Contributors are mentioned in the code where appropriate. *
7 * *
8 * Permission to use, copy, modify and distribute this software and its *
9 * documentation strictly for non-commercial purposes is hereby granted *
10 * without fee, provided that the above copyright notice appears in all *
11 * copies and that both the copyright notice and this permission notice *
12 * appear in the supporting documentation. The authors make no claims *
13 * about the suitability of this software for any purpose. It is *
14 * provided "as is" without express or implied warranty. *
15 **************************************************************************/
16
17/* $Id$ */
18
19//_________________________________________________________________________
fa42b1f3 20//--
21//-- Yves Schutz (SUBATECH)
f6019cda 22// Reconstruction class. Redesigned from the old AliReconstructionner class and
23// derived from STEER/AliReconstructor.
24//
29c9d15f 25
85c25c2e 26
f6019cda 27// --- ROOT system ---
85c25c2e 28#include <TClonesArray.h>
0c5b726e 29#include "TGeoManager.h"
30#include "TGeoMatrix.h"
f6019cda 31
32// --- Standard library ---
33
34// --- AliRoot header files ---
f6019cda 35#include "AliEMCALReconstructor.h"
5dee926e 36
aaa3cb7c 37#include "AliCodeTimer.h"
23ca956b 38#include "AliCaloCalibPedestal.h"
39#include "AliEMCALCalibData.h"
af885e0f 40#include "AliESDEvent.h"
89ffc0b0 41#include "AliESDCaloCluster.h"
0e7c6655 42#include "AliESDCaloCells.h"
6a0cf740 43#include "AliESDtrack.h"
5dee926e 44#include "AliEMCALLoader.h"
98e9578e 45#include "AliEMCALRawUtils.h"
0e7c6655 46#include "AliEMCALDigit.h"
f6019cda 47#include "AliEMCALClusterizerv1.h"
783153ff 48#include "AliEMCALClusterizerv2.h"
ee08edde 49#include "AliEMCALClusterizerNxN.h"
5dee926e 50#include "AliEMCALRecPoint.h"
dc293ae9 51#include "AliEMCALPID.h"
1d59832c 52#include "AliRawReader.h"
fa42b1f3 53#include "AliCDBEntry.h"
54#include "AliCDBManager.h"
65bdc82f 55#include "AliEMCALGeometry.h"
ac8ae9fe 56#include "AliEMCAL.h"
85c25c2e 57#include "AliESDVZERO.h"
0c5b726e 58#include "AliCDBManager.h"
72c58de0 59#include "AliRunLoader.h"
60#include "AliRun.h"
916f1e76 61#include "AliEMCALTriggerData.h"
62#include "AliEMCALTriggerElectronics.h"
fff39dd1 63#include "AliEMCALTriggerDCSConfigDB.h"
64#include "AliEMCALTriggerDCSConfig.h"
de39a0ff 65#include "AliEMCALTriggerData.h"
66#include "AliEMCALTriggerRawDigit.h"
67#include "AliEMCALTriggerPatch.h"
68#include "AliEMCALTriggerTypes.h"
1d59832c 69
85c25c2e 70ClassImp(AliEMCALReconstructor)
95ffa314 71
5970dfe2 72const AliEMCALRecParam* AliEMCALReconstructor::fgkRecParam = 0; // EMCAL rec. parameters
b57cc9b9 73AliEMCALRawUtils* AliEMCALReconstructor::fgRawUtils = 0; // EMCAL raw utilities class
74AliEMCALClusterizer* AliEMCALReconstructor::fgClusterizer = 0; // EMCAL clusterizer class
75TClonesArray* AliEMCALReconstructor::fgDigitsArr = 0; // list of digits, to be used multiple times
76TObjArray* AliEMCALReconstructor::fgClustersArr = 0; // list of clusters, to be used multiple times
4bd37bd4 77TClonesArray* AliEMCALReconstructor::fgTriggerDigits = 0; // list of trigger digits, to be used multiple times
916f1e76 78AliEMCALTriggerElectronics* AliEMCALReconstructor::fgTriggerProcessor = 0x0;
f6019cda 79//____________________________________________________________________________
18a21c7c 80AliEMCALReconstructor::AliEMCALReconstructor()
5970dfe2 81 : fGeom(0),fCalibData(0),fPedestalData(0),fTriggerData(0x0), fMatches(0x0)
f6019cda 82{
83 // ctor
c828bc97 84
85 // AliDebug(2, "Mark.");
86
65bdc82f 87 fgRawUtils = new AliEMCALRawUtils;
29c9d15f 88
72c58de0 89 //To make sure we match with the geometry in a simulation file,
90 //let's try to get it first. If not, take the default geometry
33c3c91a 91 AliRunLoader *rl = AliRunLoader::Instance();
7e1d9a9b 92 if (rl->GetAliRun()){
93 AliEMCAL * emcal = dynamic_cast<AliEMCAL*>(rl->GetAliRun()->GetDetector("EMCAL"));
94 if(emcal) fGeom = emcal->GetGeometry();
95 }
96
97 if(!fGeom) {
72c58de0 98 AliInfo(Form("Using default geometry in reconstruction"));
937d0661 99 fGeom = AliEMCALGeometry::GetInstance(AliEMCALGeometry::GetDefaultGeometryName());
65bdc82f 100 }
29c9d15f 101
0c5b726e 102 //Get calibration parameters
103 if(!fCalibData)
104 {
29c9d15f 105 AliCDBEntry *entry = (AliCDBEntry*)
106 AliCDBManager::Instance()->Get("EMCAL/Calib/Data");
107 if (entry) fCalibData = (AliEMCALCalibData*) entry->GetObject();
0c5b726e 108 }
29c9d15f 109
0c5b726e 110 if(!fCalibData)
29c9d15f 111 AliFatal("Calibration parameters not found in CDB!");
112
40164976 113 //Get calibration parameters
114 if(!fPedestalData)
115 {
de39a0ff 116 AliCDBEntry *entry = (AliCDBEntry*)
29c9d15f 117 AliCDBManager::Instance()->Get("EMCAL/Calib/Pedestals");
de39a0ff 118 if (entry) fPedestalData = (AliCaloCalibPedestal*) entry->GetObject();
40164976 119 }
de39a0ff 120
ee08edde 121 if(!fPedestalData)
122 AliFatal("Dead map not found in CDB!");
29c9d15f 123
72c58de0 124 if(!fGeom) AliFatal(Form("Could not get geometry!"));
29c9d15f 125
f62c044a 126 AliEMCALTriggerDCSConfigDB* dcsConfigDB = AliEMCALTriggerDCSConfigDB::Instance();
29c9d15f 127
f62c044a 128 const AliEMCALTriggerDCSConfig* dcsConfig = dcsConfigDB->GetTriggerDCSConfig();
29c9d15f 129
f62c044a 130 if (!dcsConfig) AliFatal("No Trigger DCS Configuration from OCDB!");
131 fgTriggerProcessor = new AliEMCALTriggerElectronics( dcsConfig );
29c9d15f 132
f62c044a 133 fTriggerData = new AliEMCALTriggerData();
29c9d15f 134
f62c044a 135 //Init temporary list of digits
4bd37bd4 136 fgDigitsArr = new TClonesArray("AliEMCALDigit",1000);
137 fgClustersArr = new TObjArray(1000);
138 fgTriggerDigits = new TClonesArray("AliEMCALTriggerRawDigit",1000);
5970dfe2 139
140 //Track matching
141 fMatches = new TList();
142 fMatches->SetOwner(kTRUE);
f6019cda 143}
144
0a4cb131 145//____________________________________________________________________________
f6019cda 146AliEMCALReconstructor::~AliEMCALReconstructor()
147{
148 // dtor
c828bc97 149
150 //AliDebug(2, "Mark.");
151
23ca956b 152 if(fGeom) delete fGeom;
80453a9f 153
154 //No need to delete, recovered from OCDB
155 //if(fCalibData) delete fCalibData;
156 //if(fPedestalData) delete fPedestalData;
de39a0ff 157
b57cc9b9 158 if(fgDigitsArr){
159 fgDigitsArr->Clear("C");
160 delete fgDigitsArr;
161 }
162
163 if(fgClustersArr){
164 fgClustersArr->Clear();
165 delete fgClustersArr;
166 }
167
4bd37bd4 168 if(fgTriggerDigits){
7dbb454b 169 fgTriggerDigits->Clear();
4bd37bd4 170 delete fgTriggerDigits;
171 }
172
23ca956b 173 if(fgRawUtils) delete fgRawUtils;
174 if(fgClusterizer) delete fgClusterizer;
f62c044a 175 if(fgTriggerProcessor) delete fgTriggerProcessor;
b57cc9b9 176
5970dfe2 177 if(fMatches) { fMatches->Delete(); delete fMatches; fMatches = 0;}
178
aaa3cb7c 179 AliCodeTimer::Instance()->Print();
f6019cda 180}
181
9519e3e3 182//____________________________________________________________________________
183void AliEMCALReconstructor::InitClusterizer() const
ee08edde 184{
9519e3e3 185 //Init the clusterizer with geometry and calibration pointers, avoid doing it twice.
186 Int_t clusterizerType = -1;
187 Int_t eventType = -1;
188 if(GetRecParam()) {
189 clusterizerType = GetRecParam()->GetClusterizerFlag();
190 eventType = GetRecParam()->GetEventSpecie();
7e1d9a9b 191 }
192 else{
9519e3e3 193 AliCDBEntry *entry = (AliCDBEntry*)
29c9d15f 194 AliCDBManager::Instance()->Get("EMCAL/Calib/RecoParam");
9519e3e3 195 //Get The reco param for the default event specie
196 if (entry) {
197 AliEMCALRecParam *recParam = (AliEMCALRecParam*)((TObjArray *) entry->GetObject())->At(0);
198 if(recParam) clusterizerType = recParam->GetClusterizerFlag();
de39a0ff 199 }
7e1d9a9b 200 }
ee08edde 201
9519e3e3 202 //Check if clusterizer previously set corresponds to what is needed for this event type
203 if(fgClusterizer){
204 if(eventType!=AliRecoParam::kCalib){
205 //printf("ReCreate clusterizer? Clusterizer set <%d>, Clusterizer in use <%s>\n",
29c9d15f 206 // clusterizerType, fgClusterizer->Version());
9519e3e3 207
783153ff 208 if (clusterizerType == AliEMCALRecParam::kClusterizerv1 && !strcmp(fgClusterizer->Version(),"clu-v1")) return;
9519e3e3 209
210 else if(clusterizerType == AliEMCALRecParam::kClusterizerNxN && !strcmp(fgClusterizer->Version(),"clu-NxN")) return;
211
783153ff 212 else if(clusterizerType == AliEMCALRecParam::kClusterizerv2 && !strcmp(fgClusterizer->Version(),"clu-v2")) return;
213
9519e3e3 214 //Need to create new clusterizer, the one set previously is not the correct one
215 delete fgClusterizer;
216 }
217 else return;
218 }
29c9d15f 219
783153ff 220 if (clusterizerType == AliEMCALRecParam::kClusterizerv1)
29c9d15f 221 {
783153ff 222 fgClusterizer = new AliEMCALClusterizerv1 (fGeom, fCalibData,fPedestalData);
29c9d15f 223 }
783153ff 224 else if (clusterizerType == AliEMCALRecParam::kClusterizerNxN)
29c9d15f 225 {
226 fgClusterizer = new AliEMCALClusterizerNxN(fGeom, fCalibData,fPedestalData);
227 }
783153ff 228 else if (clusterizerType == AliEMCALRecParam::kClusterizerv2)
229 {
230 fgClusterizer = new AliEMCALClusterizerv2 (fGeom, fCalibData,fPedestalData);
231 }
232 else
233 {
234 AliFatal(Form("Unknown clusterizer %d ", clusterizerType));
235 }
ee08edde 236}
237
238//____________________________________________________________________________
c47157cd 239void AliEMCALReconstructor::Reconstruct(TTree* digitsTree, TTree* clustersTree) const
f6019cda 240{
241 // method called by AliReconstruction;
242 // Only the clusterization is performed,; the rest of the reconstruction is done in FillESD because the track
243 // segment maker needs access to the AliESD object to retrieve the tracks reconstructed by
244 // the global tracking.
c47157cd 245 // Works on the current event.
29c9d15f 246
1a429a6b 247 AliCodeTimerAuto("",0)
29c9d15f 248
249 //Get input digits and put them in fgDigitsArr, clear the list before
85c25c2e 250 ReadDigitsArrayFromTree(digitsTree);
29c9d15f 251
9519e3e3 252 InitClusterizer();
253
0832a2bf 254 fgClusterizer->InitParameters();
4601e3a7 255 fgClusterizer->SetOutput(clustersTree);
9519e3e3 256
e853f058 257 //Skip clusterization of LED events
258 if (GetRecParam()->GetEventSpecie()!=AliRecoParam::kCalib){
9517d886 259
29c9d15f 260 if(fgDigitsArr && fgDigitsArr->GetEntries()) {
261
262 fgClusterizer->SetInput(digitsTree);
263
264 //fgClusterizer->Digits2Clusters("deb all") ; //For debugging
265 fgClusterizer->Digits2Clusters("");
266
267 fgClusterizer->Clear();
268
269 }//digits array exists and has somethind
e853f058 270 }//not a LED event
29c9d15f 271
916f1e76 272 clustersTree->Fill();
c828bc97 273
274 // Deleting the recpoints at the end of the reconstruction call
275 fgClusterizer->DeleteRecPoints();
f6019cda 276}
277
278//____________________________________________________________________________
c47157cd 279void AliEMCALReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const
29c9d15f 280
a68156e6 281{
c47157cd 282 // Conversion from raw data to
283 // EMCAL digits.
284 // Works on a single-event basis
95ffa314 285
98e9578e 286 rawReader->Reset() ;
29c9d15f 287
de39a0ff 288 fTriggerData->SetMode(1);
29c9d15f 289
b57cc9b9 290 if(fgDigitsArr) fgDigitsArr->Clear("C");
29c9d15f 291
de39a0ff 292 TClonesArray *digitsTrg = new TClonesArray("AliEMCALTriggerRawDigit", 32 * 96);
29c9d15f 293
c47157cd 294 Int_t bufsize = 32000;
b57cc9b9 295 digitsTree->Branch("EMCAL", &fgDigitsArr, bufsize);
916f1e76 296 digitsTree->Branch("EMTRG", &digitsTrg, bufsize);
29c9d15f 297
e853f058 298 //Skip calibration events do the rest
299 Bool_t doFit = kTRUE;
95ffa314 300 if ( !(GetRecParam()->FitLEDEvents()) && GetRecParam()->GetEventSpecie()==AliRecoParam::kCalib) doFit = kFALSE;
e853f058 301 if (doFit){
29c9d15f 302 //must be done here because, in constructor, option is not yet known
303 fgRawUtils->SetOption(GetOption());
304
92d9f317 305 // fgRawUtils->SetRawFormatHighLowGainFactor(GetRecParam()->GetHighLowGainFactor());
95ffa314 306
92d9f317 307 // fgRawUtils->SetRawFormatOrder(GetRecParam()->GetOrderParameter());
308 // fgRawUtils->SetRawFormatTau(GetRecParam()->GetTau());
29c9d15f 309 fgRawUtils->SetNoiseThreshold(GetRecParam()->GetNoiseThreshold());
310 fgRawUtils->SetNPedSamples(GetRecParam()->GetNPedSamples());
311 fgRawUtils->SetRemoveBadChannels(GetRecParam()->GetRemoveBadChannels());
48b634ff 312 if (!fgRawUtils->GetFittingAlgorithm()) fgRawUtils->SetFittingAlgorithm(GetRecParam()->GetFittingAlgorithm());
29c9d15f 313 fgRawUtils->SetFALTROUsage(GetRecParam()->UseFALTRO());
92d9f317 314
315 //fgRawUtils->SetTimeMin(GetRecParam()->GetTimeMin());
316 //fgRawUtils->SetTimeMax(GetRecParam()->GetTimeMax());
317
318 // fgRawUtils->SetTimeMin(-99999 );
319 // fgRawUtils->SetTimeMax( 99999 );
29c9d15f 320
321 fgRawUtils->Raw2Digits(rawReader,fgDigitsArr,fPedestalData,digitsTrg,fTriggerData);
92d9f317 322
793176c1 323 }//skip calibration event
e853f058 324 else{
29c9d15f 325 AliDebug(1," Calibration Event, skip!");
e853f058 326 }
29c9d15f 327
c615db53 328 digitsTree->Fill();
916f1e76 329 digitsTrg->Delete();
916f1e76 330 delete digitsTrg;
29c9d15f 331
a68156e6 332}
333
85c25c2e 334
a68156e6 335//____________________________________________________________________________
0e7c6655 336void AliEMCALReconstructor::FillESD(TTree* digitsTree, TTree* clustersTree,
c47157cd 337 AliESDEvent* esd) const
f6019cda 338{
98e9578e 339 // Called by AliReconstruct after Reconstruct() and global tracking and vertexing
85c25c2e 340 // and V0
c47157cd 341 // Works on the current event
de39a0ff 342 // printf(" ## AliEMCALReconstructor::FillESD() is started ### \n ");
85c25c2e 343 //return;
29c9d15f 344
7ba18c0e 345 //########################################
346 // Trigger
347 //########################################
de39a0ff 348
da6062af 349 static int saveOnce = 0;
95ffa314 350
29c9d15f 351 Int_t v0M[2] = {0, 0};
352
353 AliESDVZERO* esdV0 = esd->GetVZEROData();
354
355 if (esdV0)
356 {
357 for (Int_t i = 0; i < 32; i++)
358 {
359 v0M[0] += (Int_t)esdV0->GetAdcV0C(i);
360 v0M[1] += (Int_t)esdV0->GetAdcV0A(i);
361 }
362 }
363 else
364 {
365 AliWarning("Cannot retrieve V0 ESD! Run w/ null V0 charges");
366 }
367
368 if (fgTriggerDigits) fgTriggerDigits->Clear();
369
370 TBranch *branchtrg = digitsTree->GetBranch("EMTRG");
371
372 if (!branchtrg)
373 {
374 AliError("Can't get the branch with the EMCAL trigger digits!");
375 return;
376 }
377
378 branchtrg->SetAddress(&fgTriggerDigits);
379 branchtrg->GetEntry(0);
380
381 // Note: fgTriggerProcessor reset done at the end of this method
382 fgTriggerProcessor->Digits2Trigger(fgTriggerDigits, v0M, fTriggerData);
383
384 // Fill ESD
385 AliESDCaloTrigger* trgESD = esd->GetCaloTrigger("EMCAL");
386
387 if (trgESD)
388 {
389 trgESD->Allocate(fgTriggerDigits->GetEntriesFast());
390
391 for (Int_t i = 0; i < fgTriggerDigits->GetEntriesFast(); i++)
392 {
393 AliEMCALTriggerRawDigit* rdig = (AliEMCALTriggerRawDigit*)fgTriggerDigits->At(i);
de39a0ff 394
29c9d15f 395 Int_t px, py;
396 if (fGeom->GetPositionInEMCALFromAbsFastORIndex(rdig->GetId(), px, py))
397 {
398 Int_t a = -1, t = -1, times[10];
399
400 rdig->GetMaximum(a, t);
401 rdig->GetL0Times(times);
402
403 trgESD->Add(px, py, a, t, times, rdig->GetNL0Times(), rdig->GetL1TimeSum(), rdig->GetTriggerBits());
404 }
405 }
406
407 trgESD->SetL1Threshold(0, fTriggerData->GetL1GammaThreshold());
408
409 trgESD->SetL1Threshold(1, fTriggerData->GetL1JetThreshold() );
95ffa314 410
411 Int_t v0[2];
412 fTriggerData->GetL1V0(v0);
413
414 trgESD->SetL1V0(v0);
415 trgESD->SetL1FrameMask(fTriggerData->GetL1FrameMask());
416
417 if (!saveOnce && fTriggerData->GetL1DataDecoded())
418 {
419 int type[8] = {0};
420 fTriggerData->GetL1TriggerType(type);
421
422 esd->SetCaloTriggerType(type);
423
424 saveOnce = 1;
425 }
29c9d15f 426 }
427
428 // Resetting
429 fTriggerData->Reset();
de39a0ff 430
0e7c6655 431 //########################################
432 //##############Fill CaloCells###############
433 //########################################
29c9d15f 434
435 //Get input digits and put them in fgDigitsArr, clear the list before
b57cc9b9 436 ReadDigitsArrayFromTree(digitsTree);
29c9d15f 437
b57cc9b9 438 Int_t nDigits = fgDigitsArr->GetEntries(), idignew = 0 ;
0e7c6655 439 AliDebug(1,Form("%d digits",nDigits));
0e7c6655 440 AliESDCaloCells &emcCells = *(esd->GetEMCALCells());
441 emcCells.CreateContainer(nDigits);
c8fe2783 442 emcCells.SetType(AliVCaloCells::kEMCALCell);
0c5b726e 443 Float_t energy = 0;
783153ff 444 Float_t time = 0;
0e7c6655 445 for (Int_t idig = 0 ; idig < nDigits ; idig++) {
b57cc9b9 446 const AliEMCALDigit * dig = (const AliEMCALDigit*)fgDigitsArr->At(idig);
783153ff 447 time = dig->GetTime(); // Time already calibrated in clusterizer
448 energy = dig->GetAmplitude(); // energy calibrated in clusterizer
449 if(energy > 0 ){
450 fgClusterizer->Calibrate(energy,time,dig->GetId()); //Digits already calibrated in clusterizers
451 if(energy > 0){ //Digits tagged as bad (dead, hot, not alive) are set to 0 in calibrate, remove them
452 emcCells.SetCell(idignew,dig->GetId(),energy, time);
95ffa314 453 idignew++;
29c9d15f 454 }
0e7c6655 455 }
456 }
457 emcCells.SetNumberOfCells(idignew);
458 emcCells.Sort();
29c9d15f 459
0e7c6655 460 //------------------------------------------------------------
461 //-----------------CLUSTERS-----------------------------
462 //------------------------------------------------------------
9dce5a21 463 clustersTree->SetBranchStatus("*",0); //disable all branches
464 clustersTree->SetBranchStatus("EMCALECARP",1); //Enable only the branch we need
b57cc9b9 465 if(fgClustersArr) fgClustersArr->Clear();
0e7c6655 466 TBranch *branch = clustersTree->GetBranch("EMCALECARP");
b57cc9b9 467 branch->SetAddress(&fgClustersArr);
9dce5a21 468 branch->GetEntry(0);
469 //clustersTree->GetEvent(0);
29c9d15f 470
b57cc9b9 471 Int_t nClusters = fgClustersArr->GetEntries(), nClustersNew=0;
0e7c6655 472 AliDebug(1,Form("%d clusters",nClusters));
5970dfe2 473
85c25c2e 474
6a0cf740 475 //########################################
85c25c2e 476 //##############Fill CaloClusters#############
6a0cf740 477 //########################################
5dee926e 478 for (Int_t iClust = 0 ; iClust < nClusters ; iClust++) {
b57cc9b9 479 const AliEMCALRecPoint * clust = (const AliEMCALRecPoint*)fgClustersArr->At(iClust);
5970dfe2 480 if(!clust) continue;
c8fe2783 481 //if(clust->GetClusterType()== AliVCluster::kEMCALClusterv1) nRP++; else nPC++;
29c9d15f 482 // clust->Print(); //For debugging
a7a5421e 483 // Get information from EMCAL reconstruction points
85c60a8e 484 Float_t xyz[3];
5dee926e 485 TVector3 gpos;
486 clust->GetGlobalPosition(gpos);
35397e76 487 for (Int_t ixyz=0; ixyz<3; ixyz++)
5dee926e 488 xyz[ixyz] = gpos[ixyz];
85c25c2e 489 Float_t elipAxis[2];
490 clust->GetElipsAxis(elipAxis);
29c9d15f 491 //Create digits lists
35397e76 492 Int_t cellMult = clust->GetMultiplicity();
493 //TArrayS digiList(digitMult);
494 Float_t *amplFloat = clust->GetEnergiesList();
495 Int_t *digitInts = clust->GetAbsId();
496 TArrayS absIdList(cellMult);
eb972628 497 TArrayD fracList(cellMult);
29c9d15f 498
35397e76 499 Int_t newCellMult = 0;
500 for (Int_t iCell=0; iCell<cellMult; iCell++) {
501 if (amplFloat[iCell] > 0) {
95ffa314 502 absIdList[newCellMult] = (UShort_t)(digitInts[iCell]);
503 //Calculate Fraction
504 if(emcCells.GetCellAmplitude(digitInts[iCell])>0 && GetRecParam()->GetUnfold()){
505 fracList[newCellMult] = amplFloat[iCell]/(emcCells.GetCellAmplitude(digitInts[iCell]));//get cell calibration value
506
507 }
508 else{
509 fracList[newCellMult] = 0;
510 }
511 newCellMult++;
92da3372 512 }
92da3372 513 }
29c9d15f 514
eb972628 515 absIdList.Set(newCellMult);
516 fracList.Set(newCellMult);
517
35397e76 518 if(newCellMult > 0) { // accept cluster if it has some digit
519 nClustersNew++;
65721814 520 //Primaries
7592dfc4 521 Int_t parentMult = 0;
fa42b1f3 522 Int_t *parentList = clust->GetParents(parentMult);
a7a5421e 523 // fills the ESDCaloCluster
35397e76 524 AliESDCaloCluster * ec = new AliESDCaloCluster() ;
c8fe2783 525 ec->SetType(AliVCluster::kEMCALClusterv1);
7592dfc4 526 ec->SetPosition(xyz);
527 ec->SetE(clust->GetEnergy());
29c9d15f 528
ee08edde 529 //Distance to the nearest bad crystal
530 ec->SetDistanceToBadChannel(clust->GetDistanceToBadTower());
29c9d15f 531
35397e76 532 ec->SetNCells(newCellMult);
533 //Change type of list from short to ushort
534 UShort_t *newAbsIdList = new UShort_t[newCellMult];
ee08edde 535 Double_t *newFracList = new Double_t[newCellMult];
35397e76 536 for(Int_t i = 0; i < newCellMult ; i++) {
537 newAbsIdList[i]=absIdList[i];
de39a0ff 538 newFracList[i] =fracList[i];
35397e76 539 }
540 ec->SetCellsAbsId(newAbsIdList);
eb972628 541 ec->SetCellsAmplitudeFraction(newFracList);
c8fe2783 542 ec->SetDispersion(clust->GetDispersion());
543 ec->SetChi2(-1); //not yet implemented
35397e76 544 ec->SetM02(elipAxis[0]*elipAxis[0]) ;
545 ec->SetM20(elipAxis[1]*elipAxis[1]) ;
78902954 546 ec->SetTOF(clust->GetTime()) ; //time-of-fligh
225cd96d 547 ec->SetNExMax(clust->GetNExMax()); //number of local maxima
5970dfe2 548
29c9d15f 549
35397e76 550 TArrayI arrayParents(parentMult,parentList);
551 ec->AddLabels(arrayParents);
5970dfe2 552 //
553 //Track matching
554 //
555 fMatches->Clear();
556 Int_t nTracks = esd->GetNumberOfTracks();
557 for (Int_t itrack = 0; itrack < nTracks; itrack++)
558 {
559 AliESDtrack * track = esd->GetTrack(itrack) ; // retrieve track
560 if(track->GetEMCALcluster()==iClust)
561 {
562 Double_t dEta=-999, dPhi=-999;
563 Bool_t isMatch = CalculateResidual(track, ec, dEta, dPhi);
564 if(!isMatch)
565 {
a8fbdc61 566 // AliDebug(10, "Not good");
5970dfe2 567 continue;
5970dfe2 568 }
569 AliEMCALMatch *match = new AliEMCALMatch();
570 match->SetIndexT(itrack);
571 match->SetDistance(TMath::Sqrt(dEta*dEta+dPhi*dPhi));
572 match->SetdEta(dEta);
573 match->SetdPhi(dPhi);
574 fMatches->Add(match);
575 }
576 }
577 fMatches->Sort(kSortAscending); //Sort matched tracks from closest to furthest
578 Int_t nMatch = fMatches->GetEntries();
579 TArrayI arrayTrackMatched(nMatch);
580 for(Int_t imatch=0; imatch<nMatch; imatch++)
581 {
582 AliEMCALMatch *match = (AliEMCALMatch*)fMatches->At(imatch);
583 arrayTrackMatched[imatch] = match->GetIndexT();
584 if(imatch==0)
585 {
586 ec->SetTrackDistance(match->GetdPhi(), match->GetdEta());
587 }
588 }
589 ec->AddTracksMatched(arrayTrackMatched);
590
591 //add the cluster to the esd object
eb972628 592 esd->AddCaloCluster(ec);
5970dfe2 593
a7a5421e 594 delete ec;
85d4cbde 595 delete [] newAbsIdList ;
eb972628 596 delete [] newFracList ;
29c9d15f 597 }
598 } // cycle on clusters
5970dfe2 599
600 //
601 //Reset the index of matched cluster for tracks
602 //to the one in CaloCluster array
603 Int_t ncls = esd->GetNumberOfCaloClusters();
604 for(Int_t icl=0; icl<ncls; icl++)
605 {
606 AliESDCaloCluster *cluster = esd->GetCaloCluster(icl);
607 if(!cluster || !cluster->IsEMCAL()) continue;
608 TArrayI *trackIndex = cluster->GetTracksMatched();
609 for(Int_t itr=0; itr<trackIndex->GetSize(); itr++)
610 {
611 AliESDtrack *track = esd->GetTrack(trackIndex->At(itr));
612 track->SetEMCALcluster(cluster->GetID());
613 }
614 }
29c9d15f 615
29c9d15f 616
617 //Fill ESDCaloCluster with PID weights
618 AliEMCALPID *pid = new AliEMCALPID;
619 //pid->SetPrintInfo(kTRUE);
620 pid->SetReconstructor(kTRUE);
621 pid->RunPID(esd);
622 delete pid;
623
624 //Store EMCAL misalignment matrixes
625 FillMisalMatrixes(esd) ;
626
0c5b726e 627}
628
629//==================================================================================
630void AliEMCALReconstructor::FillMisalMatrixes(AliESDEvent* esd)const{
29c9d15f 631 //Store EMCAL matrixes in ESD Header
632
633 //Check, if matrixes was already stored
634 for(Int_t sm = 0 ; sm < fGeom->GetNumberOfSuperModules(); sm++){
635 if(esd->GetEMCALMatrix(sm)!=0)
636 return ;
637 }
638
639 //Create and store matrixes
640 if(!gGeoManager){
641 AliError("Can not store misal. matrixes: no gGeoManager! \n") ;
642 return ;
643 }
644 //Note, that owner of copied marixes will be header
7e1d9a9b 645 const Int_t bufsize = 255;
29c9d15f 646 char path[bufsize] ;
647 TGeoHMatrix * m = 0x0;
648 for(Int_t sm = 0; sm < fGeom->GetNumberOfSuperModules(); sm++){
649 snprintf(path,bufsize,"/ALIC_1/XEN1_1/SMOD_%d",sm+1) ; //In Geometry modules numbered 1,2,.,5
650 if(sm >= 10) snprintf(path,bufsize,"/ALIC_1/XEN1_1/SM10_%d",sm-10+1) ;
651
652 if (gGeoManager->CheckPath(path)){
653 gGeoManager->cd(path);
654 m = gGeoManager->GetCurrentMatrix() ;
655 // printf("================================================= \n");
656 // printf("AliEMCALReconstructor::FixMisalMatrixes(), sm %d, \n",sm);
657 // m->Print("");
658 esd->SetEMCALMatrix(new TGeoHMatrix(*m),sm) ;
659 // printf("================================================= \n");
660 }
661 else{
662 esd->SetEMCALMatrix(NULL,sm) ;
663 }
664 }
f6019cda 665}
dc293ae9 666
9517d886 667//__________________________________________________________________________
85c25c2e 668void AliEMCALReconstructor::ReadDigitsArrayFromTree(TTree *digitsTree) const
669{
29c9d15f 670 // Read the digits from the input tree
671 // See AliEMCALClusterizer::SetInput(TTree *digitsTree);
672
673 // Clear previous digits in the list
674 if(fgDigitsArr){
b57cc9b9 675 fgDigitsArr->Clear("C");
85c25c2e 676 }
29c9d15f 677 else{
678 // It should not happen, but just in case ...
679 fgDigitsArr = new TClonesArray("AliEMCALDigit",100);
680 }
681
85c25c2e 682 // Read the digits from the input tree
683 TBranch *branch = digitsTree->GetBranch("EMCAL");
684 if (!branch) {
685 AliError("can't get the branch with the EMCAL digits !");
686 return;
29c9d15f 687 }
688
85c25c2e 689 branch->SetAddress(&fgDigitsArr);
690 branch->GetEntry(0);
691}
98e9578e 692
5970dfe2 693//==================================================================================
694Bool_t AliEMCALReconstructor::CalculateResidual(AliESDtrack *track, AliESDCaloCluster *cluster, Double_t &dEta, Double_t &dPhi)const
695{
696 //
697 // calculate the residual between track and cluster
698 //
699
700 // If the esdFriend is available, use the TPCOuter point as the starting point of extrapolation
701 // Otherwise use the TPCInner point
456126ad 702 AliExternalTrackParam *trkParam = 0;
5970dfe2 703 const AliESDfriendTrack* friendTrack = track->GetFriendTrack();
704 if(friendTrack && friendTrack->GetTPCOut())
705 trkParam = const_cast<AliExternalTrackParam*>(friendTrack->GetTPCOut());
706 else
707 trkParam = const_cast<AliExternalTrackParam*>(track->GetInnerParam());
708 if(!trkParam) return kFALSE;
709
710 //Perform extrapolation
711 Double_t trkPos[3];
712 Float_t clsPos[3];
713
456126ad 714 AliExternalTrackParam trkParamTmp (*trkParam);
5970dfe2 715 cluster->GetPosition(clsPos);
716 TVector3 vec(clsPos[0],clsPos[1],clsPos[2]);
717 Double_t alpha = ((int)(vec.Phi()*TMath::RadToDeg()/20)+0.5)*20*TMath::DegToRad();
718 //Rotate to proper local coordinate
719 vec.RotateZ(-alpha);
5970dfe2 720 //extrapolation is done here
d874f457 721 if(!AliTrackerBase::PropagateTrackToBxByBz(&trkParamTmp, vec.X(), track->GetMass(), GetRecParam()->GetExtrapolateStep(), kTRUE, 0.8, -1))
5970dfe2 722 return kFALSE;
723
724 //Calculate the residuals
456126ad 725 trkParamTmp.GetXYZ(trkPos);
5970dfe2 726
727 TVector3 clsPosVec(clsPos[0],clsPos[1],clsPos[2]);
728 TVector3 trkPosVec(trkPos[0],trkPos[1],trkPos[2]);
5970dfe2 729
d874f457 730 dPhi = clsPosVec.DeltaPhi(trkPosVec);
5970dfe2 731 dEta = clsPosVec.Eta()-trkPosVec.Eta();
732
733 return kTRUE;
734}
735
736//
737//==================================================================================
738//
739AliEMCALReconstructor::AliEMCALMatch::AliEMCALMatch()
740 : TObject(),
741 fIndexT(-1),
742 fDistance(-999.),
743 fdEta(-999.),
744 fdPhi(-999.)
745{
746 //default constructor
916f1e76 747
5970dfe2 748}
749
750//
751//==================================================================================
752//
753AliEMCALReconstructor::AliEMCALMatch::AliEMCALMatch(const AliEMCALMatch& copy)
754 : TObject(),
755 fIndexT(copy.fIndexT),
756 fDistance(copy.fDistance),
757 fdEta(copy.fdEta),
758 fdPhi(copy.fdPhi)
759{
760 //copy ctor
761}
762
763//
764//==================================================================================
765//
766Int_t AliEMCALReconstructor::AliEMCALMatch::Compare(const TObject *obj) const
767{
768 //
769 // Compare wrt the residual
770 //
771
772 AliEMCALReconstructor::AliEMCALMatch *that = (AliEMCALReconstructor::AliEMCALMatch*)obj;
773
774 Double_t thisDist = fDistance;//fDistance;
775 Double_t thatDist = that->fDistance;//that->GetDistance();
776
777 if (thisDist > thatDist) return 1;
778 else if (thisDist < thatDist) return -1;
779 return 0;
780}