blob: 44f8aa728188401044d59d5d3576af1806cb4bd0 [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
13#include "src/sksl/SkSLASTNode.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050014#include "src/sksl/ir/SkSLModifiers.h"
15#include "src/sksl/ir/SkSLSymbolTable.h"
16#include "src/sksl/ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070017
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040018#ifndef SKSL_STANDALONE
Mike Kleinc0bd9f92019-04-23 12:05:21 -050019#include "include/private/SkOnce.h"
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040020#endif
21
ethannicholasb3058bd2016-07-01 08:22:01 -070022namespace SkSL {
23
John Stiles8d056592020-11-10 10:03:50 -050024static constexpr int kMaxParseDepth = 50;
John Stilesd39aec92020-12-03 14:37:16 -050025static constexpr int kMaxStructDepth = 8;
John Stilesa695d622020-11-10 10:04:26 -050026
27static bool struct_is_too_deeply_nested(const Type& type, int limit) {
28 if (limit < 0) {
29 return true;
30 }
31
John Stilesc0c51062020-12-03 17:16:29 -050032 if (type.isStruct()) {
John Stilesa695d622020-11-10 10:04:26 -050033 for (const Type::Field& f : type.fields()) {
34 if (struct_is_too_deeply_nested(*f.fType, limit - 1)) {
35 return true;
36 }
37 }
38 }
39
40 return false;
41}
ethannicholascad64162016-10-27 10:54:02 -070042
John Stiles7d5b90a2021-01-21 14:26:51 -050043static int parse_modifier_token(Token::Kind token) {
44 switch (token) {
45 case Token::Kind::TK_UNIFORM: return Modifiers::kUniform_Flag;
46 case Token::Kind::TK_CONST: return Modifiers::kConst_Flag;
47 case Token::Kind::TK_IN: return Modifiers::kIn_Flag;
48 case Token::Kind::TK_OUT: return Modifiers::kOut_Flag;
49 case Token::Kind::TK_INOUT: return Modifiers::kIn_Flag | Modifiers::kOut_Flag;
50 case Token::Kind::TK_FLAT: return Modifiers::kFlat_Flag;
51 case Token::Kind::TK_NOPERSPECTIVE: return Modifiers::kNoPerspective_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050052 case Token::Kind::TK_HASSIDEEFFECTS: return Modifiers::kHasSideEffects_Flag;
John Stiles7d5b90a2021-01-21 14:26:51 -050053 case Token::Kind::TK_VARYING: return Modifiers::kVarying_Flag;
54 case Token::Kind::TK_INLINE: return Modifiers::kInline_Flag;
55 default: return 0;
56 }
57}
58
ethannicholascad64162016-10-27 10:54:02 -070059class AutoDepth {
60public:
61 AutoDepth(Parser* p)
Ethan Nicholascf4deab2019-09-13 16:28:14 -040062 : fParser(p)
63 , fDepth(0) {}
ethannicholascad64162016-10-27 10:54:02 -070064
65 ~AutoDepth() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -040066 fParser->fDepth -= fDepth;
ethannicholascad64162016-10-27 10:54:02 -070067 }
68
Ethan Nicholascf4deab2019-09-13 16:28:14 -040069 bool increase() {
70 ++fDepth;
71 ++fParser->fDepth;
John Stiles8d056592020-11-10 10:03:50 -050072 if (fParser->fDepth > kMaxParseDepth) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070073 fParser->error(fParser->peek(), String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070074 return false;
75 }
76 return true;
77 }
78
79private:
80 Parser* fParser;
Ethan Nicholascf4deab2019-09-13 16:28:14 -040081 int fDepth;
ethannicholascad64162016-10-27 10:54:02 -070082};
83
Brian Salomon140f3da2018-08-23 13:51:27 +000084std::unordered_map<String, Parser::LayoutToken>* Parser::layoutTokens;
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040085
86void Parser::InitLayoutMap() {
Brian Salomon140f3da2018-08-23 13:51:27 +000087 layoutTokens = new std::unordered_map<String, LayoutToken>;
Brian Salomon23356442018-11-30 15:33:19 -050088 #define TOKEN(name, text) (*layoutTokens)[text] = LayoutToken::name
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040089 TOKEN(LOCATION, "location");
90 TOKEN(OFFSET, "offset");
91 TOKEN(BINDING, "binding");
92 TOKEN(INDEX, "index");
93 TOKEN(SET, "set");
94 TOKEN(BUILTIN, "builtin");
95 TOKEN(INPUT_ATTACHMENT_INDEX, "input_attachment_index");
96 TOKEN(ORIGIN_UPPER_LEFT, "origin_upper_left");
97 TOKEN(OVERRIDE_COVERAGE, "override_coverage");
98 TOKEN(BLEND_SUPPORT_ALL_EQUATIONS, "blend_support_all_equations");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040099 TOKEN(PUSH_CONSTANT, "push_constant");
100 TOKEN(POINTS, "points");
101 TOKEN(LINES, "lines");
102 TOKEN(LINE_STRIP, "line_strip");
103 TOKEN(LINES_ADJACENCY, "lines_adjacency");
104 TOKEN(TRIANGLES, "triangles");
105 TOKEN(TRIANGLE_STRIP, "triangle_strip");
106 TOKEN(TRIANGLES_ADJACENCY, "triangles_adjacency");
107 TOKEN(MAX_VERTICES, "max_vertices");
108 TOKEN(INVOCATIONS, "invocations");
Brian Osmanf59a9612020-04-15 14:18:13 -0400109 TOKEN(MARKER, "marker");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400110 TOKEN(WHEN, "when");
111 TOKEN(KEY, "key");
Michael Ludwiga4275592018-08-31 10:52:47 -0400112 TOKEN(TRACKED, "tracked");
Brian Osmanb32d66b2020-04-30 17:12:03 -0400113 TOKEN(SRGB_UNPREMUL, "srgb_unpremul");
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400114 TOKEN(CTYPE, "ctype");
Brian Osmanf28e55d2018-10-03 16:35:54 -0400115 TOKEN(SKPMCOLOR4F, "SkPMColor4f");
Mike Reedb26b4e72020-01-22 14:31:21 -0500116 TOKEN(SKV4, "SkV4");
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400117 TOKEN(SKRECT, "SkRect");
118 TOKEN(SKIRECT, "SkIRect");
119 TOKEN(SKPMCOLOR, "SkPMColor");
Mike Reedb26b4e72020-01-22 14:31:21 -0500120 TOKEN(SKM44, "SkM44");
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400121 TOKEN(BOOL, "bool");
122 TOKEN(INT, "int");
123 TOKEN(FLOAT, "float");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400124 #undef TOKEN
125}
126
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400127Parser::Parser(const char* text, size_t length, SymbolTable& symbols, ErrorReporter& errors)
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700128: fText(text)
John Stiles7cbb09c22021-01-07 16:07:00 -0500129, fPushback(Token::Kind::TK_NONE, -1, -1)
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400130, fSymbols(symbols)
ethannicholasb3058bd2016-07-01 08:22:01 -0700131, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700132 fLexer.start(text, length);
Brian Salomon3b83afe2018-08-23 11:04:36 -0400133 static const bool layoutMapInitialized = []{ return (void)InitLayoutMap(), true; }();
134 (void) layoutMapInitialized;
ethannicholasb3058bd2016-07-01 08:22:01 -0700135}
136
John Stiles3b209362020-11-16 17:03:10 -0500137template <typename... Args>
138ASTNode::ID Parser::createNode(Args&&... args) {
139 ASTNode::ID result(fFile->fNodes.size());
140 fFile->fNodes.emplace_back(&fFile->fNodes, std::forward<Args>(args)...);
141 return result;
142}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400143
John Stiles3b209362020-11-16 17:03:10 -0500144ASTNode::ID Parser::addChild(ASTNode::ID target, ASTNode::ID child) {
145 fFile->fNodes[target.fValue].addChild(child);
146 return child;
147}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400148
John Stiles3b209362020-11-16 17:03:10 -0500149void Parser::createEmptyChild(ASTNode::ID target) {
150 ASTNode::ID child(fFile->fNodes.size());
151 fFile->fNodes.emplace_back();
152 fFile->fNodes[target.fValue].addChild(child);
153}
Ethan Nicholasfc994162019-06-06 10:04:27 -0400154
Ethan Nicholasf7b88202017-09-18 14:10:39 -0400155/* (directive | section | declaration)* END_OF_FILE */
Ethan Nicholasba9a04f2020-11-06 09:28:04 -0500156std::unique_ptr<ASTFile> Parser::compilationUnit() {
John Stilesfbd050b2020-08-03 13:21:46 -0400157 fFile = std::make_unique<ASTFile>();
John Stiles3b209362020-11-16 17:03:10 -0500158 ASTNode::ID result = this->createNode(/*offset=*/0, ASTNode::Kind::kFile);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400159 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700160 for (;;) {
161 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400162 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400163 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400164 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400165 ASTNode::ID dir = this->directive();
166 if (fErrors.errorCount()) {
167 return nullptr;
168 }
169 if (dir) {
170 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700171 }
172 break;
173 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400174 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400175 ASTNode::ID section = this->section();
176 if (fErrors.errorCount()) {
177 return nullptr;
178 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400179 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400180 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400181 }
182 break;
183 }
John Stiles7cbb09c22021-01-07 16:07:00 -0500184 case Token::Kind::TK_INVALID: {
185 this->error(this->peek(), String("invalid token"));
186 return nullptr;
187 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700188 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400189 ASTNode::ID decl = this->declaration();
190 if (fErrors.errorCount()) {
191 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700192 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400193 if (decl) {
194 getNode(result).addChild(decl);
195 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700196 }
197 }
198 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400199 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700200}
201
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700202Token Parser::nextRawToken() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500203 if (fPushback.fKind != Token::Kind::TK_NONE) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700204 Token result = fPushback;
John Stiles7cbb09c22021-01-07 16:07:00 -0500205 fPushback.fKind = Token::Kind::TK_NONE;
ethannicholasb3058bd2016-07-01 08:22:01 -0700206 return result;
207 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700208 Token result = fLexer.next();
209 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400210}
211
212Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700213 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400214 while (token.fKind == Token::Kind::TK_WHITESPACE ||
215 token.fKind == Token::Kind::TK_LINE_COMMENT ||
216 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700217 token = this->nextRawToken();
218 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400219 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700220}
221
222void Parser::pushback(Token t) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500223 SkASSERT(fPushback.fKind == Token::Kind::TK_NONE);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400224 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700225}
226
227Token Parser::peek() {
John Stiles7cbb09c22021-01-07 16:07:00 -0500228 if (fPushback.fKind == Token::Kind::TK_NONE) {
Brian Osman634624a2017-08-15 11:14:30 -0400229 fPushback = this->nextToken();
230 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700231 return fPushback;
232}
233
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400234bool Parser::checkNext(Token::Kind kind, Token* result) {
John Stiles7cbb09c22021-01-07 16:07:00 -0500235 if (fPushback.fKind != Token::Kind::TK_NONE && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400236 return false;
237 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400238 Token next = this->nextToken();
239 if (next.fKind == kind) {
240 if (result) {
241 *result = next;
242 }
243 return true;
244 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400245 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400246 return false;
247}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500248
249bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700250 Token next = this->nextToken();
251 if (next.fKind == kind) {
252 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400253 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700254 }
255 return true;
256 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700257 this->error(next, "expected " + String(expected) + ", but found '" +
258 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700259 return false;
260 }
261}
262
John Stiles2630ea32020-12-04 10:51:21 -0500263bool Parser::expectIdentifier(Token* result) {
264 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", result)) {
265 return false;
266 }
267 if (this->isType(this->text(*result))) {
268 this->error(*result, "expected an identifier, but found type '" +
269 this->text(*result) + "'");
270 return false;
271 }
272 return true;
273}
274
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700275StringFragment Parser::text(Token token) {
276 return StringFragment(fText + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500277}
278
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700279void Parser::error(Token token, String msg) {
280 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700281}
282
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700283void Parser::error(int offset, String msg) {
284 fErrors.error(offset, msg);
285}
286
287bool Parser::isType(StringFragment name) {
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400288 const Symbol* s = fSymbols[name];
John Stiles2630ea32020-12-04 10:51:21 -0500289 return s && s->is<Type>();
ethannicholasb3058bd2016-07-01 08:22:01 -0700290}
291
John Stiles80b02af2021-02-12 17:07:51 -0500292bool Parser::isArrayType(ASTNode::ID type) {
293 const ASTNode& node = this->getNode(type);
294 SkASSERT(node.fKind == ASTNode::Kind::kType);
295 return node.begin() != node.end();
296}
297
Ethan Nicholas11d53972016-11-28 11:23:23 -0500298/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700299 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400300ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700301 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400302 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400303 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700304 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700305 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400306 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700307 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500308 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400309 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700310 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400311 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400312 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700313 }
314 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400315 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400316 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700317 }
John Stiles3b209362020-11-16 17:03:10 -0500318 return this->createNode(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700319 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700320 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400321 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700322 }
323}
324
Ethan Nicholas762466e2017-06-29 10:03:38 -0400325/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
326 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400327ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400328 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400329 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400330 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400331 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400332 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400333 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400334 this->nextToken();
335 Token argToken;
John Stiles2630ea32020-12-04 10:51:21 -0500336 if (!this->expectIdentifier(&argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400337 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400338 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700339 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400340 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400341 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400342 }
343 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400344 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400345 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400346 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400347 StringFragment text;
348 Token codeStart = this->nextRawToken();
349 size_t startOffset = codeStart.fOffset;
350 this->pushback(codeStart);
351 text.fChars = fText + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400352 int level = 1;
353 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700354 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400355 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400356 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400357 ++level;
358 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400359 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400360 --level;
361 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400362 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700363 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400364 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400365 default:
366 break;
367 }
368 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400369 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400370 break;
371 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400372 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700373 StringFragment name = this->text(start);
374 ++name.fChars;
375 --name.fLength;
John Stiles3b209362020-11-16 17:03:10 -0500376 return this->createNode(start.fOffset, ASTNode::Kind::kSection,
377 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400378}
379
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500380/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
381 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400382ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500383 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400384 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400385 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500386 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400387 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400388 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500389 }
390 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500391 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400392 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500393 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400394 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400395 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500396 }
John Stilese6c67c52021-01-15 12:01:00 -0500397 fSymbols.add(Type::MakeEnumType(this->text(name)));
John Stiles3b209362020-11-16 17:03:10 -0500398 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400399 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500400 Token id;
John Stiles2630ea32020-12-04 10:51:21 -0500401 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400402 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500403 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400404 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400405 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500406 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400407 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500408 }
John Stiles3b209362020-11-16 17:03:10 -0500409 ASTNode::ID child = this->addChild(
410 result, this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400411 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500412 } else {
John Stiles3b209362020-11-16 17:03:10 -0500413 this->addChild(result,
414 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500415 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400416 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
417 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400418 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500419 }
John Stiles2630ea32020-12-04 10:51:21 -0500420 if (!this->expectIdentifier(&id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400421 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500422 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400423 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400424 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500425 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400426 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500427 }
John Stiles3b209362020-11-16 17:03:10 -0500428 ASTNode::ID child = this->addChild(
429 result,
430 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400431 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500432 } else {
John Stiles3b209362020-11-16 17:03:10 -0500433 this->addChild(
434 result,
435 this->createNode(id.fOffset, ASTNode::Kind::kEnumCase, this->text(id)));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500436 }
437 }
438 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400439 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400440 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500441}
442
443/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700444 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400445ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700446 Token lookahead = this->peek();
Ethan Nicholasda6320c2020-09-02 14:08:23 -0400447 switch (lookahead.fKind) {
448 case Token::Kind::TK_ENUM:
449 return this->enumDeclaration();
450 case Token::Kind::TK_SEMICOLON:
451 this->error(lookahead.fOffset, "expected a declaration, but found ';'");
452 return ASTNode::ID::Invalid();
453 default:
454 break;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500455 }
456 Modifiers modifiers = this->modifiers();
457 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400458 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700459 // we have an identifier that's not a type, could be the start of an interface block
460 return this->interfaceBlock(modifiers);
461 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400462 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700463 return this->structVarDeclaration(modifiers);
464 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400465 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700466 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -0500467 return this->createNode(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700468 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400469 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700470 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400471 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700472 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700473 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500474 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400475 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700476 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400477 if (this->checkNext(Token::Kind::TK_LPAREN)) {
John Stiles3b209362020-11-16 17:03:10 -0500478 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kFunction);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400479 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
480 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400481 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400482 for (;;) {
483 ASTNode::ID parameter = this->parameter();
484 if (!parameter) {
485 return ASTNode::ID::Invalid();
486 }
487 ++fd.fParameterCount;
488 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400489 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400490 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700491 }
492 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700493 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400494 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400495 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400496 return ASTNode::ID::Invalid();
497 }
498 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400499 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700500 body = this->block();
501 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400502 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700503 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400504 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700505 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400506 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700507 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400508 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700509 }
510}
511
John Stiles76389b72021-01-25 13:49:05 -0500512/* (varDeclarations | expressionStatement) */
513ASTNode::ID Parser::varDeclarationsOrExpressionStatement() {
514 if (this->isType(this->text(this->peek()))) {
515 // Statements that begin with a typename are most often variable declarations, but
516 // occasionally the type is part of a constructor, and these are actually expression-
517 // statements in disguise. First, attempt the common case: parse it as a vardecl.
518 Checkpoint checkpoint(this);
John Stiles8dabeac2021-02-12 16:05:00 -0500519 VarDeclarationsPrefix prefix;
520 if (this->varDeclarationsPrefix(&prefix)) {
521 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
John Stiles76389b72021-01-25 13:49:05 -0500522 }
523
524 // If this statement wasn't actually a vardecl after all, rewind and try parsing it as an
525 // expression-statement instead.
526 checkpoint.rewind();
527 }
528
529 return this->expressionStatement();
530}
531
John Stiles8dabeac2021-02-12 16:05:00 -0500532// Helper function for varDeclarations(). If this function succeeds, we assume that the rest of the
533// statement is a variable-declaration statement, not an expression-statement.
534bool Parser::varDeclarationsPrefix(VarDeclarationsPrefix* prefixData) {
535 prefixData->modifiers = this->modifiers();
536 prefixData->type = this->type();
537 if (!prefixData->type) {
538 return false;
539 }
540 return this->expectIdentifier(&prefixData->name);
541}
542
ethannicholasb3058bd2016-07-01 08:22:01 -0700543/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400544ASTNode::ID Parser::varDeclarations() {
John Stiles8dabeac2021-02-12 16:05:00 -0500545 VarDeclarationsPrefix prefix;
546 if (!this->varDeclarationsPrefix(&prefix)) {
John Stiles76389b72021-01-25 13:49:05 -0500547 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700548 }
John Stiles8dabeac2021-02-12 16:05:00 -0500549 return this->varDeclarationEnd(prefix.modifiers, prefix.type, this->text(prefix.name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700550}
551
552/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400553ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400554 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400555 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700556 }
557 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500558 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400559 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700560 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400561 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400562 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700563 }
564 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400565 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400566 ASTNode::ID decls = this->varDeclarations();
567 if (!decls) {
568 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700569 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400570 ASTNode& declsNode = getNode(decls);
John Stiles13fc2602020-10-09 17:42:31 -0400571 Modifiers modifiers = declsNode.begin()->getModifiers();
572 if (modifiers.fFlags != Modifiers::kNo_Flag) {
573 String desc = modifiers.description();
574 desc.pop_back(); // remove trailing space
575 this->error(declsNode.fOffset,
576 "modifier '" + desc + "' is not permitted on a struct field");
577 }
578
Brian Osman00fea5b2021-01-28 09:46:30 -0500579 const Symbol* symbol = fSymbols[(declsNode.begin() + 1)->getString()];
John Stiles13fc2602020-10-09 17:42:31 -0400580 SkASSERT(symbol);
581 const Type* type = &symbol->as<Type>();
John Stiles1d757782020-11-17 09:43:22 -0500582 if (type->isOpaque()) {
583 this->error(declsNode.fOffset,
584 "opaque type '" + type->name() + "' is not permitted in a struct");
585 }
John Stiles13fc2602020-10-09 17:42:31 -0400586
Ethan Nicholasfc994162019-06-06 10:04:27 -0400587 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
588 ASTNode& var = *iter;
589 ASTNode::VarData vd = var.getVarData();
John Stiles6bef6a72020-12-02 14:26:04 -0500590
John Stilesd39aec92020-12-03 14:37:16 -0500591 // Read array size if one is present.
592 if (vd.fIsArray) {
593 const ASTNode& size = *var.begin();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400594 if (!size || size.fKind != ASTNode::Kind::kInt) {
595 this->error(declsNode.fOffset, "array size in struct field must be a constant");
596 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700597 }
John Stiles6bef6a72020-12-02 14:26:04 -0500598 if (size.getInt() <= 0 || size.getInt() > INT_MAX) {
599 this->error(declsNode.fOffset, "array size is invalid");
600 return ASTNode::ID::Invalid();
601 }
John Stilesd39aec92020-12-03 14:37:16 -0500602 // Add the array dimensions to our type.
603 int arraySize = size.getInt();
John Stilesa2179502020-12-03 14:37:57 -0500604 type = fSymbols.addArrayDimension(type, arraySize);
ethannicholasb3058bd2016-07-01 08:22:01 -0700605 }
John Stiles13fc2602020-10-09 17:42:31 -0400606
607 fields.push_back(Type::Field(modifiers, vd.fName, type));
John Stilesd39aec92020-12-03 14:37:16 -0500608 if (vd.fIsArray ? var.begin()->fNext : var.fFirstChild) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400609 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700610 }
611 }
612 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400613 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400614 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700615 }
Brian Osman6c3b23f2021-02-12 13:09:58 -0500616 if (fields.empty()) {
617 this->error(name.fOffset,
618 "struct '" + this->text(name) + "' must contain at least one field");
619 return ASTNode::ID::Invalid();
620 }
John Stilesad2d4942020-12-11 16:55:58 -0500621 std::unique_ptr<Type> newType = Type::MakeStructType(name.fOffset, this->text(name), fields);
John Stilesa695d622020-11-10 10:04:26 -0500622 if (struct_is_too_deeply_nested(*newType, kMaxStructDepth)) {
623 this->error(name.fOffset, "struct '" + this->text(name) + "' is too deeply nested");
624 return ASTNode::ID::Invalid();
625 }
626 fSymbols.add(std::move(newType));
Brian Osman00fea5b2021-01-28 09:46:30 -0500627 return this->createNode(name.fOffset, ASTNode::Kind::kType, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700628}
629
630/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400631ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
632 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700633 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400634 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700635 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400636 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400637 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400638 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700639 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400640 this->expect(Token::Kind::TK_SEMICOLON, "';'");
John Stilesdc75a972020-11-25 16:24:55 -0500641 return type;
ethannicholasb3058bd2016-07-01 08:22:01 -0700642}
643
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400644/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
645 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400646ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
John Stiles08070f62020-11-17 10:45:49 -0500647 int offset = this->peek().fOffset;
648 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kVarDeclarations);
649 this->addChild(result, this->createNode(offset, ASTNode::Kind::kModifiers, mods));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400650 getNode(result).addChild(type);
John Stilesd39aec92020-12-03 14:37:16 -0500651
John Stiles80b02af2021-02-12 17:07:51 -0500652 auto parseArrayDimensions = [&](ASTNode::ID currentVar, ASTNode::VarData* vd) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400653 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500654 if (vd->fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500655 this->error(this->peek(), "multi-dimensional arrays are not supported");
656 return false;
657 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400658 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
John Stiles3b209362020-11-16 17:03:10 -0500659 this->createEmptyChild(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700660 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400661 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700662 if (!size) {
John Stilesd39aec92020-12-03 14:37:16 -0500663 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700664 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400665 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400666 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
John Stilesd39aec92020-12-03 14:37:16 -0500667 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700668 }
669 }
John Stilesd39aec92020-12-03 14:37:16 -0500670 vd->fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700671 }
John Stilesd39aec92020-12-03 14:37:16 -0500672 return true;
673 };
674
675 auto parseInitializer = [this](ASTNode::ID currentVar) -> bool {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400676 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400677 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700678 if (!value) {
John Stilesd39aec92020-12-03 14:37:16 -0500679 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -0700680 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400681 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700682 }
John Stilesd39aec92020-12-03 14:37:16 -0500683 return true;
684 };
685
686 ASTNode::ID currentVar = this->createNode(offset, ASTNode::Kind::kVarDeclaration);
687 ASTNode::VarData vd{name, /*isArray=*/false};
688
689 getNode(result).addChild(currentVar);
690 if (!parseArrayDimensions(currentVar, &vd)) {
691 return ASTNode::ID::Invalid();
692 }
693 getNode(currentVar).setVarData(vd);
694 if (!parseInitializer(currentVar)) {
695 return ASTNode::ID::Invalid();
696 }
697
698 while (this->checkNext(Token::Kind::TK_COMMA)) {
699 Token identifierName;
John Stiles2630ea32020-12-04 10:51:21 -0500700 if (!this->expectIdentifier(&identifierName)) {
John Stilesd39aec92020-12-03 14:37:16 -0500701 return ASTNode::ID::Invalid();
702 }
703
704 currentVar = ASTNode::ID(fFile->fNodes.size());
705 vd = ASTNode::VarData{this->text(identifierName), /*isArray=*/false};
706 fFile->fNodes.emplace_back(&fFile->fNodes, offset, ASTNode::Kind::kVarDeclaration);
707
708 getNode(result).addChild(currentVar);
709 if (!parseArrayDimensions(currentVar, &vd)) {
710 return ASTNode::ID::Invalid();
711 }
712 getNode(currentVar).setVarData(vd);
713 if (!parseInitializer(currentVar)) {
714 return ASTNode::ID::Invalid();
715 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700716 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400717 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400718 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700719 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400720 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700721}
722
723/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400724ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400725 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400726 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700727 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400728 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700729 }
730 Token name;
John Stiles2630ea32020-12-04 10:51:21 -0500731 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400732 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700733 }
John Stiles3b209362020-11-16 17:03:10 -0500734 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kParameter);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400735 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
736 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400737 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stiles80b02af2021-02-12 17:07:51 -0500738 if (pd.fIsArray || this->isArrayType(type)) {
John Stilesd39aec92020-12-03 14:37:16 -0500739 this->error(this->peek(), "multi-dimensional arrays are not supported");
740 return ASTNode::ID::Invalid();
741 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700742 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400743 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400744 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700745 }
John Stilesf94348f2020-12-23 18:58:43 -0500746 StringFragment arraySizeFrag = this->text(sizeToken);
747 SKSL_INT arraySize;
748 if (!SkSL::stoi(arraySizeFrag, &arraySize)) {
749 this->error(sizeToken, "array size is too large: " + arraySizeFrag);
750 return ASTNode::ID::Invalid();
751 }
752 this->addChild(result, this->createNode(sizeToken.fOffset, ASTNode::Kind::kInt, arraySize));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400753 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400754 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700755 }
John Stilesd39aec92020-12-03 14:37:16 -0500756 pd.fIsArray = true;
ethannicholasb3058bd2016-07-01 08:22:01 -0700757 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400758 getNode(result).setParameterData(pd);
759 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700760}
761
Ethan Nicholasd608c092017-10-26 09:30:08 -0400762/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700763int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400764 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700765 return -1;
766 }
767 Token resultToken;
John Stilesf94348f2020-12-23 18:58:43 -0500768 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
769 return -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700770 }
John Stilesf94348f2020-12-23 18:58:43 -0500771 StringFragment resultFrag = this->text(resultToken);
772 SKSL_INT resultValue;
773 if (!SkSL::stoi(resultFrag, &resultValue)) {
774 this->error(resultToken, "value in layout is too large: " + resultFrag);
775 return -1;
776 }
777 return resultValue;
ethannicholasb3058bd2016-07-01 08:22:01 -0700778}
779
Ethan Nicholasd608c092017-10-26 09:30:08 -0400780/** EQ IDENTIFIER */
781StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400782 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400783 return StringFragment();
784 }
785 Token resultToken;
John Stiles2630ea32020-12-04 10:51:21 -0500786 if (!this->expectIdentifier(&resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400787 return StringFragment();
788 }
789 return this->text(resultToken);
790}
791
792
Ethan Nicholas762466e2017-06-29 10:03:38 -0400793/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400794StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400795 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400796 return "";
797 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700798 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400799 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400800 StringFragment code;
801 code.fChars = fText + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400802 int level = 1;
803 bool done = false;
804 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700805 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400806 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400807 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400808 ++level;
809 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400810 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400811 --level;
812 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400813 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400814 if (level == 1) {
815 done = true;
816 }
817 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400818 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700819 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400820 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400821 default:
822 break;
823 }
824 if (!level) {
825 done = true;
826 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400827 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400828 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400829 this->pushback(std::move(next));
830 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400831 }
832 return code;
833}
834
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400835Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400836 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400837 Token t = this->nextToken();
838 String text = this->text(t);
839 auto found = layoutTokens->find(text);
840 if (found != layoutTokens->end()) {
841 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400842 case LayoutToken::SKPMCOLOR4F:
843 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500844 case LayoutToken::SKV4:
845 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400846 case LayoutToken::SKRECT:
847 return Layout::CType::kSkRect;
848 case LayoutToken::SKIRECT:
849 return Layout::CType::kSkIRect;
850 case LayoutToken::SKPMCOLOR:
851 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400852 case LayoutToken::BOOL:
853 return Layout::CType::kBool;
854 case LayoutToken::INT:
855 return Layout::CType::kInt32;
856 case LayoutToken::FLOAT:
857 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500858 case LayoutToken::SKM44:
859 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400860 default:
861 break;
862 }
863 }
864 this->error(t, "unsupported ctype");
865 }
866 return Layout::CType::kDefault;
867}
868
ethannicholas8ac838d2016-11-22 08:39:36 -0800869/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500870Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500871 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700872 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500873 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700874 int binding = -1;
875 int index = -1;
876 int set = -1;
877 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500878 int inputAttachmentIndex = -1;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500879 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
880 int maxVertices = -1;
881 int invocations = -1;
Brian Osmanf59a9612020-04-15 14:18:13 -0400882 StringFragment marker;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400883 StringFragment when;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400884 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400885 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
886 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500887 return Layout(flags, location, offset, binding, index, set, builtin,
Brian Osman4717fbb2021-02-23 13:12:09 -0500888 inputAttachmentIndex, primitive, maxVertices, invocations, marker, when,
Brian Osmanba7ef322021-02-23 13:37:22 -0500889 ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700890 }
891 for (;;) {
892 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700893 String text = this->text(t);
Brian Osmana77ed8b2021-02-23 12:54:22 -0500894 auto setFlag = [&](Layout::Flag f) {
895 if (flags & f) {
896 this->error(t, "layout qualifier '" + text + "' appears more than once");
897 }
898 flags |= f;
899 };
900 auto setPrimitive = [&](Layout::Primitive p) {
901 if (flags & Layout::kPrimitive_Flag) {
902 this->error(t, "only one primitive-type layout qualifier is allowed");
903 }
904 flags |= Layout::kPrimitive_Flag;
905 primitive = p;
906 };
907
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400908 auto found = layoutTokens->find(text);
909 if (found != layoutTokens->end()) {
910 switch (found->second) {
Brian Osmana77ed8b2021-02-23 12:54:22 -0500911 case LayoutToken::ORIGIN_UPPER_LEFT:
912 setFlag(Layout::kOriginUpperLeft_Flag);
913 break;
914 case LayoutToken::OVERRIDE_COVERAGE:
915 setFlag(Layout::kOverrideCoverage_Flag);
916 break;
917 case LayoutToken::PUSH_CONSTANT:
918 setFlag(Layout::kPushConstant_Flag);
919 break;
920 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
921 setFlag(Layout::kBlendSupportAllEquations_Flag);
922 break;
923 case LayoutToken::TRACKED:
924 setFlag(Layout::kTracked_Flag);
925 break;
926 case LayoutToken::SRGB_UNPREMUL:
927 setFlag(Layout::kSRGBUnpremul_Flag);
928 break;
929 case LayoutToken::KEY:
930 setFlag(Layout::kKey_Flag);
931 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700932 case LayoutToken::LOCATION:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500933 setFlag(Layout::kLocation_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500934 location = this->layoutInt();
935 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700936 case LayoutToken::OFFSET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500937 setFlag(Layout::kOffset_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500938 offset = this->layoutInt();
939 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700940 case LayoutToken::BINDING:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500941 setFlag(Layout::kBinding_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500942 binding = this->layoutInt();
943 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700944 case LayoutToken::INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500945 setFlag(Layout::kIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500946 index = this->layoutInt();
947 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700948 case LayoutToken::SET:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500949 setFlag(Layout::kSet_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500950 set = this->layoutInt();
951 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700952 case LayoutToken::BUILTIN:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500953 setFlag(Layout::kBuiltin_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500954 builtin = this->layoutInt();
955 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700956 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500957 setFlag(Layout::kInputAttachmentIndex_Flag);
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500958 inputAttachmentIndex = this->layoutInt();
959 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700960 case LayoutToken::POINTS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500961 setPrimitive(Layout::kPoints_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500962 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700963 case LayoutToken::LINES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500964 setPrimitive(Layout::kLines_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500965 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700966 case LayoutToken::LINE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500967 setPrimitive(Layout::kLineStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500968 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700969 case LayoutToken::LINES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500970 setPrimitive(Layout::kLinesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500971 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700972 case LayoutToken::TRIANGLES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500973 setPrimitive(Layout::kTriangles_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500974 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700975 case LayoutToken::TRIANGLE_STRIP:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500976 setPrimitive(Layout::kTriangleStrip_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500977 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700978 case LayoutToken::TRIANGLES_ADJACENCY:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500979 setPrimitive(Layout::kTrianglesAdjacency_Primitive);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500980 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700981 case LayoutToken::MAX_VERTICES:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500982 setFlag(Layout::kMaxVertices_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500983 maxVertices = this->layoutInt();
984 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700985 case LayoutToken::INVOCATIONS:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500986 setFlag(Layout::kInvocations_Flag);
Ethan Nicholas52cad152017-02-16 16:37:32 -0500987 invocations = this->layoutInt();
988 break;
Brian Osmanf59a9612020-04-15 14:18:13 -0400989 case LayoutToken::MARKER:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500990 setFlag(Layout::kMarker_Flag);
Brian Osmanf59a9612020-04-15 14:18:13 -0400991 marker = this->layoutCode();
992 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700993 case LayoutToken::WHEN:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500994 setFlag(Layout::kWhen_Flag);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400995 when = this->layoutCode();
996 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -0400997 case LayoutToken::CTYPE:
Brian Osmana77ed8b2021-02-23 12:54:22 -0500998 setFlag(Layout::kCType_Flag);
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400999 ctype = this->layoutCType();
1000 break;
1001 default:
Brian Osmana77ed8b2021-02-23 12:54:22 -05001002 this->error(t, "'" + text + "' is not a valid layout qualifier");
Ethan Nicholasd608c092017-10-26 09:30:08 -04001003 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -05001004 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001005 } else {
Brian Osmana77ed8b2021-02-23 12:54:22 -05001006 this->error(t, "'" + text + "' is not a valid layout qualifier");
ethannicholasb3058bd2016-07-01 08:22:01 -07001007 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001008 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001009 break;
1010 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001011 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001012 break;
1013 }
1014 }
1015 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -05001016 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
Brian Osmanba7ef322021-02-23 13:37:22 -05001017 primitive, maxVertices, invocations, marker, when, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -07001018}
1019
Brian Salomonf9f45122016-11-29 11:59:17 -05001020/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Brian Osman58ee8982021-02-18 15:39:38 -05001021 VARYING | INLINE)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -05001022Modifiers Parser::modifiers() {
1023 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -07001024 int flags = 0;
1025 for (;;) {
1026 // TODO: handle duplicate / incompatible flags
John Stiles7d5b90a2021-01-21 14:26:51 -05001027 int tokenFlag = parse_modifier_token(peek().fKind);
1028 if (!tokenFlag) {
1029 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001030 }
John Stiles7d5b90a2021-01-21 14:26:51 -05001031 flags |= tokenFlag;
1032 this->nextToken();
ethannicholasb3058bd2016-07-01 08:22:01 -07001033 }
John Stiles7d5b90a2021-01-21 14:26:51 -05001034 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001035}
1036
Ethan Nicholas11d53972016-11-28 11:23:23 -05001037Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
1038 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -07001039 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001040 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001041 }
1042 return result;
1043}
1044
1045/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001046ASTNode::ID Parser::statement() {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001047 Token start = this->nextToken();
1048 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001049 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001050 return ASTNode::ID::Invalid();
1051 }
1052 this->pushback(start);
ethannicholasb3058bd2016-07-01 08:22:01 -07001053 switch (start.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001054 case Token::Kind::TK_IF: // fall through
1055 case Token::Kind::TK_STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -07001056 return this->ifStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001057 case Token::Kind::TK_FOR:
ethannicholasb3058bd2016-07-01 08:22:01 -07001058 return this->forStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001059 case Token::Kind::TK_DO:
ethannicholasb3058bd2016-07-01 08:22:01 -07001060 return this->doStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001061 case Token::Kind::TK_WHILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001062 return this->whileStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001063 case Token::Kind::TK_SWITCH: // fall through
1064 case Token::Kind::TK_STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -05001065 return this->switchStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001066 case Token::Kind::TK_RETURN:
ethannicholasb3058bd2016-07-01 08:22:01 -07001067 return this->returnStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001068 case Token::Kind::TK_BREAK:
ethannicholasb3058bd2016-07-01 08:22:01 -07001069 return this->breakStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001070 case Token::Kind::TK_CONTINUE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001071 return this->continueStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001072 case Token::Kind::TK_DISCARD:
ethannicholasb3058bd2016-07-01 08:22:01 -07001073 return this->discardStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001074 case Token::Kind::TK_LBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001075 return this->block();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001076 case Token::Kind::TK_SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001077 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -05001078 return this->createNode(start.fOffset, ASTNode::Kind::kBlock);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001079 case Token::Kind::TK_CONST:
Ethan Nicholasfc994162019-06-06 10:04:27 -04001080 return this->varDeclarations();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001081 case Token::Kind::TK_IDENTIFIER:
John Stiles76389b72021-01-25 13:49:05 -05001082 return this->varDeclarationsOrExpressionStatement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001083 default:
1084 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001085 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001086}
1087
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001088/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* QUESTION? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001089ASTNode::ID Parser::type() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001090 Token type;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001091 if (!this->expect(Token::Kind::TK_IDENTIFIER, "a type", &type)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001092 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001093 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001094 if (!this->isType(this->text(type))) {
1095 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001096 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001097 }
Brian Osman00fea5b2021-01-28 09:46:30 -05001098 ASTNode::ID result = this->createNode(type.fOffset, ASTNode::Kind::kType, this->text(type));
John Stilesd39aec92020-12-03 14:37:16 -05001099 bool isArray = false;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001100 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stilesd39aec92020-12-03 14:37:16 -05001101 if (isArray) {
1102 this->error(this->peek(), "multi-dimensional arrays are not supported");
1103 return ASTNode::ID::Invalid();
1104 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001105 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001106 SKSL_INT i;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001107 if (this->intLiteral(&i)) {
John Stiles08070f62020-11-17 10:45:49 -05001108 this->addChild(result, this->createNode(this->peek().fOffset,
1109 ASTNode::Kind::kInt, i));
Ethan Nicholas50afc172017-02-16 14:49:57 -05001110 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001111 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001112 }
1113 } else {
John Stiles3b209362020-11-16 17:03:10 -05001114 this->createEmptyChild(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001115 }
John Stilesd39aec92020-12-03 14:37:16 -05001116 isArray = true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001117 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001118 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001119 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001120}
1121
John Stiles57a996b2020-04-19 19:05:12 -07001122/* IDENTIFIER LBRACE
1123 varDeclaration+
1124 RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001125ASTNode::ID Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001126 Token name;
John Stiles2630ea32020-12-04 10:51:21 -05001127 if (!this->expectIdentifier(&name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001128 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001129 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001130 if (peek().fKind != Token::Kind::TK_LBRACE) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001131 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001132 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -07001133 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001134 this->error(name, "no type named '" + this->text(name) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001135 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001136 }
John Stiles3b209362020-11-16 17:03:10 -05001137 ASTNode::ID result = this->createNode(name.fOffset, ASTNode::Kind::kInterfaceBlock);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001138 ASTNode::InterfaceBlockData id(mods, this->text(name), 0, "", 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07001139 this->nextToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001140 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001141 ASTNode::ID decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001142 if (!decl) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001143 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001144 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001145 getNode(result).addChild(decl);
1146 ++id.fDeclarationCount;
ethannicholasb3058bd2016-07-01 08:22:01 -07001147 }
John Stiles57a996b2020-04-19 19:05:12 -07001148 if (id.fDeclarationCount == 0) {
1149 this->error(name, "interface block '" + this->text(name) +
1150 "' must contain at least one member");
1151 return ASTNode::ID::Invalid();
1152 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001153 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001154 std::vector<ASTNode> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001155 StringFragment instanceName;
1156 Token instanceNameToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001157 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &instanceNameToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001158 id.fInstanceName = this->text(instanceNameToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001159 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
John Stilesd39aec92020-12-03 14:37:16 -05001160 if (id.fIsArray) {
1161 this->error(this->peek(), "multi-dimensional arrays are not supported");
1162 return false;
1163 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001164 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001165 ASTNode::ID size = this->expression();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001166 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001167 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001168 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001169 getNode(result).addChild(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001170 } else {
John Stiles3b209362020-11-16 17:03:10 -05001171 this->createEmptyChild(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001172 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001173 this->expect(Token::Kind::TK_RBRACKET, "']'");
John Stilesd39aec92020-12-03 14:37:16 -05001174 id.fIsArray = true;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001175 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001176 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -07001177 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001178 getNode(result).setInterfaceBlockData(id);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001179 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001180 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001181}
1182
1183/* IF LPAREN expression RPAREN statement (ELSE statement)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001184ASTNode::ID Parser::ifStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001185 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001186 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_IF, &start);
1187 if (!isStatic && !this->expect(Token::Kind::TK_IF, "'if'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001188 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001189 }
John Stiles3b209362020-11-16 17:03:10 -05001190 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kIf, isStatic);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001191 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001192 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001193 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001194 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001195 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001196 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001197 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001198 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001199 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001200 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001201 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001202 ASTNode::ID ifTrue = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001203 if (!ifTrue) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001204 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001205 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001206 getNode(result).addChild(ifTrue);
1207 ASTNode::ID ifFalse;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001208 if (this->checkNext(Token::Kind::TK_ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001209 ifFalse = this->statement();
1210 if (!ifFalse) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001211 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001212 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001213 getNode(result).addChild(ifFalse);
ethannicholasb3058bd2016-07-01 08:22:01 -07001214 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001215 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001216}
1217
1218/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001219ASTNode::ID Parser::doStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001220 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001221 if (!this->expect(Token::Kind::TK_DO, "'do'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001222 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001223 }
John Stiles3b209362020-11-16 17:03:10 -05001224 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kDo);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001225 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001226 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001227 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001228 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001229 getNode(result).addChild(statement);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001230 if (!this->expect(Token::Kind::TK_WHILE, "'while'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001231 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001232 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001233 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001234 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001235 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001236 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001237 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001238 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001239 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001240 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001241 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001242 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001243 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001244 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001245 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001246 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001247 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001248}
1249
1250/* WHILE LPAREN expression RPAREN STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001251ASTNode::ID Parser::whileStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001252 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001253 if (!this->expect(Token::Kind::TK_WHILE, "'while'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001254 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001255 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001256 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001257 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001258 }
John Stiles3b209362020-11-16 17:03:10 -05001259 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kWhile);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001260 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001261 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001262 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001263 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001264 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001265 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001266 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001267 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001268 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001269 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001270 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001271 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001272 getNode(result).addChild(statement);
1273 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001274}
1275
Ethan Nicholasaf197692017-02-27 13:26:45 -05001276/* CASE expression COLON statement* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001277ASTNode::ID Parser::switchCase() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001278 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001279 if (!this->expect(Token::Kind::TK_CASE, "'case'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001280 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001281 }
John Stiles3b209362020-11-16 17:03:10 -05001282 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kSwitchCase);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001283 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001284 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001285 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001286 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001287 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001288 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001289 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001290 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001291 while (this->peek().fKind != Token::Kind::TK_RBRACE &&
1292 this->peek().fKind != Token::Kind::TK_CASE &&
1293 this->peek().fKind != Token::Kind::TK_DEFAULT) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001294 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001295 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001296 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001297 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001298 getNode(result).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001299 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001300 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001301}
1302
1303/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001304ASTNode::ID Parser::switchStatement() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001305 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001306 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_SWITCH, &start);
1307 if (!isStatic && !this->expect(Token::Kind::TK_SWITCH, "'switch'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001308 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001309 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001310 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001311 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001312 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001313 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001314 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001315 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001316 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001317 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001318 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001319 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001320 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001321 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001322 }
John Stiles3b209362020-11-16 17:03:10 -05001323 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kSwitch, isStatic);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001324 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001325 while (this->peek().fKind == Token::Kind::TK_CASE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001326 ASTNode::ID c = this->switchCase();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001327 if (!c) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001328 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001329 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001330 getNode(result).addChild(c);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001331 }
1332 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1333 // parts of the compiler may rely upon this assumption.
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001334 if (this->peek().fKind == Token::Kind::TK_DEFAULT) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001335 Token defaultStart;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001336 SkAssertResult(this->expect(Token::Kind::TK_DEFAULT, "'default'", &defaultStart));
1337 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001338 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001339 }
John Stiles3b209362020-11-16 17:03:10 -05001340 ASTNode::ID defaultCase = this->addChild(
1341 result, this->createNode(defaultStart.fOffset, ASTNode::Kind::kSwitchCase));
1342 this->createEmptyChild(defaultCase); // empty test to signify default case
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001343 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001344 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001345 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001346 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001347 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001348 getNode(defaultCase).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001349 }
Ethan Nicholasaf197692017-02-27 13:26:45 -05001350 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001351 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001352 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001353 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001354 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001355}
1356
Ethan Nicholas11d53972016-11-28 11:23:23 -05001357/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001358 STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001359ASTNode::ID Parser::forStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001360 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001361 if (!this->expect(Token::Kind::TK_FOR, "'for'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001362 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001363 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001364 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001365 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001366 }
John Stiles3b209362020-11-16 17:03:10 -05001367 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kFor);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001368 ASTNode::ID initializer;
ethannicholasb3058bd2016-07-01 08:22:01 -07001369 Token nextToken = this->peek();
1370 switch (nextToken.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001371 case Token::Kind::TK_SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001372 this->nextToken();
John Stiles3b209362020-11-16 17:03:10 -05001373 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001374 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001375 case Token::Kind::TK_CONST: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001376 initializer = this->varDeclarations();
1377 if (!initializer) {
1378 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001379 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001380 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001381 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001382 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001383 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001384 if (this->isType(this->text(nextToken))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001385 initializer = this->varDeclarations();
1386 if (!initializer) {
1387 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001388 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001389 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001390 break;
1391 }
John Stiles30212b72020-06-11 17:55:07 -04001392 [[fallthrough]];
1393 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001394 default:
1395 initializer = this->expressionStatement();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001396 if (!initializer) {
1397 return ASTNode::ID::Invalid();
1398 }
1399 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001400 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001401 ASTNode::ID test;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001402 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001403 test = this->expression();
1404 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001405 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001406 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001407 getNode(result).addChild(test);
1408 } else {
John Stiles3b209362020-11-16 17:03:10 -05001409 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001410 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001411 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001412 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001413 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001414 ASTNode::ID next;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001415 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001416 next = this->expression();
1417 if (!next) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001418 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001419 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001420 getNode(result).addChild(next);
1421 } else {
John Stiles3b209362020-11-16 17:03:10 -05001422 this->createEmptyChild(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001423 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001424 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001425 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001426 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001427 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001428 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001429 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001430 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001431 getNode(result).addChild(statement);
1432 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001433}
1434
1435/* RETURN expression? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001436ASTNode::ID Parser::returnStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001437 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001438 if (!this->expect(Token::Kind::TK_RETURN, "'return'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001439 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001440 }
John Stiles3b209362020-11-16 17:03:10 -05001441 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kReturn);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001442 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001443 ASTNode::ID expression = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001444 if (!expression) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001445 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001446 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001447 getNode(result).addChild(expression);
ethannicholasb3058bd2016-07-01 08:22:01 -07001448 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001449 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001450 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001451 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001452 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001453}
1454
1455/* BREAK SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001456ASTNode::ID Parser::breakStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001457 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001458 if (!this->expect(Token::Kind::TK_BREAK, "'break'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001459 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001460 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001461 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001462 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001463 }
John Stiles3b209362020-11-16 17:03:10 -05001464 return this->createNode(start.fOffset, ASTNode::Kind::kBreak);
ethannicholasb3058bd2016-07-01 08:22:01 -07001465}
1466
1467/* CONTINUE SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001468ASTNode::ID Parser::continueStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001469 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001470 if (!this->expect(Token::Kind::TK_CONTINUE, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001471 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001472 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001473 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001474 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001475 }
John Stiles3b209362020-11-16 17:03:10 -05001476 return this->createNode(start.fOffset, ASTNode::Kind::kContinue);
ethannicholasb3058bd2016-07-01 08:22:01 -07001477}
1478
1479/* DISCARD SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001480ASTNode::ID Parser::discardStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001481 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001482 if (!this->expect(Token::Kind::TK_DISCARD, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001483 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001484 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001485 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001486 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001487 }
John Stiles3b209362020-11-16 17:03:10 -05001488 return this->createNode(start.fOffset, ASTNode::Kind::kDiscard);
ethannicholasb3058bd2016-07-01 08:22:01 -07001489}
1490
1491/* LBRACE statement* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001492ASTNode::ID Parser::block() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001493 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001494 if (!this->expect(Token::Kind::TK_LBRACE, "'{'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001495 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001496 }
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001497 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001498 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001499 return ASTNode::ID::Invalid();
1500 }
John Stiles3b209362020-11-16 17:03:10 -05001501 ASTNode::ID result = this->createNode(start.fOffset, ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -07001502 for (;;) {
1503 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001504 case Token::Kind::TK_RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001505 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001506 return result;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001507 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001508 this->error(this->peek(), "expected '}', but found end of file");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001509 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001510 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001511 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001512 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001513 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001514 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001515 getNode(result).addChild(statement);
ethannicholasb3058bd2016-07-01 08:22:01 -07001516 }
1517 }
1518 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001519 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001520}
1521
1522/* expression SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001523ASTNode::ID Parser::expressionStatement() {
1524 ASTNode::ID expr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001525 if (expr) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001526 if (this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001527 return expr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001528 }
1529 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001530 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001531}
1532
1533/* assignmentExpression (COMMA assignmentExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001534ASTNode::ID Parser::expression() {
1535 ASTNode::ID result = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001536 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001537 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001538 }
1539 Token t;
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001540 AutoDepth depth(this);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001541 while (this->checkNext(Token::Kind::TK_COMMA, &t)) {
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001542 if (!depth.increase()) {
1543 return ASTNode::ID::Invalid();
1544 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001545 ASTNode::ID right = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001546 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001547 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001548 }
John Stiles45990502021-02-16 10:55:27 -05001549 ASTNode::ID newResult = this->createNode(t.fOffset, ASTNode::Kind::kBinary,
1550 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001551 getNode(newResult).addChild(result);
1552 getNode(newResult).addChild(right);
1553 result = newResult;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001554 }
1555 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001556}
1557
1558/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1559 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1560 assignmentExpression)*
1561 */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001562ASTNode::ID Parser::assignmentExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001563 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001564 ASTNode::ID result = this->ternaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001565 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001566 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001567 }
1568 for (;;) {
1569 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001570 case Token::Kind::TK_EQ: // fall through
1571 case Token::Kind::TK_STAREQ: // fall through
1572 case Token::Kind::TK_SLASHEQ: // fall through
1573 case Token::Kind::TK_PERCENTEQ: // fall through
1574 case Token::Kind::TK_PLUSEQ: // fall through
1575 case Token::Kind::TK_MINUSEQ: // fall through
1576 case Token::Kind::TK_SHLEQ: // fall through
1577 case Token::Kind::TK_SHREQ: // fall through
1578 case Token::Kind::TK_BITWISEANDEQ: // fall through
1579 case Token::Kind::TK_BITWISEXOREQ: // fall through
John Stiles8b3b1592020-11-23 11:06:44 -05001580 case Token::Kind::TK_BITWISEOREQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001581 if (!depth.increase()) {
1582 return ASTNode::ID::Invalid();
1583 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001584 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001585 ASTNode::ID right = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001586 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001587 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001588 }
John Stiles3b209362020-11-16 17:03:10 -05001589 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001590 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001591 getNode(newResult).addChild(result);
1592 getNode(newResult).addChild(right);
1593 result = newResult;
1594 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001595 }
1596 default:
1597 return result;
1598 }
1599 }
1600}
1601
1602/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001603ASTNode::ID Parser::ternaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001604 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001605 ASTNode::ID base = this->logicalOrExpression();
1606 if (!base) {
1607 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001608 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001609 if (this->checkNext(Token::Kind::TK_QUESTION)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001610 if (!depth.increase()) {
1611 return ASTNode::ID::Invalid();
1612 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001613 ASTNode::ID trueExpr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001614 if (!trueExpr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001615 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001616 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001617 if (this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001618 ASTNode::ID falseExpr = this->assignmentExpression();
1619 if (!falseExpr) {
1620 return ASTNode::ID::Invalid();
1621 }
John Stiles3b209362020-11-16 17:03:10 -05001622 ASTNode::ID ternary = this->createNode(getNode(base).fOffset, ASTNode::Kind::kTernary);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001623 getNode(ternary).addChild(base);
1624 getNode(ternary).addChild(trueExpr);
1625 getNode(ternary).addChild(falseExpr);
1626 return ternary;
ethannicholasb3058bd2016-07-01 08:22:01 -07001627 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001628 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001629 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001630 return base;
ethannicholasb3058bd2016-07-01 08:22:01 -07001631}
1632
1633/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001634ASTNode::ID Parser::logicalOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001635 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001636 ASTNode::ID result = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001637 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001638 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001639 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001640 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001641 while (this->checkNext(Token::Kind::TK_LOGICALOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001642 if (!depth.increase()) {
1643 return ASTNode::ID::Invalid();
1644 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001645 ASTNode::ID right = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001646 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001647 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001648 }
John Stiles3b209362020-11-16 17:03:10 -05001649 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001650 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001651 getNode(newResult).addChild(result);
1652 getNode(newResult).addChild(right);
1653 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001654 }
1655 return result;
1656}
1657
1658/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001659ASTNode::ID Parser::logicalXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001660 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001661 ASTNode::ID result = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001662 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001663 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001664 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001665 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001666 while (this->checkNext(Token::Kind::TK_LOGICALXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001667 if (!depth.increase()) {
1668 return ASTNode::ID::Invalid();
1669 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001670 ASTNode::ID right = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001671 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001672 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001673 }
John Stiles3b209362020-11-16 17:03:10 -05001674 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001675 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001676 getNode(newResult).addChild(result);
1677 getNode(newResult).addChild(right);
1678 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001679 }
1680 return result;
1681}
1682
1683/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001684ASTNode::ID Parser::logicalAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001685 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001686 ASTNode::ID result = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001687 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001688 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001689 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001690 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001691 while (this->checkNext(Token::Kind::TK_LOGICALAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001692 if (!depth.increase()) {
1693 return ASTNode::ID::Invalid();
1694 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001695 ASTNode::ID right = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001696 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001697 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001698 }
John Stiles3b209362020-11-16 17:03:10 -05001699 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001700 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001701 getNode(newResult).addChild(result);
1702 getNode(newResult).addChild(right);
1703 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001704 }
1705 return result;
1706}
1707
1708/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001709ASTNode::ID Parser::bitwiseOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001710 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001711 ASTNode::ID result = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001712 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001713 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001714 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001715 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001716 while (this->checkNext(Token::Kind::TK_BITWISEOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001717 if (!depth.increase()) {
1718 return ASTNode::ID::Invalid();
1719 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001720 ASTNode::ID right = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001721 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001722 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001723 }
John Stiles45990502021-02-16 10:55:27 -05001724 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
1725 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001726 getNode(newResult).addChild(result);
1727 getNode(newResult).addChild(right);
1728 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001729 }
1730 return result;
1731}
1732
1733/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001734ASTNode::ID Parser::bitwiseXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001735 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001736 ASTNode::ID result = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001737 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001738 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001739 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001740 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001741 while (this->checkNext(Token::Kind::TK_BITWISEXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001742 if (!depth.increase()) {
1743 return ASTNode::ID::Invalid();
1744 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001745 ASTNode::ID right = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001746 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001747 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001748 }
John Stiles45990502021-02-16 10:55:27 -05001749 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
1750 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001751 getNode(newResult).addChild(result);
1752 getNode(newResult).addChild(right);
1753 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001754 }
1755 return result;
1756}
1757
1758/* equalityExpression (BITWISEAND equalityExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001759ASTNode::ID Parser::bitwiseAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001760 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001761 ASTNode::ID result = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001762 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001763 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001764 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001765 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001766 while (this->checkNext(Token::Kind::TK_BITWISEAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001767 if (!depth.increase()) {
1768 return ASTNode::ID::Invalid();
1769 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001770 ASTNode::ID right = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001771 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001772 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001773 }
John Stiles3b209362020-11-16 17:03:10 -05001774 ASTNode::ID newResult = this->createNode(getNode(result).fOffset, ASTNode::Kind::kBinary,
John Stiles45990502021-02-16 10:55:27 -05001775 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001776 getNode(newResult).addChild(result);
1777 getNode(newResult).addChild(right);
1778 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001779 }
1780 return result;
1781}
1782
1783/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001784ASTNode::ID Parser::equalityExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001785 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001786 ASTNode::ID result = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001787 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001788 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001789 }
1790 for (;;) {
1791 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001792 case Token::Kind::TK_EQEQ: // fall through
1793 case Token::Kind::TK_NEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001794 if (!depth.increase()) {
1795 return ASTNode::ID::Invalid();
1796 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001797 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001798 ASTNode::ID right = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001799 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001800 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001801 }
John Stiles3b209362020-11-16 17:03:10 -05001802 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001803 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001804 getNode(newResult).addChild(result);
1805 getNode(newResult).addChild(right);
1806 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001807 break;
1808 }
1809 default:
1810 return result;
1811 }
1812 }
1813}
1814
1815/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001816ASTNode::ID Parser::relationalExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001817 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001818 ASTNode::ID result = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001819 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001820 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001821 }
1822 for (;;) {
1823 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001824 case Token::Kind::TK_LT: // fall through
1825 case Token::Kind::TK_GT: // fall through
1826 case Token::Kind::TK_LTEQ: // fall through
1827 case Token::Kind::TK_GTEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001828 if (!depth.increase()) {
1829 return ASTNode::ID::Invalid();
1830 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001831 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001832 ASTNode::ID right = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001833 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001834 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001835 }
John Stiles3b209362020-11-16 17:03:10 -05001836 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001837 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001838 getNode(newResult).addChild(result);
1839 getNode(newResult).addChild(right);
1840 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001841 break;
1842 }
1843 default:
1844 return result;
1845 }
1846 }
1847}
1848
1849/* additiveExpression ((SHL | SHR) additiveExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001850ASTNode::ID Parser::shiftExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001851 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001852 ASTNode::ID result = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001853 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001854 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001855 }
1856 for (;;) {
1857 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001858 case Token::Kind::TK_SHL: // fall through
1859 case Token::Kind::TK_SHR: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001860 if (!depth.increase()) {
1861 return ASTNode::ID::Invalid();
1862 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001863 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001864 ASTNode::ID right = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001865 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001866 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001867 }
John Stiles3b209362020-11-16 17:03:10 -05001868 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001869 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001870 getNode(newResult).addChild(result);
1871 getNode(newResult).addChild(right);
1872 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001873 break;
1874 }
1875 default:
1876 return result;
1877 }
1878 }
1879}
1880
1881/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001882ASTNode::ID Parser::additiveExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001883 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001884 ASTNode::ID result = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001885 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001886 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001887 }
1888 for (;;) {
1889 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001890 case Token::Kind::TK_PLUS: // fall through
1891 case Token::Kind::TK_MINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001892 if (!depth.increase()) {
1893 return ASTNode::ID::Invalid();
1894 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001895 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001896 ASTNode::ID right = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001897 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001898 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001899 }
John Stiles3b209362020-11-16 17:03:10 -05001900 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001901 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001902 getNode(newResult).addChild(result);
1903 getNode(newResult).addChild(right);
1904 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001905 break;
1906 }
1907 default:
1908 return result;
1909 }
1910 }
1911}
1912
1913/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001914ASTNode::ID Parser::multiplicativeExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001915 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001916 ASTNode::ID result = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001917 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001918 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001919 }
1920 for (;;) {
1921 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001922 case Token::Kind::TK_STAR: // fall through
1923 case Token::Kind::TK_SLASH: // fall through
1924 case Token::Kind::TK_PERCENT: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001925 if (!depth.increase()) {
1926 return ASTNode::ID::Invalid();
1927 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001928 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001929 ASTNode::ID right = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001930 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001931 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001932 }
John Stiles3b209362020-11-16 17:03:10 -05001933 ASTNode::ID newResult = this->createNode(getNode(result).fOffset,
John Stiles45990502021-02-16 10:55:27 -05001934 ASTNode::Kind::kBinary, Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001935 getNode(newResult).addChild(result);
1936 getNode(newResult).addChild(right);
1937 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001938 break;
1939 }
1940 default:
1941 return result;
1942 }
1943 }
1944}
1945
1946/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001947ASTNode::ID Parser::unaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001948 AutoDepth depth(this);
ethannicholasb3058bd2016-07-01 08:22:01 -07001949 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001950 case Token::Kind::TK_PLUS: // fall through
1951 case Token::Kind::TK_MINUS: // fall through
1952 case Token::Kind::TK_LOGICALNOT: // fall through
1953 case Token::Kind::TK_BITWISENOT: // fall through
1954 case Token::Kind::TK_PLUSPLUS: // fall through
1955 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001956 if (!depth.increase()) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001957 return ASTNode::ID::Invalid();
Ethan Nicholas6dcc3252019-02-20 15:18:36 -05001958 }
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001959 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001960 ASTNode::ID expr = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001961 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001962 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001963 }
John Stiles45990502021-02-16 10:55:27 -05001964 ASTNode::ID result = this->createNode(t.fOffset, ASTNode::Kind::kPrefix,
1965 Operator(t.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04001966 getNode(result).addChild(expr);
1967 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001968 }
1969 default:
1970 return this->postfixExpression();
1971 }
1972}
1973
1974/* term suffix* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001975ASTNode::ID Parser::postfixExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001976 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001977 ASTNode::ID result = this->term();
ethannicholasb3058bd2016-07-01 08:22:01 -07001978 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001979 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001980 }
1981 for (;;) {
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001982 Token t = this->peek();
1983 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001984 case Token::Kind::TK_FLOAT_LITERAL:
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001985 if (this->text(t)[0] != '.') {
1986 return result;
1987 }
John Stiles30212b72020-06-11 17:55:07 -04001988 [[fallthrough]];
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001989 case Token::Kind::TK_LBRACKET:
1990 case Token::Kind::TK_DOT:
1991 case Token::Kind::TK_LPAREN:
1992 case Token::Kind::TK_PLUSPLUS:
1993 case Token::Kind::TK_MINUSMINUS:
1994 case Token::Kind::TK_COLONCOLON:
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001995 if (!depth.increase()) {
1996 return ASTNode::ID::Invalid();
1997 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001998 result = this->suffix(result);
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001999 if (!result) {
2000 return ASTNode::ID::Invalid();
2001 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002002 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002003 default:
2004 return result;
2005 }
2006 }
2007}
2008
Ethan Nicholas11d53972016-11-28 11:23:23 -05002009/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
Ethan Nicholase455f652019-09-13 12:52:55 -04002010 PLUSPLUS | MINUSMINUS | COLONCOLON IDENTIFIER | FLOAT_LITERAL [IDENTIFIER] */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002011ASTNode::ID Parser::suffix(ASTNode::ID base) {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002012 SkASSERT(base);
ethannicholasb3058bd2016-07-01 08:22:01 -07002013 Token next = this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002014 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002015 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002016 return ASTNode::ID::Invalid();
2017 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002018 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002019 case Token::Kind::TK_LBRACKET: {
2020 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
John Stiles3b209362020-11-16 17:03:10 -05002021 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kIndex);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002022 getNode(result).addChild(base);
2023 return result;
ethannicholas5961bc92016-10-12 06:39:56 -07002024 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002025 ASTNode::ID e = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002026 if (!e) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002027 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002028 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002029 this->expect(Token::Kind::TK_RBRACKET, "']' to complete array access expression");
John Stiles3b209362020-11-16 17:03:10 -05002030 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kIndex);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002031 getNode(result).addChild(base);
2032 getNode(result).addChild(e);
2033 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002034 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002035 case Token::Kind::TK_COLONCOLON: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002036 int offset = this->peek().fOffset;
2037 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002038 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002039 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kScope,
2040 std::move(text));
Brian Osman6518d772020-09-10 16:50:06 -04002041 getNode(result).addChild(base);
2042 return result;
2043 }
2044 return ASTNode::ID::Invalid();
2045 }
2046 case Token::Kind::TK_DOT: {
2047 int offset = this->peek().fOffset;
2048 StringFragment text;
2049 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002050 ASTNode::ID result = this->createNode(offset, ASTNode::Kind::kField,
2051 std::move(text));
Ethan Nicholasfc994162019-06-06 10:04:27 -04002052 getNode(result).addChild(base);
2053 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002054 }
Brian Osman6518d772020-09-10 16:50:06 -04002055 [[fallthrough]];
Ethan Nicholase455f652019-09-13 12:52:55 -04002056 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002057 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholase455f652019-09-13 12:52:55 -04002058 // Swizzles that start with a constant number, e.g. '.000r', will be tokenized as
2059 // floating point literals, possibly followed by an identifier. Handle that here.
2060 StringFragment field = this->text(next);
2061 SkASSERT(field.fChars[0] == '.');
2062 ++field.fChars;
2063 --field.fLength;
2064 for (size_t i = 0; i < field.fLength; ++i) {
2065 if (field.fChars[i] != '0' && field.fChars[i] != '1') {
2066 this->error(next, "invalid swizzle");
2067 return ASTNode::ID::Invalid();
2068 }
2069 }
2070 // use the next *raw* token so we don't ignore whitespace - we only care about
2071 // identifiers that directly follow the float
2072 Token id = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002073 if (id.fKind == Token::Kind::TK_IDENTIFIER) {
Ethan Nicholase455f652019-09-13 12:52:55 -04002074 field.fLength += id.fLength;
2075 } else {
2076 this->pushback(id);
2077 }
John Stiles3b209362020-11-16 17:03:10 -05002078 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kField, field);
Ethan Nicholase455f652019-09-13 12:52:55 -04002079 getNode(result).addChild(base);
2080 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002081 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002082 case Token::Kind::TK_LPAREN: {
John Stiles3b209362020-11-16 17:03:10 -05002083 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kCall);
Ethan Nicholasfc994162019-06-06 10:04:27 -04002084 getNode(result).addChild(base);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002085 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002086 for (;;) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002087 ASTNode::ID expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002088 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002089 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002090 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002091 getNode(result).addChild(expr);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002092 if (!this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002093 break;
2094 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002095 }
2096 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002097 this->expect(Token::Kind::TK_RPAREN, "')' to complete function parameters");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002098 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002099 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002100 case Token::Kind::TK_PLUSPLUS: // fall through
2101 case Token::Kind::TK_MINUSMINUS: {
John Stiles45990502021-02-16 10:55:27 -05002102 ASTNode::ID result = this->createNode(next.fOffset, ASTNode::Kind::kPostfix,
2103 Operator(next.fKind));
Ethan Nicholasfc994162019-06-06 10:04:27 -04002104 getNode(result).addChild(base);
2105 return result;
2106 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002107 default: {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002108 this->error(next, "expected expression suffix, but found '" + this->text(next) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002109 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002110 }
2111 }
2112}
2113
Brian Osman33c64a42020-12-23 10:26:38 -05002114/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | '(' expression ')' */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002115ASTNode::ID Parser::term() {
ethannicholasb3058bd2016-07-01 08:22:01 -07002116 Token t = this->peek();
2117 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002118 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002119 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002120 if (this->identifier(&text)) {
John Stiles3b209362020-11-16 17:03:10 -05002121 return this->createNode(t.fOffset, ASTNode::Kind::kIdentifier, std::move(text));
ethannicholasb3058bd2016-07-01 08:22:01 -07002122 }
John Stiles30212b72020-06-11 17:55:07 -04002123 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002124 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002125 case Token::Kind::TK_INT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002126 SKSL_INT i;
ethannicholasb3058bd2016-07-01 08:22:01 -07002127 if (this->intLiteral(&i)) {
John Stiles3b209362020-11-16 17:03:10 -05002128 return this->createNode(t.fOffset, ASTNode::Kind::kInt, i);
ethannicholasb3058bd2016-07-01 08:22:01 -07002129 }
2130 break;
2131 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002132 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002133 SKSL_FLOAT f;
ethannicholasb3058bd2016-07-01 08:22:01 -07002134 if (this->floatLiteral(&f)) {
John Stiles3b209362020-11-16 17:03:10 -05002135 return this->createNode(t.fOffset, ASTNode::Kind::kFloat, f);
ethannicholasb3058bd2016-07-01 08:22:01 -07002136 }
2137 break;
2138 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002139 case Token::Kind::TK_TRUE_LITERAL: // fall through
2140 case Token::Kind::TK_FALSE_LITERAL: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002141 bool b;
2142 if (this->boolLiteral(&b)) {
John Stiles3b209362020-11-16 17:03:10 -05002143 return this->createNode(t.fOffset, ASTNode::Kind::kBool, b);
ethannicholasb3058bd2016-07-01 08:22:01 -07002144 }
2145 break;
2146 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002147 case Token::Kind::TK_LPAREN: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002148 this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002149 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002150 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002151 return ASTNode::ID::Invalid();
2152 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002153 ASTNode::ID result = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002154 if (result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002155 this->expect(Token::Kind::TK_RPAREN, "')' to complete expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002156 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002157 }
2158 break;
2159 }
2160 default:
2161 this->nextToken();
John Stilesf94348f2020-12-23 18:58:43 -05002162 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002163 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002164 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002165}
2166
2167/* INT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002168bool Parser::intLiteral(SKSL_INT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002169 Token t;
John Stilesf94348f2020-12-23 18:58:43 -05002170 if (!this->expect(Token::Kind::TK_INT_LITERAL, "integer literal", &t)) {
2171 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07002172 }
John Stilesf94348f2020-12-23 18:58:43 -05002173 StringFragment s = this->text(t);
2174 if (!SkSL::stoi(s, dest)) {
2175 this->error(t, "integer is too large: " + s);
2176 return false;
2177 }
2178 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07002179}
2180
John Stilesf94348f2020-12-23 18:58:43 -05002181
ethannicholasb3058bd2016-07-01 08:22:01 -07002182/* FLOAT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002183bool Parser::floatLiteral(SKSL_FLOAT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002184 Token t;
John Stilesf94348f2020-12-23 18:58:43 -05002185 if (!this->expect(Token::Kind::TK_FLOAT_LITERAL, "float literal", &t)) {
2186 return false;
ethannicholasb3058bd2016-07-01 08:22:01 -07002187 }
John Stilesf94348f2020-12-23 18:58:43 -05002188 StringFragment s = this->text(t);
2189 if (!SkSL::stod(s, dest)) {
2190 this->error(t, "floating-point value is too large: " + s);
2191 return false;
2192 }
2193 return true;
ethannicholasb3058bd2016-07-01 08:22:01 -07002194}
2195
2196/* TRUE_LITERAL | FALSE_LITERAL */
2197bool Parser::boolLiteral(bool* dest) {
2198 Token t = this->nextToken();
2199 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002200 case Token::Kind::TK_TRUE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002201 *dest = true;
2202 return true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002203 case Token::Kind::TK_FALSE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002204 *dest = false;
2205 return true;
2206 default:
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002207 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002208 return false;
2209 }
2210}
2211
2212/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002213bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002214 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002215 if (this->expect(Token::Kind::TK_IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002216 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07002217 return true;
2218 }
2219 return false;
2220}
2221
John Stilesa6841be2020-08-06 14:11:56 -04002222} // namespace SkSL