| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2016 Google Inc. | 2 * Copyright 2016 Google Inc. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license that can be | 4 * Use of this source code is governed by a BSD-style license that can be |
| 5 * found in the LICENSE file. | 5 * found in the LICENSE file. |
| 6 */ | 6 */ |
| 7 | 7 |
| 8 #ifndef SKSL_SPIRVCODEGENERATOR | 8 #ifndef SKSL_SPIRVCODEGENERATOR |
| 9 #define SKSL_SPIRVCODEGENERATOR | 9 #define SKSL_SPIRVCODEGENERATOR |
| 10 | 10 |
| 11 #include <sstream> | 11 #include <sstream> |
| 12 #include <stack> | 12 #include <stack> |
| 13 #include <tuple> | 13 #include <tuple> |
| 14 #include <unordered_map> | 14 #include <unordered_map> |
| 15 | 15 |
| 16 #include "SkSLCodeGenerator.h" | 16 #include "SkSLCodeGenerator.h" |
| 17 #include "SkSLMemoryLayout.h" | |
| 18 #include "ir/SkSLBinaryExpression.h" | 17 #include "ir/SkSLBinaryExpression.h" |
| 19 #include "ir/SkSLBoolLiteral.h" | 18 #include "ir/SkSLBoolLiteral.h" |
| 20 #include "ir/SkSLConstructor.h" | 19 #include "ir/SkSLConstructor.h" |
| 21 #include "ir/SkSLFloatLiteral.h" | 20 #include "ir/SkSLFloatLiteral.h" |
| 22 #include "ir/SkSLIfStatement.h" | 21 #include "ir/SkSLIfStatement.h" |
| 23 #include "ir/SkSLIndexExpression.h" | 22 #include "ir/SkSLIndexExpression.h" |
| 24 #include "ir/SkSLInterfaceBlock.h" | 23 #include "ir/SkSLInterfaceBlock.h" |
| 25 #include "ir/SkSLIntLiteral.h" | 24 #include "ir/SkSLIntLiteral.h" |
| 26 #include "ir/SkSLFieldAccess.h" | 25 #include "ir/SkSLFieldAccess.h" |
| 27 #include "ir/SkSLForStatement.h" | 26 #include "ir/SkSLForStatement.h" |
| (...skipping 29 matching lines...) Expand all Loading... |
| 57 // by a pointer (e.g. vector swizzles), returns 0. | 56 // by a pointer (e.g. vector swizzles), returns 0. |
| 58 virtual SpvId getPointer() = 0; | 57 virtual SpvId getPointer() = 0; |
| 59 | 58 |
| 60 virtual SpvId load(std::ostream& out) = 0; | 59 virtual SpvId load(std::ostream& out) = 0; |
| 61 | 60 |
| 62 virtual void store(SpvId value, std::ostream& out) = 0; | 61 virtual void store(SpvId value, std::ostream& out) = 0; |
| 63 }; | 62 }; |
| 64 | 63 |
| 65 SPIRVCodeGenerator(const Context* context) | 64 SPIRVCodeGenerator(const Context* context) |
| 66 : fContext(*context) | 65 : fContext(*context) |
| 67 , fDefaultLayout(MemoryLayout::k140_Standard) | |
| 68 , fCapabilities(1 << SpvCapabilityShader) | 66 , fCapabilities(1 << SpvCapabilityShader) |
| 69 , fIdCount(1) | 67 , fIdCount(1) |
| 70 , fBoolTrue(0) | 68 , fBoolTrue(0) |
| 71 , fBoolFalse(0) | 69 , fBoolFalse(0) |
| 72 , fCurrentBlock(0) { | 70 , fCurrentBlock(0) { |
| 73 this->setupIntrinsics(); | 71 this->setupIntrinsics(); |
| 74 } | 72 } |
| 75 | 73 |
| 76 void generateCode(const Program& program, std::ostream& out) override; | 74 void generateCode(const Program& program, std::ostream& out) override; |
| 77 | 75 |
| (...skipping 10 matching lines...) Expand all Loading... |
| 88 kTexture2D_SpecialIntrinsic, | 86 kTexture2D_SpecialIntrinsic, |
| 89 kTextureProj_SpecialIntrinsic | 87 kTextureProj_SpecialIntrinsic |
| 90 }; | 88 }; |
| 91 | 89 |
| 92 void setupIntrinsics(); | 90 void setupIntrinsics(); |
| 93 | 91 |
| 94 SpvId nextId(); | 92 SpvId nextId(); |
| 95 | 93 |
| 96 SpvId getType(const Type& type); | 94 SpvId getType(const Type& type); |
| 97 | 95 |
| 98 SpvId getType(const Type& type, const MemoryLayout& layout); | |
| 99 | |
| 100 SpvId getFunctionType(const FunctionDeclaration& function); | 96 SpvId getFunctionType(const FunctionDeclaration& function); |
| 101 | 97 |
| 102 SpvId getPointerType(const Type& type, SpvStorageClass_ storageClass); | 98 SpvId getPointerType(const Type& type, SpvStorageClass_ storageClass); |
| 103 | 99 |
| 104 SpvId getPointerType(const Type& type, const MemoryLayout& layout, | |
| 105 SpvStorageClass_ storageClass); | |
| 106 | |
| 107 std::vector<SpvId> getAccessChain(const Expression& expr, std::ostream& out)
; | 100 std::vector<SpvId> getAccessChain(const Expression& expr, std::ostream& out)
; |
| 108 | 101 |
| 109 void writeLayout(const Layout& layout, SpvId target); | 102 void writeLayout(const Layout& layout, SpvId target); |
| 110 | 103 |
| 111 void writeLayout(const Layout& layout, SpvId target, int member); | 104 void writeLayout(const Layout& layout, SpvId target, int member); |
| 112 | 105 |
| 113 void writeStruct(const Type& type, const MemoryLayout& layout, SpvId resultI
d); | 106 void writeStruct(const Type& type, SpvId resultId); |
| 114 | 107 |
| 115 void writeProgramElement(const ProgramElement& pe, std::ostream& out); | 108 void writeProgramElement(const ProgramElement& pe, std::ostream& out); |
| 116 | 109 |
| 117 SpvId writeInterfaceBlock(const InterfaceBlock& intf); | 110 SpvId writeInterfaceBlock(const InterfaceBlock& intf); |
| 118 | 111 |
| 119 SpvId writeFunctionStart(const FunctionDeclaration& f, std::ostream& out); | 112 SpvId writeFunctionStart(const FunctionDeclaration& f, std::ostream& out); |
| 120 | 113 |
| 121 SpvId writeFunctionDeclaration(const FunctionDeclaration& f, std::ostream& o
ut); | 114 SpvId writeFunctionDeclaration(const FunctionDeclaration& f, std::ostream& o
ut); |
| 122 | 115 |
| 123 SpvId writeFunction(const FunctionDefinition& f, std::ostream& out); | 116 SpvId writeFunction(const FunctionDefinition& f, std::ostream& out); |
| (...skipping 105 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 229 int32_t word5, int32_t word6, std::ostream& out); | 222 int32_t word5, int32_t word6, std::ostream& out); |
| 230 | 223 |
| 231 void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t w
ord3, int32_t word4, | 224 void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t w
ord3, int32_t word4, |
| 232 int32_t word5, int32_t word6, int32_t word7, std::ostr
eam& out); | 225 int32_t word5, int32_t word6, int32_t word7, std::ostr
eam& out); |
| 233 | 226 |
| 234 void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t w
ord3, int32_t word4, | 227 void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t w
ord3, int32_t word4, |
| 235 int32_t word5, int32_t word6, int32_t word7, int32_t w
ord8, | 228 int32_t word5, int32_t word6, int32_t word7, int32_t w
ord8, |
| 236 std::ostream& out); | 229 std::ostream& out); |
| 237 | 230 |
| 238 const Context& fContext; | 231 const Context& fContext; |
| 239 const MemoryLayout fDefaultLayout; | |
| 240 | 232 |
| 241 uint64_t fCapabilities; | 233 uint64_t fCapabilities; |
| 242 SpvId fIdCount; | 234 SpvId fIdCount; |
| 243 SpvId fGLSLExtendedInstructions; | 235 SpvId fGLSLExtendedInstructions; |
| 244 typedef std::tuple<IntrinsicKind, int32_t, int32_t, int32_t, int32_t> Intrin
sic; | 236 typedef std::tuple<IntrinsicKind, int32_t, int32_t, int32_t, int32_t> Intrin
sic; |
| 245 std::unordered_map<std::string, Intrinsic> fIntrinsicMap; | 237 std::unordered_map<std::string, Intrinsic> fIntrinsicMap; |
| 246 std::unordered_map<const FunctionDeclaration*, SpvId> fFunctionMap; | 238 std::unordered_map<const FunctionDeclaration*, SpvId> fFunctionMap; |
| 247 std::unordered_map<const Variable*, SpvId> fVariableMap; | 239 std::unordered_map<const Variable*, SpvId> fVariableMap; |
| 248 std::unordered_map<const Variable*, int32_t> fInterfaceBlockMap; | 240 std::unordered_map<const Variable*, int32_t> fInterfaceBlockMap; |
| 249 std::unordered_map<std::string, SpvId> fTypeMap; | 241 std::unordered_map<std::string, SpvId> fTypeMap; |
| (...skipping 16 matching lines...) Expand all Loading... |
| 266 std::stack<SpvId> fBreakTarget; | 258 std::stack<SpvId> fBreakTarget; |
| 267 std::stack<SpvId> fContinueTarget; | 259 std::stack<SpvId> fContinueTarget; |
| 268 | 260 |
| 269 friend class PointerLValue; | 261 friend class PointerLValue; |
| 270 friend class SwizzleLValue; | 262 friend class SwizzleLValue; |
| 271 }; | 263 }; |
| 272 | 264 |
| 273 } | 265 } |
| 274 | 266 |
| 275 #endif | 267 #endif |
| OLD | NEW |