add EMCal trigger in eh analysis. add hists. in Raa study

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEpidQA.h

diff --git a/PWGHF/hfe/AliHFEpidQA.h b/PWGHF/hfe/AliHFEpidQA.h
index b14b02a6ff4ea5ea59992a07628be7b02a875ac0..fcd242fe2dd23b4cba1edc50a081e09e720713cb 100644 (file)
--- a/PWGHF/hfe/AliHFEpidQA.h
+++ b/PWGHF/hfe/AliHFEpidQA.h
@@ -79,6 +79,15 @@ class AliHFEpidQA : public TObject{
     
     AliHFEtrdPIDqa *GetTRDQA() const { return fTRDpidQA; }
 
+    Int_t GetCentrality(AliVEvent*  const fInputEvent);
+    Int_t GetMultiplicityITS(AliVEvent*  const fInputEvent) const;
+    void SetPbPb() { fIsPbPb = kTRUE; };
+    void SetPP() { fIsPbPb = kFALSE; };
+    void SetPPMultiBin() { fIsppMultiBin = kFALSE; };
+    Bool_t IsPbPb() const { return fIsPbPb; };
+    Bool_t IsPP() const { return !fIsPbPb; };
+    Bool_t IsPPMultiBin() const { return fIsppMultiBin; };
+
   protected:
     enum{
       kV0pidQA = BIT(14),
@@ -109,6 +118,9 @@ class AliHFEpidQA : public TObject{
     Double_t TRDlikeTracklet(Int_t layer, AliESDtrack * const track, Double_t *likelihood);
     Int_t TRDmomBin(Double_t p) const;
 
+
+
+
  protected:
     Int_t GetMaxPID(const Double_t *pidProbs) const;
     Int_t GetPDG(Int_t index);
@@ -125,7 +137,10 @@ class AliHFEpidQA : public TObject{
     TFile             *fNNref;        // reference file for NN pid 
     TMultiLayerPerceptron *fNet[11];  //  reference networks
     Bool_t fTRDTotalChargeInSlice0;     // Fix for Foreward/Backward compatibility
-  
+
+    Bool_t             fIsPbPb;        // Analysis Type: pp or PbPb
+    Bool_t             fIsppMultiBin;  // pp multiplicity bin analysis
+
   ClassDef(AliHFEpidQA, 1)            // PID QA tool
 };
 #endif