blob: f148cf621240e34423107e3ae59ec1d55591c87c [file] [log] [blame]
ethannicholasb3058bd2016-07-01 08:22:01 -07001/*
2 * Copyright 2016 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
Ethan Nicholas11d53972016-11-28 11:23:23 -05007
Mike Klein4b432fa2019-06-06 11:44:05 -05008#include "src/sksl/SkSLParser.h"
John Stilesfbd050b2020-08-03 13:21:46 -04009
10#include <memory>
11#include "stdio.h"
12
Ethan Nicholasdaed2592021-03-04 14:30:25 -050013#include "include/private/SkSLModifiers.h"
John Stilesfbd050b2020-08-03 13:21:46 -040014#include "src/sksl/SkSLASTNode.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050015#include "src/sksl/ir/SkSLSymbolTable.h"
16#include "src/sksl/ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070017
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040018#ifndef SKSL_STANDALONE
Mike Kleinc0bd9f92019-04-23 12:05:21 -050019#include "include/private/SkOnce.h"
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040020#endif
21
ethannicholasb3058bd2016-07-01 08:22:01 -070022namespace SkSL {
23
John Stiles8d056592020-11-10 10:03:50 -050024static constexpr int kMaxParseDepth = 50;
John Stilesd39aec92020-12-03 14:37:16 -050025static constexpr int kMaxStructDepth = 8;
John Stilesa695d622020-11-10 10:04:26 -050026
27static bool struct_is_too_deeply_nested(const Type& type, int limit) {
28 if (limit < 0) {
29 return true;
30 }
31
John Stilesc0c51062020-12-03 17:16:29 -050032 if (type.isStruct()) {
John Stilesa695d622020-11-10 10:04:26 -050033 for (const Type::Field& f : type.fields()) {
34 if (struct_is_too_deeply_nested(*f.fType, limit - 1)) {
35 return true;
36 }
37 }
38 }
39
40 return false;
41}
ethannicholascad64162016-10-27 10:54:02 -070042
John Stiles7d5b90a2021-01-21 14:26:51 -050043static int parse_modifier_token(Token::Kind token) {
44 switch (token) {
45 case Token::Kind::TK_UNIFORM: return Modifiers::kUniform_Flag;
46 case Token::Kind::TK_CONST: return Modifiers::kConst_Flag;
47 case Token::Kind::TK_IN: return Modifiers::kIn_Flag;
48 case Token::Kind::TK_OUT: return Modifiers::kOut_Flag;
49 case Token::Kind::TK_INOUT: return Modifiers::kIn_Flag | Modifiers::kOut_Flag;
50 case Token::Kind::TK_FLAT: return Modifiers::kFlat_Flag;
51 case Token::Kind::TK_NOPERSPECTIVE: return Modifiers::kNoPerspective_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050052 case Token::Kind::TK_HASSIDEEFFECTS: return Modifiers::kHasSideEffects_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050053 case Token::Kind::TK_VARYING: return Modifiers::kVarying_Flag;
54 case Token::Kind::TK_INLINE: return Modifiers::kInline_Flag;
John Stiles0dd1a772021-03-09 22:14:27 -050055 case Token::Kind::TK_NOINLINE: return Modifiers::kNoInline_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050056 default: return 0;
57 }
58}
59
ethannicholascad64162016-10-27 10:54:02 -070060class AutoDepth {
61public:
62 AutoDepth(Parser* p)
Ethan Nicholascf4deab2019-09-13 16:28:14 -040063 : fParser(p)
64 , fDepth(0) {}
ethannicholascad64162016-10-27 10:54:02 -070065
66 ~AutoDepth() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -040067 fParser->fDepth -= fDepth;
ethannicholascad64162016-10-27 10:54:02 -070068 }
69
Ethan Nicholascf4deab2019-09-13 16:28:14 -040070 bool increase() {
71 ++fDepth;
72 ++fParser->fDepth;
John Stiles8d056592020-11-10 10:03:50 -050073 if (fParser->fDepth > kMaxParseDepth) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070074 fParser->error(fParser->peek(), String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070075 return false;
76 }
77 return true;
78 }
79
80private:
81 Parser* fParser;
Ethan Nicholascf4deab2019-09-13 16:28:14 -040082 int fDepth;
ethannicholascad64162016-10-27 10:54:02 -070083};
84
Brian Salomon140f3da2018-08-23 13:51:27 +000085std::unordered_map<String, Parser::LayoutToken>* Parser::layoutTokens;
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040086
87void Parser::InitLayoutMap() {
Brian Salomon140f3da2018-08-23 13:51:27 +000088 layoutTokens = new std::unordered_map<String, LayoutToken>;
Brian Salomon23356442018-11-30 15:33:19 -050089 #define TOKEN(name, text) (*layoutTokens)[text] = LayoutToken::name
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040090 TOKEN(LOCATION, "location");
91 TOKEN(OFFSET, "offset");
92 TOKEN(BINDING, "binding");
93 TOKEN(INDEX, "index");
94 TOKEN(SET, "set");
95 TOKEN(BUILTIN, "builtin");
96 TOKEN(INPUT_ATTACHMENT_INDEX, "input_attachment_index");
97 TOKEN(ORIGIN_UPPER_LEFT, "origin_upper_left");
98 TOKEN(OVERRIDE_COVERAGE, "override_coverage");
Brian Osmandc4f8cd2021-02-25 15:30:46 -050099 TOKEN(EARLY_FRAGMENT_TESTS, "early_fragment_tests");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400100 TOKEN(BLEND_SUPPORT_ALL_EQUATIONS, "blend_support_all_equations");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400101 TOKEN(PUSH_CONSTANT, "push_constant");
102 TOKEN(POINTS, "points");
103 TOKEN(LINES, "lines");
104 TOKEN(LINE_STRIP, "line_strip");
105 TOKEN(LINES_ADJACENCY, "lines_adjacency");
106 TOKEN(TRIANGLES, "triangles");
107 TOKEN(TRIANGLE_STRIP, "triangle_strip");
108 TOKEN(TRIANGLES_ADJACENCY, "triangles_adjacency");
109 TOKEN(MAX_VERTICES, "max_vertices");
110 TOKEN(INVOCATIONS, "invocations");
Brian Osmanf59a9612020-04-15 14:18:13 -0400111 TOKEN(MARKER, "marker");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400112 TOKEN(WHEN, "when");
113 TOKEN(KEY, "key");
Michael Ludwiga4275592018-08-31 10:52:47 -0400114 TOKEN(TRACKED, "tracked");
Brian Osmanb32d66b2020-04-30 17:12:03 -0400115 TOKEN(SRGB_UNPREMUL, "srgb_unpremul");
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400116 TOKEN(CTYPE, "ctype");
Brian Osmanf28e55d2018-10-03 16:35:54 -0400117 TOKEN(SKPMCOLOR4F, "SkPMColor4f");
Mike Reedb26b4e72020-01-22 14:31:21 -0500118 TOKEN(SKV4, "SkV4");
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400119 TOKEN(SKRECT, "SkRect");
120 TOKEN(SKIRECT, "SkIRect");
121 TOKEN(SKPMCOLOR, "SkPMColor");
Mike Reedb26b4e72020-01-22 14:31:21 -0500122 TOKEN(SKM44, "SkM44");
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400123 TOKEN(BOOL, "bool");
124 TOKEN(INT, "int");
125 TOKEN(FLOAT, "float");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400126 #undef TOKEN
127}
128
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400129Parser::Parser(const char* text, size_t length, SymbolTable& symbols, ErrorReporter& errors)
John Stiles18a45b32021-03-29 18:07:32 -0400130: fText(text, length)
John Stiles7cbb09c22021-01-07 16:07:00 -0500131, fPushback(Token::Kind::TK_NONE, -1, -1)
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400132, fSymbols(symbols)
ethannicholasb3058bd2016-07-01 08:22:01 -0700133, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700134 fLexer.start(text, length);
Brian Salomon3b83afe2018-08-23 11:04:36 -0400135 static const bool layoutMapInitialized = []{ return (void)InitLayoutMap(), true; }();
136 (void) layoutMapInitialized;
ethannicholasb3058bd2016-07-01 08:22:01 -0700137}
138
John Stiles3b209362020-11-16 17:03:10 -0500139template <typename... Args>
140ASTNode::ID Parser::createNode(Args&&... args) {
141 ASTNode::ID result(fFile->fNodes.size());
142 fFile->fNodes.emplace_back(&fFile->fNodes, std::forward<Args>(args)...);
143 return result;
144}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400145
John Stiles3b209362020-11-16 17:03:10 -0500146ASTNode::ID Parser::addChild(ASTNode::ID target, ASTNode::ID child) {
147 fFile->fNodes[target.fValue].addChild(child);
148 return child;
149}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400150
John Stiles3b209362020-11-16 17:03:10 -0500151void Parser::createEmptyChild(ASTNode::ID target) {
152 ASTNode::ID child(fFile->fNodes.size());
153 fFile->fNodes.emplace_back();
154 fFile->fNodes[target.fValue].addChild(child);
155}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400156
Ethan Nicholasf7b88202017-09-18 14:10:39 -0400157/* (directive | section | declaration)* END_OF_FILE */
Ethan Nicholasba9a04f2020-11-06 09:28:04 -0500158std::unique_ptr<ASTFile> Parser::compilationUnit() {
John Stilesfbd050b2020-08-03 13:21:46 -0400159 fFile = std::make_unique<ASTFile>();
John Stiles18a45b32021-03-29 18:07:32 -0400160 fFile->fNodes.reserve(fText.size() / 10); // a typical program is approx 10:1 for chars:nodes
John Stiles3b209362020-11-16 17:03:10 -0500161 ASTNode::ID result = this->createNode(/*offset=*/0, ASTNode::Kind::kFile);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400162 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700163 for (;;) {
164 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400165 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400166 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400167 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400168 ASTNode::ID dir = this->directive();
169 if (fErrors.errorCount()) {
170 return nullptr;
171 }
172 if (dir) {
173 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700174 }
175 break;
176 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400177 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400178 ASTNode::ID section = this->section();
179 if (fErrors.errorCount()) {
180 return nullptr;
181 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400182 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400183 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400184 }
185 break;
186 }
John Stiles7cbb09c22021-01-07 16:07:00 -0500187 case Token::Kind::TK_INVALID: {
188 this->error(this->peek(), String("invalid token"));
189 return nullptr;
190 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700191 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400192 ASTNode::ID decl = this->declaration();
193 if (fErrors.errorCount()) {
194 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700195 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400196 if (decl) {
197 getNode(result).addChild(decl);
198 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700199 }
200 }
201 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400202 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700203}
204
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700205Token Parser::nextRawToken() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500206 if (fPushback.fKind != Token::Kind::TK_NONE) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700207 Token result = fPushback;
John Stiles7cbb09c22021-01-07 16:07:00 -0500208 fPushback.fKind = Token::Kind::TK_NONE;
ethannicholasb3058bd2016-07-01 08:22:01 -0700209 return result;
210 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700211 Token result = fLexer.next();
212 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400213}
214
215Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700216 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400217 while (token.fKind == Token::Kind::TK_WHITESPACE ||
218 token.fKind == Token::Kind::TK_LINE_COMMENT ||
219 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700220 token = this->nextRawToken();
221 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400222 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700223}
224
225void Parser::pushback(Token t) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500226 SkASSERT(fPushback.fKind == Token::Kind::TK_NONE);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400227 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700228}
229
230Token Parser::peek() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500231 if (fPushback.fKind == Token::Kind::TK_NONE) {
Brian Osman634624a2017-08-15 11:14:30 -0400232 fPushback = this->nextToken();
233 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700234 return fPushback;
235}
236
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400237bool Parser::checkNext(Token::Kind kind, Token* result) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500238 if (fPushback.fKind != Token::Kind::TK_NONE && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400239 return false;
240 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400241 Token next = this->nextToken();
242 if (next.fKind == kind) {
243 if (result) {
244 *result = next;
245 }
246 return true;
247 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400248 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400249 return false;
250}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500251
252bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700253 Token next = this->nextToken();
254 if (next.fKind == kind) {
255 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400256 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700257 }
258 return true;
259 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700260 this->error(next, "expected " + String(expected) + ", but found '" +
261 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700262 return false;
263 }
264}
265
John Stiles2630ea32020-12-04 10:51:21 -0500266bool Parser::expectIdentifier(Token* result) {
267 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", result)) {
268 return false;
269 }
270 if (this->isType(this->text(*result))) {
271 this->error(*result, "expected an identifier, but found type '" +
272 this->text(*result) + "'");
273 return false;
274 }
275 return true;
276}
277
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700278StringFragment Parser::text(Token token) {
John Stiles18a45b32021-03-29 18:07:32 -0400279 return StringFragment(fText.begin() + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500280}
281
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700282void Parser::error(Token token, String msg) {
283 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700284}
285
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700286void Parser::error(int offset, String msg) {
287 fErrors.error(offset, msg);
288}
289
290bool Parser::isType(StringFragment name) {
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400291 const Symbol* s = fSymbols[name];
John Stiles2630ea32020-12-04 10:51:21 -0500292 return s && s->is<Type>();
ethannicholasb3058bd2016-07-01 08:22:01 -0700293}
294
John Stiles80b02af2021-02-12 17:07:51 -0500295bool Parser::isArrayType(ASTNode::ID type) {
296 const ASTNode& node = this->getNode(type);
297 SkASSERT(node.fKind == ASTNode::Kind::kType);
298 return node.begin() != node.end();
299}
300
Ethan Nicholas11d53972016-11-28 11:23:23 -0500301/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700302 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400303ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700304 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400305 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400306 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700307 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700308 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400309 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700310 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500311 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400312 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700313 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400314 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400315 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700316 }
317 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400318 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400319 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700320 }
John Stiles3b209362020-11-16 17:03:10 -0500321 return this->createNode(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700322 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700323 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400324 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700325 }
326}
327
Ethan Nicholas762466e2017-06-29 10:03:38 -0400328/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
329 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400330ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400331 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400332 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400333 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400334 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400335 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400336 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400337 this->nextToken();
338 Token argToken;
John Stiles2630ea32020-12-04 10:51:21 -0500339 if (!this->expectIdentifier(&argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400340 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400341 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700342 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400343 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400344 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400345 }
346 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400347 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400348 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400349 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400350 StringFragment text;
351 Token codeStart = this->nextRawToken();
352 size_t startOffset = codeStart.fOffset;
353 this->pushback(codeStart);
John Stiles18a45b32021-03-29 18:07:32 -0400354 text.fChars = fText.begin() + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400355 int level = 1;
356 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700357 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400358 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400359 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400360 ++level;
361 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400362 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400363 --level;
364 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400365 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700366 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400367 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400368 default:
369 break;
370 }
371 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400372 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400373 break;
374 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400375 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700376 StringFragment name = this->text(start);
377 ++name.fChars;
378 --name.fLength;
John Stiles3b209362020-11-16 17:03:10 -0500379 return this->createNode(start.fOffset, ASTNode::Kind::kSection,
380 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400381}
382
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500383/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
384 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400385ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500386 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400387 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400388 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500389 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400390 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400391 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500392 }
393 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500394 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400395 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500396 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400397 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400398 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500399 }
John Stilese6c67c52021-01-15 12:01:00 -0500400 fSymbols.add(Type::MakeEnumType(this->text(name)));
John Stiles3b209362020-11-16 17:03:10 -0500401 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400402 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500403 Token id;
John Stiles2630ea32020-12-04 10:51:21 -0500404 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400405 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500406 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400407 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400408 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500409 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400410 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500411 }
John Stiles3b209362020-11-16 17:03:10 -0500412 ASTNode::ID child = this->addChild(
413 result, this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400414 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500415 } else {
John Stiles3b209362020-11-16 17:03:10 -0500416 this->addChild(result,
417 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500418 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400419 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
420 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400421 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500422 }
John Stiles2630ea32020-12-04 10:51:21 -0500423 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400424 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500425 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400426 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400427 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500428 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400429 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500430 }
John Stiles3b209362020-11-16 17:03:10 -0500431 ASTNode::ID child = this->addChild(
432 result,
433 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400434 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500435 } else {
John Stiles3b209362020-11-16 17:03:10 -0500436 this->addChild(
437 result,
438 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500439 }
440 }
441 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400442 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400443 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500444}
445
446/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700447 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400448ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700449 Token lookahead = this->peek();
Ethan Nicholasda6320c2020-09-02 14:08:23 -0400450 switch (lookahead.fKind) {
451 case Token::Kind::TK_ENUM:
452 return this->enumDeclaration();
453 case Token::Kind::TK_SEMICOLON:
454 this->error(lookahead.fOffset, "expected a declaration, but found ';'");
455 return ASTNode::ID::Invalid();
456 default:
457 break;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500458 }
459 Modifiers modifiers = this->modifiers();
460 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400461 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700462 // we have an identifier that's not a type, could be the start of an interface block
463 return this->interfaceBlock(modifiers);
464 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400465 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700466 return this->structVarDeclaration(modifiers);
467 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400468 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700469 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -0500470 return this->createNode(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700471 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400472 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700473 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400474 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700475 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700476 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500477 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400478 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700479 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400480 if (this->checkNext(Token::Kind::TK_LPAREN)) {
John Stiles3b209362020-11-16 17:03:10 -0500481 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kFunction);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400482 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
483 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400484 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400485 for (;;) {
486 ASTNode::ID parameter = this->parameter();
487 if (!parameter) {
488 return ASTNode::ID::Invalid();
489 }
490 ++fd.fParameterCount;
491 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400492 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400493 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700494 }
495 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700496 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400497 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400498 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400499 return ASTNode::ID::Invalid();
500 }
501 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400502 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700503 body = this->block();
504 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400505 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700506 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400507 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700508 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400509 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700510 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400511 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700512 }
513}
514
John Stiles76389b72021-01-25 13:49:05 -0500515/* (varDeclarations | expressionStatement) */
516ASTNode::ID Parser::varDeclarationsOrExpressionStatement() {
517 if (this->isType(this->text(this->peek()))) {
518 // Statements that begin with a typename are most often variable declarations, but
519 // occasionally the type is part of a constructor, and these are actually expression-
520 // statements in disguise. First, attempt the common case: parse it as a vardecl.
521 Checkpoint checkpoint(this);
John Stiles8dabeac2021-02-12 16:05:00 -0500522 VarDeclarationsPrefix prefix;
523 if (this->varDeclarationsPrefix(&prefix)) {
524 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
John Stiles76389b72021-01-25 13:49:05 -0500525 }
526
527 // If this statement wasn't actually a vardecl after all, rewind and try parsing it as an
528 // expression-statement instead.
529 checkpoint.rewind();
530 }
531
532 return this->expressionStatement();
533}
534
John Stiles8dabeac2021-02-12 16:05:00 -0500535// Helper function for varDeclarations(). If this function succeeds, we assume that the rest of the
536// statement is a variable-declaration statement, not an expression-statement.
537bool Parser::varDeclarationsPrefix(VarDeclarationsPrefix* prefixData) {
538 prefixData->modifiers = this->modifiers();
539 prefixData->type = this->type();
540 if (!prefixData->type) {
541 return false;
542 }
543 return this->expectIdentifier(&prefixData->name);
544}
545
ethannicholasb3058bd2016-07-01 08:22:01 -0700546/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400547ASTNode::ID Parser::varDeclarations() {
John Stiles8dabeac2021-02-12 16:05:00 -0500548 VarDeclarationsPrefix prefix;
549 if (!this->varDeclarationsPrefix(&prefix)) {
John Stiles76389b72021-01-25 13:49:05 -0500550 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700551 }
John Stiles8dabeac2021-02-12 16:05:00 -0500552 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700553}
554
555/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400556ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400557 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400558 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700559 }
560 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500561 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400562 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700563 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400564 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400565 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700566 }
567 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400568 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400569 ASTNode::ID decls = this->varDeclarations();
570 if (!decls) {
571 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700572 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400573 ASTNode& declsNode = getNode(decls);
John Stilesece1bf02021-03-08 11:15:55 -0500574 const Modifiers& modifiers = declsNode.begin()->getModifiers();
John Stiles13fc2602020-10-09 17:42:31 -0400575 if (modifiers.fFlags != Modifiers::kNo_Flag) {
576 String desc = modifiers.description();
577 desc.pop_back(); // remove trailing space
578 this->error(declsNode.fOffset,
579 "modifier '" + desc + "' is not permitted on a struct field");
580 }
581
Brian Osman00fea5b2021-01-28 09:46:30 -0500582 const Symbol* symbol = fSymbols[(declsNode.begin() + 1)->getString()];
John Stiles13fc2602020-10-09 17:42:31 -0400583 SkASSERT(symbol);
584 const Type* type = &symbol->as<Type>();
John Stiles1d757782020-11-17 09:43:22 -0500585 if (type->isOpaque()) {
586 this->error(declsNode.fOffset,
587 "opaque type '" + type->name() + "' is not permitted in a struct");
588 }
John Stiles13fc2602020-10-09 17:42:31 -0400589
Ethan Nicholasfc994162019-06-06 10:04:27 -0400590 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
591 ASTNode& var = *iter;
John Stilesece1bf02021-03-08 11:15:55 -0500592 const ASTNode::VarData& vd = var.getVarData();
John Stiles6bef6a72020-12-02 14:26:04 -0500593
John Stilesd39aec92020-12-03 14:37:16 -0500594 // Read array size if one is present.
595 if (vd.fIsArray) {
596 const ASTNode& size = *var.begin();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400597 if (!size || size.fKind != ASTNode::Kind::kInt) {
598 this->error(declsNode.fOffset, "array size in struct field must be a constant");
599 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700600 }
John Stiles6bef6a72020-12-02 14:26:04 -0500601 if (size.getInt() <= 0 || size.getInt() > INT_MAX) {
602 this->error(declsNode.fOffset, "array size is invalid");
603 return ASTNode::ID::Invalid();
604 }
John Stilesd39aec92020-12-03 14:37:16 -0500605 // Add the array dimensions to our type.
606 int arraySize = size.getInt();
John Stilesa2179502020-12-03 14:37:57 -0500607 type = fSymbols.addArrayDimension(type, arraySize);
ethannicholasb3058bd2016-07-01 08:22:01 -0700608 }
John Stiles13fc2602020-10-09 17:42:31 -0400609
610 fields.push_back(Type::Field(modifiers, vd.fName, type));
John Stilesd39aec92020-12-03 14:37:16 -0500611 if (vd.fIsArray ? var.begin()->fNext : var.fFirstChild) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400612 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700613 }
614 }
615 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400616 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400617 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700618 }
Brian Osman6c3b23f2021-02-12 13:09:58 -0500619 if (fields.empty()) {
620 this->error(name.fOffset,
621 "struct '" + this->text(name) + "' must contain at least one field");
622 return ASTNode::ID::Invalid();
623 }
John Stilesad2d4942020-12-11 16:55:58 -0500624 std::unique_ptr<Type> newType = Type::MakeStructType(name.fOffset, this->text(name), fields);
John Stilesa695d622020-11-10 10:04:26 -0500625 if (struct_is_too_deeply_nested(*newType, kMaxStructDepth)) {
626 this->error(name.fOffset, "struct '" + this->text(name) + "' is too deeply nested");
627 return ASTNode::ID::Invalid();
628 }
629 fSymbols.add(std::move(newType));
Brian Osman00fea5b2021-01-28 09:46:30 -0500630 return this->createNode(name.fOffset, ASTNode::Kind::kType, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700631}
632
633/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400634ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
635 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700636 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400637 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700638 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400639 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400640 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400641 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700642 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400643 this->expect(Token::Kind::TK_SEMICOLON, "';'");
John Stilesdc75a972020-11-25 16:24:55 -0500644 return type;
ethannicholasb3058bd2016-07-01 08:22:01 -0700645}
646
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400647/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
648 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400649ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
John Stiles08070f62020-11-17 10:45:49 -0500650 int offset = this->peek().fOffset;
651 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kVarDeclarations);
652 this->addChild(result, this->createNode(offset, ASTNode::Kind::kModifiers, mods));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400653 getNode(result).addChild(type);
John Stilesd39aec92020-12-03 14:37:16 -0500654
John Stiles80b02af2021-02-12 17:07:51 -0500655 auto parseArrayDimensions = [&](ASTNode::ID currentVar, ASTNode::VarData* vd) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400656 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500657 if (vd->fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500658 this->error(this->peek(), "multi-dimensional arrays are not supported");
659 return false;
660 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400661 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
John Stiles3b209362020-11-16 17:03:10 -0500662 this->createEmptyChild(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700663 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400664 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700665 if (!size) {
John Stilesd39aec92020-12-03 14:37:16 -0500666 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700667 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400668 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400669 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
John Stilesd39aec92020-12-03 14:37:16 -0500670 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700671 }
672 }
John Stilesd39aec92020-12-03 14:37:16 -0500673 vd->fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700674 }
John Stilesd39aec92020-12-03 14:37:16 -0500675 return true;
676 };
677
678 auto parseInitializer = [this](ASTNode::ID currentVar) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400679 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400680 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700681 if (!value) {
John Stilesd39aec92020-12-03 14:37:16 -0500682 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700683 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400684 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700685 }
John Stilesd39aec92020-12-03 14:37:16 -0500686 return true;
687 };
688
689 ASTNode::ID currentVar = this->createNode(offset, ASTNode::Kind::kVarDeclaration);
690 ASTNode::VarData vd{name, /*isArray=*/false};
691
692 getNode(result).addChild(currentVar);
693 if (!parseArrayDimensions(currentVar, &vd)) {
694 return ASTNode::ID::Invalid();
695 }
696 getNode(currentVar).setVarData(vd);
697 if (!parseInitializer(currentVar)) {
698 return ASTNode::ID::Invalid();
699 }
700
701 while (this->checkNext(Token::Kind::TK_COMMA)) {
702 Token identifierName;
John Stiles2630ea32020-12-04 10:51:21 -0500703 if (!this->expectIdentifier(&identifierName)) {
John Stilesd39aec92020-12-03 14:37:16 -0500704 return ASTNode::ID::Invalid();
705 }
706
707 currentVar = ASTNode::ID(fFile->fNodes.size());
708 vd = ASTNode::VarData{this->text(identifierName), /*isArray=*/false};
709 fFile->fNodes.emplace_back(&fFile->fNodes, offset, ASTNode::Kind::kVarDeclaration);
710
711 getNode(result).addChild(currentVar);
712 if (!parseArrayDimensions(currentVar, &vd)) {
713 return ASTNode::ID::Invalid();
714 }
715 getNode(currentVar).setVarData(vd);
716 if (!parseInitializer(currentVar)) {
717 return ASTNode::ID::Invalid();
718 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700719 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400720 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400721 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700722 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400723 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700724}
725
726/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400727ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400728 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400729 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700730 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400731 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700732 }
733 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500734 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400735 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700736 }
John Stiles3b209362020-11-16 17:03:10 -0500737 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kParameter);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400738 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
739 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400740 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500741 if (pd.fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500742 this->error(this->peek(), "multi-dimensional arrays are not supported");
743 return ASTNode::ID::Invalid();
744 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700745 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400746 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400747 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700748 }
John Stilesf94348f2020-12-23 18:58:43 -0500749 StringFragment arraySizeFrag = this->text(sizeToken);
750 SKSL_INT arraySize;
751 if (!SkSL::stoi(arraySizeFrag, &arraySize)) {
752 this->error(sizeToken, "array size is too large: " + arraySizeFrag);
753 return ASTNode::ID::Invalid();
754 }
755 this->addChild(result, this->createNode(sizeToken.fOffset, ASTNode::Kind::kInt, arraySize));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400756 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400757 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700758 }
John Stilesd39aec92020-12-03 14:37:16 -0500759 pd.fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700760 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400761 getNode(result).setParameterData(pd);
762 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700763}
764
Ethan Nicholasd608c092017-10-26 09:30:08 -0400765/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700766int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400767 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700768 return -1;
769 }
770 Token resultToken;
John Stilesf94348f2020-12-23 18:58:43 -0500771 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
772 return -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700773 }
John Stilesf94348f2020-12-23 18:58:43 -0500774 StringFragment resultFrag = this->text(resultToken);
775 SKSL_INT resultValue;
776 if (!SkSL::stoi(resultFrag, &resultValue)) {
777 this->error(resultToken, "value in layout is too large: " + resultFrag);
778 return -1;
779 }
780 return resultValue;
ethannicholasb3058bd2016-07-01 08:22:01 -0700781}
782
Ethan Nicholasd608c092017-10-26 09:30:08 -0400783/** EQ IDENTIFIER */
784StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400785 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400786 return StringFragment();
787 }
788 Token resultToken;
John Stiles2630ea32020-12-04 10:51:21 -0500789 if (!this->expectIdentifier(&resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400790 return StringFragment();
791 }
792 return this->text(resultToken);
793}
794
795
Ethan Nicholas762466e2017-06-29 10:03:38 -0400796/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400797StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400798 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400799 return "";
800 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700801 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400802 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400803 StringFragment code;
John Stiles18a45b32021-03-29 18:07:32 -0400804 code.fChars = fText.begin() + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400805 int level = 1;
806 bool done = false;
807 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700808 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400809 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400810 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400811 ++level;
812 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400813 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400814 --level;
815 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400816 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400817 if (level == 1) {
818 done = true;
819 }
820 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400821 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700822 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400823 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400824 default:
825 break;
826 }
827 if (!level) {
828 done = true;
829 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400830 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400831 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400832 this->pushback(std::move(next));
833 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400834 }
835 return code;
836}
837
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400838Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400839 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400840 Token t = this->nextToken();
841 String text = this->text(t);
842 auto found = layoutTokens->find(text);
843 if (found != layoutTokens->end()) {
844 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400845 case LayoutToken::SKPMCOLOR4F:
846 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500847 case LayoutToken::SKV4:
848 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400849 case LayoutToken::SKRECT:
850 return Layout::CType::kSkRect;
851 case LayoutToken::SKIRECT:
852 return Layout::CType::kSkIRect;
853 case LayoutToken::SKPMCOLOR:
854 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400855 case LayoutToken::BOOL:
856 return Layout::CType::kBool;
857 case LayoutToken::INT:
858 return Layout::CType::kInt32;
859 case LayoutToken::FLOAT:
860 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500861 case LayoutToken::SKM44:
862 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400863 default:
864 break;
865 }
866 }
867 this->error(t, "unsupported ctype");
868 }
869 return Layout::CType::kDefault;
870}
871
ethannicholas8ac838d2016-11-22 08:39:36 -0800872/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500873Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500874 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700875 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500876 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700877 int binding = -1;
878 int index = -1;
879 int set = -1;
880 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500881 int inputAttachmentIndex = -1;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500882 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
883 int maxVertices = -1;
884 int invocations = -1;
Brian Osmanf59a9612020-04-15 14:18:13 -0400885 StringFragment marker;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400886 StringFragment when;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400887 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400888 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
889 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500890 return Layout(flags, location, offset, binding, index, set, builtin,
Brian Osman4717fbb2021-02-23 13:12:09 -0500891 inputAttachmentIndex, primitive, maxVertices, invocations, marker, when,
Brian Osmanba7ef322021-02-23 13:37:22 -0500892 ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700893 }
894 for (;;) {
895 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700896 String text = this->text(t);
Brian Osmana77ed8b2021-02-23 12:54:22 -0500897 auto setFlag = [&](Layout::Flag f) {
898 if (flags & f) {
899 this->error(t, "layout qualifier '" + text + "' appears more than once");
900 }
901 flags |= f;
902 };
903 auto setPrimitive = [&](Layout::Primitive p) {
904 if (flags & Layout::kPrimitive_Flag) {
905 this->error(t, "only one primitive-type layout qualifier is allowed");
906 }
907 flags |= Layout::kPrimitive_Flag;
908 primitive = p;
909 };
910
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400911 auto found = layoutTokens->find(text);
912 if (found != layoutTokens->end()) {
913 switch (found->second) {
Brian Osmana77ed8b2021-02-23 12:54:22 -0500914 case LayoutToken::ORIGIN_UPPER_LEFT:
915 setFlag(Layout::kOriginUpperLeft_Flag);
916 break;
917 case LayoutToken::OVERRIDE_COVERAGE:
918 setFlag(Layout::kOverrideCoverage_Flag);
919 break;
Brian Osmandc4f8cd2021-02-25 15:30:46 -0500920 case LayoutToken::EARLY_FRAGMENT_TESTS:
921 setFlag(Layout::kEarlyFragmentTests_Flag);
922 break;
Brian Osmana77ed8b2021-02-23 12:54:22 -0500923 case LayoutToken::PUSH_CONSTANT:
924 setFlag(Layout::kPushConstant_Flag);
925 break;
926 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
927 setFlag(Layout::kBlendSupportAllEquations_Flag);
928 break;
929 case LayoutToken::TRACKED:
930 setFlag(Layout::kTracked_Flag);
931 break;
932 case LayoutToken::SRGB_UNPREMUL:
933 setFlag(Layout::kSRGBUnpremul_Flag);
934 break;
935 case LayoutToken::KEY:
936 setFlag(Layout::kKey_Flag);
937 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700938 case LayoutToken::LOCATION:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500939 setFlag(Layout::kLocation_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500940 location = this->layoutInt();
941 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700942 case LayoutToken::OFFSET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500943 setFlag(Layout::kOffset_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500944 offset = this->layoutInt();
945 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700946 case LayoutToken::BINDING:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500947 setFlag(Layout::kBinding_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500948 binding = this->layoutInt();
949 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700950 case LayoutToken::INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500951 setFlag(Layout::kIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500952 index = this->layoutInt();
953 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700954 case LayoutToken::SET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500955 setFlag(Layout::kSet_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500956 set = this->layoutInt();
957 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700958 case LayoutToken::BUILTIN:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500959 setFlag(Layout::kBuiltin_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500960 builtin = this->layoutInt();
961 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700962 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500963 setFlag(Layout::kInputAttachmentIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500964 inputAttachmentIndex = this->layoutInt();
965 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700966 case LayoutToken::POINTS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500967 setPrimitive(Layout::kPoints_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500968 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700969 case LayoutToken::LINES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500970 setPrimitive(Layout::kLines_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500971 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700972 case LayoutToken::LINE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500973 setPrimitive(Layout::kLineStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500974 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700975 case LayoutToken::LINES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500976 setPrimitive(Layout::kLinesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500977 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700978 case LayoutToken::TRIANGLES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500979 setPrimitive(Layout::kTriangles_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500980 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700981 case LayoutToken::TRIANGLE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500982 setPrimitive(Layout::kTriangleStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500983 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700984 case LayoutToken::TRIANGLES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500985 setPrimitive(Layout::kTrianglesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500986 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700987 case LayoutToken::MAX_VERTICES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500988 setFlag(Layout::kMaxVertices_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500989 maxVertices = this->layoutInt();
990 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700991 case LayoutToken::INVOCATIONS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500992 setFlag(Layout::kInvocations_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500993 invocations = this->layoutInt();
994 break;
Brian Osmanf59a9612020-04-15 14:18:13 -0400995 case LayoutToken::MARKER:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500996 setFlag(Layout::kMarker_Flag);
Brian Osmanf59a9612020-04-15 14:18:13 -0400997 marker = this->layoutCode();
998 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700999 case LayoutToken::WHEN:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001000 setFlag(Layout::kWhen_Flag);
Ethan Nicholas762466e2017-06-29 10:03:38 -04001001 when = this->layoutCode();
1002 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -04001003 case LayoutToken::CTYPE:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001004 setFlag(Layout::kCType_Flag);
Ethan Nicholas78aceb22018-08-31 16:13:58 -04001005 ctype = this->layoutCType();
1006 break;
1007 default:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001008 this->error(t, "'" + text + "' is not a valid layout qualifier");
Ethan Nicholasd608c092017-10-26 09:30:08 -04001009 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -05001010 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001011 } else {
Brian Osmana77ed8b2021-02-23 12:54:22 -05001012 this->error(t, "'" + text + "' is not a valid layout qualifier");
ethannicholasb3058bd2016-07-01 08:22:01 -07001013 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001014 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001015 break;
1016 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001017 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001018 break;
1019 }
1020 }
1021 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -05001022 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
Brian Osmanba7ef322021-02-23 13:37:22 -05001023 primitive, maxVertices, invocations, marker, when, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -07001024}
1025
Brian Salomonf9f45122016-11-29 11:59:17 -05001026/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Brian Osman58ee8982021-02-18 15:39:38 -05001027 VARYING | INLINE)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -05001028Modifiers Parser::modifiers() {
1029 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -07001030 int flags = 0;
1031 for (;;) {
1032 // TODO: handle duplicate / incompatible flags
John Stiles7d5b90a2021-01-21 14:26:51 -05001033 int tokenFlag = parse_modifier_token(peek().fKind);
1034 if (!tokenFlag) {
1035 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001036 }
John Stiles7d5b90a2021-01-21 14:26:51 -05001037 flags |= tokenFlag;
1038 this->nextToken();
ethannicholasb3058bd2016-07-01 08:22:01 -07001039 }
John Stiles7d5b90a2021-01-21 14:26:51 -05001040 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001041}
1042
Ethan Nicholas11d53972016-11-28 11:23:23 -05001043Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
1044 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -07001045 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001046 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001047 }
1048 return result;
1049}
1050
1051/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001052ASTNode::ID Parser::statement() {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001053 Token start = this->nextToken();
1054 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001055 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001056 return ASTNode::ID::Invalid();
1057 }
1058 this->pushback(start);
ethannicholasb3058bd2016-07-01 08:22:01 -07001059 switch (start.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001060 case Token::Kind::TK_IF: // fall through
1061 case Token::Kind::TK_STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -07001062 return this->ifStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001063 case Token::Kind::TK_FOR:
ethannicholasb3058bd2016-07-01 08:22:01 -07001064 return this->forStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001065 case Token::Kind::TK_DO:
ethannicholasb3058bd2016-07-01 08:22:01 -07001066 return this->doStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001067 case Token::Kind::TK_WHILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001068 return this->whileStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001069 case Token::Kind::TK_SWITCH: // fall through
1070 case Token::Kind::TK_STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -05001071 return this->switchStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001072 case Token::Kind::TK_RETURN:
ethannicholasb3058bd2016-07-01 08:22:01 -07001073 return this->returnStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001074 case Token::Kind::TK_BREAK:
ethannicholasb3058bd2016-07-01 08:22:01 -07001075 return this->breakStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001076 case Token::Kind::TK_CONTINUE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001077 return this->continueStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001078 case Token::Kind::TK_DISCARD:
ethannicholasb3058bd2016-07-01 08:22:01 -07001079 return this->discardStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001080 case Token::Kind::TK_LBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001081 return this->block();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001082 case Token::Kind::TK_SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001083 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -05001084 return this->createNode(start.fOffset, ASTNode::Kind::kBlock);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001085 case Token::Kind::TK_CONST:
Ethan Nicholasfc994162019-06-06 10:04:27 -04001086 return this->varDeclarations();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001087 case Token::Kind::TK_IDENTIFIER:
John Stiles76389b72021-01-25 13:49:05 -05001088 return this->varDeclarationsOrExpressionStatement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001089 default:
1090 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001091 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001092}
1093
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001094/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* QUESTION? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001095ASTNode::ID Parser::type() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001096 Token type;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001097 if (!this->expect(Token::Kind::TK_IDENTIFIER, "a type", &type)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001098 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001099 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001100 if (!this->isType(this->text(type))) {
1101 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001102 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001103 }
Brian Osman00fea5b2021-01-28 09:46:30 -05001104 ASTNode::ID result = this->createNode(type.fOffset, ASTNode::Kind::kType, this->text(type));
John Stilesd39aec92020-12-03 14:37:16 -05001105 bool isArray = false;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001106 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stilesd39aec92020-12-03 14:37:16 -05001107 if (isArray) {
1108 this->error(this->peek(), "multi-dimensional arrays are not supported");
1109 return ASTNode::ID::Invalid();
1110 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001111 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001112 SKSL_INT i;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001113 if (this->intLiteral(&i)) {
John Stiles08070f62020-11-17 10:45:49 -05001114 this->addChild(result, this->createNode(this->peek().fOffset,
1115 ASTNode::Kind::kInt, i));
Ethan Nicholas50afc172017-02-16 14:49:57 -05001116 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001117 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001118 }
1119 } else {
John Stiles3b209362020-11-16 17:03:10 -05001120 this->createEmptyChild(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001121 }
John Stilesd39aec92020-12-03 14:37:16 -05001122 isArray = true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001123 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001124 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001125 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001126}
1127
John Stiles57a996b2020-04-19 19:05:12 -07001128/* IDENTIFIER LBRACE
1129 varDeclaration+
1130 RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001131ASTNode::ID Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001132 Token name;
John Stiles2630ea32020-12-04 10:51:21 -05001133 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001134 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001135 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001136 if (peek().fKind != Token::Kind::TK_LBRACE) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001137 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001138 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -07001139 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001140 this->error(name, "no type named '" + this->text(name) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001141 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001142 }
John Stiles3b209362020-11-16 17:03:10 -05001143 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kInterfaceBlock);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001144 ASTNode::InterfaceBlockData id(mods, this->text(name), 0, "", 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07001145 this->nextToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001146 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001147 ASTNode::ID decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001148 if (!decl) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001149 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001150 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001151 getNode(result).addChild(decl);
1152 ++id.fDeclarationCount;
ethannicholasb3058bd2016-07-01 08:22:01 -07001153 }
John Stiles57a996b2020-04-19 19:05:12 -07001154 if (id.fDeclarationCount == 0) {
1155 this->error(name, "interface block '" + this->text(name) +
1156 "' must contain at least one member");
1157 return ASTNode::ID::Invalid();
1158 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001159 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001160 std::vector<ASTNode> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001161 StringFragment instanceName;
1162 Token instanceNameToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001163 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &instanceNameToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001164 id.fInstanceName = this->text(instanceNameToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001165 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stilesd39aec92020-12-03 14:37:16 -05001166 if (id.fIsArray) {
1167 this->error(this->peek(), "multi-dimensional arrays are not supported");
1168 return false;
1169 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001170 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001171 ASTNode::ID size = this->expression();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001172 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001173 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001174 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001175 getNode(result).addChild(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001176 } else {
John Stiles3b209362020-11-16 17:03:10 -05001177 this->createEmptyChild(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001178 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001179 this->expect(Token::Kind::TK_RBRACKET, "']'");
John Stilesd39aec92020-12-03 14:37:16 -05001180 id.fIsArray = true;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001181 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001182 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -07001183 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001184 getNode(result).setInterfaceBlockData(id);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001185 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001186 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001187}
1188
1189/* IF LPAREN expression RPAREN statement (ELSE statement)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001190ASTNode::ID Parser::ifStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001191 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001192 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_IF, &start);
1193 if (!isStatic && !this->expect(Token::Kind::TK_IF, "'if'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001194 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001195 }
John Stiles3b209362020-11-16 17:03:10 -05001196 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kIf, isStatic);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001197 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001198 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001199 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001200 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001201 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001202 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001203 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001204 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001205 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001206 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001207 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001208 ASTNode::ID ifTrue = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001209 if (!ifTrue) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001210 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001211 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001212 getNode(result).addChild(ifTrue);
1213 ASTNode::ID ifFalse;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001214 if (this->checkNext(Token::Kind::TK_ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001215 ifFalse = this->statement();
1216 if (!ifFalse) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001217 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001218 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001219 getNode(result).addChild(ifFalse);
ethannicholasb3058bd2016-07-01 08:22:01 -07001220 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001221 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001222}
1223
1224/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001225ASTNode::ID Parser::doStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001226 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001227 if (!this->expect(Token::Kind::TK_DO, "'do'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001228 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001229 }
John Stiles3b209362020-11-16 17:03:10 -05001230 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kDo);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001231 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001232 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001233 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001234 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001235 getNode(result).addChild(statement);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001236 if (!this->expect(Token::Kind::TK_WHILE, "'while'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001237 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001238 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001239 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001240 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001241 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001242 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001243 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001244 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001245 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001246 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001247 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001248 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001249 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001250 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001251 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001252 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001253 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001254}
1255
1256/* WHILE LPAREN expression RPAREN STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001257ASTNode::ID Parser::whileStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001258 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001259 if (!this->expect(Token::Kind::TK_WHILE, "'while'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001260 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001261 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001262 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001263 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001264 }
John Stiles3b209362020-11-16 17:03:10 -05001265 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kWhile);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001266 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001267 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001268 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001269 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001270 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001271 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001272 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001273 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001274 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001275 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001276 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001277 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001278 getNode(result).addChild(statement);
1279 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001280}
1281
Ethan Nicholasaf197692017-02-27 13:26:45 -05001282/* CASE expression COLON statement* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001283ASTNode::ID Parser::switchCase() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001284 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001285 if (!this->expect(Token::Kind::TK_CASE, "'case'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001286 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001287 }
John Stiles3b209362020-11-16 17:03:10 -05001288 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kSwitchCase);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001289 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001290 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001291 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001292 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001293 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
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(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001297 while (this->peek().fKind != Token::Kind::TK_RBRACE &&
1298 this->peek().fKind != Token::Kind::TK_CASE &&
1299 this->peek().fKind != Token::Kind::TK_DEFAULT) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001300 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001301 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001302 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001303 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001304 getNode(result).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001305 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001306 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001307}
1308
1309/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001310ASTNode::ID Parser::switchStatement() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001311 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001312 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_SWITCH, &start);
1313 if (!isStatic && !this->expect(Token::Kind::TK_SWITCH, "'switch'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001314 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001315 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001316 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
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 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001320 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001321 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001322 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001323 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001324 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001325 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001326 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001327 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001328 }
John Stiles3b209362020-11-16 17:03:10 -05001329 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kSwitch, isStatic);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001330 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001331 while (this->peek().fKind == Token::Kind::TK_CASE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001332 ASTNode::ID c = this->switchCase();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001333 if (!c) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001334 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001335 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001336 getNode(result).addChild(c);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001337 }
1338 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1339 // parts of the compiler may rely upon this assumption.
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001340 if (this->peek().fKind == Token::Kind::TK_DEFAULT) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001341 Token defaultStart;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001342 SkAssertResult(this->expect(Token::Kind::TK_DEFAULT, "'default'", &defaultStart));
1343 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001344 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001345 }
John Stiles3b209362020-11-16 17:03:10 -05001346 ASTNode::ID defaultCase = this->addChild(
1347 result, this->createNode(defaultStart.fOffset, ASTNode::Kind::kSwitchCase));
1348 this->createEmptyChild(defaultCase); // empty test to signify default case
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001349 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001350 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001351 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001352 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001353 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001354 getNode(defaultCase).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001355 }
Ethan Nicholasaf197692017-02-27 13:26:45 -05001356 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001357 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001358 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001359 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001360 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001361}
1362
Ethan Nicholas11d53972016-11-28 11:23:23 -05001363/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001364 STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001365ASTNode::ID Parser::forStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001366 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001367 if (!this->expect(Token::Kind::TK_FOR, "'for'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001368 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001369 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001370 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001371 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001372 }
John Stiles3b209362020-11-16 17:03:10 -05001373 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kFor);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001374 ASTNode::ID initializer;
ethannicholasb3058bd2016-07-01 08:22:01 -07001375 Token nextToken = this->peek();
1376 switch (nextToken.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001377 case Token::Kind::TK_SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001378 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -05001379 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001380 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001381 case Token::Kind::TK_CONST: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001382 initializer = this->varDeclarations();
1383 if (!initializer) {
1384 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001385 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001386 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001387 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001388 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001389 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001390 if (this->isType(this->text(nextToken))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001391 initializer = this->varDeclarations();
1392 if (!initializer) {
1393 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001394 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001395 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001396 break;
1397 }
John Stiles30212b72020-06-11 17:55:07 -04001398 [[fallthrough]];
1399 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001400 default:
1401 initializer = this->expressionStatement();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001402 if (!initializer) {
1403 return ASTNode::ID::Invalid();
1404 }
1405 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001406 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001407 ASTNode::ID test;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001408 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001409 test = this->expression();
1410 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001411 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001412 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001413 getNode(result).addChild(test);
1414 } else {
John Stiles3b209362020-11-16 17:03:10 -05001415 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001416 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001417 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001418 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001419 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001420 ASTNode::ID next;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001421 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001422 next = this->expression();
1423 if (!next) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001424 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001425 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001426 getNode(result).addChild(next);
1427 } else {
John Stiles3b209362020-11-16 17:03:10 -05001428 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001429 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001430 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001431 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001432 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001433 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001434 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001435 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001436 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001437 getNode(result).addChild(statement);
1438 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001439}
1440
1441/* RETURN expression? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001442ASTNode::ID Parser::returnStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001443 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001444 if (!this->expect(Token::Kind::TK_RETURN, "'return'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001445 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001446 }
John Stiles3b209362020-11-16 17:03:10 -05001447 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kReturn);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001448 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001449 ASTNode::ID expression = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001450 if (!expression) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001451 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001452 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001453 getNode(result).addChild(expression);
ethannicholasb3058bd2016-07-01 08:22:01 -07001454 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001455 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001456 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001457 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001458 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001459}
1460
1461/* BREAK SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001462ASTNode::ID Parser::breakStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001463 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001464 if (!this->expect(Token::Kind::TK_BREAK, "'break'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001465 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001466 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001467 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001468 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001469 }
John Stiles3b209362020-11-16 17:03:10 -05001470 return this->createNode(start.fOffset, ASTNode::Kind::kBreak);
ethannicholasb3058bd2016-07-01 08:22:01 -07001471}
1472
1473/* CONTINUE SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001474ASTNode::ID Parser::continueStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001475 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001476 if (!this->expect(Token::Kind::TK_CONTINUE, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001477 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001478 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001479 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001480 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001481 }
John Stiles3b209362020-11-16 17:03:10 -05001482 return this->createNode(start.fOffset, ASTNode::Kind::kContinue);
ethannicholasb3058bd2016-07-01 08:22:01 -07001483}
1484
1485/* DISCARD SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001486ASTNode::ID Parser::discardStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001487 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001488 if (!this->expect(Token::Kind::TK_DISCARD, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001489 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001490 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001491 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001492 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001493 }
John Stiles3b209362020-11-16 17:03:10 -05001494 return this->createNode(start.fOffset, ASTNode::Kind::kDiscard);
ethannicholasb3058bd2016-07-01 08:22:01 -07001495}
1496
1497/* LBRACE statement* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001498ASTNode::ID Parser::block() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001499 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001500 if (!this->expect(Token::Kind::TK_LBRACE, "'{'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001501 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001502 }
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001503 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001504 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001505 return ASTNode::ID::Invalid();
1506 }
John Stiles3b209362020-11-16 17:03:10 -05001507 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -07001508 for (;;) {
1509 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001510 case Token::Kind::TK_RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001511 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001512 return result;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001513 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001514 this->error(this->peek(), "expected '}', but found end of file");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001515 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001516 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001517 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001518 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001519 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001520 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001521 getNode(result).addChild(statement);
ethannicholasb3058bd2016-07-01 08:22:01 -07001522 }
1523 }
1524 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001525 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001526}
1527
1528/* expression SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001529ASTNode::ID Parser::expressionStatement() {
1530 ASTNode::ID expr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001531 if (expr) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001532 if (this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001533 return expr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001534 }
1535 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001536 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001537}
1538
1539/* assignmentExpression (COMMA assignmentExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001540ASTNode::ID Parser::expression() {
1541 ASTNode::ID result = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001542 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001543 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001544 }
1545 Token t;
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001546 AutoDepth depth(this);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001547 while (this->checkNext(Token::Kind::TK_COMMA, &t)) {
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001548 if (!depth.increase()) {
1549 return ASTNode::ID::Invalid();
1550 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001551 ASTNode::ID right = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001552 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001553 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001554 }
John Stiles45990502021-02-16 10:55:27 -05001555 ASTNode::ID newResult = this->createNode(t.fOffset, ASTNode::Kind::kBinary,
1556 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001557 getNode(newResult).addChild(result);
1558 getNode(newResult).addChild(right);
1559 result = newResult;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001560 }
1561 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001562}
1563
1564/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1565 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1566 assignmentExpression)*
1567 */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001568ASTNode::ID Parser::assignmentExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001569 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001570 ASTNode::ID result = this->ternaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001571 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001572 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001573 }
1574 for (;;) {
1575 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001576 case Token::Kind::TK_EQ: // fall through
1577 case Token::Kind::TK_STAREQ: // fall through
1578 case Token::Kind::TK_SLASHEQ: // fall through
1579 case Token::Kind::TK_PERCENTEQ: // fall through
1580 case Token::Kind::TK_PLUSEQ: // fall through
1581 case Token::Kind::TK_MINUSEQ: // fall through
1582 case Token::Kind::TK_SHLEQ: // fall through
1583 case Token::Kind::TK_SHREQ: // fall through
1584 case Token::Kind::TK_BITWISEANDEQ: // fall through
1585 case Token::Kind::TK_BITWISEXOREQ: // fall through
John Stiles8b3b1592020-11-23 11:06:44 -05001586 case Token::Kind::TK_BITWISEOREQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001587 if (!depth.increase()) {
1588 return ASTNode::ID::Invalid();
1589 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001590 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001591 ASTNode::ID right = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001592 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001593 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001594 }
John Stiles3b209362020-11-16 17:03:10 -05001595 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001596 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001597 getNode(newResult).addChild(result);
1598 getNode(newResult).addChild(right);
1599 result = newResult;
1600 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001601 }
1602 default:
1603 return result;
1604 }
1605 }
1606}
1607
1608/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001609ASTNode::ID Parser::ternaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001610 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001611 ASTNode::ID base = this->logicalOrExpression();
1612 if (!base) {
1613 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001614 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001615 if (this->checkNext(Token::Kind::TK_QUESTION)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001616 if (!depth.increase()) {
1617 return ASTNode::ID::Invalid();
1618 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001619 ASTNode::ID trueExpr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001620 if (!trueExpr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001621 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001622 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001623 if (this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001624 ASTNode::ID falseExpr = this->assignmentExpression();
1625 if (!falseExpr) {
1626 return ASTNode::ID::Invalid();
1627 }
John Stiles3b209362020-11-16 17:03:10 -05001628 ASTNode::ID ternary = this->createNode(getNode(base).fOffset, ASTNode::Kind::kTernary);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001629 getNode(ternary).addChild(base);
1630 getNode(ternary).addChild(trueExpr);
1631 getNode(ternary).addChild(falseExpr);
1632 return ternary;
ethannicholasb3058bd2016-07-01 08:22:01 -07001633 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001634 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001635 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001636 return base;
ethannicholasb3058bd2016-07-01 08:22:01 -07001637}
1638
1639/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001640ASTNode::ID Parser::logicalOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001641 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001642 ASTNode::ID result = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001643 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001644 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001645 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001646 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001647 while (this->checkNext(Token::Kind::TK_LOGICALOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001648 if (!depth.increase()) {
1649 return ASTNode::ID::Invalid();
1650 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001651 ASTNode::ID right = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001652 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001653 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001654 }
John Stiles3b209362020-11-16 17:03:10 -05001655 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001656 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001657 getNode(newResult).addChild(result);
1658 getNode(newResult).addChild(right);
1659 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001660 }
1661 return result;
1662}
1663
1664/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001665ASTNode::ID Parser::logicalXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001666 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001667 ASTNode::ID result = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001668 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001669 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001670 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001671 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001672 while (this->checkNext(Token::Kind::TK_LOGICALXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001673 if (!depth.increase()) {
1674 return ASTNode::ID::Invalid();
1675 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001676 ASTNode::ID right = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001677 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001678 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001679 }
John Stiles3b209362020-11-16 17:03:10 -05001680 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001681 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001682 getNode(newResult).addChild(result);
1683 getNode(newResult).addChild(right);
1684 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001685 }
1686 return result;
1687}
1688
1689/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001690ASTNode::ID Parser::logicalAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001691 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001692 ASTNode::ID result = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001693 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001694 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001695 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001696 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001697 while (this->checkNext(Token::Kind::TK_LOGICALAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001698 if (!depth.increase()) {
1699 return ASTNode::ID::Invalid();
1700 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001701 ASTNode::ID right = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001702 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001703 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001704 }
John Stiles3b209362020-11-16 17:03:10 -05001705 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001706 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001707 getNode(newResult).addChild(result);
1708 getNode(newResult).addChild(right);
1709 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001710 }
1711 return result;
1712}
1713
1714/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001715ASTNode::ID Parser::bitwiseOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001716 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001717 ASTNode::ID result = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001718 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001719 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001720 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001721 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001722 while (this->checkNext(Token::Kind::TK_BITWISEOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001723 if (!depth.increase()) {
1724 return ASTNode::ID::Invalid();
1725 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001726 ASTNode::ID right = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001727 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001728 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001729 }
John Stiles45990502021-02-16 10:55:27 -05001730 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
1731 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001732 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/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001740ASTNode::ID Parser::bitwiseXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001741 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001742 ASTNode::ID result = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001743 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001744 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001745 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001746 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001747 while (this->checkNext(Token::Kind::TK_BITWISEXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001748 if (!depth.increase()) {
1749 return ASTNode::ID::Invalid();
1750 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001751 ASTNode::ID right = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001752 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001753 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001754 }
John Stiles45990502021-02-16 10:55:27 -05001755 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
1756 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001757 getNode(newResult).addChild(result);
1758 getNode(newResult).addChild(right);
1759 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001760 }
1761 return result;
1762}
1763
1764/* equalityExpression (BITWISEAND equalityExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001765ASTNode::ID Parser::bitwiseAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001766 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001767 ASTNode::ID result = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001768 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001769 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001770 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001771 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001772 while (this->checkNext(Token::Kind::TK_BITWISEAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001773 if (!depth.increase()) {
1774 return ASTNode::ID::Invalid();
1775 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001776 ASTNode::ID right = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001777 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001778 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001779 }
John Stiles3b209362020-11-16 17:03:10 -05001780 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001781 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001782 getNode(newResult).addChild(result);
1783 getNode(newResult).addChild(right);
1784 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001785 }
1786 return result;
1787}
1788
1789/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001790ASTNode::ID Parser::equalityExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001791 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001792 ASTNode::ID result = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001793 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001794 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001795 }
1796 for (;;) {
1797 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001798 case Token::Kind::TK_EQEQ: // fall through
1799 case Token::Kind::TK_NEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001800 if (!depth.increase()) {
1801 return ASTNode::ID::Invalid();
1802 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001803 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001804 ASTNode::ID right = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001805 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001806 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001807 }
John Stiles3b209362020-11-16 17:03:10 -05001808 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001809 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001810 getNode(newResult).addChild(result);
1811 getNode(newResult).addChild(right);
1812 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001813 break;
1814 }
1815 default:
1816 return result;
1817 }
1818 }
1819}
1820
1821/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001822ASTNode::ID Parser::relationalExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001823 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001824 ASTNode::ID result = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001825 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001826 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001827 }
1828 for (;;) {
1829 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001830 case Token::Kind::TK_LT: // fall through
1831 case Token::Kind::TK_GT: // fall through
1832 case Token::Kind::TK_LTEQ: // fall through
1833 case Token::Kind::TK_GTEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001834 if (!depth.increase()) {
1835 return ASTNode::ID::Invalid();
1836 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001837 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001838 ASTNode::ID right = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001839 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001840 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001841 }
John Stiles3b209362020-11-16 17:03:10 -05001842 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001843 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001844 getNode(newResult).addChild(result);
1845 getNode(newResult).addChild(right);
1846 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001847 break;
1848 }
1849 default:
1850 return result;
1851 }
1852 }
1853}
1854
1855/* additiveExpression ((SHL | SHR) additiveExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001856ASTNode::ID Parser::shiftExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001857 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001858 ASTNode::ID result = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001859 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001860 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001861 }
1862 for (;;) {
1863 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001864 case Token::Kind::TK_SHL: // fall through
1865 case Token::Kind::TK_SHR: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001866 if (!depth.increase()) {
1867 return ASTNode::ID::Invalid();
1868 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001869 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001870 ASTNode::ID right = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001871 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001872 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001873 }
John Stiles3b209362020-11-16 17:03:10 -05001874 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001875 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001876 getNode(newResult).addChild(result);
1877 getNode(newResult).addChild(right);
1878 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001879 break;
1880 }
1881 default:
1882 return result;
1883 }
1884 }
1885}
1886
1887/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001888ASTNode::ID Parser::additiveExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001889 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001890 ASTNode::ID result = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001891 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001892 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001893 }
1894 for (;;) {
1895 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001896 case Token::Kind::TK_PLUS: // fall through
1897 case Token::Kind::TK_MINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001898 if (!depth.increase()) {
1899 return ASTNode::ID::Invalid();
1900 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001901 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001902 ASTNode::ID right = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001903 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001904 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001905 }
John Stiles3b209362020-11-16 17:03:10 -05001906 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001907 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001908 getNode(newResult).addChild(result);
1909 getNode(newResult).addChild(right);
1910 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001911 break;
1912 }
1913 default:
1914 return result;
1915 }
1916 }
1917}
1918
1919/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001920ASTNode::ID Parser::multiplicativeExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001921 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001922 ASTNode::ID result = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001923 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001924 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001925 }
1926 for (;;) {
1927 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001928 case Token::Kind::TK_STAR: // fall through
1929 case Token::Kind::TK_SLASH: // fall through
1930 case Token::Kind::TK_PERCENT: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001931 if (!depth.increase()) {
1932 return ASTNode::ID::Invalid();
1933 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001934 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001935 ASTNode::ID right = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001936 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001937 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001938 }
John Stiles3b209362020-11-16 17:03:10 -05001939 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001940 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001941 getNode(newResult).addChild(result);
1942 getNode(newResult).addChild(right);
1943 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001944 break;
1945 }
1946 default:
1947 return result;
1948 }
1949 }
1950}
1951
1952/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001953ASTNode::ID Parser::unaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001954 AutoDepth depth(this);
ethannicholasb3058bd2016-07-01 08:22:01 -07001955 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001956 case Token::Kind::TK_PLUS: // fall through
1957 case Token::Kind::TK_MINUS: // fall through
1958 case Token::Kind::TK_LOGICALNOT: // fall through
1959 case Token::Kind::TK_BITWISENOT: // fall through
1960 case Token::Kind::TK_PLUSPLUS: // fall through
1961 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001962 if (!depth.increase()) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001963 return ASTNode::ID::Invalid();
Ethan Nicholas6dcc3252019-02-20 15:18:36 -05001964 }
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001965 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001966 ASTNode::ID expr = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001967 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001968 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001969 }
John Stiles45990502021-02-16 10:55:27 -05001970 ASTNode::ID result = this->createNode(t.fOffset, ASTNode::Kind::kPrefix,
1971 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001972 getNode(result).addChild(expr);
1973 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001974 }
1975 default:
1976 return this->postfixExpression();
1977 }
1978}
1979
1980/* term suffix* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001981ASTNode::ID Parser::postfixExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001982 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001983 ASTNode::ID result = this->term();
ethannicholasb3058bd2016-07-01 08:22:01 -07001984 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001985 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001986 }
1987 for (;;) {
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001988 Token t = this->peek();
1989 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001990 case Token::Kind::TK_FLOAT_LITERAL:
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001991 if (this->text(t)[0] != '.') {
1992 return result;
1993 }
John Stiles30212b72020-06-11 17:55:07 -04001994 [[fallthrough]];
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001995 case Token::Kind::TK_LBRACKET:
1996 case Token::Kind::TK_DOT:
1997 case Token::Kind::TK_LPAREN:
1998 case Token::Kind::TK_PLUSPLUS:
1999 case Token::Kind::TK_MINUSMINUS:
2000 case Token::Kind::TK_COLONCOLON:
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002001 if (!depth.increase()) {
2002 return ASTNode::ID::Invalid();
2003 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002004 result = this->suffix(result);
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002005 if (!result) {
2006 return ASTNode::ID::Invalid();
2007 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002008 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002009 default:
2010 return result;
2011 }
2012 }
2013}
2014
Ethan Nicholas11d53972016-11-28 11:23:23 -05002015/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
Ethan Nicholase455f652019-09-13 12:52:55 -04002016 PLUSPLUS | MINUSMINUS | COLONCOLON IDENTIFIER | FLOAT_LITERAL [IDENTIFIER] */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002017ASTNode::ID Parser::suffix(ASTNode::ID base) {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002018 SkASSERT(base);
ethannicholasb3058bd2016-07-01 08:22:01 -07002019 Token next = this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002020 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002021 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002022 return ASTNode::ID::Invalid();
2023 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002024 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002025 case Token::Kind::TK_LBRACKET: {
2026 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
John Stiles3b209362020-11-16 17:03:10 -05002027 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kIndex);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002028 getNode(result).addChild(base);
2029 return result;
ethannicholas5961bc92016-10-12 06:39:56 -07002030 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002031 ASTNode::ID e = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002032 if (!e) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002033 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002034 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002035 this->expect(Token::Kind::TK_RBRACKET, "']' to complete array access expression");
John Stiles3b209362020-11-16 17:03:10 -05002036 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kIndex);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002037 getNode(result).addChild(base);
2038 getNode(result).addChild(e);
2039 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002040 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002041 case Token::Kind::TK_COLONCOLON: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002042 int offset = this->peek().fOffset;
2043 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002044 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002045 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kScope,
2046 std::move(text));
Brian Osman6518d772020-09-10 16:50:06 -04002047 getNode(result).addChild(base);
2048 return result;
2049 }
2050 return ASTNode::ID::Invalid();
2051 }
2052 case Token::Kind::TK_DOT: {
2053 int offset = this->peek().fOffset;
2054 StringFragment text;
2055 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002056 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kField,
2057 std::move(text));
Ethan Nicholasfc994162019-06-06 10:04:27 -04002058 getNode(result).addChild(base);
2059 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002060 }
Brian Osman6518d772020-09-10 16:50:06 -04002061 [[fallthrough]];
Ethan Nicholase455f652019-09-13 12:52:55 -04002062 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002063 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholase455f652019-09-13 12:52:55 -04002064 // Swizzles that start with a constant number, e.g. '.000r', will be tokenized as
2065 // floating point literals, possibly followed by an identifier. Handle that here.
2066 StringFragment field = this->text(next);
2067 SkASSERT(field.fChars[0] == '.');
2068 ++field.fChars;
2069 --field.fLength;
2070 for (size_t i = 0; i < field.fLength; ++i) {
2071 if (field.fChars[i] != '0' && field.fChars[i] != '1') {
2072 this->error(next, "invalid swizzle");
2073 return ASTNode::ID::Invalid();
2074 }
2075 }
2076 // use the next *raw* token so we don't ignore whitespace - we only care about
2077 // identifiers that directly follow the float
2078 Token id = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002079 if (id.fKind == Token::Kind::TK_IDENTIFIER) {
Ethan Nicholase455f652019-09-13 12:52:55 -04002080 field.fLength += id.fLength;
2081 } else {
2082 this->pushback(id);
2083 }
John Stiles3b209362020-11-16 17:03:10 -05002084 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kField, field);
Ethan Nicholase455f652019-09-13 12:52:55 -04002085 getNode(result).addChild(base);
2086 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002087 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002088 case Token::Kind::TK_LPAREN: {
John Stiles3b209362020-11-16 17:03:10 -05002089 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kCall);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002090 getNode(result).addChild(base);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002091 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002092 for (;;) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002093 ASTNode::ID expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002094 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002095 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002096 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002097 getNode(result).addChild(expr);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002098 if (!this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002099 break;
2100 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002101 }
2102 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002103 this->expect(Token::Kind::TK_RPAREN, "')' to complete function parameters");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002104 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002105 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002106 case Token::Kind::TK_PLUSPLUS: // fall through
2107 case Token::Kind::TK_MINUSMINUS: {
John Stiles45990502021-02-16 10:55:27 -05002108 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kPostfix,
2109 Operator(next.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04002110 getNode(result).addChild(base);
2111 return result;
2112 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002113 default: {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002114 this->error(next, "expected expression suffix, but found '" + this->text(next) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002115 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002116 }
2117 }
2118}
2119
Brian Osman33c64a42020-12-23 10:26:38 -05002120/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | '(' expression ')' */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002121ASTNode::ID Parser::term() {
ethannicholasb3058bd2016-07-01 08:22:01 -07002122 Token t = this->peek();
2123 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002124 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002125 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002126 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002127 return this->createNode(t.fOffset, ASTNode::Kind::kIdentifier, std::move(text));
ethannicholasb3058bd2016-07-01 08:22:01 -07002128 }
John Stiles30212b72020-06-11 17:55:07 -04002129 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002130 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002131 case Token::Kind::TK_INT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002132 SKSL_INT i;
ethannicholasb3058bd2016-07-01 08:22:01 -07002133 if (this->intLiteral(&i)) {
John Stiles3b209362020-11-16 17:03:10 -05002134 return this->createNode(t.fOffset, ASTNode::Kind::kInt, i);
ethannicholasb3058bd2016-07-01 08:22:01 -07002135 }
2136 break;
2137 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002138 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002139 SKSL_FLOAT f;
ethannicholasb3058bd2016-07-01 08:22:01 -07002140 if (this->floatLiteral(&f)) {
John Stiles3b209362020-11-16 17:03:10 -05002141 return this->createNode(t.fOffset, ASTNode::Kind::kFloat, f);
ethannicholasb3058bd2016-07-01 08:22:01 -07002142 }
2143 break;
2144 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002145 case Token::Kind::TK_TRUE_LITERAL: // fall through
2146 case Token::Kind::TK_FALSE_LITERAL: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002147 bool b;
2148 if (this->boolLiteral(&b)) {
John Stiles3b209362020-11-16 17:03:10 -05002149 return this->createNode(t.fOffset, ASTNode::Kind::kBool, b);
ethannicholasb3058bd2016-07-01 08:22:01 -07002150 }
2151 break;
2152 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002153 case Token::Kind::TK_LPAREN: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002154 this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002155 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002156 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002157 return ASTNode::ID::Invalid();
2158 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002159 ASTNode::ID result = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002160 if (result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002161 this->expect(Token::Kind::TK_RPAREN, "')' to complete expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002162 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002163 }
2164 break;
2165 }
2166 default:
2167 this->nextToken();
John Stilesf94348f2020-12-23 18:58:43 -05002168 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002169 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002170 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002171}
2172
2173/* INT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002174bool Parser::intLiteral(SKSL_INT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002175 Token t;
John Stilesf94348f2020-12-23 18:58:43 -05002176 if (!this->expect(Token::Kind::TK_INT_LITERAL, "integer literal", &t)) {
2177 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07002178 }
John Stilesf94348f2020-12-23 18:58:43 -05002179 StringFragment s = this->text(t);
2180 if (!SkSL::stoi(s, dest)) {
2181 this->error(t, "integer is too large: " + s);
2182 return false;
2183 }
2184 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07002185}
2186
John Stilesf94348f2020-12-23 18:58:43 -05002187
ethannicholasb3058bd2016-07-01 08:22:01 -07002188/* FLOAT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002189bool Parser::floatLiteral(SKSL_FLOAT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002190 Token t;
John Stilesf94348f2020-12-23 18:58:43 -05002191 if (!this->expect(Token::Kind::TK_FLOAT_LITERAL, "float literal", &t)) {
2192 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07002193 }
John Stilesf94348f2020-12-23 18:58:43 -05002194 StringFragment s = this->text(t);
2195 if (!SkSL::stod(s, dest)) {
2196 this->error(t, "floating-point value is too large: " + s);
2197 return false;
2198 }
2199 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07002200}
2201
2202/* TRUE_LITERAL | FALSE_LITERAL */
2203bool Parser::boolLiteral(bool* dest) {
2204 Token t = this->nextToken();
2205 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002206 case Token::Kind::TK_TRUE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002207 *dest = true;
2208 return true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002209 case Token::Kind::TK_FALSE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002210 *dest = false;
2211 return true;
2212 default:
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002213 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002214 return false;
2215 }
2216}
2217
2218/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002219bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002220 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002221 if (this->expect(Token::Kind::TK_IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002222 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07002223 return true;
2224 }
2225 return false;
2226}
2227
John Stilesa6841be2020-08-06 14:11:56 -04002228} // namespace SkSL