1 #ifndef ALIHFMASSFITTER_H
2 #define ALIHFMASSFITTER_H
3 /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
6 /////////////////////////////////////////////////////////////
8 // AliHFMassFitter for the fit of invariant mass distribution
11 // Author: C.Bianchin, chiara.bianchin@pd.infn.it
12 /////////////////////////////////////////////////////////////
23 class AliHFMassFitter : public TNamed {
27 AliHFMassFitter(const TH1F* histoToFit, Double_t minvalue, Double_t maxvalue, Int_t rebin=1,Int_t fittypeb=0,Int_t fittypes=0);
28 virtual ~AliHFMassFitter();
30 AliHFMassFitter(const AliHFMassFitter &mfit);
31 AliHFMassFitter& operator=(const AliHFMassFitter &mfit);
34 void SetHisto(const TH1F *histoToFit);
35 void SetRangeFit(Double_t minvalue, Double_t maxvalue){fminMass=minvalue; fmaxMass=maxvalue;}
36 void SetMinRangeFit(Double_t minvalue){fminMass=minvalue;}
37 void SetMaxRangeFit(Double_t maxvalue){fmaxMass=maxvalue;}
38 void SetBinN(Int_t newbinN){fNbin=newbinN;}
39 void SetType(Int_t fittypeb, Int_t fittypes);
40 void SetReflectionSigmaFactor(Int_t constant) {ffactor=constant;}
41 void SetInitialGaussianMean(Double_t mean) {fMass=mean;} // change the default value of the mean
42 void SetInitialGaussianSigma(Double_t sigma) {fSigmaSgn=sigma;} // change the default value of the sigma
43 void SetSideBands(Bool_t onlysidebands=kTRUE) {fSideBands=onlysidebands;} // consider only side bands
46 TH1F* GetHistoClone() const; //return the histogram
47 void GetRangeFit(Double_t &minvalue, Double_t &maxvalue) const {minvalue=fminMass; maxvalue=fmaxMass;}
48 Double_t GetMinRangeFit()const {return fminMass;}
49 Double_t GetMaxRangeFit()const {return fmaxMass;}
50 Int_t GetBinN() const {return fNbin;}
51 void GetFitPars(Float_t* pars) const;
52 void GetTypeOfFit(Bool_t &background, Int_t &typeb) const {background = fWithBkg; typeb = ftypeOfFit4Bkg;}
53 Int_t GetReflectionSigmaFactor() const {return ffactor;}
54 Double_t GetMean() const {return fMass;}
55 Double_t GetSigma()const {return fSigmaSgn;}
56 Double_t GetChiSquare() const;
57 Double_t GetReducedChiSquare() const;
58 void GetSideBandsBounds(Int_t& lb, Int_t& hb) const;
60 void PrintParTitles() const;
62 void InitNtuParam(char *ntuname="ntupar"); // initialize TNtuple to store the parameters
63 void FillNtuParam(); //Fill the TNtuple with the current parameters
64 TNtuple* GetNtuParam() const {return fntuParam;} // return the TNtuple
65 TNtuple* NtuParamOneShot(char *ntuname="ntupar"); // the three functions above all together
66 void WriteHisto(TString path="./") const; // write the histogram
67 void WriteNtuple(TString path="./") const; // write the TNtuple
71 void IntS(Float_t *valuewitherror) const; // integral of signal given my the fit with error
72 Double_t IntTot() const {return fhistoInvMass->Integral("width");} // return total integral of the histogram
73 void Signal(Double_t nOfSigma,Double_t &signal,Double_t &errsignal) const; // signal in nsigma with error
74 void Signal(Double_t min,Double_t max,Double_t &signal,Double_t &errsignal) const; // signal in (min, max) with error
75 void Background(Double_t nOfSigma,Double_t &background,Double_t &errbackground) const; // backgournd in nsigma with error
76 void Background(Double_t min,Double_t max,Double_t &background,Double_t &errbackground) const; // backgournd in (min, max) with error
77 void Significance(Double_t nOfSigma,Double_t &significance,Double_t &errsignificance) const; // significance in nsigma with error
78 void Significance(Double_t min,Double_t max,Double_t &significance,Double_t &errsignificance) const; // significance in (min, max) with error
80 Double_t FitFunction4MassDistr (Double_t* x, Double_t* par);
81 Double_t FitFunction4Sgn (Double_t* x, Double_t* par);
82 Double_t FitFunction4Bkg (Double_t* x, Double_t* par);
83 Bool_t MassFitter(Bool_t draw=kTRUE);
84 void RebinMass(Int_t binground=1);
89 void ComputeParSize();
90 Bool_t SideBandsBounds();
91 void AddFunctionsToHisto();
93 TH1F *fhistoInvMass; // histogram to fit
94 Double_t fminMass; // lower mass limit
95 Double_t fmaxMass; // upper mass limit
96 Int_t fNbin; // number of bins
97 Int_t fParsSize; // size of fFitPars array
98 Float_t *fFitPars; //[fParsSize] array of fit parameters
99 Bool_t fWithBkg; // signal+background (kTRUE) or signal only (kFALSE)
100 Int_t ftypeOfFit4Bkg; // 0 = exponential; 1 = linear; 2 = pol2
101 Int_t ftypeOfFit4Sgn; // 0 = gaus; 1 = gaus+gaus broadened
102 Int_t ffactor; // number to multiply to the sigma of the signal to obtain the reflected gaussian
103 TNtuple *fntuParam; // contains fit parameters
104 Double_t fMass; // signal gaussian mean value
105 Double_t fSigmaSgn; // signal gaussian sigma
106 Bool_t fSideBands; // kTRUE = only side bands considered
107 Int_t fSideBandl; // left side band limit (bin number)
108 Int_t fSideBandr; // right side band limit (bin number)
109 Int_t fcounter; // internal counter
110 TList *fContourGraph; // TList of TGraph containing contour plots
111 ClassDef(AliHFMassFitter,3); // class for invariant mass fit