RootObjectMetadata.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id: RootObjectMetadata.cxx 472095 2011-12-02 10:27:01Z krasznaa $
 
 
 
// Gaudi/Athena include(s):
#include "AthenaKernel/errorcheck.h"
 
// Local include(s):
#include "RootObjectMetadata.h"
#include "isPrimitive.h"
// STL include(s):
#include <set>
 
namespace D3PD {
 
   RootObjectMetadata::RootObjectMetadata()
      : ObjectMetadata() {
 
   }
 
   RootObjectMetadata::RootObjectMetadata( const RootObjectMetadata& parent )
      : ObjectMetadata( parent ) {
 
   }
 
   StatusCode RootObjectMetadata::addVariable( const std::string& name,
                                               const std::string& type,
                                               const std::string& docstring ) {
 
      // Check that the variable has the correct prefix:
      if( ( not m_prefix.empty() ) && ( not name.starts_with( m_prefix )) ) {
         REPORT_MESSAGE_WITH_CONTEXT( MSG::ERROR, "RootObjectMetadata" )
            << "Specified variable name (" << name << ") doesn't have the "
            << "expected prefix (" << m_prefix << ")";
         return StatusCode::RECOVERABLE;
      }
 
      // Create a new variable:
      Variable var;
 
      // Remove the prefix from the variable name:
      var.setName( name.substr( m_prefix.size(), name.npos ) );
 
      // Check if it's already amongst the existing variables:
      if( m_variables.find( var ) != m_variables.end() ) {
         return StatusCode::SUCCESS;
      }
 
      // Set the type of the variable:
      var.setType( type );
 
      // Set whether the variable is a primitive:
      var.setPrimitive( isPrimitive( type ) );
 
      // Set the documentation string for the variable:
      var.setDoc( docstring );
 
      // Remember the variable:
      m_variables.insert( std::move(var) );
 
      return StatusCode::SUCCESS;
   }

   StatusCode RootObjectMetadata::checkPrefixes() {
 
      // For simplicity's sake I don't modify the original std::set,
      // but create a new one instead. This is slightly slower, but
      // much easier. In any case, performance is not a major concern
      // in this code...
      std::set< Variable > newVariables;
 
      // Loop over the variables:
      std::set< Variable >::const_iterator itr = m_variables.begin();
      std::set< Variable >::const_iterator end = m_variables.end();
      for( ; itr != end; ++itr ) {
 
         // Check if the variable name begins with the prefix (it shouldn't):
         if( ( not m_prefix.empty() ) &&
             ( itr->name().starts_with( m_prefix )) ) {
            // Create a copy of the variable, and remove the prefix from its name:
            Variable var = *itr;
            var.setName( var.name().substr( m_prefix.size(), var.name().npos ) );
            // Insert it into the new container:
            newVariables.insert( std::move(var) );
         } else {
            // Insert the variable as it is into the new container:
            newVariables.insert( *itr );
         }
      }
 
      // Swap the two containers. (This should be a fairly quick operation...)
      m_variables.swap( newVariables );
 
      return StatusCode::SUCCESS;
   }

   bool RootObjectMetadata::isObjectMetadata( const std::string& name ) {
 
      if( ( name.size() > ( RANDOM_NAME_POSTFIX_LENGTH + 1 ) ) &&
          ( name[ name.size() - ( RANDOM_NAME_POSTFIX_LENGTH + 1 ) ] == '_' ) ) {
         return true;
      } else {
         return false;
      }
   }
 
} // namespace D3PD