PoolTool.h

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2025 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#pragma once
 
#include <GaudiKernel/Incident.h>
#include <RootCnv/PoolClasses.h>
#include <algorithm>
/*
 *  Gaudi namespace declaration
 */
namespace Gaudi {

  class PoolTool : virtual public RootDataConnection::Tool {
    std::vector<Gaudi::RootRef> m_poolLinks;
 
  public:
    PoolTool( RootDataConnection* con ) { c = con; }

    string _treeName( std::string_view sr ) {
      std::string t{ sr };
      std::replace( begin( t ), end( t ), '/', '_' );
      return t;
    }
 
    RootRef poolRef( size_t i ) const override { return m_poolLinks[i]; }

    int loadRefs( std::string_view /* section */, std::string_view cnt, unsigned long entry,
                  RootObjectRefs& refs ) override {
      auto   ti = sections().find( cnt );
      TTree* t  = ( ti != sections().end() ? ti->second : nullptr );
      if ( !t ) { t = (TTree*)c->file()->Get( _treeName( cnt ).c_str() ); }
      if ( t ) {
        TBranch*   b1  = t->GetBranch( "Links" );
        TBranch*   b2  = t->GetBranch( "Refs" );
        MsgStream& msg = msgSvc();
        if ( b1 && b2 ) {
          LinkManager       lm, *plm   = &lm;
          PoolDbLinkManager mgr, *pmgr = &mgr;
          b1->SetAutoDelete( kFALSE );
          b2->SetAutoDelete( kFALSE );
          b1->SetAddress( &plm );
          b2->SetAddress( &pmgr );
          int nb1 = b1->GetEvent( entry );
          int nb2 = b2->GetEvent( entry );
          if ( nb1 > 1 && nb2 > 1 ) {
            size_t ref_size = mgr.references().size();
            refs.refs.resize( ref_size );
            msg << MSG::VERBOSE;
            for ( size_t j = 0; j < ref_size; ++j ) {
              const pair<int, int>& oid = mgr.references()[j]->m_oid;
              string                loc = mgr.links()[j].substr( 1 );
              RootRef&              r   = refs.refs[j];
              if ( oid.first >= 0 ) {
                r       = m_poolLinks[oid.first];
                r.entry = oid.second;
                r.link  = c->makeLink( loc );
                msg << "Add leaf:" << oid.first << "->" << loc << " from " << c->getDb( r.dbase ) << "#"
                    << c->getCont( r.container ) << " Link:" << c->getLink( r.link ) << " CLID:" << hex << r.clid
                    << " Typ:" << hex << int( r.svc ) << " Ent:" << dec << r.entry << endl;
              }
            }
            for ( int i = 0, n = lm.size(); i < n; ++i ) {
              const LinkManager::Link* lnk     = lm.link( i );
              int                      link_id = c->makeLink( lnk->path() );
              msg << "Add Link:" << lnk->path() << endl;
              refs.links.push_back( link_id );
            }
            return nb1 + nb2;
          }
        }
        msg << MSG::ERROR << "Failed to access POOL Ref/Link branches:" << cnt << " [" << _treeName( cnt ) << "]"
            << endmsg;
        t->Print();
      }
      return -1;
    }

    TBranch* getBranch( std::string_view /* section */, std::string_view branch_name ) override {
      auto   ti = sections().find( branch_name );
      TTree* t  = ( ti != sections().end() ? ti->second : nullptr );
      if ( t ) { return (TBranch*)t->GetListOfBranches()->At( 0 ); }
      string tname = _treeName( branch_name );
      t            = (TTree*)c->file()->Get( tname.c_str() ); // c->getSection(tname);
      if ( t ) {
        TBranch* b = (TBranch*)t->GetListOfBranches()->At( 0 );
        if ( b ) {
          sections()[std::string{ branch_name }] = t;
          return b;
        }
        incidentSvc()->fireIncident( Incident( c->pfn(), IncidentType::CorruptedInputFile ) );
        msgSvc() << MSG::ERROR << "Failed to access POOL branch:" << branch_name << " [" << tname << "]" << endmsg;
        t->Print();
      }
      return nullptr;
    }

    StatusCode saveRefs() override { return StatusCode::FAILURE; }

    StatusCode readRefs() override {
      int  i;
      char text[2048];
      msgSvc() << MSG::VERBOSE;
 
      // First read ##Params
      TTree* t = (TTree*)c->file()->Get( "##Params" );
      if ( !t ) return StatusCode::FAILURE;
      TBranch* b = t->GetBranch( "db_string" );
      if ( !b ) return StatusCode::FAILURE;
      for ( i = 0, b->SetAddress( text ); i < b->GetEntries(); ++i ) {
        b->GetEvent( i );
        char* id1 = strstr( text, "[NAME=" );
        char* id2 = strstr( text, "[VALUE=" );
        if ( id1 && id2 ) {
          id1 += 6;
          id2 += 7;
          char* id11 = strstr( id1, "]" );
          char* id22 = strstr( id2, "]" );
          if ( id11 && id22 ) {
            *id11 = 0;
            *id22 = 0;
            params().emplace_back( id1, id2 );
            msgSvc() << "Param:" << id1 << "=" << id2 << "." << endmsg;
          }
        }
      }
 
      // Read ##Links
      t = (TTree*)c->file()->Get( "##Links" );
      if ( !t ) return StatusCode::FAILURE;
      b = t->GetBranch( "db_string" );
      if ( !b ) return StatusCode::FAILURE;
      m_poolLinks.resize( (size_t)b->GetEntries() + 2 ); // Take into account the ##Links and ##Shapes entry of POOL!
      for ( i = 0, b->SetAddress( text ); i < b->GetEntries(); ++i ) {
        b->GetEvent( i );
        std::string db, container;
        int         clid = 1, technology = 0, ipar[2] = { -1, -1 };
        for ( char* p1 = (char*)text; p1; p1 = ::strchr( ++p1, '[' ) ) {
          char* p2 = ::strchr( p1, '=' );
          char* p3 = ::strchr( p1, ']' );
          if ( p2 && p3 ) {
            if ( ::strncmp( "[DB=", p1, 4 ) == 0 ) {
              *p3 = 0;
              db  = p1 + 4;
            } else if ( ::strncmp( "[CNT=", p1, 5 ) == 0 ) {
              *p3       = 0;
              container = p1 + 5;
            } else if ( ::strncmp( "[OID=", p1, 5 ) == 0 ) {
              *p3 = 0;
              ::sscanf( p1 + 5, "%08X,%08X", (unsigned int*)&ipar[0], (unsigned int*)&ipar[1] );
            } else if ( ::strncmp( "[CLID=", p1, 6 ) == 0 ) {
              *p3 = 0;
              ::sscanf( p1 + 6, "%08X", (unsigned int*)&clid );
            } else if ( ::strncmp( "[TECH=", p1, 6 ) == 0 ) {
              *p3 = 0;
              ::sscanf( p1 + 6, "%08X", (unsigned int*)&technology );
            } else {
              *p3 = *p2 = 0;
            }
            *p3 = ']';
            *p2 = '=';
          }
        }
        c->makeRef( "", clid, technology, db, container, -1, m_poolLinks[i + 2] );
        RootRef& r = m_poolLinks[i];
        msgSvc() << "Add link[" << i << "]:" << db << container << " [" << r.dbase << "," << r.container << "] "
                 << " tech:" << hex << setw( 8 ) << r.svc << " CLID:" << setw( 8 ) << r.clid << dec << endmsg;
      }
      return StatusCode::SUCCESS;
    }
  };
} // namespace Gaudi