4dfe6ef88ffe170fd9f91c4ed8036a76693ae8d5
[u/mrichter/AliRoot.git] / STAT / TStatToolkit.h
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 TH3;
19 class TString;
20 class TTree;
21 class TGraph;
22 class TGraph2D;
23  
24 class TStatToolkit : public TObject
25 {
26  public:
27   TStatToolkit();
28   virtual ~TStatToolkit();
29   //
30   //
31   //
32   static void    EvaluateUni(Int_t nvectors, Double_t *data, Double_t &mean, Double_t &sigma, Int_t hh);
33   static void    EvaluateUniExternal(Int_t nvectors, Double_t *data, Double_t &mean, Double_t &sigma, Int_t hh, Float_t externalfactor=1);
34   static Int_t  Freq(Int_t n, const Int_t *inlist, Int_t *outlist, Bool_t down);    
35   //
36   // HISTOGRAMS TOOLS
37   //
38   static  void TruncatedMean(const TH1 * his, TVectorD *param, Float_t down=0, Float_t up=1.0, Bool_t verbose=kFALSE);
39   static void LTM(TH1F * his, TVectorD *param=0 , Float_t fraction=1,  Bool_t verbose=kFALSE);
40   static Double_t  FitGaus(TH1F* his, TVectorD *param=0, TMatrixD *matrix=0, Float_t xmin=0, Float_t xmax=0,  Bool_t verbose=kFALSE);
41   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);
42   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);
43
44   static TGraph2D *  MakeStat2D(TH3 * his, Int_t delta0, Int_t delta1, Int_t type);
45   static TGraph *  MakeStat1D(TH3 * his, Int_t delta1, Int_t type);
46   //
47   // Graph tools
48   //
49   static TGraph * MakeGraphSparse(TTree * tree, const char * expr="Entry", const char * cut="1");
50   //
51   // Fitting function
52   //
53   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);
54   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);
55   //
56   //Linear fitter helper function
57   //
58   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);
59   static Int_t GetFitIndex(const TString fString, const TString subString);
60  static TString FilterFit(const TString &input, const TString filter, TVectorD &vec, TMatrixD &covar);
61  static void Update1D(Double_t delta, Double_t sigma, Int_t s1, TMatrixD &param, TMatrixD &covar);
62   static void   Constrain1D(const TString &input, const TString filter, TVectorD &param, TMatrixD & covar, Double_t mean, Double_t sigma);
63   static TString  MakeFitString(const TString &input, const TVectorD &param, const TMatrixD & covar);
64
65   //
66   // TestFunctions:
67   //
68  static  void TestGausFit(Int_t nhistos=5000);
69
70  ClassDef(TStatToolkit,0) // Various mathematical tools for physics analysis - which are not included in ROOT TMath
71  
72 };
73 #endif