MdcMcHitCnv.cxx

Go to the documentation of this file.

#ifndef MdcMcHitCnv_CXX
#define MdcMcHitCnv_CXX  1
 
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/ObjectVector.h"
 
#include "TClonesArray.h"
 
#include "EventModel/EventModel.h"
#include "McTruth/McEvent.h"   //TDS object
#include "McTruth/MdcMcHit.h"   //TDS object
#include "RootEventData/TMdcMc.h" // standard root object
#include "RootEventData/TMcEvent.h"
#include "RootCnvSvc/Mc/McCnv.h"
#include "RootCnvSvc/Mc/MdcMcHitCnv.h"
#include "RootCnvSvc/RootAddress.h"
 
 
// Instantiation of a static factory class used by clients to create
// instances of this service
//static CnvFactory<MdcMcHitCnv> s_factory;
//const ICnvFactory& MdcMcHitCnvFactory = s_factory;
 
MdcMcHitCnv::MdcMcHitCnv(ISvcLocator* svc)
: RootEventBaseCnv(classID(), svc)
{
    // Here we associate this converter with the /Event path on the TDS.
    MsgStream log(msgSvc(), "MdcMcHitCnv");
    //log << MSG::DEBUG << "Constructor called for " << objType() << endreq;
    m_rootBranchname ="m_mdcMcHitCol";
    //declareObject(EventModel::MC::MdcMcHitCol, objType(), m_rootTreename, m_rootBranchname);
    m_adresses.push_back(&m_mdcMcHitCol);
    m_mdcMcHitCol=0;
}
 
StatusCode MdcMcHitCnv::TObjectToDataObject(DataObject*& refpObject) {
    // creation of TDS object from root object
 
    MsgStream log(msgSvc(), "MdcMcHitCnv");
    log << MSG::DEBUG << "MdcMcHitCnv::TObjectToDataObject" << endreq;
    StatusCode sc=StatusCode::SUCCESS;
 
    // create the TDS location for the MdcMc Collection
    MdcMcHitCol* mdcMcTdsCol = new MdcMcHitCol;
    refpObject=mdcMcTdsCol;
 
 
    // now convert
    if (!m_mdcMcHitCol) return sc;
    TIter mdcMcIter(m_mdcMcHitCol);
    TMdcMc *mdcMcRoot = 0;
    while ((mdcMcRoot = (TMdcMc*)mdcMcIter.Next())) {
        unsigned int id = mdcMcRoot ->getId();
        unsigned int trackIndex = mdcMcRoot ->getTrackIndex();
        double xPosition =   mdcMcRoot ->getPositionX() ;
        double yPosition =   mdcMcRoot ->getPositionY() ;
        double zPosition =   mdcMcRoot ->getPositionZ() ;
        double  driftDistance= mdcMcRoot ->getDriftDistance() ;
        double depositEnergy = mdcMcRoot ->getDepositEnergy() ;
        int positionFlag = mdcMcRoot->getPositionFlag();
        double px = mdcMcRoot->getMomentumX();
        double py = mdcMcRoot->getMomentumY();
        double pz = mdcMcRoot->getMomentumZ();
        double trkLen = mdcMcRoot->getFlightLength();
        TString creatorProcess = mdcMcRoot->getCreatorProcess();
        int    digiIdx = mdcMcRoot->getDigiIdx();
        int    pdgCode = mdcMcRoot->getPDGCode();
        int    isSec   = mdcMcRoot->getIsSecondary();
        
        MdcMcHit *mdcMcTds = new MdcMcHit;
        m_common.m_rootMdcMcHitMap[mdcMcRoot] = mdcMcTds;
 
        mdcMcTds->setIdentifier(id);
        mdcMcTds->setTrackIndex(trackIndex);
        mdcMcTds->setPositionX(xPosition);
        mdcMcTds->setPositionY(yPosition);
        mdcMcTds->setPositionZ(zPosition);
        mdcMcTds->setDriftDistance(driftDistance);
        mdcMcTds->setDepositEnergy(depositEnergy);
        mdcMcTds->setPositionFlag(positionFlag);
        mdcMcTds->setMomentumX(px);
        mdcMcTds->setMomentumY(py);
        mdcMcTds->setMomentumZ(pz);
        mdcMcTds->setFlightLength(trkLen);
        mdcMcTds->setCreatorProcess(creatorProcess.Data());
        mdcMcTds->setDigiIdx(digiIdx);
        mdcMcTds->setCurrentTrackPID(pdgCode);
        mdcMcTds->setIsSecondary(isSec);
 
        mdcMcTdsCol->push_back(mdcMcTds);
    }
    //m_mdcMcHitCol->Delete();  // wensp add 2005/12/30
    delete m_mdcMcHitCol;
    m_mdcMcHitCol = 0;
    return StatusCode::SUCCESS;
}
 
StatusCode MdcMcHitCnv::DataObjectToTObject(DataObject* obj,RootAddress* rootaddr) {
 
    MsgStream log(msgSvc(), "MdcMcHitCnv");
    log << MSG::DEBUG << "MdcMcHitCnv::DataObjectToTObject" << endreq;
    StatusCode sc=StatusCode::SUCCESS;
 
    MdcMcHitCol * mdcMcHitCnvTds=dynamic_cast<MdcMcHitCol *> (obj);
    if (!mdcMcHitCnvTds) {
        log << MSG::ERROR << "Could not downcast to MdcMcHitCol" << endreq;
        return StatusCode::FAILURE;
    }
 
    DataObject *evt;
    m_eds->findObject(EventModel::MC::Event,evt);
    if (evt==NULL) {
        log << MSG::ERROR << "Could not get McEvent in TDS "  << endreq;
        return StatusCode::FAILURE;
    }
    McEvent * devtTds=dynamic_cast<McEvent *> (evt);
    if (!devtTds) {
        log << MSG::ERROR << "MdcMcHitCnv:Could not downcast to TDS McEvent" << endreq;
    }
    IOpaqueAddress *addr;
 
    m_cnvSvc->getMcCnv()->createRep(evt,addr); 
    TMcEvent *McEvt=m_cnvSvc->getMcCnv()->getWriteObject();
 
    const TObjArray *m_mdcMcHitCol = McEvt->getMdcMcHitCol();
    if (!m_mdcMcHitCol) return sc;
    McEvt->clearMdcMcHitCol(); //necessary in case there is I/O at the same time since array is static
    MdcMcHitCol::const_iterator mdcMcTds;
 
    for (mdcMcTds = mdcMcHitCnvTds->begin(); mdcMcTds != mdcMcHitCnvTds->end(); mdcMcTds++) {
        UInt_t id  = ((*mdcMcTds)->identify()).get_value() ;
        UInt_t trackIndex = (*mdcMcTds) ->getTrackIndex();
        Double_t xPosition =   (*mdcMcTds) ->getPositionX() ;
        Double_t yPosition =   (*mdcMcTds) ->getPositionY() ;
        Double_t zPosition =   (*mdcMcTds) ->getPositionZ() ;
        Double_t driftDistance=   (*mdcMcTds) ->getDriftDistance();
        Double_t depositEnergy = (*mdcMcTds) ->getDepositEnergy() ;
        Int_t positionFlag = (*mdcMcTds) ->getPositionFlag();
        Double_t px = (*mdcMcTds)->getMomentumX();
        Double_t py = (*mdcMcTds)->getMomentumY();
        Double_t pz = (*mdcMcTds)->getMomentumZ();
        Double_t trkLen = (*mdcMcTds)->getFlightLength();
        TString  creatorProcess = (*mdcMcTds)->getCreatorProcess();
        Int_t    trkIdx = (*mdcMcTds)->getDigiIdx();
        Int_t    pdgCode = (*mdcMcTds)->getCurrentTrackPID();
        Int_t    isSec   = (*mdcMcTds)->getIsSecondary();
 
        TMdcMc *mdcMcRoot = new TMdcMc();
        //m_common.m_mdcMcHitMap[(*mdcMcTds)] = mdcMcRoot;
 
        mdcMcRoot->setId(id);
        mdcMcRoot->setTrackIndex(trackIndex);
        mdcMcRoot->setPositionX(xPosition);
        mdcMcRoot->setPositionY(yPosition);
        mdcMcRoot->setPositionZ(zPosition);
        mdcMcRoot->setDriftDistance(driftDistance);
        mdcMcRoot->setDepositEnergy(depositEnergy);
        mdcMcRoot->setPositionFlag(positionFlag);
        mdcMcRoot->setMomentumX(px);
        mdcMcRoot->setMomentumY(py);
        mdcMcRoot->setMomentumZ(pz);
        mdcMcRoot->setFlightLength(trkLen);
        mdcMcRoot->setCreatorProcess(creatorProcess);
        mdcMcRoot->setDigiIdx(trkIdx);
        mdcMcRoot->setPDGCode(pdgCode);
        mdcMcRoot->setIsSecondary(isSec);
 
        McEvt->addMdcMc(mdcMcRoot);
    }
 
    return StatusCode::SUCCESS;
}
#endif