]>
Commit | Line | Data |
---|---|---|
cbedddce | 1 | #ifndef ALIANALYSISTASKSESIGNIFICANCE_H |
2 | #define ALIANALYSISTASKSESIGNIFICANCE_H | |
3 | ||
4 | /* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. * | |
5 | * See cxx source for full Copyright notice */ | |
6 | ||
27de2dfb | 7 | /* $Id$ */ |
8 | ||
cbedddce | 9 | //************************************************************************* |
10 | // AliAnalysisTaskSESignificane to calculate effects on | |
11 | // significance of D mesons cut | |
12 | // Authors: G. Ortona, ortona@to.infn.it | |
13 | // F. Prino, prino@to.infn.it | |
14 | // Renu Bala, bala@to.infn.it | |
15 | // Chiara Bianchin, cbianchi@pd.infn.it | |
16 | //************************************************************************* | |
17 | ||
18 | #include "AliAnalysisTaskSE.h" | |
19 | #include "AliAnalysisVertexingHF.h" | |
20 | ||
21 | class TH1F; | |
22 | class AliMultiDimVector; | |
23 | class AliRDHFCuts; | |
24 | ||
25 | class AliAnalysisTaskSESignificance : public AliAnalysisTaskSE | |
26 | { | |
27 | public: | |
28 | ||
33c5731e | 29 | enum FeedDownEnum {kBoth,kCharmOnly,kBeautyOnly}; |
30 | ||
cbedddce | 31 | AliAnalysisTaskSESignificance(); |
32 | AliAnalysisTaskSESignificance(const char *name, TList *listMDV,AliRDHFCuts *RDCuts, Int_t decaychannel,Int_t selectionlevel=AliRDHFCuts::kAll); | |
33 | ||
34 | virtual ~AliAnalysisTaskSESignificance(); | |
35 | ||
3aa6ce53 | 36 | Bool_t CheckConsistency(); |
cbedddce | 37 | void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;} |
33c5731e | 38 | void SetBFeedDown(FeedDownEnum flagB);//see enum |
39 | void SetDFromCharmOnly(){SetBFeedDown(kCharmOnly);} | |
40 | void SetDFromBeautyOnly(){SetBFeedDown(kBeautyOnly);} | |
cbedddce | 41 | void SetMassLimits(Float_t range,Int_t pdg); |
42 | void SetMassLimits(Float_t lowlimit, Float_t uplimit); | |
f80a6da9 | 43 | void SetNBins(Int_t nbins){fNBins=nbins;} |
44 | void SetFillWithPartAntiPartBoth(Int_t value){fPartOrAndAntiPart=value;} | |
cbedddce | 45 | //void SetMultiVector(const AliMultiDimVector *MultiDimVec){fMultiDimVec->CopyStructure(MultiDimVec);} |
f80a6da9 | 46 | Float_t GetUpperMassLimit()const {return fUpmasslimit;} |
47 | Float_t GetLowerMassLimit()const {return fLowmasslimit;} | |
48 | Int_t GetNBins()const {return fNBins;} | |
49 | Int_t GetFillWithPartAntiPartBoth()const {return fPartOrAndAntiPart;} | |
33c5731e | 50 | Int_t GetBFeedDown()const {return fBFeedDown;} |
f80a6da9 | 51 | |
cbedddce | 52 | // Implementation of interface methods |
53 | virtual void UserCreateOutputObjects(); | |
54 | virtual void LocalInit();// {Init();} | |
55 | virtual void UserExec(Option_t *option); | |
56 | virtual void Terminate(Option_t *option); | |
57 | ||
58 | private: | |
59 | ||
60 | AliAnalysisTaskSESignificance(const AliAnalysisTaskSESignificance &source); | |
196a3266 | 61 | AliAnalysisTaskSESignificance& operator=(const AliAnalysisTaskSESignificance& source); |
62 | void SetPDGCodes(); | |
cbedddce | 63 | Int_t GetHistoIndex(Int_t iPtBin) const { return iPtBin*3;} |
64 | Int_t GetSignalHistoIndex(Int_t iPtBin) const { return iPtBin*3+1;} | |
65 | Int_t GetBackgroundHistoIndex(Int_t iPtBin) const { return iPtBin*3+2;} | |
66 | Int_t GetLSHistoIndex(Int_t iPtBin)const { return iPtBin*5;} | |
f80a6da9 | 67 | |
f80a6da9 | 68 | |
196a3266 | 69 | void FillDplus(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Int_t isSel); |
70 | void FillD02p(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index, Int_t isSel); | |
71 | void FillDs(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Int_t isSel); | |
72 | void FillDstar(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Int_t isSel); | |
73 | void FillD04p(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index,Int_t isSel); | |
74 | void FillLambdac(AliAODRecoDecayHF* d,TClonesArray *arrayMC,Int_t index, Int_t isSel); | |
f80a6da9 | 75 | |
76 | ||
cbedddce | 77 | enum {kMaxPtBins=5}; |
78 | enum {kMaxCutVar=5}; | |
79 | enum {kMaxSteps=10}; | |
80 | enum {kMaxNHist=500000}; | |
5ecc6e9b | 81 | enum {kDplustoKpipi,kD0toKpi,kDstartoKpipi,kDstoKKpi,kD0toKpipipi,kLambdactopKpi}; |
cbedddce | 82 | |
83 | TList *fOutput; //! list send on output slot 0 | |
84 | TList *fCutList; //Multidimvector container | |
85 | TH1F *fMassHist[kMaxNHist]; //!hist. for inv mass | |
86 | TH1F *fSigHist[kMaxNHist]; //!hist. for inv mass (sig from MC truth) | |
87 | TH1F *fBkgHist[kMaxNHist]; //!hist. for inv mass (bkg from MC truth) | |
88 | TH1F *fRflHist[kMaxNHist]; //!hist. for inv mass (bkg from MC truth) | |
89 | TH1F* fHistNEvents; //! hist of n of aods | |
90 | Float_t fUpmasslimit; //upper inv mass limit for histos | |
91 | Float_t fLowmasslimit; //lower inv mass limit for histos | |
92 | AliRDHFCuts *fRDCuts;//prong cut values | |
93 | Int_t fNPtBins; //number of pt bins | |
94 | Bool_t fReadMC; //flag for access to MC | |
33c5731e | 95 | FeedDownEnum fBFeedDown; //flag to search for D from B decays |
cbedddce | 96 | Int_t fDecChannel; //decay channel identifier |
196a3266 | 97 | Int_t fPDGmother; // PDG code of D meson |
98 | Int_t fNProngs; // number of prong of the decay channel | |
99 | Int_t fPDGdaughters[4]; // PDG codes of daughters | |
100 | TString fBranchName; // AOD branch name for channel | |
101 | Int_t fSelectionlevel; //selection level: kALL,kTracks,kCandidate | |
102 | Int_t fNVars; // number of selection variables | |
103 | Float_t fVars[kMaxCutVar]; // array with values of cut variables | |
f80a6da9 | 104 | Int_t fNBins; //number of bins in the mass histograms |
105 | Int_t fPartOrAndAntiPart; //fill histograms with particle only (+1), antiparticle only (-1), both (0) | |
106 | ||
196a3266 | 107 | ClassDef(AliAnalysisTaskSESignificance,3); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates |
cbedddce | 108 | }; |
109 | ||
110 | #endif | |
111 | ||
112 |