TofEnergyCalibSvc.cxx

Go to the documentation of this file.

//     This service is used to read the TofEnergyCalibSvc  information from the database
//
//     an example to use this service is shown in test/test_read.cxx
//
//    the joboption for the example is shown in share/job-test.txt
//
//     the joboption for this service is shown in  share/jobOptions_TofEnergyCalibSvc.txt
 
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <sstream>
 
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/IInterface.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/Kernel.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/PropertyMgr.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "GaudiKernel/StatusCode.h"
#include "TofEnergyCalibSvc.h"
 
#include "GaudiKernel/IIncidentListener.h"
#include "GaudiKernel/IIncidentSvc.h"
#include "GaudiKernel/Incident.h"
 
#include "EventModel/Event.h"
#include "EventModel/EventHeader.h"
#include "EventModel/EventModel.h"
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/ISvcLocator.h"
 
#include <TMath.h>
 
using namespace std;
 
DECLARE_COMPONENT( TofEnergyCalibSvc )
TofEnergyCalibSvc::TofEnergyCalibSvc( const string& name, ISvcLocator* svcloc )
    : base_class( name, svcloc ) {
  // declare properties
  declareProperty( "Host", host = std::string( "bes3db2.ihep.ac.cn" ) );
  declareProperty( "DbName", dbName = std::string( "offlinedb" ) );
  declareProperty( "UserName", userName = std::string( "guest" ) );
  declareProperty( "Password", password = std::string( "guestpass" ) );
  declareProperty( "IsData", m_isData = std::string( "default" ) );
  declareProperty( "BossRelease", m_bossRelease = std::string( "default" ) );
  declareProperty( "TofEnergyPar", m_calPar = "default" );
  declareProperty( "TofEnergyBossVer", m_bossVer = "default" );
  //    m_dbsvc = DatabaseSvc::instance();
}
 
TofEnergyCalibSvc::~TofEnergyCalibSvc() {}
 
StatusCode TofEnergyCalibSvc::initialize() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "TofEnergyCalibSvc::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;
  }
 
  return StatusCode::SUCCESS;
}
 
StatusCode TofEnergyCalibSvc::finalize() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "TofEnergyCalibSvc::finalize()" << endmsg;
  //  if(m_connect_offline) delete m_connect_offline;
  return StatusCode::SUCCESS;
}
 
void TofEnergyCalibSvc::handle( const Incident& inc ) {
  MsgStream log( msgSvc(), name() );
  log << MSG::DEBUG << "handle: " << inc.type() << endmsg;
 
  if ( inc.type() == "NewRun" )
  {
    log << MSG::DEBUG << "NewRun" << endmsg;
    if ( !getTofEnergyCalibSvcInfo() )
    { log << MSG::ERROR << "can not initilize Tof energy Calib Constants" << endmsg; }
  }
}
 
// select sftver and parver frm table CalVtxLum
StatusCode TofEnergyCalibSvc::getSftParVer( std::string& SftVer, std::string& CalParVer,
                                            int& runfrm, int& runto, int RunNo,
                                            std::string BossRelease, std::string DataType ) {
  MsgStream            log( msgSvc(), "TofEnergyCalibSvc" );
  char                 stmt[300];
  int                  run_No = RunNo;
  MYSQL_RES*           res_set;
  IDatabaseSvc*        m_dbsvc;
  DatabaseRecordVector res;
  StatusCode           sc = serviceLocator()->service( "DatabaseSvc", m_dbsvc, true );
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << " Unable to find DatabaseSvc " << endmsg;
    return sc;
  }
 
  const char* bossRelease = BossRelease.c_str();
  const char* dataType    = DataType.c_str();
 
  sprintf( stmt,
           "select RunFrom,RunTo,SftVer,ParVer from CalVtxLumVer where BossRelease = '%s' and "
           "RunFrom <= %d and RunTo "
           ">= %d and DataType='%s' ",
           bossRelease, run_No, run_No, dataType );
 
  int row_no = m_dbsvc->query( "offlinedb", stmt, res );
  if ( row_no < 1 )
  {
    std::cout << "ERROR:error searching with:" << stmt << std::endl;
    exit( 1 );
  }
  if ( row_no == 1 )
  {
    DatabaseRecord* records1 = res[0];
    runfrm                   = records1->GetInt( "RunFrom" );
    runto                    = records1->GetInt( "RunTo" );
    cout << dataType << " runfrm,runto in getSftParVer is:" << runfrm << "::::" << runto
         << endl;
    SftVer    = records1->GetString( "SftVer" );
    CalParVer = records1->GetString( "ParVer" );
  }
  if ( row_no > 1 )
  {
    cout << "ERROR: " << dataType << " set overlapped run number in the table CalVtxLumVer"
         << endl;
    exit( 1 );
  }
  return StatusCode::SUCCESS;
}
bool TofEnergyCalibSvc::getTofEnergyCalibSvcInfo() {
  MsgStream                        log( msgSvc(), name() );
  SmartDataPtr<Event::EventHeader> eventHeader( m_eventSvc, "/Event/EventHeader" );
  int                              run = eventHeader->runNumber();
  // cout<<"Run in getTofEnergyCalibSvcInfo() is: "<<run<<endl;
 
  char       stmt1[400];
  StatusCode st1;
  // runfrm,runto are the value from the table ***CalConst;runfrm1,runto1 are the value from
  // the table CalVtxLumVer
  int runfrm, runfrm1;
  int runto, runto1;
std:
  string cType;
 
  if ( run < 0 )
  {
    cout << "This data is the MC sample with the Run Number: " << run << endl;
    run = -run;
  }
 
  if ( m_bossRelease == "default" ) m_bossRelease = getenv( "BES_RELEASE" );
  if ( m_isData == "default" )
  {
    cout << "Please set the type of TofEnergy" << endl;
    exit( 1 );
  }
  if ( m_isData == "Data" ) cType = "TofEnergyData";
  if ( m_isData == "MC" ) cType = "TofEnergyMC";
  st1 = getSftParVer( m_bossVer, m_calPar, runfrm1, runto1, run, m_bossRelease, cType );
  if ( cType == "TofEnergyData" )
  {
    sprintf( stmt1,
             "select CalibConst,Para1,Para2,Para3,Para4,Para5 from TofEnergyCalib where "
             "BossVer = '%s' and RunFrom <= "
             "%d and RunTo >= %d and isData = '%s'",
             m_bossVer.c_str(), run, run, m_isData.c_str() );
  }
  else if ( cType == "TofEnergyMC" )
  {
    sprintf( stmt1,
             "select CalibConst,Para1,Para2,Para3,Para4,Para5 from TofEnergySim where BossVer "
             "= '%s' and RunFrom <= %d "
             "and RunTo >= %d and isData = '%s'",
             m_bossVer.c_str(), run, run, m_isData.c_str() );
  }
  // cout<<"stmt is:"<<stmt1<<endl;
  DatabaseRecordVector res;
  int                  row_no = m_dbsvc->query( "offlinedb", stmt1, res );
  if ( row_no > 0 )
  {
    DatabaseRecord& dbrec = *res[row_no - 1];
    m_CalibConst          = dbrec.GetDouble( "CalibConst" );
    m_Para1               = dbrec.GetDouble( "Para1" );
    m_Para2               = dbrec.GetDouble( "Para2" );
    m_Para3               = dbrec.GetDouble( "Para3" );
    m_Para4               = dbrec.GetDouble( "Para4" );
    m_Para5               = dbrec.GetDouble( "Para5" );
    cout << stmt1 << endl;
    cout << "BossReleaseVer=" << m_bossRelease.c_str() << "\t"
         << "TofEnergyBossVer=" << m_bossVer.c_str() << endl;
    cout << "m_CalibConst is:" << m_CalibConst << endl;
    cout << "Para1=" << m_Para1 << "\t"
         << "Para2=" << m_Para2 << "\t"
         << "Para3=" << m_Para3 << "\t"
         << "Para4=" << m_Para4 << "\t"
         << "Para5=" << m_Para5 << endl;
    cout << "Successfully fetch TofEnergyCalibSvc information for run: " << run << endl;
    return true;
  }
  else if ( row_no <= 0 )
  {
    cout << " TofEnergyCalibSvc:: can not found TofEnergyCalibSvc information of run:" << run
         << ", boss version " << m_bossRelease << endl;
    exit( 1 );
    return false;
  }
  return true;
}