1 files changed, 48 insertions, 0 deletions
diff --git a/src/Ryujinx.Graphics.Vulkan/BufferState.cs b/src/Ryujinx.Graphics.Vulkan/BufferState.cs
new file mode 100644
index 00000000..6829f833
--- /dev/null
+++ b/src/Ryujinx.Graphics.Vulkan/BufferState.cs
@@ -0,0 +1,48 @@
+using System;
+
+namespace Ryujinx.Graphics.Vulkan
+{
+    struct BufferState : IDisposable
+    {
+        public static BufferState Null => new BufferState(null, 0, 0);
+
+        private readonly int _offset;
+        private readonly int _size;
+
+        private Auto<DisposableBuffer> _buffer;
+
+        public BufferState(Auto<DisposableBuffer> buffer, int offset, int size)
+        {
+            _buffer = buffer;
+            _offset = offset;
+            _size = size;
+            buffer?.IncrementReferenceCount();
+        }
+
+        public void BindTransformFeedbackBuffer(VulkanRenderer gd, CommandBufferScoped cbs, uint binding)
+        {
+            if (_buffer != null)
+            {
+                var buffer = _buffer.Get(cbs, _offset, _size).Value;
+
+                gd.TransformFeedbackApi.CmdBindTransformFeedbackBuffers(cbs.CommandBuffer, binding, 1, buffer, (ulong)_offset, (ulong)_size);
+            }
+        }
+
+        public void Swap(Auto<DisposableBuffer> from, Auto<DisposableBuffer> to)
+        {
+            if (_buffer == from)
+            {
+                _buffer.DecrementReferenceCount();
+                to.IncrementReferenceCount();
+
+                _buffer = to;
+            }
+        }
+
+        public void Dispose()
+        {
+            _buffer?.DecrementReferenceCount();
+        }
+    }
+}