MemoryReader.hpp

// This file is part of libAthena.
//
// libAthena is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// libAthena is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with libAthena.  If not, see <http://www.gnu.org/licenses/>

#ifndef MEMORYREADER_HPP
#define MEMORYREADER_HPP

#include "Athena/IStreamReader.hpp"
#include <string>
#include <functional>

namespace Athena
{
namespace io
{
class MemoryReader : public IStreamReader
{
public:
    MemoryReader(const atUint8* data, atUint64 length);

    MemoryReader(const std::string& filename, std::function<void(int)> progressFun = nullptr);

    virtual ~MemoryReader();

    void setEndian(Endian endian);

    Endian endian()      const;

    bool isBigEndian()   const;

    bool isLittleEndian()const;

    bool isOpen() const;

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

    inline void seekAlign32() {seek(ROUND_UP_32(m_position), SeekOrigin::Begin);}

    bool atEnd() const;

    atUint64 position() const;

    atUint64 length() const;

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


    atUint8* data() const;

    void setFilepath(const std::string& filepath);

    std::string filepath() const;

    void seekBit(int bit);

    bool readBit();

    atInt8 readByte();

    atUint8 readUByte();

    inline atInt8* readBytes(atUint64 length) {return (atInt8*)readUBytes(length);}

    atUint8* readUBytes(atUint64 length);

    atUint64 readBytesToBuf(void* buf, atUint64 len) {return readUBytesToBuf(buf, len);}
    atUint64 readUBytesToBuf(void* buf, atUint64 len);

    atInt16 readInt16();

    atUint16 readUint16();

    atInt32 readInt32();

    atUint32 readUint32();

    atInt64 readInt64();

    atUint64 readUint64();

    float readFloat();

    double readDouble();

    bool readBool();

    std::string readUnicode(atInt32 maxlen = -1);

    std::string readString(atInt32 maxlen = -1);

    void setProgressCallback(std::function<void(int)> cb);
protected:
    void loadData();
    atUint8*      m_data;
    atUint64      m_length;
    std::string m_filepath; 
    atUint64      m_position;
    atUint64      m_bitPosition;
    Endian      m_endian;
    std::function<void(int)> m_progressCallback;
};

} // io
} // Athena

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

#endif // MEMORYREADER_BASE

#endif // MEMORYREADER_HPP