18 ast_function **functions;
19 ast_value **imm_float;
20 ast_value **imm_string;
21 ast_value **imm_vector;
23 /* must be deleted first, they reference immediates and values */
24 ast_value **accessors;
26 ast_value *imm_float_zero;
27 ast_value *imm_vector_zero;
32 ast_function *function;
38 /* we store the '=' operator info */
39 const oper_info *assign_op;
41 /* TYPE_FIELD -> parser_find_fields is used instead of find_var
42 * TODO: TYPE_VECTOR -> x, y and z are accepted in the gmqcc standard
43 * anything else: type error
49 static bool GMQCC_WARN parser_pop_local(parser_t *parser);
50 static bool parse_variable(parser_t *parser, ast_block *localblock, bool nofields);
51 static ast_block* parse_block(parser_t *parser, bool warnreturn);
52 static bool parse_block_into(parser_t *parser, ast_block *block, bool warnreturn);
53 static ast_expression* parse_statement_or_block(parser_t *parser);
54 static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases);
55 static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma);
56 static ast_expression* parse_expression(parser_t *parser, bool stopatcomma);
58 static void parseerror(parser_t *parser, const char *fmt, ...)
65 con_vprintmsg(LVL_ERROR, parser->lex->tok.ctx.file, parser->lex->tok.ctx.line, "parse error", fmt, ap);
69 /* returns true if it counts as an error */
70 static bool GMQCC_WARN parsewarning(parser_t *parser, int warntype, const char *fmt, ...)
73 int lvl = LVL_WARNING;
75 if (!OPTS_WARN(warntype))
84 con_vprintmsg(lvl, parser->lex->tok.ctx.file, parser->lex->tok.ctx.line, "warning", fmt, ap);
90 static bool GMQCC_WARN genwarning(lex_ctx ctx, int warntype, const char *fmt, ...)
93 int lvl = LVL_WARNING;
95 if (!OPTS_WARN(warntype))
102 con_vprintmsg(lvl, ctx.file, ctx.line, "warning", fmt, ap);
108 /**********************************************************************
109 * some maths used for constant folding
112 vector vec3_add(vector a, vector b)
121 vector vec3_sub(vector a, vector b)
130 qcfloat vec3_mulvv(vector a, vector b)
132 return (a.x * b.x + a.y * b.y + a.z * b.z);
135 vector vec3_mulvf(vector a, float b)
144 /**********************************************************************
148 bool parser_next(parser_t *parser)
150 /* lex_do kills the previous token */
151 parser->tok = lex_do(parser->lex);
152 if (parser->tok == TOKEN_EOF)
154 if (parser->tok >= TOKEN_ERROR) {
155 parseerror(parser, "lex error");
161 #define parser_tokval(p) ((p)->lex->tok.value)
162 #define parser_token(p) (&((p)->lex->tok))
163 #define parser_ctx(p) ((p)->lex->tok.ctx)
165 static ast_value* parser_const_float(parser_t *parser, double d)
169 for (i = 0; i < vec_size(parser->imm_float); ++i) {
170 const double compare = parser->imm_float[i]->constval.vfloat;
171 if (memcmp((const void*)&compare, (const void *)&d, sizeof(double)) == 0)
172 return parser->imm_float[i];
174 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_FLOAT);
176 out->constval.vfloat = d;
177 vec_push(parser->imm_float, out);
181 static ast_value* parser_const_float_0(parser_t *parser)
183 if (!parser->imm_float_zero)
184 parser->imm_float_zero = parser_const_float(parser, 0);
185 return parser->imm_float_zero;
188 static char *parser_strdup(const char *str)
191 /* actually dup empty strings */
192 char *out = mem_a(1);
196 return util_strdup(str);
199 static ast_value* parser_const_string(parser_t *parser, const char *str)
203 for (i = 0; i < vec_size(parser->imm_string); ++i) {
204 if (!strcmp(parser->imm_string[i]->constval.vstring, str))
205 return parser->imm_string[i];
207 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_STRING);
209 out->constval.vstring = parser_strdup(str);
210 vec_push(parser->imm_string, out);
214 static ast_value* parser_const_vector(parser_t *parser, vector v)
218 for (i = 0; i < vec_size(parser->imm_vector); ++i) {
219 if (!memcmp(&parser->imm_vector[i]->constval.vvec, &v, sizeof(v)))
220 return parser->imm_vector[i];
222 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_VECTOR);
224 out->constval.vvec = v;
225 vec_push(parser->imm_vector, out);
229 static ast_value* parser_const_vector_f(parser_t *parser, float x, float y, float z)
235 return parser_const_vector(parser, v);
238 static ast_value* parser_const_vector_0(parser_t *parser)
240 if (!parser->imm_vector_zero)
241 parser->imm_vector_zero = parser_const_vector_f(parser, 0, 0, 0);
242 return parser->imm_vector_zero;
245 static ast_expression* parser_find_field(parser_t *parser, const char *name)
248 for (i = 0; i < vec_size(parser->fields); ++i) {
249 if (!strcmp(parser->fields[i].name, name))
250 return parser->fields[i].var;
255 static ast_expression* parser_find_global(parser_t *parser, const char *name)
258 for (i = 0; i < vec_size(parser->globals); ++i) {
259 if (!strcmp(parser->globals[i].name, name))
260 return parser->globals[i].var;
265 static ast_expression* parser_find_param(parser_t *parser, const char *name)
269 if (!parser->function)
271 fun = parser->function->vtype;
272 for (i = 0; i < vec_size(fun->expression.params); ++i) {
273 if (!strcmp(fun->expression.params[i]->name, name))
274 return (ast_expression*)(fun->expression.params[i]);
279 static ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto, bool *isparam)
283 for (i = vec_size(parser->locals); i > upto;) {
285 if (!strcmp(parser->locals[i].name, name))
286 return parser->locals[i].var;
289 return parser_find_param(parser, name);
292 static ast_expression* parser_find_var(parser_t *parser, const char *name)
296 v = parser_find_local(parser, name, 0, &dummy);
297 if (!v) v = parser_find_global(parser, name);
303 size_t etype; /* 0 = expression, others are operators */
307 ast_block *block; /* for commas and function calls */
316 #define SY_PAREN_EXPR '('
317 #define SY_PAREN_FUNC 'f'
318 #define SY_PAREN_INDEX '['
320 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
331 static sy_elem syblock(lex_ctx ctx, ast_block *v) {
335 e.out = (ast_expression*)v;
342 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
344 e.etype = 1 + (op - operators);
353 static sy_elem syparen(lex_ctx ctx, int p, size_t off) {
365 # define DEBUGSHUNTDO(x) x
367 # define DEBUGSHUNTDO(x)
370 /* With regular precedence rules, ent.foo[n] is the same as (ent.foo)[n],
371 * so we need to rotate it to become ent.(foo[n]).
373 static bool rotate_entfield_array_index_nodes(ast_expression **out)
375 ast_array_index *index;
376 ast_entfield *entfield;
380 ast_expression *entity;
382 lex_ctx ctx = ast_ctx(*out);
384 if (!ast_istype(*out, ast_array_index))
386 index = (ast_array_index*)*out;
388 if (!ast_istype(index->array, ast_entfield))
390 entfield = (ast_entfield*)index->array;
392 if (!ast_istype(entfield->field, ast_value))
394 field = (ast_value*)entfield->field;
397 entity = entfield->entity;
401 index = ast_array_index_new(ctx, (ast_expression*)field, sub);
402 entfield = ast_entfield_new(ctx, entity, (ast_expression*)index);
403 *out = (ast_expression*)entfield;
408 static bool parser_sy_pop(parser_t *parser, shunt *sy)
412 ast_expression *out = NULL;
413 ast_expression *exprs[3];
414 ast_block *blocks[3];
415 ast_value *asvalue[3];
416 size_t i, assignop, addop;
417 qcint generated_op = 0;
422 if (!vec_size(sy->ops)) {
423 parseerror(parser, "internal error: missing operator");
427 if (vec_last(sy->ops).paren) {
428 parseerror(parser, "unmatched parenthesis");
432 op = &operators[vec_last(sy->ops).etype - 1];
433 ctx = vec_last(sy->ops).ctx;
435 DEBUGSHUNTDO(con_out("apply %s\n", op->op));
437 if (vec_size(sy->out) < op->operands) {
438 parseerror(parser, "internal error: not enough operands: %i (operator %s (%i))", vec_size(sy->out),
439 op->op, (int)op->id);
443 vec_shrinkby(sy->ops, 1);
445 vec_shrinkby(sy->out, op->operands);
446 for (i = 0; i < op->operands; ++i) {
447 exprs[i] = sy->out[vec_size(sy->out)+i].out;
448 blocks[i] = sy->out[vec_size(sy->out)+i].block;
449 asvalue[i] = (ast_value*)exprs[i];
452 if (blocks[0] && !vec_size(blocks[0]->exprs) && op->id != opid1(',')) {
453 parseerror(parser, "internal error: operator cannot be applied on empty blocks");
457 #define NotSameType(T) \
458 (exprs[0]->expression.vtype != exprs[1]->expression.vtype || \
459 exprs[0]->expression.vtype != T)
460 #define CanConstFold1(A) \
461 (ast_istype((A), ast_value) && ((ast_value*)(A))->isconst)
462 #define CanConstFold(A, B) \
463 (CanConstFold1(A) && CanConstFold1(B))
464 #define ConstV(i) (asvalue[(i)]->constval.vvec)
465 #define ConstF(i) (asvalue[(i)]->constval.vfloat)
466 #define ConstS(i) (asvalue[(i)]->constval.vstring)
470 parseerror(parser, "internal error: unhandled operator: %s (%i)", op->op, (int)op->id);
474 if (exprs[0]->expression.vtype == TYPE_ENTITY) {
475 if (exprs[1]->expression.vtype != TYPE_FIELD) {
476 parseerror(parser, "type error: right hand of member-operand should be an entity-field");
479 out = (ast_expression*)ast_entfield_new(ctx, exprs[0], exprs[1]);
481 else if (exprs[0]->expression.vtype == TYPE_VECTOR) {
482 parseerror(parser, "internal error: vector access is not supposed to be handled at this point");
486 parseerror(parser, "type error: member-of operator on something that is not an entity or vector");
492 if (exprs[0]->expression.vtype != TYPE_ARRAY &&
493 !(exprs[0]->expression.vtype == TYPE_FIELD &&
494 exprs[0]->expression.next->expression.vtype == TYPE_ARRAY))
496 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
497 parseerror(parser, "cannot index value of type %s", ty1);
500 if (exprs[1]->expression.vtype != TYPE_FLOAT) {
501 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
502 parseerror(parser, "index must be of type float, not %s", ty1);
505 out = (ast_expression*)ast_array_index_new(ctx, exprs[0], exprs[1]);
506 if (rotate_entfield_array_index_nodes(&out))
508 if (opts_standard != COMPILER_GMQCC) {
509 /* this error doesn't need to make us bail out */
510 (void)!parsewarning(parser, WARN_EXTENSIONS,
511 "accessing array-field members of an entity without parenthesis\n"
512 " -> this is an extension from -std=gmqcc");
519 vec_push(blocks[0]->exprs, exprs[1]);
521 blocks[0] = ast_block_new(ctx);
522 vec_push(blocks[0]->exprs, exprs[0]);
523 vec_push(blocks[0]->exprs, exprs[1]);
525 if (!ast_block_set_type(blocks[0], exprs[1]))
528 vec_push(sy->out, syblock(ctx, blocks[0]));
532 switch (exprs[0]->expression.vtype) {
534 if (CanConstFold1(exprs[0]))
535 out = (ast_expression*)parser_const_float(parser, -ConstF(0));
537 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F,
538 (ast_expression*)parser_const_float_0(parser),
542 if (CanConstFold1(exprs[0]))
543 out = (ast_expression*)parser_const_vector_f(parser,
544 -ConstV(0).x, -ConstV(0).y, -ConstV(0).z);
546 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V,
547 (ast_expression*)parser_const_vector_0(parser),
551 parseerror(parser, "invalid types used in expression: cannot negate type %s",
552 type_name[exprs[0]->expression.vtype]);
558 switch (exprs[0]->expression.vtype) {
560 if (CanConstFold1(exprs[0]))
561 out = (ast_expression*)parser_const_float(parser, !ConstF(0));
563 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, exprs[0]);
566 if (CanConstFold1(exprs[0]))
567 out = (ast_expression*)parser_const_float(parser,
568 (!ConstV(0).x && !ConstV(0).y && !ConstV(0).z));
570 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_V, exprs[0]);
573 if (CanConstFold1(exprs[0]))
574 out = (ast_expression*)parser_const_float(parser, !ConstS(0) || !*ConstS(0));
576 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_S, exprs[0]);
578 /* we don't constant-fold NOT for these types */
580 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_ENT, exprs[0]);
583 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_FNC, exprs[0]);
586 parseerror(parser, "invalid types used in expression: cannot logically negate type %s",
587 type_name[exprs[0]->expression.vtype]);
593 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
594 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
596 parseerror(parser, "invalid types used in expression: cannot add type %s and %s",
597 type_name[exprs[0]->expression.vtype],
598 type_name[exprs[1]->expression.vtype]);
601 switch (exprs[0]->expression.vtype) {
603 if (CanConstFold(exprs[0], exprs[1]))
605 out = (ast_expression*)parser_const_float(parser, ConstF(0) + ConstF(1));
608 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
611 if (CanConstFold(exprs[0], exprs[1]))
612 out = (ast_expression*)parser_const_vector(parser, vec3_add(ConstV(0), ConstV(1)));
614 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
617 parseerror(parser, "invalid types used in expression: cannot add type %s and %s",
618 type_name[exprs[0]->expression.vtype],
619 type_name[exprs[1]->expression.vtype]);
624 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
625 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
627 parseerror(parser, "invalid types used in expression: cannot subtract type %s from %s",
628 type_name[exprs[1]->expression.vtype],
629 type_name[exprs[0]->expression.vtype]);
632 switch (exprs[0]->expression.vtype) {
634 if (CanConstFold(exprs[0], exprs[1]))
635 out = (ast_expression*)parser_const_float(parser, ConstF(0) - ConstF(1));
637 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
640 if (CanConstFold(exprs[0], exprs[1]))
641 out = (ast_expression*)parser_const_vector(parser, vec3_sub(ConstV(0), ConstV(1)));
643 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
646 parseerror(parser, "invalid types used in expression: cannot subtract type %s from %s",
647 type_name[exprs[1]->expression.vtype],
648 type_name[exprs[0]->expression.vtype]);
653 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
654 exprs[0]->expression.vtype != TYPE_VECTOR &&
655 exprs[0]->expression.vtype != TYPE_FLOAT &&
656 exprs[1]->expression.vtype != TYPE_VECTOR &&
657 exprs[1]->expression.vtype != TYPE_FLOAT)
659 parseerror(parser, "invalid types used in expression: cannot multiply types %s and %s",
660 type_name[exprs[1]->expression.vtype],
661 type_name[exprs[0]->expression.vtype]);
664 switch (exprs[0]->expression.vtype) {
666 if (exprs[1]->expression.vtype == TYPE_VECTOR)
668 if (CanConstFold(exprs[0], exprs[1]))
669 out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(1), ConstF(0)));
671 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
675 if (CanConstFold(exprs[0], exprs[1]))
676 out = (ast_expression*)parser_const_float(parser, ConstF(0) * ConstF(1));
678 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
682 if (exprs[1]->expression.vtype == TYPE_FLOAT)
684 if (CanConstFold(exprs[0], exprs[1]))
685 out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(0), ConstF(1)));
687 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
691 if (CanConstFold(exprs[0], exprs[1]))
692 out = (ast_expression*)parser_const_float(parser, vec3_mulvv(ConstV(0), ConstV(1)));
694 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
698 parseerror(parser, "invalid types used in expression: cannot multiply types %s and %s",
699 type_name[exprs[1]->expression.vtype],
700 type_name[exprs[0]->expression.vtype]);
705 if (NotSameType(TYPE_FLOAT)) {
706 parseerror(parser, "invalid types used in expression: cannot divide types %s and %s",
707 type_name[exprs[0]->expression.vtype],
708 type_name[exprs[1]->expression.vtype]);
711 if (CanConstFold(exprs[0], exprs[1]))
712 out = (ast_expression*)parser_const_float(parser, ConstF(0) / ConstF(1));
714 out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
718 parseerror(parser, "qc does not have a modulo operator");
722 if (NotSameType(TYPE_FLOAT)) {
723 parseerror(parser, "invalid types used in expression: cannot perform bit operations between types %s and %s",
724 type_name[exprs[0]->expression.vtype],
725 type_name[exprs[1]->expression.vtype]);
728 if (CanConstFold(exprs[0], exprs[1]))
729 out = (ast_expression*)parser_const_float(parser,
730 (op->id == opid1('|') ? (float)( ((qcint)ConstF(0)) | ((qcint)ConstF(1)) ) :
731 (float)( ((qcint)ConstF(0)) & ((qcint)ConstF(1)) ) ));
733 out = (ast_expression*)ast_binary_new(ctx,
734 (op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
738 parseerror(parser, "TODO: bitxor");
743 case opid3('<','<','='):
744 case opid3('>','>','='):
745 parseerror(parser, "TODO: shifts");
749 generated_op += 1; /* INSTR_OR */
751 generated_op += INSTR_AND;
752 if (NotSameType(TYPE_FLOAT)) {
753 parseerror(parser, "invalid types used in expression: cannot perform logical operations between types %s and %s",
754 type_name[exprs[0]->expression.vtype],
755 type_name[exprs[1]->expression.vtype]);
756 parseerror(parser, "TODO: logical ops for arbitrary types using INSTR_NOT");
757 parseerror(parser, "TODO: optional early out");
760 if (opts_standard == COMPILER_GMQCC)
761 con_out("TODO: early out logic\n");
762 if (CanConstFold(exprs[0], exprs[1]))
763 out = (ast_expression*)parser_const_float(parser,
764 (generated_op == INSTR_OR ? (ConstF(0) || ConstF(1)) : (ConstF(0) && ConstF(1))));
766 out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
770 if (exprs[1]->expression.vtype != exprs[2]->expression.vtype) {
771 ast_type_to_string(exprs[1], ty1, sizeof(ty1));
772 ast_type_to_string(exprs[2], ty2, sizeof(ty2));
773 parseerror(parser, "iperands of ternary expression must have the same type, got %s and %s", ty1, ty2);
776 if (CanConstFold1(exprs[0]))
777 out = (ConstF(0) ? exprs[1] : exprs[2]);
779 out = (ast_expression*)ast_ternary_new(ctx, exprs[0], exprs[1], exprs[2]);
783 generated_op += 1; /* INSTR_GT */
785 generated_op += 1; /* INSTR_LT */
786 case opid2('>', '='):
787 generated_op += 1; /* INSTR_GE */
788 case opid2('<', '='):
789 generated_op += INSTR_LE;
790 if (NotSameType(TYPE_FLOAT)) {
791 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
792 type_name[exprs[0]->expression.vtype],
793 type_name[exprs[1]->expression.vtype]);
796 out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
798 case opid2('!', '='):
799 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
800 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
801 type_name[exprs[0]->expression.vtype],
802 type_name[exprs[1]->expression.vtype]);
805 out = (ast_expression*)ast_binary_new(ctx, type_ne_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
807 case opid2('=', '='):
808 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
809 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
810 type_name[exprs[0]->expression.vtype],
811 type_name[exprs[1]->expression.vtype]);
814 out = (ast_expression*)ast_binary_new(ctx, type_eq_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
818 if (ast_istype(exprs[0], ast_entfield)) {
819 ast_expression *field = ((ast_entfield*)exprs[0])->field;
820 if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) &&
821 exprs[0]->expression.vtype == TYPE_FIELD &&
822 exprs[0]->expression.next->expression.vtype == TYPE_VECTOR)
824 assignop = type_storep_instr[TYPE_VECTOR];
827 assignop = type_storep_instr[exprs[0]->expression.vtype];
828 if (!ast_compare_type(field->expression.next, exprs[1])) {
829 ast_type_to_string(field->expression.next, ty1, sizeof(ty1));
830 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
831 if (opts_standard == COMPILER_QCC &&
832 field->expression.next->expression.vtype == TYPE_FUNCTION &&
833 exprs[1]->expression.vtype == TYPE_FUNCTION)
835 if (parsewarning(parser, WARN_ASSIGN_FUNCTION_TYPES,
836 "invalid types in assignment: cannot assign %s to %s", ty2, ty1))
842 parseerror(parser, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
847 if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) &&
848 exprs[0]->expression.vtype == TYPE_FIELD &&
849 exprs[0]->expression.next->expression.vtype == TYPE_VECTOR)
851 assignop = type_store_instr[TYPE_VECTOR];
854 assignop = type_store_instr[exprs[0]->expression.vtype];
857 if (assignop == AINSTR_END) {
858 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
859 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
860 parseerror(parser, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
862 else if (!ast_compare_type(exprs[0], exprs[1])) {
863 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
864 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
865 if (opts_standard == COMPILER_QCC &&
866 exprs[0]->expression.vtype == TYPE_FUNCTION &&
867 exprs[1]->expression.vtype == TYPE_FUNCTION)
869 if (parsewarning(parser, WARN_ASSIGN_FUNCTION_TYPES,
870 "invalid types in assignment: cannot assign %s to %s", ty2, ty1))
876 parseerror(parser, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
879 out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]);
881 case opid3('+','+','P'):
882 case opid3('-','-','P'):
884 if (exprs[0]->expression.vtype != TYPE_FLOAT) {
885 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
886 parseerror(parser, "invalid type for prefix increment: %s", ty1);
889 if (op->id == opid3('+','+','P'))
893 if (ast_istype(exprs[0], ast_entfield)) {
894 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop,
896 (ast_expression*)parser_const_float(parser, 1));
898 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop,
900 (ast_expression*)parser_const_float(parser, 1));
903 case opid3('S','+','+'):
904 case opid3('S','-','-'):
906 if (exprs[0]->expression.vtype != TYPE_FLOAT) {
907 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
908 parseerror(parser, "invalid type for prefix increment: %s", ty1);
911 if (op->id == opid3('+','+','P'))
915 if (ast_istype(exprs[0], ast_entfield)) {
916 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop,
918 (ast_expression*)parser_const_float(parser, 1));
920 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop,
922 (ast_expression*)parser_const_float(parser, 1));
927 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
928 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
930 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
931 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
932 parseerror(parser, "invalid types used in expression: cannot add or subtract type %s and %s",
936 if (ast_istype(exprs[0], ast_entfield))
937 assignop = type_storep_instr[exprs[0]->expression.vtype];
939 assignop = type_store_instr[exprs[0]->expression.vtype];
940 switch (exprs[0]->expression.vtype) {
942 out = (ast_expression*)ast_binstore_new(ctx, assignop,
943 (op->id == opid2('+','=') ? INSTR_ADD_F : INSTR_SUB_F),
947 out = (ast_expression*)ast_binstore_new(ctx, assignop,
948 (op->id == opid2('+','=') ? INSTR_ADD_V : INSTR_SUB_V),
952 parseerror(parser, "invalid types used in expression: cannot add or subtract type %s and %s",
953 type_name[exprs[0]->expression.vtype],
954 type_name[exprs[1]->expression.vtype]);
962 parseerror(parser, "failed to apply operand %s", op->op);
966 DEBUGSHUNTDO(con_out("applied %s\n", op->op));
967 vec_push(sy->out, syexp(ctx, out));
971 static bool parser_close_call(parser_t *parser, shunt *sy)
973 /* was a function call */
980 vec_shrinkby(sy->ops, 1);
981 fid = sy->ops[vec_size(sy->ops)].off;
983 /* out[fid] is the function
984 * everything above is parameters...
986 * 1 params = ast_expression
990 if (vec_size(sy->out) < 1 || vec_size(sy->out) <= fid) {
991 parseerror(parser, "internal error: function call needs function and parameter list...");
995 fun = sy->out[fid].out;
997 call = ast_call_new(sy->ops[vec_size(sy->ops)].ctx, fun);
999 parseerror(parser, "out of memory");
1003 if (fid+1 == vec_size(sy->out)) {
1006 } else if (fid+2 == vec_size(sy->out)) {
1008 vec_shrinkby(sy->out, 1);
1009 params = sy->out[vec_size(sy->out)].block;
1013 vec_push(call->params, sy->out[vec_size(sy->out)].out);
1015 paramcount = vec_size(params->exprs);
1016 call->params = params->exprs;
1017 params->exprs = NULL;
1020 if (!ast_call_check_types(call))
1023 parseerror(parser, "invalid function call");
1027 /* overwrite fid, the function, with a call */
1028 sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
1030 if (fun->expression.vtype != TYPE_FUNCTION) {
1031 parseerror(parser, "not a function (%s)", type_name[fun->expression.vtype]);
1035 if (!fun->expression.next) {
1036 parseerror(parser, "could not determine function return type");
1039 if (vec_size(fun->expression.params) != paramcount &&
1040 !(fun->expression.variadic &&
1041 vec_size(fun->expression.params) < paramcount))
1044 const char *fewmany = (vec_size(fun->expression.params) > paramcount) ? "few" : "many";
1046 fval = (ast_istype(fun, ast_value) ? ((ast_value*)fun) : NULL);
1047 if (opts_standard == COMPILER_GMQCC)
1050 parseerror(parser, "too %s parameters for call to %s: expected %i, got %i\n"
1051 " -> `%s` has been declared here: %s:%i",
1052 fewmany, fval->name, (int)vec_size(fun->expression.params), (int)paramcount,
1053 fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line);
1055 parseerror(parser, "too %s parameters for function call: expected %i, got %i\n"
1056 " -> `%s` has been declared here: %s:%i",
1057 fewmany, fval->name, (int)vec_size(fun->expression.params), (int)paramcount,
1058 fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line);
1064 return !parsewarning(parser, WARN_TOO_FEW_PARAMETERS,
1065 "too %s parameters for call to %s: expected %i, got %i\n"
1066 " -> `%s` has been declared here: %s:%i",
1067 fewmany, fval->name, (int)vec_size(fun->expression.params), (int)paramcount,
1068 fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line);
1070 return !parsewarning(parser, WARN_TOO_FEW_PARAMETERS,
1071 "too %s parameters for function call: expected %i, got %i\n"
1072 " -> `%s` has been declared here: %s:%i",
1073 fewmany, fval->name, (int)vec_size(fun->expression.params), (int)paramcount,
1074 fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line);
1082 static bool parser_close_paren(parser_t *parser, shunt *sy, bool functions_only)
1084 if (!vec_size(sy->ops)) {
1085 parseerror(parser, "unmatched closing paren");
1088 /* this would for bit a + (x) because there are no operators inside (x)
1089 if (sy->ops[vec_size(sy->ops)-1].paren == 1) {
1090 parseerror(parser, "empty parenthesis expression");
1094 while (vec_size(sy->ops)) {
1095 if (sy->ops[vec_size(sy->ops)-1].paren == SY_PAREN_FUNC) {
1096 if (!parser_close_call(parser, sy))
1100 if (sy->ops[vec_size(sy->ops)-1].paren == SY_PAREN_EXPR) {
1101 vec_shrinkby(sy->ops, 1);
1102 return !functions_only;
1104 if (sy->ops[vec_size(sy->ops)-1].paren == SY_PAREN_INDEX) {
1107 /* pop off the parenthesis */
1108 vec_shrinkby(sy->ops, 1);
1109 /* then apply the index operator */
1110 if (!parser_sy_pop(parser, sy))
1114 if (!parser_sy_pop(parser, sy))
1120 static void parser_reclassify_token(parser_t *parser)
1123 for (i = 0; i < operator_count; ++i) {
1124 if (!strcmp(parser_tokval(parser), operators[i].op)) {
1125 parser->tok = TOKEN_OPERATOR;
1131 static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma)
1133 ast_expression *expr = NULL;
1135 bool wantop = false;
1136 bool gotmemberof = false;
1138 /* count the parens because an if starts with one, so the
1139 * end of a condition is an unmatched closing paren
1147 parser->lex->flags.noops = false;
1149 parser_reclassify_token(parser);
1154 gotmemberof = false;
1156 parser->memberof = 0;
1158 if (parser->tok == TOKEN_IDENT)
1160 ast_expression *var;
1162 parseerror(parser, "expected operator or end of statement");
1167 if (opts_standard == COMPILER_GMQCC)
1169 if (parser->memberof == TYPE_ENTITY) {
1170 /* still get vars first since there could be a fieldpointer */
1171 var = parser_find_var(parser, parser_tokval(parser));
1173 var = parser_find_field(parser, parser_tokval(parser));
1175 else if (parser->memberof == TYPE_VECTOR)
1177 parseerror(parser, "TODO: implement effective vector member access");
1180 else if (parser->memberof) {
1181 parseerror(parser, "namespace for member not found");
1185 var = parser_find_var(parser, parser_tokval(parser));
1187 var = parser_find_var(parser, parser_tokval(parser));
1189 var = parser_find_field(parser, parser_tokval(parser));
1192 parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
1195 if (ast_istype(var, ast_value))
1196 ((ast_value*)var)->uses++;
1197 vec_push(sy.out, syexp(parser_ctx(parser), var));
1198 DEBUGSHUNTDO(con_out("push %s\n", parser_tokval(parser)));
1200 else if (parser->tok == TOKEN_FLOATCONST) {
1203 parseerror(parser, "expected operator or end of statement, got constant");
1207 val = parser_const_float(parser, (parser_token(parser)->constval.f));
1210 vec_push(sy.out, syexp(parser_ctx(parser), (ast_expression*)val));
1211 DEBUGSHUNTDO(con_out("push %g\n", parser_token(parser)->constval.f));
1213 else if (parser->tok == TOKEN_INTCONST) {
1216 parseerror(parser, "expected operator or end of statement, got constant");
1220 val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
1223 vec_push(sy.out, syexp(parser_ctx(parser), (ast_expression*)val));
1224 DEBUGSHUNTDO(con_out("push %i\n", parser_token(parser)->constval.i));
1226 else if (parser->tok == TOKEN_STRINGCONST) {
1229 parseerror(parser, "expected operator or end of statement, got constant");
1233 val = parser_const_string(parser, parser_tokval(parser));
1236 vec_push(sy.out, syexp(parser_ctx(parser), (ast_expression*)val));
1237 DEBUGSHUNTDO(con_out("push string\n"));
1239 else if (parser->tok == TOKEN_VECTORCONST) {
1242 parseerror(parser, "expected operator or end of statement, got constant");
1246 val = parser_const_vector(parser, parser_token(parser)->constval.v);
1249 vec_push(sy.out, syexp(parser_ctx(parser), (ast_expression*)val));
1250 DEBUGSHUNTDO(con_out("push '%g %g %g'\n",
1251 parser_token(parser)->constval.v.x,
1252 parser_token(parser)->constval.v.y,
1253 parser_token(parser)->constval.v.z));
1255 else if (parser->tok == '(') {
1256 parseerror(parser, "internal error: '(' should be classified as operator");
1259 else if (parser->tok == '[') {
1260 parseerror(parser, "internal error: '[' should be classified as operator");
1263 else if (parser->tok == ')') {
1265 DEBUGSHUNTDO(con_out("do[op] )\n"));
1269 /* we do expect an operator next */
1270 /* closing an opening paren */
1271 if (!parser_close_paren(parser, &sy, false))
1274 DEBUGSHUNTDO(con_out("do[nop] )\n"));
1278 /* allowed for function calls */
1279 if (!parser_close_paren(parser, &sy, true))
1284 else if (parser->tok == ']') {
1286 parseerror(parser, "operand expected");
1290 if (!parser_close_paren(parser, &sy, false))
1294 else if (parser->tok != TOKEN_OPERATOR) {
1296 parseerror(parser, "expected operator or end of statement");
1303 /* classify the operator */
1304 const oper_info *op;
1305 const oper_info *olast = NULL;
1307 for (o = 0; o < operator_count; ++o) {
1308 if ((!(operators[o].flags & OP_PREFIX) == wantop) &&
1309 /* !(operators[o].flags & OP_SUFFIX) && / * remove this */
1310 !strcmp(parser_tokval(parser), operators[o].op))
1315 if (o == operator_count) {
1316 /* no operator found... must be the end of the statement */
1319 /* found an operator */
1322 /* when declaring variables, a comma starts a new variable */
1323 if (op->id == opid1(',') && !parens && stopatcomma) {
1324 /* fixup the token */
1329 /* a colon without a pervious question mark cannot be a ternary */
1330 if (!ternaries && op->id == opid2(':','?')) {
1335 if (vec_size(sy.ops) && !vec_last(sy.ops).paren)
1336 olast = &operators[vec_last(sy.ops).etype-1];
1339 (op->prec < olast->prec) ||
1340 (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
1342 if (!parser_sy_pop(parser, &sy))
1344 if (vec_size(sy.ops) && !vec_last(sy.ops).paren)
1345 olast = &operators[vec_last(sy.ops).etype-1];
1350 if (op->id == opid1('.') && opts_standard == COMPILER_GMQCC) {
1351 /* for gmqcc standard: open up the namespace of the previous type */
1352 ast_expression *prevex = vec_last(sy.out).out;
1354 parseerror(parser, "unexpected member operator");
1357 if (prevex->expression.vtype == TYPE_ENTITY)
1358 parser->memberof = TYPE_ENTITY;
1359 else if (prevex->expression.vtype == TYPE_VECTOR)
1360 parser->memberof = TYPE_VECTOR;
1362 parseerror(parser, "type error: type has no members");
1368 if (op->id == opid1('(')) {
1370 size_t sycount = vec_size(sy.out);
1371 DEBUGSHUNTDO(con_out("push [op] (\n"));
1373 /* we expected an operator, this is the function-call operator */
1374 vec_push(sy.ops, syparen(parser_ctx(parser), SY_PAREN_FUNC, sycount-1));
1377 vec_push(sy.ops, syparen(parser_ctx(parser), SY_PAREN_EXPR, 0));
1378 DEBUGSHUNTDO(con_out("push [nop] (\n"));
1381 } else if (op->id == opid1('[')) {
1383 parseerror(parser, "unexpected array subscript");
1387 /* push both the operator and the paren, this makes life easier */
1388 vec_push(sy.ops, syop(parser_ctx(parser), op));
1389 vec_push(sy.ops, syparen(parser_ctx(parser), SY_PAREN_INDEX, 0));
1391 } else if (op->id == opid2('?',':')) {
1393 vec_push(sy.ops, syop(parser_ctx(parser), op));
1396 } else if (op->id == opid2(':','?')) {
1397 /* we don't push this operator */
1401 DEBUGSHUNTDO(con_out("push operator %s\n", op->op));
1402 vec_push(sy.ops, syop(parser_ctx(parser), op));
1403 wantop = !!(op->flags & OP_SUFFIX);
1406 if (!parser_next(parser)) {
1409 if (parser->tok == ';' ||
1410 (!parens && parser->tok == ']'))
1416 while (vec_size(sy.ops)) {
1417 if (!parser_sy_pop(parser, &sy))
1421 parser->lex->flags.noops = true;
1422 if (!vec_size(sy.out)) {
1423 parseerror(parser, "empty expression");
1426 expr = sy.out[0].out;
1429 DEBUGSHUNTDO(con_out("shunt done\n"));
1433 parser->lex->flags.noops = true;
1439 static ast_expression* parse_expression(parser_t *parser, bool stopatcomma)
1441 ast_expression *e = parse_expression_leave(parser, stopatcomma);
1444 if (!parser_next(parser)) {
1451 static bool parse_if(parser_t *parser, ast_block *block, ast_expression **out)
1454 ast_expression *cond, *ontrue, *onfalse = NULL;
1457 lex_ctx ctx = parser_ctx(parser);
1459 (void)block; /* not touching */
1461 /* skip the 'if', parse an optional 'not' and check for an opening paren */
1462 if (!parser_next(parser)) {
1463 parseerror(parser, "expected condition or 'not'");
1466 if (parser->tok == TOKEN_KEYWORD && !strcmp(parser_tokval(parser), "not")) {
1468 if (!parser_next(parser)) {
1469 parseerror(parser, "expected condition in parenthesis");
1473 if (parser->tok != '(') {
1474 parseerror(parser, "expected 'if' condition in parenthesis");
1477 /* parse into the expression */
1478 if (!parser_next(parser)) {
1479 parseerror(parser, "expected 'if' condition after opening paren");
1482 /* parse the condition */
1483 cond = parse_expression_leave(parser, false);
1487 if (parser->tok != ')') {
1488 parseerror(parser, "expected closing paren after 'if' condition");
1492 /* parse into the 'then' branch */
1493 if (!parser_next(parser)) {
1494 parseerror(parser, "expected statement for on-true branch of 'if'");
1498 ontrue = parse_statement_or_block(parser);
1503 /* check for an else */
1504 if (!strcmp(parser_tokval(parser), "else")) {
1505 /* parse into the 'else' branch */
1506 if (!parser_next(parser)) {
1507 parseerror(parser, "expected on-false branch after 'else'");
1512 onfalse = parse_statement_or_block(parser);
1521 ifthen = ast_ifthen_new(ctx, cond, onfalse, ontrue);
1523 ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
1524 *out = (ast_expression*)ifthen;
1528 static bool parse_while(parser_t *parser, ast_block *block, ast_expression **out)
1531 ast_expression *cond, *ontrue;
1533 lex_ctx ctx = parser_ctx(parser);
1535 (void)block; /* not touching */
1537 /* skip the 'while' and check for opening paren */
1538 if (!parser_next(parser) || parser->tok != '(') {
1539 parseerror(parser, "expected 'while' condition in parenthesis");
1542 /* parse into the expression */
1543 if (!parser_next(parser)) {
1544 parseerror(parser, "expected 'while' condition after opening paren");
1547 /* parse the condition */
1548 cond = parse_expression_leave(parser, false);
1552 if (parser->tok != ')') {
1553 parseerror(parser, "expected closing paren after 'while' condition");
1557 /* parse into the 'then' branch */
1558 if (!parser_next(parser)) {
1559 parseerror(parser, "expected while-loop body");
1563 ontrue = parse_statement_or_block(parser);
1569 aloop = ast_loop_new(ctx, NULL, cond, NULL, NULL, ontrue);
1570 *out = (ast_expression*)aloop;
1574 static bool parse_dowhile(parser_t *parser, ast_block *block, ast_expression **out)
1577 ast_expression *cond, *ontrue;
1579 lex_ctx ctx = parser_ctx(parser);
1581 (void)block; /* not touching */
1583 /* skip the 'do' and get the body */
1584 if (!parser_next(parser)) {
1585 parseerror(parser, "expected loop body");
1588 ontrue = parse_statement_or_block(parser);
1592 /* expect the "while" */
1593 if (parser->tok != TOKEN_KEYWORD ||
1594 strcmp(parser_tokval(parser), "while"))
1596 parseerror(parser, "expected 'while' and condition");
1601 /* skip the 'while' and check for opening paren */
1602 if (!parser_next(parser) || parser->tok != '(') {
1603 parseerror(parser, "expected 'while' condition in parenthesis");
1607 /* parse into the expression */
1608 if (!parser_next(parser)) {
1609 parseerror(parser, "expected 'while' condition after opening paren");
1613 /* parse the condition */
1614 cond = parse_expression_leave(parser, false);
1618 if (parser->tok != ')') {
1619 parseerror(parser, "expected closing paren after 'while' condition");
1625 if (!parser_next(parser) || parser->tok != ';') {
1626 parseerror(parser, "expected semicolon after condition");
1632 if (!parser_next(parser)) {
1633 parseerror(parser, "parse error");
1639 aloop = ast_loop_new(ctx, NULL, NULL, cond, NULL, ontrue);
1640 *out = (ast_expression*)aloop;
1644 static bool parse_for(parser_t *parser, ast_block *block, ast_expression **out)
1647 ast_expression *initexpr, *cond, *increment, *ontrue;
1648 size_t oldblocklocal;
1651 lex_ctx ctx = parser_ctx(parser);
1653 oldblocklocal = parser->blocklocal;
1654 parser->blocklocal = vec_size(parser->locals);
1661 /* skip the 'while' and check for opening paren */
1662 if (!parser_next(parser) || parser->tok != '(') {
1663 parseerror(parser, "expected 'for' expressions in parenthesis");
1666 /* parse into the expression */
1667 if (!parser_next(parser)) {
1668 parseerror(parser, "expected 'for' initializer after opening paren");
1672 if (parser->tok == TOKEN_TYPENAME) {
1673 if (opts_standard != COMPILER_GMQCC) {
1674 if (parsewarning(parser, WARN_EXTENSIONS,
1675 "current standard does not allow variable declarations in for-loop initializers"))
1679 parseerror(parser, "TODO: assignment of new variables to be non-const");
1681 if (!parse_variable(parser, block, true))
1684 else if (parser->tok != ';')
1686 initexpr = parse_expression_leave(parser, false);
1691 /* move on to condition */
1692 if (parser->tok != ';') {
1693 parseerror(parser, "expected semicolon after for-loop initializer");
1696 if (!parser_next(parser)) {
1697 parseerror(parser, "expected for-loop condition");
1701 /* parse the condition */
1702 if (parser->tok != ';') {
1703 cond = parse_expression_leave(parser, false);
1708 /* move on to incrementor */
1709 if (parser->tok != ';') {
1710 parseerror(parser, "expected semicolon after for-loop initializer");
1713 if (!parser_next(parser)) {
1714 parseerror(parser, "expected for-loop condition");
1718 /* parse the incrementor */
1719 if (parser->tok != ')') {
1720 increment = parse_expression_leave(parser, false);
1723 if (!ast_istype(increment, ast_store) &&
1724 !ast_istype(increment, ast_call) &&
1725 !ast_istype(increment, ast_binstore))
1727 if (genwarning(ast_ctx(increment), WARN_EFFECTLESS_STATEMENT, "statement has no effect"))
1733 if (parser->tok != ')') {
1734 parseerror(parser, "expected closing paren after 'for-loop' incrementor");
1737 /* parse into the 'then' branch */
1738 if (!parser_next(parser)) {
1739 parseerror(parser, "expected for-loop body");
1742 ontrue = parse_statement_or_block(parser);
1747 aloop = ast_loop_new(ctx, initexpr, cond, NULL, increment, ontrue);
1748 *out = (ast_expression*)aloop;
1750 while (vec_size(parser->locals) > parser->blocklocal)
1751 retval = retval && parser_pop_local(parser);
1752 parser->blocklocal = oldblocklocal;
1755 if (initexpr) ast_delete(initexpr);
1756 if (cond) ast_delete(cond);
1757 if (increment) ast_delete(increment);
1758 while (vec_size(parser->locals) > parser->blocklocal)
1759 (void)!parser_pop_local(parser);
1760 parser->blocklocal = oldblocklocal;
1764 static bool parse_return(parser_t *parser, ast_block *block, ast_expression **out)
1766 ast_expression *exp = NULL;
1767 ast_return *ret = NULL;
1768 ast_value *expected = parser->function->vtype;
1770 (void)block; /* not touching */
1772 if (!parser_next(parser)) {
1773 parseerror(parser, "expected return expression");
1777 if (parser->tok != ';') {
1778 exp = parse_expression(parser, false);
1782 if (exp->expression.vtype != expected->expression.next->expression.vtype) {
1783 parseerror(parser, "return with invalid expression");
1786 ret = ast_return_new(exp->expression.node.context, exp);
1792 if (!parser_next(parser))
1793 parseerror(parser, "parse error");
1794 if (expected->expression.next->expression.vtype != TYPE_VOID) {
1795 if (opts_standard != COMPILER_GMQCC)
1796 (void)!parsewarning(parser, WARN_MISSING_RETURN_VALUES, "return without value");
1798 parseerror(parser, "return without value");
1800 ret = ast_return_new(parser_ctx(parser), NULL);
1802 *out = (ast_expression*)ret;
1806 static bool parse_break_continue(parser_t *parser, ast_block *block, ast_expression **out, bool is_continue)
1808 lex_ctx ctx = parser_ctx(parser);
1810 (void)block; /* not touching */
1812 if (!parser_next(parser) || parser->tok != ';') {
1813 parseerror(parser, "expected semicolon");
1817 if (!parser_next(parser))
1818 parseerror(parser, "parse error");
1820 *out = (ast_expression*)ast_breakcont_new(ctx, is_continue);
1824 static bool parse_switch(parser_t *parser, ast_block *block, ast_expression **out)
1826 ast_expression *operand;
1828 ast_switch *switchnode;
1829 ast_switch_case swcase;
1831 lex_ctx ctx = parser_ctx(parser);
1833 (void)block; /* not touching */
1835 /* parse over the opening paren */
1836 if (!parser_next(parser) || parser->tok != '(') {
1837 parseerror(parser, "expected switch operand in parenthesis");
1841 /* parse into the expression */
1842 if (!parser_next(parser)) {
1843 parseerror(parser, "expected switch operand");
1846 /* parse the operand */
1847 operand = parse_expression_leave(parser, false);
1851 if (!OPTS_FLAG(RELAXED_SWITCH)) {
1852 opval = (ast_value*)operand;
1853 if (!ast_istype(operand, ast_value) || !opval->isconst) {
1854 parseerror(parser, "case on non-constant values need to be explicitly enabled via -frelaxed-switch");
1860 switchnode = ast_switch_new(ctx, operand);
1863 if (parser->tok != ')') {
1864 ast_delete(switchnode);
1865 parseerror(parser, "expected closing paren after 'switch' operand");
1869 /* parse over the opening paren */
1870 if (!parser_next(parser) || parser->tok != '{') {
1871 ast_delete(switchnode);
1872 parseerror(parser, "expected list of cases");
1876 if (!parser_next(parser)) {
1877 ast_delete(switchnode);
1878 parseerror(parser, "expected 'case' or 'default'");
1883 while (parser->tok != '}') {
1884 ast_block *caseblock;
1886 if (parser->tok != TOKEN_KEYWORD) {
1887 ast_delete(switchnode);
1888 parseerror(parser, "expected 'case' or 'default'");
1891 if (!strcmp(parser_tokval(parser), "case")) {
1892 if (!parser_next(parser)) {
1893 ast_delete(switchnode);
1894 parseerror(parser, "expected expression for case");
1897 swcase.value = parse_expression_leave(parser, false);
1898 if (!swcase.value) {
1899 ast_delete(switchnode);
1900 parseerror(parser, "expected expression for case");
1904 else if (!strcmp(parser_tokval(parser), "default")) {
1905 swcase.value = NULL;
1906 if (!parser_next(parser)) {
1907 ast_delete(switchnode);
1908 parseerror(parser, "expected colon");
1913 /* Now the colon and body */
1914 if (parser->tok != ':') {
1915 if (swcase.value) ast_unref(swcase.value);
1916 ast_delete(switchnode);
1917 parseerror(parser, "expected colon");
1921 if (!parser_next(parser)) {
1922 if (swcase.value) ast_unref(swcase.value);
1923 ast_delete(switchnode);
1924 parseerror(parser, "expected statements or case");
1927 caseblock = ast_block_new(parser_ctx(parser));
1929 if (swcase.value) ast_unref(swcase.value);
1930 ast_delete(switchnode);
1933 swcase.code = (ast_expression*)caseblock;
1934 vec_push(switchnode->cases, swcase);
1936 ast_expression *expr;
1937 if (parser->tok == '}')
1939 if (parser->tok == TOKEN_KEYWORD) {
1940 if (!strcmp(parser_tokval(parser), "case") ||
1941 !strcmp(parser_tokval(parser), "default"))
1946 if (!parse_statement(parser, caseblock, &expr, true)) {
1947 ast_delete(switchnode);
1952 vec_push(caseblock->exprs, expr);
1957 if (parser->tok != '}') {
1958 ast_delete(switchnode);
1959 parseerror(parser, "expected closing paren of case list");
1962 if (!parser_next(parser)) {
1963 ast_delete(switchnode);
1964 parseerror(parser, "parse error after switch");
1967 *out = (ast_expression*)switchnode;
1971 static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases)
1973 if (parser->tok == TOKEN_TYPENAME || parser->tok == '.')
1975 /* local variable */
1977 parseerror(parser, "cannot declare a variable from here");
1980 if (opts_standard == COMPILER_QCC) {
1981 if (parsewarning(parser, WARN_EXTENSIONS, "missing 'local' keyword when declaring a local variable"))
1984 if (!parse_variable(parser, block, false))
1989 else if (parser->tok == TOKEN_KEYWORD)
1991 if (!strcmp(parser_tokval(parser), "local"))
1994 parseerror(parser, "cannot declare a local variable here");
1997 if (!parser_next(parser)) {
1998 parseerror(parser, "expected variable declaration");
2001 if (!parse_variable(parser, block, true))
2006 else if (!strcmp(parser_tokval(parser), "return"))
2008 return parse_return(parser, block, out);
2010 else if (!strcmp(parser_tokval(parser), "if"))
2012 return parse_if(parser, block, out);
2014 else if (!strcmp(parser_tokval(parser), "while"))
2016 return parse_while(parser, block, out);
2018 else if (!strcmp(parser_tokval(parser), "do"))
2020 return parse_dowhile(parser, block, out);
2022 else if (!strcmp(parser_tokval(parser), "for"))
2024 if (opts_standard == COMPILER_QCC) {
2025 if (parsewarning(parser, WARN_EXTENSIONS, "for loops are not recognized in the original Quake C standard, to enable try an alternate standard --std=?"))
2028 return parse_for(parser, block, out);
2030 else if (!strcmp(parser_tokval(parser), "break"))
2032 return parse_break_continue(parser, block, out, false);
2034 else if (!strcmp(parser_tokval(parser), "continue"))
2036 return parse_break_continue(parser, block, out, true);
2038 else if (!strcmp(parser_tokval(parser), "switch"))
2040 return parse_switch(parser, block, out);
2042 else if (!strcmp(parser_tokval(parser), "case") ||
2043 !strcmp(parser_tokval(parser), "default"))
2046 parseerror(parser, "unexpected 'case' label");
2051 parseerror(parser, "Unexpected keyword");
2054 else if (parser->tok == '{')
2057 inner = parse_block(parser, false);
2060 *out = (ast_expression*)inner;
2065 ast_expression *exp = parse_expression(parser, false);
2069 if (!ast_istype(exp, ast_store) &&
2070 !ast_istype(exp, ast_call) &&
2071 !ast_istype(exp, ast_binstore))
2073 if (genwarning(ast_ctx(exp), WARN_EFFECTLESS_STATEMENT, "statement has no effect"))
2080 static bool GMQCC_WARN parser_pop_local(parser_t *parser)
2085 ve = &vec_last(parser->locals);
2086 if (!parser->errors) {
2087 if (ast_istype(ve->var, ast_value) && !(((ast_value*)(ve->var))->uses)) {
2088 if (parsewarning(parser, WARN_UNUSED_VARIABLE, "unused variable: `%s`", ve->name))
2093 vec_pop(parser->locals);
2097 static bool parse_block_into(parser_t *parser, ast_block *block, bool warnreturn)
2099 size_t oldblocklocal;
2102 oldblocklocal = parser->blocklocal;
2103 parser->blocklocal = vec_size(parser->locals);
2105 if (!parser_next(parser)) { /* skip the '{' */
2106 parseerror(parser, "expected function body");
2110 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
2112 ast_expression *expr;
2113 if (parser->tok == '}')
2116 if (!parse_statement(parser, block, &expr, false)) {
2117 /* parseerror(parser, "parse error"); */
2123 vec_push(block->exprs, expr);
2126 if (parser->tok != '}') {
2129 if (warnreturn && parser->function->vtype->expression.next->expression.vtype != TYPE_VOID)
2131 if (!vec_size(block->exprs) ||
2132 !ast_istype(vec_last(block->exprs), ast_return))
2134 if (parsewarning(parser, WARN_MISSING_RETURN_VALUES, "control reaches end of non-void function")) {
2140 (void)parser_next(parser);
2144 while (vec_size(parser->locals) > parser->blocklocal)
2145 retval = retval && parser_pop_local(parser);
2146 parser->blocklocal = oldblocklocal;
2150 static ast_block* parse_block(parser_t *parser, bool warnreturn)
2153 block = ast_block_new(parser_ctx(parser));
2156 if (!parse_block_into(parser, block, warnreturn)) {
2157 ast_block_delete(block);
2163 static ast_expression* parse_statement_or_block(parser_t *parser)
2165 ast_expression *expr = NULL;
2166 if (parser->tok == '{')
2167 return (ast_expression*)parse_block(parser, false);
2168 if (!parse_statement(parser, NULL, &expr, false))
2174 static bool create_vector_members(ast_value *var, varentry_t *ve)
2177 size_t len = strlen(var->name);
2179 for (i = 0; i < 3; ++i) {
2180 ve[i].var = (ast_expression*)ast_member_new(ast_ctx(var), (ast_expression*)var, i);
2184 ve[i].name = (char*)mem_a(len+3);
2186 ast_delete(ve[i].var);
2190 memcpy(ve[i].name, var->name, len);
2191 ve[i].name[len] = '_';
2192 ve[i].name[len+1] = 'x'+i;
2193 ve[i].name[len+2] = 0;
2202 ast_delete(ve[i].var);
2209 static bool parse_function_body(parser_t *parser, ast_value *var)
2211 ast_block *block = NULL;
2216 ast_expression *framenum = NULL;
2217 ast_expression *nextthink = NULL;
2218 /* None of the following have to be deleted */
2219 ast_expression *fld_think = NULL, *fld_nextthink = NULL, *fld_frame = NULL;
2220 ast_expression *gbl_time = NULL, *gbl_self = NULL;
2221 bool has_frame_think;
2225 has_frame_think = false;
2226 old = parser->function;
2228 if (var->expression.variadic) {
2229 if (parsewarning(parser, WARN_VARIADIC_FUNCTION,
2230 "variadic function with implementation will not be able to access additional parameters"))
2236 if (parser->tok == '[') {
2237 /* got a frame definition: [ framenum, nextthink ]
2238 * this translates to:
2239 * self.frame = framenum;
2240 * self.nextthink = time + 0.1;
2241 * self.think = nextthink;
2245 fld_think = parser_find_field(parser, "think");
2246 fld_nextthink = parser_find_field(parser, "nextthink");
2247 fld_frame = parser_find_field(parser, "frame");
2248 if (!fld_think || !fld_nextthink || !fld_frame) {
2249 parseerror(parser, "cannot use [frame,think] notation without the required fields");
2250 parseerror(parser, "please declare the following entityfields: `frame`, `think`, `nextthink`");
2253 gbl_time = parser_find_global(parser, "time");
2254 gbl_self = parser_find_global(parser, "self");
2255 if (!gbl_time || !gbl_self) {
2256 parseerror(parser, "cannot use [frame,think] notation without the required globals");
2257 parseerror(parser, "please declare the following globals: `time`, `self`");
2261 if (!parser_next(parser))
2264 framenum = parse_expression_leave(parser, true);
2266 parseerror(parser, "expected a framenumber constant in[frame,think] notation");
2269 if (!ast_istype(framenum, ast_value) || !( (ast_value*)framenum )->isconst) {
2270 ast_unref(framenum);
2271 parseerror(parser, "framenumber in [frame,think] notation must be a constant");
2275 if (parser->tok != ',') {
2276 ast_unref(framenum);
2277 parseerror(parser, "expected comma after frame number in [frame,think] notation");
2278 parseerror(parser, "Got a %i\n", parser->tok);
2282 if (!parser_next(parser)) {
2283 ast_unref(framenum);
2287 if (parser->tok == TOKEN_IDENT && !parser_find_var(parser, parser_tokval(parser)))
2289 /* qc allows the use of not-yet-declared functions here
2290 * - this automatically creates a prototype */
2292 ast_value *thinkfunc;
2293 ast_expression *functype = fld_think->expression.next;
2295 thinkfunc = ast_value_new(parser_ctx(parser), parser_tokval(parser), functype->expression.vtype);
2296 if (!thinkfunc || !ast_type_adopt(thinkfunc, functype)) {
2297 ast_unref(framenum);
2298 parseerror(parser, "failed to create implicit prototype for `%s`", parser_tokval(parser));
2302 if (!parser_next(parser)) {
2303 ast_unref(framenum);
2304 ast_delete(thinkfunc);
2308 varent.var = (ast_expression*)thinkfunc;
2309 varent.name = util_strdup(thinkfunc->name);
2310 vec_push(parser->globals, varent);
2311 nextthink = (ast_expression*)thinkfunc;
2314 nextthink = parse_expression_leave(parser, true);
2316 ast_unref(framenum);
2317 parseerror(parser, "expected a think-function in [frame,think] notation");
2322 if (!ast_istype(nextthink, ast_value)) {
2323 parseerror(parser, "think-function in [frame,think] notation must be a constant");
2327 if (retval && parser->tok != ']') {
2328 parseerror(parser, "expected closing `]` for [frame,think] notation");
2332 if (retval && !parser_next(parser)) {
2336 if (retval && parser->tok != '{') {
2337 parseerror(parser, "a function body has to be declared after a [frame,think] declaration");
2342 ast_unref(nextthink);
2343 ast_unref(framenum);
2347 has_frame_think = true;
2350 block = ast_block_new(parser_ctx(parser));
2352 parseerror(parser, "failed to allocate block");
2353 if (has_frame_think) {
2354 ast_unref(nextthink);
2355 ast_unref(framenum);
2360 if (has_frame_think) {
2362 ast_expression *self_frame;
2363 ast_expression *self_nextthink;
2364 ast_expression *self_think;
2365 ast_expression *time_plus_1;
2366 ast_store *store_frame;
2367 ast_store *store_nextthink;
2368 ast_store *store_think;
2370 ctx = parser_ctx(parser);
2371 self_frame = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_frame);
2372 self_nextthink = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_nextthink);
2373 self_think = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_think);
2375 time_plus_1 = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F,
2376 gbl_time, (ast_expression*)parser_const_float(parser, 0.1));
2378 if (!self_frame || !self_nextthink || !self_think || !time_plus_1) {
2379 if (self_frame) ast_delete(self_frame);
2380 if (self_nextthink) ast_delete(self_nextthink);
2381 if (self_think) ast_delete(self_think);
2382 if (time_plus_1) ast_delete(time_plus_1);
2388 store_frame = ast_store_new(ctx, INSTR_STOREP_F, self_frame, framenum);
2389 store_nextthink = ast_store_new(ctx, INSTR_STOREP_F, self_nextthink, time_plus_1);
2390 store_think = ast_store_new(ctx, INSTR_STOREP_FNC, self_think, nextthink);
2393 ast_delete(self_frame);
2396 if (!store_nextthink) {
2397 ast_delete(self_nextthink);
2401 ast_delete(self_think);
2405 if (store_frame) ast_delete(store_frame);
2406 if (store_nextthink) ast_delete(store_nextthink);
2407 if (store_think) ast_delete(store_think);
2410 vec_push(block->exprs, (ast_expression*)store_frame);
2411 vec_push(block->exprs, (ast_expression*)store_nextthink);
2412 vec_push(block->exprs, (ast_expression*)store_think);
2416 parseerror(parser, "failed to generate code for [frame,think]");
2417 ast_unref(nextthink);
2418 ast_unref(framenum);
2424 for (parami = 0; parami < vec_size(var->expression.params); ++parami) {
2427 ast_value *param = var->expression.params[parami];
2429 if (param->expression.vtype != TYPE_VECTOR &&
2430 (param->expression.vtype != TYPE_FIELD ||
2431 param->expression.next->expression.vtype != TYPE_VECTOR))
2436 if (!create_vector_members(param, ve)) {
2437 ast_block_delete(block);
2441 for (e = 0; e < 3; ++e) {
2442 vec_push(parser->locals, ve[e]);
2443 ast_block_collect(block, ve[e].var);
2444 ve[e].var = NULL; /* collected */
2448 func = ast_function_new(ast_ctx(var), var->name, var);
2450 parseerror(parser, "failed to allocate function for `%s`", var->name);
2451 ast_block_delete(block);
2454 vec_push(parser->functions, func);
2456 parser->function = func;
2457 if (!parse_block_into(parser, block, true)) {
2458 ast_block_delete(block);
2462 vec_push(func->blocks, block);
2464 parser->function = old;
2465 while (vec_size(parser->locals))
2466 retval = retval && parser_pop_local(parser);
2468 if (parser->tok == ';')
2469 return parser_next(parser);
2470 else if (opts_standard == COMPILER_QCC)
2471 parseerror(parser, "missing semicolon after function body (mandatory with -std=qcc)");
2475 vec_pop(parser->functions);
2476 ast_function_delete(func);
2477 var->constval.vfunc = NULL;
2480 while (vec_size(parser->locals)) {
2481 mem_d(vec_last(parser->locals).name);
2482 vec_pop(parser->locals);
2484 parser->function = old;
2488 static ast_expression *array_accessor_split(
2493 ast_expression *left,
2494 ast_expression *right
2500 lex_ctx ctx = ast_ctx(array);
2502 if (!left || !right) {
2503 if (left) ast_delete(left);
2504 if (right) ast_delete(right);
2508 cmp = ast_binary_new(ctx, INSTR_LT,
2509 (ast_expression*)index,
2510 (ast_expression*)parser_const_float(parser, middle));
2514 parseerror(parser, "internal error: failed to create comparison for array setter");
2518 ifthen = ast_ifthen_new(ctx, (ast_expression*)cmp, left, right);
2520 ast_delete(cmp); /* will delete left and right */
2521 parseerror(parser, "internal error: failed to create conditional jump for array setter");
2525 return (ast_expression*)ifthen;
2528 static ast_expression *array_setter_node(parser_t *parser, ast_value *array, ast_value *index, ast_value *value, size_t from, size_t afterend)
2530 lex_ctx ctx = ast_ctx(array);
2532 if (from+1 == afterend) {
2533 /* set this value */
2536 ast_array_index *subscript;
2538 int assignop = type_store_instr[value->expression.vtype];
2540 if (value->expression.vtype == TYPE_FIELD && value->expression.next->expression.vtype == TYPE_VECTOR)
2541 assignop = INSTR_STORE_V;
2543 subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)parser_const_float(parser, from));
2547 st = ast_store_new(ctx, assignop, (ast_expression*)subscript, (ast_expression*)value);
2549 ast_delete(subscript);
2553 block = ast_block_new(ctx);
2559 vec_push(block->exprs, (ast_expression*)st);
2561 ret = ast_return_new(ctx, NULL);
2567 vec_push(block->exprs, (ast_expression*)ret);
2569 return (ast_expression*)block;
2571 ast_expression *left, *right;
2572 size_t diff = afterend - from;
2573 size_t middle = from + diff/2;
2574 left = array_setter_node(parser, array, index, value, from, middle);
2575 right = array_setter_node(parser, array, index, value, middle, afterend);
2576 return array_accessor_split(parser, array, index, middle, left, right);
2580 static ast_expression *array_field_setter_node(
2589 lex_ctx ctx = ast_ctx(array);
2591 if (from+1 == afterend) {
2592 /* set this value */
2595 ast_entfield *entfield;
2596 ast_array_index *subscript;
2598 int assignop = type_storep_instr[value->expression.vtype];
2600 if (value->expression.vtype == TYPE_FIELD && value->expression.next->expression.vtype == TYPE_VECTOR)
2601 assignop = INSTR_STOREP_V;
2603 subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)parser_const_float(parser, from));
2607 entfield = ast_entfield_new_force(ctx,
2608 (ast_expression*)entity,
2609 (ast_expression*)subscript,
2610 (ast_expression*)subscript);
2612 ast_delete(subscript);
2616 st = ast_store_new(ctx, assignop, (ast_expression*)entfield, (ast_expression*)value);
2618 ast_delete(entfield);
2622 block = ast_block_new(ctx);
2628 vec_push(block->exprs, (ast_expression*)st);
2630 ret = ast_return_new(ctx, NULL);
2636 vec_push(block->exprs, (ast_expression*)ret);
2638 return (ast_expression*)block;
2640 ast_expression *left, *right;
2641 size_t diff = afterend - from;
2642 size_t middle = from + diff/2;
2643 left = array_field_setter_node(parser, array, entity, index, value, from, middle);
2644 right = array_field_setter_node(parser, array, entity, index, value, middle, afterend);
2645 return array_accessor_split(parser, array, index, middle, left, right);
2649 static ast_expression *array_getter_node(parser_t *parser, ast_value *array, ast_value *index, size_t from, size_t afterend)
2651 lex_ctx ctx = ast_ctx(array);
2653 if (from+1 == afterend) {
2655 ast_array_index *subscript;
2657 subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)parser_const_float(parser, from));
2661 ret = ast_return_new(ctx, (ast_expression*)subscript);
2663 ast_delete(subscript);
2667 return (ast_expression*)ret;
2669 ast_expression *left, *right;
2670 size_t diff = afterend - from;
2671 size_t middle = from + diff/2;
2672 left = array_getter_node(parser, array, index, from, middle);
2673 right = array_getter_node(parser, array, index, middle, afterend);
2674 return array_accessor_split(parser, array, index, middle, left, right);
2678 static bool parser_create_array_accessor(parser_t *parser, ast_value *array, const char *funcname, ast_value **out)
2680 ast_function *func = NULL;
2681 ast_value *fval = NULL;
2682 ast_block *body = NULL;
2684 fval = ast_value_new(ast_ctx(array), funcname, TYPE_FUNCTION);
2686 parseerror(parser, "failed to create accessor function value");
2690 func = ast_function_new(ast_ctx(array), funcname, fval);
2693 parseerror(parser, "failed to create accessor function node");
2697 body = ast_block_new(ast_ctx(array));
2699 parseerror(parser, "failed to create block for array accessor");
2705 vec_push(func->blocks, body);
2708 vec_push(parser->accessors, fval);
2713 static bool parser_create_array_setter(parser_t *parser, ast_value *array, const char *funcname)
2715 ast_expression *root = NULL;
2716 ast_value *index = NULL;
2717 ast_value *value = NULL;
2721 if (!ast_istype(array->expression.next, ast_value)) {
2722 parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type");
2726 if (!parser_create_array_accessor(parser, array, funcname, &fval))
2728 func = fval->constval.vfunc;
2729 fval->expression.next = (ast_expression*)ast_value_new(ast_ctx(array), "<void>", TYPE_VOID);
2731 index = ast_value_new(ast_ctx(array), "index", TYPE_FLOAT);
2732 value = ast_value_copy((ast_value*)array->expression.next);
2734 if (!index || !value) {
2735 parseerror(parser, "failed to create locals for array accessor");
2738 (void)!ast_value_set_name(value, "value"); /* not important */
2739 vec_push(fval->expression.params, index);
2740 vec_push(fval->expression.params, value);
2742 root = array_setter_node(parser, array, index, value, 0, array->expression.count);
2744 parseerror(parser, "failed to build accessor search tree");
2748 vec_push(func->blocks[0]->exprs, root);
2749 array->setter = fval;
2752 if (index) ast_delete(index);
2753 if (value) ast_delete(value);
2754 if (root) ast_delete(root);
2760 static bool parser_create_array_field_setter(parser_t *parser, ast_value *array, const char *funcname)
2762 ast_expression *root = NULL;
2763 ast_value *entity = NULL;
2764 ast_value *index = NULL;
2765 ast_value *value = NULL;
2769 if (!ast_istype(array->expression.next, ast_value)) {
2770 parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type");
2774 if (!parser_create_array_accessor(parser, array, funcname, &fval))
2776 func = fval->constval.vfunc;
2777 fval->expression.next = (ast_expression*)ast_value_new(ast_ctx(array), "<void>", TYPE_VOID);
2779 entity = ast_value_new(ast_ctx(array), "entity", TYPE_ENTITY);
2780 index = ast_value_new(ast_ctx(array), "index", TYPE_FLOAT);
2781 value = ast_value_copy((ast_value*)array->expression.next);
2782 if (!entity || !index || !value) {
2783 parseerror(parser, "failed to create locals for array accessor");
2786 (void)!ast_value_set_name(value, "value"); /* not important */
2787 vec_push(fval->expression.params, entity);
2788 vec_push(fval->expression.params, index);
2789 vec_push(fval->expression.params, value);
2791 root = array_field_setter_node(parser, array, entity, index, value, 0, array->expression.count);
2793 parseerror(parser, "failed to build accessor search tree");
2797 vec_push(func->blocks[0]->exprs, root);
2798 array->setter = fval;
2801 if (entity) ast_delete(entity);
2802 if (index) ast_delete(index);
2803 if (value) ast_delete(value);
2804 if (root) ast_delete(root);
2810 static bool parser_create_array_getter(parser_t *parser, ast_value *array, const ast_expression *elemtype, const char *funcname)
2812 ast_expression *root = NULL;
2813 ast_value *index = NULL;
2817 /* NOTE: checking array->expression.next rather than elemtype since
2818 * for fields elemtype is a temporary fieldtype.
2820 if (!ast_istype(array->expression.next, ast_value)) {
2821 parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type");
2825 if (!parser_create_array_accessor(parser, array, funcname, &fval))
2827 func = fval->constval.vfunc;
2828 fval->expression.next = ast_type_copy(ast_ctx(array), elemtype);
2830 index = ast_value_new(ast_ctx(array), "index", TYPE_FLOAT);
2833 parseerror(parser, "failed to create locals for array accessor");
2836 vec_push(fval->expression.params, index);
2838 root = array_getter_node(parser, array, index, 0, array->expression.count);
2840 parseerror(parser, "failed to build accessor search tree");
2844 vec_push(func->blocks[0]->exprs, root);
2845 array->getter = fval;
2848 if (index) ast_delete(index);
2849 if (root) ast_delete(root);
2855 static ast_value *parse_typename(parser_t *parser, ast_value **storebase);
2856 static ast_value *parse_parameter_list(parser_t *parser, ast_value *var)
2864 bool variadic = false;
2866 ctx = parser_ctx(parser);
2868 /* for the sake of less code we parse-in in this function */
2869 if (!parser_next(parser)) {
2870 parseerror(parser, "expected parameter list");
2876 /* parse variables until we hit a closing paren */
2877 while (parser->tok != ')') {
2879 /* there must be commas between them */
2880 if (parser->tok != ',') {
2881 parseerror(parser, "expected comma or end of parameter list");
2884 if (!parser_next(parser)) {
2885 parseerror(parser, "expected parameter");
2891 if (parser->tok == TOKEN_DOTS) {
2892 /* '...' indicates a varargs function */
2894 if (!parser_next(parser)) {
2895 parseerror(parser, "expected parameter");
2898 if (parser->tok != ')') {
2899 parseerror(parser, "`...` must be the last parameter of a variadic function declaration");
2905 /* for anything else just parse a typename */
2906 param = parse_typename(parser, NULL);
2909 vec_push(params, param);
2910 if (param->expression.vtype >= TYPE_VARIANT) {
2911 char typename[1024];
2912 ast_type_to_string((ast_expression*)param, typename, sizeof(typename));
2913 parseerror(parser, "type not supported as part of a parameter list: %s", typename);
2920 if (vec_size(params) > 8 && opts_standard == COMPILER_QCC)
2921 (void)!parsewarning(parser, WARN_EXTENSIONS, "more than 8 parameters are not supported by this standard");
2924 if (!parser_next(parser)) {
2925 parseerror(parser, "parse error after typename");
2929 /* now turn 'var' into a function type */
2930 fval = ast_value_new(ctx, "<type()>", TYPE_FUNCTION);
2931 fval->expression.next = (ast_expression*)var;
2932 fval->expression.variadic = variadic;
2935 var->expression.params = params;
2942 for (i = 0; i < vec_size(params); ++i)
2943 ast_delete(params[i]);
2948 static ast_value *parse_arraysize(parser_t *parser, ast_value *var)
2950 ast_expression *cexp;
2951 ast_value *cval, *tmp;
2954 ctx = parser_ctx(parser);
2956 if (!parser_next(parser)) {
2958 parseerror(parser, "expected array-size");
2962 cexp = parse_expression_leave(parser, true);
2964 if (!cexp || !ast_istype(cexp, ast_value)) {
2968 parseerror(parser, "expected array-size as constant positive integer");
2971 cval = (ast_value*)cexp;
2973 tmp = ast_value_new(ctx, "<type[]>", TYPE_ARRAY);
2974 tmp->expression.next = (ast_expression*)var;
2977 if (cval->expression.vtype == TYPE_INTEGER)
2978 tmp->expression.count = cval->constval.vint;
2979 else if (cval->expression.vtype == TYPE_FLOAT)
2980 tmp->expression.count = cval->constval.vfloat;
2984 parseerror(parser, "array-size must be a positive integer constant");
2989 if (parser->tok != ']') {
2991 parseerror(parser, "expected ']' after array-size");
2994 if (!parser_next(parser)) {
2996 parseerror(parser, "error after parsing array size");
3002 /* Parse a complete typename.
3003 * for single-variables (ie. function parameters or typedefs) storebase should be NULL
3004 * but when parsing variables separated by comma
3005 * 'storebase' should point to where the base-type should be kept.
3006 * The base type makes up every bit of type information which comes *before* the
3009 * The following will be parsed in its entirety:
3011 * The 'basetype' in this case is 'void()'
3012 * and if there's a comma after it, say:
3014 * then the type-information 'void()' can be stored in 'storebase'
3016 static ast_value *parse_typename(parser_t *parser, ast_value **storebase)
3018 ast_value *var, *tmp;
3021 const char *name = NULL;
3022 bool isfield = false;
3023 bool wasarray = false;
3025 ctx = parser_ctx(parser);
3027 /* types may start with a dot */
3028 if (parser->tok == '.') {
3030 /* if we parsed a dot we need a typename now */
3031 if (!parser_next(parser)) {
3032 parseerror(parser, "expected typename for field definition");
3035 if (parser->tok != TOKEN_TYPENAME) {
3036 parseerror(parser, "expected typename");
3041 /* generate the basic type value */
3042 var = ast_value_new(ctx, "<type>", parser_token(parser)->constval.t);
3043 /* do not yet turn into a field - remember:
3044 * .void() foo; is a field too
3045 * .void()() foo; is a function
3049 if (!parser_next(parser)) {
3051 parseerror(parser, "parse error after typename");
3055 /* an opening paren now starts the parameter-list of a function
3056 * this is where original-QC has parameter lists.
3057 * We allow a single parameter list here.
3058 * Much like fteqcc we don't allow `float()() x`
3060 if (parser->tok == '(') {
3061 var = parse_parameter_list(parser, var);
3066 /* store the base if requested */
3068 *storebase = ast_value_copy(var);
3070 tmp = ast_value_new(ctx, "<type:f>", TYPE_FIELD);
3071 tmp->expression.next = (ast_expression*)*storebase;
3076 /* there may be a name now */
3077 if (parser->tok == TOKEN_IDENT) {
3078 name = util_strdup(parser_tokval(parser));
3080 if (!parser_next(parser)) {
3082 parseerror(parser, "error after variable or field declaration");
3087 /* now this may be an array */
3088 if (parser->tok == '[') {
3090 var = parse_arraysize(parser, var);
3095 /* This is the point where we can turn it into a field */
3097 /* turn it into a field if desired */
3098 tmp = ast_value_new(ctx, "<type:f>", TYPE_FIELD);
3099 tmp->expression.next = (ast_expression*)var;
3103 /* now there may be function parens again */
3104 if (parser->tok == '(' && opts_standard == COMPILER_QCC)
3105 parseerror(parser, "C-style function syntax is not allowed in -std=qcc");
3106 if (parser->tok == '(' && wasarray)
3107 parseerror(parser, "arrays as part of a return type is not supported");
3108 while (parser->tok == '(') {
3109 var = parse_parameter_list(parser, var);
3118 /* finally name it */
3120 if (!ast_value_set_name(var, name)) {
3122 parseerror(parser, "internal error: failed to set name");
3125 /* free the name, ast_value_set_name duplicates */
3132 static bool parse_variable(parser_t *parser, ast_block *localblock, bool nofields)
3136 ast_expression *old;
3140 ast_value *basetype = NULL;
3142 bool isparam = false;
3143 bool isvector = false;
3144 bool cleanvar = true;
3145 bool wasarray = false;
3147 varentry_t varent, ve[3];
3149 /* get the first complete variable */
3150 var = parse_typename(parser, &basetype);
3153 ast_delete(basetype);
3157 memset(&varent, 0, sizeof(varent));
3158 memset(&ve, 0, sizeof(ve));
3164 /* Part 0: finish the type */
3165 if (parser->tok == '(') {
3166 if (opts_standard == COMPILER_QCC)
3167 parseerror(parser, "C-style function syntax is not allowed in -std=qcc");
3168 var = parse_parameter_list(parser, var);
3174 /* we only allow 1-dimensional arrays */
3175 if (parser->tok == '[') {
3177 var = parse_arraysize(parser, var);
3183 if (parser->tok == '(' && wasarray) {
3184 parseerror(parser, "arrays as part of a return type is not supported");
3185 /* we'll still parse the type completely for now */
3187 /* for functions returning functions */
3188 while (parser->tok == '(') {
3189 if (opts_standard == COMPILER_QCC)
3190 parseerror(parser, "C-style function syntax is not allowed in -std=qcc");
3191 var = parse_parameter_list(parser, var);
3199 * check for validity: (end_sys_..., multiple-definitions, prototypes, ...)
3200 * Also: if there was a prototype, `var` will be deleted and set to `proto` which
3201 * is then filled with the previous definition and the parameter-names replaced.
3204 /* Deal with end_sys_ vars */
3206 if (!strcmp(var->name, "end_sys_globals")) {
3207 parser->crc_globals = vec_size(parser->globals);
3210 else if (!strcmp(var->name, "end_sys_fields")) {
3211 parser->crc_fields = vec_size(parser->fields);
3214 if (was_end && var->expression.vtype == TYPE_FIELD) {
3215 if (parsewarning(parser, WARN_END_SYS_FIELDS,
3216 "global '%s' hint should not be a field",
3217 parser_tokval(parser)))
3224 if (!nofields && var->expression.vtype == TYPE_FIELD)
3226 /* deal with field declarations */
3227 old = parser_find_field(parser, var->name);
3229 if (parsewarning(parser, WARN_FIELD_REDECLARED, "field `%s` already declared here: %s:%i",
3230 var->name, ast_ctx(old).file, (int)ast_ctx(old).line))
3239 parseerror(parser, "field `%s` already declared here: %s:%i",
3240 var->name, ast_ctx(old).file, ast_ctx(old).line);
3245 if (opts_standard == COMPILER_QCC &&
3246 (old = parser_find_global(parser, var->name)))
3248 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
3249 parseerror(parser, "field `%s` already declared here: %s:%i",
3250 var->name, ast_ctx(old).file, ast_ctx(old).line);
3257 /* deal with other globals */
3258 old = parser_find_global(parser, var->name);
3259 if (old && var->expression.vtype == TYPE_FUNCTION && old->expression.vtype == TYPE_FUNCTION)
3261 /* This is a function which had a prototype */
3262 if (!ast_istype(old, ast_value)) {
3263 parseerror(parser, "internal error: prototype is not an ast_value");
3267 proto = (ast_value*)old;
3268 if (!ast_compare_type((ast_expression*)proto, (ast_expression*)var)) {
3269 parseerror(parser, "conflicting types for `%s`, previous declaration was here: %s:%i",
3271 ast_ctx(proto).file, ast_ctx(proto).line);
3275 /* we need the new parameter-names */
3276 for (i = 0; i < vec_size(proto->expression.params); ++i)
3277 ast_value_set_name(proto->expression.params[i], var->expression.params[i]->name);
3285 parseerror(parser, "global `%s` already declared here: %s:%i",
3286 var->name, ast_ctx(old).file, ast_ctx(old).line);
3290 if (opts_standard == COMPILER_QCC &&
3291 (old = parser_find_field(parser, var->name)))
3293 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
3294 parseerror(parser, "global `%s` already declared here: %s:%i",
3295 var->name, ast_ctx(old).file, ast_ctx(old).line);
3302 else /* it's not a global */
3304 old = parser_find_local(parser, var->name, parser->blocklocal, &isparam);
3305 if (old && !isparam) {
3306 parseerror(parser, "local `%s` already declared here: %s:%i",
3307 var->name, ast_ctx(old).file, (int)ast_ctx(old).line);
3311 old = parser_find_local(parser, var->name, 0, &isparam);
3312 if (old && isparam) {
3313 if (parsewarning(parser, WARN_LOCAL_SHADOWS,
3314 "local `%s` is shadowing a parameter", var->name))
3316 parseerror(parser, "local `%s` already declared here: %s:%i",
3317 var->name, ast_ctx(old).file, (int)ast_ctx(old).line);
3321 if (opts_standard != COMPILER_GMQCC) {
3330 * Create the global/local, and deal with vector types.
3333 if (var->expression.vtype == TYPE_VECTOR)
3335 else if (var->expression.vtype == TYPE_FIELD &&
3336 var->expression.next->expression.vtype == TYPE_VECTOR)
3340 if (!create_vector_members(var, ve)) {
3346 varent.name = util_strdup(var->name);
3347 varent.var = (ast_expression*)var;
3350 /* deal with global variables, fields, functions */
3351 if (!nofields && var->expression.vtype == TYPE_FIELD) {
3352 vec_push(parser->fields, varent);
3354 for (i = 0; i < 3; ++i)
3355 vec_push(parser->fields, ve[i]);
3359 vec_push(parser->globals, varent);
3361 for (i = 0; i < 3; ++i)
3362 vec_push(parser->globals, ve[i]);
3366 vec_push(parser->locals, varent);
3367 vec_push(localblock->locals, var);
3369 for (i = 0; i < 3; ++i) {
3370 vec_push(parser->locals, ve[i]);
3371 ast_block_collect(localblock, ve[i].var);
3372 ve[i].var = NULL; /* from here it's being collected in the block */
3378 ve[0].name = ve[1].name = ve[2].name = NULL;
3379 ve[0].var = ve[1].var = ve[2].var = NULL;
3385 if (var->expression.vtype == TYPE_ARRAY) {
3387 snprintf(name, sizeof(name), "%s##SET", var->name);
3388 if (!parser_create_array_setter(parser, var, name))
3390 snprintf(name, sizeof(name), "%s##GET", var->name);
3391 if (!parser_create_array_getter(parser, var, var->expression.next, name))
3394 else if (!localblock && !nofields &&
3395 var->expression.vtype == TYPE_FIELD &&
3396 var->expression.next->expression.vtype == TYPE_ARRAY)
3399 ast_expression *telem;
3401 ast_value *array = (ast_value*)var->expression.next;
3403 if (!ast_istype(var->expression.next, ast_value)) {
3404 parseerror(parser, "internal error: field element type must be an ast_value");
3408 snprintf(name, sizeof(name), "%s##SETF", var->name);
3409 if (!parser_create_array_field_setter(parser, array, name))
3412 telem = ast_type_copy(ast_ctx(var), array->expression.next);
3413 tfield = ast_value_new(ast_ctx(var), "<.type>", TYPE_FIELD);
3414 tfield->expression.next = telem;
3415 snprintf(name, sizeof(name), "%s##GETFP", var->name);
3416 if (!parser_create_array_getter(parser, array, (ast_expression*)tfield, name)) {
3424 if (parser->tok == ';') {
3425 ast_delete(basetype);
3426 if (!parser_next(parser)) {
3427 parseerror(parser, "error after variable declaration");
3433 if (parser->tok == ',')
3436 if (!var || (!localblock && !nofields && basetype->expression.vtype == TYPE_FIELD)) {
3437 parseerror(parser, "missing comma or semicolon while parsing variables");
3441 if (localblock && opts_standard == COMPILER_QCC) {
3442 if (parsewarning(parser, WARN_LOCAL_CONSTANTS,
3443 "initializing expression turns variable `%s` into a constant in this standard",
3450 if (parser->tok != '{') {
3451 if (parser->tok != '=') {
3452 parseerror(parser, "missing semicolon or initializer");
3456 if (!parser_next(parser)) {
3457 parseerror(parser, "error parsing initializer");
3461 else if (opts_standard == COMPILER_QCC) {
3462 parseerror(parser, "expected '=' before function body in this standard");
3465 if (parser->tok == '#') {
3469 parseerror(parser, "cannot declare builtins within functions");
3472 if (var->expression.vtype != TYPE_FUNCTION) {
3473 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
3476 if (!parser_next(parser)) {
3477 parseerror(parser, "expected builtin number");
3480 if (parser->tok != TOKEN_INTCONST) {
3481 parseerror(parser, "builtin number must be an integer constant");
3484 if (parser_token(parser)->constval.i <= 0) {
3485 parseerror(parser, "builtin number must be an integer greater than zero");
3489 func = ast_function_new(ast_ctx(var), var->name, var);
3491 parseerror(parser, "failed to allocate function for `%s`", var->name);
3494 vec_push(parser->functions, func);
3496 func->builtin = -parser_token(parser)->constval.i;
3498 if (!parser_next(parser)) {
3499 parseerror(parser, "expected comma or semicolon");
3500 ast_function_delete(func);
3501 var->constval.vfunc = NULL;
3505 else if (parser->tok == '{' || parser->tok == '[')
3508 parseerror(parser, "cannot declare functions within functions");
3512 if (!parse_function_body(parser, var))
3514 ast_delete(basetype);
3517 ast_expression *cexp;
3520 cexp = parse_expression_leave(parser, true);
3525 cval = (ast_value*)cexp;
3526 if (!ast_istype(cval, ast_value) || !cval->isconst)
3527 parseerror(parser, "cannot initialize a global constant variable with a non-constant expression");
3530 var->isconst = true;
3531 if (cval->expression.vtype == TYPE_STRING)
3532 var->constval.vstring = parser_strdup(cval->constval.vstring);
3534 memcpy(&var->constval, &cval->constval, sizeof(var->constval));
3538 shunt sy = { NULL, NULL };
3539 vec_push(sy.out, syexp(ast_ctx(var), (ast_expression*)var));
3540 vec_push(sy.out, syexp(ast_ctx(cexp), (ast_expression*)cexp));
3541 vec_push(sy.ops, syop(ast_ctx(var), parser->assign_op));
3542 if (!parser_sy_pop(parser, &sy))
3545 if (vec_size(sy.out) != 1 && vec_size(sy.ops) != 0)
3546 parseerror(parser, "internal error: leaked operands");
3547 vec_push(localblock->exprs, (ast_expression*)sy.out[0].out);
3555 if (parser->tok == ',') {
3556 if (!parser_next(parser)) {
3557 parseerror(parser, "expected another variable");
3561 if (parser->tok != TOKEN_IDENT) {
3562 parseerror(parser, "expected another variable");
3565 var = ast_value_copy(basetype);
3567 ast_value_set_name(var, parser_tokval(parser));
3568 if (!parser_next(parser)) {
3569 parseerror(parser, "error parsing variable declaration");
3575 if (parser->tok != ';') {
3576 parseerror(parser, "missing semicolon after variables");
3580 if (!parser_next(parser)) {
3581 parseerror(parser, "parse error after variable declaration");
3585 ast_delete(basetype);
3589 if (cleanvar && var)
3591 ast_delete(basetype);
3595 ast_delete(basetype);
3596 if (cleanvar && var)
3598 if (varent.name) mem_d(varent.name);
3599 if (ve[0].name) mem_d(ve[0].name);
3600 if (ve[1].name) mem_d(ve[1].name);
3601 if (ve[2].name) mem_d(ve[2].name);
3602 if (ve[0].var) mem_d(ve[0].var);
3603 if (ve[1].var) mem_d(ve[1].var);
3604 if (ve[2].var) mem_d(ve[2].var);
3608 static bool parser_global_statement(parser_t *parser)
3610 if (parser->tok == TOKEN_TYPENAME || parser->tok == '.')
3612 return parse_variable(parser, NULL, false);
3614 else if (parser->tok == TOKEN_KEYWORD)
3616 /* handle 'var' and 'const' */
3617 if (!strcmp(parser_tokval(parser), "var")) {
3618 if (!parser_next(parser)) {
3619 parseerror(parser, "expected variable declaration after 'var'");
3622 return parse_variable(parser, NULL, true);
3626 else if (parser->tok == '$')
3628 if (!parser_next(parser)) {
3629 parseerror(parser, "parse error");
3635 parseerror(parser, "unexpected token: %s", parser->lex->tok.value);
3641 static uint16_t progdefs_crc_sum(uint16_t old, const char *str)
3643 return util_crc16(old, str, strlen(str));
3646 static void progdefs_crc_file(const char *str)
3648 /* write to progdefs.h here */
3652 static uint16_t progdefs_crc_both(uint16_t old, const char *str)
3654 old = progdefs_crc_sum(old, str);
3655 progdefs_crc_file(str);
3659 static void generate_checksum(parser_t *parser)
3661 uint16_t crc = 0xFFFF;
3664 crc = progdefs_crc_both(crc, "\n/* file generated by qcc, do not modify */\n\ntypedef struct\n{");
3665 crc = progdefs_crc_sum(crc, "\tint\tpad[28];\n");
3667 progdefs_crc_file("\tint\tpad;\n");
3668 progdefs_crc_file("\tint\tofs_return[3];\n");
3669 progdefs_crc_file("\tint\tofs_parm0[3];\n");
3670 progdefs_crc_file("\tint\tofs_parm1[3];\n");
3671 progdefs_crc_file("\tint\tofs_parm2[3];\n");
3672 progdefs_crc_file("\tint\tofs_parm3[3];\n");
3673 progdefs_crc_file("\tint\tofs_parm4[3];\n");
3674 progdefs_crc_file("\tint\tofs_parm5[3];\n");
3675 progdefs_crc_file("\tint\tofs_parm6[3];\n");
3676 progdefs_crc_file("\tint\tofs_parm7[3];\n");
3678 for (i = 0; i < parser->crc_globals; ++i) {
3679 if (!ast_istype(parser->globals[i].var, ast_value))
3681 switch (parser->globals[i].var->expression.vtype) {
3682 case TYPE_FLOAT: crc = progdefs_crc_both(crc, "\tfloat\t"); break;
3683 case TYPE_VECTOR: crc = progdefs_crc_both(crc, "\tvec3_t\t"); break;
3684 case TYPE_STRING: crc = progdefs_crc_both(crc, "\tstring_t\t"); break;
3685 case TYPE_FUNCTION: crc = progdefs_crc_both(crc, "\tfunc_t\t"); break;
3687 crc = progdefs_crc_both(crc, "\tint\t");
3690 crc = progdefs_crc_both(crc, parser->globals[i].name);
3691 crc = progdefs_crc_both(crc, ";\n");
3693 crc = progdefs_crc_both(crc, "} globalvars_t;\n\ntypedef struct\n{\n");
3694 for (i = 0; i < parser->crc_fields; ++i) {
3695 if (!ast_istype(parser->fields[i].var, ast_value))
3697 switch (parser->fields[i].var->expression.next->expression.vtype) {
3698 case TYPE_FLOAT: crc = progdefs_crc_both(crc, "\tfloat\t"); break;
3699 case TYPE_VECTOR: crc = progdefs_crc_both(crc, "\tvec3_t\t"); break;
3700 case TYPE_STRING: crc = progdefs_crc_both(crc, "\tstring_t\t"); break;
3701 case TYPE_FUNCTION: crc = progdefs_crc_both(crc, "\tfunc_t\t"); break;
3703 crc = progdefs_crc_both(crc, "\tint\t");
3706 crc = progdefs_crc_both(crc, parser->fields[i].name);
3707 crc = progdefs_crc_both(crc, ";\n");
3709 crc = progdefs_crc_both(crc, "} entvars_t;\n\n");
3714 static parser_t *parser;
3719 parser = (parser_t*)mem_a(sizeof(parser_t));
3723 memset(parser, 0, sizeof(*parser));
3725 for (i = 0; i < operator_count; ++i) {
3726 if (operators[i].id == opid1('=')) {
3727 parser->assign_op = operators+i;
3731 if (!parser->assign_op) {
3732 printf("internal error: initializing parser: failed to find assign operator\n");
3739 bool parser_compile()
3741 /* initial lexer/parser state */
3742 parser->lex->flags.noops = true;
3744 if (parser_next(parser))
3746 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
3748 if (!parser_global_statement(parser)) {
3749 if (parser->tok == TOKEN_EOF)
3750 parseerror(parser, "unexpected eof");
3751 else if (!parser->errors)
3752 parseerror(parser, "there have been errors, bailing out");
3753 lex_close(parser->lex);
3759 parseerror(parser, "parse error");
3760 lex_close(parser->lex);
3765 lex_close(parser->lex);
3768 return !parser->errors;
3771 bool parser_compile_file(const char *filename)
3773 parser->lex = lex_open(filename);
3775 con_err("failed to open file \"%s\"\n", filename);
3778 return parser_compile();
3781 bool parser_compile_string_len(const char *name, const char *str, size_t len)
3783 parser->lex = lex_open_string(str, len, name);
3785 con_err("failed to create lexer for string \"%s\"\n", name);
3788 return parser_compile();
3791 bool parser_compile_string(const char *name, const char *str)
3793 parser->lex = lex_open_string(str, strlen(str), name);
3795 con_err("failed to create lexer for string \"%s\"\n", name);
3798 return parser_compile();
3801 void parser_cleanup()
3804 for (i = 0; i < vec_size(parser->accessors); ++i) {
3805 ast_delete(parser->accessors[i]->constval.vfunc);
3806 parser->accessors[i]->constval.vfunc = NULL;
3807 ast_delete(parser->accessors[i]);
3809 for (i = 0; i < vec_size(parser->functions); ++i) {
3810 ast_delete(parser->functions[i]);
3812 for (i = 0; i < vec_size(parser->imm_vector); ++i) {
3813 ast_delete(parser->imm_vector[i]);
3815 for (i = 0; i < vec_size(parser->imm_string); ++i) {
3816 ast_delete(parser->imm_string[i]);
3818 for (i = 0; i < vec_size(parser->imm_float); ++i) {
3819 ast_delete(parser->imm_float[i]);
3821 for (i = 0; i < vec_size(parser->fields); ++i) {
3822 ast_delete(parser->fields[i].var);
3823 mem_d(parser->fields[i].name);
3825 for (i = 0; i < vec_size(parser->globals); ++i) {
3826 ast_delete(parser->globals[i].var);
3827 mem_d(parser->globals[i].name);
3829 vec_free(parser->accessors);
3830 vec_free(parser->functions);
3831 vec_free(parser->imm_vector);
3832 vec_free(parser->imm_string);
3833 vec_free(parser->imm_float);
3834 vec_free(parser->globals);
3835 vec_free(parser->fields);
3836 vec_free(parser->locals);
3841 bool parser_finish(const char *output)
3847 if (!parser->errors)
3849 ir = ir_builder_new("gmqcc_out");
3851 con_out("failed to allocate builder\n");
3855 for (i = 0; i < vec_size(parser->fields); ++i) {
3858 if (!ast_istype(parser->fields[i].var, ast_value))
3860 field = (ast_value*)parser->fields[i].var;
3861 isconst = field->isconst;
3862 field->isconst = false;
3863 if (!ast_global_codegen((ast_value*)field, ir, true)) {
3864 con_out("failed to generate field %s\n", field->name);
3865 ir_builder_delete(ir);
3870 ast_expression *subtype;
3871 field->isconst = true;
3872 subtype = field->expression.next;
3873 ifld = ir_builder_create_field(ir, field->name, subtype->expression.vtype);
3874 if (subtype->expression.vtype == TYPE_FIELD)
3875 ifld->fieldtype = subtype->expression.next->expression.vtype;
3876 else if (subtype->expression.vtype == TYPE_FUNCTION)
3877 ifld->outtype = subtype->expression.next->expression.vtype;
3878 (void)!ir_value_set_field(field->ir_v, ifld);
3881 for (i = 0; i < vec_size(parser->globals); ++i) {
3883 if (!ast_istype(parser->globals[i].var, ast_value))
3885 asvalue = (ast_value*)(parser->globals[i].var);
3886 if (!asvalue->uses && !asvalue->isconst && asvalue->expression.vtype != TYPE_FUNCTION) {
3887 if (strcmp(asvalue->name, "end_sys_globals") &&
3888 strcmp(asvalue->name, "end_sys_fields"))
3890 retval = retval && !genwarning(ast_ctx(asvalue), WARN_UNUSED_VARIABLE,
3891 "unused global: `%s`", asvalue->name);
3894 if (!ast_global_codegen(asvalue, ir, false)) {
3895 con_out("failed to generate global %s\n", parser->globals[i].name);
3896 ir_builder_delete(ir);
3900 for (i = 0; i < vec_size(parser->imm_float); ++i) {
3901 if (!ast_global_codegen(parser->imm_float[i], ir, false)) {
3902 con_out("failed to generate global %s\n", parser->imm_float[i]->name);
3903 ir_builder_delete(ir);
3907 for (i = 0; i < vec_size(parser->imm_string); ++i) {
3908 if (!ast_global_codegen(parser->imm_string[i], ir, false)) {
3909 con_out("failed to generate global %s\n", parser->imm_string[i]->name);
3910 ir_builder_delete(ir);
3914 for (i = 0; i < vec_size(parser->imm_vector); ++i) {
3915 if (!ast_global_codegen(parser->imm_vector[i], ir, false)) {
3916 con_out("failed to generate global %s\n", parser->imm_vector[i]->name);
3917 ir_builder_delete(ir);
3921 for (i = 0; i < vec_size(parser->globals); ++i) {
3923 if (!ast_istype(parser->globals[i].var, ast_value))
3925 asvalue = (ast_value*)(parser->globals[i].var);
3926 if (asvalue->setter) {
3927 if (!ast_global_codegen(asvalue->setter, ir, false) ||
3928 !ast_function_codegen(asvalue->setter->constval.vfunc, ir) ||
3929 !ir_function_finalize(asvalue->setter->constval.vfunc->ir_func))
3931 printf("failed to generate setter for %s\n", parser->globals[i].name);
3932 ir_builder_delete(ir);
3936 if (asvalue->getter) {
3937 if (!ast_global_codegen(asvalue->getter, ir, false) ||
3938 !ast_function_codegen(asvalue->getter->constval.vfunc, ir) ||
3939 !ir_function_finalize(asvalue->getter->constval.vfunc->ir_func))
3941 printf("failed to generate getter for %s\n", parser->globals[i].name);
3942 ir_builder_delete(ir);
3947 for (i = 0; i < vec_size(parser->fields); ++i) {
3949 asvalue = (ast_value*)(parser->fields[i].var->expression.next);
3951 if (!ast_istype((ast_expression*)asvalue, ast_value))
3953 if (asvalue->expression.vtype != TYPE_ARRAY)
3955 if (asvalue->setter) {
3956 if (!ast_global_codegen(asvalue->setter, ir, false) ||
3957 !ast_function_codegen(asvalue->setter->constval.vfunc, ir) ||
3958 !ir_function_finalize(asvalue->setter->constval.vfunc->ir_func))
3960 printf("failed to generate setter for %s\n", parser->fields[i].name);
3961 ir_builder_delete(ir);
3965 if (asvalue->getter) {
3966 if (!ast_global_codegen(asvalue->getter, ir, false) ||
3967 !ast_function_codegen(asvalue->getter->constval.vfunc, ir) ||
3968 !ir_function_finalize(asvalue->getter->constval.vfunc->ir_func))
3970 printf("failed to generate getter for %s\n", parser->fields[i].name);
3971 ir_builder_delete(ir);
3976 for (i = 0; i < vec_size(parser->functions); ++i) {
3977 if (!ast_function_codegen(parser->functions[i], ir)) {
3978 con_out("failed to generate function %s\n", parser->functions[i]->name);
3979 ir_builder_delete(ir);
3984 ir_builder_dump(ir, con_out);
3985 for (i = 0; i < vec_size(parser->functions); ++i) {
3986 if (!ir_function_finalize(parser->functions[i]->ir_func)) {
3987 con_out("failed to finalize function %s\n", parser->functions[i]->name);
3988 ir_builder_delete(ir);
3995 ir_builder_dump(ir, con_out);
3997 generate_checksum(parser);
3999 if (!ir_builder_generate(ir, output)) {
4000 con_out("*** failed to generate output file\n");
4001 ir_builder_delete(ir);
4006 ir_builder_delete(ir);
4010 con_out("*** there were compile errors\n");