16 MEM_VECTOR_MAKE(varentry_t, globals);
17 MEM_VECTOR_MAKE(ast_function*, functions);
18 MEM_VECTOR_MAKE(ast_value*, imm_float);
19 MEM_VECTOR_MAKE(ast_value*, imm_string);
20 MEM_VECTOR_MAKE(ast_value*, imm_vector);
22 ast_function *function;
23 MEM_VECTOR_MAKE(varentry_t, locals);
29 MEM_VEC_FUNCTIONS(parser_t, varentry_t, globals)
30 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_float)
31 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_string)
32 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_vector)
33 MEM_VEC_FUNCTIONS(parser_t, varentry_t, locals)
34 MEM_VEC_FUNCTIONS(parser_t, ast_function*, functions)
36 void parseerror(parser_t *parser, const char *fmt, ...)
43 printf("error %s:%lu: ", parser->lex->tok->ctx.file, (unsigned long)parser->lex->tok->ctx.line);
54 bool parser_next(parser_t *parser)
56 /* lex_do kills the previous token */
57 parser->tok = lex_do(parser->lex);
58 if (parser->tok == TOKEN_EOF || parser->tok >= TOKEN_ERROR)
63 /* lift a token out of the parser so it's not destroyed by parser_next */
64 token *parser_lift(parser_t *parser)
66 token *tok = parser->lex->tok;
67 parser->lex->tok = NULL;
71 #define parser_tokval(p) (p->lex->tok->value)
72 #define parser_token(p) (p->lex->tok)
73 #define parser_ctx(p) (p->lex->tok->ctx)
75 ast_value* parser_const_float(parser_t *parser, double d)
79 for (i = 0; i < parser->imm_float_count; ++i) {
80 if (parser->imm_float[i]->constval.vfloat == d)
81 return parser->imm_float[i];
83 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_FLOAT);
85 out->constval.vfloat = d;
86 if (!parser_t_imm_float_add(parser, out)) {
87 ast_value_delete(out);
93 ast_value* parser_const_string(parser_t *parser, const char *str)
97 for (i = 0; i < parser->imm_string_count; ++i) {
98 if (!strcmp(parser->imm_string[i]->constval.vstring, str))
99 return parser->imm_string[i];
101 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_STRING);
103 out->constval.vstring = util_strdup(str);
104 if (!parser_t_imm_string_add(parser, out)) {
105 ast_value_delete(out);
111 ast_value* parser_const_vector(parser_t *parser, vector v)
115 for (i = 0; i < parser->imm_vector_count; ++i) {
116 if (!memcmp(&parser->imm_vector[i]->constval.vvec, &v, sizeof(v)))
117 return parser->imm_vector[i];
119 out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_VECTOR);
121 out->constval.vvec = v;
122 if (!parser_t_imm_vector_add(parser, out)) {
123 ast_value_delete(out);
129 ast_expression* parser_find_global(parser_t *parser, const char *name)
132 for (i = 0; i < parser->globals_count; ++i) {
133 if (!strcmp(parser->globals[i].name, name))
134 return parser->globals[i].var;
139 ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto)
143 for (i = parser->locals_count; i > upto;) {
145 if (!strcmp(parser->locals[i].name, name))
146 return parser->locals[i].var;
148 fun = parser->function->vtype;
149 for (i = 0; i < fun->expression.params_count; ++i) {
150 if (!strcmp(fun->expression.params[i]->name, name))
151 return (ast_expression*)(fun->expression.params[i]);
156 ast_expression* parser_find_var(parser_t *parser, const char *name)
159 v = parser_find_local(parser, name, 0);
160 if (!v) v = parser_find_global(parser, name);
165 MEM_VECTOR_MAKE(ast_value*, p);
167 MEM_VEC_FUNCTIONS(paramlist_t, ast_value*, p)
169 static ast_value *parser_parse_type(parser_t *parser, int basetype, bool *isfunc)
173 lex_ctx ctx = parser_ctx(parser);
174 int vtype = basetype;
178 MEM_VECTOR_INIT(¶ms, p);
182 if (parser->tok == '(') {
188 if (!parser_next(parser))
191 if (parser->tok == ')')
194 temptype = parser_token(parser)->constval.t;
195 if (!parser_next(parser))
198 param = parser_parse_type(parser, temptype, &dummy);
204 if (parser->tok == TOKEN_IDENT) {
205 /* named parameter */
206 if (!ast_value_set_name(param, parser_tokval(parser)))
208 if (!parser_next(parser))
212 if (!paramlist_t_p_add(¶ms, param)) {
213 parseerror(parser, "Out of memory while parsing typename");
217 if (parser->tok == ',')
219 if (parser->tok == ')')
221 parseerror(parser, "Unexpected token");
224 if (!parser_next(parser))
228 var = ast_value_new(ctx, "<unnamed>", vtype);
231 MEM_VECTOR_MOVE(¶ms, p, &var->expression, params);
234 for (i = 0; i < params.p_count; ++i)
235 ast_value_delete(params.p[i]);
236 MEM_VECTOR_CLEAR(¶ms, p);
242 size_t etype; /* 0 = expression, others are operators */
246 ast_block *block; /* for commas and function calls */
251 MEM_VECTOR_MAKE(sy_elem, out);
252 MEM_VECTOR_MAKE(sy_elem, ops);
254 MEM_VEC_FUNCTIONS(shunt, sy_elem, out)
255 MEM_VEC_FUNCTIONS(shunt, sy_elem, ops)
257 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
267 static sy_elem syblock(lex_ctx ctx, ast_block *v) {
270 e.out = (ast_expression*)v;
277 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
279 e.etype = 1 + (op - operators);
287 static sy_elem syparen(lex_ctx ctx, int p, size_t off) {
298 static bool parser_sy_pop(parser_t *parser, shunt *sy)
302 ast_expression *out = NULL;
303 ast_expression *exprs[3];
304 ast_block *blocks[3];
307 if (!sy->ops_count) {
308 parseerror(parser, "internal error: missing operator");
312 if (sy->ops[sy->ops_count-1].paren) {
313 parseerror(parser, "unmatched parenthesis");
317 op = &operators[sy->ops[sy->ops_count-1].etype - 1];
318 ctx = sy->ops[sy->ops_count-1].ctx;
320 if (sy->out_count < op->operands) {
321 parseerror(parser, "internal error: not enough operands: %i", sy->out_count);
327 sy->out_count -= op->operands;
328 for (i = 0; i < op->operands; ++i) {
329 exprs[i] = sy->out[sy->out_count+i].out;
330 blocks[i] = sy->out[sy->out_count+i].block;
333 if (blocks[0] && !blocks[0]->exprs_count && op->id != opid1(',')) {
334 parseerror(parser, "internal error: operator cannot be applied on empty blocks");
341 parseerror(parser, "internal error: unhandled operand");
346 if (!ast_block_exprs_add(blocks[0], exprs[1]))
349 blocks[0] = ast_block_new(ctx);
350 if (!ast_block_exprs_add(blocks[0], exprs[0]) ||
351 !ast_block_exprs_add(blocks[0], exprs[1]))
356 if (!ast_block_set_type(blocks[0], exprs[1]))
359 sy->out[sy->out_count++] = syblock(ctx, blocks[0]);
363 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
364 parseerror(parser, "Cannot add type %s and %s",
365 type_name[exprs[0]->expression.vtype],
366 type_name[exprs[1]->expression.vtype]);
369 switch (exprs[0]->expression.vtype) {
371 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
374 out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
377 parseerror(parser, "Cannot add type %s and %s",
378 type_name[exprs[0]->expression.vtype],
379 type_name[exprs[1]->expression.vtype]);
384 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
385 parseerror(parser, "Cannot subtract type %s from %s",
386 type_name[exprs[1]->expression.vtype],
387 type_name[exprs[0]->expression.vtype]);
390 switch (exprs[0]->expression.vtype) {
392 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
395 out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
398 parseerror(parser, "Cannot add type %s from %s",
399 type_name[exprs[1]->expression.vtype],
400 type_name[exprs[0]->expression.vtype]);
405 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
406 exprs[0]->expression.vtype != TYPE_VECTOR &&
407 exprs[0]->expression.vtype != TYPE_FLOAT &&
408 exprs[1]->expression.vtype != TYPE_VECTOR &&
409 exprs[1]->expression.vtype != TYPE_FLOAT)
411 parseerror(parser, "Cannot multiply type %s from %s",
412 type_name[exprs[1]->expression.vtype],
413 type_name[exprs[0]->expression.vtype]);
416 switch (exprs[0]->expression.vtype) {
418 if (exprs[1]->expression.vtype == TYPE_VECTOR)
419 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
421 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
424 if (exprs[1]->expression.vtype == TYPE_FLOAT)
425 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
427 out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
430 parseerror(parser, "Cannot add type %s from %s",
431 type_name[exprs[1]->expression.vtype],
432 type_name[exprs[0]->expression.vtype]);
437 if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
438 exprs[0]->expression.vtype != TYPE_FLOAT)
440 parseerror(parser, "Cannot divide types %s and %s",
441 type_name[exprs[0]->expression.vtype],
442 type_name[exprs[1]->expression.vtype]);
445 out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
450 out = (ast_expression*)ast_store_new(ctx,
451 type_store_instr[exprs[0]->expression.vtype],
457 parseerror(parser, "failed to apply operand %s", op->op);
461 sy->out[sy->out_count++] = syexp(ctx, out);
465 static bool parser_close_call(parser_t *parser, shunt *sy)
467 /* was a function call */
475 fid = sy->ops[sy->ops_count].off;
477 /* out[fid] is the function
478 * everything above is parameters...
480 * 1 params = ast_expression
484 if (sy->out_count < 1 || sy->out_count <= fid) {
485 parseerror(parser, "internal error: function call needs function and parameter list...");
489 fun = sy->out[fid].out;
491 call = ast_call_new(sy->ops[sy->ops_count].ctx, fun);
493 parseerror(parser, "out of memory");
497 if (fid+1 == sy->out_count) {
500 } else if (fid+2 == sy->out_count) {
503 params = sy->out[sy->out_count].block;
507 if (!ast_call_params_add(call, sy->out[sy->out_count].out)) {
508 ast_delete(sy->out[sy->out_count].out);
509 parseerror(parser, "out of memory");
513 paramcount = params->exprs_count;
514 MEM_VECTOR_MOVE(params, exprs, call, params);
518 parseerror(parser, "invalid function call");
522 /* overwrite fid, the function, with a call */
523 sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
525 if (fun->expression.vtype != TYPE_FUNCTION) {
526 parseerror(parser, "not a function");
530 if (!fun->expression.next) {
531 parseerror(parser, "could not determine function return type");
534 if (fun->expression.params_count != paramcount) {
535 parseerror(parser, "expected %i parameters, got %i", (int)fun->expression.params_count, paramcount);
543 static bool parser_close_paren(parser_t *parser, shunt *sy, bool functions_only)
545 if (!sy->ops_count) {
546 parseerror(parser, "unmatched closing paren");
549 if (sy->ops[sy->ops_count-1].paren == 1) {
550 parseerror(parser, "empty parenthesis expression");
553 while (sy->ops_count) {
554 if (sy->ops[sy->ops_count-1].paren == 'f') {
555 if (!parser_close_call(parser, sy))
559 if (sy->ops[sy->ops_count-1].paren == 1) {
561 return !functions_only;
563 if (!parser_sy_pop(parser, sy))
569 static ast_expression* parser_expression(parser_t *parser)
571 ast_expression *expr = NULL;
575 /* count the parens because an if starts with one, so the
576 * end of a condition is an unmatched closing paren
580 MEM_VECTOR_INIT(&sy, out);
581 MEM_VECTOR_INIT(&sy, ops);
587 bool nextwant = true;
588 if (parser->tok == TOKEN_IDENT)
591 ast_expression *var = parser_find_var(parser, parser_tokval(parser));
593 parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
596 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), var))) {
597 parseerror(parser, "out of memory");
601 else if (parser->tok == TOKEN_FLOATCONST) {
602 ast_value *val = parser_const_float(parser, (parser_token(parser)->constval.f));
605 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
606 parseerror(parser, "out of memory");
610 else if (parser->tok == TOKEN_INTCONST) {
611 ast_value *val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
614 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
615 parseerror(parser, "out of memory");
619 else if (parser->tok == TOKEN_STRINGCONST) {
620 ast_value *val = parser_const_string(parser, parser_tokval(parser));
623 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
624 parseerror(parser, "out of memory");
628 else if (parser->tok == TOKEN_VECTORCONST) {
629 ast_value *val = parser_const_vector(parser, parser_token(parser)->constval.v);
632 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
633 parseerror(parser, "out of memory");
637 else if (parser->tok == '(') {
639 nextwant = false; /* not expecting an operator next */
640 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1, 0))) {
641 parseerror(parser, "out of memory");
645 else if (parser->tok == ')') {
649 /* allowed for function calls */
650 if (!parser_close_paren(parser, &sy, true))
654 /* TODO: prefix operators */
655 parseerror(parser, "expected statement");
659 parser->lex->flags.noops = !wantop;
661 if (parser->tok == '(') {
663 /* we expected an operator, this is the function-call operator */
664 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 'f', sy.out_count-1))) {
665 parseerror(parser, "out of memory");
669 else if (parser->tok == ')') {
673 /* we do expect an operator next */
674 /* closing an opening paren */
675 if (!parser_close_paren(parser, &sy, false))
678 else if (parser->tok != TOKEN_OPERATOR) {
679 parseerror(parser, "expected operator or end of statement");
683 /* classify the operator */
684 /* TODO: suffix operators */
686 const oper_info *olast = NULL;
688 for (o = 0; o < operator_count; ++o) {
689 if (!(operators[o].flags & OP_PREFIX) &&
690 !(operators[o].flags & OP_SUFFIX) && /* remove this */
691 !strcmp(parser_tokval(parser), operators[o].op))
696 if (o == operator_count) {
697 /* no operator found... must be the end of the statement */
700 /* found an operator */
703 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
704 olast = &operators[sy.ops[sy.ops_count-1].etype-1];
707 (op->prec < olast->prec) ||
708 (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
710 if (!parser_sy_pop(parser, &sy))
712 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
713 olast = &operators[sy.ops[sy.ops_count-1].etype-1];
716 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
720 parser->lex->flags.noops = true;
722 if (!parser_next(parser)) {
725 if (parser->tok == ';') {
729 if (parens >= 0 && !parser_next(parser)) {
730 parseerror(parser, "Unexpected end of file");
734 while (sy.ops_count) {
735 if (!parser_sy_pop(parser, &sy))
739 parser->lex->flags.noops = true;
741 parseerror(parser, "empty expression");
744 expr = sy.out[0].out;
745 MEM_VECTOR_CLEAR(&sy, out);
746 MEM_VECTOR_CLEAR(&sy, ops);
750 parser->lex->flags.noops = true;
751 MEM_VECTOR_CLEAR(&sy, out);
752 MEM_VECTOR_CLEAR(&sy, ops);
756 static bool parser_variable(parser_t *parser, ast_block *localblock);
757 static ast_block* parser_parse_block(parser_t *parser);
758 static ast_expression* parser_parse_statement_or_block(parser_t *parser);
759 static bool parser_parse_statement(parser_t *parser, ast_block *block, ast_expression **out)
761 if (parser->tok == TOKEN_TYPENAME)
765 parseerror(parser, "cannot declare a variable from here");
768 if (!parser_variable(parser, block))
773 else if (parser->tok == TOKEN_KEYWORD)
775 if (!strcmp(parser_tokval(parser), "return"))
777 ast_expression *exp = NULL;
778 ast_return *ret = NULL;
779 ast_value *expected = parser->function->vtype;
781 if (!parser_next(parser)) {
782 parseerror(parser, "expected return expression");
786 if (parser->tok != ';') {
787 exp = parser_expression(parser);
791 if (exp->expression.vtype != expected->expression.next->expression.vtype) {
792 parseerror(parser, "return with invalid expression");
795 ret = ast_return_new(exp->expression.node.context, exp);
801 *out = (ast_expression*)ret;
802 } else if (!parser_next(parser)) {
803 parseerror(parser, "expected semicolon");
804 if (expected->expression.next->expression.vtype != TYPE_VOID) {
805 parseerror(parser, "return without value");
810 else if (!strcmp(parser_tokval(parser), "if"))
813 ast_expression *cond, *ontrue, *onfalse = NULL;
815 lex_ctx ctx = parser_ctx(parser);
817 /* skip the 'if' and check for opening paren */
818 if (!parser_next(parser) || parser->tok != '(') {
819 parseerror(parser, "expected 'if' condition in parenthesis");
822 /* parse into the expression */
823 if (!parser_next(parser)) {
824 parseerror(parser, "expected 'if' condition after opening paren");
827 /* parse the condition */
828 cond = parser_expression(parser);
832 if (parser->tok != ')') {
833 parseerror(parser, "expected closing paren after 'if' condition");
837 /* parse into the 'then' branch */
838 if (!parser_next(parser)) {
839 parseerror(parser, "expected statement for on-true branch of 'if'");
843 ontrue = parser_parse_statement_or_block(parser);
848 /* check for an else */
849 if (!strcmp(parser_tokval(parser), "else")) {
850 /* parse into the 'else' branch */
851 if (!parser_next(parser)) {
852 parseerror(parser, "expected on-false branch after 'else'");
857 onfalse = parser_parse_statement_or_block(parser);
865 ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
866 *out = (ast_expression*)ifthen;
869 parseerror(parser, "Unexpected keyword");
872 else if (parser->tok == '{')
875 inner = parser_parse_block(parser);
878 *out = (ast_expression*)inner;
883 ast_expression *exp = parser_expression(parser);
891 static void parser_pop_local(parser_t *parser)
893 parser->locals_count--;
894 mem_d(parser->locals[parser->locals_count].name);
897 static ast_block* parser_parse_block(parser_t *parser)
899 size_t oldblocklocal;
900 ast_block *block = NULL;
902 oldblocklocal = parser->blocklocal;
903 parser->blocklocal = parser->locals_count;
905 if (!parser_next(parser)) { /* skip the '{' */
906 parseerror(parser, "expected function body");
910 block = ast_block_new(parser_ctx(parser));
912 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
914 ast_expression *expr;
915 if (parser->tok == '}')
918 if (!parser_parse_statement(parser, block, &expr)) {
919 ast_block_delete(block);
925 if (!ast_block_exprs_add(block, expr)) {
927 ast_block_delete(block);
933 if (parser->tok != '}') {
934 ast_block_delete(block);
937 (void)parser_next(parser);
941 parser->blocklocal = oldblocklocal;
942 /* unroll the local vector */
943 while (parser->locals_count > parser->blocklocal)
944 parser_pop_local(parser);
948 static ast_expression* parser_parse_statement_or_block(parser_t *parser)
950 ast_expression *expr;
951 if (parser->tok == '{')
952 return (ast_expression*)parser_parse_block(parser);
953 if (!parser_parse_statement(parser, NULL, &expr))
958 static bool parser_variable(parser_t *parser, ast_block *localblock)
961 ast_function *func = NULL;
966 int basetype = parser_token(parser)->constval.t;
970 if (!parser_next(parser)) { /* skip basetype or comma */
971 parseerror(parser, "expected variable declaration");
977 ctx = parser_ctx(parser);
978 var = parser_parse_type(parser, basetype, &isfunc);
983 if (parser->tok != TOKEN_IDENT) {
984 parseerror(parser, "expected variable name\n");
988 if (!localblock && parser_find_global(parser, parser_tokval(parser))) {
989 ast_value_delete(var);
990 parseerror(parser, "global already exists: %s\n", parser_tokval(parser));
994 if (localblock && parser_find_local(parser, parser_tokval(parser), parser->blocklocal)) {
995 ast_value_delete(var);
996 parseerror(parser, "local variable already exists: %s\n", parser_tokval(parser));
1000 if (!ast_value_set_name(var, parser_tokval(parser))) {
1001 parseerror(parser, "failed to set variable name\n");
1002 ast_value_delete(var);
1007 /* a function was defined */
1010 /* turn var into a value of TYPE_FUNCTION, with the old var
1013 fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1014 func = ast_function_new(ctx, var->name, fval);
1015 if (!fval || !func) {
1016 ast_value_delete(var);
1017 if (fval) ast_value_delete(fval);
1018 if (func) ast_function_delete(func);
1022 fval->expression.next = (ast_expression*)var;
1023 MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1025 if (!parser_t_functions_add(parser, func)) {
1026 ast_value_delete(var);
1027 if (fval) ast_value_delete(fval);
1028 if (func) ast_function_delete(func);
1035 varent.name = util_strdup(var->name);
1036 varent.var = (ast_expression*)var;
1037 if (var->expression.vtype == TYPE_VECTOR)
1039 size_t len = strlen(varent.name);
1040 varentry_t vx, vy, vz;
1041 vx.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 0);
1042 vy.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 1);
1043 vz.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 2);
1044 vx.name = mem_a(len+3);
1045 vy.name = mem_a(len+3);
1046 vz.name = mem_a(len+3);
1047 strcpy(vx.name, varent.name);
1048 strcpy(vy.name, varent.name);
1049 strcpy(vz.name, varent.name);
1050 vx.name[len] = vy.name[len] = vz.name[len] = '_';
1051 vx.name[len+1] = 'x';
1052 vy.name[len+1] = 'y';
1053 vz.name[len+1] = 'z';
1054 vx.name[len+2] = vy.name[len+2] = vz.name[len+2] = 0;
1057 (void)!parser_t_globals_add(parser, varent);
1058 (void)!parser_t_globals_add(parser, vx);
1059 (void)!parser_t_globals_add(parser, vy);
1060 (void)!parser_t_globals_add(parser, vz);
1062 (void)!parser_t_locals_add(parser, varent);
1063 (void)!parser_t_locals_add(parser, vx);
1064 (void)!parser_t_locals_add(parser, vy);
1065 (void)!parser_t_locals_add(parser, vz);
1070 if ( (!localblock && !parser_t_globals_add(parser, varent)) ||
1071 ( localblock && !parser_t_locals_add(parser, varent)) )
1073 ast_value_delete(var);
1077 if (localblock && !ast_block_locals_add(localblock, var))
1079 parser_pop_local(parser);
1080 ast_value_delete(var);
1084 if (!parser_next(parser)) {
1085 ast_value_delete(var);
1089 if (parser->tok == ';') {
1090 if (!parser_next(parser))
1091 return parser->tok == TOKEN_EOF;
1095 if (parser->tok == ',') {
1100 if (parser->tok != '=') {
1101 parseerror(parser, "expected '=' or ';'");
1105 if (!parser_next(parser))
1108 if (parser->tok == '#') {
1110 parseerror(parser, "cannot declare builtins within functions");
1113 if (!isfunc || !func) {
1114 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
1117 if (!parser_next(parser)) {
1118 parseerror(parser, "expected builtin number");
1121 if (parser->tok != TOKEN_INTCONST) {
1122 parseerror(parser, "builtin number must be an integer constant");
1125 if (parser_token(parser)->constval.i <= 0) {
1126 parseerror(parser, "builtin number must be positive integer greater than zero");
1130 func->builtin = -parser_token(parser)->constval.i;
1131 } else if (parser->tok == '{') {
1134 ast_function *old = parser->function;
1137 parseerror(parser, "cannot declare functions within functions");
1141 parser->function = func;
1142 block = parser_parse_block(parser);
1143 parser->function = old;
1148 if (!ast_function_blocks_add(func, block)) {
1149 ast_block_delete(block);
1154 parseerror(parser, "TODO, const assignment");
1157 if (!parser_next(parser))
1160 if (parser->tok == ',') {
1165 if (parser->tok != ';') {
1166 parseerror(parser, "expected semicolon");
1170 (void)parser_next(parser);
1176 static bool parser_do(parser_t *parser)
1178 if (parser->tok == TOKEN_TYPENAME)
1180 return parser_variable(parser, NULL);
1182 else if (parser->tok == TOKEN_KEYWORD)
1184 /* handle 'var' and 'const' */
1187 else if (parser->tok == '.')
1189 /* entity-member declaration */
1194 parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
1200 static parser_t *parser;
1204 parser = (parser_t*)mem_a(sizeof(parser_t));
1208 memset(parser, 0, sizeof(parser));
1210 MEM_VECTOR_INIT(parser, globals);
1211 MEM_VECTOR_INIT(parser, locals);
1215 bool parser_compile(const char *filename)
1217 parser->lex = lex_open(filename);
1219 printf("failed to open file \"%s\"\n", filename);
1223 /* initial lexer/parser state */
1224 parser->lex->flags.noops = true;
1226 if (parser_next(parser))
1228 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1230 if (!parser_do(parser)) {
1231 if (parser->tok == TOKEN_EOF)
1232 parseerror(parser, "unexpected eof");
1234 parseerror(parser, "parse error\n");
1235 lex_close(parser->lex);
1242 lex_close(parser->lex);
1244 return !parser->errors;
1247 void parser_cleanup()
1250 for (i = 0; i < parser->functions_count; ++i) {
1251 ast_delete(parser->functions[i]);
1253 for (i = 0; i < parser->imm_vector_count; ++i) {
1254 ast_delete(parser->imm_vector[i]);
1256 for (i = 0; i < parser->imm_string_count; ++i) {
1257 ast_delete(parser->imm_string[i]);
1259 for (i = 0; i < parser->imm_float_count; ++i) {
1260 ast_delete(parser->imm_float[i]);
1262 for (i = 0; i < parser->globals_count; ++i) {
1263 ast_delete(parser->globals[i].var);
1264 mem_d(parser->globals[i].name);
1266 MEM_VECTOR_CLEAR(parser, globals);
1271 bool parser_finish(const char *output)
1276 if (!parser->errors)
1278 ir = ir_builder_new("gmqcc_out");
1280 printf("failed to allocate builder\n");
1284 for (i = 0; i < parser->imm_float_count; ++i) {
1285 if (!ast_global_codegen(parser->imm_float[i], ir)) {
1286 printf("failed to generate global %s\n", parser->imm_float[i]->name);
1287 ir_builder_delete(ir);
1291 for (i = 0; i < parser->imm_string_count; ++i) {
1292 if (!ast_global_codegen(parser->imm_string[i], ir)) {
1293 printf("failed to generate global %s\n", parser->imm_string[i]->name);
1294 ir_builder_delete(ir);
1298 for (i = 0; i < parser->imm_vector_count; ++i) {
1299 if (!ast_global_codegen(parser->imm_vector[i], ir)) {
1300 printf("failed to generate global %s\n", parser->imm_vector[i]->name);
1301 ir_builder_delete(ir);
1305 for (i = 0; i < parser->globals_count; ++i) {
1306 if (!ast_istype(parser->globals[i].var, ast_value))
1308 if (!ast_global_codegen((ast_value*)(parser->globals[i].var), ir)) {
1309 printf("failed to generate global %s\n", parser->globals[i].name);
1310 ir_builder_delete(ir);
1314 for (i = 0; i < parser->functions_count; ++i) {
1315 if (!ast_function_codegen(parser->functions[i], ir)) {
1316 printf("failed to generate function %s\n", parser->functions[i]->name);
1317 ir_builder_delete(ir);
1320 if (!ir_function_finalize(parser->functions[i]->ir_func)) {
1321 printf("failed to finalize function %s\n", parser->functions[i]->name);
1322 ir_builder_delete(ir);
1327 ir_builder_dump(ir, printf);
1329 if (!ir_builder_generate(ir, output)) {
1330 printf("*** failed to generate output file\n");
1331 ir_builder_delete(ir);
1335 ir_builder_delete(ir);
1339 printf("*** there were compile errors\n");