AliPhysics  vAN-20151012 (2287573)
 All Classes Namespaces Files Functions Variables Enumerations Enumerator Macros
AliAnalysisTaskSEDplus.h
Go to the documentation of this file.
1 #ifndef ALIANALYSISTASKSEDPLUS_H
2 #define ALIANALYSISTASKSEDPLUS_H
3 
4 /* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice */
6 
7 /* $Id$ */
8 
9 //*************************************************************************
17 //*************************************************************************
18 
19 #include <TROOT.h>
20 #include <TSystem.h>
21 #include <TNtuple.h>
22 #include <TH1F.h>
23 #include <TH2F.h>
24 #include <TH3F.h>
25 #include <THnSparse.h>
26 #include <TArrayD.h>
27 
29 #include "AliAnalysisTaskSE.h"
30 #include "AliAnalysisVertexingHF.h"
32 #include "AliAODMCHeader.h"
33 #include "AliAODMCParticle.h"
34 
35 class AliAnalysisTaskSEDplus : public AliAnalysisTaskSE
36 {
37  public:
38 
40  AliAnalysisTaskSEDplus(const char *name, AliRDHFCutsDplustoKpipi* analysiscuts,Int_t fillNtuple=0);
41  virtual ~AliAnalysisTaskSEDplus();
42 
43  void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
44  void SetDoLikeSign(Int_t dols=0){fDoLS=dols;}
45  void SetSystem(Int_t system=0){fSystem=system;}
46  void SetCutsDistr(Bool_t cutsDistr=kTRUE){fCutsDistr=cutsDistr;}
48  void SetDoMCAcceptanceHistos(Bool_t doMCAcc=kTRUE){fStepMCAcc=doMCAcc;}
49  void SetImpactParameterBinning(Int_t nbins, Float_t dmin, Float_t dmax){
51  fLowerImpPar=dmin;
52  fHigherImpPar=dmax;
53  }
54  void SetUseStrangeness(Bool_t uses=kTRUE){fUseStrangeness=uses;}
55  void SetMassLimits(Float_t range);
56  void SetMassLimits(Float_t lowlimit, Float_t uplimit);
57  void SetBinWidth(Float_t w);
58  void SetUseBit(Bool_t dols=kTRUE){fUseBit=dols;}
59 
63  void SetUseQuarkLevelTag(Bool_t opt){fUseQuarkTagInKine=opt;}
64 
65  Float_t GetUpperMassLimit(){return fUpmasslimit;}
66  Float_t GetLowerMassLimit(){return fLowmasslimit;}
67  Int_t GetNBinsPt(){return fNPtBins;}
68  Float_t GetBinWidth(){return fBinWidth;}
69  Int_t GetNBinsHistos();
70 
71  void LSAnalysis(TClonesArray *arrayOppositeSign,TClonesArray *arrayLikeSign,AliAODEvent *aod,AliAODVertex *vtx1, Int_t nDplusOS);
72 
73  void CreateLikeSignHistos();
76 
77  Bool_t CheckAcc(TClonesArray* arrayMC,Int_t nProng, Int_t *labDau);
78  void FillMCAcceptanceHistos(TClonesArray *arrayMC, AliAODMCHeader *mcHeader);
79 
81  virtual void UserCreateOutputObjects();
82  virtual void Init();
83  virtual void LocalInit() {Init();}
84  virtual void UserExec(Option_t *option);
85  virtual void Terminate(Option_t *option);
86 
87  private:
88 
91  Int_t GetHistoIndex(Int_t iPtBin) const { return iPtBin*3;}
92  Int_t GetSignalHistoIndex(Int_t iPtBin) const { return iPtBin*3+1;}
93  Int_t GetBackgroundHistoIndex(Int_t iPtBin) const { return iPtBin*3+2;}
94  Int_t GetLSHistoIndex(Int_t iPtBin)const { return iPtBin*5;}
95  Float_t GetTrueImpactParameter(const AliAODMCHeader *mcHeader, TClonesArray* arrayMC, const AliAODMCParticle *partDp) const;
96  Float_t GetStrangenessWeights(const AliAODRecoDecayHF3Prong* d, TClonesArray* arrayMC, Float_t factor[3]) const;
97 
98  enum {kMaxPtBins=20};
99 
100  TList *fOutput;
101  TH1F *fHistNEvents;
108  TH1F *fPtKHist[3*kMaxPtBins];
111  TH1F *fDCAHist[3*kMaxPtBins];
112  TH1F *fDLxy[3*kMaxPtBins];
113  TH1F *fDLxyTC[3*kMaxPtBins];
114  TH1F *fCosxy[3*kMaxPtBins];
115  TH1F *fCosxyTC[3*kMaxPtBins];
127  TH2F *fCorreld0Kd0pi[3];
128  TH2F *fHistCentrality[3];
129  THnSparseF *fHistMassPtImpParTC[5];
130  THnSparseF *fMCAccPrompt;
131  THnSparseF *fMCAccBFeed;
132  TH2F *fPtVsMass;
133  TH2F *fPtVsMassTC;
134  TH3F *fYVsPt;
135  TH3F *fYVsPtTC;
136  TH2F *fYVsPtSig;
137  TH2F *fYVsPtSigTC;
138  TH2F *fPhiEtaCand;
140  TH1F *fSPDMult;
141  TNtuple *fNtupleDplus;
142  Float_t fUpmasslimit;
143  Float_t fLowmasslimit;
144  Int_t fNPtBins;
145  Float_t fBinWidth;
146  TList *fListCuts;
150  Int_t fFillNtuple;
151  Bool_t fReadMC;
153  Bool_t fUseBit;
154  Bool_t fCutsDistr;
155  Bool_t fDoImpPar;
156  Bool_t fStepMCAcc;
158  Int_t fNImpParBins;
159  Float_t fLowerImpPar;
160  Float_t fHigherImpPar;
161  Int_t fDoLS;
163  Int_t fSystem;
164 
166  ClassDef(AliAnalysisTaskSEDplus,24);
167 };
169 
170 #endif
TH1F * fCosxy[3 *kMaxPtBins]
!hist. for Cosxy (LC)
THnSparseF * fMCAccPrompt
!histo for StepMCAcc for Dplus prompt (pt,y,ptB)
TH2F * fPhiEtaCand
! hist. with eta/phi distribution of candidates
Bool_t fUseStrangeness
flag for access to MC
TH1F * fSigVertHist[3 *kMaxPtBins]
!hist. for sigVert (LC)
void SetUseStrangeness(Bool_t uses=kTRUE)
AliRDHFCutsDplustoKpipi * fRDCutsAnalysis
list of cuts
virtual void UserCreateOutputObjects()
Implementation of interface methods.
TH1F * fPtpi1Hist[3 *kMaxPtBins]
!hist. for PtPi1 (LC)
void SetImpactParameterBinning(Int_t nbins, Float_t dmin, Float_t dmax)
TH3F * fYVsPt
! hist. of Y vs. Pt vs. Mass(prod. cuts)
virtual void Terminate(Option_t *option)
TH1F * fMassHistTC[3 *kMaxPtBins]
!hist. for inv mass (TC)
TH1F * fDLxy[3 *kMaxPtBins]
!hist. for DLxy (LC)
Int_t fEtaSelection
flag to do LS analysis
Float_t fLowerImpPar
nunber of bins in impact parameter histos
TH2F * fHistCentrality[3]
!hist. for cent distr (all,sel ev, )
TH1F * fPtMaxHistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 5 (LC)
TH1F * fSPDMult
! hist. of spd mult
Bool_t fReadMC
flag for filling ntuple 0 no NTuple 1 big Ntuple 2 small NTuple
Int_t GetLSHistoIndex(Int_t iPtBin) const
TH1F * fMassHistLSTC[5 *kMaxPtBins]
!hist. for LS inv mass (TC)
Int_t GetSignalHistoIndex(Int_t iPtBin) const
TList * fListCuts
width of one bin in output histos
void SetDoMCAcceptanceHistos(Bool_t doMCAcc=kTRUE)
TList * fOutput
! list send on output slot 0
void SetDoImpactParameterHistos(Bool_t doImp=kTRUE)
Bool_t fCutsDistr
flag to use bitmask
void LSAnalysis(TClonesArray *arrayOppositeSign, TClonesArray *arrayLikeSign, AliAODEvent *aod, AliAODVertex *vtx1, Int_t nDplusOS)
Int_t GetBackgroundHistoIndex(Int_t iPtBin) const
TH1F * fCosPHist[3 *kMaxPtBins]
!hist. for PointingAngle (LC)
TH2F * fPtVsMassTC
! hist. of pt vs. mass (analysis cuts)
void SetUseBit(Bool_t dols=kTRUE)
TH2F * fYVsPtSig
! hist. of Y vs. Pt (MC, only sig, prod. cuts)
THnSparseF * fHistMassPtImpParTC[5]
! histograms for impact parameter and cut variation study
Float_t fHigherImpPar
lower limit in impact parameter (um)
Class for cuts on AOD reconstructed D+->Kpipi.
TH1F * fMassHistLS[5 *kMaxPtBins]
!hist. for LS inv mass (LC)
void SetSystem(Int_t system=0)
Int_t fFillNtuple
limits for the Pt bins
TH1F * fSumd02Hist[3 *kMaxPtBins]
!hist. for sum d02 (LC)
void SetDoLikeSign(Int_t dols=0)
void FillMCAcceptanceHistos(TClonesArray *arrayMC, AliAODMCHeader *mcHeader)
TH3F * fYVsPtTC
! hist. of Y vs. Pt vs. Mass (analysis cuts)
TH1F * fMassHistTCPlus[3 *kMaxPtBins]
!hist. for D+ inv mass (TC)
TH2F * fPtVsMass
! hist. of pt vs. mass (prod. cuts)
const Bool_t doImp
Int_t fDoLS
higher limit in impact parameter (um)
TH1F * fPtpi2Hist[3 *kMaxPtBins]
!hist. for PtPi2 (LC)
TH2F * fPhiEtaCandSigReg
! hist. eta/phi of candidates in D+ mass region
Float_t GetTrueImpactParameter(const AliAODMCHeader *mcHeader, TClonesArray *arrayMC, const AliAODMCParticle *partDp) const
Int_t fNImpParBins
flag for quark/hadron level identification of prompt and feeddown
void SetCutsDistr(Bool_t cutsDistr=kTRUE)
TH1F * fDLenHistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 2 (LC)
Int_t GetHistoIndex(Int_t iPtBin) const
THnSparseF * fMCAccBFeed
!histo for StepMCAcc for Dplus FD (pt,y,ptB)
TH1F * fMassHist[3 *kMaxPtBins]
!hist. for inv mass (LC)
Bool_t CheckAcc(TClonesArray *arrayMC, Int_t nProng, Int_t *labDau)
TH1F * fSumd02HistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 3 (LC)
TH1F * fHistNEvents
!hist. for No. of events
TH1F * fCosPHistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 1 (LC)
TH1F * fPtKHist[3 *kMaxPtBins]
!hist. for PtK (LC)
Float_t fBinWidth
Number of Pt Bins.
void SetMassLimits(Float_t range)
TH2F * fYVsPtSigTC
! hist. of Y vs. Pt (MC, only sig, analysis cuts)
void SetReadMC(Bool_t readMC=kTRUE)
TH1F * fDLxyTC[3 *kMaxPtBins]
!hist. for DLxy (TC)
Int_t fSystem
eta region to accept D+ 0=all, -1 = negative, 1 = positive
TH2F * fCorreld0Kd0pi[3]
!hist. for d0k*d0pi vs. d0k*d0pi (LC)
virtual void UserExec(Option_t *option)
TH1F * fDLenHist[3 *kMaxPtBins]
!hist. for Dec Length (LC)
Bool_t fDoImpPar
flag to activate cuts distr histos
AliNormalizationCounter * fCounter
Cuts for Analysis.
TH1F * fSigVertHistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 4 (LC)
Bool_t fUseBit
flag to enhance strangeness in MC to fit to data
Double_t fArrayBinLimits[kMaxPtBins+1]
AliAnalysisTaskSEDplus & operator=(const AliAnalysisTaskSEDplus &source)
const Int_t nbins
void SetUseQuarkLevelTag(Bool_t opt)
Float_t GetStrangenessWeights(const AliAODRecoDecayHF3Prong *d, TClonesArray *arrayMC, Float_t factor[3]) const
TH1F * fDCAHist[3 *kMaxPtBins]
!hist. for DCA (LC)
Float_t fLowmasslimit
upper inv mass limit for histos
TH1F * fCosxyTC[3 *kMaxPtBins]
!hist. for Cosxy (TC)
TNtuple * fNtupleDplus
! output ntuple
Bool_t fStepMCAcc
flag to activate impact paramter histos
Int_t fNPtBins
lower inv mass limit for histos
Bool_t fUseQuarkTagInKine
flag to activate histos for StepMCAcc
TH1F * fMassHistTCMinus[3 *kMaxPtBins]
!hist. for D- inv mass (TC)
TH1F * fDCAHistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 6 (LC)
TH1F * fPtMaxHist[3 *kMaxPtBins]
!hist. for Pt Max (LC)