]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/STEERBase/AliPIDResponse.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
adapt alignment ideal OCDB generation macro and OCDB file to new DCal geometry
[u/mrichter/AliRoot.git] / STEER / STEERBase / AliPIDResponse.cxx
CommitLineData
29bf19f2 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15
4ec8e76d 16/* $Id: AliPIDResponse.cxx 46193 2010-12-21 09:00:14Z wiechula $ */
29bf19f2 17
18//-----------------------------------------------------------------
4ec8e76d 19// Base class for handling the pid response //
20// functions of all detectors //
21// and give access to the nsigmas //
22// //
23// Origin: Jens Wiechula, Uni Tuebingen, jens.wiechula@cern.ch //
29bf19f2 24//-----------------------------------------------------------------
25
4ec8e76d 26#include <TList.h>
27#include <TObjArray.h>
28#include <TPRegexp.h>
29#include <TF1.h>
f84b18dd 30#include <TH2D.h>
4ec8e76d 31#include <TSpline.h>
32#include <TFile.h>
00a38d07 33#include <TArrayI.h>
db0e2c5f 34#include <TArrayF.h>
f84b18dd 35#include <TLinearFitter.h>
5a9dc560 36#include <TSystem.h>
37#include <TMD5.h>
4ec8e76d 38
39#include <AliVEvent.h>
fd21ec8d 40#include <AliVTrack.h>
4ec8e76d 41#include <AliLog.h>
42#include <AliPID.h>
ea235c90 43#include <AliOADBContainer.h>
db0e2c5f 44#include <AliTRDPIDResponseObject.h>
9c499471 45#include <AliTRDdEdxParams.h>
b79db598 46#include <AliTOFPIDParams.h>
567624b5 47#include <AliHMPIDPIDParams.h>
29bf19f2 48
49#include "AliPIDResponse.h"
00a38d07 50#include "AliDetectorPID.h"
29bf19f2 51
80f28562 52#include "AliCentrality.h"
53
29bf19f2 54ClassImp(AliPIDResponse);
55
42fcc729 56Float_t AliPIDResponse::fgTOFmismatchProb = 0.0;
57
4ec8e76d 58AliPIDResponse::AliPIDResponse(Bool_t isMC/*=kFALSE*/) :
59TNamed("PIDResponse","PIDResponse"),
60fITSResponse(isMC),
61fTPCResponse(),
62fTRDResponse(),
63fTOFResponse(),
567624b5 64fHMPIDResponse(),
e96b9916 65fEMCALResponse(),
fd21ec8d 66fRange(5.),
67fITSPIDmethod(kITSTruncMean),
1ceae0ac 68fTuneMConData(kFALSE),
69fTuneMConDataMask(kDetTOF|kDetTPC),
4ec8e76d 70fIsMC(isMC),
a017c06a 71fCachePID(kFALSE),
4ec8e76d 72fOADBPath(),
00a38d07 73fCustomTPCpidResponse(),
4ec8e76d 74fBeamType("PP"),
75fLHCperiod(),
76fMCperiodTPC(),
fd21ec8d 77fMCperiodUser(),
ea235c90 78fCurrentFile(),
87da0205 79fCurrentAliRootRev(-1),
4ec8e76d 80fRecoPass(0),
fd21ec8d 81fRecoPassUser(-1),
1ceae0ac 82fRun(-1),
1b9e31a6 83fOldRun(-1),
78cbd205 84fResT0A(75.),
85fResT0C(65.),
86fResT0AC(55.),
644666df 87fArrPidResponseMaster(NULL),
88fResolutionCorrection(NULL),
89fOADBvoltageMaps(NULL),
87da0205 90fUseTPCEtaCorrection(kFALSE),
91fUseTPCMultiplicityCorrection(kFALSE),
644666df 92fTRDPIDResponseObject(NULL),
9c499471 93fTRDdEdxParams(NULL),
c5fb644a 94fTOFtail(0.9),
644666df 95fTOFPIDParams(NULL),
567624b5 96fHMPIDPIDParams(NULL),
644666df 97fEMCALPIDParams(NULL),
98fCurrentEvent(NULL),
42fcc729 99fCurrCentrality(0.0),
100fBeamTypeNum(kPP)
4ec8e76d 101{
102 //
103 // default ctor
104 //
a635821f 105 AliLog::SetClassDebugLevel("AliPIDResponse",0);
106 AliLog::SetClassDebugLevel("AliESDpid",0);
107 AliLog::SetClassDebugLevel("AliAODpidUtil",0);
ea235c90 108
4ec8e76d 109}
110
111//______________________________________________________________________________
112AliPIDResponse::~AliPIDResponse()
113{
114 //
115 // dtor
116 //
00a38d07 117 delete fArrPidResponseMaster;
118 delete fTRDPIDResponseObject;
9c499471 119 delete fTRDdEdxParams;
00a38d07 120 delete fTOFPIDParams;
4ec8e76d 121}
122
123//______________________________________________________________________________
124AliPIDResponse::AliPIDResponse(const AliPIDResponse &other) :
125TNamed(other),
126fITSResponse(other.fITSResponse),
127fTPCResponse(other.fTPCResponse),
128fTRDResponse(other.fTRDResponse),
129fTOFResponse(other.fTOFResponse),
567624b5 130fHMPIDResponse(other.fHMPIDResponse),
e96b9916 131fEMCALResponse(other.fEMCALResponse),
fd21ec8d 132fRange(other.fRange),
133fITSPIDmethod(other.fITSPIDmethod),
1ceae0ac 134fTuneMConData(other.fTuneMConData),
135fTuneMConDataMask(other.fTuneMConDataMask),
4ec8e76d 136fIsMC(other.fIsMC),
1c9d11be 137fCachePID(other.fCachePID),
4ec8e76d 138fOADBPath(other.fOADBPath),
00a38d07 139fCustomTPCpidResponse(other.fCustomTPCpidResponse),
4ec8e76d 140fBeamType("PP"),
141fLHCperiod(),
142fMCperiodTPC(),
fd21ec8d 143fMCperiodUser(other.fMCperiodUser),
ea235c90 144fCurrentFile(),
87da0205 145fCurrentAliRootRev(other.fCurrentAliRootRev),
4ec8e76d 146fRecoPass(0),
fd21ec8d 147fRecoPassUser(other.fRecoPassUser),
1ceae0ac 148fRun(-1),
1b9e31a6 149fOldRun(-1),
78cbd205 150fResT0A(75.),
151fResT0C(65.),
152fResT0AC(55.),
644666df 153fArrPidResponseMaster(NULL),
154fResolutionCorrection(NULL),
155fOADBvoltageMaps(NULL),
f84b18dd 156fUseTPCEtaCorrection(other.fUseTPCEtaCorrection),
87da0205 157fUseTPCMultiplicityCorrection(other.fUseTPCMultiplicityCorrection),
644666df 158fTRDPIDResponseObject(NULL),
9c499471 159fTRDdEdxParams(NULL),
c5fb644a 160fTOFtail(0.9),
644666df 161fTOFPIDParams(NULL),
567624b5 162fHMPIDPIDParams(NULL),
644666df 163fEMCALPIDParams(NULL),
164fCurrentEvent(NULL),
42fcc729 165fCurrCentrality(0.0),
166fBeamTypeNum(kPP)
4ec8e76d 167{
168 //
169 // copy ctor
170 //
171}
172
173//______________________________________________________________________________
174AliPIDResponse& AliPIDResponse::operator=(const AliPIDResponse &other)
175{
176 //
177 // copy ctor
178 //
179 if(this!=&other) {
180 delete fArrPidResponseMaster;
181 TNamed::operator=(other);
182 fITSResponse=other.fITSResponse;
183 fTPCResponse=other.fTPCResponse;
184 fTRDResponse=other.fTRDResponse;
185 fTOFResponse=other.fTOFResponse;
567624b5 186 fHMPIDResponse=other.fHMPIDResponse;
e96b9916 187 fEMCALResponse=other.fEMCALResponse;
fd21ec8d 188 fRange=other.fRange;
189 fITSPIDmethod=other.fITSPIDmethod;
4ec8e76d 190 fOADBPath=other.fOADBPath;
00a38d07 191 fCustomTPCpidResponse=other.fCustomTPCpidResponse;
1ceae0ac 192 fTuneMConData=other.fTuneMConData;
193 fTuneMConDataMask=other.fTuneMConDataMask;
4ec8e76d 194 fIsMC=other.fIsMC;
1c9d11be 195 fCachePID=other.fCachePID;
4ec8e76d 196 fBeamType="PP";
42fcc729 197 fBeamTypeNum=kPP;
4ec8e76d 198 fLHCperiod="";
199 fMCperiodTPC="";
fd21ec8d 200 fMCperiodUser=other.fMCperiodUser;
ea235c90 201 fCurrentFile="";
87da0205 202 fCurrentAliRootRev=other.fCurrentAliRootRev;
4ec8e76d 203 fRecoPass=0;
fd21ec8d 204 fRecoPassUser=other.fRecoPassUser;
1ceae0ac 205 fRun=-1;
1b9e31a6 206 fOldRun=-1;
78cbd205 207 fResT0A=75.;
208 fResT0C=65.;
209 fResT0AC=55.;
644666df 210 fArrPidResponseMaster=NULL;
211 fResolutionCorrection=NULL;
212 fOADBvoltageMaps=NULL;
1ceae0ac 213 fUseTPCEtaCorrection=other.fUseTPCEtaCorrection;
87da0205 214 fUseTPCMultiplicityCorrection=other.fUseTPCMultiplicityCorrection;
644666df 215 fTRDPIDResponseObject=NULL;
9c499471 216 fTRDdEdxParams=NULL;
644666df 217 fEMCALPIDParams=NULL;
c5fb644a 218 fTOFtail=0.9;
644666df 219 fTOFPIDParams=NULL;
567624b5 220 fHMPIDPIDParams=NULL;
e96b9916 221 fCurrentEvent=other.fCurrentEvent;
87da0205 222
4ec8e76d 223 }
224 return *this;
225}
226
fd21ec8d 227//______________________________________________________________________________
355b831b 228Float_t AliPIDResponse::NumberOfSigmas(EDetector detector, const AliVParticle *vtrack, AliPID::EParticleType type) const
fd21ec8d 229{
230 //
231 // NumberOfSigmas for 'detCode'
232 //
355b831b 233
234 const AliVTrack *track=static_cast<const AliVTrack*>(vtrack);
235 // look for cached value first
236 const AliDetectorPID *detPID=track->GetDetectorPID();
237
238 if ( detPID && detPID->HasNumberOfSigmas(detector)){
239 return detPID->GetNumberOfSigmas(detector, type);
240 } else if (fCachePID) {
241 FillTrackDetectorPID(track, detector);
242 detPID=track->GetDetectorPID();
243 return detPID->GetNumberOfSigmas(detector, type);
fd21ec8d 244 }
355b831b 245
246 return GetNumberOfSigmas(detector, track, type);
fd21ec8d 247}
248
e96b9916 249//______________________________________________________________________________
355b831b 250AliPIDResponse::EDetPidStatus AliPIDResponse::NumberOfSigmas(EDetector detCode, const AliVParticle *track,
251 AliPID::EParticleType type, Double_t &val) const
00a38d07 252{
253 //
355b831b 254 // NumberOfSigmas with detector status as return value
00a38d07 255 //
355b831b 256
257 val=NumberOfSigmas(detCode, track, type);
258 return CheckPIDStatus(detCode, (AliVTrack*)track);
00a38d07 259}
260
1c9d11be 261//______________________________________________________________________________
262// public buffered versions of the PID calculation
263//
264
00a38d07 265//______________________________________________________________________________
266Float_t AliPIDResponse::NumberOfSigmasITS(const AliVParticle *vtrack, AliPID::EParticleType type) const
267{
268 //
269 // Calculate the number of sigmas in the ITS
270 //
271
355b831b 272 return NumberOfSigmas(kITS, vtrack, type);
00a38d07 273}
274
275//______________________________________________________________________________
276Float_t AliPIDResponse::NumberOfSigmasTPC(const AliVParticle *vtrack, AliPID::EParticleType type) const
277{
278 //
279 // Calculate the number of sigmas in the TPC
280 //
281
355b831b 282 return NumberOfSigmas(kTPC, vtrack, type);
00a38d07 283}
284
644666df 285//______________________________________________________________________________
286Float_t AliPIDResponse::NumberOfSigmasTPC( const AliVParticle *vtrack,
287 AliPID::EParticleType type,
f84b18dd 288 AliTPCPIDResponse::ETPCdEdxSource dedxSource) const
644666df 289{
290 //get number of sigmas according the selected TPC gain configuration scenario
291 const AliVTrack *track=static_cast<const AliVTrack*>(vtrack);
292
87da0205 293 Float_t nSigma=fTPCResponse.GetNumberOfSigmas(track, type, dedxSource, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
644666df 294
295 return nSigma;
296}
297
5cd0300d 298//______________________________________________________________________________
299Float_t AliPIDResponse::NumberOfSigmasTRD(const AliVParticle *vtrack, AliPID::EParticleType type) const
300{
301 //
302 // Calculate the number of sigmas in the TRD
303 //
304 return NumberOfSigmas(kTRD, vtrack, type);
305}
306
00a38d07 307//______________________________________________________________________________
1c9d11be 308Float_t AliPIDResponse::NumberOfSigmasTOF(const AliVParticle *vtrack, AliPID::EParticleType type) const
00a38d07 309{
310 //
1c9d11be 311 // Calculate the number of sigmas in the TOF
00a38d07 312 //
313
355b831b 314 return NumberOfSigmas(kTOF, vtrack, type);
1c9d11be 315}
e96b9916 316
567624b5 317//______________________________________________________________________________
318Float_t AliPIDResponse::NumberOfSigmasHMPID(const AliVParticle *vtrack, AliPID::EParticleType type) const
319{
320 //
321 // Calculate the number of sigmas in the EMCAL
322 //
323
324 return NumberOfSigmas(kHMPID, vtrack, type);
325}
326
1c9d11be 327//______________________________________________________________________________
328Float_t AliPIDResponse::NumberOfSigmasEMCAL(const AliVParticle *vtrack, AliPID::EParticleType type) const
329{
330 //
331 // Calculate the number of sigmas in the EMCAL
332 //
e96b9916 333
355b831b 334 return NumberOfSigmas(kEMCAL, vtrack, type);
e96b9916 335}
336
6d0064aa 337//______________________________________________________________________________
1c9d11be 338Float_t AliPIDResponse::NumberOfSigmasEMCAL(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &eop, Double_t showershape[4]) const
339{
340 //
341 // emcal nsigma with eop and showershape
342 //
00a38d07 343 AliVTrack *track=(AliVTrack*)vtrack;
344
6d0064aa 345 AliVCluster *matchedClus = NULL;
346
347 Double_t mom = -1.;
348 Double_t pt = -1.;
349 Double_t EovP = -1.;
350 Double_t fClsE = -1.;
32fa24d6 351
352 // initialize eop and shower shape parameters
353 eop = -1.;
354 for(Int_t i = 0; i < 4; i++){
355 showershape[i] = -1.;
356 }
6d0064aa 357
358 Int_t nMatchClus = -1;
359 Int_t charge = 0;
360
361 // Track matching
362 nMatchClus = track->GetEMCALcluster();
363 if(nMatchClus > -1){
364
365 mom = track->P();
366 pt = track->Pt();
367 charge = track->Charge();
368
369 matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
370
371 if(matchedClus){
372
373 // matched cluster is EMCAL
374 if(matchedClus->IsEMCAL()){
375
376 fClsE = matchedClus->E();
377 EovP = fClsE/mom;
378
379 // fill used EMCAL variables here
380 eop = EovP; // E/p
381 showershape[0] = matchedClus->GetNCells(); // number of cells in cluster
382 showershape[1] = matchedClus->GetM02(); // long axis
383 showershape[2] = matchedClus->GetM20(); // short axis
384 showershape[3] = matchedClus->GetDispersion(); // dispersion
1c9d11be 385
386 // look for cached value first
387 const AliDetectorPID *detPID=track->GetDetectorPID();
388 const EDetector detector=kEMCAL;
389
390 if ( detPID && detPID->HasNumberOfSigmas(detector)){
391 return detPID->GetNumberOfSigmas(detector, type);
392 } else if (fCachePID) {
393 FillTrackDetectorPID(track, detector);
394 detPID=track->GetDetectorPID();
395 return detPID->GetNumberOfSigmas(detector, type);
396 }
397
398 // NSigma value really meaningful only for electrons!
399 return fEMCALResponse.GetNumberOfSigmas(pt,EovP,type,charge);
6d0064aa 400 }
401 }
402 }
403 return -999;
404}
405
567624b5 406//______________________________________________________________________________
1d59271b 407AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDelta(EDetector detector, const AliVParticle *track, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
567624b5 408{
409 //
410 //
411 //
412 val=-9999.;
413 switch (detector){
1d59271b 414 case kITS: return GetSignalDeltaITS(track,type,val,ratio); break;
415 case kTPC: return GetSignalDeltaTPC(track,type,val,ratio); break;
5cd0300d 416 case kTRD: return GetSignalDeltaTRD(track,type,val,ratio); break;
1d59271b 417 case kTOF: return GetSignalDeltaTOF(track,type,val,ratio); break;
418 case kHMPID: return GetSignalDeltaHMPID(track,type,val,ratio); break;
567624b5 419 default: return kDetNoSignal;
420 }
421 return kDetNoSignal;
422}
423
424//______________________________________________________________________________
1d59271b 425Double_t AliPIDResponse::GetSignalDelta(EDetector detCode, const AliVParticle *track, AliPID::EParticleType type, Bool_t ratio/*=kFALSE*/) const
567624b5 426{
427 //
428 //
429 //
430 Double_t val=-9999.;
1d59271b 431 EDetPidStatus stat=GetSignalDelta(detCode, track, type, val, ratio);
567624b5 432 if ( stat==kDetNoSignal ) val=-9999.;
433 return val;
434}
435
00a38d07 436//______________________________________________________________________________
355b831b 437AliPIDResponse::EDetPidStatus AliPIDResponse::ComputePIDProbability (EDetCode detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
00a38d07 438{
00a38d07 439 // Compute PID response of 'detCode'
355b831b 440
441 // find detector code from detector bit mask
442 Int_t detector=-1;
443 for (Int_t idet=0; idet<kNdetectors; ++idet) if ( (detCode&(1<<idet)) ) { detector=idet; break; }
444 if (detector==-1) return kDetNoSignal;
00a38d07 445
355b831b 446 return ComputePIDProbability((EDetector)detector, track, nSpecies, p);
00a38d07 447}
448
fd21ec8d 449//______________________________________________________________________________
355b831b 450AliPIDResponse::EDetPidStatus AliPIDResponse::ComputePIDProbability (EDetector detector, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
fd21ec8d 451{
452 //
355b831b 453 // Compute PID response of 'detector'
fd21ec8d 454 //
455
1c9d11be 456 const AliDetectorPID *detPID=track->GetDetectorPID();
355b831b 457
458 if ( detPID && detPID->HasRawProbability(detector)){
1c9d11be 459 return detPID->GetRawProbability(detector, p, nSpecies);
460 } else if (fCachePID) {
461 FillTrackDetectorPID(track, detector);
462 detPID=track->GetDetectorPID();
463 return detPID->GetRawProbability(detector, p, nSpecies);
fd21ec8d 464 }
465
355b831b 466 //if no caching return values calculated from scratch
467 return GetComputePIDProbability(detector, track, nSpecies, p);
468}
469
470//______________________________________________________________________________
471AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeITSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
472{
473 // Compute PID response for the ITS
474 return ComputePIDProbability(kITS, track, nSpecies, p);
fd21ec8d 475}
355b831b 476
fd21ec8d 477//______________________________________________________________________________
478AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTPCProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
479{
fd21ec8d 480 // Compute PID response for the TPC
355b831b 481 return ComputePIDProbability(kTPC, track, nSpecies, p);
fd21ec8d 482}
355b831b 483
fd21ec8d 484//______________________________________________________________________________
485AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTOFProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
486{
fd21ec8d 487 // Compute PID response for the
355b831b 488 return ComputePIDProbability(kTOF, track, nSpecies, p);
fd21ec8d 489}
355b831b 490
fd21ec8d 491//______________________________________________________________________________
355b831b 492AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
fd21ec8d 493{
fd21ec8d 494 // Compute PID response for the
355b831b 495 return ComputePIDProbability(kTRD, track, nSpecies, p);
fd21ec8d 496}
355b831b 497
fd21ec8d 498//______________________________________________________________________________
e96b9916 499AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeEMCALProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
fd21ec8d 500{
fd21ec8d 501 // Compute PID response for the EMCAL
355b831b 502 return ComputePIDProbability(kEMCAL, track, nSpecies, p);
1c9d11be 503}
504//______________________________________________________________________________
505AliPIDResponse::EDetPidStatus AliPIDResponse::ComputePHOSProbability (const AliVTrack */*track*/, Int_t nSpecies, Double_t p[]) const
506{
1c9d11be 507 // Compute PID response for the PHOS
1c9d11be 508
509 // set flat distribution (no decision)
00a38d07 510 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
1c9d11be 511 return kDetNoSignal;
512}
355b831b 513
1c9d11be 514//______________________________________________________________________________
515AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeHMPIDProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
516{
1c9d11be 517 // Compute PID response for the HMPID
355b831b 518 return ComputePIDProbability(kHMPID, track, nSpecies, p);
519}
fd21ec8d 520
355b831b 521//______________________________________________________________________________
522AliPIDResponse::EDetPidStatus AliPIDResponse::ComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const
523{
524 // Compute PID response for the
525 return GetComputeTRDProbability(track, nSpecies, p, PIDmethod);
526}
e96b9916 527
355b831b 528//______________________________________________________________________________
529AliPIDResponse::EDetPidStatus AliPIDResponse::CheckPIDStatus(EDetector detector, const AliVTrack *track) const
530{
531 // calculate detector pid status
532
533 const Int_t iDetCode=(Int_t)detector;
534 if (iDetCode<0||iDetCode>=kNdetectors) return kDetNoSignal;
535 const AliDetectorPID *detPID=track->GetDetectorPID();
536
537 if ( detPID ){
538 return detPID->GetPIDStatus(detector);
1c9d11be 539 } else if (fCachePID) {
540 FillTrackDetectorPID(track, detector);
541 detPID=track->GetDetectorPID();
355b831b 542 return detPID->GetPIDStatus(detector);
e96b9916 543 }
355b831b 544
545 // if not buffered and no buffering is requested
546 return GetPIDStatus(detector, track);
fd21ec8d 547}
548
4ec8e76d 549//______________________________________________________________________________
00a38d07 550void AliPIDResponse::InitialiseEvent(AliVEvent *event, Int_t pass, Int_t run)
4ec8e76d 551{
552 //
553 // Apply settings for the current event
554 //
555 fRecoPass=pass;
e96b9916 556
78cbd205 557
644666df 558 fCurrentEvent=NULL;
4ec8e76d 559 if (!event) return;
e96b9916 560 fCurrentEvent=event;
00a38d07 561 if (run>0) fRun=run;
562 else fRun=event->GetRunNumber();
4ec8e76d 563
564 if (fRun!=fOldRun){
565 ExecNewRun();
566 fOldRun=fRun;
567 }
568
569 //TPC resolution parametrisation PbPb
570 if ( fResolutionCorrection ){
571 Double_t corrSigma=fResolutionCorrection->Eval(GetTPCMultiplicityBin(event));
572 fTPCResponse.SetSigma(3.79301e-03*corrSigma, 2.21280e+04);
573 }
574
87da0205 575 // Set up TPC multiplicity for PbPb
22158469
JW		 576 if (fUseTPCMultiplicityCorrection) {
577 Int_t numESDtracks = event->GetNumberOfESDTracks();
578 if (numESDtracks < 0) {
579 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!");
580 numESDtracks = 0;
581 }
582 fTPCResponse.SetCurrentEventMultiplicity(numESDtracks);
87da0205 583 }
87da0205 584 else
585 fTPCResponse.SetCurrentEventMultiplicity(0);
586
4ec8e76d 587 //TOF resolution
b79db598 588 SetTOFResponse(event, (AliPIDResponse::EStartTimeType_t)fTOFPIDParams->GetStartTimeMethod());
589
80f28562 590
591 // Get and set centrality
592 AliCentrality *centrality = event->GetCentrality();
593 if(centrality){
594 fCurrCentrality = centrality->GetCentralityPercentile("V0M");
595 }
596 else{
597 fCurrCentrality = -1;
598 }
87da0205 599
600 // Set centrality percentile for EMCAL
601 fEMCALResponse.SetCentrality(fCurrCentrality);
602
c53e310b 603 // switch off some TOF channel according to OADB to match data TOF matching eff
604 if (fTuneMConData && ((fTuneMConDataMask & kDetTOF) == kDetTOF) && fTOFPIDParams->GetTOFmatchingLossMC() > 0.01){
605 Int_t ntrk = event->GetNumberOfTracks();
606 for(Int_t i=0;i < ntrk;i++){
607 AliVParticle *trk = event->GetTrack(i);
608 Int_t channel = GetTOFResponse().GetTOFchannel(trk);
609 Int_t swoffEachOfThem = Int_t(100./fTOFPIDParams->GetTOFmatchingLossMC() + 0.5);
610 if(!(channel%swoffEachOfThem)) ((AliVTrack *) trk)->ResetStatus(AliVTrack::kTOFout);
611 }
612 }
613
4ec8e76d 614}
615
616//______________________________________________________________________________
617void AliPIDResponse::ExecNewRun()
618{
619 //
620 // Things to Execute upon a new run
621 //
622 SetRecoInfo();
623
624 SetITSParametrisation();
625
626 SetTPCPidResponseMaster();
627 SetTPCParametrisation();
f84b18dd 628 SetTPCEtaMaps();
53d016dc 629
630 SetTRDPidResponseMaster();
5eb3e944 631 //has to precede InitializeTRDResponse(), otherwise the read-out fTRDdEdxParams is not pased in TRDResponse!
9c499471 632 SetTRDdEdxParams();
53d016dc 633 InitializeTRDResponse();
b2138b40 634
635 SetEMCALPidResponseMaster();
636 InitializeEMCALResponse();
4ec8e76d 637
b79db598 638 SetTOFPidResponseMaster();
639 InitializeTOFResponse();
644666df 640
567624b5 641 SetHMPIDPidResponseMaster();
642 InitializeHMPIDResponse();
643
644666df 644 if (fCurrentEvent) fTPCResponse.SetMagField(fCurrentEvent->GetMagneticField());
4ec8e76d 645}
646
1c9d11be 647//______________________________________________________________________________
4ec8e76d 648Double_t AliPIDResponse::GetTPCMultiplicityBin(const AliVEvent * const event)
649{
650 //
651 // Get TPC multiplicity in bins of 150
652 //
653
654 const AliVVertex* vertexTPC = event->GetPrimaryVertex();
655 Double_t tpcMulti=0.;
656 if(vertexTPC){
657 Double_t vertexContribTPC=vertexTPC->GetNContributors();
658 tpcMulti=vertexContribTPC/150.;
659 if (tpcMulti>20.) tpcMulti=20.;
660 }
661
662 return tpcMulti;
663}
664
665//______________________________________________________________________________
666void AliPIDResponse::SetRecoInfo()
667{
668 //
669 // Set reconstruction information
670 //
671
672 //reset information
673 fLHCperiod="";
674 fMCperiodTPC="";
675
676 fBeamType="";
677
678 fBeamType="PP";
42fcc729 679 fBeamTypeNum=kPP;
bbce5a64 680
681 Bool_t hasProdInfo=(fCurrentFile.BeginsWith("LHC"));
4ec8e76d 682
3e95957e 683 TPRegexp reg(".*(LHC1[1-3][a-z]+[0-9]+[a-z_]*)[/_].*");
bbce5a64 684 if (hasProdInfo) reg=TPRegexp("LHC1[1-2][a-z]+[0-9]+[a-z_]*");
ea6ba565 685 TPRegexp reg12a17("LHC1[2-4][a-z]");
1436d6bb 686
4ec8e76d 687 //find the period by run number (UGLY, but not stored in ESD and AOD... )
688 if (fRun>=114737&&fRun<=117223) { fLHCperiod="LHC10B"; fMCperiodTPC="LHC10D1"; }
689 else if (fRun>=118503&&fRun<=121040) { fLHCperiod="LHC10C"; fMCperiodTPC="LHC10D1"; }
690 else if (fRun>=122195&&fRun<=126437) { fLHCperiod="LHC10D"; fMCperiodTPC="LHC10F6A"; }
99e9d5ec 691 else if (fRun>=127710&&fRun<=130850) { fLHCperiod="LHC10E"; fMCperiodTPC="LHC10F6A"; }
4ec8e76d 692 else if (fRun>=133004&&fRun<=135029) { fLHCperiod="LHC10F"; fMCperiodTPC="LHC10F6A"; }
693 else if (fRun>=135654&&fRun<=136377) { fLHCperiod="LHC10G"; fMCperiodTPC="LHC10F6A"; }
12d3abbc 694 else if (fRun>=136851&&fRun<=139846) {
ea235c90 695 fLHCperiod="LHC10H";
696 fMCperiodTPC="LHC10H8";
697 if (reg.MatchB(fCurrentFile)) fMCperiodTPC="LHC11A10";
ef7661fd 698 // exception for 13d2 and later
699 if (fCurrentAliRootRev >= 62714) fMCperiodTPC="LHC13D2";
ea235c90 700 fBeamType="PBPB";
42fcc729 701 fBeamTypeNum=kPBPB;
ea235c90 702 }
12d3abbc 703 else if (fRun>=139847&&fRun<=146974) { fLHCperiod="LHC11A"; fMCperiodTPC="LHC10F6A"; }
704 //TODO: periods 11B (146975-150721), 11C (150722-155837) are not yet treated assume 11d for the moment
705 else if (fRun>=146975&&fRun<=155837) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
706 else if (fRun>=155838&&fRun<=159649) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
707 // also for 11e (159650-162750),f(162751-165771) use 11d
708 else if (fRun>=159650&&fRun<=162750) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
709 else if (fRun>=162751&&fRun<=165771) { fLHCperiod="LHC11D"; fMCperiodTPC="LHC10F6A"; }
00a38d07 710
12d3abbc 711 else if (fRun>=165772 && fRun<=170718) {
3077a03d 712 fLHCperiod="LHC11H";
713 fMCperiodTPC="LHC11A10";
714 fBeamType="PBPB";
42fcc729 715 fBeamTypeNum=kPBPB;
a78fd045 716 if (reg12a17.MatchB(fCurrentFile)) fMCperiodTPC="LHC12A17";
3077a03d 717 }
03696add 718 if (fRun>=170719 && fRun<=177311) {
719 fLHCperiod="LHC12A";
720 fBeamType="PP";
721 fBeamTypeNum=kPP;
722 fMCperiodTPC="LHC10F6A";
723 if (fCurrentAliRootRev >= 62714)
dbeb0132 724 fMCperiodTPC="LHC14E2";
03696add 725 }
3b3bf053 726 // for the moment use LHC12b parameters up to LHC12d
03696add 727 if (fRun>=177312 /*&& fRun<=179356*/) {
728 fLHCperiod="LHC12B";
729 fBeamType="PP";
730 fBeamTypeNum=kPP;
731 fMCperiodTPC="LHC10F6A";
732 if (fCurrentAliRootRev >= 62714)
dbeb0132 733 fMCperiodTPC="LHC14E2";
03696add 734 }
42fcc729 735// if (fRun>=179357 && fRun<=183173) { fLHCperiod="LHC12C"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
736// if (fRun>=183174 && fRun<=186345) { fLHCperiod="LHC12D"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
737// if (fRun>=186346 && fRun<=186635) { fLHCperiod="LHC12E"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
738
739// if (fRun>=186636 && fRun<=188166) { fLHCperiod="LHC12F"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
740// if (fRun >= 188167 && fRun <= 188355 ) { fLHCperiod="LHC12G"; fBeamType="PP"; fBeamTypeNum=kPP;/*fMCperiodTPC="";*/ }
741// if (fRun >= 188356 && fRun <= 188503 ) { fLHCperiod="LHC12G"; fBeamType="PPB"; fBeamTypeNum=kPPB;/*fMCperiodTPC="";*/ }
3b3bf053 742// for the moment use 12g parametrisation for all full gain runs (LHC12e+)
5b7afd75 743 if (fRun >= 186346 && fRun < 188719) { fLHCperiod="LHC12G"; fBeamType="PPB";fBeamTypeNum=kPPB; fMCperiodTPC="LHC12G"; }
87da0205 744
5b7afd75 745 // Dedicated splines for periods 12g and 12i(j) (and use more appropriate MC)
746 if (fRun >= 188720 && fRun <= 192738) {
747 fLHCperiod="LHC12H";
748 fBeamType="PP";
749 fBeamTypeNum=kPP;
750 fMCperiodTPC="LHC10F6A";
751 if (fCurrentAliRootRev >= 62714)
752 fMCperiodTPC="LHC13B2_FIXn1";
753 }
754 if (fRun >= 192739 && fRun <= 194479) {
755 fLHCperiod="LHC12I";
756 fBeamType="PP";
757 fBeamTypeNum=kPP;
758 fMCperiodTPC="LHC10F6A";
759 if (fCurrentAliRootRev >= 62714)
760 fMCperiodTPC="LHC13B2_FIXn1";
761 }
762
87da0205 763 // New parametrisation for 2013 pPb runs
5b7afd75 764 if (fRun >= 194480) {
87da0205 765 fLHCperiod="LHC13B";
766 fBeamType="PPB";
42fcc729 767 fBeamTypeNum=kPPB;
bbce5a64 768 fMCperiodTPC="LHC12G";
87da0205 769
770 if (fCurrentAliRootRev >= 61605)
771 fMCperiodTPC="LHC13B2_FIX";
bbce5a64 772 if (fCurrentAliRootRev >= 62714)
773 fMCperiodTPC="LHC13B2_FIXn1";
c4bec231 774
775 // High luminosity pPb runs require different parametrisations
776 if (fRun >= 195875 && fRun <= 197411) {
777 fLHCperiod="LHC13F";
778 }
87da0205 779 }
80ab5635 780
66de625c 781 //exception new pp MC productions from 2011 (11a periods have 10f6a splines!)
42fcc729 782 if (fBeamType=="PP" && reg.MatchB(fCurrentFile) && !fCurrentFile.Contains("LHC11a")) { fMCperiodTPC="LHC11B2"; fBeamType="PP";fBeamTypeNum=kPP; }
4a527e08 783 // exception for 11f1
bbce5a64 784 if (fCurrentFile.Contains("LHC11f1")) fMCperiodTPC="LHC11F1";
bf26ce58 785 // exception for 12f1a, 12f1b and 12i3
bbce5a64 786 if (fCurrentFile.Contains("LHC12f1") || fCurrentFile.Contains("LHC12i3")) fMCperiodTPC="LHC12F1";
c3ee524d 787 // exception for 12c4
bbce5a64 788 if (fCurrentFile.Contains("LHC12c4")) fMCperiodTPC="LHC12C4";
361813a2 789 // exception for 13d1 11d anchored prod
790 if (fLHCperiod=="LHC11D" && fCurrentFile.Contains("LHC13d1")) fMCperiodTPC="LHC13D1";
4ec8e76d 791}
792
793//______________________________________________________________________________
794void AliPIDResponse::SetITSParametrisation()
795{
796 //
797 // Set the ITS parametrisation
798 //
799}
800
f84b18dd 801
802//______________________________________________________________________________
803void AliPIDResponse::AddPointToHyperplane(TH2D* h, TLinearFitter* linExtrapolation, Int_t binX, Int_t binY)
804{
805 if (h->GetBinContent(binX, binY) <= 1e-4)
806 return; // Reject bins without content (within some numerical precision) or with strange content
807
808 Double_t coord[2] = {0, 0};
809 coord[0] = h->GetXaxis()->GetBinCenter(binX);
810 coord[1] = h->GetYaxis()->GetBinCenter(binY);
811 Double_t binError = h->GetBinError(binX, binY);
812 if (binError <= 0) {
813 binError = 1000; // Should not happen because bins without content are rejected for the map (TH2D* h)
814 printf("ERROR: This should never happen: Trying to add bin in addPointToHyperplane with error not set....\n");
815 }
816 linExtrapolation->AddPoint(coord, h->GetBinContent(binX, binY, binError));
817}
818
819
820//______________________________________________________________________________
821TH2D* AliPIDResponse::RefineHistoViaLinearInterpolation(TH2D* h, Double_t refineFactorX, Double_t refineFactorY)
822{
823 if (!h)
824 return 0x0;
825
826 // Interpolate to finer map
827 TLinearFitter* linExtrapolation = new TLinearFitter(2, "hyp2", "");
828
829 Double_t upperMapBoundY = h->GetYaxis()->GetBinUpEdge(h->GetYaxis()->GetNbins());
830 Double_t lowerMapBoundY = h->GetYaxis()->GetBinLowEdge(1);
1b45e564 831 Int_t nBinsX = 30;
f84b18dd 832 // 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,
833 // scale the number of bins correspondingly
1b45e564 834 Int_t nBinsY = TMath::Nint((upperMapBoundY - lowerMapBoundY) / (0.02 - 0.0016) * 40);
f84b18dd 835 Int_t nBinsXrefined = nBinsX * refineFactorX;
836 Int_t nBinsYrefined = nBinsY * refineFactorY;
837
838 TH2D* hRefined = new TH2D(Form("%s_refined", h->GetName()), Form("%s (refined)", h->GetTitle()),
839 nBinsXrefined, h->GetXaxis()->GetBinLowEdge(1), h->GetXaxis()->GetBinUpEdge(h->GetXaxis()->GetNbins()),
840 nBinsYrefined, lowerMapBoundY, upperMapBoundY);
841
842 for (Int_t binX = 1; binX <= nBinsXrefined; binX++) {
843 for (Int_t binY = 1; binY <= nBinsYrefined; binY++) {
844
845 hRefined->SetBinContent(binX, binY, 1); // Default value is 1
846
847 Double_t centerX = hRefined->GetXaxis()->GetBinCenter(binX);
848 Double_t centerY = hRefined->GetYaxis()->GetBinCenter(binY);
849
1b45e564 850 /*OLD
f84b18dd 851 linExtrapolation->ClearPoints();
852
853 // For interpolation: Just take the corresponding bin from the old histo.
854 // For extrapolation: take the last available bin from the old histo.
855 // If the boundaries are to be skipped, also skip the corresponding bins
856 Int_t oldBinX = h->GetXaxis()->FindBin(centerX);
857 if (oldBinX < 1)
858 oldBinX = 1;
859 if (oldBinX > nBinsX)
860 oldBinX = nBinsX;
861
862 Int_t oldBinY = h->GetYaxis()->FindBin(centerY);
863 if (oldBinY < 1)
864 oldBinY = 1;
865 if (oldBinY > nBinsY)
866 oldBinY = nBinsY;
867
868 // Neighbours left column
869 if (oldBinX >= 2) {
870 if (oldBinY >= 2) {
871 AddPointToHyperplane(h, linExtrapolation, oldBinX - 1, oldBinY - 1);
872 }
873
874 AddPointToHyperplane(h, linExtrapolation, oldBinX - 1, oldBinY);
875
876 if (oldBinY < nBinsY) {
877 AddPointToHyperplane(h, linExtrapolation, oldBinX - 1, oldBinY + 1);
878 }
879 }
880
881 // Neighbours (and point itself) same column
882 if (oldBinY >= 2) {
883 AddPointToHyperplane(h, linExtrapolation, oldBinX, oldBinY - 1);
884 }
885
886 AddPointToHyperplane(h, linExtrapolation, oldBinX, oldBinY);
887
888 if (oldBinY < nBinsY) {
889 AddPointToHyperplane(h, linExtrapolation, oldBinX, oldBinY + 1);
890 }
891
892 // Neighbours right column
893 if (oldBinX < nBinsX) {
894 if (oldBinY >= 2) {
895 AddPointToHyperplane(h, linExtrapolation, oldBinX + 1, oldBinY - 1);
896 }
897
898 AddPointToHyperplane(h, linExtrapolation, oldBinX + 1, oldBinY);
899
900 if (oldBinY < nBinsY) {
901 AddPointToHyperplane(h, linExtrapolation, oldBinX + 1, oldBinY + 1);
902 }
903 }
904
905
906 // Fit 2D-hyperplane
907 if (linExtrapolation->GetNpoints() <= 0)
908 continue;
909
910 if (linExtrapolation->Eval() != 0)// EvalRobust -> Takes much, much, [...], much more time (~hours instead of seconds)
911 continue;
912
913 // Fill the bin of the refined histogram with the extrapolated value
914 Double_t interpolatedValue = linExtrapolation->GetParameter(0) + linExtrapolation->GetParameter(1) * centerX
915 + linExtrapolation->GetParameter(2) * centerY;
916 */
f85a3764 917 Double_t interpolatedValue = h->Interpolate(centerX, centerY) ;
f84b18dd 918 hRefined->SetBinContent(binX, binY, interpolatedValue);
919 }
920 }
921
1b45e564 922
923 // Problem: Interpolation does not work before/beyond center of first/last bin (as the name suggests).
924 // Therefore, for each row in dEdx: Take last bin from old map and interpolate values from center and edge.
925 // Assume line through these points and extropolate to last bin of refined map
926 const Double_t firstOldXbinUpEdge = h->GetXaxis()->GetBinUpEdge(1);
927 const Double_t firstOldXbinCenter = h->GetXaxis()->GetBinCenter(1);
928
929 const Double_t oldXbinHalfWidth = firstOldXbinUpEdge - firstOldXbinCenter;
930
931 const Double_t lastOldXbinLowEdge = h->GetXaxis()->GetBinLowEdge(h->GetNbinsX());
932 const Double_t lastOldXbinCenter = h->GetXaxis()->GetBinCenter(h->GetNbinsX());
933
934 for (Int_t binY = 1; binY <= nBinsYrefined; binY++) {
935 Double_t centerY = hRefined->GetYaxis()->GetBinCenter(binY);
936
937 const Double_t interpolatedCenterFirstXbin = h->Interpolate(firstOldXbinCenter, centerY);
938 const Double_t interpolatedUpEdgeFirstXbin = h->Interpolate(firstOldXbinUpEdge, centerY);
939
940 const Double_t extrapolationSlopeFirstXbin = (interpolatedUpEdgeFirstXbin - interpolatedCenterFirstXbin) / oldXbinHalfWidth;
941 const Double_t extrapolationOffsetFirstXbin = interpolatedCenterFirstXbin;
942
943
944 const Double_t interpolatedCenterLastXbin = h->Interpolate(lastOldXbinCenter, centerY);
945 const Double_t interpolatedLowEdgeLastXbin = h->Interpolate(lastOldXbinLowEdge, centerY);
946
947 const Double_t extrapolationSlopeLastXbin = (interpolatedCenterLastXbin - interpolatedLowEdgeLastXbin) / oldXbinHalfWidth;
948 const Double_t extrapolationOffsetLastXbin = interpolatedCenterLastXbin;
949
950 for (Int_t binX = 1; binX <= nBinsXrefined; binX++) {
951 Double_t centerX = hRefined->GetXaxis()->GetBinCenter(binX);
952
953 if (centerX < firstOldXbinCenter) {
954 Double_t extrapolatedValue = extrapolationOffsetFirstXbin + (centerX - firstOldXbinCenter) * extrapolationSlopeFirstXbin;
955 hRefined->SetBinContent(binX, binY, extrapolatedValue);
956 }
957 else if (centerX <= lastOldXbinCenter) {
958 continue;
959 }
960 else {
961 Double_t extrapolatedValue = extrapolationOffsetLastXbin + (centerX - lastOldXbinCenter) * extrapolationSlopeLastXbin;
962 hRefined->SetBinContent(binX, binY, extrapolatedValue);
963 }
964 }
965 }
966
f84b18dd 967 delete linExtrapolation;
968
969 return hRefined;
970}
971
972//______________________________________________________________________________
973void AliPIDResponse::SetTPCEtaMaps(Double_t refineFactorMapX, Double_t refineFactorMapY,
974 Double_t refineFactorSigmaMapX, Double_t refineFactorSigmaMapY)
975{
976 //
977 // Load the TPC eta correction maps from the OADB
978 //
979
f85a3764 980 if (fUseTPCEtaCorrection == kFALSE) {
981 // Disable eta correction via setting no maps
982 if (!fTPCResponse.SetEtaCorrMap(0x0))
1b45e564 983 AliInfo("Request to disable TPC eta correction -> Eta correction has been disabled");
f85a3764 984 else
985 AliError("Request to disable TPC eta correction -> Some error occured when unloading the correction maps");
986
987 if (!fTPCResponse.SetSigmaParams(0x0, 0))
1b45e564 988 AliInfo("Request to disable TPC eta correction -> Using old parametrisation for sigma");
989 else
f85a3764 990 AliError("Request to disable TPC eta correction -> Some error occured when unloading the sigma maps");
991
992 return;
993 }
1b45e564 994
f84b18dd 995 TString dataType = "DATA";
996 TString period = fLHCperiod.IsNull() ? "No period information" : fLHCperiod;
997
998 if (fIsMC) {
87da0205 999 if (!(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) {
f85a3764 1000 period=fMCperiodTPC;
1001 dataType="MC";
1002 }
f84b18dd 1003 fRecoPass = 1;
1004
87da0205 1005 if (!(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC)) && fMCperiodTPC.IsNull()) {
f84b18dd 1006 AliFatal("MC detected, but no MC period set -> Not changing eta maps!");
1007 return;
1008 }
f84b18dd 1009 }
f85a3764 1010
1011 Int_t recopass = fRecoPass;
a2c30af1 1012 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC) )
f85a3764 1013 recopass = fRecoPassUser;
f84b18dd 1014
f85a3764 1015 TString defaultObj = Form("Default_%s_pass%d", dataType.Data(), recopass);
f84b18dd 1016
f85a3764 1017 AliInfo(Form("Current period and reco pass: %s.pass%d", period.Data(), recopass));
f84b18dd 1018
1019 // Invalidate old maps
1020 fTPCResponse.SetEtaCorrMap(0x0);
1021 fTPCResponse.SetSigmaParams(0x0, 0);
1022
1023 // Load the eta correction maps
f85a3764 1024 AliOADBContainer etaMapsCont(Form("TPCetaMaps_%s_pass%d", dataType.Data(), recopass));
f84b18dd 1025
1026 Int_t statusCont = etaMapsCont.InitFromFile(Form("%s/COMMON/PID/data/TPCetaMaps.root", fOADBPath.Data()),
f85a3764 1027 Form("TPCetaMaps_%s_pass%d", dataType.Data(), recopass));
f84b18dd 1028 if (statusCont) {
1029 AliError("Failed initializing TPC eta correction maps from OADB -> Disabled eta correction");
87da0205 1030 fUseTPCEtaCorrection = kFALSE;
f84b18dd 1031 }
1032 else {
1033 AliInfo(Form("Loading TPC eta correction map from %s/COMMON/PID/data/TPCetaMaps.root", fOADBPath.Data()));
1034
1035 TH2D* etaMap = 0x0;
1036
87da0205 1037 if (fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) {
f85a3764 1038 TString searchMap = Form("TPCetaMaps_%s_%s_pass%d", dataType.Data(), period.Data(), recopass);
f84b18dd 1039 etaMap = dynamic_cast<TH2D *>(etaMapsCont.GetDefaultObject(searchMap.Data()));
1040 if (!etaMap) {
1041 // Try default object
1042 etaMap = dynamic_cast<TH2D *>(etaMapsCont.GetDefaultObject(defaultObj.Data()));
1043 }
1044 }
1045 else {
1046 etaMap = dynamic_cast<TH2D *>(etaMapsCont.GetObject(fRun, defaultObj.Data()));
1047 }
1048
1049
1050 if (!etaMap) {
1051 AliError(Form("TPC eta correction map not found for run %d and also no default map found -> Disabled eta correction!!!", fRun));
87da0205 1052 fUseTPCEtaCorrection = kFALSE;
f84b18dd 1053 }
1054 else {
1055 TH2D* etaMapRefined = RefineHistoViaLinearInterpolation(etaMap, refineFactorMapX, refineFactorMapY);
1056
1057 if (etaMapRefined) {
1058 if (!fTPCResponse.SetEtaCorrMap(etaMapRefined)) {
1059 AliError(Form("Failed to set TPC eta correction map for run %d -> Disabled eta correction!!!", fRun));
1060 fTPCResponse.SetEtaCorrMap(0x0);
87da0205 1061 fUseTPCEtaCorrection = kFALSE;
f84b18dd 1062 }
1063 else {
5a9dc560 1064 AliInfo(Form("Loaded TPC eta correction map (refine factors %.2f/%.2f) from %s/COMMON/PID/data/TPCetaMaps.root: %s (MD5(map) = %s)",
1065 refineFactorMapX, refineFactorMapY, fOADBPath.Data(), fTPCResponse.GetEtaCorrMap()->GetTitle(),
1066 GetChecksum(fTPCResponse.GetEtaCorrMap()).Data()));
f84b18dd 1067 }
1068
1069 delete etaMapRefined;
1070 }
1071 else {
1072 AliError(Form("Failed to set TPC eta correction map for run %d (map was loaded, but couldn't be refined) -> Disabled eta correction!!!", fRun));
87da0205 1073 fUseTPCEtaCorrection = kFALSE;
f84b18dd 1074 }
1075 }
1076 }
1077
87da0205 1078 // If there was some problem loading the eta maps, it makes no sense to load the sigma maps (that require eta corrected data)
1079 if (fUseTPCEtaCorrection == kFALSE) {
1080 AliError("Failed to load TPC eta correction map required by sigma maps -> Using old parametrisation for sigma");
1081 return;
1082 }
1083
f84b18dd 1084 // Load the sigma parametrisation (1/dEdx vs tanTheta_local (~eta))
f85a3764 1085 AliOADBContainer etaSigmaMapsCont(Form("TPCetaSigmaMaps_%s_pass%d", dataType.Data(), recopass));
f84b18dd 1086
1087 statusCont = etaSigmaMapsCont.InitFromFile(Form("%s/COMMON/PID/data/TPCetaMaps.root", fOADBPath.Data()),
f85a3764 1088 Form("TPCetaSigmaMaps_%s_pass%d", dataType.Data(), recopass));
f84b18dd 1089 if (statusCont) {
1090 AliError("Failed initializing TPC eta sigma maps from OADB -> Using old sigma parametrisation");
1091 }
1092 else {
1093 AliInfo(Form("Loading TPC eta sigma map from %s/COMMON/PID/data/TPCetaMaps.root", fOADBPath.Data()));
1094
1095 TObjArray* etaSigmaPars = 0x0;
1096
87da0205 1097 if (fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) {
f85a3764 1098 TString searchMap = Form("TPCetaSigmaMaps_%s_%s_pass%d", dataType.Data(), period.Data(), recopass);
f84b18dd 1099 etaSigmaPars = dynamic_cast<TObjArray *>(etaSigmaMapsCont.GetDefaultObject(searchMap.Data()));
1100 if (!etaSigmaPars) {
1101 // Try default object
1102 etaSigmaPars = dynamic_cast<TObjArray *>(etaSigmaMapsCont.GetDefaultObject(defaultObj.Data()));
1103 }
1104 }
1105 else {
1106 etaSigmaPars = dynamic_cast<TObjArray *>(etaSigmaMapsCont.GetObject(fRun, defaultObj.Data()));
1107 }
1108
1109 if (!etaSigmaPars) {
1110 AliError(Form("TPC eta sigma parametrisation not found for run %d -> Using old sigma parametrisation!!!", fRun));
1111 }
1112 else {
1113 TH2D* etaSigmaPar1Map = dynamic_cast<TH2D *>(etaSigmaPars->FindObject("sigmaPar1Map"));
1114 TNamed* sigmaPar0Info = dynamic_cast<TNamed *>(etaSigmaPars->FindObject("sigmaPar0"));
1115 Double_t sigmaPar0 = 0.0;
1116
1117 if (sigmaPar0Info) {
1118 TString sigmaPar0String = sigmaPar0Info->GetTitle();
1119 sigmaPar0 = sigmaPar0String.Atof();
1120 }
1121 else {
1122 // Something is weired because the object for parameter 0 could not be loaded -> New sigma parametrisation can not be used!
1123 etaSigmaPar1Map = 0x0;
1124 }
1125
1126 TH2D* etaSigmaPar1MapRefined = RefineHistoViaLinearInterpolation(etaSigmaPar1Map, refineFactorSigmaMapX, refineFactorSigmaMapY);
1127
1128
1129 if (etaSigmaPar1MapRefined) {
1130 if (!fTPCResponse.SetSigmaParams(etaSigmaPar1MapRefined, sigmaPar0)) {
1131 AliError(Form("Failed to set TPC eta sigma map for run %d -> Using old sigma parametrisation!!!", fRun));
1132 fTPCResponse.SetSigmaParams(0x0, 0);
1133 }
1134 else {
5a9dc560 1135 AliInfo(Form("Loaded TPC sigma correction map (refine factors %.2f/%.2f) from %s/COMMON/PID/data/TPCetaMaps.root: %s (MD5(map) = %s, sigmaPar0 = %f)",
1136 refineFactorSigmaMapX, refineFactorSigmaMapY, fOADBPath.Data(), fTPCResponse.GetSigmaPar1Map()->GetTitle(),
1137 GetChecksum(fTPCResponse.GetSigmaPar1Map()).Data(), sigmaPar0));
f84b18dd 1138 }
1139
1140 delete etaSigmaPar1MapRefined;
1141 }
1142 else {
1143 AliError(Form("Failed to set TPC eta sigma map for run %d (map was loaded, but couldn't be refined) -> Using old sigma parametrisation!!!",
1144 fRun));
1145 }
1146 }
1147 }
1148}
1149
4ec8e76d 1150//______________________________________________________________________________
1151void AliPIDResponse::SetTPCPidResponseMaster()
1152{
1153 //
1154 // Load the TPC pid response functions from the OADB
644666df 1155 // Load the TPC voltage maps from OADB
4ec8e76d 1156 //
09b50a42 1157 //don't load twice for the moment
1158 if (fArrPidResponseMaster) return;
1159
1160
4ec8e76d 1161 //reset the PID response functions
1162 delete fArrPidResponseMaster;
644666df 1163 fArrPidResponseMaster=NULL;
4ec8e76d 1164
1165 TString fileName(Form("%s/COMMON/PID/data/TPCPIDResponse.root", fOADBPath.Data()));
644666df 1166 TFile *f=NULL;
00a38d07 1167 if (!fCustomTPCpidResponse.IsNull()) fileName=fCustomTPCpidResponse;
4ec8e76d 1168
644666df 1169 TString fileNamePIDresponse(Form("%s/COMMON/PID/data/TPCPIDResponse.root", fOADBPath.Data()));
1170 f=TFile::Open(fileNamePIDresponse.Data());
ea235c90 1171 if (f && f->IsOpen() && !f->IsZombie()){
1172 fArrPidResponseMaster=dynamic_cast<TObjArray*>(f->Get("TPCPIDResponse"));
4ec8e76d 1173 }
ea235c90 1174 delete f;
644666df 1175
1176 TString fileNameVoltageMaps(Form("%s/COMMON/PID/data/TPCvoltageSettings.root", fOADBPath.Data()));
1177 f=TFile::Open(fileNameVoltageMaps.Data());
1178 if (f && f->IsOpen() && !f->IsZombie()){
1179 fOADBvoltageMaps=dynamic_cast<AliOADBContainer*>(f->Get("TPCvoltageSettings"));
1180 }
1181 delete f;
4ec8e76d 1182
1183 if (!fArrPidResponseMaster){
644666df 1184 AliFatal(Form("Could not retrieve the TPC pid response from: %s",fileNamePIDresponse.Data()));
4ec8e76d 1185 return;
1186 }
1187 fArrPidResponseMaster->SetOwner();
644666df 1188
1189 if (!fOADBvoltageMaps)
1190 {
1191 AliFatal(Form("Could not retrieve the TPC voltage maps from: %s",fileNameVoltageMaps.Data()));
1192 }
1193 fArrPidResponseMaster->SetOwner();
4ec8e76d 1194}
1195
1196//______________________________________________________________________________
1197void AliPIDResponse::SetTPCParametrisation()
1198{
1199 //
1200 // Change BB parametrisation for current run
1201 //
1202
12d3abbc 1203 //
1204 //reset old splines
1205 //
1206 fTPCResponse.ResetSplines();
1207
4ec8e76d 1208 if (fLHCperiod.IsNull()) {
12d3abbc 1209 AliError("No period set, not changing parametrisation");
4ec8e76d 1210 return;
1211 }
1212
1213 //
1214 // Set default parametrisations for data and MC
1215 //
1216
1217 //data type
1218 TString datatype="DATA";
1219 //in case of mc fRecoPass is per default 1
1220 if (fIsMC) {
87da0205 1221 if(!(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) datatype="MC";
539a5a59 1222 fRecoPass=1;
4ec8e76d 1223 }
f84b18dd 1224
4a527e08 1225 // period
1226 TString period=fLHCperiod;
87da0205 1227 if (fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))) period=fMCperiodTPC;
4a527e08 1228
f85a3764 1229 Int_t recopass = fRecoPass;
87da0205 1230 if(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC)) recopass = fRecoPassUser;
f85a3764 1231
1232 AliInfo(Form("Searching splines for: %s %s PASS%d %s",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
4a527e08 1233 Bool_t found=kFALSE;
4ec8e76d 1234 //
1235 //set the new PID splines
1236 //
4ec8e76d 1237 if (fArrPidResponseMaster){
4ec8e76d 1238 //for MC don't use period information
644666df 1239 //if (fIsMC) period="[A-Z0-9]*";
4ec8e76d 1240 //for MC use MC period information
644666df 1241 //pattern for the default entry (valid for all particles)
de678885 1242 TPRegexp reg(Form("TSPLINE3_%s_([A-Z]*)_%s_PASS%d_%s_MEAN(_*)([A-Z1-9]*)",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
644666df 1243
f85a3764 1244 //find particle id and gain scenario
644666df 1245 for (Int_t igainScenario=0; igainScenario<AliTPCPIDResponse::fgkNumberOfGainScenarios; igainScenario++)
1246 {
1247 TObject *grAll=NULL;
1248 TString gainScenario = AliTPCPIDResponse::GainScenarioName(igainScenario);
1249 gainScenario.ToUpper();
1250 //loop over entries and filter them
1251 for (Int_t iresp=0; iresp<fArrPidResponseMaster->GetEntriesFast();++iresp)
1252 {
1253 TObject *responseFunction=fArrPidResponseMaster->At(iresp);
1254 if (responseFunction==NULL) continue;
1255 TString responseName=responseFunction->GetName();
1256
1257 if (!reg.MatchB(responseName)) continue;
1258
1259 TObjArray *arr=reg.MatchS(responseName); if (!arr) continue;
1260 TObject* tmp=NULL;
1261 tmp=arr->At(1); if (!tmp) continue;
1262 TString particleName=tmp->GetName();
1263 tmp=arr->At(3); if (!tmp) continue;
1264 TString gainScenarioName=tmp->GetName();
1265 delete arr;
1266 if (particleName.IsNull()) continue;
1267 if (!grAll && particleName=="ALL" && gainScenarioName==gainScenario) grAll=responseFunction;
1268 else
1269 {
1270 for (Int_t ispec=0; ispec<(AliTPCPIDResponse::fgkNumberOfParticleSpecies); ++ispec)
1271 {
1272 TString particle=AliPID::ParticleName(ispec);
1273 particle.ToUpper();
1274 //std::cout<<responseName<<" "<<particle<<" "<<particleName<<" "<<gainScenario<<" "<<gainScenarioName<<std::endl;
1275 if ( particle == particleName && gainScenario == gainScenarioName )
1276 {
1277 fTPCResponse.SetResponseFunction( responseFunction,
1278 (AliPID::EParticleType)ispec,
1279 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1280 fTPCResponse.SetUseDatabase(kTRUE);
5a9dc560 1281 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,responseFunction->GetName(),
1282 GetChecksum((TSpline3*)responseFunction).Data()));
644666df 1283 found=kTRUE;
644666df 1284 break;
1285 }
4ec8e76d 1286 }
1287 }
1288 }
bf26ce58 1289
1290 // Retrieve responsefunction for pions - will (if available) be used for muons if there are no dedicated muon splines.
1291 // For light nuclei, try to set the proton spline, if no dedicated splines are available.
1292 // In both cases: Use default splines, if no dedicated splines and no pion/proton splines are available.
1293 TObject* responseFunctionPion = fTPCResponse.GetResponseFunction( (AliPID::EParticleType)AliPID::kPion,
1294 (AliTPCPIDResponse::ETPCgainScenario)igainScenario);
1295 TObject* responseFunctionProton = fTPCResponse.GetResponseFunction( (AliPID::EParticleType)AliPID::kProton,
1296 (AliTPCPIDResponse::ETPCgainScenario)igainScenario);
1297
1298 for (Int_t ispec=0; ispec<(AliTPCPIDResponse::fgkNumberOfParticleSpecies); ++ispec)
644666df 1299 {
bf26ce58 1300 if (!fTPCResponse.GetResponseFunction( (AliPID::EParticleType)ispec,
1301 (AliTPCPIDResponse::ETPCgainScenario)igainScenario))
644666df 1302 {
bf26ce58 1303 if (ispec == AliPID::kMuon) { // Muons
1304 if (responseFunctionPion) {
1305 fTPCResponse.SetResponseFunction( responseFunctionPion,
1306 (AliPID::EParticleType)ispec,
1307 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1308 fTPCResponse.SetUseDatabase(kTRUE);
5a9dc560 1309 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,responseFunctionPion->GetName(),
1310 GetChecksum((TSpline3*)responseFunctionPion).Data()));
bf26ce58 1311 found=kTRUE;
1312 }
1313 else if (grAll) {
1314 fTPCResponse.SetResponseFunction( grAll,
1315 (AliPID::EParticleType)ispec,
1316 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1317 fTPCResponse.SetUseDatabase(kTRUE);
5a9dc560 1318 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,grAll->GetName(),
1319 GetChecksum((TSpline3*)grAll).Data()));
bf26ce58 1320 found=kTRUE;
1321 }
1322 //else
1323 // AliError(Form("No splines found for muons (also no pion splines and no default splines) for gain scenario %d!", igainScenario));
1324 }
1325 else if (ispec >= AliPID::kSPECIES) { // Light nuclei
1326 if (responseFunctionProton) {
1327 fTPCResponse.SetResponseFunction( responseFunctionProton,
1328 (AliPID::EParticleType)ispec,
1329 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1330 fTPCResponse.SetUseDatabase(kTRUE);
5a9dc560 1331 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,responseFunctionProton->GetName(),
1332 GetChecksum((TSpline3*)responseFunctionProton).Data()));
bf26ce58 1333 found=kTRUE;
1334 }
1335 else if (grAll) {
644666df 1336 fTPCResponse.SetResponseFunction( grAll,
1337 (AliPID::EParticleType)ispec,
1338 (AliTPCPIDResponse::ETPCgainScenario)igainScenario );
1339 fTPCResponse.SetUseDatabase(kTRUE);
5a9dc560 1340 AliInfo(Form("Adding graph: %d %d - %s (MD5(spline) = %s)",ispec,igainScenario,grAll->GetName(),
1341 GetChecksum((TSpline3*)grAll).Data()));
644666df 1342 found=kTRUE;
bf26ce58 1343 }
1344 //else
1345 // AliError(Form("No splines found for species %d (also no proton splines and no default splines) for gain scenario %d!",
1346 // ispec, igainScenario));
644666df 1347 }
4ec8e76d 1348 }
1349 }
1350 }
1351 }
644666df 1352 else AliInfo("no fArrPidResponseMaster");
4a527e08 1353
1354 if (!found){
f85a3764 1355 AliError(Form("No splines found for: %s %s PASS%d %s",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
4a527e08 1356 }
644666df 1357
87da0205 1358
1359 //
22158469 1360 // Setup multiplicity correction (only used for non-pp collisions)
87da0205 1361 //
22158469
JW		 1362
1363 const Bool_t isPP = (fBeamType.CompareTo("PP") == 0);
1364
1365 // 2013 pPb data taking at low luminosity
1366 const Bool_t isPPb2013LowLuminosity = period.Contains("LHC13B") || period.Contains("LHC13C") || period.Contains("LHC13D");
1367 // PbPb 2010, period 10h.pass2
1368 //TODO Needs further development const Bool_t is10hpass2 = period.Contains("LHC10H") && recopass == 2;
1369
eafa51b5
JW		 1370
1371 // In case of MC without(!) tune on data activated for the TPC, don't use the multiplicity correction for the moment
1372 Bool_t isMCandNotTPCtuneOnData = fIsMC && !(fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC));
1373
22158469 1374 // If correction is available, but disabled (highly NOT recommended!), print warning
eafa51b5 1375 if (!fUseTPCMultiplicityCorrection && !isPP && !isMCandNotTPCtuneOnData) {
22158469
JW		 1376 //TODO: Needs further development if (is10hpass2 || isPPb2013LowLuminosity) {
1377 if (isPPb2013LowLuminosity) {
1378 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!");
1379 }
1380 }
1381
eafa51b5 1382 if (fUseTPCMultiplicityCorrection && !isPP && !isMCandNotTPCtuneOnData) {
87da0205 1383 AliInfo("Multiplicity correction enabled!");
1384
1385 //TODO After testing, load parameters from outside
22158469 1386 /*TODO no correction for MC
87da0205 1387 if (period.Contains("LHC11A10")) {//LHC11A10A
1388 AliInfo("Using multiplicity correction parameters for 11a10!");
1389 fTPCResponse.SetParameterMultiplicityCorrection(0, 6.90133e-06);
1390 fTPCResponse.SetParameterMultiplicityCorrection(1, -1.22123e-03);
1391 fTPCResponse.SetParameterMultiplicityCorrection(2, 1.80220e-02);
1392 fTPCResponse.SetParameterMultiplicityCorrection(3, 0.1);
1393 fTPCResponse.SetParameterMultiplicityCorrection(4, 6.45306e-03);
1394
1395 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, -2.85505e-07);
1396 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, -1.31911e-06);
1397 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1398
1399 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, -4.29665e-05);
1400 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, 1.37023e-02);
1401 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, -6.36337e-01);
1402 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 1.13479e-02);
1403 }
22158469
JW		 1404 else*/ if (isPPb2013LowLuminosity) {// 2013 pPb data taking at low luminosity
1405 AliInfo("Using multiplicity correction parameters for 13b.pass2 (at least also valid for 13{c,d} and pass 3)!");
ef7661fd 1406
1407 fTPCResponse.SetParameterMultiplicityCorrection(0, -5.906e-06);
1408 fTPCResponse.SetParameterMultiplicityCorrection(1, -5.064e-04);
1409 fTPCResponse.SetParameterMultiplicityCorrection(2, -3.521e-02);
1410 fTPCResponse.SetParameterMultiplicityCorrection(3, 2.469e-02);
1411 fTPCResponse.SetParameterMultiplicityCorrection(4, 0);
1412
1413 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, -5.32e-06);
1414 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, 1.177e-05);
1415 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1416
1417 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, 0.);
1418 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, 0.);
1419 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, 0.);
1420 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 0.);
1421
1422 /* Not too bad, but far from perfect in the details
1423 fTPCResponse.SetParameterMultiplicityCorrection(0, -6.27187e-06);
1424 fTPCResponse.SetParameterMultiplicityCorrection(1, -4.60649e-04);
1425 fTPCResponse.SetParameterMultiplicityCorrection(2, -4.26450e-02);
1426 fTPCResponse.SetParameterMultiplicityCorrection(3, 2.40590e-02);
1427 fTPCResponse.SetParameterMultiplicityCorrection(4, 0);
1428
1429 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, -5.338e-06);
1430 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, 1.220e-05);
1431 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1432
1433 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, 7.89237e-05);
1434 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, -1.30662e-02);
1435 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, 8.91548e-01);
1436 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 1.47931e-02);
1437 */
1438 }
22158469
JW		 1439 /*TODO: Needs further development
1440 else if (is10hpass2) {
87da0205 1441 AliInfo("Using multiplicity correction parameters for 10h.pass2!");
1442 fTPCResponse.SetParameterMultiplicityCorrection(0, 3.21636e-07);
1443 fTPCResponse.SetParameterMultiplicityCorrection(1, -6.65876e-04);
1444 fTPCResponse.SetParameterMultiplicityCorrection(2, 1.28786e-03);
1445 fTPCResponse.SetParameterMultiplicityCorrection(3, 1.47677e-02);
1446 fTPCResponse.SetParameterMultiplicityCorrection(4, 0);
1447
1448 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(0, 7.23591e-08);
1449 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(1, 2.7469e-06);
1450 fTPCResponse.SetParameterMultiplicityCorrectionTanTheta(2, -0.5);
1451
1452 fTPCResponse.SetParameterMultiplicitySigmaCorrection(0, -1.22590e-05);
1453 fTPCResponse.SetParameterMultiplicitySigmaCorrection(1, 6.88888e-03);
1454 fTPCResponse.SetParameterMultiplicitySigmaCorrection(2, -3.20788e-01);
1455 fTPCResponse.SetParameterMultiplicitySigmaCorrection(3, 1.07345e-02);
1456 }
22158469 1457 */
87da0205 1458 else {
1459 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));
1460 fUseTPCMultiplicityCorrection = kFALSE;
1461 fTPCResponse.ResetMultiplicityCorrectionFunctions();
1462 }
1463 }
1464 else {
1465 // Just set parameters such that overall correction factor is 1, i.e. no correction.
1466 // This is just a reasonable choice for the parameters for safety reasons. Disabling
1467 // the multiplicity correction will anyhow skip the calculation of the corresponding
1468 // correction factor inside THIS class. Nevertheless, experts can access the TPCPIDResponse
1469 // directly and use it for calculations - which should still give valid results, even if
1470 // the multiplicity correction is explicitely enabled in such expert calls.
1471
eafa51b5
JW		 1472 TString reasonForDisabling = "requested by user";
1473 if (fUseTPCMultiplicityCorrection) {
1474 if (isPP)
1475 reasonForDisabling = "pp collisions";
1476 else
1477 reasonForDisabling = "MC w/o tune on data";
1478 }
1479
87da0205 1480 AliInfo(Form("Multiplicity correction %sdisabled (%s)!", fUseTPCMultiplicityCorrection ? "automatically " : "",
eafa51b5 1481 reasonForDisabling.Data()));
87da0205 1482
1483 fUseTPCMultiplicityCorrection = kFALSE;
1484 fTPCResponse.ResetMultiplicityCorrectionFunctions();
1485 }
1486
22158469
JW		 1487 if (fUseTPCMultiplicityCorrection) {
1488 for (Int_t i = 0; i <= 4 + 1; i++) {
1489 AliInfo(Form("parMultCorr: %d, %e", i, fTPCResponse.GetMultiplicityCorrectionFunction()->GetParameter(i)));
1490 }
1491 for (Int_t j = 0; j <= 2 + 1; j++) {
1492 AliInfo(Form("parMultCorrTanTheta: %d, %e", j, fTPCResponse.GetMultiplicityCorrectionFunctionTanTheta()->GetParameter(j)));
1493 }
1494 for (Int_t j = 0; j <= 3 + 1; j++) {
1495 AliInfo(Form("parMultSigmaCorr: %d, %e", j, fTPCResponse.GetMultiplicitySigmaCorrectionFunction()->GetParameter(j)));
1496 }
87da0205 1497 }
1498
4ec8e76d 1499 //
87da0205 1500 // Setup old resolution parametrisation
4ec8e76d 1501 //
1502
1503 //default
1504 fTPCResponse.SetSigma(3.79301e-03, 2.21280e+04);
1505
3b3bf053 1506 if (fRun>=122195){ //LHC10d
4ec8e76d 1507 fTPCResponse.SetSigma(2.30176e-02, 5.60422e+02);
1508 }
3b3bf053 1509
1510 if (fRun>=170719){ // LHC12a
1511 fTPCResponse.SetSigma(2.95714e-03, 1.01953e+05);
1512 }
1513
1514 if (fRun>=177312){ // LHC12b
1515 fTPCResponse.SetSigma(3.74633e-03, 7.11829e+04 );
1516 }
1517
1518 if (fRun>=186346){ // LHC12e
723c4874 1519 fTPCResponse.SetSigma(8.62022e-04, 9.08156e+05);
1520 }
1521
23425eb2 1522 if (fArrPidResponseMaster)
f85a3764 1523 fResolutionCorrection=(TF1*)fArrPidResponseMaster->FindObject(Form("TF1_%s_ALL_%s_PASS%d_%s_SIGMA",datatype.Data(),period.Data(),recopass,fBeamType.Data()));
4ec8e76d 1524
5a9dc560 1525 if (fResolutionCorrection) AliInfo(Form("Setting multiplicity correction function: %s (MD5(corr function) = %s)",
1526 fResolutionCorrection->GetName(), GetChecksum(fResolutionCorrection).Data()));
644666df 1527
1528 //read in the voltage map
12d3abbc 1529 TVectorF* gsm = 0x0;
1530 if (fOADBvoltageMaps) gsm=dynamic_cast<TVectorF*>(fOADBvoltageMaps->GetObject(fRun));
644666df 1531 if (gsm)
1532 {
1533 fTPCResponse.SetVoltageMap(*gsm);
1534 TString vals;
1535 AliInfo(Form("Reading the voltage map for run %d\n",fRun));
1536 vals="IROC A: "; for (Int_t i=0; i<18; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1537 AliInfo(vals.Data());
1538 vals="IROC C: "; for (Int_t i=18; i<36; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1539 AliInfo(vals.Data());
1540 vals="OROC A: "; for (Int_t i=36; i<54; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1541 AliInfo(vals.Data());
1542 vals="OROC C: "; for (Int_t i=54; i<72; i++){vals+=Form("%.2f ",(*gsm)[i]);}
1543 AliInfo(vals.Data());
1544 }
1545 else AliInfo("no voltage map, ideal default assumed");
4ec8e76d 1546}
1547
ea235c90 1548//______________________________________________________________________________
1549void AliPIDResponse::SetTRDPidResponseMaster()
1550{
1551 //
1552 // Load the TRD pid params and references from the OADB
1553 //
db0e2c5f 1554 if(fTRDPIDResponseObject) return;
53d016dc 1555 AliOADBContainer contParams("contParams");
1556
db0e2c5f 1557 Int_t statusResponse = contParams.InitFromFile(Form("%s/COMMON/PID/data/TRDPIDResponse.root", fOADBPath.Data()), "AliTRDPIDResponseObject");
1558 if(statusResponse){
1559 AliError("Failed initializing PID Response Object from OADB");
59a8e853 1560 } else {
db0e2c5f 1561 AliInfo(Form("Loading TRD Response from %s/COMMON/PID/data/TRDPIDResponse.root", fOADBPath.Data()));
1562 fTRDPIDResponseObject = dynamic_cast<AliTRDPIDResponseObject *>(contParams.GetObject(fRun));
1563 if(!fTRDPIDResponseObject){
1564 AliError(Form("TRD Response not found in run %d", fRun));
59a8e853 1565 }
1566 }
ea235c90 1567}
1568
1569//______________________________________________________________________________
1570void AliPIDResponse::InitializeTRDResponse(){
1571 //
1572 // Set PID Params and references to the TRD PID response
1573 //
db0e2c5f 1574 fTRDResponse.SetPIDResponseObject(fTRDPIDResponseObject);
9c499471 1575 fTRDResponse.SetdEdxParams(fTRDdEdxParams);
f2762b1c 1576}
1577
bd58d4b9 1578//______________________________________________________________________________
1579void AliPIDResponse::SetTRDSlices(UInt_t TRDslicesForPID[2],AliTRDPIDResponse::ETRDPIDMethod method) const{
1580
72abc110 1581 if(fLHCperiod.Contains("LHC10D") || fLHCperiod.Contains("LHC10E")){
bd58d4b9 1582 // backward compatibility for setting with 8 slices
1583 TRDslicesForPID[0] = 0;
1584 TRDslicesForPID[1] = 7;
f2762b1c 1585 }
bd58d4b9 1586 else{
1587 if(method==AliTRDPIDResponse::kLQ1D){
1588 TRDslicesForPID[0] = 0; // first Slice contains normalized dEdx
1589 TRDslicesForPID[1] = 0;
1590 }
1591 if(method==AliTRDPIDResponse::kLQ2D){
1592 TRDslicesForPID[0] = 1;
1593 TRDslicesForPID[1] = 7;
1594 }
db0e2c5f 1595 }
bd58d4b9 1596 AliDebug(1,Form("Slice Range set to %d - %d",TRDslicesForPID[0],TRDslicesForPID[1]));
ea235c90 1597}
1598
9c499471 1599//______________________________________________________________________________
1600void AliPIDResponse::SetTRDdEdxParams()
1601{
1602 if(fTRDdEdxParams) return;
1603
1604 const TString containerName = "TRDdEdxParamsContainer";
1605 AliOADBContainer cont(containerName.Data());
1606
1607 const TString filePathNamePackage=Form("%s/COMMON/PID/data/TRDdEdxParams.root", fOADBPath.Data());
1608
1609 const Int_t statusCont = cont.InitFromFile(filePathNamePackage.Data(), cont.GetName());
1610 if (statusCont){
1611 AliFatal("Failed initializing settings from OADB");
1612 }
1613 else{
2ddf4abb 1614 AliInfo(Form("Loading %s from %s\n", cont.GetName(), filePathNamePackage.Data()));
9c499471 1615
1616 fTRDdEdxParams = (AliTRDdEdxParams*)(cont.GetObject(fRun, "default"));
1617 //fTRDdEdxParams->Print();
1618
1619 if(!fTRDdEdxParams){
1620 AliError(Form("TRD dEdx Params default not found"));
1621 }
1622 }
1623}
1624
b79db598 1625//______________________________________________________________________________
1626void AliPIDResponse::SetTOFPidResponseMaster()
1627{
1628 //
1629 // Load the TOF pid params from the OADB
1630 //
00a38d07 1631
1632 if (fTOFPIDParams) delete fTOFPIDParams;
644666df 1633 fTOFPIDParams=NULL;
00a38d07 1634
b79db598 1635 TFile *oadbf = new TFile(Form("%s/COMMON/PID/data/TOFPIDParams.root",fOADBPath.Data()));
00a38d07 1636 if (oadbf && oadbf->IsOpen()) {
b79db598 1637 AliInfo(Form("Loading TOF Params from %s/COMMON/PID/data/TOFPIDParams.root", fOADBPath.Data()));
1638 AliOADBContainer *oadbc = (AliOADBContainer *)oadbf->Get("TOFoadb");
00a38d07 1639 if (oadbc) fTOFPIDParams = dynamic_cast<AliTOFPIDParams *>(oadbc->GetObject(fRun,"TOFparams"));
b79db598 1640 oadbf->Close();
1641 delete oadbc;
b79db598 1642 }
1643 delete oadbf;
1644
00a38d07 1645 if (!fTOFPIDParams) AliFatal("TOFPIDParams could not be retrieved");
1646}
b79db598 1647
1648//______________________________________________________________________________
1649void AliPIDResponse::InitializeTOFResponse(){
1650 //
1651 // Set PID Params to the TOF PID response
00a38d07 1652 //
1653
1654 AliInfo("TOF PID Params loaded from OADB");
1655 AliInfo(Form(" TOF resolution %5.2f [ps]",fTOFPIDParams->GetTOFresolution()));
1656 AliInfo(Form(" StartTime method %d",fTOFPIDParams->GetStartTimeMethod()));
1657 AliInfo(Form(" TOF res. mom. params: %5.2f %5.2f %5.2f %5.2f",
1658 fTOFPIDParams->GetSigParams(0),fTOFPIDParams->GetSigParams(1),fTOFPIDParams->GetSigParams(2),fTOFPIDParams->GetSigParams(3)));
c53e310b 1659 AliInfo(Form(" Fraction of tracks within gaussian behaviour: %6.4f",fTOFPIDParams->GetTOFtail()));
1660 AliInfo(Form(" MC: Fraction of tracks (percentage) to cut to fit matching in data: %6.2f%%",fTOFPIDParams->GetTOFmatchingLossMC()));
1661 AliInfo(Form(" MC: Fraction of random hits (percentage) to add to fit mismatch in data: %6.2f%%",fTOFPIDParams->GetTOFadditionalMismForMC()));
1662 AliInfo(Form(" Start Time Offset %6.2f ps",fTOFPIDParams->GetTOFtimeOffset()));
1663
b79db598 1664 for (Int_t i=0;i<4;i++) {
1665 fTOFResponse.SetTrackParameter(i,fTOFPIDParams->GetSigParams(i));
1666 }
1667 fTOFResponse.SetTimeResolution(fTOFPIDParams->GetTOFresolution());
1668
78cbd205 1669 AliInfo("TZERO resolution loaded from ESDrun/AODheader");
1670 Float_t t0Spread[4];
1671 for (Int_t i=0;i<4;i++) t0Spread[i]=fCurrentEvent->GetT0spread(i);
1672 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]));
1673 Float_t a = t0Spread[1]*t0Spread[1]-t0Spread[0]*t0Spread[0]+t0Spread[3]*t0Spread[3];
1674 Float_t c = t0Spread[2]*t0Spread[2]-t0Spread[0]*t0Spread[0]+t0Spread[3]*t0Spread[3];
1675 if ( (t0Spread[0] > 50. && t0Spread[0] < 400.) && (a > 0.) && (c>0.)) {
1676 fResT0AC=t0Spread[3];
1677 fResT0A=TMath::Sqrt(a);
1678 fResT0C=TMath::Sqrt(c);
1679 } else {
1680 AliInfo(" TZERO spreads not present or inconsistent, loading default");
1681 fResT0A=75.;
1682 fResT0C=65.;
1683 fResT0AC=55.;
1684 }
1685 AliInfo(Form(" TZERO resolution set to: T0A: %f [ps] T0C: %f [ps] T0AC %f [ps]",fResT0A,fResT0C,fResT0AC));
1686
b79db598 1687}
1688
567624b5 1689//______________________________________________________________________________
1690void AliPIDResponse::SetHMPIDPidResponseMaster()
1691{
1692 //
1693 // Load the HMPID pid params from the OADB
1694 //
1695
1696 if (fHMPIDPIDParams) delete fHMPIDPIDParams;
1697 fHMPIDPIDParams=NULL;
1698
b2f22270 1699 TFile *oadbf;
1700 if(!fIsMC) oadbf = new TFile(Form("%s/COMMON/PID/data/HMPIDPIDParams.root",fOADBPath.Data()));
1701 else oadbf = new TFile(Form("%s/COMMON/PID/MC/HMPIDPIDParams.root",fOADBPath.Data()));
567624b5 1702 if (oadbf && oadbf->IsOpen()) {
1703 AliInfo(Form("Loading HMPID Params from %s/COMMON/PID/data/HMPIDPIDParams.root", fOADBPath.Data()));
1704 AliOADBContainer *oadbc = (AliOADBContainer *)oadbf->Get("HMPoadb");
1705 if (oadbc) fHMPIDPIDParams = dynamic_cast<AliHMPIDPIDParams *>(oadbc->GetObject(fRun,"HMPparams"));
1706 oadbf->Close();
1707 delete oadbc;
1708 }
1709 delete oadbf;
1710
1711 if (!fHMPIDPIDParams) AliFatal("HMPIDPIDParams could not be retrieved");
1712}
1713
1714//______________________________________________________________________________
1715void AliPIDResponse::InitializeHMPIDResponse(){
1716 //
1717 // Set PID Params to the HMPID PID response
1718 //
1719
1720 fHMPIDResponse.SetRefIndexArray(fHMPIDPIDParams->GetHMPIDrefIndex());
1721}
b79db598 1722
1c9d11be 1723//______________________________________________________________________________
239fe91c 1724Bool_t AliPIDResponse::IdentifiedAsElectronTRD(const AliVTrack *vtrack,Double_t efficiencyLevel,Double_t centrality,AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const {
1725 // old function for compatibility
1726 Int_t ntracklets=0;
1727 return IdentifiedAsElectronTRD(vtrack,ntracklets,efficiencyLevel,centrality,PIDmethod);
1728}
1729
1730//______________________________________________________________________________
1731Bool_t AliPIDResponse::IdentifiedAsElectronTRD(const AliVTrack *vtrack, Int_t &ntracklets,Double_t efficiencyLevel,Double_t centrality,AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const {
ea235c90 1732 //
1733 // Check whether track is identified as electron under a given electron efficiency hypothesis
bd58d4b9 1734 //
239fe91c 1735 // ntracklets is the number of tracklets that has been used to calculate the PID signal
bd58d4b9 1736
ea235c90 1737 Double_t probs[AliPID::kSPECIES];
ea235c90 1738
239fe91c 1739 ntracklets =CalculateTRDResponse(vtrack,probs,PIDmethod);
1740
99e9d5ec 1741 // Take mean of the TRD momenta in the given tracklets
1742 Float_t p = 0, trdmomenta[AliVTrack::kTRDnPlanes];
1743 Int_t nmomenta = 0;
ea235c90 1744 for(Int_t iPl=0;iPl<AliVTrack::kTRDnPlanes;iPl++){
1745 if(vtrack->GetTRDmomentum(iPl) > 0.){
99e9d5ec 1746 trdmomenta[nmomenta++] = vtrack->GetTRDmomentum(iPl);
ea235c90 1747 }
1748 }
99e9d5ec 1749 p = TMath::Mean(nmomenta, trdmomenta);
ea235c90 1750
bd58d4b9 1751 return fTRDResponse.IdentifiedAsElectron(ntracklets, probs, p, efficiencyLevel,centrality,PIDmethod);
ea235c90 1752}
1753
b2138b40 1754//______________________________________________________________________________
1755void AliPIDResponse::SetEMCALPidResponseMaster()
1756{
1757 //
1758 // Load the EMCAL pid response functions from the OADB
1759 //
1760 TObjArray* fEMCALPIDParamsRun = NULL;
1761 TObjArray* fEMCALPIDParamsPass = NULL;
1762
1763 if(fEMCALPIDParams) return;
1764 AliOADBContainer contParams("contParams");
1765
1766 Int_t statusPars = contParams.InitFromFile(Form("%s/COMMON/PID/data/EMCALPIDParams.root", fOADBPath.Data()), "AliEMCALPIDParams");
1767 if(statusPars){
1768 AliError("Failed initializing PID Params from OADB");
1769 }
1770 else {
1771 AliInfo(Form("Loading EMCAL Params from %s/COMMON/PID/data/EMCALPIDParams.root", fOADBPath.Data()));
1772
1773 fEMCALPIDParamsRun = dynamic_cast<TObjArray *>(contParams.GetObject(fRun));
1774 if(fEMCALPIDParamsRun) fEMCALPIDParamsPass = dynamic_cast<TObjArray *>(fEMCALPIDParamsRun->FindObject(Form("pass%d",fRecoPass)));
1775 if(fEMCALPIDParamsPass) fEMCALPIDParams = dynamic_cast<TObjArray *>(fEMCALPIDParamsPass->FindObject(Form("EMCALPIDParams_Particles")));
1776
1777 if(!fEMCALPIDParams){
f8d39067 1778 AliInfo(Form("EMCAL Params not found in run %d pass %d", fRun, fRecoPass));
1f631618 1779 AliInfo("Will take the standard LHC11d instead ...");
b2138b40 1780
1f631618 1781 fEMCALPIDParamsRun = dynamic_cast<TObjArray *>(contParams.GetObject(156477));
1782 if(fEMCALPIDParamsRun) fEMCALPIDParamsPass = dynamic_cast<TObjArray *>(fEMCALPIDParamsRun->FindObject(Form("pass%d",1)));
b2138b40 1783 if(fEMCALPIDParamsPass) fEMCALPIDParams = dynamic_cast<TObjArray *>(fEMCALPIDParamsPass->FindObject(Form("EMCALPIDParams_Particles")));
1784
1785 if(!fEMCALPIDParams){
1f631618 1786 AliError(Form("DEFAULT EMCAL Params (LHC11d) not found in file %s/COMMON/PID/data/EMCALPIDParams.root", fOADBPath.Data()));
b2138b40 1787 }
1788 }
1789 }
1790}
1791
1792//______________________________________________________________________________
1793void AliPIDResponse::InitializeEMCALResponse(){
1794 //
1795 // Set PID Params to the EMCAL PID response
1796 //
1797 fEMCALResponse.SetPIDParams(fEMCALPIDParams);
1798
1799}
00a38d07 1800
1c9d11be 1801//______________________________________________________________________________
1802void AliPIDResponse::FillTrackDetectorPID(const AliVTrack *track, EDetector detector) const
00a38d07 1803{
1804 //
1805 // create detector PID information and setup the transient pointer in the track
1806 //
1c9d11be 1807
1808 // check if detector number is inside accepted range
1809 if (detector == kNdetectors) return;
1810
1811 // get detector pid
1812 AliDetectorPID *detPID=const_cast<AliDetectorPID*>(track->GetDetectorPID());
1813 if (!detPID) {
1814 detPID=new AliDetectorPID;
1815 (const_cast<AliVTrack*>(track))->SetDetectorPID(detPID);
1816 }
1817
1818 //check if values exist
355b831b 1819 if (detPID->HasRawProbability(detector) && detPID->HasNumberOfSigmas(detector)) return;
00a38d07 1820
1821 //TODO: which particles to include? See also the loops below...
1822 Double_t values[AliPID::kSPECIESC]={0};
1c9d11be 1823
355b831b 1824 //probabilities
1825 EDetPidStatus status=GetComputePIDProbability(detector,track,AliPID::kSPECIESC,values);
1826 detPID->SetRawProbability(detector, values, (Int_t)AliPID::kSPECIESC, status);
1827
1c9d11be 1828 //nsigmas
1829 for (Int_t ipart=0; ipart<AliPID::kSPECIESC; ++ipart)
1830 values[ipart]=GetNumberOfSigmas(detector,track,(AliPID::EParticleType)ipart);
355b831b 1831 // the pid status is the same for probabilities and nSigmas, so it is
1832 // fine to use the one from the probabilities also here
1833 detPID->SetNumberOfSigmas(detector, values, (Int_t)AliPID::kSPECIESC, status);
1c9d11be 1834
1c9d11be 1835}
1836
1837//______________________________________________________________________________
1838void AliPIDResponse::FillTrackDetectorPID()
1839{
1840 //
1841 // create detector PID information and setup the transient pointer in the track
1842 //
1843
1844 if (!fCurrentEvent) return;
00a38d07 1845
1846 for (Int_t itrack=0; itrack<fCurrentEvent->GetNumberOfTracks(); ++itrack){
1847 AliVTrack *track=dynamic_cast<AliVTrack*>(fCurrentEvent->GetTrack(itrack));
1848 if (!track) continue;
1849
00a38d07 1850 for (Int_t idet=0; idet<kNdetectors; ++idet){
1c9d11be 1851 FillTrackDetectorPID(track, (EDetector)idet);
00a38d07 1852 }
00a38d07 1853 }
1854}
1855
1c9d11be 1856//______________________________________________________________________________
5f8db5fe 1857void AliPIDResponse::SetTOFResponse(AliVEvent *vevent,EStartTimeType_t option){
1858 //
1859 // Set TOF response function
1860 // Input option for event_time used
1861 //
c53e310b 1862
5f8db5fe 1863 Float_t t0spread = 0.; //vevent->GetEventTimeSpread();
1864 if(t0spread < 10) t0spread = 80;
1865
c53e310b 1866 // T0-FILL and T0-TO offset (because of TOF misallignment
1867 Float_t starttimeoffset = 0;
1868 if(fTOFPIDParams && !(fIsMC)) starttimeoffset=fTOFPIDParams->GetTOFtimeOffset();
b3f687a1 1869 if(fTOFPIDParams){
1870 fTOFtail = fTOFPIDParams->GetTOFtail();
1871 GetTOFResponse().SetTOFtail(fTOFtail);
1872 }
5f8db5fe 1873
c53e310b 1874 // T0 from TOF algorithm
5f8db5fe 1875 Bool_t flagT0TOF=kFALSE;
1876 Bool_t flagT0T0=kFALSE;
1877 Float_t *startTime = new Float_t[fTOFResponse.GetNmomBins()];
1878 Float_t *startTimeRes = new Float_t[fTOFResponse.GetNmomBins()];
1879 Int_t *startTimeMask = new Int_t[fTOFResponse.GetNmomBins()];
1880
1881 // T0-TOF arrays
1882 Float_t *estimatedT0event = new Float_t[fTOFResponse.GetNmomBins()];
1883 Float_t *estimatedT0resolution = new Float_t[fTOFResponse.GetNmomBins()];
1884 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1885 estimatedT0event[i]=0.0;
1886 estimatedT0resolution[i]=0.0;
1887 startTimeMask[i] = 0;
1888 }
1889
78cbd205 1890 Float_t resT0A=fResT0A;
1891 Float_t resT0C=fResT0C;
1892 Float_t resT0AC=fResT0AC;
5f8db5fe 1893 if(vevent->GetT0TOF()){ // check if T0 detector information is available
1894 flagT0T0=kTRUE;
1895 }
1896
1897
1898 AliTOFHeader *tofHeader = (AliTOFHeader*)vevent->GetTOFHeader();
1899
1900 if (tofHeader) { // read global info and T0-TOF
1901 fTOFResponse.SetTimeResolution(tofHeader->GetTOFResolution());
1902 t0spread = tofHeader->GetT0spread(); // read t0 sprad
1903 if(t0spread < 10) t0spread = 80;
1904
1905 flagT0TOF=kTRUE;
1906 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){ // read T0-TOF default value
1907 startTime[i]=tofHeader->GetDefaultEventTimeVal();
1908 startTimeRes[i]=tofHeader->GetDefaultEventTimeRes();
1909 if(startTimeRes[i] < 1.e-5) startTimeRes[i] = t0spread;
c53e310b 1910
1911 if(startTimeRes[i] > t0spread - 10 && TMath::Abs(startTime[i]) < 0.001) startTime[i] = -starttimeoffset; // apply offset for T0-fill
5f8db5fe 1912 }
1913
1914 TArrayI *ibin=(TArrayI*)tofHeader->GetNvalues();
1915 TArrayF *t0Bin=(TArrayF*)tofHeader->GetEventTimeValues();
1916 TArrayF *t0ResBin=(TArrayF*)tofHeader->GetEventTimeRes();
1917 for(Int_t j=0;j < tofHeader->GetNbins();j++){ // fill T0-TOF in p-bins
1918 Int_t icurrent = (Int_t)ibin->GetAt(j);
1919 startTime[icurrent]=t0Bin->GetAt(j);
1920 startTimeRes[icurrent]=t0ResBin->GetAt(j);
1921 if(startTimeRes[icurrent] < 1.e-5) startTimeRes[icurrent] = t0spread;
c53e310b 1922 if(startTimeRes[icurrent] > t0spread - 10 && TMath::Abs(startTime[icurrent]) < 0.001) startTime[icurrent] = -starttimeoffset; // apply offset for T0-fill
5f8db5fe 1923 }
1924 }
1925
1926 // for cut of 3 sigma on t0 spread
1927 Float_t t0cut = 3 * t0spread;
1928 if(t0cut < 500) t0cut = 500;
1929
1930 if(option == kFILL_T0){ // T0-FILL is used
1931 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
c53e310b 1932 estimatedT0event[i]=0.0-starttimeoffset;
5f8db5fe 1933 estimatedT0resolution[i]=t0spread;
1934 }
1935 fTOFResponse.SetT0event(estimatedT0event);
1936 fTOFResponse.SetT0resolution(estimatedT0resolution);
1937 }
1938
1939 if(option == kTOF_T0){ // T0-TOF is used when available (T0-FILL otherwise) from ESD
1940 if(flagT0TOF){
1941 fTOFResponse.SetT0event(startTime);
1942 fTOFResponse.SetT0resolution(startTimeRes);
1943 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1944 if(startTimeRes[i]<t0spread) startTimeMask[i]=1;
1945 fTOFResponse.SetT0binMask(i,startTimeMask[i]);
1946 }
1947 }
1948 else{
1949 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
c53e310b 1950 estimatedT0event[i]=0.0-starttimeoffset;
5f8db5fe 1951 estimatedT0resolution[i]=t0spread;
1952 fTOFResponse.SetT0binMask(i,startTimeMask[i]);
1953 }
1954 fTOFResponse.SetT0event(estimatedT0event);
1955 fTOFResponse.SetT0resolution(estimatedT0resolution);
1956 }
1957 }
1958 else if(option == kBest_T0){ // T0-T0 or T0-TOF are used when available (T0-FILL otherwise) from ESD
1959 Float_t t0AC=-10000;
1960 Float_t t0A=-10000;
1961 Float_t t0C=-10000;
1962 if(flagT0T0){
c53e310b 1963 t0A= vevent->GetT0TOF()[1] - starttimeoffset;
1964 t0C= vevent->GetT0TOF()[2] - starttimeoffset;
f84b18dd 1965 // t0AC= vevent->GetT0TOF()[0];
c53e310b 1966 t0AC= t0A/resT0A/resT0A + t0C/resT0C/resT0C;
1967 resT0AC= TMath::Sqrt(1./resT0A/resT0A + 1./resT0C/resT0C);
1968 t0AC /= resT0AC*resT0AC;
5f8db5fe 1969 }
1970
1971 Float_t t0t0Best = 0;
1972 Float_t t0t0BestRes = 9999;
1973 Int_t t0used=0;
1974 if(TMath::Abs(t0A) < t0cut && TMath::Abs(t0C) < t0cut && TMath::Abs(t0C-t0A) < 500){
1975 t0t0Best = t0AC;
1976 t0t0BestRes = resT0AC;
1977 t0used=6;
1978 }
1979 else if(TMath::Abs(t0C) < t0cut){
1980 t0t0Best = t0C;
1981 t0t0BestRes = resT0C;
1982 t0used=4;
1983 }
1984 else if(TMath::Abs(t0A) < t0cut){
1985 t0t0Best = t0A;
1986 t0t0BestRes = resT0A;
1987 t0used=2;
1988 }
1989
1990 if(flagT0TOF){ // if T0-TOF info is available
1991 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
1992 if(t0t0BestRes < 999){
1993 if(startTimeRes[i] < t0spread){
1994 Double_t wtot = 1./startTimeRes[i]/startTimeRes[i] + 1./t0t0BestRes/t0t0BestRes;
1995 Double_t t0best = startTime[i]/startTimeRes[i]/startTimeRes[i] + t0t0Best/t0t0BestRes/t0t0BestRes;
1996 estimatedT0event[i]=t0best / wtot;
1997 estimatedT0resolution[i]=1./TMath::Sqrt(wtot);
1998 startTimeMask[i] = t0used+1;
1999 }
2000 else {
2001 estimatedT0event[i]=t0t0Best;
2002 estimatedT0resolution[i]=t0t0BestRes;
2003 startTimeMask[i] = t0used;
2004 }
2005 }
2006 else{
2007 estimatedT0event[i]=startTime[i];
2008 estimatedT0resolution[i]=startTimeRes[i];
2009 if(startTimeRes[i]<t0spread) startTimeMask[i]=1;
2010 }
2011 fTOFResponse.SetT0binMask(i,startTimeMask[i]);
2012 }
2013 fTOFResponse.SetT0event(estimatedT0event);
2014 fTOFResponse.SetT0resolution(estimatedT0resolution);
2015 }
2016 else{ // if no T0-TOF info is available
2017 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
2018 fTOFResponse.SetT0binMask(i,t0used);
2019 if(t0t0BestRes < 999){
2020 estimatedT0event[i]=t0t0Best;
2021 estimatedT0resolution[i]=t0t0BestRes;
2022 }
2023 else{
c53e310b 2024 estimatedT0event[i]=0.0-starttimeoffset;
5f8db5fe 2025 estimatedT0resolution[i]=t0spread;
2026 }
2027 }
2028 fTOFResponse.SetT0event(estimatedT0event);
2029 fTOFResponse.SetT0resolution(estimatedT0resolution);
2030 }
2031 }
2032
2033 else if(option == kT0_T0){ // T0-T0 is used when available (T0-FILL otherwise)
2034 Float_t t0AC=-10000;
2035 Float_t t0A=-10000;
2036 Float_t t0C=-10000;
2037 if(flagT0T0){
c53e310b 2038 t0A= vevent->GetT0TOF()[1] - starttimeoffset;
2039 t0C= vevent->GetT0TOF()[2] - starttimeoffset;
f84b18dd 2040 // t0AC= vevent->GetT0TOF()[0];
c53e310b 2041 t0AC= t0A/resT0A/resT0A + t0C/resT0C/resT0C;
2042 resT0AC= TMath::Sqrt(1./resT0A/resT0A + 1./resT0C/resT0C);
2043 t0AC /= resT0AC*resT0AC;
5f8db5fe 2044 }
2045
2046 if(TMath::Abs(t0A) < t0cut && TMath::Abs(t0C) < t0cut && TMath::Abs(t0C-t0A) < 500){
2047 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
2048 estimatedT0event[i]=t0AC;
2049 estimatedT0resolution[i]=resT0AC;
2050 fTOFResponse.SetT0binMask(i,6);
2051 }
2052 }
2053 else if(TMath::Abs(t0C) < t0cut){
2054 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
2055 estimatedT0event[i]=t0C;
2056 estimatedT0resolution[i]=resT0C;
2057 fTOFResponse.SetT0binMask(i,4);
2058 }
2059 }
2060 else if(TMath::Abs(t0A) < t0cut){
2061 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
2062 estimatedT0event[i]=t0A;
2063 estimatedT0resolution[i]=resT0A;
2064 fTOFResponse.SetT0binMask(i,2);
2065 }
2066 }
2067 else{
2068 for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
c53e310b 2069 estimatedT0event[i]= 0.0 - starttimeoffset;
5f8db5fe 2070 estimatedT0resolution[i]=t0spread;
2071 fTOFResponse.SetT0binMask(i,0);
2072 }
2073 }
2074 fTOFResponse.SetT0event(estimatedT0event);
2075 fTOFResponse.SetT0resolution(estimatedT0resolution);
2076 }
c53e310b 2077
5f8db5fe 2078 delete [] startTime;
2079 delete [] startTimeRes;
2080 delete [] startTimeMask;
2081 delete [] estimatedT0event;
2082 delete [] estimatedT0resolution;
2083}
1c9d11be 2084
2085//______________________________________________________________________________
2086// private non cached versions of the PID calculation
2087//
2088
2089
2090//______________________________________________________________________________
355b831b 2091Float_t AliPIDResponse::GetNumberOfSigmas(EDetector detector, const AliVParticle *vtrack, AliPID::EParticleType type) const
1c9d11be 2092{
2093 //
2094 // NumberOfSigmas for 'detCode'
2095 //
355b831b 2096
2097 const AliVTrack *track=static_cast<const AliVTrack*>(vtrack);
1c9d11be 2098
355b831b 2099 switch (detector){
567624b5 2100 case kITS: return GetNumberOfSigmasITS(track, type); break;
2101 case kTPC: return GetNumberOfSigmasTPC(track, type); break;
5cd0300d 2102 case kTRD: return GetNumberOfSigmasTRD(track, type); break;
567624b5 2103 case kTOF: return GetNumberOfSigmasTOF(track, type); break;
2104 case kHMPID: return GetNumberOfSigmasHMPID(track, type); break;
1c9d11be 2105 case kEMCAL: return GetNumberOfSigmasEMCAL(track, type); break;
2106 default: return -999.;
2107 }
1c9d11be 2108
355b831b 2109 return -999.;
2110}
1c9d11be 2111
2112//______________________________________________________________________________
2113Float_t AliPIDResponse::GetNumberOfSigmasITS(const AliVParticle *vtrack, AliPID::EParticleType type) const
2114{
2115 //
2116 // Calculate the number of sigmas in the ITS
2117 //
2118
2119 AliVTrack *track=(AliVTrack*)vtrack;
355b831b 2120
2121 const EDetPidStatus pidStatus=GetITSPIDStatus(track);
2122 if (pidStatus!=kDetPidOk) return -999.;
355b831b 2123
567624b5 2124 return fITSResponse.GetNumberOfSigmas(track,type);
1c9d11be 2125}
2126
2127//______________________________________________________________________________
2128Float_t AliPIDResponse::GetNumberOfSigmasTPC(const AliVParticle *vtrack, AliPID::EParticleType type) const
2129{
2130 //
2131 // Calculate the number of sigmas in the TPC
2132 //
2133
2134 AliVTrack *track=(AliVTrack*)vtrack;
355b831b 2135
2136 const EDetPidStatus pidStatus=GetTPCPIDStatus(track);
a017c06a 2137 if (pidStatus==kDetNoSignal) return -999.;
1d59271b 2138
2139 // the following call is needed in order to fill the transient data member
2140 // fTPCsignalTuned which is used in the TPCPIDResponse to judge
2141 // if using tuned on data
87da0205 2142 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))
2143 this->GetTPCsignalTunedOnData(track);
1c9d11be 2144
87da0205 2145 return fTPCResponse.GetNumberOfSigmas(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
1c9d11be 2146}
2147
5cd0300d 2148//______________________________________________________________________________
2149Float_t AliPIDResponse::GetNumberOfSigmasTRD(const AliVParticle *vtrack, AliPID::EParticleType type) const
2150{
2151 //
2152 // Calculate the number of sigmas in the TRD
2153 //
2154
2155 AliVTrack *track=(AliVTrack*)vtrack;
2156
2157 const EDetPidStatus pidStatus=GetTRDPIDStatus(track);
2158 if (pidStatus!=kDetPidOk) return -999.;
2159
2160 return fTRDResponse.GetNumberOfSigmas(track,type);
2161}
2162
1c9d11be 2163//______________________________________________________________________________
355b831b 2164Float_t AliPIDResponse::GetNumberOfSigmasTOF(const AliVParticle *vtrack, AliPID::EParticleType type) const
1c9d11be 2165{
2166 //
355b831b 2167 // Calculate the number of sigmas in the TOF
1c9d11be 2168 //
2169
2170 AliVTrack *track=(AliVTrack*)vtrack;
355b831b 2171
2172 const EDetPidStatus pidStatus=GetTOFPIDStatus(track);
2173 if (pidStatus!=kDetPidOk) return -999.;
1c9d11be 2174
355b831b 2175 return GetNumberOfSigmasTOFold(vtrack, type);
2176}
567624b5 2177//______________________________________________________________________________
2178
2179Float_t AliPIDResponse::GetNumberOfSigmasHMPID(const AliVParticle *vtrack, AliPID::EParticleType type) const
2180{
2181 //
2182 // Calculate the number of sigmas in the HMPID
2183 //
2184 AliVTrack *track=(AliVTrack*)vtrack;
2185
2186 const EDetPidStatus pidStatus=GetHMPIDPIDStatus(track);
2187 if (pidStatus!=kDetPidOk) return -999.;
2188
2189 return fHMPIDResponse.GetNumberOfSigmas(track, type);
2190}
355b831b 2191
2192//______________________________________________________________________________
2193Float_t AliPIDResponse::GetNumberOfSigmasEMCAL(const AliVParticle *vtrack, AliPID::EParticleType type) const
2194{
2195 //
2196 // Calculate the number of sigmas in the EMCAL
2197 //
1c9d11be 2198
355b831b 2199 AliVTrack *track=(AliVTrack*)vtrack;
2200
2201 const EDetPidStatus pidStatus=GetEMCALPIDStatus(track);
2202 if (pidStatus!=kDetPidOk) return -999.;
2203
2204 const Int_t nMatchClus = track->GetEMCALcluster();
2205 AliVCluster *matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
1c9d11be 2206
355b831b 2207 const Double_t mom = track->P();
2208 const Double_t pt = track->Pt();
2209 const Int_t charge = track->Charge();
2210 const Double_t fClsE = matchedClus->E();
2211 const Double_t EovP = fClsE/mom;
1c9d11be 2212
355b831b 2213 return fEMCALResponse.GetNumberOfSigmas(pt,EovP,type,charge);
1c9d11be 2214}
2215
567624b5 2216//______________________________________________________________________________
1d59271b 2217AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaITS(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
567624b5 2218{
2219 //
2220 // Signal minus expected Signal for ITS
2221 //
2222 AliVTrack *track=(AliVTrack*)vtrack;
1d59271b 2223 val=fITSResponse.GetSignalDelta(track,type,ratio);
567624b5 2224
2225 return GetITSPIDStatus(track);
2226}
2227
2228//______________________________________________________________________________
1d59271b 2229AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaTPC(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
567624b5 2230{
2231 //
2232 // Signal minus expected Signal for TPC
2233 //
2234 AliVTrack *track=(AliVTrack*)vtrack;
1d59271b 2235
2236 // the following call is needed in order to fill the transient data member
2237 // fTPCsignalTuned which is used in the TPCPIDResponse to judge
2238 // if using tuned on data
87da0205 2239 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC))
1d59271b 2240 this->GetTPCsignalTunedOnData(track);
2241
87da0205 2242 val=fTPCResponse.GetSignalDelta(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection, ratio);
567624b5 2243
2244 return GetTPCPIDStatus(track);
2245}
2246
5cd0300d 2247//______________________________________________________________________________
2248AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaTRD(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
2249{
2250 //
2251 // Signal minus expected Signal for TRD
2252 //
2253 AliVTrack *track=(AliVTrack*)vtrack;
2254 val=fTRDResponse.GetSignalDelta(track,type,ratio);
2255
2256 return GetTRDPIDStatus(track);
2257}
2258
567624b5 2259//______________________________________________________________________________
1d59271b 2260AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaTOF(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
567624b5 2261{
2262 //
2263 // Signal minus expected Signal for TOF
2264 //
2265 AliVTrack *track=(AliVTrack*)vtrack;
1d59271b 2266 val=GetSignalDeltaTOFold(track, type, ratio);
87da0205 2267
567624b5 2268 return GetTOFPIDStatus(track);
2269}
2270
2271//______________________________________________________________________________
1d59271b 2272AliPIDResponse::EDetPidStatus AliPIDResponse::GetSignalDeltaHMPID(const AliVParticle *vtrack, AliPID::EParticleType type, Double_t &val, Bool_t ratio/*=kFALSE*/) const
567624b5 2273{
2274 //
2275 // Signal minus expected Signal for HMPID
2276 //
2277 AliVTrack *track=(AliVTrack*)vtrack;
1d59271b 2278 val=fHMPIDResponse.GetSignalDelta(track, type, ratio);
567624b5 2279
2280 return GetHMPIDPIDStatus(track);
2281}
1c9d11be 2282
2283//______________________________________________________________________________
2284AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputePIDProbability (EDetector detCode, const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2285{
2286 //
2287 // Compute PID response of 'detCode'
2288 //
2289
2290 switch (detCode){
2291 case kITS: return GetComputeITSProbability(track, nSpecies, p); break;
2292 case kTPC: return GetComputeTPCProbability(track, nSpecies, p); break;
2293 case kTRD: return GetComputeTRDProbability(track, nSpecies, p); break;
2294 case kTOF: return GetComputeTOFProbability(track, nSpecies, p); break;
2295 case kPHOS: return GetComputePHOSProbability(track, nSpecies, p); break;
2296 case kEMCAL: return GetComputeEMCALProbability(track, nSpecies, p); break;
2297 case kHMPID: return GetComputeHMPIDProbability(track, nSpecies, p); break;
2298 default: return kDetNoSignal;
2299 }
2300}
2301
2302//______________________________________________________________________________
2303AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeITSProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2304{
2305 //
2306 // Compute PID response for the ITS
2307 //
2308
1c9d11be 2309 // set flat distribution (no decision)
2310 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2311
355b831b 2312 const EDetPidStatus pidStatus=GetITSPIDStatus(track);
2313 if (pidStatus!=kDetPidOk) return pidStatus;
2314
2315 if (track->GetDetectorPID()){
2316 return track->GetDetectorPID()->GetRawProbability(kITS, p, nSpecies);
2317 }
1c9d11be 2318
2319 //check for ITS standalone tracks
2320 Bool_t isSA=kTRUE;
2321 if( track->GetStatus() & AliVTrack::kTPCin ) isSA=kFALSE;
2322
2323 Double_t mom=track->P();
2324 Double_t dedx=track->GetITSsignal();
2325 Double_t momITS=mom;
2326 UChar_t clumap=track->GetITSClusterMap();
2327 Int_t nPointsForPid=0;
2328 for(Int_t i=2; i<6; i++){
2329 if(clumap&(1<<i)) ++nPointsForPid;
2330 }
2331
1c9d11be 2332 Bool_t mismatch=kTRUE/*, heavy=kTRUE*/;
bf26ce58 2333 for (Int_t j=0; j<nSpecies; j++) {
1c9d11be 2334 Double_t mass=AliPID::ParticleMassZ(j);//GeV/c^2
2335 const Double_t chargeFactor = TMath::Power(AliPID::ParticleCharge(j),2.);
2336 Double_t bethe=fITSResponse.Bethe(momITS,mass)*chargeFactor;
2337 //TODO: in case of the electron, use the SA parametrisation,
2338 // this needs to be changed if ITS provides a parametrisation
2339 // for electrons also for ITS+TPC tracks
2340 Double_t sigma=fITSResponse.GetResolution(bethe,nPointsForPid,isSA || (j==(Int_t)AliPID::kElectron));
2341 if (TMath::Abs(dedx-bethe) > fRange*sigma) {
2342 p[j]=TMath::Exp(-0.5*fRange*fRange)/sigma;
2343 } else {
2344 p[j]=TMath::Exp(-0.5*(dedx-bethe)*(dedx-bethe)/(sigma*sigma))/sigma;
2345 mismatch=kFALSE;
2346 }
1c9d11be 2347 }
2348
2349 if (mismatch){
bf26ce58 2350 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
1c9d11be 2351 }
2352
1c9d11be 2353 return kDetPidOk;
2354}
2355//______________________________________________________________________________
2356AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeTPCProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2357{
2358 //
2359 // Compute PID response for the TPC
2360 //
2361
2362 // set flat distribution (no decision)
2363 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2364
355b831b 2365 const EDetPidStatus pidStatus=GetTPCPIDStatus(track);
a017c06a 2366 if (pidStatus==kDetNoSignal) return pidStatus;
1c9d11be 2367
1c9d11be 2368 Double_t dedx=track->GetTPCsignal();
2369 Bool_t mismatch=kTRUE/*, heavy=kTRUE*/;
2370
87da0205 2371 if (fTuneMConData && ((fTuneMConDataMask & kDetTPC) == kDetTPC)) dedx = this->GetTPCsignalTunedOnData(track);
1c9d11be 2372
f84b18dd 2373 Double_t bethe = 0.;
2374 Double_t sigma = 0.;
2375
bf26ce58 2376 for (Int_t j=0; j<nSpecies; j++) {
1c9d11be 2377 AliPID::EParticleType type=AliPID::EParticleType(j);
f84b18dd 2378
87da0205 2379 bethe=fTPCResponse.GetExpectedSignal(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
2380 sigma=fTPCResponse.GetExpectedSigma(track, type, AliTPCPIDResponse::kdEdxDefault, fUseTPCEtaCorrection, fUseTPCMultiplicityCorrection);
f85a3764 2381
1c9d11be 2382 if (TMath::Abs(dedx-bethe) > fRange*sigma) {
2383 p[j]=TMath::Exp(-0.5*fRange*fRange)/sigma;
2384 } else {
2385 p[j]=TMath::Exp(-0.5*(dedx-bethe)*(dedx-bethe)/(sigma*sigma))/sigma;
2386 mismatch=kFALSE;
2387 }
2388 }
2389
2390 if (mismatch){
2391 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
1c9d11be 2392 }
2393
a017c06a 2394 return pidStatus;
1c9d11be 2395}
2396//______________________________________________________________________________
2397AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeTOFProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2398{
2399 //
57e985ed 2400 // Compute PID probabilities for TOF
1c9d11be 2401 //
4fc0f532 2402
42fcc729 2403 fgTOFmismatchProb = 1E-8;
2404
2405 // centrality --> fCurrCentrality
2406 // Beam type --> fBeamTypeNum
2407 // 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
2408 // isMC --> fIsMC
2409
2410 Int_t nTOFcluster = 0;
2411 if(track->GetTOFHeader() && track->GetTOFHeader()->GetTriggerMask()){ // N TOF clusters available
2412 nTOFcluster = track->GetTOFHeader()->GetNumberOfTOFclusters();
2413 if(fIsMC) nTOFcluster *= 1.5; // +50% in MC
2414 }
2415 else{
2416 switch(fBeamTypeNum){
2417 case kPP: // pp 7 TeV
2418 nTOFcluster = 50;
2419 break;
2420 case kPPB: // pPb 5.05 ATeV
2421 nTOFcluster = 50 + (100-fCurrCentrality)*50;
2422 break;
2423 case kPBPB: // PbPb 2.76 ATeV
2424 nTOFcluster = 50 + (100-fCurrCentrality)*150;
2425 break;
2426 }
2427 }
2428
4fc0f532 2429 //fTOFResponse.GetMismatchProbability(track->GetTOFsignal(),track->Eta()) * 0.01; // for future implementation of mismatch (i.e. 1% mismatch that should be extended for PbPb, pPb)
2430
1c9d11be 2431 // set flat distribution (no decision)
2432 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2433
355b831b 2434 const EDetPidStatus pidStatus=GetTOFPIDStatus(track);
2435 if (pidStatus!=kDetPidOk) return pidStatus;
2436
c5fb644a 2437 const Double_t meanCorrFactor = 0.07/fTOFtail; // Correction factor on the mean because of the tail (should be ~ 0.1 with tail = 1.1)
1c9d11be 2438
bf26ce58 2439 for (Int_t j=0; j<nSpecies; j++) {
1c9d11be 2440 AliPID::EParticleType type=AliPID::EParticleType(j);
355b831b 2441 const Double_t nsigmas=GetNumberOfSigmasTOFold(track,type) + meanCorrFactor;
1c9d11be 2442
355b831b 2443 const Double_t expTime = fTOFResponse.GetExpectedSignal(track,type);
2444 const Double_t sig = fTOFResponse.GetExpectedSigma(track->P(),expTime,AliPID::ParticleMassZ(type));
4fc0f532 2445
2446 if(nsigmas < fTOFtail)
2447 p[j] = TMath::Exp(-0.5*nsigmas*nsigmas)/sig;
2448 else
2449 p[j] = TMath::Exp(-(nsigmas - fTOFtail*0.5)*fTOFtail)/sig;
2450
42fcc729 2451 p[j] += fgTOFmismatchProb;
1c9d11be 2452 }
2453
1c9d11be 2454 return kDetPidOk;
2455}
239fe91c 2456
2457Int_t AliPIDResponse::CalculateTRDResponse(const AliVTrack *track,Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const
2458{
2459 // new function for backward compatibility
2460 // returns number of tracklets PID
2461
2462 UInt_t TRDslicesForPID[2];
2463 SetTRDSlices(TRDslicesForPID,PIDmethod);
2464
2465 Float_t mom[6]={0.};
2466 Double_t dedx[48]={0.}; // Allocate space for the maximum number of TRD slices
2467 Int_t nslices = TRDslicesForPID[1] - TRDslicesForPID[0] + 1;
2468 AliDebug(1, Form("First Slice: %d, Last Slice: %d, Number of slices: %d", TRDslicesForPID[0], TRDslicesForPID[1], nslices));
2469 for(UInt_t ilayer = 0; ilayer < 6; ilayer++){
2470 mom[ilayer] = track->GetTRDmomentum(ilayer);
2471 for(UInt_t islice = TRDslicesForPID[0]; islice <= TRDslicesForPID[1]; islice++){
2472 dedx[ilayer*nslices+islice-TRDslicesForPID[0]] = track->GetTRDslice(ilayer, islice);
2473 }
2474 }
2475
2476 return fTRDResponse.GetResponse(nslices, dedx, mom, p,PIDmethod);
2477
2478}
1c9d11be 2479//______________________________________________________________________________
239fe91c 2480AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeTRDProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[],AliTRDPIDResponse::ETRDPIDMethod PIDmethod) const
1c9d11be 2481{
2482 //
355b831b 2483 // Compute PID probabilities for the TRD
1c9d11be 2484 //
2485
1c9d11be 2486 // set flat distribution (no decision)
2487 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
355b831b 2488
2489 const EDetPidStatus pidStatus=GetTRDPIDStatus(track);
2490 if (pidStatus!=kDetPidOk) return pidStatus;
2491
239fe91c 2492 CalculateTRDResponse(track,p,PIDmethod);
2493
1c9d11be 2494 return kDetPidOk;
1c9d11be 2495}
355b831b 2496
1c9d11be 2497//______________________________________________________________________________
2498AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeEMCALProbability (const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2499{
2500 //
2501 // Compute PID response for the EMCAL
2502 //
2503
2504 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
355b831b 2505
2506 const EDetPidStatus pidStatus=GetEMCALPIDStatus(track);
2507 if (pidStatus!=kDetPidOk) return pidStatus;
2508
2509 const Int_t nMatchClus = track->GetEMCALcluster();
2510 AliVCluster *matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
1c9d11be 2511
355b831b 2512 const Double_t mom = track->P();
2513 const Double_t pt = track->Pt();
2514 const Int_t charge = track->Charge();
2515 const Double_t fClsE = matchedClus->E();
2516 const Double_t EovP = fClsE/mom;
1c9d11be 2517
355b831b 2518 // compute the probabilities
2519 fEMCALResponse.ComputeEMCALProbability(nSpecies,pt,EovP,charge,p);
2520 return kDetPidOk;
1c9d11be 2521}
355b831b 2522
1c9d11be 2523//______________________________________________________________________________
2524AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputePHOSProbability (const AliVTrack */*track*/, Int_t nSpecies, Double_t p[]) const
2525{
2526 //
2527 // Compute PID response for the PHOS
2528 //
2529
2530 // set flat distribution (no decision)
2531 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
2532 return kDetNoSignal;
2533}
355b831b 2534
1c9d11be 2535//______________________________________________________________________________
2536AliPIDResponse::EDetPidStatus AliPIDResponse::GetComputeHMPIDProbability(const AliVTrack *track, Int_t nSpecies, Double_t p[]) const
2537{
2538 //
2539 // Compute PID response for the HMPID
2540 //
355b831b 2541
1c9d11be 2542 // set flat distribution (no decision)
2543 for (Int_t j=0; j<nSpecies; j++) p[j]=1./nSpecies;
355b831b 2544
2545 const EDetPidStatus pidStatus=GetHMPIDPIDStatus(track);
2546 if (pidStatus!=kDetPidOk) return pidStatus;
1c9d11be 2547
567624b5 2548 fHMPIDResponse.GetProbability(track,nSpecies,p);
2549
1c9d11be 2550 return kDetPidOk;
2551}
355b831b 2552
2553//______________________________________________________________________________
2554AliPIDResponse::EDetPidStatus AliPIDResponse::GetITSPIDStatus(const AliVTrack *track) const
2555{
2556 // compute ITS pid status
2557
2558 // check status bits
2559 if ((track->GetStatus()&AliVTrack::kITSin)==0 &&
2560 (track->GetStatus()&AliVTrack::kITSout)==0) return kDetNoSignal;
2561
2562 const Float_t dEdx=track->GetITSsignal();
2563 if (dEdx<=0) return kDetNoSignal;
2564
2565 // requite at least 3 pid clusters
2566 const UChar_t clumap=track->GetITSClusterMap();
2567 Int_t nPointsForPid=0;
2568 for(Int_t i=2; i<6; i++){
2569 if(clumap&(1<<i)) ++nPointsForPid;
2570 }
2571
2572 if(nPointsForPid<3) {
2573 return kDetNoSignal;
2574 }
2575
2576 return kDetPidOk;
2577}
2578
2579//______________________________________________________________________________
2580AliPIDResponse::EDetPidStatus AliPIDResponse:: GetTPCPIDStatus(const AliVTrack *track) const
2581{
2582 // compute TPC pid status
2583
2584 // check quality of the track
2585 if ( (track->GetStatus()&AliVTrack::kTPCin )==0 && (track->GetStatus()&AliVTrack::kTPCout)==0 ) return kDetNoSignal;
2586
2587 // check pid values
2588 const Double_t dedx=track->GetTPCsignal();
2589 const UShort_t signalN=track->GetTPCsignalN();
2590 if (signalN<10 || dedx<10) return kDetNoSignal;
2591
2592 if (!(fArrPidResponseMaster && fArrPidResponseMaster->At(AliPID::kPion))) return kDetNoParams;
2593
2594 return kDetPidOk;
2595}
2596
2597//______________________________________________________________________________
2598AliPIDResponse::EDetPidStatus AliPIDResponse::GetTRDPIDStatus(const AliVTrack *track) const
2599{
2600 // compute TRD pid status
2601
2602 if((track->GetStatus()&AliVTrack::kTRDout)==0) return kDetNoSignal;
2603 return kDetPidOk;
2604}
2605
2606//______________________________________________________________________________
2607AliPIDResponse::EDetPidStatus AliPIDResponse::GetTOFPIDStatus(const AliVTrack *track) const
2608{
2609 // compute TOF pid status
2610
2611 if ((track->GetStatus()&AliVTrack::kTOFout)==0) return kDetNoSignal;
2612 if ((track->GetStatus()&AliVTrack::kTIME)==0) return kDetNoSignal;
2613
2614 return kDetPidOk;
2615}
2616
2617//______________________________________________________________________________
2618Float_t AliPIDResponse::GetTOFMismatchProbability(const AliVTrack *track) const
2619{
2620 // compute mismatch probability cross-checking at 5 sigmas with TPC
2621 // currently just implemented as a 5 sigma compatibility cut
2622
42fcc729 2623 if(!track) return fgTOFmismatchProb;
2624
355b831b 2625 // check pid status
2626 const EDetPidStatus tofStatus=GetTOFPIDStatus(track);
2627 if (tofStatus!=kDetPidOk) return 0.;
2628
2629 //mismatch
2630 const EDetPidStatus tpcStatus=GetTPCPIDStatus(track);
a017c06a 2631 if (tpcStatus==kDetNoSignal) return 0.;
355b831b 2632
2633 const Double_t meanCorrFactor = 0.11/fTOFtail; // Correction factor on the mean because of the tail (should be ~ 0.1 with tail = 1.1)
2634 Bool_t mismatch = kTRUE/*, heavy = kTRUE*/;
2635 for (Int_t j=0; j<AliPID::kSPECIESC; j++) {
2636 AliPID::EParticleType type=AliPID::EParticleType(j);
2637 const Double_t nsigmas=GetNumberOfSigmasTOFold(track,type) + meanCorrFactor;
2638
2639 if (TMath::Abs(nsigmas)<5.){
2640 const Double_t nsigmasTPC=GetNumberOfSigmasTPC(track,type);
2641 if (TMath::Abs(nsigmasTPC)<5.) mismatch=kFALSE;
2642 }
2643 }
2644
2645 if (mismatch){
2646 return 1.;
2647 }
2648
2649 return 0.;
2650}
2651
355b831b 2652//______________________________________________________________________________
2653AliPIDResponse::EDetPidStatus AliPIDResponse:: GetHMPIDPIDStatus(const AliVTrack *track) const
2654{
2655 // compute HMPID pid status
567624b5 2656
2657 Int_t ch = track->GetHMPIDcluIdx()/1000000;
2658 Double_t HMPIDsignal = track->GetHMPIDsignal();
2659
2660 if((track->GetStatus()&AliVTrack::kHMPIDpid)==0 || ch<0 || ch>6 || HMPIDsignal<0) return kDetNoSignal;
2661
355b831b 2662 return kDetPidOk;
2663}
2664
2665//______________________________________________________________________________
2666AliPIDResponse::EDetPidStatus AliPIDResponse:: GetPHOSPIDStatus(const AliVTrack */*track*/) const
2667{
2668 // compute PHOS pid status
2669 return kDetNoSignal;
2670}
2671
2672//______________________________________________________________________________
2673AliPIDResponse::EDetPidStatus AliPIDResponse:: GetEMCALPIDStatus(const AliVTrack *track) const
2674{
2675 // compute EMCAL pid status
2676
2677
2678 // Track matching
2679 const Int_t nMatchClus = track->GetEMCALcluster();
2680 if (nMatchClus<0) return kDetNoSignal;
2681
2682 AliVCluster *matchedClus = (AliVCluster*)fCurrentEvent->GetCaloCluster(nMatchClus);
2683
2684 if (!(matchedClus && matchedClus->IsEMCAL())) return kDetNoSignal;
2685
2686 const Int_t charge = track->Charge();
2687 if (TMath::Abs(charge)!=1) return kDetNoSignal;
2688
2689 if (!(fEMCALPIDParams && fEMCALPIDParams->At(AliPID::kElectron))) return kDetNoParams;
2690
2691 return kDetPidOk;
2692
2693}
2694
2695//______________________________________________________________________________
2696AliPIDResponse::EDetPidStatus AliPIDResponse::GetPIDStatus(EDetector detector, const AliVTrack *track) const
2697{
2698 //
2699 // check pid status for a track
2700 //
2701
2702 switch (detector){
2703 case kITS: return GetITSPIDStatus(track); break;
2704 case kTPC: return GetTPCPIDStatus(track); break;
2705 case kTRD: return GetTRDPIDStatus(track); break;
2706 case kTOF: return GetTOFPIDStatus(track); break;
2707 case kPHOS: return GetPHOSPIDStatus(track); break;
2708 case kEMCAL: return GetEMCALPIDStatus(track); break;
2709 case kHMPID: return GetHMPIDPIDStatus(track); break;
2710 default: return kDetNoSignal;
2711 }
2712 return kDetNoSignal;
2713
2714}
5a9dc560 2715
2716//______________________________________________________________________________
2717TString AliPIDResponse::GetChecksum(const TObject* obj) const
2718{
2719 // Return the checksum for an object obj (tested to work properly at least for histograms and TSplines).
2720
2721 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!
2722
2723 // For parallel processing, a unique file pathname is required. Uniqueness can be guaranteed by using a unique directory name
2724 UInt_t index = 0;
2725 TString uniquePathName = Form("tempChecksum_%u", index);
2726
2727 // To get a unique path name, increase the index until no directory
2728 // of such a name exists.
2729 // NOTE: gSystem->AccessPathName(...) returns kTRUE, if the access FAILED!
2730 while (!gSystem->AccessPathName(uniquePathName.Data()))
2731 uniquePathName = Form("tempChecksum_%u", ++index);
2732
2733 if (gSystem->mkdir(uniquePathName.Data()) < 0) {
2734 AliError("Could not create temporary directory to store temp file for checksum determination!");
2735 return "ERROR";
2736 }
2737
2738 TString option = "";
2739
2740 // Save object as a macro, which will be deleted immediately after the checksum has been computed
2741 // (does not work for desired data types if saved as *.root for some reason) - one only wants to compare the content, not
2742 // the modification time etc. ...
2743 if (dynamic_cast<const TH1*>(obj))
2744 option = "colz"; // Histos need this option, since w/o this option, a counter is added to the filename
2745
2746
2747 // SaveAs must be called with the fixed fileName only, since the first argument goes into the file content
2748 // for some object types. Thus, change the directory, save the file and then go back
2749 TString oldDir = gSystem->pwd();
2750 gSystem->cd(uniquePathName.Data());
2751 obj->SaveAs(fileName.Data(), option.Data());
2752 gSystem->cd(oldDir.Data());
2753
2754 // Use the file to calculate the MD5 checksum
2755 TMD5* md5 = TMD5::FileChecksum(Form("%s/%s", uniquePathName.Data(), fileName.Data()));
2756 TString checksum = md5->AsString();
2757
2758 // Clean up
2759 delete md5;
2760 gSystem->Exec(Form("rm -rf %s", uniquePathName.Data()));
2761
2762 return checksum;
2763}