RootIOEvtSelector.cc

Go to the documentation of this file.

#include "RootIOSvc/RootIOEvtSelector.h"

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

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

using namespace std;

RootIOEvtSelector::RootIOEvtSelector(const string& name, 
                                     ISvcLocator* svcloc)
    : Service(name,svcloc)
    , m_rioSvc(0)
{

}
RootIOEvtSelector::~RootIOEvtSelector()
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::DEBUG << "RootIOEvtSelector::~RootIOEvtSelector()" << endreq;
}

StatusCode RootIOEvtSelector::initialize()
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::DEBUG << "initializing @" << (void*)this << endreq;
    log << MSG::DEBUG << "called " << this->name() << endreq;

    StatusCode sc = this->Service::initialize();
    if (sc.isFailure()) return sc;

    IService* isvc=0;
    sc = serviceLocator()->service("RootIOCnvSvc", isvc, false);
    //sc = serviceLocator()->service("EventPersistencySvc", 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;
    }

    log << MSG::DEBUG << "initialized @" << (void*)this
        << " got RootIOCnvSvc @" << (void*)m_rioSvc
        << endreq;
    return sc;
}
StatusCode RootIOEvtSelector::finalize()
{
    return this->Service::finalize();
}


StatusCode RootIOEvtSelector::createContext(IEvtSelector::Context*& c) const
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::DEBUG << "createContext" << endreq;

    c = new RootIOEvtSelector::Context(this);
    return StatusCode::SUCCESS;
}

StatusCode RootIOEvtSelector::next(IEvtSelector::Context& c) const
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::DEBUG << "next" << endreq;
    return this->next(c,1);
}

StatusCode RootIOEvtSelector::next(IEvtSelector::Context& c, int jump) const
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::DEBUG << "next(" << jump << ")" << endreq;
    RootIOEvtSelector::Context* rc = 
        dynamic_cast<RootIOEvtSelector::Context*>(&c);
    if (!rc) return StatusCode::FAILURE;
    return this->setEntry(*rc,rc->m_entry+jump);
}

StatusCode RootIOEvtSelector::previous(IEvtSelector::Context& c) const
{
    return this->previous(c,1);
}

StatusCode RootIOEvtSelector::previous(IEvtSelector::Context& c, int jump) const
{
    RootIOEvtSelector::Context* rc = 
        dynamic_cast<RootIOEvtSelector::Context*>(&c);
    if (!rc) return StatusCode::FAILURE;
    rc->m_entry -= jump;
    return this->setEntry(*rc,rc->m_entry-jump);
}

StatusCode RootIOEvtSelector::last(IEvtSelector::Context&) const
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::ERROR << "last() not currently supported" << endreq;
    return StatusCode::FAILURE;
}

StatusCode RootIOEvtSelector::rewind(IEvtSelector::Context&) const
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::ERROR << "rewind() not currently supported" << endreq;
    return StatusCode::FAILURE;
}

StatusCode RootIOEvtSelector::setEntry(RootIOEvtSelector::Context& rc, int entry) const
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::DEBUG << "setEntry(" << entry << ")" << endreq;

    IRootIOSvc::InputStreamMap& ism = m_rioSvc->inputStreams();
    IRootIOSvc::InputStreamMap::iterator it, done = ism.end();
    int errors = 0;
    for (it = ism.begin(); it != done; ++it) {
        RootInputStream* ris = it->second;
        bool okay = ris->setEntry(entry);
        if (!okay) {
            log << MSG::WARNING
                << "setEntry("<<entry<<") failed to set entry on input stream for " 
                << it->first << endreq;
            ++errors;
            continue;
        }
    }

    if (errors) return StatusCode::FAILURE;
    rc.m_entry = entry;
    return StatusCode::SUCCESS;
}


StatusCode RootIOEvtSelector::createAddress(const IEvtSelector::Context& c,
                                            IOpaqueAddress*& iop) const
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::DEBUG << "createAddress" << endreq;
    const RootIOEvtSelector::Context* rc = 
        dynamic_cast<const RootIOEvtSelector::Context*>(&c);
    if (!rc) {
        log << MSG::ERROR << "Failed to get RootIOEvtSelector::Context" << endreq;
        return StatusCode::FAILURE;
    }

    RootInputAddress* ria = 
        new RootInputAddress(DataObject::classID(),"/Event");
    ria->setEntry(rc->m_entry);
    iop = ria;
    log << MSG::DEBUG << "createAddress for /Event" << endreq;
    return StatusCode::SUCCESS;
}

StatusCode RootIOEvtSelector::releaseContext(IEvtSelector::Context*& c) const

{
    RootIOEvtSelector::Context* rc = 
        dynamic_cast<RootIOEvtSelector::Context*>(c);
    if (!rc) return StatusCode::FAILURE;

    delete rc;
    c = 0;
    return StatusCode::SUCCESS;
}

StatusCode RootIOEvtSelector::resetCriteria(const string& cr,
                                            IEvtSelector::Context& c) const
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");
    log << MSG::INFO << "resetCriteria(" << cr << ")" << endreq;

    RootIOEvtSelector::Context* rc = 
        dynamic_cast<RootIOEvtSelector::Context*>(&c);
    if (!rc) return StatusCode::FAILURE;

    // what to do here???

    return StatusCode::SUCCESS;
}

StatusCode RootIOEvtSelector::queryInterface(const InterfaceID& riid, 
                                             void** ppif)
{
    MsgStream log(msgSvc(), "RootIOEvtSelector");

    if (riid == IID_IEvtSelector) {
        *ppif = (IEvtSelector*)this;
        addRef();
        log << MSG::DEBUG << "queryInterface(IID_IEvtSelector)" << endreq;
        return StatusCode::SUCCESS;
    }
    log << MSG::DEBUG << "queryInterface(" << riid << ")" << endreq;
    return Service::queryInterface(riid,ppif);
}