plotLogicalSpectra.C

Go to the documentation of this file.

// takes the output of a covariancematrixmaker job
// can be configured to plot pm 1 sigma variations
//
// 2020-09-10

enum type{
  kNominal,
  kPlus1Sigma,
  kMinus1Sigma
};

std::vector<double> getNormFactors(TH1D* h, double cutoff){
 
  std::vector<double> norms;
  for (int i = 1; i < h->GetNbinsX()+1; ++i){
    if (h->GetBinContent(i) > cutoff)
      norms.push_back(0.0005);
    else
      norms.push_back(1.0);
  }
  return norms;

};

void normaliseBins(TH1D* h, std::vector<double>& norms){
  for (int i = 0; i < h->GetNbinsX(); ++i){
    h->SetBinContent(i+1, h->GetBinContent(i+1)*norms.at(i));
  }
};


void styleHistogram(TH1D* h, type t){
  h->SetLineWidth(2);

  if (t == kPlus1Sigma)
    h->SetLineColor(kGreen+1);
  if (t == kMinus1Sigma)
    h->SetLineColor(kAzure+1);

  h->GetXaxis()->CenterTitle();
  h->GetYaxis()->CenterTitle();

  h->GetYaxis()->SetTitleOffset(0.6);

  h->GetXaxis()->SetTitle("Logical Bin Number");
  h->GetYaxis()->SetTitle("Normalised Event Count");

  h->GetYaxis()->SetRangeUser(0, h->GetBinContent(h->GetMaximumBin())*1.25);


};

void plotLogicalSpectra(bool pm1sigma = false){

  TH1D* logicalSpectra = (TH1D*)_file0->Get("covar/NominalBin");

  TCanvas* c1 = new TCanvas("c1", "c1", 2000, 500);
  c1->SetGridy();
  c1->SetLeftMargin(0.08);
  c1->SetRightMargin(0.05);

  std::vector<double> norms = getNormFactors(logicalSpectra, 100);

  normaliseBins(logicalSpectra, norms);
  styleHistogram(logicalSpectra, kNominal);

  logicalSpectra->Draw("hist");

  if (pm1sigma){

    TH1D* plus1SigmaSpectra  = (TH1D*)_file0->Get("covar/Plus1SigmaSpectrum");
    TH1D* minus1SigmaSpectra = (TH1D*)_file0->Get("covar/Minus1SigmaSpectrum");

    normaliseBins(plus1SigmaSpectra, norms);
    styleHistogram(plus1SigmaSpectra, kPlus1Sigma);

    normaliseBins(minus1SigmaSpectra, norms);
    styleHistogram(minus1SigmaSpectra, kMinus1Sigma);

    plus1SigmaSpectra ->Draw("same hist");
    minus1SigmaSpectra->Draw("same hist");
    logicalSpectra    ->Draw("same hist");

    TLegend* leg = new TLegend(0.75, 0.70, 0.85, 0.85);
    leg->AddEntry(logicalSpectra    , "Nominal Spectra", "l");
    leg->AddEntry(plus1SigmaSpectra , "+1 #sigma Spectra", "l");
    leg->AddEntry(minus1SigmaSpectra, "-1 #sigma Spectra", "l");
    leg->SetLineWidth(0);
    leg->SetFillStyle(0);
    leg->Draw("same");

  }

  c1->SaveAs("logicalBinSpectra.pdf");

}