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

#ifndef ALIAODHANDLER_H
#define ALIAODHANDLER_H
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//-------------------------------------------------------------------------
//     Implementation of the Event Handler Interface for AOD
//     Author: Andreas Morsch, CERN
//-------------------------------------------------------------------------

#include "AliVEventHandler.h"
#include "AliAODExtension.h"

class AliAODEvent;
class TFile;
class TTree;
class TObjArray;
class AliMCEventHandler;
class AliAODMCHeader;
class AliGenEventHeader;
class TMap;
class AliAnalysisFilter;

class AliAODHandler : public AliVEventHandler {
    
 public:
    AliAODHandler();
    AliAODHandler(const char* name, const char* title);
    virtual ~AliAODHandler();
    virtual void         SetOutputFileName(const char* fname);
    virtual const char*  GetOutputFileName();
    // Extra outputs as a string separated by commas
    virtual const char*  GetExtraOutputs() const;
    virtual Bool_t       Init(Option_t* option);
    virtual Bool_t       Init(TTree* /*tree*/, Option_t* /*option*/)  {return kTRUE;}
    virtual Bool_t       GetEntry() {return kTRUE;}
	    
    virtual Bool_t       BeginEvent(Long64_t /*entry*/) {fFillAOD=kFALSE; fFillExtension=kFALSE; return kTRUE;}
    virtual Bool_t       Notify() { return AliVEventHandler::Notify(); };
    virtual Bool_t       Notify(const char * /* path */) {return kTRUE;}
    virtual Bool_t       FinishEvent();
    virtual Bool_t       Terminate();
    virtual Bool_t       TerminateIO();
    //
    virtual void         SetCreateNonStandardAOD()               {fIsStandard = kFALSE;}
    virtual void         SetFillAOD(Bool_t b)                    {fFillAOD = b;}
    virtual void         SetFillExtension(Bool_t b)              {fFillExtension = b;}
    virtual void         SetFillAODforRun(Bool_t b)              {fFillAODRun = b;}
    virtual void         SetNeedsHeaderReplication()             {fNeedsHeaderReplication             = kTRUE;}
    virtual void         SetNeedsTracksBranchReplication()       {fNeedsTracksBranchReplication       = kTRUE;}
    virtual void         SetNeedsVerticesBranchReplication()     {fNeedsVerticesBranchReplication     = kTRUE;}
    virtual void         SetNeedsV0sBranchReplication()          {fNeedsV0sBranchReplication          = kTRUE;}
    virtual void         SetNeedsCascadesBranchReplication()     {fNeedsCascadesBranchReplication     = kTRUE;}
    virtual void         SetNeedsTrackletsBranchReplication()    {fNeedsTrackletsBranchReplication    = kTRUE;}
    virtual void         SetNeedsPMDClustersBranchReplication()  {fNeedsPMDClustersBranchReplication  = kTRUE;}
    virtual void         SetNeedsJetsBranchReplication()         {fNeedsJetsBranchReplication         = kTRUE;}
    virtual void         SetNeedsFMDClustersBranchReplication()  {fNeedsFMDClustersBranchReplication  = kTRUE;}
    virtual void         SetNeedsCaloClustersBranchReplication() {fNeedsCaloClustersBranchReplication = kTRUE;}
    virtual void         SetNeedsMCParticlesBranchReplication()  {fNeedsMCParticlesBranchReplication  = kTRUE;}
    virtual void         SetNeedsDimuonsBranchReplication()      {fNeedsDimuonsBranchReplication      = kTRUE;}
    virtual void         SetAODIsReplicated() {fAODIsReplicated = kTRUE;}
    virtual void         SetAODExtensionMode()                   {fIsStandard=kFALSE; fFillAOD=kFALSE; fFillAODRun=kFALSE; fFillExtension=kTRUE;}
    //
    AliAODEvent*         GetAOD()  {return fAODEvent;}
    virtual TTree*       GetTree() const {return fTreeA;}
    TObjArray*           GetExtensions() const {return fExtensions;}
    AliAODExtension*     GetExtension(const char *filename) const;
    TObjArray*           GetFilters() const {return fFilters;}
    AliAODExtension*     GetFilteredAOD(const char *filename) const;
    void                 CreateTree(Int_t flag);
    void                 FillTree();
    void                 AddAODtoTreeUserInfo();
    void                 AddBranch(const char* cname, void* addobj, const char *fname="");
  
    AliAODExtension*     AddExtension(const char *filename, const char *title="");                 
    AliAODExtension*     AddFilteredAOD(const char *filename, const char *filtername);
//    AliAODExtension*     FindExtensionContainingBranch(const char* bname) const;
    Bool_t               IsStandard()                         const {return fIsStandard;}
    Bool_t               GetFillAOD()                         const {return fFillAOD;} 
    Bool_t               NeedsHeaderReplication()             const {return  fNeedsHeaderReplication;}
    Bool_t               NeedsTracksBranchReplication()       const {return  fNeedsTracksBranchReplication;}
    Bool_t               NeedsVerticesBranchReplication()     const {return  fNeedsVerticesBranchReplication;}
    Bool_t               NeedsV0sBranchReplication()          const {return  fNeedsV0sBranchReplication;}
    Bool_t               NeedsCascadesBranchReplication()     const {return  fNeedsCascadesBranchReplication;}
    Bool_t               NeedsTrackletsBranchReplication()    const {return  fNeedsTrackletsBranchReplication;}
    Bool_t               NeedsPMDClustersBranchReplication()  const {return  fNeedsPMDClustersBranchReplication;}
    Bool_t               NeedsJetsBranchReplication()         const {return  fNeedsJetsBranchReplication;}
    Bool_t               NeedsFMDClustersBranchReplication()  const {return  fNeedsFMDClustersBranchReplication;}
    Bool_t               NeedsCaloClustersBranchReplication() const {return  fNeedsCaloClustersBranchReplication;}
    Bool_t               NeedsMCParticlesBranchReplication()  const {return  fNeedsMCParticlesBranchReplication;}
    Bool_t               NeedsDimuonsBranchReplication()      const {return  fNeedsDimuonsBranchReplication;}
    Bool_t               AODIsReplicated()                    const {return  fAODIsReplicated;}
    //
    void                 SetInputTree(TTree* /*tree*/) {;}
    void                 SetMCEventHandler(AliMCEventHandler* mcH) {fMCEventH = mcH;} // For internal use
    void StoreMCParticles(); // Store MC particles, only to be called from AliAnalyisTaskMCParticleFilter

  Bool_t HasExtensions() const;
  
  void Print(Option_t* opt="") const;
  
 private:
  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
  void PrintExtensions(const TObjArray& array) const;

 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 evt by evt)
    Bool_t                   fFillAODRun;                         // Flag for filling of the AOD tree at the end (run)
    Bool_t                   fFillExtension;                      // Flag for filling or the delta AOD tree at the end
    Bool_t                   fNeedsHeaderReplication;             // Flag for header replication
    Bool_t                   fNeedsTracksBranchReplication;       // Flag for tracks replication
    Bool_t                   fNeedsVerticesBranchReplication;     // Flag for vertices replication
    Bool_t                   fNeedsV0sBranchReplication;          // Flag for V0s replication
    Bool_t                   fNeedsCascadesBranchReplication;     // Flag for Cascade replication
    Bool_t                   fNeedsTrackletsBranchReplication;    // Flag for Tracklets replication
    Bool_t                   fNeedsPMDClustersBranchReplication;  // Flag for PMDClusters replication
    Bool_t                   fNeedsJetsBranchReplication;         // Flag for Jets replication
    Bool_t                   fNeedsFMDClustersBranchReplication;  // Flag for FMDClusters replication
    Bool_t                   fNeedsCaloClustersBranchReplication; // Flag for CaloClusters replication
    Bool_t                   fNeedsMCParticlesBranchReplication;  // Flag for MCParticles replication
    Bool_t                   fNeedsDimuonsBranchReplication;      // Flag for Dimuons 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
    TObjArray               *fExtensions;             //  List of extensions
    TObjArray               *fFilters;                //  List of filtered AOD's

  ClassDef(AliAODHandler, 7)
};

#endif
Commit	Line	Data
ec4af4c1	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 */
	5
	6	/* $Id$ */
	7
	8	//-------------------------------------------------------------------------
	9	// Implementation of the Event Handler Interface for AOD
	10	// Author: Andreas Morsch, CERN
	11	//-------------------------------------------------------------------------
	12
f3214a54	13	#include "AliVEventHandler.h"
d815d7c4	14	#include "AliAODExtension.h"
ec4af4c1	15
	16	class AliAODEvent;
	17	class TFile;
	18	class TTree;
9066c676	19	class TObjArray;
da97a08a	20	class AliMCEventHandler;
dce1b636	21	class AliAODMCHeader;
dce1b636	22	class AliGenEventHeader;
26ba01d4	23	class TMap;
26ba01d4	24	class AliAnalysisFilter;
ec4af4c1	25
f3214a54	26	class AliAODHandler : public AliVEventHandler {
ec4af4c1	27
	28	public:
	29	AliAODHandler();
	30	AliAODHandler(const char* name, const char* title);
	31	virtual ~AliAODHandler();
7970f4ac	32	virtual void SetOutputFileName(const char* fname);
7970f4ac	33	virtual const char* GetOutputFileName();
90d50a8c	34	// Extra outputs as a string separated by commas
90d50a8c	35	virtual const char* GetExtraOutputs() const;
300d5701	36	virtual Bool_t Init(Option_t* option);
300d5701	37	virtual Bool_t Init(TTree* /tree/, Option_t* /option/) {return kTRUE;}
0d6a82a5	38	virtual Bool_t GetEntry() {return kTRUE;}
0d6a82a5	39
307b6330	40	virtual Bool_t BeginEvent(Long64_t /entry/) {fFillAOD=kFALSE; fFillExtension=kFALSE; return kTRUE;}
276be602	41	virtual Bool_t Notify() { return AliVEventHandler::Notify(); };
890126ab	42	virtual Bool_t Notify(const char * /* path */) {return kTRUE;}
5f380da9	43	virtual Bool_t FinishEvent();
ec4af4c1	44	virtual Bool_t Terminate();
ec4af4c1	45	virtual Bool_t TerminateIO();
ec4af4c1	46	//
41b01ae4	47	virtual void SetCreateNonStandardAOD() {fIsStandard = kFALSE;}
41b01ae4	48	virtual void SetFillAOD(Bool_t b) {fFillAOD = b;}
307b6330	49	virtual void SetFillExtension(Bool_t b) {fFillExtension = b;}
41b01ae4	50	virtual void SetFillAODforRun(Bool_t b) {fFillAODRun = b;}
3549c522	51	virtual void SetNeedsHeaderReplication() {fNeedsHeaderReplication = kTRUE;}
	52	virtual void SetNeedsTracksBranchReplication() {fNeedsTracksBranchReplication = kTRUE;}
	53	virtual void SetNeedsVerticesBranchReplication() {fNeedsVerticesBranchReplication = kTRUE;}
	54	virtual void SetNeedsV0sBranchReplication() {fNeedsV0sBranchReplication = kTRUE;}
e0107012	55	virtual void SetNeedsCascadesBranchReplication() {fNeedsCascadesBranchReplication = kTRUE;}
3549c522	56	virtual void SetNeedsTrackletsBranchReplication() {fNeedsTrackletsBranchReplication = kTRUE;}
	57	virtual void SetNeedsPMDClustersBranchReplication() {fNeedsPMDClustersBranchReplication = kTRUE;}
	58	virtual void SetNeedsJetsBranchReplication() {fNeedsJetsBranchReplication = kTRUE;}
	59	virtual void SetNeedsFMDClustersBranchReplication() {fNeedsFMDClustersBranchReplication = kTRUE;}
75754ba8	60	virtual void SetNeedsCaloClustersBranchReplication() {fNeedsCaloClustersBranchReplication = kTRUE;}
fa8b0e56	61	virtual void SetNeedsMCParticlesBranchReplication() {fNeedsMCParticlesBranchReplication = kTRUE;}
866d8d78	62	virtual void SetNeedsDimuonsBranchReplication() {fNeedsDimuonsBranchReplication = kTRUE;}
75754ba8	63	virtual void SetAODIsReplicated() {fAODIsReplicated = kTRUE;}
ca2834f6	64	virtual void SetAODExtensionMode() {fIsStandard=kFALSE; fFillAOD=kFALSE; fFillAODRun=kFALSE; fFillExtension=kTRUE;}
78f7f935	65	//
ec4af4c1	66	AliAODEvent* GetAOD() {return fAODEvent;}
11326411	67	virtual TTree* GetTree() const {return fTreeA;}
9066c676	68	TObjArray* GetExtensions() const {return fExtensions;}
582cfeb5	69	AliAODExtension* GetExtension(const char *filename) const;
	70	TObjArray* GetFilters() const {return fFilters;}
	71	AliAODExtension* GetFilteredAOD(const char *filename) const;
954526ed	72	void CreateTree(Int_t flag);
ec4af4c1	73	void FillTree();
ec4af4c1	74	void AddAODtoTreeUserInfo();
9066c676	75	void AddBranch(const char* cname, void* addobj, const char *fname="");
14d6fad5	76
9066c676	77	AliAODExtension* AddExtension(const char filename, const char title="");
582cfeb5	78	AliAODExtension* AddFilteredAOD(const char filename, const char filtername);
14d6fad5	79	// AliAODExtension* FindExtensionContainingBranch(const char* bname) const;
3549c522	80	Bool_t IsStandard() const {return fIsStandard;}
	81	Bool_t GetFillAOD() const {return fFillAOD;}
	82	Bool_t NeedsHeaderReplication() const {return fNeedsHeaderReplication;}
	83	Bool_t NeedsTracksBranchReplication() const {return fNeedsTracksBranchReplication;}
	84	Bool_t NeedsVerticesBranchReplication() const {return fNeedsVerticesBranchReplication;}
	85	Bool_t NeedsV0sBranchReplication() const {return fNeedsV0sBranchReplication;}
e0107012	86	Bool_t NeedsCascadesBranchReplication() const {return fNeedsCascadesBranchReplication;}
3549c522	87	Bool_t NeedsTrackletsBranchReplication() const {return fNeedsTrackletsBranchReplication;}
	88	Bool_t NeedsPMDClustersBranchReplication() const {return fNeedsPMDClustersBranchReplication;}
	89	Bool_t NeedsJetsBranchReplication() const {return fNeedsJetsBranchReplication;}
	90	Bool_t NeedsFMDClustersBranchReplication() const {return fNeedsFMDClustersBranchReplication;}
	91	Bool_t NeedsCaloClustersBranchReplication() const {return fNeedsCaloClustersBranchReplication;}
	92	Bool_t NeedsMCParticlesBranchReplication() const {return fNeedsMCParticlesBranchReplication;}
866d8d78	93	Bool_t NeedsDimuonsBranchReplication() const {return fNeedsDimuonsBranchReplication;}
3549c522	94	Bool_t AODIsReplicated() const {return fAODIsReplicated;}
933fd60f	95	//
933fd60f	96	void SetInputTree(TTree* /tree/) {;}
da97a08a	97	void SetMCEventHandler(AliMCEventHandler* mcH) {fMCEventH = mcH;} // For internal use
522aca31	98	void StoreMCParticles(); // Store MC particles, only to be called from AliAnalyisTaskMCParticleFilter
522aca31	99
14d6fad5	100	Bool_t HasExtensions() const;
14d6fad5	101
26ba01d4	102	void Print(Option_t* opt="") const;
26ba01d4	103
ec4af4c1	104	private:
14d6fad5	105	void SetMCHeaderInfo(AliAODMCHeader mcHeader,AliGenEventHeader genHeader); // Utility function t catch different types of eventheaders
	106	AliAODHandler(const AliAODHandler&); // Not implemented
	107	AliAODHandler& operator=(const AliAODHandler&); // Not implemented
b5b9155e	108	void PrintExtensions(const TObjArray& array) const;
14d6fad5	109
ec4af4c1	110	private:
75754ba8	111	Bool_t fIsStandard; // Flag for standard aod creation
41b01ae4	112	Bool_t fFillAOD; // Flag for filling of the AOD tree at the end (all or nothing evt by evt)
41b01ae4	113	Bool_t fFillAODRun; // Flag for filling of the AOD tree at the end (run)
ca2834f6	114	Bool_t fFillExtension; // Flag for filling or the delta AOD tree at the end
75754ba8	115	Bool_t fNeedsHeaderReplication; // Flag for header replication
	116	Bool_t fNeedsTracksBranchReplication; // Flag for tracks replication
	117	Bool_t fNeedsVerticesBranchReplication; // Flag for vertices replication
	118	Bool_t fNeedsV0sBranchReplication; // Flag for V0s replication
e0107012	119	Bool_t fNeedsCascadesBranchReplication; // Flag for Cascade replication
75754ba8	120	Bool_t fNeedsTrackletsBranchReplication; // Flag for Tracklets replication
	121	Bool_t fNeedsPMDClustersBranchReplication; // Flag for PMDClusters replication
	122	Bool_t fNeedsJetsBranchReplication; // Flag for Jets replication
	123	Bool_t fNeedsFMDClustersBranchReplication; // Flag for FMDClusters replication
	124	Bool_t fNeedsCaloClustersBranchReplication; // Flag for CaloClusters replication
3549c522	125	Bool_t fNeedsMCParticlesBranchReplication; // Flag for MCParticles replication
866d8d78	126	Bool_t fNeedsDimuonsBranchReplication; // Flag for Dimuons replication
75754ba8	127	Bool_t fAODIsReplicated; // Flag true if replication as been executed
7c3a9fbf	128	AliAODEvent *fAODEvent; //! Pointer to the AOD event
da97a08a	129	AliMCEventHandler *fMCEventH; //! Pointer to mc event handler needed not to depend on the manager
7c3a9fbf	130	TTree *fTreeA; //! tree for AOD persistency
	131	TFile *fFileA; //! Output file
	132	TString fFileName; // Output file name
9066c676	133	TObjArray *fExtensions; // List of extensions
e0107012	134	TObjArray *fFilters; // List of filtered AOD's
582cfeb5	135
ca2834f6	136	ClassDef(AliAODHandler, 7)
582cfeb5	137	};
26ba01d4	138
ec4af4c1	139	#endif