From 080c0b052e7f975ff28a824fa5f84eb9ebd5eea1 Mon Sep 17 00:00:00 2001
From: Jake Turner <jake@evansturner.co.uk>
Date: Tue, 14 May 2024 12:42:20 +0100
Subject: [PATCH] DXIL::Program added function to ProcessFunctions via callback

void ProcessFunctions(std::function<bool(const Function *)> callback) const;
---
 renderdoc/driver/shaders/dxil/dxil_bytecode.h  |  2 ++
 renderdoc/driver/shaders/dxil/dxil_reflect.cpp | 10 ++++++++++
 2 files changed, 12 insertions(+)

diff --git a/renderdoc/driver/shaders/dxil/dxil_bytecode.h b/renderdoc/driver/shaders/dxil/dxil_bytecode.h
index 4f6759d69..aa04aebce 100644
--- a/renderdoc/driver/shaders/dxil/dxil_bytecode.h
+++ b/renderdoc/driver/shaders/dxil/dxil_bytecode.h
@@ -1280,6 +1280,8 @@ public:
   rdcarray<ShaderEntryPoint> GetEntryPoints();
   void FillEntryPointInterfaces();
   rdcstr GetResourceReferenceName(ResourceClass resClass, const BindingSlot &slot) const;
+  // false means stop processing
+  void ProcessFunctions(std::function<bool(const Function *)> callback) const;
   void FillRayPayloads(
       Program *executable,
       rdcflatmap<ShaderEntryPoint, rdcpair<DXBC::CBufferVariableType, DXBC::CBufferVariableType>>
diff --git a/renderdoc/driver/shaders/dxil/dxil_reflect.cpp b/renderdoc/driver/shaders/dxil/dxil_reflect.cpp
index 71d9eb36b..abfbf33d7 100644
--- a/renderdoc/driver/shaders/dxil/dxil_reflect.cpp
+++ b/renderdoc/driver/shaders/dxil/dxil_reflect.cpp
@@ -1758,4 +1758,14 @@ const ResourceReference *Program::GetResourceReference(const rdcstr &handleStr)
   return NULL;
 }
 
+void Program::ProcessFunctions(std::function<bool(const Function *)> callback) const
+{
+  for(const DXIL::Function *f : m_Functions)
+  {
+    // false means stop processing
+    if(!callback(f))
+      break;
+  }
+}
+
 };    // namespace DXIL