Coverity fixes (Philippe, Ivana)
[u/mrichter/AliRoot.git] / PWG3 / base / AliPtMothFromPtDaugh.h
CommitLineData
c4df3c2a 1#ifndef ALIPTMOTHFROMPTDAUGH_H
2#define ALIPTMOTHFROMPTDAUGH_H
3/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
27de2dfb 5
6/* $Id$ */
7
c4df3c2a 8////////////////////////////////////////////////////////////////////////////
9// Class to perform pt-spectra (and ptMin-spectra) extraction of mothers //
10// particles starting from measured pt-spectra of daughters particles //
11// that come from inclusive decays. //
12// //
13// Contact: Giuseppe.Bruno@ba.infn.it & Fiorella.Fionda@ba.infn.it //
14////////////////////////////////////////////////////////////////////////////
15
16class TH1F;
17class TNtuple;
18class AliStack;
19class TParticle;
20class TArrayI;
21#include <TNamed.h>
22
23//________________________________________________________________
24class AliPtMothFromPtDaugh : public TNamed {
25
26public:
27
28 typedef enum {
29 kUserAnalysis,
30 kBtoJPSI,
31 kBtoEle,
32 kBtoMuon,
33 kBtoD0
34 } Analysis_mode;
35
36 AliPtMothFromPtDaugh();
37 AliPtMothFromPtDaugh(const char* name, const char* title);
38 virtual ~AliPtMothFromPtDaugh();
39
40 Bool_t CreateWeights(); // method which create the containers for the corrections for the deconvolution
41 void DeleteWeights(); // reset the containers
42 Bool_t ReadHistoPtDaught(const TH1F *hist); // this is input, i.e. the spectrum of the daughter.
43 //
44 // setters
45 //
46 void SetDefaultAnalysis(Analysis_mode mode){fAnalysisMode = mode; InitDefaultAnalysis();}
47 void SetPdgDaugh(Int_t pdgD); // set the pdg of the daughter particle
48 void SetBeautyMothers(); // all the Beauty particles are considered as "mothers"
49 void SetPdgMothers(Int_t n_mothers,Int_t *pdgM); // define which particles are considered as possible mothers
50
51 // (here you can set the pt binning of the final spectrum, i.e. that of the mothers )
52 void SetBinsPtMoth(Double_t ptmin, Double_t ptmax,Int_t nbins,Double_t alpha=1.0); // binning of histo pt(Mothers)
53 void SetBinsPtMoth(Int_t nbins, Double_t *edgeBins); // " " " "
54 void SetBinsPtMinMoth(Double_t ptmin, Double_t ptmax,Int_t nbins,Double_t alpha=1.0); // histo of pt_min(Mothers)
55 void SetBinsPtMinMoth(Int_t nbins, Double_t *edgeBins); // histo of pt_min(Mothers)
56 //
57 void SetYmothers(Double_t yMin, Double_t yMax){fyMothMin = yMin; // define rapidity range
58 fyMothMax = yMax; SetUseEta(kFALSE);} // of the mothers
59 void SetYdaughter(Double_t yMin, Double_t yMax){fyDaughMin = yMin; // define rapidity range
60 fyDaughMax = yMax; SetUseEta(kFALSE);} // of the daughter
61
62 void SetEtaMothers(Double_t etaMin, Double_t etaMax){fyMothMin = etaMin; // define pseudo-rapidity range of mothers
63 fyMothMax = etaMax; SetUseEta(kTRUE);}
64 void SetEtaDaughter(Double_t etaMin, Double_t etaMax){fyDaughMin = etaMin; // define pseudo-rapidity range of daughter
65 fyDaughMax = etaMax; SetUseEta(kTRUE);}
66 void SetDecayNtupla(TNtuple *DecKine){fDecayKine = DecKine;} // define TNtupla with kinematic informations
67 //getters
68 Double_t* GetBinsSize(const TH1F *hist, Int_t &n) const;
69 Bool_t GetEtaMothers(Double_t &etaMin, Double_t &etaMax) const; // get pseudorapidity edges for mothers if pseudorapidity
70 // is used (return kFALSE if rapidity has been used)
71 Bool_t GetEtaDaughter(Double_t &etaMin, Double_t &etaMax) const; // get pseudorapidity edges for daughters if pseudorapidity
72 // is used (return kFALSE if rapidity has ben used)
73 Bool_t GetYMothers(Double_t &yMin, Double_t &yMax) const; // get rapidity edges for mothers if rapidity is used
74 Bool_t GetYDaughter(Double_t &yMin, Double_t &yMax) const; // get rapidity edges for daughters if rapidity is used
75
76 Int_t GetPdgDaugh() const {return fDaughter;} // return the pdg of the daughter particle
77 Int_t* GetPdgMothers(Int_t &n_mothers) const; // return the pdg codes of the mothers particles
78
79 // main method to evaluate mothers spectra (for pt and ptMin)
80 Bool_t EvaluatePtMoth();
81 void WritePtMothHistoToFile(char *fileOutName="Mothers.root");
82
83 // return values of correction factors for pt-mothers ditribution
84 Double_t GetW(Int_t i,Int_t j) const;
85 Double_t GetF(Int_t i) const;
86
87 // return values of correction factors for ptMin-mothers ditribution
88 Double_t GetWmin(Int_t i,Int_t j) const;
89 Double_t GetFmin(Int_t i) const;
90
91 // return errors on correction factors for pt-mothers distribution
92 Double_t GetStatErrW(Int_t i,Int_t j) const;
93 Double_t GetStatErrF(Int_t i) const;
94
95 // return errors on correction factors for ptMin-mothers distribution
96 Double_t GetStatErrWmin(Int_t i,Int_t j) const;
97 Double_t GetStatErrFmin(Int_t i) const;
98
99 // return pointers to pt and ptMin mothers histograms
100 TH1F* GetHistoPtMother() const {return fHistoPtMothers;}
101 TH1F* GetHistoPtMinMother() const {return fHistoPtMinMothers;}
102
103 //method to read kinematic
104 Int_t GiveBinIndex(Double_t Ptpart,const TH1F *ptHist) const;
105 Bool_t Rapidity(const TParticle *particle, Double_t &y);
106 Bool_t IsMothers(Int_t pdgCode);
107 Bool_t IsSelectedDaugh(const TParticle *part, Int_t &labelDaugh, AliStack * const stack);
108 Bool_t CutDaugh(Double_t yD,Double_t ptD);
109
110private:
111 void InitDefaultAnalysis(); // set the default analysis for one of the specific Analysis_mode, apart kUserAnalysis
112 // methods to evaluate correction factors
113 Bool_t EvaluateWij(); //
114 Bool_t EvaluateFi(); //
115
116 void SetUseEta(Bool_t useEta){fUseEta=useEta;} // decide whether to use rapidity or pseudo-rapidity cut
117 Double_t* SetBinsSize(Double_t ptmin, Double_t ptmax,Int_t nbins,Double_t alpha);
118 // method to evaluate raw-mothers spectra (for pt and ptMin)
119 Bool_t EvaluatePtMothRaw(TH1F *histoPt, TH1F *histoPtMin);
120 //method to evaluate statistical errors on extracted distribution
121 Bool_t EvaluateErrPt(Double_t *erStat);
122 Bool_t EvaluateErrPtMin(Double_t *erStat);
123 void SetPdgMothersPrivate(Int_t n_mothers,Int_t *pdgM); // define which particles are considered as possible mothers
124
125 TNtuple* fDecayKine; //Ntupla to store kinematic information of Decay
126 Double_t** fWij; //pointer to store correction factors
127 Double_t* fFi; //" " "
128 Double_t** fWijMin; //" " "
129 Double_t* fFiMin; //" " "
130 TH1F* fHistoPtDaughter; //pointers to pt-histogram of Daughther
131 TH1F* fHistoPtMothers; //pointers to ptMin-histogram of mothers
132 TH1F* fHistoPtMinMothers; //pointers for pt_min-histogram of Mothers
133 TArrayI* fMothers; //Array with pdg codes of mothers
134 Int_t fDaughter; //pdg code of daughter
135 Double_t fyMothMax; //max rapidity (or pseudorapidity) of mothers
136 Double_t fyMothMin; //min " " "
137 Double_t fyDaughMax; //max " " of daughters
138 Double_t fyDaughMin; //min " " "
139 Bool_t fUseEta; //kTRUE if pseudorapidity range is used
140 Analysis_mode fAnalysisMode; //analysis mode
141
142 AliPtMothFromPtDaugh(const AliPtMothFromPtDaugh &c);
143 AliPtMothFromPtDaugh &operator=(const AliPtMothFromPtDaugh &c);
144
145 ClassDef(AliPtMothFromPtDaugh,1); // mother pt spectrum from daughter pt spectrum
146};
147
148
149#endif