Dimitri : bug concerning relative distances fixed
[u/mrichter/AliRoot.git] / PHOS / AliPHOSTrackSegmentMakerv1.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
b2a60966 16/* $Id$ */
17
d15a28e7 18//_________________________________________________________________________
b2a60966 19// Implementation version 1 of algorithm class to construct PHOS track segments
20// Associates EMC and PPSD clusters
21// Unfolds the EMC cluster
22//
23//*-- Author: Dmitri Peressounko (RRC Ki & SUBATECH)
24//
d15a28e7 25
26// --- ROOT system ---
27
28#include "TObjArray.h"
29#include "TClonesArray.h"
9f616d61 30#include "TObjectTable.h"
d15a28e7 31
32// --- Standard library ---
33
de9ec31b 34#include <iostream.h>
d15a28e7 35
36// --- AliRoot header files ---
37
38#include "AliPHOSTrackSegmentMakerv1.h"
98cbd830 39#include "AliPHOSIndexToObject.h"
d15a28e7 40#include "AliPHOSTrackSegment.h"
41#include "AliPHOSLink.h"
42#include "AliPHOSv0.h"
43#include "AliRun.h"
44
92862013 45extern void UnfoldingChiSquare(Int_t &nPar, Double_t *Grad, Double_t & fret, Double_t *x, Int_t iflag) ;
9f616d61 46
d15a28e7 47ClassImp( AliPHOSTrackSegmentMakerv1)
48
49
50//____________________________________________________________________________
b2a60966 51 AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1() : AliPHOSTrackSegmentMaker()
d15a28e7 52{
9f616d61 53 // ctor
b2a60966 54
28c3a259 55 fR0 = 10. ;
d15a28e7 56 //clusters are sorted in "rows" and "columns" of width geom->GetCrystalSize(0),
98cbd830 57 fDelta = fR0 + fGeom->GetCrystalSize(0) ;
69183710 58 if(!gMinuit) gMinuit = new TMinuit(100) ;
6727be7e 59 fUnfoldFlag = kTRUE ;
d15a28e7 60}
61
9f616d61 62//____________________________________________________________________________
63 AliPHOSTrackSegmentMakerv1::~AliPHOSTrackSegmentMakerv1()
64{
65 // dtor
69183710 66
67 // delete gMinuit ;
9f616d61 68}
b2a60966 69
d15a28e7 70//____________________________________________________________________________
71Bool_t AliPHOSTrackSegmentMakerv1::FindFit(AliPHOSEmcRecPoint * emcRP, int * maxAt, Float_t * maxAtEnergy,
92862013 72 Int_t nPar, Float_t * fitparameters)
9f616d61 73{
b2a60966 74 // Calls TMinuit to fit the energy distribution of a cluster with several maxima
6ad0bfa0 75
9f616d61 76 gMinuit->SetPrintLevel(-1) ; // No Printout
77 gMinuit->SetFCN(UnfoldingChiSquare) ; // To set the address of the minimization function
78 gMinuit->SetObjectFit(emcRP) ; // To tranfer pointer to UnfoldingChiSquare
79
80 // filling initial values for fit parameters
d15a28e7 81 AliPHOSDigit * digit ;
9f616d61 82
83 Int_t ierflg = 0;
84 Int_t index = 0 ;
92862013 85 Int_t nDigits = (Int_t) nPar / 3 ;
9f616d61 86
d15a28e7 87 Int_t iDigit ;
9f616d61 88
89
92862013 90 for(iDigit = 0; iDigit < nDigits; iDigit++){
d15a28e7 91 digit = (AliPHOSDigit *) maxAt[iDigit];
92
92862013 93 Int_t relid[4] ;
d15a28e7 94 Float_t x ;
95 Float_t z ;
98cbd830 96 fGeom->AbsToRelNumbering(digit->GetId(), relid) ;
97 fGeom->RelPosInModule(relid, x, z) ;
d15a28e7 98
92862013 99 Float_t energy = maxAtEnergy[iDigit] ;
d15a28e7 100
9f616d61 101 gMinuit->mnparm(index, "x", x, 0.1, 0, 0, ierflg) ;
102 index++ ;
d15a28e7 103 if(ierflg != 0){
104 cout << "PHOS Unfolding> Unable to set initial value for fit procedure : x = " << x << endl ;
105 return kFALSE;
106 }
9f616d61 107 gMinuit->mnparm(index, "z", z, 0.1, 0, 0, ierflg) ;
108 index++ ;
d15a28e7 109 if(ierflg != 0){
110 cout << "PHOS Unfolding> Unable to set initial value for fit procedure : z = " << z << endl ;
111 return kFALSE;
112 }
92862013 113 gMinuit->mnparm(index, "Energy", energy , 0.05*energy, 0., 4.*energy, ierflg) ;
9f616d61 114 index++ ;
d15a28e7 115 if(ierflg != 0){
92862013 116 cout << "PHOS Unfolding> Unable to set initial value for fit procedure : energy = " << energy << endl ;
d15a28e7 117 return kFALSE;
6ad0bfa0 118 }
d15a28e7 119 }
120
9f616d61 121 Double_t p0 = 0.1 ; // "Tolerance" Evaluation stops when EDM = 0.0001*p0 ; The number of function call slightly
122 // depends on it.
123 Double_t p1 = 1.0 ;
124 Double_t p2 = 0.0 ;
125
69183710 126 gMinuit->mnexcm("SET STR", &p2, 0, ierflg) ; // force TMinuit to reduce function calls
9f616d61 127 gMinuit->mnexcm("SET GRA", &p1, 1, ierflg) ; // force TMinuit to use my gradient
d15a28e7 128 gMinuit->SetMaxIterations(5);
9f616d61 129 gMinuit->mnexcm("SET NOW", &p2 , 0, ierflg) ; // No Warnings
98cbd830 130
9f616d61 131 gMinuit->mnexcm("MIGRAD", &p0, 0, ierflg) ; // minimize
98cbd830 132
9f616d61 133 if(ierflg == 4){ // Minimum not found
134 cout << "PHOS Unfolding> Fit not converged, cluster abandoned "<< endl ;
d15a28e7 135 return kFALSE ;
136 }
92862013 137 for(index = 0; index < nPar; index++){
d15a28e7 138 Double_t err ;
139 Double_t val ;
9f616d61 140 gMinuit->GetParameter(index, val, err) ; // Returns value and error of parameter index
92862013 141 fitparameters[index] = val ;
d15a28e7 142 }
98cbd830 143
d15a28e7 144 return kTRUE;
9f616d61 145
d15a28e7 146}
9f616d61 147
d15a28e7 148//____________________________________________________________________________
98cbd830 149void AliPHOSTrackSegmentMakerv1::FillOneModule(AliPHOSRecPoint::RecPointsList * emcIn,
28c3a259 150 TArrayI * emcOut,
88714635 151 AliPHOSRecPoint::RecPointsList * ppsdIn,
28c3a259 152 TArrayI * ppsdOutUp,
153 TArrayI * ppsdOutLow,
88714635 154 Int_t & phosmod,
155 Int_t & emcStopedAt,
156 Int_t & ppsdStopedAt)
9f616d61 157{
69183710 158 // Fill xxxOut arrays with clusters from one PHOS module
9f616d61 159
160 AliPHOSEmcRecPoint * emcRecPoint ;
d15a28e7 161 AliPHOSPpsdRecPoint * ppsdRecPoint ;
162 Int_t index ;
6ad0bfa0 163
92862013 164 Int_t nEmcUnfolded = emcIn->GetEntries() ;
28c3a259 165 emcOut->Set(nEmcUnfolded);
166 Int_t inEmcOut = 0 ;
92862013 167 for(index = emcStopedAt; index < nEmcUnfolded; index++){
6ad0bfa0 168
98cbd830 169 emcRecPoint = (AliPHOSEmcRecPoint *) emcIn->At(index) ;
9f616d61 170
98cbd830 171 if(emcRecPoint->GetPHOSMod() != phosmod )
172 break ;
31aa6d6c 173
28c3a259 174 emcOut->AddAt(emcRecPoint->GetIndexInList(),inEmcOut) ;
175 inEmcOut++ ;
d15a28e7 176 }
28c3a259 177 emcOut->Set(inEmcOut) ;
178
d15a28e7 179 emcStopedAt = index ;
180
28c3a259 181 ppsdOutLow->Set(ppsdIn->GetEntries()) ;
182 ppsdOutUp->Set(ppsdIn->GetEntries()) ;
183 Int_t inPpsdLow = 0;
184 Int_t inPpsdUp = 0;
d15a28e7 185 for(index = ppsdStopedAt; index < ppsdIn->GetEntries(); index++){
83974468 186 ppsdRecPoint = (AliPHOSPpsdRecPoint *) ppsdIn->At(index) ;
92862013 187 if(ppsdRecPoint->GetPHOSMod() != phosmod )
9f616d61 188 break ;
d15a28e7 189 if(ppsdRecPoint->GetUp() )
28c3a259 190 ppsdOutUp->AddAt(index,inPpsdUp++) ;
d15a28e7 191 else
28c3a259 192 ppsdOutLow->AddAt(index,inPpsdLow++) ;
d15a28e7 193 }
28c3a259 194 ppsdOutLow->Set(inPpsdLow);
195 ppsdOutUp->Set(inPpsdUp);
d15a28e7 196 ppsdStopedAt = index ;
6ad0bfa0 197
d15a28e7 198}
199//____________________________________________________________________________
92862013 200Float_t AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcclu,AliPHOSPpsdRecPoint * PpsdClu, Bool_t &toofar)
d15a28e7 201{
b2a60966 202 // Calculates the distance between the EMC RecPoint and the PPSD RecPoint
203
92862013 204 Float_t r = fR0 ;
d15a28e7 205
206 TVector3 vecEmc ;
207 TVector3 vecPpsd ;
208
92862013 209 emcclu->GetLocalPosition(vecEmc) ;
d15a28e7 210 PpsdClu->GetLocalPosition(vecPpsd) ;
92862013 211 if(emcclu->GetPHOSMod() == PpsdClu->GetPHOSMod()){
d15a28e7 212 if(vecPpsd.X() >= vecEmc.X() - fDelta ){
213 if(vecPpsd.Z() >= vecEmc.Z() - fDelta ){
6ad0bfa0 214 // Correct to difference in CPV and EMC position due to different distance to center.
215 // we assume, that particle moves from center
98cbd830 216 Float_t dCPV = fGeom->GetIPtoOuterCoverDistance();
217 Float_t dEMC = fGeom->GetIPtoCrystalSurface() ;
92862013 218 dEMC = dEMC / dCPV ;
219 vecPpsd = dEMC * vecPpsd - vecEmc ;
220 r = vecPpsd.Mag() ;
d15a28e7 221 } // if zPpsd >= zEmc - fDelta
92862013 222 toofar = kFALSE ;
d15a28e7 223 } // if xPpsd >= xEmc - fDelta
224 else
92862013 225 toofar = kTRUE ;
d15a28e7 226 }
227 else
92862013 228 toofar = kTRUE ;
d15a28e7 229
92862013 230 return r ;
d15a28e7 231}
232
233//____________________________________________________________________________
28c3a259 234void AliPHOSTrackSegmentMakerv1::MakeLinks(TArrayI * emcRecPoints, TArrayI * ppsdRecPointsUp,
235 TArrayI * ppsdRecPointsLow, TClonesArray * linklowArray,
92862013 236 TClonesArray *linkupArray)
d15a28e7 237{
b2a60966 238 // Finds distances (links) between all EMC and PPSD clusters, which are not further apart from each other than fR0
239
d15a28e7 240
92862013 241 AliPHOSPpsdRecPoint * ppsdlow ;
242 AliPHOSPpsdRecPoint * ppsdup ;
243 AliPHOSEmcRecPoint * emcclu ;
28c3a259 244
d15a28e7 245 Int_t iLinkLow = 0 ;
246 Int_t iLinkUp = 0 ;
247
28c3a259 248 Int_t iEmcRP;
249
250 for(iEmcRP = 0; iEmcRP < emcRecPoints->GetSize(); iEmcRP++ ) {
251 emcclu = (AliPHOSEmcRecPoint *) fPlease->GimeRecPoint(emcRecPoints->At(iEmcRP),"emc") ;
92862013 252 Bool_t toofar ;
d15a28e7 253
28c3a259 254 Int_t iPpsdLow ;
255
256 for(iPpsdLow = 0; iPpsdLow < ppsdRecPointsLow->GetSize();iPpsdLow++ ) {
257
258 ppsdlow = (AliPHOSPpsdRecPoint *) fPlease->GimeRecPoint(ppsdRecPointsLow->At(iPpsdLow),"ppsd") ;
92862013 259 Float_t r = GetDistanceInPHOSPlane(emcclu, ppsdlow, toofar) ;
d15a28e7 260
92862013 261 if(toofar)
d15a28e7 262 break ;
83974468 263 if(r < fR0){
28c3a259 264 new( (*linklowArray)[iLinkLow++]) AliPHOSLink(r, iEmcRP, iPpsdLow) ;
265 }
d15a28e7 266 }
267
28c3a259 268 Int_t iPpsdUp = 0 ;
269 for(iPpsdUp = 0; iPpsdUp < ppsdRecPointsUp->GetSize();iPpsdUp++ ) {
270
271 ppsdup = (AliPHOSPpsdRecPoint *)fPlease->GimeRecPoint(ppsdRecPointsUp->At(iPpsdUp),"ppsd") ;
92862013 272 Float_t r = GetDistanceInPHOSPlane(emcclu, ppsdup, toofar) ;
d15a28e7 273
92862013 274 if(toofar)
d15a28e7 275 break ;
83974468 276 if(r < fR0) {
28c3a259 277 new( (*linkupArray)[iLinkUp++]) AliPHOSLink(r, iEmcRP, iPpsdUp) ;
278 }
d15a28e7 279 }
28c3a259 280 }
d15a28e7 281
92862013 282 linklowArray->Sort() ; //first links with smallest distances
283 linkupArray->Sort() ;
d15a28e7 284}
28c3a259 285
d15a28e7 286//____________________________________________________________________________
28c3a259 287void AliPHOSTrackSegmentMakerv1::MakePairs(TArrayI * emcRecPoints,
288 TArrayI * ppsdRecPointsUp,
289 TArrayI * ppsdRecPointsLow,
88714635 290 TClonesArray * linklowArray,
291 TClonesArray * linkupArray,
292 AliPHOSTrackSegment::TrackSegmentsList * trsl)
6ad0bfa0 293{
294
295 // Finds the smallest links and makes pairs of PPSD and EMC clusters with smallest distance
296
92862013 297 TIter nextLow(linklowArray) ;
298 TIter nextUp(linkupArray) ;
d15a28e7 299
300 AliPHOSLink * linkLow ;
301 AliPHOSLink * linkUp ;
302
28c3a259 303 Int_t emc ;
304 Int_t ppsdLow ;
305 Int_t ppsdUp ;
d15a28e7 306
28c3a259 307 AliPHOSPpsdRecPoint * nullpointer = 0 ;
308 ppsdUp = 0 ;
9f616d61 309
d15a28e7 310 while ( (linkLow = (AliPHOSLink *)nextLow() ) ){
28c3a259 311
312 emc = emcRecPoints->At(linkLow->GetEmc()) ;
313 ppsdLow = ppsdRecPointsLow->At(linkLow->GetPpsd()) ;
314 if( (emc >= 0) && (ppsdLow >= 0) ){ // RecPoints not removed yet
315
316 new( (*trsl)[fNTrackSegments] ) AliPHOSTrackSegment((AliPHOSEmcRecPoint *)fPlease->GimeRecPoint(emc,"emc"),
317 nullpointer,
318 (AliPHOSPpsdRecPoint *)fPlease->GimeRecPoint(ppsdLow,"ppsd") ) ;
319 ((AliPHOSTrackSegment* )trsl->At(fNTrackSegments))->SetIndexInList(fNTrackSegments);
320 //replace index of emc to negative and shifted index of TS
321 emcRecPoints->AddAt(-2 - fNTrackSegments,linkLow->GetEmc()) ;
322 //replace index of emc to negative and shifted index of TS
323 ppsdRecPointsLow->AddAt(-2 - fNTrackSegments,linkLow->GetPpsd()) ;
324 fNTrackSegments++ ;
325
6ad0bfa0 326 }
d15a28e7 327 }
28c3a259 328
329 while ( (linkUp = (AliPHOSLink *)nextUp() ) ){
330 emc = emcRecPoints->At(linkUp->GetEmc()) ;
331 if(emc != -1){ //without ppsd Up yet
d15a28e7 332
28c3a259 333 ppsdUp = ppsdRecPointsUp->At(linkUp->GetPpsd()) ;
334
335 if(emc >= 0){ //without ppsd Low => create new TS
336
337 fNTrackSegments = trsl->GetEntries() ;
338 new( (*trsl)[fNTrackSegments] ) AliPHOSTrackSegment((AliPHOSEmcRecPoint *) fPlease->GimeRecPoint(emc,"emc"),
339 (AliPHOSPpsdRecPoint *)fPlease->GimeRecPoint(ppsdUp,"ppsd"),
340 nullpointer) ;
341 ((AliPHOSTrackSegment *) trsl->At(fNTrackSegments))->SetIndexInList(fNTrackSegments);
342 fNTrackSegments++ ;
343 }
344 else{ // append ppsd Up to existing TS
345 ((AliPHOSTrackSegment *)trsl->At(-2-emc))->SetPpsdUpRecPoint((AliPHOSPpsdRecPoint *)fPlease->GimeRecPoint(ppsdUp,"ppsd"));
d15a28e7 346 }
28c3a259 347 emcRecPoints->AddAt(-1,linkUp->GetEmc()) ; //Mark that PPSD Up found
d15a28e7 348
28c3a259 349 }
350 }
351
352 Int_t iEmcRP ;
353 for(iEmcRP = 0; iEmcRP <emcRecPoints->GetSize() ; iEmcRP++ ){
354 emc = emcRecPoints->At(iEmcRP) ;
355 if(emc >=0 ){
356 ppsdUp = 0;
357 ppsdLow = 0;
358 new( (*trsl)[fNTrackSegments] ) AliPHOSTrackSegment((AliPHOSEmcRecPoint *) fPlease->GimeRecPoint(emc,"emc"),
359 nullpointer, nullpointer ) ;
360 ((AliPHOSTrackSegment *) trsl->At(fNTrackSegments))->SetIndexInList(fNTrackSegments);
361 fNTrackSegments++;
d15a28e7 362 }
b2a60966 363
d15a28e7 364 }
28c3a259 365
d15a28e7 366}
367
368//____________________________________________________________________________
88714635 369void AliPHOSTrackSegmentMakerv1::MakeTrackSegments(DigitsList * dl,
370 AliPHOSRecPoint::RecPointsList * emcl,
371 AliPHOSRecPoint::RecPointsList * ppsdl,
372 AliPHOSTrackSegment::TrackSegmentsList * trsl)
d15a28e7 373{
b2a60966 374 // Makes the track segments out of the list of EMC and PPSD Recpoints and stores them in a list
98cbd830 375
92862013 376 Int_t phosmod = 1 ;
9f616d61 377 Int_t emcStopedAt = 0 ;
d15a28e7 378 Int_t ppsdStopedAt = 0 ;
28c3a259 379
380 fNTrackSegments = 0 ;
381
d15a28e7 382
28c3a259 383 TArrayI * emcRecPoints = new TArrayI(1000) ; // these arrays keep indexes
384 TArrayI * ppsdRecPointsUp = new TArrayI(1000) ; // of RecPoints, which are
385 TArrayI * ppsdRecPointsLow = new TArrayI(1000) ; // kept in TClonesArray's emcl and ppsdl
d15a28e7 386
387
92862013 388 TClonesArray * linklowArray = new TClonesArray("AliPHOSLink", 100);
389 TClonesArray * linkupArray = new TClonesArray("AliPHOSLink", 100);
6ad0bfa0 390
28c3a259 391
98cbd830 392 if(fUnfoldFlag){
69183710 393 UnfoldAll(dl, emcl) ; // Unfolds all EMC clusters
98cbd830 394 }
d15a28e7 395
28c3a259 396
98cbd830 397 while(phosmod <= fGeom->GetNModules() ){
398
399 FillOneModule(emcl, emcRecPoints, ppsdl, ppsdRecPointsUp, ppsdRecPointsLow, phosmod, emcStopedAt, ppsdStopedAt) ;
400
401 MakeLinks(emcRecPoints, ppsdRecPointsUp, ppsdRecPointsLow, linklowArray, linkupArray) ;
402
92862013 403 MakePairs(emcRecPoints, ppsdRecPointsUp, ppsdRecPointsLow, linklowArray, linkupArray, trsl) ;
98cbd830 404
28c3a259 405 emcRecPoints->Reset() ;
98cbd830 406
28c3a259 407 ppsdRecPointsUp->Reset() ;
98cbd830 408
28c3a259 409 ppsdRecPointsLow->Reset() ;
98cbd830 410
92862013 411 linkupArray->Clear() ;
98cbd830 412
92862013 413 linklowArray->Clear() ;
98cbd830 414
92862013 415 phosmod++ ;
d15a28e7 416 }
28c3a259 417
92862013 418 delete emcRecPoints ;
419 emcRecPoints = 0 ;
98cbd830 420
92862013 421 delete ppsdRecPointsUp ;
422 ppsdRecPointsUp = 0 ;
9f616d61 423
92862013 424 delete ppsdRecPointsLow ;
425 ppsdRecPointsLow = 0 ;
9f616d61 426
92862013 427 delete linkupArray ;
428 linkupArray = 0 ;
9f616d61 429
92862013 430 delete linklowArray ;
431 linklowArray = 0 ;
d15a28e7 432}
433
0336aa72 434//____________________________________________________________________________
435void AliPHOSTrackSegmentMakerv1::MakeTrackSegmentsCPV(DigitsList * dl,
69183710 436 AliPHOSRecPoint::RecPointsList * emcl,
437 AliPHOSRecPoint::RecPointsList * cpvl)
0336aa72 438{
439 // Unfold clusters in EMC and CPV and refill reconstructed point lists emcl and ppsdl
440 // Yuri Kharlov. 19 October 2000
441
442 fNTrackSegments = 0 ;
443
444 TArrayI * emcRecPoints = new TArrayI(1000) ; // these arrays keep indexes
445 TArrayI * cpvRecPoints = new TArrayI(1000) ; // of RecPoints, which are kept in emcl and ppsdl
446
447 if(fUnfoldFlag){
448 UnfoldAll(dl, emcl) ; // Unfolds all EMC clusters
449 UnfoldAll(dl, cpvl) ; // Unfolds all CPV clusters
450 }
451
452// Int_t phosmod = 1 ;
453// Int_t emcStopedAt = 0 ;
454// Int_t cpvStopedAt = 0 ;
455// while(phosmod <= fGeom->GetNModules() ){
456// FillOneModule(emcl, emcRecPoints, ppsdl, cpvRecPoints, phosmod, emcStopedAt, cpvStopedAt) ;
457// emcRecPoints->Reset() ;
458// cpvRecPoints->Reset() ;
459// phosmod++ ;
460// }
461
462 delete emcRecPoints ; emcRecPoints = 0 ;
463 delete cpvRecPoints ; cpvRecPoints = 0 ;
464}
465
d15a28e7 466//____________________________________________________________________________
9f616d61 467Double_t AliPHOSTrackSegmentMakerv1::ShowerShape(Double_t r)
468{
b2a60966 469 // Shape of the shower (see PHOS TDR)
470 // If you change this function, change also the gradien evaluation in ChiSquare()
471
6ad0bfa0 472 Double_t r4 = r*r*r*r ;
473 Double_t r295 = TMath::Power(r, 2.95) ;
d15a28e7 474 Double_t shape = TMath::Exp( -r4 * (1. / (2.32 + 0.26 * r4) + 0.0316 / (1 + 0.0652 * r295) ) ) ;
475 return shape ;
476}
477
98cbd830 478//____________________________________________________________________________
479void AliPHOSTrackSegmentMakerv1::UnfoldAll(DigitsList * dl, AliPHOSRecPoint::RecPointsList * emcIn)
480{
481 // Performs unfolding of all EMC clusters, sorts them and resets indexes in RecPoints
482
483 AliPHOSEmcRecPoint * emcRecPoint ;
484 Int_t index ;
485 Int_t nEmcUnfolded = emcIn->GetEntries() ;
486
487 for(index = 0 ; index < nEmcUnfolded; index++){
488
69183710 489 emcRecPoint = (AliPHOSEmcRecPoint *) emcIn->At(index) ;
98cbd830 490
69183710 491 Int_t nMultipl = emcRecPoint->GetMultiplicity() ;
492 Int_t * maxAt = new Int_t[nMultipl] ;
98cbd830 493 Float_t * maxAtEnergy = new Float_t[nMultipl] ;
494 Int_t nMax = emcRecPoint->GetNumberOfLocalMax(maxAt, maxAtEnergy) ;
495
69183710 496 if( nMax > 1 ) { // if cluster is very flat (no pronounced maximum) then nMax = 0
98cbd830 497 UnfoldClusters(dl, emcIn, emcRecPoint, nMax, maxAt, maxAtEnergy) ;
498 emcIn->Remove(emcRecPoint);
499 emcIn->Compress() ;
500 index-- ;
501 nEmcUnfolded-- ;
502 }
503
504 delete[] maxAt ;
505 delete[] maxAtEnergy ;
506 } //Unfolding finished
507
508 emcIn->Sort() ;
509
510 // to set index to new and correct index of old RecPoints
28c3a259 511 for( index = 0 ; index < emcIn->GetEntries() ; index++){
69183710 512
98cbd830 513 ((AliPHOSEmcRecPoint *) emcIn->At(index))->SetIndexInList(index) ;
69183710 514
98cbd830 515 }
516
28c3a259 517 emcIn->Sort() ;
518
98cbd830 519}
d15a28e7 520//____________________________________________________________________________
88714635 521void AliPHOSTrackSegmentMakerv1::UnfoldClusters(DigitsList * dl,
522 AliPHOSRecPoint::RecPointsList * emcIn,
523 AliPHOSEmcRecPoint * iniEmc,
524 Int_t nMax,
525 int * maxAt,
98cbd830 526 Float_t * maxAtEnergy)
d15a28e7 527{
b2a60966 528 // Performs the unfolding of a cluster with nMax overlapping showers
529 // This is time consuming (use the (Un)SetUnfolFlag() )
530
92862013 531 Int_t nPar = 3 * nMax ;
31aa6d6c 532 Float_t * fitparameters = new Float_t[nPar] ;
98cbd830 533
9f616d61 534
92862013 535 Bool_t rv = FindFit(iniEmc, maxAt, maxAtEnergy, nPar, fitparameters) ;
31aa6d6c 536 if( !rv ) {
537 // Fit failed, return and remove cluster
538 delete[] fitparameters ;
d15a28e7 539 return ;
31aa6d6c 540 }
98cbd830 541
d15a28e7 542 Float_t xDigit ;
543 Float_t zDigit ;
92862013 544 Int_t relid[4] ;
d15a28e7 545
92862013 546 Int_t nDigits = iniEmc->GetMultiplicity() ;
d15a28e7 547 Float_t xpar ;
548 Float_t zpar ;
92862013 549 Float_t epar ;
550 Float_t distance ;
551 Float_t ratio ;
31aa6d6c 552 Float_t * efit = new Float_t[nDigits] ;
d15a28e7 553 Int_t iparam ;
554 Int_t iDigit ;
555
556 AliPHOSDigit * digit ;
557 AliPHOSEmcRecPoint * emcRP ;
6ad0bfa0 558 Int_t * emcDigits = iniEmc->GetDigitsList() ;
d15a28e7 559 Float_t * emcEnergies = iniEmc->GetEnergiesList() ;
560
561 Int_t iRecPoint = emcIn->GetEntries() ;
562
92862013 563 for(iDigit = 0 ; iDigit < nDigits ; iDigit ++){
98cbd830 564 digit = fPlease->GimeDigit( emcDigits[iDigit] ) ;
565 fGeom->AbsToRelNumbering(digit->GetId(), relid) ;
566 fGeom->RelPosInModule(relid, xDigit, zDigit) ;
92862013 567 efit[iDigit] = 0;
d15a28e7 568 iparam = 0 ;
569
92862013 570 while(iparam < nPar ){
571 xpar = fitparameters[iparam] ;
572 zpar = fitparameters[iparam+1] ;
573 epar = fitparameters[iparam+2] ;
d15a28e7 574 iparam += 3 ;
92862013 575 distance = (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar) ;
576 distance = TMath::Sqrt(distance) ;
b2a60966 577 efit[iDigit] += epar * AliPHOSTrackSegmentMakerv1::ShowerShape(distance) ;
d15a28e7 578 }
d15a28e7 579 }
580
581 iparam = 0 ;
582 Float_t eDigit ;
583
98cbd830 584
92862013 585 while(iparam < nPar ){
586 xpar = fitparameters[iparam] ;
587 zpar = fitparameters[iparam+1] ;
588 epar = fitparameters[iparam+2] ;
d15a28e7 589 iparam += 3 ;
98cbd830 590
28c3a259 591 if(iRecPoint >= emcIn->GetSize())
592 emcIn->Expand(2*iRecPoint) ;
593 (*emcIn)[iRecPoint] = new AliPHOSEmcRecPoint( iniEmc->GetLogWeightCut(), iniEmc->GetLocMaxCut() ) ;
594
98cbd830 595 emcRP = (AliPHOSEmcRecPoint *) emcIn->At(iRecPoint);
596 iRecPoint++ ;
d15a28e7 597
92862013 598 for(iDigit = 0 ; iDigit < nDigits ; iDigit ++){
98cbd830 599 digit = fPlease->GimeDigit( emcDigits[iDigit] ) ;
600 fGeom->AbsToRelNumbering(digit->GetId(), relid) ;
601 fGeom->RelPosInModule(relid, xDigit, zDigit) ;
92862013 602 distance = (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar) ;
603 distance = TMath::Sqrt(distance) ;
b2a60966 604 ratio = epar * AliPHOSTrackSegmentMakerv1::ShowerShape(distance) / efit[iDigit] ;
92862013 605 eDigit = emcEnergies[iDigit] * ratio ;
9f616d61 606 emcRP->AddDigit( *digit, eDigit ) ;
d15a28e7 607 }
608
d15a28e7 609 }
31aa6d6c 610
611 delete[] fitparameters ;
612 delete[] efit ;
613
d15a28e7 614}
6ad0bfa0 615
d15a28e7 616//______________________________________________________________________________
92862013 617void UnfoldingChiSquare(Int_t & nPar, Double_t * Grad, Double_t & fret, Double_t * x, Int_t iflag)
d15a28e7 618{
b2a60966 619 // Calculates th Chi square for the cluster unfolding minimization
620 // Number of parameters, Gradient, Chi squared, parameters, what to do
98cbd830 621
9f616d61 622 AliPHOSEmcRecPoint * emcRP = (AliPHOSEmcRecPoint *) gMinuit->GetObjectFit() ; // EmcRecPoint to fit
98cbd830 623
6ad0bfa0 624 Int_t * emcDigits = emcRP->GetDigitsList() ;
98cbd830 625
626 Int_t nOfDigits = emcRP->GetDigitsMultiplicity() ;
627
d15a28e7 628 Float_t * emcEnergies = emcRP->GetEnergiesList() ;
98cbd830 629
630 AliPHOSGeometry * geom = AliPHOSGeometry::GetInstance() ;
631
632 AliPHOSIndexToObject * please = AliPHOSIndexToObject::GetInstance() ;
633
d15a28e7 634 fret = 0. ;
635 Int_t iparam ;
636
9f616d61 637 if(iflag == 2)
92862013 638 for(iparam = 0 ; iparam < nPar ; iparam++)
9f616d61 639 Grad[iparam] = 0 ; // Will evaluate gradient
640
98cbd830 641 Double_t efit ;
642
d15a28e7 643 AliPHOSDigit * digit ;
98cbd830 644 Int_t iDigit ;
645
646 for( iDigit = 0 ; iDigit < nOfDigits ; iDigit++) {
647
648 digit = please->GimeDigit( emcDigits[iDigit] ) ;
9f616d61 649
92862013 650 Int_t relid[4] ;
d15a28e7 651 Float_t xDigit ;
652 Float_t zDigit ;
98cbd830 653
92862013 654 geom->AbsToRelNumbering(digit->GetId(), relid) ;
98cbd830 655
92862013 656 geom->RelPosInModule(relid, xDigit, zDigit) ;
98cbd830 657
9f616d61 658 if(iflag == 2){ // calculate gradient
659 Int_t iParam = 0 ;
92862013 660 efit = 0 ;
661 while(iParam < nPar ){
662 Double_t distance = (xDigit - x[iParam]) * (xDigit - x[iParam]) ;
9f616d61 663 iParam++ ;
92862013 664 distance += (zDigit - x[iParam]) * (zDigit - x[iParam]) ;
665 distance = TMath::Sqrt( distance ) ;
9f616d61 666 iParam++ ;
92862013 667 efit += x[iParam] * AliPHOSTrackSegmentMakerv1::ShowerShape(distance) ;
9f616d61 668 iParam++ ;
669 }
92862013 670 Double_t sum = 2. * (efit - emcEnergies[iDigit]) / emcEnergies[iDigit] ; // Here we assume, that sigma = sqrt(E)
9f616d61 671 iParam = 0 ;
92862013 672 while(iParam < nPar ){
9f616d61 673 Double_t xpar = x[iParam] ;
674 Double_t zpar = x[iParam+1] ;
92862013 675 Double_t epar = x[iParam+2] ;
9f616d61 676 Double_t dr = TMath::Sqrt( (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar) );
677 Double_t shape = sum * AliPHOSTrackSegmentMakerv1::ShowerShape(dr) ;
678 Double_t r4 = dr*dr*dr*dr ;
679 Double_t r295 = TMath::Power(dr,2.95) ;
680 Double_t deriv =-4. * dr*dr * ( 2.32 / ( (2.32 + 0.26 * r4) * (2.32 + 0.26 * r4) ) +
681 0.0316 * (1. + 0.0171 * r295) / ( ( 1. + 0.0652 * r295) * (1. + 0.0652 * r295) ) ) ;
682
92862013 683 Grad[iParam] += epar * shape * deriv * (xpar - xDigit) ; // Derivative over x
9f616d61 684 iParam++ ;
92862013 685 Grad[iParam] += epar * shape * deriv * (zpar - zDigit) ; // Derivative over z
9f616d61 686 iParam++ ;
687 Grad[iParam] += shape ; // Derivative over energy
688 iParam++ ;
689 }
690 }
92862013 691 efit = 0;
9f616d61 692 iparam = 0 ;
98cbd830 693
92862013 694 while(iparam < nPar ){
9f616d61 695 Double_t xpar = x[iparam] ;
696 Double_t zpar = x[iparam+1] ;
92862013 697 Double_t epar = x[iparam+2] ;
9f616d61 698 iparam += 3 ;
92862013 699 Double_t distance = (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar) ;
700 distance = TMath::Sqrt(distance) ;
701 efit += epar * AliPHOSTrackSegmentMakerv1::ShowerShape(distance) ;
9f616d61 702 }
98cbd830 703
92862013 704 fret += (efit-emcEnergies[iDigit])*(efit-emcEnergies[iDigit])/emcEnergies[iDigit] ;
9f616d61 705 // Here we assume, that sigma = sqrt(E)
d15a28e7 706 }
98cbd830 707
d15a28e7 708}