Add new version of macros for RSN analysis

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

//
// Class AliRsnAnalysisEffSE
//
// Virtual Class derivated from AliRsnVAnalysisTaskSE which will be base class
// for all RSN SE tasks
//
// authors: Martin Vala (martin.vala@cern.ch)
//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
//
#ifndef ALIRSNANALYSISEFFSE_H
#define ALIRSNANALYSISEFFSE_H

#include <TArrayD.h>
#include <TArrayI.h>
#include <TClonesArray.h>

#include "AliRsnVAnalysisTaskSE.h"
#include "AliRsnEvent.h"
#include "AliRsnMother.h"
#include "AliRsnDaughter.h"

class AliPID;

class AliCFContainer;

class AliRsnPairDef;
class AliRsnPIDIndex;
class AliRsnPIDDefESD;
class AliRsnCutSet;
class AliRsnCutManager;
class AliRsnValue;

class AliRsnAnalysisManager;
class AliRsnAnalysisEffSE : public AliRsnVAnalysisTaskSE {

public:
   AliRsnAnalysisEffSE(const char *name = "AliRsnAnalysisTaskEffSE");
   AliRsnAnalysisEffSE(const AliRsnAnalysisEffSE& copy);
   virtual ~AliRsnAnalysisEffSE() {;};

   // Implement this
   virtual void    RsnUserCreateOutputObjects();
   virtual void    RsnUserExec(Option_t*);
   virtual void    RsnTerminate(Option_t*);
   virtual Bool_t  EventProcess();

   // settings
   AliRsnCutSet*   GetEventCuts() {return &fEventCuts;}
   void            AddPairDef(AliRsnPairDef *pairDef);
   void            AddStepMC(AliRsnCutManager *mgr);
   void            AddStepESD(AliRsnCutManager *mgr);
   void            AddAxis(AliRsnValue *axis);

private:

   AliRsnAnalysisEffSE& operator=(const AliRsnAnalysisEffSE& /*copy*/) {return *this;}
   void                 ProcessEventESD(AliRsnPairDef *pairDef);
   void                 ProcessEventAOD(AliRsnPairDef *pairDef);
   void                 FillContainer(AliCFContainer *cont, const TObjArray *stepList, AliRsnPairDef *pd, Int_t firstOutStep);
   Int_t                FindESDtrack(Int_t label, AliESDEvent *esd, Bool_t rejectFakes);
   TArrayI              FindESDtracks(Int_t label, AliESDEvent *esd);
   Int_t                FindAODtrack(Int_t label, AliAODEvent *aod, Bool_t rejectFakes);
   TArrayI              FindAODtracks(Int_t label, AliAODEvent *aod);

   Bool_t                fUseITSSA;                // switch to use ITS standalone tracks
   Bool_t                fUseGlobal;               // switch to use global tracks
   TObjArray             fStepListMC;              // list of cut managers for all steps with MC
   TObjArray             fStepListESD;             // list of cut managers for all steps with ESD
   TClonesArray          fAxisList;                // list of axes of efficiency plots
   TObjArray             fPairDefList;             // decay channels
   TList                *fContainerList;           // list of CF containers
   TList                *fOutList;                 // global output list
   TArrayD               fVar;                     // list of variables of the container
   AliRsnMother          fPair;                    // interface to pair
   AliRsnDaughter        fDaughter[2];             // interface to tracks
   AliRsnCutSet          fEventCuts;               // event cuts

   ClassDef(AliRsnAnalysisEffSE, 1)
};

#endif