bin_composition_pie_chart.C File Reference

#include "OscLib/IOscCalc.h"
#include "CAFAna/Analysis/Calcs.h"
#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/Vars/FitVars.h"
#include "CAFAna/Analysis/Exposures.h"
#include "3FlavorAna/Ana2018/nue/joint_fit_2018_tools.h"
#include "Utilities/func/MathUtil.h"
#include "TCanvas.h"
#include "TH1.h"
#include "TLatex.h"
#include "TLegend.h"
#include "TMath.h"
#include "TPie.h"
#include "TPolyLine.h"
#include "TRandom3.h"

Go to the source code of this file.

Functions
void	leg (double x0, double y0, double x1, double y1)
void	pie_arc (double cx, double cy, double a0, double a1, double r, int width, int lineCol, int fillCol, double fudgeR=0)
double	IntInBin (TH1 *h, int bin)
TH1 *	PredictCosmic (std::string fname, std::string name)
void	bin_composition_pie_chart (bool isFHC)

Variables
const double	explode = .05
const double	R = .35
const int	cols [3] = {kRed-7, kBlue-7, kGray}
const int	text_cols [3] = {kRed, kBlue, kGray+2}
const int	internal_line_col = kGray+2
const std::string	component_labels [3] = {"Signal", "Beam bkg", "Cosmic bkg"}
const std::string	sample_labels [3] = {"Low CVN", "High CVN", "Peripheral"}

Function Documentation

void bin_composition_pie_chart

(

bool

isFHC

)

Definition at line 86 of file bin_composition_pie_chart.C.

References a1, b, calc, col, cols, component_labels, std::cos(), dCP, plot_validation_datamc_2018::decompStr, ana::DefaultOscCalc(), dmsq32, APDHVSetting::dummy, explode, util::frac(), GetNueCosmics2018(), GetNuePrediction2018(), hbkg, MECModelEnuComparisons::i, internal_line_col, IntInBin(), calib::j, ana::Sign::kAntiNu, ana::Current::kCC, ana::kFitDeltaInPiUnits, ana::kFitDmSq32, ana::kFitSinSqTheta23, ana::Sign::kNu, ana::Flavors::kNuMuToNuE, label, leg(), cet::sqlite::max(), pie_arc(), pot, plot_validation_datamc::pred, ana::IPrediction::Predict(), ana::IPrediction::PredictComponent(), R, central_limit::rand, sample_labels, ana::FitVarWithPrior::SetValue(), ana::ConstrainedFitVarWithPrior::SetValue(), ana::FitDmSq32::SetValue(), std::sin(), ssth23, string, PandAna.Demos.pi0_spectra::tot, ana::Spectrum::ToTH1(), x1, xbins, make_syst_table_plots::xlo, and y1.

{
  // Need a calc before we can do anything, seed at 2017 best fit
  double dCP = 1.21;
  double ssth23 = 0.446;
  double dmsq32 = 0.00244;

  osc::IOscCalcAdjustable* calc = DefaultOscCalc();

  kFitDeltaInPiUnits.SetValue(calc,dCP);
  kFitDmSq32.SetValue(calc,dmsq32);
  kFitSinSqTheta23.SetValue(calc,ssth23);

  // Pull the prediction
  std::string decompStr = isFHC ? "combo" : "prop";
  std::string beamStr = isFHC ? "fhc" : "rhc";
  const IPrediction *pred =
    GetNuePrediction2018(decompStr, calc, false, beamStr, true);

  double pot = 9.48e20; // From Nitish, update to official once committed

  TH1 *psig = pred->PredictComponent(calc, Flavors::kNuMuToNuE,
                                     Current::kCC, Sign::kNu).ToTH1(pot);
  if (!isFHC)
    psig =  pred->PredictComponent(calc, Flavors::kNuMuToNuE,
                                   Current::kCC, Sign::kAntiNu).ToTH1(pot);
  TH1 *pbb  = pred->Predict(calc).ToTH1(pot);
  pbb->Add(psig,-1); // beam bkgd = total - sig

  // And the cosmic prediction
  TH1D *pcos = GetNueCosmics2018(beamStr).first;


  TCanvas *canPie = new TCanvas("a", "b", 600, 600);

  double vals[] = { IntInBin(psig,0), IntInBin(pbb,0), IntInBin(pcos,0),
                    IntInBin(psig,1), IntInBin(pbb,1), IntInBin(pcos,1),
                    IntInBin(psig,2), IntInBin(pbb,2), IntInBin(pcos,2)};

  double tot = 0;
  for(int i = 0; i < 9; ++i) tot += vals[i];

  double tots[3];
  tots[0] = vals[0]+vals[1]+vals[2];
  tots[1] = vals[3]+vals[4]+vals[5];
  tots[2] = vals[6]+vals[7]+vals[8];


  double accum = 0;

  double tot_accum = 0;
  for(int i = 0; i < 3; ++i){
    const double a0 = tot_accum/tot * 2*TMath::Pi();
    tot_accum += tots[i];
    const double a1 = tot_accum/tot * 2*TMath::Pi();
    const double cx = .5+explode*cos((a0+a1)/2);
    const double cy = .5+explode*sin((a0+a1)/2);

    for(int j = i*3; j < (i+1)*3; ++j){
      const int col = cols[j%3];

      const double a0j = accum/tot * 2*TMath::Pi();
      accum += vals[j];
      const double a1j = accum/tot * 2*TMath::Pi();
      pie_arc(cx, cy, a0j, a1j, R, 3, internal_line_col, col, 0.02);
    }

    const std::string label = sample_labels[i];

    TLatex* ltx = new TLatex(cx+(R+.03)*cos((a0+a1)/2),
                             cy+(R+.03)*sin((a0+a1)/2),
                             label.c_str());
    ltx->SetTextAlign(22);
    double textAng = (a0+a1)/2*180/TMath::Pi()-90;
    if(textAng < -90 || textAng > +90) textAng += 180;
    ltx->SetTextAngle(textAng);

    ltx->Draw();

    // Draw the thick lines
    pie_arc(cx, cy, a0, a1, R, 5, kBlack, -1);
  }

  leg(0, .8, .25, 1);

  std::string appenStr = isFHC ? "_FHC"  : "_RHC";
  gPad->Print(("pie"+appenStr+".png").c_str());
  gPad->Print(("pie"+appenStr+".pdf").c_str());

  // Begin bars

  TRandom3 *rand = new TRandom3(); // Poisson fake data, until we have data
  TCanvas *canBars = new TCanvas();
  Double_t xbins[] = {0,1,4,5,8,9,12,13};
  TH1* htot = new TH1F("", ";;Expected Events", 7, xbins);
  TH1* htot_nofill = new TH1F("", ";;Expected Events", 7, xbins);
  TH1* hdata = new TH1F("", ";;Expected Events", 7, xbins);
  TH1* hcos = new TH1F("", "", 7, xbins);
  TH1* hbkg = new TH1F("", "", 7, xbins);
  for(int i = 0; i < 3; ++i){
    htot->SetBinContent(2*i+2, tots[i]);
    htot_nofill->SetBinContent(2*i+2, tots[i]);
    hcos->SetBinContent(2*i+2, vals[i*3+2]);
    hbkg->SetBinContent(2*i+2, vals[i*3+1] + vals[i*3+2]);
    hdata->SetBinContent(2*i+2, rand->Poisson(tots[i])); // Poisson for now
  }
  htot->SetLineWidth(5);
  htot_nofill->SetLineWidth(5);
  hdata->SetLineWidth(5);
  hdata->SetMarkerStyle(8);

  for(int i = 0; i < 3; ++i){
    htot->GetXaxis()->SetBinLabel(2+2*i, sample_labels[i].c_str());
  }

  htot->GetXaxis()->SetLabelSize(1.5*htot->GetYaxis()->GetTitleSize());
  htot->GetXaxis()->LabelsOption("h");

  htot->GetXaxis()->SetTickLength(0);

  htot->GetYaxis()->CenterTitle();
  double max = 1.40*htot->GetMaximum();
  htot->SetMaximum(max);

  htot->SetFillColor(cols[0]);
  htot->Draw("hist");
  //htot->DrawClone();
  //htot->SetFillStyle(0);

  hbkg->SetLineWidth(3);
  hbkg->SetLineColor(internal_line_col);
  hbkg->SetFillColor(cols[1]);
  hbkg->Draw("same");

  hcos->SetLineWidth(3);
  hcos->SetLineColor(internal_line_col);
  hcos->SetFillColor(cols[2]);
  hcos->Draw("same");
  htot_nofill->Draw("same,hist");
  hdata->Draw("E,P,same");

  double xlo = isFHC ? 0.125 : 0.60;
  TLegend* legBars = new TLegend(xlo, 0.60, xlo+0.25, 0.85);
  legBars->SetFillStyle(0);
  TH1* dummy = new TH1F("", "", 1, 0, 1);
  dummy->SetLineWidth(3);
  for(int i = 0; i < 3; ++i){
    dummy->SetFillColor(cols[i]);
    legBars->AddEntry(dummy->Clone(), component_labels[i].c_str(), "bf");
  }
  legBars->AddEntry(hdata, "Data", "lep");
  legBars->Draw();

  gPad->Print(("bars"+appenStr+".png").c_str());
  gPad->Print(("bars"+appenStr+".pdf").c_str());

  // Begin mosaic

  TCanvas *canMosaic = new TCanvas("c", "d", 1000, 600);

  const double padding = .1;
  const double internal_padding = .025;

  const double active_width = 1-2*padding-2*internal_padding;
  const double active_height = 1-2*padding;

  // Fraction of total number of events in each sample
  double frac[3] = {0,};
  for(int i = 0; i < 3; ++i){
    for(int j = 0; j < 3; ++j){
      frac[i] += vals[i*3+j]/tot_accum;
    }
  }

  accum = 0;
  for(int i = 0; i < 3; ++i){
    const double x0 = padding + active_width*accum;
    accum += frac[i];
    const double x1 = padding + active_width*accum;
    accum += internal_padding;
    TBox* b = new TBox(x0, padding, x1, 1-padding);
    b->SetLineWidth(5);
    b->SetFillStyle(0);

    // Total in this column
    const double tot = vals[i*3] + vals[i*3+1] + vals[i*3+2];
    double accum2 = 0;
    for(int j = 2; j >= 0; --j){
      const double y0 = padding + active_height*accum2;
      accum2 += vals[i*3+j]/tot;
      const double y1 = padding + active_height*accum2;

      TBox* b = new TBox(x0, y0, x1, y1);
      b->SetFillColor(cols[j]);
      b->SetLineColor(internal_line_col);
      b->SetLineWidth(3);
      b->Draw("l");

      if(i == 0){
        // This line is usefull to put the labels vertically
        // on the left of the figure
        //TLatex* ltx = new TLatex(padding*.75, (y0+y1)/2, component_labels[j].c_str());

        double labelsplit = 0.03;
        if (j == 2){
          TLatex* ltx1 =
            new TLatex((x0+x1)/2, (y0+y1)/2, "Cosmic bkg");
          ltx1->SetTextAlign(22);
          ltx1->Draw();
        }
        else if (j == 1){
          TLatex* ltx1 =
            new TLatex((x0+x1)/2, (y0+y1)/2, "Beam bkg");
          ltx1->SetTextAlign(22);
          ltx1->Draw();
        }
        else{
          TLatex* ltx =
            new TLatex((x0+x1)/2, (y0+y1)/2, component_labels[j].c_str());
          ltx->SetTextAlign(22);
          //ltx->SetTextAngle(90);
          //ltx->SetTextColor(text_cols[j]);
          ltx->Draw();
        }
      }
    } // end for j

    TLatex* ltx = new TLatex((x0+x1)/2, padding/2, sample_labels[i].c_str());
    ltx->SetTextAlign(22);
    ltx->Draw();

    // This is the outline we computed above. Draw it after the rest
    b->Draw("l");
  }

  gPad->Print(("mosaic"+appenStr+".png").c_str());
  gPad->Print(("mosaic"+appenStr+".pdf").c_str());
}

double IntInBin	(	TH1 *	h,
		int	bin
	)

Definition at line 68 of file bin_composition_pie_chart.C.

References bin.

{
  int EBinsPerSelBin = 9;
  int idx1 = 1 + EBinsPerSelBin*bin;
  int idx2 = EBinsPerSelBin*(bin+1);
  if (bin <= 2)
    return h->Integral(idx1,idx2);
  return h->GetBinContent(21); // Peripheral is merged to bin 30
}

void leg	(	double	x0,
		double	y0,
		double	x1,
		double	y1
	)

Definition at line 33 of file bin_composition_pie_chart.C.

References cols, component_labels, APDHVSetting::dummy, MECModelEnuComparisons::i, and leg().

{
  TLegend* leg = new TLegend(x0, y0, x1, y1);
  leg->SetFillStyle(0);
  TH1* dummy = new TH1F("", "", 1, 0, 1);
  dummy->SetLineWidth(3);
  for(int i = 0; i < 3; ++i){
    dummy->SetFillColor(cols[i]);
    leg->AddEntry(dummy->Clone(), component_labels[i].c_str(), "bf");
  }
  leg->Draw();
}

void pie_arc	(	double	cx,
		double	cy,
		double	a0,
		double	a1,
		double	r,
		int	width,
		int	lineCol,
		int	fillCol,
		double	fudgeR = 0
	)

Definition at line 46 of file bin_composition_pie_chart.C.

References a, std::cos(), MECModelEnuComparisons::i, submit_hadd::l, and std::sin().

{
  TPolyLine* l = new TPolyLine();
  l->SetLineWidth(width);
  l->SetLineColor(lineCol);

  l->SetPoint(0, cx+fudgeR*r*cos(a0), cy+fudgeR*r*sin(a0));
  for(int i = 0; i <= 100; ++i){
    const double a = a0 + (a1-a0)*i/100.;
    l->SetPoint(i+1, cx+r*cos(a), cy+r*sin(a));
  }
  l->SetPoint(102, cx+fudgeR*r*cos(a1), cy+fudgeR*r*sin(a1));
  l->SetPoint(103, cx+fudgeR*r*cos(a0), cy+fudgeR*r*sin(a0));

  l->SetFillColor(fillCol);
  l->SetFillStyle(1001);

  if(fillCol >= 0) l->Draw("f");
  l->Draw("l");
}

TH1* PredictCosmic	(	std::string	fname,
		std::string	name
	)

Definition at line 78 of file bin_composition_pie_chart.C.

References std::cos(), plot_validation_datamc::fname, ana::kAna2017Livetime, ana::kLivetime, runNovaSAM::release, runNovaSAM::ret, and ana::Spectrum::ToTH1().

{
  Spectrum *cos = LoadFromFile<Spectrum>(fname,name).release();
  TH1 *ret = cos->ToTH1(kAna2017Livetime,kLivetime);
  return ret;
}

Variable Documentation

const int cols[3] = {kRed-7, kBlue-7, kGray}

Definition at line 26 of file bin_composition_pie_chart.C.

const std::string component_labels[3] = {"Signal", "Beam bkg", "Cosmic bkg"}

Definition at line 30 of file bin_composition_pie_chart.C.

const double explode = .05

Definition at line 23 of file bin_composition_pie_chart.C.

const int internal_line_col = kGray+2

Definition at line 28 of file bin_composition_pie_chart.C.

const double R = .35

Definition at line 24 of file bin_composition_pie_chart.C.

const std::string sample_labels[3] = {"Low CVN", "High CVN", "Peripheral"}

Definition at line 31 of file bin_composition_pie_chart.C.

const int text_cols[3] = {kRed, kBlue, kGray+2}

Definition at line 27 of file bin_composition_pie_chart.C.