Updates (Chiara)
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliHFMassFitter.h
CommitLineData
fabf4d8e 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 */
5
6/////////////////////////////////////////////////////////////
7//
8// AliHFMassFitter for the fit of invariant mass distribution
9// of charmed mesons
10//
11// Author: C.Bianchin, chiara.bianchin@pd.infn.it
12/////////////////////////////////////////////////////////////
13
14#include <TNamed.h>
15#include <Riostream.h>
16#include <TH1F.h>
17#include <TF1.h>
18#include <TMath.h>
19#include <TNtuple.h>
20#include <TFile.h>
21#include <TString.h>
56cbeefb 22#include <TList.h>
fabf4d8e 23
24
25class AliHFMassFitter : public TNamed {
26
27 public:
28 AliHFMassFitter();
29 AliHFMassFitter(TH1F* histoToFit, Double_t minvalue, Double_t maxvalue, Int_t rebin=1,Int_t fittypeb=0,Int_t fittypes=0);
2f328d65 30 virtual ~AliHFMassFitter();
fabf4d8e 31
74179930 32 AliHFMassFitter(const AliHFMassFitter &mfit);
33 AliHFMassFitter& operator=(const AliHFMassFitter &mfit);
fabf4d8e 34
35 //setters
56cbeefb 36 void SetHisto(TH1F *histoToFit);
2f328d65 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);
fabf4d8e 42 void SetReflectionSigmaFactor(Int_t constant) {ffactor=constant;}
6321ee46 43 void SetInitialGaussianMean(Double_t mean) {fMass=mean;} // change the default value of the mean
44 void SetInitialGaussianSigma(Double_t sigma) {fSigmaSgn=sigma;} // change the default value of the sigma
45 void SetSideBands(Bool_t onlysidebands=kTRUE) {fSideBands=onlysidebands;} // consider only side bands
fabf4d8e 46
47 //getters
6321ee46 48 TH1F* GetHistoClone() const; //return the histogram
a60f573d 49 void GetRangeFit(Double_t &minvalue, Double_t &maxvalue) const {minvalue=fminMass; maxvalue=fmaxMass;}
2f328d65 50 Double_t GetMinRangeFit()const {return fminMass;}
51 Double_t GetMaxRangeFit()const {return fmaxMass;}
52 Int_t GetBinN() const {return fNbin;}
fabf4d8e 53 void GetFitPars(Float_t*) const;
56cbeefb 54 void GetTypeOfFit(Bool_t &background, Int_t &typeb) const {background = fWithBkg; typeb = ftypeOfFit4Bkg;}
fabf4d8e 55 Int_t GetReflectionSigmaFactor() const {return ffactor;}
0566386c 56 Double_t GetMean() const {return fMass;}
57 Double_t GetSigma()const {return fSigmaSgn;}
34c79b83 58 Double_t GetChiSquare() const;
b7d4bc49 59 Double_t GetReducedChiSquare() const;
60 void GetSideBandsBounds(Int_t&, Int_t&) const;
fabf4d8e 61
ad73ca8f 62 void PrintParTitles() const;
63
6321ee46 64 void InitNtuParam(char *ntuname="ntupar"); // initialize TNtuple to store the parameters
65 void FillNtuParam(); //Fill the TNtuple with the current parameters
66 TNtuple* GetNtuParam() const {return fntuParam;} // return the TNtuple
67 TNtuple* NtuParamOneShot(char *ntuname="ntupar"); // the three functions above all together
68 void WriteHisto(TString path="./"); // write the histogram
69 void WriteNtuple(TString path="./") const; // write the TNtuple
a60f573d 70 void DrawFit() const;
56cbeefb 71 void Reset();
72
6321ee46 73 void IntS(Float_t *valuewitherror); // integral of signal given my the fit with error
74 Double_t IntTot(){return fhistoInvMass->Integral("width");} // return total integral of the histogram
75 void Signal(Double_t nOfSigma,Double_t &signal,Double_t &errsignal) const; // signal in nsigma with error
76 void Signal(Double_t min,Double_t max,Double_t &signal,Double_t &errsignal) const; // signal in (min, max) with error
77 void Background(Double_t nOfSigma,Double_t &background,Double_t &errbackground) const; // backgournd in nsigma with error
78 void Background(Double_t min,Double_t max,Double_t &background,Double_t &errbackground) const; // backgournd in (min, max) with error
79 void Significance(Double_t nOfSigma,Double_t &significance,Double_t &errsignificance) const; // significance in nsigma with error
80 void Significance(Double_t min,Double_t max,Double_t &significance,Double_t &errsignificance) const; // significance in (min, max) with error
fabf4d8e 81
82 Double_t FitFunction4MassDistr (Double_t*, Double_t*);
83 Double_t FitFunction4Sgn (Double_t*, Double_t*);
84 Double_t FitFunction4Bkg (Double_t*, Double_t*);
34c79b83 85 Bool_t MassFitter(Bool_t draw=kTRUE);
fabf4d8e 86 void RebinMass(Int_t binground=1);
87
88
89 private:
90
2f328d65 91 void ComputeParSize();
b7d4bc49 92 Bool_t SideBandsBounds();
16856d6e 93 void AddFunctionsToHisto();
2f328d65 94
74179930 95 TH1F *fhistoInvMass; // histogram to fit
fabf4d8e 96 Double_t fminMass; // lower mass limit
97 Double_t fmaxMass; // upper mass limit
98 Int_t fNbin; // number of bins
2f328d65 99 Int_t fParsSize; // size of fFitPars array
100 Float_t *fFitPars; //[fParsSize] array of fit parameters
fabf4d8e 101 Bool_t fWithBkg; // signal+background (kTRUE) or signal only (kFALSE)
102 Int_t ftypeOfFit4Bkg; // 0 = exponential; 1 = linear; 2 = pol2
103 Int_t ftypeOfFit4Sgn; // 0 = gaus; 1 = gaus+gaus broadened
104 Int_t ffactor; // number to multiply to the sigma of the signal to obtain the reflected gaussian
105 TNtuple *fntuParam; // contains fit parameters
106 Double_t fMass; // signal gaussian mean value
107 Double_t fSigmaSgn; // signal gaussian sigma
108 Bool_t fSideBands; // kTRUE = only side bands considered
b7d4bc49 109 Int_t fSideBandl; // left side band limit (bin number)
110 Int_t fSideBandr; // right side band limit (bin number)
56cbeefb 111 Int_t fcounter; // internal counter
ad73ca8f 112 TList *fContourGraph; // TList of TGraph containing contour plots
113 ClassDef(AliHFMassFitter,3); // class for invariant mass fit
fabf4d8e 114};
115
116#endif
117
74179930 118