Common: Add helper classes for D3D11

src/common/d3d11/stream_buffer.cpp

@@ -0,0 +1,88 @@
+#include "stream_buffer.h"
+#include "YBaseLib/Log.h"
+Log_SetChannel(D3D11);
+
+namespace D3D11 {
+
+StreamBuffer::StreamBuffer() : m_size(0), m_position(0) {}
+
+StreamBuffer::StreamBuffer(ComPtr<ID3D11Buffer> buffer) : m_buffer(std::move(buffer)), m_position(0)
+{
+  D3D11_BUFFER_DESC desc;
+  m_buffer->GetDesc(&desc);
+  m_size = desc.ByteWidth;
+}
+
+StreamBuffer::~StreamBuffer()
+{
+  Release();
+}
+
+bool StreamBuffer::Create(ID3D11Device* device, D3D11_BIND_FLAG bind_flags, u32 size)
+{
+  CD3D11_BUFFER_DESC desc(size, bind_flags, D3D11_USAGE_DYNAMIC, D3D11_CPU_ACCESS_WRITE, 0, 0);
+  ComPtr<ID3D11Buffer> buffer;
+  const HRESULT hr = device->CreateBuffer(&desc, nullptr, &buffer);
+  if (FAILED(hr))
+  {
+    Log_ErrorPrintf("Creating buffer failed: 0x%08X", hr);
+    return false;
+  }
+
+  m_buffer = std::move(buffer);
+  m_size = size;
+  m_position = 0;
+  return true;
+}
+
+void StreamBuffer::Adopt(ComPtr<ID3D11Buffer> buffer)
+{
+  m_buffer = std::move(buffer);
+
+  D3D11_BUFFER_DESC desc;
+  m_buffer->GetDesc(&desc);
+  m_size = desc.ByteWidth;
+  m_position = 0;
+}
+
+void StreamBuffer::Release()
+{
+  m_buffer.Reset();
+}
+
+StreamBuffer::MappingResult StreamBuffer::Map(ID3D11DeviceContext* context, u32 alignment, u32 min_size)
+{
+  m_position = Common::AlignUp(m_position, alignment);
+
+  D3D11_MAP map;
+  if ((m_position + min_size) >= m_size)
+  {
+    // wrap around
+    m_position = 0;
+    map = D3D11_MAP_WRITE_DISCARD;
+  }
+  else
+  {
+    map = D3D11_MAP_WRITE_NO_OVERWRITE;
+  }
+
+  D3D11_MAPPED_SUBRESOURCE sr;
+  const HRESULT hr = context->Map(m_buffer.Get(), 0, map, 0, &sr);
+  if (FAILED(hr))
+  {
+    Log_ErrorPrintf("Map failed: 0x%08X", hr);
+    Panic("Map failed");
+    return {};
+  }
+
+  return MappingResult{static_cast<char*>(sr.pData) + m_position, m_position, m_position / alignment,
+                       (m_size - m_position) / alignment};
+}
+
+void StreamBuffer::Unmap(ID3D11DeviceContext* context, u32 used_size)
+{
+  context->Unmap(m_buffer.Get(), 0);
+  m_position += used_size;
+}
+
+} // namespace D3D11