MemoryReader.hpp

#ifndef MEMORYREADER_HPP
#define MEMORYREADER_HPP

#include <string>
#include <memory>
#include <functional>
#include "athena/IStreamReader.hpp"

namespace athena
{
namespace io
{
class MemoryReader : public IStreamReader
{
protected:
    MemoryReader() = default;
public:
    virtual ~MemoryReader();

    MemoryReader(const void* data, atUint64 length, bool takeOwnership=false, bool globalErr=true);

    void seek(atInt64 pos, SeekOrigin origin = SeekOrigin::Current);

    inline atUint64 position() const
    {return m_position;}

    inline atUint64 length() const
    {return m_length;}


    void setData(const atUint8* data, atUint64 length, bool takeOwnership=false);


    atUint8* data() const;

    atUint64 readUBytesToBuf(void* buf, atUint64 len);

protected:
    const void*      m_data = nullptr;
    atUint64         m_length = 0;
    atUint64         m_position = 0;
    bool             m_owns = false;
    bool             m_globalErr = true;
};

class MemoryCopyReader : public MemoryReader
{
public:
    MemoryCopyReader(const void* data, atUint64 length);

    MemoryCopyReader(const std::string& filename)
    : m_filepath(filename)
    {loadData();}

    void setData(const atUint8* data, atUint64 length);

protected:
    void loadData();
    std::unique_ptr<atUint8[]> m_dataCopy;
    std::string   m_filepath; 
};

} // io
} // Athena

#ifndef MEMORYREADER_BASE
#define MEMORYREADER_BASE() \
private: \
    typedef athena::io::MemoryReader base

#endif // MEMORYREADER_BASE

#ifndef MEMORYCOPYREADER_BASE
#define MEMORYCOPYREADER_BASE() \
private: \
    typedef athena::io::MemoryCopyReader base

#endif // MEMORYCOPYREADER_BASE

#endif // MEMORYREADER_HPP