blob: 5b74a1da3ea0001672eb0d4bb117164c8eaf518e [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_INLINE: return Modifiers::kInline_Flag;
John Stiles0dd1a772021-03-09 22:14:27 -050054 case Token::Kind::TK_NOINLINE: return Modifiers::kNoInline_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050055 default: return 0;
56 }
57}
58
ethannicholascad64162016-10-27 10:54:02 -070059class AutoDepth {
60public:
61 AutoDepth(Parser* p)
Ethan Nicholascf4deab2019-09-13 16:28:14 -040062 : fParser(p)
63 , fDepth(0) {}
ethannicholascad64162016-10-27 10:54:02 -070064
65 ~AutoDepth() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -040066 fParser->fDepth -= fDepth;
ethannicholascad64162016-10-27 10:54:02 -070067 }
68
Ethan Nicholascf4deab2019-09-13 16:28:14 -040069 bool increase() {
70 ++fDepth;
71 ++fParser->fDepth;
John Stiles8d056592020-11-10 10:03:50 -050072 if (fParser->fDepth > kMaxParseDepth) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070073 fParser->error(fParser->peek(), String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070074 return false;
75 }
76 return true;
77 }
78
79private:
80 Parser* fParser;
Ethan Nicholascf4deab2019-09-13 16:28:14 -040081 int fDepth;
ethannicholascad64162016-10-27 10:54:02 -070082};
83
Brian Salomon140f3da2018-08-23 13:51:27 +000084std::unordered_map<String, Parser::LayoutToken>* Parser::layoutTokens;
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040085
86void Parser::InitLayoutMap() {
Brian Salomon140f3da2018-08-23 13:51:27 +000087 layoutTokens = new std::unordered_map<String, LayoutToken>;
Brian Salomon23356442018-11-30 15:33:19 -050088 #define TOKEN(name, text) (*layoutTokens)[text] = LayoutToken::name
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040089 TOKEN(LOCATION, "location");
90 TOKEN(OFFSET, "offset");
91 TOKEN(BINDING, "binding");
92 TOKEN(INDEX, "index");
93 TOKEN(SET, "set");
94 TOKEN(BUILTIN, "builtin");
95 TOKEN(INPUT_ATTACHMENT_INDEX, "input_attachment_index");
96 TOKEN(ORIGIN_UPPER_LEFT, "origin_upper_left");
97 TOKEN(OVERRIDE_COVERAGE, "override_coverage");
Brian Osmandc4f8cd2021-02-25 15:30:46 -050098 TOKEN(EARLY_FRAGMENT_TESTS, "early_fragment_tests");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040099 TOKEN(BLEND_SUPPORT_ALL_EQUATIONS, "blend_support_all_equations");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400100 TOKEN(PUSH_CONSTANT, "push_constant");
101 TOKEN(POINTS, "points");
102 TOKEN(LINES, "lines");
103 TOKEN(LINE_STRIP, "line_strip");
104 TOKEN(LINES_ADJACENCY, "lines_adjacency");
105 TOKEN(TRIANGLES, "triangles");
106 TOKEN(TRIANGLE_STRIP, "triangle_strip");
107 TOKEN(TRIANGLES_ADJACENCY, "triangles_adjacency");
108 TOKEN(MAX_VERTICES, "max_vertices");
109 TOKEN(INVOCATIONS, "invocations");
Brian Osmanf59a9612020-04-15 14:18:13 -0400110 TOKEN(MARKER, "marker");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400111 TOKEN(WHEN, "when");
112 TOKEN(KEY, "key");
Michael Ludwiga4275592018-08-31 10:52:47 -0400113 TOKEN(TRACKED, "tracked");
Brian Osmanb32d66b2020-04-30 17:12:03 -0400114 TOKEN(SRGB_UNPREMUL, "srgb_unpremul");
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400115 TOKEN(CTYPE, "ctype");
Brian Osmanf28e55d2018-10-03 16:35:54 -0400116 TOKEN(SKPMCOLOR4F, "SkPMColor4f");
Mike Reedb26b4e72020-01-22 14:31:21 -0500117 TOKEN(SKV4, "SkV4");
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400118 TOKEN(SKRECT, "SkRect");
119 TOKEN(SKIRECT, "SkIRect");
120 TOKEN(SKPMCOLOR, "SkPMColor");
Mike Reedb26b4e72020-01-22 14:31:21 -0500121 TOKEN(SKM44, "SkM44");
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400122 TOKEN(BOOL, "bool");
123 TOKEN(INT, "int");
124 TOKEN(FLOAT, "float");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400125 #undef TOKEN
126}
127
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400128Parser::Parser(const char* text, size_t length, SymbolTable& symbols, ErrorReporter& errors)
John Stiles18a45b32021-03-29 18:07:32 -0400129: fText(text, length)
John Stiles7cbb09c22021-01-07 16:07:00 -0500130, fPushback(Token::Kind::TK_NONE, -1, -1)
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400131, fSymbols(symbols)
ethannicholasb3058bd2016-07-01 08:22:01 -0700132, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700133 fLexer.start(text, length);
Brian Salomon3b83afe2018-08-23 11:04:36 -0400134 static const bool layoutMapInitialized = []{ return (void)InitLayoutMap(), true; }();
135 (void) layoutMapInitialized;
ethannicholasb3058bd2016-07-01 08:22:01 -0700136}
137
John Stiles3b209362020-11-16 17:03:10 -0500138template <typename... Args>
139ASTNode::ID Parser::createNode(Args&&... args) {
140 ASTNode::ID result(fFile->fNodes.size());
141 fFile->fNodes.emplace_back(&fFile->fNodes, std::forward<Args>(args)...);
142 return result;
143}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400144
John Stiles3b209362020-11-16 17:03:10 -0500145ASTNode::ID Parser::addChild(ASTNode::ID target, ASTNode::ID child) {
146 fFile->fNodes[target.fValue].addChild(child);
147 return child;
148}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400149
John Stiles3b209362020-11-16 17:03:10 -0500150void Parser::createEmptyChild(ASTNode::ID target) {
151 ASTNode::ID child(fFile->fNodes.size());
152 fFile->fNodes.emplace_back();
153 fFile->fNodes[target.fValue].addChild(child);
154}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400155
Ethan Nicholasf7b88202017-09-18 14:10:39 -0400156/* (directive | section | declaration)* END_OF_FILE */
Ethan Nicholasba9a04f2020-11-06 09:28:04 -0500157std::unique_ptr<ASTFile> Parser::compilationUnit() {
John Stilesfbd050b2020-08-03 13:21:46 -0400158 fFile = std::make_unique<ASTFile>();
John Stiles18a45b32021-03-29 18:07:32 -0400159 fFile->fNodes.reserve(fText.size() / 10); // a typical program is approx 10:1 for chars:nodes
John Stiles3b209362020-11-16 17:03:10 -0500160 ASTNode::ID result = this->createNode(/*offset=*/0, ASTNode::Kind::kFile);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400161 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700162 for (;;) {
163 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400164 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400165 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400166 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400167 ASTNode::ID dir = this->directive();
168 if (fErrors.errorCount()) {
169 return nullptr;
170 }
171 if (dir) {
172 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700173 }
174 break;
175 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400176 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400177 ASTNode::ID section = this->section();
178 if (fErrors.errorCount()) {
179 return nullptr;
180 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400181 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400182 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400183 }
184 break;
185 }
John Stiles7cbb09c22021-01-07 16:07:00 -0500186 case Token::Kind::TK_INVALID: {
187 this->error(this->peek(), String("invalid token"));
188 return nullptr;
189 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700190 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400191 ASTNode::ID decl = this->declaration();
192 if (fErrors.errorCount()) {
193 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700194 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400195 if (decl) {
196 getNode(result).addChild(decl);
197 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700198 }
199 }
200 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400201 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700202}
203
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700204Token Parser::nextRawToken() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500205 if (fPushback.fKind != Token::Kind::TK_NONE) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700206 Token result = fPushback;
John Stiles7cbb09c22021-01-07 16:07:00 -0500207 fPushback.fKind = Token::Kind::TK_NONE;
ethannicholasb3058bd2016-07-01 08:22:01 -0700208 return result;
209 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700210 Token result = fLexer.next();
211 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400212}
213
214Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700215 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400216 while (token.fKind == Token::Kind::TK_WHITESPACE ||
217 token.fKind == Token::Kind::TK_LINE_COMMENT ||
218 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700219 token = this->nextRawToken();
220 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400221 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700222}
223
224void Parser::pushback(Token t) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500225 SkASSERT(fPushback.fKind == Token::Kind::TK_NONE);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400226 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700227}
228
229Token Parser::peek() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500230 if (fPushback.fKind == Token::Kind::TK_NONE) {
Brian Osman634624a2017-08-15 11:14:30 -0400231 fPushback = this->nextToken();
232 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700233 return fPushback;
234}
235
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400236bool Parser::checkNext(Token::Kind kind, Token* result) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500237 if (fPushback.fKind != Token::Kind::TK_NONE && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400238 return false;
239 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400240 Token next = this->nextToken();
241 if (next.fKind == kind) {
242 if (result) {
243 *result = next;
244 }
245 return true;
246 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400247 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400248 return false;
249}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500250
251bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700252 Token next = this->nextToken();
253 if (next.fKind == kind) {
254 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400255 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700256 }
257 return true;
258 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700259 this->error(next, "expected " + String(expected) + ", but found '" +
260 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700261 return false;
262 }
263}
264
John Stiles2630ea32020-12-04 10:51:21 -0500265bool Parser::expectIdentifier(Token* result) {
266 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", result)) {
267 return false;
268 }
269 if (this->isType(this->text(*result))) {
270 this->error(*result, "expected an identifier, but found type '" +
271 this->text(*result) + "'");
272 return false;
273 }
274 return true;
275}
276
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700277StringFragment Parser::text(Token token) {
John Stiles18a45b32021-03-29 18:07:32 -0400278 return StringFragment(fText.begin() + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500279}
280
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700281void Parser::error(Token token, String msg) {
282 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700283}
284
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700285void Parser::error(int offset, String msg) {
286 fErrors.error(offset, msg);
287}
288
289bool Parser::isType(StringFragment name) {
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400290 const Symbol* s = fSymbols[name];
John Stiles2630ea32020-12-04 10:51:21 -0500291 return s && s->is<Type>();
ethannicholasb3058bd2016-07-01 08:22:01 -0700292}
293
John Stiles80b02af2021-02-12 17:07:51 -0500294bool Parser::isArrayType(ASTNode::ID type) {
295 const ASTNode& node = this->getNode(type);
296 SkASSERT(node.fKind == ASTNode::Kind::kType);
297 return node.begin() != node.end();
298}
299
Ethan Nicholas11d53972016-11-28 11:23:23 -0500300/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700301 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400302ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700303 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400304 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400305 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700306 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700307 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400308 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700309 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500310 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400311 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700312 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400313 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400314 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700315 }
316 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400317 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400318 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700319 }
John Stiles3b209362020-11-16 17:03:10 -0500320 return this->createNode(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700321 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700322 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400323 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700324 }
325}
326
Ethan Nicholas762466e2017-06-29 10:03:38 -0400327/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
328 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400329ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400330 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400331 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400332 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400333 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400334 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400335 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400336 this->nextToken();
337 Token argToken;
John Stiles2630ea32020-12-04 10:51:21 -0500338 if (!this->expectIdentifier(&argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400339 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400340 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700341 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400342 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400343 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400344 }
345 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400346 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400347 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400348 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400349 StringFragment text;
350 Token codeStart = this->nextRawToken();
351 size_t startOffset = codeStart.fOffset;
352 this->pushback(codeStart);
John Stiles18a45b32021-03-29 18:07:32 -0400353 text.fChars = fText.begin() + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400354 int level = 1;
355 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700356 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400357 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400358 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400359 ++level;
360 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400361 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400362 --level;
363 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400364 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700365 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400366 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400367 default:
368 break;
369 }
370 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400371 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400372 break;
373 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400374 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700375 StringFragment name = this->text(start);
376 ++name.fChars;
377 --name.fLength;
John Stiles3b209362020-11-16 17:03:10 -0500378 return this->createNode(start.fOffset, ASTNode::Kind::kSection,
379 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400380}
381
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500382/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
383 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400384ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500385 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400386 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400387 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500388 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400389 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400390 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500391 }
392 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500393 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400394 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500395 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400396 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400397 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500398 }
John Stilese6c67c52021-01-15 12:01:00 -0500399 fSymbols.add(Type::MakeEnumType(this->text(name)));
John Stiles3b209362020-11-16 17:03:10 -0500400 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400401 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500402 Token id;
John Stiles2630ea32020-12-04 10:51:21 -0500403 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400404 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500405 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400406 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400407 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500408 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400409 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500410 }
John Stiles3b209362020-11-16 17:03:10 -0500411 ASTNode::ID child = this->addChild(
412 result, this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400413 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500414 } else {
John Stiles3b209362020-11-16 17:03:10 -0500415 this->addChild(result,
416 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500417 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400418 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
419 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400420 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500421 }
John Stiles2630ea32020-12-04 10:51:21 -0500422 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400423 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500424 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400425 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400426 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500427 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400428 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500429 }
John Stiles3b209362020-11-16 17:03:10 -0500430 ASTNode::ID child = this->addChild(
431 result,
432 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400433 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500434 } else {
John Stiles3b209362020-11-16 17:03:10 -0500435 this->addChild(
436 result,
437 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500438 }
439 }
440 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400441 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400442 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500443}
444
445/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700446 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400447ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700448 Token lookahead = this->peek();
Ethan Nicholasda6320c2020-09-02 14:08:23 -0400449 switch (lookahead.fKind) {
450 case Token::Kind::TK_ENUM:
451 return this->enumDeclaration();
452 case Token::Kind::TK_SEMICOLON:
453 this->error(lookahead.fOffset, "expected a declaration, but found ';'");
454 return ASTNode::ID::Invalid();
455 default:
456 break;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500457 }
458 Modifiers modifiers = this->modifiers();
459 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400460 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700461 // we have an identifier that's not a type, could be the start of an interface block
462 return this->interfaceBlock(modifiers);
463 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400464 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700465 return this->structVarDeclaration(modifiers);
466 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400467 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700468 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -0500469 return this->createNode(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700470 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400471 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700472 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400473 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700474 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700475 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500476 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400477 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700478 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400479 if (this->checkNext(Token::Kind::TK_LPAREN)) {
John Stiles3b209362020-11-16 17:03:10 -0500480 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kFunction);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400481 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
482 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400483 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400484 for (;;) {
485 ASTNode::ID parameter = this->parameter();
486 if (!parameter) {
487 return ASTNode::ID::Invalid();
488 }
489 ++fd.fParameterCount;
490 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400491 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400492 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700493 }
494 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700495 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400496 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400497 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400498 return ASTNode::ID::Invalid();
499 }
500 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400501 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700502 body = this->block();
503 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400504 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700505 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400506 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700507 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400508 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700509 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400510 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700511 }
512}
513
John Stiles76389b72021-01-25 13:49:05 -0500514/* (varDeclarations | expressionStatement) */
515ASTNode::ID Parser::varDeclarationsOrExpressionStatement() {
516 if (this->isType(this->text(this->peek()))) {
517 // Statements that begin with a typename are most often variable declarations, but
518 // occasionally the type is part of a constructor, and these are actually expression-
519 // statements in disguise. First, attempt the common case: parse it as a vardecl.
520 Checkpoint checkpoint(this);
John Stiles8dabeac2021-02-12 16:05:00 -0500521 VarDeclarationsPrefix prefix;
522 if (this->varDeclarationsPrefix(&prefix)) {
523 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
John Stiles76389b72021-01-25 13:49:05 -0500524 }
525
526 // If this statement wasn't actually a vardecl after all, rewind and try parsing it as an
527 // expression-statement instead.
528 checkpoint.rewind();
529 }
530
531 return this->expressionStatement();
532}
533
John Stiles8dabeac2021-02-12 16:05:00 -0500534// Helper function for varDeclarations(). If this function succeeds, we assume that the rest of the
535// statement is a variable-declaration statement, not an expression-statement.
536bool Parser::varDeclarationsPrefix(VarDeclarationsPrefix* prefixData) {
537 prefixData->modifiers = this->modifiers();
538 prefixData->type = this->type();
539 if (!prefixData->type) {
540 return false;
541 }
542 return this->expectIdentifier(&prefixData->name);
543}
544
ethannicholasb3058bd2016-07-01 08:22:01 -0700545/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400546ASTNode::ID Parser::varDeclarations() {
John Stiles8dabeac2021-02-12 16:05:00 -0500547 VarDeclarationsPrefix prefix;
548 if (!this->varDeclarationsPrefix(&prefix)) {
John Stiles76389b72021-01-25 13:49:05 -0500549 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700550 }
John Stiles8dabeac2021-02-12 16:05:00 -0500551 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700552}
553
554/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400555ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400556 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400557 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700558 }
559 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500560 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400561 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700562 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400563 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400564 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700565 }
566 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400567 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400568 ASTNode::ID decls = this->varDeclarations();
569 if (!decls) {
570 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700571 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400572 ASTNode& declsNode = getNode(decls);
John Stilesece1bf02021-03-08 11:15:55 -0500573 const Modifiers& modifiers = declsNode.begin()->getModifiers();
John Stiles13fc2602020-10-09 17:42:31 -0400574 if (modifiers.fFlags != Modifiers::kNo_Flag) {
575 String desc = modifiers.description();
576 desc.pop_back(); // remove trailing space
577 this->error(declsNode.fOffset,
578 "modifier '" + desc + "' is not permitted on a struct field");
579 }
580
Brian Osman00fea5b2021-01-28 09:46:30 -0500581 const Symbol* symbol = fSymbols[(declsNode.begin() + 1)->getString()];
John Stiles13fc2602020-10-09 17:42:31 -0400582 SkASSERT(symbol);
583 const Type* type = &symbol->as<Type>();
John Stiles1d757782020-11-17 09:43:22 -0500584 if (type->isOpaque()) {
585 this->error(declsNode.fOffset,
586 "opaque type '" + type->name() + "' is not permitted in a struct");
587 }
John Stiles13fc2602020-10-09 17:42:31 -0400588
Ethan Nicholasfc994162019-06-06 10:04:27 -0400589 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
590 ASTNode& var = *iter;
John Stilesece1bf02021-03-08 11:15:55 -0500591 const ASTNode::VarData& vd = var.getVarData();
John Stiles6bef6a72020-12-02 14:26:04 -0500592
John Stilesd39aec92020-12-03 14:37:16 -0500593 // Read array size if one is present.
594 if (vd.fIsArray) {
595 const ASTNode& size = *var.begin();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400596 if (!size || size.fKind != ASTNode::Kind::kInt) {
597 this->error(declsNode.fOffset, "array size in struct field must be a constant");
598 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700599 }
John Stiles6bef6a72020-12-02 14:26:04 -0500600 if (size.getInt() <= 0 || size.getInt() > INT_MAX) {
601 this->error(declsNode.fOffset, "array size is invalid");
602 return ASTNode::ID::Invalid();
603 }
John Stilesd39aec92020-12-03 14:37:16 -0500604 // Add the array dimensions to our type.
605 int arraySize = size.getInt();
John Stilesa2179502020-12-03 14:37:57 -0500606 type = fSymbols.addArrayDimension(type, arraySize);
ethannicholasb3058bd2016-07-01 08:22:01 -0700607 }
John Stiles13fc2602020-10-09 17:42:31 -0400608
609 fields.push_back(Type::Field(modifiers, vd.fName, type));
John Stilesd39aec92020-12-03 14:37:16 -0500610 if (vd.fIsArray ? var.begin()->fNext : var.fFirstChild) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400611 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700612 }
613 }
614 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400615 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400616 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700617 }
Brian Osman6c3b23f2021-02-12 13:09:58 -0500618 if (fields.empty()) {
619 this->error(name.fOffset,
620 "struct '" + this->text(name) + "' must contain at least one field");
621 return ASTNode::ID::Invalid();
622 }
John Stilesad2d4942020-12-11 16:55:58 -0500623 std::unique_ptr<Type> newType = Type::MakeStructType(name.fOffset, this->text(name), fields);
John Stilesa695d622020-11-10 10:04:26 -0500624 if (struct_is_too_deeply_nested(*newType, kMaxStructDepth)) {
625 this->error(name.fOffset, "struct '" + this->text(name) + "' is too deeply nested");
626 return ASTNode::ID::Invalid();
627 }
628 fSymbols.add(std::move(newType));
Brian Osman00fea5b2021-01-28 09:46:30 -0500629 return this->createNode(name.fOffset, ASTNode::Kind::kType, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700630}
631
632/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400633ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
634 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700635 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400636 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700637 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400638 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400639 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400640 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700641 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400642 this->expect(Token::Kind::TK_SEMICOLON, "';'");
John Stilesdc75a972020-11-25 16:24:55 -0500643 return type;
ethannicholasb3058bd2016-07-01 08:22:01 -0700644}
645
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400646/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
647 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400648ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
John Stiles08070f62020-11-17 10:45:49 -0500649 int offset = this->peek().fOffset;
650 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kVarDeclarations);
651 this->addChild(result, this->createNode(offset, ASTNode::Kind::kModifiers, mods));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400652 getNode(result).addChild(type);
John Stilesd39aec92020-12-03 14:37:16 -0500653
John Stiles80b02af2021-02-12 17:07:51 -0500654 auto parseArrayDimensions = [&](ASTNode::ID currentVar, ASTNode::VarData* vd) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400655 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500656 if (vd->fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500657 this->error(this->peek(), "multi-dimensional arrays are not supported");
658 return false;
659 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400660 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
John Stiles3b209362020-11-16 17:03:10 -0500661 this->createEmptyChild(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700662 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400663 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700664 if (!size) {
John Stilesd39aec92020-12-03 14:37:16 -0500665 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700666 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400667 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400668 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
John Stilesd39aec92020-12-03 14:37:16 -0500669 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700670 }
671 }
John Stilesd39aec92020-12-03 14:37:16 -0500672 vd->fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700673 }
John Stilesd39aec92020-12-03 14:37:16 -0500674 return true;
675 };
676
677 auto parseInitializer = [this](ASTNode::ID currentVar) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400678 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400679 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700680 if (!value) {
John Stilesd39aec92020-12-03 14:37:16 -0500681 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700682 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400683 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700684 }
John Stilesd39aec92020-12-03 14:37:16 -0500685 return true;
686 };
687
688 ASTNode::ID currentVar = this->createNode(offset, ASTNode::Kind::kVarDeclaration);
689 ASTNode::VarData vd{name, /*isArray=*/false};
690
691 getNode(result).addChild(currentVar);
692 if (!parseArrayDimensions(currentVar, &vd)) {
693 return ASTNode::ID::Invalid();
694 }
695 getNode(currentVar).setVarData(vd);
696 if (!parseInitializer(currentVar)) {
697 return ASTNode::ID::Invalid();
698 }
699
700 while (this->checkNext(Token::Kind::TK_COMMA)) {
701 Token identifierName;
John Stiles2630ea32020-12-04 10:51:21 -0500702 if (!this->expectIdentifier(&identifierName)) {
John Stilesd39aec92020-12-03 14:37:16 -0500703 return ASTNode::ID::Invalid();
704 }
705
706 currentVar = ASTNode::ID(fFile->fNodes.size());
707 vd = ASTNode::VarData{this->text(identifierName), /*isArray=*/false};
708 fFile->fNodes.emplace_back(&fFile->fNodes, offset, ASTNode::Kind::kVarDeclaration);
709
710 getNode(result).addChild(currentVar);
711 if (!parseArrayDimensions(currentVar, &vd)) {
712 return ASTNode::ID::Invalid();
713 }
714 getNode(currentVar).setVarData(vd);
715 if (!parseInitializer(currentVar)) {
716 return ASTNode::ID::Invalid();
717 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700718 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400719 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400720 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700721 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400722 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700723}
724
725/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400726ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400727 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400728 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700729 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400730 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700731 }
732 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500733 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400734 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700735 }
John Stiles3b209362020-11-16 17:03:10 -0500736 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kParameter);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400737 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
738 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400739 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500740 if (pd.fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500741 this->error(this->peek(), "multi-dimensional arrays are not supported");
742 return ASTNode::ID::Invalid();
743 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700744 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400745 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400746 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700747 }
John Stilesf94348f2020-12-23 18:58:43 -0500748 StringFragment arraySizeFrag = this->text(sizeToken);
749 SKSL_INT arraySize;
750 if (!SkSL::stoi(arraySizeFrag, &arraySize)) {
751 this->error(sizeToken, "array size is too large: " + arraySizeFrag);
752 return ASTNode::ID::Invalid();
753 }
754 this->addChild(result, this->createNode(sizeToken.fOffset, ASTNode::Kind::kInt, arraySize));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400755 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400756 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700757 }
John Stilesd39aec92020-12-03 14:37:16 -0500758 pd.fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700759 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400760 getNode(result).setParameterData(pd);
761 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700762}
763
Ethan Nicholasd608c092017-10-26 09:30:08 -0400764/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700765int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400766 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700767 return -1;
768 }
769 Token resultToken;
John Stilesf94348f2020-12-23 18:58:43 -0500770 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
771 return -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700772 }
John Stilesf94348f2020-12-23 18:58:43 -0500773 StringFragment resultFrag = this->text(resultToken);
774 SKSL_INT resultValue;
775 if (!SkSL::stoi(resultFrag, &resultValue)) {
776 this->error(resultToken, "value in layout is too large: " + resultFrag);
777 return -1;
778 }
779 return resultValue;
ethannicholasb3058bd2016-07-01 08:22:01 -0700780}
781
Ethan Nicholasd608c092017-10-26 09:30:08 -0400782/** EQ IDENTIFIER */
783StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400784 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400785 return StringFragment();
786 }
787 Token resultToken;
John Stiles2630ea32020-12-04 10:51:21 -0500788 if (!this->expectIdentifier(&resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400789 return StringFragment();
790 }
791 return this->text(resultToken);
792}
793
794
Ethan Nicholas762466e2017-06-29 10:03:38 -0400795/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400796StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400797 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400798 return "";
799 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700800 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400801 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400802 StringFragment code;
John Stiles18a45b32021-03-29 18:07:32 -0400803 code.fChars = fText.begin() + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400804 int level = 1;
805 bool done = false;
806 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700807 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400808 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400809 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400810 ++level;
811 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400812 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400813 --level;
814 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400815 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400816 if (level == 1) {
817 done = true;
818 }
819 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400820 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700821 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400822 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400823 default:
824 break;
825 }
826 if (!level) {
827 done = true;
828 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400829 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400830 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400831 this->pushback(std::move(next));
832 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400833 }
834 return code;
835}
836
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400837Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400838 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400839 Token t = this->nextToken();
840 String text = this->text(t);
841 auto found = layoutTokens->find(text);
842 if (found != layoutTokens->end()) {
843 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400844 case LayoutToken::SKPMCOLOR4F:
845 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500846 case LayoutToken::SKV4:
847 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400848 case LayoutToken::SKRECT:
849 return Layout::CType::kSkRect;
850 case LayoutToken::SKIRECT:
851 return Layout::CType::kSkIRect;
852 case LayoutToken::SKPMCOLOR:
853 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400854 case LayoutToken::BOOL:
855 return Layout::CType::kBool;
856 case LayoutToken::INT:
857 return Layout::CType::kInt32;
858 case LayoutToken::FLOAT:
859 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500860 case LayoutToken::SKM44:
861 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400862 default:
863 break;
864 }
865 }
866 this->error(t, "unsupported ctype");
867 }
868 return Layout::CType::kDefault;
869}
870
ethannicholas8ac838d2016-11-22 08:39:36 -0800871/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500872Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500873 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700874 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500875 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700876 int binding = -1;
877 int index = -1;
878 int set = -1;
879 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500880 int inputAttachmentIndex = -1;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500881 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
882 int maxVertices = -1;
883 int invocations = -1;
Brian Osmanf59a9612020-04-15 14:18:13 -0400884 StringFragment marker;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400885 StringFragment when;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400886 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400887 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
888 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500889 return Layout(flags, location, offset, binding, index, set, builtin,
Brian Osman4717fbb2021-02-23 13:12:09 -0500890 inputAttachmentIndex, primitive, maxVertices, invocations, marker, when,
Brian Osmanba7ef322021-02-23 13:37:22 -0500891 ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700892 }
893 for (;;) {
894 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700895 String text = this->text(t);
Brian Osmana77ed8b2021-02-23 12:54:22 -0500896 auto setFlag = [&](Layout::Flag f) {
897 if (flags & f) {
898 this->error(t, "layout qualifier '" + text + "' appears more than once");
899 }
900 flags |= f;
901 };
902 auto setPrimitive = [&](Layout::Primitive p) {
903 if (flags & Layout::kPrimitive_Flag) {
904 this->error(t, "only one primitive-type layout qualifier is allowed");
905 }
906 flags |= Layout::kPrimitive_Flag;
907 primitive = p;
908 };
909
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400910 auto found = layoutTokens->find(text);
911 if (found != layoutTokens->end()) {
912 switch (found->second) {
Brian Osmana77ed8b2021-02-23 12:54:22 -0500913 case LayoutToken::ORIGIN_UPPER_LEFT:
914 setFlag(Layout::kOriginUpperLeft_Flag);
915 break;
916 case LayoutToken::OVERRIDE_COVERAGE:
917 setFlag(Layout::kOverrideCoverage_Flag);
918 break;
Brian Osmandc4f8cd2021-02-25 15:30:46 -0500919 case LayoutToken::EARLY_FRAGMENT_TESTS:
920 setFlag(Layout::kEarlyFragmentTests_Flag);
921 break;
Brian Osmana77ed8b2021-02-23 12:54:22 -0500922 case LayoutToken::PUSH_CONSTANT:
923 setFlag(Layout::kPushConstant_Flag);
924 break;
925 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
926 setFlag(Layout::kBlendSupportAllEquations_Flag);
927 break;
928 case LayoutToken::TRACKED:
929 setFlag(Layout::kTracked_Flag);
930 break;
931 case LayoutToken::SRGB_UNPREMUL:
932 setFlag(Layout::kSRGBUnpremul_Flag);
933 break;
934 case LayoutToken::KEY:
935 setFlag(Layout::kKey_Flag);
936 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700937 case LayoutToken::LOCATION:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500938 setFlag(Layout::kLocation_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500939 location = this->layoutInt();
940 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700941 case LayoutToken::OFFSET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500942 setFlag(Layout::kOffset_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500943 offset = this->layoutInt();
944 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700945 case LayoutToken::BINDING:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500946 setFlag(Layout::kBinding_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500947 binding = this->layoutInt();
948 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700949 case LayoutToken::INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500950 setFlag(Layout::kIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500951 index = this->layoutInt();
952 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700953 case LayoutToken::SET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500954 setFlag(Layout::kSet_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500955 set = this->layoutInt();
956 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700957 case LayoutToken::BUILTIN:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500958 setFlag(Layout::kBuiltin_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500959 builtin = this->layoutInt();
960 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700961 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500962 setFlag(Layout::kInputAttachmentIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500963 inputAttachmentIndex = this->layoutInt();
964 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700965 case LayoutToken::POINTS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500966 setPrimitive(Layout::kPoints_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500967 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700968 case LayoutToken::LINES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500969 setPrimitive(Layout::kLines_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500970 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700971 case LayoutToken::LINE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500972 setPrimitive(Layout::kLineStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500973 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700974 case LayoutToken::LINES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500975 setPrimitive(Layout::kLinesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500976 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700977 case LayoutToken::TRIANGLES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500978 setPrimitive(Layout::kTriangles_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500979 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700980 case LayoutToken::TRIANGLE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500981 setPrimitive(Layout::kTriangleStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500982 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700983 case LayoutToken::TRIANGLES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500984 setPrimitive(Layout::kTrianglesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500985 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700986 case LayoutToken::MAX_VERTICES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500987 setFlag(Layout::kMaxVertices_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500988 maxVertices = this->layoutInt();
989 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700990 case LayoutToken::INVOCATIONS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500991 setFlag(Layout::kInvocations_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500992 invocations = this->layoutInt();
993 break;
Brian Osmanf59a9612020-04-15 14:18:13 -0400994 case LayoutToken::MARKER:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500995 setFlag(Layout::kMarker_Flag);
Brian Osmanf59a9612020-04-15 14:18:13 -0400996 marker = this->layoutCode();
997 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700998 case LayoutToken::WHEN:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500999 setFlag(Layout::kWhen_Flag);
Ethan Nicholas762466e2017-06-29 10:03:38 -04001000 when = this->layoutCode();
1001 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -04001002 case LayoutToken::CTYPE:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001003 setFlag(Layout::kCType_Flag);
Ethan Nicholas78aceb22018-08-31 16:13:58 -04001004 ctype = this->layoutCType();
1005 break;
1006 default:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001007 this->error(t, "'" + text + "' is not a valid layout qualifier");
Ethan Nicholasd608c092017-10-26 09:30:08 -04001008 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -05001009 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001010 } else {
Brian Osmana77ed8b2021-02-23 12:54:22 -05001011 this->error(t, "'" + text + "' is not a valid layout qualifier");
ethannicholasb3058bd2016-07-01 08:22:01 -07001012 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001013 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001014 break;
1015 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001016 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001017 break;
1018 }
1019 }
1020 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -05001021 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
Brian Osmanba7ef322021-02-23 13:37:22 -05001022 primitive, maxVertices, invocations, marker, when, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -07001023}
1024
Brian Osmane49703f2021-04-19 11:15:24 -04001025/* layout? (UNIFORM | CONST | IN | OUT | INOUT | FLAT | NOPERSPECTIVE | INLINE)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -05001026Modifiers Parser::modifiers() {
1027 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -07001028 int flags = 0;
1029 for (;;) {
1030 // TODO: handle duplicate / incompatible flags
John Stiles7d5b90a2021-01-21 14:26:51 -05001031 int tokenFlag = parse_modifier_token(peek().fKind);
1032 if (!tokenFlag) {
1033 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001034 }
John Stiles7d5b90a2021-01-21 14:26:51 -05001035 flags |= tokenFlag;
1036 this->nextToken();
ethannicholasb3058bd2016-07-01 08:22:01 -07001037 }
John Stiles7d5b90a2021-01-21 14:26:51 -05001038 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001039}
1040
Ethan Nicholas11d53972016-11-28 11:23:23 -05001041Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
1042 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -07001043 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001044 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001045 }
1046 return result;
1047}
1048
1049/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001050ASTNode::ID Parser::statement() {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001051 Token start = this->nextToken();
1052 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001053 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001054 return ASTNode::ID::Invalid();
1055 }
1056 this->pushback(start);
ethannicholasb3058bd2016-07-01 08:22:01 -07001057 switch (start.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001058 case Token::Kind::TK_IF: // fall through
1059 case Token::Kind::TK_STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -07001060 return this->ifStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001061 case Token::Kind::TK_FOR:
ethannicholasb3058bd2016-07-01 08:22:01 -07001062 return this->forStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001063 case Token::Kind::TK_DO:
ethannicholasb3058bd2016-07-01 08:22:01 -07001064 return this->doStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001065 case Token::Kind::TK_WHILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001066 return this->whileStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001067 case Token::Kind::TK_SWITCH: // fall through
1068 case Token::Kind::TK_STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -05001069 return this->switchStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001070 case Token::Kind::TK_RETURN:
ethannicholasb3058bd2016-07-01 08:22:01 -07001071 return this->returnStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001072 case Token::Kind::TK_BREAK:
ethannicholasb3058bd2016-07-01 08:22:01 -07001073 return this->breakStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001074 case Token::Kind::TK_CONTINUE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001075 return this->continueStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001076 case Token::Kind::TK_DISCARD:
ethannicholasb3058bd2016-07-01 08:22:01 -07001077 return this->discardStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001078 case Token::Kind::TK_LBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001079 return this->block();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001080 case Token::Kind::TK_SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001081 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -05001082 return this->createNode(start.fOffset, ASTNode::Kind::kBlock);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001083 case Token::Kind::TK_CONST:
Ethan Nicholasfc994162019-06-06 10:04:27 -04001084 return this->varDeclarations();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001085 case Token::Kind::TK_IDENTIFIER:
John Stiles76389b72021-01-25 13:49:05 -05001086 return this->varDeclarationsOrExpressionStatement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001087 default:
1088 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001089 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001090}
1091
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001092/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* QUESTION? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001093ASTNode::ID Parser::type() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001094 Token type;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001095 if (!this->expect(Token::Kind::TK_IDENTIFIER, "a type", &type)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001096 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001097 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001098 if (!this->isType(this->text(type))) {
1099 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001100 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001101 }
Brian Osman00fea5b2021-01-28 09:46:30 -05001102 ASTNode::ID result = this->createNode(type.fOffset, ASTNode::Kind::kType, this->text(type));
John Stilesd39aec92020-12-03 14:37:16 -05001103 bool isArray = false;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001104 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stilesd39aec92020-12-03 14:37:16 -05001105 if (isArray) {
1106 this->error(this->peek(), "multi-dimensional arrays are not supported");
1107 return ASTNode::ID::Invalid();
1108 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001109 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001110 SKSL_INT i;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001111 if (this->intLiteral(&i)) {
John Stiles08070f62020-11-17 10:45:49 -05001112 this->addChild(result, this->createNode(this->peek().fOffset,
1113 ASTNode::Kind::kInt, i));
Ethan Nicholas50afc172017-02-16 14:49:57 -05001114 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001115 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001116 }
1117 } else {
John Stiles3b209362020-11-16 17:03:10 -05001118 this->createEmptyChild(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001119 }
John Stilesd39aec92020-12-03 14:37:16 -05001120 isArray = true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001121 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001122 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001123 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001124}
1125
John Stiles57a996b2020-04-19 19:05:12 -07001126/* IDENTIFIER LBRACE
1127 varDeclaration+
1128 RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001129ASTNode::ID Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001130 Token name;
John Stiles2630ea32020-12-04 10:51:21 -05001131 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001132 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001133 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001134 if (peek().fKind != Token::Kind::TK_LBRACE) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001135 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001136 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -07001137 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001138 this->error(name, "no type named '" + this->text(name) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001139 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001140 }
John Stiles3b209362020-11-16 17:03:10 -05001141 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kInterfaceBlock);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001142 ASTNode::InterfaceBlockData id(mods, this->text(name), 0, "", 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07001143 this->nextToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001144 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001145 ASTNode::ID decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001146 if (!decl) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001147 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001148 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001149 getNode(result).addChild(decl);
1150 ++id.fDeclarationCount;
ethannicholasb3058bd2016-07-01 08:22:01 -07001151 }
John Stiles57a996b2020-04-19 19:05:12 -07001152 if (id.fDeclarationCount == 0) {
1153 this->error(name, "interface block '" + this->text(name) +
1154 "' must contain at least one member");
1155 return ASTNode::ID::Invalid();
1156 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001157 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001158 std::vector<ASTNode> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001159 StringFragment instanceName;
1160 Token instanceNameToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001161 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &instanceNameToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001162 id.fInstanceName = this->text(instanceNameToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001163 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stilesd39aec92020-12-03 14:37:16 -05001164 if (id.fIsArray) {
1165 this->error(this->peek(), "multi-dimensional arrays are not supported");
1166 return false;
1167 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001168 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001169 ASTNode::ID size = this->expression();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001170 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001171 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001172 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001173 getNode(result).addChild(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001174 } else {
John Stiles3b209362020-11-16 17:03:10 -05001175 this->createEmptyChild(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001176 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001177 this->expect(Token::Kind::TK_RBRACKET, "']'");
John Stilesd39aec92020-12-03 14:37:16 -05001178 id.fIsArray = true;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001179 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001180 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -07001181 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001182 getNode(result).setInterfaceBlockData(id);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001183 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001184 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001185}
1186
1187/* IF LPAREN expression RPAREN statement (ELSE statement)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001188ASTNode::ID Parser::ifStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001189 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001190 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_IF, &start);
1191 if (!isStatic && !this->expect(Token::Kind::TK_IF, "'if'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001192 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001193 }
John Stiles3b209362020-11-16 17:03:10 -05001194 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kIf, isStatic);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001195 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001196 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001197 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001198 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001199 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001200 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001201 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001202 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001203 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001204 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001205 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001206 ASTNode::ID ifTrue = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001207 if (!ifTrue) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001208 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001209 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001210 getNode(result).addChild(ifTrue);
1211 ASTNode::ID ifFalse;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001212 if (this->checkNext(Token::Kind::TK_ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001213 ifFalse = this->statement();
1214 if (!ifFalse) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001215 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001216 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001217 getNode(result).addChild(ifFalse);
ethannicholasb3058bd2016-07-01 08:22:01 -07001218 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001219 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001220}
1221
1222/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001223ASTNode::ID Parser::doStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001224 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001225 if (!this->expect(Token::Kind::TK_DO, "'do'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001226 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001227 }
John Stiles3b209362020-11-16 17:03:10 -05001228 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kDo);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001229 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001230 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001231 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001232 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001233 getNode(result).addChild(statement);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001234 if (!this->expect(Token::Kind::TK_WHILE, "'while'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001235 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001236 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001237 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001238 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001239 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001240 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001241 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001242 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001243 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001244 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001245 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001246 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001247 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001248 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001249 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001250 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001251 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001252}
1253
1254/* WHILE LPAREN expression RPAREN STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001255ASTNode::ID Parser::whileStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001256 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001257 if (!this->expect(Token::Kind::TK_WHILE, "'while'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001258 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001259 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001260 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001261 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001262 }
John Stiles3b209362020-11-16 17:03:10 -05001263 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kWhile);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001264 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001265 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001266 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001267 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001268 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001269 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001270 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001271 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001272 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001273 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001274 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001275 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001276 getNode(result).addChild(statement);
1277 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001278}
1279
Ethan Nicholasaf197692017-02-27 13:26:45 -05001280/* CASE expression COLON statement* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001281ASTNode::ID Parser::switchCase() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001282 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001283 if (!this->expect(Token::Kind::TK_CASE, "'case'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001284 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001285 }
John Stiles3b209362020-11-16 17:03:10 -05001286 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kSwitchCase);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001287 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001288 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001289 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001290 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001291 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001292 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001293 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001294 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001295 while (this->peek().fKind != Token::Kind::TK_RBRACE &&
1296 this->peek().fKind != Token::Kind::TK_CASE &&
1297 this->peek().fKind != Token::Kind::TK_DEFAULT) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001298 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001299 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001300 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001301 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001302 getNode(result).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001303 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001304 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001305}
1306
1307/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001308ASTNode::ID Parser::switchStatement() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001309 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001310 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_SWITCH, &start);
1311 if (!isStatic && !this->expect(Token::Kind::TK_SWITCH, "'switch'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001312 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001313 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001314 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001315 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001316 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001317 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001318 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001319 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001320 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001321 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001322 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001323 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001324 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001325 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001326 }
John Stiles3b209362020-11-16 17:03:10 -05001327 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kSwitch, isStatic);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001328 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001329 while (this->peek().fKind == Token::Kind::TK_CASE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001330 ASTNode::ID c = this->switchCase();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001331 if (!c) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001332 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001333 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001334 getNode(result).addChild(c);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001335 }
1336 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1337 // parts of the compiler may rely upon this assumption.
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001338 if (this->peek().fKind == Token::Kind::TK_DEFAULT) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001339 Token defaultStart;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001340 SkAssertResult(this->expect(Token::Kind::TK_DEFAULT, "'default'", &defaultStart));
1341 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001342 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001343 }
John Stiles3b209362020-11-16 17:03:10 -05001344 ASTNode::ID defaultCase = this->addChild(
1345 result, this->createNode(defaultStart.fOffset, ASTNode::Kind::kSwitchCase));
1346 this->createEmptyChild(defaultCase); // empty test to signify default case
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001347 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001348 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001349 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001350 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001351 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001352 getNode(defaultCase).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001353 }
Ethan Nicholasaf197692017-02-27 13:26:45 -05001354 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001355 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001356 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001357 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001358 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001359}
1360
Ethan Nicholas11d53972016-11-28 11:23:23 -05001361/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001362 STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001363ASTNode::ID Parser::forStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001364 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001365 if (!this->expect(Token::Kind::TK_FOR, "'for'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001366 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001367 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001368 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001369 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001370 }
John Stiles3b209362020-11-16 17:03:10 -05001371 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kFor);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001372 ASTNode::ID initializer;
ethannicholasb3058bd2016-07-01 08:22:01 -07001373 Token nextToken = this->peek();
1374 switch (nextToken.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001375 case Token::Kind::TK_SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001376 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -05001377 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001378 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001379 case Token::Kind::TK_CONST: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001380 initializer = this->varDeclarations();
1381 if (!initializer) {
1382 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001383 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001384 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001385 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001386 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001387 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001388 if (this->isType(this->text(nextToken))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001389 initializer = this->varDeclarations();
1390 if (!initializer) {
1391 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001392 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001393 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001394 break;
1395 }
John Stiles30212b72020-06-11 17:55:07 -04001396 [[fallthrough]];
1397 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001398 default:
1399 initializer = this->expressionStatement();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001400 if (!initializer) {
1401 return ASTNode::ID::Invalid();
1402 }
1403 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001404 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001405 ASTNode::ID test;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001406 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001407 test = this->expression();
1408 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001409 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001410 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001411 getNode(result).addChild(test);
1412 } else {
John Stiles3b209362020-11-16 17:03:10 -05001413 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001414 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001415 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001416 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001417 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001418 ASTNode::ID next;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001419 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001420 next = this->expression();
1421 if (!next) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001422 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001423 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001424 getNode(result).addChild(next);
1425 } else {
John Stiles3b209362020-11-16 17:03:10 -05001426 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001427 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001428 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001429 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001430 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001431 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001432 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001433 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001434 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001435 getNode(result).addChild(statement);
1436 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001437}
1438
1439/* RETURN expression? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001440ASTNode::ID Parser::returnStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001441 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001442 if (!this->expect(Token::Kind::TK_RETURN, "'return'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001443 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001444 }
John Stiles3b209362020-11-16 17:03:10 -05001445 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kReturn);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001446 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001447 ASTNode::ID expression = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001448 if (!expression) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001449 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001450 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001451 getNode(result).addChild(expression);
ethannicholasb3058bd2016-07-01 08:22:01 -07001452 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001453 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001454 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001455 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001456 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001457}
1458
1459/* BREAK SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001460ASTNode::ID Parser::breakStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001461 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001462 if (!this->expect(Token::Kind::TK_BREAK, "'break'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001463 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001464 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001465 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001466 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001467 }
John Stiles3b209362020-11-16 17:03:10 -05001468 return this->createNode(start.fOffset, ASTNode::Kind::kBreak);
ethannicholasb3058bd2016-07-01 08:22:01 -07001469}
1470
1471/* CONTINUE SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001472ASTNode::ID Parser::continueStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001473 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001474 if (!this->expect(Token::Kind::TK_CONTINUE, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001475 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001476 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001477 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001478 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001479 }
John Stiles3b209362020-11-16 17:03:10 -05001480 return this->createNode(start.fOffset, ASTNode::Kind::kContinue);
ethannicholasb3058bd2016-07-01 08:22:01 -07001481}
1482
1483/* DISCARD SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001484ASTNode::ID Parser::discardStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001485 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001486 if (!this->expect(Token::Kind::TK_DISCARD, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001487 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001488 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001489 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001490 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001491 }
John Stiles3b209362020-11-16 17:03:10 -05001492 return this->createNode(start.fOffset, ASTNode::Kind::kDiscard);
ethannicholasb3058bd2016-07-01 08:22:01 -07001493}
1494
1495/* LBRACE statement* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001496ASTNode::ID Parser::block() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001497 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001498 if (!this->expect(Token::Kind::TK_LBRACE, "'{'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001499 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001500 }
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001501 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001502 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001503 return ASTNode::ID::Invalid();
1504 }
John Stiles3b209362020-11-16 17:03:10 -05001505 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -07001506 for (;;) {
1507 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001508 case Token::Kind::TK_RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001509 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001510 return result;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001511 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001512 this->error(this->peek(), "expected '}', but found end of file");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001513 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001514 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001515 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001516 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001517 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001518 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001519 getNode(result).addChild(statement);
ethannicholasb3058bd2016-07-01 08:22:01 -07001520 }
1521 }
1522 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001523 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001524}
1525
1526/* expression SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001527ASTNode::ID Parser::expressionStatement() {
1528 ASTNode::ID expr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001529 if (expr) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001530 if (this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001531 return expr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001532 }
1533 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001534 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001535}
1536
1537/* assignmentExpression (COMMA assignmentExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001538ASTNode::ID Parser::expression() {
1539 ASTNode::ID result = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001540 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001541 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001542 }
1543 Token t;
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001544 AutoDepth depth(this);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001545 while (this->checkNext(Token::Kind::TK_COMMA, &t)) {
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001546 if (!depth.increase()) {
1547 return ASTNode::ID::Invalid();
1548 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001549 ASTNode::ID right = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001550 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001551 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001552 }
John Stiles45990502021-02-16 10:55:27 -05001553 ASTNode::ID newResult = this->createNode(t.fOffset, ASTNode::Kind::kBinary,
1554 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001555 getNode(newResult).addChild(result);
1556 getNode(newResult).addChild(right);
1557 result = newResult;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001558 }
1559 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001560}
1561
1562/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1563 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1564 assignmentExpression)*
1565 */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001566ASTNode::ID Parser::assignmentExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001567 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001568 ASTNode::ID result = this->ternaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001569 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001570 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001571 }
1572 for (;;) {
1573 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001574 case Token::Kind::TK_EQ: // fall through
1575 case Token::Kind::TK_STAREQ: // fall through
1576 case Token::Kind::TK_SLASHEQ: // fall through
1577 case Token::Kind::TK_PERCENTEQ: // fall through
1578 case Token::Kind::TK_PLUSEQ: // fall through
1579 case Token::Kind::TK_MINUSEQ: // fall through
1580 case Token::Kind::TK_SHLEQ: // fall through
1581 case Token::Kind::TK_SHREQ: // fall through
1582 case Token::Kind::TK_BITWISEANDEQ: // fall through
1583 case Token::Kind::TK_BITWISEXOREQ: // fall through
John Stiles8b3b1592020-11-23 11:06:44 -05001584 case Token::Kind::TK_BITWISEOREQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001585 if (!depth.increase()) {
1586 return ASTNode::ID::Invalid();
1587 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001588 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001589 ASTNode::ID right = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001590 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001591 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001592 }
John Stiles3b209362020-11-16 17:03:10 -05001593 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001594 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001595 getNode(newResult).addChild(result);
1596 getNode(newResult).addChild(right);
1597 result = newResult;
1598 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001599 }
1600 default:
1601 return result;
1602 }
1603 }
1604}
1605
1606/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001607ASTNode::ID Parser::ternaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001608 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001609 ASTNode::ID base = this->logicalOrExpression();
1610 if (!base) {
1611 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001612 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001613 if (this->checkNext(Token::Kind::TK_QUESTION)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001614 if (!depth.increase()) {
1615 return ASTNode::ID::Invalid();
1616 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001617 ASTNode::ID trueExpr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001618 if (!trueExpr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001619 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001620 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001621 if (this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001622 ASTNode::ID falseExpr = this->assignmentExpression();
1623 if (!falseExpr) {
1624 return ASTNode::ID::Invalid();
1625 }
John Stiles3b209362020-11-16 17:03:10 -05001626 ASTNode::ID ternary = this->createNode(getNode(base).fOffset, ASTNode::Kind::kTernary);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001627 getNode(ternary).addChild(base);
1628 getNode(ternary).addChild(trueExpr);
1629 getNode(ternary).addChild(falseExpr);
1630 return ternary;
ethannicholasb3058bd2016-07-01 08:22:01 -07001631 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001632 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001633 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001634 return base;
ethannicholasb3058bd2016-07-01 08:22:01 -07001635}
1636
1637/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001638ASTNode::ID Parser::logicalOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001639 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001640 ASTNode::ID result = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001641 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001642 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001643 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001644 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001645 while (this->checkNext(Token::Kind::TK_LOGICALOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001646 if (!depth.increase()) {
1647 return ASTNode::ID::Invalid();
1648 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001649 ASTNode::ID right = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001650 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001651 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001652 }
John Stiles3b209362020-11-16 17:03:10 -05001653 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001654 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001655 getNode(newResult).addChild(result);
1656 getNode(newResult).addChild(right);
1657 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001658 }
1659 return result;
1660}
1661
1662/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001663ASTNode::ID Parser::logicalXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001664 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001665 ASTNode::ID result = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001666 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001667 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001668 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001669 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001670 while (this->checkNext(Token::Kind::TK_LOGICALXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001671 if (!depth.increase()) {
1672 return ASTNode::ID::Invalid();
1673 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001674 ASTNode::ID right = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001675 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001676 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001677 }
John Stiles3b209362020-11-16 17:03:10 -05001678 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001679 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001680 getNode(newResult).addChild(result);
1681 getNode(newResult).addChild(right);
1682 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001683 }
1684 return result;
1685}
1686
1687/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001688ASTNode::ID Parser::logicalAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001689 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001690 ASTNode::ID result = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001691 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001692 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001693 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001694 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001695 while (this->checkNext(Token::Kind::TK_LOGICALAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001696 if (!depth.increase()) {
1697 return ASTNode::ID::Invalid();
1698 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001699 ASTNode::ID right = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001700 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001701 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001702 }
John Stiles3b209362020-11-16 17:03:10 -05001703 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001704 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001705 getNode(newResult).addChild(result);
1706 getNode(newResult).addChild(right);
1707 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001708 }
1709 return result;
1710}
1711
1712/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001713ASTNode::ID Parser::bitwiseOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001714 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001715 ASTNode::ID result = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001716 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001717 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001718 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001719 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001720 while (this->checkNext(Token::Kind::TK_BITWISEOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001721 if (!depth.increase()) {
1722 return ASTNode::ID::Invalid();
1723 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001724 ASTNode::ID right = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001725 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001726 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001727 }
John Stiles45990502021-02-16 10:55:27 -05001728 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
1729 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001730 getNode(newResult).addChild(result);
1731 getNode(newResult).addChild(right);
1732 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001733 }
1734 return result;
1735}
1736
1737/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001738ASTNode::ID Parser::bitwiseXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001739 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001740 ASTNode::ID result = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001741 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001742 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001743 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001744 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001745 while (this->checkNext(Token::Kind::TK_BITWISEXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001746 if (!depth.increase()) {
1747 return ASTNode::ID::Invalid();
1748 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001749 ASTNode::ID right = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001750 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001751 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001752 }
John Stiles45990502021-02-16 10:55:27 -05001753 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
1754 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001755 getNode(newResult).addChild(result);
1756 getNode(newResult).addChild(right);
1757 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001758 }
1759 return result;
1760}
1761
1762/* equalityExpression (BITWISEAND equalityExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001763ASTNode::ID Parser::bitwiseAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001764 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001765 ASTNode::ID result = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001766 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001767 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001768 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001769 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001770 while (this->checkNext(Token::Kind::TK_BITWISEAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001771 if (!depth.increase()) {
1772 return ASTNode::ID::Invalid();
1773 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001774 ASTNode::ID right = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001775 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001776 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001777 }
John Stiles3b209362020-11-16 17:03:10 -05001778 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001779 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001780 getNode(newResult).addChild(result);
1781 getNode(newResult).addChild(right);
1782 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001783 }
1784 return result;
1785}
1786
1787/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001788ASTNode::ID Parser::equalityExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001789 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001790 ASTNode::ID result = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001791 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001792 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001793 }
1794 for (;;) {
1795 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001796 case Token::Kind::TK_EQEQ: // fall through
1797 case Token::Kind::TK_NEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001798 if (!depth.increase()) {
1799 return ASTNode::ID::Invalid();
1800 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001801 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001802 ASTNode::ID right = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001803 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001804 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001805 }
John Stiles3b209362020-11-16 17:03:10 -05001806 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001807 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001808 getNode(newResult).addChild(result);
1809 getNode(newResult).addChild(right);
1810 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001811 break;
1812 }
1813 default:
1814 return result;
1815 }
1816 }
1817}
1818
1819/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001820ASTNode::ID Parser::relationalExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001821 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001822 ASTNode::ID result = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001823 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001824 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001825 }
1826 for (;;) {
1827 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001828 case Token::Kind::TK_LT: // fall through
1829 case Token::Kind::TK_GT: // fall through
1830 case Token::Kind::TK_LTEQ: // fall through
1831 case Token::Kind::TK_GTEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001832 if (!depth.increase()) {
1833 return ASTNode::ID::Invalid();
1834 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001835 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001836 ASTNode::ID right = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001837 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001838 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001839 }
John Stiles3b209362020-11-16 17:03:10 -05001840 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001841 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001842 getNode(newResult).addChild(result);
1843 getNode(newResult).addChild(right);
1844 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001845 break;
1846 }
1847 default:
1848 return result;
1849 }
1850 }
1851}
1852
1853/* additiveExpression ((SHL | SHR) additiveExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001854ASTNode::ID Parser::shiftExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001855 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001856 ASTNode::ID result = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001857 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001858 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001859 }
1860 for (;;) {
1861 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001862 case Token::Kind::TK_SHL: // fall through
1863 case Token::Kind::TK_SHR: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001864 if (!depth.increase()) {
1865 return ASTNode::ID::Invalid();
1866 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001867 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001868 ASTNode::ID right = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001869 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001870 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001871 }
John Stiles3b209362020-11-16 17:03:10 -05001872 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001873 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001874 getNode(newResult).addChild(result);
1875 getNode(newResult).addChild(right);
1876 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001877 break;
1878 }
1879 default:
1880 return result;
1881 }
1882 }
1883}
1884
1885/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001886ASTNode::ID Parser::additiveExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001887 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001888 ASTNode::ID result = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001889 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001890 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001891 }
1892 for (;;) {
1893 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001894 case Token::Kind::TK_PLUS: // fall through
1895 case Token::Kind::TK_MINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001896 if (!depth.increase()) {
1897 return ASTNode::ID::Invalid();
1898 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001899 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001900 ASTNode::ID right = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001901 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001902 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001903 }
John Stiles3b209362020-11-16 17:03:10 -05001904 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001905 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001906 getNode(newResult).addChild(result);
1907 getNode(newResult).addChild(right);
1908 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001909 break;
1910 }
1911 default:
1912 return result;
1913 }
1914 }
1915}
1916
1917/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001918ASTNode::ID Parser::multiplicativeExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001919 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001920 ASTNode::ID result = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001921 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001922 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001923 }
1924 for (;;) {
1925 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001926 case Token::Kind::TK_STAR: // fall through
1927 case Token::Kind::TK_SLASH: // fall through
1928 case Token::Kind::TK_PERCENT: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001929 if (!depth.increase()) {
1930 return ASTNode::ID::Invalid();
1931 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001932 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001933 ASTNode::ID right = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001934 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001935 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001936 }
John Stiles3b209362020-11-16 17:03:10 -05001937 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001938 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001939 getNode(newResult).addChild(result);
1940 getNode(newResult).addChild(right);
1941 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001942 break;
1943 }
1944 default:
1945 return result;
1946 }
1947 }
1948}
1949
1950/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001951ASTNode::ID Parser::unaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001952 AutoDepth depth(this);
ethannicholasb3058bd2016-07-01 08:22:01 -07001953 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001954 case Token::Kind::TK_PLUS: // fall through
1955 case Token::Kind::TK_MINUS: // fall through
1956 case Token::Kind::TK_LOGICALNOT: // fall through
1957 case Token::Kind::TK_BITWISENOT: // fall through
1958 case Token::Kind::TK_PLUSPLUS: // fall through
1959 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001960 if (!depth.increase()) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001961 return ASTNode::ID::Invalid();
Ethan Nicholas6dcc3252019-02-20 15:18:36 -05001962 }
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001963 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001964 ASTNode::ID expr = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001965 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001966 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001967 }
John Stiles45990502021-02-16 10:55:27 -05001968 ASTNode::ID result = this->createNode(t.fOffset, ASTNode::Kind::kPrefix,
1969 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001970 getNode(result).addChild(expr);
1971 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001972 }
1973 default:
1974 return this->postfixExpression();
1975 }
1976}
1977
1978/* term suffix* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001979ASTNode::ID Parser::postfixExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001980 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001981 ASTNode::ID result = this->term();
ethannicholasb3058bd2016-07-01 08:22:01 -07001982 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001983 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001984 }
1985 for (;;) {
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001986 Token t = this->peek();
1987 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001988 case Token::Kind::TK_FLOAT_LITERAL:
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001989 if (this->text(t)[0] != '.') {
1990 return result;
1991 }
John Stiles30212b72020-06-11 17:55:07 -04001992 [[fallthrough]];
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001993 case Token::Kind::TK_LBRACKET:
1994 case Token::Kind::TK_DOT:
1995 case Token::Kind::TK_LPAREN:
1996 case Token::Kind::TK_PLUSPLUS:
1997 case Token::Kind::TK_MINUSMINUS:
1998 case Token::Kind::TK_COLONCOLON:
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001999 if (!depth.increase()) {
2000 return ASTNode::ID::Invalid();
2001 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002002 result = this->suffix(result);
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002003 if (!result) {
2004 return ASTNode::ID::Invalid();
2005 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002006 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002007 default:
2008 return result;
2009 }
2010 }
2011}
2012
Ethan Nicholas11d53972016-11-28 11:23:23 -05002013/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
Ethan Nicholase455f652019-09-13 12:52:55 -04002014 PLUSPLUS | MINUSMINUS | COLONCOLON IDENTIFIER | FLOAT_LITERAL [IDENTIFIER] */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002015ASTNode::ID Parser::suffix(ASTNode::ID base) {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002016 SkASSERT(base);
ethannicholasb3058bd2016-07-01 08:22:01 -07002017 Token next = this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002018 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002019 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002020 return ASTNode::ID::Invalid();
2021 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002022 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002023 case Token::Kind::TK_LBRACKET: {
2024 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
John Stiles3b209362020-11-16 17:03:10 -05002025 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kIndex);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002026 getNode(result).addChild(base);
2027 return result;
ethannicholas5961bc92016-10-12 06:39:56 -07002028 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002029 ASTNode::ID e = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002030 if (!e) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002031 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002032 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002033 this->expect(Token::Kind::TK_RBRACKET, "']' to complete array access expression");
John Stiles3b209362020-11-16 17:03:10 -05002034 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kIndex);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002035 getNode(result).addChild(base);
2036 getNode(result).addChild(e);
2037 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002038 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002039 case Token::Kind::TK_COLONCOLON: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002040 int offset = this->peek().fOffset;
2041 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002042 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002043 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kScope,
2044 std::move(text));
Brian Osman6518d772020-09-10 16:50:06 -04002045 getNode(result).addChild(base);
2046 return result;
2047 }
2048 return ASTNode::ID::Invalid();
2049 }
2050 case Token::Kind::TK_DOT: {
2051 int offset = this->peek().fOffset;
2052 StringFragment text;
2053 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002054 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kField,
2055 std::move(text));
Ethan Nicholasfc994162019-06-06 10:04:27 -04002056 getNode(result).addChild(base);
2057 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002058 }
Brian Osman6518d772020-09-10 16:50:06 -04002059 [[fallthrough]];
Ethan Nicholase455f652019-09-13 12:52:55 -04002060 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002061 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholase455f652019-09-13 12:52:55 -04002062 // Swizzles that start with a constant number, e.g. '.000r', will be tokenized as
2063 // floating point literals, possibly followed by an identifier. Handle that here.
2064 StringFragment field = this->text(next);
2065 SkASSERT(field.fChars[0] == '.');
2066 ++field.fChars;
2067 --field.fLength;
2068 for (size_t i = 0; i < field.fLength; ++i) {
2069 if (field.fChars[i] != '0' && field.fChars[i] != '1') {
2070 this->error(next, "invalid swizzle");
2071 return ASTNode::ID::Invalid();
2072 }
2073 }
2074 // use the next *raw* token so we don't ignore whitespace - we only care about
2075 // identifiers that directly follow the float
2076 Token id = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002077 if (id.fKind == Token::Kind::TK_IDENTIFIER) {
Ethan Nicholase455f652019-09-13 12:52:55 -04002078 field.fLength += id.fLength;
2079 } else {
2080 this->pushback(id);
2081 }
John Stiles3b209362020-11-16 17:03:10 -05002082 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kField, field);
Ethan Nicholase455f652019-09-13 12:52:55 -04002083 getNode(result).addChild(base);
2084 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002085 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002086 case Token::Kind::TK_LPAREN: {
John Stiles3b209362020-11-16 17:03:10 -05002087 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kCall);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002088 getNode(result).addChild(base);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002089 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002090 for (;;) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002091 ASTNode::ID expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002092 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002093 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002094 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002095 getNode(result).addChild(expr);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002096 if (!this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002097 break;
2098 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002099 }
2100 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002101 this->expect(Token::Kind::TK_RPAREN, "')' to complete function parameters");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002102 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002103 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002104 case Token::Kind::TK_PLUSPLUS: // fall through
2105 case Token::Kind::TK_MINUSMINUS: {
John Stiles45990502021-02-16 10:55:27 -05002106 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kPostfix,
2107 Operator(next.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04002108 getNode(result).addChild(base);
2109 return result;
2110 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002111 default: {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002112 this->error(next, "expected expression suffix, but found '" + this->text(next) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002113 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002114 }
2115 }
2116}
2117
Brian Osman33c64a42020-12-23 10:26:38 -05002118/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | '(' expression ')' */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002119ASTNode::ID Parser::term() {
ethannicholasb3058bd2016-07-01 08:22:01 -07002120 Token t = this->peek();
2121 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002122 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002123 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002124 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002125 return this->createNode(t.fOffset, ASTNode::Kind::kIdentifier, std::move(text));
ethannicholasb3058bd2016-07-01 08:22:01 -07002126 }
John Stiles30212b72020-06-11 17:55:07 -04002127 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002128 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002129 case Token::Kind::TK_INT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002130 SKSL_INT i;
ethannicholasb3058bd2016-07-01 08:22:01 -07002131 if (this->intLiteral(&i)) {
John Stiles3b209362020-11-16 17:03:10 -05002132 return this->createNode(t.fOffset, ASTNode::Kind::kInt, i);
ethannicholasb3058bd2016-07-01 08:22:01 -07002133 }
2134 break;
2135 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002136 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002137 SKSL_FLOAT f;
ethannicholasb3058bd2016-07-01 08:22:01 -07002138 if (this->floatLiteral(&f)) {
John Stiles3b209362020-11-16 17:03:10 -05002139 return this->createNode(t.fOffset, ASTNode::Kind::kFloat, f);
ethannicholasb3058bd2016-07-01 08:22:01 -07002140 }
2141 break;
2142 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002143 case Token::Kind::TK_TRUE_LITERAL: // fall through
2144 case Token::Kind::TK_FALSE_LITERAL: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002145 bool b;
2146 if (this->boolLiteral(&b)) {
John Stiles3b209362020-11-16 17:03:10 -05002147 return this->createNode(t.fOffset, ASTNode::Kind::kBool, b);
ethannicholasb3058bd2016-07-01 08:22:01 -07002148 }
2149 break;
2150 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002151 case Token::Kind::TK_LPAREN: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002152 this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002153 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002154 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002155 return ASTNode::ID::Invalid();
2156 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002157 ASTNode::ID result = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002158 if (result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002159 this->expect(Token::Kind::TK_RPAREN, "')' to complete expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002160 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002161 }
2162 break;
2163 }
2164 default:
2165 this->nextToken();
John Stilesf94348f2020-12-23 18:58:43 -05002166 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002167 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002168 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002169}
2170
2171/* INT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002172bool Parser::intLiteral(SKSL_INT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002173 Token t;
John Stilesf94348f2020-12-23 18:58:43 -05002174 if (!this->expect(Token::Kind::TK_INT_LITERAL, "integer literal", &t)) {
2175 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07002176 }
John Stilesf94348f2020-12-23 18:58:43 -05002177 StringFragment s = this->text(t);
2178 if (!SkSL::stoi(s, dest)) {
2179 this->error(t, "integer is too large: " + s);
2180 return false;
2181 }
2182 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07002183}
2184
John Stilesf94348f2020-12-23 18:58:43 -05002185
ethannicholasb3058bd2016-07-01 08:22:01 -07002186/* FLOAT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002187bool Parser::floatLiteral(SKSL_FLOAT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002188 Token t;
John Stilesf94348f2020-12-23 18:58:43 -05002189 if (!this->expect(Token::Kind::TK_FLOAT_LITERAL, "float literal", &t)) {
2190 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07002191 }
John Stilesf94348f2020-12-23 18:58:43 -05002192 StringFragment s = this->text(t);
2193 if (!SkSL::stod(s, dest)) {
2194 this->error(t, "floating-point value is too large: " + s);
2195 return false;
2196 }
2197 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07002198}
2199
2200/* TRUE_LITERAL | FALSE_LITERAL */
2201bool Parser::boolLiteral(bool* dest) {
2202 Token t = this->nextToken();
2203 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002204 case Token::Kind::TK_TRUE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002205 *dest = true;
2206 return true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002207 case Token::Kind::TK_FALSE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002208 *dest = false;
2209 return true;
2210 default:
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002211 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002212 return false;
2213 }
2214}
2215
2216/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002217bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002218 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002219 if (this->expect(Token::Kind::TK_IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002220 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07002221 return true;
2222 }
2223 return false;
2224}
2225
John Stilesa6841be2020-08-06 14:11:56 -04002226} // namespace SkSL