Return null pointer if no tracks were found
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALReconstructor.cxx
CommitLineData
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
31#include "AliESD.h"
32#include "AliRunLoader.h"
33#include "AliEMCALLoader.h"
f6019cda 34#include "AliEMCALClusterizerv1.h"
5dee926e 35#include "AliEMCALRecPoint.h"
dc293ae9 36#include "AliEMCALPID.h"
1d59832c 37#include "AliRawReader.h"
38
f6019cda 39
40ClassImp(AliEMCALReconstructor)
41
42//____________________________________________________________________________
18a21c7c 43AliEMCALReconstructor::AliEMCALReconstructor()
44 : fDebug(kFALSE)
f6019cda 45{
46 // ctor
f6019cda 47}
48
0a4cb131 49//____________________________________________________________________________
18a21c7c 50AliEMCALReconstructor::AliEMCALReconstructor(const AliEMCALReconstructor & rec)
51 : AliReconstructor(rec),
52 fDebug(rec.fDebug)
0a4cb131 53{
54 //copy ctor
0a4cb131 55}
f6019cda 56
57//____________________________________________________________________________
58AliEMCALReconstructor::~AliEMCALReconstructor()
59{
60 // dtor
61}
62
63//____________________________________________________________________________
64void AliEMCALReconstructor::Reconstruct(AliRunLoader* runLoader) const
65{
66 // method called by AliReconstruction;
67 // Only the clusterization is performed,; the rest of the reconstruction is done in FillESD because the track
68 // segment maker needs access to the AliESD object to retrieve the tracks reconstructed by
69 // the global tracking.
70
71 TString headerFile(runLoader->GetFileName()) ;
b4975670 72 TString branchName(runLoader->GetEventFolder()->GetName() ) ;
f6019cda 73
74 AliEMCALClusterizerv1 clu(headerFile, branchName);
75 clu.SetEventRange(0, -1) ; // do all the events
76 if ( Debug() )
77 clu.ExecuteTask("deb all") ;
78 else
85c60a8e 79 clu.ExecuteTask("pseudo") ;
80
f6019cda 81}
82
83//____________________________________________________________________________
5dee926e 84void AliEMCALReconstructor::Reconstruct(AliRunLoader* runLoader, AliRawReader* rawreader) const
a68156e6 85{
86 // method called by AliReconstruction;
87 // Only the clusterization is performed,; the rest of the reconstruction is done in FillESD because the track
88 // segment maker needs access to the AliESD object to retrieve the tracks reconstructed by
89 // the global tracking.
90 // Here we reconstruct from Raw Data
91
92 rawreader->Reset() ;
93 TString headerFile(runLoader->GetFileName()) ;
94 TString branchName(runLoader->GetEventFolder()->GetName()) ;
85c60a8e 95
a68156e6 96 AliEMCALClusterizerv1 clu(headerFile, branchName);
97 clu.SetEventRange(0, -1) ; // do all the events
98 if ( Debug() )
85c60a8e 99 clu.ExecuteTask("deb pseudo all") ;
a68156e6 100 else
85c60a8e 101 clu.ExecuteTask("pseudo") ;
a68156e6 102
103}
104
105//____________________________________________________________________________
f6019cda 106void AliEMCALReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
107{
108 // Called by AliReconstruct after Reconstruct() and global tracking and vertxing
92da3372 109
b4975670 110 Int_t eventNumber = runLoader->GetEventNumber() ;
111
1963b290 112 TString headerFile(runLoader->GetFileName()) ;
113 TString branchName(runLoader->GetEventFolder()->GetName()) ;
85c60a8e 114 // Creates AliESDCaloCluster from AliEMCALRecPoints
5dee926e 115 AliRunLoader *rl = AliRunLoader::GetRunLoader();
116 AliEMCALLoader *emcalLoader = dynamic_cast<AliEMCALLoader*>(rl->GetDetectorLoader("EMCAL"));
117 rl->LoadRecPoints();
92da3372 118 rl->LoadKinematics(); // To get the primary label
119 rl->LoadDigits(); // To get the primary label
120 rl->LoadHits(); // To get the primary label
5dee926e 121 rl->GetEvent(eventNumber);
122 TObjArray *clusters = emcalLoader->RecPoints();
123 Int_t nClusters = clusters->GetEntries();
85c60a8e 124 esd->SetNumberOfEMCALClusters(nClusters) ;
125 esd->SetFirstEMCALCluster(esd->GetNumberOfCaloClusters()) ;
92da3372 126
5dee926e 127 for (Int_t iClust = 0 ; iClust < nClusters ; iClust++) {
128 const AliEMCALRecPoint * clust = emcalLoader->RecPoint(iClust);
85c60a8e 129
130 if (Debug()) clust->Print();
85c60a8e 131 AliESDCaloCluster * ec = new AliESDCaloCluster() ;
132 // fills the ESDCaloCluster
133 Float_t xyz[3];
5dee926e 134 TVector3 gpos;
135 clust->GetGlobalPosition(gpos);
f6019cda 136 for (Int_t ixyz=0; ixyz<3; ixyz++)
5dee926e 137 xyz[ixyz] = gpos[ixyz];
92da3372 138
85c60a8e 139 Int_t digitMult = clust->GetMultiplicity();
140 UShort_t *amplList = new UShort_t[digitMult];
141 UShort_t *timeList = new UShort_t[digitMult];
142 UShort_t *digiList = new UShort_t[digitMult];
143 Float_t *amplFloat = clust->GetEnergiesList();
144 Float_t *timeFloat = clust->GetTimeList();
145 Int_t *digitInts = clust->GetAbsId();
8a11aef1 146 Float_t elipAxis[2];
92da3372 147 clust->GetElipsAxis(elipAxis);
85c60a8e 148
92da3372 149 // Convert Float_t* and Int_t* to UShort_t* to save memory
150 Int_t newdigitMult = digitMult ;
151 for (Int_t iDigit=0; iDigit<digitMult; iDigit++) {
152 if(timeFloat[iDigit] < 65536/1e9*100){
153 amplList[iDigit] = (UShort_t)(amplFloat[iDigit]*500);
154 timeList[iDigit] = (UShort_t)(timeFloat[iDigit]*1e9*100); //Time in units of 100 ns = 0.1 ps
155 digiList[iDigit] = (UShort_t)(digitInts[iDigit]);
156 }
157 else
158 newdigitMult = newdigitMult - 1 ;
159 }
160
85c60a8e 161 ec->SetClusterType(clust->GetClusterType());
162 ec->SetGlobalPosition(xyz);
163 ec->SetClusterEnergy(clust->GetEnergy());
85c60a8e 164 ec->SetDigitAmplitude(amplList); //energies
165 ec->SetDigitTime(timeList); //times
166 ec->SetDigitIndex(digiList); //indices
92da3372 167 ec->SetNumberOfDigits(newdigitMult);
1d59832c 168 if(clust->GetClusterType()== AliESDCaloCluster::kClusterv1){
92da3372 169 ec->SetClusterDisp(clust->GetDispersion());
170 ec->SetClusterChi2(-1); //not yet implemented
171 ec->SetM02(elipAxis[0]*elipAxis[0]) ;
172 ec->SetM20(elipAxis[1]*elipAxis[1]) ;
173 ec->SetM11(-1) ; //not yet implemented
174 ec->SetPrimaryIndex(clust->GetPrimaryIndex());
175 }
85c60a8e 176 // add the cluster to the esd object
177 esd->AddCaloCluster(ec);
dc293ae9 178
179 delete ec;
f6019cda 180 }
dc293ae9 181 //Fill ESDCaloCluster with PID weights
182
183 AliEMCALPID *pid = new AliEMCALPID;
184 //pid->SetPrintInfo(kTRUE);
185 pid->SetReconstructor(kTRUE);
186 pid->RunPID(esd);
187
f6019cda 188}
dc293ae9 189