blob: 9104e5a85b065ae4b1bfe263faa073317f69c807 [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");
88 TOKEN(WHEN, "when");
89 TOKEN(KEY, "key");
Michael Ludwiga4275592018-08-31 10:52:47 -040090 TOKEN(TRACKED, "tracked");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040091 TOKEN(CTYPE, "ctype");
Brian Osmanf28e55d2018-10-03 16:35:54 -040092 TOKEN(SKPMCOLOR4F, "SkPMColor4f");
Mike Reedb26b4e72020-01-22 14:31:21 -050093 TOKEN(SKV4, "SkV4");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040094 TOKEN(SKRECT, "SkRect");
95 TOKEN(SKIRECT, "SkIRect");
96 TOKEN(SKPMCOLOR, "SkPMColor");
Mike Reedb26b4e72020-01-22 14:31:21 -050097 TOKEN(SKM44, "SkM44");
Ethan Nicholasc1c686b2019-04-02 17:30:23 -040098 TOKEN(BOOL, "bool");
99 TOKEN(INT, "int");
100 TOKEN(FLOAT, "float");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400101 #undef TOKEN
102}
103
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700104Parser::Parser(const char* text, size_t length, SymbolTable& types, ErrorReporter& errors)
105: fText(text)
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400106, fPushback(Token::Kind::TK_INVALID, -1, -1)
ethannicholasb3058bd2016-07-01 08:22:01 -0700107, fTypes(types)
108, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700109 fLexer.start(text, length);
Brian Salomon3b83afe2018-08-23 11:04:36 -0400110 static const bool layoutMapInitialized = []{ return (void)InitLayoutMap(), true; }();
111 (void) layoutMapInitialized;
ethannicholasb3058bd2016-07-01 08:22:01 -0700112}
113
Ethan Nicholasfc994162019-06-06 10:04:27 -0400114#define CREATE_NODE(result, ...) \
115 ASTNode::ID result(fFile->fNodes.size()); \
116 fFile->fNodes.emplace_back(&fFile->fNodes, __VA_ARGS__)
117
118#define RETURN_NODE(...) \
119 do { \
120 CREATE_NODE(result, __VA_ARGS__); \
121 return result; \
122 } while (false)
123
124#define CREATE_CHILD(child, target, ...) \
125 CREATE_NODE(child, __VA_ARGS__); \
126 fFile->fNodes[target.fValue].addChild(child)
127
128#define CREATE_EMPTY_CHILD(target) \
129 do { \
130 ASTNode::ID child(fFile->fNodes.size()); \
131 fFile->fNodes.emplace_back(); \
132 fFile->fNodes[target.fValue].addChild(child); \
133 } while (false)
134
Ethan Nicholasf7b88202017-09-18 14:10:39 -0400135/* (directive | section | declaration)* END_OF_FILE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400136std::unique_ptr<ASTFile> Parser::file() {
137 fFile.reset(new ASTFile());
138 CREATE_NODE(result, 0, ASTNode::Kind::kFile);
139 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700140 for (;;) {
141 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400142 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400143 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400144 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400145 ASTNode::ID dir = this->directive();
146 if (fErrors.errorCount()) {
147 return nullptr;
148 }
149 if (dir) {
150 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700151 }
152 break;
153 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400154 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400155 ASTNode::ID section = this->section();
156 if (fErrors.errorCount()) {
157 return nullptr;
158 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400159 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400160 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400161 }
162 break;
163 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700164 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400165 ASTNode::ID decl = this->declaration();
166 if (fErrors.errorCount()) {
167 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700168 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400169 if (decl) {
170 getNode(result).addChild(decl);
171 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700172 }
173 }
174 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400175 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700176}
177
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700178Token Parser::nextRawToken() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400179 if (fPushback.fKind != Token::Kind::TK_INVALID) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700180 Token result = fPushback;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400181 fPushback.fKind = Token::Kind::TK_INVALID;
ethannicholasb3058bd2016-07-01 08:22:01 -0700182 return result;
183 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700184 Token result = fLexer.next();
185 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400186}
187
188Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700189 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400190 while (token.fKind == Token::Kind::TK_WHITESPACE ||
191 token.fKind == Token::Kind::TK_LINE_COMMENT ||
192 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700193 token = this->nextRawToken();
194 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400195 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700196}
197
198void Parser::pushback(Token t) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400199 SkASSERT(fPushback.fKind == Token::Kind::TK_INVALID);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400200 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700201}
202
203Token Parser::peek() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400204 if (fPushback.fKind == Token::Kind::TK_INVALID) {
Brian Osman634624a2017-08-15 11:14:30 -0400205 fPushback = this->nextToken();
206 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700207 return fPushback;
208}
209
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400210bool Parser::checkNext(Token::Kind kind, Token* result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400211 if (fPushback.fKind != Token::Kind::TK_INVALID && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400212 return false;
213 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400214 Token next = this->nextToken();
215 if (next.fKind == kind) {
216 if (result) {
217 *result = next;
218 }
219 return true;
220 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400221 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400222 return false;
223}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500224
225bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700226 Token next = this->nextToken();
227 if (next.fKind == kind) {
228 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400229 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700230 }
231 return true;
232 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700233 this->error(next, "expected " + String(expected) + ", but found '" +
234 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700235 return false;
236 }
237}
238
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700239StringFragment Parser::text(Token token) {
240 return StringFragment(fText + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500241}
242
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700243void Parser::error(Token token, String msg) {
244 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700245}
246
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700247void Parser::error(int offset, String msg) {
248 fErrors.error(offset, msg);
249}
250
251bool Parser::isType(StringFragment name) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700252 return nullptr != fTypes[name];
253}
254
Ethan Nicholas11d53972016-11-28 11:23:23 -0500255/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700256 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400257ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700258 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400259 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400260 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700261 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700262 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400263 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700264 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400265 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400266 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700267 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400268 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400269 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700270 }
271 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400272 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400273 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700274 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400275 RETURN_NODE(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700276 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700277 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400278 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700279 }
280}
281
Ethan Nicholas762466e2017-06-29 10:03:38 -0400282/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
283 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400284ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400285 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400286 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400287 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400288 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400289 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400290 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400291 this->nextToken();
292 Token argToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400293 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400294 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400295 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700296 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400297 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400298 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400299 }
300 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400301 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400302 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400303 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400304 StringFragment text;
305 Token codeStart = this->nextRawToken();
306 size_t startOffset = codeStart.fOffset;
307 this->pushback(codeStart);
308 text.fChars = fText + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400309 int level = 1;
310 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700311 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400312 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400313 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400314 ++level;
315 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400316 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400317 --level;
318 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400319 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700320 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400321 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400322 default:
323 break;
324 }
325 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400326 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400327 break;
328 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400329 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700330 StringFragment name = this->text(start);
331 ++name.fChars;
332 --name.fLength;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400333 RETURN_NODE(start.fOffset, ASTNode::Kind::kSection,
334 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400335}
336
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500337/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
338 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400339ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500340 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400341 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400342 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500343 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400344 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400345 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500346 }
347 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400348 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400349 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500350 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400351 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400352 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500353 }
354 fTypes.add(this->text(name), std::unique_ptr<Symbol>(new Type(this->text(name),
355 Type::kEnum_Kind)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400356 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400357 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500358 Token id;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400359 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400360 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500361 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400362 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400363 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500364 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400365 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500366 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400367 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
368 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500369 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400370 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500371 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400372 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
373 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400374 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500375 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400376 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
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->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400380 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500381 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400382 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500383 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400384 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
385 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500386 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400387 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500388 }
389 }
390 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400391 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400392 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500393}
394
395/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700396 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400397ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700398 Token lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400399 if (lookahead.fKind == Token::Kind::TK_ENUM) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500400 return this->enumDeclaration();
401 }
402 Modifiers modifiers = this->modifiers();
403 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400404 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700405 // we have an identifier that's not a type, could be the start of an interface block
406 return this->interfaceBlock(modifiers);
407 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400408 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700409 return this->structVarDeclaration(modifiers);
410 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400411 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700412 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400413 RETURN_NODE(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700414 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400415 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700416 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400417 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700418 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400419 if (getNode(type).getTypeData().fIsStructDeclaration &&
420 this->checkNext(Token::Kind::TK_SEMICOLON)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400421 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700422 }
423 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400424 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400425 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700426 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400427 if (this->checkNext(Token::Kind::TK_LPAREN)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400428 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kFunction);
429 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
430 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400431 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400432 for (;;) {
433 ASTNode::ID parameter = this->parameter();
434 if (!parameter) {
435 return ASTNode::ID::Invalid();
436 }
437 ++fd.fParameterCount;
438 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400439 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400440 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700441 }
442 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700443 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400444 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400445 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400446 return ASTNode::ID::Invalid();
447 }
448 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400449 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700450 body = this->block();
451 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400452 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700453 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400454 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700455 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400456 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700457 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400458 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700459 }
460}
461
462/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400463ASTNode::ID Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500464 Modifiers modifiers = this->modifiers();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400465 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700466 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400467 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700468 }
469 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400470 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400471 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700472 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400473 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700474}
475
476/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400477ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400478 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400479 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700480 }
481 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400482 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400483 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700484 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400485 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400486 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700487 }
488 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400489 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400490 ASTNode::ID decls = this->varDeclarations();
491 if (!decls) {
492 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700493 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400494 ASTNode& declsNode = getNode(decls);
495 auto type = (const Type*) fTypes[(declsNode.begin() + 1)->getTypeData().fName];
496 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
497 ASTNode& var = *iter;
498 ASTNode::VarData vd = var.getVarData();
499 for (int j = vd.fSizeCount - 1; j >= 0; j--) {
500 const ASTNode& size = *(var.begin() + j);
501 if (!size || size.fKind != ASTNode::Kind::kInt) {
502 this->error(declsNode.fOffset, "array size in struct field must be a constant");
503 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700504 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400505 uint64_t columns = size.getInt();
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400506 String name = type->name() + "[" + to_string(columns) + "]";
Ethan Nicholas91164d12019-05-15 15:29:54 -0400507 type = (Type*) fTypes.takeOwnership(std::unique_ptr<Symbol>(
508 new Type(name,
509 Type::kArray_Kind,
510 *type,
511 (int) columns)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700512 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400513 fields.push_back(Type::Field(declsNode.begin()->getModifiers(), vd.fName, type));
514 if (vd.fSizeCount ? (var.begin() + (vd.fSizeCount - 1))->fNext : var.fFirstChild) {
515 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700516 }
517 }
518 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400519 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400520 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700521 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700522 fTypes.add(this->text(name), std::unique_ptr<Type>(new Type(name.fOffset, this->text(name),
523 fields)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400524 RETURN_NODE(name.fOffset, ASTNode::Kind::kType,
525 ASTNode::TypeData(this->text(name), true, false));
ethannicholasb3058bd2016-07-01 08:22:01 -0700526}
527
528/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400529ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
530 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700531 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400532 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700533 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400534 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400535 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400536 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700537 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400538 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400539 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700540}
541
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400542/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
543 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400544ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
545 CREATE_NODE(result, -1, ASTNode::Kind::kVarDeclarations);
546 CREATE_CHILD(modifiers, result, -1, ASTNode::Kind::kModifiers, mods);
547 getNode(result).addChild(type);
548 CREATE_NODE(currentVar, -1, ASTNode::Kind::kVarDeclaration);
549 ASTNode::VarData vd(name, 0);
550 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400551 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
552 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400553 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700554 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400555 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700556 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400557 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700558 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400559 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400560 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400561 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700562 }
563 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400564 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700565 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400566 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400567 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400568 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700569 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400570 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700571 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400572 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700573 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400574 while (this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700575 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400576 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400577 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700578 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400579 currentVar = ASTNode::ID(fFile->fNodes.size());
580 vd = ASTNode::VarData(this->text(name), 0);
581 fFile->fNodes.emplace_back(&fFile->fNodes, -1, ASTNode::Kind::kVarDeclaration);
582 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400583 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
584 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400585 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700586 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400587 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700588 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400589 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700590 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400591 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400592 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400593 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700594 }
595 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400596 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700597 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400598 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400599 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400600 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700601 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400602 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700603 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400604 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700605 }
606 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400607 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400608 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700609 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400610 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700611}
612
613/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400614ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400615 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400616 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700617 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400618 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700619 }
620 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400621 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400622 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700623 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400624 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kParameter);
625 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
626 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400627 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700628 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400629 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400630 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700631 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400632 CREATE_CHILD(child, result, sizeToken.fOffset, ASTNode::Kind::kInt,
633 SkSL::stoi(this->text(sizeToken)));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400634 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
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 ++pd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700638 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400639 getNode(result).setParameterData(pd);
640 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700641}
642
Ethan Nicholasd608c092017-10-26 09:30:08 -0400643/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700644int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400645 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700646 return -1;
647 }
648 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400649 if (this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700650 return SkSL::stoi(this->text(resultToken));
ethannicholasb3058bd2016-07-01 08:22:01 -0700651 }
652 return -1;
653}
654
Ethan Nicholasd608c092017-10-26 09:30:08 -0400655/** EQ IDENTIFIER */
656StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400657 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400658 return StringFragment();
659 }
660 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400661 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400662 return StringFragment();
663 }
664 return this->text(resultToken);
665}
666
667
Ethan Nicholas762466e2017-06-29 10:03:38 -0400668/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400669StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400670 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400671 return "";
672 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700673 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400674 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400675 StringFragment code;
676 code.fChars = fText + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400677 int level = 1;
678 bool done = false;
679 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700680 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400681 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400682 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400683 ++level;
684 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400685 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400686 --level;
687 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400688 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400689 if (level == 1) {
690 done = true;
691 }
692 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400693 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700694 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400695 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400696 default:
697 break;
698 }
699 if (!level) {
700 done = true;
701 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400702 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400703 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400704 this->pushback(std::move(next));
705 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400706 }
707 return code;
708}
709
710/** (EQ IDENTIFIER('identity'))? */
711Layout::Key Parser::layoutKey() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400712 if (this->peek().fKind == Token::Kind::TK_EQ) {
713 this->expect(Token::Kind::TK_EQ, "'='");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400714 Token key;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400715 if (this->expect(Token::Kind::TK_IDENTIFIER, "an identifer", &key)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700716 if (this->text(key) == "identity") {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400717 return Layout::kIdentity_Key;
718 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700719 this->error(key, "unsupported layout key");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400720 }
721 }
722 }
723 return Layout::kKey_Key;
724}
725
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400726Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400727 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400728 Token t = this->nextToken();
729 String text = this->text(t);
730 auto found = layoutTokens->find(text);
731 if (found != layoutTokens->end()) {
732 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400733 case LayoutToken::SKPMCOLOR4F:
734 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500735 case LayoutToken::SKV4:
736 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400737 case LayoutToken::SKRECT:
738 return Layout::CType::kSkRect;
739 case LayoutToken::SKIRECT:
740 return Layout::CType::kSkIRect;
741 case LayoutToken::SKPMCOLOR:
742 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400743 case LayoutToken::BOOL:
744 return Layout::CType::kBool;
745 case LayoutToken::INT:
746 return Layout::CType::kInt32;
747 case LayoutToken::FLOAT:
748 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500749 case LayoutToken::SKM44:
750 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400751 default:
752 break;
753 }
754 }
755 this->error(t, "unsupported ctype");
756 }
757 return Layout::CType::kDefault;
758}
759
ethannicholas8ac838d2016-11-22 08:39:36 -0800760/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500761Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500762 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700763 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500764 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700765 int binding = -1;
766 int index = -1;
767 int set = -1;
768 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500769 int inputAttachmentIndex = -1;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500770 Layout::Format format = Layout::Format::kUnspecified;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500771 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
772 int maxVertices = -1;
773 int invocations = -1;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400774 StringFragment when;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400775 Layout::Key key = Layout::kNo_Key;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400776 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400777 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
778 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500779 return Layout(flags, location, offset, binding, index, set, builtin,
780 inputAttachmentIndex, format, primitive, maxVertices, invocations, when,
781 key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700782 }
783 for (;;) {
784 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700785 String text = this->text(t);
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400786 auto found = layoutTokens->find(text);
787 if (found != layoutTokens->end()) {
788 switch (found->second) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700789 case LayoutToken::LOCATION:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500790 location = this->layoutInt();
791 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700792 case LayoutToken::OFFSET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500793 offset = this->layoutInt();
794 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700795 case LayoutToken::BINDING:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500796 binding = this->layoutInt();
797 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700798 case LayoutToken::INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500799 index = this->layoutInt();
800 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700801 case LayoutToken::SET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500802 set = this->layoutInt();
803 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700804 case LayoutToken::BUILTIN:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500805 builtin = this->layoutInt();
806 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700807 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500808 inputAttachmentIndex = this->layoutInt();
809 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700810 case LayoutToken::ORIGIN_UPPER_LEFT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500811 flags |= Layout::kOriginUpperLeft_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500812 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700813 case LayoutToken::OVERRIDE_COVERAGE:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500814 flags |= Layout::kOverrideCoverage_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500815 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700816 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500817 flags |= Layout::kBlendSupportAllEquations_Flag;
818 break;
819 case LayoutToken::BLEND_SUPPORT_MULTIPLY:
820 flags |= Layout::kBlendSupportMultiply_Flag;
821 break;
822 case LayoutToken::BLEND_SUPPORT_SCREEN:
823 flags |= Layout::kBlendSupportScreen_Flag;
824 break;
825 case LayoutToken::BLEND_SUPPORT_OVERLAY:
826 flags |= Layout::kBlendSupportOverlay_Flag;
827 break;
828 case LayoutToken::BLEND_SUPPORT_DARKEN:
829 flags |= Layout::kBlendSupportDarken_Flag;
830 break;
831 case LayoutToken::BLEND_SUPPORT_LIGHTEN:
832 flags |= Layout::kBlendSupportLighten_Flag;
833 break;
834 case LayoutToken::BLEND_SUPPORT_COLORDODGE:
835 flags |= Layout::kBlendSupportColorDodge_Flag;
836 break;
837 case LayoutToken::BLEND_SUPPORT_COLORBURN:
838 flags |= Layout::kBlendSupportColorBurn_Flag;
839 break;
840 case LayoutToken::BLEND_SUPPORT_HARDLIGHT:
841 flags |= Layout::kBlendSupportHardLight_Flag;
842 break;
843 case LayoutToken::BLEND_SUPPORT_SOFTLIGHT:
844 flags |= Layout::kBlendSupportSoftLight_Flag;
845 break;
846 case LayoutToken::BLEND_SUPPORT_DIFFERENCE:
847 flags |= Layout::kBlendSupportDifference_Flag;
848 break;
849 case LayoutToken::BLEND_SUPPORT_EXCLUSION:
850 flags |= Layout::kBlendSupportExclusion_Flag;
851 break;
852 case LayoutToken::BLEND_SUPPORT_HSL_HUE:
853 flags |= Layout::kBlendSupportHSLHue_Flag;
854 break;
855 case LayoutToken::BLEND_SUPPORT_HSL_SATURATION:
856 flags |= Layout::kBlendSupportHSLSaturation_Flag;
857 break;
858 case LayoutToken::BLEND_SUPPORT_HSL_COLOR:
859 flags |= Layout::kBlendSupportHSLColor_Flag;
860 break;
861 case LayoutToken::BLEND_SUPPORT_HSL_LUMINOSITY:
862 flags |= Layout::kBlendSupportHSLLuminosity_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500863 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700864 case LayoutToken::PUSH_CONSTANT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500865 flags |= Layout::kPushConstant_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500866 break;
Michael Ludwiga4275592018-08-31 10:52:47 -0400867 case LayoutToken::TRACKED:
868 flags |= Layout::kTracked_Flag;
869 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700870 case LayoutToken::POINTS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500871 primitive = Layout::kPoints_Primitive;
872 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700873 case LayoutToken::LINES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500874 primitive = Layout::kLines_Primitive;
875 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700876 case LayoutToken::LINE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500877 primitive = Layout::kLineStrip_Primitive;
878 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700879 case LayoutToken::LINES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500880 primitive = Layout::kLinesAdjacency_Primitive;
881 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700882 case LayoutToken::TRIANGLES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500883 primitive = Layout::kTriangles_Primitive;
884 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700885 case LayoutToken::TRIANGLE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500886 primitive = Layout::kTriangleStrip_Primitive;
887 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700888 case LayoutToken::TRIANGLES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500889 primitive = Layout::kTrianglesAdjacency_Primitive;
890 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700891 case LayoutToken::MAX_VERTICES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500892 maxVertices = this->layoutInt();
893 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700894 case LayoutToken::INVOCATIONS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500895 invocations = this->layoutInt();
896 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700897 case LayoutToken::WHEN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400898 when = this->layoutCode();
899 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700900 case LayoutToken::KEY:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400901 key = this->layoutKey();
902 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -0400903 case LayoutToken::CTYPE:
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400904 ctype = this->layoutCType();
905 break;
906 default:
907 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
Ethan Nicholasd608c092017-10-26 09:30:08 -0400908 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500909 }
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400910 } else if (Layout::ReadFormat(text, &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500911 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700912 } else {
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400913 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
ethannicholasb3058bd2016-07-01 08:22:01 -0700914 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400915 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700916 break;
917 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400918 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700919 break;
920 }
921 }
922 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500923 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
924 format, primitive, maxVertices, invocations, when, key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700925}
926
Brian Salomonf9f45122016-11-29 11:59:17 -0500927/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Ethan Nicholasa7ceb502019-01-11 10:31:48 -0500928 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT | BUFFER | PLS | PLSIN |
Brian Osman3c358422020-03-23 10:44:12 -0400929 PLSOUT | VARYING)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500930Modifiers Parser::modifiers() {
931 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700932 int flags = 0;
933 for (;;) {
934 // TODO: handle duplicate / incompatible flags
935 switch (peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400936 case Token::Kind::TK_UNIFORM:
ethannicholasb3058bd2016-07-01 08:22:01 -0700937 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500938 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700939 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400940 case Token::Kind::TK_CONST:
ethannicholasb3058bd2016-07-01 08:22:01 -0700941 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500942 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700943 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400944 case Token::Kind::TK_IN:
ethannicholasb3058bd2016-07-01 08:22:01 -0700945 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500946 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700947 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400948 case Token::Kind::TK_OUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700949 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500950 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700951 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400952 case Token::Kind::TK_INOUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700953 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500954 flags |= Modifiers::kIn_Flag;
955 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700956 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400957 case Token::Kind::TK_FLAT:
ethannicholasf789b382016-08-03 12:43:36 -0700958 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500959 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700960 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400961 case Token::Kind::TK_NOPERSPECTIVE:
ethannicholasf789b382016-08-03 12:43:36 -0700962 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500963 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700964 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400965 case Token::Kind::TK_READONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500966 this->nextToken();
967 flags |= Modifiers::kReadOnly_Flag;
968 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400969 case Token::Kind::TK_WRITEONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500970 this->nextToken();
971 flags |= Modifiers::kWriteOnly_Flag;
972 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400973 case Token::Kind::TK_COHERENT:
Brian Salomonf9f45122016-11-29 11:59:17 -0500974 this->nextToken();
975 flags |= Modifiers::kCoherent_Flag;
976 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400977 case Token::Kind::TK_VOLATILE:
Brian Salomonf9f45122016-11-29 11:59:17 -0500978 this->nextToken();
979 flags |= Modifiers::kVolatile_Flag;
980 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400981 case Token::Kind::TK_RESTRICT:
Brian Salomonf9f45122016-11-29 11:59:17 -0500982 this->nextToken();
983 flags |= Modifiers::kRestrict_Flag;
984 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400985 case Token::Kind::TK_BUFFER:
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400986 this->nextToken();
987 flags |= Modifiers::kBuffer_Flag;
988 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400989 case Token::Kind::TK_HASSIDEEFFECTS:
Ethan Nicholascb670962017-04-20 19:31:52 -0400990 this->nextToken();
991 flags |= Modifiers::kHasSideEffects_Flag;
992 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400993 case Token::Kind::TK_PLS:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -0500994 this->nextToken();
995 flags |= Modifiers::kPLS_Flag;
996 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400997 case Token::Kind::TK_PLSIN:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -0500998 this->nextToken();
999 flags |= Modifiers::kPLSIn_Flag;
1000 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001001 case Token::Kind::TK_PLSOUT:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001002 this->nextToken();
1003 flags |= Modifiers::kPLSOut_Flag;
1004 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001005 case Token::Kind::TK_VARYING:
Brian Osman3c358422020-03-23 10:44:12 -04001006 this->nextToken();
1007 flags |= Modifiers::kVarying_Flag;
1008 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001009 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001010 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001011 }
1012 }
1013}
1014
Ethan Nicholas11d53972016-11-28 11:23:23 -05001015Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
1016 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -07001017 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001018 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001019 }
1020 return result;
1021}
1022
1023/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001024ASTNode::ID Parser::statement() {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001025 Token start = this->nextToken();
1026 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001027 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001028 return ASTNode::ID::Invalid();
1029 }
1030 this->pushback(start);
ethannicholasb3058bd2016-07-01 08:22:01 -07001031 switch (start.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001032 case Token::Kind::TK_IF: // fall through
1033 case Token::Kind::TK_STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -07001034 return this->ifStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001035 case Token::Kind::TK_FOR:
ethannicholasb3058bd2016-07-01 08:22:01 -07001036 return this->forStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001037 case Token::Kind::TK_DO:
ethannicholasb3058bd2016-07-01 08:22:01 -07001038 return this->doStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001039 case Token::Kind::TK_WHILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001040 return this->whileStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001041 case Token::Kind::TK_SWITCH: // fall through
1042 case Token::Kind::TK_STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -05001043 return this->switchStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001044 case Token::Kind::TK_RETURN:
ethannicholasb3058bd2016-07-01 08:22:01 -07001045 return this->returnStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001046 case Token::Kind::TK_BREAK:
ethannicholasb3058bd2016-07-01 08:22:01 -07001047 return this->breakStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001048 case Token::Kind::TK_CONTINUE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001049 return this->continueStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001050 case Token::Kind::TK_DISCARD:
ethannicholasb3058bd2016-07-01 08:22:01 -07001051 return this->discardStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001052 case Token::Kind::TK_LBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001053 return this->block();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001054 case Token::Kind::TK_SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001055 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001056 RETURN_NODE(start.fOffset, ASTNode::Kind::kBlock);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001057 case Token::Kind::TK_CONST:
Ethan Nicholasfc994162019-06-06 10:04:27 -04001058 return this->varDeclarations();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001059 case Token::Kind::TK_IDENTIFIER:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001060 if (this->isType(this->text(start))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001061 return this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001062 }
1063 // fall through
1064 default:
1065 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001066 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001067}
1068
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001069/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* QUESTION? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001070ASTNode::ID Parser::type() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001071 Token type;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001072 if (!this->expect(Token::Kind::TK_IDENTIFIER, "a type", &type)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001073 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001074 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001075 if (!this->isType(this->text(type))) {
1076 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001077 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001078 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001079 CREATE_NODE(result, type.fOffset, ASTNode::Kind::kType);
1080 ASTNode::TypeData td(this->text(type), false, false);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001081 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1082 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001083 SKSL_INT i;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001084 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001085 CREATE_CHILD(child, result, -1, ASTNode::Kind::kInt, i);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001086 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001087 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001088 }
1089 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001090 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001091 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001092 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001093 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001094 td.fIsNullable = this->checkNext(Token::Kind::TK_QUESTION);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001095 getNode(result).setTypeData(td);
1096 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001097}
1098
Ethan Nicholas50afc172017-02-16 14:49:57 -05001099/* IDENTIFIER LBRACE varDeclaration* RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001100ASTNode::ID Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001101 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001102 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001103 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001104 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001105 if (peek().fKind != Token::Kind::TK_LBRACE) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001106 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001107 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -07001108 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001109 this->error(name, "no type named '" + this->text(name) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001110 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001111 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001112 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kInterfaceBlock);
1113 ASTNode::InterfaceBlockData id(mods, this->text(name), 0, "", 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07001114 this->nextToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001115 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001116 ASTNode::ID decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001117 if (!decl) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001118 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001119 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001120 getNode(result).addChild(decl);
1121 ++id.fDeclarationCount;
ethannicholasb3058bd2016-07-01 08:22:01 -07001122 }
1123 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001124 std::vector<ASTNode> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001125 StringFragment instanceName;
1126 Token instanceNameToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001127 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &instanceNameToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001128 id.fInstanceName = this->text(instanceNameToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001129 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1130 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001131 ASTNode::ID size = this->expression();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001132 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001133 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001134 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001135 getNode(result).addChild(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001136 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001137 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001138 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001139 ++id.fSizeCount;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001140 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001141 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001142 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -07001143 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001144 getNode(result).setInterfaceBlockData(id);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001145 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001146 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001147}
1148
1149/* IF LPAREN expression RPAREN statement (ELSE statement)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001150ASTNode::ID Parser::ifStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001151 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001152 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_IF, &start);
1153 if (!isStatic && !this->expect(Token::Kind::TK_IF, "'if'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001154 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001155 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001156 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kIf, isStatic);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001157 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001158 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001159 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001160 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001161 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001162 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001163 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001164 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001165 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001166 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001167 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001168 ASTNode::ID ifTrue = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001169 if (!ifTrue) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001170 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001171 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001172 getNode(result).addChild(ifTrue);
1173 ASTNode::ID ifFalse;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001174 if (this->checkNext(Token::Kind::TK_ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001175 ifFalse = this->statement();
1176 if (!ifFalse) {
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 getNode(result).addChild(ifFalse);
ethannicholasb3058bd2016-07-01 08:22:01 -07001180 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001181 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001182}
1183
1184/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001185ASTNode::ID Parser::doStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001186 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001187 if (!this->expect(Token::Kind::TK_DO, "'do'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001188 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001189 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001190 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kDo);
1191 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001192 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001193 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001194 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001195 getNode(result).addChild(statement);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001196 if (!this->expect(Token::Kind::TK_WHILE, "'while'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001197 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001198 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001199 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001200 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001201 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001202 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001203 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001204 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001205 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001206 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001207 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001208 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001209 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001210 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001211 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001212 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001213 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001214}
1215
1216/* WHILE LPAREN expression RPAREN STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001217ASTNode::ID Parser::whileStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001218 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001219 if (!this->expect(Token::Kind::TK_WHILE, "'while'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001220 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001221 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001222 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
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 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kWhile);
1226 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001227 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001228 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001229 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001230 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001231 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001232 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001233 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001234 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001235 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001236 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001237 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001238 getNode(result).addChild(statement);
1239 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001240}
1241
Ethan Nicholasaf197692017-02-27 13:26:45 -05001242/* CASE expression COLON statement* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001243ASTNode::ID Parser::switchCase() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001244 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001245 if (!this->expect(Token::Kind::TK_CASE, "'case'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001246 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001247 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001248 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitchCase);
1249 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001250 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001251 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001252 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001253 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001254 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001255 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001256 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001257 while (this->peek().fKind != Token::Kind::TK_RBRACE &&
1258 this->peek().fKind != Token::Kind::TK_CASE &&
1259 this->peek().fKind != Token::Kind::TK_DEFAULT) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001260 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001261 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001262 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001263 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001264 getNode(result).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001265 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001266 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001267}
1268
1269/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001270ASTNode::ID Parser::switchStatement() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001271 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001272 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_SWITCH, &start);
1273 if (!isStatic && !this->expect(Token::Kind::TK_SWITCH, "'switch'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001274 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001275 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001276 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001277 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001278 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001279 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001280 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001281 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001282 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001283 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001284 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001285 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001286 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001287 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001288 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001289 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitch, isStatic);
1290 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001291 while (this->peek().fKind == Token::Kind::TK_CASE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001292 ASTNode::ID c = this->switchCase();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001293 if (!c) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001294 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001295 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001296 getNode(result).addChild(c);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001297 }
1298 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1299 // parts of the compiler may rely upon this assumption.
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001300 if (this->peek().fKind == Token::Kind::TK_DEFAULT) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001301 Token defaultStart;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001302 SkAssertResult(this->expect(Token::Kind::TK_DEFAULT, "'default'", &defaultStart));
1303 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001304 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001305 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001306 CREATE_CHILD(defaultCase, result, defaultStart.fOffset, ASTNode::Kind::kSwitchCase);
1307 CREATE_EMPTY_CHILD(defaultCase); // empty test to signify default case
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001308 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001309 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001310 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001311 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001312 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001313 getNode(defaultCase).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001314 }
Ethan Nicholasaf197692017-02-27 13:26:45 -05001315 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001316 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001317 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001318 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001319 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001320}
1321
Ethan Nicholas11d53972016-11-28 11:23:23 -05001322/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001323 STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001324ASTNode::ID Parser::forStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001325 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001326 if (!this->expect(Token::Kind::TK_FOR, "'for'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001327 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001328 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001329 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001330 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001331 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001332 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kFor);
1333 ASTNode::ID initializer;
ethannicholasb3058bd2016-07-01 08:22:01 -07001334 Token nextToken = this->peek();
1335 switch (nextToken.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001336 case Token::Kind::TK_SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001337 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001338 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001339 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001340 case Token::Kind::TK_CONST: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001341 initializer = this->varDeclarations();
1342 if (!initializer) {
1343 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001344 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001345 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001346 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001347 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001348 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001349 if (this->isType(this->text(nextToken))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001350 initializer = this->varDeclarations();
1351 if (!initializer) {
1352 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001353 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001354 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001355 break;
1356 }
ethannicholasa54401d2016-10-14 08:37:32 -07001357 } // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001358 default:
1359 initializer = this->expressionStatement();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001360 if (!initializer) {
1361 return ASTNode::ID::Invalid();
1362 }
1363 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001364 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001365 ASTNode::ID test;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001366 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001367 test = this->expression();
1368 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001369 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001370 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001371 getNode(result).addChild(test);
1372 } else {
1373 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001374 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001375 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001376 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001377 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001378 ASTNode::ID next;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001379 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001380 next = this->expression();
1381 if (!next) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001382 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001383 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001384 getNode(result).addChild(next);
1385 } else {
1386 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001387 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001388 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
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 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001392 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001393 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001394 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001395 getNode(result).addChild(statement);
1396 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001397}
1398
1399/* RETURN expression? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001400ASTNode::ID Parser::returnStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001401 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001402 if (!this->expect(Token::Kind::TK_RETURN, "'return'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001403 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001404 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001405 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kReturn);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001406 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001407 ASTNode::ID expression = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001408 if (!expression) {
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 getNode(result).addChild(expression);
ethannicholasb3058bd2016-07-01 08:22:01 -07001412 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001413 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001414 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001415 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001416 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001417}
1418
1419/* BREAK SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001420ASTNode::ID Parser::breakStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001421 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001422 if (!this->expect(Token::Kind::TK_BREAK, "'break'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001423 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001424 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001425 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001426 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001427 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001428 RETURN_NODE(start.fOffset, ASTNode::Kind::kBreak);
ethannicholasb3058bd2016-07-01 08:22:01 -07001429}
1430
1431/* CONTINUE SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001432ASTNode::ID Parser::continueStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001433 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001434 if (!this->expect(Token::Kind::TK_CONTINUE, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001435 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001436 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001437 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001438 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001439 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001440 RETURN_NODE(start.fOffset, ASTNode::Kind::kContinue);
ethannicholasb3058bd2016-07-01 08:22:01 -07001441}
1442
1443/* DISCARD SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001444ASTNode::ID Parser::discardStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001445 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001446 if (!this->expect(Token::Kind::TK_DISCARD, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001447 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001448 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001449 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001450 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001451 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001452 RETURN_NODE(start.fOffset, ASTNode::Kind::kDiscard);
ethannicholasb3058bd2016-07-01 08:22:01 -07001453}
1454
1455/* LBRACE statement* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001456ASTNode::ID Parser::block() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001457 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001458 if (!this->expect(Token::Kind::TK_LBRACE, "'{'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001459 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001460 }
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001461 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001462 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001463 return ASTNode::ID::Invalid();
1464 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001465 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -07001466 for (;;) {
1467 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001468 case Token::Kind::TK_RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001469 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001470 return result;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001471 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001472 this->error(this->peek(), "expected '}', but found end of file");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001473 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001474 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001475 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001476 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001477 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001478 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001479 getNode(result).addChild(statement);
ethannicholasb3058bd2016-07-01 08:22:01 -07001480 }
1481 }
1482 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001483 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001484}
1485
1486/* expression SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001487ASTNode::ID Parser::expressionStatement() {
1488 ASTNode::ID expr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001489 if (expr) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001490 if (this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001491 return expr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001492 }
1493 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001494 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001495}
1496
1497/* assignmentExpression (COMMA assignmentExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001498ASTNode::ID Parser::expression() {
1499 ASTNode::ID result = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001500 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001501 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001502 }
1503 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001504 while (this->checkNext(Token::Kind::TK_COMMA, &t)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001505 ASTNode::ID right = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001506 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001507 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001508 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001509 CREATE_NODE(newResult, t.fOffset, ASTNode::Kind::kBinary, std::move(t));
1510 getNode(newResult).addChild(result);
1511 getNode(newResult).addChild(right);
1512 result = newResult;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001513 }
1514 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001515}
1516
1517/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1518 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1519 assignmentExpression)*
1520 */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001521ASTNode::ID Parser::assignmentExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001522 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001523 ASTNode::ID result = this->ternaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001524 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001525 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001526 }
1527 for (;;) {
1528 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001529 case Token::Kind::TK_EQ: // fall through
1530 case Token::Kind::TK_STAREQ: // fall through
1531 case Token::Kind::TK_SLASHEQ: // fall through
1532 case Token::Kind::TK_PERCENTEQ: // fall through
1533 case Token::Kind::TK_PLUSEQ: // fall through
1534 case Token::Kind::TK_MINUSEQ: // fall through
1535 case Token::Kind::TK_SHLEQ: // fall through
1536 case Token::Kind::TK_SHREQ: // fall through
1537 case Token::Kind::TK_BITWISEANDEQ: // fall through
1538 case Token::Kind::TK_BITWISEXOREQ: // fall through
1539 case Token::Kind::TK_BITWISEOREQ: // fall through
1540 case Token::Kind::TK_LOGICALANDEQ: // fall through
1541 case Token::Kind::TK_LOGICALXOREQ: // fall through
1542 case Token::Kind::TK_LOGICALOREQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001543 if (!depth.increase()) {
1544 return ASTNode::ID::Invalid();
1545 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001546 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001547 ASTNode::ID right = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001548 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001549 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001550 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001551 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1552 std::move(t));
1553 getNode(newResult).addChild(result);
1554 getNode(newResult).addChild(right);
1555 result = newResult;
1556 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001557 }
1558 default:
1559 return result;
1560 }
1561 }
1562}
1563
1564/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001565ASTNode::ID Parser::ternaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001566 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001567 ASTNode::ID base = this->logicalOrExpression();
1568 if (!base) {
1569 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001570 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001571 if (this->checkNext(Token::Kind::TK_QUESTION)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001572 if (!depth.increase()) {
1573 return ASTNode::ID::Invalid();
1574 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001575 ASTNode::ID trueExpr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001576 if (!trueExpr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001577 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001578 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001579 if (this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001580 ASTNode::ID falseExpr = this->assignmentExpression();
1581 if (!falseExpr) {
1582 return ASTNode::ID::Invalid();
1583 }
1584 CREATE_NODE(ternary, getNode(base).fOffset, ASTNode::Kind::kTernary);
1585 getNode(ternary).addChild(base);
1586 getNode(ternary).addChild(trueExpr);
1587 getNode(ternary).addChild(falseExpr);
1588 return ternary;
ethannicholasb3058bd2016-07-01 08:22:01 -07001589 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001590 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001591 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001592 return base;
ethannicholasb3058bd2016-07-01 08:22:01 -07001593}
1594
1595/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001596ASTNode::ID Parser::logicalOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001597 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001598 ASTNode::ID result = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001599 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001600 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001601 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001602 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001603 while (this->checkNext(Token::Kind::TK_LOGICALOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001604 if (!depth.increase()) {
1605 return ASTNode::ID::Invalid();
1606 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001607 ASTNode::ID right = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001608 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001609 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001610 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001611 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1612 getNode(newResult).addChild(result);
1613 getNode(newResult).addChild(right);
1614 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001615 }
1616 return result;
1617}
1618
1619/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001620ASTNode::ID Parser::logicalXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001621 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001622 ASTNode::ID result = this->logicalAndExpression();
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_LOGICALXOR, &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->logicalAndExpression();
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/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001644ASTNode::ID Parser::logicalAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001645 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001646 ASTNode::ID result = this->bitwiseOrExpression();
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_LOGICALAND, &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->bitwiseOrExpression();
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/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001668ASTNode::ID Parser::bitwiseOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001669 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001670 ASTNode::ID result = this->bitwiseXorExpression();
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_BITWISEOR, &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->bitwiseXorExpression();
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/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001692ASTNode::ID Parser::bitwiseXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001693 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001694 ASTNode::ID result = this->bitwiseAndExpression();
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_BITWISEXOR, &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->bitwiseAndExpression();
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/* equalityExpression (BITWISEAND equalityExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001716ASTNode::ID Parser::bitwiseAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001717 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001718 ASTNode::ID result = this->equalityExpression();
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_BITWISEAND, &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->equalityExpression();
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/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001740ASTNode::ID Parser::equalityExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001741 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001742 ASTNode::ID result = this->relationalExpression();
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 }
1746 for (;;) {
1747 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001748 case Token::Kind::TK_EQEQ: // fall through
1749 case Token::Kind::TK_NEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001750 if (!depth.increase()) {
1751 return ASTNode::ID::Invalid();
1752 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001753 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001754 ASTNode::ID right = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001755 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001756 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001757 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001758 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1759 std::move(t));
1760 getNode(newResult).addChild(result);
1761 getNode(newResult).addChild(right);
1762 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001763 break;
1764 }
1765 default:
1766 return result;
1767 }
1768 }
1769}
1770
1771/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001772ASTNode::ID Parser::relationalExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001773 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001774 ASTNode::ID result = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001775 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001776 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001777 }
1778 for (;;) {
1779 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001780 case Token::Kind::TK_LT: // fall through
1781 case Token::Kind::TK_GT: // fall through
1782 case Token::Kind::TK_LTEQ: // fall through
1783 case Token::Kind::TK_GTEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001784 if (!depth.increase()) {
1785 return ASTNode::ID::Invalid();
1786 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001787 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001788 ASTNode::ID right = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001789 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001790 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001791 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001792 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1793 std::move(t));
1794 getNode(newResult).addChild(result);
1795 getNode(newResult).addChild(right);
1796 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001797 break;
1798 }
1799 default:
1800 return result;
1801 }
1802 }
1803}
1804
1805/* additiveExpression ((SHL | SHR) additiveExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001806ASTNode::ID Parser::shiftExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001807 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001808 ASTNode::ID result = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001809 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001810 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001811 }
1812 for (;;) {
1813 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001814 case Token::Kind::TK_SHL: // fall through
1815 case Token::Kind::TK_SHR: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001816 if (!depth.increase()) {
1817 return ASTNode::ID::Invalid();
1818 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001819 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001820 ASTNode::ID right = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001821 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001822 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001823 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001824 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1825 std::move(t));
1826 getNode(newResult).addChild(result);
1827 getNode(newResult).addChild(right);
1828 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001829 break;
1830 }
1831 default:
1832 return result;
1833 }
1834 }
1835}
1836
1837/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001838ASTNode::ID Parser::additiveExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001839 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001840 ASTNode::ID result = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001841 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001842 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001843 }
1844 for (;;) {
1845 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001846 case Token::Kind::TK_PLUS: // fall through
1847 case Token::Kind::TK_MINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001848 if (!depth.increase()) {
1849 return ASTNode::ID::Invalid();
1850 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001851 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001852 ASTNode::ID right = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001853 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001854 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001855 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001856 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1857 std::move(t));
1858 getNode(newResult).addChild(result);
1859 getNode(newResult).addChild(right);
1860 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001861 break;
1862 }
1863 default:
1864 return result;
1865 }
1866 }
1867}
1868
1869/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001870ASTNode::ID Parser::multiplicativeExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001871 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001872 ASTNode::ID result = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001873 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001874 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001875 }
1876 for (;;) {
1877 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001878 case Token::Kind::TK_STAR: // fall through
1879 case Token::Kind::TK_SLASH: // fall through
1880 case Token::Kind::TK_PERCENT: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001881 if (!depth.increase()) {
1882 return ASTNode::ID::Invalid();
1883 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001884 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001885 ASTNode::ID right = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001886 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001887 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001888 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001889 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1890 std::move(t));
1891 getNode(newResult).addChild(result);
1892 getNode(newResult).addChild(right);
1893 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001894 break;
1895 }
1896 default:
1897 return result;
1898 }
1899 }
1900}
1901
1902/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001903ASTNode::ID Parser::unaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001904 AutoDepth depth(this);
ethannicholasb3058bd2016-07-01 08:22:01 -07001905 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001906 case Token::Kind::TK_PLUS: // fall through
1907 case Token::Kind::TK_MINUS: // fall through
1908 case Token::Kind::TK_LOGICALNOT: // fall through
1909 case Token::Kind::TK_BITWISENOT: // fall through
1910 case Token::Kind::TK_PLUSPLUS: // fall through
1911 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001912 if (!depth.increase()) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001913 return ASTNode::ID::Invalid();
Ethan Nicholas6dcc3252019-02-20 15:18:36 -05001914 }
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001915 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001916 ASTNode::ID expr = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001917 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001918 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001919 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001920 CREATE_NODE(result, t.fOffset, ASTNode::Kind::kPrefix, std::move(t));
1921 getNode(result).addChild(expr);
1922 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001923 }
1924 default:
1925 return this->postfixExpression();
1926 }
1927}
1928
1929/* term suffix* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001930ASTNode::ID Parser::postfixExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001931 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001932 ASTNode::ID result = this->term();
ethannicholasb3058bd2016-07-01 08:22:01 -07001933 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001934 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001935 }
1936 for (;;) {
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001937 Token t = this->peek();
1938 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001939 case Token::Kind::TK_FLOAT_LITERAL:
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001940 if (this->text(t)[0] != '.') {
1941 return result;
1942 }
1943 // fall through
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001944 case Token::Kind::TK_LBRACKET:
1945 case Token::Kind::TK_DOT:
1946 case Token::Kind::TK_LPAREN:
1947 case Token::Kind::TK_PLUSPLUS:
1948 case Token::Kind::TK_MINUSMINUS:
1949 case Token::Kind::TK_COLONCOLON:
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001950 if (!depth.increase()) {
1951 return ASTNode::ID::Invalid();
1952 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001953 result = this->suffix(result);
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001954 if (!result) {
1955 return ASTNode::ID::Invalid();
1956 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001957 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001958 default:
1959 return result;
1960 }
1961 }
1962}
1963
Ethan Nicholas11d53972016-11-28 11:23:23 -05001964/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
Ethan Nicholase455f652019-09-13 12:52:55 -04001965 PLUSPLUS | MINUSMINUS | COLONCOLON IDENTIFIER | FLOAT_LITERAL [IDENTIFIER] */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001966ASTNode::ID Parser::suffix(ASTNode::ID base) {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001967 SkASSERT(base);
ethannicholasb3058bd2016-07-01 08:22:01 -07001968 Token next = this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001969 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001970 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001971 return ASTNode::ID::Invalid();
1972 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001973 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001974 case Token::Kind::TK_LBRACKET: {
1975 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001976 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
1977 getNode(result).addChild(base);
1978 return result;
ethannicholas5961bc92016-10-12 06:39:56 -07001979 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001980 ASTNode::ID e = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001981 if (!e) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001982 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001983 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001984 this->expect(Token::Kind::TK_RBRACKET, "']' to complete array access expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001985 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
1986 getNode(result).addChild(base);
1987 getNode(result).addChild(e);
1988 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001989 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001990 case Token::Kind::TK_DOT: // fall through
1991 case Token::Kind::TK_COLONCOLON: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001992 int offset = this->peek().fOffset;
1993 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07001994 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001995 CREATE_NODE(result, offset, ASTNode::Kind::kField, std::move(text));
1996 getNode(result).addChild(base);
1997 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001998 }
Ethan Nicholase455f652019-09-13 12:52:55 -04001999 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002000 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholase455f652019-09-13 12:52:55 -04002001 // Swizzles that start with a constant number, e.g. '.000r', will be tokenized as
2002 // floating point literals, possibly followed by an identifier. Handle that here.
2003 StringFragment field = this->text(next);
2004 SkASSERT(field.fChars[0] == '.');
2005 ++field.fChars;
2006 --field.fLength;
2007 for (size_t i = 0; i < field.fLength; ++i) {
2008 if (field.fChars[i] != '0' && field.fChars[i] != '1') {
2009 this->error(next, "invalid swizzle");
2010 return ASTNode::ID::Invalid();
2011 }
2012 }
2013 // use the next *raw* token so we don't ignore whitespace - we only care about
2014 // identifiers that directly follow the float
2015 Token id = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002016 if (id.fKind == Token::Kind::TK_IDENTIFIER) {
Ethan Nicholase455f652019-09-13 12:52:55 -04002017 field.fLength += id.fLength;
2018 } else {
2019 this->pushback(id);
2020 }
2021 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kField, field);
2022 getNode(result).addChild(base);
2023 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002024 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002025 case Token::Kind::TK_LPAREN: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002026 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kCall);
2027 getNode(result).addChild(base);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002028 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002029 for (;;) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002030 ASTNode::ID expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002031 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002032 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002033 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002034 getNode(result).addChild(expr);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002035 if (!this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002036 break;
2037 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002038 }
2039 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002040 this->expect(Token::Kind::TK_RPAREN, "')' to complete function parameters");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002041 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002042 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002043 case Token::Kind::TK_PLUSPLUS: // fall through
2044 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002045 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kPostfix, next);
2046 getNode(result).addChild(base);
2047 return result;
2048 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002049 default: {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002050 this->error(next, "expected expression suffix, but found '" + this->text(next) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002051 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002052 }
2053 }
2054}
2055
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002056/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | NULL_LITERAL | '(' expression ')' */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002057ASTNode::ID Parser::term() {
ethannicholasb3058bd2016-07-01 08:22:01 -07002058 Token t = this->peek();
2059 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002060 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002061 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002062 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002063 RETURN_NODE(t.fOffset, ASTNode::Kind::kIdentifier, std::move(text));
ethannicholasb3058bd2016-07-01 08:22:01 -07002064 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002065 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002066 case Token::Kind::TK_INT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002067 SKSL_INT i;
ethannicholasb3058bd2016-07-01 08:22:01 -07002068 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002069 RETURN_NODE(t.fOffset, ASTNode::Kind::kInt, i);
ethannicholasb3058bd2016-07-01 08:22:01 -07002070 }
2071 break;
2072 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002073 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002074 SKSL_FLOAT f;
ethannicholasb3058bd2016-07-01 08:22:01 -07002075 if (this->floatLiteral(&f)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002076 RETURN_NODE(t.fOffset, ASTNode::Kind::kFloat, f);
ethannicholasb3058bd2016-07-01 08:22:01 -07002077 }
2078 break;
2079 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002080 case Token::Kind::TK_TRUE_LITERAL: // fall through
2081 case Token::Kind::TK_FALSE_LITERAL: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002082 bool b;
2083 if (this->boolLiteral(&b)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002084 RETURN_NODE(t.fOffset, ASTNode::Kind::kBool, b);
ethannicholasb3058bd2016-07-01 08:22:01 -07002085 }
2086 break;
2087 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002088 case Token::Kind::TK_NULL_LITERAL:
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002089 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04002090 RETURN_NODE(t.fOffset, ASTNode::Kind::kNull);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002091 case Token::Kind::TK_LPAREN: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002092 this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002093 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002094 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002095 return ASTNode::ID::Invalid();
2096 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002097 ASTNode::ID result = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002098 if (result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002099 this->expect(Token::Kind::TK_RPAREN, "')' to complete expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002100 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002101 }
2102 break;
2103 }
2104 default:
2105 this->nextToken();
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002106 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002107 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002108 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002109}
2110
2111/* INT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002112bool Parser::intLiteral(SKSL_INT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002113 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002114 if (this->expect(Token::Kind::TK_INT_LITERAL, "integer literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002115 *dest = SkSL::stol(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002116 return true;
2117 }
2118 return false;
2119}
2120
2121/* FLOAT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002122bool Parser::floatLiteral(SKSL_FLOAT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002123 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002124 if (this->expect(Token::Kind::TK_FLOAT_LITERAL, "float literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002125 *dest = SkSL::stod(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002126 return true;
2127 }
2128 return false;
2129}
2130
2131/* TRUE_LITERAL | FALSE_LITERAL */
2132bool Parser::boolLiteral(bool* dest) {
2133 Token t = this->nextToken();
2134 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002135 case Token::Kind::TK_TRUE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002136 *dest = true;
2137 return true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002138 case Token::Kind::TK_FALSE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002139 *dest = false;
2140 return true;
2141 default:
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002142 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002143 return false;
2144 }
2145}
2146
2147/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002148bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002149 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002150 if (this->expect(Token::Kind::TK_IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002151 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07002152 return true;
2153 }
2154 return false;
2155}
2156
2157} // namespace