blob: eb5c107b29775befbbd5dbae68a492b6c4f33554 [file] [log] [blame]
ethannicholasb3058bd2016-07-01 08:22:01 -07001/*
2 * Copyright 2016 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
Ethan Nicholas11d53972016-11-28 11:23:23 -05007
Mike Klein4b432fa2019-06-06 11:44:05 -05008#include "src/sksl/SkSLParser.h"
John Stilesfbd050b2020-08-03 13:21:46 -04009
10#include <memory>
11#include "stdio.h"
12
Ethan Nicholasdaed2592021-03-04 14:30:25 -050013#include "include/private/SkSLModifiers.h"
John Stilesfbd050b2020-08-03 13:21:46 -040014#include "src/sksl/SkSLASTNode.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050015#include "src/sksl/ir/SkSLSymbolTable.h"
16#include "src/sksl/ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070017
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040018#ifndef SKSL_STANDALONE
Mike Kleinc0bd9f92019-04-23 12:05:21 -050019#include "include/private/SkOnce.h"
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040020#endif
21
ethannicholasb3058bd2016-07-01 08:22:01 -070022namespace SkSL {
23
John Stiles8d056592020-11-10 10:03:50 -050024static constexpr int kMaxParseDepth = 50;
John Stilesd39aec92020-12-03 14:37:16 -050025static constexpr int kMaxStructDepth = 8;
John Stilesa695d622020-11-10 10:04:26 -050026
27static bool struct_is_too_deeply_nested(const Type& type, int limit) {
28 if (limit < 0) {
29 return true;
30 }
31
John Stilesc0c51062020-12-03 17:16:29 -050032 if (type.isStruct()) {
John Stilesa695d622020-11-10 10:04:26 -050033 for (const Type::Field& f : type.fields()) {
34 if (struct_is_too_deeply_nested(*f.fType, limit - 1)) {
35 return true;
36 }
37 }
38 }
39
40 return false;
41}
ethannicholascad64162016-10-27 10:54:02 -070042
John Stiles7d5b90a2021-01-21 14:26:51 -050043static int parse_modifier_token(Token::Kind token) {
44 switch (token) {
45 case Token::Kind::TK_UNIFORM: return Modifiers::kUniform_Flag;
46 case Token::Kind::TK_CONST: return Modifiers::kConst_Flag;
47 case Token::Kind::TK_IN: return Modifiers::kIn_Flag;
48 case Token::Kind::TK_OUT: return Modifiers::kOut_Flag;
49 case Token::Kind::TK_INOUT: return Modifiers::kIn_Flag | Modifiers::kOut_Flag;
50 case Token::Kind::TK_FLAT: return Modifiers::kFlat_Flag;
51 case Token::Kind::TK_NOPERSPECTIVE: return Modifiers::kNoPerspective_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050052 case Token::Kind::TK_HASSIDEEFFECTS: return Modifiers::kHasSideEffects_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050053 case Token::Kind::TK_VARYING: return Modifiers::kVarying_Flag;
54 case Token::Kind::TK_INLINE: return Modifiers::kInline_Flag;
John Stiles0dd1a772021-03-09 22:14:27 -050055 case Token::Kind::TK_NOINLINE: return Modifiers::kNoInline_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050056 default: return 0;
57 }
58}
59
ethannicholascad64162016-10-27 10:54:02 -070060class AutoDepth {
61public:
62 AutoDepth(Parser* p)
Ethan Nicholascf4deab2019-09-13 16:28:14 -040063 : fParser(p)
64 , fDepth(0) {}
ethannicholascad64162016-10-27 10:54:02 -070065
66 ~AutoDepth() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -040067 fParser->fDepth -= fDepth;
ethannicholascad64162016-10-27 10:54:02 -070068 }
69
Ethan Nicholascf4deab2019-09-13 16:28:14 -040070 bool increase() {
71 ++fDepth;
72 ++fParser->fDepth;
John Stiles8d056592020-11-10 10:03:50 -050073 if (fParser->fDepth > kMaxParseDepth) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070074 fParser->error(fParser->peek(), String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070075 return false;
76 }
77 return true;
78 }
79
80private:
81 Parser* fParser;
Ethan Nicholascf4deab2019-09-13 16:28:14 -040082 int fDepth;
ethannicholascad64162016-10-27 10:54:02 -070083};
84
Brian Salomon140f3da2018-08-23 13:51:27 +000085std::unordered_map<String, Parser::LayoutToken>* Parser::layoutTokens;
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040086
87void Parser::InitLayoutMap() {
Brian Salomon140f3da2018-08-23 13:51:27 +000088 layoutTokens = new std::unordered_map<String, LayoutToken>;
Brian Salomon23356442018-11-30 15:33:19 -050089 #define TOKEN(name, text) (*layoutTokens)[text] = LayoutToken::name
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040090 TOKEN(LOCATION, "location");
91 TOKEN(OFFSET, "offset");
92 TOKEN(BINDING, "binding");
93 TOKEN(INDEX, "index");
94 TOKEN(SET, "set");
95 TOKEN(BUILTIN, "builtin");
96 TOKEN(INPUT_ATTACHMENT_INDEX, "input_attachment_index");
97 TOKEN(ORIGIN_UPPER_LEFT, "origin_upper_left");
98 TOKEN(OVERRIDE_COVERAGE, "override_coverage");
Brian Osmandc4f8cd2021-02-25 15:30:46 -050099 TOKEN(EARLY_FRAGMENT_TESTS, "early_fragment_tests");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400100 TOKEN(BLEND_SUPPORT_ALL_EQUATIONS, "blend_support_all_equations");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400101 TOKEN(PUSH_CONSTANT, "push_constant");
102 TOKEN(POINTS, "points");
103 TOKEN(LINES, "lines");
104 TOKEN(LINE_STRIP, "line_strip");
105 TOKEN(LINES_ADJACENCY, "lines_adjacency");
106 TOKEN(TRIANGLES, "triangles");
107 TOKEN(TRIANGLE_STRIP, "triangle_strip");
108 TOKEN(TRIANGLES_ADJACENCY, "triangles_adjacency");
109 TOKEN(MAX_VERTICES, "max_vertices");
110 TOKEN(INVOCATIONS, "invocations");
Brian Osmanf59a9612020-04-15 14:18:13 -0400111 TOKEN(MARKER, "marker");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400112 TOKEN(WHEN, "when");
113 TOKEN(KEY, "key");
Michael Ludwiga4275592018-08-31 10:52:47 -0400114 TOKEN(TRACKED, "tracked");
Brian Osmanb32d66b2020-04-30 17:12:03 -0400115 TOKEN(SRGB_UNPREMUL, "srgb_unpremul");
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400116 TOKEN(CTYPE, "ctype");
Brian Osmanf28e55d2018-10-03 16:35:54 -0400117 TOKEN(SKPMCOLOR4F, "SkPMColor4f");
Mike Reedb26b4e72020-01-22 14:31:21 -0500118 TOKEN(SKV4, "SkV4");
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400119 TOKEN(SKRECT, "SkRect");
120 TOKEN(SKIRECT, "SkIRect");
121 TOKEN(SKPMCOLOR, "SkPMColor");
Mike Reedb26b4e72020-01-22 14:31:21 -0500122 TOKEN(SKM44, "SkM44");
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400123 TOKEN(BOOL, "bool");
124 TOKEN(INT, "int");
125 TOKEN(FLOAT, "float");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400126 #undef TOKEN
127}
128
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400129Parser::Parser(const char* text, size_t length, SymbolTable& symbols, ErrorReporter& errors)
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700130: fText(text)
John Stiles7cbb09c22021-01-07 16:07:00 -0500131, fPushback(Token::Kind::TK_NONE, -1, -1)
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400132, fSymbols(symbols)
ethannicholasb3058bd2016-07-01 08:22:01 -0700133, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700134 fLexer.start(text, length);
Brian Salomon3b83afe2018-08-23 11:04:36 -0400135 static const bool layoutMapInitialized = []{ return (void)InitLayoutMap(), true; }();
136 (void) layoutMapInitialized;
ethannicholasb3058bd2016-07-01 08:22:01 -0700137}
138
John Stiles3b209362020-11-16 17:03:10 -0500139template <typename... Args>
140ASTNode::ID Parser::createNode(Args&&... args) {
141 ASTNode::ID result(fFile->fNodes.size());
142 fFile->fNodes.emplace_back(&fFile->fNodes, std::forward<Args>(args)...);
143 return result;
144}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400145
John Stiles3b209362020-11-16 17:03:10 -0500146ASTNode::ID Parser::addChild(ASTNode::ID target, ASTNode::ID child) {
147 fFile->fNodes[target.fValue].addChild(child);
148 return child;
149}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400150
John Stiles3b209362020-11-16 17:03:10 -0500151void Parser::createEmptyChild(ASTNode::ID target) {
152 ASTNode::ID child(fFile->fNodes.size());
153 fFile->fNodes.emplace_back();
154 fFile->fNodes[target.fValue].addChild(child);
155}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400156
Ethan Nicholasf7b88202017-09-18 14:10:39 -0400157/* (directive | section | declaration)* END_OF_FILE */
Ethan Nicholasba9a04f2020-11-06 09:28:04 -0500158std::unique_ptr<ASTFile> Parser::compilationUnit() {
John Stilesfbd050b2020-08-03 13:21:46 -0400159 fFile = std::make_unique<ASTFile>();
John Stiles3b209362020-11-16 17:03:10 -0500160 ASTNode::ID result = this->createNode(/*offset=*/0, ASTNode::Kind::kFile);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400161 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700162 for (;;) {
163 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400164 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400165 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400166 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400167 ASTNode::ID dir = this->directive();
168 if (fErrors.errorCount()) {
169 return nullptr;
170 }
171 if (dir) {
172 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700173 }
174 break;
175 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400176 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400177 ASTNode::ID section = this->section();
178 if (fErrors.errorCount()) {
179 return nullptr;
180 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400181 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400182 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400183 }
184 break;
185 }
John Stiles7cbb09c22021-01-07 16:07:00 -0500186 case Token::Kind::TK_INVALID: {
187 this->error(this->peek(), String("invalid token"));
188 return nullptr;
189 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700190 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400191 ASTNode::ID decl = this->declaration();
192 if (fErrors.errorCount()) {
193 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700194 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400195 if (decl) {
196 getNode(result).addChild(decl);
197 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700198 }
199 }
200 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400201 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700202}
203
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700204Token Parser::nextRawToken() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500205 if (fPushback.fKind != Token::Kind::TK_NONE) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700206 Token result = fPushback;
John Stiles7cbb09c22021-01-07 16:07:00 -0500207 fPushback.fKind = Token::Kind::TK_NONE;
ethannicholasb3058bd2016-07-01 08:22:01 -0700208 return result;
209 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700210 Token result = fLexer.next();
211 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400212}
213
214Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700215 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400216 while (token.fKind == Token::Kind::TK_WHITESPACE ||
217 token.fKind == Token::Kind::TK_LINE_COMMENT ||
218 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700219 token = this->nextRawToken();
220 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400221 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700222}
223
224void Parser::pushback(Token t) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500225 SkASSERT(fPushback.fKind == Token::Kind::TK_NONE);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400226 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700227}
228
229Token Parser::peek() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500230 if (fPushback.fKind == Token::Kind::TK_NONE) {
Brian Osman634624a2017-08-15 11:14:30 -0400231 fPushback = this->nextToken();
232 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700233 return fPushback;
234}
235
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400236bool Parser::checkNext(Token::Kind kind, Token* result) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500237 if (fPushback.fKind != Token::Kind::TK_NONE && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400238 return false;
239 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400240 Token next = this->nextToken();
241 if (next.fKind == kind) {
242 if (result) {
243 *result = next;
244 }
245 return true;
246 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400247 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400248 return false;
249}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500250
251bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700252 Token next = this->nextToken();
253 if (next.fKind == kind) {
254 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400255 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700256 }
257 return true;
258 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700259 this->error(next, "expected " + String(expected) + ", but found '" +
260 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700261 return false;
262 }
263}
264
John Stiles2630ea32020-12-04 10:51:21 -0500265bool Parser::expectIdentifier(Token* result) {
266 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", result)) {
267 return false;
268 }
269 if (this->isType(this->text(*result))) {
270 this->error(*result, "expected an identifier, but found type '" +
271 this->text(*result) + "'");
272 return false;
273 }
274 return true;
275}
276
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700277StringFragment Parser::text(Token token) {
278 return StringFragment(fText + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500279}
280
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700281void Parser::error(Token token, String msg) {
282 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700283}
284
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700285void Parser::error(int offset, String msg) {
286 fErrors.error(offset, msg);
287}
288
289bool Parser::isType(StringFragment name) {
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400290 const Symbol* s = fSymbols[name];
John Stiles2630ea32020-12-04 10:51:21 -0500291 return s && s->is<Type>();
ethannicholasb3058bd2016-07-01 08:22:01 -0700292}
293
John Stiles80b02af2021-02-12 17:07:51 -0500294bool Parser::isArrayType(ASTNode::ID type) {
295 const ASTNode& node = this->getNode(type);
296 SkASSERT(node.fKind == ASTNode::Kind::kType);
297 return node.begin() != node.end();
298}
299
Ethan Nicholas11d53972016-11-28 11:23:23 -0500300/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700301 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400302ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700303 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400304 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400305 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700306 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700307 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400308 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700309 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500310 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400311 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700312 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400313 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400314 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700315 }
316 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400317 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400318 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700319 }
John Stiles3b209362020-11-16 17:03:10 -0500320 return this->createNode(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700321 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700322 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400323 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700324 }
325}
326
Ethan Nicholas762466e2017-06-29 10:03:38 -0400327/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
328 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400329ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400330 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400331 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400332 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400333 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400334 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400335 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400336 this->nextToken();
337 Token argToken;
John Stiles2630ea32020-12-04 10:51:21 -0500338 if (!this->expectIdentifier(&argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400339 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400340 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700341 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400342 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400343 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400344 }
345 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400346 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400347 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400348 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400349 StringFragment text;
350 Token codeStart = this->nextRawToken();
351 size_t startOffset = codeStart.fOffset;
352 this->pushback(codeStart);
353 text.fChars = fText + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400354 int level = 1;
355 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700356 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400357 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400358 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400359 ++level;
360 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400361 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400362 --level;
363 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400364 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700365 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400366 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400367 default:
368 break;
369 }
370 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400371 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400372 break;
373 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400374 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700375 StringFragment name = this->text(start);
376 ++name.fChars;
377 --name.fLength;
John Stiles3b209362020-11-16 17:03:10 -0500378 return this->createNode(start.fOffset, ASTNode::Kind::kSection,
379 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400380}
381
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500382/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
383 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400384ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500385 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400386 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400387 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500388 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400389 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400390 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500391 }
392 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500393 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400394 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500395 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400396 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400397 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500398 }
John Stilese6c67c52021-01-15 12:01:00 -0500399 fSymbols.add(Type::MakeEnumType(this->text(name)));
John Stiles3b209362020-11-16 17:03:10 -0500400 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400401 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500402 Token id;
John Stiles2630ea32020-12-04 10:51:21 -0500403 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400404 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500405 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400406 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400407 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500408 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400409 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500410 }
John Stiles3b209362020-11-16 17:03:10 -0500411 ASTNode::ID child = this->addChild(
412 result, this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400413 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500414 } else {
John Stiles3b209362020-11-16 17:03:10 -0500415 this->addChild(result,
416 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500417 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400418 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
419 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400420 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500421 }
John Stiles2630ea32020-12-04 10:51:21 -0500422 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400423 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500424 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400425 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400426 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500427 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400428 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500429 }
John Stiles3b209362020-11-16 17:03:10 -0500430 ASTNode::ID child = this->addChild(
431 result,
432 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400433 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500434 } else {
John Stiles3b209362020-11-16 17:03:10 -0500435 this->addChild(
436 result,
437 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500438 }
439 }
440 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400441 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400442 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500443}
444
445/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700446 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400447ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700448 Token lookahead = this->peek();
Ethan Nicholasda6320c2020-09-02 14:08:23 -0400449 switch (lookahead.fKind) {
450 case Token::Kind::TK_ENUM:
451 return this->enumDeclaration();
452 case Token::Kind::TK_SEMICOLON:
453 this->error(lookahead.fOffset, "expected a declaration, but found ';'");
454 return ASTNode::ID::Invalid();
455 default:
456 break;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500457 }
458 Modifiers modifiers = this->modifiers();
459 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400460 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700461 // we have an identifier that's not a type, could be the start of an interface block
462 return this->interfaceBlock(modifiers);
463 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400464 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700465 return this->structVarDeclaration(modifiers);
466 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400467 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700468 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -0500469 return this->createNode(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700470 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400471 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700472 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400473 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700474 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700475 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500476 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400477 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700478 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400479 if (this->checkNext(Token::Kind::TK_LPAREN)) {
John Stiles3b209362020-11-16 17:03:10 -0500480 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kFunction);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400481 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
482 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400483 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400484 for (;;) {
485 ASTNode::ID parameter = this->parameter();
486 if (!parameter) {
487 return ASTNode::ID::Invalid();
488 }
489 ++fd.fParameterCount;
490 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400491 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400492 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700493 }
494 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700495 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400496 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400497 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400498 return ASTNode::ID::Invalid();
499 }
500 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400501 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700502 body = this->block();
503 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400504 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700505 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400506 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700507 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400508 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700509 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400510 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700511 }
512}
513
John Stiles76389b72021-01-25 13:49:05 -0500514/* (varDeclarations | expressionStatement) */
515ASTNode::ID Parser::varDeclarationsOrExpressionStatement() {
516 if (this->isType(this->text(this->peek()))) {
517 // Statements that begin with a typename are most often variable declarations, but
518 // occasionally the type is part of a constructor, and these are actually expression-
519 // statements in disguise. First, attempt the common case: parse it as a vardecl.
520 Checkpoint checkpoint(this);
John Stiles8dabeac2021-02-12 16:05:00 -0500521 VarDeclarationsPrefix prefix;
522 if (this->varDeclarationsPrefix(&prefix)) {
523 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
John Stiles76389b72021-01-25 13:49:05 -0500524 }
525
526 // If this statement wasn't actually a vardecl after all, rewind and try parsing it as an
527 // expression-statement instead.
528 checkpoint.rewind();
529 }
530
531 return this->expressionStatement();
532}
533
John Stiles8dabeac2021-02-12 16:05:00 -0500534// Helper function for varDeclarations(). If this function succeeds, we assume that the rest of the
535// statement is a variable-declaration statement, not an expression-statement.
536bool Parser::varDeclarationsPrefix(VarDeclarationsPrefix* prefixData) {
537 prefixData->modifiers = this->modifiers();
538 prefixData->type = this->type();
539 if (!prefixData->type) {
540 return false;
541 }
542 return this->expectIdentifier(&prefixData->name);
543}
544
ethannicholasb3058bd2016-07-01 08:22:01 -0700545/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400546ASTNode::ID Parser::varDeclarations() {
John Stiles8dabeac2021-02-12 16:05:00 -0500547 VarDeclarationsPrefix prefix;
548 if (!this->varDeclarationsPrefix(&prefix)) {
John Stiles76389b72021-01-25 13:49:05 -0500549 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700550 }
John Stiles8dabeac2021-02-12 16:05:00 -0500551 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700552}
553
554/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400555ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400556 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400557 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700558 }
559 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500560 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400561 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700562 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400563 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400564 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700565 }
566 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400567 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400568 ASTNode::ID decls = this->varDeclarations();
569 if (!decls) {
570 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700571 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400572 ASTNode& declsNode = getNode(decls);
John Stilesece1bf02021-03-08 11:15:55 -0500573 const Modifiers& modifiers = declsNode.begin()->getModifiers();
John Stiles13fc2602020-10-09 17:42:31 -0400574 if (modifiers.fFlags != Modifiers::kNo_Flag) {
575 String desc = modifiers.description();
576 desc.pop_back(); // remove trailing space
577 this->error(declsNode.fOffset,
578 "modifier '" + desc + "' is not permitted on a struct field");
579 }
580
Brian Osman00fea5b2021-01-28 09:46:30 -0500581 const Symbol* symbol = fSymbols[(declsNode.begin() + 1)->getString()];
John Stiles13fc2602020-10-09 17:42:31 -0400582 SkASSERT(symbol);
583 const Type* type = &symbol->as<Type>();
John Stiles1d757782020-11-17 09:43:22 -0500584 if (type->isOpaque()) {
585 this->error(declsNode.fOffset,
586 "opaque type '" + type->name() + "' is not permitted in a struct");
587 }
John Stiles13fc2602020-10-09 17:42:31 -0400588
Ethan Nicholasfc994162019-06-06 10:04:27 -0400589 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
590 ASTNode& var = *iter;
John Stilesece1bf02021-03-08 11:15:55 -0500591 const ASTNode::VarData& vd = var.getVarData();
John Stiles6bef6a72020-12-02 14:26:04 -0500592
John Stilesd39aec92020-12-03 14:37:16 -0500593 // Read array size if one is present.
594 if (vd.fIsArray) {
595 const ASTNode& size = *var.begin();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400596 if (!size || size.fKind != ASTNode::Kind::kInt) {
597 this->error(declsNode.fOffset, "array size in struct field must be a constant");
598 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700599 }
John Stiles6bef6a72020-12-02 14:26:04 -0500600 if (size.getInt() <= 0 || size.getInt() > INT_MAX) {
601 this->error(declsNode.fOffset, "array size is invalid");
602 return ASTNode::ID::Invalid();
603 }
John Stilesd39aec92020-12-03 14:37:16 -0500604 // Add the array dimensions to our type.
605 int arraySize = size.getInt();
John Stilesa2179502020-12-03 14:37:57 -0500606 type = fSymbols.addArrayDimension(type, arraySize);
ethannicholasb3058bd2016-07-01 08:22:01 -0700607 }
John Stiles13fc2602020-10-09 17:42:31 -0400608
609 fields.push_back(Type::Field(modifiers, vd.fName, type));
John Stilesd39aec92020-12-03 14:37:16 -0500610 if (vd.fIsArray ? var.begin()->fNext : var.fFirstChild) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400611 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700612 }
613 }
614 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400615 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400616 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700617 }
Brian Osman6c3b23f2021-02-12 13:09:58 -0500618 if (fields.empty()) {
619 this->error(name.fOffset,
620 "struct '" + this->text(name) + "' must contain at least one field");
621 return ASTNode::ID::Invalid();
622 }
John Stilesad2d4942020-12-11 16:55:58 -0500623 std::unique_ptr<Type> newType = Type::MakeStructType(name.fOffset, this->text(name), fields);
John Stilesa695d622020-11-10 10:04:26 -0500624 if (struct_is_too_deeply_nested(*newType, kMaxStructDepth)) {
625 this->error(name.fOffset, "struct '" + this->text(name) + "' is too deeply nested");
626 return ASTNode::ID::Invalid();
627 }
628 fSymbols.add(std::move(newType));
Brian Osman00fea5b2021-01-28 09:46:30 -0500629 return this->createNode(name.fOffset, ASTNode::Kind::kType, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700630}
631
632/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400633ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
634 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700635 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400636 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700637 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400638 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400639 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400640 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700641 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400642 this->expect(Token::Kind::TK_SEMICOLON, "';'");
John Stilesdc75a972020-11-25 16:24:55 -0500643 return type;
ethannicholasb3058bd2016-07-01 08:22:01 -0700644}
645
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400646/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
647 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400648ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
John Stiles08070f62020-11-17 10:45:49 -0500649 int offset = this->peek().fOffset;
650 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kVarDeclarations);
651 this->addChild(result, this->createNode(offset, ASTNode::Kind::kModifiers, mods));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400652 getNode(result).addChild(type);
John Stilesd39aec92020-12-03 14:37:16 -0500653
John Stiles80b02af2021-02-12 17:07:51 -0500654 auto parseArrayDimensions = [&](ASTNode::ID currentVar, ASTNode::VarData* vd) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400655 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500656 if (vd->fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500657 this->error(this->peek(), "multi-dimensional arrays are not supported");
658 return false;
659 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400660 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
John Stiles3b209362020-11-16 17:03:10 -0500661 this->createEmptyChild(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700662 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400663 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700664 if (!size) {
John Stilesd39aec92020-12-03 14:37:16 -0500665 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700666 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400667 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400668 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
John Stilesd39aec92020-12-03 14:37:16 -0500669 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700670 }
671 }
John Stilesd39aec92020-12-03 14:37:16 -0500672 vd->fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700673 }
John Stilesd39aec92020-12-03 14:37:16 -0500674 return true;
675 };
676
677 auto parseInitializer = [this](ASTNode::ID currentVar) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400678 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400679 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700680 if (!value) {
John Stilesd39aec92020-12-03 14:37:16 -0500681 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700682 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400683 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700684 }
John Stilesd39aec92020-12-03 14:37:16 -0500685 return true;
686 };
687
688 ASTNode::ID currentVar = this->createNode(offset, ASTNode::Kind::kVarDeclaration);
689 ASTNode::VarData vd{name, /*isArray=*/false};
690
691 getNode(result).addChild(currentVar);
692 if (!parseArrayDimensions(currentVar, &vd)) {
693 return ASTNode::ID::Invalid();
694 }
695 getNode(currentVar).setVarData(vd);
696 if (!parseInitializer(currentVar)) {
697 return ASTNode::ID::Invalid();
698 }
699
700 while (this->checkNext(Token::Kind::TK_COMMA)) {
701 Token identifierName;
John Stiles2630ea32020-12-04 10:51:21 -0500702 if (!this->expectIdentifier(&identifierName)) {
John Stilesd39aec92020-12-03 14:37:16 -0500703 return ASTNode::ID::Invalid();
704 }
705
706 currentVar = ASTNode::ID(fFile->fNodes.size());
707 vd = ASTNode::VarData{this->text(identifierName), /*isArray=*/false};
708 fFile->fNodes.emplace_back(&fFile->fNodes, offset, ASTNode::Kind::kVarDeclaration);
709
710 getNode(result).addChild(currentVar);
711 if (!parseArrayDimensions(currentVar, &vd)) {
712 return ASTNode::ID::Invalid();
713 }
714 getNode(currentVar).setVarData(vd);
715 if (!parseInitializer(currentVar)) {
716 return ASTNode::ID::Invalid();
717 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700718 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400719 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400720 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700721 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400722 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700723}
724
725/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400726ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400727 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400728 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700729 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400730 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700731 }
732 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500733 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400734 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700735 }
John Stiles3b209362020-11-16 17:03:10 -0500736 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kParameter);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400737 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
738 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400739 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500740 if (pd.fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500741 this->error(this->peek(), "multi-dimensional arrays are not supported");
742 return ASTNode::ID::Invalid();
743 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700744 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400745 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400746 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700747 }
John Stilesf94348f2020-12-23 18:58:43 -0500748 StringFragment arraySizeFrag = this->text(sizeToken);
749 SKSL_INT arraySize;
750 if (!SkSL::stoi(arraySizeFrag, &arraySize)) {
751 this->error(sizeToken, "array size is too large: " + arraySizeFrag);
752 return ASTNode::ID::Invalid();
753 }
754 this->addChild(result, this->createNode(sizeToken.fOffset, ASTNode::Kind::kInt, arraySize));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400755 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400756 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700757 }
John Stilesd39aec92020-12-03 14:37:16 -0500758 pd.fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700759 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400760 getNode(result).setParameterData(pd);
761 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700762}
763
Ethan Nicholasd608c092017-10-26 09:30:08 -0400764/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700765int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400766 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700767 return -1;
768 }
769 Token resultToken;
John Stilesf94348f2020-12-23 18:58:43 -0500770 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
771 return -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700772 }
John Stilesf94348f2020-12-23 18:58:43 -0500773 StringFragment resultFrag = this->text(resultToken);
774 SKSL_INT resultValue;
775 if (!SkSL::stoi(resultFrag, &resultValue)) {
776 this->error(resultToken, "value in layout is too large: " + resultFrag);
777 return -1;
778 }
779 return resultValue;
ethannicholasb3058bd2016-07-01 08:22:01 -0700780}
781
Ethan Nicholasd608c092017-10-26 09:30:08 -0400782/** EQ IDENTIFIER */
783StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400784 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400785 return StringFragment();
786 }
787 Token resultToken;
John Stiles2630ea32020-12-04 10:51:21 -0500788 if (!this->expectIdentifier(&resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400789 return StringFragment();
790 }
791 return this->text(resultToken);
792}
793
794
Ethan Nicholas762466e2017-06-29 10:03:38 -0400795/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400796StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400797 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400798 return "";
799 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700800 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400801 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400802 StringFragment code;
803 code.fChars = fText + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400804 int level = 1;
805 bool done = false;
806 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700807 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400808 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400809 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400810 ++level;
811 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400812 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400813 --level;
814 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400815 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400816 if (level == 1) {
817 done = true;
818 }
819 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400820 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700821 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400822 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400823 default:
824 break;
825 }
826 if (!level) {
827 done = true;
828 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400829 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400830 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400831 this->pushback(std::move(next));
832 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400833 }
834 return code;
835}
836
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400837Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400838 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400839 Token t = this->nextToken();
840 String text = this->text(t);
841 auto found = layoutTokens->find(text);
842 if (found != layoutTokens->end()) {
843 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400844 case LayoutToken::SKPMCOLOR4F:
845 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500846 case LayoutToken::SKV4:
847 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400848 case LayoutToken::SKRECT:
849 return Layout::CType::kSkRect;
850 case LayoutToken::SKIRECT:
851 return Layout::CType::kSkIRect;
852 case LayoutToken::SKPMCOLOR:
853 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400854 case LayoutToken::BOOL:
855 return Layout::CType::kBool;
856 case LayoutToken::INT:
857 return Layout::CType::kInt32;
858 case LayoutToken::FLOAT:
859 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500860 case LayoutToken::SKM44:
861 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400862 default:
863 break;
864 }
865 }
866 this->error(t, "unsupported ctype");
867 }
868 return Layout::CType::kDefault;
869}
870
ethannicholas8ac838d2016-11-22 08:39:36 -0800871/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500872Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500873 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700874 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500875 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700876 int binding = -1;
877 int index = -1;
878 int set = -1;
879 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500880 int inputAttachmentIndex = -1;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500881 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
882 int maxVertices = -1;
883 int invocations = -1;
Brian Osmanf59a9612020-04-15 14:18:13 -0400884 StringFragment marker;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400885 StringFragment when;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400886 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400887 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
888 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500889 return Layout(flags, location, offset, binding, index, set, builtin,
Brian Osman4717fbb2021-02-23 13:12:09 -0500890 inputAttachmentIndex, primitive, maxVertices, invocations, marker, when,
Brian Osmanba7ef322021-02-23 13:37:22 -0500891 ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700892 }
893 for (;;) {
894 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700895 String text = this->text(t);
Brian Osmana77ed8b2021-02-23 12:54:22 -0500896 auto setFlag = [&](Layout::Flag f) {
897 if (flags & f) {
898 this->error(t, "layout qualifier '" + text + "' appears more than once");
899 }
900 flags |= f;
901 };
902 auto setPrimitive = [&](Layout::Primitive p) {
903 if (flags & Layout::kPrimitive_Flag) {
904 this->error(t, "only one primitive-type layout qualifier is allowed");
905 }
906 flags |= Layout::kPrimitive_Flag;
907 primitive = p;
908 };
909
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400910 auto found = layoutTokens->find(text);
911 if (found != layoutTokens->end()) {
912 switch (found->second) {
Brian Osmana77ed8b2021-02-23 12:54:22 -0500913 case LayoutToken::ORIGIN_UPPER_LEFT:
914 setFlag(Layout::kOriginUpperLeft_Flag);
915 break;
916 case LayoutToken::OVERRIDE_COVERAGE:
917 setFlag(Layout::kOverrideCoverage_Flag);
918 break;
Brian Osmandc4f8cd2021-02-25 15:30:46 -0500919 case LayoutToken::EARLY_FRAGMENT_TESTS:
920 setFlag(Layout::kEarlyFragmentTests_Flag);
921 break;
Brian Osmana77ed8b2021-02-23 12:54:22 -0500922 case LayoutToken::PUSH_CONSTANT:
923 setFlag(Layout::kPushConstant_Flag);
924 break;
925 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
926 setFlag(Layout::kBlendSupportAllEquations_Flag);
927 break;
928 case LayoutToken::TRACKED:
929 setFlag(Layout::kTracked_Flag);
930 break;
931 case LayoutToken::SRGB_UNPREMUL:
932 setFlag(Layout::kSRGBUnpremul_Flag);
933 break;
934 case LayoutToken::KEY:
935 setFlag(Layout::kKey_Flag);
936 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700937 case LayoutToken::LOCATION:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500938 setFlag(Layout::kLocation_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500939 location = this->layoutInt();
940 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700941 case LayoutToken::OFFSET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500942 setFlag(Layout::kOffset_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500943 offset = this->layoutInt();
944 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700945 case LayoutToken::BINDING:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500946 setFlag(Layout::kBinding_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500947 binding = this->layoutInt();
948 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700949 case LayoutToken::INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500950 setFlag(Layout::kIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500951 index = this->layoutInt();
952 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700953 case LayoutToken::SET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500954 setFlag(Layout::kSet_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500955 set = this->layoutInt();
956 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700957 case LayoutToken::BUILTIN:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500958 setFlag(Layout::kBuiltin_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500959 builtin = this->layoutInt();
960 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700961 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500962 setFlag(Layout::kInputAttachmentIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500963 inputAttachmentIndex = this->layoutInt();
964 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700965 case LayoutToken::POINTS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500966 setPrimitive(Layout::kPoints_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500967 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700968 case LayoutToken::LINES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500969 setPrimitive(Layout::kLines_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500970 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700971 case LayoutToken::LINE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500972 setPrimitive(Layout::kLineStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500973 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700974 case LayoutToken::LINES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500975 setPrimitive(Layout::kLinesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500976 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700977 case LayoutToken::TRIANGLES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500978 setPrimitive(Layout::kTriangles_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500979 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700980 case LayoutToken::TRIANGLE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500981 setPrimitive(Layout::kTriangleStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500982 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700983 case LayoutToken::TRIANGLES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500984 setPrimitive(Layout::kTrianglesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500985 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700986 case LayoutToken::MAX_VERTICES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500987 setFlag(Layout::kMaxVertices_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500988 maxVertices = this->layoutInt();
989 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700990 case LayoutToken::INVOCATIONS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500991 setFlag(Layout::kInvocations_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500992 invocations = this->layoutInt();
993 break;
Brian Osmanf59a9612020-04-15 14:18:13 -0400994 case LayoutToken::MARKER:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500995 setFlag(Layout::kMarker_Flag);
Brian Osmanf59a9612020-04-15 14:18:13 -0400996 marker = this->layoutCode();
997 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700998 case LayoutToken::WHEN:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500999 setFlag(Layout::kWhen_Flag);
Ethan Nicholas762466e2017-06-29 10:03:38 -04001000 when = this->layoutCode();
1001 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -04001002 case LayoutToken::CTYPE:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001003 setFlag(Layout::kCType_Flag);
Ethan Nicholas78aceb22018-08-31 16:13:58 -04001004 ctype = this->layoutCType();
1005 break;
1006 default:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001007 this->error(t, "'" + text + "' is not a valid layout qualifier");
Ethan Nicholasd608c092017-10-26 09:30:08 -04001008 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -05001009 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001010 } else {
Brian Osmana77ed8b2021-02-23 12:54:22 -05001011 this->error(t, "'" + text + "' is not a valid layout qualifier");
ethannicholasb3058bd2016-07-01 08:22:01 -07001012 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001013 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001014 break;
1015 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001016 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001017 break;
1018 }
1019 }
1020 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -05001021 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
Brian Osmanba7ef322021-02-23 13:37:22 -05001022 primitive, maxVertices, invocations, marker, when, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -07001023}
1024
Brian Salomonf9f45122016-11-29 11:59:17 -05001025/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Brian Osman58ee8982021-02-18 15:39:38 -05001026 VARYING | INLINE)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -05001027Modifiers Parser::modifiers() {
1028 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -07001029 int flags = 0;
1030 for (;;) {
1031 // TODO: handle duplicate / incompatible flags
John Stiles7d5b90a2021-01-21 14:26:51 -05001032 int tokenFlag = parse_modifier_token(peek().fKind);
1033 if (!tokenFlag) {
1034 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001035 }
John Stiles7d5b90a2021-01-21 14:26:51 -05001036 flags |= tokenFlag;
1037 this->nextToken();
ethannicholasb3058bd2016-07-01 08:22:01 -07001038 }
John Stiles7d5b90a2021-01-21 14:26:51 -05001039 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001040}
1041
Ethan Nicholas11d53972016-11-28 11:23:23 -05001042Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
1043 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -07001044 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001045 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001046 }
1047 return result;
1048}
1049
1050/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001051ASTNode::ID Parser::statement() {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001052 Token start = this->nextToken();
1053 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001054 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001055 return ASTNode::ID::Invalid();
1056 }
1057 this->pushback(start);
ethannicholasb3058bd2016-07-01 08:22:01 -07001058 switch (start.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001059 case Token::Kind::TK_IF: // fall through
1060 case Token::Kind::TK_STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -07001061 return this->ifStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001062 case Token::Kind::TK_FOR:
ethannicholasb3058bd2016-07-01 08:22:01 -07001063 return this->forStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001064 case Token::Kind::TK_DO:
ethannicholasb3058bd2016-07-01 08:22:01 -07001065 return this->doStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001066 case Token::Kind::TK_WHILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001067 return this->whileStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001068 case Token::Kind::TK_SWITCH: // fall through
1069 case Token::Kind::TK_STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -05001070 return this->switchStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001071 case Token::Kind::TK_RETURN:
ethannicholasb3058bd2016-07-01 08:22:01 -07001072 return this->returnStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001073 case Token::Kind::TK_BREAK:
ethannicholasb3058bd2016-07-01 08:22:01 -07001074 return this->breakStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001075 case Token::Kind::TK_CONTINUE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001076 return this->continueStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001077 case Token::Kind::TK_DISCARD:
ethannicholasb3058bd2016-07-01 08:22:01 -07001078 return this->discardStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001079 case Token::Kind::TK_LBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001080 return this->block();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001081 case Token::Kind::TK_SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001082 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -05001083 return this->createNode(start.fOffset, ASTNode::Kind::kBlock);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001084 case Token::Kind::TK_CONST:
Ethan Nicholasfc994162019-06-06 10:04:27 -04001085 return this->varDeclarations();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001086 case Token::Kind::TK_IDENTIFIER:
John Stiles76389b72021-01-25 13:49:05 -05001087 return this->varDeclarationsOrExpressionStatement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001088 default:
1089 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001090 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001091}
1092
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001093/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* QUESTION? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001094ASTNode::ID Parser::type() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001095 Token type;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001096 if (!this->expect(Token::Kind::TK_IDENTIFIER, "a type", &type)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001097 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001098 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001099 if (!this->isType(this->text(type))) {
1100 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001101 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001102 }
Brian Osman00fea5b2021-01-28 09:46:30 -05001103 ASTNode::ID result = this->createNode(type.fOffset, ASTNode::Kind::kType, this->text(type));
John Stilesd39aec92020-12-03 14:37:16 -05001104 bool isArray = false;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001105 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stilesd39aec92020-12-03 14:37:16 -05001106 if (isArray) {
1107 this->error(this->peek(), "multi-dimensional arrays are not supported");
1108 return ASTNode::ID::Invalid();
1109 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001110 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001111 SKSL_INT i;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001112 if (this->intLiteral(&i)) {
John Stiles08070f62020-11-17 10:45:49 -05001113 this->addChild(result, this->createNode(this->peek().fOffset,
1114 ASTNode::Kind::kInt, i));
Ethan Nicholas50afc172017-02-16 14:49:57 -05001115 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001116 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001117 }
1118 } else {
John Stiles3b209362020-11-16 17:03:10 -05001119 this->createEmptyChild(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001120 }
John Stilesd39aec92020-12-03 14:37:16 -05001121 isArray = true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001122 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001123 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001124 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001125}
1126
John Stiles57a996b2020-04-19 19:05:12 -07001127/* IDENTIFIER LBRACE
1128 varDeclaration+
1129 RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001130ASTNode::ID Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001131 Token name;
John Stiles2630ea32020-12-04 10:51:21 -05001132 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001133 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001134 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001135 if (peek().fKind != Token::Kind::TK_LBRACE) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001136 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001137 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -07001138 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001139 this->error(name, "no type named '" + this->text(name) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001140 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001141 }
John Stiles3b209362020-11-16 17:03:10 -05001142 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kInterfaceBlock);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001143 ASTNode::InterfaceBlockData id(mods, this->text(name), 0, "", 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07001144 this->nextToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001145 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001146 ASTNode::ID decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001147 if (!decl) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001148 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001149 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001150 getNode(result).addChild(decl);
1151 ++id.fDeclarationCount;
ethannicholasb3058bd2016-07-01 08:22:01 -07001152 }
John Stiles57a996b2020-04-19 19:05:12 -07001153 if (id.fDeclarationCount == 0) {
1154 this->error(name, "interface block '" + this->text(name) +
1155 "' must contain at least one member");
1156 return ASTNode::ID::Invalid();
1157 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001158 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001159 std::vector<ASTNode> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001160 StringFragment instanceName;
1161 Token instanceNameToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001162 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &instanceNameToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001163 id.fInstanceName = this->text(instanceNameToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001164 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stilesd39aec92020-12-03 14:37:16 -05001165 if (id.fIsArray) {
1166 this->error(this->peek(), "multi-dimensional arrays are not supported");
1167 return false;
1168 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001169 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001170 ASTNode::ID size = this->expression();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001171 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001172 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001173 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001174 getNode(result).addChild(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001175 } else {
John Stiles3b209362020-11-16 17:03:10 -05001176 this->createEmptyChild(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001177 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001178 this->expect(Token::Kind::TK_RBRACKET, "']'");
John Stilesd39aec92020-12-03 14:37:16 -05001179 id.fIsArray = true;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001180 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001181 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -07001182 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001183 getNode(result).setInterfaceBlockData(id);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001184 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001185 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001186}
1187
1188/* IF LPAREN expression RPAREN statement (ELSE statement)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001189ASTNode::ID Parser::ifStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001190 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001191 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_IF, &start);
1192 if (!isStatic && !this->expect(Token::Kind::TK_IF, "'if'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001193 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001194 }
John Stiles3b209362020-11-16 17:03:10 -05001195 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kIf, isStatic);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001196 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001197 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001198 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001199 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001200 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001201 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001202 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001203 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001204 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001205 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001206 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001207 ASTNode::ID ifTrue = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001208 if (!ifTrue) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001209 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001210 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001211 getNode(result).addChild(ifTrue);
1212 ASTNode::ID ifFalse;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001213 if (this->checkNext(Token::Kind::TK_ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001214 ifFalse = this->statement();
1215 if (!ifFalse) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001216 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001217 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001218 getNode(result).addChild(ifFalse);
ethannicholasb3058bd2016-07-01 08:22:01 -07001219 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001220 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001221}
1222
1223/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001224ASTNode::ID Parser::doStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001225 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001226 if (!this->expect(Token::Kind::TK_DO, "'do'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001227 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001228 }
John Stiles3b209362020-11-16 17:03:10 -05001229 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kDo);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001230 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001231 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001232 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001233 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001234 getNode(result).addChild(statement);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001235 if (!this->expect(Token::Kind::TK_WHILE, "'while'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001236 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001237 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001238 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001239 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001240 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001241 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001242 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001243 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001244 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001245 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001246 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001247 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001248 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001249 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001250 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001251 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001252 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001253}
1254
1255/* WHILE LPAREN expression RPAREN STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001256ASTNode::ID Parser::whileStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001257 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001258 if (!this->expect(Token::Kind::TK_WHILE, "'while'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001259 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001260 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001261 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001262 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001263 }
John Stiles3b209362020-11-16 17:03:10 -05001264 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kWhile);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001265 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001266 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001267 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001268 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001269 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001270 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001271 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001272 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001273 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001274 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001275 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001276 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001277 getNode(result).addChild(statement);
1278 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001279}
1280
Ethan Nicholasaf197692017-02-27 13:26:45 -05001281/* CASE expression COLON statement* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001282ASTNode::ID Parser::switchCase() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001283 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001284 if (!this->expect(Token::Kind::TK_CASE, "'case'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001285 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001286 }
John Stiles3b209362020-11-16 17:03:10 -05001287 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kSwitchCase);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001288 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001289 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001290 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001291 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001292 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001293 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001294 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001295 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001296 while (this->peek().fKind != Token::Kind::TK_RBRACE &&
1297 this->peek().fKind != Token::Kind::TK_CASE &&
1298 this->peek().fKind != Token::Kind::TK_DEFAULT) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001299 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001300 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001301 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001302 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001303 getNode(result).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001304 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001305 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001306}
1307
1308/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001309ASTNode::ID Parser::switchStatement() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001310 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001311 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_SWITCH, &start);
1312 if (!isStatic && !this->expect(Token::Kind::TK_SWITCH, "'switch'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001313 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001314 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001315 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001316 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001317 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001318 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001319 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001320 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001321 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001322 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001323 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001324 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001325 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001326 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001327 }
John Stiles3b209362020-11-16 17:03:10 -05001328 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kSwitch, isStatic);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001329 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001330 while (this->peek().fKind == Token::Kind::TK_CASE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001331 ASTNode::ID c = this->switchCase();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001332 if (!c) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001333 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001334 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001335 getNode(result).addChild(c);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001336 }
1337 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1338 // parts of the compiler may rely upon this assumption.
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001339 if (this->peek().fKind == Token::Kind::TK_DEFAULT) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001340 Token defaultStart;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001341 SkAssertResult(this->expect(Token::Kind::TK_DEFAULT, "'default'", &defaultStart));
1342 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001343 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001344 }
John Stiles3b209362020-11-16 17:03:10 -05001345 ASTNode::ID defaultCase = this->addChild(
1346 result, this->createNode(defaultStart.fOffset, ASTNode::Kind::kSwitchCase));
1347 this->createEmptyChild(defaultCase); // empty test to signify default case
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001348 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001349 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001350 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001351 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001352 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001353 getNode(defaultCase).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001354 }
Ethan Nicholasaf197692017-02-27 13:26:45 -05001355 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001356 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001357 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001358 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001359 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001360}
1361
Ethan Nicholas11d53972016-11-28 11:23:23 -05001362/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001363 STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001364ASTNode::ID Parser::forStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001365 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001366 if (!this->expect(Token::Kind::TK_FOR, "'for'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001367 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001368 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001369 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001370 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001371 }
John Stiles3b209362020-11-16 17:03:10 -05001372 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kFor);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001373 ASTNode::ID initializer;
ethannicholasb3058bd2016-07-01 08:22:01 -07001374 Token nextToken = this->peek();
1375 switch (nextToken.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001376 case Token::Kind::TK_SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001377 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -05001378 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001379 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001380 case Token::Kind::TK_CONST: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001381 initializer = this->varDeclarations();
1382 if (!initializer) {
1383 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001384 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001385 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001386 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001387 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001388 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001389 if (this->isType(this->text(nextToken))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001390 initializer = this->varDeclarations();
1391 if (!initializer) {
1392 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001393 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001394 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001395 break;
1396 }
John Stiles30212b72020-06-11 17:55:07 -04001397 [[fallthrough]];
1398 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001399 default:
1400 initializer = this->expressionStatement();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001401 if (!initializer) {
1402 return ASTNode::ID::Invalid();
1403 }
1404 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001405 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001406 ASTNode::ID test;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001407 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001408 test = this->expression();
1409 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001410 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001411 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001412 getNode(result).addChild(test);
1413 } else {
John Stiles3b209362020-11-16 17:03:10 -05001414 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001415 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001416 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001417 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001418 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001419 ASTNode::ID next;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001420 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001421 next = this->expression();
1422 if (!next) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001423 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001424 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001425 getNode(result).addChild(next);
1426 } else {
John Stiles3b209362020-11-16 17:03:10 -05001427 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001428 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001429 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001430 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001431 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001432 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001433 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001434 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001435 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001436 getNode(result).addChild(statement);
1437 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001438}
1439
1440/* RETURN expression? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001441ASTNode::ID Parser::returnStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001442 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001443 if (!this->expect(Token::Kind::TK_RETURN, "'return'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001444 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001445 }
John Stiles3b209362020-11-16 17:03:10 -05001446 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kReturn);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001447 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001448 ASTNode::ID expression = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001449 if (!expression) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001450 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001451 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001452 getNode(result).addChild(expression);
ethannicholasb3058bd2016-07-01 08:22:01 -07001453 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001454 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001455 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001456 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001457 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001458}
1459
1460/* BREAK SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001461ASTNode::ID Parser::breakStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001462 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001463 if (!this->expect(Token::Kind::TK_BREAK, "'break'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001464 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001465 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001466 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001467 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001468 }
John Stiles3b209362020-11-16 17:03:10 -05001469 return this->createNode(start.fOffset, ASTNode::Kind::kBreak);
ethannicholasb3058bd2016-07-01 08:22:01 -07001470}
1471
1472/* CONTINUE SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001473ASTNode::ID Parser::continueStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001474 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001475 if (!this->expect(Token::Kind::TK_CONTINUE, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001476 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001477 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001478 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001479 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001480 }
John Stiles3b209362020-11-16 17:03:10 -05001481 return this->createNode(start.fOffset, ASTNode::Kind::kContinue);
ethannicholasb3058bd2016-07-01 08:22:01 -07001482}
1483
1484/* DISCARD SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001485ASTNode::ID Parser::discardStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001486 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001487 if (!this->expect(Token::Kind::TK_DISCARD, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001488 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001489 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001490 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001491 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001492 }
John Stiles3b209362020-11-16 17:03:10 -05001493 return this->createNode(start.fOffset, ASTNode::Kind::kDiscard);
ethannicholasb3058bd2016-07-01 08:22:01 -07001494}
1495
1496/* LBRACE statement* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001497ASTNode::ID Parser::block() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001498 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001499 if (!this->expect(Token::Kind::TK_LBRACE, "'{'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001500 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001501 }
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001502 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001503 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001504 return ASTNode::ID::Invalid();
1505 }
John Stiles3b209362020-11-16 17:03:10 -05001506 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -07001507 for (;;) {
1508 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001509 case Token::Kind::TK_RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001510 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001511 return result;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001512 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001513 this->error(this->peek(), "expected '}', but found end of file");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001514 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001515 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001516 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001517 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001518 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001519 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001520 getNode(result).addChild(statement);
ethannicholasb3058bd2016-07-01 08:22:01 -07001521 }
1522 }
1523 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001524 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001525}
1526
1527/* expression SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001528ASTNode::ID Parser::expressionStatement() {
1529 ASTNode::ID expr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001530 if (expr) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001531 if (this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001532 return expr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001533 }
1534 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001535 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001536}
1537
1538/* assignmentExpression (COMMA assignmentExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001539ASTNode::ID Parser::expression() {
1540 ASTNode::ID result = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001541 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001542 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001543 }
1544 Token t;
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001545 AutoDepth depth(this);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001546 while (this->checkNext(Token::Kind::TK_COMMA, &t)) {
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001547 if (!depth.increase()) {
1548 return ASTNode::ID::Invalid();
1549 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001550 ASTNode::ID right = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001551 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001552 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001553 }
John Stiles45990502021-02-16 10:55:27 -05001554 ASTNode::ID newResult = this->createNode(t.fOffset, ASTNode::Kind::kBinary,
1555 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001556 getNode(newResult).addChild(result);
1557 getNode(newResult).addChild(right);
1558 result = newResult;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001559 }
1560 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001561}
1562
1563/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1564 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1565 assignmentExpression)*
1566 */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001567ASTNode::ID Parser::assignmentExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001568 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001569 ASTNode::ID result = this->ternaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001570 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001571 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001572 }
1573 for (;;) {
1574 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001575 case Token::Kind::TK_EQ: // fall through
1576 case Token::Kind::TK_STAREQ: // fall through
1577 case Token::Kind::TK_SLASHEQ: // fall through
1578 case Token::Kind::TK_PERCENTEQ: // fall through
1579 case Token::Kind::TK_PLUSEQ: // fall through
1580 case Token::Kind::TK_MINUSEQ: // fall through
1581 case Token::Kind::TK_SHLEQ: // fall through
1582 case Token::Kind::TK_SHREQ: // fall through
1583 case Token::Kind::TK_BITWISEANDEQ: // fall through
1584 case Token::Kind::TK_BITWISEXOREQ: // fall through
John Stiles8b3b1592020-11-23 11:06:44 -05001585 case Token::Kind::TK_BITWISEOREQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001586 if (!depth.increase()) {
1587 return ASTNode::ID::Invalid();
1588 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001589 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001590 ASTNode::ID right = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001591 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001592 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001593 }
John Stiles3b209362020-11-16 17:03:10 -05001594 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001595 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001596 getNode(newResult).addChild(result);
1597 getNode(newResult).addChild(right);
1598 result = newResult;
1599 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001600 }
1601 default:
1602 return result;
1603 }
1604 }
1605}
1606
1607/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001608ASTNode::ID Parser::ternaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001609 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001610 ASTNode::ID base = this->logicalOrExpression();
1611 if (!base) {
1612 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001613 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001614 if (this->checkNext(Token::Kind::TK_QUESTION)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001615 if (!depth.increase()) {
1616 return ASTNode::ID::Invalid();
1617 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001618 ASTNode::ID trueExpr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001619 if (!trueExpr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001620 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001621 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001622 if (this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001623 ASTNode::ID falseExpr = this->assignmentExpression();
1624 if (!falseExpr) {
1625 return ASTNode::ID::Invalid();
1626 }
John Stiles3b209362020-11-16 17:03:10 -05001627 ASTNode::ID ternary = this->createNode(getNode(base).fOffset, ASTNode::Kind::kTernary);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001628 getNode(ternary).addChild(base);
1629 getNode(ternary).addChild(trueExpr);
1630 getNode(ternary).addChild(falseExpr);
1631 return ternary;
ethannicholasb3058bd2016-07-01 08:22:01 -07001632 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001633 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001634 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001635 return base;
ethannicholasb3058bd2016-07-01 08:22:01 -07001636}
1637
1638/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001639ASTNode::ID Parser::logicalOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001640 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001641 ASTNode::ID result = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001642 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001643 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001644 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001645 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001646 while (this->checkNext(Token::Kind::TK_LOGICALOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001647 if (!depth.increase()) {
1648 return ASTNode::ID::Invalid();
1649 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001650 ASTNode::ID right = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001651 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001652 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001653 }
John Stiles3b209362020-11-16 17:03:10 -05001654 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001655 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001656 getNode(newResult).addChild(result);
1657 getNode(newResult).addChild(right);
1658 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001659 }
1660 return result;
1661}
1662
1663/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001664ASTNode::ID Parser::logicalXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001665 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001666 ASTNode::ID result = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001667 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001668 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001669 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001670 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001671 while (this->checkNext(Token::Kind::TK_LOGICALXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001672 if (!depth.increase()) {
1673 return ASTNode::ID::Invalid();
1674 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001675 ASTNode::ID right = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001676 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001677 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001678 }
John Stiles3b209362020-11-16 17:03:10 -05001679 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001680 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001681 getNode(newResult).addChild(result);
1682 getNode(newResult).addChild(right);
1683 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001684 }
1685 return result;
1686}
1687
1688/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001689ASTNode::ID Parser::logicalAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001690 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001691 ASTNode::ID result = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001692 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001693 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001694 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001695 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001696 while (this->checkNext(Token::Kind::TK_LOGICALAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001697 if (!depth.increase()) {
1698 return ASTNode::ID::Invalid();
1699 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001700 ASTNode::ID right = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001701 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001702 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001703 }
John Stiles3b209362020-11-16 17:03:10 -05001704 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001705 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001706 getNode(newResult).addChild(result);
1707 getNode(newResult).addChild(right);
1708 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001709 }
1710 return result;
1711}
1712
1713/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001714ASTNode::ID Parser::bitwiseOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001715 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001716 ASTNode::ID result = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001717 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001718 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001719 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001720 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001721 while (this->checkNext(Token::Kind::TK_BITWISEOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001722 if (!depth.increase()) {
1723 return ASTNode::ID::Invalid();
1724 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001725 ASTNode::ID right = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001726 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001727 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001728 }
John Stiles45990502021-02-16 10:55:27 -05001729 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
1730 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001731 getNode(newResult).addChild(result);
1732 getNode(newResult).addChild(right);
1733 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001734 }
1735 return result;
1736}
1737
1738/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001739ASTNode::ID Parser::bitwiseXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001740 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001741 ASTNode::ID result = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001742 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001743 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001744 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001745 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001746 while (this->checkNext(Token::Kind::TK_BITWISEXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001747 if (!depth.increase()) {
1748 return ASTNode::ID::Invalid();
1749 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001750 ASTNode::ID right = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001751 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001752 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001753 }
John Stiles45990502021-02-16 10:55:27 -05001754 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
1755 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001756 getNode(newResult).addChild(result);
1757 getNode(newResult).addChild(right);
1758 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001759 }
1760 return result;
1761}
1762
1763/* equalityExpression (BITWISEAND equalityExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001764ASTNode::ID Parser::bitwiseAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001765 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001766 ASTNode::ID result = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001767 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001768 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001769 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001770 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001771 while (this->checkNext(Token::Kind::TK_BITWISEAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001772 if (!depth.increase()) {
1773 return ASTNode::ID::Invalid();
1774 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001775 ASTNode::ID right = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001776 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001777 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001778 }
John Stiles3b209362020-11-16 17:03:10 -05001779 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001780 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001781 getNode(newResult).addChild(result);
1782 getNode(newResult).addChild(right);
1783 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001784 }
1785 return result;
1786}
1787
1788/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001789ASTNode::ID Parser::equalityExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001790 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001791 ASTNode::ID result = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001792 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001793 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001794 }
1795 for (;;) {
1796 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001797 case Token::Kind::TK_EQEQ: // fall through
1798 case Token::Kind::TK_NEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001799 if (!depth.increase()) {
1800 return ASTNode::ID::Invalid();
1801 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001802 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001803 ASTNode::ID right = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001804 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001805 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001806 }
John Stiles3b209362020-11-16 17:03:10 -05001807 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001808 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001809 getNode(newResult).addChild(result);
1810 getNode(newResult).addChild(right);
1811 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001812 break;
1813 }
1814 default:
1815 return result;
1816 }
1817 }
1818}
1819
1820/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001821ASTNode::ID Parser::relationalExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001822 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001823 ASTNode::ID result = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001824 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001825 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001826 }
1827 for (;;) {
1828 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001829 case Token::Kind::TK_LT: // fall through
1830 case Token::Kind::TK_GT: // fall through
1831 case Token::Kind::TK_LTEQ: // fall through
1832 case Token::Kind::TK_GTEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001833 if (!depth.increase()) {
1834 return ASTNode::ID::Invalid();
1835 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001836 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001837 ASTNode::ID right = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001838 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001839 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001840 }
John Stiles3b209362020-11-16 17:03:10 -05001841 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001842 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001843 getNode(newResult).addChild(result);
1844 getNode(newResult).addChild(right);
1845 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001846 break;
1847 }
1848 default:
1849 return result;
1850 }
1851 }
1852}
1853
1854/* additiveExpression ((SHL | SHR) additiveExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001855ASTNode::ID Parser::shiftExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001856 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001857 ASTNode::ID result = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001858 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001859 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001860 }
1861 for (;;) {
1862 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001863 case Token::Kind::TK_SHL: // fall through
1864 case Token::Kind::TK_SHR: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001865 if (!depth.increase()) {
1866 return ASTNode::ID::Invalid();
1867 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001868 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001869 ASTNode::ID right = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001870 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001871 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001872 }
John Stiles3b209362020-11-16 17:03:10 -05001873 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001874 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001875 getNode(newResult).addChild(result);
1876 getNode(newResult).addChild(right);
1877 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001878 break;
1879 }
1880 default:
1881 return result;
1882 }
1883 }
1884}
1885
1886/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001887ASTNode::ID Parser::additiveExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001888 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001889 ASTNode::ID result = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001890 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001891 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001892 }
1893 for (;;) {
1894 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001895 case Token::Kind::TK_PLUS: // fall through
1896 case Token::Kind::TK_MINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001897 if (!depth.increase()) {
1898 return ASTNode::ID::Invalid();
1899 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001900 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001901 ASTNode::ID right = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001902 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001903 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001904 }
John Stiles3b209362020-11-16 17:03:10 -05001905 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001906 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001907 getNode(newResult).addChild(result);
1908 getNode(newResult).addChild(right);
1909 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001910 break;
1911 }
1912 default:
1913 return result;
1914 }
1915 }
1916}
1917
1918/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001919ASTNode::ID Parser::multiplicativeExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001920 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001921 ASTNode::ID result = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001922 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001923 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001924 }
1925 for (;;) {
1926 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001927 case Token::Kind::TK_STAR: // fall through
1928 case Token::Kind::TK_SLASH: // fall through
1929 case Token::Kind::TK_PERCENT: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001930 if (!depth.increase()) {
1931 return ASTNode::ID::Invalid();
1932 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001933 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001934 ASTNode::ID right = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001935 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001936 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001937 }
John Stiles3b209362020-11-16 17:03:10 -05001938 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001939 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001940 getNode(newResult).addChild(result);
1941 getNode(newResult).addChild(right);
1942 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001943 break;
1944 }
1945 default:
1946 return result;
1947 }
1948 }
1949}
1950
1951/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001952ASTNode::ID Parser::unaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001953 AutoDepth depth(this);
ethannicholasb3058bd2016-07-01 08:22:01 -07001954 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001955 case Token::Kind::TK_PLUS: // fall through
1956 case Token::Kind::TK_MINUS: // fall through
1957 case Token::Kind::TK_LOGICALNOT: // fall through
1958 case Token::Kind::TK_BITWISENOT: // fall through
1959 case Token::Kind::TK_PLUSPLUS: // fall through
1960 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001961 if (!depth.increase()) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001962 return ASTNode::ID::Invalid();
Ethan Nicholas6dcc3252019-02-20 15:18:36 -05001963 }
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001964 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001965 ASTNode::ID expr = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001966 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001967 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001968 }
John Stiles45990502021-02-16 10:55:27 -05001969 ASTNode::ID result = this->createNode(t.fOffset, ASTNode::Kind::kPrefix,
1970 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001971 getNode(result).addChild(expr);
1972 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001973 }
1974 default:
1975 return this->postfixExpression();
1976 }
1977}
1978
1979/* term suffix* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001980ASTNode::ID Parser::postfixExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001981 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001982 ASTNode::ID result = this->term();
ethannicholasb3058bd2016-07-01 08:22:01 -07001983 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001984 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001985 }
1986 for (;;) {
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001987 Token t = this->peek();
1988 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001989 case Token::Kind::TK_FLOAT_LITERAL:
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001990 if (this->text(t)[0] != '.') {
1991 return result;
1992 }
John Stiles30212b72020-06-11 17:55:07 -04001993 [[fallthrough]];
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001994 case Token::Kind::TK_LBRACKET:
1995 case Token::Kind::TK_DOT:
1996 case Token::Kind::TK_LPAREN:
1997 case Token::Kind::TK_PLUSPLUS:
1998 case Token::Kind::TK_MINUSMINUS:
1999 case Token::Kind::TK_COLONCOLON:
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002000 if (!depth.increase()) {
2001 return ASTNode::ID::Invalid();
2002 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002003 result = this->suffix(result);
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002004 if (!result) {
2005 return ASTNode::ID::Invalid();
2006 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002007 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002008 default:
2009 return result;
2010 }
2011 }
2012}
2013
Ethan Nicholas11d53972016-11-28 11:23:23 -05002014/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
Ethan Nicholase455f652019-09-13 12:52:55 -04002015 PLUSPLUS | MINUSMINUS | COLONCOLON IDENTIFIER | FLOAT_LITERAL [IDENTIFIER] */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002016ASTNode::ID Parser::suffix(ASTNode::ID base) {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002017 SkASSERT(base);
ethannicholasb3058bd2016-07-01 08:22:01 -07002018 Token next = this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002019 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002020 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002021 return ASTNode::ID::Invalid();
2022 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002023 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002024 case Token::Kind::TK_LBRACKET: {
2025 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
John Stiles3b209362020-11-16 17:03:10 -05002026 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kIndex);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002027 getNode(result).addChild(base);
2028 return result;
ethannicholas5961bc92016-10-12 06:39:56 -07002029 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002030 ASTNode::ID e = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002031 if (!e) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002032 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002033 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002034 this->expect(Token::Kind::TK_RBRACKET, "']' to complete array access expression");
John Stiles3b209362020-11-16 17:03:10 -05002035 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kIndex);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002036 getNode(result).addChild(base);
2037 getNode(result).addChild(e);
2038 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002039 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002040 case Token::Kind::TK_COLONCOLON: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002041 int offset = this->peek().fOffset;
2042 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002043 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002044 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kScope,
2045 std::move(text));
Brian Osman6518d772020-09-10 16:50:06 -04002046 getNode(result).addChild(base);
2047 return result;
2048 }
2049 return ASTNode::ID::Invalid();
2050 }
2051 case Token::Kind::TK_DOT: {
2052 int offset = this->peek().fOffset;
2053 StringFragment text;
2054 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002055 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kField,
2056 std::move(text));
Ethan Nicholasfc994162019-06-06 10:04:27 -04002057 getNode(result).addChild(base);
2058 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002059 }
Brian Osman6518d772020-09-10 16:50:06 -04002060 [[fallthrough]];
Ethan Nicholase455f652019-09-13 12:52:55 -04002061 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002062 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholase455f652019-09-13 12:52:55 -04002063 // Swizzles that start with a constant number, e.g. '.000r', will be tokenized as
2064 // floating point literals, possibly followed by an identifier. Handle that here.
2065 StringFragment field = this->text(next);
2066 SkASSERT(field.fChars[0] == '.');
2067 ++field.fChars;
2068 --field.fLength;
2069 for (size_t i = 0; i < field.fLength; ++i) {
2070 if (field.fChars[i] != '0' && field.fChars[i] != '1') {
2071 this->error(next, "invalid swizzle");
2072 return ASTNode::ID::Invalid();
2073 }
2074 }
2075 // use the next *raw* token so we don't ignore whitespace - we only care about
2076 // identifiers that directly follow the float
2077 Token id = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002078 if (id.fKind == Token::Kind::TK_IDENTIFIER) {
Ethan Nicholase455f652019-09-13 12:52:55 -04002079 field.fLength += id.fLength;
2080 } else {
2081 this->pushback(id);
2082 }
John Stiles3b209362020-11-16 17:03:10 -05002083 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kField, field);
Ethan Nicholase455f652019-09-13 12:52:55 -04002084 getNode(result).addChild(base);
2085 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002086 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002087 case Token::Kind::TK_LPAREN: {
John Stiles3b209362020-11-16 17:03:10 -05002088 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kCall);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002089 getNode(result).addChild(base);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002090 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002091 for (;;) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002092 ASTNode::ID expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002093 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002094 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002095 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002096 getNode(result).addChild(expr);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002097 if (!this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002098 break;
2099 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002100 }
2101 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002102 this->expect(Token::Kind::TK_RPAREN, "')' to complete function parameters");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002103 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002104 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002105 case Token::Kind::TK_PLUSPLUS: // fall through
2106 case Token::Kind::TK_MINUSMINUS: {
John Stiles45990502021-02-16 10:55:27 -05002107 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kPostfix,
2108 Operator(next.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04002109 getNode(result).addChild(base);
2110 return result;
2111 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002112 default: {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002113 this->error(next, "expected expression suffix, but found '" + this->text(next) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002114 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002115 }
2116 }
2117}
2118
Brian Osman33c64a42020-12-23 10:26:38 -05002119/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | '(' expression ')' */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002120ASTNode::ID Parser::term() {
ethannicholasb3058bd2016-07-01 08:22:01 -07002121 Token t = this->peek();
2122 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002123 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002124 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002125 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002126 return this->createNode(t.fOffset, ASTNode::Kind::kIdentifier, std::move(text));
ethannicholasb3058bd2016-07-01 08:22:01 -07002127 }
John Stiles30212b72020-06-11 17:55:07 -04002128 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002129 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002130 case Token::Kind::TK_INT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002131 SKSL_INT i;
ethannicholasb3058bd2016-07-01 08:22:01 -07002132 if (this->intLiteral(&i)) {
John Stiles3b209362020-11-16 17:03:10 -05002133 return this->createNode(t.fOffset, ASTNode::Kind::kInt, i);
ethannicholasb3058bd2016-07-01 08:22:01 -07002134 }
2135 break;
2136 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002137 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002138 SKSL_FLOAT f;
ethannicholasb3058bd2016-07-01 08:22:01 -07002139 if (this->floatLiteral(&f)) {
John Stiles3b209362020-11-16 17:03:10 -05002140 return this->createNode(t.fOffset, ASTNode::Kind::kFloat, f);
ethannicholasb3058bd2016-07-01 08:22:01 -07002141 }
2142 break;
2143 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002144 case Token::Kind::TK_TRUE_LITERAL: // fall through
2145 case Token::Kind::TK_FALSE_LITERAL: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002146 bool b;
2147 if (this->boolLiteral(&b)) {
John Stiles3b209362020-11-16 17:03:10 -05002148 return this->createNode(t.fOffset, ASTNode::Kind::kBool, b);
ethannicholasb3058bd2016-07-01 08:22:01 -07002149 }
2150 break;
2151 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002152 case Token::Kind::TK_LPAREN: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002153 this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002154 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002155 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002156 return ASTNode::ID::Invalid();
2157 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002158 ASTNode::ID result = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002159 if (result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002160 this->expect(Token::Kind::TK_RPAREN, "')' to complete expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002161 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002162 }
2163 break;
2164 }
2165 default:
2166 this->nextToken();
John Stilesf94348f2020-12-23 18:58:43 -05002167 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002168 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002169 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002170}
2171
2172/* INT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002173bool Parser::intLiteral(SKSL_INT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002174 Token t;
John Stilesf94348f2020-12-23 18:58:43 -05002175 if (!this->expect(Token::Kind::TK_INT_LITERAL, "integer literal", &t)) {
2176 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07002177 }
John Stilesf94348f2020-12-23 18:58:43 -05002178 StringFragment s = this->text(t);
2179 if (!SkSL::stoi(s, dest)) {
2180 this->error(t, "integer is too large: " + s);
2181 return false;
2182 }
2183 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07002184}
2185
John Stilesf94348f2020-12-23 18:58:43 -05002186
ethannicholasb3058bd2016-07-01 08:22:01 -07002187/* FLOAT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002188bool Parser::floatLiteral(SKSL_FLOAT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002189 Token t;
John Stilesf94348f2020-12-23 18:58:43 -05002190 if (!this->expect(Token::Kind::TK_FLOAT_LITERAL, "float literal", &t)) {
2191 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07002192 }
John Stilesf94348f2020-12-23 18:58:43 -05002193 StringFragment s = this->text(t);
2194 if (!SkSL::stod(s, dest)) {
2195 this->error(t, "floating-point value is too large: " + s);
2196 return false;
2197 }
2198 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07002199}
2200
2201/* TRUE_LITERAL | FALSE_LITERAL */
2202bool Parser::boolLiteral(bool* dest) {
2203 Token t = this->nextToken();
2204 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002205 case Token::Kind::TK_TRUE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002206 *dest = true;
2207 return true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002208 case Token::Kind::TK_FALSE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002209 *dest = false;
2210 return true;
2211 default:
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002212 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002213 return false;
2214 }
2215}
2216
2217/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002218bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002219 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002220 if (this->expect(Token::Kind::TK_IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002221 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07002222 return true;
2223 }
2224 return false;
2225}
2226
John Stilesa6841be2020-08-06 14:11:56 -04002227} // namespace SkSL