1
0
mirror of https://github.com/doitsujin/dxvk.git synced 2024-12-11 19:24:11 +01:00
dxvk/src/spirv/spirv_instruction.h

128 lines
2.9 KiB
C
Raw Normal View History

2017-10-17 13:02:57 +02:00
#pragma once
#include <spirv/spirv.hpp>
#include <spirv/GLSL.std.450.hpp>
2017-10-17 13:02:57 +02:00
#include "spirv_include.h"
2017-10-17 13:02:57 +02:00
namespace dxvk {
/**
* \brief SPIR-V instruction
*
* Helps parsing a single instruction, providing
* access to the op code, instruction length and
* instruction arguments.
*/
class SpirvInstruction {
2017-10-17 13:02:57 +02:00
public:
SpirvInstruction() { }
SpirvInstruction(uint32_t* code, uint32_t size)
2017-10-17 13:02:57 +02:00
: m_code(code), m_size(size) { }
/**
* \brief SPIR-V Op code
* \returns The op code
*/
spv::Op opCode() const {
return static_cast<spv::Op>(
this->arg(0) & spv::OpCodeMask);
2017-10-17 13:02:57 +02:00
}
/**
* \brief Instruction length
* \returns Number of DWORDs
*/
uint32_t length() const {
return this->arg(0) >> spv::WordCountShift;
2017-10-17 13:02:57 +02:00
}
/**
* \brief Argument value
*
* Retrieves an argument DWORD. Note that some instructions
* take 64-bit arguments which require more than one DWORD.
* Arguments start at index 1. Calling this method with an
* argument ID of 0 will return the opcode token.
* \param [in] id Argument index, starting at 1
* \returns The argument value
*/
uint32_t arg(uint32_t id) const {
return id < m_size ? m_code[id] : 0;
}
/**
* \brief Changes the value of an argument
*
* \param [in] id Argument index, starting at 1
* \param [in] word New argument word
*/
void setArg(uint32_t id, uint32_t word) const {
if (id < m_size)
m_code[id] = word;
}
2017-10-17 13:02:57 +02:00
private:
uint32_t* m_code = nullptr;
uint32_t m_size = 0;
2017-10-17 13:02:57 +02:00
};
/**
* \brief SPIR-V instruction
*
* Convenient iterator that can be used
* to process raw SPIR-V shader code.
*/
class SpirvInstructionIterator {
2017-10-17 13:02:57 +02:00
public:
SpirvInstructionIterator() { }
SpirvInstructionIterator(uint32_t* code, uint32_t size)
2017-10-17 13:02:57 +02:00
: m_code(size != 0 ? code : nullptr), m_size(size) {
if ((size >= 5) && (m_code[0] == spv::MagicNumber))
this->advance(5);
}
SpirvInstructionIterator& operator ++ () {
this->advance(SpirvInstruction(m_code, m_size).length());
2017-10-17 13:02:57 +02:00
return *this;
}
SpirvInstruction operator * () const {
return SpirvInstruction(m_code, m_size);
2017-10-17 13:02:57 +02:00
}
bool operator == (const SpirvInstructionIterator& other) const {
2017-10-17 13:02:57 +02:00
return this->m_code == other.m_code
&& this->m_size == other.m_size;
}
bool operator != (const SpirvInstructionIterator& other) const {
2017-10-17 13:02:57 +02:00
return this->m_code != other.m_code
&& this->m_size != other.m_size;
}
private:
uint32_t* m_code = nullptr;
uint32_t m_size = 0;
2017-10-17 13:02:57 +02:00
void advance(uint32_t n) {
if (n < m_size) {
2017-10-17 13:02:57 +02:00
m_code += n;
m_size -= n;
} else {
m_code = nullptr;
m_size = 0;
}
}
};
}