Mostly minor style modifications to be ready for cloning with EMCAL
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALClusterizerv1.cxx
CommitLineData
483b0559 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 **************************************************************************/
483b0559 15/* $Id$ */
16
17/* $Log:
18 1 October 2000. Yuri Kharlov:
19 AreNeighbours()
20 PPSD upper layer is considered if number of layers>1
21
22 18 October 2000. Yuri Kharlov:
23 AliEMCALClusterizerv1()
24 CPV clusterizing parameters added
25
26 MakeClusters()
27 After first PPSD digit remove EMC digits only once
28*/
29//*-- Author: Yves Schutz (SUBATECH) & Dmitri Peressounko (SUBATECH & Kurchatov Institute)
30//////////////////////////////////////////////////////////////////////////////
31// Clusterization class. Performs clusterization (collects neighbouring active cells) and
32// unfolds the clusters having several local maxima.
33// Results are stored in TreeR#, branches EMCALTowerRP (EMC recPoints),
34// EMCALPreShoRP (CPV RecPoints) and AliEMCALClusterizer (Clusterizer with all
35// parameters including input digits branch title, thresholds etc.)
36// This TTask is normally called from Reconstructioner, but can as well be used in
37// standalone mode.
38// Use Case:
39// root [0] AliEMCALClusterizerv1 * cl = new AliEMCALClusterizerv1("galice.root")
40// Warning in <TDatabasePDG::TDatabasePDG>: object already instantiated
41// //reads gAlice from header file "..."
42// root [1] cl->ExecuteTask()
43// //finds RecPoints in all events stored in galice.root
44// root [2] cl->SetDigitsBranch("digits2")
45// //sets another title for Digitis (input) branch
46// root [3] cl->SetRecPointsBranch("recp2")
47// //sets another title four output branches
48// root [4] cl->SetTowerLocalMaxCut(0.03)
49// //set clusterization parameters
50// root [5] cl->ExecuteTask("deb all time")
51// //once more finds RecPoints options are
52// // deb - print number of found rec points
53// // deb all - print number of found RecPoints and some their characteristics
54// // time - print benchmarking results
55
56// --- ROOT system ---
57
58#include "TROOT.h"
59#include "TFile.h"
60#include "TFolder.h"
61#include "TMath.h"
62#include "TMinuit.h"
63#include "TTree.h"
64#include "TSystem.h"
65#include "TBenchmark.h"
66
67// --- Standard library ---
68
69#include <iostream.h>
70#include <iomanip.h>
71
72// --- AliRoot header files ---
73
74#include "AliEMCALClusterizerv1.h"
75#include "AliEMCALDigit.h"
76#include "AliEMCALDigitizer.h"
77#include "AliEMCALTowerRecPoint.h"
78#include "AliEMCAL.h"
79#include "AliEMCALGetter.h"
80#include "AliRun.h"
81
82ClassImp(AliEMCALClusterizerv1)
83
84//____________________________________________________________________________
85 AliEMCALClusterizerv1::AliEMCALClusterizerv1() : AliEMCALClusterizer()
86{
87 // default ctor (to be used mainly by Streamer)
88
839828a6 89 InitParameters() ;
ef305168 90 fDefaultInit = kTRUE ;
483b0559 91}
92
93//____________________________________________________________________________
94AliEMCALClusterizerv1::AliEMCALClusterizerv1(const char* headerFile,const char* name)
95:AliEMCALClusterizer(headerFile, name)
96{
97 // ctor with the indication of the file where header Tree and digits Tree are stored
98
839828a6 99 InitParameters() ;
ef305168 100 fDefaultInit = kFALSE ;
483b0559 101 Init() ;
102
103}
104//____________________________________________________________________________
105 AliEMCALClusterizerv1::~AliEMCALClusterizerv1()
106{
ef305168 107 // dtor
108 // fDefaultInit = kTRUE if Clusterizer created by default ctor (to get just the parameters)
839828a6 109
ef305168 110 if (!fDefaultInit) {
111 AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
112
113 // remove the task from the folder list
114 gime->RemoveTask("C",GetName()) ;
115
116 // remove the RecPoints from the folder list
117 TString name(GetName()) ;
118 name.Remove(name.Index(":")) ;
119 gime->RemoveObjects("D", name) ; // Digits
120 gime->RemoveObjects("RT", name) ; // TowerRecPoints
121 gime->RemoveObjects("RP", name) ; // PreShoRecPoints
122
123 // Delete gAlice
124 gime->CloseFile() ;
125
126 }
127}
128
129//____________________________________________________________________________
130const TString AliEMCALClusterizerv1::BranchName() const
131{
132 TString branchName(GetName() ) ;
133 branchName.Remove(branchName.Index(Version())-1) ;
134 return branchName ;
483b0559 135}
839828a6 136
483b0559 137//____________________________________________________________________________
138Float_t AliEMCALClusterizerv1::Calibrate(Int_t amp, Bool_t inpresho) const
139{
140 if ( inpresho ) // calibrate as pre shower
141 return -fADCpedestalPreSho + amp * fADCchannelPreSho ;
142
143 else //calibrate as tower
144 return -fADCpedestalTower + amp * fADCchannelTower ;
145}
146//____________________________________________________________________________
147void AliEMCALClusterizerv1::Exec(Option_t * option)
148{
149 // Steering method
150
151 if( strcmp(GetName(), "")== 0 )
152 Init() ;
153
154 if(strstr(option,"tim"))
155 gBenchmark->Start("EMCALClusterizer");
156
157 if(strstr(option,"print"))
158 Print("") ;
159
160 gAlice->GetEvent(0) ;
161
162 //check, if the branch with name of this" already exits?
163 TObjArray * lob = (TObjArray*)gAlice->TreeR()->GetListOfBranches() ;
164 TIter next(lob) ;
165 TBranch * branch = 0 ;
166 Bool_t emcaltowerfound = kFALSE, emcalpreshofound = kFALSE, clusterizerfound = kFALSE ;
167
168 TString branchname = GetName() ;
169 branchname.Remove(branchname.Index(Version())-1) ;
170
171 while ( (branch = (TBranch*)next()) && (!emcaltowerfound || !emcalpreshofound || !clusterizerfound) ) {
172 if ( (strcmp(branch->GetName(), "EMCALTowerRP")==0) && (strcmp(branch->GetTitle(), branchname.Data())==0) )
173 emcaltowerfound = kTRUE ;
174
175 else if ( (strcmp(branch->GetName(), "EMCALPreShoRP")==0) && (strcmp(branch->GetTitle(), branchname.Data())==0) )
176 emcalpreshofound = kTRUE ;
177
178 else if ((strcmp(branch->GetName(), "AliEMCALClusterizer")==0) && (strcmp(branch->GetTitle(), GetName())==0) )
179 clusterizerfound = kTRUE ;
180 }
181
182 if ( emcalpreshofound || emcaltowerfound || clusterizerfound ) {
183 cerr << "WARNING: AliEMCALClusterizer::Exec -> Tower(PreSho)RecPoints and/or Clusterizer branch with name "
184 << branchname.Data() << " already exits" << endl ;
185 return ;
186 }
187
188 AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
189 Int_t nevents = (Int_t) gAlice->TreeE()->GetEntries() ;
190 Int_t ievent ;
191
192 for(ievent = 0; ievent < nevents; ievent++){
193
194 if(ievent == 0)
195 GetCalibrationParameters() ;
196
197 fNumberOfTowerClusters = fNumberOfPreShoClusters = 0 ;
198
199 gime->Event(ievent,"D") ;
200
201 MakeClusters() ;
202
203 if(fToUnfold)
204 MakeUnfolding() ;
205
206 WriteRecPoints(ievent) ;
207
208 if(strstr(option,"deb"))
209 PrintRecPoints(option) ;
210
211 //increment the total number of digits per run
212 fRecPointsInRun += gime->TowerRecPoints()->GetEntriesFast() ;
213 fRecPointsInRun += gime->PreShowerRecPoints()->GetEntriesFast() ;
214 }
215
216 if(strstr(option,"tim")){
217 gBenchmark->Stop("EMCALClusterizer");
218 cout << "AliEMCALClusterizer:" << endl ;
219 cout << " took " << gBenchmark->GetCpuTime("EMCALClusterizer") << " seconds for Clusterizing "
220 << gBenchmark->GetCpuTime("EMCALClusterizer")/nevents << " seconds per event " << endl ;
221 cout << endl ;
222 }
223
224}
225
226//____________________________________________________________________________
a0636361 227Bool_t AliEMCALClusterizerv1::FindFit(AliEMCALTowerRecPoint * emcRP, AliEMCALDigit ** maxAt, Float_t * maxAtEnergy,
483b0559 228 Int_t nPar, Float_t * fitparameters) const
229{
230 // Calls TMinuit to fit the energy distribution of a cluster with several maxima
231 // The initial values for fitting procedure are set equal to the positions of local maxima.
232 // Cluster will be fitted as a superposition of nPar/3 electromagnetic showers
233
234 AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
235 TClonesArray * digits = gime->Digits() ;
236
237
238 gMinuit->mncler(); // Reset Minuit's list of paramters
239 gMinuit->SetPrintLevel(-1) ; // No Printout
240 gMinuit->SetFCN(AliEMCALClusterizerv1::UnfoldingChiSquare) ;
241 // To set the address of the minimization function
242
243 TList * toMinuit = new TList();
244 toMinuit->AddAt(emcRP,0) ;
245 toMinuit->AddAt(digits,1) ;
246
247 gMinuit->SetObjectFit(toMinuit) ; // To tranfer pointer to UnfoldingChiSquare
248
249 // filling initial values for fit parameters
250 AliEMCALDigit * digit ;
251
252 Int_t ierflg = 0;
253 Int_t index = 0 ;
254 Int_t nDigits = (Int_t) nPar / 3 ;
255
256 Int_t iDigit ;
257
258 AliEMCALGeometry * geom = gime->EMCALGeometry() ;
259
260 for(iDigit = 0; iDigit < nDigits; iDigit++){
a0636361 261 digit = maxAt[iDigit];
483b0559 262
263 Int_t relid[4] ;
264 Float_t x = 0.;
265 Float_t z = 0.;
266 geom->AbsToRelNumbering(digit->GetId(), relid) ;
267 geom->PosInAlice(relid, x, z) ;
268
269 Float_t energy = maxAtEnergy[iDigit] ;
270
271 gMinuit->mnparm(index, "x", x, 0.1, 0, 0, ierflg) ;
272 index++ ;
273 if(ierflg != 0){
274 cout << "EMCAL Unfolding> Unable to set initial value for fit procedure : x = " << x << endl ;
275 return kFALSE;
276 }
277 gMinuit->mnparm(index, "z", z, 0.1, 0, 0, ierflg) ;
278 index++ ;
279 if(ierflg != 0){
280 cout << "EMCAL Unfolding> Unable to set initial value for fit procedure : z = " << z << endl ;
281 return kFALSE;
282 }
283 gMinuit->mnparm(index, "Energy", energy , 0.05*energy, 0., 4.*energy, ierflg) ;
284 index++ ;
285 if(ierflg != 0){
286 cout << "EMCAL Unfolding> Unable to set initial value for fit procedure : energy = " << energy << endl ;
287 return kFALSE;
288 }
289 }
290
291 Double_t p0 = 0.1 ; // "Tolerance" Evaluation stops when EDM = 0.0001*p0 ; The number of function call slightly
292 // depends on it.
293 Double_t p1 = 1.0 ;
294 Double_t p2 = 0.0 ;
295
296 gMinuit->mnexcm("SET STR", &p2, 0, ierflg) ; // force TMinuit to reduce function calls
297 gMinuit->mnexcm("SET GRA", &p1, 1, ierflg) ; // force TMinuit to use my gradient
298 gMinuit->SetMaxIterations(5);
299 gMinuit->mnexcm("SET NOW", &p2 , 0, ierflg) ; // No Warnings
300
301 gMinuit->mnexcm("MIGRAD", &p0, 0, ierflg) ; // minimize
302
303 if(ierflg == 4){ // Minimum not found
304 cout << "EMCAL Unfolding> Fit not converged, cluster abandoned "<< endl ;
305 return kFALSE ;
306 }
307 for(index = 0; index < nPar; index++){
308 Double_t err ;
309 Double_t val ;
310 gMinuit->GetParameter(index, val, err) ; // Returns value and error of parameter index
311 fitparameters[index] = val ;
312 }
313
314 delete toMinuit ;
315 return kTRUE;
316
317}
318
319//____________________________________________________________________________
320void AliEMCALClusterizerv1::GetCalibrationParameters()
321{
322 AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
323 TString branchname = GetName() ;
324 branchname.Remove(branchname.Index(Version())-1) ;
325
326 AliEMCALDigitizer * dig = gime->Digitizer(branchname) ;
327
328 fADCchannelTower = dig->GetTowerchannel() ;
329 fADCpedestalTower = dig->GetTowerpedestal();
330
331 fADCchannelPreSho = dig->GetPreShochannel() ;
332 fADCpedestalPreSho = dig->GetPreShopedestal() ;
333
334}
335//____________________________________________________________________________
336void AliEMCALClusterizerv1::Init()
337{
338 // Make all memory allocations which can not be done in default constructor.
339 // Attach the Clusterizer task to the list of EMCAL tasks
340
341 if ( strcmp(GetTitle(), "") == 0 )
342 SetTitle("galice.root") ;
343
344 TString branchname = GetName() ;
345 branchname.Remove(branchname.Index(Version())-1) ;
346
347 AliEMCALGetter * gime = AliEMCALGetter::GetInstance(GetTitle(), branchname, "update") ;
348 if ( gime == 0 ) {
349 cerr << "ERROR: AliEMCALClusterizerv1::Init -> Could not obtain the Getter object !" << endl ;
350 return ;
351 }
352
353 const AliEMCALGeometry * geom = gime->EMCALGeometry() ;
354 fNTowers = geom->GetNZ() * geom->GetNPhi() ;
355
356 if(!gMinuit)
357 gMinuit = new TMinuit(100) ;
358
359 gime->PostClusterizer(this) ;
360 // create a folder on the white board
361 gime->PostRecPoints(branchname ) ;
362
363 gime->PostDigits(branchname) ;
364 gime->PostDigitizer(branchname) ;
365
366}
367
368//____________________________________________________________________________
839828a6 369void AliEMCALClusterizerv1::InitParameters()
370{
371 fNumberOfPreShoClusters = fNumberOfTowerClusters = 0 ;
372
373
374
375 fPreShoClusteringThreshold = 0.0001;
376 fTowerClusteringThreshold = 0.2;
377
378 fTowerLocMaxCut = 0.03 ;
379 fPreShoLocMaxCut = 0.03 ;
380
381 fW0 = 4.5 ;
382 fW0CPV = 4.0 ;
383
384 fTimeGate = 1.e-8 ;
385
386 fToUnfold = kFALSE ;
387
388 fHeaderFileName = GetTitle() ;
389 fDigitsBranchTitle = GetName() ;
390
391 TString clusterizerName( GetName()) ;
392 if (clusterizerName.IsNull() )
393 clusterizerName = "Default" ;
394 clusterizerName.Append(":") ;
395 clusterizerName.Append(Version()) ;
396 SetName(clusterizerName) ;
397 fRecPointsInRun = 0 ;
398}
399
400//____________________________________________________________________________
483b0559 401Int_t AliEMCALClusterizerv1::AreNeighbours(AliEMCALDigit * d1, AliEMCALDigit * d2)const
402{
403 // Gives the neighbourness of two digits = 0 are not neighbour but continue searching
404 // = 1 are neighbour
405 // = 2 are not neighbour but do not continue searching
406 // neighbours are defined as digits having at least a common vertex
407 // The order of d1 and d2 is important: first (d1) should be a digit already in a cluster
408 // which is compared to a digit (d2) not yet in a cluster
409
410 AliEMCALGeometry * geom = AliEMCALGetter::GetInstance()->EMCALGeometry() ;
411
412 Int_t rv = 0 ;
413
414 Int_t relid1[4] ;
415 geom->AbsToRelNumbering(d1->GetId(), relid1) ;
416
417 Int_t relid2[4] ;
418 geom->AbsToRelNumbering(d2->GetId(), relid2) ;
419
420 if ( (relid1[0] == relid2[0]) && (relid1[1]==relid2[1]) ) { // inside the same EMCAL Arm
421 Int_t rowdiff = TMath::Abs( relid1[2] - relid2[2] ) ;
422 Int_t coldiff = TMath::Abs( relid1[3] - relid2[3] ) ;
423
424 if (( coldiff <= 1 ) && ( rowdiff <= 1 )){
425 if((relid1[1] != 0) || (TMath::Abs(d1->GetTime() - d2->GetTime() ) < fTimeGate))
426 rv = 1 ;
427 }
428 else {
429 if((relid2[2] > relid1[2]) && (relid2[3] > relid1[3]+1))
430 rv = 2; // Difference in row numbers is too large to look further
431 }
432
433 }
434 else {
435
436 if( (relid1[0] < relid2[0]) || (relid1[1] != relid2[1]) )
437 rv=2 ;
438
439 }
440
441 return rv ;
442}
443
444
445//____________________________________________________________________________
446Bool_t AliEMCALClusterizerv1::IsInTower(AliEMCALDigit * digit) const
447{
448 // Tells if (true) or not (false) the digit is in a EMCAL-Tower
449
450 Bool_t rv = kFALSE ;
451 if (!digit->IsInPreShower())
452 rv = kTRUE;
453 return rv ;
454}
455
456//____________________________________________________________________________
457Bool_t AliEMCALClusterizerv1::IsInPreShower(AliEMCALDigit * digit) const
458{
459 // Tells if (true) or not (false) the digit is in a EMCAL-PreShower
460
461 Bool_t rv = kFALSE ;
462 if (digit->IsInPreShower())
463 rv = kTRUE;
464 return rv ;
465}
466
467//____________________________________________________________________________
468void AliEMCALClusterizerv1::WriteRecPoints(Int_t event)
469{
470
471 // Creates new branches with given title
472 // fills and writes into TreeR.
473
483b0559 474 AliEMCALGetter *gime = AliEMCALGetter::GetInstance() ;
ef305168 475 TObjArray * towerRecPoints = gime->TowerRecPoints(BranchName()) ;
476 TObjArray * preshoRecPoints = gime->PreShowerRecPoints(BranchName()) ;
477 TClonesArray * digits = gime->Digits(BranchName()) ;
839828a6 478 TTree * treeR ;
483b0559 479
839828a6 480 if (!gAlice->TreeR() )
481 gAlice->MakeTree("R", fSplitFile);
482 treeR = gAlice->TreeR() ;
483
484 Int_t index ;
483b0559 485 //Evaluate position, dispersion and other RecPoint properties...
486 for(index = 0; index < towerRecPoints->GetEntries(); index++)
487 (dynamic_cast<AliEMCALTowerRecPoint *>(towerRecPoints->At(index)))->EvalAll(fW0,digits) ;
488
489 towerRecPoints->Sort() ;
490
491 for(index = 0; index < towerRecPoints->GetEntries(); index++)
492 (dynamic_cast<AliEMCALTowerRecPoint *>(towerRecPoints->At(index)))->SetIndexInList(index) ;
493
494 towerRecPoints->Expand(towerRecPoints->GetEntriesFast()) ;
495
ef305168 496 //Now the same for pre shower
483b0559 497 for(index = 0; index < preshoRecPoints->GetEntries(); index++)
498 (dynamic_cast<AliEMCALRecPoint *>(preshoRecPoints->At(index)))->EvalAll(fW0CPV,digits) ;
499
500 preshoRecPoints->Sort() ;
501
502 for(index = 0; index < preshoRecPoints->GetEntries(); index++)
503 (dynamic_cast<AliEMCALRecPoint *>(preshoRecPoints->At(index)))->SetIndexInList(index) ;
504
505 preshoRecPoints->Expand(preshoRecPoints->GetEntriesFast()) ;
506
507 //Make branches in TreeR for RecPoints and Clusterizer
483b0559 508
483b0559 509
510 Int_t bufferSize = 32000 ;
511 Int_t splitlevel = 0 ;
512
ef305168 513 //First Tower branch
839828a6 514 TBranch * emcBranch = treeR->Branch("EMCALTowerRP","TObjArray",&towerRecPoints,bufferSize,splitlevel);
ef305168 515 emcBranch->SetTitle(BranchName());
839828a6 516
483b0559 517
ef305168 518 //Now Pre Shower branch
839828a6 519 TBranch * cpvBranch = treeR->Branch("EMCALPreShoRP","TObjArray",&preshoRecPoints,bufferSize,splitlevel);
ef305168 520 cpvBranch->SetTitle(BranchName());
839828a6 521
483b0559 522
523 //And Finally clusterizer branch
ef305168 524 AliEMCALClusterizerv1 * cl = (AliEMCALClusterizerv1*)gime->Clusterizer(BranchName()) ;
839828a6 525 TBranch * clusterizerBranch = treeR->Branch("AliEMCALClusterizer","AliEMCALClusterizerv1",
483b0559 526 &cl,bufferSize,splitlevel);
ef305168 527 clusterizerBranch->SetTitle(BranchName());
839828a6 528
483b0559 529 emcBranch ->Fill() ;
530 cpvBranch ->Fill() ;
531 clusterizerBranch->Fill() ;
532
839828a6 533 treeR->AutoSave() ; //Write(0,kOverwrite) ;
483b0559 534
535}
536
537//____________________________________________________________________________
538void AliEMCALClusterizerv1::MakeClusters()
539{
540 // Steering method to construct the clusters stored in a list of Reconstructed Points
541 // A cluster is defined as a list of neighbour digits
542
543 TString branchName(GetName()) ;
544 branchName.Remove(branchName.Index(Version())-1) ;
545
546 AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
547
548 TObjArray * towerRecPoints = gime->TowerRecPoints(branchName) ;
549 TObjArray * preshoRecPoints = gime->PreShowerRecPoints(branchName) ;
550 towerRecPoints->Delete() ;
551 preshoRecPoints->Delete() ;
552
553 TClonesArray * digits = gime->Digits(branchName) ;
554 TClonesArray * digitsC = dynamic_cast<TClonesArray*>(digits->Clone()) ;
555
556
557 // Clusterization starts
558
559 TIter nextdigit(digitsC) ;
560 AliEMCALDigit * digit ;
561 Bool_t notremoved = kTRUE ;
562
563 while ( (digit = dynamic_cast<AliEMCALDigit *>(nextdigit())) ) { // scan over the list of digitsC
564 AliEMCALRecPoint * clu = 0 ;
565
566 TArrayI clusterdigitslist(1500) ;
567 Int_t index ;
568
569 if (( IsInTower (digit) && Calibrate(digit->GetAmp(),digit->IsInPreShower()) > fTowerClusteringThreshold ) ||
570 ( IsInPreShower (digit) && Calibrate(digit->GetAmp(),digit->IsInPreShower()) > fPreShoClusteringThreshold ) ) {
571
572 Int_t iDigitInCluster = 0 ;
573
574 if ( IsInTower(digit) ) {
575 // start a new Tower RecPoint
576 if(fNumberOfTowerClusters >= towerRecPoints->GetSize())
577 towerRecPoints->Expand(2*fNumberOfTowerClusters+1) ;
578
579 towerRecPoints->AddAt(new AliEMCALTowerRecPoint(""), fNumberOfTowerClusters) ;
580 clu = dynamic_cast<AliEMCALTowerRecPoint *>(towerRecPoints->At(fNumberOfTowerClusters)) ;
581 fNumberOfTowerClusters++ ;
582 clu->AddDigit(*digit, Calibrate(digit->GetAmp(),digit->IsInPreShower())) ;
583 clusterdigitslist[iDigitInCluster] = digit->GetIndexInList() ;
584 iDigitInCluster++ ;
585 digitsC->Remove(digit) ;
586
587 } else {
588
589 // start a new Pre Shower cluster
590 if(fNumberOfPreShoClusters >= preshoRecPoints->GetSize())
591 preshoRecPoints->Expand(2*fNumberOfPreShoClusters+1);
592
593 preshoRecPoints->AddAt(new AliEMCALTowerRecPoint(""), fNumberOfPreShoClusters) ;
594
595 clu = dynamic_cast<AliEMCALTowerRecPoint *>(preshoRecPoints->At(fNumberOfPreShoClusters)) ;
596 fNumberOfPreShoClusters++ ;
597 clu->AddDigit(*digit, Calibrate(digit->GetAmp(),digit->IsInPreShower() ) );
598 clusterdigitslist[iDigitInCluster] = digit->GetIndexInList() ;
599 iDigitInCluster++ ;
600 digitsC->Remove(digit) ;
601 nextdigit.Reset() ;
602
603 // Here we remove remaining Tower digits, which cannot make a cluster
604
605 if( notremoved ) {
606 while( ( digit = dynamic_cast<AliEMCALDigit *>(nextdigit()) ) ) {
607 if( IsInTower(digit) )
608 digitsC->Remove(digit) ;
609 else
610 break ;
611 }
612 notremoved = kFALSE ;
613 }
614
615 } // else
616
617 nextdigit.Reset() ;
618
619 AliEMCALDigit * digitN ;
620 index = 0 ;
621 while (index < iDigitInCluster){ // scan over digits already in cluster
622 digit = (AliEMCALDigit*)digits->At(clusterdigitslist[index]) ;
623 index++ ;
624 while ( (digitN = (AliEMCALDigit *)nextdigit()) ) { // scan over the reduced list of digits
625 Int_t ineb = AreNeighbours(digit, digitN); // call (digit,digitN) in THAT oder !!!!!
626 switch (ineb ) {
627 case 0 : // not a neighbour
628 break ;
629 case 1 : // are neighbours
630 clu->AddDigit(*digitN, Calibrate( digitN->GetAmp(), digitN->IsInPreShower() ) ) ;
631 clusterdigitslist[iDigitInCluster] = digitN->GetIndexInList() ;
632 iDigitInCluster++ ;
633 digitsC->Remove(digitN) ;
634 break ;
635 case 2 : // too far from each other
636 goto endofloop;
637 } // switch
638
639 } // while digitN
640
641 endofloop: ;
642 nextdigit.Reset() ;
643
644 } // loop over cluster
645
646 } // energy theshold
647
648
649 } // while digit
650
651 delete digitsC ;
652
653}
654
655//____________________________________________________________________________
656void AliEMCALClusterizerv1::MakeUnfolding()
657{
658 Fatal("AliEMCALClusterizerv1::MakeUnfolding", "--> Unfolding not implemented") ;
659
660// // Unfolds clusters using the shape of an ElectroMagnetic shower
661// // Performs unfolding of all EMC/CPV clusters
662
663// AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
664
665// const AliEMCALGeometry * geom = gime->EMCALGeometry() ;
666// TObjArray * emcRecPoints = gime->TowerRecPoints() ;
667// TObjArray * cpvRecPoints = gime->PreShoRecPoints() ;
668// TClonesArray * digits = gime->Digits() ;
669
670// // Unfold first EMC clusters
671// if(fNumberOfTowerClusters > 0){
672
673// Int_t nModulesToUnfold = geom->GetNModules() ;
674
675// Int_t numberofNotUnfolded = fNumberOfTowerClusters ;
676// Int_t index ;
677// for(index = 0 ; index < numberofNotUnfolded ; index++){
678
679// AliEMCALTowerRecPoint * emcRecPoint = (AliEMCALTowerRecPoint *) emcRecPoints->At(index) ;
680// if(emcRecPoint->GetEMCALMod()> nModulesToUnfold)
681// break ;
682
683// Int_t nMultipl = emcRecPoint->GetMultiplicity() ;
684// Int_t * maxAt = new Int_t[nMultipl] ;
685// Float_t * maxAtEnergy = new Float_t[nMultipl] ;
686// Int_t nMax = emcRecPoint->GetNumberOfLocalMax(maxAt, maxAtEnergy,fTowerLocMaxCut,digits) ;
687
688// if( nMax > 1 ) { // if cluster is very flat (no pronounced maximum) then nMax = 0
689// UnfoldCluster(emcRecPoint, nMax, maxAt, maxAtEnergy) ;
690// emcRecPoints->Remove(emcRecPoint);
691// emcRecPoints->Compress() ;
692// index-- ;
693// fNumberOfTowerClusters -- ;
694// numberofNotUnfolded-- ;
695// }
696
697// delete[] maxAt ;
698// delete[] maxAtEnergy ;
699// }
700// }
701// // Unfolding of EMC clusters finished
702
703
704// // Unfold now CPV clusters
705// if(fNumberOfPreShoClusters > 0){
706
707// Int_t nModulesToUnfold = geom->GetNModules() ;
708
709// Int_t numberofPreShoNotUnfolded = fNumberOfPreShoClusters ;
710// Int_t index ;
711// for(index = 0 ; index < numberofPreShoNotUnfolded ; index++){
712
713// AliEMCALRecPoint * recPoint = (AliEMCALRecPoint *) cpvRecPoints->At(index) ;
714
715// if(recPoint->GetEMCALMod()> nModulesToUnfold)
716// break ;
717
718// AliEMCALTowerRecPoint * emcRecPoint = (AliEMCALTowerRecPoint*) recPoint ;
719
720// Int_t nMultipl = emcRecPoint->GetMultiplicity() ;
721// Int_t * maxAt = new Int_t[nMultipl] ;
722// Float_t * maxAtEnergy = new Float_t[nMultipl] ;
723// Int_t nMax = emcRecPoint->GetNumberOfLocalMax(maxAt, maxAtEnergy,fPreShoLocMaxCut,digits) ;
724
725// if( nMax > 1 ) { // if cluster is very flat (no pronounced maximum) then nMax = 0
726// UnfoldCluster(emcRecPoint, nMax, maxAt, maxAtEnergy) ;
727// cpvRecPoints->Remove(emcRecPoint);
728// cpvRecPoints->Compress() ;
729// index-- ;
730// numberofPreShoNotUnfolded-- ;
731// fNumberOfPreShoClusters-- ;
732// }
733
734// delete[] maxAt ;
735// delete[] maxAtEnergy ;
736// }
737// }
738// //Unfolding of PreSho clusters finished
739
740}
741
742//____________________________________________________________________________
743Double_t AliEMCALClusterizerv1::ShowerShape(Double_t r)
744{
745 // Shape of the shower (see EMCAL TDR)
746 // If you change this function, change also the gradient evaluation in ChiSquare()
747
748 Double_t r4 = r*r*r*r ;
749 Double_t r295 = TMath::Power(r, 2.95) ;
750 Double_t shape = TMath::Exp( -r4 * (1. / (2.32 + 0.26 * r4) + 0.0316 / (1 + 0.0652 * r295) ) ) ;
751 return shape ;
752}
753
754//____________________________________________________________________________
755void AliEMCALClusterizerv1::UnfoldCluster(AliEMCALTowerRecPoint * iniTower,
756 Int_t nMax,
a0636361 757 AliEMCALDigit ** maxAt,
483b0559 758 Float_t * maxAtEnergy)
759{
760 // Performs the unfolding of a cluster with nMax overlapping showers
761
762 Fatal("AliEMCALClusterizerv1::UnfoldCluster", "--> Unfolding not implemented") ;
763
764 // AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
765// const AliEMCALGeometry * geom = gime->EMCALGeometry() ;
766// const TClonesArray * digits = gime->Digits() ;
767// TObjArray * emcRecPoints = gime->TowerRecPoints() ;
768// TObjArray * cpvRecPoints = gime->PreShoRecPoints() ;
769
770// Int_t nPar = 3 * nMax ;
771// Float_t * fitparameters = new Float_t[nPar] ;
772
773// Bool_t rv = FindFit(iniTower, maxAt, maxAtEnergy, nPar, fitparameters) ;
774// if( !rv ) {
775// // Fit failed, return and remove cluster
776// delete[] fitparameters ;
777// return ;
778// }
779
780// // create ufolded rec points and fill them with new energy lists
781// // First calculate energy deposited in each sell in accordance with fit (without fluctuations): efit[]
782// // and later correct this number in acordance with actual energy deposition
783
784// Int_t nDigits = iniTower->GetMultiplicity() ;
785// Float_t * efit = new Float_t[nDigits] ;
786// Float_t xDigit=0.,zDigit=0.,distance=0. ;
787// Float_t xpar=0.,zpar=0.,epar=0. ;
788// Int_t relid[4] ;
789// AliEMCALDigit * digit = 0 ;
790// Int_t * emcDigits = iniTower->GetDigitsList() ;
791
792// Int_t iparam ;
793// Int_t iDigit ;
794// for(iDigit = 0 ; iDigit < nDigits ; iDigit ++){
795// digit = (AliEMCALDigit*) digits->At(emcDigits[iDigit] ) ;
796// geom->AbsToRelNumbering(digit->GetId(), relid) ;
797// geom->RelPosInModule(relid, xDigit, zDigit) ;
798// efit[iDigit] = 0;
799
800// iparam = 0 ;
801// while(iparam < nPar ){
802// xpar = fitparameters[iparam] ;
803// zpar = fitparameters[iparam+1] ;
804// epar = fitparameters[iparam+2] ;
805// iparam += 3 ;
806// distance = (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar) ;
807// distance = TMath::Sqrt(distance) ;
808// efit[iDigit] += epar * ShowerShape(distance) ;
809// }
810// }
811
812
813// // Now create new RecPoints and fill energy lists with efit corrected to fluctuations
814// // so that energy deposited in each cell is distributed betwin new clusters proportionally
815// // to its contribution to efit
816
817// Float_t * emcEnergies = iniTower->GetEnergiesList() ;
818// Float_t ratio ;
819
820// iparam = 0 ;
821// while(iparam < nPar ){
822// xpar = fitparameters[iparam] ;
823// zpar = fitparameters[iparam+1] ;
824// epar = fitparameters[iparam+2] ;
825// iparam += 3 ;
826
827// AliEMCALTowerRecPoint * emcRP = 0 ;
828
829// if(iniTower->IsTower()){ //create new entries in fTowerRecPoints...
830
831// if(fNumberOfTowerClusters >= emcRecPoints->GetSize())
832// emcRecPoints->Expand(2*fNumberOfTowerClusters) ;
833
834// (*emcRecPoints)[fNumberOfTowerClusters] = new AliEMCALTowerRecPoint("") ;
835// emcRP = (AliEMCALTowerRecPoint *) emcRecPoints->At(fNumberOfTowerClusters);
836// fNumberOfTowerClusters++ ;
837// }
838// else{//create new entries in fPreShoRecPoints
839// if(fNumberOfPreShoClusters >= cpvRecPoints->GetSize())
840// cpvRecPoints->Expand(2*fNumberOfPreShoClusters) ;
841
842// (*cpvRecPoints)[fNumberOfPreShoClusters] = new AliEMCALPreShoRecPoint("") ;
843// emcRP = (AliEMCALTowerRecPoint *) cpvRecPoints->At(fNumberOfPreShoClusters);
844// fNumberOfPreShoClusters++ ;
845// }
846
847// Float_t eDigit ;
848// for(iDigit = 0 ; iDigit < nDigits ; iDigit ++){
849// digit = (AliEMCALDigit*) digits->At( emcDigits[iDigit] ) ;
850// geom->AbsToRelNumbering(digit->GetId(), relid) ;
851// geom->RelPosInModule(relid, xDigit, zDigit) ;
852// distance = (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar) ;
853// distance = TMath::Sqrt(distance) ;
854// ratio = epar * ShowerShape(distance) / efit[iDigit] ;
855// eDigit = emcEnergies[iDigit] * ratio ;
856// emcRP->AddDigit( *digit, eDigit ) ;
857// }
858// }
859
860// delete[] fitparameters ;
861// delete[] efit ;
862
863}
864
865//_____________________________________________________________________________
866void AliEMCALClusterizerv1::UnfoldingChiSquare(Int_t & nPar, Double_t * Grad, Double_t & fret, Double_t * x, Int_t iflag)
867{
868 // Calculates the Chi square for the cluster unfolding minimization
869 // Number of parameters, Gradient, Chi squared, parameters, what to do
870
871 abort() ;
872 // Fatal("AliEMCALClusterizerv1::UnfoldingChiSquare","-->Unfolding not implemented") ;
873
874// TList * toMinuit = (TList*) gMinuit->GetObjectFit() ;
875
876// AliEMCALTowerRecPoint * emcRP = (AliEMCALTowerRecPoint*) toMinuit->At(0) ;
877// TClonesArray * digits = (TClonesArray*)toMinuit->At(1) ;
878
879
880
881// // AliEMCALTowerRecPoint * emcRP = (AliEMCALTowerRecPoint *) gMinuit->GetObjectFit() ; // TowerRecPoint to fit
882
883// Int_t * emcDigits = emcRP->GetDigitsList() ;
884
885// Int_t nOdigits = emcRP->GetDigitsMultiplicity() ;
886
887// Float_t * emcEnergies = emcRP->GetEnergiesList() ;
888
889// const AliEMCALGeometry * geom = AliEMCALGetter::GetInstance()->EMCALGeometry() ;
890// fret = 0. ;
891// Int_t iparam ;
892
893// if(iflag == 2)
894// for(iparam = 0 ; iparam < nPar ; iparam++)
895// Grad[iparam] = 0 ; // Will evaluate gradient
896
897// Double_t efit ;
898
899// AliEMCALDigit * digit ;
900// Int_t iDigit ;
901
902// for( iDigit = 0 ; iDigit < nOdigits ; iDigit++) {
903
904// digit = (AliEMCALDigit*) digits->At( emcDigits[iDigit] ) ;
905
906// Int_t relid[4] ;
907// Float_t xDigit ;
908// Float_t zDigit ;
909
910// geom->AbsToRelNumbering(digit->GetId(), relid) ;
911
912// geom->RelPosInModule(relid, xDigit, zDigit) ;
913
914// if(iflag == 2){ // calculate gradient
915// Int_t iParam = 0 ;
916// efit = 0 ;
917// while(iParam < nPar ){
918// Double_t distance = (xDigit - x[iParam]) * (xDigit - x[iParam]) ;
919// iParam++ ;
920// distance += (zDigit - x[iParam]) * (zDigit - x[iParam]) ;
921// distance = TMath::Sqrt( distance ) ;
922// iParam++ ;
923// efit += x[iParam] * ShowerShape(distance) ;
924// iParam++ ;
925// }
926// Double_t sum = 2. * (efit - emcEnergies[iDigit]) / emcEnergies[iDigit] ; // Here we assume, that sigma = sqrt(E)
927// iParam = 0 ;
928// while(iParam < nPar ){
929// Double_t xpar = x[iParam] ;
930// Double_t zpar = x[iParam+1] ;
931// Double_t epar = x[iParam+2] ;
932// Double_t dr = TMath::Sqrt( (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar) );
933// Double_t shape = sum * ShowerShape(dr) ;
934// Double_t r4 = dr*dr*dr*dr ;
935// Double_t r295 = TMath::Power(dr,2.95) ;
936// Double_t deriv =-4. * dr*dr * ( 2.32 / ( (2.32 + 0.26 * r4) * (2.32 + 0.26 * r4) ) +
937// 0.0316 * (1. + 0.0171 * r295) / ( ( 1. + 0.0652 * r295) * (1. + 0.0652 * r295) ) ) ;
938
939// Grad[iParam] += epar * shape * deriv * (xpar - xDigit) ; // Derivative over x
940// iParam++ ;
941// Grad[iParam] += epar * shape * deriv * (zpar - zDigit) ; // Derivative over z
942// iParam++ ;
943// Grad[iParam] += shape ; // Derivative over energy
944// iParam++ ;
945// }
946// }
947// efit = 0;
948// iparam = 0 ;
949
950// while(iparam < nPar ){
951// Double_t xpar = x[iparam] ;
952// Double_t zpar = x[iparam+1] ;
953// Double_t epar = x[iparam+2] ;
954// iparam += 3 ;
955// Double_t distance = (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar) ;
956// distance = TMath::Sqrt(distance) ;
957// efit += epar * ShowerShape(distance) ;
958// }
959
960// fret += (efit-emcEnergies[iDigit])*(efit-emcEnergies[iDigit])/emcEnergies[iDigit] ;
961// // Here we assume, that sigma = sqrt(E)
962// }
963
964}
965
966//____________________________________________________________________________
967void AliEMCALClusterizerv1::Print(Option_t * option)const
968{
969 // Print clusterizer parameters
970
971 if( strcmp(GetName(), "") !=0 ){
972
973 // Print parameters
974
975 TString taskName(GetName()) ;
976 taskName.ReplaceAll(Version(), "") ;
977
978 cout << "---------------"<< taskName.Data() << " " << GetTitle()<< "-----------" << endl
979 << "Clusterizing digits from the file: " << fHeaderFileName.Data() << endl
980 << " Branch: " << fDigitsBranchTitle.Data() << endl
981 << endl
982 << " EMC Clustering threshold = " << fTowerClusteringThreshold << endl
983 << " EMC Local Maximum cut = " << fTowerLocMaxCut << endl
984 << " EMC Logarothmic weight = " << fW0 << endl
985 << endl
986 << " CPV Clustering threshold = " << fPreShoClusteringThreshold << endl
987 << " CPV Local Maximum cut = " << fPreShoLocMaxCut << endl
988 << " CPV Logarothmic weight = " << fW0CPV << endl
989 << endl ;
990 if(fToUnfold)
991 cout << " Unfolding on " << endl ;
992 else
993 cout << " Unfolding off " << endl ;
994
995 cout << "------------------------------------------------------------------" <<endl ;
996 }
997 else
998 cout << " AliEMCALClusterizerv1 not initialized " << endl ;
999}
1000//____________________________________________________________________________
1001void AliEMCALClusterizerv1::PrintRecPoints(Option_t * option)
1002{
1003 // Prints list of RecPoints produced at the current pass of AliEMCALClusterizer
1004
1005 TObjArray * towerRecPoints = AliEMCALGetter::GetInstance()->TowerRecPoints() ;
1006 TObjArray * preshoRecPoints = AliEMCALGetter::GetInstance()->PreShowerRecPoints() ;
1007
1008 cout << "AliEMCALClusterizerv1: : event "<<gAlice->GetEvNumber() << endl ;
1009 cout << " Found "<< towerRecPoints->GetEntriesFast() << " TOWER Rec Points and "
1010 << preshoRecPoints->GetEntriesFast() << " PRE SHOWER RecPoints" << endl ;
1011
1012 fRecPointsInRun += towerRecPoints->GetEntriesFast() ;
1013 fRecPointsInRun += preshoRecPoints->GetEntriesFast() ;
1014
1015 if(strstr(option,"all")) {
1016
1017 cout << "Tower clusters " << endl ;
1018 cout << " Index Ene(MeV) Multi Module phi r theta Lambda 1 Lambda 2 # of prim Primaries list " << endl;
1019
1020 Int_t index ;
1021 for (index = 0 ; index < towerRecPoints->GetEntries() ; index++) {
1022 AliEMCALTowerRecPoint * rp = dynamic_cast<AliEMCALTowerRecPoint * >(towerRecPoints->At(index)) ;
1023 TVector3 globalpos;
1024 rp->GetGlobalPosition(globalpos);
1025 Float_t lambda[2];
1026 rp->GetElipsAxis(lambda);
1027 Int_t * primaries;
1028 Int_t nprimaries;
1029 primaries = rp->GetPrimaries(nprimaries);
1030
1031 cout << setw(4) << rp->GetIndexInList() << " "
1032 << setw(7) << setprecision(3) << rp->GetEnergy() << " "
1033 << setw(3) << rp->GetMultiplicity() << " "
1034 << setw(1) << rp->GetEMCALArm() << " "
1035 << setw(5) << setprecision(4) << globalpos.X() << " "
1036 << setw(5) << setprecision(4) << globalpos.Y() << " "
1037 << setw(5) << setprecision(4) << globalpos.Z() << " "
1038 << setw(4) << setprecision(2) << lambda[0] << " "
1039 << setw(4) << setprecision(2) << lambda[1] << " "
1040 << setw(2) << nprimaries << " " ;
1041
1042 for (Int_t iprimary=0; iprimary<nprimaries; iprimary++)
1043 cout << setw(4) << primaries[iprimary] << " " ;
1044 cout << endl ;
1045 }
1046
1047 //Now plot Pre shower recPoints
1048
1049 cout << "-----------------------------------------------------------------------"<<endl ;
1050
1051 cout << "PreShower clusters " << endl ;
1052 cout << " Index Ene(MeV) Multi Module phi r theta Lambda 1 Lambda 2 # of prim Primaries list " << endl;
1053
1054 for (index = 0 ; index < preshoRecPoints->GetEntries() ; index++) {
1055 AliEMCALTowerRecPoint * rp = dynamic_cast<AliEMCALTowerRecPoint *>(preshoRecPoints->At(index)) ;
1056 TVector3 globalpos;
1057 rp->GetGlobalPosition(globalpos);
1058 Float_t lambda[2];
1059 rp->GetElipsAxis(lambda);
1060 Int_t * primaries;
1061 Int_t nprimaries;
1062 primaries = rp->GetPrimaries(nprimaries);
1063
1064 cout << setw(4) << rp->GetIndexInList() << " "
1065 << setw(7) << setprecision(3) << rp->GetEnergy() << " "
1066 << setw(3) << rp->GetMultiplicity() << " "
1067 << setw(1) << rp->GetEMCALArm() << " "
1068 << setw(5) << setprecision(4) << globalpos.X() << " "
1069 << setw(5) << setprecision(4) << globalpos.Y() << " "
1070 << setw(5) << setprecision(4) << globalpos.Z() << " "
1071 << setw(4) << setprecision(2) << lambda[0] << " "
1072 << setw(4) << setprecision(2) << lambda[1] << " "
1073 << setw(2) << nprimaries << " " ;
1074
1075 for (Int_t iprimary=0; iprimary<nprimaries; iprimary++)
1076 cout << setw(4) << primaries[iprimary] << " " ;
1077 cout << endl ;
1078 }
1079
1080 cout << "-----------------------------------------------------------------------"<<endl ;
1081 }
1082}
1083