MdcAliEvent.cxx

Go to the documentation of this file.

#include "MdcAlignAlg/MdcAliEvent.h"
 
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/IMessageSvc.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/PropertyMgr.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "GaudiKernel/StatusCode.h"
 
#include "BesTimerSvc/BesTimerSvc.h"
#include "BesTimerSvc/IBesTimerSvc.h"
 
#include "EventModel/Event.h"
#include "RawEvent/RawDataUtil.h"
 
#include "MdcCalibFunSvc/IMdcCalibFunSvc.h"
#include "MdcCalibFunSvc/MdcCalibFunSvc.h"
 
#include "EvTimeEvent/RecEsTime.h"
#include "MdcRecEvent/RecMdcHit.h"
#include "MdcRecEvent/RecMdcKalTrack.h"
#include "MdcRecEvent/RecMdcTrack.h"
#include "ReconEvent/ReconEvent.h"
 
#include "Identifier/Identifier.h"
#include "Identifier/MdcID.h"
 
#include <iostream>
#include <math.h>
 
using namespace Event;
 
StatusCode MdcAliEvent::setRecEvent() {
  IMessageSvc* msgSvc;
  Gaudi::svcLocator()->service( "MessageSvc", msgSvc );
  MsgStream log( msgSvc, "MdcAliEvent" );
  log << MSG::INFO << "MdcAliEvent::setRecEvent" << endmsg;
 
  IDataProviderSvc* eventSvc = NULL;
  Gaudi::svcLocator()->service( "EventDataSvc", eventSvc );
 
  // get EsTimeCol
  SmartDataPtr<RecEsTimeCol> aevtimeCol( eventSvc, "/Event/Recon/RecEsTimeCol" );
  if ( !aevtimeCol || ( aevtimeCol->size() == 0 ) )
  {
    m_tes    = -9999.0;
    m_esFlag = -1;
  }
  else
  {
    RecEsTimeCol::iterator iter_evt = aevtimeCol->begin();
    for ( ; iter_evt != aevtimeCol->end(); iter_evt++ )
    {
      m_tes    = ( *iter_evt )->getTest();
      m_esFlag = ( *iter_evt )->getStat();
    }
  }
 
  m_esCutFg    = false;
  bool flagTes = false;
  for ( int iEs = 0; iEs < m_param.nEsFlag; iEs++ )
  {
    if ( m_esFlag == m_param.esFlag[iEs] )
    {
      flagTes = true;
      break;
    }
  }
  if ( flagTes && ( m_tes > m_param.tesMin ) && ( m_tes < m_param.tesMax ) ) m_esCutFg = true;
 
  SmartDataPtr<RecMdcTrackCol> newtrkCol( eventSvc, "/Event/Recon/RecMdcTrackCol" );
  if ( !newtrkCol )
  {
    log << MSG::ERROR << "Could not find RecMdcTrackCol" << endmsg;
    return ( StatusCode::FAILURE );
  }
 
  int                      i = 0;
  MdcAliRecTrk*            rectrk;
  RecMdcTrackCol::iterator it_trk = newtrkCol->begin();
  for ( ; it_trk != newtrkCol->end(); it_trk++ )
  {
    rectrk = new MdcAliRecTrk();
    rectrk->setRecTrk( it_trk );
    m_rectrk.push_back( rectrk );
 
    i++;
  }
 
  return StatusCode::SUCCESS;
}
 
StatusCode MdcAliEvent::setKalEvent() {
  IMessageSvc* msgSvc;
  Gaudi::svcLocator()->service( "MessageSvc", msgSvc );
  MsgStream log( msgSvc, "MdcAliEvent" );
  log << MSG::INFO << "MdcAliEvent::setKalEvent" << endmsg;
 
  IDataProviderSvc* eventSvc = NULL;
  Gaudi::svcLocator()->service( "EventDataSvc", eventSvc );
 
  // get EsTimeCol
  SmartDataPtr<RecEsTimeCol> aevtimeCol( eventSvc, "/Event/Recon/RecEsTimeCol" );
  if ( !aevtimeCol || ( aevtimeCol->size() == 0 ) )
  {
    m_tes    = -9999.0;
    m_esFlag = -1;
  }
  else
  {
    RecEsTimeCol::iterator iter_evt = aevtimeCol->begin();
    for ( ; iter_evt != aevtimeCol->end(); iter_evt++ )
    {
      m_tes    = ( *iter_evt )->getTest();
      m_esFlag = ( *iter_evt )->getStat();
    }
  }
 
  m_esCutFg    = false;
  bool flagTes = false;
  for ( int iEs = 0; iEs < m_param.nEsFlag; iEs++ )
  {
    if ( m_esFlag == m_param.esFlag[iEs] )
    {
      flagTes = true;
      break;
    }
  }
  if ( flagTes && ( m_tes > m_param.tesMin ) && ( m_tes < m_param.tesMax ) ) m_esCutFg = true;
 
  SmartDataPtr<RecMdcKalTrackCol> kaltrkCol( eventSvc, "/Event/Recon/RecMdcKalTrackCol" );
  if ( !kaltrkCol )
  {
    log << MSG::FATAL << "Could not find RecMdcKalTrackCol" << endmsg;
    return StatusCode::FAILURE;
  }
 
  int                         i = 0;
  MdcAliRecTrk*               rectrk;
  RecMdcKalTrackCol::iterator iter_trk = kaltrkCol->begin();
  for ( ; iter_trk != kaltrkCol->end(); iter_trk++ )
  {
    rectrk = new MdcAliRecTrk();
    rectrk->setKalTrk( iter_trk );
    m_rectrk.push_back( rectrk );
 
    i++;
  }
  log << MSG::INFO << "KalTrack ntrk = " << i << endmsg;
 
  return StatusCode::SUCCESS;
}
 
void MdcAliEvent::clear() {
  unsigned int i;
  for ( i = 0; i < m_rectrk.size(); i++ ) { delete m_rectrk[i]; }
  m_rectrk.clear();
}