blob: 00378708e240d654aecf8c4a54fdf956c1ca7e75 [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
ethannicholasb3058bd2016-07-01 08:22:01 -07008#include "stdio.h"
Ethan Nicholasfc994162019-06-06 10:04:27 -04009#include "src/sksl/SkSLASTNode.h"
Mike Klein4b432fa2019-06-06 11:44:05 -050010#include "src/sksl/SkSLParser.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050011#include "src/sksl/ir/SkSLModifiers.h"
12#include "src/sksl/ir/SkSLSymbolTable.h"
13#include "src/sksl/ir/SkSLType.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070014
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040015#ifndef SKSL_STANDALONE
Mike Kleinc0bd9f92019-04-23 12:05:21 -050016#include "include/private/SkOnce.h"
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040017#endif
18
ethannicholasb3058bd2016-07-01 08:22:01 -070019namespace SkSL {
20
ethannicholascad64162016-10-27 10:54:02 -070021#define MAX_PARSE_DEPTH 50
22
23class AutoDepth {
24public:
25 AutoDepth(Parser* p)
Ethan Nicholascf4deab2019-09-13 16:28:14 -040026 : fParser(p)
27 , fDepth(0) {}
ethannicholascad64162016-10-27 10:54:02 -070028
29 ~AutoDepth() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -040030 fParser->fDepth -= fDepth;
ethannicholascad64162016-10-27 10:54:02 -070031 }
32
Ethan Nicholascf4deab2019-09-13 16:28:14 -040033 bool increase() {
34 ++fDepth;
35 ++fParser->fDepth;
ethannicholascad64162016-10-27 10:54:02 -070036 if (fParser->fDepth > MAX_PARSE_DEPTH) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -070037 fParser->error(fParser->peek(), String("exceeded max parse depth"));
ethannicholascad64162016-10-27 10:54:02 -070038 return false;
39 }
40 return true;
41 }
42
43private:
44 Parser* fParser;
Ethan Nicholascf4deab2019-09-13 16:28:14 -040045 int fDepth;
ethannicholascad64162016-10-27 10:54:02 -070046};
47
Brian Salomon140f3da2018-08-23 13:51:27 +000048std::unordered_map<String, Parser::LayoutToken>* Parser::layoutTokens;
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040049
50void Parser::InitLayoutMap() {
Brian Salomon140f3da2018-08-23 13:51:27 +000051 layoutTokens = new std::unordered_map<String, LayoutToken>;
Brian Salomon23356442018-11-30 15:33:19 -050052 #define TOKEN(name, text) (*layoutTokens)[text] = LayoutToken::name
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040053 TOKEN(LOCATION, "location");
54 TOKEN(OFFSET, "offset");
55 TOKEN(BINDING, "binding");
56 TOKEN(INDEX, "index");
57 TOKEN(SET, "set");
58 TOKEN(BUILTIN, "builtin");
59 TOKEN(INPUT_ATTACHMENT_INDEX, "input_attachment_index");
60 TOKEN(ORIGIN_UPPER_LEFT, "origin_upper_left");
61 TOKEN(OVERRIDE_COVERAGE, "override_coverage");
62 TOKEN(BLEND_SUPPORT_ALL_EQUATIONS, "blend_support_all_equations");
63 TOKEN(BLEND_SUPPORT_MULTIPLY, "blend_support_multiply");
64 TOKEN(BLEND_SUPPORT_SCREEN, "blend_support_screen");
65 TOKEN(BLEND_SUPPORT_OVERLAY, "blend_support_overlay");
66 TOKEN(BLEND_SUPPORT_DARKEN, "blend_support_darken");
67 TOKEN(BLEND_SUPPORT_LIGHTEN, "blend_support_lighten");
68 TOKEN(BLEND_SUPPORT_COLORDODGE, "blend_support_colordodge");
69 TOKEN(BLEND_SUPPORT_COLORBURN, "blend_support_colorburn");
70 TOKEN(BLEND_SUPPORT_HARDLIGHT, "blend_support_hardlight");
71 TOKEN(BLEND_SUPPORT_SOFTLIGHT, "blend_support_softlight");
72 TOKEN(BLEND_SUPPORT_DIFFERENCE, "blend_support_difference");
73 TOKEN(BLEND_SUPPORT_EXCLUSION, "blend_support_exclusion");
74 TOKEN(BLEND_SUPPORT_HSL_HUE, "blend_support_hsl_hue");
75 TOKEN(BLEND_SUPPORT_HSL_SATURATION, "blend_support_hsl_saturation");
76 TOKEN(BLEND_SUPPORT_HSL_COLOR, "blend_support_hsl_color");
77 TOKEN(BLEND_SUPPORT_HSL_LUMINOSITY, "blend_support_hsl_luminosity");
78 TOKEN(PUSH_CONSTANT, "push_constant");
79 TOKEN(POINTS, "points");
80 TOKEN(LINES, "lines");
81 TOKEN(LINE_STRIP, "line_strip");
82 TOKEN(LINES_ADJACENCY, "lines_adjacency");
83 TOKEN(TRIANGLES, "triangles");
84 TOKEN(TRIANGLE_STRIP, "triangle_strip");
85 TOKEN(TRIANGLES_ADJACENCY, "triangles_adjacency");
86 TOKEN(MAX_VERTICES, "max_vertices");
87 TOKEN(INVOCATIONS, "invocations");
Brian Osmanf59a9612020-04-15 14:18:13 -040088 TOKEN(MARKER, "marker");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -040089 TOKEN(WHEN, "when");
90 TOKEN(KEY, "key");
Michael Ludwiga4275592018-08-31 10:52:47 -040091 TOKEN(TRACKED, "tracked");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040092 TOKEN(CTYPE, "ctype");
Brian Osmanf28e55d2018-10-03 16:35:54 -040093 TOKEN(SKPMCOLOR4F, "SkPMColor4f");
Mike Reedb26b4e72020-01-22 14:31:21 -050094 TOKEN(SKV4, "SkV4");
Ethan Nicholas78aceb22018-08-31 16:13:58 -040095 TOKEN(SKRECT, "SkRect");
96 TOKEN(SKIRECT, "SkIRect");
97 TOKEN(SKPMCOLOR, "SkPMColor");
Mike Reedb26b4e72020-01-22 14:31:21 -050098 TOKEN(SKM44, "SkM44");
Ethan Nicholasc1c686b2019-04-02 17:30:23 -040099 TOKEN(BOOL, "bool");
100 TOKEN(INT, "int");
101 TOKEN(FLOAT, "float");
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400102 #undef TOKEN
103}
104
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700105Parser::Parser(const char* text, size_t length, SymbolTable& types, ErrorReporter& errors)
106: fText(text)
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400107, fPushback(Token::Kind::TK_INVALID, -1, -1)
ethannicholasb3058bd2016-07-01 08:22:01 -0700108, fTypes(types)
109, fErrors(errors) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700110 fLexer.start(text, length);
Brian Salomon3b83afe2018-08-23 11:04:36 -0400111 static const bool layoutMapInitialized = []{ return (void)InitLayoutMap(), true; }();
112 (void) layoutMapInitialized;
ethannicholasb3058bd2016-07-01 08:22:01 -0700113}
114
Ethan Nicholasfc994162019-06-06 10:04:27 -0400115#define CREATE_NODE(result, ...) \
116 ASTNode::ID result(fFile->fNodes.size()); \
117 fFile->fNodes.emplace_back(&fFile->fNodes, __VA_ARGS__)
118
119#define RETURN_NODE(...) \
120 do { \
121 CREATE_NODE(result, __VA_ARGS__); \
122 return result; \
123 } while (false)
124
125#define CREATE_CHILD(child, target, ...) \
126 CREATE_NODE(child, __VA_ARGS__); \
127 fFile->fNodes[target.fValue].addChild(child)
128
129#define CREATE_EMPTY_CHILD(target) \
130 do { \
131 ASTNode::ID child(fFile->fNodes.size()); \
132 fFile->fNodes.emplace_back(); \
133 fFile->fNodes[target.fValue].addChild(child); \
134 } while (false)
135
Ethan Nicholasf7b88202017-09-18 14:10:39 -0400136/* (directive | section | declaration)* END_OF_FILE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400137std::unique_ptr<ASTFile> Parser::file() {
138 fFile.reset(new ASTFile());
139 CREATE_NODE(result, 0, ASTNode::Kind::kFile);
140 fFile->fRoot = result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700141 for (;;) {
142 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400143 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholasfc994162019-06-06 10:04:27 -0400144 return std::move(fFile);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400145 case Token::Kind::TK_DIRECTIVE: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400146 ASTNode::ID dir = this->directive();
147 if (fErrors.errorCount()) {
148 return nullptr;
149 }
150 if (dir) {
151 getNode(result).addChild(dir);
ethannicholasb3058bd2016-07-01 08:22:01 -0700152 }
153 break;
154 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400155 case Token::Kind::TK_SECTION: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400156 ASTNode::ID section = this->section();
157 if (fErrors.errorCount()) {
158 return nullptr;
159 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400160 if (section) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400161 getNode(result).addChild(section);
Ethan Nicholas762466e2017-06-29 10:03:38 -0400162 }
163 break;
164 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700165 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400166 ASTNode::ID decl = this->declaration();
167 if (fErrors.errorCount()) {
168 return nullptr;
ethannicholasb3058bd2016-07-01 08:22:01 -0700169 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400170 if (decl) {
171 getNode(result).addChild(decl);
172 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700173 }
174 }
175 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400176 return std::move(fFile);
ethannicholasb3058bd2016-07-01 08:22:01 -0700177}
178
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700179Token Parser::nextRawToken() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400180 if (fPushback.fKind != Token::Kind::TK_INVALID) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700181 Token result = fPushback;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400182 fPushback.fKind = Token::Kind::TK_INVALID;
ethannicholasb3058bd2016-07-01 08:22:01 -0700183 return result;
184 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700185 Token result = fLexer.next();
186 return result;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400187}
188
189Token Parser::nextToken() {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700190 Token token = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400191 while (token.fKind == Token::Kind::TK_WHITESPACE ||
192 token.fKind == Token::Kind::TK_LINE_COMMENT ||
193 token.fKind == Token::Kind::TK_BLOCK_COMMENT) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700194 token = this->nextRawToken();
195 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400196 return token;
ethannicholasb3058bd2016-07-01 08:22:01 -0700197}
198
199void Parser::pushback(Token t) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400200 SkASSERT(fPushback.fKind == Token::Kind::TK_INVALID);
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400201 fPushback = std::move(t);
ethannicholasb3058bd2016-07-01 08:22:01 -0700202}
203
204Token Parser::peek() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400205 if (fPushback.fKind == Token::Kind::TK_INVALID) {
Brian Osman634624a2017-08-15 11:14:30 -0400206 fPushback = this->nextToken();
207 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700208 return fPushback;
209}
210
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400211bool Parser::checkNext(Token::Kind kind, Token* result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400212 if (fPushback.fKind != Token::Kind::TK_INVALID && fPushback.fKind != kind) {
Brian Osman634624a2017-08-15 11:14:30 -0400213 return false;
214 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400215 Token next = this->nextToken();
216 if (next.fKind == kind) {
217 if (result) {
218 *result = next;
219 }
220 return true;
221 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400222 this->pushback(std::move(next));
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400223 return false;
224}
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500225
226bool Parser::expect(Token::Kind kind, const char* expected, Token* result) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700227 Token next = this->nextToken();
228 if (next.fKind == kind) {
229 if (result) {
Brian Osman634624a2017-08-15 11:14:30 -0400230 *result = std::move(next);
ethannicholasb3058bd2016-07-01 08:22:01 -0700231 }
232 return true;
233 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700234 this->error(next, "expected " + String(expected) + ", but found '" +
235 this->text(next) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -0700236 return false;
237 }
238}
239
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700240StringFragment Parser::text(Token token) {
241 return StringFragment(fText + token.fOffset, token.fLength);
Ethan Nicholas9e1138d2016-11-21 10:39:35 -0500242}
243
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700244void Parser::error(Token token, String msg) {
245 this->error(token.fOffset, msg);
ethannicholasb3058bd2016-07-01 08:22:01 -0700246}
247
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700248void Parser::error(int offset, String msg) {
249 fErrors.error(offset, msg);
250}
251
252bool Parser::isType(StringFragment name) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700253 return nullptr != fTypes[name];
254}
255
Ethan Nicholas11d53972016-11-28 11:23:23 -0500256/* DIRECTIVE(#version) INT_LITERAL ("es" | "compatibility")? |
ethannicholas5961bc92016-10-12 06:39:56 -0700257 DIRECTIVE(#extension) IDENTIFIER COLON IDENTIFIER */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400258ASTNode::ID Parser::directive() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700259 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400260 if (!this->expect(Token::Kind::TK_DIRECTIVE, "a directive", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400261 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700262 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700263 StringFragment text = this->text(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400264 if (text == "#extension") {
ethannicholasb3058bd2016-07-01 08:22:01 -0700265 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400266 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400267 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700268 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400269 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400270 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700271 }
272 // FIXME: need to start paying attention to this token
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400273 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400274 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700275 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400276 RETURN_NODE(start.fOffset, ASTNode::Kind::kExtension, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700277 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700278 this->error(start, "unsupported directive '" + this->text(start) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400279 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700280 }
281}
282
Ethan Nicholas762466e2017-06-29 10:03:38 -0400283/* SECTION LBRACE (LPAREN IDENTIFIER RPAREN)? <any sequence of tokens with balanced braces>
284 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400285ASTNode::ID Parser::section() {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400286 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400287 if (!this->expect(Token::Kind::TK_SECTION, "a section token", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400288 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400289 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400290 StringFragment argument;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400291 if (this->peek().fKind == Token::Kind::TK_LPAREN) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400292 this->nextToken();
293 Token argToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400294 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &argToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400295 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400296 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700297 argument = this->text(argToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400298 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400299 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400300 }
301 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400302 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400303 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400304 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400305 StringFragment text;
306 Token codeStart = this->nextRawToken();
307 size_t startOffset = codeStart.fOffset;
308 this->pushback(codeStart);
309 text.fChars = fText + startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400310 int level = 1;
311 for (;;) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700312 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400313 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400314 case Token::Kind::TK_LBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400315 ++level;
316 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400317 case Token::Kind::TK_RBRACE:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400318 --level;
319 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400320 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700321 this->error(start, "reached end of file while parsing section");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400322 return ASTNode::ID::Invalid();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400323 default:
324 break;
325 }
326 if (!level) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400327 text.fLength = next.fOffset - startOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400328 break;
329 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400330 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700331 StringFragment name = this->text(start);
332 ++name.fChars;
333 --name.fLength;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400334 RETURN_NODE(start.fOffset, ASTNode::Kind::kSection,
335 ASTNode::SectionData(name, argument, text));
Ethan Nicholas762466e2017-06-29 10:03:38 -0400336}
337
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500338/* ENUM CLASS IDENTIFIER LBRACE (IDENTIFIER (EQ expression)? (COMMA IDENTIFIER (EQ expression))*)?
339 RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400340ASTNode::ID Parser::enumDeclaration() {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500341 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400342 if (!this->expect(Token::Kind::TK_ENUM, "'enum'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400343 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500344 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400345 if (!this->expect(Token::Kind::TK_CLASS, "'class'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400346 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500347 }
348 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400349 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400350 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500351 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400352 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400353 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500354 }
355 fTypes.add(this->text(name), std::unique_ptr<Symbol>(new Type(this->text(name),
356 Type::kEnum_Kind)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400357 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kEnum, this->text(name));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400358 if (!this->checkNext(Token::Kind::TK_RBRACE)) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500359 Token id;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400360 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400361 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500362 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400363 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400364 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500365 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400366 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500367 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400368 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
369 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500370 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400371 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500372 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400373 while (!this->checkNext(Token::Kind::TK_RBRACE)) {
374 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400375 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500376 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400377 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &id)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400378 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500379 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400380 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400381 ASTNode::ID value = this->assignmentExpression();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500382 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400383 return ASTNode::ID::Invalid();
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500384 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400385 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
386 getNode(child).addChild(value);
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500387 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400388 CREATE_CHILD(child, result, id.fOffset, ASTNode::Kind::kEnumCase, this->text(id));
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500389 }
390 }
391 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400392 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400393 return result;
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500394}
395
396/* enumDeclaration | modifiers (structVarDeclaration | type IDENTIFIER ((LPAREN parameter
ethannicholasb3058bd2016-07-01 08:22:01 -0700397 (COMMA parameter)* RPAREN (block | SEMICOLON)) | SEMICOLON) | interfaceBlock) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400398ASTNode::ID Parser::declaration() {
ethannicholasb3058bd2016-07-01 08:22:01 -0700399 Token lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400400 if (lookahead.fKind == Token::Kind::TK_ENUM) {
Ethan Nicholasaae47c82017-11-10 15:34:03 -0500401 return this->enumDeclaration();
402 }
403 Modifiers modifiers = this->modifiers();
404 lookahead = this->peek();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400405 if (lookahead.fKind == Token::Kind::TK_IDENTIFIER && !this->isType(this->text(lookahead))) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700406 // we have an identifier that's not a type, could be the start of an interface block
407 return this->interfaceBlock(modifiers);
408 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400409 if (lookahead.fKind == Token::Kind::TK_STRUCT) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700410 return this->structVarDeclaration(modifiers);
411 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400412 if (lookahead.fKind == Token::Kind::TK_SEMICOLON) {
ethannicholas5961bc92016-10-12 06:39:56 -0700413 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400414 RETURN_NODE(lookahead.fOffset, ASTNode::Kind::kModifiers, modifiers);
ethannicholas5961bc92016-10-12 06:39:56 -0700415 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400416 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700417 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400418 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700419 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400420 if (getNode(type).getTypeData().fIsStructDeclaration &&
421 this->checkNext(Token::Kind::TK_SEMICOLON)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400422 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700423 }
424 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400425 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400426 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700427 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400428 if (this->checkNext(Token::Kind::TK_LPAREN)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400429 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kFunction);
430 ASTNode::FunctionData fd(modifiers, this->text(name), 0);
431 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400432 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400433 for (;;) {
434 ASTNode::ID parameter = this->parameter();
435 if (!parameter) {
436 return ASTNode::ID::Invalid();
437 }
438 ++fd.fParameterCount;
439 getNode(result).addChild(parameter);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400440 if (!this->checkNext(Token::Kind::TK_COMMA)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400441 break;
ethannicholasb3058bd2016-07-01 08:22:01 -0700442 }
443 }
ethannicholasb3058bd2016-07-01 08:22:01 -0700444 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400445 getNode(result).setFunctionData(fd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400446 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400447 return ASTNode::ID::Invalid();
448 }
449 ASTNode::ID body;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400450 if (!this->checkNext(Token::Kind::TK_SEMICOLON)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700451 body = this->block();
452 if (!body) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400453 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700454 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400455 getNode(result).addChild(body);
ethannicholasb3058bd2016-07-01 08:22:01 -0700456 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400457 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700458 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400459 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700460 }
461}
462
463/* modifiers type IDENTIFIER varDeclarationEnd */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400464ASTNode::ID Parser::varDeclarations() {
Ethan Nicholas11d53972016-11-28 11:23:23 -0500465 Modifiers modifiers = this->modifiers();
Ethan Nicholasfc994162019-06-06 10:04:27 -0400466 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700467 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400468 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700469 }
470 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400471 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400472 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700473 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400474 return this->varDeclarationEnd(modifiers, type, this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700475}
476
477/* STRUCT IDENTIFIER LBRACE varDeclaration* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400478ASTNode::ID Parser::structDeclaration() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400479 if (!this->expect(Token::Kind::TK_STRUCT, "'struct'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400480 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700481 }
482 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400483 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400484 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700485 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400486 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400487 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700488 }
489 std::vector<Type::Field> fields;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400490 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400491 ASTNode::ID decls = this->varDeclarations();
492 if (!decls) {
493 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700494 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400495 ASTNode& declsNode = getNode(decls);
496 auto type = (const Type*) fTypes[(declsNode.begin() + 1)->getTypeData().fName];
497 for (auto iter = declsNode.begin() + 2; iter != declsNode.end(); ++iter) {
498 ASTNode& var = *iter;
499 ASTNode::VarData vd = var.getVarData();
500 for (int j = vd.fSizeCount - 1; j >= 0; j--) {
501 const ASTNode& size = *(var.begin() + j);
502 if (!size || size.fKind != ASTNode::Kind::kInt) {
503 this->error(declsNode.fOffset, "array size in struct field must be a constant");
504 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700505 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400506 uint64_t columns = size.getInt();
Ethan Nicholas0df1b042017-03-31 13:56:23 -0400507 String name = type->name() + "[" + to_string(columns) + "]";
Ethan Nicholas91164d12019-05-15 15:29:54 -0400508 type = (Type*) fTypes.takeOwnership(std::unique_ptr<Symbol>(
509 new Type(name,
510 Type::kArray_Kind,
511 *type,
512 (int) columns)));
ethannicholasb3058bd2016-07-01 08:22:01 -0700513 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400514 fields.push_back(Type::Field(declsNode.begin()->getModifiers(), vd.fName, type));
515 if (vd.fSizeCount ? (var.begin() + (vd.fSizeCount - 1))->fNext : var.fFirstChild) {
516 this->error(declsNode.fOffset, "initializers are not permitted on struct fields");
ethannicholasb3058bd2016-07-01 08:22:01 -0700517 }
518 }
519 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400520 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400521 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700522 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700523 fTypes.add(this->text(name), std::unique_ptr<Type>(new Type(name.fOffset, this->text(name),
524 fields)));
Ethan Nicholasfc994162019-06-06 10:04:27 -0400525 RETURN_NODE(name.fOffset, ASTNode::Kind::kType,
526 ASTNode::TypeData(this->text(name), true, false));
ethannicholasb3058bd2016-07-01 08:22:01 -0700527}
528
529/* structDeclaration ((IDENTIFIER varDeclarationEnd) | SEMICOLON) */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400530ASTNode::ID Parser::structVarDeclaration(Modifiers modifiers) {
531 ASTNode::ID type = this->structDeclaration();
ethannicholasb3058bd2016-07-01 08:22:01 -0700532 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400533 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700534 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -0400535 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400536 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400537 return this->varDeclarationEnd(modifiers, std::move(type), this->text(name));
ethannicholasb3058bd2016-07-01 08:22:01 -0700538 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400539 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -0400540 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700541}
542
Ethan Nicholas4b330df2017-05-17 10:52:55 -0400543/* (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)? (COMMA IDENTIFER
544 (LBRACKET expression? RBRACKET)* (EQ assignmentExpression)?)* SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400545ASTNode::ID Parser::varDeclarationEnd(Modifiers mods, ASTNode::ID type, StringFragment name) {
546 CREATE_NODE(result, -1, ASTNode::Kind::kVarDeclarations);
547 CREATE_CHILD(modifiers, result, -1, ASTNode::Kind::kModifiers, mods);
548 getNode(result).addChild(type);
549 CREATE_NODE(currentVar, -1, ASTNode::Kind::kVarDeclaration);
550 ASTNode::VarData vd(name, 0);
551 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400552 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
553 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400554 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700555 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400556 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700557 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400558 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700559 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400560 getNode(currentVar).addChild(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400561 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400562 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700563 }
564 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400565 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700566 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400567 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400568 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400569 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700570 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400571 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700572 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400573 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700574 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400575 while (this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700576 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400577 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400578 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700579 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400580 currentVar = ASTNode::ID(fFile->fNodes.size());
581 vd = ASTNode::VarData(this->text(name), 0);
582 fFile->fNodes.emplace_back(&fFile->fNodes, -1, ASTNode::Kind::kVarDeclaration);
583 getNode(result).addChild(currentVar);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400584 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
585 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400586 CREATE_EMPTY_CHILD(currentVar);
ethannicholasb3058bd2016-07-01 08:22:01 -0700587 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400588 ASTNode::ID size = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700589 if (!size) {
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(size);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400593 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400594 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700595 }
596 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400597 ++vd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700598 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400599 getNode(currentVar).setVarData(vd);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400600 if (this->checkNext(Token::Kind::TK_EQ)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400601 ASTNode::ID value = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -0700602 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400603 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700604 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400605 getNode(currentVar).addChild(value);
ethannicholasb3058bd2016-07-01 08:22:01 -0700606 }
607 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400608 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
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 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700612}
613
614/* modifiers type IDENTIFIER (LBRACKET INT_LITERAL RBRACKET)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400615ASTNode::ID Parser::parameter() {
Ethan Nicholasc6f5e102017-03-31 14:53:17 -0400616 Modifiers modifiers = this->modifiersWithDefaults(0);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400617 ASTNode::ID type = this->type();
ethannicholasb3058bd2016-07-01 08:22:01 -0700618 if (!type) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400619 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700620 }
621 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400622 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400623 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700624 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400625 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kParameter);
626 ASTNode::ParameterData pd(modifiers, this->text(name), 0);
627 getNode(result).addChild(type);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400628 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700629 Token sizeToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400630 if (!this->expect(Token::Kind::TK_INT_LITERAL, "a positive integer", &sizeToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400631 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700632 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400633 CREATE_CHILD(child, result, sizeToken.fOffset, ASTNode::Kind::kInt,
634 SkSL::stoi(this->text(sizeToken)));
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400635 if (!this->expect(Token::Kind::TK_RBRACKET, "']'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400636 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -0700637 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400638 ++pd.fSizeCount;
ethannicholasb3058bd2016-07-01 08:22:01 -0700639 }
Ethan Nicholasfc994162019-06-06 10:04:27 -0400640 getNode(result).setParameterData(pd);
641 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -0700642}
643
Ethan Nicholasd608c092017-10-26 09:30:08 -0400644/** EQ INT_LITERAL */
ethannicholasb3058bd2016-07-01 08:22:01 -0700645int Parser::layoutInt() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400646 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700647 return -1;
648 }
649 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400650 if (this->expect(Token::Kind::TK_INT_LITERAL, "a non-negative integer", &resultToken)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700651 return SkSL::stoi(this->text(resultToken));
ethannicholasb3058bd2016-07-01 08:22:01 -0700652 }
653 return -1;
654}
655
Ethan Nicholasd608c092017-10-26 09:30:08 -0400656/** EQ IDENTIFIER */
657StringFragment Parser::layoutIdentifier() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400658 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400659 return StringFragment();
660 }
661 Token resultToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400662 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &resultToken)) {
Ethan Nicholasd608c092017-10-26 09:30:08 -0400663 return StringFragment();
664 }
665 return this->text(resultToken);
666}
667
668
Ethan Nicholas762466e2017-06-29 10:03:38 -0400669/** EQ <any sequence of tokens with balanced parentheses and no top-level comma> */
Ethan Nicholasfc994162019-06-06 10:04:27 -0400670StringFragment Parser::layoutCode() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400671 if (!this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400672 return "";
673 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700674 Token start = this->nextRawToken();
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400675 this->pushback(start);
Ethan Nicholasfc994162019-06-06 10:04:27 -0400676 StringFragment code;
677 code.fChars = fText + start.fOffset;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400678 int level = 1;
679 bool done = false;
680 while (!done) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700681 Token next = this->nextRawToken();
Ethan Nicholas762466e2017-06-29 10:03:38 -0400682 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400683 case Token::Kind::TK_LPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400684 ++level;
685 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400686 case Token::Kind::TK_RPAREN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400687 --level;
688 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400689 case Token::Kind::TK_COMMA:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400690 if (level == 1) {
691 done = true;
692 }
693 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400694 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700695 this->error(start, "reached end of file while parsing layout");
Ethan Nicholas6d71f492019-06-10 16:58:37 -0400696 return "";
Ethan Nicholas762466e2017-06-29 10:03:38 -0400697 default:
698 break;
699 }
700 if (!level) {
701 done = true;
702 }
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400703 if (done) {
Ethan Nicholasfc994162019-06-06 10:04:27 -0400704 code.fLength = next.fOffset - start.fOffset;
Ethan Nicholas08b79b72017-08-14 10:35:37 -0400705 this->pushback(std::move(next));
706 }
Ethan Nicholas762466e2017-06-29 10:03:38 -0400707 }
708 return code;
709}
710
711/** (EQ IDENTIFIER('identity'))? */
712Layout::Key Parser::layoutKey() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400713 if (this->peek().fKind == Token::Kind::TK_EQ) {
714 this->expect(Token::Kind::TK_EQ, "'='");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400715 Token key;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400716 if (this->expect(Token::Kind::TK_IDENTIFIER, "an identifer", &key)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700717 if (this->text(key) == "identity") {
Ethan Nicholas762466e2017-06-29 10:03:38 -0400718 return Layout::kIdentity_Key;
719 } else {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700720 this->error(key, "unsupported layout key");
Ethan Nicholas762466e2017-06-29 10:03:38 -0400721 }
722 }
723 }
724 return Layout::kKey_Key;
725}
726
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400727Layout::CType Parser::layoutCType() {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400728 if (this->expect(Token::Kind::TK_EQ, "'='")) {
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400729 Token t = this->nextToken();
730 String text = this->text(t);
731 auto found = layoutTokens->find(text);
732 if (found != layoutTokens->end()) {
733 switch (found->second) {
Brian Osmanf28e55d2018-10-03 16:35:54 -0400734 case LayoutToken::SKPMCOLOR4F:
735 return Layout::CType::kSkPMColor4f;
Mike Reedb26b4e72020-01-22 14:31:21 -0500736 case LayoutToken::SKV4:
737 return Layout::CType::kSkV4;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400738 case LayoutToken::SKRECT:
739 return Layout::CType::kSkRect;
740 case LayoutToken::SKIRECT:
741 return Layout::CType::kSkIRect;
742 case LayoutToken::SKPMCOLOR:
743 return Layout::CType::kSkPMColor;
Ethan Nicholasc1c686b2019-04-02 17:30:23 -0400744 case LayoutToken::BOOL:
745 return Layout::CType::kBool;
746 case LayoutToken::INT:
747 return Layout::CType::kInt32;
748 case LayoutToken::FLOAT:
749 return Layout::CType::kFloat;
Mike Reedb26b4e72020-01-22 14:31:21 -0500750 case LayoutToken::SKM44:
751 return Layout::CType::kSkM44;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400752 default:
753 break;
754 }
755 }
756 this->error(t, "unsupported ctype");
757 }
758 return Layout::CType::kDefault;
759}
760
ethannicholas8ac838d2016-11-22 08:39:36 -0800761/* LAYOUT LPAREN IDENTIFIER (EQ INT_LITERAL)? (COMMA IDENTIFIER (EQ INT_LITERAL)?)* RPAREN */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500762Layout Parser::layout() {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500763 int flags = 0;
ethannicholasb3058bd2016-07-01 08:22:01 -0700764 int location = -1;
Ethan Nicholas19671772016-11-28 16:30:17 -0500765 int offset = -1;
ethannicholasb3058bd2016-07-01 08:22:01 -0700766 int binding = -1;
767 int index = -1;
768 int set = -1;
769 int builtin = -1;
Greg Daniel64773e62016-11-22 09:44:03 -0500770 int inputAttachmentIndex = -1;
Ethan Nicholas11d53972016-11-28 11:23:23 -0500771 Layout::Format format = Layout::Format::kUnspecified;
Ethan Nicholas52cad152017-02-16 16:37:32 -0500772 Layout::Primitive primitive = Layout::kUnspecified_Primitive;
773 int maxVertices = -1;
774 int invocations = -1;
Brian Osmanf59a9612020-04-15 14:18:13 -0400775 StringFragment marker;
Ethan Nicholasfc994162019-06-06 10:04:27 -0400776 StringFragment when;
Ethan Nicholas762466e2017-06-29 10:03:38 -0400777 Layout::Key key = Layout::kNo_Key;
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400778 Layout::CType ctype = Layout::CType::kDefault;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400779 if (this->checkNext(Token::Kind::TK_LAYOUT)) {
780 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500781 return Layout(flags, location, offset, binding, index, set, builtin,
Brian Osmanf59a9612020-04-15 14:18:13 -0400782 inputAttachmentIndex, format, primitive, maxVertices, invocations, marker,
783 when, key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700784 }
785 for (;;) {
786 Token t = this->nextToken();
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700787 String text = this->text(t);
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400788 auto found = layoutTokens->find(text);
789 if (found != layoutTokens->end()) {
790 switch (found->second) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700791 case LayoutToken::LOCATION:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500792 location = this->layoutInt();
793 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700794 case LayoutToken::OFFSET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500795 offset = this->layoutInt();
796 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700797 case LayoutToken::BINDING:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500798 binding = this->layoutInt();
799 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700800 case LayoutToken::INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500801 index = this->layoutInt();
802 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700803 case LayoutToken::SET:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500804 set = this->layoutInt();
805 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700806 case LayoutToken::BUILTIN:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500807 builtin = this->layoutInt();
808 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700809 case LayoutToken::INPUT_ATTACHMENT_INDEX:
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500810 inputAttachmentIndex = this->layoutInt();
811 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700812 case LayoutToken::ORIGIN_UPPER_LEFT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500813 flags |= Layout::kOriginUpperLeft_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500814 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700815 case LayoutToken::OVERRIDE_COVERAGE:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500816 flags |= Layout::kOverrideCoverage_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500817 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700818 case LayoutToken::BLEND_SUPPORT_ALL_EQUATIONS:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500819 flags |= Layout::kBlendSupportAllEquations_Flag;
820 break;
821 case LayoutToken::BLEND_SUPPORT_MULTIPLY:
822 flags |= Layout::kBlendSupportMultiply_Flag;
823 break;
824 case LayoutToken::BLEND_SUPPORT_SCREEN:
825 flags |= Layout::kBlendSupportScreen_Flag;
826 break;
827 case LayoutToken::BLEND_SUPPORT_OVERLAY:
828 flags |= Layout::kBlendSupportOverlay_Flag;
829 break;
830 case LayoutToken::BLEND_SUPPORT_DARKEN:
831 flags |= Layout::kBlendSupportDarken_Flag;
832 break;
833 case LayoutToken::BLEND_SUPPORT_LIGHTEN:
834 flags |= Layout::kBlendSupportLighten_Flag;
835 break;
836 case LayoutToken::BLEND_SUPPORT_COLORDODGE:
837 flags |= Layout::kBlendSupportColorDodge_Flag;
838 break;
839 case LayoutToken::BLEND_SUPPORT_COLORBURN:
840 flags |= Layout::kBlendSupportColorBurn_Flag;
841 break;
842 case LayoutToken::BLEND_SUPPORT_HARDLIGHT:
843 flags |= Layout::kBlendSupportHardLight_Flag;
844 break;
845 case LayoutToken::BLEND_SUPPORT_SOFTLIGHT:
846 flags |= Layout::kBlendSupportSoftLight_Flag;
847 break;
848 case LayoutToken::BLEND_SUPPORT_DIFFERENCE:
849 flags |= Layout::kBlendSupportDifference_Flag;
850 break;
851 case LayoutToken::BLEND_SUPPORT_EXCLUSION:
852 flags |= Layout::kBlendSupportExclusion_Flag;
853 break;
854 case LayoutToken::BLEND_SUPPORT_HSL_HUE:
855 flags |= Layout::kBlendSupportHSLHue_Flag;
856 break;
857 case LayoutToken::BLEND_SUPPORT_HSL_SATURATION:
858 flags |= Layout::kBlendSupportHSLSaturation_Flag;
859 break;
860 case LayoutToken::BLEND_SUPPORT_HSL_COLOR:
861 flags |= Layout::kBlendSupportHSLColor_Flag;
862 break;
863 case LayoutToken::BLEND_SUPPORT_HSL_LUMINOSITY:
864 flags |= Layout::kBlendSupportHSLLuminosity_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500865 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700866 case LayoutToken::PUSH_CONSTANT:
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500867 flags |= Layout::kPushConstant_Flag;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500868 break;
Michael Ludwiga4275592018-08-31 10:52:47 -0400869 case LayoutToken::TRACKED:
870 flags |= Layout::kTracked_Flag;
871 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700872 case LayoutToken::POINTS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500873 primitive = Layout::kPoints_Primitive;
874 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700875 case LayoutToken::LINES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500876 primitive = Layout::kLines_Primitive;
877 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700878 case LayoutToken::LINE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500879 primitive = Layout::kLineStrip_Primitive;
880 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700881 case LayoutToken::LINES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500882 primitive = Layout::kLinesAdjacency_Primitive;
883 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700884 case LayoutToken::TRIANGLES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500885 primitive = Layout::kTriangles_Primitive;
886 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700887 case LayoutToken::TRIANGLE_STRIP:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500888 primitive = Layout::kTriangleStrip_Primitive;
889 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700890 case LayoutToken::TRIANGLES_ADJACENCY:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500891 primitive = Layout::kTrianglesAdjacency_Primitive;
892 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700893 case LayoutToken::MAX_VERTICES:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500894 maxVertices = this->layoutInt();
895 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700896 case LayoutToken::INVOCATIONS:
Ethan Nicholas52cad152017-02-16 16:37:32 -0500897 invocations = this->layoutInt();
898 break;
Brian Osmanf59a9612020-04-15 14:18:13 -0400899 case LayoutToken::MARKER:
900 marker = this->layoutCode();
901 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700902 case LayoutToken::WHEN:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400903 when = this->layoutCode();
904 break;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -0700905 case LayoutToken::KEY:
Ethan Nicholas762466e2017-06-29 10:03:38 -0400906 key = this->layoutKey();
907 break;
Ethan Nicholasd608c092017-10-26 09:30:08 -0400908 case LayoutToken::CTYPE:
Ethan Nicholas78aceb22018-08-31 16:13:58 -0400909 ctype = this->layoutCType();
910 break;
911 default:
912 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
Ethan Nicholasd608c092017-10-26 09:30:08 -0400913 break;
Ethan Nicholas3614d9a2017-02-15 12:33:30 -0500914 }
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400915 } else if (Layout::ReadFormat(text, &format)) {
Brian Salomon2a51de82016-11-16 12:06:01 -0500916 // AST::ReadFormat stored the result in 'format'.
ethannicholasb3058bd2016-07-01 08:22:01 -0700917 } else {
Ethan Nicholasb93af7e2018-07-24 11:28:52 -0400918 this->error(t, ("'" + text + "' is not a valid layout qualifier").c_str());
ethannicholasb3058bd2016-07-01 08:22:01 -0700919 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400920 if (this->checkNext(Token::Kind::TK_RPAREN)) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700921 break;
922 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400923 if (!this->expect(Token::Kind::TK_COMMA, "','")) {
ethannicholasb3058bd2016-07-01 08:22:01 -0700924 break;
925 }
926 }
927 }
Ethan Nicholas39204fd2017-11-27 13:12:30 -0500928 return Layout(flags, location, offset, binding, index, set, builtin, inputAttachmentIndex,
Brian Osmanf59a9612020-04-15 14:18:13 -0400929 format, primitive, maxVertices, invocations, marker, when, key, ctype);
ethannicholasb3058bd2016-07-01 08:22:01 -0700930}
931
Brian Salomonf9f45122016-11-29 11:59:17 -0500932/* layout? (UNIFORM | CONST | IN | OUT | INOUT | LOWP | MEDIUMP | HIGHP | FLAT | NOPERSPECTIVE |
Ethan Nicholasa7ceb502019-01-11 10:31:48 -0500933 READONLY | WRITEONLY | COHERENT | VOLATILE | RESTRICT | BUFFER | PLS | PLSIN |
Brian Osman3c358422020-03-23 10:44:12 -0400934 PLSOUT | VARYING)* */
Ethan Nicholas11d53972016-11-28 11:23:23 -0500935Modifiers Parser::modifiers() {
936 Layout layout = this->layout();
ethannicholasb3058bd2016-07-01 08:22:01 -0700937 int flags = 0;
938 for (;;) {
939 // TODO: handle duplicate / incompatible flags
940 switch (peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400941 case Token::Kind::TK_UNIFORM:
ethannicholasb3058bd2016-07-01 08:22:01 -0700942 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500943 flags |= Modifiers::kUniform_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700944 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400945 case Token::Kind::TK_CONST:
ethannicholasb3058bd2016-07-01 08:22:01 -0700946 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500947 flags |= Modifiers::kConst_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700948 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400949 case Token::Kind::TK_IN:
ethannicholasb3058bd2016-07-01 08:22:01 -0700950 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500951 flags |= Modifiers::kIn_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700952 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400953 case Token::Kind::TK_OUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700954 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500955 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700956 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400957 case Token::Kind::TK_INOUT:
ethannicholasb3058bd2016-07-01 08:22:01 -0700958 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500959 flags |= Modifiers::kIn_Flag;
960 flags |= Modifiers::kOut_Flag;
ethannicholasb3058bd2016-07-01 08:22:01 -0700961 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400962 case Token::Kind::TK_FLAT:
ethannicholasf789b382016-08-03 12:43:36 -0700963 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500964 flags |= Modifiers::kFlat_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700965 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400966 case Token::Kind::TK_NOPERSPECTIVE:
ethannicholasf789b382016-08-03 12:43:36 -0700967 this->nextToken();
Ethan Nicholas11d53972016-11-28 11:23:23 -0500968 flags |= Modifiers::kNoPerspective_Flag;
ethannicholasf789b382016-08-03 12:43:36 -0700969 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400970 case Token::Kind::TK_READONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500971 this->nextToken();
972 flags |= Modifiers::kReadOnly_Flag;
973 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400974 case Token::Kind::TK_WRITEONLY:
Brian Salomonf9f45122016-11-29 11:59:17 -0500975 this->nextToken();
976 flags |= Modifiers::kWriteOnly_Flag;
977 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400978 case Token::Kind::TK_COHERENT:
Brian Salomonf9f45122016-11-29 11:59:17 -0500979 this->nextToken();
980 flags |= Modifiers::kCoherent_Flag;
981 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400982 case Token::Kind::TK_VOLATILE:
Brian Salomonf9f45122016-11-29 11:59:17 -0500983 this->nextToken();
984 flags |= Modifiers::kVolatile_Flag;
985 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400986 case Token::Kind::TK_RESTRICT:
Brian Salomonf9f45122016-11-29 11:59:17 -0500987 this->nextToken();
988 flags |= Modifiers::kRestrict_Flag;
989 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400990 case Token::Kind::TK_BUFFER:
Ethan Nicholas0dd30d92017-05-01 16:57:07 -0400991 this->nextToken();
992 flags |= Modifiers::kBuffer_Flag;
993 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400994 case Token::Kind::TK_HASSIDEEFFECTS:
Ethan Nicholascb670962017-04-20 19:31:52 -0400995 this->nextToken();
996 flags |= Modifiers::kHasSideEffects_Flag;
997 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -0400998 case Token::Kind::TK_PLS:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -0500999 this->nextToken();
1000 flags |= Modifiers::kPLS_Flag;
1001 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001002 case Token::Kind::TK_PLSIN:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001003 this->nextToken();
1004 flags |= Modifiers::kPLSIn_Flag;
1005 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001006 case Token::Kind::TK_PLSOUT:
Ethan Nicholasa7ceb502019-01-11 10:31:48 -05001007 this->nextToken();
1008 flags |= Modifiers::kPLSOut_Flag;
1009 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001010 case Token::Kind::TK_VARYING:
Brian Osman3c358422020-03-23 10:44:12 -04001011 this->nextToken();
1012 flags |= Modifiers::kVarying_Flag;
1013 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001014 default:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001015 return Modifiers(layout, flags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001016 }
1017 }
1018}
1019
Ethan Nicholas11d53972016-11-28 11:23:23 -05001020Modifiers Parser::modifiersWithDefaults(int defaultFlags) {
1021 Modifiers result = this->modifiers();
ethannicholasb3058bd2016-07-01 08:22:01 -07001022 if (!result.fFlags) {
Ethan Nicholas11d53972016-11-28 11:23:23 -05001023 return Modifiers(result.fLayout, defaultFlags);
ethannicholasb3058bd2016-07-01 08:22:01 -07001024 }
1025 return result;
1026}
1027
1028/* ifStatement | forStatement | doStatement | whileStatement | block | expression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001029ASTNode::ID Parser::statement() {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001030 Token start = this->nextToken();
1031 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001032 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001033 return ASTNode::ID::Invalid();
1034 }
1035 this->pushback(start);
ethannicholasb3058bd2016-07-01 08:22:01 -07001036 switch (start.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001037 case Token::Kind::TK_IF: // fall through
1038 case Token::Kind::TK_STATIC_IF:
ethannicholasb3058bd2016-07-01 08:22:01 -07001039 return this->ifStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001040 case Token::Kind::TK_FOR:
ethannicholasb3058bd2016-07-01 08:22:01 -07001041 return this->forStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001042 case Token::Kind::TK_DO:
ethannicholasb3058bd2016-07-01 08:22:01 -07001043 return this->doStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001044 case Token::Kind::TK_WHILE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001045 return this->whileStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001046 case Token::Kind::TK_SWITCH: // fall through
1047 case Token::Kind::TK_STATIC_SWITCH:
Ethan Nicholasaf197692017-02-27 13:26:45 -05001048 return this->switchStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001049 case Token::Kind::TK_RETURN:
ethannicholasb3058bd2016-07-01 08:22:01 -07001050 return this->returnStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001051 case Token::Kind::TK_BREAK:
ethannicholasb3058bd2016-07-01 08:22:01 -07001052 return this->breakStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001053 case Token::Kind::TK_CONTINUE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001054 return this->continueStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001055 case Token::Kind::TK_DISCARD:
ethannicholasb3058bd2016-07-01 08:22:01 -07001056 return this->discardStatement();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001057 case Token::Kind::TK_LBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001058 return this->block();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001059 case Token::Kind::TK_SEMICOLON:
Ethan Nicholas11d53972016-11-28 11:23:23 -05001060 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001061 RETURN_NODE(start.fOffset, ASTNode::Kind::kBlock);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001062 case Token::Kind::TK_CONST:
Ethan Nicholasfc994162019-06-06 10:04:27 -04001063 return this->varDeclarations();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001064 case Token::Kind::TK_IDENTIFIER:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001065 if (this->isType(this->text(start))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001066 return this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001067 }
1068 // fall through
1069 default:
1070 return this->expressionStatement();
Ethan Nicholas11d53972016-11-28 11:23:23 -05001071 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001072}
1073
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05001074/* IDENTIFIER(type) (LBRACKET intLiteral? RBRACKET)* QUESTION? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001075ASTNode::ID Parser::type() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001076 Token type;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001077 if (!this->expect(Token::Kind::TK_IDENTIFIER, "a type", &type)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001078 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001079 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001080 if (!this->isType(this->text(type))) {
1081 this->error(type, ("no type named '" + this->text(type) + "'").c_str());
Ethan Nicholasfc994162019-06-06 10:04:27 -04001082 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001083 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001084 CREATE_NODE(result, type.fOffset, ASTNode::Kind::kType);
1085 ASTNode::TypeData td(this->text(type), false, false);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001086 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1087 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001088 SKSL_INT i;
Ethan Nicholas50afc172017-02-16 14:49:57 -05001089 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001090 CREATE_CHILD(child, result, -1, ASTNode::Kind::kInt, i);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001091 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001092 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001093 }
1094 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001095 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001096 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001097 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001098 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001099 td.fIsNullable = this->checkNext(Token::Kind::TK_QUESTION);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001100 getNode(result).setTypeData(td);
1101 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001102}
1103
John Stiles57a996b2020-04-19 19:05:12 -07001104/* IDENTIFIER LBRACE
1105 varDeclaration+
1106 RBRACE (IDENTIFIER (LBRACKET expression? RBRACKET)*)? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001107ASTNode::ID Parser::interfaceBlock(Modifiers mods) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001108 Token name;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001109 if (!this->expect(Token::Kind::TK_IDENTIFIER, "an identifier", &name)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001110 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001111 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001112 if (peek().fKind != Token::Kind::TK_LBRACE) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001113 // we only get into interfaceBlock if we found a top-level identifier which was not a type.
Ethan Nicholas11d53972016-11-28 11:23:23 -05001114 // 99% of the time, the user was not actually intending to create an interface block, so
ethannicholasb3058bd2016-07-01 08:22:01 -07001115 // it's better to report it as an unknown type
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001116 this->error(name, "no type named '" + this->text(name) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001117 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001118 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001119 CREATE_NODE(result, name.fOffset, ASTNode::Kind::kInterfaceBlock);
1120 ASTNode::InterfaceBlockData id(mods, this->text(name), 0, "", 0);
ethannicholasb3058bd2016-07-01 08:22:01 -07001121 this->nextToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001122 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001123 ASTNode::ID decl = this->varDeclarations();
ethannicholasb3058bd2016-07-01 08:22:01 -07001124 if (!decl) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001125 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001126 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001127 getNode(result).addChild(decl);
1128 ++id.fDeclarationCount;
ethannicholasb3058bd2016-07-01 08:22:01 -07001129 }
John Stiles57a996b2020-04-19 19:05:12 -07001130 if (id.fDeclarationCount == 0) {
1131 this->error(name, "interface block '" + this->text(name) +
1132 "' must contain at least one member");
1133 return ASTNode::ID::Invalid();
1134 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001135 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001136 std::vector<ASTNode> sizes;
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001137 StringFragment instanceName;
1138 Token instanceNameToken;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001139 if (this->checkNext(Token::Kind::TK_IDENTIFIER, &instanceNameToken)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001140 id.fInstanceName = this->text(instanceNameToken);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001141 while (this->checkNext(Token::Kind::TK_LBRACKET)) {
1142 if (this->peek().fKind != Token::Kind::TK_RBRACKET) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001143 ASTNode::ID size = this->expression();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001144 if (!size) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001145 return ASTNode::ID::Invalid();
Ethan Nicholas50afc172017-02-16 14:49:57 -05001146 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001147 getNode(result).addChild(size);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001148 } else {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001149 CREATE_EMPTY_CHILD(result);
Ethan Nicholas50afc172017-02-16 14:49:57 -05001150 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001151 ++id.fSizeCount;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001152 this->expect(Token::Kind::TK_RBRACKET, "']'");
Ethan Nicholas50afc172017-02-16 14:49:57 -05001153 }
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001154 instanceName = this->text(instanceNameToken);
ethannicholasb3058bd2016-07-01 08:22:01 -07001155 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001156 getNode(result).setInterfaceBlockData(id);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001157 this->expect(Token::Kind::TK_SEMICOLON, "';'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001158 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001159}
1160
1161/* IF LPAREN expression RPAREN statement (ELSE statement)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001162ASTNode::ID Parser::ifStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001163 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001164 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_IF, &start);
1165 if (!isStatic && !this->expect(Token::Kind::TK_IF, "'if'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001166 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001167 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001168 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kIf, isStatic);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001169 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001170 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001171 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001172 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001173 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001174 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001175 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001176 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001177 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001178 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001179 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001180 ASTNode::ID ifTrue = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001181 if (!ifTrue) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001182 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001183 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001184 getNode(result).addChild(ifTrue);
1185 ASTNode::ID ifFalse;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001186 if (this->checkNext(Token::Kind::TK_ELSE)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001187 ifFalse = this->statement();
1188 if (!ifFalse) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001189 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001190 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001191 getNode(result).addChild(ifFalse);
ethannicholasb3058bd2016-07-01 08:22:01 -07001192 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001193 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001194}
1195
1196/* DO statement WHILE LPAREN expression RPAREN SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001197ASTNode::ID Parser::doStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001198 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001199 if (!this->expect(Token::Kind::TK_DO, "'do'", &start)) {
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 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kDo);
1203 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001204 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001205 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001206 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001207 getNode(result).addChild(statement);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001208 if (!this->expect(Token::Kind::TK_WHILE, "'while'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001209 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001210 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001211 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001212 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001213 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001214 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001215 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001216 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001217 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001218 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001219 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001220 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001221 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001222 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001223 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001224 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001225 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001226}
1227
1228/* WHILE LPAREN expression RPAREN STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001229ASTNode::ID Parser::whileStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001230 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001231 if (!this->expect(Token::Kind::TK_WHILE, "'while'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001232 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001233 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001234 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001235 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001236 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001237 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kWhile);
1238 ASTNode::ID test = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001239 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001240 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001241 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001242 getNode(result).addChild(test);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001243 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001244 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001245 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001246 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001247 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001248 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001249 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001250 getNode(result).addChild(statement);
1251 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001252}
1253
Ethan Nicholasaf197692017-02-27 13:26:45 -05001254/* CASE expression COLON statement* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001255ASTNode::ID Parser::switchCase() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001256 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001257 if (!this->expect(Token::Kind::TK_CASE, "'case'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001258 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001259 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001260 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitchCase);
1261 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001262 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001263 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001264 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001265 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001266 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001267 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001268 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001269 while (this->peek().fKind != Token::Kind::TK_RBRACE &&
1270 this->peek().fKind != Token::Kind::TK_CASE &&
1271 this->peek().fKind != Token::Kind::TK_DEFAULT) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001272 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001273 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001274 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001275 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001276 getNode(result).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001277 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001278 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001279}
1280
1281/* SWITCH LPAREN expression RPAREN LBRACE switchCase* (DEFAULT COLON statement*)? RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001282ASTNode::ID Parser::switchStatement() {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001283 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001284 bool isStatic = this->checkNext(Token::Kind::TK_STATIC_SWITCH, &start);
1285 if (!isStatic && !this->expect(Token::Kind::TK_SWITCH, "'switch'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001286 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001287 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001288 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001289 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001290 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001291 ASTNode::ID value = this->expression();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001292 if (!value) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001293 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001294 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001295 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001296 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001297 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001298 if (!this->expect(Token::Kind::TK_LBRACE, "'{'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001299 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001300 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001301 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kSwitch, isStatic);
1302 getNode(result).addChild(value);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001303 while (this->peek().fKind == Token::Kind::TK_CASE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001304 ASTNode::ID c = this->switchCase();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001305 if (!c) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001306 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001307 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001308 getNode(result).addChild(c);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001309 }
1310 // Requiring default: to be last (in defiance of C and GLSL) was a deliberate decision. Other
1311 // parts of the compiler may rely upon this assumption.
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001312 if (this->peek().fKind == Token::Kind::TK_DEFAULT) {
Ethan Nicholasaf197692017-02-27 13:26:45 -05001313 Token defaultStart;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001314 SkAssertResult(this->expect(Token::Kind::TK_DEFAULT, "'default'", &defaultStart));
1315 if (!this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001316 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001317 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001318 CREATE_CHILD(defaultCase, result, defaultStart.fOffset, ASTNode::Kind::kSwitchCase);
1319 CREATE_EMPTY_CHILD(defaultCase); // empty test to signify default case
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001320 while (this->peek().fKind != Token::Kind::TK_RBRACE) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001321 ASTNode::ID s = this->statement();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001322 if (!s) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001323 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001324 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001325 getNode(defaultCase).addChild(s);
Ethan Nicholasaf197692017-02-27 13:26:45 -05001326 }
Ethan Nicholasaf197692017-02-27 13:26:45 -05001327 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001328 if (!this->expect(Token::Kind::TK_RBRACE, "'}'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001329 return ASTNode::ID::Invalid();
Ethan Nicholasaf197692017-02-27 13:26:45 -05001330 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001331 return result;
Ethan Nicholasaf197692017-02-27 13:26:45 -05001332}
1333
Ethan Nicholas11d53972016-11-28 11:23:23 -05001334/* FOR LPAREN (declaration | expression)? SEMICOLON expression? SEMICOLON expression? RPAREN
ethannicholasb3058bd2016-07-01 08:22:01 -07001335 STATEMENT */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001336ASTNode::ID Parser::forStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001337 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001338 if (!this->expect(Token::Kind::TK_FOR, "'for'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001339 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001340 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001341 if (!this->expect(Token::Kind::TK_LPAREN, "'('")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001342 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001343 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001344 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kFor);
1345 ASTNode::ID initializer;
ethannicholasb3058bd2016-07-01 08:22:01 -07001346 Token nextToken = this->peek();
1347 switch (nextToken.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001348 case Token::Kind::TK_SEMICOLON:
ethannicholas22f939e2016-10-13 13:25:34 -07001349 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001350 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001351 break;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001352 case Token::Kind::TK_CONST: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001353 initializer = this->varDeclarations();
1354 if (!initializer) {
1355 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001356 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001357 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001358 break;
ethannicholasa54401d2016-10-14 08:37:32 -07001359 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001360 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001361 if (this->isType(this->text(nextToken))) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001362 initializer = this->varDeclarations();
1363 if (!initializer) {
1364 return ASTNode::ID::Invalid();
ethannicholasa54401d2016-10-14 08:37:32 -07001365 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001366 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001367 break;
1368 }
ethannicholasa54401d2016-10-14 08:37:32 -07001369 } // fall through
ethannicholasb3058bd2016-07-01 08:22:01 -07001370 default:
1371 initializer = this->expressionStatement();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001372 if (!initializer) {
1373 return ASTNode::ID::Invalid();
1374 }
1375 getNode(result).addChild(initializer);
ethannicholasb3058bd2016-07-01 08:22:01 -07001376 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001377 ASTNode::ID test;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001378 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001379 test = this->expression();
1380 if (!test) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001381 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001382 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001383 getNode(result).addChild(test);
1384 } else {
1385 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001386 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001387 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001388 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001389 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001390 ASTNode::ID next;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001391 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07001392 next = this->expression();
1393 if (!next) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001394 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001395 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001396 getNode(result).addChild(next);
1397 } else {
1398 CREATE_EMPTY_CHILD(result);
ethannicholasb3058bd2016-07-01 08:22:01 -07001399 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001400 if (!this->expect(Token::Kind::TK_RPAREN, "')'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001401 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001402 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001403 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001404 if (!statement) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001405 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001406 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001407 getNode(result).addChild(statement);
1408 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001409}
1410
1411/* RETURN expression? SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001412ASTNode::ID Parser::returnStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001413 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001414 if (!this->expect(Token::Kind::TK_RETURN, "'return'", &start)) {
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 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kReturn);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001418 if (this->peek().fKind != Token::Kind::TK_SEMICOLON) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001419 ASTNode::ID expression = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001420 if (!expression) {
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(expression);
ethannicholasb3058bd2016-07-01 08:22:01 -07001424 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001425 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001426 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001427 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001428 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001429}
1430
1431/* BREAK SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001432ASTNode::ID Parser::breakStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001433 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001434 if (!this->expect(Token::Kind::TK_BREAK, "'break'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001435 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001436 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001437 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001438 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001439 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001440 RETURN_NODE(start.fOffset, ASTNode::Kind::kBreak);
ethannicholasb3058bd2016-07-01 08:22:01 -07001441}
1442
1443/* CONTINUE SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001444ASTNode::ID Parser::continueStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001445 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001446 if (!this->expect(Token::Kind::TK_CONTINUE, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001447 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001448 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001449 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001450 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001451 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001452 RETURN_NODE(start.fOffset, ASTNode::Kind::kContinue);
ethannicholasb3058bd2016-07-01 08:22:01 -07001453}
1454
1455/* DISCARD SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001456ASTNode::ID Parser::discardStatement() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001457 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001458 if (!this->expect(Token::Kind::TK_DISCARD, "'continue'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001459 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001460 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001461 if (!this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001462 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001463 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001464 RETURN_NODE(start.fOffset, ASTNode::Kind::kDiscard);
ethannicholasb3058bd2016-07-01 08:22:01 -07001465}
1466
1467/* LBRACE statement* RBRACE */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001468ASTNode::ID Parser::block() {
ethannicholasb3058bd2016-07-01 08:22:01 -07001469 Token start;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001470 if (!this->expect(Token::Kind::TK_LBRACE, "'{'", &start)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001471 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001472 }
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001473 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001474 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001475 return ASTNode::ID::Invalid();
1476 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001477 CREATE_NODE(result, start.fOffset, ASTNode::Kind::kBlock);
ethannicholasb3058bd2016-07-01 08:22:01 -07001478 for (;;) {
1479 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001480 case Token::Kind::TK_RBRACE:
ethannicholasb3058bd2016-07-01 08:22:01 -07001481 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001482 return result;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001483 case Token::Kind::TK_END_OF_FILE:
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07001484 this->error(this->peek(), "expected '}', but found end of file");
Ethan Nicholasfc994162019-06-06 10:04:27 -04001485 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001486 default: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001487 ASTNode::ID statement = this->statement();
ethannicholasb3058bd2016-07-01 08:22:01 -07001488 if (!statement) {
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 getNode(result).addChild(statement);
ethannicholasb3058bd2016-07-01 08:22:01 -07001492 }
1493 }
1494 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001495 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001496}
1497
1498/* expression SEMICOLON */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001499ASTNode::ID Parser::expressionStatement() {
1500 ASTNode::ID expr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001501 if (expr) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001502 if (this->expect(Token::Kind::TK_SEMICOLON, "';'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001503 return expr;
ethannicholasb3058bd2016-07-01 08:22:01 -07001504 }
1505 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001506 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001507}
1508
1509/* assignmentExpression (COMMA assignmentExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001510ASTNode::ID Parser::expression() {
1511 ASTNode::ID result = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001512 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001513 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001514 }
1515 Token t;
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001516 AutoDepth depth(this);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001517 while (this->checkNext(Token::Kind::TK_COMMA, &t)) {
Ethan Nicholasb67d0562020-04-30 16:10:00 -04001518 if (!depth.increase()) {
1519 return ASTNode::ID::Invalid();
1520 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001521 ASTNode::ID right = this->assignmentExpression();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001522 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001523 return ASTNode::ID::Invalid();
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001524 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001525 CREATE_NODE(newResult, t.fOffset, ASTNode::Kind::kBinary, std::move(t));
1526 getNode(newResult).addChild(result);
1527 getNode(newResult).addChild(right);
1528 result = newResult;
Ethan Nicholas4b330df2017-05-17 10:52:55 -04001529 }
1530 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001531}
1532
1533/* ternaryExpression ((EQEQ | STAREQ | SLASHEQ | PERCENTEQ | PLUSEQ | MINUSEQ | SHLEQ | SHREQ |
1534 BITWISEANDEQ | BITWISEXOREQ | BITWISEOREQ | LOGICALANDEQ | LOGICALXOREQ | LOGICALOREQ)
1535 assignmentExpression)*
1536 */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001537ASTNode::ID Parser::assignmentExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001538 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001539 ASTNode::ID result = this->ternaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001540 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001541 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001542 }
1543 for (;;) {
1544 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001545 case Token::Kind::TK_EQ: // fall through
1546 case Token::Kind::TK_STAREQ: // fall through
1547 case Token::Kind::TK_SLASHEQ: // fall through
1548 case Token::Kind::TK_PERCENTEQ: // fall through
1549 case Token::Kind::TK_PLUSEQ: // fall through
1550 case Token::Kind::TK_MINUSEQ: // fall through
1551 case Token::Kind::TK_SHLEQ: // fall through
1552 case Token::Kind::TK_SHREQ: // fall through
1553 case Token::Kind::TK_BITWISEANDEQ: // fall through
1554 case Token::Kind::TK_BITWISEXOREQ: // fall through
1555 case Token::Kind::TK_BITWISEOREQ: // fall through
1556 case Token::Kind::TK_LOGICALANDEQ: // fall through
1557 case Token::Kind::TK_LOGICALXOREQ: // fall through
1558 case Token::Kind::TK_LOGICALOREQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001559 if (!depth.increase()) {
1560 return ASTNode::ID::Invalid();
1561 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001562 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001563 ASTNode::ID right = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001564 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001565 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001566 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001567 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1568 std::move(t));
1569 getNode(newResult).addChild(result);
1570 getNode(newResult).addChild(right);
1571 result = newResult;
1572 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001573 }
1574 default:
1575 return result;
1576 }
1577 }
1578}
1579
1580/* logicalOrExpression ('?' expression ':' assignmentExpression)? */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001581ASTNode::ID Parser::ternaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001582 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001583 ASTNode::ID base = this->logicalOrExpression();
1584 if (!base) {
1585 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001586 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001587 if (this->checkNext(Token::Kind::TK_QUESTION)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001588 if (!depth.increase()) {
1589 return ASTNode::ID::Invalid();
1590 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001591 ASTNode::ID trueExpr = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001592 if (!trueExpr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001593 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001594 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001595 if (this->expect(Token::Kind::TK_COLON, "':'")) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001596 ASTNode::ID falseExpr = this->assignmentExpression();
1597 if (!falseExpr) {
1598 return ASTNode::ID::Invalid();
1599 }
1600 CREATE_NODE(ternary, getNode(base).fOffset, ASTNode::Kind::kTernary);
1601 getNode(ternary).addChild(base);
1602 getNode(ternary).addChild(trueExpr);
1603 getNode(ternary).addChild(falseExpr);
1604 return ternary;
ethannicholasb3058bd2016-07-01 08:22:01 -07001605 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001606 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001607 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001608 return base;
ethannicholasb3058bd2016-07-01 08:22:01 -07001609}
1610
1611/* logicalXorExpression (LOGICALOR logicalXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001612ASTNode::ID Parser::logicalOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001613 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001614 ASTNode::ID result = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001615 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001616 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001617 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001618 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001619 while (this->checkNext(Token::Kind::TK_LOGICALOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001620 if (!depth.increase()) {
1621 return ASTNode::ID::Invalid();
1622 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001623 ASTNode::ID right = this->logicalXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001624 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001625 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001626 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001627 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1628 getNode(newResult).addChild(result);
1629 getNode(newResult).addChild(right);
1630 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001631 }
1632 return result;
1633}
1634
1635/* logicalAndExpression (LOGICALXOR logicalAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001636ASTNode::ID Parser::logicalXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001637 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001638 ASTNode::ID result = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001639 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001640 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001641 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001642 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001643 while (this->checkNext(Token::Kind::TK_LOGICALXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001644 if (!depth.increase()) {
1645 return ASTNode::ID::Invalid();
1646 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001647 ASTNode::ID right = this->logicalAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001648 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001649 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001650 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001651 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1652 getNode(newResult).addChild(result);
1653 getNode(newResult).addChild(right);
1654 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001655 }
1656 return result;
1657}
1658
1659/* bitwiseOrExpression (LOGICALAND bitwiseOrExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001660ASTNode::ID Parser::logicalAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001661 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001662 ASTNode::ID result = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001663 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001664 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001665 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001666 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001667 while (this->checkNext(Token::Kind::TK_LOGICALAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001668 if (!depth.increase()) {
1669 return ASTNode::ID::Invalid();
1670 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001671 ASTNode::ID right = this->bitwiseOrExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001672 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001673 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001674 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001675 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1676 getNode(newResult).addChild(result);
1677 getNode(newResult).addChild(right);
1678 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001679 }
1680 return result;
1681}
1682
1683/* bitwiseXorExpression (BITWISEOR bitwiseXorExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001684ASTNode::ID Parser::bitwiseOrExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001685 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001686 ASTNode::ID result = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001687 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001688 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001689 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001690 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001691 while (this->checkNext(Token::Kind::TK_BITWISEOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001692 if (!depth.increase()) {
1693 return ASTNode::ID::Invalid();
1694 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001695 ASTNode::ID right = this->bitwiseXorExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001696 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001697 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001698 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001699 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1700 getNode(newResult).addChild(result);
1701 getNode(newResult).addChild(right);
1702 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001703 }
1704 return result;
1705}
1706
1707/* bitwiseAndExpression (BITWISEXOR bitwiseAndExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001708ASTNode::ID Parser::bitwiseXorExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001709 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001710 ASTNode::ID result = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001711 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001712 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001713 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001714 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001715 while (this->checkNext(Token::Kind::TK_BITWISEXOR, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001716 if (!depth.increase()) {
1717 return ASTNode::ID::Invalid();
1718 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001719 ASTNode::ID right = this->bitwiseAndExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001720 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001721 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001722 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001723 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1724 getNode(newResult).addChild(result);
1725 getNode(newResult).addChild(right);
1726 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001727 }
1728 return result;
1729}
1730
1731/* equalityExpression (BITWISEAND equalityExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001732ASTNode::ID Parser::bitwiseAndExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001733 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001734 ASTNode::ID result = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001735 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001736 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001737 }
Ethan Nicholas0c9d13b2017-05-08 16:18:19 -04001738 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001739 while (this->checkNext(Token::Kind::TK_BITWISEAND, &t)) {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001740 if (!depth.increase()) {
1741 return ASTNode::ID::Invalid();
1742 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001743 ASTNode::ID right = this->equalityExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001744 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001745 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001746 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001747 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary, std::move(t));
1748 getNode(newResult).addChild(result);
1749 getNode(newResult).addChild(right);
1750 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001751 }
1752 return result;
1753}
1754
1755/* relationalExpression ((EQEQ | NEQ) relationalExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001756ASTNode::ID Parser::equalityExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001757 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001758 ASTNode::ID result = this->relationalExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001759 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001760 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001761 }
1762 for (;;) {
1763 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001764 case Token::Kind::TK_EQEQ: // fall through
1765 case Token::Kind::TK_NEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001766 if (!depth.increase()) {
1767 return ASTNode::ID::Invalid();
1768 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001769 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001770 ASTNode::ID right = this->relationalExpression();
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,
1775 std::move(t));
1776 getNode(newResult).addChild(result);
1777 getNode(newResult).addChild(right);
1778 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001779 break;
1780 }
1781 default:
1782 return result;
1783 }
1784 }
1785}
1786
1787/* shiftExpression ((LT | GT | LTEQ | GTEQ) shiftExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001788ASTNode::ID Parser::relationalExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001789 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001790 ASTNode::ID result = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001791 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001792 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001793 }
1794 for (;;) {
1795 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001796 case Token::Kind::TK_LT: // fall through
1797 case Token::Kind::TK_GT: // fall through
1798 case Token::Kind::TK_LTEQ: // fall through
1799 case Token::Kind::TK_GTEQ: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001800 if (!depth.increase()) {
1801 return ASTNode::ID::Invalid();
1802 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001803 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001804 ASTNode::ID right = this->shiftExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001805 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001806 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001807 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001808 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1809 std::move(t));
1810 getNode(newResult).addChild(result);
1811 getNode(newResult).addChild(right);
1812 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001813 break;
1814 }
1815 default:
1816 return result;
1817 }
1818 }
1819}
1820
1821/* additiveExpression ((SHL | SHR) additiveExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001822ASTNode::ID Parser::shiftExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001823 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001824 ASTNode::ID result = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001825 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001826 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001827 }
1828 for (;;) {
1829 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001830 case Token::Kind::TK_SHL: // fall through
1831 case Token::Kind::TK_SHR: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001832 if (!depth.increase()) {
1833 return ASTNode::ID::Invalid();
1834 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001835 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001836 ASTNode::ID right = this->additiveExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001837 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001838 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001839 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001840 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1841 std::move(t));
1842 getNode(newResult).addChild(result);
1843 getNode(newResult).addChild(right);
1844 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001845 break;
1846 }
1847 default:
1848 return result;
1849 }
1850 }
1851}
1852
1853/* multiplicativeExpression ((PLUS | MINUS) multiplicativeExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001854ASTNode::ID Parser::additiveExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001855 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001856 ASTNode::ID result = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001857 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001858 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001859 }
1860 for (;;) {
1861 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001862 case Token::Kind::TK_PLUS: // fall through
1863 case Token::Kind::TK_MINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001864 if (!depth.increase()) {
1865 return ASTNode::ID::Invalid();
1866 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001867 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001868 ASTNode::ID right = this->multiplicativeExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001869 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001870 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001871 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001872 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1873 std::move(t));
1874 getNode(newResult).addChild(result);
1875 getNode(newResult).addChild(right);
1876 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001877 break;
1878 }
1879 default:
1880 return result;
1881 }
1882 }
1883}
1884
1885/* unaryExpression ((STAR | SLASH | PERCENT) unaryExpression)* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001886ASTNode::ID Parser::multiplicativeExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001887 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001888 ASTNode::ID result = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001889 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001890 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001891 }
1892 for (;;) {
1893 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001894 case Token::Kind::TK_STAR: // fall through
1895 case Token::Kind::TK_SLASH: // fall through
1896 case Token::Kind::TK_PERCENT: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001897 if (!depth.increase()) {
1898 return ASTNode::ID::Invalid();
1899 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001900 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001901 ASTNode::ID right = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001902 if (!right) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001903 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001904 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001905 CREATE_NODE(newResult, getNode(result).fOffset, ASTNode::Kind::kBinary,
1906 std::move(t));
1907 getNode(newResult).addChild(result);
1908 getNode(newResult).addChild(right);
1909 result = newResult;
ethannicholasb3058bd2016-07-01 08:22:01 -07001910 break;
1911 }
1912 default:
1913 return result;
1914 }
1915 }
1916}
1917
1918/* postfixExpression | (PLUS | MINUS | NOT | PLUSPLUS | MINUSMINUS) unaryExpression */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001919ASTNode::ID Parser::unaryExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001920 AutoDepth depth(this);
ethannicholasb3058bd2016-07-01 08:22:01 -07001921 switch (this->peek().fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001922 case Token::Kind::TK_PLUS: // fall through
1923 case Token::Kind::TK_MINUS: // fall through
1924 case Token::Kind::TK_LOGICALNOT: // fall through
1925 case Token::Kind::TK_BITWISENOT: // fall through
1926 case Token::Kind::TK_PLUSPLUS: // fall through
1927 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001928 if (!depth.increase()) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001929 return ASTNode::ID::Invalid();
Ethan Nicholas6dcc3252019-02-20 15:18:36 -05001930 }
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001931 Token t = this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04001932 ASTNode::ID expr = this->unaryExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001933 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001934 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001935 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001936 CREATE_NODE(result, t.fOffset, ASTNode::Kind::kPrefix, std::move(t));
1937 getNode(result).addChild(expr);
1938 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07001939 }
1940 default:
1941 return this->postfixExpression();
1942 }
1943}
1944
1945/* term suffix* */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001946ASTNode::ID Parser::postfixExpression() {
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001947 AutoDepth depth(this);
Ethan Nicholasfc994162019-06-06 10:04:27 -04001948 ASTNode::ID result = this->term();
ethannicholasb3058bd2016-07-01 08:22:01 -07001949 if (!result) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001950 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001951 }
1952 for (;;) {
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001953 Token t = this->peek();
1954 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001955 case Token::Kind::TK_FLOAT_LITERAL:
Ethan Nicholas5a9a0b32019-09-17 16:18:22 -04001956 if (this->text(t)[0] != '.') {
1957 return result;
1958 }
1959 // fall through
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001960 case Token::Kind::TK_LBRACKET:
1961 case Token::Kind::TK_DOT:
1962 case Token::Kind::TK_LPAREN:
1963 case Token::Kind::TK_PLUSPLUS:
1964 case Token::Kind::TK_MINUSMINUS:
1965 case Token::Kind::TK_COLONCOLON:
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001966 if (!depth.increase()) {
1967 return ASTNode::ID::Invalid();
1968 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001969 result = this->suffix(result);
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001970 if (!result) {
1971 return ASTNode::ID::Invalid();
1972 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001973 break;
ethannicholasb3058bd2016-07-01 08:22:01 -07001974 default:
1975 return result;
1976 }
1977 }
1978}
1979
Ethan Nicholas11d53972016-11-28 11:23:23 -05001980/* LBRACKET expression? RBRACKET | DOT IDENTIFIER | LPAREN parameters RPAREN |
Ethan Nicholase455f652019-09-13 12:52:55 -04001981 PLUSPLUS | MINUSMINUS | COLONCOLON IDENTIFIER | FLOAT_LITERAL [IDENTIFIER] */
Ethan Nicholasfc994162019-06-06 10:04:27 -04001982ASTNode::ID Parser::suffix(ASTNode::ID base) {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04001983 SkASSERT(base);
ethannicholasb3058bd2016-07-01 08:22:01 -07001984 Token next = this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001985 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04001986 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04001987 return ASTNode::ID::Invalid();
1988 }
ethannicholasb3058bd2016-07-01 08:22:01 -07001989 switch (next.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04001990 case Token::Kind::TK_LBRACKET: {
1991 if (this->checkNext(Token::Kind::TK_RBRACKET)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001992 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
1993 getNode(result).addChild(base);
1994 return result;
ethannicholas5961bc92016-10-12 06:39:56 -07001995 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04001996 ASTNode::ID e = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07001997 if (!e) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04001998 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07001999 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002000 this->expect(Token::Kind::TK_RBRACKET, "']' to complete array access expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002001 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kIndex);
2002 getNode(result).addChild(base);
2003 getNode(result).addChild(e);
2004 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002005 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002006 case Token::Kind::TK_DOT: // fall through
2007 case Token::Kind::TK_COLONCOLON: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002008 int offset = this->peek().fOffset;
2009 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002010 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002011 CREATE_NODE(result, offset, ASTNode::Kind::kField, std::move(text));
2012 getNode(result).addChild(base);
2013 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002014 }
Ethan Nicholase455f652019-09-13 12:52:55 -04002015 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002016 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholase455f652019-09-13 12:52:55 -04002017 // Swizzles that start with a constant number, e.g. '.000r', will be tokenized as
2018 // floating point literals, possibly followed by an identifier. Handle that here.
2019 StringFragment field = this->text(next);
2020 SkASSERT(field.fChars[0] == '.');
2021 ++field.fChars;
2022 --field.fLength;
2023 for (size_t i = 0; i < field.fLength; ++i) {
2024 if (field.fChars[i] != '0' && field.fChars[i] != '1') {
2025 this->error(next, "invalid swizzle");
2026 return ASTNode::ID::Invalid();
2027 }
2028 }
2029 // use the next *raw* token so we don't ignore whitespace - we only care about
2030 // identifiers that directly follow the float
2031 Token id = this->nextRawToken();
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002032 if (id.fKind == Token::Kind::TK_IDENTIFIER) {
Ethan Nicholase455f652019-09-13 12:52:55 -04002033 field.fLength += id.fLength;
2034 } else {
2035 this->pushback(id);
2036 }
2037 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kField, field);
2038 getNode(result).addChild(base);
2039 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002040 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002041 case Token::Kind::TK_LPAREN: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002042 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kCall);
2043 getNode(result).addChild(base);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002044 if (this->peek().fKind != Token::Kind::TK_RPAREN) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002045 for (;;) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002046 ASTNode::ID expr = this->assignmentExpression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002047 if (!expr) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002048 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002049 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002050 getNode(result).addChild(expr);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002051 if (!this->checkNext(Token::Kind::TK_COMMA)) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002052 break;
2053 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002054 }
2055 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002056 this->expect(Token::Kind::TK_RPAREN, "')' to complete function parameters");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002057 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002058 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002059 case Token::Kind::TK_PLUSPLUS: // fall through
2060 case Token::Kind::TK_MINUSMINUS: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002061 CREATE_NODE(result, next.fOffset, ASTNode::Kind::kPostfix, next);
2062 getNode(result).addChild(base);
2063 return result;
2064 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002065 default: {
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002066 this->error(next, "expected expression suffix, but found '" + this->text(next) + "'");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002067 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002068 }
2069 }
2070}
2071
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002072/* IDENTIFIER | intLiteral | floatLiteral | boolLiteral | NULL_LITERAL | '(' expression ')' */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002073ASTNode::ID Parser::term() {
ethannicholasb3058bd2016-07-01 08:22:01 -07002074 Token t = this->peek();
2075 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002076 case Token::Kind::TK_IDENTIFIER: {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002077 StringFragment text;
ethannicholasb3058bd2016-07-01 08:22:01 -07002078 if (this->identifier(&text)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002079 RETURN_NODE(t.fOffset, ASTNode::Kind::kIdentifier, std::move(text));
ethannicholasb3058bd2016-07-01 08:22:01 -07002080 }
ethannicholasb3058bd2016-07-01 08:22:01 -07002081 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002082 case Token::Kind::TK_INT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002083 SKSL_INT i;
ethannicholasb3058bd2016-07-01 08:22:01 -07002084 if (this->intLiteral(&i)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002085 RETURN_NODE(t.fOffset, ASTNode::Kind::kInt, i);
ethannicholasb3058bd2016-07-01 08:22:01 -07002086 }
2087 break;
2088 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002089 case Token::Kind::TK_FLOAT_LITERAL: {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002090 SKSL_FLOAT f;
ethannicholasb3058bd2016-07-01 08:22:01 -07002091 if (this->floatLiteral(&f)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002092 RETURN_NODE(t.fOffset, ASTNode::Kind::kFloat, f);
ethannicholasb3058bd2016-07-01 08:22:01 -07002093 }
2094 break;
2095 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002096 case Token::Kind::TK_TRUE_LITERAL: // fall through
2097 case Token::Kind::TK_FALSE_LITERAL: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002098 bool b;
2099 if (this->boolLiteral(&b)) {
Ethan Nicholasfc994162019-06-06 10:04:27 -04002100 RETURN_NODE(t.fOffset, ASTNode::Kind::kBool, b);
ethannicholasb3058bd2016-07-01 08:22:01 -07002101 }
2102 break;
2103 }
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002104 case Token::Kind::TK_NULL_LITERAL:
Ethan Nicholasee1c8a72019-02-22 10:50:47 -05002105 this->nextToken();
Ethan Nicholasfc994162019-06-06 10:04:27 -04002106 RETURN_NODE(t.fOffset, ASTNode::Kind::kNull);
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002107 case Token::Kind::TK_LPAREN: {
ethannicholasb3058bd2016-07-01 08:22:01 -07002108 this->nextToken();
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002109 AutoDepth depth(this);
Ethan Nicholascf4deab2019-09-13 16:28:14 -04002110 if (!depth.increase()) {
Ethan Nicholas4e3b0112019-06-07 16:49:07 -04002111 return ASTNode::ID::Invalid();
2112 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002113 ASTNode::ID result = this->expression();
ethannicholasb3058bd2016-07-01 08:22:01 -07002114 if (result) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002115 this->expect(Token::Kind::TK_RPAREN, "')' to complete expression");
Ethan Nicholasfc994162019-06-06 10:04:27 -04002116 return result;
ethannicholasb3058bd2016-07-01 08:22:01 -07002117 }
2118 break;
2119 }
2120 default:
2121 this->nextToken();
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002122 this->error(t.fOffset, "expected expression, but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002123 }
Ethan Nicholasfc994162019-06-06 10:04:27 -04002124 return ASTNode::ID::Invalid();
ethannicholasb3058bd2016-07-01 08:22:01 -07002125}
2126
2127/* INT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002128bool Parser::intLiteral(SKSL_INT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002129 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002130 if (this->expect(Token::Kind::TK_INT_LITERAL, "integer literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002131 *dest = SkSL::stol(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002132 return true;
2133 }
2134 return false;
2135}
2136
2137/* FLOAT_LITERAL */
Ethan Nicholasfc994162019-06-06 10:04:27 -04002138bool Parser::floatLiteral(SKSL_FLOAT* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002139 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002140 if (this->expect(Token::Kind::TK_FLOAT_LITERAL, "float literal", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002141 *dest = SkSL::stod(this->text(t));
ethannicholasb3058bd2016-07-01 08:22:01 -07002142 return true;
2143 }
2144 return false;
2145}
2146
2147/* TRUE_LITERAL | FALSE_LITERAL */
2148bool Parser::boolLiteral(bool* dest) {
2149 Token t = this->nextToken();
2150 switch (t.fKind) {
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002151 case Token::Kind::TK_TRUE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002152 *dest = true;
2153 return true;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002154 case Token::Kind::TK_FALSE_LITERAL:
ethannicholasb3058bd2016-07-01 08:22:01 -07002155 *dest = false;
2156 return true;
2157 default:
Ethan Nicholas0c8582e2019-07-19 09:26:46 -04002158 this->error(t, "expected 'true' or 'false', but found '" + this->text(t) + "'");
ethannicholasb3058bd2016-07-01 08:22:01 -07002159 return false;
2160 }
2161}
2162
2163/* IDENTIFIER */
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002164bool Parser::identifier(StringFragment* dest) {
ethannicholasb3058bd2016-07-01 08:22:01 -07002165 Token t;
Ethan Nicholas5a9e7fb2020-04-17 12:45:51 -04002166 if (this->expect(Token::Kind::TK_IDENTIFIER, "identifier", &t)) {
Ethan Nicholas5b5f0962017-09-11 13:50:14 -07002167 *dest = this->text(t);
ethannicholasb3058bd2016-07-01 08:22:01 -07002168 return true;
2169 }
2170 return false;
2171}
2172
2173} // namespace