blob: f712b34e1a5164206e93647116e607cc0ed50a27 [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 Nicholas0dd30d92017-05-01 16:57:07 -040015static_assert(YY_FLEX_MAJOR_VERSION * 10000 + YY_FLEX_MINOR_VERSION * 100 +
16 YY_FLEX_SUBMINOR_VERSION >= 20601,
Ethan Nicholas94421942017-03-31 17:06:42 -040017 "we require Flex 2.6.1 or better for security reasons");
Ethan Nicholasbfe15f62017-03-01 11:46:51 -050018#undef register
ethannicholasb3058bd2016-07-01 08:22:01 -070019#ifdef __clang__
20#pragma clang diagnostic pop
21#endif
ethannicholasf789b382016-08-03 12:43:36 -070022#ifdef __GNUC__
23#pragma GCC diagnostic pop
24#endif
25#ifdef _MSC_VER
26#pragma warning(pop)
27#endif
ethannicholasb3058bd2016-07-01 08:22:01 -070028
Ethan Nicholasbfe15f62017-03-01 11:46:51 -050029#include "lex.layout.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070030#include "ast/SkSLASTBinaryExpression.h"
31#include "ast/SkSLASTBlock.h"
32#include "ast/SkSLASTBoolLiteral.h"
33#include "ast/SkSLASTBreakStatement.h"
34#include "ast/SkSLASTCallSuffix.h"
35#include "ast/SkSLASTContinueStatement.h"
36#include "ast/SkSLASTDiscardStatement.h"
37#include "ast/SkSLASTDoStatement.h"
38#include "ast/SkSLASTExpression.h"
39#include "ast/SkSLASTExpressionStatement.h"
40#include "ast/SkSLASTExtension.h"
41#include "ast/SkSLASTFieldSuffix.h"
42#include "ast/SkSLASTFloatLiteral.h"
43#include "ast/SkSLASTForStatement.h"
44#include "ast/SkSLASTFunction.h"
45#include "ast/SkSLASTIdentifier.h"
46#include "ast/SkSLASTIfStatement.h"
47#include "ast/SkSLASTIndexSuffix.h"
48#include "ast/SkSLASTInterfaceBlock.h"
49#include "ast/SkSLASTIntLiteral.h"
ethannicholas5961bc92016-10-12 06:39:56 -070050#include "ast/SkSLASTModifiersDeclaration.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070051#include "ast/SkSLASTParameter.h"
ethannicholas5961bc92016-10-12 06:39:56 -070052#include "ast/SkSLASTPrecision.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070053#include "ast/SkSLASTPrefixExpression.h"
54#include "ast/SkSLASTReturnStatement.h"
Ethan Nicholas762466e2017-06-29 10:03:38 -040055#include "ast/SkSLASTSection.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070056#include "ast/SkSLASTStatement.h"
57#include "ast/SkSLASTSuffixExpression.h"
Ethan Nicholasaf197692017-02-27 13:26:45 -050058#include "ast/SkSLASTSwitchCase.h"
59#include "ast/SkSLASTSwitchStatement.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070060#include "ast/SkSLASTTernaryExpression.h"
61#include "ast/SkSLASTType.h"
62#include "ast/SkSLASTVarDeclaration.h"
63#include "ast/SkSLASTVarDeclarationStatement.h"
64#include "ast/SkSLASTWhileStatement.h"
65#include "ir/SkSLSymbolTable.h"
Ethan Nicholas11d53972016-11-28 11:23:23 -050066#include "ir/SkSLModifiers.h"
ethannicholasd598f792016-07-25 10:08:54 -070067#include "ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070068
69namespace SkSL {
70
ethannicholascad64162016-10-27 10:54:02 -070071#define MAX_PARSE_DEPTH 50
72
73class AutoDepth {
74public:
75 AutoDepth(Parser* p)
76 : fParser(p) {
77 fParser->fDepth++;
78 }
79
80 ~AutoDepth() {
81 fParser->fDepth--;
82 }
83
84 bool checkValid() {
85 if (fParser->fDepth > MAX_PARSE_DEPTH) {
Ethan Nicholas0df1b042017-03-31 13:56:23 -040086 fParser->error(fParser->peek().fPosition, String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070087 return false;
88 }
89 return true;
90 }
91
92private:
93 Parser* fParser;
94};
95
Ethan Nicholas0df1b042017-03-31 13:56:23 -040096Parser::Parser(String text, SymbolTable& types, ErrorReporter& errors)
97: fPushback(Position(-1, -1), Token::INVALID_TOKEN, String())
ethannicholasb3058bd2016-07-01 08:22:01 -070098, fTypes(types)
99, fErrors(errors) {
100 sksllex_init(&fScanner);
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500101 layoutlex_init(&fLayoutScanner);
ethannicholasb3058bd2016-07-01 08:22:01 -0700102 fBuffer = sksl_scan_string(text.c_str(), fScanner);
103 skslset_lineno(1, fScanner);
ethannicholasb3058bd2016-07-01 08:22:01 -0700104}
105
106Parser::~Parser() {
107 sksl_delete_buffer(fBuffer, fScanner);
ethannicholas69a7e7c2016-07-01 19:09:27 -0700108 sksllex_destroy(fScanner);
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500109 layoutlex_destroy(fLayoutScanner);
ethannicholasb3058bd2016-07-01 08:22:01 -0700110}
111
Ethan Nicholas762466e2017-06-29 10:03:38 -0400112/* (precision | directive | section | declaration)* END_OF_FILE */
ethannicholasb3058bd2016-07-01 08:22:01 -0700113std::vector<std::unique_ptr<ASTDeclaration>> Parser::file() {
114 std::vector<std::unique_ptr<ASTDeclaration>> result;
115 for (;;) {
116 switch (this->peek().fKind) {
117 case Token::END_OF_FILE:
118 return result;
ethannicholas5961bc92016-10-12 06:39:56 -0700119 case Token::PRECISION: {
120 std::unique_ptr<ASTDeclaration> precision = this->precision();
121 if (precision) {
122 result.push_back(std::move(precision));
123 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700124 break;
ethannicholas5961bc92016-10-12 06:39:56 -0700125 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700126 case Token::DIRECTIVE: {
127 std::unique_ptr<ASTDeclaration> decl = this->directive();
128 if (decl) {
129 result.push_back(std::move(decl));
130 }
131 break;
132 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400133 case Token::SECTION: {
134 std::unique_ptr<ASTDeclaration> section = this->section();
135 if (section) {
136 result.push_back(std::move(section));
137 }
138 break;
139 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700140 default: {
141 std::unique_ptr<ASTDeclaration> decl = this->declaration();
142 if (!decl) {
143 continue;
144 }
145 result.push_back(std::move(decl));
146 }
147 }
148 }
149}
150
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400151Token Parser::nextRawToken(bool needText) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700152 if (fPushback.fKind != Token::INVALID_TOKEN) {
Brian Osman634624a2017-08-15 11:14:30 -0400153 Token result(std::move(fPushback));
ethannicholasb3058bd2016-07-01 08:22:01 -0700154 fPushback.fKind = Token::INVALID_TOKEN;
Brian Osman634624a2017-08-15 11:14:30 -0400155 fPushback.fText.clear();
ethannicholasb3058bd2016-07-01 08:22:01 -0700156 return result;
157 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400158 Token::Kind kind = (Token::Kind) sksllex(fScanner);
159 if (!needText) {
160 switch (kind) {
161 case Token::Kind::DIRECTIVE: // fall through
162 case Token::Kind::IDENTIFIER: // fall through
163 case Token::Kind::INT_LITERAL: // fall through
164 case Token::Kind::FLOAT_LITERAL: // fall through
165 case Token::Kind::SECTION:
166 needText = true;
167 default:
168 break;
169 }
170 }
171 static String unavailable("<unavailable>");
172 return Token(Position(skslget_lineno(fScanner), -1), kind,
173 needText ? String(skslget_text(fScanner)) : unavailable);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400174}
175
176Token Parser::nextToken() {
177 Token token;
178 do {
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400179 token = this->nextRawToken(false);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400180 } while (token.fKind == Token::WHITESPACE);
181 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700182}
183
184void Parser::pushback(Token t) {
185 ASSERT(fPushback.fKind == Token::INVALID_TOKEN);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400186 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700187}
188
189Token Parser::peek() {
Brian Osman634624a2017-08-15 11:14:30 -0400190 if (fPushback.fKind == Token::INVALID_TOKEN) {
191 fPushback = this->nextToken();
192 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700193 return fPushback;
194}
195
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400196bool Parser::checkNext(Token::Kind kind, Token* result) {
Brian Osman634624a2017-08-15 11:14:30 -0400197 if (fPushback.fKind != Token::INVALID_TOKEN && fPushback.fKind != kind) {
198 return false;
199 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400200 Token next = this->nextToken();
201 if (next.fKind == kind) {
202 if (result) {
203 *result = next;
204 }
205 return true;
206 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400207 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400208 return false;
209}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500210
211bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700212 Token next = this->nextToken();
213 if (next.fKind == kind) {
214 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400215 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700216 }
217 return true;
218 } else {
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500219 if (next.fText.size()) {
Brian Osman634624a2017-08-15 11:14:30 -0400220 this->error(next.fPosition, "expected " + String(expected) + ", but found '" +
221 next.fText + "'");
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500222 } else {
223 this->error(next.fPosition, "parse error, recompile in debug mode for details");
224 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700225 return false;
226 }
227}
228
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500229void Parser::error(Position p, const char* msg) {
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400230 this->error(p, String(msg));
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500231}
232
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400233void Parser::error(Position p, String msg) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700234 fErrors.error(p, msg);
235}
236
Brian Osman634624a2017-08-15 11:14:30 -0400237bool Parser::isType(const String& name) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700238 return nullptr != fTypes[name];
239}
240
241/* PRECISION (LOWP | MEDIUMP | HIGHP) type SEMICOLON */
ethannicholas5961bc92016-10-12 06:39:56 -0700242std::unique_ptr<ASTDeclaration> Parser::precision() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700243 if (!this->expect(Token::PRECISION, "'precision'")) {
ethannicholas5961bc92016-10-12 06:39:56 -0700244 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700245 }
ethannicholas5961bc92016-10-12 06:39:56 -0700246 Modifiers::Flag result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700247 Token p = this->nextToken();
248 switch (p.fKind) {
ethannicholas5961bc92016-10-12 06:39:56 -0700249 case Token::LOWP:
250 result = Modifiers::kLowp_Flag;
251 break;
252 case Token::MEDIUMP:
253 result = Modifiers::kMediump_Flag;
254 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700255 case Token::HIGHP:
ethannicholas5961bc92016-10-12 06:39:56 -0700256 result = Modifiers::kHighp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700257 break;
258 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500259 this->error(p.fPosition, "expected 'lowp', 'mediump', or 'highp', but found '" +
ethannicholasb3058bd2016-07-01 08:22:01 -0700260 p.fText + "'");
ethannicholas5961bc92016-10-12 06:39:56 -0700261 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700262 }
ethannicholas5961bc92016-10-12 06:39:56 -0700263 // FIXME handle the type
ethannicholasb3058bd2016-07-01 08:22:01 -0700264 if (!this->type()) {
ethannicholas5961bc92016-10-12 06:39:56 -0700265 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700266 }
267 this->expect(Token::SEMICOLON, "';'");
ethannicholas5961bc92016-10-12 06:39:56 -0700268 return std::unique_ptr<ASTDeclaration>(new ASTPrecision(p.fPosition, result));
ethannicholasb3058bd2016-07-01 08:22:01 -0700269}
270
Ethan Nicholas11d53972016-11-28 11:23:23 -0500271/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700272 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
ethannicholasb3058bd2016-07-01 08:22:01 -0700273std::unique_ptr<ASTDeclaration> Parser::directive() {
274 Token start;
275 if (!this->expect(Token::DIRECTIVE, "a directive", &start)) {
276 return nullptr;
277 }
278 if (start.fText == "#version") {
279 this->expect(Token::INT_LITERAL, "a version number");
ethannicholas5961bc92016-10-12 06:39:56 -0700280 Token next = this->peek();
281 if (next.fText == "es" || next.fText == "compatibility") {
282 this->nextToken();
283 }
284 // version is ignored for now; it will eventually become an error when we stop pretending
285 // to be GLSL
ethannicholasb3058bd2016-07-01 08:22:01 -0700286 return nullptr;
287 } else if (start.fText == "#extension") {
288 Token name;
289 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
290 return nullptr;
291 }
292 if (!this->expect(Token::COLON, "':'")) {
293 return nullptr;
294 }
295 // FIXME: need to start paying attention to this token
296 if (!this->expect(Token::IDENTIFIER, "an identifier")) {
297 return nullptr;
298 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500299 return std::unique_ptr<ASTDeclaration>(new ASTExtension(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700300 std::move(name.fText)));
301 } else {
302 this->error(start.fPosition, "unsupported directive '" + start.fText + "'");
303 return nullptr;
304 }
305}
306
Ethan Nicholas762466e2017-06-29 10:03:38 -0400307/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
308 RBRACE */
309std::unique_ptr<ASTDeclaration> Parser::section() {
310 Token start;
311 if (!this->expect(Token::SECTION, "a section token", &start)) {
312 return nullptr;
313 }
314 String argument;
315 if (this->peek().fKind == Token::LPAREN) {
316 this->nextToken();
317 Token argToken;
318 if (!this->expect(Token::IDENTIFIER, "an identifier", &argToken)) {
319 return nullptr;
320 }
321 argument = argToken.fText;
322 if (!this->expect(Token::RPAREN, "')'")) {
323 return nullptr;
324 }
325 }
326 if (!this->expect(Token::LBRACE, "'{'")) {
327 return nullptr;
328 }
329 String text;
330 int level = 1;
331 for (;;) {
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400332 Token next = this->nextRawToken(true);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400333 switch (next.fKind) {
334 case Token::LBRACE:
335 ++level;
336 break;
337 case Token::RBRACE:
338 --level;
339 break;
340 case Token::END_OF_FILE:
341 this->error(start.fPosition, "reached end of file while parsing section");
342 return nullptr;
343 default:
344 break;
345 }
346 if (!level) {
347 break;
348 }
349 text += next.fText;
350 }
351 return std::unique_ptr<ASTDeclaration>(new ASTSection(start.fPosition,
352 String(start.fText.c_str() + 1),
353 argument,
354 text));
355}
356
Ethan Nicholas11d53972016-11-28 11:23:23 -0500357/* modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700358 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
359std::unique_ptr<ASTDeclaration> Parser::declaration() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500360 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700361 Token lookahead = this->peek();
362 if (lookahead.fKind == Token::IDENTIFIER && !this->isType(lookahead.fText)) {
363 // we have an identifier that's not a type, could be the start of an interface block
364 return this->interfaceBlock(modifiers);
365 }
366 if (lookahead.fKind == Token::STRUCT) {
367 return this->structVarDeclaration(modifiers);
368 }
ethannicholas5961bc92016-10-12 06:39:56 -0700369 if (lookahead.fKind == Token::SEMICOLON) {
370 this->nextToken();
371 return std::unique_ptr<ASTDeclaration>(new ASTModifiersDeclaration(modifiers));
372 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700373 std::unique_ptr<ASTType> type(this->type());
374 if (!type) {
375 return nullptr;
376 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400377 if (type->fKind == ASTType::kStruct_Kind && this->checkNext(Token::SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700378 return nullptr;
379 }
380 Token name;
381 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
382 return nullptr;
383 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400384 if (this->checkNext(Token::LPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700385 std::vector<std::unique_ptr<ASTParameter>> parameters;
386 while (this->peek().fKind != Token::RPAREN) {
387 if (parameters.size() > 0) {
388 if (!this->expect(Token::COMMA, "','")) {
389 return nullptr;
390 }
391 }
392 std::unique_ptr<ASTParameter> parameter = this->parameter();
393 if (!parameter) {
394 return nullptr;
395 }
396 parameters.push_back(std::move(parameter));
397 }
398 this->nextToken();
399 std::unique_ptr<ASTBlock> body;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400400 if (!this->checkNext(Token::SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700401 body = this->block();
402 if (!body) {
403 return nullptr;
404 }
405 }
Ethan Nicholascb670962017-04-20 19:31:52 -0400406 return std::unique_ptr<ASTDeclaration>(new ASTFunction(name.fPosition,
407 modifiers,
408 std::move(type),
Ethan Nicholas11d53972016-11-28 11:23:23 -0500409 std::move(name.fText),
410 std::move(parameters),
ethannicholasb3058bd2016-07-01 08:22:01 -0700411 std::move(body)));
412 } else {
Brian Osman634624a2017-08-15 11:14:30 -0400413 return this->varDeclarationEnd(modifiers, std::move(type), std::move(name.fText));
ethannicholasb3058bd2016-07-01 08:22:01 -0700414 }
415}
416
417/* modifiers type IDENTIFIER varDeclarationEnd */
ethannicholas14fe8cc2016-09-07 13:37:16 -0700418std::unique_ptr<ASTVarDeclarations> Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500419 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700420 std::unique_ptr<ASTType> type(this->type());
421 if (!type) {
422 return nullptr;
423 }
424 Token name;
425 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
426 return nullptr;
427 }
428 return this->varDeclarationEnd(modifiers, std::move(type), std::move(name.fText));
429}
430
431/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
432std::unique_ptr<ASTType> Parser::structDeclaration() {
433 if (!this->expect(Token::STRUCT, "'struct'")) {
434 return nullptr;
435 }
436 Token name;
437 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
438 return nullptr;
439 }
440 if (!this->expect(Token::LBRACE, "'{'")) {
441 return nullptr;
442 }
443 std::vector<Type::Field> fields;
444 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700445 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700446 if (!decl) {
447 return nullptr;
448 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700449 for (const auto& var : decl->fVars) {
ethannicholasd598f792016-07-25 10:08:54 -0700450 auto type = (const Type*) fTypes[decl->fType->fName];
ethannicholas14fe8cc2016-09-07 13:37:16 -0700451 for (int i = (int) var.fSizes.size() - 1; i >= 0; i--) {
ethannicholasdd4645b2016-10-14 12:14:46 -0700452 if (!var.fSizes[i] || var.fSizes[i]->fKind != ASTExpression::kInt_Kind) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700453 this->error(decl->fPosition, "array size in struct field must be a constant");
ethannicholasdd4645b2016-10-14 12:14:46 -0700454 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700455 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700456 uint64_t columns = ((ASTIntLiteral&) *var.fSizes[i]).fValue;
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400457 String name = type->name() + "[" + to_string(columns) + "]";
ethannicholasd598f792016-07-25 10:08:54 -0700458 type = new Type(name, Type::kArray_Kind, *type, (int) columns);
459 fTypes.takeOwnership((Type*) type);
ethannicholasb3058bd2016-07-01 08:22:01 -0700460 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700461 fields.push_back(Type::Field(decl->fModifiers, var.fName, type));
462 if (var.fValue) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700463 this->error(decl->fPosition, "initializers are not permitted on struct fields");
464 }
465 }
466 }
467 if (!this->expect(Token::RBRACE, "'}'")) {
468 return nullptr;
469 }
Ethan Nicholas19671772016-11-28 16:30:17 -0500470 fTypes.add(name.fText, std::unique_ptr<Type>(new Type(name.fPosition, name.fText, fields)));
Ethan Nicholas11d53972016-11-28 11:23:23 -0500471 return std::unique_ptr<ASTType>(new ASTType(name.fPosition, name.fText,
Ethan Nicholas50afc172017-02-16 14:49:57 -0500472 ASTType::kStruct_Kind, std::vector<int>()));
ethannicholasb3058bd2016-07-01 08:22:01 -0700473}
474
475/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500476std::unique_ptr<ASTVarDeclarations> Parser::structVarDeclaration(Modifiers modifiers) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700477 std::unique_ptr<ASTType> type = this->structDeclaration();
478 if (!type) {
479 return nullptr;
480 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400481 Token name;
482 if (this->checkNext(Token::IDENTIFIER, &name)) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500483 std::unique_ptr<ASTVarDeclarations> result = this->varDeclarationEnd(modifiers,
484 std::move(type),
ethannicholas14fe8cc2016-09-07 13:37:16 -0700485 std::move(name.fText));
ethannicholasb3058bd2016-07-01 08:22:01 -0700486 if (result) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700487 for (const auto& var : result->fVars) {
488 if (var.fValue) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500489 this->error(var.fValue->fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -0700490 "struct variables cannot be initialized");
491 }
492 }
493 }
494 return result;
495 }
496 this->expect(Token::SEMICOLON, "';'");
497 return nullptr;
498}
499
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400500/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
501 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500502std::unique_ptr<ASTVarDeclarations> Parser::varDeclarationEnd(Modifiers mods,
ethannicholas14fe8cc2016-09-07 13:37:16 -0700503 std::unique_ptr<ASTType> type,
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400504 String name) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700505 std::vector<ASTVarDeclaration> vars;
ethannicholasb3058bd2016-07-01 08:22:01 -0700506 std::vector<std::unique_ptr<ASTExpression>> currentVarSizes;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400507 while (this->checkNext(Token::LBRACKET)) {
508 if (this->checkNext(Token::RBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700509 currentVarSizes.push_back(nullptr);
510 } else {
511 std::unique_ptr<ASTExpression> size(this->expression());
512 if (!size) {
513 return nullptr;
514 }
515 currentVarSizes.push_back(std::move(size));
516 if (!this->expect(Token::RBRACKET, "']'")) {
517 return nullptr;
518 }
519 }
520 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700521 std::unique_ptr<ASTExpression> value;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400522 if (this->checkNext(Token::EQ)) {
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400523 value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700524 if (!value) {
525 return nullptr;
526 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700527 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700528 vars.emplace_back(std::move(name), std::move(currentVarSizes), std::move(value));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400529 while (this->checkNext(Token::COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700530 Token name;
531 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
532 return nullptr;
533 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700534 currentVarSizes.clear();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700535 value.reset();
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400536 while (this->checkNext(Token::LBRACKET)) {
537 if (this->checkNext(Token::RBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700538 currentVarSizes.push_back(nullptr);
539 } else {
540 std::unique_ptr<ASTExpression> size(this->expression());
541 if (!size) {
542 return nullptr;
543 }
544 currentVarSizes.push_back(std::move(size));
545 if (!this->expect(Token::RBRACKET, "']'")) {
546 return nullptr;
547 }
548 }
549 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400550 if (this->checkNext(Token::EQ)) {
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400551 value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700552 if (!value) {
553 return nullptr;
554 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700555 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700556 vars.emplace_back(std::move(name.fText), std::move(currentVarSizes), std::move(value));
ethannicholasb3058bd2016-07-01 08:22:01 -0700557 }
558 if (!this->expect(Token::SEMICOLON, "';'")) {
559 return nullptr;
560 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700561 return std::unique_ptr<ASTVarDeclarations>(new ASTVarDeclarations(std::move(mods),
562 std::move(type),
563 std::move(vars)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700564}
565
566/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
567std::unique_ptr<ASTParameter> Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400568 Modifiers modifiers = this->modifiersWithDefaults(0);
ethannicholasb3058bd2016-07-01 08:22:01 -0700569 std::unique_ptr<ASTType> type = this->type();
570 if (!type) {
571 return nullptr;
572 }
573 Token name;
574 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
575 return nullptr;
576 }
577 std::vector<int> sizes;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400578 while (this->checkNext(Token::LBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700579 Token sizeToken;
580 if (!this->expect(Token::INT_LITERAL, "a positive integer", &sizeToken)) {
581 return nullptr;
582 }
583 sizes.push_back(SkSL::stoi(sizeToken.fText));
584 if (!this->expect(Token::RBRACKET, "']'")) {
585 return nullptr;
586 }
587 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500588 return std::unique_ptr<ASTParameter>(new ASTParameter(name.fPosition, modifiers,
589 std::move(type), name.fText,
ethannicholasb3058bd2016-07-01 08:22:01 -0700590 std::move(sizes)));
591}
592
593/** (EQ INT_LITERAL)? */
594int Parser::layoutInt() {
595 if (!this->expect(Token::EQ, "'='")) {
596 return -1;
597 }
598 Token resultToken;
599 if (this->expect(Token::INT_LITERAL, "a non-negative integer", &resultToken)) {
600 return SkSL::stoi(resultToken.fText);
601 }
602 return -1;
603}
604
Ethan Nicholas762466e2017-06-29 10:03:38 -0400605/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
606String Parser::layoutCode() {
607 if (!this->expect(Token::EQ, "'='")) {
608 return "";
609 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400610 Token start = this->nextRawToken(true);
611 this->pushback(start);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400612 String code;
613 int level = 1;
614 bool done = false;
615 while (!done) {
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400616 Token next = this->nextRawToken(true);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400617 switch (next.fKind) {
618 case Token::LPAREN:
619 ++level;
620 break;
621 case Token::RPAREN:
622 --level;
623 break;
624 case Token::COMMA:
625 if (level == 1) {
626 done = true;
627 }
628 break;
629 case Token::END_OF_FILE:
630 this->error(start.fPosition, "reached end of file while parsing layout");
631 return nullptr;
632 default:
633 break;
634 }
635 if (!level) {
636 done = true;
637 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400638 if (done) {
639 this->pushback(std::move(next));
640 }
641 else {
642 code += next.fText;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400643 }
644 }
645 return code;
646}
647
648/** (EQ IDENTIFIER('identity'))? */
649Layout::Key Parser::layoutKey() {
650 if (this->peek().fKind == Token::EQ) {
651 this->expect(Token::EQ, "'='");
652 Token key;
653 if (this->expect(Token::IDENTIFIER, "an identifer", &key)) {
654 if (key.fText == "identity") {
655 return Layout::kIdentity_Key;
656 } else {
657 this->error(key.fPosition, "unsupported layout key");
658 }
659 }
660 }
661 return Layout::kKey_Key;
662}
663
ethannicholas8ac838d2016-11-22 08:39:36 -0800664/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500665Layout Parser::layout() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700666 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500667 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700668 int binding = -1;
669 int index = -1;
670 int set = -1;
671 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500672 int inputAttachmentIndex = -1;
ethannicholasf789b382016-08-03 12:43:36 -0700673 bool originUpperLeft = false;
ethannicholas5961bc92016-10-12 06:39:56 -0700674 bool overrideCoverage = false;
675 bool blendSupportAllEquations = false;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500676 Layout::Format format = Layout::Format::kUnspecified;
ethannicholas8ac838d2016-11-22 08:39:36 -0800677 bool pushConstant = false;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500678 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
679 int maxVertices = -1;
680 int invocations = -1;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400681 String when;
682 Layout::Key key = Layout::kNo_Key;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400683 if (this->checkNext(Token::LAYOUT)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700684 if (!this->expect(Token::LPAREN, "'('")) {
Ethan Nicholas19671772016-11-28 16:30:17 -0500685 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
Ethan Nicholas11d53972016-11-28 11:23:23 -0500686 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas762466e2017-06-29 10:03:38 -0400687 pushConstant, primitive, maxVertices, invocations, when, key);
ethannicholasb3058bd2016-07-01 08:22:01 -0700688 }
689 for (;;) {
690 Token t = this->nextToken();
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500691 YY_BUFFER_STATE buffer;
692 buffer = layout_scan_string(t.fText.c_str(), fLayoutScanner);
693 int token = layoutlex(fLayoutScanner);
694 layout_delete_buffer(buffer, fLayoutScanner);
695 if (token != Token::INVALID_TOKEN) {
696 switch (token) {
697 case Token::LOCATION:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500698 location = this->layoutInt();
699 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500700 case Token::OFFSET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500701 offset = this->layoutInt();
702 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500703 case Token::BINDING:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500704 binding = this->layoutInt();
705 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500706 case Token::INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500707 index = this->layoutInt();
708 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500709 case Token::SET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500710 set = this->layoutInt();
711 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500712 case Token::BUILTIN:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500713 builtin = this->layoutInt();
714 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500715 case Token::INPUT_ATTACHMENT_INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500716 inputAttachmentIndex = this->layoutInt();
717 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500718 case Token::ORIGIN_UPPER_LEFT:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500719 originUpperLeft = true;
720 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500721 case Token::OVERRIDE_COVERAGE:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500722 overrideCoverage = true;
723 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500724 case Token::BLEND_SUPPORT_ALL_EQUATIONS:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500725 blendSupportAllEquations = true;
726 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500727 case Token::PUSH_CONSTANT:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500728 pushConstant = true;
729 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500730 case Token::POINTS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500731 primitive = Layout::kPoints_Primitive;
732 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500733 case Token::LINES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500734 primitive = Layout::kLines_Primitive;
735 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500736 case Token::LINE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500737 primitive = Layout::kLineStrip_Primitive;
738 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500739 case Token::LINES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500740 primitive = Layout::kLinesAdjacency_Primitive;
741 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500742 case Token::TRIANGLES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500743 primitive = Layout::kTriangles_Primitive;
744 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500745 case Token::TRIANGLE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500746 primitive = Layout::kTriangleStrip_Primitive;
747 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500748 case Token::TRIANGLES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500749 primitive = Layout::kTrianglesAdjacency_Primitive;
750 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500751 case Token::MAX_VERTICES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500752 maxVertices = this->layoutInt();
753 break;
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500754 case Token::INVOCATIONS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500755 invocations = this->layoutInt();
756 break;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400757 case Token::WHEN:
758 when = this->layoutCode();
759 break;
760 case Token::KEY:
761 key = this->layoutKey();
762 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500763 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500764 } else if (Layout::ReadFormat(t.fText, &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500765 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700766 } else {
Greg Daniel64773e62016-11-22 09:44:03 -0500767 this->error(t.fPosition, ("'" + t.fText +
ethannicholasb3058bd2016-07-01 08:22:01 -0700768 "' is not a valid layout qualifier").c_str());
769 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400770 if (this->checkNext(Token::RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700771 break;
772 }
773 if (!this->expect(Token::COMMA, "','")) {
774 break;
775 }
776 }
777 }
Ethan Nicholas19671772016-11-28 16:30:17 -0500778 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
779 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas762466e2017-06-29 10:03:38 -0400780 pushConstant, primitive, maxVertices, invocations, when, key);
ethannicholasb3058bd2016-07-01 08:22:01 -0700781}
782
Brian Salomonf9f45122016-11-29 11:59:17 -0500783/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400784 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT | BUFFER)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500785Modifiers Parser::modifiers() {
786 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700787 int flags = 0;
788 for (;;) {
789 // TODO: handle duplicate / incompatible flags
790 switch (peek().fKind) {
791 case Token::UNIFORM:
792 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500793 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700794 break;
795 case Token::CONST:
796 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500797 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700798 break;
799 case Token::IN:
800 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500801 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700802 break;
803 case Token::OUT:
804 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500805 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700806 break;
807 case Token::INOUT:
808 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500809 flags |= Modifiers::kIn_Flag;
810 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700811 break;
812 case Token::LOWP:
813 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500814 flags |= Modifiers::kLowp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700815 break;
816 case Token::MEDIUMP:
817 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500818 flags |= Modifiers::kMediump_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700819 break;
820 case Token::HIGHP:
821 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500822 flags |= Modifiers::kHighp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700823 break;
ethannicholasf789b382016-08-03 12:43:36 -0700824 case Token::FLAT:
825 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500826 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700827 break;
828 case Token::NOPERSPECTIVE:
829 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500830 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700831 break;
Brian Salomonf9f45122016-11-29 11:59:17 -0500832 case Token::READONLY:
833 this->nextToken();
834 flags |= Modifiers::kReadOnly_Flag;
835 break;
836 case Token::WRITEONLY:
837 this->nextToken();
838 flags |= Modifiers::kWriteOnly_Flag;
839 break;
840 case Token::COHERENT:
841 this->nextToken();
842 flags |= Modifiers::kCoherent_Flag;
843 break;
844 case Token::VOLATILE:
845 this->nextToken();
846 flags |= Modifiers::kVolatile_Flag;
847 break;
848 case Token::RESTRICT:
849 this->nextToken();
850 flags |= Modifiers::kRestrict_Flag;
851 break;
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400852 case Token::BUFFER:
853 this->nextToken();
854 flags |= Modifiers::kBuffer_Flag;
855 break;
Ethan Nicholascb670962017-04-20 19:31:52 -0400856 case Token::HASSIDEEFFECTS:
857 this->nextToken();
858 flags |= Modifiers::kHasSideEffects_Flag;
859 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700860 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500861 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700862 }
863 }
864}
865
Ethan Nicholas11d53972016-11-28 11:23:23 -0500866Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
867 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700868 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500869 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700870 }
871 return result;
872}
873
874/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
875std::unique_ptr<ASTStatement> Parser::statement() {
876 Token start = this->peek();
877 switch (start.fKind) {
Ethan Nicholas5ac13c22017-05-10 15:06:17 -0400878 case Token::IF: // fall through
879 case Token::STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -0700880 return this->ifStatement();
881 case Token::FOR:
882 return this->forStatement();
883 case Token::DO:
884 return this->doStatement();
885 case Token::WHILE:
886 return this->whileStatement();
Ethan Nicholas5ac13c22017-05-10 15:06:17 -0400887 case Token::SWITCH: // fall through
888 case Token::STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -0500889 return this->switchStatement();
ethannicholasb3058bd2016-07-01 08:22:01 -0700890 case Token::RETURN:
891 return this->returnStatement();
892 case Token::BREAK:
893 return this->breakStatement();
894 case Token::CONTINUE:
895 return this->continueStatement();
896 case Token::DISCARD:
897 return this->discardStatement();
898 case Token::LBRACE:
899 return this->block();
900 case Token::SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500901 this->nextToken();
902 return std::unique_ptr<ASTStatement>(new ASTBlock(start.fPosition,
ethannicholas0730be72016-09-01 07:59:02 -0700903 std::vector<std::unique_ptr<ASTStatement>>()));
ethannicholasb3058bd2016-07-01 08:22:01 -0700904 case Token::CONST: // fall through
905 case Token::HIGHP: // fall through
906 case Token::MEDIUMP: // fall through
907 case Token::LOWP: {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700908 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700909 if (!decl) {
910 return nullptr;
911 }
912 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(std::move(decl)));
913 }
914 case Token::IDENTIFIER:
915 if (this->isType(start.fText)) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700916 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700917 if (!decl) {
918 return nullptr;
919 }
920 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
921 std::move(decl)));
922 }
923 // fall through
924 default:
925 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500926 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700927}
928
Ethan Nicholas50afc172017-02-16 14:49:57 -0500929/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* */
ethannicholasb3058bd2016-07-01 08:22:01 -0700930std::unique_ptr<ASTType> Parser::type() {
931 Token type;
932 if (!this->expect(Token::IDENTIFIER, "a type", &type)) {
933 return nullptr;
934 }
935 if (!this->isType(type.fText)) {
936 this->error(type.fPosition, ("no type named '" + type.fText + "'").c_str());
937 return nullptr;
938 }
Ethan Nicholas50afc172017-02-16 14:49:57 -0500939 std::vector<int> sizes;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400940 while (this->checkNext(Token::LBRACKET)) {
Ethan Nicholas50afc172017-02-16 14:49:57 -0500941 if (this->peek().fKind != Token::RBRACKET) {
942 int64_t i;
943 if (this->intLiteral(&i)) {
944 sizes.push_back(i);
945 } else {
946 return nullptr;
947 }
948 } else {
949 sizes.push_back(-1);
950 }
951 this->expect(Token::RBRACKET, "']'");
952 }
Ethan Nicholas11d53972016-11-28 11:23:23 -0500953 return std::unique_ptr<ASTType>(new ASTType(type.fPosition, std::move(type.fText),
Ethan Nicholas50afc172017-02-16 14:49:57 -0500954 ASTType::kIdentifier_Kind, sizes));
ethannicholasb3058bd2016-07-01 08:22:01 -0700955}
956
Ethan Nicholas50afc172017-02-16 14:49:57 -0500957/* IDENTIFIER LBRACE varDeclaration* RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500958std::unique_ptr<ASTDeclaration> Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700959 Token name;
960 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
961 return nullptr;
962 }
963 if (peek().fKind != Token::LBRACE) {
964 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -0500965 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -0700966 // it's better to report it as an unknown type
967 this->error(name.fPosition, "no type named '" + name.fText + "'");
968 return nullptr;
969 }
970 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500971 std::vector<std::unique_ptr<ASTVarDeclarations>> decls;
ethannicholasb3058bd2016-07-01 08:22:01 -0700972 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700973 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700974 if (!decl) {
975 return nullptr;
976 }
977 decls.push_back(std::move(decl));
978 }
979 this->nextToken();
Ethan Nicholas50afc172017-02-16 14:49:57 -0500980 std::vector<std::unique_ptr<ASTExpression>> sizes;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400981 Token instanceName;
982 if (this->checkNext(Token::IDENTIFIER, &instanceName)) {
983 while (this->checkNext(Token::LBRACKET)) {
Ethan Nicholas50afc172017-02-16 14:49:57 -0500984 if (this->peek().fKind != Token::RBRACKET) {
985 std::unique_ptr<ASTExpression> size = this->expression();
986 if (!size) {
987 return nullptr;
988 }
989 sizes.push_back(std::move(size));
990 } else {
991 sizes.push_back(nullptr);
992 }
993 this->expect(Token::RBRACKET, "']'");
994 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700995 }
996 this->expect(Token::SEMICOLON, "';'");
Ethan Nicholas11d53972016-11-28 11:23:23 -0500997 return std::unique_ptr<ASTDeclaration>(new ASTInterfaceBlock(name.fPosition, mods,
Brian Osman634624a2017-08-15 11:14:30 -0400998 std::move(name.fText),
999 std::move(decls),
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001000 std::move(instanceName.fText),
Ethan Nicholas50afc172017-02-16 14:49:57 -05001001 std::move(sizes)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001002}
1003
1004/* IF LPAREN expression RPAREN statement (ELSE statement)? */
1005std::unique_ptr<ASTIfStatement> Parser::ifStatement() {
1006 Token start;
Ethan Nicholas5ac13c22017-05-10 15:06:17 -04001007 bool isStatic = this->checkNext(Token::STATIC_IF, &start);
1008 if (!isStatic && !this->expect(Token::IF, "'if'", &start)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001009 return nullptr;
1010 }
1011 if (!this->expect(Token::LPAREN, "'('")) {
1012 return nullptr;
1013 }
1014 std::unique_ptr<ASTExpression> test(this->expression());
1015 if (!test) {
1016 return nullptr;
1017 }
1018 if (!this->expect(Token::RPAREN, "')'")) {
1019 return nullptr;
1020 }
1021 std::unique_ptr<ASTStatement> ifTrue(this->statement());
1022 if (!ifTrue) {
1023 return nullptr;
1024 }
1025 std::unique_ptr<ASTStatement> ifFalse;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001026 if (this->checkNext(Token::ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001027 ifFalse = this->statement();
1028 if (!ifFalse) {
1029 return nullptr;
1030 }
1031 }
Ethan Nicholas5ac13c22017-05-10 15:06:17 -04001032 return std::unique_ptr<ASTIfStatement>(new ASTIfStatement(start.fPosition,
1033 isStatic,
1034 std::move(test),
Ethan Nicholas11d53972016-11-28 11:23:23 -05001035 std::move(ifTrue),
ethannicholasb3058bd2016-07-01 08:22:01 -07001036 std::move(ifFalse)));
1037}
1038
1039/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
1040std::unique_ptr<ASTDoStatement> Parser::doStatement() {
1041 Token start;
1042 if (!this->expect(Token::DO, "'do'", &start)) {
1043 return nullptr;
1044 }
1045 std::unique_ptr<ASTStatement> statement(this->statement());
1046 if (!statement) {
1047 return nullptr;
1048 }
1049 if (!this->expect(Token::WHILE, "'while'")) {
1050 return nullptr;
1051 }
1052 if (!this->expect(Token::LPAREN, "'('")) {
1053 return nullptr;
1054 }
1055 std::unique_ptr<ASTExpression> test(this->expression());
1056 if (!test) {
1057 return nullptr;
1058 }
1059 if (!this->expect(Token::RPAREN, "')'")) {
1060 return nullptr;
1061 }
1062 if (!this->expect(Token::SEMICOLON, "';'")) {
1063 return nullptr;
1064 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001065 return std::unique_ptr<ASTDoStatement>(new ASTDoStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001066 std::move(statement),
1067 std::move(test)));
1068}
1069
1070/* WHILE LPAREN expression RPAREN STATEMENT */
1071std::unique_ptr<ASTWhileStatement> Parser::whileStatement() {
1072 Token start;
1073 if (!this->expect(Token::WHILE, "'while'", &start)) {
1074 return nullptr;
1075 }
1076 if (!this->expect(Token::LPAREN, "'('")) {
1077 return nullptr;
1078 }
1079 std::unique_ptr<ASTExpression> test(this->expression());
1080 if (!test) {
1081 return nullptr;
1082 }
1083 if (!this->expect(Token::RPAREN, "')'")) {
1084 return nullptr;
1085 }
1086 std::unique_ptr<ASTStatement> statement(this->statement());
1087 if (!statement) {
1088 return nullptr;
1089 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001090 return std::unique_ptr<ASTWhileStatement>(new ASTWhileStatement(start.fPosition,
1091 std::move(test),
ethannicholasb3058bd2016-07-01 08:22:01 -07001092 std::move(statement)));
1093}
1094
Ethan Nicholasaf197692017-02-27 13:26:45 -05001095/* CASE expression COLON statement* */
1096std::unique_ptr<ASTSwitchCase> Parser::switchCase() {
1097 Token start;
1098 if (!this->expect(Token::CASE, "'case'", &start)) {
1099 return nullptr;
1100 }
1101 std::unique_ptr<ASTExpression> value = this->expression();
1102 if (!value) {
1103 return nullptr;
1104 }
1105 if (!this->expect(Token::COLON, "':'")) {
1106 return nullptr;
1107 }
1108 std::vector<std::unique_ptr<ASTStatement>> statements;
1109 while (this->peek().fKind != Token::RBRACE && this->peek().fKind != Token::CASE &&
1110 this->peek().fKind != Token::DEFAULT) {
1111 std::unique_ptr<ASTStatement> s = this->statement();
1112 if (!s) {
1113 return nullptr;
1114 }
1115 statements.push_back(std::move(s));
1116 }
1117 return std::unique_ptr<ASTSwitchCase>(new ASTSwitchCase(start.fPosition, std::move(value),
1118 std::move(statements)));
1119}
1120
1121/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
1122std::unique_ptr<ASTStatement> Parser::switchStatement() {
1123 Token start;
Ethan Nicholas5ac13c22017-05-10 15:06:17 -04001124 bool isStatic = this->checkNext(Token::STATIC_SWITCH, &start);
1125 if (!isStatic && !this->expect(Token::SWITCH, "'switch'", &start)) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001126 return nullptr;
1127 }
1128 if (!this->expect(Token::LPAREN, "'('")) {
1129 return nullptr;
1130 }
1131 std::unique_ptr<ASTExpression> value(this->expression());
1132 if (!value) {
1133 return nullptr;
1134 }
1135 if (!this->expect(Token::RPAREN, "')'")) {
1136 return nullptr;
1137 }
1138 if (!this->expect(Token::LBRACE, "'{'")) {
1139 return nullptr;
1140 }
1141 std::vector<std::unique_ptr<ASTSwitchCase>> cases;
1142 while (this->peek().fKind == Token::CASE) {
1143 std::unique_ptr<ASTSwitchCase> c = this->switchCase();
1144 if (!c) {
1145 return nullptr;
1146 }
1147 cases.push_back(std::move(c));
1148 }
1149 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1150 // parts of the compiler may rely upon this assumption.
1151 if (this->peek().fKind == Token::DEFAULT) {
1152 Token defaultStart;
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001153 ASSERT_RESULT(this->expect(Token::DEFAULT, "'default'", &defaultStart));
Ethan Nicholasaf197692017-02-27 13:26:45 -05001154 if (!this->expect(Token::COLON, "':'")) {
1155 return nullptr;
1156 }
1157 std::vector<std::unique_ptr<ASTStatement>> statements;
1158 while (this->peek().fKind != Token::RBRACE) {
1159 std::unique_ptr<ASTStatement> s = this->statement();
1160 if (!s) {
1161 return nullptr;
1162 }
1163 statements.push_back(std::move(s));
1164 }
1165 cases.emplace_back(new ASTSwitchCase(defaultStart.fPosition, nullptr,
1166 std::move(statements)));
1167 }
1168 if (!this->expect(Token::RBRACE, "'}'")) {
1169 return nullptr;
1170 }
1171 return std::unique_ptr<ASTStatement>(new ASTSwitchStatement(start.fPosition,
Ethan Nicholas5ac13c22017-05-10 15:06:17 -04001172 isStatic,
Ethan Nicholasaf197692017-02-27 13:26:45 -05001173 std::move(value),
1174 std::move(cases)));
1175}
1176
Ethan Nicholas11d53972016-11-28 11:23:23 -05001177/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001178 STATEMENT */
1179std::unique_ptr<ASTForStatement> Parser::forStatement() {
1180 Token start;
1181 if (!this->expect(Token::FOR, "'for'", &start)) {
1182 return nullptr;
1183 }
1184 if (!this->expect(Token::LPAREN, "'('")) {
1185 return nullptr;
1186 }
1187 std::unique_ptr<ASTStatement> initializer;
1188 Token nextToken = this->peek();
1189 switch (nextToken.fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001190 case Token::SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001191 this->nextToken();
ethannicholasb3058bd2016-07-01 08:22:01 -07001192 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001193 case Token::CONST: {
1194 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
1195 if (!vd) {
1196 return nullptr;
1197 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001198 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001199 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001200 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001201 }
1202 case Token::IDENTIFIER: {
ethannicholasb3058bd2016-07-01 08:22:01 -07001203 if (this->isType(nextToken.fText)) {
ethannicholasa54401d2016-10-14 08:37:32 -07001204 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
1205 if (!vd) {
1206 return nullptr;
1207 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001208 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001209 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001210 break;
1211 }
ethannicholasa54401d2016-10-14 08:37:32 -07001212 } // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001213 default:
1214 initializer = this->expressionStatement();
1215 }
1216 std::unique_ptr<ASTExpression> test;
1217 if (this->peek().fKind != Token::SEMICOLON) {
1218 test = this->expression();
1219 if (!test) {
1220 return nullptr;
1221 }
1222 }
1223 if (!this->expect(Token::SEMICOLON, "';'")) {
1224 return nullptr;
1225 }
1226 std::unique_ptr<ASTExpression> next;
ethannicholas22f939e2016-10-13 13:25:34 -07001227 if (this->peek().fKind != Token::RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001228 next = this->expression();
1229 if (!next) {
1230 return nullptr;
1231 }
1232 }
1233 if (!this->expect(Token::RPAREN, "')'")) {
1234 return nullptr;
1235 }
1236 std::unique_ptr<ASTStatement> statement(this->statement());
1237 if (!statement) {
1238 return nullptr;
1239 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001240 return std::unique_ptr<ASTForStatement>(new ASTForStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001241 std::move(initializer),
1242 std::move(test), std::move(next),
1243 std::move(statement)));
1244}
1245
1246/* RETURN expression? SEMICOLON */
1247std::unique_ptr<ASTReturnStatement> Parser::returnStatement() {
1248 Token start;
1249 if (!this->expect(Token::RETURN, "'return'", &start)) {
1250 return nullptr;
1251 }
1252 std::unique_ptr<ASTExpression> expression;
1253 if (this->peek().fKind != Token::SEMICOLON) {
1254 expression = this->expression();
1255 if (!expression) {
1256 return nullptr;
1257 }
1258 }
1259 if (!this->expect(Token::SEMICOLON, "';'")) {
1260 return nullptr;
1261 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001262 return std::unique_ptr<ASTReturnStatement>(new ASTReturnStatement(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001263 std::move(expression)));
1264}
1265
1266/* BREAK SEMICOLON */
1267std::unique_ptr<ASTBreakStatement> Parser::breakStatement() {
1268 Token start;
1269 if (!this->expect(Token::BREAK, "'break'", &start)) {
1270 return nullptr;
1271 }
1272 if (!this->expect(Token::SEMICOLON, "';'")) {
1273 return nullptr;
1274 }
1275 return std::unique_ptr<ASTBreakStatement>(new ASTBreakStatement(start.fPosition));
1276}
1277
1278/* CONTINUE SEMICOLON */
1279std::unique_ptr<ASTContinueStatement> Parser::continueStatement() {
1280 Token start;
1281 if (!this->expect(Token::CONTINUE, "'continue'", &start)) {
1282 return nullptr;
1283 }
1284 if (!this->expect(Token::SEMICOLON, "';'")) {
1285 return nullptr;
1286 }
1287 return std::unique_ptr<ASTContinueStatement>(new ASTContinueStatement(start.fPosition));
1288}
1289
1290/* DISCARD SEMICOLON */
1291std::unique_ptr<ASTDiscardStatement> Parser::discardStatement() {
1292 Token start;
1293 if (!this->expect(Token::DISCARD, "'continue'", &start)) {
1294 return nullptr;
1295 }
1296 if (!this->expect(Token::SEMICOLON, "';'")) {
1297 return nullptr;
1298 }
1299 return std::unique_ptr<ASTDiscardStatement>(new ASTDiscardStatement(start.fPosition));
1300}
1301
1302/* LBRACE statement* RBRACE */
1303std::unique_ptr<ASTBlock> Parser::block() {
ethannicholascad64162016-10-27 10:54:02 -07001304 AutoDepth depth(this);
1305 if (!depth.checkValid()) {
1306 return nullptr;
1307 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001308 Token start;
1309 if (!this->expect(Token::LBRACE, "'{'", &start)) {
1310 return nullptr;
1311 }
1312 std::vector<std::unique_ptr<ASTStatement>> statements;
1313 for (;;) {
1314 switch (this->peek().fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001315 case Token::RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001316 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001317 return std::unique_ptr<ASTBlock>(new ASTBlock(start.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001318 std::move(statements)));
Ethan Nicholas11d53972016-11-28 11:23:23 -05001319 case Token::END_OF_FILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001320 this->error(this->peek().fPosition, "expected '}', but found end of file");
1321 return nullptr;
1322 default: {
1323 std::unique_ptr<ASTStatement> statement = this->statement();
1324 if (!statement) {
1325 return nullptr;
1326 }
1327 statements.push_back(std::move(statement));
1328 }
1329 }
1330 }
1331}
1332
1333/* expression SEMICOLON */
1334std::unique_ptr<ASTExpressionStatement> Parser::expressionStatement() {
1335 std::unique_ptr<ASTExpression> expr = this->expression();
1336 if (expr) {
1337 if (this->expect(Token::SEMICOLON, "';'")) {
1338 ASTExpressionStatement* result = new ASTExpressionStatement(std::move(expr));
1339 return std::unique_ptr<ASTExpressionStatement>(result);
1340 }
1341 }
1342 return nullptr;
1343}
1344
1345/* assignmentExpression */
1346std::unique_ptr<ASTExpression> Parser::expression() {
ethannicholascad64162016-10-27 10:54:02 -07001347 AutoDepth depth(this);
1348 if (!depth.checkValid()) {
1349 return nullptr;
1350 }
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001351 return this->commaExpression();
1352}
1353
1354/* assignmentExpression (COMMA assignmentExpression)* */
1355std::unique_ptr<ASTExpression> Parser::commaExpression() {
1356 std::unique_ptr<ASTExpression> result = this->assignmentExpression();
1357 if (!result) {
1358 return nullptr;
1359 }
1360 Token t;
1361 while (this->checkNext(Token::COMMA, &t)) {
1362 std::unique_ptr<ASTExpression> right = this->commaExpression();
1363 if (!right) {
1364 return nullptr;
1365 }
Brian Osman634624a2017-08-15 11:14:30 -04001366 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001367 }
1368 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001369}
1370
1371/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1372 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1373 assignmentExpression)*
1374 */
1375std::unique_ptr<ASTExpression> Parser::assignmentExpression() {
1376 std::unique_ptr<ASTExpression> result = this->ternaryExpression();
1377 if (!result) {
1378 return nullptr;
1379 }
1380 for (;;) {
1381 switch (this->peek().fKind) {
1382 case Token::EQ: // fall through
1383 case Token::STAREQ: // fall through
1384 case Token::SLASHEQ: // fall through
1385 case Token::PERCENTEQ: // fall through
1386 case Token::PLUSEQ: // fall through
1387 case Token::MINUSEQ: // fall through
1388 case Token::SHLEQ: // fall through
1389 case Token::SHREQ: // fall through
1390 case Token::BITWISEANDEQ: // fall through
1391 case Token::BITWISEXOREQ: // fall through
1392 case Token::BITWISEOREQ: // fall through
1393 case Token::LOGICALANDEQ: // fall through
1394 case Token::LOGICALXOREQ: // fall through
1395 case Token::LOGICALOREQ: {
1396 Token t = this->nextToken();
1397 std::unique_ptr<ASTExpression> right = this->assignmentExpression();
1398 if (!right) {
1399 return nullptr;
1400 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001401 result = std::unique_ptr<ASTExpression>(new ASTBinaryExpression(std::move(result),
Brian Osman634624a2017-08-15 11:14:30 -04001402 std::move(t),
ethannicholasb3058bd2016-07-01 08:22:01 -07001403 std::move(right)));
1404 }
1405 default:
1406 return result;
1407 }
1408 }
1409}
1410
1411/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
1412std::unique_ptr<ASTExpression> Parser::ternaryExpression() {
1413 std::unique_ptr<ASTExpression> result = this->logicalOrExpression();
1414 if (!result) {
1415 return nullptr;
1416 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001417 if (this->checkNext(Token::QUESTION)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001418 std::unique_ptr<ASTExpression> trueExpr = this->expression();
1419 if (!trueExpr) {
1420 return nullptr;
1421 }
1422 if (this->expect(Token::COLON, "':'")) {
1423 std::unique_ptr<ASTExpression> falseExpr = this->assignmentExpression();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001424 return std::unique_ptr<ASTExpression>(new ASTTernaryExpression(std::move(result),
1425 std::move(trueExpr),
ethannicholasb3058bd2016-07-01 08:22:01 -07001426 std::move(falseExpr)));
1427 }
1428 return nullptr;
1429 }
1430 return result;
1431}
1432
1433/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
1434std::unique_ptr<ASTExpression> Parser::logicalOrExpression() {
1435 std::unique_ptr<ASTExpression> result = this->logicalXorExpression();
1436 if (!result) {
1437 return nullptr;
1438 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001439 Token t;
1440 while (this->checkNext(Token::LOGICALOR, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001441 std::unique_ptr<ASTExpression> right = this->logicalXorExpression();
1442 if (!right) {
1443 return nullptr;
1444 }
Brian Osman634624a2017-08-15 11:14:30 -04001445 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001446 }
1447 return result;
1448}
1449
1450/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
1451std::unique_ptr<ASTExpression> Parser::logicalXorExpression() {
1452 std::unique_ptr<ASTExpression> result = this->logicalAndExpression();
1453 if (!result) {
1454 return nullptr;
1455 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001456 Token t;
1457 while (this->checkNext(Token::LOGICALXOR, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001458 std::unique_ptr<ASTExpression> right = this->logicalAndExpression();
1459 if (!right) {
1460 return nullptr;
1461 }
Brian Osman634624a2017-08-15 11:14:30 -04001462 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001463 }
1464 return result;
1465}
1466
1467/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
1468std::unique_ptr<ASTExpression> Parser::logicalAndExpression() {
1469 std::unique_ptr<ASTExpression> result = this->bitwiseOrExpression();
1470 if (!result) {
1471 return nullptr;
1472 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001473 Token t;
1474 while (this->checkNext(Token::LOGICALAND, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001475 std::unique_ptr<ASTExpression> right = this->bitwiseOrExpression();
1476 if (!right) {
1477 return nullptr;
1478 }
Brian Osman634624a2017-08-15 11:14:30 -04001479 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001480 }
1481 return result;
1482}
1483
1484/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
1485std::unique_ptr<ASTExpression> Parser::bitwiseOrExpression() {
1486 std::unique_ptr<ASTExpression> result = this->bitwiseXorExpression();
1487 if (!result) {
1488 return nullptr;
1489 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001490 Token t;
1491 while (this->checkNext(Token::BITWISEOR, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001492 std::unique_ptr<ASTExpression> right = this->bitwiseXorExpression();
1493 if (!right) {
1494 return nullptr;
1495 }
Brian Osman634624a2017-08-15 11:14:30 -04001496 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001497 }
1498 return result;
1499}
1500
1501/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
1502std::unique_ptr<ASTExpression> Parser::bitwiseXorExpression() {
1503 std::unique_ptr<ASTExpression> result = this->bitwiseAndExpression();
1504 if (!result) {
1505 return nullptr;
1506 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001507 Token t;
1508 while (this->checkNext(Token::BITWISEXOR, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001509 std::unique_ptr<ASTExpression> right = this->bitwiseAndExpression();
1510 if (!right) {
1511 return nullptr;
1512 }
Brian Osman634624a2017-08-15 11:14:30 -04001513 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001514 }
1515 return result;
1516}
1517
1518/* equalityExpression (BITWISEAND equalityExpression)* */
1519std::unique_ptr<ASTExpression> Parser::bitwiseAndExpression() {
1520 std::unique_ptr<ASTExpression> result = this->equalityExpression();
1521 if (!result) {
1522 return nullptr;
1523 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001524 Token t;
1525 while (this->checkNext(Token::BITWISEAND, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001526 std::unique_ptr<ASTExpression> right = this->equalityExpression();
1527 if (!right) {
1528 return nullptr;
1529 }
Brian Osman634624a2017-08-15 11:14:30 -04001530 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001531 }
1532 return result;
1533}
1534
1535/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
1536std::unique_ptr<ASTExpression> Parser::equalityExpression() {
1537 std::unique_ptr<ASTExpression> result = this->relationalExpression();
1538 if (!result) {
1539 return nullptr;
1540 }
1541 for (;;) {
1542 switch (this->peek().fKind) {
1543 case Token::EQEQ: // fall through
1544 case Token::NEQ: {
1545 Token t = this->nextToken();
1546 std::unique_ptr<ASTExpression> right = this->relationalExpression();
1547 if (!right) {
1548 return nullptr;
1549 }
Brian Osman634624a2017-08-15 11:14:30 -04001550 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001551 break;
1552 }
1553 default:
1554 return result;
1555 }
1556 }
1557}
1558
1559/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
1560std::unique_ptr<ASTExpression> Parser::relationalExpression() {
1561 std::unique_ptr<ASTExpression> result = this->shiftExpression();
1562 if (!result) {
1563 return nullptr;
1564 }
1565 for (;;) {
1566 switch (this->peek().fKind) {
1567 case Token::LT: // fall through
1568 case Token::GT: // fall through
1569 case Token::LTEQ: // fall through
1570 case Token::GTEQ: {
1571 Token t = this->nextToken();
1572 std::unique_ptr<ASTExpression> right = this->shiftExpression();
1573 if (!right) {
1574 return nullptr;
1575 }
Brian Osman634624a2017-08-15 11:14:30 -04001576 result.reset(new ASTBinaryExpression(std::move(result), std::move(t),
1577 std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001578 break;
1579 }
1580 default:
1581 return result;
1582 }
1583 }
1584}
1585
1586/* additiveExpression ((SHL | SHR) additiveExpression)* */
1587std::unique_ptr<ASTExpression> Parser::shiftExpression() {
1588 std::unique_ptr<ASTExpression> result = this->additiveExpression();
1589 if (!result) {
1590 return nullptr;
1591 }
1592 for (;;) {
1593 switch (this->peek().fKind) {
1594 case Token::SHL: // fall through
1595 case Token::SHR: {
1596 Token t = this->nextToken();
1597 std::unique_ptr<ASTExpression> right = this->additiveExpression();
1598 if (!right) {
1599 return nullptr;
1600 }
Brian Osman634624a2017-08-15 11:14:30 -04001601 result.reset(new ASTBinaryExpression(std::move(result), std::move(t),
1602 std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001603 break;
1604 }
1605 default:
1606 return result;
1607 }
1608 }
1609}
1610
1611/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
1612std::unique_ptr<ASTExpression> Parser::additiveExpression() {
1613 std::unique_ptr<ASTExpression> result = this->multiplicativeExpression();
1614 if (!result) {
1615 return nullptr;
1616 }
1617 for (;;) {
1618 switch (this->peek().fKind) {
1619 case Token::PLUS: // fall through
1620 case Token::MINUS: {
1621 Token t = this->nextToken();
1622 std::unique_ptr<ASTExpression> right = this->multiplicativeExpression();
1623 if (!right) {
1624 return nullptr;
1625 }
Brian Osman634624a2017-08-15 11:14:30 -04001626 result.reset(new ASTBinaryExpression(std::move(result), std::move(t),
1627 std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001628 break;
1629 }
1630 default:
1631 return result;
1632 }
1633 }
1634}
1635
1636/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
1637std::unique_ptr<ASTExpression> Parser::multiplicativeExpression() {
1638 std::unique_ptr<ASTExpression> result = this->unaryExpression();
1639 if (!result) {
1640 return nullptr;
1641 }
1642 for (;;) {
1643 switch (this->peek().fKind) {
1644 case Token::STAR: // fall through
1645 case Token::SLASH: // fall through
1646 case Token::PERCENT: {
1647 Token t = this->nextToken();
1648 std::unique_ptr<ASTExpression> right = this->unaryExpression();
1649 if (!right) {
1650 return nullptr;
1651 }
Brian Osman634624a2017-08-15 11:14:30 -04001652 result.reset(new ASTBinaryExpression(std::move(result), std::move(t),
1653 std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001654 break;
1655 }
1656 default:
1657 return result;
1658 }
1659 }
1660}
1661
1662/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
1663std::unique_ptr<ASTExpression> Parser::unaryExpression() {
1664 switch (this->peek().fKind) {
ethannicholas5961bc92016-10-12 06:39:56 -07001665 case Token::PLUS: // fall through
1666 case Token::MINUS: // fall through
1667 case Token::LOGICALNOT: // fall through
1668 case Token::BITWISENOT: // fall through
1669 case Token::PLUSPLUS: // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001670 case Token::MINUSMINUS: {
1671 Token t = this->nextToken();
1672 std::unique_ptr<ASTExpression> expr = this->unaryExpression();
1673 if (!expr) {
1674 return nullptr;
1675 }
Brian Osman634624a2017-08-15 11:14:30 -04001676 return std::unique_ptr<ASTExpression>(new ASTPrefixExpression(std::move(t),
1677 std::move(expr)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001678 }
1679 default:
1680 return this->postfixExpression();
1681 }
1682}
1683
1684/* term suffix* */
1685std::unique_ptr<ASTExpression> Parser::postfixExpression() {
1686 std::unique_ptr<ASTExpression> result = this->term();
1687 if (!result) {
1688 return nullptr;
1689 }
1690 for (;;) {
1691 switch (this->peek().fKind) {
1692 case Token::LBRACKET: // fall through
1693 case Token::DOT: // fall through
1694 case Token::LPAREN: // fall through
1695 case Token::PLUSPLUS: // fall through
1696 case Token::MINUSMINUS: {
1697 std::unique_ptr<ASTSuffix> s = this->suffix();
1698 if (!s) {
1699 return nullptr;
1700 }
1701 result.reset(new ASTSuffixExpression(std::move(result), std::move(s)));
1702 break;
1703 }
1704 default:
1705 return result;
1706 }
1707 }
1708}
1709
Ethan Nicholas11d53972016-11-28 11:23:23 -05001710/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
ethannicholasb3058bd2016-07-01 08:22:01 -07001711 PLUSPLUS | MINUSMINUS */
1712std::unique_ptr<ASTSuffix> Parser::suffix() {
1713 Token next = this->nextToken();
1714 switch (next.fKind) {
1715 case Token::LBRACKET: {
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001716 if (this->checkNext(Token::RBRACKET)) {
ethannicholas5961bc92016-10-12 06:39:56 -07001717 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(next.fPosition));
1718 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001719 std::unique_ptr<ASTExpression> e = this->expression();
1720 if (!e) {
1721 return nullptr;
1722 }
1723 this->expect(Token::RBRACKET, "']' to complete array access expression");
1724 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(std::move(e)));
1725 }
1726 case Token::DOT: {
1727 Position pos = this->peek().fPosition;
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001728 String text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001729 if (this->identifier(&text)) {
1730 return std::unique_ptr<ASTSuffix>(new ASTFieldSuffix(pos, std::move(text)));
1731 }
1732 return nullptr;
1733 }
1734 case Token::LPAREN: {
1735 std::vector<std::unique_ptr<ASTExpression>> parameters;
1736 if (this->peek().fKind != Token::RPAREN) {
1737 for (;;) {
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001738 std::unique_ptr<ASTExpression> expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001739 if (!expr) {
1740 return nullptr;
1741 }
1742 parameters.push_back(std::move(expr));
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001743 if (!this->checkNext(Token::COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001744 break;
1745 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001746 }
1747 }
1748 this->expect(Token::RPAREN, "')' to complete function parameters");
Ethan Nicholas11d53972016-11-28 11:23:23 -05001749 return std::unique_ptr<ASTSuffix>(new ASTCallSuffix(next.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001750 std::move(parameters)));
1751 }
1752 case Token::PLUSPLUS:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001753 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fPosition,
ethannicholasb3058bd2016-07-01 08:22:01 -07001754 ASTSuffix::kPostIncrement_Kind));
1755 case Token::MINUSMINUS:
1756 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fPosition,
1757 ASTSuffix::kPostDecrement_Kind));
1758 default: {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001759 this->error(next.fPosition, "expected expression suffix, but found '" + next.fText +
ethannicholasb3058bd2016-07-01 08:22:01 -07001760 "'\n");
1761 return nullptr;
1762 }
1763 }
1764}
1765
1766/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | '(' expression ')' */
1767std::unique_ptr<ASTExpression> Parser::term() {
1768 std::unique_ptr<ASTExpression> result;
1769 Token t = this->peek();
1770 switch (t.fKind) {
1771 case Token::IDENTIFIER: {
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001772 String text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001773 if (this->identifier(&text)) {
1774 result.reset(new ASTIdentifier(t.fPosition, std::move(text)));
1775 }
1776 break;
1777 }
1778 case Token::INT_LITERAL: {
1779 int64_t i;
1780 if (this->intLiteral(&i)) {
1781 result.reset(new ASTIntLiteral(t.fPosition, i));
1782 }
1783 break;
1784 }
1785 case Token::FLOAT_LITERAL: {
1786 double f;
1787 if (this->floatLiteral(&f)) {
1788 result.reset(new ASTFloatLiteral(t.fPosition, f));
1789 }
1790 break;
1791 }
1792 case Token::TRUE_LITERAL: // fall through
1793 case Token::FALSE_LITERAL: {
1794 bool b;
1795 if (this->boolLiteral(&b)) {
1796 result.reset(new ASTBoolLiteral(t.fPosition, b));
1797 }
1798 break;
1799 }
1800 case Token::LPAREN: {
1801 this->nextToken();
1802 result = this->expression();
1803 if (result) {
1804 this->expect(Token::RPAREN, "')' to complete expression");
1805 }
1806 break;
1807 }
1808 default:
1809 this->nextToken();
1810 this->error(t.fPosition, "expected expression, but found '" + t.fText + "'\n");
1811 result = nullptr;
1812 }
1813 return result;
1814}
1815
1816/* INT_LITERAL */
1817bool Parser::intLiteral(int64_t* dest) {
1818 Token t;
1819 if (this->expect(Token::INT_LITERAL, "integer literal", &t)) {
1820 *dest = SkSL::stol(t.fText);
1821 return true;
1822 }
1823 return false;
1824}
1825
1826/* FLOAT_LITERAL */
1827bool Parser::floatLiteral(double* dest) {
1828 Token t;
1829 if (this->expect(Token::FLOAT_LITERAL, "float literal", &t)) {
1830 *dest = SkSL::stod(t.fText);
1831 return true;
1832 }
1833 return false;
1834}
1835
1836/* TRUE_LITERAL | FALSE_LITERAL */
1837bool Parser::boolLiteral(bool* dest) {
1838 Token t = this->nextToken();
1839 switch (t.fKind) {
1840 case Token::TRUE_LITERAL:
1841 *dest = true;
1842 return true;
1843 case Token::FALSE_LITERAL:
1844 *dest = false;
1845 return true;
1846 default:
1847 this->error(t.fPosition, "expected 'true' or 'false', but found '" + t.fText + "'\n");
1848 return false;
1849 }
1850}
1851
1852/* IDENTIFIER */
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001853bool Parser::identifier(String* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001854 Token t;
1855 if (this->expect(Token::IDENTIFIER, "identifier", &t)) {
Brian Osman634624a2017-08-15 11:14:30 -04001856 *dest = std::move(t.fText);
ethannicholasb3058bd2016-07-01 08:22:01 -07001857 return true;
1858 }
1859 return false;
1860}
1861
1862} // namespace