Coverity warning 22314

[u/mrichter/AliRoot.git] / PWGPP / pid / AliAnalysisTaskK0sBayes.h

diff --git a/PWGPP/pid/AliAnalysisTaskK0sBayes.h b/PWGPP/pid/AliAnalysisTaskK0sBayes.h
index 7e81d0cab03f85850ba33e452f52c092fb77054c..a1410223ea7ba36063a487b8fb58634ef7fd3c05 100644 (file)
--- a/PWGPP/pid/AliAnalysisTaskK0sBayes.h
+++ b/PWGPP/pid/AliAnalysisTaskK0sBayes.h
@@ -13,6 +13,7 @@
 #include <AliAnalysisTaskSE.h>
 #include <AliAODEvent.h>
 #include "AliPIDperfContainer.h"
+#include "AliPIDperfCut.h"
 
 class TH1F;
 class TH2F;
@@ -42,6 +43,8 @@ class AliAnalysisTaskK0sBayes : public AliAnalysisTaskSE {
 
   void SetTPCclusterN(Int_t ncl){fNcluster=ncl;};
 
+  void SetCorrEP(Bool_t flag=kTRUE) {fToEP = flag;};
+
   static const Int_t nPtBin = 13;  //! # pt ks bins
 
   void SetFilterBit(Int_t fb){fFilterBit=fb;};
@@ -50,6 +53,10 @@ class AliAnalysisTaskK0sBayes : public AliAnalysisTaskSE {
   void SetTypeCollisions(Int_t type){fTypeCol = type;}; // 0=pp, 1=pPb, 2=PbPb
   Int_t GetTypeCollisions() const {return fTypeCol;}; 
 
+  void SetPIDuserCut(AliPIDperfCut *usercut){fPIDuserCut = usercut;};
+
+  void SetRefSpecies(Int_t species) {fSpeciesRef = species;}
+
  private:
   AliAnalysisTaskK0sBayes(const AliAnalysisTaskK0sBayes &old); 
   AliAnalysisTaskK0sBayes& operator=(const AliAnalysisTaskK0sBayes &source); 
@@ -123,6 +130,8 @@ class AliAnalysisTaskK0sBayes : public AliAnalysisTaskSE {
   AliPIDCombined *fPIDCombined;  //! PID combined object
   AliPIDperfContainer* fContPid;      //! results for positive
   AliPIDperfContainer* fContPid2;     //! results for negative
+  AliPIDperfContainer* fContUser;     //! results for positive user cut
+  AliPIDperfContainer* fContUser2;    //! results for negative user cut
 
   Int_t fNK0s; //! number of K0s in my private selection
   Float_t fPhiK0s[1000]; //! phi of K0s in my private selection
@@ -139,8 +148,12 @@ class AliAnalysisTaskK0sBayes : public AliAnalysisTaskSE {
   TH1D *fHchannelTOFdistr; //! TOF channel distance w.r.t. IP
 
   Int_t fTypeCol; // type of collision system (0=pp, 1=pPb, 2=PbPb)
+  AliPIDperfCut *fPIDuserCut;  // pid user cut to be cheked
+
+  Bool_t fToEP;        // correlate to TPC EP
 
-  ClassDef(AliAnalysisTaskK0sBayes, 2);    //Analysis task for bayesian (K0s)
+  Int_t fSpeciesRef;    // reference species (default = 2, pion)
+  ClassDef(AliAnalysisTaskK0sBayes, 5);    //Analysis task for bayesian (K0s)
 };
 
 #endif