]> git.uio.no Git - u/mrichter/AliRoot.git/blame - TENDER/TenderSupplies/AliPHOSTenderSupply.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Merge branch 'feature-movesplit'
[u/mrichter/AliRoot.git] / TENDER / TenderSupplies / AliPHOSTenderSupply.cxx
CommitLineData
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
49ClassImp(AliPHOSTenderSupply)
50
51AliPHOSTenderSupply::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//_____________________________________________________
72AliPHOSTenderSupply::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//_____________________________________________________
93AliPHOSTenderSupply::~AliPHOSTenderSupply()
94{
95 //Destructor
96 if(fPHOSCalibData)
97 delete fPHOSCalibData;
98 fPHOSCalibData=0x0 ;
99}
100
101//_____________________________________________________
102void 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");
9c4b77df 162 if(fIsMC) //use excatly the same geometry as in simulation
163 geomContainer.InitFromFile("$ALICE_ROOT/OADB/PHOS/PHOSMCGeometry.root","PHOSMCRotationMatrixes");
164 else //Use best approaximation to real geometry
165 geomContainer.InitFromFile("$ALICE_ROOT/OADB/PHOS/PHOSGeometry.root","PHOSRotationMatrixes");
f7d0ec90 166 TObjArray *matrixes = (TObjArray*)geomContainer.GetObject(runNumber,"PHOSRotationMatrixes");
167 fPHOSGeo = AliPHOSGeometry::GetInstance("IHEP") ;
168 for(Int_t mod=0; mod<5; mod++) {
169 if(!matrixes->At(mod)) continue;
170 fPHOSGeo->SetMisalMatrix(((TGeoHMatrix*)matrixes->At(mod)),mod) ;
171 printf(".........Adding Matrix(%d), geo=%p\n",mod,fPHOSGeo) ;
e1c158c2 172 ((TGeoHMatrix*)matrixes->At(mod))->Print() ;
9c4b77df 173 }
f7d0ec90 174 }
175
176 //Init Bad channels map
177 if(!fUsePrivateBadMap){
178 AliOADBContainer badmapContainer(Form("phosBadMap"));
179 badmapContainer.InitFromFile("$ALICE_ROOT/OADB/PHOS/PHOSBadMaps.root","phosBadMap");
180 TObjArray *maps = (TObjArray*)badmapContainer.GetObject(runNumber,"phosBadMap");
181 if(!maps){
182 AliError(Form("Can not read Bad map for run %d. \n You may choose to use your map with ForceUsingBadMap()\n",runNumber)) ;
183 }
184 else{
185 AliInfo(Form("Setting PHOS bad map with name %s \n",maps->GetName())) ;
186 for(Int_t mod=0; mod<5;mod++){
187 if(fPHOSBadMap[mod])
188 delete fPHOSBadMap[mod] ;
189 TH2I * h = (TH2I*)maps->At(mod) ;
190 if(h)
191 fPHOSBadMap[mod]=new TH2I(*h) ;
192 }
193 }
194 }
195
196 if(!fUsePrivateCalib){
197 if(fIsMC){ //re/de-calibration for MC productions
198 //Init recalibration
199 AliOADBContainer calibContainer("phosRecalibration");
200 calibContainer.InitFromFile("$ALICE_ROOT/OADB/PHOS/PHOSMCCalibrations.root","phosRecalibration");
201
202 TObjArray *recalib = (TObjArray*)calibContainer.GetObject(runNumber,"PHOSRecalibration");
203 if(!recalib){
204 AliFatal(Form("Can not read calibrations for run %d\n. You may choose your specific calibration with ForceUsingCalibration()\n",runNumber)) ;
205 }
206 else{
207 //Now try to find object with proper name
208 for(Int_t i=0; i<recalib->GetEntriesFast(); i++){
209 AliPHOSCalibData * tmp = (AliPHOSCalibData*)recalib->At(i) ;
210 if(fMCProduction.CompareTo(tmp->GetName())==0){
211 fPHOSCalibData = tmp ;
212 break ;
213 }
214 }
215 if(!fPHOSCalibData) {
216 AliFatal(Form("Can not find calibration for run %d, and name %s \n",runNumber, fMCProduction.Data())) ;
217 }
218 }
219
220 }
221 else{ //real data
222 //Init recalibration
223 //Check the pass1-pass2-pass3 reconstruction
224 AliOADBContainer calibContainer("phosRecalibration");
225 calibContainer.InitFromFile("$ALICE_ROOT/OADB/PHOS/PHOSCalibrations.root","phosRecalibration");
226 TObjArray *recalib = (TObjArray*)calibContainer.GetObject(runNumber,"PHOSRecalibration");
227 if(!recalib){
228 AliFatal(Form("Can not read calibrations for run %d\n. You may choose your specific calibration with ForceUsingCalibration()\n",runNumber)) ;
229 }
230 else{
231 fPHOSCalibData = (AliPHOSCalibData*)recalib->At(fRecoPass-1) ;
232 if(!fPHOSCalibData) {
233 AliFatal(Form("Can not find calibration for run %d, pass %d \n",runNumber, fRecoPass)) ;
234 }
235 }
236 }
237 }
238
239}
240
241//_____________________________________________________
242void AliPHOSTenderSupply::ProcessEvent()
243{
244 //Choose PHOS clusters and recalibrate them
245 //that it recalculate energy, position and distance
246 //to closest track extrapolation
247
248 AliESDEvent *esd = 0x0 ;
249 AliAODEvent *aod = 0x0 ;
250 if(fTender){
251 esd = fTender->GetEvent();
252 if(!esd)
253 return ;
254 }
255 else{
256 if(!fTask){
257 return ;
258 }
259 esd = dynamic_cast<AliESDEvent*>(fTask->InputEvent()) ;
260 aod = dynamic_cast<AliAODEvent*>(fTask->InputEvent()) ;
261 if(!esd && !aod)
262 return ;
263 }
264
265 if(!fPHOSCalibData
266 || (fTender && fTender->RunChanged())){ //In case of Task init called automatically
267 InitTender();
268
269 }
270
271 TVector3 vertex ;
272 if(esd){
273 const AliESDVertex *esdVertex = esd->GetPrimaryVertex();
274 vertex.SetXYZ(esdVertex->GetX(),esdVertex->GetY(),esdVertex->GetZ());
275 }
276 else{//AOD
277 const AliAODVertex *aodVertex = aod->GetPrimaryVertex();
278 vertex.SetXYZ(aodVertex->GetX(),aodVertex->GetY(),aodVertex->GetZ());
279 }
280 if(vertex.Mag()>99.) //vertex not defined?
281 vertex.SetXYZ(0.,0.,0.) ;
282
283
284 //For re-calibration
285 const Double_t logWeight=4.5 ;
286
287 if(esd){ //To avoid multiple if in loops we made
288 //almost identical pecies of code. Please apply changes to both!!!
289 Int_t multClust=esd->GetNumberOfCaloClusters();
290 AliESDCaloCells * cells = esd->GetPHOSCells() ;
291
292 for (Int_t i=0; i<multClust; i++) {
293 AliESDCaloCluster *clu = esd->GetCaloCluster(i);
294 if ( !clu->IsPHOS()) continue;
295
4b2c1eeb 296
297 //Apply re-Calibreation
298 AliPHOSEsdCluster cluPHOS(*clu);
299 cluPHOS.Recalibrate(fPHOSCalibData,cells); // modify the cell energies
300 cluPHOS.EvalAll(logWeight,vertex); // recalculate the cluster parameters
301 cluPHOS.SetE(CorrectNonlinearity(cluPHOS.E()));// Users's nonlinearity
302
f7d0ec90 303 Float_t position[3];
4b2c1eeb 304 cluPHOS.GetPosition(position);
77e3cbcb 305 clu->SetPosition(position); //rec.point position in MARS
f7d0ec90 306 TVector3 global(position) ;
307 Int_t relId[4] ;
308 fPHOSGeo->GlobalPos2RelId(global,relId) ;
309 Int_t mod = relId[0] ;
310 Int_t cellX = relId[2];
311 Int_t cellZ = relId[3] ;
312 if ( !IsGoodChannel(mod,cellX,cellZ) ) {
313 clu->SetE(0.) ;
314 continue ;
315 }
4b2c1eeb 316
5bd0f028 317 Double_t ecore=CoreEnergy(&cluPHOS) ;
318 ecore=CorrectNonlinearity(ecore) ;
319
77e3cbcb 320 clu->SetE(cluPHOS.E()); //total particle energy
321 clu->SetCoreEnergy(ecore); //core particle energy
322
323 //Eval FullDispersion
324 clu->SetDispersion(TestFullLambda(clu->E(),cluPHOS.GetM20(),cluPHOS.GetM02())) ;
f7d0ec90 325 //Eval CoreDispersion
326 Double_t m02=0.,m20=0.;
327 EvalLambdas(&cluPHOS,m02, m20);
77e3cbcb 328 clu->SetChi2(TestCoreLambda(clu->E(),m20,m02)); //not yet implemented
329 clu->SetM02(m02) ; //second moment M2x
330 clu->SetM20(m20) ; //second moment M2z
f7d0ec90 331
77e3cbcb 332 //correct distance to track
f7d0ec90 333 Double_t dx=0.,dz=0. ;
334 fPHOSGeo->GlobalPos2RelId(global,relId) ;
335 TVector3 locPos;
336 fPHOSGeo->Global2Local(locPos,global,mod) ;
337
338 Double_t pttrack=0.;
339 Int_t charge=0;
340 FindTrackMatching(mod,&locPos,dx,dz,pttrack,charge) ;
341 Double_t r=TestCPV(dx, dz, pttrack,charge) ;
342 clu->SetTrackDistance(dx,dz);
343
344 clu->SetEmcCpvDistance(r);
77e3cbcb 345
d4657394 346 Double_t tof=EvalTOF(&cluPHOS,cells);
1e7fec1c 347// if(TMath::Abs(tof-clu->GetTOF())>100.e-9) //something wrong in cell TOF!
348// tof=clu->GetTOF() ;
d4657394 349 clu->SetTOF(tof);
393b3556 350 Double_t minDist=clu->GetDistanceToBadChannel() ;//Already calculated
351 DistanceToBadChannel(mod,&locPos,minDist);
352 clu->SetDistanceToBadChannel(minDist) ;
f7d0ec90 353
5b25db73 354 Double_t ecross = EvalEcross(&cluPHOS);
355 clu->SetMCEnergyFraction(ecross) ;
f7d0ec90 356 }
357 }
358 else{//AOD
359 Int_t multClust=aod->GetNumberOfCaloClusters();
360 AliAODCaloCells * cells = aod->GetPHOSCells() ;
361
362 for (Int_t i=0; i<multClust; i++) {
363 AliAODCaloCluster *clu = aod->GetCaloCluster(i);
364 if ( !clu->IsPHOS()) continue;
365
4b2c1eeb 366
367 //Apply re-Calibreation
368 AliPHOSAodCluster cluPHOS(*clu);
369 cluPHOS.Recalibrate(fPHOSCalibData,cells); // modify the cell energies
370 cluPHOS.EvalAll(logWeight,vertex); // recalculate the cluster parameters
371 cluPHOS.SetE(CorrectNonlinearity(cluPHOS.E()));// Users's nonlinearity
372
f7d0ec90 373 Float_t position[3];
4b2c1eeb 374 cluPHOS.GetPosition(position);
375 clu->SetPosition(position); //rec.point position in MARS
f7d0ec90 376 TVector3 global(position) ;
377 Int_t relId[4] ;
378 fPHOSGeo->GlobalPos2RelId(global,relId) ;
379 Int_t mod = relId[0] ;
380 Int_t cellX = relId[2];
381 Int_t cellZ = relId[3] ;
382 if ( !IsGoodChannel(mod,cellX,cellZ) ) {
383 clu->SetE(0.) ;
384 continue ;
385 }
386 TVector3 locPosOld; //Use it to re-calculate distance to track
387 fPHOSGeo->Global2Local(locPosOld,global,mod) ;
4b2c1eeb 388
5bd0f028 389 Double_t ecore=CoreEnergy(&cluPHOS) ;
390 ecore=CorrectNonlinearity(ecore) ;
77e3cbcb 391
5bd0f028 392 clu->SetE(cluPHOS.E()); //total particle energy
53442b36 393 clu->SetCoreEnergy(ecore); //core particle energy
77e3cbcb 394
395 //Eval FullDispersion
396 clu->SetDispersion(TestFullLambda(clu->E(),cluPHOS.GetM20(),cluPHOS.GetM02())) ;
397 //Eval CoreDispersion
398 Double_t m02=0.,m20=0.;
399 EvalLambdas(&cluPHOS,m02, m20);
400 clu->SetChi2(TestCoreLambda(clu->E(),m20,m02)); //not yet implemented
401 clu->SetM02(m02) ; //second moment M2x
402 clu->SetM20(m20) ; //second moment M2z
403
f7d0ec90 404 //correct distance to track
405 Double_t dx=clu->GetTrackDx() ;
406 Double_t dz=clu->GetTrackDz() ;
407 TVector3 locPos;
408 fPHOSGeo->Global2Local(locPos,global,mod) ;
409 if(dx!=-999.){ //there is matched track
410 dx+=locPos.X()-locPosOld.X() ;
411 dz+=locPos.Z()-locPosOld.Z() ;
412 clu->SetTrackDistance(dx,dz);
413 }
414 Double_t r = 999. ; //Big distance
415 int nTracksMatched = clu->GetNTracksMatched();
416 if(nTracksMatched > 0) {
417 AliVTrack* track = dynamic_cast<AliVTrack*> (clu->GetTrackMatched(0));
418 if ( track ) {
419 Double_t pttrack = track->Pt();
420 Short_t charge = track->Charge();
421 r=TestCPV(dx, dz, pttrack,charge) ;
422 }
423 }
424 clu->SetEmcCpvDistance(r); //Distance in sigmas
77e3cbcb 425
426
d4657394 427 Double_t tof=EvalTOF(&cluPHOS,cells);
1e7fec1c 428// if(TMath::Abs(tof-clu->GetTOF())>100.e-9) //something wrong in cell TOF!
429// tof=clu->GetTOF() ;
d4657394 430 clu->SetTOF(tof);
393b3556 431 Double_t minDist=clu->GetDistanceToBadChannel() ;//Already calculated
432 DistanceToBadChannel(mod,&locPos,minDist);
433 clu->SetDistanceToBadChannel(minDist) ;
5b25db73 434
435 Double_t ecross = EvalEcross(&cluPHOS);
436 clu->SetMCEnergyFraction(ecross) ;
f7d0ec90 437 }
438 }
439
440}
441//___________________________________________________________________________________________________
442void AliPHOSTenderSupply::FindTrackMatching(Int_t mod,TVector3 *locpos,
443 Double_t &dx, Double_t &dz,
444 Double_t &pt,Int_t &charge){
445 //Find track with closest extrapolation to cluster
446 AliESDEvent *esd = 0x0 ;
447 if(fTender)
448 esd= fTender->GetEvent();
449 else{
450 esd= dynamic_cast<AliESDEvent*>(fTask->InputEvent());
451 }
452
453 if(!esd){
454 AliError("ESD is not found") ;
455 return ;
456 }
457 Double_t magF = esd->GetMagneticField();
458
459 Double_t magSign = 1.0;
460 if(magF<0)magSign = -1.0;
461
462 if (!TGeoGlobalMagField::Instance()->GetField()) {
ac45cf45 463 AliError("Margnetic filed was not initialized, use default") ;
f7d0ec90 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//____________________________________________________________
537Float_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 }
b6f2b22a 543 if(fNonlinearityVersion=="MC"){ //Default + some correction
544 return (0.0241+1.0504*en+0.000249*en*en)*fNonlinearityParams[0]*(1+fNonlinearityParams[1]/(1.+en*en/fNonlinearityParams[2]/fNonlinearityParams[2])) ;
545 }
f7d0ec90 546
547 if(fNonlinearityVersion=="NoCorrection"){
548 return en ;
549 }
550 if(fNonlinearityVersion=="Gustavo2005"){
551 return fNonlinearityParams[0]+fNonlinearityParams[1]*en + fNonlinearityParams[2]*en*en ;
552 }
553 if(fNonlinearityVersion=="Henrik2010"){
554 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])) ;
555 }
556
557 return en ;
558}
559//_____________________________________________________________________________
77e3cbcb 560Double_t AliPHOSTenderSupply::TestCoreLambda(Double_t pt,Double_t l1,Double_t l2){
f7d0ec90 561//Parameterization for core dispersion
562//For R=4.5
563 Double_t l1Mean = 1.150200 + 0.097886/(1.+1.486645*pt+0.000038*pt*pt) ;
564 Double_t l2Mean = 1.574706 + 0.997966*exp(-0.895075*pt)-0.010666*pt ;
565 Double_t l1Sigma = 0.100255 + 0.337177*exp(-0.517684*pt)+0.001170*pt ;
566 Double_t l2Sigma = 0.232580 + 0.573401*exp(-0.735903*pt)-0.002325*pt ;
567 Double_t c = -0.110983 -0.017353/(1.-1.836995*pt+0.934517*pt*pt) ;
568
569/*
570 //Parameterizatino for full dispersion
571 Double_t l2Mean = 1.53126+9.50835e+06/(1.+1.08728e+07*pt+1.73420e+06*pt*pt) ;
572 Double_t l1Mean = 1.12365+0.123770*TMath::Exp(-pt*0.246551)+5.30000e-03*pt ;
573 Double_t l2Sigma = 6.48260e-02+7.60261e+10/(1.+1.53012e+11*pt+5.01265e+05*pt*pt)+9.00000e-03*pt;
574 Double_t l1Sigma = 4.44719e-04+6.99839e-01/(1.+1.22497e+00*pt+6.78604e-07*pt*pt)+9.00000e-03*pt;
575 Double_t c=-0.35-0.550*TMath::Exp(-0.390730*pt) ;
576*/
577 Double_t R2=0.5*(l1-l1Mean)*(l1-l1Mean)/l1Sigma/l1Sigma +
578 0.5*(l2-l2Mean)*(l2-l2Mean)/l2Sigma/l2Sigma +
579 0.5*c*(l1-l1Mean)*(l2-l2Mean)/l1Sigma/l2Sigma ;
77e3cbcb 580 return R2 ;
581
582}
583//_____________________________________________________________________________
584Double_t AliPHOSTenderSupply::TestFullLambda(Double_t pt,Double_t l1,Double_t l2){
585//Parameterization for full dispersion
586 //Parameterizatino for full dispersion
587 Double_t l2Mean = 1.53126+9.50835e+06/(1.+1.08728e+07*pt+1.73420e+06*pt*pt) ;
588 Double_t l1Mean = 1.12365+0.123770*TMath::Exp(-pt*0.246551)+5.30000e-03*pt ;
589 Double_t l2Sigma = 6.48260e-02+7.60261e+10/(1.+1.53012e+11*pt+5.01265e+05*pt*pt)+9.00000e-03*pt;
590 Double_t l1Sigma = 4.44719e-04+6.99839e-01/(1.+1.22497e+00*pt+6.78604e-07*pt*pt)+9.00000e-03*pt;
591 Double_t c=-0.35-0.550*TMath::Exp(-0.390730*pt) ;
592
593 Double_t R2=0.5*(l1-l1Mean)*(l1-l1Mean)/l1Sigma/l1Sigma +
594 0.5*(l2-l2Mean)*(l2-l2Mean)/l2Sigma/l2Sigma +
595 0.5*c*(l1-l1Mean)*(l2-l2Mean)/l1Sigma/l2Sigma ;
f7d0ec90 596 return R2 ;
597
598}
599//____________________________________________________________________________
600Double_t AliPHOSTenderSupply::TestCPV(Double_t dx, Double_t dz, Double_t pt, Int_t charge){
601 //Parameterization of LHC10h period
602 //_true if neutral_
603
604 Double_t meanX=0;
605 Double_t meanZ=0.;
606 Double_t sx=TMath::Min(5.4,2.59719e+02*TMath::Exp(-pt/1.02053e-01)+
607 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);
608 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) ;
609
610 Double_t mf = 0.; //Positive for ++ and negative for --
611 if(fTender){
612 AliESDEvent *esd = fTender->GetEvent();
613 mf = esd->GetMagneticField();
614 }
615 else{
616 if(fTask){
617 AliESDEvent *esd= dynamic_cast<AliESDEvent*>(fTask->InputEvent());
618 if(esd)
619 mf = esd->GetMagneticField();
620 else{
621 AliAODEvent *aod= dynamic_cast<AliAODEvent*>(fTask->InputEvent());
622 if(aod)
623 mf = aod->GetMagneticField();
624 }
625 }else{
626 AliError("Neither Tender nor Task defined") ;
627 }
628 }
629
630 if(mf<0.){ //field --
631 meanZ = -0.468318 ;
632 if(charge>0)
633 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)) ;
634 else
635 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)) ;
636 }
637 else{ //Field ++
638 meanZ= -0.468318;
639 if(charge>0)
640 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)) ;
641 else
642 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)) ;
643 }
644
645 Double_t rz=(dz-meanZ)/sz ;
646 Double_t rx=(dx-meanX)/sx ;
647 return TMath::Sqrt(rx*rx+rz*rz) ;
648}
649
650//________________________________________________________________________
651Bool_t AliPHOSTenderSupply::IsGoodChannel(Int_t mod, Int_t ix, Int_t iz)
652{
653 //Check if this channel belogs to the good ones
654
655 if(mod>4 || mod<1){
656// AliError(Form("No bad map for PHOS module %d ",mod)) ;
657 return kTRUE ;
658 }
659 if(!fPHOSBadMap[mod]){
660// AliError(Form("No Bad map for PHOS module %d",mod)) ;
661 return kTRUE ;
662 }
663 if(fPHOSBadMap[mod]->GetBinContent(ix,iz)>0)
664 return kFALSE ;
665 else
666 return kTRUE ;
667}
668//________________________________________________________________________
669void AliPHOSTenderSupply::ForceUsingBadMap(const char * filename){
670 //Read TH2I histograms with bad maps from local or alien file
671 TFile * fbm = TFile::Open(filename) ;
672 if(!fbm || !fbm->IsOpen()){
673 AliError(Form("Can not open BadMaps file %s",filename)) ;
674 return ;
675 }
676 gROOT->cd() ;
677 char key[55] ;
678 for(Int_t mod=1;mod<4; mod++){
679 snprintf(key,55,"PHOS_BadMap_mod%d",mod) ;
680 TH2I * h = (TH2I*)fbm->Get(key) ;
681 if(h)
682 fPHOSBadMap[mod] = new TH2I(*h) ;
683 }
684 fbm->Close() ;
685 fUsePrivateBadMap=kTRUE ;
686}
687//________________________________________________________________________
688void AliPHOSTenderSupply::ForceUsingCalibration(const char * filename){
689 //Read PHOS recalibration parameters from the file.
690 //We assume that file contains single entry: AliPHOSCalibData
691 TFile * fc = TFile::Open(filename) ;
692 if(!fc || !fc->IsOpen()){
693 AliFatal(Form("Can not open Calibration file %s",filename)) ;
694 return ;
695 }
696 fPHOSCalibData = (AliPHOSCalibData*)fc->Get("PHOSCalibration") ;
697 fc->Close() ;
698 fUsePrivateCalib=kTRUE;
699}
700//________________________________________________________________________
701void AliPHOSTenderSupply::CorrectPHOSMisalignment(TVector3 &global,Int_t mod){
702 //Correct for PHOS modules misalignment
703
704 //correct misalignment
705 const Float_t shiftX[6]={0.,-2.3,-2.11,-1.53,0.,0.} ;
706 const Float_t shiftZ[6]={0.,-0.4, 0.52, 0.8,0.,0.} ;
707 TVector3 localPos ;
708 fPHOSGeo->Global2Local(localPos,global,mod) ;
709 fPHOSGeo->Local2Global(mod,localPos.X()+shiftX[mod],localPos.Z()+shiftZ[mod],global);
710}
711//________________________________________________________________________
712void AliPHOSTenderSupply::EvalLambdas(AliVCluster * clu, Double_t &m02, Double_t &m20){
713 //calculate dispecrsion of the cluster in the circle with radius distanceCut around the maximum
714
715 const Double_t rCut=4.5 ;
716
717 Double32_t * elist = clu->GetCellsAmplitudeFraction() ;
718// Calculates the center of gravity in the local PHOS-module coordinates
719 Float_t wtot = 0;
720 Double_t xc[100]={0} ;
721 Double_t zc[100]={0} ;
722 Double_t x = 0 ;
723 Double_t z = 0 ;
724 Int_t mulDigit=TMath::Min(100,clu->GetNCells()) ;
725 const Double_t logWeight=4.5 ;
726 for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) {
727 Int_t relid[4] ;
728 Float_t xi ;
729 Float_t zi ;
730 fPHOSGeo->AbsToRelNumbering(clu->GetCellAbsId(iDigit), relid) ;
731 fPHOSGeo->RelPosInModule(relid, xi, zi);
732 xc[iDigit]=xi ;
733 zc[iDigit]=zi ;
734 if (clu->E()>0 && elist[iDigit]>0) {
735 Float_t w = TMath::Max( 0., logWeight + TMath::Log( elist[iDigit] / clu->E() ) ) ;
736 x += xc[iDigit] * w ;
737 z += zc[iDigit] * w ;
738 wtot += w ;
739 }
740 }
741 if (wtot>0) {
742 x /= wtot ;
743 z /= wtot ;
744 }
745
746 wtot = 0. ;
747 Double_t dxx = 0.;
748 Double_t dzz = 0.;
749 Double_t dxz = 0.;
750 Double_t xCut = 0. ;
751 Double_t zCut = 0. ;
752 for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) {
753 if (clu->E()>0 && elist[iDigit]>0.) {
754 Double_t w = TMath::Max( 0., logWeight + TMath::Log( elist[iDigit] / clu->E() ) ) ;
755 Double_t xi= xc[iDigit] ;
756 Double_t zi= zc[iDigit] ;
757 if((xi-x)*(xi-x)+(zi-z)*(zi-z) < rCut*rCut){
758 xCut += w * xi ;
759 zCut += w * zi ;
760 dxx += w * xi * xi ;
761 dzz += w * zi * zi ;
762 dxz += w * xi * zi ;
763 wtot += w ;
764 }
765 }
766
767 }
768 if (wtot>0) {
769 xCut/= wtot ;
770 zCut/= wtot ;
771 dxx /= wtot ;
772 dzz /= wtot ;
773 dxz /= wtot ;
774 dxx -= xCut * xCut ;
775 dzz -= zCut * zCut ;
776 dxz -= xCut * zCut ;
777
778 m02 = 0.5 * (dxx + dzz) + TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz ) ;
779 m20 = 0.5 * (dxx + dzz) - TMath::Sqrt( 0.25 * (dxx - dzz) * (dxx - dzz) + dxz * dxz ) ;
780 }
781 else {
782 m20=m02=0.;
783 }
784
785}
5bd0f028 786//____________________________________________________________________________
787Double_t AliPHOSTenderSupply::CoreEnergy(AliVCluster * clu){
788 //calculate energy of the cluster in the circle with radius distanceCut around the maximum
789
790 //Can not use already calculated coordinates?
791 //They have incidence correction...
792 const Double_t distanceCut =3.5 ;
793 const Double_t logWeight=4.5 ;
794
795 Double32_t * elist = clu->GetCellsAmplitudeFraction() ;
796// Calculates the center of gravity in the local PHOS-module coordinates
797 Float_t wtot = 0;
798 Double_t xc[100]={0} ;
799 Double_t zc[100]={0} ;
800 Double_t x = 0 ;
801 Double_t z = 0 ;
802 Int_t mulDigit=TMath::Min(100,clu->GetNCells()) ;
803 for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) {
804 Int_t relid[4] ;
805 Float_t xi ;
806 Float_t zi ;
807 fPHOSGeo->AbsToRelNumbering(clu->GetCellAbsId(iDigit), relid) ;
808 fPHOSGeo->RelPosInModule(relid, xi, zi);
809 xc[iDigit]=xi ;
810 zc[iDigit]=zi ;
811 if (clu->E()>0 && elist[iDigit]>0) {
812 Float_t w = TMath::Max( 0., logWeight + TMath::Log( elist[iDigit] / clu->E() ) ) ;
813 x += xc[iDigit] * w ;
814 z += zc[iDigit] * w ;
815 wtot += w ;
816 }
817 }
818 if (wtot>0) {
819 x /= wtot ;
820 z /= wtot ;
821 }
822 Double_t coreE=0. ;
823 for(Int_t iDigit=0; iDigit < mulDigit; iDigit++) {
824 Double_t distance = TMath::Sqrt((xc[iDigit]-x)*(xc[iDigit]-x)+(zc[iDigit]-z)*(zc[iDigit]-z)) ;
825 if(distance < distanceCut)
826 coreE += elist[iDigit] ;
827 }
828 //Apply non-linearity correction
829 return coreE ;
830}
53442b36 831//____________________________________________________________________________
832Double_t AliPHOSTenderSupply::EvalEcross(AliVCluster * clu){
833 //Calculate propoerion of the cluster energy in cross around the
834 //cell with maximal energy deposition. Can be used to reject exotic clusters
835
836 Double32_t * elist = clu->GetCellsAmplitudeFraction() ;
837 Int_t mulDigit=clu->GetNCells() ;
838 // Calculates the center of gravity in the local PHOS-module coordinates
839 //Find cell with max E
840 Double_t eMax=0.;
841 Int_t iMax=0;
842 for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) {
843 if(elist[iDigit]>eMax){
844 eMax=elist[iDigit] ;
845 iMax=iDigit ;
846 }
847 }
848 //Calculate e in cross
849 Double_t eCross=0 ;
850 Int_t relidMax[4] ;
851 fPHOSGeo->AbsToRelNumbering(clu->GetCellAbsId(iMax), relidMax) ;
852 Int_t relid[4] ;
853 for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) {
854 fPHOSGeo->AbsToRelNumbering(clu->GetCellAbsId(iDigit), relid) ;
855 if(TMath::Abs(relid[2]-relidMax[2])+TMath::Abs(relid[3]-relidMax[3])==1)
856 eCross+= elist[iDigit] ;
857 }
858 if(eMax>0)
859 return 1.-eCross/eMax ;
860 else
861 return 0 ;
862}
863
864
f7d0ec90 865//________________________________________________________________________
866Double_t AliPHOSTenderSupply::EvalTOF(AliVCluster * clu,AliVCaloCells * cells){
867 //Evaluate TOF of the cluster after re-calibration
868 //TOF here is weighted average of digits
869 // -within 50ns from the most energetic cell
870 // -not too soft.
1e7fec1c 871
f7d0ec90 872 Double32_t * elist = clu->GetCellsAmplitudeFraction() ;
873 Int_t mulDigit=clu->GetNCells() ;
874
875 Float_t tMax= 0.; //Time at the maximum
876 Float_t eMax=0. ;
877 for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) {
878 Int_t absId=clu->GetCellAbsId(iDigit) ;
abd8ae50 879 Bool_t isHG=cells->GetCellHighGain(absId) ;
f7d0ec90 880 if( elist[iDigit]>eMax){
881 tMax=CalibrateTOF(cells->GetCellTime(absId),absId,isHG) ;
882 eMax=elist[iDigit] ;
883 }
884 }
f7d0ec90 885
886 //Try to improve accuracy
887 //Do not account time of soft cells:
888 // const Double_t part=0.5 ;
889 Double_t eMin=TMath::Min(0.5,0.2*eMax) ;
890 Float_t wtot = 0.;
891 Double_t t = 0. ;
892 for(Int_t iDigit=0; iDigit<mulDigit; iDigit++) {
893 Int_t absId=clu->GetCellAbsId(iDigit) ;
abd8ae50 894 Bool_t isHG=cells->GetCellHighGain(absId) ;
1e7fec1c 895
f7d0ec90 896 Double_t ti=CalibrateTOF(cells->GetCellTime(absId),absId,isHG) ;
897 if(TMath::Abs(ti-tMax)>50.e-9) //remove soft cells with wrong time
898 continue ;
899
900 //Remove too soft cells
901 if(elist[iDigit]<eMin)
902 continue ;
903
904 if(elist[iDigit]>0){
905 //weight = 1./sigma^2
906 //Sigma is parameterization of TOF resolution 16.05.2013
907 Double_t wi2=0.;
908 if(isHG)
1e7fec1c 909 wi2=1./(2.4 + 3.9/elist[iDigit]) ;
f7d0ec90 910 else
1e7fec1c 911 wi2=1./(2.4 + 3.9/(0.1*elist[iDigit])) ; //E of LG digit is 1/16 of correcponding HG
f7d0ec90 912 t+=ti*wi2 ;
913 wtot+=wi2 ;
914 }
915 }
916 if(wtot>0){
917 t=t/wtot ;
1e7fec1c 918 }
919 else{
920 t=tMax ;
f7d0ec90 921 }
922
923 return t ;
924
925}
926//________________________________________________________________________
927Double_t AliPHOSTenderSupply::CalibrateTOF(Double_t tof, Int_t absId, Bool_t isHG){
928 //Apply time re-calibration separately for HG and LG channels
929 //By default (if not filled) shifts are zero.
930
931 Int_t relId[4];
932 fPHOSGeo->AbsToRelNumbering(absId,relId) ;
933 Int_t module = relId[0];
934 Int_t column = relId[3];
935 Int_t row = relId[2];
936 if(isHG)
937 tof-=fPHOSCalibData->GetTimeShiftEmc(module, column, row);
1e7fec1c 938 else{
f7d0ec90 939 tof-=fPHOSCalibData->GetLGTimeShiftEmc(module, column, row);
1e7fec1c 940 }
f7d0ec90 941 return tof ;
942
943}
393b3556 944//________________________________________________________________________
945void AliPHOSTenderSupply::DistanceToBadChannel(Int_t mod, TVector3 * locPos, Double_t &minDist){
946 //Check if distance to bad channel was reduced
947 Int_t range = minDist/2.2 +1 ; //Distance at which bad channels should be serached
948
949 Int_t relid[4]={0,0,0,0} ;
950 fPHOSGeo->RelPosToRelId(mod, locPos->X(), locPos->Z(), relid) ;
951 Int_t xmin=TMath::Max(1,relid[2]-range) ;
952 Int_t xmax=TMath::Min(64,relid[2]+range) ;
953 Int_t zmin=TMath::Max(1,relid[3]-range) ;
954 Int_t zmax=TMath::Min(56,relid[3]+range) ;
955
956 Float_t x=0.,z=0.;
957 for(Int_t ix=xmin;ix<=xmax;ix++){
958 for(Int_t iz=zmin;iz<=zmax;iz++){
959 if(fPHOSBadMap[mod]->GetBinContent(ix,iz)>0){ //Bad channel
960 Int_t relidBC[4]={mod,0,ix,iz} ;
961 fPHOSGeo->RelPosInModule(relidBC,x,z);
962 Double_t dist = TMath::Sqrt((x-locPos->X())*(x-locPos->X()) + (z-locPos->Z())*(z-locPos->Z()));
963 if(dist<minDist) minDist = dist;
964 }
965 }
966 }
967
968}
969
e1c158c2 970