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     ast_value *fun;
  94     for (i = parser->locals_count; i > upto;) {
  95         --i;
  96         if (!strcmp(parser->locals[i]->name, name))
  97             return parser->locals[i];
  98     }
  99     fun = parser->function->vtype;
 100     for (i = 0; i < fun->expression.params_count; ++i) {
 101         if (!strcmp(fun->expression.params[i]->name, name))
 102             return fun->expression.params[i];
 103     }
 104     return NULL;
 105 }
 106
 107 ast_value* parser_find_var(parser_t *parser, const char *name)
 108 {
 109     ast_value *v;
 110     v         = parser_find_local(parser, name, 0);
 111     if (!v) v = parser_find_global(parser, name);
 112     return v;
 113 }
 114
 115 typedef struct {
 116     MEM_VECTOR_MAKE(ast_value*, p);
 117 } paramlist_t;
 118 MEM_VEC_FUNCTIONS(paramlist_t, ast_value*, p)
 119
 120 static ast_value *parser_parse_type(parser_t *parser, int basetype, bool *isfunc)
 121 {
 122     paramlist_t params;
 123     ast_value *var;
 124     lex_ctx   ctx = parser_ctx(parser);
 125     int vtype = basetype;
 126     int temptype;
 127     size_t i;
 128
 129     MEM_VECTOR_INIT(&params, p);
 130
 131     *isfunc = false;
 132
 133     if (parser->tok == '(') {
 134         *isfunc = true;
 135         while (true) {
 136             ast_value *param;
 137             bool dummy;
 138
 139             if (!parser_next(parser))
 140                 goto on_error;
 141
 142             if (parser->tok == ')')
 143                 break;
 144
 145             temptype = parser_token(parser)->constval.t;
 146             if (!parser_next(parser))
 147                 goto on_error;
 148
 149             param = parser_parse_type(parser, temptype, &dummy);
 150             (void)dummy;
 151
 152             if (!param)
 153                 goto on_error;
 154
 155             if (parser->tok == TOKEN_IDENT) {
 156                 /* named parameter */
 157                 if (!ast_value_set_name(param, parser_tokval(parser)))
 158                     goto on_error;
 159                 if (!parser_next(parser))
 160                     goto on_error;
 161             }
 162
 163             if (!paramlist_t_p_add(&params, param)) {
 164                 parseerror(parser, "Out of memory while parsing typename");
 165                 goto on_error;
 166             }
 167
 168             if (parser->tok == ',')
 169                 continue;
 170             if (parser->tok == ')')
 171                 break;
 172             parseerror(parser, "Unexpected token");
 173             goto on_error;
 174         }
 175         if (!parser_next(parser))
 176             goto on_error;
 177     }
 178
 179     var = ast_value_new(ctx, "<unnamed>", vtype);
 180     if (!var)
 181         goto on_error;
 182     MEM_VECTOR_MOVE(&params, p, &var->expression, params);
 183     return var;
 184 on_error:
 185     for (i = 0; i < params.p_count; ++i)
 186         ast_value_delete(params.p[i]);
 187     MEM_VECTOR_CLEAR(&params, p);
 188     return NULL;
 189 }
 190
 191 typedef struct
 192 {
 193     size_t etype; /* 0 = expression, others are operators */
 194     int             paren;
 195     size_t          off;
 196     ast_expression *out;
 197     ast_value      *value; /* need to know if we can assign */
 198     ast_block      *block; /* for commas and function calls */
 199     lex_ctx ctx;
 200 } sy_elem;
 201 typedef struct
 202 {
 203     MEM_VECTOR_MAKE(sy_elem, out);
 204     MEM_VECTOR_MAKE(sy_elem, ops);
 205 } shunt;
 206 MEM_VEC_FUNCTIONS(shunt, sy_elem, out)
 207 MEM_VEC_FUNCTIONS(shunt, sy_elem, ops)
 208
 209 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
 210     sy_elem e;
 211     e.etype = 0;
 212     e.out   = v;
 213     e.value = NULL;
 214     e.block = NULL;
 215     e.ctx   = ctx;
 216     e.paren = 0;
 217     return e;
 218 }
 219 static sy_elem syval(lex_ctx ctx, ast_value *v) {
 220     sy_elem e;
 221     e.etype = 0;
 222     e.out   = (ast_expression*)v;
 223     e.value = v;
 224     e.block = NULL;
 225     e.ctx   = ctx;
 226     e.paren = 0;
 227     return e;
 228 }
 229
 230 static sy_elem syblock(lex_ctx ctx, ast_block *v) {
 231     sy_elem e;
 232     e.etype = 0;
 233     e.out   = (ast_expression*)v;
 234     e.value = NULL;
 235     e.block = v;
 236     e.ctx   = ctx;
 237     e.paren = 0;
 238     return e;
 239 }
 240
 241 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
 242     sy_elem e;
 243     e.etype = 1 + (op - operators);
 244     e.out   = NULL;
 245     e.value = NULL;
 246     e.block = NULL;
 247     e.ctx   = ctx;
 248     e.paren = 0;
 249     return e;
 250 }
 251
 252 static sy_elem syparen(lex_ctx ctx, int p, size_t off) {
 253     sy_elem e;
 254     e.etype = 0;
 255     e.off   = off;
 256     e.out   = NULL;
 257     e.value = NULL;
 258     e.block = NULL;
 259     e.ctx   = ctx;
 260     e.paren = p;
 261     return e;
 262 }
 263
 264 static bool parser_sy_pop(parser_t *parser, shunt *sy)
 265 {
 266     const oper_info *op;
 267     lex_ctx ctx;
 268     ast_expression *out = NULL;
 269     ast_expression *exprs[3];
 270     ast_value      *vars[3];
 271     ast_block      *blocks[3];
 272     size_t i;
 273
 274     if (!sy->ops_count) {
 275         parseerror(parser, "internal error: missing operator");
 276         return false;
 277     }
 278
 279     if (sy->ops[sy->ops_count-1].paren) {
 280         parseerror(parser, "unmatched parenthesis");
 281         return false;
 282     }
 283
 284     op = &operators[sy->ops[sy->ops_count-1].etype - 1];
 285     ctx = sy->ops[sy->ops_count-1].ctx;
 286
 287     if (sy->out_count < op->operands) {
 288         parseerror(parser, "internal error: not enough operands: %i", sy->out_count);
 289         return false;
 290     }
 291
 292     sy->ops_count--;
 293
 294     sy->out_count -= op->operands;
 295     for (i = 0; i < op->operands; ++i) {
 296         exprs[i]  = sy->out[sy->out_count+i].out;
 297         vars[i]   = sy->out[sy->out_count+i].value;
 298         blocks[i] = sy->out[sy->out_count+i].block;
 299     }
 300
 301     if (blocks[0] && !blocks[0]->exprs_count && op->id != opid1(',')) {
 302         parseerror(parser, "internal error: operator cannot be applied on empty blocks");
 303         return false;
 304     }
 305
 306     switch (op->id)
 307     {
 308         default:
 309             parseerror(parser, "internal error: unhandled operand");
 310             return false;
 311
 312         case opid1(','):
 313             if (blocks[0]) {
 314                 if (!ast_block_exprs_add(blocks[0], exprs[1]))
 315                     return false;
 316             } else {
 317                 blocks[0] = ast_block_new(ctx);
 318                 if (!ast_block_exprs_add(blocks[0], exprs[0]) ||
 319                     !ast_block_exprs_add(blocks[0], exprs[1]))
 320                 {
 321                     return false;
 322                 }
 323             }
 324             if (!ast_block_set_type(blocks[0], exprs[1]))
 325                 return false;
 326
 327             sy->out[sy->out_count++] = syblock(ctx, blocks[0]);
 328             return true;
 329
 330         case opid1('+'):
 331             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 332                 parseerror(parser, "Cannot add type %s and %s",
 333                            type_name[exprs[0]->expression.vtype],
 334                            type_name[exprs[1]->expression.vtype]);
 335                 return false;
 336             }
 337             switch (exprs[0]->expression.vtype) {
 338                 case TYPE_FLOAT:
 339                     out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
 340                     break;
 341                 case TYPE_VECTOR:
 342                     out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
 343                     break;
 344                 default:
 345                     parseerror(parser, "Cannot add type %s and %s",
 346                                type_name[exprs[0]->expression.vtype],
 347                                type_name[exprs[1]->expression.vtype]);
 348                     return false;
 349             };
 350             break;
 351         case opid1('-'):
 352             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 353                 parseerror(parser, "Cannot subtract type %s from %s",
 354                            type_name[exprs[1]->expression.vtype],
 355                            type_name[exprs[0]->expression.vtype]);
 356                 return false;
 357             }
 358             switch (exprs[0]->expression.vtype) {
 359                 case TYPE_FLOAT:
 360                     out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
 361                     break;
 362                 case TYPE_VECTOR:
 363                     out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
 364                     break;
 365                 default:
 366                     parseerror(parser, "Cannot add type %s from %s",
 367                                type_name[exprs[1]->expression.vtype],
 368                                type_name[exprs[0]->expression.vtype]);
 369                     return false;
 370             };
 371             break;
 372         case opid1('*'):
 373             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
 374                 exprs[0]->expression.vtype != TYPE_VECTOR &&
 375                 exprs[0]->expression.vtype != TYPE_FLOAT &&
 376                 exprs[1]->expression.vtype != TYPE_VECTOR &&
 377                 exprs[1]->expression.vtype != TYPE_FLOAT)
 378             {
 379                 parseerror(parser, "Cannot multiply type %s from %s",
 380                            type_name[exprs[1]->expression.vtype],
 381                            type_name[exprs[0]->expression.vtype]);
 382                 return false;
 383             }
 384             switch (exprs[0]->expression.vtype) {
 385                 case TYPE_FLOAT:
 386                     if (exprs[1]->expression.vtype == TYPE_VECTOR)
 387                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
 388                     else
 389                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
 390                     break;
 391                 case TYPE_VECTOR:
 392                     if (exprs[1]->expression.vtype == TYPE_FLOAT)
 393                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
 394                     else
 395                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
 396                     break;
 397                 default:
 398                     parseerror(parser, "Cannot add type %s from %s",
 399                                type_name[exprs[1]->expression.vtype],
 400                                type_name[exprs[0]->expression.vtype]);
 401                     return false;
 402             };
 403             break;
 404         case opid1('/'):
 405             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
 406                 exprs[0]->expression.vtype != TYPE_FLOAT)
 407             {
 408                 parseerror(parser, "Cannot divide types %s and %s",
 409                            type_name[exprs[0]->expression.vtype],
 410                            type_name[exprs[1]->expression.vtype]);
 411                 return false;
 412             }
 413             out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
 414             break;
 415
 416
 417         case opid1('='):
 418             if (!vars[0]) {
 419                 parseerror(parser, "Cannot assign to non-variable");
 420                 return false;
 421             }
 422             out = (ast_expression*)ast_store_new(ctx,
 423                                                  type_store_instr[vars[0]->expression.vtype],
 424                                                  vars[0], exprs[1]);
 425             break;
 426     }
 427
 428     if (!out) {
 429         parseerror(parser, "failed to apply operand %s", op->op);
 430         return false;
 431     }
 432
 433     sy->out[sy->out_count++] = syexp(ctx, out);
 434     return true;
 435 }
 436
 437 static bool parser_close_call(parser_t *parser, shunt *sy)
 438 {
 439     /* was a function call */
 440     ast_expression *fun;
 441     ast_call       *call;
 442
 443     size_t          fid;
 444     size_t          paramcount;
 445
 446     sy->ops_count--;
 447     fid = sy->ops[sy->ops_count].off;
 448
 449     /* out[fid] is the function
 450      * everything above is parameters...
 451      * 0 params = nothing
 452      * 1 params = ast_expression
 453      * more = ast_block
 454      */
 455
 456     if (sy->out_count < 1 || sy->out_count <= fid) {
 457         parseerror(parser, "internal error: function call needs function and parameter list...");
 458         return false;
 459     }
 460
 461     fun = sy->out[fid].out;
 462
 463     call = ast_call_new(sy->ops[sy->ops_count].ctx, fun);
 464     if (!call) {
 465         parseerror(parser, "out of memory");
 466         return false;
 467     }
 468
 469     printf("fid = %i, out_count = %i\n", (int)fid, (int)sy->out_count);
 470
 471     if (fid+1 == sy->out_count) {
 472         /* no arguments */
 473         paramcount = 0;
 474     } else if (fid+2 == sy->out_count) {
 475         ast_block *params;
 476         sy->out_count--;
 477         params = sy->out[sy->out_count].block;
 478         if (!params) {
 479             /* 1 param */
 480             paramcount = 1;
 481             if (!ast_call_params_add(call, sy->out[sy->out_count].out)) {
 482                 ast_delete(sy->out[sy->out_count].out);
 483                 parseerror(parser, "out of memory");
 484                 return false;
 485             }
 486         } else {
 487             paramcount = params->exprs_count;
 488             MEM_VECTOR_MOVE(params, exprs, call, params);
 489             ast_delete(params);
 490         }
 491     } else {
 492         parseerror(parser, "invalid function call");
 493         return false;
 494     }
 495
 496     /* overwrite fid, the function, with a call */
 497     sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
 498
 499     if (fun->expression.vtype != TYPE_FUNCTION) {
 500         parseerror(parser, "not a function");
 501         return false;
 502     }
 503
 504     if (!fun->expression.next) {
 505         parseerror(parser, "could not determine function return type");
 506         return false;
 507     } else {
 508         if (fun->expression.params_count != paramcount) {
 509             parseerror(parser, "expected %i parameters, got %i", (int)fun->expression.params_count, paramcount);
 510             return false;
 511         }
 512     }
 513
 514     return true;
 515 }
 516
 517 static bool parser_close_paren(parser_t *parser, shunt *sy, bool functions_only)
 518 {
 519     if (!sy->ops_count) {
 520         parseerror(parser, "unmatched closing paren");
 521         return false;
 522     }
 523     if (sy->ops[sy->ops_count-1].paren == 1) {
 524         parseerror(parser, "empty parenthesis expression");
 525         return false;
 526     }
 527     while (sy->ops_count) {
 528         if (sy->ops[sy->ops_count-1].paren == 'f') {
 529             if (!parser_close_call(parser, sy))
 530                 return false;
 531             break;
 532         }
 533         if (sy->ops[sy->ops_count-1].paren == 1) {
 534             sy->ops_count--;
 535             return !functions_only;
 536         }
 537         if (!parser_sy_pop(parser, sy))
 538             return false;
 539     }
 540     return true;
 541 }
 542
 543 static ast_expression* parser_expression(parser_t *parser)
 544 {
 545     ast_expression *expr = NULL;
 546     shunt sy;
 547     bool wantop = false;
 548
 549     MEM_VECTOR_INIT(&sy, out);
 550     MEM_VECTOR_INIT(&sy, ops);
 551
 552     while (true)
 553     {
 554         if (!wantop)
 555         {
 556             bool nextwant = true;
 557             if (parser->tok == TOKEN_IDENT)
 558             {
 559                 /* variable */
 560                 ast_value *var = parser_find_var(parser, parser_tokval(parser));
 561                 if (!var) {
 562                     parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
 563                     goto onerr;
 564                 }
 565                 if (!shunt_out_add(&sy, syval(parser_ctx(parser), var))) {
 566                     parseerror(parser, "out of memory");
 567                     goto onerr;
 568                 }
 569             }
 570             else if (parser->tok == TOKEN_FLOATCONST) {
 571                 ast_value *val = parser_const_float(parser, (parser_token(parser)->constval.f));
 572                 if (!val)
 573                     return false;
 574                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 575                     parseerror(parser, "out of memory");
 576                     goto onerr;
 577                 }
 578             }
 579             else if (parser->tok == TOKEN_INTCONST) {
 580                 ast_value *val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
 581                 if (!val)
 582                     return false;
 583                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 584                     parseerror(parser, "out of memory");
 585                     goto onerr;
 586                 }
 587             }
 588             else if (parser->tok == '(') {
 589                 nextwant = false; /* not expecting an operator next */
 590                 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1, 0))) {
 591                     parseerror(parser, "out of memory");
 592                     goto onerr;
 593                 }
 594             }
 595             else if (parser->tok == ')') {
 596                 /* allowed for function calls */
 597                 if (!parser_close_paren(parser, &sy, true))
 598                     goto onerr;
 599             }
 600             else {
 601                 /* TODO: prefix operators */
 602                 parseerror(parser, "expected statement");
 603                 goto onerr;
 604             }
 605             wantop = nextwant;
 606             parser->lex->flags.noops = !wantop;
 607         } else {
 608             if (parser->tok == '(') {
 609                 /* we expected an operator, this is the function-call operator */
 610                 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 'f', sy.out_count-1))) {
 611                     parseerror(parser, "out of memory");
 612                     goto onerr;
 613                 }
 614             }
 615             else if (parser->tok == ')') {
 616                 /* we do expect an operator next */
 617                 /* closing an opening paren */
 618                 if (!parser_close_paren(parser, &sy, false))
 619                     goto onerr;
 620             }
 621             else if (parser->tok != TOKEN_OPERATOR) {
 622                 parseerror(parser, "expected operator or end of statement");
 623                 goto onerr;
 624             }
 625             else {
 626                 /* classify the operator */
 627                 /* TODO: suffix operators */
 628                 const oper_info *op;
 629                 const oper_info *olast = NULL;
 630                 size_t o;
 631                 for (o = 0; o < operator_count; ++o) {
 632                     if (!(operators[o].flags & OP_PREFIX) &&
 633                         !(operators[o].flags & OP_SUFFIX) && /* remove this */
 634                         !strcmp(parser_tokval(parser), operators[o].op))
 635                     {
 636                         break;
 637                     }
 638                 }
 639                 if (o == operator_count) {
 640                     /* no operator found... must be the end of the statement */
 641                     break;
 642                 }
 643                 /* found an operator */
 644                 op = &operators[o];
 645
 646                 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 647                     olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 648
 649                 while (olast && (
 650                         (op->prec < olast->prec) ||
 651                         (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
 652                 {
 653                     if (!parser_sy_pop(parser, &sy))
 654                         goto onerr;
 655                     if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 656                         olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 657                 }
 658
 659                 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
 660                     goto onerr;
 661             }
 662             wantop = false;
 663             parser->lex->flags.noops = true;
 664         }
 665         if (!parser_next(parser)) {
 666             goto onerr;
 667         }
 668         if (parser->tok == ';') {
 669             break;
 670         }
 671     }
 672     if (!parser_next(parser)) {
 673         parseerror(parser, "Unexpected end of file");
 674         goto onerr;
 675     }
 676
 677     while (sy.ops_count) {
 678         if (!parser_sy_pop(parser, &sy))
 679             goto onerr;
 680     }
 681
 682     parser->lex->flags.noops = true;
 683     if (!sy.out_count) {
 684         parseerror(parser, "empty expression");
 685         expr = NULL;
 686     } else
 687         expr = sy.out[0].out;
 688     MEM_VECTOR_CLEAR(&sy, out);
 689     MEM_VECTOR_CLEAR(&sy, ops);
 690     return expr;
 691
 692 onerr:
 693     parser->lex->flags.noops = true;
 694     MEM_VECTOR_CLEAR(&sy, out);
 695     MEM_VECTOR_CLEAR(&sy, ops);
 696     return NULL;
 697 }
 698
 699 static bool parser_variable(parser_t *parser, ast_block *localblock);
 700 static bool parser_body_do(parser_t *parser, ast_block *block)
 701 {
 702     if (parser->tok == TOKEN_TYPENAME)
 703     {
 704         /* local variable */
 705         if (!parser_variable(parser, block))
 706             return false;
 707         return true;
 708     }
 709     else if (parser->tok == TOKEN_KEYWORD)
 710     {
 711         if (!strcmp(parser_tokval(parser), "return"))
 712         {
 713             ast_expression *exp;
 714             ast_return *ret;
 715
 716             if (!parser_next(parser)) {
 717                 parseerror(parser, "expected return expression");
 718                 return false;
 719             }
 720
 721             exp = parser_expression(parser);
 722             if (!exp)
 723                 return false;
 724             ret = ast_return_new(exp->expression.node.context, exp);
 725             if (!ret) {
 726                 ast_delete(exp);
 727                 return false;
 728             }
 729             if (!ast_block_exprs_add(block, (ast_expression*)ret)) {
 730                 ast_delete(ret);
 731                 return false;
 732             }
 733             return true;
 734         }
 735         parseerror(parser, "Unexpected keyword");
 736         return false;
 737     }
 738     else if (parser->tok == '{')
 739     {
 740         /* a block */
 741         parseerror(parser, "TODO: inner blocks: %s", parser_tokval(parser));
 742         return false;
 743     }
 744     else
 745     {
 746         ast_expression *exp = parser_expression(parser);
 747         if (!exp)
 748             return false;
 749         if (!ast_block_exprs_add(block, exp)) {
 750             ast_delete(exp);
 751             return false;
 752         }
 753         return true;
 754     }
 755 }
 756
 757 static ast_block* parser_parse_block(parser_t *parser)
 758 {
 759     size_t oldblocklocal;
 760     ast_block *block = NULL;
 761
 762     oldblocklocal = parser->blocklocal;
 763     parser->blocklocal = parser->locals_count;
 764
 765     if (!parser_next(parser)) { /* skip the '{' */
 766         parseerror(parser, "expected function body");
 767         goto cleanup;
 768     }
 769
 770     block = ast_block_new(parser_ctx(parser));
 771
 772     while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
 773     {
 774         if (parser->tok == '}')
 775             break;
 776
 777         if (!parser_body_do(parser, block)) {
 778             ast_block_delete(block);
 779             block = NULL;
 780             goto cleanup;
 781         }
 782     }
 783
 784     if (parser->tok != '}') {
 785         ast_block_delete(block);
 786         block = NULL;
 787     } else {
 788         (void)parser_next(parser);
 789     }
 790
 791 cleanup:
 792     parser->blocklocal = oldblocklocal;
 793     return block;
 794 }
 795
 796 static bool parser_variable(parser_t *parser, ast_block *localblock)
 797 {
 798     bool          isfunc = false;
 799     ast_function *func = NULL;
 800     lex_ctx       ctx;
 801     ast_value    *var;
 802
 803     int basetype = parser_token(parser)->constval.t;
 804
 805     while (true)
 806     {
 807         if (!parser_next(parser)) { /* skip basetype or comma */
 808             parseerror(parser, "expected variable declaration");
 809             return false;
 810         }
 811
 812         isfunc = false;
 813         func = NULL;
 814         ctx = parser_ctx(parser);
 815         var = parser_parse_type(parser, basetype, &isfunc);
 816
 817         if (!var)
 818             return false;
 819
 820         if (parser->tok != TOKEN_IDENT) {
 821             parseerror(parser, "expected variable name\n");
 822             return false;
 823         }
 824
 825         if (!localblock && parser_find_global(parser, parser_tokval(parser))) {
 826             ast_value_delete(var);
 827             parseerror(parser, "global already exists: %s\n", parser_tokval(parser));
 828             return false;
 829         }
 830
 831         if (localblock && parser_find_local(parser, parser_tokval(parser), parser->blocklocal)) {
 832             ast_value_delete(var);
 833             parseerror(parser, "local variable already exists: %s\n", parser_tokval(parser));
 834             return false;
 835         }
 836
 837         if (!ast_value_set_name(var, parser_tokval(parser))) {
 838             parseerror(parser, "failed to set variable name\n");
 839             ast_value_delete(var);
 840             return false;
 841         }
 842
 843         if (isfunc) {
 844             /* a function was defined */
 845             ast_value *fval;
 846
 847             /* turn var into a value of TYPE_FUNCTION, with the old var
 848              * as return type
 849              */
 850             fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
 851             func = ast_function_new(ctx, var->name, fval);
 852             if (!fval || !func) {
 853                 ast_value_delete(var);
 854                 if (fval) ast_value_delete(fval);
 855                 if (func) ast_function_delete(func);
 856                 return false;
 857             }
 858
 859             fval->expression.next = (ast_expression*)var;
 860             MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
 861
 862             if (!parser_t_functions_add(parser, func)) {
 863                 ast_value_delete(var);
 864                 if (fval) ast_value_delete(fval);
 865                 if (func) ast_function_delete(func);
 866                 return false;
 867             }
 868
 869             var = fval;
 870         }
 871
 872         if ( (!localblock && !parser_t_globals_add(parser, var)) ||
 873              ( localblock && !parser_t_locals_add(parser, var)) )
 874         {
 875             ast_value_delete(var);
 876             return false;
 877         }
 878         if (localblock && !ast_block_locals_add(localblock, var))
 879         {
 880             parser->locals_count--;
 881             ast_value_delete(var);
 882             return false;
 883         }
 884
 885         if (!parser_next(parser)) {
 886             ast_value_delete(var);
 887             return false;
 888         }
 889
 890         if (parser->tok == ';') {
 891             if (!parser_next(parser))
 892                 return parser->tok == TOKEN_EOF;
 893             return true;
 894         }
 895
 896         if (parser->tok == ',') {
 897             /* another var */
 898             continue;
 899         }
 900
 901         if (parser->tok != '=') {
 902             parseerror(parser, "expected '=' or ';'");
 903             return false;
 904         }
 905
 906         if (!parser_next(parser))
 907             return false;
 908
 909         if (parser->tok == '#') {
 910             if (localblock) {
 911                 parseerror(parser, "cannot declare builtins within functions");
 912                 return false;
 913             }
 914             if (!isfunc || !func) {
 915                 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
 916                 return false;
 917             }
 918             if (!parser_next(parser)) {
 919                 parseerror(parser, "expected builtin number");
 920                 return false;
 921             }
 922             if (parser->tok != TOKEN_INTCONST) {
 923                 parseerror(parser, "builtin number must be an integer constant");
 924                 return false;
 925             }
 926             if (parser_token(parser)->constval.i <= 0) {
 927                 parseerror(parser, "builtin number must be positive integer greater than zero");
 928                 return false;
 929             }
 930
 931             func->builtin = -parser_token(parser)->constval.i;
 932         } else if (parser->tok == '{') {
 933             /* function body */
 934             ast_block *block;
 935             ast_function *old = parser->function;
 936
 937             if (localblock) {
 938                 parseerror(parser, "cannot declare functions within functions");
 939                 return false;
 940             }
 941
 942             parser->function = func;
 943             block = parser_parse_block(parser);
 944             parser->function = old;
 945
 946             if (!block)
 947                 return false;
 948
 949             if (!ast_function_blocks_add(func, block)) {
 950                 ast_block_delete(block);
 951                 return false;
 952             }
 953             return true;
 954         } else {
 955             parseerror(parser, "TODO, const assignment");
 956         }
 957
 958         if (!parser_next(parser))
 959             return false;
 960
 961         if (parser->tok == ',') {
 962             /* another */
 963             continue;
 964         }
 965
 966         if (parser->tok != ';') {
 967             parseerror(parser, "expected semicolon");
 968             return false;
 969         }
 970
 971         (void)parser_next(parser);
 972
 973         return true;
 974     }
 975 }
 976
 977 static bool parser_do(parser_t *parser)
 978 {
 979     if (parser->tok == TOKEN_TYPENAME)
 980     {
 981         return parser_variable(parser, NULL);
 982     }
 983     else if (parser->tok == TOKEN_KEYWORD)
 984     {
 985         /* handle 'var' and 'const' */
 986         return false;
 987     }
 988     else if (parser->tok == '.')
 989     {
 990         /* entity-member declaration */
 991         return false;
 992     }
 993     else
 994     {
 995         parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
 996         return false;
 997     }
 998     return true;
 999 }
1000
1001 bool parser_compile(const char *filename)
1002 {
1003     size_t i;
1004     parser_t *parser;
1005     ir_builder *ir;
1006
1007     parser = (parser_t*)mem_a(sizeof(parser_t));
1008     if (!parser)
1009         return false;
1010
1011     memset(parser, 0, sizeof(parser));
1012
1013     MEM_VECTOR_INIT(parser, globals);
1014     MEM_VECTOR_INIT(parser, locals);
1015     parser->lex = lex_open(filename);
1016
1017     if (!parser->lex) {
1018         printf("failed to open file \"%s\"\n", filename);
1019         return false;
1020     }
1021
1022     /* initial lexer/parser state */
1023     parser->lex->flags.noops = true;
1024
1025     if (parser_next(parser))
1026     {
1027         while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1028         {
1029             if (!parser_do(parser)) {
1030                 if (parser->tok == TOKEN_EOF)
1031                     parseerror(parser, "unexpected eof");
1032                 else
1033                     parseerror(parser, "parse error\n");
1034                 lex_close(parser->lex);
1035                 mem_d(parser);
1036                 return false;
1037             }
1038         }
1039     }
1040
1041     lex_close(parser->lex);
1042
1043     ir = ir_builder_new("gmqcc_out");
1044     if (!ir) {
1045         printf("failed to allocate builder\n");
1046         goto cleanup;
1047     }
1048
1049     for (i = 0; i < parser->imm_float_count; ++i) {
1050         if (!ast_global_codegen(parser->imm_float[i], ir)) {
1051             printf("failed to generate global %s\n", parser->imm_float[i]->name);
1052         }
1053     }
1054     for (i = 0; i < parser->globals_count; ++i) {
1055         if (!ast_global_codegen(parser->globals[i], ir)) {
1056             printf("failed to generate global %s\n", parser->globals[i]->name);
1057         }
1058     }
1059     for (i = 0; i < parser->functions_count; ++i) {
1060         if (!ast_function_codegen(parser->functions[i], ir)) {
1061             printf("failed to generate function %s\n", parser->functions[i]->name);
1062         }
1063         if (!ir_function_finalize(parser->functions[i]->ir_func)) {
1064             printf("failed to finalize function %s\n", parser->functions[i]->name);
1065         }
1066     }
1067
1068     ir_builder_dump(ir, printf);
1069
1070 cleanup:
1071     for (i = 0; i < parser->globals_count; ++i) {
1072         ast_value_delete(parser->globals[i]);
1073     }
1074     MEM_VECTOR_CLEAR(parser, globals);
1075
1076     mem_d(parser);
1077     return true;
1078 }