RootEventBaseCnv.cxx

Go to the documentation of this file.

#define _RootEventBaseCnv_CXX
 
#include "TClass.h"
#include "TObject.h"
#include "TROOT.h"
 
#include "GaudiKernel/CommonMessaging.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/IAddressCreator.h"
#include "GaudiKernel/IConverter.h"
#include "GaudiKernel/IDataManagerSvc.h"
#include "GaudiKernel/IEvtSelector.h"
#include "GaudiKernel/IService.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/RegistryEntry.h"
#include "GaudiKernel/SmartIF.h"
 
#include "RootCnvSvc/IRootCnvSvc.h"
#include "RootCnvSvc/RootAddress.h"
#include "RootCnvSvc/RootCnvSvc.h"
#include "RootCnvSvc/RootEventBaseCnv.h"
#include "RootCnvSvc/RootEvtSelector.h"
#include "RootEventData/TDigiEvent.h"
 
extern const InterfaceID IID_IRootCnvSvc;
 
using namespace DataSvcHelpers;
 
RootEventBaseCnv::RootEventBaseCnv( const CLID& clid, ISvcLocator* svc )
    : Converter( ROOT_StorageType, clid, svc )
    , m_cnvSvc( 0 )
    , m_rootBranchname( "" )
    , m_rootTreename( "Event" )
    , m_evtsel( 0 ) {
 
  m_branchNr             = -1;
  m_branchNrDst          = -1;
  m_branchNrMc           = -1;
  m_branchNrRecon        = -1;
  m_branchNrEvtRec       = -1;
  m_branchNrEvtHeader    = -1;
  m_branchNrEvtNavigator = -1;
  m_objRead              = 0;
  CLID_top               = 0;
  m_branchNumbers        = new TArrayS( 0 );
}
 
StatusCode RootEventBaseCnv::createRep( DataObject* obj, IOpaqueAddress*& addr ) {
  // Purpose and Method: Convert the transient object to ROOT
 
  MsgStream log( msgSvc(), "RootEventBaseCnv" );
 
  StatusCode sc = StatusCode::SUCCESS;
  // get the corresponding address
  RootAddress* rootaddr;
  sc = m_cnvSvc->createAddress( obj, addr );
 
  rootaddr = dynamic_cast<RootAddress*>( addr );
 
  if ( sc.isFailure() || !rootaddr )
  {
    log << MSG::ERROR << "Could not create address for clid " << obj->clID() << ", objname "
        << obj->name() << endmsg;
    return StatusCode::FAILURE;
  }
 
  //  do the real conversion in the derived converter
  sc = DataObjectToTObject( obj, rootaddr );
 
  delete addr;
  addr = NULL;
 
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Could not transform object" << endmsg;
    return sc;
  }
 
  return StatusCode::SUCCESS;
}
StatusCode RootEventBaseCnv::fillRepRefs( IOpaqueAddress*, DataObject* ) {
  // Purpose and Method:  Resolve the references of the converted object.
  //     It is expected that derived classes will override this method.
  MsgStream log( msgSvc(), "RootEventBaseCnv" );
  return StatusCode::SUCCESS;
}
 
StatusCode RootEventBaseCnv::fillObjRefs( IOpaqueAddress*, DataObject* ) {
  // Purpose and Method:  Resolve the references of the converted object.
  //     It is expected that derived classes will override this method.
  MsgStream log( msgSvc(), "RootEventBaseCnv" );
  return StatusCode::SUCCESS;
}
 
StatusCode RootEventBaseCnv::initialize() {
 
  MsgStream log( msgSvc(), "RootEventBaseCnv" );
  log << MSG::INFO << "RootEventBaseCnv::initialize" << endmsg;
 
  StatusCode status = Converter::initialize();
 
  m_rootInterface = RootInterface::Instance( "RootEventBaseCnv" );
  if ( !m_rootInterface )
    log << MSG::ERROR << "Unable to start Root interface within RootCnvSvc" << endmsg;
 
  StatusCode sc;
  // MsgStream  log( msgSvc(), "RootEventBaseCnv" );
 
  // Get DataSvc
  IService* isvc = 0;
  sc             = serviceLocator()->getService( "EventDataSvc", isvc, true );
  if ( sc.isSuccess() )
  { sc = isvc->queryInterface( IDataProviderSvc::interfaceID(), (void**)&m_eds ); }
  if ( sc.isFailure() )
  { log << MSG::ERROR << "Unable start EventData service within RootEventBaseCnv" << endmsg; }
 
  ///////////////////// liangyt, try to get event id from evt selector
  ////////////////////////////////// / the former method to retrieve the event number of one
  /// branch is to calculate/////////////////// / division of branch number of one event (brN)
  /// and total readed branch num(entryN)/////////////// / there are some disadvantages for
  /// this method. NOW, we try to get event id from evt selector////
 
  sc = serviceLocator()->getService( "RootEvtSelector", isvc, false );
  if ( !sc.isSuccess() ) sc = serviceLocator()->getService( "EventSelector", isvc, false );
  if ( sc.isSuccess() )
  { sc = isvc->queryInterface( IID_IRootEvtSelector, (void**)&m_evtsel ); }
  if ( sc.isFailure() )
  {
    log << MSG::WARNING << "Unable to start event selector service within RootCnvSvc"
        << endmsg;
  }
  ////////////////////////
 
  // if ( status.isSuccess() )
  // {
  //   IService* isvc = 0;
  //   status         = serviceLocator()->service( "RootCnvSvc", isvc, false );
  //   if ( !status.isSuccess() )
  //   {
  //     log << MSG::DEBUG << "retrieved RootCnvSvc" << endmsg;
  //     status = serviceLocator()->service( "EventCnvSvc", isvc, true );
  //   }
  //   else
  //   {
  //     log << MSG::DEBUG << "failed to retrieved RootCnvSvc" << endmsg;
  //     status = isvc->queryInterface( IID_IRootCnvSvc, (void**)&m_cnvSvc );
  //   }
  // }
 
  if ( !status.isSuccess() ) return status;
 
  status = service( "RootCnvSvc", m_cnvSvc, false );
  if ( !status.isSuccess() ) status = service( "EventCnvSvc", m_cnvSvc, true );
 
  return status;
}
 
StatusCode RootEventBaseCnv::finalize() {
  if ( m_cnvSvc )
  {
    m_cnvSvc->release();
    m_cnvSvc = 0;
  }
  return Converter::finalize();
}
 
void RootEventBaseCnv::declareObject( const std::string& path, const CLID& cl,
                                      const std::string& treename,
                                      const std::string& branchname ) {
  // Purpose and Method:  Save the path on the TDS, treename, pathname in the m_leaves vector,
  //   corresponding to the DataObject that the converter handles.
  m_leaves.push_back( RootCnvSvc::Leaf( path, cl, treename, branchname ) );
}
 
StatusCode RootEventBaseCnv::createObj( IOpaqueAddress* addr, DataObject*& refpObject ) {
  // transform ROOT object to TDS object
  MsgStream log( msgSvc(), "RootEventBaseCnv" );
  log << MSG::DEBUG << "RootEventBaseCnv::createObj with clid " << addr->clID() << endmsg;
  StatusCode sc;
 
  // add 2005-11-29
  // log<<MSG::INFO<<"######### RootEventBaseCnv ::createObj begin of createObj:
  // m_branchNumbers
  // "<<m_branchNumbers->GetSize()<<"###############"<<endmsg;
 
  RootAddress* raddr = dynamic_cast<RootAddress*>( addr );
  if ( !raddr )
  {
    log << MSG::ERROR << "Could not downcast to Root address" << endmsg;
    return StatusCode::FAILURE;
  }
 
  static int temp    = 0; // control the begin of each files 2005-12-01
  static int entryN  = 0; // control the event number of each files 2005-21-01
  static int brN     = 0; // control munber of branch of the tree;
  int        lastBrn = brN;
  // lastBrn = brN;
  static int  branchN = 0;
  static bool isSet   = true;
 
  static int  entryBefore      = 0;
  static bool addEntryEachFile = true;
 
  if ( m_rootInterface->getENDFILE() || ( temp > 0 && temp < branchN ) )
  { // if the file has get the end:y the go to next file to create a new tree
 
    if ( m_rootInterface->getENDFILE() ) { entryN = 0; }
 
    temp++;
 
    delete m_branchNumbers;
    m_branchNumbers = new TArrayS( 0 );
 
    if ( temp == branchN ) { temp = 0; }
  }
 
  if ( m_rootInterface->getENDFILE() ) addEntryEachFile = true;
  // the 2nd method
  if ( m_evtsel->getRecId() - entryBefore == 0 )
  { // first event in this file
    delete m_branchNumbers;
    m_branchNumbers = new TArrayS( 0 );
  }
 
  // new method to initialize the branchNumber
  if ( m_currentFileName == "" ) m_currentFileName = m_rootInterface->getCurrentFileName();
  if ( !( m_currentFileName == m_rootInterface->getCurrentFileName() ) )
  {
    m_currentFileName = m_rootInterface->getCurrentFileName();
    delete m_branchNumbers;
    m_branchNumbers = new TArrayS( 0 );
  }
  //----------------------------------------
 
  if ( m_branchNumbers->GetSize() <= 0 )
  {
    if ( isSet ) brN++;
    int branchNumber;
    for ( int nb = 0; nb < raddr->getNrBranches(); nb++ )
    {
      sc = m_rootInterface->setBranchAddress( raddr->getTreename().c_str(),
                                              raddr->getBranchname( nb ).c_str(),
                                              m_adresses[nb], branchNumber );
      if ( !sc.isSuccess() )
      {
        if ( isSet )
          brN--; // liangyt: if fail to retrieve this branch, this will be not a effective
                 // branch.
        // entryN++;   //liangyt: this is the second method
        if ( temp > 0 ) temp--; // temp > 0 means recording effective branch number.
        return sc;
      }
      m_branchNumbers->Set( nb + 1 );
      m_branchNumbers->AddAt( branchNumber, nb );
    }
  }
 
  //// after a file initialized, m_rootInterface get entries
  if ( addEntryEachFile && ( m_evtsel->getRecId() > entryBefore ) )
  { // for a new file, add entry for ONLY one time.
    entryBefore += m_rootInterface->getEntries();
    addEntryEachFile = false;
  }
 
  if ( lastBrn == brN && isSet )
  {
    branchN = brN;
    isSet   = false;
  }
 
  if ( isSet == false )
    log << MSG::INFO << " 1st method set event as : " << int( entryN / branchN ) << endmsg;
  if ( isSet == false )
    raddr->setEntryNr( int( entryN / branchN ) ); // former method, keep it to be backup.
  if ( m_evtsel ) log << MSG::INFO << " event id = " << m_evtsel->getRecId() << endmsg;
 
  int eventID = 0;
  if ( entryBefore > m_evtsel->getRecId() )
    eventID = m_evtsel->getRecId() + m_rootInterface->getEntries() - entryBefore;
  else if ( entryBefore == m_evtsel->getRecId() ) eventID = 0;
  else log << MSG::ERROR << "eventId error!!!" << endmsg;
 
  log << MSG::INFO << " 2nd  method set event as : " << eventID << endmsg;
 
  if ( m_evtsel ) raddr->setEntryNr( eventID );
 
  // add dengzy
  if ( m_rootInterface->getSelectFromTag() != 0 )
  {
    if ( m_evtsel ) raddr->setEntryNr( m_evtsel->getRecId() );
  } // end of add by dengzy
 
  // read branch
 
  if ( m_branchNumbers->GetSize() > 0 )
  {
    int nbtot = 0, nb;
    for ( int ib = 0; ib < m_branchNumbers->GetSize(); ib++ )
    {
      // sc=m_rootInterface->getBranchEntry(m_branchNumbers->At(ib),raddr->getEntryNr(),nb);
      // change to get branch entry with addr(set address for each entry) liangyt
      sc = m_rootInterface->getBranchEntry( m_branchNumbers->At( ib ), raddr->getEntryNr(),
                                            m_adresses[ib], nb );
      if ( sc.isFailure() )
      {
        log << MSG::ERROR << "Could not read branch " << raddr->getBranchname( nb ) << endmsg;
        return sc;
      }
      nbtot += nb;
    }
  }
 
  else
  { // get ROOT object
    if ( CLID_top )
    {
      IConverter*       p   = conversionSvc()->converter( CLID_top );
      RootEventBaseCnv* cnv = dynamic_cast<RootEventBaseCnv*>( p );
      if ( !cnv )
      {
        log << MSG::ERROR << "Could not downcast to RootEventBaseCnv " << endmsg;
        return StatusCode::FAILURE;
      }
      m_objRead = cnv->getReadObject();
    }
  }
 
  // do concrete transformation in derived converter
  sc = TObjectToDataObject( refpObject );
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Could not transform object" << endmsg;
    return sc;
  }
 
  // verify if we have to register
  IRegistry* ent = addr->registry();
  if ( ent == 0 )
  {
    sc = m_eds->registerObject( raddr->getPath(), refpObject );
    if ( sc.isFailure() )
    {
      log << MSG::ERROR << "Could not register object " << raddr->getPath() << " status "
          << sc.getCode() << endmsg;
    }
    //    }
  }
 
  entryN++;
  return StatusCode::SUCCESS;
}