Updates and additions: Classes for signal and spectrum extraction; saving of

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

diff --git a/PWG3/dielectron/AliAnalysisTaskMultiDielectron.h b/PWG3/dielectron/AliAnalysisTaskMultiDielectron.h
index 7762bf2bb662c2f8fedad07d671c02cb0fc47ddf..a9942d715a7662489383b995ad7db917bdf06c3e 100644 (file)
--- a/PWG3/dielectron/AliAnalysisTaskMultiDielectron.h
+++ b/PWG3/dielectron/AliAnalysisTaskMultiDielectron.h
@@ -26,11 +26,13 @@ class AliAnalysisTaskMultiDielectron : public AliAnalysisTaskSE {
 public:
   AliAnalysisTaskMultiDielectron();
   AliAnalysisTaskMultiDielectron(const char *name);
-  virtual ~AliAnalysisTaskMultiDielectron(){}
+  virtual ~AliAnalysisTaskMultiDielectron(){  }
 
-  virtual void  UserExec(Option_t *option);
-  virtual void  UserCreateOutputObjects();
+  virtual void UserExec(Option_t *option);
+  virtual void UserCreateOutputObjects();
+  virtual void FinishTaskOutput();
   
+  void UsePhysicsSelection(Bool_t phy=kTRUE) {fSelectPhysics=phy;}
   
   void AddDielectron(AliDielectron * const die) { fListDielectron.Add(die); }
   
@@ -39,6 +41,8 @@ private:
   TList fListDielectron;             // List of dielectron framework instances
   TList fListHistos;                 //! List of histogram manager lists in the framework classes
   TList fListCF;                     //! List with CF Managers
+
+  Bool_t fSelectPhysics;             // Whether to use physics selection
   
   AliAnalysisTaskMultiDielectron(const AliAnalysisTaskMultiDielectron &c);
   AliAnalysisTaskMultiDielectron& operator= (const AliAnalysisTaskMultiDielectron &c);