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 /////////////////////////////////////////////////////////////
15 #include <Riostream.h>
25 class AliHFMassFitter : public TNamed {
29 AliHFMassFitter(TH1F* histoToFit, Double_t minvalue, Double_t maxvalue, Int_t rebin=1,Int_t fittypeb=0,Int_t fittypes=0);
30 virtual ~AliHFMassFitter();
32 AliHFMassFitter(const AliHFMassFitter &mfit);
33 AliHFMassFitter& operator=(const AliHFMassFitter &mfit);
36 void SetHisto(TH1F *histoToFit);
37 void SetRangeFit(Double_t minvalue, Double_t maxvalue){fminMass=minvalue; fmaxMass=maxvalue;}
38 void SetMinRangeFit(Double_t minvalue){fminMass=minvalue;}
39 void SetMaxRangeFit(Double_t maxvalue){fmaxMass=maxvalue;}
40 void SetBinN(Int_t newbinN){fNbin=newbinN;}
41 void SetType(Int_t fittypeb, Int_t fittypes);
42 void SetReflectionSigmaFactor(Int_t constant) {ffactor=constant;}
43 void SetInitialGaussianMean(Double_t mean) {fMass=mean;}
44 void SetInitialGaussianSigma(Double_t sigma) {fSigmaSgn=sigma;}
45 void SetSideBands(Bool_t onlysidebands=kTRUE) {fSideBands=onlysidebands;}
48 TH1F* GetHistoClone() const;
49 Double_t GetMinRangeFit()const {return fminMass;}
50 Double_t GetMaxRangeFit()const {return fmaxMass;}
51 Int_t GetBinN() const {return fNbin;}
52 void GetFitPars(Float_t*) const;
53 void GetTypeOfFit(Bool_t &background, Int_t &typeb) const {background = fWithBkg; typeb = ftypeOfFit4Bkg;}
54 Int_t GetReflectionSigmaFactor() const {return ffactor;}
55 Double_t GetMean() const {return fMass;}
56 Double_t GetSigma()const {return fSigmaSgn;}
57 Double_t GetChiSquare() const;
59 void InitNtuParam(char *ntuname="ntupar");
61 TNtuple* GetNtuParam() const {return fntuParam;}
62 TNtuple* NtuParamOneShot(char *ntuname="ntupar");
64 void WriteHisto(TString path="./");
65 void WriteNtuple(TString path="./") const;
68 void Signal(Double_t nOfSigma,Double_t &signal,Double_t &errsignal) const;
69 void Background(Double_t nOfSigma,Double_t &background,Double_t &errbackground) const;
70 void Significance(Double_t nOfSigma,Double_t &significance,Double_t &errsignificance) const;
72 Double_t FitFunction4MassDistr (Double_t*, Double_t*);
73 Double_t FitFunction4Sgn (Double_t*, Double_t*);
74 Double_t FitFunction4Bkg (Double_t*, Double_t*);
75 Bool_t MassFitter(Bool_t draw=kTRUE);
76 void RebinMass(Int_t binground=1);
81 void ComputeParSize();
83 TH1F *fhistoInvMass; // histogram to fit
84 Double_t fminMass; // lower mass limit
85 Double_t fmaxMass; // upper mass limit
86 Int_t fNbin; // number of bins
87 Int_t fParsSize; // size of fFitPars array
88 Float_t *fFitPars; //[fParsSize] array of fit parameters
89 Bool_t fWithBkg; // signal+background (kTRUE) or signal only (kFALSE)
90 Int_t ftypeOfFit4Bkg; // 0 = exponential; 1 = linear; 2 = pol2
91 Int_t ftypeOfFit4Sgn; // 0 = gaus; 1 = gaus+gaus broadened
92 Int_t ffactor; // number to multiply to the sigma of the signal to obtain the reflected gaussian
93 TNtuple *fntuParam; // contains fit parameters
94 Double_t fMass; // signal gaussian mean value
95 Double_t fSigmaSgn; // signal gaussian sigma
96 Bool_t fSideBands; // kTRUE = only side bands considered
97 Int_t fcounter; // internal counter
99 ClassDef(AliHFMassFitter,1); // class for invariant mass fit