]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PHOS/AliPHOSEmcRecPoint.cxx
Dimitri just makes it work
[u/mrichter/AliRoot.git] / PHOS / AliPHOSEmcRecPoint.cxx
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 //  RecPoint implementation for PHOS-EMC 
20 //  An EmcRecPoint is a cluster of digits   
21 //           
22 //*-- Author: Dmitri Peressounko (RRC KI & SUBATECH)
23
24
25 // --- ROOT system ---
26 #include "TPad.h"
27 #include "TH2.h"
28 #include "TMath.h" 
29 #include "TCanvas.h" 
30
31 // --- Standard library ---
32
33 #include <iostream.h> 
34
35 // --- AliRoot header files ---
36
37 #include "AliPHOSGeometry.h"
38 #include "AliPHOSEmcRecPoint.h"
39 #include "AliRun.h"
40 #include "AliPHOSIndexToObject.h"
41
42 ClassImp(AliPHOSEmcRecPoint)
43
44 //____________________________________________________________________________
45 AliPHOSEmcRecPoint::AliPHOSEmcRecPoint(Float_t W0, Float_t LocMaxCut)
46   : AliPHOSRecPoint()
47 {
48   // ctor
49
50   fMulDigit   = 0 ;  
51   fAmp   = 0. ;   
52   fEnergyList = new Float_t[fMaxDigit]; 
53   AliPHOSGeometry * phosgeom =  (AliPHOSGeometry *) fGeom ;
54   fDelta     =  phosgeom->GetCrystalSize(0) ; 
55   fW0        = W0 ;          
56   fLocMaxCut = LocMaxCut ; 
57   fLocPos.SetX(1000000.)  ;      //Local position should be evaluated
58   
59 }
60
61 //____________________________________________________________________________
62 AliPHOSEmcRecPoint::~AliPHOSEmcRecPoint()
63 {
64   // dtor
65   if ( fEnergyList )
66     delete[] fEnergyList ; 
67 }
68
69 //____________________________________________________________________________
70 void AliPHOSEmcRecPoint::AddDigit(AliPHOSDigit & digit, Float_t Energy)
71 {
72   // Adds a digit to the RecPoint
73   //  and accumulates the total amplitude and the multiplicity 
74   
75   if ( fMulDigit >= fMaxDigit ) { // increase the size of the lists 
76     fMaxDigit*=2 ; 
77     Int_t * tempo = new ( Int_t[fMaxDigit] ) ; 
78     Float_t * tempoE =  new ( Float_t[fMaxDigit] ) ;
79
80     Int_t index ;     
81     for ( index = 0 ; index < fMulDigit ; index++ ){
82       tempo[index]  = fDigitsList[index] ;
83       tempoE[index] = fEnergyList[index] ; 
84     }
85     
86     delete [] fDigitsList ; 
87     fDigitsList =  new ( Int_t[fMaxDigit] ) ;
88  
89     delete [] fEnergyList ;
90     fEnergyList =  new ( Float_t[fMaxDigit] ) ;
91
92     for ( index = 0 ; index < fMulDigit ; index++ ){
93       fDigitsList[index] = tempo[index] ;
94       fEnergyList[index] = tempoE[index] ; 
95     }
96  
97     delete [] tempo ;
98     delete [] tempoE ; 
99   } // if
100   
101   fDigitsList[fMulDigit]   = digit.GetIndexInList()  ; 
102   fEnergyList[fMulDigit]   = Energy ;
103   fMulDigit++ ; 
104   fAmp += Energy ; 
105 }
106
107 //____________________________________________________________________________
108 Bool_t AliPHOSEmcRecPoint::AreNeighbours(AliPHOSDigit * digit1, AliPHOSDigit * digit2 ) 
109 {
110   // Tells if (true) or not (false) two digits are neighbors)
111   
112   Bool_t aren = kFALSE ;
113   
114   AliPHOSGeometry * phosgeom =  (AliPHOSGeometry *) fGeom ;
115   Int_t relid1[4] ; 
116   phosgeom->AbsToRelNumbering(digit1->GetId(), relid1) ; 
117
118   Int_t relid2[4] ; 
119   phosgeom->AbsToRelNumbering(digit2->GetId(), relid2) ; 
120   
121   Int_t rowdiff = TMath::Abs( relid1[2] - relid2[2] ) ;  
122   Int_t coldiff = TMath::Abs( relid1[3] - relid2[3] ) ;  
123
124   if (( coldiff <= 1 )  && ( rowdiff <= 1 ) && (coldiff + rowdiff > 0)) 
125     aren = kTRUE ;
126   
127   return aren ;
128 }
129
130 //____________________________________________________________________________
131 Int_t AliPHOSEmcRecPoint::Compare(TObject * obj)
132 {
133   // Compares two RecPoints according to their position in the PHOS modules
134
135   Int_t rv ; 
136
137   AliPHOSEmcRecPoint * clu = (AliPHOSEmcRecPoint *)obj ; 
138
139  
140   Int_t phosmod1 = this->GetPHOSMod() ;
141   Int_t phosmod2 = clu->GetPHOSMod() ;
142
143   TVector3 locpos1; 
144   this->GetLocalPosition(locpos1) ;
145   TVector3 locpos2;  
146   clu->GetLocalPosition(locpos2) ;  
147
148   if(phosmod1 == phosmod2 ) {
149     Int_t rowdif = (Int_t)TMath::Ceil(locpos1.X()/fDelta)-(Int_t)TMath::Ceil(locpos2.X()/fDelta) ;
150     if (rowdif> 0) 
151       rv = -1 ;
152      else if(rowdif < 0) 
153        rv = 1 ;
154     else if(locpos1.Z()>locpos2.Z()) 
155       rv = -1 ;
156     else 
157       rv = 1 ; 
158      }
159
160   else {
161     if(phosmod1 < phosmod2 ) 
162       rv = -1 ;
163     else 
164       rv = 1 ;
165   }
166
167   return rv ; 
168 }
169
170 //______________________________________________________________________________
171 void AliPHOSEmcRecPoint::ExecuteEvent(Int_t event, Int_t px, Int_t py)
172 {
173   // Execute action corresponding to one event
174   //  This member function is called when a AliPHOSRecPoint is clicked with the locator
175   //
176   //  If Left button is clicked on AliPHOSRecPoint, the digits are switched on    
177   //  and switched off when the mouse button is released.
178   //
179
180   //   static Int_t pxold, pyold;
181
182   AliPHOSIndexToObject * please =  AliPHOSIndexToObject::GetInstance() ; 
183   
184   static TGraph *  digitgraph = 0 ;
185   
186   if (!gPad->IsEditable()) return;
187   
188   TH2F * histo = 0 ;
189   TCanvas * histocanvas ; 
190   
191   switch (event) {
192     
193   case kButton1Down: {
194     AliPHOSDigit * digit ;
195     AliPHOSGeometry * phosgeom =  (AliPHOSGeometry *) fGeom ;
196     Int_t iDigit;
197     Int_t relid[4] ;
198     
199     const Int_t kMulDigit = AliPHOSEmcRecPoint::GetDigitsMultiplicity() ; 
200     Float_t * xi = new Float_t[kMulDigit] ; 
201     Float_t * zi = new Float_t[kMulDigit] ; 
202     
203     // create the histogram for the single cluster 
204     // 1. gets histogram boundaries
205     Float_t ximax = -999. ; 
206     Float_t zimax = -999. ; 
207     Float_t ximin = 999. ; 
208     Float_t zimin = 999. ;
209     
210     for(iDigit=0; iDigit<kMulDigit; iDigit++) {
211       digit = (AliPHOSDigit *) ( please->GimeDigit(fDigitsList[iDigit]) ) ;
212       phosgeom->AbsToRelNumbering(digit->GetId(), relid) ;
213       phosgeom->RelPosInModule(relid, xi[iDigit], zi[iDigit]);
214       if ( xi[iDigit] > ximax )
215         ximax = xi[iDigit] ; 
216       if ( xi[iDigit] < ximin )
217         ximin = xi[iDigit] ; 
218       if ( zi[iDigit] > zimax )
219         zimax = zi[iDigit] ; 
220       if ( zi[iDigit] < zimin )
221         zimin = zi[iDigit] ;     
222     }
223     ximax += phosgeom->GetCrystalSize(0) / 2. ;
224     zimax += phosgeom->GetCrystalSize(2) / 2. ;
225     ximin -= phosgeom->GetCrystalSize(0) / 2. ;
226     zimin -= phosgeom->GetCrystalSize(2) / 2. ;
227     Int_t xdim = (int)( (ximax - ximin ) / phosgeom->GetCrystalSize(0) + 0.5  ) ; 
228     Int_t zdim = (int)( (zimax - zimin ) / phosgeom->GetCrystalSize(2) + 0.5 ) ;
229     
230     // 2. gets the histogram title
231     
232     Text_t title[100] ; 
233     sprintf(title,"Energy=%1.2f GeV ; Digits ; %d ", GetEnergy(), GetDigitsMultiplicity()) ;
234     
235     if (!histo) {
236       delete histo ; 
237       histo = 0 ; 
238     }
239     histo = new TH2F("cluster3D", title,  xdim, ximin, ximax, zdim, zimin, zimax)  ;
240     
241     Float_t x, z ; 
242     for(iDigit=0; iDigit<kMulDigit; iDigit++) {
243       digit = (AliPHOSDigit *) ( please->GimeDigit(fDigitsList[iDigit]) ) ;
244       phosgeom->AbsToRelNumbering(digit->GetId(), relid) ;
245       phosgeom->RelPosInModule(relid, x, z);
246       histo->Fill(x, z, fEnergyList[iDigit] ) ;
247     }
248     
249     if (!digitgraph) {
250       digitgraph = new TGraph(kMulDigit,xi,zi);
251       digitgraph-> SetMarkerStyle(5) ; 
252       digitgraph-> SetMarkerSize(1.) ;
253       digitgraph-> SetMarkerColor(1) ;
254       digitgraph-> Paint("P") ;
255     }
256     
257     Print() ;
258     histocanvas = new TCanvas("cluser", "a single cluster", 600, 500) ; 
259     histocanvas->Draw() ; 
260     histo->Draw("lego1") ; 
261     
262     delete[] xi ; 
263     delete[] zi ; 
264     
265     break;
266   }
267   
268   case kButton1Up: 
269     if (digitgraph) {
270       delete digitgraph  ;
271       digitgraph = 0 ;
272     }
273     break;
274   
275    }
276 }
277
278 //____________________________________________________________________________
279 Float_t  AliPHOSEmcRecPoint::GetDispersion() 
280 {
281   // Calculates the dispersion of the shower at the origine of the RecPoint
282
283   AliPHOSIndexToObject * please =  AliPHOSIndexToObject::GetInstance() ; 
284
285   Float_t d    = 0 ;
286   Float_t wtot = 0 ;
287
288   TVector3 locpos;
289   GetLocalPosition(locpos);
290   Float_t x = locpos.X() ;
291   Float_t z = locpos.Z() ;
292
293   AliPHOSDigit * digit ;
294   AliPHOSGeometry * phosgeom =  (AliPHOSGeometry *) fGeom ;
295   
296   Int_t iDigit;
297   for(iDigit=0; iDigit < fMulDigit; iDigit++) {
298     digit = (AliPHOSDigit *) ( please->GimeDigit(fDigitsList[iDigit]) ) ;
299     Int_t relid[4] ;
300     Float_t xi ;
301     Float_t zi ;
302     phosgeom->AbsToRelNumbering(digit->GetId(), relid) ;
303     phosgeom->RelPosInModule(relid, xi, zi);
304     Float_t w = TMath::Max(0.,fW0+TMath::Log(fEnergyList[iDigit]/fAmp ) ) ;
305     d += w*((xi-x)*(xi-x) + (zi-z)*(zi-z) ) ; 
306     wtot+=w ;
307   }
308
309   d /= wtot ;
310
311   return TMath::Sqrt(d) ;
312 }
313
314 //____________________________________________________________________________
315 void  AliPHOSEmcRecPoint::GetElipsAxis(Float_t * lambda)
316 {
317   // Calculates the axis of the shower ellipsoid
318
319   AliPHOSIndexToObject * please =  AliPHOSIndexToObject::GetInstance() ; 
320
321   Double_t wtot = 0. ;
322   Double_t x    = 0.;
323   Double_t z    = 0.;
324   Double_t dxx  = 0.;
325   Double_t dzz  = 0.;
326   Double_t dxz  = 0.;
327
328   AliPHOSDigit * digit ;
329   AliPHOSGeometry * phosgeom =  (AliPHOSGeometry *) fGeom ;
330   Int_t iDigit;
331
332   for(iDigit=0; iDigit<fMulDigit; iDigit++) {
333     digit = (AliPHOSDigit *) ( please->GimeDigit(fDigitsList[iDigit]) ) ;
334     Int_t relid[4] ;
335     Float_t xi ;
336     Float_t zi ;
337     phosgeom->AbsToRelNumbering(digit->GetId(), relid) ;
338     phosgeom->RelPosInModule(relid, xi, zi);
339     Double_t w = TMath::Max(0.,fW0+TMath::Log(fEnergyList[iDigit]/fAmp ) ) ;
340     dxx  += w * xi * xi ;
341     x    += w * xi ;
342     dzz  += w * zi * zi ;
343     z    += w * zi ; 
344     dxz  += w * xi * zi ; 
345     wtot += w ;
346   }
347   dxx /= wtot ;
348   x   /= wtot ;
349   dxx -= x * x ;
350   dzz /= wtot ;
351   z   /= wtot ;
352   dzz -= z * z ;
353   dxz /= wtot ;
354   dxz -= x * z ;
355
356 //   //Apply correction due to non-perpendicular incidence
357 //   Double_t CosX ;
358 //   Double_t CosZ ;
359 //   Double_t DistanceToIP= (Double_t ) ((AliPHOSGeometry *) fGeom)->GetIPtoCrystalSurface() ;
360   
361 //   CosX = DistanceToIP/TMath::Sqrt(DistanceToIP*DistanceToIP+x*x) ;
362 //   CosZ = DistanceToIP/TMath::Sqrt(DistanceToIP*DistanceToIP+z*z) ;
363
364 //   dxx = dxx/(CosX*CosX) ;
365 //   dzz = dzz/(CosZ*CosZ) ;
366 //   dxz = dxz/(CosX*CosZ) ;
367
368
369   lambda[0] =  0.5 * (dxx + dzz) + TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz )  ;
370   if(lambda[0] > 0)
371     lambda[0] = TMath::Sqrt(lambda[0]) ;
372
373   lambda[1] =  0.5 * (dxx + dzz) - TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz )  ;
374   if(lambda[1] > 0) //To avoid exception if numerical errors lead to negative lambda.
375     lambda[1] = TMath::Sqrt(lambda[1]) ;
376   else
377     lambda[1]= 0. ;
378 }
379
380 //____________________________________________________________________________
381 void AliPHOSEmcRecPoint::GetLocalPosition(TVector3 &LPos)
382 {
383   // Calculates the center of gravity in the local PHOS-module coordinates 
384
385   AliPHOSIndexToObject * please =  AliPHOSIndexToObject::GetInstance() ; 
386
387   if( fLocPos.X() < 1000000.) { // already evaluated
388    LPos = fLocPos ;
389    return ;
390   }
391
392   Float_t wtot = 0. ;
393  
394   Int_t relid[4] ;
395
396   Float_t x = 0. ;
397   Float_t z = 0. ;
398   
399   AliPHOSDigit * digit ;
400
401   AliPHOSGeometry * phosgeom =  (AliPHOSGeometry *) fGeom ;
402
403   Int_t iDigit;
404
405
406   for(iDigit=0; iDigit<fMulDigit; iDigit++) {
407     digit = (AliPHOSDigit *) ( please->GimeDigit(fDigitsList[iDigit]) );
408
409     Float_t xi ;
410     Float_t zi ;
411     phosgeom->AbsToRelNumbering(digit->GetId(), relid) ;
412     phosgeom->RelPosInModule(relid, xi, zi);
413     Float_t w = TMath::Max( 0., fW0 + TMath::Log( fEnergyList[iDigit] / fAmp ) ) ;
414     x    += xi * w ;
415     z    += zi * w ;
416     wtot += w ;
417
418   }
419
420   x /= wtot ;
421   z /= wtot ;
422   fLocPos.SetX(x)  ;
423   fLocPos.SetY(0.) ;
424   fLocPos.SetZ(z)  ;
425
426   LPos = fLocPos ;
427 }
428
429 //____________________________________________________________________________
430 Float_t AliPHOSEmcRecPoint::GetMaximalEnergy(void)
431 {
432   // Finds the maximum energy in the cluster
433   
434   Float_t menergy = 0. ;
435
436   Int_t iDigit;
437
438   for(iDigit=0; iDigit<fMulDigit; iDigit++) {
439  
440     if(fEnergyList[iDigit] > menergy) 
441       menergy = fEnergyList[iDigit] ;
442   }
443   return menergy ;
444 }
445
446 //____________________________________________________________________________
447 Int_t AliPHOSEmcRecPoint::GetMultiplicityAtLevel(const Float_t H) 
448 {
449   // Calculates the multiplicity of digits with energy larger than H*energy 
450   
451   Int_t multipl   = 0 ;
452   Int_t iDigit ;
453   for(iDigit=0; iDigit<fMulDigit; iDigit++) {
454
455     if(fEnergyList[iDigit] > H * fAmp) 
456       multipl++ ;
457   }
458   return multipl ;
459 }
460
461 //____________________________________________________________________________
462 Int_t  AliPHOSEmcRecPoint::GetNumberOfLocalMax(Int_t *  maxAt, Float_t * maxAtEnergy) 
463
464   // Calculates the number of local maxima in the cluster using fLocalMaxCut as the minimum
465   //  energy difference between two local maxima
466
467   AliPHOSIndexToObject * please =  AliPHOSIndexToObject::GetInstance() ; 
468
469   AliPHOSDigit * digit ;
470   AliPHOSDigit * digitN ;
471   
472
473   Int_t iDigitN ;
474   Int_t iDigit ;
475
476   for(iDigit = 0; iDigit < fMulDigit; iDigit++){
477     maxAt[iDigit] = (Int_t) ( please->GimeDigit(fDigitsList[iDigit]) ) ;
478   }
479   
480   for(iDigit = 0 ; iDigit < fMulDigit; iDigit++) {   
481     if(maxAt[iDigit] != -1) {
482       digit = (AliPHOSDigit *) maxAt[iDigit] ;
483           
484       for(iDigitN = 0; iDigitN < fMulDigit; iDigitN++) {        
485         digitN = (AliPHOSDigit *) ( please->GimeDigit(fDigitsList[iDigitN]) ) ; 
486         
487         if ( AreNeighbours(digit, digitN) ) {
488           if (fEnergyList[iDigit] > fEnergyList[iDigitN] ) {    
489             maxAt[iDigitN] = -1 ;
490             // but may be digit too is not local max ?
491             if(fEnergyList[iDigit] < fEnergyList[iDigitN] + fLocMaxCut) 
492               maxAt[iDigit] = -1 ;
493           }
494           else {
495             maxAt[iDigit] = -1 ;
496             // but may be digitN too is not local max ?
497             if(fEnergyList[iDigit] > fEnergyList[iDigitN] - fLocMaxCut) 
498               maxAt[iDigitN] = -1 ; 
499           } 
500         } // if Areneighbours
501       } // while digitN
502     } // slot not empty
503   } // while digit
504   
505   iDigitN = 0 ;
506   for(iDigit = 0; iDigit < fMulDigit; iDigit++) { 
507     if(maxAt[iDigit] != -1){
508       maxAt[iDigitN] = maxAt[iDigit] ;
509       maxAtEnergy[iDigitN] = fEnergyList[iDigit] ;
510       iDigitN++ ; 
511     }
512   }
513   return iDigitN ;
514 }
515
516
517 // //____________________________________________________________________________
518 // AliPHOSEmcRecPoint& AliPHOSEmcRecPoint::operator = (AliPHOSEmcRecPoint Clu) 
519 // {
520 //   int * dl = Clu.GetDigitsList() ; 
521  
522 //  if(fDigitsList) 
523 //     delete fDigitsList  ;
524
525 //   AliPHOSDigit * digit ;
526  
527 //   Int_t iDigit;
528
529 //   for(iDigit=0; iDigit<fMulDigit; iDigit++) {
530 //     digit = (AliPHOSDigit *) dl[iDigit];
531 //     AddDigit(*digit) ;
532 //   }
533
534 //   fAmp       = Clu.GetTotalEnergy() ;
535 //   fGeom      = Clu.GetGeom() ;
536 //   TVector3 locpos;
537 //   Clu.GetLocalPosition(locpos) ;
538 //   fLocPos    = locpos;
539 //   fMulDigit       = Clu.GetMultiplicity() ;
540 //   fMaxDigit  = Clu.GetMaximumMultiplicity() ;
541 //   fPHOSMod   = Clu.GetPHOSMod() ;
542 //   fW0        = Clu.GetLogWeightCut() ; 
543 //   fDelta     = Clu.GetDelta() ;
544 //   fLocMaxCut = Clu.GetLocMaxCut() ;
545   
546 //   delete dl ; 
547  
548 //   return *this ;
549 // }
550
551 //____________________________________________________________________________
552 void AliPHOSEmcRecPoint::Print(Option_t * option) 
553 {
554   // Print the list of digits belonging to the cluster
555   
556   cout << "AliPHOSEmcRecPoint: " << endl ;
557
558   AliPHOSDigit * digit ; 
559   Int_t iDigit;
560   AliPHOSGeometry * phosgeom =  (AliPHOSGeometry *) fGeom ;
561
562   Float_t xi ;
563   Float_t zi ;
564   Int_t relid[4] ; 
565   
566   AliPHOSIndexToObject * please =  AliPHOSIndexToObject::GetInstance() ; 
567
568   for(iDigit=0; iDigit<fMulDigit; iDigit++) {
569     digit = please->GimeDigit( fDigitsList[iDigit] ) ; 
570     phosgeom->AbsToRelNumbering(digit->GetId(), relid) ;
571     phosgeom->RelPosInModule(relid, xi, zi);
572     cout << " Id = " << digit->GetId() ;  
573     cout << "   module  = " << relid[0] ;  
574     cout << "   x  = " << xi ;  
575     cout << "   z  = " << zi ;  
576     cout << "   Energy = " << fEnergyList[iDigit] << endl ;
577   }
578   cout << "       Multiplicity    = " << fMulDigit  << endl ;
579   cout << "       Cluster Energy  = " << fAmp << endl ;
580   cout << "       Stored at position " << GetIndexInList() << endl ; 
581  
582 }
583 //______________________________________________________________________________
584 // void AliPHOSEmcRecPoint::Streamer(TBuffer &R__b)
585 // {
586 //    // Stream an object of class AliPHOSEmcRecPoint.
587
588 //    if (R__b.IsReading()) {
589 //       Version_t R__v = R__b.ReadVersion(); if (R__v) { }
590 //       AliPHOSRecPoint::Streamer(R__b);
591 //       R__b >> fDelta;
592 //       fEnergyList = new Float_t[fMulDigit] ; 
593 //       R__b.ReadFastArray(fEnergyList, fMulDigit);
594 //       R__b >> fLocMaxCut;
595 //       R__b >> fW0;
596 //    } else {
597 //       R__b.WriteVersion(AliPHOSEmcRecPoint::IsA());
598 //       AliPHOSRecPoint::Streamer(R__b);
599 //       R__b << fDelta;
600 //       R__b.WriteFastArray(fEnergyList, fMulDigit);
601 //       R__b << fLocMaxCut;
602 //       R__b << fW0;
603 //    }
604 // }
605