MdcPrintSvc.cxx

Go to the documentation of this file.

#include "MdcPrintSvc.h"
 
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/DataSvc.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/IInterface.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/Kernel.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "GaudiKernel/StatusCode.h"
 
#include "EvTimeEvent/RecEsTime.h"
#include "EventModel/EventHeader.h"
#include "EventModel/EventModel.h"
#include "Identifier/MdcID.h"
#include "McTruth/McParticle.h"
#include "McTruth/MdcMcHit.h"
#include "MdcRawEvent/MdcDigi.h"
#include "MdcRecEvent/RecMdcHit.h"
#include "MdcRecEvent/RecMdcTrack.h"
#include "RawEvent/RawDataUtil.h"
 
#include <iomanip>
#include <iostream>
 
using namespace std;
using namespace Event;
DECLARE_COMPONENT( MdcPrintSvc )
MdcPrintSvc::MdcPrintSvc( const string& name, ISvcLocator* svcloc )
    : base_class( name, svcloc ) {
 
  // declare properties
  declareProperty( "printHit", m_printHit = 1 );
  declareProperty( "printTrackDetail", m_printTrackDetail = 1 );
  declareProperty( "printErrMat", m_printErrMat = 0 );
  declareProperty( "printTdcAdc", m_printTdcAdc = 0 );
  declareProperty( "printMcTk", m_printMcTk = 0 );
}
 
MdcPrintSvc::~MdcPrintSvc() {}
 
StatusCode MdcPrintSvc::initialize() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "MdcPrintSvc::initialize()" << endmsg;
 
  StatusCode sc = Service::initialize();
  if ( sc.isFailure() ) return sc;
 
  ISvcLocator* svcLocator = Gaudi::svcLocator();
  svcLocator->service( "EventDataSvc", eventSvc, true );
  if ( NULL == eventSvc )
  {
    std::cout << "ERROR Could not load service DataSvc" << std::endl;
    return StatusCode::FAILURE;
  }
 
  IRawDataProviderSvc* irawDataProviderSvc;
  sc                   = service( "RawDataProviderSvc", irawDataProviderSvc );
  m_rawDataProviderSvc = irawDataProviderSvc;
  if ( sc.isFailure() )
  {
    log << MSG::FATAL << "Could not load RawDataProviderSvc!" << endmsg;
    return StatusCode::FAILURE;
  }
 
  return StatusCode::SUCCESS;
}
 
StatusCode MdcPrintSvc::finalize() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "MdcPrintSvc::finalize()" << endmsg;
 
  return StatusCode::SUCCESS;
}
 
/*StatusCode MdcPrintSvc::queryInterface(const InterfaceID& riid, void** ppvInterface){
  if( IID_IMdcPrintSvc.versionMatch(riid) ){
    *ppvInterface = static_cast<IMdcPrintSvc*> (this);
  } else{
    return Service::queryInterface(riid, ppvInterface);
  }
  return StatusCode::SUCCESS;
}*/
 
void MdcPrintSvc::printDigi( uint32_t getDigiFlag ) const {
  MdcDigiVec           mdcDigiVec = m_rawDataProviderSvc->getMdcDigiVec( getDigiFlag );
  MdcDigiCol::iterator iter       = mdcDigiVec.begin();
  std::cout << "//==== " << name() << " print MdcDigiVec, nDigi=" << mdcDigiVec.size() << " :"
            << std::endl;
  for ( int iDigi = 0; iter != mdcDigiVec.end(); iter++, iDigi++ )
  {
    long l = MdcID::layer( ( *iter )->identify() );
    long w = MdcID::wire( ( *iter )->identify() );
    cout << "(" << l << "," << w;
    if ( m_printMcTk )
    {
      int tkTruth = ( *iter )->getTrackIndex();
      cout << ";" << tkTruth;
    }
    if ( m_printTdcAdc )
    {
      double tdc = RawDataUtil::MdcTime( ( *iter )->getTimeChannel() );
      double adc = RawDataUtil::MdcCharge( ( *iter )->getChargeChannel() );
      cout << ",t " << tdc << ",c " << adc;
    }
    cout << ")";
    if ( iDigi % 4 == 0 ) std::cout << std::endl;
  }
  std::cout << std::endl;
}
 
void MdcPrintSvc::printRecMdcTrackCol() const {
  MsgStream log( msgSvc(), name() );
 
  SmartDataPtr<RecMdcTrackCol> recMdcTrackCol( eventSvc, "/Event/Recon/RecMdcTrackCol" );
  if ( !recMdcTrackCol )
  {
    log << MSG::WARNING << "Could not find RecMdcTrackCol" << endmsg;
    return;
  }
 
  std::cout << "//======= " << name() << " RecMdcTrackCol size = " << recMdcTrackCol->size()
            << std::endl;
  RecMdcTrackCol::iterator it = recMdcTrackCol->begin();
  for ( ; it != recMdcTrackCol->end(); it++ ) { printRecMdcTrack( *it ); }
}
 
void MdcPrintSvc::printRecMdcTrack( const RecMdcTrack* tk ) const {
  std::cout << "//==== " << name() << " print RecMdcTrack No." << tk->trackId() << " :"
            << std::endl;
  cout << " dr " << tk->helix( 0 ) << " phi0 " << tk->helix( 1 ) << " cpa " << tk->helix( 2 )
       << " z0 " << tk->helix( 3 ) << " tanl " << tk->helix( 4 ) << endl;
  if ( m_printTrackDetail )
  {
    std::cout << " q " << tk->charge() << " theta " << tk->theta() << " phi " << tk->phi()
              << " x0 " << tk->x() << " y0 " << tk->y() << " z0 " << tk->z() << " r0 "
              << tk->r() << endl;
    std::cout << " p " << tk->p() << " pt " << tk->pxy() << " pxyz(" << tk->px() << ","
              << tk->py() << "," << tk->pz() << ")" << endl;
    std::cout << " tkStat " << tk->stat() << " chi2 " << tk->chi2() << " ndof " << tk->ndof()
              << " nhit " << tk->getNhits() << " nst " << tk->nster() << endl;
    if ( m_printErrMat )
    {
      std::cout << "errmat   " << std::endl;
      for ( int i = 0; i < 15; i++ ) { std::cout << " " << tk->err( i ); }
      std::cout << "   " << std::endl;
    }
  }
 
  if ( m_printHit )
  {
    int haveDigi[43][288];
    if ( m_printMcTk )
    {
      for ( int ii = 0; ii < 43; ii++ )
      {
        for ( int jj = 0; jj < 43; jj++ ) { haveDigi[ii][jj] = -9999; }
      }
      MdcDigiVec           mdcDigiVec = m_rawDataProviderSvc->getMdcDigiVec();
      MdcDigiCol::iterator iter       = mdcDigiVec.begin();
      std::cout << name() << "//==== " << name()
                << " print MdcDigiVec, nDigi=" << mdcDigiVec.size() << " :" << std::endl;
      for ( int iDigi = 0; iter != mdcDigiVec.end(); iter++, iDigi++ )
      {
        long l         = MdcID::layer( ( *iter )->identify() );
        long w         = MdcID::wire( ( *iter )->identify() );
        haveDigi[l][w] = ( *iter )->getTrackIndex();
      }
    }
 
    int nhits = tk->getVecHits().size();
    std::cout << "nHits=" << nhits << std::endl;
    cout << "No. ";
    if ( m_printMcTk ) cout << "mcTkId ";
    cout << "(layer,wire,lr) stat z" << endl;
    for ( int ii = 0; ii < nhits; ii++ )
    {
      Identifier id( tk->getVecHits()[ii]->getMdcId() );
      int        layer = MdcID::layer( id );
      int        wire  = MdcID::wire( id );
      cout << ii << ":";
      if ( m_printMcTk ) { cout << haveDigi[layer][wire]; }
      cout << " (" << layer << "," << wire << "," << tk->getVecHits()[ii]->getFlagLR() << ")  "
           << tk->getVecHits()[ii]->getStat() << "  " << tk->getVecHits()[ii]->getZhit() << " "
           << std::endl;
    } // end of hit list
    std::cout << "  " << std::endl;
  }
}
 
void MdcPrintSvc::printMdcMcHitCol() const {
  MsgStream log( msgSvc(), name() );
 
  SmartDataPtr<MdcMcHitCol> mdcMcHitCol( eventSvc, "/Event/MC/MdcMcHitCol" );
  if ( !mdcMcHitCol )
  {
    log << MSG::WARNING << "Could not find MdcMcHitCol" << endmsg;
    return;
  }
 
  int ihit = 0;
  cout << "//==== " << name() << " print MdcMcHitCol nMcHit=" << mdcMcHitCol->size() << endl;
  if ( mdcMcHitCol->size() > 0 )
  { cout << "No. (layer,wire) mcTk  (x,y,z)   lr  driftdist(cm)  energy" << endl; }
  MdcMcHitCol::iterator iter_mchit = mdcMcHitCol->begin();
  for ( ; iter_mchit != mdcMcHitCol->end(); iter_mchit++, ihit++ )
  {
    const Identifier id       = ( *iter_mchit )->identify();
    int              layer    = MdcID::layer( id );
    int              wire     = MdcID::wire( id );
    int              iMcTk    = ( *iter_mchit )->getTrackIndex();
    double           mcX      = ( *iter_mchit )->getPositionX() / 10.;
    double           mcY      = ( *iter_mchit )->getPositionY() / 10.;
    double           mcZ      = ( *iter_mchit )->getPositionZ() / 10.;
    double           mcLR     = ( *iter_mchit )->getPositionFlag();
    double           mcDrift  = ( *iter_mchit )->getDriftDistance() / 10.;
    double           mcEnergy = ( *iter_mchit )->getDepositEnergy() / 10.;
    if ( mcLR == 0 ) mcLR = -1;
    cout << ihit << " (" << layer << "," << wire << ") " << iMcTk << " (" << mcX << "," << mcY
         << "," << mcZ << ")  " << mcLR << "  " << mcDrift << "  " << mcEnergy << endl;
  }
}