DQMHistAnalysisOutputImages.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.                  *
 **************************************************************************/
//+
// File : DQMHistAnalysisOutputImages.cc
// Description : DQM Output, send Canvases to jsroot server.
//-
 
 
#include <dqm/analysis/modules/DQMHistAnalysisOutputImages.h>
#include <TROOT.h>
#include <TClass.h>
#include <TObject.h>
#include <TCanvas.h>
#include <ctime>
#include <filesystem>
 
using namespace std;
using namespace Belle2;
 
//-----------------------------------------------------------------
//                 Register the Module
//-----------------------------------------------------------------
REG_MODULE(DQMHistAnalysisOutputImages);
 
//-----------------------------------------------------------------
//                 Implementation
//-----------------------------------------------------------------
 
DQMHistAnalysisOutputImagesModule::DQMHistAnalysisOutputImagesModule()
  : DQMHistAnalysisModule()
{
  setDescription("Module to produce output images during 'events' for dqm steering scripts.");
  //Parameter definition
  addParam("CanvasSaveDefault", m_canvasSaveDefault, "Save untagged canvases", false);
  addParam("OutputPath", m_outputPath, "Set output path", std::string(""));
  addParam("asPNG", m_asPNG, "Save as PNG files", true);
  addParam("asJPEG", m_asJPEG, "Save as JPEG files", false);
  addParam("asPDF", m_asPDF, "Save as PDF files", false);
  addParam("asJSON", m_asJSON, "Save as JSON files", false);
  addParam("asROOT", m_asROOT, "Save as ROOT files", false);
  addParam("useExpRun", m_useExpRun, "crate exp/run subdirs", false);
  B2DEBUG(20, "DQMHistAnalysisOutputImages: Constructor done.");
}
 
void DQMHistAnalysisOutputImagesModule::initialize()
{
  m_evtMetaDataPtr.isRequired();
}
 
void DQMHistAnalysisOutputImagesModule::event()
{
  B2DEBUG(20, "DQMHistAnalysisOutputImages: event called.");
  if (!m_evtMetaDataPtr.isValid()) {
    B2ERROR("No valid EventMetaData.");
    return;
  }
 
  TSeqCollection* seq = gROOT->GetListOfCanvases();
  TIter nextkey(seq);
  TObject* obj = 0;
 
  std::filesystem::path outpath = m_outputPath;
  if (m_useExpRun) outpath = outpath / std::to_string(m_evtMetaDataPtr->getExperiment()) / std::to_string(
                                 m_evtMetaDataPtr->getRun()); // add exp/run
 
  auto& clist = getCanvasUpdatedList();
  int saved_canvases = 0;
 
  while ((obj = dynamic_cast<TObject*>(nextkey()))) {
    if (auto c = dynamic_cast<TCanvas*>(obj); c != nullptr) {
      auto process_canvas = m_canvasSaveDefault;
 
      auto it = clist.find(c->GetName());
      if (it != clist.end()) {
        process_canvas = it->second;
      }
      if (!process_canvas) continue;
      saved_canvases++;
 
      std::string cname = c->GetName();
      create_directories((outpath / cname).parent_path()); // create recursive path, even if it exist
      if (m_asPNG) c->Print((outpath / (cname + ".png")).c_str());
      if (m_asJPEG) c->Print((outpath / (cname + ".jpg")).c_str());
      if (m_asPDF) c->Print((outpath / (cname + ".pdf")).c_str());
      if (m_asROOT) c->Print((outpath / (cname + ".root")).c_str());
      if (m_asJSON) c->Print((outpath / (cname + ".json")).c_str());
    }
  }
  B2INFO("Saved " << saved_canvases << " of " << seq->GetEntries() << " objects.");
}