BeamEnergySvc.cxx

Go to the documentation of this file.

#include "GaudiKernel/MsgStream.h"
#include "BeamEnergySvc/BeamEnergySvc.h"
 
BeamEnergySvc::BeamEnergySvc(const std::string& name, ISvcLocator* svcLoc)
: Service(name, svcLoc),
  m_run(-1),
  m_isRunValid(false),
  m_beamE(0)
{
  //declareProperty("", m_);
  declareProperty("Host" ,      m_host = std::string("bes3db2.ihep.ac.cn"));
  declareProperty("DbName" ,    m_dbName = std::string("run"));
  declareProperty("UserName" ,  m_userName = std::string("guest"));
  declareProperty("Password" ,  m_password = std::string("guestpass"));
  declareProperty("PortNum" ,   m_port_num = 3306); 
}
 
BeamEnergySvc::~BeamEnergySvc()
{
}
 
StatusCode BeamEnergySvc::initialize()
{
  MsgStream log(messageService(), name());
  log << MSG::INFO << "@initialize()" << endreq;
 
  StatusCode sc = Service::initialize();
 
  return sc;
}
 
StatusCode BeamEnergySvc::finalize()
{
  MsgStream log(messageService(), name());
  log << MSG::INFO << "@initialize()" << endreq;
 
  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;
}
 
MYSQL* BeamEnergySvc::OpenDb() const {
 
  const char* host     = m_host.c_str() ;
  const char* user     = m_userName.c_str() ;
  const char* passwd   = m_password.c_str() ;
  const char* db       = m_dbName.c_str() ;
  unsigned int port_num = m_port_num;
 
  MYSQL* mysql = mysql_init(NULL);
  mysql = mysql_real_connect(mysql, host, user, passwd, db, port_num,
      NULL,  // socket
      0);    // client_flag
 
  if (mysql == NULL) {
    fprintf(stderr, "can not open database: offlinedb\n");
  }
 
  return mysql;
}
 
 
void BeamEnergySvc::CloseDb(MYSQL* mysql) const {
  mysql_close(mysql);
}
 
 
void BeamEnergySvc::ReadDb(int run ){
  MsgStream log(messageService(), name());
  if ( run < 0 )
  {
    log << MSG::INFO << "This data is the MC sample with the Run Number: " << run << endreq;
    run = -run;
  }
  m_run= run;
  m_isRunValid = false;
 
  MYSQL* mysql=OpenDb();
 
  char stmt[1024];
  snprintf(stmt, 1024,
      "select BER_PRB, BPR_PRB "
      "from RunParams where run_number = %d", run);
  if (mysql_real_query(mysql, stmt, strlen(stmt))) {
    fprintf(stderr, "query error\n");
    return ;
  }
 
 
  MYSQL_RES* result_set = mysql_store_result(mysql);
  MYSQL_ROW row = mysql_fetch_row(result_set);
  if (!row) {
    fprintf(stderr, "cannot find data for RunNo %d\n", run);
    return ;
  }
 
  m_isRunValid = true;
 
  double E_E=0, E_P=0;
  sscanf(row[0], "%lf", &E_E);
  sscanf(row[1], "%lf", &E_P);
 
  m_beamE=(E_E+E_P)/2.0;
 
  // Free the memory to prevent memory leak!
  mysql_free_result(result_set);
 
  CloseDb(mysql);
}
 
bool BeamEnergySvc::isRunValid(int run) {
  if (m_run != run) {
    ReadDb(run);
  }
  return m_isRunValid;
}
 
 
double BeamEnergySvc::getbeamE(int run){
  if (!isRunValid(run)){
    std::cerr << "ERROR in BeamEnergySvc: runNo is invalid!" << std::endl;
  }
  return m_beamE;
}