blob: 478d6fd673774f9aa81ea1d96b6002d5264f89c5 [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
ethannicholasb3058bd2016-07-01 08:22:01 -07008#include "stdio.h"
9#include "SkSLParser.h"
10#include "SkSLToken.h"
11
12#define register
13#ifdef __clang__
14#pragma clang diagnostic push
15#pragma clang diagnostic ignored "-Wunneeded-internal-declaration"
ethannicholas8e5a5882016-07-06 06:55:44 -070016#pragma clang diagnostic ignored "-Wnull-conversion"
ethannicholasf789b382016-08-03 12:43:36 -070017#pragma clang diagnostic ignored "-Wsign-compare"
18#endif
19#ifdef __GNUC__
20#pragma GCC diagnostic push
21#pragma GCC diagnostic ignored "-Wsign-compare"
22#endif
23#ifdef _MSC_VER
24#pragma warning(push)
25#pragma warning(disable:4018)
ethannicholasb3058bd2016-07-01 08:22:01 -070026#endif
27#include "lex.sksl.c"
28#ifdef __clang__
29#pragma clang diagnostic pop
30#endif
ethannicholasf789b382016-08-03 12:43:36 -070031#ifdef __GNUC__
32#pragma GCC diagnostic pop
33#endif
34#ifdef _MSC_VER
35#pragma warning(pop)
36#endif
ethannicholasb3058bd2016-07-01 08:22:01 -070037#undef register
38
39#include "ast/SkSLASTBinaryExpression.h"
40#include "ast/SkSLASTBlock.h"
41#include "ast/SkSLASTBoolLiteral.h"
42#include "ast/SkSLASTBreakStatement.h"
43#include "ast/SkSLASTCallSuffix.h"
44#include "ast/SkSLASTContinueStatement.h"
45#include "ast/SkSLASTDiscardStatement.h"
46#include "ast/SkSLASTDoStatement.h"
47#include "ast/SkSLASTExpression.h"
48#include "ast/SkSLASTExpressionStatement.h"
49#include "ast/SkSLASTExtension.h"
50#include "ast/SkSLASTFieldSuffix.h"
51#include "ast/SkSLASTFloatLiteral.h"
52#include "ast/SkSLASTForStatement.h"
53#include "ast/SkSLASTFunction.h"
54#include "ast/SkSLASTIdentifier.h"
55#include "ast/SkSLASTIfStatement.h"
56#include "ast/SkSLASTIndexSuffix.h"
57#include "ast/SkSLASTInterfaceBlock.h"
58#include "ast/SkSLASTIntLiteral.h"
ethannicholas5961bc92016-10-12 06:39:56 -070059#include "ast/SkSLASTModifiersDeclaration.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070060#include "ast/SkSLASTParameter.h"
ethannicholas5961bc92016-10-12 06:39:56 -070061#include "ast/SkSLASTPrecision.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070062#include "ast/SkSLASTPrefixExpression.h"
63#include "ast/SkSLASTReturnStatement.h"
64#include "ast/SkSLASTStatement.h"
65#include "ast/SkSLASTSuffixExpression.h"
Ethan Nicholasaf197692017-02-27 13:26:45 -050066#include "ast/SkSLASTSwitchCase.h"
67#include "ast/SkSLASTSwitchStatement.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070068#include "ast/SkSLASTTernaryExpression.h"
69#include "ast/SkSLASTType.h"
70#include "ast/SkSLASTVarDeclaration.h"
71#include "ast/SkSLASTVarDeclarationStatement.h"
72#include "ast/SkSLASTWhileStatement.h"
73#include "ir/SkSLSymbolTable.h"
Ethan Nicholas11d53972016-11-28 11:23:23 -050074#include "ir/SkSLModifiers.h"
ethannicholasd598f792016-07-25 10:08:54 -070075#include "ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070076
77namespace SkSL {
78
ethannicholascad64162016-10-27 10:54:02 -070079#define MAX_PARSE_DEPTH 50
80
81class AutoDepth {
82public:
83 AutoDepth(Parser* p)
84 : fParser(p) {
85 fParser->fDepth++;
86 }
87
88 ~AutoDepth() {
89 fParser->fDepth--;
90 }
91
92 bool checkValid() {
93 if (fParser->fDepth > MAX_PARSE_DEPTH) {
Ethan Nicholas9e1138d2016-11-21 10:39:35 -050094 fParser->error(fParser->peek().fPosition, SkString("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070095 return false;
96 }
97 return true;
98 }
99
100private:
101 Parser* fParser;
102};
103
Ethan Nicholas11d53972016-11-28 11:23:23 -0500104Parser::Parser(SkString text, SymbolTable& types, ErrorReporter& errors)
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500105: fPushback(Position(-1, -1), Token::INVALID_TOKEN, SkString())
ethannicholasb3058bd2016-07-01 08:22:01 -0700106, fTypes(types)
107, fErrors(errors) {
108 sksllex_init(&fScanner);
109 fBuffer = sksl_scan_string(text.c_str(), fScanner);
110 skslset_lineno(1, fScanner);
111
112 if (false) {
113 // avoid unused warning
114 yyunput(0, nullptr, fScanner);
115 }
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500116
117 fLayoutKeys[SkString("location")] = kLocation_LayoutKey;
118 fLayoutKeys[SkString("offset")] = kOffset_LayoutKey;
119 fLayoutKeys[SkString("binding")] = kBinding_LayoutKey;
120 fLayoutKeys[SkString("index")] = kIndex_LayoutKey;
121 fLayoutKeys[SkString("set")] = kSet_LayoutKey;
122 fLayoutKeys[SkString("builtin")] = kBuiltin_LayoutKey;
123 fLayoutKeys[SkString("input_attachment_index")] = kInputAttachmentIndex_LayoutKey;
124 fLayoutKeys[SkString("origin_upper_left")] = kOriginUpperLeft_LayoutKey;
125 fLayoutKeys[SkString("override_coverage")] = kOverrideCoverage_LayoutKey;
126 fLayoutKeys[SkString("blend_support_all_equations")] = kBlendSupportAllEquations_LayoutKey;
127 fLayoutKeys[SkString("push_constant")] = kPushConstant_LayoutKey;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500128 fLayoutKeys[SkString("points")] = kPoints_LayoutKey;
129 fLayoutKeys[SkString("lines")] = kLines_LayoutKey;
130 fLayoutKeys[SkString("line_strip")] = kLineStrip_LayoutKey;
131 fLayoutKeys[SkString("lines_adjacency")] = kLinesAdjacency_LayoutKey;
132 fLayoutKeys[SkString("triangles")] = kTriangles_LayoutKey;
133 fLayoutKeys[SkString("triangle_strip")] = kTriangleStrip_LayoutKey;
134 fLayoutKeys[SkString("triangles_adjacency")] = kTrianglesAdjacency_LayoutKey;
135 fLayoutKeys[SkString("max_vertices")] = kMaxVertices_LayoutKey;
136 fLayoutKeys[SkString("invocations")] = kInvocations_LayoutKey;
ethannicholasb3058bd2016-07-01 08:22:01 -0700137}
138
139Parser::~Parser() {
140 sksl_delete_buffer(fBuffer, fScanner);
ethannicholas69a7e7c2016-07-01 19:09:27 -0700141 sksllex_destroy(fScanner);
ethannicholasb3058bd2016-07-01 08:22:01 -0700142}
143
144/* (precision | directive | declaration)* END_OF_FILE */
145std::vector<std::unique_ptr<ASTDeclaration>> Parser::file() {
146 std::vector<std::unique_ptr<ASTDeclaration>> result;
147 for (;;) {
148 switch (this->peek().fKind) {
149 case Token::END_OF_FILE:
150 return result;
ethannicholas5961bc92016-10-12 06:39:56 -0700151 case Token::PRECISION: {
152 std::unique_ptr<ASTDeclaration> precision = this->precision();
153 if (precision) {
154 result.push_back(std::move(precision));
155 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700156 break;
ethannicholas5961bc92016-10-12 06:39:56 -0700157 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700158 case Token::DIRECTIVE: {
159 std::unique_ptr<ASTDeclaration> decl = this->directive();
160 if (decl) {
161 result.push_back(std::move(decl));
162 }
163 break;
164 }
165 default: {
166 std::unique_ptr<ASTDeclaration> decl = this->declaration();
167 if (!decl) {
168 continue;
169 }
170 result.push_back(std::move(decl));
171 }
172 }
173 }
174}
175
176Token Parser::nextToken() {
177 if (fPushback.fKind != Token::INVALID_TOKEN) {
178 Token result = fPushback;
179 fPushback.fKind = Token::INVALID_TOKEN;
180 fPushback.fText = "";
181 return result;
182 }
183 int token = sksllex(fScanner);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500184 SkString text;
ethannicholas471e8942016-10-28 09:02:46 -0700185 switch ((Token::Kind) token) {
186 case Token::IDENTIFIER: // fall through
187 case Token::INT_LITERAL: // fall through
188 case Token::FLOAT_LITERAL: // fall through
189 case Token::DIRECTIVE:
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500190 text = SkString(skslget_text(fScanner));
ethannicholas471e8942016-10-28 09:02:46 -0700191 break;
192 default:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500193#ifdef SK_DEBUG
194 text = SkString(skslget_text(fScanner));
195#endif
ethannicholas471e8942016-10-28 09:02:46 -0700196 break;
197 }
198 return Token(Position(skslget_lineno(fScanner), -1), (Token::Kind) token, text);
ethannicholasb3058bd2016-07-01 08:22:01 -0700199}
200
201void Parser::pushback(Token t) {
202 ASSERT(fPushback.fKind == Token::INVALID_TOKEN);
203 fPushback = t;
204}
205
206Token Parser::peek() {
207 fPushback = this->nextToken();
208 return fPushback;
209}
210
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500211
212bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
213 return this->expect(kind, SkString(expected), result);
214}
215
216bool Parser::expect(Token::Kind kind, SkString expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700217 Token next = this->nextToken();
218 if (next.fKind == kind) {
219 if (result) {
220 *result = next;
221 }
222 return true;
223 } else {
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500224 if (next.fText.size()) {
225 this->error(next.fPosition, "expected " + expected + ", but found '" + next.fText +
226 "'");
227 } else {
228 this->error(next.fPosition, "parse error, recompile in debug mode for details");
229 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700230 return false;
231 }
232}
233
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500234void Parser::error(Position p, const char* msg) {
235 this->error(p, SkString(msg));
236}
237
238void Parser::error(Position p, SkString msg) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700239 fErrors.error(p, msg);
240}
241
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500242bool Parser::isType(SkString name) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700243 return nullptr != fTypes[name];
244}
245
246/* PRECISION (LOWP | MEDIUMP | HIGHP) type SEMICOLON */
ethannicholas5961bc92016-10-12 06:39:56 -0700247std::unique_ptr<ASTDeclaration> Parser::precision() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700248 if (!this->expect(Token::PRECISION, "'precision'")) {
ethannicholas5961bc92016-10-12 06:39:56 -0700249 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700250 }
ethannicholas5961bc92016-10-12 06:39:56 -0700251 Modifiers::Flag result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700252 Token p = this->nextToken();
253 switch (p.fKind) {
ethannicholas5961bc92016-10-12 06:39:56 -0700254 case Token::LOWP:
255 result = Modifiers::kLowp_Flag;
256 break;
257 case Token::MEDIUMP:
258 result = Modifiers::kMediump_Flag;
259 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700260 case Token::HIGHP:
ethannicholas5961bc92016-10-12 06:39:56 -0700261 result = Modifiers::kHighp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700262 break;
263 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500264 this->error(p.fPosition, "expected 'lowp', 'mediump', or 'highp', but found '" +
ethannicholasb3058bd2016-07-01 08:22:01 -0700265 p.fText + "'");
ethannicholas5961bc92016-10-12 06:39:56 -0700266 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700267 }
ethannicholas5961bc92016-10-12 06:39:56 -0700268 // FIXME handle the type
ethannicholasb3058bd2016-07-01 08:22:01 -0700269 if (!this->type()) {
ethannicholas5961bc92016-10-12 06:39:56 -0700270 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700271 }
272 this->expect(Token::SEMICOLON, "';'");
ethannicholas5961bc92016-10-12 06:39:56 -0700273 return std::unique_ptr<ASTDeclaration>(new ASTPrecision(p.fPosition, result));
ethannicholasb3058bd2016-07-01 08:22:01 -0700274}
275
Ethan Nicholas11d53972016-11-28 11:23:23 -0500276/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700277 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
ethannicholasb3058bd2016-07-01 08:22:01 -0700278std::unique_ptr<ASTDeclaration> Parser::directive() {
279 Token start;
280 if (!this->expect(Token::DIRECTIVE, "a directive", &start)) {
281 return nullptr;
282 }
283 if (start.fText == "#version") {
284 this->expect(Token::INT_LITERAL, "a version number");
ethannicholas5961bc92016-10-12 06:39:56 -0700285 Token next = this->peek();
286 if (next.fText == "es" || next.fText == "compatibility") {
287 this->nextToken();
288 }
289 // version is ignored for now; it will eventually become an error when we stop pretending
290 // to be GLSL
ethannicholasb3058bd2016-07-01 08:22:01 -0700291 return nullptr;
292 } else if (start.fText == "#extension") {
293 Token name;
294 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
295 return nullptr;
296 }
297 if (!this->expect(Token::COLON, "':'")) {
298 return nullptr;
299 }
300 // FIXME: need to start paying attention to this token
301 if (!this->expect(Token::IDENTIFIER, "an identifier")) {
302 return nullptr;
303 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500304 return std::unique_ptr<ASTDeclaration>(new ASTExtension(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700305 std::move(name.fText)));
306 } else {
307 this->error(start.fPosition, "unsupported directive '" + start.fText + "'");
308 return nullptr;
309 }
310}
311
Ethan Nicholas11d53972016-11-28 11:23:23 -0500312/* modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700313 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
314std::unique_ptr<ASTDeclaration> Parser::declaration() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500315 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700316 Token lookahead = this->peek();
317 if (lookahead.fKind == Token::IDENTIFIER && !this->isType(lookahead.fText)) {
318 // we have an identifier that's not a type, could be the start of an interface block
319 return this->interfaceBlock(modifiers);
320 }
321 if (lookahead.fKind == Token::STRUCT) {
322 return this->structVarDeclaration(modifiers);
323 }
ethannicholas5961bc92016-10-12 06:39:56 -0700324 if (lookahead.fKind == Token::SEMICOLON) {
325 this->nextToken();
326 return std::unique_ptr<ASTDeclaration>(new ASTModifiersDeclaration(modifiers));
327 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700328 std::unique_ptr<ASTType> type(this->type());
329 if (!type) {
330 return nullptr;
331 }
332 if (type->fKind == ASTType::kStruct_Kind && peek().fKind == Token::SEMICOLON) {
333 this->nextToken();
334 return nullptr;
335 }
336 Token name;
337 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
338 return nullptr;
339 }
340 if (!modifiers.fFlags && this->peek().fKind == Token::LPAREN) {
341 this->nextToken();
342 std::vector<std::unique_ptr<ASTParameter>> parameters;
343 while (this->peek().fKind != Token::RPAREN) {
344 if (parameters.size() > 0) {
345 if (!this->expect(Token::COMMA, "','")) {
346 return nullptr;
347 }
348 }
349 std::unique_ptr<ASTParameter> parameter = this->parameter();
350 if (!parameter) {
351 return nullptr;
352 }
353 parameters.push_back(std::move(parameter));
354 }
355 this->nextToken();
356 std::unique_ptr<ASTBlock> body;
357 if (this->peek().fKind == Token::SEMICOLON) {
358 this->nextToken();
359 } else {
360 body = this->block();
361 if (!body) {
362 return nullptr;
363 }
364 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500365 return std::unique_ptr<ASTDeclaration>(new ASTFunction(name.fPosition, std::move(type),
366 std::move(name.fText),
367 std::move(parameters),
ethannicholasb3058bd2016-07-01 08:22:01 -0700368 std::move(body)));
369 } else {
370 return this->varDeclarationEnd(modifiers, std::move(type), name.fText);
371 }
372}
373
374/* modifiers type IDENTIFIER varDeclarationEnd */
ethannicholas14fe8cc2016-09-07 13:37:16 -0700375std::unique_ptr<ASTVarDeclarations> Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500376 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700377 std::unique_ptr<ASTType> type(this->type());
378 if (!type) {
379 return nullptr;
380 }
381 Token name;
382 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
383 return nullptr;
384 }
385 return this->varDeclarationEnd(modifiers, std::move(type), std::move(name.fText));
386}
387
388/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
389std::unique_ptr<ASTType> Parser::structDeclaration() {
390 if (!this->expect(Token::STRUCT, "'struct'")) {
391 return nullptr;
392 }
393 Token name;
394 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
395 return nullptr;
396 }
397 if (!this->expect(Token::LBRACE, "'{'")) {
398 return nullptr;
399 }
400 std::vector<Type::Field> fields;
401 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700402 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700403 if (!decl) {
404 return nullptr;
405 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700406 for (const auto& var : decl->fVars) {
ethannicholasd598f792016-07-25 10:08:54 -0700407 auto type = (const Type*) fTypes[decl->fType->fName];
ethannicholas14fe8cc2016-09-07 13:37:16 -0700408 for (int i = (int) var.fSizes.size() - 1; i >= 0; i--) {
ethannicholasdd4645b2016-10-14 12:14:46 -0700409 if (!var.fSizes[i] || var.fSizes[i]->fKind != ASTExpression::kInt_Kind) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700410 this->error(decl->fPosition, "array size in struct field must be a constant");
ethannicholasdd4645b2016-10-14 12:14:46 -0700411 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700412 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700413 uint64_t columns = ((ASTIntLiteral&) *var.fSizes[i]).fValue;
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500414 SkString name = type->name() + "[" + to_string(columns) + "]";
ethannicholasd598f792016-07-25 10:08:54 -0700415 type = new Type(name, Type::kArray_Kind, *type, (int) columns);
416 fTypes.takeOwnership((Type*) type);
ethannicholasb3058bd2016-07-01 08:22:01 -0700417 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700418 fields.push_back(Type::Field(decl->fModifiers, var.fName, type));
419 if (var.fValue) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700420 this->error(decl->fPosition, "initializers are not permitted on struct fields");
421 }
422 }
423 }
424 if (!this->expect(Token::RBRACE, "'}'")) {
425 return nullptr;
426 }
Ethan Nicholas19671772016-11-28 16:30:17 -0500427 fTypes.add(name.fText, std::unique_ptr<Type>(new Type(name.fPosition, name.fText, fields)));
Ethan Nicholas11d53972016-11-28 11:23:23 -0500428 return std::unique_ptr<ASTType>(new ASTType(name.fPosition, name.fText,
Ethan Nicholas50afc172017-02-16 14:49:57 -0500429 ASTType::kStruct_Kind, std::vector<int>()));
ethannicholasb3058bd2016-07-01 08:22:01 -0700430}
431
432/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500433std::unique_ptr<ASTVarDeclarations> Parser::structVarDeclaration(Modifiers modifiers) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700434 std::unique_ptr<ASTType> type = this->structDeclaration();
435 if (!type) {
436 return nullptr;
437 }
438 if (peek().fKind == Token::IDENTIFIER) {
439 Token name = this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500440 std::unique_ptr<ASTVarDeclarations> result = this->varDeclarationEnd(modifiers,
441 std::move(type),
ethannicholas14fe8cc2016-09-07 13:37:16 -0700442 std::move(name.fText));
ethannicholasb3058bd2016-07-01 08:22:01 -0700443 if (result) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700444 for (const auto& var : result->fVars) {
445 if (var.fValue) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500446 this->error(var.fValue->fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700447 "struct variables cannot be initialized");
448 }
449 }
450 }
451 return result;
452 }
453 this->expect(Token::SEMICOLON, "';'");
454 return nullptr;
455}
456
Ethan Nicholas11d53972016-11-28 11:23:23 -0500457/* (LBRACKET expression? RBRACKET)* (EQ expression)? (COMMA IDENTIFER
ethannicholasb3058bd2016-07-01 08:22:01 -0700458 (LBRACKET expression? RBRACKET)* (EQ expression)?)* SEMICOLON */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500459std::unique_ptr<ASTVarDeclarations> Parser::varDeclarationEnd(Modifiers mods,
ethannicholas14fe8cc2016-09-07 13:37:16 -0700460 std::unique_ptr<ASTType> type,
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500461 SkString name) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700462 std::vector<ASTVarDeclaration> vars;
ethannicholasb3058bd2016-07-01 08:22:01 -0700463 std::vector<std::unique_ptr<ASTExpression>> currentVarSizes;
464 while (this->peek().fKind == Token::LBRACKET) {
465 this->nextToken();
466 if (this->peek().fKind == Token::RBRACKET) {
467 this->nextToken();
468 currentVarSizes.push_back(nullptr);
469 } else {
470 std::unique_ptr<ASTExpression> size(this->expression());
471 if (!size) {
472 return nullptr;
473 }
474 currentVarSizes.push_back(std::move(size));
475 if (!this->expect(Token::RBRACKET, "']'")) {
476 return nullptr;
477 }
478 }
479 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700480 std::unique_ptr<ASTExpression> value;
ethannicholasb3058bd2016-07-01 08:22:01 -0700481 if (this->peek().fKind == Token::EQ) {
482 this->nextToken();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700483 value = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700484 if (!value) {
485 return nullptr;
486 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700487 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700488 vars.emplace_back(std::move(name), std::move(currentVarSizes), std::move(value));
ethannicholasb3058bd2016-07-01 08:22:01 -0700489 while (this->peek().fKind == Token::COMMA) {
490 this->nextToken();
491 Token name;
492 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
493 return nullptr;
494 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700495 currentVarSizes.clear();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700496 value.reset();
ethannicholasb3058bd2016-07-01 08:22:01 -0700497 while (this->peek().fKind == Token::LBRACKET) {
498 this->nextToken();
499 if (this->peek().fKind == Token::RBRACKET) {
500 this->nextToken();
501 currentVarSizes.push_back(nullptr);
502 } else {
503 std::unique_ptr<ASTExpression> size(this->expression());
504 if (!size) {
505 return nullptr;
506 }
507 currentVarSizes.push_back(std::move(size));
508 if (!this->expect(Token::RBRACKET, "']'")) {
509 return nullptr;
510 }
511 }
512 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700513 if (this->peek().fKind == Token::EQ) {
514 this->nextToken();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700515 value = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700516 if (!value) {
517 return nullptr;
518 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700519 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700520 vars.emplace_back(std::move(name.fText), std::move(currentVarSizes), std::move(value));
ethannicholasb3058bd2016-07-01 08:22:01 -0700521 }
522 if (!this->expect(Token::SEMICOLON, "';'")) {
523 return nullptr;
524 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700525 return std::unique_ptr<ASTVarDeclarations>(new ASTVarDeclarations(std::move(mods),
526 std::move(type),
527 std::move(vars)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700528}
529
530/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
531std::unique_ptr<ASTParameter> Parser::parameter() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500532 Modifiers modifiers = this->modifiersWithDefaults(Modifiers::kIn_Flag);
ethannicholasb3058bd2016-07-01 08:22:01 -0700533 std::unique_ptr<ASTType> type = this->type();
534 if (!type) {
535 return nullptr;
536 }
537 Token name;
538 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
539 return nullptr;
540 }
541 std::vector<int> sizes;
542 while (this->peek().fKind == Token::LBRACKET) {
543 this->nextToken();
544 Token sizeToken;
545 if (!this->expect(Token::INT_LITERAL, "a positive integer", &sizeToken)) {
546 return nullptr;
547 }
548 sizes.push_back(SkSL::stoi(sizeToken.fText));
549 if (!this->expect(Token::RBRACKET, "']'")) {
550 return nullptr;
551 }
552 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500553 return std::unique_ptr<ASTParameter>(new ASTParameter(name.fPosition, modifiers,
554 std::move(type), name.fText,
ethannicholasb3058bd2016-07-01 08:22:01 -0700555 std::move(sizes)));
556}
557
558/** (EQ INT_LITERAL)? */
559int Parser::layoutInt() {
560 if (!this->expect(Token::EQ, "'='")) {
561 return -1;
562 }
563 Token resultToken;
564 if (this->expect(Token::INT_LITERAL, "a non-negative integer", &resultToken)) {
565 return SkSL::stoi(resultToken.fText);
566 }
567 return -1;
568}
569
ethannicholas8ac838d2016-11-22 08:39:36 -0800570/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500571Layout Parser::layout() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700572 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500573 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700574 int binding = -1;
575 int index = -1;
576 int set = -1;
577 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500578 int inputAttachmentIndex = -1;
ethannicholasf789b382016-08-03 12:43:36 -0700579 bool originUpperLeft = false;
ethannicholas5961bc92016-10-12 06:39:56 -0700580 bool overrideCoverage = false;
581 bool blendSupportAllEquations = false;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500582 Layout::Format format = Layout::Format::kUnspecified;
ethannicholas8ac838d2016-11-22 08:39:36 -0800583 bool pushConstant = false;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500584 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
585 int maxVertices = -1;
586 int invocations = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700587 if (this->peek().fKind == Token::LAYOUT) {
588 this->nextToken();
589 if (!this->expect(Token::LPAREN, "'('")) {
Ethan Nicholas19671772016-11-28 16:30:17 -0500590 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
Ethan Nicholas11d53972016-11-28 11:23:23 -0500591 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas52cad152017-02-16 16:37:32 -0500592 pushConstant, primitive, maxVertices, invocations);
ethannicholasb3058bd2016-07-01 08:22:01 -0700593 }
594 for (;;) {
595 Token t = this->nextToken();
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500596 auto found = fLayoutKeys.find(t.fText);
597 if (found != fLayoutKeys.end()) {
598 switch (found->second) {
599 case kLocation_LayoutKey:
600 location = this->layoutInt();
601 break;
602 case kOffset_LayoutKey:
603 offset = this->layoutInt();
604 break;
605 case kBinding_LayoutKey:
606 binding = this->layoutInt();
607 break;
608 case kIndex_LayoutKey:
609 index = this->layoutInt();
610 break;
611 case kSet_LayoutKey:
612 set = this->layoutInt();
613 break;
614 case kBuiltin_LayoutKey:
615 builtin = this->layoutInt();
616 break;
617 case kInputAttachmentIndex_LayoutKey:
618 inputAttachmentIndex = this->layoutInt();
619 break;
620 case kOriginUpperLeft_LayoutKey:
621 originUpperLeft = true;
622 break;
623 case kOverrideCoverage_LayoutKey:
624 overrideCoverage = true;
625 break;
626 case kBlendSupportAllEquations_LayoutKey:
627 blendSupportAllEquations = true;
628 break;
629 case kPushConstant_LayoutKey:
630 pushConstant = true;
631 break;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500632 case kPoints_LayoutKey:
633 primitive = Layout::kPoints_Primitive;
634 break;
635 case kLines_LayoutKey:
636 primitive = Layout::kLines_Primitive;
637 break;
638 case kLineStrip_LayoutKey:
639 primitive = Layout::kLineStrip_Primitive;
640 break;
641 case kLinesAdjacency_LayoutKey:
642 primitive = Layout::kLinesAdjacency_Primitive;
643 break;
644 case kTriangles_LayoutKey:
645 primitive = Layout::kTriangles_Primitive;
646 break;
647 case kTriangleStrip_LayoutKey:
648 primitive = Layout::kTriangleStrip_Primitive;
649 break;
650 case kTrianglesAdjacency_LayoutKey:
651 primitive = Layout::kTrianglesAdjacency_Primitive;
652 break;
653 case kMaxVertices_LayoutKey:
654 maxVertices = this->layoutInt();
655 break;
656 case kInvocations_LayoutKey:
657 invocations = this->layoutInt();
658 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500659 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500660 } else if (Layout::ReadFormat(t.fText, &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500661 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700662 } else {
Greg Daniel64773e62016-11-22 09:44:03 -0500663 this->error(t.fPosition, ("'" + t.fText +
ethannicholasb3058bd2016-07-01 08:22:01 -0700664 "' is not a valid layout qualifier").c_str());
665 }
666 if (this->peek().fKind == Token::RPAREN) {
667 this->nextToken();
668 break;
669 }
670 if (!this->expect(Token::COMMA, "','")) {
671 break;
672 }
673 }
674 }
Ethan Nicholas19671772016-11-28 16:30:17 -0500675 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
676 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas52cad152017-02-16 16:37:32 -0500677 pushConstant, primitive, maxVertices, invocations);
ethannicholasb3058bd2016-07-01 08:22:01 -0700678}
679
Brian Salomonf9f45122016-11-29 11:59:17 -0500680/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
681 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500682Modifiers Parser::modifiers() {
683 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700684 int flags = 0;
685 for (;;) {
686 // TODO: handle duplicate / incompatible flags
687 switch (peek().fKind) {
688 case Token::UNIFORM:
689 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500690 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700691 break;
692 case Token::CONST:
693 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500694 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700695 break;
696 case Token::IN:
697 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500698 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700699 break;
700 case Token::OUT:
701 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500702 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700703 break;
704 case Token::INOUT:
705 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500706 flags |= Modifiers::kIn_Flag;
707 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700708 break;
709 case Token::LOWP:
710 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500711 flags |= Modifiers::kLowp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700712 break;
713 case Token::MEDIUMP:
714 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500715 flags |= Modifiers::kMediump_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700716 break;
717 case Token::HIGHP:
718 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500719 flags |= Modifiers::kHighp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700720 break;
ethannicholasf789b382016-08-03 12:43:36 -0700721 case Token::FLAT:
722 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500723 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700724 break;
725 case Token::NOPERSPECTIVE:
726 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500727 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700728 break;
Brian Salomonf9f45122016-11-29 11:59:17 -0500729 case Token::READONLY:
730 this->nextToken();
731 flags |= Modifiers::kReadOnly_Flag;
732 break;
733 case Token::WRITEONLY:
734 this->nextToken();
735 flags |= Modifiers::kWriteOnly_Flag;
736 break;
737 case Token::COHERENT:
738 this->nextToken();
739 flags |= Modifiers::kCoherent_Flag;
740 break;
741 case Token::VOLATILE:
742 this->nextToken();
743 flags |= Modifiers::kVolatile_Flag;
744 break;
745 case Token::RESTRICT:
746 this->nextToken();
747 flags |= Modifiers::kRestrict_Flag;
748 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700749 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500750 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700751 }
752 }
753}
754
Ethan Nicholas11d53972016-11-28 11:23:23 -0500755Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
756 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700757 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500758 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700759 }
760 return result;
761}
762
763/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
764std::unique_ptr<ASTStatement> Parser::statement() {
765 Token start = this->peek();
766 switch (start.fKind) {
767 case Token::IF:
768 return this->ifStatement();
769 case Token::FOR:
770 return this->forStatement();
771 case Token::DO:
772 return this->doStatement();
773 case Token::WHILE:
774 return this->whileStatement();
Ethan Nicholasaf197692017-02-27 13:26:45 -0500775 case Token::SWITCH:
776 return this->switchStatement();
ethannicholasb3058bd2016-07-01 08:22:01 -0700777 case Token::RETURN:
778 return this->returnStatement();
779 case Token::BREAK:
780 return this->breakStatement();
781 case Token::CONTINUE:
782 return this->continueStatement();
783 case Token::DISCARD:
784 return this->discardStatement();
785 case Token::LBRACE:
786 return this->block();
787 case Token::SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500788 this->nextToken();
789 return std::unique_ptr<ASTStatement>(new ASTBlock(start.fPosition,
ethannicholas0730be72016-09-01 07:59:02 -0700790 std::vector<std::unique_ptr<ASTStatement>>()));
ethannicholasb3058bd2016-07-01 08:22:01 -0700791 case Token::CONST: // fall through
792 case Token::HIGHP: // fall through
793 case Token::MEDIUMP: // fall through
794 case Token::LOWP: {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700795 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700796 if (!decl) {
797 return nullptr;
798 }
799 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(std::move(decl)));
800 }
801 case Token::IDENTIFIER:
802 if (this->isType(start.fText)) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700803 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700804 if (!decl) {
805 return nullptr;
806 }
807 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
808 std::move(decl)));
809 }
810 // fall through
811 default:
812 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500813 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700814}
815
Ethan Nicholas50afc172017-02-16 14:49:57 -0500816/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* */
ethannicholasb3058bd2016-07-01 08:22:01 -0700817std::unique_ptr<ASTType> Parser::type() {
818 Token type;
819 if (!this->expect(Token::IDENTIFIER, "a type", &type)) {
820 return nullptr;
821 }
822 if (!this->isType(type.fText)) {
823 this->error(type.fPosition, ("no type named '" + type.fText + "'").c_str());
824 return nullptr;
825 }
Ethan Nicholas50afc172017-02-16 14:49:57 -0500826 std::vector<int> sizes;
827 while (this->peek().fKind == Token::LBRACKET) {
828 this->expect(Token::LBRACKET, "'['");
829 if (this->peek().fKind != Token::RBRACKET) {
830 int64_t i;
831 if (this->intLiteral(&i)) {
832 sizes.push_back(i);
833 } else {
834 return nullptr;
835 }
836 } else {
837 sizes.push_back(-1);
838 }
839 this->expect(Token::RBRACKET, "']'");
840 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500841 return std::unique_ptr<ASTType>(new ASTType(type.fPosition, std::move(type.fText),
Ethan Nicholas50afc172017-02-16 14:49:57 -0500842 ASTType::kIdentifier_Kind, sizes));
ethannicholasb3058bd2016-07-01 08:22:01 -0700843}
844
Ethan Nicholas50afc172017-02-16 14:49:57 -0500845/* IDENTIFIER LBRACE varDeclaration* RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500846std::unique_ptr<ASTDeclaration> Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700847 Token name;
848 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
849 return nullptr;
850 }
851 if (peek().fKind != Token::LBRACE) {
852 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -0500853 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -0700854 // it's better to report it as an unknown type
855 this->error(name.fPosition, "no type named '" + name.fText + "'");
856 return nullptr;
857 }
858 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500859 std::vector<std::unique_ptr<ASTVarDeclarations>> decls;
ethannicholasb3058bd2016-07-01 08:22:01 -0700860 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700861 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700862 if (!decl) {
863 return nullptr;
864 }
865 decls.push_back(std::move(decl));
866 }
867 this->nextToken();
Ethan Nicholas50afc172017-02-16 14:49:57 -0500868 SkString instanceName;
869 std::vector<std::unique_ptr<ASTExpression>> sizes;
ethannicholasb3058bd2016-07-01 08:22:01 -0700870 if (this->peek().fKind == Token::IDENTIFIER) {
Ethan Nicholas50afc172017-02-16 14:49:57 -0500871 instanceName = this->nextToken().fText;
872 while (this->peek().fKind == Token::LBRACKET) {
873 this->expect(Token::LBRACKET, "'['");
874 if (this->peek().fKind != Token::RBRACKET) {
875 std::unique_ptr<ASTExpression> size = this->expression();
876 if (!size) {
877 return nullptr;
878 }
879 sizes.push_back(std::move(size));
880 } else {
881 sizes.push_back(nullptr);
882 }
883 this->expect(Token::RBRACKET, "']'");
884 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700885 }
886 this->expect(Token::SEMICOLON, "';'");
Ethan Nicholas11d53972016-11-28 11:23:23 -0500887 return std::unique_ptr<ASTDeclaration>(new ASTInterfaceBlock(name.fPosition, mods,
Ethan Nicholas50afc172017-02-16 14:49:57 -0500888 name.fText, std::move(decls),
889 std::move(instanceName),
890 std::move(sizes)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700891}
892
893/* IF LPAREN expression RPAREN statement (ELSE statement)? */
894std::unique_ptr<ASTIfStatement> Parser::ifStatement() {
895 Token start;
896 if (!this->expect(Token::IF, "'if'", &start)) {
897 return nullptr;
898 }
899 if (!this->expect(Token::LPAREN, "'('")) {
900 return nullptr;
901 }
902 std::unique_ptr<ASTExpression> test(this->expression());
903 if (!test) {
904 return nullptr;
905 }
906 if (!this->expect(Token::RPAREN, "')'")) {
907 return nullptr;
908 }
909 std::unique_ptr<ASTStatement> ifTrue(this->statement());
910 if (!ifTrue) {
911 return nullptr;
912 }
913 std::unique_ptr<ASTStatement> ifFalse;
914 if (this->peek().fKind == Token::ELSE) {
915 this->nextToken();
916 ifFalse = this->statement();
917 if (!ifFalse) {
918 return nullptr;
919 }
920 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500921 return std::unique_ptr<ASTIfStatement>(new ASTIfStatement(start.fPosition, std::move(test),
922 std::move(ifTrue),
ethannicholasb3058bd2016-07-01 08:22:01 -0700923 std::move(ifFalse)));
924}
925
926/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
927std::unique_ptr<ASTDoStatement> Parser::doStatement() {
928 Token start;
929 if (!this->expect(Token::DO, "'do'", &start)) {
930 return nullptr;
931 }
932 std::unique_ptr<ASTStatement> statement(this->statement());
933 if (!statement) {
934 return nullptr;
935 }
936 if (!this->expect(Token::WHILE, "'while'")) {
937 return nullptr;
938 }
939 if (!this->expect(Token::LPAREN, "'('")) {
940 return nullptr;
941 }
942 std::unique_ptr<ASTExpression> test(this->expression());
943 if (!test) {
944 return nullptr;
945 }
946 if (!this->expect(Token::RPAREN, "')'")) {
947 return nullptr;
948 }
949 if (!this->expect(Token::SEMICOLON, "';'")) {
950 return nullptr;
951 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500952 return std::unique_ptr<ASTDoStatement>(new ASTDoStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700953 std::move(statement),
954 std::move(test)));
955}
956
957/* WHILE LPAREN expression RPAREN STATEMENT */
958std::unique_ptr<ASTWhileStatement> Parser::whileStatement() {
959 Token start;
960 if (!this->expect(Token::WHILE, "'while'", &start)) {
961 return nullptr;
962 }
963 if (!this->expect(Token::LPAREN, "'('")) {
964 return nullptr;
965 }
966 std::unique_ptr<ASTExpression> test(this->expression());
967 if (!test) {
968 return nullptr;
969 }
970 if (!this->expect(Token::RPAREN, "')'")) {
971 return nullptr;
972 }
973 std::unique_ptr<ASTStatement> statement(this->statement());
974 if (!statement) {
975 return nullptr;
976 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500977 return std::unique_ptr<ASTWhileStatement>(new ASTWhileStatement(start.fPosition,
978 std::move(test),
ethannicholasb3058bd2016-07-01 08:22:01 -0700979 std::move(statement)));
980}
981
Ethan Nicholasaf197692017-02-27 13:26:45 -0500982/* CASE expression COLON statement* */
983std::unique_ptr<ASTSwitchCase> Parser::switchCase() {
984 Token start;
985 if (!this->expect(Token::CASE, "'case'", &start)) {
986 return nullptr;
987 }
988 std::unique_ptr<ASTExpression> value = this->expression();
989 if (!value) {
990 return nullptr;
991 }
992 if (!this->expect(Token::COLON, "':'")) {
993 return nullptr;
994 }
995 std::vector<std::unique_ptr<ASTStatement>> statements;
996 while (this->peek().fKind != Token::RBRACE && this->peek().fKind != Token::CASE &&
997 this->peek().fKind != Token::DEFAULT) {
998 std::unique_ptr<ASTStatement> s = this->statement();
999 if (!s) {
1000 return nullptr;
1001 }
1002 statements.push_back(std::move(s));
1003 }
1004 return std::unique_ptr<ASTSwitchCase>(new ASTSwitchCase(start.fPosition, std::move(value),
1005 std::move(statements)));
1006}
1007
1008/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
1009std::unique_ptr<ASTStatement> Parser::switchStatement() {
1010 Token start;
1011 if (!this->expect(Token::SWITCH, "'switch'", &start)) {
1012 return nullptr;
1013 }
1014 if (!this->expect(Token::LPAREN, "'('")) {
1015 return nullptr;
1016 }
1017 std::unique_ptr<ASTExpression> value(this->expression());
1018 if (!value) {
1019 return nullptr;
1020 }
1021 if (!this->expect(Token::RPAREN, "')'")) {
1022 return nullptr;
1023 }
1024 if (!this->expect(Token::LBRACE, "'{'")) {
1025 return nullptr;
1026 }
1027 std::vector<std::unique_ptr<ASTSwitchCase>> cases;
1028 while (this->peek().fKind == Token::CASE) {
1029 std::unique_ptr<ASTSwitchCase> c = this->switchCase();
1030 if (!c) {
1031 return nullptr;
1032 }
1033 cases.push_back(std::move(c));
1034 }
1035 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1036 // parts of the compiler may rely upon this assumption.
1037 if (this->peek().fKind == Token::DEFAULT) {
1038 Token defaultStart;
1039 SkAssertResult(this->expect(Token::DEFAULT, "'default'", &defaultStart));
1040 if (!this->expect(Token::COLON, "':'")) {
1041 return nullptr;
1042 }
1043 std::vector<std::unique_ptr<ASTStatement>> statements;
1044 while (this->peek().fKind != Token::RBRACE) {
1045 std::unique_ptr<ASTStatement> s = this->statement();
1046 if (!s) {
1047 return nullptr;
1048 }
1049 statements.push_back(std::move(s));
1050 }
1051 cases.emplace_back(new ASTSwitchCase(defaultStart.fPosition, nullptr,
1052 std::move(statements)));
1053 }
1054 if (!this->expect(Token::RBRACE, "'}'")) {
1055 return nullptr;
1056 }
1057 return std::unique_ptr<ASTStatement>(new ASTSwitchStatement(start.fPosition,
1058 std::move(value),
1059 std::move(cases)));
1060}
1061
Ethan Nicholas11d53972016-11-28 11:23:23 -05001062/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001063 STATEMENT */
1064std::unique_ptr<ASTForStatement> Parser::forStatement() {
1065 Token start;
1066 if (!this->expect(Token::FOR, "'for'", &start)) {
1067 return nullptr;
1068 }
1069 if (!this->expect(Token::LPAREN, "'('")) {
1070 return nullptr;
1071 }
1072 std::unique_ptr<ASTStatement> initializer;
1073 Token nextToken = this->peek();
1074 switch (nextToken.fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001075 case Token::SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001076 this->nextToken();
ethannicholasb3058bd2016-07-01 08:22:01 -07001077 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001078 case Token::CONST: {
1079 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
1080 if (!vd) {
1081 return nullptr;
1082 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001083 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001084 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001085 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001086 }
1087 case Token::IDENTIFIER: {
ethannicholasb3058bd2016-07-01 08:22:01 -07001088 if (this->isType(nextToken.fText)) {
ethannicholasa54401d2016-10-14 08:37:32 -07001089 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
1090 if (!vd) {
1091 return nullptr;
1092 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001093 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001094 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001095 break;
1096 }
ethannicholasa54401d2016-10-14 08:37:32 -07001097 } // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001098 default:
1099 initializer = this->expressionStatement();
1100 }
1101 std::unique_ptr<ASTExpression> test;
1102 if (this->peek().fKind != Token::SEMICOLON) {
1103 test = this->expression();
1104 if (!test) {
1105 return nullptr;
1106 }
1107 }
1108 if (!this->expect(Token::SEMICOLON, "';'")) {
1109 return nullptr;
1110 }
1111 std::unique_ptr<ASTExpression> next;
ethannicholas22f939e2016-10-13 13:25:34 -07001112 if (this->peek().fKind != Token::RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001113 next = this->expression();
1114 if (!next) {
1115 return nullptr;
1116 }
1117 }
1118 if (!this->expect(Token::RPAREN, "')'")) {
1119 return nullptr;
1120 }
1121 std::unique_ptr<ASTStatement> statement(this->statement());
1122 if (!statement) {
1123 return nullptr;
1124 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001125 return std::unique_ptr<ASTForStatement>(new ASTForStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001126 std::move(initializer),
1127 std::move(test), std::move(next),
1128 std::move(statement)));
1129}
1130
1131/* RETURN expression? SEMICOLON */
1132std::unique_ptr<ASTReturnStatement> Parser::returnStatement() {
1133 Token start;
1134 if (!this->expect(Token::RETURN, "'return'", &start)) {
1135 return nullptr;
1136 }
1137 std::unique_ptr<ASTExpression> expression;
1138 if (this->peek().fKind != Token::SEMICOLON) {
1139 expression = this->expression();
1140 if (!expression) {
1141 return nullptr;
1142 }
1143 }
1144 if (!this->expect(Token::SEMICOLON, "';'")) {
1145 return nullptr;
1146 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001147 return std::unique_ptr<ASTReturnStatement>(new ASTReturnStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001148 std::move(expression)));
1149}
1150
1151/* BREAK SEMICOLON */
1152std::unique_ptr<ASTBreakStatement> Parser::breakStatement() {
1153 Token start;
1154 if (!this->expect(Token::BREAK, "'break'", &start)) {
1155 return nullptr;
1156 }
1157 if (!this->expect(Token::SEMICOLON, "';'")) {
1158 return nullptr;
1159 }
1160 return std::unique_ptr<ASTBreakStatement>(new ASTBreakStatement(start.fPosition));
1161}
1162
1163/* CONTINUE SEMICOLON */
1164std::unique_ptr<ASTContinueStatement> Parser::continueStatement() {
1165 Token start;
1166 if (!this->expect(Token::CONTINUE, "'continue'", &start)) {
1167 return nullptr;
1168 }
1169 if (!this->expect(Token::SEMICOLON, "';'")) {
1170 return nullptr;
1171 }
1172 return std::unique_ptr<ASTContinueStatement>(new ASTContinueStatement(start.fPosition));
1173}
1174
1175/* DISCARD SEMICOLON */
1176std::unique_ptr<ASTDiscardStatement> Parser::discardStatement() {
1177 Token start;
1178 if (!this->expect(Token::DISCARD, "'continue'", &start)) {
1179 return nullptr;
1180 }
1181 if (!this->expect(Token::SEMICOLON, "';'")) {
1182 return nullptr;
1183 }
1184 return std::unique_ptr<ASTDiscardStatement>(new ASTDiscardStatement(start.fPosition));
1185}
1186
1187/* LBRACE statement* RBRACE */
1188std::unique_ptr<ASTBlock> Parser::block() {
ethannicholascad64162016-10-27 10:54:02 -07001189 AutoDepth depth(this);
1190 if (!depth.checkValid()) {
1191 return nullptr;
1192 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001193 Token start;
1194 if (!this->expect(Token::LBRACE, "'{'", &start)) {
1195 return nullptr;
1196 }
1197 std::vector<std::unique_ptr<ASTStatement>> statements;
1198 for (;;) {
1199 switch (this->peek().fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001200 case Token::RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001201 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001202 return std::unique_ptr<ASTBlock>(new ASTBlock(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001203 std::move(statements)));
Ethan Nicholas11d53972016-11-28 11:23:23 -05001204 case Token::END_OF_FILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001205 this->error(this->peek().fPosition, "expected '}', but found end of file");
1206 return nullptr;
1207 default: {
1208 std::unique_ptr<ASTStatement> statement = this->statement();
1209 if (!statement) {
1210 return nullptr;
1211 }
1212 statements.push_back(std::move(statement));
1213 }
1214 }
1215 }
1216}
1217
1218/* expression SEMICOLON */
1219std::unique_ptr<ASTExpressionStatement> Parser::expressionStatement() {
1220 std::unique_ptr<ASTExpression> expr = this->expression();
1221 if (expr) {
1222 if (this->expect(Token::SEMICOLON, "';'")) {
1223 ASTExpressionStatement* result = new ASTExpressionStatement(std::move(expr));
1224 return std::unique_ptr<ASTExpressionStatement>(result);
1225 }
1226 }
1227 return nullptr;
1228}
1229
1230/* assignmentExpression */
1231std::unique_ptr<ASTExpression> Parser::expression() {
ethannicholascad64162016-10-27 10:54:02 -07001232 AutoDepth depth(this);
1233 if (!depth.checkValid()) {
1234 return nullptr;
1235 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001236 return this->assignmentExpression();
1237}
1238
1239/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1240 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1241 assignmentExpression)*
1242 */
1243std::unique_ptr<ASTExpression> Parser::assignmentExpression() {
1244 std::unique_ptr<ASTExpression> result = this->ternaryExpression();
1245 if (!result) {
1246 return nullptr;
1247 }
1248 for (;;) {
1249 switch (this->peek().fKind) {
1250 case Token::EQ: // fall through
1251 case Token::STAREQ: // fall through
1252 case Token::SLASHEQ: // fall through
1253 case Token::PERCENTEQ: // fall through
1254 case Token::PLUSEQ: // fall through
1255 case Token::MINUSEQ: // fall through
1256 case Token::SHLEQ: // fall through
1257 case Token::SHREQ: // fall through
1258 case Token::BITWISEANDEQ: // fall through
1259 case Token::BITWISEXOREQ: // fall through
1260 case Token::BITWISEOREQ: // fall through
1261 case Token::LOGICALANDEQ: // fall through
1262 case Token::LOGICALXOREQ: // fall through
1263 case Token::LOGICALOREQ: {
1264 Token t = this->nextToken();
1265 std::unique_ptr<ASTExpression> right = this->assignmentExpression();
1266 if (!right) {
1267 return nullptr;
1268 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001269 result = std::unique_ptr<ASTExpression>(new ASTBinaryExpression(std::move(result),
1270 t,
ethannicholasb3058bd2016-07-01 08:22:01 -07001271 std::move(right)));
1272 }
1273 default:
1274 return result;
1275 }
1276 }
1277}
1278
1279/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
1280std::unique_ptr<ASTExpression> Parser::ternaryExpression() {
1281 std::unique_ptr<ASTExpression> result = this->logicalOrExpression();
1282 if (!result) {
1283 return nullptr;
1284 }
1285 if (this->peek().fKind == Token::QUESTION) {
1286 Token question = this->nextToken();
1287 std::unique_ptr<ASTExpression> trueExpr = this->expression();
1288 if (!trueExpr) {
1289 return nullptr;
1290 }
1291 if (this->expect(Token::COLON, "':'")) {
1292 std::unique_ptr<ASTExpression> falseExpr = this->assignmentExpression();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001293 return std::unique_ptr<ASTExpression>(new ASTTernaryExpression(std::move(result),
1294 std::move(trueExpr),
ethannicholasb3058bd2016-07-01 08:22:01 -07001295 std::move(falseExpr)));
1296 }
1297 return nullptr;
1298 }
1299 return result;
1300}
1301
1302/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
1303std::unique_ptr<ASTExpression> Parser::logicalOrExpression() {
1304 std::unique_ptr<ASTExpression> result = this->logicalXorExpression();
1305 if (!result) {
1306 return nullptr;
1307 }
1308 while (this->peek().fKind == Token::LOGICALOR) {
1309 Token t = this->nextToken();
1310 std::unique_ptr<ASTExpression> right = this->logicalXorExpression();
1311 if (!right) {
1312 return nullptr;
1313 }
1314 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1315 }
1316 return result;
1317}
1318
1319/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
1320std::unique_ptr<ASTExpression> Parser::logicalXorExpression() {
1321 std::unique_ptr<ASTExpression> result = this->logicalAndExpression();
1322 if (!result) {
1323 return nullptr;
1324 }
1325 while (this->peek().fKind == Token::LOGICALXOR) {
1326 Token t = this->nextToken();
1327 std::unique_ptr<ASTExpression> right = this->logicalAndExpression();
1328 if (!right) {
1329 return nullptr;
1330 }
1331 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1332 }
1333 return result;
1334}
1335
1336/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
1337std::unique_ptr<ASTExpression> Parser::logicalAndExpression() {
1338 std::unique_ptr<ASTExpression> result = this->bitwiseOrExpression();
1339 if (!result) {
1340 return nullptr;
1341 }
1342 while (this->peek().fKind == Token::LOGICALAND) {
1343 Token t = this->nextToken();
1344 std::unique_ptr<ASTExpression> right = this->bitwiseOrExpression();
1345 if (!right) {
1346 return nullptr;
1347 }
1348 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1349 }
1350 return result;
1351}
1352
1353/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
1354std::unique_ptr<ASTExpression> Parser::bitwiseOrExpression() {
1355 std::unique_ptr<ASTExpression> result = this->bitwiseXorExpression();
1356 if (!result) {
1357 return nullptr;
1358 }
1359 while (this->peek().fKind == Token::BITWISEOR) {
1360 Token t = this->nextToken();
1361 std::unique_ptr<ASTExpression> right = this->bitwiseXorExpression();
1362 if (!right) {
1363 return nullptr;
1364 }
1365 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1366 }
1367 return result;
1368}
1369
1370/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
1371std::unique_ptr<ASTExpression> Parser::bitwiseXorExpression() {
1372 std::unique_ptr<ASTExpression> result = this->bitwiseAndExpression();
1373 if (!result) {
1374 return nullptr;
1375 }
1376 while (this->peek().fKind == Token::BITWISEXOR) {
1377 Token t = this->nextToken();
1378 std::unique_ptr<ASTExpression> right = this->bitwiseAndExpression();
1379 if (!right) {
1380 return nullptr;
1381 }
1382 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1383 }
1384 return result;
1385}
1386
1387/* equalityExpression (BITWISEAND equalityExpression)* */
1388std::unique_ptr<ASTExpression> Parser::bitwiseAndExpression() {
1389 std::unique_ptr<ASTExpression> result = this->equalityExpression();
1390 if (!result) {
1391 return nullptr;
1392 }
1393 while (this->peek().fKind == Token::BITWISEAND) {
1394 Token t = this->nextToken();
1395 std::unique_ptr<ASTExpression> right = this->equalityExpression();
1396 if (!right) {
1397 return nullptr;
1398 }
1399 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1400 }
1401 return result;
1402}
1403
1404/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
1405std::unique_ptr<ASTExpression> Parser::equalityExpression() {
1406 std::unique_ptr<ASTExpression> result = this->relationalExpression();
1407 if (!result) {
1408 return nullptr;
1409 }
1410 for (;;) {
1411 switch (this->peek().fKind) {
1412 case Token::EQEQ: // fall through
1413 case Token::NEQ: {
1414 Token t = this->nextToken();
1415 std::unique_ptr<ASTExpression> right = this->relationalExpression();
1416 if (!right) {
1417 return nullptr;
1418 }
1419 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1420 break;
1421 }
1422 default:
1423 return result;
1424 }
1425 }
1426}
1427
1428/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
1429std::unique_ptr<ASTExpression> Parser::relationalExpression() {
1430 std::unique_ptr<ASTExpression> result = this->shiftExpression();
1431 if (!result) {
1432 return nullptr;
1433 }
1434 for (;;) {
1435 switch (this->peek().fKind) {
1436 case Token::LT: // fall through
1437 case Token::GT: // fall through
1438 case Token::LTEQ: // fall through
1439 case Token::GTEQ: {
1440 Token t = this->nextToken();
1441 std::unique_ptr<ASTExpression> right = this->shiftExpression();
1442 if (!right) {
1443 return nullptr;
1444 }
1445 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1446 break;
1447 }
1448 default:
1449 return result;
1450 }
1451 }
1452}
1453
1454/* additiveExpression ((SHL | SHR) additiveExpression)* */
1455std::unique_ptr<ASTExpression> Parser::shiftExpression() {
1456 std::unique_ptr<ASTExpression> result = this->additiveExpression();
1457 if (!result) {
1458 return nullptr;
1459 }
1460 for (;;) {
1461 switch (this->peek().fKind) {
1462 case Token::SHL: // fall through
1463 case Token::SHR: {
1464 Token t = this->nextToken();
1465 std::unique_ptr<ASTExpression> right = this->additiveExpression();
1466 if (!right) {
1467 return nullptr;
1468 }
1469 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1470 break;
1471 }
1472 default:
1473 return result;
1474 }
1475 }
1476}
1477
1478/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
1479std::unique_ptr<ASTExpression> Parser::additiveExpression() {
1480 std::unique_ptr<ASTExpression> result = this->multiplicativeExpression();
1481 if (!result) {
1482 return nullptr;
1483 }
1484 for (;;) {
1485 switch (this->peek().fKind) {
1486 case Token::PLUS: // fall through
1487 case Token::MINUS: {
1488 Token t = this->nextToken();
1489 std::unique_ptr<ASTExpression> right = this->multiplicativeExpression();
1490 if (!right) {
1491 return nullptr;
1492 }
1493 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1494 break;
1495 }
1496 default:
1497 return result;
1498 }
1499 }
1500}
1501
1502/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
1503std::unique_ptr<ASTExpression> Parser::multiplicativeExpression() {
1504 std::unique_ptr<ASTExpression> result = this->unaryExpression();
1505 if (!result) {
1506 return nullptr;
1507 }
1508 for (;;) {
1509 switch (this->peek().fKind) {
1510 case Token::STAR: // fall through
1511 case Token::SLASH: // fall through
1512 case Token::PERCENT: {
1513 Token t = this->nextToken();
1514 std::unique_ptr<ASTExpression> right = this->unaryExpression();
1515 if (!right) {
1516 return nullptr;
1517 }
1518 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1519 break;
1520 }
1521 default:
1522 return result;
1523 }
1524 }
1525}
1526
1527/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
1528std::unique_ptr<ASTExpression> Parser::unaryExpression() {
1529 switch (this->peek().fKind) {
ethannicholas5961bc92016-10-12 06:39:56 -07001530 case Token::PLUS: // fall through
1531 case Token::MINUS: // fall through
1532 case Token::LOGICALNOT: // fall through
1533 case Token::BITWISENOT: // fall through
1534 case Token::PLUSPLUS: // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001535 case Token::MINUSMINUS: {
1536 Token t = this->nextToken();
1537 std::unique_ptr<ASTExpression> expr = this->unaryExpression();
1538 if (!expr) {
1539 return nullptr;
1540 }
1541 return std::unique_ptr<ASTExpression>(new ASTPrefixExpression(t, std::move(expr)));
1542 }
1543 default:
1544 return this->postfixExpression();
1545 }
1546}
1547
1548/* term suffix* */
1549std::unique_ptr<ASTExpression> Parser::postfixExpression() {
1550 std::unique_ptr<ASTExpression> result = this->term();
1551 if (!result) {
1552 return nullptr;
1553 }
1554 for (;;) {
1555 switch (this->peek().fKind) {
1556 case Token::LBRACKET: // fall through
1557 case Token::DOT: // fall through
1558 case Token::LPAREN: // fall through
1559 case Token::PLUSPLUS: // fall through
1560 case Token::MINUSMINUS: {
1561 std::unique_ptr<ASTSuffix> s = this->suffix();
1562 if (!s) {
1563 return nullptr;
1564 }
1565 result.reset(new ASTSuffixExpression(std::move(result), std::move(s)));
1566 break;
1567 }
1568 default:
1569 return result;
1570 }
1571 }
1572}
1573
Ethan Nicholas11d53972016-11-28 11:23:23 -05001574/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
ethannicholasb3058bd2016-07-01 08:22:01 -07001575 PLUSPLUS | MINUSMINUS */
1576std::unique_ptr<ASTSuffix> Parser::suffix() {
1577 Token next = this->nextToken();
1578 switch (next.fKind) {
1579 case Token::LBRACKET: {
ethannicholas5961bc92016-10-12 06:39:56 -07001580 if (this->peek().fKind == Token::RBRACKET) {
1581 this->nextToken();
1582 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(next.fPosition));
1583 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001584 std::unique_ptr<ASTExpression> e = this->expression();
1585 if (!e) {
1586 return nullptr;
1587 }
1588 this->expect(Token::RBRACKET, "']' to complete array access expression");
1589 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(std::move(e)));
1590 }
1591 case Token::DOT: {
1592 Position pos = this->peek().fPosition;
Ethan Nicholas9e1138d2016-11-21 10:39:35 -05001593 SkString text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001594 if (this->identifier(&text)) {
1595 return std::unique_ptr<ASTSuffix>(new ASTFieldSuffix(pos, std::move(text)));
1596 }
1597 return nullptr;
1598 }
1599 case Token::LPAREN: {
1600 std::vector<std::unique_ptr<ASTExpression>> parameters;
1601 if (this->peek().fKind != Token::RPAREN) {
1602 for (;;) {
1603 std::unique_ptr<ASTExpression> expr = this->expression();
1604 if (!expr) {
1605 return nullptr;
1606 }
1607 parameters.push_back(std::move(expr));
1608 if (this->peek().fKind != Token::COMMA) {
1609 break;
1610 }
1611 this->nextToken();
1612 }
1613 }
1614 this->expect(Token::RPAREN, "')' to complete function parameters");
Ethan Nicholas11d53972016-11-28 11:23:23 -05001615 return std::unique_ptr<ASTSuffix>(new ASTCallSuffix(next.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001616 std::move(parameters)));
1617 }
1618 case Token::PLUSPLUS:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001619 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001620 ASTSuffix::kPostIncrement_Kind));
1621 case Token::MINUSMINUS:
1622 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fPosition,
1623 ASTSuffix::kPostDecrement_Kind));
1624 default: {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001625 this->error(next.fPosition, "expected expression suffix, but found '" + next.fText +
ethannicholasb3058bd2016-07-01 08:22:01 -07001626 "'\n");
1627 return nullptr;
1628 }
1629 }
1630}
1631
1632/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | '(' expression ')' */
1633std::unique_ptr<ASTExpression> Parser::term() {
1634 std::unique_ptr<ASTExpression> result;
1635 Token t = this->peek();
1636 switch (t.fKind) {
1637 case Token::IDENTIFIER: {
Ethan Nicholas9e1138d2016-11-21 10:39:35 -05001638 SkString text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001639 if (this->identifier(&text)) {
1640 result.reset(new ASTIdentifier(t.fPosition, std::move(text)));
1641 }
1642 break;
1643 }
1644 case Token::INT_LITERAL: {
1645 int64_t i;
1646 if (this->intLiteral(&i)) {
1647 result.reset(new ASTIntLiteral(t.fPosition, i));
1648 }
1649 break;
1650 }
1651 case Token::FLOAT_LITERAL: {
1652 double f;
1653 if (this->floatLiteral(&f)) {
1654 result.reset(new ASTFloatLiteral(t.fPosition, f));
1655 }
1656 break;
1657 }
1658 case Token::TRUE_LITERAL: // fall through
1659 case Token::FALSE_LITERAL: {
1660 bool b;
1661 if (this->boolLiteral(&b)) {
1662 result.reset(new ASTBoolLiteral(t.fPosition, b));
1663 }
1664 break;
1665 }
1666 case Token::LPAREN: {
1667 this->nextToken();
1668 result = this->expression();
1669 if (result) {
1670 this->expect(Token::RPAREN, "')' to complete expression");
1671 }
1672 break;
1673 }
1674 default:
1675 this->nextToken();
1676 this->error(t.fPosition, "expected expression, but found '" + t.fText + "'\n");
1677 result = nullptr;
1678 }
1679 return result;
1680}
1681
1682/* INT_LITERAL */
1683bool Parser::intLiteral(int64_t* dest) {
1684 Token t;
1685 if (this->expect(Token::INT_LITERAL, "integer literal", &t)) {
1686 *dest = SkSL::stol(t.fText);
1687 return true;
1688 }
1689 return false;
1690}
1691
1692/* FLOAT_LITERAL */
1693bool Parser::floatLiteral(double* dest) {
1694 Token t;
1695 if (this->expect(Token::FLOAT_LITERAL, "float literal", &t)) {
1696 *dest = SkSL::stod(t.fText);
1697 return true;
1698 }
1699 return false;
1700}
1701
1702/* TRUE_LITERAL | FALSE_LITERAL */
1703bool Parser::boolLiteral(bool* dest) {
1704 Token t = this->nextToken();
1705 switch (t.fKind) {
1706 case Token::TRUE_LITERAL:
1707 *dest = true;
1708 return true;
1709 case Token::FALSE_LITERAL:
1710 *dest = false;
1711 return true;
1712 default:
1713 this->error(t.fPosition, "expected 'true' or 'false', but found '" + t.fText + "'\n");
1714 return false;
1715 }
1716}
1717
1718/* IDENTIFIER */
Ethan Nicholas9e1138d2016-11-21 10:39:35 -05001719bool Parser::identifier(SkString* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001720 Token t;
1721 if (this->expect(Token::IDENTIFIER, "identifier", &t)) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001722 *dest = t.fText;
ethannicholasb3058bd2016-07-01 08:22:01 -07001723 return true;
1724 }
1725 return false;
1726}
1727
1728} // namespace