Coding conventions (Gustavo)
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALRecPoint.cxx
CommitLineData
ab48128d 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/* $Id$ */
16//_________________________________________________________________________
70a93198 17// Reconstructed Points for the EMCAL
18// A RecPoint is a cluster of digits
d64c959b 19//*-- Author: Yves Schutz (SUBATECH)
70a93198 20//*-- Author: Dmitri Peressounko (RRC KI & SUBATECH)
21//*-- Author: Heather Gray (LBL) merged AliEMCALRecPoint and AliEMCALTowerRecPoint 02/04
ab48128d 22
23// --- ROOT system ---
1d59832c 24class Riostream;
e52475ed 25#include <TPad.h>
1d59832c 26class TGraph;
27class TPaveText;
e52475ed 28#include <TClonesArray.h>
29#include <TMath.h>
ab48128d 30
31// --- Standard library ---
ab48128d 32
33// --- AliRoot header files ---
1d59832c 34//#include "AliGenerator.h"
35class AliGenerator;
4635df1f 36#include "AliRunLoader.h"
37#include "AliRun.h"
1d59832c 38class AliEMCAL;
4635df1f 39#include "AliEMCALLoader.h"
ab48128d 40#include "AliEMCALGeometry.h"
4635df1f 41#include "AliEMCALHit.h"
ab48128d 42#include "AliEMCALDigit.h"
43#include "AliEMCALRecPoint.h"
ab48128d 44
45ClassImp(AliEMCALRecPoint)
46
ab48128d 47//____________________________________________________________________________
48AliEMCALRecPoint::AliEMCALRecPoint()
49 : AliRecPoint()
50{
51 // ctor
85c60a8e 52 fClusterType = -1;
692088ae 53 fMaxTrack = 0 ;
70a93198 54 fMulDigit = 0 ;
87cdc3be 55 fMaxParent = 0;
56 fMulParent = 0;
70a93198 57 fAmp = 0. ;
58 fCoreEnergy = 0 ;
59 fEnergyList = 0 ;
85c60a8e 60 fTimeList = 0 ;
e52475ed 61 fAbsIdList = 0;
87cdc3be 62 fParentsList = 0;
70a93198 63 fTime = 0. ;
e52475ed 64 // fLocPos.SetX(1.e+6) ; //Local position should be evaluated
70a93198 65 fCoreRadius = 10; //HG Check this
4635df1f 66
67 AliRunLoader *rl = AliRunLoader::GetRunLoader();
14ce0a6e 68 fGeomPtr = dynamic_cast<AliEMCAL*>(rl->GetAliRun()->GetDetector("EMCAL"))->GetGeometry();
69 //fGeomPtr = AliEMCALGeometry::GetInstance();
70 fGeomPtr->GetTransformationForSM(); // Global <-> Local
ab48128d 71}
72
73//____________________________________________________________________________
74AliEMCALRecPoint::AliEMCALRecPoint(const char * opt) : AliRecPoint(opt)
75{
76 // ctor
85c60a8e 77 fClusterType = -1;
ff1e7e2f 78 fMaxTrack = 1000 ;
79 fMaxParent = 1000;
87cdc3be 80 fMulDigit = 0 ;
81 fMulParent = 0;
70a93198 82 fAmp = 0. ;
83 fCoreEnergy = 0 ;
84 fEnergyList = 0 ;
85c60a8e 85 fTimeList = 0 ;
e52475ed 86 fAbsIdList = 0;
87cdc3be 87 fParentsList = new Int_t[fMaxParent];
70a93198 88 fTime = -1. ;
e52475ed 89 //fLocPos.SetX(1.e+6) ; //Local position should be evaluated
70a93198 90 fCoreRadius = 10; //HG Check this
14ce0a6e 91 //fGeomPtr = AliEMCALGeometry::GetInstance();
4635df1f 92 AliRunLoader *rl = AliRunLoader::GetRunLoader();
14ce0a6e 93 fGeomPtr = dynamic_cast<AliEMCAL*>(rl->GetAliRun()->GetDetector("EMCAL"))->GetGeometry();
94 fGeomPtr->GetTransformationForSM(); // Global <-> Local
70a93198 95}
96//____________________________________________________________________________
97AliEMCALRecPoint::~AliEMCALRecPoint()
98{
99 // dtor
100 if ( fEnergyList )
101 delete[] fEnergyList ;
85c60a8e 102 if ( fTimeList )
103 delete[] fTimeList ;
e52475ed 104 if ( fAbsIdList )
105 delete[] fAbsIdList ;
87cdc3be 106 if ( fParentsList)
107 delete[] fParentsList;
70a93198 108}
109
110//____________________________________________________________________________
111void AliEMCALRecPoint::AddDigit(AliEMCALDigit & digit, Float_t Energy)
112{
113 // Adds a digit to the RecPoint
114 // and accumulates the total amplitude and the multiplicity
115
116 if(fEnergyList == 0)
117 fEnergyList = new Float_t[fMaxDigit];
85c60a8e 118 if(fTimeList == 0)
119 fTimeList = new Float_t[fMaxDigit];
e52475ed 120 if(fAbsIdList == 0) {
121 fAbsIdList = new Int_t[fMaxDigit];
14ce0a6e 122 fSuperModuleNumber = fGeomPtr->GetSuperModuleNumber(digit.GetId());
e52475ed 123 }
70a93198 124
125 if ( fMulDigit >= fMaxDigit ) { // increase the size of the lists
126 fMaxDigit*=2 ;
a64a06d6 127 Int_t * tempo = new Int_t[fMaxDigit];
128 Float_t * tempoE = new Float_t[fMaxDigit];
85c60a8e 129 Float_t * tempoT = new Float_t[fMaxDigit];
e52475ed 130 Int_t * tempoId = new Int_t[fMaxDigit];
70a93198 131
132 Int_t index ;
133 for ( index = 0 ; index < fMulDigit ; index++ ){
e52475ed 134 tempo[index] = fDigitsList[index] ;
135 tempoE[index] = fEnergyList[index] ;
85c60a8e 136 tempoT[index] = fTimeList[index] ;
e52475ed 137 tempoId[index] = fAbsIdList[index] ;
70a93198 138 }
139
140 delete [] fDigitsList ;
a64a06d6 141 fDigitsList = new Int_t[fMaxDigit];
70a93198 142
143 delete [] fEnergyList ;
a64a06d6 144 fEnergyList = new Float_t[fMaxDigit];
70a93198 145
85c60a8e 146 delete [] fTimeList ;
147 fTimeList = new Float_t[fMaxDigit];
148
e52475ed 149 delete [] fAbsIdList ;
150 fAbsIdList = new Int_t[fMaxDigit];
151
70a93198 152 for ( index = 0 ; index < fMulDigit ; index++ ){
153 fDigitsList[index] = tempo[index] ;
154 fEnergyList[index] = tempoE[index] ;
85c60a8e 155 fTimeList[index] = tempoT[index] ;
e52475ed 156 fAbsIdList[index] = tempoId[index] ;
70a93198 157 }
158
159 delete [] tempo ;
160 delete [] tempoE ;
85c60a8e 161 delete [] tempoT ;
e52475ed 162 delete [] tempoId ;
70a93198 163 } // if
164
165 fDigitsList[fMulDigit] = digit.GetIndexInList() ;
166 fEnergyList[fMulDigit] = Energy ;
85c60a8e 167 fTimeList[fMulDigit] = digit.GetTime() ;
e52475ed 168 fAbsIdList[fMulDigit] = digit.GetId();
70a93198 169 fMulDigit++ ;
170 fAmp += Energy ;
171
172}
173//____________________________________________________________________________
174Bool_t AliEMCALRecPoint::AreNeighbours(AliEMCALDigit * digit1, AliEMCALDigit * digit2 ) const
175{
176 // Tells if (true) or not (false) two digits are neighbours
177 // A neighbour is defined as being two digits which share a corner
178
e52475ed 179 static Bool_t areNeighbours = kFALSE ;
180 static Int_t nSupMod=0, nTower=0, nIphi=0, nIeta=0;
181 static int nSupMod1=0, nTower1=0, nIphi1=0, nIeta1=0;
182 static Int_t relid1[2] , relid2[2] ; // ieta, iphi
183 static Int_t rowdiff=0, coldiff=0;
70a93198 184
e52475ed 185 areNeighbours = kFALSE ;
186
14ce0a6e 187 fGeomPtr->GetCellIndex(digit1->GetId(), nSupMod,nTower,nIphi,nIeta);
188 fGeomPtr->GetCellPhiEtaIndexInSModule(nSupMod,nTower,nIphi,nIeta, relid1[0],relid1[1]);
e52475ed 189
14ce0a6e 190 fGeomPtr->GetCellIndex(digit2->GetId(), nSupMod1,nTower1,nIphi1,nIeta1);
191 fGeomPtr->GetCellPhiEtaIndexInSModule(nSupMod1,nTower1,nIphi1,nIeta1, relid2[0],relid2[1]);
70a93198 192
e52475ed 193 rowdiff = TMath::Abs( relid1[0] - relid2[0] ) ;
194 coldiff = TMath::Abs( relid1[1] - relid2[1] ) ;
70a93198 195
196 if (( coldiff <= 1 ) && ( rowdiff <= 1 ) && (coldiff + rowdiff > 0))
e52475ed 197 areNeighbours = kTRUE ;
ab48128d 198
70a93198 199 return areNeighbours;
200}
201
202//____________________________________________________________________________
203Int_t AliEMCALRecPoint::Compare(const TObject * obj) const
204{
205 // Compares two RecPoints according to their position in the EMCAL modules
206
207 Float_t delta = 1 ; //Width of "Sorting row". If you change this
208 //value (what is senseless) change as well delta in
209 //AliEMCALTrackSegmentMakerv* and other RecPoints...
210 Int_t rv ;
211
212 AliEMCALRecPoint * clu = (AliEMCALRecPoint *)obj ;
213
214 TVector3 locpos1;
215 GetLocalPosition(locpos1);
216 TVector3 locpos2;
217 clu->GetLocalPosition(locpos2);
218
9848d950 219 Int_t rowdif = (Int_t)(TMath::Ceil(locpos1.X()/delta)-TMath::Ceil(locpos2.X()/delta)) ;
70a93198 220 if (rowdif> 0)
221 rv = 1 ;
222 else if(rowdif < 0)
223 rv = -1 ;
224 else if(locpos1.Y()>locpos2.Y())
225 rv = -1 ;
226 else
227 rv = 1 ;
228
229 return rv ;
ab48128d 230}
231
232//____________________________________________________________________________
233Int_t AliEMCALRecPoint::DistancetoPrimitive(Int_t px, Int_t py)
234{
235 // Compute distance from point px,py to a AliEMCALRecPoint considered as a Tmarker
236 // Compute the closest distance of approach from point px,py to this marker.
237 // The distance is computed in pixels units.
70a93198 238 // HG Still need to update -> Not sure what this should achieve
ab48128d 239
240 TVector3 pos(0.,0.,0.) ;
70a93198 241 GetLocalPosition(pos) ;
ab48128d 242 Float_t x = pos.X() ;
70a93198 243 Float_t y = pos.Y() ;
ab48128d 244 const Int_t kMaxDiff = 10;
245 Int_t pxm = gPad->XtoAbsPixel(x);
246 Int_t pym = gPad->YtoAbsPixel(y);
247 Int_t dist = (px-pxm)*(px-pxm) + (py-pym)*(py-pym);
248
249 if (dist > kMaxDiff) return 9999;
250 return dist;
251}
252
253//___________________________________________________________________________
254 void AliEMCALRecPoint::Draw(Option_t *option)
255 {
256 // Draw this AliEMCALRecPoint with its current attributes
257
258 AppendPad(option);
259 }
260
261//______________________________________________________________________________
70a93198 262void AliEMCALRecPoint::ExecuteEvent(Int_t /*event*/, Int_t, Int_t)
ab48128d 263{
264 // Execute action corresponding to one event
265 // This member function is called when a AliEMCALRecPoint is clicked with the locator
266 //
267 // If Left button is clicked on AliEMCALRecPoint, the digits are switched on
268 // and switched off when the mouse button is released.
269
270 // static Int_t pxold, pyold;
271
70a93198 272 /* static TGraph * digitgraph = 0 ;
ab48128d 273 static TPaveText* clustertext = 0 ;
274
275 if (!gPad->IsEditable()) return;
276
277 switch (event) {
278
279
280 case kButton1Down:{
281 AliEMCALDigit * digit ;
88cb7938 282 AliEMCALGeometry * emcalgeom = (AliEMCALGetter::Instance())->EMCALGeometry() ;
ab48128d 283
284 Int_t iDigit;
70a93198 285 Int_t relid[2] ;
ab48128d 286
287 const Int_t kMulDigit=AliEMCALRecPoint::GetDigitsMultiplicity() ;
288 Float_t * xi = new Float_t [kMulDigit] ;
289 Float_t * zi = new Float_t [kMulDigit] ;
290
291 for(iDigit = 0; iDigit < kMulDigit; iDigit++) {
292 Fatal("AliEMCALRecPoint::ExecuteEvent", " -> Something wrong with the code");
293 digit = 0 ; //dynamic_cast<AliEMCALDigit *>((fDigitsList)[iDigit]);
294 emcalgeom->AbsToRelNumbering(digit->GetId(), relid) ;
295 emcalgeom->PosInAlice(relid, xi[iDigit], zi[iDigit]) ;
296 }
297
298 if (!digitgraph) {
299 digitgraph = new TGraph(fMulDigit,xi,zi);
300 digitgraph-> SetMarkerStyle(5) ;
301 digitgraph-> SetMarkerSize(1.) ;
302 digitgraph-> SetMarkerColor(1) ;
303 digitgraph-> Draw("P") ;
304 }
305 if (!clustertext) {
306
307 TVector3 pos(0.,0.,0.) ;
308 GetLocalPosition(pos) ;
309 clustertext = new TPaveText(pos.X()-10,pos.Z()+10,pos.X()+50,pos.Z()+35,"") ;
310 Text_t line1[40] ;
311 Text_t line2[40] ;
312 sprintf(line1,"Energy=%1.2f GeV",GetEnergy()) ;
313 sprintf(line2,"%d Digits",GetDigitsMultiplicity()) ;
314 clustertext ->AddText(line1) ;
315 clustertext ->AddText(line2) ;
316 clustertext ->Draw("");
317 }
318 gPad->Update() ;
9e5d2067 319 Print("") ;
ab48128d 320 delete[] xi ;
321 delete[] zi ;
322 }
323
324break;
325
326 case kButton1Up:
327 if (digitgraph) {
328 delete digitgraph ;
329 digitgraph = 0 ;
330 }
331 if (clustertext) {
332 delete clustertext ;
333 clustertext = 0 ;
334 }
335
336 break;
337
70a93198 338 }*/
339}
340//____________________________________________________________________________
341void AliEMCALRecPoint::EvalAll(Float_t logWeight,TClonesArray * digits)
342{
343 // Evaluates all shower parameters
344
345 EvalLocalPosition(logWeight, digits) ;
500aeccc 346 // printf("eval position done\n");
70a93198 347 EvalElipsAxis(logWeight, digits) ;
500aeccc 348 // printf("eval axis done\n");
70a93198 349 EvalDispersion(logWeight, digits) ;
500aeccc 350 // printf("eval dispersion done\n");
4635df1f 351 //EvalCoreEnergy(logWeight, digits);
1963b290 352 // printf("eval energy done\n");
70a93198 353 EvalTime(digits) ;
500aeccc 354 // printf("eval time done\n");
70a93198 355
87cdc3be 356 EvalPrimaries(digits) ;
500aeccc 357 // printf("eval pri done\n");
87cdc3be 358 EvalParents(digits);
500aeccc 359 // printf("eval parent done\n");
70a93198 360}
361
362//____________________________________________________________________________
363void AliEMCALRecPoint::EvalDispersion(Float_t logWeight, TClonesArray * digits)
364{
365 // Calculates the dispersion of the shower at the origin of the RecPoint
366
e52475ed 367 Double_t d = 0., wtot = 0., w = 0., xyzi[3], diff=0.;
368 Int_t iDigit=0, nstat=0, i=0;
70a93198 369 AliEMCALDigit * digit ;
5dee926e 370
70a93198 371 // Calculates the centre of gravity in the local EMCAL-module coordinates
e52475ed 372 if (!fLocPos.Mag())
70a93198 373 EvalLocalPosition(logWeight, digits) ;
374
70a93198 375 // Calculates the dispersion in coordinates
70a93198 376 for(iDigit=0; iDigit < fMulDigit; iDigit++) {
377 digit = (AliEMCALDigit *) digits->At(fDigitsList[iDigit]) ;
1963b290 378
14ce0a6e 379 fGeomPtr->RelPosCellInSModule(digit->GetId(), xyzi[0], xyzi[1], xyzi[2]);
e52475ed 380 w = TMath::Max(0.,logWeight+TMath::Log(fEnergyList[iDigit]/fAmp ) ) ;
70a93198 381
e52475ed 382 if(w>0.0) {
383 wtot += w;
384 nstat++;
385 for(i=0; i<3; i++ ) {
386 diff = xyzi[i] - double(fLocPos[i]);
387 d += w * diff*diff;
388 }
389 }
ab48128d 390 }
70a93198 391
e52475ed 392 if ( wtot > 0 && nstat>1) d /= wtot ;
393 else d = 0. ;
70a93198 394
395 fDispersion = TMath::Sqrt(d) ;
ab48128d 396}
70a93198 397
ab48128d 398//____________________________________________________________________________
70a93198 399void AliEMCALRecPoint::EvalLocalPosition(Float_t logWeight, TClonesArray * digits)
88cb7938 400{
70a93198 401 // Calculates the center of gravity in the local EMCAL-module coordinates
e52475ed 402 // Info("Print", " logWeight %f : cluster energy %f ", logWeight, fAmp); // for testing
70a93198 403
e52475ed 404 AliEMCALDigit * digit;
e52475ed 405 Int_t i=0, nstat=0;
406 Double_t clXYZ[3]={0.,0.,0.}, clRmsXYZ[3]={0.,0.,0.}, xyzi[3], wtot=0., w=0.;
70a93198 407
e52475ed 408 for(Int_t iDigit=0; iDigit<fMulDigit; iDigit++) {
70a93198 409 digit = dynamic_cast<AliEMCALDigit *>(digits->At(fDigitsList[iDigit])) ;
410
14ce0a6e 411 fGeomPtr->RelPosCellInSModule(digit->GetId(), xyzi[0], xyzi[1], xyzi[2]);
e52475ed 412 // printf(" Id %i : Local x,y,z %f %f %f \n", digit->GetId(), xyzi[0], xyzi[1], xyzi[2]);
413
414 if(logWeight > 0.0) w = TMath::Max( 0., logWeight + TMath::Log( fEnergyList[iDigit] / fAmp ));
415 else w = fEnergyList[iDigit]; // just energy
ab48128d 416
e52475ed 417 if(w>0.0) {
418 wtot += w ;
419 nstat++;
420 for(i=0; i<3; i++ ) {
421 clXYZ[i] += (w*xyzi[i]);
422 clRmsXYZ[i] += (w*xyzi[i]*xyzi[i]);
423 }
424 }
425 }
426 // cout << " wtot " << wtot << endl;
70a93198 427 if ( wtot > 0 ) {
e52475ed 428 // xRMS = TMath::Sqrt(x2m - xMean*xMean);
429 for(i=0; i<3; i++ ) {
430 clXYZ[i] /= wtot;
431 if(nstat>1) {
432 clRmsXYZ[i] /= (wtot*wtot);
433 clRmsXYZ[i] = clRmsXYZ[i] - clXYZ[i]*clXYZ[i];
434 if(clRmsXYZ[i] > 0.0) {
435 clRmsXYZ[i] = TMath::Sqrt(clRmsXYZ[i]);
436 } else clRmsXYZ[i] = 0;
437 } else clRmsXYZ[i] = 0;
438 }
70a93198 439 } else {
e52475ed 440 for(i=0; i<3; i++ ) {
441 clXYZ[i] = clRmsXYZ[i] = -1.;
442 }
70a93198 443 }
e52475ed 444 // clRmsXYZ[i] ??
445 fLocPos.SetX(clXYZ[0]);
446 fLocPos.SetY(clXYZ[1]);
447 fLocPos.SetZ(clXYZ[2]);
70a93198 448
1963b290 449// if (gDebug==2)
500aeccc 450// printf("EvalLocalPosition: eta,phi,r = %f,%f,%f", fLocPos.X(), fLocPos.Y(), fLocPos.Z()) ;
e52475ed 451 fLocPosM = 0 ; // covariance matrix
ab48128d 452}
453
e52475ed 454//void AliEMCALRecPoint::EvalLocalPositionSimple()
455//{ // Weight is proportional of cell energy
456//}
457
70a93198 458//______________________________________________________________________________
459void AliEMCALRecPoint::EvalCoreEnergy(Float_t logWeight, TClonesArray * digits)
460{
461 // This function calculates energy in the core,
4635df1f 462 // i.e. within a radius rad = fCoreEnergy around the center. Beyond this radius
70a93198 463 // in accordance with shower profile the energy deposition
464 // should be less than 2%
465
466 AliEMCALDigit * digit ;
467 const Float_t kDeg2Rad = TMath::DegToRad() ;
5dee926e 468
70a93198 469 Int_t iDigit;
470
e52475ed 471 if (!fLocPos.Mag()) {
70a93198 472 EvalLocalPosition(logWeight, digits);
473 }
474
475 for(iDigit=0; iDigit < fMulDigit; iDigit++) {
476 digit = (AliEMCALDigit *) ( digits->At(fDigitsList[iDigit]) ) ;
4635df1f 477
478 Float_t ieta = 0.0;
479 Float_t iphi = 0.0;
14ce0a6e 480 //fGeomPtr->PosInAlice(digit->GetId(), ieta, iphi);
481 fGeomPtr->EtaPhiFromIndex(digit->GetId(),ieta, iphi) ;
4635df1f 482 iphi = iphi * kDeg2Rad;
70a93198 483
4635df1f 484 Float_t distance = TMath::Sqrt((ieta-fLocPos.X())*(ieta-fLocPos.X())+(iphi-fLocPos.Y())*(iphi-fLocPos.Y())) ;
70a93198 485 if(distance < fCoreRadius)
486 fCoreEnergy += fEnergyList[iDigit] ;
487 }
488
489}
ab48128d 490//____________________________________________________________________________
70a93198 491void AliEMCALRecPoint::EvalElipsAxis(Float_t logWeight,TClonesArray * digits)
ab48128d 492{
70a93198 493 // Calculates the axis of the shower ellipsoid in eta and phi
ab48128d 494
70a93198 495 Double_t wtot = 0. ;
496 Double_t x = 0.;
497 Double_t z = 0.;
498 Double_t dxx = 0.;
499 Double_t dzz = 0.;
500 Double_t dxz = 0.;
ab48128d 501
ff1e7e2f 502 const Float_t kDeg2Rad = TMath::DegToRad();
70a93198 503 AliEMCALDigit * digit ;
504
14ce0a6e 505 TString gn(fGeomPtr->GetName());
70a93198 506
507 Int_t iDigit;
ab48128d 508
70a93198 509 for(iDigit=0; iDigit<fMulDigit; iDigit++) {
510 digit = (AliEMCALDigit *) digits->At(fDigitsList[iDigit]) ;
511 Float_t etai = 0. ;
512 Float_t phii = 0. ;
e52475ed 513 if(gn.Contains("SHISH")) { // have to be change - Feb 28, 2006
1963b290 514 //copied for shish-kebab geometry, ieta,iphi is cast as float as eta,phi conversion
515 // for this geometry does not exist
516 int nSupMod=0, nTower=0, nIphi=0, nIeta=0;
517 int iphi=0, ieta=0;
14ce0a6e 518 fGeomPtr->GetCellIndex(digit->GetId(), nSupMod,nTower,nIphi,nIeta);
519 fGeomPtr->GetCellPhiEtaIndexInSModule(nSupMod,nTower,nIphi,nIeta, iphi,ieta);
1963b290 520 etai=(Float_t)ieta;
521 phii=(Float_t)iphi;
522 } else {
14ce0a6e 523 fGeomPtr->EtaPhiFromIndex(digit->GetId(), etai, phii);
1963b290 524 phii = phii * kDeg2Rad;
525 }
526
70a93198 527 Double_t w = TMath::Max(0.,logWeight+TMath::Log(fEnergyList[iDigit]/fAmp ) ) ;
ff1e7e2f 528
70a93198 529 dxx += w * etai * etai ;
530 x += w * etai ;
531 dzz += w * phii * phii ;
532 z += w * phii ;
1963b290 533
ff1e7e2f 534 dxz += w * etai * phii ;
1963b290 535
70a93198 536 wtot += w ;
537 }
ff1e7e2f 538
70a93198 539 if ( wtot > 0 ) {
540 dxx /= wtot ;
541 x /= wtot ;
542 dxx -= x * x ;
543 dzz /= wtot ;
544 z /= wtot ;
545 dzz -= z * z ;
546 dxz /= wtot ;
547 dxz -= x * z ;
ab48128d 548
70a93198 549 fLambda[0] = 0.5 * (dxx + dzz) + TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz ) ;
550 if(fLambda[0] > 0)
551 fLambda[0] = TMath::Sqrt(fLambda[0]) ;
552 else
553 fLambda[0] = 0;
554
555 fLambda[1] = 0.5 * (dxx + dzz) - TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz ) ;
ff1e7e2f 556
70a93198 557 if(fLambda[1] > 0) //To avoid exception if numerical errors lead to negative lambda.
558 fLambda[1] = TMath::Sqrt(fLambda[1]) ;
559 else
560 fLambda[1]= 0. ;
561 } else {
562 fLambda[0]= 0. ;
563 fLambda[1]= 0. ;
ab48128d 564 }
ff1e7e2f 565
1963b290 566 // printf("Evalaxis: lambdas = %f,%f", fLambda[0],fLambda[1]) ;
ff1e7e2f 567
ab48128d 568}
569
570//______________________________________________________________________________
571void AliEMCALRecPoint::EvalPrimaries(TClonesArray * digits)
572{
573 // Constructs the list of primary particles (tracks) which have contributed to this RecPoint
574
575 AliEMCALDigit * digit ;
576 Int_t * tempo = new Int_t[fMaxTrack] ;
577
578 Int_t index ;
579 for ( index = 0 ; index < GetDigitsMultiplicity() ; index++ ) { // all digits
580 digit = dynamic_cast<AliEMCALDigit *>(digits->At( fDigitsList[index] )) ;
581 Int_t nprimaries = digit->GetNprimary() ;
5c0368b8 582 if ( nprimaries == 0 ) continue ;
ab48128d 583 Int_t * newprimaryarray = new Int_t[nprimaries] ;
584 Int_t ii ;
585 for ( ii = 0 ; ii < nprimaries ; ii++)
586 newprimaryarray[ii] = digit->GetPrimary(ii+1) ;
587
588 Int_t jndex ;
589 for ( jndex = 0 ; jndex < nprimaries ; jndex++ ) { // all primaries in digit
590 if ( fMulTrack > fMaxTrack ) {
f792c312 591 fMulTrack = fMaxTrack ;
9859bfc0 592 Error("GetNprimaries", "increase fMaxTrack ") ;
ab48128d 593 break ;
594 }
595 Int_t newprimary = newprimaryarray[jndex] ;
596 Int_t kndex ;
597 Bool_t already = kFALSE ;
598 for ( kndex = 0 ; kndex < fMulTrack ; kndex++ ) { //check if not already stored
599 if ( newprimary == tempo[kndex] ){
600 already = kTRUE ;
601 break ;
602 }
603 } // end of check
5c0368b8 604 if ( !already && (fMulTrack < fMaxTrack)) { // store it
ab48128d 605 tempo[fMulTrack] = newprimary ;
606 fMulTrack++ ;
607 } // store it
608 } // all primaries in digit
f792c312 609 delete [] newprimaryarray ;
ab48128d 610 } // all digits
611
612
613 fTracksList = new Int_t[fMulTrack] ;
614 for(index = 0; index < fMulTrack; index++)
615 fTracksList[index] = tempo[index] ;
616
f792c312 617 delete [] tempo ;
ab48128d 618
619}
7ee5c5be 620
87cdc3be 621//______________________________________________________________________________
622void AliEMCALRecPoint::EvalParents(TClonesArray * digits)
623{
624 // Constructs the list of parent particles (tracks) which have contributed to this RecPoint
625
626 AliEMCALDigit * digit ;
627 Int_t * tempo = new Int_t[fMaxParent] ;
628
629 Int_t index ;
630 for ( index = 0 ; index < GetDigitsMultiplicity() ; index++ ) { // all digits
631 digit = dynamic_cast<AliEMCALDigit *>(digits->At( fDigitsList[index] )) ;
632 Int_t nparents = digit->GetNiparent() ;
5c0368b8 633 if ( nparents == 0 ) continue ;
87cdc3be 634 Int_t * newparentarray = new Int_t[nparents] ;
635 Int_t ii ;
636 for ( ii = 0 ; ii < nparents ; ii++)
637 newparentarray[ii] = digit->GetIparent(ii+1) ;
638
639 Int_t jndex ;
640 for ( jndex = 0 ; jndex < nparents ; jndex++ ) { // all primaries in digit
641 if ( fMulParent > fMaxParent ) {
642 fMulTrack = - 1 ;
643 Error("GetNiparent", "increase fMaxParent") ;
644 break ;
645 }
646 Int_t newparent = newparentarray[jndex] ;
647 Int_t kndex ;
648 Bool_t already = kFALSE ;
f1d429fd 649 for ( kndex = 0 ; kndex < fMulParent ; kndex++ ) { //check if not already stored
87cdc3be 650 if ( newparent == tempo[kndex] ){
651 already = kTRUE ;
652 break ;
653 }
654 } // end of check
5c0368b8 655 if ( !already && (fMulTrack < fMaxTrack)) { // store it
87cdc3be 656 tempo[fMulParent] = newparent ;
657 fMulParent++ ;
658 } // store it
659 } // all parents in digit
27e2a47c 660 delete [] newparentarray ;
87cdc3be 661 } // all digits
662
27e2a47c 663 if (fMulParent>0) {
664 fParentsList = new Int_t[fMulParent] ;
665 for(index = 0; index < fMulParent; index++)
666 fParentsList[index] = tempo[index] ;
667 }
87cdc3be 668
27e2a47c 669 delete [] tempo ;
87cdc3be 670
671}
672
ab48128d 673//____________________________________________________________________________
70a93198 674void AliEMCALRecPoint::GetLocalPosition(TVector3 & lpos) const
675{
676 // returns the position of the cluster in the local reference system of ALICE
70a93198 677
678 lpos.SetX(fLocPos.X()) ;
679 lpos.SetY(fLocPos.Y()) ;
680 lpos.SetZ(fLocPos.Z()) ;
681}
682
683//____________________________________________________________________________
ab48128d 684void AliEMCALRecPoint::GetGlobalPosition(TVector3 & gpos) const
685{
686 // returns the position of the cluster in the global reference system of ALICE
70a93198 687 // These are now the Cartesian X, Y and Z
e52475ed 688 // cout<<" geom "<<geom<<endl;
14ce0a6e 689 fGeomPtr->GetGlobal(fLocPos, gpos, fSuperModuleNumber);
70a93198 690}
691
692//____________________________________________________________________________
693Float_t AliEMCALRecPoint::GetMaximalEnergy(void) const
694{
695 // Finds the maximum energy in the cluster
ab48128d 696
70a93198 697 Float_t menergy = 0. ;
698
699 Int_t iDigit;
700
701 for(iDigit=0; iDigit<fMulDigit; iDigit++) {
702
703 if(fEnergyList[iDigit] > menergy)
704 menergy = fEnergyList[iDigit] ;
705 }
706 return menergy ;
ab48128d 707}
708
aad8e277 709//____________________________________________________________________________
70a93198 710Int_t AliEMCALRecPoint::GetMultiplicityAtLevel(Float_t H) const
aad8e277 711{
70a93198 712 // Calculates the multiplicity of digits with energy larger than H*energy
713
714 Int_t multipl = 0 ;
715 Int_t iDigit ;
716 for(iDigit=0; iDigit<fMulDigit; iDigit++) {
717
718 if(fEnergyList[iDigit] > H * fAmp)
719 multipl++ ;
720 }
721 return multipl ;
722}
723
724//____________________________________________________________________________
725Int_t AliEMCALRecPoint::GetNumberOfLocalMax(AliEMCALDigit ** maxAt, Float_t * maxAtEnergy,
726 Float_t locMaxCut,TClonesArray * digits) const
727{
728 // Calculates the number of local maxima in the cluster using fLocalMaxCut as the minimum
729 // energy difference between two local maxima
730
731 AliEMCALDigit * digit ;
732 AliEMCALDigit * digitN ;
733
734 Int_t iDigitN ;
735 Int_t iDigit ;
736
737 for(iDigit = 0; iDigit < fMulDigit; iDigit++)
738 maxAt[iDigit] = (AliEMCALDigit*) digits->At(fDigitsList[iDigit]) ;
739
740 for(iDigit = 0 ; iDigit < fMulDigit; iDigit++) {
741 if(maxAt[iDigit]) {
742 digit = maxAt[iDigit] ;
743
744 for(iDigitN = 0; iDigitN < fMulDigit; iDigitN++) {
745 digitN = (AliEMCALDigit *) digits->At(fDigitsList[iDigitN]) ;
746
747 if ( AreNeighbours(digit, digitN) ) {
748 if (fEnergyList[iDigit] > fEnergyList[iDigitN] ) {
749 maxAt[iDigitN] = 0 ;
750 // but may be digit too is not local max ?
751 if(fEnergyList[iDigit] < fEnergyList[iDigitN] + locMaxCut)
752 maxAt[iDigit] = 0 ;
753 }
754 else {
755 maxAt[iDigit] = 0 ;
756 // but may be digitN too is not local max ?
757 if(fEnergyList[iDigit] > fEnergyList[iDigitN] - locMaxCut)
758 maxAt[iDigitN] = 0 ;
759 }
760 } // if Areneighbours
761 } // while digitN
762 } // slot not empty
763 } // while digit
764
765 iDigitN = 0 ;
766 for(iDigit = 0; iDigit < fMulDigit; iDigit++) {
767 if(maxAt[iDigit] ){
768 maxAt[iDigitN] = maxAt[iDigit] ;
769 maxAtEnergy[iDigitN] = fEnergyList[iDigit] ;
770 iDigitN++ ;
771 }
772 }
773 return iDigitN ;
774}
4635df1f 775
776//____________________________________________________________________________
777Int_t AliEMCALRecPoint::GetPrimaryIndex() const
778{
779 // Get the primary track index in TreeK which deposits the most energy
780 // in Digits which forms RecPoint. Kinematics, Hits and Digits must be
781 // loaded before the call of the method.
782
783 AliRunLoader *rl = AliRunLoader::GetRunLoader();
784 if (!rl)
785 AliError(Form(" No Runloader ")) ;
786
787 AliEMCALLoader *emcalLoader = dynamic_cast<AliEMCALLoader*>
788 (rl->GetDetectorLoader("EMCAL"));
789
790 // Get the list of digits forming this RecPoint
791 Int_t nDigits = fMulDigit ;
792 Int_t *digitList = fDigitsList ;
793
794 // Find the digit with maximum amplitude
795 AliEMCALDigit *digit = 0;
796 TClonesArray *digits = emcalLoader->Digits();
797 Int_t maxAmp = 0;
798 Int_t bestDigitIndex = -1;
799 for (Int_t iDigit=0; iDigit<nDigits; iDigit++) {
800 digit = static_cast<AliEMCALDigit *>(digits->At(digitList[iDigit]));
801 if (digit->GetAmp() > maxAmp) {
802 maxAmp = digit->GetAmp();
803 bestDigitIndex = iDigit;
804 }
805 }
806
807 digit = static_cast<AliEMCALDigit *>(digits->At(digitList[bestDigitIndex]));
808
809 // Get the list of hits producing this digit,
810 // find which hit has deposited more energy
811 // and find the primary track.
812
813 AliEMCALHit *hit = 0;
814 TClonesArray *hits = emcalLoader->Hits();
815
816 Double_t maxedep = 0;
817 Int_t maxtrack = -1;
818 Int_t nHits = hits ->GetEntries();
819 Int_t id = digit->GetId();
820 for (Int_t iHit=0; iHit<nHits; iHit++) {
821 hit = static_cast<AliEMCALHit*> (hits->At(iHit)) ;
822 if(hit->GetId() == id){
823 Double_t edep = hit->GetEnergy();
824 Int_t track = hit->GetIparent();//Primary();
825 if(edep > maxedep){
826 maxedep = edep;
827 maxtrack = track;
828 }
829 }
830 }
831 if (maxtrack != -1) return maxtrack;
832 return -12345; // no track found :(
833}
834
70a93198 835//____________________________________________________________________________
836void AliEMCALRecPoint::EvalTime(TClonesArray * digits){
837 // time is set to the time of the digit with the maximum energy
838
839 Float_t maxE = 0;
840 Int_t maxAt = 0;
841 for(Int_t idig=0; idig < fMulDigit; idig++){
842 if(fEnergyList[idig] > maxE){
843 maxE = fEnergyList[idig] ;
844 maxAt = idig;
845 }
846 }
847 fTime = ((AliEMCALDigit*) digits->At(fDigitsList[maxAt]))->GetTime() ;
aad8e277 848
aad8e277 849}
ab48128d 850
851//______________________________________________________________________________
852void AliEMCALRecPoint::Paint(Option_t *)
853{
854 // Paint this ALiRecPoint as a TMarker with its current attributes
855
856 TVector3 pos(0.,0.,0.) ;
857 GetLocalPosition(pos) ;
858 Coord_t x = pos.X() ;
859 Coord_t y = pos.Z() ;
860 Color_t markercolor = 1 ;
861 Size_t markersize = 1. ;
862 Style_t markerstyle = 5 ;
863
864 if (!gPad->IsBatch()) {
865 gVirtualX->SetMarkerColor(markercolor) ;
866 gVirtualX->SetMarkerSize (markersize) ;
867 gVirtualX->SetMarkerStyle(markerstyle) ;
868 }
869 gPad->SetAttMarkerPS(markercolor,markerstyle,markersize) ;
870 gPad->PaintPolyMarker(1,&x,&y,"") ;
871}
70a93198 872
873//______________________________________________________________________________
874Float_t AliEMCALRecPoint::EtaToTheta(Float_t arg) const
875{
876 //Converts Theta (Radians) to Eta(Radians)
877 return (2.*TMath::ATan(TMath::Exp(-arg)));
878}
879
880//______________________________________________________________________________
881Float_t AliEMCALRecPoint::ThetaToEta(Float_t arg) const
882{
883 //Converts Eta (Radians) to Theta(Radians)
884 return (-1 * TMath::Log(TMath::Tan(0.5 * arg)));
885}
261b1065 886
887//____________________________________________________________________________
888void AliEMCALRecPoint::Print(Option_t *) const
889{
890 // Print the list of digits belonging to the cluster
e52475ed 891 return;
261b1065 892 TString message ;
4800667c 893 message = "AliEMCALRecPoint:\n" ;
261b1065 894 message += " digits # = " ;
895 Info("Print", message.Data()) ;
896
897 Int_t iDigit;
898 for(iDigit=0; iDigit<fMulDigit; iDigit++)
899 printf(" %d ", fDigitsList[iDigit] ) ;
e52475ed 900 printf("\n");
901
261b1065 902 Info("Print", " Energies = ") ;
903 for(iDigit=0; iDigit<fMulDigit; iDigit++)
904 printf(" %f ", fEnergyList[iDigit] ) ;
e52475ed 905 printf("\n");
906
907 Info("Print", "\n Abs Ids = ") ;
908 for(iDigit=0; iDigit<fMulDigit; iDigit++)
909 printf(" %i ", fAbsIdList[iDigit] ) ;
910 printf("\n");
911
912 Info("Print", " Primaries ") ;
261b1065 913 for(iDigit = 0;iDigit < fMulTrack; iDigit++)
914 printf(" %d ", fTracksList[iDigit]) ;
e52475ed 915
916 printf("\n Local x %6.2f y %7.2f z %7.1f \n", fLocPos[0], fLocPos[1], fLocPos[2]);
917
85c60a8e 918 message = " ClusterType = %d" ;
919 message += " Multiplicity = %d" ;
261b1065 920 message += " Cluster Energy = %f" ;
921 message += " Core energy = %f" ;
922 message += " Core radius = %f" ;
923 message += " Number of primaries %d" ;
924 message += " Stored at position %d" ;
85c60a8e 925 Info("Print", message.Data(), fClusterType, fMulDigit, fAmp, fCoreEnergy, fCoreRadius, fMulTrack, GetIndexInList() ) ;
261b1065 926}