Removing the hard-wired particle masses (B. Hippolyte)

[u/mrichter/AliRoot.git] / STEER / AliAODHandler.h

diff --git a/STEER/AliAODHandler.h b/STEER/AliAODHandler.h
index 7f154ad822775d82aacd8163fd636ce7861109ba..b2fcc3633a7675e5e4853673e2b87c3513d67966 100644 (file)
--- a/STEER/AliAODHandler.h
+++ b/STEER/AliAODHandler.h
@@ -15,6 +15,9 @@
 class AliAODEvent;
 class TFile;
 class TTree;
+class AliMCEventHandler;
+class AliAODMCHeader;
+class AliGenEventHeader;
 
 
 
@@ -28,7 +31,7 @@ class AliAODHandler : public AliVEventHandler {
     virtual const char*  GetOutputFileName();
     virtual Bool_t       Init(Option_t* option);
     virtual Bool_t       Init(TTree* /*tree*/, Option_t* /*option*/)  {return kTRUE;}
-    virtual Bool_t       BeginEvent(Long64_t /*entry*/)  {return kTRUE;}
+    virtual Bool_t       BeginEvent(Long64_t /*entry*/){return kTRUE;}
     virtual Bool_t       Notify() { return AliVEventHandler::Notify(); };
     virtual Bool_t       Notify(const char * /* path */) {return kTRUE;}
     virtual Bool_t       FinishEvent();
@@ -36,6 +39,7 @@ class AliAODHandler : public AliVEventHandler {
     virtual Bool_t       TerminateIO();
     //
     virtual void         SetCreateNonStandardAOD()   {fIsStandard = kFALSE;}
+    virtual void         SetFillAOD(Bool_t b)      {fFillAOD = b;}
     virtual void         SetNeedsHeaderReplication() {fNeedsHeaderReplication = kTRUE;}
     virtual void         SetNeedsTracksBranchReplication() {fNeedsTracksBranchReplication = kTRUE;}
     virtual void         SetNeedsVerticesBranchReplication() {fNeedsVerticesBranchReplication = kTRUE;}
@@ -54,6 +58,7 @@ class AliAODHandler : public AliVEventHandler {
     void                 AddAODtoTreeUserInfo();
     void                 AddBranch(const char* cname, void* addobj);
     Bool_t               IsStandard() {return fIsStandard;}
+    Bool_t               GetFillAOD(){return fFillAOD;} 
     Bool_t               NeedsHeaderReplication() {return  fNeedsHeaderReplication;}
     Bool_t               NeedsTracksBranchReplication() {return  fNeedsTracksBranchReplication;}
     Bool_t               NeedsVerticesBranchReplication() {return  fNeedsVerticesBranchReplication;}
@@ -66,11 +71,15 @@ class AliAODHandler : public AliVEventHandler {
     Bool_t               AODIsReplicated() {return fAODIsReplicated;}
     //
     void                 SetInputTree(TTree* /*tree*/) {;}
+    void                 SetMCEventHandler(AliMCEventHandler* mcH) {fMCEventH = mcH;} // For internal use
  private:
+    void StoreMCParticles();
+    void SetMCHeaderInfo(AliAODMCHeader *mcHeader,AliGenEventHeader *genHeader); // Utility function t catch different types of eventheaders
     AliAODHandler(const AliAODHandler&);             // Not implemented
     AliAODHandler& operator=(const AliAODHandler&);  // Not implemented
  private:
     Bool_t                   fIsStandard;                         // Flag for standard aod creation
+    Bool_t                   fFillAOD;                          // Flag for filling of the AOD tree at the end (all or nothing)
     Bool_t                   fNeedsHeaderReplication;             // Flag for header replication
     Bool_t                   fNeedsTracksBranchReplication;       // Flag for tracks replication
     Bool_t                   fNeedsVerticesBranchReplication;     // Flag for vertices replication
@@ -81,12 +90,12 @@ class AliAODHandler : public AliVEventHandler {
     Bool_t                   fNeedsFMDClustersBranchReplication;  // Flag for FMDClusters replication
     Bool_t                   fNeedsCaloClustersBranchReplication; // Flag for CaloClusters replication
     Bool_t                   fAODIsReplicated;                    // Flag true if replication as been executed
-    
     AliAODEvent             *fAODEvent;               //! Pointer to the AOD event
+    AliMCEventHandler       *fMCEventH;               //! Pointer to mc event handler needed not to depend on the manager
     TTree                   *fTreeA;                  //! tree for AOD persistency
     TFile                   *fFileA;                  //! Output file
     TString                  fFileName;               //  Output file name
-    ClassDef(AliAODHandler, 2);
+    ClassDef(AliAODHandler, 3)
 };
 
 #endif