CbmQaCheckerProfile1DHandler.cxx

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer] */

 
#include "CbmQaCheckerProfile1DHandler.h"
 
#include "CbmQaCheckerHist1DHandler.h"
#include "CbmQaCheckerTypedefs.h"
#include "Logger.h"
#include "TCanvas.h"
#include "TError.h"
#include "TFolder.h"
#include "TGraphAsymmErrors.h"
#include "TLegend.h"
#include "TMath.h"
#include "TMultiGraph.h"
#include "TObject.h"
#include "TProfile.h"
#include "TRatioPlot.h"
#include "TStyle.h"
 
#include <limits>
 
using cbm::qa::checker::Profile1DHandler;
 
// ---------------------------------------------------------------------------------------------------------------------
//
Profile1DHandler::Profile1DHandler(int iObject, int iFile, int iDataset) : Hist1DHandler(iObject, iFile, iDataset) {}
 
// ---------------------------------------------------------------------------------------------------------------------
//
void Profile1DHandler::CreateCanvases(Option_t* opt)
{
  int nVersions = fpObjDB->GetNofVersions();
  int iDef      = fpObjDB->GetDefaultID();
 
  // ----- Option parsing
  std::string sOption = opt;
  for (auto& ch : sOption) {
    ch = std::tolower(ch);
  }
  bool bDrawRatio = sOption.find("r") != std::string::npos;
  bool bDrawDiff  = sOption.find("d") != std::string::npos;
 
  // Axis titles
  TString xAxisTitle = static_cast<TProfile*>(fvpObjects[0])->GetXaxis()->GetTitle();
  TString yAxisTitle = static_cast<TProfile*>(fvpObjects[0])->GetYaxis()->GetTitle();
 
  // Original histograms
  {
    TString sCanvN = fsBaseName + "_cmp_canvas";
    TString sCanvT = TString("Comparison result for \"") + fsBaseName + "\"";
    fpCanvasCmp    = std::make_shared<TCanvas>(sCanvN, sCanvT, 500, 500);
    fpCanvasCmp->SetMargin(0.20, 0.05, 0.20, 0.10);
    fpCanvasCmp->cd();
    auto* pMultiGraphOrig = new TMultiGraph(fsBaseName.c_str(), fvpObjects[0]->GetTitle());
    for (int iV = 0; iV < nVersions; ++iV) {
      auto* pCopy = new TGraphAsymmErrors((TH1*) fvpObjects[iV]);
      pCopy->SetMarkerStyle(20);
      pCopy->SetTitle(fpObjDB->GetVersionLabel(iV).c_str());
      pMultiGraphOrig->Add(pCopy, "P");
    }
    pMultiGraphOrig->GetXaxis()->SetTitle(xAxisTitle);
    pMultiGraphOrig->GetYaxis()->SetTitle(yAxisTitle);
    pMultiGraphOrig->Draw("A pmc plc");
    fpCanvasCmp->BuildLegend();
  }
 
  auto* pDefault = static_cast<TProfile*>(fvpObjects[iDef])->ProjectionX();
 
  // Histogram ratios to default
  if (bDrawRatio) {
    TString sCanvN = fsBaseName + "_ratio_canvas";
    TString sCanvT = TString("Comparison result for \"") + fsBaseName + "\" (ratio)";
    fpCanvasRat    = std::make_shared<TCanvas>(sCanvN, sCanvT, 500, 500);
    fpCanvasRat->SetMargin(0.20, 0.05, 0.20, 0.10);
    fpCanvasRat->cd();
    TString titleRatio     = Form("Ratio to %s", fpObjDB->GetVersionLabel(iDef).c_str());
    auto* pMultiGraphRatio = new TMultiGraph((fsBaseName + "_ratio").c_str(), titleRatio);
    for (int iV = 0; iV < nVersions; ++iV) {
      if (iV == iDef) {
        continue;
      }
      auto* pRatio        = static_cast<TProfile*>(fvpObjects[iV])->ProjectionX();
      auto currErrorLevel = gErrorIgnoreLevel;
      gErrorIgnoreLevel   = kError;
      auto* pCopy         = new TGraphAsymmErrors(pRatio, pDefault, "pois");
      gErrorIgnoreLevel   = currErrorLevel;
      pCopy->GetYaxis()->SetTitle("ratio");
      pCopy->SetTitle(fpObjDB->GetVersionLabel(iV).c_str());
      pCopy->SetMarkerStyle(20);
      pMultiGraphRatio->Add(pCopy, "P");
      if (pRatio) {
        delete pRatio;
        pRatio = nullptr;
      }
    }
    pMultiGraphRatio->GetXaxis()->SetTitle(xAxisTitle);
    pMultiGraphRatio->GetYaxis()->SetTitle("ratio");
    pMultiGraphRatio->Draw("A pmc plc");
    fpCanvasRat->BuildLegend();
  }
 
  // Histogram ratios to default
  if (bDrawDiff) {
    TString sCanvN = fsBaseName + "_diff_canvas";
    TString sCanvT = TString("Comparison result for \"") + fsBaseName + "\" (ratio)";
    fpCanvasDiff   = std::make_shared<TCanvas>(sCanvN, sCanvT, 500, 500);
    fpCanvasDiff->SetMargin(0.20, 0.05, 0.20, 0.10);
    fpCanvasDiff->cd();
    TString titleDiff     = Form("Difference from %s", fpObjDB->GetVersionLabel(iDef).c_str());
    auto* pMultiGraphDiff = new TMultiGraph((fsBaseName + "_diff").c_str(), titleDiff);
    for (int iV = 0; iV < nVersions; ++iV) {
      if (iV == iDef) {
        continue;
      }
      auto* pDiff = static_cast<TProfile*>(fvpObjects[iV])->ProjectionX();
      pDiff->Add(pDefault, -1.);
      auto* pCopy = new TGraphAsymmErrors(pDiff);
      pCopy->GetYaxis()->SetTitle("difference");
      pCopy->SetTitle(fpObjDB->GetVersionLabel(iV).c_str());
      pCopy->SetMarkerStyle(20);
      pMultiGraphDiff->Add(pCopy, "P");
      if (pDiff) {
        delete pDiff;
        pDiff = nullptr;
      }
    }
    pMultiGraphDiff->GetXaxis()->SetTitle(xAxisTitle);
    pMultiGraphDiff->GetYaxis()->SetTitle("difference");
    pMultiGraphDiff->Draw("A pmc plc");
    fpCanvasDiff->BuildLegend();
  }
 
  if (pDefault) {
    delete pDefault;
    pDefault = nullptr;
  }
}