X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PWG0%2FAliPWG0Helper.h;h=b90fca5c56738e0c869f3f9b727ecee3d7a3c315;hb=a7f69e56435b56065311a77072e9a3b9ecdd0564;hp=a897e7b7402eecb107b020c0eff71777af052446;hpb=25db2d858f9607f41e278ae0c0f283b097d44676;p=u%2Fmrichter%2FAliRoot.git

diff --git a/PWG0/AliPWG0Helper.h b/PWG0/AliPWG0Helper.h
index a897e7b7402..b90fca5c567 100644
--- a/PWG0/AliPWG0Helper.h
+++ b/PWG0/AliPWG0Helper.h
@@ -7,23 +7,57 @@
 
 // static helper functions
 
-class AliESD;
+class AliESDEvent;
+class AliESDVertex;
 class TParticle;
+class TH1;
+class TH2;
 class TH3;
+class AliHeader;
+class AliGenEventHeader;
+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, Bool_t debug = kFALSE);
+    enum Trigger { kAcceptAll = -1, kMB1 = 0, kMB2, kMB3, kSPDFASTOR, kOfflineMB1, kOfflineMB2, kOfflineMB3, kOfflineFASTOR }; // MB1, MB2, MB3 definition from ALICE-INT-2005-025
+    enum AnalysisMode { kInvalid = -1, kSPD = 0x1, kTPC = 0x2, kTPCITS = 0x4, kFieldOn = 0x8 };
+    // in case we want to use bitmaps...
+    enum MCProcessType { kInvalidProcess = -1, kND = 0x1, kDD = 0x2, kSD = 0x4 };
 
-    static void CreateProjections(TH3* hist);
-    static void CreateDividedProjections(TH3* hist, TH3* hist2, const char* axis = 0);
+    static Bool_t IsEventTriggered(const AliESDEvent* aEsd, Trigger trigger);
+    static Bool_t IsEventTriggered(ULong64_t triggerMask, Trigger trigger);
+    static const AliESDVertex* GetVertex(AliESDEvent* aEsd, AnalysisMode analysisMethod, Bool_t debug = kFALSE, Bool_t bRedoTPC = kFALSE);
+    static Bool_t TestVertex(const AliESDVertex* vertex, AnalysisMode analysisMode, Bool_t debug = kFALSE);
+
+    static Bool_t IsPrimaryCharged(TParticle* aParticle, Int_t aTotalPrimaries, Bool_t adebug = kFALSE);
+
+    static AliPWG0Helper::MCProcessType GetEventProcessType(AliHeader* aHeader, Bool_t adebug = kFALSE);
+    static AliPWG0Helper::MCProcessType GetPythiaEventProcessType(AliGenEventHeader* aHeader, Bool_t adebug = kFALSE);
+    static AliPWG0Helper::MCProcessType GetDPMjetEventProcessType(AliGenEventHeader* aHeader, Bool_t adebug = kFALSE);
+    static Int_t GetLastProcessType() { return fgLastProcessType; }
+
+    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 NormalizeToBinWidth(TH1* hist);
+    static void NormalizeToBinWidth(TH2* hist);
+
+    static void PrintConf(AnalysisMode analysisMode, Trigger trigger);
+
   protected:
+    static Int_t fgLastProcessType;    // stores the raw value of the last process type extracnted
+ 
     ClassDef(AliPWG0Helper, 0)
+
+  private:
+    AliPWG0Helper(const AliPWG0Helper&);
+    AliPWG0Helper& operator=(const AliPWG0Helper&);
 };
 
 #endif