AliAnalysisTaskSE.h

Go to the documentation of this file.

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

#include "AliAnalysisTask.h"
#include "AliVEvent.h"

class AliVfriendEvent;
class AliVEventHandler;
class AliAODEvent;
class AliAODHeader;
class AliTOFHeader;
class AliAODVZERO;
class AliAODTracklets;
class AliAODCaloCells;
class AliAODCaloTrigger;
class AliMCEvent;
class AliMCEventHandler;
class AliInputEventHandler;
class AliMultiInputEventHandler;
class AliAnalysisCuts;
class AliESDfriend;
class AliEventTag;
class AliTrackSelectionFactory;
class AliVTrackSelection;

class TTree;
class TList;


class AliAnalysisTaskSE : public AliAnalysisTask
{
 public:
    AliAnalysisTaskSE();
    AliAnalysisTaskSE(const char* name);
    AliAnalysisTaskSE(const AliAnalysisTaskSE& obj);
    AliAnalysisTaskSE& operator=(const AliAnalysisTaskSE& other);
    virtual ~AliAnalysisTaskSE() {;}
    // Implementation of interface methods
    virtual void   ConnectInputData(Option_t *option = "");
    virtual void   CreateOutputObjects();
    virtual void   Exec(Option_t* option);
    virtual void   SetDebugLevel(Int_t level) {fDebug = level;}
    virtual void   Init() {;}
    virtual Bool_t Notify();
    // To be implemented by user
    virtual void   UserCreateOutputObjects()  {;}
    virtual void   UserExec(Option_t* /*option*/) {;}
    virtual void   UserExecMix(Option_t */*option*/) {;}
    virtual Bool_t UserNotify() {return kTRUE;}
    virtual void   NotifyRun()  {;}
    
    // Helpers for adding branches to the AOD
    virtual void   AddAODBranch(const char* cname, void* addobj, const char *fname="");
    // Event Selection
    virtual void   SelectCollisionCandidates(UInt_t offlineTriggerMask = AliVEvent::kMB) {fOfflineTriggerMask = offlineTriggerMask;}
    // Loading the declared input branches
    void           LoadBranches() const;
 // Getters
    virtual Int_t         DebugLevel() const  {return fDebug;     }
    virtual AliVEvent*    InputEvent() const  {return fInputEvent;}
    virtual AliVfriendEvent* ESDfriend()  const  {return fESDfriend; }
    virtual AliAODEvent*  AODEvent()   const  {return fOutputAOD; }
    virtual TTree*        OutputTree() const  {return fTreeA;     }
    virtual AliMCEvent*   MCEvent()    const  {return fMCEvent;   }
    virtual Long64_t      Entry()      const  {return fEntry;     }
    virtual const AliEventTag *EventTag() const;
    virtual const char*   CurrentFileName();
    virtual Bool_t        IsStandardAOD() const;
    virtual TList*        GetQAHistos()   const {return fHistosQA;}
    virtual Bool_t        IsEventInBinZero() { return kFALSE;}
    virtual UInt_t        GetCollisionCandidates() const { return fOfflineTriggerMask;}

    void SetTrackSelectionFactory(AliTrackSelectionFactory *factory)  { fTrackSelectionFactory = factory; }
    void SetTrackSelection(AliVTrackSelection *sel)                   { fTrackSelection = sel; }

 protected:
    void ConnectMultiHandler();
    void DisconnectMultiHandler();

    TObjArray *GetAcceptedTracks();

  protected:
    Int_t                 fDebug;           
    // IO
    Int_t                 fEntry;           
    AliVEvent*            fInputEvent;      
    AliVfriendEvent*      fESDfriend;       
    AliVEventHandler* fInputHandler;    
    AliAODEvent*          fOutputAOD;       
    AliMCEvent*           fMCEvent;         
    TTree*                fTreeA;           
    Int_t                 fCurrentRunNumber;
    // Output histos for QA
    TList*                fHistosQA;        
    // Provisions for replication
    static AliAODHeader*    fgAODHeader;        
    static AliTOFHeader*    fgTOFHeader;        
    static AliAODVZERO*     fgAODVZERO;         
    static TClonesArray*    fgAODTracks;        
    static TClonesArray*    fgAODVertices;      
    static TClonesArray*    fgAODV0s;           
    static TClonesArray*    fgAODPMDClusters;   
    static TClonesArray*    fgAODJets;          
    static TClonesArray*    fgAODFMDClusters;   
    static TClonesArray*    fgAODCaloClusters;  
    static AliAODCaloTrigger* fgAODEMCALTrigger; 
    static AliAODCaloTrigger* fgAODPHOSTrigger;  
    static TClonesArray*    fgAODMCParticles;   
    static AliAODTracklets* fgAODTracklets;     
    static AliAODCaloCells* fgAODEmcalCells;    
    static AliAODCaloCells* fgAODPhosCells;     
    static TClonesArray*    fgAODDimuons;       
    static TClonesArray*    fgAODHmpidRings;    
    // Event Selection
    UInt_t fOfflineTriggerMask;   
    // Event Mixing
    AliMultiInputEventHandler *fMultiInputHandler;  
    AliInputEventHandler      *fMCEventHandler;     
    AliTrackSelectionFactory  *fTrackSelectionFactory; 
    AliVTrackSelection        *fTrackSelection;        
    ClassDef(AliAnalysisTaskSE, 5); // Analysis task for standard jet analysis
};
 
#endif