From da2af696718d8e5c8d57c5dbce903cf340325cea Mon Sep 17 00:00:00 2001
From: baldurk <baldurk@baldurk.org>
Date: Mon, 11 May 2020 17:07:24 +0100
Subject: [PATCH] Add support for specifying strip restart value in D3D12 PSOs

---
 util/test/demos/d3d12/d3d12_helpers.cpp | 6 ++++++
 util/test/demos/d3d12/d3d12_helpers.h   | 2 ++
 2 files changed, 8 insertions(+)

diff --git a/util/test/demos/d3d12/d3d12_helpers.cpp b/util/test/demos/d3d12/d3d12_helpers.cpp
index 4d7e07fbc..72d404e6e 100644
--- a/util/test/demos/d3d12/d3d12_helpers.cpp
+++ b/util/test/demos/d3d12/d3d12_helpers.cpp
@@ -965,6 +965,12 @@ D3D12PSOCreator &D3D12PSOCreator::InputLayout()
   return InputLayout(m_Test->DefaultInputLayout());
 }
 
+D3D12PSOCreator &D3D12PSOCreator::StripRestart(D3D12_INDEX_BUFFER_STRIP_CUT_VALUE stripCut)
+{
+  GraphicsDesc.IBStripCutValue = stripCut;
+  return *this;
+}
+
 D3D12PSOCreator &D3D12PSOCreator::Topology(D3D12_PRIMITIVE_TOPOLOGY_TYPE topo)
 {
   GraphicsDesc.PrimitiveTopologyType = topo;
diff --git a/util/test/demos/d3d12/d3d12_helpers.h b/util/test/demos/d3d12/d3d12_helpers.h
index c61f4b962..717400be9 100644
--- a/util/test/demos/d3d12/d3d12_helpers.h
+++ b/util/test/demos/d3d12/d3d12_helpers.h
@@ -87,6 +87,8 @@ public:
   D3D12PSOCreator &InputLayout(const std::vector<D3D12_INPUT_ELEMENT_DESC> &elements);
   D3D12PSOCreator &InputLayout();
 
+  D3D12PSOCreator &StripRestart(D3D12_INDEX_BUFFER_STRIP_CUT_VALUE stripCut);
+
   D3D12PSOCreator &Topology(D3D12_PRIMITIVE_TOPOLOGY_TYPE topo);
 
   D3D12PSOCreator &RootSig(ID3D12RootSignaturePtr rootSig);