1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
16 /* $Id: AliPIDResponse.cxx 46193 2010-12-21 09:00:14Z wiechula $ */
18 //-----------------------------------------------------------------
19 // Base class for handling the pid response //
20 // functions of all detectors //
21 // and give access to the nsigmas //
23 // Origin: Jens Wiechula, Uni Tuebingen, jens.wiechula@cern.ch //
24 //-----------------------------------------------------------------
27 #include <TObjArray.h>
35 #include <TLinearFitter.h>
39 #include <AliVEvent.h>
40 #include <AliVTrack.h>
43 #include <AliOADBContainer.h>
44 #include <AliTRDPIDResponseObject.h>
45 #include <AliTRDdEdxParams.h>
46 #include <AliTOFPIDParams.h>
47 #include <AliHMPIDPIDParams.h>
49 #include "AliPIDResponse.h"
50 #include "AliDetectorPID.h"
52 #include "AliCentrality.h"
54 ClassImp(AliPIDResponse);
56 Float_t AliPIDResponse::fgTOFmismatchProb = 0.0;
58 AliPIDResponse::AliPIDResponse(Bool_t isMC/*=kFALSE*/) :
59 TNamed("PIDResponse","PIDResponse"),
67 fITSPIDmethod(kITSTruncMean),
68 fTuneMConData(kFALSE),
69 fTuneMConDataMask(kDetTOF|kDetTPC),
73 fCustomTPCpidResponse(),
80 fCurrentAliRootRev(-1),
88 fArrPidResponseMaster(NULL),
89 fResolutionCorrection(NULL),
90 fOADBvoltageMaps(NULL),
91 fUseTPCEtaCorrection(kFALSE),
92 fUseTPCMultiplicityCorrection(kFALSE),
93 fTRDPIDResponseObject(NULL),
97 fHMPIDPIDParams(NULL),
98 fEMCALPIDParams(NULL),
100 fCurrCentrality(0.0),
107 AliLog::SetClassDebugLevel("AliPIDResponse",0);
108 AliLog::SetClassDebugLevel("AliESDpid",0);
109 AliLog::SetClassDebugLevel("AliAODpidUtil",0);
113 //______________________________________________________________________________
114 AliPIDResponse::~AliPIDResponse()
119 delete fArrPidResponseMaster;
120 delete fTRDPIDResponseObject;
121 delete fTRDdEdxParams;
122 delete fTOFPIDParams;
125 //______________________________________________________________________________
126 AliPIDResponse::AliPIDResponse(const AliPIDResponse &other) :
128 fITSResponse(other.fITSResponse),
129 fTPCResponse(other.fTPCResponse),
130 fTRDResponse(other.fTRDResponse),
131 fTOFResponse(other.fTOFResponse),
132 fHMPIDResponse(other.fHMPIDResponse),
133 fEMCALResponse(other.fEMCALResponse),
134 fRange(other.fRange),
135 fITSPIDmethod(other.fITSPIDmethod),
136 fTuneMConData(other.fTuneMConData),
137 fTuneMConDataMask(other.fTuneMConDataMask),
139 fCachePID(other.fCachePID),
140 fOADBPath(other.fOADBPath),
141 fCustomTPCpidResponse(other.fCustomTPCpidResponse),
142 fCustomTPCetaMaps(other.fCustomTPCetaMaps),
146 fMCperiodUser(other.fMCperiodUser),
148 fCurrentAliRootRev(other.fCurrentAliRootRev),
150 fRecoPassUser(other.fRecoPassUser),
156 fArrPidResponseMaster(NULL),
157 fResolutionCorrection(NULL),
158 fOADBvoltageMaps(NULL),
159 fUseTPCEtaCorrection(other.fUseTPCEtaCorrection),
160 fUseTPCMultiplicityCorrection(other.fUseTPCMultiplicityCorrection),
161 fTRDPIDResponseObject(NULL),
162 fTRDdEdxParams(NULL),
165 fHMPIDPIDParams(NULL),
166 fEMCALPIDParams(NULL),
168 fCurrCentrality(0.0),
170 fNoTOFmism(other.fNoTOFmism)
177 //______________________________________________________________________________
178 AliPIDResponse& AliPIDResponse::operator=(const AliPIDResponse &other)
184 delete fArrPidResponseMaster;
185 TNamed::operator=(other);
186 fITSResponse=other.fITSResponse;
187 fTPCResponse=other.fTPCResponse;
188 fTRDResponse=other.fTRDResponse;
189 fTOFResponse=other.fTOFResponse;
190 fHMPIDResponse=other.fHMPIDResponse;
191 fEMCALResponse=other.fEMCALResponse;
193 fITSPIDmethod=other.fITSPIDmethod;
194 fOADBPath=other.fOADBPath;
195 fCustomTPCpidResponse=other.fCustomTPCpidResponse;
196 fCustomTPCetaMaps=other.fCustomTPCetaMaps;
197 fTuneMConData=other.fTuneMConData;
198 fTuneMConDataMask=other.fTuneMConDataMask;
200 fCachePID=other.fCachePID;
205 fMCperiodUser=other.fMCperiodUser;
207 fCurrentAliRootRev=other.fCurrentAliRootRev;
209 fRecoPassUser=other.fRecoPassUser;
215 fArrPidResponseMaster=NULL;
216 fResolutionCorrection=NULL;
217 fOADBvoltageMaps=NULL;
218 fUseTPCEtaCorrection=other.fUseTPCEtaCorrection;
219 fUseTPCMultiplicityCorrection=other.fUseTPCMultiplicityCorrection;
220 fTRDPIDResponseObject=NULL;
222 fEMCALPIDParams=NULL;
225 fHMPIDPIDParams=NULL;
226 fCurrentEvent=other.fCurrentEvent;
227 fNoTOFmism = other.fNoTOFmism;
233 //______________________________________________________________________________
234 Float_t AliPIDResponse::NumberOfSigmas(EDetector detector, const AliVParticle *vtrack, AliPID::EParticleType type) const
237 // NumberOfSigmas for 'detCode'
240 const AliVTrack *track=static_cast<const AliVTrack*>(vtrack);
241 // look for cached value first
242 const AliDetectorPID *detPID=track->GetDetectorPID();
244 if ( detPID && detPID->HasNumberOfSigmas(detector)){
245 return detPID->GetNumberOfSigmas(detector, type);
246 } else if (fCachePID) {
247 FillTrackDetectorPID(track, detector);
248 detPID=track->GetDetectorPID();
249 return detPID->GetNumberOfSigmas(detector, type);
252 return GetNumberOfSigmas(detector, track, type);
255 //______________________________________________________________________________
256 AliPIDResponse::EDetPidStatus AliPIDResponse::NumberOfSigmas(EDetector detCode, const AliVParticle *track,
257 AliPID::EParticleType type, Double_t &val) const
260 // NumberOfSigmas with detector status as return value
263 val=NumberOfSigmas(detCode, track, type);
264 return CheckPIDStatus(detCode, (AliVTrack*)track);
267 //______________________________________________________________________________
268 // public buffered versions of the PID calculation
271 //______________________________________________________________________________
272 Float_t AliPIDResponse::NumberOfSigmasITS(const AliVParticle *vtrack, AliPID::EParticleType type) const
275 // Calculate the number of sigmas in the ITS
278 return NumberOfSigmas(kITS, vtrack, type);
281 //______________________________________________________________________________
282 Float_t AliPIDResponse::NumberOfSigmasTPC(const AliVParticle *vtrack, AliPID::EParticleType type) const
285 // Calculate the number of sigmas in the TPC
288 return NumberOfSigmas(kTPC, vtrack, type);
291 //______________________________________________________________________________
292 Float_t AliPIDResponse::NumberOfSigmasTPC( const AliVParticle *vtrack,
293 AliPID::EParticleType type,
294 AliTPCPIDResponse::ETPCdEdxSource dedxSource) const
296 //get number of sigmas according the selected TPC gain configuration scenario
297 const AliVTrack *track=static_cast<const AliVTrack*>(vtrack);
299 Float_t nSigma=fTPCResponse.GetNumberOfSigmas(track, type, dedxSource, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
304 //______________________________________________________________________________
305 Float_t AliPIDResponse::NumberOfSigmasTRD(const AliVParticle *vtrack, AliPID::EParticleType type) const
308 // Calculate the number of sigmas in the TRD
310 return NumberOfSigmas(kTRD, vtrack, type);
313 //______________________________________________________________________________
314 Float_t AliPIDResponse::NumberOfSigmasTOF(const AliVParticle *vtrack, AliPID::EParticleType type) const
317 // Calculate the number of sigmas in the TOF
320 return NumberOfSigmas(kTOF, vtrack, type);
323 //______________________________________________________________________________
324 Float_t AliPIDResponse::NumberOfSigmasHMPID(const AliVParticle *vtrack, AliPID::EParticleType type) const
327 // Calculate the number of sigmas in the EMCAL
330 return NumberOfSigmas(kHMPID, vtrack, type);
333 //______________________________________________________________________________
334 Float_t AliPIDResponse::NumberOfSigmasEMCAL(const AliVParticle *vtrack, AliPID::EParticleType type) const
337 // Calculate the number of sigmas in the EMCAL
340 return NumberOfSigmas(kEMCAL, vtrack, type);
343 //______________________________________________________________________________
344 Float_t AliPIDResponse::NumberOfSigmasEMCAL(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &eop, Double_t showershape[4]) const
347 // emcal nsigma with eop and showershape
349 AliVTrack *track=(AliVTrack*)vtrack;
351 AliVCluster *matchedClus = NULL;
356 Double_t fClsE = -1.;
358 // initialize eop and shower shape parameters
360 for(Int_t i = 0; i < 4; i++){
361 showershape[i] = -1.;
364 Int_t nMatchClus = -1;
368 nMatchClus = track->GetEMCALcluster();
373 charge = track->Charge();
375 matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
379 // matched cluster is EMCAL
380 if(matchedClus->IsEMCAL()){
382 fClsE = matchedClus->E();
385 // fill used EMCAL variables here
387 showershape[0] = matchedClus->GetNCells(); // number of cells in cluster
388 showershape[1] = matchedClus->GetM02(); // long axis
389 showershape[2] = matchedClus->GetM20(); // short axis
390 showershape[3] = matchedClus->GetDispersion(); // dispersion
392 // look for cached value first
393 const AliDetectorPID *detPID=track->GetDetectorPID();
394 const EDetector detector=kEMCAL;
396 if ( detPID && detPID->HasNumberOfSigmas(detector)){
397 return detPID->GetNumberOfSigmas(detector, type);
398 } else if (fCachePID) {
399 FillTrackDetectorPID(track, detector);
400 detPID=track->GetDetectorPID();
401 return detPID->GetNumberOfSigmas(detector, type);
404 // NSigma value really meaningful only for electrons!
405 return fEMCALResponse.GetNumberOfSigmas(pt,EovP,type,charge);
412 //______________________________________________________________________________
413 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDelta(EDetector detector, const AliVParticle *track, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
420 case kITS: return GetSignalDeltaITS(track,type,val,ratio); break;
421 case kTPC: return GetSignalDeltaTPC(track,type,val,ratio); break;
422 case kTRD: return GetSignalDeltaTRD(track,type,val,ratio); break;
423 case kTOF: return GetSignalDeltaTOF(track,type,val,ratio); break;
424 case kHMPID: return GetSignalDeltaHMPID(track,type,val,ratio); break;
425 default: return kDetNoSignal;
430 //______________________________________________________________________________
431 Double_t AliPIDResponse::GetSignalDelta(EDetector detCode, const AliVParticle *track, AliPID::EParticleType type, Bool_t ratio/*=kFALSE*/) const
437 EDetPidStatus stat=GetSignalDelta(detCode, track, type, val, ratio);
438 if ( stat==kDetNoSignal ) val=-9999.;
442 //______________________________________________________________________________
443 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputePIDProbability (EDetCode detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
445 // Compute PID response of 'detCode'
447 // find detector code from detector bit mask
449 for (Int_t idet=0; idet<kNdetectors; ++idet) if ( (detCode&(1<<idet)) ) { detector=idet; break; }
450 if (detector==-1) return kDetNoSignal;
452 return ComputePIDProbability((EDetector)detector, track, nSpecies, p);
455 //______________________________________________________________________________
456 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputePIDProbability (EDetector detector, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
459 // Compute PID response of 'detector'
462 const AliDetectorPID *detPID=track->GetDetectorPID();
464 if ( detPID && detPID->HasRawProbability(detector)){
465 return detPID->GetRawProbability(detector, p, nSpecies);
466 } else if (fCachePID) {
467 FillTrackDetectorPID(track, detector);
468 detPID=track->GetDetectorPID();
469 return detPID->GetRawProbability(detector, p, nSpecies);
472 //if no caching return values calculated from scratch
473 return GetComputePIDProbability(detector, track, nSpecies, p);
476 //______________________________________________________________________________
477 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeITSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
479 // Compute PID response for the ITS
480 return ComputePIDProbability(kITS, track, nSpecies, p);
483 //______________________________________________________________________________
484 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTPCProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
486 // Compute PID response for the TPC
487 return ComputePIDProbability(kTPC, track, nSpecies, p);
490 //______________________________________________________________________________
491 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTOFProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
493 // Compute PID response for the
494 return ComputePIDProbability(kTOF, track, nSpecies, p);
497 //______________________________________________________________________________
498 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
500 // Compute PID response for the
501 return ComputePIDProbability(kTRD, track, nSpecies, p);
504 //______________________________________________________________________________
505 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeEMCALProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
507 // Compute PID response for the EMCAL
508 return ComputePIDProbability(kEMCAL, track, nSpecies, p);
510 //______________________________________________________________________________
511 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputePHOSProbability (const AliVTrack */*track*/, Int_t nSpecies, Double_t p[]) const
513 // Compute PID response for the PHOS
515 // set flat distribution (no decision)
516 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
520 //______________________________________________________________________________
521 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeHMPIDProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
523 // Compute PID response for the HMPID
524 return ComputePIDProbability(kHMPID, track, nSpecies, p);
527 //______________________________________________________________________________
528 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const
530 // Compute PID response for the
531 return GetComputeTRDProbability(track, nSpecies, p, PIDmethod);
534 //______________________________________________________________________________
535 AliPIDResponse::EDetPidStatus AliPIDResponse::CheckPIDStatus(EDetector detector, const AliVTrack *track) const
537 // calculate detector pid status
539 const Int_t iDetCode=(Int_t)detector;
540 if (iDetCode<0||iDetCode>=kNdetectors) return kDetNoSignal;
541 const AliDetectorPID *detPID=track->GetDetectorPID();
544 return detPID->GetPIDStatus(detector);
545 } else if (fCachePID) {
546 FillTrackDetectorPID(track, detector);
547 detPID=track->GetDetectorPID();
548 return detPID->GetPIDStatus(detector);
551 // if not buffered and no buffering is requested
552 return GetPIDStatus(detector, track);
555 //______________________________________________________________________________
556 void AliPIDResponse::InitialiseEvent(AliVEvent *event, Int_t pass, Int_t run)
559 // Apply settings for the current event
568 else fRun=event->GetRunNumber();
575 //TPC resolution parametrisation PbPb
576 if ( fResolutionCorrection ){
577 Double_t corrSigma=fResolutionCorrection->Eval(GetTPCMultiplicityBin(event));
578 fTPCResponse.SetSigma(3.79301e-03*corrSigma, 2.21280e+04);
581 // Set up TPC multiplicity for PbPb
582 if (fUseTPCMultiplicityCorrection) {
583 Int_t numESDtracks = event->GetNumberOfESDTracks();
584 if (numESDtracks < 0) {
585 AliError("Cannot obtain event multiplicity (number of ESD tracks < 0). If you are using AODs, this might be a too old production. Please disable the multiplicity correction to get a reliable PID result!");
588 fTPCResponse.SetCurrentEventMultiplicity(numESDtracks);
591 fTPCResponse.SetCurrentEventMultiplicity(0);
594 SetTOFResponse(event, (AliPIDResponse::EStartTimeType_t)fTOFPIDParams->GetStartTimeMethod());
597 // Get and set centrality
598 AliCentrality *centrality = event->GetCentrality();
600 fCurrCentrality = centrality->GetCentralityPercentile("V0M");
603 fCurrCentrality = -1;
606 // Set centrality percentile for EMCAL
607 fEMCALResponse.SetCentrality(fCurrCentrality);
609 // switch off some TOF channel according to OADB to match data TOF matching eff
610 if (fTuneMConData && ((fTuneMConDataMask & kDetTOF) == kDetTOF) && fTOFPIDParams->GetTOFmatchingLossMC() > 0.01){
611 Int_t ntrk = event->GetNumberOfTracks();
612 for(Int_t i=0;i < ntrk;i++){
613 AliVParticle *trk = event->GetTrack(i);
614 Int_t channel = GetTOFResponse().GetTOFchannel(trk);
615 Int_t swoffEachOfThem = Int_t(100./fTOFPIDParams->GetTOFmatchingLossMC() + 0.5);
616 if(!(channel%swoffEachOfThem)) ((AliVTrack *) trk)->ResetStatus(AliVTrack::kTOFout);
622 //______________________________________________________________________________
623 void AliPIDResponse::ExecNewRun()
626 // Things to Execute upon a new run
630 SetITSParametrisation();
632 SetTPCPidResponseMaster();
633 SetTPCParametrisation();
636 SetTRDPidResponseMaster();
637 //has to precede InitializeTRDResponse(), otherwise the read-out fTRDdEdxParams is not pased in TRDResponse!
639 InitializeTRDResponse();
641 SetEMCALPidResponseMaster();
642 InitializeEMCALResponse();
644 SetTOFPidResponseMaster();
645 InitializeTOFResponse();
647 SetHMPIDPidResponseMaster();
648 InitializeHMPIDResponse();
650 if (fCurrentEvent) fTPCResponse.SetMagField(fCurrentEvent->GetMagneticField());
653 //______________________________________________________________________________
654 Double_t AliPIDResponse::GetTPCMultiplicityBin(const AliVEvent * const event)
657 // Get TPC multiplicity in bins of 150
660 const AliVVertex* vertexTPC = event->GetPrimaryVertex();
661 Double_t tpcMulti=0.;
663 Double_t vertexContribTPC=vertexTPC->GetNContributors();
664 tpcMulti=vertexContribTPC/150.;
665 if (tpcMulti>20.) tpcMulti=20.;
671 //______________________________________________________________________________
672 void AliPIDResponse::SetRecoInfo()
675 // Set reconstruction information
687 Bool_t hasProdInfo=(fCurrentFile.BeginsWith("LHC"));
689 TPRegexp reg(".*(LHC1[1-3][a-z]+[0-9]+[a-z_]*)[/_].*");
690 if (hasProdInfo) reg=TPRegexp("LHC1[1-2][a-z]+[0-9]+[a-z_]*");
691 TPRegexp reg12a17("LHC1[2-4][a-z]");
693 //find the period by run number (UGLY, but not stored in ESD and AOD... )
694 if (fRun>=114737&&fRun<=117223) { fLHCperiod="LHC10B"; fMCperiodTPC="LHC10D1"; }
695 else if (fRun>=118503&&fRun<=121040) { fLHCperiod="LHC10C"; fMCperiodTPC="LHC10D1"; }
696 else if (fRun>=122195&&fRun<=126437) { fLHCperiod="LHC10D"; fMCperiodTPC="LHC10F6A"; }
697 else if (fRun>=127710&&fRun<=130850) { fLHCperiod="LHC10E"; fMCperiodTPC="LHC10F6A"; }
698 else if (fRun>=133004&&fRun<=135029) { fLHCperiod="LHC10F"; fMCperiodTPC="LHC10F6A"; }
699 else if (fRun>=135654&&fRun<=136377) { fLHCperiod="LHC10G"; fMCperiodTPC="LHC10F6A"; }
700 else if (fRun>=136851&&fRun<=139846) {
702 fMCperiodTPC="LHC10H8";
703 if (reg.MatchB(fCurrentFile)) fMCperiodTPC="LHC11A10";
704 // exception for 13d2 and later
705 if (fCurrentAliRootRev >= 62714) fMCperiodTPC="LHC13D2";
709 else if (fRun>=139847&&fRun<=146974) { fLHCperiod="LHC11A"; fMCperiodTPC="LHC10F6A"; }
710 //TODO: periods 11B (146975-150721), 11C (150722-155837) are not yet treated assume 11d for the moment
711 else if (fRun>=146975&&fRun<=155837) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
712 else if (fRun>=155838&&fRun<=159649) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
713 // also for 11e (159650-162750),f(162751-165771) use 11d
714 else if (fRun>=159650&&fRun<=162750) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
715 else if (fRun>=162751&&fRun<=165771) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
717 else if (fRun>=165772 && fRun<=170718) {
719 fMCperiodTPC="LHC11A10";
722 if (reg12a17.MatchB(fCurrentFile)) fMCperiodTPC="LHC12A17";
724 if (fRun>=170719 && fRun<=177311) {
728 fMCperiodTPC="LHC10F6A";
729 if (fCurrentAliRootRev >= 62714)
730 fMCperiodTPC="LHC14E2";
732 // for the moment use LHC12b parameters up to LHC12d
733 if (fRun>=177312 /*&& fRun<=179356*/) {
737 fMCperiodTPC="LHC10F6A";
738 if (fCurrentAliRootRev >= 62714)
739 fMCperiodTPC="LHC14E2";
741 // if (fRun>=179357 && fRun<=183173) { fLHCperiod="LHC12C"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
742 // if (fRun>=183174 && fRun<=186345) { fLHCperiod="LHC12D"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
743 // if (fRun>=186346 && fRun<=186635) { fLHCperiod="LHC12E"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
745 // if (fRun>=186636 && fRun<=188166) { fLHCperiod="LHC12F"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
746 // if (fRun >= 188167 && fRun <= 188355 ) { fLHCperiod="LHC12G"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
747 // if (fRun >= 188356 && fRun <= 188503 ) { fLHCperiod="LHC12G"; fBeamType="PPB"; fBeamTypeNum=kPPB;/*fMCperiodTPC="";*/ }
748 // for the moment use 12g parametrisation for all full gain runs (LHC12e+)
749 if (fRun >= 186346 && fRun < 188719) { fLHCperiod="LHC12G"; fBeamType="PPB";fBeamTypeNum=kPPB; fMCperiodTPC="LHC12G"; }
751 // Dedicated splines for periods 12g and 12i(j) (and use more appropriate MC)
752 if (fRun >= 188720 && fRun <= 192738) {
756 fMCperiodTPC="LHC10F6A";
757 if (fCurrentAliRootRev >= 62714)
758 fMCperiodTPC="LHC13B2_FIXn1";
760 if (fRun >= 192739 && fRun <= 194479) {
764 fMCperiodTPC="LHC10F6A";
765 if (fCurrentAliRootRev >= 62714)
766 fMCperiodTPC="LHC13B2_FIXn1";
769 // New parametrisation for 2013 pPb runs
770 if (fRun >= 194480) {
774 fMCperiodTPC="LHC12G";
776 if (fCurrentAliRootRev >= 61605)
777 fMCperiodTPC="LHC13B2_FIX";
778 if (fCurrentAliRootRev >= 62714)
779 fMCperiodTPC="LHC13B2_FIXn1";
781 // High luminosity pPb runs require different parametrisations
782 if (fRun >= 195875 && fRun <= 197411) {
787 //exception new pp MC productions from 2011 (11a periods have 10f6a splines!)
788 if (fBeamType=="PP" && reg.MatchB(fCurrentFile) && !fCurrentFile.Contains("LHC11a")) { fMCperiodTPC="LHC11B2"; fBeamType="PP";fBeamTypeNum=kPP; }
789 // exception for 11f1
790 if (fCurrentFile.Contains("LHC11f1")) fMCperiodTPC="LHC11F1";
791 // exception for 12f1a, 12f1b and 12i3
792 if (fCurrentFile.Contains("LHC12f1") || fCurrentFile.Contains("LHC12i3")) fMCperiodTPC="LHC12F1";
793 // exception for 12c4
794 if (fCurrentFile.Contains("LHC12c4")) fMCperiodTPC="LHC12C4";
795 // exception for 13d1 11d anchored prod
796 if (fLHCperiod=="LHC11D" && fCurrentFile.Contains("LHC13d1")) fMCperiodTPC="LHC13D1";
799 //______________________________________________________________________________
800 void AliPIDResponse::SetITSParametrisation()
803 // Set the ITS parametrisation
808 //______________________________________________________________________________
809 void AliPIDResponse::AddPointToHyperplane(TH2D* h, TLinearFitter* linExtrapolation, Int_t binX, Int_t binY)
811 if (h->GetBinContent(binX, binY) <= 1e-4)
812 return; // Reject bins without content (within some numerical precision) or with strange content
814 Double_t coord[2] = {0, 0};
815 coord[0] = h->GetXaxis()->GetBinCenter(binX);
816 coord[1] = h->GetYaxis()->GetBinCenter(binY);
817 Double_t binError = h->GetBinError(binX, binY);
819 binError = 1000; // Should not happen because bins without content are rejected for the map (TH2D* h)
820 printf("ERROR: This should never happen: Trying to add bin in addPointToHyperplane with error not set....\n");
822 linExtrapolation->AddPoint(coord, h->GetBinContent(binX, binY, binError));
826 //______________________________________________________________________________
827 TH2D* AliPIDResponse::RefineHistoViaLinearInterpolation(TH2D* h, Double_t refineFactorX, Double_t refineFactorY)
832 // Interpolate to finer map
833 TLinearFitter* linExtrapolation = new TLinearFitter(2, "hyp2", "");
835 Double_t upperMapBoundY = h->GetYaxis()->GetBinUpEdge(h->GetYaxis()->GetNbins());
836 Double_t lowerMapBoundY = h->GetYaxis()->GetBinLowEdge(1);
838 // Binning was find to yield good results, if 40 bins are chosen for the range 0.0016 to 0.02. For the new variable range,
839 // scale the number of bins correspondingly
840 Int_t nBinsY = TMath::Nint((upperMapBoundY - lowerMapBoundY) / (0.02 - 0.0016) * 40);
841 Int_t nBinsXrefined = nBinsX * refineFactorX;
842 Int_t nBinsYrefined = nBinsY * refineFactorY;
844 TH2D* hRefined = new TH2D(Form("%s_refined", h->GetName()), Form("%s (refined)", h->GetTitle()),
845 nBinsXrefined, h->GetXaxis()->GetBinLowEdge(1), h->GetXaxis()->GetBinUpEdge(h->GetXaxis()->GetNbins()),
846 nBinsYrefined, lowerMapBoundY, upperMapBoundY);
848 for (Int_t binX = 1; binX <= nBinsXrefined; binX++) {
849 for (Int_t binY = 1; binY <= nBinsYrefined; binY++) {
851 hRefined->SetBinContent(binX, binY, 1); // Default value is 1
853 Double_t centerX = hRefined->GetXaxis()->GetBinCenter(binX);
854 Double_t centerY = hRefined->GetYaxis()->GetBinCenter(binY);
857 linExtrapolation->ClearPoints();
859 // For interpolation: Just take the corresponding bin from the old histo.
860 // For extrapolation: take the last available bin from the old histo.
861 // If the boundaries are to be skipped, also skip the corresponding bins
862 Int_t oldBinX = h->GetXaxis()->FindBin(centerX);
865 if (oldBinX > nBinsX)
868 Int_t oldBinY = h->GetYaxis()->FindBin(centerY);
871 if (oldBinY > nBinsY)
874 // Neighbours left column
877 AddPointToHyperplane(h, linExtrapolation, oldBinX - 1, oldBinY - 1);
880 AddPointToHyperplane(h, linExtrapolation, oldBinX - 1, oldBinY);
882 if (oldBinY < nBinsY) {
883 AddPointToHyperplane(h, linExtrapolation, oldBinX - 1, oldBinY + 1);
887 // Neighbours (and point itself) same column
889 AddPointToHyperplane(h, linExtrapolation, oldBinX, oldBinY - 1);
892 AddPointToHyperplane(h, linExtrapolation, oldBinX, oldBinY);
894 if (oldBinY < nBinsY) {
895 AddPointToHyperplane(h, linExtrapolation, oldBinX, oldBinY + 1);
898 // Neighbours right column
899 if (oldBinX < nBinsX) {
901 AddPointToHyperplane(h, linExtrapolation, oldBinX + 1, oldBinY - 1);
904 AddPointToHyperplane(h, linExtrapolation, oldBinX + 1, oldBinY);
906 if (oldBinY < nBinsY) {
907 AddPointToHyperplane(h, linExtrapolation, oldBinX + 1, oldBinY + 1);
913 if (linExtrapolation->GetNpoints() <= 0)
916 if (linExtrapolation->Eval() != 0)// EvalRobust -> Takes much, much, [...], much more time (~hours instead of seconds)
919 // Fill the bin of the refined histogram with the extrapolated value
920 Double_t interpolatedValue = linExtrapolation->GetParameter(0) + linExtrapolation->GetParameter(1) * centerX
921 + linExtrapolation->GetParameter(2) * centerY;
923 Double_t interpolatedValue = h->Interpolate(centerX, centerY) ;
924 hRefined->SetBinContent(binX, binY, interpolatedValue);
929 // Problem: Interpolation does not work before/beyond center of first/last bin (as the name suggests).
930 // Therefore, for each row in dEdx: Take last bin from old map and interpolate values from center and edge.
931 // Assume line through these points and extropolate to last bin of refined map
932 const Double_t firstOldXbinUpEdge = h->GetXaxis()->GetBinUpEdge(1);
933 const Double_t firstOldXbinCenter = h->GetXaxis()->GetBinCenter(1);
935 const Double_t oldXbinHalfWidth = firstOldXbinUpEdge - firstOldXbinCenter;
937 const Double_t lastOldXbinLowEdge = h->GetXaxis()->GetBinLowEdge(h->GetNbinsX());
938 const Double_t lastOldXbinCenter = h->GetXaxis()->GetBinCenter(h->GetNbinsX());
940 for (Int_t binY = 1; binY <= nBinsYrefined; binY++) {
941 Double_t centerY = hRefined->GetYaxis()->GetBinCenter(binY);
943 const Double_t interpolatedCenterFirstXbin = h->Interpolate(firstOldXbinCenter, centerY);
944 const Double_t interpolatedUpEdgeFirstXbin = h->Interpolate(firstOldXbinUpEdge, centerY);
946 const Double_t extrapolationSlopeFirstXbin = (interpolatedUpEdgeFirstXbin - interpolatedCenterFirstXbin) / oldXbinHalfWidth;
947 const Double_t extrapolationOffsetFirstXbin = interpolatedCenterFirstXbin;
950 const Double_t interpolatedCenterLastXbin = h->Interpolate(lastOldXbinCenter, centerY);
951 const Double_t interpolatedLowEdgeLastXbin = h->Interpolate(lastOldXbinLowEdge, centerY);
953 const Double_t extrapolationSlopeLastXbin = (interpolatedCenterLastXbin - interpolatedLowEdgeLastXbin) / oldXbinHalfWidth;
954 const Double_t extrapolationOffsetLastXbin = interpolatedCenterLastXbin;
956 for (Int_t binX = 1; binX <= nBinsXrefined; binX++) {
957 Double_t centerX = hRefined->GetXaxis()->GetBinCenter(binX);
959 if (centerX < firstOldXbinCenter) {
960 Double_t extrapolatedValue = extrapolationOffsetFirstXbin + (centerX - firstOldXbinCenter) * extrapolationSlopeFirstXbin;
961 hRefined->SetBinContent(binX, binY, extrapolatedValue);
963 else if (centerX <= lastOldXbinCenter) {
967 Double_t extrapolatedValue = extrapolationOffsetLastXbin + (centerX - lastOldXbinCenter) * extrapolationSlopeLastXbin;
968 hRefined->SetBinContent(binX, binY, extrapolatedValue);
973 delete linExtrapolation;
978 //______________________________________________________________________________
979 void AliPIDResponse::SetTPCEtaMaps(Double_t refineFactorMapX, Double_t refineFactorMapY,
980 Double_t refineFactorSigmaMapX, Double_t refineFactorSigmaMapY)
983 // Load the TPC eta correction maps from the OADB
986 if (fUseTPCEtaCorrection == kFALSE) {
987 // Disable eta correction via setting no maps
988 if (!fTPCResponse.SetEtaCorrMap(0x0))
989 AliInfo("Request to disable TPC eta correction -> Eta correction has been disabled");
991 AliError("Request to disable TPC eta correction -> Some error occured when unloading the correction maps");
993 if (!fTPCResponse.SetSigmaParams(0x0, 0))
994 AliInfo("Request to disable TPC eta correction -> Using old parametrisation for sigma");
996 AliError("Request to disable TPC eta correction -> Some error occured when unloading the sigma maps");
1001 TString dataType = "DATA";
1002 TString period = fLHCperiod.IsNull() ? "No period information" : fLHCperiod;
1005 if (!(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) {
1006 period=fMCperiodTPC;
1011 if (!(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC)) && fMCperiodTPC.IsNull()) {
1012 AliFatal("MC detected, but no MC period set -> Not changing eta maps!");
1017 Int_t recopass = fRecoPass;
1018 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC) )
1019 recopass = fRecoPassUser;
1021 TString defaultObj = Form("Default_%s_pass%d", dataType.Data(), recopass);
1023 AliInfo(Form("Current period and reco pass: %s.pass%d", period.Data(), recopass));
1025 // Invalidate old maps
1026 fTPCResponse.SetEtaCorrMap(0x0);
1027 fTPCResponse.SetSigmaParams(0x0, 0);
1030 TString fileNameMaps(Form("%s/COMMON/PID/data/TPCetaMaps.root", fOADBPath.Data()));
1031 if (!fCustomTPCetaMaps.IsNull()) fileNameMaps=fCustomTPCetaMaps;
1033 // Load the eta correction maps
1034 AliOADBContainer etaMapsCont(Form("TPCetaMaps_%s_pass%d", dataType.Data(), recopass));
1036 Int_t statusCont = etaMapsCont.InitFromFile(fileNameMaps.Data(), Form("TPCetaMaps_%s_pass%d", dataType.Data(), recopass));
1038 AliError("Failed initializing TPC eta correction maps from OADB -> Disabled eta correction");
1039 fUseTPCEtaCorrection = kFALSE;
1042 AliInfo(Form("Loading TPC eta correction map from %s", fileNameMaps.Data()));
1046 if (fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) {
1047 TString searchMap = Form("TPCetaMaps_%s_%s_pass%d", dataType.Data(), period.Data(), recopass);
1048 etaMap = dynamic_cast<TH2D *>(etaMapsCont.GetDefaultObject(searchMap.Data()));
1050 // Try default object
1051 etaMap = dynamic_cast<TH2D *>(etaMapsCont.GetDefaultObject(defaultObj.Data()));
1055 etaMap = dynamic_cast<TH2D *>(etaMapsCont.GetObject(fRun, defaultObj.Data()));
1060 AliError(Form("TPC eta correction map not found for run %d and also no default map found -> Disabled eta correction!!!", fRun));
1061 fUseTPCEtaCorrection = kFALSE;
1064 TH2D* etaMapRefined = RefineHistoViaLinearInterpolation(etaMap, refineFactorMapX, refineFactorMapY);
1066 if (etaMapRefined) {
1067 if (!fTPCResponse.SetEtaCorrMap(etaMapRefined)) {
1068 AliError(Form("Failed to set TPC eta correction map for run %d -> Disabled eta correction!!!", fRun));
1069 fTPCResponse.SetEtaCorrMap(0x0);
1070 fUseTPCEtaCorrection = kFALSE;
1073 AliInfo(Form("Loaded TPC eta correction map (refine factors %.2f/%.2f) from %s: %s (MD5(map) = %s)",
1074 refineFactorMapX, refineFactorMapY, fileNameMaps.Data(), fTPCResponse.GetEtaCorrMap()->GetTitle(),
1075 GetChecksum(fTPCResponse.GetEtaCorrMap()).Data()));
1078 delete etaMapRefined;
1081 AliError(Form("Failed to set TPC eta correction map for run %d (map was loaded, but couldn't be refined) -> Disabled eta correction!!!", fRun));
1082 fUseTPCEtaCorrection = kFALSE;
1087 // If there was some problem loading the eta maps, it makes no sense to load the sigma maps (that require eta corrected data)
1088 if (fUseTPCEtaCorrection == kFALSE) {
1089 AliError("Failed to load TPC eta correction map required by sigma maps -> Using old parametrisation for sigma");
1093 // Load the sigma parametrisation (1/dEdx vs tanTheta_local (~eta))
1094 AliOADBContainer etaSigmaMapsCont(Form("TPCetaSigmaMaps_%s_pass%d", dataType.Data(), recopass));
1096 statusCont = etaSigmaMapsCont.InitFromFile(fileNameMaps.Data(), Form("TPCetaSigmaMaps_%s_pass%d", dataType.Data(), recopass));
1098 AliError("Failed initializing TPC eta sigma maps from OADB -> Using old sigma parametrisation");
1101 AliInfo(Form("Loading TPC eta sigma map from %s", fileNameMaps.Data()));
1103 TObjArray* etaSigmaPars = 0x0;
1105 if (fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) {
1106 TString searchMap = Form("TPCetaSigmaMaps_%s_%s_pass%d", dataType.Data(), period.Data(), recopass);
1107 etaSigmaPars = dynamic_cast<TObjArray *>(etaSigmaMapsCont.GetDefaultObject(searchMap.Data()));
1108 if (!etaSigmaPars) {
1109 // Try default object
1110 etaSigmaPars = dynamic_cast<TObjArray *>(etaSigmaMapsCont.GetDefaultObject(defaultObj.Data()));
1114 etaSigmaPars = dynamic_cast<TObjArray *>(etaSigmaMapsCont.GetObject(fRun, defaultObj.Data()));
1117 if (!etaSigmaPars) {
1118 AliError(Form("TPC eta sigma parametrisation not found for run %d -> Using old sigma parametrisation!!!", fRun));
1121 TH2D* etaSigmaPar1Map = dynamic_cast<TH2D *>(etaSigmaPars->FindObject("sigmaPar1Map"));
1122 TNamed* sigmaPar0Info = dynamic_cast<TNamed *>(etaSigmaPars->FindObject("sigmaPar0"));
1123 Double_t sigmaPar0 = 0.0;
1125 if (sigmaPar0Info) {
1126 TString sigmaPar0String = sigmaPar0Info->GetTitle();
1127 sigmaPar0 = sigmaPar0String.Atof();
1130 // Something is weired because the object for parameter 0 could not be loaded -> New sigma parametrisation can not be used!
1131 etaSigmaPar1Map = 0x0;
1134 TH2D* etaSigmaPar1MapRefined = RefineHistoViaLinearInterpolation(etaSigmaPar1Map, refineFactorSigmaMapX, refineFactorSigmaMapY);
1137 if (etaSigmaPar1MapRefined) {
1138 if (!fTPCResponse.SetSigmaParams(etaSigmaPar1MapRefined, sigmaPar0)) {
1139 AliError(Form("Failed to set TPC eta sigma map for run %d -> Using old sigma parametrisation!!!", fRun));
1140 fTPCResponse.SetSigmaParams(0x0, 0);
1143 AliInfo(Form("Loaded TPC sigma correction map (refine factors %.2f/%.2f) from %s: %s (MD5(map) = %s, sigmaPar0 = %f)",
1144 refineFactorSigmaMapX, refineFactorSigmaMapY, fileNameMaps.Data(), fTPCResponse.GetSigmaPar1Map()->GetTitle(),
1145 GetChecksum(fTPCResponse.GetSigmaPar1Map()).Data(), sigmaPar0));
1148 delete etaSigmaPar1MapRefined;
1151 AliError(Form("Failed to set TPC eta sigma map for run %d (map was loaded, but couldn't be refined) -> Using old sigma parametrisation!!!",
1158 //______________________________________________________________________________
1159 void AliPIDResponse::SetTPCPidResponseMaster()
1162 // Load the TPC pid response functions from the OADB
1163 // Load the TPC voltage maps from OADB
1165 //don't load twice for the moment
1166 if (fArrPidResponseMaster) return;
1169 //reset the PID response functions
1170 delete fArrPidResponseMaster;
1171 fArrPidResponseMaster=NULL;
1175 TString fileNamePIDresponse(Form("%s/COMMON/PID/data/TPCPIDResponse.root", fOADBPath.Data()));
1176 if (!fCustomTPCpidResponse.IsNull()) fileNamePIDresponse=fCustomTPCpidResponse;
1178 f=TFile::Open(fileNamePIDresponse.Data());
1179 if (f && f->IsOpen() && !f->IsZombie()){
1180 fArrPidResponseMaster=dynamic_cast<TObjArray*>(f->Get("TPCPIDResponse"));
1184 TString fileNameVoltageMaps(Form("%s/COMMON/PID/data/TPCvoltageSettings.root", fOADBPath.Data()));
1185 f=TFile::Open(fileNameVoltageMaps.Data());
1186 if (f && f->IsOpen() && !f->IsZombie()){
1187 fOADBvoltageMaps=dynamic_cast<AliOADBContainer*>(f->Get("TPCvoltageSettings"));
1191 if (!fArrPidResponseMaster){
1192 AliFatal(Form("Could not retrieve the TPC pid response from: %s",fileNamePIDresponse.Data()));
1195 fArrPidResponseMaster->SetOwner();
1197 if (!fOADBvoltageMaps)
1199 AliFatal(Form("Could not retrieve the TPC voltage maps from: %s",fileNameVoltageMaps.Data()));
1201 fArrPidResponseMaster->SetOwner();
1204 //______________________________________________________________________________
1205 void AliPIDResponse::SetTPCParametrisation()
1208 // Change BB parametrisation for current run
1214 fTPCResponse.ResetSplines();
1216 if (fLHCperiod.IsNull()) {
1217 AliError("No period set, not changing parametrisation");
1222 // Set default parametrisations for data and MC
1226 TString datatype="DATA";
1227 //in case of mc fRecoPass is per default 1
1229 if(!(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) datatype="MC";
1234 TString period=fLHCperiod;
1235 if (fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) period=fMCperiodTPC;
1237 Int_t recopass = fRecoPass;
1238 if(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC)) recopass = fRecoPassUser;
1240 AliInfo(Form("Searching splines for: %s %s PASS%d %s",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
1241 Bool_t found=kFALSE;
1243 //set the new PID splines
1245 if (fArrPidResponseMaster){
1246 //for MC don't use period information
1247 //if (fIsMC) period="[A-Z0-9]*";
1248 //for MC use MC period information
1249 //pattern for the default entry (valid for all particles)
1250 TPRegexp reg(Form("TSPLINE3_%s_([A-Z]*)_%s_PASS%d_%s_MEAN(_*)([A-Z1-9]*)",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
1252 //find particle id and gain scenario
1253 for (Int_t igainScenario=0; igainScenario<AliTPCPIDResponse::fgkNumberOfGainScenarios; igainScenario++)
1255 TObject *grAll=NULL;
1256 TString gainScenario = AliTPCPIDResponse::GainScenarioName(igainScenario);
1257 gainScenario.ToUpper();
1258 //loop over entries and filter them
1259 for (Int_t iresp=0; iresp<fArrPidResponseMaster->GetEntriesFast();++iresp)
1261 TObject *responseFunction=fArrPidResponseMaster->At(iresp);
1262 if (responseFunction==NULL) continue;
1263 TString responseName=responseFunction->GetName();
1265 if (!reg.MatchB(responseName)) continue;
1267 TObjArray *arr=reg.MatchS(responseName); if (!arr) continue;
1269 tmp=arr->At(1); if (!tmp) continue;
1270 TString particleName=tmp->GetName();
1271 tmp=arr->At(3); if (!tmp) continue;
1272 TString gainScenarioName=tmp->GetName();
1274 if (particleName.IsNull()) continue;
1275 if (!grAll && particleName=="ALL" && gainScenarioName==gainScenario) grAll=responseFunction;
1278 for (Int_t ispec=0; ispec<(AliTPCPIDResponse::fgkNumberOfParticleSpecies); ++ispec)
1280 TString particle=AliPID::ParticleName(ispec);
1282 //std::cout<<responseName<<" "<<particle<<" "<<particleName<<" "<<gainScenario<<" "<<gainScenarioName<<std::endl;
1283 if ( particle == particleName && gainScenario == gainScenarioName )
1285 fTPCResponse.SetResponseFunction( responseFunction,
1286 (AliPID::EParticleType)ispec,
1287 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1288 fTPCResponse.SetUseDatabase(kTRUE);
1289 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,responseFunction->GetName(),
1290 GetChecksum((TSpline3*)responseFunction).Data()));
1298 // Retrieve responsefunction for pions - will (if available) be used for muons if there are no dedicated muon splines.
1299 // For light nuclei, try to set the proton spline, if no dedicated splines are available.
1300 // In both cases: Use default splines, if no dedicated splines and no pion/proton splines are available.
1301 TObject* responseFunctionPion = fTPCResponse.GetResponseFunction( (AliPID::EParticleType)AliPID::kPion,
1302 (AliTPCPIDResponse::ETPCgainScenario)igainScenario);
1303 TObject* responseFunctionProton = fTPCResponse.GetResponseFunction( (AliPID::EParticleType)AliPID::kProton,
1304 (AliTPCPIDResponse::ETPCgainScenario)igainScenario);
1306 for (Int_t ispec=0; ispec<(AliTPCPIDResponse::fgkNumberOfParticleSpecies); ++ispec)
1308 if (!fTPCResponse.GetResponseFunction( (AliPID::EParticleType)ispec,
1309 (AliTPCPIDResponse::ETPCgainScenario)igainScenario))
1311 if (ispec == AliPID::kMuon) { // Muons
1312 if (responseFunctionPion) {
1313 fTPCResponse.SetResponseFunction( responseFunctionPion,
1314 (AliPID::EParticleType)ispec,
1315 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1316 fTPCResponse.SetUseDatabase(kTRUE);
1317 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,responseFunctionPion->GetName(),
1318 GetChecksum((TSpline3*)responseFunctionPion).Data()));
1322 fTPCResponse.SetResponseFunction( grAll,
1323 (AliPID::EParticleType)ispec,
1324 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1325 fTPCResponse.SetUseDatabase(kTRUE);
1326 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,grAll->GetName(),
1327 GetChecksum((TSpline3*)grAll).Data()));
1331 // AliError(Form("No splines found for muons (also no pion splines and no default splines) for gain scenario %d!", igainScenario));
1333 else if (ispec >= AliPID::kSPECIES) { // Light nuclei
1334 if (responseFunctionProton) {
1335 fTPCResponse.SetResponseFunction( responseFunctionProton,
1336 (AliPID::EParticleType)ispec,
1337 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1338 fTPCResponse.SetUseDatabase(kTRUE);
1339 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,responseFunctionProton->GetName(),
1340 GetChecksum((TSpline3*)responseFunctionProton).Data()));
1344 fTPCResponse.SetResponseFunction( grAll,
1345 (AliPID::EParticleType)ispec,
1346 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1347 fTPCResponse.SetUseDatabase(kTRUE);
1348 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,grAll->GetName(),
1349 GetChecksum((TSpline3*)grAll).Data()));
1353 // AliError(Form("No splines found for species %d (also no proton splines and no default splines) for gain scenario %d!",
1354 // ispec, igainScenario));
1360 else AliInfo("no fArrPidResponseMaster");
1363 AliError(Form("No splines found for: %s %s PASS%d %s",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
1368 // Setup multiplicity correction (only used for non-pp collisions)
1371 const Bool_t isPP = (fBeamType.CompareTo("PP") == 0);
1373 // 2013 pPb data taking at low luminosity
1374 const Bool_t isPPb2013LowLuminosity = period.Contains("LHC13B") || period.Contains("LHC13C") || period.Contains("LHC13D");
1375 // PbPb 2010, period 10h.pass2
1376 //TODO Needs further development const Bool_t is10hpass2 = period.Contains("LHC10H") && recopass == 2;
1379 // In case of MC without(!) tune on data activated for the TPC, don't use the multiplicity correction for the moment
1380 Bool_t isMCandNotTPCtuneOnData = fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC));
1382 // If correction is available, but disabled (highly NOT recommended!), print warning
1383 if (!fUseTPCMultiplicityCorrection && !isPP && !isMCandNotTPCtuneOnData) {
1384 //TODO: Needs further development if (is10hpass2 || isPPb2013LowLuminosity) {
1385 if (isPPb2013LowLuminosity) {
1386 AliWarning("Mulitplicity correction disabled, but correction parameters for this period exist. It is highly recommended to use enable the correction. Otherwise the splines might be off!");
1390 if (fUseTPCMultiplicityCorrection && !isPP && !isMCandNotTPCtuneOnData) {
1391 AliInfo("Multiplicity correction enabled!");
1393 //TODO After testing, load parameters from outside
1394 /*TODO no correction for MC
1395 if (period.Contains("LHC11A10")) {//LHC11A10A
1396 AliInfo("Using multiplicity correction parameters for 11a10!");
1397 fTPCResponse.SetParameterMultiplicityCorrection(0, 6.90133e-06);
1398 fTPCResponse.SetParameterMultiplicityCorrection(1, -1.22123e-03);
1399 fTPCResponse.SetParameterMultiplicityCorrection(2, 1.80220e-02);
1400 fTPCResponse.SetParameterMultiplicityCorrection(3, 0.1);
1401 fTPCResponse.SetParameterMultiplicityCorrection(4, 6.45306e-03);
1403 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, -2.85505e-07);
1404 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, -1.31911e-06);
1405 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1407 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, -4.29665e-05);
1408 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, 1.37023e-02);
1409 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, -6.36337e-01);
1410 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 1.13479e-02);
1412 else*/ if (isPPb2013LowLuminosity) {// 2013 pPb data taking at low luminosity
1413 AliInfo("Using multiplicity correction parameters for 13b.pass2 (at least also valid for 13{c,d} and pass 3)!");
1415 fTPCResponse.SetParameterMultiplicityCorrection(0, -5.906e-06);
1416 fTPCResponse.SetParameterMultiplicityCorrection(1, -5.064e-04);
1417 fTPCResponse.SetParameterMultiplicityCorrection(2, -3.521e-02);
1418 fTPCResponse.SetParameterMultiplicityCorrection(3, 2.469e-02);
1419 fTPCResponse.SetParameterMultiplicityCorrection(4, 0);
1421 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, -5.32e-06);
1422 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, 1.177e-05);
1423 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1425 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, 0.);
1426 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, 0.);
1427 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, 0.);
1428 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 0.);
1430 /* Not too bad, but far from perfect in the details
1431 fTPCResponse.SetParameterMultiplicityCorrection(0, -6.27187e-06);
1432 fTPCResponse.SetParameterMultiplicityCorrection(1, -4.60649e-04);
1433 fTPCResponse.SetParameterMultiplicityCorrection(2, -4.26450e-02);
1434 fTPCResponse.SetParameterMultiplicityCorrection(3, 2.40590e-02);
1435 fTPCResponse.SetParameterMultiplicityCorrection(4, 0);
1437 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, -5.338e-06);
1438 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, 1.220e-05);
1439 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1441 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, 7.89237e-05);
1442 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, -1.30662e-02);
1443 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, 8.91548e-01);
1444 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 1.47931e-02);
1447 /*TODO: Needs further development
1448 else if (is10hpass2) {
1449 AliInfo("Using multiplicity correction parameters for 10h.pass2!");
1450 fTPCResponse.SetParameterMultiplicityCorrection(0, 3.21636e-07);
1451 fTPCResponse.SetParameterMultiplicityCorrection(1, -6.65876e-04);
1452 fTPCResponse.SetParameterMultiplicityCorrection(2, 1.28786e-03);
1453 fTPCResponse.SetParameterMultiplicityCorrection(3, 1.47677e-02);
1454 fTPCResponse.SetParameterMultiplicityCorrection(4, 0);
1456 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, 7.23591e-08);
1457 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, 2.7469e-06);
1458 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1460 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, -1.22590e-05);
1461 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, 6.88888e-03);
1462 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, -3.20788e-01);
1463 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 1.07345e-02);
1467 AliError(Form("Multiplicity correction is enabled, but no multiplicity correction parameters have been found for period %s.pass%d -> Mulitplicity correction DISABLED!", period.Data(), recopass));
1468 fUseTPCMultiplicityCorrection = kFALSE;
1469 fTPCResponse.ResetMultiplicityCorrectionFunctions();
1473 // Just set parameters such that overall correction factor is 1, i.e. no correction.
1474 // This is just a reasonable choice for the parameters for safety reasons. Disabling
1475 // the multiplicity correction will anyhow skip the calculation of the corresponding
1476 // correction factor inside THIS class. Nevertheless, experts can access the TPCPIDResponse
1477 // directly and use it for calculations - which should still give valid results, even if
1478 // the multiplicity correction is explicitely enabled in such expert calls.
1480 TString reasonForDisabling = "requested by user";
1481 if (fUseTPCMultiplicityCorrection) {
1483 reasonForDisabling = "pp collisions";
1485 reasonForDisabling = "MC w/o tune on data";
1488 AliInfo(Form("Multiplicity correction %sdisabled (%s)!", fUseTPCMultiplicityCorrection ? "automatically " : "",
1489 reasonForDisabling.Data()));
1491 fUseTPCMultiplicityCorrection = kFALSE;
1492 fTPCResponse.ResetMultiplicityCorrectionFunctions();
1495 if (fUseTPCMultiplicityCorrection) {
1496 for (Int_t i = 0; i <= 4 + 1; i++) {
1497 AliInfo(Form("parMultCorr: %d, %e", i, fTPCResponse.GetMultiplicityCorrectionFunction()->GetParameter(i)));
1499 for (Int_t j = 0; j <= 2 + 1; j++) {
1500 AliInfo(Form("parMultCorrTanTheta: %d, %e", j, fTPCResponse.GetMultiplicityCorrectionFunctionTanTheta()->GetParameter(j)));
1502 for (Int_t j = 0; j <= 3 + 1; j++) {
1503 AliInfo(Form("parMultSigmaCorr: %d, %e", j, fTPCResponse.GetMultiplicitySigmaCorrectionFunction()->GetParameter(j)));
1508 // Setup old resolution parametrisation
1512 fTPCResponse.SetSigma(3.79301e-03, 2.21280e+04);
1514 if (fRun>=122195){ //LHC10d
1515 fTPCResponse.SetSigma(2.30176e-02, 5.60422e+02);
1518 if (fRun>=170719){ // LHC12a
1519 fTPCResponse.SetSigma(2.95714e-03, 1.01953e+05);
1522 if (fRun>=177312){ // LHC12b
1523 fTPCResponse.SetSigma(3.74633e-03, 7.11829e+04 );
1526 if (fRun>=186346){ // LHC12e
1527 fTPCResponse.SetSigma(8.62022e-04, 9.08156e+05);
1530 if (fArrPidResponseMaster)
1531 fResolutionCorrection=(TF1*)fArrPidResponseMaster->FindObject(Form("TF1_%s_ALL_%s_PASS%d_%s_SIGMA",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
1533 if (fResolutionCorrection) AliInfo(Form("Setting multiplicity correction function: %s (MD5(corr function) = %s)",
1534 fResolutionCorrection->GetName(), GetChecksum(fResolutionCorrection).Data()));
1536 //read in the voltage map
1537 TVectorF* gsm = 0x0;
1538 if (fOADBvoltageMaps) gsm=dynamic_cast<TVectorF*>(fOADBvoltageMaps->GetObject(fRun));
1541 fTPCResponse.SetVoltageMap(*gsm);
1543 AliInfo(Form("Reading the voltage map for run %d\n",fRun));
1544 vals="IROC A: "; for (Int_t i=0; i<18; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1545 AliInfo(vals.Data());
1546 vals="IROC C: "; for (Int_t i=18; i<36; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1547 AliInfo(vals.Data());
1548 vals="OROC A: "; for (Int_t i=36; i<54; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1549 AliInfo(vals.Data());
1550 vals="OROC C: "; for (Int_t i=54; i<72; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1551 AliInfo(vals.Data());
1553 else AliInfo("no voltage map, ideal default assumed");
1556 //______________________________________________________________________________
1557 void AliPIDResponse::SetTRDPidResponseMaster()
1560 // Load the TRD pid params and references from the OADB
1562 if(fTRDPIDResponseObject) return;
1563 AliOADBContainer contParams("contParams");
1565 Int_t statusResponse = contParams.InitFromFile(Form("%s/COMMON/PID/data/TRDPIDResponse.root", fOADBPath.Data()), "AliTRDPIDResponseObject");
1567 AliError("Failed initializing PID Response Object from OADB");
1569 AliInfo(Form("Loading TRD Response from %s/COMMON/PID/data/TRDPIDResponse.root", fOADBPath.Data()));
1570 fTRDPIDResponseObject = dynamic_cast<AliTRDPIDResponseObject *>(contParams.GetObject(fRun));
1571 if(!fTRDPIDResponseObject){
1572 AliError(Form("TRD Response not found in run %d", fRun));
1577 //______________________________________________________________________________
1578 void AliPIDResponse::InitializeTRDResponse(){
1580 // Set PID Params and references to the TRD PID response
1582 fTRDResponse.SetPIDResponseObject(fTRDPIDResponseObject);
1583 fTRDResponse.SetdEdxParams(fTRDdEdxParams);
1586 //______________________________________________________________________________
1587 void AliPIDResponse::SetTRDSlices(UInt_t TRDslicesForPID[2],AliTRDPIDResponse::ETRDPIDMethod method) const{
1589 if(fLHCperiod.Contains("LHC10D") || fLHCperiod.Contains("LHC10E")){
1590 // backward compatibility for setting with 8 slices
1591 TRDslicesForPID[0] = 0;
1592 TRDslicesForPID[1] = 7;
1595 if(method==AliTRDPIDResponse::kLQ1D){
1596 TRDslicesForPID[0] = 0; // first Slice contains normalized dEdx
1597 TRDslicesForPID[1] = 0;
1599 if(method==AliTRDPIDResponse::kLQ2D){
1600 TRDslicesForPID[0] = 1;
1601 TRDslicesForPID[1] = 7;
1604 AliDebug(1,Form("Slice Range set to %d - %d",TRDslicesForPID[0],TRDslicesForPID[1]));
1606 //______________________________________________________________________________
1607 void AliPIDResponse::SetTRDdEdxParams()
1609 if(fTRDdEdxParams) return;
1611 const TString containerName = "TRDdEdxParamsContainer";
1612 AliOADBContainer cont(containerName.Data());
1614 const TString filePathNamePackage=Form("%s/COMMON/PID/data/TRDdEdxParams.root", fOADBPath.Data());
1616 const Int_t statusCont = cont.InitFromFile(filePathNamePackage.Data(), cont.GetName());
1618 AliFatal("Failed initializing settings from OADB");
1621 AliInfo(Form("Loading %s from %s\n", cont.GetName(), filePathNamePackage.Data()));
1623 fTRDdEdxParams = (AliTRDdEdxParams*)(cont.GetObject(fRun, "default"));
1624 //fTRDdEdxParams->Print();
1626 if(!fTRDdEdxParams){
1627 AliError(Form("TRD dEdx Params default not found"));
1632 //______________________________________________________________________________
1633 void AliPIDResponse::SetTOFPidResponseMaster()
1636 // Load the TOF pid params from the OADB
1639 if (fTOFPIDParams) delete fTOFPIDParams;
1642 TFile *oadbf = new TFile(Form("%s/COMMON/PID/data/TOFPIDParams.root",fOADBPath.Data()));
1643 if (oadbf && oadbf->IsOpen()) {
1644 AliInfo(Form("Loading TOF Params from %s/COMMON/PID/data/TOFPIDParams.root", fOADBPath.Data()));
1645 AliOADBContainer *oadbc = (AliOADBContainer *)oadbf->Get("TOFoadb");
1646 if (oadbc) fTOFPIDParams = dynamic_cast<AliTOFPIDParams *>(oadbc->GetObject(fRun,"TOFparams"));
1652 if (!fTOFPIDParams) AliFatal("TOFPIDParams could not be retrieved");
1655 //______________________________________________________________________________
1656 void AliPIDResponse::InitializeTOFResponse(){
1658 // Set PID Params to the TOF PID response
1661 AliInfo("TOF PID Params loaded from OADB");
1662 AliInfo(Form(" TOF resolution %5.2f [ps]",fTOFPIDParams->GetTOFresolution()));
1663 AliInfo(Form(" StartTime method %d",fTOFPIDParams->GetStartTimeMethod()));
1664 AliInfo(Form(" TOF res. mom. params: %5.2f %5.2f %5.2f %5.2f",
1665 fTOFPIDParams->GetSigParams(0),fTOFPIDParams->GetSigParams(1),fTOFPIDParams->GetSigParams(2),fTOFPIDParams->GetSigParams(3)));
1666 AliInfo(Form(" Fraction of tracks within gaussian behaviour: %6.4f",fTOFPIDParams->GetTOFtail()));
1667 AliInfo(Form(" MC: Fraction of tracks (percentage) to cut to fit matching in data: %6.2f%%",fTOFPIDParams->GetTOFmatchingLossMC()));
1668 AliInfo(Form(" MC: Fraction of random hits (percentage) to add to fit mismatch in data: %6.2f%%",fTOFPIDParams->GetTOFadditionalMismForMC()));
1669 AliInfo(Form(" Start Time Offset %6.2f ps",fTOFPIDParams->GetTOFtimeOffset()));
1671 for (Int_t i=0;i<4;i++) {
1672 fTOFResponse.SetTrackParameter(i,fTOFPIDParams->GetSigParams(i));
1674 fTOFResponse.SetTimeResolution(fTOFPIDParams->GetTOFresolution());
1676 AliInfo("TZERO resolution loaded from ESDrun/AODheader");
1677 Float_t t0Spread[4];
1678 for (Int_t i=0;i<4;i++) t0Spread[i]=fCurrentEvent->GetT0spread(i);
1679 AliInfo(Form(" TZERO spreads from data: (A+C)/2 %f A %f C %f (A'-C')/2: %f",t0Spread[0],t0Spread[1],t0Spread[2],t0Spread[3]));
1680 Float_t a = t0Spread[1]*t0Spread[1]-t0Spread[0]*t0Spread[0]+t0Spread[3]*t0Spread[3];
1681 Float_t c = t0Spread[2]*t0Spread[2]-t0Spread[0]*t0Spread[0]+t0Spread[3]*t0Spread[3];
1682 if ( (t0Spread[0] > 50. && t0Spread[0] < 400.) && (a > 0.) && (c>0.)) {
1683 fResT0AC=t0Spread[3];
1684 fResT0A=TMath::Sqrt(a);
1685 fResT0C=TMath::Sqrt(c);
1687 AliInfo(" TZERO spreads not present or inconsistent, loading default");
1692 AliInfo(Form(" TZERO resolution set to: T0A: %f [ps] T0C: %f [ps] T0AC %f [ps]",fResT0A,fResT0C,fResT0AC));
1696 //______________________________________________________________________________
1697 void AliPIDResponse::SetHMPIDPidResponseMaster()
1700 // Load the HMPID pid params from the OADB
1703 if (fHMPIDPIDParams) delete fHMPIDPIDParams;
1704 fHMPIDPIDParams=NULL;
1707 if(!fIsMC) oadbf = new TFile(Form("%s/COMMON/PID/data/HMPIDPIDParams.root",fOADBPath.Data()));
1708 else oadbf = new TFile(Form("%s/COMMON/PID/MC/HMPIDPIDParams.root",fOADBPath.Data()));
1709 if (oadbf && oadbf->IsOpen()) {
1710 AliInfo(Form("Loading HMPID Params from %s/COMMON/PID/data/HMPIDPIDParams.root", fOADBPath.Data()));
1711 AliOADBContainer *oadbc = (AliOADBContainer *)oadbf->Get("HMPoadb");
1712 if (oadbc) fHMPIDPIDParams = dynamic_cast<AliHMPIDPIDParams *>(oadbc->GetObject(fRun,"HMPparams"));
1718 if (!fHMPIDPIDParams) AliFatal("HMPIDPIDParams could not be retrieved");
1721 //______________________________________________________________________________
1722 void AliPIDResponse::InitializeHMPIDResponse(){
1724 // Set PID Params to the HMPID PID response
1727 fHMPIDResponse.SetRefIndexArray(fHMPIDPIDParams->GetHMPIDrefIndex());
1730 //______________________________________________________________________________
1731 Bool_t AliPIDResponse::IdentifiedAsElectronTRD(const AliVTrack *vtrack,Double_t efficiencyLevel,Double_t centrality,AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const {
1732 // old function for compatibility
1734 return IdentifiedAsElectronTRD(vtrack,ntracklets,efficiencyLevel,centrality,PIDmethod);
1737 //______________________________________________________________________________
1738 Bool_t AliPIDResponse::IdentifiedAsElectronTRD(const AliVTrack *vtrack, Int_t &ntracklets,Double_t efficiencyLevel,Double_t centrality,AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const {
1740 // Check whether track is identified as electron under a given electron efficiency hypothesis
1742 // ntracklets is the number of tracklets that has been used to calculate the PID signal
1744 Double_t probs[AliPID::kSPECIES];
1746 ntracklets =CalculateTRDResponse(vtrack,probs,PIDmethod);
1748 // Take mean of the TRD momenta in the given tracklets
1749 Float_t p = 0, trdmomenta[AliVTrack::kTRDnPlanes];
1751 for(Int_t iPl=0;iPl<AliVTrack::kTRDnPlanes;iPl++){
1752 if(vtrack->GetTRDmomentum(iPl) > 0.){
1753 trdmomenta[nmomenta++] = vtrack->GetTRDmomentum(iPl);
1756 p = TMath::Mean(nmomenta, trdmomenta);
1758 return fTRDResponse.IdentifiedAsElectron(ntracklets, probs, p, efficiencyLevel,centrality,PIDmethod);
1761 //______________________________________________________________________________
1762 void AliPIDResponse::SetEMCALPidResponseMaster()
1765 // Load the EMCAL pid response functions from the OADB
1767 TObjArray* fEMCALPIDParamsRun = NULL;
1768 TObjArray* fEMCALPIDParamsPass = NULL;
1770 if(fEMCALPIDParams) return;
1771 AliOADBContainer contParams("contParams");
1773 Int_t statusPars = contParams.InitFromFile(Form("%s/COMMON/PID/data/EMCALPIDParams.root", fOADBPath.Data()), "AliEMCALPIDParams");
1775 AliError("Failed initializing PID Params from OADB");
1778 AliInfo(Form("Loading EMCAL Params from %s/COMMON/PID/data/EMCALPIDParams.root", fOADBPath.Data()));
1780 fEMCALPIDParamsRun = dynamic_cast<TObjArray *>(contParams.GetObject(fRun));
1781 if(fEMCALPIDParamsRun) fEMCALPIDParamsPass = dynamic_cast<TObjArray *>(fEMCALPIDParamsRun->FindObject(Form("pass%d",fRecoPass)));
1782 if(fEMCALPIDParamsPass) fEMCALPIDParams = dynamic_cast<TObjArray *>(fEMCALPIDParamsPass->FindObject(Form("EMCALPIDParams_Particles")));
1784 if(!fEMCALPIDParams){
1785 AliInfo(Form("EMCAL Params not found in run %d pass %d", fRun, fRecoPass));
1786 AliInfo("Will take the standard LHC11d instead ...");
1788 fEMCALPIDParamsRun = dynamic_cast<TObjArray *>(contParams.GetObject(156477));
1789 if(fEMCALPIDParamsRun) fEMCALPIDParamsPass = dynamic_cast<TObjArray *>(fEMCALPIDParamsRun->FindObject(Form("pass%d",1)));
1790 if(fEMCALPIDParamsPass) fEMCALPIDParams = dynamic_cast<TObjArray *>(fEMCALPIDParamsPass->FindObject(Form("EMCALPIDParams_Particles")));
1792 if(!fEMCALPIDParams){
1793 AliError(Form("DEFAULT EMCAL Params (LHC11d) not found in file %s/COMMON/PID/data/EMCALPIDParams.root", fOADBPath.Data()));
1799 //______________________________________________________________________________
1800 void AliPIDResponse::InitializeEMCALResponse(){
1802 // Set PID Params to the EMCAL PID response
1804 fEMCALResponse.SetPIDParams(fEMCALPIDParams);
1808 //______________________________________________________________________________
1809 void AliPIDResponse::FillTrackDetectorPID(const AliVTrack *track, EDetector detector) const
1812 // create detector PID information and setup the transient pointer in the track
1815 // check if detector number is inside accepted range
1816 if (detector == kNdetectors) return;
1819 AliDetectorPID *detPID=const_cast<AliDetectorPID*>(track->GetDetectorPID());
1821 detPID=new AliDetectorPID;
1822 (const_cast<AliVTrack*>(track))->SetDetectorPID(detPID);
1825 //check if values exist
1826 if (detPID->HasRawProbability(detector) && detPID->HasNumberOfSigmas(detector)) return;
1828 //TODO: which particles to include? See also the loops below...
1829 Double_t values[AliPID::kSPECIESC]={0};
1832 EDetPidStatus status=GetComputePIDProbability(detector,track,AliPID::kSPECIESC,values);
1833 detPID->SetRawProbability(detector, values, (Int_t)AliPID::kSPECIESC, status);
1836 for (Int_t ipart=0; ipart<AliPID::kSPECIESC; ++ipart)
1837 values[ipart]=GetNumberOfSigmas(detector,track,(AliPID::EParticleType)ipart);
1838 // the pid status is the same for probabilities and nSigmas, so it is
1839 // fine to use the one from the probabilities also here
1840 detPID->SetNumberOfSigmas(detector, values, (Int_t)AliPID::kSPECIESC, status);
1844 //______________________________________________________________________________
1845 void AliPIDResponse::FillTrackDetectorPID()
1848 // create detector PID information and setup the transient pointer in the track
1851 if (!fCurrentEvent) return;
1853 for (Int_t itrack=0; itrack<fCurrentEvent->GetNumberOfTracks(); ++itrack){
1854 AliVTrack *track=dynamic_cast<AliVTrack*>(fCurrentEvent->GetTrack(itrack));
1855 if (!track) continue;
1857 for (Int_t idet=0; idet<kNdetectors; ++idet){
1858 FillTrackDetectorPID(track, (EDetector)idet);
1863 //______________________________________________________________________________
1864 void AliPIDResponse::SetTOFResponse(AliVEvent *vevent,EStartTimeType_t option){
1866 // Set TOF response function
1867 // Input option for event_time used
1870 Float_t t0spread = 0.; //vevent->GetEventTimeSpread();
1871 if(t0spread < 10) t0spread = 80;
1873 // T0-FILL and T0-TO offset (because of TOF misallignment
1874 Float_t starttimeoffset = 0;
1875 if(fTOFPIDParams && !(fIsMC)) starttimeoffset=fTOFPIDParams->GetTOFtimeOffset();
1877 fTOFtail = fTOFPIDParams->GetTOFtail();
1878 GetTOFResponse().SetTOFtail(fTOFtail);
1881 // T0 from TOF algorithm
1882 Bool_t flagT0TOF=kFALSE;
1883 Bool_t flagT0T0=kFALSE;
1884 Float_t *startTime = new Float_t[fTOFResponse.GetNmomBins()];
1885 Float_t *startTimeRes = new Float_t[fTOFResponse.GetNmomBins()];
1886 Int_t *startTimeMask = new Int_t[fTOFResponse.GetNmomBins()];
1889 Float_t *estimatedT0event = new Float_t[fTOFResponse.GetNmomBins()];
1890 Float_t *estimatedT0resolution = new Float_t[fTOFResponse.GetNmomBins()];
1891 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1892 estimatedT0event[i]=0.0;
1893 estimatedT0resolution[i]=0.0;
1894 startTimeMask[i] = 0;
1897 Float_t resT0A=fResT0A;
1898 Float_t resT0C=fResT0C;
1899 Float_t resT0AC=fResT0AC;
1900 if(vevent->GetT0TOF()){ // check if T0 detector information is available
1905 AliTOFHeader *tofHeader = (AliTOFHeader*)vevent->GetTOFHeader();
1907 if (tofHeader) { // read global info and T0-TOF
1908 // fTOFResponse.SetTimeResolution(tofHeader->GetTOFResolution()); // read from OADB in the initialization
1909 t0spread = tofHeader->GetT0spread(); // read t0 sprad
1910 if(t0spread < 10) t0spread = 80;
1913 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){ // read T0-TOF default value
1914 startTime[i]=tofHeader->GetDefaultEventTimeVal();
1915 startTimeRes[i]=tofHeader->GetDefaultEventTimeRes();
1916 if(startTimeRes[i] < 1.e-5) startTimeRes[i] = t0spread;
1918 if(startTimeRes[i] > t0spread - 10 && TMath::Abs(startTime[i]) < 0.001) startTime[i] = -starttimeoffset; // apply offset for T0-fill
1921 TArrayI *ibin=(TArrayI*)tofHeader->GetNvalues();
1922 TArrayF *t0Bin=(TArrayF*)tofHeader->GetEventTimeValues();
1923 TArrayF *t0ResBin=(TArrayF*)tofHeader->GetEventTimeRes();
1924 for(Int_t j=0;j < tofHeader->GetNbins();j++){ // fill T0-TOF in p-bins
1925 Int_t icurrent = (Int_t)ibin->GetAt(j);
1926 startTime[icurrent]=t0Bin->GetAt(j);
1927 startTimeRes[icurrent]=t0ResBin->GetAt(j);
1928 if(startTimeRes[icurrent] < 1.e-5) startTimeRes[icurrent] = t0spread;
1929 if(startTimeRes[icurrent] > t0spread - 10 && TMath::Abs(startTime[icurrent]) < 0.001) startTime[icurrent] = -starttimeoffset; // apply offset for T0-fill
1933 // for cut of 3 sigma on t0 spread
1934 Float_t t0cut = 3 * t0spread;
1935 if(t0cut < 500) t0cut = 500;
1937 if(option == kFILL_T0){ // T0-FILL is used
1938 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1939 estimatedT0event[i]=0.0-starttimeoffset;
1940 estimatedT0resolution[i]=t0spread;
1942 fTOFResponse.SetT0event(estimatedT0event);
1943 fTOFResponse.SetT0resolution(estimatedT0resolution);
1946 if(option == kTOF_T0){ // T0-TOF is used when available (T0-FILL otherwise) from ESD
1948 fTOFResponse.SetT0event(startTime);
1949 fTOFResponse.SetT0resolution(startTimeRes);
1950 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1951 if(startTimeRes[i]<t0spread) startTimeMask[i]=1;
1952 fTOFResponse.SetT0binMask(i,startTimeMask[i]);
1956 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1957 estimatedT0event[i]=0.0-starttimeoffset;
1958 estimatedT0resolution[i]=t0spread;
1959 fTOFResponse.SetT0binMask(i,startTimeMask[i]);
1961 fTOFResponse.SetT0event(estimatedT0event);
1962 fTOFResponse.SetT0resolution(estimatedT0resolution);
1965 else if(option == kBest_T0){ // T0-T0 or T0-TOF are used when available (T0-FILL otherwise) from ESD
1966 Float_t t0AC=-10000;
1970 t0A= vevent->GetT0TOF()[1] - starttimeoffset;
1971 t0C= vevent->GetT0TOF()[2] - starttimeoffset;
1972 // t0AC= vevent->GetT0TOF()[0];
1973 t0AC= t0A/resT0A/resT0A + t0C/resT0C/resT0C;
1974 resT0AC= 1./TMath::Sqrt(1./resT0A/resT0A + 1./resT0C/resT0C);
1975 t0AC *= resT0AC*resT0AC;
1978 Float_t t0t0Best = 0;
1979 Float_t t0t0BestRes = 9999;
1981 if(TMath::Abs(t0A) < t0cut && TMath::Abs(t0C) < t0cut && TMath::Abs(t0C-t0A) < 500){
1983 t0t0BestRes = resT0AC;
1986 else if(TMath::Abs(t0C) < t0cut){
1988 t0t0BestRes = resT0C;
1991 else if(TMath::Abs(t0A) < t0cut){
1993 t0t0BestRes = resT0A;
1997 if(flagT0TOF){ // if T0-TOF info is available
1998 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1999 if(t0t0BestRes < 999){
2000 if(startTimeRes[i] < t0spread){
2001 Double_t wtot = 1./startTimeRes[i]/startTimeRes[i] + 1./t0t0BestRes/t0t0BestRes;
2002 Double_t t0best = startTime[i]/startTimeRes[i]/startTimeRes[i] + t0t0Best/t0t0BestRes/t0t0BestRes;
2003 estimatedT0event[i]=t0best / wtot;
2004 estimatedT0resolution[i]=1./TMath::Sqrt(wtot);
2005 startTimeMask[i] = t0used+1;
2008 estimatedT0event[i]=t0t0Best;
2009 estimatedT0resolution[i]=t0t0BestRes;
2010 startTimeMask[i] = t0used;
2014 estimatedT0event[i]=startTime[i];
2015 estimatedT0resolution[i]=startTimeRes[i];
2016 if(startTimeRes[i]<t0spread) startTimeMask[i]=1;
2018 fTOFResponse.SetT0binMask(i,startTimeMask[i]);
2020 fTOFResponse.SetT0event(estimatedT0event);
2021 fTOFResponse.SetT0resolution(estimatedT0resolution);
2023 else{ // if no T0-TOF info is available
2024 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
2025 fTOFResponse.SetT0binMask(i,t0used);
2026 if(t0t0BestRes < 999){
2027 estimatedT0event[i]=t0t0Best;
2028 estimatedT0resolution[i]=t0t0BestRes;
2031 estimatedT0event[i]=0.0-starttimeoffset;
2032 estimatedT0resolution[i]=t0spread;
2035 fTOFResponse.SetT0event(estimatedT0event);
2036 fTOFResponse.SetT0resolution(estimatedT0resolution);
2040 else if(option == kT0_T0){ // T0-T0 is used when available (T0-FILL otherwise)
2041 Float_t t0AC=-10000;
2045 t0A= vevent->GetT0TOF()[1] - starttimeoffset;
2046 t0C= vevent->GetT0TOF()[2] - starttimeoffset;
2047 // t0AC= vevent->GetT0TOF()[0];
2048 t0AC= t0A/resT0A/resT0A + t0C/resT0C/resT0C;
2049 resT0AC= 1./TMath::Sqrt(1./resT0A/resT0A + 1./resT0C/resT0C);
2050 t0AC *= resT0AC*resT0AC;
2053 if(TMath::Abs(t0A) < t0cut && TMath::Abs(t0C) < t0cut && TMath::Abs(t0C-t0A) < 500){
2054 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
2055 estimatedT0event[i]=t0AC;
2056 estimatedT0resolution[i]=resT0AC;
2057 fTOFResponse.SetT0binMask(i,6);
2060 else if(TMath::Abs(t0C) < t0cut){
2061 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
2062 estimatedT0event[i]=t0C;
2063 estimatedT0resolution[i]=resT0C;
2064 fTOFResponse.SetT0binMask(i,4);
2067 else if(TMath::Abs(t0A) < t0cut){
2068 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
2069 estimatedT0event[i]=t0A;
2070 estimatedT0resolution[i]=resT0A;
2071 fTOFResponse.SetT0binMask(i,2);
2075 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
2076 estimatedT0event[i]= 0.0 - starttimeoffset;
2077 estimatedT0resolution[i]=t0spread;
2078 fTOFResponse.SetT0binMask(i,0);
2081 fTOFResponse.SetT0event(estimatedT0event);
2082 fTOFResponse.SetT0resolution(estimatedT0resolution);
2085 delete [] startTime;
2086 delete [] startTimeRes;
2087 delete [] startTimeMask;
2088 delete [] estimatedT0event;
2089 delete [] estimatedT0resolution;
2092 //______________________________________________________________________________
2093 // private non cached versions of the PID calculation
2097 //______________________________________________________________________________
2098 Float_t AliPIDResponse::GetNumberOfSigmas(EDetector detector, const AliVParticle *vtrack, AliPID::EParticleType type) const
2101 // NumberOfSigmas for 'detCode'
2104 const AliVTrack *track=static_cast<const AliVTrack*>(vtrack);
2107 case kITS: return GetNumberOfSigmasITS(track, type); break;
2108 case kTPC: return GetNumberOfSigmasTPC(track, type); break;
2109 case kTRD: return GetNumberOfSigmasTRD(track, type); break;
2110 case kTOF: return GetNumberOfSigmasTOF(track, type); break;
2111 case kHMPID: return GetNumberOfSigmasHMPID(track, type); break;
2112 case kEMCAL: return GetNumberOfSigmasEMCAL(track, type); break;
2113 default: return -999.;
2119 //______________________________________________________________________________
2120 Float_t AliPIDResponse::GetNumberOfSigmasITS(const AliVParticle *vtrack, AliPID::EParticleType type) const
2123 // Calculate the number of sigmas in the ITS
2126 AliVTrack *track=(AliVTrack*)vtrack;
2128 const EDetPidStatus pidStatus=GetITSPIDStatus(track);
2129 if (pidStatus!=kDetPidOk) return -999.;
2131 return fITSResponse.GetNumberOfSigmas(track,type);
2134 //______________________________________________________________________________
2135 Float_t AliPIDResponse::GetNumberOfSigmasTPC(const AliVParticle *vtrack, AliPID::EParticleType type) const
2138 // Calculate the number of sigmas in the TPC
2141 AliVTrack *track=(AliVTrack*)vtrack;
2143 const EDetPidStatus pidStatus=GetTPCPIDStatus(track);
2144 if (pidStatus==kDetNoSignal) return -999.;
2146 // the following call is needed in order to fill the transient data member
2147 // fTPCsignalTuned which is used in the TPCPIDResponse to judge
2148 // if using tuned on data
2149 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))
2150 this->GetTPCsignalTunedOnData(track);
2152 return fTPCResponse.GetNumberOfSigmas(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
2155 //______________________________________________________________________________
2156 Float_t AliPIDResponse::GetNumberOfSigmasTRD(const AliVParticle *vtrack, AliPID::EParticleType type) const
2159 // Calculate the number of sigmas in the TRD
2162 AliVTrack *track=(AliVTrack*)vtrack;
2164 const EDetPidStatus pidStatus=GetTRDPIDStatus(track);
2165 if (pidStatus!=kDetPidOk) return -999.;
2167 return fTRDResponse.GetNumberOfSigmas(track,type);
2170 //______________________________________________________________________________
2171 Float_t AliPIDResponse::GetNumberOfSigmasTOF(const AliVParticle *vtrack, AliPID::EParticleType type) const
2174 // Calculate the number of sigmas in the TOF
2177 AliVTrack *track=(AliVTrack*)vtrack;
2179 const EDetPidStatus pidStatus=GetTOFPIDStatus(track);
2180 if (pidStatus!=kDetPidOk) return -999.;
2182 return GetNumberOfSigmasTOFold(vtrack, type);
2184 //______________________________________________________________________________
2186 Float_t AliPIDResponse::GetNumberOfSigmasHMPID(const AliVParticle *vtrack, AliPID::EParticleType type) const
2189 // Calculate the number of sigmas in the HMPID
2191 AliVTrack *track=(AliVTrack*)vtrack;
2193 const EDetPidStatus pidStatus=GetHMPIDPIDStatus(track);
2194 if (pidStatus!=kDetPidOk) return -999.;
2196 return fHMPIDResponse.GetNumberOfSigmas(track, type);
2199 //______________________________________________________________________________
2200 Float_t AliPIDResponse::GetNumberOfSigmasEMCAL(const AliVParticle *vtrack, AliPID::EParticleType type) const
2203 // Calculate the number of sigmas in the EMCAL
2206 AliVTrack *track=(AliVTrack*)vtrack;
2208 const EDetPidStatus pidStatus=GetEMCALPIDStatus(track);
2209 if (pidStatus!=kDetPidOk) return -999.;
2211 const Int_t nMatchClus = track->GetEMCALcluster();
2212 AliVCluster *matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
2214 const Double_t mom = track->P();
2215 const Double_t pt = track->Pt();
2216 const Int_t charge = track->Charge();
2217 const Double_t fClsE = matchedClus->E();
2218 const Double_t EovP = fClsE/mom;
2220 return fEMCALResponse.GetNumberOfSigmas(pt,EovP,type,charge);
2223 //______________________________________________________________________________
2224 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaITS(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2227 // Signal minus expected Signal for ITS
2229 AliVTrack *track=(AliVTrack*)vtrack;
2230 val=fITSResponse.GetSignalDelta(track,type,ratio);
2232 return GetITSPIDStatus(track);
2235 //______________________________________________________________________________
2236 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaTPC(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2239 // Signal minus expected Signal for TPC
2241 AliVTrack *track=(AliVTrack*)vtrack;
2243 // the following call is needed in order to fill the transient data member
2244 // fTPCsignalTuned which is used in the TPCPIDResponse to judge
2245 // if using tuned on data
2246 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))
2247 this->GetTPCsignalTunedOnData(track);
2249 val=fTPCResponse.GetSignalDelta(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection, ratio);
2251 return GetTPCPIDStatus(track);
2254 //______________________________________________________________________________
2255 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaTRD(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2258 // Signal minus expected Signal for TRD
2260 AliVTrack *track=(AliVTrack*)vtrack;
2261 val=fTRDResponse.GetSignalDelta(track,type,ratio);
2263 return GetTRDPIDStatus(track);
2266 //______________________________________________________________________________
2267 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaTOF(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2270 // Signal minus expected Signal for TOF
2272 AliVTrack *track=(AliVTrack*)vtrack;
2273 val=GetSignalDeltaTOFold(track, type, ratio);
2275 return GetTOFPIDStatus(track);
2278 //______________________________________________________________________________
2279 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaHMPID(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2282 // Signal minus expected Signal for HMPID
2284 AliVTrack *track=(AliVTrack*)vtrack;
2285 val=fHMPIDResponse.GetSignalDelta(track, type, ratio);
2287 return GetHMPIDPIDStatus(track);
2290 //______________________________________________________________________________
2291 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputePIDProbability (EDetector detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2294 // Compute PID response of 'detCode'
2298 case kITS: return GetComputeITSProbability(track, nSpecies, p); break;
2299 case kTPC: return GetComputeTPCProbability(track, nSpecies, p); break;
2300 case kTRD: return GetComputeTRDProbability(track, nSpecies, p); break;
2301 case kTOF: return GetComputeTOFProbability(track, nSpecies, p); break;
2302 case kPHOS: return GetComputePHOSProbability(track, nSpecies, p); break;
2303 case kEMCAL: return GetComputeEMCALProbability(track, nSpecies, p); break;
2304 case kHMPID: return GetComputeHMPIDProbability(track, nSpecies, p); break;
2305 default: return kDetNoSignal;
2309 //______________________________________________________________________________
2310 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeITSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2313 // Compute PID response for the ITS
2316 // set flat distribution (no decision)
2317 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2319 const EDetPidStatus pidStatus=GetITSPIDStatus(track);
2320 if (pidStatus!=kDetPidOk) return pidStatus;
2322 if (track->GetDetectorPID()){
2323 return track->GetDetectorPID()->GetRawProbability(kITS, p, nSpecies);
2326 //check for ITS standalone tracks
2328 if( track->GetStatus() & AliVTrack::kTPCin ) isSA=kFALSE;
2330 Double_t mom=track->P();
2331 Double_t dedx=track->GetITSsignal();
2332 Double_t momITS=mom;
2333 UChar_t clumap=track->GetITSClusterMap();
2334 Int_t nPointsForPid=0;
2335 for(Int_t i=2; i<6; i++){
2336 if(clumap&(1<<i)) ++nPointsForPid;
2339 Bool_t mismatch=kTRUE/*, heavy=kTRUE*/;
2340 for (Int_t j=0; j<nSpecies; j++) {
2341 Double_t mass=AliPID::ParticleMassZ(j);//GeV/c^2
2342 const Double_t chargeFactor = TMath::Power(AliPID::ParticleCharge(j),2.);
2343 Double_t bethe=fITSResponse.Bethe(momITS,mass)*chargeFactor;
2344 //TODO: in case of the electron, use the SA parametrisation,
2345 // this needs to be changed if ITS provides a parametrisation
2346 // for electrons also for ITS+TPC tracks
2347 Double_t sigma=fITSResponse.GetResolution(bethe,nPointsForPid,isSA || (j==(Int_t)AliPID::kElectron));
2348 if (TMath::Abs(dedx-bethe) > fRange*sigma) {
2349 p[j]=TMath::Exp(-0.5*fRange*fRange)/sigma;
2351 p[j]=TMath::Exp(-0.5*(dedx-bethe)*(dedx-bethe)/(sigma*sigma))/sigma;
2357 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2362 //______________________________________________________________________________
2363 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeTPCProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2366 // Compute PID response for the TPC
2369 // set flat distribution (no decision)
2370 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2372 const EDetPidStatus pidStatus=GetTPCPIDStatus(track);
2373 if (pidStatus==kDetNoSignal) return pidStatus;
2375 Double_t dedx=track->GetTPCsignal();
2376 Bool_t mismatch=kTRUE/*, heavy=kTRUE*/;
2378 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC)) dedx = this->GetTPCsignalTunedOnData(track);
2380 Double_t bethe = 0.;
2381 Double_t sigma = 0.;
2383 for (Int_t j=0; j<nSpecies; j++) {
2384 AliPID::EParticleType type=AliPID::EParticleType(j);
2386 bethe=fTPCResponse.GetExpectedSignal(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
2387 sigma=fTPCResponse.GetExpectedSigma(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
2389 if (TMath::Abs(dedx-bethe) > fRange*sigma) {
2390 p[j]=TMath::Exp(-0.5*fRange*fRange)/sigma;
2392 p[j]=TMath::Exp(-0.5*(dedx-bethe)*(dedx-bethe)/(sigma*sigma))/sigma;
2398 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2403 //______________________________________________________________________________
2404 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeTOFProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[],Bool_t kNoMism) const
2407 // Compute PID probabilities for TOF
2410 fgTOFmismatchProb = 1E-8;
2412 // centrality --> fCurrCentrality
2413 // Beam type --> fBeamTypeNum
2414 // N TOF cluster --> TOF header --> to get the TOF header we need to add a virtual method in AliVTrack extended to ESD and AOD tracks
2416 Float_t pt = track->Pt();
2417 Float_t mismPropagationFactor[10] = {1.,1.,1.,1.,1.,1.,1.,1.,1.,1.};
2418 if(! (kNoMism | fNoTOFmism)){ // this flag allows to disable mismatch for iterative procedure to get priors
2419 mismPropagationFactor[3] = 1 + TMath::Exp(1 - 1.12*pt);// it has to be alligned with the one in AliPIDCombined
2420 mismPropagationFactor[4] = 1 + 1./(4.71114 - 5.72372*pt + 2.94715*pt*pt);// it has to be alligned with the one in AliPIDCombined
2422 Int_t nTOFcluster = 0;
2423 if(track->GetTOFHeader() && track->GetTOFHeader()->GetTriggerMask() && track->GetTOFHeader()->GetNumberOfTOFclusters() > -1){ // N TOF clusters available
2424 nTOFcluster = track->GetTOFHeader()->GetNumberOfTOFclusters();
2425 if(fIsMC) nTOFcluster = Int_t(nTOFcluster * 1.5); // +50% in MC
2428 switch(fBeamTypeNum){
2432 case kPPB: // pPb 5.05 ATeV
2433 nTOFcluster = Int_t(308 - 2.12*fCurrCentrality + TMath::Exp(4.917 -0.1604*fCurrCentrality));
2435 case kPBPB: // PbPb 2.76 ATeV
2436 nTOFcluster = Int_t(TMath::Exp(9.4 - 0.022*fCurrCentrality));
2441 switch(fBeamTypeNum){ // matching window factors for 3 cm and 10 cm (about (10/3)^2)
2442 case kPP: // pp 7 TeV
2445 case kPPB: // pPb 5.05 ATeV
2448 case kPBPB: // pPb 5.05 ATeV
2449 // nTOFcluster *= 1;
2453 if(nTOFcluster < 0) nTOFcluster = 10;
2456 fgTOFmismatchProb=fTOFResponse.GetMismatchProbability(track->GetTOFsignal(),track->Eta()) * nTOFcluster *6E-6 * (1 + 2.90505e-01/pt/pt); // mism weight * tof occupancy (including matching window factor) * pt dependence
2460 // set flat distribution (no decision)
2461 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2463 const EDetPidStatus pidStatus=GetTOFPIDStatus(track);
2464 if (pidStatus!=kDetPidOk) return pidStatus;
2466 const Double_t meanCorrFactor = 0.07/fTOFtail; // Correction factor on the mean because of the tail (should be ~ 0.1 with tail = 1.1)
2468 for (Int_t j=0; j<nSpecies; j++) {
2469 AliPID::EParticleType type=AliPID::EParticleType(j);
2470 const Double_t nsigmas=GetNumberOfSigmasTOFold(track,type) + meanCorrFactor;
2472 const Double_t expTime = fTOFResponse.GetExpectedSignal(track,type);
2473 const Double_t sig = fTOFResponse.GetExpectedSigma(track->P(),expTime,AliPID::ParticleMassZ(type));
2475 if(nsigmas < fTOFtail)
2476 p[j] = TMath::Exp(-0.5*nsigmas*nsigmas)/sig;
2478 p[j] = TMath::Exp(-(nsigmas - fTOFtail*0.5)*fTOFtail)/sig;
2480 p[j] += fgTOFmismatchProb*mismPropagationFactor[j];
2486 Int_t AliPIDResponse::CalculateTRDResponse(const AliVTrack *track,Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const
2488 // new function for backward compatibility
2489 // returns number of tracklets PID
2491 UInt_t TRDslicesForPID[2];
2492 SetTRDSlices(TRDslicesForPID,PIDmethod);
2494 Float_t mom[6]={0.};
2495 Double_t dedx[48]={0.}; // Allocate space for the maximum number of TRD slices
2496 Int_t nslices = TRDslicesForPID[1] - TRDslicesForPID[0] + 1;
2497 AliDebug(1, Form("First Slice: %d, Last Slice: %d, Number of slices: %d", TRDslicesForPID[0], TRDslicesForPID[1], nslices));
2498 for(UInt_t ilayer = 0; ilayer < 6; ilayer++){
2499 mom[ilayer] = track->GetTRDmomentum(ilayer);
2500 for(UInt_t islice = TRDslicesForPID[0]; islice <= TRDslicesForPID[1]; islice++){
2501 dedx[ilayer*nslices+islice-TRDslicesForPID[0]] = track->GetTRDslice(ilayer, islice);
2505 return fTRDResponse.GetResponse(nslices, dedx, mom, p,PIDmethod);
2508 //______________________________________________________________________________
2509 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const
2512 // Compute PID probabilities for the TRD
2515 // set flat distribution (no decision)
2516 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2518 const EDetPidStatus pidStatus=GetTRDPIDStatus(track);
2519 if (pidStatus!=kDetPidOk) return pidStatus;
2521 CalculateTRDResponse(track,p,PIDmethod);
2526 //______________________________________________________________________________
2527 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeEMCALProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2530 // Compute PID response for the EMCAL
2533 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2535 const EDetPidStatus pidStatus=GetEMCALPIDStatus(track);
2536 if (pidStatus!=kDetPidOk) return pidStatus;
2538 const Int_t nMatchClus = track->GetEMCALcluster();
2539 AliVCluster *matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
2541 const Double_t mom = track->P();
2542 const Double_t pt = track->Pt();
2543 const Int_t charge = track->Charge();
2544 const Double_t fClsE = matchedClus->E();
2545 const Double_t EovP = fClsE/mom;
2547 // compute the probabilities
2548 fEMCALResponse.ComputeEMCALProbability(nSpecies,pt,EovP,charge,p);
2552 //______________________________________________________________________________
2553 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputePHOSProbability (const AliVTrack */*track*/, Int_t nSpecies, Double_t p[]) const
2556 // Compute PID response for the PHOS
2559 // set flat distribution (no decision)
2560 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2561 return kDetNoSignal;
2564 //______________________________________________________________________________
2565 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeHMPIDProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2568 // Compute PID response for the HMPID
2571 // set flat distribution (no decision)
2572 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2574 const EDetPidStatus pidStatus=GetHMPIDPIDStatus(track);
2575 if (pidStatus!=kDetPidOk) return pidStatus;
2577 fHMPIDResponse.GetProbability(track,nSpecies,p);
2582 //______________________________________________________________________________
2583 AliPIDResponse::EDetPidStatus AliPIDResponse::GetITSPIDStatus(const AliVTrack *track) const
2585 // compute ITS pid status
2587 // check status bits
2588 if ((track->GetStatus()&AliVTrack::kITSin)==0 &&
2589 (track->GetStatus()&AliVTrack::kITSout)==0) return kDetNoSignal;
2591 const Float_t dEdx=track->GetITSsignal();
2592 if (dEdx<=0) return kDetNoSignal;
2594 // requite at least 3 pid clusters
2595 const UChar_t clumap=track->GetITSClusterMap();
2596 Int_t nPointsForPid=0;
2597 for(Int_t i=2; i<6; i++){
2598 if(clumap&(1<<i)) ++nPointsForPid;
2601 if(nPointsForPid<3) {
2602 return kDetNoSignal;
2608 //______________________________________________________________________________
2609 AliPIDResponse::EDetPidStatus AliPIDResponse:: GetTPCPIDStatus(const AliVTrack *track) const
2611 // compute TPC pid status
2613 // check quality of the track
2614 if ( (track->GetStatus()&AliVTrack::kTPCin )==0 && (track->GetStatus()&AliVTrack::kTPCout)==0 ) return kDetNoSignal;
2617 const Double_t dedx=track->GetTPCsignal();
2618 const UShort_t signalN=track->GetTPCsignalN();
2619 if (signalN<10 || dedx<10) return kDetNoSignal;
2621 if (!(fArrPidResponseMaster && fArrPidResponseMaster->At(AliPID::kPion))) return kDetNoParams;
2626 //______________________________________________________________________________
2627 AliPIDResponse::EDetPidStatus AliPIDResponse::GetTRDPIDStatus(const AliVTrack *track) const
2629 // compute TRD pid status
2631 if((track->GetStatus()&AliVTrack::kTRDout)==0) return kDetNoSignal;
2635 //______________________________________________________________________________
2636 AliPIDResponse::EDetPidStatus AliPIDResponse::GetTOFPIDStatus(const AliVTrack *track) const
2638 // compute TOF pid status
2640 if ((track->GetStatus()&AliVTrack::kTOFout)==0) return kDetNoSignal;
2641 if ((track->GetStatus()&AliVTrack::kTIME)==0) return kDetNoSignal;
2646 //______________________________________________________________________________
2647 Float_t AliPIDResponse::GetTOFMismatchProbability(const AliVTrack *track) const
2649 // compute mismatch probability cross-checking at 5 sigmas with TPC
2650 // currently just implemented as a 5 sigma compatibility cut
2652 if(!track) return fgTOFmismatchProb;
2655 const EDetPidStatus tofStatus=GetTOFPIDStatus(track);
2656 if (tofStatus!=kDetPidOk) return 0.;
2659 const EDetPidStatus tpcStatus=GetTPCPIDStatus(track);
2660 if (tpcStatus==kDetNoSignal) return 0.;
2662 const Double_t meanCorrFactor = 0.11/fTOFtail; // Correction factor on the mean because of the tail (should be ~ 0.1 with tail = 1.1)
2663 Bool_t mismatch = kTRUE/*, heavy = kTRUE*/;
2664 for (Int_t j=0; j<AliPID::kSPECIESC; j++) {
2665 AliPID::EParticleType type=AliPID::EParticleType(j);
2666 const Double_t nsigmas=GetNumberOfSigmasTOFold(track,type) + meanCorrFactor;
2668 if (TMath::Abs(nsigmas)<5.){
2669 const Double_t nsigmasTPC=GetNumberOfSigmasTPC(track,type);
2670 if (TMath::Abs(nsigmasTPC)<5.) mismatch=kFALSE;
2681 //______________________________________________________________________________
2682 AliPIDResponse::EDetPidStatus AliPIDResponse:: GetHMPIDPIDStatus(const AliVTrack *track) const
2684 // compute HMPID pid status
2686 Int_t ch = track->GetHMPIDcluIdx()/1000000;
2687 Double_t HMPIDsignal = track->GetHMPIDsignal();
2689 if((track->GetStatus()&AliVTrack::kHMPIDpid)==0 || ch<0 || ch>6 || HMPIDsignal<0) return kDetNoSignal;
2694 //______________________________________________________________________________
2695 AliPIDResponse::EDetPidStatus AliPIDResponse:: GetPHOSPIDStatus(const AliVTrack */*track*/) const
2697 // compute PHOS pid status
2698 return kDetNoSignal;
2701 //______________________________________________________________________________
2702 AliPIDResponse::EDetPidStatus AliPIDResponse:: GetEMCALPIDStatus(const AliVTrack *track) const
2704 // compute EMCAL pid status
2708 const Int_t nMatchClus = track->GetEMCALcluster();
2709 if (nMatchClus<0) return kDetNoSignal;
2711 AliVCluster *matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
2713 if (!(matchedClus && matchedClus->IsEMCAL())) return kDetNoSignal;
2715 const Int_t charge = track->Charge();
2716 if (TMath::Abs(charge)!=1) return kDetNoSignal;
2718 if (!(fEMCALPIDParams && fEMCALPIDParams->At(AliPID::kElectron))) return kDetNoParams;
2724 //______________________________________________________________________________
2725 AliPIDResponse::EDetPidStatus AliPIDResponse::GetPIDStatus(EDetector detector, const AliVTrack *track) const
2728 // check pid status for a track
2732 case kITS: return GetITSPIDStatus(track); break;
2733 case kTPC: return GetTPCPIDStatus(track); break;
2734 case kTRD: return GetTRDPIDStatus(track); break;
2735 case kTOF: return GetTOFPIDStatus(track); break;
2736 case kPHOS: return GetPHOSPIDStatus(track); break;
2737 case kEMCAL: return GetEMCALPIDStatus(track); break;
2738 case kHMPID: return GetHMPIDPIDStatus(track); break;
2739 default: return kDetNoSignal;
2741 return kDetNoSignal;
2745 //______________________________________________________________________________
2746 TString AliPIDResponse::GetChecksum(const TObject* obj) const
2748 // Return the checksum for an object obj (tested to work properly at least for histograms and TSplines).
2750 TString fileName = Form("tempChecksum.C"); // File name must be fixed for data type "TSpline3", since the file name will end up in the file content!
2752 // For parallel processing, a unique file pathname is required. Uniqueness can be guaranteed by using a unique directory name
2754 TString uniquePathName = Form("tempChecksum_%u", index);
2756 // To get a unique path name, increase the index until no directory
2757 // of such a name exists.
2758 // NOTE: gSystem->AccessPathName(...) returns kTRUE, if the access FAILED!
2759 while (!gSystem->AccessPathName(uniquePathName.Data()))
2760 uniquePathName = Form("tempChecksum_%u", ++index);
2762 if (gSystem->mkdir(uniquePathName.Data()) < 0) {
2763 AliError("Could not create temporary directory to store temp file for checksum determination!");
2767 TString option = "";
2769 // Save object as a macro, which will be deleted immediately after the checksum has been computed
2770 // (does not work for desired data types if saved as *.root for some reason) - one only wants to compare the content, not
2771 // the modification time etc. ...
2772 if (dynamic_cast<const TH1*>(obj))
2773 option = "colz"; // Histos need this option, since w/o this option, a counter is added to the filename
2776 // SaveAs must be called with the fixed fileName only, since the first argument goes into the file content
2777 // for some object types. Thus, change the directory, save the file and then go back
2778 TString oldDir = gSystem->pwd();
2779 gSystem->cd(uniquePathName.Data());
2780 obj->SaveAs(fileName.Data(), option.Data());
2781 gSystem->cd(oldDir.Data());
2783 // Use the file to calculate the MD5 checksum
2784 TMD5* md5 = TMD5::FileChecksum(Form("%s/%s", uniquePathName.Data(), fileName.Data()));
2785 TString checksum = md5->AsString();
2789 gSystem->Exec(Form("rm -rf %s", uniquePathName.Data()));