/*
 * AIM tools
 * Copyright (C) 2015 lzwdgc
 *
 * This program 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.
 *
 * This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#pragma once

#include <memory>
#include <stdint.h>
#include <string>
#include <vector>

#define READ(b, var) b.read(&var, sizeof(var))
#define READ_NOTHROW(b, var) b.read(&var, sizeof(var), true)
#define READ_N(b, var, sz) b.read(&var, sz)
#define WRITE(b, var) b.write(&var, sizeof(var))

std::string version();
std::vector<uint8_t> readFile(const std::string &fn);
void writeFile(const std::string &fn, const std::vector<uint8_t> &data);

class buffer
{
public:
    buffer();
    buffer(size_t size);
    buffer(const std::vector<uint8_t> &buf, uint32_t data_offset = 0);
    buffer(buffer &rhs, uint32_t size);
    buffer(buffer &rhs, uint32_t size, uint32_t offset);

    uint32_t read(void *dst, uint32_t size, bool nothrow = false) const;
    uint32_t readfrom(void *dst, uint32_t size, uint32_t offset, bool nothrow = false) const;
    uint32_t write(const void *src, uint32_t size, bool nothrow = false);
    template <typename T>
    uint32_t write(const T &src)
    {
        return write(&src, sizeof(src));
    }

    void seek(uint32_t size) const;
    void skip(int n) const;
    bool eof() const;
    bool check(int index) const;
    void reset() const;

    uint32_t index() const;
    uint32_t size() const;
    const std::vector<uint8_t> &buf() const;

private:
    std::shared_ptr<std::vector<uint8_t>> buf_;
    mutable uint32_t index_ = 0;
    mutable uint8_t *ptr = 0;
    mutable uint32_t data_offset = 0;
    mutable uint32_t size_ = 0;
};