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