/home/bes3soft/bes3soft/Boss/7.0.2/dist/7.0.2/Event/RootCnvSvc/RootCnvSvc-02-01-12/src/RootEventBaseCnv.cxx

Go to the documentation of this file.

#define _RootEventBaseCnv_CXX

#include "TROOT.h"
#include "TObject.h"
#include "TClass.h"

#include "GaudiKernel/IDataManagerSvc.h"
#include "GaudiKernel/IService.h"
#include "GaudiKernel/IEvtSelector.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/RegistryEntry.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/IConverter.h"
#include "GaudiKernel/IAddressCreator.h"
#include "GaudiKernel/SmartIF.h"


#include "RootEventData/TDigiEvent.h"
#include "RootCnvSvc/RootEvtSelector.h"
#include "RootCnvSvc/RootEventBaseCnv.h"
#include "RootCnvSvc/RootCnvSvc.h"
#include "RootCnvSvc/RootAddress.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)
{
    
    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" << endreq;
    } 

   m_rootInterface=RootInterface::Instance(log);
   if (!m_rootInterface)      log << MSG::ERROR << "Unable to start Root interface within RootCnvSvc" << endreq;

   
   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" << endreq;
   }

   
    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()<<endreq;
    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" << endreq;
    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()   {
 
  StatusCode status = Converter::initialize();

  if ( status.isSuccess() )   {
    IService* isvc = 0;
    status = serviceLocator()->service("RootCnvSvc", isvc, false);
    if ( !status.isSuccess() )   status = serviceLocator()->service("EventCnvSvc", isvc, true);
    if ( status.isSuccess() )   {
      status = isvc->queryInterface(IID_IRootCnvSvc, (void**)&m_cnvSvc);
    }
  }
  
  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()<< endreq;
  StatusCode sc;

  // add 2005-11-29
  // log<<MSG::INFO<<"######### RootEventBaseCnv ::createObj begin of createObj:  m_branchNumbers "<<m_branchNumbers->GetSize()<<"###############"<<endreq;

  RootAddress *raddr=dynamic_cast<RootAddress *>(addr);
  if (!raddr) {
    log << MSG::ERROR << "Could not downcast to Root address" << endreq;
    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);
     
    }
  }

  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)<<endreq;
  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()<<endreq;


  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!!!"<<endreq;
  
  log << MSG::INFO <<" 2nd  method set event as : "<<eventID<<endreq;
  
  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) << endreq;
        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 "  << endreq;
        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" << endreq;
    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()<<endreq;
       }
       //    }
  }

  entryN++;
  return StatusCode::SUCCESS;
}

---