Update of the HFE package

[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEefficiency.h

diff --git a/PWG3/hfe/AliHFEefficiency.h b/PWG3/hfe/AliHFEefficiency.h
index 6de24e3d225fa53c67ee6fd08bd3281f568cce50..6de6d7eff231088d5b2959a61a0255eb41bdfe98 100644 (file)
--- a/PWG3/hfe/AliHFEefficiency.h
+++ b/PWG3/hfe/AliHFEefficiency.h
@@ -49,6 +49,8 @@ class AliHFEefficiency : public AliAnalysisTaskSE{
 
     void Load(const char* filename = "EffTask.root");
     void PostProcess();
+    Bool_t IsRunningTerminate() const { return TestBit(kTerminate); }
+    void SetRunTerminate(Bool_t terminate = kTRUE) { SetBit(kTerminate, terminate); }
 
     void CalculatePTsmearing();
     void DrawPtResolution(TList *l);
@@ -58,6 +60,9 @@ class AliHFEefficiency : public AliAnalysisTaskSE{
       kNTracks,
       kPt
     };
+    enum{ // Bit Definition
+      kTerminate = BIT(18)
+    };
     AliHFEefficiency(const AliHFEefficiency &);
     AliHFEefficiency &operator=(const AliHFEefficiency &);
     void DrawSignalEfficiency(AliCFEffGrid *eff, AliCFContainer *cont, Int_t var);