Core/DescriptorHeap.cpp

   1 //
   2 // Copyright (c) Microsoft. All rights reserved.
   3 // This code is licensed under the MIT License (MIT).
   4 // THIS CODE IS PROVIDED *AS IS* WITHOUT WARRANTY OF
   5 // ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING ANY
   6 // IMPLIED WARRANTIES OF FITNESS FOR A PARTICULAR
   7 // PURPOSE, MERCHANTABILITY, OR NON-INFRINGEMENT.
   8 //
   9 // Developed by Minigraph
  10 //
  11 // Author:  James Stanard
  12 //
  13
  14 #include "pch.h"
  15 #include "DescriptorHeap.h"
  16 #include "GraphicsCore.h"
  17 #include "CommandListManager.h"
  18
  19 using namespace Graphics;
  20
  21 //
  22 // DescriptorAllocator implementation
  23 //
  24 std::mutex DescriptorAllocator::sm_AllocationMutex;
  25 std::vector<Microsoft::WRL::ComPtr<ID3D12DescriptorHeap>> DescriptorAllocator::sm_DescriptorHeapPool;
  26
  27 void DescriptorAllocator::DestroyAll(void)
  28 {
  29     sm_DescriptorHeapPool.clear();
  30 }
  31
  32 ID3D12DescriptorHeap* DescriptorAllocator::RequestNewHeap(D3D12_DESCRIPTOR_HEAP_TYPE Type)
  33 {
  34     std::lock_guard<std::mutex> LockGuard(sm_AllocationMutex);
  35
  36     D3D12_DESCRIPTOR_HEAP_DESC Desc;
  37     Desc.Type = Type;
  38     Desc.NumDescriptors = sm_NumDescriptorsPerHeap;
  39     Desc.Flags = D3D12_DESCRIPTOR_HEAP_FLAG_NONE;
  40     Desc.NodeMask = 1;
  41
  42     Microsoft::WRL::ComPtr<ID3D12DescriptorHeap> pHeap;
  43     ASSERT_SUCCEEDED(Graphics::g_Device->CreateDescriptorHeap(&Desc, MY_IID_PPV_ARGS(&pHeap)));
  44     sm_DescriptorHeapPool.emplace_back(pHeap);
  45     return pHeap.Get();
  46 }
  47
  48 D3D12_CPU_DESCRIPTOR_HANDLE DescriptorAllocator::Allocate( uint32_t Count )
  49 {
  50     if (m_CurrentHeap == nullptr || m_RemainingFreeHandles < Count)
  51     {
  52         m_CurrentHeap = RequestNewHeap(m_Type);
  53         m_CurrentHandle = m_CurrentHeap->GetCPUDescriptorHandleForHeapStart();
  54         m_RemainingFreeHandles = sm_NumDescriptorsPerHeap;
  55
  56         if (m_DescriptorSize == 0)
  57             m_DescriptorSize = Graphics::g_Device->GetDescriptorHandleIncrementSize(m_Type);
  58     }
  59
  60     D3D12_CPU_DESCRIPTOR_HANDLE ret = m_CurrentHandle;
  61     m_CurrentHandle.ptr += Count * m_DescriptorSize;
  62     m_RemainingFreeHandles -= Count;
  63     return ret;
  64 }
  65
  66 //
  67 // UserDescriptorHeap implementation
  68 //
  69
  70 void UserDescriptorHeap::Create( const std::wstring& DebugHeapName )
  71 {
  72     ASSERT_SUCCEEDED(Graphics::g_Device->CreateDescriptorHeap(&m_HeapDesc, MY_IID_PPV_ARGS(m_Heap.ReleaseAndGetAddressOf())));
  73 #ifdef RELEASE
  74     (void)DebugHeapName;
  75 #else
  76     m_Heap->SetName(DebugHeapName.c_str());
  77 #endif
  78
  79     m_DescriptorSize = Graphics::g_Device->GetDescriptorHandleIncrementSize(m_HeapDesc.Type);
  80     m_NumFreeDescriptors = m_HeapDesc.NumDescriptors;
  81     m_FirstHandle = DescriptorHandle( m_Heap->GetCPUDescriptorHandleForHeapStart(),  m_Heap->GetGPUDescriptorHandleForHeapStart() );
  82     m_NextFreeHandle = m_FirstHandle;
  83 }
  84
  85 DescriptorHandle UserDescriptorHeap::Alloc( uint32_t Count )
  86 {
  87     ASSERT(HasAvailableSpace(Count), "Descriptor Heap out of space.  Increase heap size.");
  88     DescriptorHandle ret = m_NextFreeHandle;
  89     m_NextFreeHandle += Count * m_DescriptorSize;
  90     return ret;
  91 }
  92
  93 bool UserDescriptorHeap::ValidateHandle( const DescriptorHandle& DHandle ) const
  94 {
  95     if (DHandle.GetCpuHandle().ptr < m_FirstHandle.GetCpuHandle().ptr ||
  96         DHandle.GetCpuHandle().ptr >= m_FirstHandle.GetCpuHandle().ptr + m_HeapDesc.NumDescriptors * m_DescriptorSize)
  97         return false;
  98
  99     if (DHandle.GetGpuHandle().ptr - m_FirstHandle.GetGpuHandle().ptr !=
 100         DHandle.GetCpuHandle().ptr - m_FirstHandle.GetCpuHandle().ptr)
 101         return false;
 102
 103     return true;
 104 }