blob: 4262f2ff484e8bc86ae6f29b85ed3c6621f22d03 [file] [log] [blame]
ethannicholasb3058bd2016-07-01 08:22:01 -07001/*
2 * Copyright 2016 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
Ethan Nicholas11d53972016-11-28 11:23:23 -05007
ethannicholasb3058bd2016-07-01 08:22:01 -07008#include "stdio.h"
Ethan Nicholasfc994162019-06-06 10:04:27 -04009#include "src/sksl/SkSLASTNode.h"
Mike Klein4b432fa2019-06-06 11:44:05 -050010#include "src/sksl/SkSLParser.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050011#include "src/sksl/ir/SkSLModifiers.h"
12#include "src/sksl/ir/SkSLSymbolTable.h"
13#include "src/sksl/ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070014
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040015#ifndef SKSL_STANDALONE
Mike Kleinc0bd9f92019-04-23 12:05:21 -050016#include "include/private/SkOnce.h"
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040017#endif
18
ethannicholasb3058bd2016-07-01 08:22:01 -070019namespace SkSL {
20
ethannicholascad64162016-10-27 10:54:02 -070021#define MAX_PARSE_DEPTH 50
22
23class AutoDepth {
24public:
25 AutoDepth(Parser* p)
Ethan Nicholascf4deab2019-09-13 16:28:14 -040026 : fParser(p)
27 , fDepth(0) {}
ethannicholascad64162016-10-27 10:54:02 -070028
29 ~AutoDepth() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -040030 fParser->fDepth -= fDepth;
ethannicholascad64162016-10-27 10:54:02 -070031 }
32
Ethan Nicholascf4deab2019-09-13 16:28:14 -040033 bool increase() {
34 ++fDepth;
35 ++fParser->fDepth;
ethannicholascad64162016-10-27 10:54:02 -070036 if (fParser->fDepth > MAX_PARSE_DEPTH) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070037 fParser->error(fParser->peek(), String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070038 return false;
39 }
40 return true;
41 }
42
43private:
44 Parser* fParser;
Ethan Nicholascf4deab2019-09-13 16:28:14 -040045 int fDepth;
ethannicholascad64162016-10-27 10:54:02 -070046};
47
Brian Salomon140f3da2018-08-23 13:51:27 +000048std::unordered_map<String, Parser::LayoutToken>* Parser::layoutTokens;
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040049
50void Parser::InitLayoutMap() {
Brian Salomon140f3da2018-08-23 13:51:27 +000051 layoutTokens = new std::unordered_map<String, LayoutToken>;
Brian Salomon23356442018-11-30 15:33:19 -050052 #define TOKEN(name, text) (*layoutTokens)[text] = LayoutToken::name
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040053 TOKEN(LOCATION, "location");
54 TOKEN(OFFSET, "offset");
55 TOKEN(BINDING, "binding");
56 TOKEN(INDEX, "index");
57 TOKEN(SET, "set");
58 TOKEN(BUILTIN, "builtin");
59 TOKEN(INPUT_ATTACHMENT_INDEX, "input_attachment_index");
60 TOKEN(ORIGIN_UPPER_LEFT, "origin_upper_left");
61 TOKEN(OVERRIDE_COVERAGE, "override_coverage");
62 TOKEN(BLEND_SUPPORT_ALL_EQUATIONS, "blend_support_all_equations");
63 TOKEN(BLEND_SUPPORT_MULTIPLY, "blend_support_multiply");
64 TOKEN(BLEND_SUPPORT_SCREEN, "blend_support_screen");
65 TOKEN(BLEND_SUPPORT_OVERLAY, "blend_support_overlay");
66 TOKEN(BLEND_SUPPORT_DARKEN, "blend_support_darken");
67 TOKEN(BLEND_SUPPORT_LIGHTEN, "blend_support_lighten");
68 TOKEN(BLEND_SUPPORT_COLORDODGE, "blend_support_colordodge");
69 TOKEN(BLEND_SUPPORT_COLORBURN, "blend_support_colorburn");
70 TOKEN(BLEND_SUPPORT_HARDLIGHT, "blend_support_hardlight");
71 TOKEN(BLEND_SUPPORT_SOFTLIGHT, "blend_support_softlight");
72 TOKEN(BLEND_SUPPORT_DIFFERENCE, "blend_support_difference");
73 TOKEN(BLEND_SUPPORT_EXCLUSION, "blend_support_exclusion");
74 TOKEN(BLEND_SUPPORT_HSL_HUE, "blend_support_hsl_hue");
75 TOKEN(BLEND_SUPPORT_HSL_SATURATION, "blend_support_hsl_saturation");
76 TOKEN(BLEND_SUPPORT_HSL_COLOR, "blend_support_hsl_color");
77 TOKEN(BLEND_SUPPORT_HSL_LUMINOSITY, "blend_support_hsl_luminosity");
78 TOKEN(PUSH_CONSTANT, "push_constant");
79 TOKEN(POINTS, "points");
80 TOKEN(LINES, "lines");
81 TOKEN(LINE_STRIP, "line_strip");
82 TOKEN(LINES_ADJACENCY, "lines_adjacency");
83 TOKEN(TRIANGLES, "triangles");
84 TOKEN(TRIANGLE_STRIP, "triangle_strip");
85 TOKEN(TRIANGLES_ADJACENCY, "triangles_adjacency");
86 TOKEN(MAX_VERTICES, "max_vertices");
87 TOKEN(INVOCATIONS, "invocations");
Brian Osmanf59a9612020-04-15 14:18:13 -040088 TOKEN(MARKER, "marker");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040089 TOKEN(WHEN, "when");
90 TOKEN(KEY, "key");
Michael Ludwiga4275592018-08-31 10:52:47 -040091 TOKEN(TRACKED, "tracked");
Brian Osmanb32d66b2020-04-30 17:12:03 -040092 TOKEN(SRGB_UNPREMUL, "srgb_unpremul");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040093 TOKEN(CTYPE, "ctype");
Brian Osmanf28e55d2018-10-03 16:35:54 -040094 TOKEN(SKPMCOLOR4F, "SkPMColor4f");
Mike Reedb26b4e72020-01-22 14:31:21 -050095 TOKEN(SKV4, "SkV4");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040096 TOKEN(SKRECT, "SkRect");
97 TOKEN(SKIRECT, "SkIRect");
98 TOKEN(SKPMCOLOR, "SkPMColor");
Mike Reedb26b4e72020-01-22 14:31:21 -050099 TOKEN(SKM44, "SkM44");
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400100 TOKEN(BOOL, "bool");
101 TOKEN(INT, "int");
102 TOKEN(FLOAT, "float");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400103 #undef TOKEN
104}
105
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400106Parser::Parser(const char* text, size_t length, SymbolTable& symbols, ErrorReporter& errors)
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700107: fText(text)
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400108, fPushback(Token::Kind::TK_INVALID, -1, -1)
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400109, fSymbols(symbols)
ethannicholasb3058bd2016-07-01 08:22:01 -0700110, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700111 fLexer.start(text, length);
Brian Salomon3b83afe2018-08-23 11:04:36 -0400112 static const bool layoutMapInitialized = []{ return (void)InitLayoutMap(), true; }();
113 (void) layoutMapInitialized;
ethannicholasb3058bd2016-07-01 08:22:01 -0700114}
115
Ethan Nicholasfc994162019-06-06 10:04:27 -0400116#define CREATE_NODE(result, ...) \
117 ASTNode::ID result(fFile->fNodes.size()); \
118 fFile->fNodes.emplace_back(&fFile->fNodes, __VA_ARGS__)
119
120#define RETURN_NODE(...) \
121 do { \
122 CREATE_NODE(result, __VA_ARGS__); \
123 return result; \
124 } while (false)
125
126#define CREATE_CHILD(child, target, ...) \
127 CREATE_NODE(child, __VA_ARGS__); \
128 fFile->fNodes[target.fValue].addChild(child)
129
130#define CREATE_EMPTY_CHILD(target) \
131 do { \
132 ASTNode::ID child(fFile->fNodes.size()); \
133 fFile->fNodes.emplace_back(); \
134 fFile->fNodes[target.fValue].addChild(child); \
135 } while (false)
136
Ethan Nicholasf7b88202017-09-18 14:10:39 -0400137/* (directive | section | declaration)* END_OF_FILE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400138std::unique_ptr<ASTFile> Parser::file() {
139 fFile.reset(new ASTFile());
140 CREATE_NODE(result, 0, ASTNode::Kind::kFile);
141 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700142 for (;;) {
143 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400144 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400145 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400146 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400147 ASTNode::ID dir = this->directive();
148 if (fErrors.errorCount()) {
149 return nullptr;
150 }
151 if (dir) {
152 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700153 }
154 break;
155 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400156 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400157 ASTNode::ID section = this->section();
158 if (fErrors.errorCount()) {
159 return nullptr;
160 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400161 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400162 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400163 }
164 break;
165 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700166 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400167 ASTNode::ID decl = this->declaration();
168 if (fErrors.errorCount()) {
169 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700170 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400171 if (decl) {
172 getNode(result).addChild(decl);
173 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700174 }
175 }
176 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400177 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700178}
179
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700180Token Parser::nextRawToken() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400181 if (fPushback.fKind != Token::Kind::TK_INVALID) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700182 Token result = fPushback;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400183 fPushback.fKind = Token::Kind::TK_INVALID;
ethannicholasb3058bd2016-07-01 08:22:01 -0700184 return result;
185 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700186 Token result = fLexer.next();
187 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400188}
189
190Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700191 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400192 while (token.fKind == Token::Kind::TK_WHITESPACE ||
193 token.fKind == Token::Kind::TK_LINE_COMMENT ||
194 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700195 token = this->nextRawToken();
196 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400197 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700198}
199
200void Parser::pushback(Token t) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400201 SkASSERT(fPushback.fKind == Token::Kind::TK_INVALID);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400202 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700203}
204
205Token Parser::peek() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400206 if (fPushback.fKind == Token::Kind::TK_INVALID) {
Brian Osman634624a2017-08-15 11:14:30 -0400207 fPushback = this->nextToken();
208 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700209 return fPushback;
210}
211
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400212bool Parser::checkNext(Token::Kind kind, Token* result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400213 if (fPushback.fKind != Token::Kind::TK_INVALID && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400214 return false;
215 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400216 Token next = this->nextToken();
217 if (next.fKind == kind) {
218 if (result) {
219 *result = next;
220 }
221 return true;
222 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400223 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400224 return false;
225}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500226
227bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700228 Token next = this->nextToken();
229 if (next.fKind == kind) {
230 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400231 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700232 }
233 return true;
234 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700235 this->error(next, "expected " + String(expected) + ", but found '" +
236 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700237 return false;
238 }
239}
240
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700241StringFragment Parser::text(Token token) {
242 return StringFragment(fText + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500243}
244
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700245void Parser::error(Token token, String msg) {
246 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700247}
248
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700249void Parser::error(int offset, String msg) {
250 fErrors.error(offset, msg);
251}
252
253bool Parser::isType(StringFragment name) {
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400254 const Symbol* s = fSymbols[name];
255 return s && s->fKind == Symbol::kType_Kind;
ethannicholasb3058bd2016-07-01 08:22:01 -0700256}
257
Ethan Nicholas11d53972016-11-28 11:23:23 -0500258/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700259 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400260ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700261 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400262 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400263 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700264 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700265 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400266 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700267 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400268 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400269 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700270 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400271 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400272 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700273 }
274 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400275 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400276 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700277 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400278 RETURN_NODE(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700279 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700280 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400281 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700282 }
283}
284
Ethan Nicholas762466e2017-06-29 10:03:38 -0400285/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
286 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400287ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400288 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400289 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400290 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400291 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400292 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400293 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400294 this->nextToken();
295 Token argToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400296 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400297 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400298 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700299 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400300 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400301 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400302 }
303 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400304 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400305 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400306 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400307 StringFragment text;
308 Token codeStart = this->nextRawToken();
309 size_t startOffset = codeStart.fOffset;
310 this->pushback(codeStart);
311 text.fChars = fText + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400312 int level = 1;
313 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700314 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400315 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400316 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400317 ++level;
318 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400319 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400320 --level;
321 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400322 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700323 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400324 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400325 default:
326 break;
327 }
328 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400329 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400330 break;
331 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400332 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700333 StringFragment name = this->text(start);
334 ++name.fChars;
335 --name.fLength;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400336 RETURN_NODE(start.fOffset, ASTNode::Kind::kSection,
337 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400338}
339
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500340/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
341 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400342ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500343 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400344 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400345 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500346 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400347 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400348 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500349 }
350 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400351 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400352 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500353 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400354 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400355 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500356 }
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400357 fSymbols.add(this->text(name), std::unique_ptr<Symbol>(new Type(this->text(name),
358 Type::kEnum_Kind)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400359 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400360 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500361 Token id;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400362 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400363 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500364 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400365 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400366 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500367 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400368 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500369 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400370 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
371 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500372 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400373 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500374 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400375 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
376 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400377 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500378 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400379 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
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->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400383 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500384 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400385 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500386 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400387 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
388 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500389 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400390 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500391 }
392 }
393 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400394 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400395 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500396}
397
398/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700399 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400400ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700401 Token lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400402 if (lookahead.fKind == Token::Kind::TK_ENUM) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500403 return this->enumDeclaration();
404 }
405 Modifiers modifiers = this->modifiers();
406 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400407 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700408 // we have an identifier that's not a type, could be the start of an interface block
409 return this->interfaceBlock(modifiers);
410 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400411 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700412 return this->structVarDeclaration(modifiers);
413 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400414 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700415 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400416 RETURN_NODE(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700417 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400418 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700419 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400420 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700421 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400422 if (getNode(type).getTypeData().fIsStructDeclaration &&
423 this->checkNext(Token::Kind::TK_SEMICOLON)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400424 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700425 }
426 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400427 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400428 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700429 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400430 if (this->checkNext(Token::Kind::TK_LPAREN)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400431 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kFunction);
432 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
433 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400434 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400435 for (;;) {
436 ASTNode::ID parameter = this->parameter();
437 if (!parameter) {
438 return ASTNode::ID::Invalid();
439 }
440 ++fd.fParameterCount;
441 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400442 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400443 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700444 }
445 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700446 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400447 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400448 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400449 return ASTNode::ID::Invalid();
450 }
451 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400452 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700453 body = this->block();
454 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400455 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700456 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400457 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700458 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400459 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700460 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400461 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700462 }
463}
464
465/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400466ASTNode::ID Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500467 Modifiers modifiers = this->modifiers();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400468 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700469 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400470 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700471 }
472 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400473 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400474 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700475 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400476 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700477}
478
479/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400480ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400481 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400482 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700483 }
484 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400485 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400486 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700487 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400488 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400489 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700490 }
491 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400492 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400493 ASTNode::ID decls = this->varDeclarations();
494 if (!decls) {
495 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700496 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400497 ASTNode& declsNode = getNode(decls);
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400498 const Symbol* symbol = fSymbols[(declsNode.begin() + 1)->getTypeData().fName];
499 SkASSERT(symbol && symbol->fKind == Symbol::kType_Kind);
500 const Type* type = (const Type*) symbol;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400501 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
502 ASTNode& var = *iter;
503 ASTNode::VarData vd = var.getVarData();
504 for (int j = vd.fSizeCount - 1; j >= 0; j--) {
505 const ASTNode& size = *(var.begin() + j);
506 if (!size || size.fKind != ASTNode::Kind::kInt) {
507 this->error(declsNode.fOffset, "array size in struct field must be a constant");
508 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700509 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400510 uint64_t columns = size.getInt();
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400511 String name = type->name() + "[" + to_string(columns) + "]";
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400512 type = (Type*) fSymbols.takeOwnership(std::unique_ptr<const Symbol>(
Ethan Nicholas91164d12019-05-15 15:29:54 -0400513 new Type(name,
514 Type::kArray_Kind,
515 *type,
516 (int) columns)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700517 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400518 fields.push_back(Type::Field(declsNode.begin()->getModifiers(), vd.fName, type));
519 if (vd.fSizeCount ? (var.begin() + (vd.fSizeCount - 1))->fNext : var.fFirstChild) {
520 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700521 }
522 }
523 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400524 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400525 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700526 }
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400527 fSymbols.add(this->text(name), std::unique_ptr<Type>(new Type(name.fOffset, this->text(name),
528 fields)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400529 RETURN_NODE(name.fOffset, ASTNode::Kind::kType,
530 ASTNode::TypeData(this->text(name), true, false));
ethannicholasb3058bd2016-07-01 08:22:01 -0700531}
532
533/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400534ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
535 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700536 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400537 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700538 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400539 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400540 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400541 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700542 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400543 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400544 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700545}
546
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400547/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
548 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400549ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
550 CREATE_NODE(result, -1, ASTNode::Kind::kVarDeclarations);
551 CREATE_CHILD(modifiers, result, -1, ASTNode::Kind::kModifiers, mods);
552 getNode(result).addChild(type);
553 CREATE_NODE(currentVar, -1, ASTNode::Kind::kVarDeclaration);
554 ASTNode::VarData vd(name, 0);
555 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400556 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
557 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400558 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700559 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400560 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700561 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400562 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700563 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400564 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400565 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400566 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700567 }
568 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400569 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700570 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400571 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400572 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400573 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700574 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400575 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700576 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400577 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700578 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400579 while (this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700580 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400581 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400582 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700583 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400584 currentVar = ASTNode::ID(fFile->fNodes.size());
585 vd = ASTNode::VarData(this->text(name), 0);
586 fFile->fNodes.emplace_back(&fFile->fNodes, -1, ASTNode::Kind::kVarDeclaration);
587 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400588 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
589 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400590 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700591 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400592 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700593 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400594 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700595 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400596 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400597 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400598 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700599 }
600 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400601 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700602 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400603 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400604 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400605 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700606 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400607 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700608 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400609 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700610 }
611 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400612 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400613 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700614 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400615 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700616}
617
618/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400619ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400620 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400621 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700622 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400623 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700624 }
625 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400626 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400627 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700628 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400629 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kParameter);
630 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
631 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400632 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700633 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400634 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400635 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700636 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400637 CREATE_CHILD(child, result, sizeToken.fOffset, ASTNode::Kind::kInt,
638 SkSL::stoi(this->text(sizeToken)));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400639 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400640 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700641 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400642 ++pd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700643 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400644 getNode(result).setParameterData(pd);
645 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700646}
647
Ethan Nicholasd608c092017-10-26 09:30:08 -0400648/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700649int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400650 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700651 return -1;
652 }
653 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400654 if (this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700655 return SkSL::stoi(this->text(resultToken));
ethannicholasb3058bd2016-07-01 08:22:01 -0700656 }
657 return -1;
658}
659
Ethan Nicholasd608c092017-10-26 09:30:08 -0400660/** EQ IDENTIFIER */
661StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400662 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400663 return StringFragment();
664 }
665 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400666 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400667 return StringFragment();
668 }
669 return this->text(resultToken);
670}
671
672
Ethan Nicholas762466e2017-06-29 10:03:38 -0400673/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400674StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400675 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400676 return "";
677 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700678 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400679 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400680 StringFragment code;
681 code.fChars = fText + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400682 int level = 1;
683 bool done = false;
684 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700685 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400686 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400687 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400688 ++level;
689 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400690 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400691 --level;
692 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400693 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400694 if (level == 1) {
695 done = true;
696 }
697 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400698 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700699 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400700 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400701 default:
702 break;
703 }
704 if (!level) {
705 done = true;
706 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400707 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400708 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400709 this->pushback(std::move(next));
710 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400711 }
712 return code;
713}
714
715/** (EQ IDENTIFIER('identity'))? */
716Layout::Key Parser::layoutKey() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400717 if (this->peek().fKind == Token::Kind::TK_EQ) {
718 this->expect(Token::Kind::TK_EQ, "'='");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400719 Token key;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400720 if (this->expect(Token::Kind::TK_IDENTIFIER, "an identifer", &key)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700721 if (this->text(key) == "identity") {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400722 return Layout::kIdentity_Key;
723 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700724 this->error(key, "unsupported layout key");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400725 }
726 }
727 }
728 return Layout::kKey_Key;
729}
730
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400731Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400732 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400733 Token t = this->nextToken();
734 String text = this->text(t);
735 auto found = layoutTokens->find(text);
736 if (found != layoutTokens->end()) {
737 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400738 case LayoutToken::SKPMCOLOR4F:
739 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500740 case LayoutToken::SKV4:
741 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400742 case LayoutToken::SKRECT:
743 return Layout::CType::kSkRect;
744 case LayoutToken::SKIRECT:
745 return Layout::CType::kSkIRect;
746 case LayoutToken::SKPMCOLOR:
747 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400748 case LayoutToken::BOOL:
749 return Layout::CType::kBool;
750 case LayoutToken::INT:
751 return Layout::CType::kInt32;
752 case LayoutToken::FLOAT:
753 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500754 case LayoutToken::SKM44:
755 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400756 default:
757 break;
758 }
759 }
760 this->error(t, "unsupported ctype");
761 }
762 return Layout::CType::kDefault;
763}
764
ethannicholas8ac838d2016-11-22 08:39:36 -0800765/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500766Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500767 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700768 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500769 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700770 int binding = -1;
771 int index = -1;
772 int set = -1;
773 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500774 int inputAttachmentIndex = -1;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500775 Layout::Format format = Layout::Format::kUnspecified;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500776 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
777 int maxVertices = -1;
778 int invocations = -1;
Brian Osmanf59a9612020-04-15 14:18:13 -0400779 StringFragment marker;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400780 StringFragment when;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400781 Layout::Key key = Layout::kNo_Key;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400782 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400783 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
784 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500785 return Layout(flags, location, offset, binding, index, set, builtin,
Brian Osmanf59a9612020-04-15 14:18:13 -0400786 inputAttachmentIndex, format, primitive, maxVertices, invocations, marker,
787 when, key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700788 }
789 for (;;) {
790 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700791 String text = this->text(t);
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400792 auto found = layoutTokens->find(text);
793 if (found != layoutTokens->end()) {
794 switch (found->second) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700795 case LayoutToken::LOCATION:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500796 location = this->layoutInt();
797 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700798 case LayoutToken::OFFSET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500799 offset = this->layoutInt();
800 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700801 case LayoutToken::BINDING:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500802 binding = this->layoutInt();
803 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700804 case LayoutToken::INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500805 index = this->layoutInt();
806 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700807 case LayoutToken::SET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500808 set = this->layoutInt();
809 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700810 case LayoutToken::BUILTIN:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500811 builtin = this->layoutInt();
812 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700813 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500814 inputAttachmentIndex = this->layoutInt();
815 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700816 case LayoutToken::ORIGIN_UPPER_LEFT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500817 flags |= Layout::kOriginUpperLeft_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500818 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700819 case LayoutToken::OVERRIDE_COVERAGE:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500820 flags |= Layout::kOverrideCoverage_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500821 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700822 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500823 flags |= Layout::kBlendSupportAllEquations_Flag;
824 break;
825 case LayoutToken::BLEND_SUPPORT_MULTIPLY:
826 flags |= Layout::kBlendSupportMultiply_Flag;
827 break;
828 case LayoutToken::BLEND_SUPPORT_SCREEN:
829 flags |= Layout::kBlendSupportScreen_Flag;
830 break;
831 case LayoutToken::BLEND_SUPPORT_OVERLAY:
832 flags |= Layout::kBlendSupportOverlay_Flag;
833 break;
834 case LayoutToken::BLEND_SUPPORT_DARKEN:
835 flags |= Layout::kBlendSupportDarken_Flag;
836 break;
837 case LayoutToken::BLEND_SUPPORT_LIGHTEN:
838 flags |= Layout::kBlendSupportLighten_Flag;
839 break;
840 case LayoutToken::BLEND_SUPPORT_COLORDODGE:
841 flags |= Layout::kBlendSupportColorDodge_Flag;
842 break;
843 case LayoutToken::BLEND_SUPPORT_COLORBURN:
844 flags |= Layout::kBlendSupportColorBurn_Flag;
845 break;
846 case LayoutToken::BLEND_SUPPORT_HARDLIGHT:
847 flags |= Layout::kBlendSupportHardLight_Flag;
848 break;
849 case LayoutToken::BLEND_SUPPORT_SOFTLIGHT:
850 flags |= Layout::kBlendSupportSoftLight_Flag;
851 break;
852 case LayoutToken::BLEND_SUPPORT_DIFFERENCE:
853 flags |= Layout::kBlendSupportDifference_Flag;
854 break;
855 case LayoutToken::BLEND_SUPPORT_EXCLUSION:
856 flags |= Layout::kBlendSupportExclusion_Flag;
857 break;
858 case LayoutToken::BLEND_SUPPORT_HSL_HUE:
859 flags |= Layout::kBlendSupportHSLHue_Flag;
860 break;
861 case LayoutToken::BLEND_SUPPORT_HSL_SATURATION:
862 flags |= Layout::kBlendSupportHSLSaturation_Flag;
863 break;
864 case LayoutToken::BLEND_SUPPORT_HSL_COLOR:
865 flags |= Layout::kBlendSupportHSLColor_Flag;
866 break;
867 case LayoutToken::BLEND_SUPPORT_HSL_LUMINOSITY:
868 flags |= Layout::kBlendSupportHSLLuminosity_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500869 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700870 case LayoutToken::PUSH_CONSTANT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500871 flags |= Layout::kPushConstant_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500872 break;
Michael Ludwiga4275592018-08-31 10:52:47 -0400873 case LayoutToken::TRACKED:
874 flags |= Layout::kTracked_Flag;
875 break;
Brian Osmanb32d66b2020-04-30 17:12:03 -0400876 case LayoutToken::SRGB_UNPREMUL:
877 flags |= Layout::kSRGBUnpremul_Flag;
878 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700879 case LayoutToken::POINTS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500880 primitive = Layout::kPoints_Primitive;
881 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700882 case LayoutToken::LINES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500883 primitive = Layout::kLines_Primitive;
884 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700885 case LayoutToken::LINE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500886 primitive = Layout::kLineStrip_Primitive;
887 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700888 case LayoutToken::LINES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500889 primitive = Layout::kLinesAdjacency_Primitive;
890 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700891 case LayoutToken::TRIANGLES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500892 primitive = Layout::kTriangles_Primitive;
893 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700894 case LayoutToken::TRIANGLE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500895 primitive = Layout::kTriangleStrip_Primitive;
896 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700897 case LayoutToken::TRIANGLES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500898 primitive = Layout::kTrianglesAdjacency_Primitive;
899 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700900 case LayoutToken::MAX_VERTICES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500901 maxVertices = this->layoutInt();
902 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700903 case LayoutToken::INVOCATIONS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500904 invocations = this->layoutInt();
905 break;
Brian Osmanf59a9612020-04-15 14:18:13 -0400906 case LayoutToken::MARKER:
907 marker = this->layoutCode();
908 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700909 case LayoutToken::WHEN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400910 when = this->layoutCode();
911 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700912 case LayoutToken::KEY:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400913 key = this->layoutKey();
914 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -0400915 case LayoutToken::CTYPE:
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400916 ctype = this->layoutCType();
917 break;
918 default:
919 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
Ethan Nicholasd608c092017-10-26 09:30:08 -0400920 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500921 }
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400922 } else if (Layout::ReadFormat(text, &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500923 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700924 } else {
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400925 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
ethannicholasb3058bd2016-07-01 08:22:01 -0700926 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400927 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700928 break;
929 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400930 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700931 break;
932 }
933 }
934 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500935 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
Brian Osmanf59a9612020-04-15 14:18:13 -0400936 format, primitive, maxVertices, invocations, marker, when, key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700937}
938
Brian Salomonf9f45122016-11-29 11:59:17 -0500939/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Ethan Nicholasa7ceb502019-01-11 10:31:48 -0500940 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT | BUFFER | PLS | PLSIN |
Brian Osman3c358422020-03-23 10:44:12 -0400941 PLSOUT | VARYING)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500942Modifiers Parser::modifiers() {
943 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700944 int flags = 0;
945 for (;;) {
946 // TODO: handle duplicate / incompatible flags
947 switch (peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400948 case Token::Kind::TK_UNIFORM:
ethannicholasb3058bd2016-07-01 08:22:01 -0700949 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500950 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700951 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400952 case Token::Kind::TK_CONST:
ethannicholasb3058bd2016-07-01 08:22:01 -0700953 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500954 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700955 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400956 case Token::Kind::TK_IN:
ethannicholasb3058bd2016-07-01 08:22:01 -0700957 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500958 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700959 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400960 case Token::Kind::TK_OUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700961 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500962 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700963 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400964 case Token::Kind::TK_INOUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700965 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500966 flags |= Modifiers::kIn_Flag;
967 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700968 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400969 case Token::Kind::TK_FLAT:
ethannicholasf789b382016-08-03 12:43:36 -0700970 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500971 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700972 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400973 case Token::Kind::TK_NOPERSPECTIVE:
ethannicholasf789b382016-08-03 12:43:36 -0700974 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500975 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700976 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400977 case Token::Kind::TK_READONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500978 this->nextToken();
979 flags |= Modifiers::kReadOnly_Flag;
980 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400981 case Token::Kind::TK_WRITEONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500982 this->nextToken();
983 flags |= Modifiers::kWriteOnly_Flag;
984 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400985 case Token::Kind::TK_COHERENT:
Brian Salomonf9f45122016-11-29 11:59:17 -0500986 this->nextToken();
987 flags |= Modifiers::kCoherent_Flag;
988 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400989 case Token::Kind::TK_VOLATILE:
Brian Salomonf9f45122016-11-29 11:59:17 -0500990 this->nextToken();
991 flags |= Modifiers::kVolatile_Flag;
992 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400993 case Token::Kind::TK_RESTRICT:
Brian Salomonf9f45122016-11-29 11:59:17 -0500994 this->nextToken();
995 flags |= Modifiers::kRestrict_Flag;
996 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400997 case Token::Kind::TK_BUFFER:
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400998 this->nextToken();
999 flags |= Modifiers::kBuffer_Flag;
1000 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001001 case Token::Kind::TK_HASSIDEEFFECTS:
Ethan Nicholascb670962017-04-20 19:31:52 -04001002 this->nextToken();
1003 flags |= Modifiers::kHasSideEffects_Flag;
1004 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001005 case Token::Kind::TK_PLS:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001006 this->nextToken();
1007 flags |= Modifiers::kPLS_Flag;
1008 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001009 case Token::Kind::TK_PLSIN:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001010 this->nextToken();
1011 flags |= Modifiers::kPLSIn_Flag;
1012 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001013 case Token::Kind::TK_PLSOUT:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001014 this->nextToken();
1015 flags |= Modifiers::kPLSOut_Flag;
1016 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001017 case Token::Kind::TK_VARYING:
Brian Osman3c358422020-03-23 10:44:12 -04001018 this->nextToken();
1019 flags |= Modifiers::kVarying_Flag;
1020 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001021 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001022 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001023 }
1024 }
1025}
1026
Ethan Nicholas11d53972016-11-28 11:23:23 -05001027Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
1028 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -07001029 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001030 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001031 }
1032 return result;
1033}
1034
1035/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001036ASTNode::ID Parser::statement() {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001037 Token start = this->nextToken();
1038 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001039 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001040 return ASTNode::ID::Invalid();
1041 }
1042 this->pushback(start);
ethannicholasb3058bd2016-07-01 08:22:01 -07001043 switch (start.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001044 case Token::Kind::TK_IF: // fall through
1045 case Token::Kind::TK_STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -07001046 return this->ifStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001047 case Token::Kind::TK_FOR:
ethannicholasb3058bd2016-07-01 08:22:01 -07001048 return this->forStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001049 case Token::Kind::TK_DO:
ethannicholasb3058bd2016-07-01 08:22:01 -07001050 return this->doStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001051 case Token::Kind::TK_WHILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001052 return this->whileStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001053 case Token::Kind::TK_SWITCH: // fall through
1054 case Token::Kind::TK_STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -05001055 return this->switchStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001056 case Token::Kind::TK_RETURN:
ethannicholasb3058bd2016-07-01 08:22:01 -07001057 return this->returnStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001058 case Token::Kind::TK_BREAK:
ethannicholasb3058bd2016-07-01 08:22:01 -07001059 return this->breakStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001060 case Token::Kind::TK_CONTINUE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001061 return this->continueStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001062 case Token::Kind::TK_DISCARD:
ethannicholasb3058bd2016-07-01 08:22:01 -07001063 return this->discardStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001064 case Token::Kind::TK_LBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001065 return this->block();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001066 case Token::Kind::TK_SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001067 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001068 RETURN_NODE(start.fOffset, ASTNode::Kind::kBlock);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001069 case Token::Kind::TK_CONST:
Ethan Nicholasfc994162019-06-06 10:04:27 -04001070 return this->varDeclarations();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001071 case Token::Kind::TK_IDENTIFIER:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001072 if (this->isType(this->text(start))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001073 return this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001074 }
John Stiles30212b72020-06-11 17:55:07 -04001075 [[fallthrough]];
ethannicholasb3058bd2016-07-01 08:22:01 -07001076 default:
1077 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001078 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001079}
1080
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001081/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* QUESTION? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001082ASTNode::ID Parser::type() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001083 Token type;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001084 if (!this->expect(Token::Kind::TK_IDENTIFIER, "a type", &type)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001085 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001086 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001087 if (!this->isType(this->text(type))) {
1088 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001089 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001090 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001091 CREATE_NODE(result, type.fOffset, ASTNode::Kind::kType);
1092 ASTNode::TypeData td(this->text(type), false, false);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001093 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1094 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001095 SKSL_INT i;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001096 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001097 CREATE_CHILD(child, result, -1, ASTNode::Kind::kInt, i);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001098 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001099 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001100 }
1101 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001102 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001103 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001104 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001105 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001106 td.fIsNullable = this->checkNext(Token::Kind::TK_QUESTION);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001107 getNode(result).setTypeData(td);
1108 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001109}
1110
John Stiles57a996b2020-04-19 19:05:12 -07001111/* IDENTIFIER LBRACE
1112 varDeclaration+
1113 RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001114ASTNode::ID Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001115 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001116 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001117 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001118 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001119 if (peek().fKind != Token::Kind::TK_LBRACE) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001120 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001121 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -07001122 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001123 this->error(name, "no type named '" + this->text(name) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001124 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001125 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001126 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kInterfaceBlock);
1127 ASTNode::InterfaceBlockData id(mods, this->text(name), 0, "", 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07001128 this->nextToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001129 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001130 ASTNode::ID decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001131 if (!decl) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001132 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001133 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001134 getNode(result).addChild(decl);
1135 ++id.fDeclarationCount;
ethannicholasb3058bd2016-07-01 08:22:01 -07001136 }
John Stiles57a996b2020-04-19 19:05:12 -07001137 if (id.fDeclarationCount == 0) {
1138 this->error(name, "interface block '" + this->text(name) +
1139 "' must contain at least one member");
1140 return ASTNode::ID::Invalid();
1141 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001142 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001143 std::vector<ASTNode> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001144 StringFragment instanceName;
1145 Token instanceNameToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001146 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &instanceNameToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001147 id.fInstanceName = this->text(instanceNameToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001148 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1149 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001150 ASTNode::ID size = this->expression();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001151 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001152 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001153 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001154 getNode(result).addChild(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001155 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001156 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001157 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001158 ++id.fSizeCount;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001159 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001160 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001161 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -07001162 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001163 getNode(result).setInterfaceBlockData(id);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001164 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001165 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001166}
1167
1168/* IF LPAREN expression RPAREN statement (ELSE statement)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001169ASTNode::ID Parser::ifStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001170 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001171 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_IF, &start);
1172 if (!isStatic && !this->expect(Token::Kind::TK_IF, "'if'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001173 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001174 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001175 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kIf, isStatic);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001176 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001177 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001178 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001179 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001180 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001181 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001182 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001183 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001184 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001185 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001186 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001187 ASTNode::ID ifTrue = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001188 if (!ifTrue) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001189 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001190 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001191 getNode(result).addChild(ifTrue);
1192 ASTNode::ID ifFalse;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001193 if (this->checkNext(Token::Kind::TK_ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001194 ifFalse = this->statement();
1195 if (!ifFalse) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001196 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001197 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001198 getNode(result).addChild(ifFalse);
ethannicholasb3058bd2016-07-01 08:22:01 -07001199 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001200 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001201}
1202
1203/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001204ASTNode::ID Parser::doStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001205 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001206 if (!this->expect(Token::Kind::TK_DO, "'do'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001207 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001208 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001209 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kDo);
1210 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001211 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001212 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001213 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001214 getNode(result).addChild(statement);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001215 if (!this->expect(Token::Kind::TK_WHILE, "'while'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001216 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001217 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001218 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001219 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001220 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001221 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001222 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001223 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001224 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001225 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001226 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001227 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001228 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001229 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001230 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001231 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001232 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001233}
1234
1235/* WHILE LPAREN expression RPAREN STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001236ASTNode::ID Parser::whileStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001237 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001238 if (!this->expect(Token::Kind::TK_WHILE, "'while'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001239 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001240 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001241 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001242 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001243 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001244 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kWhile);
1245 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001246 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001247 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001248 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001249 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001250 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001251 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001252 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001253 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001254 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001255 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001256 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001257 getNode(result).addChild(statement);
1258 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001259}
1260
Ethan Nicholasaf197692017-02-27 13:26:45 -05001261/* CASE expression COLON statement* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001262ASTNode::ID Parser::switchCase() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001263 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001264 if (!this->expect(Token::Kind::TK_CASE, "'case'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001265 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001266 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001267 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitchCase);
1268 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001269 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001270 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001271 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001272 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001273 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001274 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001275 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001276 while (this->peek().fKind != Token::Kind::TK_RBRACE &&
1277 this->peek().fKind != Token::Kind::TK_CASE &&
1278 this->peek().fKind != Token::Kind::TK_DEFAULT) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001279 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001280 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001281 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001282 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001283 getNode(result).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001284 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001285 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001286}
1287
1288/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001289ASTNode::ID Parser::switchStatement() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001290 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001291 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_SWITCH, &start);
1292 if (!isStatic && !this->expect(Token::Kind::TK_SWITCH, "'switch'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001293 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001294 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001295 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001296 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001297 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001298 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001299 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001300 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001301 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001302 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001303 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001304 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001305 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001306 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001307 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001308 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitch, isStatic);
1309 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001310 while (this->peek().fKind == Token::Kind::TK_CASE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001311 ASTNode::ID c = this->switchCase();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001312 if (!c) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001313 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001314 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001315 getNode(result).addChild(c);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001316 }
1317 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1318 // parts of the compiler may rely upon this assumption.
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001319 if (this->peek().fKind == Token::Kind::TK_DEFAULT) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001320 Token defaultStart;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001321 SkAssertResult(this->expect(Token::Kind::TK_DEFAULT, "'default'", &defaultStart));
1322 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001323 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001324 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001325 CREATE_CHILD(defaultCase, result, defaultStart.fOffset, ASTNode::Kind::kSwitchCase);
1326 CREATE_EMPTY_CHILD(defaultCase); // empty test to signify default case
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001327 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001328 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001329 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001330 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001331 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001332 getNode(defaultCase).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001333 }
Ethan Nicholasaf197692017-02-27 13:26:45 -05001334 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001335 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001336 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001337 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001338 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001339}
1340
Ethan Nicholas11d53972016-11-28 11:23:23 -05001341/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001342 STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001343ASTNode::ID Parser::forStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001344 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001345 if (!this->expect(Token::Kind::TK_FOR, "'for'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001346 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001347 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001348 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001349 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001350 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001351 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kFor);
1352 ASTNode::ID initializer;
ethannicholasb3058bd2016-07-01 08:22:01 -07001353 Token nextToken = this->peek();
1354 switch (nextToken.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001355 case Token::Kind::TK_SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001356 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001357 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001358 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001359 case Token::Kind::TK_CONST: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001360 initializer = this->varDeclarations();
1361 if (!initializer) {
1362 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001363 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001364 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001365 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001366 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001367 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001368 if (this->isType(this->text(nextToken))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001369 initializer = this->varDeclarations();
1370 if (!initializer) {
1371 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001372 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001373 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001374 break;
1375 }
John Stiles30212b72020-06-11 17:55:07 -04001376 [[fallthrough]];
1377 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001378 default:
1379 initializer = this->expressionStatement();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001380 if (!initializer) {
1381 return ASTNode::ID::Invalid();
1382 }
1383 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001384 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001385 ASTNode::ID test;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001386 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001387 test = this->expression();
1388 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001389 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001390 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001391 getNode(result).addChild(test);
1392 } else {
1393 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001394 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001395 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001396 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001397 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001398 ASTNode::ID next;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001399 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001400 next = this->expression();
1401 if (!next) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001402 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001403 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001404 getNode(result).addChild(next);
1405 } else {
1406 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001407 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001408 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001409 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001410 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001411 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001412 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001413 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001414 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001415 getNode(result).addChild(statement);
1416 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001417}
1418
1419/* RETURN expression? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001420ASTNode::ID Parser::returnStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001421 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001422 if (!this->expect(Token::Kind::TK_RETURN, "'return'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001423 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001424 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001425 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kReturn);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001426 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001427 ASTNode::ID expression = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001428 if (!expression) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001429 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001430 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001431 getNode(result).addChild(expression);
ethannicholasb3058bd2016-07-01 08:22:01 -07001432 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001433 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001434 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001435 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001436 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001437}
1438
1439/* BREAK SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001440ASTNode::ID Parser::breakStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001441 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001442 if (!this->expect(Token::Kind::TK_BREAK, "'break'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001443 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001444 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001445 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001446 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001447 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001448 RETURN_NODE(start.fOffset, ASTNode::Kind::kBreak);
ethannicholasb3058bd2016-07-01 08:22:01 -07001449}
1450
1451/* CONTINUE SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001452ASTNode::ID Parser::continueStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001453 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001454 if (!this->expect(Token::Kind::TK_CONTINUE, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001455 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001456 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001457 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001458 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001459 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001460 RETURN_NODE(start.fOffset, ASTNode::Kind::kContinue);
ethannicholasb3058bd2016-07-01 08:22:01 -07001461}
1462
1463/* DISCARD SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001464ASTNode::ID Parser::discardStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001465 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001466 if (!this->expect(Token::Kind::TK_DISCARD, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001467 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001468 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001469 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001470 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001471 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001472 RETURN_NODE(start.fOffset, ASTNode::Kind::kDiscard);
ethannicholasb3058bd2016-07-01 08:22:01 -07001473}
1474
1475/* LBRACE statement* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001476ASTNode::ID Parser::block() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001477 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001478 if (!this->expect(Token::Kind::TK_LBRACE, "'{'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001479 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001480 }
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001481 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001482 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001483 return ASTNode::ID::Invalid();
1484 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001485 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -07001486 for (;;) {
1487 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001488 case Token::Kind::TK_RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001489 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001490 return result;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001491 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001492 this->error(this->peek(), "expected '}', but found end of file");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001493 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001494 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001495 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001496 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001497 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001498 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001499 getNode(result).addChild(statement);
ethannicholasb3058bd2016-07-01 08:22:01 -07001500 }
1501 }
1502 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001503 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001504}
1505
1506/* expression SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001507ASTNode::ID Parser::expressionStatement() {
1508 ASTNode::ID expr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001509 if (expr) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001510 if (this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001511 return expr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001512 }
1513 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001514 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001515}
1516
1517/* assignmentExpression (COMMA assignmentExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001518ASTNode::ID Parser::expression() {
1519 ASTNode::ID result = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001520 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001521 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001522 }
1523 Token t;
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001524 AutoDepth depth(this);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001525 while (this->checkNext(Token::Kind::TK_COMMA, &t)) {
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001526 if (!depth.increase()) {
1527 return ASTNode::ID::Invalid();
1528 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001529 ASTNode::ID right = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001530 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001531 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001532 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001533 CREATE_NODE(newResult, t.fOffset, ASTNode::Kind::kBinary, std::move(t));
1534 getNode(newResult).addChild(result);
1535 getNode(newResult).addChild(right);
1536 result = newResult;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001537 }
1538 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001539}
1540
1541/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1542 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1543 assignmentExpression)*
1544 */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001545ASTNode::ID Parser::assignmentExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001546 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001547 ASTNode::ID result = this->ternaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001548 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001549 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001550 }
1551 for (;;) {
1552 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001553 case Token::Kind::TK_EQ: // fall through
1554 case Token::Kind::TK_STAREQ: // fall through
1555 case Token::Kind::TK_SLASHEQ: // fall through
1556 case Token::Kind::TK_PERCENTEQ: // fall through
1557 case Token::Kind::TK_PLUSEQ: // fall through
1558 case Token::Kind::TK_MINUSEQ: // fall through
1559 case Token::Kind::TK_SHLEQ: // fall through
1560 case Token::Kind::TK_SHREQ: // fall through
1561 case Token::Kind::TK_BITWISEANDEQ: // fall through
1562 case Token::Kind::TK_BITWISEXOREQ: // fall through
1563 case Token::Kind::TK_BITWISEOREQ: // fall through
1564 case Token::Kind::TK_LOGICALANDEQ: // fall through
1565 case Token::Kind::TK_LOGICALXOREQ: // fall through
1566 case Token::Kind::TK_LOGICALOREQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001567 if (!depth.increase()) {
1568 return ASTNode::ID::Invalid();
1569 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001570 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001571 ASTNode::ID right = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001572 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001573 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001574 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001575 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1576 std::move(t));
1577 getNode(newResult).addChild(result);
1578 getNode(newResult).addChild(right);
1579 result = newResult;
1580 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001581 }
1582 default:
1583 return result;
1584 }
1585 }
1586}
1587
1588/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001589ASTNode::ID Parser::ternaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001590 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001591 ASTNode::ID base = this->logicalOrExpression();
1592 if (!base) {
1593 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001594 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001595 if (this->checkNext(Token::Kind::TK_QUESTION)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001596 if (!depth.increase()) {
1597 return ASTNode::ID::Invalid();
1598 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001599 ASTNode::ID trueExpr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001600 if (!trueExpr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001601 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001602 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001603 if (this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001604 ASTNode::ID falseExpr = this->assignmentExpression();
1605 if (!falseExpr) {
1606 return ASTNode::ID::Invalid();
1607 }
1608 CREATE_NODE(ternary, getNode(base).fOffset, ASTNode::Kind::kTernary);
1609 getNode(ternary).addChild(base);
1610 getNode(ternary).addChild(trueExpr);
1611 getNode(ternary).addChild(falseExpr);
1612 return ternary;
ethannicholasb3058bd2016-07-01 08:22:01 -07001613 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001614 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001615 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001616 return base;
ethannicholasb3058bd2016-07-01 08:22:01 -07001617}
1618
1619/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001620ASTNode::ID Parser::logicalOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001621 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001622 ASTNode::ID result = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001623 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001624 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001625 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001626 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001627 while (this->checkNext(Token::Kind::TK_LOGICALOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001628 if (!depth.increase()) {
1629 return ASTNode::ID::Invalid();
1630 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001631 ASTNode::ID right = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001632 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001633 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001634 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001635 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1636 getNode(newResult).addChild(result);
1637 getNode(newResult).addChild(right);
1638 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001639 }
1640 return result;
1641}
1642
1643/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001644ASTNode::ID Parser::logicalXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001645 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001646 ASTNode::ID result = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001647 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001648 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001649 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001650 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001651 while (this->checkNext(Token::Kind::TK_LOGICALXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001652 if (!depth.increase()) {
1653 return ASTNode::ID::Invalid();
1654 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001655 ASTNode::ID right = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001656 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001657 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001658 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001659 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1660 getNode(newResult).addChild(result);
1661 getNode(newResult).addChild(right);
1662 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001663 }
1664 return result;
1665}
1666
1667/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001668ASTNode::ID Parser::logicalAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001669 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001670 ASTNode::ID result = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001671 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001672 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001673 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001674 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001675 while (this->checkNext(Token::Kind::TK_LOGICALAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001676 if (!depth.increase()) {
1677 return ASTNode::ID::Invalid();
1678 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001679 ASTNode::ID right = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001680 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001681 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001682 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001683 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1684 getNode(newResult).addChild(result);
1685 getNode(newResult).addChild(right);
1686 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001687 }
1688 return result;
1689}
1690
1691/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001692ASTNode::ID Parser::bitwiseOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001693 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001694 ASTNode::ID result = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001695 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001696 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001697 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001698 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001699 while (this->checkNext(Token::Kind::TK_BITWISEOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001700 if (!depth.increase()) {
1701 return ASTNode::ID::Invalid();
1702 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001703 ASTNode::ID right = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001704 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001705 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001706 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001707 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1708 getNode(newResult).addChild(result);
1709 getNode(newResult).addChild(right);
1710 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001711 }
1712 return result;
1713}
1714
1715/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001716ASTNode::ID Parser::bitwiseXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001717 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001718 ASTNode::ID result = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001719 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001720 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001721 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001722 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001723 while (this->checkNext(Token::Kind::TK_BITWISEXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001724 if (!depth.increase()) {
1725 return ASTNode::ID::Invalid();
1726 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001727 ASTNode::ID right = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001728 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001729 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001730 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001731 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1732 getNode(newResult).addChild(result);
1733 getNode(newResult).addChild(right);
1734 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001735 }
1736 return result;
1737}
1738
1739/* equalityExpression (BITWISEAND equalityExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001740ASTNode::ID Parser::bitwiseAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001741 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001742 ASTNode::ID result = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001743 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001744 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001745 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001746 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001747 while (this->checkNext(Token::Kind::TK_BITWISEAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001748 if (!depth.increase()) {
1749 return ASTNode::ID::Invalid();
1750 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001751 ASTNode::ID right = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001752 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001753 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001754 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001755 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1756 getNode(newResult).addChild(result);
1757 getNode(newResult).addChild(right);
1758 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001759 }
1760 return result;
1761}
1762
1763/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001764ASTNode::ID Parser::equalityExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001765 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001766 ASTNode::ID result = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001767 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001768 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001769 }
1770 for (;;) {
1771 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001772 case Token::Kind::TK_EQEQ: // fall through
1773 case Token::Kind::TK_NEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001774 if (!depth.increase()) {
1775 return ASTNode::ID::Invalid();
1776 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001777 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001778 ASTNode::ID right = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001779 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001780 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001781 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001782 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1783 std::move(t));
1784 getNode(newResult).addChild(result);
1785 getNode(newResult).addChild(right);
1786 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001787 break;
1788 }
1789 default:
1790 return result;
1791 }
1792 }
1793}
1794
1795/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001796ASTNode::ID Parser::relationalExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001797 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001798 ASTNode::ID result = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001799 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001800 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001801 }
1802 for (;;) {
1803 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001804 case Token::Kind::TK_LT: // fall through
1805 case Token::Kind::TK_GT: // fall through
1806 case Token::Kind::TK_LTEQ: // fall through
1807 case Token::Kind::TK_GTEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001808 if (!depth.increase()) {
1809 return ASTNode::ID::Invalid();
1810 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001811 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001812 ASTNode::ID right = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001813 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001814 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001815 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001816 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1817 std::move(t));
1818 getNode(newResult).addChild(result);
1819 getNode(newResult).addChild(right);
1820 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001821 break;
1822 }
1823 default:
1824 return result;
1825 }
1826 }
1827}
1828
1829/* additiveExpression ((SHL | SHR) additiveExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001830ASTNode::ID Parser::shiftExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001831 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001832 ASTNode::ID result = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001833 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001834 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001835 }
1836 for (;;) {
1837 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001838 case Token::Kind::TK_SHL: // fall through
1839 case Token::Kind::TK_SHR: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001840 if (!depth.increase()) {
1841 return ASTNode::ID::Invalid();
1842 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001843 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001844 ASTNode::ID right = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001845 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001846 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001847 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001848 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1849 std::move(t));
1850 getNode(newResult).addChild(result);
1851 getNode(newResult).addChild(right);
1852 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001853 break;
1854 }
1855 default:
1856 return result;
1857 }
1858 }
1859}
1860
1861/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001862ASTNode::ID Parser::additiveExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001863 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001864 ASTNode::ID result = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001865 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001866 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001867 }
1868 for (;;) {
1869 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001870 case Token::Kind::TK_PLUS: // fall through
1871 case Token::Kind::TK_MINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001872 if (!depth.increase()) {
1873 return ASTNode::ID::Invalid();
1874 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001875 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001876 ASTNode::ID right = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001877 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001878 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001879 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001880 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1881 std::move(t));
1882 getNode(newResult).addChild(result);
1883 getNode(newResult).addChild(right);
1884 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001885 break;
1886 }
1887 default:
1888 return result;
1889 }
1890 }
1891}
1892
1893/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001894ASTNode::ID Parser::multiplicativeExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001895 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001896 ASTNode::ID result = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001897 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001898 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001899 }
1900 for (;;) {
1901 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001902 case Token::Kind::TK_STAR: // fall through
1903 case Token::Kind::TK_SLASH: // fall through
1904 case Token::Kind::TK_PERCENT: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001905 if (!depth.increase()) {
1906 return ASTNode::ID::Invalid();
1907 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001908 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001909 ASTNode::ID right = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001910 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001911 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001912 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001913 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1914 std::move(t));
1915 getNode(newResult).addChild(result);
1916 getNode(newResult).addChild(right);
1917 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001918 break;
1919 }
1920 default:
1921 return result;
1922 }
1923 }
1924}
1925
1926/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001927ASTNode::ID Parser::unaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001928 AutoDepth depth(this);
ethannicholasb3058bd2016-07-01 08:22:01 -07001929 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001930 case Token::Kind::TK_PLUS: // fall through
1931 case Token::Kind::TK_MINUS: // fall through
1932 case Token::Kind::TK_LOGICALNOT: // fall through
1933 case Token::Kind::TK_BITWISENOT: // fall through
1934 case Token::Kind::TK_PLUSPLUS: // fall through
1935 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001936 if (!depth.increase()) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001937 return ASTNode::ID::Invalid();
Ethan Nicholas6dcc3252019-02-20 15:18:36 -05001938 }
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001939 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001940 ASTNode::ID expr = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001941 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001942 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001943 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001944 CREATE_NODE(result, t.fOffset, ASTNode::Kind::kPrefix, std::move(t));
1945 getNode(result).addChild(expr);
1946 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001947 }
1948 default:
1949 return this->postfixExpression();
1950 }
1951}
1952
1953/* term suffix* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001954ASTNode::ID Parser::postfixExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001955 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001956 ASTNode::ID result = this->term();
ethannicholasb3058bd2016-07-01 08:22:01 -07001957 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001958 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001959 }
1960 for (;;) {
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001961 Token t = this->peek();
1962 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001963 case Token::Kind::TK_FLOAT_LITERAL:
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001964 if (this->text(t)[0] != '.') {
1965 return result;
1966 }
John Stiles30212b72020-06-11 17:55:07 -04001967 [[fallthrough]];
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001968 case Token::Kind::TK_LBRACKET:
1969 case Token::Kind::TK_DOT:
1970 case Token::Kind::TK_LPAREN:
1971 case Token::Kind::TK_PLUSPLUS:
1972 case Token::Kind::TK_MINUSMINUS:
1973 case Token::Kind::TK_COLONCOLON:
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001974 if (!depth.increase()) {
1975 return ASTNode::ID::Invalid();
1976 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001977 result = this->suffix(result);
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001978 if (!result) {
1979 return ASTNode::ID::Invalid();
1980 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001981 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001982 default:
1983 return result;
1984 }
1985 }
1986}
1987
Ethan Nicholas11d53972016-11-28 11:23:23 -05001988/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
Ethan Nicholase455f652019-09-13 12:52:55 -04001989 PLUSPLUS | MINUSMINUS | COLONCOLON IDENTIFIER | FLOAT_LITERAL [IDENTIFIER] */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001990ASTNode::ID Parser::suffix(ASTNode::ID base) {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001991 SkASSERT(base);
ethannicholasb3058bd2016-07-01 08:22:01 -07001992 Token next = this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001993 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001994 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001995 return ASTNode::ID::Invalid();
1996 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001997 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001998 case Token::Kind::TK_LBRACKET: {
1999 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002000 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
2001 getNode(result).addChild(base);
2002 return result;
ethannicholas5961bc92016-10-12 06:39:56 -07002003 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002004 ASTNode::ID e = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002005 if (!e) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002006 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002007 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002008 this->expect(Token::Kind::TK_RBRACKET, "']' to complete array access expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002009 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
2010 getNode(result).addChild(base);
2011 getNode(result).addChild(e);
2012 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002013 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002014 case Token::Kind::TK_DOT: // fall through
2015 case Token::Kind::TK_COLONCOLON: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002016 int offset = this->peek().fOffset;
2017 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002018 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002019 CREATE_NODE(result, offset, ASTNode::Kind::kField, std::move(text));
2020 getNode(result).addChild(base);
2021 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002022 }
John Stiles30212b72020-06-11 17:55:07 -04002023 [[fallthrough]]; // FIXME(ethannicholas)
Ethan Nicholase455f652019-09-13 12:52:55 -04002024 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002025 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholase455f652019-09-13 12:52:55 -04002026 // Swizzles that start with a constant number, e.g. '.000r', will be tokenized as
2027 // floating point literals, possibly followed by an identifier. Handle that here.
2028 StringFragment field = this->text(next);
2029 SkASSERT(field.fChars[0] == '.');
2030 ++field.fChars;
2031 --field.fLength;
2032 for (size_t i = 0; i < field.fLength; ++i) {
2033 if (field.fChars[i] != '0' && field.fChars[i] != '1') {
2034 this->error(next, "invalid swizzle");
2035 return ASTNode::ID::Invalid();
2036 }
2037 }
2038 // use the next *raw* token so we don't ignore whitespace - we only care about
2039 // identifiers that directly follow the float
2040 Token id = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002041 if (id.fKind == Token::Kind::TK_IDENTIFIER) {
Ethan Nicholase455f652019-09-13 12:52:55 -04002042 field.fLength += id.fLength;
2043 } else {
2044 this->pushback(id);
2045 }
2046 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kField, field);
2047 getNode(result).addChild(base);
2048 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002049 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002050 case Token::Kind::TK_LPAREN: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002051 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kCall);
2052 getNode(result).addChild(base);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002053 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002054 for (;;) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002055 ASTNode::ID expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002056 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002057 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002058 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002059 getNode(result).addChild(expr);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002060 if (!this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002061 break;
2062 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002063 }
2064 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002065 this->expect(Token::Kind::TK_RPAREN, "')' to complete function parameters");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002066 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002067 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002068 case Token::Kind::TK_PLUSPLUS: // fall through
2069 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002070 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kPostfix, next);
2071 getNode(result).addChild(base);
2072 return result;
2073 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002074 default: {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002075 this->error(next, "expected expression suffix, but found '" + this->text(next) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002076 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002077 }
2078 }
2079}
2080
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002081/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | NULL_LITERAL | '(' expression ')' */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002082ASTNode::ID Parser::term() {
ethannicholasb3058bd2016-07-01 08:22:01 -07002083 Token t = this->peek();
2084 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002085 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002086 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002087 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002088 RETURN_NODE(t.fOffset, ASTNode::Kind::kIdentifier, std::move(text));
ethannicholasb3058bd2016-07-01 08:22:01 -07002089 }
John Stiles30212b72020-06-11 17:55:07 -04002090 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002091 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002092 case Token::Kind::TK_INT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002093 SKSL_INT i;
ethannicholasb3058bd2016-07-01 08:22:01 -07002094 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002095 RETURN_NODE(t.fOffset, ASTNode::Kind::kInt, i);
ethannicholasb3058bd2016-07-01 08:22:01 -07002096 }
2097 break;
2098 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002099 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002100 SKSL_FLOAT f;
ethannicholasb3058bd2016-07-01 08:22:01 -07002101 if (this->floatLiteral(&f)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002102 RETURN_NODE(t.fOffset, ASTNode::Kind::kFloat, f);
ethannicholasb3058bd2016-07-01 08:22:01 -07002103 }
2104 break;
2105 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002106 case Token::Kind::TK_TRUE_LITERAL: // fall through
2107 case Token::Kind::TK_FALSE_LITERAL: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002108 bool b;
2109 if (this->boolLiteral(&b)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002110 RETURN_NODE(t.fOffset, ASTNode::Kind::kBool, b);
ethannicholasb3058bd2016-07-01 08:22:01 -07002111 }
2112 break;
2113 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002114 case Token::Kind::TK_NULL_LITERAL:
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002115 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04002116 RETURN_NODE(t.fOffset, ASTNode::Kind::kNull);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002117 case Token::Kind::TK_LPAREN: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002118 this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002119 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002120 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002121 return ASTNode::ID::Invalid();
2122 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002123 ASTNode::ID result = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002124 if (result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002125 this->expect(Token::Kind::TK_RPAREN, "')' to complete expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002126 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002127 }
2128 break;
2129 }
2130 default:
2131 this->nextToken();
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002132 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002133 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002134 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002135}
2136
2137/* INT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002138bool Parser::intLiteral(SKSL_INT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002139 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002140 if (this->expect(Token::Kind::TK_INT_LITERAL, "integer literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002141 *dest = SkSL::stol(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002142 return true;
2143 }
2144 return false;
2145}
2146
2147/* FLOAT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002148bool Parser::floatLiteral(SKSL_FLOAT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002149 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002150 if (this->expect(Token::Kind::TK_FLOAT_LITERAL, "float literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002151 *dest = SkSL::stod(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002152 return true;
2153 }
2154 return false;
2155}
2156
2157/* TRUE_LITERAL | FALSE_LITERAL */
2158bool Parser::boolLiteral(bool* dest) {
2159 Token t = this->nextToken();
2160 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002161 case Token::Kind::TK_TRUE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002162 *dest = true;
2163 return true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002164 case Token::Kind::TK_FALSE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002165 *dest = false;
2166 return true;
2167 default:
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002168 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002169 return false;
2170 }
2171}
2172
2173/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002174bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002175 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002176 if (this->expect(Token::Kind::TK_IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002177 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07002178 return true;
2179 }
2180 return false;
2181}
2182
2183} // namespace