blob: d7e19ea87ea32559e4ee0fc14c82f1648ded4362 [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
Mike Klein4b432fa2019-06-06 11:44:05 -05008#include "src/sksl/SkSLParser.h"
John Stilesfbd050b2020-08-03 13:21:46 -04009
10#include <memory>
11#include "stdio.h"
12
13#include "src/sksl/SkSLASTNode.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050014#include "src/sksl/ir/SkSLModifiers.h"
15#include "src/sksl/ir/SkSLSymbolTable.h"
16#include "src/sksl/ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070017
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040018#ifndef SKSL_STANDALONE
Mike Kleinc0bd9f92019-04-23 12:05:21 -050019#include "include/private/SkOnce.h"
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040020#endif
21
ethannicholasb3058bd2016-07-01 08:22:01 -070022namespace SkSL {
23
ethannicholascad64162016-10-27 10:54:02 -070024#define MAX_PARSE_DEPTH 50
25
26class AutoDepth {
27public:
28 AutoDepth(Parser* p)
Ethan Nicholascf4deab2019-09-13 16:28:14 -040029 : fParser(p)
30 , fDepth(0) {}
ethannicholascad64162016-10-27 10:54:02 -070031
32 ~AutoDepth() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -040033 fParser->fDepth -= fDepth;
ethannicholascad64162016-10-27 10:54:02 -070034 }
35
Ethan Nicholascf4deab2019-09-13 16:28:14 -040036 bool increase() {
37 ++fDepth;
38 ++fParser->fDepth;
ethannicholascad64162016-10-27 10:54:02 -070039 if (fParser->fDepth > MAX_PARSE_DEPTH) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070040 fParser->error(fParser->peek(), String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070041 return false;
42 }
43 return true;
44 }
45
46private:
47 Parser* fParser;
Ethan Nicholascf4deab2019-09-13 16:28:14 -040048 int fDepth;
ethannicholascad64162016-10-27 10:54:02 -070049};
50
Brian Salomon140f3da2018-08-23 13:51:27 +000051std::unordered_map<String, Parser::LayoutToken>* Parser::layoutTokens;
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040052
53void Parser::InitLayoutMap() {
Brian Salomon140f3da2018-08-23 13:51:27 +000054 layoutTokens = new std::unordered_map<String, LayoutToken>;
Brian Salomon23356442018-11-30 15:33:19 -050055 #define TOKEN(name, text) (*layoutTokens)[text] = LayoutToken::name
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040056 TOKEN(LOCATION, "location");
57 TOKEN(OFFSET, "offset");
58 TOKEN(BINDING, "binding");
59 TOKEN(INDEX, "index");
60 TOKEN(SET, "set");
61 TOKEN(BUILTIN, "builtin");
62 TOKEN(INPUT_ATTACHMENT_INDEX, "input_attachment_index");
63 TOKEN(ORIGIN_UPPER_LEFT, "origin_upper_left");
64 TOKEN(OVERRIDE_COVERAGE, "override_coverage");
65 TOKEN(BLEND_SUPPORT_ALL_EQUATIONS, "blend_support_all_equations");
66 TOKEN(BLEND_SUPPORT_MULTIPLY, "blend_support_multiply");
67 TOKEN(BLEND_SUPPORT_SCREEN, "blend_support_screen");
68 TOKEN(BLEND_SUPPORT_OVERLAY, "blend_support_overlay");
69 TOKEN(BLEND_SUPPORT_DARKEN, "blend_support_darken");
70 TOKEN(BLEND_SUPPORT_LIGHTEN, "blend_support_lighten");
71 TOKEN(BLEND_SUPPORT_COLORDODGE, "blend_support_colordodge");
72 TOKEN(BLEND_SUPPORT_COLORBURN, "blend_support_colorburn");
73 TOKEN(BLEND_SUPPORT_HARDLIGHT, "blend_support_hardlight");
74 TOKEN(BLEND_SUPPORT_SOFTLIGHT, "blend_support_softlight");
75 TOKEN(BLEND_SUPPORT_DIFFERENCE, "blend_support_difference");
76 TOKEN(BLEND_SUPPORT_EXCLUSION, "blend_support_exclusion");
77 TOKEN(BLEND_SUPPORT_HSL_HUE, "blend_support_hsl_hue");
78 TOKEN(BLEND_SUPPORT_HSL_SATURATION, "blend_support_hsl_saturation");
79 TOKEN(BLEND_SUPPORT_HSL_COLOR, "blend_support_hsl_color");
80 TOKEN(BLEND_SUPPORT_HSL_LUMINOSITY, "blend_support_hsl_luminosity");
81 TOKEN(PUSH_CONSTANT, "push_constant");
82 TOKEN(POINTS, "points");
83 TOKEN(LINES, "lines");
84 TOKEN(LINE_STRIP, "line_strip");
85 TOKEN(LINES_ADJACENCY, "lines_adjacency");
86 TOKEN(TRIANGLES, "triangles");
87 TOKEN(TRIANGLE_STRIP, "triangle_strip");
88 TOKEN(TRIANGLES_ADJACENCY, "triangles_adjacency");
89 TOKEN(MAX_VERTICES, "max_vertices");
90 TOKEN(INVOCATIONS, "invocations");
Brian Osmanf59a9612020-04-15 14:18:13 -040091 TOKEN(MARKER, "marker");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040092 TOKEN(WHEN, "when");
93 TOKEN(KEY, "key");
Michael Ludwiga4275592018-08-31 10:52:47 -040094 TOKEN(TRACKED, "tracked");
Brian Osmanb32d66b2020-04-30 17:12:03 -040095 TOKEN(SRGB_UNPREMUL, "srgb_unpremul");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040096 TOKEN(CTYPE, "ctype");
Brian Osmanf28e55d2018-10-03 16:35:54 -040097 TOKEN(SKPMCOLOR4F, "SkPMColor4f");
Mike Reedb26b4e72020-01-22 14:31:21 -050098 TOKEN(SKV4, "SkV4");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040099 TOKEN(SKRECT, "SkRect");
100 TOKEN(SKIRECT, "SkIRect");
101 TOKEN(SKPMCOLOR, "SkPMColor");
Mike Reedb26b4e72020-01-22 14:31:21 -0500102 TOKEN(SKM44, "SkM44");
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400103 TOKEN(BOOL, "bool");
104 TOKEN(INT, "int");
105 TOKEN(FLOAT, "float");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400106 #undef TOKEN
107}
108
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400109Parser::Parser(const char* text, size_t length, SymbolTable& symbols, ErrorReporter& errors)
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700110: fText(text)
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400111, fPushback(Token::Kind::TK_INVALID, -1, -1)
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400112, fSymbols(symbols)
ethannicholasb3058bd2016-07-01 08:22:01 -0700113, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700114 fLexer.start(text, length);
Brian Salomon3b83afe2018-08-23 11:04:36 -0400115 static const bool layoutMapInitialized = []{ return (void)InitLayoutMap(), true; }();
116 (void) layoutMapInitialized;
ethannicholasb3058bd2016-07-01 08:22:01 -0700117}
118
Ethan Nicholasfc994162019-06-06 10:04:27 -0400119#define CREATE_NODE(result, ...) \
120 ASTNode::ID result(fFile->fNodes.size()); \
121 fFile->fNodes.emplace_back(&fFile->fNodes, __VA_ARGS__)
122
123#define RETURN_NODE(...) \
124 do { \
125 CREATE_NODE(result, __VA_ARGS__); \
126 return result; \
127 } while (false)
128
129#define CREATE_CHILD(child, target, ...) \
130 CREATE_NODE(child, __VA_ARGS__); \
131 fFile->fNodes[target.fValue].addChild(child)
132
133#define CREATE_EMPTY_CHILD(target) \
134 do { \
135 ASTNode::ID child(fFile->fNodes.size()); \
136 fFile->fNodes.emplace_back(); \
137 fFile->fNodes[target.fValue].addChild(child); \
138 } while (false)
139
Ethan Nicholasf7b88202017-09-18 14:10:39 -0400140/* (directive | section | declaration)* END_OF_FILE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400141std::unique_ptr<ASTFile> Parser::file() {
John Stilesfbd050b2020-08-03 13:21:46 -0400142 fFile = std::make_unique<ASTFile>();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400143 CREATE_NODE(result, 0, ASTNode::Kind::kFile);
144 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700145 for (;;) {
146 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400147 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400148 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400149 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400150 ASTNode::ID dir = this->directive();
151 if (fErrors.errorCount()) {
152 return nullptr;
153 }
154 if (dir) {
155 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700156 }
157 break;
158 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400159 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400160 ASTNode::ID section = this->section();
161 if (fErrors.errorCount()) {
162 return nullptr;
163 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400164 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400165 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400166 }
167 break;
168 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700169 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400170 ASTNode::ID decl = this->declaration();
171 if (fErrors.errorCount()) {
172 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700173 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400174 if (decl) {
175 getNode(result).addChild(decl);
176 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700177 }
178 }
179 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400180 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700181}
182
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700183Token Parser::nextRawToken() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400184 if (fPushback.fKind != Token::Kind::TK_INVALID) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700185 Token result = fPushback;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400186 fPushback.fKind = Token::Kind::TK_INVALID;
ethannicholasb3058bd2016-07-01 08:22:01 -0700187 return result;
188 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700189 Token result = fLexer.next();
190 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400191}
192
193Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700194 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400195 while (token.fKind == Token::Kind::TK_WHITESPACE ||
196 token.fKind == Token::Kind::TK_LINE_COMMENT ||
197 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700198 token = this->nextRawToken();
199 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400200 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700201}
202
203void Parser::pushback(Token t) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400204 SkASSERT(fPushback.fKind == Token::Kind::TK_INVALID);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400205 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700206}
207
208Token Parser::peek() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400209 if (fPushback.fKind == Token::Kind::TK_INVALID) {
Brian Osman634624a2017-08-15 11:14:30 -0400210 fPushback = this->nextToken();
211 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700212 return fPushback;
213}
214
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400215bool Parser::checkNext(Token::Kind kind, Token* result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400216 if (fPushback.fKind != Token::Kind::TK_INVALID && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400217 return false;
218 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400219 Token next = this->nextToken();
220 if (next.fKind == kind) {
221 if (result) {
222 *result = next;
223 }
224 return true;
225 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400226 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400227 return false;
228}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500229
230bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700231 Token next = this->nextToken();
232 if (next.fKind == kind) {
233 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400234 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700235 }
236 return true;
237 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700238 this->error(next, "expected " + String(expected) + ", but found '" +
239 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700240 return false;
241 }
242}
243
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700244StringFragment Parser::text(Token token) {
245 return StringFragment(fText + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500246}
247
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700248void Parser::error(Token token, String msg) {
249 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700250}
251
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700252void Parser::error(int offset, String msg) {
253 fErrors.error(offset, msg);
254}
255
256bool Parser::isType(StringFragment name) {
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400257 const Symbol* s = fSymbols[name];
258 return s && s->fKind == Symbol::kType_Kind;
ethannicholasb3058bd2016-07-01 08:22:01 -0700259}
260
Ethan Nicholas11d53972016-11-28 11:23:23 -0500261/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700262 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400263ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700264 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400265 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400266 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700267 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700268 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400269 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700270 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400271 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400272 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700273 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400274 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400275 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700276 }
277 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400278 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400279 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700280 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400281 RETURN_NODE(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700282 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700283 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400284 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700285 }
286}
287
Ethan Nicholas762466e2017-06-29 10:03:38 -0400288/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
289 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400290ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400291 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400292 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400293 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400294 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400295 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400296 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400297 this->nextToken();
298 Token argToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400299 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400300 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400301 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700302 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400303 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400304 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400305 }
306 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400307 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400308 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400309 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400310 StringFragment text;
311 Token codeStart = this->nextRawToken();
312 size_t startOffset = codeStart.fOffset;
313 this->pushback(codeStart);
314 text.fChars = fText + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400315 int level = 1;
316 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700317 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400318 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400319 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400320 ++level;
321 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400322 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400323 --level;
324 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400325 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700326 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400327 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400328 default:
329 break;
330 }
331 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400332 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400333 break;
334 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400335 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700336 StringFragment name = this->text(start);
337 ++name.fChars;
338 --name.fLength;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400339 RETURN_NODE(start.fOffset, ASTNode::Kind::kSection,
340 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400341}
342
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500343/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
344 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400345ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500346 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400347 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400348 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500349 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400350 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400351 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500352 }
353 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400354 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400355 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500356 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400357 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400358 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500359 }
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400360 fSymbols.add(this->text(name), std::unique_ptr<Symbol>(new Type(this->text(name),
361 Type::kEnum_Kind)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400362 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400363 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500364 Token id;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400365 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400366 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500367 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400368 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400369 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500370 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400371 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500372 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400373 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
374 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500375 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400376 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500377 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400378 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
379 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400380 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500381 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400382 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400383 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500384 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400385 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400386 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500387 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400388 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500389 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400390 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
391 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500392 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400393 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500394 }
395 }
396 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400397 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400398 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500399}
400
401/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700402 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400403ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700404 Token lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400405 if (lookahead.fKind == Token::Kind::TK_ENUM) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500406 return this->enumDeclaration();
407 }
408 Modifiers modifiers = this->modifiers();
409 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400410 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700411 // we have an identifier that's not a type, could be the start of an interface block
412 return this->interfaceBlock(modifiers);
413 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400414 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700415 return this->structVarDeclaration(modifiers);
416 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400417 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700418 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400419 RETURN_NODE(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700420 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400421 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700422 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400423 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700424 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400425 if (getNode(type).getTypeData().fIsStructDeclaration &&
426 this->checkNext(Token::Kind::TK_SEMICOLON)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400427 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700428 }
429 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400430 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400431 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700432 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400433 if (this->checkNext(Token::Kind::TK_LPAREN)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400434 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kFunction);
435 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
436 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400437 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400438 for (;;) {
439 ASTNode::ID parameter = this->parameter();
440 if (!parameter) {
441 return ASTNode::ID::Invalid();
442 }
443 ++fd.fParameterCount;
444 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400445 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400446 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700447 }
448 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700449 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400450 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400451 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400452 return ASTNode::ID::Invalid();
453 }
454 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400455 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700456 body = this->block();
457 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400458 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700459 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400460 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700461 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400462 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700463 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400464 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700465 }
466}
467
468/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400469ASTNode::ID Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500470 Modifiers modifiers = this->modifiers();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400471 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700472 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400473 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700474 }
475 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400476 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400477 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700478 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400479 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700480}
481
482/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400483ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400484 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400485 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700486 }
487 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400488 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400489 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700490 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400491 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400492 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700493 }
494 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400495 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400496 ASTNode::ID decls = this->varDeclarations();
497 if (!decls) {
498 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700499 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400500 ASTNode& declsNode = getNode(decls);
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400501 const Symbol* symbol = fSymbols[(declsNode.begin() + 1)->getTypeData().fName];
502 SkASSERT(symbol && symbol->fKind == Symbol::kType_Kind);
503 const Type* type = (const Type*) symbol;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400504 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
505 ASTNode& var = *iter;
506 ASTNode::VarData vd = var.getVarData();
507 for (int j = vd.fSizeCount - 1; j >= 0; j--) {
508 const ASTNode& size = *(var.begin() + j);
509 if (!size || size.fKind != ASTNode::Kind::kInt) {
510 this->error(declsNode.fOffset, "array size in struct field must be a constant");
511 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700512 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400513 uint64_t columns = size.getInt();
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400514 String name = type->name() + "[" + to_string(columns) + "]";
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400515 type = (Type*) fSymbols.takeOwnership(std::unique_ptr<const Symbol>(
Ethan Nicholas91164d12019-05-15 15:29:54 -0400516 new Type(name,
517 Type::kArray_Kind,
518 *type,
519 (int) columns)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700520 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400521 fields.push_back(Type::Field(declsNode.begin()->getModifiers(), vd.fName, type));
522 if (vd.fSizeCount ? (var.begin() + (vd.fSizeCount - 1))->fNext : var.fFirstChild) {
523 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700524 }
525 }
526 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400527 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400528 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700529 }
John Stilesfbd050b2020-08-03 13:21:46 -0400530 fSymbols.add(this->text(name), std::make_unique<Type>(name.fOffset, this->text(name), fields));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400531 RETURN_NODE(name.fOffset, ASTNode::Kind::kType,
532 ASTNode::TypeData(this->text(name), true, false));
ethannicholasb3058bd2016-07-01 08:22:01 -0700533}
534
535/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400536ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
537 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700538 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400539 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700540 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400541 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400542 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400543 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700544 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400545 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400546 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700547}
548
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400549/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
550 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400551ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
552 CREATE_NODE(result, -1, ASTNode::Kind::kVarDeclarations);
553 CREATE_CHILD(modifiers, result, -1, ASTNode::Kind::kModifiers, mods);
554 getNode(result).addChild(type);
555 CREATE_NODE(currentVar, -1, ASTNode::Kind::kVarDeclaration);
556 ASTNode::VarData vd(name, 0);
557 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400558 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
559 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400560 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700561 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400562 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700563 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400564 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700565 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400566 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400567 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400568 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700569 }
570 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400571 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700572 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400573 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400574 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400575 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700576 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400577 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700578 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400579 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700580 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400581 while (this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700582 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400583 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400584 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700585 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400586 currentVar = ASTNode::ID(fFile->fNodes.size());
587 vd = ASTNode::VarData(this->text(name), 0);
588 fFile->fNodes.emplace_back(&fFile->fNodes, -1, ASTNode::Kind::kVarDeclaration);
589 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400590 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
591 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400592 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700593 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400594 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700595 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400596 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700597 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400598 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400599 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400600 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700601 }
602 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400603 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700604 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400605 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400606 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400607 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700608 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400609 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700610 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400611 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700612 }
613 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400614 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400615 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700616 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400617 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700618}
619
620/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400621ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400622 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400623 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700624 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400625 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700626 }
627 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400628 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400629 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700630 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400631 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kParameter);
632 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
633 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400634 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700635 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400636 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400637 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700638 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400639 CREATE_CHILD(child, result, sizeToken.fOffset, ASTNode::Kind::kInt,
640 SkSL::stoi(this->text(sizeToken)));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400641 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400642 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700643 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400644 ++pd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700645 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400646 getNode(result).setParameterData(pd);
647 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700648}
649
Ethan Nicholasd608c092017-10-26 09:30:08 -0400650/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700651int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400652 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700653 return -1;
654 }
655 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400656 if (this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700657 return SkSL::stoi(this->text(resultToken));
ethannicholasb3058bd2016-07-01 08:22:01 -0700658 }
659 return -1;
660}
661
Ethan Nicholasd608c092017-10-26 09:30:08 -0400662/** EQ IDENTIFIER */
663StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400664 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400665 return StringFragment();
666 }
667 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400668 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400669 return StringFragment();
670 }
671 return this->text(resultToken);
672}
673
674
Ethan Nicholas762466e2017-06-29 10:03:38 -0400675/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400676StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400677 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400678 return "";
679 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700680 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400681 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400682 StringFragment code;
683 code.fChars = fText + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400684 int level = 1;
685 bool done = false;
686 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700687 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400688 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400689 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400690 ++level;
691 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400692 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400693 --level;
694 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400695 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400696 if (level == 1) {
697 done = true;
698 }
699 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400700 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700701 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400702 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400703 default:
704 break;
705 }
706 if (!level) {
707 done = true;
708 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400709 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400710 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400711 this->pushback(std::move(next));
712 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400713 }
714 return code;
715}
716
717/** (EQ IDENTIFIER('identity'))? */
718Layout::Key Parser::layoutKey() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400719 if (this->peek().fKind == Token::Kind::TK_EQ) {
720 this->expect(Token::Kind::TK_EQ, "'='");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400721 Token key;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400722 if (this->expect(Token::Kind::TK_IDENTIFIER, "an identifer", &key)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700723 if (this->text(key) == "identity") {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400724 return Layout::kIdentity_Key;
725 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700726 this->error(key, "unsupported layout key");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400727 }
728 }
729 }
730 return Layout::kKey_Key;
731}
732
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400733Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400734 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400735 Token t = this->nextToken();
736 String text = this->text(t);
737 auto found = layoutTokens->find(text);
738 if (found != layoutTokens->end()) {
739 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400740 case LayoutToken::SKPMCOLOR4F:
741 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500742 case LayoutToken::SKV4:
743 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400744 case LayoutToken::SKRECT:
745 return Layout::CType::kSkRect;
746 case LayoutToken::SKIRECT:
747 return Layout::CType::kSkIRect;
748 case LayoutToken::SKPMCOLOR:
749 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400750 case LayoutToken::BOOL:
751 return Layout::CType::kBool;
752 case LayoutToken::INT:
753 return Layout::CType::kInt32;
754 case LayoutToken::FLOAT:
755 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500756 case LayoutToken::SKM44:
757 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400758 default:
759 break;
760 }
761 }
762 this->error(t, "unsupported ctype");
763 }
764 return Layout::CType::kDefault;
765}
766
ethannicholas8ac838d2016-11-22 08:39:36 -0800767/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500768Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500769 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700770 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500771 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700772 int binding = -1;
773 int index = -1;
774 int set = -1;
775 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500776 int inputAttachmentIndex = -1;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500777 Layout::Format format = Layout::Format::kUnspecified;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500778 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
779 int maxVertices = -1;
780 int invocations = -1;
Brian Osmanf59a9612020-04-15 14:18:13 -0400781 StringFragment marker;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400782 StringFragment when;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400783 Layout::Key key = Layout::kNo_Key;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400784 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400785 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
786 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500787 return Layout(flags, location, offset, binding, index, set, builtin,
Brian Osmanf59a9612020-04-15 14:18:13 -0400788 inputAttachmentIndex, format, primitive, maxVertices, invocations, marker,
789 when, key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700790 }
791 for (;;) {
792 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700793 String text = this->text(t);
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400794 auto found = layoutTokens->find(text);
795 if (found != layoutTokens->end()) {
796 switch (found->second) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700797 case LayoutToken::LOCATION:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500798 location = this->layoutInt();
799 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700800 case LayoutToken::OFFSET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500801 offset = this->layoutInt();
802 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700803 case LayoutToken::BINDING:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500804 binding = this->layoutInt();
805 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700806 case LayoutToken::INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500807 index = this->layoutInt();
808 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700809 case LayoutToken::SET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500810 set = this->layoutInt();
811 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700812 case LayoutToken::BUILTIN:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500813 builtin = this->layoutInt();
814 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700815 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500816 inputAttachmentIndex = this->layoutInt();
817 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700818 case LayoutToken::ORIGIN_UPPER_LEFT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500819 flags |= Layout::kOriginUpperLeft_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500820 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700821 case LayoutToken::OVERRIDE_COVERAGE:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500822 flags |= Layout::kOverrideCoverage_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500823 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700824 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500825 flags |= Layout::kBlendSupportAllEquations_Flag;
826 break;
827 case LayoutToken::BLEND_SUPPORT_MULTIPLY:
828 flags |= Layout::kBlendSupportMultiply_Flag;
829 break;
830 case LayoutToken::BLEND_SUPPORT_SCREEN:
831 flags |= Layout::kBlendSupportScreen_Flag;
832 break;
833 case LayoutToken::BLEND_SUPPORT_OVERLAY:
834 flags |= Layout::kBlendSupportOverlay_Flag;
835 break;
836 case LayoutToken::BLEND_SUPPORT_DARKEN:
837 flags |= Layout::kBlendSupportDarken_Flag;
838 break;
839 case LayoutToken::BLEND_SUPPORT_LIGHTEN:
840 flags |= Layout::kBlendSupportLighten_Flag;
841 break;
842 case LayoutToken::BLEND_SUPPORT_COLORDODGE:
843 flags |= Layout::kBlendSupportColorDodge_Flag;
844 break;
845 case LayoutToken::BLEND_SUPPORT_COLORBURN:
846 flags |= Layout::kBlendSupportColorBurn_Flag;
847 break;
848 case LayoutToken::BLEND_SUPPORT_HARDLIGHT:
849 flags |= Layout::kBlendSupportHardLight_Flag;
850 break;
851 case LayoutToken::BLEND_SUPPORT_SOFTLIGHT:
852 flags |= Layout::kBlendSupportSoftLight_Flag;
853 break;
854 case LayoutToken::BLEND_SUPPORT_DIFFERENCE:
855 flags |= Layout::kBlendSupportDifference_Flag;
856 break;
857 case LayoutToken::BLEND_SUPPORT_EXCLUSION:
858 flags |= Layout::kBlendSupportExclusion_Flag;
859 break;
860 case LayoutToken::BLEND_SUPPORT_HSL_HUE:
861 flags |= Layout::kBlendSupportHSLHue_Flag;
862 break;
863 case LayoutToken::BLEND_SUPPORT_HSL_SATURATION:
864 flags |= Layout::kBlendSupportHSLSaturation_Flag;
865 break;
866 case LayoutToken::BLEND_SUPPORT_HSL_COLOR:
867 flags |= Layout::kBlendSupportHSLColor_Flag;
868 break;
869 case LayoutToken::BLEND_SUPPORT_HSL_LUMINOSITY:
870 flags |= Layout::kBlendSupportHSLLuminosity_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500871 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700872 case LayoutToken::PUSH_CONSTANT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500873 flags |= Layout::kPushConstant_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500874 break;
Michael Ludwiga4275592018-08-31 10:52:47 -0400875 case LayoutToken::TRACKED:
876 flags |= Layout::kTracked_Flag;
877 break;
Brian Osmanb32d66b2020-04-30 17:12:03 -0400878 case LayoutToken::SRGB_UNPREMUL:
879 flags |= Layout::kSRGBUnpremul_Flag;
880 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700881 case LayoutToken::POINTS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500882 primitive = Layout::kPoints_Primitive;
883 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700884 case LayoutToken::LINES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500885 primitive = Layout::kLines_Primitive;
886 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700887 case LayoutToken::LINE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500888 primitive = Layout::kLineStrip_Primitive;
889 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700890 case LayoutToken::LINES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500891 primitive = Layout::kLinesAdjacency_Primitive;
892 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700893 case LayoutToken::TRIANGLES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500894 primitive = Layout::kTriangles_Primitive;
895 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700896 case LayoutToken::TRIANGLE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500897 primitive = Layout::kTriangleStrip_Primitive;
898 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700899 case LayoutToken::TRIANGLES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500900 primitive = Layout::kTrianglesAdjacency_Primitive;
901 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700902 case LayoutToken::MAX_VERTICES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500903 maxVertices = this->layoutInt();
904 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700905 case LayoutToken::INVOCATIONS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500906 invocations = this->layoutInt();
907 break;
Brian Osmanf59a9612020-04-15 14:18:13 -0400908 case LayoutToken::MARKER:
909 marker = this->layoutCode();
910 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700911 case LayoutToken::WHEN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400912 when = this->layoutCode();
913 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700914 case LayoutToken::KEY:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400915 key = this->layoutKey();
916 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -0400917 case LayoutToken::CTYPE:
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400918 ctype = this->layoutCType();
919 break;
920 default:
921 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
Ethan Nicholasd608c092017-10-26 09:30:08 -0400922 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500923 }
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400924 } else if (Layout::ReadFormat(text, &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500925 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700926 } else {
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400927 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
ethannicholasb3058bd2016-07-01 08:22:01 -0700928 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400929 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700930 break;
931 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400932 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700933 break;
934 }
935 }
936 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500937 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
Brian Osmanf59a9612020-04-15 14:18:13 -0400938 format, primitive, maxVertices, invocations, marker, when, key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700939}
940
Brian Salomonf9f45122016-11-29 11:59:17 -0500941/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Ethan Nicholasa7ceb502019-01-11 10:31:48 -0500942 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT | BUFFER | PLS | PLSIN |
Brian Osman3c358422020-03-23 10:44:12 -0400943 PLSOUT | VARYING)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500944Modifiers Parser::modifiers() {
945 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700946 int flags = 0;
947 for (;;) {
948 // TODO: handle duplicate / incompatible flags
949 switch (peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400950 case Token::Kind::TK_UNIFORM:
ethannicholasb3058bd2016-07-01 08:22:01 -0700951 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500952 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700953 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400954 case Token::Kind::TK_CONST:
ethannicholasb3058bd2016-07-01 08:22:01 -0700955 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500956 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700957 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400958 case Token::Kind::TK_IN:
ethannicholasb3058bd2016-07-01 08:22:01 -0700959 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500960 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700961 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400962 case Token::Kind::TK_OUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700963 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500964 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700965 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400966 case Token::Kind::TK_INOUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700967 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500968 flags |= Modifiers::kIn_Flag;
969 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700970 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400971 case Token::Kind::TK_FLAT:
ethannicholasf789b382016-08-03 12:43:36 -0700972 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500973 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700974 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400975 case Token::Kind::TK_NOPERSPECTIVE:
ethannicholasf789b382016-08-03 12:43:36 -0700976 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500977 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700978 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400979 case Token::Kind::TK_READONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500980 this->nextToken();
981 flags |= Modifiers::kReadOnly_Flag;
982 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400983 case Token::Kind::TK_WRITEONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500984 this->nextToken();
985 flags |= Modifiers::kWriteOnly_Flag;
986 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400987 case Token::Kind::TK_COHERENT:
Brian Salomonf9f45122016-11-29 11:59:17 -0500988 this->nextToken();
989 flags |= Modifiers::kCoherent_Flag;
990 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400991 case Token::Kind::TK_VOLATILE:
Brian Salomonf9f45122016-11-29 11:59:17 -0500992 this->nextToken();
993 flags |= Modifiers::kVolatile_Flag;
994 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400995 case Token::Kind::TK_RESTRICT:
Brian Salomonf9f45122016-11-29 11:59:17 -0500996 this->nextToken();
997 flags |= Modifiers::kRestrict_Flag;
998 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400999 case Token::Kind::TK_BUFFER:
Ethan Nicholas0dd30d92017-05-01 16:57:07 -04001000 this->nextToken();
1001 flags |= Modifiers::kBuffer_Flag;
1002 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001003 case Token::Kind::TK_HASSIDEEFFECTS:
Ethan Nicholascb670962017-04-20 19:31:52 -04001004 this->nextToken();
1005 flags |= Modifiers::kHasSideEffects_Flag;
1006 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001007 case Token::Kind::TK_PLS:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001008 this->nextToken();
1009 flags |= Modifiers::kPLS_Flag;
1010 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001011 case Token::Kind::TK_PLSIN:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001012 this->nextToken();
1013 flags |= Modifiers::kPLSIn_Flag;
1014 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001015 case Token::Kind::TK_PLSOUT:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001016 this->nextToken();
1017 flags |= Modifiers::kPLSOut_Flag;
1018 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001019 case Token::Kind::TK_VARYING:
Brian Osman3c358422020-03-23 10:44:12 -04001020 this->nextToken();
1021 flags |= Modifiers::kVarying_Flag;
1022 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001023 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001024 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001025 }
1026 }
1027}
1028
Ethan Nicholas11d53972016-11-28 11:23:23 -05001029Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
1030 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -07001031 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001032 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001033 }
1034 return result;
1035}
1036
1037/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001038ASTNode::ID Parser::statement() {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001039 Token start = this->nextToken();
1040 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001041 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001042 return ASTNode::ID::Invalid();
1043 }
1044 this->pushback(start);
ethannicholasb3058bd2016-07-01 08:22:01 -07001045 switch (start.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001046 case Token::Kind::TK_IF: // fall through
1047 case Token::Kind::TK_STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -07001048 return this->ifStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001049 case Token::Kind::TK_FOR:
ethannicholasb3058bd2016-07-01 08:22:01 -07001050 return this->forStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001051 case Token::Kind::TK_DO:
ethannicholasb3058bd2016-07-01 08:22:01 -07001052 return this->doStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001053 case Token::Kind::TK_WHILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001054 return this->whileStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001055 case Token::Kind::TK_SWITCH: // fall through
1056 case Token::Kind::TK_STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -05001057 return this->switchStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001058 case Token::Kind::TK_RETURN:
ethannicholasb3058bd2016-07-01 08:22:01 -07001059 return this->returnStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001060 case Token::Kind::TK_BREAK:
ethannicholasb3058bd2016-07-01 08:22:01 -07001061 return this->breakStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001062 case Token::Kind::TK_CONTINUE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001063 return this->continueStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001064 case Token::Kind::TK_DISCARD:
ethannicholasb3058bd2016-07-01 08:22:01 -07001065 return this->discardStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001066 case Token::Kind::TK_LBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001067 return this->block();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001068 case Token::Kind::TK_SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001069 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001070 RETURN_NODE(start.fOffset, ASTNode::Kind::kBlock);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001071 case Token::Kind::TK_CONST:
Ethan Nicholasfc994162019-06-06 10:04:27 -04001072 return this->varDeclarations();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001073 case Token::Kind::TK_IDENTIFIER:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001074 if (this->isType(this->text(start))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001075 return this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001076 }
John Stiles30212b72020-06-11 17:55:07 -04001077 [[fallthrough]];
ethannicholasb3058bd2016-07-01 08:22:01 -07001078 default:
1079 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001080 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001081}
1082
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001083/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* QUESTION? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001084ASTNode::ID Parser::type() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001085 Token type;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001086 if (!this->expect(Token::Kind::TK_IDENTIFIER, "a type", &type)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001087 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001088 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001089 if (!this->isType(this->text(type))) {
1090 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001091 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001092 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001093 CREATE_NODE(result, type.fOffset, ASTNode::Kind::kType);
1094 ASTNode::TypeData td(this->text(type), false, false);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001095 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1096 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001097 SKSL_INT i;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001098 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001099 CREATE_CHILD(child, result, -1, ASTNode::Kind::kInt, i);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001100 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001101 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001102 }
1103 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001104 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001105 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001106 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001107 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001108 td.fIsNullable = this->checkNext(Token::Kind::TK_QUESTION);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001109 getNode(result).setTypeData(td);
1110 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001111}
1112
John Stiles57a996b2020-04-19 19:05:12 -07001113/* IDENTIFIER LBRACE
1114 varDeclaration+
1115 RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001116ASTNode::ID Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001117 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001118 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001119 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001120 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001121 if (peek().fKind != Token::Kind::TK_LBRACE) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001122 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001123 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -07001124 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001125 this->error(name, "no type named '" + this->text(name) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001126 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001127 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001128 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kInterfaceBlock);
1129 ASTNode::InterfaceBlockData id(mods, this->text(name), 0, "", 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07001130 this->nextToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001131 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001132 ASTNode::ID decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001133 if (!decl) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001134 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001135 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001136 getNode(result).addChild(decl);
1137 ++id.fDeclarationCount;
ethannicholasb3058bd2016-07-01 08:22:01 -07001138 }
John Stiles57a996b2020-04-19 19:05:12 -07001139 if (id.fDeclarationCount == 0) {
1140 this->error(name, "interface block '" + this->text(name) +
1141 "' must contain at least one member");
1142 return ASTNode::ID::Invalid();
1143 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001144 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001145 std::vector<ASTNode> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001146 StringFragment instanceName;
1147 Token instanceNameToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001148 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &instanceNameToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001149 id.fInstanceName = this->text(instanceNameToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001150 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1151 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001152 ASTNode::ID size = this->expression();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001153 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001154 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001155 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001156 getNode(result).addChild(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001157 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001158 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001159 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001160 ++id.fSizeCount;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001161 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001162 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001163 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -07001164 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001165 getNode(result).setInterfaceBlockData(id);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001166 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001167 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001168}
1169
1170/* IF LPAREN expression RPAREN statement (ELSE statement)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001171ASTNode::ID Parser::ifStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001172 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001173 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_IF, &start);
1174 if (!isStatic && !this->expect(Token::Kind::TK_IF, "'if'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001175 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001176 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001177 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kIf, isStatic);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001178 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001179 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001180 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001181 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001182 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001183 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001184 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001185 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001186 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001187 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001188 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001189 ASTNode::ID ifTrue = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001190 if (!ifTrue) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001191 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001192 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001193 getNode(result).addChild(ifTrue);
1194 ASTNode::ID ifFalse;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001195 if (this->checkNext(Token::Kind::TK_ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001196 ifFalse = this->statement();
1197 if (!ifFalse) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001198 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001199 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001200 getNode(result).addChild(ifFalse);
ethannicholasb3058bd2016-07-01 08:22:01 -07001201 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001202 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001203}
1204
1205/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001206ASTNode::ID Parser::doStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001207 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001208 if (!this->expect(Token::Kind::TK_DO, "'do'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001209 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001210 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001211 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kDo);
1212 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001213 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001214 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001215 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001216 getNode(result).addChild(statement);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001217 if (!this->expect(Token::Kind::TK_WHILE, "'while'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001218 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001219 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001220 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001221 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001222 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001223 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001224 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001225 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001226 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001227 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001228 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001229 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001230 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001231 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001232 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001233 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001234 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001235}
1236
1237/* WHILE LPAREN expression RPAREN STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001238ASTNode::ID Parser::whileStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001239 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001240 if (!this->expect(Token::Kind::TK_WHILE, "'while'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001241 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001242 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001243 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001244 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001245 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001246 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kWhile);
1247 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001248 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001249 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001250 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001251 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001252 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001253 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001254 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001255 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001256 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001257 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001258 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001259 getNode(result).addChild(statement);
1260 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001261}
1262
Ethan Nicholasaf197692017-02-27 13:26:45 -05001263/* CASE expression COLON statement* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001264ASTNode::ID Parser::switchCase() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001265 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001266 if (!this->expect(Token::Kind::TK_CASE, "'case'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001267 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001268 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001269 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitchCase);
1270 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001271 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001272 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001273 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001274 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001275 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001276 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001277 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001278 while (this->peek().fKind != Token::Kind::TK_RBRACE &&
1279 this->peek().fKind != Token::Kind::TK_CASE &&
1280 this->peek().fKind != Token::Kind::TK_DEFAULT) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001281 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001282 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001283 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001284 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001285 getNode(result).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001286 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001287 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001288}
1289
1290/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001291ASTNode::ID Parser::switchStatement() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001292 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001293 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_SWITCH, &start);
1294 if (!isStatic && !this->expect(Token::Kind::TK_SWITCH, "'switch'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001295 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001296 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001297 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001298 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001299 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001300 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001301 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001302 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001303 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001304 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001305 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001306 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001307 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001308 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001309 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001310 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitch, isStatic);
1311 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001312 while (this->peek().fKind == Token::Kind::TK_CASE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001313 ASTNode::ID c = this->switchCase();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001314 if (!c) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001315 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001316 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001317 getNode(result).addChild(c);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001318 }
1319 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1320 // parts of the compiler may rely upon this assumption.
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001321 if (this->peek().fKind == Token::Kind::TK_DEFAULT) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001322 Token defaultStart;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001323 SkAssertResult(this->expect(Token::Kind::TK_DEFAULT, "'default'", &defaultStart));
1324 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001325 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001326 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001327 CREATE_CHILD(defaultCase, result, defaultStart.fOffset, ASTNode::Kind::kSwitchCase);
1328 CREATE_EMPTY_CHILD(defaultCase); // empty test to signify default case
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001329 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001330 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001331 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001332 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001333 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001334 getNode(defaultCase).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001335 }
Ethan Nicholasaf197692017-02-27 13:26:45 -05001336 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001337 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001338 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001339 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001340 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001341}
1342
Ethan Nicholas11d53972016-11-28 11:23:23 -05001343/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001344 STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001345ASTNode::ID Parser::forStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001346 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001347 if (!this->expect(Token::Kind::TK_FOR, "'for'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001348 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001349 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001350 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001351 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001352 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001353 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kFor);
1354 ASTNode::ID initializer;
ethannicholasb3058bd2016-07-01 08:22:01 -07001355 Token nextToken = this->peek();
1356 switch (nextToken.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001357 case Token::Kind::TK_SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001358 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001359 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001360 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001361 case Token::Kind::TK_CONST: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001362 initializer = this->varDeclarations();
1363 if (!initializer) {
1364 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001365 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001366 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001367 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001368 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001369 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001370 if (this->isType(this->text(nextToken))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001371 initializer = this->varDeclarations();
1372 if (!initializer) {
1373 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001374 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001375 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001376 break;
1377 }
John Stiles30212b72020-06-11 17:55:07 -04001378 [[fallthrough]];
1379 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001380 default:
1381 initializer = this->expressionStatement();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001382 if (!initializer) {
1383 return ASTNode::ID::Invalid();
1384 }
1385 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001386 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001387 ASTNode::ID test;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001388 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001389 test = this->expression();
1390 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001391 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001392 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001393 getNode(result).addChild(test);
1394 } else {
1395 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001396 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001397 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001398 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001399 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001400 ASTNode::ID next;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001401 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001402 next = this->expression();
1403 if (!next) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001404 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001405 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001406 getNode(result).addChild(next);
1407 } else {
1408 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001409 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001410 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001411 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001412 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001413 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001414 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001415 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001416 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001417 getNode(result).addChild(statement);
1418 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001419}
1420
1421/* RETURN expression? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001422ASTNode::ID Parser::returnStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001423 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001424 if (!this->expect(Token::Kind::TK_RETURN, "'return'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001425 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001426 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001427 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kReturn);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001428 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001429 ASTNode::ID expression = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001430 if (!expression) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001431 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001432 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001433 getNode(result).addChild(expression);
ethannicholasb3058bd2016-07-01 08:22:01 -07001434 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001435 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001436 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001437 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001438 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001439}
1440
1441/* BREAK SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001442ASTNode::ID Parser::breakStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001443 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001444 if (!this->expect(Token::Kind::TK_BREAK, "'break'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001445 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001446 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001447 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001448 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001449 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001450 RETURN_NODE(start.fOffset, ASTNode::Kind::kBreak);
ethannicholasb3058bd2016-07-01 08:22:01 -07001451}
1452
1453/* CONTINUE SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001454ASTNode::ID Parser::continueStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001455 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001456 if (!this->expect(Token::Kind::TK_CONTINUE, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001457 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001458 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001459 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001460 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001461 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001462 RETURN_NODE(start.fOffset, ASTNode::Kind::kContinue);
ethannicholasb3058bd2016-07-01 08:22:01 -07001463}
1464
1465/* DISCARD SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001466ASTNode::ID Parser::discardStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001467 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001468 if (!this->expect(Token::Kind::TK_DISCARD, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001469 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001470 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001471 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001472 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001473 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001474 RETURN_NODE(start.fOffset, ASTNode::Kind::kDiscard);
ethannicholasb3058bd2016-07-01 08:22:01 -07001475}
1476
1477/* LBRACE statement* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001478ASTNode::ID Parser::block() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001479 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001480 if (!this->expect(Token::Kind::TK_LBRACE, "'{'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001481 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001482 }
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001483 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001484 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001485 return ASTNode::ID::Invalid();
1486 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001487 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -07001488 for (;;) {
1489 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001490 case Token::Kind::TK_RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001491 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001492 return result;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001493 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001494 this->error(this->peek(), "expected '}', but found end of file");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001495 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001496 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001497 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001498 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001499 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001500 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001501 getNode(result).addChild(statement);
ethannicholasb3058bd2016-07-01 08:22:01 -07001502 }
1503 }
1504 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001505 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001506}
1507
1508/* expression SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001509ASTNode::ID Parser::expressionStatement() {
1510 ASTNode::ID expr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001511 if (expr) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001512 if (this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001513 return expr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001514 }
1515 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001516 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001517}
1518
1519/* assignmentExpression (COMMA assignmentExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001520ASTNode::ID Parser::expression() {
1521 ASTNode::ID result = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001522 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001523 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001524 }
1525 Token t;
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001526 AutoDepth depth(this);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001527 while (this->checkNext(Token::Kind::TK_COMMA, &t)) {
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001528 if (!depth.increase()) {
1529 return ASTNode::ID::Invalid();
1530 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001531 ASTNode::ID right = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001532 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001533 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001534 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001535 CREATE_NODE(newResult, t.fOffset, ASTNode::Kind::kBinary, std::move(t));
1536 getNode(newResult).addChild(result);
1537 getNode(newResult).addChild(right);
1538 result = newResult;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001539 }
1540 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001541}
1542
1543/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1544 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1545 assignmentExpression)*
1546 */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001547ASTNode::ID Parser::assignmentExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001548 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001549 ASTNode::ID result = this->ternaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001550 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001551 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001552 }
1553 for (;;) {
1554 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001555 case Token::Kind::TK_EQ: // fall through
1556 case Token::Kind::TK_STAREQ: // fall through
1557 case Token::Kind::TK_SLASHEQ: // fall through
1558 case Token::Kind::TK_PERCENTEQ: // fall through
1559 case Token::Kind::TK_PLUSEQ: // fall through
1560 case Token::Kind::TK_MINUSEQ: // fall through
1561 case Token::Kind::TK_SHLEQ: // fall through
1562 case Token::Kind::TK_SHREQ: // fall through
1563 case Token::Kind::TK_BITWISEANDEQ: // fall through
1564 case Token::Kind::TK_BITWISEXOREQ: // fall through
1565 case Token::Kind::TK_BITWISEOREQ: // fall through
1566 case Token::Kind::TK_LOGICALANDEQ: // fall through
1567 case Token::Kind::TK_LOGICALXOREQ: // fall through
1568 case Token::Kind::TK_LOGICALOREQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001569 if (!depth.increase()) {
1570 return ASTNode::ID::Invalid();
1571 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001572 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001573 ASTNode::ID right = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001574 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001575 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001576 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001577 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1578 std::move(t));
1579 getNode(newResult).addChild(result);
1580 getNode(newResult).addChild(right);
1581 result = newResult;
1582 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001583 }
1584 default:
1585 return result;
1586 }
1587 }
1588}
1589
1590/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001591ASTNode::ID Parser::ternaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001592 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001593 ASTNode::ID base = this->logicalOrExpression();
1594 if (!base) {
1595 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001596 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001597 if (this->checkNext(Token::Kind::TK_QUESTION)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001598 if (!depth.increase()) {
1599 return ASTNode::ID::Invalid();
1600 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001601 ASTNode::ID trueExpr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001602 if (!trueExpr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001603 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001604 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001605 if (this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001606 ASTNode::ID falseExpr = this->assignmentExpression();
1607 if (!falseExpr) {
1608 return ASTNode::ID::Invalid();
1609 }
1610 CREATE_NODE(ternary, getNode(base).fOffset, ASTNode::Kind::kTernary);
1611 getNode(ternary).addChild(base);
1612 getNode(ternary).addChild(trueExpr);
1613 getNode(ternary).addChild(falseExpr);
1614 return ternary;
ethannicholasb3058bd2016-07-01 08:22:01 -07001615 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001616 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001617 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001618 return base;
ethannicholasb3058bd2016-07-01 08:22:01 -07001619}
1620
1621/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001622ASTNode::ID Parser::logicalOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001623 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001624 ASTNode::ID result = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001625 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001626 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001627 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001628 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001629 while (this->checkNext(Token::Kind::TK_LOGICALOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001630 if (!depth.increase()) {
1631 return ASTNode::ID::Invalid();
1632 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001633 ASTNode::ID right = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001634 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001635 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001636 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001637 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1638 getNode(newResult).addChild(result);
1639 getNode(newResult).addChild(right);
1640 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001641 }
1642 return result;
1643}
1644
1645/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001646ASTNode::ID Parser::logicalXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001647 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001648 ASTNode::ID result = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001649 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001650 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001651 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001652 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001653 while (this->checkNext(Token::Kind::TK_LOGICALXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001654 if (!depth.increase()) {
1655 return ASTNode::ID::Invalid();
1656 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001657 ASTNode::ID right = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001658 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001659 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001660 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001661 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1662 getNode(newResult).addChild(result);
1663 getNode(newResult).addChild(right);
1664 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001665 }
1666 return result;
1667}
1668
1669/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001670ASTNode::ID Parser::logicalAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001671 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001672 ASTNode::ID result = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001673 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001674 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001675 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001676 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001677 while (this->checkNext(Token::Kind::TK_LOGICALAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001678 if (!depth.increase()) {
1679 return ASTNode::ID::Invalid();
1680 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001681 ASTNode::ID right = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001682 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001683 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001684 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001685 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1686 getNode(newResult).addChild(result);
1687 getNode(newResult).addChild(right);
1688 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001689 }
1690 return result;
1691}
1692
1693/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001694ASTNode::ID Parser::bitwiseOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001695 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001696 ASTNode::ID result = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001697 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001698 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001699 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001700 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001701 while (this->checkNext(Token::Kind::TK_BITWISEOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001702 if (!depth.increase()) {
1703 return ASTNode::ID::Invalid();
1704 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001705 ASTNode::ID right = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001706 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001707 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001708 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001709 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1710 getNode(newResult).addChild(result);
1711 getNode(newResult).addChild(right);
1712 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001713 }
1714 return result;
1715}
1716
1717/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001718ASTNode::ID Parser::bitwiseXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001719 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001720 ASTNode::ID result = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001721 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001722 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001723 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001724 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001725 while (this->checkNext(Token::Kind::TK_BITWISEXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001726 if (!depth.increase()) {
1727 return ASTNode::ID::Invalid();
1728 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001729 ASTNode::ID right = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001730 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001731 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001732 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001733 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1734 getNode(newResult).addChild(result);
1735 getNode(newResult).addChild(right);
1736 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001737 }
1738 return result;
1739}
1740
1741/* equalityExpression (BITWISEAND equalityExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001742ASTNode::ID Parser::bitwiseAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001743 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001744 ASTNode::ID result = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001745 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001746 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001747 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001748 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001749 while (this->checkNext(Token::Kind::TK_BITWISEAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001750 if (!depth.increase()) {
1751 return ASTNode::ID::Invalid();
1752 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001753 ASTNode::ID right = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001754 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001755 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001756 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001757 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1758 getNode(newResult).addChild(result);
1759 getNode(newResult).addChild(right);
1760 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001761 }
1762 return result;
1763}
1764
1765/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001766ASTNode::ID Parser::equalityExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001767 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001768 ASTNode::ID result = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001769 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001770 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001771 }
1772 for (;;) {
1773 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001774 case Token::Kind::TK_EQEQ: // fall through
1775 case Token::Kind::TK_NEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001776 if (!depth.increase()) {
1777 return ASTNode::ID::Invalid();
1778 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001779 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001780 ASTNode::ID right = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001781 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001782 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001783 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001784 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1785 std::move(t));
1786 getNode(newResult).addChild(result);
1787 getNode(newResult).addChild(right);
1788 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001789 break;
1790 }
1791 default:
1792 return result;
1793 }
1794 }
1795}
1796
1797/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001798ASTNode::ID Parser::relationalExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001799 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001800 ASTNode::ID result = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001801 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001802 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001803 }
1804 for (;;) {
1805 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001806 case Token::Kind::TK_LT: // fall through
1807 case Token::Kind::TK_GT: // fall through
1808 case Token::Kind::TK_LTEQ: // fall through
1809 case Token::Kind::TK_GTEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001810 if (!depth.increase()) {
1811 return ASTNode::ID::Invalid();
1812 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001813 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001814 ASTNode::ID right = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001815 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001816 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001817 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001818 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1819 std::move(t));
1820 getNode(newResult).addChild(result);
1821 getNode(newResult).addChild(right);
1822 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001823 break;
1824 }
1825 default:
1826 return result;
1827 }
1828 }
1829}
1830
1831/* additiveExpression ((SHL | SHR) additiveExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001832ASTNode::ID Parser::shiftExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001833 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001834 ASTNode::ID result = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001835 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001836 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001837 }
1838 for (;;) {
1839 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001840 case Token::Kind::TK_SHL: // fall through
1841 case Token::Kind::TK_SHR: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001842 if (!depth.increase()) {
1843 return ASTNode::ID::Invalid();
1844 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001845 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001846 ASTNode::ID right = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001847 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001848 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001849 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001850 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1851 std::move(t));
1852 getNode(newResult).addChild(result);
1853 getNode(newResult).addChild(right);
1854 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001855 break;
1856 }
1857 default:
1858 return result;
1859 }
1860 }
1861}
1862
1863/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001864ASTNode::ID Parser::additiveExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001865 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001866 ASTNode::ID result = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001867 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001868 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001869 }
1870 for (;;) {
1871 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001872 case Token::Kind::TK_PLUS: // fall through
1873 case Token::Kind::TK_MINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001874 if (!depth.increase()) {
1875 return ASTNode::ID::Invalid();
1876 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001877 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001878 ASTNode::ID right = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001879 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001880 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001881 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001882 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1883 std::move(t));
1884 getNode(newResult).addChild(result);
1885 getNode(newResult).addChild(right);
1886 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001887 break;
1888 }
1889 default:
1890 return result;
1891 }
1892 }
1893}
1894
1895/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001896ASTNode::ID Parser::multiplicativeExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001897 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001898 ASTNode::ID result = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001899 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001900 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001901 }
1902 for (;;) {
1903 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001904 case Token::Kind::TK_STAR: // fall through
1905 case Token::Kind::TK_SLASH: // fall through
1906 case Token::Kind::TK_PERCENT: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001907 if (!depth.increase()) {
1908 return ASTNode::ID::Invalid();
1909 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001910 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001911 ASTNode::ID right = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001912 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001913 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001914 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001915 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1916 std::move(t));
1917 getNode(newResult).addChild(result);
1918 getNode(newResult).addChild(right);
1919 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001920 break;
1921 }
1922 default:
1923 return result;
1924 }
1925 }
1926}
1927
1928/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001929ASTNode::ID Parser::unaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001930 AutoDepth depth(this);
ethannicholasb3058bd2016-07-01 08:22:01 -07001931 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001932 case Token::Kind::TK_PLUS: // fall through
1933 case Token::Kind::TK_MINUS: // fall through
1934 case Token::Kind::TK_LOGICALNOT: // fall through
1935 case Token::Kind::TK_BITWISENOT: // fall through
1936 case Token::Kind::TK_PLUSPLUS: // fall through
1937 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001938 if (!depth.increase()) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001939 return ASTNode::ID::Invalid();
Ethan Nicholas6dcc3252019-02-20 15:18:36 -05001940 }
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001941 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001942 ASTNode::ID expr = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001943 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001944 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001945 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001946 CREATE_NODE(result, t.fOffset, ASTNode::Kind::kPrefix, std::move(t));
1947 getNode(result).addChild(expr);
1948 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001949 }
1950 default:
1951 return this->postfixExpression();
1952 }
1953}
1954
1955/* term suffix* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001956ASTNode::ID Parser::postfixExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001957 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001958 ASTNode::ID result = this->term();
ethannicholasb3058bd2016-07-01 08:22:01 -07001959 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001960 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001961 }
1962 for (;;) {
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001963 Token t = this->peek();
1964 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001965 case Token::Kind::TK_FLOAT_LITERAL:
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001966 if (this->text(t)[0] != '.') {
1967 return result;
1968 }
John Stiles30212b72020-06-11 17:55:07 -04001969 [[fallthrough]];
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001970 case Token::Kind::TK_LBRACKET:
1971 case Token::Kind::TK_DOT:
1972 case Token::Kind::TK_LPAREN:
1973 case Token::Kind::TK_PLUSPLUS:
1974 case Token::Kind::TK_MINUSMINUS:
1975 case Token::Kind::TK_COLONCOLON:
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001976 if (!depth.increase()) {
1977 return ASTNode::ID::Invalid();
1978 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001979 result = this->suffix(result);
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001980 if (!result) {
1981 return ASTNode::ID::Invalid();
1982 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001983 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001984 default:
1985 return result;
1986 }
1987 }
1988}
1989
Ethan Nicholas11d53972016-11-28 11:23:23 -05001990/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
Ethan Nicholase455f652019-09-13 12:52:55 -04001991 PLUSPLUS | MINUSMINUS | COLONCOLON IDENTIFIER | FLOAT_LITERAL [IDENTIFIER] */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001992ASTNode::ID Parser::suffix(ASTNode::ID base) {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001993 SkASSERT(base);
ethannicholasb3058bd2016-07-01 08:22:01 -07001994 Token next = this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001995 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001996 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001997 return ASTNode::ID::Invalid();
1998 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001999 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002000 case Token::Kind::TK_LBRACKET: {
2001 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002002 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
2003 getNode(result).addChild(base);
2004 return result;
ethannicholas5961bc92016-10-12 06:39:56 -07002005 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002006 ASTNode::ID e = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002007 if (!e) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002008 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002009 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002010 this->expect(Token::Kind::TK_RBRACKET, "']' to complete array access expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002011 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
2012 getNode(result).addChild(base);
2013 getNode(result).addChild(e);
2014 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002015 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002016 case Token::Kind::TK_DOT: // fall through
2017 case Token::Kind::TK_COLONCOLON: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002018 int offset = this->peek().fOffset;
2019 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002020 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002021 CREATE_NODE(result, offset, ASTNode::Kind::kField, std::move(text));
2022 getNode(result).addChild(base);
2023 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002024 }
John Stiles30212b72020-06-11 17:55:07 -04002025 [[fallthrough]]; // FIXME(ethannicholas)
Ethan Nicholase455f652019-09-13 12:52:55 -04002026 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002027 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholase455f652019-09-13 12:52:55 -04002028 // Swizzles that start with a constant number, e.g. '.000r', will be tokenized as
2029 // floating point literals, possibly followed by an identifier. Handle that here.
2030 StringFragment field = this->text(next);
2031 SkASSERT(field.fChars[0] == '.');
2032 ++field.fChars;
2033 --field.fLength;
2034 for (size_t i = 0; i < field.fLength; ++i) {
2035 if (field.fChars[i] != '0' && field.fChars[i] != '1') {
2036 this->error(next, "invalid swizzle");
2037 return ASTNode::ID::Invalid();
2038 }
2039 }
2040 // use the next *raw* token so we don't ignore whitespace - we only care about
2041 // identifiers that directly follow the float
2042 Token id = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002043 if (id.fKind == Token::Kind::TK_IDENTIFIER) {
Ethan Nicholase455f652019-09-13 12:52:55 -04002044 field.fLength += id.fLength;
2045 } else {
2046 this->pushback(id);
2047 }
2048 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kField, field);
2049 getNode(result).addChild(base);
2050 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002051 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002052 case Token::Kind::TK_LPAREN: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002053 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kCall);
2054 getNode(result).addChild(base);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002055 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002056 for (;;) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002057 ASTNode::ID expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002058 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002059 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002060 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002061 getNode(result).addChild(expr);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002062 if (!this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002063 break;
2064 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002065 }
2066 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002067 this->expect(Token::Kind::TK_RPAREN, "')' to complete function parameters");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002068 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002069 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002070 case Token::Kind::TK_PLUSPLUS: // fall through
2071 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002072 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kPostfix, next);
2073 getNode(result).addChild(base);
2074 return result;
2075 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002076 default: {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002077 this->error(next, "expected expression suffix, but found '" + this->text(next) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002078 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002079 }
2080 }
2081}
2082
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002083/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | NULL_LITERAL | '(' expression ')' */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002084ASTNode::ID Parser::term() {
ethannicholasb3058bd2016-07-01 08:22:01 -07002085 Token t = this->peek();
2086 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002087 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002088 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002089 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002090 RETURN_NODE(t.fOffset, ASTNode::Kind::kIdentifier, std::move(text));
ethannicholasb3058bd2016-07-01 08:22:01 -07002091 }
John Stiles30212b72020-06-11 17:55:07 -04002092 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002093 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002094 case Token::Kind::TK_INT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002095 SKSL_INT i;
ethannicholasb3058bd2016-07-01 08:22:01 -07002096 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002097 RETURN_NODE(t.fOffset, ASTNode::Kind::kInt, i);
ethannicholasb3058bd2016-07-01 08:22:01 -07002098 }
2099 break;
2100 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002101 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002102 SKSL_FLOAT f;
ethannicholasb3058bd2016-07-01 08:22:01 -07002103 if (this->floatLiteral(&f)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002104 RETURN_NODE(t.fOffset, ASTNode::Kind::kFloat, f);
ethannicholasb3058bd2016-07-01 08:22:01 -07002105 }
2106 break;
2107 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002108 case Token::Kind::TK_TRUE_LITERAL: // fall through
2109 case Token::Kind::TK_FALSE_LITERAL: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002110 bool b;
2111 if (this->boolLiteral(&b)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002112 RETURN_NODE(t.fOffset, ASTNode::Kind::kBool, b);
ethannicholasb3058bd2016-07-01 08:22:01 -07002113 }
2114 break;
2115 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002116 case Token::Kind::TK_NULL_LITERAL:
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002117 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04002118 RETURN_NODE(t.fOffset, ASTNode::Kind::kNull);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002119 case Token::Kind::TK_LPAREN: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002120 this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002121 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002122 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002123 return ASTNode::ID::Invalid();
2124 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002125 ASTNode::ID result = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002126 if (result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002127 this->expect(Token::Kind::TK_RPAREN, "')' to complete expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002128 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002129 }
2130 break;
2131 }
2132 default:
2133 this->nextToken();
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002134 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002135 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002136 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002137}
2138
2139/* INT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002140bool Parser::intLiteral(SKSL_INT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002141 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002142 if (this->expect(Token::Kind::TK_INT_LITERAL, "integer literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002143 *dest = SkSL::stol(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002144 return true;
2145 }
2146 return false;
2147}
2148
2149/* FLOAT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002150bool Parser::floatLiteral(SKSL_FLOAT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002151 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002152 if (this->expect(Token::Kind::TK_FLOAT_LITERAL, "float literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002153 *dest = SkSL::stod(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002154 return true;
2155 }
2156 return false;
2157}
2158
2159/* TRUE_LITERAL | FALSE_LITERAL */
2160bool Parser::boolLiteral(bool* dest) {
2161 Token t = this->nextToken();
2162 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002163 case Token::Kind::TK_TRUE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002164 *dest = true;
2165 return true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002166 case Token::Kind::TK_FALSE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002167 *dest = false;
2168 return true;
2169 default:
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002170 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002171 return false;
2172 }
2173}
2174
2175/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002176bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002177 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002178 if (this->expect(Token::Kind::TK_IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002179 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07002180 return true;
2181 }
2182 return false;
2183}
2184
2185} // namespace