RootAsciiDumperAlg.cxx

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
// RootAsciiDumperAlg.cxx 
// Implementation file for class RootAsciiDumperAlg
// Author: S.Binet<binet@cern.ch>
 
// AthenaRootComps includes
#include "RootAsciiDumperAlg.h"
 
// STL includes
#include <sstream>
#include <stdio.h>
#include <stdexcept>
 
#include <inttypes.h>
 
// linux i/o includes
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <format>
 
// FrameWork includes
#include "Gaudi/Property.h"
#include "AthContainers/ConstAccessor.h"
 
// SGTools
#include "SGTools/BuiltinsClids.h"  // to put ints,... in evtstore
#include "SGTools/StlVectorClids.h" // to put std::vectors... in evtstore
 
namespace Athena {

// Public methods: 
 
// Athena Algorithm's Hooks
StatusCode RootAsciiDumperAlg::initialize()
{
  ATH_MSG_INFO ("Initializing " << name() << "...");
  ATH_CHECK( m_eiKey.initialize() );
  ATH_CHECK( m_el_jetcone_dr.initialize() );
  ATH_CHECK( m_el_n.initialize() );
  ATH_CHECK( m_el_eta.initialize() );
  ATH_CHECK( m_evtnbr.initialize() );
  ATH_CHECK( m_runnbr.initialize() );
 
  ATH_MSG_INFO("dumping data into file ["
               << m_ofname << "]...");
  if (m_ofname.empty()) {
    ATH_MSG_ERROR("cannot dump data into an empty file name!");
    return StatusCode::FAILURE;
  }
  m_ofd = open(m_ofname.value().c_str(), 
               O_WRONLY | O_CREAT | O_TRUNC,
               S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
 
  if (m_ofd < 0) {
    ATH_MSG_ERROR("problem opening file [" << m_ofname << "] with "
                  "write permissions.");
    return StatusCode::FAILURE;
  }
 
  return StatusCode::SUCCESS;
}
 
StatusCode RootAsciiDumperAlg::finalize()
{
  ATH_MSG_INFO ("Finalizing " << name() << "...");
 
  if (m_ofd > 0) {
    fflush(NULL);
    if (close(m_ofd) < 0) {
      ATH_MSG_WARNING("problem while closing [" << m_ofname << "]");
    }
  }
 
  return StatusCode::SUCCESS;
}
 
StatusCode RootAsciiDumperAlg::execute(const EventContext& ctx)
{  
  ATH_MSG_DEBUG ("Executing " << name() << "...");
 
 
  uint64_t nevts = m_nentries;
  m_nentries += 1;
 
  SG::ReadHandle<uint32_t> runnbr (m_runnbr, ctx);
  SG::ReadHandle<uint32_t> evtnbr (m_evtnbr, ctx);
 
  SG::ReadHandle<int32_t> el_n (m_el_n, ctx);
 
  SG::ReadHandle<xAOD::EventInfo> ei (m_eiKey, ctx);
  static const SG::ConstAccessor<std::string> tupleName ("tupleName");
  static const SG::ConstAccessor<std::string> collectionName ("collectionName");
  std::string collName = collectionName(*ei);
  std::string::size_type pos = collName.rfind ("/");
  if (pos != std::string::npos) {
    collName.erase (0, pos+1);
  }
 
  std::string buf = std::format(
        "{:03}.{} = {}\n"
        "{:03}.{} = {}\n"
        "{:03}.{} = {}\n"
        "{:03}.{} = {}\n"
        "{:03}.{} = {}\n",
        nevts, "collectionName", collName,
        nevts, "tupleName", tupleName(*ei),
        nevts, "RunNumber", *runnbr,
        nevts, "EventNumber", *evtnbr,
        nevts, "el_n", *el_n
  );
 
    if (buf.empty()) {
        throw std::runtime_error("Empty buffer in RootAsciiDumperAlg::execute");
    }
 
    if (write(m_ofd, buf.data(), buf.size()) == -1) {
        throw std::runtime_error("Failed to write buffer to file descriptor");
    }
  
 
  if (*el_n > 0) {
    SG::ReadHandle<std::vector<float> > el_eta (m_el_eta, ctx);
    SG::ReadHandle<std::vector<std::vector<float> > > el_jetcone_dr (m_el_jetcone_dr, ctx);
    
    std::ostringstream bufv;
    for (int32_t ii = 0; ii < *el_n; ++ii) {
        bufv << (*el_eta)[ii];
        if (ii != (*el_n) - 1) {
            bufv << ", ";
        }
    }
 
    buf = std::format(
        "{:03}.{} = [{}]\n",
        nevts,
        "el_eta",
        bufv.str()
    );
 
    if (buf.empty()) {
        throw std::runtime_error("Empty buffer in RootAsciiDumperAlg::execute");
    }
 
    if (write(m_ofd, buf.data(), buf.size()) == -1) {
        throw std::runtime_error("Failed to write buffer to file descriptor");
    }
    bufv.str("");//clear
    for (int32_t ii = 0; ii < *el_n; ++ii) {
        bufv << "[";
        for (std::size_t jj = 0; jj < (*el_jetcone_dr)[ii].size(); ++jj) {
            bufv << (*el_jetcone_dr)[ii][jj];
            if (jj + 1 < (*el_jetcone_dr)[ii].size()) {
                bufv << ", ";
            }
        }
        bufv << "]";
        if (ii + 1 < *el_n) {
            bufv << ", ";
        }
    }
 
    buf = std::format(
        "{:03}.{} = [{}]\n",
        nevts,
        "el_jetcone_dr",
        bufv.str()
    );
 
    if (write(m_ofd, buf.data(), buf.size()) == -1) {
        throw std::runtime_error("Failed to write buffer to file descriptor");
    }
  }
 
  return StatusCode::SUCCESS;
}
 
} //> end namespace Athena