#include <TFile.h>
#include <TDirectory.h>
#include <TH2.h>
#include <TCanvas.h>

void loose()
 {
  TFile *f1 = TFile::Open("tt_final_new.root");
//  TH2D *h1; gDirectory->GetObject("third_tight_mu_pt_eta", h1);
  TH2D *h2; gDirectory->GetObject("third_loose_mu_pt_eta", h2);
//  TH2D *nt2; gDirectory->GetObject("nt2_pt_eta", nt2);
//   h1->Sumw2(kTRUE); h1->SetTitle("tight muons");
//  h2->Sumw2(kTRUE); h2->SetTitle("loose muons");
//  nt2->Sumw2(kTRUE); nt2->SetTitle("b-jets");
 /* 
{
  double bins_x[] = {20, 40, 70};
  double bins_y[] = {0, 0.5,1, 1.5, 2.5};
  TH2D *h1 = new TH2D("3rd tight muon", "tight_ttbar_Z;Pt_{T}^{} [GeV/c];#eta",
                     (sizeof(bins_x) / sizeof(bins_x[0]) - 1), bins_x,
                     (sizeof(bins_y) / sizeof(bins_y[0]) - 1), bins_y);
  for (int i = 1; i <= h1->GetNbinsX(); i++)
    for (int j = 1; j <= h1->GetNbinsY(); j++)
      h1->SetBinContent(i, j, (i * i + j * j * j));
  h1->Draw("colz text e");
}  
*/

{


  double bins_x[] = { 20, 40, 70};
  double bins_y[] = {0, 0.5,1, 1.5, 2.5};
  TH2D *h2 = new TH2D("3rd loose muon", "loose_ttbar;Pt_{T}^{} [GeV/c];#eta",
                     (sizeof(bins_x) / sizeof(bins_x[0]) - 1), bins_x,
                     (sizeof(bins_y) / sizeof(bins_y[0]) - 1), bins_y);
  for (int i = 1; i <= h2->GetNbinsX(); i++)
    for (int j = 1; j <= h2->GetNbinsY(); j++)
      h2->SetBinContent(i, j, (i * i + j * j * j));
  h2->Draw("colz text e");
}

/*{
  double bins_x[] = { 20, 40, 70};
  double bins_y[] = {0, 0.5,1, 1.5, 2.5};
  TH2D *nt2 = new TH2D("nt2", "loose_ttbar_Z;Pt_{T}^{} [GeV/c];#eta",
                     (sizeof(bins_x) / sizeof(bins_x[0]) - 1), bins_x,
                     (sizeof(bins_y) / sizeof(bins_y[0]) - 1), bins_y);
  for (int i = 1; i <= nt2->GetNbinsX(); i++)
    for (int j = 1; j <= nt2->GetNbinsY(); j++)
      nt2->SetBinContent(i, j, (i * i + j * j * j));
  nt2->Draw("colz text e");
}


 
 TH2D *f = ((TH2D*)(h1->Clone()));
  f->SetNameTitle("f", "tight/loose");
  f->Divide(h2);
 
{
  double bins_x[] = { 20, 40, 70};
  double bins_y[] = {0, 0.5,1, 1.5, 2.5};
  TH2D *f = new TH2D("tight/loose", "loose_ttbar_Z;Pt_{T}^{} [GeV/c];#eta",
                     (sizeof(bins_x) / sizeof(bins_x[0]) - 1), bins_x,
                     (sizeof(bins_y) / sizeof(bins_y[0]) - 1), bins_y);
  for (int i = 1; i <= f->GetNbinsX(); i++)
    for (int j = 1; j <= f->GetNbinsY(); j++)
      f->SetBinContent(i, j, (i * i + j * j * j));
  f->Draw("colz text e");
}

 double mul = 1.0;
  if (h2->GetEntries()) mul = h1->GetEntries() / h2->GetEntries();
  if (mul != 1.0) mul /= (1.0 - mul);
  
  TH2D *Nt3 = ((TH2D*)(nt2->Clone()));
  Nt3->SetNameTitle("Nt3", "normalized b-jets");
  Nt3->Scale(mul);
*/  
  TCanvas *c1 = new TCanvas("c1", "c1", 800, 600) ;
//  c1->Divide(2, 1);
//  c1->cd(1);
  h2->Draw("colz text e");
//  c1->cd(2);
//  h1->Draw("colz text e");
/*  c1->cd(3);
  f->Draw("colz text e");
  c1->cd(4);
  nt2->Draw("colz text e");
*/
c1->SaveAs("loose_ttbar.pdf");
}