blob: cc47577e93309ce15b0872eaeb03c63e0a560357 [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"
66#include "ast/SkSLASTTernaryExpression.h"
67#include "ast/SkSLASTType.h"
68#include "ast/SkSLASTVarDeclaration.h"
69#include "ast/SkSLASTVarDeclarationStatement.h"
70#include "ast/SkSLASTWhileStatement.h"
71#include "ir/SkSLSymbolTable.h"
Ethan Nicholas11d53972016-11-28 11:23:23 -050072#include "ir/SkSLModifiers.h"
ethannicholasd598f792016-07-25 10:08:54 -070073#include "ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070074
75namespace SkSL {
76
ethannicholascad64162016-10-27 10:54:02 -070077#define MAX_PARSE_DEPTH 50
78
79class AutoDepth {
80public:
81 AutoDepth(Parser* p)
82 : fParser(p) {
83 fParser->fDepth++;
84 }
85
86 ~AutoDepth() {
87 fParser->fDepth--;
88 }
89
90 bool checkValid() {
91 if (fParser->fDepth > MAX_PARSE_DEPTH) {
Ethan Nicholas9e1138d2016-11-21 10:39:35 -050092 fParser->error(fParser->peek().fPosition, SkString("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070093 return false;
94 }
95 return true;
96 }
97
98private:
99 Parser* fParser;
100};
101
Ethan Nicholas11d53972016-11-28 11:23:23 -0500102Parser::Parser(SkString text, SymbolTable& types, ErrorReporter& errors)
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500103: fPushback(Position(-1, -1), Token::INVALID_TOKEN, SkString())
ethannicholasb3058bd2016-07-01 08:22:01 -0700104, fTypes(types)
105, fErrors(errors) {
106 sksllex_init(&fScanner);
107 fBuffer = sksl_scan_string(text.c_str(), fScanner);
108 skslset_lineno(1, fScanner);
109
110 if (false) {
111 // avoid unused warning
112 yyunput(0, nullptr, fScanner);
113 }
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500114
115 fLayoutKeys[SkString("location")] = kLocation_LayoutKey;
116 fLayoutKeys[SkString("offset")] = kOffset_LayoutKey;
117 fLayoutKeys[SkString("binding")] = kBinding_LayoutKey;
118 fLayoutKeys[SkString("index")] = kIndex_LayoutKey;
119 fLayoutKeys[SkString("set")] = kSet_LayoutKey;
120 fLayoutKeys[SkString("builtin")] = kBuiltin_LayoutKey;
121 fLayoutKeys[SkString("input_attachment_index")] = kInputAttachmentIndex_LayoutKey;
122 fLayoutKeys[SkString("origin_upper_left")] = kOriginUpperLeft_LayoutKey;
123 fLayoutKeys[SkString("override_coverage")] = kOverrideCoverage_LayoutKey;
124 fLayoutKeys[SkString("blend_support_all_equations")] = kBlendSupportAllEquations_LayoutKey;
125 fLayoutKeys[SkString("push_constant")] = kPushConstant_LayoutKey;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500126 fLayoutKeys[SkString("points")] = kPoints_LayoutKey;
127 fLayoutKeys[SkString("lines")] = kLines_LayoutKey;
128 fLayoutKeys[SkString("line_strip")] = kLineStrip_LayoutKey;
129 fLayoutKeys[SkString("lines_adjacency")] = kLinesAdjacency_LayoutKey;
130 fLayoutKeys[SkString("triangles")] = kTriangles_LayoutKey;
131 fLayoutKeys[SkString("triangle_strip")] = kTriangleStrip_LayoutKey;
132 fLayoutKeys[SkString("triangles_adjacency")] = kTrianglesAdjacency_LayoutKey;
133 fLayoutKeys[SkString("max_vertices")] = kMaxVertices_LayoutKey;
134 fLayoutKeys[SkString("invocations")] = kInvocations_LayoutKey;
ethannicholasb3058bd2016-07-01 08:22:01 -0700135}
136
137Parser::~Parser() {
138 sksl_delete_buffer(fBuffer, fScanner);
ethannicholas69a7e7c2016-07-01 19:09:27 -0700139 sksllex_destroy(fScanner);
ethannicholasb3058bd2016-07-01 08:22:01 -0700140}
141
142/* (precision | directive | declaration)* END_OF_FILE */
143std::vector<std::unique_ptr<ASTDeclaration>> Parser::file() {
144 std::vector<std::unique_ptr<ASTDeclaration>> result;
145 for (;;) {
146 switch (this->peek().fKind) {
147 case Token::END_OF_FILE:
148 return result;
ethannicholas5961bc92016-10-12 06:39:56 -0700149 case Token::PRECISION: {
150 std::unique_ptr<ASTDeclaration> precision = this->precision();
151 if (precision) {
152 result.push_back(std::move(precision));
153 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700154 break;
ethannicholas5961bc92016-10-12 06:39:56 -0700155 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700156 case Token::DIRECTIVE: {
157 std::unique_ptr<ASTDeclaration> decl = this->directive();
158 if (decl) {
159 result.push_back(std::move(decl));
160 }
161 break;
162 }
163 default: {
164 std::unique_ptr<ASTDeclaration> decl = this->declaration();
165 if (!decl) {
166 continue;
167 }
168 result.push_back(std::move(decl));
169 }
170 }
171 }
172}
173
174Token Parser::nextToken() {
175 if (fPushback.fKind != Token::INVALID_TOKEN) {
176 Token result = fPushback;
177 fPushback.fKind = Token::INVALID_TOKEN;
178 fPushback.fText = "";
179 return result;
180 }
181 int token = sksllex(fScanner);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500182 SkString text;
ethannicholas471e8942016-10-28 09:02:46 -0700183 switch ((Token::Kind) token) {
184 case Token::IDENTIFIER: // fall through
185 case Token::INT_LITERAL: // fall through
186 case Token::FLOAT_LITERAL: // fall through
187 case Token::DIRECTIVE:
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500188 text = SkString(skslget_text(fScanner));
ethannicholas471e8942016-10-28 09:02:46 -0700189 break;
190 default:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500191#ifdef SK_DEBUG
192 text = SkString(skslget_text(fScanner));
193#endif
ethannicholas471e8942016-10-28 09:02:46 -0700194 break;
195 }
196 return Token(Position(skslget_lineno(fScanner), -1), (Token::Kind) token, text);
ethannicholasb3058bd2016-07-01 08:22:01 -0700197}
198
199void Parser::pushback(Token t) {
200 ASSERT(fPushback.fKind == Token::INVALID_TOKEN);
201 fPushback = t;
202}
203
204Token Parser::peek() {
205 fPushback = this->nextToken();
206 return fPushback;
207}
208
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500209
210bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
211 return this->expect(kind, SkString(expected), result);
212}
213
214bool Parser::expect(Token::Kind kind, SkString expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700215 Token next = this->nextToken();
216 if (next.fKind == kind) {
217 if (result) {
218 *result = next;
219 }
220 return true;
221 } else {
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500222 if (next.fText.size()) {
223 this->error(next.fPosition, "expected " + expected + ", but found '" + next.fText +
224 "'");
225 } else {
226 this->error(next.fPosition, "parse error, recompile in debug mode for details");
227 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700228 return false;
229 }
230}
231
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500232void Parser::error(Position p, const char* msg) {
233 this->error(p, SkString(msg));
234}
235
236void Parser::error(Position p, SkString msg) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700237 fErrors.error(p, msg);
238}
239
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500240bool Parser::isType(SkString name) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700241 return nullptr != fTypes[name];
242}
243
244/* PRECISION (LOWP | MEDIUMP | HIGHP) type SEMICOLON */
ethannicholas5961bc92016-10-12 06:39:56 -0700245std::unique_ptr<ASTDeclaration> Parser::precision() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700246 if (!this->expect(Token::PRECISION, "'precision'")) {
ethannicholas5961bc92016-10-12 06:39:56 -0700247 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700248 }
ethannicholas5961bc92016-10-12 06:39:56 -0700249 Modifiers::Flag result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700250 Token p = this->nextToken();
251 switch (p.fKind) {
ethannicholas5961bc92016-10-12 06:39:56 -0700252 case Token::LOWP:
253 result = Modifiers::kLowp_Flag;
254 break;
255 case Token::MEDIUMP:
256 result = Modifiers::kMediump_Flag;
257 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700258 case Token::HIGHP:
ethannicholas5961bc92016-10-12 06:39:56 -0700259 result = Modifiers::kHighp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700260 break;
261 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500262 this->error(p.fPosition, "expected 'lowp', 'mediump', or 'highp', but found '" +
ethannicholasb3058bd2016-07-01 08:22:01 -0700263 p.fText + "'");
ethannicholas5961bc92016-10-12 06:39:56 -0700264 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700265 }
ethannicholas5961bc92016-10-12 06:39:56 -0700266 // FIXME handle the type
ethannicholasb3058bd2016-07-01 08:22:01 -0700267 if (!this->type()) {
ethannicholas5961bc92016-10-12 06:39:56 -0700268 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700269 }
270 this->expect(Token::SEMICOLON, "';'");
ethannicholas5961bc92016-10-12 06:39:56 -0700271 return std::unique_ptr<ASTDeclaration>(new ASTPrecision(p.fPosition, result));
ethannicholasb3058bd2016-07-01 08:22:01 -0700272}
273
Ethan Nicholas11d53972016-11-28 11:23:23 -0500274/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700275 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
ethannicholasb3058bd2016-07-01 08:22:01 -0700276std::unique_ptr<ASTDeclaration> Parser::directive() {
277 Token start;
278 if (!this->expect(Token::DIRECTIVE, "a directive", &start)) {
279 return nullptr;
280 }
281 if (start.fText == "#version") {
282 this->expect(Token::INT_LITERAL, "a version number");
ethannicholas5961bc92016-10-12 06:39:56 -0700283 Token next = this->peek();
284 if (next.fText == "es" || next.fText == "compatibility") {
285 this->nextToken();
286 }
287 // version is ignored for now; it will eventually become an error when we stop pretending
288 // to be GLSL
ethannicholasb3058bd2016-07-01 08:22:01 -0700289 return nullptr;
290 } else if (start.fText == "#extension") {
291 Token name;
292 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
293 return nullptr;
294 }
295 if (!this->expect(Token::COLON, "':'")) {
296 return nullptr;
297 }
298 // FIXME: need to start paying attention to this token
299 if (!this->expect(Token::IDENTIFIER, "an identifier")) {
300 return nullptr;
301 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500302 return std::unique_ptr<ASTDeclaration>(new ASTExtension(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700303 std::move(name.fText)));
304 } else {
305 this->error(start.fPosition, "unsupported directive '" + start.fText + "'");
306 return nullptr;
307 }
308}
309
Ethan Nicholas11d53972016-11-28 11:23:23 -0500310/* modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700311 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
312std::unique_ptr<ASTDeclaration> Parser::declaration() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500313 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700314 Token lookahead = this->peek();
315 if (lookahead.fKind == Token::IDENTIFIER && !this->isType(lookahead.fText)) {
316 // we have an identifier that's not a type, could be the start of an interface block
317 return this->interfaceBlock(modifiers);
318 }
319 if (lookahead.fKind == Token::STRUCT) {
320 return this->structVarDeclaration(modifiers);
321 }
ethannicholas5961bc92016-10-12 06:39:56 -0700322 if (lookahead.fKind == Token::SEMICOLON) {
323 this->nextToken();
324 return std::unique_ptr<ASTDeclaration>(new ASTModifiersDeclaration(modifiers));
325 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700326 std::unique_ptr<ASTType> type(this->type());
327 if (!type) {
328 return nullptr;
329 }
330 if (type->fKind == ASTType::kStruct_Kind && peek().fKind == Token::SEMICOLON) {
331 this->nextToken();
332 return nullptr;
333 }
334 Token name;
335 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
336 return nullptr;
337 }
338 if (!modifiers.fFlags && this->peek().fKind == Token::LPAREN) {
339 this->nextToken();
340 std::vector<std::unique_ptr<ASTParameter>> parameters;
341 while (this->peek().fKind != Token::RPAREN) {
342 if (parameters.size() > 0) {
343 if (!this->expect(Token::COMMA, "','")) {
344 return nullptr;
345 }
346 }
347 std::unique_ptr<ASTParameter> parameter = this->parameter();
348 if (!parameter) {
349 return nullptr;
350 }
351 parameters.push_back(std::move(parameter));
352 }
353 this->nextToken();
354 std::unique_ptr<ASTBlock> body;
355 if (this->peek().fKind == Token::SEMICOLON) {
356 this->nextToken();
357 } else {
358 body = this->block();
359 if (!body) {
360 return nullptr;
361 }
362 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500363 return std::unique_ptr<ASTDeclaration>(new ASTFunction(name.fPosition, std::move(type),
364 std::move(name.fText),
365 std::move(parameters),
ethannicholasb3058bd2016-07-01 08:22:01 -0700366 std::move(body)));
367 } else {
368 return this->varDeclarationEnd(modifiers, std::move(type), name.fText);
369 }
370}
371
372/* modifiers type IDENTIFIER varDeclarationEnd */
ethannicholas14fe8cc2016-09-07 13:37:16 -0700373std::unique_ptr<ASTVarDeclarations> Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500374 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700375 std::unique_ptr<ASTType> type(this->type());
376 if (!type) {
377 return nullptr;
378 }
379 Token name;
380 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
381 return nullptr;
382 }
383 return this->varDeclarationEnd(modifiers, std::move(type), std::move(name.fText));
384}
385
386/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
387std::unique_ptr<ASTType> Parser::structDeclaration() {
388 if (!this->expect(Token::STRUCT, "'struct'")) {
389 return nullptr;
390 }
391 Token name;
392 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
393 return nullptr;
394 }
395 if (!this->expect(Token::LBRACE, "'{'")) {
396 return nullptr;
397 }
398 std::vector<Type::Field> fields;
399 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700400 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700401 if (!decl) {
402 return nullptr;
403 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700404 for (const auto& var : decl->fVars) {
ethannicholasd598f792016-07-25 10:08:54 -0700405 auto type = (const Type*) fTypes[decl->fType->fName];
ethannicholas14fe8cc2016-09-07 13:37:16 -0700406 for (int i = (int) var.fSizes.size() - 1; i >= 0; i--) {
ethannicholasdd4645b2016-10-14 12:14:46 -0700407 if (!var.fSizes[i] || var.fSizes[i]->fKind != ASTExpression::kInt_Kind) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700408 this->error(decl->fPosition, "array size in struct field must be a constant");
ethannicholasdd4645b2016-10-14 12:14:46 -0700409 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700410 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700411 uint64_t columns = ((ASTIntLiteral&) *var.fSizes[i]).fValue;
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500412 SkString name = type->name() + "[" + to_string(columns) + "]";
ethannicholasd598f792016-07-25 10:08:54 -0700413 type = new Type(name, Type::kArray_Kind, *type, (int) columns);
414 fTypes.takeOwnership((Type*) type);
ethannicholasb3058bd2016-07-01 08:22:01 -0700415 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700416 fields.push_back(Type::Field(decl->fModifiers, var.fName, type));
417 if (var.fValue) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700418 this->error(decl->fPosition, "initializers are not permitted on struct fields");
419 }
420 }
421 }
422 if (!this->expect(Token::RBRACE, "'}'")) {
423 return nullptr;
424 }
Ethan Nicholas19671772016-11-28 16:30:17 -0500425 fTypes.add(name.fText, std::unique_ptr<Type>(new Type(name.fPosition, name.fText, fields)));
Ethan Nicholas11d53972016-11-28 11:23:23 -0500426 return std::unique_ptr<ASTType>(new ASTType(name.fPosition, name.fText,
Ethan Nicholas50afc172017-02-16 14:49:57 -0500427 ASTType::kStruct_Kind, std::vector<int>()));
ethannicholasb3058bd2016-07-01 08:22:01 -0700428}
429
430/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500431std::unique_ptr<ASTVarDeclarations> Parser::structVarDeclaration(Modifiers modifiers) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700432 std::unique_ptr<ASTType> type = this->structDeclaration();
433 if (!type) {
434 return nullptr;
435 }
436 if (peek().fKind == Token::IDENTIFIER) {
437 Token name = this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500438 std::unique_ptr<ASTVarDeclarations> result = this->varDeclarationEnd(modifiers,
439 std::move(type),
ethannicholas14fe8cc2016-09-07 13:37:16 -0700440 std::move(name.fText));
ethannicholasb3058bd2016-07-01 08:22:01 -0700441 if (result) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700442 for (const auto& var : result->fVars) {
443 if (var.fValue) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500444 this->error(var.fValue->fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700445 "struct variables cannot be initialized");
446 }
447 }
448 }
449 return result;
450 }
451 this->expect(Token::SEMICOLON, "';'");
452 return nullptr;
453}
454
Ethan Nicholas11d53972016-11-28 11:23:23 -0500455/* (LBRACKET expression? RBRACKET)* (EQ expression)? (COMMA IDENTIFER
ethannicholasb3058bd2016-07-01 08:22:01 -0700456 (LBRACKET expression? RBRACKET)* (EQ expression)?)* SEMICOLON */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500457std::unique_ptr<ASTVarDeclarations> Parser::varDeclarationEnd(Modifiers mods,
ethannicholas14fe8cc2016-09-07 13:37:16 -0700458 std::unique_ptr<ASTType> type,
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500459 SkString name) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700460 std::vector<ASTVarDeclaration> vars;
ethannicholasb3058bd2016-07-01 08:22:01 -0700461 std::vector<std::unique_ptr<ASTExpression>> currentVarSizes;
462 while (this->peek().fKind == Token::LBRACKET) {
463 this->nextToken();
464 if (this->peek().fKind == Token::RBRACKET) {
465 this->nextToken();
466 currentVarSizes.push_back(nullptr);
467 } else {
468 std::unique_ptr<ASTExpression> size(this->expression());
469 if (!size) {
470 return nullptr;
471 }
472 currentVarSizes.push_back(std::move(size));
473 if (!this->expect(Token::RBRACKET, "']'")) {
474 return nullptr;
475 }
476 }
477 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700478 std::unique_ptr<ASTExpression> value;
ethannicholasb3058bd2016-07-01 08:22:01 -0700479 if (this->peek().fKind == Token::EQ) {
480 this->nextToken();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700481 value = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700482 if (!value) {
483 return nullptr;
484 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700485 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700486 vars.emplace_back(std::move(name), std::move(currentVarSizes), std::move(value));
ethannicholasb3058bd2016-07-01 08:22:01 -0700487 while (this->peek().fKind == Token::COMMA) {
488 this->nextToken();
489 Token name;
490 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
491 return nullptr;
492 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700493 currentVarSizes.clear();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700494 value.reset();
ethannicholasb3058bd2016-07-01 08:22:01 -0700495 while (this->peek().fKind == Token::LBRACKET) {
496 this->nextToken();
497 if (this->peek().fKind == Token::RBRACKET) {
498 this->nextToken();
499 currentVarSizes.push_back(nullptr);
500 } else {
501 std::unique_ptr<ASTExpression> size(this->expression());
502 if (!size) {
503 return nullptr;
504 }
505 currentVarSizes.push_back(std::move(size));
506 if (!this->expect(Token::RBRACKET, "']'")) {
507 return nullptr;
508 }
509 }
510 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700511 if (this->peek().fKind == Token::EQ) {
512 this->nextToken();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700513 value = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700514 if (!value) {
515 return nullptr;
516 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700517 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700518 vars.emplace_back(std::move(name.fText), std::move(currentVarSizes), std::move(value));
ethannicholasb3058bd2016-07-01 08:22:01 -0700519 }
520 if (!this->expect(Token::SEMICOLON, "';'")) {
521 return nullptr;
522 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700523 return std::unique_ptr<ASTVarDeclarations>(new ASTVarDeclarations(std::move(mods),
524 std::move(type),
525 std::move(vars)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700526}
527
528/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
529std::unique_ptr<ASTParameter> Parser::parameter() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500530 Modifiers modifiers = this->modifiersWithDefaults(Modifiers::kIn_Flag);
ethannicholasb3058bd2016-07-01 08:22:01 -0700531 std::unique_ptr<ASTType> type = this->type();
532 if (!type) {
533 return nullptr;
534 }
535 Token name;
536 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
537 return nullptr;
538 }
539 std::vector<int> sizes;
540 while (this->peek().fKind == Token::LBRACKET) {
541 this->nextToken();
542 Token sizeToken;
543 if (!this->expect(Token::INT_LITERAL, "a positive integer", &sizeToken)) {
544 return nullptr;
545 }
546 sizes.push_back(SkSL::stoi(sizeToken.fText));
547 if (!this->expect(Token::RBRACKET, "']'")) {
548 return nullptr;
549 }
550 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500551 return std::unique_ptr<ASTParameter>(new ASTParameter(name.fPosition, modifiers,
552 std::move(type), name.fText,
ethannicholasb3058bd2016-07-01 08:22:01 -0700553 std::move(sizes)));
554}
555
556/** (EQ INT_LITERAL)? */
557int Parser::layoutInt() {
558 if (!this->expect(Token::EQ, "'='")) {
559 return -1;
560 }
561 Token resultToken;
562 if (this->expect(Token::INT_LITERAL, "a non-negative integer", &resultToken)) {
563 return SkSL::stoi(resultToken.fText);
564 }
565 return -1;
566}
567
ethannicholas8ac838d2016-11-22 08:39:36 -0800568/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500569Layout Parser::layout() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700570 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500571 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700572 int binding = -1;
573 int index = -1;
574 int set = -1;
575 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500576 int inputAttachmentIndex = -1;
ethannicholasf789b382016-08-03 12:43:36 -0700577 bool originUpperLeft = false;
ethannicholas5961bc92016-10-12 06:39:56 -0700578 bool overrideCoverage = false;
579 bool blendSupportAllEquations = false;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500580 Layout::Format format = Layout::Format::kUnspecified;
ethannicholas8ac838d2016-11-22 08:39:36 -0800581 bool pushConstant = false;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500582 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
583 int maxVertices = -1;
584 int invocations = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700585 if (this->peek().fKind == Token::LAYOUT) {
586 this->nextToken();
587 if (!this->expect(Token::LPAREN, "'('")) {
Ethan Nicholas19671772016-11-28 16:30:17 -0500588 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
Ethan Nicholas11d53972016-11-28 11:23:23 -0500589 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas52cad152017-02-16 16:37:32 -0500590 pushConstant, primitive, maxVertices, invocations);
ethannicholasb3058bd2016-07-01 08:22:01 -0700591 }
592 for (;;) {
593 Token t = this->nextToken();
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500594 auto found = fLayoutKeys.find(t.fText);
595 if (found != fLayoutKeys.end()) {
596 switch (found->second) {
597 case kLocation_LayoutKey:
598 location = this->layoutInt();
599 break;
600 case kOffset_LayoutKey:
601 offset = this->layoutInt();
602 break;
603 case kBinding_LayoutKey:
604 binding = this->layoutInt();
605 break;
606 case kIndex_LayoutKey:
607 index = this->layoutInt();
608 break;
609 case kSet_LayoutKey:
610 set = this->layoutInt();
611 break;
612 case kBuiltin_LayoutKey:
613 builtin = this->layoutInt();
614 break;
615 case kInputAttachmentIndex_LayoutKey:
616 inputAttachmentIndex = this->layoutInt();
617 break;
618 case kOriginUpperLeft_LayoutKey:
619 originUpperLeft = true;
620 break;
621 case kOverrideCoverage_LayoutKey:
622 overrideCoverage = true;
623 break;
624 case kBlendSupportAllEquations_LayoutKey:
625 blendSupportAllEquations = true;
626 break;
627 case kPushConstant_LayoutKey:
628 pushConstant = true;
629 break;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500630 case kPoints_LayoutKey:
631 primitive = Layout::kPoints_Primitive;
632 break;
633 case kLines_LayoutKey:
634 primitive = Layout::kLines_Primitive;
635 break;
636 case kLineStrip_LayoutKey:
637 primitive = Layout::kLineStrip_Primitive;
638 break;
639 case kLinesAdjacency_LayoutKey:
640 primitive = Layout::kLinesAdjacency_Primitive;
641 break;
642 case kTriangles_LayoutKey:
643 primitive = Layout::kTriangles_Primitive;
644 break;
645 case kTriangleStrip_LayoutKey:
646 primitive = Layout::kTriangleStrip_Primitive;
647 break;
648 case kTrianglesAdjacency_LayoutKey:
649 primitive = Layout::kTrianglesAdjacency_Primitive;
650 break;
651 case kMaxVertices_LayoutKey:
652 maxVertices = this->layoutInt();
653 break;
654 case kInvocations_LayoutKey:
655 invocations = this->layoutInt();
656 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500657 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500658 } else if (Layout::ReadFormat(t.fText, &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500659 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700660 } else {
Greg Daniel64773e62016-11-22 09:44:03 -0500661 this->error(t.fPosition, ("'" + t.fText +
ethannicholasb3058bd2016-07-01 08:22:01 -0700662 "' is not a valid layout qualifier").c_str());
663 }
664 if (this->peek().fKind == Token::RPAREN) {
665 this->nextToken();
666 break;
667 }
668 if (!this->expect(Token::COMMA, "','")) {
669 break;
670 }
671 }
672 }
Ethan Nicholas19671772016-11-28 16:30:17 -0500673 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
674 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas52cad152017-02-16 16:37:32 -0500675 pushConstant, primitive, maxVertices, invocations);
ethannicholasb3058bd2016-07-01 08:22:01 -0700676}
677
Brian Salomonf9f45122016-11-29 11:59:17 -0500678/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
679 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500680Modifiers Parser::modifiers() {
681 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700682 int flags = 0;
683 for (;;) {
684 // TODO: handle duplicate / incompatible flags
685 switch (peek().fKind) {
686 case Token::UNIFORM:
687 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500688 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700689 break;
690 case Token::CONST:
691 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500692 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700693 break;
694 case Token::IN:
695 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500696 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700697 break;
698 case Token::OUT:
699 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500700 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700701 break;
702 case Token::INOUT:
703 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500704 flags |= Modifiers::kIn_Flag;
705 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700706 break;
707 case Token::LOWP:
708 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500709 flags |= Modifiers::kLowp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700710 break;
711 case Token::MEDIUMP:
712 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500713 flags |= Modifiers::kMediump_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700714 break;
715 case Token::HIGHP:
716 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500717 flags |= Modifiers::kHighp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700718 break;
ethannicholasf789b382016-08-03 12:43:36 -0700719 case Token::FLAT:
720 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500721 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700722 break;
723 case Token::NOPERSPECTIVE:
724 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500725 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700726 break;
Brian Salomonf9f45122016-11-29 11:59:17 -0500727 case Token::READONLY:
728 this->nextToken();
729 flags |= Modifiers::kReadOnly_Flag;
730 break;
731 case Token::WRITEONLY:
732 this->nextToken();
733 flags |= Modifiers::kWriteOnly_Flag;
734 break;
735 case Token::COHERENT:
736 this->nextToken();
737 flags |= Modifiers::kCoherent_Flag;
738 break;
739 case Token::VOLATILE:
740 this->nextToken();
741 flags |= Modifiers::kVolatile_Flag;
742 break;
743 case Token::RESTRICT:
744 this->nextToken();
745 flags |= Modifiers::kRestrict_Flag;
746 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700747 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500748 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700749 }
750 }
751}
752
Ethan Nicholas11d53972016-11-28 11:23:23 -0500753Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
754 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700755 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500756 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700757 }
758 return result;
759}
760
761/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
762std::unique_ptr<ASTStatement> Parser::statement() {
763 Token start = this->peek();
764 switch (start.fKind) {
765 case Token::IF:
766 return this->ifStatement();
767 case Token::FOR:
768 return this->forStatement();
769 case Token::DO:
770 return this->doStatement();
771 case Token::WHILE:
772 return this->whileStatement();
773 case Token::RETURN:
774 return this->returnStatement();
775 case Token::BREAK:
776 return this->breakStatement();
777 case Token::CONTINUE:
778 return this->continueStatement();
779 case Token::DISCARD:
780 return this->discardStatement();
781 case Token::LBRACE:
782 return this->block();
783 case Token::SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500784 this->nextToken();
785 return std::unique_ptr<ASTStatement>(new ASTBlock(start.fPosition,
ethannicholas0730be72016-09-01 07:59:02 -0700786 std::vector<std::unique_ptr<ASTStatement>>()));
ethannicholasb3058bd2016-07-01 08:22:01 -0700787 case Token::CONST: // fall through
788 case Token::HIGHP: // fall through
789 case Token::MEDIUMP: // fall through
790 case Token::LOWP: {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700791 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700792 if (!decl) {
793 return nullptr;
794 }
795 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(std::move(decl)));
796 }
797 case Token::IDENTIFIER:
798 if (this->isType(start.fText)) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700799 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700800 if (!decl) {
801 return nullptr;
802 }
803 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
804 std::move(decl)));
805 }
806 // fall through
807 default:
808 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500809 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700810}
811
Ethan Nicholas50afc172017-02-16 14:49:57 -0500812/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* */
ethannicholasb3058bd2016-07-01 08:22:01 -0700813std::unique_ptr<ASTType> Parser::type() {
814 Token type;
815 if (!this->expect(Token::IDENTIFIER, "a type", &type)) {
816 return nullptr;
817 }
818 if (!this->isType(type.fText)) {
819 this->error(type.fPosition, ("no type named '" + type.fText + "'").c_str());
820 return nullptr;
821 }
Ethan Nicholas50afc172017-02-16 14:49:57 -0500822 std::vector<int> sizes;
823 while (this->peek().fKind == Token::LBRACKET) {
824 this->expect(Token::LBRACKET, "'['");
825 if (this->peek().fKind != Token::RBRACKET) {
826 int64_t i;
827 if (this->intLiteral(&i)) {
828 sizes.push_back(i);
829 } else {
830 return nullptr;
831 }
832 } else {
833 sizes.push_back(-1);
834 }
835 this->expect(Token::RBRACKET, "']'");
836 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500837 return std::unique_ptr<ASTType>(new ASTType(type.fPosition, std::move(type.fText),
Ethan Nicholas50afc172017-02-16 14:49:57 -0500838 ASTType::kIdentifier_Kind, sizes));
ethannicholasb3058bd2016-07-01 08:22:01 -0700839}
840
Ethan Nicholas50afc172017-02-16 14:49:57 -0500841/* IDENTIFIER LBRACE varDeclaration* RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500842std::unique_ptr<ASTDeclaration> Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700843 Token name;
844 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
845 return nullptr;
846 }
847 if (peek().fKind != Token::LBRACE) {
848 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -0500849 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -0700850 // it's better to report it as an unknown type
851 this->error(name.fPosition, "no type named '" + name.fText + "'");
852 return nullptr;
853 }
854 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500855 std::vector<std::unique_ptr<ASTVarDeclarations>> decls;
ethannicholasb3058bd2016-07-01 08:22:01 -0700856 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700857 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700858 if (!decl) {
859 return nullptr;
860 }
861 decls.push_back(std::move(decl));
862 }
863 this->nextToken();
Ethan Nicholas50afc172017-02-16 14:49:57 -0500864 SkString instanceName;
865 std::vector<std::unique_ptr<ASTExpression>> sizes;
ethannicholasb3058bd2016-07-01 08:22:01 -0700866 if (this->peek().fKind == Token::IDENTIFIER) {
Ethan Nicholas50afc172017-02-16 14:49:57 -0500867 instanceName = this->nextToken().fText;
868 while (this->peek().fKind == Token::LBRACKET) {
869 this->expect(Token::LBRACKET, "'['");
870 if (this->peek().fKind != Token::RBRACKET) {
871 std::unique_ptr<ASTExpression> size = this->expression();
872 if (!size) {
873 return nullptr;
874 }
875 sizes.push_back(std::move(size));
876 } else {
877 sizes.push_back(nullptr);
878 }
879 this->expect(Token::RBRACKET, "']'");
880 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700881 }
882 this->expect(Token::SEMICOLON, "';'");
Ethan Nicholas11d53972016-11-28 11:23:23 -0500883 return std::unique_ptr<ASTDeclaration>(new ASTInterfaceBlock(name.fPosition, mods,
Ethan Nicholas50afc172017-02-16 14:49:57 -0500884 name.fText, std::move(decls),
885 std::move(instanceName),
886 std::move(sizes)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700887}
888
889/* IF LPAREN expression RPAREN statement (ELSE statement)? */
890std::unique_ptr<ASTIfStatement> Parser::ifStatement() {
891 Token start;
892 if (!this->expect(Token::IF, "'if'", &start)) {
893 return nullptr;
894 }
895 if (!this->expect(Token::LPAREN, "'('")) {
896 return nullptr;
897 }
898 std::unique_ptr<ASTExpression> test(this->expression());
899 if (!test) {
900 return nullptr;
901 }
902 if (!this->expect(Token::RPAREN, "')'")) {
903 return nullptr;
904 }
905 std::unique_ptr<ASTStatement> ifTrue(this->statement());
906 if (!ifTrue) {
907 return nullptr;
908 }
909 std::unique_ptr<ASTStatement> ifFalse;
910 if (this->peek().fKind == Token::ELSE) {
911 this->nextToken();
912 ifFalse = this->statement();
913 if (!ifFalse) {
914 return nullptr;
915 }
916 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500917 return std::unique_ptr<ASTIfStatement>(new ASTIfStatement(start.fPosition, std::move(test),
918 std::move(ifTrue),
ethannicholasb3058bd2016-07-01 08:22:01 -0700919 std::move(ifFalse)));
920}
921
922/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
923std::unique_ptr<ASTDoStatement> Parser::doStatement() {
924 Token start;
925 if (!this->expect(Token::DO, "'do'", &start)) {
926 return nullptr;
927 }
928 std::unique_ptr<ASTStatement> statement(this->statement());
929 if (!statement) {
930 return nullptr;
931 }
932 if (!this->expect(Token::WHILE, "'while'")) {
933 return nullptr;
934 }
935 if (!this->expect(Token::LPAREN, "'('")) {
936 return nullptr;
937 }
938 std::unique_ptr<ASTExpression> test(this->expression());
939 if (!test) {
940 return nullptr;
941 }
942 if (!this->expect(Token::RPAREN, "')'")) {
943 return nullptr;
944 }
945 if (!this->expect(Token::SEMICOLON, "';'")) {
946 return nullptr;
947 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500948 return std::unique_ptr<ASTDoStatement>(new ASTDoStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700949 std::move(statement),
950 std::move(test)));
951}
952
953/* WHILE LPAREN expression RPAREN STATEMENT */
954std::unique_ptr<ASTWhileStatement> Parser::whileStatement() {
955 Token start;
956 if (!this->expect(Token::WHILE, "'while'", &start)) {
957 return nullptr;
958 }
959 if (!this->expect(Token::LPAREN, "'('")) {
960 return nullptr;
961 }
962 std::unique_ptr<ASTExpression> test(this->expression());
963 if (!test) {
964 return nullptr;
965 }
966 if (!this->expect(Token::RPAREN, "')'")) {
967 return nullptr;
968 }
969 std::unique_ptr<ASTStatement> statement(this->statement());
970 if (!statement) {
971 return nullptr;
972 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500973 return std::unique_ptr<ASTWhileStatement>(new ASTWhileStatement(start.fPosition,
974 std::move(test),
ethannicholasb3058bd2016-07-01 08:22:01 -0700975 std::move(statement)));
976}
977
Ethan Nicholas11d53972016-11-28 11:23:23 -0500978/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -0700979 STATEMENT */
980std::unique_ptr<ASTForStatement> Parser::forStatement() {
981 Token start;
982 if (!this->expect(Token::FOR, "'for'", &start)) {
983 return nullptr;
984 }
985 if (!this->expect(Token::LPAREN, "'('")) {
986 return nullptr;
987 }
988 std::unique_ptr<ASTStatement> initializer;
989 Token nextToken = this->peek();
990 switch (nextToken.fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500991 case Token::SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -0700992 this->nextToken();
ethannicholasb3058bd2016-07-01 08:22:01 -0700993 break;
ethannicholasa54401d2016-10-14 08:37:32 -0700994 case Token::CONST: {
995 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
996 if (!vd) {
997 return nullptr;
998 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700999 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001000 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001001 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001002 }
1003 case Token::IDENTIFIER: {
ethannicholasb3058bd2016-07-01 08:22:01 -07001004 if (this->isType(nextToken.fText)) {
ethannicholasa54401d2016-10-14 08:37:32 -07001005 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
1006 if (!vd) {
1007 return nullptr;
1008 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001009 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001010 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001011 break;
1012 }
ethannicholasa54401d2016-10-14 08:37:32 -07001013 } // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001014 default:
1015 initializer = this->expressionStatement();
1016 }
1017 std::unique_ptr<ASTExpression> test;
1018 if (this->peek().fKind != Token::SEMICOLON) {
1019 test = this->expression();
1020 if (!test) {
1021 return nullptr;
1022 }
1023 }
1024 if (!this->expect(Token::SEMICOLON, "';'")) {
1025 return nullptr;
1026 }
1027 std::unique_ptr<ASTExpression> next;
ethannicholas22f939e2016-10-13 13:25:34 -07001028 if (this->peek().fKind != Token::RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001029 next = this->expression();
1030 if (!next) {
1031 return nullptr;
1032 }
1033 }
1034 if (!this->expect(Token::RPAREN, "')'")) {
1035 return nullptr;
1036 }
1037 std::unique_ptr<ASTStatement> statement(this->statement());
1038 if (!statement) {
1039 return nullptr;
1040 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001041 return std::unique_ptr<ASTForStatement>(new ASTForStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001042 std::move(initializer),
1043 std::move(test), std::move(next),
1044 std::move(statement)));
1045}
1046
1047/* RETURN expression? SEMICOLON */
1048std::unique_ptr<ASTReturnStatement> Parser::returnStatement() {
1049 Token start;
1050 if (!this->expect(Token::RETURN, "'return'", &start)) {
1051 return nullptr;
1052 }
1053 std::unique_ptr<ASTExpression> expression;
1054 if (this->peek().fKind != Token::SEMICOLON) {
1055 expression = this->expression();
1056 if (!expression) {
1057 return nullptr;
1058 }
1059 }
1060 if (!this->expect(Token::SEMICOLON, "';'")) {
1061 return nullptr;
1062 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001063 return std::unique_ptr<ASTReturnStatement>(new ASTReturnStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001064 std::move(expression)));
1065}
1066
1067/* BREAK SEMICOLON */
1068std::unique_ptr<ASTBreakStatement> Parser::breakStatement() {
1069 Token start;
1070 if (!this->expect(Token::BREAK, "'break'", &start)) {
1071 return nullptr;
1072 }
1073 if (!this->expect(Token::SEMICOLON, "';'")) {
1074 return nullptr;
1075 }
1076 return std::unique_ptr<ASTBreakStatement>(new ASTBreakStatement(start.fPosition));
1077}
1078
1079/* CONTINUE SEMICOLON */
1080std::unique_ptr<ASTContinueStatement> Parser::continueStatement() {
1081 Token start;
1082 if (!this->expect(Token::CONTINUE, "'continue'", &start)) {
1083 return nullptr;
1084 }
1085 if (!this->expect(Token::SEMICOLON, "';'")) {
1086 return nullptr;
1087 }
1088 return std::unique_ptr<ASTContinueStatement>(new ASTContinueStatement(start.fPosition));
1089}
1090
1091/* DISCARD SEMICOLON */
1092std::unique_ptr<ASTDiscardStatement> Parser::discardStatement() {
1093 Token start;
1094 if (!this->expect(Token::DISCARD, "'continue'", &start)) {
1095 return nullptr;
1096 }
1097 if (!this->expect(Token::SEMICOLON, "';'")) {
1098 return nullptr;
1099 }
1100 return std::unique_ptr<ASTDiscardStatement>(new ASTDiscardStatement(start.fPosition));
1101}
1102
1103/* LBRACE statement* RBRACE */
1104std::unique_ptr<ASTBlock> Parser::block() {
ethannicholascad64162016-10-27 10:54:02 -07001105 AutoDepth depth(this);
1106 if (!depth.checkValid()) {
1107 return nullptr;
1108 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001109 Token start;
1110 if (!this->expect(Token::LBRACE, "'{'", &start)) {
1111 return nullptr;
1112 }
1113 std::vector<std::unique_ptr<ASTStatement>> statements;
1114 for (;;) {
1115 switch (this->peek().fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001116 case Token::RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001117 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001118 return std::unique_ptr<ASTBlock>(new ASTBlock(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001119 std::move(statements)));
Ethan Nicholas11d53972016-11-28 11:23:23 -05001120 case Token::END_OF_FILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001121 this->error(this->peek().fPosition, "expected '}', but found end of file");
1122 return nullptr;
1123 default: {
1124 std::unique_ptr<ASTStatement> statement = this->statement();
1125 if (!statement) {
1126 return nullptr;
1127 }
1128 statements.push_back(std::move(statement));
1129 }
1130 }
1131 }
1132}
1133
1134/* expression SEMICOLON */
1135std::unique_ptr<ASTExpressionStatement> Parser::expressionStatement() {
1136 std::unique_ptr<ASTExpression> expr = this->expression();
1137 if (expr) {
1138 if (this->expect(Token::SEMICOLON, "';'")) {
1139 ASTExpressionStatement* result = new ASTExpressionStatement(std::move(expr));
1140 return std::unique_ptr<ASTExpressionStatement>(result);
1141 }
1142 }
1143 return nullptr;
1144}
1145
1146/* assignmentExpression */
1147std::unique_ptr<ASTExpression> Parser::expression() {
ethannicholascad64162016-10-27 10:54:02 -07001148 AutoDepth depth(this);
1149 if (!depth.checkValid()) {
1150 return nullptr;
1151 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001152 return this->assignmentExpression();
1153}
1154
1155/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1156 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1157 assignmentExpression)*
1158 */
1159std::unique_ptr<ASTExpression> Parser::assignmentExpression() {
1160 std::unique_ptr<ASTExpression> result = this->ternaryExpression();
1161 if (!result) {
1162 return nullptr;
1163 }
1164 for (;;) {
1165 switch (this->peek().fKind) {
1166 case Token::EQ: // fall through
1167 case Token::STAREQ: // fall through
1168 case Token::SLASHEQ: // fall through
1169 case Token::PERCENTEQ: // fall through
1170 case Token::PLUSEQ: // fall through
1171 case Token::MINUSEQ: // fall through
1172 case Token::SHLEQ: // fall through
1173 case Token::SHREQ: // fall through
1174 case Token::BITWISEANDEQ: // fall through
1175 case Token::BITWISEXOREQ: // fall through
1176 case Token::BITWISEOREQ: // fall through
1177 case Token::LOGICALANDEQ: // fall through
1178 case Token::LOGICALXOREQ: // fall through
1179 case Token::LOGICALOREQ: {
1180 Token t = this->nextToken();
1181 std::unique_ptr<ASTExpression> right = this->assignmentExpression();
1182 if (!right) {
1183 return nullptr;
1184 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001185 result = std::unique_ptr<ASTExpression>(new ASTBinaryExpression(std::move(result),
1186 t,
ethannicholasb3058bd2016-07-01 08:22:01 -07001187 std::move(right)));
1188 }
1189 default:
1190 return result;
1191 }
1192 }
1193}
1194
1195/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
1196std::unique_ptr<ASTExpression> Parser::ternaryExpression() {
1197 std::unique_ptr<ASTExpression> result = this->logicalOrExpression();
1198 if (!result) {
1199 return nullptr;
1200 }
1201 if (this->peek().fKind == Token::QUESTION) {
1202 Token question = this->nextToken();
1203 std::unique_ptr<ASTExpression> trueExpr = this->expression();
1204 if (!trueExpr) {
1205 return nullptr;
1206 }
1207 if (this->expect(Token::COLON, "':'")) {
1208 std::unique_ptr<ASTExpression> falseExpr = this->assignmentExpression();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001209 return std::unique_ptr<ASTExpression>(new ASTTernaryExpression(std::move(result),
1210 std::move(trueExpr),
ethannicholasb3058bd2016-07-01 08:22:01 -07001211 std::move(falseExpr)));
1212 }
1213 return nullptr;
1214 }
1215 return result;
1216}
1217
1218/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
1219std::unique_ptr<ASTExpression> Parser::logicalOrExpression() {
1220 std::unique_ptr<ASTExpression> result = this->logicalXorExpression();
1221 if (!result) {
1222 return nullptr;
1223 }
1224 while (this->peek().fKind == Token::LOGICALOR) {
1225 Token t = this->nextToken();
1226 std::unique_ptr<ASTExpression> right = this->logicalXorExpression();
1227 if (!right) {
1228 return nullptr;
1229 }
1230 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1231 }
1232 return result;
1233}
1234
1235/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
1236std::unique_ptr<ASTExpression> Parser::logicalXorExpression() {
1237 std::unique_ptr<ASTExpression> result = this->logicalAndExpression();
1238 if (!result) {
1239 return nullptr;
1240 }
1241 while (this->peek().fKind == Token::LOGICALXOR) {
1242 Token t = this->nextToken();
1243 std::unique_ptr<ASTExpression> right = this->logicalAndExpression();
1244 if (!right) {
1245 return nullptr;
1246 }
1247 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1248 }
1249 return result;
1250}
1251
1252/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
1253std::unique_ptr<ASTExpression> Parser::logicalAndExpression() {
1254 std::unique_ptr<ASTExpression> result = this->bitwiseOrExpression();
1255 if (!result) {
1256 return nullptr;
1257 }
1258 while (this->peek().fKind == Token::LOGICALAND) {
1259 Token t = this->nextToken();
1260 std::unique_ptr<ASTExpression> right = this->bitwiseOrExpression();
1261 if (!right) {
1262 return nullptr;
1263 }
1264 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1265 }
1266 return result;
1267}
1268
1269/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
1270std::unique_ptr<ASTExpression> Parser::bitwiseOrExpression() {
1271 std::unique_ptr<ASTExpression> result = this->bitwiseXorExpression();
1272 if (!result) {
1273 return nullptr;
1274 }
1275 while (this->peek().fKind == Token::BITWISEOR) {
1276 Token t = this->nextToken();
1277 std::unique_ptr<ASTExpression> right = this->bitwiseXorExpression();
1278 if (!right) {
1279 return nullptr;
1280 }
1281 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1282 }
1283 return result;
1284}
1285
1286/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
1287std::unique_ptr<ASTExpression> Parser::bitwiseXorExpression() {
1288 std::unique_ptr<ASTExpression> result = this->bitwiseAndExpression();
1289 if (!result) {
1290 return nullptr;
1291 }
1292 while (this->peek().fKind == Token::BITWISEXOR) {
1293 Token t = this->nextToken();
1294 std::unique_ptr<ASTExpression> right = this->bitwiseAndExpression();
1295 if (!right) {
1296 return nullptr;
1297 }
1298 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1299 }
1300 return result;
1301}
1302
1303/* equalityExpression (BITWISEAND equalityExpression)* */
1304std::unique_ptr<ASTExpression> Parser::bitwiseAndExpression() {
1305 std::unique_ptr<ASTExpression> result = this->equalityExpression();
1306 if (!result) {
1307 return nullptr;
1308 }
1309 while (this->peek().fKind == Token::BITWISEAND) {
1310 Token t = this->nextToken();
1311 std::unique_ptr<ASTExpression> right = this->equalityExpression();
1312 if (!right) {
1313 return nullptr;
1314 }
1315 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1316 }
1317 return result;
1318}
1319
1320/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
1321std::unique_ptr<ASTExpression> Parser::equalityExpression() {
1322 std::unique_ptr<ASTExpression> result = this->relationalExpression();
1323 if (!result) {
1324 return nullptr;
1325 }
1326 for (;;) {
1327 switch (this->peek().fKind) {
1328 case Token::EQEQ: // fall through
1329 case Token::NEQ: {
1330 Token t = this->nextToken();
1331 std::unique_ptr<ASTExpression> right = this->relationalExpression();
1332 if (!right) {
1333 return nullptr;
1334 }
1335 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1336 break;
1337 }
1338 default:
1339 return result;
1340 }
1341 }
1342}
1343
1344/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
1345std::unique_ptr<ASTExpression> Parser::relationalExpression() {
1346 std::unique_ptr<ASTExpression> result = this->shiftExpression();
1347 if (!result) {
1348 return nullptr;
1349 }
1350 for (;;) {
1351 switch (this->peek().fKind) {
1352 case Token::LT: // fall through
1353 case Token::GT: // fall through
1354 case Token::LTEQ: // fall through
1355 case Token::GTEQ: {
1356 Token t = this->nextToken();
1357 std::unique_ptr<ASTExpression> right = this->shiftExpression();
1358 if (!right) {
1359 return nullptr;
1360 }
1361 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1362 break;
1363 }
1364 default:
1365 return result;
1366 }
1367 }
1368}
1369
1370/* additiveExpression ((SHL | SHR) additiveExpression)* */
1371std::unique_ptr<ASTExpression> Parser::shiftExpression() {
1372 std::unique_ptr<ASTExpression> result = this->additiveExpression();
1373 if (!result) {
1374 return nullptr;
1375 }
1376 for (;;) {
1377 switch (this->peek().fKind) {
1378 case Token::SHL: // fall through
1379 case Token::SHR: {
1380 Token t = this->nextToken();
1381 std::unique_ptr<ASTExpression> right = this->additiveExpression();
1382 if (!right) {
1383 return nullptr;
1384 }
1385 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1386 break;
1387 }
1388 default:
1389 return result;
1390 }
1391 }
1392}
1393
1394/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
1395std::unique_ptr<ASTExpression> Parser::additiveExpression() {
1396 std::unique_ptr<ASTExpression> result = this->multiplicativeExpression();
1397 if (!result) {
1398 return nullptr;
1399 }
1400 for (;;) {
1401 switch (this->peek().fKind) {
1402 case Token::PLUS: // fall through
1403 case Token::MINUS: {
1404 Token t = this->nextToken();
1405 std::unique_ptr<ASTExpression> right = this->multiplicativeExpression();
1406 if (!right) {
1407 return nullptr;
1408 }
1409 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1410 break;
1411 }
1412 default:
1413 return result;
1414 }
1415 }
1416}
1417
1418/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
1419std::unique_ptr<ASTExpression> Parser::multiplicativeExpression() {
1420 std::unique_ptr<ASTExpression> result = this->unaryExpression();
1421 if (!result) {
1422 return nullptr;
1423 }
1424 for (;;) {
1425 switch (this->peek().fKind) {
1426 case Token::STAR: // fall through
1427 case Token::SLASH: // fall through
1428 case Token::PERCENT: {
1429 Token t = this->nextToken();
1430 std::unique_ptr<ASTExpression> right = this->unaryExpression();
1431 if (!right) {
1432 return nullptr;
1433 }
1434 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1435 break;
1436 }
1437 default:
1438 return result;
1439 }
1440 }
1441}
1442
1443/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
1444std::unique_ptr<ASTExpression> Parser::unaryExpression() {
1445 switch (this->peek().fKind) {
ethannicholas5961bc92016-10-12 06:39:56 -07001446 case Token::PLUS: // fall through
1447 case Token::MINUS: // fall through
1448 case Token::LOGICALNOT: // fall through
1449 case Token::BITWISENOT: // fall through
1450 case Token::PLUSPLUS: // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001451 case Token::MINUSMINUS: {
1452 Token t = this->nextToken();
1453 std::unique_ptr<ASTExpression> expr = this->unaryExpression();
1454 if (!expr) {
1455 return nullptr;
1456 }
1457 return std::unique_ptr<ASTExpression>(new ASTPrefixExpression(t, std::move(expr)));
1458 }
1459 default:
1460 return this->postfixExpression();
1461 }
1462}
1463
1464/* term suffix* */
1465std::unique_ptr<ASTExpression> Parser::postfixExpression() {
1466 std::unique_ptr<ASTExpression> result = this->term();
1467 if (!result) {
1468 return nullptr;
1469 }
1470 for (;;) {
1471 switch (this->peek().fKind) {
1472 case Token::LBRACKET: // fall through
1473 case Token::DOT: // fall through
1474 case Token::LPAREN: // fall through
1475 case Token::PLUSPLUS: // fall through
1476 case Token::MINUSMINUS: {
1477 std::unique_ptr<ASTSuffix> s = this->suffix();
1478 if (!s) {
1479 return nullptr;
1480 }
1481 result.reset(new ASTSuffixExpression(std::move(result), std::move(s)));
1482 break;
1483 }
1484 default:
1485 return result;
1486 }
1487 }
1488}
1489
Ethan Nicholas11d53972016-11-28 11:23:23 -05001490/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
ethannicholasb3058bd2016-07-01 08:22:01 -07001491 PLUSPLUS | MINUSMINUS */
1492std::unique_ptr<ASTSuffix> Parser::suffix() {
1493 Token next = this->nextToken();
1494 switch (next.fKind) {
1495 case Token::LBRACKET: {
ethannicholas5961bc92016-10-12 06:39:56 -07001496 if (this->peek().fKind == Token::RBRACKET) {
1497 this->nextToken();
1498 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(next.fPosition));
1499 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001500 std::unique_ptr<ASTExpression> e = this->expression();
1501 if (!e) {
1502 return nullptr;
1503 }
1504 this->expect(Token::RBRACKET, "']' to complete array access expression");
1505 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(std::move(e)));
1506 }
1507 case Token::DOT: {
1508 Position pos = this->peek().fPosition;
Ethan Nicholas9e1138d2016-11-21 10:39:35 -05001509 SkString text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001510 if (this->identifier(&text)) {
1511 return std::unique_ptr<ASTSuffix>(new ASTFieldSuffix(pos, std::move(text)));
1512 }
1513 return nullptr;
1514 }
1515 case Token::LPAREN: {
1516 std::vector<std::unique_ptr<ASTExpression>> parameters;
1517 if (this->peek().fKind != Token::RPAREN) {
1518 for (;;) {
1519 std::unique_ptr<ASTExpression> expr = this->expression();
1520 if (!expr) {
1521 return nullptr;
1522 }
1523 parameters.push_back(std::move(expr));
1524 if (this->peek().fKind != Token::COMMA) {
1525 break;
1526 }
1527 this->nextToken();
1528 }
1529 }
1530 this->expect(Token::RPAREN, "')' to complete function parameters");
Ethan Nicholas11d53972016-11-28 11:23:23 -05001531 return std::unique_ptr<ASTSuffix>(new ASTCallSuffix(next.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001532 std::move(parameters)));
1533 }
1534 case Token::PLUSPLUS:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001535 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001536 ASTSuffix::kPostIncrement_Kind));
1537 case Token::MINUSMINUS:
1538 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fPosition,
1539 ASTSuffix::kPostDecrement_Kind));
1540 default: {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001541 this->error(next.fPosition, "expected expression suffix, but found '" + next.fText +
ethannicholasb3058bd2016-07-01 08:22:01 -07001542 "'\n");
1543 return nullptr;
1544 }
1545 }
1546}
1547
1548/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | '(' expression ')' */
1549std::unique_ptr<ASTExpression> Parser::term() {
1550 std::unique_ptr<ASTExpression> result;
1551 Token t = this->peek();
1552 switch (t.fKind) {
1553 case Token::IDENTIFIER: {
Ethan Nicholas9e1138d2016-11-21 10:39:35 -05001554 SkString text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001555 if (this->identifier(&text)) {
1556 result.reset(new ASTIdentifier(t.fPosition, std::move(text)));
1557 }
1558 break;
1559 }
1560 case Token::INT_LITERAL: {
1561 int64_t i;
1562 if (this->intLiteral(&i)) {
1563 result.reset(new ASTIntLiteral(t.fPosition, i));
1564 }
1565 break;
1566 }
1567 case Token::FLOAT_LITERAL: {
1568 double f;
1569 if (this->floatLiteral(&f)) {
1570 result.reset(new ASTFloatLiteral(t.fPosition, f));
1571 }
1572 break;
1573 }
1574 case Token::TRUE_LITERAL: // fall through
1575 case Token::FALSE_LITERAL: {
1576 bool b;
1577 if (this->boolLiteral(&b)) {
1578 result.reset(new ASTBoolLiteral(t.fPosition, b));
1579 }
1580 break;
1581 }
1582 case Token::LPAREN: {
1583 this->nextToken();
1584 result = this->expression();
1585 if (result) {
1586 this->expect(Token::RPAREN, "')' to complete expression");
1587 }
1588 break;
1589 }
1590 default:
1591 this->nextToken();
1592 this->error(t.fPosition, "expected expression, but found '" + t.fText + "'\n");
1593 result = nullptr;
1594 }
1595 return result;
1596}
1597
1598/* INT_LITERAL */
1599bool Parser::intLiteral(int64_t* dest) {
1600 Token t;
1601 if (this->expect(Token::INT_LITERAL, "integer literal", &t)) {
1602 *dest = SkSL::stol(t.fText);
1603 return true;
1604 }
1605 return false;
1606}
1607
1608/* FLOAT_LITERAL */
1609bool Parser::floatLiteral(double* dest) {
1610 Token t;
1611 if (this->expect(Token::FLOAT_LITERAL, "float literal", &t)) {
1612 *dest = SkSL::stod(t.fText);
1613 return true;
1614 }
1615 return false;
1616}
1617
1618/* TRUE_LITERAL | FALSE_LITERAL */
1619bool Parser::boolLiteral(bool* dest) {
1620 Token t = this->nextToken();
1621 switch (t.fKind) {
1622 case Token::TRUE_LITERAL:
1623 *dest = true;
1624 return true;
1625 case Token::FALSE_LITERAL:
1626 *dest = false;
1627 return true;
1628 default:
1629 this->error(t.fPosition, "expected 'true' or 'false', but found '" + t.fText + "'\n");
1630 return false;
1631 }
1632}
1633
1634/* IDENTIFIER */
Ethan Nicholas9e1138d2016-11-21 10:39:35 -05001635bool Parser::identifier(SkString* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001636 Token t;
1637 if (this->expect(Token::IDENTIFIER, "identifier", &t)) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001638 *dest = t.fText;
ethannicholasb3058bd2016-07-01 08:22:01 -07001639 return true;
1640 }
1641 return false;
1642}
1643
1644} // namespace