The Gaudi Framework  v29r0 (ff2e7097)
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  { ROOT_READ_ERROR = 0x2, ROOT_OPEN_ERROR = 0x4 }
 
typedef std::vector< std::stringStringVec
 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 * > Sections
 Definition of tree sections. More...
 
typedef std::vector< ContainerSectionContainerSections
 Definition of container sections to handle merged files. More...
 
typedef std::map< std::string, ContainerSectionsMergeSections
 Definition of database section to handle merged files. More...
 
typedef std::vector< RootRefLinkSections
 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

MsgStreammsgSvc () const
 Allow access to printer service. More...
 
ToolmakeTool ()
 Create file access tool to encapsulate POOL compatibiliy. More...
 
 RootDataConnection (const IInterface *own, const std::string &nam, std::shared_ptr< RootConnectionSetup > setup)
 Standard constructor. More...
 
 ~RootDataConnection () override=default
 Standard destructor. 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...
 
Tooltool () const
 Access tool. More...
 
const MergeSectionsmergeSections () const
 Access merged data section inventory. More...
 
const StringVecmergeFIDs () 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 (const std::string &msg) const
 Error handler when bad write statements occur. More...
 
std::pair< const RootRef *, const ContainerSection * > getMergeSection (const std::string &container, int entry) const
 Access link section for single container and entry. More...
 
void enableStatistics (const std::string &section)
 Enable TTreePerStats. More...
 
void saveStatistics (const std::string &statisticsFile)
 Save TTree access statistics if required. More...
 
int loadObj (const std::string &section, const std::string &cnt, unsigned long entry, DataObject *&pObj)
 Load object. More...
 
int loadRefs (const std::string &section, const std::string &cnt, unsigned long entry, RootObjectRefs &refs)
 Load references object. More...
 
std::pair< int, unsigned long > saveObj (const std::string &section, const std::string &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 (const std::string &section, const std::string &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 (const std::string &sect, bool create=false)
 Access TTree section from section name. The section is created if required. More...
 
TBranch * getBranch (const std::string &section, const std::string &branch_name)
 Access data branch by name: Get existing branch in read only mode. More...
 
TBranch * getBranch (const std::string &section, const std::string &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 (IRegistry *pA, RootRef &ref)
 Create reference object from registry entry. More...
 
void makeRef (const std::string &name, long clid, int tech, const std::string &db, const std::string &cnt, int entry, RootRef &ref)
 Create reference object from values. More...
 
int makeLink (const std::string &p)
 Convert path string to path index. More...
 
const std::stringgetDb (int which) const
 Access database/file name from saved index. More...
 
const std::stringgetCont (int which) const
 Access container name from saved index. More...
 
const std::stringgetLink (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::stringname () const
 Connection name. More...
 
void setFID (std::string fid)
 Set file ID. More...
 
const std::stringfid () const
 Access file id. More...
 
const std::stringpfn () 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 IInterfaceowner () const
 Owner instance. More...
 

Public Attributes

std::unique_ptr< Toolm_tool
 

Protected Member Functions

const std::stringempty () const
 Empty string reference. More...
 
StatusCode saveRefs ()
 Internal helper to save/update reference tables. More...
 

Protected Attributes

std::shared_ptr< RootConnectionSetupm_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 IInterfacem_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 94 of file RootDataConnection.h.

Member Typedef Documentation

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

Client set.

Definition at line 144 of file RootDataConnection.h.

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

Definition of container sections to handle merged files.

Definition at line 138 of file RootDataConnection.h.

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

Link sections definition.

Definition at line 142 of file RootDataConnection.h.

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

Definition of database section to handle merged files.

Definition at line 140 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 134 of file RootDataConnection.h.

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

Definition of tree sections.

Definition at line 136 of file RootDataConnection.h.

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

Type definition for string maps.

Definition at line 132 of file RootDataConnection.h.

Member Enumeration Documentation

anonymous enum
Enumerator
ROOT_READ_ERROR 
ROOT_OPEN_ERROR 

Definition at line 97 of file RootDataConnection.h.

Constructor & Destructor Documentation

RootDataConnection::RootDataConnection ( const IInterface own,
const std::string nam,
std::shared_ptr< RootConnectionSetup setup 
)

Standard constructor.

Definition at line 138 of file RootDataConnection.cpp.

140  : IDataConnection( owner, fname ), m_setup( std::move( setup ) )
141 { // 01234567890123456789012345678901234567890
142  // Check if FID: A82A3BD8-7ECB-DC11-8DC0-000423D950B0
143  if ( fname.length() == 36 && fname[8] == '-' && fname[13] == '-' && fname[18] == '-' && fname[23] == '-' ) {
144  m_name = "FID:" + fname;
145  }
146  m_age = 0;
147  m_file.reset();
148  addClient( owner );
149 }
void addClient(const IInterface *client)
Add new client to this data source.
int m_age
Age counter.
std::string m_name
Connection name/identifier.
IDataConnection(const IInterface *own, std::string nam)
Standard constructor.
T reset(T...args)
T move(T...args)
const IInterface * owner() const
Owner instance.
std::shared_ptr< RootConnectionSetup > m_setup
Reference to the setup structure.
std::unique_ptr< TFile > m_file
Reference to ROOT file.
Gaudi::RootDataConnection::~RootDataConnection ( )
overridedefault

Standard destructor.

Member Function Documentation

void RootDataConnection::addClient ( const IInterface client)

Add new client to this data source.

Definition at line 152 of file RootDataConnection.cpp.

152 { m_clients.insert( client ); }
Clients m_clients
Client list.
T insert(T...args)
void RootDataConnection::badWriteError ( const std::string msg) const

Error handler when bad write statements occur.

Definition at line 170 of file RootDataConnection.cpp.

171 {
172  msgSvc() << MSG::ERROR << "File:" << fid() << "Failed action:" << msg << endmsg;
173 }
const std::string & fid() const
Access file id.
MsgStream & msgSvc() const
Allow access to printer service.
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:209
StatusCode RootDataConnection::connectRead ( )
overridevirtual

Open data stream in read mode.

Connect the file in READ mode.

Implements Gaudi::IDataConnection.

Definition at line 215 of file RootDataConnection.cpp.

216 {
217  m_file.reset( TFile::Open( m_pfn.c_str() ) );
218  if ( !m_file || m_file->IsZombie() ) {
219  m_file.reset();
220  return StatusCode::FAILURE;
221  }
223  msgSvc() << MSG::DEBUG << "Opened file " << m_pfn << " in mode READ. [" << m_fid << "]" << endmsg << MSG::DEBUG;
224  if ( msgSvc().isActive() ) m_file->ls();
225  msgSvc() << MSG::VERBOSE;
226  if ( msgSvc().isActive() ) m_file->Print();
227  if ( makeTool() ) {
228  sc = m_tool->readRefs();
229  sc.ignore();
230 #if ROOT_VERSION_CODE >= ROOT_VERSION( 5, 33, 0 )
231  if ( sc.getCode() == ROOT_READ_ERROR ) {
232  IIncidentSvc* inc = m_setup->incidentSvc();
233  if ( inc ) {
234  inc->fireIncident( Incident( pfn(), IncidentType::CorruptedInputFile ) );
235  }
236  }
237 #endif
238  }
239  if ( !sc.isSuccess() ) return sc;
240  bool need_fid = m_fid == m_pfn;
241  string fid = m_fid;
242  m_mergeFIDs.clear();
243  for ( auto& elem : m_params ) {
244  if ( elem.first == "FID" ) {
245  m_mergeFIDs.push_back( elem.second );
246  if ( elem.second != m_fid ) {
247  msgSvc() << MSG::DEBUG << "Check FID param:" << elem.second << endmsg;
248  // if ( m_fid == m_pfn ) {
249  m_fid = elem.second;
250  //}
251  }
252  }
253  }
254  if ( !need_fid && fid != m_fid ) {
255  msgSvc() << MSG::ERROR << "FID mismatch:" << fid << "(Catalog) != " << m_fid << "(file)" << endmsg
256  << "for PFN:" << m_pfn << endmsg;
257  return StatusCode::FAILURE;
258  }
259  msgSvc() << MSG::DEBUG << "Using FID " << m_fid << " from params table...." << endmsg << "for PFN:" << m_pfn
260  << endmsg;
261  return sc;
262 }
const std::string & fid() const
Access file id.
std::unique_ptr< Tool > m_tool
unsigned long getCode() const
Get the status code by value.
Definition: StatusCode.h:94
std::string m_fid
File ID of the connection.
bool isSuccess() const
Test for a status code of SUCCESS.
Definition: StatusCode.h:75
const std::string & pfn() const
Access physical file name.
ParamMap m_params
Parameter map for file parameters.
StringVec m_mergeFIDs
Map containing merge FIDs.
std::string m_pfn
Physical file name of the connection.
T push_back(T...args)
virtual void fireIncident(const Incident &incident)=0
Fire an Incident.
This class is used for returning status codes from appropriate routines.
Definition: StatusCode.h:28
T reset(T...args)
T clear(T...args)
Tool * makeTool()
Create file access tool to encapsulate POOL compatibiliy.
std::shared_ptr< RootConnectionSetup > m_setup
Reference to the setup structure.
T c_str(T...args)
Base class for all Incidents (computing events).
Definition: Incident.h:17
MsgStream & msgSvc() const
Allow access to printer service.
std::unique_ptr< TFile > m_file
Reference to ROOT file.
void ignore() const
Definition: StatusCode.h:109
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:209
The interface implemented by the IncidentSvc service.
Definition: IIncidentSvc.h:23
StatusCode RootDataConnection::connectWrite ( IoType  typ)
overridevirtual

Open data stream in write mode.

Implements Gaudi::IDataConnection.

Definition at line 265 of file RootDataConnection.cpp.

266 {
267  int compress = RootConnectionSetup::compression();
268  msgSvc() << MSG::DEBUG;
269  switch ( typ ) {
270  case CREATE:
271  resetAge();
272  m_file.reset( TFile::Open( m_pfn.c_str(), "CREATE", "Root event data", compress ) );
273  m_refs = new TTree( "Refs", "Root reference data" );
274  msgSvc() << "Opened file " << m_pfn << " in mode CREATE. [" << m_fid << "]" << endmsg;
275  m_params.emplace_back( "PFN", m_pfn );
276  if ( m_fid != m_pfn ) {
277  m_params.emplace_back( "FID", m_fid );
278  }
279  makeTool();
280  break;
281  case RECREATE:
282  resetAge();
283  m_file.reset( TFile::Open( m_pfn.c_str(), "RECREATE", "Root event data", compress ) );
284  msgSvc() << "Opened file " << m_pfn << " in mode RECREATE. [" << m_fid << "]" << endmsg;
285  m_refs = new TTree( "Refs", "Root reference data" );
286  m_params.emplace_back( "PFN", m_pfn );
287  if ( m_fid != m_pfn ) {
288  m_params.emplace_back( "FID", m_fid );
289  }
290  makeTool();
291  break;
292  case UPDATE:
293  resetAge();
294  m_file.reset( TFile::Open( m_pfn.c_str(), "UPDATE", "Root event data", compress ) );
295  msgSvc() << "Opened file " << m_pfn << " in mode UPDATE. [" << m_fid << "]" << endmsg;
296  if ( m_file && !m_file->IsZombie() ) {
297  if ( makeTool() ) {
298  StatusCode sc = m_tool->readRefs();
299  sc.ignore();
300  if ( sc.getCode() == ROOT_READ_ERROR ) {
301 #if ROOT_VERSION_CODE >= ROOT_VERSION( 5, 33, 0 )
302  IIncidentSvc* inc = m_setup->incidentSvc();
303  if ( inc ) {
304  inc->fireIncident( Incident( pfn(), IncidentType::CorruptedInputFile ) );
305  }
306 #endif
307  }
308  return sc;
309  }
310  TDirectory::TContext ctxt( m_file.get() );
311  m_refs = new TTree( "Refs", "Root reference data" );
312  makeTool();
313  return StatusCode::SUCCESS;
314  }
315  break;
316  default:
317  m_refs = nullptr;
318  m_file.reset();
319  return StatusCode::FAILURE;
320  }
322 }
std::unique_ptr< Tool > m_tool
unsigned long getCode() const
Get the status code by value.
Definition: StatusCode.h:94
std::string m_fid
File ID of the connection.
static int compression()
Access to global compression level.
const std::string & pfn() const
Access physical file name.
ParamMap m_params
Parameter map for file parameters.
std::string m_pfn
Physical file name of the connection.
virtual void fireIncident(const Incident &incident)=0
Fire an Incident.
This class is used for returning status codes from appropriate routines.
Definition: StatusCode.h:28
T reset(T...args)
Tool * makeTool()
Create file access tool to encapsulate POOL compatibiliy.
T get(T...args)
std::shared_ptr< RootConnectionSetup > m_setup
Reference to the setup structure.
void resetAge()
Reset age.
T c_str(T...args)
Base class for all Incidents (computing events).
Definition: Incident.h:17
MsgStream & msgSvc() const
Allow access to printer service.
std::unique_ptr< TFile > m_file
Reference to ROOT file.
void ignore() const
Definition: StatusCode.h:109
TTree * m_refs
Pointer to the reference tree.
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:209
The interface implemented by the IncidentSvc service.
Definition: IIncidentSvc.h:23
T emplace_back(T...args)
StatusCode RootDataConnection::disconnect ( )
overridevirtual

Release data stream and release implementation dependent resources.

Implements Gaudi::IDataConnection.

Definition at line 325 of file RootDataConnection.cpp.

326 {
327  if ( m_file ) {
328  if ( !m_file->IsZombie() ) {
329  if ( m_file->IsWritable() ) {
330  msgSvc() << MSG::DEBUG;
331  TDirectory::TContext ctxt( m_file.get() );
332  if ( m_refs ) {
333  if ( !m_tool->saveRefs().isSuccess() ) badWriteError( "Saving References" );
334  if ( m_refs->Write() < 0 ) badWriteError( "Write Reference branch" );
335  }
336  for ( auto& i : m_sections ) {
337  if ( i.second ) {
338  if ( i.second->Write() < 0 ) badWriteError( "Write section:" + i.first );
339  msgSvc() << "Disconnect section " << i.first << " " << i.second->GetName() << endmsg;
340  }
341  }
342  m_sections.clear();
343  }
344  msgSvc() << MSG::DEBUG;
345  if ( msgSvc().isActive() ) m_file->ls();
346  msgSvc() << MSG::VERBOSE;
347  if ( msgSvc().isActive() ) m_file->Print();
348  m_file->Close();
349  }
350  msgSvc() << MSG::DEBUG << "Disconnected file " << m_pfn << " " << m_file->GetName() << endmsg;
351  m_file.reset();
352  m_tool.reset();
353  }
354  return StatusCode::SUCCESS;
355 }
std::unique_ptr< Tool > m_tool
std::string m_pfn
Physical file name of the connection.
void clear(STATE_TYPE _i=std::ios_base::failbit)
Definition: MsgStream.h:187
Sections m_sections
Tree sections in TFile.
T reset(T...args)
T get(T...args)
MsgStream & msgSvc() const
Allow access to printer service.
std::unique_ptr< TFile > m_file
Reference to ROOT file.
TTree * m_refs
Pointer to the reference tree.
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:209
void badWriteError(const std::string &msg) const
Error handler when bad write statements occur.
CSTR RootDataConnection::empty ( ) const
protected

Empty string reference.

Definition at line 460 of file RootDataConnection.cpp.

460 { return s_empty; }
void RootDataConnection::enableStatistics ( const std::string section)

Enable TTreePerStats.

Definition at line 186 of file RootDataConnection.cpp.

187 {
188  if ( m_statistics ) {
189  TTree* t = getSection( section, false );
190  if ( t ) {
191  m_statistics.reset( new TTreePerfStats( ( section + "_ioperf" ).c_str(), t ) );
192  return;
193  }
194  msgSvc() << MSG::WARNING << "Failed to enable perfstats for tree:" << section << endmsg;
195  return;
196  }
197  msgSvc() << MSG::INFO << "Perfstats are ALREADY ENABLED." << endmsg;
198 }
std::unique_ptr< TTreePerfStats > m_statistics
I/O read statistics from TTree.
TTree * getSection(const std::string &sect, bool create=false)
Access TTree section from section name. The section is created if required.
T reset(T...args)
MsgStream & msgSvc() const
Allow access to printer service.
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:209
TFile* Gaudi::RootDataConnection::file ( ) const
inline

Direct access to TFile structure.

Definition at line 249 of file RootDataConnection.h.

249 { return m_file.get(); }
T get(T...args)
std::unique_ptr< TFile > m_file
Reference to ROOT file.
TBranch* Gaudi::RootDataConnection::getBranch ( const std::string section,
const std::string branch_name 
)
inline

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

Definition at line 309 of file RootDataConnection.h.

310  {
311  return m_tool->getBranch( section, branch_name );
312  }
std::unique_ptr< Tool > m_tool
TBranch* Gaudi::RootDataConnection::getBranch ( const std::string section,
const std::string branch_name,
TClass *  cl,
void *  ptr,
int  buff_siz,
int  split_lvl 
)

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

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

Access container name from saved index.

Definition at line 330 of file RootDataConnection.h.

331  {
332  return ( which >= 0 ) && ( size_t( which ) < m_conts.size() ) ? *( m_conts.begin() + which ) : empty();
333  }
StringVec m_conts
Map containing external container names.
const std::string & empty() const
Empty string reference.
T size(T...args)
T begin(T...args)
def which(executable)
Definition: BaseTest.py:570
CSTR RootDataConnection::getDb ( int  which) const

Access database/file name from saved index.

Definition at line 450 of file RootDataConnection.cpp.

451 {
452  if ( ( which >= 0 ) && ( size_t( which ) < m_dbs.size() ) ) {
453  if ( *( m_dbs.begin() + which ) == s_local ) return m_fid;
454  return *( m_dbs.begin() + which );
455  }
456  return s_empty;
457 }
std::string m_fid
File ID of the connection.
T size(T...args)
T begin(T...args)
def which(executable)
Definition: BaseTest.py:570
StringVec m_dbs
Map containing external database file names (fids)
const std::string& Gaudi::RootDataConnection::getLink ( int  which) const
inline

Access link name from saved index.

Definition at line 336 of file RootDataConnection.h.

337  {
338  return ( which >= 0 ) && ( size_t( which ) < m_links.size() ) ? *( m_links.begin() + which ) : empty();
339  }
StringVec m_links
Map containing internal links names.
const std::string & empty() const
Empty string reference.
T size(T...args)
T begin(T...args)
def which(executable)
Definition: BaseTest.py:570
pair< const RootRef *, const RootDataConnection::ContainerSection * > RootDataConnection::getMergeSection ( const std::string container,
int  entry 
) const

Access link section for single container and entry.

Definition at line 578 of file RootDataConnection.cpp.

579 {
580  // size_t idx = cont.find('/',1);
581  // string container = cont[0]=='/' ? cont.substr(1,idx==string::npos?idx:idx-1) : cont;
582  auto i = m_mergeSects.find( container );
583  if ( i != m_mergeSects.end() ) {
584  size_t cnt = 0;
585  const ContainerSections& s = ( *i ).second;
586  for ( auto j = s.cbegin(); j != s.cend(); ++j, ++cnt ) {
587  const ContainerSection& c = *j;
588  if ( entry >= c.start && entry < ( c.start + c.length ) ) {
589  if ( m_linkSects.size() > cnt ) {
590  if ( msgSvc().isActive() ) {
591  msgSvc() << MSG::VERBOSE << "MergeSection for:" << container << " [" << entry << "]" << endmsg
592  << "FID:" << m_fid << " -> PFN:" << m_pfn << endmsg;
593  }
594  return {&( m_linkSects[cnt] ), &c};
595  }
596  }
597  }
598  }
599  msgSvc() << MSG::DEBUG << "Return INVALID MergeSection for:" << container << " [" << entry << "]" << endmsg
600  << "FID:" << m_fid << " -> PFN:" << m_pfn << endmsg;
601  return {nullptr, nullptr};
602 }
MergeSections m_mergeSects
Database section map for merged files.
std::vector< ContainerSection > ContainerSections
Definition of container sections to handle merged files.
std::string m_fid
File ID of the connection.
bool isActive() const
Accessor: is MsgStream active.
Definition: MsgStream.h:116
T end(T...args)
std::string m_pfn
Physical file name of the connection.
T find(T...args)
T size(T...args)
LinkSections m_linkSects
Database link sections.
MsgStream & msgSvc() const
Allow access to printer service.
string s
Definition: gaudirun.py:253
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:209
TTree * RootDataConnection::getSection ( const std::string sect,
bool  create = false 
)

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

Definition at line 358 of file RootDataConnection.cpp.

359 {
360  TTree* t = m_sections[section];
361  if ( !t ) {
362  t = (TTree*)m_file->Get( section.c_str() );
363  if ( !t && create ) {
364  TDirectory::TContext ctxt( m_file.get() );
365  t = new TTree( section.c_str(), "Root data for Gaudi" );
366  }
367  if ( t ) {
368  int cacheSize = m_setup->cacheSize;
369  if ( create ) {
370  // t->SetAutoFlush(100);
371  }
372  if ( section == m_setup->loadSection && cacheSize > -2 ) {
373  MsgStream& msg = msgSvc();
374  int learnEntries = m_setup->learnEntries;
375  t->SetCacheSize( cacheSize );
376  t->SetCacheLearnEntries( learnEntries );
377  msg << MSG::DEBUG;
378  if ( create ) {
379  msg << "Tree:" << section << "Setting up tree cache:" << cacheSize << endmsg;
380  } else {
381  const StringVec& vB = m_setup->vetoBranches;
382  const StringVec& cB = m_setup->cacheBranches;
383  msg << "Tree:" << section << " Setting up tree cache:" << cacheSize << " Add all branches." << endmsg;
384  msg << "Tree:" << section << " Learn for " << learnEntries << " entries." << endmsg;
385 
386  if ( cB.empty() && vB.empty() ) {
387  msg << "Adding (default) all branches to tree cache." << endmsg;
388  t->AddBranchToCache( "*", kTRUE );
389  }
390  if ( cB.size() == 1 && cB[0] == "*" ) {
391  msg << "Adding all branches to tree cache according to option \"CacheBranches\"." << endmsg;
392  t->AddBranchToCache( "*", kTRUE );
393  } else {
394  for ( TIter it( t->GetListOfBranches() ); it.Next(); ) {
395  const char* n = ( (TNamed*)( *it ) )->GetName();
396  bool add = false, veto = false;
397  for ( const auto& i : cB ) {
398  if ( !match_wild( n, ( i ).c_str() ) ) continue;
399  add = true;
400  break;
401  }
402  for ( auto i = vB.cbegin(); !add && i != vB.cend(); ++i ) {
403  if ( !match_wild( n, ( *i ).c_str() ) ) continue;
404  veto = true;
405  break;
406  }
407  if ( add && !veto ) {
408  msg << "Add " << n << " to branch cache." << endmsg;
409  t->AddBranchToCache( n, kTRUE );
410  } else {
411  msg << "Do not cache branch " << n << endmsg;
412  }
413  }
414  }
415  }
416  }
417  m_sections[section] = t;
418  }
419  }
420  return t;
421 }
Definition of the MsgStream class used to transmit messages.
Definition: MsgStream.h:24
std::vector< std::string > StringVec
Type definition for string maps.
Sections m_sections
Tree sections in TFile.
T get(T...args)
std::shared_ptr< RootConnectionSetup > m_setup
Reference to the setup structure.
MsgStream & msgSvc() const
Allow access to printer service.
std::unique_ptr< TFile > m_file
Reference to ROOT file.
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:209
bool Gaudi::RootDataConnection::isConnected ( ) const
inlineoverridevirtual

Check if connected to data source.

Implements Gaudi::IDataConnection.

Definition at line 251 of file RootDataConnection.h.

251 { return bool( m_file ); }
std::unique_ptr< TFile > m_file
Reference to ROOT file.
bool Gaudi::RootDataConnection::isWritable ( ) const
inline

Is the file writable?

Definition at line 253 of file RootDataConnection.h.

253 { return m_file && m_file->IsWritable(); }
std::unique_ptr< TFile > m_file
Reference to ROOT file.
int RootDataConnection::loadObj ( const std::string section,
const std::string cnt,
unsigned long  entry,
DataObject *&  pObj 
)

Load object.

Definition at line 504 of file RootDataConnection.cpp.

505 {
506  TBranch* b = getBranch( section, cnt );
507  if ( b ) {
508  TClass* cl = gROOT->GetClass( b->GetClassName(), kTRUE );
509  if ( cl ) {
510  int nb = -1;
511  pObj = (DataObject*)cl->New();
512  {
513  DataObjectPush push( pObj );
514  b->SetAddress( &pObj );
515  if ( section == m_setup->loadSection ) {
516  TTree* t = b->GetTree();
517  if ( Long64_t( entry ) != t->GetReadEntry() ) {
518  t->LoadTree( Long64_t( entry ) );
519  }
520  }
521  nb = b->GetEntry( entry );
522  msgSvc() << MSG::VERBOSE;
523  if ( msgSvc().isActive() ) {
524  msgSvc() << "Load [" << entry << "] --> " << section << ":" << cnt << " " << nb << " bytes." << endmsg;
525  }
526  if ( nb < 0 ) { // This is definitely an error...ROOT says if reads fail, -1 is issued.
527 #if ROOT_VERSION_CODE >= ROOT_VERSION( 5, 33, 0 )
528  IIncidentSvc* inc = m_setup->incidentSvc();
529  if ( inc ) {
530  inc->fireIncident( Incident( pfn(), IncidentType::CorruptedInputFile ) );
531  }
532 #endif
533  } else if ( nb == 0 && pObj->clID() == CLID_DataObject ) {
534  TFile* f = b->GetFile();
535  int vsn = f->GetVersion();
536  if ( vsn < 52400 ) {
537  // For Gaudi v21r5 (ROOT 5.24.00b) DataObject::m_version was not written!
538  // Still this call be well be successful.
539  nb = 1;
540  } else if ( vsn > 1000000 && ( vsn % 1000000 ) < 52400 ) {
541  // dto. Some POOL files have for unknown reasons a version
542  // not according to ROOT standards. Hack this explicitly.
543  nb = 1;
544  }
545  }
546  if ( nb < 0 ) {
547  delete pObj;
548  pObj = nullptr;
549  }
550  }
551  return nb;
552  }
553  }
554  return -1;
555 }
const std::string & pfn() const
Access physical file name.
virtual void fireIncident(const Incident &incident)=0
Fire an Incident.
virtual const CLID & clID() const
Retrieve reference to class definition structure.
Definition: DataObject.cpp:62
TBranch * getBranch(const std::string &section, const std::string &branch_name)
Access data branch by name: Get existing branch in read only mode.
std::shared_ptr< RootConnectionSetup > m_setup
Reference to the setup structure.
Base class for all Incidents (computing events).
Definition: Incident.h:17
MsgStream & msgSvc() const
Allow access to printer service.
A DataObject is the base class of any identifiable object on any data store.
Definition: DataObject.h:29
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:209
The interface implemented by the IncidentSvc service.
Definition: IIncidentSvc.h:23
int RootDataConnection::loadRefs ( const std::string section,
const std::string cnt,
unsigned long  entry,
RootObjectRefs refs 
)

Load references object.

Definition at line 558 of file RootDataConnection.cpp.

560 {
561  int nbytes = m_tool->loadRefs( section, cnt, entry, refs );
562 #if ROOT_VERSION_CODE >= ROOT_VERSION( 5, 33, 0 )
563  if ( nbytes < 0 ) {
564  // This is definitely an error:
565  // -- Either branch not preesent at all or
566  // -- ROOT I/O error, which issues -1
567  IIncidentSvc* inc = m_setup->incidentSvc();
568  if ( inc ) {
569  inc->fireIncident( Incident( pfn(), IncidentType::CorruptedInputFile ) );
570  }
571  }
572 #endif
573  return nbytes;
574 }
std::unique_ptr< Tool > m_tool
const std::string & pfn() const
Access physical file name.
virtual void fireIncident(const Incident &incident)=0
Fire an Incident.
std::shared_ptr< RootConnectionSetup > m_setup
Reference to the setup structure.
Base class for all Incidents (computing events).
Definition: Incident.h:17
The interface implemented by the IncidentSvc service.
Definition: IIncidentSvc.h:23
bool RootDataConnection::lookupClient ( const IInterface client) const

Lookup client for this data source.

Definition at line 163 of file RootDataConnection.cpp.

164 {
165  auto i = m_clients.find( client );
166  return i != m_clients.end();
167 }
Clients m_clients
Client list.
T end(T...args)
T find(T...args)
int RootDataConnection::makeLink ( const std::string p)

Convert path string to path index.

Definition at line 441 of file RootDataConnection.cpp.

442 {
443  auto ip = std::find( std::begin( m_links ), std::end( m_links ), p );
444  if ( ip != std::end( m_links ) ) return std::distance( std::begin( m_links ), ip );
445  m_links.push_back( p );
446  return m_links.size() - 1;
447 }
T distance(T...args)
T end(T...args)
StringVec m_links
Map containing internal links names.
T push_back(T...args)
T find(T...args)
T size(T...args)
T begin(T...args)
void RootDataConnection::makeRef ( IRegistry pA,
RootRef ref 
)

Create reference object from registry entry.

Definition at line 605 of file RootDataConnection.cpp.

606 {
607  IOpaqueAddress* pA = pR->address();
608  makeRef( pR->name(), pA->clID(), pA->svcType(), pA->par()[0], pA->par()[1], -1, ref );
609 }
virtual const CLID & clID() const =0
Retrieve class information from link.
virtual const std::string * par() const =0
Retrieve String parameters.
void makeRef(IRegistry *pA, RootRef &ref)
Create reference object from registry entry.
virtual long svcType() const =0
Retrieve service type.
Opaque address interface definition.
virtual IOpaqueAddress * address() const =0
Retrieve opaque storage address.
void Gaudi::RootDataConnection::makeRef ( const std::string name,
long  clid,
int  tech,
const std::string db,
const std::string cnt,
int  entry,
RootRef ref 
)

Create reference object from values.

RootDataConnection::Tool * RootDataConnection::makeTool ( )

Create file access tool to encapsulate POOL compatibiliy.

Definition at line 201 of file RootDataConnection.cpp.

202 {
203  if ( !m_refs ) m_refs = (TTree*)m_file->Get( "Refs" );
204  if ( m_refs ) m_tool.reset( new RootTool( this ) );
205 #ifdef __POOL_COMPATIBILITY
206  else if ( m_file->Get( "##Links" ) != nullptr )
207  m_tool.reset( new PoolTool( this ) );
208 #endif
209  else
210  m_tool.reset();
211  return m_tool.get();
212 }
std::unique_ptr< Tool > m_tool
Description:
Definition: RootTool.h:17
std::unique_ptr< TFile > m_file
Reference to ROOT file.
TTree * m_refs
Pointer to the reference tree.
const StringVec& Gaudi::RootDataConnection::mergeFIDs ( ) const
inline

Access merged FIDs.

Definition at line 259 of file RootDataConnection.h.

259 { return m_mergeFIDs; }
StringVec m_mergeFIDs
Map containing merge FIDs.
const MergeSections& Gaudi::RootDataConnection::mergeSections ( ) const
inline

Access merged data section inventory.

Definition at line 257 of file RootDataConnection.h.

257 { return m_mergeSects; }
MergeSections m_mergeSects
Database section map for merged files.
MsgStream& Gaudi::RootDataConnection::msgSvc ( ) const
inline

Allow access to printer service.

Definition at line 147 of file RootDataConnection.h.

147 { return m_setup->msgSvc(); }
std::shared_ptr< RootConnectionSetup > m_setup
Reference to the setup structure.
StatusCode Gaudi::RootDataConnection::read ( void *  const,
size_t   
)
inlineoverridevirtual

Read root byte buffer from input stream.

Implements Gaudi::IDataConnection.

Definition at line 299 of file RootDataConnection.h.

size_t RootDataConnection::removeClient ( const IInterface client)

Remove client from this data source.

Definition at line 155 of file RootDataConnection.cpp.

156 {
157  auto i = m_clients.find( client );
158  if ( i != m_clients.end() ) m_clients.erase( i );
159  return m_clients.size();
160 }
Clients m_clients
Client list.
T end(T...args)
T erase(T...args)
T find(T...args)
T size(T...args)
pair< int, unsigned long > RootDataConnection::save ( const std::string section,
const std::string 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 471 of file RootDataConnection.cpp.

473 {
474  split_lvl = 0;
475  TBranch* b = getBranch( section, cnt, cl, pObj ? &pObj : nullptr, buff_siz, split_lvl );
476  if ( b ) {
477  Long64_t evt = b->GetEntries();
478  // msgSvc() << MSG::DEBUG << cnt.c_str() << " Obj:" << (void*)pObj
479  // << " Split:" << split_lvl << " Buffer size:" << buff_siz << endl;
480  if ( fill_missing ) {
481  Long64_t num, nevt = b->GetTree()->GetEntries();
482  if ( nevt > evt ) {
483  b->SetAddress( nullptr );
484  num = nevt - evt;
485  while ( num > 0 ) {
486  b->Fill();
487  --num;
488  }
489  msgSvc() << MSG::DEBUG << "Added " << long( nevt - evt ) << " / Tree: " << nevt
490  << " / Branch: " << b->GetEntries() + 1 << " NULL entries to:" << cnt << endmsg;
491  evt = b->GetEntries();
492  }
493  }
494  b->SetAddress( &pObj );
495  return {b->Fill(), evt};
496  }
497  if ( pObj ) {
498  msgSvc() << MSG::ERROR << "Failed to access branch " << m_name << "/" << cnt << endmsg;
499  }
500  return {-1, ~0};
501 }
std::string m_name
Connection name/identifier.
TBranch * getBranch(const std::string &section, const std::string &branch_name)
Access data branch by name: Get existing branch in read only mode.
MsgStream & msgSvc() const
Allow access to printer service.
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:209
evt
Definition: IOTest.py:96
pair< int, unsigned long > RootDataConnection::saveObj ( const std::string section,
const std::string 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 463 of file RootDataConnection.cpp.

465 {
466  DataObjectPush push( pObj );
467  return save( section, cnt, cl, pObj, buff_siz, split_lvl, fill );
468 }
GAUDI_API void fill(AIDA::IHistogram1D *histo, const double value, const double weight=1.0)
simple function to fill AIDA::IHistogram1D objects
Definition: Fill.cpp:35
std::pair< int, unsigned long > save(const std::string &section, const std::string &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.
StatusCode Gaudi::RootDataConnection::saveRefs ( )
protected

Internal helper to save/update reference tables.

void RootDataConnection::saveStatistics ( const std::string statisticsFile)

Save TTree access statistics if required.

Definition at line 176 of file RootDataConnection.cpp.

177 {
178  if ( m_statistics ) {
179  m_statistics->Print();
180  if ( !statisticsFile.empty() ) m_statistics->SaveAs( statisticsFile.c_str() );
182  }
183 }
T empty(T...args)
std::unique_ptr< TTreePerfStats > m_statistics
I/O read statistics from TTree.
T reset(T...args)
T c_str(T...args)
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 303 of file RootDataConnection.h.

303 { return -1; }
Tool* Gaudi::RootDataConnection::tool ( ) const
inline

Access tool.

Definition at line 255 of file RootDataConnection.h.

255 { return m_tool.get(); }
std::unique_ptr< Tool > m_tool
StatusCode Gaudi::RootDataConnection::write ( const void *  ,
int   
)
inlineoverridevirtual

Write root byte buffer to output stream.

Implements Gaudi::IDataConnection.

Definition at line 301 of file RootDataConnection.h.

Friends And Related Function Documentation

friend class Tool
friend

Definition at line 237 of file RootDataConnection.h.

Member Data Documentation

Clients Gaudi::RootDataConnection::m_clients
protected

Client list.

Definition at line 175 of file RootDataConnection.h.

StringVec Gaudi::RootDataConnection::m_conts
protected

Map containing external container names.

Definition at line 163 of file RootDataConnection.h.

StringVec Gaudi::RootDataConnection::m_dbs
protected

Map containing external database file names (fids)

Definition at line 161 of file RootDataConnection.h.

std::string Gaudi::RootDataConnection::m_empty
protected

Buffer for empty string reference.

Definition at line 177 of file RootDataConnection.h.

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

Reference to ROOT file.

Definition at line 155 of file RootDataConnection.h.

StringVec Gaudi::RootDataConnection::m_links
protected

Map containing internal links names.

Definition at line 165 of file RootDataConnection.h.

LinkSections Gaudi::RootDataConnection::m_linkSects
protected

Database link sections.

Definition at line 173 of file RootDataConnection.h.

StringVec Gaudi::RootDataConnection::m_mergeFIDs
protected

Map containing merge FIDs.

Definition at line 167 of file RootDataConnection.h.

MergeSections Gaudi::RootDataConnection::m_mergeSects
protected

Database section map for merged files.

Definition at line 171 of file RootDataConnection.h.

ParamMap Gaudi::RootDataConnection::m_params
protected

Parameter map for file parameters.

Definition at line 169 of file RootDataConnection.h.

TTree* Gaudi::RootDataConnection::m_refs = nullptr
protected

Pointer to the reference tree.

Definition at line 157 of file RootDataConnection.h.

Sections Gaudi::RootDataConnection::m_sections
protected

Tree sections in TFile.

Definition at line 159 of file RootDataConnection.h.

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

Reference to the setup structure.

Definition at line 151 of file RootDataConnection.h.

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

I/O read statistics from TTree.

Definition at line 153 of file RootDataConnection.h.

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

Definition at line 236 of file RootDataConnection.h.


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