RecMdcTrackCnv.cxx

Go to the documentation of this file.

#ifndef RecMdcTrackCnv_CXX
#define RecMdcTrackCnv_CXX  1
 
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/ObjectVector.h"
 
#include "TClonesArray.h"
#include "Identifier/MdcID.h"
#include "Identifier/Identifier.h"
 
#include "EventModel/EventModel.h"
 
#include "ReconEvent/ReconEvent.h"  
#include "MdcRecEvent/RecMdcTrack.h"
#include "MdcRecEvent/RecMdcHit.h"
#include "RootEventData/TRecMdcTrack.h"
#include "RootEventData/TRecTrackEvent.h"
 
#include "RootCnvSvc/Rec/RecTrackCnv.h"
#include "RootCnvSvc/Rec/RecMdcTrackCnv.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" << endreq;
 
    // 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" << endreq;
        return( StatusCode::FAILURE);
    }
    SmartDataPtr<RecMdcHitCol> recMdcHitCol(dataSvc,"/Event/Recon/RecMdcHitCol");
    if (!recMdcHitCol) {
        log << MSG::FATAL << "Could not find RecMdcHitCol" << endreq;
        return( StatusCode::FAILURE);
    }
    //RecMdcHitCol::iterator iter = recMdcHitCol->begin();
    //for (;iter != recMdcHitCol->end(); iter++ ) {
    //  int tkId = (*iter)->getTrkId(); 
    //  hitRefVec[tkId].push_back(*iter);
    //}
 
    SmartDataPtr<RecCgemClusterCol> aCgemClusterCol(dataSvc,"/Event/Recon/RecCgemClusterCol");
    if(!aCgemClusterCol) {
        log << MSG::FATAL << "Could not find RecCgemClusterCol" << endreq;
        return( StatusCode::FAILURE);
    }
    //---------------------------------------
    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();       
        //int nCgemXClusters=recMdcTrackRoot->nCgemXClusters();
        //int nCgemVClusters=recMdcTrackRoot->nCgemVClusters();
        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 );  
        //recMdcTrack->setNCgemXCluster( nCgemXClusters ); 
        //recMdcTrack->setNCgemVCluster( nCgemVClusters ); 
 
        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;
        } 
 
        // --- CGEM clusters (added on 6 Apr, 2021, L.L. Wang)
        ClusterRefVec clusterRefVec;
        map<int,int> clusterFitStat;
        vector<Int_t> vecClusterId = recMdcTrackRoot->vecClusterId();
        cout<<"recMdcTrackRoot gets a vecClusterId with size "<<vecClusterId.size()<<endl;
        vector<Int_t>::iterator it_clusterId = vecClusterId.begin();
        RecCgemClusterCol::iterator iter_cluster=aCgemClusterCol->begin();
        //for(; iter_cluster!=aCgemClusterCol->end(); iter_cluster++)
        for(; it_clusterId!=vecClusterId.end(); it_clusterId++)
        {
            //if(trackId==(*iter_cluster)->getTrkId())
            iter_cluster=aCgemClusterCol->begin()+(*it_clusterId);
            if((*iter_cluster)->getclusterid()==(*it_clusterId))
            {
                const RecCgemCluster* recCgemCluster = (*iter_cluster);
                clusterRefVec.push_back(recCgemCluster);
                int clusterid = recCgemCluster->getclusterid();
                clusterFitStat[clusterid] = 1;
            }
            else cout<<"wrong clusterRefVec assignment in RecMdcTrackCnv"<<endl;
        }
        //cout<<"recMdcTrack set a clusterRefVec with size "<<clusterRefVec.size()<<endl;
        recMdcTrack->setVecClusters(clusterRefVec,clusterFitStat);
 
        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" << endreq;
 
    RecMdcTrackCol * mdcTrackColTds=dynamic_cast<RecMdcTrackCol *> (obj);
    if (!mdcTrackColTds) {
        log << MSG::ERROR << "Could not downcast to MdcTrackCol" << 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 << "RecMdcTrackCnv:Could not downcast to TDS ReconEvent" << endreq;
    }
 
    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)<<endreq;
            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();       
        //Int_t      nCgemXCluster = (*recMdcTrack)->nCgemXCluster();       
        //Int_t      nCgemVCluster = (*recMdcTrack)->nCgemVCluster();       
 
        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()
            <<endreq;
 
        recMdcTrackRoot->setStat( stat );  
        recMdcTrackRoot->setChi2( chi );  
        recMdcTrackRoot->setNdof( ndof );
        recMdcTrackRoot->setNlayer( nlayer );  
        //recMdcTrackRoot->setNCgemXClusters(nCgemXCluster);
        //recMdcTrackRoot->setNCgemVClusters(nCgemVCluster);
        recMdcTrackRoot->setErr( er );   
        recMdcTrackRoot->setNhits( nh );  
        recMdcTrackRoot->setNster( ns  );  
        // recMdcTrackRoot->setVecHits(vecHits);
        recMdcTrackRoot->setVX0( vx0 );
        recMdcTrackRoot->setVY0( vy0 );
        recMdcTrackRoot->setVZ0( vz0 );
        recMdcTrackRoot->setFiTerm( fiterm );
 
        vector<Int_t> vecClusterId;
        ClusterRefVec clusterRefVec = (*recMdcTrack)->getVecClusters();
        ClusterRefVec::iterator itCluster=clusterRefVec.begin();
        for(; itCluster!=clusterRefVec.end(); itCluster++)
        {
            Int_t clusterId= (*itCluster)->getclusterid();
            vecClusterId.push_back(clusterId);
        }
        recMdcTrackRoot->setVecClusterId(vecClusterId);
        //cout<<"recMdcTrackRoot set a vecClusterId with size "<<vecClusterId.size()<<endl;
 
        recEvt->addRecMdcTrack(recMdcTrackRoot);
    }
 
 
    return StatusCode::SUCCESS;
}
#endif