1 |
#include <iostream> |
2 |
#include <fstream> |
3 |
#include <stdio.h> |
4 |
//#include <vector.h> |
5 |
#include <math.h> |
6 |
#include <TH1F.h> |
7 |
#include <TH2F.h> |
8 |
#include <TH1D.h> |
9 |
#include <TH2D.h> |
10 |
#include <TVectorT.h> |
11 |
#include <TFile.h> |
12 |
#include <TROOT.h> |
13 |
#include <TList.h> |
14 |
#include <TString.h> |
15 |
#include <TLine.h> |
16 |
#include <TGraphAsymmErrors.h> |
17 |
#include <TGraphErrors.h> |
18 |
#include <TChain.h> |
19 |
#include <TCutG.h> |
20 |
#include <TF1.h> |
21 |
#include <TCanvas.h> |
22 |
#include <TObjString.h> |
23 |
#include <TMath.h> |
24 |
#include <TMVA/Timer.h> |
25 |
#include <RanGen.h> |
26 |
|
27 |
|
28 |
#ifndef PAMUNFOLD_H |
29 |
#define PAMUNFOLD_H |
30 |
|
31 |
using std::vector; |
32 |
|
33 |
class PamUnfold: public TNamed{ |
34 |
public: |
35 |
PamUnfold(TString name, TString title); |
36 |
~PamUnfold(); |
37 |
PamUnfold(TString name, TString title, TH1D* measured, TH2D* smearing); //passing _measured and _smearing |
38 |
|
39 |
void AddExcludedBin(Int_t bin); |
40 |
|
41 |
TH1D* GetMeasured(); |
42 |
TH2D* GetSmearing(); |
43 |
TH1D* GetUnfolded(); |
44 |
TList* GetBinHistList(); |
45 |
|
46 |
void SetMeasured(TH1D*); |
47 |
void SetSmearing(TH2D*); |
48 |
void SetPrior(TH1D*); |
49 |
void SetNormalization(TH1D*); |
50 |
void SetImproved(Bool_t improv=kTRUE); |
51 |
void SetSmoothing(Bool_t smooth, TString smooth_opt="ROOT"); |
52 |
void SetNsamples(UInt_t nsamples); |
53 |
void SetMaxSteps(UInt_t max_steps); |
54 |
void SetMinChi2(Double_t min_chi2); |
55 |
|
56 |
Bool_t IsSmNormalized(); |
57 |
|
58 |
void Unfold(); |
59 |
void ImprovedUnfold(); |
60 |
void IterativeUnfolding(UInt_t niter=0, TList* list=NULL); |
61 |
|
62 |
void Draw(TString path); |
63 |
|
64 |
static void WMovAvSmooth(TH1D* input, vector<Int_t>&excl); |
65 |
static Double_t GetChi2H( TH1D* h1, TH1D* h2 ); |
66 |
|
67 |
|
68 |
|
69 |
private: |
70 |
TH1D* _measured; |
71 |
TH1D* _unfolded; |
72 |
TH1D* _old_unfolded; |
73 |
TH2D* _smearing; |
74 |
TH2D* _smearing_sample; |
75 |
TH1D* _prior; |
76 |
TH1D* _norm; |
77 |
TList* _bin_list; |
78 |
TList* _bin_hist_list; |
79 |
Bool_t _smooth; |
80 |
TString _smooth_opt; |
81 |
UInt_t _max_steps; |
82 |
Bool_t _is_improved; |
83 |
UInt_t _nsamples; |
84 |
Double_t _min_chi2; |
85 |
vector<Int_t> _excluded_bins; |
86 |
RanGen* rangen; |
87 |
TCanvas* canv; |
88 |
|
89 |
Bool_t IsBinningOK(); |
90 |
void Init(); |
91 |
void NormalizeMatrix(); |
92 |
void SampleMatrix(); |
93 |
void BuildFlux(); |
94 |
void FindSplitting(Int_t n, Int_t &nx, Int_t &ny); |
95 |
|
96 |
ClassDef(PamUnfold, 1); |
97 |
}; |
98 |
|
99 |
|
100 |
#endif |