]>
Commit | Line | Data |
---|---|---|
f7d0ec90 | 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 | ||
17 | /////////////////////////////////////////////////////////////////////////////// | |
18 | // // | |
19 | // PHOS tender, recalibrate PHOS clusters // | |
20 | // and do track matching // | |
21 | // Author : Dmitri Peressounko (RRC KI) // | |
22 | // // | |
23 | /////////////////////////////////////////////////////////////////////////////// | |
24 | ||
25 | #include "TROOT.h" | |
26 | #include "TH2.h" | |
27 | #include "TFile.h" | |
28 | ||
29 | #include <AliLog.h> | |
30 | #include <AliVEvent.h> | |
31 | #include <AliAODEvent.h> | |
32 | #include <AliESDEvent.h> | |
33 | #include <AliAnalysisManager.h> | |
34 | #include <AliTender.h> | |
35 | #include <AliCDBManager.h> | |
36 | #include "AliMagF.h" | |
37 | #include "TGeoGlobalMagField.h" | |
38 | ||
39 | #include "AliVCluster.h" | |
40 | #include "AliPHOSTenderSupply.h" | |
41 | #include "AliPHOSCalibData.h" | |
42 | #include "AliPHOSGeometry.h" | |
43 | #include "AliPHOSEsdCluster.h" | |
44 | #include "AliPHOSAodCluster.h" | |
45 | #include "AliOADBContainer.h" | |
46 | #include "AliAODCaloCells.h" | |
47 | #include "AliESDCaloCells.h" | |
48 | ||
49 | ClassImp(AliPHOSTenderSupply) | |
50 | ||
51 | AliPHOSTenderSupply::AliPHOSTenderSupply() : | |
52 | AliTenderSupply() | |
53 | ,fOCDBpass("local://OCDB") | |
54 | ,fNonlinearityVersion("Default") | |
55 | ,fPHOSGeo(0x0) | |
56 | ,fRecoPass(-1) //to be defined | |
57 | ,fUsePrivateBadMap(0) | |
58 | ,fUsePrivateCalib(0) | |
59 | ,fPHOSCalibData(0x0) | |
60 | ,fTask(0x0) | |
61 | ,fIsMC(kFALSE) | |
62 | ,fMCProduction("") | |
63 | { | |
64 | // | |
65 | // default ctor | |
66 | // | |
67 | for(Int_t i=0;i<10;i++)fNonlinearityParams[i]=0. ; | |
68 | for(Int_t mod=0;mod<5;mod++)fPHOSBadMap[mod]=0x0 ; | |
69 | } | |
70 | ||
71 | //_____________________________________________________ | |
72 | AliPHOSTenderSupply::AliPHOSTenderSupply(const char *name, const AliTender *tender) : | |
73 | AliTenderSupply(name,tender) | |
74 | ,fOCDBpass("alien:///alice/cern.ch/user/p/prsnko/PHOSrecalibrations/") | |
75 | ,fNonlinearityVersion("Default") | |
76 | ,fPHOSGeo(0x0) | |
77 | ,fRecoPass(-1) //to be defined | |
78 | ,fUsePrivateBadMap(0) | |
79 | ,fUsePrivateCalib(0) | |
80 | ,fPHOSCalibData(0x0) | |
81 | ,fTask(0x0) | |
82 | ,fIsMC(kFALSE) | |
83 | ,fMCProduction("") | |
84 | { | |
85 | // | |
86 | // named ctor | |
87 | // | |
88 | for(Int_t i=0;i<10;i++)fNonlinearityParams[i]=0. ; | |
89 | for(Int_t mod=0;mod<5;mod++)fPHOSBadMap[mod]=0x0 ; | |
90 | } | |
91 | ||
92 | //_____________________________________________________ | |
93 | AliPHOSTenderSupply::~AliPHOSTenderSupply() | |
94 | { | |
95 | //Destructor | |
96 | if(fPHOSCalibData) | |
97 | delete fPHOSCalibData; | |
98 | fPHOSCalibData=0x0 ; | |
99 | } | |
100 | ||
101 | //_____________________________________________________ | |
102 | void AliPHOSTenderSupply::InitTender() | |
103 | { | |
104 | // | |
105 | // Initialise PHOS tender | |
106 | // | |
107 | Int_t runNumber = 0; | |
108 | if(fTender) | |
109 | runNumber = fTender->GetRun(); | |
110 | else{ | |
111 | if(!fTask){ | |
112 | AliError("Neither Tender not Taks was not set") ; | |
113 | return ; | |
114 | } | |
115 | AliAODEvent *aod = dynamic_cast<AliAODEvent*>(fTask->InputEvent()) ; | |
116 | if(aod) | |
117 | runNumber = aod->GetRunNumber() ; | |
118 | else{ | |
119 | AliESDEvent *esd = dynamic_cast<AliESDEvent*>(fTask->InputEvent()) ; | |
120 | if(esd) | |
121 | runNumber = esd->GetRunNumber() ; | |
122 | else{ | |
123 | AliError("Taks does not contain neither ESD nor AOD") ; | |
124 | return ; | |
125 | } | |
126 | } | |
127 | } | |
128 | ||
129 | //In MC always reco pass 1 | |
130 | if(fIsMC) | |
131 | fRecoPass=1 ; | |
132 | ||
133 | if(fRecoPass<0){ //not defined yet | |
134 | // read if from filename. | |
135 | AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); | |
136 | TTree * t = mgr->GetTree(); | |
137 | if(t){ | |
138 | TFile * f = t->GetCurrentFile() ; | |
139 | if(f){ | |
140 | TString fname(f->GetName()); | |
141 | if(fname.Contains("pass1")) | |
142 | fRecoPass=1; | |
143 | else | |
144 | if(fname.Contains("pass2")) | |
145 | fRecoPass=2; | |
146 | else | |
147 | if(fname.Contains("pass3")) | |
148 | fRecoPass=3; | |
149 | else | |
150 | if(fname.Contains("pass4")) | |
151 | fRecoPass=4; | |
152 | } | |
153 | } | |
154 | if(fRecoPass<0){ | |
155 | AliError("Can not find pass number from file name, set it manually"); | |
156 | } | |
157 | } | |
158 | ||
159 | //Init geometry | |
160 | if(!fPHOSGeo){ | |
161 | AliOADBContainer geomContainer("phosGeo"); | |
162 | geomContainer.InitFromFile("$ALICE_ROOT/OADB/PHOS/PHOSGeometry.root","PHOSRotationMatrixes"); | |
163 | TObjArray *matrixes = (TObjArray*)geomContainer.GetObject(runNumber,"PHOSRotationMatrixes"); | |
164 | fPHOSGeo = AliPHOSGeometry::GetInstance("IHEP") ; | |
165 | for(Int_t mod=0; mod<5; mod++) { | |
166 | if(!matrixes->At(mod)) continue; | |
167 | fPHOSGeo->SetMisalMatrix(((TGeoHMatrix*)matrixes->At(mod)),mod) ; | |
168 | printf(".........Adding Matrix(%d), geo=%p\n",mod,fPHOSGeo) ; | |
e1c158c2 | 169 | ((TGeoHMatrix*)matrixes->At(mod))->Print() ; |
f7d0ec90 | 170 | } |
171 | } | |
172 | ||
173 | //Init Bad channels map | |
174 | if(!fUsePrivateBadMap){ | |
175 | AliOADBContainer badmapContainer(Form("phosBadMap")); | |
176 | badmapContainer.InitFromFile("$ALICE_ROOT/OADB/PHOS/PHOSBadMaps.root","phosBadMap"); | |
177 | TObjArray *maps = (TObjArray*)badmapContainer.GetObject(runNumber,"phosBadMap"); | |
178 | if(!maps){ | |
179 | AliError(Form("Can not read Bad map for run %d. \n You may choose to use your map with ForceUsingBadMap()\n",runNumber)) ; | |
180 | } | |
181 | else{ | |
182 | AliInfo(Form("Setting PHOS bad map with name %s \n",maps->GetName())) ; | |
183 | for(Int_t mod=0; mod<5;mod++){ | |
184 | if(fPHOSBadMap[mod]) | |
185 | delete fPHOSBadMap[mod] ; | |
186 | TH2I * h = (TH2I*)maps->At(mod) ; | |
187 | if(h) | |
188 | fPHOSBadMap[mod]=new TH2I(*h) ; | |
189 | } | |
190 | } | |
191 | } | |
192 | ||
193 | if(!fUsePrivateCalib){ | |
194 | if(fIsMC){ //re/de-calibration for MC productions | |
195 | //Init recalibration | |
196 | AliOADBContainer calibContainer("phosRecalibration"); | |
197 | calibContainer.InitFromFile("$ALICE_ROOT/OADB/PHOS/PHOSMCCalibrations.root","phosRecalibration"); | |
198 | ||
199 | TObjArray *recalib = (TObjArray*)calibContainer.GetObject(runNumber,"PHOSRecalibration"); | |
200 | if(!recalib){ | |
201 | AliFatal(Form("Can not read calibrations for run %d\n. You may choose your specific calibration with ForceUsingCalibration()\n",runNumber)) ; | |
202 | } | |
203 | else{ | |
204 | //Now try to find object with proper name | |
205 | for(Int_t i=0; i<recalib->GetEntriesFast(); i++){ | |
206 | AliPHOSCalibData * tmp = (AliPHOSCalibData*)recalib->At(i) ; | |
207 | if(fMCProduction.CompareTo(tmp->GetName())==0){ | |
208 | fPHOSCalibData = tmp ; | |
209 | break ; | |
210 | } | |
211 | } | |
212 | if(!fPHOSCalibData) { | |
213 | AliFatal(Form("Can not find calibration for run %d, and name %s \n",runNumber, fMCProduction.Data())) ; | |
214 | } | |
215 | } | |
216 | ||
217 | } | |
218 | else{ //real data | |
219 | //Init recalibration | |
220 | //Check the pass1-pass2-pass3 reconstruction | |
221 | AliOADBContainer calibContainer("phosRecalibration"); | |
222 | calibContainer.InitFromFile("$ALICE_ROOT/OADB/PHOS/PHOSCalibrations.root","phosRecalibration"); | |
223 | TObjArray *recalib = (TObjArray*)calibContainer.GetObject(runNumber,"PHOSRecalibration"); | |
224 | if(!recalib){ | |
225 | AliFatal(Form("Can not read calibrations for run %d\n. You may choose your specific calibration with ForceUsingCalibration()\n",runNumber)) ; | |
226 | } | |
227 | else{ | |
228 | fPHOSCalibData = (AliPHOSCalibData*)recalib->At(fRecoPass-1) ; | |
229 | if(!fPHOSCalibData) { | |
230 | AliFatal(Form("Can not find calibration for run %d, pass %d \n",runNumber, fRecoPass)) ; | |
231 | } | |
232 | } | |
233 | } | |
234 | } | |
235 | ||
236 | } | |
237 | ||
238 | //_____________________________________________________ | |
239 | void AliPHOSTenderSupply::ProcessEvent() | |
240 | { | |
241 | //Choose PHOS clusters and recalibrate them | |
242 | //that it recalculate energy, position and distance | |
243 | //to closest track extrapolation | |
244 | ||
245 | AliESDEvent *esd = 0x0 ; | |
246 | AliAODEvent *aod = 0x0 ; | |
247 | if(fTender){ | |
248 | esd = fTender->GetEvent(); | |
249 | if(!esd) | |
250 | return ; | |
251 | } | |
252 | else{ | |
253 | if(!fTask){ | |
254 | return ; | |
255 | } | |
256 | esd = dynamic_cast<AliESDEvent*>(fTask->InputEvent()) ; | |
257 | aod = dynamic_cast<AliAODEvent*>(fTask->InputEvent()) ; | |
258 | if(!esd && !aod) | |
259 | return ; | |
260 | } | |
261 | ||
262 | if(!fPHOSCalibData | |
263 | || (fTender && fTender->RunChanged())){ //In case of Task init called automatically | |
264 | InitTender(); | |
265 | ||
266 | } | |
267 | ||
268 | TVector3 vertex ; | |
269 | if(esd){ | |
270 | const AliESDVertex *esdVertex = esd->GetPrimaryVertex(); | |
271 | vertex.SetXYZ(esdVertex->GetX(),esdVertex->GetY(),esdVertex->GetZ()); | |
272 | } | |
273 | else{//AOD | |
274 | const AliAODVertex *aodVertex = aod->GetPrimaryVertex(); | |
275 | vertex.SetXYZ(aodVertex->GetX(),aodVertex->GetY(),aodVertex->GetZ()); | |
276 | } | |
277 | if(vertex.Mag()>99.) //vertex not defined? | |
278 | vertex.SetXYZ(0.,0.,0.) ; | |
279 | ||
280 | ||
281 | //For re-calibration | |
282 | const Double_t logWeight=4.5 ; | |
283 | ||
284 | if(esd){ //To avoid multiple if in loops we made | |
285 | //almost identical pecies of code. Please apply changes to both!!! | |
286 | Int_t multClust=esd->GetNumberOfCaloClusters(); | |
287 | AliESDCaloCells * cells = esd->GetPHOSCells() ; | |
288 | ||
289 | for (Int_t i=0; i<multClust; i++) { | |
290 | AliESDCaloCluster *clu = esd->GetCaloCluster(i); | |
291 | if ( !clu->IsPHOS()) continue; | |
292 | ||
293 | Float_t position[3]; | |
294 | clu->GetPosition(position); | |
295 | TVector3 global(position) ; | |
296 | Int_t relId[4] ; | |
297 | fPHOSGeo->GlobalPos2RelId(global,relId) ; | |
298 | Int_t mod = relId[0] ; | |
299 | Int_t cellX = relId[2]; | |
300 | Int_t cellZ = relId[3] ; | |
301 | if ( !IsGoodChannel(mod,cellX,cellZ) ) { | |
302 | clu->SetE(0.) ; | |
303 | continue ; | |
304 | } | |
305 | ||
306 | //Apply re-Calibreation | |
307 | AliPHOSEsdCluster cluPHOS(*clu); | |
308 | cluPHOS.Recalibrate(fPHOSCalibData,cells); // modify the cell energies | |
309 | cluPHOS.EvalAll(logWeight,vertex); // recalculate the cluster parameters | |
310 | cluPHOS.SetE(CorrectNonlinearity(cluPHOS.E()));// Users's nonlinearity | |
311 | ||
5bd0f028 | 312 | Double_t ecore=CoreEnergy(&cluPHOS) ; |
313 | ecore=CorrectNonlinearity(ecore) ; | |
314 | ||
f7d0ec90 | 315 | //Correct Misalignment |
e1c158c2 | 316 | // CorrectPHOSMisalignment(global,mod) ; |
317 | // position[0]=global.X() ; | |
318 | // position[1]=global.Y() ; | |
319 | // position[2]=global.Z() ; | |
320 | // cluPHOS.GetPosition(position) ; | |
f7d0ec90 | 321 | |
322 | //Eval CoreDispersion | |
323 | Double_t m02=0.,m20=0.; | |
324 | EvalLambdas(&cluPHOS,m02, m20); | |
325 | clu->SetDispersion(TestLambda(clu->E(),m20,m02)) ; | |
326 | ||
327 | Float_t xyz[3]; | |
328 | cluPHOS.GetPosition(xyz); | |
329 | clu->SetPosition(xyz); //rec.point position in MARS | |
5bd0f028 | 330 | clu->SetE(cluPHOS.E()); //total particle energy |
331 | clu->SetMCEnergyFraction(ecore); //core particle energy | |
332 | ||
f7d0ec90 | 333 | // clu->SetDispersion(cluPHOS.GetDispersion()); //cluster dispersion |
334 | // ec->SetPID(rp->GetPID()) ; //array of particle identification | |
335 | clu->SetM02(cluPHOS.GetM02()) ; //second moment M2x | |
336 | clu->SetM20(cluPHOS.GetM20()) ; //second moment M2z | |
337 | Double_t dx=0.,dz=0. ; | |
338 | fPHOSGeo->GlobalPos2RelId(global,relId) ; | |
339 | TVector3 locPos; | |
340 | fPHOSGeo->Global2Local(locPos,global,mod) ; | |
341 | ||
342 | Double_t pttrack=0.; | |
343 | Int_t charge=0; | |
344 | FindTrackMatching(mod,&locPos,dx,dz,pttrack,charge) ; | |
345 | Double_t r=TestCPV(dx, dz, pttrack,charge) ; | |
346 | clu->SetTrackDistance(dx,dz); | |
347 | ||
348 | clu->SetEmcCpvDistance(r); | |
349 | clu->SetChi2(TestLambda(clu->E(),clu->GetM20(),clu->GetM02())); //not yet implemented | |
d4657394 | 350 | Double_t tof=EvalTOF(&cluPHOS,cells); |
351 | if(TMath::Abs(tof-clu->GetTOF())>100.e-9) //something wrong in cell TOF! | |
352 | tof=clu->GetTOF() ; | |
353 | clu->SetTOF(tof); | |
393b3556 | 354 | Double_t minDist=clu->GetDistanceToBadChannel() ;//Already calculated |
355 | DistanceToBadChannel(mod,&locPos,minDist); | |
356 | clu->SetDistanceToBadChannel(minDist) ; | |
f7d0ec90 | 357 | |
358 | } | |
359 | } | |
360 | else{//AOD | |
361 | Int_t multClust=aod->GetNumberOfCaloClusters(); | |
362 | AliAODCaloCells * cells = aod->GetPHOSCells() ; | |
363 | ||
364 | for (Int_t i=0; i<multClust; i++) { | |
365 | AliAODCaloCluster *clu = aod->GetCaloCluster(i); | |
366 | if ( !clu->IsPHOS()) continue; | |
367 | ||
368 | Float_t position[3]; | |
369 | clu->GetPosition(position); | |
370 | TVector3 global(position) ; | |
371 | Int_t relId[4] ; | |
372 | fPHOSGeo->GlobalPos2RelId(global,relId) ; | |
373 | Int_t mod = relId[0] ; | |
374 | Int_t cellX = relId[2]; | |
375 | Int_t cellZ = relId[3] ; | |
376 | if ( !IsGoodChannel(mod,cellX,cellZ) ) { | |
377 | clu->SetE(0.) ; | |
378 | continue ; | |
379 | } | |
380 | TVector3 locPosOld; //Use it to re-calculate distance to track | |
381 | fPHOSGeo->Global2Local(locPosOld,global,mod) ; | |
382 | ||
383 | //Apply re-Calibreation | |
384 | AliPHOSAodCluster cluPHOS(*clu); | |
385 | cluPHOS.Recalibrate(fPHOSCalibData,cells); // modify the cell energies | |
386 | cluPHOS.EvalAll(logWeight,vertex); // recalculate the cluster parameters | |
387 | cluPHOS.SetE(CorrectNonlinearity(cluPHOS.E()));// Users's nonlinearity | |
388 | ||
5bd0f028 | 389 | Double_t ecore=CoreEnergy(&cluPHOS) ; |
390 | ecore=CorrectNonlinearity(ecore) ; | |
391 | ||
392 | ||
f7d0ec90 | 393 | //Correct Misalignment |
e1c158c2 | 394 | // cluPHOS.GetPosition(position); |
395 | // global.SetXYZ(position[0],position[1],position[2]); | |
396 | // CorrectPHOSMisalignment(global,mod) ; | |
397 | // position[0]=global.X() ; | |
398 | // position[1]=global.Y() ; | |
399 | // position[2]=global.Z() ; | |
f7d0ec90 | 400 | |
401 | clu->SetPosition(position); //rec.point position in MARS | |
5bd0f028 | 402 | clu->SetE(cluPHOS.E()); //total particle energy |
403 | clu->SetMCEnergyFraction(ecore); //core particle energy | |
f7d0ec90 | 404 | clu->SetDispersion(cluPHOS.GetDispersion()); //cluster dispersion |
405 | // ec->SetPID(rp->GetPID()) ; //array of particle identification | |
406 | clu->SetM02(cluPHOS.GetM02()) ; //second moment M2x | |
407 | clu->SetM20(cluPHOS.GetM20()) ; //second moment M2z | |
408 | //correct distance to track | |
409 | Double_t dx=clu->GetTrackDx() ; | |
410 | Double_t dz=clu->GetTrackDz() ; | |
411 | TVector3 locPos; | |
412 | fPHOSGeo->Global2Local(locPos,global,mod) ; | |
413 | if(dx!=-999.){ //there is matched track | |
414 | dx+=locPos.X()-locPosOld.X() ; | |
415 | dz+=locPos.Z()-locPosOld.Z() ; | |
416 | clu->SetTrackDistance(dx,dz); | |
417 | } | |
418 | Double_t r = 999. ; //Big distance | |
419 | int nTracksMatched = clu->GetNTracksMatched(); | |
420 | if(nTracksMatched > 0) { | |
421 | AliVTrack* track = dynamic_cast<AliVTrack*> (clu->GetTrackMatched(0)); | |
422 | if ( track ) { | |
423 | Double_t pttrack = track->Pt(); | |
424 | Short_t charge = track->Charge(); | |
425 | r=TestCPV(dx, dz, pttrack,charge) ; | |
426 | } | |
427 | } | |
428 | clu->SetEmcCpvDistance(r); //Distance in sigmas | |
429 | ||
430 | clu->SetChi2(TestLambda(clu->E(),clu->GetM20(),clu->GetM02())); //not yet implemented | |
d4657394 | 431 | Double_t tof=EvalTOF(&cluPHOS,cells); |
432 | if(TMath::Abs(tof-clu->GetTOF())>100.e-9) //something wrong in cell TOF! | |
433 | tof=clu->GetTOF() ; | |
434 | clu->SetTOF(tof); | |
393b3556 | 435 | Double_t minDist=clu->GetDistanceToBadChannel() ;//Already calculated |
436 | DistanceToBadChannel(mod,&locPos,minDist); | |
437 | clu->SetDistanceToBadChannel(minDist) ; | |
f7d0ec90 | 438 | } |
439 | } | |
440 | ||
441 | } | |
442 | //___________________________________________________________________________________________________ | |
443 | void AliPHOSTenderSupply::FindTrackMatching(Int_t mod,TVector3 *locpos, | |
444 | Double_t &dx, Double_t &dz, | |
445 | Double_t &pt,Int_t &charge){ | |
446 | //Find track with closest extrapolation to cluster | |
447 | AliESDEvent *esd = 0x0 ; | |
448 | if(fTender) | |
449 | esd= fTender->GetEvent(); | |
450 | else{ | |
451 | esd= dynamic_cast<AliESDEvent*>(fTask->InputEvent()); | |
452 | } | |
453 | ||
454 | if(!esd){ | |
455 | AliError("ESD is not found") ; | |
456 | return ; | |
457 | } | |
458 | Double_t magF = esd->GetMagneticField(); | |
459 | ||
460 | Double_t magSign = 1.0; | |
461 | if(magF<0)magSign = -1.0; | |
462 | ||
463 | if (!TGeoGlobalMagField::Instance()->GetField()) { | |
464 | AliMagF* field = new AliMagF("Maps","Maps", magSign, magSign, AliMagF::k5kG); | |
465 | TGeoGlobalMagField::Instance()->SetField(field); | |
466 | } | |
467 | ||
468 | // *** Start the matching | |
469 | Int_t nt=0; | |
470 | nt = esd->GetNumberOfTracks(); | |
471 | //Calculate actual distance to PHOS module | |
472 | TVector3 globaPos ; | |
473 | fPHOSGeo->Local2Global(mod, 0.,0., globaPos) ; | |
474 | const Double_t rPHOS = globaPos.Pt() ; //Distance to center of PHOS module | |
475 | const Double_t kYmax = 72.+10. ; //Size of the module (with some reserve) in phi direction | |
476 | const Double_t kZmax = 64.+10. ; //Size of the module (with some reserve) in z direction | |
477 | const Double_t kAlpha0=330./180.*TMath::Pi() ; //First PHOS module angular direction | |
478 | const Double_t kAlpha= 20./180.*TMath::Pi() ; //PHOS module angular size | |
479 | Double_t minDistance = 1.e6; | |
480 | ||
481 | ||
482 | Double_t gposTrack[3] ; | |
483 | ||
484 | Double_t bz = ((AliMagF*)TGeoGlobalMagField::Instance()->GetField())->SolenoidField(); | |
485 | bz = TMath::Sign(0.5*kAlmost0Field,bz) + bz; | |
486 | ||
487 | Double_t b[3]; | |
488 | ||
489 | for (Int_t i=0; i<nt; i++) { | |
490 | AliESDtrack *esdTrack=esd->GetTrack(i); | |
491 | ||
492 | // Skip the tracks having "wrong" status (has to be checked/tuned) | |
493 | ULong_t status = esdTrack->GetStatus(); | |
494 | if ((status & AliESDtrack::kTPCout) == 0) continue; | |
495 | ||
496 | //Continue extrapolation from TPC outer surface | |
497 | const AliExternalTrackParam *outerParam=esdTrack->GetOuterParam(); | |
498 | if (!outerParam) continue; | |
499 | AliExternalTrackParam t(*outerParam); | |
500 | ||
501 | t.GetBxByBz(b) ; | |
502 | //Direction to the current PHOS module | |
503 | Double_t phiMod=kAlpha0-kAlpha*mod ; | |
504 | if(!t.Rotate(phiMod)) | |
505 | continue ; | |
506 | ||
507 | Double_t y; // Some tracks do not reach the PHOS | |
508 | if (!t.GetYAt(rPHOS,bz,y)) continue; // because of the bending | |
509 | ||
510 | Double_t z; | |
511 | if(!t.GetZAt(rPHOS,bz,z)) | |
512 | continue ; | |
513 | if (TMath::Abs(z) > kZmax) | |
514 | continue; // Some tracks miss the PHOS in Z | |
515 | if(TMath::Abs(y) < kYmax){ | |
516 | t.PropagateToBxByBz(rPHOS,b); // Propagate to the matching module | |
517 | //t.CorrectForMaterial(...); // Correct for the TOF material, if needed | |
518 | t.GetXYZ(gposTrack) ; | |
519 | TVector3 globalPositionTr(gposTrack) ; | |
520 | TVector3 localPositionTr ; | |
521 | fPHOSGeo->Global2Local(localPositionTr,globalPositionTr,mod) ; | |
522 | Double_t ddx = locpos->X()-localPositionTr.X(); | |
523 | Double_t ddz = locpos->Z()-localPositionTr.Z(); | |
524 | Double_t d2 = ddx*ddx + ddz*ddz; | |
525 | if(d2 < minDistance) { | |
526 | dx = ddx ; | |
527 | dz = ddz ; | |
528 | minDistance=d2 ; | |
529 | pt=esdTrack->Pt() ; | |
530 | charge=esdTrack->Charge() ; | |
531 | } | |
532 | } | |
533 | }//Scanned all tracks | |
534 | ||
535 | } | |
536 | //____________________________________________________________ | |
537 | Float_t AliPHOSTenderSupply::CorrectNonlinearity(Float_t en){ | |
538 | ||
539 | //For backward compatibility, if no RecoParameters found | |
540 | if(fNonlinearityVersion=="Default"){ | |
541 | return 0.0241+1.0504*en+0.000249*en*en ; | |
542 | } | |
543 | ||
544 | if(fNonlinearityVersion=="NoCorrection"){ | |
545 | return en ; | |
546 | } | |
547 | if(fNonlinearityVersion=="Gustavo2005"){ | |
548 | return fNonlinearityParams[0]+fNonlinearityParams[1]*en + fNonlinearityParams[2]*en*en ; | |
549 | } | |
550 | if(fNonlinearityVersion=="Henrik2010"){ | |
551 | return en*(fNonlinearityParams[0]+fNonlinearityParams[1]*TMath::Exp(-en*fNonlinearityParams[2]))*(1.+fNonlinearityParams[3]*TMath::Exp(-en*fNonlinearityParams[4]))*(1.+fNonlinearityParams[6]/(en*en+fNonlinearityParams[5])) ; | |
552 | } | |
553 | ||
554 | return en ; | |
555 | } | |
556 | //_____________________________________________________________________________ | |
557 | Double_t AliPHOSTenderSupply::TestLambda(Double_t pt,Double_t l1,Double_t l2){ | |
558 | //Parameterization for core dispersion | |
559 | //For R=4.5 | |
560 | Double_t l1Mean = 1.150200 + 0.097886/(1.+1.486645*pt+0.000038*pt*pt) ; | |
561 | Double_t l2Mean = 1.574706 + 0.997966*exp(-0.895075*pt)-0.010666*pt ; | |
562 | Double_t l1Sigma = 0.100255 + 0.337177*exp(-0.517684*pt)+0.001170*pt ; | |
563 | Double_t l2Sigma = 0.232580 + 0.573401*exp(-0.735903*pt)-0.002325*pt ; | |
564 | Double_t c = -0.110983 -0.017353/(1.-1.836995*pt+0.934517*pt*pt) ; | |
565 | ||
566 | /* | |
567 | //Parameterizatino for full dispersion | |
568 | Double_t l2Mean = 1.53126+9.50835e+06/(1.+1.08728e+07*pt+1.73420e+06*pt*pt) ; | |
569 | Double_t l1Mean = 1.12365+0.123770*TMath::Exp(-pt*0.246551)+5.30000e-03*pt ; | |
570 | Double_t l2Sigma = 6.48260e-02+7.60261e+10/(1.+1.53012e+11*pt+5.01265e+05*pt*pt)+9.00000e-03*pt; | |
571 | Double_t l1Sigma = 4.44719e-04+6.99839e-01/(1.+1.22497e+00*pt+6.78604e-07*pt*pt)+9.00000e-03*pt; | |
572 | Double_t c=-0.35-0.550*TMath::Exp(-0.390730*pt) ; | |
573 | */ | |
574 | Double_t R2=0.5*(l1-l1Mean)*(l1-l1Mean)/l1Sigma/l1Sigma + | |
575 | 0.5*(l2-l2Mean)*(l2-l2Mean)/l2Sigma/l2Sigma + | |
576 | 0.5*c*(l1-l1Mean)*(l2-l2Mean)/l1Sigma/l2Sigma ; | |
577 | return R2 ; | |
578 | ||
579 | } | |
580 | //____________________________________________________________________________ | |
581 | Double_t AliPHOSTenderSupply::TestCPV(Double_t dx, Double_t dz, Double_t pt, Int_t charge){ | |
582 | //Parameterization of LHC10h period | |
583 | //_true if neutral_ | |
584 | ||
585 | Double_t meanX=0; | |
586 | Double_t meanZ=0.; | |
587 | Double_t sx=TMath::Min(5.4,2.59719e+02*TMath::Exp(-pt/1.02053e-01)+ | |
588 | 6.58365e-01*5.91917e-01*5.91917e-01/((pt-9.61306e-01)*(pt-9.61306e-01)+5.91917e-01*5.91917e-01)+1.59219); | |
589 | Double_t sz=TMath::Min(2.75,4.90341e+02*1.91456e-02*1.91456e-02/(pt*pt+1.91456e-02*1.91456e-02)+1.60) ; | |
590 | ||
591 | Double_t mf = 0.; //Positive for ++ and negative for -- | |
592 | if(fTender){ | |
593 | AliESDEvent *esd = fTender->GetEvent(); | |
594 | mf = esd->GetMagneticField(); | |
595 | } | |
596 | else{ | |
597 | if(fTask){ | |
598 | AliESDEvent *esd= dynamic_cast<AliESDEvent*>(fTask->InputEvent()); | |
599 | if(esd) | |
600 | mf = esd->GetMagneticField(); | |
601 | else{ | |
602 | AliAODEvent *aod= dynamic_cast<AliAODEvent*>(fTask->InputEvent()); | |
603 | if(aod) | |
604 | mf = aod->GetMagneticField(); | |
605 | } | |
606 | }else{ | |
607 | AliError("Neither Tender nor Task defined") ; | |
608 | } | |
609 | } | |
610 | ||
611 | if(mf<0.){ //field -- | |
612 | meanZ = -0.468318 ; | |
613 | if(charge>0) | |
614 | meanX=TMath::Min(7.3, 3.89994*1.20679*1.20679/(pt*pt+1.20679*1.20679)+0.249029+2.49088e+07*TMath::Exp(-pt*3.33650e+01)) ; | |
615 | else | |
616 | meanX=-TMath::Min(7.7,3.86040*0.912499*0.912499/(pt*pt+0.912499*0.912499)+1.23114+4.48277e+05*TMath::Exp(-pt*2.57070e+01)) ; | |
617 | } | |
618 | else{ //Field ++ | |
619 | meanZ= -0.468318; | |
620 | if(charge>0) | |
621 | meanX=-TMath::Min(8.0,3.86040*1.31357*1.31357/(pt*pt+1.31357*1.31357)+0.880579+7.56199e+06*TMath::Exp(-pt*3.08451e+01)) ; | |
622 | else | |
623 | meanX= TMath::Min(6.85, 3.89994*1.16240*1.16240/(pt*pt+1.16240*1.16240)-0.120787+2.20275e+05*TMath::Exp(-pt*2.40913e+01)) ; | |
624 | } | |
625 | ||
626 | Double_t rz=(dz-meanZ)/sz ; | |
627 | Double_t rx=(dx-meanX)/sx ; | |
628 | return TMath::Sqrt(rx*rx+rz*rz) ; | |
629 | } | |
630 | ||
631 | //________________________________________________________________________ | |
632 | Bool_t AliPHOSTenderSupply::IsGoodChannel(Int_t mod, Int_t ix, Int_t iz) | |
633 | { | |
634 | //Check if this channel belogs to the good ones | |
635 | ||
636 | if(mod>4 || mod<1){ | |
637 | // AliError(Form("No bad map for PHOS module %d ",mod)) ; | |
638 | return kTRUE ; | |
639 | } | |
640 | if(!fPHOSBadMap[mod]){ | |
641 | // AliError(Form("No Bad map for PHOS module %d",mod)) ; | |
642 | return kTRUE ; | |
643 | } | |
644 | if(fPHOSBadMap[mod]->GetBinContent(ix,iz)>0) | |
645 | return kFALSE ; | |
646 | else | |
647 | return kTRUE ; | |
648 | } | |
649 | //________________________________________________________________________ | |
650 | void AliPHOSTenderSupply::ForceUsingBadMap(const char * filename){ | |
651 | //Read TH2I histograms with bad maps from local or alien file | |
652 | TFile * fbm = TFile::Open(filename) ; | |
653 | if(!fbm || !fbm->IsOpen()){ | |
654 | AliError(Form("Can not open BadMaps file %s",filename)) ; | |
655 | return ; | |
656 | } | |
657 | gROOT->cd() ; | |
658 | char key[55] ; | |
659 | for(Int_t mod=1;mod<4; mod++){ | |
660 | snprintf(key,55,"PHOS_BadMap_mod%d",mod) ; | |
661 | TH2I * h = (TH2I*)fbm->Get(key) ; | |
662 | if(h) | |
663 | fPHOSBadMap[mod] = new TH2I(*h) ; | |
664 | } | |
665 | fbm->Close() ; | |
666 | fUsePrivateBadMap=kTRUE ; | |
667 | } | |
668 | //________________________________________________________________________ | |
669 | void AliPHOSTenderSupply::ForceUsingCalibration(const char * filename){ | |
670 | //Read PHOS recalibration parameters from the file. | |
671 | //We assume that file contains single entry: AliPHOSCalibData | |
672 | TFile * fc = TFile::Open(filename) ; | |
673 | if(!fc || !fc->IsOpen()){ | |
674 | AliFatal(Form("Can not open Calibration file %s",filename)) ; | |
675 | return ; | |
676 | } | |
677 | fPHOSCalibData = (AliPHOSCalibData*)fc->Get("PHOSCalibration") ; | |
678 | fc->Close() ; | |
679 | fUsePrivateCalib=kTRUE; | |
680 | } | |
681 | //________________________________________________________________________ | |
682 | void AliPHOSTenderSupply::CorrectPHOSMisalignment(TVector3 &global,Int_t mod){ | |
683 | //Correct for PHOS modules misalignment | |
684 | ||
685 | //correct misalignment | |
686 | const Float_t shiftX[6]={0.,-2.3,-2.11,-1.53,0.,0.} ; | |
687 | const Float_t shiftZ[6]={0.,-0.4, 0.52, 0.8,0.,0.} ; | |
688 | TVector3 localPos ; | |
689 | fPHOSGeo->Global2Local(localPos,global,mod) ; | |
690 | fPHOSGeo->Local2Global(mod,localPos.X()+shiftX[mod],localPos.Z()+shiftZ[mod],global); | |
691 | } | |
692 | //________________________________________________________________________ | |
693 | void AliPHOSTenderSupply::EvalLambdas(AliVCluster * clu, Double_t &m02, Double_t &m20){ | |
694 | //calculate dispecrsion of the cluster in the circle with radius distanceCut around the maximum | |
695 | ||
696 | const Double_t rCut=4.5 ; | |
697 | ||
698 | Double32_t * elist = clu->GetCellsAmplitudeFraction() ; | |
699 | // Calculates the center of gravity in the local PHOS-module coordinates | |
700 | Float_t wtot = 0; | |
701 | Double_t xc[100]={0} ; | |
702 | Double_t zc[100]={0} ; | |
703 | Double_t x = 0 ; | |
704 | Double_t z = 0 ; | |
705 | Int_t mulDigit=TMath::Min(100,clu->GetNCells()) ; | |
706 | const Double_t logWeight=4.5 ; | |
707 | for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) { | |
708 | Int_t relid[4] ; | |
709 | Float_t xi ; | |
710 | Float_t zi ; | |
711 | fPHOSGeo->AbsToRelNumbering(clu->GetCellAbsId(iDigit), relid) ; | |
712 | fPHOSGeo->RelPosInModule(relid, xi, zi); | |
713 | xc[iDigit]=xi ; | |
714 | zc[iDigit]=zi ; | |
715 | if (clu->E()>0 && elist[iDigit]>0) { | |
716 | Float_t w = TMath::Max( 0., logWeight + TMath::Log( elist[iDigit] / clu->E() ) ) ; | |
717 | x += xc[iDigit] * w ; | |
718 | z += zc[iDigit] * w ; | |
719 | wtot += w ; | |
720 | } | |
721 | } | |
722 | if (wtot>0) { | |
723 | x /= wtot ; | |
724 | z /= wtot ; | |
725 | } | |
726 | ||
727 | wtot = 0. ; | |
728 | Double_t dxx = 0.; | |
729 | Double_t dzz = 0.; | |
730 | Double_t dxz = 0.; | |
731 | Double_t xCut = 0. ; | |
732 | Double_t zCut = 0. ; | |
733 | for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) { | |
734 | if (clu->E()>0 && elist[iDigit]>0.) { | |
735 | Double_t w = TMath::Max( 0., logWeight + TMath::Log( elist[iDigit] / clu->E() ) ) ; | |
736 | Double_t xi= xc[iDigit] ; | |
737 | Double_t zi= zc[iDigit] ; | |
738 | if((xi-x)*(xi-x)+(zi-z)*(zi-z) < rCut*rCut){ | |
739 | xCut += w * xi ; | |
740 | zCut += w * zi ; | |
741 | dxx += w * xi * xi ; | |
742 | dzz += w * zi * zi ; | |
743 | dxz += w * xi * zi ; | |
744 | wtot += w ; | |
745 | } | |
746 | } | |
747 | ||
748 | } | |
749 | if (wtot>0) { | |
750 | xCut/= wtot ; | |
751 | zCut/= wtot ; | |
752 | dxx /= wtot ; | |
753 | dzz /= wtot ; | |
754 | dxz /= wtot ; | |
755 | dxx -= xCut * xCut ; | |
756 | dzz -= zCut * zCut ; | |
757 | dxz -= xCut * zCut ; | |
758 | ||
759 | m02 = 0.5 * (dxx + dzz) + TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz ) ; | |
760 | m20 = 0.5 * (dxx + dzz) - TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz ) ; | |
761 | } | |
762 | else { | |
763 | m20=m02=0.; | |
764 | } | |
765 | ||
766 | } | |
5bd0f028 | 767 | //____________________________________________________________________________ |
768 | Double_t AliPHOSTenderSupply::CoreEnergy(AliVCluster * clu){ | |
769 | //calculate energy of the cluster in the circle with radius distanceCut around the maximum | |
770 | ||
771 | //Can not use already calculated coordinates? | |
772 | //They have incidence correction... | |
773 | const Double_t distanceCut =3.5 ; | |
774 | const Double_t logWeight=4.5 ; | |
775 | ||
776 | Double32_t * elist = clu->GetCellsAmplitudeFraction() ; | |
777 | // Calculates the center of gravity in the local PHOS-module coordinates | |
778 | Float_t wtot = 0; | |
779 | Double_t xc[100]={0} ; | |
780 | Double_t zc[100]={0} ; | |
781 | Double_t x = 0 ; | |
782 | Double_t z = 0 ; | |
783 | Int_t mulDigit=TMath::Min(100,clu->GetNCells()) ; | |
784 | for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) { | |
785 | Int_t relid[4] ; | |
786 | Float_t xi ; | |
787 | Float_t zi ; | |
788 | fPHOSGeo->AbsToRelNumbering(clu->GetCellAbsId(iDigit), relid) ; | |
789 | fPHOSGeo->RelPosInModule(relid, xi, zi); | |
790 | xc[iDigit]=xi ; | |
791 | zc[iDigit]=zi ; | |
792 | if (clu->E()>0 && elist[iDigit]>0) { | |
793 | Float_t w = TMath::Max( 0., logWeight + TMath::Log( elist[iDigit] / clu->E() ) ) ; | |
794 | x += xc[iDigit] * w ; | |
795 | z += zc[iDigit] * w ; | |
796 | wtot += w ; | |
797 | } | |
798 | } | |
799 | if (wtot>0) { | |
800 | x /= wtot ; | |
801 | z /= wtot ; | |
802 | } | |
803 | Double_t coreE=0. ; | |
804 | for(Int_t iDigit=0; iDigit < mulDigit; iDigit++) { | |
805 | Double_t distance = TMath::Sqrt((xc[iDigit]-x)*(xc[iDigit]-x)+(zc[iDigit]-z)*(zc[iDigit]-z)) ; | |
806 | if(distance < distanceCut) | |
807 | coreE += elist[iDigit] ; | |
808 | } | |
809 | //Apply non-linearity correction | |
810 | return coreE ; | |
811 | } | |
f7d0ec90 | 812 | //________________________________________________________________________ |
813 | Double_t AliPHOSTenderSupply::EvalTOF(AliVCluster * clu,AliVCaloCells * cells){ | |
814 | //Evaluate TOF of the cluster after re-calibration | |
815 | //TOF here is weighted average of digits | |
816 | // -within 50ns from the most energetic cell | |
817 | // -not too soft. | |
818 | ||
819 | ||
820 | Double32_t * elist = clu->GetCellsAmplitudeFraction() ; | |
821 | Int_t mulDigit=clu->GetNCells() ; | |
822 | ||
823 | Float_t tMax= 0.; //Time at the maximum | |
824 | Float_t eMax=0. ; | |
825 | for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) { | |
826 | Int_t absId=clu->GetCellAbsId(iDigit) ; | |
827 | Bool_t isHG=kTRUE ; | |
828 | if(cells->GetCellMCLabel(absId)==-2) //This is LG digit. No statistics to calibrate LG timing, remove them from TOF calculation | |
829 | isHG=kFALSE ; | |
830 | if( elist[iDigit]>eMax){ | |
831 | tMax=CalibrateTOF(cells->GetCellTime(absId),absId,isHG) ; | |
832 | eMax=elist[iDigit] ; | |
833 | } | |
834 | } | |
f7d0ec90 | 835 | |
836 | //Try to improve accuracy | |
837 | //Do not account time of soft cells: | |
838 | // const Double_t part=0.5 ; | |
839 | Double_t eMin=TMath::Min(0.5,0.2*eMax) ; | |
840 | Float_t wtot = 0.; | |
841 | Double_t t = 0. ; | |
842 | for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) { | |
843 | Int_t absId=clu->GetCellAbsId(iDigit) ; | |
844 | Bool_t isHG=kTRUE ; | |
845 | if(cells->GetCellMCLabel(absId)==-2) //This is LG digit. No statistics to calibrate LG timing, remove them from TOF calculation | |
846 | isHG=kFALSE ; | |
847 | ||
848 | Double_t ti=CalibrateTOF(cells->GetCellTime(absId),absId,isHG) ; | |
849 | if(TMath::Abs(ti-tMax)>50.e-9) //remove soft cells with wrong time | |
850 | continue ; | |
851 | ||
852 | //Remove too soft cells | |
853 | if(elist[iDigit]<eMin) | |
854 | continue ; | |
855 | ||
856 | if(elist[iDigit]>0){ | |
857 | //weight = 1./sigma^2 | |
858 | //Sigma is parameterization of TOF resolution 16.05.2013 | |
859 | Double_t wi2=0.; | |
860 | if(isHG) | |
861 | wi2=1./(2.4e-9 + 3.9e-9/elist[iDigit]) ; | |
862 | else | |
863 | wi2=1./(2.4e-9 + 3.9e-9/(0.1*elist[iDigit])) ; //E of LG digit is 1/16 of correcponding HG | |
864 | t+=ti*wi2 ; | |
865 | wtot+=wi2 ; | |
866 | } | |
867 | } | |
868 | if(wtot>0){ | |
869 | t=t/wtot ; | |
870 | } | |
871 | ||
872 | return t ; | |
873 | ||
874 | } | |
875 | //________________________________________________________________________ | |
876 | Double_t AliPHOSTenderSupply::CalibrateTOF(Double_t tof, Int_t absId, Bool_t isHG){ | |
877 | //Apply time re-calibration separately for HG and LG channels | |
878 | //By default (if not filled) shifts are zero. | |
879 | ||
880 | Int_t relId[4]; | |
881 | fPHOSGeo->AbsToRelNumbering(absId,relId) ; | |
882 | Int_t module = relId[0]; | |
883 | Int_t column = relId[3]; | |
884 | Int_t row = relId[2]; | |
885 | if(isHG) | |
886 | tof-=fPHOSCalibData->GetTimeShiftEmc(module, column, row); | |
887 | else | |
888 | tof-=fPHOSCalibData->GetLGTimeShiftEmc(module, column, row); | |
889 | ||
890 | return tof ; | |
891 | ||
892 | } | |
393b3556 | 893 | //________________________________________________________________________ |
894 | void AliPHOSTenderSupply::DistanceToBadChannel(Int_t mod, TVector3 * locPos, Double_t &minDist){ | |
895 | //Check if distance to bad channel was reduced | |
896 | Int_t range = minDist/2.2 +1 ; //Distance at which bad channels should be serached | |
897 | ||
898 | Int_t relid[4]={0,0,0,0} ; | |
899 | fPHOSGeo->RelPosToRelId(mod, locPos->X(), locPos->Z(), relid) ; | |
900 | Int_t xmin=TMath::Max(1,relid[2]-range) ; | |
901 | Int_t xmax=TMath::Min(64,relid[2]+range) ; | |
902 | Int_t zmin=TMath::Max(1,relid[3]-range) ; | |
903 | Int_t zmax=TMath::Min(56,relid[3]+range) ; | |
904 | ||
905 | Float_t x=0.,z=0.; | |
906 | for(Int_t ix=xmin;ix<=xmax;ix++){ | |
907 | for(Int_t iz=zmin;iz<=zmax;iz++){ | |
908 | if(fPHOSBadMap[mod]->GetBinContent(ix,iz)>0){ //Bad channel | |
909 | Int_t relidBC[4]={mod,0,ix,iz} ; | |
910 | fPHOSGeo->RelPosInModule(relidBC,x,z); | |
911 | Double_t dist = TMath::Sqrt((x-locPos->X())*(x-locPos->X()) + (z-locPos->Z())*(z-locPos->Z())); | |
912 | if(dist<minDist) minDist = dist; | |
913 | } | |
914 | } | |
915 | } | |
916 | ||
917 | } | |
918 | ||
e1c158c2 | 919 |