PWGHF/correlationHF/AliAnalysisTaskSED0Correlations.h

   1 #ifndef ALIANALYSISTASKSED0CORRELATIONS_H
   2 #define ALIANALYSISTASKSED0CORRELATIONS_H
   3
   4 /* Copyright(c) 1998-2012, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 /* $Id$ */
   8
   9 //*************************************************************************
  10 // Class AliAnalysisTaskSED0Correlations
  11 // AliAnalysisTaskSE for D0 candidates (2Prongs) and hadrons correlations
  12 // Authors:
  13 // C.Bianchin, chiara.bianchin@pd.infn.it
  14 // F.Colamaria, fabio.colamaria@ba.infn.it
  15 //*************************************************************************
  16
  17 #include <TROOT.h>
  18 #include <TSystem.h>
  19 #include <TNtuple.h>
  20 #include <TTree.h>
  21 #include <TH1F.h>
  22 #include <THnSparse.h>
  23
  24 #include "AliAnalysisTaskSE.h"
  25 #include "AliRDHFCutsD0toKpi.h"
  26 #include "AliHFAssociatedTrackCuts.h"
  27 #include "AliHFCorrelator.h"
  28 #include "AliNormalizationCounter.h"
  29
  30 using std::vector;
  31
  32 class AliAODEvent;
  33
  34 class AliAnalysisTaskSED0Correlations : public AliAnalysisTaskSE
  35 {
  36  public:
  37
  38   AliAnalysisTaskSED0Correlations();
  39   AliAnalysisTaskSED0Correlations(const char *name,AliRDHFCutsD0toKpi* cutsD0, AliHFAssociatedTrackCuts* cutsTrk);
  40   virtual ~AliAnalysisTaskSED0Correlations();
  41
  42   // Implementation of interface methods
  43   virtual void UserCreateOutputObjects();
  44   virtual void Init();
  45   virtual void LocalInit() {Init();}
  46   virtual void UserExec(Option_t *option);
  47   virtual void Terminate(Option_t *option);
  48
  49   void SetReadMC(Bool_t readMC=kFALSE){fReadMC=readMC;}
  50   void SetMCReconstructedTracks(Bool_t recoTrMC=kTRUE){fRecoTr=recoTrMC;}
  51   void SetMCReconstructedD0(Bool_t recoD0MC=kTRUE){fRecoD0=recoD0MC;}
  52   void SetMCSelEventType(Bool_t sel=kFALSE){fSelEvType=sel;}
  53   void SetFillOnlyD0D0bar(Int_t flagfill){fFillOnlyD0D0bar=flagfill;}
  54   void SetSystem(Int_t sys){fSys=sys;}
  55   void SetRejectSDDClusters(Bool_t flag) {fIsRejectSDDClusters=flag; }
  56   void SetFillGlobalPlots(Bool_t fill=kTRUE){fFillGlobal=fill;}
  57
  58   Int_t  GetReadMC() const {return fReadMC;}
  59   Int_t  GetMCReconstructedTracks() const {return fRecoTr;}
  60   Int_t  GetMCReconstructedD0() const {return fRecoD0;}
  61   Int_t  GetMCSelEventType() const {return fSelEvType;}
  62   Int_t  GetFillOnlyD0D0bar() const {return fFillOnlyD0D0bar;}
  63   Int_t  GetSystem() const {return fSys;}
  64   Bool_t GetRejectSDDClusters() const {return fIsRejectSDDClusters;}
  65   Bool_t GetFillGlobalPlots() const {return fFillGlobal;}
  66   Double_t GetEtaForCorrel() {return fEtaForCorrel;}
  67
  68   //correlations setters/printers
  69   void SetNPtBinsCorr(Int_t nbins) {fNPtBinsCorr = nbins;}
  70   void SetPtBinsLimsCorr(Double_t* ptlims) {for(int i=0;i<=fNPtBinsCorr;i++) {fBinLimsCorr.push_back(ptlims[i]);}}
  71   void SetPtBinsLimsCorr(Float_t* ptlims) {for(int i=0;i<=fNPtBinsCorr;i++) {fBinLimsCorr.push_back((Double_t)ptlims[i]);}}
  72   void SetPtTreshLow(Double_t* pttreshlow) {for(int i=0;i<fNPtBinsCorr;i++) {fPtThreshLow.push_back(pttreshlow[i]);}}
  73   void SetPtTreshUp(Double_t* pttreshup) {for(int i=0;i<fNPtBinsCorr;i++) {fPtThreshUp.push_back(pttreshup[i]);}}
  74   void PrintBinsAndLimits();
  75   Int_t PtBinCorr(Double_t pt) const;
  76   void SetEvMixing(Bool_t mix) {fMixing=mix;}
  77   void SetEtaForCorrel(Double_t etacorr) {fEtaForCorrel=etacorr;}
  78
  79   enum PartType {kTrack,kKCharg,kK0};
  80   enum FillType {kSE, kME}; //for single event or event mixing histos fill
  81
  82  private:
  83
  84   AliAnalysisTaskSED0Correlations(const AliAnalysisTaskSED0Correlations &source);
  85   AliAnalysisTaskSED0Correlations& operator=(const AliAnalysisTaskSED0Correlations& source);
  86   void FillMassHists(AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout);
  87   Int_t CheckD0Origin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
  88   //correlation methods
  89   void CreateCorrelationsObjs();
  90   void CalculateCorrelations(AliAODRecoDecayHF2Prong* d, Int_t labD0=-1, TClonesArray* mcArray=0x0);
  91   void CalculateCorrelationsMCKine(AliAODMCParticle* d, TClonesArray* mcArray=0x0);
  92   void FillSparsePlots(TClonesArray* arrayMC, Double_t mInv[], Int_t origD0, Int_t PdgD0, AliReducedParticle* track, Int_t ptbin, Int_t type, Double_t wg=1.);
  93   Int_t CheckTrackOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
  94   Bool_t IsDDaughter(AliAODMCParticle* d, AliAODMCParticle* track, TClonesArray* mcArray) const;
  95   Bool_t SelectV0(AliAODv0* v0, AliAODVertex *vtx, Int_t option, Int_t idArrayV0[][2]) const;
  96
  97   Int_t             fNPtBinsCorr;        // number of pt bins per correlations
  98   std::vector<Double_t>  fBinLimsCorr;        // limits of pt bins per correlations
  99   std::vector<Double_t>  fPtThreshLow;        // pT treshold of hadrons - low
 100   std::vector<Double_t>  fPtThreshUp;         // pT treshold of hadrons - up
 101
 102   Int_t     fEvents;                    // EventCounter
 103   Bool_t    fAlreadyFilled;             // D0 in an event already analyzed (for track distribution plots)
 104   TList    *fOutputMass;                //!list send on output slot 1
 105   TList    *fOutputCorr;                //!list of correlation histos, output slot 5
 106   TList    *fOutputStudy;               //!list of histos with MC distributions, output slot 6
 107   TH1F     *fNentries;                  //!histogram with number of events on output slot 2
 108   AliRDHFCutsD0toKpi *fCutsD0;          // Cuts for D0, output 3
 109   AliHFAssociatedTrackCuts *fCutsTracks;// Cuts for tracks and K0, output 7
 110   AliHFCorrelator* fCorrelatorTr;       // Correlator for tracks
 111   AliHFCorrelator* fCorrelatorKc;       // Correlator for charged K
 112   AliHFCorrelator* fCorrelatorK0;       // Correlator for K0
 113   Bool_t    fReadMC;                    // flag for MC array: kTRUE = read it, kFALSE = do not read it
 114   Bool_t    fRecoTr;                    // flag for using MC reconstructed (kTRUE) or pure kinematic MC (kFALSE) - Associated tracks
 115   Bool_t    fRecoD0;                    // flag for using MC reconstructed (kTRUE) or pure kinematic MC (kFALSE) - D0
 116   Bool_t    fSelEvType;                 // flag for enabling selection of event tpye (PP, GS, FE, ...) on MC analysis
 117   Bool_t    fMixing;                    // flag to enable also event mixing
 118   AliNormalizationCounter *fCounter;    //!AliNormalizationCounter on output slot 4
 119   Int_t     fNPtBins;                   // Number of pt bins
 120   Int_t     fFillOnlyD0D0bar;           // flag to fill mass histogram with D0/D0bar only (0 = fill with both, 1 = fill with D0 only, 2 = fill with D0bar only)
 121   Int_t     fIsSelectedCandidate;       // selection outcome
 122   Int_t     fSys;                       // fSys=0 -> p-p; fSys=1 ->PbPb
 123   Double_t  fEtaForCorrel;              // cut for D0 eta to enable correlation with associated particles
 124   Bool_t    fIsRejectSDDClusters;       // flag to reject events with SDD clusters
 125   Bool_t    fFillGlobal;                // flag to fill global plots (in loops on tracks and V0 for each event)
 126
 127   ClassDef(AliAnalysisTaskSED0Correlations,2); // AliAnalysisTaskSE for D0->Kpi
 128 };
 129
 130 #endif
 131