/home/bes3soft/bes3soft/Boss/7.0.2/dist/7.0.2/Reconstruction/DTagAlg/DTagAlg-00-01-04/src/ReadBeamInfFromDb.cxx

Go to the documentation of this file.

#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/IJobOptionsSvc.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/PropertyMgr.h"


#include "DTagAlg/ReadBeamInfFromDb.h"
#include <string>
#include <string.h>
ReadBeamInfFromDb::ReadBeamInfFromDb() :
  m_run(-1),
  m_isRunValid(false),
  m_beamE(0),
  m_beta(0.011,0,0),
  m_usecbE(false){
}

MYSQL* ReadBeamInfFromDb::OpenDb() const {
  
  const char host[]     = "bes3db2.ihep.ac.cn";
  const char user[]     = "guest";
  const char passwd[]   = "guestpass";
  //const char db[]       = "run";
  const char db[]       = "offlinedb";
  unsigned int port_num = 3306;

  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 ReadBeamInfFromDb::CloseDb(MYSQL* mysql) const {
  mysql_close(mysql);
}


double ReadBeamInfFromDb::ReadDb(int run){
  
  m_run=run;
  m_isRunValid = false;
  
  //read db directly
  //turned off now
  /*
  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");
    CloseDb(mysql);
    return m_beamE ;
  }
  
  
  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);
    mysql_free_result(result_set);
    CloseDb(mysql);
    return m_beamE;
  }
  


  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);
  */
  
  //read db use service
  Gaudi::svcLocator()->service("DatabaseSvc",m_dbsvc,true);
  //calibrated beam Energy
  if(m_usecbE){
    char stmt1[400];
    snprintf(stmt1, 1024,
             "select beam_energy, px, py, pz "
             "from RunParams where run_number = %d", run);
    DatabaseRecordVector res;
    int row_no = m_dbsvc->query("offlinedb", stmt1, res);
    if(row_no==0){
      return m_beamE;
    }
    
    DatabaseRecord* records = res[0];
    double bE=0;
    bE = records->GetDouble("beam_energy");
    m_beamE=bE;

    double px=records->GetDouble("px");
    double py=records->GetDouble("py");
    double pz=records->GetDouble("pz");

    m_beta.setX(px);
    m_beta.setY(py);
    m_beta.setZ(pz);
    
    //std::cout<<"beam e is:"<<bE<<", px="<<px<<",py="<<py<<",pz="<<pz<<std::endl;
  }
  //use online beam Energy 
  else{
    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){
      return m_beamE;
    }
    
    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;
  }

  
  m_isRunValid = true;
  return m_beamE;
}

bool ReadBeamInfFromDb::isRunValid(int run) {
  
  if (run == -1 || m_run != run) {
    ReadDb(run);
    return m_isRunValid;
  }
  
  return false;
  
}


double ReadBeamInfFromDb::getbeamE(int run, double defaultbeamE){
  int absrun=fabs(run);
  
  if (!isRunValid(absrun)){
    return defaultbeamE;
    fprintf(stderr, "ERROR in ReadBeamInfFromDb: runNo is invalid!\n");
  }
  return m_beamE;
}

---