timeResSplitCases.cxx

Go to the documentation of this file.

 
#include "timeResSplitCases.h"
 
#include "CxxUtils/checker_macros.h"
ATLAS_NO_CHECK_FILE_THREAD_SAFETY;  // single-threaded plotting application
 
#include "TCanvas.h"
#include "TEnv.h"
#include "THStack.h"
#include "TLegend.h"
#include "TPad.h"
 
#include <iostream>
 
void timeResSplitCases(TString config_file_name = "") {
  if (config_file_name != "") {
    g_config_file_name = config_file_name;
  }
 
  TEnv env(g_config_file_name);
 
  g_input_file_path = env.GetValue("input_file_path_TTI", "default");
  g_output_plot_path = env.GetValue("output_plot_path", "default");
  dataset_name = env.GetValue("dataset_name", "default");
  g_dataset_description = env.GetValue("dataset_description", "My Sample");
  print_png = env.GetValue("print_png", false);
  do_zoom = env.GetValue("do_zoom", false);
  g_use_log_scale = env.GetValue("use_log_scale", false);
  work_status = env.GetValue("work_status", "default");
  g_time_acc_tool = env.GetValue(
      "time_acc_tool",
      "HGTD_TrkTimePerformanceStudies.ExpertTrackTimeFromClusters/");
  g_track_selection_tool = env.GetValue(
      "track_selection_tool",
      "HGTD_TrkTimePerformanceStudies.AllTracksSelection/");
 
  TFile* file = TFile::Open(g_input_file_path, "READ");
  if (file == nullptr || file->IsZombie()) {
    std::cout << "could not open input file " << g_input_file_path << std::endl;
    return;
  }
 
  SetAtlasStyle();
 
  plot(file);
 
}
 
void plot(TFile* file) {
 
  std::vector<Color_t> colors = {
      kTeal + 4, kTeal + 2, kTeal, kTeal - 9, kMagenta, kRed, kRed, kRed, kRed};
 
  std::vector<TH1F *> hists;
  std::cout << g_track_selection_tool + g_time_acc_tool << std::endl;
 
  for (const TString& primes_fraction : primes_fractions) {
    TString plotname = "m_hist_timeres_outlier_cases" + primes_fraction;
    auto hist = (TH1F*)file->Get(g_track_selection_tool + g_time_acc_tool +
                                 plotname);
    if (hist == nullptr) {
      std::cout << "could not find " << plotname << std::endl;
      return;
    }
    hists.push_back(hist);
  }
 
  auto overall_hist =
      new TH1F("overall_hist", ";t_{reco} - t_{truth} [ns]; number of tracks",
               200, -.4, 0.4);
  for (const auto &h : hists) {
    overall_hist->Add(h);
  }
 
  THStack *stack_hist =
      new THStack("stack_hist", ";t_{reco} - t_{truth} [ns]; Number of Tracks");
 
  for (size_t i = 0; i < hists.size(); i++) {
    hists.at(i)->SetFillColor(colors.at(i));
    hists.at(i)->SetLineWidth(0.);
    stack_hist->Add(hists.at(i));
  }
 
  stack_hist->SetMaximum(stack_hist->GetMaximum() * 1.7);
 
  if (do_zoom) {
    stack_hist->SetMaximum(100);
  }
 
  TCanvas *c = new TCanvas();
  c->SetTicks(1, 1);
  if (g_use_log_scale) {
    c->SetLogy();
  }
 
  auto hist = new TH1F(
      "axishist", ";t_{reco} - t_{truth} [ns]; Number of Tracks", 10, -0.4, 0.4);
 
  if (g_use_log_scale) {
    hist->SetAxisRange(1, 1.e8, "Y");
  } else {
    hist->SetAxisRange(0, 0.08, "Y");
  }
 
  hist->Draw();
 
  stack_hist->Draw("hist same");
 
  overall_hist->SetMarkerColor(kBlack);
  overall_hist->SetLineWidth(0);
  overall_hist->SetMarkerStyle(20);
  overall_hist->SetMarkerSize(0);
 
  c->Update();
 
  gPad->RedrawAxis();
 
 
  Color_t text_color = kBlack;
  atlas::ATLAS_LABEL(0.19, 0.88, text_color);
  atlas::myText(0.31, 0.88, text_color, work_status);
 
  float label_text_size = 0.05;
 
  atlas::myText(0.19, 0.83, text_color, g_dataset_description.Data(), label_text_size);
  atlas::myText(0.19, 0.77, text_color, "Timing scenario \"Initial\"",
                label_text_size);
 
 
  auto legend = new TLegend(0.62, 0.65, 0.9, 0.92);
 
  legend->SetTextFont(42);
  legend->SetBorderSize(0);
  legend->SetTextSize(0.035);
 
  for (size_t i = 0; i < labels.size(); i++) {
    legend->AddEntry(hists.at(i), labels.at(i), "f");
  }
 
  legend->Draw("same");
 
  TString plot_name =
      Form("%s/%s_%s.pdf", g_output_plot_path.Data(),
           dataset_name.ReplaceAll(" ", "").ReplaceAll(",", "_").Data(),
           descr.Data());
  if (g_use_log_scale) {
    plot_name.ReplaceAll(".pdf", "LogScale.pdf");
  }
  c->Print(plot_name);
  if (print_png)
    c->Print(plot_name.ReplaceAll(".pdf", ".png"));
 
  plot_name.ReplaceAll(".png", ".C");
  c->Print(plot_name);
 
}
 
int main(int argc, char* argv[]) {
  if (argc != 2) {
    std::cout << "Syntax: " << argv[0] << " <config file>" << std::endl;
    return 1;
  }
  timeResSplitCases(argv[1]);
  return 0;
}