Plotting_Program/purity_plots.C at master · YonsiG/Plotting_Program · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
//Time:2021.6.29
//Last Modified:2021.6.29

#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
#include "TMath.h"
#include "TH1D.h"
#include <iostream>
#include <fstream>
#include <vector>
#include <TH2.h>
#include <TStyle.h>
#include <TCanvas.h>
#include "THStack.h"
#include "TH3F.h"
#include <TApplication.h>
#include <TEnv.h>
#include <TComplex.h>
#include <TGraph.h>
#include <TProfile.h>
#include <TProfile2D.h>
#include <TPad.h>
#include <TCanvas.h>
#include <TLegend.h>

using namespace std;

int main()
{
   bool isLog = 0;

   TFile *signalfile = new TFile("../../outfiles/C2V_3/samples_1_3_22/WZH_3_scaled.root");

   string histName = "WZH/plots_01_20_22/VBFJet_DeltaEta_3_0";

   TCanvas *c = new TCanvas("plot1", "", 800, 600);
   TPad *pad1 = new TPad("pad1", "pad1", 0, 0, 1.0, 1.0);
   THStack *stack = new THStack("ts1", "");
   c->SetMargin(0.1, 0.05, 0.1, 0.05);

   TH1D *plot_2pass = (TH1D *)signalfile->Get("VBFJet_DeltaEta_2match0");
   TH1D *plot_1pass = (TH1D *)signalfile->Get("VBFJet_DeltaEta_1match0");
   TH1D *plot_0pass = (TH1D *)signalfile->Get("VBFJet_DeltaEta_0match0");

   for (int ibin = 1; ibin < plot_2pass->GetNbinsX() + 1; ibin++)
   {
      plot_2pass->SetBinError(ibin, 0);
      plot_1pass->SetBinError(ibin, 0);
      plot_0pass->SetBinError(ibin, 0);
   }

   plot_2pass->SetFillColor(kCyan);
   plot_1pass->SetFillColor(kPink);
   plot_0pass->SetFillColor(kYellow);

   /****************N1T*****************/
   pad1->Draw();
   if (isLog)
      pad1->SetLogy();
   pad1->cd();

   plot_2pass->SetTitle("");
   plot_2pass->SetLineWidth(2.0);
   plot_2pass->SetStats(0);
   plot_2pass->GetYaxis()->SetTitle("Entries");
   plot_2pass->GetYaxis()->SetTitleSize(0.04);
   plot_2pass->GetYaxis()->SetTitleOffset(1);
   plot_2pass->GetYaxis()->SetTitleFont(70);
   plot_2pass->GetYaxis()->SetLabelSize(0.03);
   plot_2pass->GetYaxis()->SetLabelFont(70);
   plot_2pass->GetXaxis()->SetTitle("fatjet DDBvL score / GeV");
   plot_2pass->GetXaxis()->SetTitleSize(0.04);
   plot_2pass->GetXaxis()->SetTitleOffset(1);
   plot_2pass->GetXaxis()->SetTitleFont(70);
   plot_2pass->GetXaxis()->SetLabelSize(0.03);
   plot_2pass->GetXaxis()->SetLabelFont(70);

   stack->Add(plot_0pass);
   stack->Add(plot_1pass);
   stack->Add(plot_2pass);
   stack->Draw("H");

   TLegend *legend = new TLegend(0.65, 0.75, 0.88, 0.88);
   legend->AddEntry(plot_2pass, "2 matches", "f");
   legend->AddEntry(plot_1pass, "1 match", "f");
   legend->AddEntry(plot_0pass, "0 match", "f");
   legend->SetTextFont(20);
   legend->SetTextSize(0.025);
   legend->SetBorderSize(0);
   legend->Draw("same");
   //   legend -> SetHeader("ATLAS Internal");

   if (isLog)
      c->SaveAs(("../../plots/C2V_3/" + histName + "_log.pdf").c_str()); //save plots to disk
   if (!isLog)
      c->SaveAs(("../../plots/C2V_3/" + histName + "_linear.pdf").c_str());
}