lee/renderdoc
1
0
Fork 0
mirror of https://github.com/baldurk/renderdoc.git synced 2026-09-22 21:55:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a SPIR-V editor helper to add or re-use a builtin input

Browse Source
This commit is contained in:
baldurk
2023-10-30 10:24:30 +00:00
parent f87c50af25
commit 2b779038af
2 changed files with 100 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
renderdoc/driver/shaders/spirv/spirv_editor.cpp
+84
View File
@@ -403,6 +403,54 @@ Iter Editor::GetEntry(Id id)
return Iter();
}
rdcpair<Id, Id> Editor::AddBuiltinInputLoad(OperationList &ops, ShaderStage stage, BuiltIn builtin,
Id type)
{
BuiltinInputData &data = builtinInputs[builtin];
Id ptrType = DeclareType(Pointer(type, StorageClass::Input));
if(data.variable == Id())
{
const DataType &dataType = dataTypes[type];
Id var = AddVariable(OpVariable(ptrType, MakeId(), StorageClass::Input));
AddDecoration(OpDecorate(var, DecorationParam<Decoration::BuiltIn>(builtin)));
// Fragment shader inputs that are signed or unsigned integers, integer vectors, or any
// double-precision floating-point type must be decorated with Flat.
if(dataType.scalar().type != Op::TypeFloat && stage == ShaderStage::Pixel)
AddDecoration(OpDecorate(var, Decoration::Flat));
builtinInputs[builtin] = {var, ptrType, {}};
return {ops.add(OpLoad(type, MakeId(), var)), var};
}
Id varType = dataTypes[data.type].InnerType();
Id ret;
if(data.chain.empty())
{
ret = ops.add(OpLoad(varType, MakeId(), data.variable));
}
else
{
rdcarray<rdcspv::Id> chain;
for(uint32_t accessIdx : data.chain)
chain.push_back(AddConstantImmediate<uint32_t>(accessIdx));
Id subElement = ops.add(OpAccessChain(ptrType, MakeId(), data.variable, chain));
ret = ops.add(rdcspv::OpLoad(varType, MakeId(), subElement));
}
if(varType != type)
ret = ops.add(rdcspv::OpBitcast(type, MakeId(), ret));
return {ret, Id()};
}
Id Editor::DeclareStructType(const rdcarray<Id> &members)
{
Id typeId = MakeId();
@@ -544,6 +592,42 @@ void Editor::UnregisterOp(Iter it)
}
}
void Editor::RegisterBuiltinMembers(rdcspv::Id baseId, const rdcarray<uint32_t> chainSoFar,
const DataType *type)
{
uint32_t i = 0;
for(const DataType::Child &member : type->children)
{
if(member.decorations.flags & Decorations::HasBuiltIn)
{
rdcarray<uint32_t> chain = chainSoFar;
chain.push_back(i);
builtinInputs[member.decorations.builtIn] = {baseId, member.type, chain};
}
i++;
}
}
void Editor::PostParse()
{
Processor::PostParse();
for(const Variable &v : globals)
{
if(v.storage == StorageClass::Input)
{
if(decorations[v.id].flags & Decorations::HasBuiltIn)
{
builtinInputs[decorations[v.id].builtIn] = {v.id, v.type, {}};
}
else
{
RegisterBuiltinMembers(v.id, {}, &dataTypes[v.type]);
}
}
}
}
void Editor::addWords(size_t offs, int32_t num)
{
// look through every section, any that are >= this point, adjust the offsets
renderdoc/driver/shaders/spirv/spirv_editor.h
+16
View File
@@ -174,6 +174,9 @@ public:
return it->second;
}
rdcpair<Id, Id> AddBuiltinInputLoad(OperationList &ops, ShaderStage stage, BuiltIn builtin,
Id type);
Id DeclareStructType(const rdcarray<Id> &members);
// helper for AddConstant
@@ -252,6 +255,19 @@ private:
virtual void RegisterOp(Iter iter);
virtual void UnregisterOp(Iter iter);
virtual void PostParse();
void RegisterBuiltinMembers(rdcspv::Id baseId, const rdcarray<uint32_t> chainSoFar,
const DataType *type);
struct BuiltinInputData
{
Id variable;
Id type;
rdcarray<uint32_t> chain;
};
std::map<BuiltIn, BuiltinInputData> builtinInputs;
std::map<Id, Binding> bindings;