DedxCurSvc.cxx

Go to the documentation of this file.

//********************************************************************
// DedxCurSvc 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 "DedxCurSvc.h"
#include "EventModel/Event.h"
#include "EventModel/EventHeader.h"
#include "EventModel/EventModel.h"
#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 "TBuffer.h"
#include "TBufferFile.h"
#include "TTree.h"
 
#include <math.h>
 
using namespace std;
 
DECLARE_COMPONENT( DedxCurSvc )
 
DedxCurSvc::DedxCurSvc( const std::string& name, ISvcLocator* svcloc )
    : base_class( name, svcloc ) {
  declareProperty( "BossRelease", m_bossRelease = "default" );
  declareProperty( "BossVer", m_sftver = "default" );
  declareProperty( "ParVer", m_calParVer = "default" );
  declareProperty( "DbStatus", m_dbStatus = "OK" );
  declareProperty( "Type", m_type = "default" );
}
 
/*StatusCode DedxCurSvc::queryInterface (const InterfaceID& riid, void** ppvInterface ){
 
  if ( IID_IDedxCurSvc.versionMatch(riid) ) {
    *ppvInterface = static_cast<IDedxCurSvc*> (this);
  }
  else return Service::queryInterface(riid, ppvInterface) ;
  return StatusCode::SUCCESS;
}
 
*/
StatusCode DedxCurSvc::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_runFromMax = 0;
  m_runToMin   = 0;
 
  return sc;
}
 
StatusCode DedxCurSvc::finalize() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << name() << ": End of Run" << endmsg;
  return StatusCode::SUCCESS;
}
 
DedxCurSvc::~DedxCurSvc() {}
 
void DedxCurSvc::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 ( inc.type() == "NewRun" )
  {
    log << MSG::DEBUG << "New Run:" << run << endmsg;
    getDedxCurveInfo();
  }
}
 
void DedxCurSvc::getDedxCurveInfo() {
  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;
    if ( m_type == "Sim" )
    {
      run = -run;
      log << MSG::INFO << "Reverse the sign of Run Number" << endmsg;
    }
  }
  //  unsigned long *lengths;
  MYSQL_RES* res_set;
  MYSQL_ROW  row;
  char       stmt1[400];
 
  std::string sftver;
 
  if ( m_sftver == "default" && m_bossRelease == "default" )
  {
    log << MSG::FATAL << " BossRelease and BossVer for DedxCurve not set!" << endmsg;
    exit( 1 );
  }
 
  if ( ( run < m_runFromMax || run > m_runToMin ) )
  {
    if ( m_sftver == "default" )
    {
      sprintf( stmt1,
               "select RunFrom, RunTo, SftVer, ParVer from CalVtxLumVer where BossRelease = "
               "'%s' and RunFrom <= %d and RunTo >= %d and DataType = 'DedxCurve'",
               m_bossRelease.c_str(), run, run );
      std::cout << stmt1 << std::endl;
      DatabaseRecordVector rest;
      int                  row_no = m_dbsvc->query( "offlinedb", stmt1, rest );
      if ( row_no == 0 )
      {
        log << MSG::FATAL << "can not find result for DedxCur with: " << stmt1 << endmsg;
        exit( 1 );
      }
      if ( row_no > 1 )
      {
        log << MSG::FATAL << "find more than 1 results for DedxCur with: " << stmt1 << endmsg;
        exit( 1 );
      }
      DatabaseRecord* recordst = rest[0];
      sftver                   = recordst->GetString( "SftVer" );
      m_calParVer              = recordst->GetString( "ParVer" );
      m_runFromMax             = atoi( ( recordst->GetString( "RunFrom" ) ).c_str() );
      m_runToMin               = atoi( ( recordst->GetString( "RunTo" ) ).c_str() );
      log << MSG::INFO << "get from CalVtxLumVer, m_runFromMax: " << m_runFromMax
          << " m_runToMin: " << m_runToMin << endmsg;
    }
    else sftver = m_sftver;
 
    if ( m_calParVer != "default" )
      sprintf( stmt1,
               "select RunFrom, RunTo, DedxCurvePar,DedxSigmaPar, SftVer, CurveFileName from "
               "DedxCurvePar where SftVer = '%s' and RunFrom <= %d and RunTo >= %d and "
               "DedxCurveParVer = %s and Status='%s'",
               sftver.c_str(), run, run, m_calParVer.c_str(), m_dbStatus.c_str() );
    else
      sprintf( stmt1,
               "select RunFrom, RunTo, DedxCurvePar,DedxSigmaPar, SftVer, CurveFileName from "
               "DedxCurvePar where SftVer = '%s' and RunFrom <= %d and RunTo >= %d and "
               "Status='%s' order by DedxCurveParVer desc",
               sftver.c_str(), run, run, m_dbStatus.c_str() );
    std::cout << stmt1 << std::endl;
 
    DatabaseRecordVector res;
    int                  row_no = m_dbsvc->query( "offlinedb", stmt1, res );
    if ( row_no == 0 )
    {
      log << MSG::FATAL << "can not find result for DedxCur with: " << stmt1 << endmsg;
      exit( 1 );
    }
    if ( row_no > 1 )
    {
      log << MSG::FATAL << "find more than 1 results for DedxCur with: " << stmt1 << endmsg;
      exit( 1 );
    }
 
    DatabaseRecord* records = res[0];
    int             runFrom = atoi( ( records->GetString( "RunFrom" ) ).c_str() );
    int             runTo   = atoi( ( records->GetString( "RunTo" ) ).c_str() );
    std::cout << "before output DedxCurveService version" << endl;
    std::cout << " SftVer is "
              << records->GetString( "SftVer" )
              //         << " CalVerSft is "<<records->GetString("CalParVer")
              << " File name is " << records->GetString( "CurveFileName" ) << std::endl;
    std::cout << "after output DedxCurveService version" << endl;
 
    if ( m_sftver == "default" )
    {
      m_runFromMax = runFrom > m_runFromMax ? runFrom : m_runFromMax;
      m_runToMin   = runTo < m_runToMin ? runTo : m_runToMin;
    }
    else
    {
      m_runFromMax = runFrom;
      m_runToMin   = runTo;
    }
    log << MSG::INFO << "get from DedxCurvePar runFrom: " << runFrom << " runTo: " << runTo
        << endmsg;
    log << MSG::INFO << "m_runFromMax: " << m_runFromMax << " m_runToMin: " << m_runToMin
        << endmsg;
 
    TBuffer* buf1 = new TBufferFile( TBuffer::kRead );
    buf1->SetBuffer( ( *records )["DedxCurvePar"], 327680, kFALSE );
    TBuffer* buf2 = new TBufferFile( TBuffer::kRead );
    buf2->SetBuffer( ( *records )["DedxSigmaPar"], 327680, kFALSE );
 
    TTree* curvetree = new TTree();
    curvetree->Streamer( *buf1 );
 
    TTree* sigmatree = new TTree();
    sigmatree->Streamer( *buf2 );
 
    double curve[50];
    double sigma[50];
    int    SigmaSize;
    int    CurveSize;
    curvetree->SetBranchAddress( "curve", curve );
    curvetree->SetBranchAddress( "CurveSize", &CurveSize );
    sigmatree->SetBranchAddress( "sigma", sigma );
    sigmatree->SetBranchAddress( "SigmaSize", &SigmaSize );
 
    Int_t nentries_curve = (Int_t)curvetree->GetEntries();
    Int_t nentries_sigma = (Int_t)sigmatree->GetEntries();
    curvetree->GetEntry( 0 );
    if ( CurveSize > 50 )
    {
      log << MSG::ERROR << "CurveSize larger than designed number" << endmsg;
      exit( 1 );
    }
    for ( int i = 0; i < CurveSize; i++ ) { m_curve[i] = curve[i]; }
 
    sigmatree->GetEntry( 0 );
    if ( SigmaSize > 50 )
    {
      log << MSG::ERROR << "SigmaSize larger than designed number" << endmsg;
      exit( 1 );
    }
    for ( int i = 0; i < SigmaSize; i++ ) { m_sigma[i] = sigma[i]; }
    m_curve_size = CurveSize;
    m_sigma_size = SigmaSize;
    delete curvetree;
    delete sigmatree;
  }
  return;
}