Adding Id to PWG3 classes for better tracking of the coverity defect fixes (Ivana)

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAODPidHF.h

diff --git a/PWG3/vertexingHF/AliAODPidHF.h b/PWG3/vertexingHF/AliAODPidHF.h
index 04ffc310d73444183a95770dc7cecef774dedd88..3028a34e3d1d9f6913a1ea561933dfa5ea3e3e6c 100644 (file)
--- a/PWG3/vertexingHF/AliAODPidHF.h
+++ b/PWG3/vertexingHF/AliAODPidHF.h
@@ -2,7 +2,9 @@
 #define ALIAODPIDHF_H
 
 /* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
- *  * See cxx source for full Copyright notice                               */
+ * See cxx source for full Copyright notice                               */
+
+/* $Id$ */ 
 
 //***********************************************************
 //// Class AliAODPidHF
@@ -12,6 +14,7 @@
 
 #include "AliAODPid.h"
 #include "AliAODTrack.h"
+#include "AliTPCPIDResponse.h"
 
 class AliAODPidHF : public AliAODPid{
 
@@ -23,26 +26,62 @@ class AliAODPidHF : public AliAODPid{
  virtual ~AliAODPidHF();
 
  //Setters
- void SetSigma(Double_t *sigma){fnSigma=sigma;return;}
+ void SetSigma(Double_t *sigma){
+    for(Int_t i=0; i<fnNSigma; i++) fnSigma[i]=sigma[i];
+     }
  void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;}
+ void SetSigmaForTPC(Double_t *sigma){for(Int_t i=0;i<3;i++) fnSigma[i]=sigma[i];return;}
+ void SetSigmaForTOF(Double_t sigma){fnSigma[3]=sigma;return;}
+ void SetSigmaForITS(Double_t sigma){fnSigma[4]=sigma;return;}
  void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;}
  void SetPriors(Double_t *priors){fPriors=priors;return;}
- void SetPLimit(Double_t *plim){fPLimit=plim;return;}
+ void SetPLimit(Double_t *plim){for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
+ void SetPLimit(Double_t *plim,Int_t npLim){fnPLimit=npLim;for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
  void SetAsym(Bool_t asym){fAsym=asym;return;}
+ void SetTPC(Bool_t tpc){fTPC=tpc;return;}
+ void SetTOF(Bool_t tof){fTOF=tof;return;}
+ void SetITS(Bool_t its){fITS=its;return;}
+ void SetTRD(Bool_t trd){fTRD=trd;return;}
+ void SetMatch(Int_t match){fMatch=match;return;}
+ void SetCompat(Bool_t comp){fCompat=comp;return;}
+ void SetMC(Bool_t mc){fMC=mc;return;}
+ void SetOnePad(Bool_t onepad){fOnePad=onepad;return;}
+ void SetPbPb(Bool_t pbpb){fPbPb=pbpb;return;}
  
+ //Getters
+ Double_t GetSigma(Int_t idet) const{return fnSigma[idet];}
+ Double_t GetTofSigma() const{return fTOFSigma;}
+ //void GetPriors(Double_t *priors) const{priors=fPriors;return;}
+ //void GetPLimit(Double_t *plim) const{plim=fPLimit;}
+ void GetPriors(Double_t *priors) const{for(Int_t i=0;i<fnPriors;i++){priors[i]=fPriors[i];}return;}
+ void GetPLimit(Double_t *plim) const{for(Int_t i=0;i<fnPLimit;i++){plim[i]=fPLimit[i];}return;}
+ Bool_t GetAsym() const{return fAsym;}
+ Bool_t GetTPC() const{return fTPC;}
+ Bool_t GetTOF() const{return fTOF;}
+ Bool_t GetITS() const{return fITS;}
+ Bool_t GetTRD() const{return fTRD;}
+ Int_t GetMatch() const{return fMatch;}
+ Bool_t GetCompat() const{return fCompat;}
+ Bool_t GetMC() const{return fMC;}
+ Bool_t GetOnePad() const{return fOnePad;}
+ Bool_t GetPbPb() const{return fPbPb;}
 
  Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
  Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
  Bool_t IsPionRaw (AliAODTrack *track, TString detector) const;
  Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const;
  Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const;
- void BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid) const;
+ void BayesianProbability(AliAODTrack *track,Double_t *pid) const;
  void CombinedProbability(AliAODTrack *track,Bool_t *type) const; //0 = pion, 1 = kaon, 2 = proton
  Bool_t CheckStatus(AliAODTrack *track,TString detectors) const;
 
  Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const;
  Int_t MatchTPCTOF(AliAODTrack *track,Int_t mode,Int_t specie,Bool_t compat);
 
+ Int_t MakeRawPid(AliAODTrack *track,Int_t specie); //general method to perform PID using raw signals
+
+ Bool_t IsTOFPiKexcluded(AliAODTrack *track,Double_t nsigmaK);
+
 
 
  protected:
@@ -54,22 +93,32 @@ class AliAODPidHF : public AliAODPid{
  void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob) const;
  void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob) const;
  void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob) const;
+ void SetBetheBloch(AliTPCPIDResponse tpcResp) const;
 
  private:
-
- Int_t fnNSigma; // size of the nsigma array
- Double_t *fnSigma; //[fnNSigma] sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS 
+ Int_t fnNSigma; // number of sigmas
+ Double_t *fnSigma; // [fnNSigma], sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS 
  Double_t fTOFSigma; // TOF precision 
- Int_t fnPriors;    // size of the priors array
- Double_t *fPriors; //[fnPriors] set of priors
- Int_t fnPLimit;  // size of the plimit array
- Double_t *fPLimit; //[fnPLimit] limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1]
+ Int_t fnPriors; //number of priors
+ Double_t *fPriors; // [fnPriors], set of priors
+ Int_t fnPLimit; //number of Plimit
+ Double_t *fPLimit; // [fnPLimit], limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1]
  Bool_t fAsym; // asimmetric PID required
+ Bool_t fTPC; //switch to include or exclude TPC 
+ Bool_t fTOF; // switch to include or exclude TOF
+ Bool_t fITS; //switch to include or exclude ITS
+ Bool_t fTRD; // switch to include or exclude TRD
+ Int_t fMatch; //switch to combine the info from more detectors: 1 = || , 2 = &, 3 = p region
+ Bool_t fCompat; // compatibility region : useful only if fMatch=1
+ Bool_t fMC; // MC(kTRUE) or real data (kFALSE, default option)
+ Bool_t fOnePad; //  real data with one pad clusters 
+ Bool_t fPbPb; //  real data PbPb 
  
 
 
- ClassDef(AliAODPidHF,4) // AliAODPid for heavy flavor PID
+ ClassDef(AliAODPidHF,7) // AliAODPid for heavy flavor PID
 
 };
 
 #endif
+