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"
14 #include "AliAODExtension.h"
20 class AliMCEventHandler;
22 class AliGenEventHeader;
24 class AliAnalysisFilter;
26 class AliAODHandler : public AliVEventHandler {
30 AliAODHandler(const char* name, const char* title);
31 virtual ~AliAODHandler();
32 virtual void SetOutputFileName(const char* fname);
33 virtual const char* GetOutputFileName();
34 // Extra outputs as a string separated by commas
35 virtual const char* GetExtraOutputs() const;
36 virtual Bool_t Init(Option_t* option);
37 virtual Bool_t Init(TTree* /*tree*/, Option_t* /*option*/) {return kTRUE;}
38 virtual Bool_t GetEntry() {return kTRUE;}
40 virtual Bool_t BeginEvent(Long64_t /*entry*/) {fFillAOD=kFALSE; fFillExtension=kFALSE; return kTRUE;}
41 virtual Bool_t Notify() { return AliVEventHandler::Notify(); };
42 virtual Bool_t Notify(const char * /* path */) {return kTRUE;}
43 virtual Bool_t FinishEvent();
44 virtual Bool_t Terminate();
45 virtual Bool_t TerminateIO();
47 virtual void SetCreateNonStandardAOD() {Changed(); fIsStandard = kFALSE;}
48 virtual void SetFillAOD(Bool_t b) {Changed(); fFillAOD = b;}
49 virtual void SetFillExtension(Bool_t b) {Changed(); fFillExtension = b;}
50 virtual void SetFillAODforRun(Bool_t b) {Changed(); fFillAODRun = b;}
51 virtual void SetNeedsHeaderReplication() {fNeedsHeaderReplication = kTRUE;}
52 virtual void SetNeedsTracksBranchReplication() {fNeedsTracksBranchReplication = kTRUE;}
53 virtual void SetNeedsVerticesBranchReplication() {fNeedsVerticesBranchReplication = kTRUE;}
54 virtual void SetNeedsV0sBranchReplication() {fNeedsV0sBranchReplication = kTRUE;}
55 virtual void SetNeedsCascadesBranchReplication() {fNeedsCascadesBranchReplication = kTRUE;}
56 virtual void SetNeedsTrackletsBranchReplication() {fNeedsTrackletsBranchReplication = kTRUE;}
57 virtual void SetNeedsPMDClustersBranchReplication() {fNeedsPMDClustersBranchReplication = kTRUE;}
58 virtual void SetNeedsJetsBranchReplication() {fNeedsJetsBranchReplication = kTRUE;}
59 virtual void SetNeedsFMDClustersBranchReplication() {fNeedsFMDClustersBranchReplication = kTRUE;}
60 virtual void SetNeedsCaloClustersBranchReplication() {fNeedsCaloClustersBranchReplication = kTRUE;}
61 virtual void SetNeedsCaloTriggerBranchReplication() {fNeedsCaloTriggerBranchReplication = kTRUE;}
62 virtual void SetNeedsMCParticlesBranchReplication() {fNeedsMCParticlesBranchReplication = kTRUE;}
63 virtual void SetNeedsDimuonsBranchReplication() {fNeedsDimuonsBranchReplication = kTRUE;}
64 virtual void SetNeedsHMPIDBranchReplication() {fNeedsHMPIDBranchReplication = kTRUE;}
65 virtual void SetAODIsReplicated() {Changed(); fAODIsReplicated = kTRUE;}
66 virtual void SetAODExtensionMode() {Changed(); fIsStandard=kFALSE; fFillAOD=kFALSE; fFillAODRun=kFALSE; fFillExtension=kTRUE;}
68 AliAODEvent* GetAOD() {return fAODEvent;}
69 virtual TTree* GetTree() const {return fTreeA;}
70 TObjArray* GetExtensions() const {return fExtensions;}
71 AliAODExtension* GetExtension(const char *filename) const;
72 TObjArray* GetFilters() const {return fFilters;}
73 AliAODExtension* GetFilteredAOD(const char *filename) const;
74 void CreateTree(Int_t flag);
76 void AddAODtoTreeUserInfo();
77 void AddBranch(const char* cname, void* addobj, const char *fname="");
79 AliAODExtension* AddExtension(const char *filename, const char *title="");
80 AliAODExtension* AddFilteredAOD(const char *filename, const char *filtername);
81 // AliAODExtension* FindExtensionContainingBranch(const char* bname) const;
82 Bool_t IsStandard() const {return fIsStandard;}
83 Bool_t GetFillAOD() const {return fFillAOD;}
84 Bool_t NeedsHeaderReplication() const {return fNeedsHeaderReplication;}
85 Bool_t NeedsTracksBranchReplication() const {return fNeedsTracksBranchReplication;}
86 Bool_t NeedsVerticesBranchReplication() const {return fNeedsVerticesBranchReplication;}
87 Bool_t NeedsV0sBranchReplication() const {return fNeedsV0sBranchReplication;}
88 Bool_t NeedsCascadesBranchReplication() const {return fNeedsCascadesBranchReplication;}
89 Bool_t NeedsTrackletsBranchReplication() const {return fNeedsTrackletsBranchReplication;}
90 Bool_t NeedsPMDClustersBranchReplication() const {return fNeedsPMDClustersBranchReplication;}
91 Bool_t NeedsJetsBranchReplication() const {return fNeedsJetsBranchReplication;}
92 Bool_t NeedsFMDClustersBranchReplication() const {return fNeedsFMDClustersBranchReplication;}
93 Bool_t NeedsCaloClustersBranchReplication() const {return fNeedsCaloClustersBranchReplication;}
94 Bool_t NeedsCaloTriggerBranchReplication() const {return fNeedsCaloTriggerBranchReplication;}
95 Bool_t NeedsMCParticlesBranchReplication() const {return fNeedsMCParticlesBranchReplication;}
96 Bool_t NeedsDimuonsBranchReplication() const {return fNeedsDimuonsBranchReplication;}
97 Bool_t NeedsHMPIDBranchReplication() const {return fNeedsHMPIDBranchReplication;}
98 Bool_t AODIsReplicated() const {return fAODIsReplicated;}
100 void SetInputTree(TTree* /*tree*/) {;}
101 void SetMCEventHandler(AliMCEventHandler* mcH) {fMCEventH = mcH;} // For internal use
102 void StoreMCParticles(); // Store MC particles, only to be called from AliAnalyisTaskMCParticleFilter
104 Bool_t HasExtensions() const;
106 void Print(Option_t* opt="") const;
109 void SetMCHeaderInfo(AliAODMCHeader *mcHeader,AliGenEventHeader *genHeader); // Utility function t catch different types of eventheaders
110 AliAODHandler(const AliAODHandler&); // Not implemented
111 AliAODHandler& operator=(const AliAODHandler&); // Not implemented
112 void PrintExtensions(const TObjArray& array) const;
115 Bool_t fIsStandard; // Flag for standard aod creation
116 Bool_t fFillAOD; // Flag for filling of the AOD tree at the end (all or nothing evt by evt)
117 Bool_t fFillAODRun; // Flag for filling of the AOD tree at the end (run)
118 Bool_t fFillExtension; // Flag for filling or the delta AOD tree at the end
119 Bool_t fNeedsHeaderReplication; // Flag for header replication
120 Bool_t fNeedsTracksBranchReplication; // Flag for tracks replication
121 Bool_t fNeedsVerticesBranchReplication; // Flag for vertices replication
122 Bool_t fNeedsV0sBranchReplication; // Flag for V0s replication
123 Bool_t fNeedsCascadesBranchReplication; // Flag for Cascade replication
124 Bool_t fNeedsTrackletsBranchReplication; // Flag for Tracklets replication
125 Bool_t fNeedsPMDClustersBranchReplication; // Flag for PMDClusters replication
126 Bool_t fNeedsJetsBranchReplication; // Flag for Jets replication
127 Bool_t fNeedsFMDClustersBranchReplication; // Flag for FMDClusters replication
128 Bool_t fNeedsCaloClustersBranchReplication; // Flag for CaloClusters replication
129 Bool_t fNeedsCaloTriggerBranchReplication; // Flag for Calo Trigger replication
130 Bool_t fNeedsMCParticlesBranchReplication; // Flag for MCParticles replication
131 Bool_t fNeedsDimuonsBranchReplication; // Flag for Dimuons replication
132 Bool_t fNeedsHMPIDBranchReplication; // Flag for HMPID replication
133 Bool_t fAODIsReplicated; // Flag true if replication as been executed
134 AliAODEvent *fAODEvent; //! Pointer to the AOD event
135 AliMCEventHandler *fMCEventH; //! Pointer to mc event handler needed not to depend on the manager
136 TTree *fTreeA; //! tree for AOD persistency
137 TFile *fFileA; //! Output file
138 TString fFileName; // Output file name
139 TObjArray *fExtensions; // List of extensions
140 TObjArray *fFilters; // List of filtered AOD's
142 ClassDef(AliAODHandler, 7)