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