TofQCorrSvc.cxx

Go to the documentation of this file.

//********************************************************************
// TofQCorrSvc package
// This Service is used for TOF measured pulse height correction.
// For BESIII, from run 7991 to run 10881,
// TOF electronics QTC correction are done online.
// This service is used for convert the corrected Q back to RAW Q
//                      Sun Shengsen   2009-9-17
//********************************************************************
 
#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 <math.h>
 
#include "CalibData/Tof/TofCalibData.h"
#include "CalibData/Tof/bTofCalibBase.h"
#include "EventModel/Event.h"
#include "EventModel/EventHeader.h"
#include "EventModel/EventModel.h"
 
#include "TofQCorrSvc.h"
 
using namespace std;
DECLARE_COMPONENT( TofQCorrSvc )
 
TofQCorrSvc::TofQCorrSvc( const std::string& name, ISvcLocator* svcloc )
    : base_class( name, svcloc ) {}
 
/*StatusCode TofQCorrSvc::queryInterface (const InterfaceID& riid, void** ppvInterface ){
 
  if ( IID_ITofQCorrSvc.versionMatch(riid) ) {
    *ppvInterface = static_cast<ITofQCorrSvc*> (this);
  } else {
    return Service::queryInterface(riid, ppvInterface) ;
  }
  return StatusCode::SUCCESS;
}
*/
 
StatusCode TofQCorrSvc::initialize() {
 
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << name() << ": Start of run initialisation" << 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;
  }
 
  StatusCode scc;
  log << MSG::INFO << "setProperties()" << endmsg;
  //  scc = setProperties();
  m_runNo = 0;
  return sc;
}
 
StatusCode TofQCorrSvc::finalize() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << name() << ": End of Run" << endmsg;
  return StatusCode::SUCCESS;
}
 
TofQCorrSvc::~TofQCorrSvc() {}
 
const double TofQCorrSvc::getBarEast( int layer, int no ) { return m_Bar_East[layer][no]; }
 
const double TofQCorrSvc::getBarWest( int layer, int no ) { return m_Bar_West[layer][no]; }
 
const double TofQCorrSvc::getEndcap( int layer, int no ) { return m_Endcap[layer][no]; }
 
void TofQCorrSvc::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;
    getTofQCorrInfo();
  }
}
 
void TofQCorrSvc::getTofQCorrInfo() {
  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;
    run = -run;
  }
  //  unsigned long *lengths;
  char stmt1[200];
  sprintf( stmt1,
           "select TOFCalib "
           "from RunParams where run_number = %d",
           run );
 
  DatabaseRecordVector res;
  int                  row_no = m_dbsvc->query( "run", stmt1, res );
  if ( !row_no )
  {
    log << MSG::ERROR << "search errror" << endmsg;
    exit( 1 );
  }
  long int        runNo;
  DatabaseRecord* records1 = res[0];
  sscanf( ( *records1 )["TOFCalib"], "%ld", &runNo );
 
  char stmt2[200];
  sprintf( stmt2,
           "select be,end,counter,a1,a2,a3 "
           "from TOF_Q where run_number = %ld",
           runNo );
 
  res.clear();
  row_no = m_dbsvc->query( "run", stmt2, res );
  if ( !row_no )
  {
    log << MSG::ERROR << "search errror" << endmsg;
    exit( 1 );
  }
 
  long int be, end, counter;
  double   sql_result[3];
 
  for ( int jjj = 0; jjj < row_no; jjj++ )
  {
    DatabaseRecord* records2 = res[jjj];
    sscanf( ( *records2 )["be"], "%ld", &be );
    sscanf( ( *records2 )["end"], "%ld", &end );
    sscanf( ( *records2 )["counter"], "%ld", &counter );
    sscanf( ( *records2 )["a1"], "%lf", &sql_result[0] );
    sscanf( ( *records2 )["a2"], "%lf", &sql_result[1] );
    sscanf( ( *records2 )["a3"], "%lf", &sql_result[2] );
    if ( be == 0 && end == 0 )
    {
      m_Endcap[counter][0] = sql_result[0];
      m_Endcap[counter][1] = sql_result[1];
      m_Endcap[counter][2] = sql_result[2];
    }
    if ( be == 1 && end == 0 )
    {
      m_Bar_East[counter][0] = sql_result[0];
      m_Bar_East[counter][1] = sql_result[1];
      m_Bar_East[counter][2] = sql_result[2];
    }
    if ( be == 1 && end == 1 )
    {
      m_Bar_West[counter][0] = sql_result[0];
      m_Bar_West[counter][1] = sql_result[1];
      m_Bar_West[counter][2] = sql_result[2];
    }
  }
  res.clear();
}
 
const double TofQCorrSvc::BQRaw1( int id, double q ) {
  MsgStream log( msgSvc(), name() );
  double    qraw = -999.0;
  if ( id < 0 || id > 175 )
  {
    log << MSG::WARNING << "BQRaw1 :  TofId is out of range, tofid = " << id << endmsg;
    return qraw;
  }
  if ( fabs( q - 10000.0 ) < 1.0e-6 ) { return 10000.0; }
  else if ( fabs( q + 999.0 ) < 1.0e-6 ) { return -999.0; }
  double a = m_Bar_East[id][2];
  double b = m_Bar_East[id][1];
  double c = m_Bar_East[id][0] - q / 2.5;
  qraw     = ( sqrt( b * b - 4.0 * a * c ) - b ) / ( 2.0 * a );
 
  return qraw;
}
 
const double TofQCorrSvc::BQRaw2( int id, double q ) {
  MsgStream log( msgSvc(), name() );
  double    qraw = -999.0;
  if ( id < 0 || id > 175 )
  {
    log << MSG::WARNING << "BQRaw2 :  TofId is out of range, tofid = " << id << endmsg;
    return qraw;
  }
  if ( fabs( q - 10000.0 ) < 1.0e-6 ) { return 10000.0; }
  else if ( fabs( q + 999.0 ) < 1.0e-6 ) { return -999.0; }
  double a = m_Bar_West[id][2];
  double b = m_Bar_West[id][1];
  double c = m_Bar_West[id][0] - q / 2.5;
  qraw     = ( sqrt( b * b - 4.0 * a * c ) - b ) / ( 2.0 * a );
  return qraw;
}
 
const double TofQCorrSvc::EQRaw( int id, double q ) {
  MsgStream log( msgSvc(), name() );
  double    qraw = -999.0;
  if ( id < 0 || id > 95 )
  {
    log << MSG::WARNING << "EQRaw :  TofId is out of range, tofid = " << id << endmsg;
    return qraw;
  }
  if ( fabs( q - 10000.0 ) < 1.0e-6 ) { return 10000.0; }
  else if ( fabs( q + 999.0 ) < 1.0e-6 ) { return -999.0; }
  double a = m_Endcap[id][2];
  double b = m_Endcap[id][1];
  double c = m_Endcap[id][0] - q / 2.5;
  qraw     = ( sqrt( b * b - 4.0 * a * c ) - b ) / ( 2.0 * a );
  return qraw;
}