RootFileHandler.cpp

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2026 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#include "RootFileHandler.h"
#include <GaudiKernel/IFileMgr.h>
#include <GaudiKernel/MsgStream.h>
#include <TFile.h>
#include <TROOT.h>
 
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 
RootFileHandler::RootFileHandler( IMessageSvc* msg ) : m_log( msg, "RootFileHandler" ) {
  // Protect against multiple instances of TROOT
  if ( !gROOT ) { static TROOT root( "root", "ROOT I/O" ); }
  m_level = msg->outputLevel( "RootFileHandler" );
}
 
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 
Io::open_t RootFileHandler::openRootFile( const std::string& n, const Io::IoFlags& f, const std::string& desc,
                                          Io::Fd& fd, void*& ptr ) {
 
  m_log.setLevel( m_level );
 
  if ( m_log.level() <= MSG::DEBUG )
    m_log << MSG::DEBUG << "openRootFile(\"" << n << "\"," << f << "," << desc << ")" << endmsg;
 
  ptr = nullptr;
  fd  = -1;
 
  std::string opt;
 
  if ( f == Io::READ ) {
    opt = "READ";
  } else if ( f == ( Io::WRITE | Io::CREATE | Io::EXCL ) ) {
    opt = "NEW";
  } else if ( f == ( Io::WRITE | Io::CREATE ) ) {
    opt = "RECREATE";
  } else if ( f | Io::APPEND ) {
    opt = "UPDATE";
  } else {
    m_log << MSG::ERROR << "Don't know how to handle IoFlag " << f << endmsg;
    return 1;
  }
 
  std::unique_ptr<TFile> tf;
 
  try {
    tf.reset( TFile::Open( n.c_str(), opt.c_str() ) );
  } catch ( const std::exception& Exception ) {
    m_log << MSG::ERROR << "exception caught while trying to open root"
          << " file for reading: " << Exception.what() << std::endl
          << "  -> file probably corrupt." << endmsg;
    return 1;
  } catch ( ... ) {
    m_log << MSG::ERROR << "Problems opening input file  \"" << n << "\": probably corrupt" << endmsg;
    return 1;
  }
 
  if ( !tf || !tf->IsOpen() ) {
    m_log << MSG::ERROR << "Unable to open ROOT file \"" << n << "\" with options \"" << opt << "\"" << endmsg;
 
    tf.reset();
    return 1;
  }
 
  fd = tf->GetFd();
 
  ptr = tf.release();
 
  if ( m_log.level() <= MSG::DEBUG ) m_log << MSG::DEBUG << "opened TFile " << ptr << " Fd: " << fd << endmsg;
 
  return 0;
}
 
//* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *//
 
Io::close_t RootFileHandler::closeRootFile( void* ptr ) {
 
  if ( m_log.level() <= MSG::DEBUG ) m_log << MSG::DEBUG << "closeRootFile(ptr:" << ptr << ")" << endmsg;
 
  if ( !ptr ) {
    m_log << MSG::ERROR << "Unable to close file: ptr == 0" << endmsg;
    return -1;
  }
 
  TFile* tf = static_cast<TFile*>( ptr );
 
  try {
    tf->Close();
  } catch ( const std::exception& Exception ) {
    m_log << MSG::ERROR << "exception caught while trying to close root"
          << " file" << Exception.what() << endmsg;
    return -1;
  } catch ( ... ) {
    m_log << MSG::ERROR << "Problems closing ROOT file  \"" << tf->GetName() << "\"" << endmsg;
    return -1;
  }
 
  return 0;
}
 
//* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *//
 
Io::reopen_t RootFileHandler::reopenRootFile( void*, const Io::IoFlags& ) {
 
  m_log << MSG::ERROR << "reopen not implemented" << endmsg;
  return -1;
}