5 * Permission is hereby granted, free of charge, to any person obtaining a copy of
6 * this software and associated documentation files (the "Software"), to deal in
7 * the Software without restriction, including without limitation the rights to
8 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
9 * of the Software, and to permit persons to whom the Software is furnished to do
10 * so, subject to the following conditions:
12 * The above copyright notice and this permission notice shall be included in all
13 * copies or substantial portions of the Software.
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
29 #define PARSER_HT_FIELDS 0
30 #define PARSER_HT_GLOBALS 1
31 /* beginning of locals */
32 #define PARSER_HT_LOCALS 2
34 #define PARSER_HT_SIZE 1024
35 #define TYPEDEF_HT_SIZE 16
41 ast_expression **globals;
42 ast_expression **fields;
43 ast_function **functions;
44 ast_value **imm_float;
45 ast_value **imm_string;
46 ast_value **imm_vector;
49 /* must be deleted first, they reference immediates and values */
50 ast_value **accessors;
52 ast_value *imm_float_zero;
53 ast_value *imm_float_one;
54 ast_value *imm_vector_zero;
59 ast_function *function;
61 /* A list of hashtables for each scope */
67 /* not to be used directly, we use the hash table */
68 ast_expression **_locals;
70 ast_value **_typedefs;
71 size_t *_blocktypedefs;
75 /* we store the '=' operator info */
76 const oper_info *assign_op;
78 /* TYPE_FIELD -> parser_find_fields is used instead of find_var
79 * TODO: TYPE_VECTOR -> x, y and z are accepted in the gmqcc standard
80 * anything else: type error
85 static void parser_enterblock(parser_t *parser);
86 static bool parser_leaveblock(parser_t *parser);
87 static void parser_addlocal(parser_t *parser, const char *name, ast_expression *e);
88 static bool parse_typedef(parser_t *parser);
89 static bool parse_variable(parser_t *parser, ast_block *localblock, bool nofields, bool is_const, ast_value *cached_typedef);
90 static ast_block* parse_block(parser_t *parser, bool warnreturn);
91 static bool parse_block_into(parser_t *parser, ast_block *block, bool warnreturn);
92 static ast_expression* parse_statement_or_block(parser_t *parser);
93 static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases);
94 static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma);
95 static ast_expression* parse_expression(parser_t *parser, bool stopatcomma);
97 static void parseerror(parser_t *parser, const char *fmt, ...)
104 con_vprintmsg(LVL_ERROR, parser->lex->tok.ctx.file, parser->lex->tok.ctx.line, "parse error", fmt, ap);
108 /* returns true if it counts as an error */
109 static bool GMQCC_WARN parsewarning(parser_t *parser, int warntype, const char *fmt, ...)
112 int lvl = LVL_WARNING;
114 if (!OPTS_WARN(warntype))
123 con_vprintmsg(lvl, parser->lex->tok.ctx.file, parser->lex->tok.ctx.line, "warning", fmt, ap);
129 static bool GMQCC_WARN genwarning(lex_ctx ctx, int warntype, const char *fmt, ...)
132 int lvl = LVL_WARNING;
134 if (!OPTS_WARN(warntype))
141 con_vprintmsg(lvl, ctx.file, ctx.line, "warning", fmt, ap);
147 /**********************************************************************
148 * some maths used for constant folding
151 vector vec3_add(vector a, vector b)
160 vector vec3_sub(vector a, vector b)
169 qcfloat vec3_mulvv(vector a, vector b)
171 return (a.x * b.x + a.y * b.y + a.z * b.z);
174 vector vec3_mulvf(vector a, float b)
183 /**********************************************************************
187 bool parser_next(parser_t *parser)
189 /* lex_do kills the previous token */
190 parser->tok = lex_do(parser->lex);
191 if (parser->tok == TOKEN_EOF)
193 if (parser->tok >= TOKEN_ERROR) {
194 parseerror(parser, "lex error");
200 #define parser_tokval(p) ((p)->lex->tok.value)
201 #define parser_token(p) (&((p)->lex->tok))
202 #define parser_ctx(p) ((p)->lex->tok.ctx)
204 static ast_value* parser_const_float(parser_t *parser, double d)
208 for (i = 0; i < vec_size(parser->imm_float); ++i) {
209 const double compare = parser->imm_float[i]->constval.vfloat;
210 if (memcmp((const void*)&compare, (const void *)&d, sizeof(double)) == 0)
211 return parser->imm_float[i];
213 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_FLOAT);
214 out->constant = true;
215 out->hasvalue = true;
216 out->constval.vfloat = d;
217 vec_push(parser->imm_float, out);
221 static ast_value* parser_const_float_0(parser_t *parser)
223 if (!parser->imm_float_zero)
224 parser->imm_float_zero = parser_const_float(parser, 0);
225 return parser->imm_float_zero;
228 static ast_value* parser_const_float_1(parser_t *parser)
230 if (!parser->imm_float_one)
231 parser->imm_float_one = parser_const_float(parser, 1);
232 return parser->imm_float_one;
235 static char *parser_strdup(const char *str)
238 /* actually dup empty strings */
239 char *out = mem_a(1);
243 return util_strdup(str);
246 static ast_value* parser_const_string(parser_t *parser, const char *str, bool dotranslate)
250 for (i = 0; i < vec_size(parser->imm_string); ++i) {
251 if (!strcmp(parser->imm_string[i]->constval.vstring, str))
252 return parser->imm_string[i];
256 snprintf(name, sizeof(name), "dotranslate_%lu", (unsigned long)(parser->translated++));
257 out = ast_value_new(parser_ctx(parser), name, TYPE_STRING);
259 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_STRING);
260 out->constant = true;
261 out->hasvalue = true;
262 out->constval.vstring = parser_strdup(str);
263 vec_push(parser->imm_string, out);
267 static ast_value* parser_const_vector(parser_t *parser, vector v)
271 for (i = 0; i < vec_size(parser->imm_vector); ++i) {
272 if (!memcmp(&parser->imm_vector[i]->constval.vvec, &v, sizeof(v)))
273 return parser->imm_vector[i];
275 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_VECTOR);
276 out->constant = true;
277 out->hasvalue = true;
278 out->constval.vvec = v;
279 vec_push(parser->imm_vector, out);
283 static ast_value* parser_const_vector_f(parser_t *parser, float x, float y, float z)
289 return parser_const_vector(parser, v);
292 static ast_value* parser_const_vector_0(parser_t *parser)
294 if (!parser->imm_vector_zero)
295 parser->imm_vector_zero = parser_const_vector_f(parser, 0, 0, 0);
296 return parser->imm_vector_zero;
299 static ast_expression* parser_find_field(parser_t *parser, const char *name)
301 return util_htget(parser->htfields, name);
304 static ast_expression* parser_find_global(parser_t *parser, const char *name)
306 return util_htget(parser->htglobals, name);
309 static ast_expression* parser_find_param(parser_t *parser, const char *name)
313 if (!parser->function)
315 fun = parser->function->vtype;
316 for (i = 0; i < vec_size(fun->expression.params); ++i) {
317 if (!strcmp(fun->expression.params[i]->name, name))
318 return (ast_expression*)(fun->expression.params[i]);
323 static ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto, bool *isparam)
328 hash = util_hthash(parser->htglobals, name);
331 for (i = vec_size(parser->variables); i > upto;) {
333 if ( (e = util_htgeth(parser->variables[i], name, hash)) )
337 return parser_find_param(parser, name);
340 static ast_expression* parser_find_var(parser_t *parser, const char *name)
344 v = parser_find_local(parser, name, 0, &dummy);
345 if (!v) v = parser_find_global(parser, name);
349 static ast_value* parser_find_typedef(parser_t *parser, const char *name, size_t upto)
353 hash = util_hthash(parser->typedefs[0], name);
355 for (i = vec_size(parser->typedefs); i > upto;) {
357 if ( (e = (ast_value*)util_htgeth(parser->typedefs[i], name, hash)) )
365 size_t etype; /* 0 = expression, others are operators */
369 ast_block *block; /* for commas and function calls */
378 #define SY_PAREN_EXPR '('
379 #define SY_PAREN_FUNC 'f'
380 #define SY_PAREN_INDEX '['
382 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
393 static sy_elem syblock(lex_ctx ctx, ast_block *v) {
397 e.out = (ast_expression*)v;
404 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
406 e.etype = 1 + (op - operators);
415 static sy_elem syparen(lex_ctx ctx, int p, size_t off) {
427 # define DEBUGSHUNTDO(x) x
429 # define DEBUGSHUNTDO(x)
432 /* With regular precedence rules, ent.foo[n] is the same as (ent.foo)[n],
433 * so we need to rotate it to become ent.(foo[n]).
435 static bool rotate_entfield_array_index_nodes(ast_expression **out)
437 ast_array_index *index;
438 ast_entfield *entfield;
442 ast_expression *entity;
444 lex_ctx ctx = ast_ctx(*out);
446 if (!ast_istype(*out, ast_array_index))
448 index = (ast_array_index*)*out;
450 if (!ast_istype(index->array, ast_entfield))
452 entfield = (ast_entfield*)index->array;
454 if (!ast_istype(entfield->field, ast_value))
456 field = (ast_value*)entfield->field;
459 entity = entfield->entity;
463 index = ast_array_index_new(ctx, (ast_expression*)field, sub);
464 entfield = ast_entfield_new(ctx, entity, (ast_expression*)index);
465 *out = (ast_expression*)entfield;
470 static bool parser_sy_pop(parser_t *parser, shunt *sy)
474 ast_expression *out = NULL;
475 ast_expression *exprs[3];
476 ast_block *blocks[3];
477 ast_value *asvalue[3];
478 size_t i, assignop, addop, subop;
479 qcint generated_op = 0;
484 if (!vec_size(sy->ops)) {
485 parseerror(parser, "internal error: missing operator");
489 if (vec_last(sy->ops).paren) {
490 parseerror(parser, "unmatched parenthesis");
494 op = &operators[vec_last(sy->ops).etype - 1];
495 ctx = vec_last(sy->ops).ctx;
497 DEBUGSHUNTDO(con_out("apply %s\n", op->op));
499 if (vec_size(sy->out) < op->operands) {
500 parseerror(parser, "internal error: not enough operands: %i (operator %s (%i))", vec_size(sy->out),
501 op->op, (int)op->id);
505 vec_shrinkby(sy->ops, 1);
507 vec_shrinkby(sy->out, op->operands);
508 for (i = 0; i < op->operands; ++i) {
509 exprs[i] = sy->out[vec_size(sy->out)+i].out;
510 blocks[i] = sy->out[vec_size(sy->out)+i].block;
511 asvalue[i] = (ast_value*)exprs[i];
514 if (blocks[0] && !vec_size(blocks[0]->exprs) && op->id != opid1(',')) {
515 parseerror(parser, "internal error: operator cannot be applied on empty blocks");
519 #define NotSameType(T) \
520 (exprs[0]->expression.vtype != exprs[1]->expression.vtype || \
521 exprs[0]->expression.vtype != T)
522 #define CanConstFold1(A) \
523 (ast_istype((A), ast_value) && ((ast_value*)(A))->hasvalue && ((ast_value*)(A))->constant)
524 #define CanConstFold(A, B) \
525 (CanConstFold1(A) && CanConstFold1(B))
526 #define ConstV(i) (asvalue[(i)]->constval.vvec)
527 #define ConstF(i) (asvalue[(i)]->constval.vfloat)
528 #define ConstS(i) (asvalue[(i)]->constval.vstring)
532 parseerror(parser, "internal error: unhandled operator: %s (%i)", op->op, (int)op->id);
536 if (exprs[0]->expression.vtype == TYPE_ENTITY) {
537 if (exprs[1]->expression.vtype != TYPE_FIELD) {
538 parseerror(parser, "type error: right hand of member-operand should be an entity-field");
541 out = (ast_expression*)ast_entfield_new(ctx, exprs[0], exprs[1]);
543 else if (exprs[0]->expression.vtype == TYPE_VECTOR) {
544 parseerror(parser, "internal error: vector access is not supposed to be handled at this point");
548 parseerror(parser, "type error: member-of operator on something that is not an entity or vector");
554 if (exprs[0]->expression.vtype != TYPE_ARRAY &&
555 !(exprs[0]->expression.vtype == TYPE_FIELD &&
556 exprs[0]->expression.next->expression.vtype == TYPE_ARRAY))
558 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
559 parseerror(parser, "cannot index value of type %s", ty1);
562 if (exprs[1]->expression.vtype != TYPE_FLOAT) {
563 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
564 parseerror(parser, "index must be of type float, not %s", ty1);
567 out = (ast_expression*)ast_array_index_new(ctx, exprs[0], exprs[1]);
568 if (rotate_entfield_array_index_nodes(&out))
570 if (opts_standard != COMPILER_GMQCC) {
571 /* this error doesn't need to make us bail out */
572 (void)!parsewarning(parser, WARN_EXTENSIONS,
573 "accessing array-field members of an entity without parenthesis\n"
574 " -> this is an extension from -std=gmqcc");
581 ast_block_add_expr(blocks[0], exprs[1]);
583 blocks[0] = ast_block_new(ctx);
584 ast_block_add_expr(blocks[0], exprs[0]);
585 ast_block_add_expr(blocks[0], exprs[1]);
587 if (!ast_block_set_type(blocks[0], exprs[1]))
590 vec_push(sy->out, syblock(ctx, blocks[0]));
594 switch (exprs[0]->expression.vtype) {
596 if (CanConstFold1(exprs[0]))
597 out = (ast_expression*)parser_const_float(parser, -ConstF(0));
599 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F,
600 (ast_expression*)parser_const_float_0(parser),
604 if (CanConstFold1(exprs[0]))
605 out = (ast_expression*)parser_const_vector_f(parser,
606 -ConstV(0).x, -ConstV(0).y, -ConstV(0).z);
608 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V,
609 (ast_expression*)parser_const_vector_0(parser),
613 parseerror(parser, "invalid types used in expression: cannot negate type %s",
614 type_name[exprs[0]->expression.vtype]);
620 switch (exprs[0]->expression.vtype) {
622 if (CanConstFold1(exprs[0]))
623 out = (ast_expression*)parser_const_float(parser, !ConstF(0));
625 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, exprs[0]);
628 if (CanConstFold1(exprs[0]))
629 out = (ast_expression*)parser_const_float(parser,
630 (!ConstV(0).x && !ConstV(0).y && !ConstV(0).z));
632 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_V, exprs[0]);
635 if (CanConstFold1(exprs[0]))
636 out = (ast_expression*)parser_const_float(parser, !ConstS(0) || !*ConstS(0));
638 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_S, exprs[0]);
640 /* we don't constant-fold NOT for these types */
642 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_ENT, exprs[0]);
645 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_FNC, exprs[0]);
648 parseerror(parser, "invalid types used in expression: cannot logically negate type %s",
649 type_name[exprs[0]->expression.vtype]);
655 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
656 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
658 parseerror(parser, "invalid types used in expression: cannot add type %s and %s",
659 type_name[exprs[0]->expression.vtype],
660 type_name[exprs[1]->expression.vtype]);
663 switch (exprs[0]->expression.vtype) {
665 if (CanConstFold(exprs[0], exprs[1]))
667 out = (ast_expression*)parser_const_float(parser, ConstF(0) + ConstF(1));
670 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
673 if (CanConstFold(exprs[0], exprs[1]))
674 out = (ast_expression*)parser_const_vector(parser, vec3_add(ConstV(0), ConstV(1)));
676 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
679 parseerror(parser, "invalid types used in expression: cannot add type %s and %s",
680 type_name[exprs[0]->expression.vtype],
681 type_name[exprs[1]->expression.vtype]);
686 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
687 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
689 parseerror(parser, "invalid types used in expression: cannot subtract type %s from %s",
690 type_name[exprs[1]->expression.vtype],
691 type_name[exprs[0]->expression.vtype]);
694 switch (exprs[0]->expression.vtype) {
696 if (CanConstFold(exprs[0], exprs[1]))
697 out = (ast_expression*)parser_const_float(parser, ConstF(0) - ConstF(1));
699 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
702 if (CanConstFold(exprs[0], exprs[1]))
703 out = (ast_expression*)parser_const_vector(parser, vec3_sub(ConstV(0), ConstV(1)));
705 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
708 parseerror(parser, "invalid types used in expression: cannot subtract type %s from %s",
709 type_name[exprs[1]->expression.vtype],
710 type_name[exprs[0]->expression.vtype]);
715 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
716 exprs[0]->expression.vtype != TYPE_VECTOR &&
717 exprs[0]->expression.vtype != TYPE_FLOAT &&
718 exprs[1]->expression.vtype != TYPE_VECTOR &&
719 exprs[1]->expression.vtype != TYPE_FLOAT)
721 parseerror(parser, "invalid types used in expression: cannot multiply types %s and %s",
722 type_name[exprs[1]->expression.vtype],
723 type_name[exprs[0]->expression.vtype]);
726 switch (exprs[0]->expression.vtype) {
728 if (exprs[1]->expression.vtype == TYPE_VECTOR)
730 if (CanConstFold(exprs[0], exprs[1]))
731 out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(1), ConstF(0)));
733 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
737 if (CanConstFold(exprs[0], exprs[1]))
738 out = (ast_expression*)parser_const_float(parser, ConstF(0) * ConstF(1));
740 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
744 if (exprs[1]->expression.vtype == TYPE_FLOAT)
746 if (CanConstFold(exprs[0], exprs[1]))
747 out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(0), ConstF(1)));
749 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
753 if (CanConstFold(exprs[0], exprs[1]))
754 out = (ast_expression*)parser_const_float(parser, vec3_mulvv(ConstV(0), ConstV(1)));
756 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
760 parseerror(parser, "invalid types used in expression: cannot multiply types %s and %s",
761 type_name[exprs[1]->expression.vtype],
762 type_name[exprs[0]->expression.vtype]);
767 if (NotSameType(TYPE_FLOAT)) {
768 parseerror(parser, "invalid types used in expression: cannot divide types %s and %s",
769 type_name[exprs[0]->expression.vtype],
770 type_name[exprs[1]->expression.vtype]);
773 if (CanConstFold(exprs[0], exprs[1]))
774 out = (ast_expression*)parser_const_float(parser, ConstF(0) / ConstF(1));
776 out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
780 parseerror(parser, "qc does not have a modulo operator");
784 if (NotSameType(TYPE_FLOAT)) {
785 parseerror(parser, "invalid types used in expression: cannot perform bit operations between types %s and %s",
786 type_name[exprs[0]->expression.vtype],
787 type_name[exprs[1]->expression.vtype]);
790 if (CanConstFold(exprs[0], exprs[1]))
791 out = (ast_expression*)parser_const_float(parser,
792 (op->id == opid1('|') ? (float)( ((qcint)ConstF(0)) | ((qcint)ConstF(1)) ) :
793 (float)( ((qcint)ConstF(0)) & ((qcint)ConstF(1)) ) ));
795 out = (ast_expression*)ast_binary_new(ctx,
796 (op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
800 parseerror(parser, "TODO: bitxor");
805 case opid3('<','<','='):
806 case opid3('>','>','='):
807 parseerror(parser, "TODO: shifts");
811 generated_op += 1; /* INSTR_OR */
813 generated_op += INSTR_AND;
815 if (NotSameType(TYPE_FLOAT)) {
816 parseerror(parser, "invalid types used in expression: cannot perform logical operations between types %s and %s",
817 type_name[exprs[0]->expression.vtype],
818 type_name[exprs[1]->expression.vtype]);
819 parseerror(parser, "TODO: logical ops for arbitrary types using INSTR_NOT");
820 parseerror(parser, "TODO: optional early out");
824 if (opts_standard == COMPILER_GMQCC)
825 con_out("TODO: early out logic\n");
826 if (CanConstFold(exprs[0], exprs[1]))
827 out = (ast_expression*)parser_const_float(parser,
828 (generated_op == INSTR_OR ? (ConstF(0) || ConstF(1)) : (ConstF(0) && ConstF(1))));
830 out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
834 if (exprs[1]->expression.vtype != exprs[2]->expression.vtype) {
835 ast_type_to_string(exprs[1], ty1, sizeof(ty1));
836 ast_type_to_string(exprs[2], ty2, sizeof(ty2));
837 parseerror(parser, "iperands of ternary expression must have the same type, got %s and %s", ty1, ty2);
840 if (CanConstFold1(exprs[0]))
841 out = (ConstF(0) ? exprs[1] : exprs[2]);
843 out = (ast_expression*)ast_ternary_new(ctx, exprs[0], exprs[1], exprs[2]);
847 generated_op += 1; /* INSTR_GT */
849 generated_op += 1; /* INSTR_LT */
850 case opid2('>', '='):
851 generated_op += 1; /* INSTR_GE */
852 case opid2('<', '='):
853 generated_op += INSTR_LE;
854 if (NotSameType(TYPE_FLOAT)) {
855 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
856 type_name[exprs[0]->expression.vtype],
857 type_name[exprs[1]->expression.vtype]);
860 out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
862 case opid2('!', '='):
863 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
864 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
865 type_name[exprs[0]->expression.vtype],
866 type_name[exprs[1]->expression.vtype]);
869 out = (ast_expression*)ast_binary_new(ctx, type_ne_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
871 case opid2('=', '='):
872 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
873 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
874 type_name[exprs[0]->expression.vtype],
875 type_name[exprs[1]->expression.vtype]);
878 out = (ast_expression*)ast_binary_new(ctx, type_eq_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
882 if (ast_istype(exprs[0], ast_entfield)) {
883 ast_expression *field = ((ast_entfield*)exprs[0])->field;
884 if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) &&
885 exprs[0]->expression.vtype == TYPE_FIELD &&
886 exprs[0]->expression.next->expression.vtype == TYPE_VECTOR)
888 assignop = type_storep_instr[TYPE_VECTOR];
891 assignop = type_storep_instr[exprs[0]->expression.vtype];
892 if (!ast_compare_type(field->expression.next, exprs[1])) {
893 ast_type_to_string(field->expression.next, ty1, sizeof(ty1));
894 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
895 if (opts_standard == COMPILER_QCC &&
896 field->expression.next->expression.vtype == TYPE_FUNCTION &&
897 exprs[1]->expression.vtype == TYPE_FUNCTION)
899 if (parsewarning(parser, WARN_ASSIGN_FUNCTION_TYPES,
900 "invalid types in assignment: cannot assign %s to %s", ty2, ty1))
906 parseerror(parser, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
911 if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) &&
912 exprs[0]->expression.vtype == TYPE_FIELD &&
913 exprs[0]->expression.next->expression.vtype == TYPE_VECTOR)
915 assignop = type_store_instr[TYPE_VECTOR];
918 assignop = type_store_instr[exprs[0]->expression.vtype];
921 if (assignop == AINSTR_END) {
922 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
923 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
924 parseerror(parser, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
926 else if (!ast_compare_type(exprs[0], exprs[1])) {
927 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
928 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
929 if (opts_standard == COMPILER_QCC &&
930 exprs[0]->expression.vtype == TYPE_FUNCTION &&
931 exprs[1]->expression.vtype == TYPE_FUNCTION)
933 if (parsewarning(parser, WARN_ASSIGN_FUNCTION_TYPES,
934 "invalid types in assignment: cannot assign %s to %s", ty2, ty1))
940 parseerror(parser, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
943 out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]);
945 case opid3('+','+','P'):
946 case opid3('-','-','P'):
948 if (exprs[0]->expression.vtype != TYPE_FLOAT) {
949 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
950 parseerror(parser, "invalid type for prefix increment: %s", ty1);
953 if (op->id == opid3('+','+','P'))
957 if (ast_istype(exprs[0], ast_entfield)) {
958 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop,
960 (ast_expression*)parser_const_float_1(parser));
962 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop,
964 (ast_expression*)parser_const_float_1(parser));
967 case opid3('S','+','+'):
968 case opid3('S','-','-'):
970 if (exprs[0]->expression.vtype != TYPE_FLOAT) {
971 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
972 parseerror(parser, "invalid type for suffix increment: %s", ty1);
975 if (op->id == opid3('S','+','+')) {
982 if (ast_istype(exprs[0], ast_entfield)) {
983 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop,
985 (ast_expression*)parser_const_float_1(parser));
987 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop,
989 (ast_expression*)parser_const_float_1(parser));
993 out = (ast_expression*)ast_binary_new(ctx, subop,
995 (ast_expression*)parser_const_float_1(parser));
999 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
1000 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
1002 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1003 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1004 parseerror(parser, "invalid types used in expression: cannot add or subtract type %s and %s",
1008 if (ast_istype(exprs[0], ast_entfield))
1009 assignop = type_storep_instr[exprs[0]->expression.vtype];
1011 assignop = type_store_instr[exprs[0]->expression.vtype];
1012 switch (exprs[0]->expression.vtype) {
1014 out = (ast_expression*)ast_binstore_new(ctx, assignop,
1015 (op->id == opid2('+','=') ? INSTR_ADD_F : INSTR_SUB_F),
1016 exprs[0], exprs[1]);
1019 out = (ast_expression*)ast_binstore_new(ctx, assignop,
1020 (op->id == opid2('+','=') ? INSTR_ADD_V : INSTR_SUB_V),
1021 exprs[0], exprs[1]);
1024 parseerror(parser, "invalid types used in expression: cannot add or subtract type %s and %s",
1025 type_name[exprs[0]->expression.vtype],
1026 type_name[exprs[1]->expression.vtype]);
1030 case opid2('*','='):
1031 case opid2('/','='):
1032 if (exprs[1]->expression.vtype != TYPE_FLOAT ||
1033 !(exprs[0]->expression.vtype == TYPE_FLOAT ||
1034 exprs[0]->expression.vtype == TYPE_VECTOR))
1036 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1037 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1038 parseerror(parser, "invalid types used in expression: %s and %s",
1042 if (ast_istype(exprs[0], ast_entfield))
1043 assignop = type_storep_instr[exprs[0]->expression.vtype];
1045 assignop = type_store_instr[exprs[0]->expression.vtype];
1046 switch (exprs[0]->expression.vtype) {
1048 out = (ast_expression*)ast_binstore_new(ctx, assignop,
1049 (op->id == opid2('*','=') ? INSTR_MUL_F : INSTR_DIV_F),
1050 exprs[0], exprs[1]);
1053 if (op->id == opid2('*','=')) {
1054 out = (ast_expression*)ast_binstore_new(ctx, assignop, INSTR_MUL_VF,
1055 exprs[0], exprs[1]);
1057 /* there's no DIV_VF */
1058 out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F,
1059 (ast_expression*)parser_const_float_1(parser),
1063 out = (ast_expression*)ast_binstore_new(ctx, assignop, INSTR_MUL_VF,
1068 parseerror(parser, "invalid types used in expression: cannot add or subtract type %s and %s",
1069 type_name[exprs[0]->expression.vtype],
1070 type_name[exprs[1]->expression.vtype]);
1074 case opid2('&','='):
1075 case opid2('|','='):
1076 if (NotSameType(TYPE_FLOAT)) {
1077 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1078 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1079 parseerror(parser, "invalid types used in expression: %s and %s",
1083 if (ast_istype(exprs[0], ast_entfield))
1084 assignop = type_storep_instr[exprs[0]->expression.vtype];
1086 assignop = type_store_instr[exprs[0]->expression.vtype];
1087 out = (ast_expression*)ast_binstore_new(ctx, assignop,
1088 (op->id == opid2('&','=') ? INSTR_BITAND : INSTR_BITOR),
1089 exprs[0], exprs[1]);
1091 case opid3('&','~','='):
1092 /* This is like: a &= ~(b);
1093 * But QC has no bitwise-not, so we implement it as
1096 if (NotSameType(TYPE_FLOAT)) {
1097 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1098 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1099 parseerror(parser, "invalid types used in expression: %s and %s",
1103 if (ast_istype(exprs[0], ast_entfield))
1104 assignop = type_storep_instr[exprs[0]->expression.vtype];
1106 assignop = type_store_instr[exprs[0]->expression.vtype];
1107 out = (ast_expression*)ast_binary_new(ctx, INSTR_BITAND, exprs[0], exprs[1]);
1110 out = (ast_expression*)ast_binstore_new(ctx, assignop, INSTR_SUB_F, exprs[0], out);
1116 parseerror(parser, "failed to apply operand %s", op->op);
1120 DEBUGSHUNTDO(con_out("applied %s\n", op->op));
1121 vec_push(sy->out, syexp(ctx, out));
1125 static bool parser_close_call(parser_t *parser, shunt *sy)
1127 /* was a function call */
1128 ast_expression *fun;
1134 vec_shrinkby(sy->ops, 1);
1135 fid = sy->ops[vec_size(sy->ops)].off;
1137 /* out[fid] is the function
1138 * everything above is parameters...
1139 * 0 params = nothing
1140 * 1 params = ast_expression
1144 if (vec_size(sy->out) < 1 || vec_size(sy->out) <= fid) {
1145 parseerror(parser, "internal error: function call needs function and parameter list...");
1149 fun = sy->out[fid].out;
1151 call = ast_call_new(sy->ops[vec_size(sy->ops)].ctx, fun);
1153 parseerror(parser, "out of memory");
1157 if (fid+1 == vec_size(sy->out)) {
1160 } else if (fid+2 == vec_size(sy->out)) {
1162 vec_shrinkby(sy->out, 1);
1163 params = sy->out[vec_size(sy->out)].block;
1167 vec_push(call->params, sy->out[vec_size(sy->out)].out);
1169 paramcount = vec_size(params->exprs);
1170 call->params = params->exprs;
1171 params->exprs = NULL;
1174 if (!ast_call_check_types(call))
1177 parseerror(parser, "invalid function call");
1181 /* overwrite fid, the function, with a call */
1182 sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
1184 if (fun->expression.vtype != TYPE_FUNCTION) {
1185 parseerror(parser, "not a function (%s)", type_name[fun->expression.vtype]);
1189 if (!fun->expression.next) {
1190 parseerror(parser, "could not determine function return type");
1193 if (vec_size(fun->expression.params) != paramcount &&
1194 !(fun->expression.variadic &&
1195 vec_size(fun->expression.params) < paramcount))
1198 const char *fewmany = (vec_size(fun->expression.params) > paramcount) ? "few" : "many";
1200 fval = (ast_istype(fun, ast_value) ? ((ast_value*)fun) : NULL);
1201 if (opts_standard == COMPILER_GMQCC)
1204 parseerror(parser, "too %s parameters for call to %s: expected %i, got %i\n"
1205 " -> `%s` has been declared here: %s:%i",
1206 fewmany, fval->name, (int)vec_size(fun->expression.params), (int)paramcount,
1207 fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line);
1209 parseerror(parser, "too %s parameters for function call: expected %i, got %i\n"
1210 " -> `%s` has been declared here: %s:%i",
1211 fewmany, fval->name, (int)vec_size(fun->expression.params), (int)paramcount,
1212 fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line);
1218 return !parsewarning(parser, WARN_TOO_FEW_PARAMETERS,
1219 "too %s parameters for call to %s: expected %i, got %i\n"
1220 " -> `%s` has been declared here: %s:%i",
1221 fewmany, fval->name, (int)vec_size(fun->expression.params), (int)paramcount,
1222 fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line);
1224 return !parsewarning(parser, WARN_TOO_FEW_PARAMETERS,
1225 "too %s parameters for function call: expected %i, got %i\n"
1226 " -> `%s` has been declared here: %s:%i",
1227 fewmany, fval->name, (int)vec_size(fun->expression.params), (int)paramcount,
1228 fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line);
1236 static bool parser_close_paren(parser_t *parser, shunt *sy, bool functions_only)
1238 if (!vec_size(sy->ops)) {
1239 parseerror(parser, "unmatched closing paren");
1242 /* this would for bit a + (x) because there are no operators inside (x)
1243 if (sy->ops[vec_size(sy->ops)-1].paren == 1) {
1244 parseerror(parser, "empty parenthesis expression");
1248 while (vec_size(sy->ops)) {
1249 if (sy->ops[vec_size(sy->ops)-1].paren == SY_PAREN_FUNC) {
1250 if (!parser_close_call(parser, sy))
1254 if (sy->ops[vec_size(sy->ops)-1].paren == SY_PAREN_EXPR) {
1255 vec_shrinkby(sy->ops, 1);
1256 return !functions_only;
1258 if (sy->ops[vec_size(sy->ops)-1].paren == SY_PAREN_INDEX) {
1261 /* pop off the parenthesis */
1262 vec_shrinkby(sy->ops, 1);
1263 /* then apply the index operator */
1264 if (!parser_sy_pop(parser, sy))
1268 if (!parser_sy_pop(parser, sy))
1274 static void parser_reclassify_token(parser_t *parser)
1277 for (i = 0; i < operator_count; ++i) {
1278 if (!strcmp(parser_tokval(parser), operators[i].op)) {
1279 parser->tok = TOKEN_OPERATOR;
1285 static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma)
1287 ast_expression *expr = NULL;
1289 bool wantop = false;
1290 bool gotmemberof = false;
1292 /* count the parens because an if starts with one, so the
1293 * end of a condition is an unmatched closing paren
1301 parser->lex->flags.noops = false;
1303 parser_reclassify_token(parser);
1308 gotmemberof = false;
1310 parser->memberof = 0;
1312 if (OPTS_FLAG(TRANSLATABLE_STRINGS) &&
1313 parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "_"))
1315 /* a translatable string */
1319 parseerror(parser, "expected operator or end of statement, got constant");
1323 parser->lex->flags.noops = true;
1324 if (!parser_next(parser) || parser->tok != '(') {
1325 parseerror(parser, "use _(\"string\") to create a translatable string constant");
1328 parser->lex->flags.noops = false;
1329 if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) {
1330 parseerror(parser, "expected a constant string in translatable-string extension");
1333 val = parser_const_string(parser, parser_tokval(parser), true);
1337 vec_push(sy.out, syexp(parser_ctx(parser), (ast_expression*)val));
1338 DEBUGSHUNTDO(con_out("push string\n"));
1340 if (!parser_next(parser) || parser->tok != ')') {
1341 parseerror(parser, "expected closing paren after translatable string");
1345 else if (parser->tok == TOKEN_IDENT)
1347 ast_expression *var;
1349 parseerror(parser, "expected operator or end of statement");
1354 if (opts_standard == COMPILER_GMQCC)
1356 if (parser->memberof == TYPE_ENTITY) {
1357 /* still get vars first since there could be a fieldpointer */
1358 var = parser_find_var(parser, parser_tokval(parser));
1360 var = parser_find_field(parser, parser_tokval(parser));
1362 else if (parser->memberof == TYPE_VECTOR)
1364 parseerror(parser, "TODO: implement effective vector member access");
1367 else if (parser->memberof) {
1368 parseerror(parser, "namespace for member not found");
1372 var = parser_find_var(parser, parser_tokval(parser));
1374 var = parser_find_var(parser, parser_tokval(parser));
1376 var = parser_find_field(parser, parser_tokval(parser));
1379 parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
1382 if (ast_istype(var, ast_value))
1383 ((ast_value*)var)->uses++;
1384 vec_push(sy.out, syexp(parser_ctx(parser), var));
1385 DEBUGSHUNTDO(con_out("push %s\n", parser_tokval(parser)));
1387 else if (parser->tok == TOKEN_FLOATCONST) {
1390 parseerror(parser, "expected operator or end of statement, got constant");
1394 val = parser_const_float(parser, (parser_token(parser)->constval.f));
1397 vec_push(sy.out, syexp(parser_ctx(parser), (ast_expression*)val));
1398 DEBUGSHUNTDO(con_out("push %g\n", parser_token(parser)->constval.f));
1400 else if (parser->tok == TOKEN_INTCONST) {
1403 parseerror(parser, "expected operator or end of statement, got constant");
1407 val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
1410 vec_push(sy.out, syexp(parser_ctx(parser), (ast_expression*)val));
1411 DEBUGSHUNTDO(con_out("push %i\n", parser_token(parser)->constval.i));
1413 else if (parser->tok == TOKEN_STRINGCONST) {
1416 parseerror(parser, "expected operator or end of statement, got constant");
1420 val = parser_const_string(parser, parser_tokval(parser), false);
1423 vec_push(sy.out, syexp(parser_ctx(parser), (ast_expression*)val));
1424 DEBUGSHUNTDO(con_out("push string\n"));
1426 else if (parser->tok == TOKEN_VECTORCONST) {
1429 parseerror(parser, "expected operator or end of statement, got constant");
1433 val = parser_const_vector(parser, parser_token(parser)->constval.v);
1436 vec_push(sy.out, syexp(parser_ctx(parser), (ast_expression*)val));
1437 DEBUGSHUNTDO(con_out("push '%g %g %g'\n",
1438 parser_token(parser)->constval.v.x,
1439 parser_token(parser)->constval.v.y,
1440 parser_token(parser)->constval.v.z));
1442 else if (parser->tok == '(') {
1443 parseerror(parser, "internal error: '(' should be classified as operator");
1446 else if (parser->tok == '[') {
1447 parseerror(parser, "internal error: '[' should be classified as operator");
1450 else if (parser->tok == ')') {
1452 DEBUGSHUNTDO(con_out("do[op] )\n"));
1456 /* we do expect an operator next */
1457 /* closing an opening paren */
1458 if (!parser_close_paren(parser, &sy, false))
1461 DEBUGSHUNTDO(con_out("do[nop] )\n"));
1465 /* allowed for function calls */
1466 if (!parser_close_paren(parser, &sy, true))
1471 else if (parser->tok == ']') {
1473 parseerror(parser, "operand expected");
1477 if (!parser_close_paren(parser, &sy, false))
1481 else if (parser->tok != TOKEN_OPERATOR) {
1483 parseerror(parser, "expected operator or end of statement");
1490 /* classify the operator */
1491 const oper_info *op;
1492 const oper_info *olast = NULL;
1494 for (o = 0; o < operator_count; ++o) {
1495 if ((!(operators[o].flags & OP_PREFIX) == wantop) &&
1496 /* !(operators[o].flags & OP_SUFFIX) && / * remove this */
1497 !strcmp(parser_tokval(parser), operators[o].op))
1502 if (o == operator_count) {
1503 /* no operator found... must be the end of the statement */
1506 /* found an operator */
1509 /* when declaring variables, a comma starts a new variable */
1510 if (op->id == opid1(',') && !parens && stopatcomma) {
1511 /* fixup the token */
1516 /* a colon without a pervious question mark cannot be a ternary */
1517 if (!ternaries && op->id == opid2(':','?')) {
1522 if (vec_size(sy.ops) && !vec_last(sy.ops).paren)
1523 olast = &operators[vec_last(sy.ops).etype-1];
1526 (op->prec < olast->prec) ||
1527 (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
1529 if (!parser_sy_pop(parser, &sy))
1531 if (vec_size(sy.ops) && !vec_last(sy.ops).paren)
1532 olast = &operators[vec_last(sy.ops).etype-1];
1537 if (op->id == opid1('.') && opts_standard == COMPILER_GMQCC) {
1538 /* for gmqcc standard: open up the namespace of the previous type */
1539 ast_expression *prevex = vec_last(sy.out).out;
1541 parseerror(parser, "unexpected member operator");
1544 if (prevex->expression.vtype == TYPE_ENTITY)
1545 parser->memberof = TYPE_ENTITY;
1546 else if (prevex->expression.vtype == TYPE_VECTOR)
1547 parser->memberof = TYPE_VECTOR;
1549 parseerror(parser, "type error: type has no members");
1555 if (op->id == opid1('(')) {
1557 size_t sycount = vec_size(sy.out);
1558 DEBUGSHUNTDO(con_out("push [op] (\n"));
1560 /* we expected an operator, this is the function-call operator */
1561 vec_push(sy.ops, syparen(parser_ctx(parser), SY_PAREN_FUNC, sycount-1));
1564 vec_push(sy.ops, syparen(parser_ctx(parser), SY_PAREN_EXPR, 0));
1565 DEBUGSHUNTDO(con_out("push [nop] (\n"));
1568 } else if (op->id == opid1('[')) {
1570 parseerror(parser, "unexpected array subscript");
1574 /* push both the operator and the paren, this makes life easier */
1575 vec_push(sy.ops, syop(parser_ctx(parser), op));
1576 vec_push(sy.ops, syparen(parser_ctx(parser), SY_PAREN_INDEX, 0));
1578 } else if (op->id == opid2('?',':')) {
1580 vec_push(sy.ops, syop(parser_ctx(parser), op));
1583 } else if (op->id == opid2(':','?')) {
1584 /* we don't push this operator */
1588 DEBUGSHUNTDO(con_out("push operator %s\n", op->op));
1589 vec_push(sy.ops, syop(parser_ctx(parser), op));
1590 wantop = !!(op->flags & OP_SUFFIX);
1593 if (!parser_next(parser)) {
1596 if (parser->tok == ';' ||
1597 (!parens && parser->tok == ']'))
1603 while (vec_size(sy.ops)) {
1604 if (!parser_sy_pop(parser, &sy))
1608 parser->lex->flags.noops = true;
1609 if (!vec_size(sy.out)) {
1610 parseerror(parser, "empty expression");
1613 expr = sy.out[0].out;
1616 DEBUGSHUNTDO(con_out("shunt done\n"));
1620 parser->lex->flags.noops = true;
1626 static ast_expression* parse_expression(parser_t *parser, bool stopatcomma)
1628 ast_expression *e = parse_expression_leave(parser, stopatcomma);
1631 if (!parser_next(parser)) {
1638 static void parser_enterblock(parser_t *parser)
1640 vec_push(parser->variables, util_htnew(PARSER_HT_SIZE));
1641 vec_push(parser->_blocklocals, vec_size(parser->_locals));
1642 vec_push(parser->typedefs, util_htnew(TYPEDEF_HT_SIZE));
1643 vec_push(parser->_blocktypedefs, vec_size(parser->_typedefs));
1646 static bool parser_leaveblock(parser_t *parser)
1649 size_t locals, typedefs;
1651 if (vec_size(parser->variables) <= PARSER_HT_LOCALS) {
1652 parseerror(parser, "internal error: parser_leaveblock with no block");
1656 util_htdel(vec_last(parser->variables));
1657 vec_pop(parser->variables);
1658 if (!vec_size(parser->_blocklocals)) {
1659 parseerror(parser, "internal error: parser_leaveblock with no block (2)");
1663 locals = vec_last(parser->_blocklocals);
1664 vec_pop(parser->_blocklocals);
1665 while (vec_size(parser->_locals) != locals) {
1666 ast_expression *e = vec_last(parser->_locals);
1667 ast_value *v = (ast_value*)e;
1668 vec_pop(parser->_locals);
1669 if (ast_istype(e, ast_value) && !v->uses) {
1670 if (parsewarning(parser, WARN_UNUSED_VARIABLE, "unused variable: `%s`", v->name))
1675 typedefs = vec_last(parser->_blocktypedefs);
1676 while (vec_size(parser->_typedefs) != typedefs) {
1677 ast_delete(vec_last(parser->_typedefs));
1678 vec_pop(parser->_typedefs);
1680 util_htdel(vec_last(parser->typedefs));
1681 vec_pop(parser->typedefs);
1686 static void parser_addlocal(parser_t *parser, const char *name, ast_expression *e)
1688 vec_push(parser->_locals, e);
1689 util_htset(vec_last(parser->variables), name, (void*)e);
1692 static bool parse_if(parser_t *parser, ast_block *block, ast_expression **out)
1695 ast_expression *cond, *ontrue, *onfalse = NULL;
1698 lex_ctx ctx = parser_ctx(parser);
1700 (void)block; /* not touching */
1702 /* skip the 'if', parse an optional 'not' and check for an opening paren */
1703 if (!parser_next(parser)) {
1704 parseerror(parser, "expected condition or 'not'");
1707 if (parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "not")) {
1709 if (!parser_next(parser)) {
1710 parseerror(parser, "expected condition in parenthesis");
1714 if (parser->tok != '(') {
1715 parseerror(parser, "expected 'if' condition in parenthesis");
1718 /* parse into the expression */
1719 if (!parser_next(parser)) {
1720 parseerror(parser, "expected 'if' condition after opening paren");
1723 /* parse the condition */
1724 cond = parse_expression_leave(parser, false);
1728 if (parser->tok != ')') {
1729 parseerror(parser, "expected closing paren after 'if' condition");
1733 /* parse into the 'then' branch */
1734 if (!parser_next(parser)) {
1735 parseerror(parser, "expected statement for on-true branch of 'if'");
1739 ontrue = parse_statement_or_block(parser);
1744 /* check for an else */
1745 if (!strcmp(parser_tokval(parser), "else")) {
1746 /* parse into the 'else' branch */
1747 if (!parser_next(parser)) {
1748 parseerror(parser, "expected on-false branch after 'else'");
1753 onfalse = parse_statement_or_block(parser);
1762 ifthen = ast_ifthen_new(ctx, cond, onfalse, ontrue);
1764 ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
1765 *out = (ast_expression*)ifthen;
1769 static bool parse_while(parser_t *parser, ast_block *block, ast_expression **out)
1772 ast_expression *cond, *ontrue;
1774 lex_ctx ctx = parser_ctx(parser);
1776 (void)block; /* not touching */
1778 /* skip the 'while' and check for opening paren */
1779 if (!parser_next(parser) || parser->tok != '(') {
1780 parseerror(parser, "expected 'while' condition in parenthesis");
1783 /* parse into the expression */
1784 if (!parser_next(parser)) {
1785 parseerror(parser, "expected 'while' condition after opening paren");
1788 /* parse the condition */
1789 cond = parse_expression_leave(parser, false);
1793 if (parser->tok != ')') {
1794 parseerror(parser, "expected closing paren after 'while' condition");
1798 /* parse into the 'then' branch */
1799 if (!parser_next(parser)) {
1800 parseerror(parser, "expected while-loop body");
1804 ontrue = parse_statement_or_block(parser);
1810 aloop = ast_loop_new(ctx, NULL, cond, NULL, NULL, ontrue);
1811 *out = (ast_expression*)aloop;
1815 static bool parse_dowhile(parser_t *parser, ast_block *block, ast_expression **out)
1818 ast_expression *cond, *ontrue;
1820 lex_ctx ctx = parser_ctx(parser);
1822 (void)block; /* not touching */
1824 /* skip the 'do' and get the body */
1825 if (!parser_next(parser)) {
1826 parseerror(parser, "expected loop body");
1829 ontrue = parse_statement_or_block(parser);
1833 /* expect the "while" */
1834 if (parser->tok != TOKEN_KEYWORD ||
1835 strcmp(parser_tokval(parser), "while"))
1837 parseerror(parser, "expected 'while' and condition");
1842 /* skip the 'while' and check for opening paren */
1843 if (!parser_next(parser) || parser->tok != '(') {
1844 parseerror(parser, "expected 'while' condition in parenthesis");
1848 /* parse into the expression */
1849 if (!parser_next(parser)) {
1850 parseerror(parser, "expected 'while' condition after opening paren");
1854 /* parse the condition */
1855 cond = parse_expression_leave(parser, false);
1859 if (parser->tok != ')') {
1860 parseerror(parser, "expected closing paren after 'while' condition");
1866 if (!parser_next(parser) || parser->tok != ';') {
1867 parseerror(parser, "expected semicolon after condition");
1873 if (!parser_next(parser)) {
1874 parseerror(parser, "parse error");
1880 aloop = ast_loop_new(ctx, NULL, NULL, cond, NULL, ontrue);
1881 *out = (ast_expression*)aloop;
1885 static bool parse_for(parser_t *parser, ast_block *block, ast_expression **out)
1888 ast_expression *initexpr, *cond, *increment, *ontrue;
1892 lex_ctx ctx = parser_ctx(parser);
1894 parser_enterblock(parser);
1901 /* skip the 'while' and check for opening paren */
1902 if (!parser_next(parser) || parser->tok != '(') {
1903 parseerror(parser, "expected 'for' expressions in parenthesis");
1906 /* parse into the expression */
1907 if (!parser_next(parser)) {
1908 parseerror(parser, "expected 'for' initializer after opening paren");
1913 if (parser->tok == TOKEN_IDENT)
1914 typevar = parser_find_typedef(parser, parser_tokval(parser), 0);
1916 if (typevar || parser->tok == TOKEN_TYPENAME) {
1917 if (opts_standard != COMPILER_GMQCC) {
1918 if (parsewarning(parser, WARN_EXTENSIONS,
1919 "current standard does not allow variable declarations in for-loop initializers"))
1923 parseerror(parser, "TODO: assignment of new variables to be non-const");
1925 if (!parse_variable(parser, block, true, false, typevar))
1928 else if (parser->tok != ';')
1930 initexpr = parse_expression_leave(parser, false);
1935 /* move on to condition */
1936 if (parser->tok != ';') {
1937 parseerror(parser, "expected semicolon after for-loop initializer");
1940 if (!parser_next(parser)) {
1941 parseerror(parser, "expected for-loop condition");
1945 /* parse the condition */
1946 if (parser->tok != ';') {
1947 cond = parse_expression_leave(parser, false);
1952 /* move on to incrementor */
1953 if (parser->tok != ';') {
1954 parseerror(parser, "expected semicolon after for-loop initializer");
1957 if (!parser_next(parser)) {
1958 parseerror(parser, "expected for-loop condition");
1962 /* parse the incrementor */
1963 if (parser->tok != ')') {
1964 increment = parse_expression_leave(parser, false);
1967 if (!ast_side_effects(increment)) {
1968 if (genwarning(ast_ctx(increment), WARN_EFFECTLESS_STATEMENT, "statement has no effect"))
1974 if (parser->tok != ')') {
1975 parseerror(parser, "expected closing paren after 'for-loop' incrementor");
1978 /* parse into the 'then' branch */
1979 if (!parser_next(parser)) {
1980 parseerror(parser, "expected for-loop body");
1983 ontrue = parse_statement_or_block(parser);
1988 aloop = ast_loop_new(ctx, initexpr, cond, NULL, increment, ontrue);
1989 *out = (ast_expression*)aloop;
1991 if (!parser_leaveblock(parser))
1995 if (initexpr) ast_delete(initexpr);
1996 if (cond) ast_delete(cond);
1997 if (increment) ast_delete(increment);
1998 (void)!parser_leaveblock(parser);
2002 static bool parse_return(parser_t *parser, ast_block *block, ast_expression **out)
2004 ast_expression *exp = NULL;
2005 ast_return *ret = NULL;
2006 ast_value *expected = parser->function->vtype;
2008 (void)block; /* not touching */
2010 if (!parser_next(parser)) {
2011 parseerror(parser, "expected return expression");
2015 if (parser->tok != ';') {
2016 exp = parse_expression(parser, false);
2020 if (exp->expression.vtype != expected->expression.next->expression.vtype) {
2021 parseerror(parser, "return with invalid expression");
2024 ret = ast_return_new(exp->expression.node.context, exp);
2030 if (!parser_next(parser))
2031 parseerror(parser, "parse error");
2032 if (expected->expression.next->expression.vtype != TYPE_VOID) {
2033 if (opts_standard != COMPILER_GMQCC)
2034 (void)!parsewarning(parser, WARN_MISSING_RETURN_VALUES, "return without value");
2036 parseerror(parser, "return without value");
2038 ret = ast_return_new(parser_ctx(parser), NULL);
2040 *out = (ast_expression*)ret;
2044 static bool parse_break_continue(parser_t *parser, ast_block *block, ast_expression **out, bool is_continue)
2046 lex_ctx ctx = parser_ctx(parser);
2048 (void)block; /* not touching */
2050 if (!parser_next(parser) || parser->tok != ';') {
2051 parseerror(parser, "expected semicolon");
2055 if (!parser_next(parser))
2056 parseerror(parser, "parse error");
2058 *out = (ast_expression*)ast_breakcont_new(ctx, is_continue);
2062 static bool parse_switch(parser_t *parser, ast_block *block, ast_expression **out)
2064 ast_expression *operand;
2066 ast_switch *switchnode;
2067 ast_switch_case swcase;
2069 lex_ctx ctx = parser_ctx(parser);
2071 (void)block; /* not touching */
2073 /* parse over the opening paren */
2074 if (!parser_next(parser) || parser->tok != '(') {
2075 parseerror(parser, "expected switch operand in parenthesis");
2079 /* parse into the expression */
2080 if (!parser_next(parser)) {
2081 parseerror(parser, "expected switch operand");
2084 /* parse the operand */
2085 operand = parse_expression_leave(parser, false);
2089 if (!OPTS_FLAG(RELAXED_SWITCH)) {
2090 opval = (ast_value*)operand;
2091 if (!ast_istype(operand, ast_value) || !opval->constant) {
2092 parseerror(parser, "case on non-constant values need to be explicitly enabled via -frelaxed-switch");
2098 switchnode = ast_switch_new(ctx, operand);
2101 if (parser->tok != ')') {
2102 ast_delete(switchnode);
2103 parseerror(parser, "expected closing paren after 'switch' operand");
2107 /* parse over the opening paren */
2108 if (!parser_next(parser) || parser->tok != '{') {
2109 ast_delete(switchnode);
2110 parseerror(parser, "expected list of cases");
2114 if (!parser_next(parser)) {
2115 ast_delete(switchnode);
2116 parseerror(parser, "expected 'case' or 'default'");
2121 while (parser->tok != '}') {
2122 ast_block *caseblock;
2124 if (parser->tok != TOKEN_KEYWORD) {
2125 ast_delete(switchnode);
2126 parseerror(parser, "expected 'case' or 'default'");
2129 if (!strcmp(parser_tokval(parser), "case")) {
2130 if (!parser_next(parser)) {
2131 ast_delete(switchnode);
2132 parseerror(parser, "expected expression for case");
2135 swcase.value = parse_expression_leave(parser, false);
2136 if (!swcase.value) {
2137 ast_delete(switchnode);
2138 parseerror(parser, "expected expression for case");
2142 else if (!strcmp(parser_tokval(parser), "default")) {
2143 swcase.value = NULL;
2144 if (!parser_next(parser)) {
2145 ast_delete(switchnode);
2146 parseerror(parser, "expected colon");
2151 /* Now the colon and body */
2152 if (parser->tok != ':') {
2153 if (swcase.value) ast_unref(swcase.value);
2154 ast_delete(switchnode);
2155 parseerror(parser, "expected colon");
2159 if (!parser_next(parser)) {
2160 if (swcase.value) ast_unref(swcase.value);
2161 ast_delete(switchnode);
2162 parseerror(parser, "expected statements or case");
2165 caseblock = ast_block_new(parser_ctx(parser));
2167 if (swcase.value) ast_unref(swcase.value);
2168 ast_delete(switchnode);
2171 swcase.code = (ast_expression*)caseblock;
2172 vec_push(switchnode->cases, swcase);
2174 ast_expression *expr;
2175 if (parser->tok == '}')
2177 if (parser->tok == TOKEN_KEYWORD) {
2178 if (!strcmp(parser_tokval(parser), "case") ||
2179 !strcmp(parser_tokval(parser), "default"))
2184 if (!parse_statement(parser, caseblock, &expr, true)) {
2185 ast_delete(switchnode);
2190 ast_block_add_expr(caseblock, expr);
2195 if (parser->tok != '}') {
2196 ast_delete(switchnode);
2197 parseerror(parser, "expected closing paren of case list");
2200 if (!parser_next(parser)) {
2201 ast_delete(switchnode);
2202 parseerror(parser, "parse error after switch");
2205 *out = (ast_expression*)switchnode;
2209 static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases)
2211 ast_value *typevar = NULL;
2212 if (parser->tok == TOKEN_IDENT)
2213 typevar = parser_find_typedef(parser, parser_tokval(parser), 0);
2215 if (typevar || parser->tok == TOKEN_TYPENAME || parser->tok == '.')
2217 /* local variable */
2219 parseerror(parser, "cannot declare a variable from here");
2222 if (opts_standard == COMPILER_QCC) {
2223 if (parsewarning(parser, WARN_EXTENSIONS, "missing 'local' keyword when declaring a local variable"))
2226 if (!parse_variable(parser, block, false, false, typevar))
2231 else if (parser->tok == TOKEN_KEYWORD)
2233 if (!strcmp(parser_tokval(parser), "local"))
2236 parseerror(parser, "cannot declare a local variable here");
2239 if (!parser_next(parser)) {
2240 parseerror(parser, "expected variable declaration");
2243 if (!parse_variable(parser, block, true, false, NULL))
2248 else if (!strcmp(parser_tokval(parser), "return"))
2250 return parse_return(parser, block, out);
2252 else if (!strcmp(parser_tokval(parser), "if"))
2254 return parse_if(parser, block, out);
2256 else if (!strcmp(parser_tokval(parser), "while"))
2258 return parse_while(parser, block, out);
2260 else if (!strcmp(parser_tokval(parser), "do"))
2262 return parse_dowhile(parser, block, out);
2264 else if (!strcmp(parser_tokval(parser), "for"))
2266 if (opts_standard == COMPILER_QCC) {
2267 if (parsewarning(parser, WARN_EXTENSIONS, "for loops are not recognized in the original Quake C standard, to enable try an alternate standard --std=?"))
2270 return parse_for(parser, block, out);
2272 else if (!strcmp(parser_tokval(parser), "break"))
2274 return parse_break_continue(parser, block, out, false);
2276 else if (!strcmp(parser_tokval(parser), "continue"))
2278 return parse_break_continue(parser, block, out, true);
2280 else if (!strcmp(parser_tokval(parser), "switch"))
2282 return parse_switch(parser, block, out);
2284 else if (!strcmp(parser_tokval(parser), "case") ||
2285 !strcmp(parser_tokval(parser), "default"))
2288 parseerror(parser, "unexpected 'case' label");
2293 else if (!strcmp(parser_tokval(parser), "typedef"))
2295 if (!parser_next(parser)) {
2296 parseerror(parser, "expected type definition after 'typedef'");
2299 return parse_typedef(parser);
2301 parseerror(parser, "Unexpected keyword");
2304 else if (parser->tok == '{')
2307 inner = parse_block(parser, false);
2310 *out = (ast_expression*)inner;
2315 ast_expression *exp = parse_expression(parser, false);
2319 if (!ast_side_effects(exp)) {
2320 if (genwarning(ast_ctx(exp), WARN_EFFECTLESS_STATEMENT, "statement has no effect"))
2327 static bool parse_block_into(parser_t *parser, ast_block *block, bool warnreturn)
2331 parser_enterblock(parser);
2333 if (!parser_next(parser)) { /* skip the '{' */
2334 parseerror(parser, "expected function body");
2338 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
2340 ast_expression *expr = NULL;
2341 if (parser->tok == '}')
2344 if (!parse_statement(parser, block, &expr, false)) {
2345 /* parseerror(parser, "parse error"); */
2351 ast_block_add_expr(block, expr);
2354 if (parser->tok != '}') {
2357 if (warnreturn && parser->function->vtype->expression.next->expression.vtype != TYPE_VOID)
2359 if (!vec_size(block->exprs) ||
2360 !ast_istype(vec_last(block->exprs), ast_return))
2362 if (parsewarning(parser, WARN_MISSING_RETURN_VALUES, "control reaches end of non-void function")) {
2368 (void)parser_next(parser);
2372 if (!parser_leaveblock(parser))
2374 return retval && !!block;
2377 static ast_block* parse_block(parser_t *parser, bool warnreturn)
2380 block = ast_block_new(parser_ctx(parser));
2383 if (!parse_block_into(parser, block, warnreturn)) {
2384 ast_block_delete(block);
2390 static ast_expression* parse_statement_or_block(parser_t *parser)
2392 ast_expression *expr = NULL;
2393 if (parser->tok == '{')
2394 return (ast_expression*)parse_block(parser, false);
2395 if (!parse_statement(parser, NULL, &expr, false))
2400 static bool create_vector_members(ast_value *var, ast_member **me)
2403 size_t len = strlen(var->name);
2405 for (i = 0; i < 3; ++i) {
2406 char *name = mem_a(len+3);
2407 memcpy(name, var->name, len);
2409 name[len+1] = 'x'+i;
2411 me[i] = ast_member_new(ast_ctx(var), (ast_expression*)var, i, name);
2420 do { ast_member_delete(me[--i]); } while(i);
2424 static bool parse_function_body(parser_t *parser, ast_value *var)
2426 ast_block *block = NULL;
2431 ast_expression *framenum = NULL;
2432 ast_expression *nextthink = NULL;
2433 /* None of the following have to be deleted */
2434 ast_expression *fld_think = NULL, *fld_nextthink = NULL, *fld_frame = NULL;
2435 ast_expression *gbl_time = NULL, *gbl_self = NULL;
2436 bool has_frame_think;
2440 has_frame_think = false;
2441 old = parser->function;
2443 if (var->expression.variadic) {
2444 if (parsewarning(parser, WARN_VARIADIC_FUNCTION,
2445 "variadic function with implementation will not be able to access additional parameters"))
2451 if (parser->tok == '[') {
2452 /* got a frame definition: [ framenum, nextthink ]
2453 * this translates to:
2454 * self.frame = framenum;
2455 * self.nextthink = time + 0.1;
2456 * self.think = nextthink;
2460 fld_think = parser_find_field(parser, "think");
2461 fld_nextthink = parser_find_field(parser, "nextthink");
2462 fld_frame = parser_find_field(parser, "frame");
2463 if (!fld_think || !fld_nextthink || !fld_frame) {
2464 parseerror(parser, "cannot use [frame,think] notation without the required fields");
2465 parseerror(parser, "please declare the following entityfields: `frame`, `think`, `nextthink`");
2468 gbl_time = parser_find_global(parser, "time");
2469 gbl_self = parser_find_global(parser, "self");
2470 if (!gbl_time || !gbl_self) {
2471 parseerror(parser, "cannot use [frame,think] notation without the required globals");
2472 parseerror(parser, "please declare the following globals: `time`, `self`");
2476 if (!parser_next(parser))
2479 framenum = parse_expression_leave(parser, true);
2481 parseerror(parser, "expected a framenumber constant in[frame,think] notation");
2484 if (!ast_istype(framenum, ast_value) || !( (ast_value*)framenum )->hasvalue) {
2485 ast_unref(framenum);
2486 parseerror(parser, "framenumber in [frame,think] notation must be a constant");
2490 if (parser->tok != ',') {
2491 ast_unref(framenum);
2492 parseerror(parser, "expected comma after frame number in [frame,think] notation");
2493 parseerror(parser, "Got a %i\n", parser->tok);
2497 if (!parser_next(parser)) {
2498 ast_unref(framenum);
2502 if (parser->tok == TOKEN_IDENT && !parser_find_var(parser, parser_tokval(parser)))
2504 /* qc allows the use of not-yet-declared functions here
2505 * - this automatically creates a prototype */
2506 ast_value *thinkfunc;
2507 ast_expression *functype = fld_think->expression.next;
2509 thinkfunc = ast_value_new(parser_ctx(parser), parser_tokval(parser), functype->expression.vtype);
2510 if (!thinkfunc || !ast_type_adopt(thinkfunc, functype)) {
2511 ast_unref(framenum);
2512 parseerror(parser, "failed to create implicit prototype for `%s`", parser_tokval(parser));
2516 if (!parser_next(parser)) {
2517 ast_unref(framenum);
2518 ast_delete(thinkfunc);
2522 vec_push(parser->globals, (ast_expression*)thinkfunc);
2523 util_htset(parser->htglobals, thinkfunc->name, thinkfunc);
2524 nextthink = (ast_expression*)thinkfunc;
2527 nextthink = parse_expression_leave(parser, true);
2529 ast_unref(framenum);
2530 parseerror(parser, "expected a think-function in [frame,think] notation");
2535 if (!ast_istype(nextthink, ast_value)) {
2536 parseerror(parser, "think-function in [frame,think] notation must be a constant");
2540 if (retval && parser->tok != ']') {
2541 parseerror(parser, "expected closing `]` for [frame,think] notation");
2545 if (retval && !parser_next(parser)) {
2549 if (retval && parser->tok != '{') {
2550 parseerror(parser, "a function body has to be declared after a [frame,think] declaration");
2555 ast_unref(nextthink);
2556 ast_unref(framenum);
2560 has_frame_think = true;
2563 block = ast_block_new(parser_ctx(parser));
2565 parseerror(parser, "failed to allocate block");
2566 if (has_frame_think) {
2567 ast_unref(nextthink);
2568 ast_unref(framenum);
2573 if (has_frame_think) {
2575 ast_expression *self_frame;
2576 ast_expression *self_nextthink;
2577 ast_expression *self_think;
2578 ast_expression *time_plus_1;
2579 ast_store *store_frame;
2580 ast_store *store_nextthink;
2581 ast_store *store_think;
2583 ctx = parser_ctx(parser);
2584 self_frame = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_frame);
2585 self_nextthink = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_nextthink);
2586 self_think = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_think);
2588 time_plus_1 = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F,
2589 gbl_time, (ast_expression*)parser_const_float(parser, 0.1));
2591 if (!self_frame || !self_nextthink || !self_think || !time_plus_1) {
2592 if (self_frame) ast_delete(self_frame);
2593 if (self_nextthink) ast_delete(self_nextthink);
2594 if (self_think) ast_delete(self_think);
2595 if (time_plus_1) ast_delete(time_plus_1);
2601 store_frame = ast_store_new(ctx, INSTR_STOREP_F, self_frame, framenum);
2602 store_nextthink = ast_store_new(ctx, INSTR_STOREP_F, self_nextthink, time_plus_1);
2603 store_think = ast_store_new(ctx, INSTR_STOREP_FNC, self_think, nextthink);
2606 ast_delete(self_frame);
2609 if (!store_nextthink) {
2610 ast_delete(self_nextthink);
2614 ast_delete(self_think);
2618 if (store_frame) ast_delete(store_frame);
2619 if (store_nextthink) ast_delete(store_nextthink);
2620 if (store_think) ast_delete(store_think);
2623 ast_block_add_expr(block, (ast_expression*)store_frame);
2624 ast_block_add_expr(block, (ast_expression*)store_nextthink);
2625 ast_block_add_expr(block, (ast_expression*)store_think);
2629 parseerror(parser, "failed to generate code for [frame,think]");
2630 ast_unref(nextthink);
2631 ast_unref(framenum);
2637 parser_enterblock(parser);
2639 for (parami = 0; parami < vec_size(var->expression.params); ++parami) {
2641 ast_value *param = var->expression.params[parami];
2644 if (param->expression.vtype != TYPE_VECTOR &&
2645 (param->expression.vtype != TYPE_FIELD ||
2646 param->expression.next->expression.vtype != TYPE_VECTOR))
2651 if (!create_vector_members(param, me)) {
2652 ast_block_delete(block);
2656 for (e = 0; e < 3; ++e) {
2657 parser_addlocal(parser, me[e]->name, (ast_expression*)me[e]);
2658 ast_block_collect(block, (ast_expression*)me[e]);
2662 func = ast_function_new(ast_ctx(var), var->name, var);
2664 parseerror(parser, "failed to allocate function for `%s`", var->name);
2665 ast_block_delete(block);
2668 vec_push(parser->functions, func);
2670 parser->function = func;
2671 if (!parse_block_into(parser, block, true)) {
2672 ast_block_delete(block);
2676 vec_push(func->blocks, block);
2678 parser->function = old;
2679 if (!parser_leaveblock(parser))
2681 if (vec_size(parser->variables) != PARSER_HT_LOCALS) {
2682 parseerror(parser, "internal error: local scopes left");
2686 if (parser->tok == ';')
2687 return parser_next(parser);
2688 else if (opts_standard == COMPILER_QCC)
2689 parseerror(parser, "missing semicolon after function body (mandatory with -std=qcc)");
2693 vec_pop(parser->functions);
2694 ast_function_delete(func);
2695 var->constval.vfunc = NULL;
2698 (void)!parser_leaveblock(parser);
2699 parser->function = old;
2703 static ast_expression *array_accessor_split(
2708 ast_expression *left,
2709 ast_expression *right
2715 lex_ctx ctx = ast_ctx(array);
2717 if (!left || !right) {
2718 if (left) ast_delete(left);
2719 if (right) ast_delete(right);
2723 cmp = ast_binary_new(ctx, INSTR_LT,
2724 (ast_expression*)index,
2725 (ast_expression*)parser_const_float(parser, middle));
2729 parseerror(parser, "internal error: failed to create comparison for array setter");
2733 ifthen = ast_ifthen_new(ctx, (ast_expression*)cmp, left, right);
2735 ast_delete(cmp); /* will delete left and right */
2736 parseerror(parser, "internal error: failed to create conditional jump for array setter");
2740 return (ast_expression*)ifthen;
2743 static ast_expression *array_setter_node(parser_t *parser, ast_value *array, ast_value *index, ast_value *value, size_t from, size_t afterend)
2745 lex_ctx ctx = ast_ctx(array);
2747 if (from+1 == afterend) {
2748 /* set this value */
2751 ast_array_index *subscript;
2753 int assignop = type_store_instr[value->expression.vtype];
2755 if (value->expression.vtype == TYPE_FIELD && value->expression.next->expression.vtype == TYPE_VECTOR)
2756 assignop = INSTR_STORE_V;
2758 subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)parser_const_float(parser, from));
2762 st = ast_store_new(ctx, assignop, (ast_expression*)subscript, (ast_expression*)value);
2764 ast_delete(subscript);
2768 block = ast_block_new(ctx);
2774 ast_block_add_expr(block, (ast_expression*)st);
2776 ret = ast_return_new(ctx, NULL);
2782 ast_block_add_expr(block, (ast_expression*)ret);
2784 return (ast_expression*)block;
2786 ast_expression *left, *right;
2787 size_t diff = afterend - from;
2788 size_t middle = from + diff/2;
2789 left = array_setter_node(parser, array, index, value, from, middle);
2790 right = array_setter_node(parser, array, index, value, middle, afterend);
2791 return array_accessor_split(parser, array, index, middle, left, right);
2795 static ast_expression *array_field_setter_node(
2804 lex_ctx ctx = ast_ctx(array);
2806 if (from+1 == afterend) {
2807 /* set this value */
2810 ast_entfield *entfield;
2811 ast_array_index *subscript;
2813 int assignop = type_storep_instr[value->expression.vtype];
2815 if (value->expression.vtype == TYPE_FIELD && value->expression.next->expression.vtype == TYPE_VECTOR)
2816 assignop = INSTR_STOREP_V;
2818 subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)parser_const_float(parser, from));
2822 entfield = ast_entfield_new_force(ctx,
2823 (ast_expression*)entity,
2824 (ast_expression*)subscript,
2825 (ast_expression*)subscript);
2827 ast_delete(subscript);
2831 st = ast_store_new(ctx, assignop, (ast_expression*)entfield, (ast_expression*)value);
2833 ast_delete(entfield);
2837 block = ast_block_new(ctx);
2843 ast_block_add_expr(block, (ast_expression*)st);
2845 ret = ast_return_new(ctx, NULL);
2851 ast_block_add_expr(block, (ast_expression*)ret);
2853 return (ast_expression*)block;
2855 ast_expression *left, *right;
2856 size_t diff = afterend - from;
2857 size_t middle = from + diff/2;
2858 left = array_field_setter_node(parser, array, entity, index, value, from, middle);
2859 right = array_field_setter_node(parser, array, entity, index, value, middle, afterend);
2860 return array_accessor_split(parser, array, index, middle, left, right);
2864 static ast_expression *array_getter_node(parser_t *parser, ast_value *array, ast_value *index, size_t from, size_t afterend)
2866 lex_ctx ctx = ast_ctx(array);
2868 if (from+1 == afterend) {
2870 ast_array_index *subscript;
2872 subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)parser_const_float(parser, from));
2876 ret = ast_return_new(ctx, (ast_expression*)subscript);
2878 ast_delete(subscript);
2882 return (ast_expression*)ret;
2884 ast_expression *left, *right;
2885 size_t diff = afterend - from;
2886 size_t middle = from + diff/2;
2887 left = array_getter_node(parser, array, index, from, middle);
2888 right = array_getter_node(parser, array, index, middle, afterend);
2889 return array_accessor_split(parser, array, index, middle, left, right);
2893 static bool parser_create_array_accessor(parser_t *parser, ast_value *array, const char *funcname, ast_value **out)
2895 ast_function *func = NULL;
2896 ast_value *fval = NULL;
2897 ast_block *body = NULL;
2899 fval = ast_value_new(ast_ctx(array), funcname, TYPE_FUNCTION);
2901 parseerror(parser, "failed to create accessor function value");
2905 func = ast_function_new(ast_ctx(array), funcname, fval);
2908 parseerror(parser, "failed to create accessor function node");
2912 body = ast_block_new(ast_ctx(array));
2914 parseerror(parser, "failed to create block for array accessor");
2920 vec_push(func->blocks, body);
2923 vec_push(parser->accessors, fval);
2928 static bool parser_create_array_setter(parser_t *parser, ast_value *array, const char *funcname)
2930 ast_expression *root = NULL;
2931 ast_value *index = NULL;
2932 ast_value *value = NULL;
2936 if (!ast_istype(array->expression.next, ast_value)) {
2937 parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type");
2941 if (!parser_create_array_accessor(parser, array, funcname, &fval))
2943 func = fval->constval.vfunc;
2944 fval->expression.next = (ast_expression*)ast_value_new(ast_ctx(array), "<void>", TYPE_VOID);
2946 index = ast_value_new(ast_ctx(array), "index", TYPE_FLOAT);
2947 value = ast_value_copy((ast_value*)array->expression.next);
2949 if (!index || !value) {
2950 parseerror(parser, "failed to create locals for array accessor");
2953 (void)!ast_value_set_name(value, "value"); /* not important */
2954 vec_push(fval->expression.params, index);
2955 vec_push(fval->expression.params, value);
2957 root = array_setter_node(parser, array, index, value, 0, array->expression.count);
2959 parseerror(parser, "failed to build accessor search tree");
2963 ast_block_add_expr(func->blocks[0], root);
2964 array->setter = fval;
2967 if (index) ast_delete(index);
2968 if (value) ast_delete(value);
2969 if (root) ast_delete(root);
2975 static bool parser_create_array_field_setter(parser_t *parser, ast_value *array, const char *funcname)
2977 ast_expression *root = NULL;
2978 ast_value *entity = NULL;
2979 ast_value *index = NULL;
2980 ast_value *value = NULL;
2984 if (!ast_istype(array->expression.next, ast_value)) {
2985 parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type");
2989 if (!parser_create_array_accessor(parser, array, funcname, &fval))
2991 func = fval->constval.vfunc;
2992 fval->expression.next = (ast_expression*)ast_value_new(ast_ctx(array), "<void>", TYPE_VOID);
2994 entity = ast_value_new(ast_ctx(array), "entity", TYPE_ENTITY);
2995 index = ast_value_new(ast_ctx(array), "index", TYPE_FLOAT);
2996 value = ast_value_copy((ast_value*)array->expression.next);
2997 if (!entity || !index || !value) {
2998 parseerror(parser, "failed to create locals for array accessor");
3001 (void)!ast_value_set_name(value, "value"); /* not important */
3002 vec_push(fval->expression.params, entity);
3003 vec_push(fval->expression.params, index);
3004 vec_push(fval->expression.params, value);
3006 root = array_field_setter_node(parser, array, entity, index, value, 0, array->expression.count);
3008 parseerror(parser, "failed to build accessor search tree");
3012 ast_block_add_expr(func->blocks[0], root);
3013 array->setter = fval;
3016 if (entity) ast_delete(entity);
3017 if (index) ast_delete(index);
3018 if (value) ast_delete(value);
3019 if (root) ast_delete(root);
3025 static bool parser_create_array_getter(parser_t *parser, ast_value *array, const ast_expression *elemtype, const char *funcname)
3027 ast_expression *root = NULL;
3028 ast_value *index = NULL;
3032 /* NOTE: checking array->expression.next rather than elemtype since
3033 * for fields elemtype is a temporary fieldtype.
3035 if (!ast_istype(array->expression.next, ast_value)) {
3036 parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type");
3040 if (!parser_create_array_accessor(parser, array, funcname, &fval))
3042 func = fval->constval.vfunc;
3043 fval->expression.next = ast_type_copy(ast_ctx(array), elemtype);
3045 index = ast_value_new(ast_ctx(array), "index", TYPE_FLOAT);
3048 parseerror(parser, "failed to create locals for array accessor");
3051 vec_push(fval->expression.params, index);
3053 root = array_getter_node(parser, array, index, 0, array->expression.count);
3055 parseerror(parser, "failed to build accessor search tree");
3059 ast_block_add_expr(func->blocks[0], root);
3060 array->getter = fval;
3063 if (index) ast_delete(index);
3064 if (root) ast_delete(root);
3070 static ast_value *parse_typename(parser_t *parser, ast_value **storebase, ast_value *cached_typedef);
3071 static ast_value *parse_parameter_list(parser_t *parser, ast_value *var)
3079 bool variadic = false;
3081 ctx = parser_ctx(parser);
3083 /* for the sake of less code we parse-in in this function */
3084 if (!parser_next(parser)) {
3085 parseerror(parser, "expected parameter list");
3091 /* parse variables until we hit a closing paren */
3092 while (parser->tok != ')') {
3094 /* there must be commas between them */
3095 if (parser->tok != ',') {
3096 parseerror(parser, "expected comma or end of parameter list");
3099 if (!parser_next(parser)) {
3100 parseerror(parser, "expected parameter");
3106 if (parser->tok == TOKEN_DOTS) {
3107 /* '...' indicates a varargs function */
3109 if (!parser_next(parser)) {
3110 parseerror(parser, "expected parameter");
3113 if (parser->tok != ')') {
3114 parseerror(parser, "`...` must be the last parameter of a variadic function declaration");
3120 /* for anything else just parse a typename */
3121 param = parse_typename(parser, NULL, NULL);
3124 vec_push(params, param);
3125 if (param->expression.vtype >= TYPE_VARIANT) {
3126 char typename[1024];
3127 ast_type_to_string((ast_expression*)param, typename, sizeof(typename));
3128 parseerror(parser, "type not supported as part of a parameter list: %s", typename);
3135 if (vec_size(params) > 8 && opts_standard == COMPILER_QCC)
3136 (void)!parsewarning(parser, WARN_EXTENSIONS, "more than 8 parameters are not supported by this standard");
3139 if (!parser_next(parser)) {
3140 parseerror(parser, "parse error after typename");
3144 /* now turn 'var' into a function type */
3145 fval = ast_value_new(ctx, "<type()>", TYPE_FUNCTION);
3146 fval->expression.next = (ast_expression*)var;
3147 fval->expression.variadic = variadic;
3150 var->expression.params = params;
3157 for (i = 0; i < vec_size(params); ++i)
3158 ast_delete(params[i]);
3163 static ast_value *parse_arraysize(parser_t *parser, ast_value *var)
3165 ast_expression *cexp;
3166 ast_value *cval, *tmp;
3169 ctx = parser_ctx(parser);
3171 if (!parser_next(parser)) {
3173 parseerror(parser, "expected array-size");
3177 cexp = parse_expression_leave(parser, true);
3179 if (!cexp || !ast_istype(cexp, ast_value)) {
3183 parseerror(parser, "expected array-size as constant positive integer");
3186 cval = (ast_value*)cexp;
3188 tmp = ast_value_new(ctx, "<type[]>", TYPE_ARRAY);
3189 tmp->expression.next = (ast_expression*)var;
3192 if (cval->expression.vtype == TYPE_INTEGER)
3193 tmp->expression.count = cval->constval.vint;
3194 else if (cval->expression.vtype == TYPE_FLOAT)
3195 tmp->expression.count = cval->constval.vfloat;
3199 parseerror(parser, "array-size must be a positive integer constant");
3204 if (parser->tok != ']') {
3206 parseerror(parser, "expected ']' after array-size");
3209 if (!parser_next(parser)) {
3211 parseerror(parser, "error after parsing array size");
3217 /* Parse a complete typename.
3218 * for single-variables (ie. function parameters or typedefs) storebase should be NULL
3219 * but when parsing variables separated by comma
3220 * 'storebase' should point to where the base-type should be kept.
3221 * The base type makes up every bit of type information which comes *before* the
3224 * The following will be parsed in its entirety:
3226 * The 'basetype' in this case is 'void()'
3227 * and if there's a comma after it, say:
3229 * then the type-information 'void()' can be stored in 'storebase'
3231 static ast_value *parse_typename(parser_t *parser, ast_value **storebase, ast_value *cached_typedef)
3233 ast_value *var, *tmp;
3236 const char *name = NULL;
3237 bool isfield = false;
3238 bool wasarray = false;
3239 size_t morefields = 0;
3241 ctx = parser_ctx(parser);
3243 /* types may start with a dot */
3244 if (parser->tok == '.') {
3246 /* if we parsed a dot we need a typename now */
3247 if (!parser_next(parser)) {
3248 parseerror(parser, "expected typename for field definition");
3252 /* Further dots are handled seperately because they won't be part of the
3255 while (parser->tok == '.') {
3257 if (!parser_next(parser)) {
3258 parseerror(parser, "expected typename for field definition");
3263 if (parser->tok == TOKEN_IDENT)
3264 cached_typedef = parser_find_typedef(parser, parser_tokval(parser), 0);
3265 if (!cached_typedef && parser->tok != TOKEN_TYPENAME) {
3266 parseerror(parser, "expected typename");
3271 /* generate the basic type value */
3272 if (cached_typedef) {
3273 var = ast_value_copy(cached_typedef);
3274 ast_value_set_name(var, "<type(from_def)>");
3276 var = ast_value_new(ctx, "<type>", parser_token(parser)->constval.t);
3278 for (; morefields; --morefields) {
3279 tmp = ast_value_new(ctx, "<.type>", TYPE_FIELD);
3280 tmp->expression.next = (ast_expression*)var;
3284 /* do not yet turn into a field - remember:
3285 * .void() foo; is a field too
3286 * .void()() foo; is a function
3290 if (!parser_next(parser)) {
3292 parseerror(parser, "parse error after typename");
3296 /* an opening paren now starts the parameter-list of a function
3297 * this is where original-QC has parameter lists.
3298 * We allow a single parameter list here.
3299 * Much like fteqcc we don't allow `float()() x`
3301 if (parser->tok == '(') {
3302 var = parse_parameter_list(parser, var);
3307 /* store the base if requested */
3309 *storebase = ast_value_copy(var);
3311 tmp = ast_value_new(ctx, "<type:f>", TYPE_FIELD);
3312 tmp->expression.next = (ast_expression*)*storebase;
3317 /* there may be a name now */
3318 if (parser->tok == TOKEN_IDENT) {
3319 name = util_strdup(parser_tokval(parser));
3321 if (!parser_next(parser)) {
3323 parseerror(parser, "error after variable or field declaration");
3328 /* now this may be an array */
3329 if (parser->tok == '[') {
3331 var = parse_arraysize(parser, var);
3336 /* This is the point where we can turn it into a field */
3338 /* turn it into a field if desired */
3339 tmp = ast_value_new(ctx, "<type:f>", TYPE_FIELD);
3340 tmp->expression.next = (ast_expression*)var;
3344 /* now there may be function parens again */
3345 if (parser->tok == '(' && opts_standard == COMPILER_QCC)
3346 parseerror(parser, "C-style function syntax is not allowed in -std=qcc");
3347 if (parser->tok == '(' && wasarray)
3348 parseerror(parser, "arrays as part of a return type is not supported");
3349 while (parser->tok == '(') {
3350 var = parse_parameter_list(parser, var);
3359 /* finally name it */
3361 if (!ast_value_set_name(var, name)) {
3363 parseerror(parser, "internal error: failed to set name");
3366 /* free the name, ast_value_set_name duplicates */
3373 static bool parse_typedef(parser_t *parser)
3375 ast_value *typevar, *oldtype;
3376 ast_expression *old;
3378 typevar = parse_typename(parser, NULL, NULL);
3383 if ( (old = parser_find_var(parser, typevar->name)) ) {
3384 parseerror(parser, "cannot define a type with the same name as a variable: %s\n"
3385 " -> `%s` has been declared here: %s:%i",
3386 typevar->name, ast_ctx(old).file, ast_ctx(old).line);
3387 ast_delete(typevar);
3391 if ( (oldtype = parser_find_typedef(parser, typevar->name, vec_last(parser->_blocktypedefs))) ) {
3392 parseerror(parser, "type `%s` has already been declared here: %s:%i",
3393 typevar->name, ast_ctx(oldtype).file, ast_ctx(oldtype).line);
3394 ast_delete(typevar);
3398 vec_push(parser->_typedefs, typevar);
3399 util_htset(vec_last(parser->typedefs), typevar->name, typevar);
3401 if (parser->tok != ';') {
3402 parseerror(parser, "expected semicolon after typedef");
3405 if (!parser_next(parser)) {
3406 parseerror(parser, "parse error after typedef");
3413 static bool parse_variable(parser_t *parser, ast_block *localblock, bool nofields, bool is_const, ast_value *cached_typedef)
3417 ast_expression *old;
3421 ast_value *basetype = NULL;
3423 bool isparam = false;
3424 bool isvector = false;
3425 bool cleanvar = true;
3426 bool wasarray = false;
3430 /* get the first complete variable */
3431 var = parse_typename(parser, &basetype, cached_typedef);
3434 ast_delete(basetype);
3442 /* Part 0: finish the type */
3443 if (parser->tok == '(') {
3444 if (opts_standard == COMPILER_QCC)
3445 parseerror(parser, "C-style function syntax is not allowed in -std=qcc");
3446 var = parse_parameter_list(parser, var);
3452 /* we only allow 1-dimensional arrays */
3453 if (parser->tok == '[') {
3455 var = parse_arraysize(parser, var);
3461 if (parser->tok == '(' && wasarray) {
3462 parseerror(parser, "arrays as part of a return type is not supported");
3463 /* we'll still parse the type completely for now */
3465 /* for functions returning functions */
3466 while (parser->tok == '(') {
3467 if (opts_standard == COMPILER_QCC)
3468 parseerror(parser, "C-style function syntax is not allowed in -std=qcc");
3469 var = parse_parameter_list(parser, var);
3477 var->constant = true;
3480 * check for validity: (end_sys_..., multiple-definitions, prototypes, ...)
3481 * Also: if there was a prototype, `var` will be deleted and set to `proto` which
3482 * is then filled with the previous definition and the parameter-names replaced.
3485 /* Deal with end_sys_ vars */
3487 if (!strcmp(var->name, "end_sys_globals")) {
3488 parser->crc_globals = vec_size(parser->globals);
3491 else if (!strcmp(var->name, "end_sys_fields")) {
3492 parser->crc_fields = vec_size(parser->fields);
3495 if (was_end && var->expression.vtype == TYPE_FIELD) {
3496 if (parsewarning(parser, WARN_END_SYS_FIELDS,
3497 "global '%s' hint should not be a field",
3498 parser_tokval(parser)))
3505 if (!nofields && var->expression.vtype == TYPE_FIELD)
3507 /* deal with field declarations */
3508 old = parser_find_field(parser, var->name);
3510 if (parsewarning(parser, WARN_FIELD_REDECLARED, "field `%s` already declared here: %s:%i",
3511 var->name, ast_ctx(old).file, (int)ast_ctx(old).line))
3520 parseerror(parser, "field `%s` already declared here: %s:%i",
3521 var->name, ast_ctx(old).file, ast_ctx(old).line);
3526 if (opts_standard == COMPILER_QCC &&
3527 (old = parser_find_global(parser, var->name)))
3529 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
3530 parseerror(parser, "field `%s` already declared here: %s:%i",
3531 var->name, ast_ctx(old).file, ast_ctx(old).line);
3538 /* deal with other globals */
3539 old = parser_find_global(parser, var->name);
3540 if (old && var->expression.vtype == TYPE_FUNCTION && old->expression.vtype == TYPE_FUNCTION)
3542 /* This is a function which had a prototype */
3543 if (!ast_istype(old, ast_value)) {
3544 parseerror(parser, "internal error: prototype is not an ast_value");
3548 proto = (ast_value*)old;
3549 if (!ast_compare_type((ast_expression*)proto, (ast_expression*)var)) {
3550 parseerror(parser, "conflicting types for `%s`, previous declaration was here: %s:%i",
3552 ast_ctx(proto).file, ast_ctx(proto).line);
3556 /* we need the new parameter-names */
3557 for (i = 0; i < vec_size(proto->expression.params); ++i)
3558 ast_value_set_name(proto->expression.params[i], var->expression.params[i]->name);
3566 if (opts_standard == COMPILER_GMQCC) {
3567 parseerror(parser, "global `%s` already declared here: %s:%i",
3568 var->name, ast_ctx(old).file, ast_ctx(old).line);
3572 if (parsewarning(parser, WARN_DOUBLE_DECLARATION,
3573 "global `%s` already declared here: %s:%i",
3574 var->name, ast_ctx(old).file, ast_ctx(old).line))
3581 if (opts_standard == COMPILER_QCC &&
3582 (old = parser_find_field(parser, var->name)))
3584 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
3585 parseerror(parser, "global `%s` already declared here: %s:%i",
3586 var->name, ast_ctx(old).file, ast_ctx(old).line);
3593 else /* it's not a global */
3595 old = parser_find_local(parser, var->name, vec_size(parser->variables)-1, &isparam);
3596 if (old && !isparam) {
3597 parseerror(parser, "local `%s` already declared here: %s:%i",
3598 var->name, ast_ctx(old).file, (int)ast_ctx(old).line);
3602 old = parser_find_local(parser, var->name, 0, &isparam);
3603 if (old && isparam) {
3604 if (parsewarning(parser, WARN_LOCAL_SHADOWS,
3605 "local `%s` is shadowing a parameter", var->name))
3607 parseerror(parser, "local `%s` already declared here: %s:%i",
3608 var->name, ast_ctx(old).file, (int)ast_ctx(old).line);
3612 if (opts_standard != COMPILER_GMQCC) {
3621 * Create the global/local, and deal with vector types.
3624 if (var->expression.vtype == TYPE_VECTOR)
3626 else if (var->expression.vtype == TYPE_FIELD &&
3627 var->expression.next->expression.vtype == TYPE_VECTOR)
3631 if (!create_vector_members(var, me)) {
3638 /* deal with global variables, fields, functions */
3639 if (!nofields && var->expression.vtype == TYPE_FIELD) {
3640 vec_push(parser->fields, (ast_expression*)var);
3641 util_htset(parser->htfields, var->name, var);
3643 for (i = 0; i < 3; ++i) {
3644 vec_push(parser->fields, (ast_expression*)me[i]);
3645 util_htset(parser->htfields, me[i]->name, me[i]);
3650 vec_push(parser->globals, (ast_expression*)var);
3651 util_htset(parser->htglobals, var->name, var);
3653 for (i = 0; i < 3; ++i) {
3654 vec_push(parser->globals, (ast_expression*)me[i]);
3655 util_htset(parser->htglobals, me[i]->name, me[i]);
3660 vec_push(localblock->locals, var);
3661 parser_addlocal(parser, var->name, (ast_expression*)var);
3663 for (i = 0; i < 3; ++i) {
3664 parser_addlocal(parser, me[i]->name, (ast_expression*)me[i]);
3665 ast_block_collect(localblock, (ast_expression*)me[i]);
3671 me[0] = me[1] = me[2] = NULL;
3676 if (var->expression.vtype == TYPE_ARRAY) {
3678 snprintf(name, sizeof(name), "%s##SET", var->name);
3679 if (!parser_create_array_setter(parser, var, name))
3681 snprintf(name, sizeof(name), "%s##GET", var->name);
3682 if (!parser_create_array_getter(parser, var, var->expression.next, name))
3685 else if (!localblock && !nofields &&
3686 var->expression.vtype == TYPE_FIELD &&
3687 var->expression.next->expression.vtype == TYPE_ARRAY)
3690 ast_expression *telem;
3692 ast_value *array = (ast_value*)var->expression.next;
3694 if (!ast_istype(var->expression.next, ast_value)) {
3695 parseerror(parser, "internal error: field element type must be an ast_value");
3699 snprintf(name, sizeof(name), "%s##SETF", var->name);
3700 if (!parser_create_array_field_setter(parser, array, name))
3703 telem = ast_type_copy(ast_ctx(var), array->expression.next);
3704 tfield = ast_value_new(ast_ctx(var), "<.type>", TYPE_FIELD);
3705 tfield->expression.next = telem;
3706 snprintf(name, sizeof(name), "%s##GETFP", var->name);
3707 if (!parser_create_array_getter(parser, array, (ast_expression*)tfield, name)) {
3715 if (parser->tok == ';') {
3716 ast_delete(basetype);
3717 if (!parser_next(parser)) {
3718 parseerror(parser, "error after variable declaration");
3724 if (parser->tok == ',')
3727 if (!var || (!localblock && !nofields && basetype->expression.vtype == TYPE_FIELD)) {
3728 parseerror(parser, "missing comma or semicolon while parsing variables");
3732 if (localblock && opts_standard == COMPILER_QCC) {
3733 if (parsewarning(parser, WARN_LOCAL_CONSTANTS,
3734 "initializing expression turns variable `%s` into a constant in this standard",
3741 if (parser->tok != '{') {
3742 if (parser->tok != '=') {
3743 parseerror(parser, "missing semicolon or initializer, got: `%s`", parser_tokval(parser));
3747 if (!parser_next(parser)) {
3748 parseerror(parser, "error parsing initializer");
3752 else if (opts_standard == COMPILER_QCC) {
3753 parseerror(parser, "expected '=' before function body in this standard");
3756 if (parser->tok == '#') {
3757 ast_function *func = NULL;
3760 parseerror(parser, "cannot declare builtins within functions");
3763 if (var->expression.vtype != TYPE_FUNCTION) {
3764 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
3767 if (!parser_next(parser)) {
3768 parseerror(parser, "expected builtin number");
3771 if (parser->tok != TOKEN_INTCONST) {
3772 parseerror(parser, "builtin number must be an integer constant");
3775 if (parser_token(parser)->constval.i <= 0) {
3776 parseerror(parser, "builtin number must be an integer greater than zero");
3780 if (var->hasvalue) {
3781 (void)!parsewarning(parser, WARN_DOUBLE_DECLARATION,
3782 "builtin `%s` has already been defined\n"
3783 " -> previous declaration here: %s:%i",
3784 var->name, ast_ctx(var).file, (int)ast_ctx(var).line);
3788 func = ast_function_new(ast_ctx(var), var->name, var);
3790 parseerror(parser, "failed to allocate function for `%s`", var->name);
3793 vec_push(parser->functions, func);
3795 func->builtin = -parser_token(parser)->constval.i;
3798 if (!parser_next(parser)) {
3799 parseerror(parser, "expected comma or semicolon");
3801 ast_function_delete(func);
3802 var->constval.vfunc = NULL;
3806 else if (parser->tok == '{' || parser->tok == '[')
3809 parseerror(parser, "cannot declare functions within functions");
3813 if (!parse_function_body(parser, var))
3815 ast_delete(basetype);
3818 ast_expression *cexp;
3821 cexp = parse_expression_leave(parser, true);
3826 cval = (ast_value*)cexp;
3827 if (!ast_istype(cval, ast_value) || !cval->hasvalue)
3828 parseerror(parser, "cannot initialize a global constant variable with a non-constant expression");
3831 if (opts_standard != COMPILER_GMQCC && !OPTS_FLAG(INITIALIZED_NONCONSTANTS))
3832 var->constant = true;
3833 var->hasvalue = true;
3834 if (cval->expression.vtype == TYPE_STRING)
3835 var->constval.vstring = parser_strdup(cval->constval.vstring);
3837 memcpy(&var->constval, &cval->constval, sizeof(var->constval));
3841 shunt sy = { NULL, NULL };
3842 vec_push(sy.out, syexp(ast_ctx(var), (ast_expression*)var));
3843 vec_push(sy.out, syexp(ast_ctx(cexp), (ast_expression*)cexp));
3844 vec_push(sy.ops, syop(ast_ctx(var), parser->assign_op));
3845 if (!parser_sy_pop(parser, &sy))
3848 if (vec_size(sy.out) != 1 && vec_size(sy.ops) != 0)
3849 parseerror(parser, "internal error: leaked operands");
3850 ast_block_add_expr(localblock, (ast_expression*)sy.out[0].out);
3858 if (parser->tok == ',') {
3859 if (!parser_next(parser)) {
3860 parseerror(parser, "expected another variable");
3864 if (parser->tok != TOKEN_IDENT) {
3865 parseerror(parser, "expected another variable");
3868 var = ast_value_copy(basetype);
3870 ast_value_set_name(var, parser_tokval(parser));
3871 if (!parser_next(parser)) {
3872 parseerror(parser, "error parsing variable declaration");
3878 if (parser->tok != ';') {
3879 parseerror(parser, "missing semicolon after variables");
3883 if (!parser_next(parser)) {
3884 parseerror(parser, "parse error after variable declaration");
3888 ast_delete(basetype);
3892 if (cleanvar && var)
3894 ast_delete(basetype);
3898 ast_delete(basetype);
3899 if (cleanvar && var)
3901 if (me[0]) ast_member_delete(me[0]);
3902 if (me[1]) ast_member_delete(me[1]);
3903 if (me[2]) ast_member_delete(me[2]);
3907 static bool parser_global_statement(parser_t *parser)
3909 ast_value *istype = NULL;
3910 if (parser->tok == TOKEN_IDENT)
3911 istype = parser_find_typedef(parser, parser_tokval(parser), 0);
3913 if (istype || parser->tok == TOKEN_TYPENAME || parser->tok == '.')
3915 return parse_variable(parser, NULL, false, false, istype);
3917 else if (parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "var"))
3919 if (!strcmp(parser_tokval(parser), "var")) {
3920 if (!parser_next(parser)) {
3921 parseerror(parser, "expected variable declaration after 'var'");
3924 return parse_variable(parser, NULL, true, false, NULL);
3927 else if (parser->tok == TOKEN_KEYWORD)
3929 if (!strcmp(parser_tokval(parser), "const")) {
3930 if (!parser_next(parser)) {
3931 parseerror(parser, "expected variable declaration after 'const'");
3934 if (parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "var")) {
3935 (void)!parsewarning(parser, WARN_CONST_VAR, "ignoring `var` after const qualifier");
3936 if (!parser_next(parser)) {
3937 parseerror(parser, "expected variable declaration after 'const var'");
3941 return parse_variable(parser, NULL, true, true, NULL);
3943 else if (!strcmp(parser_tokval(parser), "typedef")) {
3944 if (!parser_next(parser)) {
3945 parseerror(parser, "expected type definition after 'typedef'");
3948 return parse_typedef(parser);
3950 parseerror(parser, "unrecognized keyword `%s`", parser_tokval(parser));
3953 else if (parser->tok == '$')
3955 if (!parser_next(parser)) {
3956 parseerror(parser, "parse error");
3962 parseerror(parser, "unexpected token: %s", parser->lex->tok.value);
3968 static uint16_t progdefs_crc_sum(uint16_t old, const char *str)
3970 return util_crc16(old, str, strlen(str));
3973 static void progdefs_crc_file(const char *str)
3975 /* write to progdefs.h here */
3979 static uint16_t progdefs_crc_both(uint16_t old, const char *str)
3981 old = progdefs_crc_sum(old, str);
3982 progdefs_crc_file(str);
3986 static void generate_checksum(parser_t *parser)
3988 uint16_t crc = 0xFFFF;
3992 crc = progdefs_crc_both(crc, "\n/* file generated by qcc, do not modify */\n\ntypedef struct\n{");
3993 crc = progdefs_crc_sum(crc, "\tint\tpad[28];\n");
3995 progdefs_crc_file("\tint\tpad;\n");
3996 progdefs_crc_file("\tint\tofs_return[3];\n");
3997 progdefs_crc_file("\tint\tofs_parm0[3];\n");
3998 progdefs_crc_file("\tint\tofs_parm1[3];\n");
3999 progdefs_crc_file("\tint\tofs_parm2[3];\n");
4000 progdefs_crc_file("\tint\tofs_parm3[3];\n");
4001 progdefs_crc_file("\tint\tofs_parm4[3];\n");
4002 progdefs_crc_file("\tint\tofs_parm5[3];\n");
4003 progdefs_crc_file("\tint\tofs_parm6[3];\n");
4004 progdefs_crc_file("\tint\tofs_parm7[3];\n");
4006 for (i = 0; i < parser->crc_globals; ++i) {
4007 if (!ast_istype(parser->globals[i], ast_value))
4009 value = (ast_value*)(parser->globals[i]);
4010 switch (value->expression.vtype) {
4011 case TYPE_FLOAT: crc = progdefs_crc_both(crc, "\tfloat\t"); break;
4012 case TYPE_VECTOR: crc = progdefs_crc_both(crc, "\tvec3_t\t"); break;
4013 case TYPE_STRING: crc = progdefs_crc_both(crc, "\tstring_t\t"); break;
4014 case TYPE_FUNCTION: crc = progdefs_crc_both(crc, "\tfunc_t\t"); break;
4016 crc = progdefs_crc_both(crc, "\tint\t");
4019 crc = progdefs_crc_both(crc, value->name);
4020 crc = progdefs_crc_both(crc, ";\n");
4022 crc = progdefs_crc_both(crc, "} globalvars_t;\n\ntypedef struct\n{\n");
4023 for (i = 0; i < parser->crc_fields; ++i) {
4024 if (!ast_istype(parser->fields[i], ast_value))
4026 value = (ast_value*)(parser->fields[i]);
4027 switch (value->expression.next->expression.vtype) {
4028 case TYPE_FLOAT: crc = progdefs_crc_both(crc, "\tfloat\t"); break;
4029 case TYPE_VECTOR: crc = progdefs_crc_both(crc, "\tvec3_t\t"); break;
4030 case TYPE_STRING: crc = progdefs_crc_both(crc, "\tstring_t\t"); break;
4031 case TYPE_FUNCTION: crc = progdefs_crc_both(crc, "\tfunc_t\t"); break;
4033 crc = progdefs_crc_both(crc, "\tint\t");
4036 crc = progdefs_crc_both(crc, value->name);
4037 crc = progdefs_crc_both(crc, ";\n");
4039 crc = progdefs_crc_both(crc, "} entvars_t;\n\n");
4044 static parser_t *parser;
4050 parser = (parser_t*)mem_a(sizeof(parser_t));
4054 memset(parser, 0, sizeof(*parser));
4056 for (i = 0; i < operator_count; ++i) {
4057 if (operators[i].id == opid1('=')) {
4058 parser->assign_op = operators+i;
4062 if (!parser->assign_op) {
4063 printf("internal error: initializing parser: failed to find assign operator\n");
4068 vec_push(parser->variables, parser->htfields = util_htnew(PARSER_HT_SIZE));
4069 vec_push(parser->variables, parser->htglobals = util_htnew(PARSER_HT_SIZE));
4070 vec_push(parser->typedefs, util_htnew(TYPEDEF_HT_SIZE));
4071 vec_push(parser->_blocktypedefs, 0);
4075 bool parser_compile()
4077 /* initial lexer/parser state */
4078 parser->lex->flags.noops = true;
4080 if (parser_next(parser))
4082 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
4084 if (!parser_global_statement(parser)) {
4085 if (parser->tok == TOKEN_EOF)
4086 parseerror(parser, "unexpected eof");
4087 else if (!parser->errors)
4088 parseerror(parser, "there have been errors, bailing out");
4089 lex_close(parser->lex);
4095 parseerror(parser, "parse error");
4096 lex_close(parser->lex);
4101 lex_close(parser->lex);
4104 return !parser->errors;
4107 bool parser_compile_file(const char *filename)
4109 parser->lex = lex_open(filename);
4111 con_err("failed to open file \"%s\"\n", filename);
4114 return parser_compile();
4117 bool parser_compile_string_len(const char *name, const char *str, size_t len)
4119 parser->lex = lex_open_string(str, len, name);
4121 con_err("failed to create lexer for string \"%s\"\n", name);
4124 return parser_compile();
4127 bool parser_compile_string(const char *name, const char *str)
4129 parser->lex = lex_open_string(str, strlen(str), name);
4131 con_err("failed to create lexer for string \"%s\"\n", name);
4134 return parser_compile();
4137 void parser_cleanup()
4140 for (i = 0; i < vec_size(parser->accessors); ++i) {
4141 ast_delete(parser->accessors[i]->constval.vfunc);
4142 parser->accessors[i]->constval.vfunc = NULL;
4143 ast_delete(parser->accessors[i]);
4145 for (i = 0; i < vec_size(parser->functions); ++i) {
4146 ast_delete(parser->functions[i]);
4148 for (i = 0; i < vec_size(parser->imm_vector); ++i) {
4149 ast_delete(parser->imm_vector[i]);
4151 for (i = 0; i < vec_size(parser->imm_string); ++i) {
4152 ast_delete(parser->imm_string[i]);
4154 for (i = 0; i < vec_size(parser->imm_float); ++i) {
4155 ast_delete(parser->imm_float[i]);
4157 for (i = 0; i < vec_size(parser->fields); ++i) {
4158 ast_delete(parser->fields[i]);
4160 for (i = 0; i < vec_size(parser->globals); ++i) {
4161 ast_delete(parser->globals[i]);
4163 vec_free(parser->accessors);
4164 vec_free(parser->functions);
4165 vec_free(parser->imm_vector);
4166 vec_free(parser->imm_string);
4167 vec_free(parser->imm_float);
4168 vec_free(parser->globals);
4169 vec_free(parser->fields);
4171 for (i = 0; i < vec_size(parser->variables); ++i)
4172 util_htdel(parser->variables[i]);
4173 vec_free(parser->variables);
4174 vec_free(parser->_blocklocals);
4175 vec_free(parser->_locals);
4177 for (i = 0; i < vec_size(parser->_typedefs); ++i)
4178 ast_delete(parser->_typedefs[i]);
4179 vec_free(parser->_typedefs);
4180 for (i = 0; i < vec_size(parser->typedefs); ++i)
4181 util_htdel(parser->typedefs[i]);
4182 vec_free(parser->typedefs);
4183 vec_free(parser->_blocktypedefs);
4188 bool parser_finish(const char *output)
4194 if (!parser->errors)
4196 ir = ir_builder_new("gmqcc_out");
4198 con_out("failed to allocate builder\n");
4202 for (i = 0; i < vec_size(parser->fields); ++i) {
4205 if (!ast_istype(parser->fields[i], ast_value))
4207 field = (ast_value*)parser->fields[i];
4208 hasvalue = field->hasvalue;
4209 field->hasvalue = false;
4210 if (!ast_global_codegen((ast_value*)field, ir, true)) {
4211 con_out("failed to generate field %s\n", field->name);
4212 ir_builder_delete(ir);
4217 ast_expression *subtype;
4218 field->hasvalue = true;
4219 subtype = field->expression.next;
4220 ifld = ir_builder_create_field(ir, field->name, subtype->expression.vtype);
4221 if (subtype->expression.vtype == TYPE_FIELD)
4222 ifld->fieldtype = subtype->expression.next->expression.vtype;
4223 else if (subtype->expression.vtype == TYPE_FUNCTION)
4224 ifld->outtype = subtype->expression.next->expression.vtype;
4225 (void)!ir_value_set_field(field->ir_v, ifld);
4228 for (i = 0; i < vec_size(parser->globals); ++i) {
4230 if (!ast_istype(parser->globals[i], ast_value))
4232 asvalue = (ast_value*)(parser->globals[i]);
4233 if (!asvalue->uses && !asvalue->hasvalue && asvalue->expression.vtype != TYPE_FUNCTION) {
4234 if (strcmp(asvalue->name, "end_sys_globals") &&
4235 strcmp(asvalue->name, "end_sys_fields"))
4237 retval = retval && !genwarning(ast_ctx(asvalue), WARN_UNUSED_VARIABLE,
4238 "unused global: `%s`", asvalue->name);
4241 if (!ast_global_codegen(asvalue, ir, false)) {
4242 con_out("failed to generate global %s\n", asvalue->name);
4243 ir_builder_delete(ir);
4247 for (i = 0; i < vec_size(parser->imm_float); ++i) {
4248 if (!ast_global_codegen(parser->imm_float[i], ir, false)) {
4249 con_out("failed to generate global %s\n", parser->imm_float[i]->name);
4250 ir_builder_delete(ir);
4254 for (i = 0; i < vec_size(parser->imm_string); ++i) {
4255 if (!ast_global_codegen(parser->imm_string[i], ir, false)) {
4256 con_out("failed to generate global %s\n", parser->imm_string[i]->name);
4257 ir_builder_delete(ir);
4261 for (i = 0; i < vec_size(parser->imm_vector); ++i) {
4262 if (!ast_global_codegen(parser->imm_vector[i], ir, false)) {
4263 con_out("failed to generate global %s\n", parser->imm_vector[i]->name);
4264 ir_builder_delete(ir);
4268 for (i = 0; i < vec_size(parser->globals); ++i) {
4270 if (!ast_istype(parser->globals[i], ast_value))
4272 asvalue = (ast_value*)(parser->globals[i]);
4273 if (asvalue->setter) {
4274 if (!ast_global_codegen(asvalue->setter, ir, false) ||
4275 !ast_function_codegen(asvalue->setter->constval.vfunc, ir) ||
4276 !ir_function_finalize(asvalue->setter->constval.vfunc->ir_func))
4278 printf("failed to generate setter for %s\n", asvalue->name);
4279 ir_builder_delete(ir);
4283 if (asvalue->getter) {
4284 if (!ast_global_codegen(asvalue->getter, ir, false) ||
4285 !ast_function_codegen(asvalue->getter->constval.vfunc, ir) ||
4286 !ir_function_finalize(asvalue->getter->constval.vfunc->ir_func))
4288 printf("failed to generate getter for %s\n", asvalue->name);
4289 ir_builder_delete(ir);
4294 for (i = 0; i < vec_size(parser->fields); ++i) {
4296 asvalue = (ast_value*)(parser->fields[i]->expression.next);
4298 if (!ast_istype((ast_expression*)asvalue, ast_value))
4300 if (asvalue->expression.vtype != TYPE_ARRAY)
4302 if (asvalue->setter) {
4303 if (!ast_global_codegen(asvalue->setter, ir, false) ||
4304 !ast_function_codegen(asvalue->setter->constval.vfunc, ir) ||
4305 !ir_function_finalize(asvalue->setter->constval.vfunc->ir_func))
4307 printf("failed to generate setter for %s\n", asvalue->name);
4308 ir_builder_delete(ir);
4312 if (asvalue->getter) {
4313 if (!ast_global_codegen(asvalue->getter, ir, false) ||
4314 !ast_function_codegen(asvalue->getter->constval.vfunc, ir) ||
4315 !ir_function_finalize(asvalue->getter->constval.vfunc->ir_func))
4317 printf("failed to generate getter for %s\n", asvalue->name);
4318 ir_builder_delete(ir);
4323 for (i = 0; i < vec_size(parser->functions); ++i) {
4324 if (!ast_function_codegen(parser->functions[i], ir)) {
4325 con_out("failed to generate function %s\n", parser->functions[i]->name);
4326 ir_builder_delete(ir);
4331 ir_builder_dump(ir, con_out);
4332 for (i = 0; i < vec_size(parser->functions); ++i) {
4333 if (!ir_function_finalize(parser->functions[i]->ir_func)) {
4334 con_out("failed to finalize function %s\n", parser->functions[i]->name);
4335 ir_builder_delete(ir);
4342 ir_builder_dump(ir, con_out);
4344 generate_checksum(parser);
4346 if (!ir_builder_generate(ir, output)) {
4347 con_out("*** failed to generate output file\n");
4348 ir_builder_delete(ir);
4353 ir_builder_delete(ir);
4357 con_out("*** there were compile errors\n");