RecMdcTrackCnv.cxx

Go to the documentation of this file.

#ifndef RecMdcTrackCnv_CXX
#define RecMdcTrackCnv_CXX 1
 
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/ObjectVector.h"
 
#include "Identifier/Identifier.h"
#include "Identifier/MdcID.h"
#include "TClonesArray.h"
 
#include "EventModel/EventModel.h"
 
#include "MdcRecEvent/RecMdcHit.h"
#include "MdcRecEvent/RecMdcTrack.h"
#include "ReconEvent/ReconEvent.h"
#include "RootEventData/TRecMdcTrack.h"
#include "RootEventData/TRecTrackEvent.h"
 
#include "RootCnvSvc/Rec/RecMdcTrackCnv.h"
#include "RootCnvSvc/Rec/RecTrackCnv.h"
#include "RootCnvSvc/RootAddress.h"
 
#include <vector>
 
using namespace std;
 
// Instantiation of a static factory class used by clients to create
// instances of this service
// static CnvFactory<RecMdcTrackCnv> s_factory;
// const ICnvFactory& RecMdcTrackCnvFactory = s_factory;
 
RecMdcTrackCnv::RecMdcTrackCnv( ISvcLocator* svc ) : RootEventBaseCnv( classID(), svc ) {
  // Here we associate this converter with the /Event path on the TDS.
  MsgStream log( msgSvc(), "RecMdcTrackCnv" );
  // m_rootTreename ="Rec";
  m_rootBranchname = "m_recMdcTrackCol";
  // declareObject(EventModel::Recon::RecMdcTrackCol, objType(), m_rootTreename,
  // m_rootBranchname);
  m_adresses.push_back( &m_recMdcTrackCol );
  m_recMdcTrackCol = 0;
}
 
StatusCode RecMdcTrackCnv::TObjectToDataObject( DataObject*& refpObject ) {
  // creation of TDS object from root object
  MsgStream log( msgSvc(), "RecMdcTrackCnv" );
  log << MSG::DEBUG << "RecMdcTrackCnv::TObjectToDataObject" << endmsg;
 
  // create the TDS location for the RecMdcTrack Collection
  RecMdcTrackCol* recMdcTrackCol = new RecMdcTrackCol;
  refpObject                     = recMdcTrackCol;
  // now convert
  if ( !m_recMdcTrackCol ) return StatusCode::SUCCESS;
  TIter mdcTrackIter( m_recMdcTrackCol );
 
  //----------get hit smartRefVector--------
  // HitRefVec *hitRefVec;
  // int nTk=0;
  // while (mdcTrackIter.Next()){ nTk++; }
  // hitRefVec = new HitRefVec[nTk];
 
  IDataProviderSvc* dataSvc = 0;
  StatusCode        sc      = serviceLocator()->getService(
      "EventDataSvc", IDataProviderSvc::interfaceID(), (IInterface*&)dataSvc );
  if ( !sc.isSuccess() )
  {
    log << MSG::FATAL << "Could not get EventDataSvc in RecMdcTrackCnv" << endmsg;
    return ( StatusCode::FAILURE );
  }
  SmartDataPtr<RecMdcHitCol> recMdcHitCol( dataSvc, "/Event/Recon/RecMdcHitCol" );
  if ( !recMdcHitCol )
  {
    log << MSG::FATAL << "Could not find RecMdcHitCol" << endmsg;
    return ( StatusCode::FAILURE );
  }
  // RecMdcHitCol::iterator iter = recMdcHitCol->begin();
  // for (;iter != recMdcHitCol->end(); iter++ ) {
  //   int tkId = (*iter)->getTrkId();
  //   hitRefVec[tkId].push_back(*iter);
  // }
 
  //---------------------------------------
  mdcTrackIter                  = m_recMdcTrackCol;
  TRecMdcTrack* recMdcTrackRoot = 0;
  while ( ( recMdcTrackRoot = (TRecMdcTrack*)mdcTrackIter.Next() ) )
  {
    int    trackId = recMdcTrackRoot->trackId();
    int    charge  = recMdcTrackRoot->charge();
    double helix[5];
    for ( int i = 0; i < 5; i++ ) { helix[i] = recMdcTrackRoot->helix( i ); }
    double pxy    = recMdcTrackRoot->pxy();
    double px     = recMdcTrackRoot->px();
    double py     = recMdcTrackRoot->py();
    double pz     = recMdcTrackRoot->pz();
    double p      = recMdcTrackRoot->p();
    double theta  = recMdcTrackRoot->theta();
    double phi    = recMdcTrackRoot->phi();
    double x      = recMdcTrackRoot->x();
    double y      = recMdcTrackRoot->y();
    double z      = recMdcTrackRoot->z();
    double r      = recMdcTrackRoot->r();
    int    stat   = recMdcTrackRoot->stat();
    double chi    = recMdcTrackRoot->chi2();
    int    ndof   = recMdcTrackRoot->ndof();
    int    nlayer = recMdcTrackRoot->nlayer();
    double err[15];
    for ( int i = 0; i < 15; i++ ) { err[i] = recMdcTrackRoot->err( i ); }
    int nh = recMdcTrackRoot->nhits();
    int ns = recMdcTrackRoot->nster();
    // vector<int> vecHits = recMdcTrackRoot->vecHits();
    double       vx0                               = recMdcTrackRoot->vx0();
    double       vy0                               = recMdcTrackRoot->vy0();
    double       vz0                               = recMdcTrackRoot->vz0();
    double       fiterm                            = recMdcTrackRoot->fiTerm();
    RecMdcTrack* recMdcTrack                       = new RecMdcTrack();
    m_common.m_rootRecMdcTrackMap[recMdcTrackRoot] = recMdcTrack;
 
    recMdcTrack->setTrackId( trackId );
    recMdcTrack->setCharge( charge );
    recMdcTrack->setHelix( helix );
    recMdcTrack->setPxy( pxy );
    recMdcTrack->setPx( px );
    recMdcTrack->setPy( py );
    recMdcTrack->setPz( pz );
    recMdcTrack->setP( p );
    recMdcTrack->setTheta( theta );
    recMdcTrack->setPhi( phi );
    recMdcTrack->setX( x );
    recMdcTrack->setY( y );
    recMdcTrack->setZ( z );
    recMdcTrack->setR( r );
    recMdcTrack->setStat( stat );
    recMdcTrack->setChi2( chi );
    recMdcTrack->setNdof( ndof );
    recMdcTrack->setError( err );
    recMdcTrack->setNhits( nh );
    recMdcTrack->setNster( ns );
    recMdcTrack->setNlayer( nlayer );
    // recMdcTrack->setVecHits( vecHits );
    // rec
    recMdcTrack->setVX0( vx0 );
    recMdcTrack->setVY0( vy0 );
    recMdcTrack->setVZ0( vz0 );
    recMdcTrack->setFiTerm( fiterm );
 
    HitRefVec theHitRefVec;
 
    RecMdcHitCol::iterator iter = recMdcHitCol->begin();
    for ( ; iter != recMdcHitCol->end(); iter++ )
    {
 
      // cout<<" (*iter)->getTrkId(): "<<(*iter)->getTrkId()<<endl;
      if ( ( *iter )->getTrkId() == trackId )
      {
        SmartRef<RecMdcHit> refhit( *iter );
        theHitRefVec.push_back( refhit );
      }
    }
 
    recMdcTrack->setVecHits( theHitRefVec );
 
    int nhits = recMdcTrack->getVecHits().size();
 
    // std::cout<<" mdc hits: "<<nhits<< std::endl;
 
    for ( int ii = 0; ii < nhits; ii++ )
    {
 
      // cout<<"ddl: "<<(recMdcTrack->getVecHits()[ii])->getDriftDistLeft()<<endl;
      // cout<<"erddl: "<<(recMdcTrack->getVecHits()[ii])->getErrDriftDistLeft()<<endl;
      Identifier id( recMdcTrack->getVecHits()[ii]->getMdcId() );
      int        layer = MdcID::layer( id );
      int        wire  = MdcID::wire( id );
      // cout<<"layer: "<<layer<<" wire: "<<wire<<endl;
    }
    recMdcTrackCol->push_back( recMdcTrack );
  }
 
  delete m_recMdcTrackCol;
  m_recMdcTrackCol = 0;
  return StatusCode::SUCCESS;
}
 
StatusCode RecMdcTrackCnv::DataObjectToTObject( DataObject* obj, RootAddress* rootaddr ) {
 
  MsgStream log( msgSvc(), "RecMdcTrackCnv" );
  log << MSG::DEBUG << "RecMdcTrackCnv::DataObjectToTObject" << endmsg;
 
  RecMdcTrackCol* mdcTrackColTds = dynamic_cast<RecMdcTrackCol*>( obj );
  if ( !mdcTrackColTds )
  {
    log << MSG::ERROR << "Could not downcast to MdcTrackCol" << endmsg;
    return StatusCode::FAILURE;
  }
  DataObject* evt;
  StatusCode  sc = m_eds->findObject( EventModel::Recon::Event, evt );
  if ( !sc.isSuccess() )
  {
    log << MSG::ERROR << "Could not get ReconEvent in TDS " << endmsg;
    return StatusCode::FAILURE;
  }
  ReconEvent* devtTds = dynamic_cast<ReconEvent*>( evt );
  if ( !devtTds )
  { log << MSG::ERROR << "RecMdcTrackCnv:Could not downcast to TDS ReconEvent" << endmsg; }
 
  IOpaqueAddress* addr;
 
  m_cnvSvc->getRecTrackCnv()->createRep( evt, addr );
  TRecTrackEvent* recEvt = m_cnvSvc->getRecTrackCnv()->getWriteObject();
 
  const TObjArray* m_recMdcTrackCol = recEvt->getRecMdcTrackCol();
  if ( !m_recMdcTrackCol ) return StatusCode::SUCCESS;
  // necessary in case there is I/O at the same time since array is static
  recEvt->clearRecMdcTrackCol();
 
  RecMdcTrackCol::const_iterator recMdcTrack;
  for ( recMdcTrack = mdcTrackColTds->begin(); recMdcTrack != mdcTrackColTds->end();
        recMdcTrack++ )
  {
    Int_t    trackId = ( *recMdcTrack )->trackId();
    Double_t helix[5];
    for ( int i = 0; i < 5; i++ )
    {
      log << MSG::INFO << " recMdcTrack.helix(" << i << "): " << ( *recMdcTrack )->helix( i )
          << endmsg;
      helix[i] = ( *recMdcTrack )->helix( i );
    }
 
    Int_t    stat   = ( *recMdcTrack )->stat();
    Double_t chi    = ( *recMdcTrack )->chi2();
    Int_t    ndof   = ( *recMdcTrack )->ndof();
    Int_t    ns     = ( *recMdcTrack )->nster();
    Int_t    nlayer = ( *recMdcTrack )->nlayer();
 
    Double_t er[15];
    for ( Int_t i = 0; i < 15; i++ ) { er[i] = ( *recMdcTrack )->err( i ); }
    Int_t nh = ( *recMdcTrack )->getNhits();
    // vector<Int_t> vecHits = (*recMdcTrack)->vecHits();
    Double_t vx0    = ( *recMdcTrack )->getVX0();
    Double_t vy0    = ( *recMdcTrack )->getVY0();
    Double_t vz0    = ( *recMdcTrack )->getVZ0();
    Double_t fiterm = ( *recMdcTrack )->getFiTerm();
 
    TRecMdcTrack* recMdcTrackRoot = new TRecMdcTrack();
    // m_common.m_recMdcTrackMap[(*recMdcTrack)] = recMdcTrackRoot;
 
    recMdcTrackRoot->setTrackId( trackId );
    recMdcTrackRoot->setHelix( helix );
    log << MSG::INFO << " test,recMdcTrackRoot.px: " << recMdcTrackRoot->px()
        << " recMdcTrackRoot.py: " << recMdcTrackRoot->py()
        << " recMdcTrackRoot.pz: " << recMdcTrackRoot->pz() << endmsg;
 
    recMdcTrackRoot->setStat( stat );
    recMdcTrackRoot->setChi2( chi );
    recMdcTrackRoot->setNdof( ndof );
    recMdcTrackRoot->setNlayer( nlayer );
    recMdcTrackRoot->setErr( er );
    recMdcTrackRoot->setNhits( nh );
    recMdcTrackRoot->setNster( ns );
    // recMdcTrackRoot->setVecHits(vecHits);
    recMdcTrackRoot->setVX0( vx0 );
    recMdcTrackRoot->setVY0( vy0 );
    recMdcTrackRoot->setVZ0( vz0 );
    recMdcTrackRoot->setFiTerm( fiterm );
 
    recEvt->addRecMdcTrack( recMdcTrackRoot );
  }
 
  return StatusCode::SUCCESS;
}
#endif