TemplateStudies/fastplot.C at master · c-alexe/TemplateStudies · 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
{

  int do_pull = 1;

  TString systname = "scale_A4";

  TString proc = "A4";

  vector<TString> phasespace = {"x0p30_y3p00",
				"x0p40_y3p50"
  };

  vector<TString> runs = {"wp",
			  //"wm", "z"
  };
  vector<TString> dx = {"1", "2", "3", "4"
			//"6",
			//"7","8","9","10","11","12"
  };
  vector<TString> dy = {"2", "4", "6"
			//"2",
			//"4"
  };

  for(int ip = 0; ip<phasespace.size(); ip++ ){
    for(int ip2 = 0; ip2<runs.size(); ip2++ ){
          for(int idx = 0; idx<dx.size(); idx++ ){
	    for(int idy = 0; idy<dy.size(); idy++ ){
	      TFile* fin = TFile::Open("fout_fit_"+systname+"_"+phasespace[ip]+"_"+runs[ip2]+"_"+proc+"_x"+dx[idx]+"_y"+dy[idy]+".root", "READ");
	      TH1D* hinfo = (TH1D*)fin->Get("h_info");
	      TH1D* hstart = (TH1D*)fin->Get("h_start");
	      int ns = hinfo->GetBinContent(3);
	      TCanvas* c = new TCanvas("c", "canvas", 1200, 600);
	      c->Divide(2,1);
	      for(int i = 0; i < ns; i++){
 		if(do_pull){
		  TH2D* e_i = (TH2D*)fin->Get(Form("h_pull_%d",i));
		  TH2D* d_i = (TH2D*)fin->Get(Form("h_data_%d",i));
		  TH2D* d_i_clone = (TH2D*) d_i->Clone(phasespace[ip]+"_"+runs[ip2]+"_"+proc+"_x"+dx[idx]+"_y"+dy[idy]+"_"+TString(Form("%d",i)));
		  for(int ix=1; ix<=d_i_clone->GetXaxis()->GetNbins(); ix++){
		    for(int iy=1; iy<=d_i_clone->GetYaxis()->GetNbins(); iy++){
		      double pull = (d_i->GetBinContent(ix,iy) - hstart->GetBinContent(ix,iy))/hstart->GetBinError(ix,iy);
		      d_i_clone->SetBinContent(ix,iy, pull );
		    }
		  }
		  d_i_clone->SetMinimum(-3);
		  d_i_clone->SetMaximum(+3);
		  d_i_clone->SetStats(0);
		  c->cd(1);
		  d_i_clone->Draw("colz");
		  e_i->SetStats(0);
		  e_i->SetMinimum(-3);
		  e_i->SetMaximum(+3);
		  c->cd(2);
		  gPad->SetRightMargin(0.15);
		  e_i->Draw("COLZ");
		}
		else{
		  TH2D* e_i = (TH2D*)fin->Get(Form("h_exp_%d",i));
		  TH2D* d_i = (TH2D*)fin->Get(Form("h_data_%d",i));
		  TH2D* d_i_clone = (TH2D*) d_i->Clone(phasespace[ip]+"_"+runs[ip2]+"_"+proc+"_x"+dx[idx]+"_y"+dy[idy]+"_"+TString(Form("%d",i)));
		  d_i_clone->Divide(hstart);
		  d_i_clone->SetMinimum(0.5);
		  d_i_clone->SetMaximum(1.5);
		  d_i_clone->SetStats(0);
		  c->cd(1);
		  d_i_clone->Draw("colz");

		  e_i->Divide(d_i);
		  e_i->SetStats(0);
		  e_i->SetMinimum(0.95);
		  e_i->SetMaximum(1.05);

		  c->cd(2);
		  gPad->SetRightMargin(0.15);
		  e_i->Draw("COLZ");
		}
		c->SaveAs("plots/"+systname+"_"+TString(Form("_var%d", i))+"_"+phasespace[ip]+"_"+runs[ip2]+"_"+proc+"_x"+dx[idx]+"_y"+dy[idy]+".png");
		//if(i==1) return;
	      }
	      fin->Close();
	    }
	  }
    }
  }

  return;
}