397f5e92e78ef593486f5425fea79e65a1f7ebe5
[u/mrichter/AliRoot.git] / PHOS / AliPHOSClusterizerv1.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 //_________________________________________________________________________
17 // A brief description of the class
18 //*-- Author : Yves Schutz  SUBATECH 
19 //////////////////////////////////////////////////////////////////////////////
20
21 // --- ROOT system ---
22
23 #include "TMath.h" 
24
25 // --- Standard library ---
26
27 #include <iostream>
28
29 // --- AliRoot header files ---
30
31 #include "AliPHOSClusterizerv1.h"
32 #include "AliPHOSDigit.h"
33 #include "AliPHOSEmcRecPoint.h"
34 #include "AliPHOSPpsdRecPoint.h"
35 #include "AliPHOSv0.h" 
36 #include "AliRun.h" 
37
38 ClassImp(AliPHOSClusterizerv1)
39
40 //____________________________________________________________________________
41 AliPHOSClusterizerv1::AliPHOSClusterizerv1()
42 {
43   fA                       = 0.;
44   fB                       = 0.01 ;
45   fNumberOfEmcClusters     = 0 ; 
46   fNumberOfPpsdClusters    = 0 ; 
47   fEmcClusteringThreshold  = 0.1;   
48   fEmcEnergyThreshold      = 0.01;    
49   fPpsdClusteringThreshold = 0.00000015; 
50   fPpsdEnergyThreshold     = 0.0000001;  
51   fW0                      = 4.5 ;
52   fLocMaxCut               = 0.06 ;
53 }
54
55 //____________________________________________________________________________
56 Int_t AliPHOSClusterizerv1::AreNeighbours(AliPHOSDigit * d1, AliPHOSDigit * d2)
57 {
58   // neigbours are defined as digits having at least common vertex
59   // The order of A and B in AreNeighbours(A,B) is important: first (A) should be digit 
60   // in cluster, which compared with digits not clusterized yet  
61   Int_t rv = 0 ; 
62
63   AliPHOSGeometry * geom = AliPHOSGeometry::GetInstance() ;
64
65   Int_t relid1[4] ; 
66   geom->AbsToRelNumbering(d1->GetId(), relid1) ; 
67
68   Int_t relid2[4] ; 
69   geom->AbsToRelNumbering(d2->GetId(), relid2) ; 
70  
71   if ( (relid1[0] == relid2[0]) && (relid1[1]==relid2[1]) ) { // inside the same PHOS module and the same PPSD Module 
72     Int_t rowdiff = TMath::Abs( relid1[2] - relid2[2] ) ;  
73     Int_t coldiff = TMath::Abs( relid1[3] - relid2[3] ) ;  
74     
75     if (( coldiff <= 1 )  && ( rowdiff <= 1 )){
76       rv = 1 ; 
77     }
78     else {
79       if((relid2[2] > relid1[2]) && (relid2[3] > relid1[3]+1)) 
80         rv = 2; //  Difference in row numbers is too large to look further 
81     }
82
83   } 
84   else {
85     
86     if( (relid1[0] < relid2[0]) || (relid1[1] < relid2[1]) )  
87       rv=2 ;
88
89   }
90   
91   return rv ; 
92 }
93
94 //____________________________________________________________________________
95 void AliPHOSClusterizerv1::FillandSort(const DigitsList * dl, TObjArray * tl) 
96 {
97   // copies the digits with energy above thershold and sorts the list
98   // according to increasing Id number
99
100   cout << "HOLA 1" << endl;
101   AliPHOSGeometry * geom = AliPHOSGeometry::GetInstance() ;
102   Int_t relid[4] ;  
103   
104   TIter next(dl) ; 
105   AliPHOSDigit * digit ;
106   
107  
108
109   while ( (digit = (AliPHOSDigit *)next()) ) { 
110
111 //     cout << " clusterizerv1 " << endl ;
112 //     int nprim = digit->GetNprimary() ;
113 //     int * aprim = digit->GetPrimary() ;
114 //     for ( int ii = 0 ; ii < nprim ; ii++)
115 //       cout << ii << " prim = " << aprim[ii] << endl ;
116
117     Int_t id    = digit->GetId() ; 
118     Float_t ene = Calibrate(digit->GetAmp()) ; 
119     geom->AbsToRelNumbering(id, relid) ;
120     if(relid[1]==0){ // EMC
121       if ( ene > fEmcEnergyThreshold )
122         tl->Add(digit) ;
123     }
124
125     else { //Ppsd
126       if ( ene > fPpsdEnergyThreshold )
127         tl->Add(digit) ; 
128     }
129
130   }
131   cout << tl << endl;
132
133   TIter next2(tl) ; Int_t iii=0;
134   cout << "Entries is " << tl->GetEntries() << tl->GetLast() << endl;
135   while ( (digit = (AliPHOSDigit *)next2()) ) { 
136     cout << iii++ << "   "  << digit << " Amplitud is  " << digit->GetAmp() << endl;
137   }
138   cout << "HOLA de nuevo " << tl << endl;
139   tl->Print();
140   cout << "sort" << endl ;
141   tl->Sort() ; 
142 }
143
144 //____________________________________________________________________________
145 void AliPHOSClusterizerv1:: GetNumberOfClustersFound(Int_t * numb) 
146 {
147  numb[0] = fNumberOfEmcClusters ; 
148  numb[1] = fNumberOfPpsdClusters ; 
149 }
150
151 //____________________________________________________________________________
152 Bool_t AliPHOSClusterizerv1::IsInEmc(AliPHOSDigit * digit) 
153 {
154   Bool_t rv = kFALSE ; 
155
156   AliPHOSGeometry * geom = AliPHOSGeometry::GetInstance() ;  
157
158   Int_t relid[4] ; 
159   geom->AbsToRelNumbering(digit->GetId(), relid) ; 
160
161   if ( relid[1] == 0  )
162     rv = kTRUE; 
163
164   return rv ; 
165 }
166
167 //____________________________________________________________________________
168 void AliPHOSClusterizerv1::MakeClusters(const DigitsList * dl, RecPointsList * emcl, RecPointsList * ppsdl)
169 {
170   // Fill and sort the working digits list
171   TObjArray tempodigitslist( dl->GetEntries() ) ;
172   FillandSort(dl, &tempodigitslist) ; 
173
174   // Clusterization starts  
175   TIter nextdigit(&tempodigitslist) ; 
176   AliPHOSDigit * digit ; 
177   Bool_t notremoved = kTRUE ;
178
179   while ( (digit = (AliPHOSDigit *)nextdigit()) ) { // scan over the list of digits
180     AliPHOSRecPoint * clu ; 
181    
182     int * clusterdigitslist[dl->GetEntries()] ;   
183     Int_t index ;
184     if (( ( IsInEmc(digit) ) && ( Calibrate(digit->GetAmp() ) > fEmcClusteringThreshold ) ) || 
185         ( ( !IsInEmc(digit) ) && ( Calibrate(digit->GetAmp() ) > fPpsdClusteringThreshold ) ) ) {
186   
187       Int_t iDigitInCluster = 0 ; 
188
189       if  ( IsInEmc(digit) ) {   
190         new ((*emcl)[fNumberOfEmcClusters]) AliPHOSEmcRecPoint(fW0, fLocMaxCut) ; // start a new EMC RecPoint
191         clu = (AliPHOSEmcRecPoint *) (*emcl)[fNumberOfEmcClusters] ; 
192         fNumberOfEmcClusters++ ; 
193         clu->AddDigit(*digit, Calibrate(digit->GetAmp())) ; 
194
195         clusterdigitslist[iDigitInCluster] = (int* ) digit ;    
196         iDigitInCluster++ ; 
197         tempodigitslist.Remove(digit) ; 
198       }
199
200       else { 
201         new ((*ppsdl)[fNumberOfPpsdClusters]) AliPHOSPpsdRecPoint() ; // start a new PPSD cluster
202         clu =  (AliPHOSPpsdRecPoint *) ppsdl->At(fNumberOfPpsdClusters)  ;  
203         fNumberOfPpsdClusters++ ; 
204         clu->AddDigit(*digit, Calibrate(digit->GetAmp()) ) ;    
205         clusterdigitslist[iDigitInCluster] = (int* ) digit ;    
206         iDigitInCluster++ ; 
207         tempodigitslist.Remove(digit) ; 
208         nextdigit.Reset() ;
209         
210         // Here we remove resting EMC digits, which cannot make cluster
211
212         if( notremoved ) { 
213           
214           while( ( digit = (AliPHOSDigit *)nextdigit() ) ) {
215             
216             if( IsInEmc(digit) ) 
217               tempodigitslist.Remove(digit) ;
218             else 
219               break ;
220           
221           } // while digit  
222           
223         } // if notremoved 
224         
225       } // else        
226       
227       nextdigit.Reset() ;
228
229       AliPHOSDigit * digitN ; 
230       index = 0 ;
231       while (index < iDigitInCluster){ // scan over digits already in cluster 
232         digit = (AliPHOSDigit *) clusterdigitslist[index]  ;      
233         index++ ; 
234         while ( (digitN = (AliPHOSDigit *)nextdigit()) ) { // scan over the reduced list of digits 
235           Int_t ineb = AreNeighbours(digit, digitN);   //  call (digit,digitN) in THAT oder !!!!!
236           switch (ineb ) {
237           case 0 :   // not a neighbour
238             break ;      
239           case 1 :   // are neighbours 
240             clu->AddDigit( *digitN, Calibrate( digitN->GetAmp() ) ) ;
241             clusterdigitslist[iDigitInCluster] =(int*) digitN ; 
242             iDigitInCluster++ ; 
243             tempodigitslist.Remove(digitN) ;
244             break ;
245           case 2 :   // too far from each other
246             goto endofloop;   
247           } // switch
248           
249         } // while digitN
250
251       endofloop: ;
252         nextdigit.Reset() ; 
253         
254       } // loop over cluster     
255  
256    }  //below energy theshold  
257   
258   } // while digit
259
260   tempodigitslist.Clear() ; 
261 }
262
263 //____________________________________________________________________________
264 void AliPHOSClusterizerv1::PrintParameters() 
265 {
266   cout << "PHOS Clusterizer version 1 :" << endl 
267        << "                       EMC  Clustering threshold = " << fEmcClusteringThreshold << endl
268        << "                       EMC  Energy threshold     = " << fEmcEnergyThreshold << endl                  
269        << "                      PPSD  Clustering threshold = " << fPpsdClusteringThreshold << endl
270        << "                      PPSD  Energy threshold     = " << fPpsdEnergyThreshold << endl ;                
271 }