MrpcRecTDS.cxx

Go to the documentation of this file.

#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/IMessageSvc.h"
#include "GaudiKernel/StatusCode.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "ReconEvent/ReconEvent.h"
#include "TofRecEvent/RecTofTrack.h"
#include "MrpcRec/MrpcTrack.h"
#include "MrpcRec/MrpcRecTDS.h"
 
StatusCode MrpcRecTDS::InitializeTDS() {
  if( RegisterReconEvent() == StatusCode::FAILURE ) return StatusCode::FAILURE;
  if( RegisterNullRecTofTrackCol() == StatusCode::FAILURE ) return StatusCode::FAILURE;
  return StatusCode::SUCCESS;
}
 
 
StatusCode MrpcRecTDS::RegisterNullRecTofTrackCol() {
  IMessageSvc* msgSvc;
  Gaudi::svcLocator()->service("MessageSvc", msgSvc);
  MsgStream log(msgSvc, "MrpcRecTDS");
  log << MSG::INFO << "Initialize: MrpcRecTDS::RegisterNullRecTofTrack()" << endreq;
 
  IDataProviderSvc* eventSvc;
  Gaudi::svcLocator()->service("EventDataSvc", eventSvc);
 
  RecTofTrackCol* recTofTrackCol = new RecTofTrackCol;
 
// Check whether the RecTofTrackCol has been registered.
  StatusCode sc;
  DataObject *aRecTofTrackCol;
  eventSvc->findObject( "/Event/Recon/RecTofTrackCol", aRecTofTrackCol );
  if( aRecTofTrackCol != NULL ) {
    sc = eventSvc->unregisterObject( "/Event/Recon/RecTofTrackCol" );
    if( sc != StatusCode::SUCCESS ) {
      log << MSG::FATAL << "Initialize: Could not unregister RecTofTrack collection!" << endreq;
      delete recTofTrackCol;
      return StatusCode::FAILURE;
    }
  }
  delete aRecTofTrackCol;
 
  sc = eventSvc->registerObject("/Event/Recon/RecTofTrackCol", recTofTrackCol);
  if( sc != StatusCode::SUCCESS ) {
    log << MSG::FATAL << "Initialize: MrpcRec could not register Null Rec TOF track collection!" << endreq;
    delete recTofTrackCol;
    return StatusCode::FAILURE;
  }
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode MrpcRecTDS::RegisterTDS( int runNumber, int eventNumber, std::vector<MrpcTrack*>*& tofTrackVec) {
  //  if( RegisterReconEvent() == StatusCode::FAILURE ) return StatusCode::FAILURE;
  if( RegisterRecTofTrackCol( tofTrackVec ) == StatusCode::FAILURE ) return StatusCode::FAILURE;
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode MrpcRecTDS::RegisterReconEvent() {
  IMessageSvc* msgSvc;
  Gaudi::svcLocator()->service("MessageSvc", msgSvc);
  MsgStream log(msgSvc, "MrpcRecTDS");
 
  IDataProviderSvc* eventSvc;
  Gaudi::svcLocator()->service("EventDataSvc", eventSvc);
 
// Register Reconstructed Track To TDS
  DataObject *aReconEvent ;
  eventSvc->findObject("/Event/Recon",aReconEvent);
  if(aReconEvent==NULL) {
    log << MSG::INFO << "MrpcRecTDS::RegisterReconEvent()" << endreq;
    aReconEvent = new ReconEvent();
    StatusCode sc = eventSvc->registerObject("/Event/Recon",aReconEvent);
    if(sc!=StatusCode::SUCCESS) {
      log << MSG::FATAL << "MrpcRec could not register ReconEvent!" <<endreq;
      return StatusCode::FAILURE;
    }
  } 
  return StatusCode::SUCCESS;
}
 
 
StatusCode MrpcRecTDS::RegisterRecTofTrackCol( std::vector<MrpcTrack*>*& tofTrackVec ) {
  IMessageSvc* msgSvc;
  Gaudi::svcLocator()->service("MessageSvc", msgSvc);
  MsgStream log(msgSvc, "MrpcRecTDS");
  log << MSG::INFO << "MrpcRecTDS::RegisterRecTofTrack()" << endreq;
 
  IDataProviderSvc* eventSvc;
  Gaudi::svcLocator()->service("EventDataSvc", eventSvc);
 
  RecTofTrackCol* recTofTrackCol = new RecTofTrackCol;
 
  std::vector<MrpcTrack*>::iterator iter = tofTrackVec->begin();
  for( ; iter < tofTrackVec->end(); iter++ ) {
    (*iter)->convert2RecTofTrackCol( recTofTrackCol );
  }
 
// Check whether the RecTofTrackCol has been registered.
  StatusCode sc;
  DataObject *aRecTofTrackCol;
  eventSvc->findObject( "/Event/Recon/RecTofTrackCol", aRecTofTrackCol );
  if( aRecTofTrackCol != NULL ) {
 
    sc = eventSvc->unregisterObject( "/Event/Recon/RecTofTrackCol" );
 
    delete aRecTofTrackCol;
 
    if( sc != StatusCode::SUCCESS ) {
      log << MSG::FATAL << "Could not unregister RecTofTrack collection!" << endreq;
 
      RecTofTrackCol::iterator iter_tof = recTofTrackCol->begin();
      for( ; iter_tof!=recTofTrackCol->end(); iter_tof++ ) {
    delete (*iter_tof);
      }
      recTofTrackCol->clear();
      delete recTofTrackCol;
 
      std::vector<MrpcTrack*>::iterator iter_data = tofTrackVec->begin();
      for( ; iter_data!=tofTrackVec->end(); iter_data++ ) {
    delete (*iter_data);
      }
      tofTrackVec->clear();
      delete tofTrackVec;
 
      return StatusCode::FAILURE;
    }
  }
 
  sc = eventSvc->registerObject("/Event/Recon/RecTofTrackCol", recTofTrackCol);
  if( sc != StatusCode::SUCCESS ) {
    log << MSG::FATAL << "MrpcRec could not register Rec TOF track collection!" << endreq;
    RecTofTrackCol::iterator iter_tof = recTofTrackCol->begin();
    for( ; iter_tof!=recTofTrackCol->end(); iter_tof++ ) {
      delete (*iter_tof);
    }
    recTofTrackCol->clear();
    delete recTofTrackCol;
    return StatusCode::FAILURE;
  }
 
  return StatusCode::SUCCESS;
}