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