Adding a TStatToolkit
[u/mrichter/AliRoot.git] / STAT / TStatToolkit.h
1 #ifndef ALIMATHBASE_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  
8 #include "TObject.h"
9 #include "TVectorD.h"
10 #include "TMatrixD.h"
11 #include "TGraph2D.h"
12 #include "TGraph.h"
13
14 class TH1F;
15 class TH3;
16 class TString;
17 class TTree;
18  
19 class TStatToolkit : public TObject
20 {
21  public:
22   TStatToolkit();
23   virtual ~TStatToolkit();
24   //
25   //
26   //
27   static void    EvaluateUni(Int_t nvectors, Double_t *data, Double_t &mean, Double_t &sigma, Int_t hh);
28   static void    EvaluateUniExternal(Int_t nvectors, Double_t *data, Double_t &mean, Double_t &sigma, Int_t hh, Float_t externalfactor=1);
29   static Int_t  Freq(Int_t n, const Int_t *inlist, Int_t *outlist, Bool_t down);    
30   //
31   // HISTOGRAMS TOOLS
32   //
33   static  void TruncatedMean(TH1F * his, TVectorD *param, Float_t down=0, Float_t up=1.0, Bool_t verbose=kFALSE);
34   static void LTM(TH1F * his, TVectorD *param=0 , Float_t fraction=1,  Bool_t verbose=kFALSE);
35   static Double_t  FitGaus(TH1F* his, TVectorD *param=0, TMatrixD *matrix=0, Float_t xmin=0, Float_t xmax=0,  Bool_t verbose=kFALSE);
36   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);
37   static Float_t  GetCOG(Short_t *arr, Int_t nBins, Float_t xMin, Float_t xMax, Float_t *rms=0, Float_t *sum=0);
38
39   static TGraph2D *  MakeStat2D(TH3 * his, Int_t delta0, Int_t delta1, Int_t type);
40   static TGraph *  MakeStat1D(TH3 * his, Int_t delta1, Int_t type);
41
42   //
43   // Fitting function
44   //
45   static TString* FitPlane(TTree * tree, const char* drawCommand, const char* formula, const char* cuts, Double_t & chi2, Int_t &npoints,  TVectorD &fitParam, TMatrixD &covMatrix, Int_t start=0, Int_t stop=10000000);
46
47
48
49
50   //
51   // TestFunctions:
52   //
53  static  void TestGausFit(Int_t nhistos=5000);
54     
55  ClassDef(TStatToolkit,0) // Various mathematical tools for physics analysis - which are not included in ROOT TMath
56  
57 };
58 #endif