You're breathtaking!

Source/Engine/GraphicsDevice/DirectX/DX12/ResourceOwnerDX12.h

@@ -0,0 +1,120 @@
+// Copyright (c) 2012-2020 Wojciech Figat. All rights reserved.
+
+#pragma once
+
+#include "Engine/Core/Delegate.h"
+#include "Engine/Graphics/GPUResourceState.h"
+#include "../IncludeDirectXHeaders.h"
+
+#if GRAPHICS_API_DIRECTX12
+
+class GPUResource;
+class GPUContextDX12;
+class GPUAsyncContextDX12;
+
+/// <summary>
+/// Default amount of frames to wait until resource delete.
+/// </summary>
+#define DX12_RESOURCE_DELETE_SAFE_FRAMES_COUNT 10
+
+/// <summary>
+/// Custom resource state used to indicate invalid state (useful for debugging resource tracking issues).
+/// </summary>
+#define D3D12_RESOURCE_STATE_CORRUPT (D3D12_RESOURCE_STATES)-1
+
+/// <summary>
+/// Tracking of per-resource or per-subresource state for D3D12 resources that require to issue resource access barriers during rendering.
+/// </summary>
+class ResourceStateDX12 : public GPUResourceState<D3D12_RESOURCE_STATES, D3D12_RESOURCE_STATE_CORRUPT>
+{
+public:
+
+    /// <summary>
+    /// Returns true if resource state transition is needed in order to use resource in given state.
+    /// </summary>
+    /// <param name="currentState">The current resource state.</param>
+    /// <param name="targeState">the destination resource state.</param>
+    /// <returns>True if need to perform a transition, otherwise false.</returns>
+    static bool IsTransitionNeeded(D3D12_RESOURCE_STATES currentState, D3D12_RESOURCE_STATES targeState)
+    {
+        // If 'targeState' is a subset of 'before', then there's no need for a transition
+        // Note: COMMON is an oddball state that doesn't follow the RESOURE_STATE pattern of 
+        // having exactly one bit set so we need to special case these
+        return currentState != targeState && ((currentState | targeState) != currentState || targeState == D3D12_RESOURCE_STATE_COMMON);
+    }
+};
+
+/// <summary>
+/// Base class for objects in DirectX 12 layer that can own a resource
+/// </summary>
+class ResourceOwnerDX12
+{
+    friend GPUContextDX12;
+    friend GPUAsyncContextDX12;
+
+protected:
+
+    ID3D12Resource* _resource;
+    uint32 _subresourcesCount;
+
+    ResourceOwnerDX12()
+        : _resource(nullptr)
+        , _subresourcesCount(0)
+    {
+    }
+
+    ~ResourceOwnerDX12()
+    {
+    }
+
+public:
+
+    /// <summary>
+    /// Action called on resource release event.
+    /// </summary>
+    Delegate<ResourceOwnerDX12*> OnRelease;
+
+    /// <summary>
+    /// The resource state tracking helper. Used for resource barriers.
+    /// </summary>
+    ResourceStateDX12 State;
+
+public:
+
+    /// <summary>
+    /// Gets the subresources count.
+    /// </summary>
+    /// <returns>The subresources count.</returns>
+    FORCE_INLINE uint32 GetSubresourcesCount() const
+    {
+        return _subresourcesCount;
+    }
+
+    /// <summary>
+    /// Gets DirectX 12 resource object handle
+    /// </summary>
+    /// <returns>DirectX 12 resource object handle</returns>
+    FORCE_INLINE ID3D12Resource* GetResource() const
+    {
+        return _resource;
+    }
+
+    /// <summary>
+    /// Gets resource owner object as a GPUResource type or returns null if cannot perform cast.
+    /// </summary>
+    /// <returns>GPU Resource or null if cannot cast.</returns>
+    virtual GPUResource* AsGPUResource() const = 0;
+
+protected:
+
+    FORCE_INLINE void initResource(ID3D12Resource* resource, const D3D12_RESOURCE_STATES initialState, const D3D12_RESOURCE_DESC& desc, bool usePerSubresourceTracking = false)
+    {
+        initResource(resource, initialState, desc.DepthOrArraySize * desc.MipLevels, usePerSubresourceTracking);
+    }
+
+    void initResource(ID3D12Resource* resource, const D3D12_RESOURCE_STATES initialState, const uint32 subresourceCount, bool usePerSubresourceTracking = false);
+    void initResource(const D3D12_RESOURCE_STATES initialState, const uint32 subresourceCount, bool usePerSubresourceTracking = false);
+    void releaseResource(uint32 safeFrameCount = DX12_RESOURCE_DELETE_SAFE_FRAMES_COUNT);
+};
+
+#endif