STEER/STEERBase/AliInputEventHandler.h

   1 #ifndef ALIINPUTEVENTHANDLER_H
   2 #define ALIINPUTEVENTHANDLER_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 //     Input Handler realisation of the AliVEventHandler interface
  10 //     Author: Andreas Morsch, CERN
  11 //-------------------------------------------------------------------------
  12
  13 #include "AliVEventHandler.h"
  14 #include <TTree.h>
  15
  16
  17 class AliVEvent;
  18 class AliVCuts;
  19 class AliRunTag;
  20 class AliEventTag;
  21 class AliPIDResponse;
  22
  23 class AliInputEventHandler : public AliVEventHandler {
  24
  25  public:
  26  enum EInputHandlerFlags {
  27     kUserCallSelectionMask = BIT(14), // Watch out for defining base class bits
  28     kCheckStatistics       = BIT(15)
  29  };
  30     AliInputEventHandler();
  31     AliInputEventHandler(const char* name, const char* title);
  32     virtual ~AliInputEventHandler();
  33     virtual void         SetInputFileName(const char* fname);
  34     virtual const char  *GetInputFileName() const                     {return fInputFileName;}
  35     virtual void         SetOutputFileName(const char* /*fname*/) {;}
  36     virtual const char  *GetOutputFileName() const                    {return 0;}
  37     virtual Bool_t       Init(Option_t* opt) {if(fMixingHandler) fMixingHandler->Init(opt);return kTRUE;}
  38     virtual Bool_t       Init(TTree* tree, Option_t* opt) {if(fMixingHandler) fMixingHandler->Init(tree,opt);return kTRUE;}
  39     virtual Bool_t       GetEntry() {if(fMixingHandler) fMixingHandler->GetEntry(); return kTRUE;}
  40     virtual Bool_t       BeginEvent(Long64_t entry) {if(fMixingHandler) fMixingHandler->BeginEvent(entry);return kTRUE;}
  41
  42     virtual Bool_t       Notify()      { return AliVEventHandler::Notify();}
  43     virtual Bool_t       Notify(const char *path) {if(fMixingHandler) fMixingHandler->Notify(path);return kTRUE;}
  44     virtual Bool_t       FinishEvent() {if(fMixingHandler) fMixingHandler->FinishEvent();return kTRUE;}
  45     virtual Bool_t       Terminate()   {if(fMixingHandler) fMixingHandler->Terminate();return kTRUE;}
  46     virtual Bool_t       TerminateIO() {if(fMixingHandler) fMixingHandler->TerminateIO();return kTRUE;}
  47
  48     // Setters
  49     virtual void         SetInputTree(TTree* tree)                    {fTree = tree;}
  50     virtual void         SetEventSelection(AliVCuts* cuts)            {if (fEventCuts) Changed(); fEventCuts = cuts;}
  51     virtual void         SetUserCallSelectionMask(Bool_t flag=kTRUE)  {TObject::SetBit(kUserCallSelectionMask,flag);}
  52     virtual void         SetCheckStatistics(Bool_t flag=kTRUE)        {Changed(); TObject::SetBit(kCheckStatistics,flag);}
  53     //
  54     void SetInactiveBranches(const char* branches) {Changed(); fBranches   = branches;}
  55     void SetActiveBranches  (const char* branches) {Changed(); fBranchesOn = branches;}
  56      // Getters
  57     virtual AliVEvent   *GetEvent()        const                      {return 0;}
  58     virtual const AliEventTag   *GetEventTag() const                  {return 0;}
  59     virtual AliRunTag   *GetRunTag()       const                      {return 0;}
  60     // Get the statistics object (currently TH2). Option can be BIN0.
  61     virtual TObject     *GetStatistics(Option_t *option="") const;
  62     virtual Option_t    *GetAnalysisType() const                      {return 0;}
  63     virtual TTree       *GetTree( )        const                      {return fTree;}
  64     virtual AliVCuts    *GetEventSelection() const                    {return fEventCuts;}
  65     virtual Long64_t     GetReadEntry()    const;
  66     virtual Bool_t       IsCheckStatistics() const                    {return TObject::TestBit(kCheckStatistics);}
  67     virtual Bool_t       IsUserCallSelectionMask() const              {return TObject::TestBit(kUserCallSelectionMask);}
  68     virtual Bool_t       NewEvent()
  69         {Bool_t ne = fNewEvent; fNewEvent = kFALSE; return ne;}
  70     virtual UInt_t       IsEventSelected()
  71         {return fIsSelectedResult;}
  72     TList       *GetUserInfo() const                         {return fUserInfo;}
  73     // Mixing
  74     void SetMixingHandler(AliInputEventHandler* mixing) {Changed(); fMixingHandler = mixing;}
  75     AliInputEventHandler* MixingHandler()               {return fMixingHandler;}
  76     // Parent Handler
  77     void SetParentHandler(AliInputEventHandler* parent) {Changed(); fParentHandler = parent;}
  78     AliInputEventHandler* ParentHandler()               {return fParentHandler;}
  79
  80     //PID response
  81     virtual AliPIDResponse* GetPIDResponse() {return 0x0;}
  82     virtual void CreatePIDResponse(Bool_t /*isMC*/=kFALSE) {;}
  83
  84  protected:
  85     void SwitchOffBranches() const;
  86     void SwitchOnBranches()  const;
  87  private:
  88     AliInputEventHandler(const AliInputEventHandler& handler);
  89     AliInputEventHandler& operator=(const AliInputEventHandler& handler);
  90  protected:
  91     TTree          *fTree;         //! Pointer to the tree
  92     TString         fBranches;     //  List of branches to be switched off (separated by space)
  93     TString         fBranchesOn;   //  List of branches to be switched on  (separated by space)
  94     TString         fInputFileName; // Name of the input file
  95     Bool_t          fNewEvent;     //  New event flag
  96     AliVCuts*       fEventCuts;    //  Cuts on the event level
  97     UInt_t          fIsSelectedResult; //  Selection result
  98     AliInputEventHandler* fMixingHandler; // Optionla plugin for mixing
  99     AliInputEventHandler* fParentHandler; // optional pointer to parent handlers (used in AliMultiInputEventHandler)
 100     TList           *fUserInfo;     //! transient user info for current tree
 101     ClassDef(AliInputEventHandler, 7);
 102 };
 103
 104 #endif