OpenDX/src/dxso/dxso_reader.h

#pragma once

#include "dxso_include.h"

#include "../dxbc/dxbc_tag.h"

#include <cstdint>

namespace dxvk {

  /**
    * \brief DXSO (d3d9) bytecode reader
    *
    * Holds references to the shader byte code and
    * provides methods to read
    */
  class DxsoReader {

  public:

    DxsoReader(const char* data)
      : DxsoReader(data, 0) { }

    size_t pos() {
      return m_pos;
    }

    auto readu32() { return this->readNum<uint32_t> (); }
    auto readf32() { return this->readNum<float>    (); }

    DxbcTag readTag();

    void read(void* dst, size_t n);

    void skip(size_t n);

    void store(std::ostream&& stream, size_t size) const;

    const char* currentPtr() {
      return m_data + m_pos;
    }

  private:

    DxsoReader(const char* data, size_t pos)
      : m_data(data), m_pos(pos) { }

    const char* m_data = nullptr;
    size_t      m_pos = 0;

    template<typename T>
    T readNum() {
      T result;
      this->read(&result, sizeof(result));
      return result;
    }

  };

}