]>
Commit | Line | Data |
---|---|---|
38a7a276 | 1 | #ifndef TSTATTOOLKIT_H |
21f3a443 | 2 | #define TSTATTOOLKIT_H |
3 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
4 | * See cxx source for full Copyright notice */ | |
5 | ||
3d7cc0b4 | 6 | // |
7 | // some utilities which do net exist in the standard ROOT | |
8 | // | |
21f3a443 | 9 | |
10 | #include "TObject.h" | |
11 | #include "TVectorD.h" | |
12 | #include "TMatrixD.h" | |
3d7cc0b4 | 13 | //#include "TGraph2D.h" |
14 | //#include "TGraph.h" | |
21f3a443 | 15 | |
16 | class TH1F; | |
3d7cc0b4 | 17 | class TH1; |
21f3a443 | 18 | class TH3; |
19 | class TString; | |
20 | class TTree; | |
3d7cc0b4 | 21 | class TGraph; |
22 | class TGraph2D; | |
377a7d60 | 23 | class TCanvas; |
24 | class TMultiGraph; | |
1a8f4649 | 25 | class TGraphErrors; |
377a7d60 | 26 | |
21f3a443 | 27 | class TStatToolkit : public TObject |
28 | { | |
29 | public: | |
30 | TStatToolkit(); | |
31 | virtual ~TStatToolkit(); | |
32 | // | |
33 | // | |
34 | // | |
35 | static void EvaluateUni(Int_t nvectors, Double_t *data, Double_t &mean, Double_t &sigma, Int_t hh); | |
36 | static void EvaluateUniExternal(Int_t nvectors, Double_t *data, Double_t &mean, Double_t &sigma, Int_t hh, Float_t externalfactor=1); | |
37 | static Int_t Freq(Int_t n, const Int_t *inlist, Int_t *outlist, Bool_t down); | |
38 | // | |
39 | // HISTOGRAMS TOOLS | |
40 | // | |
3d7cc0b4 | 41 | static void TruncatedMean(const TH1 * his, TVectorD *param, Float_t down=0, Float_t up=1.0, Bool_t verbose=kFALSE); |
21f3a443 | 42 | static void LTM(TH1F * his, TVectorD *param=0 , Float_t fraction=1, Bool_t verbose=kFALSE); |
94a43b22 | 43 | static Double_t FitGaus(TH1* his, TVectorD *param=0, TMatrixD *matrix=0, Float_t xmin=0, Float_t xmax=0, Bool_t verbose=kFALSE); |
21f3a443 | 44 | static Double_t FitGaus(Float_t *arr, Int_t nBins, Float_t xMin, Float_t xMax, TVectorD *param=0, TMatrixD *matrix=0, Bool_t verbose=kFALSE); |
3d7cc0b4 | 45 | static Float_t GetCOG(const Short_t *arr, Int_t nBins, Float_t xMin, Float_t xMax, Float_t *rms=0, Float_t *sum=0); |
21f3a443 | 46 | |
47 | static TGraph2D * MakeStat2D(TH3 * his, Int_t delta0, Int_t delta1, Int_t type); | |
48 | static TGraph * MakeStat1D(TH3 * his, Int_t delta1, Int_t type); | |
df0a2a0a | 49 | // |
50 | // Graph tools | |
51 | // | |
377a7d60 | 52 | static TGraph * MakeGraphSparse(TTree * tree, const char * expr="Entry", const char * cut="1", Int_t mstyle=25, Int_t mcolor=1, Float_t msize=-1, Float_t offset=0.0); |
1a8f4649 | 53 | static TGraphErrors * MakeGraphErrors(TTree * tree, const char * expr="Entry", const char * cut="1", Int_t mstyle=25, Int_t mcolor=1, Float_t msize=-1, Float_t offset=0.0); |
54 | ||
21f3a443 | 55 | // |
56 | // Fitting function | |
57 | // | |
88b1c775 | 58 | static TString* FitPlane(TTree * tree, const char* drawCommand, const char* formula, const char* cuts, Double_t & chi2, Int_t &npoints, TVectorD &fitParam, TMatrixD &covMatrix, Float_t frac=-1, Int_t start=0, Int_t stop=10000000, Bool_t fix0=kFALSE); |
cb1d20de | 59 | static TString* FitPlaneFixed(TTree * tree, const char* drawCommand, const char* formula, const char* cuts, Double_t & chi2, Int_t &npoints, TVectorD &fitParam, TMatrixD &covMatrix, Float_t frac=-1, Int_t start=0, Int_t stop=10000000); |
7c9cf6e4 | 60 | // |
61 | //Linear fitter helper function | |
62 | // | |
cb1d20de | 63 | static TString* FitPlaneConstrain(TTree * tree, const char* drawCommand, const char* formula, const char* cuts, Double_t & chi2, Int_t &npoints, TVectorD &fitParam, TMatrixD &covMatrix, Float_t frac=-1, Int_t start=0, Int_t stop=10000000, Double_t constrain=-1); |
3d7cc0b4 | 64 | static Int_t GetFitIndex(const TString fString, const TString subString); |
65 | static TString FilterFit(const TString &input, const TString filter, TVectorD &vec, TMatrixD &covar); | |
7c9cf6e4 | 66 | static void Update1D(Double_t delta, Double_t sigma, Int_t s1, TMatrixD ¶m, TMatrixD &covar); |
3d7cc0b4 | 67 | static void Constrain1D(const TString &input, const TString filter, TVectorD ¶m, TMatrixD & covar, Double_t mean, Double_t sigma); |
ae45c94d | 68 | static TString MakeFitString(const TString &input, const TVectorD ¶m, const TMatrixD & covar, Bool_t verbose=kFALSE); |
377a7d60 | 69 | // |
70 | // TTree function for the trneding | |
71 | // | |
72 | static Int_t MakeStatAlias(TTree * tree, const char * expr, const char * cut, const char * alias); | |
73 | static Int_t SetStatusAlias(TTree * tree, const char * expr, const char * cut, const char * alias); | |
74 | static TMultiGraph* MakeStatusMultGr(TTree * tree, const char * expr, const char * cut, const char * alias, Int_t igr); | |
75 | static void AddStatusPad(TCanvas* c1, Float_t padratio, Float_t bottommargin); | |
76 | static void DrawStatusGraphs(TObjArray* oaMultGr); | |
21f3a443 | 77 | // |
78 | // TestFunctions: | |
79 | // | |
80 | static void TestGausFit(Int_t nhistos=5000); | |
7c9cf6e4 | 81 | |
21f3a443 | 82 | ClassDef(TStatToolkit,0) // Various mathematical tools for physics analysis - which are not included in ROOT TMath |
83 | ||
84 | }; | |
85 | #endif |