parser.c

   1 #include <stdio.h>
   2 #include <stdarg.h>
   3
   4 #include "gmqcc.h"
   5 #include "lexer.h"
   6
   7 typedef struct {
   8     lex_file *lex;
   9     int      tok;
  10
  11     MEM_VECTOR_MAKE(ast_value*, globals);
  12     MEM_VECTOR_MAKE(ast_function*, functions);
  13     MEM_VECTOR_MAKE(ast_value*, imm_float);
  14
  15     ast_function *function;
  16     MEM_VECTOR_MAKE(ast_value*, locals);
  17     size_t blocklocal;
  18 } parser_t;
  19
  20 MEM_VEC_FUNCTIONS(parser_t, ast_value*, globals)
  21 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_float)
  22 MEM_VEC_FUNCTIONS(parser_t, ast_value*, locals)
  23 MEM_VEC_FUNCTIONS(parser_t, ast_function*, functions)
  24
  25 void parseerror(parser_t *parser, const char *fmt, ...)
  26 {
  27         va_list ap;
  28
  29     if (parser)
  30             printf("error %s:%lu: ", parser->lex->tok->ctx.file, (unsigned long)parser->lex->tok->ctx.line);
  31         else
  32             printf("error: ");
  33
  34         va_start(ap, fmt);
  35         vprintf(fmt, ap);
  36         va_end(ap);
  37
  38         printf("\n");
  39 }
  40
  41 bool parser_next(parser_t *parser)
  42 {
  43     /* lex_do kills the previous token */
  44     parser->tok = lex_do(parser->lex);
  45     if (parser->tok == TOKEN_EOF || parser->tok >= TOKEN_ERROR)
  46         return false;
  47     return true;
  48 }
  49
  50 /* lift a token out of the parser so it's not destroyed by parser_next */
  51 token *parser_lift(parser_t *parser)
  52 {
  53     token *tok = parser->lex->tok;
  54     parser->lex->tok = NULL;
  55     return tok;
  56 }
  57
  58 #define parser_tokval(p) (p->lex->tok->value)
  59 #define parser_token(p)  (p->lex->tok)
  60 #define parser_ctx(p)    (p->lex->tok->ctx)
  61
  62 ast_value* parser_const_float(parser_t *parser, double d)
  63 {
  64     size_t i;
  65     ast_value *out;
  66     for (i = 0; i < parser->imm_float_count; ++i) {
  67         if (parser->imm_float[i]->constval.vfloat == d)
  68             return parser->imm_float[i];
  69     }
  70     out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_FLOAT);
  71     out->isconst = true;
  72     out->constval.vfloat = d;
  73     if (!parser_t_imm_float_add(parser, out)) {
  74         ast_value_delete(out);
  75         return NULL;
  76     }
  77     return out;
  78 }
  79
  80 ast_value* parser_find_global(parser_t *parser, const char *name)
  81 {
  82     size_t i;
  83     for (i = 0; i < parser->globals_count; ++i) {
  84         if (!strcmp(parser->globals[i]->name, name))
  85             return parser->globals[i];
  86     }
  87     return NULL;
  88 }
  89
  90 ast_value* parser_find_local(parser_t *parser, const char *name, size_t upto)
  91 {
  92     size_t i;
  93     for (i = parser->locals_count; i > upto;) {
  94         --i;
  95         if (!strcmp(parser->locals[i]->name, name))
  96             return parser->locals[i];
  97     }
  98     return NULL;
  99 }
 100
 101 ast_value* parser_find_var(parser_t *parser, const char *name)
 102 {
 103     ast_value *v;
 104     v         = parser_find_local(parser, name, 0);
 105     if (!v) v = parser_find_global(parser, name);
 106     return v;
 107 }
 108
 109 typedef struct {
 110     MEM_VECTOR_MAKE(ast_value*, p);
 111 } paramlist_t;
 112 MEM_VEC_FUNCTIONS(paramlist_t, ast_value*, p)
 113
 114 static ast_value *parser_parse_type(parser_t *parser, int basetype, bool *isfunc)
 115 {
 116     paramlist_t params;
 117     ast_value *var;
 118     lex_ctx   ctx = parser_ctx(parser);
 119     int vtype = basetype;
 120     int temptype;
 121     size_t i;
 122
 123     MEM_VECTOR_INIT(&params, p);
 124
 125     *isfunc = false;
 126
 127     if (parser->tok == '(') {
 128         *isfunc = true;
 129         while (true) {
 130             ast_value *param;
 131             bool dummy;
 132
 133             if (!parser_next(parser))
 134                 goto on_error;
 135
 136             if (parser->tok == ')')
 137                 break;
 138
 139             temptype = parser_token(parser)->constval.t;
 140             if (!parser_next(parser))
 141                 goto on_error;
 142
 143             param = parser_parse_type(parser, temptype, &dummy);
 144             (void)dummy;
 145
 146             if (!param)
 147                 goto on_error;
 148
 149             if (parser->tok == TOKEN_IDENT) {
 150                 /* named parameter */
 151                 if (!ast_value_set_name(param, parser_tokval(parser)))
 152                     goto on_error;
 153                 if (!parser_next(parser))
 154                     goto on_error;
 155             }
 156
 157             if (!paramlist_t_p_add(&params, param)) {
 158                 parseerror(parser, "Out of memory while parsing typename");
 159                 goto on_error;
 160             }
 161
 162             if (parser->tok == ',')
 163                 continue;
 164             if (parser->tok == ')')
 165                 break;
 166             parseerror(parser, "Unexpected token");
 167             goto on_error;
 168         }
 169         if (!parser_next(parser))
 170             goto on_error;
 171     }
 172
 173     var = ast_value_new(ctx, "<unnamed>", vtype);
 174     if (!var)
 175         goto on_error;
 176     MEM_VECTOR_MOVE(&params, p, var, params);
 177     return var;
 178 on_error:
 179     for (i = 0; i < params.p_count; ++i)
 180         ast_value_delete(params.p[i]);
 181     MEM_VECTOR_CLEAR(&params, p);
 182     return NULL;
 183 }
 184
 185 typedef struct
 186 {
 187     size_t etype; /* 0 = expression, others are operators */
 188     int             paren;
 189     ast_expression *out;
 190     ast_value      *value; /* need to know if we can assign */
 191     lex_ctx ctx;
 192 } sy_elem;
 193 typedef struct
 194 {
 195     MEM_VECTOR_MAKE(sy_elem, out);
 196     MEM_VECTOR_MAKE(sy_elem, ops);
 197 } shunt;
 198 MEM_VEC_FUNCTIONS(shunt, sy_elem, out)
 199 MEM_VEC_FUNCTIONS(shunt, sy_elem, ops)
 200
 201 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
 202     sy_elem e;
 203     e.etype = 0;
 204     e.out   = v;
 205     e.value = NULL;
 206     e.ctx   = ctx;
 207     e.paren = 0;
 208     return e;
 209 }
 210 static sy_elem syval(lex_ctx ctx, ast_value *v) {
 211     sy_elem e;
 212     e.etype = 0;
 213     e.out   = (ast_expression*)v;
 214     e.value = v;
 215     e.ctx   = ctx;
 216     e.paren = 0;
 217     return e;
 218 }
 219
 220 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
 221     sy_elem e;
 222     e.etype = 1 + (op - operators);
 223     e.out   = NULL;
 224     e.value = NULL;
 225     e.ctx   = ctx;
 226     e.paren = 0;
 227     return e;
 228 }
 229
 230 static sy_elem syparen(lex_ctx ctx, int p) {
 231     sy_elem e;
 232     e.etype = 0;
 233     e.out   = NULL;
 234     e.value = NULL;
 235     e.ctx   = ctx;
 236     e.paren = p;
 237     return e;
 238 }
 239
 240 static bool parser_sy_pop(parser_t *parser, shunt *sy)
 241 {
 242     const oper_info *op;
 243     lex_ctx ctx;
 244     ast_expression *out = NULL;
 245     ast_expression *exprs[3];
 246     ast_value      *vars[3];
 247     size_t i;
 248
 249     if (!sy->ops_count) {
 250         parseerror(parser, "internal error: missing operator");
 251         return false;
 252     }
 253
 254     if (sy->ops[sy->ops_count-1].paren) {
 255         parseerror(parser, "unmatched parenthesis");
 256         return false;
 257     }
 258
 259     op = &operators[sy->ops[sy->ops_count-1].etype - 1];
 260     ctx = sy->ops[sy->ops_count-1].ctx;
 261
 262     if (sy->out_count < op->operands) {
 263         parseerror(parser, "internal error: not enough operands: %i", sy->out_count);
 264         return false;
 265     }
 266
 267     sy->ops_count--;
 268
 269     sy->out_count -= op->operands;
 270     for (i = 0; i < op->operands; ++i) {
 271         exprs[i] = sy->out[sy->out_count+i].out;
 272         vars[i]  = sy->out[sy->out_count+i].value;
 273     }
 274
 275     printf("Applying operator %s\n", op->op);
 276     switch (op->id)
 277     {
 278         default:
 279             parseerror(parser, "internal error: unhandled operand");
 280             return false;
 281
 282         case opid1('+'):
 283             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 284                 parseerror(parser, "Cannot add type %s and %s",
 285                            type_name[exprs[0]->expression.vtype],
 286                            type_name[exprs[1]->expression.vtype]);
 287                 return false;
 288             }
 289             switch (exprs[0]->expression.vtype) {
 290                 case TYPE_FLOAT:
 291                     out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
 292                     break;
 293                 case TYPE_VECTOR:
 294                     out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
 295                     break;
 296                 default:
 297                     parseerror(parser, "Cannot add type %s and %s",
 298                                type_name[exprs[0]->expression.vtype],
 299                                type_name[exprs[1]->expression.vtype]);
 300                     return false;
 301             };
 302             break;
 303         case opid1('-'):
 304             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 305                 parseerror(parser, "Cannot subtract type %s from %s",
 306                            type_name[exprs[1]->expression.vtype],
 307                            type_name[exprs[0]->expression.vtype]);
 308                 return false;
 309             }
 310             switch (exprs[0]->expression.vtype) {
 311                 case TYPE_FLOAT:
 312                     out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
 313                     break;
 314                 case TYPE_VECTOR:
 315                     out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
 316                     break;
 317                 default:
 318                     parseerror(parser, "Cannot add type %s from %s",
 319                                type_name[exprs[1]->expression.vtype],
 320                                type_name[exprs[0]->expression.vtype]);
 321                     return false;
 322             };
 323             break;
 324         case opid1('*'):
 325             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
 326                 exprs[0]->expression.vtype != TYPE_VECTOR &&
 327                 exprs[0]->expression.vtype != TYPE_FLOAT &&
 328                 exprs[1]->expression.vtype != TYPE_VECTOR &&
 329                 exprs[1]->expression.vtype != TYPE_FLOAT)
 330             {
 331                 parseerror(parser, "Cannot multiply type %s from %s",
 332                            type_name[exprs[1]->expression.vtype],
 333                            type_name[exprs[0]->expression.vtype]);
 334                 return false;
 335             }
 336             switch (exprs[0]->expression.vtype) {
 337                 case TYPE_FLOAT:
 338                     if (exprs[1]->expression.vtype == TYPE_VECTOR)
 339                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
 340                     else
 341                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
 342                     break;
 343                 case TYPE_VECTOR:
 344                     if (exprs[1]->expression.vtype == TYPE_FLOAT)
 345                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
 346                     else
 347                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
 348                     break;
 349                 default:
 350                     parseerror(parser, "Cannot add type %s from %s",
 351                                type_name[exprs[1]->expression.vtype],
 352                                type_name[exprs[0]->expression.vtype]);
 353                     return false;
 354             };
 355             break;
 356         case opid1('/'):
 357             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
 358                 exprs[0]->expression.vtype != TYPE_FLOAT)
 359             {
 360                 parseerror(parser, "Cannot divide types %s and %s",
 361                            type_name[exprs[0]->expression.vtype],
 362                            type_name[exprs[1]->expression.vtype]);
 363                 return false;
 364             }
 365             out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
 366             break;
 367
 368
 369         case opid1('='):
 370             if (!vars[0]) {
 371                 parseerror(parser, "Cannot assign to non-variable");
 372                 return false;
 373             }
 374             out = (ast_expression*)ast_store_new(ctx,
 375                                                  type_store_instr[vars[0]->expression.vtype],
 376                                                  vars[0], exprs[1]);
 377             break;
 378     }
 379
 380     if (!out) {
 381         parseerror(parser, "failed to apply operand %s", op->op);
 382         return false;
 383     }
 384
 385     sy->out[sy->out_count++] = syexp(ctx, out);
 386     return true;
 387 }
 388
 389 static ast_expression* parser_expression(parser_t *parser)
 390 {
 391     ast_expression *expr = NULL;
 392     shunt sy;
 393     bool wantop = false;
 394
 395     MEM_VECTOR_INIT(&sy, out);
 396     MEM_VECTOR_INIT(&sy, ops);
 397
 398     while (true)
 399     {
 400         if (!wantop)
 401         {
 402             bool nextwant = true;
 403             if (parser->tok == TOKEN_IDENT)
 404             {
 405                 /* variable */
 406                 ast_value *var = parser_find_var(parser, parser_tokval(parser));
 407                 if (!var) {
 408                     parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
 409                     goto onerr;
 410                 }
 411                 if (!shunt_out_add(&sy, syval(parser_ctx(parser), var))) {
 412                     parseerror(parser, "out of memory");
 413                     goto onerr;
 414                 }
 415                 printf("Added: %s\n", var->name);
 416             }
 417             else if (parser->tok == TOKEN_FLOATCONST) {
 418                 ast_value *val = parser_const_float(parser, (parser_token(parser)->constval.f));
 419                 if (!val)
 420                     return false;
 421                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 422                     parseerror(parser, "out of memory");
 423                     goto onerr;
 424                 }
 425             }
 426             else if (parser->tok == TOKEN_INTCONST) {
 427                 ast_value *val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
 428                 if (!val)
 429                     return false;
 430                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 431                     parseerror(parser, "out of memory");
 432                     goto onerr;
 433                 }
 434                 printf("Added: %i\n", parser_token(parser)->constval.i);
 435             }
 436             else if (parser->tok == '(') {
 437                 nextwant = false; /* not expecting an operator next */
 438                 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1))) {
 439                     parseerror(parser, "out of memory");
 440                     goto onerr;
 441                 }
 442             }
 443             else {
 444                 /* TODO: prefix operators */
 445                 parseerror(parser, "expected statement");
 446                 goto onerr;
 447             }
 448             wantop = nextwant;
 449             parser->lex->flags.noops = !wantop;
 450         } else {
 451             if (parser->tok == ')') {
 452                 /* we do expect an operator next */
 453                 /* closing an opening paren */
 454                 printf("Applying closing paren\n");
 455                 if (!sy.ops_count) {
 456                     parseerror(parser, "unmatched closing paren");
 457                     goto onerr;
 458                 }
 459                 if (sy.ops[sy.ops_count-1].paren == 1) {
 460                     parseerror(parser, "empty parenthesis expression");
 461                     goto onerr;
 462                 }
 463                 while (sy.ops_count) {
 464                     if (sy.ops[sy.ops_count-1].paren == 1) {
 465                         sy.ops_count--;
 466                         break;
 467                     }
 468                     if (!parser_sy_pop(parser, &sy))
 469                         goto onerr;
 470                 }
 471             }
 472             else if (parser->tok != TOKEN_OPERATOR) {
 473                 parseerror(parser, "expected operator or end of statement");
 474                 goto onerr;
 475             }
 476             else {
 477                 /* classify the operator */
 478                 /* TODO: suffix operators */
 479                 const oper_info *op;
 480                 const oper_info *olast = NULL;
 481                 size_t o;
 482                 for (o = 0; o < operator_count; ++o) {
 483                     if (!(operators[o].flags & OP_PREFIX) &&
 484                         !(operators[o].flags & OP_SUFFIX) && /* remove this */
 485                         !strcmp(parser_tokval(parser), operators[o].op))
 486                     {
 487                         break;
 488                     }
 489                 }
 490                 if (o == operator_count) {
 491                     /* no operator found... must be the end of the statement */
 492                     break;
 493                 }
 494                 /* found an operator */
 495                 op = &operators[o];
 496
 497                 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 498                     olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 499
 500                 while (olast && (
 501                         (op->prec < olast->prec) ||
 502                         (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
 503                 {
 504                     if (!parser_sy_pop(parser, &sy))
 505                         goto onerr;
 506                     if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 507                         olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 508                 }
 509
 510                 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
 511                     goto onerr;
 512                 printf("Added op %s\n", op->op);
 513             }
 514             wantop = false;
 515             parser->lex->flags.noops = true;
 516         }
 517         if (!parser_next(parser)) {
 518             goto onerr;
 519         }
 520         if (parser->tok == ';') {
 521             printf("End of statement\n");
 522             break;
 523         }
 524     }
 525     if (!parser_next(parser)) {
 526         parseerror(parser, "Unexpected end of file");
 527         goto onerr;
 528     }
 529
 530     while (sy.ops_count) {
 531         if (!parser_sy_pop(parser, &sy))
 532             goto onerr;
 533     }
 534
 535     parser->lex->flags.noops = true;
 536     if (!sy.out_count) {
 537         parseerror(parser, "empty expression");
 538         expr = NULL;
 539     } else
 540         expr = sy.out[0].out;
 541     MEM_VECTOR_CLEAR(&sy, out);
 542     MEM_VECTOR_CLEAR(&sy, ops);
 543     return expr;
 544
 545 onerr:
 546     parser->lex->flags.noops = true;
 547     MEM_VECTOR_CLEAR(&sy, out);
 548     MEM_VECTOR_CLEAR(&sy, ops);
 549     return NULL;
 550 }
 551
 552 static bool parser_variable(parser_t *parser, ast_block *localblock);
 553 static bool parser_body_do(parser_t *parser, ast_block *block)
 554 {
 555     if (parser->tok == TOKEN_TYPENAME)
 556     {
 557         /* local variable */
 558         if (!parser_variable(parser, block))
 559             return false;
 560         return true;
 561     }
 562     else if (parser->tok == TOKEN_KEYWORD)
 563     {
 564         if (!strcmp(parser_tokval(parser), "return"))
 565         {
 566             ast_expression *exp;
 567             ast_return *ret;
 568
 569             if (!parser_next(parser)) {
 570                 parseerror(parser, "expected return expression");
 571                 return false;
 572             }
 573
 574             exp = parser_expression(parser);
 575             if (!exp)
 576                 return false;
 577             ret = ast_return_new(exp->expression.node.context, exp);
 578             if (!ret) {
 579                 ast_delete(exp);
 580                 return false;
 581             }
 582             if (!ast_block_exprs_add(block, (ast_expression*)ret)) {
 583                 ast_delete(ret);
 584                 return false;
 585             }
 586             return true;
 587         }
 588         parseerror(parser, "Unexpected keyword");
 589         return false;
 590     }
 591     else if (parser->tok == '{')
 592     {
 593         /* a block */
 594         parseerror(parser, "TODO: inner blocks: %s", parser_tokval(parser));
 595         return false;
 596     }
 597     else
 598     {
 599         ast_expression *exp = parser_expression(parser);
 600         if (!exp)
 601             return false;
 602         if (!ast_block_exprs_add(block, exp)) {
 603             ast_delete(exp);
 604             return false;
 605         }
 606         return true;
 607     }
 608 }
 609
 610 static ast_block* parser_parse_block(parser_t *parser)
 611 {
 612     size_t oldblocklocal;
 613     ast_block *block = NULL;
 614
 615     oldblocklocal = parser->blocklocal;
 616     parser->blocklocal = parser->locals_count;
 617
 618     if (!parser_next(parser)) { /* skip the '{' */
 619         parseerror(parser, "expected function body");
 620         goto cleanup;
 621     }
 622
 623     block = ast_block_new(parser_ctx(parser));
 624
 625     while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
 626     {
 627         if (parser->tok == '}')
 628             break;
 629
 630         if (!parser_body_do(parser, block)) {
 631             ast_block_delete(block);
 632             block = NULL;
 633             goto cleanup;
 634         }
 635     }
 636
 637     if (parser->tok != '}') {
 638         ast_block_delete(block);
 639         block = NULL;
 640     } else {
 641         (void)parser_next(parser);
 642     }
 643
 644 cleanup:
 645     parser->blocklocal = oldblocklocal;
 646     return block;
 647 }
 648
 649 static bool parser_variable(parser_t *parser, ast_block *localblock)
 650 {
 651     bool          isfunc = false;
 652     ast_function *func = NULL;
 653     lex_ctx       ctx;
 654     ast_value    *var;
 655
 656     int basetype = parser_token(parser)->constval.t;
 657
 658     while (true)
 659     {
 660         if (!parser_next(parser)) { /* skip basetype or comma */
 661             parseerror(parser, "expected variable declaration");
 662             return false;
 663         }
 664
 665         isfunc = false;
 666         func = NULL;
 667         ctx = parser_ctx(parser);
 668         var = parser_parse_type(parser, basetype, &isfunc);
 669
 670         if (!var)
 671             return false;
 672
 673         if (parser->tok != TOKEN_IDENT) {
 674             parseerror(parser, "expected variable name\n");
 675             return false;
 676         }
 677
 678         if (!localblock && parser_find_global(parser, parser_tokval(parser))) {
 679             ast_value_delete(var);
 680             parseerror(parser, "global already exists: %s\n", parser_tokval(parser));
 681             return false;
 682         }
 683
 684         if (localblock && parser_find_local(parser, parser_tokval(parser), parser->blocklocal)) {
 685             ast_value_delete(var);
 686             parseerror(parser, "local variable already exists: %s\n", parser_tokval(parser));
 687             return false;
 688         }
 689
 690         if (!ast_value_set_name(var, parser_tokval(parser))) {
 691             parseerror(parser, "failed to set variable name\n");
 692             ast_value_delete(var);
 693             return false;
 694         }
 695
 696         if (isfunc) {
 697             /* a function was defined */
 698             ast_value *fval;
 699
 700             /* turn var into a value of TYPE_FUNCTION, with the old var
 701              * as return type
 702              */
 703             fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
 704             func = ast_function_new(ctx, var->name, fval);
 705             if (!fval || !func) {
 706                 ast_value_delete(var);
 707                 if (fval) ast_value_delete(fval);
 708                 if (func) ast_function_delete(func);
 709                 return false;
 710             }
 711
 712             fval->expression.next = (ast_expression*)var;
 713             MEM_VECTOR_MOVE(var, params, fval, params);
 714
 715             if (!parser_t_functions_add(parser, func)) {
 716                 ast_value_delete(var);
 717                 if (fval) ast_value_delete(fval);
 718                 if (func) ast_function_delete(func);
 719                 return false;
 720             }
 721
 722             var = fval;
 723         }
 724
 725         if ( (!localblock && !parser_t_globals_add(parser, var)) ||
 726              ( localblock && !parser_t_locals_add(parser, var)) )
 727         {
 728             ast_value_delete(var);
 729             return false;
 730         }
 731         if (localblock && !ast_block_locals_add(localblock, var))
 732         {
 733             parser->locals_count--;
 734             ast_value_delete(var);
 735             return false;
 736         }
 737
 738         if (!parser_next(parser)) {
 739             ast_value_delete(var);
 740             return false;
 741         }
 742
 743         if (parser->tok == ';') {
 744             if (!parser_next(parser))
 745                 return parser->tok == TOKEN_EOF;
 746             return true;
 747         }
 748
 749         if (parser->tok == ',') {
 750             /* another var */
 751             continue;
 752         }
 753
 754         if (parser->tok != '=') {
 755             parseerror(parser, "expected '=' or ';'");
 756             return false;
 757         }
 758
 759         if (!parser_next(parser))
 760             return false;
 761
 762         if (parser->tok == '#') {
 763             if (localblock) {
 764                 parseerror(parser, "cannot declare builtins within functions");
 765                 return false;
 766             }
 767             if (!isfunc || !func) {
 768                 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
 769                 return false;
 770             }
 771             if (!parser_next(parser)) {
 772                 parseerror(parser, "expected builtin number");
 773                 return false;
 774             }
 775             if (parser->tok != TOKEN_INTCONST) {
 776                 parseerror(parser, "builtin number must be an integer constant");
 777                 return false;
 778             }
 779             if (parser_token(parser)->constval.i <= 0) {
 780                 parseerror(parser, "builtin number must be positive integer greater than zero");
 781                 return false;
 782             }
 783
 784             func->builtin = -parser_token(parser)->constval.i;
 785         } else if (parser->tok == '{') {
 786             /* function body */
 787             ast_block *block;
 788             ast_function *old = parser->function;
 789
 790             if (localblock) {
 791                 parseerror(parser, "cannot declare functions within functions");
 792                 return false;
 793             }
 794
 795             parser->function = func;
 796             block = parser_parse_block(parser);
 797             parser->function = old;
 798
 799             if (!block)
 800                 return false;
 801
 802             if (!ast_function_blocks_add(func, block)) {
 803                 ast_block_delete(block);
 804                 return false;
 805             }
 806             return true;
 807         } else {
 808             parseerror(parser, "TODO, const assignment");
 809         }
 810
 811         if (!parser_next(parser))
 812             return false;
 813
 814         if (parser->tok == ',') {
 815             /* another */
 816             continue;
 817         }
 818
 819         if (parser->tok != ';') {
 820             parseerror(parser, "expected semicolon");
 821             return false;
 822         }
 823
 824         (void)parser_next(parser);
 825
 826         return true;
 827     }
 828 }
 829
 830 static bool parser_do(parser_t *parser)
 831 {
 832     if (parser->tok == TOKEN_TYPENAME)
 833     {
 834         return parser_variable(parser, NULL);
 835     }
 836     else if (parser->tok == TOKEN_KEYWORD)
 837     {
 838         /* handle 'var' and 'const' */
 839         return false;
 840     }
 841     else if (parser->tok == '.')
 842     {
 843         /* entity-member declaration */
 844         return false;
 845     }
 846     else
 847     {
 848         parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
 849         return false;
 850     }
 851     return true;
 852 }
 853
 854 bool parser_compile(const char *filename)
 855 {
 856     size_t i;
 857     parser_t *parser;
 858     ir_builder *ir;
 859
 860     parser = (parser_t*)mem_a(sizeof(parser_t));
 861     if (!parser)
 862         return false;
 863
 864     memset(parser, 0, sizeof(parser));
 865
 866     MEM_VECTOR_INIT(parser, globals);
 867     MEM_VECTOR_INIT(parser, locals);
 868     parser->lex = lex_open(filename);
 869
 870     if (!parser->lex) {
 871         printf("failed to open file \"%s\"\n", filename);
 872         return false;
 873     }
 874
 875     /* initial lexer/parser state */
 876     parser->lex->flags.noops = true;
 877
 878     if (parser_next(parser))
 879     {
 880         while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
 881         {
 882             if (!parser_do(parser)) {
 883                 if (parser->tok == TOKEN_EOF)
 884                     parseerror(parser, "unexpected eof");
 885                 else
 886                     parseerror(parser, "parse error\n");
 887                 lex_close(parser->lex);
 888                 mem_d(parser);
 889                 return false;
 890             }
 891         }
 892     }
 893
 894     lex_close(parser->lex);
 895
 896     ir = ir_builder_new("gmqcc_out");
 897     if (!ir) {
 898         printf("failed to allocate builder\n");
 899         goto cleanup;
 900     }
 901
 902     for (i = 0; i < parser->imm_float_count; ++i) {
 903         if (!ast_global_codegen(parser->imm_float[i], ir)) {
 904             printf("failed to generate global %s\n", parser->imm_float[i]->name);
 905         }
 906     }
 907     for (i = 0; i < parser->globals_count; ++i) {
 908         if (!ast_global_codegen(parser->globals[i], ir)) {
 909             printf("failed to generate global %s\n", parser->globals[i]->name);
 910         }
 911     }
 912     for (i = 0; i < parser->functions_count; ++i) {
 913         if (!ast_function_codegen(parser->functions[i], ir)) {
 914             printf("failed to generate function %s\n", parser->functions[i]->name);
 915         }
 916         if (!ir_function_finalize(parser->functions[i]->ir_func)) {
 917             printf("failed to finalize function %s\n", parser->functions[i]->name);
 918         }
 919     }
 920
 921     ir_builder_dump(ir, printf);
 922
 923 cleanup:
 924     for (i = 0; i < parser->globals_count; ++i) {
 925         ast_value_delete(parser->globals[i]);
 926     }
 927     MEM_VECTOR_CLEAR(parser, globals);
 928
 929     mem_d(parser);
 930     return true;
 931 }