From e6bf418c308e882f76e2e30a12c462141ceccac6 Mon Sep 17 00:00:00 2001
From: Jake Turner <jake@evansturner.co.uk>
Date: Wed, 1 May 2024 13:35:52 +0100
Subject: [PATCH] RD DXIL Disassembly: decode "dx.op.bufferStore"
 &"dx.op.rawBufferStore"

Example Output

call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %Exposure_UAV_structbuf, i32 1, i32 0, float %75, float undef, float undef, float undef, i8 1, i32 4)
call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %Exposure_UAV_structbuf, i32 2, i32 0, float %FMin, float undef, float undef, float undef, i8 1, i32 4)
call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %Exposure_UAV_structbuf, i32 3, i32 0, float %59, float undef, float undef, float undef, i8 1, i32 4)

Becomes

Exposure[1] = {_75};
Exposure[2] = {_FMin};
Exposure[3] = {_59};
---
 .../driver/shaders/dxil/dxil_disassemble.cpp  | 70 +++++++++++++++++++
 1 file changed, 70 insertions(+)

diff --git a/renderdoc/driver/shaders/dxil/dxil_disassemble.cpp b/renderdoc/driver/shaders/dxil/dxil_disassemble.cpp
index 40770814f..f44175723 100644
--- a/renderdoc/driver/shaders/dxil/dxil_disassemble.cpp
+++ b/renderdoc/driver/shaders/dxil/dxil_disassemble.cpp
@@ -2809,6 +2809,76 @@ void Program::MakeRDDisassemblyString(const DXBC::Reflection *reflection)
                 showDxFuncName = true;
               }
             }
+            else if(showDxFuncName && funcCallName.beginsWith("dx.op.bufferStore") ||
+                    funcCallName.beginsWith("dx.op.rawBufferStore"))
+            {
+              if(funcCallName.beginsWith("dx.op.bufferStore"))
+              {
+                // uav,coord0,coord1,value0,value1,value2,value3,mask
+                showDxFuncName = false;
+                uint32_t dxopCode;
+                RDCASSERT(getival<uint32_t>(inst.args[0], dxopCode));
+                RDCASSERTEQUAL(dxopCode, 69);
+              }
+              else
+              {
+                // uav, index, elementOffset, value0, value1, value2, value3, mask, alignment
+                showDxFuncName = false;
+                uint32_t dxopCode;
+                RDCASSERT(getival<uint32_t>(inst.args[0], dxopCode));
+                RDCASSERTEQUAL(dxopCode, 140);
+              }
+
+              rdcstr handleStr = ArgToString(inst.args[1], false);
+              if(resHandles.count(handleStr) > 0)
+              {
+                uint32_t offset = 0;
+                bool validElementOffset = !isUndef(inst.args[3]);
+                bool constantElementOffset = validElementOffset && getival(inst.args[3], offset);
+
+                lineStr += resHandles[handleStr].name;
+                uint32_t index;
+                if(getival(inst.args[2], index))
+                {
+                  if((offset == 0) || (index > 0))
+                    lineStr += "[" + ToStr(index) + "]";
+                }
+                else
+                {
+                  lineStr += "[" + ArgToString(inst.args[2], false) + "]";
+                }
+                if(validElementOffset)
+                {
+                  if(constantElementOffset)
+                  {
+                    if(offset > 0)
+                      lineStr += " + " + ToStr(offset) + " bytes";
+                  }
+                  else
+                  {
+                    lineStr += " + " + ArgToString(inst.args[3], false) + " bytes";
+                  }
+                }
+                lineStr += " = ";
+                lineStr += "{";
+                bool needComma = false;
+                for(uint32_t a = 4; a < 8; ++a)
+                {
+                  if(!isUndef(inst.args[a]))
+                  {
+                    if(needComma)
+                      lineStr += ", ";
+                    lineStr += ArgToString(inst.args[a], false);
+                    needComma = true;
+                  }
+                }
+                lineStr += "}";
+              }
+              else
+              {
+                showDxFuncName = true;
+              }
+            }
             else if(funcCallName.beginsWith("llvm.dbg."))
             {
             }