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