TofEnergyRec.cxx

Go to the documentation of this file.

// add bhabha selection code 9/17
//
//  Package: TofEnergyRec
//  BESIII Tof Energy Reconstruction Algorithm.
//  Created by He Miao (EPC IHEP)
//
#include "DstEvent/TofHitStatus.h"
#include "EvTimeEvent/RecEsTime.h"
#include "EventModel/EventHeader.h"
#include "EventModel/EventModel.h"
#include "ExtEvent/RecExtTrack.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/PropertyMgr.h"
#include "GaudiKernel/SmartDataLocator.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "McTruth/McParticle.h"
#include "McTruth/TofMcHit.h"
#include "RawDataProviderSvc/TofData.h"
#include "ReconEvent/ReconEvent.h"
#include "TofCaliSvc/ITofCaliSvc.h"
#include "TofGeomSvc/ITofGeomSvc.h"
#include "TofRecEvent/RecBTofCalHit.h"
#include "TofRecEvent/RecBTofHit.h"
#include "TofRecEvent/RecETofCalHit.h"
#include "TofRecEvent/RecETofHit.h"
#include "TofRecEvent/RecTofTrack.h"
 
#include "RawDataProviderSvc/IRawDataProviderSvc.h"
#include "TofEnergyRec.h"
#include "TofShower.h"
#include <iostream>
 
using namespace std;
using namespace Event;
 
/////////////////////////////////////////////////////////////////////////////
 
IRawDataProviderSvc* tofDigiSvc;
DECLARE_COMPONENT( TofEnergyRec )
TofEnergyRec::TofEnergyRec( const std::string& name, ISvcLocator* pSvcLocator )
    : Algorithm( name, pSvcLocator ) {
  declareProperty( "Output", m_output = false );
  declareProperty( "EventNb", m_eventNb = 0 );
  declareProperty( "CalibConst", m_calibConst = 0.0085 );
  declareProperty( "IsData", m_isData = true );
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 
StatusCode TofEnergyRec::initialize() {
 
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "TofEnergyRec in initialize()" << endmsg;
 
  m_event     = 0;
  m_tofShower = new TofShower;
  m_tofShower->setCalibConst( m_calibConst );
  m_tofShower->setIsData( m_isData );
  // m_tofShower->readCalibPar();
 
  if ( m_output )
  {
    cout << "--------Book Tof Energy Ntuple!" << endl;
    NTuplePtr nt( ntupleSvc(), "FILE209/sci" );
    NTuplePtr nt1( ntupleSvc(), "FILE209/shower" );
    NTuplePtr nt2( ntupleSvc(), "FILE209/seed" );
    if ( nt && nt1 && nt2 )
    {
      m_tuple  = nt;
      m_tuple1 = nt1;
      m_tuple2 = nt2;
    }
    else
    {
      m_tuple  = ntupleSvc()->book( "FILE209/sci", CLID_ColumnWiseTuple, "TofEnergyRec" );
      m_tuple1 = ntupleSvc()->book( "FILE209/shower", CLID_ColumnWiseTuple, "TofEnergyRec" );
      m_tuple2 = ntupleSvc()->book( "FILE209/seed", CLID_ColumnWiseTuple, "TofEnergyRec" );
      if ( m_tuple && m_tuple1 && m_tuple2 )
      { m_tofShower->BookNtuple( m_tuple, m_tuple1, m_tuple2 ); }
      else
      {
        log << MSG::ERROR << " ----- TofShower cannot book N-tuple:" << long( m_tuple )
            << endmsg;
      }
    }
  }
 
  return StatusCode::SUCCESS;
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 
StatusCode TofEnergyRec::execute() {
 
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "TofEnergyRec in execute()!!" << endmsg;
 
  SmartDataPtr<Event::EventHeader> eventHeader( eventSvc(), "/Event/EventHeader" );
  if ( !eventHeader )
  {
    log << MSG::FATAL << "TofEnergyRec could not find Event Header!" << endmsg;
    return StatusCode::FAILURE;
  }
  int run   = eventHeader->runNumber();
  int event = eventHeader->eventNumber();
  if ( m_eventNb != 0 && m_event % m_eventNb == 0 )
    std::cout << "--------event: " << m_event << ", run= " << run << "  event= " << event
              << std::endl;
  log << MSG::INFO << "run= " << run << "  event= " << event << endmsg;
  m_event++;
 
  IRawDataProviderSvc* tofDigiSvc;
  StatusCode           sc = service( "RawDataProviderSvc", tofDigiSvc );
  if ( sc != StatusCode::SUCCESS )
  {
    log << MSG::FATAL << "TofEnergyRec could not find RawDataProviderSvc!" << endmsg;
    return StatusCode::SUCCESS;
  }
 
  // check whether the Recon has been already registered
  DataObject* aReconEvent;
  eventSvc()->findObject( "/Event/Recon", aReconEvent );
  if ( aReconEvent == NULL )
  {
    log << MSG::INFO << "Register ReconEvent" << endmsg;
    // then register Recon
    aReconEvent = new ReconEvent();
    sc          = eventSvc()->registerObject( "/Event/Recon", aReconEvent );
    if ( sc != StatusCode::SUCCESS )
    {
      log << MSG::FATAL << "Could not register ReconEvent" << endmsg;
      return StatusCode::FAILURE;
    }
  }
 
  // Register RecTofTrackCol
  SmartDataPtr<RecTofTrackCol> tofTrackCol( eventSvc(), EventModel::Recon::RecTofTrackCol );
  if ( !tofTrackCol )
  {
    log << MSG::INFO << "Register RecTofTrackCol!" << endmsg;
    tofTrackCol = new RecTofTrackCol;
    sc          = eventSvc()->registerObject( EventModel::Recon::RecTofTrackCol, tofTrackCol );
    if ( sc != StatusCode::SUCCESS )
    { log << MSG::FATAL << "Could not register RecTofTrackCol!" << endmsg; }
  }
 
  // Retrieve RecEsTimeCol
  SmartDataPtr<RecEsTimeCol> estimeCol( eventSvc(), "/Event/Recon/RecEsTimeCol" );
  if ( !estimeCol || ( estimeCol->size() == 0 ) )
  {
    log << MSG::WARNING << "TofRec Could not find RecEsTimeCol!   Run = " << run
        << " Event = " << event << endmsg;
    return StatusCode::SUCCESS;
  }
  RecEsTimeCol::iterator iter_ESTime = estimeCol->begin();
 
  double t0  = 0;
  t0         = ( *iter_ESTime )->getTest();
  int t0Stat = ( *iter_ESTime )->getStat();
 
  std::vector<TofData*> tofDataVec;
  tofDataVec = tofDigiSvc->tofDataVectorTof();
 
  m_tofShower->findShower( tofDataVec, tofTrackCol, t0 );
 
  return StatusCode::SUCCESS;
}
 
//   * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 
StatusCode TofEnergyRec::finalize() {
  cout << "Total event:" << m_event << endl;
  delete m_tofShower;
  return StatusCode::SUCCESS;
}