Update

[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnVAnalysisTask.h

//
// Class AliRsnVAnalysisTask
//
// Virtual Class derivated from AliAnalysisTaskSE which will be base class
// for all RSN Multi tasks
//
// authors: Martin Vala (martin.vala@cern.ch)
//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
//

#ifndef ALIRSNVANALYSISTASK_H
#define ALIRSNVANALYSISTASK_H

#include <TH1.h>

#include "AliAnalysisTaskSE.h"

#include "AliRsnEvent.h"
#include "AliRsnVATProcessInfo.h"
#include <AliMixInputEventHandler.h>

class AliESDEvent;
class AliAODEvent;
class AliMCEvent;

class AliRsnVAnalysisTask : public AliAnalysisTaskSE
{
  public:
  
    AliRsnVAnalysisTask(const char *name = "AliRsnVAnalysisTask", Bool_t mcOnly = kFALSE);
    AliRsnVAnalysisTask(const AliRsnVAnalysisTask& copy);
    AliRsnVAnalysisTask& operator= (const AliRsnVAnalysisTask& /*copy*/) { return *this; }
    virtual ~AliRsnVAnalysisTask() {/* Does nothing*/;}

    // basic interface methods
    virtual void    LocalInit();
    virtual Bool_t  UserNotify();
    virtual void    ConnectInputData(Option_t *opt);
    virtual void    UserCreateOutputObjects();
    virtual void    UserExec(Option_t* opt);
		virtual void    UserExecMix(Option_t* option = "");
    virtual void    Terminate(Option_t* opt);

    // customized methods (to be implemented in derived classes)
    virtual void    RsnUserCreateOutputObjects();
    virtual void    RsnUserExec(Option_t*);
    virtual void    RsnUserExecMix(Option_t*);
    virtual void    RsnTerminate(Option_t*);

    // event pre-processing functions
    virtual Bool_t  EventProcess();

    // getters
    AliRsnEvent*           GetRsnEvent(Int_t i=0) {return &fRsnEvent[i];}
    AliRsnVATProcessInfo*  GetInfo()     {return &fTaskInfo;}

    // setters
    void SetMCOnly(Bool_t mcOnly = kTRUE)                           {fMCOnly = mcOnly;}
    void SetLogType(AliLog::EType_t type, const char *classes = "") {fLogType = type; fLogClassesString = classes;}
    void SetPrintInfoNumber(const Long64_t &num = 100)              {fTaskInfo.SetPrintInfoNumber(num);}
    
    void SetMixing(Bool_t doMix = kTRUE) {fIsMixing = doMix;}
    Bool_t IsMixing() { return fIsMixing;}
    
    void UseMixingRange(Bool_t useMixRange = kTRUE) {fUseMixingRange = useMixRange;}
    Bool_t IsUsingMixingRange() { return fUseMixingRange;}

  protected:

    AliLog::EType_t         fLogType;          //  log type
    TString                 fLogClassesString; //  all classes string divided with ":"

    AliESDEvent            *fESDEvent[2];         //  ESD event
    AliMCEvent             *fMCEvent[2];          //  MC event
    AliAODEvent            *fAODEventIn[2];       //  AOD event from input
    AliAODEvent            *fAODEventOut[2];      //  AOD event from output from previous taks

    Bool_t                  fIsMixing;        // flag is using mixing
    Bool_t                  fMCOnly;           //  use only MC information
    AliRsnEvent             fRsnEvent[2];         //  interface to event for RSN package

    TList                  *fInfoList;         //! output list for informations
    AliRsnVATProcessInfo    fTaskInfo;         //  task info
    
    AliMixInputEventHandler *fMixedEH;         //! mixed event hadnler
    Bool_t 									fUseMixingRange;   //

    void                    SetupMixingEvents();
    void                    SetDebugForAllClasses();

    ClassDef(AliRsnVAnalysisTask, 1)
};

#endif