| /* Copyright (c) 2015-2016 The Khronos Group Inc. |
| * Copyright (c) 2015-2016 Valve Corporation |
| * Copyright (c) 2015-2016 LunarG, Inc. |
| * Copyright (C) 2015-2016 Google Inc. |
| * |
| * Permission is hereby granted, free of charge, to any person obtaining a copy |
| * of this software and/or associated documentation files (the "Materials"), to |
| * deal in the Materials without restriction, including without limitation the |
| * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or |
| * sell copies of the Materials, and to permit persons to whom the Materials |
| * are furnished to do so, subject to the following conditions: |
| * |
| * The above copyright notice(s) and this permission notice shall be included |
| * in all copies or substantial portions of the Materials. |
| * |
| * THE MATERIALS ARE PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. |
| * |
| * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, |
| * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR |
| * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE MATERIALS OR THE |
| * USE OR OTHER DEALINGS IN THE MATERIALS |
| * |
| * Author: Courtney Goeltzenleuchter <courtneygo@google.com> |
| * Author: Tobin Ehlis <tobine@google.com> |
| * Author: Chris Forbes <chrisf@ijw.co.nz> |
| * Author: Mark Lobodzinski <mark@lunarg.com> |
| */ |
| |
| // Enable mem_tracker merged code |
| #define MTMERGE 1 |
| |
| #pragma once |
| #include "vulkan/vk_layer.h" |
| #include <atomic> |
| #include <vector> |
| #include <unordered_map> |
| #include <unordered_set> |
| #include <memory> |
| #include <functional> |
| |
| using std::vector; |
| using std::unordered_set; |
| |
| #if MTMERGE |
| // Mem Tracker ERROR codes |
| typedef enum _MEM_TRACK_ERROR { |
| MEMTRACK_NONE, // Used for INFO & other non-error messages |
| MEMTRACK_INVALID_CB, // Cmd Buffer invalid |
| MEMTRACK_INVALID_MEM_OBJ, // Invalid Memory Object |
| MEMTRACK_INVALID_ALIASING, // Invalid Memory Aliasing |
| MEMTRACK_INVALID_LAYOUT, // Invalid Layout |
| MEMTRACK_INTERNAL_ERROR, // Bug in Mem Track Layer internal data structures |
| MEMTRACK_FREED_MEM_REF, // MEM Obj freed while it still has obj and/or CB refs |
| MEMTRACK_MEM_OBJ_CLEAR_EMPTY_BINDINGS, // Clearing bindings on mem obj that doesn't have any bindings |
| MEMTRACK_MISSING_MEM_BINDINGS, // Trying to retrieve mem bindings, but none found (may be internal error) |
| MEMTRACK_INVALID_OBJECT, // Attempting to reference generic VK Object that is invalid |
| MEMTRACK_MEMORY_BINDING_ERROR, // Error during one of many calls that bind memory to object or CB |
| MEMTRACK_MEMORY_LEAK, // Failure to call vkFreeMemory on Mem Obj prior to DestroyDevice |
| MEMTRACK_INVALID_STATE, // Memory not in the correct state |
| MEMTRACK_RESET_CB_WHILE_IN_FLIGHT, // vkResetCommandBuffer() called on a CB that hasn't completed |
| MEMTRACK_INVALID_FENCE_STATE, // Invalid Fence State signaled or used |
| MEMTRACK_REBIND_OBJECT, // Non-sparse object bindings are immutable |
| MEMTRACK_INVALID_USAGE_FLAG, // Usage flags specified at image/buffer create conflict w/ use of object |
| MEMTRACK_INVALID_MAP, // Size flag specified at alloc is too small for mapping range |
| } MEM_TRACK_ERROR; |
| |
| // MemTracker Semaphore states |
| typedef enum SemaphoreState { |
| MEMTRACK_SEMAPHORE_STATE_UNSET, // Semaphore is in an undefined state |
| MEMTRACK_SEMAPHORE_STATE_SIGNALLED, // Semaphore has is in signalled state |
| MEMTRACK_SEMAPHORE_STATE_WAIT, // Semaphore is in wait state |
| } SemaphoreState; |
| |
| struct MemRange { |
| VkDeviceSize offset; |
| VkDeviceSize size; |
| }; |
| |
| /* |
| * MTMTODO : Update this comment |
| * Data Structure overview |
| * There are 4 global STL(' maps |
| * cbMap -- map of command Buffer (CB) objects to MT_CB_INFO structures |
| * Each MT_CB_INFO struct has an stl list container with |
| * memory objects that are referenced by this CB |
| * memObjMap -- map of Memory Objects to MT_MEM_OBJ_INFO structures |
| * Each MT_MEM_OBJ_INFO has two stl list containers with: |
| * -- all CBs referencing this mem obj |
| * -- all VK Objects that are bound to this memory |
| * objectMap -- map of objects to MT_OBJ_INFO structures |
| * |
| * Algorithm overview |
| * These are the primary events that should happen related to different objects |
| * 1. Command buffers |
| * CREATION - Add object,structure to map |
| * CMD BIND - If mem associated, add mem reference to list container |
| * DESTROY - Remove from map, decrement (and report) mem references |
| * 2. Mem Objects |
| * CREATION - Add object,structure to map |
| * OBJ BIND - Add obj structure to list container for that mem node |
| * CMB BIND - If mem-related add CB structure to list container for that mem node |
| * DESTROY - Flag as errors any remaining refs and remove from map |
| * 3. Generic Objects |
| * MEM BIND - DESTROY any previous binding, Add obj node w/ ref to map, add obj ref to list container for that mem node |
| * DESTROY - If mem bound, remove reference list container for that memInfo, remove object ref from map |
| */ |
| // TODO : Is there a way to track when Cmd Buffer finishes & remove mem references at that point? |
| // TODO : Could potentially store a list of freed mem allocs to flag when they're incorrectly used |
| |
| // Simple struct to hold handle and type of object so they can be uniquely identified and looked up in appropriate map |
| struct MT_OBJ_HANDLE_TYPE { |
| uint64_t handle; |
| VkDebugReportObjectTypeEXT type; |
| }; |
| |
| bool operator==(MT_OBJ_HANDLE_TYPE a, MT_OBJ_HANDLE_TYPE b) noexcept { |
| return a.handle == b.handle && a.type == b.type; |
| } |
| |
| namespace std { |
| template<> |
| struct hash<MT_OBJ_HANDLE_TYPE> { |
| size_t operator()(MT_OBJ_HANDLE_TYPE obj) const noexcept { |
| return hash<uint64_t>()(obj.handle) ^ |
| hash<uint32_t>()(obj.type); |
| } |
| }; |
| } |
| |
| struct MEMORY_RANGE { |
| uint64_t handle; |
| VkDeviceMemory memory; |
| VkDeviceSize start; |
| VkDeviceSize end; |
| }; |
| |
| // Data struct for tracking memory object |
| struct DEVICE_MEM_INFO { |
| void *object; // Dispatchable object used to create this memory (device of swapchain) |
| bool valid; // Stores if the memory has valid data or not |
| VkDeviceMemory mem; |
| VkMemoryAllocateInfo allocInfo; |
| unordered_set<MT_OBJ_HANDLE_TYPE> objBindings; // objects bound to this memory |
| unordered_set<VkCommandBuffer> commandBufferBindings; // cmd buffers referencing this memory |
| vector<MEMORY_RANGE> bufferRanges; |
| vector<MEMORY_RANGE> imageRanges; |
| VkImage image; // If memory is bound to image, this will have VkImage handle, else VK_NULL_HANDLE |
| MemRange memRange; |
| void *pData, *pDriverData; |
| }; |
| |
| // This only applies to Buffers and Images, which can have memory bound to them |
| struct MT_OBJ_BINDING_INFO { |
| VkDeviceMemory mem; |
| bool valid; // If this is a swapchain image backing memory is not a MT_MEM_OBJ_INFO so store it here. |
| union create_info { |
| VkImageCreateInfo image; |
| VkBufferCreateInfo buffer; |
| } create_info; |
| }; |
| |
| struct MT_FB_ATTACHMENT_INFO { |
| VkImage image; |
| VkDeviceMemory mem; |
| }; |
| |
| struct MT_PASS_ATTACHMENT_INFO { |
| uint32_t attachment; |
| VkAttachmentLoadOp load_op; |
| VkAttachmentStoreOp store_op; |
| }; |
| |
| // Associate fenceId with a fence object |
| struct MT_FENCE_INFO { |
| uint64_t fenceId; // Sequence number for fence at last submit |
| VkQueue queue; // Queue that this fence is submitted against or NULL |
| VkSwapchainKHR swapchain; // Swapchain that this fence is submitted against or NULL |
| VkBool32 firstTimeFlag; // Fence was created in signaled state, avoid warnings for first use |
| VkFenceCreateInfo createInfo; |
| }; |
| |
| // Track Queue information |
| struct MT_QUEUE_INFO { |
| uint64_t lastRetiredId; |
| uint64_t lastSubmittedId; |
| list<VkCommandBuffer> pQueueCommandBuffers; |
| list<VkDeviceMemory> pMemRefList; |
| }; |
| |
| struct MT_DESCRIPTOR_SET_INFO { |
| std::vector<VkImageView> images; |
| std::vector<VkBuffer> buffers; |
| }; |
| |
| // Track Swapchain Information |
| struct MT_SWAP_CHAIN_INFO { |
| VkSwapchainCreateInfoKHR createInfo; |
| std::vector<VkImage> images; |
| }; |
| |
| #endif |
| // Draw State ERROR codes |
| typedef enum _DRAW_STATE_ERROR { |
| // TODO: Remove the comments here or expand them. There isn't any additional information in the |
| // comments than in the name in almost all cases. |
| DRAWSTATE_NONE, // Used for INFO & other non-error messages |
| DRAWSTATE_INTERNAL_ERROR, // Error with DrawState internal data structures |
| DRAWSTATE_NO_PIPELINE_BOUND, // Unable to identify a bound pipeline |
| DRAWSTATE_INVALID_POOL, // Invalid DS pool |
| DRAWSTATE_INVALID_SET, // Invalid DS |
| DRAWSTATE_INVALID_RENDER_AREA, // Invalid renderArea |
| DRAWSTATE_INVALID_LAYOUT, // Invalid DS layout |
| DRAWSTATE_INVALID_IMAGE_LAYOUT, // Invalid Image layout |
| DRAWSTATE_INVALID_PIPELINE, // Invalid Pipeline handle referenced |
| DRAWSTATE_INVALID_PIPELINE_LAYOUT, // Invalid PipelineLayout |
| DRAWSTATE_INVALID_PIPELINE_CREATE_STATE, // Attempt to create a pipeline |
| // with invalid state |
| DRAWSTATE_INVALID_COMMAND_BUFFER, // Invalid CommandBuffer referenced |
| DRAWSTATE_INVALID_BARRIER, // Invalid Barrier |
| DRAWSTATE_INVALID_BUFFER, // Invalid Buffer |
| DRAWSTATE_INVALID_QUERY, // Invalid Query |
| DRAWSTATE_INVALID_FENCE, // Invalid Fence |
| DRAWSTATE_INVALID_SEMAPHORE, // Invalid Semaphore |
| DRAWSTATE_INVALID_EVENT, // Invalid Event |
| DRAWSTATE_VTX_INDEX_OUT_OF_BOUNDS, // binding in vkCmdBindVertexData() too |
| // large for PSO's |
| // pVertexBindingDescriptions array |
| DRAWSTATE_VTX_INDEX_ALIGNMENT_ERROR, // binding offset in |
| // vkCmdBindIndexBuffer() out of |
| // alignment based on indexType |
| // DRAWSTATE_MISSING_DOT_PROGRAM, // No "dot" program in order |
| // to generate png image |
| DRAWSTATE_OUT_OF_MEMORY, // malloc failed |
| DRAWSTATE_INVALID_DESCRIPTOR_SET, // Descriptor Set handle is unknown |
| DRAWSTATE_DESCRIPTOR_TYPE_MISMATCH, // Type in layout vs. update are not the |
| // same |
| DRAWSTATE_DESCRIPTOR_STAGEFLAGS_MISMATCH, // StageFlags in layout are not |
| // the same throughout a single |
| // VkWriteDescriptorSet update |
| DRAWSTATE_DESCRIPTOR_UPDATE_OUT_OF_BOUNDS, // Descriptors set for update out |
| // of bounds for corresponding |
| // layout section |
| DRAWSTATE_DESCRIPTOR_POOL_EMPTY, // Attempt to allocate descriptor from a |
| // pool with no more descriptors of that |
| // type available |
| DRAWSTATE_CANT_FREE_FROM_NON_FREE_POOL, // Invalid to call |
| // vkFreeDescriptorSets on Sets |
| // allocated from a NON_FREE Pool |
| DRAWSTATE_INVALID_UPDATE_INDEX, // Index of requested update is invalid for |
| // specified descriptors set |
| DRAWSTATE_INVALID_UPDATE_STRUCT, // Struct in DS Update tree is of invalid |
| // type |
| DRAWSTATE_NUM_SAMPLES_MISMATCH, // Number of samples in bound PSO does not |
| // match number in FB of current RenderPass |
| DRAWSTATE_NO_END_COMMAND_BUFFER, // Must call vkEndCommandBuffer() before |
| // QueueSubmit on that commandBuffer |
| DRAWSTATE_NO_BEGIN_COMMAND_BUFFER, // Binding cmds or calling End on CB that |
| // never had vkBeginCommandBuffer() |
| // called on it |
| DRAWSTATE_COMMAND_BUFFER_SINGLE_SUBMIT_VIOLATION, // Cmd Buffer created with |
| // VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT |
| // flag is submitted |
| // multiple times |
| DRAWSTATE_INVALID_SECONDARY_COMMAND_BUFFER, // vkCmdExecuteCommands() called |
| // with a primary commandBuffer |
| // in pCommandBuffers array |
| DRAWSTATE_VIEWPORT_NOT_BOUND, // Draw submitted with no viewport state bound |
| DRAWSTATE_SCISSOR_NOT_BOUND, // Draw submitted with no scissor state bound |
| DRAWSTATE_LINE_WIDTH_NOT_BOUND, // Draw submitted with no line width state |
| // bound |
| DRAWSTATE_DEPTH_BIAS_NOT_BOUND, // Draw submitted with no depth bias state |
| // bound |
| DRAWSTATE_BLEND_NOT_BOUND, // Draw submitted with no blend state bound when |
| // color write enabled |
| DRAWSTATE_DEPTH_BOUNDS_NOT_BOUND, // Draw submitted with no depth bounds |
| // state bound when depth enabled |
| DRAWSTATE_STENCIL_NOT_BOUND, // Draw submitted with no stencil state bound |
| // when stencil enabled |
| DRAWSTATE_INDEX_BUFFER_NOT_BOUND, // Draw submitted with no depth-stencil |
| // state bound when depth write enabled |
| DRAWSTATE_PIPELINE_LAYOUTS_INCOMPATIBLE, // Draw submitted PSO Pipeline |
| // layout that's not compatible |
| // with layout from |
| // BindDescriptorSets |
| DRAWSTATE_RENDERPASS_INCOMPATIBLE, // Incompatible renderpasses between |
| // secondary cmdBuffer and primary |
| // cmdBuffer or framebuffer |
| DRAWSTATE_FRAMEBUFFER_INCOMPATIBLE, // Incompatible framebuffer between |
| // secondary cmdBuffer and active |
| // renderPass |
| DRAWSTATE_INVALID_RENDERPASS, // Use of a NULL or otherwise invalid |
| // RenderPass object |
| DRAWSTATE_INVALID_RENDERPASS_CMD, // Invalid cmd submitted while a |
| // RenderPass is active |
| DRAWSTATE_NO_ACTIVE_RENDERPASS, // Rendering cmd submitted without an active |
| // RenderPass |
| DRAWSTATE_DESCRIPTOR_SET_NOT_UPDATED, // DescriptorSet bound but it was |
| // never updated. This is a warning |
| // code. |
| DRAWSTATE_DESCRIPTOR_SET_NOT_BOUND, // DescriptorSet used by pipeline at |
| // draw time is not bound, or has been |
| // disturbed (which would have flagged |
| // previous warning) |
| DRAWSTATE_INVALID_DYNAMIC_OFFSET_COUNT, // DescriptorSets bound with |
| // different number of dynamic |
| // descriptors that were included in |
| // dynamicOffsetCount |
| DRAWSTATE_CLEAR_CMD_BEFORE_DRAW, // Clear cmd issued before any Draw in |
| // CommandBuffer, should use RenderPass Ops |
| // instead |
| DRAWSTATE_BEGIN_CB_INVALID_STATE, // CB state at Begin call is bad. Can be |
| // Primary/Secondary CB created with |
| // mismatched FB/RP information or CB in |
| // RECORDING state |
| DRAWSTATE_INVALID_CB_SIMULTANEOUS_USE, // CmdBuffer is being used in |
| // violation of |
| // VK_COMMAND_BUFFER_USAGE_SIMULTANEOUS_USE_BIT |
| // rules (i.e. simultaneous use w/o |
| // that bit set) |
| DRAWSTATE_INVALID_COMMAND_BUFFER_RESET, // Attempting to call Reset (or |
| // Begin on recorded cmdBuffer) that |
| // was allocated from Pool w/o |
| // VK_COMMAND_POOL_CREATE_RESET_COMMAND_BUFFER_BIT |
| // bit set |
| DRAWSTATE_VIEWPORT_SCISSOR_MISMATCH, // Count for viewports and scissors |
| // mismatch and/or state doesn't match |
| // count |
| DRAWSTATE_INVALID_IMAGE_ASPECT, // Image aspect is invalid for the current |
| // operation |
| DRAWSTATE_MISSING_ATTACHMENT_REFERENCE, // Attachment reference must be |
| // present in active subpass |
| DRAWSTATE_SAMPLER_DESCRIPTOR_ERROR, // A Descriptor of *_SAMPLER type is |
| // being updated with an invalid or bad |
| // Sampler |
| DRAWSTATE_INCONSISTENT_IMMUTABLE_SAMPLER_UPDATE, // Descriptors of |
| // *COMBINED_IMAGE_SAMPLER |
| // type are being updated |
| // where some, but not all, |
| // of the updates use |
| // immutable samplers |
| DRAWSTATE_IMAGEVIEW_DESCRIPTOR_ERROR, // A Descriptor of *_IMAGE or |
| // *_ATTACHMENT type is being updated |
| // with an invalid or bad ImageView |
| DRAWSTATE_BUFFERVIEW_DESCRIPTOR_ERROR, // A Descriptor of *_TEXEL_BUFFER |
| // type is being updated with an |
| // invalid or bad BufferView |
| DRAWSTATE_BUFFERINFO_DESCRIPTOR_ERROR, // A Descriptor of |
| // *_[UNIFORM|STORAGE]_BUFFER_[DYNAMIC] |
| // type is being updated with an |
| // invalid or bad BufferView |
| DRAWSTATE_DYNAMIC_OFFSET_OVERFLOW, // At draw time the dynamic offset |
| // combined with buffer offset and range |
| // oversteps size of buffer |
| DRAWSTATE_DOUBLE_DESTROY, // Destroying an object twice |
| DRAWSTATE_OBJECT_INUSE, // Destroying or modifying an object in use by a |
| // command buffer |
| DRAWSTATE_QUEUE_FORWARD_PROGRESS, // Queue cannot guarantee forward progress |
| DRAWSTATE_INVALID_UNIFORM_BUFFER_OFFSET, // Dynamic Uniform Buffer Offsets |
| // violate device limit |
| DRAWSTATE_INVALID_STORAGE_BUFFER_OFFSET, // Dynamic Storage Buffer Offsets |
| // violate device limit |
| DRAWSTATE_INDEPENDENT_BLEND, // If independent blending is not enabled, all |
| // elements of pAttachmentsMustBeIdentical |
| DRAWSTATE_DISABLED_LOGIC_OP, // If logic operations is not enabled, logicOpEnable |
| // must be VK_FALSE |
| DRAWSTATE_INVALID_LOGIC_OP, // If logicOpEnable is VK_TRUE, logicOp must |
| // must be a valid VkLogicOp value |
| DRAWSTATE_INVALID_QUEUE_INDEX, // Specified queue index exceeds number |
| // of queried queue families |
| DRAWSTATE_PUSH_CONSTANTS_ERROR, // Push constants exceed maxPushConstantSize |
| } DRAW_STATE_ERROR; |
| |
| typedef enum _SHADER_CHECKER_ERROR { |
| SHADER_CHECKER_NONE, |
| SHADER_CHECKER_INTERFACE_TYPE_MISMATCH, // Type mismatch between shader stages or shader and pipeline |
| SHADER_CHECKER_OUTPUT_NOT_CONSUMED, // Entry appears in output interface, but missing in input |
| SHADER_CHECKER_INPUT_NOT_PRODUCED, // Entry appears in input interface, but missing in output |
| SHADER_CHECKER_NON_SPIRV_SHADER, // Shader image is not SPIR-V |
| SHADER_CHECKER_INCONSISTENT_SPIRV, // General inconsistency within a SPIR-V module |
| SHADER_CHECKER_UNKNOWN_STAGE, // Stage is not supported by analysis |
| SHADER_CHECKER_INCONSISTENT_VI, // VI state contains conflicting binding or attrib descriptions |
| SHADER_CHECKER_MISSING_DESCRIPTOR, // Shader attempts to use a descriptor binding not declared in the layout |
| SHADER_CHECKER_BAD_SPECIALIZATION, // Specialization map entry points outside specialization data block |
| SHADER_CHECKER_MISSING_ENTRYPOINT, // Shader module does not contain the requested entrypoint |
| SHADER_CHECKER_PUSH_CONSTANT_OUT_OF_RANGE, // Push constant variable is not in a push constant range |
| SHADER_CHECKER_PUSH_CONSTANT_NOT_ACCESSIBLE_FROM_STAGE, // Push constant range exists, but not accessible from stage |
| SHADER_CHECKER_DESCRIPTOR_TYPE_MISMATCH, // Descriptor type does not match shader resource type |
| SHADER_CHECKER_DESCRIPTOR_NOT_ACCESSIBLE_FROM_STAGE, // Descriptor used by shader, but not accessible from stage |
| SHADER_CHECKER_FEATURE_NOT_ENABLED, // Shader uses capability requiring a feature not enabled on device |
| SHADER_CHECKER_BAD_CAPABILITY, // Shader uses capability not supported by Vulkan (OpenCL features) |
| } SHADER_CHECKER_ERROR; |
| |
| typedef enum _DRAW_TYPE { |
| DRAW = 0, |
| DRAW_INDEXED = 1, |
| DRAW_INDIRECT = 2, |
| DRAW_INDEXED_INDIRECT = 3, |
| DRAW_BEGIN_RANGE = DRAW, |
| DRAW_END_RANGE = DRAW_INDEXED_INDIRECT, |
| NUM_DRAW_TYPES = (DRAW_END_RANGE - DRAW_BEGIN_RANGE + 1), |
| } DRAW_TYPE; |
| |
| typedef struct _SHADER_DS_MAPPING { |
| uint32_t slotCount; |
| VkDescriptorSetLayoutCreateInfo *pShaderMappingSlot; |
| } SHADER_DS_MAPPING; |
| |
| typedef struct _GENERIC_HEADER { |
| VkStructureType sType; |
| const void *pNext; |
| } GENERIC_HEADER; |
| |
| typedef struct _PIPELINE_NODE { |
| VkPipeline pipeline; |
| VkGraphicsPipelineCreateInfo graphicsPipelineCI; |
| VkPipelineVertexInputStateCreateInfo vertexInputCI; |
| VkPipelineInputAssemblyStateCreateInfo iaStateCI; |
| VkPipelineTessellationStateCreateInfo tessStateCI; |
| VkPipelineViewportStateCreateInfo vpStateCI; |
| VkPipelineRasterizationStateCreateInfo rsStateCI; |
| VkPipelineMultisampleStateCreateInfo msStateCI; |
| VkPipelineColorBlendStateCreateInfo cbStateCI; |
| VkPipelineDepthStencilStateCreateInfo dsStateCI; |
| VkPipelineDynamicStateCreateInfo dynStateCI; |
| VkPipelineShaderStageCreateInfo vsCI; |
| VkPipelineShaderStageCreateInfo tcsCI; |
| VkPipelineShaderStageCreateInfo tesCI; |
| VkPipelineShaderStageCreateInfo gsCI; |
| VkPipelineShaderStageCreateInfo fsCI; |
| // Compute shader is include in VkComputePipelineCreateInfo |
| VkComputePipelineCreateInfo computePipelineCI; |
| // Flag of which shader stages are active for this pipeline |
| uint32_t active_shaders; |
| // Capture which slots (set#->bindings) are actually used by the shaders of this pipeline |
| unordered_map<uint32_t, unordered_set<uint32_t>> active_slots; |
| // Vtx input info (if any) |
| std::vector<VkVertexInputBindingDescription> vertexBindingDescriptions; |
| std::vector<VkVertexInputAttributeDescription> vertexAttributeDescriptions; |
| std::vector<VkPipelineColorBlendAttachmentState> attachments; |
| bool blendConstantsEnabled; // Blend constants enabled for any attachments |
| // Default constructor |
| _PIPELINE_NODE() |
| : pipeline{}, graphicsPipelineCI{}, vertexInputCI{}, iaStateCI{}, tessStateCI{}, vpStateCI{}, rsStateCI{}, msStateCI{}, |
| cbStateCI{}, dsStateCI{}, dynStateCI{}, vsCI{}, tcsCI{}, tesCI{}, gsCI{}, fsCI{}, computePipelineCI{}, active_shaders(0), |
| active_slots(), vertexBindingDescriptions(), vertexAttributeDescriptions(), attachments(), blendConstantsEnabled(false) {} |
| } PIPELINE_NODE; |
| |
| class BASE_NODE { |
| public: |
| std::atomic_int in_use; |
| }; |
| |
| typedef struct _SAMPLER_NODE { |
| VkSampler sampler; |
| VkSamplerCreateInfo createInfo; |
| |
| _SAMPLER_NODE(const VkSampler *ps, const VkSamplerCreateInfo *pci) : sampler(*ps), createInfo(*pci){}; |
| } SAMPLER_NODE; |
| |
| class IMAGE_NODE : public BASE_NODE { |
| public: |
| VkImageCreateInfo createInfo; |
| VkDeviceMemory mem; |
| VkDeviceSize memOffset; |
| VkDeviceSize memSize; |
| }; |
| |
| typedef struct _IMAGE_LAYOUT_NODE { |
| VkImageLayout layout; |
| VkFormat format; |
| } IMAGE_LAYOUT_NODE; |
| |
| class IMAGE_CMD_BUF_LAYOUT_NODE { |
| public: |
| IMAGE_CMD_BUF_LAYOUT_NODE() {} |
| IMAGE_CMD_BUF_LAYOUT_NODE(VkImageLayout initialLayoutInput, VkImageLayout layoutInput) |
| : initialLayout(initialLayoutInput), layout(layoutInput) {} |
| |
| VkImageLayout initialLayout; |
| VkImageLayout layout; |
| }; |
| |
| class BUFFER_NODE : public BASE_NODE { |
| public: |
| using BASE_NODE::in_use; |
| unique_ptr<VkBufferCreateInfo> create_info; |
| }; |
| |
| // Store the DAG. |
| struct DAGNode { |
| uint32_t pass; |
| std::vector<uint32_t> prev; |
| std::vector<uint32_t> next; |
| }; |
| |
| struct RENDER_PASS_NODE { |
| VkRenderPassCreateInfo const *pCreateInfo; |
| VkFramebuffer fb; |
| vector<bool> hasSelfDependency; |
| vector<DAGNode> subpassToNode; |
| vector<vector<VkFormat>> subpassColorFormats; |
| vector<MT_PASS_ATTACHMENT_INFO> attachments; |
| unordered_map<uint32_t, bool> attachment_first_read; |
| unordered_map<uint32_t, VkImageLayout> attachment_first_layout; |
| |
| RENDER_PASS_NODE(VkRenderPassCreateInfo const *pCreateInfo) : pCreateInfo(pCreateInfo), fb(VK_NULL_HANDLE) { |
| uint32_t i; |
| |
| subpassColorFormats.reserve(pCreateInfo->subpassCount); |
| for (i = 0; i < pCreateInfo->subpassCount; i++) { |
| const VkSubpassDescription *subpass = &pCreateInfo->pSubpasses[i]; |
| vector<VkFormat> color_formats; |
| uint32_t j; |
| |
| color_formats.reserve(subpass->colorAttachmentCount); |
| for (j = 0; j < subpass->colorAttachmentCount; j++) { |
| const uint32_t att = subpass->pColorAttachments[j].attachment; |
| const VkFormat format = pCreateInfo->pAttachments[att].format; |
| |
| color_formats.push_back(format); |
| } |
| |
| subpassColorFormats.push_back(color_formats); |
| } |
| } |
| }; |
| |
| class PHYS_DEV_PROPERTIES_NODE { |
| public: |
| VkPhysicalDeviceProperties properties; |
| VkPhysicalDeviceFeatures features; |
| vector<VkQueueFamilyProperties> queue_family_properties; |
| }; |
| |
| class FENCE_NODE : public BASE_NODE { |
| public: |
| using BASE_NODE::in_use; |
| #if MTMERGE |
| uint64_t fenceId; // Sequence number for fence at last submit |
| VkSwapchainKHR swapchain; // Swapchain that this fence is submitted against or NULL |
| VkBool32 firstTimeFlag; // Fence was created in signaled state, avoid warnings for first use |
| VkFenceCreateInfo createInfo; |
| #endif |
| VkQueue queue; |
| vector<VkCommandBuffer> cmdBuffers; |
| bool needsSignaled; |
| vector<VkFence> priorFences; |
| |
| // Default constructor |
| FENCE_NODE() : queue(NULL), needsSignaled(VK_FALSE){}; |
| }; |
| |
| class SEMAPHORE_NODE : public BASE_NODE { |
| public: |
| using BASE_NODE::in_use; |
| uint32_t signaled; |
| SemaphoreState state; |
| VkQueue queue; |
| }; |
| |
| class EVENT_NODE : public BASE_NODE { |
| public: |
| using BASE_NODE::in_use; |
| bool needsSignaled; |
| VkPipelineStageFlags stageMask; |
| }; |
| |
| class QUEUE_NODE { |
| public: |
| VkDevice device; |
| vector<VkFence> lastFences; |
| #if MTMERGE |
| uint64_t lastRetiredId; |
| uint64_t lastSubmittedId; |
| // MTMTODO : merge cmd_buffer data structs here |
| list<VkCommandBuffer> pQueueCommandBuffers; |
| list<VkDeviceMemory> pMemRefList; |
| #endif |
| vector<VkCommandBuffer> untrackedCmdBuffers; |
| unordered_set<VkCommandBuffer> inFlightCmdBuffers; |
| unordered_map<VkEvent, VkPipelineStageFlags> eventToStageMap; |
| }; |
| |
| class QUERY_POOL_NODE : public BASE_NODE { |
| public: |
| VkQueryPoolCreateInfo createInfo; |
| }; |
| |
| class FRAMEBUFFER_NODE { |
| public: |
| VkFramebufferCreateInfo createInfo; |
| unordered_set<VkCommandBuffer> referencingCmdBuffers; |
| vector<MT_FB_ATTACHMENT_INFO> attachments; |
| }; |
| |
| // Descriptor Data structures |
| // Layout Node has the core layout data |
| typedef struct _LAYOUT_NODE { |
| VkDescriptorSetLayout layout; |
| VkDescriptorSetLayoutCreateInfo createInfo; |
| uint32_t startIndex; // 1st index of this layout |
| uint32_t endIndex; // last index of this layout |
| uint32_t dynamicDescriptorCount; // Total count of dynamic descriptors used |
| // by this layout |
| vector<VkDescriptorType> descriptorTypes; // Type per descriptor in this |
| // layout to verify correct |
| // updates |
| vector<VkShaderStageFlags> stageFlags; // stageFlags per descriptor in this |
| // layout to verify correct updates |
| unordered_map<uint32_t, uint32_t> bindingToIndexMap; // map set binding # to |
| // createInfo.pBindings index |
| // Default constructor |
| _LAYOUT_NODE() : layout{}, createInfo{}, startIndex(0), endIndex(0), dynamicDescriptorCount(0){}; |
| } LAYOUT_NODE; |
| |
| // Store layouts and pushconstants for PipelineLayout |
| struct PIPELINE_LAYOUT_NODE { |
| vector<VkDescriptorSetLayout> descriptorSetLayouts; |
| vector<VkPushConstantRange> pushConstantRanges; |
| }; |
| |
| class SET_NODE : public BASE_NODE { |
| public: |
| using BASE_NODE::in_use; |
| VkDescriptorSet set; |
| VkDescriptorPool pool; |
| // Head of LL of all Update structs for this set |
| GENERIC_HEADER *pUpdateStructs; |
| // Total num of descriptors in this set (count of its layout plus all prior layouts) |
| uint32_t descriptorCount; |
| vector<GENERIC_HEADER*> pDescriptorUpdates; // Vector where each index points to update node for its slot |
| LAYOUT_NODE *pLayout; // Layout for this set |
| SET_NODE *pNext; |
| unordered_set<VkCommandBuffer> boundCmdBuffers; // Cmd buffers that this set has been bound to |
| SET_NODE() : set(VK_NULL_HANDLE), pool(VK_NULL_HANDLE), pUpdateStructs(nullptr), pLayout(nullptr), pNext(nullptr){}; |
| }; |
| |
| typedef struct _DESCRIPTOR_POOL_NODE { |
| VkDescriptorPool pool; |
| uint32_t maxSets; // Max descriptor sets allowed in this pool |
| uint32_t availableSets; // Available descriptor sets in this pool |
| |
| VkDescriptorPoolCreateInfo createInfo; |
| SET_NODE *pSets; // Head of LL of sets for this Pool |
| vector<uint32_t> maxDescriptorTypeCount; // Max # of descriptors of each type in this pool |
| vector<uint32_t> availableDescriptorTypeCount; // Available # of descriptors of each type in this pool |
| |
| _DESCRIPTOR_POOL_NODE(const VkDescriptorPool pool, const VkDescriptorPoolCreateInfo *pCreateInfo) |
| : pool(pool), maxSets(pCreateInfo->maxSets), availableSets(pCreateInfo->maxSets), createInfo(*pCreateInfo), pSets(NULL), |
| maxDescriptorTypeCount(VK_DESCRIPTOR_TYPE_RANGE_SIZE), availableDescriptorTypeCount(VK_DESCRIPTOR_TYPE_RANGE_SIZE) { |
| if (createInfo.poolSizeCount) { // Shadow type struct from ptr into local struct |
| size_t poolSizeCountSize = createInfo.poolSizeCount * sizeof(VkDescriptorPoolSize); |
| createInfo.pPoolSizes = new VkDescriptorPoolSize[poolSizeCountSize]; |
| memcpy((void *)createInfo.pPoolSizes, pCreateInfo->pPoolSizes, poolSizeCountSize); |
| // Now set max counts for each descriptor type based on count of that type times maxSets |
| uint32_t i = 0; |
| for (i = 0; i < createInfo.poolSizeCount; ++i) { |
| uint32_t typeIndex = static_cast<uint32_t>(createInfo.pPoolSizes[i].type); |
| maxDescriptorTypeCount[typeIndex] = createInfo.pPoolSizes[i].descriptorCount; |
| availableDescriptorTypeCount[typeIndex] = maxDescriptorTypeCount[typeIndex]; |
| } |
| } else { |
| createInfo.pPoolSizes = NULL; // Make sure this is NULL so we don't try to clean it up |
| } |
| } |
| ~_DESCRIPTOR_POOL_NODE() { |
| delete[] createInfo.pPoolSizes; |
| // TODO : pSets are currently freed in deletePools function which uses freeShadowUpdateTree function |
| // need to migrate that struct to smart ptrs for auto-cleanup |
| } |
| } DESCRIPTOR_POOL_NODE; |
| |
| // Cmd Buffer Tracking |
| typedef enum _CMD_TYPE { |
| CMD_BINDPIPELINE, |
| CMD_BINDPIPELINEDELTA, |
| CMD_SETVIEWPORTSTATE, |
| CMD_SETSCISSORSTATE, |
| CMD_SETLINEWIDTHSTATE, |
| CMD_SETDEPTHBIASSTATE, |
| CMD_SETBLENDSTATE, |
| CMD_SETDEPTHBOUNDSSTATE, |
| CMD_SETSTENCILREADMASKSTATE, |
| CMD_SETSTENCILWRITEMASKSTATE, |
| CMD_SETSTENCILREFERENCESTATE, |
| CMD_BINDDESCRIPTORSETS, |
| CMD_BINDINDEXBUFFER, |
| CMD_BINDVERTEXBUFFER, |
| CMD_DRAW, |
| CMD_DRAWINDEXED, |
| CMD_DRAWINDIRECT, |
| CMD_DRAWINDEXEDINDIRECT, |
| CMD_DISPATCH, |
| CMD_DISPATCHINDIRECT, |
| CMD_COPYBUFFER, |
| CMD_COPYIMAGE, |
| CMD_BLITIMAGE, |
| CMD_COPYBUFFERTOIMAGE, |
| CMD_COPYIMAGETOBUFFER, |
| CMD_CLONEIMAGEDATA, |
| CMD_UPDATEBUFFER, |
| CMD_FILLBUFFER, |
| CMD_CLEARCOLORIMAGE, |
| CMD_CLEARATTACHMENTS, |
| CMD_CLEARDEPTHSTENCILIMAGE, |
| CMD_RESOLVEIMAGE, |
| CMD_SETEVENT, |
| CMD_RESETEVENT, |
| CMD_WAITEVENTS, |
| CMD_PIPELINEBARRIER, |
| CMD_BEGINQUERY, |
| CMD_ENDQUERY, |
| CMD_RESETQUERYPOOL, |
| CMD_COPYQUERYPOOLRESULTS, |
| CMD_WRITETIMESTAMP, |
| CMD_PUSHCONSTANTS, |
| CMD_INITATOMICCOUNTERS, |
| CMD_LOADATOMICCOUNTERS, |
| CMD_SAVEATOMICCOUNTERS, |
| CMD_BEGINRENDERPASS, |
| CMD_NEXTSUBPASS, |
| CMD_ENDRENDERPASS, |
| CMD_EXECUTECOMMANDS, |
| } CMD_TYPE; |
| // Data structure for holding sequence of cmds in cmd buffer |
| typedef struct _CMD_NODE { |
| CMD_TYPE type; |
| uint64_t cmdNumber; |
| } CMD_NODE; |
| |
| typedef enum _CB_STATE { |
| CB_NEW, // Newly created CB w/o any cmds |
| CB_RECORDING, // BeginCB has been called on this CB |
| CB_RECORDED, // EndCB has been called on this CB |
| CB_INVALID // CB had a bound descriptor set destroyed or updated |
| } CB_STATE; |
| // CB Status -- used to track status of various bindings on cmd buffer objects |
| typedef VkFlags CBStatusFlags; |
| typedef enum _CBStatusFlagBits { |
| // clang-format off |
| CBSTATUS_NONE = 0x00000000, // No status is set |
| CBSTATUS_VIEWPORT_SET = 0x00000001, // Viewport has been set |
| CBSTATUS_LINE_WIDTH_SET = 0x00000002, // Line width has been set |
| CBSTATUS_DEPTH_BIAS_SET = 0x00000004, // Depth bias has been set |
| CBSTATUS_BLEND_CONSTANTS_SET = 0x00000008, // Blend constants state has been set |
| CBSTATUS_DEPTH_BOUNDS_SET = 0x00000010, // Depth bounds state object has been set |
| CBSTATUS_STENCIL_READ_MASK_SET = 0x00000020, // Stencil read mask has been set |
| CBSTATUS_STENCIL_WRITE_MASK_SET = 0x00000040, // Stencil write mask has been set |
| CBSTATUS_STENCIL_REFERENCE_SET = 0x00000080, // Stencil reference has been set |
| CBSTATUS_INDEX_BUFFER_BOUND = 0x00000100, // Index buffer has been set |
| CBSTATUS_SCISSOR_SET = 0x00000200, // Scissor has been set |
| CBSTATUS_ALL = 0x000003FF, // All dynamic state set |
| // clang-format on |
| } CBStatusFlagBits; |
| |
| typedef struct stencil_data { |
| uint32_t compareMask; |
| uint32_t writeMask; |
| uint32_t reference; |
| } CBStencilData; |
| |
| typedef struct _DRAW_DATA { vector<VkBuffer> buffers; } DRAW_DATA; |
| |
| struct ImageSubresourcePair { |
| VkImage image; |
| bool hasSubresource; |
| VkImageSubresource subresource; |
| }; |
| |
| bool operator==(const ImageSubresourcePair &img1, const ImageSubresourcePair &img2) { |
| if (img1.image != img2.image || img1.hasSubresource != img2.hasSubresource) |
| return false; |
| return !img1.hasSubresource || |
| (img1.subresource.aspectMask == img2.subresource.aspectMask && img1.subresource.mipLevel == img2.subresource.mipLevel && |
| img1.subresource.arrayLayer == img2.subresource.arrayLayer); |
| } |
| |
| namespace std { |
| template <> struct hash<ImageSubresourcePair> { |
| size_t operator()(ImageSubresourcePair img) const throw() { |
| size_t hashVal = hash<uint64_t>()(reinterpret_cast<uint64_t &>(img.image)); |
| hashVal ^= hash<bool>()(img.hasSubresource); |
| if (img.hasSubresource) { |
| hashVal ^= hash<uint32_t>()(reinterpret_cast<uint32_t &>(img.subresource.aspectMask)); |
| hashVal ^= hash<uint32_t>()(img.subresource.mipLevel); |
| hashVal ^= hash<uint32_t>()(img.subresource.arrayLayer); |
| } |
| return hashVal; |
| } |
| }; |
| } |
| |
| struct QueryObject { |
| VkQueryPool pool; |
| uint32_t index; |
| }; |
| |
| bool operator==(const QueryObject &query1, const QueryObject &query2) { |
| return (query1.pool == query2.pool && query1.index == query2.index); |
| } |
| |
| namespace std { |
| template <> struct hash<QueryObject> { |
| size_t operator()(QueryObject query) const throw() { |
| return hash<uint64_t>()((uint64_t)(query.pool)) ^ hash<uint32_t>()(query.index); |
| } |
| }; |
| } |
| // Track last states that are bound per pipeline bind point (Gfx & Compute) |
| struct LAST_BOUND_STATE { |
| VkPipeline pipeline; |
| VkPipelineLayout pipelineLayout; |
| // Track each set that has been bound |
| // TODO : can unique be global per CB? (do we care about Gfx vs. Compute?) |
| unordered_set<VkDescriptorSet> uniqueBoundSets; |
| // Ordered bound set tracking where index is set# that given set is bound to |
| vector<VkDescriptorSet> boundDescriptorSets; |
| // one dynamic offset per dynamic descriptor bound to this CB |
| vector<uint32_t> dynamicOffsets; |
| void reset() { |
| pipeline = VK_NULL_HANDLE; |
| pipelineLayout = VK_NULL_HANDLE; |
| uniqueBoundSets.clear(); |
| boundDescriptorSets.clear(); |
| dynamicOffsets.clear(); |
| } |
| }; |
| // Cmd Buffer Wrapper Struct |
| struct GLOBAL_CB_NODE { |
| VkCommandBuffer commandBuffer; |
| VkCommandBufferAllocateInfo createInfo; |
| VkCommandBufferBeginInfo beginInfo; |
| VkCommandBufferInheritanceInfo inheritanceInfo; |
| // VkFence fence; // fence tracking this cmd buffer |
| VkDevice device; // device this CB belongs to |
| uint64_t numCmds; // number of cmds in this CB |
| uint64_t drawCount[NUM_DRAW_TYPES]; // Count of each type of draw in this CB |
| CB_STATE state; // Track cmd buffer update state |
| uint64_t submitCount; // Number of times CB has been submitted |
| CBStatusFlags status; // Track status of various bindings on cmd buffer |
| vector<CMD_NODE> cmds; // vector of commands bound to this command buffer |
| // Currently storing "lastBound" objects on per-CB basis |
| // long-term may want to create caches of "lastBound" states and could have |
| // each individual CMD_NODE referencing its own "lastBound" state |
| // VkPipeline lastBoundPipeline; |
| // VkPipelineLayout lastBoundPipelineLayout; |
| // // Capture unique std::set of descriptorSets that are bound to this CB. |
| // std::set<VkDescriptorSet> uniqueBoundSets; |
| // vector<VkDescriptorSet> boundDescriptorSets; // Index is set# that given set is bound to |
| // Store last bound state for Gfx & Compute pipeline bind points |
| LAST_BOUND_STATE lastBound[VK_PIPELINE_BIND_POINT_RANGE_SIZE]; |
| |
| vector<uint32_t> dynamicOffsets; |
| vector<VkViewport> viewports; |
| vector<VkRect2D> scissors; |
| VkRenderPassBeginInfo activeRenderPassBeginInfo; |
| uint64_t fenceId; |
| VkFence lastSubmittedFence; |
| VkQueue lastSubmittedQueue; |
| VkRenderPass activeRenderPass; |
| VkSubpassContents activeSubpassContents; |
| uint32_t activeSubpass; |
| VkFramebuffer framebuffer; |
| // Track descriptor sets that are destroyed or updated while bound to CB |
| // TODO : These data structures relate to tracking resources that invalidate |
| // a cmd buffer that references them. Need to unify how we handle these |
| // cases so we don't have different tracking data for each type. |
| std::set<VkDescriptorSet> destroyedSets; |
| std::set<VkDescriptorSet> updatedSets; |
| unordered_set<VkFramebuffer> destroyedFramebuffers; |
| vector<VkEvent> waitedEvents; |
| vector<VkSemaphore> semaphores; |
| vector<VkEvent> events; |
| unordered_map<QueryObject, vector<VkEvent>> waitedEventsBeforeQueryReset; |
| unordered_map<QueryObject, bool> queryToStateMap; // 0 is unavailable, 1 is available |
| unordered_set<QueryObject> activeQueries; |
| unordered_set<QueryObject> startedQueries; |
| unordered_map<ImageSubresourcePair, IMAGE_CMD_BUF_LAYOUT_NODE> imageLayoutMap; |
| unordered_map<VkImage, vector<ImageSubresourcePair>> imageSubresourceMap; |
| unordered_map<VkEvent, VkPipelineStageFlags> eventToStageMap; |
| vector<DRAW_DATA> drawData; |
| DRAW_DATA currentDrawData; |
| VkCommandBuffer primaryCommandBuffer; |
| // Track images and buffers that are updated by this CB at the point of a draw |
| unordered_set<VkImageView> updateImages; |
| unordered_set<VkBuffer> updateBuffers; |
| // If cmd buffer is primary, track secondary command buffers pending |
| // execution |
| std::unordered_set<VkCommandBuffer> secondaryCommandBuffers; |
| // MTMTODO : Scrub these data fields and merge active sets w/ lastBound as appropriate |
| vector<std::function<VkBool32()>> validate_functions; |
| std::unordered_set<VkDeviceMemory> memObjs; |
| vector<std::function<bool(VkQueue)>> eventUpdates; |
| }; |
| |
| class SWAPCHAIN_NODE { |
| public: |
| VkSwapchainCreateInfoKHR createInfo; |
| uint32_t *pQueueFamilyIndices; |
| std::vector<VkImage> images; |
| SWAPCHAIN_NODE(const VkSwapchainCreateInfoKHR *pCreateInfo) : createInfo(*pCreateInfo), pQueueFamilyIndices(NULL) { |
| if (pCreateInfo->queueFamilyIndexCount && pCreateInfo->imageSharingMode == VK_SHARING_MODE_CONCURRENT) { |
| pQueueFamilyIndices = new uint32_t[pCreateInfo->queueFamilyIndexCount]; |
| memcpy(pQueueFamilyIndices, pCreateInfo->pQueueFamilyIndices, pCreateInfo->queueFamilyIndexCount * sizeof(uint32_t)); |
| createInfo.pQueueFamilyIndices = pQueueFamilyIndices; |
| } |
| } |
| ~SWAPCHAIN_NODE() { delete[] pQueueFamilyIndices; } |
| }; |