Gaudi::RootDataConnection Class Reference

Concrete implementation of the IDataConnection interface to access ROOT files. More...

#include <GaudiRootCnv/RootDataConnection.h>

Inheritance diagram for Gaudi::RootDataConnection:

Collaboration diagram for Gaudi::RootDataConnection:

Classes
class	ContainerSection
	Internal helper class, which described a TBranch section in a ROOT file. More...
class	Tool
	Helper class to facilitate an abstraction layer for reading POOL style files with this package. More...

Public Types
enum	Status : StatusCode::code_t { Status::ROOT_READ_ERROR = 0x2, Status::ROOT_OPEN_ERROR = 0x4 }
typedef std::vector< std::string >	StringVec
	Type definition for string maps. More...
typedef std::vector< std::pair< std::string, std::string > >	ParamMap
	Type definition for the parameter map. More...
typedef std::map< std::string, TTree *, std::less<> >	Sections
	Definition of tree sections. More...
typedef std::vector< ContainerSection >	ContainerSections
	Definition of container sections to handle merged files. More...
typedef std::map< std::string, ContainerSections, std::less<> >	MergeSections
	Definition of database section to handle merged files. More...
typedef std::vector< RootRef >	LinkSections
	Link sections definition. More...
typedef std::set< const IInterface * >	Clients
	Client set. More...
Public Types inherited from Gaudi::IDataConnection
enum	IoType { READ = 1 << 1, UPDATE = 1 << 2, CREATE = 1 << 3, RECREATE = ( 1 << 4 ) + ( 1 << 3 ) }
	I/O Connection types. More...
enum	IoStatus { BAD_DATA_CONNECTION = 4 }
	Status Code on bad file connection. More...

Public Member Functions
MsgStream &	msgSvc () const
	Allow access to printer service. More...
Tool *	makeTool ()
	Create file access tool to encapsulate POOL compatibiliy. More...
	RootDataConnection (const IInterface *own, boost::string_ref nam, std::shared_ptr< RootConnectionSetup > setup)
	Standard constructor. More...
TFile *	file () const
	Direct access to TFile structure. More...
bool	isConnected () const override
	Check if connected to data source. More...
bool	isWritable () const
	Is the file writable? More...
Tool *	tool () const
	Access tool. More...
const MergeSections &	mergeSections () const
	Access merged data section inventory. More...
const StringVec &	mergeFIDs () const
	Access merged FIDs. More...
void	addClient (const IInterface *client)
	Add new client to this data source. More...
size_t	removeClient (const IInterface *client)
	Remove client from this data source. More...
bool	lookupClient (const IInterface *client) const
	Lookup client for this data source. More...
void	badWriteError (boost::string_ref msg) const
	Error handler when bad write statements occur. More...
std::pair< const RootRef *, const ContainerSection * >	getMergeSection (boost::string_ref container, int entry) const
	Access link section for single container and entry. More...
void	enableStatistics (boost::string_ref section)
	Enable TTreePerStats. More...
void	saveStatistics (boost::string_ref statisticsFile)
	Save TTree access statistics if required. More...
int	loadObj (boost::string_ref section, boost::string_ref cnt, unsigned long entry, DataObject *&pObj)
	Load object. More...
int	loadRefs (boost::string_ref section, boost::string_ref cnt, unsigned long entry, RootObjectRefs &refs)
	Load references object. More...
std::pair< int, unsigned long >	saveObj (boost::string_ref section, boost::string_ref cnt, TClass *cl, DataObject *pObj, int buff_siz, int split_lvl, bool fill_missing=false)
	Save object of a given class to section and container. More...
std::pair< int, unsigned long >	save (boost::string_ref section, boost::string_ref cnt, TClass *cl, void *pObj, int buff_siz, int split_lvl, bool fill_missing=false)
	Save object of a given class to section and container. More...
StatusCode	connectRead () override
	Open data stream in read mode. More...
StatusCode	connectWrite (IoType typ) override
	Open data stream in write mode. More...
StatusCode	disconnect () override
	Release data stream and release implementation dependent resources. More...
StatusCode	read (void *const, size_t) override
	Read root byte buffer from input stream. More...
StatusCode	write (const void *, int) override
	Write root byte buffer to output stream. More...
long long int	seek (long long int, int) override
	Seek on the file described by ioDesc. Arguments as in ::seek() More...
TTree *	getSection (boost::string_ref sect, bool create=false)
	Access TTree section from section name. The section is created if required. More...
TBranch *	getBranch (boost::string_ref section, boost::string_ref branch_name)
	Access data branch by name: Get existing branch in read only mode. More...
TBranch *	getBranch (boost::string_ref section, boost::string_ref branch_name, TClass *cl, void *ptr, int buff_siz, int split_lvl)
	Access data branch by name: Get existing branch in write mode. More...
void	makeRef (const IRegistry &pA, RootRef &ref)
	Create reference object from registry entry. More...
void	makeRef (boost::string_ref name, long clid, int tech, boost::string_ref db, boost::string_ref cnt, int entry, RootRef &ref)
	Create reference object from values. More...
int	makeLink (boost::string_ref p)
	Convert path string to path index. More...
const std::string &	getDb (int which) const
	Access database/file name from saved index. More...
const std::string &	getCont (int which) const
	Access container name from saved index. More...
const std::string &	getLink (int which) const
	Access link name from saved index. More...
Public Member Functions inherited from Gaudi::IDataConnection
	IDataConnection (const IInterface *own, std::string nam)
	Standard constructor. More...
virtual	~IDataConnection ()=default
	Standard destructor. More...
const std::string &	name () const
	Connection name. More...
void	setFID (std::string fid)
	Set file ID. More...
const std::string &	fid () const
	Access file id. More...
const std::string &	pfn () const
	Access physical file name. More...
void	setPFN (std::string fn)
	Set physical file name. More...
int	ageFile ()
	Increase age of I/O source. More...
void	resetAge ()
	Reset age. More...
int	age () const
	Access age counter. More...
const IInterface *	owner () const
	Owner instance. More...

Public Attributes
std::unique_ptr< Tool >	m_tool

Protected Member Functions
const std::string &	empty () const
	Empty string reference. More...
StatusCode	saveRefs ()
	Internal helper to save/update reference tables. More...

Protected Attributes
std::shared_ptr< RootConnectionSetup >	m_setup
	Reference to the setup structure. More...
std::unique_ptr< TTreePerfStats >	m_statistics
	I/O read statistics from TTree. More...
std::unique_ptr< TFile >	m_file
	Reference to ROOT file. More...
TTree *	m_refs = nullptr
	Pointer to the reference tree. More...
Sections	m_sections
	Tree sections in TFile. More...
StringVec	m_dbs
	Map containing external database file names (fids) More...
StringVec	m_conts
	Map containing external container names. More...
StringVec	m_links
	Map containing internal links names. More...
StringVec	m_mergeFIDs
	Map containing merge FIDs. More...
ParamMap	m_params
	Parameter map for file parameters. More...
MergeSections	m_mergeSects
	Database section map for merged files. More...
LinkSections	m_linkSects
	Database link sections. More...
Clients	m_clients
	Client list. More...
std::string	m_empty
	Buffer for empty string reference. More...
Protected Attributes inherited from Gaudi::IDataConnection
std::string	m_name
	Connection name/identifier. More...
std::string	m_fid
	File ID of the connection. More...
std::string	m_pfn
	Physical file name of the connection. More...
int	m_age = 0
	Age counter. More...
const IInterface *	m_owner = nullptr
	Owner pointer. More...

Friends
class	Tool

Detailed Description

Concrete implementation of the IDataConnection interface to access ROOT files.

Author

M.Frank

Version

1.0

Date

20/12/2009

Definition at line 93 of file RootDataConnection.h.

Member Typedef Documentation

typedef std::set<const IInterface*> Gaudi::RootDataConnection::Clients

Client set.

Definition at line 141 of file RootDataConnection.h.

typedef std::vector<ContainerSection> Gaudi::RootDataConnection::ContainerSections

Definition of container sections to handle merged files.

Definition at line 135 of file RootDataConnection.h.

typedef std::vector<RootRef> Gaudi::RootDataConnection::LinkSections

Link sections definition.

Definition at line 139 of file RootDataConnection.h.

typedef std::map<std::string, ContainerSections, std::less<> > Gaudi::RootDataConnection::MergeSections

Definition of database section to handle merged files.

Definition at line 137 of file RootDataConnection.h.

typedef std::vector<std::pair<std::string, std::string> > Gaudi::RootDataConnection::ParamMap

Type definition for the parameter map.

Definition at line 131 of file RootDataConnection.h.

typedef std::map<std::string, TTree*, std::less<> > Gaudi::RootDataConnection::Sections

Definition of tree sections.

Definition at line 133 of file RootDataConnection.h.

typedef std::vector<std::string> Gaudi::RootDataConnection::StringVec

Type definition for string maps.

Definition at line 129 of file RootDataConnection.h.

Member Enumeration Documentation

enum Gaudi::RootDataConnection::Status : StatusCode::code_t

strong

Enumerator
ROOT_READ_ERROR
ROOT_OPEN_ERROR

Definition at line 95 of file RootDataConnection.h.

: StatusCode::code_t { ROOT_READ_ERROR = 0x2, ROOT_OPEN_ERROR = 0x4 };

Constructor & Destructor Documentation

RootDataConnection::RootDataConnection	(	const IInterface *	own,
		boost::string_ref	nam,
		std::shared_ptr< RootConnectionSetup >	setup
	)

Standard constructor.

Definition at line 153 of file RootDataConnection.cpp.

    : IDataConnection( owner, fname.to_string() )
    , m_setup( std::move( setup ) ) { //               01234567890123456789012345678901234567890
  // Check if FID: A82A3BD8-7ECB-DC11-8DC0-000423D950B0
  if ( fname.size() == 36 && fname[8] == '-' && fname[13] == '-' && fname[18] == '-' && fname[23] == '-' ) {
    m_name = "FID:";
    m_name.append( fname.data(), fname.size() );
  }
  m_age = 0;
  m_file.reset();
  addClient( owner );
}

Member Function Documentation

void RootDataConnection::addClient

(

const IInterface *

client

)

Add new client to this data source.

Definition at line 168 of file RootDataConnection.cpp.

{ m_clients.insert( client ); }

void RootDataConnection::badWriteError

(

boost::string_ref

msg

)

const

Error handler when bad write statements occur.

Definition at line 184 of file RootDataConnection.cpp.

                                                                  {
  msgSvc() << MSG::ERROR << "File:" << fid() << "Failed action:" << msg << endmsg;
}

StatusCode RootDataConnection::connectRead ( )

overridevirtual

Open data stream in read mode.

Connect the file in READ mode.

Implements Gaudi::IDataConnection.

Definition at line 225 of file RootDataConnection.cpp.

                                           {
  m_file.reset( TFile::Open( m_pfn.c_str() ) );
  if ( !m_file || m_file->IsZombie() ) {
    m_file.reset();
    return StatusCode::FAILURE;
  }
  StatusCode sc = StatusCode::FAILURE;
  msgSvc() << MSG::DEBUG << "Opened file " << m_pfn << " in mode READ. [" << m_fid << "]" << endmsg << MSG::DEBUG;
  if ( msgSvc().isActive() ) m_file->ls();
  msgSvc() << MSG::VERBOSE;
  if ( msgSvc().isActive() ) m_file->Print();
  if ( makeTool() ) {
    sc = m_tool->readRefs();
    sc.ignore();
#if ROOT_VERSION_CODE >= ROOT_VERSION( 5, 33, 0 )
    if ( sc == Status::ROOT_READ_ERROR ) {
      IIncidentSvc* inc = m_setup->incidentSvc();
      if ( inc ) { inc->fireIncident( Incident( pfn(), IncidentType::CorruptedInputFile ) ); }
    }
#endif
  }
  if ( !sc.isSuccess() ) return sc;
  bool   need_fid = m_fid == m_pfn;
  string fid      = m_fid;
  m_mergeFIDs.clear();
  for ( auto& elem : m_params ) {
    if ( elem.first == "FID" ) {
      m_mergeFIDs.push_back( elem.second );
      if ( elem.second != m_fid ) {
        msgSvc() << MSG::DEBUG << "Check FID param:" << elem.second << endmsg;
        // if ( m_fid == m_pfn ) {
        m_fid = elem.second;
        //}
      }
    }
  }
  if ( !need_fid && fid != m_fid ) {
    msgSvc() << MSG::ERROR << "FID mismatch:" << fid << "(Catalog) != " << m_fid << "(file)" << endmsg
             << "for PFN:" << m_pfn << endmsg;
    return StatusCode::FAILURE;
  }
  msgSvc() << MSG::DEBUG << "Using FID " << m_fid << " from params table...." << endmsg << "for PFN:" << m_pfn
           << endmsg;
  return sc;
}

StatusCode RootDataConnection::connectWrite ( IoType  typ )

overridevirtual

Open data stream in write mode.

Implements Gaudi::IDataConnection.

Definition at line 272 of file RootDataConnection.cpp.

                                                        {
  int compress = RootConnectionSetup::compression();
  msgSvc() << MSG::DEBUG;
  switch ( typ ) {
  case CREATE:
    resetAge();
    m_file.reset( TFile::Open( m_pfn.c_str(), "CREATE", "Root event data", compress ) );
    m_refs = new TTree( "Refs", "Root reference data" );
    msgSvc() << "Opened file " << m_pfn << " in mode CREATE. [" << m_fid << "]" << endmsg;
    m_params.emplace_back( "PFN", m_pfn );
    if ( m_fid != m_pfn ) { m_params.emplace_back( "FID", m_fid ); }
    makeTool();
    break;
  case RECREATE:
    resetAge();
    m_file.reset( TFile::Open( m_pfn.c_str(), "RECREATE", "Root event data", compress ) );
    msgSvc() << "Opened file " << m_pfn << " in mode RECREATE. [" << m_fid << "]" << endmsg;
    m_refs = new TTree( "Refs", "Root reference data" );
    m_params.emplace_back( "PFN", m_pfn );
    if ( m_fid != m_pfn ) { m_params.emplace_back( "FID", m_fid ); }
    makeTool();
    break;
  case UPDATE:
    resetAge();
    m_file.reset( TFile::Open( m_pfn.c_str(), "UPDATE", "Root event data", compress ) );
    msgSvc() << "Opened file " << m_pfn << " in mode UPDATE. [" << m_fid << "]" << endmsg;
    if ( m_file && !m_file->IsZombie() ) {
      if ( makeTool() ) {
        StatusCode sc = m_tool->readRefs();
        sc.ignore();
        if ( sc == Status::ROOT_READ_ERROR ) {
#if ROOT_VERSION_CODE >= ROOT_VERSION( 5, 33, 0 )
          IIncidentSvc* inc = m_setup->incidentSvc();
          if ( inc ) { inc->fireIncident( Incident( pfn(), IncidentType::CorruptedInputFile ) ); }
#endif
        }
        return sc;
      }
      TDirectory::TContext ctxt( m_file.get() );
      m_refs = new TTree( "Refs", "Root reference data" );
      makeTool();
      return StatusCode::SUCCESS;
    }
    break;
  default:
    m_refs = nullptr;
    m_file.reset();
    return StatusCode::FAILURE;
  }
  return m_file ? StatusCode::SUCCESS : StatusCode::FAILURE;
}

StatusCode RootDataConnection::disconnect ( )

overridevirtual

Release data stream and release implementation dependent resources.

Implements Gaudi::IDataConnection.

Definition at line 325 of file RootDataConnection.cpp.

                                          {
  if ( m_file ) {
    if ( !m_file->IsZombie() ) {
      if ( m_file->IsWritable() ) {
        msgSvc() << MSG::DEBUG;
        TDirectory::TContext ctxt( m_file.get() );
        if ( m_refs ) {
          if ( !m_tool->saveRefs().isSuccess() ) badWriteError( "Saving References" );
          if ( m_refs->Write() < 0 ) badWriteError( "Write Reference branch" );
        }
        for ( auto& i : m_sections ) {
          if ( i.second ) {
            if ( i.second->Write() < 0 ) badWriteError( "Write section:" + i.first );
            msgSvc() << "Disconnect section " << i.first << " " << i.second->GetName() << endmsg;
          }
        }
        m_sections.clear();
      }
      msgSvc() << MSG::DEBUG;
      if ( msgSvc().isActive() ) m_file->ls();
      msgSvc() << MSG::VERBOSE;
      if ( msgSvc().isActive() ) m_file->Print();
      m_file->Close();
    }
    msgSvc() << MSG::DEBUG << "Disconnected file " << m_pfn << " " << m_file->GetName() << endmsg;
    m_file.reset();
    m_tool.reset();
  }
  return StatusCode::SUCCESS;
}

CSTR RootDataConnection::empty ( ) const

protected

Empty string reference.

Definition at line 456 of file RootDataConnection.cpp.

{ return s_empty; }

void RootDataConnection::enableStatistics

(

boost::string_ref

section

)

Enable TTreePerStats.

Definition at line 198 of file RootDataConnection.cpp.

                                                                   {
  if ( m_statistics ) {
    TTree* t = getSection( section, false );
    if ( t ) {
      m_statistics.reset( new TTreePerfStats( ( section.to_string() + "_ioperf" ).c_str(), t ) );
      return;
    }
    msgSvc() << MSG::WARNING << "Failed to enable perfstats for tree:" << section << endmsg;
    return;
  }
  msgSvc() << MSG::INFO << "Perfstats are ALREADY ENABLED." << endmsg;
}

TFile* Gaudi::RootDataConnection::file ( ) const

inline

Direct access to TFile structure.

Definition at line 243 of file RootDataConnection.h.

{ return m_file.get(); }

TBranch* Gaudi::RootDataConnection::getBranch ( boost::string_ref  section, boost::string_ref  branch_name  )

inline

Access data branch by name: Get existing branch in read only mode.

Definition at line 303 of file RootDataConnection.h.

                                                                               {
      return m_tool->getBranch( section, branch_name );
    }

TBranch * RootDataConnection::getBranch	(	boost::string_ref	section,
		boost::string_ref	branch_name,
		TClass *	cl,
		void *	ptr,
		int	buff_siz,
		int	split_lvl
	)

Access data branch by name: Get existing branch in write mode.

Definition at line 423 of file RootDataConnection.cpp.

                                                                      {
  string n = branch_name.to_string();
  std::replace_if( begin( n ), end( n ), []( const char c ) { return !isalnum( c ); }, '_' );
  n += ".";
  TTree*   t = getSection( section, true );
  TBranch* b = t->GetBranch( n.c_str() );
  if ( !b && cl && m_file->IsWritable() ) {
    b = t->Branch( n.c_str(), cl->GetName(), (void*)( ptr ? &ptr : nullptr ), buff_siz, split_lvl );
  }
  if ( !b ) b = t->GetBranch( branch_name.to_string().c_str() );
  if ( b ) b->SetAutoDelete( kFALSE );
  return b;
}

const std::string& Gaudi::RootDataConnection::getCont ( int  which ) const

inline

Access container name from saved index.

Definition at line 323 of file RootDataConnection.h.

                                                {
      return ( which >= 0 ) && ( size_t( which ) < m_conts.size() ) ? *( m_conts.begin() + which ) : empty();
    }

CSTR RootDataConnection::getDb

(

int

which

)

const

Access database/file name from saved index.

Definition at line 447 of file RootDataConnection.cpp.

                                                {
  if ( ( which >= 0 ) && ( size_t( which ) < m_dbs.size() ) ) {
    if ( *( m_dbs.begin() + which ) == s_local ) return m_fid;
    return *( m_dbs.begin() + which );
  }
  return s_empty;
}

const std::string& Gaudi::RootDataConnection::getLink ( int  which ) const

inline

Access link name from saved index.

Definition at line 328 of file RootDataConnection.h.

                                                {
      return ( which >= 0 ) && ( size_t( which ) < m_links.size() ) ? *( m_links.begin() + which ) : empty();
    }

pair< const RootRef *, const RootDataConnection::ContainerSection * > RootDataConnection::getMergeSection	(	boost::string_ref	container,
		int	entry
	)		const

Access link section for single container and entry.

Definition at line 563 of file RootDataConnection.cpp.

                                                                                {
  // size_t idx = cont.find('/',1);
  // string container = cont[0]=='/' ? cont.substr(1,idx==string::npos?idx:idx-1) : cont;
  auto i = m_mergeSects.find( container );
  if ( i != m_mergeSects.end() ) {
    size_t                   cnt = 0;
    const ContainerSections& s   = ( *i ).second;
    for ( auto j = s.cbegin(); j != s.cend(); ++j, ++cnt ) {
      const ContainerSection& c = *j;
      if ( entry >= c.start && entry < ( c.start + c.length ) ) {
        if ( m_linkSects.size() > cnt ) {
          if ( msgSvc().isActive() ) {
            msgSvc() << MSG::VERBOSE << "MergeSection for:" << container << "  [" << entry << "]" << endmsg
                     << "FID:" << m_fid << " -> PFN:" << m_pfn << endmsg;
          }
          return {&( m_linkSects[cnt] ), &c};
        }
      }
    }
  }
  msgSvc() << MSG::DEBUG << "Return INVALID MergeSection for:" << container << "  [" << entry << "]" << endmsg
           << "FID:" << m_fid << " -> PFN:" << m_pfn << endmsg;
  return {nullptr, nullptr};
}

TTree * RootDataConnection::getSection	(	boost::string_ref	sect,
		bool	create = false
	)

Access TTree section from section name. The section is created if required.

Definition at line 357 of file RootDataConnection.cpp.

                                                                            {
  auto   it = m_sections.find( section );
  TTree* t  = ( it != m_sections.end() ? it->second : nullptr );
  if ( !t ) {
    t = (TTree*)m_file->Get( section.to_string().c_str() );
    if ( !t && create ) {
      TDirectory::TContext ctxt( m_file.get() );
      t = new TTree( section.to_string().c_str(), "Root data for Gaudi" );
    }
    if ( t ) {
      int cacheSize = m_setup->cacheSize;
      if ( create ) {
        // t->SetAutoFlush(100);
      }
      if ( section == m_setup->loadSection && cacheSize > -2 ) {
        MsgStream& msg          = msgSvc();
        int        learnEntries = m_setup->learnEntries;
        t->SetCacheSize( cacheSize );
        t->SetCacheLearnEntries( learnEntries );
        msg << MSG::DEBUG;
        if ( create ) {
          msg << "Tree:" << section << "Setting up tree cache:" << cacheSize << endmsg;
        } else {
          const StringVec& vB = m_setup->vetoBranches;
          const StringVec& cB = m_setup->cacheBranches;
          msg << "Tree:" << section << " Setting up tree cache:" << cacheSize << " Add all branches." << endmsg;
          msg << "Tree:" << section << " Learn for " << learnEntries << " entries." << endmsg;

          if ( cB.empty() && vB.empty() ) {
            msg << "Adding (default) all branches to tree cache." << endmsg;
            t->AddBranchToCache( "*", kTRUE );
          }
          if ( cB.size() == 1 && cB[0] == "*" ) {
            msg << "Adding all branches to tree cache according to option \"CacheBranches\"." << endmsg;
            t->AddBranchToCache( "*", kTRUE );
          } else {
            for ( TIter it( t->GetListOfBranches() ); it.Next(); ) {
              const char* n   = ( (TNamed*)( *it ) )->GetName();
              bool        add = false, veto = false;
              for ( const auto& i : cB ) {
                if ( !match_wild( n, ( i ).c_str() ) ) continue;
                add = true;
                break;
              }
              for ( auto i = vB.cbegin(); !add && i != vB.cend(); ++i ) {
                if ( !match_wild( n, ( *i ).c_str() ) ) continue;
                veto = true;
                break;
              }
              if ( add && !veto ) {
                msg << "Add " << n << " to branch cache." << endmsg;
                t->AddBranchToCache( n, kTRUE );
              } else {
                msg << "Do not cache branch " << n << endmsg;
              }
            }
          }
        }
      }
      m_sections[section.to_string()] = t;
    }
  }
  return t;
}

bool Gaudi::RootDataConnection::isConnected ( ) const

inlineoverridevirtual

Check if connected to data source.

Implements Gaudi::IDataConnection.

Definition at line 245 of file RootDataConnection.h.

{ return bool( m_file ); }

bool Gaudi::RootDataConnection::isWritable ( ) const

inline

Is the file writable?

Definition at line 247 of file RootDataConnection.h.

{ return m_file && m_file->IsWritable(); }

int RootDataConnection::loadObj	(	boost::string_ref	section,
		boost::string_ref	cnt,
		unsigned long	entry,
		DataObject *&	pObj
	)

Load object.

Definition at line 496 of file RootDataConnection.cpp.

                                                     {
  TBranch* b = getBranch( section, cnt );
  if ( b ) {
    TClass* cl = gROOT->GetClass( b->GetClassName(), kTRUE );
    if ( cl ) {
      int nb = -1;
      pObj   = (DataObject*)cl->New();
      {
        DataObjectPush push( pObj );
        b->SetAddress( &pObj );
        if ( section == m_setup->loadSection ) {
          TTree* t = b->GetTree();
          if ( Long64_t( entry ) != t->GetReadEntry() ) { t->LoadTree( Long64_t( entry ) ); }
        }
        nb = b->GetEntry( entry );
        msgSvc() << MSG::VERBOSE;
        if ( msgSvc().isActive() ) {
          msgSvc() << "Load [" << entry << "] --> " << section << ":" << cnt << "  " << nb << " bytes." << endmsg;
        }
        if ( nb < 0 ) { // This is definitely an error...ROOT says if reads fail, -1 is issued.
#if ROOT_VERSION_CODE >= ROOT_VERSION( 5, 33, 0 )
          IIncidentSvc* inc = m_setup->incidentSvc();
          if ( inc ) { inc->fireIncident( Incident( pfn(), IncidentType::CorruptedInputFile ) ); }
#endif
        } else if ( nb == 0 && pObj->clID() == CLID_DataObject ) {
          TFile* f   = b->GetFile();
          int    vsn = f->GetVersion();
          if ( vsn < 52400 ) {
            // For Gaudi v21r5 (ROOT 5.24.00b) DataObject::m_version was not written!
            // Still this call be well be successful.
            nb = 1;
          } else if ( vsn > 1000000 && ( vsn % 1000000 ) < 52400 ) {
            // dto. Some POOL files have for unknown reasons a version
            // not according to ROOT standards. Hack this explicitly.
            nb = 1;
          }
        }
        if ( nb < 0 ) {
          delete pObj;
          pObj = nullptr;
        }
      }
      return nb;
    }
  }
  return -1;
}

int RootDataConnection::loadRefs	(	boost::string_ref	section,
		boost::string_ref	cnt,
		unsigned long	entry,
		RootObjectRefs &	refs
	)

Load references object.

Definition at line 546 of file RootDataConnection.cpp.

                                                         {
  int nbytes = m_tool->loadRefs( section, cnt, entry, refs );
#if ROOT_VERSION_CODE >= ROOT_VERSION( 5, 33, 0 )
  if ( nbytes < 0 ) {
    // This is definitely an error:
    // -- Either branch not preesent at all or
    // -- ROOT I/O error, which issues -1
    IIncidentSvc* inc = m_setup->incidentSvc();
    if ( inc ) { inc->fireIncident( Incident( pfn(), IncidentType::CorruptedInputFile ) ); }
  }
#endif
  return nbytes;
}

bool RootDataConnection::lookupClient

(

const IInterface *

client

)

const

Lookup client for this data source.

Definition at line 178 of file RootDataConnection.cpp.

                                                                      {
  auto i = m_clients.find( client );
  return i != m_clients.end();
}

int RootDataConnection::makeLink

(

boost::string_ref

p

)

Convert path string to path index.

Definition at line 439 of file RootDataConnection.cpp.

                                                    {
  auto ip = std::find( std::begin( m_links ), std::end( m_links ), p );
  if ( ip != std::end( m_links ) ) return std::distance( std::begin( m_links ), ip );
  m_links.push_back( p.to_string() );
  return m_links.size() - 1;
}

void RootDataConnection::makeRef	(	const IRegistry &	pA,
		RootRef &	ref
	)

Create reference object from registry entry.

Definition at line 589 of file RootDataConnection.cpp.

                                                                    {
  IOpaqueAddress* pA = pR.address();
  makeRef( pR.name(), pA->clID(), pA->svcType(), pA->par()[0], pA->par()[1], -1, ref );
}

void RootDataConnection::makeRef	(	boost::string_ref	name,
		long	clid,
		int	tech,
		boost::string_ref	db,
		boost::string_ref	cnt,
		int	entry,
		RootRef &	ref
	)

Create reference object from values.

Definition at line 595 of file RootDataConnection.cpp.

                                                                                 {
  auto db   = ( dbase == m_fid ? boost::string_ref{s_local} : dbase );
  ref.entry = entry;

  int cdb = -1;
  if ( !db.empty() ) {
    auto idb = std::find_if( m_dbs.begin(), m_dbs.end(), [&]( const std::string& i ) { return i == db; } );
    cdb      = std::distance( m_dbs.begin(), idb );
    if ( idb == m_dbs.end() ) m_dbs.push_back( db.to_string() );
  }

  int ccnt = -1;
  if ( !cnt.empty() ) {
    auto icnt = std::find_if( m_conts.begin(), m_conts.end(), [&]( const std::string& i ) { return i == cnt; } );
    ccnt      = std::distance( m_conts.begin(), icnt );
    if ( icnt == m_conts.end() ) m_conts.push_back( cnt.to_string() );
  }

  int clnk = -1;
  if ( !name.empty() ) {
    auto ilnk = std::find_if( m_links.begin(), m_links.end(), [&]( const std::string& i ) { return i == name; } );
    clnk      = std::distance( m_links.begin(), ilnk );
    if ( ilnk == m_links.end() ) m_links.push_back( name.to_string() );
  }

  ref.dbase     = cdb;
  ref.container = ccnt;
  ref.link      = clnk;
  ref.clid      = clid;
  ref.svc       = tech;
  if ( ref.svc == POOL_ROOT_StorageType || ref.svc == POOL_ROOTKEY_StorageType ||
       ref.svc == POOL_ROOTTREE_StorageType ) {
    ref.svc = ROOT_StorageType;
  }
}

RootDataConnection::Tool * RootDataConnection::makeTool

(

)

Create file access tool to encapsulate POOL compatibiliy.

Definition at line 212 of file RootDataConnection.cpp.

                                                     {
  if ( !m_refs ) m_refs = (TTree*)m_file->Get( "Refs" );
  if ( m_refs ) m_tool.reset( new RootTool( this ) );
#ifdef __POOL_COMPATIBILITY
  else if ( m_file->Get( "##Links" ) != nullptr )
    m_tool.reset( new PoolTool( this ) );
#endif
  else
    m_tool.reset();
  return m_tool.get();
}

const StringVec& Gaudi::RootDataConnection::mergeFIDs ( ) const

inline

Access merged FIDs.

Definition at line 253 of file RootDataConnection.h.

{ return m_mergeFIDs; }

const MergeSections& Gaudi::RootDataConnection::mergeSections ( ) const

inline

Access merged data section inventory.

Definition at line 251 of file RootDataConnection.h.

{ return m_mergeSects; }

MsgStream& Gaudi::RootDataConnection::msgSvc ( ) const

inline

Allow access to printer service.

Definition at line 144 of file RootDataConnection.h.

{ return m_setup->msgSvc(); }

StatusCode Gaudi::RootDataConnection::read ( void *  const, size_t    )

inlineoverridevirtual

Read root byte buffer from input stream.

Implements Gaudi::IDataConnection.

Definition at line 293 of file RootDataConnection.h.

{ return StatusCode::FAILURE; }

size_t RootDataConnection::removeClient

(

const IInterface *

client

)

Remove client from this data source.

Definition at line 171 of file RootDataConnection.cpp.

                                                                  {
  auto i = m_clients.find( client );
  if ( i != m_clients.end() ) m_clients.erase( i );
  return m_clients.size();
}

pair< int, unsigned long > RootDataConnection::save	(	boost::string_ref	section,
		boost::string_ref	cnt,
		TClass *	cl,
		void *	pObj,
		int	buff_siz,
		int	split_lvl,
		bool	fill_missing = false
	)

Save object of a given class to section and container.

Definition at line 466 of file RootDataConnection.cpp.

                                                                                                                {
  split_lvl  = 0;
  TBranch* b = getBranch( section, cnt, cl, pObj ? &pObj : nullptr, buff_siz, split_lvl );
  if ( b ) {
    Long64_t evt = b->GetEntries();
    // msgSvc() << MSG::DEBUG << cnt.c_str() << " Obj:" << (void*)pObj
    //         << " Split:" << split_lvl << " Buffer size:" << buff_siz << endl;
    if ( fill_missing ) {
      Long64_t num, nevt = b->GetTree()->GetEntries();
      if ( nevt > evt ) {
        b->SetAddress( nullptr );
        num = nevt - evt;
        while ( num > 0 ) {
          b->Fill();
          --num;
        }
        msgSvc() << MSG::DEBUG << "Added " << long( nevt - evt ) << " / Tree: " << nevt
                 << " / Branch: " << b->GetEntries() + 1 << " NULL entries to:" << cnt << endmsg;
        evt = b->GetEntries();
      }
    }
    b->SetAddress( &pObj );
    return {b->Fill(), evt};
  }
  if ( pObj ) { msgSvc() << MSG::ERROR << "Failed to access branch " << m_name << "/" << cnt << endmsg; }
  return {-1, ~0};
}

pair< int, unsigned long > RootDataConnection::saveObj	(	boost::string_ref	section,
		boost::string_ref	cnt,
		TClass *	cl,
		DataObject *	pObj,
		int	buff_siz,
		int	split_lvl,
		bool	fill_missing = false
	)

Save object of a given class to section and container.

Definition at line 459 of file RootDataConnection.cpp.

                                                                                                                 {
  DataObjectPush push( pObj );
  return save( section, cnt, cl, pObj, buff_siz, split_lvl, fill );
}

StatusCode Gaudi::RootDataConnection::saveRefs ( )

protected

Internal helper to save/update reference tables.

void RootDataConnection::saveStatistics

(

boost::string_ref

statisticsFile

)

Save TTree access statistics if required.

Definition at line 189 of file RootDataConnection.cpp.

                                                                        {
  if ( m_statistics ) {
    m_statistics->Print();
    if ( !statisticsFile.empty() ) m_statistics->SaveAs( statisticsFile.to_string().c_str() );
    m_statistics.reset();
  }
}

long long int Gaudi::RootDataConnection::seek ( long long int  , int    )

inlineoverridevirtual

Seek on the file described by ioDesc. Arguments as in ::seek()

Implements Gaudi::IDataConnection.

Definition at line 297 of file RootDataConnection.h.

{ return -1; }

Tool* Gaudi::RootDataConnection::tool ( ) const

inline

Access tool.

Definition at line 249 of file RootDataConnection.h.

{ return m_tool.get(); }

StatusCode Gaudi::RootDataConnection::write ( const void *  , int    )

inlineoverridevirtual

Write root byte buffer to output stream.

Implements Gaudi::IDataConnection.

Definition at line 295 of file RootDataConnection.h.

{ return StatusCode::FAILURE; }

Friends And Related Function Documentation

friend class Tool

friend

Definition at line 233 of file RootDataConnection.h.

Member Data Documentation

Clients Gaudi::RootDataConnection::m_clients

protected

Client list.

Definition at line 172 of file RootDataConnection.h.

StringVec Gaudi::RootDataConnection::m_conts

protected

Map containing external container names.

Definition at line 160 of file RootDataConnection.h.

StringVec Gaudi::RootDataConnection::m_dbs

protected

Map containing external database file names (fids)

Definition at line 158 of file RootDataConnection.h.

std::string Gaudi::RootDataConnection::m_empty

protected

Buffer for empty string reference.

Definition at line 174 of file RootDataConnection.h.

std::unique_ptr<TFile> Gaudi::RootDataConnection::m_file

protected

Reference to ROOT file.

Definition at line 152 of file RootDataConnection.h.

StringVec Gaudi::RootDataConnection::m_links

protected

Map containing internal links names.

Definition at line 162 of file RootDataConnection.h.

LinkSections Gaudi::RootDataConnection::m_linkSects

protected

Database link sections.

Definition at line 170 of file RootDataConnection.h.

StringVec Gaudi::RootDataConnection::m_mergeFIDs

protected

Map containing merge FIDs.

Definition at line 164 of file RootDataConnection.h.

MergeSections Gaudi::RootDataConnection::m_mergeSects

protected

Database section map for merged files.

Definition at line 168 of file RootDataConnection.h.

ParamMap Gaudi::RootDataConnection::m_params

protected

Parameter map for file parameters.

Definition at line 166 of file RootDataConnection.h.

TTree* Gaudi::RootDataConnection::m_refs = nullptr

protected

Pointer to the reference tree.

Definition at line 154 of file RootDataConnection.h.

Sections Gaudi::RootDataConnection::m_sections

protected

Tree sections in TFile.

Definition at line 156 of file RootDataConnection.h.

std::shared_ptr<RootConnectionSetup> Gaudi::RootDataConnection::m_setup

protected

Reference to the setup structure.

Definition at line 148 of file RootDataConnection.h.

std::unique_ptr<TTreePerfStats> Gaudi::RootDataConnection::m_statistics

protected

I/O read statistics from TTree.

Definition at line 150 of file RootDataConnection.h.

std::unique_ptr<Tool> Gaudi::RootDataConnection::m_tool

Definition at line 232 of file RootDataConnection.h.

---

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

• RootCnv/RootCnv/RootDataConnection.h
• RootCnv/src/RootDataConnection.cpp