xAODRNFileReadTest.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
 
// System include(s):
#include <cstring>
#include <memory>
#include <vector>
#include <string>
 
// ROOT include(s):
#include <TFile.h>
#include <TError.h>
#include <TClass.h>
 
// Local include(s):
#include "xAODRootAccess/Init.h"
#include "xAODRootAccess/REvent.h"
#include "xAODRootAccess/tools/Message.h"

static const char* const APP_NAME = "xAODFileReadTest";

#define R_CHECK( EXP )                                                     \
   do {                                                                    \
      const auto result = EXP;                                             \
      if( ! result.isSuccess() ) {                                         \
         Error( APP_NAME, XAOD_MESSAGE( "Failed to execute: %s" ), #EXP ); \
         return 1;                                                         \
      }                                                                    \
   } while( 0 )

class REventClass : public xAOD::Experimental::REvent {
public:
   // Inherit the constructor(s).
   using xAOD::Experimental::REvent::REvent;

   StatusCode loadInputObjects() {
      // Get the event format object:
      const xAOD::EventFormat* ef = this->inputEventFormat();
      if( ! ef ) {
         return StatusCode::FAILURE;
      }
      // Loop over the objects of the file:
      for( const auto & ef_itr : *ef ) {
         // A helper object:
         const xAOD::EventFormatElement& efe = ef_itr.second;
         // Skip auxiliary objects:
         if( efe.branchName().rfind( "Aux." ) ==
             ( efe.branchName().size() - 4 ) ) {
            continue;
         }
         // And dynamic variables:
         if( efe.parentName() != "" ) {
            continue;
         }
         // Skip auxiliary stores with non-standard names:
         ::TClass* cl = ::TClass::GetClass( efe.className().c_str(), kTRUE,
                                            kTRUE );
         if( ! cl ) continue;
         if( cl->InheritsFrom( "SG::IConstAuxStore" ) ) {
            continue;
         }
         // Get the type of the branch:
         const std::type_info* ti = cl->GetTypeInfo();
         if( ! ti ) {
            ::Warning( "REventClass::loadInputObjects",
                       "Couldn't find std::type_info object for type %s "
                       "(key=%s)", cl->GetName(), efe.branchName().c_str() );
            continue;
         }
         // Check if the branch exists in the file:
         static constexpr bool METADATA = false;
         if( ! this->contains( efe.branchName(), *ti, METADATA ) ) {
            continue;
         }
         // Try to load the object/container:
         static constexpr bool SILENT = false;
         if( ! this->getInputObject( efe.branchName(), *ti, SILENT, METADATA ) ) {
            Error( "REventClass::loadInputObjects",
                   XAOD_MESSAGE( "Couldn't load object: %s" ),
                   efe.branchName().c_str() );
            return StatusCode::FAILURE;
         }
      }
      // Return gracefully:
      return StatusCode::SUCCESS;
   }
}; // class REventClass
 
int main( int argc, char* argv[] ) {
 
   // Initialise the environment:
   R_CHECK( xAOD::Init() );
 
   // Provide usage instructions if not enough options were given:
   if( ( argc < 2 ) ||
       ( ! strcmp( argv[ 1 ], "-h" ) ) ||
       ( ! strcmp( argv[ 1 ], "--help" ) ) ) {
      Info( APP_NAME, "Usage: %s [options] <file1> [file2]...", APP_NAME );
      Info( APP_NAME, "Options:" );
      Info( APP_NAME, "   -m <access mode>" );
      Info( APP_NAME, "      0: Branch access" );
      Info( APP_NAME, "      1: Class access (default)" );
      Info( APP_NAME, "      2: Athena access" );
      return 1;
   }
 
   // Decode the command line options:
   std::vector< std::string > fileNames;
   for( int i = 1; i < argc; ++i ) {
      fileNames.push_back( argv[ i ] );
   }
 
   // Create the "REvent" object used for the test:
   REventClass event;
 
   // Loop over the specified input files:
   for( const std::string& fileName : fileNames ) {
 
      // // Open the file:
      Info( APP_NAME, "Opening file: %s", fileName.c_str() );
 
      // Give it to REvent:
 
      // Call readFrom to setup REvent for reading
      R_CHECK( event.readFrom( fileName ) );
 
      // Loop over the file's events:
      const Long64_t entries = event.getEntries();
      for( Long64_t entry = 0; entry < entries; ++entry ) {
 
         // Load the event:
         if( event.getEntry( entry ) < 0 ) {
            Error( APP_NAME,
                   XAOD_MESSAGE( "Failed to load entry %i from file: %s" ),
                   static_cast< int >( entry ), fileName.c_str() );
            return 1;
         }
 
         // Load the event:
         R_CHECK( event.loadInputObjects() );
 
         // Tell the user where we are:
         if( ( entry % 1000 == 0 ) || ( entry +1 == entries ) ) {
            Info( APP_NAME, "===>>> Loaded entry %i / %i <<<===",
                  static_cast< int >( entry ), static_cast< int >( entries ) );
         }
      }
   }
 
   // Return gracefully:
   return 0;
}