[d3d11] Save a few CPU cycles in Map/MapBuffer

src/d3d11/d3d11_buffer.h

@@ -80,17 +80,18 @@ namespace dxvk {
     }
     
     VkDeviceSize GetSize() const {
-      return m_buffer->info().size;
+      return m_desc.ByteWidth;
+    }
+
+    DxvkPhysicalBufferSlice DiscardSlice() {
+      m_mapped = m_buffer->allocPhysicalSlice();
+      return m_mapped;
     }
 
     DxvkPhysicalBufferSlice GetMappedSlice() const {
       return m_mapped;
     }
 
-    void SetMappedSlice(const DxvkPhysicalBufferSlice& slice) {
-      m_mapped = slice;
-    }
-
     D3D10Buffer* GetD3D10Iface() {
       return &m_d3d10;
     }

src/d3d11/d3d11_context_def.cpp

@@ -159,9 +159,6 @@ namespace dxvk {
     D3D11Buffer* pBuffer = static_cast<D3D11Buffer*>(pResource);
     const Rc<DxvkBuffer> buffer = pBuffer->GetBuffer();
     
-    D3D11_BUFFER_DESC bufferDesc;
-    pBuffer->GetDesc(&bufferDesc);
-    
     if (!(buffer->memFlags() & VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT)) {
       Logger::err("D3D11: Cannot map a device-local buffer");
       return E_INVALIDARG;
@@ -173,7 +170,7 @@ namespace dxvk {
     pMapEntry->RowPitch     = pBuffer->GetSize();
     pMapEntry->DepthPitch   = pBuffer->GetSize();
     
-    if (bufferDesc.Usage == D3D11_USAGE_DYNAMIC && m_parent->GetOptions()->dcMapSpeedHack) {
+    if (pBuffer->Desc()->Usage == D3D11_USAGE_DYNAMIC && m_parent->GetOptions()->dcMapSpeedHack) {
       // For resources that cannot be written by the GPU,
       // we may write to the buffer resource directly and
       // just swap in the physical buffer slice as needed.
@@ -240,10 +237,7 @@ namespace dxvk {
     const D3D11DeferredContextMapEntry* pMapEntry) {
     D3D11Buffer* pBuffer = static_cast<D3D11Buffer*>(pResource);
     
-    D3D11_BUFFER_DESC bufferDesc;
+    if (pBuffer->Desc()->Usage == D3D11_USAGE_DYNAMIC && m_parent->GetOptions()->dcMapSpeedHack) {
-    pBuffer->GetDesc(&bufferDesc);
-    
-    if (bufferDesc.Usage == D3D11_USAGE_DYNAMIC && m_parent->GetOptions()->dcMapSpeedHack) {
       EmitCs([
         cDstBuffer = pBuffer->GetBuffer(),
         cPhysSlice = pMapEntry->BufferSlice

src/d3d11/d3d11_context_imm.cpp

@@ -157,28 +157,32 @@ namespace dxvk {
           D3D11_MAP                   MapType,
           UINT                        MapFlags,
           D3D11_MAPPED_SUBRESOURCE*   pMappedResource) {
-    if (pResource == nullptr)
+    if (!pResource || !pMappedResource)
-      return DXGI_ERROR_INVALID_CALL;
+      return E_INVALIDARG;
     
-    if (pMappedResource != nullptr) {
+    D3D11_RESOURCE_DIMENSION resourceDim = D3D11_RESOURCE_DIMENSION_UNKNOWN;
+    pResource->GetType(&resourceDim);
+
+    HRESULT hr;
+    
+    if (resourceDim == D3D11_RESOURCE_DIMENSION_BUFFER) {
+      hr = MapBuffer(
+        static_cast<D3D11Buffer*>(pResource),
+        MapType, MapFlags, pMappedResource);
+    } else {
+      hr = MapImage(
+        GetCommonTexture(pResource),
+        Subresource, MapType, MapFlags,
+        pMappedResource);
+    }
+
+    if (FAILED(hr)) {
       pMappedResource->pData      = nullptr;
       pMappedResource->RowPitch   = 0;
       pMappedResource->DepthPitch = 0;
     }
 
-    D3D11_RESOURCE_DIMENSION resourceDim = D3D11_RESOURCE_DIMENSION_UNKNOWN;
+    return hr;
-    pResource->GetType(&resourceDim);
-    
-    if (resourceDim == D3D11_RESOURCE_DIMENSION_BUFFER) {
-      return MapBuffer(
-        static_cast<D3D11Buffer*>(pResource),
-        MapType, MapFlags, pMappedResource);
-    } else {
-      return MapImage(
-        GetCommonTexture(pResource),
-        Subresource, MapType, MapFlags,
-        pMappedResource);
-    }
   }
   
   
@@ -338,16 +342,22 @@ namespace dxvk {
       // Allocate a new backing slice for the buffer and set
       // it as the 'new' mapped slice. This assumes that the
       // only way to invalidate a buffer is by mapping it.
-      auto physicalSlice = buffer->allocPhysicalSlice();
+      auto physicalSlice = pResource->DiscardSlice();
-      pResource->SetMappedSlice(physicalSlice);
+      pMappedResource->pData      = physicalSlice.mapPtr(0);
+      pMappedResource->RowPitch   = pResource->GetSize();
+      pMappedResource->DepthPitch = pResource->GetSize();
       
       EmitCs([
-        cBuffer        = buffer,
+        cBuffer        = std::move(buffer),
-        cPhysicalSlice = physicalSlice
+        cPhysicalSlice = std::move(physicalSlice)
       ] (DxvkContext* ctx) {
         ctx->invalidateBuffer(cBuffer, cPhysicalSlice);
       });
-    } else if (MapType != D3D11_MAP_WRITE_NO_OVERWRITE) {
+
+      return S_OK;
+    } else {
+      // Wait until the resource is no longer in use
+      if (MapType != D3D11_MAP_WRITE_NO_OVERWRITE) {
         if (!WaitForResource(buffer->resource(), MapFlags))
           return DXGI_ERROR_WAS_STILL_DRAWING;
       }
@@ -362,6 +372,7 @@ namespace dxvk {
       pMappedResource->DepthPitch = pResource->GetSize();
       return S_OK;
     }
+  }
   
   
   HRESULT D3D11ImmediateContext::MapImage(