ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [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 | */ |
Mike Klein | 6ad9909 | 2016-10-26 10:35:22 -0400 | [diff] [blame] | 7 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 8 | #include "src/sksl/SkSLCompiler.h" |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 9 | |
John Stiles | fbd050b | 2020-08-03 13:21:46 -0400 | [diff] [blame] | 10 | #include <memory> |
John Stiles | b8e010c | 2020-08-11 18:05:39 -0400 | [diff] [blame] | 11 | #include <unordered_set> |
John Stiles | fbd050b | 2020-08-03 13:21:46 -0400 | [diff] [blame] | 12 | |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 13 | #include "include/sksl/DSLCore.h" |
Leon Scroggins | b66214e | 2021-02-11 17:14:18 -0500 | [diff] [blame] | 14 | #include "src/core/SkTraceEvent.h" |
John Stiles | f3a28db | 2021-03-10 23:00:47 -0500 | [diff] [blame] | 15 | #include "src/sksl/SkSLConstantFolder.h" |
Ethan Nicholas | dd2fdea | 2021-07-20 15:23:04 -0400 | [diff] [blame] | 16 | #include "src/sksl/SkSLDSLParser.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 17 | #include "src/sksl/SkSLIRGenerator.h" |
Ethan Nicholas | 58014fa | 2021-09-29 17:18:18 -0400 | [diff] [blame] | 18 | #include "src/sksl/SkSLIntrinsicMap.h" |
Brian Osman | 0018501 | 2021-02-04 16:07:11 -0500 | [diff] [blame] | 19 | #include "src/sksl/SkSLOperators.h" |
John Stiles | 270cec2 | 2021-02-17 12:59:36 -0500 | [diff] [blame] | 20 | #include "src/sksl/SkSLProgramSettings.h" |
Ethan Nicholas | c18bb51 | 2020-07-28 14:46:53 -0400 | [diff] [blame] | 21 | #include "src/sksl/SkSLRehydrator.h" |
Ethan Nicholas | c845272 | 2021-10-07 10:47:32 -0400 | [diff] [blame] | 22 | #include "src/sksl/SkSLThreadContext.h" |
John Stiles | 3738ef5 | 2021-04-13 10:41:57 -0400 | [diff] [blame] | 23 | #include "src/sksl/codegen/SkSLGLSLCodeGenerator.h" |
John Stiles | 3738ef5 | 2021-04-13 10:41:57 -0400 | [diff] [blame] | 24 | #include "src/sksl/codegen/SkSLMetalCodeGenerator.h" |
| 25 | #include "src/sksl/codegen/SkSLSPIRVCodeGenerator.h" |
| 26 | #include "src/sksl/codegen/SkSLSPIRVtoHLSL.h" |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 27 | #include "src/sksl/dsl/priv/DSLWriter.h" |
| 28 | #include "src/sksl/dsl/priv/DSL_priv.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 29 | #include "src/sksl/ir/SkSLExpression.h" |
| 30 | #include "src/sksl/ir/SkSLExpressionStatement.h" |
| 31 | #include "src/sksl/ir/SkSLFunctionCall.h" |
John Stiles | 7591d4b | 2021-09-13 13:32:06 -0400 | [diff] [blame] | 32 | #include "src/sksl/ir/SkSLLiteral.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 33 | #include "src/sksl/ir/SkSLModifiersDeclaration.h" |
| 34 | #include "src/sksl/ir/SkSLNop.h" |
| 35 | #include "src/sksl/ir/SkSLSymbolTable.h" |
| 36 | #include "src/sksl/ir/SkSLTernaryExpression.h" |
| 37 | #include "src/sksl/ir/SkSLUnresolvedFunction.h" |
| 38 | #include "src/sksl/ir/SkSLVarDeclarations.h" |
Ethan Nicholas | 1780755 | 2021-10-01 09:42:36 -0400 | [diff] [blame] | 39 | #include "src/sksl/transform/SkSLProgramWriter.h" |
John Stiles | 7176637 | 2021-10-06 15:54:55 -0400 | [diff] [blame] | 40 | #include "src/sksl/transform/SkSLTransform.h" |
John Stiles | e615000 | 2020-10-05 12:03:53 -0400 | [diff] [blame] | 41 | #include "src/utils/SkBitSet.h" |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 42 | |
Ethan Nicholas | b33fa3f | 2020-08-06 13:00:19 -0400 | [diff] [blame] | 43 | #include <fstream> |
| 44 | |
Ethan Nicholas | a11035b | 2019-11-26 16:27:47 -0500 | [diff] [blame] | 45 | #if !defined(SKSL_STANDALONE) & SK_SUPPORT_GPU |
| 46 | #include "include/gpu/GrContextOptions.h" |
| 47 | #include "src/gpu/GrShaderCaps.h" |
| 48 | #endif |
| 49 | |
Ethan Nicholas | a6ae1f7 | 2017-03-16 09:56:54 -0400 | [diff] [blame] | 50 | #ifdef SK_ENABLE_SPIRV_VALIDATION |
| 51 | #include "spirv-tools/libspirv.hpp" |
| 52 | #endif |
| 53 | |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 54 | #if defined(SKSL_STANDALONE) |
Ethan Nicholas | c18bb51 | 2020-07-28 14:46:53 -0400 | [diff] [blame] | 55 | |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 56 | // In standalone mode, we load the textual sksl source files. GN generates or copies these files |
| 57 | // to the skslc executable directory. The "data" in this mode is just the filename. |
| 58 | #define MODULE_DATA(name) MakeModulePath("sksl_" #name ".sksl") |
| 59 | |
| 60 | #else |
| 61 | |
| 62 | // At runtime, we load the dehydrated sksl data files. The data is a (pointer, size) pair. |
Ethan Nicholas | c18bb51 | 2020-07-28 14:46:53 -0400 | [diff] [blame] | 63 | #include "src/sksl/generated/sksl_frag.dehydrated.sksl" |
Ethan Nicholas | c18bb51 | 2020-07-28 14:46:53 -0400 | [diff] [blame] | 64 | #include "src/sksl/generated/sksl_gpu.dehydrated.sksl" |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 65 | #include "src/sksl/generated/sksl_public.dehydrated.sksl" |
Brian Osman | cbb60bd | 2021-04-12 09:49:20 -0400 | [diff] [blame] | 66 | #include "src/sksl/generated/sksl_rt_shader.dehydrated.sksl" |
Ethan Nicholas | c18bb51 | 2020-07-28 14:46:53 -0400 | [diff] [blame] | 67 | #include "src/sksl/generated/sksl_vert.dehydrated.sksl" |
| 68 | |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 69 | #define MODULE_DATA(name) MakeModuleData(SKSL_INCLUDE_sksl_##name,\ |
| 70 | SKSL_INCLUDE_sksl_##name##_LENGTH) |
Ethan Nicholas | c18bb51 | 2020-07-28 14:46:53 -0400 | [diff] [blame] | 71 | |
| 72 | #endif |
Ethan Nicholas | 0d99766 | 2019-04-08 09:46:01 -0400 | [diff] [blame] | 73 | |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 74 | namespace SkSL { |
| 75 | |
John Stiles | 7247b48 | 2021-03-08 10:40:35 -0500 | [diff] [blame] | 76 | // These flags allow tools like Viewer or Nanobench to override the compiler's ProgramSettings. |
John Stiles | 2ee4d7a | 2021-03-30 10:30:47 -0400 | [diff] [blame] | 77 | Compiler::OverrideFlag Compiler::sOptimizer = OverrideFlag::kDefault; |
| 78 | Compiler::OverrideFlag Compiler::sInliner = OverrideFlag::kDefault; |
John Stiles | 8ef4d6c | 2021-03-05 16:01:45 -0500 | [diff] [blame] | 79 | |
John Stiles | 47c0a74 | 2021-02-09 09:30:35 -0500 | [diff] [blame] | 80 | using RefKind = VariableReference::RefKind; |
| 81 | |
Brian Osman | 88cda17 | 2020-10-09 12:05:16 -0400 | [diff] [blame] | 82 | class AutoSource { |
| 83 | public: |
Ethan Nicholas | b449fff | 2021-08-04 15:06:37 -0400 | [diff] [blame] | 84 | AutoSource(Compiler* compiler, const char* source) |
John Stiles | a289ac2 | 2021-05-06 07:35:35 -0400 | [diff] [blame] | 85 | : fCompiler(compiler) { |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 86 | SkASSERT(!fCompiler->errorReporter().source()); |
| 87 | fCompiler->errorReporter().setSource(source); |
Brian Osman | 88cda17 | 2020-10-09 12:05:16 -0400 | [diff] [blame] | 88 | } |
| 89 | |
John Stiles | a289ac2 | 2021-05-06 07:35:35 -0400 | [diff] [blame] | 90 | ~AutoSource() { |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 91 | fCompiler->errorReporter().setSource(nullptr); |
John Stiles | a289ac2 | 2021-05-06 07:35:35 -0400 | [diff] [blame] | 92 | } |
Brian Osman | 88cda17 | 2020-10-09 12:05:16 -0400 | [diff] [blame] | 93 | |
| 94 | Compiler* fCompiler; |
Brian Osman | 88cda17 | 2020-10-09 12:05:16 -0400 | [diff] [blame] | 95 | }; |
| 96 | |
John Stiles | a935c3f | 2021-02-25 10:35:49 -0500 | [diff] [blame] | 97 | class AutoProgramConfig { |
| 98 | public: |
| 99 | AutoProgramConfig(std::shared_ptr<Context>& context, ProgramConfig* config) |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 100 | : fContext(context.get()) |
| 101 | , fOldConfig(fContext->fConfig) { |
John Stiles | a935c3f | 2021-02-25 10:35:49 -0500 | [diff] [blame] | 102 | fContext->fConfig = config; |
| 103 | } |
| 104 | |
| 105 | ~AutoProgramConfig() { |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 106 | fContext->fConfig = fOldConfig; |
John Stiles | a935c3f | 2021-02-25 10:35:49 -0500 | [diff] [blame] | 107 | } |
| 108 | |
| 109 | Context* fContext; |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 110 | ProgramConfig* fOldConfig; |
John Stiles | a935c3f | 2021-02-25 10:35:49 -0500 | [diff] [blame] | 111 | }; |
| 112 | |
John Stiles | 10d39d9 | 2021-05-04 16:13:14 -0400 | [diff] [blame] | 113 | class AutoModifiersPool { |
| 114 | public: |
| 115 | AutoModifiersPool(std::shared_ptr<Context>& context, ModifiersPool* modifiersPool) |
| 116 | : fContext(context.get()) { |
| 117 | SkASSERT(!fContext->fModifiersPool); |
| 118 | fContext->fModifiersPool = modifiersPool; |
| 119 | } |
| 120 | |
| 121 | ~AutoModifiersPool() { |
| 122 | fContext->fModifiersPool = nullptr; |
| 123 | } |
| 124 | |
| 125 | Context* fContext; |
| 126 | }; |
| 127 | |
John Stiles | d6a5f449 | 2021-02-11 15:46:11 -0500 | [diff] [blame] | 128 | Compiler::Compiler(const ShaderCapsClass* caps) |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 129 | : fErrorReporter(this) |
John Stiles | 2dda4b6 | 2021-09-16 13:18:10 -0400 | [diff] [blame] | 130 | , fContext(std::make_shared<Context>(fErrorReporter, *caps, fMangler)) |
John Stiles | a47b351 | 2021-05-04 16:15:00 -0400 | [diff] [blame] | 131 | , fInliner(fContext.get()) { |
John Stiles | c1a98b8 | 2021-02-24 13:35:02 -0500 | [diff] [blame] | 132 | SkASSERT(caps); |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 133 | fRootModule.fSymbols = this->makeRootSymbolTable(); |
| 134 | fPrivateModule.fSymbols = this->makePrivateSymbolTable(fRootModule.fSymbols); |
John Stiles | c1a98b8 | 2021-02-24 13:35:02 -0500 | [diff] [blame] | 135 | fIRGenerator = std::make_unique<IRGenerator>(fContext.get()); |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 136 | } |
| 137 | |
| 138 | Compiler::~Compiler() {} |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 139 | |
John Stiles | 54e7c05 | 2021-01-11 14:22:36 -0500 | [diff] [blame] | 140 | #define TYPE(t) fContext->fTypes.f ## t .get() |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 141 | |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 142 | std::shared_ptr<SymbolTable> Compiler::makeRootSymbolTable() { |
Ethan Nicholas | c7774a7 | 2021-08-27 15:34:05 -0400 | [diff] [blame] | 143 | auto rootSymbolTable = std::make_shared<SymbolTable>(*fContext, /*builtin=*/true); |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 144 | |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 145 | const SkSL::Symbol* rootTypes[] = { |
| 146 | TYPE(Void), |
Brian Salomon | bf7b620 | 2016-11-11 16:08:03 -0500 | [diff] [blame] | 147 | |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 148 | TYPE( Float), TYPE( Float2), TYPE( Float3), TYPE( Float4), |
| 149 | TYPE( Half), TYPE( Half2), TYPE( Half3), TYPE( Half4), |
| 150 | TYPE( Int), TYPE( Int2), TYPE( Int3), TYPE( Int4), |
John Stiles | 823c504 | 2021-08-17 12:09:00 -0400 | [diff] [blame] | 151 | TYPE( UInt), TYPE( UInt2), TYPE( UInt3), TYPE( UInt4), |
| 152 | TYPE( Short), TYPE( Short2), TYPE( Short3), TYPE( Short4), |
| 153 | TYPE(UShort), TYPE(UShort2), TYPE(UShort3), TYPE(UShort4), |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 154 | TYPE( Bool), TYPE( Bool2), TYPE( Bool3), TYPE( Bool4), |
Brian Salomon | 2a51de8 | 2016-11-16 12:06:01 -0500 | [diff] [blame] | 155 | |
John Stiles | 823c504 | 2021-08-17 12:09:00 -0400 | [diff] [blame] | 156 | TYPE(Float2x2), TYPE(Float2x3), TYPE(Float2x4), |
| 157 | TYPE(Float3x2), TYPE(Float3x3), TYPE(Float3x4), |
| 158 | TYPE(Float4x2), TYPE(Float4x3), TYPE(Float4x4), |
| 159 | |
| 160 | TYPE(Half2x2), TYPE(Half2x3), TYPE(Half2x4), |
| 161 | TYPE(Half3x2), TYPE(Half3x3), TYPE(Half3x4), |
| 162 | TYPE(Half4x2), TYPE(Half4x3), TYPE(Half4x4), |
Greg Daniel | 64773e6 | 2016-11-22 09:44:03 -0500 | [diff] [blame] | 163 | |
Brian Osman | c63f431 | 2020-12-23 11:44:14 -0500 | [diff] [blame] | 164 | TYPE(SquareMat), TYPE(SquareHMat), |
John Stiles | 823c504 | 2021-08-17 12:09:00 -0400 | [diff] [blame] | 165 | TYPE(Mat), TYPE(HMat), |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 166 | |
John Stiles | 823c504 | 2021-08-17 12:09:00 -0400 | [diff] [blame] | 167 | // TODO(skia:12349): generic short/ushort |
| 168 | TYPE(GenType), TYPE(GenIType), TYPE(GenUType), |
| 169 | TYPE(GenHType), /* (GenSType) (GenUSType) */ |
| 170 | TYPE(GenBType), |
| 171 | |
| 172 | TYPE(Vec), TYPE(IVec), TYPE(UVec), |
| 173 | TYPE(HVec), TYPE(SVec), TYPE(USVec), |
| 174 | TYPE(BVec), |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 175 | |
Brian Osman | 14d0096 | 2021-04-02 17:04:35 -0400 | [diff] [blame] | 176 | TYPE(ColorFilter), |
| 177 | TYPE(Shader), |
John Stiles | bb2ef92 | 2021-07-26 08:32:07 -0400 | [diff] [blame] | 178 | TYPE(Blender), |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 179 | }; |
| 180 | |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 181 | for (const SkSL::Symbol* type : rootTypes) { |
| 182 | rootSymbolTable->addWithoutOwnership(type); |
| 183 | } |
| 184 | |
| 185 | return rootSymbolTable; |
| 186 | } |
| 187 | |
| 188 | std::shared_ptr<SymbolTable> Compiler::makePrivateSymbolTable(std::shared_ptr<SymbolTable> parent) { |
| 189 | auto privateSymbolTable = std::make_shared<SymbolTable>(parent, /*builtin=*/true); |
| 190 | |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 191 | const SkSL::Symbol* privateTypes[] = { |
| 192 | TYPE(Sampler1D), TYPE(Sampler2D), TYPE(Sampler3D), |
| 193 | TYPE(SamplerExternalOES), |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 194 | TYPE(Sampler2DRect), |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 195 | |
| 196 | TYPE(ISampler2D), |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 197 | TYPE(SubpassInput), TYPE(SubpassInputMS), |
| 198 | |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 199 | TYPE(Sampler), |
| 200 | TYPE(Texture2D), |
| 201 | }; |
| 202 | |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 203 | for (const SkSL::Symbol* type : privateTypes) { |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 204 | privateSymbolTable->addWithoutOwnership(type); |
Brian Osman | b06301e | 2020-11-06 11:45:36 -0500 | [diff] [blame] | 205 | } |
| 206 | |
Brian Osman | 3887a01 | 2020-09-30 13:22:27 -0400 | [diff] [blame] | 207 | // sk_Caps is "builtin", but all references to it are resolved to Settings, so we don't need to |
| 208 | // treat it as builtin (ie, no need to clone it into the Program). |
Ethan Nicholas | 89cfde1 | 2021-09-27 11:20:34 -0400 | [diff] [blame] | 209 | privateSymbolTable->add(std::make_unique<Variable>(/*line=*/-1, |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 210 | fCoreModifiers.add(Modifiers{}), |
| 211 | "sk_Caps", |
| 212 | fContext->fTypes.fSkCaps.get(), |
| 213 | /*builtin=*/false, |
| 214 | Variable::Storage::kGlobal)); |
Ethan Nicholas | 3605ace | 2016-11-21 15:59:48 -0500 | [diff] [blame] | 215 | |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 216 | return privateSymbolTable; |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 217 | } |
| 218 | |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 219 | #undef TYPE |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 220 | |
Brian Osman | 5626998 | 2020-11-20 12:38:07 -0500 | [diff] [blame] | 221 | const ParsedModule& Compiler::loadGPUModule() { |
| 222 | if (!fGPUModule.fSymbols) { |
John Stiles | dbd4e6f | 2021-02-16 13:29:15 -0500 | [diff] [blame] | 223 | fGPUModule = this->parseModule(ProgramKind::kFragment, MODULE_DATA(gpu), fPrivateModule); |
Brian Osman | 5626998 | 2020-11-20 12:38:07 -0500 | [diff] [blame] | 224 | } |
| 225 | return fGPUModule; |
| 226 | } |
| 227 | |
| 228 | const ParsedModule& Compiler::loadFragmentModule() { |
| 229 | if (!fFragmentModule.fSymbols) { |
John Stiles | dbd4e6f | 2021-02-16 13:29:15 -0500 | [diff] [blame] | 230 | fFragmentModule = this->parseModule(ProgramKind::kFragment, MODULE_DATA(frag), |
Brian Osman | 5626998 | 2020-11-20 12:38:07 -0500 | [diff] [blame] | 231 | this->loadGPUModule()); |
| 232 | } |
| 233 | return fFragmentModule; |
| 234 | } |
| 235 | |
| 236 | const ParsedModule& Compiler::loadVertexModule() { |
| 237 | if (!fVertexModule.fSymbols) { |
John Stiles | dbd4e6f | 2021-02-16 13:29:15 -0500 | [diff] [blame] | 238 | fVertexModule = this->parseModule(ProgramKind::kVertex, MODULE_DATA(vert), |
Brian Osman | 5626998 | 2020-11-20 12:38:07 -0500 | [diff] [blame] | 239 | this->loadGPUModule()); |
| 240 | } |
| 241 | return fVertexModule; |
| 242 | } |
| 243 | |
Brian Osman | cbb60bd | 2021-04-12 09:49:20 -0400 | [diff] [blame] | 244 | static void add_glsl_type_aliases(SkSL::SymbolTable* symbols, const SkSL::BuiltinTypes& types) { |
| 245 | // Add some aliases to the runtime effect modules so that it's friendlier, and more like GLSL |
| 246 | symbols->addAlias("vec2", types.fFloat2.get()); |
| 247 | symbols->addAlias("vec3", types.fFloat3.get()); |
| 248 | symbols->addAlias("vec4", types.fFloat4.get()); |
| 249 | |
| 250 | symbols->addAlias("ivec2", types.fInt2.get()); |
| 251 | symbols->addAlias("ivec3", types.fInt3.get()); |
| 252 | symbols->addAlias("ivec4", types.fInt4.get()); |
| 253 | |
| 254 | symbols->addAlias("bvec2", types.fBool2.get()); |
| 255 | symbols->addAlias("bvec3", types.fBool3.get()); |
| 256 | symbols->addAlias("bvec4", types.fBool4.get()); |
| 257 | |
| 258 | symbols->addAlias("mat2", types.fFloat2x2.get()); |
| 259 | symbols->addAlias("mat3", types.fFloat3x3.get()); |
| 260 | symbols->addAlias("mat4", types.fFloat4x4.get()); |
| 261 | } |
| 262 | |
Brian Osman | a8b897b | 2021-08-30 16:40:44 -0400 | [diff] [blame] | 263 | const ParsedModule& Compiler::loadPublicModule() { |
| 264 | if (!fPublicModule.fSymbols) { |
| 265 | fPublicModule = this->parseModule(ProgramKind::kGeneric, MODULE_DATA(public), fRootModule); |
| 266 | add_glsl_type_aliases(fPublicModule.fSymbols.get(), fContext->fTypes); |
Brian Osman | cbb60bd | 2021-04-12 09:49:20 -0400 | [diff] [blame] | 267 | } |
Brian Osman | a8b897b | 2021-08-30 16:40:44 -0400 | [diff] [blame] | 268 | return fPublicModule; |
Brian Osman | cbb60bd | 2021-04-12 09:49:20 -0400 | [diff] [blame] | 269 | } |
| 270 | |
| 271 | const ParsedModule& Compiler::loadRuntimeShaderModule() { |
| 272 | if (!fRuntimeShaderModule.fSymbols) { |
| 273 | fRuntimeShaderModule = this->parseModule( |
| 274 | ProgramKind::kRuntimeShader, MODULE_DATA(rt_shader), this->loadPublicModule()); |
Brian Osman | cbb60bd | 2021-04-12 09:49:20 -0400 | [diff] [blame] | 275 | } |
| 276 | return fRuntimeShaderModule; |
| 277 | } |
| 278 | |
John Stiles | dbd4e6f | 2021-02-16 13:29:15 -0500 | [diff] [blame] | 279 | const ParsedModule& Compiler::moduleForProgramKind(ProgramKind kind) { |
Brian Osman | 88cda17 | 2020-10-09 12:05:16 -0400 | [diff] [blame] | 280 | switch (kind) { |
Brian Osman | a8b897b | 2021-08-30 16:40:44 -0400 | [diff] [blame] | 281 | case ProgramKind::kVertex: return this->loadVertexModule(); break; |
| 282 | case ProgramKind::kFragment: return this->loadFragmentModule(); break; |
| 283 | case ProgramKind::kRuntimeColorFilter: return this->loadPublicModule(); break; |
| 284 | case ProgramKind::kRuntimeShader: return this->loadRuntimeShaderModule(); break; |
| 285 | case ProgramKind::kRuntimeBlender: return this->loadPublicModule(); break; |
| 286 | case ProgramKind::kGeneric: return this->loadPublicModule(); break; |
Brian Osman | 88cda17 | 2020-10-09 12:05:16 -0400 | [diff] [blame] | 287 | } |
| 288 | SkUNREACHABLE; |
Ethan Nicholas | c18bb51 | 2020-07-28 14:46:53 -0400 | [diff] [blame] | 289 | } |
| 290 | |
John Stiles | dbd4e6f | 2021-02-16 13:29:15 -0500 | [diff] [blame] | 291 | LoadedModule Compiler::loadModule(ProgramKind kind, |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 292 | ModuleData data, |
John Stiles | a935c3f | 2021-02-25 10:35:49 -0500 | [diff] [blame] | 293 | std::shared_ptr<SymbolTable> base, |
| 294 | bool dehydrate) { |
| 295 | if (dehydrate) { |
| 296 | // NOTE: This is a workaround. When dehydrating includes, skslc doesn't know which module |
| 297 | // it's preparing, nor what the correct base module is. We can't use 'Root', because many |
| 298 | // GPU intrinsics reference private types, like samplers or textures. Today, 'Private' does |
| 299 | // contain the union of all known types, so this is safe. If we ever have types that only |
| 300 | // exist in 'Public' (for example), this logic needs to be smarter (by choosing the correct |
| 301 | // base for the module we're compiling). |
John Stiles | b624b72 | 2021-08-13 12:16:13 -0400 | [diff] [blame] | 302 | base = fPrivateModule.fSymbols; |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 303 | } |
John Stiles | a935c3f | 2021-02-25 10:35:49 -0500 | [diff] [blame] | 304 | SkASSERT(base); |
| 305 | |
John Stiles | a47b351 | 2021-05-04 16:15:00 -0400 | [diff] [blame] | 306 | // Put the core-module modifier pool into the context. |
| 307 | AutoModifiersPool autoPool(fContext, &fCoreModifiers); |
John Stiles | 10d39d9 | 2021-05-04 16:13:14 -0400 | [diff] [blame] | 308 | |
John Stiles | a935c3f | 2021-02-25 10:35:49 -0500 | [diff] [blame] | 309 | // Built-in modules always use default program settings. |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 310 | Program::Settings settings; |
| 311 | settings.fReplaceSettings = !dehydrate; |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 312 | |
| 313 | #if defined(SKSL_STANDALONE) |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 314 | SkASSERT(this->errorCount() == 0); |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 315 | SkASSERT(data.fPath); |
| 316 | std::ifstream in(data.fPath); |
John Stiles | d51c979 | 2021-03-18 11:40:14 -0400 | [diff] [blame] | 317 | String text{std::istreambuf_iterator<char>(in), std::istreambuf_iterator<char>()}; |
Ethan Nicholas | b33fa3f | 2020-08-06 13:00:19 -0400 | [diff] [blame] | 318 | if (in.rdstate()) { |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 319 | printf("error reading %s\n", data.fPath); |
Ethan Nicholas | b33fa3f | 2020-08-06 13:00:19 -0400 | [diff] [blame] | 320 | abort(); |
| 321 | } |
Brian Osman | 88cda17 | 2020-10-09 12:05:16 -0400 | [diff] [blame] | 322 | ParsedModule baseModule = {base, /*fIntrinsics=*/nullptr}; |
Ethan Nicholas | 051aeb7 | 2021-09-24 16:39:19 -0400 | [diff] [blame] | 323 | LoadedModule result = DSLParser(this, settings, kind, |
| 324 | std::move(text)).moduleInheritingFrom(std::move(baseModule)); |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 325 | if (this->errorCount()) { |
Ethan Nicholas | 8da1e65 | 2019-05-24 11:01:59 -0400 | [diff] [blame] | 326 | printf("Unexpected errors: %s\n", this->fErrorText.c_str()); |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 327 | SkDEBUGFAILF("%s %s\n", data.fPath, this->fErrorText.c_str()); |
Ethan Nicholas | 8da1e65 | 2019-05-24 11:01:59 -0400 | [diff] [blame] | 328 | } |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 329 | #else |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 330 | ProgramConfig config; |
John Stiles | c876489 | 2021-10-04 11:56:30 -0400 | [diff] [blame] | 331 | config.fIsBuiltinCode = true; |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 332 | config.fKind = kind; |
| 333 | config.fSettings = settings; |
| 334 | AutoProgramConfig autoConfig(fContext, &config); |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 335 | SkASSERT(data.fData && (data.fSize != 0)); |
John Stiles | 10d39d9 | 2021-05-04 16:13:14 -0400 | [diff] [blame] | 336 | Rehydrator rehydrator(fContext.get(), base, data.fData, data.fSize); |
Ethan Nicholas | 051aeb7 | 2021-09-24 16:39:19 -0400 | [diff] [blame] | 337 | LoadedModule result = { kind, rehydrator.symbolTable(), rehydrator.elements() }; |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 338 | #endif |
| 339 | |
Ethan Nicholas | 051aeb7 | 2021-09-24 16:39:19 -0400 | [diff] [blame] | 340 | return result; |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 341 | } |
| 342 | |
John Stiles | dbd4e6f | 2021-02-16 13:29:15 -0500 | [diff] [blame] | 343 | ParsedModule Compiler::parseModule(ProgramKind kind, ModuleData data, const ParsedModule& base) { |
John Stiles | a935c3f | 2021-02-25 10:35:49 -0500 | [diff] [blame] | 344 | LoadedModule module = this->loadModule(kind, data, base.fSymbols, /*dehydrate=*/false); |
Brian Osman | 0006ad0 | 2020-11-18 15:38:39 -0500 | [diff] [blame] | 345 | this->optimize(module); |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 346 | |
| 347 | // For modules that just declare (but don't define) intrinsic functions, there will be no new |
| 348 | // program elements. In that case, we can share our parent's intrinsic map: |
Brian Osman | 0006ad0 | 2020-11-18 15:38:39 -0500 | [diff] [blame] | 349 | if (module.fElements.empty()) { |
John Stiles | 10d39d9 | 2021-05-04 16:13:14 -0400 | [diff] [blame] | 350 | return ParsedModule{module.fSymbols, base.fIntrinsics}; |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 351 | } |
| 352 | |
Ethan Nicholas | 58014fa | 2021-09-29 17:18:18 -0400 | [diff] [blame] | 353 | auto intrinsics = std::make_shared<IntrinsicMap>(base.fIntrinsics.get()); |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 354 | |
| 355 | // Now, transfer all of the program elements to an intrinsic map. This maps certain types of |
| 356 | // global objects to the declaring ProgramElement. |
Brian Osman | 0006ad0 | 2020-11-18 15:38:39 -0500 | [diff] [blame] | 357 | for (std::unique_ptr<ProgramElement>& element : module.fElements) { |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 358 | switch (element->kind()) { |
| 359 | case ProgramElement::Kind::kFunction: { |
| 360 | const FunctionDefinition& f = element->as<FunctionDefinition>(); |
Ethan Nicholas | 0a5d096 | 2020-10-14 13:33:18 -0400 | [diff] [blame] | 361 | SkASSERT(f.declaration().isBuiltin()); |
| 362 | intrinsics->insertOrDie(f.declaration().description(), std::move(element)); |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 363 | break; |
| 364 | } |
John Stiles | 569249b | 2020-11-03 12:18:22 -0500 | [diff] [blame] | 365 | case ProgramElement::Kind::kFunctionPrototype: { |
| 366 | // These are already in the symbol table. |
| 367 | break; |
| 368 | } |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 369 | case ProgramElement::Kind::kGlobalVar: { |
Ethan Nicholas | c51f33e | 2020-10-13 13:49:44 -0400 | [diff] [blame] | 370 | const GlobalVarDeclaration& global = element->as<GlobalVarDeclaration>(); |
| 371 | const Variable& var = global.declaration()->as<VarDeclaration>().var(); |
| 372 | SkASSERT(var.isBuiltin()); |
Ethan Nicholas | d2e0960 | 2021-06-10 11:21:59 -0400 | [diff] [blame] | 373 | intrinsics->insertOrDie(String(var.name()), std::move(element)); |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 374 | break; |
| 375 | } |
| 376 | case ProgramElement::Kind::kInterfaceBlock: { |
Ethan Nicholas | eaf4788 | 2020-10-15 10:10:08 -0400 | [diff] [blame] | 377 | const Variable& var = element->as<InterfaceBlock>().variable(); |
| 378 | SkASSERT(var.isBuiltin()); |
Ethan Nicholas | d2e0960 | 2021-06-10 11:21:59 -0400 | [diff] [blame] | 379 | intrinsics->insertOrDie(String(var.name()), std::move(element)); |
Brian Osman | 3d87e9f | 2020-10-08 11:50:22 -0400 | [diff] [blame] | 380 | break; |
| 381 | } |
| 382 | default: |
| 383 | printf("Unsupported element: %s\n", element->description().c_str()); |
| 384 | SkASSERT(false); |
| 385 | break; |
| 386 | } |
| 387 | } |
| 388 | |
John Stiles | 10d39d9 | 2021-05-04 16:13:14 -0400 | [diff] [blame] | 389 | return ParsedModule{module.fSymbols, std::move(intrinsics)}; |
Ethan Nicholas | 8da1e65 | 2019-05-24 11:01:59 -0400 | [diff] [blame] | 390 | } |
| 391 | |
John Stiles | c876489 | 2021-10-04 11:56:30 -0400 | [diff] [blame] | 392 | std::unique_ptr<Program> Compiler::convertProgram(ProgramKind kind, |
| 393 | String text, |
| 394 | Program::Settings settings) { |
Brian Osman | 7a20b5c | 2021-03-15 16:23:33 -0400 | [diff] [blame] | 395 | TRACE_EVENT0("skia.shaders", "SkSL::Compiler::convertProgram"); |
Leon Scroggins | b66214e | 2021-02-11 17:14:18 -0500 | [diff] [blame] | 396 | |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 397 | SkASSERT(!settings.fExternalFunctions || (kind == ProgramKind::kGeneric)); |
Ethan Nicholas | 91164d1 | 2019-05-15 15:29:54 -0400 | [diff] [blame] | 398 | |
John Stiles | 2ee4d7a | 2021-03-30 10:30:47 -0400 | [diff] [blame] | 399 | // Honor our optimization-override flags. |
| 400 | switch (sOptimizer) { |
| 401 | case OverrideFlag::kDefault: |
| 402 | break; |
| 403 | case OverrideFlag::kOff: |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 404 | settings.fOptimize = false; |
John Stiles | 2ee4d7a | 2021-03-30 10:30:47 -0400 | [diff] [blame] | 405 | break; |
| 406 | case OverrideFlag::kOn: |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 407 | settings.fOptimize = true; |
John Stiles | 2ee4d7a | 2021-03-30 10:30:47 -0400 | [diff] [blame] | 408 | break; |
| 409 | } |
| 410 | |
| 411 | switch (sInliner) { |
| 412 | case OverrideFlag::kDefault: |
| 413 | break; |
| 414 | case OverrideFlag::kOff: |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 415 | settings.fInlineThreshold = 0; |
John Stiles | 2ee4d7a | 2021-03-30 10:30:47 -0400 | [diff] [blame] | 416 | break; |
| 417 | case OverrideFlag::kOn: |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 418 | if (settings.fInlineThreshold == 0) { |
| 419 | settings.fInlineThreshold = kDefaultInlineThreshold; |
John Stiles | 2ee4d7a | 2021-03-30 10:30:47 -0400 | [diff] [blame] | 420 | } |
| 421 | break; |
| 422 | } |
John Stiles | 7247b48 | 2021-03-08 10:40:35 -0500 | [diff] [blame] | 423 | |
| 424 | // Disable optimization settings that depend on a parent setting which has been disabled. |
Ethan Nicholas | 55a63af | 2021-05-18 10:12:58 -0400 | [diff] [blame] | 425 | settings.fInlineThreshold *= (int)settings.fOptimize; |
| 426 | settings.fRemoveDeadFunctions &= settings.fOptimize; |
| 427 | settings.fRemoveDeadVariables &= settings.fOptimize; |
John Stiles | 7247b48 | 2021-03-08 10:40:35 -0500 | [diff] [blame] | 428 | |
John Stiles | addccaf | 2021-08-02 19:03:30 -0400 | [diff] [blame] | 429 | // Runtime effects always allow narrowing conversions. |
| 430 | if (ProgramConfig::IsRuntimeEffect(kind)) { |
| 431 | settings.fAllowNarrowingConversions = true; |
| 432 | } |
| 433 | |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 434 | this->resetErrors(); |
John Stiles | 10d39d9 | 2021-05-04 16:13:14 -0400 | [diff] [blame] | 435 | fInliner.reset(); |
Brian Osman | 88cda17 | 2020-10-09 12:05:16 -0400 | [diff] [blame] | 436 | |
Ethan Nicholas | dd2fdea | 2021-07-20 15:23:04 -0400 | [diff] [blame] | 437 | settings.fDSLMangling = false; |
Ethan Nicholas | 051aeb7 | 2021-09-24 16:39:19 -0400 | [diff] [blame] | 438 | return DSLParser(this, settings, kind, std::move(text)).program(); |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 439 | } |
| 440 | |
Brian Osman | 0006ad0 | 2020-11-18 15:38:39 -0500 | [diff] [blame] | 441 | bool Compiler::optimize(LoadedModule& module) { |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 442 | SkASSERT(!this->errorCount()); |
Brian Osman | 0006ad0 | 2020-11-18 15:38:39 -0500 | [diff] [blame] | 443 | |
John Stiles | 270cec2 | 2021-02-17 12:59:36 -0500 | [diff] [blame] | 444 | // Create a temporary program configuration with default settings. |
| 445 | ProgramConfig config; |
John Stiles | c876489 | 2021-10-04 11:56:30 -0400 | [diff] [blame] | 446 | config.fIsBuiltinCode = true; |
John Stiles | 270cec2 | 2021-02-17 12:59:36 -0500 | [diff] [blame] | 447 | config.fKind = module.fKind; |
John Stiles | a935c3f | 2021-02-25 10:35:49 -0500 | [diff] [blame] | 448 | AutoProgramConfig autoConfig(fContext, &config); |
John Stiles | 270cec2 | 2021-02-17 12:59:36 -0500 | [diff] [blame] | 449 | |
John Stiles | d120464 | 2021-02-17 16:30:02 -0500 | [diff] [blame] | 450 | // Reset the Inliner. |
John Stiles | 10d39d9 | 2021-05-04 16:13:14 -0400 | [diff] [blame] | 451 | fInliner.reset(); |
John Stiles | 270cec2 | 2021-02-17 12:59:36 -0500 | [diff] [blame] | 452 | |
| 453 | std::unique_ptr<ProgramUsage> usage = Analysis::GetUsage(module); |
Brian Osman | 0006ad0 | 2020-11-18 15:38:39 -0500 | [diff] [blame] | 454 | |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 455 | while (this->errorCount() == 0) { |
Brian Osman | 0006ad0 | 2020-11-18 15:38:39 -0500 | [diff] [blame] | 456 | // Perform inline-candidate analysis and inline any functions deemed suitable. |
John Stiles | 3ff77f4 | 2021-09-06 22:17:58 -0400 | [diff] [blame] | 457 | if (!this->runInliner(module.fElements, module.fSymbols, usage.get())) { |
Brian Osman | 0006ad0 | 2020-11-18 15:38:39 -0500 | [diff] [blame] | 458 | break; |
| 459 | } |
| 460 | } |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 461 | return this->errorCount() == 0; |
Brian Osman | 0006ad0 | 2020-11-18 15:38:39 -0500 | [diff] [blame] | 462 | } |
| 463 | |
Ethan Nicholas | 0054311 | 2018-07-31 09:44:36 -0400 | [diff] [blame] | 464 | bool Compiler::optimize(Program& program) { |
John Stiles | 7247b48 | 2021-03-08 10:40:35 -0500 | [diff] [blame] | 465 | // The optimizer only needs to run when it is enabled. |
| 466 | if (!program.fConfig->fSettings.fOptimize) { |
| 467 | return true; |
| 468 | } |
| 469 | |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 470 | SkASSERT(!this->errorCount()); |
Brian Osman | 010ce6a | 2020-10-19 16:34:10 -0400 | [diff] [blame] | 471 | ProgramUsage* usage = program.fUsage.get(); |
John Stiles | 7954d6c | 2020-09-01 10:53:02 -0400 | [diff] [blame] | 472 | |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 473 | if (this->errorCount() == 0) { |
John Stiles | 87fc657 | 2021-04-01 14:56:34 +0000 | [diff] [blame] | 474 | // Run the inliner only once; it is expensive! Multiple passes can occasionally shake out |
| 475 | // more wins, but it's diminishing returns. |
John Stiles | bcaacec | 2021-10-06 20:06:55 -0400 | [diff] [blame] | 476 | this->runInliner(program.fOwnedElements, program.fSymbols, usage); |
Ethan Nicholas | 34b19c5 | 2020-09-14 11:33:47 -0400 | [diff] [blame] | 477 | |
John Stiles | 8097076 | 2021-09-18 12:42:10 -0400 | [diff] [blame] | 478 | // Unreachable code can confuse some drivers, so it's worth removing. (skia:12012) |
John Stiles | 042a197 | 2021-10-06 20:37:39 -0400 | [diff] [blame] | 479 | Transform::EliminateUnreachableCode(program, usage); |
John Stiles | 8097076 | 2021-09-18 12:42:10 -0400 | [diff] [blame] | 480 | |
John Stiles | 042a197 | 2021-10-06 20:37:39 -0400 | [diff] [blame] | 481 | while (Transform::EliminateDeadFunctions(program, usage)) { |
John Stiles | b666458 | 2021-03-19 09:46:00 -0400 | [diff] [blame] | 482 | // Removing dead functions may cause more functions to become unreferenced. Try again. |
Ethan Nicholas | 34b19c5 | 2020-09-14 11:33:47 -0400 | [diff] [blame] | 483 | } |
John Stiles | a7827d6 | 2021-10-06 16:51:57 -0400 | [diff] [blame] | 484 | while (Transform::EliminateDeadLocalVariables(program, usage)) { |
John Stiles | b666458 | 2021-03-19 09:46:00 -0400 | [diff] [blame] | 485 | // Removing dead variables may cause more variables to become unreferenced. Try again. |
| 486 | } |
John Stiles | 25be58e | 2021-05-20 14:38:40 -0400 | [diff] [blame] | 487 | |
John Stiles | 042a197 | 2021-10-06 20:37:39 -0400 | [diff] [blame] | 488 | Transform::EliminateDeadGlobalVariables(program, usage); |
Ethan Nicholas | 0054311 | 2018-07-31 09:44:36 -0400 | [diff] [blame] | 489 | } |
John Stiles | bb1505f | 2021-02-12 09:17:53 -0500 | [diff] [blame] | 490 | |
John Stiles | 2ecc595 | 2021-09-01 14:41:36 -0400 | [diff] [blame] | 491 | return this->errorCount() == 0; |
| 492 | } |
| 493 | |
John Stiles | 3ff77f4 | 2021-09-06 22:17:58 -0400 | [diff] [blame] | 494 | bool Compiler::runInliner(const std::vector<std::unique_ptr<ProgramElement>>& elements, |
| 495 | std::shared_ptr<SymbolTable> symbols, |
| 496 | ProgramUsage* usage) { |
| 497 | // The program's SymbolTable was taken out of the IRGenerator when the program was bundled, but |
| 498 | // the inliner relies (indirectly) on having a valid SymbolTable in the IRGenerator. |
| 499 | // In particular, inlining can turn a non-optimizable expression like `normalize(myVec)` into |
| 500 | // `normalize(vec2(7))`, which is now optimizable. The optimizer can use DSL to simplify this |
| 501 | // expression--e.g., in the case of normalize, using DSL's Length(). The DSL relies on |
| 502 | // irGenerator.convertIdentifier() to look up `length`. convertIdentifier() needs a valid symbol |
| 503 | // table to find the declaration of `length`. To allow this chain of events to succeed, we |
| 504 | // re-insert the program's symbol table back into the IRGenerator temporarily. |
| 505 | SkASSERT(!fIRGenerator->fSymbolTable); |
| 506 | fIRGenerator->fSymbolTable = symbols; |
| 507 | |
| 508 | bool result = fInliner.analyze(elements, symbols, usage); |
| 509 | |
| 510 | fIRGenerator->fSymbolTable = nullptr; |
| 511 | return result; |
| 512 | } |
| 513 | |
John Stiles | 2ecc595 | 2021-09-01 14:41:36 -0400 | [diff] [blame] | 514 | bool Compiler::finalize(Program& program) { |
| 515 | // Do a pass looking for @if/@switch statements that didn't optimize away, or dangling |
| 516 | // FunctionReference or TypeReference expressions. Report these as errors. |
| 517 | Analysis::VerifyStaticTestsAndExpressions(program); |
| 518 | |
John Stiles | 9d82e61 | 2021-09-03 08:39:54 -0400 | [diff] [blame] | 519 | // Verify that the program conforms to ES2 limitations. |
John Stiles | 2ecc595 | 2021-09-01 14:41:36 -0400 | [diff] [blame] | 520 | if (fContext->fConfig->strictES2Mode() && this->errorCount() == 0) { |
John Stiles | 9d82e61 | 2021-09-03 08:39:54 -0400 | [diff] [blame] | 521 | // Enforce Appendix A, Section 5 of the GLSL ES 1.00 spec -- Indexing. This logic assumes |
| 522 | // that all loops meet the criteria of Section 4, and if they don't, could crash. |
John Stiles | bcaacec | 2021-10-06 20:06:55 -0400 | [diff] [blame] | 523 | for (const auto& pe : program.fOwnedElements) { |
John Stiles | 2ecc595 | 2021-09-01 14:41:36 -0400 | [diff] [blame] | 524 | Analysis::ValidateIndexingForES2(*pe, this->errorReporter()); |
| 525 | } |
John Stiles | 9d82e61 | 2021-09-03 08:39:54 -0400 | [diff] [blame] | 526 | // Verify that the program size is reasonable after unrolling and inlining. This also |
| 527 | // issues errors for static recursion and overly-deep function-call chains. |
John Stiles | 61e5e20 | 2021-09-02 09:56:31 -0400 | [diff] [blame] | 528 | Analysis::CheckProgramUnrolledSize(program); |
John Stiles | bb1505f | 2021-02-12 09:17:53 -0500 | [diff] [blame] | 529 | } |
| 530 | |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 531 | return this->errorCount() == 0; |
Ethan Nicholas | 0054311 | 2018-07-31 09:44:36 -0400 | [diff] [blame] | 532 | } |
| 533 | |
Brian Osman | fb32ddf | 2019-06-18 10:14:20 -0400 | [diff] [blame] | 534 | #if defined(SKSL_STANDALONE) || SK_SUPPORT_GPU |
| 535 | |
Ethan Nicholas | 0054311 | 2018-07-31 09:44:36 -0400 | [diff] [blame] | 536 | bool Compiler::toSPIRV(Program& program, OutputStream& out) { |
Brian Osman | 7a20b5c | 2021-03-15 16:23:33 -0400 | [diff] [blame] | 537 | TRACE_EVENT0("skia.shaders", "SkSL::Compiler::toSPIRV"); |
Ethan Nicholas | b449fff | 2021-08-04 15:06:37 -0400 | [diff] [blame] | 538 | AutoSource as(this, program.fSource->c_str()); |
Brian Salomon | d8d85b9 | 2021-07-07 09:41:17 -0400 | [diff] [blame] | 539 | ProgramSettings settings; |
| 540 | settings.fDSLUseMemoryPool = false; |
| 541 | dsl::Start(this, program.fConfig->fKind, settings); |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 542 | dsl::SetErrorReporter(&fErrorReporter); |
Ethan Nicholas | c845272 | 2021-10-07 10:47:32 -0400 | [diff] [blame] | 543 | ThreadContext::IRGenerator().fSymbolTable = program.fSymbols; |
Ethan Nicholas | a6ae1f7 | 2017-03-16 09:56:54 -0400 | [diff] [blame] | 544 | #ifdef SK_ENABLE_SPIRV_VALIDATION |
Ethan Nicholas | 0df1b04 | 2017-03-31 13:56:23 -0400 | [diff] [blame] | 545 | StringStream buffer; |
Ethan Nicholas | 3abc6c6 | 2021-08-13 11:20:09 -0400 | [diff] [blame] | 546 | SPIRVCodeGenerator cg(fContext.get(), &program, &buffer); |
Ethan Nicholas | a6ae1f7 | 2017-03-16 09:56:54 -0400 | [diff] [blame] | 547 | bool result = cg.generateCode(); |
John Stiles | 270cec2 | 2021-02-17 12:59:36 -0500 | [diff] [blame] | 548 | if (result && program.fConfig->fSettings.fValidateSPIRV) { |
Ethan Nicholas | a6ae1f7 | 2017-03-16 09:56:54 -0400 | [diff] [blame] | 549 | spvtools::SpirvTools tools(SPV_ENV_VULKAN_1_0); |
Ethan Nicholas | 762466e | 2017-06-29 10:03:38 -0400 | [diff] [blame] | 550 | const String& data = buffer.str(); |
Ethan Nicholas | d9d33c3 | 2018-06-12 11:05:59 -0400 | [diff] [blame] | 551 | SkASSERT(0 == data.size() % 4); |
Brian Osman | 8d09d4a | 2020-11-24 15:51:06 -0500 | [diff] [blame] | 552 | String errors; |
| 553 | auto dumpmsg = [&errors](spv_message_level_t, const char*, const spv_position_t&, |
| 554 | const char* m) { |
| 555 | errors.appendf("SPIR-V validation error: %s\n", m); |
Ethan Nicholas | a6ae1f7 | 2017-03-16 09:56:54 -0400 | [diff] [blame] | 556 | }; |
| 557 | tools.SetMessageConsumer(dumpmsg); |
Brian Osman | 8d09d4a | 2020-11-24 15:51:06 -0500 | [diff] [blame] | 558 | |
| 559 | // Verify that the SPIR-V we produced is valid. At runtime, we will abort() with a message |
| 560 | // explaining the error. In standalone mode (skslc), we will send the message, plus the |
| 561 | // entire disassembled SPIR-V (for easier context & debugging) as *our* error message. |
| 562 | result = tools.Validate((const uint32_t*) data.c_str(), data.size() / 4); |
| 563 | |
| 564 | if (!result) { |
| 565 | #if defined(SKSL_STANDALONE) |
| 566 | // Convert the string-stream to a SPIR-V disassembly. |
| 567 | std::string disassembly; |
| 568 | if (tools.Disassemble((const uint32_t*)data.data(), data.size() / 4, &disassembly)) { |
| 569 | errors.append(disassembly); |
| 570 | } |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 571 | this->errorReporter().error(-1, errors); |
Ethan Nicholas | c973d26 | 2021-09-17 16:10:29 -0400 | [diff] [blame] | 572 | this->errorReporter().reportPendingErrors(PositionInfo()); |
Brian Osman | 8d09d4a | 2020-11-24 15:51:06 -0500 | [diff] [blame] | 573 | #else |
| 574 | SkDEBUGFAILF("%s", errors.c_str()); |
| 575 | #endif |
| 576 | } |
Ethan Nicholas | 762466e | 2017-06-29 10:03:38 -0400 | [diff] [blame] | 577 | out.write(data.c_str(), data.size()); |
Ethan Nicholas | a6ae1f7 | 2017-03-16 09:56:54 -0400 | [diff] [blame] | 578 | } |
| 579 | #else |
Ethan Nicholas | 3abc6c6 | 2021-08-13 11:20:09 -0400 | [diff] [blame] | 580 | SPIRVCodeGenerator cg(fContext.get(), &program, &out); |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 581 | bool result = cg.generateCode(); |
Ethan Nicholas | a6ae1f7 | 2017-03-16 09:56:54 -0400 | [diff] [blame] | 582 | #endif |
Brian Salomon | d8d85b9 | 2021-07-07 09:41:17 -0400 | [diff] [blame] | 583 | dsl::End(); |
Ethan Nicholas | ce33f10 | 2016-12-09 17:22:59 -0500 | [diff] [blame] | 584 | return result; |
| 585 | } |
| 586 | |
Ethan Nicholas | 0054311 | 2018-07-31 09:44:36 -0400 | [diff] [blame] | 587 | bool Compiler::toSPIRV(Program& program, String* out) { |
Ethan Nicholas | 0df1b04 | 2017-03-31 13:56:23 -0400 | [diff] [blame] | 588 | StringStream buffer; |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 589 | bool result = this->toSPIRV(program, buffer); |
| 590 | if (result) { |
Ethan Nicholas | 762466e | 2017-06-29 10:03:38 -0400 | [diff] [blame] | 591 | *out = buffer.str(); |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 592 | } |
| 593 | return result; |
| 594 | } |
| 595 | |
Ethan Nicholas | 0054311 | 2018-07-31 09:44:36 -0400 | [diff] [blame] | 596 | bool Compiler::toGLSL(Program& program, OutputStream& out) { |
Brian Osman | 7a20b5c | 2021-03-15 16:23:33 -0400 | [diff] [blame] | 597 | TRACE_EVENT0("skia.shaders", "SkSL::Compiler::toGLSL"); |
Ethan Nicholas | b449fff | 2021-08-04 15:06:37 -0400 | [diff] [blame] | 598 | AutoSource as(this, program.fSource->c_str()); |
Ethan Nicholas | 3abc6c6 | 2021-08-13 11:20:09 -0400 | [diff] [blame] | 599 | GLSLCodeGenerator cg(fContext.get(), &program, &out); |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 600 | bool result = cg.generateCode(); |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 601 | return result; |
| 602 | } |
| 603 | |
Ethan Nicholas | 0054311 | 2018-07-31 09:44:36 -0400 | [diff] [blame] | 604 | bool Compiler::toGLSL(Program& program, String* out) { |
Ethan Nicholas | 0df1b04 | 2017-03-31 13:56:23 -0400 | [diff] [blame] | 605 | StringStream buffer; |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 606 | bool result = this->toGLSL(program, buffer); |
| 607 | if (result) { |
Ethan Nicholas | 762466e | 2017-06-29 10:03:38 -0400 | [diff] [blame] | 608 | *out = buffer.str(); |
Ethan Nicholas | 941e7e2 | 2016-12-12 15:33:30 -0500 | [diff] [blame] | 609 | } |
| 610 | return result; |
| 611 | } |
| 612 | |
Brian Osman | c024391 | 2020-02-19 15:35:26 -0500 | [diff] [blame] | 613 | bool Compiler::toHLSL(Program& program, String* out) { |
| 614 | String spirv; |
| 615 | if (!this->toSPIRV(program, &spirv)) { |
| 616 | return false; |
| 617 | } |
| 618 | |
| 619 | return SPIRVtoHLSL(spirv, out); |
| 620 | } |
| 621 | |
Ethan Nicholas | 0054311 | 2018-07-31 09:44:36 -0400 | [diff] [blame] | 622 | bool Compiler::toMetal(Program& program, OutputStream& out) { |
Brian Osman | 7a20b5c | 2021-03-15 16:23:33 -0400 | [diff] [blame] | 623 | TRACE_EVENT0("skia.shaders", "SkSL::Compiler::toMetal"); |
Ethan Nicholas | b449fff | 2021-08-04 15:06:37 -0400 | [diff] [blame] | 624 | AutoSource as(this, program.fSource->c_str()); |
Ethan Nicholas | 3abc6c6 | 2021-08-13 11:20:09 -0400 | [diff] [blame] | 625 | MetalCodeGenerator cg(fContext.get(), &program, &out); |
Ethan Nicholas | cc30577 | 2017-10-13 16:17:45 -0400 | [diff] [blame] | 626 | bool result = cg.generateCode(); |
Ethan Nicholas | cc30577 | 2017-10-13 16:17:45 -0400 | [diff] [blame] | 627 | return result; |
| 628 | } |
| 629 | |
Ethan Nicholas | 0054311 | 2018-07-31 09:44:36 -0400 | [diff] [blame] | 630 | bool Compiler::toMetal(Program& program, String* out) { |
Timothy Liang | b8eeb80 | 2018-07-23 16:46:16 -0400 | [diff] [blame] | 631 | StringStream buffer; |
| 632 | bool result = this->toMetal(program, buffer); |
| 633 | if (result) { |
| 634 | *out = buffer.str(); |
| 635 | } |
| 636 | return result; |
| 637 | } |
| 638 | |
Ethan Nicholas | 2a479a5 | 2020-08-18 16:29:45 -0400 | [diff] [blame] | 639 | #endif // defined(SKSL_STANDALONE) || SK_SUPPORT_GPU |
Brian Osman | 2e29ab5 | 2019-09-20 12:19:11 -0400 | [diff] [blame] | 640 | |
Ethan Nicholas | 3272412 | 2021-09-07 13:49:07 -0400 | [diff] [blame] | 641 | void Compiler::handleError(skstd::string_view msg, PositionInfo pos) { |
Ethan Nicholas | a40ddcd | 2021-08-06 09:17:18 -0400 | [diff] [blame] | 642 | fErrorText += "error: " + (pos.line() >= 1 ? to_string(pos.line()) + ": " : "") + msg + "\n"; |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 643 | } |
| 644 | |
Ethan Nicholas | 9504614 | 2021-01-07 10:57:27 -0500 | [diff] [blame] | 645 | String Compiler::errorText(bool showCount) { |
| 646 | if (showCount) { |
| 647 | this->writeErrorCount(); |
| 648 | } |
Ethan Nicholas | 0df1b04 | 2017-03-31 13:56:23 -0400 | [diff] [blame] | 649 | String result = fErrorText; |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 650 | this->resetErrors(); |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 651 | return result; |
| 652 | } |
| 653 | |
| 654 | void Compiler::writeErrorCount() { |
Ethan Nicholas | 4a5e22a | 2021-08-13 17:29:51 -0400 | [diff] [blame] | 655 | int count = this->errorCount(); |
| 656 | if (count) { |
| 657 | fErrorText += to_string(count) + " error"; |
| 658 | if (count > 1) { |
ethannicholas | b3058bd | 2016-07-01 08:22:01 -0700 | [diff] [blame] | 659 | fErrorText += "s"; |
| 660 | } |
| 661 | fErrorText += "\n"; |
| 662 | } |
| 663 | } |
| 664 | |
John Stiles | a6841be | 2020-08-06 14:11:56 -0400 | [diff] [blame] | 665 | } // namespace SkSL |