ANALYSIS/AliAnalysisTask.h

   1 #ifndef ALIANALYSISTASK_H
   2 #define ALIANALYSISTASK_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7 // Author: Andrei Gheata, 31/05/2006
   8
   9 //==============================================================================
  10 //   AliAnalysysTask - Class representing a basic analysis task. Any
  11 // user-defined task should derive from it and implement the Exec() virtual
  12 // method.
  13 //==============================================================================
  14
  15 #ifndef ROOT_TTask
  16 #include "TTask.h"
  17 #endif
  18
  19 #ifndef ROOT_TObjArray
  20 #include "TObjArray.h"
  21 #endif
  22
  23 class TClass;
  24 class TFile;
  25 class AliAnalysisDataSlot;
  26 class AliAnalysisDataContainer;
  27
  28 class AliAnalysisTask : public TTask {
  29  public:
  30   enum EAnalysisTaskFlags {
  31     kTaskUsed    = BIT(14),
  32     kTaskZombie  = BIT(15),
  33     kTaskChecked = BIT(16),
  34     kTaskPostEventLoop = BIT(17)
  35   };
  36
  37  protected:
  38   Bool_t                    fReady;         // Flag if the task is ready
  39   Bool_t                    fInitialized;   // True if Init() was called
  40   Int_t                     fNinputs;       // Number of inputs
  41   Int_t                     fNoutputs;      // Number of outputs
  42   Bool_t                   *fOutputReady;   //[fNoutputs] Flags for output readyness
  43   TObject                  *fPublishedData; //! published data
  44   TObjArray                *fInputs;        // Array of input slots
  45   TObjArray                *fOutputs;       // Array of output slots
  46
  47   // Define the input/output slots (called by user in the ctor of the derived class)
  48   //=== CALL IN THE CONSTRUCTOR OF DERIVED CLASS TO DEFINE INPUTS/OUTPUTS ===
  49   void                      DefineInput(Int_t islot, TClass *type);
  50   void                      DefineOutput(Int_t islot, TClass *type);
  51   //=====================================================================
  52
  53   //=====================================================================
  54   // === OVERLOAD THIS TO CONNECT TREE BRANCHES AT INPUT SLOTS. YOU
  55   // SHOULD DEFINE HERE ALSO THE OBJECTS TO BE CONNECTED TO YOUR OUTPUTS
  56   virtual void              ConnectInputData(Option_t *option="");
  57   //=====================================================================
  58
  59   // Post output data (called by Exec() when data is ready)
  60   //=== CALL IN EXEC() FOR EACH OUTPUT WHEN READY ===
  61   Bool_t                    PostData(Int_t iout, TObject *data, Option_t *option="");
  62   //=====================================================================
  63
  64   // === USE THIS FIRST IN YOUR ConnectInputData() TO CHECH IF A BRANCH IS ALREADY CONNECTED
  65   // TO SOME ADDRESS.
  66   char                     *GetBranchAddress(Int_t islot, const char *branch) const;
  67   // === CALL THIS AFTERWARDS IN ConnectInputData() IF THE BRANCH ADDRESS IS NOT YET SET
  68   Bool_t                    SetBranchAddress(Int_t islot, const char *branch, void *address) const;
  69   //=====================================================================
  70   //=== CALL IN ConnectInputData() TO ENABLE ONLY EXPLICIT BRANCHES NEEDED FOR THIS TASK EXECUTION
  71   void                      EnableBranch(Int_t islot, const char *bname) const;
  72   //=====================================================================
  73   // === CALL THIS IN CreateOutputObjects BEFORE CREATING THE OBJECT FOR EACH
  74   // OUTPUT IOUT THAT HAS TO BE WRITTEN TO A FILE
  75   TFile                    *OpenFile(Int_t iout, Option_t *option="RECREATE") const;
  76
  77 public:
  78   AliAnalysisTask();
  79   AliAnalysisTask(const char *name, const char *title);
  80   AliAnalysisTask(const AliAnalysisTask &task);
  81   virtual ~AliAnalysisTask();
  82
  83   // Assignment
  84   AliAnalysisTask& operator=(const AliAnalysisTask &task);
  85   //=====================================================================
  86   // === OVERLOAD THIS AND CREATE YOUR OUTPUT OBJECTS (HISTOGRAMS,DATA) HERE
  87   virtual void              CreateOutputObjects();
  88   // === OVERLOAD THIS IF YOU NEED TO INITIALIZE YOUR CLASS ON THE CLIENT
  89   virtual void              LocalInit();
  90   // === OVERLOAD THIS IF YOU NEED TO TREAT INPUT FILE CHANGE
  91   virtual Bool_t            Notify();
  92   // Conect inputs/outputs to data containers (by AliAnalysisModule)
  93   Bool_t                    ConnectInput(Int_t islot, AliAnalysisDataContainer *cont);
  94   Bool_t                    ConnectOutput(Int_t islot, AliAnalysisDataContainer *cont);
  95   // Check connectivity
  96   Bool_t                    AreSlotsConnected();
  97   // Check if data for all inputs is ready
  98   void                      CheckNotify(Bool_t init=kFALSE);
  99   // Check if there are illegal circular dependencies
 100   Bool_t                    CheckCircularDeps();
 101   // Getters
 102   Int_t                     GetNinputs() const  {return fNinputs;}
 103   Int_t                     GetNoutputs() const {return fNoutputs;}
 104   TObject                  *GetPublishedData() const {return fPublishedData;}
 105   AliAnalysisDataSlot      *GetInputSlot(Int_t islot) const  {return (AliAnalysisDataSlot*)fInputs->At(islot);}
 106   AliAnalysisDataSlot      *GetOutputSlot(Int_t islot) const {return (AliAnalysisDataSlot*)fOutputs->At(islot);}
 107   TClass                   *GetInputType(Int_t islot) const;
 108   TClass                   *GetOutputType(Int_t islot) const;
 109   // === USE THIS TO RETREIVE INPUT DATA AND STATICALLY CAST IT TO THE DECLARED TYPE
 110   TObject                  *GetInputData(Int_t islot) const;
 111   TObject                  *GetOutputData(Int_t islot) const;
 112   Bool_t                    IsOutputReady(Int_t islot) const {return fOutputReady[islot];}
 113   Bool_t                    IsChecked() const  {return TObject::TestBit(kTaskChecked);}
 114   Bool_t                    IsPostEventLoop() const {return TObject::TestBit(kTaskPostEventLoop);}
 115   Bool_t                    IsInitialized() const  {return fInitialized;}
 116   Bool_t                    IsReady() const  {return fReady;}
 117   Bool_t                    IsUsed() const   {return TObject::TestBit(kTaskUsed);}
 118   Bool_t                    IsZombie() const {return TObject::TestBit(kTaskZombie);}
 119   void                      PrintTask(Option_t *option="all", Int_t indent=0) const;
 120   void                      PrintContainers(Option_t *option="all", Int_t indent=0) const;
 121   void                      SetChecked(Bool_t flag=kTRUE) {TObject::SetBit(kTaskChecked,flag);}
 122   void                      SetPostEventLoop(Bool_t flag=kTRUE);
 123   void                      SetUsed(Bool_t flag=kTRUE);
 124   void                      SetZombie(Bool_t flag=kTRUE) {TObject::SetBit(kTaskZombie,flag);}
 125   // Main task execution
 126   //=== IMPLEMENT THIS !!! ==============================================
 127   virtual void              Exec(Option_t *option) = 0;
 128   //=====================================================================
 129   Bool_t                    HasExecuted() const {return fHasExecuted;}
 130   //=====================================================================
 131   // === OVERLOAD THIS IF YOU WANT TO DO SOMETHING WITH THE OUTPUT
 132   virtual void              Terminate(Option_t *option="");
 133   //=====================================================================
 134
 135   ClassDef(AliAnalysisTask,1)  // Class describing an analysis task
 136 };
 137 #endif