X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PWGPP%2Fpid%2FAliAnalysisTaskK0sBayes.h;h=a1410223ea7ba36063a487b8fb58634ef7fd3c05;hb=c95d10435ed795929c1be06a1439e1c0e1983b4c;hp=7e81d0cab03f85850ba33e452f52c092fb77054c;hpb=9c668341640bd6339bb3cfb4084b98cce19e544d;p=u%2Fmrichter%2FAliRoot.git

diff --git a/PWGPP/pid/AliAnalysisTaskK0sBayes.h b/PWGPP/pid/AliAnalysisTaskK0sBayes.h
index 7e81d0cab03..a1410223ea7 100644
--- 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