You're breathtaking!

Source/Engine/GraphicsDevice/Vulkan/ResourceOwnerVulkan.h

@@ -0,0 +1,67 @@
+// Copyright (c) 2012-2020 Wojciech Figat. All rights reserved.
+
+#pragma once
+
+#include "Engine/Graphics/GPUResourceState.h"
+#include "IncludeVulkanHeaders.h"
+
+#if GRAPHICS_API_VULKAN
+
+class GPUResource;
+class GPUContextVulkan;
+
+/// <summary>
+/// Vulkan resource state used to indicate invalid state (useful for debugging resource tracking issues).
+/// </summary>
+#define VK_IMAGE_LAYOUT_CORRUPT VK_IMAGE_LAYOUT_MAX_ENUM
+
+/// <summary>
+/// Tracking of per-resource or per-subresource state for Vulkan resources that require to issue resource access barriers during rendering.
+/// </summary>
+class ResourceStateVulkan : public GPUResourceState<VkImageLayout, VK_IMAGE_LAYOUT_CORRUPT>
+{
+};
+
+/// <summary>
+/// Base class for objects in Vulkan backend that can own a resource.
+/// </summary>
+class ResourceOwnerVulkan
+{
+    friend GPUContextVulkan;
+
+public:
+
+    virtual ~ResourceOwnerVulkan()
+    {
+    }
+
+public:
+
+    /// <summary>
+    /// The resource state tracking helper. Used for resource barriers.
+    /// </summary>
+    ResourceStateVulkan State;
+
+    /// <summary>
+    /// The array size (for textures).
+    /// </summary>
+    int32 ArraySlices;
+
+public:
+
+    /// <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:
+
+    void initResource(VkImageLayout initialState, int32 mipLevels = 1, int32 arraySize = 1, bool usePerSubresourceTracking = false)
+    {
+        State.Initialize(mipLevels * arraySize, initialState, usePerSubresourceTracking);
+        ArraySlices = arraySize;
+    }
+};
+
+#endif