1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
16 //_________________________________________________________________________
17 // Reconstructed Points for the EMCAL
18 // A RecPoint is a cluster of digits
19 //*-- Author: Yves Schutz (SUBATECH)
20 //*-- Author: Dmitri Peressounko (RRC KI & SUBATECH)
21 //*-- Author: Heather Gray (LBL) merged AliEMCALRecPoint and AliEMCALTowerRecPoint 02/04
23 // --- ROOT system ---
26 #include "TPaveText.h"
27 #include "TClonesArray.h"
30 // --- Standard library ---
32 // --- AliRoot header files ---
33 #include "AliGenerator.h"
34 #include "AliEMCALGeometry.h"
35 #include "AliEMCALDigit.h"
36 #include "AliEMCALRecPoint.h"
37 #include "AliEMCALGetter.h"
39 ClassImp(AliEMCALRecPoint)
42 //____________________________________________________________________________
43 AliEMCALRecPoint::AliEMCALRecPoint()
56 fLocPos.SetX(0.) ; //Local position should be evaluated
57 fCoreRadius = 10; //HG Check this
60 //____________________________________________________________________________
61 AliEMCALRecPoint::AliEMCALRecPoint(const char * opt) : AliRecPoint(opt)
71 fParentsList = new Int_t[fMaxParent];
73 fLocPos.SetX(1000000.) ; //Local position should be evaluated
74 fCoreRadius = 10; //HG Check this
76 //____________________________________________________________________________
77 AliEMCALRecPoint::~AliEMCALRecPoint()
81 delete[] fEnergyList ;
83 delete[] fParentsList;
86 //____________________________________________________________________________
87 void AliEMCALRecPoint::AddDigit(AliEMCALDigit & digit, Float_t Energy)
89 // Adds a digit to the RecPoint
90 // and accumulates the total amplitude and the multiplicity
93 fEnergyList = new Float_t[fMaxDigit];
95 if ( fMulDigit >= fMaxDigit ) { // increase the size of the lists
97 Int_t * tempo = new Int_t[fMaxDigit];
98 Float_t * tempoE = new Float_t[fMaxDigit];
101 for ( index = 0 ; index < fMulDigit ; index++ ){
102 tempo[index] = fDigitsList[index] ;
103 tempoE[index] = fEnergyList[index] ;
106 delete [] fDigitsList ;
107 fDigitsList = new Int_t[fMaxDigit];
109 delete [] fEnergyList ;
110 fEnergyList = new Float_t[fMaxDigit];
112 for ( index = 0 ; index < fMulDigit ; index++ ){
113 fDigitsList[index] = tempo[index] ;
114 fEnergyList[index] = tempoE[index] ;
121 fDigitsList[fMulDigit] = digit.GetIndexInList() ;
122 fEnergyList[fMulDigit] = Energy ;
127 //____________________________________________________________________________
128 Bool_t AliEMCALRecPoint::AreNeighbours(AliEMCALDigit * digit1, AliEMCALDigit * digit2 ) const
130 // Tells if (true) or not (false) two digits are neighbours
131 // A neighbour is defined as being two digits which share a corner
133 Bool_t areNeighbours = kFALSE ;
135 AliEMCALGeometry * geom = (AliEMCALGetter::Instance())->EMCALGeometry();
138 geom->AbsToRelNumbering(digit1->GetId(), relid1) ;
141 geom->AbsToRelNumbering(digit2->GetId(), relid2) ;
143 Int_t rowdiff = TMath::Abs( relid1[0] - relid2[0] ) ;
144 Int_t coldiff = TMath::Abs( relid1[1] - relid2[1] ) ;
146 if (( coldiff <= 1 ) && ( rowdiff <= 1 ) && (coldiff + rowdiff > 0))
147 areNeighbours = kTRUE ;
149 return areNeighbours;
152 //____________________________________________________________________________
153 Int_t AliEMCALRecPoint::Compare(const TObject * obj) const
155 // Compares two RecPoints according to their position in the EMCAL modules
157 Float_t delta = 1 ; //Width of "Sorting row". If you change this
158 //value (what is senseless) change as well delta in
159 //AliEMCALTrackSegmentMakerv* and other RecPoints...
162 AliEMCALRecPoint * clu = (AliEMCALRecPoint *)obj ;
165 GetLocalPosition(locpos1);
167 clu->GetLocalPosition(locpos2);
169 Int_t rowdif = (Int_t)TMath::Ceil(locpos1.X()/delta)-(Int_t)TMath::Ceil(locpos2.X()/delta) ;
174 else if(locpos1.Y()>locpos2.Y())
182 //____________________________________________________________________________
183 Int_t AliEMCALRecPoint::DistancetoPrimitive(Int_t px, Int_t py)
185 // Compute distance from point px,py to a AliEMCALRecPoint considered as a Tmarker
186 // Compute the closest distance of approach from point px,py to this marker.
187 // The distance is computed in pixels units.
188 // HG Still need to update -> Not sure what this should achieve
190 TVector3 pos(0.,0.,0.) ;
191 GetLocalPosition(pos) ;
192 Float_t x = pos.X() ;
193 Float_t y = pos.Y() ;
194 const Int_t kMaxDiff = 10;
195 Int_t pxm = gPad->XtoAbsPixel(x);
196 Int_t pym = gPad->YtoAbsPixel(y);
197 Int_t dist = (px-pxm)*(px-pxm) + (py-pym)*(py-pym);
199 if (dist > kMaxDiff) return 9999;
203 //___________________________________________________________________________
204 void AliEMCALRecPoint::Draw(Option_t *option)
206 // Draw this AliEMCALRecPoint with its current attributes
211 //______________________________________________________________________________
212 void AliEMCALRecPoint::ExecuteEvent(Int_t /*event*/, Int_t, Int_t)
214 // Execute action corresponding to one event
215 // This member function is called when a AliEMCALRecPoint is clicked with the locator
217 // If Left button is clicked on AliEMCALRecPoint, the digits are switched on
218 // and switched off when the mouse button is released.
220 // static Int_t pxold, pyold;
222 /* static TGraph * digitgraph = 0 ;
223 static TPaveText* clustertext = 0 ;
225 if (!gPad->IsEditable()) return;
231 AliEMCALDigit * digit ;
232 AliEMCALGeometry * emcalgeom = (AliEMCALGetter::Instance())->EMCALGeometry() ;
237 const Int_t kMulDigit=AliEMCALRecPoint::GetDigitsMultiplicity() ;
238 Float_t * xi = new Float_t [kMulDigit] ;
239 Float_t * zi = new Float_t [kMulDigit] ;
241 for(iDigit = 0; iDigit < kMulDigit; iDigit++) {
242 Fatal("AliEMCALRecPoint::ExecuteEvent", " -> Something wrong with the code");
243 digit = 0 ; //dynamic_cast<AliEMCALDigit *>((fDigitsList)[iDigit]);
244 emcalgeom->AbsToRelNumbering(digit->GetId(), relid) ;
245 emcalgeom->PosInAlice(relid, xi[iDigit], zi[iDigit]) ;
249 digitgraph = new TGraph(fMulDigit,xi,zi);
250 digitgraph-> SetMarkerStyle(5) ;
251 digitgraph-> SetMarkerSize(1.) ;
252 digitgraph-> SetMarkerColor(1) ;
253 digitgraph-> Draw("P") ;
257 TVector3 pos(0.,0.,0.) ;
258 GetLocalPosition(pos) ;
259 clustertext = new TPaveText(pos.X()-10,pos.Z()+10,pos.X()+50,pos.Z()+35,"") ;
262 sprintf(line1,"Energy=%1.2f GeV",GetEnergy()) ;
263 sprintf(line2,"%d Digits",GetDigitsMultiplicity()) ;
264 clustertext ->AddText(line1) ;
265 clustertext ->AddText(line2) ;
266 clustertext ->Draw("");
290 //____________________________________________________________________________
291 void AliEMCALRecPoint::EvalAll(Float_t logWeight,TClonesArray * digits)
293 // Evaluates all shower parameters
295 EvalLocalPosition(logWeight, digits) ;
296 EvalElipsAxis(logWeight, digits) ;
297 EvalDispersion(logWeight, digits) ;
298 EvalCoreEnergy(logWeight, digits);
301 EvalPrimaries(digits) ;
305 //____________________________________________________________________________
306 void AliEMCALRecPoint::EvalDispersion(Float_t logWeight, TClonesArray * digits)
308 // Calculates the dispersion of the shower at the origin of the RecPoint
313 AliEMCALDigit * digit ;
315 AliEMCALGeometry * geom = (AliEMCALGetter::Instance())->EMCALGeometry();
317 // Calculates the centre of gravity in the local EMCAL-module coordinates
320 if (!fLocPos.X() || !fLocPos.Y())
321 EvalLocalPosition(logWeight, digits) ;
323 const Float_t kDeg2Rad = TMath::DegToRad() ;
325 Float_t cluEta = fLocPos.X() ;
326 Float_t cluPhi = fLocPos.Y() ;
327 Float_t cluR = fLocPos.Z() ;
330 printf("EvalDispersion: eta,phi,r = %f,%f,%f", cluEta, cluPhi, cluR) ;
332 // Calculates the dispersion in coordinates
334 for(iDigit=0; iDigit < fMulDigit; iDigit++) {
335 digit = (AliEMCALDigit *) digits->At(fDigitsList[iDigit]) ;
338 geom->EtaPhiFromIndex(digit->GetId(), etai, phii);
339 phii = phii * kDeg2Rad;
341 printf("EvalDispersion: id = %d, etai,phii = %f,%f", digit->GetId(), etai, phii) ;
343 Float_t w = TMath::Max(0.,logWeight+TMath::Log(fEnergyList[iDigit]/fAmp ) ) ;
344 d += w * ( (etai-cluEta)*(etai-cluEta) + (phii-cluPhi)*(phii-cluPhi) ) ;
353 fDispersion = TMath::Sqrt(d) ;
357 //____________________________________________________________________________
358 void AliEMCALRecPoint::EvalLocalPosition(Float_t logWeight, TClonesArray * digits)
360 // Calculates the center of gravity in the local EMCAL-module coordinates
365 AliEMCALDigit * digit ;
366 AliEMCALGeometry * geom = (AliEMCALGetter::Instance())->EMCALGeometry();
370 const Float_t kDeg2Rad = TMath::DegToRad();
372 for(iDigit=0; iDigit<fMulDigit; iDigit++) {
373 digit = dynamic_cast<AliEMCALDigit *>(digits->At(fDigitsList[iDigit])) ;
377 geom->EtaPhiFromIndex(digit->GetId(), etai, phii);
378 phii = phii * kDeg2Rad;
379 Float_t w = TMath::Max( 0., logWeight + TMath::Log( fEnergyList[iDigit] / fAmp ) ) ;
380 cluEta += (etai * w) ;
381 cluPhi += (phii * w );
393 fLocPos.SetX(cluEta);
394 fLocPos.SetY(cluPhi);
395 fLocPos.SetZ(geom->GetIP2ECASection());
398 printf("EvalLocalPosition: eta,phi,r = %f,%f,%f", fLocPos.X(), fLocPos.Y(), fLocPos.Z()) ;
402 //______________________________________________________________________________
403 void AliEMCALRecPoint::EvalCoreEnergy(Float_t logWeight, TClonesArray * digits)
405 // This function calculates energy in the core,
406 // i.e. within a radius rad = 3cm around the center. Beyond this radius
407 // in accordance with shower profile the energy deposition
408 // should be less than 2%
410 AliEMCALDigit * digit ;
411 const Float_t kDeg2Rad = TMath::DegToRad() ;
412 AliEMCALGeometry * geom = (AliEMCALGetter::Instance())->EMCALGeometry();
415 if (!fLocPos.X() || !fLocPos.Y() ) {
416 EvalLocalPosition(logWeight, digits);
419 for(iDigit=0; iDigit < fMulDigit; iDigit++) {
420 digit = (AliEMCALDigit *) ( digits->At(fDigitsList[iDigit]) ) ;
423 geom->PosInAlice(digit->GetId(), etai, phii);
424 phii = phii * kDeg2Rad;
426 Float_t distance = TMath::Sqrt((etai-fLocPos.X())*(etai-fLocPos.X())+(phii-fLocPos.Y())*(phii-fLocPos.Y())) ;
427 if(distance < fCoreRadius)
428 fCoreEnergy += fEnergyList[iDigit] ;
432 //____________________________________________________________________________
433 void AliEMCALRecPoint::EvalElipsAxis(Float_t logWeight,TClonesArray * digits)
435 // Calculates the axis of the shower ellipsoid in eta and phi
444 AliEMCALDigit * digit ;
446 AliEMCALGeometry * geom = (AliEMCALGetter::Instance())->EMCALGeometry();
450 for(iDigit=0; iDigit<fMulDigit; iDigit++) {
451 digit = (AliEMCALDigit *) digits->At(fDigitsList[iDigit]) ;
454 geom->EtaPhiFromIndex(digit->GetId(), etai, phii);
455 Double_t w = TMath::Max(0.,logWeight+TMath::Log(fEnergyList[iDigit]/fAmp ) ) ;
456 dxx += w * etai * etai ;
458 dzz += w * phii * phii ;
460 dxz += w * etai * etai ;
473 fLambda[0] = 0.5 * (dxx + dzz) + TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz ) ;
475 fLambda[0] = TMath::Sqrt(fLambda[0]) ;
479 fLambda[1] = 0.5 * (dxx + dzz) - TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz ) ;
480 if(fLambda[1] > 0) //To avoid exception if numerical errors lead to negative lambda.
481 fLambda[1] = TMath::Sqrt(fLambda[1]) ;
490 //______________________________________________________________________________
491 void AliEMCALRecPoint::EvalPrimaries(TClonesArray * digits)
493 // Constructs the list of primary particles (tracks) which have contributed to this RecPoint
495 AliEMCALDigit * digit ;
496 Int_t * tempo = new Int_t[fMaxTrack] ;
499 for ( index = 0 ; index < GetDigitsMultiplicity() ; index++ ) { // all digits
500 digit = dynamic_cast<AliEMCALDigit *>(digits->At( fDigitsList[index] )) ;
501 Int_t nprimaries = digit->GetNprimary() ;
502 Int_t * newprimaryarray = new Int_t[nprimaries] ;
504 for ( ii = 0 ; ii < nprimaries ; ii++)
505 newprimaryarray[ii] = digit->GetPrimary(ii+1) ;
508 for ( jndex = 0 ; jndex < nprimaries ; jndex++ ) { // all primaries in digit
509 if ( fMulTrack > fMaxTrack ) {
510 fMulTrack = fMaxTrack ;
511 Error("GetNprimaries", "increase fMaxTrack ") ;
514 Int_t newprimary = newprimaryarray[jndex] ;
516 Bool_t already = kFALSE ;
517 for ( kndex = 0 ; kndex < fMulTrack ; kndex++ ) { //check if not already stored
518 if ( newprimary == tempo[kndex] ){
523 if ( !already) { // store it
524 tempo[fMulTrack] = newprimary ;
527 } // all primaries in digit
528 delete [] newprimaryarray ;
532 fTracksList = new Int_t[fMulTrack] ;
533 for(index = 0; index < fMulTrack; index++)
534 fTracksList[index] = tempo[index] ;
540 //______________________________________________________________________________
541 void AliEMCALRecPoint::EvalParents(TClonesArray * digits)
543 // Constructs the list of parent particles (tracks) which have contributed to this RecPoint
545 AliEMCALDigit * digit ;
546 Int_t * tempo = new Int_t[fMaxParent] ;
549 for ( index = 0 ; index < GetDigitsMultiplicity() ; index++ ) { // all digits
550 digit = dynamic_cast<AliEMCALDigit *>(digits->At( fDigitsList[index] )) ;
551 Int_t nparents = digit->GetNiparent() ;
552 Int_t * newparentarray = new Int_t[nparents] ;
554 for ( ii = 0 ; ii < nparents ; ii++)
555 newparentarray[ii] = digit->GetIparent(ii+1) ;
558 for ( jndex = 0 ; jndex < nparents ; jndex++ ) { // all primaries in digit
559 if ( fMulParent > fMaxParent ) {
561 Error("GetNiparent", "increase fMaxParent") ;
564 Int_t newparent = newparentarray[jndex] ;
566 Bool_t already = kFALSE ;
567 for ( kndex = 0 ; kndex < fMulParent ; kndex++ ) { //check if not already stored
568 if ( newparent == tempo[kndex] ){
573 if ( !already) { // store it
574 tempo[fMulParent] = newparent ;
577 } // all parents in digit
578 delete newparentarray ;
582 fParentsList = new Int_t[fMulParent] ;
583 for(index = 0; index < fMulParent; index++)
584 fParentsList[index] = tempo[index] ;
590 //____________________________________________________________________________
591 void AliEMCALRecPoint::GetLocalPosition(TVector3 & lpos) const
593 // returns the position of the cluster in the local reference system of ALICE
594 // X = eta, Y = phi, Z = r (a constant for the EMCAL)
596 lpos.SetX(fLocPos.X()) ;
597 lpos.SetY(fLocPos.Y()) ;
598 lpos.SetZ(fLocPos.Z()) ;
601 //____________________________________________________________________________
602 void AliEMCALRecPoint::GetGlobalPosition(TVector3 & gpos) const
604 // returns the position of the cluster in the global reference system of ALICE
605 // These are now the Cartesian X, Y and Z
607 AliEMCALGeometry * geom = (AliEMCALGetter::Instance())->EMCALGeometry();
608 Int_t absid = geom->TowerIndexFromEtaPhi(fLocPos.X(), TMath::RadToDeg()*fLocPos.Y());
609 geom->XYZFromIndex(absid, gpos);
612 //____________________________________________________________________________
613 Float_t AliEMCALRecPoint::GetMaximalEnergy(void) const
615 // Finds the maximum energy in the cluster
617 Float_t menergy = 0. ;
621 for(iDigit=0; iDigit<fMulDigit; iDigit++) {
623 if(fEnergyList[iDigit] > menergy)
624 menergy = fEnergyList[iDigit] ;
629 //____________________________________________________________________________
630 Int_t AliEMCALRecPoint::GetMultiplicityAtLevel(Float_t H) const
632 // Calculates the multiplicity of digits with energy larger than H*energy
636 for(iDigit=0; iDigit<fMulDigit; iDigit++) {
638 if(fEnergyList[iDigit] > H * fAmp)
644 //____________________________________________________________________________
645 Int_t AliEMCALRecPoint::GetNumberOfLocalMax(AliEMCALDigit ** maxAt, Float_t * maxAtEnergy,
646 Float_t locMaxCut,TClonesArray * digits) const
648 // Calculates the number of local maxima in the cluster using fLocalMaxCut as the minimum
649 // energy difference between two local maxima
651 AliEMCALDigit * digit ;
652 AliEMCALDigit * digitN ;
657 for(iDigit = 0; iDigit < fMulDigit; iDigit++)
658 maxAt[iDigit] = (AliEMCALDigit*) digits->At(fDigitsList[iDigit]) ;
660 for(iDigit = 0 ; iDigit < fMulDigit; iDigit++) {
662 digit = maxAt[iDigit] ;
664 for(iDigitN = 0; iDigitN < fMulDigit; iDigitN++) {
665 digitN = (AliEMCALDigit *) digits->At(fDigitsList[iDigitN]) ;
667 if ( AreNeighbours(digit, digitN) ) {
668 if (fEnergyList[iDigit] > fEnergyList[iDigitN] ) {
670 // but may be digit too is not local max ?
671 if(fEnergyList[iDigit] < fEnergyList[iDigitN] + locMaxCut)
676 // but may be digitN too is not local max ?
677 if(fEnergyList[iDigit] > fEnergyList[iDigitN] - locMaxCut)
680 } // if Areneighbours
686 for(iDigit = 0; iDigit < fMulDigit; iDigit++) {
688 maxAt[iDigitN] = maxAt[iDigit] ;
689 maxAtEnergy[iDigitN] = fEnergyList[iDigit] ;
695 //____________________________________________________________________________
696 void AliEMCALRecPoint::EvalTime(TClonesArray * digits){
697 // time is set to the time of the digit with the maximum energy
701 for(Int_t idig=0; idig < fMulDigit; idig++){
702 if(fEnergyList[idig] > maxE){
703 maxE = fEnergyList[idig] ;
707 fTime = ((AliEMCALDigit*) digits->At(fDigitsList[maxAt]))->GetTime() ;
711 //______________________________________________________________________________
712 void AliEMCALRecPoint::Paint(Option_t *)
714 // Paint this ALiRecPoint as a TMarker with its current attributes
716 TVector3 pos(0.,0.,0.) ;
717 GetLocalPosition(pos) ;
718 Coord_t x = pos.X() ;
719 Coord_t y = pos.Z() ;
720 Color_t markercolor = 1 ;
721 Size_t markersize = 1. ;
722 Style_t markerstyle = 5 ;
724 if (!gPad->IsBatch()) {
725 gVirtualX->SetMarkerColor(markercolor) ;
726 gVirtualX->SetMarkerSize (markersize) ;
727 gVirtualX->SetMarkerStyle(markerstyle) ;
729 gPad->SetAttMarkerPS(markercolor,markerstyle,markersize) ;
730 gPad->PaintPolyMarker(1,&x,&y,"") ;
733 //______________________________________________________________________________
734 Float_t AliEMCALRecPoint::EtaToTheta(Float_t arg) const
736 //Converts Theta (Radians) to Eta(Radians)
737 return (2.*TMath::ATan(TMath::Exp(-arg)));
740 //______________________________________________________________________________
741 Float_t AliEMCALRecPoint::ThetaToEta(Float_t arg) const
743 //Converts Eta (Radians) to Theta(Radians)
744 return (-1 * TMath::Log(TMath::Tan(0.5 * arg)));
747 //____________________________________________________________________________
748 void AliEMCALRecPoint::Print(Option_t *) const
750 // Print the list of digits belonging to the cluster
753 message = "AliPHOSEmcRecPoint:\n" ;
754 message += " digits # = " ;
755 Info("Print", message.Data()) ;
758 for(iDigit=0; iDigit<fMulDigit; iDigit++)
759 printf(" %d ", fDigitsList[iDigit] ) ;
761 Info("Print", " Energies = ") ;
762 for(iDigit=0; iDigit<fMulDigit; iDigit++)
763 printf(" %f ", fEnergyList[iDigit] ) ;
765 Info("Print", " Primaries ") ;
766 for(iDigit = 0;iDigit < fMulTrack; iDigit++)
767 printf(" %d ", fTracksList[iDigit]) ;
769 message = " Multiplicity = %d" ;
770 message += " Cluster Energy = %f" ;
771 message += " Core energy = %f" ;
772 message += " Core radius = %f" ;
773 message += " Number of primaries %d" ;
774 message += " Stored at position %d" ;
776 Info("Print", message.Data(), fMulDigit, fAmp, fCoreEnergy, fCoreRadius, fMulTrack, GetIndexInList() ) ;