]>
Commit | Line | Data |
---|---|---|
1 | #ifndef TSTATTOOLKIT_H | |
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 | ||
6 | // | |
7 | // some utilities which do net exist in the standard ROOT | |
8 | // | |
9 | ||
10 | #include "TObject.h" | |
11 | #include "TVectorD.h" | |
12 | #include "TMatrixD.h" | |
13 | //#include "TGraph2D.h" | |
14 | //#include "TGraph.h" | |
15 | ||
16 | class TH1F; | |
17 | class TH1; | |
18 | class TH2; | |
19 | class TH3; | |
20 | class TString; | |
21 | class TTree; | |
22 | class TGraph; | |
23 | class TGraph2D; | |
24 | class TCanvas; | |
25 | class TMultiGraph; | |
26 | class TGraphErrors; | |
27 | class TObjString; | |
28 | class TCut; | |
29 | class TTreeSRedirector; | |
30 | ||
31 | ||
32 | class TStatToolkit : public TObject | |
33 | { | |
34 | public: | |
35 | TStatToolkit(); | |
36 | virtual ~TStatToolkit(); | |
37 | // | |
38 | // | |
39 | // | |
40 | static void EvaluateUni(Int_t nvectors, Double_t *data, Double_t &mean, Double_t &sigma, Int_t hh); | |
41 | static void EvaluateUniExternal(Int_t nvectors, Double_t *data, Double_t &mean, Double_t &sigma, Int_t hh, Float_t externalfactor=1); | |
42 | static Int_t Freq(Int_t n, const Int_t *inlist, Int_t *outlist, Bool_t down); | |
43 | // | |
44 | // HISTOGRAMS TOOLS | |
45 | // | |
46 | static void TruncatedMean(const TH1 * his, TVectorD *param, Float_t down=0, Float_t up=1.0, Bool_t verbose=kFALSE); | |
47 | static void MedianFilter(TH1 * his1D, Int_t nmedian); | |
48 | static Bool_t LTMHisto(TH1 * his, TVectorD ¶m , Float_t fraction=1); | |
49 | // | |
50 | static void LTM(TH1 * his, TVectorD *param=0 , Float_t fraction=1, Bool_t verbose=kFALSE); | |
51 | static Double_t FitGaus(TH1* his, TVectorD *param=0, TMatrixD *matrix=0, Float_t xmin=0, Float_t xmax=0, Bool_t verbose=kFALSE); | |
52 | 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); | |
53 | 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); | |
54 | ||
55 | static TGraph2D * MakeStat2D(TH3 * his, Int_t delta0, Int_t delta1, Int_t type); | |
56 | static TGraphErrors * MakeStat1D(TH2 * his, Int_t deltaBin, Double_t fraction, Int_t returnType, Int_t markerStyle, Int_t markerColor); | |
57 | // | |
58 | // Graph tools | |
59 | // | |
60 | 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); | |
61 | 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); | |
62 | ||
63 | // | |
64 | // Fitting function | |
65 | // | |
66 | 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); | |
67 | 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); | |
68 | // | |
69 | //Linear fitter helper function | |
70 | // | |
71 | 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); | |
72 | static Int_t GetFitIndex(const TString fString, const TString subString); | |
73 | static TString FilterFit(const TString &input, const TString filter, TVectorD &vec, TMatrixD &covar); | |
74 | static void Update1D(Double_t delta, Double_t sigma, Int_t s1, TMatrixD ¶m, TMatrixD &covar); | |
75 | static void Constrain1D(const TString &input, const TString filter, TVectorD ¶m, TMatrixD & covar, Double_t mean, Double_t sigma); | |
76 | static TString MakeFitString(const TString &input, const TVectorD ¶m, const TMatrixD & covar, Bool_t verbose=kFALSE); | |
77 | // | |
78 | // TTree function for the trending | |
79 | // | |
80 | static Int_t MakeStatAlias(TTree * tree, const char * expr, const char * cut, const char * alias); | |
81 | static Int_t SetStatusAlias(TTree * tree, const char * expr, const char * cut, const char * alias); | |
82 | static TMultiGraph* MakeStatusMultGr(TTree * tree, const char * expr, const char * cut, const char * alias, Int_t igr=0); | |
83 | static void AddStatusPad(TCanvas* c1, Float_t padratio, Float_t bottommargin); | |
84 | static void DrawStatusGraphs(TObjArray* oaMultGr); | |
85 | static TTree* WriteStatusToTree(TObject* oStatusGr); | |
86 | static TMultiGraph* MakeStatusLines(TTree * tree, const char * expr, const char * cut, const char * alias); | |
87 | static void MakeSummaryTree(TTree* treeIn, TTreeSRedirector *pcstream, TObjString& sumID, TCut &selection); | |
88 | // | |
89 | // TTree function for robust draw | |
90 | // | |
91 | static TH1* DrawHistogram(TTree * tree, const char* drawCommand, const char* cuts = "1", const char* hname = "histo", const char* htitle = "histo", Int_t nsigma = 4, Float_t fraction = 0.75); | |
92 | // | |
93 | // TestFunctions: | |
94 | // | |
95 | static void TestGausFit(Int_t nhistos=5000); | |
96 | static void CheckTreeAliases(TTree * tree, Int_t ncheck); | |
97 | ClassDef(TStatToolkit,0) // Various mathematical tools for physics analysis - which are not included in ROOT TMath | |
98 | ||
99 | }; | |
100 | #endif |