New Version from Alla
[u/mrichter/AliRoot.git] / PHOS / AliPHOSClusterizerv1.cxx
CommitLineData
d15a28e7 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
9f616d61 27#include <iostream>
d15a28e7 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
38ClassImp(AliPHOSClusterizerv1)
39
40//____________________________________________________________________________
41AliPHOSClusterizerv1::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//____________________________________________________________________________
56Int_t AliPHOSClusterizerv1::AreNeighbours(AliPHOSDigit * d1, AliPHOSDigit * d2)
57{
9f616d61 58 // neigbours are defined as digits having at least common vertex
d15a28e7 59 // The order of A and B in AreNeighbours(A,B) is important: first (A) should be digit
9f616d61 60 // in cluster, which compared with digits not clusterized yet
d15a28e7 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//____________________________________________________________________________
95void 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
d15a28e7 100 AliPHOSGeometry * geom = AliPHOSGeometry::GetInstance() ;
101 Int_t relid[4] ;
102
103 TIter next(dl) ;
104 AliPHOSDigit * digit ;
105
106 while ( (digit = (AliPHOSDigit *)next()) ) {
107
108 Int_t id = digit->GetId() ;
109 Float_t ene = Calibrate(digit->GetAmp()) ;
110 geom->AbsToRelNumbering(id, relid) ;
111
112 if(relid[1]==0){ // EMC
113 if ( ene > fEmcEnergyThreshold )
114 tl->Add(digit) ;
115 }
116
117 else { //Ppsd
118 if ( ene > fPpsdEnergyThreshold )
119 tl->Add(digit) ;
120 }
121
122 }
123 tl->Sort() ;
124}
125
126//____________________________________________________________________________
127void AliPHOSClusterizerv1:: GetNumberOfClustersFound(Int_t * numb)
128{
129 numb[0] = fNumberOfEmcClusters ;
130 numb[1] = fNumberOfPpsdClusters ;
131}
132
133//____________________________________________________________________________
134Bool_t AliPHOSClusterizerv1::IsInEmc(AliPHOSDigit * digit)
135{
9f616d61 136 Bool_t rv = kFALSE ;
d15a28e7 137
138 AliPHOSGeometry * geom = AliPHOSGeometry::GetInstance() ;
139
140 Int_t relid[4] ;
141 geom->AbsToRelNumbering(digit->GetId(), relid) ;
142
143 if ( relid[1] == 0 )
144 rv = kTRUE;
145
146 return rv ;
147}
148
149//____________________________________________________________________________
150void AliPHOSClusterizerv1::MakeClusters(const DigitsList * dl, RecPointsList * emcl, RecPointsList * ppsdl)
151{
d15a28e7 152 // Fill and sort the working digits list
153 TObjArray TempoDigitsList( dl->GetEntries() ) ;
9f616d61 154 FillandSort(dl, &TempoDigitsList) ;
d15a28e7 155
156 // Clusterization starts
157 TIter nextdigit(&TempoDigitsList) ;
158 AliPHOSDigit * digit ;
159 Bool_t NotRemoved = kTRUE ;
9f616d61 160 cout << "Cluster1" << endl;
d15a28e7 161 while ( (digit = (AliPHOSDigit *)nextdigit()) ) { // scan over the list of digits
162 AliPHOSRecPoint * clu ;
9f616d61 163
d15a28e7 164 int * ClusterDigitsList[dl->GetEntries()] ;
165 Int_t index ;
9f616d61 166 if (( ( IsInEmc(digit) ) && ( Calibrate(digit->GetAmp() ) > fEmcClusteringThreshold ) ) ||
167 ( ( !IsInEmc(digit) ) && ( Calibrate(digit->GetAmp() ) > fPpsdClusteringThreshold ) ) ) {
d15a28e7 168
169 Int_t iDigitInCluster = 0 ;
170
9f616d61 171 if ( IsInEmc(digit) ) {
172 emcl->Print("");
173 cout << "Cluster2 EMC" << endl;
174 new ((*emcl)[fNumberOfEmcClusters]) AliPHOSEmcRecPoint(fW0, fLocMaxCut) ; // start a new EMC RecPoint
175 clu = (AliPHOSEmcRecPoint *) (*emcl)[fNumberOfEmcClusters] ;
176 fNumberOfEmcClusters++ ;
177 clu->AddDigit(*digit, Calibrate(digit->GetAmp())) ;
d15a28e7 178
9f616d61 179 ClusterDigitsList[iDigitInCluster] = (int* ) digit ;
180 iDigitInCluster++ ;
d15a28e7 181 TempoDigitsList.Remove(digit) ;
182 }
183
184 else {
9f616d61 185 cout << "Cluster2 PPSD" << endl;
186 new ((*ppsdl)[fNumberOfPpsdClusters]) AliPHOSPpsdRecPoint() ; // start a new PPSD cluster
187 clu = (AliPHOSPpsdRecPoint *) ppsdl->At(fNumberOfPpsdClusters) ;
188 fNumberOfPpsdClusters++ ;
189 clu->AddDigit(*digit, 0.) ;
190 ClusterDigitsList[iDigitInCluster] = (int* ) digit ;
191 iDigitInCluster++ ;
192 TempoDigitsList.Remove(digit) ;
d15a28e7 193 nextdigit.Reset() ;
194
9f616d61 195 // Here we remove resting EMC digits, which cannot make cluster
196
197 if( NotRemoved ) {
d15a28e7 198
9f616d61 199 while( ( digit = (AliPHOSDigit *)nextdigit() ) ) {
d15a28e7 200
9f616d61 201 if( IsInEmc(digit) )
202 TempoDigitsList.Remove(digit) ;
d15a28e7 203 else
204 break ;
9f616d61 205
206 } // while digit
d15a28e7 207
208 } // if NotRemoved
209
210 } // else
211
212 nextdigit.Reset() ;
213
214 AliPHOSDigit * digitN ;
215 index = 0 ;
9f616d61 216 while (index < iDigitInCluster){ // scan over digits already in cluster
217 digit = (AliPHOSDigit *) ClusterDigitsList[index] ;
218 index++ ;
d15a28e7 219 while ( (digitN = (AliPHOSDigit *)nextdigit()) ) { // scan over the reduced list of digits
220 Int_t ineb = AreNeighbours(digit, digitN); // call (digit,digitN) in THAT oder !!!!!
221 switch (ineb ) {
9f616d61 222 case 0 : // not a neighbour
d15a28e7 223 break ;
9f616d61 224 case 1 : // are neighbours
225 clu->AddDigit( *digitN, Calibrate( digitN->GetAmp() ) ) ;
226 ClusterDigitsList[iDigitInCluster] =(int*) digitN ;
227 iDigitInCluster++ ;
d15a28e7 228 TempoDigitsList.Remove(digitN) ;
229 break ;
9f616d61 230 case 2 : // too far from each other
d15a28e7 231 goto endofloop;
232 } // switch
233
234 } // while digitN
235
236 endofloop: ;
237 nextdigit.Reset() ;
238
239 } // loop over cluster
240
241 } //below energy theshold
242
243 } // while digit
244
245 TempoDigitsList.Clear() ;
246}
247
248//____________________________________________________________________________
249void AliPHOSClusterizerv1::PrintParameters()
250{
251 cout << "PHOS Clusterizer version 1 :" << endl
252 << " EMC Clustering threshold = " << fEmcClusteringThreshold << endl
253 << " EMC Energy threshold = " << fEmcEnergyThreshold << endl
254 << " PPSD Clustering threshold = " << fPpsdClusteringThreshold << endl
255 << " PPSD Energy threshold = " << fPpsdEnergyThreshold << endl ;
256}