Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 1 | /* |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 2 | * Copyright (c) 2015-2016 The Khronos Group Inc. |
| 3 | * Copyright (c) 2015-2016 Valve Corporation |
| 4 | * Copyright (c) 2015-2016 LunarG, Inc. |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 5 | * |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 6 | * Permission is hereby granted, free of charge, to any person obtaining a copy |
| 7 | * of this software and/or associated documentation files (the "Materials"), to |
| 8 | * deal in the Materials without restriction, including without limitation the |
| 9 | * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or |
| 10 | * sell copies of the Materials, and to permit persons to whom the Materials are |
| 11 | * furnished to do so, subject to the following conditions: |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 12 | * |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 13 | * The above copyright notice(s) and this permission notice shall be included in |
| 14 | * all copies or substantial portions of the Materials. |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 15 | * |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 16 | * THE MATERIALS ARE PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 17 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 18 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. |
| 19 | * |
| 20 | * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, |
| 21 | * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR |
| 22 | * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE MATERIALS OR THE |
| 23 | * USE OR OTHER DEALINGS IN THE MATERIALS. |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 24 | * |
Courtney Goeltzenleuchter | 0555952 | 2015-10-30 11:14:30 -0600 | [diff] [blame] | 25 | * Author: Courtney Goeltzenleuchter <courtney@LunarG.com> |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 26 | */ |
| 27 | |
| 28 | #ifndef VKRENDERFRAMEWORK_H |
| 29 | #define VKRENDERFRAMEWORK_H |
| 30 | |
Michael Lentine | 83ab341 | 2015-11-03 16:20:30 -0800 | [diff] [blame] | 31 | #ifdef ANDROID |
| 32 | #include "vktestframeworkandroid.h" |
| 33 | class VkImageObj; |
| 34 | #else |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 35 | #include "vktestframework.h" |
Michael Lentine | 83ab341 | 2015-11-03 16:20:30 -0800 | [diff] [blame] | 36 | #endif |
David Pinedo | abd0772 | 2015-11-24 09:00:24 -0700 | [diff] [blame] | 37 | #include "vulkan/vk_lunarg_debug_marker.h" |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 38 | |
Michael Lentine | 83ab341 | 2015-11-03 16:20:30 -0800 | [diff] [blame] | 39 | #include <vector> |
| 40 | |
| 41 | using namespace std; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 42 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 43 | class VkDeviceObj : public vk_testing::Device { |
| 44 | public: |
Tony Barbour | d1c3572 | 2015-04-16 15:59:00 -0600 | [diff] [blame] | 45 | VkDeviceObj(uint32_t id, VkPhysicalDevice obj); |
Courtney Goeltzenleuchter | 5bac609 | 2015-07-07 11:47:33 -0600 | [diff] [blame] | 46 | VkDeviceObj(uint32_t id, VkPhysicalDevice obj, |
| 47 | std::vector<const char *> &layers, |
| 48 | std::vector<const char *> &extension_names); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 49 | |
Chia-I Wu | f368b60 | 2015-07-03 10:41:20 +0800 | [diff] [blame] | 50 | VkDevice device() { return handle(); } |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 51 | void get_device_queue(); |
| 52 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 53 | uint32_t id; |
| 54 | VkPhysicalDeviceProperties props; |
| 55 | const VkQueueFamilyProperties *queue_props; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 56 | |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 57 | VkQueue m_queue; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 58 | }; |
| 59 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 60 | class VkDepthStencilObj : public vk_testing::Image { |
| 61 | public: |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 62 | VkDepthStencilObj(); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 63 | void Init(VkDeviceObj *device, int32_t width, int32_t height, |
| 64 | VkFormat format); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 65 | bool Initialized(); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 66 | VkImageView *BindInfo(); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 67 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 68 | protected: |
| 69 | VkDeviceObj *m_device; |
| 70 | bool m_initialized; |
| 71 | vk_testing::ImageView m_imageView; |
| 72 | VkFormat m_depth_stencil_fmt; |
| 73 | VkImageView m_attachmentBindInfo; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 74 | }; |
| 75 | |
Tony Barbour | fe3351b | 2015-07-28 10:17:20 -0600 | [diff] [blame] | 76 | class VkCommandBufferObj; |
| 77 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 78 | class VkRenderFramework : public VkTestFramework { |
| 79 | public: |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 80 | VkRenderFramework(); |
| 81 | ~VkRenderFramework(); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 82 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 83 | VkDevice device() { return m_device->device(); } |
| 84 | VkPhysicalDevice gpu() { return objs[0]; } |
| 85 | VkRenderPass renderPass() { return m_renderPass; } |
| 86 | VkFramebuffer framebuffer() { return m_framebuffer; } |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 87 | void InitViewport(float width, float height); |
| 88 | void InitViewport(); |
| 89 | void InitRenderTarget(); |
| 90 | void InitRenderTarget(uint32_t targets); |
Courtney Goeltzenleuchter | 5861a1b | 2015-09-01 17:30:39 -0600 | [diff] [blame] | 91 | void InitRenderTarget(VkImageView *dsBinding); |
| 92 | void InitRenderTarget(uint32_t targets, VkImageView *dsBinding); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 93 | void InitFramework(); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 94 | void InitFramework(std::vector<const char *> instance_layer_names, |
| 95 | std::vector<const char *> device_layer_names, |
| 96 | std::vector<const char *> instance_extension_names, |
| 97 | std::vector<const char *> device_extension_names, |
| 98 | PFN_vkDebugReportCallbackEXT = NULL, |
| 99 | void *userData = NULL); |
Courtney Goeltzenleuchter | cd69eee | 2015-07-06 09:10:47 -0600 | [diff] [blame] | 100 | |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 101 | void ShutdownFramework(); |
| 102 | void InitState(); |
| 103 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 104 | const VkRenderPassBeginInfo &renderPassBeginInfo() const { |
| 105 | return m_renderPassBeginInfo; |
| 106 | } |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 107 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 108 | protected: |
| 109 | VkApplicationInfo app_info; |
| 110 | VkInstance inst; |
| 111 | VkPhysicalDevice objs[16]; |
| 112 | uint32_t gpu_count; |
| 113 | VkDeviceObj *m_device; |
| 114 | VkCommandPool m_commandPool; |
| 115 | VkCommandBufferObj *m_commandBuffer; |
| 116 | VkRenderPass m_renderPass; |
| 117 | VkFramebuffer m_framebuffer; |
| 118 | std::vector<VkViewport> m_viewports; |
| 119 | std::vector<VkRect2D> m_scissors; |
| 120 | float m_lineWidth; |
| 121 | float m_depthBiasConstantFactor; |
| 122 | float m_depthBiasClamp; |
| 123 | float m_depthBiasSlopeFactor; |
| 124 | float m_blendConstants[4]; |
| 125 | float m_minDepthBounds; |
| 126 | float m_maxDepthBounds; |
| 127 | uint32_t m_compareMask; |
| 128 | uint32_t m_writeMask; |
| 129 | uint32_t m_reference; |
| 130 | std::vector<VkClearValue> m_renderPassClearValues; |
| 131 | VkRenderPassBeginInfo m_renderPassBeginInfo; |
| 132 | vector<VkImageObj *> m_renderTargets; |
| 133 | float m_width, m_height; |
| 134 | VkFormat m_render_target_fmt; |
| 135 | VkFormat m_depth_stencil_fmt; |
| 136 | VkClearColorValue m_clear_color; |
| 137 | bool m_clear_via_load_op; |
| 138 | float m_depth_clear_color; |
| 139 | uint32_t m_stencil_clear_color; |
| 140 | VkDepthStencilObj *m_depthStencil; |
| 141 | PFN_vkCreateDebugReportCallbackEXT m_CreateDebugReportCallback; |
Courtney Goeltzenleuchter | 011ccf8 | 2015-12-15 09:25:15 -0700 | [diff] [blame] | 142 | PFN_vkDestroyDebugReportCallbackEXT m_DestroyDebugReportCallback; |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 143 | PFN_vkDebugReportMessageEXT m_DebugReportMessage; |
| 144 | VkDebugReportCallbackEXT m_globalMsgCallback; |
| 145 | VkDebugReportCallbackEXT m_devMsgCallback; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 146 | |
| 147 | /* |
| 148 | * SetUp and TearDown are called by the Google Test framework |
| 149 | * to initialize a test framework based on this class. |
| 150 | */ |
| 151 | virtual void SetUp() { |
| 152 | this->app_info.sType = VK_STRUCTURE_TYPE_APPLICATION_INFO; |
| 153 | this->app_info.pNext = NULL; |
Chia-I Wu | 3432a0c | 2015-10-27 18:04:07 +0800 | [diff] [blame] | 154 | this->app_info.pApplicationName = "base"; |
| 155 | this->app_info.applicationVersion = 1; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 156 | this->app_info.pEngineName = "unittest"; |
| 157 | this->app_info.engineVersion = 1; |
Tony Barbour | aabb359 | 2016-02-25 13:58:50 -0700 | [diff] [blame] | 158 | this->app_info.apiVersion = VK_MAKE_VERSION(1, 0, 0); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 159 | |
| 160 | InitFramework(); |
| 161 | } |
| 162 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 163 | virtual void TearDown() { ShutdownFramework(); } |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 164 | }; |
| 165 | |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 166 | class VkDescriptorSetObj; |
| 167 | class VkIndexBufferObj; |
| 168 | class VkConstantBufferObj; |
| 169 | class VkPipelineObj; |
| 170 | class VkDescriptorSetObj; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 171 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 172 | class VkCommandBufferObj : public vk_testing::CommandBuffer { |
| 173 | public: |
Chia-I Wu | 3432a0c | 2015-10-27 18:04:07 +0800 | [diff] [blame] | 174 | VkCommandBufferObj(VkDeviceObj *device, VkCommandPool pool); |
| 175 | VkCommandBuffer GetBufferHandle(); |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 176 | VkResult BeginCommandBuffer(); |
Chia-I Wu | 3432a0c | 2015-10-27 18:04:07 +0800 | [diff] [blame] | 177 | VkResult BeginCommandBuffer(VkCommandBufferBeginInfo *pInfo); |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 178 | VkResult EndCommandBuffer(); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 179 | void PipelineBarrier(VkPipelineStageFlags src_stages, |
| 180 | VkPipelineStageFlags dest_stages, |
| 181 | VkDependencyFlags dependencyFlags, |
| 182 | uint32_t memoryBarrierCount, |
| 183 | const VkMemoryBarrier *pMemoryBarriers, |
| 184 | uint32_t bufferMemoryBarrierCount, |
| 185 | const VkBufferMemoryBarrier *pBufferMemoryBarriers, |
| 186 | uint32_t imageMemoryBarrierCount, |
| 187 | const VkImageMemoryBarrier *pImageMemoryBarriers); |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 188 | void AddRenderTarget(VkImageObj *renderTarget); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 189 | void AddDepthStencil(); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 190 | void ClearAllBuffers(VkClearColorValue clear_color, float depth_clear_color, |
| 191 | uint32_t stencil_clear_color, |
| 192 | VkDepthStencilObj *depthStencilObj); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 193 | void PrepareAttachments(); |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 194 | void BindPipeline(VkPipelineObj &pipeline); |
| 195 | void BindDescriptorSet(VkDescriptorSetObj &descriptorSet); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 196 | void BindVertexBuffer(VkConstantBufferObj *vertexBuffer, |
| 197 | VkDeviceSize offset, uint32_t binding); |
Cody Northrop | d2ad034 | 2015-08-05 11:15:02 -0600 | [diff] [blame] | 198 | void BindIndexBuffer(VkIndexBufferObj *indexBuffer, VkDeviceSize offset); |
Chia-I Wu | 08accc6 | 2015-07-07 11:50:03 +0800 | [diff] [blame] | 199 | void BeginRenderPass(const VkRenderPassBeginInfo &info); |
Chia-I Wu | 0b50a1c | 2015-06-26 15:34:39 +0800 | [diff] [blame] | 200 | void EndRenderPass(); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 201 | void FillBuffer(VkBuffer buffer, VkDeviceSize offset, |
| 202 | VkDeviceSize fill_size, uint32_t data); |
| 203 | void Draw(uint32_t vertexCount, uint32_t instanceCount, |
| 204 | uint32_t firstVertex, uint32_t firstInstance); |
| 205 | void DrawIndexed(uint32_t indexCount, uint32_t instanceCount, |
| 206 | uint32_t firstIndex, int32_t vertexOffset, |
| 207 | uint32_t firstInstance); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 208 | void QueueCommandBuffer(); |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 209 | void QueueCommandBuffer(VkFence fence); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 210 | void SetViewport(uint32_t firstViewport, uint32_t viewportCount, |
| 211 | const VkViewport *pViewports); |
| 212 | void SetScissor(uint32_t firstScissor, uint32_t scissorCount, |
| 213 | const VkRect2D *pScissors); |
Courtney Goeltzenleuchter | 49c7308 | 2015-09-17 15:06:17 -0600 | [diff] [blame] | 214 | void SetLineWidth(float lineWidth); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 215 | void SetDepthBias(float depthBiasConstantFactor, float depthBiasClamp, |
| 216 | float depthBiasSlopeFactor); |
Chia-I Wu | 2bfb33c | 2015-10-26 17:24:52 +0800 | [diff] [blame] | 217 | void SetBlendConstants(const float blendConstants[4]); |
Courtney Goeltzenleuchter | 49c7308 | 2015-09-17 15:06:17 -0600 | [diff] [blame] | 218 | void SetDepthBounds(float minDepthBounds, float maxDepthBounds); |
Chia-I Wu | 1b99bb2 | 2015-10-27 19:25:11 +0800 | [diff] [blame] | 219 | void SetStencilReadMask(VkStencilFaceFlags faceMask, uint32_t compareMask); |
| 220 | void SetStencilWriteMask(VkStencilFaceFlags faceMask, uint32_t writeMask); |
| 221 | void SetStencilReference(VkStencilFaceFlags faceMask, uint32_t reference); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 222 | void UpdateBuffer(VkBuffer buffer, VkDeviceSize dstOffset, |
| 223 | VkDeviceSize dataSize, const uint32_t *pData); |
| 224 | void CopyImage(VkImage srcImage, VkImageLayout srcImageLayout, |
| 225 | VkImage dstImage, VkImageLayout dstImageLayout, |
| 226 | uint32_t regionCount, const VkImageCopy *pRegions); |
| 227 | void ResolveImage(VkImage srcImage, VkImageLayout srcImageLayout, |
| 228 | VkImage dstImage, VkImageLayout dstImageLayout, |
| 229 | uint32_t regionCount, const VkImageResolve *pRegions); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 230 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 231 | protected: |
| 232 | VkDeviceObj *m_device; |
| 233 | vector<VkImageObj *> m_renderTargets; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 234 | }; |
| 235 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 236 | class VkConstantBufferObj : public vk_testing::Buffer { |
| 237 | public: |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 238 | VkConstantBufferObj(VkDeviceObj *device); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 239 | VkConstantBufferObj(VkDeviceObj *device, int constantCount, |
| 240 | int constantSize, const void *data); |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 241 | ~VkConstantBufferObj(); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 242 | void BufferMemoryBarrier( |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 243 | VkFlags srcAccessMask = VK_ACCESS_HOST_WRITE_BIT | |
| 244 | VK_ACCESS_SHADER_WRITE_BIT | |
| 245 | VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT | |
| 246 | VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_WRITE_BIT | |
| 247 | VK_ACCESS_TRANSFER_WRITE_BIT, |
| 248 | VkFlags dstAccessMask = VK_ACCESS_HOST_READ_BIT | |
| 249 | VK_ACCESS_INDIRECT_COMMAND_READ_BIT | |
| 250 | VK_ACCESS_INDEX_READ_BIT | |
| 251 | VK_ACCESS_VERTEX_ATTRIBUTE_READ_BIT | |
| 252 | VK_ACCESS_UNIFORM_READ_BIT | |
| 253 | VK_ACCESS_SHADER_READ_BIT | |
| 254 | VK_ACCESS_COLOR_ATTACHMENT_READ_BIT | |
| 255 | VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_READ_BIT | |
| 256 | VK_ACCESS_MEMORY_READ_BIT); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 257 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 258 | void Bind(VkCommandBuffer commandBuffer, VkDeviceSize offset, |
| 259 | uint32_t binding); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 260 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 261 | VkDescriptorBufferInfo m_descriptorBufferInfo; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 262 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 263 | protected: |
| 264 | VkDeviceObj *m_device; |
| 265 | vk_testing::BufferView m_bufferView; |
| 266 | int m_numVertices; |
| 267 | int m_stride; |
| 268 | vk_testing::CommandPool *m_commandPool; |
| 269 | VkCommandBufferObj *m_commandBuffer; |
| 270 | vk_testing::Fence m_fence; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 271 | }; |
| 272 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 273 | class VkIndexBufferObj : public VkConstantBufferObj { |
| 274 | public: |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 275 | VkIndexBufferObj(VkDeviceObj *device); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 276 | void CreateAndInitBuffer(int numIndexes, VkIndexType dataFormat, |
| 277 | const void *data); |
Chia-I Wu | 3432a0c | 2015-10-27 18:04:07 +0800 | [diff] [blame] | 278 | void Bind(VkCommandBuffer commandBuffer, VkDeviceSize offset); |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 279 | VkIndexType GetIndexType(); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 280 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 281 | protected: |
| 282 | VkIndexType m_indexType; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 283 | }; |
| 284 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 285 | class VkImageObj : public vk_testing::Image { |
| 286 | public: |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 287 | VkImageObj(VkDeviceObj *dev); |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 288 | bool IsCompatible(VkFlags usage, VkFlags features); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 289 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 290 | public: |
| 291 | void init(uint32_t w, uint32_t h, VkFormat fmt, VkFlags usage, |
| 292 | VkImageTiling tiling = VK_IMAGE_TILING_LINEAR, |
| 293 | VkMemoryPropertyFlags reqs = 0); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 294 | |
| 295 | // void clear( CommandBuffer*, uint32_t[4] ); |
| 296 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 297 | void layout(VkImageLayout layout) { |
Courtney Goeltzenleuchter | 4cb6d92 | 2015-10-23 13:38:14 -0600 | [diff] [blame] | 298 | m_descriptorImageInfo.imageLayout = layout; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 299 | } |
| 300 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 301 | VkDeviceMemory memory() const { return Image::memory().handle(); } |
Chia-I Wu | 681d7a0 | 2015-07-03 13:44:34 +0800 | [diff] [blame] | 302 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 303 | void *MapMemory() { return Image::memory().map(); } |
Chia-I Wu | 681d7a0 | 2015-07-03 13:44:34 +0800 | [diff] [blame] | 304 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 305 | void UnmapMemory() { Image::memory().unmap(); } |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 306 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 307 | void ImageMemoryBarrier(VkCommandBufferObj *cmd, VkImageAspectFlags aspect, |
| 308 | VkFlags output_mask, VkFlags input_mask, |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 309 | VkImageLayout image_layout); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 310 | |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 311 | VkResult CopyImage(VkImageObj &src_image); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 312 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 313 | VkImage image() const { return handle(); } |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 314 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 315 | VkImageView targetView(VkFormat format) { |
| 316 | if (!m_targetView.initialized()) { |
Courtney Goeltzenleuchter | 5861a1b | 2015-09-01 17:30:39 -0600 | [diff] [blame] | 317 | VkImageViewCreateInfo createView = {}; |
| 318 | createView.sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO; |
| 319 | createView.image = handle(); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 320 | createView.viewType = VK_IMAGE_VIEW_TYPE_2D; |
Courtney Goeltzenleuchter | 5861a1b | 2015-09-01 17:30:39 -0600 | [diff] [blame] | 321 | createView.format = format; |
Chia-I Wu | ab83a0e | 2015-10-27 19:00:15 +0800 | [diff] [blame] | 322 | createView.components.r = VK_COMPONENT_SWIZZLE_R; |
| 323 | createView.components.g = VK_COMPONENT_SWIZZLE_G; |
| 324 | createView.components.b = VK_COMPONENT_SWIZZLE_B; |
| 325 | createView.components.a = VK_COMPONENT_SWIZZLE_A; |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 326 | createView.subresourceRange = {VK_IMAGE_ASPECT_COLOR_BIT, 0, 1, 0, |
| 327 | 1}; |
Courtney Goeltzenleuchter | 5861a1b | 2015-09-01 17:30:39 -0600 | [diff] [blame] | 328 | createView.flags = 0; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 329 | m_targetView.init(*m_device, createView); |
| 330 | } |
Chia-I Wu | 3158bf3 | 2015-07-03 11:49:42 +0800 | [diff] [blame] | 331 | return m_targetView.handle(); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 332 | } |
| 333 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 334 | void SetLayout(VkCommandBufferObj *cmd_buf, VkImageAspectFlagBits aspect, |
| 335 | VkImageLayout image_layout); |
Courtney Goeltzenleuchter | 908e767 | 2015-10-21 17:00:51 -0600 | [diff] [blame] | 336 | void SetLayout(VkImageAspectFlagBits aspect, VkImageLayout image_layout); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 337 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 338 | VkImageLayout layout() const { return m_descriptorImageInfo.imageLayout; } |
| 339 | uint32_t width() const { return extent().width; } |
| 340 | uint32_t height() const { return extent().height; } |
| 341 | VkDeviceObj *device() const { return m_device; } |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 342 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 343 | protected: |
| 344 | VkDeviceObj *m_device; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 345 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 346 | vk_testing::ImageView m_targetView; |
| 347 | VkDescriptorImageInfo m_descriptorImageInfo; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 348 | }; |
| 349 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 350 | class VkTextureObj : public VkImageObj { |
| 351 | public: |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 352 | VkTextureObj(VkDeviceObj *device, uint32_t *colors = NULL); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 353 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 354 | VkDescriptorImageInfo m_imageInfo; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 355 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 356 | protected: |
| 357 | VkDeviceObj *m_device; |
| 358 | vk_testing::ImageView m_textureView; |
| 359 | VkDeviceSize m_rowPitch; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 360 | }; |
| 361 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 362 | class VkSamplerObj : public vk_testing::Sampler { |
| 363 | public: |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 364 | VkSamplerObj(VkDeviceObj *device); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 365 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 366 | protected: |
| 367 | VkDeviceObj *m_device; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 368 | }; |
| 369 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 370 | class VkDescriptorSetObj : public vk_testing::DescriptorPool { |
| 371 | public: |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 372 | VkDescriptorSetObj(VkDeviceObj *device); |
| 373 | ~VkDescriptorSetObj(); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 374 | |
| 375 | int AppendDummy(); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 376 | int AppendBuffer(VkDescriptorType type, |
| 377 | VkConstantBufferObj &constantBuffer); |
| 378 | int AppendSamplerTexture(VkSamplerObj *sampler, VkTextureObj *texture); |
Chia-I Wu | 3432a0c | 2015-10-27 18:04:07 +0800 | [diff] [blame] | 379 | void CreateVKDescriptorSet(VkCommandBufferObj *commandBuffer); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 380 | |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 381 | VkDescriptorSet GetDescriptorSetHandle() const; |
Mark Lobodzinski | 0fadf5f | 2015-04-17 14:11:39 -0500 | [diff] [blame] | 382 | VkPipelineLayout GetPipelineLayout() const; |
Tony Barbour | 26aa42b | 2016-03-02 15:35:27 -0700 | [diff] [blame] | 383 | int GetTypeCounts() {return m_type_counts.size();} |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 384 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 385 | protected: |
| 386 | VkDeviceObj *m_device; |
| 387 | vector<VkDescriptorPoolSize> m_type_counts; |
| 388 | int m_nextSlot; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 389 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 390 | vector<VkDescriptorImageInfo> m_imageSamplerDescriptors; |
| 391 | vector<VkWriteDescriptorSet> m_writes; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 392 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 393 | vk_testing::DescriptorSetLayout m_layout; |
| 394 | vk_testing::PipelineLayout m_pipeline_layout; |
Tony Barbour | 26aa42b | 2016-03-02 15:35:27 -0700 | [diff] [blame] | 395 | vk_testing::DescriptorSet *m_set = NULL; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 396 | }; |
| 397 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 398 | class VkShaderObj : public vk_testing::ShaderModule { |
| 399 | public: |
| 400 | VkShaderObj(VkDeviceObj *device, const char *shaderText, |
Chris Forbes | 4f9b0e6 | 2015-12-30 11:49:04 +1300 | [diff] [blame] | 401 | VkShaderStageFlagBits stage, VkRenderFramework *framework, |
| 402 | char const *name = "main"); |
Chia-I Wu | 28e0691 | 2015-10-31 00:31:16 +0800 | [diff] [blame] | 403 | VkPipelineShaderStageCreateInfo GetStageCreateInfo() const; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 404 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 405 | protected: |
| 406 | VkPipelineShaderStageCreateInfo stage_info; |
| 407 | VkShaderStageFlagBits m_stage; |
Chris Forbes | 4f9b0e6 | 2015-12-30 11:49:04 +1300 | [diff] [blame] | 408 | char const *m_name; |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 409 | VkDeviceObj *m_device; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 410 | }; |
| 411 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 412 | class VkPipelineObj : public vk_testing::Pipeline { |
| 413 | public: |
Tony Barbour | 6918cd5 | 2015-04-09 12:58:51 -0600 | [diff] [blame] | 414 | VkPipelineObj(VkDeviceObj *device); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 415 | void AddShader(VkShaderObj *shaderObj); |
| 416 | void AddVertexInputAttribs(VkVertexInputAttributeDescription *vi_attrib, |
| 417 | int count); |
| 418 | void AddVertexInputBindings(VkVertexInputBindingDescription *vi_binding, |
| 419 | int count); |
| 420 | void AddColorAttachment(uint32_t binding, |
| 421 | const VkPipelineColorBlendAttachmentState *att); |
Courtney Goeltzenleuchter | 507ba97 | 2015-09-30 16:16:57 -0600 | [diff] [blame] | 422 | void MakeDynamic(VkDynamicState state); |
Chia-I Wu | 08accc6 | 2015-07-07 11:50:03 +0800 | [diff] [blame] | 423 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 424 | void AddColorAttachment() { |
Tony Barbour | dd6e32e | 2015-07-10 15:29:03 -0600 | [diff] [blame] | 425 | VkPipelineColorBlendAttachmentState att = {}; |
Chia-I Wu | 08accc6 | 2015-07-07 11:50:03 +0800 | [diff] [blame] | 426 | att.blendEnable = VK_FALSE; |
Chia-I Wu | 1b99bb2 | 2015-10-27 19:25:11 +0800 | [diff] [blame] | 427 | att.colorWriteMask = 0xf; |
Chia-I Wu | 08accc6 | 2015-07-07 11:50:03 +0800 | [diff] [blame] | 428 | AddColorAttachment(0, &att); |
| 429 | } |
| 430 | |
Tony Barbour | dd6e32e | 2015-07-10 15:29:03 -0600 | [diff] [blame] | 431 | void SetDepthStencil(VkPipelineDepthStencilStateCreateInfo *); |
Tony Barbour | 0ec8cd5 | 2015-08-06 09:27:11 -0600 | [diff] [blame] | 432 | void SetMSAA(VkPipelineMultisampleStateCreateInfo *ms_state); |
Tobin Ehlis | d332f28 | 2015-10-02 11:00:56 -0600 | [diff] [blame] | 433 | void SetViewport(vector<VkViewport> viewports); |
| 434 | void SetScissor(vector<VkRect2D> scissors); |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 435 | VkResult CreateVKPipeline(VkPipelineLayout layout, |
| 436 | VkRenderPass render_pass); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 437 | |
Karl Schultz | 6addd81 | 2016-02-02 17:17:23 -0700 | [diff] [blame] | 438 | protected: |
| 439 | VkPipelineVertexInputStateCreateInfo m_vi_state; |
| 440 | VkPipelineInputAssemblyStateCreateInfo m_ia_state; |
| 441 | VkPipelineRasterizationStateCreateInfo m_rs_state; |
| 442 | VkPipelineColorBlendStateCreateInfo m_cb_state; |
| 443 | VkPipelineDepthStencilStateCreateInfo m_ds_state; |
| 444 | VkPipelineViewportStateCreateInfo m_vp_state; |
| 445 | VkPipelineMultisampleStateCreateInfo m_ms_state; |
| 446 | vector<VkDynamicState> m_dynamic_state_enables; |
| 447 | vector<VkViewport> m_viewports; |
| 448 | vector<VkRect2D> m_scissors; |
| 449 | VkDeviceObj *m_device; |
| 450 | vector<VkShaderObj *> m_shaderObjs; |
| 451 | vector<int> m_vertexBufferBindings; |
| 452 | vector<VkPipelineColorBlendAttachmentState> m_colorAttachments; |
| 453 | int m_vertexBufferCount; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 454 | }; |
| 455 | |
| 456 | #endif // VKRENDERFRAMEWORK_H |