Ana01PaperPlot.C

Go to the documentation of this file.

#include <vector>
#include <iostream>
#include <iomanip>

#include "TString.h"
#include "TH1.h"
#include "TH2.h"
#include "TCanvas.h"
#include "TArrow.h"
#include "TAxis.h"
#include "TROOT.h"
#include "TStyle.h"
#include "TLine.h"
#include "TLegend.h"
#include "TCollection.h"
#include "TGraph.h"
#include "TObjArray.h"
#include "TObject.h"
#include "TVirtualPad.h"
#include "TFile.h"
#include "TMarker.h"
#include "TLatex.h"
#include "TMath.h"
#include "TColor.h"
#include "TFrame.h"

#include "CAFAna/Analysis/Style.h"
#include "Utilities/rootlogon.C"


void DrawAxes(TH2* surface)
{
  double th24lo =  0.;
  double th24hi = 40.;

  double th34lo =  0.;
  double th34hi = 40.;

  TString dummyName = surface->GetName();
  dummyName += "_dummy";
  TH2* hDummy = (TH2*)surface->Clone(dummyName);
  hDummy->SetDirectory(0);
  hDummy->Reset();
  hDummy->SetTitle("");
  hDummy->GetXaxis()->SetRangeUser(th34lo, th34hi);
  hDummy->GetXaxis()->SetTitle("#theta_{34} (deg.)");
  hDummy->GetXaxis()->CenterTitle();
  hDummy->GetXaxis()->SetTitleOffset(0.9);
  hDummy->GetYaxis()->SetRangeUser(th24lo, th24hi);
  hDummy->GetYaxis()->SetTitle("#theta_{24} (deg.)");
  hDummy->GetYaxis()->CenterTitle();
  hDummy->GetYaxis()->SetTitleOffset(0.9);
  hDummy->GetXaxis()->SetTitleSize(0.07);
  hDummy->GetYaxis()->SetTitleSize(0.07);
  hDummy->GetXaxis()->SetLabelSize(0.06);
  hDummy->GetYaxis()->SetLabelSize(0.06);

  hDummy->Draw();

  return;
}


vector<TGraph*> GetContour(TH2* surface, double up, int color, int style, int fillcolor, bool fill=false)
{
  TVirtualPad* bak = gPad;

  const bool wasbatch = gROOT->IsBatch();
  gROOT->SetBatch(); // User doesn't want to see our temporary canvas
  gROOT->ForceStyle();

  TCanvas tmp;
  gStyle->SetOptStat(0);
  gPad->SetFillStyle(0);
  
  vector<TGraph*> results;

  tmp.cd();
  double minLike = surface->GetBinContent( surface->GetMinimumBin() );
  double maxLike = surface->GetBinContent( surface->GetMaximumBin() );
  cout << "TH1 Name = " << surface->GetName() << " : minLike = " << minLike << "  maxLike = " << maxLike << " up = " << up << endl;
  const double level = up;
  surface->SetContour(1, &level);
  //surface->SetContourLevel(0, up - minLike);
  surface->Smooth();
  surface->Draw("cont list");
  tmp.Update();
  tmp.Paint();

  gROOT->SetBatch(wasbatch);
  gPad = bak;

  TObjArray* contArr = (TObjArray*)gROOT->GetListOfSpecials()->FindObject("contours");
  TList* contList = (TList*)contArr->At(0);
  TIter next(contList);
  TGraph* gr(0);
  while ( gr = (TGraph*)next() )
    {
      TGraph* toSave = (TGraph*)gr->Clone();
      TGraph* toSave2 = (TGraph*)gr->Clone();
      toSave->SetLineWidth(2);
      toSave->SetLineColor(color);
      toSave->SetLineStyle(style);
      toSave2->SetLineWidth(2);
      toSave2->SetLineColor(color);
      toSave2->SetLineStyle(style);
      if(fill){
        toSave->SetPoint(toSave->GetN(), 0, 0);
        toSave->SetFillColor(fillcolor);
        toSave->Draw("f same");
      }
      toSave->Draw("l same");
      results.push_back( toSave );
      results.push_back( toSave2 );
    }
  return results;
}


void Ana01PaperPlot()
{

  const bool wasbatch = gROOT->IsBatch();
  
  gROOT->ForceStyle();

  gStyle->SetFillStyle(0);
  gStyle->SetPadBottomMargin(0.15);
  gStyle->SetPadLeftMargin(0.15);
  gStyle->SetLabelOffset(0.01,"y");
  
  double up68 = TMath::ChisquareQuantile(0.68, 1);
  double up90 = TMath::ChisquareQuantile(0.90, 1);

  TFile fFCfile("$NOVA_ANA/steriles/Ana01/input_FC_ana01_t2434_paperplot.root","READ");
  //  TFile fNus17file("$NOVA_ANA/steriles/Nus17/plot_predInterpSysts_nus17_shape_2D_lower.root","READ");
  TH2F *hFCcorr;
  //TH2F *hNus17;
  
  TCanvas *c1 = (TCanvas*)fFCfile.Get("cFCChi2");
  gROOT->SetBatch();
  gPad->SetFillStyle(0);
  hFCcorr = (TH2F*)c1->GetPrimitive("chi_me"); //hFCChi
  hFCcorr->SetDirectory(0);
  c1->Clear();
  c1->Draw();
  hFCcorr->Draw();
  
  std::string cname = "cFCcorr";
  std::string ctitle = "FC Correction";
  gROOT->SetBatch(wasbatch);
  TCanvas* c = new TCanvas(cname.c_str(), ctitle.c_str());
  gPad->SetFillStyle(0);
  DrawAxes(hFCcorr);
  TH2* hDummy2 = (TH2*)hFCcorr->Clone("hFCcorr");
  
  //TH2* hDummy1 = (TH2*)fNus17file.Get("h34vs24_9e20_shapestatS");

  Color_t kSigmaConfidenceColor   = kAzure+2;
  //Color_t k17SigmaConfidenceColor = kAzure+2;

  Int_t kFillColor68 = TColor::GetColorTransparent(kSigmaConfidenceColor, .20);
  Int_t kFillColor90 = TColor::GetColorTransparent(kSigmaConfidenceColor, .35);
  //Int_t k17FillColor68 = TColor::GetColorTransparent(k17SigmaConfidenceColor, .20);
  //Int_t k17FillColor90 = TColor::GetColorTransparent(k17SigmaConfidenceColor, .35);

  vector<TGraph*> vecFC68 = GetContour(hDummy2, up68, kBlack, kDashed, kFillColor68, true);
  vector<TGraph*> vecFC90 = GetContour(hDummy2, up90, kBlack, kSolid, kFillColor90, true);
  
  //vector<TGraph*> vec1768 = GetContour(hDummy1, up68, kAzure+2, kDashed, k17FillColor68, true);
  //vector<TGraph*> vec1790 = GetContour(hDummy1, up90, kAzure+2, kSolid, k17FillColor90, true);

  
  TLatex *tex = new TLatex();
  tex->SetNDC();
  tex->SetTextFont(42);
  tex->SetTextSize(0.05);
  tex->SetTextAlign(11);
  tex->DrawLatex(0.2, 0.80, "NOvA 6.05#times10^{20} POT-equiv.");
  tex->DrawLatex(0.2, 0.8-1.*0.055, "#Deltam^{2}_{32} = 2.44#times10^{-3} eV^{2}");
  tex->DrawLatex(0.2, 0.8-2.*0.055, "#theta_{13} = 8.5#circ, ^{}#theta_{23} = 45#circ");
  tex->DrawLatex(0.2, 0.8-3.*0.055, "#Deltam^{2}_{41} = 0.5 eV^{2}");
    
  TLegend* leg = new TLegend(0.62, 0.53, 0.87, 0.73);
  leg->SetBorderSize(0);
  leg->SetFillColor(kWhite);
  leg->SetFillStyle(0);
  leg->SetFillStyle(0);
  leg->SetTextFont(42);

  leg->AddEntry(vecFC68[0],"68% C.L.","L");
  leg->AddEntry(vecFC90[0],"90% C.L.","L");
  //leg->AddEntry(vec1768[0],"NOvA 2017: 68% C.L.","L");
  //leg->AddEntry(vec1790[0],"NOvA 2017: 90% C.L.","L");

  leg->SetTextSize(0.05);
  leg->Draw();
  
  gPad->RedrawAxis();

  gPad->Modified();
  gPad->Update();

  gPad->Print("ana01_paper_plot_fc_t2434.pdf");

}