]>
Commit | Line | Data |
---|---|---|
0dd3a2ac | 1 | #ifndef ALISPLINEFIT_H |
2 | #define ALISPLINEFIT_H | |
3 | /* Copyright(c) 2006-07, ALICE Experiment at CERN, All rights reserved. * | |
4 | * See cxx source for full Copyright notice */ | |
5 | ||
6 | #include "TGraph.h" | |
7 | #include "TGraphSmooth.h" | |
8 | #include "TRandom.h" | |
9 | #include "TSpline.h" | |
10 | #include "TLinearFitter.h" | |
11 | #include "TDecompSVD.h" | |
12 | #include "TDecompSparse.h" | |
13 | #include "TMatrixDSparse.h" | |
14 | #include "TF1.h" | |
15 | #include "TH1F.h" | |
16 | #include "TObject.h" | |
17 | #include "TClonesArray.h" | |
18 | ||
19 | #include "TTreeStream.h" | |
20 | ||
21 | class AliSplineFit : public TObject { | |
22 | public: | |
23 | AliSplineFit(); | |
24 | AliSplineFit(const AliSplineFit&); | |
25 | ~AliSplineFit(); | |
26 | AliSplineFit& operator=(const AliSplineFit&); | |
27 | Double_t Eval(Double_t x, Int_t deriv=0) const; | |
28 | void InitKnots(TGraph * graph, Int_t min, Int_t iter, Double_t maxDelta); | |
29 | void MakeKnots0(TGraph * graph, Double_t maxdelta, Int_t minpoints); | |
30 | void SplineFit(Int_t nder); | |
31 | void MakeSmooth(TGraph * graph, Float_t ratio, char * type); | |
32 | void Update(TSpline3 *spline, Int_t nknots); | |
8625679a | 33 | void Cleanup(); |
0dd3a2ac | 34 | Int_t GetKnots() const {return fN;} |
35 | Double_t* GetX() const {return fX;} | |
36 | Double_t* GetY0() const {return fY0;} | |
37 | Double_t* GetY1() const {return fY1;} | |
38 | // | |
39 | // Test functions | |
40 | // | |
41 | TGraph * MakeGraph(Double_t xmin, Double_t xmax, Int_t npoints, Int_t deriv=0) const ; | |
42 | TGraph * MakeDiff(TGraph * graph) const ; | |
43 | TH1F * MakeDiffHisto(TGraph * graph) const; | |
44 | // | |
45 | static void Test(Int_t npoints=2000, Int_t ntracks=100, Float_t snoise=0.05); | |
46 | // | |
47 | static TGraph * GenerGraph(Int_t npoints, Double_t fraction, Double_t s1, Double_t s2, Double_t s3, Int_t der=0); | |
48 | static TGraph * GenerNoise(TGraph * graph0, Double_t s0); | |
49 | ||
50 | protected: | |
51 | ||
52 | // | |
53 | // working parameters for spline fit | |
54 | // | |
55 | Int_t OptimizeKnots(Int_t nIter); | |
56 | Float_t CheckKnot(Int_t iKnot); | |
57 | Bool_t RefitKnot(Int_t iKnot); | |
58 | // | |
59 | Bool_t fBDump; // dump debug information flag | |
60 | TGraph *fGraph; //! initial graph | |
61 | Int_t fNmin; // number of points per one knot in iteration 0 | |
62 | Double_t fSigma; // locally estimated sigma | |
63 | Double_t fMaxDelta;// maximal deviation of the spline fit | |
64 | Int_t fN0; // number of knots in iteration 0 | |
65 | TClonesArray *fParams; // object array of parameters in knots | |
66 | TClonesArray *fCovars; // object array of covariance in knots | |
8625679a | 67 | Int_t *fIndex; //[fN0] index of point corresponding to knot |
deebe992 | 68 | static TLinearFitter* fitterStatic(); // static fitter to save processing time |
0dd3a2ac | 69 | // |
70 | // | |
71 | // | |
72 | Int_t fN; // number of knots after compression | |
73 | Double_t fChi2; // chi2 per degree of freedom | |
8625679a | 74 | Double_t *fX; //[fN] - xknot value |
75 | Double_t *fY0; //[fN] - y value at X | |
76 | Double_t *fY1; //[fN] - y derivative value at X | |
77 | Double_t *fChi2I; //[fN] - chi2 on interval | |
78 | ClassDef(AliSplineFit, 1); | |
0dd3a2ac | 79 | }; |
80 | #endif |