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