1 |
{
|
2 |
TFile * in_file = new TFile("TMVA-phys-13jan2009.root", "READ");
|
3 |
TFile * in_file2 = new TFile("TMVA-qcd-09jan2009.root", "READ");
|
4 |
TH1F * _sig = (TH1F *)in_file.Get("Method_Likelihood/Likelihood/MVA_Likelihood_S");
|
5 |
TH1F * _bg = (TH1F *)in_file.Get("Method_Likelihood/Likelihood/MVA_Likelihood_B");
|
6 |
TH1F * _qcd = (TH1F *)in_file2.Get("Method_Likelihood/Likelihood/MVA_Likelihood_B");
|
7 |
|
8 |
// output file
|
9 |
TFile * out_file = new TFile("rarfit_templates-13jan2009.root", "RECREATE");
|
10 |
// create new TTree
|
11 |
TTree * sig_template = new TTree("signal_template","signal_template");
|
12 |
TTree * bg_template = new TTree("background_template","background_template");
|
13 |
TTree * qcd_template = new TTree("qcd_template","qcd_template");
|
14 |
|
15 |
Double_t _ll = -1.0;
|
16 |
TBranch * b_ll = sig_template->Branch("MVA_Likelihood", &_ll, "MVA_Likelihood/D");
|
17 |
// loop over bins in the signal histogram
|
18 |
int n_bins = _sig->GetNbinsX();
|
19 |
double integral = _sig->Integral();
|
20 |
double n_entries = _sig->GetEntries();
|
21 |
double scale_factor = n_entries/integral; // histogram is scaled, so need to scale it back
|
22 |
for (int bin=1; bin != n_bins+1; bin++){
|
23 |
_ll = _sig->GetBinCenter(bin);
|
24 |
//loop over bin content
|
25 |
int bin_entries = (int)(_sig->GetBinContent(bin) * scale_factor + 0.5);
|
26 |
for (int _entry=0; _entry!=bin_entries; _entry++){
|
27 |
if (_ll < 0.0) cout << _ll << endl;
|
28 |
sig_template->Fill();
|
29 |
}
|
30 |
}
|
31 |
|
32 |
_ll = -1.0;
|
33 |
b_ll = bg_template->Branch("MVA_Likelihood", &_ll, "MVA_Likelihood/D");
|
34 |
// loop over bins in the background histogram
|
35 |
n_bins = _bg->GetNbinsX();
|
36 |
integral = _bg->Integral();
|
37 |
n_entries = _bg->GetEntries();
|
38 |
scale_factor = n_entries/integral; // histogram is scaled, so need to scale it back
|
39 |
for (int bin=1; bin != n_bins+1; bin++){
|
40 |
_ll = _bg->GetBinCenter(bin);
|
41 |
//loop over bin content
|
42 |
int bin_entries = (int)(_bg->GetBinContent(bin) * scale_factor + 0.5);
|
43 |
for (int _entry=0; _entry!=bin_entries; _entry++){
|
44 |
if (_ll < 0.0) cout << _ll << endl;
|
45 |
bg_template->Fill();
|
46 |
}
|
47 |
}
|
48 |
|
49 |
_ll = -1.0;
|
50 |
b_ll = qcd_template->Branch("MVA_Likelihood", &_ll, "MVA_Likelihood/D");
|
51 |
// loop over bins in the background histogram
|
52 |
n_bins = _qcd->GetNbinsX();
|
53 |
integral = _qcd->Integral();
|
54 |
n_entries = _qcd->GetEntries();
|
55 |
scale_factor = n_entries/integral; // histogram is scaled, so need to scale it back
|
56 |
for (int bin=1; bin != n_bins+1; bin++){
|
57 |
_ll = _qcd->GetBinCenter(bin);
|
58 |
//loop over bin content
|
59 |
int bin_entries = (int)(_qcd->GetBinContent(bin) * scale_factor + 0.5);
|
60 |
for (int _entry=0; _entry!=bin_entries; _entry++){
|
61 |
if (_ll < 0.0) cout << _ll << endl;
|
62 |
qcd_template->Fill();
|
63 |
}
|
64 |
}
|
65 |
|
66 |
out_file->cd();
|
67 |
|
68 |
out_file->Write();
|
69 |
delete out_file;
|
70 |
delete in_file;
|
71 |
}
|
72 |
|