blob: 9ba858e19953fb9bb58efe7b89f8d6d8ea1a7950 [file] [log] [blame]
ethannicholasb3058bd2016-07-01 08:22:01 -07001/*
2 * Copyright 2016 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
Ethan Nicholas11d53972016-11-28 11:23:23 -05007
Mike Klein4b432fa2019-06-06 11:44:05 -05008#include "src/sksl/SkSLParser.h"
John Stilesfbd050b2020-08-03 13:21:46 -04009
10#include <memory>
11#include "stdio.h"
12
Ethan Nicholasdaed2592021-03-04 14:30:25 -050013#include "include/private/SkSLModifiers.h"
John Stilesfbd050b2020-08-03 13:21:46 -040014#include "src/sksl/SkSLASTNode.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050015#include "src/sksl/ir/SkSLSymbolTable.h"
16#include "src/sksl/ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070017
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040018#ifndef SKSL_STANDALONE
Mike Kleinc0bd9f92019-04-23 12:05:21 -050019#include "include/private/SkOnce.h"
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040020#endif
21
ethannicholasb3058bd2016-07-01 08:22:01 -070022namespace SkSL {
23
John Stiles8d056592020-11-10 10:03:50 -050024static constexpr int kMaxParseDepth = 50;
John Stilesd39aec92020-12-03 14:37:16 -050025static constexpr int kMaxStructDepth = 8;
John Stilesa695d622020-11-10 10:04:26 -050026
27static bool struct_is_too_deeply_nested(const Type& type, int limit) {
28 if (limit < 0) {
29 return true;
30 }
31
John Stilesc0c51062020-12-03 17:16:29 -050032 if (type.isStruct()) {
John Stilesa695d622020-11-10 10:04:26 -050033 for (const Type::Field& f : type.fields()) {
34 if (struct_is_too_deeply_nested(*f.fType, limit - 1)) {
35 return true;
36 }
37 }
38 }
39
40 return false;
41}
ethannicholascad64162016-10-27 10:54:02 -070042
John Stiles7d5b90a2021-01-21 14:26:51 -050043static int parse_modifier_token(Token::Kind token) {
44 switch (token) {
45 case Token::Kind::TK_UNIFORM: return Modifiers::kUniform_Flag;
46 case Token::Kind::TK_CONST: return Modifiers::kConst_Flag;
47 case Token::Kind::TK_IN: return Modifiers::kIn_Flag;
48 case Token::Kind::TK_OUT: return Modifiers::kOut_Flag;
49 case Token::Kind::TK_INOUT: return Modifiers::kIn_Flag | Modifiers::kOut_Flag;
50 case Token::Kind::TK_FLAT: return Modifiers::kFlat_Flag;
51 case Token::Kind::TK_NOPERSPECTIVE: return Modifiers::kNoPerspective_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050052 case Token::Kind::TK_HASSIDEEFFECTS: return Modifiers::kHasSideEffects_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050053 case Token::Kind::TK_VARYING: return Modifiers::kVarying_Flag;
54 case Token::Kind::TK_INLINE: return Modifiers::kInline_Flag;
55 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)
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700129: fText(text)
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 Stiles3b209362020-11-16 17:03:10 -0500159 ASTNode::ID result = this->createNode(/*offset=*/0, ASTNode::Kind::kFile);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400160 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700161 for (;;) {
162 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400163 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400164 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400165 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400166 ASTNode::ID dir = this->directive();
167 if (fErrors.errorCount()) {
168 return nullptr;
169 }
170 if (dir) {
171 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700172 }
173 break;
174 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400175 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400176 ASTNode::ID section = this->section();
177 if (fErrors.errorCount()) {
178 return nullptr;
179 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400180 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400181 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400182 }
183 break;
184 }
John Stiles7cbb09c22021-01-07 16:07:00 -0500185 case Token::Kind::TK_INVALID: {
186 this->error(this->peek(), String("invalid token"));
187 return nullptr;
188 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700189 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400190 ASTNode::ID decl = this->declaration();
191 if (fErrors.errorCount()) {
192 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700193 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400194 if (decl) {
195 getNode(result).addChild(decl);
196 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700197 }
198 }
199 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400200 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700201}
202
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700203Token Parser::nextRawToken() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500204 if (fPushback.fKind != Token::Kind::TK_NONE) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700205 Token result = fPushback;
John Stiles7cbb09c22021-01-07 16:07:00 -0500206 fPushback.fKind = Token::Kind::TK_NONE;
ethannicholasb3058bd2016-07-01 08:22:01 -0700207 return result;
208 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700209 Token result = fLexer.next();
210 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400211}
212
213Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700214 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400215 while (token.fKind == Token::Kind::TK_WHITESPACE ||
216 token.fKind == Token::Kind::TK_LINE_COMMENT ||
217 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700218 token = this->nextRawToken();
219 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400220 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700221}
222
223void Parser::pushback(Token t) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500224 SkASSERT(fPushback.fKind == Token::Kind::TK_NONE);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400225 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700226}
227
228Token Parser::peek() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500229 if (fPushback.fKind == Token::Kind::TK_NONE) {
Brian Osman634624a2017-08-15 11:14:30 -0400230 fPushback = this->nextToken();
231 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700232 return fPushback;
233}
234
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400235bool Parser::checkNext(Token::Kind kind, Token* result) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500236 if (fPushback.fKind != Token::Kind::TK_NONE && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400237 return false;
238 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400239 Token next = this->nextToken();
240 if (next.fKind == kind) {
241 if (result) {
242 *result = next;
243 }
244 return true;
245 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400246 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400247 return false;
248}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500249
250bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700251 Token next = this->nextToken();
252 if (next.fKind == kind) {
253 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400254 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700255 }
256 return true;
257 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700258 this->error(next, "expected " + String(expected) + ", but found '" +
259 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700260 return false;
261 }
262}
263
John Stiles2630ea32020-12-04 10:51:21 -0500264bool Parser::expectIdentifier(Token* result) {
265 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", result)) {
266 return false;
267 }
268 if (this->isType(this->text(*result))) {
269 this->error(*result, "expected an identifier, but found type '" +
270 this->text(*result) + "'");
271 return false;
272 }
273 return true;
274}
275
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700276StringFragment Parser::text(Token token) {
277 return StringFragment(fText + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500278}
279
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700280void Parser::error(Token token, String msg) {
281 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700282}
283
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700284void Parser::error(int offset, String msg) {
285 fErrors.error(offset, msg);
286}
287
288bool Parser::isType(StringFragment name) {
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400289 const Symbol* s = fSymbols[name];
John Stiles2630ea32020-12-04 10:51:21 -0500290 return s && s->is<Type>();
ethannicholasb3058bd2016-07-01 08:22:01 -0700291}
292
John Stiles80b02af2021-02-12 17:07:51 -0500293bool Parser::isArrayType(ASTNode::ID type) {
294 const ASTNode& node = this->getNode(type);
295 SkASSERT(node.fKind == ASTNode::Kind::kType);
296 return node.begin() != node.end();
297}
298
Ethan Nicholas11d53972016-11-28 11:23:23 -0500299/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700300 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400301ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700302 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400303 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400304 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700305 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700306 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400307 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700308 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500309 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400310 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700311 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400312 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400313 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700314 }
315 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400316 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400317 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700318 }
John Stiles3b209362020-11-16 17:03:10 -0500319 return this->createNode(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700320 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700321 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400322 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700323 }
324}
325
Ethan Nicholas762466e2017-06-29 10:03:38 -0400326/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
327 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400328ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400329 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400330 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400331 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400332 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400333 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400334 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400335 this->nextToken();
336 Token argToken;
John Stiles2630ea32020-12-04 10:51:21 -0500337 if (!this->expectIdentifier(&argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400338 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400339 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700340 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400341 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400342 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400343 }
344 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400345 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400346 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400347 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400348 StringFragment text;
349 Token codeStart = this->nextRawToken();
350 size_t startOffset = codeStart.fOffset;
351 this->pushback(codeStart);
352 text.fChars = fText + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400353 int level = 1;
354 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700355 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400356 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400357 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400358 ++level;
359 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400360 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400361 --level;
362 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400363 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700364 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400365 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400366 default:
367 break;
368 }
369 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400370 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400371 break;
372 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400373 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700374 StringFragment name = this->text(start);
375 ++name.fChars;
376 --name.fLength;
John Stiles3b209362020-11-16 17:03:10 -0500377 return this->createNode(start.fOffset, ASTNode::Kind::kSection,
378 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400379}
380
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500381/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
382 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400383ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500384 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400385 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400386 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500387 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400388 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400389 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500390 }
391 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500392 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400393 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500394 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400395 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400396 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500397 }
John Stilese6c67c52021-01-15 12:01:00 -0500398 fSymbols.add(Type::MakeEnumType(this->text(name)));
John Stiles3b209362020-11-16 17:03:10 -0500399 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400400 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500401 Token id;
John Stiles2630ea32020-12-04 10:51:21 -0500402 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400403 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500404 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400405 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400406 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500407 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400408 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500409 }
John Stiles3b209362020-11-16 17:03:10 -0500410 ASTNode::ID child = this->addChild(
411 result, this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400412 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500413 } else {
John Stiles3b209362020-11-16 17:03:10 -0500414 this->addChild(result,
415 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500416 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400417 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
418 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400419 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500420 }
John Stiles2630ea32020-12-04 10:51:21 -0500421 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400422 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500423 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400424 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400425 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500426 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400427 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500428 }
John Stiles3b209362020-11-16 17:03:10 -0500429 ASTNode::ID child = this->addChild(
430 result,
431 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400432 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500433 } else {
John Stiles3b209362020-11-16 17:03:10 -0500434 this->addChild(
435 result,
436 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500437 }
438 }
439 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400440 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400441 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500442}
443
444/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700445 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400446ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700447 Token lookahead = this->peek();
Ethan Nicholasda6320c2020-09-02 14:08:23 -0400448 switch (lookahead.fKind) {
449 case Token::Kind::TK_ENUM:
450 return this->enumDeclaration();
451 case Token::Kind::TK_SEMICOLON:
452 this->error(lookahead.fOffset, "expected a declaration, but found ';'");
453 return ASTNode::ID::Invalid();
454 default:
455 break;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500456 }
457 Modifiers modifiers = this->modifiers();
458 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400459 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700460 // we have an identifier that's not a type, could be the start of an interface block
461 return this->interfaceBlock(modifiers);
462 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400463 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700464 return this->structVarDeclaration(modifiers);
465 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400466 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700467 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -0500468 return this->createNode(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700469 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400470 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700471 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400472 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700473 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700474 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500475 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400476 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700477 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400478 if (this->checkNext(Token::Kind::TK_LPAREN)) {
John Stiles3b209362020-11-16 17:03:10 -0500479 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kFunction);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400480 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
481 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400482 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400483 for (;;) {
484 ASTNode::ID parameter = this->parameter();
485 if (!parameter) {
486 return ASTNode::ID::Invalid();
487 }
488 ++fd.fParameterCount;
489 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400490 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400491 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700492 }
493 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700494 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400495 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400496 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400497 return ASTNode::ID::Invalid();
498 }
499 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400500 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700501 body = this->block();
502 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400503 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700504 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400505 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700506 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400507 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700508 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400509 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700510 }
511}
512
John Stiles76389b72021-01-25 13:49:05 -0500513/* (varDeclarations | expressionStatement) */
514ASTNode::ID Parser::varDeclarationsOrExpressionStatement() {
515 if (this->isType(this->text(this->peek()))) {
516 // Statements that begin with a typename are most often variable declarations, but
517 // occasionally the type is part of a constructor, and these are actually expression-
518 // statements in disguise. First, attempt the common case: parse it as a vardecl.
519 Checkpoint checkpoint(this);
John Stiles8dabeac2021-02-12 16:05:00 -0500520 VarDeclarationsPrefix prefix;
521 if (this->varDeclarationsPrefix(&prefix)) {
522 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
John Stiles76389b72021-01-25 13:49:05 -0500523 }
524
525 // If this statement wasn't actually a vardecl after all, rewind and try parsing it as an
526 // expression-statement instead.
527 checkpoint.rewind();
528 }
529
530 return this->expressionStatement();
531}
532
John Stiles8dabeac2021-02-12 16:05:00 -0500533// Helper function for varDeclarations(). If this function succeeds, we assume that the rest of the
534// statement is a variable-declaration statement, not an expression-statement.
535bool Parser::varDeclarationsPrefix(VarDeclarationsPrefix* prefixData) {
536 prefixData->modifiers = this->modifiers();
537 prefixData->type = this->type();
538 if (!prefixData->type) {
539 return false;
540 }
541 return this->expectIdentifier(&prefixData->name);
542}
543
ethannicholasb3058bd2016-07-01 08:22:01 -0700544/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400545ASTNode::ID Parser::varDeclarations() {
John Stiles8dabeac2021-02-12 16:05:00 -0500546 VarDeclarationsPrefix prefix;
547 if (!this->varDeclarationsPrefix(&prefix)) {
John Stiles76389b72021-01-25 13:49:05 -0500548 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700549 }
John Stiles8dabeac2021-02-12 16:05:00 -0500550 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700551}
552
553/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400554ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400555 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400556 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700557 }
558 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500559 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400560 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700561 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400562 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400563 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700564 }
565 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400566 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400567 ASTNode::ID decls = this->varDeclarations();
568 if (!decls) {
569 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700570 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400571 ASTNode& declsNode = getNode(decls);
John Stilesece1bf02021-03-08 11:15:55 -0500572 const Modifiers& modifiers = declsNode.begin()->getModifiers();
John Stiles13fc2602020-10-09 17:42:31 -0400573 if (modifiers.fFlags != Modifiers::kNo_Flag) {
574 String desc = modifiers.description();
575 desc.pop_back(); // remove trailing space
576 this->error(declsNode.fOffset,
577 "modifier '" + desc + "' is not permitted on a struct field");
578 }
579
Brian Osman00fea5b2021-01-28 09:46:30 -0500580 const Symbol* symbol = fSymbols[(declsNode.begin() + 1)->getString()];
John Stiles13fc2602020-10-09 17:42:31 -0400581 SkASSERT(symbol);
582 const Type* type = &symbol->as<Type>();
John Stiles1d757782020-11-17 09:43:22 -0500583 if (type->isOpaque()) {
584 this->error(declsNode.fOffset,
585 "opaque type '" + type->name() + "' is not permitted in a struct");
586 }
John Stiles13fc2602020-10-09 17:42:31 -0400587
Ethan Nicholasfc994162019-06-06 10:04:27 -0400588 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
589 ASTNode& var = *iter;
John Stilesece1bf02021-03-08 11:15:55 -0500590 const ASTNode::VarData& vd = var.getVarData();
John Stiles6bef6a72020-12-02 14:26:04 -0500591
John Stilesd39aec92020-12-03 14:37:16 -0500592 // Read array size if one is present.
593 if (vd.fIsArray) {
594 const ASTNode& size = *var.begin();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400595 if (!size || size.fKind != ASTNode::Kind::kInt) {
596 this->error(declsNode.fOffset, "array size in struct field must be a constant");
597 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700598 }
John Stiles6bef6a72020-12-02 14:26:04 -0500599 if (size.getInt() <= 0 || size.getInt() > INT_MAX) {
600 this->error(declsNode.fOffset, "array size is invalid");
601 return ASTNode::ID::Invalid();
602 }
John Stilesd39aec92020-12-03 14:37:16 -0500603 // Add the array dimensions to our type.
604 int arraySize = size.getInt();
John Stilesa2179502020-12-03 14:37:57 -0500605 type = fSymbols.addArrayDimension(type, arraySize);
ethannicholasb3058bd2016-07-01 08:22:01 -0700606 }
John Stiles13fc2602020-10-09 17:42:31 -0400607
608 fields.push_back(Type::Field(modifiers, vd.fName, type));
John Stilesd39aec92020-12-03 14:37:16 -0500609 if (vd.fIsArray ? var.begin()->fNext : var.fFirstChild) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400610 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700611 }
612 }
613 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400614 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400615 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700616 }
Brian Osman6c3b23f2021-02-12 13:09:58 -0500617 if (fields.empty()) {
618 this->error(name.fOffset,
619 "struct '" + this->text(name) + "' must contain at least one field");
620 return ASTNode::ID::Invalid();
621 }
John Stilesad2d4942020-12-11 16:55:58 -0500622 std::unique_ptr<Type> newType = Type::MakeStructType(name.fOffset, this->text(name), fields);
John Stilesa695d622020-11-10 10:04:26 -0500623 if (struct_is_too_deeply_nested(*newType, kMaxStructDepth)) {
624 this->error(name.fOffset, "struct '" + this->text(name) + "' is too deeply nested");
625 return ASTNode::ID::Invalid();
626 }
627 fSymbols.add(std::move(newType));
Brian Osman00fea5b2021-01-28 09:46:30 -0500628 return this->createNode(name.fOffset, ASTNode::Kind::kType, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700629}
630
631/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400632ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
633 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700634 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400635 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700636 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400637 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400638 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400639 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700640 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400641 this->expect(Token::Kind::TK_SEMICOLON, "';'");
John Stilesdc75a972020-11-25 16:24:55 -0500642 return type;
ethannicholasb3058bd2016-07-01 08:22:01 -0700643}
644
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400645/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
646 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400647ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
John Stiles08070f62020-11-17 10:45:49 -0500648 int offset = this->peek().fOffset;
649 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kVarDeclarations);
650 this->addChild(result, this->createNode(offset, ASTNode::Kind::kModifiers, mods));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400651 getNode(result).addChild(type);
John Stilesd39aec92020-12-03 14:37:16 -0500652
John Stiles80b02af2021-02-12 17:07:51 -0500653 auto parseArrayDimensions = [&](ASTNode::ID currentVar, ASTNode::VarData* vd) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400654 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500655 if (vd->fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500656 this->error(this->peek(), "multi-dimensional arrays are not supported");
657 return false;
658 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400659 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
John Stiles3b209362020-11-16 17:03:10 -0500660 this->createEmptyChild(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700661 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400662 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700663 if (!size) {
John Stilesd39aec92020-12-03 14:37:16 -0500664 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700665 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400666 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400667 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
John Stilesd39aec92020-12-03 14:37:16 -0500668 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700669 }
670 }
John Stilesd39aec92020-12-03 14:37:16 -0500671 vd->fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700672 }
John Stilesd39aec92020-12-03 14:37:16 -0500673 return true;
674 };
675
676 auto parseInitializer = [this](ASTNode::ID currentVar) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400677 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400678 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700679 if (!value) {
John Stilesd39aec92020-12-03 14:37:16 -0500680 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700681 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400682 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700683 }
John Stilesd39aec92020-12-03 14:37:16 -0500684 return true;
685 };
686
687 ASTNode::ID currentVar = this->createNode(offset, ASTNode::Kind::kVarDeclaration);
688 ASTNode::VarData vd{name, /*isArray=*/false};
689
690 getNode(result).addChild(currentVar);
691 if (!parseArrayDimensions(currentVar, &vd)) {
692 return ASTNode::ID::Invalid();
693 }
694 getNode(currentVar).setVarData(vd);
695 if (!parseInitializer(currentVar)) {
696 return ASTNode::ID::Invalid();
697 }
698
699 while (this->checkNext(Token::Kind::TK_COMMA)) {
700 Token identifierName;
John Stiles2630ea32020-12-04 10:51:21 -0500701 if (!this->expectIdentifier(&identifierName)) {
John Stilesd39aec92020-12-03 14:37:16 -0500702 return ASTNode::ID::Invalid();
703 }
704
705 currentVar = ASTNode::ID(fFile->fNodes.size());
706 vd = ASTNode::VarData{this->text(identifierName), /*isArray=*/false};
707 fFile->fNodes.emplace_back(&fFile->fNodes, offset, ASTNode::Kind::kVarDeclaration);
708
709 getNode(result).addChild(currentVar);
710 if (!parseArrayDimensions(currentVar, &vd)) {
711 return ASTNode::ID::Invalid();
712 }
713 getNode(currentVar).setVarData(vd);
714 if (!parseInitializer(currentVar)) {
715 return ASTNode::ID::Invalid();
716 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700717 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400718 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400719 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700720 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400721 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700722}
723
724/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400725ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400726 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400727 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700728 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400729 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700730 }
731 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500732 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400733 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700734 }
John Stiles3b209362020-11-16 17:03:10 -0500735 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kParameter);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400736 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
737 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400738 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500739 if (pd.fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500740 this->error(this->peek(), "multi-dimensional arrays are not supported");
741 return ASTNode::ID::Invalid();
742 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700743 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400744 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400745 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700746 }
John Stilesf94348f2020-12-23 18:58:43 -0500747 StringFragment arraySizeFrag = this->text(sizeToken);
748 SKSL_INT arraySize;
749 if (!SkSL::stoi(arraySizeFrag, &arraySize)) {
750 this->error(sizeToken, "array size is too large: " + arraySizeFrag);
751 return ASTNode::ID::Invalid();
752 }
753 this->addChild(result, this->createNode(sizeToken.fOffset, ASTNode::Kind::kInt, arraySize));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400754 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400755 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700756 }
John Stilesd39aec92020-12-03 14:37:16 -0500757 pd.fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700758 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400759 getNode(result).setParameterData(pd);
760 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700761}
762
Ethan Nicholasd608c092017-10-26 09:30:08 -0400763/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700764int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400765 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700766 return -1;
767 }
768 Token resultToken;
John Stilesf94348f2020-12-23 18:58:43 -0500769 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
770 return -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700771 }
John Stilesf94348f2020-12-23 18:58:43 -0500772 StringFragment resultFrag = this->text(resultToken);
773 SKSL_INT resultValue;
774 if (!SkSL::stoi(resultFrag, &resultValue)) {
775 this->error(resultToken, "value in layout is too large: " + resultFrag);
776 return -1;
777 }
778 return resultValue;
ethannicholasb3058bd2016-07-01 08:22:01 -0700779}
780
Ethan Nicholasd608c092017-10-26 09:30:08 -0400781/** EQ IDENTIFIER */
782StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400783 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400784 return StringFragment();
785 }
786 Token resultToken;
John Stiles2630ea32020-12-04 10:51:21 -0500787 if (!this->expectIdentifier(&resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400788 return StringFragment();
789 }
790 return this->text(resultToken);
791}
792
793
Ethan Nicholas762466e2017-06-29 10:03:38 -0400794/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400795StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400796 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400797 return "";
798 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700799 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400800 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400801 StringFragment code;
802 code.fChars = fText + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400803 int level = 1;
804 bool done = false;
805 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700806 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400807 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400808 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400809 ++level;
810 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400811 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400812 --level;
813 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400814 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400815 if (level == 1) {
816 done = true;
817 }
818 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400819 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700820 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400821 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400822 default:
823 break;
824 }
825 if (!level) {
826 done = true;
827 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400828 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400829 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400830 this->pushback(std::move(next));
831 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400832 }
833 return code;
834}
835
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400836Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400837 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400838 Token t = this->nextToken();
839 String text = this->text(t);
840 auto found = layoutTokens->find(text);
841 if (found != layoutTokens->end()) {
842 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400843 case LayoutToken::SKPMCOLOR4F:
844 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500845 case LayoutToken::SKV4:
846 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400847 case LayoutToken::SKRECT:
848 return Layout::CType::kSkRect;
849 case LayoutToken::SKIRECT:
850 return Layout::CType::kSkIRect;
851 case LayoutToken::SKPMCOLOR:
852 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400853 case LayoutToken::BOOL:
854 return Layout::CType::kBool;
855 case LayoutToken::INT:
856 return Layout::CType::kInt32;
857 case LayoutToken::FLOAT:
858 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500859 case LayoutToken::SKM44:
860 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400861 default:
862 break;
863 }
864 }
865 this->error(t, "unsupported ctype");
866 }
867 return Layout::CType::kDefault;
868}
869
ethannicholas8ac838d2016-11-22 08:39:36 -0800870/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500871Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500872 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700873 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500874 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700875 int binding = -1;
876 int index = -1;
877 int set = -1;
878 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500879 int inputAttachmentIndex = -1;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500880 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
881 int maxVertices = -1;
882 int invocations = -1;
Brian Osmanf59a9612020-04-15 14:18:13 -0400883 StringFragment marker;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400884 StringFragment when;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400885 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400886 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
887 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500888 return Layout(flags, location, offset, binding, index, set, builtin,
Brian Osman4717fbb2021-02-23 13:12:09 -0500889 inputAttachmentIndex, primitive, maxVertices, invocations, marker, when,
Brian Osmanba7ef322021-02-23 13:37:22 -0500890 ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700891 }
892 for (;;) {
893 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700894 String text = this->text(t);
Brian Osmana77ed8b2021-02-23 12:54:22 -0500895 auto setFlag = [&](Layout::Flag f) {
896 if (flags & f) {
897 this->error(t, "layout qualifier '" + text + "' appears more than once");
898 }
899 flags |= f;
900 };
901 auto setPrimitive = [&](Layout::Primitive p) {
902 if (flags & Layout::kPrimitive_Flag) {
903 this->error(t, "only one primitive-type layout qualifier is allowed");
904 }
905 flags |= Layout::kPrimitive_Flag;
906 primitive = p;
907 };
908
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400909 auto found = layoutTokens->find(text);
910 if (found != layoutTokens->end()) {
911 switch (found->second) {
Brian Osmana77ed8b2021-02-23 12:54:22 -0500912 case LayoutToken::ORIGIN_UPPER_LEFT:
913 setFlag(Layout::kOriginUpperLeft_Flag);
914 break;
915 case LayoutToken::OVERRIDE_COVERAGE:
916 setFlag(Layout::kOverrideCoverage_Flag);
917 break;
Brian Osmandc4f8cd2021-02-25 15:30:46 -0500918 case LayoutToken::EARLY_FRAGMENT_TESTS:
919 setFlag(Layout::kEarlyFragmentTests_Flag);
920 break;
Brian Osmana77ed8b2021-02-23 12:54:22 -0500921 case LayoutToken::PUSH_CONSTANT:
922 setFlag(Layout::kPushConstant_Flag);
923 break;
924 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
925 setFlag(Layout::kBlendSupportAllEquations_Flag);
926 break;
927 case LayoutToken::TRACKED:
928 setFlag(Layout::kTracked_Flag);
929 break;
930 case LayoutToken::SRGB_UNPREMUL:
931 setFlag(Layout::kSRGBUnpremul_Flag);
932 break;
933 case LayoutToken::KEY:
934 setFlag(Layout::kKey_Flag);
935 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700936 case LayoutToken::LOCATION:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500937 setFlag(Layout::kLocation_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500938 location = this->layoutInt();
939 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700940 case LayoutToken::OFFSET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500941 setFlag(Layout::kOffset_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500942 offset = this->layoutInt();
943 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700944 case LayoutToken::BINDING:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500945 setFlag(Layout::kBinding_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500946 binding = this->layoutInt();
947 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700948 case LayoutToken::INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500949 setFlag(Layout::kIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500950 index = this->layoutInt();
951 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700952 case LayoutToken::SET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500953 setFlag(Layout::kSet_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500954 set = this->layoutInt();
955 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700956 case LayoutToken::BUILTIN:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500957 setFlag(Layout::kBuiltin_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500958 builtin = this->layoutInt();
959 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700960 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500961 setFlag(Layout::kInputAttachmentIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500962 inputAttachmentIndex = this->layoutInt();
963 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700964 case LayoutToken::POINTS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500965 setPrimitive(Layout::kPoints_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500966 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700967 case LayoutToken::LINES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500968 setPrimitive(Layout::kLines_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500969 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700970 case LayoutToken::LINE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500971 setPrimitive(Layout::kLineStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500972 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700973 case LayoutToken::LINES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500974 setPrimitive(Layout::kLinesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500975 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700976 case LayoutToken::TRIANGLES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500977 setPrimitive(Layout::kTriangles_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500978 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700979 case LayoutToken::TRIANGLE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500980 setPrimitive(Layout::kTriangleStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500981 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700982 case LayoutToken::TRIANGLES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500983 setPrimitive(Layout::kTrianglesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500984 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700985 case LayoutToken::MAX_VERTICES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500986 setFlag(Layout::kMaxVertices_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500987 maxVertices = this->layoutInt();
988 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700989 case LayoutToken::INVOCATIONS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500990 setFlag(Layout::kInvocations_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500991 invocations = this->layoutInt();
992 break;
Brian Osmanf59a9612020-04-15 14:18:13 -0400993 case LayoutToken::MARKER:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500994 setFlag(Layout::kMarker_Flag);
Brian Osmanf59a9612020-04-15 14:18:13 -0400995 marker = this->layoutCode();
996 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700997 case LayoutToken::WHEN:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500998 setFlag(Layout::kWhen_Flag);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400999 when = this->layoutCode();
1000 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -04001001 case LayoutToken::CTYPE:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001002 setFlag(Layout::kCType_Flag);
Ethan Nicholas78aceb22018-08-31 16:13:58 -04001003 ctype = this->layoutCType();
1004 break;
1005 default:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001006 this->error(t, "'" + text + "' is not a valid layout qualifier");
Ethan Nicholasd608c092017-10-26 09:30:08 -04001007 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -05001008 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001009 } else {
Brian Osmana77ed8b2021-02-23 12:54:22 -05001010 this->error(t, "'" + text + "' is not a valid layout qualifier");
ethannicholasb3058bd2016-07-01 08:22:01 -07001011 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001012 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001013 break;
1014 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001015 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001016 break;
1017 }
1018 }
1019 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -05001020 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
Brian Osmanba7ef322021-02-23 13:37:22 -05001021 primitive, maxVertices, invocations, marker, when, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -07001022}
1023
Brian Salomonf9f45122016-11-29 11:59:17 -05001024/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Brian Osman58ee8982021-02-18 15:39:38 -05001025 VARYING | 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