PWGHF/correlationHF/AliAnalysisTaskDStarCorrelations.h

   1 #ifndef AliAnalysisTaskDStarCorrelations_H
   2 #define AliAnalysisTaskDStarCorrelations_H
   3
   4 /**************************************************************************
   5  * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
   6  *                                                                        *
   7  * Author: The ALICE Off-line Project.                                    *
   8  * Contributors are mentioned in the code where appropriate.              *
   9  *                                                                        *
  10  * Permission to use, copy, modify and distribute this software and its   *
  11  * documentation strictly for non-commercial purposes is hereby granted   *
  12  * without fee, provided that the above copyright notice appears in all   *
  13  * copies and that both the copyright notice and this permission notice   *
  14  * appear in the supporting documentation. The authors make no claims     *
  15  * about the suitability of this software for any purpose. It is          *
  16  * provided "as is" without express or implied warranty.                  *
  17  **************************************************************************/
  18
  19 /* $Id: AliAnalysisTaskDStarCorrelations.h 65139 2013-11-25 14:47:45Z fprino $ */
  20
  21 //-----------------------------------------------------------------------
  22 //
  23 //
  24 //                                                 Author S.Bjelogrlic
  25 //                         Utrecht University
  26 //                      sandro.bjelogrlic@cern.ch
  27 //
  28 //-----------------------------------------------------------------------
  29
  30 #include <TH2F.h>
  31 #include <TH1D.h>
  32 #include <TH3D.h>
  33 #include "AliAnalysisTaskSE.h"
  34 #include "AliAODEvent.h"
  35 #include "AliRDHFCutsDStartoKpipi.h"
  36 #include "AliEventPoolManager.h"
  37 #include "AliAODRecoCascadeHF.h"
  38 #include "AliHFAssociatedTrackCuts.h"
  39 #include "AliNormalizationCounter.h"
  40 #include "AliHFCorrelator.h"
  41 #include <THnSparse.h>
  42 #include "AliAnalysisUtils.h"
  43 #include "AliVertexingHFUtils.h"
  44 class TParticle ;
  45 class TClonesArray ;
  46 class AliAODMCParticle;
  47 class AliAODEvent;
  48 class AliVParticle;
  49 class TObjArray;
  50 class AliEventPoolManager;
  51 class AliESDEvent;
  52
  53
  54
  55 class AliAnalysisTaskDStarCorrelations : public AliAnalysisTaskSE
  56 {
  57
  58  public :
  59
  60   enum CollSyst {pp,pA,AA};
  61     enum DEffVariable{kNone,kMult,kCentr,kRapidity,kEta};
  62     enum BkgMethod{kDZeroSB, kDStarSB};
  63
  64   AliAnalysisTaskDStarCorrelations();
  65   AliAnalysisTaskDStarCorrelations(const Char_t* name,AliRDHFCutsDStartoKpipi* cuts, AliHFAssociatedTrackCuts *AsscCuts, AliAnalysisTaskDStarCorrelations::CollSyst syst,Bool_t mode);
  66   virtual ~AliAnalysisTaskDStarCorrelations();
  67
  68
  69
  70   // Implementation of interface methods
  71   virtual void UserCreateOutputObjects();
  72   virtual void Init();
  73   virtual void LocalInit() {Init();}
  74   virtual void UserExec(Option_t *option);
  75   virtual void Terminate(Option_t *option);
  76
  77   void DefineThNSparseForAnalysis();
  78   void DefineHistoForAnalysis();
  79   void EnlargeDZeroMassWindow();
  80     Bool_t IsDDaughter(AliAODMCParticle* d, AliAODMCParticle* track) const ;
  81
  82   // checker for event mixing
  83   void EventMixingChecks(AliAODEvent * AOD);
  84
  85     // setters
  86     void SetCorrelator(Int_t l) {fselect = l;} // select 1 for hadrons, 2 for Kaons, 3 for Kzeros
  87     void SetMonteCarlo(Bool_t k) {fmontecarlo = k;}
  88     void SetUseMixing (Bool_t j) {fmixing = j;}
  89     void SetUseFullMode (Bool_t j) {fFullmode = j;}
  90     void SetCollSys(CollSyst system){fSystem=system;} // select between pp (kFALSE) or PbPb (kTRUE)
  91     void SetEfficiencyVariable(DEffVariable var){fEfficiencyVariable = var;} // set the efficiency variable to use
  92     void SetBkgEstimationMethod(BkgMethod var){fBkgMethod = var;} // set the efficiency variable to use
  93     void SetUseReconstruction(Bool_t reco){fReco = reco;}
  94     void SetUseEfficiencyCorrection(Bool_t correction){fUseEfficiencyCorrection = correction;} // setter for using the single track efficiency correction
  95     void SetUseDmesonEfficiencyCorrection(Bool_t correction){fUseDmesonEfficiencyCorrection = correction;} // setter for using the single track efficiency correction
  96     void SetUseCentrality (Bool_t j) {fUseCentrality = j;} // switch for centrality (kTRUE)/multiplicity(kFALSE)
  97     void SetUseHadronicChannelAtKineLevel (Bool_t use){fUseHadronicChannelAtKineLevel = use;}
  98     void SetUseRemoveMoreThanOneCDmesonCandidate (Bool_t use){fRemoveMoreThanOneDmesonCandidate = use;}
  99
 100
 101     void SetNofPhiBins(Int_t nbins){fPhiBins = nbins;} // number of delta phi bins
 102     void SetLevelOfDebug(Int_t debug){fDebugLevel=debug;} // set debug level
 103     void SetUseDisplacement(Int_t m) {fDisplacement=m;} // select 0 for no displ, 1 for abs displ, 2 for d0/sigma_d0
 104
 105
 106     void SetDim(){fDim = 4;
 107         fDMesonSigmas = new Float_t[4];} // standard definedt = cannot be changed from outside
 108
 109     void SetMaxDStarEta(Double_t eta){fMaxEtaDStar = eta;} // maximum eta D*
 110
 111
 112   void SetDMesonSigmas(Float_t DStarWin, Float_t D0Win, Float_t SBmin, Float_t SBmax){
 113     fDMesonSigmas[0] = DStarWin;
 114     fDMesonSigmas[1] = D0Win;
 115     fDMesonSigmas[2] = SBmin;
 116     fDMesonSigmas[3] = SBmax;
 117
 118   }
 119
 120
 121
 122
 123
 124   void SetDeffMapvsPt(TH1D * map){fDeffMapvsPt = map;}
 125   void SetDeffMapvsPtvsMult(TH2D * map){fDeffMapvsPtvsMult = (TH2D*)map;}
 126   void SetDeffMapvsPtvsMultvsEta(TH2D * map){fDeffMapvsPtvsEta = map;}
 127
 128
 129
 130
 131 private:
 132
 133   AliAnalysisTaskDStarCorrelations(const AliAnalysisTaskDStarCorrelations &source);
 134   AliAnalysisTaskDStarCorrelations& operator=(const AliAnalysisTaskDStarCorrelations& source);
 135
 136   TObject* fhandler; //! Analysis Handler
 137   TClonesArray* fmcArray; //mcarray
 138   AliNormalizationCounter *fCounter; // counter
 139   AliHFCorrelator * fCorrelator; // object for correlations
 140
 141
 142
 143   Int_t fselect; // select what to correlate with a D* 1-chargedtracks,2-chargedkaons,3-k0s
 144   Bool_t fmontecarlo;//switch for MC
 145   Bool_t fmixing;// switch for event mixing
 146   Bool_t fFullmode;
 147   CollSyst fSystem; // pp, pPb or PbPb
 148   DEffVariable  fEfficiencyVariable; // set second variable to study efficiency (mult, centr, y, eta)
 149   BkgMethod fBkgMethod; // bkg estimation method (dstar or dzero sidebands)
 150   Bool_t fReco; // use reconstruction or MC truth
 151   Bool_t fUseEfficiencyCorrection; // boolean variable to use or not the efficiency correction
 152   Bool_t fUseDmesonEfficiencyCorrection; // boolean flag for the use of Dmeson efficiency correction
 153   Bool_t fUseCentrality;// boolean to switch in between centrality or multiplicity
 154   Bool_t fUseHadronicChannelAtKineLevel; //
 155   Bool_t fRemoveMoreThanOneDmesonCandidate; // flag to remove a second, 3rd etc candidate if there is any - useful in PbPb
 156
 157   Int_t fPhiBins;
 158   Int_t fEvents; //! number of event
 159   Int_t fDebugLevel; //! debug level
 160   Int_t fDisplacement; // set 0 for no displacement cut, 1 for absolute d0, 2 for d0/sigma_d0
 161   Int_t fDim;//
 162   Int_t fNofPtBins;
 163      Double_t fMaxEtaDStar;
 164   Float_t *fDMesonSigmas;//[fDim]
 165   Float_t * fD0Window;  //[fNofPtBins]
 166
 167
 168
 169
 170   TList *fOutput;                  //! user output data
 171     TList *fDmesonOutput; //!output related to d meson
 172     TList *fTracksOutput; //!output related to tracks
 173     TList *fEMOutput; //! output with EM checks
 174     TList *fCorrelationOutput; // ! output with correlation sparses
 175   TList *fOutputMC;                //! outpu for MC
 176   AliRDHFCutsDStartoKpipi *fCuts;  // Cuts D*
 177   AliHFAssociatedTrackCuts *fAssocCuts; // cuts for associated
 178   AliAnalysisUtils *fUtils;
 179   AliAODTracklets * fTracklets; // AliAODtracklets
 180
 181   TH1D * fDeffMapvsPt; // histo for Deff mappin
 182   TH2D * fDeffMapvsPtvsMult; // histo for Deff mappin
 183   TH2D * fDeffMapvsPtvsEta; // histo for Deff mappin
 184
 185   ClassDef(AliAnalysisTaskDStarCorrelations,8); // class for D meson correlations
 186
 187 };
 188
 189 #endif