MdcFastTrkAlg.cxx

Go to the documentation of this file.

#include "MdcFastTrkAlg.h"
#include "FTFinder.h"
 
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/PropertyMgr.h"
#include "GaudiKernel/SmartDataPtr.h"
 
#include "EventModel/Event.h"
#include "EventModel/EventHeader.h"
#include "MdcGeomSvc/IMdcGeomSvc.h"
#include "MdcGeomSvc/MdcGeoLayer.h"
#include "MdcGeomSvc/MdcGeoWire.h"
#include "MdcRawEvent/MdcDigi.h"
 
#include <fstream>
 
#ifndef OnlineMode
#  include "McTruth/McKine.h"
#  include "ntupleItem.h"
#endif
 
#include "MdcRecEvent/RecMdcTrack.h"
 
#include <iostream>
#include <vector>
 
using namespace std;
 
//...Globals...
 
#ifndef OnlineMode
// MCTruth
 
int num_2Dtrk, num_3Dtrk;
int num_finaltrk;
 
extern NTuple::Item<long>   g_ntrkMC, g_eventNo;
extern NTuple::Array<float> g_theta0MC, g_phi0MC;
extern NTuple::Array<float> g_pxMC, g_pyMC, g_pzMC, g_ptMC;
 
// recon
extern NTuple::Item<long>   g_ntrk;
extern NTuple::Item<float>  g_eventtime;
extern NTuple::Array<float> g_px, g_py, g_pz, g_pt, g_p;
extern NTuple::Array<float> g_phi, g_theta;
extern NTuple::Array<float> g_vx, g_vy, g_vz;
extern NTuple::Array<float> g_dr, g_phi0, g_kappa, g_dz, g_tanl;
extern NTuple::Item<float>  g_estime;
 
extern IHistogram1D* g_sigmaxy;
extern IHistogram1D* g_sigmaz;
extern IHistogram1D* g_chi2xy;
extern IHistogram1D* g_chi2sz;
extern IHistogram1D* g_ncellMC;
extern IHistogram1D* g_ncell;
extern IHistogram1D* g_naxialhit;
extern IHistogram1D* g_nstereohit;
extern IHistogram1D* g_nhit;
extern IHistogram2D* g_hitmap[20];
#endif
 
/////////////////////////////////////////////////////////////////////////////
DECLARE_COMPONENT( MdcFastTrkAlg )
MdcFastTrkAlg::MdcFastTrkAlg( const std::string& name, ISvcLocator* pSvcLocator )
    : Algorithm( name, pSvcLocator ) {
  // Declare the properties
  declareProperty( "bunchtime", m_bunchtime_MC = 8.0 );
  declareProperty( "T0cal", m_mdc_tcal = 200.0 );
  m_ftFinder = 0;
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode MdcFastTrkAlg::initialize() {
 
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "in initialize()" << endmsg;
 
#ifndef OnlineMode
  NTuplePtr nt( ntupleSvc(), "FILE102/n1" );
  if ( nt ) m_tuple = nt;
  else
  {
    m_tuple = ntupleSvc()->book( "FILE102/n1", CLID_ColumnWiseTuple, "MdcRecEvent" );
    if ( m_tuple )
    {
      m_tuple->addItem( "eventNo", g_eventNo );
      m_tuple->addItem( "NtrackMC", g_ntrkMC, 0, 50 );
      m_tuple->addItem( "MCtheta0", g_ntrkMC, g_theta0MC );
      m_tuple->addItem( "MCphi0", g_ntrkMC, g_phi0MC );
      m_tuple->addItem( "pxMC", g_ntrkMC, g_pxMC );
      m_tuple->addItem( "pyMC", g_ntrkMC, g_pyMC );
      m_tuple->addItem( "pzMC", g_ntrkMC, g_pzMC );
      m_tuple->addItem( "ptMC", g_ntrkMC, g_ptMC );
      m_tuple->addItem( "Ntrack", g_ntrk, 0, 50 );
      m_tuple->addItem( "px", g_ntrk, g_px );
      m_tuple->addItem( "py", g_ntrk, g_py );
      m_tuple->addItem( "pz", g_ntrk, g_pz );
      m_tuple->addItem( "pt", g_ntrk, g_pt );
      m_tuple->addItem( "p", g_ntrk, g_p );
      m_tuple->addItem( "phi", g_ntrk, g_phi );
      m_tuple->addItem( "theta", g_ntrk, g_theta );
      m_tuple->addItem( "vx", g_ntrk, g_vx );
      m_tuple->addItem( "vy", g_ntrk, g_vy );
      m_tuple->addItem( "vz", g_ntrk, g_vz );
      m_tuple->addItem( "dr", g_ntrk, g_dr );
      m_tuple->addItem( "phi0", g_ntrk, g_phi0 );
      m_tuple->addItem( "kappa", g_ntrk, g_kappa );
      m_tuple->addItem( "dz", g_ntrk, g_dz );
      m_tuple->addItem( "tanl", g_ntrk, g_tanl );
      m_tuple->addItem( "eventtime", g_eventtime );
      m_tuple->addItem( "Testime", g_estime );
    }
    else
    { // did not manage to book the N tuple....
      log << MSG::ERROR << "Cannot book N-tuple:" << long( m_tuple ) << endmsg;
      return StatusCode::FAILURE;
    }
  }
 
  g_sigmaxy    = histoSvc()->book( "sigmaxy", "1D sigmaxy", 100, -0.2, 0.2 );
  g_sigmaz     = histoSvc()->book( "sigmaz", "1D sigmaz", 100, -3.5, 3.5 );
  g_chi2xy     = histoSvc()->book( "chi2xy", "1D chi2xy", 100, -0.01, 0.01 );
  g_chi2sz     = histoSvc()->book( "chi2sz", "1D chi2sz", 100, -8.0, 8. );
  g_ncellMC    = histoSvc()->book( "ncellmc", "1D ncellMC", 100, 0.0, 7000. );
  g_ncell      = histoSvc()->book( "ncell", "1D ncell", 100, 0.0, 7000. );
  g_naxialhit  = histoSvc()->book( "naxialhit", "2D axial hit", 30, 0.0, 30. );
  g_nstereohit = histoSvc()->book( "nstereohit", "2D stereo hit", 30, 0.0, 30. );
  g_nhit       = histoSvc()->book( "nhit", "2D hit", 50, 0.0, 50. );
  g_hitmap[0]  = histoSvc()->book( "hm0", "2d hitmap1", 80, -80., 80., 80, -80., 80. );
  g_hitmap[1]  = histoSvc()->book( "hm1", "2d hitmap2", 80, -80., 80., 80, -80., 80. );
  g_hitmap[2]  = histoSvc()->book( "hm2", "2d hitmap3", 80, -80., 80., 80, -80., 80. );
  g_hitmap[3]  = histoSvc()->book( "hm3", "2d hitmap4", 80, -80., 80., 80, -80., 80. );
  g_hitmap[4]  = histoSvc()->book( "hm4", "2d hitmap5", 80, -80., 80., 80, -80., 80. );
  g_hitmap[5]  = histoSvc()->book( "hm5", "2d hitmap6", 80, -80., 80., 80, -80., 80. );
  g_hitmap[6]  = histoSvc()->book( "hm6", "2d hitmap7", 80, -80., 80., 80, -80., 80. );
  g_hitmap[7]  = histoSvc()->book( "hm7", "2d hitmap8", 80, -80., 80., 80, -80., 80. );
  g_hitmap[8]  = histoSvc()->book( "hm8", "2d hitmap9", 80, -80., 80., 80, -80., 80. );
  g_hitmap[9]  = histoSvc()->book( "hm9", "2d hitmap10", 80, -80., 80., 80, -80., 80. );
  g_hitmap[10] = histoSvc()->book( "hm10", "2d hitmap11", 80, -80., 80., 80, -80., 80. );
 
  num_2Dtrk    = 0;
  num_3Dtrk    = 0;
  num_finaltrk = 0;
#endif
 
  m_ftFinder = new FTFinder();
  m_ftFinder->init();
  m_ftFinder->setBunchtime( m_bunchtime_MC );
  m_ftFinder->setT0cal( m_mdc_tcal );
  m_ftFinder->setAlgorithmPointer( this );
 
#ifndef OnlineMode
  StatusCode sc = service( "BesTimerSvc", m_timersvc );
  if ( sc.isFailure() )
  {
    log << MSG::WARNING << name() << ": Unable to locate BesTimer Service" << endmsg;
    return StatusCode::FAILURE;
  }
 
  m_timer[1] = m_timersvc->addItem( "Execution" );
  m_timer[1]->propName( "nExecution" );
#endif
 
  return StatusCode::SUCCESS;
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode MdcFastTrkAlg::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;
 
#ifndef OnlineMode
  m_timer[1]->start();
#endif
 
  m_ftFinder->event();
 
#ifndef OnlineMode
  m_timer[1]->stop();
 
  g_eventtime = m_timer[1]->elapsed();
 
  if ( m_tuple )
  {
    StatusCode status = m_tuple->write();
    if ( !status.isSuccess() ) { log << MSG::ERROR << "Can't fill ntuple!" << endmsg; }
  }
#endif
 
  return StatusCode::SUCCESS;
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode MdcFastTrkAlg::finalize() {
 
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "in finalize()" << endmsg;
 
  m_ftFinder->term();
#ifndef OnlineMode
  log << MSG::DEBUG << "num_2Dtrk: " << num_2Dtrk << " num_3Dtrk: " << num_3Dtrk
      << " num_finaltrk: " << num_finaltrk << endmsg;
  m_timersvc->print();
#endif
 
  return StatusCode::SUCCESS;
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode MdcFastTrkAlg::beginRun() {
  info() << "in beginRun()" << endmsg;
  m_ftFinder->begin_run();
 
  return StatusCode::SUCCESS;
}