BeamEnergySvc.cxx

Go to the documentation of this file.

#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/IIncidentListener.h"
#include "GaudiKernel/IIncidentSvc.h"
#include "GaudiKernel/IInterface.h"
#include "GaudiKernel/Incident.h"
#include "GaudiKernel/Kernel.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/Service.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "GaudiKernel/StatusCode.h"
 
#include "BeamEnergySvc.h"
#include "DatabaseSvc/DatabaseRecord.h"
#include "EventModel/Event.h"
#include "EventModel/EventHeader.h"
#include "EventModel/EventModel.h"
 
#include <math.h>
 
using namespace std;
 
DECLARE_COMPONENT( BeamEnergySvc )
BeamEnergySvc::BeamEnergySvc( const std::string& name, ISvcLocator* svcloc )
    : base_class( name, svcloc ), m_run( -1 ), m_isRunValid( false ), m_beamE( 0 ) {
  declareProperty( "Host", m_host = "bes3db2.ihep.ac.cn" );
}
 
BeamEnergySvc::~BeamEnergySvc() {}
 
StatusCode BeamEnergySvc::initialize() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "@initialize()" << endmsg;
 
  StatusCode sc = Service::initialize();
 
  if ( sc.isFailure() ) return sc;
 
  IIncidentSvc* incsvc;
  sc           = service( "IncidentSvc", incsvc );
  int priority = 100;
  if ( sc.isSuccess() ) { incsvc->addListener( this, "NewRun", priority ); }
 
  sc = serviceLocator()->service( "DatabaseSvc", m_dbsvc, true );
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Unable to find DatabaseSvc " << endmsg;
    return sc;
  }
 
  sc = serviceLocator()->service( "EventDataSvc", m_eventSvc, true );
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Unable to find EventDataSvc " << endmsg;
    return sc;
  }
  m_run = 0;
  return sc;
}
 
StatusCode BeamEnergySvc::finalize() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "@initialize()" << endmsg;
 
  StatusCode sc = Service::finalize();
 
  return sc;
}
 
/*StatusCode BeamEnergySvc::queryInterface(const InterfaceID& riid, void** ppvIF)
{
  if ( IBeamEnergySvc::interfaceID().versionMatch(riid) ) {
    *ppvIF = dynamic_cast<IBeamEnergySvc*>(this);
  }
  else {
    return Service::queryInterface(riid, ppvIF);
  }
  addRef();
  return StatusCode::SUCCESS;
}*/
 
void BeamEnergySvc::handle( const Incident& inc ) {
  MsgStream log( msgSvc(), name() );
  log << MSG::DEBUG << "handle: " << inc.type() << endmsg;
 
  SmartDataPtr<Event::EventHeader> eventHeader( m_eventSvc, "/Event/EventHeader" );
  int                              run = eventHeader->runNumber();
  //  if( run<0 || run>11000 ) return;
 
  if ( inc.type() == "NewRun" )
  {
    log << MSG::DEBUG << "New Run:" << run << endmsg;
    getBeamEnergyInfo();
  }
}
 
bool BeamEnergySvc::isRunValid() { return m_isRunValid; }
 
double BeamEnergySvc::getbeamE() {
  if ( !isRunValid() )
  { std::cerr << "ERROR in BeamEnergySvc: runNo is invalid!" << std::endl; }
  return m_beamE;
}
 
void BeamEnergySvc::getBeamEnergyInfo() {
  MsgStream                        log( msgSvc(), name() );
  SmartDataPtr<Event::EventHeader> eventHeader( m_eventSvc, "/Event/EventHeader" );
  int                              run = eventHeader->runNumber();
 
  if ( run < 0 )
  {
    log << MSG::INFO << "This data is the MC sample with the Run Number: " << run << endmsg;
    m_isRunValid = true;
    m_beamE      = 1.8865;
    return;
    // run=-run;
  }
  m_run = run;
  char stmt1[400];
 
  snprintf( stmt1, 1024,
            "select BER_PRB, BPR_PRB "
            "from RunParams where run_number = %d",
            run );
 
  DatabaseRecordVector res;
  int                  row_no = m_dbsvc->query( "run", stmt1, res );
  if ( row_no == 0 )
  {
    log << MSG::FATAL << "can not find result for BeamEnergySvc" << endmsg;
    exit( 1 );
  }
 
  m_isRunValid = true;
 
  DatabaseRecord* records = res[0];
  double          E_E = 0, E_P = 0;
  E_E     = records->GetDouble( "BER_PRB" );
  E_P     = records->GetDouble( "BPR_PRB" );
  m_beamE = ( E_E + E_P ) / 2.0;
 
  return;
}