1 /**************************************************************************
2 * Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
5 * Author: The ALICE Off-line Project. *
6 * Contributors are mentioned in the code where appropriate. *
8 * Permission to use, copy, modify and distribute this software and its *
9 * documentation strictly for non-commercial purposes is hereby granted *
10 * without fee, provided that the above copyright notice appears in all *
11 * copies and that both the copyright notice and this permission notice *
12 * appear in the supporting documentation. The authors make no claims *
13 * about the suitability of this software for any purpose. It is *
14 * provided "as is" without express or implied warranty. *
15 * *************************************************************************/
17 //***********************************************************
19 // class for PID with AliAODRecoDecayHF
20 // Authors: D. Caffarri caffarri@pd.infn.it, A.Dainese andrea.dainese@pd.infn.it, S. Dash dash@to.infn.it, F. Prino prino@to.infn.it, R. Romita r.romita@gsi.de, Y. Wang yifei@pi0.physi.uni-heidelberg.de
21 //***********************************************************
22 #include "AliAODPidHF.h"
23 #include "AliAODPid.h"
25 #include "AliTPCPIDResponse.h"
26 #include "AliITSPIDResponse.h"
27 #include "AliTOFPIDResponse.h"
28 #include "AliAODpidUtil.h"
29 #include "AliESDtrack.h"
34 //------------------------------
35 AliAODPidHF::AliAODPidHF():
54 // Default constructor
56 fPLimit=new Double_t[fnPLimit];
57 fnSigma=new Double_t[fnNSigma];
58 fPriors=new Double_t[fnPriors];
60 for(Int_t i=0;i<fnNSigma;i++){
63 for(Int_t i=0;i<fnPriors;i++){
66 for(Int_t i=0;i<fnPLimit;i++){
71 //----------------------
72 AliAODPidHF::~AliAODPidHF()
75 // if(fPLimit) delete fPLimit;
76 // if(fnSigma) delete fnSigma;
77 // if(fPriors) delete fPriors;
79 //------------------------
80 AliAODPidHF::AliAODPidHF(const AliAODPidHF& pid) :
82 fnNSigma(pid.fnNSigma),
84 fTOFSigma(pid.fTOFSigma),
85 fnPriors(pid.fnPriors),
87 fnPLimit(pid.fnPLimit),
99 for(Int_t i=0;i<5;i++){
100 fPriors[i]=pid.fPriors[i];
105 //----------------------
106 Int_t AliAODPidHF::RawSignalPID(AliAODTrack *track, TString detector) const{
107 // raw PID for single detectors, returns the particle type with smaller sigma
109 if(detector.Contains("ITS")) return ApplyPidITSRaw(track,specie);
110 if(detector.Contains("TPC")) return ApplyPidTPCRaw(track,specie);
111 if(detector.Contains("TOF")) return ApplyPidTOFRaw(track,specie);
116 //---------------------------
117 Bool_t AliAODPidHF::IsKaonRaw(AliAODTrack *track, TString detector) const{
118 // checks if the track can be a kaon, raw PID applied for single detectors
121 if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,3);
122 if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,3);
123 if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,3);
125 if(specie==3) return kTRUE;
128 //---------------------------
129 Bool_t AliAODPidHF::IsPionRaw (AliAODTrack *track, TString detector) const{
130 // checks if the track can be a pion, raw PID applied for single detectors
134 if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,2);
135 if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,2);
136 if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,2);
138 if(specie==2) return kTRUE;
141 //---------------------------
142 Bool_t AliAODPidHF::IsProtonRaw (AliAODTrack *track, TString detector) const{
143 // checks if the track can be a proton raw PID applied for single detectors
146 if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,4);
147 if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,4);
148 if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,4);
150 if(specie==4) return kTRUE;
154 //--------------------------
155 Bool_t AliAODPidHF::IsElectronRaw(AliAODTrack *track, TString detector) const{
156 // checks if the track can be an electron raw PID applied for single detectors
159 if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,0);
160 if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,0);
161 if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,0);
163 if(specie==0) return kTRUE;
167 //--------------------------
168 Int_t AliAODPidHF::ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const{
169 // n-sigma cut, TPC PID
171 if(!CheckStatus(track,"TPC")) return 0;
172 AliAODPid *pidObj = track->GetDetPid();
174 Double_t dedx=pidObj->GetTPCsignal();
175 Double_t mom = pidObj->GetTPCmomentum();
176 AliTPCPIDResponse tpcResponse;
178 if(specie<0){ // from RawSignalPID : should return the particle specie to wich the de/dx is closer to the bethe-block curve -> performance to be checked
179 Double_t nsigmaMax=fnSigma[0];
180 for(Int_t ipart=0;ipart<5;ipart++){
181 AliPID::EParticleType type=AliPID::EParticleType(ipart);
182 Double_t nsigma = TMath::Abs(tpcResponse.GetNumberOfSigmas(mom,dedx,track->GetTPCNcls(),type));
183 if((nsigma<nsigmaMax) && (nsigma<fnSigma[0])) {
188 }else{ // asks only for one particle specie
189 AliPID::EParticleType type=AliPID::EParticleType(specie);
190 Double_t nsigma = TMath::Abs(tpcResponse.GetNumberOfSigmas(mom,dedx,track->GetTPCNcls(),type));
191 if (nsigma>fnSigma[0]) {
201 //----------------------------
202 Int_t AliAODPidHF::ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const{
203 // truncated mean, ITS PID
205 if(!CheckStatus(track,"ITS")) return 0;
207 Double_t mom=track->P();
208 AliAODPid *pidObj = track->GetDetPid();
210 Double_t dedx=pidObj->GetITSsignal();
211 UChar_t clumap=track->GetITSClusterMap();
212 Int_t nPointsForPid=0;
213 for(Int_t i=2; i<6; i++){
214 if(clumap&(1<<i)) ++nPointsForPid;
218 if(track->GetStatus() & AliESDtrack::kTPCin) isSA = kFALSE;
220 AliITSPIDResponse itsResponse;
222 if(specie<0){ // from RawSignalPID : should return the particle specie to wich the de/dx is closer to the bethe-block curve -> performance to be checked
223 Double_t nsigmaMax=fnSigma[4];
224 for(Int_t ipart=0;ipart<5;ipart++){
225 AliPID::EParticleType type=AliPID::EParticleType(ipart);
226 Double_t nsigma = TMath::Abs(itsResponse.GetNumberOfSigmas(mom,dedx,type,nPointsForPid,isSA));
227 if((nsigma<nsigmaMax) && (nsigma<fnSigma[4])) {
232 }else{ // asks only for one particle specie
233 AliPID::EParticleType type=AliPID::EParticleType(specie);
234 Double_t nsigma = TMath::Abs(itsResponse.GetNumberOfSigmas(mom,dedx,type));
235 if (nsigma>fnSigma[4]) {
243 //----------------------------
244 Int_t AliAODPidHF::ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const{
245 // n-sigma cut, TOF PID
247 if(!CheckStatus(track,"TOF")) return 0;
249 Double_t time[AliPID::kSPECIESN];
250 AliAODPid *pidObj = track->GetDetPid();
251 pidObj->GetIntegratedTimes(time);
252 Double_t sigTOF=pidObj->GetTOFsignal();
253 // AliTOFPIDResponse tofResponse;
256 if(specie<0){ // from RawSignalPID : should return the particle specie to wich the de/dx is closer to the bethe-block curve -> performance to be checked
257 Double_t nsigmaMax=fTOFSigma*fnSigma[3];
258 for(Int_t ipart=0;ipart<5;ipart++){
259 //AliPID::EParticleType type=AliPID::EParticleType(ipart);
260 //Double_t nsigma = tofResponse.GetExpectedSigma(track->P(),time[type],AliPID::ParticleMass(type));
261 Double_t nsigma=TMath::Abs(sigTOF-time[ipart]);
262 if((nsigma<nsigmaMax) && (nsigma<fnSigma[3]*fTOFSigma)) {
267 }else{ // asks only for one particle specie
268 //AliPID::EParticleType type=AliPID::EParticleType(specie);
269 //Double_t nsigma = TMath::Abs(tofResponse.GetExpectedSigma(track->P(),time[type],AliPID::ParticleMass(type)));
270 Double_t nsigma=TMath::Abs(sigTOF-time[specie]);
271 if (nsigma>fnSigma[3]*fTOFSigma) {
280 //------------------------------
281 void AliAODPidHF::CombinedProbability(AliAODTrack *track,Bool_t *type) const{
282 // combined PID stored inside the AOD track
284 const Double_t *pid=track->PID();
287 for (Int_t i=0; i<10; i++) {
288 if (pid[i]>max) {k=i; max=pid[i];}
291 if(k==2) type[0]=kTRUE;
292 if(k==3) type[1]=kTRUE;
293 if(k==4) type[2]=kTRUE;
297 //--------------------
298 void AliAODPidHF::BayesianProbability(AliAODTrack *track,Double_t *pid) const{
299 // bayesian PID for single detectors or combined
301 if(fITS && !fTPC && !fTOF) {BayesianProbabilityITS(track,pid);return;}
302 if(fTPC && !fITS && !fTOF) {BayesianProbabilityTPC(track,pid);return;}
303 if(fTOF && !fITS && !fTPC) {BayesianProbabilityTOF(track,pid);return;}
305 Double_t probITS[5]={1.,1.,1.,1.,1.};
306 Double_t probTPC[5]={1.,1.,1.,1.,1.};
307 Double_t probTOF[5]={1.,1.,1.,1.,1.};
308 if(fITS) BayesianProbabilityITS(track,probITS);
309 if(fTPC) BayesianProbabilityTPC(track,probTPC);
310 if(fTOF) BayesianProbabilityTOF(track,probTOF);
311 Double_t probTot[5]={0.,0.,0.,0.,0.};
312 for(Int_t i=0;i<5;i++){
313 probTot[i]=probITS[i]*probTPC[i]*probTOF[i];
315 for(Int_t i2=0;i2<5;i2++){
316 pid[i2]=probTot[i2]*fPriors[i2]/(probTot[0]*fPriors[0]+probTot[1]*fPriors[1]+probTot[2]*fPriors[2]+probTot[3]*fPriors[3]+probTot[4]*fPriors[4]);
322 //------------------------------------
323 void AliAODPidHF::BayesianProbabilityITS(AliAODTrack *track,Double_t *prob) const{
325 // bayesian PID for ITS
327 Double_t itspid[AliPID::kSPECIES];
328 pid.MakeITSPID(track,itspid);
329 for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
330 if(fTOF || fTPC || fTRD){
331 prob[ind]=itspid[ind];
333 prob[ind]=itspid[ind]*fPriors[ind]/(itspid[0]*fPriors[0]+itspid[1]*fPriors[1]+itspid[2]*fPriors[2]+itspid[3]*fPriors[3]+itspid[4]*fPriors[4]);
339 //------------------------------------
340 void AliAODPidHF::BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob) const{
341 // bayesian PID for TPC
344 Double_t tpcpid[AliPID::kSPECIES];
345 pid.MakeTPCPID(track,tpcpid);
346 for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
347 if(fTOF || fITS || fTRD){
348 prob[ind]=tpcpid[ind];
350 prob[ind]=tpcpid[ind]*fPriors[ind]/(tpcpid[0]*fPriors[0]+tpcpid[1]*fPriors[1]+tpcpid[2]*fPriors[2]+tpcpid[3]*fPriors[3]+tpcpid[4]*fPriors[4]);
356 //------------------------------------
357 void AliAODPidHF::BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob) const{
358 // bayesian PID for TOF
361 Double_t tofpid[AliPID::kSPECIES];
362 pid.MakeTOFPID(track,tofpid);
363 for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
364 if(fTPC || fITS || fTRD){
365 prob[ind]=tofpid[ind];
367 prob[ind]=tofpid[ind]*fPriors[ind]/(tofpid[0]*fPriors[0]+tofpid[1]*fPriors[1]+tofpid[2]*fPriors[2]+tofpid[3]*fPriors[3]+tofpid[4]*fPriors[4]);
373 //---------------------------------
374 void AliAODPidHF::BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob) const{
375 // bayesian PID for TRD
378 Double_t trdpid[AliPID::kSPECIES];
379 pid.MakeTRDPID(track,trdpid);
380 for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
381 if(fTPC || fITS || fTOF){
382 prob[ind]=trdpid[ind];
384 prob[ind]=trdpid[ind]*fPriors[ind]/(trdpid[0]*fPriors[0]+trdpid[1]*fPriors[1]+trdpid[2]*fPriors[2]+trdpid[3]*fPriors[3]+trdpid[4]*fPriors[4]);
390 //--------------------------------
391 Bool_t AliAODPidHF::CheckStatus(AliAODTrack *track,TString detectors) const{
393 // Quality cuts on the tracks, detector by detector
395 if(detectors.Contains("ITS")){
396 if ((track->GetStatus()&AliESDtrack::kITSin)==0) return kFALSE;
397 UChar_t clumap=track->GetITSClusterMap();
398 Int_t nPointsForPid=0;
399 for(Int_t i=2; i<6; i++){
400 if(clumap&(1<<i)) ++nPointsForPid;
402 if(nPointsForPid<3) return kFALSE;
405 if(detectors.Contains("TPC")){
406 if ((track->GetStatus()&AliESDtrack::kTPCin )==0) return kFALSE;
407 UShort_t nTPCClus=track->GetTPCClusterMap().CountBits();
408 if (nTPCClus<70) return kFALSE;
411 if(detectors.Contains("TOF")){
412 if ((track->GetStatus()&AliESDtrack::kTOFout )==0) return kFALSE;
413 if ((track->GetStatus()&AliESDtrack::kTIME )==0) return kFALSE;
414 if ((track->GetStatus()&AliESDtrack::kTOFpid )==0) return kFALSE;
418 if(detectors.Contains("TRD")){
419 if ((track->GetStatus()&AliESDtrack::kTRDout )==0) return kFALSE;
420 AliAODPid *pidObj = track->GetDetPid();
421 Float_t *mom=pidObj->GetTRDmomentum();
423 for(Int_t iPl=0;iPl<6;iPl++){
424 if(mom[iPl]>0.) ntracklets++;
426 if(ntracklets<4) return kFALSE;
431 //--------------------------------------------
432 Bool_t AliAODPidHF::TPCRawAsym(AliAODTrack* track,Int_t specie) const{
433 // TPC nsigma cut PID, different sigmas in different p bins
435 if(!CheckStatus(track,"TPC")) return kFALSE;
436 AliAODPid *pidObj = track->GetDetPid();
437 Double_t mom = pidObj->GetTPCmomentum();
438 Double_t dedx=pidObj->GetTPCsignal();
440 AliTPCPIDResponse tpcResponse;
441 AliPID::EParticleType type=AliPID::EParticleType(specie);
442 Double_t nsigma = TMath::Abs(tpcResponse.GetNumberOfSigmas(mom,dedx,track->GetTPCNcls(),type));
444 if(mom<fPLimit[0] && nsigma<fnSigma[0]) return kTRUE;
445 if(mom<fPLimit[1] && mom>fPLimit[0] && nsigma<fnSigma[1]) return kTRUE;
446 if(mom>fPLimit[1] && nsigma<fnSigma[2]) return kTRUE;
451 Int_t AliAODPidHF::MatchTPCTOF(AliAODTrack *track,Int_t mode,Int_t specie,Bool_t compat){
452 // combination of the PID info coming from TPC and TOF
454 //TOF || TPC (a la' Andrea R.)
456 // for the single detectors: -1 = kFALSE, 1 = kTRUE, 0 = compatible
457 // the method returns the sum of the response of the 2 detectors
458 if(fTPC && fTOF) {if(!CheckStatus(track,"TPC") && !CheckStatus(track,"TOF")) return 0;}
463 if(CheckStatus(track,"TPC")) {
465 if(TPCRawAsym(track,specie)) {
471 if(specie==2 && IsPionRaw(track,"TPC")) {
476 if(specie==3 && IsKaonRaw(track,"TPC")) {
481 if(specie==4 && IsProtonRaw(track,"TPC")) {
490 if(compat && tTPCinfo<0){
491 Double_t sig0tmp=fnSigma[0];
493 if(specie==2 && IsPionRaw(track,"TPC")) tTPCinfo=0;
494 if(specie==3 && IsKaonRaw(track,"TPC")) tTPCinfo=0;
495 if(specie==4 && IsProtonRaw(track,"TPC")) tTPCinfo=0;
504 if(!CheckStatus(track,"TOF") && fTPC) return tTPCinfo;
508 if(specie==2 && IsPionRaw(track,"TOF")) tTOFinfo=1;
509 if(specie==3 && IsKaonRaw(track,"TOF")) tTOFinfo=1;
510 if(specie==4 && IsProtonRaw(track,"TOF")) tTOFinfo=1;
512 if(compat && tTOFinfo>0){
513 Double_t ptrack=track->P();
514 if(ptrack>1.5) tTOFinfo=0;
518 if(tTPCinfo+tTOFinfo==0 && fITS){
519 if(!CheckStatus(track,"ITS")) return tTPCinfo+tTOFinfo;
521 if(specie==2 && IsPionRaw(track,"TOF")) tITSinfo=1;
522 if(specie==3 && IsKaonRaw(track,"TOF")) tITSinfo=1;
523 if(specie==4 && IsProtonRaw(track,"TOF")) tITSinfo=1;
527 return tTPCinfo+tTOFinfo;
530 //TPC & TOF (a la' Yifei)
531 // convention: -1 = kFALSE, 1 = kTRUE, 0 = not identified
534 if(fTPC && CheckStatus(track,"TPC")) {
537 if(!TPCRawAsym(track,specie)) tTPCinfo=-1;
539 if(specie==2 && !IsPionRaw(track,"TPC")) tTPCinfo=-1;
540 if(specie==3 && !IsKaonRaw(track,"TPC")) tTPCinfo=-1;
541 if(specie==4 && !IsProtonRaw(track,"TPC")) tTPCinfo=-1;
547 if(fTPC && !CheckStatus(track,"TOF")) return tTPCinfo;
549 if(specie==2 && !IsPionRaw(track,"TOF")) tTOFinfo=-1;
550 if(specie==3 && !IsKaonRaw(track,"TOF")) tTOFinfo=-1;
551 if(specie==4 && !IsProtonRaw(track,"TOF")) tTOFinfo=-1;
554 if(tTOFinfo==1 && tTPCinfo==1) return 1;
556 if(tTPCinfo+tTOFinfo==0 && fITS){
557 if(!CheckStatus(track,"ITS")) return tTPCinfo+tTOFinfo;
559 if(specie==2 && IsPionRaw(track,"TOF")) tITSinfo=1;
560 if(specie==3 && IsKaonRaw(track,"TOF")) tITSinfo=1;
561 if(specie==4 && IsProtonRaw(track,"TOF")) tITSinfo=1;
570 //TPC for p<fPLimit[0], TOF for p>=fPLimit[0] (a la' Andrea A.)
571 // convention (temporary): -1 = kFALSE, 1 = kTRUE, 0 = not identified
573 if(fTPC && fTOF) if(!CheckStatus(track,"TPC") && !CheckStatus(track,"TOF")) return 0;
575 Double_t ptrack=track->P();
580 if(ptrack>=fPLimit[0] && ptrack<fPLimit[1] && fTPC) {
581 if(!CheckStatus(track,"TPC")) return 0;
583 if(TPCRawAsym(track,specie)) tTPCinfo=1;
585 if(specie==2 && IsPionRaw(track,"TPC")) tTPCinfo=1;
586 if(specie==3 && IsKaonRaw(track,"TPC")) tTPCinfo=1;
587 if(specie==4 && IsProtonRaw(track,"TPC")) tTPCinfo=1;
593 if(ptrack>=fPLimit[1] && fTOF){
594 if(!CheckStatus(track,"TOF")) return 0;
595 if(specie==2 && IsPionRaw(track,"TOF")) tTOFinfo=1;
596 if(specie==3 && IsKaonRaw(track,"TOF")) tTOFinfo=1;
597 if(specie==4 && IsProtonRaw(track,"TOF")) tTOFinfo=1;
602 if(ptrack<fPLimit[0] && fITS){
603 if(!CheckStatus(track,"ITS")) return 0;
604 if(specie==2 && IsPionRaw(track,"ITS")) tITSinfo=1;
605 if(specie==3 && IsKaonRaw(track,"ITS")) tITSinfo=1;
606 if(specie==4 && IsProtonRaw(track,"ITS")) tITSinfo=1;
615 //----------------------------------
616 Int_t AliAODPidHF::MakeRawPid(AliAODTrack *track, Int_t specie){
617 // general method to compute PID
619 return MatchTPCTOF(track,fMatch,specie,fCompat);
621 if(fTPC && !fTOF && !fITS) {
622 Int_t tTPCres=ApplyPidTPCRaw(track,specie);
623 if(tTPCres==specie){return 1;}else{return tTPCres;};
625 AliError("You should enable just one detector if you don't want to match");
628 if(fTOF && !fTPC && !fITS) {
629 Int_t tTOFres=ApplyPidTOFRaw(track,specie);
630 if(tTOFres==specie){return 1;}else{return tTOFres;};
632 AliError("You should enable just one detector if you don't want to match");
636 if(fITS && !fTPC && !fTOF) {
637 Int_t tITSres=ApplyPidITSRaw(track,specie);
638 if(tITSres==specie){return 1;}else{return tITSres;};
640 AliError("You should enable just one detector if you don't want to match");