blob: a2456cece9cfcbb2cb0b325326e8cb132f636105 [file] [log] [blame]
ethannicholasb3058bd2016-07-01 08:22:01 -07001/*
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 */
Ethan Nicholas11d53972016-11-28 11:23:23 -05007
Mike Kleinc0bd9f92019-04-23 12:05:21 -05008#include "src/sksl/SkSLIRGenerator.h"
ethannicholasb3058bd2016-07-01 08:22:01 -07009
10#include "limits.h"
Ethan Nicholasaf197692017-02-27 13:26:45 -050011#include <unordered_set>
ethannicholasb3058bd2016-07-01 08:22:01 -070012
Mike Kleinc0bd9f92019-04-23 12:05:21 -050013#include "src/sksl/SkSLCompiler.h"
14#include "src/sksl/SkSLParser.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050015#include "src/sksl/ir/SkSLBinaryExpression.h"
16#include "src/sksl/ir/SkSLBoolLiteral.h"
17#include "src/sksl/ir/SkSLBreakStatement.h"
18#include "src/sksl/ir/SkSLConstructor.h"
19#include "src/sksl/ir/SkSLContinueStatement.h"
20#include "src/sksl/ir/SkSLDiscardStatement.h"
21#include "src/sksl/ir/SkSLDoStatement.h"
22#include "src/sksl/ir/SkSLEnum.h"
23#include "src/sksl/ir/SkSLExpressionStatement.h"
Ethan Nicholas9e6a3932019-05-17 16:31:21 -040024#include "src/sksl/ir/SkSLExternalFunctionCall.h"
Ethan Nicholas91164d12019-05-15 15:29:54 -040025#include "src/sksl/ir/SkSLExternalValueReference.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050026#include "src/sksl/ir/SkSLField.h"
27#include "src/sksl/ir/SkSLFieldAccess.h"
28#include "src/sksl/ir/SkSLFloatLiteral.h"
29#include "src/sksl/ir/SkSLForStatement.h"
30#include "src/sksl/ir/SkSLFunctionCall.h"
31#include "src/sksl/ir/SkSLFunctionDeclaration.h"
32#include "src/sksl/ir/SkSLFunctionDefinition.h"
33#include "src/sksl/ir/SkSLFunctionReference.h"
34#include "src/sksl/ir/SkSLIfStatement.h"
35#include "src/sksl/ir/SkSLIndexExpression.h"
36#include "src/sksl/ir/SkSLIntLiteral.h"
37#include "src/sksl/ir/SkSLInterfaceBlock.h"
38#include "src/sksl/ir/SkSLLayout.h"
Chris Daltonb0fd4b12019-10-29 13:41:22 -060039#include "src/sksl/ir/SkSLNop.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050040#include "src/sksl/ir/SkSLNullLiteral.h"
41#include "src/sksl/ir/SkSLPostfixExpression.h"
42#include "src/sksl/ir/SkSLPrefixExpression.h"
43#include "src/sksl/ir/SkSLReturnStatement.h"
44#include "src/sksl/ir/SkSLSetting.h"
45#include "src/sksl/ir/SkSLSwitchCase.h"
46#include "src/sksl/ir/SkSLSwitchStatement.h"
47#include "src/sksl/ir/SkSLSwizzle.h"
48#include "src/sksl/ir/SkSLTernaryExpression.h"
49#include "src/sksl/ir/SkSLUnresolvedFunction.h"
50#include "src/sksl/ir/SkSLVarDeclarations.h"
51#include "src/sksl/ir/SkSLVarDeclarationsStatement.h"
52#include "src/sksl/ir/SkSLVariable.h"
53#include "src/sksl/ir/SkSLVariableReference.h"
54#include "src/sksl/ir/SkSLWhileStatement.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070055
56namespace SkSL {
57
58class AutoSymbolTable {
59public:
Ethan Nicholas11d53972016-11-28 11:23:23 -050060 AutoSymbolTable(IRGenerator* ir)
ethannicholasb3058bd2016-07-01 08:22:01 -070061 : fIR(ir)
62 , fPrevious(fIR->fSymbolTable) {
63 fIR->pushSymbolTable();
64 }
65
66 ~AutoSymbolTable() {
67 fIR->popSymbolTable();
Ethan Nicholasd9d33c32018-06-12 11:05:59 -040068 SkASSERT(fPrevious == fIR->fSymbolTable);
ethannicholasb3058bd2016-07-01 08:22:01 -070069 }
70
71 IRGenerator* fIR;
72 std::shared_ptr<SymbolTable> fPrevious;
73};
74
ethannicholas22f939e2016-10-13 13:25:34 -070075class AutoLoopLevel {
76public:
Ethan Nicholas11d53972016-11-28 11:23:23 -050077 AutoLoopLevel(IRGenerator* ir)
ethannicholas22f939e2016-10-13 13:25:34 -070078 : fIR(ir) {
79 fIR->fLoopLevel++;
80 }
81
82 ~AutoLoopLevel() {
83 fIR->fLoopLevel--;
84 }
85
86 IRGenerator* fIR;
87};
88
Ethan Nicholasaf197692017-02-27 13:26:45 -050089class AutoSwitchLevel {
90public:
91 AutoSwitchLevel(IRGenerator* ir)
92 : fIR(ir) {
93 fIR->fSwitchLevel++;
94 }
95
96 ~AutoSwitchLevel() {
97 fIR->fSwitchLevel--;
98 }
99
100 IRGenerator* fIR;
101};
102
Ethan Nicholas11d53972016-11-28 11:23:23 -0500103IRGenerator::IRGenerator(const Context* context, std::shared_ptr<SymbolTable> symbolTable,
ethannicholasb3058bd2016-07-01 08:22:01 -0700104 ErrorReporter& errorReporter)
ethannicholasd598f792016-07-25 10:08:54 -0700105: fContext(*context)
106, fCurrentFunction(nullptr)
Ethan Nicholas762466e2017-06-29 10:03:38 -0400107, fRootSymbolTable(symbolTable)
108, fSymbolTable(symbolTable)
ethannicholas22f939e2016-10-13 13:25:34 -0700109, fLoopLevel(0)
Ethan Nicholasaf197692017-02-27 13:26:45 -0500110, fSwitchLevel(0)
ethannicholasd598f792016-07-25 10:08:54 -0700111, fErrors(errorReporter) {}
ethannicholasb3058bd2016-07-01 08:22:01 -0700112
113void IRGenerator::pushSymbolTable() {
Ethan Nicholas8feeff92017-03-30 14:11:58 -0400114 fSymbolTable.reset(new SymbolTable(std::move(fSymbolTable), &fErrors));
ethannicholasb3058bd2016-07-01 08:22:01 -0700115}
116
117void IRGenerator::popSymbolTable() {
118 fSymbolTable = fSymbolTable->fParent;
119}
120
Ethan Nicholas762466e2017-06-29 10:03:38 -0400121static void fill_caps(const SKSL_CAPS_CLASS& caps,
122 std::unordered_map<String, Program::Settings::Value>* capsMap) {
Brian Salomon23356442018-11-30 15:33:19 -0500123#define CAP(name) \
124 capsMap->insert(std::make_pair(String(#name), Program::Settings::Value(caps.name())))
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500125 CAP(fbFetchSupport);
Brian Salomond4013302018-04-04 13:58:33 +0000126 CAP(fbFetchNeedsCustomOutput);
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500127 CAP(flatInterpolationSupport);
128 CAP(noperspectiveInterpolationSupport);
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500129 CAP(externalTextureSupport);
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500130 CAP(mustEnableAdvBlendEqs);
131 CAP(mustEnableSpecificAdvBlendEqs);
132 CAP(mustDeclareFragmentShaderOutput);
Michael Ludwig4f94ef62018-09-12 15:22:16 -0400133 CAP(mustDoOpBetweenFloorAndAbs);
Brian Salomonf8c187c2019-12-19 14:41:57 -0500134 CAP(mustGuardDivisionEvenAfterExplicitZeroCheck);
135 CAP(inBlendModesFailRandomlyForAllZeroVec);
Michael Ludwig24d438b2018-09-12 15:22:50 -0400136 CAP(atan2ImplementedAsAtanYOverX);
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500137 CAP(canUseAnyFunctionInShader);
Chris Dalton47c8ed32017-11-15 18:27:09 -0700138 CAP(floatIs32Bits);
Ethan Nicholas07990de2017-07-18 09:47:43 -0400139 CAP(integerSupport);
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500140#undef CAP
141}
142
Ethan Nicholas3c6ae622018-04-24 13:06:09 -0400143void IRGenerator::start(const Program::Settings* settings,
144 std::vector<std::unique_ptr<ProgramElement>>* inherited) {
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500145 fSettings = settings;
146 fCapsMap.clear();
147 if (settings->fCaps) {
148 fill_caps(*settings->fCaps, &fCapsMap);
Ethan Nicholas00543112018-07-31 09:44:36 -0400149 } else {
150 fCapsMap.insert(std::make_pair(String("integerSupport"),
151 Program::Settings::Value(true)));
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500152 }
Ethan Nicholas3605ace2016-11-21 15:59:48 -0500153 this->pushSymbolTable();
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400154 fInvocations = -1;
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500155 fInputs.reset();
Robert Phillipsfe8da172018-01-24 14:52:02 +0000156 fSkPerVertex = nullptr;
157 fRTAdjust = nullptr;
158 fRTAdjustInterfaceBlock = nullptr;
Ethan Nicholas3c6ae622018-04-24 13:06:09 -0400159 if (inherited) {
160 for (const auto& e : *inherited) {
161 if (e->fKind == ProgramElement::kInterfaceBlock_Kind) {
162 InterfaceBlock& intf = (InterfaceBlock&) *e;
163 if (intf.fVariable.fName == Compiler::PERVERTEX_NAME) {
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400164 SkASSERT(!fSkPerVertex);
Ethan Nicholas3c6ae622018-04-24 13:06:09 -0400165 fSkPerVertex = &intf.fVariable;
166 }
167 }
168 }
169 }
Ethan Nicholasdb80f692019-11-22 14:06:12 -0500170 SkASSERT(fIntrinsics);
171 for (auto& pair : *fIntrinsics) {
172 pair.second.second = false;
173 }
Ethan Nicholas3605ace2016-11-21 15:59:48 -0500174}
175
Ethan Nicholasfc994162019-06-06 10:04:27 -0400176std::unique_ptr<Extension> IRGenerator::convertExtension(int offset, StringFragment name) {
177 return std::unique_ptr<Extension>(new Extension(offset, name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700178}
179
Ethan Nicholasfc994162019-06-06 10:04:27 -0400180void IRGenerator::finish() {
181 this->popSymbolTable();
182 fSettings = nullptr;
183}
184
185std::unique_ptr<Statement> IRGenerator::convertStatement(const ASTNode& statement) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700186 switch (statement.fKind) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400187 case ASTNode::Kind::kBlock:
188 return this->convertBlock(statement);
189 case ASTNode::Kind::kVarDeclarations:
190 return this->convertVarDeclarationStatement(statement);
191 case ASTNode::Kind::kIf:
192 return this->convertIf(statement);
193 case ASTNode::Kind::kFor:
194 return this->convertFor(statement);
195 case ASTNode::Kind::kWhile:
196 return this->convertWhile(statement);
197 case ASTNode::Kind::kDo:
198 return this->convertDo(statement);
199 case ASTNode::Kind::kSwitch:
200 return this->convertSwitch(statement);
201 case ASTNode::Kind::kReturn:
202 return this->convertReturn(statement);
203 case ASTNode::Kind::kBreak:
204 return this->convertBreak(statement);
205 case ASTNode::Kind::kContinue:
206 return this->convertContinue(statement);
207 case ASTNode::Kind::kDiscard:
208 return this->convertDiscard(statement);
209 default:
210 // it's an expression
211 std::unique_ptr<Statement> result = this->convertExpressionStatement(statement);
Robert Phillipsfe8da172018-01-24 14:52:02 +0000212 if (fRTAdjust && Program::kGeometry_Kind == fKind) {
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400213 SkASSERT(result->fKind == Statement::kExpression_Kind);
Robert Phillipsfe8da172018-01-24 14:52:02 +0000214 Expression& expr = *((ExpressionStatement&) *result).fExpression;
215 if (expr.fKind == Expression::kFunctionCall_Kind) {
216 FunctionCall& fc = (FunctionCall&) expr;
217 if (fc.fFunction.fBuiltin && fc.fFunction.fName == "EmitVertex") {
218 std::vector<std::unique_ptr<Statement>> statements;
219 statements.push_back(getNormalizeSkPositionCode());
220 statements.push_back(std::move(result));
221 return std::unique_ptr<Block>(new Block(statement.fOffset,
222 std::move(statements),
223 fSymbolTable));
224 }
225 }
226 }
227 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700228 }
229}
230
Ethan Nicholasfc994162019-06-06 10:04:27 -0400231std::unique_ptr<Block> IRGenerator::convertBlock(const ASTNode& block) {
232 SkASSERT(block.fKind == ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -0700233 AutoSymbolTable table(this);
234 std::vector<std::unique_ptr<Statement>> statements;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400235 for (const auto& child : block) {
236 std::unique_ptr<Statement> statement = this->convertStatement(child);
ethannicholasb3058bd2016-07-01 08:22:01 -0700237 if (!statement) {
238 return nullptr;
239 }
240 statements.push_back(std::move(statement));
241 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700242 return std::unique_ptr<Block>(new Block(block.fOffset, std::move(statements), fSymbolTable));
ethannicholasb3058bd2016-07-01 08:22:01 -0700243}
244
Ethan Nicholasfc994162019-06-06 10:04:27 -0400245std::unique_ptr<Statement> IRGenerator::convertVarDeclarationStatement(const ASTNode& s) {
246 SkASSERT(s.fKind == ASTNode::Kind::kVarDeclarations);
247 auto decl = this->convertVarDeclarations(s, Variable::kLocal_Storage);
ethannicholasb3058bd2016-07-01 08:22:01 -0700248 if (!decl) {
249 return nullptr;
250 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700251 return std::unique_ptr<Statement>(new VarDeclarationsStatement(std::move(decl)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700252}
253
Ethan Nicholasfc994162019-06-06 10:04:27 -0400254std::unique_ptr<VarDeclarations> IRGenerator::convertVarDeclarations(const ASTNode& decls,
ethannicholas14fe8cc2016-09-07 13:37:16 -0700255 Variable::Storage storage) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400256 SkASSERT(decls.fKind == ASTNode::Kind::kVarDeclarations);
257 auto iter = decls.begin();
258 const Modifiers& modifiers = iter++->getModifiers();
259 const ASTNode& rawType = *(iter++);
Ethan Nicholas82a62d22017-11-07 14:42:10 +0000260 std::vector<std::unique_ptr<VarDeclaration>> variables;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400261 const Type* baseType = this->convertType(rawType);
ethannicholasb3058bd2016-07-01 08:22:01 -0700262 if (!baseType) {
263 return nullptr;
264 }
Brian Osman2fe83fe2019-12-16 13:17:59 -0500265 if (fKind != Program::kFragmentProcessor_Kind) {
266 if ((modifiers.fFlags & Modifiers::kIn_Flag) &&
267 baseType->kind() == Type::Kind::kMatrix_Kind) {
268 fErrors.error(decls.fOffset, "'in' variables may not have matrix type");
269 }
Brian Osman088913a2019-12-19 15:44:56 -0500270 if ((modifiers.fFlags & Modifiers::kIn_Flag) &&
271 (modifiers.fFlags & Modifiers::kUniform_Flag)) {
272 fErrors.error(decls.fOffset,
273 "'in uniform' variables only permitted within fragment processors");
274 }
Brian Osman2fe83fe2019-12-16 13:17:59 -0500275 if (modifiers.fLayout.fWhen.fLength) {
276 fErrors.error(decls.fOffset, "'when' is only permitted within fragment processors");
277 }
278 if (modifiers.fLayout.fFlags & Layout::kTracked_Flag) {
279 fErrors.error(decls.fOffset, "'tracked' is only permitted within fragment processors");
280 }
281 if (modifiers.fLayout.fCType != Layout::CType::kDefault) {
282 fErrors.error(decls.fOffset, "'ctype' is only permitted within fragment processors");
283 }
284 if (modifiers.fLayout.fKey) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400285 fErrors.error(decls.fOffset, "'key' is only permitted within fragment processors");
Ethan Nicholasbcd51e82019-04-09 10:40:41 -0400286 }
Brian Osman2fe83fe2019-12-16 13:17:59 -0500287 }
288 if (modifiers.fLayout.fKey && (modifiers.fFlags & Modifiers::kUniform_Flag)) {
289 fErrors.error(decls.fOffset, "'key' is not permitted on 'uniform' variables");
Ethan Nicholasbcd51e82019-04-09 10:40:41 -0400290 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400291 for (; iter != decls.end(); ++iter) {
292 const ASTNode& varDecl = *iter;
293 if (modifiers.fLayout.fLocation == 0 && modifiers.fLayout.fIndex == 0 &&
294 (modifiers.fFlags & Modifiers::kOut_Flag) && fKind == Program::kFragment_Kind &&
295 varDecl.getVarData().fName != "sk_FragColor") {
296 fErrors.error(varDecl.fOffset,
Ethan Nicholas6c942712018-03-16 09:45:11 -0400297 "out location=0, index=0 is reserved for sk_FragColor");
298 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400299 const ASTNode::VarData& varData = varDecl.getVarData();
ethannicholasd598f792016-07-25 10:08:54 -0700300 const Type* type = baseType;
ethannicholas14fe8cc2016-09-07 13:37:16 -0700301 std::vector<std::unique_ptr<Expression>> sizes;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400302 auto iter = varDecl.begin();
Brian Osman82d49702019-12-30 13:44:01 -0500303 if (varData.fSizeCount > 0 && (modifiers.fFlags & Modifiers::kIn_Flag)) {
304 fErrors.error(varDecl.fOffset, "'in' variables may not have array type");
305 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400306 for (size_t i = 0; i < varData.fSizeCount; ++i, ++iter) {
307 const ASTNode& rawSize = *iter;
ethannicholas14fe8cc2016-09-07 13:37:16 -0700308 if (rawSize) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400309 auto size = this->coerce(this->convertExpression(rawSize), *fContext.fInt_Type);
ethannicholasb3058bd2016-07-01 08:22:01 -0700310 if (!size) {
311 return nullptr;
312 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700313 String name(type->fName);
Ethan Nicholas50afc172017-02-16 14:49:57 -0500314 int64_t count;
ethannicholasb3058bd2016-07-01 08:22:01 -0700315 if (size->fKind == Expression::kIntLiteral_Kind) {
316 count = ((IntLiteral&) *size).fValue;
317 if (count <= 0) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700318 fErrors.error(size->fOffset, "array size must be positive");
Ethan Nicholas66d80062019-09-09 14:50:51 -0400319 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700320 }
321 name += "[" + to_string(count) + "]";
322 } else {
Ethan Nicholas66d80062019-09-09 14:50:51 -0400323 fErrors.error(size->fOffset, "array size must be specified");
324 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700325 }
Ethan Nicholas91164d12019-05-15 15:29:54 -0400326 type = (Type*) fSymbolTable->takeOwnership(
327 std::unique_ptr<Symbol>(new Type(name,
328 Type::kArray_Kind,
329 *type,
330 (int) count)));
ethannicholas14fe8cc2016-09-07 13:37:16 -0700331 sizes.push_back(std::move(size));
ethannicholasb3058bd2016-07-01 08:22:01 -0700332 } else {
Ethan Nicholas91164d12019-05-15 15:29:54 -0400333 type = (Type*) fSymbolTable->takeOwnership(
334 std::unique_ptr<Symbol>(new Type(type->name() + "[]",
335 Type::kArray_Kind,
336 *type,
337 -1)));
ethannicholas14fe8cc2016-09-07 13:37:16 -0700338 sizes.push_back(nullptr);
ethannicholasb3058bd2016-07-01 08:22:01 -0700339 }
340 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400341 auto var = std::unique_ptr<Variable>(new Variable(varDecl.fOffset, modifiers,
342 varData.fName, *type, storage));
Robert Phillipsfe8da172018-01-24 14:52:02 +0000343 if (var->fName == Compiler::RTADJUST_NAME) {
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400344 SkASSERT(!fRTAdjust);
345 SkASSERT(var->fType == *fContext.fFloat4_Type);
Robert Phillipsfe8da172018-01-24 14:52:02 +0000346 fRTAdjust = var.get();
347 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700348 std::unique_ptr<Expression> value;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400349 if (iter != varDecl.end()) {
350 value = this->convertExpression(*iter);
ethannicholasb3058bd2016-07-01 08:22:01 -0700351 if (!value) {
352 return nullptr;
353 }
ethannicholasd598f792016-07-25 10:08:54 -0700354 value = this->coerce(std::move(value), *type);
Ethan Nicholas68dd2c12018-03-01 15:05:17 -0500355 if (!value) {
356 return nullptr;
357 }
Ethan Nicholascb670962017-04-20 19:31:52 -0400358 var->fWriteCount = 1;
Ethan Nicholas8f6c2ab2018-01-17 13:51:52 -0500359 var->fInitialValue = value.get();
ethannicholasb3058bd2016-07-01 08:22:01 -0700360 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400361 if (storage == Variable::kGlobal_Storage && var->fName == "sk_FragColor" &&
362 (*fSymbolTable)[var->fName]) {
ethannicholas5961bc92016-10-12 06:39:56 -0700363 // already defined, ignore
Ethan Nicholasfc994162019-06-06 10:04:27 -0400364 } else if (storage == Variable::kGlobal_Storage && (*fSymbolTable)[var->fName] &&
365 (*fSymbolTable)[var->fName]->fKind == Symbol::kVariable_Kind &&
366 ((Variable*) (*fSymbolTable)[var->fName])->fModifiers.fLayout.fBuiltin >= 0) {
ethannicholasf789b382016-08-03 12:43:36 -0700367 // already defined, just update the modifiers
Ethan Nicholasfc994162019-06-06 10:04:27 -0400368 Variable* old = (Variable*) (*fSymbolTable)[var->fName];
ethannicholasf789b382016-08-03 12:43:36 -0700369 old->fModifiers = var->fModifiers;
370 } else {
Ethan Nicholas82a62d22017-11-07 14:42:10 +0000371 variables.emplace_back(new VarDeclaration(var.get(), std::move(sizes),
372 std::move(value)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400373 StringFragment name = var->fName;
374 fSymbolTable->add(name, std::move(var));
ethannicholasf789b382016-08-03 12:43:36 -0700375 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700376 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400377 return std::unique_ptr<VarDeclarations>(new VarDeclarations(decls.fOffset,
ethannicholas14fe8cc2016-09-07 13:37:16 -0700378 baseType,
379 std::move(variables)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700380}
381
Ethan Nicholasfc994162019-06-06 10:04:27 -0400382std::unique_ptr<ModifiersDeclaration> IRGenerator::convertModifiersDeclaration(const ASTNode& m) {
383 SkASSERT(m.fKind == ASTNode::Kind::kModifiers);
384 Modifiers modifiers = m.getModifiers();
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400385 if (modifiers.fLayout.fInvocations != -1) {
Ethan Nicholasf06576b2019-04-03 15:45:25 -0400386 if (fKind != Program::kGeometry_Kind) {
387 fErrors.error(m.fOffset, "'invocations' is only legal in geometry shaders");
388 return nullptr;
389 }
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400390 fInvocations = modifiers.fLayout.fInvocations;
Chris Daltonf1b47bb2017-10-06 11:57:51 -0600391 if (fSettings->fCaps && !fSettings->fCaps->gsInvocationsSupport()) {
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400392 modifiers.fLayout.fInvocations = -1;
393 Variable* invocationId = (Variable*) (*fSymbolTable)["sk_InvocationID"];
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400394 SkASSERT(invocationId);
Ethan Nicholasd1d52562018-03-20 16:30:34 -0400395 invocationId->fModifiers.fFlags = 0;
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400396 invocationId->fModifiers.fLayout.fBuiltin = -1;
397 if (modifiers.fLayout.description() == "") {
398 return nullptr;
399 }
400 }
401 }
402 if (modifiers.fLayout.fMaxVertices != -1 && fInvocations > 0 && fSettings->fCaps &&
Chris Daltonf1b47bb2017-10-06 11:57:51 -0600403 !fSettings->fCaps->gsInvocationsSupport()) {
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400404 modifiers.fLayout.fMaxVertices *= fInvocations;
405 }
406 return std::unique_ptr<ModifiersDeclaration>(new ModifiersDeclaration(modifiers));
ethannicholas5961bc92016-10-12 06:39:56 -0700407}
408
Ethan Nicholasfc994162019-06-06 10:04:27 -0400409std::unique_ptr<Statement> IRGenerator::convertIf(const ASTNode& n) {
410 SkASSERT(n.fKind == ASTNode::Kind::kIf);
411 auto iter = n.begin();
412 std::unique_ptr<Expression> test = this->coerce(this->convertExpression(*(iter++)),
ethannicholasd598f792016-07-25 10:08:54 -0700413 *fContext.fBool_Type);
ethannicholasb3058bd2016-07-01 08:22:01 -0700414 if (!test) {
415 return nullptr;
416 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400417 std::unique_ptr<Statement> ifTrue = this->convertStatement(*(iter++));
ethannicholasb3058bd2016-07-01 08:22:01 -0700418 if (!ifTrue) {
419 return nullptr;
420 }
421 std::unique_ptr<Statement> ifFalse;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400422 if (iter != n.end()) {
423 ifFalse = this->convertStatement(*(iter++));
ethannicholasb3058bd2016-07-01 08:22:01 -0700424 if (!ifFalse) {
425 return nullptr;
426 }
427 }
ethannicholas08a92112016-11-09 13:26:45 -0800428 if (test->fKind == Expression::kBoolLiteral_Kind) {
429 // static boolean value, fold down to a single branch
430 if (((BoolLiteral&) *test).fValue) {
431 return ifTrue;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400432 } else if (ifFalse) {
ethannicholas08a92112016-11-09 13:26:45 -0800433 return ifFalse;
434 } else {
435 // False & no else clause. Not an error, so don't return null!
436 std::vector<std::unique_ptr<Statement>> empty;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400437 return std::unique_ptr<Statement>(new Block(n.fOffset, std::move(empty),
ethannicholas08a92112016-11-09 13:26:45 -0800438 fSymbolTable));
439 }
440 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400441 return std::unique_ptr<Statement>(new IfStatement(n.fOffset, n.getBool(), std::move(test),
ethannicholasb3058bd2016-07-01 08:22:01 -0700442 std::move(ifTrue), std::move(ifFalse)));
443}
444
Ethan Nicholasfc994162019-06-06 10:04:27 -0400445std::unique_ptr<Statement> IRGenerator::convertFor(const ASTNode& f) {
446 SkASSERT(f.fKind == ASTNode::Kind::kFor);
ethannicholas22f939e2016-10-13 13:25:34 -0700447 AutoLoopLevel level(this);
ethannicholasb3058bd2016-07-01 08:22:01 -0700448 AutoSymbolTable table(this);
ethannicholas22f939e2016-10-13 13:25:34 -0700449 std::unique_ptr<Statement> initializer;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400450 auto iter = f.begin();
451 if (*iter) {
452 initializer = this->convertStatement(*iter);
ethannicholas22f939e2016-10-13 13:25:34 -0700453 if (!initializer) {
454 return nullptr;
455 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700456 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400457 ++iter;
ethannicholas22f939e2016-10-13 13:25:34 -0700458 std::unique_ptr<Expression> test;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400459 if (*iter) {
460 test = this->coerce(this->convertExpression(*iter), *fContext.fBool_Type);
ethannicholas22f939e2016-10-13 13:25:34 -0700461 if (!test) {
462 return nullptr;
463 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700464 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400465 ++iter;
ethannicholas22f939e2016-10-13 13:25:34 -0700466 std::unique_ptr<Expression> next;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400467 if (*iter) {
468 next = this->convertExpression(*iter);
ethannicholas22f939e2016-10-13 13:25:34 -0700469 if (!next) {
470 return nullptr;
471 }
472 this->checkValid(*next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700473 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400474 ++iter;
475 std::unique_ptr<Statement> statement = this->convertStatement(*iter);
ethannicholasb3058bd2016-07-01 08:22:01 -0700476 if (!statement) {
477 return nullptr;
478 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700479 return std::unique_ptr<Statement>(new ForStatement(f.fOffset, std::move(initializer),
ethannicholasb3058bd2016-07-01 08:22:01 -0700480 std::move(test), std::move(next),
ethannicholasd598f792016-07-25 10:08:54 -0700481 std::move(statement), fSymbolTable));
ethannicholasb3058bd2016-07-01 08:22:01 -0700482}
483
Ethan Nicholasfc994162019-06-06 10:04:27 -0400484std::unique_ptr<Statement> IRGenerator::convertWhile(const ASTNode& w) {
485 SkASSERT(w.fKind == ASTNode::Kind::kWhile);
ethannicholas22f939e2016-10-13 13:25:34 -0700486 AutoLoopLevel level(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400487 auto iter = w.begin();
488 std::unique_ptr<Expression> test = this->coerce(this->convertExpression(*(iter++)),
ethannicholasd598f792016-07-25 10:08:54 -0700489 *fContext.fBool_Type);
ethannicholasb3058bd2016-07-01 08:22:01 -0700490 if (!test) {
491 return nullptr;
492 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400493 std::unique_ptr<Statement> statement = this->convertStatement(*(iter++));
ethannicholasb3058bd2016-07-01 08:22:01 -0700494 if (!statement) {
495 return nullptr;
496 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700497 return std::unique_ptr<Statement>(new WhileStatement(w.fOffset, std::move(test),
ethannicholasb3058bd2016-07-01 08:22:01 -0700498 std::move(statement)));
499}
500
Ethan Nicholasfc994162019-06-06 10:04:27 -0400501std::unique_ptr<Statement> IRGenerator::convertDo(const ASTNode& d) {
502 SkASSERT(d.fKind == ASTNode::Kind::kDo);
ethannicholas22f939e2016-10-13 13:25:34 -0700503 AutoLoopLevel level(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400504 auto iter = d.begin();
505 std::unique_ptr<Statement> statement = this->convertStatement(*(iter++));
506 if (!statement) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700507 return nullptr;
508 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400509 std::unique_ptr<Expression> test = this->coerce(this->convertExpression(*(iter++)),
510 *fContext.fBool_Type);
511 if (!test) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700512 return nullptr;
513 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700514 return std::unique_ptr<Statement>(new DoStatement(d.fOffset, std::move(statement),
ethannicholasb3058bd2016-07-01 08:22:01 -0700515 std::move(test)));
516}
517
Ethan Nicholasfc994162019-06-06 10:04:27 -0400518std::unique_ptr<Statement> IRGenerator::convertSwitch(const ASTNode& s) {
519 SkASSERT(s.fKind == ASTNode::Kind::kSwitch);
Ethan Nicholasaf197692017-02-27 13:26:45 -0500520 AutoSwitchLevel level(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400521 auto iter = s.begin();
522 std::unique_ptr<Expression> value = this->convertExpression(*(iter++));
Ethan Nicholasaf197692017-02-27 13:26:45 -0500523 if (!value) {
524 return nullptr;
525 }
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500526 if (value->fType != *fContext.fUInt_Type && value->fType.kind() != Type::kEnum_Kind) {
Ethan Nicholasaf197692017-02-27 13:26:45 -0500527 value = this->coerce(std::move(value), *fContext.fInt_Type);
528 if (!value) {
529 return nullptr;
530 }
531 }
532 AutoSymbolTable table(this);
533 std::unordered_set<int> caseValues;
534 std::vector<std::unique_ptr<SwitchCase>> cases;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400535 for (; iter != s.end(); ++iter) {
536 const ASTNode& c = *iter;
537 SkASSERT(c.fKind == ASTNode::Kind::kSwitchCase);
Ethan Nicholasaf197692017-02-27 13:26:45 -0500538 std::unique_ptr<Expression> caseValue;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400539 auto childIter = c.begin();
540 if (*childIter) {
541 caseValue = this->convertExpression(*childIter);
Ethan Nicholasaf197692017-02-27 13:26:45 -0500542 if (!caseValue) {
543 return nullptr;
544 }
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500545 caseValue = this->coerce(std::move(caseValue), value->fType);
546 if (!caseValue) {
547 return nullptr;
Ethan Nicholasaf197692017-02-27 13:26:45 -0500548 }
549 if (!caseValue->isConstant()) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700550 fErrors.error(caseValue->fOffset, "case value must be a constant");
Ethan Nicholasaf197692017-02-27 13:26:45 -0500551 return nullptr;
552 }
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500553 int64_t v;
554 this->getConstantInt(*caseValue, &v);
Ethan Nicholasaf197692017-02-27 13:26:45 -0500555 if (caseValues.find(v) != caseValues.end()) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700556 fErrors.error(caseValue->fOffset, "duplicate case value");
Ethan Nicholasaf197692017-02-27 13:26:45 -0500557 }
558 caseValues.insert(v);
559 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400560 ++childIter;
Ethan Nicholasaf197692017-02-27 13:26:45 -0500561 std::vector<std::unique_ptr<Statement>> statements;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400562 for (; childIter != c.end(); ++childIter) {
563 std::unique_ptr<Statement> converted = this->convertStatement(*childIter);
Ethan Nicholasaf197692017-02-27 13:26:45 -0500564 if (!converted) {
565 return nullptr;
566 }
567 statements.push_back(std::move(converted));
568 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400569 cases.emplace_back(new SwitchCase(c.fOffset, std::move(caseValue),
Ethan Nicholasaf197692017-02-27 13:26:45 -0500570 std::move(statements)));
571 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400572 return std::unique_ptr<Statement>(new SwitchStatement(s.fOffset, s.getBool(),
Ethan Nicholasc432b0c2017-07-18 13:22:37 -0400573 std::move(value), std::move(cases),
574 fSymbolTable));
Ethan Nicholasaf197692017-02-27 13:26:45 -0500575}
576
Ethan Nicholasfc994162019-06-06 10:04:27 -0400577std::unique_ptr<Statement> IRGenerator::convertExpressionStatement(const ASTNode& s) {
578 std::unique_ptr<Expression> e = this->convertExpression(s);
ethannicholasb3058bd2016-07-01 08:22:01 -0700579 if (!e) {
580 return nullptr;
581 }
582 this->checkValid(*e);
583 return std::unique_ptr<Statement>(new ExpressionStatement(std::move(e)));
584}
585
Ethan Nicholasfc994162019-06-06 10:04:27 -0400586std::unique_ptr<Statement> IRGenerator::convertReturn(const ASTNode& r) {
587 SkASSERT(r.fKind == ASTNode::Kind::kReturn);
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400588 SkASSERT(fCurrentFunction);
Robert Phillipsfe8da172018-01-24 14:52:02 +0000589 // early returns from a vertex main function will bypass the sk_Position normalization, so
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400590 // SkASSERT that we aren't doing that. It is of course possible to fix this by adding a
Robert Phillipsfe8da172018-01-24 14:52:02 +0000591 // normalization before each return, but it will probably never actually be necessary.
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400592 SkASSERT(Program::kVertex_Kind != fKind || !fRTAdjust || "main" != fCurrentFunction->fName);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400593 if (r.begin() != r.end()) {
594 std::unique_ptr<Expression> result = this->convertExpression(*r.begin());
ethannicholasb3058bd2016-07-01 08:22:01 -0700595 if (!result) {
596 return nullptr;
597 }
ethannicholasd598f792016-07-25 10:08:54 -0700598 if (fCurrentFunction->fReturnType == *fContext.fVoid_Type) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700599 fErrors.error(result->fOffset, "may not return a value from a void function");
ethannicholasb3058bd2016-07-01 08:22:01 -0700600 } else {
601 result = this->coerce(std::move(result), fCurrentFunction->fReturnType);
602 if (!result) {
603 return nullptr;
604 }
605 }
606 return std::unique_ptr<Statement>(new ReturnStatement(std::move(result)));
607 } else {
ethannicholasd598f792016-07-25 10:08:54 -0700608 if (fCurrentFunction->fReturnType != *fContext.fVoid_Type) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700609 fErrors.error(r.fOffset, "expected function to return '" +
Ethan Nicholas2a099da2020-01-02 14:40:54 -0500610 fCurrentFunction->fReturnType.displayName() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700611 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700612 return std::unique_ptr<Statement>(new ReturnStatement(r.fOffset));
ethannicholasb3058bd2016-07-01 08:22:01 -0700613 }
614}
615
Ethan Nicholasfc994162019-06-06 10:04:27 -0400616std::unique_ptr<Statement> IRGenerator::convertBreak(const ASTNode& b) {
617 SkASSERT(b.fKind == ASTNode::Kind::kBreak);
Ethan Nicholasaf197692017-02-27 13:26:45 -0500618 if (fLoopLevel > 0 || fSwitchLevel > 0) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700619 return std::unique_ptr<Statement>(new BreakStatement(b.fOffset));
ethannicholas22f939e2016-10-13 13:25:34 -0700620 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700621 fErrors.error(b.fOffset, "break statement must be inside a loop or switch");
ethannicholas22f939e2016-10-13 13:25:34 -0700622 return nullptr;
623 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700624}
625
Ethan Nicholasfc994162019-06-06 10:04:27 -0400626std::unique_ptr<Statement> IRGenerator::convertContinue(const ASTNode& c) {
627 SkASSERT(c.fKind == ASTNode::Kind::kContinue);
ethannicholas22f939e2016-10-13 13:25:34 -0700628 if (fLoopLevel > 0) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700629 return std::unique_ptr<Statement>(new ContinueStatement(c.fOffset));
ethannicholas22f939e2016-10-13 13:25:34 -0700630 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700631 fErrors.error(c.fOffset, "continue statement must be inside a loop");
ethannicholas22f939e2016-10-13 13:25:34 -0700632 return nullptr;
633 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700634}
635
Ethan Nicholasfc994162019-06-06 10:04:27 -0400636std::unique_ptr<Statement> IRGenerator::convertDiscard(const ASTNode& d) {
637 SkASSERT(d.fKind == ASTNode::Kind::kDiscard);
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700638 return std::unique_ptr<Statement>(new DiscardStatement(d.fOffset));
ethannicholasb3058bd2016-07-01 08:22:01 -0700639}
640
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500641std::unique_ptr<Block> IRGenerator::applyInvocationIDWorkaround(std::unique_ptr<Block> main) {
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400642 Layout invokeLayout;
643 Modifiers invokeModifiers(invokeLayout, Modifiers::kHasSideEffects_Flag);
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700644 FunctionDeclaration* invokeDecl = new FunctionDeclaration(-1,
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400645 invokeModifiers,
646 "_invoke",
647 std::vector<const Variable*>(),
648 *fContext.fVoid_Type);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500649 fProgramElements->push_back(std::unique_ptr<ProgramElement>(
650 new FunctionDefinition(-1, *invokeDecl, std::move(main))));
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400651 fSymbolTable->add(invokeDecl->fName, std::unique_ptr<FunctionDeclaration>(invokeDecl));
652
Ethan Nicholas82a62d22017-11-07 14:42:10 +0000653 std::vector<std::unique_ptr<VarDeclaration>> variables;
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400654 Variable* loopIdx = (Variable*) (*fSymbolTable)["sk_InvocationID"];
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400655 SkASSERT(loopIdx);
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700656 std::unique_ptr<Expression> test(new BinaryExpression(-1,
657 std::unique_ptr<Expression>(new VariableReference(-1, *loopIdx)),
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400658 Token::LT,
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700659 std::unique_ptr<IntLiteral>(new IntLiteral(fContext, -1, fInvocations)),
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400660 *fContext.fBool_Type));
661 std::unique_ptr<Expression> next(new PostfixExpression(
662 std::unique_ptr<Expression>(
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700663 new VariableReference(-1,
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400664 *loopIdx,
665 VariableReference::kReadWrite_RefKind)),
666 Token::PLUSPLUS));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400667 ASTNode endPrimitiveID(&fFile->fNodes, -1, ASTNode::Kind::kIdentifier, "EndPrimitive");
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400668 std::unique_ptr<Expression> endPrimitive = this->convertExpression(endPrimitiveID);
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400669 SkASSERT(endPrimitive);
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400670
671 std::vector<std::unique_ptr<Statement>> loopBody;
672 std::vector<std::unique_ptr<Expression>> invokeArgs;
673 loopBody.push_back(std::unique_ptr<Statement>(new ExpressionStatement(
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700674 this->call(-1,
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400675 *invokeDecl,
676 std::vector<std::unique_ptr<Expression>>()))));
677 loopBody.push_back(std::unique_ptr<Statement>(new ExpressionStatement(
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700678 this->call(-1,
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400679 std::move(endPrimitive),
680 std::vector<std::unique_ptr<Expression>>()))));
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700681 std::unique_ptr<Expression> assignment(new BinaryExpression(-1,
682 std::unique_ptr<Expression>(new VariableReference(-1, *loopIdx)),
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400683 Token::EQ,
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700684 std::unique_ptr<IntLiteral>(new IntLiteral(fContext, -1, 0)),
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400685 *fContext.fInt_Type));
686 std::unique_ptr<Statement> initializer(new ExpressionStatement(std::move(assignment)));
687 std::unique_ptr<Statement> loop = std::unique_ptr<Statement>(
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700688 new ForStatement(-1,
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400689 std::move(initializer),
690 std::move(test),
691 std::move(next),
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700692 std::unique_ptr<Block>(new Block(-1, std::move(loopBody))),
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400693 fSymbolTable));
694 std::vector<std::unique_ptr<Statement>> children;
695 children.push_back(std::move(loop));
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700696 return std::unique_ptr<Block>(new Block(-1, std::move(children)));
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400697}
698
Robert Phillipsfe8da172018-01-24 14:52:02 +0000699std::unique_ptr<Statement> IRGenerator::getNormalizeSkPositionCode() {
Ethan Nicholasb809efb2018-04-12 14:39:21 -0400700 // sk_Position = float4(sk_Position.xy * rtAdjust.xz + sk_Position.ww * rtAdjust.yw,
Robert Phillipsfe8da172018-01-24 14:52:02 +0000701 // 0,
702 // sk_Position.w);
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400703 SkASSERT(fSkPerVertex && fRTAdjust);
Robert Phillipsfe8da172018-01-24 14:52:02 +0000704 #define REF(var) std::unique_ptr<Expression>(\
705 new VariableReference(-1, *var, VariableReference::kRead_RefKind))
706 #define FIELD(var, idx) std::unique_ptr<Expression>(\
707 new FieldAccess(REF(var), idx, FieldAccess::kAnonymousInterfaceBlock_OwnerKind))
708 #define POS std::unique_ptr<Expression>(new FieldAccess(REF(fSkPerVertex), 0, \
709 FieldAccess::kAnonymousInterfaceBlock_OwnerKind))
710 #define ADJUST (fRTAdjustInterfaceBlock ? \
711 FIELD(fRTAdjustInterfaceBlock, fRTAdjustFieldIndex) : \
712 REF(fRTAdjust))
Ethan Nicholasb809efb2018-04-12 14:39:21 -0400713 #define SWIZZLE(expr, ...) std::unique_ptr<Expression>(new Swizzle(fContext, expr, \
714 { __VA_ARGS__ }))
715 #define OP(left, op, right) std::unique_ptr<Expression>( \
716 new BinaryExpression(-1, left, op, right, \
717 *fContext.fFloat2_Type))
Robert Phillipsfe8da172018-01-24 14:52:02 +0000718 std::vector<std::unique_ptr<Expression>> children;
Ethan Nicholasb809efb2018-04-12 14:39:21 -0400719 children.push_back(OP(OP(SWIZZLE(POS, 0, 1), Token::STAR, SWIZZLE(ADJUST, 0, 2)),
Robert Phillipsfe8da172018-01-24 14:52:02 +0000720 Token::PLUS,
Ethan Nicholasb809efb2018-04-12 14:39:21 -0400721 OP(SWIZZLE(POS, 3, 3), Token::STAR, SWIZZLE(ADJUST, 1, 3))));
Robert Phillipsfe8da172018-01-24 14:52:02 +0000722 children.push_back(std::unique_ptr<Expression>(new FloatLiteral(fContext, -1, 0.0)));
723 children.push_back(SWIZZLE(POS, 3));
724 std::unique_ptr<Expression> result = OP(POS, Token::EQ,
725 std::unique_ptr<Expression>(new Constructor(-1,
726 *fContext.fFloat4_Type,
727 std::move(children))));
728 return std::unique_ptr<Statement>(new ExpressionStatement(std::move(result)));
729}
730
Ethan Nicholasfc994162019-06-06 10:04:27 -0400731void IRGenerator::convertFunction(const ASTNode& f) {
732 auto iter = f.begin();
733 const Type* returnType = this->convertType(*(iter++));
ethannicholasb3058bd2016-07-01 08:22:01 -0700734 if (!returnType) {
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400735 return;
ethannicholasb3058bd2016-07-01 08:22:01 -0700736 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400737 const ASTNode::FunctionData& fd = f.getFunctionData();
ethannicholasd598f792016-07-25 10:08:54 -0700738 std::vector<const Variable*> parameters;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400739 for (size_t i = 0; i < fd.fParameterCount; ++i) {
740 const ASTNode& param = *(iter++);
741 SkASSERT(param.fKind == ASTNode::Kind::kParameter);
742 ASTNode::ParameterData pd = param.getParameterData();
743 auto paramIter = param.begin();
744 const Type* type = this->convertType(*(paramIter++));
ethannicholasb3058bd2016-07-01 08:22:01 -0700745 if (!type) {
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400746 return;
ethannicholasb3058bd2016-07-01 08:22:01 -0700747 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400748 for (int j = (int) pd.fSizeCount; j >= 1; j--) {
749 int size = (param.begin() + j)->getInt();
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400750 String name = type->name() + "[" + to_string(size) + "]";
Ethan Nicholas91164d12019-05-15 15:29:54 -0400751 type = (Type*) fSymbolTable->takeOwnership(
752 std::unique_ptr<Symbol>(new Type(std::move(name),
753 Type::kArray_Kind,
754 *type,
755 size)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700756 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400757 StringFragment name = pd.fName;
Ethan Nicholas91164d12019-05-15 15:29:54 -0400758 Variable* var = (Variable*) fSymbolTable->takeOwnership(
Ethan Nicholasfc994162019-06-06 10:04:27 -0400759 std::unique_ptr<Symbol>(new Variable(param.fOffset,
760 pd.fModifiers,
Ethan Nicholas91164d12019-05-15 15:29:54 -0400761 name,
762 *type,
763 Variable::kParameter_Storage)));
ethannicholasd598f792016-07-25 10:08:54 -0700764 parameters.push_back(var);
ethannicholasb3058bd2016-07-01 08:22:01 -0700765 }
766
Ethan Nicholasfc994162019-06-06 10:04:27 -0400767 if (fd.fName == "main") {
Ethan Nicholas0d997662019-04-08 09:46:01 -0400768 switch (fKind) {
769 case Program::kPipelineStage_Kind: {
770 bool valid;
771 switch (parameters.size()) {
Brian Osman7353dc52020-02-07 13:37:12 -0500772 case 2:
773 valid = parameters[0]->fType == *fContext.fFloat2_Type &&
Ethan Nicholas0d997662019-04-08 09:46:01 -0400774 parameters[0]->fModifiers.fFlags == 0 &&
Brian Osman7353dc52020-02-07 13:37:12 -0500775 parameters[1]->fType == *fContext.fHalf4_Type &&
776 parameters[1]->fModifiers.fFlags == (Modifiers::kIn_Flag |
Ethan Nicholas0d997662019-04-08 09:46:01 -0400777 Modifiers::kOut_Flag);
778 break;
779 case 1:
780 valid = parameters[0]->fType == *fContext.fHalf4_Type &&
781 parameters[0]->fModifiers.fFlags == (Modifiers::kIn_Flag |
782 Modifiers::kOut_Flag);
783 break;
784 default:
785 valid = false;
786 }
787 if (!valid) {
Brian Osman7353dc52020-02-07 13:37:12 -0500788 fErrors.error(f.fOffset, "pipeline stage 'main' must be declared main(float2, "
789 "inout half4) or main(inout half4)");
Ethan Nicholas0d997662019-04-08 09:46:01 -0400790 return;
791 }
792 break;
Ethan Nicholasa70693b2019-03-04 13:07:36 -0500793 }
Ethan Nicholas746035a2019-04-23 13:31:09 -0400794 case Program::kGeneric_Kind:
Ethan Nicholas0d997662019-04-08 09:46:01 -0400795 break;
Ethan Nicholas0d997662019-04-08 09:46:01 -0400796 default:
797 if (parameters.size()) {
798 fErrors.error(f.fOffset, "shader 'main' must have zero parameters");
799 }
Ethan Nicholas00543112018-07-31 09:44:36 -0400800 }
801 }
802
ethannicholasb3058bd2016-07-01 08:22:01 -0700803 // find existing declaration
ethannicholasd598f792016-07-25 10:08:54 -0700804 const FunctionDeclaration* decl = nullptr;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400805 auto entry = (*fSymbolTable)[fd.fName];
ethannicholasb3058bd2016-07-01 08:22:01 -0700806 if (entry) {
ethannicholasd598f792016-07-25 10:08:54 -0700807 std::vector<const FunctionDeclaration*> functions;
ethannicholasb3058bd2016-07-01 08:22:01 -0700808 switch (entry->fKind) {
809 case Symbol::kUnresolvedFunction_Kind:
ethannicholasd598f792016-07-25 10:08:54 -0700810 functions = ((UnresolvedFunction*) entry)->fFunctions;
ethannicholasb3058bd2016-07-01 08:22:01 -0700811 break;
812 case Symbol::kFunctionDeclaration_Kind:
ethannicholasd598f792016-07-25 10:08:54 -0700813 functions.push_back((FunctionDeclaration*) entry);
ethannicholasb3058bd2016-07-01 08:22:01 -0700814 break;
815 default:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400816 fErrors.error(f.fOffset, "symbol '" + fd.fName + "' was already defined");
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400817 return;
ethannicholasb3058bd2016-07-01 08:22:01 -0700818 }
819 for (const auto& other : functions) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400820 SkASSERT(other->fName == fd.fName);
ethannicholasb3058bd2016-07-01 08:22:01 -0700821 if (parameters.size() == other->fParameters.size()) {
822 bool match = true;
823 for (size_t i = 0; i < parameters.size(); i++) {
824 if (parameters[i]->fType != other->fParameters[i]->fType) {
825 match = false;
826 break;
827 }
828 }
829 if (match) {
ethannicholasd598f792016-07-25 10:08:54 -0700830 if (*returnType != other->fReturnType) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400831 FunctionDeclaration newDecl(f.fOffset, fd.fModifiers, fd.fName, parameters,
Ethan Nicholascb670962017-04-20 19:31:52 -0400832 *returnType);
Ethan Nicholas2a099da2020-01-02 14:40:54 -0500833 fErrors.error(f.fOffset, "functions '" + newDecl.declaration() +
834 "' and '" + other->declaration() +
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700835 "' differ only in return type");
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400836 return;
ethannicholasb3058bd2016-07-01 08:22:01 -0700837 }
838 decl = other;
839 for (size_t i = 0; i < parameters.size(); i++) {
840 if (parameters[i]->fModifiers != other->fParameters[i]->fModifiers) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700841 fErrors.error(f.fOffset, "modifiers on parameter " +
842 to_string((uint64_t) i + 1) +
843 " differ between declaration and "
844 "definition");
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400845 return;
ethannicholasb3058bd2016-07-01 08:22:01 -0700846 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700847 }
Ethan Nicholasdb80f692019-11-22 14:06:12 -0500848 if (other->fDefined && !other->fBuiltin) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700849 fErrors.error(f.fOffset, "duplicate definition of " +
Ethan Nicholas2a099da2020-01-02 14:40:54 -0500850 other->declaration());
ethannicholasb3058bd2016-07-01 08:22:01 -0700851 }
852 break;
853 }
854 }
855 }
856 }
857 if (!decl) {
858 // couldn't find an existing declaration
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700859 auto newDecl = std::unique_ptr<FunctionDeclaration>(new FunctionDeclaration(f.fOffset,
Ethan Nicholasfc994162019-06-06 10:04:27 -0400860 fd.fModifiers,
861 fd.fName,
ethannicholas471e8942016-10-28 09:02:46 -0700862 parameters,
863 *returnType));
864 decl = newDecl.get();
865 fSymbolTable->add(decl->fName, std::move(newDecl));
ethannicholasb3058bd2016-07-01 08:22:01 -0700866 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400867 if (iter != f.end()) {
868 // compile body
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400869 SkASSERT(!fCurrentFunction);
ethannicholasd598f792016-07-25 10:08:54 -0700870 fCurrentFunction = decl;
871 decl->fDefined = true;
872 std::shared_ptr<SymbolTable> old = fSymbolTable;
873 AutoSymbolTable table(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400874 if (fd.fName == "main" && fKind == Program::kPipelineStage_Kind) {
Brian Osman7353dc52020-02-07 13:37:12 -0500875 if (parameters.size() == 2) {
876 parameters[0]->fModifiers.fLayout.fBuiltin = SK_MAIN_COORDS_BUILTIN;
877 parameters[1]->fModifiers.fLayout.fBuiltin = SK_OUTCOLOR_BUILTIN;
Ethan Nicholasa70693b2019-03-04 13:07:36 -0500878 } else {
879 SkASSERT(parameters.size() == 1);
880 parameters[0]->fModifiers.fLayout.fBuiltin = SK_OUTCOLOR_BUILTIN;
881 }
Ethan Nicholas00543112018-07-31 09:44:36 -0400882 }
ethannicholasd598f792016-07-25 10:08:54 -0700883 for (size_t i = 0; i < parameters.size(); i++) {
884 fSymbolTable->addWithoutOwnership(parameters[i]->fName, decl->fParameters[i]);
ethannicholasb3058bd2016-07-01 08:22:01 -0700885 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400886 bool needInvocationIDWorkaround = fInvocations != -1 && fd.fName == "main" &&
Chris Daltonf1b47bb2017-10-06 11:57:51 -0600887 fSettings->fCaps &&
888 !fSettings->fCaps->gsInvocationsSupport();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400889 std::unique_ptr<Block> body = this->convertBlock(*iter);
ethannicholasd598f792016-07-25 10:08:54 -0700890 fCurrentFunction = nullptr;
891 if (!body) {
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -0400892 return;
893 }
894 if (needInvocationIDWorkaround) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500895 body = this->applyInvocationIDWorkaround(std::move(body));
ethannicholasd598f792016-07-25 10:08:54 -0700896 }
Ethan Nicholascb670962017-04-20 19:31:52 -0400897 // conservatively assume all user-defined functions have side effects
898 ((Modifiers&) decl->fModifiers).fFlags |= Modifiers::kHasSideEffects_Flag;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400899 if (Program::kVertex_Kind == fKind && fd.fName == "main" && fRTAdjust) {
Robert Phillipsfe8da172018-01-24 14:52:02 +0000900 body->fStatements.insert(body->fStatements.end(), this->getNormalizeSkPositionCode());
901 }
Ethan Nicholasdb80f692019-11-22 14:06:12 -0500902 std::unique_ptr<FunctionDefinition> result(new FunctionDefinition(f.fOffset, *decl,
903 std::move(body)));
904 result->fSource = &f;
905 fProgramElements->push_back(std::move(result));
ethannicholasb3058bd2016-07-01 08:22:01 -0700906 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700907}
908
Ethan Nicholasfc994162019-06-06 10:04:27 -0400909std::unique_ptr<InterfaceBlock> IRGenerator::convertInterfaceBlock(const ASTNode& intf) {
910 SkASSERT(intf.fKind == ASTNode::Kind::kInterfaceBlock);
911 ASTNode::InterfaceBlockData id = intf.getInterfaceBlockData();
ethannicholasb3058bd2016-07-01 08:22:01 -0700912 std::shared_ptr<SymbolTable> old = fSymbolTable;
Ethan Nicholas68dd2c12018-03-01 15:05:17 -0500913 this->pushSymbolTable();
914 std::shared_ptr<SymbolTable> symbols = fSymbolTable;
ethannicholasb3058bd2016-07-01 08:22:01 -0700915 std::vector<Type::Field> fields;
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400916 bool haveRuntimeArray = false;
Robert Phillipsfe8da172018-01-24 14:52:02 +0000917 bool foundRTAdjust = false;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400918 auto iter = intf.begin();
919 for (size_t i = 0; i < id.fDeclarationCount; ++i) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700920 std::unique_ptr<VarDeclarations> decl = this->convertVarDeclarations(
Ethan Nicholasfc994162019-06-06 10:04:27 -0400921 *(iter++),
Ethan Nicholasa7ceb502019-01-11 10:31:48 -0500922 Variable::kInterfaceBlock_Storage);
ethannicholas7effa7a2016-10-14 09:56:33 -0700923 if (!decl) {
924 return nullptr;
925 }
Ethan Nicholas82a62d22017-11-07 14:42:10 +0000926 for (const auto& stmt : decl->fVars) {
927 VarDeclaration& vd = (VarDeclaration&) *stmt;
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400928 if (haveRuntimeArray) {
Ethan Nicholas82a62d22017-11-07 14:42:10 +0000929 fErrors.error(decl->fOffset,
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400930 "only the last entry in an interface block may be a runtime-sized "
931 "array");
932 }
Robert Phillipsfe8da172018-01-24 14:52:02 +0000933 if (vd.fVar == fRTAdjust) {
934 foundRTAdjust = true;
Ethan Nicholasd9d33c32018-06-12 11:05:59 -0400935 SkASSERT(vd.fVar->fType == *fContext.fFloat4_Type);
Robert Phillipsfe8da172018-01-24 14:52:02 +0000936 fRTAdjustFieldIndex = fields.size();
937 }
Ethan Nicholas82a62d22017-11-07 14:42:10 +0000938 fields.push_back(Type::Field(vd.fVar->fModifiers, vd.fVar->fName,
939 &vd.fVar->fType));
940 if (vd.fValue) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700941 fErrors.error(decl->fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -0700942 "initializers are not permitted on interface block fields");
943 }
Ethan Nicholas82a62d22017-11-07 14:42:10 +0000944 if (vd.fVar->fModifiers.fFlags & (Modifiers::kIn_Flag |
Ethan Nicholasfc994162019-06-06 10:04:27 -0400945 Modifiers::kOut_Flag |
946 Modifiers::kUniform_Flag |
947 Modifiers::kBuffer_Flag |
948 Modifiers::kConst_Flag)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700949 fErrors.error(decl->fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -0700950 "interface block fields may not have storage qualifiers");
951 }
Ethan Nicholas82a62d22017-11-07 14:42:10 +0000952 if (vd.fVar->fType.kind() == Type::kArray_Kind &&
953 vd.fVar->fType.columns() == -1) {
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400954 haveRuntimeArray = true;
955 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500956 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700957 }
Ethan Nicholas68dd2c12018-03-01 15:05:17 -0500958 this->popSymbolTable();
Ethan Nicholas91164d12019-05-15 15:29:54 -0400959 Type* type = (Type*) old->takeOwnership(std::unique_ptr<Symbol>(new Type(intf.fOffset,
Ethan Nicholasfc994162019-06-06 10:04:27 -0400960 id.fTypeName,
Ethan Nicholas91164d12019-05-15 15:29:54 -0400961 fields)));
Ethan Nicholas50afc172017-02-16 14:49:57 -0500962 std::vector<std::unique_ptr<Expression>> sizes;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400963 for (size_t i = 0; i < id.fSizeCount; ++i) {
964 const ASTNode& size = *(iter++);
Ethan Nicholas50afc172017-02-16 14:49:57 -0500965 if (size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400966 std::unique_ptr<Expression> converted = this->convertExpression(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -0500967 if (!converted) {
968 return nullptr;
969 }
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400970 String name = type->fName;
Ethan Nicholas50afc172017-02-16 14:49:57 -0500971 int64_t count;
972 if (converted->fKind == Expression::kIntLiteral_Kind) {
973 count = ((IntLiteral&) *converted).fValue;
974 if (count <= 0) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700975 fErrors.error(converted->fOffset, "array size must be positive");
Ethan Nicholas66d80062019-09-09 14:50:51 -0400976 return nullptr;
Ethan Nicholas50afc172017-02-16 14:49:57 -0500977 }
978 name += "[" + to_string(count) + "]";
979 } else {
Ethan Nicholas66d80062019-09-09 14:50:51 -0400980 fErrors.error(intf.fOffset, "array size must be specified");
981 return nullptr;
Ethan Nicholas50afc172017-02-16 14:49:57 -0500982 }
Ethan Nicholas91164d12019-05-15 15:29:54 -0400983 type = (Type*) symbols->takeOwnership(std::unique_ptr<Symbol>(
984 new Type(name,
985 Type::kArray_Kind,
986 *type,
987 (int) count)));
Ethan Nicholas50afc172017-02-16 14:49:57 -0500988 sizes.push_back(std::move(converted));
989 } else {
Ethan Nicholas66d80062019-09-09 14:50:51 -0400990 fErrors.error(intf.fOffset, "array size must be specified");
991 return nullptr;
Ethan Nicholas50afc172017-02-16 14:49:57 -0500992 }
993 }
Ethan Nicholas91164d12019-05-15 15:29:54 -0400994 Variable* var = (Variable*) old->takeOwnership(std::unique_ptr<Symbol>(
995 new Variable(intf.fOffset,
Ethan Nicholasfc994162019-06-06 10:04:27 -0400996 id.fModifiers,
997 id.fInstanceName.fLength ? id.fInstanceName : id.fTypeName,
Ethan Nicholas91164d12019-05-15 15:29:54 -0400998 *type,
999 Variable::kGlobal_Storage)));
Robert Phillipsfe8da172018-01-24 14:52:02 +00001000 if (foundRTAdjust) {
1001 fRTAdjustInterfaceBlock = var;
1002 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001003 if (id.fInstanceName.fLength) {
1004 old->addWithoutOwnership(id.fInstanceName, var);
ethannicholasb3058bd2016-07-01 08:22:01 -07001005 } else {
1006 for (size_t i = 0; i < fields.size(); i++) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001007 old->add(fields[i].fName, std::unique_ptr<Field>(new Field(intf.fOffset, *var,
ethannicholasd598f792016-07-25 10:08:54 -07001008 (int) i)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001009 }
1010 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001011 return std::unique_ptr<InterfaceBlock>(new InterfaceBlock(intf.fOffset,
Ethan Nicholas8feeff92017-03-30 14:11:58 -04001012 var,
Ethan Nicholasfc994162019-06-06 10:04:27 -04001013 id.fTypeName,
1014 id.fInstanceName,
Ethan Nicholas50afc172017-02-16 14:49:57 -05001015 std::move(sizes),
Ethan Nicholas68dd2c12018-03-01 15:05:17 -05001016 symbols));
ethannicholasb3058bd2016-07-01 08:22:01 -07001017}
1018
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001019void IRGenerator::getConstantInt(const Expression& value, int64_t* out) {
1020 switch (value.fKind) {
1021 case Expression::kIntLiteral_Kind:
1022 *out = ((const IntLiteral&) value).fValue;
1023 break;
1024 case Expression::kVariableReference_Kind: {
1025 const Variable& var = ((VariableReference&) value).fVariable;
1026 if ((var.fModifiers.fFlags & Modifiers::kConst_Flag) &&
1027 var.fInitialValue) {
1028 this->getConstantInt(*var.fInitialValue, out);
1029 }
1030 break;
1031 }
1032 default:
1033 fErrors.error(value.fOffset, "expected a constant int");
1034 }
1035}
1036
Ethan Nicholasfc994162019-06-06 10:04:27 -04001037void IRGenerator::convertEnum(const ASTNode& e) {
1038 SkASSERT(e.fKind == ASTNode::Kind::kEnum);
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001039 std::vector<Variable*> variables;
1040 int64_t currentValue = 0;
1041 Layout layout;
Ethan Nicholasfc994162019-06-06 10:04:27 -04001042 ASTNode enumType(e.fNodes, e.fOffset, ASTNode::Kind::kType,
1043 ASTNode::TypeData(e.getString(), false, false));
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001044 const Type* type = this->convertType(enumType);
1045 Modifiers modifiers(layout, Modifiers::kConst_Flag);
Ethan Nicholaseace9352018-10-15 20:09:54 +00001046 std::shared_ptr<SymbolTable> symbols(new SymbolTable(fSymbolTable, &fErrors));
1047 fSymbolTable = symbols;
Ethan Nicholasfc994162019-06-06 10:04:27 -04001048 for (auto iter = e.begin(); iter != e.end(); ++iter) {
1049 const ASTNode& child = *iter;
1050 SkASSERT(child.fKind == ASTNode::Kind::kEnumCase);
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001051 std::unique_ptr<Expression> value;
Ethan Nicholasfc994162019-06-06 10:04:27 -04001052 if (child.begin() != child.end()) {
1053 value = this->convertExpression(*child.begin());
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001054 if (!value) {
Ethan Nicholaseace9352018-10-15 20:09:54 +00001055 fSymbolTable = symbols->fParent;
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001056 return;
1057 }
1058 this->getConstantInt(*value, &currentValue);
1059 }
1060 value = std::unique_ptr<Expression>(new IntLiteral(fContext, e.fOffset, currentValue));
1061 ++currentValue;
Ethan Nicholasfc994162019-06-06 10:04:27 -04001062 auto var = std::unique_ptr<Variable>(new Variable(e.fOffset, modifiers, child.getString(),
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001063 *type, Variable::kGlobal_Storage,
1064 value.get()));
1065 variables.push_back(var.get());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001066 symbols->add(child.getString(), std::move(var));
Ethan Nicholas91164d12019-05-15 15:29:54 -04001067 symbols->takeOwnership(std::move(value));
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001068 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001069 fProgramElements->push_back(std::unique_ptr<ProgramElement>(new Enum(e.fOffset, e.getString(),
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001070 symbols)));
Ethan Nicholaseace9352018-10-15 20:09:54 +00001071 fSymbolTable = symbols->fParent;
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001072}
1073
Ethan Nicholasfc994162019-06-06 10:04:27 -04001074const Type* IRGenerator::convertType(const ASTNode& type) {
1075 ASTNode::TypeData td = type.getTypeData();
1076 const Symbol* result = (*fSymbolTable)[td.fName];
ethannicholasb3058bd2016-07-01 08:22:01 -07001077 if (result && result->fKind == Symbol::kType_Kind) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001078 if (td.fIsNullable) {
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001079 if (((Type&) *result) == *fContext.fFragmentProcessor_Type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001080 if (type.begin() != type.end()) {
1081 fErrors.error(type.fOffset, "type '" + td.fName + "' may not be used in "
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001082 "an array");
1083 }
Ethan Nicholas91164d12019-05-15 15:29:54 -04001084 result = fSymbolTable->takeOwnership(std::unique_ptr<Symbol>(
1085 new Type(String(result->fName) + "?",
1086 Type::kNullable_Kind,
1087 (const Type&) *result)));
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001088 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001089 fErrors.error(type.fOffset, "type '" + td.fName + "' may not be nullable");
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001090 }
1091 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001092 for (const auto& size : type) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001093 String name(result->fName);
1094 name += "[";
Ethan Nicholasfc994162019-06-06 10:04:27 -04001095 if (size) {
1096 name += to_string(size.getInt());
Ethan Nicholas50afc172017-02-16 14:49:57 -05001097 }
1098 name += "]";
Ethan Nicholas91164d12019-05-15 15:29:54 -04001099 result = (Type*) fSymbolTable->takeOwnership(std::unique_ptr<Symbol>(
1100 new Type(name,
1101 Type::kArray_Kind,
1102 (const Type&) *result,
Brian Osmanb9416ca2019-06-12 16:18:32 -04001103 size ? size.getInt()
1104 : 0)));
Ethan Nicholas50afc172017-02-16 14:49:57 -05001105 }
ethannicholasd598f792016-07-25 10:08:54 -07001106 return (const Type*) result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001107 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001108 fErrors.error(type.fOffset, "unknown type '" + td.fName + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07001109 return nullptr;
1110}
1111
Ethan Nicholasfc994162019-06-06 10:04:27 -04001112std::unique_ptr<Expression> IRGenerator::convertExpression(const ASTNode& expr) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001113 switch (expr.fKind) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001114 case ASTNode::Kind::kBinary:
1115 return this->convertBinaryExpression(expr);
1116 case ASTNode::Kind::kBool:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001117 return std::unique_ptr<Expression>(new BoolLiteral(fContext, expr.fOffset,
Ethan Nicholasfc994162019-06-06 10:04:27 -04001118 expr.getBool()));
1119 case ASTNode::Kind::kCall:
1120 return this->convertCallExpression(expr);
1121 case ASTNode::Kind::kField:
1122 return this->convertFieldExpression(expr);
1123 case ASTNode::Kind::kFloat:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001124 return std::unique_ptr<Expression>(new FloatLiteral(fContext, expr.fOffset,
Ethan Nicholasfc994162019-06-06 10:04:27 -04001125 expr.getFloat()));
1126 case ASTNode::Kind::kIdentifier:
1127 return this->convertIdentifier(expr);
1128 case ASTNode::Kind::kIndex:
1129 return this->convertIndexExpression(expr);
1130 case ASTNode::Kind::kInt:
1131 return std::unique_ptr<Expression>(new IntLiteral(fContext, expr.fOffset,
1132 expr.getInt()));
1133 case ASTNode::Kind::kNull:
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001134 return std::unique_ptr<Expression>(new NullLiteral(fContext, expr.fOffset));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001135 case ASTNode::Kind::kPostfix:
1136 return this->convertPostfixExpression(expr);
1137 case ASTNode::Kind::kPrefix:
1138 return this->convertPrefixExpression(expr);
1139 case ASTNode::Kind::kTernary:
1140 return this->convertTernaryExpression(expr);
ethannicholasb3058bd2016-07-01 08:22:01 -07001141 default:
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001142#ifdef SK_DEBUG
Ethan Nicholasfc994162019-06-06 10:04:27 -04001143 ABORT("unsupported expression: %s\n", expr.description().c_str());
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001144#endif
1145 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001146 }
1147}
1148
Ethan Nicholasfc994162019-06-06 10:04:27 -04001149std::unique_ptr<Expression> IRGenerator::convertIdentifier(const ASTNode& identifier) {
1150 SkASSERT(identifier.fKind == ASTNode::Kind::kIdentifier);
1151 const Symbol* result = (*fSymbolTable)[identifier.getString()];
ethannicholasb3058bd2016-07-01 08:22:01 -07001152 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001153 fErrors.error(identifier.fOffset, "unknown identifier '" + identifier.getString() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07001154 return nullptr;
1155 }
1156 switch (result->fKind) {
1157 case Symbol::kFunctionDeclaration_Kind: {
ethannicholasd598f792016-07-25 10:08:54 -07001158 std::vector<const FunctionDeclaration*> f = {
1159 (const FunctionDeclaration*) result
ethannicholasb3058bd2016-07-01 08:22:01 -07001160 };
ethannicholasd598f792016-07-25 10:08:54 -07001161 return std::unique_ptr<FunctionReference>(new FunctionReference(fContext,
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001162 identifier.fOffset,
ethannicholasd598f792016-07-25 10:08:54 -07001163 f));
ethannicholasb3058bd2016-07-01 08:22:01 -07001164 }
1165 case Symbol::kUnresolvedFunction_Kind: {
ethannicholasd598f792016-07-25 10:08:54 -07001166 const UnresolvedFunction* f = (const UnresolvedFunction*) result;
1167 return std::unique_ptr<FunctionReference>(new FunctionReference(fContext,
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001168 identifier.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07001169 f->fFunctions));
1170 }
1171 case Symbol::kVariable_Kind: {
Ethan Nicholas38657112017-02-09 17:01:22 -05001172 const Variable* var = (const Variable*) result;
Ethan Nicholascd700e92018-08-24 16:43:57 -04001173 switch (var->fModifiers.fLayout.fBuiltin) {
1174 case SK_WIDTH_BUILTIN:
1175 fInputs.fRTWidth = true;
1176 break;
1177 case SK_HEIGHT_BUILTIN:
Greg Daniele6ab9982018-08-22 13:56:32 +00001178 fInputs.fRTHeight = true;
Ethan Nicholascd700e92018-08-24 16:43:57 -04001179 break;
1180#ifndef SKSL_STANDALONE
1181 case SK_FRAGCOORD_BUILTIN:
Brian Osman9f313b62019-10-02 12:03:11 -04001182 fInputs.fFlipY = true;
1183 if (fSettings->fFlipY &&
1184 (!fSettings->fCaps ||
1185 !fSettings->fCaps->fragCoordConventionsExtensionString())) {
1186 fInputs.fRTHeight = true;
Ethan Nicholascd700e92018-08-24 16:43:57 -04001187 }
Greg Daniele6ab9982018-08-22 13:56:32 +00001188#endif
Ethan Nicholascd700e92018-08-24 16:43:57 -04001189 }
Ethan Nicholas33c59ed2019-08-13 10:21:38 -04001190 if (fKind == Program::kFragmentProcessor_Kind &&
1191 (var->fModifiers.fFlags & Modifiers::kIn_Flag) &&
1192 !(var->fModifiers.fFlags & Modifiers::kUniform_Flag) &&
1193 !var->fModifiers.fLayout.fKey &&
1194 var->fModifiers.fLayout.fBuiltin == -1 &&
1195 var->fType.nonnullable() != *fContext.fFragmentProcessor_Type &&
1196 var->fType.kind() != Type::kSampler_Kind) {
1197 bool valid = false;
1198 for (const auto& decl : fFile->root()) {
1199 if (decl.fKind == ASTNode::Kind::kSection) {
1200 ASTNode::SectionData section = decl.getSectionData();
1201 if (section.fName == "setData") {
1202 valid = true;
1203 break;
1204 }
1205 }
1206 }
1207 if (!valid) {
1208 fErrors.error(identifier.fOffset, "'in' variable must be either 'uniform' or "
1209 "'layout(key)', or there must be a custom "
1210 "@setData function");
1211 }
1212 }
Ethan Nicholas86a43402017-01-19 13:32:00 -05001213 // default to kRead_RefKind; this will be corrected later if the variable is written to
1214 return std::unique_ptr<VariableReference>(new VariableReference(
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001215 identifier.fOffset,
Ethan Nicholas86a43402017-01-19 13:32:00 -05001216 *var,
1217 VariableReference::kRead_RefKind));
ethannicholasb3058bd2016-07-01 08:22:01 -07001218 }
1219 case Symbol::kField_Kind: {
ethannicholasd598f792016-07-25 10:08:54 -07001220 const Field* field = (const Field*) result;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001221 VariableReference* base = new VariableReference(identifier.fOffset, field->fOwner,
Ethan Nicholas86a43402017-01-19 13:32:00 -05001222 VariableReference::kRead_RefKind);
ethannicholasf789b382016-08-03 12:43:36 -07001223 return std::unique_ptr<Expression>(new FieldAccess(
1224 std::unique_ptr<Expression>(base),
1225 field->fFieldIndex,
1226 FieldAccess::kAnonymousInterfaceBlock_OwnerKind));
ethannicholasb3058bd2016-07-01 08:22:01 -07001227 }
1228 case Symbol::kType_Kind: {
ethannicholasd598f792016-07-25 10:08:54 -07001229 const Type* t = (const Type*) result;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001230 return std::unique_ptr<TypeReference>(new TypeReference(fContext, identifier.fOffset,
ethannicholasd598f792016-07-25 10:08:54 -07001231 *t));
ethannicholasb3058bd2016-07-01 08:22:01 -07001232 }
Ethan Nicholas91164d12019-05-15 15:29:54 -04001233 case Symbol::kExternal_Kind: {
1234 ExternalValue* r = (ExternalValue*) result;
1235 return std::unique_ptr<ExternalValueReference>(
1236 new ExternalValueReference(identifier.fOffset, r));
1237 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001238 default:
1239 ABORT("unsupported symbol type %d\n", result->fKind);
1240 }
Ethan Nicholasc0709392017-06-27 11:20:22 -04001241}
1242
Ethan Nicholasfc994162019-06-06 10:04:27 -04001243std::unique_ptr<Section> IRGenerator::convertSection(const ASTNode& s) {
1244 ASTNode::SectionData section = s.getSectionData();
1245 return std::unique_ptr<Section>(new Section(s.fOffset, section.fName, section.fArgument,
1246 section.fText));
Ethan Nicholas762466e2017-06-29 10:03:38 -04001247}
1248
1249
Ethan Nicholas11d53972016-11-28 11:23:23 -05001250std::unique_ptr<Expression> IRGenerator::coerce(std::unique_ptr<Expression> expr,
ethannicholasd598f792016-07-25 10:08:54 -07001251 const Type& type) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001252 if (!expr) {
1253 return nullptr;
1254 }
ethannicholasd598f792016-07-25 10:08:54 -07001255 if (expr->fType == type) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001256 return expr;
1257 }
1258 this->checkValid(*expr);
ethannicholasd598f792016-07-25 10:08:54 -07001259 if (expr->fType == *fContext.fInvalid_Type) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001260 return nullptr;
1261 }
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001262 if (expr->coercionCost(type) == INT_MAX) {
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001263 fErrors.error(expr->fOffset, "expected '" + type.displayName() + "', but found '" +
1264 expr->fType.displayName() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07001265 return nullptr;
1266 }
ethannicholasd598f792016-07-25 10:08:54 -07001267 if (type.kind() == Type::kScalar_Kind) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001268 std::vector<std::unique_ptr<Expression>> args;
1269 args.push_back(std::move(expr));
Ethan Nicholase1f55022019-02-05 17:17:40 -05001270 std::unique_ptr<Expression> ctor;
1271 if (type == *fContext.fFloatLiteral_Type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001272 ctor = this->convertIdentifier(ASTNode(&fFile->fNodes, -1, ASTNode::Kind::kIdentifier,
1273 "float"));
Ethan Nicholase1f55022019-02-05 17:17:40 -05001274 } else if (type == *fContext.fIntLiteral_Type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001275 ctor = this->convertIdentifier(ASTNode(&fFile->fNodes, -1, ASTNode::Kind::kIdentifier,
1276 "int"));
Ethan Nicholase1f55022019-02-05 17:17:40 -05001277 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001278 ctor = this->convertIdentifier(ASTNode(&fFile->fNodes, -1, ASTNode::Kind::kIdentifier,
1279 type.fName));
Ethan Nicholase1f55022019-02-05 17:17:40 -05001280 }
1281 if (!ctor) {
1282 printf("error, null identifier: %s\n", String(type.fName).c_str());
1283 }
Ethan Nicholasd9d33c32018-06-12 11:05:59 -04001284 SkASSERT(ctor);
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001285 return this->call(-1, std::move(ctor), std::move(args));
ethannicholasb3058bd2016-07-01 08:22:01 -07001286 }
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001287 if (expr->fKind == Expression::kNullLiteral_Kind) {
1288 SkASSERT(type.kind() == Type::kNullable_Kind);
1289 return std::unique_ptr<Expression>(new NullLiteral(expr->fOffset, type));
1290 }
ethannicholas5961bc92016-10-12 06:39:56 -07001291 std::vector<std::unique_ptr<Expression>> args;
1292 args.push_back(std::move(expr));
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001293 return std::unique_ptr<Expression>(new Constructor(-1, type, std::move(args)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001294}
1295
ethannicholasf789b382016-08-03 12:43:36 -07001296static bool is_matrix_multiply(const Type& left, const Type& right) {
1297 if (left.kind() == Type::kMatrix_Kind) {
1298 return right.kind() == Type::kMatrix_Kind || right.kind() == Type::kVector_Kind;
1299 }
1300 return left.kind() == Type::kVector_Kind && right.kind() == Type::kMatrix_Kind;
1301}
ethannicholasea4567c2016-10-17 11:24:37 -07001302
ethannicholasb3058bd2016-07-01 08:22:01 -07001303/**
1304 * Determines the operand and result types of a binary expression. Returns true if the expression is
1305 * legal, false otherwise. If false, the values of the out parameters are undefined.
1306 */
Ethan Nicholas11d53972016-11-28 11:23:23 -05001307static bool determine_binary_type(const Context& context,
1308 Token::Kind op,
1309 const Type& left,
1310 const Type& right,
ethannicholasd598f792016-07-25 10:08:54 -07001311 const Type** outLeftType,
1312 const Type** outRightType,
1313 const Type** outResultType,
ethannicholasb3058bd2016-07-01 08:22:01 -07001314 bool tryFlipped) {
1315 bool isLogical;
ethannicholasea4567c2016-10-17 11:24:37 -07001316 bool validMatrixOrVectorOp;
ethannicholasb3058bd2016-07-01 08:22:01 -07001317 switch (op) {
ethannicholasea4567c2016-10-17 11:24:37 -07001318 case Token::EQ:
1319 *outLeftType = &left;
1320 *outRightType = &left;
1321 *outResultType = &left;
1322 return right.canCoerceTo(left);
ethannicholasb3058bd2016-07-01 08:22:01 -07001323 case Token::EQEQ: // fall through
ethannicholasea4567c2016-10-17 11:24:37 -07001324 case Token::NEQ:
Ethan Nicholas23463002018-03-28 15:16:15 -04001325 if (right.canCoerceTo(left)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001326 *outLeftType = &left;
Ethan Nicholas23463002018-03-28 15:16:15 -04001327 *outRightType = &left;
1328 *outResultType = context.fBool_Type.get();
1329 return true;
1330 } if (left.canCoerceTo(right)) {
1331 *outLeftType = &right;
Ethan Nicholasaae47c82017-11-10 15:34:03 -05001332 *outRightType = &right;
1333 *outResultType = context.fBool_Type.get();
1334 return true;
1335 }
Ethan Nicholas23463002018-03-28 15:16:15 -04001336 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07001337 case Token::LT: // fall through
1338 case Token::GT: // fall through
1339 case Token::LTEQ: // fall through
1340 case Token::GTEQ:
1341 isLogical = true;
ethannicholasea4567c2016-10-17 11:24:37 -07001342 validMatrixOrVectorOp = false;
ethannicholasb3058bd2016-07-01 08:22:01 -07001343 break;
1344 case Token::LOGICALOR: // fall through
1345 case Token::LOGICALAND: // fall through
1346 case Token::LOGICALXOR: // fall through
1347 case Token::LOGICALOREQ: // fall through
1348 case Token::LOGICALANDEQ: // fall through
1349 case Token::LOGICALXOREQ:
ethannicholasd598f792016-07-25 10:08:54 -07001350 *outLeftType = context.fBool_Type.get();
1351 *outRightType = context.fBool_Type.get();
1352 *outResultType = context.fBool_Type.get();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001353 return left.canCoerceTo(*context.fBool_Type) &&
ethannicholasd598f792016-07-25 10:08:54 -07001354 right.canCoerceTo(*context.fBool_Type);
Ethan Nicholas11d53972016-11-28 11:23:23 -05001355 case Token::STAREQ:
Ethan Nicholasf7b88202017-09-18 14:10:39 -04001356 if (left.kind() == Type::kScalar_Kind) {
1357 *outLeftType = &left;
1358 *outRightType = &left;
1359 *outResultType = &left;
1360 return right.canCoerceTo(left);
1361 }
1362 // fall through
1363 case Token::STAR:
ethannicholasf789b382016-08-03 12:43:36 -07001364 if (is_matrix_multiply(left, right)) {
1365 // determine final component type
1366 if (determine_binary_type(context, Token::STAR, left.componentType(),
1367 right.componentType(), outLeftType, outRightType,
1368 outResultType, false)) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001369 *outLeftType = &(*outResultType)->toCompound(context, left.columns(),
Brian Salomon23356442018-11-30 15:33:19 -05001370 left.rows());
Ethan Nicholas11d53972016-11-28 11:23:23 -05001371 *outRightType = &(*outResultType)->toCompound(context, right.columns(),
Brian Salomon23356442018-11-30 15:33:19 -05001372 right.rows());
ethannicholasf789b382016-08-03 12:43:36 -07001373 int leftColumns = left.columns();
1374 int leftRows = left.rows();
1375 int rightColumns;
1376 int rightRows;
1377 if (right.kind() == Type::kVector_Kind) {
1378 // matrix * vector treats the vector as a column vector, so we need to
1379 // transpose it
1380 rightColumns = right.rows();
1381 rightRows = right.columns();
Ethan Nicholasd9d33c32018-06-12 11:05:59 -04001382 SkASSERT(rightColumns == 1);
ethannicholasf789b382016-08-03 12:43:36 -07001383 } else {
1384 rightColumns = right.columns();
1385 rightRows = right.rows();
1386 }
1387 if (rightColumns > 1) {
1388 *outResultType = &(*outResultType)->toCompound(context, rightColumns,
1389 leftRows);
1390 } else {
1391 // result was a column vector, transpose it back to a row
1392 *outResultType = &(*outResultType)->toCompound(context, leftRows,
1393 rightColumns);
1394 }
1395 return leftColumns == rightRows;
1396 } else {
1397 return false;
1398 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001399 }
ethannicholasea4567c2016-10-17 11:24:37 -07001400 isLogical = false;
1401 validMatrixOrVectorOp = true;
1402 break;
Ethan Nicholasf7b88202017-09-18 14:10:39 -04001403 case Token::PLUSEQ:
1404 case Token::MINUSEQ:
1405 case Token::SLASHEQ:
1406 case Token::PERCENTEQ:
1407 case Token::SHLEQ:
1408 case Token::SHREQ:
1409 if (left.kind() == Type::kScalar_Kind) {
1410 *outLeftType = &left;
1411 *outRightType = &left;
1412 *outResultType = &left;
1413 return right.canCoerceTo(left);
1414 }
1415 // fall through
ethannicholasea4567c2016-10-17 11:24:37 -07001416 case Token::PLUS: // fall through
ethannicholasea4567c2016-10-17 11:24:37 -07001417 case Token::MINUS: // fall through
ethannicholasea4567c2016-10-17 11:24:37 -07001418 case Token::SLASH: // fall through
ethannicholasea4567c2016-10-17 11:24:37 -07001419 isLogical = false;
1420 validMatrixOrVectorOp = true;
1421 break;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001422 case Token::COMMA:
1423 *outLeftType = &left;
1424 *outRightType = &right;
1425 *outResultType = &right;
1426 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07001427 default:
1428 isLogical = false;
ethannicholasea4567c2016-10-17 11:24:37 -07001429 validMatrixOrVectorOp = false;
ethannicholasb3058bd2016-07-01 08:22:01 -07001430 }
ethannicholasea4567c2016-10-17 11:24:37 -07001431 bool isVectorOrMatrix = left.kind() == Type::kVector_Kind || left.kind() == Type::kMatrix_Kind;
Ethan Nicholasf7b88202017-09-18 14:10:39 -04001432 if (left.kind() == Type::kScalar_Kind && right.kind() == Type::kScalar_Kind &&
1433 right.canCoerceTo(left)) {
1434 if (left.priority() > right.priority()) {
1435 *outLeftType = &left;
1436 *outRightType = &left;
1437 } else {
1438 *outLeftType = &right;
1439 *outRightType = &right;
1440 }
1441 if (isLogical) {
1442 *outResultType = context.fBool_Type.get();
1443 } else {
1444 *outResultType = &left;
1445 }
1446 return true;
1447 }
1448 if (right.canCoerceTo(left) && isVectorOrMatrix && validMatrixOrVectorOp) {
ethannicholasd598f792016-07-25 10:08:54 -07001449 *outLeftType = &left;
1450 *outRightType = &left;
ethannicholasb3058bd2016-07-01 08:22:01 -07001451 if (isLogical) {
ethannicholasd598f792016-07-25 10:08:54 -07001452 *outResultType = context.fBool_Type.get();
ethannicholasb3058bd2016-07-01 08:22:01 -07001453 } else {
ethannicholasd598f792016-07-25 10:08:54 -07001454 *outResultType = &left;
ethannicholasb3058bd2016-07-01 08:22:01 -07001455 }
1456 return true;
1457 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001458 if ((left.kind() == Type::kVector_Kind || left.kind() == Type::kMatrix_Kind) &&
ethannicholasd598f792016-07-25 10:08:54 -07001459 (right.kind() == Type::kScalar_Kind)) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001460 if (determine_binary_type(context, op, left.componentType(), right, outLeftType,
ethannicholasd598f792016-07-25 10:08:54 -07001461 outRightType, outResultType, false)) {
1462 *outLeftType = &(*outLeftType)->toCompound(context, left.columns(), left.rows());
ethannicholasb3058bd2016-07-01 08:22:01 -07001463 if (!isLogical) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001464 *outResultType = &(*outResultType)->toCompound(context, left.columns(),
ethannicholasd598f792016-07-25 10:08:54 -07001465 left.rows());
ethannicholasb3058bd2016-07-01 08:22:01 -07001466 }
1467 return true;
1468 }
1469 return false;
1470 }
1471 if (tryFlipped) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001472 return determine_binary_type(context, op, right, left, outRightType, outLeftType,
ethannicholasd598f792016-07-25 10:08:54 -07001473 outResultType, false);
ethannicholasb3058bd2016-07-01 08:22:01 -07001474 }
1475 return false;
1476}
1477
Michael Ludwig7b429ae2018-09-06 17:01:38 -04001478static std::unique_ptr<Expression> short_circuit_boolean(const Context& context,
1479 const Expression& left,
1480 Token::Kind op,
1481 const Expression& right) {
1482 SkASSERT(left.fKind == Expression::kBoolLiteral_Kind);
1483 bool leftVal = ((BoolLiteral&) left).fValue;
1484 if (op == Token::LOGICALAND) {
1485 // (true && expr) -> (expr) and (false && expr) -> (false)
1486 return leftVal ? right.clone()
1487 : std::unique_ptr<Expression>(new BoolLiteral(context, left.fOffset, false));
1488 } else if (op == Token::LOGICALOR) {
1489 // (true || expr) -> (true) and (false || expr) -> (expr)
1490 return leftVal ? std::unique_ptr<Expression>(new BoolLiteral(context, left.fOffset, true))
1491 : right.clone();
Noah Lavine334d0ba2019-12-18 23:03:49 -05001492 } else if (op == Token::LOGICALXOR) {
1493 // (true ^^ expr) -> !(expr) and (false ^^ expr) -> (expr)
1494 return leftVal ? std::unique_ptr<Expression>(new PrefixExpression(Token::LOGICALNOT,
1495 right.clone()))
1496 : right.clone();
Michael Ludwig7b429ae2018-09-06 17:01:38 -04001497 } else {
Michael Ludwig7b429ae2018-09-06 17:01:38 -04001498 return nullptr;
1499 }
1500}
1501
ethannicholas08a92112016-11-09 13:26:45 -08001502std::unique_ptr<Expression> IRGenerator::constantFold(const Expression& left,
1503 Token::Kind op,
Ethan Nicholas86a43402017-01-19 13:32:00 -05001504 const Expression& right) const {
Michael Ludwig7b429ae2018-09-06 17:01:38 -04001505 // If the left side is a constant boolean literal, the right side does not need to be constant
1506 // for short circuit optimizations to allow the constant to be folded.
1507 if (left.fKind == Expression::kBoolLiteral_Kind && !right.isConstant()) {
1508 return short_circuit_boolean(fContext, left, op, right);
1509 } else if (right.fKind == Expression::kBoolLiteral_Kind && !left.isConstant()) {
1510 // There aren't side effects in SKSL within expressions, so (left OP right) is equivalent to
1511 // (right OP left) for short-circuit optimizations
1512 return short_circuit_boolean(fContext, right, op, left);
1513 }
1514
1515 // Other than the short-circuit cases above, constant folding requires both sides to be constant
Ethan Nicholascb670962017-04-20 19:31:52 -04001516 if (!left.isConstant() || !right.isConstant()) {
1517 return nullptr;
1518 }
ethannicholas08a92112016-11-09 13:26:45 -08001519 // Note that we expressly do not worry about precision and overflow here -- we use the maximum
1520 // precision to calculate the results and hope the result makes sense. The plan is to move the
1521 // Skia caps into SkSL, so we have access to all of them including the precisions of the various
1522 // types, which will let us be more intelligent about this.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001523 if (left.fKind == Expression::kBoolLiteral_Kind &&
ethannicholas08a92112016-11-09 13:26:45 -08001524 right.fKind == Expression::kBoolLiteral_Kind) {
1525 bool leftVal = ((BoolLiteral&) left).fValue;
1526 bool rightVal = ((BoolLiteral&) right).fValue;
1527 bool result;
1528 switch (op) {
1529 case Token::LOGICALAND: result = leftVal && rightVal; break;
1530 case Token::LOGICALOR: result = leftVal || rightVal; break;
1531 case Token::LOGICALXOR: result = leftVal ^ rightVal; break;
1532 default: return nullptr;
1533 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001534 return std::unique_ptr<Expression>(new BoolLiteral(fContext, left.fOffset, result));
ethannicholas08a92112016-11-09 13:26:45 -08001535 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001536 #define RESULT(t, op) std::unique_ptr<Expression>(new t ## Literal(fContext, left.fOffset, \
ethannicholas08a92112016-11-09 13:26:45 -08001537 leftVal op rightVal))
1538 if (left.fKind == Expression::kIntLiteral_Kind && right.fKind == Expression::kIntLiteral_Kind) {
1539 int64_t leftVal = ((IntLiteral&) left).fValue;
1540 int64_t rightVal = ((IntLiteral&) right).fValue;
1541 switch (op) {
Ethan Nicholascb670962017-04-20 19:31:52 -04001542 case Token::PLUS: return RESULT(Int, +);
1543 case Token::MINUS: return RESULT(Int, -);
1544 case Token::STAR: return RESULT(Int, *);
Ethan Nicholas9a5610e2017-01-03 15:16:29 -05001545 case Token::SLASH:
1546 if (rightVal) {
1547 return RESULT(Int, /);
1548 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001549 fErrors.error(right.fOffset, "division by zero");
Ethan Nicholas9a5610e2017-01-03 15:16:29 -05001550 return nullptr;
Ethan Nicholas2503ab62017-01-05 10:44:25 -05001551 case Token::PERCENT:
1552 if (rightVal) {
1553 return RESULT(Int, %);
1554 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001555 fErrors.error(right.fOffset, "division by zero");
Ethan Nicholas2503ab62017-01-05 10:44:25 -05001556 return nullptr;
ethannicholas08a92112016-11-09 13:26:45 -08001557 case Token::BITWISEAND: return RESULT(Int, &);
1558 case Token::BITWISEOR: return RESULT(Int, |);
1559 case Token::BITWISEXOR: return RESULT(Int, ^);
ethannicholas08a92112016-11-09 13:26:45 -08001560 case Token::EQEQ: return RESULT(Bool, ==);
1561 case Token::NEQ: return RESULT(Bool, !=);
1562 case Token::GT: return RESULT(Bool, >);
1563 case Token::GTEQ: return RESULT(Bool, >=);
1564 case Token::LT: return RESULT(Bool, <);
1565 case Token::LTEQ: return RESULT(Bool, <=);
Ethan Nicholasfeba68a2019-06-10 09:56:29 -04001566 case Token::SHL:
1567 if (rightVal >= 0 && rightVal <= 31) {
1568 return RESULT(Int, <<);
1569 }
1570 fErrors.error(right.fOffset, "shift value out of range");
1571 return nullptr;
1572 case Token::SHR:
1573 if (rightVal >= 0 && rightVal <= 31) {
1574 return RESULT(Int, >>);
1575 }
1576 fErrors.error(right.fOffset, "shift value out of range");
1577 return nullptr;
1578
1579 default:
1580 return nullptr;
ethannicholas08a92112016-11-09 13:26:45 -08001581 }
1582 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001583 if (left.fKind == Expression::kFloatLiteral_Kind &&
ethannicholas08a92112016-11-09 13:26:45 -08001584 right.fKind == Expression::kFloatLiteral_Kind) {
1585 double leftVal = ((FloatLiteral&) left).fValue;
1586 double rightVal = ((FloatLiteral&) right).fValue;
1587 switch (op) {
1588 case Token::PLUS: return RESULT(Float, +);
1589 case Token::MINUS: return RESULT(Float, -);
1590 case Token::STAR: return RESULT(Float, *);
Ethan Nicholas9a5610e2017-01-03 15:16:29 -05001591 case Token::SLASH:
1592 if (rightVal) {
1593 return RESULT(Float, /);
1594 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001595 fErrors.error(right.fOffset, "division by zero");
Ethan Nicholas9a5610e2017-01-03 15:16:29 -05001596 return nullptr;
Ethan Nicholascb670962017-04-20 19:31:52 -04001597 case Token::EQEQ: return RESULT(Bool, ==);
1598 case Token::NEQ: return RESULT(Bool, !=);
1599 case Token::GT: return RESULT(Bool, >);
1600 case Token::GTEQ: return RESULT(Bool, >=);
1601 case Token::LT: return RESULT(Bool, <);
1602 case Token::LTEQ: return RESULT(Bool, <=);
ethannicholas08a92112016-11-09 13:26:45 -08001603 default: return nullptr;
1604 }
1605 }
Ethan Nicholase1f55022019-02-05 17:17:40 -05001606 if (left.fType.kind() == Type::kVector_Kind && left.fType.componentType().isFloat() &&
Ethan Nicholascb670962017-04-20 19:31:52 -04001607 left.fType == right.fType) {
Ethan Nicholascb670962017-04-20 19:31:52 -04001608 std::vector<std::unique_ptr<Expression>> args;
Ethan Nicholas4cf5fd92019-06-10 16:15:56 -04001609 #define RETURN_VEC_COMPONENTWISE_RESULT(op) \
1610 for (int i = 0; i < left.fType.columns(); i++) { \
1611 float value = left.getFVecComponent(i) op \
1612 right.getFVecComponent(i); \
1613 args.emplace_back(new FloatLiteral(fContext, -1, value)); \
1614 } \
1615 return std::unique_ptr<Expression>(new Constructor(-1, left.fType, \
Brian Salomon23356442018-11-30 15:33:19 -05001616 std::move(args)))
Ethan Nicholascb670962017-04-20 19:31:52 -04001617 switch (op) {
Ethan Nicholas3deaeb22017-04-25 14:42:11 -04001618 case Token::EQEQ:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001619 return std::unique_ptr<Expression>(new BoolLiteral(fContext, -1,
Ethan Nicholas3deaeb22017-04-25 14:42:11 -04001620 left.compareConstant(fContext, right)));
1621 case Token::NEQ:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001622 return std::unique_ptr<Expression>(new BoolLiteral(fContext, -1,
Ethan Nicholas3deaeb22017-04-25 14:42:11 -04001623 !left.compareConstant(fContext, right)));
Ethan Nicholascb670962017-04-20 19:31:52 -04001624 case Token::PLUS: RETURN_VEC_COMPONENTWISE_RESULT(+);
1625 case Token::MINUS: RETURN_VEC_COMPONENTWISE_RESULT(-);
1626 case Token::STAR: RETURN_VEC_COMPONENTWISE_RESULT(*);
Ethan Nicholas4cf5fd92019-06-10 16:15:56 -04001627 case Token::SLASH:
1628 for (int i = 0; i < left.fType.columns(); i++) {
1629 SKSL_FLOAT rvalue = right.getFVecComponent(i);
1630 if (rvalue == 0.0) {
1631 fErrors.error(right.fOffset, "division by zero");
1632 return nullptr;
1633 }
1634 float value = left.getFVecComponent(i) / rvalue;
1635 args.emplace_back(new FloatLiteral(fContext, -1, value));
1636 }
1637 return std::unique_ptr<Expression>(new Constructor(-1, left.fType,
1638 std::move(args)));
Ethan Nicholascb670962017-04-20 19:31:52 -04001639 default: return nullptr;
1640 }
1641 }
Ethan Nicholas3deaeb22017-04-25 14:42:11 -04001642 if (left.fType.kind() == Type::kMatrix_Kind &&
1643 right.fType.kind() == Type::kMatrix_Kind &&
1644 left.fKind == right.fKind) {
1645 switch (op) {
1646 case Token::EQEQ:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001647 return std::unique_ptr<Expression>(new BoolLiteral(fContext, -1,
Ethan Nicholas3deaeb22017-04-25 14:42:11 -04001648 left.compareConstant(fContext, right)));
1649 case Token::NEQ:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001650 return std::unique_ptr<Expression>(new BoolLiteral(fContext, -1,
Ethan Nicholas3deaeb22017-04-25 14:42:11 -04001651 !left.compareConstant(fContext, right)));
1652 default:
1653 return nullptr;
1654 }
1655 }
ethannicholas08a92112016-11-09 13:26:45 -08001656 #undef RESULT
1657 return nullptr;
1658}
1659
Ethan Nicholasfc994162019-06-06 10:04:27 -04001660std::unique_ptr<Expression> IRGenerator::convertBinaryExpression(const ASTNode& expression) {
1661 SkASSERT(expression.fKind == ASTNode::Kind::kBinary);
1662 auto iter = expression.begin();
1663 std::unique_ptr<Expression> left = this->convertExpression(*(iter++));
ethannicholasb3058bd2016-07-01 08:22:01 -07001664 if (!left) {
1665 return nullptr;
1666 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001667 std::unique_ptr<Expression> right = this->convertExpression(*(iter++));
ethannicholasb3058bd2016-07-01 08:22:01 -07001668 if (!right) {
1669 return nullptr;
1670 }
ethannicholasd598f792016-07-25 10:08:54 -07001671 const Type* leftType;
1672 const Type* rightType;
1673 const Type* resultType;
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001674 const Type* rawLeftType;
1675 if (left->fKind == Expression::kIntLiteral_Kind && right->fType.isInteger()) {
1676 rawLeftType = &right->fType;
1677 } else {
1678 rawLeftType = &left->fType;
1679 }
1680 const Type* rawRightType;
1681 if (right->fKind == Expression::kIntLiteral_Kind && left->fType.isInteger()) {
1682 rawRightType = &left->fType;
1683 } else {
1684 rawRightType = &right->fType;
1685 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001686 Token::Kind op = expression.getToken().fKind;
1687 if (!determine_binary_type(fContext, op, *rawLeftType, *rawRightType, &leftType, &rightType,
1688 &resultType, !Compiler::IsAssignment(op))) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001689 fErrors.error(expression.fOffset, String("type mismatch: '") +
Ethan Nicholasfc994162019-06-06 10:04:27 -04001690 Compiler::OperatorName(expression.getToken().fKind) +
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001691 "' cannot operate on '" + left->fType.displayName() +
1692 "', '" + right->fType.displayName() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07001693 return nullptr;
1694 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001695 if (Compiler::IsAssignment(op)) {
1696 this->setRefKind(*left, op != Token::EQ ? VariableReference::kReadWrite_RefKind :
1697 VariableReference::kWrite_RefKind);
ethannicholasea4567c2016-10-17 11:24:37 -07001698 }
1699 left = this->coerce(std::move(left), *leftType);
1700 right = this->coerce(std::move(right), *rightType);
1701 if (!left || !right) {
1702 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001703 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001704 std::unique_ptr<Expression> result = this->constantFold(*left.get(), op, *right.get());
ethannicholas08a92112016-11-09 13:26:45 -08001705 if (!result) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001706 result = std::unique_ptr<Expression>(new BinaryExpression(expression.fOffset,
ethannicholas08a92112016-11-09 13:26:45 -08001707 std::move(left),
Ethan Nicholasfc994162019-06-06 10:04:27 -04001708 op,
ethannicholas08a92112016-11-09 13:26:45 -08001709 std::move(right),
1710 *resultType));
1711 }
1712 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001713}
1714
Ethan Nicholasfc994162019-06-06 10:04:27 -04001715std::unique_ptr<Expression> IRGenerator::convertTernaryExpression(const ASTNode& node) {
1716 SkASSERT(node.fKind == ASTNode::Kind::kTernary);
1717 auto iter = node.begin();
1718 std::unique_ptr<Expression> test = this->coerce(this->convertExpression(*(iter++)),
ethannicholasd598f792016-07-25 10:08:54 -07001719 *fContext.fBool_Type);
ethannicholasb3058bd2016-07-01 08:22:01 -07001720 if (!test) {
1721 return nullptr;
1722 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001723 std::unique_ptr<Expression> ifTrue = this->convertExpression(*(iter++));
ethannicholasb3058bd2016-07-01 08:22:01 -07001724 if (!ifTrue) {
1725 return nullptr;
1726 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001727 std::unique_ptr<Expression> ifFalse = this->convertExpression(*(iter++));
ethannicholasb3058bd2016-07-01 08:22:01 -07001728 if (!ifFalse) {
1729 return nullptr;
1730 }
ethannicholasd598f792016-07-25 10:08:54 -07001731 const Type* trueType;
1732 const Type* falseType;
1733 const Type* resultType;
1734 if (!determine_binary_type(fContext, Token::EQEQ, ifTrue->fType, ifFalse->fType, &trueType,
Ethan Nicholas2be687a2017-01-03 16:44:39 -05001735 &falseType, &resultType, true) || trueType != falseType) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001736 fErrors.error(node.fOffset, "ternary operator result mismatch: '" +
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001737 ifTrue->fType.displayName() + "', '" +
1738 ifFalse->fType.displayName() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07001739 return nullptr;
1740 }
ethannicholasd598f792016-07-25 10:08:54 -07001741 ifTrue = this->coerce(std::move(ifTrue), *trueType);
Ethan Nicholas2be687a2017-01-03 16:44:39 -05001742 if (!ifTrue) {
1743 return nullptr;
1744 }
ethannicholasd598f792016-07-25 10:08:54 -07001745 ifFalse = this->coerce(std::move(ifFalse), *falseType);
Ethan Nicholas2be687a2017-01-03 16:44:39 -05001746 if (!ifFalse) {
1747 return nullptr;
1748 }
ethannicholas08a92112016-11-09 13:26:45 -08001749 if (test->fKind == Expression::kBoolLiteral_Kind) {
1750 // static boolean test, just return one of the branches
1751 if (((BoolLiteral&) *test).fValue) {
1752 return ifTrue;
1753 } else {
1754 return ifFalse;
1755 }
1756 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001757 return std::unique_ptr<Expression>(new TernaryExpression(node.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07001758 std::move(test),
Ethan Nicholas11d53972016-11-28 11:23:23 -05001759 std::move(ifTrue),
ethannicholasb3058bd2016-07-01 08:22:01 -07001760 std::move(ifFalse)));
1761}
1762
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001763std::unique_ptr<Expression> IRGenerator::call(int offset,
Ethan Nicholas11d53972016-11-28 11:23:23 -05001764 const FunctionDeclaration& function,
ethannicholasd598f792016-07-25 10:08:54 -07001765 std::vector<std::unique_ptr<Expression>> arguments) {
Ethan Nicholasdb80f692019-11-22 14:06:12 -05001766 if (function.fBuiltin) {
Ethan Nicholasb962eff2020-01-23 16:49:41 -05001767 auto found = fIntrinsics->find(function.declaration());
Ethan Nicholasdb80f692019-11-22 14:06:12 -05001768 if (found != fIntrinsics->end() && !found->second.second) {
1769 found->second.second = true;
1770 const FunctionDeclaration* old = fCurrentFunction;
1771 fCurrentFunction = nullptr;
1772 this->convertFunction(*((FunctionDefinition&) *found->second.first).fSource);
1773 fCurrentFunction = old;
1774 }
1775 }
ethannicholasd598f792016-07-25 10:08:54 -07001776 if (function.fParameters.size() != arguments.size()) {
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001777 String msg = "call to '" + function.fName + "' expected " +
Ethan Nicholas11d53972016-11-28 11:23:23 -05001778 to_string((uint64_t) function.fParameters.size()) +
ethannicholasb3058bd2016-07-01 08:22:01 -07001779 " argument";
ethannicholasd598f792016-07-25 10:08:54 -07001780 if (function.fParameters.size() != 1) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001781 msg += "s";
1782 }
ethannicholas5961bc92016-10-12 06:39:56 -07001783 msg += ", but found " + to_string((uint64_t) arguments.size());
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001784 fErrors.error(offset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -07001785 return nullptr;
1786 }
ethannicholas471e8942016-10-28 09:02:46 -07001787 std::vector<const Type*> types;
1788 const Type* returnType;
1789 if (!function.determineFinalTypes(arguments, &types, &returnType)) {
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001790 String msg = "no match for " + function.fName + "(";
1791 String separator;
ethannicholas471e8942016-10-28 09:02:46 -07001792 for (size_t i = 0; i < arguments.size(); i++) {
1793 msg += separator;
1794 separator = ", ";
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001795 msg += arguments[i]->fType.displayName();
ethannicholas471e8942016-10-28 09:02:46 -07001796 }
1797 msg += ")";
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001798 fErrors.error(offset, msg);
ethannicholas471e8942016-10-28 09:02:46 -07001799 return nullptr;
1800 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001801 for (size_t i = 0; i < arguments.size(); i++) {
ethannicholas471e8942016-10-28 09:02:46 -07001802 arguments[i] = this->coerce(std::move(arguments[i]), *types[i]);
ethannicholasea4567c2016-10-17 11:24:37 -07001803 if (!arguments[i]) {
1804 return nullptr;
1805 }
ethannicholasd598f792016-07-25 10:08:54 -07001806 if (arguments[i] && (function.fParameters[i]->fModifiers.fFlags & Modifiers::kOut_Flag)) {
Ethan Nicholas8f7e28f2018-03-26 14:24:27 -04001807 this->setRefKind(*arguments[i],
1808 function.fParameters[i]->fModifiers.fFlags & Modifiers::kIn_Flag ?
1809 VariableReference::kReadWrite_RefKind :
1810 VariableReference::kPointer_RefKind);
ethannicholasb3058bd2016-07-01 08:22:01 -07001811 }
1812 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001813 return std::unique_ptr<FunctionCall>(new FunctionCall(offset, *returnType, function,
ethannicholasb3058bd2016-07-01 08:22:01 -07001814 std::move(arguments)));
1815}
1816
1817/**
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001818 * Determines the cost of coercing the arguments of a function to the required types. Cost has no
1819 * particular meaning other than "lower costs are preferred". Returns INT_MAX if the call is not
1820 * valid.
ethannicholasb3058bd2016-07-01 08:22:01 -07001821 */
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001822int IRGenerator::callCost(const FunctionDeclaration& function,
1823 const std::vector<std::unique_ptr<Expression>>& arguments) {
ethannicholasd598f792016-07-25 10:08:54 -07001824 if (function.fParameters.size() != arguments.size()) {
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001825 return INT_MAX;
ethannicholasb3058bd2016-07-01 08:22:01 -07001826 }
1827 int total = 0;
ethannicholas471e8942016-10-28 09:02:46 -07001828 std::vector<const Type*> types;
1829 const Type* ignored;
1830 if (!function.determineFinalTypes(arguments, &types, &ignored)) {
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001831 return INT_MAX;
ethannicholas471e8942016-10-28 09:02:46 -07001832 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001833 for (size_t i = 0; i < arguments.size(); i++) {
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001834 int cost = arguments[i]->coercionCost(*types[i]);
1835 if (cost != INT_MAX) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001836 total += cost;
1837 } else {
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001838 return INT_MAX;
ethannicholasb3058bd2016-07-01 08:22:01 -07001839 }
1840 }
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001841 return total;
ethannicholasb3058bd2016-07-01 08:22:01 -07001842}
1843
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001844std::unique_ptr<Expression> IRGenerator::call(int offset,
Ethan Nicholas11d53972016-11-28 11:23:23 -05001845 std::unique_ptr<Expression> functionValue,
ethannicholasb3058bd2016-07-01 08:22:01 -07001846 std::vector<std::unique_ptr<Expression>> arguments) {
Ethan Nicholas9e6a3932019-05-17 16:31:21 -04001847 switch (functionValue->fKind) {
1848 case Expression::kTypeReference_Kind:
1849 return this->convertConstructor(offset,
1850 ((TypeReference&) *functionValue).fValue,
1851 std::move(arguments));
1852 case Expression::kExternalValue_Kind: {
1853 ExternalValue* v = ((ExternalValueReference&) *functionValue).fValue;
1854 if (!v->canCall()) {
1855 fErrors.error(offset, "this external value is not a function");
1856 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001857 }
Ethan Nicholas9e6a3932019-05-17 16:31:21 -04001858 int count = v->callParameterCount();
1859 if (count != (int) arguments.size()) {
1860 fErrors.error(offset, "external function expected " + to_string(count) +
1861 " arguments, but found " + to_string((int) arguments.size()));
1862 return nullptr;
1863 }
1864 static constexpr int PARAMETER_MAX = 16;
1865 SkASSERT(count < PARAMETER_MAX);
1866 const Type* types[PARAMETER_MAX];
1867 v->getCallParameterTypes(types);
1868 for (int i = 0; i < count; ++i) {
1869 arguments[i] = this->coerce(std::move(arguments[i]), *types[i]);
1870 if (!arguments[i]) {
1871 return nullptr;
1872 }
1873 }
1874 return std::unique_ptr<Expression>(new ExternalFunctionCall(offset, v->callReturnType(),
1875 v, std::move(arguments)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001876 }
Ethan Nicholas9e6a3932019-05-17 16:31:21 -04001877 case Expression::kFunctionReference_Kind: {
1878 FunctionReference* ref = (FunctionReference*) functionValue.get();
1879 int bestCost = INT_MAX;
1880 const FunctionDeclaration* best = nullptr;
1881 if (ref->fFunctions.size() > 1) {
1882 for (const auto& f : ref->fFunctions) {
1883 int cost = this->callCost(*f, arguments);
1884 if (cost < bestCost) {
1885 bestCost = cost;
1886 best = f;
1887 }
1888 }
1889 if (best) {
1890 return this->call(offset, *best, std::move(arguments));
1891 }
1892 String msg = "no match for " + ref->fFunctions[0]->fName + "(";
1893 String separator;
1894 for (size_t i = 0; i < arguments.size(); i++) {
1895 msg += separator;
1896 separator = ", ";
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001897 msg += arguments[i]->fType.displayName();
Ethan Nicholas9e6a3932019-05-17 16:31:21 -04001898 }
1899 msg += ")";
1900 fErrors.error(offset, msg);
1901 return nullptr;
1902 }
1903 return this->call(offset, *ref->fFunctions[0], std::move(arguments));
ethannicholasb3058bd2016-07-01 08:22:01 -07001904 }
Ethan Nicholas9e6a3932019-05-17 16:31:21 -04001905 default:
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001906 fErrors.error(offset, "not a function");
Ethan Nicholas9e6a3932019-05-17 16:31:21 -04001907 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001908 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001909}
1910
Ethan Nicholas84645e32017-02-09 13:57:14 -05001911std::unique_ptr<Expression> IRGenerator::convertNumberConstructor(
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001912 int offset,
Ethan Nicholas11d53972016-11-28 11:23:23 -05001913 const Type& type,
ethannicholasb3058bd2016-07-01 08:22:01 -07001914 std::vector<std::unique_ptr<Expression>> args) {
Ethan Nicholasd9d33c32018-06-12 11:05:59 -04001915 SkASSERT(type.isNumber());
Ethan Nicholas84645e32017-02-09 13:57:14 -05001916 if (args.size() != 1) {
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001917 fErrors.error(offset, "invalid arguments to '" + type.displayName() +
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001918 "' constructor, (expected exactly 1 argument, but found " +
1919 to_string((uint64_t) args.size()) + ")");
ethannicholasb3058bd2016-07-01 08:22:01 -07001920 return nullptr;
1921 }
Ethan Nicholasf7b88202017-09-18 14:10:39 -04001922 if (type == args[0]->fType) {
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001923 return std::move(args[0]);
1924 }
Ethan Nicholasf7b88202017-09-18 14:10:39 -04001925 if (type.isFloat() && args.size() == 1 && args[0]->fKind == Expression::kFloatLiteral_Kind) {
1926 double value = ((FloatLiteral&) *args[0]).fValue;
Ethan Nicholas00543112018-07-31 09:44:36 -04001927 return std::unique_ptr<Expression>(new FloatLiteral(offset, value, &type));
Ethan Nicholasdcba08e2017-08-02 10:52:54 -04001928 }
1929 if (type.isFloat() && args.size() == 1 && args[0]->fKind == Expression::kIntLiteral_Kind) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001930 int64_t value = ((IntLiteral&) *args[0]).fValue;
Ethan Nicholas00543112018-07-31 09:44:36 -04001931 return std::unique_ptr<Expression>(new FloatLiteral(offset, (double) value, &type));
ethannicholasb3058bd2016-07-01 08:22:01 -07001932 }
Ethan Nicholas84645e32017-02-09 13:57:14 -05001933 if (args[0]->fKind == Expression::kIntLiteral_Kind && (type == *fContext.fInt_Type ||
1934 type == *fContext.fUInt_Type)) {
Ethan Nicholas00543112018-07-31 09:44:36 -04001935 return std::unique_ptr<Expression>(new IntLiteral(offset,
Ethan Nicholas84645e32017-02-09 13:57:14 -05001936 ((IntLiteral&) *args[0]).fValue,
1937 &type));
ethannicholasb3058bd2016-07-01 08:22:01 -07001938 }
Ethan Nicholas84645e32017-02-09 13:57:14 -05001939 if (args[0]->fType == *fContext.fBool_Type) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001940 std::unique_ptr<IntLiteral> zero(new IntLiteral(fContext, offset, 0));
1941 std::unique_ptr<IntLiteral> one(new IntLiteral(fContext, offset, 1));
Ethan Nicholas84645e32017-02-09 13:57:14 -05001942 return std::unique_ptr<Expression>(
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001943 new TernaryExpression(offset, std::move(args[0]),
Ethan Nicholas84645e32017-02-09 13:57:14 -05001944 this->coerce(std::move(one), type),
1945 this->coerce(std::move(zero),
1946 type)));
1947 }
1948 if (!args[0]->fType.isNumber()) {
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001949 fErrors.error(offset, "invalid argument to '" + type.displayName() +
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001950 "' constructor (expected a number or bool, but found '" +
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001951 args[0]->fType.displayName() + "')");
Ethan Nicholas84645e32017-02-09 13:57:14 -05001952 return nullptr;
1953 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001954 return std::unique_ptr<Expression>(new Constructor(offset, type, std::move(args)));
Ethan Nicholas84645e32017-02-09 13:57:14 -05001955}
1956
1957int component_count(const Type& type) {
1958 switch (type.kind()) {
1959 case Type::kVector_Kind:
1960 return type.columns();
1961 case Type::kMatrix_Kind:
1962 return type.columns() * type.rows();
1963 default:
1964 return 1;
1965 }
1966}
1967
1968std::unique_ptr<Expression> IRGenerator::convertCompoundConstructor(
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001969 int offset,
Ethan Nicholas84645e32017-02-09 13:57:14 -05001970 const Type& type,
1971 std::vector<std::unique_ptr<Expression>> args) {
Ethan Nicholasd9d33c32018-06-12 11:05:59 -04001972 SkASSERT(type.kind() == Type::kVector_Kind || type.kind() == Type::kMatrix_Kind);
Ethan Nicholas84645e32017-02-09 13:57:14 -05001973 if (type.kind() == Type::kMatrix_Kind && args.size() == 1 &&
1974 args[0]->fType.kind() == Type::kMatrix_Kind) {
1975 // matrix from matrix is always legal
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001976 return std::unique_ptr<Expression>(new Constructor(offset, type, std::move(args)));
Ethan Nicholas84645e32017-02-09 13:57:14 -05001977 }
1978 int actual = 0;
1979 int expected = type.rows() * type.columns();
1980 if (args.size() != 1 || expected != component_count(args[0]->fType) ||
1981 type.componentType().isNumber() != args[0]->fType.componentType().isNumber()) {
ethannicholas5961bc92016-10-12 06:39:56 -07001982 for (size_t i = 0; i < args.size(); i++) {
Ethan Nicholas84645e32017-02-09 13:57:14 -05001983 if (args[i]->fType.kind() == Type::kVector_Kind) {
Ethan Nicholas49a36ba2017-02-09 17:04:23 +00001984 if (type.componentType().isNumber() !=
1985 args[i]->fType.componentType().isNumber()) {
Ethan Nicholas2a099da2020-01-02 14:40:54 -05001986 fErrors.error(offset, "'" + args[i]->fType.displayName() + "' is not a valid "
1987 "parameter to '" + type.displayName() +
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001988 "' constructor");
Ethan Nicholas49a36ba2017-02-09 17:04:23 +00001989 return nullptr;
1990 }
Ethan Nicholas84645e32017-02-09 13:57:14 -05001991 actual += args[i]->fType.columns();
Ethan Nicholas49a36ba2017-02-09 17:04:23 +00001992 } else if (args[i]->fType.kind() == Type::kScalar_Kind) {
1993 actual += 1;
1994 if (type.kind() != Type::kScalar_Kind) {
1995 args[i] = this->coerce(std::move(args[i]), type.componentType());
1996 if (!args[i]) {
1997 return nullptr;
1998 }
1999 }
2000 } else {
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002001 fErrors.error(offset, "'" + args[i]->fType.displayName() + "' is not a valid "
2002 "parameter to '" + type.displayName() + "' constructor");
Ethan Nicholas49a36ba2017-02-09 17:04:23 +00002003 return nullptr;
2004 }
2005 }
Ethan Nicholas84645e32017-02-09 13:57:14 -05002006 if (actual != 1 && actual != expected) {
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002007 fErrors.error(offset, "invalid arguments to '" + type.displayName() +
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002008 "' constructor (expected " + to_string(expected) +
2009 " scalars, but found " + to_string(actual) + ")");
Ethan Nicholas49a36ba2017-02-09 17:04:23 +00002010 return nullptr;
2011 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002012 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002013 return std::unique_ptr<Expression>(new Constructor(offset, type, std::move(args)));
ethannicholasb3058bd2016-07-01 08:22:01 -07002014}
2015
Ethan Nicholas84645e32017-02-09 13:57:14 -05002016std::unique_ptr<Expression> IRGenerator::convertConstructor(
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002017 int offset,
Ethan Nicholas84645e32017-02-09 13:57:14 -05002018 const Type& type,
2019 std::vector<std::unique_ptr<Expression>> args) {
2020 // FIXME: add support for structs
2021 Type::Kind kind = type.kind();
2022 if (args.size() == 1 && args[0]->fType == type) {
2023 // argument is already the right type, just return it
2024 return std::move(args[0]);
2025 }
2026 if (type.isNumber()) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002027 return this->convertNumberConstructor(offset, type, std::move(args));
Ethan Nicholas84645e32017-02-09 13:57:14 -05002028 } else if (kind == Type::kArray_Kind) {
2029 const Type& base = type.componentType();
2030 for (size_t i = 0; i < args.size(); i++) {
2031 args[i] = this->coerce(std::move(args[i]), base);
2032 if (!args[i]) {
2033 return nullptr;
2034 }
2035 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002036 return std::unique_ptr<Expression>(new Constructor(offset, type, std::move(args)));
Ethan Nicholas84645e32017-02-09 13:57:14 -05002037 } else if (kind == Type::kVector_Kind || kind == Type::kMatrix_Kind) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002038 return this->convertCompoundConstructor(offset, type, std::move(args));
Ethan Nicholas84645e32017-02-09 13:57:14 -05002039 } else {
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002040 fErrors.error(offset, "cannot construct '" + type.displayName() + "'");
Ethan Nicholas84645e32017-02-09 13:57:14 -05002041 return nullptr;
2042 }
2043}
2044
Ethan Nicholasfc994162019-06-06 10:04:27 -04002045std::unique_ptr<Expression> IRGenerator::convertPrefixExpression(const ASTNode& expression) {
2046 SkASSERT(expression.fKind == ASTNode::Kind::kPrefix);
2047 std::unique_ptr<Expression> base = this->convertExpression(*expression.begin());
ethannicholasb3058bd2016-07-01 08:22:01 -07002048 if (!base) {
2049 return nullptr;
2050 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002051 switch (expression.getToken().fKind) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002052 case Token::PLUS:
Ethan Nicholase1f55022019-02-05 17:17:40 -05002053 if (!base->fType.isNumber() && base->fType.kind() != Type::kVector_Kind &&
2054 base->fType != *fContext.fFloatLiteral_Type) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002055 fErrors.error(expression.fOffset,
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002056 "'+' cannot operate on '" + base->fType.displayName() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002057 return nullptr;
2058 }
2059 return base;
2060 case Token::MINUS:
ethannicholasb3058bd2016-07-01 08:22:01 -07002061 if (base->fKind == Expression::kIntLiteral_Kind) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002062 return std::unique_ptr<Expression>(new IntLiteral(fContext, base->fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07002063 -((IntLiteral&) *base).fValue));
2064 }
2065 if (base->fKind == Expression::kFloatLiteral_Kind) {
2066 double value = -((FloatLiteral&) *base).fValue;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002067 return std::unique_ptr<Expression>(new FloatLiteral(fContext, base->fOffset,
ethannicholasd598f792016-07-25 10:08:54 -07002068 value));
ethannicholasb3058bd2016-07-01 08:22:01 -07002069 }
Ethan Nicholase1f55022019-02-05 17:17:40 -05002070 if (!base->fType.isNumber() && base->fType.kind() != Type::kVector_Kind) {
2071 fErrors.error(expression.fOffset,
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002072 "'-' cannot operate on '" + base->fType.displayName() + "'");
Ethan Nicholase1f55022019-02-05 17:17:40 -05002073 return nullptr;
2074 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002075 return std::unique_ptr<Expression>(new PrefixExpression(Token::MINUS, std::move(base)));
2076 case Token::PLUSPLUS:
ethannicholasd598f792016-07-25 10:08:54 -07002077 if (!base->fType.isNumber()) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002078 fErrors.error(expression.fOffset,
Ethan Nicholasfc994162019-06-06 10:04:27 -04002079 String("'") + Compiler::OperatorName(expression.getToken().fKind) +
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002080 "' cannot operate on '" + base->fType.displayName() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002081 return nullptr;
2082 }
Ethan Nicholas8f7e28f2018-03-26 14:24:27 -04002083 this->setRefKind(*base, VariableReference::kReadWrite_RefKind);
ethannicholasb3058bd2016-07-01 08:22:01 -07002084 break;
Ethan Nicholas11d53972016-11-28 11:23:23 -05002085 case Token::MINUSMINUS:
ethannicholasd598f792016-07-25 10:08:54 -07002086 if (!base->fType.isNumber()) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002087 fErrors.error(expression.fOffset,
Ethan Nicholasfc994162019-06-06 10:04:27 -04002088 String("'") + Compiler::OperatorName(expression.getToken().fKind) +
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002089 "' cannot operate on '" + base->fType.displayName() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002090 return nullptr;
2091 }
Ethan Nicholas8f7e28f2018-03-26 14:24:27 -04002092 this->setRefKind(*base, VariableReference::kReadWrite_RefKind);
ethannicholasb3058bd2016-07-01 08:22:01 -07002093 break;
ethannicholas5961bc92016-10-12 06:39:56 -07002094 case Token::LOGICALNOT:
ethannicholasd598f792016-07-25 10:08:54 -07002095 if (base->fType != *fContext.fBool_Type) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002096 fErrors.error(expression.fOffset,
Ethan Nicholasfc994162019-06-06 10:04:27 -04002097 String("'") + Compiler::OperatorName(expression.getToken().fKind) +
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002098 "' cannot operate on '" + base->fType.displayName() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002099 return nullptr;
2100 }
ethannicholas08a92112016-11-09 13:26:45 -08002101 if (base->fKind == Expression::kBoolLiteral_Kind) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002102 return std::unique_ptr<Expression>(new BoolLiteral(fContext, base->fOffset,
ethannicholas08a92112016-11-09 13:26:45 -08002103 !((BoolLiteral&) *base).fValue));
2104 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002105 break;
ethannicholas5961bc92016-10-12 06:39:56 -07002106 case Token::BITWISENOT:
Brian Osman73fb39c2019-09-24 16:22:55 -04002107 if (base->fType != *fContext.fInt_Type && base->fType != *fContext.fUInt_Type) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002108 fErrors.error(expression.fOffset,
Ethan Nicholasfc994162019-06-06 10:04:27 -04002109 String("'") + Compiler::OperatorName(expression.getToken().fKind) +
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002110 "' cannot operate on '" + base->fType.displayName() + "'");
ethannicholas5961bc92016-10-12 06:39:56 -07002111 return nullptr;
2112 }
2113 break;
Ethan Nicholas11d53972016-11-28 11:23:23 -05002114 default:
ethannicholasb3058bd2016-07-01 08:22:01 -07002115 ABORT("unsupported prefix operator\n");
2116 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002117 return std::unique_ptr<Expression>(new PrefixExpression(expression.getToken().fKind,
ethannicholasb3058bd2016-07-01 08:22:01 -07002118 std::move(base)));
2119}
2120
2121std::unique_ptr<Expression> IRGenerator::convertIndex(std::unique_ptr<Expression> base,
Ethan Nicholasfc994162019-06-06 10:04:27 -04002122 const ASTNode& index) {
Ethan Nicholas50afc172017-02-16 14:49:57 -05002123 if (base->fKind == Expression::kTypeReference_Kind) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002124 if (index.fKind == ASTNode::Kind::kInt) {
Ethan Nicholas50afc172017-02-16 14:49:57 -05002125 const Type& oldType = ((TypeReference&) *base).fValue;
Ethan Nicholasfc994162019-06-06 10:04:27 -04002126 SKSL_INT size = index.getInt();
Ethan Nicholas91164d12019-05-15 15:29:54 -04002127 Type* newType = (Type*) fSymbolTable->takeOwnership(std::unique_ptr<Symbol>(
2128 new Type(oldType.name() + "[" + to_string(size) + "]",
Ethan Nicholasfc994162019-06-06 10:04:27 -04002129 Type::kArray_Kind, oldType, size)));
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002130 return std::unique_ptr<Expression>(new TypeReference(fContext, base->fOffset,
Ethan Nicholas50afc172017-02-16 14:49:57 -05002131 *newType));
2132
2133 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002134 fErrors.error(base->fOffset, "array size must be a constant");
Ethan Nicholas50afc172017-02-16 14:49:57 -05002135 return nullptr;
2136 }
2137 }
ethannicholas5961bc92016-10-12 06:39:56 -07002138 if (base->fType.kind() != Type::kArray_Kind && base->fType.kind() != Type::kMatrix_Kind &&
2139 base->fType.kind() != Type::kVector_Kind) {
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002140 fErrors.error(base->fOffset, "expected array, but found '" + base->fType.displayName() +
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002141 "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002142 return nullptr;
2143 }
2144 std::unique_ptr<Expression> converted = this->convertExpression(index);
2145 if (!converted) {
2146 return nullptr;
2147 }
ethannicholas5961bc92016-10-12 06:39:56 -07002148 if (converted->fType != *fContext.fUInt_Type) {
2149 converted = this->coerce(std::move(converted), *fContext.fInt_Type);
2150 if (!converted) {
2151 return nullptr;
2152 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002153 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05002154 return std::unique_ptr<Expression>(new IndexExpression(fContext, std::move(base),
ethannicholasd598f792016-07-25 10:08:54 -07002155 std::move(converted)));
ethannicholasb3058bd2016-07-01 08:22:01 -07002156}
2157
2158std::unique_ptr<Expression> IRGenerator::convertField(std::unique_ptr<Expression> base,
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002159 StringFragment field) {
Ethan Nicholas91164d12019-05-15 15:29:54 -04002160 if (base->fKind == Expression::kExternalValue_Kind) {
2161 ExternalValue& ev = *((ExternalValueReference&) *base).fValue;
2162 ExternalValue* result = ev.getChild(String(field).c_str());
2163 if (!result) {
2164 fErrors.error(base->fOffset, "external value does not have a child named '" + field +
2165 "'");
2166 return nullptr;
2167 }
2168 return std::unique_ptr<Expression>(new ExternalValueReference(base->fOffset, result));
2169 }
ethannicholasd598f792016-07-25 10:08:54 -07002170 auto fields = base->fType.fields();
ethannicholasb3058bd2016-07-01 08:22:01 -07002171 for (size_t i = 0; i < fields.size(); i++) {
2172 if (fields[i].fName == field) {
2173 return std::unique_ptr<Expression>(new FieldAccess(std::move(base), (int) i));
2174 }
2175 }
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002176 fErrors.error(base->fOffset, "type '" + base->fType.displayName() + "' does not have a "
Ethan Nicholasb962eff2020-01-23 16:49:41 -05002177 "field named '" + field + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002178 return nullptr;
2179}
2180
2181std::unique_ptr<Expression> IRGenerator::convertSwizzle(std::unique_ptr<Expression> base,
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002182 StringFragment fields) {
ethannicholasd598f792016-07-25 10:08:54 -07002183 if (base->fType.kind() != Type::kVector_Kind) {
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002184 fErrors.error(base->fOffset, "cannot swizzle type '" + base->fType.displayName() + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002185 return nullptr;
2186 }
2187 std::vector<int> swizzleComponents;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002188 for (size_t i = 0; i < fields.fLength; i++) {
Ethan Nicholas9e1138d2016-11-21 10:39:35 -05002189 switch (fields[i]) {
Ethan Nicholasac285b12019-02-12 16:05:18 -05002190 case '0':
Ethan Nicholasac285b12019-02-12 16:05:18 -05002191 swizzleComponents.push_back(SKSL_SWIZZLE_0);
2192 break;
2193 case '1':
Ethan Nicholasac285b12019-02-12 16:05:18 -05002194 swizzleComponents.push_back(SKSL_SWIZZLE_1);
2195 break;
Ethan Nicholase455f652019-09-13 12:52:55 -04002196 case 'x':
2197 case 'r':
Ethan Nicholas11d53972016-11-28 11:23:23 -05002198 case 's':
Ethan Nicholase455f652019-09-13 12:52:55 -04002199 case 'L':
ethannicholasb3058bd2016-07-01 08:22:01 -07002200 swizzleComponents.push_back(0);
2201 break;
Ethan Nicholase455f652019-09-13 12:52:55 -04002202 case 'y':
2203 case 'g':
ethannicholasb3058bd2016-07-01 08:22:01 -07002204 case 't':
Ethan Nicholase455f652019-09-13 12:52:55 -04002205 case 'T':
ethannicholasd598f792016-07-25 10:08:54 -07002206 if (base->fType.columns() >= 2) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002207 swizzleComponents.push_back(1);
2208 break;
2209 }
2210 // fall through
Ethan Nicholase455f652019-09-13 12:52:55 -04002211 case 'z':
2212 case 'b':
Ethan Nicholas11d53972016-11-28 11:23:23 -05002213 case 'p':
Ethan Nicholase455f652019-09-13 12:52:55 -04002214 case 'R':
ethannicholasd598f792016-07-25 10:08:54 -07002215 if (base->fType.columns() >= 3) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002216 swizzleComponents.push_back(2);
2217 break;
2218 }
2219 // fall through
Ethan Nicholase455f652019-09-13 12:52:55 -04002220 case 'w':
2221 case 'a':
ethannicholasb3058bd2016-07-01 08:22:01 -07002222 case 'q':
Ethan Nicholase455f652019-09-13 12:52:55 -04002223 case 'B':
ethannicholasd598f792016-07-25 10:08:54 -07002224 if (base->fType.columns() >= 4) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002225 swizzleComponents.push_back(3);
2226 break;
2227 }
2228 // fall through
2229 default:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002230 fErrors.error(base->fOffset, String::printf("invalid swizzle component '%c'",
2231 fields[i]));
ethannicholasb3058bd2016-07-01 08:22:01 -07002232 return nullptr;
2233 }
2234 }
Ethan Nicholasd9d33c32018-06-12 11:05:59 -04002235 SkASSERT(swizzleComponents.size() > 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07002236 if (swizzleComponents.size() > 4) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002237 fErrors.error(base->fOffset, "too many components in swizzle mask '" + fields + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002238 return nullptr;
2239 }
ethannicholasd598f792016-07-25 10:08:54 -07002240 return std::unique_ptr<Expression>(new Swizzle(fContext, std::move(base), swizzleComponents));
ethannicholasb3058bd2016-07-01 08:22:01 -07002241}
2242
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002243std::unique_ptr<Expression> IRGenerator::getCap(int offset, String name) {
Ethan Nicholas941e7e22016-12-12 15:33:30 -05002244 auto found = fCapsMap.find(name);
2245 if (found == fCapsMap.end()) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002246 fErrors.error(offset, "unknown capability flag '" + name + "'");
Ethan Nicholas3605ace2016-11-21 15:59:48 -05002247 return nullptr;
2248 }
Ethan Nicholas762466e2017-06-29 10:03:38 -04002249 String fullName = "sk_Caps." + name;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002250 return std::unique_ptr<Expression>(new Setting(offset, fullName,
2251 found->second.literal(fContext, offset)));
Ethan Nicholas762466e2017-06-29 10:03:38 -04002252}
2253
Ethan Nicholas00543112018-07-31 09:44:36 -04002254std::unique_ptr<Expression> IRGenerator::getArg(int offset, String name) const {
Ethan Nicholas762466e2017-06-29 10:03:38 -04002255 auto found = fSettings->fArgs.find(name);
2256 if (found == fSettings->fArgs.end()) {
Ethan Nicholas762466e2017-06-29 10:03:38 -04002257 return nullptr;
Ethan Nicholas3605ace2016-11-21 15:59:48 -05002258 }
Ethan Nicholas762466e2017-06-29 10:03:38 -04002259 String fullName = "sk_Args." + name;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002260 return std::unique_ptr<Expression>(new Setting(offset,
Ethan Nicholas762466e2017-06-29 10:03:38 -04002261 fullName,
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002262 found->second.literal(fContext, offset)));
Ethan Nicholas3605ace2016-11-21 15:59:48 -05002263}
2264
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002265std::unique_ptr<Expression> IRGenerator::convertTypeField(int offset, const Type& type,
2266 StringFragment field) {
2267 std::unique_ptr<Expression> result;
2268 for (const auto& e : *fProgramElements) {
2269 if (e->fKind == ProgramElement::kEnum_Kind && type.name() == ((Enum&) *e).fTypeName) {
2270 std::shared_ptr<SymbolTable> old = fSymbolTable;
2271 fSymbolTable = ((Enum&) *e).fSymbols;
Ethan Nicholasfc994162019-06-06 10:04:27 -04002272 result = convertIdentifier(ASTNode(&fFile->fNodes, offset, ASTNode::Kind::kIdentifier,
2273 field));
Ethan Nicholasdb80f692019-11-22 14:06:12 -05002274 SkASSERT(result->fKind == Expression::kVariableReference_Kind);
2275 const Variable& v = ((VariableReference&) *result).fVariable;
2276 SkASSERT(v.fInitialValue);
2277 SkASSERT(v.fInitialValue->fKind == Expression::kIntLiteral_Kind);
2278 result.reset(new IntLiteral(offset, ((IntLiteral&) *v.fInitialValue).fValue, &type));
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002279 fSymbolTable = old;
Ethan Nicholasdb80f692019-11-22 14:06:12 -05002280 break;
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002281 }
2282 }
2283 if (!result) {
Ethan Nicholasdb80f692019-11-22 14:06:12 -05002284 auto found = fIntrinsics->find(type.fName);
2285 if (found != fIntrinsics->end()) {
2286 SkASSERT(!found->second.second);
2287 found->second.second = true;
2288 fProgramElements->push_back(found->second.first->clone());
2289 return this->convertTypeField(offset, type, field);
2290 }
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002291 fErrors.error(offset, "type '" + type.fName + "' does not have a field named '" + field +
2292 "'");
2293 }
2294 return result;
2295}
2296
Ethan Nicholasfc994162019-06-06 10:04:27 -04002297std::unique_ptr<Expression> IRGenerator::convertIndexExpression(const ASTNode& index) {
2298 SkASSERT(index.fKind == ASTNode::Kind::kIndex);
2299 auto iter = index.begin();
2300 std::unique_ptr<Expression> base = this->convertExpression(*(iter++));
ethannicholasb3058bd2016-07-01 08:22:01 -07002301 if (!base) {
2302 return nullptr;
2303 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002304 if (iter != index.end()) {
2305 return this->convertIndex(std::move(base), *(iter++));
2306 } else if (base->fKind == Expression::kTypeReference_Kind) {
2307 const Type& oldType = ((TypeReference&) *base).fValue;
2308 Type* newType = (Type*) fSymbolTable->takeOwnership(std::unique_ptr<Symbol>(
Ethan Nicholas91164d12019-05-15 15:29:54 -04002309 new Type(oldType.name() + "[]",
2310 Type::kArray_Kind,
2311 oldType,
2312 -1)));
Ethan Nicholasfc994162019-06-06 10:04:27 -04002313 return std::unique_ptr<Expression>(new TypeReference(fContext, base->fOffset,
2314 *newType));
ethannicholasb3058bd2016-07-01 08:22:01 -07002315 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002316 fErrors.error(index.fOffset, "'[]' must follow a type name");
2317 return nullptr;
2318}
2319
2320std::unique_ptr<Expression> IRGenerator::convertCallExpression(const ASTNode& callNode) {
2321 SkASSERT(callNode.fKind == ASTNode::Kind::kCall);
2322 auto iter = callNode.begin();
2323 std::unique_ptr<Expression> base = this->convertExpression(*(iter++));
2324 if (!base) {
2325 return nullptr;
2326 }
2327 std::vector<std::unique_ptr<Expression>> arguments;
2328 for (; iter != callNode.end(); ++iter) {
2329 std::unique_ptr<Expression> converted = this->convertExpression(*iter);
2330 if (!converted) {
2331 return nullptr;
2332 }
2333 arguments.push_back(std::move(converted));
2334 }
2335 return this->call(callNode.fOffset, std::move(base), std::move(arguments));
2336}
2337
2338std::unique_ptr<Expression> IRGenerator::convertFieldExpression(const ASTNode& fieldNode) {
2339 std::unique_ptr<Expression> base = this->convertExpression(*fieldNode.begin());
2340 if (!base) {
2341 return nullptr;
2342 }
2343 StringFragment field = fieldNode.getString();
2344 if (base->fType == *fContext.fSkCaps_Type) {
2345 return this->getCap(fieldNode.fOffset, field);
2346 }
2347 if (base->fType == *fContext.fSkArgs_Type) {
2348 return this->getArg(fieldNode.fOffset, field);
2349 }
2350 if (base->fKind == Expression::kTypeReference_Kind) {
2351 return this->convertTypeField(base->fOffset, ((TypeReference&) *base).fValue,
2352 field);
2353 }
2354 if (base->fKind == Expression::kExternalValue_Kind) {
2355 return this->convertField(std::move(base), field);
2356 }
2357 switch (base->fType.kind()) {
2358 case Type::kVector_Kind:
2359 return this->convertSwizzle(std::move(base), field);
2360 case Type::kOther_Kind:
2361 case Type::kStruct_Kind:
2362 return this->convertField(std::move(base), field);
2363 default:
2364 fErrors.error(base->fOffset, "cannot swizzle value of type '" +
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002365 base->fType.displayName() + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002366 return nullptr;
2367 }
2368}
2369
2370std::unique_ptr<Expression> IRGenerator::convertPostfixExpression(const ASTNode& expression) {
2371 std::unique_ptr<Expression> base = this->convertExpression(*expression.begin());
2372 if (!base) {
2373 return nullptr;
2374 }
2375 if (!base->fType.isNumber()) {
2376 fErrors.error(expression.fOffset,
2377 "'" + String(Compiler::OperatorName(expression.getToken().fKind)) +
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002378 "' cannot operate on '" + base->fType.displayName() + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002379 return nullptr;
2380 }
2381 this->setRefKind(*base, VariableReference::kReadWrite_RefKind);
2382 return std::unique_ptr<Expression>(new PostfixExpression(std::move(base),
2383 expression.getToken().fKind));
ethannicholasb3058bd2016-07-01 08:22:01 -07002384}
2385
2386void IRGenerator::checkValid(const Expression& expr) {
2387 switch (expr.fKind) {
2388 case Expression::kFunctionReference_Kind:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002389 fErrors.error(expr.fOffset, "expected '(' to begin function call");
ethannicholasb3058bd2016-07-01 08:22:01 -07002390 break;
2391 case Expression::kTypeReference_Kind:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002392 fErrors.error(expr.fOffset, "expected '(' to begin constructor invocation");
ethannicholasb3058bd2016-07-01 08:22:01 -07002393 break;
2394 default:
ethannicholasea4567c2016-10-17 11:24:37 -07002395 if (expr.fType == *fContext.fInvalid_Type) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002396 fErrors.error(expr.fOffset, "invalid expression");
ethannicholasea4567c2016-10-17 11:24:37 -07002397 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002398 }
2399}
2400
Ethan Nicholascb0f4092019-04-19 11:26:50 -04002401bool IRGenerator::checkSwizzleWrite(const Swizzle& swizzle) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002402 int bits = 0;
2403 for (int idx : swizzle.fComponents) {
Ethan Nicholascb0f4092019-04-19 11:26:50 -04002404 if (idx < 0) {
2405 fErrors.error(swizzle.fOffset, "cannot write to a swizzle mask containing a constant");
2406 return false;
2407 }
2408 SkASSERT(idx <= 3);
ethannicholasb3058bd2016-07-01 08:22:01 -07002409 int bit = 1 << idx;
2410 if (bits & bit) {
Ethan Nicholascb0f4092019-04-19 11:26:50 -04002411 fErrors.error(swizzle.fOffset,
2412 "cannot write to the same swizzle field more than once");
2413 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07002414 }
2415 bits |= bit;
2416 }
Ethan Nicholascb0f4092019-04-19 11:26:50 -04002417 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07002418}
2419
Ethan Nicholas8f7e28f2018-03-26 14:24:27 -04002420void IRGenerator::setRefKind(const Expression& expr, VariableReference::RefKind kind) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002421 switch (expr.fKind) {
2422 case Expression::kVariableReference_Kind: {
ethannicholasd598f792016-07-25 10:08:54 -07002423 const Variable& var = ((VariableReference&) expr).fVariable;
ethannicholasb3058bd2016-07-01 08:22:01 -07002424 if (var.fModifiers.fFlags & (Modifiers::kConst_Flag | Modifiers::kUniform_Flag)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002425 fErrors.error(expr.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07002426 "cannot modify immutable variable '" + var.fName + "'");
2427 }
Ethan Nicholas8f7e28f2018-03-26 14:24:27 -04002428 ((VariableReference&) expr).setRefKind(kind);
ethannicholasb3058bd2016-07-01 08:22:01 -07002429 break;
2430 }
2431 case Expression::kFieldAccess_Kind:
Ethan Nicholas8f7e28f2018-03-26 14:24:27 -04002432 this->setRefKind(*((FieldAccess&) expr).fBase, kind);
ethannicholasb3058bd2016-07-01 08:22:01 -07002433 break;
Ethan Nicholascb0f4092019-04-19 11:26:50 -04002434 case Expression::kSwizzle_Kind: {
2435 const Swizzle& swizzle = (Swizzle&) expr;
2436 this->checkSwizzleWrite(swizzle);
2437 this->setRefKind(*swizzle.fBase, kind);
ethannicholasb3058bd2016-07-01 08:22:01 -07002438 break;
Ethan Nicholascb0f4092019-04-19 11:26:50 -04002439 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002440 case Expression::kIndex_Kind:
Ethan Nicholas8f7e28f2018-03-26 14:24:27 -04002441 this->setRefKind(*((IndexExpression&) expr).fBase, kind);
ethannicholasb3058bd2016-07-01 08:22:01 -07002442 break;
Ethan Nicholasa583b812018-01-18 13:32:11 -05002443 case Expression::kTernary_Kind: {
2444 TernaryExpression& t = (TernaryExpression&) expr;
Ethan Nicholas8f7e28f2018-03-26 14:24:27 -04002445 this->setRefKind(*t.fIfTrue, kind);
2446 this->setRefKind(*t.fIfFalse, kind);
Ethan Nicholasa583b812018-01-18 13:32:11 -05002447 break;
2448 }
Ethan Nicholas91164d12019-05-15 15:29:54 -04002449 case Expression::kExternalValue_Kind: {
2450 const ExternalValue& v = *((ExternalValueReference&) expr).fValue;
2451 if (!v.canWrite()) {
2452 fErrors.error(expr.fOffset,
2453 "cannot modify immutable external value '" + v.fName + "'");
2454 }
2455 break;
2456 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002457 default:
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002458 fErrors.error(expr.fOffset, "cannot assign to this expression");
ethannicholasb3058bd2016-07-01 08:22:01 -07002459 break;
2460 }
2461}
2462
Robert Phillipsfe8da172018-01-24 14:52:02 +00002463void IRGenerator::convertProgram(Program::Kind kind,
2464 const char* text,
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002465 size_t length,
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002466 SymbolTable& types,
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002467 std::vector<std::unique_ptr<ProgramElement>>* out) {
Robert Phillipsfe8da172018-01-24 14:52:02 +00002468 fKind = kind;
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002469 fProgramElements = out;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002470 Parser parser(text, length, types, fErrors);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002471 fFile = parser.file();
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002472 if (fErrors.errorCount()) {
2473 return;
2474 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002475 SkASSERT(fFile);
2476 for (const auto& decl : fFile->root()) {
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002477 switch (decl.fKind) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002478 case ASTNode::Kind::kVarDeclarations: {
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002479 std::unique_ptr<VarDeclarations> s = this->convertVarDeclarations(
Ethan Nicholasfc994162019-06-06 10:04:27 -04002480 decl,
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002481 Variable::kGlobal_Storage);
2482 if (s) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002483 fProgramElements->push_back(std::move(s));
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002484 }
2485 break;
2486 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002487 case ASTNode::Kind::kEnum: {
2488 this->convertEnum(decl);
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002489 break;
2490 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002491 case ASTNode::Kind::kFunction: {
2492 this->convertFunction(decl);
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002493 break;
2494 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002495 case ASTNode::Kind::kModifiers: {
2496 std::unique_ptr<ModifiersDeclaration> f = this->convertModifiersDeclaration(decl);
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002497 if (f) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002498 fProgramElements->push_back(std::move(f));
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002499 }
2500 break;
2501 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002502 case ASTNode::Kind::kInterfaceBlock: {
2503 std::unique_ptr<InterfaceBlock> i = this->convertInterfaceBlock(decl);
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002504 if (i) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002505 fProgramElements->push_back(std::move(i));
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002506 }
2507 break;
2508 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002509 case ASTNode::Kind::kExtension: {
2510 std::unique_ptr<Extension> e = this->convertExtension(decl.fOffset,
2511 decl.getString());
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002512 if (e) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002513 fProgramElements->push_back(std::move(e));
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002514 }
2515 break;
2516 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002517 case ASTNode::Kind::kSection: {
2518 std::unique_ptr<Section> s = this->convertSection(decl);
Ethan Nicholas762466e2017-06-29 10:03:38 -04002519 if (s) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -05002520 fProgramElements->push_back(std::move(s));
Ethan Nicholas762466e2017-06-29 10:03:38 -04002521 }
2522 break;
2523 }
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002524 default:
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002525#ifdef SK_DEBUG
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002526 ABORT("unsupported declaration: %s\n", decl.description().c_str());
Ethan Nicholas2a099da2020-01-02 14:40:54 -05002527#endif
2528 break;
Ethan Nicholas7da6dfa2017-06-21 11:25:18 -04002529 }
2530 }
2531}
2532
2533
ethannicholasb3058bd2016-07-01 08:22:01 -07002534}