TreeTofCalibDataCnv.cxx

Go to the documentation of this file.

#include "TreeTofCalibDataCnv.h"
#include "CalibData/Tof/TofCalibData.h"
#include "CalibDataSvc/IInstrumentName.h"
#include "CalibMySQLCnvSvc/TreeAddress.h"
#include "GaudiKernel/MsgStream.h"
#include "TBuffer.h"
#include "TDirectory.h"
#include "TFile.h"
#include "TObject.h"
#include "TTree.h"
 
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/GenericAddress.h"
#include "GaudiKernel/IAddressCreator.h"
#include "GaudiKernel/IConversionSvc.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/IOpaqueAddress.h"
 
#include "CalibDataSvc/ICalibMetaCnvSvc.h"
#include "CalibDataSvc/ICalibTreeSvc.h" //maybe
 
// Temporary.  Hope to find a better way to do this
#include "CalibData/CalibModel.h"
using namespace CalibData;
// static CnvFactory<TreeTofCalibDataCnv> TofCal_factory;
// const  ICnvFactory& TreeTofCalibDataCnvFactory = TofCal_factory;
 
const unsigned int nBarAtten  = 8;
const unsigned int nBarSpeed  = 2;
const unsigned int nBarPar    = 10;
const unsigned int nBarParOff = 20;
const unsigned int nBarSigma  = 5;
const unsigned int nBarSigCnt = 10;
 
const unsigned int nBarSigCor = 8;
const unsigned int nBarOffset = 2;
 
const unsigned int nEndAtten = 5;
const unsigned int nEndSpeed = 4;
const unsigned int nEndPar   = 8;
const unsigned int nEndSigma = 5;
 
const unsigned int nEtfSpeed = 4;
const unsigned int nEtfPar   = 20;
const unsigned int nEtfBunch = 4;
 
TreeTofCalibDataCnv::TreeTofCalibDataCnv( ISvcLocator* svc )
    : TreeCalBaseCnv( svc, CLID_Calib_TofCal ) {}
 
const CLID& TreeTofCalibDataCnv::objType() const { return CLID_Calib_TofCal; }
 
const CLID& TreeTofCalibDataCnv::classID() { return CLID_Calib_TofCal; }
 
StatusCode TreeTofCalibDataCnv::i_createObj( IOpaqueAddress* addr, DataObject*& refpObject ) {
 
  MsgStream log( msgSvc(), "TreeTofCalibDataCnv" );
  log << MSG::DEBUG << "SetProperty" << endmsg;
 
  TreeAddress* add = dynamic_cast<TreeAddress*>( addr );
 
  DatabaseRecord* records = add->pp();
 
  TBufferFile* buf1 = new TBufferFile( TBuffer::kRead );
  TBufferFile* buf2 = new TBufferFile( TBuffer::kRead );
  TBufferFile* buf3 = new TBufferFile( TBuffer::kRead );
  TBufferFile* buf4 = new TBufferFile( TBuffer::kRead );
  TBufferFile* buf5 = 0;
  TBufferFile* buf6 = 0;
 
  buf1->SetBuffer( ( *records )["BarTofPar"], 2048000, kFALSE );
  buf2->SetBuffer( ( *records )["EndTofPar"], 2048000, kFALSE );
  buf3->SetBuffer( ( *records )["TofConPar"], 2048000, kFALSE );
  buf4->SetBuffer( ( *records )["BarTof"], 2048000, kFALSE );
  if ( ( *records )["EtfTofPar"] )
  {
    buf5 = new TBufferFile( TBuffer::kRead );
    buf5->SetBuffer( ( *records )["EtfTofPar"], 2048000, kFALSE );
  }
  if ( ( *records )["EtfTofBunch"] )
  {
    buf6 = new TBufferFile( TBuffer::kRead );
    buf6->SetBuffer( ( *records )["EtfTofBunch"], 2048000, kFALSE );
  }
 
  std::cout << " SftVer is " << ( *records )["SftVer"];
  std::cout << " CalVerSft is " << ( *records )["CalParVer"];
  std::cout << "  File name  is " << ( *records )["FileName"] << std::endl;
 
  TTree* btoftree       = new TTree();
  TTree* etoftree       = new TTree();
  TTree* etftree        = new TTree();
  TTree* etfbunchtree   = new TTree();
  TTree* btofcommontree = new TTree();
  TTree* CalibInfo      = new TTree();
 
  btoftree->Streamer( *buf1 );
  etoftree->Streamer( *buf2 );
  btofcommontree->Streamer( *buf3 );
  CalibInfo->Streamer( *buf4 );
  if ( buf5 ) { etftree->Streamer( *buf5 ); }
 
  if ( buf6 ) { etfbunchtree->Streamer( *buf6 ); }
 
  CalibData::bTofCalibBase       bTof;
  CalibData::eTofCalibBase       eTof;
  CalibData::etfCalibBase        etf;
  CalibData::etfBunchCalibBase   etfBunch;
  CalibData::bTofCommonCalibBase bTofCommon;
  CalibData::tofCalibInfoBase    tofinfo;
 
  std::vector<CalibData::bTofCalibBase>       tmpbTof;
  std::vector<CalibData::eTofCalibBase>       tmpeTof;
  std::vector<CalibData::etfCalibBase>        tmpetf;
  std::vector<CalibData::etfBunchCalibBase>   tmpetfBunch;
  std::vector<CalibData::bTofCommonCalibBase> tmpbTofCommon;
  std::vector<CalibData::tofCalibInfoBase>    tofinfoCol;
 
  // Read in the object
  int cnt;
  // read btoftree ------------------------------------------------------------
  double cnvBarAtten[nBarAtten];
  double cnvBarSpeed[nBarSpeed];
  double cnvBarPar1[nBarPar];
  double cnvBarPar2[nBarPar];
  double cnvBarParOff1_bunch0[nBarParOff];
  double cnvBarParOff2_bunch0[nBarParOff];
  double cnvBarParOff1_bunch1[nBarParOff];
  double cnvBarParOff2_bunch1[nBarParOff];
  double cnvBarParOff1_bunch2[nBarParOff];
  double cnvBarParOff2_bunch2[nBarParOff];
  double cnvBarParOff1_bunch3[nBarParOff];
  double cnvBarParOff2_bunch3[nBarParOff];
  double cnvFLeft[nBarSigma];
  double cnvFRight[nBarSigma];
  double cnvFCount[nBarSigCnt];
 
  char brname[10];
  for ( unsigned int i = 0; i < nBarAtten; i++ )
  {
    sprintf( brname, "Atten%i", i );
    btoftree->SetBranchAddress( brname, &cnvBarAtten[i] );
  }
  for ( unsigned int i = 0; i < nBarSpeed; i++ )
  {
    sprintf( brname, "Speed%i", i );
    btoftree->SetBranchAddress( brname, &cnvBarSpeed[i] );
  }
  for ( unsigned int i = 0; i < nBarPar; i++ )
  {
    sprintf( brname, "P%i", i );
    btoftree->SetBranchAddress( brname, &cnvBarPar1[i] );
  }
  for ( unsigned int i = 0; i < nBarPar; i++ )
  {
    sprintf( brname, "P%i", i + nBarPar );
    btoftree->SetBranchAddress( brname, &cnvBarPar2[i] );
  }
  for ( unsigned int i = 0; i < nBarParOff; i++ )
  {
    sprintf( brname, "Bunch0_Poff%i", i );
    btoftree->SetBranchAddress( brname, &cnvBarParOff1_bunch0[i] );
  }
  for ( unsigned int i = 0; i < nBarParOff; i++ )
  {
    sprintf( brname, "Bunch0_Poff%i", i + nBarParOff );
    btoftree->SetBranchAddress( brname, &cnvBarParOff2_bunch0[i] );
  }
  for ( unsigned int i = 0; i < nBarParOff; i++ )
  {
    sprintf( brname, "Bunch1_Poff%i", i );
    btoftree->SetBranchAddress( brname, &cnvBarParOff1_bunch1[i] );
  }
  for ( unsigned int i = 0; i < nBarParOff; i++ )
  {
    sprintf( brname, "Bunch1_Poff%i", i + nBarParOff );
    btoftree->SetBranchAddress( brname, &cnvBarParOff2_bunch1[i] );
  }
  for ( unsigned int i = 0; i < nBarParOff; i++ )
  {
    sprintf( brname, "Bunch2_Poff%i", i );
    btoftree->SetBranchAddress( brname, &cnvBarParOff1_bunch2[i] );
  }
  for ( unsigned int i = 0; i < nBarParOff; i++ )
  {
    sprintf( brname, "Bunch2_Poff%i", i + nBarParOff );
    btoftree->SetBranchAddress( brname, &cnvBarParOff2_bunch2[i] );
  }
  for ( unsigned int i = 0; i < nBarParOff; i++ )
  {
    sprintf( brname, "Bunch3_Poff%i", i );
    btoftree->SetBranchAddress( brname, &cnvBarParOff1_bunch3[i] );
  }
  for ( unsigned int i = 0; i < nBarParOff; i++ )
  {
    sprintf( brname, "Bunch3_Poff%i", i + nBarParOff );
    btoftree->SetBranchAddress( brname, &cnvBarParOff2_bunch3[i] );
  }
  for ( unsigned int i = 0; i < nBarSigma; i++ )
  {
    sprintf( brname, "FLeft%i", i );
    btoftree->SetBranchAddress( brname, &cnvFLeft[i] );
  }
  for ( unsigned int i = 0; i < nBarSigma; i++ )
  {
    sprintf( brname, "FRight%i", i );
    btoftree->SetBranchAddress( brname, &cnvFRight[i] );
  }
  for ( unsigned int i = 0; i < nBarSigCnt; i++ )
  {
    sprintf( brname, "FCounter%i", i );
    btoftree->SetBranchAddress( brname, &cnvFCount[i] );
  }
 
  unsigned int btofN =
      btoftree->GetEntries(); // Modify for n flies   176 ==> btoftree->GetEntries() wensp
  for ( cnt = 0; cnt < btofN; cnt++ )
  {
    btoftree->GetEntry( cnt );
    bTof.setAtten( cnvBarAtten );
    bTof.setSpeed( cnvBarSpeed );
    bTof.setP1( cnvBarPar1 );
    bTof.setP2( cnvBarPar2 );
    bTof.setPoff1_bunch0( cnvBarParOff1_bunch0 );
    bTof.setPoff2_bunch0( cnvBarParOff2_bunch0 );
    bTof.setPoff1_bunch1( cnvBarParOff1_bunch1 );
    bTof.setPoff2_bunch1( cnvBarParOff2_bunch1 );
    bTof.setPoff1_bunch2( cnvBarParOff1_bunch2 );
    bTof.setPoff2_bunch2( cnvBarParOff2_bunch2 );
    bTof.setPoff1_bunch3( cnvBarParOff1_bunch3 );
    bTof.setPoff2_bunch3( cnvBarParOff2_bunch3 );
    bTof.setFPleft( cnvFLeft );
    bTof.setFPright( cnvFRight );
    bTof.setFPcounter( cnvFCount );
    tmpbTof.push_back( bTof );
  }
 
  // read etoftree
  double cnvEndAtten[nEndAtten];
  double cnvEndSpeed[nEndSpeed];
  double cnvEndPar[nEndPar];
  double cnvEndFPar[nEndSigma];
 
  char ecname[10];
  for ( unsigned int i = 0; i < nEndAtten; i++ )
  {
    sprintf( ecname, "Atten%i", i );
    etoftree->SetBranchAddress( ecname, &cnvEndAtten[i] );
  }
  for ( unsigned int i = 0; i < nEndSpeed; i++ )
  {
    sprintf( ecname, "Speed%i", i );
    etoftree->SetBranchAddress( ecname, &cnvEndSpeed[i] );
  }
  for ( unsigned int i = 0; i < nEndPar; i++ )
  {
    sprintf( ecname, "P%i", i );
    etoftree->SetBranchAddress( ecname, &cnvEndPar[i] );
  }
  for ( unsigned int i = 0; i < nEndSigma; i++ )
  {
    sprintf( ecname, "FCounter%i", i );
    etoftree->SetBranchAddress( ecname, &cnvEndFPar[i] );
  }
 
  unsigned int etofN =
      etoftree->GetEntries(); // Modify for n flies   96 ==> etoftree->GetEntries() wensp
  for ( cnt = 0; cnt < etofN; cnt++ )
  {
    etoftree->GetEntry( cnt );
    eTof.setAtten( cnvEndAtten );
    eTof.setSpeed( cnvEndSpeed );
    eTof.setP( cnvEndPar );
    eTof.setFP( cnvEndFPar );
    tmpeTof.push_back( eTof );
  }
 
  // read etftree
  if ( etftree->GetEntries() != 0 )
  {
 
    double cnvEtfSpeed[nEtfSpeed];
    double cnvEtfPar[nEtfPar];
    double cnvEtfPar1[nEtfPar];
    double cnvEtfPar2[nEtfPar];
 
    char etfname[10];
    for ( unsigned int i = 0; i < nEtfSpeed; i++ )
    {
      sprintf( etfname, "Speed%i", i );
      etftree->SetBranchAddress( etfname, &cnvEtfSpeed[i] );
    }
    for ( unsigned int i = 0; i < nEtfPar; i++ )
    {
      sprintf( etfname, "P%i", i );
      etftree->SetBranchAddress( etfname, &cnvEtfPar[i] );
      sprintf( etfname, "P%i", nEtfPar + i );
      etftree->SetBranchAddress( etfname, &cnvEtfPar1[i] );
      sprintf( etfname, "P%i", 2 * nEtfPar + i );
      etftree->SetBranchAddress( etfname, &cnvEtfPar2[i] );
    }
 
    unsigned int etfN =
        etftree->GetEntries(); // Modify for n flies   72*12 ==> etftree->GetEntries() wensp
    for ( cnt = 0; cnt < etfN; cnt++ )
    {
      etftree->GetEntry( cnt );
      etf.setSpeed( cnvEtfSpeed );
      etf.setP( cnvEtfPar );
      etf.setP1( cnvEtfPar1 );
      etf.setP2( cnvEtfPar2 );
      tmpetf.push_back( etf );
    }
  }
 
  // read etftree
  if ( etfbunchtree->GetEntries() != 0 )
  {
    double cnvEtfBunchP[nEtfBunch];
    char   etfbunchname[10];
    for ( unsigned int i = 0; i < nEtfBunch; i++ )
    {
      sprintf( etfbunchname, "pbunch%i", i );
      etfbunchtree->SetBranchAddress( etfbunchname, &cnvEtfBunchP[i] );
    }
    int entries = etfbunchtree->GetEntries();
    for ( cnt = 0; cnt < entries; cnt++ )
    {
      etfbunchtree->GetEntry( cnt );
      etfBunch.setPBunch( cnvEtfBunchP );
      tmpetfBunch.push_back( etfBunch );
    }
  }
 
  // read bTofCommonCalibBase
  double cnvBarSigCor[nBarSigCor];
  double cnvBarOffset[nBarOffset];
  for ( unsigned int i = 0; i < nBarSigCor; i++ )
  {
    sprintf( brname, "sigmaCorr%i", i );
    btofcommontree->SetBranchAddress( brname, &cnvBarSigCor[i] );
  }
  for ( unsigned int i = 0; i < nBarOffset; i++ )
  {
    sprintf( brname, "t0offset%i", i );
    btofcommontree->SetBranchAddress( brname, &cnvBarOffset[i] );
  }
 
  int entries = btofcommontree->GetEntries();
  for ( cnt = 0; cnt < entries; cnt++ )
  {
    btofcommontree->GetEntry( cnt );
    bTofCommon.setSigmaCorr( cnvBarSigCor );
    bTofCommon.setOffset( cnvBarOffset );
    tmpbTofCommon.push_back( bTofCommon );
  }
 
  int m_runFrom, m_runTo, m_eventFrom, m_eventTo; // Modify for n files, wensp
  int m_run1, m_run2, m_version;
  int m_qCorr, m_qElec, m_misLable;
  int m_tofidEast[5], m_tofidWest[5], m_tofidEndcap[5];
 
  if ( CalibInfo->GetBranchStatus( "runFrom" ) )
  {
    CalibInfo->SetBranchAddress( "runFrom", &m_runFrom );
    CalibInfo->SetBranchAddress( "runTo", &m_runTo );
    CalibInfo->SetBranchAddress( "eventFrom", &m_eventFrom );
    CalibInfo->SetBranchAddress( "eventTo", &m_eventTo );
  }
  else
  {
    m_runFrom   = -1;
    m_runTo     = -1;
    m_eventFrom = -1;
    m_eventTo   = -1;
  }
  CalibInfo->SetBranchAddress( "Run1", &m_run1 );
  CalibInfo->SetBranchAddress( "Run2", &m_run2 );
  CalibInfo->SetBranchAddress( "Version", &m_version );
  CalibInfo->SetBranchAddress( "ebrId0", &m_tofidEast[0] );
  CalibInfo->SetBranchAddress( "ebrId1", &m_tofidEast[1] );
  CalibInfo->SetBranchAddress( "ebrId2", &m_tofidEast[2] );
  CalibInfo->SetBranchAddress( "ebrId3", &m_tofidEast[3] );
  CalibInfo->SetBranchAddress( "ebrId4", &m_tofidEast[4] );
  CalibInfo->SetBranchAddress( "ecId0", &m_tofidEndcap[0] );
  CalibInfo->SetBranchAddress( "ecId1", &m_tofidEndcap[1] );
  CalibInfo->SetBranchAddress( "ecId2", &m_tofidEndcap[2] );
  CalibInfo->SetBranchAddress( "ecId3", &m_tofidEndcap[3] );
  CalibInfo->SetBranchAddress( "ecId4", &m_tofidEndcap[4] );
  CalibInfo->SetBranchAddress( "wbrId0", &m_tofidWest[0] );
  CalibInfo->SetBranchAddress( "wbrId1", &m_tofidWest[1] );
  CalibInfo->SetBranchAddress( "wbrId2", &m_tofidWest[2] );
  CalibInfo->SetBranchAddress( "wbrId3", &m_tofidWest[3] );
  CalibInfo->SetBranchAddress( "wbrId4", &m_tofidWest[4] );
  CalibInfo->SetBranchAddress( "misLable", &m_misLable );
  CalibInfo->SetBranchAddress( "qCorr", &m_qCorr );
  CalibInfo->SetBranchAddress( "qElec", &m_qElec );
 
  entries = CalibInfo->GetEntries();
  for ( cnt = 0; cnt < entries; cnt++ )
  {
    CalibInfo->GetEntry( cnt );
    tofinfo.setRunFrom( m_runFrom );
    tofinfo.setRunTo( m_runTo );
    tofinfo.setEventFrom( m_eventFrom );
    tofinfo.setEventTo( m_eventTo );
    tofinfo.setRunBegin( m_run1 );
    tofinfo.setRunEnd( m_run2 );
    tofinfo.setVersion( m_version );
    tofinfo.setQCorr( m_qCorr );
    tofinfo.setQElec( m_qElec );
    tofinfo.setMisLable( m_misLable );
    tofinfo.setBrEast( m_tofidEast );
    tofinfo.setBrWest( m_tofidWest );
    tofinfo.setEndcap( m_tofidEndcap );
    tofinfoCol.push_back( tofinfo );
  }
 
  CalibData::TofCalibData* tmpObject = new CalibData::TofCalibData(
      tmpbTof, tmpbTofCommon, tmpeTof, tmpetf, tmpetfBunch, tofinfoCol );
 
  refpObject = tmpObject;
  delete btoftree;
  delete etoftree;
  delete etftree;
  delete etfbunchtree;
  delete btofcommontree;
  delete CalibInfo;
 
  return StatusCode::SUCCESS;
}