| 1 |
mayorov |
1.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 |