#include <RNTupleAuxDynReader.h>

Inheritance diagram for RootAuxDynIO::RNTupleAuxDynReader:

Collaboration diagram for RootAuxDynIO::RNTupleAuxDynReader:

Classes
struct	FieldInfo

Public Member Functions
	RNTupleAuxDynReader (const std::string &field_name, const std::string &field_type, RNTupleReader *reader)
	create Reader for Aux attributes of an AuxContainer object stored in a given field More...

void	init (bool standalone)
	initialize once the mode of the Aux store is known More...

virtual void	addReaderToObject (void object, size_t row, std::recursive_mutex iomtx=nullptr) override final
	attach RNTupleAuxStore to the current Aux container @object More...

virtual const SG::auxid_set_t &	auxIDs () const override final
	Aux IDs of all the Aux attributes belonging to the Aux container being read. More...

void	addBytes (size_t bytes)

virtual size_t	getBytesRead () const override final

virtual void	resetBytesRead () override final

const FieldInfo &	getFieldInfo (const SG::auxid_t &auxid, const SG::AuxStoreInternal &store)
	get field informatino for `auxid` More...

RNTupleReader *	getNativeReader ()

virtual	~RNTupleAuxDynReader ()

bool	msgLvl (const MSG::Level lvl) const
	Test the output level. More...

MsgStream &	msg () const
	The standard message stream. More...

MsgStream &	msg (const MSG::Level lvl) const
	The standard message stream. More...

void	setLevel (MSG::Level lvl)
	Change the current logging level. More...

Protected Attributes
SG::auxid_set_t	m_auxids

std::string	m_storeFieldName

size_t	m_bytesRead = 0

int	m_storeHolderOffset = -1

bool	m_initialized = false

std::string	m_key

std::map< SG::auxid_t, FieldInfo >	m_fieldInfos

RNTupleReader *	m_ntupleReader

Private Member Functions
void	initMessaging () const
	Initialize our message level and MessageSvc. More...

Private Attributes
std::string	m_nm
	Message source name. More...

boost::thread_specific_ptr< MsgStream >	m_msg_tls
	MsgStream instance (a std::cout like with print-out levels) More...

std::atomic< IMessageSvc * >	m_imsg { nullptr }
	MessageSvc pointer. More...

std::atomic< MSG::Level >	m_lvl { MSG::NIL }
	Current logging level. More...

std::atomic_flag m_initialized	ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
	Messaging initialized (initMessaging) More...

Detailed Description

Definition at line 27 of file RNTupleAuxDynReader.h.

Constructor & Destructor Documentation

◆ RNTupleAuxDynReader()

RootAuxDynIO::RNTupleAuxDynReader::RNTupleAuxDynReader	(	const std::string &	field_name,
		const std::string &	field_type,
		RNTupleReader *	reader
	)

create Reader for Aux attributes of an AuxContainer object stored in a given field

Definition at line 113 of file RNTupleAuxDynReader.cxx.

       : AthMessaging( std::string("RNTupleAuxDynReader[")+field_name+"]" ),
         m_storeFieldName( field_name ),
         m_ntupleReader( reader )
    {
       const std::string field_prefix = field_type + "_";
       if( m_storeFieldName.rfind( field_type, 0 ) != std::string::npos ) {
          m_key = m_storeFieldName.substr( field_type.size()+1 );
       }
       ATH_MSG_VERBOSE("field name=" << m_storeFieldName << "  field_prefix=" << field_prefix << "  key=" << m_key);
       TClass *tc = TClass::GetClass( field_type.c_str() );
       if( tc ) {
          TClass *storeTC = tc->GetBaseClass("SG::IAuxStoreHolder");
          if( storeTC ) {
             m_storeHolderOffset = tc->GetBaseClassOffset( storeTC );
          } else {
             throw std::runtime_error(string("Class ") + tc->GetName() +" does not implement SG::IAuxStoreHolder");
          }
       } else {
          throw std::runtime_error(string("Class ") + field_type +" could not be found");
       }
    }

◆ ~RNTupleAuxDynReader()

virtual RootAuxDynIO::RNTupleAuxDynReader::~RNTupleAuxDynReader ( )

inlinevirtual

Definition at line 74 of file RNTupleAuxDynReader.h.

74 {}

Member Function Documentation

◆ addBytes()

void RootAuxDynIO::RNTupleAuxDynReader::addBytes ( size_t bytes )

inline

Definition at line 97 of file RNTupleAuxDynReader.h.

                                                           {
       m_bytesRead += bytes;
    }

◆ addReaderToObject()

void RootAuxDynIO::RNTupleAuxDynReader::addReaderToObject	(	void *	object,
		size_t	row,
		std::recursive_mutex *	iomtx = `nullptr`
	)

finaloverridevirtual

attach RNTupleAuxStore to the current Aux container @object

Implements RootAuxDynIO::IRootAuxDynReader.

Definition at line 255 of file RNTupleAuxDynReader.cxx.

    {
       auto store_holder = reinterpret_cast<SG::IAuxStoreHolder*>((char*)object + m_storeHolderOffset);
       bool standalone { store_holder->getStoreType()==SG::IAuxStoreHolder::AST_ObjectStore };
       if( !m_initialized )
          init(standalone);
       store_holder->setStore( new RNTupleAuxDynStore(*this, row, standalone, iomtx) );
    }

◆ auxIDs()

const SG::auxid_set_t & RootAuxDynIO::RNTupleAuxDynReader::auxIDs ( ) const

inlinefinaloverridevirtual

Aux IDs of all the Aux attributes belonging to the Aux container being read.

Implements RootAuxDynIO::IRootAuxDynReader.

Definition at line 109 of file RNTupleAuxDynReader.h.

                                                                  {
       return m_auxids;
    }

◆ getBytesRead()

size_t RootAuxDynIO::RNTupleAuxDynReader::getBytesRead ( ) const

inlinefinaloverridevirtual

Implements RootAuxDynIO::IRootAuxDynReader.

Definition at line 101 of file RNTupleAuxDynReader.h.

                                                          {
       return m_bytesRead;
    }

◆ getFieldInfo()

const RNTupleAuxDynReader::FieldInfo & RootAuxDynIO::RNTupleAuxDynReader::getFieldInfo	(	const SG::auxid_t &	auxid,
		const SG::AuxStoreInternal &	store
	)

get field informatino for auxid

Definition at line 182 of file RNTupleAuxDynReader.cxx.

    {
       FieldInfo& fieldInfo = m_fieldInfos[auxid];
       if( fieldInfo.status == FieldInfo::NotInitialized )
       {
          SG::AuxTypeRegistry& reg = SG::AuxTypeRegistry::instance();
          fieldInfo.auxid = auxid;
          fieldInfo.attribName = reg.getName(auxid);
          // Don't match this attribute if it's been renamed.
          // For example, suppose we've renamed attribute `foo' to `foo_old',
          // and someone then comes and asks for `foo'.
          // `foo' will not be found in the m_fieldMap test below
          // (`foo_old' will be in this map).  However, in the following
          // else clause, we'll recreate the field name from `foo'.
          // This field exists (renaming is only in the transient store),
          // so if we didn't have the condition here, then we'd then
          // make a `foo' attribute from that field.
          if( reg.inputRename(m_key, fieldInfo.attribName) != fieldInfo.attribName ) {
             fieldInfo.status = FieldInfo::NotFound;
             return fieldInfo;
          }
  
          bool isFieldFound = false;
  
          if( !fieldInfo.fieldName.empty() ) {
             const string field_prefix = m_storeFieldName + ':';
             const string attr_infile = fieldInfo.fieldName.substr(field_prefix.size());
             isFieldFound = (attr_infile == fieldInfo.attribName);
          }
  
          if( !isFieldFound ) {
             // mark initialized here so it remembers this field was not found
             fieldInfo.status = FieldInfo::NotFound;
             return fieldInfo;
          }
  
          if( !store.standalone() and fieldInfo.fieldName.rfind("SG::PackedContainer<", 0) == 0 )
             fieldInfo.isPackedContainer = true;
  
          /*
            string elem_tname, branch_tname;
            const type_info* ti = getAuxElementType( fieldInfo.tclass, typ, store.standalone(),
            elem_tname, branch_tname );
            const type_info* reg_ti = reg.getType(auxid);
            if( ti && ti != reg_ti && strcmp(ti->name(), reg_ti->name()) != 0 )
            {
            // type in registry is different than type in the file.
            // will need to use ROOT auto schema evolution
            fieldInfo.needsSE = true;
            errorcheck::ReportMessage msg (MSG::INFO, ERRORCHECK_ARGS, "RNTupleAuxDynReader");
            msg << "attribute " << fieldInfo.attribName << " (id=" << auxid <<
            " typename=" << SG::AuxTypeRegistry::instance().getType(auxid)->name()
            << ") has different type than the branch " << branch_tname;
  
            const std::type_info *tinf =  store.getIOType(auxid);
            fieldInfo.SE_tclass  = TClass::GetClass(*tinf);
            fieldInfo.SE_edt = kOther_t;
            if( !fieldInfo.SE_tclass  ) {
            fieldInfo.SE_edt = TDataType::GetType(*tinf);
            if( fieldInfo.SE_edt <=0 ) {
            fieldInfo.status = FieldInfo::TypeError;
            throw string("Error getting ROOT type for AUX branch ") + fieldInfo.branch->GetName()
            + " typeinfo=" + tinf->name();
            }
            }
            }
          */
          fieldInfo.status = FieldInfo::Initialized;
       }
       return fieldInfo;
    }

◆ getNativeReader()

RNTupleReader * RootAuxDynIO::RNTupleAuxDynReader::getNativeReader ( )

inline

Definition at line 113 of file RNTupleAuxDynReader.h.

                                                                {
       return m_ntupleReader;
    }

◆ init()

void RootAuxDynIO::RNTupleAuxDynReader::init ( bool standalone )

initialize once the mode of the Aux store is known

Definition at line 141 of file RNTupleAuxDynReader.cxx.

    {
       if( m_initialized )  return;
    
       const SG::AuxTypeRegistry& reg = SG::AuxTypeRegistry::instance();
       const string field_prefix = m_storeFieldName + ':';
       const auto& desc = m_ntupleReader->GetDescriptor();
 #if ROOT_VERSION_CODE >= ROOT_VERSION( 6, 31, 0 )
       for( const auto &f : desc.GetTopLevelFields() ) {
 #else
       for( const auto &f : desc->GetTopLevelFields() ) {
 #endif
          const string field_name = f.GetFieldName();
          if( field_name.rfind(field_prefix,0) == 0 ) {
             const string attr_infile = field_name.substr(field_prefix.size());
             const string attr_name = reg.inputRename(m_key, attr_infile);
             const string field_type = f.GetTypeName();
  
             SG::auxid_t auxid = getAuxIdForAttribute(attr_name, field_type, standalone);
             // add AuxID to the list
             // May still be null if we don't have a dictionary for this field
             if( auxid != SG::null_auxid ) {
                m_auxids.insert(auxid);
                m_fieldInfos[auxid].fieldName = field_name;
 #if ROOT_VERSION_CODE >= ROOT_VERSION( 6, 31, 0 )
                m_fieldInfos[auxid].view_p = std::make_unique<RNTupleView<void,true>>(
                   m_ntupleReader->GetView<void>(field_name, nullptr) );
 #endif
             } else {
                errorcheck::ReportMessage msg (MSG::WARNING, ERRORCHECK_ARGS, "RNTupleAuxDynReader::init");
                msg << "Could not find auxid for " << attr_infile << " type: " << field_type
                    << "  standalone=" << standalone;
             }
          } 
       }
       m_initialized = true;
    }

◆ initMessaging()

void AthMessaging::initMessaging ( ) const

privateinherited

Initialize our message level and MessageSvc.

This method should only be called once.

Definition at line 39 of file AthMessaging.cxx.

 {
   m_imsg = Athena::getMessageSvc();
   m_lvl = m_imsg ?
     static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
     MSG::INFO;
 }

◆ msg() [1/2]

MsgStream & AthMessaging::msg ( ) const

inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 164 of file AthMessaging.h.

 {
   MsgStream* ms = m_msg_tls.get();
   if (!ms) {
     if (!m_initialized.test_and_set()) initMessaging();
     ms = new MsgStream(m_imsg,m_nm);
     m_msg_tls.reset( ms );
   }
  
   ms->setLevel (m_lvl);
   return *ms;
 }

◆ msg() [2/2]

MsgStream & AthMessaging::msg ( const MSG::Level lvl ) const

inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 179 of file AthMessaging.h.

180 { return msg() << lvl; }

◆ msgLvl()

bool AthMessaging::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Test the output level.

Parameters

lvl	The message level to test against

Returns: boolean Indicating if messages at given level will be printed

Return values

true	Messages at level "lvl" will be printed

Definition at line 151 of file AthMessaging.h.

 {
   if (!m_initialized.test_and_set()) initMessaging();
   if (m_lvl <= lvl) {
     msg() << lvl;
     return true;
   } else {
     return false;
   }
 }

◆ resetBytesRead()

void RootAuxDynIO::RNTupleAuxDynReader::resetBytesRead ( )

inlinefinaloverridevirtual

Implements RootAuxDynIO::IRootAuxDynReader.

Definition at line 105 of file RNTupleAuxDynReader.h.

                                                    {
       m_bytesRead = 0;
    }

◆ setLevel()

void AthMessaging::setLevel ( MSG::Level lvl )

inherited

Change the current logging level.

Use this rather than msg().setLevel() for proper operation with MT.

Definition at line 28 of file AthMessaging.cxx.

 {
   m_lvl = lvl;
 }

Member Data Documentation

◆ ATLAS_THREAD_SAFE

std::atomic_flag m_initialized AthMessaging::ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT

mutableprivateinherited

Messaging initialized (initMessaging)

Definition at line 141 of file AthMessaging.h.

◆ m_auxids

SG::auxid_set_t RootAuxDynIO::RNTupleAuxDynReader::m_auxids

protected

Definition at line 78 of file RNTupleAuxDynReader.h.

◆ m_bytesRead

size_t RootAuxDynIO::RNTupleAuxDynReader::m_bytesRead = 0

protected

Definition at line 82 of file RNTupleAuxDynReader.h.

◆ m_fieldInfos

std::map<SG::auxid_t, FieldInfo> RootAuxDynIO::RNTupleAuxDynReader::m_fieldInfos

protected

Definition at line 89 of file RNTupleAuxDynReader.h.

◆ m_imsg

std::atomic<IMessageSvc*> AthMessaging::m_imsg { nullptr }

mutableprivateinherited

MessageSvc pointer.

Definition at line 135 of file AthMessaging.h.

◆ m_initialized

bool RootAuxDynIO::RNTupleAuxDynReader::m_initialized = false

protected

Definition at line 85 of file RNTupleAuxDynReader.h.

◆ m_key

std::string RootAuxDynIO::RNTupleAuxDynReader::m_key

protected

Definition at line 86 of file RNTupleAuxDynReader.h.

◆ m_lvl

std::atomic<MSG::Level> AthMessaging::m_lvl { MSG::NIL }

mutableprivateinherited

Current logging level.

Definition at line 138 of file AthMessaging.h.

◆ m_msg_tls

boost::thread_specific_ptr<MsgStream> AthMessaging::m_msg_tls

mutableprivateinherited

MsgStream instance (a std::cout like with print-out levels)

Definition at line 132 of file AthMessaging.h.

◆ m_nm

std::string AthMessaging::m_nm

privateinherited

Message source name.

Definition at line 129 of file AthMessaging.h.

◆ m_ntupleReader

RNTupleReader* RootAuxDynIO::RNTupleAuxDynReader::m_ntupleReader

protected

Definition at line 92 of file RNTupleAuxDynReader.h.

◆ m_storeFieldName

std::string RootAuxDynIO::RNTupleAuxDynReader::m_storeFieldName

protected

Definition at line 80 of file RNTupleAuxDynReader.h.

◆ m_storeHolderOffset

int RootAuxDynIO::RNTupleAuxDynReader::m_storeHolderOffset = -1

protected

Definition at line 84 of file RNTupleAuxDynReader.h.

The documentation for this class was generated from the following files:

Classes

Public Member Functions

Protected Attributes

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ RNTupleAuxDynReader()

◆ ~RNTupleAuxDynReader()

Member Function Documentation

◆ addBytes()

◆ addReaderToObject()

◆ auxIDs()

◆ getBytesRead()

◆ getFieldInfo()

◆ getNativeReader()

◆ init()

◆ initMessaging()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ resetBytesRead()

◆ setLevel()

Member Data Documentation

◆ ATLAS_THREAD_SAFE

◆ m_auxids

◆ m_bytesRead

◆ m_fieldInfos

◆ m_imsg

◆ m_initialized

◆ m_key

◆ m_lvl

◆ m_msg_tls

◆ m_nm

◆ m_ntupleReader

◆ m_storeFieldName

◆ m_storeHolderOffset