DQMHistReferenceModule.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#include <dqm/analysis/modules/DQMHistReferenceModule.h>
#include <TROOT.h>
#include <TKey.h>
 
using namespace std;
using namespace Belle2;
 
//-----------------------------------------------------------------
//                 Register the Module
//-----------------------------------------------------------------
REG_MODULE(DQMHistReference);
 
//-----------------------------------------------------------------
//                 Implementation
//-----------------------------------------------------------------
 
DQMHistReferenceModule::DQMHistReferenceModule() : DQMHistAnalysisModule()
{
  //Parameter definition
  addParam("ReferenceFile", m_referenceFileName, "Name of the reference histrogram files", string(""));
  B2DEBUG(1, "DQMHistReference: Constructor done.");
}
 
void DQMHistReferenceModule::beginRun()
{
  B2DEBUG(1, "DQMHistReference: beginRun called.");
 
  loadReferenceHistos();
}
 
void DQMHistReferenceModule::loadReferenceHistos()
{
  TH1::AddDirectory(false); // do not store any histograms
 
  B2DEBUG(1, "DQMHistReference: reading references from input root file");
 
  string run_type = getRunType();
  if (run_type == "") run_type = "default";
 
  B2INFO("DQMHistReference: run_type " << run_type);
 
  TFile* refFile = new TFile(m_referenceFileName.c_str(), "READ");
 
  if (refFile->IsZombie()) {
    B2INFO("DQMHistReference: reference file " << m_referenceFileName << " does not exist. No references will be used!");
    refFile->Close();
    delete refFile;
    return;
  }
 
  B2INFO("DQMHistReference: use reference file " << m_referenceFileName);
 
  TIter nextRefDirKey(refFile->GetListOfKeys());
  TKey* refDirKey;
  while ((refDirKey = dynamic_cast<TKey*>(nextRefDirKey()))) {
    if (refDirKey->IsFolder() && string(refDirKey->GetName()) == string("ref")) {
      TDirectory* refDir = static_cast<TDirectory*>(refDirKey->ReadObj()); // ReadObj -> I own it, delete later
      TIter nextDetDirKey(refDir->GetListOfKeys());
      TKey* detDirKey;
      // detector folders
      while ((detDirKey = dynamic_cast<TKey*>(nextDetDirKey()))) {
        if (!detDirKey->IsFolder()) continue;
        TDirectory* detDir = static_cast<TDirectory*>(detDirKey->ReadObj());
        TIter nextRunTypeDirKey(detDir->GetListOfKeys()); // ReadObj -> Now I own this, so delete later
        TKey* runtypeDirKey;
        TDirectory* runtypeDir = nullptr;
        // run type folders (get the run type corresponding folder or use default one)
        while ((runtypeDirKey = dynamic_cast<TKey*>(nextRunTypeDirKey()))) {
          if (!runtypeDirKey->IsFolder()) continue;
          if (string(runtypeDirKey->GetName()) == run_type) {
            if (runtypeDir) delete runtypeDir; // if default was loaded before
            runtypeDir = static_cast<TDirectory*>(runtypeDirKey->ReadObj()); // ReadObj -> I own it, delete later
            break; // break directly, otherwise "default" could overwrite it
          }
          // else we would check if default, which we load as backup
          // ReadObj -> I own it
          if (string(runtypeDirKey->GetName()) == "default") runtypeDir = static_cast<TDirectory*>(runtypeDirKey->ReadObj());
        }
        string detName = detDir->GetName();
        // Attention, runtypeDir and runtypeDirKey could be zero here
        if (!runtypeDir) {
          B2INFO("No run type specific or default references available for " << detName);
        } else {
          B2INFO("Reading reference histograms for " << detName << " from run type folder: " << runtypeDir->GetName());
 
          TIter nextHistkey(runtypeDir->GetListOfKeys());
          TKey* histKey;
          // now read histograms
          while ((histKey = dynamic_cast<TKey*>(nextHistkey()))) {
            if (histKey->IsFolder()) continue;
            if (gROOT->GetClass(histKey->GetClassName())->InheritsFrom("TH1")) { // maybe not needed with the dynamic cast check below
              auto h = dynamic_cast<TH1*>(histKey->ReadObj());
              if (h) addRefHist(detName, h); // ReadObj -> I own it, tranfer ownership to function
            }
          }
          delete runtypeDir; // always non-zero as checked above ... runtype or "default"
        }
        delete detDir; // always non-zero ... detector subdir name
      }
      delete refDir; // always non-zero ... "ref" folder
    }
  }
 
  B2INFO("DQMHistReference: read references done");
  refFile->Close();
  delete refFile;
}