STEER/AliAODHandler.h

   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
  13 #include "AliVEventHandler.h"
  14
  15 class AliAODEvent;
  16 class TFile;
  17 class TTree;
  18 class TObjArray;
  19 class AliMCEventHandler;
  20 class AliAODMCHeader;
  21 class AliAODExtension;
  22 class AliGenEventHeader;
  23
  24
  25
  26 class AliAODHandler : public AliVEventHandler {
  27
  28  public:
  29     AliAODHandler();
  30     AliAODHandler(const char* name, const char* title);
  31     virtual ~AliAODHandler();
  32     virtual void         SetOutputFileName(const char* fname);
  33     virtual const char*  GetOutputFileName();
  34     virtual Bool_t       Init(Option_t* option);
  35     virtual Bool_t       Init(TTree* /*tree*/, Option_t* /*option*/)  {return kTRUE;}
  36     virtual Bool_t       BeginEvent(Long64_t /*entry*/){return kTRUE;}
  37     virtual Bool_t       Notify() { return AliVEventHandler::Notify(); };
  38     virtual Bool_t       Notify(const char * /* path */) {return kTRUE;}
  39     virtual Bool_t       FinishEvent();
  40     virtual Bool_t       Terminate();
  41     virtual Bool_t       TerminateIO();
  42     //
  43     virtual void         SetCreateNonStandardAOD()   {fIsStandard = kFALSE;}
  44     virtual void         SetFillAOD(Bool_t b)      {fFillAOD = b;}
  45     virtual void         SetNeedsHeaderReplication()             {fNeedsHeaderReplication             = kTRUE;}
  46     virtual void         SetNeedsTracksBranchReplication()       {fNeedsTracksBranchReplication       = kTRUE;}
  47     virtual void         SetNeedsVerticesBranchReplication()     {fNeedsVerticesBranchReplication     = kTRUE;}
  48     virtual void         SetNeedsV0sBranchReplication()          {fNeedsV0sBranchReplication          = kTRUE;}
  49     virtual void         SetNeedsTrackletsBranchReplication()    {fNeedsTrackletsBranchReplication    = kTRUE;}
  50     virtual void         SetNeedsPMDClustersBranchReplication()  {fNeedsPMDClustersBranchReplication  = kTRUE;}
  51     virtual void         SetNeedsJetsBranchReplication()         {fNeedsJetsBranchReplication         = kTRUE;}
  52     virtual void         SetNeedsFMDClustersBranchReplication()  {fNeedsFMDClustersBranchReplication  = kTRUE;}
  53     virtual void         SetNeedsCaloClustersBranchReplication() {fNeedsCaloClustersBranchReplication = kTRUE;}
  54     virtual void         SetNeedsMCParticlesBranchReplication()  {fNeedsCaloClustersBranchReplication = kTRUE;}
  55     virtual void         SetAODIsReplicated() {fAODIsReplicated = kTRUE;}
  56     //
  57     AliAODEvent*         GetAOD()  {return fAODEvent;}
  58     virtual TTree*       GetTree() const {return fTreeA;}
  59     TObjArray*           GetExtensions() const {return fExtensions;}
  60     AliAODExtension*     GetExtension(const char *filename) const;
  61     TObjArray*           GetFilters() const {return fFilters;}
  62     AliAODExtension*     GetFilteredAOD(const char *filename) const;
  63     void                 CreateTree(Int_t flag);
  64     void                 FillTree();
  65     void                 AddAODtoTreeUserInfo();
  66     void                 AddBranch(const char* cname, void* addobj, const char *fname="");
  67     AliAODExtension*     AddExtension(const char *filename, const char *title="");
  68     AliAODExtension*     AddFilteredAOD(const char *filename, const char *filtername);
  69     Bool_t               IsStandard()                         const {return fIsStandard;}
  70     Bool_t               GetFillAOD()                         const {return fFillAOD;}
  71     Bool_t               NeedsHeaderReplication()             const {return  fNeedsHeaderReplication;}
  72     Bool_t               NeedsTracksBranchReplication()       const {return  fNeedsTracksBranchReplication;}
  73     Bool_t               NeedsVerticesBranchReplication()     const {return  fNeedsVerticesBranchReplication;}
  74     Bool_t               NeedsV0sBranchReplication()          const {return  fNeedsV0sBranchReplication;}
  75     Bool_t               NeedsTrackletsBranchReplication()    const {return  fNeedsTrackletsBranchReplication;}
  76     Bool_t               NeedsPMDClustersBranchReplication()  const {return  fNeedsPMDClustersBranchReplication;}
  77     Bool_t               NeedsJetsBranchReplication()         const {return  fNeedsJetsBranchReplication;}
  78     Bool_t               NeedsFMDClustersBranchReplication()  const {return  fNeedsFMDClustersBranchReplication;}
  79     Bool_t               NeedsCaloClustersBranchReplication() const {return  fNeedsCaloClustersBranchReplication;}
  80     Bool_t               NeedsMCParticlesBranchReplication()  const {return  fNeedsMCParticlesBranchReplication;}
  81     Bool_t               AODIsReplicated()                    const {return  fAODIsReplicated;}
  82     //
  83     void                 SetInputTree(TTree* /*tree*/) {;}
  84     void                 SetMCEventHandler(AliMCEventHandler* mcH) {fMCEventH = mcH;} // For internal use
  85     void StoreMCParticles(); // Store MC particles, only to be called from AliAnalyisTaskMCParticleFilter
  86
  87  private:
  88     void SetMCHeaderInfo(AliAODMCHeader *mcHeader,AliGenEventHeader *genHeader); // Utility function t catch different types of eventheaders
  89     AliAODHandler(const AliAODHandler&);             // Not implemented
  90     AliAODHandler& operator=(const AliAODHandler&);  // Not implemented
  91  private:
  92     Bool_t                   fIsStandard;                         // Flag for standard aod creation
  93     Bool_t                   fFillAOD;                            // Flag for filling of the AOD tree at the end (all or nothing)
  94     Bool_t                   fNeedsHeaderReplication;             // Flag for header replication
  95     Bool_t                   fNeedsTracksBranchReplication;       // Flag for tracks replication
  96     Bool_t                   fNeedsVerticesBranchReplication;     // Flag for vertices replication
  97     Bool_t                   fNeedsV0sBranchReplication;          // Flag for V0s replication
  98     Bool_t                   fNeedsTrackletsBranchReplication;    // Flag for Tracklets replication
  99     Bool_t                   fNeedsPMDClustersBranchReplication;  // Flag for PMDClusters replication
 100     Bool_t                   fNeedsJetsBranchReplication;         // Flag for Jets replication
 101     Bool_t                   fNeedsFMDClustersBranchReplication;  // Flag for FMDClusters replication
 102     Bool_t                   fNeedsCaloClustersBranchReplication; // Flag for CaloClusters replication
 103     Bool_t                   fNeedsMCParticlesBranchReplication;  // Flag for MCParticles replication
 104     Bool_t                   fAODIsReplicated;                    // Flag true if replication as been executed
 105     AliAODEvent             *fAODEvent;               //! Pointer to the AOD event
 106     AliMCEventHandler       *fMCEventH;               //! Pointer to mc event handler needed not to depend on the manager
 107     TTree                   *fTreeA;                  //! tree for AOD persistency
 108     TFile                   *fFileA;                  //! Output file
 109     TString                  fFileName;               //  Output file name
 110     TObjArray               *fExtensions;             //  List of extensions
 111     TObjArray               *fFilters;                // List of filtered AOD's
 112     ClassDef(AliAODHandler, 5)
 113 };
 114
 115 //-------------------------------------------------------------------------
 116 //     Support class for AOD extensions. This is created by the user analysis
 117 //     that requires a separate file for some AOD branches. The name of the
 118 //     AliAODExtension object is the file name where the AOD branches will be
 119 //     stored.
 120 //     Author: Andrei Gheata, CERN
 121 //-------------------------------------------------------------------------
 122
 123 class AliAODExtension : public TNamed {
 124
 125 public:
 126
 127 enum EAliAODExtensionFlags {
 128    kFilteredAOD      = BIT(14)
 129 };
 130
 131     AliAODExtension() : TNamed(), fAODEvent(0), fTreeE(0), fFileE(0), fNtotal(0), fNpassed(0), fSelected(kFALSE) {}
 132     AliAODExtension(const char* name, const char* title, Bool_t isfilter=kFALSE);
 133     virtual ~AliAODExtension();
 134     void                 AddBranch(const char* cname, void* addobj);
 135     Bool_t               FinishEvent();
 136     Int_t                GetNtotal() const         {return fNtotal;}
 137     Int_t                GetNpassed() const        {return fNpassed;}
 138     const char*          GetOutputFileName() const {return TNamed::GetName();}
 139     AliAODEvent*         GetAOD() const            {return fAODEvent;}
 140     TTree*               GetTree() const           {return fTreeE;}
 141     Bool_t               Init(Option_t *option);
 142     Bool_t               IsFilteredAOD() const     {return TObject::TestBit(kFilteredAOD);}
 143     Bool_t               IsEventSelected() const   {return fSelected;}
 144     void                 SelectEvent(Bool_t flag=kTRUE)  {fSelected = flag;}
 145     void                 SetEvent(AliAODEvent *event);
 146     void                 SetOutputFileName(const char* fname) {TNamed::SetName(fname);}
 147     Bool_t               TerminateIO();
 148  private:
 149     AliAODExtension(const AliAODExtension&);             // Not implemented
 150     AliAODExtension& operator=(const AliAODExtension&);  // Not implemented
 151
 152  private:
 153     AliAODEvent             *fAODEvent;               //! Pointer to the AOD event
 154     TTree                   *fTreeE;                  //! tree for AOD persistency
 155     TFile                   *fFileE;                  //! Output file
 156     Int_t                    fNtotal;                 //! Number of processed events
 157     Int_t                    fNpassed;                //! Number of events that passed the filter
 158     Bool_t                   fSelected;               //! Select current event for filtered AOD's. Made false at event start.
 159     ClassDef(AliAODExtension, 1)                      // Support for extra AOD branches in a separate AOD file
 160 };
 161 #endif