MdcMakeHits.cxx

Go to the documentation of this file.

#include "MdcMakeHits.h"
#include "EventModel/Event.h"
#include "GaudiKernel/IDataManagerSvc.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/INTupleSvc.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/PropertyMgr.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "Identifier/MdcID.h"
#include "MdcCalibFunSvc/IMdcCalibFunSvc.h"
#include "MdcData/MdcHit.h"
#include "MdcData/MdcHitMap.h"
#include "MdcGeom/MdcDetector.h"
#include "MdcGeom/MdcLayer.h"
#include "MdcGeom/MdcSWire.h"
#include "MdcRawEvent/MdcDigi.h"
#include "RawDataProviderSvc/IRawDataProviderSvc.h"
#include "RawDataProviderSvc/MdcRawDataProvider.h"
#include <CLHEP/Geometry/Point3D.h>
#include <CLHEP/Units/PhysicalConstants.h>
 
#include <iostream>
#include <vector>
using namespace std;
using namespace Event;
 
extern const CLID& CLID_MdcHitCol;
 
/////////////////////////////////////////////////////////////////////////////
DECLARE_COMPONENT( MdcMakeHits )
MdcMakeHits::MdcMakeHits( const std::string& name, ISvcLocator* pSvcLocator )
    : Algorithm( name, pSvcLocator ) {
  // Declare the properties
  declareProperty( "cosmicFit", m_cosmicFit = false );
  declareProperty( "doSag", m_doSag = false );
  declareProperty( "countPropTime", m_countPropTime = true );
 
  declareProperty( "getDigiFlag", m_getDigiFlag = 0 );
  declareProperty( "maxMdcDigi", m_maxMdcDigi = 0 );
  declareProperty( "keepBadTdc", m_keepBadTdc = 0 );
  declareProperty( "dropHot", m_dropHot = 0 );
  declareProperty( "keepUnmatch", m_keepUnmatch = 0 );
  declareProperty( "minMdcDigi", m_minMdcDigi = 0 );
  declareProperty( "combineTracking", m_combineTracking = false ); // yzhang 2010-05-28
 
#ifdef MDCPATREC_RESLAYER
  declareProperty( "resLayer", m_resLayer = -1 );
#endif
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode MdcMakeHits::beginRun() {
  _gm = MdcDetector::instance( m_doSag );
  if ( NULL == _gm ) return StatusCode::FAILURE;
  return StatusCode::SUCCESS;
}
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode MdcMakeHits::initialize() {
 
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "in initialize()" << endmsg;
 
  StatusCode sc;
 
  sc = service( "MdcCalibFunSvc", m_mdcCalibFunSvc );
  if ( sc.isFailure() ) { log << MSG::FATAL << "Could not load MdcCalibFunSvc!" << endmsg; }
 
  sc = service( "RawDataProviderSvc", m_rawDataProviderSvc );
  if ( sc.isFailure() )
  {
    log << MSG::FATAL << "Could not load RawDataProviderSvc!" << endmsg;
    return StatusCode::FAILURE;
  }
 
  return StatusCode::SUCCESS;
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode MdcMakeHits::execute() {
  if ( !m_beginRun )
  {
    StatusCode sc = beginRun();
    if ( sc.isFailure() )
    {
      error() << "beginRun failed" << endmsg;
      return StatusCode::FAILURE;
    }
    m_beginRun = true;
  }
 
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "in execute()" << endmsg;
  StatusCode sc;
 
  DataObject* pnode = 0;
  sc                = eventSvc()->retrieveObject( "/Event/Hit", pnode );
  if ( !sc.isSuccess() )
  {
    pnode = new DataObject;
    sc    = eventSvc()->registerObject( "/Event/Hit", pnode );
    if ( !sc.isSuccess() )
    {
      log << MSG::FATAL << " Could not register hit branch" << endmsg;
      return StatusCode::FAILURE;
    }
  }
 
  SmartIF<IDataManagerSvc> dataManSvc( eventSvc() );
  DataObject*              hitCol;
  eventSvc()->findObject( "/Event/Hit/MdcHitCol", hitCol );
  if ( hitCol != NULL )
  {
    dataManSvc->clearSubTree( "/Event/Hit/MdcHitCol" );
    eventSvc()->unregisterObject( "/Event/Hit/MdcHitCol" );
  }
  MdcHitCol* mdcHitCol = new MdcHitCol;
  sc                   = eventSvc()->registerObject( "/Event/Hit/MdcHitCol", mdcHitCol );
  if ( !sc.isSuccess() )
  {
    log << MSG::FATAL << " Could not register hit collection" << endmsg;
    return StatusCode::FAILURE;
  }
 
  DataObject* hitMap;
  eventSvc()->findObject( "/Event/Hit/MdcHitMap", hitMap );
  if ( hitMap != NULL )
  {
    dataManSvc->clearSubTree( "/Event/Hit/MdcHitMap" );
    eventSvc()->unregisterObject( "/Event/Hit/MdcHitMap" );
  }
  MdcHitMap* mdcHitMap = new MdcHitMap( *_gm );
  sc                   = eventSvc()->registerObject( "/Event/Hit/MdcHitMap", mdcHitMap );
  if ( !sc.isSuccess() )
  {
    log << MSG::FATAL << " Could not register hit map" << endmsg;
    return StatusCode::FAILURE;
  }
 
  // retrieve Mdc digi vector form RawDataProviderSvc
  uint32_t getDigiFlag = 0;
  getDigiFlag += m_maxMdcDigi;
  if ( m_dropHot || m_combineTracking ) getDigiFlag |= MdcRawDataProvider::b_dropHot;
  if ( m_keepBadTdc ) getDigiFlag |= MdcRawDataProvider::b_keepBadTdc;
  if ( m_keepUnmatch ) getDigiFlag |= MdcRawDataProvider::b_keepUnmatch;
  MdcDigiVec mdcDigiVec = m_rawDataProviderSvc->getMdcDigiVec( getDigiFlag );
 
  // std::cout<<__FILE__<<" nDigi = "<<mdcDigiVec.size()<<  std::endl;
  if ( (int)mdcDigiVec.size() < m_minMdcDigi )
  {
    log << MSG::WARNING << " Skip this event for MdcDigiVec.size() < " << m_minMdcDigi
        << endmsg;
    return StatusCode::SUCCESS;
  }
 
  MdcDigiVec::iterator iter = mdcDigiVec.begin();
  for ( ; iter != mdcDigiVec.end(); iter++ )
  {
    const MdcDigi* aDigi = *iter;
    MdcHit*        hit   = new MdcHit( aDigi, _gm );
    hit->setCalibSvc( m_mdcCalibFunSvc );
    hit->setCosmicFit( m_cosmicFit );
    hit->setCountPropTime( m_countPropTime );
#ifdef MDCPATREC_RESLAYER
    hit->setResLayer( m_resLayer );
#endif
    mdcHitCol->push_back( hit );
    mdcHitMap->addHit( *hit );
  }
  return StatusCode::SUCCESS;
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode MdcMakeHits::finalize() {
 
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "in finalize()" << endmsg;
 
  return StatusCode::SUCCESS;
}