[u/mrichter/AliRoot.git] / STEER / STEERBase / AliTRDNDFast.h

// Author: Daniel.Lohner@cern.ch

#ifndef ALIROOT_AliTRDNDFast
#define ALIROOT_AliTRDNDFast

#ifndef ROOT_TH1
#include "TH1F.h"
#endif
#ifndef ROOT_TArrayF
#include "TArrayF.h"
#endif
#ifndef ROOT_TF2
#include "TF1.h"
#endif
#ifndef ROOT_TMath
#include "TMath.h"
#endif
#ifndef ROOT_TRandom
#include "TRandom.h"
#endif

using namespace std;

extern Double_t langaufun(Double_t *x,Double_t *par);

class AliTRDNDFast : public TObject {

public:
    static const Int_t kNpar = 5;

    AliTRDNDFast();
    AliTRDNDFast(const char *name,Int_t ndim,Int_t nbins,Double_t xlow,Double_t xup);
    AliTRDNDFast(const AliTRDNDFast&);
    AliTRDNDFast &operator=(const AliTRDNDFast &ref);
    virtual ~AliTRDNDFast();
    
    TF1 *FitLandau(TString name,TH1F *htemp,Double_t range[2],TString option);

    void Build(TH1F **hdEdx,TString path="");
    void Build(Double_t **pars);
    Double_t Eval(Double_t *point) const;
    void Random(Double_t *point) const;
    Int_t GetNDim(){return fNDim;};
    Double_t GetParam(Int_t dim,Int_t par){if((dim>=0)&&(dim<fNDim)&&(par>=0)&&(par<kNpar)){return fPars[par].GetAt(dim);}else{return 0;}};
    void PrintPars();
    static void Random(Double_t *point,AliTRDNDFast *nd0,AliTRDNDFast *nd1,Double_t w0,Double_t w1);

private:

    void ScaleLangauFun(TF1 *func,Double_t mpv);
    TF1 *GetLangauFun(TString funcname,Double_t range[2],Double_t scalefactor=1);
    void BuildHistos();
    void Init();
    void Cleanup();
    
    static Int_t BinarySearchInterpolation(Int_t start,Int_t end,Double_t *a0,Double_t *a1,Double_t w0,Double_t w1,Double_t val);
    static Double_t GetRandomInterpolation(TH1F *hist0,TH1F *hist1,Double_t w0,Double_t w1);

    Int_t fNDim; // Dimensions
    TString fTitle; //title
    TF1 **fFunc; //! functions, do not store
    TH1F **fHistos; //[fNDim] Histograms
    TArrayF fPars[kNpar]; // parameters

    ClassDef(AliTRDNDFast,1)  //Fast TRD ND class
};

#endif
Commit	Line	Data
12ece7a3	1	// Author: Daniel.Lohner@cern.ch
	2
	3	#ifndef ALIROOT_AliTRDNDFast
	4	#define ALIROOT_AliTRDNDFast
	5
	6	#ifndef ROOT_TH1
	7	#include "TH1F.h"
	8	#endif
	9	#ifndef ROOT_TArrayF
	10	#include "TArrayF.h"
	11	#endif
	12	#ifndef ROOT_TF2
	13	#include "TF1.h"
	14	#endif
	15	#ifndef ROOT_TMath
	16	#include "TMath.h"
	17	#endif
	18	#ifndef ROOT_TRandom
	19	#include "TRandom.h"
	20	#endif
	21
	22	using namespace std;
	23
	24	extern Double_t langaufun(Double_t x,Double_t par);
	25
	26	class AliTRDNDFast : public TObject {
	27
	28	public:
	29	static const Int_t kNpar = 5;
	30
	31	AliTRDNDFast();
	32	AliTRDNDFast(const char *name,Int_t ndim,Int_t nbins,Double_t xlow,Double_t xup);
	33	AliTRDNDFast(const AliTRDNDFast&);
	34	AliTRDNDFast &operator=(const AliTRDNDFast &ref);
	35	virtual ~AliTRDNDFast();
	36
	37	TF1 FitLandau(TString name,TH1F htemp,Double_t range[2],TString option);
	38
	39	void Build(TH1F **hdEdx,TString path="");
	40	void Build(Double_t **pars);
	41	Double_t Eval(Double_t *point) const;
	42	void Random(Double_t *point) const;
	43	Int_t GetNDim(){return fNDim;};
	44	Double_t GetParam(Int_t dim,Int_t par){if((dim>=0)&&(dim<fNDim)&&(par>=0)&&(par<kNpar)){return fPars[par].GetAt(dim);}else{return 0;}};
	45	void PrintPars();
	46	static void Random(Double_t point,AliTRDNDFast nd0,AliTRDNDFast *nd1,Double_t w0,Double_t w1);
	47
	48	private:
	49
	50	void ScaleLangauFun(TF1 *func,Double_t mpv);
	51	TF1 *GetLangauFun(TString funcname,Double_t range[2],Double_t scalefactor=1);
	52	void BuildHistos();
	53	void Init();
	54	void Cleanup();
	55
	56	static Int_t BinarySearchInterpolation(Int_t start,Int_t end,Double_t a0,Double_t a1,Double_t w0,Double_t w1,Double_t val);
	57	static Double_t GetRandomInterpolation(TH1F hist0,TH1F hist1,Double_t w0,Double_t w1);
	58
	59	Int_t fNDim; // Dimensions
	60	TString fTitle; //title
	61	TF1 **fFunc; //! functions, do not store
	62	TH1F **fHistos; //[fNDim] Histograms
	63	TArrayF fPars[kNpar]; // parameters
	64
65	ClassDef(AliTRDNDFast,1) //Fast TRD ND class
66	};
67
68	#endif