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 <AliTOFPIDParams.h>
46 #include <AliHMPIDPIDParams.h>
48 #include "AliPIDResponse.h"
49 #include "AliDetectorPID.h"
51 #include "AliCentrality.h"
53 ClassImp(AliPIDResponse);
55 AliPIDResponse::AliPIDResponse(Bool_t isMC/*=kFALSE*/) :
56 TNamed("PIDResponse","PIDResponse"),
64 fITSPIDmethod(kITSTruncMean),
65 fTuneMConData(kFALSE),
66 fTuneMConDataMask(kDetTOF|kDetTPC),
70 fCustomTPCpidResponse(),
76 fCurrentAliRootRev(-1),
84 fArrPidResponseMaster(NULL),
85 fResolutionCorrection(NULL),
86 fOADBvoltageMaps(NULL),
87 fUseTPCEtaCorrection(kFALSE),
88 fUseTPCMultiplicityCorrection(kFALSE),
89 fTRDPIDResponseObject(NULL),
92 fHMPIDPIDParams(NULL),
93 fEMCALPIDParams(NULL),
100 AliLog::SetClassDebugLevel("AliPIDResponse",0);
101 AliLog::SetClassDebugLevel("AliESDpid",0);
102 AliLog::SetClassDebugLevel("AliAODpidUtil",0);
106 //______________________________________________________________________________
107 AliPIDResponse::~AliPIDResponse()
112 delete fArrPidResponseMaster;
113 delete fTRDPIDResponseObject;
114 delete fTOFPIDParams;
117 //______________________________________________________________________________
118 AliPIDResponse::AliPIDResponse(const AliPIDResponse &other) :
120 fITSResponse(other.fITSResponse),
121 fTPCResponse(other.fTPCResponse),
122 fTRDResponse(other.fTRDResponse),
123 fTOFResponse(other.fTOFResponse),
124 fHMPIDResponse(other.fHMPIDResponse),
125 fEMCALResponse(other.fEMCALResponse),
126 fRange(other.fRange),
127 fITSPIDmethod(other.fITSPIDmethod),
128 fTuneMConData(other.fTuneMConData),
129 fTuneMConDataMask(other.fTuneMConDataMask),
131 fCachePID(other.fCachePID),
132 fOADBPath(other.fOADBPath),
133 fCustomTPCpidResponse(other.fCustomTPCpidResponse),
137 fMCperiodUser(other.fMCperiodUser),
139 fCurrentAliRootRev(other.fCurrentAliRootRev),
141 fRecoPassUser(other.fRecoPassUser),
147 fArrPidResponseMaster(NULL),
148 fResolutionCorrection(NULL),
149 fOADBvoltageMaps(NULL),
150 fUseTPCEtaCorrection(other.fUseTPCEtaCorrection),
151 fUseTPCMultiplicityCorrection(other.fUseTPCMultiplicityCorrection),
152 fTRDPIDResponseObject(NULL),
155 fHMPIDPIDParams(NULL),
156 fEMCALPIDParams(NULL),
165 //______________________________________________________________________________
166 AliPIDResponse& AliPIDResponse::operator=(const AliPIDResponse &other)
172 delete fArrPidResponseMaster;
173 TNamed::operator=(other);
174 fITSResponse=other.fITSResponse;
175 fTPCResponse=other.fTPCResponse;
176 fTRDResponse=other.fTRDResponse;
177 fTOFResponse=other.fTOFResponse;
178 fHMPIDResponse=other.fHMPIDResponse;
179 fEMCALResponse=other.fEMCALResponse;
181 fITSPIDmethod=other.fITSPIDmethod;
182 fOADBPath=other.fOADBPath;
183 fCustomTPCpidResponse=other.fCustomTPCpidResponse;
184 fTuneMConData=other.fTuneMConData;
185 fTuneMConDataMask=other.fTuneMConDataMask;
187 fCachePID=other.fCachePID;
191 fMCperiodUser=other.fMCperiodUser;
193 fCurrentAliRootRev=other.fCurrentAliRootRev;
195 fRecoPassUser=other.fRecoPassUser;
201 fArrPidResponseMaster=NULL;
202 fResolutionCorrection=NULL;
203 fOADBvoltageMaps=NULL;
204 fUseTPCEtaCorrection=other.fUseTPCEtaCorrection;
205 fUseTPCMultiplicityCorrection=other.fUseTPCMultiplicityCorrection;
206 fTRDPIDResponseObject=NULL;
207 fEMCALPIDParams=NULL;
210 fHMPIDPIDParams=NULL;
211 fCurrentEvent=other.fCurrentEvent;
217 //______________________________________________________________________________
218 Float_t AliPIDResponse::NumberOfSigmas(EDetector detector, const AliVParticle *vtrack, AliPID::EParticleType type) const
221 // NumberOfSigmas for 'detCode'
224 const AliVTrack *track=static_cast<const AliVTrack*>(vtrack);
225 // look for cached value first
226 const AliDetectorPID *detPID=track->GetDetectorPID();
228 if ( detPID && detPID->HasNumberOfSigmas(detector)){
229 return detPID->GetNumberOfSigmas(detector, type);
230 } else if (fCachePID) {
231 FillTrackDetectorPID(track, detector);
232 detPID=track->GetDetectorPID();
233 return detPID->GetNumberOfSigmas(detector, type);
236 return GetNumberOfSigmas(detector, track, type);
239 //______________________________________________________________________________
240 AliPIDResponse::EDetPidStatus AliPIDResponse::NumberOfSigmas(EDetector detCode, const AliVParticle *track,
241 AliPID::EParticleType type, Double_t &val) const
244 // NumberOfSigmas with detector status as return value
247 val=NumberOfSigmas(detCode, track, type);
248 return CheckPIDStatus(detCode, (AliVTrack*)track);
251 //______________________________________________________________________________
252 // public buffered versions of the PID calculation
255 //______________________________________________________________________________
256 Float_t AliPIDResponse::NumberOfSigmasITS(const AliVParticle *vtrack, AliPID::EParticleType type) const
259 // Calculate the number of sigmas in the ITS
262 return NumberOfSigmas(kITS, vtrack, type);
265 //______________________________________________________________________________
266 Float_t AliPIDResponse::NumberOfSigmasTPC(const AliVParticle *vtrack, AliPID::EParticleType type) const
269 // Calculate the number of sigmas in the TPC
272 return NumberOfSigmas(kTPC, vtrack, type);
275 //______________________________________________________________________________
276 Float_t AliPIDResponse::NumberOfSigmasTPC( const AliVParticle *vtrack,
277 AliPID::EParticleType type,
278 AliTPCPIDResponse::ETPCdEdxSource dedxSource) const
280 //get number of sigmas according the selected TPC gain configuration scenario
281 const AliVTrack *track=static_cast<const AliVTrack*>(vtrack);
283 Float_t nSigma=fTPCResponse.GetNumberOfSigmas(track, type, dedxSource, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
288 //______________________________________________________________________________
289 Float_t AliPIDResponse::NumberOfSigmasTOF(const AliVParticle *vtrack, AliPID::EParticleType type) const
292 // Calculate the number of sigmas in the TOF
295 return NumberOfSigmas(kTOF, vtrack, type);
298 //______________________________________________________________________________
299 Float_t AliPIDResponse::NumberOfSigmasHMPID(const AliVParticle *vtrack, AliPID::EParticleType type) const
302 // Calculate the number of sigmas in the EMCAL
305 return NumberOfSigmas(kHMPID, vtrack, type);
308 //______________________________________________________________________________
309 Float_t AliPIDResponse::NumberOfSigmasEMCAL(const AliVParticle *vtrack, AliPID::EParticleType type) const
312 // Calculate the number of sigmas in the EMCAL
315 return NumberOfSigmas(kEMCAL, vtrack, type);
318 //______________________________________________________________________________
319 Float_t AliPIDResponse::NumberOfSigmasEMCAL(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &eop, Double_t showershape[4]) const
322 // emcal nsigma with eop and showershape
324 AliVTrack *track=(AliVTrack*)vtrack;
326 AliVCluster *matchedClus = NULL;
331 Double_t fClsE = -1.;
333 // initialize eop and shower shape parameters
335 for(Int_t i = 0; i < 4; i++){
336 showershape[i] = -1.;
339 Int_t nMatchClus = -1;
343 nMatchClus = track->GetEMCALcluster();
348 charge = track->Charge();
350 matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
354 // matched cluster is EMCAL
355 if(matchedClus->IsEMCAL()){
357 fClsE = matchedClus->E();
360 // fill used EMCAL variables here
362 showershape[0] = matchedClus->GetNCells(); // number of cells in cluster
363 showershape[1] = matchedClus->GetM02(); // long axis
364 showershape[2] = matchedClus->GetM20(); // short axis
365 showershape[3] = matchedClus->GetDispersion(); // dispersion
367 // look for cached value first
368 const AliDetectorPID *detPID=track->GetDetectorPID();
369 const EDetector detector=kEMCAL;
371 if ( detPID && detPID->HasNumberOfSigmas(detector)){
372 return detPID->GetNumberOfSigmas(detector, type);
373 } else if (fCachePID) {
374 FillTrackDetectorPID(track, detector);
375 detPID=track->GetDetectorPID();
376 return detPID->GetNumberOfSigmas(detector, type);
379 // NSigma value really meaningful only for electrons!
380 return fEMCALResponse.GetNumberOfSigmas(pt,EovP,type,charge);
387 //______________________________________________________________________________
388 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDelta(EDetector detector, const AliVParticle *track, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
395 case kITS: return GetSignalDeltaITS(track,type,val,ratio); break;
396 case kTPC: return GetSignalDeltaTPC(track,type,val,ratio); break;
397 case kTOF: return GetSignalDeltaTOF(track,type,val,ratio); break;
398 case kHMPID: return GetSignalDeltaHMPID(track,type,val,ratio); break;
399 default: return kDetNoSignal;
404 //______________________________________________________________________________
405 Double_t AliPIDResponse::GetSignalDelta(EDetector detCode, const AliVParticle *track, AliPID::EParticleType type, Bool_t ratio/*=kFALSE*/) const
411 EDetPidStatus stat=GetSignalDelta(detCode, track, type, val, ratio);
412 if ( stat==kDetNoSignal ) val=-9999.;
416 //______________________________________________________________________________
417 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputePIDProbability (EDetCode detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
419 // Compute PID response of 'detCode'
421 // find detector code from detector bit mask
423 for (Int_t idet=0; idet<kNdetectors; ++idet) if ( (detCode&(1<<idet)) ) { detector=idet; break; }
424 if (detector==-1) return kDetNoSignal;
426 return ComputePIDProbability((EDetector)detector, track, nSpecies, p);
429 //______________________________________________________________________________
430 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputePIDProbability (EDetector detector, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
433 // Compute PID response of 'detector'
436 const AliDetectorPID *detPID=track->GetDetectorPID();
438 if ( detPID && detPID->HasRawProbability(detector)){
439 return detPID->GetRawProbability(detector, p, nSpecies);
440 } else if (fCachePID) {
441 FillTrackDetectorPID(track, detector);
442 detPID=track->GetDetectorPID();
443 return detPID->GetRawProbability(detector, p, nSpecies);
446 //if no caching return values calculated from scratch
447 return GetComputePIDProbability(detector, track, nSpecies, p);
450 //______________________________________________________________________________
451 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeITSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
453 // Compute PID response for the ITS
454 return ComputePIDProbability(kITS, track, nSpecies, p);
457 //______________________________________________________________________________
458 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTPCProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
460 // Compute PID response for the TPC
461 return ComputePIDProbability(kTPC, track, nSpecies, p);
464 //______________________________________________________________________________
465 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTOFProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
467 // Compute PID response for the
468 return ComputePIDProbability(kTOF, track, nSpecies, p);
471 //______________________________________________________________________________
472 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
474 // Compute PID response for the
475 return ComputePIDProbability(kTRD, track, nSpecies, p);
478 //______________________________________________________________________________
479 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeEMCALProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
481 // Compute PID response for the EMCAL
482 return ComputePIDProbability(kEMCAL, track, nSpecies, p);
484 //______________________________________________________________________________
485 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputePHOSProbability (const AliVTrack */*track*/, Int_t nSpecies, Double_t p[]) const
487 // Compute PID response for the PHOS
489 // set flat distribution (no decision)
490 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
494 //______________________________________________________________________________
495 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeHMPIDProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
497 // Compute PID response for the HMPID
498 return ComputePIDProbability(kHMPID, track, nSpecies, p);
501 //______________________________________________________________________________
502 AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const
504 // Compute PID response for the
505 return GetComputeTRDProbability(track, nSpecies, p, PIDmethod);
508 //______________________________________________________________________________
509 AliPIDResponse::EDetPidStatus AliPIDResponse::CheckPIDStatus(EDetector detector, const AliVTrack *track) const
511 // calculate detector pid status
513 const Int_t iDetCode=(Int_t)detector;
514 if (iDetCode<0||iDetCode>=kNdetectors) return kDetNoSignal;
515 const AliDetectorPID *detPID=track->GetDetectorPID();
518 return detPID->GetPIDStatus(detector);
519 } else if (fCachePID) {
520 FillTrackDetectorPID(track, detector);
521 detPID=track->GetDetectorPID();
522 return detPID->GetPIDStatus(detector);
525 // if not buffered and no buffering is requested
526 return GetPIDStatus(detector, track);
529 //______________________________________________________________________________
530 void AliPIDResponse::InitialiseEvent(AliVEvent *event, Int_t pass, Int_t run)
533 // Apply settings for the current event
542 else fRun=event->GetRunNumber();
549 //TPC resolution parametrisation PbPb
550 if ( fResolutionCorrection ){
551 Double_t corrSigma=fResolutionCorrection->Eval(GetTPCMultiplicityBin(event));
552 fTPCResponse.SetSigma(3.79301e-03*corrSigma, 2.21280e+04);
555 // Set up TPC multiplicity for PbPb
556 //TODO Will NOT give the desired number for AODs -> Needs new variable/function in future.
557 // Fatal, if AOD event and correction enabled
558 //printf("DETECTED class: %s (%d)\n\n\n\n", event->IsA()->GetName(), fUseTPCMultiplicityCorrection);//TODO
559 if (fUseTPCMultiplicityCorrection && strcmp(event->IsA()->GetName(), "AliESDEvent") != 0) {
560 AliFatal("TPC multiplicity correction is enabled, but will NOT work for AOD events, only for ESD => Disabled multiplicity correction!");
561 fUseTPCMultiplicityCorrection = kFALSE;
564 if (fUseTPCMultiplicityCorrection)
565 fTPCResponse.SetCurrentEventMultiplicity(event->GetNumberOfTracks());
567 fTPCResponse.SetCurrentEventMultiplicity(0);
570 SetTOFResponse(event, (AliPIDResponse::EStartTimeType_t)fTOFPIDParams->GetStartTimeMethod());
573 // Get and set centrality
574 AliCentrality *centrality = event->GetCentrality();
576 fCurrCentrality = centrality->GetCentralityPercentile("V0M");
579 fCurrCentrality = -1;
582 // Set centrality percentile for EMCAL
583 fEMCALResponse.SetCentrality(fCurrCentrality);
585 // switch off some TOF channel according to OADB to match data TOF matching eff
586 if (fTuneMConData && ((fTuneMConDataMask & kDetTOF) == kDetTOF) && fTOFPIDParams->GetTOFmatchingLossMC() > 0.01){
587 Int_t ntrk = event->GetNumberOfTracks();
588 for(Int_t i=0;i < ntrk;i++){
589 AliVParticle *trk = event->GetTrack(i);
590 Int_t channel = GetTOFResponse().GetTOFchannel(trk);
591 Int_t swoffEachOfThem = Int_t(100./fTOFPIDParams->GetTOFmatchingLossMC() + 0.5);
592 if(!(channel%swoffEachOfThem)) ((AliVTrack *) trk)->ResetStatus(AliVTrack::kTOFout);
598 //______________________________________________________________________________
599 void AliPIDResponse::ExecNewRun()
602 // Things to Execute upon a new run
606 SetITSParametrisation();
608 SetTPCPidResponseMaster();
609 SetTPCParametrisation();
612 SetTRDPidResponseMaster();
613 InitializeTRDResponse();
615 SetEMCALPidResponseMaster();
616 InitializeEMCALResponse();
618 SetTOFPidResponseMaster();
619 InitializeTOFResponse();
621 SetHMPIDPidResponseMaster();
622 InitializeHMPIDResponse();
624 if (fCurrentEvent) fTPCResponse.SetMagField(fCurrentEvent->GetMagneticField());
627 //______________________________________________________________________________
628 Double_t AliPIDResponse::GetTPCMultiplicityBin(const AliVEvent * const event)
631 // Get TPC multiplicity in bins of 150
634 const AliVVertex* vertexTPC = event->GetPrimaryVertex();
635 Double_t tpcMulti=0.;
637 Double_t vertexContribTPC=vertexTPC->GetNContributors();
638 tpcMulti=vertexContribTPC/150.;
639 if (tpcMulti>20.) tpcMulti=20.;
645 //______________________________________________________________________________
646 void AliPIDResponse::SetRecoInfo()
649 // Set reconstruction information
660 Bool_t hasProdInfo=(fCurrentFile.BeginsWith("LHC"));
662 TPRegexp reg(".*(LHC1[1-3][a-z]+[0-9]+[a-z_]*)/.*");
663 if (hasProdInfo) reg=TPRegexp("LHC1[1-2][a-z]+[0-9]+[a-z_]*");
664 TPRegexp reg12a17("LHC1[2-3][a-z]");
666 //find the period by run number (UGLY, but not stored in ESD and AOD... )
667 if (fRun>=114737&&fRun<=117223) { fLHCperiod="LHC10B"; fMCperiodTPC="LHC10D1"; }
668 else if (fRun>=118503&&fRun<=121040) { fLHCperiod="LHC10C"; fMCperiodTPC="LHC10D1"; }
669 else if (fRun>=122195&&fRun<=126437) { fLHCperiod="LHC10D"; fMCperiodTPC="LHC10F6A"; }
670 else if (fRun>=127710&&fRun<=130850) { fLHCperiod="LHC10E"; fMCperiodTPC="LHC10F6A"; }
671 else if (fRun>=133004&&fRun<=135029) { fLHCperiod="LHC10F"; fMCperiodTPC="LHC10F6A"; }
672 else if (fRun>=135654&&fRun<=136377) { fLHCperiod="LHC10G"; fMCperiodTPC="LHC10F6A"; }
673 else if (fRun>=136851&&fRun<=139846) {
675 fMCperiodTPC="LHC10H8";
676 if (reg.MatchB(fCurrentFile)) fMCperiodTPC="LHC11A10";
677 // exception for 13d2 and later
678 if (fCurrentAliRootRev >= 62714) fMCperiodTPC="LHC13D2";
681 else if (fRun>=139847&&fRun<=146974) { fLHCperiod="LHC11A"; fMCperiodTPC="LHC10F6A"; }
682 //TODO: periods 11B (146975-150721), 11C (150722-155837) are not yet treated assume 11d for the moment
683 else if (fRun>=146975&&fRun<=155837) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
684 else if (fRun>=155838&&fRun<=159649) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
685 // also for 11e (159650-162750),f(162751-165771) use 11d
686 else if (fRun>=159650&&fRun<=162750) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
687 else if (fRun>=162751&&fRun<=165771) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
689 else if (fRun>=165772 && fRun<=170718) {
691 fMCperiodTPC="LHC11A10";
693 if (reg12a17.MatchB(fCurrentFile)) fMCperiodTPC="LHC12A17";
695 if (fRun>=170719 && fRun<=177311) { fLHCperiod="LHC12A"; fBeamType="PP"; /*fMCperiodTPC="";*/ }
696 // for the moment use LHC12b parameters up to LHC12d
697 if (fRun>=177312 /*&& fRun<=179356*/) { fLHCperiod="LHC12B"; fBeamType="PP"; /*fMCperiodTPC="";*/ }
698 // if (fRun>=179357 && fRun<=183173) { fLHCperiod="LHC12C"; fBeamType="PP"; /*fMCperiodTPC="";*/ }
699 // if (fRun>=183174 && fRun<=186345) { fLHCperiod="LHC12D"; fBeamType="PP"; /*fMCperiodTPC="";*/ }
700 // if (fRun>=186346 && fRun<=186635) { fLHCperiod="LHC12E"; fBeamType="PP"; /*fMCperiodTPC="";*/ }
702 // if (fRun>=186636 && fRun<=188166) { fLHCperiod="LHC12F"; fBeamType="PP"; /*fMCperiodTPC="";*/ }
703 // if (fRun >= 188167 && fRun <= 188355 ) { fLHCperiod="LHC12G"; fBeamType="PP"; /*fMCperiodTPC="";*/ }
704 // if (fRun >= 188356 && fRun <= 188503 ) { fLHCperiod="LHC12G"; fBeamType="PPB"; /*fMCperiodTPC="";*/ }
705 // for the moment use 12g parametrisation for all full gain runs (LHC12e+)
706 if (fRun >= 186346 && fRun < 194480) { fLHCperiod="LHC12G"; fBeamType="PPB"; fMCperiodTPC="LHC12G"; }
708 // New parametrisation for 2013 pPb runs
709 if (fRun >= 194480) {
712 fMCperiodTPC="LHC12G";
714 if (fCurrentAliRootRev >= 61605)
715 fMCperiodTPC="LHC13B2_FIX";
716 if (fCurrentAliRootRev >= 62714)
717 fMCperiodTPC="LHC13B2_FIXn1";
719 // High luminosity pPb runs require different parametrisations
720 if (fRun >= 195875 && fRun <= 197411) {
725 //exception new pp MC productions from 2011 (11a periods have 10f6a splines!)
726 if (fBeamType=="PP" && reg.MatchB(fCurrentFile) && !fCurrentFile.Contains("LHC11a")) { fMCperiodTPC="LHC11B2"; fBeamType="PP"; }
727 // exception for 11f1
728 if (fCurrentFile.Contains("LHC11f1")) fMCperiodTPC="LHC11F1";
729 // exception for 12f1a, 12f1b and 12i3
730 if (fCurrentFile.Contains("LHC12f1") || fCurrentFile.Contains("LHC12i3")) fMCperiodTPC="LHC12F1";
731 // exception for 12c4
732 if (fCurrentFile.Contains("LHC12c4")) fMCperiodTPC="LHC12C4";
733 // exception for 12d and 13d pp periods
734 if (fBeamType=="PP" && fCurrentAliRootRev >= 61605) fMCperiodTPC="LHC13D1";
737 //______________________________________________________________________________
738 void AliPIDResponse::SetITSParametrisation()
741 // Set the ITS parametrisation
746 //______________________________________________________________________________
747 void AliPIDResponse::AddPointToHyperplane(TH2D* h, TLinearFitter* linExtrapolation, Int_t binX, Int_t binY)
749 if (h->GetBinContent(binX, binY) <= 1e-4)
750 return; // Reject bins without content (within some numerical precision) or with strange content
752 Double_t coord[2] = {0, 0};
753 coord[0] = h->GetXaxis()->GetBinCenter(binX);
754 coord[1] = h->GetYaxis()->GetBinCenter(binY);
755 Double_t binError = h->GetBinError(binX, binY);
757 binError = 1000; // Should not happen because bins without content are rejected for the map (TH2D* h)
758 printf("ERROR: This should never happen: Trying to add bin in addPointToHyperplane with error not set....\n");
760 linExtrapolation->AddPoint(coord, h->GetBinContent(binX, binY, binError));
764 //______________________________________________________________________________
765 TH2D* AliPIDResponse::RefineHistoViaLinearInterpolation(TH2D* h, Double_t refineFactorX, Double_t refineFactorY)
770 // Interpolate to finer map
771 TLinearFitter* linExtrapolation = new TLinearFitter(2, "hyp2", "");
773 Double_t upperMapBoundY = h->GetYaxis()->GetBinUpEdge(h->GetYaxis()->GetNbins());
774 Double_t lowerMapBoundY = h->GetYaxis()->GetBinLowEdge(1);
776 // 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,
777 // scale the number of bins correspondingly
778 Int_t nBinsY = TMath::Nint((upperMapBoundY - lowerMapBoundY) / (0.02 - 0.0016) * 40);
779 Int_t nBinsXrefined = nBinsX * refineFactorX;
780 Int_t nBinsYrefined = nBinsY * refineFactorY;
782 TH2D* hRefined = new TH2D(Form("%s_refined", h->GetName()), Form("%s (refined)", h->GetTitle()),
783 nBinsXrefined, h->GetXaxis()->GetBinLowEdge(1), h->GetXaxis()->GetBinUpEdge(h->GetXaxis()->GetNbins()),
784 nBinsYrefined, lowerMapBoundY, upperMapBoundY);
786 for (Int_t binX = 1; binX <= nBinsXrefined; binX++) {
787 for (Int_t binY = 1; binY <= nBinsYrefined; binY++) {
789 hRefined->SetBinContent(binX, binY, 1); // Default value is 1
791 Double_t centerX = hRefined->GetXaxis()->GetBinCenter(binX);
792 Double_t centerY = hRefined->GetYaxis()->GetBinCenter(binY);
795 linExtrapolation->ClearPoints();
797 // For interpolation: Just take the corresponding bin from the old histo.
798 // For extrapolation: take the last available bin from the old histo.
799 // If the boundaries are to be skipped, also skip the corresponding bins
800 Int_t oldBinX = h->GetXaxis()->FindBin(centerX);
803 if (oldBinX > nBinsX)
806 Int_t oldBinY = h->GetYaxis()->FindBin(centerY);
809 if (oldBinY > nBinsY)
812 // Neighbours left column
815 AddPointToHyperplane(h, linExtrapolation, oldBinX - 1, oldBinY - 1);
818 AddPointToHyperplane(h, linExtrapolation, oldBinX - 1, oldBinY);
820 if (oldBinY < nBinsY) {
821 AddPointToHyperplane(h, linExtrapolation, oldBinX - 1, oldBinY + 1);
825 // Neighbours (and point itself) same column
827 AddPointToHyperplane(h, linExtrapolation, oldBinX, oldBinY - 1);
830 AddPointToHyperplane(h, linExtrapolation, oldBinX, oldBinY);
832 if (oldBinY < nBinsY) {
833 AddPointToHyperplane(h, linExtrapolation, oldBinX, oldBinY + 1);
836 // Neighbours right column
837 if (oldBinX < nBinsX) {
839 AddPointToHyperplane(h, linExtrapolation, oldBinX + 1, oldBinY - 1);
842 AddPointToHyperplane(h, linExtrapolation, oldBinX + 1, oldBinY);
844 if (oldBinY < nBinsY) {
845 AddPointToHyperplane(h, linExtrapolation, oldBinX + 1, oldBinY + 1);
851 if (linExtrapolation->GetNpoints() <= 0)
854 if (linExtrapolation->Eval() != 0)// EvalRobust -> Takes much, much, [...], much more time (~hours instead of seconds)
857 // Fill the bin of the refined histogram with the extrapolated value
858 Double_t interpolatedValue = linExtrapolation->GetParameter(0) + linExtrapolation->GetParameter(1) * centerX
859 + linExtrapolation->GetParameter(2) * centerY;
861 Double_t interpolatedValue = h->Interpolate(centerX, centerY) ;
862 hRefined->SetBinContent(binX, binY, interpolatedValue);
867 // Problem: Interpolation does not work before/beyond center of first/last bin (as the name suggests).
868 // Therefore, for each row in dEdx: Take last bin from old map and interpolate values from center and edge.
869 // Assume line through these points and extropolate to last bin of refined map
870 const Double_t firstOldXbinUpEdge = h->GetXaxis()->GetBinUpEdge(1);
871 const Double_t firstOldXbinCenter = h->GetXaxis()->GetBinCenter(1);
873 const Double_t oldXbinHalfWidth = firstOldXbinUpEdge - firstOldXbinCenter;
875 const Double_t lastOldXbinLowEdge = h->GetXaxis()->GetBinLowEdge(h->GetNbinsX());
876 const Double_t lastOldXbinCenter = h->GetXaxis()->GetBinCenter(h->GetNbinsX());
878 for (Int_t binY = 1; binY <= nBinsYrefined; binY++) {
879 Double_t centerY = hRefined->GetYaxis()->GetBinCenter(binY);
881 const Double_t interpolatedCenterFirstXbin = h->Interpolate(firstOldXbinCenter, centerY);
882 const Double_t interpolatedUpEdgeFirstXbin = h->Interpolate(firstOldXbinUpEdge, centerY);
884 const Double_t extrapolationSlopeFirstXbin = (interpolatedUpEdgeFirstXbin - interpolatedCenterFirstXbin) / oldXbinHalfWidth;
885 const Double_t extrapolationOffsetFirstXbin = interpolatedCenterFirstXbin;
888 const Double_t interpolatedCenterLastXbin = h->Interpolate(lastOldXbinCenter, centerY);
889 const Double_t interpolatedLowEdgeLastXbin = h->Interpolate(lastOldXbinLowEdge, centerY);
891 const Double_t extrapolationSlopeLastXbin = (interpolatedCenterLastXbin - interpolatedLowEdgeLastXbin) / oldXbinHalfWidth;
892 const Double_t extrapolationOffsetLastXbin = interpolatedCenterLastXbin;
894 for (Int_t binX = 1; binX <= nBinsXrefined; binX++) {
895 Double_t centerX = hRefined->GetXaxis()->GetBinCenter(binX);
897 if (centerX < firstOldXbinCenter) {
898 Double_t extrapolatedValue = extrapolationOffsetFirstXbin + (centerX - firstOldXbinCenter) * extrapolationSlopeFirstXbin;
899 hRefined->SetBinContent(binX, binY, extrapolatedValue);
901 else if (centerX <= lastOldXbinCenter) {
905 Double_t extrapolatedValue = extrapolationOffsetLastXbin + (centerX - lastOldXbinCenter) * extrapolationSlopeLastXbin;
906 hRefined->SetBinContent(binX, binY, extrapolatedValue);
911 delete linExtrapolation;
916 //______________________________________________________________________________
917 void AliPIDResponse::SetTPCEtaMaps(Double_t refineFactorMapX, Double_t refineFactorMapY,
918 Double_t refineFactorSigmaMapX, Double_t refineFactorSigmaMapY)
921 // Load the TPC eta correction maps from the OADB
924 if (fUseTPCEtaCorrection == kFALSE) {
925 // Disable eta correction via setting no maps
926 if (!fTPCResponse.SetEtaCorrMap(0x0))
927 AliInfo("Request to disable TPC eta correction -> Eta correction has been disabled");
929 AliError("Request to disable TPC eta correction -> Some error occured when unloading the correction maps");
931 if (!fTPCResponse.SetSigmaParams(0x0, 0))
932 AliInfo("Request to disable TPC eta correction -> Using old parametrisation for sigma");
934 AliError("Request to disable TPC eta correction -> Some error occured when unloading the sigma maps");
939 TString dataType = "DATA";
940 TString period = fLHCperiod.IsNull() ? "No period information" : fLHCperiod;
943 if (!(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) {
949 if (!(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC)) && fMCperiodTPC.IsNull()) {
950 AliFatal("MC detected, but no MC period set -> Not changing eta maps!");
955 Int_t recopass = fRecoPass;
956 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC) )
957 recopass = fRecoPassUser;
959 TString defaultObj = Form("Default_%s_pass%d", dataType.Data(), recopass);
961 AliInfo(Form("Current period and reco pass: %s.pass%d", period.Data(), recopass));
963 // Invalidate old maps
964 fTPCResponse.SetEtaCorrMap(0x0);
965 fTPCResponse.SetSigmaParams(0x0, 0);
967 // Load the eta correction maps
968 AliOADBContainer etaMapsCont(Form("TPCetaMaps_%s_pass%d", dataType.Data(), recopass));
970 Int_t statusCont = etaMapsCont.InitFromFile(Form("%s/COMMON/PID/data/TPCetaMaps.root", fOADBPath.Data()),
971 Form("TPCetaMaps_%s_pass%d", dataType.Data(), recopass));
973 AliError("Failed initializing TPC eta correction maps from OADB -> Disabled eta correction");
974 fUseTPCEtaCorrection = kFALSE;
977 AliInfo(Form("Loading TPC eta correction map from %s/COMMON/PID/data/TPCetaMaps.root", fOADBPath.Data()));
981 if (fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) {
982 TString searchMap = Form("TPCetaMaps_%s_%s_pass%d", dataType.Data(), period.Data(), recopass);
983 etaMap = dynamic_cast<TH2D *>(etaMapsCont.GetDefaultObject(searchMap.Data()));
985 // Try default object
986 etaMap = dynamic_cast<TH2D *>(etaMapsCont.GetDefaultObject(defaultObj.Data()));
990 etaMap = dynamic_cast<TH2D *>(etaMapsCont.GetObject(fRun, defaultObj.Data()));
995 AliError(Form("TPC eta correction map not found for run %d and also no default map found -> Disabled eta correction!!!", fRun));
996 fUseTPCEtaCorrection = kFALSE;
999 TH2D* etaMapRefined = RefineHistoViaLinearInterpolation(etaMap, refineFactorMapX, refineFactorMapY);
1001 if (etaMapRefined) {
1002 if (!fTPCResponse.SetEtaCorrMap(etaMapRefined)) {
1003 AliError(Form("Failed to set TPC eta correction map for run %d -> Disabled eta correction!!!", fRun));
1004 fTPCResponse.SetEtaCorrMap(0x0);
1005 fUseTPCEtaCorrection = kFALSE;
1008 AliInfo(Form("Loaded TPC eta correction map (refine factors %.2f/%.2f) from %s/COMMON/PID/data/TPCetaMaps.root: %s (MD5(map) = %s)",
1009 refineFactorMapX, refineFactorMapY, fOADBPath.Data(), fTPCResponse.GetEtaCorrMap()->GetTitle(),
1010 GetChecksum(fTPCResponse.GetEtaCorrMap()).Data()));
1013 delete etaMapRefined;
1016 AliError(Form("Failed to set TPC eta correction map for run %d (map was loaded, but couldn't be refined) -> Disabled eta correction!!!", fRun));
1017 fUseTPCEtaCorrection = kFALSE;
1022 // If there was some problem loading the eta maps, it makes no sense to load the sigma maps (that require eta corrected data)
1023 if (fUseTPCEtaCorrection == kFALSE) {
1024 AliError("Failed to load TPC eta correction map required by sigma maps -> Using old parametrisation for sigma");
1028 // Load the sigma parametrisation (1/dEdx vs tanTheta_local (~eta))
1029 AliOADBContainer etaSigmaMapsCont(Form("TPCetaSigmaMaps_%s_pass%d", dataType.Data(), recopass));
1031 statusCont = etaSigmaMapsCont.InitFromFile(Form("%s/COMMON/PID/data/TPCetaMaps.root", fOADBPath.Data()),
1032 Form("TPCetaSigmaMaps_%s_pass%d", dataType.Data(), recopass));
1034 AliError("Failed initializing TPC eta sigma maps from OADB -> Using old sigma parametrisation");
1037 AliInfo(Form("Loading TPC eta sigma map from %s/COMMON/PID/data/TPCetaMaps.root", fOADBPath.Data()));
1039 TObjArray* etaSigmaPars = 0x0;
1041 if (fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) {
1042 TString searchMap = Form("TPCetaSigmaMaps_%s_%s_pass%d", dataType.Data(), period.Data(), recopass);
1043 etaSigmaPars = dynamic_cast<TObjArray *>(etaSigmaMapsCont.GetDefaultObject(searchMap.Data()));
1044 if (!etaSigmaPars) {
1045 // Try default object
1046 etaSigmaPars = dynamic_cast<TObjArray *>(etaSigmaMapsCont.GetDefaultObject(defaultObj.Data()));
1050 etaSigmaPars = dynamic_cast<TObjArray *>(etaSigmaMapsCont.GetObject(fRun, defaultObj.Data()));
1053 if (!etaSigmaPars) {
1054 AliError(Form("TPC eta sigma parametrisation not found for run %d -> Using old sigma parametrisation!!!", fRun));
1057 TH2D* etaSigmaPar1Map = dynamic_cast<TH2D *>(etaSigmaPars->FindObject("sigmaPar1Map"));
1058 TNamed* sigmaPar0Info = dynamic_cast<TNamed *>(etaSigmaPars->FindObject("sigmaPar0"));
1059 Double_t sigmaPar0 = 0.0;
1061 if (sigmaPar0Info) {
1062 TString sigmaPar0String = sigmaPar0Info->GetTitle();
1063 sigmaPar0 = sigmaPar0String.Atof();
1066 // Something is weired because the object for parameter 0 could not be loaded -> New sigma parametrisation can not be used!
1067 etaSigmaPar1Map = 0x0;
1070 TH2D* etaSigmaPar1MapRefined = RefineHistoViaLinearInterpolation(etaSigmaPar1Map, refineFactorSigmaMapX, refineFactorSigmaMapY);
1073 if (etaSigmaPar1MapRefined) {
1074 if (!fTPCResponse.SetSigmaParams(etaSigmaPar1MapRefined, sigmaPar0)) {
1075 AliError(Form("Failed to set TPC eta sigma map for run %d -> Using old sigma parametrisation!!!", fRun));
1076 fTPCResponse.SetSigmaParams(0x0, 0);
1079 AliInfo(Form("Loaded TPC sigma correction map (refine factors %.2f/%.2f) from %s/COMMON/PID/data/TPCetaMaps.root: %s (MD5(map) = %s, sigmaPar0 = %f)",
1080 refineFactorSigmaMapX, refineFactorSigmaMapY, fOADBPath.Data(), fTPCResponse.GetSigmaPar1Map()->GetTitle(),
1081 GetChecksum(fTPCResponse.GetSigmaPar1Map()).Data(), sigmaPar0));
1084 delete etaSigmaPar1MapRefined;
1087 AliError(Form("Failed to set TPC eta sigma map for run %d (map was loaded, but couldn't be refined) -> Using old sigma parametrisation!!!",
1094 //______________________________________________________________________________
1095 void AliPIDResponse::SetTPCPidResponseMaster()
1098 // Load the TPC pid response functions from the OADB
1099 // Load the TPC voltage maps from OADB
1101 //don't load twice for the moment
1102 if (fArrPidResponseMaster) return;
1105 //reset the PID response functions
1106 delete fArrPidResponseMaster;
1107 fArrPidResponseMaster=NULL;
1109 TString fileName(Form("%s/COMMON/PID/data/TPCPIDResponse.root", fOADBPath.Data()));
1111 if (!fCustomTPCpidResponse.IsNull()) fileName=fCustomTPCpidResponse;
1113 TString fileNamePIDresponse(Form("%s/COMMON/PID/data/TPCPIDResponse.root", fOADBPath.Data()));
1114 f=TFile::Open(fileNamePIDresponse.Data());
1115 if (f && f->IsOpen() && !f->IsZombie()){
1116 fArrPidResponseMaster=dynamic_cast<TObjArray*>(f->Get("TPCPIDResponse"));
1120 TString fileNameVoltageMaps(Form("%s/COMMON/PID/data/TPCvoltageSettings.root", fOADBPath.Data()));
1121 f=TFile::Open(fileNameVoltageMaps.Data());
1122 if (f && f->IsOpen() && !f->IsZombie()){
1123 fOADBvoltageMaps=dynamic_cast<AliOADBContainer*>(f->Get("TPCvoltageSettings"));
1127 if (!fArrPidResponseMaster){
1128 AliFatal(Form("Could not retrieve the TPC pid response from: %s",fileNamePIDresponse.Data()));
1131 fArrPidResponseMaster->SetOwner();
1133 if (!fOADBvoltageMaps)
1135 AliFatal(Form("Could not retrieve the TPC voltage maps from: %s",fileNameVoltageMaps.Data()));
1137 fArrPidResponseMaster->SetOwner();
1140 //______________________________________________________________________________
1141 void AliPIDResponse::SetTPCParametrisation()
1144 // Change BB parametrisation for current run
1150 fTPCResponse.ResetSplines();
1152 if (fLHCperiod.IsNull()) {
1153 AliError("No period set, not changing parametrisation");
1158 // Set default parametrisations for data and MC
1162 TString datatype="DATA";
1163 //in case of mc fRecoPass is per default 1
1165 if(!(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) datatype="MC";
1170 TString period=fLHCperiod;
1171 if (fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) period=fMCperiodTPC;
1173 Int_t recopass = fRecoPass;
1174 if(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC)) recopass = fRecoPassUser;
1176 AliInfo(Form("Searching splines for: %s %s PASS%d %s",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
1177 Bool_t found=kFALSE;
1179 //set the new PID splines
1181 if (fArrPidResponseMaster){
1182 //for MC don't use period information
1183 //if (fIsMC) period="[A-Z0-9]*";
1184 //for MC use MC period information
1185 //pattern for the default entry (valid for all particles)
1186 TPRegexp reg(Form("TSPLINE3_%s_([A-Z]*)_%s_PASS%d_%s_MEAN(_*)([A-Z1-9]*)",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
1188 //find particle id and gain scenario
1189 for (Int_t igainScenario=0; igainScenario<AliTPCPIDResponse::fgkNumberOfGainScenarios; igainScenario++)
1191 TObject *grAll=NULL;
1192 TString gainScenario = AliTPCPIDResponse::GainScenarioName(igainScenario);
1193 gainScenario.ToUpper();
1194 //loop over entries and filter them
1195 for (Int_t iresp=0; iresp<fArrPidResponseMaster->GetEntriesFast();++iresp)
1197 TObject *responseFunction=fArrPidResponseMaster->At(iresp);
1198 if (responseFunction==NULL) continue;
1199 TString responseName=responseFunction->GetName();
1201 if (!reg.MatchB(responseName)) continue;
1203 TObjArray *arr=reg.MatchS(responseName); if (!arr) continue;
1205 tmp=arr->At(1); if (!tmp) continue;
1206 TString particleName=tmp->GetName();
1207 tmp=arr->At(3); if (!tmp) continue;
1208 TString gainScenarioName=tmp->GetName();
1210 if (particleName.IsNull()) continue;
1211 if (!grAll && particleName=="ALL" && gainScenarioName==gainScenario) grAll=responseFunction;
1214 for (Int_t ispec=0; ispec<(AliTPCPIDResponse::fgkNumberOfParticleSpecies); ++ispec)
1216 TString particle=AliPID::ParticleName(ispec);
1218 //std::cout<<responseName<<" "<<particle<<" "<<particleName<<" "<<gainScenario<<" "<<gainScenarioName<<std::endl;
1219 if ( particle == particleName && gainScenario == gainScenarioName )
1221 fTPCResponse.SetResponseFunction( responseFunction,
1222 (AliPID::EParticleType)ispec,
1223 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1224 fTPCResponse.SetUseDatabase(kTRUE);
1225 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,responseFunction->GetName(),
1226 GetChecksum((TSpline3*)responseFunction).Data()));
1234 // Retrieve responsefunction for pions - will (if available) be used for muons if there are no dedicated muon splines.
1235 // For light nuclei, try to set the proton spline, if no dedicated splines are available.
1236 // In both cases: Use default splines, if no dedicated splines and no pion/proton splines are available.
1237 TObject* responseFunctionPion = fTPCResponse.GetResponseFunction( (AliPID::EParticleType)AliPID::kPion,
1238 (AliTPCPIDResponse::ETPCgainScenario)igainScenario);
1239 TObject* responseFunctionProton = fTPCResponse.GetResponseFunction( (AliPID::EParticleType)AliPID::kProton,
1240 (AliTPCPIDResponse::ETPCgainScenario)igainScenario);
1242 for (Int_t ispec=0; ispec<(AliTPCPIDResponse::fgkNumberOfParticleSpecies); ++ispec)
1244 if (!fTPCResponse.GetResponseFunction( (AliPID::EParticleType)ispec,
1245 (AliTPCPIDResponse::ETPCgainScenario)igainScenario))
1247 if (ispec == AliPID::kMuon) { // Muons
1248 if (responseFunctionPion) {
1249 fTPCResponse.SetResponseFunction( responseFunctionPion,
1250 (AliPID::EParticleType)ispec,
1251 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1252 fTPCResponse.SetUseDatabase(kTRUE);
1253 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,responseFunctionPion->GetName(),
1254 GetChecksum((TSpline3*)responseFunctionPion).Data()));
1258 fTPCResponse.SetResponseFunction( grAll,
1259 (AliPID::EParticleType)ispec,
1260 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1261 fTPCResponse.SetUseDatabase(kTRUE);
1262 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,grAll->GetName(),
1263 GetChecksum((TSpline3*)grAll).Data()));
1267 // AliError(Form("No splines found for muons (also no pion splines and no default splines) for gain scenario %d!", igainScenario));
1269 else if (ispec >= AliPID::kSPECIES) { // Light nuclei
1270 if (responseFunctionProton) {
1271 fTPCResponse.SetResponseFunction( responseFunctionProton,
1272 (AliPID::EParticleType)ispec,
1273 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1274 fTPCResponse.SetUseDatabase(kTRUE);
1275 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,responseFunctionProton->GetName(),
1276 GetChecksum((TSpline3*)responseFunctionProton).Data()));
1280 fTPCResponse.SetResponseFunction( grAll,
1281 (AliPID::EParticleType)ispec,
1282 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1283 fTPCResponse.SetUseDatabase(kTRUE);
1284 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,grAll->GetName(),
1285 GetChecksum((TSpline3*)grAll).Data()));
1289 // AliError(Form("No splines found for species %d (also no proton splines and no default splines) for gain scenario %d!",
1290 // ispec, igainScenario));
1296 else AliInfo("no fArrPidResponseMaster");
1299 AliError(Form("No splines found for: %s %s PASS%d %s",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
1304 // Setup multiplicity correction
1306 if (fUseTPCMultiplicityCorrection && !(fBeamType.CompareTo("PP") == 0)) {
1307 AliInfo("Multiplicity correction enabled!");
1309 //TODO After testing, load parameters from outside
1310 /*TODO now correction for MC
1311 if (period.Contains("LHC11A10")) {//LHC11A10A
1312 AliInfo("Using multiplicity correction parameters for 11a10!");
1313 fTPCResponse.SetParameterMultiplicityCorrection(0, 6.90133e-06);
1314 fTPCResponse.SetParameterMultiplicityCorrection(1, -1.22123e-03);
1315 fTPCResponse.SetParameterMultiplicityCorrection(2, 1.80220e-02);
1316 fTPCResponse.SetParameterMultiplicityCorrection(3, 0.1);
1317 fTPCResponse.SetParameterMultiplicityCorrection(4, 6.45306e-03);
1319 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, -2.85505e-07);
1320 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, -1.31911e-06);
1321 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1323 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, -4.29665e-05);
1324 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, 1.37023e-02);
1325 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, -6.36337e-01);
1326 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 1.13479e-02);
1328 else*/ if (period.Contains("LHC13B") || period.Contains("LHC13C") || period.Contains("LHC13D")) {// 2013 pPb data taking at low luminosity
1329 AliInfo("Using multiplicity correction parameters for 13b.pass2!");
1331 fTPCResponse.SetParameterMultiplicityCorrection(0, -5.906e-06);
1332 fTPCResponse.SetParameterMultiplicityCorrection(1, -5.064e-04);
1333 fTPCResponse.SetParameterMultiplicityCorrection(2, -3.521e-02);
1334 fTPCResponse.SetParameterMultiplicityCorrection(3, 2.469e-02);
1335 fTPCResponse.SetParameterMultiplicityCorrection(4, 0);
1337 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, -5.32e-06);
1338 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, 1.177e-05);
1339 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1341 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, 0.);
1342 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, 0.);
1343 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, 0.);
1344 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 0.);
1346 /* Not too bad, but far from perfect in the details
1347 fTPCResponse.SetParameterMultiplicityCorrection(0, -6.27187e-06);
1348 fTPCResponse.SetParameterMultiplicityCorrection(1, -4.60649e-04);
1349 fTPCResponse.SetParameterMultiplicityCorrection(2, -4.26450e-02);
1350 fTPCResponse.SetParameterMultiplicityCorrection(3, 2.40590e-02);
1351 fTPCResponse.SetParameterMultiplicityCorrection(4, 0);
1353 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, -5.338e-06);
1354 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, 1.220e-05);
1355 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1357 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, 7.89237e-05);
1358 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, -1.30662e-02);
1359 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, 8.91548e-01);
1360 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 1.47931e-02);
1363 else if (period.Contains("LHC10H") && recopass == 2) {
1364 AliInfo("Using multiplicity correction parameters for 10h.pass2!");
1365 fTPCResponse.SetParameterMultiplicityCorrection(0, 3.21636e-07);
1366 fTPCResponse.SetParameterMultiplicityCorrection(1, -6.65876e-04);
1367 fTPCResponse.SetParameterMultiplicityCorrection(2, 1.28786e-03);
1368 fTPCResponse.SetParameterMultiplicityCorrection(3, 1.47677e-02);
1369 fTPCResponse.SetParameterMultiplicityCorrection(4, 0);
1371 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, 7.23591e-08);
1372 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, 2.7469e-06);
1373 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1375 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, -1.22590e-05);
1376 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, 6.88888e-03);
1377 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, -3.20788e-01);
1378 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 1.07345e-02);
1381 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));
1382 fUseTPCMultiplicityCorrection = kFALSE;
1383 fTPCResponse.ResetMultiplicityCorrectionFunctions();
1387 // Just set parameters such that overall correction factor is 1, i.e. no correction.
1388 // This is just a reasonable choice for the parameters for safety reasons. Disabling
1389 // the multiplicity correction will anyhow skip the calculation of the corresponding
1390 // correction factor inside THIS class. Nevertheless, experts can access the TPCPIDResponse
1391 // directly and use it for calculations - which should still give valid results, even if
1392 // the multiplicity correction is explicitely enabled in such expert calls.
1394 AliInfo(Form("Multiplicity correction %sdisabled (%s)!", fUseTPCMultiplicityCorrection ? "automatically " : "",
1395 fUseTPCMultiplicityCorrection ? "pp collisions" : "requested by user"));
1397 fUseTPCMultiplicityCorrection = kFALSE;
1398 fTPCResponse.ResetMultiplicityCorrectionFunctions();
1403 for (Int_t i = 0; i <= 4 + 1; i++) {
1404 printf("parMultCorr: %d, %e\n", i, fTPCResponse.GetMultiplicityCorrectionFunction()->GetParameter(i));
1406 for (Int_t j = 0; j <= 2 + 1; j++) {
1407 printf("parMultCorrTanTheta: %d, %e\n", j, fTPCResponse.GetMultiplicityCorrectionFunctionTanTheta()->GetParameter(j));
1409 for (Int_t j = 0; j <= 3 + 1; j++) {
1410 printf("parMultSigmaCorr: %d, %e\n", j, fTPCResponse.GetMultiplicitySigmaCorrectionFunction()->GetParameter(j));
1417 // Setup old resolution parametrisation
1421 fTPCResponse.SetSigma(3.79301e-03, 2.21280e+04);
1423 if (fRun>=122195){ //LHC10d
1424 fTPCResponse.SetSigma(2.30176e-02, 5.60422e+02);
1427 if (fRun>=170719){ // LHC12a
1428 fTPCResponse.SetSigma(2.95714e-03, 1.01953e+05);
1431 if (fRun>=177312){ // LHC12b
1432 fTPCResponse.SetSigma(3.74633e-03, 7.11829e+04 );
1435 if (fRun>=186346){ // LHC12e
1436 fTPCResponse.SetSigma(8.62022e-04, 9.08156e+05);
1439 if (fArrPidResponseMaster)
1440 fResolutionCorrection=(TF1*)fArrPidResponseMaster->FindObject(Form("TF1_%s_ALL_%s_PASS%d_%s_SIGMA",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
1442 if (fResolutionCorrection) AliInfo(Form("Setting multiplicity correction function: %s (MD5(corr function) = %s)",
1443 fResolutionCorrection->GetName(), GetChecksum(fResolutionCorrection).Data()));
1445 //read in the voltage map
1446 TVectorF* gsm = 0x0;
1447 if (fOADBvoltageMaps) gsm=dynamic_cast<TVectorF*>(fOADBvoltageMaps->GetObject(fRun));
1450 fTPCResponse.SetVoltageMap(*gsm);
1452 AliInfo(Form("Reading the voltage map for run %d\n",fRun));
1453 vals="IROC A: "; for (Int_t i=0; i<18; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1454 AliInfo(vals.Data());
1455 vals="IROC C: "; for (Int_t i=18; i<36; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1456 AliInfo(vals.Data());
1457 vals="OROC A: "; for (Int_t i=36; i<54; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1458 AliInfo(vals.Data());
1459 vals="OROC C: "; for (Int_t i=54; i<72; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1460 AliInfo(vals.Data());
1462 else AliInfo("no voltage map, ideal default assumed");
1465 //______________________________________________________________________________
1466 void AliPIDResponse::SetTRDPidResponseMaster()
1469 // Load the TRD pid params and references from the OADB
1471 if(fTRDPIDResponseObject) return;
1472 AliOADBContainer contParams("contParams");
1474 Int_t statusResponse = contParams.InitFromFile(Form("%s/COMMON/PID/data/TRDPIDResponse.root", fOADBPath.Data()), "AliTRDPIDResponseObject");
1476 AliError("Failed initializing PID Response Object from OADB");
1478 AliInfo(Form("Loading TRD Response from %s/COMMON/PID/data/TRDPIDResponse.root", fOADBPath.Data()));
1479 fTRDPIDResponseObject = dynamic_cast<AliTRDPIDResponseObject *>(contParams.GetObject(fRun));
1480 if(!fTRDPIDResponseObject){
1481 AliError(Form("TRD Response not found in run %d", fRun));
1486 //______________________________________________________________________________
1487 void AliPIDResponse::InitializeTRDResponse(){
1489 // Set PID Params and references to the TRD PID response
1491 fTRDResponse.SetPIDResponseObject(fTRDPIDResponseObject);
1494 //______________________________________________________________________________
1495 void AliPIDResponse::SetTRDSlices(UInt_t TRDslicesForPID[2],AliTRDPIDResponse::ETRDPIDMethod method) const{
1497 if(fLHCperiod.Contains("LHC10D") || fLHCperiod.Contains("LHC10E")){
1498 // backward compatibility for setting with 8 slices
1499 TRDslicesForPID[0] = 0;
1500 TRDslicesForPID[1] = 7;
1503 if(method==AliTRDPIDResponse::kLQ1D){
1504 TRDslicesForPID[0] = 0; // first Slice contains normalized dEdx
1505 TRDslicesForPID[1] = 0;
1507 if(method==AliTRDPIDResponse::kLQ2D){
1508 TRDslicesForPID[0] = 1;
1509 TRDslicesForPID[1] = 7;
1512 AliDebug(1,Form("Slice Range set to %d - %d",TRDslicesForPID[0],TRDslicesForPID[1]));
1515 //______________________________________________________________________________
1516 void AliPIDResponse::SetTOFPidResponseMaster()
1519 // Load the TOF pid params from the OADB
1522 if (fTOFPIDParams) delete fTOFPIDParams;
1525 TFile *oadbf = new TFile(Form("%s/COMMON/PID/data/TOFPIDParams.root",fOADBPath.Data()));
1526 if (oadbf && oadbf->IsOpen()) {
1527 AliInfo(Form("Loading TOF Params from %s/COMMON/PID/data/TOFPIDParams.root", fOADBPath.Data()));
1528 AliOADBContainer *oadbc = (AliOADBContainer *)oadbf->Get("TOFoadb");
1529 if (oadbc) fTOFPIDParams = dynamic_cast<AliTOFPIDParams *>(oadbc->GetObject(fRun,"TOFparams"));
1535 if (!fTOFPIDParams) AliFatal("TOFPIDParams could not be retrieved");
1538 //______________________________________________________________________________
1539 void AliPIDResponse::InitializeTOFResponse(){
1541 // Set PID Params to the TOF PID response
1544 AliInfo("TOF PID Params loaded from OADB");
1545 AliInfo(Form(" TOF resolution %5.2f [ps]",fTOFPIDParams->GetTOFresolution()));
1546 AliInfo(Form(" StartTime method %d",fTOFPIDParams->GetStartTimeMethod()));
1547 AliInfo(Form(" TOF res. mom. params: %5.2f %5.2f %5.2f %5.2f",
1548 fTOFPIDParams->GetSigParams(0),fTOFPIDParams->GetSigParams(1),fTOFPIDParams->GetSigParams(2),fTOFPIDParams->GetSigParams(3)));
1549 AliInfo(Form(" Fraction of tracks within gaussian behaviour: %6.4f",fTOFPIDParams->GetTOFtail()));
1550 AliInfo(Form(" MC: Fraction of tracks (percentage) to cut to fit matching in data: %6.2f%%",fTOFPIDParams->GetTOFmatchingLossMC()));
1551 AliInfo(Form(" MC: Fraction of random hits (percentage) to add to fit mismatch in data: %6.2f%%",fTOFPIDParams->GetTOFadditionalMismForMC()));
1552 AliInfo(Form(" Start Time Offset %6.2f ps",fTOFPIDParams->GetTOFtimeOffset()));
1554 for (Int_t i=0;i<4;i++) {
1555 fTOFResponse.SetTrackParameter(i,fTOFPIDParams->GetSigParams(i));
1557 fTOFResponse.SetTimeResolution(fTOFPIDParams->GetTOFresolution());
1559 AliInfo("TZERO resolution loaded from ESDrun/AODheader");
1560 Float_t t0Spread[4];
1561 for (Int_t i=0;i<4;i++) t0Spread[i]=fCurrentEvent->GetT0spread(i);
1562 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]));
1563 Float_t a = t0Spread[1]*t0Spread[1]-t0Spread[0]*t0Spread[0]+t0Spread[3]*t0Spread[3];
1564 Float_t c = t0Spread[2]*t0Spread[2]-t0Spread[0]*t0Spread[0]+t0Spread[3]*t0Spread[3];
1565 if ( (t0Spread[0] > 50. && t0Spread[0] < 400.) && (a > 0.) && (c>0.)) {
1566 fResT0AC=t0Spread[3];
1567 fResT0A=TMath::Sqrt(a);
1568 fResT0C=TMath::Sqrt(c);
1570 AliInfo(" TZERO spreads not present or inconsistent, loading default");
1575 AliInfo(Form(" TZERO resolution set to: T0A: %f [ps] T0C: %f [ps] T0AC %f [ps]",fResT0A,fResT0C,fResT0AC));
1579 //______________________________________________________________________________
1580 void AliPIDResponse::SetHMPIDPidResponseMaster()
1583 // Load the HMPID pid params from the OADB
1586 if (fHMPIDPIDParams) delete fHMPIDPIDParams;
1587 fHMPIDPIDParams=NULL;
1589 TFile *oadbf = new TFile(Form("%s/COMMON/PID/data/HMPIDPIDParams.root",fOADBPath.Data()));
1590 if (oadbf && oadbf->IsOpen()) {
1591 AliInfo(Form("Loading HMPID Params from %s/COMMON/PID/data/HMPIDPIDParams.root", fOADBPath.Data()));
1592 AliOADBContainer *oadbc = (AliOADBContainer *)oadbf->Get("HMPoadb");
1593 if (oadbc) fHMPIDPIDParams = dynamic_cast<AliHMPIDPIDParams *>(oadbc->GetObject(fRun,"HMPparams"));
1599 if (!fHMPIDPIDParams) AliFatal("HMPIDPIDParams could not be retrieved");
1602 //______________________________________________________________________________
1603 void AliPIDResponse::InitializeHMPIDResponse(){
1605 // Set PID Params to the HMPID PID response
1608 fHMPIDResponse.SetRefIndexArray(fHMPIDPIDParams->GetHMPIDrefIndex());
1611 //______________________________________________________________________________
1612 Bool_t AliPIDResponse::IdentifiedAsElectronTRD(const AliVTrack *vtrack, Double_t efficiencyLevel,Double_t centrality,AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const {
1614 // Check whether track is identified as electron under a given electron efficiency hypothesis
1617 Double_t probs[AliPID::kSPECIES];
1618 ComputeTRDProbability(vtrack, AliPID::kSPECIES, probs,PIDmethod);
1620 Int_t ntracklets = vtrack->GetTRDntrackletsPID();
1621 // Take mean of the TRD momenta in the given tracklets
1622 Float_t p = 0, trdmomenta[AliVTrack::kTRDnPlanes];
1624 for(Int_t iPl=0;iPl<AliVTrack::kTRDnPlanes;iPl++){
1625 if(vtrack->GetTRDmomentum(iPl) > 0.){
1626 trdmomenta[nmomenta++] = vtrack->GetTRDmomentum(iPl);
1629 p = TMath::Mean(nmomenta, trdmomenta);
1631 return fTRDResponse.IdentifiedAsElectron(ntracklets, probs, p, efficiencyLevel,centrality,PIDmethod);
1634 //______________________________________________________________________________
1635 void AliPIDResponse::SetEMCALPidResponseMaster()
1638 // Load the EMCAL pid response functions from the OADB
1640 TObjArray* fEMCALPIDParamsRun = NULL;
1641 TObjArray* fEMCALPIDParamsPass = NULL;
1643 if(fEMCALPIDParams) return;
1644 AliOADBContainer contParams("contParams");
1646 Int_t statusPars = contParams.InitFromFile(Form("%s/COMMON/PID/data/EMCALPIDParams.root", fOADBPath.Data()), "AliEMCALPIDParams");
1648 AliError("Failed initializing PID Params from OADB");
1651 AliInfo(Form("Loading EMCAL Params from %s/COMMON/PID/data/EMCALPIDParams.root", fOADBPath.Data()));
1653 fEMCALPIDParamsRun = dynamic_cast<TObjArray *>(contParams.GetObject(fRun));
1654 if(fEMCALPIDParamsRun) fEMCALPIDParamsPass = dynamic_cast<TObjArray *>(fEMCALPIDParamsRun->FindObject(Form("pass%d",fRecoPass)));
1655 if(fEMCALPIDParamsPass) fEMCALPIDParams = dynamic_cast<TObjArray *>(fEMCALPIDParamsPass->FindObject(Form("EMCALPIDParams_Particles")));
1657 if(!fEMCALPIDParams){
1658 AliInfo(Form("EMCAL Params not found in run %d pass %d", fRun, fRecoPass));
1659 AliInfo("Will take the standard LHC11d instead ...");
1661 fEMCALPIDParamsRun = dynamic_cast<TObjArray *>(contParams.GetObject(156477));
1662 if(fEMCALPIDParamsRun) fEMCALPIDParamsPass = dynamic_cast<TObjArray *>(fEMCALPIDParamsRun->FindObject(Form("pass%d",1)));
1663 if(fEMCALPIDParamsPass) fEMCALPIDParams = dynamic_cast<TObjArray *>(fEMCALPIDParamsPass->FindObject(Form("EMCALPIDParams_Particles")));
1665 if(!fEMCALPIDParams){
1666 AliError(Form("DEFAULT EMCAL Params (LHC11d) not found in file %s/COMMON/PID/data/EMCALPIDParams.root", fOADBPath.Data()));
1672 //______________________________________________________________________________
1673 void AliPIDResponse::InitializeEMCALResponse(){
1675 // Set PID Params to the EMCAL PID response
1677 fEMCALResponse.SetPIDParams(fEMCALPIDParams);
1681 //______________________________________________________________________________
1682 void AliPIDResponse::FillTrackDetectorPID(const AliVTrack *track, EDetector detector) const
1685 // create detector PID information and setup the transient pointer in the track
1688 // check if detector number is inside accepted range
1689 if (detector == kNdetectors) return;
1692 AliDetectorPID *detPID=const_cast<AliDetectorPID*>(track->GetDetectorPID());
1694 detPID=new AliDetectorPID;
1695 (const_cast<AliVTrack*>(track))->SetDetectorPID(detPID);
1698 //check if values exist
1699 if (detPID->HasRawProbability(detector) && detPID->HasNumberOfSigmas(detector)) return;
1701 //TODO: which particles to include? See also the loops below...
1702 Double_t values[AliPID::kSPECIESC]={0};
1705 EDetPidStatus status=GetComputePIDProbability(detector,track,AliPID::kSPECIESC,values);
1706 detPID->SetRawProbability(detector, values, (Int_t)AliPID::kSPECIESC, status);
1709 for (Int_t ipart=0; ipart<AliPID::kSPECIESC; ++ipart)
1710 values[ipart]=GetNumberOfSigmas(detector,track,(AliPID::EParticleType)ipart);
1711 // the pid status is the same for probabilities and nSigmas, so it is
1712 // fine to use the one from the probabilities also here
1713 detPID->SetNumberOfSigmas(detector, values, (Int_t)AliPID::kSPECIESC, status);
1717 //______________________________________________________________________________
1718 void AliPIDResponse::FillTrackDetectorPID()
1721 // create detector PID information and setup the transient pointer in the track
1724 if (!fCurrentEvent) return;
1726 for (Int_t itrack=0; itrack<fCurrentEvent->GetNumberOfTracks(); ++itrack){
1727 AliVTrack *track=dynamic_cast<AliVTrack*>(fCurrentEvent->GetTrack(itrack));
1728 if (!track) continue;
1730 for (Int_t idet=0; idet<kNdetectors; ++idet){
1731 FillTrackDetectorPID(track, (EDetector)idet);
1736 //______________________________________________________________________________
1737 void AliPIDResponse::SetTOFResponse(AliVEvent *vevent,EStartTimeType_t option){
1739 // Set TOF response function
1740 // Input option for event_time used
1743 Float_t t0spread = 0.; //vevent->GetEventTimeSpread();
1744 if(t0spread < 10) t0spread = 80;
1746 // T0-FILL and T0-TO offset (because of TOF misallignment
1747 Float_t starttimeoffset = 0;
1748 if(fTOFPIDParams && !(fIsMC)) starttimeoffset=fTOFPIDParams->GetTOFtimeOffset();
1750 fTOFtail = fTOFPIDParams->GetTOFtail();
1751 GetTOFResponse().SetTOFtail(fTOFtail);
1754 // T0 from TOF algorithm
1755 Bool_t flagT0TOF=kFALSE;
1756 Bool_t flagT0T0=kFALSE;
1757 Float_t *startTime = new Float_t[fTOFResponse.GetNmomBins()];
1758 Float_t *startTimeRes = new Float_t[fTOFResponse.GetNmomBins()];
1759 Int_t *startTimeMask = new Int_t[fTOFResponse.GetNmomBins()];
1762 Float_t *estimatedT0event = new Float_t[fTOFResponse.GetNmomBins()];
1763 Float_t *estimatedT0resolution = new Float_t[fTOFResponse.GetNmomBins()];
1764 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1765 estimatedT0event[i]=0.0;
1766 estimatedT0resolution[i]=0.0;
1767 startTimeMask[i] = 0;
1770 Float_t resT0A=fResT0A;
1771 Float_t resT0C=fResT0C;
1772 Float_t resT0AC=fResT0AC;
1773 if(vevent->GetT0TOF()){ // check if T0 detector information is available
1778 AliTOFHeader *tofHeader = (AliTOFHeader*)vevent->GetTOFHeader();
1780 if (tofHeader) { // read global info and T0-TOF
1781 fTOFResponse.SetTimeResolution(tofHeader->GetTOFResolution());
1782 t0spread = tofHeader->GetT0spread(); // read t0 sprad
1783 if(t0spread < 10) t0spread = 80;
1786 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){ // read T0-TOF default value
1787 startTime[i]=tofHeader->GetDefaultEventTimeVal();
1788 startTimeRes[i]=tofHeader->GetDefaultEventTimeRes();
1789 if(startTimeRes[i] < 1.e-5) startTimeRes[i] = t0spread;
1791 if(startTimeRes[i] > t0spread - 10 && TMath::Abs(startTime[i]) < 0.001) startTime[i] = -starttimeoffset; // apply offset for T0-fill
1794 TArrayI *ibin=(TArrayI*)tofHeader->GetNvalues();
1795 TArrayF *t0Bin=(TArrayF*)tofHeader->GetEventTimeValues();
1796 TArrayF *t0ResBin=(TArrayF*)tofHeader->GetEventTimeRes();
1797 for(Int_t j=0;j < tofHeader->GetNbins();j++){ // fill T0-TOF in p-bins
1798 Int_t icurrent = (Int_t)ibin->GetAt(j);
1799 startTime[icurrent]=t0Bin->GetAt(j);
1800 startTimeRes[icurrent]=t0ResBin->GetAt(j);
1801 if(startTimeRes[icurrent] < 1.e-5) startTimeRes[icurrent] = t0spread;
1802 if(startTimeRes[icurrent] > t0spread - 10 && TMath::Abs(startTime[icurrent]) < 0.001) startTime[icurrent] = -starttimeoffset; // apply offset for T0-fill
1806 // for cut of 3 sigma on t0 spread
1807 Float_t t0cut = 3 * t0spread;
1808 if(t0cut < 500) t0cut = 500;
1810 if(option == kFILL_T0){ // T0-FILL is used
1811 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1812 estimatedT0event[i]=0.0-starttimeoffset;
1813 estimatedT0resolution[i]=t0spread;
1815 fTOFResponse.SetT0event(estimatedT0event);
1816 fTOFResponse.SetT0resolution(estimatedT0resolution);
1819 if(option == kTOF_T0){ // T0-TOF is used when available (T0-FILL otherwise) from ESD
1821 fTOFResponse.SetT0event(startTime);
1822 fTOFResponse.SetT0resolution(startTimeRes);
1823 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1824 if(startTimeRes[i]<t0spread) startTimeMask[i]=1;
1825 fTOFResponse.SetT0binMask(i,startTimeMask[i]);
1829 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1830 estimatedT0event[i]=0.0-starttimeoffset;
1831 estimatedT0resolution[i]=t0spread;
1832 fTOFResponse.SetT0binMask(i,startTimeMask[i]);
1834 fTOFResponse.SetT0event(estimatedT0event);
1835 fTOFResponse.SetT0resolution(estimatedT0resolution);
1838 else if(option == kBest_T0){ // T0-T0 or T0-TOF are used when available (T0-FILL otherwise) from ESD
1839 Float_t t0AC=-10000;
1843 t0A= vevent->GetT0TOF()[1] - starttimeoffset;
1844 t0C= vevent->GetT0TOF()[2] - starttimeoffset;
1845 // t0AC= vevent->GetT0TOF()[0];
1846 t0AC= t0A/resT0A/resT0A + t0C/resT0C/resT0C;
1847 resT0AC= TMath::Sqrt(1./resT0A/resT0A + 1./resT0C/resT0C);
1848 t0AC /= resT0AC*resT0AC;
1851 Float_t t0t0Best = 0;
1852 Float_t t0t0BestRes = 9999;
1854 if(TMath::Abs(t0A) < t0cut && TMath::Abs(t0C) < t0cut && TMath::Abs(t0C-t0A) < 500){
1856 t0t0BestRes = resT0AC;
1859 else if(TMath::Abs(t0C) < t0cut){
1861 t0t0BestRes = resT0C;
1864 else if(TMath::Abs(t0A) < t0cut){
1866 t0t0BestRes = resT0A;
1870 if(flagT0TOF){ // if T0-TOF info is available
1871 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1872 if(t0t0BestRes < 999){
1873 if(startTimeRes[i] < t0spread){
1874 Double_t wtot = 1./startTimeRes[i]/startTimeRes[i] + 1./t0t0BestRes/t0t0BestRes;
1875 Double_t t0best = startTime[i]/startTimeRes[i]/startTimeRes[i] + t0t0Best/t0t0BestRes/t0t0BestRes;
1876 estimatedT0event[i]=t0best / wtot;
1877 estimatedT0resolution[i]=1./TMath::Sqrt(wtot);
1878 startTimeMask[i] = t0used+1;
1881 estimatedT0event[i]=t0t0Best;
1882 estimatedT0resolution[i]=t0t0BestRes;
1883 startTimeMask[i] = t0used;
1887 estimatedT0event[i]=startTime[i];
1888 estimatedT0resolution[i]=startTimeRes[i];
1889 if(startTimeRes[i]<t0spread) startTimeMask[i]=1;
1891 fTOFResponse.SetT0binMask(i,startTimeMask[i]);
1893 fTOFResponse.SetT0event(estimatedT0event);
1894 fTOFResponse.SetT0resolution(estimatedT0resolution);
1896 else{ // if no T0-TOF info is available
1897 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1898 fTOFResponse.SetT0binMask(i,t0used);
1899 if(t0t0BestRes < 999){
1900 estimatedT0event[i]=t0t0Best;
1901 estimatedT0resolution[i]=t0t0BestRes;
1904 estimatedT0event[i]=0.0-starttimeoffset;
1905 estimatedT0resolution[i]=t0spread;
1908 fTOFResponse.SetT0event(estimatedT0event);
1909 fTOFResponse.SetT0resolution(estimatedT0resolution);
1913 else if(option == kT0_T0){ // T0-T0 is used when available (T0-FILL otherwise)
1914 Float_t t0AC=-10000;
1918 t0A= vevent->GetT0TOF()[1] - starttimeoffset;
1919 t0C= vevent->GetT0TOF()[2] - starttimeoffset;
1920 // t0AC= vevent->GetT0TOF()[0];
1921 t0AC= t0A/resT0A/resT0A + t0C/resT0C/resT0C;
1922 resT0AC= TMath::Sqrt(1./resT0A/resT0A + 1./resT0C/resT0C);
1923 t0AC /= resT0AC*resT0AC;
1926 if(TMath::Abs(t0A) < t0cut && TMath::Abs(t0C) < t0cut && TMath::Abs(t0C-t0A) < 500){
1927 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1928 estimatedT0event[i]=t0AC;
1929 estimatedT0resolution[i]=resT0AC;
1930 fTOFResponse.SetT0binMask(i,6);
1933 else if(TMath::Abs(t0C) < t0cut){
1934 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1935 estimatedT0event[i]=t0C;
1936 estimatedT0resolution[i]=resT0C;
1937 fTOFResponse.SetT0binMask(i,4);
1940 else if(TMath::Abs(t0A) < t0cut){
1941 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1942 estimatedT0event[i]=t0A;
1943 estimatedT0resolution[i]=resT0A;
1944 fTOFResponse.SetT0binMask(i,2);
1948 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1949 estimatedT0event[i]= 0.0 - starttimeoffset;
1950 estimatedT0resolution[i]=t0spread;
1951 fTOFResponse.SetT0binMask(i,0);
1954 fTOFResponse.SetT0event(estimatedT0event);
1955 fTOFResponse.SetT0resolution(estimatedT0resolution);
1958 delete [] startTime;
1959 delete [] startTimeRes;
1960 delete [] startTimeMask;
1961 delete [] estimatedT0event;
1962 delete [] estimatedT0resolution;
1965 //______________________________________________________________________________
1966 // private non cached versions of the PID calculation
1970 //______________________________________________________________________________
1971 Float_t AliPIDResponse::GetNumberOfSigmas(EDetector detector, const AliVParticle *vtrack, AliPID::EParticleType type) const
1974 // NumberOfSigmas for 'detCode'
1977 const AliVTrack *track=static_cast<const AliVTrack*>(vtrack);
1980 case kITS: return GetNumberOfSigmasITS(track, type); break;
1981 case kTPC: return GetNumberOfSigmasTPC(track, type); break;
1982 case kTOF: return GetNumberOfSigmasTOF(track, type); break;
1983 case kHMPID: return GetNumberOfSigmasHMPID(track, type); break;
1984 case kEMCAL: return GetNumberOfSigmasEMCAL(track, type); break;
1985 default: return -999.;
1991 //______________________________________________________________________________
1992 Float_t AliPIDResponse::GetNumberOfSigmasITS(const AliVParticle *vtrack, AliPID::EParticleType type) const
1995 // Calculate the number of sigmas in the ITS
1998 AliVTrack *track=(AliVTrack*)vtrack;
2000 const EDetPidStatus pidStatus=GetITSPIDStatus(track);
2001 if (pidStatus!=kDetPidOk) return -999.;
2003 return fITSResponse.GetNumberOfSigmas(track,type);
2006 //______________________________________________________________________________
2007 Float_t AliPIDResponse::GetNumberOfSigmasTPC(const AliVParticle *vtrack, AliPID::EParticleType type) const
2010 // Calculate the number of sigmas in the TPC
2013 AliVTrack *track=(AliVTrack*)vtrack;
2015 const EDetPidStatus pidStatus=GetTPCPIDStatus(track);
2016 if (pidStatus!=kDetPidOk) return -999.;
2018 // the following call is needed in order to fill the transient data member
2019 // fTPCsignalTuned which is used in the TPCPIDResponse to judge
2020 // if using tuned on data
2021 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))
2022 this->GetTPCsignalTunedOnData(track);
2024 return fTPCResponse.GetNumberOfSigmas(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
2027 //______________________________________________________________________________
2028 Float_t AliPIDResponse::GetNumberOfSigmasTOF(const AliVParticle *vtrack, AliPID::EParticleType type) const
2031 // Calculate the number of sigmas in the TOF
2034 AliVTrack *track=(AliVTrack*)vtrack;
2036 const EDetPidStatus pidStatus=GetTOFPIDStatus(track);
2037 if (pidStatus!=kDetPidOk) return -999.;
2039 return GetNumberOfSigmasTOFold(vtrack, type);
2041 //______________________________________________________________________________
2043 Float_t AliPIDResponse::GetNumberOfSigmasHMPID(const AliVParticle *vtrack, AliPID::EParticleType type) const
2046 // Calculate the number of sigmas in the HMPID
2048 AliVTrack *track=(AliVTrack*)vtrack;
2050 const EDetPidStatus pidStatus=GetHMPIDPIDStatus(track);
2051 if (pidStatus!=kDetPidOk) return -999.;
2053 return fHMPIDResponse.GetNumberOfSigmas(track, type);
2056 //______________________________________________________________________________
2057 Float_t AliPIDResponse::GetNumberOfSigmasEMCAL(const AliVParticle *vtrack, AliPID::EParticleType type) const
2060 // Calculate the number of sigmas in the EMCAL
2063 AliVTrack *track=(AliVTrack*)vtrack;
2065 const EDetPidStatus pidStatus=GetEMCALPIDStatus(track);
2066 if (pidStatus!=kDetPidOk) return -999.;
2068 const Int_t nMatchClus = track->GetEMCALcluster();
2069 AliVCluster *matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
2071 const Double_t mom = track->P();
2072 const Double_t pt = track->Pt();
2073 const Int_t charge = track->Charge();
2074 const Double_t fClsE = matchedClus->E();
2075 const Double_t EovP = fClsE/mom;
2077 return fEMCALResponse.GetNumberOfSigmas(pt,EovP,type,charge);
2080 //______________________________________________________________________________
2081 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaITS(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2084 // Signal minus expected Signal for ITS
2086 AliVTrack *track=(AliVTrack*)vtrack;
2087 val=fITSResponse.GetSignalDelta(track,type,ratio);
2089 return GetITSPIDStatus(track);
2092 //______________________________________________________________________________
2093 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaTPC(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2096 // Signal minus expected Signal for TPC
2098 AliVTrack *track=(AliVTrack*)vtrack;
2100 // the following call is needed in order to fill the transient data member
2101 // fTPCsignalTuned which is used in the TPCPIDResponse to judge
2102 // if using tuned on data
2103 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))
2104 this->GetTPCsignalTunedOnData(track);
2106 val=fTPCResponse.GetSignalDelta(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection, ratio);
2108 return GetTPCPIDStatus(track);
2111 //______________________________________________________________________________
2112 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaTOF(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2115 // Signal minus expected Signal for TOF
2117 AliVTrack *track=(AliVTrack*)vtrack;
2118 val=GetSignalDeltaTOFold(track, type, ratio);
2120 return GetTOFPIDStatus(track);
2123 //______________________________________________________________________________
2124 AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaHMPID(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2127 // Signal minus expected Signal for HMPID
2129 AliVTrack *track=(AliVTrack*)vtrack;
2130 val=fHMPIDResponse.GetSignalDelta(track, type, ratio);
2132 return GetHMPIDPIDStatus(track);
2135 //______________________________________________________________________________
2136 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputePIDProbability (EDetector detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2139 // Compute PID response of 'detCode'
2143 case kITS: return GetComputeITSProbability(track, nSpecies, p); break;
2144 case kTPC: return GetComputeTPCProbability(track, nSpecies, p); break;
2145 case kTRD: return GetComputeTRDProbability(track, nSpecies, p); break;
2146 case kTOF: return GetComputeTOFProbability(track, nSpecies, p); break;
2147 case kPHOS: return GetComputePHOSProbability(track, nSpecies, p); break;
2148 case kEMCAL: return GetComputeEMCALProbability(track, nSpecies, p); break;
2149 case kHMPID: return GetComputeHMPIDProbability(track, nSpecies, p); break;
2150 default: return kDetNoSignal;
2154 //______________________________________________________________________________
2155 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeITSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2158 // Compute PID response for the ITS
2161 // set flat distribution (no decision)
2162 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2164 const EDetPidStatus pidStatus=GetITSPIDStatus(track);
2165 if (pidStatus!=kDetPidOk) return pidStatus;
2167 if (track->GetDetectorPID()){
2168 return track->GetDetectorPID()->GetRawProbability(kITS, p, nSpecies);
2171 //check for ITS standalone tracks
2173 if( track->GetStatus() & AliVTrack::kTPCin ) isSA=kFALSE;
2175 Double_t mom=track->P();
2176 Double_t dedx=track->GetITSsignal();
2177 Double_t momITS=mom;
2178 UChar_t clumap=track->GetITSClusterMap();
2179 Int_t nPointsForPid=0;
2180 for(Int_t i=2; i<6; i++){
2181 if(clumap&(1<<i)) ++nPointsForPid;
2184 Bool_t mismatch=kTRUE/*, heavy=kTRUE*/;
2185 for (Int_t j=0; j<nSpecies; j++) {
2186 Double_t mass=AliPID::ParticleMassZ(j);//GeV/c^2
2187 const Double_t chargeFactor = TMath::Power(AliPID::ParticleCharge(j),2.);
2188 Double_t bethe=fITSResponse.Bethe(momITS,mass)*chargeFactor;
2189 //TODO: in case of the electron, use the SA parametrisation,
2190 // this needs to be changed if ITS provides a parametrisation
2191 // for electrons also for ITS+TPC tracks
2192 Double_t sigma=fITSResponse.GetResolution(bethe,nPointsForPid,isSA || (j==(Int_t)AliPID::kElectron));
2193 if (TMath::Abs(dedx-bethe) > fRange*sigma) {
2194 p[j]=TMath::Exp(-0.5*fRange*fRange)/sigma;
2196 p[j]=TMath::Exp(-0.5*(dedx-bethe)*(dedx-bethe)/(sigma*sigma))/sigma;
2202 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2207 //______________________________________________________________________________
2208 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeTPCProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2211 // Compute PID response for the TPC
2214 // set flat distribution (no decision)
2215 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2217 const EDetPidStatus pidStatus=GetTPCPIDStatus(track);
2218 if (pidStatus!=kDetPidOk) return pidStatus;
2220 Double_t dedx=track->GetTPCsignal();
2221 Bool_t mismatch=kTRUE/*, heavy=kTRUE*/;
2223 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC)) dedx = this->GetTPCsignalTunedOnData(track);
2225 Double_t bethe = 0.;
2226 Double_t sigma = 0.;
2228 for (Int_t j=0; j<nSpecies; j++) {
2229 AliPID::EParticleType type=AliPID::EParticleType(j);
2231 bethe=fTPCResponse.GetExpectedSignal(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
2232 sigma=fTPCResponse.GetExpectedSigma(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
2234 if (TMath::Abs(dedx-bethe) > fRange*sigma) {
2235 p[j]=TMath::Exp(-0.5*fRange*fRange)/sigma;
2237 p[j]=TMath::Exp(-0.5*(dedx-bethe)*(dedx-bethe)/(sigma*sigma))/sigma;
2243 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2248 //______________________________________________________________________________
2249 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeTOFProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2252 // Compute PID probabilities for TOF
2255 // set flat distribution (no decision)
2256 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2258 const EDetPidStatus pidStatus=GetTOFPIDStatus(track);
2259 if (pidStatus!=kDetPidOk) return pidStatus;
2261 const Double_t meanCorrFactor = 0.07/fTOFtail; // Correction factor on the mean because of the tail (should be ~ 0.1 with tail = 1.1)
2263 for (Int_t j=0; j<nSpecies; j++) {
2264 AliPID::EParticleType type=AliPID::EParticleType(j);
2265 const Double_t nsigmas=GetNumberOfSigmasTOFold(track,type) + meanCorrFactor;
2267 const Double_t expTime = fTOFResponse.GetExpectedSignal(track,type);
2268 const Double_t sig = fTOFResponse.GetExpectedSigma(track->P(),expTime,AliPID::ParticleMassZ(type));
2269 if (TMath::Abs(nsigmas) > (fRange+2)) {
2270 if(nsigmas < fTOFtail)
2271 p[j] = TMath::Exp(-0.5*(fRange+2)*(fRange+2))/sig;
2273 p[j] = TMath::Exp(-(fRange+2 - fTOFtail*0.5)*fTOFtail)/sig;
2275 if(nsigmas < fTOFtail)
2276 p[j] = TMath::Exp(-0.5*nsigmas*nsigmas)/sig;
2278 p[j] = TMath::Exp(-(nsigmas - fTOFtail*0.5)*fTOFtail)/sig;
2284 //______________________________________________________________________________
2285 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod/*=AliTRDPIDResponse::kLQ1D*/) const
2288 // Compute PID probabilities for the TRD
2291 // set flat distribution (no decision)
2292 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2294 const EDetPidStatus pidStatus=GetTRDPIDStatus(track);
2295 if (pidStatus!=kDetPidOk) return pidStatus;
2297 UInt_t TRDslicesForPID[2];
2298 SetTRDSlices(TRDslicesForPID,PIDmethod);
2300 Float_t mom[6]={0.};
2301 Double_t dedx[48]={0.}; // Allocate space for the maximum number of TRD slices
2302 Int_t nslices = TRDslicesForPID[1] - TRDslicesForPID[0] + 1;
2303 AliDebug(1, Form("First Slice: %d, Last Slice: %d, Number of slices: %d", TRDslicesForPID[0], TRDslicesForPID[1], nslices));
2304 for(UInt_t ilayer = 0; ilayer < 6; ilayer++){
2305 mom[ilayer] = track->GetTRDmomentum(ilayer);
2306 for(UInt_t islice = TRDslicesForPID[0]; islice <= TRDslicesForPID[1]; islice++){
2307 dedx[ilayer*nslices+islice-TRDslicesForPID[0]] = track->GetTRDslice(ilayer, islice);
2311 fTRDResponse.GetResponse(nslices, dedx, mom, p,PIDmethod);
2315 //______________________________________________________________________________
2316 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeEMCALProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2319 // Compute PID response for the EMCAL
2322 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2324 const EDetPidStatus pidStatus=GetEMCALPIDStatus(track);
2325 if (pidStatus!=kDetPidOk) return pidStatus;
2327 const Int_t nMatchClus = track->GetEMCALcluster();
2328 AliVCluster *matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
2330 const Double_t mom = track->P();
2331 const Double_t pt = track->Pt();
2332 const Int_t charge = track->Charge();
2333 const Double_t fClsE = matchedClus->E();
2334 const Double_t EovP = fClsE/mom;
2336 // compute the probabilities
2337 fEMCALResponse.ComputeEMCALProbability(nSpecies,pt,EovP,charge,p);
2341 //______________________________________________________________________________
2342 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputePHOSProbability (const AliVTrack */*track*/, Int_t nSpecies, Double_t p[]) const
2345 // Compute PID response for the PHOS
2348 // set flat distribution (no decision)
2349 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2350 return kDetNoSignal;
2353 //______________________________________________________________________________
2354 AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeHMPIDProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2357 // Compute PID response for the HMPID
2360 // set flat distribution (no decision)
2361 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2363 const EDetPidStatus pidStatus=GetHMPIDPIDStatus(track);
2364 if (pidStatus!=kDetPidOk) return pidStatus;
2366 fHMPIDResponse.GetProbability(track,nSpecies,p);
2371 //______________________________________________________________________________
2372 AliPIDResponse::EDetPidStatus AliPIDResponse::GetITSPIDStatus(const AliVTrack *track) const
2374 // compute ITS pid status
2376 // check status bits
2377 if ((track->GetStatus()&AliVTrack::kITSin)==0 &&
2378 (track->GetStatus()&AliVTrack::kITSout)==0) return kDetNoSignal;
2380 const Float_t dEdx=track->GetITSsignal();
2381 if (dEdx<=0) return kDetNoSignal;
2383 // requite at least 3 pid clusters
2384 const UChar_t clumap=track->GetITSClusterMap();
2385 Int_t nPointsForPid=0;
2386 for(Int_t i=2; i<6; i++){
2387 if(clumap&(1<<i)) ++nPointsForPid;
2390 if(nPointsForPid<3) {
2391 return kDetNoSignal;
2397 //______________________________________________________________________________
2398 AliPIDResponse::EDetPidStatus AliPIDResponse:: GetTPCPIDStatus(const AliVTrack *track) const
2400 // compute TPC pid status
2402 // check quality of the track
2403 if ( (track->GetStatus()&AliVTrack::kTPCin )==0 && (track->GetStatus()&AliVTrack::kTPCout)==0 ) return kDetNoSignal;
2406 const Double_t dedx=track->GetTPCsignal();
2407 const UShort_t signalN=track->GetTPCsignalN();
2408 if (signalN<10 || dedx<10) return kDetNoSignal;
2410 if (!(fArrPidResponseMaster && fArrPidResponseMaster->At(AliPID::kPion))) return kDetNoParams;
2415 //______________________________________________________________________________
2416 AliPIDResponse::EDetPidStatus AliPIDResponse::GetTRDPIDStatus(const AliVTrack *track) const
2418 // compute TRD pid status
2420 if((track->GetStatus()&AliVTrack::kTRDout)==0) return kDetNoSignal;
2424 //______________________________________________________________________________
2425 AliPIDResponse::EDetPidStatus AliPIDResponse::GetTOFPIDStatus(const AliVTrack *track) const
2427 // compute TOF pid status
2429 if ((track->GetStatus()&AliVTrack::kTOFout)==0) return kDetNoSignal;
2430 if ((track->GetStatus()&AliVTrack::kTIME)==0) return kDetNoSignal;
2435 //______________________________________________________________________________
2436 Float_t AliPIDResponse::GetTOFMismatchProbability(const AliVTrack *track) const
2438 // compute mismatch probability cross-checking at 5 sigmas with TPC
2439 // currently just implemented as a 5 sigma compatibility cut
2442 const EDetPidStatus tofStatus=GetTOFPIDStatus(track);
2443 if (tofStatus!=kDetPidOk) return 0.;
2446 const EDetPidStatus tpcStatus=GetTPCPIDStatus(track);
2447 if (tpcStatus!=kDetPidOk) return 0.;
2449 const Double_t meanCorrFactor = 0.11/fTOFtail; // Correction factor on the mean because of the tail (should be ~ 0.1 with tail = 1.1)
2450 Bool_t mismatch = kTRUE/*, heavy = kTRUE*/;
2451 for (Int_t j=0; j<AliPID::kSPECIESC; j++) {
2452 AliPID::EParticleType type=AliPID::EParticleType(j);
2453 const Double_t nsigmas=GetNumberOfSigmasTOFold(track,type) + meanCorrFactor;
2455 if (TMath::Abs(nsigmas)<5.){
2456 const Double_t nsigmasTPC=GetNumberOfSigmasTPC(track,type);
2457 if (TMath::Abs(nsigmasTPC)<5.) mismatch=kFALSE;
2468 //______________________________________________________________________________
2469 AliPIDResponse::EDetPidStatus AliPIDResponse:: GetHMPIDPIDStatus(const AliVTrack *track) const
2471 // compute HMPID pid status
2473 Int_t ch = track->GetHMPIDcluIdx()/1000000;
2474 Double_t HMPIDsignal = track->GetHMPIDsignal();
2476 if((track->GetStatus()&AliVTrack::kHMPIDpid)==0 || ch<0 || ch>6 || HMPIDsignal<0) return kDetNoSignal;
2481 //______________________________________________________________________________
2482 AliPIDResponse::EDetPidStatus AliPIDResponse:: GetPHOSPIDStatus(const AliVTrack */*track*/) const
2484 // compute PHOS pid status
2485 return kDetNoSignal;
2488 //______________________________________________________________________________
2489 AliPIDResponse::EDetPidStatus AliPIDResponse:: GetEMCALPIDStatus(const AliVTrack *track) const
2491 // compute EMCAL pid status
2495 const Int_t nMatchClus = track->GetEMCALcluster();
2496 if (nMatchClus<0) return kDetNoSignal;
2498 AliVCluster *matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
2500 if (!(matchedClus && matchedClus->IsEMCAL())) return kDetNoSignal;
2502 const Int_t charge = track->Charge();
2503 if (TMath::Abs(charge)!=1) return kDetNoSignal;
2505 if (!(fEMCALPIDParams && fEMCALPIDParams->At(AliPID::kElectron))) return kDetNoParams;
2511 //______________________________________________________________________________
2512 AliPIDResponse::EDetPidStatus AliPIDResponse::GetPIDStatus(EDetector detector, const AliVTrack *track) const
2515 // check pid status for a track
2519 case kITS: return GetITSPIDStatus(track); break;
2520 case kTPC: return GetTPCPIDStatus(track); break;
2521 case kTRD: return GetTRDPIDStatus(track); break;
2522 case kTOF: return GetTOFPIDStatus(track); break;
2523 case kPHOS: return GetPHOSPIDStatus(track); break;
2524 case kEMCAL: return GetEMCALPIDStatus(track); break;
2525 case kHMPID: return GetHMPIDPIDStatus(track); break;
2526 default: return kDetNoSignal;
2528 return kDetNoSignal;
2532 //______________________________________________________________________________
2533 TString AliPIDResponse::GetChecksum(const TObject* obj) const
2535 // Return the checksum for an object obj (tested to work properly at least for histograms and TSplines).
2537 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!
2539 // For parallel processing, a unique file pathname is required. Uniqueness can be guaranteed by using a unique directory name
2541 TString uniquePathName = Form("tempChecksum_%u", index);
2543 // To get a unique path name, increase the index until no directory
2544 // of such a name exists.
2545 // NOTE: gSystem->AccessPathName(...) returns kTRUE, if the access FAILED!
2546 while (!gSystem->AccessPathName(uniquePathName.Data()))
2547 uniquePathName = Form("tempChecksum_%u", ++index);
2549 if (gSystem->mkdir(uniquePathName.Data()) < 0) {
2550 AliError("Could not create temporary directory to store temp file for checksum determination!");
2554 TString option = "";
2556 // Save object as a macro, which will be deleted immediately after the checksum has been computed
2557 // (does not work for desired data types if saved as *.root for some reason) - one only wants to compare the content, not
2558 // the modification time etc. ...
2559 if (dynamic_cast<const TH1*>(obj))
2560 option = "colz"; // Histos need this option, since w/o this option, a counter is added to the filename
2563 // SaveAs must be called with the fixed fileName only, since the first argument goes into the file content
2564 // for some object types. Thus, change the directory, save the file and then go back
2565 TString oldDir = gSystem->pwd();
2566 gSystem->cd(uniquePathName.Data());
2567 obj->SaveAs(fileName.Data(), option.Data());
2568 gSystem->cd(oldDir.Data());
2570 // Use the file to calculate the MD5 checksum
2571 TMD5* md5 = TMD5::FileChecksum(Form("%s/%s", uniquePathName.Data(), fileName.Data()));
2572 TString checksum = md5->AsString();
2576 gSystem->Exec(Form("rm -rf %s", uniquePathName.Data()));