blob: cdbf25e0d58475aeda974a41002ce047c03ed647 [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
Ethan Nicholasbfe15f62017-03-01 11:46:51 -050013#include "disable_flex_warnings.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070014#include "lex.sksl.c"
Ethan Nicholasbfe15f62017-03-01 11:46:51 -050015#undef register
ethannicholasb3058bd2016-07-01 08:22:01 -070016#ifdef __clang__
17#pragma clang diagnostic pop
18#endif
ethannicholasf789b382016-08-03 12:43:36 -070019#ifdef __GNUC__
20#pragma GCC diagnostic pop
21#endif
22#ifdef _MSC_VER
23#pragma warning(pop)
24#endif
ethannicholasb3058bd2016-07-01 08:22:01 -070025
Ethan Nicholasbfe15f62017-03-01 11:46:51 -050026#include "lex.layout.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070027#include "ast/SkSLASTBinaryExpression.h"
28#include "ast/SkSLASTBlock.h"
29#include "ast/SkSLASTBoolLiteral.h"
30#include "ast/SkSLASTBreakStatement.h"
31#include "ast/SkSLASTCallSuffix.h"
32#include "ast/SkSLASTContinueStatement.h"
33#include "ast/SkSLASTDiscardStatement.h"
34#include "ast/SkSLASTDoStatement.h"
35#include "ast/SkSLASTExpression.h"
36#include "ast/SkSLASTExpressionStatement.h"
37#include "ast/SkSLASTExtension.h"
38#include "ast/SkSLASTFieldSuffix.h"
39#include "ast/SkSLASTFloatLiteral.h"
40#include "ast/SkSLASTForStatement.h"
41#include "ast/SkSLASTFunction.h"
42#include "ast/SkSLASTIdentifier.h"
43#include "ast/SkSLASTIfStatement.h"
44#include "ast/SkSLASTIndexSuffix.h"
45#include "ast/SkSLASTInterfaceBlock.h"
46#include "ast/SkSLASTIntLiteral.h"
ethannicholas5961bc92016-10-12 06:39:56 -070047#include "ast/SkSLASTModifiersDeclaration.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070048#include "ast/SkSLASTParameter.h"
ethannicholas5961bc92016-10-12 06:39:56 -070049#include "ast/SkSLASTPrecision.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070050#include "ast/SkSLASTPrefixExpression.h"
51#include "ast/SkSLASTReturnStatement.h"
52#include "ast/SkSLASTStatement.h"
53#include "ast/SkSLASTSuffixExpression.h"
Ethan Nicholasaf197692017-02-27 13:26:45 -050054#include "ast/SkSLASTSwitchCase.h"
55#include "ast/SkSLASTSwitchStatement.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070056#include "ast/SkSLASTTernaryExpression.h"
57#include "ast/SkSLASTType.h"
58#include "ast/SkSLASTVarDeclaration.h"
59#include "ast/SkSLASTVarDeclarationStatement.h"
60#include "ast/SkSLASTWhileStatement.h"
61#include "ir/SkSLSymbolTable.h"
Ethan Nicholas11d53972016-11-28 11:23:23 -050062#include "ir/SkSLModifiers.h"
ethannicholasd598f792016-07-25 10:08:54 -070063#include "ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070064
65namespace SkSL {
66
ethannicholascad64162016-10-27 10:54:02 -070067#define MAX_PARSE_DEPTH 50
68
69class AutoDepth {
70public:
71 AutoDepth(Parser* p)
72 : fParser(p) {
73 fParser->fDepth++;
74 }
75
76 ~AutoDepth() {
77 fParser->fDepth--;
78 }
79
80 bool checkValid() {
81 if (fParser->fDepth > MAX_PARSE_DEPTH) {
Ethan Nicholas0df1b042017-03-31 13:56:23 -040082 fParser->error(fParser->peek().fPosition, String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070083 return false;
84 }
85 return true;
86 }
87
88private:
89 Parser* fParser;
90};
91
Ethan Nicholas0df1b042017-03-31 13:56:23 -040092Parser::Parser(String text, SymbolTable& types, ErrorReporter& errors)
93: fPushback(Position(-1, -1), Token::INVALID_TOKEN, String())
ethannicholasb3058bd2016-07-01 08:22:01 -070094, fTypes(types)
95, fErrors(errors) {
96 sksllex_init(&fScanner);
Ethan Nicholasbfe15f62017-03-01 11:46:51 -050097 layoutlex_init(&fLayoutScanner);
ethannicholasb3058bd2016-07-01 08:22:01 -070098 fBuffer = sksl_scan_string(text.c_str(), fScanner);
99 skslset_lineno(1, fScanner);
100
101 if (false) {
102 // avoid unused warning
103 yyunput(0, nullptr, fScanner);
104 }
105}
106
107Parser::~Parser() {
108 sksl_delete_buffer(fBuffer, fScanner);
ethannicholas69a7e7c2016-07-01 19:09:27 -0700109 sksllex_destroy(fScanner);
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500110 layoutlex_destroy(fLayoutScanner);
ethannicholasb3058bd2016-07-01 08:22:01 -0700111}
112
113/* (precision | directive | declaration)* END_OF_FILE */
114std::vector<std::unique_ptr<ASTDeclaration>> Parser::file() {
115 std::vector<std::unique_ptr<ASTDeclaration>> result;
116 for (;;) {
117 switch (this->peek().fKind) {
118 case Token::END_OF_FILE:
119 return result;
ethannicholas5961bc92016-10-12 06:39:56 -0700120 case Token::PRECISION: {
121 std::unique_ptr<ASTDeclaration> precision = this->precision();
122 if (precision) {
123 result.push_back(std::move(precision));
124 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700125 break;
ethannicholas5961bc92016-10-12 06:39:56 -0700126 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700127 case Token::DIRECTIVE: {
128 std::unique_ptr<ASTDeclaration> decl = this->directive();
129 if (decl) {
130 result.push_back(std::move(decl));
131 }
132 break;
133 }
134 default: {
135 std::unique_ptr<ASTDeclaration> decl = this->declaration();
136 if (!decl) {
137 continue;
138 }
139 result.push_back(std::move(decl));
140 }
141 }
142 }
143}
144
145Token Parser::nextToken() {
146 if (fPushback.fKind != Token::INVALID_TOKEN) {
147 Token result = fPushback;
148 fPushback.fKind = Token::INVALID_TOKEN;
149 fPushback.fText = "";
150 return result;
151 }
152 int token = sksllex(fScanner);
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400153 String text;
ethannicholas471e8942016-10-28 09:02:46 -0700154 switch ((Token::Kind) token) {
155 case Token::IDENTIFIER: // fall through
156 case Token::INT_LITERAL: // fall through
157 case Token::FLOAT_LITERAL: // fall through
158 case Token::DIRECTIVE:
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400159 text = String(skslget_text(fScanner));
ethannicholas471e8942016-10-28 09:02:46 -0700160 break;
161 default:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500162#ifdef SK_DEBUG
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400163 text = String(skslget_text(fScanner));
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500164#endif
ethannicholas471e8942016-10-28 09:02:46 -0700165 break;
166 }
167 return Token(Position(skslget_lineno(fScanner), -1), (Token::Kind) token, text);
ethannicholasb3058bd2016-07-01 08:22:01 -0700168}
169
170void Parser::pushback(Token t) {
171 ASSERT(fPushback.fKind == Token::INVALID_TOKEN);
172 fPushback = t;
173}
174
175Token Parser::peek() {
176 fPushback = this->nextToken();
177 return fPushback;
178}
179
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500180
181bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400182 return this->expect(kind, String(expected), result);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500183}
184
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400185bool Parser::expect(Token::Kind kind, String expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700186 Token next = this->nextToken();
187 if (next.fKind == kind) {
188 if (result) {
189 *result = next;
190 }
191 return true;
192 } else {
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500193 if (next.fText.size()) {
194 this->error(next.fPosition, "expected " + expected + ", but found '" + next.fText +
195 "'");
196 } else {
197 this->error(next.fPosition, "parse error, recompile in debug mode for details");
198 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700199 return false;
200 }
201}
202
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500203void Parser::error(Position p, const char* msg) {
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400204 this->error(p, String(msg));
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500205}
206
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400207void Parser::error(Position p, String msg) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700208 fErrors.error(p, msg);
209}
210
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400211bool Parser::isType(String name) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700212 return nullptr != fTypes[name];
213}
214
215/* PRECISION (LOWP | MEDIUMP | HIGHP) type SEMICOLON */
ethannicholas5961bc92016-10-12 06:39:56 -0700216std::unique_ptr<ASTDeclaration> Parser::precision() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700217 if (!this->expect(Token::PRECISION, "'precision'")) {
ethannicholas5961bc92016-10-12 06:39:56 -0700218 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700219 }
ethannicholas5961bc92016-10-12 06:39:56 -0700220 Modifiers::Flag result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700221 Token p = this->nextToken();
222 switch (p.fKind) {
ethannicholas5961bc92016-10-12 06:39:56 -0700223 case Token::LOWP:
224 result = Modifiers::kLowp_Flag;
225 break;
226 case Token::MEDIUMP:
227 result = Modifiers::kMediump_Flag;
228 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700229 case Token::HIGHP:
ethannicholas5961bc92016-10-12 06:39:56 -0700230 result = Modifiers::kHighp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700231 break;
232 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500233 this->error(p.fPosition, "expected 'lowp', 'mediump', or 'highp', but found '" +
ethannicholasb3058bd2016-07-01 08:22:01 -0700234 p.fText + "'");
ethannicholas5961bc92016-10-12 06:39:56 -0700235 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700236 }
ethannicholas5961bc92016-10-12 06:39:56 -0700237 // FIXME handle the type
ethannicholasb3058bd2016-07-01 08:22:01 -0700238 if (!this->type()) {
ethannicholas5961bc92016-10-12 06:39:56 -0700239 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700240 }
241 this->expect(Token::SEMICOLON, "';'");
ethannicholas5961bc92016-10-12 06:39:56 -0700242 return std::unique_ptr<ASTDeclaration>(new ASTPrecision(p.fPosition, result));
ethannicholasb3058bd2016-07-01 08:22:01 -0700243}
244
Ethan Nicholas11d53972016-11-28 11:23:23 -0500245/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700246 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
ethannicholasb3058bd2016-07-01 08:22:01 -0700247std::unique_ptr<ASTDeclaration> Parser::directive() {
248 Token start;
249 if (!this->expect(Token::DIRECTIVE, "a directive", &start)) {
250 return nullptr;
251 }
252 if (start.fText == "#version") {
253 this->expect(Token::INT_LITERAL, "a version number");
ethannicholas5961bc92016-10-12 06:39:56 -0700254 Token next = this->peek();
255 if (next.fText == "es" || next.fText == "compatibility") {
256 this->nextToken();
257 }
258 // version is ignored for now; it will eventually become an error when we stop pretending
259 // to be GLSL
ethannicholasb3058bd2016-07-01 08:22:01 -0700260 return nullptr;
261 } else if (start.fText == "#extension") {
262 Token name;
263 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
264 return nullptr;
265 }
266 if (!this->expect(Token::COLON, "':'")) {
267 return nullptr;
268 }
269 // FIXME: need to start paying attention to this token
270 if (!this->expect(Token::IDENTIFIER, "an identifier")) {
271 return nullptr;
272 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500273 return std::unique_ptr<ASTDeclaration>(new ASTExtension(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700274 std::move(name.fText)));
275 } else {
276 this->error(start.fPosition, "unsupported directive '" + start.fText + "'");
277 return nullptr;
278 }
279}
280
Ethan Nicholas11d53972016-11-28 11:23:23 -0500281/* modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700282 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
283std::unique_ptr<ASTDeclaration> Parser::declaration() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500284 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700285 Token lookahead = this->peek();
286 if (lookahead.fKind == Token::IDENTIFIER && !this->isType(lookahead.fText)) {
287 // we have an identifier that's not a type, could be the start of an interface block
288 return this->interfaceBlock(modifiers);
289 }
290 if (lookahead.fKind == Token::STRUCT) {
291 return this->structVarDeclaration(modifiers);
292 }
ethannicholas5961bc92016-10-12 06:39:56 -0700293 if (lookahead.fKind == Token::SEMICOLON) {
294 this->nextToken();
295 return std::unique_ptr<ASTDeclaration>(new ASTModifiersDeclaration(modifiers));
296 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700297 std::unique_ptr<ASTType> type(this->type());
298 if (!type) {
299 return nullptr;
300 }
301 if (type->fKind == ASTType::kStruct_Kind && peek().fKind == Token::SEMICOLON) {
302 this->nextToken();
303 return nullptr;
304 }
305 Token name;
306 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
307 return nullptr;
308 }
309 if (!modifiers.fFlags && this->peek().fKind == Token::LPAREN) {
310 this->nextToken();
311 std::vector<std::unique_ptr<ASTParameter>> parameters;
312 while (this->peek().fKind != Token::RPAREN) {
313 if (parameters.size() > 0) {
314 if (!this->expect(Token::COMMA, "','")) {
315 return nullptr;
316 }
317 }
318 std::unique_ptr<ASTParameter> parameter = this->parameter();
319 if (!parameter) {
320 return nullptr;
321 }
322 parameters.push_back(std::move(parameter));
323 }
324 this->nextToken();
325 std::unique_ptr<ASTBlock> body;
326 if (this->peek().fKind == Token::SEMICOLON) {
327 this->nextToken();
328 } else {
329 body = this->block();
330 if (!body) {
331 return nullptr;
332 }
333 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500334 return std::unique_ptr<ASTDeclaration>(new ASTFunction(name.fPosition, std::move(type),
335 std::move(name.fText),
336 std::move(parameters),
ethannicholasb3058bd2016-07-01 08:22:01 -0700337 std::move(body)));
338 } else {
339 return this->varDeclarationEnd(modifiers, std::move(type), name.fText);
340 }
341}
342
343/* modifiers type IDENTIFIER varDeclarationEnd */
ethannicholas14fe8cc2016-09-07 13:37:16 -0700344std::unique_ptr<ASTVarDeclarations> Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500345 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700346 std::unique_ptr<ASTType> type(this->type());
347 if (!type) {
348 return nullptr;
349 }
350 Token name;
351 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
352 return nullptr;
353 }
354 return this->varDeclarationEnd(modifiers, std::move(type), std::move(name.fText));
355}
356
357/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
358std::unique_ptr<ASTType> Parser::structDeclaration() {
359 if (!this->expect(Token::STRUCT, "'struct'")) {
360 return nullptr;
361 }
362 Token name;
363 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
364 return nullptr;
365 }
366 if (!this->expect(Token::LBRACE, "'{'")) {
367 return nullptr;
368 }
369 std::vector<Type::Field> fields;
370 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700371 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700372 if (!decl) {
373 return nullptr;
374 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700375 for (const auto& var : decl->fVars) {
ethannicholasd598f792016-07-25 10:08:54 -0700376 auto type = (const Type*) fTypes[decl->fType->fName];
ethannicholas14fe8cc2016-09-07 13:37:16 -0700377 for (int i = (int) var.fSizes.size() - 1; i >= 0; i--) {
ethannicholasdd4645b2016-10-14 12:14:46 -0700378 if (!var.fSizes[i] || var.fSizes[i]->fKind != ASTExpression::kInt_Kind) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700379 this->error(decl->fPosition, "array size in struct field must be a constant");
ethannicholasdd4645b2016-10-14 12:14:46 -0700380 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700381 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700382 uint64_t columns = ((ASTIntLiteral&) *var.fSizes[i]).fValue;
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400383 String name = type->name() + "[" + to_string(columns) + "]";
ethannicholasd598f792016-07-25 10:08:54 -0700384 type = new Type(name, Type::kArray_Kind, *type, (int) columns);
385 fTypes.takeOwnership((Type*) type);
ethannicholasb3058bd2016-07-01 08:22:01 -0700386 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700387 fields.push_back(Type::Field(decl->fModifiers, var.fName, type));
388 if (var.fValue) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700389 this->error(decl->fPosition, "initializers are not permitted on struct fields");
390 }
391 }
392 }
393 if (!this->expect(Token::RBRACE, "'}'")) {
394 return nullptr;
395 }
Ethan Nicholas19671772016-11-28 16:30:17 -0500396 fTypes.add(name.fText, std::unique_ptr<Type>(new Type(name.fPosition, name.fText, fields)));
Ethan Nicholas11d53972016-11-28 11:23:23 -0500397 return std::unique_ptr<ASTType>(new ASTType(name.fPosition, name.fText,
Ethan Nicholas50afc172017-02-16 14:49:57 -0500398 ASTType::kStruct_Kind, std::vector<int>()));
ethannicholasb3058bd2016-07-01 08:22:01 -0700399}
400
401/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500402std::unique_ptr<ASTVarDeclarations> Parser::structVarDeclaration(Modifiers modifiers) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700403 std::unique_ptr<ASTType> type = this->structDeclaration();
404 if (!type) {
405 return nullptr;
406 }
407 if (peek().fKind == Token::IDENTIFIER) {
408 Token name = this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500409 std::unique_ptr<ASTVarDeclarations> result = this->varDeclarationEnd(modifiers,
410 std::move(type),
ethannicholas14fe8cc2016-09-07 13:37:16 -0700411 std::move(name.fText));
ethannicholasb3058bd2016-07-01 08:22:01 -0700412 if (result) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700413 for (const auto& var : result->fVars) {
414 if (var.fValue) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500415 this->error(var.fValue->fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700416 "struct variables cannot be initialized");
417 }
418 }
419 }
420 return result;
421 }
422 this->expect(Token::SEMICOLON, "';'");
423 return nullptr;
424}
425
Ethan Nicholas11d53972016-11-28 11:23:23 -0500426/* (LBRACKET expression? RBRACKET)* (EQ expression)? (COMMA IDENTIFER
ethannicholasb3058bd2016-07-01 08:22:01 -0700427 (LBRACKET expression? RBRACKET)* (EQ expression)?)* SEMICOLON */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500428std::unique_ptr<ASTVarDeclarations> Parser::varDeclarationEnd(Modifiers mods,
ethannicholas14fe8cc2016-09-07 13:37:16 -0700429 std::unique_ptr<ASTType> type,
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400430 String name) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700431 std::vector<ASTVarDeclaration> vars;
ethannicholasb3058bd2016-07-01 08:22:01 -0700432 std::vector<std::unique_ptr<ASTExpression>> currentVarSizes;
433 while (this->peek().fKind == Token::LBRACKET) {
434 this->nextToken();
435 if (this->peek().fKind == Token::RBRACKET) {
436 this->nextToken();
437 currentVarSizes.push_back(nullptr);
438 } else {
439 std::unique_ptr<ASTExpression> size(this->expression());
440 if (!size) {
441 return nullptr;
442 }
443 currentVarSizes.push_back(std::move(size));
444 if (!this->expect(Token::RBRACKET, "']'")) {
445 return nullptr;
446 }
447 }
448 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700449 std::unique_ptr<ASTExpression> value;
ethannicholasb3058bd2016-07-01 08:22:01 -0700450 if (this->peek().fKind == Token::EQ) {
451 this->nextToken();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700452 value = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700453 if (!value) {
454 return nullptr;
455 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700456 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700457 vars.emplace_back(std::move(name), std::move(currentVarSizes), std::move(value));
ethannicholasb3058bd2016-07-01 08:22:01 -0700458 while (this->peek().fKind == Token::COMMA) {
459 this->nextToken();
460 Token name;
461 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
462 return nullptr;
463 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700464 currentVarSizes.clear();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700465 value.reset();
ethannicholasb3058bd2016-07-01 08:22:01 -0700466 while (this->peek().fKind == Token::LBRACKET) {
467 this->nextToken();
468 if (this->peek().fKind == Token::RBRACKET) {
469 this->nextToken();
470 currentVarSizes.push_back(nullptr);
471 } else {
472 std::unique_ptr<ASTExpression> size(this->expression());
473 if (!size) {
474 return nullptr;
475 }
476 currentVarSizes.push_back(std::move(size));
477 if (!this->expect(Token::RBRACKET, "']'")) {
478 return nullptr;
479 }
480 }
481 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700482 if (this->peek().fKind == Token::EQ) {
483 this->nextToken();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700484 value = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700485 if (!value) {
486 return nullptr;
487 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700488 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700489 vars.emplace_back(std::move(name.fText), std::move(currentVarSizes), std::move(value));
ethannicholasb3058bd2016-07-01 08:22:01 -0700490 }
491 if (!this->expect(Token::SEMICOLON, "';'")) {
492 return nullptr;
493 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700494 return std::unique_ptr<ASTVarDeclarations>(new ASTVarDeclarations(std::move(mods),
495 std::move(type),
496 std::move(vars)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700497}
498
499/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
500std::unique_ptr<ASTParameter> Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400501 Modifiers modifiers = this->modifiersWithDefaults(0);
ethannicholasb3058bd2016-07-01 08:22:01 -0700502 std::unique_ptr<ASTType> type = this->type();
503 if (!type) {
504 return nullptr;
505 }
506 Token name;
507 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
508 return nullptr;
509 }
510 std::vector<int> sizes;
511 while (this->peek().fKind == Token::LBRACKET) {
512 this->nextToken();
513 Token sizeToken;
514 if (!this->expect(Token::INT_LITERAL, "a positive integer", &sizeToken)) {
515 return nullptr;
516 }
517 sizes.push_back(SkSL::stoi(sizeToken.fText));
518 if (!this->expect(Token::RBRACKET, "']'")) {
519 return nullptr;
520 }
521 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500522 return std::unique_ptr<ASTParameter>(new ASTParameter(name.fPosition, modifiers,
523 std::move(type), name.fText,
ethannicholasb3058bd2016-07-01 08:22:01 -0700524 std::move(sizes)));
525}
526
527/** (EQ INT_LITERAL)? */
528int Parser::layoutInt() {
529 if (!this->expect(Token::EQ, "'='")) {
530 return -1;
531 }
532 Token resultToken;
533 if (this->expect(Token::INT_LITERAL, "a non-negative integer", &resultToken)) {
534 return SkSL::stoi(resultToken.fText);
535 }
536 return -1;
537}
538
ethannicholas8ac838d2016-11-22 08:39:36 -0800539/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500540Layout Parser::layout() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700541 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500542 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700543 int binding = -1;
544 int index = -1;
545 int set = -1;
546 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500547 int inputAttachmentIndex = -1;
ethannicholasf789b382016-08-03 12:43:36 -0700548 bool originUpperLeft = false;
ethannicholas5961bc92016-10-12 06:39:56 -0700549 bool overrideCoverage = false;
550 bool blendSupportAllEquations = false;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500551 Layout::Format format = Layout::Format::kUnspecified;
ethannicholas8ac838d2016-11-22 08:39:36 -0800552 bool pushConstant = false;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500553 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
554 int maxVertices = -1;
555 int invocations = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700556 if (this->peek().fKind == Token::LAYOUT) {
557 this->nextToken();
558 if (!this->expect(Token::LPAREN, "'('")) {
Ethan Nicholas19671772016-11-28 16:30:17 -0500559 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
Ethan Nicholas11d53972016-11-28 11:23:23 -0500560 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas52cad152017-02-16 16:37:32 -0500561 pushConstant, primitive, maxVertices, invocations);
ethannicholasb3058bd2016-07-01 08:22:01 -0700562 }
563 for (;;) {
564 Token t = this->nextToken();
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500565 YY_BUFFER_STATE buffer;
566 buffer = layout_scan_string(t.fText.c_str(), fLayoutScanner);
567 int token = layoutlex(fLayoutScanner);
568 layout_delete_buffer(buffer, fLayoutScanner);
569 if (token != Token::INVALID_TOKEN) {
570 switch (token) {
571 case Token::LOCATION:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500572 location = this->layoutInt();
573 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500574 case Token::OFFSET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500575 offset = this->layoutInt();
576 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500577 case Token::BINDING:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500578 binding = this->layoutInt();
579 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500580 case Token::INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500581 index = this->layoutInt();
582 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500583 case Token::SET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500584 set = this->layoutInt();
585 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500586 case Token::BUILTIN:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500587 builtin = this->layoutInt();
588 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500589 case Token::INPUT_ATTACHMENT_INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500590 inputAttachmentIndex = this->layoutInt();
591 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500592 case Token::ORIGIN_UPPER_LEFT:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500593 originUpperLeft = true;
594 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500595 case Token::OVERRIDE_COVERAGE:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500596 overrideCoverage = true;
597 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500598 case Token::BLEND_SUPPORT_ALL_EQUATIONS:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500599 blendSupportAllEquations = true;
600 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500601 case Token::PUSH_CONSTANT:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500602 pushConstant = true;
603 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500604 case Token::POINTS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500605 primitive = Layout::kPoints_Primitive;
606 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500607 case Token::LINES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500608 primitive = Layout::kLines_Primitive;
609 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500610 case Token::LINE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500611 primitive = Layout::kLineStrip_Primitive;
612 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500613 case Token::LINES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500614 primitive = Layout::kLinesAdjacency_Primitive;
615 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500616 case Token::TRIANGLES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500617 primitive = Layout::kTriangles_Primitive;
618 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500619 case Token::TRIANGLE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500620 primitive = Layout::kTriangleStrip_Primitive;
621 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500622 case Token::TRIANGLES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500623 primitive = Layout::kTrianglesAdjacency_Primitive;
624 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500625 case Token::MAX_VERTICES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500626 maxVertices = this->layoutInt();
627 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500628 case Token::INVOCATIONS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500629 invocations = this->layoutInt();
630 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500631 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500632 } else if (Layout::ReadFormat(t.fText, &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500633 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700634 } else {
Greg Daniel64773e62016-11-22 09:44:03 -0500635 this->error(t.fPosition, ("'" + t.fText +
ethannicholasb3058bd2016-07-01 08:22:01 -0700636 "' is not a valid layout qualifier").c_str());
637 }
638 if (this->peek().fKind == Token::RPAREN) {
639 this->nextToken();
640 break;
641 }
642 if (!this->expect(Token::COMMA, "','")) {
643 break;
644 }
645 }
646 }
Ethan Nicholas19671772016-11-28 16:30:17 -0500647 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
648 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas52cad152017-02-16 16:37:32 -0500649 pushConstant, primitive, maxVertices, invocations);
ethannicholasb3058bd2016-07-01 08:22:01 -0700650}
651
Brian Salomonf9f45122016-11-29 11:59:17 -0500652/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
653 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500654Modifiers Parser::modifiers() {
655 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700656 int flags = 0;
657 for (;;) {
658 // TODO: handle duplicate / incompatible flags
659 switch (peek().fKind) {
660 case Token::UNIFORM:
661 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500662 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700663 break;
664 case Token::CONST:
665 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500666 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700667 break;
668 case Token::IN:
669 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500670 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700671 break;
672 case Token::OUT:
673 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500674 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700675 break;
676 case Token::INOUT:
677 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500678 flags |= Modifiers::kIn_Flag;
679 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700680 break;
681 case Token::LOWP:
682 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500683 flags |= Modifiers::kLowp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700684 break;
685 case Token::MEDIUMP:
686 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500687 flags |= Modifiers::kMediump_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700688 break;
689 case Token::HIGHP:
690 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500691 flags |= Modifiers::kHighp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700692 break;
ethannicholasf789b382016-08-03 12:43:36 -0700693 case Token::FLAT:
694 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500695 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700696 break;
697 case Token::NOPERSPECTIVE:
698 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500699 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700700 break;
Brian Salomonf9f45122016-11-29 11:59:17 -0500701 case Token::READONLY:
702 this->nextToken();
703 flags |= Modifiers::kReadOnly_Flag;
704 break;
705 case Token::WRITEONLY:
706 this->nextToken();
707 flags |= Modifiers::kWriteOnly_Flag;
708 break;
709 case Token::COHERENT:
710 this->nextToken();
711 flags |= Modifiers::kCoherent_Flag;
712 break;
713 case Token::VOLATILE:
714 this->nextToken();
715 flags |= Modifiers::kVolatile_Flag;
716 break;
717 case Token::RESTRICT:
718 this->nextToken();
719 flags |= Modifiers::kRestrict_Flag;
720 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700721 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500722 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700723 }
724 }
725}
726
Ethan Nicholas11d53972016-11-28 11:23:23 -0500727Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
728 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700729 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500730 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700731 }
732 return result;
733}
734
735/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
736std::unique_ptr<ASTStatement> Parser::statement() {
737 Token start = this->peek();
738 switch (start.fKind) {
739 case Token::IF:
740 return this->ifStatement();
741 case Token::FOR:
742 return this->forStatement();
743 case Token::DO:
744 return this->doStatement();
745 case Token::WHILE:
746 return this->whileStatement();
Ethan Nicholasaf197692017-02-27 13:26:45 -0500747 case Token::SWITCH:
748 return this->switchStatement();
ethannicholasb3058bd2016-07-01 08:22:01 -0700749 case Token::RETURN:
750 return this->returnStatement();
751 case Token::BREAK:
752 return this->breakStatement();
753 case Token::CONTINUE:
754 return this->continueStatement();
755 case Token::DISCARD:
756 return this->discardStatement();
757 case Token::LBRACE:
758 return this->block();
759 case Token::SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500760 this->nextToken();
761 return std::unique_ptr<ASTStatement>(new ASTBlock(start.fPosition,
ethannicholas0730be72016-09-01 07:59:02 -0700762 std::vector<std::unique_ptr<ASTStatement>>()));
ethannicholasb3058bd2016-07-01 08:22:01 -0700763 case Token::CONST: // fall through
764 case Token::HIGHP: // fall through
765 case Token::MEDIUMP: // fall through
766 case Token::LOWP: {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700767 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700768 if (!decl) {
769 return nullptr;
770 }
771 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(std::move(decl)));
772 }
773 case Token::IDENTIFIER:
774 if (this->isType(start.fText)) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700775 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700776 if (!decl) {
777 return nullptr;
778 }
779 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
780 std::move(decl)));
781 }
782 // fall through
783 default:
784 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500785 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700786}
787
Ethan Nicholas50afc172017-02-16 14:49:57 -0500788/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* */
ethannicholasb3058bd2016-07-01 08:22:01 -0700789std::unique_ptr<ASTType> Parser::type() {
790 Token type;
791 if (!this->expect(Token::IDENTIFIER, "a type", &type)) {
792 return nullptr;
793 }
794 if (!this->isType(type.fText)) {
795 this->error(type.fPosition, ("no type named '" + type.fText + "'").c_str());
796 return nullptr;
797 }
Ethan Nicholas50afc172017-02-16 14:49:57 -0500798 std::vector<int> sizes;
799 while (this->peek().fKind == Token::LBRACKET) {
800 this->expect(Token::LBRACKET, "'['");
801 if (this->peek().fKind != Token::RBRACKET) {
802 int64_t i;
803 if (this->intLiteral(&i)) {
804 sizes.push_back(i);
805 } else {
806 return nullptr;
807 }
808 } else {
809 sizes.push_back(-1);
810 }
811 this->expect(Token::RBRACKET, "']'");
812 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500813 return std::unique_ptr<ASTType>(new ASTType(type.fPosition, std::move(type.fText),
Ethan Nicholas50afc172017-02-16 14:49:57 -0500814 ASTType::kIdentifier_Kind, sizes));
ethannicholasb3058bd2016-07-01 08:22:01 -0700815}
816
Ethan Nicholas50afc172017-02-16 14:49:57 -0500817/* IDENTIFIER LBRACE varDeclaration* RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500818std::unique_ptr<ASTDeclaration> Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700819 Token name;
820 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
821 return nullptr;
822 }
823 if (peek().fKind != Token::LBRACE) {
824 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -0500825 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -0700826 // it's better to report it as an unknown type
827 this->error(name.fPosition, "no type named '" + name.fText + "'");
828 return nullptr;
829 }
830 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500831 std::vector<std::unique_ptr<ASTVarDeclarations>> decls;
ethannicholasb3058bd2016-07-01 08:22:01 -0700832 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700833 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700834 if (!decl) {
835 return nullptr;
836 }
837 decls.push_back(std::move(decl));
838 }
839 this->nextToken();
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400840 String instanceName;
Ethan Nicholas50afc172017-02-16 14:49:57 -0500841 std::vector<std::unique_ptr<ASTExpression>> sizes;
ethannicholasb3058bd2016-07-01 08:22:01 -0700842 if (this->peek().fKind == Token::IDENTIFIER) {
Ethan Nicholas50afc172017-02-16 14:49:57 -0500843 instanceName = this->nextToken().fText;
844 while (this->peek().fKind == Token::LBRACKET) {
845 this->expect(Token::LBRACKET, "'['");
846 if (this->peek().fKind != Token::RBRACKET) {
847 std::unique_ptr<ASTExpression> size = this->expression();
848 if (!size) {
849 return nullptr;
850 }
851 sizes.push_back(std::move(size));
852 } else {
853 sizes.push_back(nullptr);
854 }
855 this->expect(Token::RBRACKET, "']'");
856 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700857 }
858 this->expect(Token::SEMICOLON, "';'");
Ethan Nicholas11d53972016-11-28 11:23:23 -0500859 return std::unique_ptr<ASTDeclaration>(new ASTInterfaceBlock(name.fPosition, mods,
Ethan Nicholas50afc172017-02-16 14:49:57 -0500860 name.fText, std::move(decls),
861 std::move(instanceName),
862 std::move(sizes)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700863}
864
865/* IF LPAREN expression RPAREN statement (ELSE statement)? */
866std::unique_ptr<ASTIfStatement> Parser::ifStatement() {
867 Token start;
868 if (!this->expect(Token::IF, "'if'", &start)) {
869 return nullptr;
870 }
871 if (!this->expect(Token::LPAREN, "'('")) {
872 return nullptr;
873 }
874 std::unique_ptr<ASTExpression> test(this->expression());
875 if (!test) {
876 return nullptr;
877 }
878 if (!this->expect(Token::RPAREN, "')'")) {
879 return nullptr;
880 }
881 std::unique_ptr<ASTStatement> ifTrue(this->statement());
882 if (!ifTrue) {
883 return nullptr;
884 }
885 std::unique_ptr<ASTStatement> ifFalse;
886 if (this->peek().fKind == Token::ELSE) {
887 this->nextToken();
888 ifFalse = this->statement();
889 if (!ifFalse) {
890 return nullptr;
891 }
892 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500893 return std::unique_ptr<ASTIfStatement>(new ASTIfStatement(start.fPosition, std::move(test),
894 std::move(ifTrue),
ethannicholasb3058bd2016-07-01 08:22:01 -0700895 std::move(ifFalse)));
896}
897
898/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
899std::unique_ptr<ASTDoStatement> Parser::doStatement() {
900 Token start;
901 if (!this->expect(Token::DO, "'do'", &start)) {
902 return nullptr;
903 }
904 std::unique_ptr<ASTStatement> statement(this->statement());
905 if (!statement) {
906 return nullptr;
907 }
908 if (!this->expect(Token::WHILE, "'while'")) {
909 return nullptr;
910 }
911 if (!this->expect(Token::LPAREN, "'('")) {
912 return nullptr;
913 }
914 std::unique_ptr<ASTExpression> test(this->expression());
915 if (!test) {
916 return nullptr;
917 }
918 if (!this->expect(Token::RPAREN, "')'")) {
919 return nullptr;
920 }
921 if (!this->expect(Token::SEMICOLON, "';'")) {
922 return nullptr;
923 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500924 return std::unique_ptr<ASTDoStatement>(new ASTDoStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700925 std::move(statement),
926 std::move(test)));
927}
928
929/* WHILE LPAREN expression RPAREN STATEMENT */
930std::unique_ptr<ASTWhileStatement> Parser::whileStatement() {
931 Token start;
932 if (!this->expect(Token::WHILE, "'while'", &start)) {
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 std::unique_ptr<ASTStatement> statement(this->statement());
946 if (!statement) {
947 return nullptr;
948 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500949 return std::unique_ptr<ASTWhileStatement>(new ASTWhileStatement(start.fPosition,
950 std::move(test),
ethannicholasb3058bd2016-07-01 08:22:01 -0700951 std::move(statement)));
952}
953
Ethan Nicholasaf197692017-02-27 13:26:45 -0500954/* CASE expression COLON statement* */
955std::unique_ptr<ASTSwitchCase> Parser::switchCase() {
956 Token start;
957 if (!this->expect(Token::CASE, "'case'", &start)) {
958 return nullptr;
959 }
960 std::unique_ptr<ASTExpression> value = this->expression();
961 if (!value) {
962 return nullptr;
963 }
964 if (!this->expect(Token::COLON, "':'")) {
965 return nullptr;
966 }
967 std::vector<std::unique_ptr<ASTStatement>> statements;
968 while (this->peek().fKind != Token::RBRACE && this->peek().fKind != Token::CASE &&
969 this->peek().fKind != Token::DEFAULT) {
970 std::unique_ptr<ASTStatement> s = this->statement();
971 if (!s) {
972 return nullptr;
973 }
974 statements.push_back(std::move(s));
975 }
976 return std::unique_ptr<ASTSwitchCase>(new ASTSwitchCase(start.fPosition, std::move(value),
977 std::move(statements)));
978}
979
980/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
981std::unique_ptr<ASTStatement> Parser::switchStatement() {
982 Token start;
983 if (!this->expect(Token::SWITCH, "'switch'", &start)) {
984 return nullptr;
985 }
986 if (!this->expect(Token::LPAREN, "'('")) {
987 return nullptr;
988 }
989 std::unique_ptr<ASTExpression> value(this->expression());
990 if (!value) {
991 return nullptr;
992 }
993 if (!this->expect(Token::RPAREN, "')'")) {
994 return nullptr;
995 }
996 if (!this->expect(Token::LBRACE, "'{'")) {
997 return nullptr;
998 }
999 std::vector<std::unique_ptr<ASTSwitchCase>> cases;
1000 while (this->peek().fKind == Token::CASE) {
1001 std::unique_ptr<ASTSwitchCase> c = this->switchCase();
1002 if (!c) {
1003 return nullptr;
1004 }
1005 cases.push_back(std::move(c));
1006 }
1007 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1008 // parts of the compiler may rely upon this assumption.
1009 if (this->peek().fKind == Token::DEFAULT) {
1010 Token defaultStart;
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001011 ASSERT_RESULT(this->expect(Token::DEFAULT, "'default'", &defaultStart));
Ethan Nicholasaf197692017-02-27 13:26:45 -05001012 if (!this->expect(Token::COLON, "':'")) {
1013 return nullptr;
1014 }
1015 std::vector<std::unique_ptr<ASTStatement>> statements;
1016 while (this->peek().fKind != Token::RBRACE) {
1017 std::unique_ptr<ASTStatement> s = this->statement();
1018 if (!s) {
1019 return nullptr;
1020 }
1021 statements.push_back(std::move(s));
1022 }
1023 cases.emplace_back(new ASTSwitchCase(defaultStart.fPosition, nullptr,
1024 std::move(statements)));
1025 }
1026 if (!this->expect(Token::RBRACE, "'}'")) {
1027 return nullptr;
1028 }
1029 return std::unique_ptr<ASTStatement>(new ASTSwitchStatement(start.fPosition,
1030 std::move(value),
1031 std::move(cases)));
1032}
1033
Ethan Nicholas11d53972016-11-28 11:23:23 -05001034/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001035 STATEMENT */
1036std::unique_ptr<ASTForStatement> Parser::forStatement() {
1037 Token start;
1038 if (!this->expect(Token::FOR, "'for'", &start)) {
1039 return nullptr;
1040 }
1041 if (!this->expect(Token::LPAREN, "'('")) {
1042 return nullptr;
1043 }
1044 std::unique_ptr<ASTStatement> initializer;
1045 Token nextToken = this->peek();
1046 switch (nextToken.fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001047 case Token::SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001048 this->nextToken();
ethannicholasb3058bd2016-07-01 08:22:01 -07001049 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001050 case Token::CONST: {
1051 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
1052 if (!vd) {
1053 return nullptr;
1054 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001055 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001056 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001057 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001058 }
1059 case Token::IDENTIFIER: {
ethannicholasb3058bd2016-07-01 08:22:01 -07001060 if (this->isType(nextToken.fText)) {
ethannicholasa54401d2016-10-14 08:37:32 -07001061 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
1062 if (!vd) {
1063 return nullptr;
1064 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001065 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001066 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001067 break;
1068 }
ethannicholasa54401d2016-10-14 08:37:32 -07001069 } // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001070 default:
1071 initializer = this->expressionStatement();
1072 }
1073 std::unique_ptr<ASTExpression> test;
1074 if (this->peek().fKind != Token::SEMICOLON) {
1075 test = this->expression();
1076 if (!test) {
1077 return nullptr;
1078 }
1079 }
1080 if (!this->expect(Token::SEMICOLON, "';'")) {
1081 return nullptr;
1082 }
1083 std::unique_ptr<ASTExpression> next;
ethannicholas22f939e2016-10-13 13:25:34 -07001084 if (this->peek().fKind != Token::RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001085 next = this->expression();
1086 if (!next) {
1087 return nullptr;
1088 }
1089 }
1090 if (!this->expect(Token::RPAREN, "')'")) {
1091 return nullptr;
1092 }
1093 std::unique_ptr<ASTStatement> statement(this->statement());
1094 if (!statement) {
1095 return nullptr;
1096 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001097 return std::unique_ptr<ASTForStatement>(new ASTForStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001098 std::move(initializer),
1099 std::move(test), std::move(next),
1100 std::move(statement)));
1101}
1102
1103/* RETURN expression? SEMICOLON */
1104std::unique_ptr<ASTReturnStatement> Parser::returnStatement() {
1105 Token start;
1106 if (!this->expect(Token::RETURN, "'return'", &start)) {
1107 return nullptr;
1108 }
1109 std::unique_ptr<ASTExpression> expression;
1110 if (this->peek().fKind != Token::SEMICOLON) {
1111 expression = this->expression();
1112 if (!expression) {
1113 return nullptr;
1114 }
1115 }
1116 if (!this->expect(Token::SEMICOLON, "';'")) {
1117 return nullptr;
1118 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001119 return std::unique_ptr<ASTReturnStatement>(new ASTReturnStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001120 std::move(expression)));
1121}
1122
1123/* BREAK SEMICOLON */
1124std::unique_ptr<ASTBreakStatement> Parser::breakStatement() {
1125 Token start;
1126 if (!this->expect(Token::BREAK, "'break'", &start)) {
1127 return nullptr;
1128 }
1129 if (!this->expect(Token::SEMICOLON, "';'")) {
1130 return nullptr;
1131 }
1132 return std::unique_ptr<ASTBreakStatement>(new ASTBreakStatement(start.fPosition));
1133}
1134
1135/* CONTINUE SEMICOLON */
1136std::unique_ptr<ASTContinueStatement> Parser::continueStatement() {
1137 Token start;
1138 if (!this->expect(Token::CONTINUE, "'continue'", &start)) {
1139 return nullptr;
1140 }
1141 if (!this->expect(Token::SEMICOLON, "';'")) {
1142 return nullptr;
1143 }
1144 return std::unique_ptr<ASTContinueStatement>(new ASTContinueStatement(start.fPosition));
1145}
1146
1147/* DISCARD SEMICOLON */
1148std::unique_ptr<ASTDiscardStatement> Parser::discardStatement() {
1149 Token start;
1150 if (!this->expect(Token::DISCARD, "'continue'", &start)) {
1151 return nullptr;
1152 }
1153 if (!this->expect(Token::SEMICOLON, "';'")) {
1154 return nullptr;
1155 }
1156 return std::unique_ptr<ASTDiscardStatement>(new ASTDiscardStatement(start.fPosition));
1157}
1158
1159/* LBRACE statement* RBRACE */
1160std::unique_ptr<ASTBlock> Parser::block() {
ethannicholascad64162016-10-27 10:54:02 -07001161 AutoDepth depth(this);
1162 if (!depth.checkValid()) {
1163 return nullptr;
1164 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001165 Token start;
1166 if (!this->expect(Token::LBRACE, "'{'", &start)) {
1167 return nullptr;
1168 }
1169 std::vector<std::unique_ptr<ASTStatement>> statements;
1170 for (;;) {
1171 switch (this->peek().fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001172 case Token::RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001173 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001174 return std::unique_ptr<ASTBlock>(new ASTBlock(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001175 std::move(statements)));
Ethan Nicholas11d53972016-11-28 11:23:23 -05001176 case Token::END_OF_FILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001177 this->error(this->peek().fPosition, "expected '}', but found end of file");
1178 return nullptr;
1179 default: {
1180 std::unique_ptr<ASTStatement> statement = this->statement();
1181 if (!statement) {
1182 return nullptr;
1183 }
1184 statements.push_back(std::move(statement));
1185 }
1186 }
1187 }
1188}
1189
1190/* expression SEMICOLON */
1191std::unique_ptr<ASTExpressionStatement> Parser::expressionStatement() {
1192 std::unique_ptr<ASTExpression> expr = this->expression();
1193 if (expr) {
1194 if (this->expect(Token::SEMICOLON, "';'")) {
1195 ASTExpressionStatement* result = new ASTExpressionStatement(std::move(expr));
1196 return std::unique_ptr<ASTExpressionStatement>(result);
1197 }
1198 }
1199 return nullptr;
1200}
1201
1202/* assignmentExpression */
1203std::unique_ptr<ASTExpression> Parser::expression() {
ethannicholascad64162016-10-27 10:54:02 -07001204 AutoDepth depth(this);
1205 if (!depth.checkValid()) {
1206 return nullptr;
1207 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001208 return this->assignmentExpression();
1209}
1210
1211/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1212 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1213 assignmentExpression)*
1214 */
1215std::unique_ptr<ASTExpression> Parser::assignmentExpression() {
1216 std::unique_ptr<ASTExpression> result = this->ternaryExpression();
1217 if (!result) {
1218 return nullptr;
1219 }
1220 for (;;) {
1221 switch (this->peek().fKind) {
1222 case Token::EQ: // fall through
1223 case Token::STAREQ: // fall through
1224 case Token::SLASHEQ: // fall through
1225 case Token::PERCENTEQ: // fall through
1226 case Token::PLUSEQ: // fall through
1227 case Token::MINUSEQ: // fall through
1228 case Token::SHLEQ: // fall through
1229 case Token::SHREQ: // fall through
1230 case Token::BITWISEANDEQ: // fall through
1231 case Token::BITWISEXOREQ: // fall through
1232 case Token::BITWISEOREQ: // fall through
1233 case Token::LOGICALANDEQ: // fall through
1234 case Token::LOGICALXOREQ: // fall through
1235 case Token::LOGICALOREQ: {
1236 Token t = this->nextToken();
1237 std::unique_ptr<ASTExpression> right = this->assignmentExpression();
1238 if (!right) {
1239 return nullptr;
1240 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001241 result = std::unique_ptr<ASTExpression>(new ASTBinaryExpression(std::move(result),
1242 t,
ethannicholasb3058bd2016-07-01 08:22:01 -07001243 std::move(right)));
1244 }
1245 default:
1246 return result;
1247 }
1248 }
1249}
1250
1251/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
1252std::unique_ptr<ASTExpression> Parser::ternaryExpression() {
1253 std::unique_ptr<ASTExpression> result = this->logicalOrExpression();
1254 if (!result) {
1255 return nullptr;
1256 }
1257 if (this->peek().fKind == Token::QUESTION) {
1258 Token question = this->nextToken();
1259 std::unique_ptr<ASTExpression> trueExpr = this->expression();
1260 if (!trueExpr) {
1261 return nullptr;
1262 }
1263 if (this->expect(Token::COLON, "':'")) {
1264 std::unique_ptr<ASTExpression> falseExpr = this->assignmentExpression();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001265 return std::unique_ptr<ASTExpression>(new ASTTernaryExpression(std::move(result),
1266 std::move(trueExpr),
ethannicholasb3058bd2016-07-01 08:22:01 -07001267 std::move(falseExpr)));
1268 }
1269 return nullptr;
1270 }
1271 return result;
1272}
1273
1274/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
1275std::unique_ptr<ASTExpression> Parser::logicalOrExpression() {
1276 std::unique_ptr<ASTExpression> result = this->logicalXorExpression();
1277 if (!result) {
1278 return nullptr;
1279 }
1280 while (this->peek().fKind == Token::LOGICALOR) {
1281 Token t = this->nextToken();
1282 std::unique_ptr<ASTExpression> right = this->logicalXorExpression();
1283 if (!right) {
1284 return nullptr;
1285 }
1286 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1287 }
1288 return result;
1289}
1290
1291/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
1292std::unique_ptr<ASTExpression> Parser::logicalXorExpression() {
1293 std::unique_ptr<ASTExpression> result = this->logicalAndExpression();
1294 if (!result) {
1295 return nullptr;
1296 }
1297 while (this->peek().fKind == Token::LOGICALXOR) {
1298 Token t = this->nextToken();
1299 std::unique_ptr<ASTExpression> right = this->logicalAndExpression();
1300 if (!right) {
1301 return nullptr;
1302 }
1303 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1304 }
1305 return result;
1306}
1307
1308/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
1309std::unique_ptr<ASTExpression> Parser::logicalAndExpression() {
1310 std::unique_ptr<ASTExpression> result = this->bitwiseOrExpression();
1311 if (!result) {
1312 return nullptr;
1313 }
1314 while (this->peek().fKind == Token::LOGICALAND) {
1315 Token t = this->nextToken();
1316 std::unique_ptr<ASTExpression> right = this->bitwiseOrExpression();
1317 if (!right) {
1318 return nullptr;
1319 }
1320 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1321 }
1322 return result;
1323}
1324
1325/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
1326std::unique_ptr<ASTExpression> Parser::bitwiseOrExpression() {
1327 std::unique_ptr<ASTExpression> result = this->bitwiseXorExpression();
1328 if (!result) {
1329 return nullptr;
1330 }
1331 while (this->peek().fKind == Token::BITWISEOR) {
1332 Token t = this->nextToken();
1333 std::unique_ptr<ASTExpression> right = this->bitwiseXorExpression();
1334 if (!right) {
1335 return nullptr;
1336 }
1337 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1338 }
1339 return result;
1340}
1341
1342/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
1343std::unique_ptr<ASTExpression> Parser::bitwiseXorExpression() {
1344 std::unique_ptr<ASTExpression> result = this->bitwiseAndExpression();
1345 if (!result) {
1346 return nullptr;
1347 }
1348 while (this->peek().fKind == Token::BITWISEXOR) {
1349 Token t = this->nextToken();
1350 std::unique_ptr<ASTExpression> right = this->bitwiseAndExpression();
1351 if (!right) {
1352 return nullptr;
1353 }
1354 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1355 }
1356 return result;
1357}
1358
1359/* equalityExpression (BITWISEAND equalityExpression)* */
1360std::unique_ptr<ASTExpression> Parser::bitwiseAndExpression() {
1361 std::unique_ptr<ASTExpression> result = this->equalityExpression();
1362 if (!result) {
1363 return nullptr;
1364 }
1365 while (this->peek().fKind == Token::BITWISEAND) {
1366 Token t = this->nextToken();
1367 std::unique_ptr<ASTExpression> right = this->equalityExpression();
1368 if (!right) {
1369 return nullptr;
1370 }
1371 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1372 }
1373 return result;
1374}
1375
1376/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
1377std::unique_ptr<ASTExpression> Parser::equalityExpression() {
1378 std::unique_ptr<ASTExpression> result = this->relationalExpression();
1379 if (!result) {
1380 return nullptr;
1381 }
1382 for (;;) {
1383 switch (this->peek().fKind) {
1384 case Token::EQEQ: // fall through
1385 case Token::NEQ: {
1386 Token t = this->nextToken();
1387 std::unique_ptr<ASTExpression> right = this->relationalExpression();
1388 if (!right) {
1389 return nullptr;
1390 }
1391 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1392 break;
1393 }
1394 default:
1395 return result;
1396 }
1397 }
1398}
1399
1400/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
1401std::unique_ptr<ASTExpression> Parser::relationalExpression() {
1402 std::unique_ptr<ASTExpression> result = this->shiftExpression();
1403 if (!result) {
1404 return nullptr;
1405 }
1406 for (;;) {
1407 switch (this->peek().fKind) {
1408 case Token::LT: // fall through
1409 case Token::GT: // fall through
1410 case Token::LTEQ: // fall through
1411 case Token::GTEQ: {
1412 Token t = this->nextToken();
1413 std::unique_ptr<ASTExpression> right = this->shiftExpression();
1414 if (!right) {
1415 return nullptr;
1416 }
1417 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1418 break;
1419 }
1420 default:
1421 return result;
1422 }
1423 }
1424}
1425
1426/* additiveExpression ((SHL | SHR) additiveExpression)* */
1427std::unique_ptr<ASTExpression> Parser::shiftExpression() {
1428 std::unique_ptr<ASTExpression> result = this->additiveExpression();
1429 if (!result) {
1430 return nullptr;
1431 }
1432 for (;;) {
1433 switch (this->peek().fKind) {
1434 case Token::SHL: // fall through
1435 case Token::SHR: {
1436 Token t = this->nextToken();
1437 std::unique_ptr<ASTExpression> right = this->additiveExpression();
1438 if (!right) {
1439 return nullptr;
1440 }
1441 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1442 break;
1443 }
1444 default:
1445 return result;
1446 }
1447 }
1448}
1449
1450/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
1451std::unique_ptr<ASTExpression> Parser::additiveExpression() {
1452 std::unique_ptr<ASTExpression> result = this->multiplicativeExpression();
1453 if (!result) {
1454 return nullptr;
1455 }
1456 for (;;) {
1457 switch (this->peek().fKind) {
1458 case Token::PLUS: // fall through
1459 case Token::MINUS: {
1460 Token t = this->nextToken();
1461 std::unique_ptr<ASTExpression> right = this->multiplicativeExpression();
1462 if (!right) {
1463 return nullptr;
1464 }
1465 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1466 break;
1467 }
1468 default:
1469 return result;
1470 }
1471 }
1472}
1473
1474/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
1475std::unique_ptr<ASTExpression> Parser::multiplicativeExpression() {
1476 std::unique_ptr<ASTExpression> result = this->unaryExpression();
1477 if (!result) {
1478 return nullptr;
1479 }
1480 for (;;) {
1481 switch (this->peek().fKind) {
1482 case Token::STAR: // fall through
1483 case Token::SLASH: // fall through
1484 case Token::PERCENT: {
1485 Token t = this->nextToken();
1486 std::unique_ptr<ASTExpression> right = this->unaryExpression();
1487 if (!right) {
1488 return nullptr;
1489 }
1490 result.reset(new ASTBinaryExpression(std::move(result), t, std::move(right)));
1491 break;
1492 }
1493 default:
1494 return result;
1495 }
1496 }
1497}
1498
1499/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
1500std::unique_ptr<ASTExpression> Parser::unaryExpression() {
1501 switch (this->peek().fKind) {
ethannicholas5961bc92016-10-12 06:39:56 -07001502 case Token::PLUS: // fall through
1503 case Token::MINUS: // fall through
1504 case Token::LOGICALNOT: // fall through
1505 case Token::BITWISENOT: // fall through
1506 case Token::PLUSPLUS: // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001507 case Token::MINUSMINUS: {
1508 Token t = this->nextToken();
1509 std::unique_ptr<ASTExpression> expr = this->unaryExpression();
1510 if (!expr) {
1511 return nullptr;
1512 }
1513 return std::unique_ptr<ASTExpression>(new ASTPrefixExpression(t, std::move(expr)));
1514 }
1515 default:
1516 return this->postfixExpression();
1517 }
1518}
1519
1520/* term suffix* */
1521std::unique_ptr<ASTExpression> Parser::postfixExpression() {
1522 std::unique_ptr<ASTExpression> result = this->term();
1523 if (!result) {
1524 return nullptr;
1525 }
1526 for (;;) {
1527 switch (this->peek().fKind) {
1528 case Token::LBRACKET: // fall through
1529 case Token::DOT: // fall through
1530 case Token::LPAREN: // fall through
1531 case Token::PLUSPLUS: // fall through
1532 case Token::MINUSMINUS: {
1533 std::unique_ptr<ASTSuffix> s = this->suffix();
1534 if (!s) {
1535 return nullptr;
1536 }
1537 result.reset(new ASTSuffixExpression(std::move(result), std::move(s)));
1538 break;
1539 }
1540 default:
1541 return result;
1542 }
1543 }
1544}
1545
Ethan Nicholas11d53972016-11-28 11:23:23 -05001546/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
ethannicholasb3058bd2016-07-01 08:22:01 -07001547 PLUSPLUS | MINUSMINUS */
1548std::unique_ptr<ASTSuffix> Parser::suffix() {
1549 Token next = this->nextToken();
1550 switch (next.fKind) {
1551 case Token::LBRACKET: {
ethannicholas5961bc92016-10-12 06:39:56 -07001552 if (this->peek().fKind == Token::RBRACKET) {
1553 this->nextToken();
1554 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(next.fPosition));
1555 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001556 std::unique_ptr<ASTExpression> e = this->expression();
1557 if (!e) {
1558 return nullptr;
1559 }
1560 this->expect(Token::RBRACKET, "']' to complete array access expression");
1561 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(std::move(e)));
1562 }
1563 case Token::DOT: {
1564 Position pos = this->peek().fPosition;
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001565 String text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001566 if (this->identifier(&text)) {
1567 return std::unique_ptr<ASTSuffix>(new ASTFieldSuffix(pos, std::move(text)));
1568 }
1569 return nullptr;
1570 }
1571 case Token::LPAREN: {
1572 std::vector<std::unique_ptr<ASTExpression>> parameters;
1573 if (this->peek().fKind != Token::RPAREN) {
1574 for (;;) {
1575 std::unique_ptr<ASTExpression> expr = this->expression();
1576 if (!expr) {
1577 return nullptr;
1578 }
1579 parameters.push_back(std::move(expr));
1580 if (this->peek().fKind != Token::COMMA) {
1581 break;
1582 }
1583 this->nextToken();
1584 }
1585 }
1586 this->expect(Token::RPAREN, "')' to complete function parameters");
Ethan Nicholas11d53972016-11-28 11:23:23 -05001587 return std::unique_ptr<ASTSuffix>(new ASTCallSuffix(next.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001588 std::move(parameters)));
1589 }
1590 case Token::PLUSPLUS:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001591 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001592 ASTSuffix::kPostIncrement_Kind));
1593 case Token::MINUSMINUS:
1594 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fPosition,
1595 ASTSuffix::kPostDecrement_Kind));
1596 default: {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001597 this->error(next.fPosition, "expected expression suffix, but found '" + next.fText +
ethannicholasb3058bd2016-07-01 08:22:01 -07001598 "'\n");
1599 return nullptr;
1600 }
1601 }
1602}
1603
1604/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | '(' expression ')' */
1605std::unique_ptr<ASTExpression> Parser::term() {
1606 std::unique_ptr<ASTExpression> result;
1607 Token t = this->peek();
1608 switch (t.fKind) {
1609 case Token::IDENTIFIER: {
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001610 String text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001611 if (this->identifier(&text)) {
1612 result.reset(new ASTIdentifier(t.fPosition, std::move(text)));
1613 }
1614 break;
1615 }
1616 case Token::INT_LITERAL: {
1617 int64_t i;
1618 if (this->intLiteral(&i)) {
1619 result.reset(new ASTIntLiteral(t.fPosition, i));
1620 }
1621 break;
1622 }
1623 case Token::FLOAT_LITERAL: {
1624 double f;
1625 if (this->floatLiteral(&f)) {
1626 result.reset(new ASTFloatLiteral(t.fPosition, f));
1627 }
1628 break;
1629 }
1630 case Token::TRUE_LITERAL: // fall through
1631 case Token::FALSE_LITERAL: {
1632 bool b;
1633 if (this->boolLiteral(&b)) {
1634 result.reset(new ASTBoolLiteral(t.fPosition, b));
1635 }
1636 break;
1637 }
1638 case Token::LPAREN: {
1639 this->nextToken();
1640 result = this->expression();
1641 if (result) {
1642 this->expect(Token::RPAREN, "')' to complete expression");
1643 }
1644 break;
1645 }
1646 default:
1647 this->nextToken();
1648 this->error(t.fPosition, "expected expression, but found '" + t.fText + "'\n");
1649 result = nullptr;
1650 }
1651 return result;
1652}
1653
1654/* INT_LITERAL */
1655bool Parser::intLiteral(int64_t* dest) {
1656 Token t;
1657 if (this->expect(Token::INT_LITERAL, "integer literal", &t)) {
1658 *dest = SkSL::stol(t.fText);
1659 return true;
1660 }
1661 return false;
1662}
1663
1664/* FLOAT_LITERAL */
1665bool Parser::floatLiteral(double* dest) {
1666 Token t;
1667 if (this->expect(Token::FLOAT_LITERAL, "float literal", &t)) {
1668 *dest = SkSL::stod(t.fText);
1669 return true;
1670 }
1671 return false;
1672}
1673
1674/* TRUE_LITERAL | FALSE_LITERAL */
1675bool Parser::boolLiteral(bool* dest) {
1676 Token t = this->nextToken();
1677 switch (t.fKind) {
1678 case Token::TRUE_LITERAL:
1679 *dest = true;
1680 return true;
1681 case Token::FALSE_LITERAL:
1682 *dest = false;
1683 return true;
1684 default:
1685 this->error(t.fPosition, "expected 'true' or 'false', but found '" + t.fText + "'\n");
1686 return false;
1687 }
1688}
1689
1690/* IDENTIFIER */
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001691bool Parser::identifier(String* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001692 Token t;
1693 if (this->expect(Token::IDENTIFIER, "identifier", &t)) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001694 *dest = t.fText;
ethannicholasb3058bd2016-07-01 08:22:01 -07001695 return true;
1696 }
1697 return false;
1698}
1699
1700} // namespace