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_function *function;
24 MEM_VECTOR_MAKE(varentry_t, locals);
29 /* TYPE_FIELD -> parser_find_fields is used instead of find_var
30 * TODO: TYPE_VECTOR -> x, y and z are accepted in the gmqcc standard
31 * anything else: type error
36 MEM_VEC_FUNCTIONS(parser_t, varentry_t, globals)
37 MEM_VEC_FUNCTIONS(parser_t, varentry_t, fields)
38 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_float)
39 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_string)
40 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_vector)
41 MEM_VEC_FUNCTIONS(parser_t, varentry_t, locals)
42 MEM_VEC_FUNCTIONS(parser_t, ast_function*, functions)
44 void parseerror(parser_t *parser, const char *fmt, ...)
51 printf("error %s:%lu: ", parser->lex->tok->ctx.file, (unsigned long)parser->lex->tok->ctx.line);
62 bool parser_next(parser_t *parser)
64 /* lex_do kills the previous token */
65 parser->tok = lex_do(parser->lex);
66 if (parser->tok == TOKEN_EOF || parser->tok >= TOKEN_ERROR)
71 /* lift a token out of the parser so it's not destroyed by parser_next */
72 token *parser_lift(parser_t *parser)
74 token *tok = parser->lex->tok;
75 parser->lex->tok = NULL;
79 #define parser_tokval(p) (p->lex->tok->value)
80 #define parser_token(p) (p->lex->tok)
81 #define parser_ctx(p) (p->lex->tok->ctx)
83 ast_value* parser_const_float(parser_t *parser, double d)
87 for (i = 0; i < parser->imm_float_count; ++i) {
88 if (parser->imm_float[i]->constval.vfloat == d)
89 return parser->imm_float[i];
91 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_FLOAT);
93 out->constval.vfloat = d;
94 if (!parser_t_imm_float_add(parser, out)) {
95 ast_value_delete(out);
101 ast_value* parser_const_string(parser_t *parser, const char *str)
105 for (i = 0; i < parser->imm_string_count; ++i) {
106 if (!strcmp(parser->imm_string[i]->constval.vstring, str))
107 return parser->imm_string[i];
109 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_STRING);
111 out->constval.vstring = util_strdup(str);
112 if (!parser_t_imm_string_add(parser, out)) {
113 ast_value_delete(out);
119 ast_value* parser_const_vector(parser_t *parser, vector v)
123 for (i = 0; i < parser->imm_vector_count; ++i) {
124 if (!memcmp(&parser->imm_vector[i]->constval.vvec, &v, sizeof(v)))
125 return parser->imm_vector[i];
127 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_VECTOR);
129 out->constval.vvec = v;
130 if (!parser_t_imm_vector_add(parser, out)) {
131 ast_value_delete(out);
137 ast_expression* parser_find_field(parser_t *parser, const char *name)
140 for (i = 0; i < parser->fields_count; ++i) {
141 if (!strcmp(parser->fields[i].name, name))
142 return parser->fields[i].var;
147 ast_expression* parser_find_global(parser_t *parser, const char *name)
150 for (i = 0; i < parser->globals_count; ++i) {
151 if (!strcmp(parser->globals[i].name, name))
152 return parser->globals[i].var;
157 ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto)
161 for (i = parser->locals_count; i > upto;) {
163 if (!strcmp(parser->locals[i].name, name))
164 return parser->locals[i].var;
166 fun = parser->function->vtype;
167 for (i = 0; i < fun->expression.params_count; ++i) {
168 if (!strcmp(fun->expression.params[i]->name, name))
169 return (ast_expression*)(fun->expression.params[i]);
174 ast_expression* parser_find_var(parser_t *parser, const char *name)
177 v = parser_find_local(parser, name, 0);
178 if (!v) v = parser_find_global(parser, name);
183 MEM_VECTOR_MAKE(ast_value*, p);
185 MEM_VEC_FUNCTIONS(paramlist_t, ast_value*, p)
187 static ast_value *parser_parse_type(parser_t *parser, int basetype, bool *isfunc)
191 lex_ctx ctx = parser_ctx(parser);
192 int vtype = basetype;
196 MEM_VECTOR_INIT(¶ms, p);
200 if (parser->tok == '(') {
206 if (!parser_next(parser))
209 if (parser->tok == ')')
212 temptype = parser_token(parser)->constval.t;
213 if (!parser_next(parser))
216 param = parser_parse_type(parser, temptype, &dummy);
222 if (parser->tok == TOKEN_IDENT) {
223 /* named parameter */
224 if (!ast_value_set_name(param, parser_tokval(parser)))
226 if (!parser_next(parser))
230 if (!paramlist_t_p_add(¶ms, param)) {
231 parseerror(parser, "Out of memory while parsing typename");
235 if (parser->tok == ',')
237 if (parser->tok == ')')
239 parseerror(parser, "Unexpected token");
242 if (!parser_next(parser))
246 var = ast_value_new(ctx, "<unnamed>", vtype);
249 MEM_VECTOR_MOVE(¶ms, p, &var->expression, params);
252 for (i = 0; i < params.p_count; ++i)
253 ast_value_delete(params.p[i]);
254 MEM_VECTOR_CLEAR(¶ms, p);
260 size_t etype; /* 0 = expression, others are operators */
264 ast_block *block; /* for commas and function calls */
269 MEM_VECTOR_MAKE(sy_elem, out);
270 MEM_VECTOR_MAKE(sy_elem, ops);
272 MEM_VEC_FUNCTIONS(shunt, sy_elem, out)
273 MEM_VEC_FUNCTIONS(shunt, sy_elem, ops)
275 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
285 static sy_elem syblock(lex_ctx ctx, ast_block *v) {
288 e.out = (ast_expression*)v;
295 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
297 e.etype = 1 + (op - operators);
305 static sy_elem syparen(lex_ctx ctx, int p, size_t off) {
317 # define DEBUGSHUNTDO(x) x
319 # define DEBUGSHUNTDO(x)
322 static bool parser_sy_pop(parser_t *parser, shunt *sy)
326 ast_expression *out = NULL;
327 ast_expression *exprs[3];
328 ast_block *blocks[3];
330 qcint generated_op = 0;
332 if (!sy->ops_count) {
333 parseerror(parser, "internal error: missing operator");
337 if (sy->ops[sy->ops_count-1].paren) {
338 parseerror(parser, "unmatched parenthesis");
342 op = &operators[sy->ops[sy->ops_count-1].etype - 1];
343 ctx = sy->ops[sy->ops_count-1].ctx;
345 DEBUGSHUNTDO(printf("apply %s\n", op->op));
347 if (sy->out_count < op->operands) {
348 parseerror(parser, "internal error: not enough operands: %i", sy->out_count);
354 sy->out_count -= op->operands;
355 for (i = 0; i < op->operands; ++i) {
356 exprs[i] = sy->out[sy->out_count+i].out;
357 blocks[i] = sy->out[sy->out_count+i].block;
360 if (blocks[0] && !blocks[0]->exprs_count && op->id != opid1(',')) {
361 parseerror(parser, "internal error: operator cannot be applied on empty blocks");
365 #define NotSameType(T) \
366 (exprs[0]->expression.vtype != exprs[1]->expression.vtype || \
367 exprs[0]->expression.vtype != T)
371 parseerror(parser, "internal error: unhandled operand");
375 if (exprs[0]->expression.vtype == TYPE_ENTITY) {
376 if (exprs[1]->expression.vtype != TYPE_FIELD) {
377 parseerror(parser, "type error: right hand of member-operand should be an entity-field");
380 out = (ast_expression*)ast_entfield_new(ctx, exprs[0], exprs[1]);
382 else if (exprs[0]->expression.vtype == TYPE_VECTOR) {
383 parseerror(parser, "internal error: vector access is not supposed to be handled at this point");
387 parseerror(parser, "type error: member-of operator on something that is not an entity or vector");
394 if (!ast_block_exprs_add(blocks[0], exprs[1]))
397 blocks[0] = ast_block_new(ctx);
398 if (!ast_block_exprs_add(blocks[0], exprs[0]) ||
399 !ast_block_exprs_add(blocks[0], exprs[1]))
404 if (!ast_block_set_type(blocks[0], exprs[1]))
407 sy->out[sy->out_count++] = syblock(ctx, blocks[0]);
411 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
412 parseerror(parser, "Cannot add type %s and %s",
413 type_name[exprs[0]->expression.vtype],
414 type_name[exprs[1]->expression.vtype]);
417 if (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) {
418 parseerror(parser, "type error: %s - %s not defined",
419 type_name[exprs[0]->expression.vtype],
420 type_name[exprs[1]->expression.vtype]);
423 switch (exprs[0]->expression.vtype) {
425 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
428 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
431 parseerror(parser, "type error: cannot add type %s to %s",
432 type_name[exprs[0]->expression.vtype],
433 type_name[exprs[1]->expression.vtype]);
438 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
439 parseerror(parser, "type error: cannot subtract type %s from %s",
440 type_name[exprs[1]->expression.vtype],
441 type_name[exprs[0]->expression.vtype]);
444 if (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) {
445 parseerror(parser, "type error: %s - %s not defined",
446 type_name[exprs[0]->expression.vtype],
447 type_name[exprs[1]->expression.vtype]);
450 switch (exprs[0]->expression.vtype) {
452 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
455 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
458 parseerror(parser, "type error: cannot subtract type %s from %s",
459 type_name[exprs[1]->expression.vtype],
460 type_name[exprs[0]->expression.vtype]);
465 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
466 exprs[0]->expression.vtype != TYPE_VECTOR &&
467 exprs[0]->expression.vtype != TYPE_FLOAT &&
468 exprs[1]->expression.vtype != TYPE_VECTOR &&
469 exprs[1]->expression.vtype != TYPE_FLOAT)
471 parseerror(parser, "type error: cannot multiply type %s by %s",
472 type_name[exprs[1]->expression.vtype],
473 type_name[exprs[0]->expression.vtype]);
476 switch (exprs[0]->expression.vtype) {
478 if (exprs[1]->expression.vtype == TYPE_VECTOR)
479 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
481 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
484 if (exprs[1]->expression.vtype == TYPE_FLOAT)
485 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
487 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
490 parseerror(parser, "type error: cannot multiplye type %s by %s",
491 type_name[exprs[1]->expression.vtype],
492 type_name[exprs[0]->expression.vtype]);
497 if (NotSameType(TYPE_FLOAT)) {
498 parseerror(parser, "type error: cannot divide types %s and %s",
499 type_name[exprs[0]->expression.vtype],
500 type_name[exprs[1]->expression.vtype]);
503 out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
507 if (NotSameType(TYPE_FLOAT)) {
508 parseerror(parser, "type error: cannot perform bit operations on types %s and %s",
509 type_name[exprs[0]->expression.vtype],
510 type_name[exprs[1]->expression.vtype]);
513 out = (ast_expression*)ast_binary_new(ctx,
514 (op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
519 generated_op += 1; /* INSTR_GT */
521 generated_op += 1; /* INSTR_LT */
522 case opid2('>', '='):
523 generated_op += 1; /* INSTR_GE */
524 case opid2('<', '='):
525 generated_op += INSTR_LE;
526 if (NotSameType(TYPE_FLOAT)) {
527 parseerror(parser, "type error: cannot compare types %s and %s",
528 type_name[exprs[0]->expression.vtype],
529 type_name[exprs[1]->expression.vtype]);
532 out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
534 case opid2('!', '='):
535 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
536 parseerror(parser, "type error: cannot compare types %s and %s",
537 type_name[exprs[0]->expression.vtype],
538 type_name[exprs[1]->expression.vtype]);
541 out = (ast_expression*)ast_binary_new(ctx, type_ne_op[exprs[0]->expression.vtype], exprs[0], exprs[1]);
543 case opid2('=', '='):
544 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
545 parseerror(parser, "type error: cannot compare types %s and %s",
546 type_name[exprs[0]->expression.vtype],
547 type_name[exprs[1]->expression.vtype]);
550 out = (ast_expression*)ast_binary_new(ctx, type_eq_op[exprs[0]->expression.vtype], exprs[0], exprs[1]);
555 if (ast_istype(exprs[0], ast_entfield))
556 assignop = type_storep_instr[exprs[0]->expression.vtype];
558 assignop = type_store_instr[exprs[0]->expression.vtype];
559 out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]);
565 parseerror(parser, "failed to apply operand %s", op->op);
569 DEBUGSHUNTDO(printf("applied %s\n", op->op));
570 sy->out[sy->out_count++] = syexp(ctx, out);
574 static bool parser_close_call(parser_t *parser, shunt *sy)
576 /* was a function call */
584 fid = sy->ops[sy->ops_count].off;
586 /* out[fid] is the function
587 * everything above is parameters...
589 * 1 params = ast_expression
593 if (sy->out_count < 1 || sy->out_count <= fid) {
594 parseerror(parser, "internal error: function call needs function and parameter list...");
598 fun = sy->out[fid].out;
600 call = ast_call_new(sy->ops[sy->ops_count].ctx, fun);
602 parseerror(parser, "out of memory");
606 if (fid+1 == sy->out_count) {
609 } else if (fid+2 == sy->out_count) {
612 params = sy->out[sy->out_count].block;
616 if (!ast_call_params_add(call, sy->out[sy->out_count].out)) {
617 ast_delete(sy->out[sy->out_count].out);
618 parseerror(parser, "out of memory");
622 paramcount = params->exprs_count;
623 MEM_VECTOR_MOVE(params, exprs, call, params);
627 parseerror(parser, "invalid function call");
631 /* overwrite fid, the function, with a call */
632 sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
634 if (fun->expression.vtype != TYPE_FUNCTION) {
635 parseerror(parser, "not a function");
639 if (!fun->expression.next) {
640 parseerror(parser, "could not determine function return type");
643 if (fun->expression.params_count != paramcount) {
644 parseerror(parser, "expected %i parameters, got %i", (int)fun->expression.params_count, paramcount);
652 static bool parser_close_paren(parser_t *parser, shunt *sy, bool functions_only)
654 if (!sy->ops_count) {
655 parseerror(parser, "unmatched closing paren");
658 if (sy->ops[sy->ops_count-1].paren == 1) {
659 parseerror(parser, "empty parenthesis expression");
662 while (sy->ops_count) {
663 if (sy->ops[sy->ops_count-1].paren == 'f') {
664 if (!parser_close_call(parser, sy))
668 if (sy->ops[sy->ops_count-1].paren == 1) {
670 return !functions_only;
672 if (!parser_sy_pop(parser, sy))
678 static ast_expression* parser_expression(parser_t *parser)
680 ast_expression *expr = NULL;
683 bool gotmemberof = false;
685 /* count the parens because an if starts with one, so the
686 * end of a condition is an unmatched closing paren
690 MEM_VECTOR_INIT(&sy, out);
691 MEM_VECTOR_INIT(&sy, ops);
698 parser->memberof = 0;
701 bool nextwant = true;
702 if (parser->tok == TOKEN_IDENT)
706 if (opts_standard == COMPILER_GMQCC)
708 if (parser->memberof == TYPE_ENTITY)
709 var = parser_find_field(parser, parser_tokval(parser));
710 else if (parser->memberof == TYPE_VECTOR)
712 parseerror(parser, "TODO: implement effective vector member access");
715 else if (parser->memberof) {
716 parseerror(parser, "namespace for member not found");
720 var = parser_find_var(parser, parser_tokval(parser));
722 var = parser_find_var(parser, parser_tokval(parser));
724 var = parser_find_field(parser, parser_tokval(parser));
727 parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
730 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), var))) {
731 parseerror(parser, "out of memory");
734 DEBUGSHUNTDO(printf("push %s\n", parser_tokval(parser)));
736 else if (parser->tok == TOKEN_FLOATCONST) {
737 ast_value *val = parser_const_float(parser, (parser_token(parser)->constval.f));
740 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
741 parseerror(parser, "out of memory");
744 DEBUGSHUNTDO(printf("push %g\n", parser_token(parser)->constval.f));
746 else if (parser->tok == TOKEN_INTCONST) {
747 ast_value *val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
750 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
751 parseerror(parser, "out of memory");
754 DEBUGSHUNTDO(printf("push %i\n", parser_token(parser)->constval.i));
756 else if (parser->tok == TOKEN_STRINGCONST) {
757 ast_value *val = parser_const_string(parser, parser_tokval(parser));
760 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
761 parseerror(parser, "out of memory");
764 DEBUGSHUNTDO(printf("push string\n"));
766 else if (parser->tok == TOKEN_VECTORCONST) {
767 ast_value *val = parser_const_vector(parser, parser_token(parser)->constval.v);
770 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
771 parseerror(parser, "out of memory");
774 DEBUGSHUNTDO(printf("push '%g %g %g'\n",
775 parser_token(parser)->constval.v.x,
776 parser_token(parser)->constval.v.y,
777 parser_token(parser)->constval.v.z));
779 else if (parser->tok == '(') {
781 nextwant = false; /* not expecting an operator next */
782 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1, 0))) {
783 parseerror(parser, "out of memory");
786 DEBUGSHUNTDO(printf("push (\n"));
788 else if (parser->tok == ')') {
789 DEBUGSHUNTDO(printf("do[nop] )\n"));
793 /* allowed for function calls */
794 if (!parser_close_paren(parser, &sy, true))
798 /* TODO: prefix operators */
799 parseerror(parser, "expected statement");
803 parser->lex->flags.noops = !wantop;
805 bool nextwant = false;
806 if (parser->tok == '(') {
807 DEBUGSHUNTDO(printf("push (\n"));
809 /* we expected an operator, this is the function-call operator */
810 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 'f', sy.out_count-1))) {
811 parseerror(parser, "out of memory");
815 else if (parser->tok == ')') {
816 DEBUGSHUNTDO(printf("do[op] )\n"));
820 /* we do expect an operator next */
821 /* closing an opening paren */
822 if (!parser_close_paren(parser, &sy, false))
826 else if (parser->tok != TOKEN_OPERATOR) {
827 parseerror(parser, "expected operator or end of statement");
831 /* classify the operator */
832 /* TODO: suffix operators */
834 const oper_info *olast = NULL;
836 for (o = 0; o < operator_count; ++o) {
837 if (!(operators[o].flags & OP_PREFIX) &&
838 !(operators[o].flags & OP_SUFFIX) && /* remove this */
839 !strcmp(parser_tokval(parser), operators[o].op))
844 if (o == operator_count) {
845 /* no operator found... must be the end of the statement */
848 /* found an operator */
850 if (op->id == opid1('.')) {
851 /* for gmqcc standard: open up the namespace of the previous type */
852 ast_expression *prevex = sy.out[sy.out_count-1].out;
854 parseerror(parser, "unexpected member operator");
857 if (prevex->expression.vtype == TYPE_ENTITY)
858 parser->memberof = TYPE_ENTITY;
859 else if (prevex->expression.vtype == TYPE_VECTOR)
860 parser->memberof = TYPE_VECTOR;
862 parseerror(parser, "type error: type has no members");
868 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
869 olast = &operators[sy.ops[sy.ops_count-1].etype-1];
872 (op->prec < olast->prec) ||
873 (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
875 if (!parser_sy_pop(parser, &sy))
877 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
878 olast = &operators[sy.ops[sy.ops_count-1].etype-1];
883 DEBUGSHUNTDO(printf("push operator %s\n", op->op));
884 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
888 parser->lex->flags.noops = !wantop;
890 if (!parser_next(parser)) {
893 if (parser->tok == ';') {
897 if (parens >= 0 && !parser_next(parser)) {
898 parseerror(parser, "Unexpected end of file");
902 while (sy.ops_count) {
903 if (!parser_sy_pop(parser, &sy))
907 parser->lex->flags.noops = true;
909 parseerror(parser, "empty expression");
912 expr = sy.out[0].out;
913 MEM_VECTOR_CLEAR(&sy, out);
914 MEM_VECTOR_CLEAR(&sy, ops);
915 DEBUGSHUNTDO(printf("shut done\n"));
919 parser->lex->flags.noops = true;
920 MEM_VECTOR_CLEAR(&sy, out);
921 MEM_VECTOR_CLEAR(&sy, ops);
925 static bool parser_variable(parser_t *parser, ast_block *localblock);
926 static ast_block* parser_parse_block(parser_t *parser);
927 static ast_expression* parser_parse_statement_or_block(parser_t *parser);
928 static bool parser_parse_statement(parser_t *parser, ast_block *block, ast_expression **out)
930 if (parser->tok == TOKEN_TYPENAME)
934 parseerror(parser, "cannot declare a variable from here");
937 if (!parser_variable(parser, block))
942 else if (parser->tok == TOKEN_KEYWORD)
944 if (!strcmp(parser_tokval(parser), "return"))
946 ast_expression *exp = NULL;
947 ast_return *ret = NULL;
948 ast_value *expected = parser->function->vtype;
950 if (!parser_next(parser)) {
951 parseerror(parser, "expected return expression");
955 if (parser->tok != ';') {
956 exp = parser_expression(parser);
960 if (exp->expression.vtype != expected->expression.next->expression.vtype) {
961 parseerror(parser, "return with invalid expression");
964 ret = ast_return_new(exp->expression.node.context, exp);
970 *out = (ast_expression*)ret;
971 } else if (!parser_next(parser)) {
972 parseerror(parser, "expected semicolon");
973 if (expected->expression.next->expression.vtype != TYPE_VOID) {
974 parseerror(parser, "return without value");
979 else if (!strcmp(parser_tokval(parser), "if"))
982 ast_expression *cond, *ontrue, *onfalse = NULL;
984 lex_ctx ctx = parser_ctx(parser);
986 /* skip the 'if' and check for opening paren */
987 if (!parser_next(parser) || parser->tok != '(') {
988 parseerror(parser, "expected 'if' condition in parenthesis");
991 /* parse into the expression */
992 if (!parser_next(parser)) {
993 parseerror(parser, "expected 'if' condition after opening paren");
996 /* parse the condition */
997 cond = parser_expression(parser);
1001 if (parser->tok != ')') {
1002 parseerror(parser, "expected closing paren after 'if' condition");
1006 /* parse into the 'then' branch */
1007 if (!parser_next(parser)) {
1008 parseerror(parser, "expected statement for on-true branch of 'if'");
1012 ontrue = parser_parse_statement_or_block(parser);
1017 /* check for an else */
1018 if (!strcmp(parser_tokval(parser), "else")) {
1019 /* parse into the 'else' branch */
1020 if (!parser_next(parser)) {
1021 parseerror(parser, "expected on-false branch after 'else'");
1026 onfalse = parser_parse_statement_or_block(parser);
1034 ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
1035 *out = (ast_expression*)ifthen;
1038 parseerror(parser, "Unexpected keyword");
1041 else if (parser->tok == '{')
1044 inner = parser_parse_block(parser);
1047 *out = (ast_expression*)inner;
1052 ast_expression *exp = parser_expression(parser);
1060 static void parser_pop_local(parser_t *parser)
1062 parser->locals_count--;
1063 mem_d(parser->locals[parser->locals_count].name);
1066 static ast_block* parser_parse_block(parser_t *parser)
1068 size_t oldblocklocal;
1069 ast_block *block = NULL;
1071 oldblocklocal = parser->blocklocal;
1072 parser->blocklocal = parser->locals_count;
1074 if (!parser_next(parser)) { /* skip the '{' */
1075 parseerror(parser, "expected function body");
1079 block = ast_block_new(parser_ctx(parser));
1081 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1083 ast_expression *expr;
1084 if (parser->tok == '}')
1087 if (!parser_parse_statement(parser, block, &expr)) {
1088 ast_block_delete(block);
1094 if (!ast_block_exprs_add(block, expr)) {
1096 ast_block_delete(block);
1102 if (parser->tok != '}') {
1103 ast_block_delete(block);
1106 (void)parser_next(parser);
1110 while (parser->locals_count > parser->blocklocal)
1111 parser_pop_local(parser);
1112 parser->blocklocal = oldblocklocal;
1113 /* unroll the local vector */
1117 static ast_expression* parser_parse_statement_or_block(parser_t *parser)
1119 ast_expression *expr;
1120 if (parser->tok == '{')
1121 return (ast_expression*)parser_parse_block(parser);
1122 if (!parser_parse_statement(parser, NULL, &expr))
1127 static bool parser_variable(parser_t *parser, ast_block *localblock)
1129 bool isfunc = false;
1130 ast_function *func = NULL;
1135 int basetype = parser_token(parser)->constval.t;
1139 if (!parser_next(parser)) { /* skip basetype or comma */
1140 parseerror(parser, "expected variable declaration");
1146 ctx = parser_ctx(parser);
1147 var = parser_parse_type(parser, basetype, &isfunc);
1152 if (parser->tok != TOKEN_IDENT) {
1153 parseerror(parser, "expected variable name\n");
1157 if (!localblock && parser_find_global(parser, parser_tokval(parser))) {
1158 ast_value_delete(var);
1159 parseerror(parser, "global already exists: %s\n", parser_tokval(parser));
1163 if (localblock && parser_find_local(parser, parser_tokval(parser), parser->blocklocal)) {
1164 ast_value_delete(var);
1165 parseerror(parser, "local variable already exists: %s\n", parser_tokval(parser));
1169 if (!ast_value_set_name(var, parser_tokval(parser))) {
1170 parseerror(parser, "failed to set variable name\n");
1171 ast_value_delete(var);
1176 /* a function was defined */
1179 /* turn var into a value of TYPE_FUNCTION, with the old var
1182 fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1183 func = ast_function_new(ctx, var->name, fval);
1184 if (!fval || !func) {
1185 ast_value_delete(var);
1186 if (fval) ast_value_delete(fval);
1187 if (func) ast_function_delete(func);
1191 fval->expression.next = (ast_expression*)var;
1192 MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1194 if (!parser_t_functions_add(parser, func)) {
1195 ast_value_delete(var);
1196 if (fval) ast_value_delete(fval);
1197 if (func) ast_function_delete(func);
1204 varent.name = util_strdup(var->name);
1205 varent.var = (ast_expression*)var;
1206 if (var->expression.vtype == TYPE_VECTOR)
1208 size_t len = strlen(varent.name);
1209 varentry_t vx, vy, vz;
1210 vx.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 0);
1211 vy.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 1);
1212 vz.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 2);
1213 vx.name = mem_a(len+3);
1214 vy.name = mem_a(len+3);
1215 vz.name = mem_a(len+3);
1216 strcpy(vx.name, varent.name);
1217 strcpy(vy.name, varent.name);
1218 strcpy(vz.name, varent.name);
1219 vx.name[len] = vy.name[len] = vz.name[len] = '_';
1220 vx.name[len+1] = 'x';
1221 vy.name[len+1] = 'y';
1222 vz.name[len+1] = 'z';
1223 vx.name[len+2] = vy.name[len+2] = vz.name[len+2] = 0;
1226 (void)!parser_t_globals_add(parser, varent);
1227 (void)!parser_t_globals_add(parser, vx);
1228 (void)!parser_t_globals_add(parser, vy);
1229 (void)!parser_t_globals_add(parser, vz);
1231 (void)!parser_t_locals_add(parser, varent);
1232 (void)!parser_t_locals_add(parser, vx);
1233 (void)!parser_t_locals_add(parser, vy);
1234 (void)!parser_t_locals_add(parser, vz);
1239 if ( (!localblock && !parser_t_globals_add(parser, varent)) ||
1240 ( localblock && !parser_t_locals_add(parser, varent)) )
1242 ast_value_delete(var);
1246 if (localblock && !ast_block_locals_add(localblock, var))
1248 parser_pop_local(parser);
1249 ast_value_delete(var);
1253 if (!parser_next(parser)) {
1254 ast_value_delete(var);
1258 if (parser->tok == ';') {
1259 if (!parser_next(parser))
1260 return parser->tok == TOKEN_EOF;
1264 if (parser->tok == ',') {
1269 if (parser->tok != '=') {
1270 parseerror(parser, "expected '=' or ';'");
1274 if (!parser_next(parser))
1277 if (parser->tok == '#') {
1279 parseerror(parser, "cannot declare builtins within functions");
1282 if (!isfunc || !func) {
1283 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
1286 if (!parser_next(parser)) {
1287 parseerror(parser, "expected builtin number");
1290 if (parser->tok != TOKEN_INTCONST) {
1291 parseerror(parser, "builtin number must be an integer constant");
1294 if (parser_token(parser)->constval.i <= 0) {
1295 parseerror(parser, "builtin number must be positive integer greater than zero");
1299 func->builtin = -parser_token(parser)->constval.i;
1300 } else if (parser->tok == '{') {
1303 ast_function *old = parser->function;
1306 parseerror(parser, "cannot declare functions within functions");
1310 parser->function = func;
1311 block = parser_parse_block(parser);
1312 parser->function = old;
1317 if (!ast_function_blocks_add(func, block)) {
1318 ast_block_delete(block);
1323 parseerror(parser, "TODO, const assignment");
1326 if (!parser_next(parser))
1329 if (parser->tok == ',') {
1334 if (parser->tok != ';') {
1335 parseerror(parser, "expected semicolon");
1339 (void)parser_next(parser);
1345 static bool parser_do(parser_t *parser)
1347 if (parser->tok == TOKEN_TYPENAME)
1349 return parser_variable(parser, NULL);
1351 else if (parser->tok == TOKEN_KEYWORD)
1353 /* handle 'var' and 'const' */
1356 else if (parser->tok == '.')
1360 bool isfunc = false;
1362 lex_ctx ctx = parser_ctx(parser);
1365 /* entity-member declaration */
1366 if (!parser_next(parser) || parser->tok != TOKEN_TYPENAME) {
1367 parseerror(parser, "expected member variable definition");
1371 /* remember the base/return type */
1372 basetype = parser_token(parser)->constval.t;
1374 /* parse into the declaration */
1375 if (!parser_next(parser)) {
1376 parseerror(parser, "expected field def");
1380 /* parse the field type fully */
1381 var = parser_parse_type(parser, basetype, &isfunc);
1385 /* now the field name */
1386 if (parser->tok != TOKEN_IDENT) {
1387 parseerror(parser, "expected field name");
1392 /* check for an existing field
1393 * in original qc we also have to check for an existing
1394 * global named like the field
1396 if (opts_standard == COMPILER_QCC) {
1397 if (parser_find_global(parser, parser_tokval(parser))) {
1398 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
1403 if (parser_find_field(parser, parser_tokval(parser))) {
1404 parseerror(parser, "field %s already exists", parser_tokval(parser));
1409 /* if it was a function, turn it into a function */
1412 /* turn var into a value of TYPE_FUNCTION, with the old var
1415 fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1417 ast_value_delete(var);
1418 ast_value_delete(fval);
1422 fval->expression.next = (ast_expression*)var;
1423 MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1428 /* turn it into a field */
1429 fld = ast_value_new(ctx, parser_tokval(parser), TYPE_FIELD);
1430 fld->expression.next = (ast_expression*)var;
1432 varent.var = (ast_expression*)fld;
1433 if (var->expression.vtype == TYPE_VECTOR)
1435 /* create _x, _y and _z fields as well */
1436 parseerror(parser, "TODO: vector field members (_x,_y,_z)");
1441 varent.name = util_strdup(fld->name);
1442 (void)!parser_t_fields_add(parser, varent);
1444 /* end with a semicolon */
1445 if (!parser_next(parser) || parser->tok != ';') {
1446 parseerror(parser, "semicolon expected");
1450 /* skip the semicolon */
1451 if (!parser_next(parser))
1452 return parser->tok == TOKEN_EOF;
1458 parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
1464 static parser_t *parser;
1468 parser = (parser_t*)mem_a(sizeof(parser_t));
1472 memset(parser, 0, sizeof(parser));
1474 MEM_VECTOR_INIT(parser, globals);
1475 MEM_VECTOR_INIT(parser, locals);
1479 bool parser_compile(const char *filename)
1481 parser->lex = lex_open(filename);
1483 printf("failed to open file \"%s\"\n", filename);
1487 /* initial lexer/parser state */
1488 parser->lex->flags.noops = true;
1490 if (parser_next(parser))
1492 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1494 if (!parser_do(parser)) {
1495 if (parser->tok == TOKEN_EOF)
1496 parseerror(parser, "unexpected eof");
1498 parseerror(parser, "parse error\n");
1499 lex_close(parser->lex);
1506 lex_close(parser->lex);
1508 return !parser->errors;
1511 void parser_cleanup()
1514 for (i = 0; i < parser->functions_count; ++i) {
1515 ast_delete(parser->functions[i]);
1517 for (i = 0; i < parser->imm_vector_count; ++i) {
1518 ast_delete(parser->imm_vector[i]);
1520 for (i = 0; i < parser->imm_string_count; ++i) {
1521 ast_delete(parser->imm_string[i]);
1523 for (i = 0; i < parser->imm_float_count; ++i) {
1524 ast_delete(parser->imm_float[i]);
1526 for (i = 0; i < parser->globals_count; ++i) {
1527 ast_delete(parser->globals[i].var);
1528 mem_d(parser->globals[i].name);
1530 MEM_VECTOR_CLEAR(parser, globals);
1535 bool parser_finish(const char *output)
1540 if (!parser->errors)
1542 ir = ir_builder_new("gmqcc_out");
1544 printf("failed to allocate builder\n");
1548 for (i = 0; i < parser->imm_float_count; ++i) {
1549 if (!ast_global_codegen(parser->imm_float[i], ir)) {
1550 printf("failed to generate global %s\n", parser->imm_float[i]->name);
1551 ir_builder_delete(ir);
1555 for (i = 0; i < parser->imm_string_count; ++i) {
1556 if (!ast_global_codegen(parser->imm_string[i], ir)) {
1557 printf("failed to generate global %s\n", parser->imm_string[i]->name);
1558 ir_builder_delete(ir);
1562 for (i = 0; i < parser->imm_vector_count; ++i) {
1563 if (!ast_global_codegen(parser->imm_vector[i], ir)) {
1564 printf("failed to generate global %s\n", parser->imm_vector[i]->name);
1565 ir_builder_delete(ir);
1569 for (i = 0; i < parser->fields_count; ++i) {
1572 if (!ast_istype(parser->fields[i].var, ast_value))
1574 field = (ast_value*)parser->fields[i].var;
1575 isconst = field->isconst;
1576 field->isconst = false;
1577 if (!ast_global_codegen((ast_value*)field, ir)) {
1578 printf("failed to generate field %s\n", field->name);
1579 ir_builder_delete(ir);
1584 ast_expression *subtype;
1585 field->isconst = true;
1586 subtype = field->expression.next;
1587 ifld = ir_builder_create_field(ir, field->name, subtype->expression.vtype);
1588 if (subtype->expression.vtype == TYPE_FIELD)
1589 ifld->fieldtype = subtype->expression.next->expression.vtype;
1590 else if (subtype->expression.vtype == TYPE_FUNCTION)
1591 ifld->outtype = subtype->expression.next->expression.vtype;
1592 (void)!ir_value_set_field(field->ir_v, ifld);
1595 for (i = 0; i < parser->globals_count; ++i) {
1596 if (!ast_istype(parser->globals[i].var, ast_value))
1598 if (!ast_global_codegen((ast_value*)(parser->globals[i].var), ir)) {
1599 printf("failed to generate global %s\n", parser->globals[i].name);
1600 ir_builder_delete(ir);
1604 for (i = 0; i < parser->functions_count; ++i) {
1605 if (!ast_function_codegen(parser->functions[i], ir)) {
1606 printf("failed to generate function %s\n", parser->functions[i]->name);
1607 ir_builder_delete(ir);
1610 if (!ir_function_finalize(parser->functions[i]->ir_func)) {
1611 printf("failed to finalize function %s\n", parser->functions[i]->name);
1612 ir_builder_delete(ir);
1617 ir_builder_dump(ir, printf);
1619 if (!ir_builder_generate(ir, output)) {
1620 printf("*** failed to generate output file\n");
1621 ir_builder_delete(ir);
1625 ir_builder_delete(ir);
1629 printf("*** there were compile errors\n");