RootCnvSvc-04-01-15/src/Rec/RecCgemTrackCnv.cxx

Go to the documentation of this file.

#ifndef RecCgemTrackCnv_CXX
#define RecCgemTrackCnv_CXX  1
 
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/ObjectVector.h"
 
#include "TClonesArray.h"
#include "Identifier/CgemID.h"
#include "Identifier/Identifier.h"
 
#include "EventModel/EventModel.h"
 
#include "ReconEvent/ReconEvent.h"  
#include "CgemRecEvent/RecCgemTrack.h"
#include "CgemRecEvent/RecCgemHit.h"
#include "RootEventData/TRecCgemTrack.h"
#include "RootEventData/TRecTrackEvent.h"
 
#include "RootCnvSvc/Rec/RecTrackCnv.h"
#include "RootCnvSvc/Rec/RecCgemTrackCnv.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<RecCgemTrackCnv> s_factory;
//const ICnvFactory& RecCgemTrackCnvFactory = s_factory;
 
RecCgemTrackCnv::RecCgemTrackCnv(ISvcLocator* svc)
: RootEventBaseCnv(classID(), svc)
{
    // Here we associate this converter with the /Event path on the TDS.
    MsgStream log(msgSvc(), "RecCgemTrackCnv");
    //m_rootTreename ="Rec";
    m_rootBranchname ="m_recCgemTrackCol";
    //declareObject(EventModel::Recon::RecCgemTrackCol, objType(), m_rootTreename, m_rootBranchname);
    m_adresses.push_back(&m_recCgemTrackCol);
    m_recCgemTrackCol=0;
}
 
StatusCode RecCgemTrackCnv::TObjectToDataObject(DataObject*& refpObject) {
  // creation of TDS object from root object
  MsgStream log(msgSvc(), "RecCgemTrackCnv");
  log << MSG::DEBUG << "RecCgemTrackCnv::TObjectToDataObject" << endreq;
 
  // create the TDS location for the RecCgemTrack Collection
  RecCgemTrackCol* recCgemTrackCol = new RecCgemTrackCol;
  refpObject = recCgemTrackCol;
  // now convert
  if (!m_recCgemTrackCol) return StatusCode::SUCCESS;
  TIter cgemTrackIter(m_recCgemTrackCol);
 
  //----------get hit smartRefVector--------
  //HitRefVec *hitRefVec;
  //int nTk=0;
  //while (cgemTrackIter.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 RecCgemTrackCnv" << endreq;
    return( StatusCode::FAILURE);
  }
  SmartDataPtr<RecCgemHitCol> recCgemHitCol(dataSvc,"/Event/Recon/RecCgemHitCol");
  if (!recCgemHitCol) {
    log << MSG::FATAL << "Could not find RecCgemHitCol" << endreq;
    return( StatusCode::FAILURE);
  }
  //RecCgemHitCol::iterator iter = recCgemHitCol->begin();
  //for (;iter != recCgemHitCol->end(); iter++ ) {
  //  int tkId = (*iter)->getTrkId(); 
  //  hitRefVec[tkId].push_back(*iter);
  //}
 
  //---------------------------------------
  cgemTrackIter = m_recCgemTrackCol;
  TRecCgemTrack *recCgemTrackRoot = 0;
  while ((recCgemTrackRoot = (TRecCgemTrack*)cgemTrackIter.Next())) {
    int      trackId =  recCgemTrackRoot->trackId();
    int      charge =   recCgemTrackRoot->charge();     
    double   cgemHelix[5];
    for(int i=0;i<5; i++){
      cgemHelix[i] = recCgemTrackRoot->helix(i);
    }
    double   pxy   = recCgemTrackRoot->pxy();
    double   px    = recCgemTrackRoot->px();
    double   py    = recCgemTrackRoot->py();
    double   pz    = recCgemTrackRoot->pz();
    double   p     = recCgemTrackRoot->p();
    double   theta = recCgemTrackRoot->theta();
    double   phi   = recCgemTrackRoot->phi();
    double   x     = recCgemTrackRoot->x();
    double   y     = recCgemTrackRoot->y();
    double   z     = recCgemTrackRoot->z();
    double   r     = recCgemTrackRoot->r();
    int      stat  = recCgemTrackRoot->stat();      
    double   chi   = recCgemTrackRoot->chi2();      
    int      ndof  = recCgemTrackRoot->ndof();       
    double err[15];
    for (int i=0; i<15; i++){
      err[i] = recCgemTrackRoot->err(i);
    }
    int      nh     = recCgemTrackRoot->nhits();     
    // vector<int> vecHits = recCgemTrackRoot->vecHits();
    double   vx0    = recCgemTrackRoot->vx0();
    double   vy0    = recCgemTrackRoot->vy0();
    double   vz0    = recCgemTrackRoot->vz0();
    double   fiterm = recCgemTrackRoot->fiTerm();
    RecCgemTrack *recCgemTrack = new RecCgemTrack();
    m_common.m_rootRecCgemTrackMap[recCgemTrackRoot] = recCgemTrack;
 
    recCgemTrack->setTrackId( trackId );
    recCgemTrack->setCharge( charge );
    recCgemTrack->setHelix( cgemHelix );
    recCgemTrack->setPxy( pxy );   
    recCgemTrack->setPx( px );   
    recCgemTrack->setPy( py );   
    recCgemTrack->setPz( pz );   
    recCgemTrack->setP( p );   
    recCgemTrack->setTheta( theta );   
    recCgemTrack->setPhi( phi );   
    recCgemTrack->setX( x );   
    recCgemTrack->setY( y );    
    recCgemTrack->setZ( z );   
    recCgemTrack->setR( r );   
    recCgemTrack->setStat( stat );  
    recCgemTrack->setChi2( chi );  
    recCgemTrack->setNdof( ndof );  
    recCgemTrack->setError( err );   
    recCgemTrack->setNhits( nh );  
    // recCgemTrack->setVecHits( vecHits );
    //rec
    recCgemTrack->setVX0( vx0 );   
    recCgemTrack->setVY0( vy0 );    
    recCgemTrack->setVZ0( vz0 );   
    recCgemTrack->setFiTerm( fiterm );  
 
    RecCgemHitRefVec theHitRefVec;
 
    RecCgemHitCol::iterator iter = recCgemHitCol->begin();
    for (;iter != recCgemHitCol->end(); iter++){
  
      //cout<<" (*iter)->getTrkId(): "<<(*iter)->getTrkId()<<endl;
      if((*iter)->getTrkId() == trackId){
    SmartRef<RecCgemHit> refhit(*iter);
    theHitRefVec.push_back(refhit);
      }
    }
 
    recCgemTrack->setVecHits(theHitRefVec);
    
    int nhits = recCgemTrack->getVecHits().size();
          
    //std::cout<<" cgem hits: "<<nhits<< std::endl;
    
    for(int ii=0; ii <nhits ; ii++){
 
      //cout<<"ddl: "<<(recCgemTrack->getVecHits()[ii])->getDriftDistLeft()<<endl;
      //cout<<"erddl: "<<(recCgemTrack->getVecHits()[ii])->getErrDriftDistLeft()<<endl;
      Identifier id(recCgemTrack->getVecHits()[ii]->getCgemId());
      int layer = CgemID::layer(id);
      int sheet = CgemID::sheet(id);
      int strip = CgemID::strip(id);
      //cout<<"layer: "<<layer<<" wire: "<<wire<<endl;
    } 
      recCgemTrackCol->push_back(recCgemTrack);
  }
 
  delete m_recCgemTrackCol;
  m_recCgemTrackCol = 0;  
  return StatusCode::SUCCESS;
}
 
 
 
 
StatusCode RecCgemTrackCnv::DataObjectToTObject(DataObject* obj,RootAddress* rootaddr) {
 
  MsgStream log(msgSvc(), "RecCgemTrackCnv");
  log << MSG::DEBUG << "RecCgemTrackCnv::DataObjectToTObject" << endreq;
 
  RecCgemTrackCol * cgemTrackColTds=dynamic_cast<RecCgemTrackCol *> (obj);
  if (!cgemTrackColTds) {
    log << MSG::ERROR << "Could not downcast to CgemTrackCol" << endreq;
    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 "  << endreq;
    return StatusCode::FAILURE;
  }
  ReconEvent * devtTds=dynamic_cast<ReconEvent *> (evt);
  if (!devtTds) {
    log << MSG::ERROR << "RecCgemTrackCnv:Could not downcast to TDS ReconEvent" << endreq;
  }
 
  IOpaqueAddress *addr;
 
  m_cnvSvc->getRecTrackCnv()->createRep(evt,addr);
  TRecTrackEvent *recEvt=m_cnvSvc->getRecTrackCnv()->getWriteObject();
 
  const TObjArray *m_recCgemTrackCol = recEvt->getRecCgemTrackCol();
  if (!m_recCgemTrackCol) return StatusCode::SUCCESS;
  //necessary in case there is I/O at the same time since array is static
  recEvt->clearRecCgemTrackCol();
 
  RecCgemTrackCol::const_iterator recCgemTrack;
  for (recCgemTrack = cgemTrackColTds->begin(); recCgemTrack != cgemTrackColTds->end(); recCgemTrack++) {
     Int_t      trackId =(*recCgemTrack)->trackId();
     Double_t   cgemHelix[5];
     for(int i=0;i<5;i++){
       log<<MSG::INFO<<" recCgemTrack.cgemHelix("<<i<<"): "<<(*recCgemTrack)->helix(i)<<endreq;
       cgemHelix[i] = (*recCgemTrack)->helix(i);     
     }
 
     Int_t      stat =  (*recCgemTrack)->stat();      
     Double_t   chi =   (*recCgemTrack)->chi2();      
     Int_t      ndof =  (*recCgemTrack)->ndof();       
 
     Double_t er[15];
     for (Int_t i=0; i<15; i++){
       er[i] = (*recCgemTrack)->err(i);
     }
     Int_t      nh =    (*recCgemTrack)->getNhits();     
     // vector<Int_t> vecHits = (*recCgemTrack)->vecHits();
     Double_t   vx0 =   (*recCgemTrack)->getVX0();         
     Double_t   vy0 =   (*recCgemTrack)->getVY0();         
     Double_t   vz0 =   (*recCgemTrack)->getVZ0();
     Double_t   fiterm =(*recCgemTrack)->getFiTerm();
 
     TRecCgemTrack *recCgemTrackRoot = new TRecCgemTrack();
     //m_common.m_recCgemTrackMap[(*recCgemTrack)] = recCgemTrackRoot;
 
     recCgemTrackRoot->setTrackId( trackId );
     recCgemTrackRoot->setHelix(cgemHelix);
     log<<MSG::INFO<<" test,recCgemTrackRoot.px: "<<recCgemTrackRoot->px()
                   <<" recCgemTrackRoot.py: "<<recCgemTrackRoot->py()
               <<" recCgemTrackRoot.pz: "<<recCgemTrackRoot->pz()
           <<endreq;
       
     recCgemTrackRoot->setStat( stat );  
     recCgemTrackRoot->setChi2( chi );  
     recCgemTrackRoot->setNdof( ndof );  
     recCgemTrackRoot->setErr( er );   
     recCgemTrackRoot->setNhits( nh );  
     // recCgemTrackRoot->setVecHits(vecHits);
     recCgemTrackRoot->setVX0( vx0 );
     recCgemTrackRoot->setVY0( vy0 );
     recCgemTrackRoot->setVZ0( vz0 );
     recCgemTrackRoot->setFiTerm( fiterm );
 
     recEvt->addRecCgemTrack(recCgemTrackRoot);
  }
 
 
  return StatusCode::SUCCESS;
}
#endif