RootIOAddress.h

Go to the documentation of this file.

#ifndef ROOTIOADDRESS_H
#define ROOTIOADDRESS_H

#include "RootIOSvc/RootInputStream.h"
#include "RootIOSvc/RootOutputStream.h"


#include "GaudiKernel/GenericAddress.h"

#include <string>

namespace RootIO {
    std::string branchname(const std::string tespath);
}


template<class StreamType>
class RootIOAddress : public GenericAddress
{
    mutable StreamType* m_stream;
public:
    RootIOAddress(const CLID& clid, const std::string& tespath)
        : GenericAddress(ROOT_StorageType,clid,tespath), m_stream(0) {}


    // Store underlying stream for later use by converter
    void setStream(StreamType* stream) { m_stream = stream; }

    // Converter uses this
    StreamType* stream() const { return m_stream; }


    const std::string& tespath() const { return m_par[0]; }

    std::string tesdir() const {
        std::string::size_type slash = m_par[0].rfind('/');
        if (!slash || slash == std::string::npos) return "/";
        return m_par[0].substr(0,slash);
    }

    std::string treename() const {
        std::string::size_type slash = m_par[0].rfind('/');
        if (!slash || slash == std::string::npos) return m_par[0];
        return m_par[0].substr(slash+1);
    }

    std::string branchname() const {
        return RootIO::branchname(tespath());
    }

    int entry() const { return m_ipar[0]; }
    void setEntry(int entry) { m_ipar[0] = entry; }
};


class RootInputAddress : public RootIOAddress<RootInputStream>
{
    bool m_read_done;
public:
    // Initial construction
    RootInputAddress(const CLID& clid, const std::string& tespath);
    int fileNumber() { return m_ipar[1]; }
    void setFileNumber(int fn) { m_ipar[1] = fn; }
    bool readIsDone() { return m_read_done; }
    void readDone(bool tf = true) { m_read_done = tf; }
};

class RootOutputAddress : public RootIOAddress<RootOutputStream>
{
public:
    // Initial construction
    RootOutputAddress(const CLID& clid, const std::string& tespath);
};

#endif  // ROOTIOADDRESS_H