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

347 lines
10 KiB
C++

#pragma once
#include "spirv_code_buffer.h"
namespace dxvk {
/**
* \brief SPIR-V module
*
* This class generates a code buffer containing a full
* SPIR-V shader module. Ensures that the module layout
* is valid, as defined in the SPIR-V 1.0 specification,
* section 2.4 "Logical Layout of a Module".
*/
class SpirvModule {
public:
SpirvModule();
~SpirvModule();
SpirvCodeBuffer compile() const;
uint32_t allocateId();
void enableCapability(
spv::Capability capability);
void addEntryPoint(
uint32_t entryPointId,
spv::ExecutionModel executionModel,
const char* name,
uint32_t interfaceCount,
const uint32_t* interfaceIds);
void setMemoryModel(
spv::AddressingModel addressModel,
spv::MemoryModel memoryModel);
void enableEarlyFragmentTests(
uint32_t entryPointId);
void setLocalSize(
uint32_t entryPointId,
uint32_t x,
uint32_t y,
uint32_t z);
void setOriginUpperLeft(
uint32_t entryPointId);
void setDebugName(
uint32_t expressionId,
const char* debugName);
void setDebugMemberName(
uint32_t structId,
uint32_t memberId,
const char* debugName);
uint32_t constBool(
bool v);
uint32_t consti32(
int32_t v);
uint32_t consti64(
int64_t v);
uint32_t constu32(
uint32_t v);
uint32_t constu64(
uint64_t v);
uint32_t constf32(
float v);
uint32_t constf64(
double v);
uint32_t constComposite(
uint32_t typeId,
uint32_t constCount,
const uint32_t* constIds);
void decorate(
uint32_t object,
spv::Decoration decoration);
void decorateArrayStride(
uint32_t object,
uint32_t stride);
void decorateBinding(
uint32_t object,
uint32_t binding);
void decorateBlock(
uint32_t object);
void decorateBuiltIn(
uint32_t object,
spv::BuiltIn builtIn);
void decorateComponent(
uint32_t object,
uint32_t location);
void decorateDescriptorSet(
uint32_t object,
uint32_t set);
void decorateLocation(
uint32_t object,
uint32_t location);
void memberDecorateBuiltIn(
uint32_t structId,
uint32_t memberId,
spv::BuiltIn builtIn);
void memberDecorateOffset(
uint32_t structId,
uint32_t memberId,
uint32_t offset);
uint32_t defVoidType();
uint32_t defBoolType();
uint32_t defIntType(
uint32_t width,
uint32_t isSigned);
uint32_t defFloatType(
uint32_t width);
uint32_t defVectorType(
uint32_t elementType,
uint32_t elementCount);
uint32_t defMatrixType(
uint32_t columnType,
uint32_t columnCount);
uint32_t defArrayType(
uint32_t typeId,
uint32_t length);
uint32_t defArrayTypeUnique(
uint32_t typeId,
uint32_t length);
uint32_t defRuntimeArrayType(
uint32_t typeId);
uint32_t defRuntimeArrayTypeUnique(
uint32_t typeId);
uint32_t defFunctionType(
uint32_t returnType,
uint32_t argCount,
const uint32_t* argTypes);
uint32_t defStructType(
uint32_t memberCount,
const uint32_t* memberTypes);
uint32_t defStructTypeUnique(
uint32_t memberCount,
const uint32_t* memberTypes);
uint32_t defPointerType(
uint32_t variableType,
spv::StorageClass storageClass);
uint32_t defSamplerType();
uint32_t defImageType(
uint32_t sampledType,
spv::Dim dimensionality,
uint32_t depth,
uint32_t arrayed,
uint32_t multisample,
uint32_t sampled,
spv::ImageFormat format);
uint32_t defSampledImageType(
uint32_t imageType);
uint32_t newVar(
uint32_t pointerType,
spv::StorageClass storageClass);
void functionBegin(
uint32_t returnType,
uint32_t functionId,
uint32_t functionType,
spv::FunctionControlMask functionControl);
uint32_t functionParameter(
uint32_t parameterType);
void functionEnd();
uint32_t opAccessChain(
uint32_t resultType,
uint32_t composite,
uint32_t indexCount,
const uint32_t* indexArray);
uint32_t opBitcast(
uint32_t resultType,
uint32_t operand);
uint32_t opCompositeConstruct(
uint32_t resultType,
uint32_t valueCount,
const uint32_t* valueArray);
uint32_t opCompositeExtract(
uint32_t resultType,
uint32_t composite,
uint32_t indexCount,
const uint32_t* indexArray);
uint32_t opCompositeInsert(
uint32_t resultType,
uint32_t object,
uint32_t composite,
uint32_t indexCount,
const uint32_t* indexArray);
uint32_t opVectorShuffle(
uint32_t resultType,
uint32_t vectorLeft,
uint32_t vectorRight,
uint32_t indexCount,
const uint32_t* indexArray);
uint32_t opSNegate(
uint32_t resultType,
uint32_t operand);
uint32_t opFNegate(
uint32_t resultType,
uint32_t operand);
uint32_t opSAbs(
uint32_t resultType,
uint32_t operand);
uint32_t opFAbs(
uint32_t resultType,
uint32_t operand);
uint32_t opIAdd(
uint32_t resultType,
uint32_t a,
uint32_t b);
uint32_t opFAdd(
uint32_t resultType,
uint32_t a,
uint32_t b);
uint32_t opIMul(
uint32_t resultType,
uint32_t a,
uint32_t b);
uint32_t opFMul(
uint32_t resultType,
uint32_t a,
uint32_t b);
uint32_t opFClamp(
uint32_t resultType,
uint32_t x,
uint32_t minVal,
uint32_t maxVal);
uint32_t opDot(
uint32_t resultType,
uint32_t vector1,
uint32_t vector2);
uint32_t opInverseSqrt(
uint32_t resultType,
uint32_t x);
uint32_t opFunctionCall(
uint32_t resultType,
uint32_t functionId,
uint32_t argCount,
const uint32_t* argIds);
void opLabel(
uint32_t labelId);
uint32_t opLoad(
uint32_t typeId,
uint32_t pointerId);
void opStore(
uint32_t pointerId,
uint32_t valueId);
uint32_t opSampledImage(
uint32_t resultType,
uint32_t image,
uint32_t sampler);
uint32_t opImageSampleImplicitLod(
uint32_t resultType,
uint32_t sampledImage,
uint32_t coordinates);
void opReturn();
private:
uint32_t m_id = 1;
uint32_t m_instExtGlsl450 = 0;
SpirvCodeBuffer m_capabilities;
SpirvCodeBuffer m_instExt;
SpirvCodeBuffer m_memoryModel;
SpirvCodeBuffer m_entryPoints;
SpirvCodeBuffer m_execModeInfo;
SpirvCodeBuffer m_debugNames;
SpirvCodeBuffer m_annotations;
SpirvCodeBuffer m_typeConstDefs;
SpirvCodeBuffer m_variables;
SpirvCodeBuffer m_code;
uint32_t defType(
spv::Op op,
uint32_t argCount,
const uint32_t* argIds);
void instImportGlsl450();
};
}