source/backends/vulkan/pipelinestate_backend.cpp

   1 #include <msp/core/hash.h>
   2 #include "batch.h"
   3 #include "blend.h"
   4 #include "buffer.h"
   5 #include "depthtest.h"
   6 #include "device.h"
   7 #include "framebuffer.h"
   8 #include "pipelinestate.h"
   9 #include "pipelinestate_backend.h"
  10 #include "program.h"
  11 #include "rect.h"
  12 #include "sampler.h"
  13 #include "stenciltest.h"
  14 #include "structurebuilder.h"
  15 #include "texture.h"
  16 #include "uniformblock.h"
  17 #include "vertexsetup.h"
  18 #include "vulkan.h"
  19
  20 using namespace std;
  21
  22 namespace Msp {
  23 namespace GL {
  24
  25 VulkanPipelineState::VulkanPipelineState():
  26         device(Device::get_current())
  27 { }
  28
  29 VulkanPipelineState::VulkanPipelineState(VulkanPipelineState &&other):
  30         device(other.device),
  31         handle(other.handle)
  32 { }
  33
  34 void VulkanPipelineState::update() const
  35 {
  36         const PipelineState &self = *static_cast<const PipelineState *>(this);
  37
  38         if(self.changes&PipelineState::VERTEX_SETUP)
  39                 self.vertex_setup->refresh();
  40
  41         constexpr unsigned pipeline_mask = PipelineState::SHPROG|PipelineState::VERTEX_SETUP|PipelineState::FACE_CULL|
  42                 PipelineState::DEPTH_TEST|PipelineState::STENCIL_TEST|PipelineState::BLEND|PipelineState::PRIMITIVE_TYPE;
  43         if(self.changes&pipeline_mask)
  44                 handle = device.get_pipeline_cache().get_pipeline(self);
  45
  46         if(self.changes&(PipelineState::SHPROG|PipelineState::UNIFORMS|PipelineState::TEXTURES))
  47         {
  48                 unsigned changed_sets = (self.changes&PipelineState::SHPROG ? ~0U : 0U);
  49                 for(const PipelineState::BoundUniformBlock &u: self.uniform_blocks)
  50                         if(u.changed || changed_sets==~0U)
  51                         {
  52                                 if(u.block)
  53                                         u.used = self.shprog->uses_uniform_block_binding(u.binding);
  54                                 if(u.binding>=0)
  55                                         changed_sets |= 1<<(u.binding>>20);
  56                                 u.changed = false;
  57                         }
  58                 for(const PipelineState::BoundTexture &t: self.textures)
  59                         if(t.changed || changed_sets==~0U)
  60                         {
  61                                 if(t.texture && t.sampler)
  62                                         t.used = self.shprog->uses_texture_binding(t.binding);
  63                                 changed_sets |= 1<<(t.binding>>20);
  64                                 if(t.texture && t.level>=0)
  65                                         t.texture->refresh_mip_views();
  66                                 if(t.sampler)
  67                                         t.sampler->refresh();
  68                                 t.changed = false;
  69                         }
  70
  71                 descriptor_set_handles.resize(self.shprog->get_n_descriptor_sets());
  72                 for(unsigned i=0; i<descriptor_set_handles.size(); ++i)
  73                         if(changed_sets&(1<<i))
  74                                 descriptor_set_handles[i] = device.get_pipeline_cache().get_descriptor_set(self, i);
  75         }
  76
  77         self.changes = 0;
  78 }
  79
  80 uint64_t VulkanPipelineState::compute_hash() const
  81 {
  82         const PipelineState &self = *static_cast<const PipelineState *>(this);
  83         const FrameFormat &format = self.framebuffer->get_format();
  84
  85         uint64_t result = hash<64>(self.shprog);
  86         result = hash_update<64>(result, self.vertex_setup->compute_hash());
  87         result = hash_round<64>(result, self.primitive_type);
  88
  89         if(self.front_face!=NON_MANIFOLD && self.face_cull!=NO_CULL)
  90         {
  91                 result = hash_round<64>(result, self.front_face);
  92                 result = hash_round<64>(result, self.face_cull);
  93         }
  94
  95         result = hash_round<64>(result, format.get_samples());
  96
  97         if(const DepthTest *depth_test = self.depth_test)
  98                 if(depth_test->enabled)
  99                 {
 100                         result = hash_round<64>(result, depth_test->compare);
 101                         result = hash_update<64>(result, depth_test->write);
 102                 }
 103
 104         if(const StencilTest *stencil_test = self.stencil_test)
 105                 if(stencil_test->enabled)
 106                 {
 107                         result = hash_round<64>(result, stencil_test->compare);
 108                         result = hash_round<64>(result, stencil_test->stencil_fail_op);
 109                         result = hash_round<64>(result, stencil_test->depth_fail_op);
 110                         result = hash_round<64>(result, stencil_test->depth_pass_op);
 111                         result = hash_update<64>(result, stencil_test->reference);
 112                 }
 113
 114         if(const Blend *blend = self.blend)
 115                 if(blend->enabled)
 116                 {
 117                         result = hash_round<64>(result, blend->equation);
 118                         result = hash_round<64>(result, blend->src_factor);
 119                         result = hash_round<64>(result, blend->dst_factor);
 120                         result = hash_round<64>(result, blend->write_mask);
 121                 }
 122
 123         for(FrameAttachment a: format)
 124                 result = hash_update<64>(result, a);
 125
 126         return result;
 127 }
 128
 129 void VulkanPipelineState::fill_creation_info(vector<char> &buffer) const
 130 {
 131         const PipelineState &self = *static_cast<const PipelineState *>(this);
 132
 133         const FrameFormat &format = self.framebuffer->get_format();
 134         VkRenderPass render_pass = device.get_pipeline_cache().get_render_pass(format, false, false, false);
 135
 136         unsigned n_color_attachments = 0;
 137         for(FrameAttachment a: format)
 138         {
 139                 unsigned attach_pt = get_attach_point(a);
 140                 if(attach_pt!=get_attach_point(DEPTH_ATTACHMENT) && attach_pt!=get_attach_point(STENCIL_ATTACHMENT))
 141                         ++n_color_attachments;
 142         }
 143
 144         StructureBuilder sb(buffer, 10);
 145         VkGraphicsPipelineCreateInfo *&pipeline_info = sb.add<VkGraphicsPipelineCreateInfo>();
 146         VkPipelineInputAssemblyStateCreateInfo *&input_assembly_info = sb.add<VkPipelineInputAssemblyStateCreateInfo>();
 147         VkPipelineViewportStateCreateInfo *&viewport_info = sb.add<VkPipelineViewportStateCreateInfo>();
 148         VkPipelineRasterizationStateCreateInfo *&raster_info = sb.add<VkPipelineRasterizationStateCreateInfo>();
 149         VkPipelineMultisampleStateCreateInfo *&multisample_info = sb.add<VkPipelineMultisampleStateCreateInfo>();
 150         VkPipelineDepthStencilStateCreateInfo *&depth_stencil_info = sb.add<VkPipelineDepthStencilStateCreateInfo>();
 151         VkPipelineColorBlendStateCreateInfo *&blend_info = sb.add<VkPipelineColorBlendStateCreateInfo>();
 152         VkPipelineColorBlendAttachmentState *&blend_attachments = sb.add<VkPipelineColorBlendAttachmentState>(n_color_attachments);
 153         VkPipelineDynamicStateCreateInfo *&dynamic_info = sb.add<VkPipelineDynamicStateCreateInfo>();
 154         VkDynamicState *&dynamic_states = sb.add<VkDynamicState>(2);
 155
 156         input_assembly_info->sType = VK_STRUCTURE_TYPE_PIPELINE_INPUT_ASSEMBLY_STATE_CREATE_INFO;
 157         input_assembly_info->topology = static_cast<VkPrimitiveTopology>(get_vulkan_primitive_type(self.primitive_type));
 158         input_assembly_info->primitiveRestartEnable = true;
 159
 160         viewport_info->sType = VK_STRUCTURE_TYPE_PIPELINE_VIEWPORT_STATE_CREATE_INFO;
 161         viewport_info->viewportCount = 1;
 162         viewport_info->pViewports = 0;
 163         viewport_info->scissorCount = 1;
 164         viewport_info->pScissors = 0;
 165
 166         raster_info->sType = VK_STRUCTURE_TYPE_PIPELINE_RASTERIZATION_STATE_CREATE_INFO;
 167         raster_info->depthClampEnable = VK_FALSE;
 168         raster_info->rasterizerDiscardEnable = VK_FALSE;
 169         raster_info->polygonMode = VK_POLYGON_MODE_FILL;
 170         if(self.face_cull==NO_CULL || self.front_face==NON_MANIFOLD)
 171         {
 172                 raster_info->cullMode = VK_CULL_MODE_NONE;
 173                 raster_info->frontFace = VK_FRONT_FACE_COUNTER_CLOCKWISE;
 174         }
 175         else
 176         {
 177                 raster_info->cullMode = (self.face_cull==CULL_FRONT ? VK_CULL_MODE_FRONT_BIT : VK_CULL_MODE_BACK_BIT);
 178                 raster_info->frontFace = (self.front_face==CLOCKWISE ? VK_FRONT_FACE_CLOCKWISE : VK_FRONT_FACE_COUNTER_CLOCKWISE);
 179         }
 180         raster_info->depthBiasEnable = VK_FALSE;
 181         raster_info->depthBiasConstantFactor = 0.0f;
 182         raster_info->depthBiasClamp = 0.0f;
 183         raster_info->depthBiasSlopeFactor = 0.0f;
 184         raster_info->lineWidth = 1.0f;
 185
 186         multisample_info->sType = VK_STRUCTURE_TYPE_PIPELINE_MULTISAMPLE_STATE_CREATE_INFO;
 187         multisample_info->rasterizationSamples = static_cast<VkSampleCountFlagBits>(get_vulkan_samples(format.get_samples()));
 188         multisample_info->sampleShadingEnable = VK_FALSE;
 189         multisample_info->minSampleShading = 1.0f;
 190         multisample_info->pSampleMask = 0;
 191         multisample_info->alphaToCoverageEnable = VK_FALSE;
 192         multisample_info->alphaToOneEnable = VK_FALSE;
 193
 194         depth_stencil_info->sType = VK_STRUCTURE_TYPE_PIPELINE_DEPTH_STENCIL_STATE_CREATE_INFO;
 195         if(const DepthTest *depth_test = self.depth_test)
 196         {
 197                 depth_stencil_info->depthTestEnable = depth_test->enabled;
 198                 depth_stencil_info->depthWriteEnable = depth_test->write;
 199                 depth_stencil_info->depthCompareOp = static_cast<VkCompareOp>(get_vulkan_predicate(depth_test->compare));
 200                 depth_stencil_info->depthBoundsTestEnable = VK_FALSE;
 201         }
 202         if(const StencilTest *stencil_test = self.stencil_test)
 203         {
 204                 depth_stencil_info->stencilTestEnable = stencil_test->enabled;
 205                 depth_stencil_info->front.failOp = static_cast<VkStencilOp>(get_vulkan_stencil_op(stencil_test->stencil_fail_op));
 206                 depth_stencil_info->front.passOp = static_cast<VkStencilOp>(get_vulkan_stencil_op(stencil_test->depth_pass_op));
 207                 depth_stencil_info->front.depthFailOp = static_cast<VkStencilOp>(get_vulkan_stencil_op(stencil_test->depth_fail_op));
 208                 depth_stencil_info->front.compareOp = static_cast<VkCompareOp>(get_vulkan_predicate(stencil_test->compare));
 209                 depth_stencil_info->front.compareMask = 0xFFFFFFFFU;
 210                 depth_stencil_info->front.writeMask = 0xFFFFFFFFU;
 211                 depth_stencil_info->front.reference = stencil_test->reference;
 212                 depth_stencil_info->back = depth_stencil_info->front;
 213         }
 214
 215         if(const Blend *blend = self.blend)
 216         {
 217                 for(unsigned i=0; i<n_color_attachments; ++i)
 218                 {
 219                         blend_attachments[i].blendEnable = blend->enabled;
 220                         blend_attachments[i].srcColorBlendFactor = static_cast<VkBlendFactor>(get_vulkan_blend_factor(blend->src_factor));
 221                         blend_attachments[i].dstColorBlendFactor = static_cast<VkBlendFactor>(get_vulkan_blend_factor(blend->dst_factor));
 222                         blend_attachments[i].colorBlendOp = static_cast<VkBlendOp>(get_vulkan_blend_equation(blend->equation));
 223                         blend_attachments[i].srcAlphaBlendFactor = blend_attachments[i].srcColorBlendFactor;
 224                         blend_attachments[i].dstAlphaBlendFactor = blend_attachments[i].dstColorBlendFactor;
 225                         blend_attachments[i].alphaBlendOp = blend_attachments[i].colorBlendOp;
 226                         blend_attachments[i].colorWriteMask = get_vulkan_color_mask(blend->write_mask);
 227                 }
 228         }
 229         else
 230         {
 231                 for(unsigned i=0; i<n_color_attachments; ++i)
 232                         blend_attachments[i].colorWriteMask = VK_COLOR_COMPONENT_R_BIT|VK_COLOR_COMPONENT_G_BIT|VK_COLOR_COMPONENT_B_BIT|VK_COLOR_COMPONENT_A_BIT;
 233         }
 234
 235         blend_info->sType = VK_STRUCTURE_TYPE_PIPELINE_COLOR_BLEND_STATE_CREATE_INFO;
 236         blend_info->attachmentCount = n_color_attachments;
 237         blend_info->pAttachments = blend_attachments;
 238
 239         dynamic_states[0] = VK_DYNAMIC_STATE_VIEWPORT;
 240         dynamic_states[1] = VK_DYNAMIC_STATE_SCISSOR;
 241
 242         dynamic_info->sType = VK_STRUCTURE_TYPE_PIPELINE_DYNAMIC_STATE_CREATE_INFO;
 243         dynamic_info->dynamicStateCount = 2;
 244         dynamic_info->pDynamicStates = dynamic_states;
 245
 246         pipeline_info->sType = VK_STRUCTURE_TYPE_GRAPHICS_PIPELINE_CREATE_INFO;
 247
 248         pipeline_info->pInputAssemblyState = input_assembly_info;
 249         pipeline_info->pTessellationState = 0;
 250         pipeline_info->pViewportState = viewport_info;
 251         pipeline_info->pRasterizationState = raster_info;
 252         pipeline_info->pMultisampleState = multisample_info;
 253         pipeline_info->pDepthStencilState = depth_stencil_info;
 254         pipeline_info->pColorBlendState = blend_info;
 255         pipeline_info->pDynamicState = dynamic_info;
 256         pipeline_info->renderPass = handle_cast<::VkRenderPass>(render_pass);
 257         pipeline_info->subpass = 0;
 258
 259         if(self.shprog)
 260         {
 261                 pipeline_info->stageCount = self.shprog->n_stages;
 262                 pipeline_info->pStages = reinterpret_cast<const VkPipelineShaderStageCreateInfo *>(self.shprog->creation_info.data());
 263                 pipeline_info->layout = handle_cast<::VkPipelineLayout>(self.shprog->layout_handle);
 264         }
 265
 266         if(self.vertex_setup)
 267                 pipeline_info->pVertexInputState = reinterpret_cast<const VkPipelineVertexInputStateCreateInfo *>(self.vertex_setup->creation_info.data());
 268 }
 269
 270 uint64_t VulkanPipelineState::compute_descriptor_set_hash(unsigned index) const
 271 {
 272         const PipelineState &self = *static_cast<const PipelineState *>(this);
 273
 274         uint64_t result = hash<64>(0, 0);
 275         for(const PipelineState::BoundUniformBlock &b: self.uniform_blocks)
 276                 if(b.used && b.binding>=0 && static_cast<unsigned>(b.binding>>20)==index)
 277                 {
 278                         result = hash_update<64>(result, b.binding);
 279                         result = hash_update<64>(result, reinterpret_cast<uintptr_t>(b.block));
 280                         result = hash_update<64>(result, reinterpret_cast<uintptr_t>(b.buffer->handle));
 281                 }
 282         for(const PipelineState::BoundTexture &t: self.textures)
 283                 if(t.used && (t.binding>>20)==index)
 284                 {
 285                         result = hash_update<64>(result, t.binding);
 286                         result = hash_update<64>(result, reinterpret_cast<uintptr_t>(t.texture->handle));
 287                         result = hash_update<64>(result, reinterpret_cast<uintptr_t>(t.sampler->handle));
 288                         result = hash_update<64>(result, t.level);
 289                 }
 290
 291         return result;
 292 }
 293
 294 VkDescriptorSetLayout VulkanPipelineState::get_descriptor_set_layout(unsigned index) const
 295 {
 296         return static_cast<const PipelineState *>(this)->shprog->desc_set_layout_handles[index];
 297 }
 298
 299 unsigned VulkanPipelineState::fill_descriptor_writes(unsigned index, vector<char> &buffer) const
 300 {
 301         const PipelineState &self = *static_cast<const PipelineState *>(this);
 302
 303         unsigned n_buffers = 0;
 304         for(const PipelineState::BoundUniformBlock &u: self.uniform_blocks)
 305                 if(u.used && u.binding>=0 && static_cast<unsigned>(u.binding>>20)==index)
 306                         ++n_buffers;
 307         unsigned n_images = 0;
 308         for(const PipelineState::BoundTexture &t: self.textures)
 309                 if(t.used && (t.binding>>20)==index)
 310                         ++n_images;
 311         unsigned n_writes = n_buffers+n_images;
 312
 313         StructureBuilder sb(buffer, 3);
 314         VkWriteDescriptorSet *&writes = sb.add<VkWriteDescriptorSet>(n_writes);
 315         VkDescriptorBufferInfo *&buffers = sb.add<VkDescriptorBufferInfo>(n_buffers);
 316         VkDescriptorImageInfo *&images = sb.add<VkDescriptorImageInfo>(n_images);
 317
 318         VkWriteDescriptorSet *write_ptr = writes;
 319         VkDescriptorBufferInfo *buffer_ptr = buffers;
 320         VkDescriptorImageInfo *image_ptr = images;
 321
 322         for(const PipelineState::BoundUniformBlock &u: self.uniform_blocks)
 323                 if(u.used && u.binding>=0 && static_cast<unsigned>(u.binding>>20)==index)
 324                 {
 325                         buffer_ptr->buffer = handle_cast<::VkBuffer>(u.buffer->handle);
 326                         buffer_ptr->offset = u.block->get_offset();
 327                         buffer_ptr->range = u.block->get_data_size();
 328
 329                         write_ptr->sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET;
 330                         write_ptr->dstBinding = u.binding&0xFFFFF;
 331                         write_ptr->descriptorCount = 1;
 332                         write_ptr->descriptorType = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC;
 333                         write_ptr->pBufferInfo = buffer_ptr;
 334
 335                         ++buffer_ptr;
 336                         ++write_ptr;
 337                 }
 338
 339         for(const PipelineState::BoundTexture &t: self.textures)
 340                 if(t.used && (t.binding>>20)==index)
 341                 {
 342                         image_ptr->sampler = handle_cast<::VkSampler>(t.sampler->handle);
 343                         if(t.level<0)
 344                                 image_ptr->imageView = handle_cast<::VkImageView>(t.texture->view_handle);
 345                         else
 346                                 image_ptr->imageView = handle_cast<::VkImageView>(t.texture->mip_view_handles[t.level]);
 347                         image_ptr->imageLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;
 348
 349                         write_ptr->sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET;
 350                         write_ptr->dstBinding = t.binding&0xFFFFF;
 351                         write_ptr->descriptorCount = 1;
 352                         write_ptr->descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER;
 353                         write_ptr->pImageInfo = image_ptr;
 354
 355                         ++image_ptr;
 356                         ++write_ptr;
 357                 }
 358
 359         return n_writes;
 360 }
 361
 362 void VulkanPipelineState::apply(VkCommandBuffer command_buffer, unsigned frame, bool negative_viewport) const
 363 {
 364         const PipelineState &self = *static_cast<const PipelineState *>(this);
 365         const VulkanFunctions &vk = device.get_functions();
 366
 367         vk.CmdBindPipeline(command_buffer, VK_PIPELINE_BIND_POINT_GRAPHICS, handle);
 368         if(const VertexSetup *vs = self.vertex_setup)
 369         {
 370                 vk.CmdBindVertexBuffers(command_buffer, 0, vs->n_bindings, vs->buffers, vs->offsets);
 371                 VkIndexType index_type = static_cast<VkIndexType>(get_vulkan_index_type(vs->get_index_type()));
 372                 vk.CmdBindIndexBuffer(command_buffer, vs->get_index_buffer()->handle, 0, index_type);
 373         }
 374
 375         if(!self.uniform_blocks.empty())
 376         {
 377                 const PipelineState::BoundUniformBlock &first_block = self.uniform_blocks.front();
 378                 if(first_block.used && first_block.binding==ReflectData::PUSH_CONSTANT)
 379                 {
 380                         const UniformBlock &pc_block = *first_block.block;
 381                         vk.CmdPushConstants(command_buffer, self.shprog->layout_handle, self.shprog->stage_flags,
 382                                 pc_block.get_offset(), pc_block.get_data_size(), pc_block.get_data_pointer());
 383                 }
 384         }
 385
 386         if(!descriptor_set_handles.empty())
 387         {
 388                 vector<uint32_t> dynamic_offsets;
 389                 dynamic_offsets.reserve(self.uniform_blocks.size());
 390                 for(const PipelineState::BoundUniformBlock &u: self.uniform_blocks)
 391                         if(u.used && u.binding>=0)
 392                         {
 393                                 if(u.buffer->get_usage()==STREAMING)
 394                                         dynamic_offsets.push_back(frame*u.buffer->get_size());
 395                                 else
 396                                         dynamic_offsets.push_back(0);
 397                         }
 398
 399                 vk.CmdBindDescriptorSets(command_buffer, VK_PIPELINE_BIND_POINT_GRAPHICS, self.shprog->layout_handle,
 400                         0, descriptor_set_handles.size(), descriptor_set_handles.data(), dynamic_offsets.size(), dynamic_offsets.data());
 401         }
 402
 403         VkViewport viewport = { };
 404         if(self.viewport)
 405         {
 406                 viewport.x = self.viewport->left;
 407                 viewport.y = self.viewport->bottom;
 408                 viewport.width = self.viewport->width;
 409                 viewport.height = self.viewport->height;
 410         }
 411         else
 412         {
 413                 viewport.x = 0;
 414                 viewport.y = 0;
 415                 viewport.width = self.framebuffer->get_width();
 416                 viewport.height = self.framebuffer->get_height();
 417         }
 418         if(negative_viewport)
 419         {
 420                 viewport.y += viewport.height;
 421                 viewport.height = -viewport.height;
 422         }
 423         viewport.minDepth = 0.0f;
 424         viewport.maxDepth = 1.0f;
 425         vk.CmdSetViewport(command_buffer, 0, 1, &viewport);
 426
 427         VkRect2D scissor = { };
 428         if(self.scissor)
 429         {
 430                 scissor.offset.x = self.scissor->left;
 431                 scissor.offset.y = self.scissor->bottom;
 432                 scissor.extent.width = self.scissor->width;
 433                 scissor.extent.height = self.scissor->height;
 434         }
 435         else
 436         {
 437                 scissor.offset.x = 0;
 438                 scissor.offset.y = 0;
 439                 scissor.extent.width = self.framebuffer->get_width();
 440                 scissor.extent.height = self.framebuffer->get_height();
 441         }
 442         vk.CmdSetScissor(command_buffer, 0, 1, &scissor);
 443 }
 444
 445 } // namespace GL
 446 } // namespace Msp