PWGPP-3, ATO-19 - Reduce size of the space points for alignment (Laser calibration...

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

diff --git a/PWGPP/pid/AliAnalysisTaskPhiBayes.h b/PWGPP/pid/AliAnalysisTaskPhiBayes.h
index 6fd3dc255a17d6dfcc19cb04198eafd22874edd5..49c0deb9b63fcf47dc2c68aacf4510bd295fb7a8 100644 (file)
--- a/PWGPP/pid/AliAnalysisTaskPhiBayes.h
+++ b/PWGPP/pid/AliAnalysisTaskPhiBayes.h
@@ -13,6 +13,7 @@
 #include <AliAnalysisTaskSE.h>
 #include <AliAODEvent.h>
 #include "AliPIDperfContainer.h"
+#include "AliPIDperfCut.h"
 
 class TH1F;
 class TH2F;
@@ -42,11 +43,19 @@ class AliAnalysisTaskPhiBayes : public AliAnalysisTaskSE {
 
   void SetTPCclusterN(Int_t ncl){fNcluster=ncl;};
 
+  void SetCorrEP(Bool_t flag=kTRUE) {fToEP = flag;};
+
   static const Int_t nPtBin = 13;  //! # pt phi bins
 
   void SetFilterBit(Int_t fb){fFilterBit=fb;};
   Int_t GetFilterBit() const {return fFilterBit;};
 
+  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:
   AliAnalysisTaskPhiBayes(const AliAnalysisTaskPhiBayes &old); 
   AliAnalysisTaskPhiBayes& operator=(const AliAnalysisTaskPhiBayes &source); 
@@ -116,11 +125,19 @@ class AliAnalysisTaskPhiBayes : 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
 
   TH1F *fHmismTOF;         //! TOF mismatch distribution
   TH1D *fHchannelTOFdistr; //! TOF channel distance w.r.t. IP
 
-  ClassDef(AliAnalysisTaskPhiBayes, 1);    //Analysis task for bayesian (K0s)
+  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
+  Int_t fSpeciesRef;    // reference species (default = 3, kaon)
+
+  ClassDef(AliAnalysisTaskPhiBayes, 5);    //Analysis task for bayesian (K0s)
 };
 
 #endif