CbmQaCheckerObjectHandler.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 "CbmQaCheckerObjectHandler.h"
 
#include "CbmQaCheckerResult.h"
#include "Logger.h"
#include "TCanvas.h"
#include "TDirectory.h"
#include "TNamed.h"
 
using cbm::qa::checker::ECmpInference;
using cbm::qa::checker::ObjectHandler;
using cbm::qa::checker::Result;
 
// ---------------------------------------------------------------------------------------------------------------------
//
ObjectHandler::ObjectHandler(int iObject, int iFile, int iDataset, const char* objType)
  : fsObjType(objType)
  , fObjectID(iObject)
  , fFileID(iFile)
  , fDatasetID(iDataset)
{
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
ObjectHandler::~ObjectHandler()
{
  for (auto* pObj : fvpObjects) {
    if (pObj) {
      delete pObj;
      pObj = nullptr;
    }
  }
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
void ObjectHandler::AddObjects(const std::vector<TNamed*>& vpObj)
{
  // ----- Check input
  LOG_IF(fatal, !fpObjDB) << "ObjectHandler: object database was not defined";
  LOG_IF(fatal, !fpOutDir) << "ObjectHandler: output directory was not defined";
  LOG_IF(fatal, (int) vpObj.size() != fpObjDB->GetNofVersions())
    << "ObjectHandler: Attempt to add vector with object pointers of different to the one of version labels";
 
  TDirectory* pCurrDir = gDirectory;
  gDirectory           = fpOutDir;
  for (int iVer = 0; iVer < fpObjDB->GetNofVersions(); ++iVer) {
    if (!fsBaseName.size()) {
      fsBaseName = vpObj[iVer]->GetName();
    }
    else {
      LOG_IF(fatal, strcmp(fsBaseName.c_str(), vpObj[iVer]->GetName()))
        << "Hist1DHandler: attempt to add object of different name " << fsBaseName << " vs. " << vpObj[iVer]->GetName();
    }
    const char* cloneName = Form("%s_orig_%s", fsBaseName.data(), fpObjDB->GetVersionLabel(iVer).data());
    fvpObjects.push_back((TNamed*) vpObj[iVer]->Clone(cloneName));
  }
  gDirectory = pCurrDir;
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
std::vector<Result> ObjectHandler::CompareWithDefault()
{
  // All object versions are equal to default one (H0)
  std::vector<Result> res(fpObjDB->GetNofVersions(), Result(ECmpInference::StronglyEqual, true, 1., 1., 1.));
  int iDef = fpObjDB->GetDefaultID();
  for (int iV = 0; iV < (int) fvpObjects.size(); ++iV) {
    if (iV == iDef) {
      continue;
    }
    res[iV] = this->Compare(iV);
  }
  return res;
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
void ObjectHandler::SetOutputDirectory(TDirectory* pDir)
{
  LOG_IF(fatal, !pDir) << "ObjectHandler: attempt to pass nullptr as a folder";
  fpOutDir = pDir;
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
void ObjectHandler::Write()
{
  fpOutDir->cd();
  for (auto pObj : fvpObjects) {
    pObj->Write();
  }
 
  auto WriteCanvas = [](auto canvas) -> void {
    if (canvas.get()) canvas->Write();
  };
 
  WriteCanvas(fpCanvasCmp);
  WriteCanvas(fpCanvasRat);
  WriteCanvas(fpCanvasDiff);
}