Major update of the framework

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

diff --git a/PWG3/dielectron/AliAnalysisTaskDielectronSE.h b/PWG3/dielectron/AliAnalysisTaskDielectronSE.h
index f5503d7ca2c05b37475ffc7d9c1db552d11a5a44..dc5ff3ef198c9276c7bab7ce1f4a88299417edd3 100644 (file)
--- a/PWG3/dielectron/AliAnalysisTaskDielectronSE.h
+++ b/PWG3/dielectron/AliAnalysisTaskDielectronSE.h
@@ -18,6 +18,7 @@
 #include "AliAnalysisTaskSE.h"
 
 class AliDielectron;
+class TH1D;
 
 class AliAnalysisTaskDielectronSE : public AliAnalysisTaskSE {
   
@@ -28,6 +29,10 @@ public:
 
   virtual void  UserExec(Option_t *option);
   virtual void  UserCreateOutputObjects();
+
+  void UsePhysicsSelection(Bool_t phy=kTRUE) {fSelectPhysics=phy;}
+  void SetTriggerMask(UInt_t mask) {fTriggerMask=mask;}
+  UInt_t GetTriggerMask() const { return fTriggerMask; }
   
   void SetDielectron(AliDielectron * const die) { fDielectron = die; }
   
@@ -35,6 +40,11 @@ private:
   
   AliDielectron *fDielectron;             // Dielectron framework object
 
+  Bool_t fSelectPhysics;             // Whether to use physics selection
+  UInt_t fTriggerMask;               // Event trigger mask
+  
+  TH1D *fEventStat;                  //! Histogram with event statistics
+  
   AliAnalysisTaskDielectronSE(const AliAnalysisTaskDielectronSE &c);
   AliAnalysisTaskDielectronSE& operator= (const AliAnalysisTaskDielectronSE &c);