o update dielectron package

[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliAnalysisTaskDielectronSE.h

diff --git a/PWGDQ/dielectron/AliAnalysisTaskDielectronSE.h b/PWGDQ/dielectron/AliAnalysisTaskDielectronSE.h
index 6718328035170643e75609f46742885631436daa..6ddae5b4956eda0e209c32dcbbe0099ad9fe8b4f 100644 (file)
--- a/PWGDQ/dielectron/AliAnalysisTaskDielectronSE.h
+++ b/PWGDQ/dielectron/AliAnalysisTaskDielectronSE.h
@@ -16,9 +16,9 @@
 //#####################################################
 
 #include "AliAnalysisTaskSE.h"
-#include "AliDielectronPID.h"
 
 class AliDielectron;
+class AliTriggerAnalysis;
 class TH1D;
 
 class AliAnalysisTaskDielectronSE : public AliAnalysisTaskSE {
@@ -30,22 +30,31 @@ public:
 
   virtual void  UserExec(Option_t *option);
   virtual void  UserCreateOutputObjects();
-  //temporary
-  virtual void NotifyRun(){AliDielectronPID::SetCorrVal((Double_t)fCurrentRunNumber);}
   
   void UsePhysicsSelection(Bool_t phy=kTRUE) {fSelectPhysics=phy;}
   void SetTriggerMask(UInt_t mask) {fTriggerMask=mask;}
   UInt_t GetTriggerMask() const { return fTriggerMask; }
+
+  void SetEventFilter(AliAnalysisCuts * const filter) {fEventFilter=filter;}
+  void SetTriggerOnV0AND(Bool_t v0and=kTRUE)    { fTriggerOnV0AND=v0and;    }
+  void SetRejectPileup(Bool_t pileup=kTRUE)     { fRejectPileup=pileup;     }
   
   void SetDielectron(AliDielectron * const die) { fDielectron = die; }
   
 private:
+  enum {kAllEvents=0, kSelectedEvents, kV0andEvents, kFilteredEvents, kPileupEvents, kNbinsEvent};
   
   AliDielectron *fDielectron;             // Dielectron framework object
 
   Bool_t fSelectPhysics;             // Whether to use physics selection
   UInt_t fTriggerMask;               // Event trigger mask
-  
+  Bool_t fTriggerOnV0AND;            // if to trigger on V0and
+  Bool_t fRejectPileup;              // pileup rejection wanted
+
+  AliTriggerAnalysis *fTriggerAnalysis; //! trigger analysis class
+
+  AliAnalysisCuts *fEventFilter;     // event filter
+
   TH1D *fEventStat;                  //! Histogram with event statistics
   
   AliAnalysisTaskDielectronSE(const AliAnalysisTaskDielectronSE &c);