AliPhysics  1168478 (1168478)
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
AliAnalysisTaskSEDs.h
Go to the documentation of this file.
1 #ifndef ALIANALYSISTASKDS_H
2 #define ALIANALYSISTASKDS_H
3 
4 /* Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice */
6 
7 /* $Id$ */
8 
15 
16 #include <TROOT.h>
17 #include <TSystem.h>
18 #include <TH1F.h>
19 #include <TH2F.h>
20 #include <TH3F.h>
21 #include <THnSparse.h>
22 
23 #include "AliAnalysisTaskSE.h"
24 #include "AliRDHFCutsDstoKKpi.h"
25 #include "AliLog.h"
26 
28 
29 
31 {
32  public:
33 
35  AliAnalysisTaskSEDs(const char *name, AliRDHFCutsDstoKKpi* analysiscuts, Int_t fillNtuple=0);
36  virtual ~AliAnalysisTaskSEDs();
37  void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
39  if(fReadMC) fWriteOnlySignal=opt;
40  else AliError("fReadMC has to be kTRUE");
41  }
42  void SetFillNtuple(Int_t fill=0){fFillNtuple=fill;}
43  void SetFillNSparse(Bool_t fill=kTRUE){fFillSparse=fill;}
44  void SetFillNSparseDplus(Bool_t fill=kTRUE){fFillSparseDplus=fill;if(fill)fFillSparse=fill;}
46  void SetMassRange(Double_t rang=0.4){fMassRange=rang;}
47  void SetDoCutVarHistos(Bool_t opt=kTRUE) {fDoCutVarHistos=opt;}
48  void SetUseSelectionBit(Bool_t opt=kFALSE){ fUseSelectionBit=opt;}
50  void SetUseRotBkg(Bool_t flag=kFALSE) {fDoRotBkg=flag;}
51  void SetUseBkgFromPhiSB(Bool_t flag=kFALSE) {fDoBkgPhiSB=flag;}
54  void SetFillTracklets(Bool_t flag) {fUseTrkl=flag;}
55  Bool_t CheckDaugAcc(TClonesArray* arrayMC,Int_t nProng, Int_t *labDau);
56  Bool_t GetUseWeight() const {return fUseWeight;}
57  void FillMCGenAccHistos(TClonesArray *arrayMC, AliAODMCHeader *mcHeader);
58  void GenerateRotBkg(AliAODRecoDecayHF3Prong *d, Int_t dec, Int_t iPtBin);
59 
64 
65  void SetInvMassBinSize(Double_t binsiz=0.002){fMassBinSize=binsiz;}
66  void SetPtBins(Int_t n, Float_t* lim);
68  void SetSystem(Int_t system){fSystem = system;}
69 
71 
73  virtual void UserCreateOutputObjects();
74  virtual void Init();
75  virtual void LocalInit() {Init();}
76  virtual void UserExec(Option_t *option);
77  virtual void Terminate(Option_t *option);
78 
79  private:
80  Int_t GetHistoIndex(Int_t iPtBin) const { return iPtBin*4;}
81  Int_t GetSignalHistoIndex(Int_t iPtBin) const { return iPtBin*4+1;}
82  Int_t GetBackgroundHistoIndex(Int_t iPtBin) const { return iPtBin*4+2;}
83  Int_t GetReflSignalHistoIndex(Int_t iPtBin) const { return iPtBin*4+3;}
84 
86 
89 
91  TH1F* fHistNEvents;
92  TH1F* fHistoPtWeight; //-> user-defined histogram to calculate the Pt weights
93  TH1F* fChanHist[4];
94  TH1F* fMassHist[4*kMaxPtBins];
111  TH1F* fDCAHist[4*kMaxPtBins];
128  TH1F *fHistCentrality[3];
143  TNtuple *fNtupleDs;
145  Int_t fSystem;
162  UChar_t fNPtBins;
164  TList *fListCuts; //list of cuts
171 
172 
175 
176  THnSparseF *fnSparse;
177  THnSparseF *fnSparseIP;
178  THnSparseF *fnSparseMC[4];
179  THnSparseF *fnSparseMCDplus[4];
184  THnSparseF *fImpParSparse;
185  THnSparseF *fImpParSparseMC;
186 
187 
189  ClassDef(AliAnalysisTaskSEDs,24);
190 };
192 
193 #endif
194 
195 
virtual void LocalInit()
THnSparseF * fnSparseMCDplus[4]
TH1F * fPtMaxHist[4 *kMaxPtBins]
! hist. for Pt Max (Prod Cuts)
AliAnalysisTaskSEDs & operator=(const AliAnalysisTaskSEDs &source)
void SetFillNSparseDplus(Bool_t fill=kTRUE)
THnSparseF * fnSparseMC[4]
!<!THnSparse for topomatic variable
double Double_t
Definition: External.C:58
Definition: External.C:260
Definition: External.C:236
TH1F * fPtProng1Hist[4 *kMaxPtBins]
! hist. for DCA (Prod Cuts)
void SetUseRotBkg(Bool_t flag=kFALSE)
TH3F * fCosPxyHist3D
! cosPxy vs Ds mass vs pt
TH1F * fDLenxyHist[4 *kMaxPtBins]
! hist. of norm decay length XY (sig,bkg,tot)
void SetInvMassBinSize(Double_t binsiz=0.002)
TH1F * fNormIPHist[4 *kMaxPtBins]
! hist. for topomatic variable
TH3F * fPtProng1Hist3D
! Pt prong1 vs Ds mass vs pt
void SetUseSelectionBit(Bool_t opt=kFALSE)
void SetPtBins(Int_t n, Float_t *lim)
TH1F * fNDLenxyHist[4 *kMaxPtBins]
! hist. of decay length XY (sig,bkg,tot)
TH2F * fDalitzK0st[4 *kMaxPtBins]
! dalitz plot via K0* (sig,bkg,tot)
Bool_t fFillSparseDplus
flag for usage of THnSparse
Int_t GetHistoIndex(Int_t iPtBin) const
void SetSystem(Int_t system)
THnSparseF * fImpParSparse
!<!THnSparse for MC for D+->kkpi
TH3F * fSigVertHist3D
! SigVert vs Ds mass vs pt
Double_t fMassRange
limits for pt bins
TH2F * fPtVsMass
! hist. of pt vs. mass (prod. cuts)
TH1F * fPtProng0Hist[4 *kMaxPtBins]
! hist. for Pt Max (Prod Cuts)
void SetFillNtuple(Int_t fill=0)
TH2F * fYVsPtSig
! hist. of Y vs. Pt (MC, only sig, prod. cuts)
virtual void UserCreateOutputObjects()
Implementation of interface methods.
Bool_t fFillSparse
flag for usage of HasSelectionBit
TH1F * fSumd02Hist[4 *kMaxPtBins]
! hist. for sum d02 (Prod Cuts)
TH3F * fCosPiKPhiHist3D
! cosPiKPhi vs Ds mass vs pt
TH1F * fMassHistPhi[4 *kMaxPtBins]
! hist. of mass spectra via phi (sig,bkg,tot)
TH1F * fCosPiKPhiHist[4 *kMaxPtBins]
! hist. for CosPiKPhi
Bool_t fDoBkgPhiSB
flag to create rotational bkg (rotating pi track)
THnSparseF * fnSparse
Cuts for Analysis.
TNtuple * fNtupleDs
! output ntuple
THnSparseF * fImpParSparseMC
!<!THnSparse for imp. par. on data
TH1F * fMassRotBkgHistPhi[kMaxPtBins]
! hist. of bkg generated from rot. of the pion
void GenerateRotBkg(AliAODRecoDecayHF3Prong *d, Int_t dec, Int_t iPtBin)
Bool_t fDoRotBkg
flag for usage of sparse for imp. parameter
TH3F * fPtProng0Hist3D
! Pt prong0 vs Ds mass vs pt
TH1F * fHistCentrality[3]
!hist. for cent distr (all,sel ev, )
void SetPtWeightsFromFONLL5anddataoverLHC16i2a()
TH3F * fDCAHist3D
! DCA vs Ds mass vs pt
int Int_t
Definition: External.C:63
TH3F * fCosPHist3D
! cosP vs Ds mass vs pt
void SetWriteOnlySignalInNtuple(Bool_t opt=kTRUE)
AliNormalizationCounter * fCounter
flag to set mass window of phi meson (when using pion rotation to create bkg)
float Float_t
Definition: External.C:68
TH2F * fHistCentralityMult[3]
!hist. for cent distr vs mult (all,sel ev, )
TH1F * fHistNEvents
! hist. for No. of events
Double_t GetPtWeightFromHistogram(Double_t pt)
AliRDHFCutsDstoKKpi * fAnalysisCuts
TList * fOutput
! list send on output slot 0
void SetPhiMassRange4RotBkg(Double_t range)
TH2F * fHistAllV0multNTPCout
! histo for V0mult vs #tracks TPCout (all)
Bool_t fUseTrkl
flag to decide whether to use pt-weights != 1 when filling the container or not
TH1F * fCosPxyHist[4 *kMaxPtBins]
! hist. of cosXY pointing angle (sig,bkg,tot)
void SetPtWeightsFromFONLL5andTAMUoverLHC16i2abc()
void SetMassRange(Double_t rang=0.4)
THnSparseF * fnSparseIP
!<!THnSparse for candidates on data
TH3F * fPtProng2Hist3D
! Pt prong2 vs Ds mass vs pt
Bool_t CheckDaugAcc(TClonesArray *arrayMC, Int_t nProng, Int_t *labDau)
TH2F * fDalitzPhi[4 *kMaxPtBins]
! dalitz plot via phi (sig,bkg,tot)
TH2F * fHistSelV0multNTPCout
! histo for V0mult vs #tracks TPCout (sel)
TH3F * fDLenHist3D
! Dlen vs Ds mass vs pt
TH1F * fMassHistKpi[kMaxPtBins]
! hist. of mass spectra of Kpi
TH1F * fMassRSBkgHistPhi[kMaxPtBins]
! hist. of bkg generated from right phi sideband + pion
TH3F * fNormIPHist3D
! nIP vs Ds mass vs pt
TH1F * fPtCandHist[4 *kMaxPtBins]
! hist. for Pt Max (Prod Cuts)
void FillMCGenAccHistos(TClonesArray *arrayMC, AliAODMCHeader *mcHeader)
Bool_t fReadMC
0 = pp, 1 = pPb,PbPb
Int_t fAODProtection
flag to fill sparse with Ntracklets
TH1F * fCosPiDsHist[4 *kMaxPtBins]
! hist. for CosPiDs
Int_t GetSignalHistoIndex(Int_t iPtBin) const
TH1F * fMassLSBkgHistPhi[kMaxPtBins]
! hist. of bkg generated from left phi sideband + pion
virtual void UserExec(Option_t *option)
void SetFillTracklets(Bool_t flag)
Int_t GetReflSignalHistoIndex(Int_t iPtBin) const
virtual void Terminate(Option_t *option)
TH1F * fCosPHist[4 *kMaxPtBins]
! hist. of cos pointing angle (sig,bkg,tot)
TH3F * fDLenxyHist3D
! Dlenxy vs Ds mass vs pt
Bool_t fFillImpParSparse
flag for usage of THnSparse
Bool_t fDoCutV0multTPCout
flag to create bkg from phi sidebands
void SetFillNSparseImpPar(Bool_t fill=kTRUE)
Bool_t fUseWeight
flag to activate cut on V0mult vs #tracks TPCout
void SetUseBkgFromPhiSB(Bool_t flag=kFALSE)
TH1F * fDLenHist[4 *kMaxPtBins]
! hist. of decay length (sig,bkg,tot)
void SetReadMC(Bool_t readMC=kTRUE)
Bool_t fUseSelectionBit
flag to create and fill histos with distributions of cut variables
void SetUseCutV0multVsTPCout(Bool_t flag)
void SetFillNSparse(Bool_t fill=kTRUE)
const char Option_t
Definition: External.C:48
void SetDoCutVarHistos(Bool_t opt=kTRUE)
TH1F * fChanHist[4]
! hist. with KKpi and piKK candidates (sig,bkg,tot)
TH1F * fMassHist[4 *kMaxPtBins]
! hist. of mass spectra (sig,bkg,tot)
TH1F * fDCAHist[4 *kMaxPtBins]
! hist. for DCA (Prod Cuts)
Bool_t fDoCutVarHistos
flag to control ntuple writing in MC
Bool_t GetUseWeight() const
TH1F * fPtProng2Hist[4 *kMaxPtBins]
! hist. for DCA (Prod Cuts)
bool Bool_t
Definition: External.C:53
void SetAODMismatchProtection(Int_t opt=1)
void SetAnalysisCuts(AliRDHFCutsDstoKKpi *cuts)
void SetPtWeightsFromFONLL5andBAMPSoverLHC16i2abc()
TH2F * fDalitz[4 *kMaxPtBins]
! dalitz plot (sig,bkg,tot)
TH1F * fSigVertHist[4 *kMaxPtBins]
! hist. for sigVert (Prod Cuts)
TH3F * fNDLenxyHist3D
! NDlenxy vs Ds mass vs pt
TH2F * fPtVsMassPhi
! hist. of pt vs. mass (phi selection)
TH1F * fMassHistKK[kMaxPtBins]
! hist. of mass spectra of KK
TH1F * fMassHistK0st[4 *kMaxPtBins]
! hist. of mass spectra via K0* (sig,bkg,tot)
TH2F * fYVsPt
! hist. of Y vs. Pt (prod. cuts)
TH2F * fPtVsMassK0st
! hist. of pt vs. mass (K0* selection)
TList * fListCuts
number of Pt bins
UChar_t fNPtBins
-1: no protection, 0: check AOD/dAOD nEvents only, 1: check AOD/dAOD nEvents + TProcessID names ...
Bool_t fWriteOnlySignal
flag for access to MC
TH3F * fCosPiDsHist3D
! cosPiDs vs Ds mass vs pt
Double_t fMassBinSize
range for mass histogram
Double_t fminMass
bin size for inv. mass histo
Int_t GetBackgroundHistoIndex(Int_t iPtBin) const
Float_t fPtLimits[kMaxPtBins+1]