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

Go to the documentation of this file.

#ifndef RECCGEMCLUSTERCNV_CXX
#define RECCGEMCLUSTERCNV_CXX
 
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/ObjectVector.h"
 
#include "TClonesArray.h"
 
#include "EventModel/EventModel.h"
 
#include "ReconEvent/ReconEvent.h"
#include "CgemRecEvent/RecCgemCluster.h"
 
#include "RootEventData/TRecCgemCluster.h"
#include "RootEventData/TRecTrackEvent.h"
 
#include "RootCnvSvc/Rec/RecTrackCnv.h"
#include "RootCnvSvc/Rec/RecCgemClusterCnv.h"
#include "RootCnvSvc/RootAddress.h"
 
#include <vector>
 
using namespace std;
 
RecCgemClusterCnv::RecCgemClusterCnv(ISvcLocator* svc)
    :RootEventBaseCnv(classID(), svc)
{
 
        MsgStream log(msgSvc(),"RecCgemClusterCnv");
        log << MSG::INFO << " RecCgemClusterCnv ------ " << endreq;
        m_rootBranchname = "m_recCgemClusterCol";
        m_adresses.push_back(&m_recCgemClusterCol);
        m_recCgemClusterCol = 0;
}
StatusCode RecCgemClusterCnv::TObjectToDataObject(DataObject*& refpObject)
{
    MsgStream log(msgSvc(),"RecCgemClusterCnv");
    log << MSG::DEBUG << "RecCgemClusterCnv::TObjectToDataObject" << endreq;
 
 
    RecCgemClusterCol* recCgemClusterCol = new RecCgemClusterCol;
    refpObject = recCgemClusterCol;
 
 
    //convert//
    if (!m_recCgemClusterCol) return StatusCode::SUCCESS;
    TIter cgemClusterIter(m_recCgemClusterCol);
    TRecCgemCluster *recCgemClusterRoot = 0;
    while ((recCgemClusterRoot = (TRecCgemCluster*)cgemClusterIter.Next()))
    {
        int clusterid = recCgemClusterRoot->getclusterid();
        int layerid   = recCgemClusterRoot->getlayerid();
        int sheetid   = recCgemClusterRoot->getsheetid();
        int flag      = recCgemClusterRoot->getflag();
        double energydeposit = recCgemClusterRoot->getenergydeposit();
        double recphi  = recCgemClusterRoot->getrecphi();
        //double recpositiony  = recCgemClusterRoot->getrecpositiony();
        double recv  = recCgemClusterRoot->getrecv();
 
        double recZ = recCgemClusterRoot->getRecZ();
        int trkid = recCgemClusterRoot->getTrkId();
        int clusterflag_first = recCgemClusterRoot->getclusterflagb();
        int clusterflag_second = recCgemClusterRoot->getclusterflage();
 
        RecCgemCluster* recCgemCluster = new RecCgemCluster();
        m_common.m_rootRecCgemClusterMap[recCgemClusterRoot] = recCgemCluster;
 
        recCgemCluster->setclusterid(clusterid);
        recCgemCluster->setlayerid(layerid);
        recCgemCluster->setsheetid(sheetid);
        recCgemCluster->setflag(flag);
        recCgemCluster->setenergydeposit(energydeposit);
        recCgemCluster->setrecphi(recphi);
        //recCgemCluster->setrecpositiony(recpositiony);
        recCgemCluster->setrecv(recv);
 
        recCgemCluster->setRecZ(recZ);
        recCgemCluster->setTrkId(trkid);
        recCgemCluster->setclusterflag(clusterflag_first,clusterflag_second);
 
        recCgemClusterCol->push_back(recCgemCluster);
    }
    return StatusCode::SUCCESS;
}
StatusCode RecCgemClusterCnv::DataObjectToTObject(DataObject* obj,RootAddress* rootaddr)
{
    MsgStream log(msgSvc(), "RecCgemClusterCnv");
    log << MSG::DEBUG << "RecCgemClusterCnv::DataObjectToTObject" << endreq;
 
    RecCgemClusterCol * cgemClusterColTds=dynamic_cast<RecCgemClusterCol *> (obj);
    if (!cgemClusterColTds) {
        log << MSG::ERROR << "Could not downcast to RecCgemClusterCol" << endreq;
        return StatusCode::FAILURE;
    }
 
    DataObject *evt;
    m_eds->findObject(EventModel::Recon::Event,evt);
    if (evt==NULL) {
        log << MSG::ERROR << "Could not get ReconEvent in TDS "  << endreq;
        return StatusCode::FAILURE;
    }
 
    ReconEvent * devtTds=dynamic_cast<ReconEvent *> (evt);
    if (!devtTds) {
        log << MSG::ERROR << "RecCgemHitCnv:Could not downcast to TDS ReconEvent" << endreq;
    }
 
    IOpaqueAddress *addr;
 
    m_cnvSvc->getRecTrackCnv()->createRep(evt,addr);
    TRecTrackEvent *recEvt=m_cnvSvc->getRecTrackCnv()->getWriteObject();
 
    const TObjArray *m_recCgemClusterCol = recEvt->getRecCgemClusterCol();
    if (!m_recCgemClusterCol) return StatusCode::SUCCESS;
    recEvt->clearRecCgemClusterCol();
 
    RecCgemClusterCol::const_iterator recCgemCluster;
    for (recCgemCluster = cgemClusterColTds->begin(); recCgemCluster != cgemClusterColTds->end(); recCgemCluster++)
    {
        Int_t layerid = (*recCgemCluster)->getlayerid();
        Int_t sheetid = (*recCgemCluster)->getsheetid();
        Int_t clusterid = (*recCgemCluster)->getclusterid();
        Int_t flag    = (*recCgemCluster)->getflag();
        Double_t energydeposit = (*recCgemCluster)->getenergydeposit();
        Double_t recphi  = (*recCgemCluster)->getrecphi();
        //Double_t recpositiony  = (*recCgemCluster)->getrecpositiony();
        Double_t recv  = (*recCgemCluster)->getrecv();
 
        Int_t trkid = (*recCgemCluster)->getTrkId();
        Double_t recZ = (*recCgemCluster)->getRecZ();
        Int_t clusterflag_b = (*recCgemCluster)->getclusterflagb();
        Int_t clusterflag_e = (*recCgemCluster)->getclusterflage();
 
        TRecCgemCluster *recCgemClusterRoot = new TRecCgemCluster();
 
        recCgemClusterRoot->setlayerid(layerid);
        recCgemClusterRoot->setsheetid(sheetid);
        recCgemClusterRoot->setclusterid(clusterid);
        recCgemClusterRoot->setflag(flag);
        recCgemClusterRoot->setenergydeposit(energydeposit);
        recCgemClusterRoot->setrecphi(recphi);
        //recCgemClusterRoot->setrecpositiony(recpositiony);
        recCgemClusterRoot->setrecv(recv);
 
        recCgemClusterRoot->setTrkId(trkid);
        recCgemClusterRoot->setRecZ(recZ);
        recCgemClusterRoot->setclusterflag(clusterflag_b,clusterflag_e);
        Int_t stripid[2][2] = {-9999,-9999,-9999,-9999};
        if(flag==0||flag==1){
            stripid[flag][0] = clusterflag_b;
            stripid[flag][1] = clusterflag_e;
        }else if(flag==2||flag==3){
            for(RecCgemClusterCol::const_iterator iter = cgemClusterColTds->begin(); iter != cgemClusterColTds->end(); iter++){
                if((*iter)->getclusterid() == clusterflag_b || (*iter)->getclusterid() == clusterflag_e){
                    Int_t Flag = (*iter)->getflag();
                    if(Flag==0||Flag==1){
                        stripid[Flag][0] = (*iter)->getclusterflagb();
                        stripid[Flag][1] = (*iter)->getclusterflage();
                    }
                }
            }
        }
        recCgemClusterRoot->setstripid(stripid);
 
        recEvt->addRecCgemCluster(recCgemClusterRoot);
    }
    return StatusCode::SUCCESS;
}
#endif