blob: fa80885ddadbe827fc4baa976561dcd51dc0fa4a [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
ethannicholascad64162016-10-27 10:54:02 -070024#define MAX_PARSE_DEPTH 50
25
26class AutoDepth {
27public:
28 AutoDepth(Parser* p)
Ethan Nicholascf4deab2019-09-13 16:28:14 -040029 : fParser(p)
30 , fDepth(0) {}
ethannicholascad64162016-10-27 10:54:02 -070031
32 ~AutoDepth() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -040033 fParser->fDepth -= fDepth;
ethannicholascad64162016-10-27 10:54:02 -070034 }
35
Ethan Nicholascf4deab2019-09-13 16:28:14 -040036 bool increase() {
37 ++fDepth;
38 ++fParser->fDepth;
ethannicholascad64162016-10-27 10:54:02 -070039 if (fParser->fDepth > MAX_PARSE_DEPTH) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070040 fParser->error(fParser->peek(), String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070041 return false;
42 }
43 return true;
44 }
45
46private:
47 Parser* fParser;
Ethan Nicholascf4deab2019-09-13 16:28:14 -040048 int fDepth;
ethannicholascad64162016-10-27 10:54:02 -070049};
50
Brian Salomon140f3da2018-08-23 13:51:27 +000051std::unordered_map<String, Parser::LayoutToken>* Parser::layoutTokens;
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040052
53void Parser::InitLayoutMap() {
Brian Salomon140f3da2018-08-23 13:51:27 +000054 layoutTokens = new std::unordered_map<String, LayoutToken>;
Brian Salomon23356442018-11-30 15:33:19 -050055 #define TOKEN(name, text) (*layoutTokens)[text] = LayoutToken::name
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040056 TOKEN(LOCATION, "location");
57 TOKEN(OFFSET, "offset");
58 TOKEN(BINDING, "binding");
59 TOKEN(INDEX, "index");
60 TOKEN(SET, "set");
61 TOKEN(BUILTIN, "builtin");
62 TOKEN(INPUT_ATTACHMENT_INDEX, "input_attachment_index");
63 TOKEN(ORIGIN_UPPER_LEFT, "origin_upper_left");
64 TOKEN(OVERRIDE_COVERAGE, "override_coverage");
65 TOKEN(BLEND_SUPPORT_ALL_EQUATIONS, "blend_support_all_equations");
66 TOKEN(BLEND_SUPPORT_MULTIPLY, "blend_support_multiply");
67 TOKEN(BLEND_SUPPORT_SCREEN, "blend_support_screen");
68 TOKEN(BLEND_SUPPORT_OVERLAY, "blend_support_overlay");
69 TOKEN(BLEND_SUPPORT_DARKEN, "blend_support_darken");
70 TOKEN(BLEND_SUPPORT_LIGHTEN, "blend_support_lighten");
71 TOKEN(BLEND_SUPPORT_COLORDODGE, "blend_support_colordodge");
72 TOKEN(BLEND_SUPPORT_COLORBURN, "blend_support_colorburn");
73 TOKEN(BLEND_SUPPORT_HARDLIGHT, "blend_support_hardlight");
74 TOKEN(BLEND_SUPPORT_SOFTLIGHT, "blend_support_softlight");
75 TOKEN(BLEND_SUPPORT_DIFFERENCE, "blend_support_difference");
76 TOKEN(BLEND_SUPPORT_EXCLUSION, "blend_support_exclusion");
77 TOKEN(BLEND_SUPPORT_HSL_HUE, "blend_support_hsl_hue");
78 TOKEN(BLEND_SUPPORT_HSL_SATURATION, "blend_support_hsl_saturation");
79 TOKEN(BLEND_SUPPORT_HSL_COLOR, "blend_support_hsl_color");
80 TOKEN(BLEND_SUPPORT_HSL_LUMINOSITY, "blend_support_hsl_luminosity");
81 TOKEN(PUSH_CONSTANT, "push_constant");
82 TOKEN(POINTS, "points");
83 TOKEN(LINES, "lines");
84 TOKEN(LINE_STRIP, "line_strip");
85 TOKEN(LINES_ADJACENCY, "lines_adjacency");
86 TOKEN(TRIANGLES, "triangles");
87 TOKEN(TRIANGLE_STRIP, "triangle_strip");
88 TOKEN(TRIANGLES_ADJACENCY, "triangles_adjacency");
89 TOKEN(MAX_VERTICES, "max_vertices");
90 TOKEN(INVOCATIONS, "invocations");
Brian Osmanf59a9612020-04-15 14:18:13 -040091 TOKEN(MARKER, "marker");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040092 TOKEN(WHEN, "when");
93 TOKEN(KEY, "key");
Michael Ludwiga4275592018-08-31 10:52:47 -040094 TOKEN(TRACKED, "tracked");
Brian Osmanb32d66b2020-04-30 17:12:03 -040095 TOKEN(SRGB_UNPREMUL, "srgb_unpremul");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040096 TOKEN(CTYPE, "ctype");
Brian Osmanf28e55d2018-10-03 16:35:54 -040097 TOKEN(SKPMCOLOR4F, "SkPMColor4f");
Mike Reedb26b4e72020-01-22 14:31:21 -050098 TOKEN(SKV4, "SkV4");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040099 TOKEN(SKRECT, "SkRect");
100 TOKEN(SKIRECT, "SkIRect");
101 TOKEN(SKPMCOLOR, "SkPMColor");
Mike Reedb26b4e72020-01-22 14:31:21 -0500102 TOKEN(SKM44, "SkM44");
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400103 TOKEN(BOOL, "bool");
104 TOKEN(INT, "int");
105 TOKEN(FLOAT, "float");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400106 #undef TOKEN
107}
108
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400109Parser::Parser(const char* text, size_t length, SymbolTable& symbols, ErrorReporter& errors)
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700110: fText(text)
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400111, fPushback(Token::Kind::TK_INVALID, -1, -1)
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400112, fSymbols(symbols)
ethannicholasb3058bd2016-07-01 08:22:01 -0700113, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700114 fLexer.start(text, length);
Brian Salomon3b83afe2018-08-23 11:04:36 -0400115 static const bool layoutMapInitialized = []{ return (void)InitLayoutMap(), true; }();
116 (void) layoutMapInitialized;
ethannicholasb3058bd2016-07-01 08:22:01 -0700117}
118
Ethan Nicholasfc994162019-06-06 10:04:27 -0400119#define CREATE_NODE(result, ...) \
120 ASTNode::ID result(fFile->fNodes.size()); \
121 fFile->fNodes.emplace_back(&fFile->fNodes, __VA_ARGS__)
122
123#define RETURN_NODE(...) \
124 do { \
125 CREATE_NODE(result, __VA_ARGS__); \
126 return result; \
127 } while (false)
128
129#define CREATE_CHILD(child, target, ...) \
130 CREATE_NODE(child, __VA_ARGS__); \
131 fFile->fNodes[target.fValue].addChild(child)
132
133#define CREATE_EMPTY_CHILD(target) \
134 do { \
135 ASTNode::ID child(fFile->fNodes.size()); \
136 fFile->fNodes.emplace_back(); \
137 fFile->fNodes[target.fValue].addChild(child); \
138 } while (false)
139
Ethan Nicholasf7b88202017-09-18 14:10:39 -0400140/* (directive | section | declaration)* END_OF_FILE */
Ethan Nicholasba9a04f2020-11-06 09:28:04 -0500141std::unique_ptr<ASTFile> Parser::compilationUnit() {
John Stilesfbd050b2020-08-03 13:21:46 -0400142 fFile = std::make_unique<ASTFile>();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400143 CREATE_NODE(result, 0, ASTNode::Kind::kFile);
144 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700145 for (;;) {
146 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400147 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400148 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400149 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400150 ASTNode::ID dir = this->directive();
151 if (fErrors.errorCount()) {
152 return nullptr;
153 }
154 if (dir) {
155 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700156 }
157 break;
158 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400159 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400160 ASTNode::ID section = this->section();
161 if (fErrors.errorCount()) {
162 return nullptr;
163 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400164 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400165 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400166 }
167 break;
168 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700169 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400170 ASTNode::ID decl = this->declaration();
171 if (fErrors.errorCount()) {
172 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700173 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400174 if (decl) {
175 getNode(result).addChild(decl);
176 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700177 }
178 }
179 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400180 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700181}
182
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700183Token Parser::nextRawToken() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400184 if (fPushback.fKind != Token::Kind::TK_INVALID) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700185 Token result = fPushback;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400186 fPushback.fKind = Token::Kind::TK_INVALID;
ethannicholasb3058bd2016-07-01 08:22:01 -0700187 return result;
188 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700189 Token result = fLexer.next();
190 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400191}
192
193Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700194 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400195 while (token.fKind == Token::Kind::TK_WHITESPACE ||
196 token.fKind == Token::Kind::TK_LINE_COMMENT ||
197 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700198 token = this->nextRawToken();
199 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400200 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700201}
202
203void Parser::pushback(Token t) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400204 SkASSERT(fPushback.fKind == Token::Kind::TK_INVALID);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400205 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700206}
207
208Token Parser::peek() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400209 if (fPushback.fKind == Token::Kind::TK_INVALID) {
Brian Osman634624a2017-08-15 11:14:30 -0400210 fPushback = this->nextToken();
211 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700212 return fPushback;
213}
214
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400215bool Parser::checkNext(Token::Kind kind, Token* result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400216 if (fPushback.fKind != Token::Kind::TK_INVALID && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400217 return false;
218 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400219 Token next = this->nextToken();
220 if (next.fKind == kind) {
221 if (result) {
222 *result = next;
223 }
224 return true;
225 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400226 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400227 return false;
228}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500229
230bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700231 Token next = this->nextToken();
232 if (next.fKind == kind) {
233 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400234 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700235 }
236 return true;
237 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700238 this->error(next, "expected " + String(expected) + ", but found '" +
239 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700240 return false;
241 }
242}
243
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700244StringFragment Parser::text(Token token) {
245 return StringFragment(fText + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500246}
247
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700248void Parser::error(Token token, String msg) {
249 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700250}
251
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700252void Parser::error(int offset, String msg) {
253 fErrors.error(offset, msg);
254}
255
256bool Parser::isType(StringFragment name) {
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400257 const Symbol* s = fSymbols[name];
Ethan Nicholase6592142020-09-08 10:22:09 -0400258 return s && s->kind() == Symbol::Kind::kType;
ethannicholasb3058bd2016-07-01 08:22:01 -0700259}
260
Ethan Nicholas11d53972016-11-28 11:23:23 -0500261/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700262 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400263ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700264 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400265 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400266 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700267 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700268 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400269 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700270 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400271 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400272 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700273 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400274 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400275 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700276 }
277 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400278 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400279 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700280 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400281 RETURN_NODE(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700282 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700283 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400284 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700285 }
286}
287
Ethan Nicholas762466e2017-06-29 10:03:38 -0400288/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
289 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400290ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400291 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400292 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400293 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400294 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400295 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400296 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400297 this->nextToken();
298 Token argToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400299 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400300 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400301 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700302 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400303 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400304 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400305 }
306 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400307 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400308 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400309 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400310 StringFragment text;
311 Token codeStart = this->nextRawToken();
312 size_t startOffset = codeStart.fOffset;
313 this->pushback(codeStart);
314 text.fChars = fText + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400315 int level = 1;
316 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700317 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400318 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400319 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400320 ++level;
321 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400322 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400323 --level;
324 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400325 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700326 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400327 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400328 default:
329 break;
330 }
331 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400332 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400333 break;
334 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400335 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700336 StringFragment name = this->text(start);
337 ++name.fChars;
338 --name.fLength;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400339 RETURN_NODE(start.fOffset, ASTNode::Kind::kSection,
340 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400341}
342
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500343/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
344 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400345ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500346 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400347 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400348 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500349 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400350 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400351 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500352 }
353 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400354 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400355 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500356 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400357 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400358 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500359 }
John Stilesb8cc6652020-10-08 09:12:07 -0400360 fSymbols.add(std::make_unique<Type>(this->text(name), Type::TypeKind::kEnum));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400361 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400362 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500363 Token id;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400364 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400365 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500366 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400367 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400368 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500369 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400370 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500371 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400372 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
373 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500374 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400375 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500376 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400377 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
378 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400379 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500380 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400381 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400382 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500383 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400384 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400385 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500386 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400387 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500388 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400389 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
390 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500391 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400392 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500393 }
394 }
395 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400396 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400397 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500398}
399
400/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700401 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400402ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700403 Token lookahead = this->peek();
Ethan Nicholasda6320c2020-09-02 14:08:23 -0400404 switch (lookahead.fKind) {
405 case Token::Kind::TK_ENUM:
406 return this->enumDeclaration();
407 case Token::Kind::TK_SEMICOLON:
408 this->error(lookahead.fOffset, "expected a declaration, but found ';'");
409 return ASTNode::ID::Invalid();
410 default:
411 break;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500412 }
413 Modifiers modifiers = this->modifiers();
414 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400415 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700416 // we have an identifier that's not a type, could be the start of an interface block
417 return this->interfaceBlock(modifiers);
418 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400419 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700420 return this->structVarDeclaration(modifiers);
421 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400422 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700423 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400424 RETURN_NODE(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700425 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400426 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700427 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400428 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700429 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400430 if (getNode(type).getTypeData().fIsStructDeclaration &&
431 this->checkNext(Token::Kind::TK_SEMICOLON)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400432 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700433 }
434 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400435 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400436 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700437 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400438 if (this->checkNext(Token::Kind::TK_LPAREN)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400439 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kFunction);
440 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
441 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400442 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400443 for (;;) {
444 ASTNode::ID parameter = this->parameter();
445 if (!parameter) {
446 return ASTNode::ID::Invalid();
447 }
448 ++fd.fParameterCount;
449 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400450 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400451 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700452 }
453 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700454 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400455 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400456 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400457 return ASTNode::ID::Invalid();
458 }
459 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400460 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700461 body = this->block();
462 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400463 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700464 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400465 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700466 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400467 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700468 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400469 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700470 }
471}
472
473/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400474ASTNode::ID Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500475 Modifiers modifiers = this->modifiers();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400476 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700477 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400478 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700479 }
480 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400481 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400482 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700483 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400484 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700485}
486
487/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400488ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400489 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400490 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700491 }
492 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400493 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400494 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700495 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400496 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400497 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700498 }
499 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400500 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400501 ASTNode::ID decls = this->varDeclarations();
502 if (!decls) {
503 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700504 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400505 ASTNode& declsNode = getNode(decls);
John Stiles13fc2602020-10-09 17:42:31 -0400506 Modifiers modifiers = declsNode.begin()->getModifiers();
507 if (modifiers.fFlags != Modifiers::kNo_Flag) {
508 String desc = modifiers.description();
509 desc.pop_back(); // remove trailing space
510 this->error(declsNode.fOffset,
511 "modifier '" + desc + "' is not permitted on a struct field");
512 }
513
Ethan Nicholasc18bb512020-07-28 14:46:53 -0400514 const Symbol* symbol = fSymbols[(declsNode.begin() + 1)->getTypeData().fName];
John Stiles13fc2602020-10-09 17:42:31 -0400515 SkASSERT(symbol);
516 const Type* type = &symbol->as<Type>();
517
Ethan Nicholasfc994162019-06-06 10:04:27 -0400518 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
519 ASTNode& var = *iter;
520 ASTNode::VarData vd = var.getVarData();
521 for (int j = vd.fSizeCount - 1; j >= 0; j--) {
522 const ASTNode& size = *(var.begin() + j);
523 if (!size || size.fKind != ASTNode::Kind::kInt) {
524 this->error(declsNode.fOffset, "array size in struct field must be a constant");
525 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700526 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400527 uint64_t columns = size.getInt();
John Stilesf621e232020-08-25 13:33:02 -0400528 String typeName = type->name() + "[" + to_string(columns) + "]";
John Stiles3ae071e2020-08-05 15:29:29 -0400529 type = fSymbols.takeOwnershipOfSymbol(
Ethan Nicholase6592142020-09-08 10:22:09 -0400530 std::make_unique<Type>(typeName, Type::TypeKind::kArray, *type,
531 (int)columns));
ethannicholasb3058bd2016-07-01 08:22:01 -0700532 }
John Stiles13fc2602020-10-09 17:42:31 -0400533
534 fields.push_back(Type::Field(modifiers, vd.fName, type));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400535 if (vd.fSizeCount ? (var.begin() + (vd.fSizeCount - 1))->fNext : var.fFirstChild) {
536 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700537 }
538 }
539 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400540 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400541 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700542 }
John Stilesb8cc6652020-10-08 09:12:07 -0400543 fSymbols.add(std::make_unique<Type>(name.fOffset, this->text(name), fields));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400544 RETURN_NODE(name.fOffset, ASTNode::Kind::kType,
545 ASTNode::TypeData(this->text(name), true, false));
ethannicholasb3058bd2016-07-01 08:22:01 -0700546}
547
548/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400549ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
550 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700551 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400552 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700553 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400554 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400555 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400556 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700557 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400558 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400559 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700560}
561
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400562/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
563 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400564ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
565 CREATE_NODE(result, -1, ASTNode::Kind::kVarDeclarations);
566 CREATE_CHILD(modifiers, result, -1, ASTNode::Kind::kModifiers, mods);
567 getNode(result).addChild(type);
568 CREATE_NODE(currentVar, -1, ASTNode::Kind::kVarDeclaration);
569 ASTNode::VarData vd(name, 0);
570 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400571 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
572 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400573 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700574 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400575 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700576 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400577 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700578 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400579 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400580 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400581 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700582 }
583 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400584 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700585 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400586 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400587 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400588 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700589 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400590 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700591 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400592 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700593 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400594 while (this->checkNext(Token::Kind::TK_COMMA)) {
John Stilesf621e232020-08-25 13:33:02 -0400595 Token identifierName;
596 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &identifierName)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400597 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700598 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400599 currentVar = ASTNode::ID(fFile->fNodes.size());
John Stilesf621e232020-08-25 13:33:02 -0400600 vd = ASTNode::VarData(this->text(identifierName), 0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400601 fFile->fNodes.emplace_back(&fFile->fNodes, -1, ASTNode::Kind::kVarDeclaration);
602 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400603 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
604 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400605 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700606 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400607 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700608 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400609 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700610 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400611 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400612 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400613 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700614 }
615 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400616 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700617 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400618 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400619 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400620 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700621 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400622 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700623 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400624 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700625 }
626 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400627 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400628 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700629 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400630 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700631}
632
633/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400634ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400635 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400636 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700637 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400638 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700639 }
640 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400641 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400642 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700643 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400644 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kParameter);
645 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
646 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400647 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700648 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400649 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400650 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700651 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400652 CREATE_CHILD(child, result, sizeToken.fOffset, ASTNode::Kind::kInt,
653 SkSL::stoi(this->text(sizeToken)));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400654 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400655 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700656 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400657 ++pd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700658 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400659 getNode(result).setParameterData(pd);
660 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700661}
662
Ethan Nicholasd608c092017-10-26 09:30:08 -0400663/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700664int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400665 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700666 return -1;
667 }
668 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400669 if (this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700670 return SkSL::stoi(this->text(resultToken));
ethannicholasb3058bd2016-07-01 08:22:01 -0700671 }
672 return -1;
673}
674
Ethan Nicholasd608c092017-10-26 09:30:08 -0400675/** EQ IDENTIFIER */
676StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400677 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400678 return StringFragment();
679 }
680 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400681 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400682 return StringFragment();
683 }
684 return this->text(resultToken);
685}
686
687
Ethan Nicholas762466e2017-06-29 10:03:38 -0400688/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400689StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400690 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400691 return "";
692 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700693 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400694 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400695 StringFragment code;
696 code.fChars = fText + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400697 int level = 1;
698 bool done = false;
699 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700700 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400701 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400702 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400703 ++level;
704 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400705 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400706 --level;
707 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400708 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400709 if (level == 1) {
710 done = true;
711 }
712 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400713 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700714 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400715 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400716 default:
717 break;
718 }
719 if (!level) {
720 done = true;
721 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400722 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400723 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400724 this->pushback(std::move(next));
725 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400726 }
727 return code;
728}
729
730/** (EQ IDENTIFIER('identity'))? */
731Layout::Key Parser::layoutKey() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400732 if (this->peek().fKind == Token::Kind::TK_EQ) {
733 this->expect(Token::Kind::TK_EQ, "'='");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400734 Token key;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400735 if (this->expect(Token::Kind::TK_IDENTIFIER, "an identifer", &key)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700736 if (this->text(key) == "identity") {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400737 return Layout::kIdentity_Key;
738 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700739 this->error(key, "unsupported layout key");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400740 }
741 }
742 }
743 return Layout::kKey_Key;
744}
745
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400746Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400747 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400748 Token t = this->nextToken();
749 String text = this->text(t);
750 auto found = layoutTokens->find(text);
751 if (found != layoutTokens->end()) {
752 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400753 case LayoutToken::SKPMCOLOR4F:
754 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500755 case LayoutToken::SKV4:
756 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400757 case LayoutToken::SKRECT:
758 return Layout::CType::kSkRect;
759 case LayoutToken::SKIRECT:
760 return Layout::CType::kSkIRect;
761 case LayoutToken::SKPMCOLOR:
762 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400763 case LayoutToken::BOOL:
764 return Layout::CType::kBool;
765 case LayoutToken::INT:
766 return Layout::CType::kInt32;
767 case LayoutToken::FLOAT:
768 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500769 case LayoutToken::SKM44:
770 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400771 default:
772 break;
773 }
774 }
775 this->error(t, "unsupported ctype");
776 }
777 return Layout::CType::kDefault;
778}
779
ethannicholas8ac838d2016-11-22 08:39:36 -0800780/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500781Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500782 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700783 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500784 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700785 int binding = -1;
786 int index = -1;
787 int set = -1;
788 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500789 int inputAttachmentIndex = -1;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500790 Layout::Format format = Layout::Format::kUnspecified;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500791 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
792 int maxVertices = -1;
793 int invocations = -1;
Brian Osmanf59a9612020-04-15 14:18:13 -0400794 StringFragment marker;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400795 StringFragment when;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400796 Layout::Key key = Layout::kNo_Key;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400797 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400798 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
799 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500800 return Layout(flags, location, offset, binding, index, set, builtin,
Brian Osmanf59a9612020-04-15 14:18:13 -0400801 inputAttachmentIndex, format, primitive, maxVertices, invocations, marker,
802 when, key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700803 }
804 for (;;) {
805 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700806 String text = this->text(t);
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400807 auto found = layoutTokens->find(text);
808 if (found != layoutTokens->end()) {
809 switch (found->second) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700810 case LayoutToken::LOCATION:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500811 location = this->layoutInt();
812 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700813 case LayoutToken::OFFSET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500814 offset = this->layoutInt();
815 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700816 case LayoutToken::BINDING:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500817 binding = this->layoutInt();
818 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700819 case LayoutToken::INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500820 index = this->layoutInt();
821 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700822 case LayoutToken::SET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500823 set = this->layoutInt();
824 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700825 case LayoutToken::BUILTIN:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500826 builtin = this->layoutInt();
827 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700828 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500829 inputAttachmentIndex = this->layoutInt();
830 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700831 case LayoutToken::ORIGIN_UPPER_LEFT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500832 flags |= Layout::kOriginUpperLeft_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500833 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700834 case LayoutToken::OVERRIDE_COVERAGE:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500835 flags |= Layout::kOverrideCoverage_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500836 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700837 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500838 flags |= Layout::kBlendSupportAllEquations_Flag;
839 break;
840 case LayoutToken::BLEND_SUPPORT_MULTIPLY:
841 flags |= Layout::kBlendSupportMultiply_Flag;
842 break;
843 case LayoutToken::BLEND_SUPPORT_SCREEN:
844 flags |= Layout::kBlendSupportScreen_Flag;
845 break;
846 case LayoutToken::BLEND_SUPPORT_OVERLAY:
847 flags |= Layout::kBlendSupportOverlay_Flag;
848 break;
849 case LayoutToken::BLEND_SUPPORT_DARKEN:
850 flags |= Layout::kBlendSupportDarken_Flag;
851 break;
852 case LayoutToken::BLEND_SUPPORT_LIGHTEN:
853 flags |= Layout::kBlendSupportLighten_Flag;
854 break;
855 case LayoutToken::BLEND_SUPPORT_COLORDODGE:
856 flags |= Layout::kBlendSupportColorDodge_Flag;
857 break;
858 case LayoutToken::BLEND_SUPPORT_COLORBURN:
859 flags |= Layout::kBlendSupportColorBurn_Flag;
860 break;
861 case LayoutToken::BLEND_SUPPORT_HARDLIGHT:
862 flags |= Layout::kBlendSupportHardLight_Flag;
863 break;
864 case LayoutToken::BLEND_SUPPORT_SOFTLIGHT:
865 flags |= Layout::kBlendSupportSoftLight_Flag;
866 break;
867 case LayoutToken::BLEND_SUPPORT_DIFFERENCE:
868 flags |= Layout::kBlendSupportDifference_Flag;
869 break;
870 case LayoutToken::BLEND_SUPPORT_EXCLUSION:
871 flags |= Layout::kBlendSupportExclusion_Flag;
872 break;
873 case LayoutToken::BLEND_SUPPORT_HSL_HUE:
874 flags |= Layout::kBlendSupportHSLHue_Flag;
875 break;
876 case LayoutToken::BLEND_SUPPORT_HSL_SATURATION:
877 flags |= Layout::kBlendSupportHSLSaturation_Flag;
878 break;
879 case LayoutToken::BLEND_SUPPORT_HSL_COLOR:
880 flags |= Layout::kBlendSupportHSLColor_Flag;
881 break;
882 case LayoutToken::BLEND_SUPPORT_HSL_LUMINOSITY:
883 flags |= Layout::kBlendSupportHSLLuminosity_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500884 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700885 case LayoutToken::PUSH_CONSTANT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500886 flags |= Layout::kPushConstant_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500887 break;
Michael Ludwiga4275592018-08-31 10:52:47 -0400888 case LayoutToken::TRACKED:
889 flags |= Layout::kTracked_Flag;
890 break;
Brian Osmanb32d66b2020-04-30 17:12:03 -0400891 case LayoutToken::SRGB_UNPREMUL:
892 flags |= Layout::kSRGBUnpremul_Flag;
893 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700894 case LayoutToken::POINTS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500895 primitive = Layout::kPoints_Primitive;
896 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700897 case LayoutToken::LINES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500898 primitive = Layout::kLines_Primitive;
899 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700900 case LayoutToken::LINE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500901 primitive = Layout::kLineStrip_Primitive;
902 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700903 case LayoutToken::LINES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500904 primitive = Layout::kLinesAdjacency_Primitive;
905 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700906 case LayoutToken::TRIANGLES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500907 primitive = Layout::kTriangles_Primitive;
908 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700909 case LayoutToken::TRIANGLE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500910 primitive = Layout::kTriangleStrip_Primitive;
911 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700912 case LayoutToken::TRIANGLES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500913 primitive = Layout::kTrianglesAdjacency_Primitive;
914 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700915 case LayoutToken::MAX_VERTICES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500916 maxVertices = this->layoutInt();
917 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700918 case LayoutToken::INVOCATIONS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500919 invocations = this->layoutInt();
920 break;
Brian Osmanf59a9612020-04-15 14:18:13 -0400921 case LayoutToken::MARKER:
922 marker = this->layoutCode();
923 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700924 case LayoutToken::WHEN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400925 when = this->layoutCode();
926 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700927 case LayoutToken::KEY:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400928 key = this->layoutKey();
929 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -0400930 case LayoutToken::CTYPE:
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400931 ctype = this->layoutCType();
932 break;
933 default:
934 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
Ethan Nicholasd608c092017-10-26 09:30:08 -0400935 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500936 }
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400937 } else if (Layout::ReadFormat(text, &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500938 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700939 } else {
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400940 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
ethannicholasb3058bd2016-07-01 08:22:01 -0700941 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400942 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700943 break;
944 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400945 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700946 break;
947 }
948 }
949 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500950 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
Brian Osmanf59a9612020-04-15 14:18:13 -0400951 format, primitive, maxVertices, invocations, marker, when, key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700952}
953
Brian Salomonf9f45122016-11-29 11:59:17 -0500954/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Ethan Nicholasa7ceb502019-01-11 10:31:48 -0500955 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT | BUFFER | PLS | PLSIN |
Ethan Nicholasf3c8f5d2020-08-20 13:09:14 +0000956 PLSOUT | VARYING | INLINE)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500957Modifiers Parser::modifiers() {
958 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700959 int flags = 0;
960 for (;;) {
961 // TODO: handle duplicate / incompatible flags
962 switch (peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400963 case Token::Kind::TK_UNIFORM:
ethannicholasb3058bd2016-07-01 08:22:01 -0700964 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500965 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700966 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400967 case Token::Kind::TK_CONST:
ethannicholasb3058bd2016-07-01 08:22:01 -0700968 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500969 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700970 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400971 case Token::Kind::TK_IN:
ethannicholasb3058bd2016-07-01 08:22:01 -0700972 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500973 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700974 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400975 case Token::Kind::TK_OUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700976 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500977 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700978 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400979 case Token::Kind::TK_INOUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700980 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500981 flags |= Modifiers::kIn_Flag;
982 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700983 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400984 case Token::Kind::TK_FLAT:
ethannicholasf789b382016-08-03 12:43:36 -0700985 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500986 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700987 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400988 case Token::Kind::TK_NOPERSPECTIVE:
ethannicholasf789b382016-08-03 12:43:36 -0700989 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500990 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700991 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400992 case Token::Kind::TK_READONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500993 this->nextToken();
994 flags |= Modifiers::kReadOnly_Flag;
995 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400996 case Token::Kind::TK_WRITEONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500997 this->nextToken();
998 flags |= Modifiers::kWriteOnly_Flag;
999 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001000 case Token::Kind::TK_COHERENT:
Brian Salomonf9f45122016-11-29 11:59:17 -05001001 this->nextToken();
1002 flags |= Modifiers::kCoherent_Flag;
1003 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001004 case Token::Kind::TK_VOLATILE:
Brian Salomonf9f45122016-11-29 11:59:17 -05001005 this->nextToken();
1006 flags |= Modifiers::kVolatile_Flag;
1007 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001008 case Token::Kind::TK_RESTRICT:
Brian Salomonf9f45122016-11-29 11:59:17 -05001009 this->nextToken();
1010 flags |= Modifiers::kRestrict_Flag;
1011 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001012 case Token::Kind::TK_BUFFER:
Ethan Nicholas0dd30d92017-05-01 16:57:07 -04001013 this->nextToken();
1014 flags |= Modifiers::kBuffer_Flag;
1015 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001016 case Token::Kind::TK_HASSIDEEFFECTS:
Ethan Nicholascb670962017-04-20 19:31:52 -04001017 this->nextToken();
1018 flags |= Modifiers::kHasSideEffects_Flag;
1019 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001020 case Token::Kind::TK_PLS:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001021 this->nextToken();
1022 flags |= Modifiers::kPLS_Flag;
1023 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001024 case Token::Kind::TK_PLSIN:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001025 this->nextToken();
1026 flags |= Modifiers::kPLSIn_Flag;
1027 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001028 case Token::Kind::TK_PLSOUT:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001029 this->nextToken();
1030 flags |= Modifiers::kPLSOut_Flag;
1031 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001032 case Token::Kind::TK_VARYING:
Brian Osman3c358422020-03-23 10:44:12 -04001033 this->nextToken();
1034 flags |= Modifiers::kVarying_Flag;
1035 break;
Ethan Nicholasf3c8f5d2020-08-20 13:09:14 +00001036 case Token::Kind::TK_INLINE:
1037 this->nextToken();
1038 flags |= Modifiers::kInline_Flag;
1039 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001040 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001041 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001042 }
1043 }
1044}
1045
Ethan Nicholas11d53972016-11-28 11:23:23 -05001046Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
1047 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -07001048 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001049 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001050 }
1051 return result;
1052}
1053
1054/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001055ASTNode::ID Parser::statement() {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001056 Token start = this->nextToken();
1057 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001058 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001059 return ASTNode::ID::Invalid();
1060 }
1061 this->pushback(start);
ethannicholasb3058bd2016-07-01 08:22:01 -07001062 switch (start.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001063 case Token::Kind::TK_IF: // fall through
1064 case Token::Kind::TK_STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -07001065 return this->ifStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001066 case Token::Kind::TK_FOR:
ethannicholasb3058bd2016-07-01 08:22:01 -07001067 return this->forStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001068 case Token::Kind::TK_DO:
ethannicholasb3058bd2016-07-01 08:22:01 -07001069 return this->doStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001070 case Token::Kind::TK_WHILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001071 return this->whileStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001072 case Token::Kind::TK_SWITCH: // fall through
1073 case Token::Kind::TK_STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -05001074 return this->switchStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001075 case Token::Kind::TK_RETURN:
ethannicholasb3058bd2016-07-01 08:22:01 -07001076 return this->returnStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001077 case Token::Kind::TK_BREAK:
ethannicholasb3058bd2016-07-01 08:22:01 -07001078 return this->breakStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001079 case Token::Kind::TK_CONTINUE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001080 return this->continueStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001081 case Token::Kind::TK_DISCARD:
ethannicholasb3058bd2016-07-01 08:22:01 -07001082 return this->discardStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001083 case Token::Kind::TK_LBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001084 return this->block();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001085 case Token::Kind::TK_SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001086 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001087 RETURN_NODE(start.fOffset, ASTNode::Kind::kBlock);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001088 case Token::Kind::TK_CONST:
Ethan Nicholasfc994162019-06-06 10:04:27 -04001089 return this->varDeclarations();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001090 case Token::Kind::TK_IDENTIFIER:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001091 if (this->isType(this->text(start))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001092 return this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001093 }
John Stiles30212b72020-06-11 17:55:07 -04001094 [[fallthrough]];
ethannicholasb3058bd2016-07-01 08:22:01 -07001095 default:
1096 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001097 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001098}
1099
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001100/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* QUESTION? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001101ASTNode::ID Parser::type() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001102 Token type;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001103 if (!this->expect(Token::Kind::TK_IDENTIFIER, "a type", &type)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001104 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001105 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001106 if (!this->isType(this->text(type))) {
1107 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001108 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001109 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001110 CREATE_NODE(result, type.fOffset, ASTNode::Kind::kType);
1111 ASTNode::TypeData td(this->text(type), false, false);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001112 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1113 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001114 SKSL_INT i;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001115 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001116 CREATE_CHILD(child, result, -1, ASTNode::Kind::kInt, i);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001117 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001118 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001119 }
1120 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001121 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001122 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001123 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001124 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001125 td.fIsNullable = this->checkNext(Token::Kind::TK_QUESTION);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001126 getNode(result).setTypeData(td);
1127 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001128}
1129
John Stiles57a996b2020-04-19 19:05:12 -07001130/* IDENTIFIER LBRACE
1131 varDeclaration+
1132 RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001133ASTNode::ID Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001134 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001135 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001136 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001137 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001138 if (peek().fKind != Token::Kind::TK_LBRACE) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001139 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001140 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -07001141 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001142 this->error(name, "no type named '" + this->text(name) + "'");
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 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kInterfaceBlock);
1146 ASTNode::InterfaceBlockData id(mods, this->text(name), 0, "", 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07001147 this->nextToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001148 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001149 ASTNode::ID decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001150 if (!decl) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001151 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001152 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001153 getNode(result).addChild(decl);
1154 ++id.fDeclarationCount;
ethannicholasb3058bd2016-07-01 08:22:01 -07001155 }
John Stiles57a996b2020-04-19 19:05:12 -07001156 if (id.fDeclarationCount == 0) {
1157 this->error(name, "interface block '" + this->text(name) +
1158 "' must contain at least one member");
1159 return ASTNode::ID::Invalid();
1160 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001161 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001162 std::vector<ASTNode> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001163 StringFragment instanceName;
1164 Token instanceNameToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001165 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &instanceNameToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001166 id.fInstanceName = this->text(instanceNameToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001167 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1168 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001169 ASTNode::ID size = this->expression();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001170 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001171 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001172 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001173 getNode(result).addChild(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001174 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001175 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001176 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001177 ++id.fSizeCount;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001178 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001179 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001180 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -07001181 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001182 getNode(result).setInterfaceBlockData(id);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001183 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001184 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001185}
1186
1187/* IF LPAREN expression RPAREN statement (ELSE statement)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001188ASTNode::ID Parser::ifStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001189 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001190 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_IF, &start);
1191 if (!isStatic && !this->expect(Token::Kind::TK_IF, "'if'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001192 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001193 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001194 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kIf, isStatic);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001195 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001196 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001197 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001198 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001199 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001200 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001201 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001202 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001203 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001204 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001205 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001206 ASTNode::ID ifTrue = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001207 if (!ifTrue) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001208 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001209 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001210 getNode(result).addChild(ifTrue);
1211 ASTNode::ID ifFalse;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001212 if (this->checkNext(Token::Kind::TK_ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001213 ifFalse = this->statement();
1214 if (!ifFalse) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001215 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001216 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001217 getNode(result).addChild(ifFalse);
ethannicholasb3058bd2016-07-01 08:22:01 -07001218 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001219 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001220}
1221
1222/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001223ASTNode::ID Parser::doStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001224 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001225 if (!this->expect(Token::Kind::TK_DO, "'do'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001226 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001227 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001228 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kDo);
1229 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001230 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001231 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001232 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001233 getNode(result).addChild(statement);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001234 if (!this->expect(Token::Kind::TK_WHILE, "'while'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001235 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001236 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001237 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001238 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001239 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001240 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001241 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001242 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001243 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001244 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001245 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001246 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001247 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001248 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001249 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001250 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001251 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001252}
1253
1254/* WHILE LPAREN expression RPAREN STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001255ASTNode::ID Parser::whileStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001256 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001257 if (!this->expect(Token::Kind::TK_WHILE, "'while'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001258 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001259 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001260 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001261 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001262 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001263 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kWhile);
1264 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001265 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001266 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001267 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001268 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001269 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001270 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001271 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001272 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001273 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001274 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001275 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001276 getNode(result).addChild(statement);
1277 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001278}
1279
Ethan Nicholasaf197692017-02-27 13:26:45 -05001280/* CASE expression COLON statement* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001281ASTNode::ID Parser::switchCase() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001282 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001283 if (!this->expect(Token::Kind::TK_CASE, "'case'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001284 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001285 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001286 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitchCase);
1287 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001288 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001289 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001290 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001291 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001292 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001293 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001294 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001295 while (this->peek().fKind != Token::Kind::TK_RBRACE &&
1296 this->peek().fKind != Token::Kind::TK_CASE &&
1297 this->peek().fKind != Token::Kind::TK_DEFAULT) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001298 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001299 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001300 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001301 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001302 getNode(result).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001303 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001304 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001305}
1306
1307/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001308ASTNode::ID Parser::switchStatement() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001309 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001310 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_SWITCH, &start);
1311 if (!isStatic && !this->expect(Token::Kind::TK_SWITCH, "'switch'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001312 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001313 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001314 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001315 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001316 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001317 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001318 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001319 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001320 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001321 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001322 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001323 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001324 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001325 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001326 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001327 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitch, isStatic);
1328 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001329 while (this->peek().fKind == Token::Kind::TK_CASE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001330 ASTNode::ID c = this->switchCase();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001331 if (!c) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001332 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001333 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001334 getNode(result).addChild(c);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001335 }
1336 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1337 // parts of the compiler may rely upon this assumption.
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001338 if (this->peek().fKind == Token::Kind::TK_DEFAULT) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001339 Token defaultStart;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001340 SkAssertResult(this->expect(Token::Kind::TK_DEFAULT, "'default'", &defaultStart));
1341 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001342 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001343 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001344 CREATE_CHILD(defaultCase, result, defaultStart.fOffset, ASTNode::Kind::kSwitchCase);
1345 CREATE_EMPTY_CHILD(defaultCase); // empty test to signify default case
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001346 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001347 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001348 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001349 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001350 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001351 getNode(defaultCase).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001352 }
Ethan Nicholasaf197692017-02-27 13:26:45 -05001353 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001354 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001355 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001356 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001357 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001358}
1359
Ethan Nicholas11d53972016-11-28 11:23:23 -05001360/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001361 STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001362ASTNode::ID Parser::forStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001363 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001364 if (!this->expect(Token::Kind::TK_FOR, "'for'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001365 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001366 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001367 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001368 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001369 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001370 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kFor);
1371 ASTNode::ID initializer;
ethannicholasb3058bd2016-07-01 08:22:01 -07001372 Token nextToken = this->peek();
1373 switch (nextToken.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001374 case Token::Kind::TK_SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001375 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001376 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001377 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001378 case Token::Kind::TK_CONST: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001379 initializer = this->varDeclarations();
1380 if (!initializer) {
1381 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001382 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001383 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001384 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001385 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001386 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001387 if (this->isType(this->text(nextToken))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001388 initializer = this->varDeclarations();
1389 if (!initializer) {
1390 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001391 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001392 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001393 break;
1394 }
John Stiles30212b72020-06-11 17:55:07 -04001395 [[fallthrough]];
1396 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001397 default:
1398 initializer = this->expressionStatement();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001399 if (!initializer) {
1400 return ASTNode::ID::Invalid();
1401 }
1402 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001403 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001404 ASTNode::ID test;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001405 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001406 test = this->expression();
1407 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001408 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001409 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001410 getNode(result).addChild(test);
1411 } else {
1412 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001413 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001414 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001415 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001416 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001417 ASTNode::ID next;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001418 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001419 next = this->expression();
1420 if (!next) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001421 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001422 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001423 getNode(result).addChild(next);
1424 } else {
1425 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001426 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001427 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001428 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001429 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001430 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001431 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001432 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001433 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001434 getNode(result).addChild(statement);
1435 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001436}
1437
1438/* RETURN expression? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001439ASTNode::ID Parser::returnStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001440 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001441 if (!this->expect(Token::Kind::TK_RETURN, "'return'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001442 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001443 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001444 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kReturn);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001445 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001446 ASTNode::ID expression = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001447 if (!expression) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001448 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001449 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001450 getNode(result).addChild(expression);
ethannicholasb3058bd2016-07-01 08:22:01 -07001451 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001452 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001453 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001454 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001455 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001456}
1457
1458/* BREAK SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001459ASTNode::ID Parser::breakStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001460 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001461 if (!this->expect(Token::Kind::TK_BREAK, "'break'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001462 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001463 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001464 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001465 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001466 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001467 RETURN_NODE(start.fOffset, ASTNode::Kind::kBreak);
ethannicholasb3058bd2016-07-01 08:22:01 -07001468}
1469
1470/* CONTINUE SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001471ASTNode::ID Parser::continueStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001472 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001473 if (!this->expect(Token::Kind::TK_CONTINUE, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001474 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001475 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001476 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001477 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001478 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001479 RETURN_NODE(start.fOffset, ASTNode::Kind::kContinue);
ethannicholasb3058bd2016-07-01 08:22:01 -07001480}
1481
1482/* DISCARD SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001483ASTNode::ID Parser::discardStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001484 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001485 if (!this->expect(Token::Kind::TK_DISCARD, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001486 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001487 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001488 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001489 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001490 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001491 RETURN_NODE(start.fOffset, ASTNode::Kind::kDiscard);
ethannicholasb3058bd2016-07-01 08:22:01 -07001492}
1493
1494/* LBRACE statement* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001495ASTNode::ID Parser::block() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001496 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001497 if (!this->expect(Token::Kind::TK_LBRACE, "'{'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001498 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001499 }
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001500 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001501 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001502 return ASTNode::ID::Invalid();
1503 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001504 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -07001505 for (;;) {
1506 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001507 case Token::Kind::TK_RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001508 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001509 return result;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001510 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001511 this->error(this->peek(), "expected '}', but found end of file");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001512 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001513 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001514 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001515 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001516 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001517 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001518 getNode(result).addChild(statement);
ethannicholasb3058bd2016-07-01 08:22:01 -07001519 }
1520 }
1521 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001522 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001523}
1524
1525/* expression SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001526ASTNode::ID Parser::expressionStatement() {
1527 ASTNode::ID expr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001528 if (expr) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001529 if (this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001530 return expr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001531 }
1532 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001533 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001534}
1535
1536/* assignmentExpression (COMMA assignmentExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001537ASTNode::ID Parser::expression() {
1538 ASTNode::ID result = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001539 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001540 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001541 }
1542 Token t;
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001543 AutoDepth depth(this);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001544 while (this->checkNext(Token::Kind::TK_COMMA, &t)) {
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001545 if (!depth.increase()) {
1546 return ASTNode::ID::Invalid();
1547 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001548 ASTNode::ID right = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001549 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001550 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001551 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001552 CREATE_NODE(newResult, t.fOffset, ASTNode::Kind::kBinary, std::move(t));
1553 getNode(newResult).addChild(result);
1554 getNode(newResult).addChild(right);
1555 result = newResult;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001556 }
1557 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001558}
1559
1560/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1561 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1562 assignmentExpression)*
1563 */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001564ASTNode::ID Parser::assignmentExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001565 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001566 ASTNode::ID result = this->ternaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001567 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001568 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001569 }
1570 for (;;) {
1571 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001572 case Token::Kind::TK_EQ: // fall through
1573 case Token::Kind::TK_STAREQ: // fall through
1574 case Token::Kind::TK_SLASHEQ: // fall through
1575 case Token::Kind::TK_PERCENTEQ: // fall through
1576 case Token::Kind::TK_PLUSEQ: // fall through
1577 case Token::Kind::TK_MINUSEQ: // fall through
1578 case Token::Kind::TK_SHLEQ: // fall through
1579 case Token::Kind::TK_SHREQ: // fall through
1580 case Token::Kind::TK_BITWISEANDEQ: // fall through
1581 case Token::Kind::TK_BITWISEXOREQ: // fall through
1582 case Token::Kind::TK_BITWISEOREQ: // fall through
1583 case Token::Kind::TK_LOGICALANDEQ: // fall through
1584 case Token::Kind::TK_LOGICALXOREQ: // fall through
1585 case Token::Kind::TK_LOGICALOREQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001586 if (!depth.increase()) {
1587 return ASTNode::ID::Invalid();
1588 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001589 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001590 ASTNode::ID right = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001591 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001592 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001593 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001594 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1595 std::move(t));
1596 getNode(newResult).addChild(result);
1597 getNode(newResult).addChild(right);
1598 result = newResult;
1599 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001600 }
1601 default:
1602 return result;
1603 }
1604 }
1605}
1606
1607/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001608ASTNode::ID Parser::ternaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001609 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001610 ASTNode::ID base = this->logicalOrExpression();
1611 if (!base) {
1612 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001613 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001614 if (this->checkNext(Token::Kind::TK_QUESTION)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001615 if (!depth.increase()) {
1616 return ASTNode::ID::Invalid();
1617 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001618 ASTNode::ID trueExpr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001619 if (!trueExpr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001620 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001621 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001622 if (this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001623 ASTNode::ID falseExpr = this->assignmentExpression();
1624 if (!falseExpr) {
1625 return ASTNode::ID::Invalid();
1626 }
1627 CREATE_NODE(ternary, getNode(base).fOffset, ASTNode::Kind::kTernary);
1628 getNode(ternary).addChild(base);
1629 getNode(ternary).addChild(trueExpr);
1630 getNode(ternary).addChild(falseExpr);
1631 return ternary;
ethannicholasb3058bd2016-07-01 08:22:01 -07001632 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001633 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001634 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001635 return base;
ethannicholasb3058bd2016-07-01 08:22:01 -07001636}
1637
1638/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001639ASTNode::ID Parser::logicalOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001640 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001641 ASTNode::ID result = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001642 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001643 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001644 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001645 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001646 while (this->checkNext(Token::Kind::TK_LOGICALOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001647 if (!depth.increase()) {
1648 return ASTNode::ID::Invalid();
1649 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001650 ASTNode::ID right = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001651 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001652 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001653 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001654 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1655 getNode(newResult).addChild(result);
1656 getNode(newResult).addChild(right);
1657 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001658 }
1659 return result;
1660}
1661
1662/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001663ASTNode::ID Parser::logicalXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001664 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001665 ASTNode::ID result = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001666 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001667 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001668 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001669 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001670 while (this->checkNext(Token::Kind::TK_LOGICALXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001671 if (!depth.increase()) {
1672 return ASTNode::ID::Invalid();
1673 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001674 ASTNode::ID right = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001675 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001676 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001677 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001678 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1679 getNode(newResult).addChild(result);
1680 getNode(newResult).addChild(right);
1681 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001682 }
1683 return result;
1684}
1685
1686/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001687ASTNode::ID Parser::logicalAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001688 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001689 ASTNode::ID result = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001690 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001691 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001692 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001693 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001694 while (this->checkNext(Token::Kind::TK_LOGICALAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001695 if (!depth.increase()) {
1696 return ASTNode::ID::Invalid();
1697 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001698 ASTNode::ID right = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001699 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001700 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001701 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001702 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1703 getNode(newResult).addChild(result);
1704 getNode(newResult).addChild(right);
1705 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001706 }
1707 return result;
1708}
1709
1710/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001711ASTNode::ID Parser::bitwiseOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001712 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001713 ASTNode::ID result = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001714 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001715 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001716 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001717 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001718 while (this->checkNext(Token::Kind::TK_BITWISEOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001719 if (!depth.increase()) {
1720 return ASTNode::ID::Invalid();
1721 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001722 ASTNode::ID right = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001723 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001724 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001725 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001726 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1727 getNode(newResult).addChild(result);
1728 getNode(newResult).addChild(right);
1729 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001730 }
1731 return result;
1732}
1733
1734/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001735ASTNode::ID Parser::bitwiseXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001736 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001737 ASTNode::ID result = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001738 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001739 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001740 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001741 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001742 while (this->checkNext(Token::Kind::TK_BITWISEXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001743 if (!depth.increase()) {
1744 return ASTNode::ID::Invalid();
1745 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001746 ASTNode::ID right = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001747 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001748 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001749 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001750 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1751 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 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001774 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1775 getNode(newResult).addChild(result);
1776 getNode(newResult).addChild(right);
1777 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001778 }
1779 return result;
1780}
1781
1782/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001783ASTNode::ID Parser::equalityExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001784 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001785 ASTNode::ID result = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001786 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001787 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001788 }
1789 for (;;) {
1790 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001791 case Token::Kind::TK_EQEQ: // fall through
1792 case Token::Kind::TK_NEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001793 if (!depth.increase()) {
1794 return ASTNode::ID::Invalid();
1795 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001796 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001797 ASTNode::ID right = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001798 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001799 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001800 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001801 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1802 std::move(t));
1803 getNode(newResult).addChild(result);
1804 getNode(newResult).addChild(right);
1805 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001806 break;
1807 }
1808 default:
1809 return result;
1810 }
1811 }
1812}
1813
1814/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001815ASTNode::ID Parser::relationalExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001816 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001817 ASTNode::ID result = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001818 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001819 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001820 }
1821 for (;;) {
1822 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001823 case Token::Kind::TK_LT: // fall through
1824 case Token::Kind::TK_GT: // fall through
1825 case Token::Kind::TK_LTEQ: // fall through
1826 case Token::Kind::TK_GTEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001827 if (!depth.increase()) {
1828 return ASTNode::ID::Invalid();
1829 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001830 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001831 ASTNode::ID right = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001832 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001833 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001834 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001835 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1836 std::move(t));
1837 getNode(newResult).addChild(result);
1838 getNode(newResult).addChild(right);
1839 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001840 break;
1841 }
1842 default:
1843 return result;
1844 }
1845 }
1846}
1847
1848/* additiveExpression ((SHL | SHR) additiveExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001849ASTNode::ID Parser::shiftExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001850 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001851 ASTNode::ID result = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001852 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001853 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001854 }
1855 for (;;) {
1856 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001857 case Token::Kind::TK_SHL: // fall through
1858 case Token::Kind::TK_SHR: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001859 if (!depth.increase()) {
1860 return ASTNode::ID::Invalid();
1861 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001862 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001863 ASTNode::ID right = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001864 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001865 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001866 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001867 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1868 std::move(t));
1869 getNode(newResult).addChild(result);
1870 getNode(newResult).addChild(right);
1871 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001872 break;
1873 }
1874 default:
1875 return result;
1876 }
1877 }
1878}
1879
1880/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001881ASTNode::ID Parser::additiveExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001882 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001883 ASTNode::ID result = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001884 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001885 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001886 }
1887 for (;;) {
1888 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001889 case Token::Kind::TK_PLUS: // fall through
1890 case Token::Kind::TK_MINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001891 if (!depth.increase()) {
1892 return ASTNode::ID::Invalid();
1893 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001894 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001895 ASTNode::ID right = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001896 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001897 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001898 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001899 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1900 std::move(t));
1901 getNode(newResult).addChild(result);
1902 getNode(newResult).addChild(right);
1903 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001904 break;
1905 }
1906 default:
1907 return result;
1908 }
1909 }
1910}
1911
1912/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001913ASTNode::ID Parser::multiplicativeExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001914 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001915 ASTNode::ID result = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001916 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001917 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001918 }
1919 for (;;) {
1920 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001921 case Token::Kind::TK_STAR: // fall through
1922 case Token::Kind::TK_SLASH: // fall through
1923 case Token::Kind::TK_PERCENT: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001924 if (!depth.increase()) {
1925 return ASTNode::ID::Invalid();
1926 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001927 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001928 ASTNode::ID right = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001929 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001930 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001931 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001932 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1933 std::move(t));
1934 getNode(newResult).addChild(result);
1935 getNode(newResult).addChild(right);
1936 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001937 break;
1938 }
1939 default:
1940 return result;
1941 }
1942 }
1943}
1944
1945/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001946ASTNode::ID Parser::unaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001947 AutoDepth depth(this);
ethannicholasb3058bd2016-07-01 08:22:01 -07001948 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001949 case Token::Kind::TK_PLUS: // fall through
1950 case Token::Kind::TK_MINUS: // fall through
1951 case Token::Kind::TK_LOGICALNOT: // fall through
1952 case Token::Kind::TK_BITWISENOT: // fall through
1953 case Token::Kind::TK_PLUSPLUS: // fall through
1954 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001955 if (!depth.increase()) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001956 return ASTNode::ID::Invalid();
Ethan Nicholas6dcc3252019-02-20 15:18:36 -05001957 }
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001958 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001959 ASTNode::ID expr = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001960 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001961 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001962 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001963 CREATE_NODE(result, t.fOffset, ASTNode::Kind::kPrefix, std::move(t));
1964 getNode(result).addChild(expr);
1965 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001966 }
1967 default:
1968 return this->postfixExpression();
1969 }
1970}
1971
1972/* term suffix* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001973ASTNode::ID Parser::postfixExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001974 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001975 ASTNode::ID result = this->term();
ethannicholasb3058bd2016-07-01 08:22:01 -07001976 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001977 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001978 }
1979 for (;;) {
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001980 Token t = this->peek();
1981 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001982 case Token::Kind::TK_FLOAT_LITERAL:
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001983 if (this->text(t)[0] != '.') {
1984 return result;
1985 }
John Stiles30212b72020-06-11 17:55:07 -04001986 [[fallthrough]];
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001987 case Token::Kind::TK_LBRACKET:
1988 case Token::Kind::TK_DOT:
1989 case Token::Kind::TK_LPAREN:
1990 case Token::Kind::TK_PLUSPLUS:
1991 case Token::Kind::TK_MINUSMINUS:
1992 case Token::Kind::TK_COLONCOLON:
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001993 if (!depth.increase()) {
1994 return ASTNode::ID::Invalid();
1995 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001996 result = this->suffix(result);
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001997 if (!result) {
1998 return ASTNode::ID::Invalid();
1999 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002000 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002001 default:
2002 return result;
2003 }
2004 }
2005}
2006
Ethan Nicholas11d53972016-11-28 11:23:23 -05002007/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
Ethan Nicholase455f652019-09-13 12:52:55 -04002008 PLUSPLUS | MINUSMINUS | COLONCOLON IDENTIFIER | FLOAT_LITERAL [IDENTIFIER] */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002009ASTNode::ID Parser::suffix(ASTNode::ID base) {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002010 SkASSERT(base);
ethannicholasb3058bd2016-07-01 08:22:01 -07002011 Token next = this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002012 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002013 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002014 return ASTNode::ID::Invalid();
2015 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002016 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002017 case Token::Kind::TK_LBRACKET: {
2018 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002019 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
2020 getNode(result).addChild(base);
2021 return result;
ethannicholas5961bc92016-10-12 06:39:56 -07002022 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002023 ASTNode::ID e = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002024 if (!e) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002025 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002026 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002027 this->expect(Token::Kind::TK_RBRACKET, "']' to complete array access expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002028 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
2029 getNode(result).addChild(base);
2030 getNode(result).addChild(e);
2031 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002032 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002033 case Token::Kind::TK_COLONCOLON: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002034 int offset = this->peek().fOffset;
2035 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002036 if (this->identifier(&text)) {
Brian Osman6518d772020-09-10 16:50:06 -04002037 CREATE_NODE(result, offset, ASTNode::Kind::kScope, std::move(text));
2038 getNode(result).addChild(base);
2039 return result;
2040 }
2041 return ASTNode::ID::Invalid();
2042 }
2043 case Token::Kind::TK_DOT: {
2044 int offset = this->peek().fOffset;
2045 StringFragment text;
2046 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002047 CREATE_NODE(result, offset, ASTNode::Kind::kField, std::move(text));
2048 getNode(result).addChild(base);
2049 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002050 }
Brian Osman6518d772020-09-10 16:50:06 -04002051 [[fallthrough]];
Ethan Nicholase455f652019-09-13 12:52:55 -04002052 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002053 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholase455f652019-09-13 12:52:55 -04002054 // Swizzles that start with a constant number, e.g. '.000r', will be tokenized as
2055 // floating point literals, possibly followed by an identifier. Handle that here.
2056 StringFragment field = this->text(next);
2057 SkASSERT(field.fChars[0] == '.');
2058 ++field.fChars;
2059 --field.fLength;
2060 for (size_t i = 0; i < field.fLength; ++i) {
2061 if (field.fChars[i] != '0' && field.fChars[i] != '1') {
2062 this->error(next, "invalid swizzle");
2063 return ASTNode::ID::Invalid();
2064 }
2065 }
2066 // use the next *raw* token so we don't ignore whitespace - we only care about
2067 // identifiers that directly follow the float
2068 Token id = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002069 if (id.fKind == Token::Kind::TK_IDENTIFIER) {
Ethan Nicholase455f652019-09-13 12:52:55 -04002070 field.fLength += id.fLength;
2071 } else {
2072 this->pushback(id);
2073 }
2074 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kField, field);
2075 getNode(result).addChild(base);
2076 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002077 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002078 case Token::Kind::TK_LPAREN: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002079 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kCall);
2080 getNode(result).addChild(base);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002081 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002082 for (;;) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002083 ASTNode::ID expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002084 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002085 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002086 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002087 getNode(result).addChild(expr);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002088 if (!this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002089 break;
2090 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002091 }
2092 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002093 this->expect(Token::Kind::TK_RPAREN, "')' to complete function parameters");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002094 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002095 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002096 case Token::Kind::TK_PLUSPLUS: // fall through
2097 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002098 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kPostfix, next);
2099 getNode(result).addChild(base);
2100 return result;
2101 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002102 default: {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002103 this->error(next, "expected expression suffix, but found '" + this->text(next) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002104 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002105 }
2106 }
2107}
2108
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002109/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | NULL_LITERAL | '(' expression ')' */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002110ASTNode::ID Parser::term() {
ethannicholasb3058bd2016-07-01 08:22:01 -07002111 Token t = this->peek();
2112 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002113 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002114 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002115 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002116 RETURN_NODE(t.fOffset, ASTNode::Kind::kIdentifier, std::move(text));
ethannicholasb3058bd2016-07-01 08:22:01 -07002117 }
John Stiles30212b72020-06-11 17:55:07 -04002118 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07002119 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002120 case Token::Kind::TK_INT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002121 SKSL_INT i;
ethannicholasb3058bd2016-07-01 08:22:01 -07002122 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002123 RETURN_NODE(t.fOffset, ASTNode::Kind::kInt, i);
ethannicholasb3058bd2016-07-01 08:22:01 -07002124 }
2125 break;
2126 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002127 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002128 SKSL_FLOAT f;
ethannicholasb3058bd2016-07-01 08:22:01 -07002129 if (this->floatLiteral(&f)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002130 RETURN_NODE(t.fOffset, ASTNode::Kind::kFloat, f);
ethannicholasb3058bd2016-07-01 08:22:01 -07002131 }
2132 break;
2133 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002134 case Token::Kind::TK_TRUE_LITERAL: // fall through
2135 case Token::Kind::TK_FALSE_LITERAL: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002136 bool b;
2137 if (this->boolLiteral(&b)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002138 RETURN_NODE(t.fOffset, ASTNode::Kind::kBool, b);
ethannicholasb3058bd2016-07-01 08:22:01 -07002139 }
2140 break;
2141 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002142 case Token::Kind::TK_NULL_LITERAL:
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002143 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04002144 RETURN_NODE(t.fOffset, ASTNode::Kind::kNull);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002145 case Token::Kind::TK_LPAREN: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002146 this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002147 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002148 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002149 return ASTNode::ID::Invalid();
2150 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002151 ASTNode::ID result = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002152 if (result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002153 this->expect(Token::Kind::TK_RPAREN, "')' to complete expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002154 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002155 }
2156 break;
2157 }
2158 default:
2159 this->nextToken();
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002160 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002161 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002162 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002163}
2164
2165/* INT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002166bool Parser::intLiteral(SKSL_INT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002167 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002168 if (this->expect(Token::Kind::TK_INT_LITERAL, "integer literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002169 *dest = SkSL::stol(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002170 return true;
2171 }
2172 return false;
2173}
2174
2175/* FLOAT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002176bool Parser::floatLiteral(SKSL_FLOAT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002177 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002178 if (this->expect(Token::Kind::TK_FLOAT_LITERAL, "float literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002179 *dest = SkSL::stod(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002180 return true;
2181 }
2182 return false;
2183}
2184
2185/* TRUE_LITERAL | FALSE_LITERAL */
2186bool Parser::boolLiteral(bool* dest) {
2187 Token t = this->nextToken();
2188 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002189 case Token::Kind::TK_TRUE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002190 *dest = true;
2191 return true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002192 case Token::Kind::TK_FALSE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002193 *dest = false;
2194 return true;
2195 default:
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002196 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002197 return false;
2198 }
2199}
2200
2201/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002202bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002203 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002204 if (this->expect(Token::Kind::TK_IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002205 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07002206 return true;
2207 }
2208 return false;
2209}
2210
John Stilesa6841be2020-08-06 14:11:56 -04002211} // namespace SkSL