AliPhysics  879ce3b (879ce3b)
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros
Go to the documentation of this file.
4 /* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice */
7 /* $Id$ */
9 //*************************************************************************
17 //*************************************************************************
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>
29 #include "AliAnalysisTaskSE.h"
30 #include "AliAnalysisVertexingHF.h"
32 #include "AliAODMCHeader.h"
33 #include "AliAODMCParticle.h"
35 class AliAnalysisTaskSEDplus : public AliAnalysisTaskSE
36 {
37  public:
40  AliAnalysisTaskSEDplus(const char *name, AliRDHFCutsDplustoKpipi* analysiscuts,Int_t fillNtuple=0);
41  virtual ~AliAnalysisTaskSEDplus();
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;}
63  void SetUseQuarkLevelTag(Bool_t opt){fUseQuarkTagInKine=opt;}
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();
71  void LSAnalysis(TClonesArray *arrayOppositeSign,TClonesArray *arrayLikeSign,AliAODEvent *aod,AliAODVertex *vtx1, Int_t nDplusOS);
73  void CreateLikeSignHistos();
77  Bool_t CheckAcc(TClonesArray* arrayMC,Int_t nProng, Int_t *labDau);
78  void FillMCAcceptanceHistos(TClonesArray *arrayMC, AliAODMCHeader *mcHeader);
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);
87  private:
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;
98  enum {kMaxPtBins=20};
101  TList *fOutput;
102  TH1F *fHistNEvents;
109  TH1F *fPtKHist[3*kMaxPtBins];
112  TH1F *fDCAHist[3*kMaxPtBins];
113  TH1F *fDLxy[3*kMaxPtBins];
114  TH1F *fCosxy[3*kMaxPtBins];
125  TH2F *fCorreld0Kd0pi[3];
126  TH2F *fHistCentrality[3];
127  THnSparseF *fHistMassPtImpPar[5];
128  THnSparseF *fMCAccPrompt;
129  THnSparseF *fMCAccBFeed;
131  TH2F *fPtVsMass;
132  TH3F *fYVsPtNoPid;
133  TH3F *fYVsPt;
135  TH2F *fYVsPtSig;
136  TH2F *fPhiEtaCand;
138  TH1F *fSPDMult;
139  TNtuple *fNtupleDplus;
140  Float_t fUpmasslimit;
141  Float_t fLowmasslimit;
142  Int_t fNPtBins;
143  Float_t fBinWidth;
144  TList *fListCuts;
148  Int_t fFillNtuple;
149  Bool_t fReadMC;
151  Bool_t fUseBit;
152  Bool_t fCutsDistr;
153  Bool_t fDoImpPar;
154  Bool_t fStepMCAcc;
156  Int_t fNImpParBins;
157  Float_t fLowerImpPar;
158  Float_t fHigherImpPar;
159  Int_t fDoLS;
161  Int_t fSystem;
164  ClassDef(AliAnalysisTaskSEDplus,25);
165 };
168 #endif
TH1F * fCosxy[3 *kMaxPtBins]
!hist. for Cosxy (topol+PID)
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 (topol+PID)
void SetUseStrangeness(Bool_t uses=kTRUE)
AliRDHFCutsDplustoKpipi * fRDCutsAnalysis
list of cuts
virtual void UserCreateOutputObjects()
Implementation of interface methods.
THnSparseF * fHistMassPtImpPar[5]
! histograms for impact parameter and cut variation study
TH2F * fYVsPtSigNoPid
! hist. of Y vs. Pt (MC, only sig, w/o PID)
TH1F * fPtpi1Hist[3 *kMaxPtBins]
!hist. for PtPi1 (topol+PID)
void SetImpactParameterBinning(Int_t nbins, Float_t dmin, Float_t dmax)
TH3F * fYVsPt
! hist. of Y vs. Pt vs. Mass (topol+PID cuts)
virtual void Terminate(Option_t *option)
TH1F * fDLxy[3 *kMaxPtBins]
!hist. for DLxy (topol+PID)
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 (topol+PID)
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
Int_t GetSignalHistoIndex(Int_t iPtBin) const
TH1F * fMassHistNoPid[3 *kMaxPtBins]
!hist. for inv mass (w/o PID)
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 (topol+PID)
void SetUseBit(Bool_t dols=kTRUE)
TH2F * fYVsPtSig
! hist. of Y vs. Pt (MC, only sig, topol+PID cuts)
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 (topol+PID)
void SetSystem(Int_t system=0)
Int_t fFillNtuple
limits for the Pt bins
TH1F * fSumd02Hist[3 *kMaxPtBins]
!hist. for sum d02 (topol+PID)
void SetDoLikeSign(Int_t dols=0)
void FillMCAcceptanceHistos(TClonesArray *arrayMC, AliAODMCHeader *mcHeader)
TH1F * fMassHistPlus[3 *kMaxPtBins]
!hist. for D+ inv mass (topol+PID cuts)
TH2F * fPtVsMass
! hist. of pt vs. mass (topol+PID cuts)
const Bool_t doImp
Int_t fDoLS
higher limit in impact parameter (um)
TH1F * fPtpi2Hist[3 *kMaxPtBins]
!hist. for PtPi2 (topol+PID)
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 (topol+PID)
TH3F * fYVsPtNoPid
! hist. of Y vs. Pt vs. Mass(w/o PID)
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 (topol+PID cuts)
TH2F * fPtVsMassNoPid
! hist. of pt vs. mass (w/o PID)
Bool_t CheckAcc(TClonesArray *arrayMC, Int_t nProng, Int_t *labDau)
TH1F * fSumd02HistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 3 (topol+PID)
TH1F * fHistNEvents
!hist. for No. of events
TH1F * fCosPHistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 1 (topol+PID)
TH1F * fPtKHist[3 *kMaxPtBins]
!hist. for PtK (topol+PID)
Float_t fBinWidth
Number of Pt Bins.
void SetMassLimits(Float_t range)
void SetReadMC(Bool_t readMC=kTRUE)
Int_t fSystem
eta region to accept D+ 0=all, -1 = negative, 1 = positive
TH2F * fCorreld0Kd0pi[3]
!hist. for d0k*d0pi vs. d0k*d0pi (topol+PID)
virtual void UserExec(Option_t *option)
TH1F * fDLenHist[3 *kMaxPtBins]
!hist. for Dec Length (topol+PID)
Bool_t fDoImpPar
flag to activate cuts distr histos
AliNormalizationCounter * fCounter
Cuts for Analysis.
TH1F * fSigVertHistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 4 (topol+PID)
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 (topol+PID)
Float_t fLowmasslimit
upper inv mass limit for histos
TH1F * fMassHistMinus[3 *kMaxPtBins]
!hist. for D- inv mass (topol+PID cuts)
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 * fDCAHistLS[3 *kMaxPtBins]
!hist. for LS cuts variable 6 (topol+PID)
TH1F * fPtMaxHist[3 *kMaxPtBins]
!hist. for Pt Max (topol+PID)