coding convention
[u/mrichter/AliRoot.git] / PHOS / AliPHOSClusterizerv2.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 //  Clusterization class for IHEP reconstruction.
20 // Performs clusterization (collects neighbouring active cells)
21 // It differs from AliPHOSClusterizerv1 in neighbour definition only
22 //*-- Author: Boris Polichtchouk, IHEP
23
24 // --- ROOT system ---
25 #include "TBenchmark.h"
26
27 // --- Standard library ---
28
29 // --- AliRoot header files ---
30 #include "AliPHOSClusterizerv2.h"
31 #include "AliPHOSGetter.h"
32 #include "TFolder.h"
33 #include "AliPHOSEvalRecPoint.h"
34 #include "AliPHOSRecCpvManager.h"
35 #include "AliPHOSRecEmcManager.h"
36 #include "AliPHOSGeometry.h"
37
38 ClassImp(AliPHOSClusterizerv2)
39
40 //____________________________________________________________________________
41 AliPHOSClusterizerv2::AliPHOSClusterizerv2() : AliPHOSClusterizerv1() 
42 {}
43
44 //____________________________________________________________________________
45 AliPHOSClusterizerv2::AliPHOSClusterizerv2(const char * headerFile, const char * name):
46 AliPHOSClusterizerv1(headerFile,name)
47 {}
48
49 //____________________________________________________________________________
50 AliPHOSClusterizerv2::AliPHOSClusterizerv2(const AliPHOSClusterizerv2 & clu):
51 AliPHOSClusterizerv1(clu)
52 {}
53
54 //____________________________________________________________________________
55 void AliPHOSClusterizerv2::GetNumberOfClustersFound(int* numb) const
56 {
57   // Returns the number of found EMC and CPV rec.points
58
59   AliPHOSGetter * gime = AliPHOSGetter::Instance() ;   
60   numb[0] = gime->EmcRecPoints()->GetEntries();  
61   numb[1] = gime->CpvRecPoints()->GetEntries();  
62 }
63
64 //____________________________________________________________________________
65 void AliPHOSClusterizerv2::Exec(Option_t* option)
66 {
67   // Steering method
68
69   if(strstr(option,"tim"))
70     gBenchmark->Start("PHOSClusterizer"); 
71   
72   if(strstr(option,"print"))
73     Print() ; 
74
75   AliPHOSGetter * gime = AliPHOSGetter::Instance() ; 
76
77   TFolder* wPoolF =  gime->PhosLoader()->GetDetectorDataFolder();
78   
79   TObjArray* wPool = new TObjArray(400);
80   wPool->SetName("SmartPoints");
81   wPoolF->Add(wPool);
82   wPoolF->Add(this);
83
84   Int_t nevents = gime->MaxEvent() ;
85   Int_t ievent ;
86
87   for(ievent = 0; ievent<nevents; ievent++) {
88     
89     gime->Event(ievent,"D") ;
90     
91     Info("Exec", "MakeClusters invoked..") ;
92     MakeClusters() ;
93     Info("Exec", "MakeClusters done.") ;
94
95
96     //SmartRecPoints will communicate with wPool.
97
98     AliPHOSEvalRecPoint* rp=0;
99
100     // CPV reconstruction
101
102     AliPHOSRecCpvManager* recCpv = new AliPHOSRecCpvManager();
103     wPoolF->Add(recCpv);
104
105     Int_t iPoint; //loop variable
106
107     for(iPoint=0; iPoint<gime->CpvRecPoints()->GetEntriesFast(); iPoint++) {
108       rp = new AliPHOSEvalRecPoint(iPoint, kTRUE);
109       rp->MakeJob();
110     }
111
112     AliPHOSEvalRecPoint pt;
113     pt.UpdateWorkingPool();
114
115     TObjArray * cpvRecPoints = gime->CpvRecPoints() ; 
116     Int_t nOldCpv = cpvRecPoints->GetEntries();
117     cpvRecPoints->Delete();
118     cpvRecPoints->Compress();
119
120     Int_t i; //loop variable
121
122     for(i=0; i<wPool->GetEntries(); i++)
123       cpvRecPoints->Add(wPool->At(i));
124
125     wPool->Clear();
126     wPool->Compress();
127
128     wPoolF->Remove(recCpv);
129     delete recCpv;
130
131     Info("Exec", "       %d", gime->CpvRecPoints()->GetEntries() ) ;
132     Info("Exec", "       %d cpvRecPoints", cpvRecPoints->GetEntries() ) ;
133
134
135     // Now Emc reconstruction
136
137     AliPHOSRecEmcManager* recEmc = new AliPHOSRecEmcManager();
138     wPoolF->Add(recEmc);
139
140     for(iPoint=0; iPoint<gime->EmcRecPoints()->GetEntriesFast(); iPoint++) {
141       rp = new AliPHOSEvalRecPoint(iPoint, kFALSE);
142       rp->MakeJob();
143     }
144
145     pt.UpdateWorkingPool();
146
147     TObjArray * emcRecPoints = gime->EmcRecPoints() ; 
148     Int_t nOldEmc = emcRecPoints->GetEntries();
149     emcRecPoints->Delete();
150     emcRecPoints->Compress();
151
152     for(i=0; i<wPool->GetEntries(); i++)
153       emcRecPoints->Add(wPool->At(i));
154
155     wPool->Clear();
156     wPool->Compress();
157
158     wPoolF->Remove(recEmc);
159     delete recEmc;
160
161     TString message ; 
162     message  = "       %d  OLD cpvRecPoints\n" ; 
163     message += "       %d\n" ; 
164     message += "       %d cpvRecPoints\n" ; 
165
166     message += "       %d OLD emcRecPoints " ; 
167     message += "       %d\n" ;
168     message += "       %d emcRecPoints\n" ;
169
170     Info("Exec", message.Data(), 
171          nOldCpv, 
172          gime->CpvRecPoints()->GetEntries(),cpvRecPoints->GetEntries(), 
173          nOldEmc, 
174          gime->EmcRecPoints()->GetEntries(), emcRecPoints->GetEntries() ) ; 
175
176     WriteRecPoints();
177
178
179   } // loop over events
180
181   if(strstr(option,"tim")) {
182     gBenchmark->Stop("PHOSClusterizer");
183     Info("Exec","took %f seconds for Clusterizing", gBenchmark->GetCpuTime("PHOSClusterizer") ) ;
184   }
185 }
186
187 //____________________________________________________________________________
188 Int_t AliPHOSClusterizerv2::AreNeighbours(AliPHOSDigit* d1, AliPHOSDigit* d2) const
189 {
190   // Points are neighbours if they have common edge.
191   // Points with common vertex are NOT neighbours.
192   // This treatment of neighbourship is the part of 
193   // IHEP algorithm of clusterization.
194
195   // Gives the neighbourness of two digits = 0 are not neighbour but continue searching 
196   //                                       = 1 are neighbour
197   //                                       = 2 are not neighbour but do not continue searching
198   // The order of d1 and d2 is important: first (d1) should be a digit already in a cluster 
199   // which is compared to a digit (d2)  not yet in a cluster  
200
201   const AliPHOSGeometry * geom = AliPHOSGetter::Instance()->PHOSGeometry();
202
203   Int_t rv = 0 ; 
204
205   Int_t relid1[4] ; 
206   geom->AbsToRelNumbering(d1->GetId(), relid1) ; 
207
208   Int_t relid2[4] ; 
209   geom->AbsToRelNumbering(d2->GetId(), relid2) ; 
210  
211   if ( (relid1[0] == relid2[0]) && (relid1[1]==relid2[1]) ) { // inside the same PHOS module and the same PPSD Module 
212     Int_t rowdiff = TMath::Abs( relid1[2] - relid2[2] ) ;  
213     Int_t coldiff = TMath::Abs( relid1[3] - relid2[3] ) ;  
214     
215     if ( ( (coldiff < 1) && (rowdiff <= 1) ) || ( ( coldiff <= 1 )  && ( rowdiff < 1 ) ) ){
216       rv = 1 ; 
217     }
218     else {
219       if((relid2[2] > relid1[2]) && (relid2[3] > relid1[3]+1)) 
220         rv = 2; //  Difference in row numbers is too large to look further 
221     }
222
223   } 
224   else {
225     
226     if( (relid1[0] < relid2[0]) || (relid1[1] < relid2[1]) )  
227       rv=2 ;
228
229   }
230
231   return rv ; 
232
233 }