blob: 9bfa62f2f55f2ea69a2e6ee8197bfad989bf6c6b [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"
ethannicholasb3058bd2016-07-01 08:22:01 -070010#include "ast/SkSLASTBinaryExpression.h"
11#include "ast/SkSLASTBlock.h"
12#include "ast/SkSLASTBoolLiteral.h"
13#include "ast/SkSLASTBreakStatement.h"
14#include "ast/SkSLASTCallSuffix.h"
15#include "ast/SkSLASTContinueStatement.h"
16#include "ast/SkSLASTDiscardStatement.h"
17#include "ast/SkSLASTDoStatement.h"
18#include "ast/SkSLASTExpression.h"
19#include "ast/SkSLASTExpressionStatement.h"
20#include "ast/SkSLASTExtension.h"
21#include "ast/SkSLASTFieldSuffix.h"
22#include "ast/SkSLASTFloatLiteral.h"
23#include "ast/SkSLASTForStatement.h"
24#include "ast/SkSLASTFunction.h"
25#include "ast/SkSLASTIdentifier.h"
26#include "ast/SkSLASTIfStatement.h"
27#include "ast/SkSLASTIndexSuffix.h"
28#include "ast/SkSLASTInterfaceBlock.h"
29#include "ast/SkSLASTIntLiteral.h"
ethannicholas5961bc92016-10-12 06:39:56 -070030#include "ast/SkSLASTModifiersDeclaration.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070031#include "ast/SkSLASTParameter.h"
32#include "ast/SkSLASTPrefixExpression.h"
33#include "ast/SkSLASTReturnStatement.h"
Ethan Nicholas762466e2017-06-29 10:03:38 -040034#include "ast/SkSLASTSection.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070035#include "ast/SkSLASTStatement.h"
36#include "ast/SkSLASTSuffixExpression.h"
Ethan Nicholasaf197692017-02-27 13:26:45 -050037#include "ast/SkSLASTSwitchCase.h"
38#include "ast/SkSLASTSwitchStatement.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070039#include "ast/SkSLASTTernaryExpression.h"
40#include "ast/SkSLASTType.h"
41#include "ast/SkSLASTVarDeclaration.h"
42#include "ast/SkSLASTVarDeclarationStatement.h"
43#include "ast/SkSLASTWhileStatement.h"
44#include "ir/SkSLSymbolTable.h"
Ethan Nicholas11d53972016-11-28 11:23:23 -050045#include "ir/SkSLModifiers.h"
ethannicholasd598f792016-07-25 10:08:54 -070046#include "ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070047
48namespace SkSL {
49
ethannicholascad64162016-10-27 10:54:02 -070050#define MAX_PARSE_DEPTH 50
51
52class AutoDepth {
53public:
54 AutoDepth(Parser* p)
55 : fParser(p) {
56 fParser->fDepth++;
57 }
58
59 ~AutoDepth() {
60 fParser->fDepth--;
61 }
62
63 bool checkValid() {
64 if (fParser->fDepth > MAX_PARSE_DEPTH) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070065 fParser->error(fParser->peek(), String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070066 return false;
67 }
68 return true;
69 }
70
71private:
72 Parser* fParser;
73};
74
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070075Parser::Parser(const char* text, size_t length, SymbolTable& types, ErrorReporter& errors)
76: fText(text)
77, fPushback(Token::INVALID, -1, -1)
ethannicholasb3058bd2016-07-01 08:22:01 -070078, fTypes(types)
79, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070080 fLexer.start(text, length);
ethannicholasb3058bd2016-07-01 08:22:01 -070081}
82
Ethan Nicholasf7b88202017-09-18 14:10:39 -040083/* (directive | section | declaration)* END_OF_FILE */
ethannicholasb3058bd2016-07-01 08:22:01 -070084std::vector<std::unique_ptr<ASTDeclaration>> Parser::file() {
85 std::vector<std::unique_ptr<ASTDeclaration>> result;
86 for (;;) {
87 switch (this->peek().fKind) {
88 case Token::END_OF_FILE:
89 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -070090 case Token::DIRECTIVE: {
91 std::unique_ptr<ASTDeclaration> decl = this->directive();
92 if (decl) {
93 result.push_back(std::move(decl));
94 }
95 break;
96 }
Ethan Nicholas762466e2017-06-29 10:03:38 -040097 case Token::SECTION: {
98 std::unique_ptr<ASTDeclaration> section = this->section();
99 if (section) {
100 result.push_back(std::move(section));
101 }
102 break;
103 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700104 default: {
105 std::unique_ptr<ASTDeclaration> decl = this->declaration();
106 if (!decl) {
107 continue;
108 }
109 result.push_back(std::move(decl));
110 }
111 }
112 }
113}
114
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700115Token Parser::nextRawToken() {
116 if (fPushback.fKind != Token::INVALID) {
117 Token result = fPushback;
118 fPushback.fKind = Token::INVALID;
ethannicholasb3058bd2016-07-01 08:22:01 -0700119 return result;
120 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700121 Token result = fLexer.next();
122 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400123}
124
125Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700126 Token token = this->nextRawToken();
127 while (token.fKind == Token::WHITESPACE || token.fKind == Token::LINE_COMMENT ||
128 token.fKind == Token::BLOCK_COMMENT) {
129 token = this->nextRawToken();
130 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400131 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700132}
133
134void Parser::pushback(Token t) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700135 ASSERT(fPushback.fKind == Token::INVALID);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400136 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700137}
138
139Token Parser::peek() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700140 if (fPushback.fKind == Token::INVALID) {
Brian Osman634624a2017-08-15 11:14:30 -0400141 fPushback = this->nextToken();
142 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700143 return fPushback;
144}
145
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400146bool Parser::checkNext(Token::Kind kind, Token* result) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700147 if (fPushback.fKind != Token::INVALID && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400148 return false;
149 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400150 Token next = this->nextToken();
151 if (next.fKind == kind) {
152 if (result) {
153 *result = next;
154 }
155 return true;
156 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400157 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400158 return false;
159}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500160
161bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700162 Token next = this->nextToken();
163 if (next.fKind == kind) {
164 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400165 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700166 }
167 return true;
168 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700169 this->error(next, "expected " + String(expected) + ", but found '" +
170 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700171 return false;
172 }
173}
174
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700175StringFragment Parser::text(Token token) {
176 return StringFragment(fText + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500177}
178
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700179void Parser::error(Token token, String msg) {
180 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700181}
182
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700183void Parser::error(int offset, String msg) {
184 fErrors.error(offset, msg);
185}
186
187bool Parser::isType(StringFragment name) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700188 return nullptr != fTypes[name];
189}
190
Ethan Nicholas11d53972016-11-28 11:23:23 -0500191/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700192 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
ethannicholasb3058bd2016-07-01 08:22:01 -0700193std::unique_ptr<ASTDeclaration> Parser::directive() {
194 Token start;
195 if (!this->expect(Token::DIRECTIVE, "a directive", &start)) {
196 return nullptr;
197 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700198 StringFragment text = this->text(start);
199 if (text == "#version") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700200 this->expect(Token::INT_LITERAL, "a version number");
ethannicholas5961bc92016-10-12 06:39:56 -0700201 Token next = this->peek();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700202 StringFragment nextText = this->text(next);
203 if (nextText == "es" || nextText == "compatibility") {
ethannicholas5961bc92016-10-12 06:39:56 -0700204 this->nextToken();
205 }
206 // version is ignored for now; it will eventually become an error when we stop pretending
207 // to be GLSL
ethannicholasb3058bd2016-07-01 08:22:01 -0700208 return nullptr;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700209 } else if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700210 Token name;
211 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
212 return nullptr;
213 }
214 if (!this->expect(Token::COLON, "':'")) {
215 return nullptr;
216 }
217 // FIXME: need to start paying attention to this token
218 if (!this->expect(Token::IDENTIFIER, "an identifier")) {
219 return nullptr;
220 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700221 return std::unique_ptr<ASTDeclaration>(new ASTExtension(start.fOffset,
222 String(this->text(name))));
ethannicholasb3058bd2016-07-01 08:22:01 -0700223 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700224 this->error(start, "unsupported directive '" + this->text(start) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700225 return nullptr;
226 }
227}
228
Ethan Nicholas762466e2017-06-29 10:03:38 -0400229/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
230 RBRACE */
231std::unique_ptr<ASTDeclaration> Parser::section() {
232 Token start;
233 if (!this->expect(Token::SECTION, "a section token", &start)) {
234 return nullptr;
235 }
236 String argument;
237 if (this->peek().fKind == Token::LPAREN) {
238 this->nextToken();
239 Token argToken;
240 if (!this->expect(Token::IDENTIFIER, "an identifier", &argToken)) {
241 return nullptr;
242 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700243 argument = this->text(argToken);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400244 if (!this->expect(Token::RPAREN, "')'")) {
245 return nullptr;
246 }
247 }
248 if (!this->expect(Token::LBRACE, "'{'")) {
249 return nullptr;
250 }
251 String text;
252 int level = 1;
253 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700254 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400255 switch (next.fKind) {
256 case Token::LBRACE:
257 ++level;
258 break;
259 case Token::RBRACE:
260 --level;
261 break;
262 case Token::END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700263 this->error(start, "reached end of file while parsing section");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400264 return nullptr;
265 default:
266 break;
267 }
268 if (!level) {
269 break;
270 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700271 text += this->text(next);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400272 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700273 StringFragment name = this->text(start);
274 ++name.fChars;
275 --name.fLength;
276 return std::unique_ptr<ASTDeclaration>(new ASTSection(start.fOffset,
277 String(name),
Ethan Nicholas762466e2017-06-29 10:03:38 -0400278 argument,
279 text));
280}
281
Ethan Nicholas11d53972016-11-28 11:23:23 -0500282/* modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700283 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
284std::unique_ptr<ASTDeclaration> Parser::declaration() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500285 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700286 Token lookahead = this->peek();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700287 if (lookahead.fKind == Token::IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700288 // we have an identifier that's not a type, could be the start of an interface block
289 return this->interfaceBlock(modifiers);
290 }
291 if (lookahead.fKind == Token::STRUCT) {
292 return this->structVarDeclaration(modifiers);
293 }
ethannicholas5961bc92016-10-12 06:39:56 -0700294 if (lookahead.fKind == Token::SEMICOLON) {
295 this->nextToken();
296 return std::unique_ptr<ASTDeclaration>(new ASTModifiersDeclaration(modifiers));
297 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700298 std::unique_ptr<ASTType> type(this->type());
299 if (!type) {
300 return nullptr;
301 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400302 if (type->fKind == ASTType::kStruct_Kind && this->checkNext(Token::SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700303 return nullptr;
304 }
305 Token name;
306 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
307 return nullptr;
308 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400309 if (this->checkNext(Token::LPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700310 std::vector<std::unique_ptr<ASTParameter>> parameters;
311 while (this->peek().fKind != Token::RPAREN) {
312 if (parameters.size() > 0) {
313 if (!this->expect(Token::COMMA, "','")) {
314 return nullptr;
315 }
316 }
317 std::unique_ptr<ASTParameter> parameter = this->parameter();
318 if (!parameter) {
319 return nullptr;
320 }
321 parameters.push_back(std::move(parameter));
322 }
323 this->nextToken();
324 std::unique_ptr<ASTBlock> body;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400325 if (!this->checkNext(Token::SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700326 body = this->block();
327 if (!body) {
328 return nullptr;
329 }
330 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700331 return std::unique_ptr<ASTDeclaration>(new ASTFunction(name.fOffset,
Ethan Nicholascb670962017-04-20 19:31:52 -0400332 modifiers,
333 std::move(type),
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700334 this->text(name),
Ethan Nicholas11d53972016-11-28 11:23:23 -0500335 std::move(parameters),
ethannicholasb3058bd2016-07-01 08:22:01 -0700336 std::move(body)));
337 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700338 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700339 }
340}
341
342/* modifiers type IDENTIFIER varDeclarationEnd */
ethannicholas14fe8cc2016-09-07 13:37:16 -0700343std::unique_ptr<ASTVarDeclarations> Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500344 Modifiers modifiers = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700345 std::unique_ptr<ASTType> type(this->type());
346 if (!type) {
347 return nullptr;
348 }
349 Token name;
350 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
351 return nullptr;
352 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700353 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700354}
355
356/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
357std::unique_ptr<ASTType> Parser::structDeclaration() {
358 if (!this->expect(Token::STRUCT, "'struct'")) {
359 return nullptr;
360 }
361 Token name;
362 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
363 return nullptr;
364 }
365 if (!this->expect(Token::LBRACE, "'{'")) {
366 return nullptr;
367 }
368 std::vector<Type::Field> fields;
369 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700370 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700371 if (!decl) {
372 return nullptr;
373 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700374 for (const auto& var : decl->fVars) {
ethannicholasd598f792016-07-25 10:08:54 -0700375 auto type = (const Type*) fTypes[decl->fType->fName];
ethannicholas14fe8cc2016-09-07 13:37:16 -0700376 for (int i = (int) var.fSizes.size() - 1; i >= 0; i--) {
ethannicholasdd4645b2016-10-14 12:14:46 -0700377 if (!var.fSizes[i] || var.fSizes[i]->fKind != ASTExpression::kInt_Kind) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700378 this->error(decl->fOffset, "array size in struct field must be a constant");
ethannicholasdd4645b2016-10-14 12:14:46 -0700379 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700380 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700381 uint64_t columns = ((ASTIntLiteral&) *var.fSizes[i]).fValue;
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400382 String name = type->name() + "[" + to_string(columns) + "]";
ethannicholasd598f792016-07-25 10:08:54 -0700383 type = new Type(name, Type::kArray_Kind, *type, (int) columns);
384 fTypes.takeOwnership((Type*) type);
ethannicholasb3058bd2016-07-01 08:22:01 -0700385 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700386 fields.push_back(Type::Field(decl->fModifiers, var.fName, type));
387 if (var.fValue) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700388 this->error(decl->fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700389 }
390 }
391 }
392 if (!this->expect(Token::RBRACE, "'}'")) {
393 return nullptr;
394 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700395 fTypes.add(this->text(name), std::unique_ptr<Type>(new Type(name.fOffset, this->text(name),
396 fields)));
397 return std::unique_ptr<ASTType>(new ASTType(name.fOffset, this->text(name),
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 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400407 Token name;
408 if (this->checkNext(Token::IDENTIFIER, &name)) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500409 std::unique_ptr<ASTVarDeclarations> result = this->varDeclarationEnd(modifiers,
410 std::move(type),
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700411 this->text(name));
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 Nicholas5b5f0962017-09-11 13:50:14 -0700415 this->error(var.fValue->fOffset,
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 Nicholas4b330df2017-05-17 10:52:55 -0400426/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
427 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* 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 Nicholas5b5f0962017-09-11 13:50:14 -0700430 StringFragment 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;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400433 while (this->checkNext(Token::LBRACKET)) {
434 if (this->checkNext(Token::RBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700435 currentVarSizes.push_back(nullptr);
436 } else {
437 std::unique_ptr<ASTExpression> size(this->expression());
438 if (!size) {
439 return nullptr;
440 }
441 currentVarSizes.push_back(std::move(size));
442 if (!this->expect(Token::RBRACKET, "']'")) {
443 return nullptr;
444 }
445 }
446 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700447 std::unique_ptr<ASTExpression> value;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400448 if (this->checkNext(Token::EQ)) {
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400449 value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700450 if (!value) {
451 return nullptr;
452 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700453 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700454 vars.emplace_back(name, std::move(currentVarSizes), std::move(value));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400455 while (this->checkNext(Token::COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700456 Token name;
457 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
458 return nullptr;
459 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700460 currentVarSizes.clear();
ethannicholas14fe8cc2016-09-07 13:37:16 -0700461 value.reset();
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400462 while (this->checkNext(Token::LBRACKET)) {
463 if (this->checkNext(Token::RBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700464 currentVarSizes.push_back(nullptr);
465 } else {
466 std::unique_ptr<ASTExpression> size(this->expression());
467 if (!size) {
468 return nullptr;
469 }
470 currentVarSizes.push_back(std::move(size));
471 if (!this->expect(Token::RBRACKET, "']'")) {
472 return nullptr;
473 }
474 }
475 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400476 if (this->checkNext(Token::EQ)) {
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400477 value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700478 if (!value) {
479 return nullptr;
480 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700481 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700482 vars.emplace_back(this->text(name), std::move(currentVarSizes), std::move(value));
ethannicholasb3058bd2016-07-01 08:22:01 -0700483 }
484 if (!this->expect(Token::SEMICOLON, "';'")) {
485 return nullptr;
486 }
ethannicholas14fe8cc2016-09-07 13:37:16 -0700487 return std::unique_ptr<ASTVarDeclarations>(new ASTVarDeclarations(std::move(mods),
488 std::move(type),
489 std::move(vars)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700490}
491
492/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
493std::unique_ptr<ASTParameter> Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400494 Modifiers modifiers = this->modifiersWithDefaults(0);
ethannicholasb3058bd2016-07-01 08:22:01 -0700495 std::unique_ptr<ASTType> type = this->type();
496 if (!type) {
497 return nullptr;
498 }
499 Token name;
500 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
501 return nullptr;
502 }
503 std::vector<int> sizes;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400504 while (this->checkNext(Token::LBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700505 Token sizeToken;
506 if (!this->expect(Token::INT_LITERAL, "a positive integer", &sizeToken)) {
507 return nullptr;
508 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700509 sizes.push_back(SkSL::stoi(this->text(sizeToken)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700510 if (!this->expect(Token::RBRACKET, "']'")) {
511 return nullptr;
512 }
513 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700514 return std::unique_ptr<ASTParameter>(new ASTParameter(name.fOffset, modifiers, std::move(type),
515 this->text(name), std::move(sizes)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700516}
517
518/** (EQ INT_LITERAL)? */
519int Parser::layoutInt() {
520 if (!this->expect(Token::EQ, "'='")) {
521 return -1;
522 }
523 Token resultToken;
524 if (this->expect(Token::INT_LITERAL, "a non-negative integer", &resultToken)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700525 return SkSL::stoi(this->text(resultToken));
ethannicholasb3058bd2016-07-01 08:22:01 -0700526 }
527 return -1;
528}
529
Ethan Nicholas762466e2017-06-29 10:03:38 -0400530/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
531String Parser::layoutCode() {
532 if (!this->expect(Token::EQ, "'='")) {
533 return "";
534 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700535 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400536 this->pushback(start);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400537 String code;
538 int level = 1;
539 bool done = false;
540 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700541 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400542 switch (next.fKind) {
543 case Token::LPAREN:
544 ++level;
545 break;
546 case Token::RPAREN:
547 --level;
548 break;
549 case Token::COMMA:
550 if (level == 1) {
551 done = true;
552 }
553 break;
554 case Token::END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700555 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400556 return nullptr;
557 default:
558 break;
559 }
560 if (!level) {
561 done = true;
562 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400563 if (done) {
564 this->pushback(std::move(next));
565 }
566 else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700567 code += this->text(next);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400568 }
569 }
570 return code;
571}
572
573/** (EQ IDENTIFIER('identity'))? */
574Layout::Key Parser::layoutKey() {
575 if (this->peek().fKind == Token::EQ) {
576 this->expect(Token::EQ, "'='");
577 Token key;
578 if (this->expect(Token::IDENTIFIER, "an identifer", &key)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700579 if (this->text(key) == "identity") {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400580 return Layout::kIdentity_Key;
581 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700582 this->error(key, "unsupported layout key");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400583 }
584 }
585 }
586 return Layout::kKey_Key;
587}
588
ethannicholas8ac838d2016-11-22 08:39:36 -0800589/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500590Layout Parser::layout() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700591 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500592 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700593 int binding = -1;
594 int index = -1;
595 int set = -1;
596 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500597 int inputAttachmentIndex = -1;
ethannicholasf789b382016-08-03 12:43:36 -0700598 bool originUpperLeft = false;
ethannicholas5961bc92016-10-12 06:39:56 -0700599 bool overrideCoverage = false;
600 bool blendSupportAllEquations = false;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500601 Layout::Format format = Layout::Format::kUnspecified;
ethannicholas8ac838d2016-11-22 08:39:36 -0800602 bool pushConstant = false;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500603 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
604 int maxVertices = -1;
605 int invocations = -1;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400606 String when;
607 Layout::Key key = Layout::kNo_Key;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400608 if (this->checkNext(Token::LAYOUT)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700609 if (!this->expect(Token::LPAREN, "'('")) {
Ethan Nicholas19671772016-11-28 16:30:17 -0500610 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
Ethan Nicholas11d53972016-11-28 11:23:23 -0500611 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas762466e2017-06-29 10:03:38 -0400612 pushConstant, primitive, maxVertices, invocations, when, key);
ethannicholasb3058bd2016-07-01 08:22:01 -0700613 }
614 for (;;) {
615 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700616 String text = this->text(t);
617 fLayoutLexer.start(text.c_str(), text.size());
618 int token = fLayoutLexer.next().fKind;
619 if (token != LayoutToken::INVALID) {
Ethan Nicholasbfe15f62017-03-01 11:46:51 -0500620 switch (token) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700621 case LayoutToken::LOCATION:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500622 location = this->layoutInt();
623 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700624 case LayoutToken::OFFSET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500625 offset = this->layoutInt();
626 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700627 case LayoutToken::BINDING:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500628 binding = this->layoutInt();
629 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700630 case LayoutToken::INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500631 index = this->layoutInt();
632 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700633 case LayoutToken::SET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500634 set = this->layoutInt();
635 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700636 case LayoutToken::BUILTIN:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500637 builtin = this->layoutInt();
638 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700639 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500640 inputAttachmentIndex = this->layoutInt();
641 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700642 case LayoutToken::ORIGIN_UPPER_LEFT:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500643 originUpperLeft = true;
644 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700645 case LayoutToken::OVERRIDE_COVERAGE:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500646 overrideCoverage = true;
647 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700648 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500649 blendSupportAllEquations = true;
650 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700651 case LayoutToken::PUSH_CONSTANT:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500652 pushConstant = true;
653 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700654 case LayoutToken::POINTS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500655 primitive = Layout::kPoints_Primitive;
656 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700657 case LayoutToken::LINES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500658 primitive = Layout::kLines_Primitive;
659 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700660 case LayoutToken::LINE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500661 primitive = Layout::kLineStrip_Primitive;
662 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700663 case LayoutToken::LINES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500664 primitive = Layout::kLinesAdjacency_Primitive;
665 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700666 case LayoutToken::TRIANGLES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500667 primitive = Layout::kTriangles_Primitive;
668 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700669 case LayoutToken::TRIANGLE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500670 primitive = Layout::kTriangleStrip_Primitive;
671 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700672 case LayoutToken::TRIANGLES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500673 primitive = Layout::kTrianglesAdjacency_Primitive;
674 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700675 case LayoutToken::MAX_VERTICES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500676 maxVertices = this->layoutInt();
677 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700678 case LayoutToken::INVOCATIONS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500679 invocations = this->layoutInt();
680 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700681 case LayoutToken::WHEN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400682 when = this->layoutCode();
683 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700684 case LayoutToken::KEY:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400685 key = this->layoutKey();
686 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700687 default:
688 ASSERT(false);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500689 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700690 } else if (Layout::ReadFormat(this->text(t), &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500691 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700692 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700693 this->error(t, ("'" + this->text(t) + "' is not a valid layout qualifier").c_str());
ethannicholasb3058bd2016-07-01 08:22:01 -0700694 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400695 if (this->checkNext(Token::RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700696 break;
697 }
698 if (!this->expect(Token::COMMA, "','")) {
699 break;
700 }
701 }
702 }
Ethan Nicholas19671772016-11-28 16:30:17 -0500703 return Layout(location, offset, binding, index, set, builtin, inputAttachmentIndex,
704 originUpperLeft, overrideCoverage, blendSupportAllEquations, format,
Ethan Nicholas762466e2017-06-29 10:03:38 -0400705 pushConstant, primitive, maxVertices, invocations, when, key);
ethannicholasb3058bd2016-07-01 08:22:01 -0700706}
707
Brian Salomonf9f45122016-11-29 11:59:17 -0500708/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400709 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT | BUFFER)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500710Modifiers Parser::modifiers() {
711 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700712 int flags = 0;
713 for (;;) {
714 // TODO: handle duplicate / incompatible flags
715 switch (peek().fKind) {
716 case Token::UNIFORM:
717 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500718 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700719 break;
720 case Token::CONST:
721 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500722 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700723 break;
724 case Token::IN:
725 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500726 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700727 break;
728 case Token::OUT:
729 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500730 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700731 break;
732 case Token::INOUT:
733 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500734 flags |= Modifiers::kIn_Flag;
735 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700736 break;
737 case Token::LOWP:
738 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500739 flags |= Modifiers::kLowp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700740 break;
741 case Token::MEDIUMP:
742 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500743 flags |= Modifiers::kMediump_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700744 break;
745 case Token::HIGHP:
746 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500747 flags |= Modifiers::kHighp_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700748 break;
ethannicholasf789b382016-08-03 12:43:36 -0700749 case Token::FLAT:
750 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500751 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700752 break;
753 case Token::NOPERSPECTIVE:
754 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500755 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700756 break;
Brian Salomonf9f45122016-11-29 11:59:17 -0500757 case Token::READONLY:
758 this->nextToken();
759 flags |= Modifiers::kReadOnly_Flag;
760 break;
761 case Token::WRITEONLY:
762 this->nextToken();
763 flags |= Modifiers::kWriteOnly_Flag;
764 break;
765 case Token::COHERENT:
766 this->nextToken();
767 flags |= Modifiers::kCoherent_Flag;
768 break;
769 case Token::VOLATILE:
770 this->nextToken();
771 flags |= Modifiers::kVolatile_Flag;
772 break;
773 case Token::RESTRICT:
774 this->nextToken();
775 flags |= Modifiers::kRestrict_Flag;
776 break;
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400777 case Token::BUFFER:
778 this->nextToken();
779 flags |= Modifiers::kBuffer_Flag;
780 break;
Ethan Nicholascb670962017-04-20 19:31:52 -0400781 case Token::HASSIDEEFFECTS:
782 this->nextToken();
783 flags |= Modifiers::kHasSideEffects_Flag;
784 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700785 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500786 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700787 }
788 }
789}
790
Ethan Nicholas11d53972016-11-28 11:23:23 -0500791Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
792 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -0700793 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500794 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -0700795 }
796 return result;
797}
798
799/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
800std::unique_ptr<ASTStatement> Parser::statement() {
801 Token start = this->peek();
802 switch (start.fKind) {
Ethan Nicholas5ac13c22017-05-10 15:06:17 -0400803 case Token::IF: // fall through
804 case Token::STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -0700805 return this->ifStatement();
806 case Token::FOR:
807 return this->forStatement();
808 case Token::DO:
809 return this->doStatement();
810 case Token::WHILE:
811 return this->whileStatement();
Ethan Nicholas5ac13c22017-05-10 15:06:17 -0400812 case Token::SWITCH: // fall through
813 case Token::STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -0500814 return this->switchStatement();
ethannicholasb3058bd2016-07-01 08:22:01 -0700815 case Token::RETURN:
816 return this->returnStatement();
817 case Token::BREAK:
818 return this->breakStatement();
819 case Token::CONTINUE:
820 return this->continueStatement();
821 case Token::DISCARD:
822 return this->discardStatement();
823 case Token::LBRACE:
824 return this->block();
825 case Token::SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -0500826 this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700827 return std::unique_ptr<ASTStatement>(new ASTBlock(start.fOffset,
ethannicholas0730be72016-09-01 07:59:02 -0700828 std::vector<std::unique_ptr<ASTStatement>>()));
ethannicholasb3058bd2016-07-01 08:22:01 -0700829 case Token::CONST: // fall through
830 case Token::HIGHP: // fall through
831 case Token::MEDIUMP: // fall through
832 case Token::LOWP: {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700833 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700834 if (!decl) {
835 return nullptr;
836 }
837 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(std::move(decl)));
838 }
839 case Token::IDENTIFIER:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700840 if (this->isType(this->text(start))) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700841 auto decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700842 if (!decl) {
843 return nullptr;
844 }
845 return std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
846 std::move(decl)));
847 }
848 // fall through
849 default:
850 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500851 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700852}
853
Ethan Nicholas50afc172017-02-16 14:49:57 -0500854/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* */
ethannicholasb3058bd2016-07-01 08:22:01 -0700855std::unique_ptr<ASTType> Parser::type() {
856 Token type;
857 if (!this->expect(Token::IDENTIFIER, "a type", &type)) {
858 return nullptr;
859 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700860 if (!this->isType(this->text(type))) {
861 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
ethannicholasb3058bd2016-07-01 08:22:01 -0700862 return nullptr;
863 }
Ethan Nicholas50afc172017-02-16 14:49:57 -0500864 std::vector<int> sizes;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400865 while (this->checkNext(Token::LBRACKET)) {
Ethan Nicholas50afc172017-02-16 14:49:57 -0500866 if (this->peek().fKind != Token::RBRACKET) {
867 int64_t i;
868 if (this->intLiteral(&i)) {
869 sizes.push_back(i);
870 } else {
871 return nullptr;
872 }
873 } else {
874 sizes.push_back(-1);
875 }
876 this->expect(Token::RBRACKET, "']'");
877 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700878 return std::unique_ptr<ASTType>(new ASTType(type.fOffset, this->text(type),
Ethan Nicholas50afc172017-02-16 14:49:57 -0500879 ASTType::kIdentifier_Kind, sizes));
ethannicholasb3058bd2016-07-01 08:22:01 -0700880}
881
Ethan Nicholas50afc172017-02-16 14:49:57 -0500882/* IDENTIFIER LBRACE varDeclaration* RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500883std::unique_ptr<ASTDeclaration> Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700884 Token name;
885 if (!this->expect(Token::IDENTIFIER, "an identifier", &name)) {
886 return nullptr;
887 }
888 if (peek().fKind != Token::LBRACE) {
889 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -0500890 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -0700891 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700892 this->error(name, "no type named '" + this->text(name) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700893 return nullptr;
894 }
895 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500896 std::vector<std::unique_ptr<ASTVarDeclarations>> decls;
ethannicholasb3058bd2016-07-01 08:22:01 -0700897 while (this->peek().fKind != Token::RBRACE) {
ethannicholas14fe8cc2016-09-07 13:37:16 -0700898 std::unique_ptr<ASTVarDeclarations> decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -0700899 if (!decl) {
900 return nullptr;
901 }
902 decls.push_back(std::move(decl));
903 }
904 this->nextToken();
Ethan Nicholas50afc172017-02-16 14:49:57 -0500905 std::vector<std::unique_ptr<ASTExpression>> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700906 StringFragment instanceName;
907 Token instanceNameToken;
908 if (this->checkNext(Token::IDENTIFIER, &instanceNameToken)) {
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400909 while (this->checkNext(Token::LBRACKET)) {
Ethan Nicholas50afc172017-02-16 14:49:57 -0500910 if (this->peek().fKind != Token::RBRACKET) {
911 std::unique_ptr<ASTExpression> size = this->expression();
912 if (!size) {
913 return nullptr;
914 }
915 sizes.push_back(std::move(size));
916 } else {
917 sizes.push_back(nullptr);
918 }
919 this->expect(Token::RBRACKET, "']'");
920 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700921 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -0700922 }
923 this->expect(Token::SEMICOLON, "';'");
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700924 return std::unique_ptr<ASTDeclaration>(new ASTInterfaceBlock(name.fOffset, mods,
925 this->text(name),
Brian Osman634624a2017-08-15 11:14:30 -0400926 std::move(decls),
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700927 instanceName,
Ethan Nicholas50afc172017-02-16 14:49:57 -0500928 std::move(sizes)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700929}
930
931/* IF LPAREN expression RPAREN statement (ELSE statement)? */
932std::unique_ptr<ASTIfStatement> Parser::ifStatement() {
933 Token start;
Ethan Nicholas5ac13c22017-05-10 15:06:17 -0400934 bool isStatic = this->checkNext(Token::STATIC_IF, &start);
935 if (!isStatic && !this->expect(Token::IF, "'if'", &start)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700936 return nullptr;
937 }
938 if (!this->expect(Token::LPAREN, "'('")) {
939 return nullptr;
940 }
941 std::unique_ptr<ASTExpression> test(this->expression());
942 if (!test) {
943 return nullptr;
944 }
945 if (!this->expect(Token::RPAREN, "')'")) {
946 return nullptr;
947 }
948 std::unique_ptr<ASTStatement> ifTrue(this->statement());
949 if (!ifTrue) {
950 return nullptr;
951 }
952 std::unique_ptr<ASTStatement> ifFalse;
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400953 if (this->checkNext(Token::ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700954 ifFalse = this->statement();
955 if (!ifFalse) {
956 return nullptr;
957 }
958 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700959 return std::unique_ptr<ASTIfStatement>(new ASTIfStatement(start.fOffset,
Ethan Nicholas5ac13c22017-05-10 15:06:17 -0400960 isStatic,
961 std::move(test),
Ethan Nicholas11d53972016-11-28 11:23:23 -0500962 std::move(ifTrue),
ethannicholasb3058bd2016-07-01 08:22:01 -0700963 std::move(ifFalse)));
964}
965
966/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
967std::unique_ptr<ASTDoStatement> Parser::doStatement() {
968 Token start;
969 if (!this->expect(Token::DO, "'do'", &start)) {
970 return nullptr;
971 }
972 std::unique_ptr<ASTStatement> statement(this->statement());
973 if (!statement) {
974 return nullptr;
975 }
976 if (!this->expect(Token::WHILE, "'while'")) {
977 return nullptr;
978 }
979 if (!this->expect(Token::LPAREN, "'('")) {
980 return nullptr;
981 }
982 std::unique_ptr<ASTExpression> test(this->expression());
983 if (!test) {
984 return nullptr;
985 }
986 if (!this->expect(Token::RPAREN, "')'")) {
987 return nullptr;
988 }
989 if (!this->expect(Token::SEMICOLON, "';'")) {
990 return nullptr;
991 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700992 return std::unique_ptr<ASTDoStatement>(new ASTDoStatement(start.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -0700993 std::move(statement),
994 std::move(test)));
995}
996
997/* WHILE LPAREN expression RPAREN STATEMENT */
998std::unique_ptr<ASTWhileStatement> Parser::whileStatement() {
999 Token start;
1000 if (!this->expect(Token::WHILE, "'while'", &start)) {
1001 return nullptr;
1002 }
1003 if (!this->expect(Token::LPAREN, "'('")) {
1004 return nullptr;
1005 }
1006 std::unique_ptr<ASTExpression> test(this->expression());
1007 if (!test) {
1008 return nullptr;
1009 }
1010 if (!this->expect(Token::RPAREN, "')'")) {
1011 return nullptr;
1012 }
1013 std::unique_ptr<ASTStatement> statement(this->statement());
1014 if (!statement) {
1015 return nullptr;
1016 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001017 return std::unique_ptr<ASTWhileStatement>(new ASTWhileStatement(start.fOffset,
Ethan Nicholas11d53972016-11-28 11:23:23 -05001018 std::move(test),
ethannicholasb3058bd2016-07-01 08:22:01 -07001019 std::move(statement)));
1020}
1021
Ethan Nicholasaf197692017-02-27 13:26:45 -05001022/* CASE expression COLON statement* */
1023std::unique_ptr<ASTSwitchCase> Parser::switchCase() {
1024 Token start;
1025 if (!this->expect(Token::CASE, "'case'", &start)) {
1026 return nullptr;
1027 }
1028 std::unique_ptr<ASTExpression> value = this->expression();
1029 if (!value) {
1030 return nullptr;
1031 }
1032 if (!this->expect(Token::COLON, "':'")) {
1033 return nullptr;
1034 }
1035 std::vector<std::unique_ptr<ASTStatement>> statements;
1036 while (this->peek().fKind != Token::RBRACE && this->peek().fKind != Token::CASE &&
1037 this->peek().fKind != Token::DEFAULT) {
1038 std::unique_ptr<ASTStatement> s = this->statement();
1039 if (!s) {
1040 return nullptr;
1041 }
1042 statements.push_back(std::move(s));
1043 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001044 return std::unique_ptr<ASTSwitchCase>(new ASTSwitchCase(start.fOffset, std::move(value),
Ethan Nicholasaf197692017-02-27 13:26:45 -05001045 std::move(statements)));
1046}
1047
1048/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
1049std::unique_ptr<ASTStatement> Parser::switchStatement() {
1050 Token start;
Ethan Nicholas5ac13c22017-05-10 15:06:17 -04001051 bool isStatic = this->checkNext(Token::STATIC_SWITCH, &start);
1052 if (!isStatic && !this->expect(Token::SWITCH, "'switch'", &start)) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001053 return nullptr;
1054 }
1055 if (!this->expect(Token::LPAREN, "'('")) {
1056 return nullptr;
1057 }
1058 std::unique_ptr<ASTExpression> value(this->expression());
1059 if (!value) {
1060 return nullptr;
1061 }
1062 if (!this->expect(Token::RPAREN, "')'")) {
1063 return nullptr;
1064 }
1065 if (!this->expect(Token::LBRACE, "'{'")) {
1066 return nullptr;
1067 }
1068 std::vector<std::unique_ptr<ASTSwitchCase>> cases;
1069 while (this->peek().fKind == Token::CASE) {
1070 std::unique_ptr<ASTSwitchCase> c = this->switchCase();
1071 if (!c) {
1072 return nullptr;
1073 }
1074 cases.push_back(std::move(c));
1075 }
1076 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1077 // parts of the compiler may rely upon this assumption.
1078 if (this->peek().fKind == Token::DEFAULT) {
1079 Token defaultStart;
Ethan Nicholas0df1b042017-03-31 13:56:23 -04001080 ASSERT_RESULT(this->expect(Token::DEFAULT, "'default'", &defaultStart));
Ethan Nicholasaf197692017-02-27 13:26:45 -05001081 if (!this->expect(Token::COLON, "':'")) {
1082 return nullptr;
1083 }
1084 std::vector<std::unique_ptr<ASTStatement>> statements;
1085 while (this->peek().fKind != Token::RBRACE) {
1086 std::unique_ptr<ASTStatement> s = this->statement();
1087 if (!s) {
1088 return nullptr;
1089 }
1090 statements.push_back(std::move(s));
1091 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001092 cases.emplace_back(new ASTSwitchCase(defaultStart.fOffset, nullptr,
Ethan Nicholasaf197692017-02-27 13:26:45 -05001093 std::move(statements)));
1094 }
1095 if (!this->expect(Token::RBRACE, "'}'")) {
1096 return nullptr;
1097 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001098 return std::unique_ptr<ASTStatement>(new ASTSwitchStatement(start.fOffset,
Ethan Nicholas5ac13c22017-05-10 15:06:17 -04001099 isStatic,
Ethan Nicholasaf197692017-02-27 13:26:45 -05001100 std::move(value),
1101 std::move(cases)));
1102}
1103
Ethan Nicholas11d53972016-11-28 11:23:23 -05001104/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001105 STATEMENT */
1106std::unique_ptr<ASTForStatement> Parser::forStatement() {
1107 Token start;
1108 if (!this->expect(Token::FOR, "'for'", &start)) {
1109 return nullptr;
1110 }
1111 if (!this->expect(Token::LPAREN, "'('")) {
1112 return nullptr;
1113 }
1114 std::unique_ptr<ASTStatement> initializer;
1115 Token nextToken = this->peek();
1116 switch (nextToken.fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001117 case Token::SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001118 this->nextToken();
ethannicholasb3058bd2016-07-01 08:22:01 -07001119 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001120 case Token::CONST: {
1121 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
1122 if (!vd) {
1123 return nullptr;
1124 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001125 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001126 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001127 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001128 }
1129 case Token::IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001130 if (this->isType(this->text(nextToken))) {
ethannicholasa54401d2016-10-14 08:37:32 -07001131 std::unique_ptr<ASTVarDeclarations> vd = this->varDeclarations();
1132 if (!vd) {
1133 return nullptr;
1134 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001135 initializer = std::unique_ptr<ASTStatement>(new ASTVarDeclarationStatement(
ethannicholasa54401d2016-10-14 08:37:32 -07001136 std::move(vd)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001137 break;
1138 }
ethannicholasa54401d2016-10-14 08:37:32 -07001139 } // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001140 default:
1141 initializer = this->expressionStatement();
1142 }
1143 std::unique_ptr<ASTExpression> test;
1144 if (this->peek().fKind != Token::SEMICOLON) {
1145 test = this->expression();
1146 if (!test) {
1147 return nullptr;
1148 }
1149 }
1150 if (!this->expect(Token::SEMICOLON, "';'")) {
1151 return nullptr;
1152 }
1153 std::unique_ptr<ASTExpression> next;
ethannicholas22f939e2016-10-13 13:25:34 -07001154 if (this->peek().fKind != Token::RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001155 next = this->expression();
1156 if (!next) {
1157 return nullptr;
1158 }
1159 }
1160 if (!this->expect(Token::RPAREN, "')'")) {
1161 return nullptr;
1162 }
1163 std::unique_ptr<ASTStatement> statement(this->statement());
1164 if (!statement) {
1165 return nullptr;
1166 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001167 return std::unique_ptr<ASTForStatement>(new ASTForStatement(start.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07001168 std::move(initializer),
1169 std::move(test), std::move(next),
1170 std::move(statement)));
1171}
1172
1173/* RETURN expression? SEMICOLON */
1174std::unique_ptr<ASTReturnStatement> Parser::returnStatement() {
1175 Token start;
1176 if (!this->expect(Token::RETURN, "'return'", &start)) {
1177 return nullptr;
1178 }
1179 std::unique_ptr<ASTExpression> expression;
1180 if (this->peek().fKind != Token::SEMICOLON) {
1181 expression = this->expression();
1182 if (!expression) {
1183 return nullptr;
1184 }
1185 }
1186 if (!this->expect(Token::SEMICOLON, "';'")) {
1187 return nullptr;
1188 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001189 return std::unique_ptr<ASTReturnStatement>(new ASTReturnStatement(start.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07001190 std::move(expression)));
1191}
1192
1193/* BREAK SEMICOLON */
1194std::unique_ptr<ASTBreakStatement> Parser::breakStatement() {
1195 Token start;
1196 if (!this->expect(Token::BREAK, "'break'", &start)) {
1197 return nullptr;
1198 }
1199 if (!this->expect(Token::SEMICOLON, "';'")) {
1200 return nullptr;
1201 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001202 return std::unique_ptr<ASTBreakStatement>(new ASTBreakStatement(start.fOffset));
ethannicholasb3058bd2016-07-01 08:22:01 -07001203}
1204
1205/* CONTINUE SEMICOLON */
1206std::unique_ptr<ASTContinueStatement> Parser::continueStatement() {
1207 Token start;
1208 if (!this->expect(Token::CONTINUE, "'continue'", &start)) {
1209 return nullptr;
1210 }
1211 if (!this->expect(Token::SEMICOLON, "';'")) {
1212 return nullptr;
1213 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001214 return std::unique_ptr<ASTContinueStatement>(new ASTContinueStatement(start.fOffset));
ethannicholasb3058bd2016-07-01 08:22:01 -07001215}
1216
1217/* DISCARD SEMICOLON */
1218std::unique_ptr<ASTDiscardStatement> Parser::discardStatement() {
1219 Token start;
1220 if (!this->expect(Token::DISCARD, "'continue'", &start)) {
1221 return nullptr;
1222 }
1223 if (!this->expect(Token::SEMICOLON, "';'")) {
1224 return nullptr;
1225 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001226 return std::unique_ptr<ASTDiscardStatement>(new ASTDiscardStatement(start.fOffset));
ethannicholasb3058bd2016-07-01 08:22:01 -07001227}
1228
1229/* LBRACE statement* RBRACE */
1230std::unique_ptr<ASTBlock> Parser::block() {
ethannicholascad64162016-10-27 10:54:02 -07001231 AutoDepth depth(this);
1232 if (!depth.checkValid()) {
1233 return nullptr;
1234 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001235 Token start;
1236 if (!this->expect(Token::LBRACE, "'{'", &start)) {
1237 return nullptr;
1238 }
1239 std::vector<std::unique_ptr<ASTStatement>> statements;
1240 for (;;) {
1241 switch (this->peek().fKind) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001242 case Token::RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001243 this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001244 return std::unique_ptr<ASTBlock>(new ASTBlock(start.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07001245 std::move(statements)));
Ethan Nicholas11d53972016-11-28 11:23:23 -05001246 case Token::END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001247 this->error(this->peek(), "expected '}', but found end of file");
ethannicholasb3058bd2016-07-01 08:22:01 -07001248 return nullptr;
1249 default: {
1250 std::unique_ptr<ASTStatement> statement = this->statement();
1251 if (!statement) {
1252 return nullptr;
1253 }
1254 statements.push_back(std::move(statement));
1255 }
1256 }
1257 }
1258}
1259
1260/* expression SEMICOLON */
1261std::unique_ptr<ASTExpressionStatement> Parser::expressionStatement() {
1262 std::unique_ptr<ASTExpression> expr = this->expression();
1263 if (expr) {
1264 if (this->expect(Token::SEMICOLON, "';'")) {
1265 ASTExpressionStatement* result = new ASTExpressionStatement(std::move(expr));
1266 return std::unique_ptr<ASTExpressionStatement>(result);
1267 }
1268 }
1269 return nullptr;
1270}
1271
1272/* assignmentExpression */
1273std::unique_ptr<ASTExpression> Parser::expression() {
ethannicholascad64162016-10-27 10:54:02 -07001274 AutoDepth depth(this);
1275 if (!depth.checkValid()) {
1276 return nullptr;
1277 }
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001278 return this->commaExpression();
1279}
1280
1281/* assignmentExpression (COMMA assignmentExpression)* */
1282std::unique_ptr<ASTExpression> Parser::commaExpression() {
1283 std::unique_ptr<ASTExpression> result = this->assignmentExpression();
1284 if (!result) {
1285 return nullptr;
1286 }
1287 Token t;
1288 while (this->checkNext(Token::COMMA, &t)) {
1289 std::unique_ptr<ASTExpression> right = this->commaExpression();
1290 if (!right) {
1291 return nullptr;
1292 }
Brian Osman634624a2017-08-15 11:14:30 -04001293 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001294 }
1295 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001296}
1297
1298/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1299 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1300 assignmentExpression)*
1301 */
1302std::unique_ptr<ASTExpression> Parser::assignmentExpression() {
1303 std::unique_ptr<ASTExpression> result = this->ternaryExpression();
1304 if (!result) {
1305 return nullptr;
1306 }
1307 for (;;) {
1308 switch (this->peek().fKind) {
1309 case Token::EQ: // fall through
1310 case Token::STAREQ: // fall through
1311 case Token::SLASHEQ: // fall through
1312 case Token::PERCENTEQ: // fall through
1313 case Token::PLUSEQ: // fall through
1314 case Token::MINUSEQ: // fall through
1315 case Token::SHLEQ: // fall through
1316 case Token::SHREQ: // fall through
1317 case Token::BITWISEANDEQ: // fall through
1318 case Token::BITWISEXOREQ: // fall through
1319 case Token::BITWISEOREQ: // fall through
1320 case Token::LOGICALANDEQ: // fall through
1321 case Token::LOGICALXOREQ: // fall through
1322 case Token::LOGICALOREQ: {
1323 Token t = this->nextToken();
1324 std::unique_ptr<ASTExpression> right = this->assignmentExpression();
1325 if (!right) {
1326 return nullptr;
1327 }
Ethan Nicholas11d53972016-11-28 11:23:23 -05001328 result = std::unique_ptr<ASTExpression>(new ASTBinaryExpression(std::move(result),
Brian Osman634624a2017-08-15 11:14:30 -04001329 std::move(t),
ethannicholasb3058bd2016-07-01 08:22:01 -07001330 std::move(right)));
1331 }
1332 default:
1333 return result;
1334 }
1335 }
1336}
1337
1338/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
1339std::unique_ptr<ASTExpression> Parser::ternaryExpression() {
1340 std::unique_ptr<ASTExpression> result = this->logicalOrExpression();
1341 if (!result) {
1342 return nullptr;
1343 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001344 if (this->checkNext(Token::QUESTION)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001345 std::unique_ptr<ASTExpression> trueExpr = this->expression();
1346 if (!trueExpr) {
1347 return nullptr;
1348 }
1349 if (this->expect(Token::COLON, "':'")) {
1350 std::unique_ptr<ASTExpression> falseExpr = this->assignmentExpression();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001351 return std::unique_ptr<ASTExpression>(new ASTTernaryExpression(std::move(result),
1352 std::move(trueExpr),
ethannicholasb3058bd2016-07-01 08:22:01 -07001353 std::move(falseExpr)));
1354 }
1355 return nullptr;
1356 }
1357 return result;
1358}
1359
1360/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
1361std::unique_ptr<ASTExpression> Parser::logicalOrExpression() {
1362 std::unique_ptr<ASTExpression> result = this->logicalXorExpression();
1363 if (!result) {
1364 return nullptr;
1365 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001366 Token t;
1367 while (this->checkNext(Token::LOGICALOR, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001368 std::unique_ptr<ASTExpression> right = this->logicalXorExpression();
1369 if (!right) {
1370 return nullptr;
1371 }
Brian Osman634624a2017-08-15 11:14:30 -04001372 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001373 }
1374 return result;
1375}
1376
1377/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
1378std::unique_ptr<ASTExpression> Parser::logicalXorExpression() {
1379 std::unique_ptr<ASTExpression> result = this->logicalAndExpression();
1380 if (!result) {
1381 return nullptr;
1382 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001383 Token t;
1384 while (this->checkNext(Token::LOGICALXOR, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001385 std::unique_ptr<ASTExpression> right = this->logicalAndExpression();
1386 if (!right) {
1387 return nullptr;
1388 }
Brian Osman634624a2017-08-15 11:14:30 -04001389 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001390 }
1391 return result;
1392}
1393
1394/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
1395std::unique_ptr<ASTExpression> Parser::logicalAndExpression() {
1396 std::unique_ptr<ASTExpression> result = this->bitwiseOrExpression();
1397 if (!result) {
1398 return nullptr;
1399 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001400 Token t;
1401 while (this->checkNext(Token::LOGICALAND, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001402 std::unique_ptr<ASTExpression> right = this->bitwiseOrExpression();
1403 if (!right) {
1404 return nullptr;
1405 }
Brian Osman634624a2017-08-15 11:14:30 -04001406 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001407 }
1408 return result;
1409}
1410
1411/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
1412std::unique_ptr<ASTExpression> Parser::bitwiseOrExpression() {
1413 std::unique_ptr<ASTExpression> result = this->bitwiseXorExpression();
1414 if (!result) {
1415 return nullptr;
1416 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001417 Token t;
1418 while (this->checkNext(Token::BITWISEOR, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001419 std::unique_ptr<ASTExpression> right = this->bitwiseXorExpression();
1420 if (!right) {
1421 return nullptr;
1422 }
Brian Osman634624a2017-08-15 11:14:30 -04001423 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001424 }
1425 return result;
1426}
1427
1428/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
1429std::unique_ptr<ASTExpression> Parser::bitwiseXorExpression() {
1430 std::unique_ptr<ASTExpression> result = this->bitwiseAndExpression();
1431 if (!result) {
1432 return nullptr;
1433 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001434 Token t;
1435 while (this->checkNext(Token::BITWISEXOR, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001436 std::unique_ptr<ASTExpression> right = this->bitwiseAndExpression();
1437 if (!right) {
1438 return nullptr;
1439 }
Brian Osman634624a2017-08-15 11:14:30 -04001440 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001441 }
1442 return result;
1443}
1444
1445/* equalityExpression (BITWISEAND equalityExpression)* */
1446std::unique_ptr<ASTExpression> Parser::bitwiseAndExpression() {
1447 std::unique_ptr<ASTExpression> result = this->equalityExpression();
1448 if (!result) {
1449 return nullptr;
1450 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001451 Token t;
1452 while (this->checkNext(Token::BITWISEAND, &t)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001453 std::unique_ptr<ASTExpression> right = this->equalityExpression();
1454 if (!right) {
1455 return nullptr;
1456 }
Brian Osman634624a2017-08-15 11:14:30 -04001457 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001458 }
1459 return result;
1460}
1461
1462/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
1463std::unique_ptr<ASTExpression> Parser::equalityExpression() {
1464 std::unique_ptr<ASTExpression> result = this->relationalExpression();
1465 if (!result) {
1466 return nullptr;
1467 }
1468 for (;;) {
1469 switch (this->peek().fKind) {
1470 case Token::EQEQ: // fall through
1471 case Token::NEQ: {
1472 Token t = this->nextToken();
1473 std::unique_ptr<ASTExpression> right = this->relationalExpression();
1474 if (!right) {
1475 return nullptr;
1476 }
Brian Osman634624a2017-08-15 11:14:30 -04001477 result.reset(new ASTBinaryExpression(std::move(result), std::move(t), std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001478 break;
1479 }
1480 default:
1481 return result;
1482 }
1483 }
1484}
1485
1486/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
1487std::unique_ptr<ASTExpression> Parser::relationalExpression() {
1488 std::unique_ptr<ASTExpression> result = this->shiftExpression();
1489 if (!result) {
1490 return nullptr;
1491 }
1492 for (;;) {
1493 switch (this->peek().fKind) {
1494 case Token::LT: // fall through
1495 case Token::GT: // fall through
1496 case Token::LTEQ: // fall through
1497 case Token::GTEQ: {
1498 Token t = this->nextToken();
1499 std::unique_ptr<ASTExpression> right = this->shiftExpression();
1500 if (!right) {
1501 return nullptr;
1502 }
Brian Osman634624a2017-08-15 11:14:30 -04001503 result.reset(new ASTBinaryExpression(std::move(result), std::move(t),
1504 std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001505 break;
1506 }
1507 default:
1508 return result;
1509 }
1510 }
1511}
1512
1513/* additiveExpression ((SHL | SHR) additiveExpression)* */
1514std::unique_ptr<ASTExpression> Parser::shiftExpression() {
1515 std::unique_ptr<ASTExpression> result = this->additiveExpression();
1516 if (!result) {
1517 return nullptr;
1518 }
1519 for (;;) {
1520 switch (this->peek().fKind) {
1521 case Token::SHL: // fall through
1522 case Token::SHR: {
1523 Token t = this->nextToken();
1524 std::unique_ptr<ASTExpression> right = this->additiveExpression();
1525 if (!right) {
1526 return nullptr;
1527 }
Brian Osman634624a2017-08-15 11:14:30 -04001528 result.reset(new ASTBinaryExpression(std::move(result), std::move(t),
1529 std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001530 break;
1531 }
1532 default:
1533 return result;
1534 }
1535 }
1536}
1537
1538/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
1539std::unique_ptr<ASTExpression> Parser::additiveExpression() {
1540 std::unique_ptr<ASTExpression> result = this->multiplicativeExpression();
1541 if (!result) {
1542 return nullptr;
1543 }
1544 for (;;) {
1545 switch (this->peek().fKind) {
1546 case Token::PLUS: // fall through
1547 case Token::MINUS: {
1548 Token t = this->nextToken();
1549 std::unique_ptr<ASTExpression> right = this->multiplicativeExpression();
1550 if (!right) {
1551 return nullptr;
1552 }
Brian Osman634624a2017-08-15 11:14:30 -04001553 result.reset(new ASTBinaryExpression(std::move(result), std::move(t),
1554 std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001555 break;
1556 }
1557 default:
1558 return result;
1559 }
1560 }
1561}
1562
1563/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
1564std::unique_ptr<ASTExpression> Parser::multiplicativeExpression() {
1565 std::unique_ptr<ASTExpression> result = this->unaryExpression();
1566 if (!result) {
1567 return nullptr;
1568 }
1569 for (;;) {
1570 switch (this->peek().fKind) {
1571 case Token::STAR: // fall through
1572 case Token::SLASH: // fall through
1573 case Token::PERCENT: {
1574 Token t = this->nextToken();
1575 std::unique_ptr<ASTExpression> right = this->unaryExpression();
1576 if (!right) {
1577 return nullptr;
1578 }
Brian Osman634624a2017-08-15 11:14:30 -04001579 result.reset(new ASTBinaryExpression(std::move(result), std::move(t),
1580 std::move(right)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001581 break;
1582 }
1583 default:
1584 return result;
1585 }
1586 }
1587}
1588
1589/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
1590std::unique_ptr<ASTExpression> Parser::unaryExpression() {
1591 switch (this->peek().fKind) {
ethannicholas5961bc92016-10-12 06:39:56 -07001592 case Token::PLUS: // fall through
1593 case Token::MINUS: // fall through
1594 case Token::LOGICALNOT: // fall through
1595 case Token::BITWISENOT: // fall through
1596 case Token::PLUSPLUS: // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001597 case Token::MINUSMINUS: {
1598 Token t = this->nextToken();
1599 std::unique_ptr<ASTExpression> expr = this->unaryExpression();
1600 if (!expr) {
1601 return nullptr;
1602 }
Brian Osman634624a2017-08-15 11:14:30 -04001603 return std::unique_ptr<ASTExpression>(new ASTPrefixExpression(std::move(t),
1604 std::move(expr)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001605 }
1606 default:
1607 return this->postfixExpression();
1608 }
1609}
1610
1611/* term suffix* */
1612std::unique_ptr<ASTExpression> Parser::postfixExpression() {
1613 std::unique_ptr<ASTExpression> result = this->term();
1614 if (!result) {
1615 return nullptr;
1616 }
1617 for (;;) {
1618 switch (this->peek().fKind) {
1619 case Token::LBRACKET: // fall through
1620 case Token::DOT: // fall through
1621 case Token::LPAREN: // fall through
1622 case Token::PLUSPLUS: // fall through
1623 case Token::MINUSMINUS: {
1624 std::unique_ptr<ASTSuffix> s = this->suffix();
1625 if (!s) {
1626 return nullptr;
1627 }
1628 result.reset(new ASTSuffixExpression(std::move(result), std::move(s)));
1629 break;
1630 }
1631 default:
1632 return result;
1633 }
1634 }
1635}
1636
Ethan Nicholas11d53972016-11-28 11:23:23 -05001637/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
ethannicholasb3058bd2016-07-01 08:22:01 -07001638 PLUSPLUS | MINUSMINUS */
1639std::unique_ptr<ASTSuffix> Parser::suffix() {
1640 Token next = this->nextToken();
1641 switch (next.fKind) {
1642 case Token::LBRACKET: {
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001643 if (this->checkNext(Token::RBRACKET)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001644 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(next.fOffset));
ethannicholas5961bc92016-10-12 06:39:56 -07001645 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001646 std::unique_ptr<ASTExpression> e = this->expression();
1647 if (!e) {
1648 return nullptr;
1649 }
1650 this->expect(Token::RBRACKET, "']' to complete array access expression");
1651 return std::unique_ptr<ASTSuffix>(new ASTIndexSuffix(std::move(e)));
1652 }
1653 case Token::DOT: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001654 int offset = this->peek().fOffset;
1655 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001656 if (this->identifier(&text)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001657 return std::unique_ptr<ASTSuffix>(new ASTFieldSuffix(offset, std::move(text)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001658 }
1659 return nullptr;
1660 }
1661 case Token::LPAREN: {
1662 std::vector<std::unique_ptr<ASTExpression>> parameters;
1663 if (this->peek().fKind != Token::RPAREN) {
1664 for (;;) {
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001665 std::unique_ptr<ASTExpression> expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001666 if (!expr) {
1667 return nullptr;
1668 }
1669 parameters.push_back(std::move(expr));
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001670 if (!this->checkNext(Token::COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001671 break;
1672 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001673 }
1674 }
1675 this->expect(Token::RPAREN, "')' to complete function parameters");
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001676 return std::unique_ptr<ASTSuffix>(new ASTCallSuffix(next.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07001677 std::move(parameters)));
1678 }
1679 case Token::PLUSPLUS:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001680 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07001681 ASTSuffix::kPostIncrement_Kind));
1682 case Token::MINUSMINUS:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001683 return std::unique_ptr<ASTSuffix>(new ASTSuffix(next.fOffset,
ethannicholasb3058bd2016-07-01 08:22:01 -07001684 ASTSuffix::kPostDecrement_Kind));
1685 default: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001686 this->error(next, "expected expression suffix, but found '" + this->text(next) +
ethannicholasb3058bd2016-07-01 08:22:01 -07001687 "'\n");
1688 return nullptr;
1689 }
1690 }
1691}
1692
1693/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | '(' expression ')' */
1694std::unique_ptr<ASTExpression> Parser::term() {
1695 std::unique_ptr<ASTExpression> result;
1696 Token t = this->peek();
1697 switch (t.fKind) {
1698 case Token::IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001699 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001700 if (this->identifier(&text)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001701 result.reset(new ASTIdentifier(t.fOffset, std::move(text)));
ethannicholasb3058bd2016-07-01 08:22:01 -07001702 }
1703 break;
1704 }
1705 case Token::INT_LITERAL: {
1706 int64_t i;
1707 if (this->intLiteral(&i)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001708 result.reset(new ASTIntLiteral(t.fOffset, i));
ethannicholasb3058bd2016-07-01 08:22:01 -07001709 }
1710 break;
1711 }
1712 case Token::FLOAT_LITERAL: {
1713 double f;
1714 if (this->floatLiteral(&f)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001715 result.reset(new ASTFloatLiteral(t.fOffset, f));
ethannicholasb3058bd2016-07-01 08:22:01 -07001716 }
1717 break;
1718 }
1719 case Token::TRUE_LITERAL: // fall through
1720 case Token::FALSE_LITERAL: {
1721 bool b;
1722 if (this->boolLiteral(&b)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001723 result.reset(new ASTBoolLiteral(t.fOffset, b));
ethannicholasb3058bd2016-07-01 08:22:01 -07001724 }
1725 break;
1726 }
1727 case Token::LPAREN: {
1728 this->nextToken();
1729 result = this->expression();
1730 if (result) {
1731 this->expect(Token::RPAREN, "')' to complete expression");
1732 }
1733 break;
1734 }
1735 default:
1736 this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001737 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'\n");
ethannicholasb3058bd2016-07-01 08:22:01 -07001738 result = nullptr;
1739 }
1740 return result;
1741}
1742
1743/* INT_LITERAL */
1744bool Parser::intLiteral(int64_t* dest) {
1745 Token t;
1746 if (this->expect(Token::INT_LITERAL, "integer literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001747 *dest = SkSL::stol(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07001748 return true;
1749 }
1750 return false;
1751}
1752
1753/* FLOAT_LITERAL */
1754bool Parser::floatLiteral(double* dest) {
1755 Token t;
1756 if (this->expect(Token::FLOAT_LITERAL, "float literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001757 *dest = SkSL::stod(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07001758 return true;
1759 }
1760 return false;
1761}
1762
1763/* TRUE_LITERAL | FALSE_LITERAL */
1764bool Parser::boolLiteral(bool* dest) {
1765 Token t = this->nextToken();
1766 switch (t.fKind) {
1767 case Token::TRUE_LITERAL:
1768 *dest = true;
1769 return true;
1770 case Token::FALSE_LITERAL:
1771 *dest = false;
1772 return true;
1773 default:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001774 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'\n");
ethannicholasb3058bd2016-07-01 08:22:01 -07001775 return false;
1776 }
1777}
1778
1779/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001780bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001781 Token t;
1782 if (this->expect(Token::IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001783 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07001784 return true;
1785 }
1786 return false;
1787}
1788
1789} // namespace