bitstream.hpp

#ifdef NALL_MOSAIC_INTERNAL_HPP

namespace nall {
namespace mosaic {

struct bitstream {
  filemap fp;
  uint8_t *data;
  unsigned size;
  bool readonly;
  bool endian;

  inline bool read(uint64_t addr) const {
    if(data == nullptr || (addr >> 3) >= size) return 0;
    unsigned mask = endian == 0 ? (0x01 << (addr & 7)) : (0x80 >> (addr & 7));
    return data[addr >> 3] & mask;
  }

  inline void write(uint64_t addr, bool value) {
    if(data == nullptr || readonly == true || (addr >> 3) >= size) return;
    unsigned mask = endian == 0 ? (0x01 << (addr & 7)) : (0x80 >> (addr & 7));
    if(value == 0) data[addr >> 3] &= ~mask;
    if(value == 1) data[addr >> 3] |=  mask;
  }

  inline bool open(const string &filename) {
    readonly = false;
    if(fp.open(filename, filemap::mode::readwrite) == false) {
      readonly = true;
      if(fp.open(filename, filemap::mode::read) == false) {
        return false;
      }
    }
    data = fp.data();
    size = fp.size();
    return true;
  }

  inline void close() {
    fp.close();
    data = nullptr;
  }

  inline bitstream() : data(nullptr), endian(1) {
  }

  inline ~bitstream() {
    close();
  }
};

}
}

#endif