The emc and ppsd multiplicity was not reinitialized from event to the next causing...
[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 = 4. ;   
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                                                 TObjArray * emcOut, 
151                                                 AliPHOSRecPoint::RecPointsList * ppsdIn, 
152                                                 TObjArray * ppsdOutUp,
153                                                 TObjArray * 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   for(index = emcStopedAt; index < nEmcUnfolded; index++){
166
167     emcRecPoint = (AliPHOSEmcRecPoint *) emcIn->At(index) ;
168     
169     if(emcRecPoint->GetPHOSMod() != phosmod )  
170       break ;
171     
172     emcOut->Add(emcRecPoint) ;
173   }
174   emcStopedAt = index ;
175
176   for(index = ppsdStopedAt; index < ppsdIn->GetEntries(); index++){
177     ppsdRecPoint = (AliPHOSPpsdRecPoint *) ppsdIn->At(index) ;
178     if(ppsdRecPoint->GetPHOSMod() != phosmod )   
179       break ;
180     if(ppsdRecPoint->GetUp() ) 
181       ppsdOutUp->Add(ppsdRecPoint) ;
182     else  
183       ppsdOutLow->Add(ppsdRecPoint) ;
184   }
185   ppsdStopedAt = index ;
186    
187   emcOut->Sort() ;
188   ppsdOutUp->Sort() ;
189   ppsdOutLow->Sort() ;   
190 }
191 //____________________________________________________________________________
192 Float_t  AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcclu,AliPHOSPpsdRecPoint * PpsdClu, Bool_t &toofar)
193 {
194   // Calculates the distance between the EMC RecPoint and the PPSD RecPoint
195  
196   Float_t r = fR0 ;
197  
198   TVector3 vecEmc ;
199   TVector3 vecPpsd ;
200   
201   emcclu->GetLocalPosition(vecEmc) ;
202   PpsdClu->GetLocalPosition(vecPpsd)  ; 
203   if(emcclu->GetPHOSMod() == PpsdClu->GetPHOSMod()){ 
204     if(vecPpsd.X() >= vecEmc.X() - fDelta ){ 
205       if(vecPpsd.Z() >= vecEmc.Z() - fDelta ){
206         // Correct to difference in CPV and EMC position due to different distance to center.
207         // we assume, that particle moves from center
208         Float_t dCPV = fGeom->GetIPtoOuterCoverDistance();
209         Float_t dEMC = fGeom->GetIPtoCrystalSurface() ;
210         dEMC         = dEMC / dCPV ;
211         vecPpsd = dEMC * vecPpsd  - vecEmc ; 
212         r = vecPpsd.Mag() ;
213       } // if  zPpsd >= zEmc - fDelta
214       toofar = kFALSE ;
215     } // if  xPpsd >= xEmc - fDelta
216     else 
217       toofar = kTRUE ;
218   } 
219   else 
220     toofar = kTRUE ;
221   
222   return r ;
223 }
224
225 //____________________________________________________________________________
226 void  AliPHOSTrackSegmentMakerv1::MakeLinks(TObjArray * emcRecPoints, TObjArray * ppsdRecPointsUp, 
227                                      TObjArray * ppsdRecPointsLow, TClonesArray * linklowArray, 
228                                      TClonesArray *linkupArray) 
229
230   // Finds distances (links) between all EMC and PPSD clusters, which are not further apart from each other than fR0 
231   
232   TIter nextEmc(emcRecPoints) ;
233   Int_t iEmcClu = 0 ; 
234   
235   AliPHOSPpsdRecPoint * ppsdlow ; 
236   AliPHOSPpsdRecPoint * ppsdup ;
237   AliPHOSEmcRecPoint * emcclu ;
238   
239   Int_t iLinkLow = 0 ;
240   Int_t iLinkUp  = 0 ;
241   
242   while( (emcclu = (AliPHOSEmcRecPoint*)nextEmc() ) ) {
243     Bool_t toofar ;
244     TIter nextPpsdLow(ppsdRecPointsLow ) ;
245     Int_t iPpsdLow = 0 ;
246     
247     while( (ppsdlow = (AliPHOSPpsdRecPoint*)nextPpsdLow() ) ) { 
248       Float_t r = GetDistanceInPHOSPlane(emcclu, ppsdlow, toofar) ;
249       
250       if(toofar) 
251         break ;  
252       if(r < fR0){
253         new( (*linklowArray)[iLinkLow++]) AliPHOSLink(r, iEmcClu, iPpsdLow) ;
254      }
255       iPpsdLow++ ;
256       
257     }
258     
259     TIter nextPpsdUp(ppsdRecPointsUp ) ;
260     Int_t iPpsdUp = 0 ;
261     
262     while( (ppsdup = (AliPHOSPpsdRecPoint*)nextPpsdUp() ) ) { 
263       Float_t r = GetDistanceInPHOSPlane(emcclu, ppsdup, toofar) ;
264       
265       if(toofar)
266         break ;  
267       if(r < fR0) { 
268         new( (*linkupArray)[iLinkUp++]) AliPHOSLink(r, iEmcClu, iPpsdUp) ;
269       }
270       iPpsdUp++ ;
271       
272     }
273     
274     iEmcClu++ ; 
275     
276   } // while nextEmC
277   
278   linklowArray->Sort() ; //first links with smallest distances
279   linkupArray->Sort() ;
280 }
281     
282 //____________________________________________________________________________
283 void  AliPHOSTrackSegmentMakerv1::MakePairs(TObjArray * emcRecPoints, 
284                                             TObjArray * ppsdRecPointsUp, 
285                                             TObjArray * ppsdRecPointsLow, 
286                                             TClonesArray * linklowArray, 
287                                             TClonesArray * linkupArray, 
288                                             AliPHOSTrackSegment::TrackSegmentsList * trsl) 
289
290
291   // Finds the smallest links and makes pairs of PPSD and EMC clusters with smallest distance 
292   
293   TIter nextLow(linklowArray) ;
294   TIter nextUp(linkupArray) ;
295   
296   AliPHOSLink * linkLow ;
297   AliPHOSLink * linkUp ;
298
299   AliPHOSEmcRecPoint * emc ;
300   AliPHOSPpsdRecPoint * ppsdLow ;
301   AliPHOSPpsdRecPoint * ppsdUp ;
302
303   AliPHOSRecPoint * nullpointer = 0 ;
304
305   while ( (linkLow =  (AliPHOSLink *)nextLow() ) ){
306     emc = (AliPHOSEmcRecPoint *) emcRecPoints->At(linkLow->GetEmc()) ;
307     ppsdLow = (AliPHOSPpsdRecPoint *) ppsdRecPointsLow->At(linkLow->GetPpsd()) ;
308     if( (emc) && (ppsdLow) ){ // RecPoints not removed yet 
309          ppsdUp = 0 ;
310          
311          while ( (linkUp =  (AliPHOSLink *)nextUp() ) ){  
312            if(linkLow->GetEmc() == linkUp->GetEmc() ){
313              ppsdUp = (AliPHOSPpsdRecPoint *) ppsdRecPointsUp->At(linkUp->GetPpsd()) ;
314              break ;
315            }
316         
317          } // while nextUp
318          
319          nextUp.Reset();
320 //          AliPHOSTrackSegment * subtr = new AliPHOSTrackSegment(emc, ppsdUp, ppsdLow ) ;
321 //       trsl->Add(subtr) ; 
322          fNTrackSegments = trsl->GetEntries() ; 
323          new( (*trsl)[fNTrackSegments] ) AliPHOSTrackSegment(emc, ppsdUp, ppsdLow ) ;
324          fNTrackSegments++ ;
325          emcRecPoints->AddAt(nullpointer,linkLow->GetEmc()) ;     
326          ppsdRecPointsLow->AddAt(nullpointer,linkLow->GetPpsd()) ;
327          
328          if(ppsdUp)  
329            ppsdRecPointsUp->AddAt(nullpointer,linkUp->GetPpsd()) ;
330          
331     } 
332   } 
333    
334   TIter nextEmc(emcRecPoints) ;          
335   nextEmc.Reset() ;
336
337   while( (emc = (AliPHOSEmcRecPoint*)nextEmc()) ){ //to create pairs if no ppsdlow
338     ppsdLow = 0 ; 
339     ppsdUp  = 0 ;
340     
341     while ( (linkUp =  (AliPHOSLink *)nextUp() ) ){
342       
343       if(emcRecPoints->IndexOf(emc) == linkUp->GetEmc() ){
344         ppsdUp = (AliPHOSPpsdRecPoint *) ppsdRecPointsUp->At(linkUp->GetPpsd()) ;
345         break ;
346       }
347       
348     }
349
350     fNTrackSegments = trsl->GetEntries() ; 
351     new( (*trsl)[fNTrackSegments] ) AliPHOSTrackSegment(emc, ppsdUp, ppsdLow ) ;
352     fNTrackSegments++ ;
353     
354
355     if(ppsdUp)  
356       ppsdRecPointsUp->AddAt(nullpointer,linkUp->GetPpsd()) ;
357   }
358      
359 }
360
361 //____________________________________________________________________________
362 void  AliPHOSTrackSegmentMakerv1::MakeTrackSegments(DigitsList * dl, 
363                                                     AliPHOSRecPoint::RecPointsList * emcl, 
364                                                     AliPHOSRecPoint::RecPointsList * ppsdl, 
365                                                     AliPHOSTrackSegment::TrackSegmentsList * trsl)
366 {
367   // Makes the track segments out of the list of EMC and PPSD Recpoints and stores them in a list
368   
369   Int_t phosmod      = 1 ;
370   Int_t emcStopedAt  = 0 ; 
371   Int_t ppsdStopedAt = 0 ; 
372   
373   TObjArray * emcRecPoints     = new TObjArray(100) ;  // these arrays keep pointers 
374   TObjArray * ppsdRecPointsUp  = new TObjArray(100) ;  // to RecPoints, which are 
375   TObjArray * ppsdRecPointsLow = new TObjArray(100) ;  // kept in TClonesArray's emcl and ppsdl
376   
377   
378   TClonesArray * linklowArray = new TClonesArray("AliPHOSLink", 100);
379   TClonesArray * linkupArray  = new TClonesArray("AliPHOSLink", 100); 
380
381   if(fUnfoldFlag){
382     UnfoldAll(dl, emcl) ; // Unfolds all EMC clusters
383   }
384
385   while(phosmod <= fGeom->GetNModules() ){
386     
387     FillOneModule(emcl, emcRecPoints, ppsdl, ppsdRecPointsUp, ppsdRecPointsLow, phosmod, emcStopedAt, ppsdStopedAt) ;
388     
389     MakeLinks(emcRecPoints, ppsdRecPointsUp, ppsdRecPointsLow, linklowArray, linkupArray) ; 
390     
391     MakePairs(emcRecPoints, ppsdRecPointsUp, ppsdRecPointsLow, linklowArray, linkupArray, trsl) ;
392     
393     emcRecPoints->Clear() ;
394     
395     ppsdRecPointsUp->Clear() ;
396     
397     ppsdRecPointsLow->Clear() ;
398     
399     linkupArray->Clear() ;
400     
401     linklowArray->Clear() ;
402     
403     phosmod++ ; 
404   }
405   delete emcRecPoints ; 
406   emcRecPoints = 0 ; 
407   
408   delete ppsdRecPointsUp ; 
409   ppsdRecPointsUp = 0 ; 
410
411   delete ppsdRecPointsLow ; 
412   ppsdRecPointsLow = 0 ; 
413
414   delete linkupArray ; 
415   linkupArray = 0  ; 
416
417   delete linklowArray ; 
418   linklowArray = 0 ; 
419 }
420
421 //____________________________________________________________________________
422 Double_t  AliPHOSTrackSegmentMakerv1::ShowerShape(Double_t r)
423
424   // Shape of the shower (see PHOS TDR)
425   // If you change this function, change also the gradien evaluation  in ChiSquare()
426
427   Double_t r4    = r*r*r*r ;
428   Double_t r295  = TMath::Power(r, 2.95) ;
429   Double_t shape = TMath::Exp( -r4 * (1. / (2.32 + 0.26 * r4) + 0.0316 / (1 + 0.0652 * r295) ) ) ;
430   return shape ;
431 }
432
433 //____________________________________________________________________________
434 void  AliPHOSTrackSegmentMakerv1::UnfoldAll(DigitsList * dl, AliPHOSRecPoint::RecPointsList * emcIn) 
435 {
436   // Performs unfolding of all EMC clusters, sorts them and resets indexes in RecPoints
437
438   AliPHOSEmcRecPoint *  emcRecPoint  ; 
439   Int_t index ;
440   Int_t nEmcUnfolded = emcIn->GetEntries() ;
441   
442   for(index = 0 ; index < nEmcUnfolded; index++){
443
444     emcRecPoint = (AliPHOSEmcRecPoint *) emcIn->At(index) ;
445     
446     Int_t nMultipl = emcRecPoint->GetMultiplicity() ; 
447     Int_t * maxAt = new Int_t[nMultipl] ;
448     Float_t * maxAtEnergy = new Float_t[nMultipl] ;
449     Int_t nMax = emcRecPoint->GetNumberOfLocalMax(maxAt, maxAtEnergy) ;
450     
451     if( nMax > 1 ) {     // if cluster is very flat (no pronounced maximum) then nMax = 0 
452       
453       UnfoldClusters(dl, emcIn, emcRecPoint, nMax, maxAt, maxAtEnergy) ;
454       emcIn->Remove(emcRecPoint); 
455       emcIn->Compress() ;
456       index-- ;
457       nEmcUnfolded-- ;
458     }
459     
460     delete[] maxAt ; 
461     delete[] maxAtEnergy ; 
462   } //Unfolding finished
463
464   emcIn->Sort() ;
465
466   // to set index to new and correct index of old RecPoints
467   for( index = 0 ; index < emcIn->GetEntriesFast() ; index++){
468     
469     ((AliPHOSEmcRecPoint *) emcIn->At(index))->SetIndexInList(index) ;   
470     
471   }
472
473 }
474 //____________________________________________________________________________
475 void  AliPHOSTrackSegmentMakerv1::UnfoldClusters(DigitsList * dl, 
476                                                  AliPHOSRecPoint::RecPointsList * emcIn, 
477                                                  AliPHOSEmcRecPoint * iniEmc, 
478                                                  Int_t nMax, 
479                                                  int * maxAt, 
480                                                  Float_t * maxAtEnergy)
481
482   // Performs the unfolding of a cluster with nMax overlapping showers 
483   // This is time consuming (use the (Un)SetUnfolFlag()  )
484
485   Int_t nPar = 3 * nMax ;
486   Float_t * fitparameters = new Float_t[nPar] ;
487
488
489   Bool_t rv = FindFit(iniEmc, maxAt, maxAtEnergy, nPar, fitparameters) ;
490   if( !rv ) {
491     // Fit failed, return and remove cluster
492     delete[] fitparameters ; 
493     return ;
494   }
495
496
497   Float_t xDigit ;
498   Float_t zDigit ;
499   Int_t relid[4] ;
500
501   Int_t nDigits = iniEmc->GetMultiplicity() ;  
502   Float_t xpar  ;
503   Float_t zpar  ;
504   Float_t epar  ;
505   Float_t distance ;
506   Float_t ratio ;
507   Float_t * efit = new Float_t[nDigits] ;
508   Int_t iparam ;
509   Int_t iDigit ;
510   
511   AliPHOSDigit * digit ;
512   AliPHOSEmcRecPoint * emcRP ;  
513   Int_t * emcDigits = iniEmc->GetDigitsList() ;
514   Float_t * emcEnergies = iniEmc->GetEnergiesList() ;
515
516   Int_t iRecPoint = emcIn->GetEntries() ;
517
518   for(iDigit = 0 ; iDigit < nDigits ; iDigit ++){
519     digit = fPlease->GimeDigit( emcDigits[iDigit] ) ;   
520     fGeom->AbsToRelNumbering(digit->GetId(), relid) ;
521     fGeom->RelPosInModule(relid, xDigit, zDigit) ;
522     efit[iDigit] = 0;
523     iparam = 0 ;
524     
525     while(iparam < nPar ){
526       xpar = fitparameters[iparam] ;
527       zpar = fitparameters[iparam+1] ;
528       epar = fitparameters[iparam+2] ;
529       iparam += 3 ;
530       distance = (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar)  ;
531       distance =  TMath::Sqrt(distance) ;
532       efit[iDigit] += epar * AliPHOSTrackSegmentMakerv1::ShowerShape(distance) ;
533     }
534   }
535
536   iparam = 0 ;
537   Float_t eDigit ;
538
539
540   while(iparam < nPar ){
541     xpar = fitparameters[iparam] ;
542     zpar = fitparameters[iparam+1] ;
543     epar = fitparameters[iparam+2] ;
544     iparam += 3 ;
545
546    (*emcIn)[iRecPoint] = new AliPHOSEmcRecPoint( iniEmc->GetLogWeightCut(), iniEmc->GetLocMaxCut() ) ;
547
548     emcRP = (AliPHOSEmcRecPoint *) emcIn->At(iRecPoint);
549     iRecPoint++ ;
550
551     for(iDigit = 0 ; iDigit < nDigits ; iDigit ++){
552       digit = fPlease->GimeDigit( emcDigits[iDigit] ) ; 
553       fGeom->AbsToRelNumbering(digit->GetId(), relid) ;
554       fGeom->RelPosInModule(relid, xDigit, zDigit) ;
555       distance = (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar)  ;
556       distance =  TMath::Sqrt(distance) ;
557       ratio = epar * AliPHOSTrackSegmentMakerv1::ShowerShape(distance) / efit[iDigit] ; 
558       eDigit = emcEnergies[iDigit] * ratio ;
559       emcRP->AddDigit( *digit, eDigit ) ;
560     }
561
562   }
563   
564   delete[] fitparameters ; 
565   delete[] efit ; 
566
567 }
568
569 //______________________________________________________________________________
570 void UnfoldingChiSquare(Int_t & nPar, Double_t * Grad, Double_t & fret, Double_t * x, Int_t iflag)
571 {
572   // Calculates th Chi square for the cluster unfolding minimization
573   // Number of parameters, Gradient, Chi squared, parameters, what to do
574   
575   AliPHOSEmcRecPoint * emcRP = (AliPHOSEmcRecPoint *) gMinuit->GetObjectFit() ; // EmcRecPoint to fit
576
577   Int_t * emcDigits     = emcRP->GetDigitsList() ;
578
579   Int_t nOfDigits = emcRP->GetDigitsMultiplicity() ; 
580
581   Float_t * emcEnergies = emcRP->GetEnergiesList() ;
582
583   AliPHOSGeometry * geom = AliPHOSGeometry::GetInstance() ;
584
585   AliPHOSIndexToObject * please = AliPHOSIndexToObject::GetInstance() ;
586
587   fret = 0. ;     
588   Int_t iparam ;
589
590   if(iflag == 2)
591     for(iparam = 0 ; iparam < nPar ; iparam++)    
592       Grad[iparam] = 0 ; // Will evaluate gradient
593   
594   Double_t efit ;    
595
596   AliPHOSDigit * digit ;
597   Int_t iDigit ;
598
599   for( iDigit = 0 ; iDigit < nOfDigits ; iDigit++) {
600
601     digit = please->GimeDigit( emcDigits[iDigit] ) ; 
602
603     Int_t relid[4] ;
604     Float_t xDigit ;
605     Float_t zDigit ;
606
607     geom->AbsToRelNumbering(digit->GetId(), relid) ;
608
609     geom->RelPosInModule(relid, xDigit, zDigit) ;
610
611      if(iflag == 2){  // calculate gradient
612        Int_t iParam = 0 ;
613        efit = 0 ;
614        while(iParam < nPar ){
615          Double_t distance = (xDigit - x[iParam]) * (xDigit - x[iParam]) ;
616          iParam++ ; 
617          distance += (zDigit - x[iParam]) * (zDigit - x[iParam]) ; 
618          distance = TMath::Sqrt( distance ) ; 
619          iParam++ ;      
620          efit += x[iParam] * AliPHOSTrackSegmentMakerv1::ShowerShape(distance) ;
621          iParam++ ;
622        }
623        Double_t sum = 2. * (efit - emcEnergies[iDigit]) / emcEnergies[iDigit] ; // Here we assume, that sigma = sqrt(E) 
624        iParam = 0 ;
625        while(iParam < nPar ){
626          Double_t xpar = x[iParam] ;
627          Double_t zpar = x[iParam+1] ;
628          Double_t epar = x[iParam+2] ;
629          Double_t dr = TMath::Sqrt( (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar) );
630          Double_t shape = sum * AliPHOSTrackSegmentMakerv1::ShowerShape(dr) ;
631          Double_t r4 = dr*dr*dr*dr ;
632          Double_t r295 = TMath::Power(dr,2.95) ;
633          Double_t deriv =-4. * dr*dr * ( 2.32 / ( (2.32 + 0.26 * r4) * (2.32 + 0.26 * r4) ) +
634                                          0.0316 * (1. + 0.0171 * r295) / ( ( 1. + 0.0652 * r295) * (1. + 0.0652 * r295) ) ) ;
635          
636          Grad[iParam] += epar * shape * deriv * (xpar - xDigit) ;  // Derivative over x    
637          iParam++ ; 
638          Grad[iParam] += epar * shape * deriv * (zpar - zDigit) ;  // Derivative over z         
639          iParam++ ; 
640          Grad[iParam] += shape ;                                  // Derivative over energy             
641          iParam++ ; 
642        }
643      }
644      efit = 0;
645      iparam = 0 ;
646
647      while(iparam < nPar ){
648        Double_t xpar = x[iparam] ;
649        Double_t zpar = x[iparam+1] ;
650        Double_t epar = x[iparam+2] ;
651        iparam += 3 ;
652        Double_t distance = (xDigit - xpar) * (xDigit - xpar) + (zDigit - zpar) * (zDigit - zpar)  ;
653        distance =  TMath::Sqrt(distance) ;
654        efit += epar * AliPHOSTrackSegmentMakerv1::ShowerShape(distance) ;
655      }
656
657      fret += (efit-emcEnergies[iDigit])*(efit-emcEnergies[iDigit])/emcEnergies[iDigit] ; 
658      // Here we assume, that sigma = sqrt(E)
659   }
660
661 }