added function from Christian to enable branches with new ESD format

[u/mrichter/AliRoot.git] / PWG0 / AliPWG0Helper.h

diff --git a/PWG0/AliPWG0Helper.h b/PWG0/AliPWG0Helper.h
index 3e79132e279f46f04ae05a6860cfbf90d59792c0..04fb01db8aa7aaa267c1f7fd1926eaa1a5239eff 100644 (file)
--- a/PWG0/AliPWG0Helper.h
+++ b/PWG0/AliPWG0Helper.h
@@ -8,20 +8,40 @@
 // static helper functions
 
 class AliESD;
+class AliESDVertex;
 class TParticle;
-class TH3F;
+class TH3;
+class AliHeader;
+class AliStack;
+class TTree;
 
 class AliPWG0Helper : public TObject
 {
   public:
-    static Bool_t IsEventTriggered(AliESD* aEsd);
-    static Bool_t IsVertexReconstructed(AliESD* aEsd);
-    static Bool_t IsPrimaryCharged(TParticle* aParticle, Int_t aTotalPrimaries);
+    enum Trigger { kMB1 = 0, kMB2 }; // definition from ALICE-INT-2005-025
 
-    static void CreateProjections(TH3F* hist);
+    static Bool_t IsEventTriggered(const AliESD* aEsd, Trigger trigger = kMB2);
+    static Bool_t IsEventTriggered(ULong64_t triggerMask, Trigger trigger = kMB2);
+    static Bool_t IsVertexReconstructed(const AliESD* aEsd);
+    static Bool_t IsVertexReconstructed(const AliESDVertex* vtxESD);
+    static Bool_t IsPrimaryCharged(TParticle* aParticle, Int_t aTotalPrimaries, Bool_t adebug = kFALSE);
+
+    static Int_t GetPythiaEventProcessType(AliHeader* aHeader, Bool_t adebug = kFALSE);
+    static TParticle* FindPrimaryMother(AliStack* stack, Int_t label);
+    static Int_t FindPrimaryMotherLabel(AliStack* stack, Int_t label);
+
+    static void CreateProjections(TH3* hist, Bool_t save = kFALSE);
+    static void CreateDividedProjections(TH3* hist, TH3* hist2, const char* axis = 0, Bool_t putErrors = kFALSE, Bool_t save = kFALSE);
+    static const char* GetAxisTitle(TH3* hist, const char axis);
+
+    static void SetBranchStatusRecursive(TTree* tree, char *bname, Bool_t status, Bool_t debug = kFALSE);
 
   protected:
     ClassDef(AliPWG0Helper, 0)
+
+  private:
+    AliPWG0Helper(const AliPWG0Helper&);
+    AliPWG0Helper& operator=(const AliPWG0Helper&);
 };
 
 #endif