16 MEM_VECTOR_MAKE(varentry_t, globals);
17 MEM_VECTOR_MAKE(varentry_t, fields);
18 MEM_VECTOR_MAKE(ast_function*, functions);
19 MEM_VECTOR_MAKE(ast_value*, imm_float);
20 MEM_VECTOR_MAKE(ast_value*, imm_string);
21 MEM_VECTOR_MAKE(ast_value*, imm_vector);
23 ast_value *imm_float_zero;
24 ast_value *imm_vector_zero;
26 ast_function *function;
27 MEM_VECTOR_MAKE(varentry_t, locals);
32 /* TYPE_FIELD -> parser_find_fields is used instead of find_var
33 * TODO: TYPE_VECTOR -> x, y and z are accepted in the gmqcc standard
34 * anything else: type error
39 MEM_VEC_FUNCTIONS(parser_t, varentry_t, globals)
40 MEM_VEC_FUNCTIONS(parser_t, varentry_t, fields)
41 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_float)
42 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_string)
43 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_vector)
44 MEM_VEC_FUNCTIONS(parser_t, varentry_t, locals)
45 MEM_VEC_FUNCTIONS(parser_t, ast_function*, functions)
47 static void parser_pop_local(parser_t *parser);
48 static bool parser_variable(parser_t *parser, ast_block *localblock);
49 static ast_block* parser_parse_block(parser_t *parser);
50 static bool parser_parse_block_into(parser_t *parser, ast_block *block);
51 static ast_expression* parser_parse_statement_or_block(parser_t *parser);
52 static ast_expression* parser_expression_leave(parser_t *parser, bool stopatcomma);
53 static ast_expression* parser_expression(parser_t *parser, bool stopatcomma);
55 void parseerror(parser_t *parser, const char *fmt, ...)
62 vprintmsg(LVL_ERROR, parser->lex->tok->ctx.file, parser->lex->tok->ctx.line, "parse error", fmt, ap);
66 /* returns true if it counts as an error */
67 bool GMQCC_WARN parsewarning(parser_t *parser, int warntype, const char *fmt, ...)
70 int lvl = LVL_WARNING;
72 if (!OPTS_WARN(warntype))
81 vprintmsg(lvl, parser->lex->tok->ctx.file, parser->lex->tok->ctx.line, "warning", fmt, ap);
87 /**********************************************************************
88 * some maths used for constant folding
91 vector vec3_add(vector a, vector b)
100 vector vec3_sub(vector a, vector b)
109 qcfloat vec3_mulvv(vector a, vector b)
111 return (a.x * b.x + a.y * b.y + a.z * b.z);
114 vector vec3_mulvf(vector a, float b)
123 /**********************************************************************
127 bool parser_next(parser_t *parser)
129 /* lex_do kills the previous token */
130 parser->tok = lex_do(parser->lex);
131 if (parser->tok == TOKEN_EOF || parser->tok >= TOKEN_ERROR)
136 /* lift a token out of the parser so it's not destroyed by parser_next */
137 token *parser_lift(parser_t *parser)
139 token *tok = parser->lex->tok;
140 parser->lex->tok = NULL;
144 #define parser_tokval(p) (p->lex->tok->value)
145 #define parser_token(p) (p->lex->tok)
146 #define parser_ctx(p) (p->lex->tok->ctx)
148 ast_value* parser_const_float(parser_t *parser, double d)
152 for (i = 0; i < parser->imm_float_count; ++i) {
153 if (parser->imm_float[i]->constval.vfloat == d)
154 return parser->imm_float[i];
156 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_FLOAT);
158 out->constval.vfloat = d;
159 if (!parser_t_imm_float_add(parser, out)) {
160 ast_value_delete(out);
166 ast_value* parser_const_float_0(parser_t *parser)
168 if (!parser->imm_float_zero)
169 parser->imm_float_zero = parser_const_float(parser, 0);
170 return parser->imm_float_zero;
173 char *parser_strdup(const char *str)
176 /* actually dup empty strings */
177 char *out = mem_a(1);
181 return util_strdup(str);
184 ast_value* parser_const_string(parser_t *parser, const char *str)
188 for (i = 0; i < parser->imm_string_count; ++i) {
189 if (!strcmp(parser->imm_string[i]->constval.vstring, str))
190 return parser->imm_string[i];
192 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_STRING);
194 out->constval.vstring = parser_strdup(str);
195 if (!parser_t_imm_string_add(parser, out)) {
196 ast_value_delete(out);
202 ast_value* parser_const_vector(parser_t *parser, vector v)
206 for (i = 0; i < parser->imm_vector_count; ++i) {
207 if (!memcmp(&parser->imm_vector[i]->constval.vvec, &v, sizeof(v)))
208 return parser->imm_vector[i];
210 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_VECTOR);
212 out->constval.vvec = v;
213 if (!parser_t_imm_vector_add(parser, out)) {
214 ast_value_delete(out);
220 ast_value* parser_const_vector_f(parser_t *parser, float x, float y, float z)
226 return parser_const_vector(parser, v);
229 ast_value* parser_const_vector_0(parser_t *parser)
231 if (!parser->imm_vector_zero)
232 parser->imm_vector_zero = parser_const_vector_f(parser, 0, 0, 0);
233 return parser->imm_vector_zero;
236 ast_expression* parser_find_field(parser_t *parser, const char *name)
239 for (i = 0; i < parser->fields_count; ++i) {
240 if (!strcmp(parser->fields[i].name, name))
241 return parser->fields[i].var;
246 ast_expression* parser_find_global(parser_t *parser, const char *name)
249 for (i = 0; i < parser->globals_count; ++i) {
250 if (!strcmp(parser->globals[i].name, name))
251 return parser->globals[i].var;
256 ast_expression* parser_find_param(parser_t *parser, const char *name)
260 if (!parser->function)
262 fun = parser->function->vtype;
263 for (i = 0; i < fun->expression.params_count; ++i) {
264 if (!strcmp(fun->expression.params[i]->name, name))
265 return (ast_expression*)(fun->expression.params[i]);
270 ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto, bool *isparam)
274 for (i = parser->locals_count; i > upto;) {
276 if (!strcmp(parser->locals[i].name, name))
277 return parser->locals[i].var;
280 return parser_find_param(parser, name);
283 ast_expression* parser_find_var(parser_t *parser, const char *name)
287 v = parser_find_local(parser, name, 0, &dummy);
288 if (!v) v = parser_find_global(parser, name);
293 MEM_VECTOR_MAKE(ast_value*, p);
295 MEM_VEC_FUNCTIONS(paramlist_t, ast_value*, p)
297 static ast_value *parser_parse_type(parser_t *parser, int basetype, bool *isfunc)
301 lex_ctx ctx = parser_ctx(parser);
302 int vtype = basetype;
306 MEM_VECTOR_INIT(¶ms, p);
310 if (parser->tok == '(') {
315 bool isfield = false;
318 if (!parser_next(parser))
321 if (parser->tok == ')')
324 if (parser->tok == '.') {
326 if (!parser_next(parser)) {
327 parseerror(parser, "expected field parameter type");
332 temptype = parser_token(parser)->constval.t;
333 if (!parser_next(parser))
336 param = parser_parse_type(parser, temptype, &dummy);
342 if (parser->tok == TOKEN_IDENT) {
343 /* named parameter */
344 if (!ast_value_set_name(param, parser_tokval(parser)))
346 if (!parser_next(parser))
351 fld = ast_value_new(ctx, param->name, TYPE_FIELD);
352 fld->expression.next = (ast_expression*)param;
356 if (!paramlist_t_p_add(¶ms, param)) {
357 parseerror(parser, "Out of memory while parsing typename");
361 if (parser->tok == ',')
363 if (parser->tok == ')')
365 parseerror(parser, "Unexpected token");
368 if (!parser_next(parser))
372 var = ast_value_new(ctx, "<unnamed>", vtype);
375 MEM_VECTOR_MOVE(¶ms, p, &var->expression, params);
378 for (i = 0; i < params.p_count; ++i)
379 ast_value_delete(params.p[i]);
380 MEM_VECTOR_CLEAR(¶ms, p);
386 size_t etype; /* 0 = expression, others are operators */
390 ast_block *block; /* for commas and function calls */
395 MEM_VECTOR_MAKE(sy_elem, out);
396 MEM_VECTOR_MAKE(sy_elem, ops);
398 MEM_VEC_FUNCTIONS(shunt, sy_elem, out)
399 MEM_VEC_FUNCTIONS(shunt, sy_elem, ops)
401 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
411 static sy_elem syblock(lex_ctx ctx, ast_block *v) {
414 e.out = (ast_expression*)v;
421 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
423 e.etype = 1 + (op - operators);
431 static sy_elem syparen(lex_ctx ctx, int p, size_t off) {
443 # define DEBUGSHUNTDO(x) x
445 # define DEBUGSHUNTDO(x)
448 static bool parser_sy_pop(parser_t *parser, shunt *sy)
452 ast_expression *out = NULL;
453 ast_expression *exprs[3];
454 ast_block *blocks[3];
455 ast_value *asvalue[3];
457 qcint generated_op = 0;
459 if (!sy->ops_count) {
460 parseerror(parser, "internal error: missing operator");
464 if (sy->ops[sy->ops_count-1].paren) {
465 parseerror(parser, "unmatched parenthesis");
469 op = &operators[sy->ops[sy->ops_count-1].etype - 1];
470 ctx = sy->ops[sy->ops_count-1].ctx;
472 DEBUGSHUNTDO(printf("apply %s\n", op->op));
474 if (sy->out_count < op->operands) {
475 parseerror(parser, "internal error: not enough operands: %i (operator %s (%i))", sy->out_count,
476 op->op, (int)op->id);
482 sy->out_count -= op->operands;
483 for (i = 0; i < op->operands; ++i) {
484 exprs[i] = sy->out[sy->out_count+i].out;
485 blocks[i] = sy->out[sy->out_count+i].block;
486 asvalue[i] = (ast_value*)exprs[i];
489 if (blocks[0] && !blocks[0]->exprs_count && op->id != opid1(',')) {
490 parseerror(parser, "internal error: operator cannot be applied on empty blocks");
494 #define NotSameType(T) \
495 (exprs[0]->expression.vtype != exprs[1]->expression.vtype || \
496 exprs[0]->expression.vtype != T)
497 #define CanConstFold1(A) \
498 (ast_istype((A), ast_value) && ((ast_value*)(A))->isconst)
499 #define CanConstFold(A, B) \
500 (CanConstFold1(A) && CanConstFold1(B))
501 #define ConstV(i) (asvalue[(i)]->constval.vvec)
502 #define ConstF(i) (asvalue[(i)]->constval.vfloat)
503 #define ConstS(i) (asvalue[(i)]->constval.vstring)
507 parseerror(parser, "internal error: unhandled operator: %s (%i)", op->op, (int)op->id);
511 if (exprs[0]->expression.vtype == TYPE_ENTITY) {
512 if (exprs[1]->expression.vtype != TYPE_FIELD) {
513 parseerror(parser, "type error: right hand of member-operand should be an entity-field");
516 out = (ast_expression*)ast_entfield_new(ctx, exprs[0], exprs[1]);
518 else if (exprs[0]->expression.vtype == TYPE_VECTOR) {
519 parseerror(parser, "internal error: vector access is not supposed to be handled at this point");
523 parseerror(parser, "type error: member-of operator on something that is not an entity or vector");
530 if (!ast_block_exprs_add(blocks[0], exprs[1]))
533 blocks[0] = ast_block_new(ctx);
534 if (!ast_block_exprs_add(blocks[0], exprs[0]) ||
535 !ast_block_exprs_add(blocks[0], exprs[1]))
540 if (!ast_block_set_type(blocks[0], exprs[1]))
543 sy->out[sy->out_count++] = syblock(ctx, blocks[0]);
547 switch (exprs[0]->expression.vtype) {
549 if (CanConstFold1(exprs[0]))
550 out = (ast_expression*)parser_const_float(parser, -ConstF(0));
552 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F,
553 (ast_expression*)parser_const_float_0(parser),
557 if (CanConstFold1(exprs[0]))
558 out = (ast_expression*)parser_const_vector_f(parser,
559 -ConstV(0).x, -ConstV(0).y, -ConstV(0).z);
561 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V,
562 (ast_expression*)parser_const_vector_0(parser),
566 parseerror(parser, "invalid types used in expression: cannot negate type %s",
567 type_name[exprs[0]->expression.vtype]);
573 switch (exprs[0]->expression.vtype) {
575 if (CanConstFold1(exprs[0]))
576 out = (ast_expression*)parser_const_float(parser, !ConstF(0));
578 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, exprs[0]);
581 if (CanConstFold1(exprs[0]))
582 out = (ast_expression*)parser_const_float(parser,
583 (!ConstV(0).x && !ConstV(0).y && !ConstV(0).z));
585 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_V, exprs[0]);
588 if (CanConstFold1(exprs[0]))
589 out = (ast_expression*)parser_const_float(parser, !ConstS(0) || !*ConstS(0));
591 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_S, exprs[0]);
593 /* we don't constant-fold NOT for these types */
595 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_ENT, exprs[0]);
598 out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_FNC, exprs[0]);
601 parseerror(parser, "invalid types used in expression: cannot logically negate type %s",
602 type_name[exprs[0]->expression.vtype]);
608 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
609 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
611 parseerror(parser, "invalid types used in expression: cannot add type %s and %s",
612 type_name[exprs[0]->expression.vtype],
613 type_name[exprs[1]->expression.vtype]);
616 switch (exprs[0]->expression.vtype) {
618 if (CanConstFold(exprs[0], exprs[1]))
620 out = (ast_expression*)parser_const_float(parser, ConstF(0) + ConstF(1));
623 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
626 if (CanConstFold(exprs[0], exprs[1]))
627 out = (ast_expression*)parser_const_vector(parser, vec3_add(ConstV(0), ConstV(1)));
629 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
632 parseerror(parser, "invalid types used in expression: cannot add type %s and %s",
633 type_name[exprs[0]->expression.vtype],
634 type_name[exprs[1]->expression.vtype]);
639 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
640 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
642 parseerror(parser, "invalid types used in expression: cannot subtract type %s from %s",
643 type_name[exprs[1]->expression.vtype],
644 type_name[exprs[0]->expression.vtype]);
647 switch (exprs[0]->expression.vtype) {
649 if (CanConstFold(exprs[0], exprs[1]))
650 out = (ast_expression*)parser_const_float(parser, ConstF(0) - ConstF(1));
652 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
655 if (CanConstFold(exprs[0], exprs[1]))
656 out = (ast_expression*)parser_const_vector(parser, vec3_sub(ConstV(0), ConstV(1)));
658 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
661 parseerror(parser, "invalid types used in expression: cannot subtract type %s from %s",
662 type_name[exprs[1]->expression.vtype],
663 type_name[exprs[0]->expression.vtype]);
668 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
669 exprs[0]->expression.vtype != TYPE_VECTOR &&
670 exprs[0]->expression.vtype != TYPE_FLOAT &&
671 exprs[1]->expression.vtype != TYPE_VECTOR &&
672 exprs[1]->expression.vtype != TYPE_FLOAT)
674 parseerror(parser, "invalid types used in expression: cannot multiply types %s and %s",
675 type_name[exprs[1]->expression.vtype],
676 type_name[exprs[0]->expression.vtype]);
679 switch (exprs[0]->expression.vtype) {
681 if (exprs[1]->expression.vtype == TYPE_VECTOR)
683 if (CanConstFold(exprs[0], exprs[1]))
684 out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(1), ConstF(0)));
686 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
690 if (CanConstFold(exprs[0], exprs[1]))
691 out = (ast_expression*)parser_const_float(parser, ConstF(0) * ConstF(1));
693 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
697 if (exprs[1]->expression.vtype == TYPE_FLOAT)
699 if (CanConstFold(exprs[0], exprs[1]))
700 out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(0), ConstF(1)));
702 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
706 if (CanConstFold(exprs[0], exprs[1]))
707 out = (ast_expression*)parser_const_float(parser, vec3_mulvv(ConstV(0), ConstV(1)));
709 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
713 parseerror(parser, "invalid types used in expression: cannot multiply types %s and %s",
714 type_name[exprs[1]->expression.vtype],
715 type_name[exprs[0]->expression.vtype]);
720 if (NotSameType(TYPE_FLOAT)) {
721 parseerror(parser, "invalid types used in expression: cannot divide types %s and %s",
722 type_name[exprs[0]->expression.vtype],
723 type_name[exprs[1]->expression.vtype]);
726 if (CanConstFold(exprs[0], exprs[1]))
727 out = (ast_expression*)parser_const_float(parser, ConstF(0) / ConstF(1));
729 out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
733 parseerror(parser, "qc does not have a modulo operator");
737 if (NotSameType(TYPE_FLOAT)) {
738 parseerror(parser, "invalid types used in expression: cannot perform bit operations between types %s and %s",
739 type_name[exprs[0]->expression.vtype],
740 type_name[exprs[1]->expression.vtype]);
743 if (CanConstFold(exprs[0], exprs[1]))
744 out = (ast_expression*)parser_const_float(parser,
745 (op->id == opid1('|') ? (float)( ((qcint)ConstF(0)) | ((qcint)ConstF(1)) ) :
746 (float)( ((qcint)ConstF(0)) & ((qcint)ConstF(1)) ) ));
748 out = (ast_expression*)ast_binary_new(ctx,
749 (op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
753 parseerror(parser, "TODO: bitxor");
758 case opid3('<','<','='):
759 case opid3('>','>','='):
760 parseerror(parser, "TODO: shifts");
764 generated_op += 1; /* INSTR_OR */
766 generated_op += INSTR_AND;
767 if (NotSameType(TYPE_FLOAT)) {
768 parseerror(parser, "invalid types used in expression: cannot perform logical operations between types %s and %s",
769 type_name[exprs[0]->expression.vtype],
770 type_name[exprs[1]->expression.vtype]);
771 parseerror(parser, "TODO: logical ops for arbitrary types using INSTR_NOT");
772 parseerror(parser, "TODO: optional early out");
775 if (opts_standard == COMPILER_GMQCC)
776 printf("TODO: early out logic\n");
777 if (CanConstFold(exprs[0], exprs[1]))
778 out = (ast_expression*)parser_const_float(parser,
779 (generated_op == INSTR_OR ? (ConstF(0) || ConstF(1)) : (ConstF(0) && ConstF(1))));
781 out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
785 generated_op += 1; /* INSTR_GT */
787 generated_op += 1; /* INSTR_LT */
788 case opid2('>', '='):
789 generated_op += 1; /* INSTR_GE */
790 case opid2('<', '='):
791 generated_op += INSTR_LE;
792 if (NotSameType(TYPE_FLOAT)) {
793 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
794 type_name[exprs[0]->expression.vtype],
795 type_name[exprs[1]->expression.vtype]);
798 out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
800 case opid2('!', '='):
801 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
802 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
803 type_name[exprs[0]->expression.vtype],
804 type_name[exprs[1]->expression.vtype]);
807 out = (ast_expression*)ast_binary_new(ctx, type_ne_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
809 case opid2('=', '='):
810 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
811 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
812 type_name[exprs[0]->expression.vtype],
813 type_name[exprs[1]->expression.vtype]);
816 out = (ast_expression*)ast_binary_new(ctx, type_eq_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
820 if (ast_istype(exprs[0], ast_entfield))
821 assignop = type_storep_instr[exprs[0]->expression.vtype];
823 assignop = type_store_instr[exprs[0]->expression.vtype];
824 out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]);
828 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
829 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
831 parseerror(parser, "invalid types used in expression: cannot add or subtract type %s and %s",
832 type_name[exprs[0]->expression.vtype],
833 type_name[exprs[1]->expression.vtype]);
836 if (ast_istype(exprs[0], ast_entfield))
837 assignop = type_storep_instr[exprs[0]->expression.vtype];
839 assignop = type_store_instr[exprs[0]->expression.vtype];
840 switch (exprs[0]->expression.vtype) {
842 out = (ast_expression*)ast_binstore_new(ctx, assignop,
843 (op->id == opid2('+','=') ? INSTR_ADD_F : INSTR_SUB_F),
847 out = (ast_expression*)ast_binstore_new(ctx, assignop,
848 (op->id == opid2('+','=') ? INSTR_ADD_V : INSTR_SUB_V),
852 parseerror(parser, "invalid types used in expression: cannot add or subtract type %s and %s",
853 type_name[exprs[0]->expression.vtype],
854 type_name[exprs[1]->expression.vtype]);
862 parseerror(parser, "failed to apply operand %s", op->op);
866 DEBUGSHUNTDO(printf("applied %s\n", op->op));
867 sy->out[sy->out_count++] = syexp(ctx, out);
871 static bool parser_close_call(parser_t *parser, shunt *sy)
873 /* was a function call */
881 fid = sy->ops[sy->ops_count].off;
883 /* out[fid] is the function
884 * everything above is parameters...
886 * 1 params = ast_expression
890 if (sy->out_count < 1 || sy->out_count <= fid) {
891 parseerror(parser, "internal error: function call needs function and parameter list...");
895 fun = sy->out[fid].out;
897 call = ast_call_new(sy->ops[sy->ops_count].ctx, fun);
899 parseerror(parser, "out of memory");
903 if (fid+1 == sy->out_count) {
906 } else if (fid+2 == sy->out_count) {
909 params = sy->out[sy->out_count].block;
913 if (!ast_call_params_add(call, sy->out[sy->out_count].out)) {
914 ast_delete(sy->out[sy->out_count].out);
915 parseerror(parser, "out of memory");
919 paramcount = params->exprs_count;
920 MEM_VECTOR_MOVE(params, exprs, call, params);
924 parseerror(parser, "invalid function call");
928 /* overwrite fid, the function, with a call */
929 sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
931 if (fun->expression.vtype != TYPE_FUNCTION) {
932 parseerror(parser, "not a function (%s)", type_name[fun->expression.vtype]);
936 if (!fun->expression.next) {
937 parseerror(parser, "could not determine function return type");
940 if (fun->expression.params_count != paramcount) {
942 fval = (ast_istype(fun, ast_value) ? ((ast_value*)fun) : NULL);
943 if (opts_standard == COMPILER_GMQCC)
946 parseerror(parser, "too few parameters for call to %s: expected %i, got %i",
947 fval->name, (int)fun->expression.params_count, paramcount);
949 parseerror(parser, "too few parameters for function call: expected %i, got %i",
950 (int)fun->expression.params_count, paramcount);
956 return !parsewarning(parser, WARN_TOO_FEW_PARAMETERS,
957 "too few parameters for call to %s: expected %i, got %i",
958 fval->name, (int)fun->expression.params_count, paramcount);
960 return !parsewarning(parser, WARN_TOO_FEW_PARAMETERS,
961 "too few parameters for function call: expected %i, got %i",
962 (int)fun->expression.params_count, paramcount);
970 static bool parser_close_paren(parser_t *parser, shunt *sy, bool functions_only)
972 if (!sy->ops_count) {
973 parseerror(parser, "unmatched closing paren");
976 if (sy->ops[sy->ops_count-1].paren == 1) {
977 parseerror(parser, "empty parenthesis expression");
980 while (sy->ops_count) {
981 if (sy->ops[sy->ops_count-1].paren == 'f') {
982 if (!parser_close_call(parser, sy))
986 if (sy->ops[sy->ops_count-1].paren == 1) {
988 return !functions_only;
990 if (!parser_sy_pop(parser, sy))
996 static void parser_reclassify_token(parser_t *parser)
999 for (i = 0; i < operator_count; ++i) {
1000 if (!strcmp(parser_tokval(parser), operators[i].op)) {
1001 parser->tok = TOKEN_OPERATOR;
1007 static ast_expression* parser_expression_leave(parser_t *parser, bool stopatcomma)
1009 ast_expression *expr = NULL;
1011 bool wantop = false;
1012 bool gotmemberof = false;
1014 /* count the parens because an if starts with one, so the
1015 * end of a condition is an unmatched closing paren
1019 MEM_VECTOR_INIT(&sy, out);
1020 MEM_VECTOR_INIT(&sy, ops);
1022 parser->lex->flags.noops = false;
1024 parser_reclassify_token(parser);
1029 gotmemberof = false;
1031 parser->memberof = 0;
1033 if (parser->tok == TOKEN_IDENT)
1035 ast_expression *var;
1037 parseerror(parser, "expected operator or end of statement");
1042 if (opts_standard == COMPILER_GMQCC)
1044 if (parser->memberof == TYPE_ENTITY) {
1045 /* still get vars first since there could be a fieldpointer */
1046 var = parser_find_var(parser, parser_tokval(parser));
1048 var = parser_find_field(parser, parser_tokval(parser));
1050 else if (parser->memberof == TYPE_VECTOR)
1052 parseerror(parser, "TODO: implement effective vector member access");
1055 else if (parser->memberof) {
1056 parseerror(parser, "namespace for member not found");
1060 var = parser_find_var(parser, parser_tokval(parser));
1062 var = parser_find_var(parser, parser_tokval(parser));
1064 var = parser_find_field(parser, parser_tokval(parser));
1067 parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
1070 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), var))) {
1071 parseerror(parser, "out of memory");
1074 DEBUGSHUNTDO(printf("push %s\n", parser_tokval(parser)));
1076 else if (parser->tok == TOKEN_FLOATCONST) {
1079 parseerror(parser, "expected operator or end of statement, got constant");
1083 val = parser_const_float(parser, (parser_token(parser)->constval.f));
1086 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1087 parseerror(parser, "out of memory");
1090 DEBUGSHUNTDO(printf("push %g\n", parser_token(parser)->constval.f));
1092 else if (parser->tok == TOKEN_INTCONST) {
1095 parseerror(parser, "expected operator or end of statement, got constant");
1099 val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
1102 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1103 parseerror(parser, "out of memory");
1106 DEBUGSHUNTDO(printf("push %i\n", parser_token(parser)->constval.i));
1108 else if (parser->tok == TOKEN_STRINGCONST) {
1111 parseerror(parser, "expected operator or end of statement, got constant");
1115 val = parser_const_string(parser, parser_tokval(parser));
1118 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1119 parseerror(parser, "out of memory");
1122 DEBUGSHUNTDO(printf("push string\n"));
1124 else if (parser->tok == TOKEN_VECTORCONST) {
1127 parseerror(parser, "expected operator or end of statement, got constant");
1131 val = parser_const_vector(parser, parser_token(parser)->constval.v);
1134 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1135 parseerror(parser, "out of memory");
1138 DEBUGSHUNTDO(printf("push '%g %g %g'\n",
1139 parser_token(parser)->constval.v.x,
1140 parser_token(parser)->constval.v.y,
1141 parser_token(parser)->constval.v.z));
1143 else if (parser->tok == '(') {
1144 parseerror(parser, "internal error: '(' should be classified as operator");
1147 else if (parser->tok == ')') {
1149 DEBUGSHUNTDO(printf("do[op] )\n"));
1153 /* we do expect an operator next */
1154 /* closing an opening paren */
1155 if (!parser_close_paren(parser, &sy, false))
1158 DEBUGSHUNTDO(printf("do[nop] )\n"));
1162 /* allowed for function calls */
1163 if (!parser_close_paren(parser, &sy, true))
1168 else if (parser->tok != TOKEN_OPERATOR) {
1170 parseerror(parser, "expected operator or end of statement");
1177 /* classify the operator */
1178 /* TODO: suffix operators */
1179 const oper_info *op;
1180 const oper_info *olast = NULL;
1182 for (o = 0; o < operator_count; ++o) {
1183 if ((!(operators[o].flags & OP_PREFIX) == wantop) &&
1184 !(operators[o].flags & OP_SUFFIX) && /* remove this */
1185 !strcmp(parser_tokval(parser), operators[o].op))
1190 if (o == operator_count) {
1191 /* no operator found... must be the end of the statement */
1194 /* found an operator */
1197 /* when declaring variables, a comma starts a new variable */
1198 if (op->id == opid1(',') && !parens && stopatcomma)
1201 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
1202 olast = &operators[sy.ops[sy.ops_count-1].etype-1];
1205 (op->prec < olast->prec) ||
1206 (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
1208 if (!parser_sy_pop(parser, &sy))
1210 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
1211 olast = &operators[sy.ops[sy.ops_count-1].etype-1];
1216 if (op->id == opid1('.') && opts_standard == COMPILER_GMQCC) {
1217 /* for gmqcc standard: open up the namespace of the previous type */
1218 ast_expression *prevex = sy.out[sy.out_count-1].out;
1220 parseerror(parser, "unexpected member operator");
1223 if (prevex->expression.vtype == TYPE_ENTITY)
1224 parser->memberof = TYPE_ENTITY;
1225 else if (prevex->expression.vtype == TYPE_VECTOR)
1226 parser->memberof = TYPE_VECTOR;
1228 parseerror(parser, "type error: type has no members");
1234 if (op->id == opid1('(')) {
1236 DEBUGSHUNTDO(printf("push (\n"));
1238 /* we expected an operator, this is the function-call operator */
1239 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 'f', sy.out_count-1))) {
1240 parseerror(parser, "out of memory");
1245 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1, 0))) {
1246 parseerror(parser, "out of memory");
1249 DEBUGSHUNTDO(printf("push (\n"));
1253 DEBUGSHUNTDO(printf("push operator %s\n", op->op));
1254 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
1259 if (!parser_next(parser)) {
1262 if (parser->tok == ';') {
1267 while (sy.ops_count) {
1268 if (!parser_sy_pop(parser, &sy))
1272 parser->lex->flags.noops = true;
1273 if (!sy.out_count) {
1274 parseerror(parser, "empty expression");
1277 expr = sy.out[0].out;
1278 MEM_VECTOR_CLEAR(&sy, out);
1279 MEM_VECTOR_CLEAR(&sy, ops);
1280 DEBUGSHUNTDO(printf("shunt done\n"));
1284 parser->lex->flags.noops = true;
1285 MEM_VECTOR_CLEAR(&sy, out);
1286 MEM_VECTOR_CLEAR(&sy, ops);
1290 static ast_expression* parser_expression(parser_t *parser, bool stopatcomma)
1292 ast_expression *e = parser_expression_leave(parser, stopatcomma);
1295 if (!parser_next(parser)) {
1302 static bool parser_parse_if(parser_t *parser, ast_block *block, ast_expression **out)
1305 ast_expression *cond, *ontrue, *onfalse = NULL;
1307 lex_ctx ctx = parser_ctx(parser);
1309 /* skip the 'if' and check for opening paren */
1310 if (!parser_next(parser) || parser->tok != '(') {
1311 parseerror(parser, "expected 'if' condition in parenthesis");
1314 /* parse into the expression */
1315 if (!parser_next(parser)) {
1316 parseerror(parser, "expected 'if' condition after opening paren");
1319 /* parse the condition */
1320 cond = parser_expression_leave(parser, false);
1324 if (parser->tok != ')') {
1325 parseerror(parser, "expected closing paren after 'if' condition");
1329 /* parse into the 'then' branch */
1330 if (!parser_next(parser)) {
1331 parseerror(parser, "expected statement for on-true branch of 'if'");
1335 ontrue = parser_parse_statement_or_block(parser);
1340 /* check for an else */
1341 if (!strcmp(parser_tokval(parser), "else")) {
1342 /* parse into the 'else' branch */
1343 if (!parser_next(parser)) {
1344 parseerror(parser, "expected on-false branch after 'else'");
1349 onfalse = parser_parse_statement_or_block(parser);
1357 ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
1358 *out = (ast_expression*)ifthen;
1362 static bool parser_parse_while(parser_t *parser, ast_block *block, ast_expression **out)
1365 ast_expression *cond, *ontrue;
1367 lex_ctx ctx = parser_ctx(parser);
1369 /* skip the 'while' and check for opening paren */
1370 if (!parser_next(parser) || parser->tok != '(') {
1371 parseerror(parser, "expected 'while' condition in parenthesis");
1374 /* parse into the expression */
1375 if (!parser_next(parser)) {
1376 parseerror(parser, "expected 'while' condition after opening paren");
1379 /* parse the condition */
1380 cond = parser_expression_leave(parser, false);
1384 if (parser->tok != ')') {
1385 parseerror(parser, "expected closing paren after 'while' condition");
1389 /* parse into the 'then' branch */
1390 if (!parser_next(parser)) {
1391 parseerror(parser, "expected while-loop body");
1395 ontrue = parser_parse_statement_or_block(parser);
1401 aloop = ast_loop_new(ctx, NULL, cond, NULL, NULL, ontrue);
1402 *out = (ast_expression*)aloop;
1406 static bool parser_parse_dowhile(parser_t *parser, ast_block *block, ast_expression **out)
1409 ast_expression *cond, *ontrue;
1411 lex_ctx ctx = parser_ctx(parser);
1413 /* skip the 'do' and get the body */
1414 if (!parser_next(parser)) {
1415 parseerror(parser, "expected loop body");
1418 ontrue = parser_parse_statement_or_block(parser);
1422 /* expect the "while" */
1423 if (parser->tok != TOKEN_KEYWORD ||
1424 strcmp(parser_tokval(parser), "while"))
1426 parseerror(parser, "expected 'while' and condition");
1431 /* skip the 'while' and check for opening paren */
1432 if (!parser_next(parser) || parser->tok != '(') {
1433 parseerror(parser, "expected 'while' condition in parenthesis");
1437 /* parse into the expression */
1438 if (!parser_next(parser)) {
1439 parseerror(parser, "expected 'while' condition after opening paren");
1443 /* parse the condition */
1444 cond = parser_expression_leave(parser, false);
1448 if (parser->tok != ')') {
1449 parseerror(parser, "expected closing paren after 'while' condition");
1455 if (!parser_next(parser) || parser->tok != ';') {
1456 parseerror(parser, "expected semicolon after condition");
1462 if (!parser_next(parser)) {
1463 parseerror(parser, "parse error");
1469 aloop = ast_loop_new(ctx, NULL, NULL, cond, NULL, ontrue);
1470 *out = (ast_expression*)aloop;
1474 static bool parser_parse_for(parser_t *parser, ast_block *block, ast_expression **out)
1477 ast_expression *initexpr, *cond, *increment, *ontrue;
1478 size_t oldblocklocal;
1480 lex_ctx ctx = parser_ctx(parser);
1482 oldblocklocal = parser->blocklocal;
1483 parser->blocklocal = parser->locals_count;
1490 /* skip the 'while' and check for opening paren */
1491 if (!parser_next(parser) || parser->tok != '(') {
1492 parseerror(parser, "expected 'for' expressions in parenthesis");
1495 /* parse into the expression */
1496 if (!parser_next(parser)) {
1497 parseerror(parser, "expected 'for' initializer after opening paren");
1501 if (parser->tok == TOKEN_TYPENAME) {
1502 if (opts_standard != COMPILER_GMQCC) {
1503 if (parsewarning(parser, WARN_EXTENSIONS,
1504 "current standard does not allow variable declarations in for-loop initializers"))
1508 parseerror(parser, "TODO: assignment of new variables to be non-const");
1510 if (!parser_variable(parser, block))
1513 else if (parser->tok != ';')
1515 initexpr = parser_expression_leave(parser, false);
1520 /* move on to condition */
1521 if (parser->tok != ';') {
1522 parseerror(parser, "expected semicolon after for-loop initializer");
1525 if (!parser_next(parser)) {
1526 parseerror(parser, "expected for-loop condition");
1530 /* parse the condition */
1531 if (parser->tok != ';') {
1532 cond = parser_expression_leave(parser, false);
1537 /* move on to incrementor */
1538 if (parser->tok != ';') {
1539 parseerror(parser, "expected semicolon after for-loop initializer");
1542 if (!parser_next(parser)) {
1543 parseerror(parser, "expected for-loop condition");
1547 /* parse the incrementor */
1548 if (parser->tok != ')') {
1549 increment = parser_expression_leave(parser, false);
1555 if (parser->tok != ')') {
1556 parseerror(parser, "expected closing paren after 'for-loop' incrementor");
1559 /* parse into the 'then' branch */
1560 if (!parser_next(parser)) {
1561 parseerror(parser, "expected for-loop body");
1564 ontrue = parser_parse_statement_or_block(parser);
1569 aloop = ast_loop_new(ctx, initexpr, cond, NULL, increment, ontrue);
1570 *out = (ast_expression*)aloop;
1572 while (parser->locals_count > parser->blocklocal)
1573 parser_pop_local(parser);
1574 parser->blocklocal = oldblocklocal;
1577 if (initexpr) ast_delete(initexpr);
1578 if (cond) ast_delete(cond);
1579 if (increment) ast_delete(increment);
1580 while (parser->locals_count > parser->blocklocal)
1581 parser_pop_local(parser);
1582 parser->blocklocal = oldblocklocal;
1586 static bool parser_parse_statement(parser_t *parser, ast_block *block, ast_expression **out)
1588 if (parser->tok == TOKEN_TYPENAME)
1590 /* local variable */
1592 parseerror(parser, "cannot declare a variable from here");
1595 if (opts_standard == COMPILER_QCC) {
1596 if (parsewarning(parser, WARN_EXTENSIONS, "missing 'local' keyword when declaring a local variable"))
1599 if (!parser_variable(parser, block))
1604 else if (parser->tok == TOKEN_KEYWORD)
1606 if (!strcmp(parser_tokval(parser), "local"))
1609 parseerror(parser, "cannot declare a local variable here");
1612 if (!parser_next(parser)) {
1613 parseerror(parser, "expected variable declaration");
1616 if (!parser_variable(parser, block))
1621 else if (!strcmp(parser_tokval(parser), "return"))
1623 ast_expression *exp = NULL;
1624 ast_return *ret = NULL;
1625 ast_value *expected = parser->function->vtype;
1627 if (!parser_next(parser)) {
1628 parseerror(parser, "expected return expression");
1632 if (parser->tok != ';') {
1633 exp = parser_expression(parser, false);
1637 if (exp->expression.vtype != expected->expression.next->expression.vtype) {
1638 parseerror(parser, "return with invalid expression");
1641 ret = ast_return_new(exp->expression.node.context, exp);
1647 if (!parser_next(parser))
1648 parseerror(parser, "parse error");
1649 if (expected->expression.next->expression.vtype != TYPE_VOID) {
1650 if (opts_standard != COMPILER_GMQCC)
1651 (void)!parsewarning(parser, WARN_MISSING_RETURN_VALUES, "return without value");
1653 parseerror(parser, "return without value");
1655 ret = ast_return_new(parser_ctx(parser), NULL);
1657 *out = (ast_expression*)ret;
1660 else if (!strcmp(parser_tokval(parser), "if"))
1662 return parser_parse_if(parser, block, out);
1664 else if (!strcmp(parser_tokval(parser), "while"))
1666 return parser_parse_while(parser, block, out);
1668 else if (!strcmp(parser_tokval(parser), "do"))
1670 return parser_parse_dowhile(parser, block, out);
1672 else if (!strcmp(parser_tokval(parser), "for"))
1674 if (opts_standard == COMPILER_QCC) {
1675 if (parsewarning(parser, WARN_EXTENSIONS, "for loops are not recognized in the original Quake C standard, to enable try an alternate standard --std=?"))
1678 return parser_parse_for(parser, block, out);
1680 parseerror(parser, "Unexpected keyword");
1683 else if (parser->tok == '{')
1686 inner = parser_parse_block(parser);
1689 *out = (ast_expression*)inner;
1694 ast_expression *exp = parser_expression(parser, false);
1702 static void parser_pop_local(parser_t *parser)
1704 parser->locals_count--;
1705 mem_d(parser->locals[parser->locals_count].name);
1708 static bool parser_parse_block_into(parser_t *parser, ast_block *block)
1710 size_t oldblocklocal;
1712 oldblocklocal = parser->blocklocal;
1713 parser->blocklocal = parser->locals_count;
1715 if (!parser_next(parser)) { /* skip the '{' */
1716 parseerror(parser, "expected function body");
1720 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1722 ast_expression *expr;
1723 if (parser->tok == '}')
1726 if (!parser_parse_statement(parser, block, &expr)) {
1727 parseerror(parser, "parse error");
1733 if (!ast_block_exprs_add(block, expr)) {
1740 if (parser->tok != '}') {
1743 (void)parser_next(parser);
1747 while (parser->locals_count > parser->blocklocal)
1748 parser_pop_local(parser);
1749 parser->blocklocal = oldblocklocal;
1753 static ast_block* parser_parse_block(parser_t *parser)
1756 block = ast_block_new(parser_ctx(parser));
1759 if (!parser_parse_block_into(parser, block)) {
1760 ast_block_delete(block);
1766 static ast_expression* parser_parse_statement_or_block(parser_t *parser)
1768 ast_expression *expr = NULL;
1769 if (parser->tok == '{')
1770 return (ast_expression*)parser_parse_block(parser);
1771 if (!parser_parse_statement(parser, NULL, &expr))
1776 static bool create_vector_members(parser_t *parser, ast_value *var,
1777 varentry_t *vx, varentry_t *vy, varentry_t *vz)
1779 size_t len = strlen(var->name);
1780 vx->var = (ast_expression*)ast_member_new(ast_ctx(var), (ast_expression*)var, 0);
1782 parseerror(parser, "failed to create vector members (out of memory)");
1786 vy->var = (ast_expression*)ast_member_new(ast_ctx(var), (ast_expression*)var, 1);
1788 ast_delete(vx->var);
1789 parseerror(parser, "failed to create vector members (out of memory)");
1793 vz->var = (ast_expression*)ast_member_new(ast_ctx(var), (ast_expression*)var, 2);
1795 ast_delete(vy->var);
1796 ast_delete(vx->var);
1797 parseerror(parser, "failed to create vector members (out of memory)");
1801 if ( !(vx->name = (char*)mem_a(len+3)) ) {
1802 ast_delete(vz->var);
1803 ast_delete(vy->var);
1804 ast_delete(vx->var);
1805 parseerror(parser, "failed to create vector members (out of memory)");
1808 if ( !(vy->name = (char*)mem_a(len+3)) ) {
1810 ast_delete(vz->var);
1811 ast_delete(vy->var);
1812 ast_delete(vx->var);
1813 parseerror(parser, "failed to create vector members (out of memory)");
1816 if ( !(vz->name = (char*)mem_a(len+3)) ) {
1819 ast_delete(vz->var);
1820 ast_delete(vy->var);
1821 ast_delete(vx->var);
1822 parseerror(parser, "failed to create vector members (out of memory)");
1826 memcpy(vx->name, var->name, len);
1827 memcpy(vy->name, var->name, len);
1828 memcpy(vz->name, var->name, len);
1829 vx->name[len] = vy->name[len] = vz->name[len] = '_';
1830 vx->name[len+1] = 'x';
1831 vy->name[len+1] = 'y';
1832 vz->name[len+1] = 'z';
1833 vx->name[len+2] = vy->name[len+2] = vz->name[len+2] = 0;
1837 static bool parser_variable(parser_t *parser, ast_block *localblock)
1839 bool isfunc = false;
1840 ast_function *func = NULL;
1844 ast_expression *olddecl;
1849 int basetype = parser_token(parser)->constval.t;
1855 if (!parser_next(parser)) { /* skip basetype or comma */
1856 parseerror(parser, "expected variable declaration");
1864 ctx = parser_ctx(parser);
1865 var = parser_parse_type(parser, basetype, &isfunc);
1870 if (parser->tok != TOKEN_IDENT) {
1871 parseerror(parser, "expected variable name\n");
1876 if (!localblock && (olddecl = parser_find_global(parser, parser_tokval(parser)))) {
1877 ast_value_delete(var);
1878 parseerror(parser, "global `%s` already declared here: %s:%i",
1879 parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1884 olddecl = parser_find_local(parser, parser_tokval(parser), parser->blocklocal, &isparam);
1885 if (opts_standard == COMPILER_GMQCC)
1890 ast_value_delete(var);
1891 parseerror(parser, "local `%s` already declared here: %s:%i",
1892 parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1897 if( (!isparam && olddecl) || (olddecl = parser_find_local(parser, parser_tokval(parser), 0, &isparam)) )
1899 if (parsewarning(parser, WARN_LOCAL_SHADOWS,
1900 "local `%s` is shadowing a parameter", parser_tokval(parser)))
1902 ast_value_delete(var);
1903 parseerror(parser, "local `%s` already declared here: %s:%i",
1904 parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1913 ast_value_delete(var);
1915 parsewarning(parser, WARN_LOCAL_SHADOWS,
1916 "a parameter is shadowing local `%s`", parser_tokval(parser)))
1922 parseerror(parser, "local `%s` already declared here: %s:%i",
1923 parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1932 if (!ast_value_set_name(var, parser_tokval(parser))) {
1933 parseerror(parser, "failed to set variable name\n");
1934 ast_value_delete(var);
1939 /* a function was defined */
1941 ast_value *proto = NULL;
1945 olddecl = parser_find_global(parser, parser_tokval(parser));
1947 olddecl = parser_find_local(parser, parser_tokval(parser), parser->blocklocal, &dummy);
1950 /* we had a prototype */
1951 if (!ast_istype(olddecl, ast_value)) {
1952 /* theoretically not possible you think?
1958 parseerror(parser, "cannot declare a function with the same name as a vector's member: %s",
1959 parser_tokval(parser));
1960 ast_value_delete(var);
1964 proto = (ast_value*)olddecl;
1967 /* turn var into a value of TYPE_FUNCTION, with the old var
1970 fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1971 func = ast_function_new(ctx, var->name, fval);
1972 if (!fval || !func) {
1973 ast_value_delete(var);
1974 if (fval) ast_value_delete(fval);
1975 if (func) ast_function_delete(func);
1979 fval->expression.next = (ast_expression*)var;
1980 MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1982 /* we compare the type late here, but it's easier than
1983 * messing with the parameter-vector etc. earlier
1987 if (!ast_compare_type((ast_expression*)proto, (ast_expression*)fval)) {
1988 parseerror(parser, "conflicting types for `%s`, previous declaration was here: %s:%i",
1990 ast_ctx(proto).file, ast_ctx(proto).line);
1991 ast_function_delete(func);
1992 ast_value_delete(fval);
1995 /* copy over the parameter names */
1996 for (param = 0; param < fval->expression.params_count; ++param)
1997 ast_value_set_name(proto->expression.params[param], fval->expression.params[param]->name);
1999 /* now ditch the rest of the new data */
2000 ast_function_delete(func);
2001 ast_value_delete(fval);
2003 func = proto->constval.vfunc;
2008 if (!parser_t_functions_add(parser, func)) {
2009 ast_function_delete(func);
2010 ast_value_delete(fval);
2019 varent.name = util_strdup(var->name);
2020 varent.var = (ast_expression*)var;
2021 if (var->expression.vtype == TYPE_VECTOR)
2023 varentry_t vx, vy, vz;
2024 if (!create_vector_members(parser, var, &vx, &vy, &vz)) {
2030 (void)!parser_t_globals_add(parser, varent);
2031 (void)!parser_t_globals_add(parser, vx);
2032 (void)!parser_t_globals_add(parser, vy);
2033 (void)!parser_t_globals_add(parser, vz);
2035 (void)!parser_t_locals_add(parser, varent);
2036 (void)!parser_t_locals_add(parser, vx);
2037 (void)!parser_t_locals_add(parser, vy);
2038 (void)!parser_t_locals_add(parser, vz);
2039 if (!ast_block_locals_add(localblock, var) ||
2040 !ast_block_collect(localblock, vx.var) ||
2041 !ast_block_collect(localblock, vy.var) ||
2042 !ast_block_collect(localblock, vz.var))
2044 parser_pop_local(parser);
2045 parser_pop_local(parser);
2046 parser_pop_local(parser);
2047 parser_pop_local(parser);
2048 ast_value_delete(var);
2055 if ( (!localblock && !parser_t_globals_add(parser, varent)) ||
2056 ( localblock && !parser_t_locals_add(parser, varent)) )
2058 ast_value_delete(var);
2061 if (localblock && !ast_block_locals_add(localblock, var))
2063 parser_pop_local(parser);
2064 ast_value_delete(var);
2071 if (!parser_next(parser)) {
2072 ast_value_delete(var);
2076 if (parser->tok == ';') {
2077 if (!parser_next(parser))
2078 return parser->tok == TOKEN_EOF;
2082 if (parser->tok == ',') {
2087 if (parser->tok != '=') {
2088 parseerror(parser, "expected '=' or ';'");
2092 if (!parser_next(parser))
2095 if (parser->tok == '#') {
2097 parseerror(parser, "cannot declare builtins within functions");
2100 if (!isfunc || !func) {
2101 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
2104 if (!parser_next(parser)) {
2105 parseerror(parser, "expected builtin number");
2108 if (parser->tok != TOKEN_INTCONST) {
2109 parseerror(parser, "builtin number must be an integer constant");
2112 if (parser_token(parser)->constval.i <= 0) {
2113 parseerror(parser, "builtin number must be positive integer greater than zero");
2117 func->builtin = -parser_token(parser)->constval.i;
2119 if (!parser_next(parser))
2121 } else if (parser->tok == '{') {
2125 ast_function *old = parser->function;
2128 parseerror(parser, "cannot declare functions within functions");
2132 block = ast_block_new(parser_ctx(parser));
2134 parseerror(parser, "failed to allocate block");
2138 for (parami = 0; parami < var->expression.params_count; ++parami) {
2139 ast_value *param = var->expression.params[parami];
2140 varentry_t vx, vy, vz;
2142 if (param->expression.vtype != TYPE_VECTOR &&
2143 (param->expression.vtype != TYPE_FIELD ||
2144 param->expression.next->expression.vtype != TYPE_VECTOR))
2149 if (!create_vector_members(parser, param, &vx, &vy, &vz)) {
2150 ast_block_delete(block);
2154 (void)!parser_t_locals_add(parser, vx);
2155 (void)!parser_t_locals_add(parser, vy);
2156 (void)!parser_t_locals_add(parser, vz);
2157 if (!ast_block_collect(block, vx.var) ||
2158 !ast_block_collect(block, vy.var) ||
2159 !ast_block_collect(block, vz.var) )
2161 parser_pop_local(parser);
2162 parser_pop_local(parser);
2163 parser_pop_local(parser);
2164 ast_block_delete(block);
2169 parser->function = func;
2170 if (!parser_parse_block_into(parser, block)) {
2171 ast_block_delete(block);
2172 parser->function = old;
2175 parser->function = old;
2180 if (!ast_function_blocks_add(func, block)) {
2181 ast_block_delete(block);
2185 if (parser->tok == ';')
2186 return parser_next(parser) || parser->tok == TOKEN_EOF;
2187 else if (opts_standard == COMPILER_QCC)
2188 parseerror(parser, "missing semicolon after function body (mandatory with -std=qcc)");
2191 ast_expression *cexp;
2194 cexp = parser_expression_leave(parser, true);
2195 cval = (ast_value*)cexp;
2196 if (!ast_istype(cval, ast_value) || !cval->isconst)
2197 parseerror(parser, "cannot initialize a global constant variable with a non-constant expression");
2200 var->isconst = true;
2201 memcpy(&var->constval, &cval->constval, sizeof(var->constval));
2202 memset(&cval->constval, 0, sizeof(cval->constval));
2207 if (parser->tok == ',') {
2212 if (parser->tok != ';') {
2213 parseerror(parser, "missing semicolon");
2217 (void)parser_next(parser);
2223 static bool parser_do(parser_t *parser)
2225 if (parser->tok == TOKEN_TYPENAME)
2227 return parser_variable(parser, NULL);
2229 else if (parser->tok == TOKEN_KEYWORD)
2231 /* handle 'var' and 'const' */
2234 else if (parser->tok == '.')
2239 ast_expression *oldex;
2240 bool isfunc = false;
2242 lex_ctx ctx = parser_ctx(parser);
2245 /* entity-member declaration */
2246 if (!parser_next(parser) || parser->tok != TOKEN_TYPENAME) {
2247 parseerror(parser, "expected member variable definition");
2251 /* remember the base/return type */
2252 basetype = parser_token(parser)->constval.t;
2254 /* parse into the declaration */
2255 if (!parser_next(parser)) {
2256 parseerror(parser, "expected field def");
2260 /* parse the field type fully */
2261 typevar = var = parser_parse_type(parser, basetype, &isfunc);
2266 var = ast_value_copy(typevar);
2267 /* now the field name */
2268 if (parser->tok != TOKEN_IDENT) {
2269 parseerror(parser, "expected field name");
2274 /* check for an existing field
2275 * in original qc we also have to check for an existing
2276 * global named like the field
2278 if (opts_standard == COMPILER_QCC) {
2279 if (parser_find_global(parser, parser_tokval(parser))) {
2280 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
2288 fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
2290 ast_value_delete(var);
2293 fval->expression.next = (ast_expression*)var;
2294 MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
2298 /* turn it into a field */
2299 fld = ast_value_new(ctx, parser_tokval(parser), TYPE_FIELD);
2300 fld->expression.next = (ast_expression*)var;
2302 if ( (oldex = parser_find_field(parser, parser_tokval(parser)))) {
2303 if (ast_istype(oldex, ast_member)) {
2304 parseerror(parser, "cannot declare a field with the same name as a vector component, component %s has been declared here: %s:%i",
2305 parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line);
2309 if (!ast_istype(oldex, ast_value)) {
2310 /* not possible / sanity check */
2311 parseerror(parser, "internal error: %s is not an ast_value", parser_tokval(parser));
2316 if (!ast_compare_type(oldex, (ast_expression*)fld)) {
2317 parseerror(parser, "field %s has previously been declared with a different type here: %s:%i",
2318 parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line);
2322 if (parsewarning(parser, WARN_FIELD_REDECLARED, "field %s has already been declared here: %s:%i",
2323 parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line))
2334 varent.var = (ast_expression*)fld;
2335 varent.name = util_strdup(fld->name);
2336 (void)!parser_t_fields_add(parser, varent);
2338 if (var->expression.vtype == TYPE_VECTOR)
2340 /* create _x, _y and _z fields as well */
2341 varentry_t vx, vy, vz;
2342 if (!create_vector_members(parser, fld, &vx, &vy, &vz)) {
2346 (void)!parser_t_fields_add(parser, vx);
2347 (void)!parser_t_fields_add(parser, vy);
2348 (void)!parser_t_fields_add(parser, vz);
2352 if (!parser_next(parser)) {
2353 parseerror(parser, "expected semicolon or another field name");
2356 if (parser->tok == ';')
2358 if (parser->tok != ',' || !parser_next(parser)) {
2359 parseerror(parser, "expected semicolon or another field name");
2363 ast_delete(typevar);
2365 /* skip the semicolon */
2366 if (!parser_next(parser))
2367 return parser->tok == TOKEN_EOF;
2371 else if (parser->tok == '$')
2373 if (!parser_next(parser)) {
2374 parseerror(parser, "parse error");
2380 parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
2386 static parser_t *parser;
2390 parser = (parser_t*)mem_a(sizeof(parser_t));
2394 memset(parser, 0, sizeof(*parser));
2398 bool parser_compile(const char *filename)
2400 parser->lex = lex_open(filename);
2402 printf("failed to open file \"%s\"\n", filename);
2406 /* initial lexer/parser state */
2407 parser->lex->flags.noops = true;
2409 if (parser_next(parser))
2411 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
2413 if (!parser_do(parser)) {
2414 if (parser->tok == TOKEN_EOF)
2415 parseerror(parser, "unexpected eof");
2416 else if (!parser->errors)
2417 parseerror(parser, "parse error\n");
2418 lex_close(parser->lex);
2425 lex_close(parser->lex);
2428 return !parser->errors;
2431 void parser_cleanup()
2434 for (i = 0; i < parser->functions_count; ++i) {
2435 ast_delete(parser->functions[i]);
2437 for (i = 0; i < parser->imm_vector_count; ++i) {
2438 ast_delete(parser->imm_vector[i]);
2440 for (i = 0; i < parser->imm_string_count; ++i) {
2441 ast_delete(parser->imm_string[i]);
2443 for (i = 0; i < parser->imm_float_count; ++i) {
2444 ast_delete(parser->imm_float[i]);
2446 for (i = 0; i < parser->fields_count; ++i) {
2447 ast_delete(parser->fields[i].var);
2448 mem_d(parser->fields[i].name);
2450 for (i = 0; i < parser->globals_count; ++i) {
2451 ast_delete(parser->globals[i].var);
2452 mem_d(parser->globals[i].name);
2454 MEM_VECTOR_CLEAR(parser, functions);
2455 MEM_VECTOR_CLEAR(parser, imm_vector);
2456 MEM_VECTOR_CLEAR(parser, imm_string);
2457 MEM_VECTOR_CLEAR(parser, imm_float);
2458 MEM_VECTOR_CLEAR(parser, globals);
2459 MEM_VECTOR_CLEAR(parser, fields);
2460 MEM_VECTOR_CLEAR(parser, locals);
2465 bool parser_finish(const char *output)
2470 if (!parser->errors)
2472 ir = ir_builder_new("gmqcc_out");
2474 printf("failed to allocate builder\n");
2478 for (i = 0; i < parser->imm_float_count; ++i) {
2479 if (!ast_global_codegen(parser->imm_float[i], ir)) {
2480 printf("failed to generate global %s\n", parser->imm_float[i]->name);
2481 ir_builder_delete(ir);
2485 for (i = 0; i < parser->imm_string_count; ++i) {
2486 if (!ast_global_codegen(parser->imm_string[i], ir)) {
2487 printf("failed to generate global %s\n", parser->imm_string[i]->name);
2488 ir_builder_delete(ir);
2492 for (i = 0; i < parser->imm_vector_count; ++i) {
2493 if (!ast_global_codegen(parser->imm_vector[i], ir)) {
2494 printf("failed to generate global %s\n", parser->imm_vector[i]->name);
2495 ir_builder_delete(ir);
2499 for (i = 0; i < parser->fields_count; ++i) {
2502 if (!ast_istype(parser->fields[i].var, ast_value))
2504 field = (ast_value*)parser->fields[i].var;
2505 isconst = field->isconst;
2506 field->isconst = false;
2507 if (!ast_global_codegen((ast_value*)field, ir)) {
2508 printf("failed to generate field %s\n", field->name);
2509 ir_builder_delete(ir);
2514 ast_expression *subtype;
2515 field->isconst = true;
2516 subtype = field->expression.next;
2517 ifld = ir_builder_create_field(ir, field->name, subtype->expression.vtype);
2518 if (subtype->expression.vtype == TYPE_FIELD)
2519 ifld->fieldtype = subtype->expression.next->expression.vtype;
2520 else if (subtype->expression.vtype == TYPE_FUNCTION)
2521 ifld->outtype = subtype->expression.next->expression.vtype;
2522 (void)!ir_value_set_field(field->ir_v, ifld);
2525 for (i = 0; i < parser->globals_count; ++i) {
2526 if (!ast_istype(parser->globals[i].var, ast_value))
2528 if (!ast_global_codegen((ast_value*)(parser->globals[i].var), ir)) {
2529 printf("failed to generate global %s\n", parser->globals[i].name);
2530 ir_builder_delete(ir);
2534 for (i = 0; i < parser->functions_count; ++i) {
2535 if (!ast_function_codegen(parser->functions[i], ir)) {
2536 printf("failed to generate function %s\n", parser->functions[i]->name);
2537 ir_builder_delete(ir);
2540 if (!ir_function_finalize(parser->functions[i]->ir_func)) {
2541 printf("failed to finalize function %s\n", parser->functions[i]->name);
2542 ir_builder_delete(ir);
2548 ir_builder_dump(ir, printf);
2550 if (!ir_builder_generate(ir, output)) {
2551 printf("*** failed to generate output file\n");
2552 ir_builder_delete(ir);
2556 ir_builder_delete(ir);
2560 printf("*** there were compile errors\n");