jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2016 Google Inc. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
| 7 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 8 | #include "include/gpu/GrContext.h" |
| 9 | #include "src/gpu/GrContextPriv.h" |
| 10 | #include "src/gpu/GrPersistentCacheUtils.h" |
| 11 | #include "src/gpu/GrShaderCaps.h" |
Brian Osman | ac9be9d | 2019-05-01 10:29:34 -0400 | [diff] [blame^] | 12 | #include "src/gpu/GrShaderUtils.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 13 | #include "src/gpu/GrStencilSettings.h" |
| 14 | #include "src/gpu/vk/GrVkDescriptorSetManager.h" |
| 15 | #include "src/gpu/vk/GrVkGpu.h" |
| 16 | #include "src/gpu/vk/GrVkPipelineStateBuilder.h" |
| 17 | #include "src/gpu/vk/GrVkRenderPass.h" |
| 18 | #include "src/gpu/vk/GrVkRenderTarget.h" |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 19 | |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 20 | typedef size_t shader_size; |
| 21 | |
egdaniel | 22281c1 | 2016-03-23 13:49:40 -0700 | [diff] [blame] | 22 | GrVkPipelineState* GrVkPipelineStateBuilder::CreatePipelineState( |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 23 | GrVkGpu* gpu, |
Robert Phillips | d0fe875 | 2019-01-31 14:13:59 -0500 | [diff] [blame] | 24 | GrRenderTarget* renderTarget, GrSurfaceOrigin origin, |
Brian Salomon | ff168d9 | 2018-06-23 15:17:27 -0400 | [diff] [blame] | 25 | const GrPrimitiveProcessor& primProc, |
Greg Daniel | 9a51a86 | 2018-11-30 10:18:14 -0500 | [diff] [blame] | 26 | const GrTextureProxy* const primProcProxies[], |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 27 | const GrPipeline& pipeline, |
| 28 | const GrStencilSettings& stencil, |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 29 | GrPrimitiveType primitiveType, |
| 30 | Desc* desc, |
Greg Daniel | 99b88e0 | 2018-10-03 15:31:20 -0400 | [diff] [blame] | 31 | VkRenderPass compatibleRenderPass) { |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 32 | // create a builder. This will be handed off to effects so they can use it to add |
| 33 | // uniforms, varyings, textures, etc |
Robert Phillips | d0fe875 | 2019-01-31 14:13:59 -0500 | [diff] [blame] | 34 | GrVkPipelineStateBuilder builder(gpu, renderTarget, origin, pipeline, primProc, |
| 35 | primProcProxies, desc); |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 36 | |
Ethan Nicholas | 2983f40 | 2017-05-08 09:36:08 -0400 | [diff] [blame] | 37 | if (!builder.emitAndInstallProcs()) { |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 38 | return nullptr; |
| 39 | } |
| 40 | |
Greg Daniel | 99b88e0 | 2018-10-03 15:31:20 -0400 | [diff] [blame] | 41 | return builder.finalize(stencil, primitiveType, compatibleRenderPass, desc); |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 42 | } |
| 43 | |
egdaniel | 22281c1 | 2016-03-23 13:49:40 -0700 | [diff] [blame] | 44 | GrVkPipelineStateBuilder::GrVkPipelineStateBuilder(GrVkGpu* gpu, |
Robert Phillips | d0fe875 | 2019-01-31 14:13:59 -0500 | [diff] [blame] | 45 | GrRenderTarget* renderTarget, |
| 46 | GrSurfaceOrigin origin, |
egdaniel | 22281c1 | 2016-03-23 13:49:40 -0700 | [diff] [blame] | 47 | const GrPipeline& pipeline, |
| 48 | const GrPrimitiveProcessor& primProc, |
Greg Daniel | 9a51a86 | 2018-11-30 10:18:14 -0500 | [diff] [blame] | 49 | const GrTextureProxy* const primProcProxies[], |
Ethan Nicholas | 3865711 | 2017-02-09 17:01:22 -0500 | [diff] [blame] | 50 | GrProgramDesc* desc) |
Robert Phillips | d0fe875 | 2019-01-31 14:13:59 -0500 | [diff] [blame] | 51 | : INHERITED(renderTarget, origin, primProc, primProcProxies, pipeline, desc) |
Brian Salomon | ff168d9 | 2018-06-23 15:17:27 -0400 | [diff] [blame] | 52 | , fGpu(gpu) |
| 53 | , fVaryingHandler(this) |
| 54 | , fUniformHandler(this) {} |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 55 | |
egdaniel | 22281c1 | 2016-03-23 13:49:40 -0700 | [diff] [blame] | 56 | const GrCaps* GrVkPipelineStateBuilder::caps() const { |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 57 | return fGpu->caps(); |
| 58 | } |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 59 | |
Brian Salomon | 99938a8 | 2016-11-21 13:41:08 -0500 | [diff] [blame] | 60 | void GrVkPipelineStateBuilder::finalizeFragmentOutputColor(GrShaderVar& outputColor) { |
Brian Salomon | 6039768 | 2016-11-22 15:06:46 -0500 | [diff] [blame] | 61 | outputColor.addLayoutQualifier("location = 0, index = 0"); |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 62 | } |
| 63 | |
Brian Salomon | 99938a8 | 2016-11-21 13:41:08 -0500 | [diff] [blame] | 64 | void GrVkPipelineStateBuilder::finalizeFragmentSecondaryColor(GrShaderVar& outputColor) { |
Brian Salomon | 6039768 | 2016-11-22 15:06:46 -0500 | [diff] [blame] | 65 | outputColor.addLayoutQualifier("location = 0, index = 1"); |
egdaniel | d632bb4 | 2016-03-30 12:06:48 -0700 | [diff] [blame] | 66 | } |
| 67 | |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 68 | bool GrVkPipelineStateBuilder::createVkShaderModule(VkShaderStageFlagBits stage, |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 69 | const SkSL::String& sksl, |
egdaniel | 22281c1 | 2016-03-23 13:49:40 -0700 | [diff] [blame] | 70 | VkShaderModule* shaderModule, |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 71 | VkPipelineShaderStageCreateInfo* stageInfo, |
Ethan Nicholas | 3865711 | 2017-02-09 17:01:22 -0500 | [diff] [blame] | 72 | const SkSL::Program::Settings& settings, |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 73 | Desc* desc, |
| 74 | SkSL::String* outSPIRV, |
| 75 | SkSL::Program::Inputs* outInputs) { |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 76 | if (!GrCompileVkShaderModule(fGpu, sksl, stage, shaderModule, |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 77 | stageInfo, settings, outSPIRV, outInputs)) { |
| 78 | return false; |
| 79 | } |
| 80 | if (outInputs->fRTHeight) { |
| 81 | this->addRTHeightUniform(SKSL_RTHEIGHT_NAME); |
| 82 | } |
| 83 | if (outInputs->fFlipY) { |
| 84 | desc->setSurfaceOriginKey(GrGLSLFragmentShaderBuilder::KeyForSurfaceOrigin( |
Robert Phillips | d0fe875 | 2019-01-31 14:13:59 -0500 | [diff] [blame] | 85 | this->origin())); |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 86 | } |
| 87 | return true; |
| 88 | } |
| 89 | |
| 90 | bool GrVkPipelineStateBuilder::installVkShaderModule(VkShaderStageFlagBits stage, |
| 91 | const GrGLSLShaderBuilder& builder, |
| 92 | VkShaderModule* shaderModule, |
| 93 | VkPipelineShaderStageCreateInfo* stageInfo, |
| 94 | SkSL::String spirv, |
| 95 | SkSL::Program::Inputs inputs) { |
| 96 | if (!GrInstallVkShaderModule(fGpu, spirv, stage, shaderModule, stageInfo)) { |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 97 | return false; |
| 98 | } |
Greg Daniel | e6ab998 | 2018-08-22 13:56:32 +0000 | [diff] [blame] | 99 | if (inputs.fRTHeight) { |
| 100 | this->addRTHeightUniform(SKSL_RTHEIGHT_NAME); |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 101 | } |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 102 | return true; |
| 103 | } |
| 104 | |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 105 | static constexpr SkFourByteTag kSPIRV_Tag = SkSetFourByteTag('S', 'P', 'R', 'V'); |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 106 | static constexpr SkFourByteTag kSKSL_Tag = SkSetFourByteTag('S', 'K', 'S', 'L'); |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 107 | |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 108 | int GrVkPipelineStateBuilder::loadShadersFromCache(const SkData& cached, |
| 109 | VkShaderModule* outVertShaderModule, |
| 110 | VkShaderModule* outFragShaderModule, |
| 111 | VkShaderModule* outGeomShaderModule, |
| 112 | VkPipelineShaderStageCreateInfo* outStageInfo) { |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 113 | SkSL::String shaders[kGrShaderTypeCount]; |
| 114 | SkSL::Program::Inputs inputs[kGrShaderTypeCount]; |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 115 | |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 116 | if (kSPIRV_Tag != GrPersistentCacheUtils::UnpackCachedShaders(&cached, shaders, |
| 117 | inputs, kGrShaderTypeCount)) { |
| 118 | return 0; |
| 119 | } |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 120 | |
| 121 | SkAssertResult(this->installVkShaderModule(VK_SHADER_STAGE_VERTEX_BIT, |
| 122 | fVS, |
| 123 | outVertShaderModule, |
| 124 | &outStageInfo[0], |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 125 | shaders[kVertex_GrShaderType], |
| 126 | inputs[kVertex_GrShaderType])); |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 127 | |
| 128 | SkAssertResult(this->installVkShaderModule(VK_SHADER_STAGE_FRAGMENT_BIT, |
| 129 | fFS, |
| 130 | outFragShaderModule, |
| 131 | &outStageInfo[1], |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 132 | shaders[kFragment_GrShaderType], |
| 133 | inputs[kFragment_GrShaderType])); |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 134 | |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 135 | if (!shaders[kGeometry_GrShaderType].empty()) { |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 136 | SkAssertResult(this->installVkShaderModule(VK_SHADER_STAGE_GEOMETRY_BIT, |
| 137 | fGS, |
| 138 | outGeomShaderModule, |
| 139 | &outStageInfo[2], |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 140 | shaders[kGeometry_GrShaderType], |
| 141 | inputs[kGeometry_GrShaderType])); |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 142 | return 3; |
| 143 | } else { |
| 144 | return 2; |
Ethan Nicholas | 3865711 | 2017-02-09 17:01:22 -0500 | [diff] [blame] | 145 | } |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 146 | } |
| 147 | |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 148 | void GrVkPipelineStateBuilder::storeShadersInCache(const SkSL::String shaders[], |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 149 | const SkSL::Program::Inputs inputs[], |
| 150 | bool isSkSL) { |
Greg Daniel | a870b46 | 2019-01-08 15:49:46 -0500 | [diff] [blame] | 151 | Desc* desc = static_cast<Desc*>(this->desc()); |
Greg Daniel | a870b46 | 2019-01-08 15:49:46 -0500 | [diff] [blame] | 152 | sk_sp<SkData> key = SkData::MakeWithoutCopy(desc->asKey(), desc->shaderKeyLength()); |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 153 | sk_sp<SkData> data = GrPersistentCacheUtils::PackCachedShaders(isSkSL ? kSKSL_Tag : kSPIRV_Tag, |
| 154 | shaders, |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 155 | inputs, kGrShaderTypeCount); |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 156 | this->gpu()->getContext()->priv().getPersistentCache()->store(*key, *data); |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 157 | } |
| 158 | |
csmartdalton | c633abb | 2016-11-01 08:55:55 -0700 | [diff] [blame] | 159 | GrVkPipelineState* GrVkPipelineStateBuilder::finalize(const GrStencilSettings& stencil, |
| 160 | GrPrimitiveType primitiveType, |
Greg Daniel | 99b88e0 | 2018-10-03 15:31:20 -0400 | [diff] [blame] | 161 | VkRenderPass compatibleRenderPass, |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 162 | Desc* desc) { |
Brian Salomon | 662ea4b | 2018-07-12 14:53:49 -0400 | [diff] [blame] | 163 | VkDescriptorSetLayout dsLayout[2]; |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 164 | VkPipelineLayout pipelineLayout; |
Greg Daniel | 852d715 | 2017-05-08 12:10:12 -0400 | [diff] [blame] | 165 | VkShaderModule vertShaderModule = VK_NULL_HANDLE; |
| 166 | VkShaderModule geomShaderModule = VK_NULL_HANDLE; |
| 167 | VkShaderModule fragShaderModule = VK_NULL_HANDLE; |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 168 | |
egdaniel | 707bbd6 | 2016-07-26 07:19:47 -0700 | [diff] [blame] | 169 | GrVkResourceProvider& resourceProvider = fGpu->resourceProvider(); |
Jim Van Verth | 684d7ce | 2016-11-17 13:30:22 -0500 | [diff] [blame] | 170 | // These layouts are not owned by the PipelineStateBuilder and thus should not be destroyed |
egdaniel | 707bbd6 | 2016-07-26 07:19:47 -0700 | [diff] [blame] | 171 | dsLayout[GrVkUniformHandler::kUniformBufferDescSet] = resourceProvider.getUniformDSLayout(); |
| 172 | |
| 173 | GrVkDescriptorSetManager::Handle samplerDSHandle; |
Greg Daniel | a754378 | 2017-05-02 14:01:43 -0400 | [diff] [blame] | 174 | resourceProvider.getSamplerDescriptorSetHandle(VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, |
| 175 | fUniformHandler, &samplerDSHandle); |
egdaniel | 707bbd6 | 2016-07-26 07:19:47 -0700 | [diff] [blame] | 176 | dsLayout[GrVkUniformHandler::kSamplerDescSet] = |
| 177 | resourceProvider.getSamplerDSLayout(samplerDSHandle); |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 178 | |
| 179 | // Create the VkPipelineLayout |
| 180 | VkPipelineLayoutCreateInfo layoutCreateInfo; |
| 181 | memset(&layoutCreateInfo, 0, sizeof(VkPipelineLayoutCreateFlags)); |
| 182 | layoutCreateInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO; |
| 183 | layoutCreateInfo.pNext = 0; |
| 184 | layoutCreateInfo.flags = 0; |
Brian Salomon | 662ea4b | 2018-07-12 14:53:49 -0400 | [diff] [blame] | 185 | layoutCreateInfo.setLayoutCount = 2; |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 186 | layoutCreateInfo.pSetLayouts = dsLayout; |
| 187 | layoutCreateInfo.pushConstantRangeCount = 0; |
| 188 | layoutCreateInfo.pPushConstantRanges = nullptr; |
| 189 | |
| 190 | GR_VK_CALL_ERRCHECK(fGpu->vkInterface(), CreatePipelineLayout(fGpu->device(), |
| 191 | &layoutCreateInfo, |
| 192 | nullptr, |
| 193 | &pipelineLayout)); |
| 194 | |
| 195 | // We need to enable the following extensions so that the compiler can correctly make spir-v |
| 196 | // from our glsl shaders. |
| 197 | fVS.extensions().appendf("#extension GL_ARB_separate_shader_objects : enable\n"); |
| 198 | fFS.extensions().appendf("#extension GL_ARB_separate_shader_objects : enable\n"); |
| 199 | fVS.extensions().appendf("#extension GL_ARB_shading_language_420pack : enable\n"); |
| 200 | fFS.extensions().appendf("#extension GL_ARB_shading_language_420pack : enable\n"); |
| 201 | |
| 202 | this->finalizeShaders(); |
| 203 | |
Greg Daniel | 852d715 | 2017-05-08 12:10:12 -0400 | [diff] [blame] | 204 | VkPipelineShaderStageCreateInfo shaderStageInfo[3]; |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 205 | SkSL::Program::Settings settings; |
Ethan Nicholas | e468286 | 2017-07-18 14:08:18 -0400 | [diff] [blame] | 206 | settings.fCaps = this->caps()->shaderCaps(); |
Robert Phillips | d0fe875 | 2019-01-31 14:13:59 -0500 | [diff] [blame] | 207 | settings.fFlipY = this->origin() != kTopLeft_GrSurfaceOrigin; |
Robert Phillips | c1541ae | 2019-02-04 12:05:37 -0500 | [diff] [blame] | 208 | settings.fSharpenTextures = |
Robert Phillips | 9da87e0 | 2019-02-04 13:26:26 -0500 | [diff] [blame] | 209 | this->gpu()->getContext()->priv().options().fSharpenMipmappedTextures; |
Brian Salomon | dc09213 | 2018-04-04 10:14:16 -0400 | [diff] [blame] | 210 | SkASSERT(!this->fragColorIsInOut()); |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 211 | |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 212 | sk_sp<SkData> cached; |
Robert Phillips | 9da87e0 | 2019-02-04 13:26:26 -0500 | [diff] [blame] | 213 | auto persistentCache = fGpu->getContext()->priv().getPersistentCache(); |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 214 | if (persistentCache) { |
Greg Daniel | a870b46 | 2019-01-08 15:49:46 -0500 | [diff] [blame] | 215 | sk_sp<SkData> key = SkData::MakeWithoutCopy(desc->asKey(), desc->shaderKeyLength()); |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 216 | cached = persistentCache->load(*key); |
Greg Daniel | 852d715 | 2017-05-08 12:10:12 -0400 | [diff] [blame] | 217 | } |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 218 | bool binaryCache = true; |
| 219 | #if GR_TEST_UTILS |
| 220 | binaryCache = !fGpu->getContext()->priv().options().fCacheSKSL; |
| 221 | #endif |
| 222 | int numShaderStages = 0; |
| 223 | if (cached && binaryCache) { |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 224 | numShaderStages = this->loadShadersFromCache(*cached, &vertShaderModule, &fragShaderModule, |
| 225 | &geomShaderModule, shaderStageInfo); |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 226 | } |
| 227 | |
| 228 | if (!numShaderStages) { |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 229 | numShaderStages = 2; // We always have at least vertex and fragment stages. |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 230 | SkSL::String shaders[kGrShaderTypeCount]; |
| 231 | SkSL::Program::Inputs inputs[kGrShaderTypeCount]; |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 232 | |
| 233 | SkSL::String* sksl[kGrShaderTypeCount] = { |
| 234 | &fVS.fCompilerString, |
| 235 | &fGS.fCompilerString, |
| 236 | &fFS.fCompilerString, |
| 237 | }; |
| 238 | #if GR_TEST_UTILS |
| 239 | SkSL::String cached_sksl[kGrShaderTypeCount]; |
| 240 | if (cached) { |
| 241 | if (kSKSL_Tag == GrPersistentCacheUtils::UnpackCachedShaders( |
| 242 | cached.get(), cached_sksl, inputs, kGrShaderTypeCount)) { |
| 243 | for (int i = 0; i < kGrShaderTypeCount; ++i) { |
| 244 | sksl[i] = &cached_sksl[i]; |
| 245 | } |
| 246 | } |
| 247 | } |
| 248 | #endif |
| 249 | |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 250 | SkAssertResult(this->createVkShaderModule(VK_SHADER_STAGE_VERTEX_BIT, |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 251 | *sksl[kVertex_GrShaderType], |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 252 | &vertShaderModule, |
| 253 | &shaderStageInfo[0], |
| 254 | settings, |
| 255 | desc, |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 256 | &shaders[kVertex_GrShaderType], |
| 257 | &inputs[kVertex_GrShaderType])); |
Greg Daniel | 852d715 | 2017-05-08 12:10:12 -0400 | [diff] [blame] | 258 | |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 259 | SkAssertResult(this->createVkShaderModule(VK_SHADER_STAGE_FRAGMENT_BIT, |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 260 | *sksl[kFragment_GrShaderType], |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 261 | &fragShaderModule, |
| 262 | &shaderStageInfo[1], |
| 263 | settings, |
| 264 | desc, |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 265 | &shaders[kFragment_GrShaderType], |
| 266 | &inputs[kFragment_GrShaderType])); |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 267 | |
| 268 | if (this->primitiveProcessor().willUseGeoShader()) { |
| 269 | SkAssertResult(this->createVkShaderModule(VK_SHADER_STAGE_GEOMETRY_BIT, |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 270 | *sksl[kGeometry_GrShaderType], |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 271 | &geomShaderModule, |
| 272 | &shaderStageInfo[2], |
| 273 | settings, |
| 274 | desc, |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 275 | &shaders[kGeometry_GrShaderType], |
| 276 | &inputs[kGeometry_GrShaderType])); |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 277 | ++numShaderStages; |
| 278 | } |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 279 | if (persistentCache && !cached) { |
| 280 | bool isSkSL = false; |
| 281 | #if GR_TEST_UTILS |
| 282 | if (fGpu->getContext()->priv().options().fCacheSKSL) { |
| 283 | for (int i = 0; i < kGrShaderTypeCount; ++i) { |
Brian Osman | ac9be9d | 2019-05-01 10:29:34 -0400 | [diff] [blame^] | 284 | shaders[i] = GrShaderUtils::PrettyPrint(*sksl[i]); |
Brian Osman | fd7657c | 2019-04-25 11:34:07 -0400 | [diff] [blame] | 285 | } |
| 286 | isSkSL = true; |
| 287 | } |
| 288 | #endif |
| 289 | this->storeShadersInCache(shaders, inputs, isSkSL); |
Ethan Nicholas | 92e01cf | 2018-12-19 13:12:10 -0500 | [diff] [blame] | 290 | } |
| 291 | } |
Chris Dalton | d7291ba | 2019-03-07 14:17:03 -0700 | [diff] [blame] | 292 | GrVkPipeline* pipeline = resourceProvider.createPipeline( |
Chris Dalton | 71713f6 | 2019-04-18 12:41:03 -0600 | [diff] [blame] | 293 | this->renderTarget()->numColorSamples(), fPrimProc, fPipeline, stencil, this->origin(), |
| 294 | shaderStageInfo, numShaderStages, primitiveType, compatibleRenderPass, pipelineLayout); |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 295 | GR_VK_CALL(fGpu->vkInterface(), DestroyShaderModule(fGpu->device(), vertShaderModule, |
| 296 | nullptr)); |
| 297 | GR_VK_CALL(fGpu->vkInterface(), DestroyShaderModule(fGpu->device(), fragShaderModule, |
| 298 | nullptr)); |
Greg Daniel | 852d715 | 2017-05-08 12:10:12 -0400 | [diff] [blame] | 299 | // This if check should not be needed since calling destroy on a VK_NULL_HANDLE is allowed. |
| 300 | // However this is causing a crash in certain drivers (e.g. NVidia). |
| 301 | if (this->primitiveProcessor().willUseGeoShader()) { |
| 302 | GR_VK_CALL(fGpu->vkInterface(), DestroyShaderModule(fGpu->device(), geomShaderModule, |
| 303 | nullptr)); |
| 304 | } |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 305 | |
| 306 | if (!pipeline) { |
| 307 | GR_VK_CALL(fGpu->vkInterface(), DestroyPipelineLayout(fGpu->device(), pipelineLayout, |
| 308 | nullptr)); |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 309 | return nullptr; |
| 310 | } |
| 311 | |
egdaniel | 22281c1 | 2016-03-23 13:49:40 -0700 | [diff] [blame] | 312 | return new GrVkPipelineState(fGpu, |
egdaniel | 22281c1 | 2016-03-23 13:49:40 -0700 | [diff] [blame] | 313 | pipeline, |
| 314 | pipelineLayout, |
egdaniel | 707bbd6 | 2016-07-26 07:19:47 -0700 | [diff] [blame] | 315 | samplerDSHandle, |
egdaniel | 22281c1 | 2016-03-23 13:49:40 -0700 | [diff] [blame] | 316 | fUniformHandles, |
| 317 | fUniformHandler.fUniforms, |
Greg Daniel | 18f9602 | 2017-05-04 15:09:03 -0400 | [diff] [blame] | 318 | fUniformHandler.fCurrentGeometryUBOOffset, |
egdaniel | 22281c1 | 2016-03-23 13:49:40 -0700 | [diff] [blame] | 319 | fUniformHandler.fCurrentFragmentUBOOffset, |
Greg Daniel | 7a82edf | 2018-12-04 10:54:34 -0500 | [diff] [blame] | 320 | fUniformHandler.fSamplers, |
Robert Phillips | 369e8b7 | 2017-08-01 16:13:04 -0400 | [diff] [blame] | 321 | std::move(fGeometryProcessor), |
| 322 | std::move(fXferProcessor), |
Brian Salomon | 4d3f517 | 2018-06-07 14:42:52 -0400 | [diff] [blame] | 323 | std::move(fFragmentProcessors), |
| 324 | fFragmentProcessorCnt); |
jvanverth | 992ad36 | 2016-02-26 09:21:02 -0800 | [diff] [blame] | 325 | } |
egdaniel | 707bbd6 | 2016-07-26 07:19:47 -0700 | [diff] [blame] | 326 | |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 327 | ////////////////////////////////////////////////////////////////////////////// |
| 328 | |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 329 | bool GrVkPipelineStateBuilder::Desc::Build(Desc* desc, |
Robert Phillips | d0fe875 | 2019-01-31 14:13:59 -0500 | [diff] [blame] | 330 | GrRenderTarget* renderTarget, |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 331 | const GrPrimitiveProcessor& primProc, |
| 332 | const GrPipeline& pipeline, |
| 333 | const GrStencilSettings& stencil, |
| 334 | GrPrimitiveType primitiveType, |
Greg Daniel | 7a82edf | 2018-12-04 10:54:34 -0500 | [diff] [blame] | 335 | GrVkGpu* gpu) { |
Chris Dalton | d7291ba | 2019-03-07 14:17:03 -0700 | [diff] [blame] | 336 | if (!INHERITED::Build(desc, renderTarget, primProc, |
Robert Phillips | d0fe875 | 2019-01-31 14:13:59 -0500 | [diff] [blame] | 337 | primitiveType == GrPrimitiveType::kPoints, pipeline, gpu)) { |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 338 | return false; |
| 339 | } |
| 340 | |
| 341 | GrProcessorKeyBuilder b(&desc->key()); |
Greg Daniel | a870b46 | 2019-01-08 15:49:46 -0500 | [diff] [blame] | 342 | |
| 343 | b.add32(GrVkGpu::kShader_PersistentCacheKeyType); |
| 344 | int keyLength = desc->key().count(); |
| 345 | SkASSERT(0 == (keyLength % 4)); |
| 346 | desc->fShaderKeyLength = SkToU32(keyLength); |
| 347 | |
Robert Phillips | d0fe875 | 2019-01-31 14:13:59 -0500 | [diff] [blame] | 348 | GrVkRenderTarget* vkRT = (GrVkRenderTarget*)renderTarget; |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 349 | vkRT->simpleRenderPass()->genKey(&b); |
| 350 | |
| 351 | stencil.genKey(&b); |
| 352 | |
Jim Van Verth | 1223e7f | 2019-02-28 17:38:35 -0500 | [diff] [blame] | 353 | b.add32(pipeline.getBlendInfoKey()); |
Brian Salomon | 1471df9 | 2018-06-08 10:49:00 -0400 | [diff] [blame] | 354 | |
| 355 | b.add32((uint32_t)primitiveType); |
| 356 | |
| 357 | return true; |
| 358 | } |