RootIOBaseCnv.cc

Go to the documentation of this file.

#include "RootIOSvc/RootIOBaseCnv.h"
#include "RootIOSvc/RootIOAddress.h"

#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"

RootIOBaseCnv::RootIOBaseCnv(const CLID& clid, ISvcLocator* svc)
    : Converter(ROOT_StorageType,clid,svc)
    , m_rioSvc(0)
{
}
      
RootIOBaseCnv::~RootIOBaseCnv()
{
}

StatusCode RootIOBaseCnv::initialize()
{
    StatusCode sc = Converter::initialize();
    if (sc.isFailure()) return sc;

    MsgStream log(msgSvc(), "RootIOBaseCnv");

    IService* isvc=0;
    sc = serviceLocator()->service("RootIOCnvSvc", isvc, false);
    if (sc.isFailure()) {
        log << MSG::ERROR << "Conversion service RootIOCnvSvc"
            << " could not be retrieved" << endreq;
        return sc;
    }
    isvc->addRef();

    sc = isvc->queryInterface(IRootIOSvc::interfaceID(), (void**)&m_rioSvc);
    if (sc.isFailure()) {
        log << MSG::ERROR << "Conversion service RootIOCnvSvc"
            << " does not implement IRootIOCnvSvc" << endreq;
        return sc;
    }

    sc = isvc->queryInterface(IConversionSvc::interfaceID(),(void**)&m_cnvSvc);
    if (sc.isFailure()) {
        log << MSG::ERROR 
            << "RootIOCnvSvc does not implement IConversionSvc" << endreq;
        return sc;
    }    

    return sc;
}


StatusCode RootIOBaseCnv::createObj(IOpaqueAddress* addr, DataObject*& dat)
{
    MsgStream log(msgSvc(), "RootIOBaseCnv::createObj");
    log << MSG::DEBUG << "createObj " << objType() << endreq;

    RootInputAddress* ria = dynamic_cast<RootInputAddress*>(addr);
    if (!ria) {
        log << MSG::ERROR << "was not given a RootInputAddress" << endreq;
        return StatusCode::FAILURE;
    }
    
    RootInputStream* ris = ria->stream();
    if (!ris) {
        log << MSG::ERROR << "RootInputAddress did not have stream" << endreq;
        return StatusCode::FAILURE;
    }

    // Read in persistent object unless already done so.
    if (!ria->readIsDone()) {
        bool okay = false;
        int fileNumber = ria->fileNumber();

        // Set entry but don't read.
        if (fileNumber == -1) {
            log << MSG::DEBUG << "setEntry(" << ria->entry() <<")"<< endreq;
            okay = ris->setEntry(ria->entry(),false);
        }
        else {
            log << MSG::DEBUG << "setFileEntry(" << fileNumber 
                <<","<< ria->entry() <<")"<< endreq;
            okay = ris->setFileEntry(fileNumber,ria->entry(),false);
            if (!okay) {
                log << MSG::ERROR << "failed to set " << objType() 
                    << " entry to " << ria->entry() << " fileNumber=" << fileNumber
                    << endreq;
                return StatusCode::FAILURE;
            }
        }

        // Now read
        okay = ris->read();
        if (!okay) {
            log << MSG::ERROR << "failed to read " << objType() 
                << " entry " << ria->entry()
                << endreq;
            return StatusCode::FAILURE;
        }
        ria->readDone();
    }

    m_ris = ris;
    return PerObjectToTranObject(dat);
}

StatusCode RootIOBaseCnv::createRep(DataObject* obj, 
                                    IOpaqueAddress*& addr)   
{

    // Purpose and Method: Convert the transient object to ROOT

    MsgStream log(msgSvc(), "RootIOBaseCnv");
    log << MSG::DEBUG << "createRep " << objType() << endreq;

    // get the corresponding address 
    RootOutputAddress *roa = 0;
    StatusCode sc = m_rioSvc->createAddress(obj,roa);
    addr = roa;
    
    //  do the real conversion in the derived converter
    sc = TranObjectToPerObject(*obj,*roa);
    if (sc.isFailure()) {
        log << MSG::ERROR << "Could not transform object" << endreq;
        return sc;
    }

    return StatusCode::SUCCESS;
}

StatusCode RootIOBaseCnv::fillRepRefs(IOpaqueAddress* /*pAddress*/,DataObject* /*pObject*/)
{
    MsgStream log(msgSvc(), "RootIOBaseCnv");
    log << MSG::DEBUG << "fillRepRefs " << objType() << endreq;

    return StatusCode::SUCCESS;
}

StatusCode RootIOBaseCnv::fillObjRefs(IOpaqueAddress* /*pAddress*/,DataObject* /*pObject*/)
{
    return StatusCode::SUCCESS;
}

StatusCode RootIOBaseCnv::finalize()
{
    return StatusCode::SUCCESS;
}


int RootIOBaseCnv::commit(const RootOutputAddress& roa) 
{
    RootOutputStream* ros = m_rioSvc->outputStream(roa);
    if (!ros) return StatusCode::FAILURE;
    //int entries = ros->entries();
    int entries = ros->fileEntries();
    bool okay = ros->write();
    if (okay) return entries;
    return -1;
}


RootIOBaseCnv* RootIOBaseCnv::otherConverter(int clID)
{
    MsgStream log(msgSvc(), "RootIOBaseCnv::otherConverter");

    IConversionSvc* cnvSvc = this->conversionSvc();
    if (!cnvSvc) {
        log << MSG::WARNING
            << "Failed to get conversion service" << endreq;
        return 0;
    }
    IConverter* cnv = cnvSvc->converter(clID);
    if (!cnv) {
        log << MSG::WARNING
            << "Failed to get converter for class ID #" << clID << endreq;
        return 0;
    }

    RootIOBaseCnv* base = dynamic_cast<RootIOBaseCnv*>(cnv);
    if (!base) {
        log << MSG::WARNING
            << "Failed to dynamic_cast converter to RootIOBaseCnv" << endreq;
    }
    return base;
}