1 #ifndef ALIAODHANDLER_H
2 #define ALIAODHANDLER_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
8 //-------------------------------------------------------------------------
9 // Implementation of the Event Handler Interface for AOD
10 // Author: Andreas Morsch, CERN
11 //-------------------------------------------------------------------------
13 #include "AliVEventHandler.h"
18 class AliMCEventHandler;
20 class AliGenEventHeader;
24 class AliAODHandler : public AliVEventHandler {
28 AliAODHandler(const char* name, const char* title);
29 virtual ~AliAODHandler();
30 virtual void SetOutputFileName(const char* fname);
31 virtual const char* GetOutputFileName();
32 virtual Bool_t Init(Option_t* option);
33 virtual Bool_t Init(TTree* /*tree*/, Option_t* /*option*/) {return kTRUE;}
34 virtual Bool_t BeginEvent(Long64_t /*entry*/){return kTRUE;}
35 virtual Bool_t Notify() { return AliVEventHandler::Notify(); };
36 virtual Bool_t Notify(const char * /* path */) {return kTRUE;}
37 virtual Bool_t FinishEvent();
38 virtual Bool_t Terminate();
39 virtual Bool_t TerminateIO();
41 virtual void SetCreateNonStandardAOD() {fIsStandard = kFALSE;}
42 virtual void SetFillAOD(Bool_t b) {fFillAOD = b;}
43 virtual void SetNeedsHeaderReplication() {fNeedsHeaderReplication = kTRUE;}
44 virtual void SetNeedsTracksBranchReplication() {fNeedsTracksBranchReplication = kTRUE;}
45 virtual void SetNeedsVerticesBranchReplication() {fNeedsVerticesBranchReplication = kTRUE;}
46 virtual void SetNeedsV0sBranchReplication() {fNeedsV0sBranchReplication = kTRUE;}
47 virtual void SetNeedsTrackletsBranchReplication() {fNeedsTrackletsBranchReplication = kTRUE;}
48 virtual void SetNeedsPMDClustersBranchReplication() {fNeedsPMDClustersBranchReplication = kTRUE;}
49 virtual void SetNeedsJetsBranchReplication() {fNeedsJetsBranchReplication = kTRUE;}
50 virtual void SetNeedsFMDClustersBranchReplication() {fNeedsFMDClustersBranchReplication = kTRUE;}
51 virtual void SetNeedsCaloClustersBranchReplication() {fNeedsCaloClustersBranchReplication = kTRUE;}
52 virtual void SetAODIsReplicated() {fAODIsReplicated = kTRUE;}
54 AliAODEvent* GetAOD() {return fAODEvent;}
55 virtual TTree* GetTree() const {return fTreeA;}
56 void CreateTree(Int_t flag);
58 void AddAODtoTreeUserInfo();
59 void AddBranch(const char* cname, void* addobj);
60 Bool_t IsStandard() {return fIsStandard;}
61 Bool_t GetFillAOD(){return fFillAOD;}
62 Bool_t NeedsHeaderReplication() {return fNeedsHeaderReplication;}
63 Bool_t NeedsTracksBranchReplication() {return fNeedsTracksBranchReplication;}
64 Bool_t NeedsVerticesBranchReplication() {return fNeedsVerticesBranchReplication;}
65 Bool_t NeedsV0sBranchReplication() {return fNeedsV0sBranchReplication;}
66 Bool_t NeedsTrackletsBranchReplication() {return fNeedsTrackletsBranchReplication;}
67 Bool_t NeedsPMDClustersBranchReplication() {return fNeedsPMDClustersBranchReplication;}
68 Bool_t NeedsJetsBranchReplication() {return fNeedsJetsBranchReplication;}
69 Bool_t NeedsFMDClustersBranchReplication() {return fNeedsFMDClustersBranchReplication;}
70 Bool_t NeedsCaloClustersBranchReplication() {return fNeedsCaloClustersBranchReplication;}
71 Bool_t AODIsReplicated() {return fAODIsReplicated;}
73 void SetInputTree(TTree* /*tree*/) {;}
74 void SetMCEventHandler(AliMCEventHandler* mcH) {fMCEventH = mcH;} // For internal use
76 void StoreMCParticles();
77 void SetMCHeaderInfo(AliAODMCHeader *mcHeader,AliGenEventHeader *genHeader); // Utility function t catch different types of eventheaders
78 AliAODHandler(const AliAODHandler&); // Not implemented
79 AliAODHandler& operator=(const AliAODHandler&); // Not implemented
81 Bool_t fIsStandard; // Flag for standard aod creation
82 Bool_t fFillAOD; // Flag for filling of the AOD tree at the end (all or nothing)
83 Bool_t fNeedsHeaderReplication; // Flag for header replication
84 Bool_t fNeedsTracksBranchReplication; // Flag for tracks replication
85 Bool_t fNeedsVerticesBranchReplication; // Flag for vertices replication
86 Bool_t fNeedsV0sBranchReplication; // Flag for V0s replication
87 Bool_t fNeedsTrackletsBranchReplication; // Flag for Tracklets replication
88 Bool_t fNeedsPMDClustersBranchReplication; // Flag for PMDClusters replication
89 Bool_t fNeedsJetsBranchReplication; // Flag for Jets replication
90 Bool_t fNeedsFMDClustersBranchReplication; // Flag for FMDClusters replication
91 Bool_t fNeedsCaloClustersBranchReplication; // Flag for CaloClusters replication
92 Bool_t fAODIsReplicated; // Flag true if replication as been executed
93 AliAODEvent *fAODEvent; //! Pointer to the AOD event
94 AliMCEventHandler *fMCEventH; //! Pointer to mc event handler needed not to depend on the manager
95 TTree *fTreeA; //! tree for AOD persistency
96 TFile *fFileA; //! Output file
97 TString fFileName; // Output file name
98 ClassDef(AliAODHandler, 3)