RootCnvSvc-04-01-15/src/Dst/CgemTrackCnv.cxx

Go to the documentation of this file.

#ifndef CgemTrackCnv_CXX
#define CgemTrackCnv_CXX  1
 
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/ObjectVector.h"
#include "TClonesArray.h"
#include "EventModel/EventModel.h"
#include "DstEvent/DstEvent.h"  //TDS object  
#include "DstEvent/DstCgemTrack.h"   //TDS object
#include "RootEventData/TCgemTrack.h" // standard root object
#include "RootEventData/TDstEvent.h"
#include "RootCnvSvc/Dst/DstCnv.h"
#include "RootCnvSvc/Dst/CgemTrackCnv.h"
#include "RootCnvSvc/RootAddress.h"
#include <vector>
#include "CLHEP/Matrix/Vector.h"
#include "CLHEP/Matrix/SymMatrix.h"
#include "CLHEP/Geometry/Point3D.h"
 
using namespace std;
 
 
CgemTrackCnv::CgemTrackCnv(ISvcLocator* svc)
: RootEventBaseCnv(classID(), svc)
{
    // Here we associate this converter with the /Event path on the TDS.
    MsgStream log(msgSvc(), "CgemTrackCnv");
    m_rootBranchname ="m_cgemTrackCol";
    m_adresses.push_back(&m_cgemTrackCol);
    m_cgemTrackCol=0;
}
 
StatusCode CgemTrackCnv::TObjectToDataObject(DataObject*& refpObject) {
    // creation of TDS object from root object
    MsgStream log(msgSvc(), "CgemTrackCnv");
//myModify
    log << MSG::INFO << "My Message::CgemTrackCnv::TObjectToDataObject()"<<endreq;
 
    log << MSG::DEBUG << "CgemTrackCnv::TObjectToDataObject" << endreq;
    StatusCode sc=StatusCode::SUCCESS;
 
    // create the TDS location for the CgemTrack Collection
    DstCgemTrackCol* cgemTrackTdsCol = new DstCgemTrackCol;
    refpObject=cgemTrackTdsCol;
 
    // now convert
    if (!m_cgemTrackCol) return sc;
    TIter cgemTrackIter(m_cgemTrackCol);
    TCgemTrack *cgemTrackRoot = 0;
    while ((cgemTrackRoot = (TCgemTrack*)cgemTrackIter.Next())) {
 
     int      trackId = cgemTrackRoot->trackId();
     int      charge  = cgemTrackRoot->charge();     
     double   helix[5];
     for(int i=0; i<5; i++){
       helix[i] = cgemTrackRoot->helix(i);
       log<<MSG::INFO<<"in CgemTrackCnv, helix["<<i<<"]: "<<helix[i]<<endreq;
     }
     double   pxy   = cgemTrackRoot->pxy();
     double   px    = cgemTrackRoot->px();
     double   py    = cgemTrackRoot->py();
     double   pz    = cgemTrackRoot->pz();
     double   p     = cgemTrackRoot->p();
     double   theta = cgemTrackRoot->theta();
     double   phi   = cgemTrackRoot->phi();
     double   x     = cgemTrackRoot->x();
     double   y     = cgemTrackRoot->y();
     double   z     = cgemTrackRoot->z();
     double   r     = cgemTrackRoot->r();
     int      stat  = cgemTrackRoot->stat();
     double   chi   = cgemTrackRoot->chi2();
     int      ndof  = cgemTrackRoot->ndof();
     
     double err[15];
     for (int i=0; i<15; i++){
       err[i] = cgemTrackRoot->err(i);
     }
   
     double poca[3];
     poca[0] = x;
     poca[1] = y;
     poca[2] = z;
 
//myModify
    log << MSG::INFO << "My Message::CgemTrackCnv::TObjectToDataObject::charge:"<<charge<<",p:"<<p<<",r:"<<r<<",z:"<<z<<endreq;
 
     DstCgemTrack *cgemTrackTds = new DstCgemTrack();
     m_common.m_rootCgemTrackMap[cgemTrackRoot] = cgemTrackTds;
 
     cgemTrackTds->setTrackId( trackId );
    log<<MSG::INFO<<" dstCgemTrack trkid: "<<cgemTrackTds->trackId()<<endreq;
  
     cgemTrackTds->setCharge( charge );
     cgemTrackTds->setHelix(helix);
     cgemTrackTds->setPoca(poca);
     
     log<<MSG::INFO<<"dstCgemTrack helix: "<<cgemTrackTds->helix()<<endreq;
    
     cgemTrackTds->setPxy( pxy );   
     cgemTrackTds->setPx( px );   
     cgemTrackTds->setPy( py );   
     cgemTrackTds->setPz( pz );   
     cgemTrackTds->setP( p );   
     cgemTrackTds->setTheta( theta );   
     cgemTrackTds->setPhi( phi );   
     cgemTrackTds->setX( x );   
     cgemTrackTds->setY( y );    
     cgemTrackTds->setZ( z );   
     cgemTrackTds->setR( r );   
     cgemTrackTds->setStat( stat );  
     cgemTrackTds->setChi2( chi );  
     cgemTrackTds->setNdof( ndof );  
     cgemTrackTds->setError( err );   
     
     cgemTrackTdsCol->push_back(cgemTrackTds);
    
     }
    delete m_cgemTrackCol;
    m_cgemTrackCol = 0;  
//myModify
    log << MSG::INFO << "My Message::CgemTrackCnv::TObjectToDataObject()-finished"<<endreq;
 
    return StatusCode::SUCCESS;
}
 
StatusCode CgemTrackCnv::DataObjectToTObject(DataObject* obj,RootAddress* rootaddr) {
 
  MsgStream log(msgSvc(), "CgemTrackCnv");
//myModify
    log << MSG::INFO << "My Message::CgemTrackCnv::DataObjectToTObject("<<rootaddr->getTreename()<<")"<<endreq;
 
  log << MSG::DEBUG << "CgemTrackCnv::DataObjectToTObject" << endreq;
  StatusCode sc=StatusCode::SUCCESS;
 
  DstCgemTrackCol * cgemTrackColTds=dynamic_cast<DstCgemTrackCol *> (obj);
  if (!cgemTrackColTds) {
    log << MSG::ERROR << "Could not downcast to CgemTrackCol" << endreq;
//myModify
    log << MSG::INFO << "My Message::CgemTrackCnv::DataObjectToTObject("<<rootaddr->getTreename()<<")-finished"<<endreq;
 
    return StatusCode::FAILURE;
  }
 
  DataObject *evt;
  m_eds->findObject(EventModel::Dst::Event,evt);
  if (evt==NULL) {
    log << MSG::ERROR << "Could not get DstEvent in TDS "  << endreq;
//myModify
    log << MSG::INFO << "My Message::CgemTrackCnv::DataObjectToTObject("<<rootaddr->getTreename()<<")-finished"<<endreq;
 
    return StatusCode::FAILURE;
  }
  DstEvent * devtTds=dynamic_cast<DstEvent *> (evt);
  if (!devtTds) {
    log << MSG::ERROR << "CgemTrackCnv:Could not downcast to TDS DstEvent" << endreq;
  }
  IOpaqueAddress *addr;
 
//myModify
    log << MSG::INFO << "My Message::CgemTrackCnv::DataObjectToTObject("<<rootaddr->getTreename()<<")::DstCnv"<<endreq;
 
  m_cnvSvc->getDstCnv()->createRep(evt,addr); 
  TDstEvent *recEvt=m_cnvSvc->getDstCnv()->getWriteObject();
 
  const TObjArray *m_cgemTrackCol = recEvt->getCgemTrackCol();
  if (!m_cgemTrackCol) return sc;
  recEvt->clearCgemTrackCol(); //necessary in case there is I/O at the same time since array is static
  DstCgemTrackCol::const_iterator cgemTrackTds;
 
  for (cgemTrackTds = cgemTrackColTds->begin(); cgemTrackTds != cgemTrackColTds->end(); cgemTrackTds++) {
     Int_t      trackId =(*cgemTrackTds)->trackId();
     Double_t   helix[5];
     for(int i=0;i<5;i++){
       helix[i] = (*cgemTrackTds)->helix(i) ; 
     }    
     Int_t    stat = (*cgemTrackTds)->stat();      
     Double_t chi =  (*cgemTrackTds)->chi2();      
     Int_t    ndof = (*cgemTrackTds)->ndof();       
     Double_t err[15];
     for (Int_t i=0; i<15; i++){  
       err[i] = (*cgemTrackTds)->err(i);
     }
     Int_t fL = (*cgemTrackTds)->firstLayer();
     Int_t lL = (*cgemTrackTds)->lastLayer();
 
     TCgemTrack *cgemTrackRoot = new TCgemTrack();
     //m_common.m_cgemTrackMap[(*cgemTrackTds)] = cgemTrackRoot;
     cgemTrackRoot->setTrackId( trackId );
    
     cgemTrackRoot->setHelix(helix);
     log<<MSG::INFO<<"test,trackId: "<<cgemTrackRoot->trackId()<<endreq;
     log<<MSG::INFO<<"test,px: "<<cgemTrackRoot->px()
                   <<"test,py: "<<cgemTrackRoot->py()
               <<"test,pz: "<<cgemTrackRoot->pz()<<endreq;
 
     cgemTrackRoot->setStat( stat );  
     cgemTrackRoot->setChi2( chi );  
     cgemTrackRoot->setNdof( ndof );  
     cgemTrackRoot->setErr( err );   
     cgemTrackRoot->setFirstLayer(fL);
     cgemTrackRoot->setLastLayer(lL);
     
     recEvt->addCgemTrack(cgemTrackRoot);
  }
 
//myModify
    log << MSG::INFO << "My Message::CgemTrackCnv::DataObjectToTObject("<<rootaddr->getTreename()<<")-finished"<<endreq;
 
  return StatusCode::SUCCESS;
}
#endif