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