PWGJE/FlavourJetTasks/AliAnalysisTaskEmcalJetV0CF.h

   1 #ifndef ALIANALYSISTASKEMCALJETV0CF_H
   2 #define ALIANALYSISTASKEMCALJETV0CF_H
   3
   4 #include <TH1D.h>
   5 #include "AliAnalysisTaskEmcalJet.h"
   6 //=============================================================================
   7
   8 class TClonesArray;
   9 class TVector3;
  10
  11 class AliAODEvent;
  12 class AliESDEvent;
  13 class AliCentrality;
  14
  15 class AliParticleContainer;
  16 class AliClusterContainer;
  17 class AliJetContainer;
  18 //=============================================================================
  19
  20 class AliAnalysisTaskEmcalJetV0CF : public AliAnalysisTaskEmcalJet {
  21
  22  public :
  23
  24   AliAnalysisTaskEmcalJetV0CF();
  25   AliAnalysisTaskEmcalJetV0CF(const char *name, const Bool_t bHistos=kTRUE);
  26   virtual ~AliAnalysisTaskEmcalJetV0CF();
  27
  28   virtual void Init();
  29   virtual void LocalInit() { Init(); }
  30   virtual void UserCreateOutputObjects();
  31   virtual void Terminate(Option_t *opt);
  32
  33   void SetKaCutNS(Double_t d) { fKaCutNS = d; }
  34   void SetLaCutNS(Double_t d) { fLaCutNS = d; }
  35   void SetHistoKshortInvM(TH1D const *h) { fHistoKshortInvM = new TH1D(*h); }
  36   void SetHistoLambdaInvM(TH1D const *h) { fHistoLambdaInvM = new TH1D(*h); }
  37   void SetHistoAntiLaInvM(TH1D const *h) { fHistoAntiLaInvM = new TH1D(*h); }
  38   void SetV0EtaRange(Double_t dMin, Double_t dMax) { fV0CutMinEta = dMin, fV0CutMaxEta = dMax; }
  39 //=============================================================================
  40
  41  protected :
  42
  43   virtual void   ExecOnce();
  44   virtual Bool_t FillGeneralHistograms();
  45   virtual Bool_t FillHistograms();
  46   virtual Bool_t IsEventSelected();
  47   virtual Bool_t RetrieveEventObjects();
  48   virtual Bool_t Run();
  49 //=============================================================================
  50
  51  private :
  52
  53   AliAnalysisTaskEmcalJetV0CF(const AliAnalysisTaskEmcalJetV0CF &);
  54   AliAnalysisTaskEmcalJetV0CF& operator=(const AliAnalysisTaskEmcalJetV0CF &);
  55
  56   Bool_t FillRecoInfo();
  57   Bool_t FillKineInfo();
  58   void CreateUserOutputHistograms();
  59
  60   Bool_t IsV0InJet(TVector3 vV0, Double_t dJetPtMin);
  61
  62   Double_t fKaCutNS; //
  63   Double_t fLaCutNS; //
  64
  65   Double_t fV0CutMinEta; //
  66   Double_t fV0CutMaxEta; //
  67
  68   AliAODEvent   *fEventAOD;  //!
  69   AliESDEvent   *fEventESD;  //!
  70   AliCentrality *fCentInfo;  //!
  71
  72   AliJetContainer      *fJetsContRD;          //!
  73   AliParticleContainer *fTracksContRD;        //!
  74   AliClusterContainer  *fCaloClustersContRD;  //!
  75
  76   AliJetContainer      *fJetsContMC;    //!
  77   AliParticleContainer *fTracksContMC;  //!
  78
  79   TClonesArray *fV0s;  //!
  80
  81   TH1D *fHistoKshortInvM;  //!
  82   TH1D *fHistoLambdaInvM;  //!
  83   TH1D *fHistoAntiLaInvM;  //!
  84
  85   TList *fListUserOutputs; //!
  86
  87   ClassDef(AliAnalysisTaskEmcalJetV0CF, 1);
  88 };
  89
  90 #endif