parser.c

   1 #include <stdio.h>
   2 #include <stdarg.h>
   3
   4 #include "gmqcc.h"
   5 #include "lexer.h"
   6
   7 typedef struct {
   8     char *name;
   9     ast_expression *var;
  10 } varentry_t;
  11
  12 typedef struct {
  13     lex_file *lex;
  14     int      tok;
  15
  16     MEM_VECTOR_MAKE(varentry_t, globals);
  17     MEM_VECTOR_MAKE(varentry_t, fields);
  18     MEM_VECTOR_MAKE(ast_function*, functions);
  19     MEM_VECTOR_MAKE(ast_value*, imm_float);
  20     MEM_VECTOR_MAKE(ast_value*, imm_string);
  21     MEM_VECTOR_MAKE(ast_value*, imm_vector);
  22
  23     ast_function *function;
  24     MEM_VECTOR_MAKE(varentry_t, locals);
  25     size_t blocklocal;
  26
  27     size_t errors;
  28
  29     /* TYPE_FIELD -> parser_find_fields is used instead of find_var
  30      * TODO: TYPE_VECTOR -> x, y and z are accepted in the gmqcc standard
  31      * anything else: type error
  32      */
  33     qcint  memberof;
  34 } parser_t;
  35
  36 MEM_VEC_FUNCTIONS(parser_t, varentry_t, globals)
  37 MEM_VEC_FUNCTIONS(parser_t, varentry_t, fields)
  38 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_float)
  39 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_string)
  40 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_vector)
  41 MEM_VEC_FUNCTIONS(parser_t, varentry_t, locals)
  42 MEM_VEC_FUNCTIONS(parser_t, ast_function*, functions)
  43
  44 void parseerror(parser_t *parser, const char *fmt, ...)
  45 {
  46         va_list ap;
  47
  48         parser->errors++;
  49
  50     if (parser)
  51             printf("error %s:%lu: ", parser->lex->tok->ctx.file, (unsigned long)parser->lex->tok->ctx.line);
  52         else
  53             printf("error: ");
  54
  55         va_start(ap, fmt);
  56         vprintf(fmt, ap);
  57         va_end(ap);
  58
  59         printf("\n");
  60 }
  61
  62 bool parser_next(parser_t *parser)
  63 {
  64     /* lex_do kills the previous token */
  65     parser->tok = lex_do(parser->lex);
  66     if (parser->tok == TOKEN_EOF || parser->tok >= TOKEN_ERROR)
  67         return false;
  68     return true;
  69 }
  70
  71 /* lift a token out of the parser so it's not destroyed by parser_next */
  72 token *parser_lift(parser_t *parser)
  73 {
  74     token *tok = parser->lex->tok;
  75     parser->lex->tok = NULL;
  76     return tok;
  77 }
  78
  79 #define parser_tokval(p) (p->lex->tok->value)
  80 #define parser_token(p)  (p->lex->tok)
  81 #define parser_ctx(p)    (p->lex->tok->ctx)
  82
  83 ast_value* parser_const_float(parser_t *parser, double d)
  84 {
  85     size_t i;
  86     ast_value *out;
  87     for (i = 0; i < parser->imm_float_count; ++i) {
  88         if (parser->imm_float[i]->constval.vfloat == d)
  89             return parser->imm_float[i];
  90     }
  91     out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_FLOAT);
  92     out->isconst = true;
  93     out->constval.vfloat = d;
  94     if (!parser_t_imm_float_add(parser, out)) {
  95         ast_value_delete(out);
  96         return NULL;
  97     }
  98     return out;
  99 }
 100
 101 ast_value* parser_const_string(parser_t *parser, const char *str)
 102 {
 103     size_t i;
 104     ast_value *out;
 105     for (i = 0; i < parser->imm_string_count; ++i) {
 106         if (!strcmp(parser->imm_string[i]->constval.vstring, str))
 107             return parser->imm_string[i];
 108     }
 109     out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_STRING);
 110     out->isconst = true;
 111     out->constval.vstring = util_strdup(str);
 112     if (!parser_t_imm_string_add(parser, out)) {
 113         ast_value_delete(out);
 114         return NULL;
 115     }
 116     return out;
 117 }
 118
 119 ast_value* parser_const_vector(parser_t *parser, vector v)
 120 {
 121     size_t i;
 122     ast_value *out;
 123     for (i = 0; i < parser->imm_vector_count; ++i) {
 124         if (!memcmp(&parser->imm_vector[i]->constval.vvec, &v, sizeof(v)))
 125             return parser->imm_vector[i];
 126     }
 127     out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_VECTOR);
 128     out->isconst = true;
 129     out->constval.vvec = v;
 130     if (!parser_t_imm_vector_add(parser, out)) {
 131         ast_value_delete(out);
 132         return NULL;
 133     }
 134     return out;
 135 }
 136
 137 ast_expression* parser_find_field(parser_t *parser, const char *name)
 138 {
 139     size_t i;
 140     for (i = 0; i < parser->fields_count; ++i) {
 141         if (!strcmp(parser->fields[i].name, name))
 142             return parser->fields[i].var;
 143     }
 144     return NULL;
 145 }
 146
 147 ast_expression* parser_find_global(parser_t *parser, const char *name)
 148 {
 149     size_t i;
 150     for (i = 0; i < parser->globals_count; ++i) {
 151         if (!strcmp(parser->globals[i].name, name))
 152             return parser->globals[i].var;
 153     }
 154     return NULL;
 155 }
 156
 157 ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto)
 158 {
 159     size_t i;
 160     ast_value *fun;
 161     for (i = parser->locals_count; i > upto;) {
 162         --i;
 163         if (!strcmp(parser->locals[i].name, name))
 164             return parser->locals[i].var;
 165     }
 166     fun = parser->function->vtype;
 167     for (i = 0; i < fun->expression.params_count; ++i) {
 168         if (!strcmp(fun->expression.params[i]->name, name))
 169             return (ast_expression*)(fun->expression.params[i]);
 170     }
 171     return NULL;
 172 }
 173
 174 ast_expression* parser_find_var(parser_t *parser, const char *name)
 175 {
 176     ast_expression *v;
 177     v         = parser_find_local(parser, name, 0);
 178     if (!v) v = parser_find_global(parser, name);
 179     return v;
 180 }
 181
 182 typedef struct {
 183     MEM_VECTOR_MAKE(ast_value*, p);
 184 } paramlist_t;
 185 MEM_VEC_FUNCTIONS(paramlist_t, ast_value*, p)
 186
 187 static ast_value *parser_parse_type(parser_t *parser, int basetype, bool *isfunc)
 188 {
 189     paramlist_t params;
 190     ast_value *var;
 191     lex_ctx   ctx = parser_ctx(parser);
 192     int vtype = basetype;
 193     int temptype;
 194     size_t i;
 195
 196     MEM_VECTOR_INIT(&params, p);
 197
 198     *isfunc = false;
 199
 200     if (parser->tok == '(') {
 201         *isfunc = true;
 202         while (true) {
 203             ast_value *param;
 204             bool dummy;
 205
 206             if (!parser_next(parser))
 207                 goto on_error;
 208
 209             if (parser->tok == ')')
 210                 break;
 211
 212             temptype = parser_token(parser)->constval.t;
 213             if (!parser_next(parser))
 214                 goto on_error;
 215
 216             param = parser_parse_type(parser, temptype, &dummy);
 217             (void)dummy;
 218
 219             if (!param)
 220                 goto on_error;
 221
 222             if (parser->tok == TOKEN_IDENT) {
 223                 /* named parameter */
 224                 if (!ast_value_set_name(param, parser_tokval(parser)))
 225                     goto on_error;
 226                 if (!parser_next(parser))
 227                     goto on_error;
 228             }
 229
 230             if (!paramlist_t_p_add(&params, param)) {
 231                 parseerror(parser, "Out of memory while parsing typename");
 232                 goto on_error;
 233             }
 234
 235             if (parser->tok == ',')
 236                 continue;
 237             if (parser->tok == ')')
 238                 break;
 239             parseerror(parser, "Unexpected token");
 240             goto on_error;
 241         }
 242         if (!parser_next(parser))
 243             goto on_error;
 244     }
 245
 246     var = ast_value_new(ctx, "<unnamed>", vtype);
 247     if (!var)
 248         goto on_error;
 249     MEM_VECTOR_MOVE(&params, p, &var->expression, params);
 250     return var;
 251 on_error:
 252     for (i = 0; i < params.p_count; ++i)
 253         ast_value_delete(params.p[i]);
 254     MEM_VECTOR_CLEAR(&params, p);
 255     return NULL;
 256 }
 257
 258 typedef struct
 259 {
 260     size_t etype; /* 0 = expression, others are operators */
 261     int             paren;
 262     size_t          off;
 263     ast_expression *out;
 264     ast_block      *block; /* for commas and function calls */
 265     lex_ctx ctx;
 266 } sy_elem;
 267 typedef struct
 268 {
 269     MEM_VECTOR_MAKE(sy_elem, out);
 270     MEM_VECTOR_MAKE(sy_elem, ops);
 271 } shunt;
 272 MEM_VEC_FUNCTIONS(shunt, sy_elem, out)
 273 MEM_VEC_FUNCTIONS(shunt, sy_elem, ops)
 274
 275 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
 276     sy_elem e;
 277     e.etype = 0;
 278     e.out   = v;
 279     e.block = NULL;
 280     e.ctx   = ctx;
 281     e.paren = 0;
 282     return e;
 283 }
 284
 285 static sy_elem syblock(lex_ctx ctx, ast_block *v) {
 286     sy_elem e;
 287     e.etype = 0;
 288     e.out   = (ast_expression*)v;
 289     e.block = v;
 290     e.ctx   = ctx;
 291     e.paren = 0;
 292     return e;
 293 }
 294
 295 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
 296     sy_elem e;
 297     e.etype = 1 + (op - operators);
 298     e.out   = NULL;
 299     e.block = NULL;
 300     e.ctx   = ctx;
 301     e.paren = 0;
 302     return e;
 303 }
 304
 305 static sy_elem syparen(lex_ctx ctx, int p, size_t off) {
 306     sy_elem e;
 307     e.etype = 0;
 308     e.off   = off;
 309     e.out   = NULL;
 310     e.block = NULL;
 311     e.ctx   = ctx;
 312     e.paren = p;
 313     return e;
 314 }
 315
 316 #ifdef DEBUGSHUNT
 317 # define DEBUGSHUNTDO(x) x
 318 #else
 319 # define DEBUGSHUNTDO(x)
 320 #endif
 321
 322 static bool parser_sy_pop(parser_t *parser, shunt *sy)
 323 {
 324     const oper_info *op;
 325     lex_ctx ctx;
 326     ast_expression *out = NULL;
 327     ast_expression *exprs[3];
 328     ast_block      *blocks[3];
 329     size_t i, assignop;
 330     qcint  generated_op = 0;
 331
 332     if (!sy->ops_count) {
 333         parseerror(parser, "internal error: missing operator");
 334         return false;
 335     }
 336
 337     if (sy->ops[sy->ops_count-1].paren) {
 338         parseerror(parser, "unmatched parenthesis");
 339         return false;
 340     }
 341
 342     op = &operators[sy->ops[sy->ops_count-1].etype - 1];
 343     ctx = sy->ops[sy->ops_count-1].ctx;
 344
 345     DEBUGSHUNTDO(printf("apply %s\n", op->op));
 346
 347     if (sy->out_count < op->operands) {
 348         parseerror(parser, "internal error: not enough operands: %i", sy->out_count);
 349         return false;
 350     }
 351
 352     sy->ops_count--;
 353
 354     sy->out_count -= op->operands;
 355     for (i = 0; i < op->operands; ++i) {
 356         exprs[i]  = sy->out[sy->out_count+i].out;
 357         blocks[i] = sy->out[sy->out_count+i].block;
 358     }
 359
 360     if (blocks[0] && !blocks[0]->exprs_count && op->id != opid1(',')) {
 361         parseerror(parser, "internal error: operator cannot be applied on empty blocks");
 362         return false;
 363     }
 364
 365 #define NotSameType(T) \
 366              (exprs[0]->expression.vtype != exprs[1]->expression.vtype || \
 367               exprs[0]->expression.vtype != T)
 368     switch (op->id)
 369     {
 370         default:
 371             parseerror(parser, "internal error: unhandled operand");
 372             return false;
 373
 374         case opid1('.'):
 375             if (exprs[0]->expression.vtype == TYPE_ENTITY) {
 376                 if (exprs[1]->expression.vtype != TYPE_FIELD) {
 377                     parseerror(parser, "type error: right hand of member-operand should be an entity-field");
 378                     return false;
 379                 }
 380                 out = (ast_expression*)ast_entfield_new(ctx, exprs[0], exprs[1]);
 381             }
 382             else if (exprs[0]->expression.vtype == TYPE_VECTOR) {
 383                 parseerror(parser, "internal error: vector access is not supposed to be handled at this point");
 384                 return false;
 385             }
 386             else {
 387                 parseerror(parser, "type error: member-of operator on something that is not an entity or vector");
 388                 return false;
 389             }
 390             break;
 391
 392         case opid1(','):
 393             if (blocks[0]) {
 394                 if (!ast_block_exprs_add(blocks[0], exprs[1]))
 395                     return false;
 396             } else {
 397                 blocks[0] = ast_block_new(ctx);
 398                 if (!ast_block_exprs_add(blocks[0], exprs[0]) ||
 399                     !ast_block_exprs_add(blocks[0], exprs[1]))
 400                 {
 401                     return false;
 402                 }
 403             }
 404             if (!ast_block_set_type(blocks[0], exprs[1]))
 405                 return false;
 406
 407             sy->out[sy->out_count++] = syblock(ctx, blocks[0]);
 408             return true;
 409
 410         case opid1('+'):
 411             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 412                 parseerror(parser, "Cannot add type %s and %s",
 413                            type_name[exprs[0]->expression.vtype],
 414                            type_name[exprs[1]->expression.vtype]);
 415                 return false;
 416             }
 417             if (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) {
 418                 parseerror(parser, "type error: %s - %s not defined",
 419                            type_name[exprs[0]->expression.vtype],
 420                            type_name[exprs[1]->expression.vtype]);
 421                 return false;
 422             }
 423             switch (exprs[0]->expression.vtype) {
 424                 case TYPE_FLOAT:
 425                     out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
 426                     break;
 427                 case TYPE_VECTOR:
 428                     out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
 429                     break;
 430                 default:
 431                     parseerror(parser, "type error: cannot add type %s to %s",
 432                                type_name[exprs[0]->expression.vtype],
 433                                type_name[exprs[1]->expression.vtype]);
 434                     return false;
 435             };
 436             break;
 437         case opid1('-'):
 438             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 439                 parseerror(parser, "type error: cannot subtract type %s from %s",
 440                            type_name[exprs[1]->expression.vtype],
 441                            type_name[exprs[0]->expression.vtype]);
 442                 return false;
 443             }
 444             if (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) {
 445                 parseerror(parser, "type error: %s - %s not defined",
 446                            type_name[exprs[0]->expression.vtype],
 447                            type_name[exprs[1]->expression.vtype]);
 448                 return false;
 449             }
 450             switch (exprs[0]->expression.vtype) {
 451                 case TYPE_FLOAT:
 452                     out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
 453                     break;
 454                 case TYPE_VECTOR:
 455                     out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
 456                     break;
 457                 default:
 458                     parseerror(parser, "type error: cannot subtract type %s from %s",
 459                                type_name[exprs[1]->expression.vtype],
 460                                type_name[exprs[0]->expression.vtype]);
 461                     return false;
 462             };
 463             break;
 464         case opid1('*'):
 465             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
 466                 exprs[0]->expression.vtype != TYPE_VECTOR &&
 467                 exprs[0]->expression.vtype != TYPE_FLOAT &&
 468                 exprs[1]->expression.vtype != TYPE_VECTOR &&
 469                 exprs[1]->expression.vtype != TYPE_FLOAT)
 470             {
 471                 parseerror(parser, "type error: cannot multiply type %s by %s",
 472                            type_name[exprs[1]->expression.vtype],
 473                            type_name[exprs[0]->expression.vtype]);
 474                 return false;
 475             }
 476             switch (exprs[0]->expression.vtype) {
 477                 case TYPE_FLOAT:
 478                     if (exprs[1]->expression.vtype == TYPE_VECTOR)
 479                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
 480                     else
 481                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
 482                     break;
 483                 case TYPE_VECTOR:
 484                     if (exprs[1]->expression.vtype == TYPE_FLOAT)
 485                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
 486                     else
 487                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
 488                     break;
 489                 default:
 490                     parseerror(parser, "type error: cannot multiplye type %s by %s",
 491                                type_name[exprs[1]->expression.vtype],
 492                                type_name[exprs[0]->expression.vtype]);
 493                     return false;
 494             };
 495             break;
 496         case opid1('/'):
 497             if (NotSameType(TYPE_FLOAT)) {
 498                 parseerror(parser, "type error: cannot divide types %s and %s",
 499                            type_name[exprs[0]->expression.vtype],
 500                            type_name[exprs[1]->expression.vtype]);
 501                 return false;
 502             }
 503             out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
 504             break;
 505         case opid1('%'):
 506             parseerror(parser, "qc does not have a modulo operator");
 507             return false;
 508         case opid1('|'):
 509         case opid1('&'):
 510             if (NotSameType(TYPE_FLOAT)) {
 511                 parseerror(parser, "type error: cannot perform bit operations on types %s and %s",
 512                            type_name[exprs[0]->expression.vtype],
 513                            type_name[exprs[1]->expression.vtype]);
 514                 return false;
 515             }
 516             out = (ast_expression*)ast_binary_new(ctx,
 517                                                   (op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
 518                                                   exprs[0], exprs[1]);
 519             break;
 520         case opid1('^'):
 521             parseerror(parser, "TODO: bitxor");
 522             return false;
 523
 524         case opid2('<','<'):
 525         case opid2('>','>'):
 526             parseerror(parser, "TODO: shifts");
 527             return false;
 528
 529         case opid2('|','|'):
 530             generated_op += 1; /* INSTR_OR */
 531         case opid2('&','&'):
 532             generated_op += INSTR_AND;
 533             if (NotSameType(TYPE_FLOAT)) {
 534                 parseerror(parser, "type error: cannot apply logical operation on types %s and %s",
 535                            type_name[exprs[0]->expression.vtype],
 536                            type_name[exprs[1]->expression.vtype]);
 537                 parseerror(parser, "TODO: logical ops for arbitrary types using INSTR_NOT");
 538                 parseerror(parser, "TODO: optional early out");
 539                 return false;
 540             }
 541             if (opts_standard == COMPILER_GMQCC)
 542                 printf("TODO: early out logic\n");
 543             out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
 544             break;
 545
 546         case opid1('>'):
 547             generated_op += 1; /* INSTR_GT */
 548         case opid1('<'):
 549             generated_op += 1; /* INSTR_LT */
 550         case opid2('>', '='):
 551             generated_op += 1; /* INSTR_GE */
 552         case opid2('<', '='):
 553             generated_op += INSTR_LE;
 554             if (NotSameType(TYPE_FLOAT)) {
 555                 parseerror(parser, "type error: cannot compare types %s and %s",
 556                            type_name[exprs[0]->expression.vtype],
 557                            type_name[exprs[1]->expression.vtype]);
 558                 return false;
 559             }
 560             out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
 561             break;
 562         case opid2('!', '='):
 563             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 564                 parseerror(parser, "type error: cannot compare types %s and %s",
 565                            type_name[exprs[0]->expression.vtype],
 566                            type_name[exprs[1]->expression.vtype]);
 567                 return false;
 568             }
 569             out = (ast_expression*)ast_binary_new(ctx, type_ne_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
 570             break;
 571         case opid2('=', '='):
 572             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 573                 parseerror(parser, "type error: cannot compare types %s and %s",
 574                            type_name[exprs[0]->expression.vtype],
 575                            type_name[exprs[1]->expression.vtype]);
 576                 return false;
 577             }
 578             out = (ast_expression*)ast_binary_new(ctx, type_eq_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
 579             break;
 580
 581         case opid1('='):
 582             if (ast_istype(exprs[0], ast_entfield))
 583                 assignop = type_storep_instr[exprs[0]->expression.vtype];
 584             else
 585                 assignop = type_store_instr[exprs[0]->expression.vtype];
 586             out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]);
 587             break;
 588     }
 589 #undef NotSameType
 590
 591     if (!out) {
 592         parseerror(parser, "failed to apply operand %s", op->op);
 593         return false;
 594     }
 595
 596     DEBUGSHUNTDO(printf("applied %s\n", op->op));
 597     sy->out[sy->out_count++] = syexp(ctx, out);
 598     return true;
 599 }
 600
 601 static bool parser_close_call(parser_t *parser, shunt *sy)
 602 {
 603     /* was a function call */
 604     ast_expression *fun;
 605     ast_call       *call;
 606
 607     size_t          fid;
 608     size_t          paramcount;
 609
 610     sy->ops_count--;
 611     fid = sy->ops[sy->ops_count].off;
 612
 613     /* out[fid] is the function
 614      * everything above is parameters...
 615      * 0 params = nothing
 616      * 1 params = ast_expression
 617      * more = ast_block
 618      */
 619
 620     if (sy->out_count < 1 || sy->out_count <= fid) {
 621         parseerror(parser, "internal error: function call needs function and parameter list...");
 622         return false;
 623     }
 624
 625     fun = sy->out[fid].out;
 626
 627     call = ast_call_new(sy->ops[sy->ops_count].ctx, fun);
 628     if (!call) {
 629         parseerror(parser, "out of memory");
 630         return false;
 631     }
 632
 633     if (fid+1 == sy->out_count) {
 634         /* no arguments */
 635         paramcount = 0;
 636     } else if (fid+2 == sy->out_count) {
 637         ast_block *params;
 638         sy->out_count--;
 639         params = sy->out[sy->out_count].block;
 640         if (!params) {
 641             /* 1 param */
 642             paramcount = 1;
 643             if (!ast_call_params_add(call, sy->out[sy->out_count].out)) {
 644                 ast_delete(sy->out[sy->out_count].out);
 645                 parseerror(parser, "out of memory");
 646                 return false;
 647             }
 648         } else {
 649             paramcount = params->exprs_count;
 650             MEM_VECTOR_MOVE(params, exprs, call, params);
 651             ast_delete(params);
 652         }
 653     } else {
 654         parseerror(parser, "invalid function call");
 655         return false;
 656     }
 657
 658     /* overwrite fid, the function, with a call */
 659     sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
 660
 661     if (fun->expression.vtype != TYPE_FUNCTION) {
 662         parseerror(parser, "not a function");
 663         return false;
 664     }
 665
 666     if (!fun->expression.next) {
 667         parseerror(parser, "could not determine function return type");
 668         return false;
 669     } else {
 670         if (fun->expression.params_count != paramcount) {
 671             parseerror(parser, "expected %i parameters, got %i", (int)fun->expression.params_count, paramcount);
 672             return false;
 673         }
 674     }
 675
 676     return true;
 677 }
 678
 679 static bool parser_close_paren(parser_t *parser, shunt *sy, bool functions_only)
 680 {
 681     if (!sy->ops_count) {
 682         parseerror(parser, "unmatched closing paren");
 683         return false;
 684     }
 685     if (sy->ops[sy->ops_count-1].paren == 1) {
 686         parseerror(parser, "empty parenthesis expression");
 687         return false;
 688     }
 689     while (sy->ops_count) {
 690         if (sy->ops[sy->ops_count-1].paren == 'f') {
 691             if (!parser_close_call(parser, sy))
 692                 return false;
 693             break;
 694         }
 695         if (sy->ops[sy->ops_count-1].paren == 1) {
 696             sy->ops_count--;
 697             return !functions_only;
 698         }
 699         if (!parser_sy_pop(parser, sy))
 700             return false;
 701     }
 702     return true;
 703 }
 704
 705 static ast_expression* parser_expression(parser_t *parser)
 706 {
 707     ast_expression *expr = NULL;
 708     shunt sy;
 709     bool wantop = false;
 710     bool gotmemberof = false;
 711
 712     /* count the parens because an if starts with one, so the
 713      * end of a condition is an unmatched closing paren
 714      */
 715     int parens = 0;
 716
 717     MEM_VECTOR_INIT(&sy, out);
 718     MEM_VECTOR_INIT(&sy, ops);
 719
 720     while (true)
 721     {
 722         if (gotmemberof)
 723             gotmemberof = false;
 724         else
 725             parser->memberof = 0;
 726         if (!wantop)
 727         {
 728             bool nextwant = true;
 729             if (parser->tok == TOKEN_IDENT)
 730             {
 731                 /* variable */
 732                 ast_expression *var;
 733                 if (opts_standard == COMPILER_GMQCC)
 734                 {
 735                     if (parser->memberof == TYPE_ENTITY)
 736                         var = parser_find_field(parser, parser_tokval(parser));
 737                     else if (parser->memberof == TYPE_VECTOR)
 738                     {
 739                         parseerror(parser, "TODO: implement effective vector member access");
 740                         goto onerr;
 741                     }
 742                     else if (parser->memberof) {
 743                         parseerror(parser, "namespace for member not found");
 744                         goto onerr;
 745                     }
 746                     else
 747                         var = parser_find_var(parser, parser_tokval(parser));
 748                 } else {
 749                     var = parser_find_var(parser, parser_tokval(parser));
 750                     if (!var)
 751                         var = parser_find_field(parser, parser_tokval(parser));
 752                 }
 753                 if (!var) {
 754                     parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
 755                     goto onerr;
 756                 }
 757                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), var))) {
 758                     parseerror(parser, "out of memory");
 759                     goto onerr;
 760                 }
 761                 DEBUGSHUNTDO(printf("push %s\n", parser_tokval(parser)));
 762             }
 763             else if (parser->tok == TOKEN_FLOATCONST) {
 764                 ast_value *val = parser_const_float(parser, (parser_token(parser)->constval.f));
 765                 if (!val)
 766                     return false;
 767                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 768                     parseerror(parser, "out of memory");
 769                     goto onerr;
 770                 }
 771                 DEBUGSHUNTDO(printf("push %g\n", parser_token(parser)->constval.f));
 772             }
 773             else if (parser->tok == TOKEN_INTCONST) {
 774                 ast_value *val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
 775                 if (!val)
 776                     return false;
 777                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 778                     parseerror(parser, "out of memory");
 779                     goto onerr;
 780                 }
 781                 DEBUGSHUNTDO(printf("push %i\n", parser_token(parser)->constval.i));
 782             }
 783             else if (parser->tok == TOKEN_STRINGCONST) {
 784                 ast_value *val = parser_const_string(parser, parser_tokval(parser));
 785                 if (!val)
 786                     return false;
 787                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 788                     parseerror(parser, "out of memory");
 789                     goto onerr;
 790                 }
 791                 DEBUGSHUNTDO(printf("push string\n"));
 792             }
 793             else if (parser->tok == TOKEN_VECTORCONST) {
 794                 ast_value *val = parser_const_vector(parser, parser_token(parser)->constval.v);
 795                 if (!val)
 796                     return false;
 797                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 798                     parseerror(parser, "out of memory");
 799                     goto onerr;
 800                 }
 801                 DEBUGSHUNTDO(printf("push '%g %g %g'\n",
 802                                     parser_token(parser)->constval.v.x,
 803                                     parser_token(parser)->constval.v.y,
 804                                     parser_token(parser)->constval.v.z));
 805             }
 806             else if (parser->tok == '(') {
 807                 ++parens;
 808                 nextwant = false; /* not expecting an operator next */
 809                 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1, 0))) {
 810                     parseerror(parser, "out of memory");
 811                     goto onerr;
 812                 }
 813                 DEBUGSHUNTDO(printf("push (\n"));
 814             }
 815             else if (parser->tok == ')') {
 816                 DEBUGSHUNTDO(printf("do[nop] )\n"));
 817                 --parens;
 818                 if (parens < 0)
 819                     break;
 820                 /* allowed for function calls */
 821                 if (!parser_close_paren(parser, &sy, true))
 822                     goto onerr;
 823             }
 824             else {
 825                 /* TODO: prefix operators */
 826                 parseerror(parser, "expected statement");
 827                 goto onerr;
 828             }
 829             wantop = nextwant;
 830             parser->lex->flags.noops = !wantop;
 831         } else {
 832             bool nextwant = false;
 833             if (parser->tok == '(') {
 834                 DEBUGSHUNTDO(printf("push (\n"));
 835                 ++parens;
 836                 /* we expected an operator, this is the function-call operator */
 837                 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 'f', sy.out_count-1))) {
 838                     parseerror(parser, "out of memory");
 839                     goto onerr;
 840                 }
 841             }
 842             else if (parser->tok == ')') {
 843                 DEBUGSHUNTDO(printf("do[op] )\n"));
 844                 --parens;
 845                 if (parens < 0)
 846                     break;
 847                 /* we do expect an operator next */
 848                 /* closing an opening paren */
 849                 if (!parser_close_paren(parser, &sy, false))
 850                     goto onerr;
 851                 nextwant = true;
 852             }
 853             else if (parser->tok != TOKEN_OPERATOR) {
 854                 parseerror(parser, "expected operator or end of statement");
 855                 goto onerr;
 856             }
 857             else {
 858                 /* classify the operator */
 859                 /* TODO: suffix operators */
 860                 const oper_info *op;
 861                 const oper_info *olast = NULL;
 862                 size_t o;
 863                 for (o = 0; o < operator_count; ++o) {
 864                     if (!(operators[o].flags & OP_PREFIX) &&
 865                         !(operators[o].flags & OP_SUFFIX) && /* remove this */
 866                         !strcmp(parser_tokval(parser), operators[o].op))
 867                     {
 868                         break;
 869                     }
 870                 }
 871                 if (o == operator_count) {
 872                     /* no operator found... must be the end of the statement */
 873                     break;
 874                 }
 875                 /* found an operator */
 876                 op = &operators[o];
 877                 if (op->id == opid1('.')) {
 878                     /* for gmqcc standard: open up the namespace of the previous type */
 879                     ast_expression *prevex = sy.out[sy.out_count-1].out;
 880                     if (!prevex) {
 881                         parseerror(parser, "unexpected member operator");
 882                         goto onerr;
 883                     }
 884                     if (prevex->expression.vtype == TYPE_ENTITY)
 885                         parser->memberof = TYPE_ENTITY;
 886                     else if (prevex->expression.vtype == TYPE_VECTOR)
 887                         parser->memberof = TYPE_VECTOR;
 888                     else {
 889                         parseerror(parser, "type error: type has no members");
 890                         goto onerr;
 891                     }
 892                     gotmemberof = true;
 893                 }
 894
 895                 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 896                     olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 897
 898                 while (olast && (
 899                         (op->prec < olast->prec) ||
 900                         (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
 901                 {
 902                     if (!parser_sy_pop(parser, &sy))
 903                         goto onerr;
 904                     if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 905                         olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 906                     else
 907                         olast = NULL;
 908                 }
 909
 910                 DEBUGSHUNTDO(printf("push operator %s\n", op->op));
 911                 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
 912                     goto onerr;
 913             }
 914             wantop = nextwant;
 915             parser->lex->flags.noops = !wantop;
 916         }
 917         if (!parser_next(parser)) {
 918             goto onerr;
 919         }
 920         if (parser->tok == ';') {
 921             break;
 922         }
 923     }
 924     if (parens >= 0 && !parser_next(parser)) {
 925         parseerror(parser, "Unexpected end of file");
 926         goto onerr;
 927     }
 928
 929     while (sy.ops_count) {
 930         if (!parser_sy_pop(parser, &sy))
 931             goto onerr;
 932     }
 933
 934     parser->lex->flags.noops = true;
 935     if (!sy.out_count) {
 936         parseerror(parser, "empty expression");
 937         expr = NULL;
 938     } else
 939         expr = sy.out[0].out;
 940     MEM_VECTOR_CLEAR(&sy, out);
 941     MEM_VECTOR_CLEAR(&sy, ops);
 942     DEBUGSHUNTDO(printf("shut done\n"));
 943     return expr;
 944
 945 onerr:
 946     parser->lex->flags.noops = true;
 947     MEM_VECTOR_CLEAR(&sy, out);
 948     MEM_VECTOR_CLEAR(&sy, ops);
 949     return NULL;
 950 }
 951
 952 static bool parser_variable(parser_t *parser, ast_block *localblock);
 953 static ast_block* parser_parse_block(parser_t *parser);
 954 static ast_expression* parser_parse_statement_or_block(parser_t *parser);
 955 static bool parser_parse_statement(parser_t *parser, ast_block *block, ast_expression **out)
 956 {
 957     if (parser->tok == TOKEN_TYPENAME)
 958     {
 959         /* local variable */
 960         if (!block) {
 961             parseerror(parser, "cannot declare a variable from here");
 962             return false;
 963         }
 964         if (!parser_variable(parser, block))
 965             return false;
 966         *out = NULL;
 967         return true;
 968     }
 969     else if (parser->tok == TOKEN_KEYWORD)
 970     {
 971         if (!strcmp(parser_tokval(parser), "return"))
 972         {
 973             ast_expression *exp = NULL;
 974             ast_return     *ret = NULL;
 975             ast_value      *expected = parser->function->vtype;
 976
 977             if (!parser_next(parser)) {
 978                 parseerror(parser, "expected return expression");
 979                 return false;
 980             }
 981
 982             if (parser->tok != ';') {
 983                 exp = parser_expression(parser);
 984                 if (!exp)
 985                     return false;
 986
 987                 if (exp->expression.vtype != expected->expression.next->expression.vtype) {
 988                     parseerror(parser, "return with invalid expression");
 989                 }
 990
 991                 ret = ast_return_new(exp->expression.node.context, exp);
 992                 if (!ret) {
 993                     ast_delete(exp);
 994                     return false;
 995                 }
 996
 997                 *out = (ast_expression*)ret;
 998             } else if (!parser_next(parser)) {
 999                 parseerror(parser, "expected semicolon");
1000                 if (expected->expression.next->expression.vtype != TYPE_VOID) {
1001                     parseerror(parser, "return without value");
1002                 }
1003             }
1004             return true;
1005         }
1006         else if (!strcmp(parser_tokval(parser), "if"))
1007         {
1008             ast_ifthen *ifthen;
1009             ast_expression *cond, *ontrue, *onfalse = NULL;
1010
1011             lex_ctx ctx = parser_ctx(parser);
1012
1013             /* skip the 'if' and check for opening paren */
1014             if (!parser_next(parser) || parser->tok != '(') {
1015                 parseerror(parser, "expected 'if' condition in parenthesis");
1016                 return false;
1017             }
1018             /* parse into the expression */
1019             if (!parser_next(parser)) {
1020                 parseerror(parser, "expected 'if' condition after opening paren");
1021                 return false;
1022             }
1023             /* parse the condition */
1024             cond = parser_expression(parser);
1025             if (!cond)
1026                 return false;
1027             /* closing paren */
1028             if (parser->tok != ')') {
1029                 parseerror(parser, "expected closing paren after 'if' condition");
1030                 ast_delete(cond);
1031                 return false;
1032             }
1033             /* parse into the 'then' branch */
1034             if (!parser_next(parser)) {
1035                 parseerror(parser, "expected statement for on-true branch of 'if'");
1036                 ast_delete(cond);
1037                 return false;
1038             }
1039             ontrue = parser_parse_statement_or_block(parser);
1040             if (!ontrue) {
1041                 ast_delete(cond);
1042                 return false;
1043             }
1044             /* check for an else */
1045             if (!strcmp(parser_tokval(parser), "else")) {
1046                 /* parse into the 'else' branch */
1047                 if (!parser_next(parser)) {
1048                     parseerror(parser, "expected on-false branch after 'else'");
1049                     ast_delete(ontrue);
1050                     ast_delete(cond);
1051                     return false;
1052                 }
1053                 onfalse = parser_parse_statement_or_block(parser);
1054                 if (!onfalse) {
1055                     ast_delete(ontrue);
1056                     ast_delete(cond);
1057                     return false;
1058                 }
1059             }
1060
1061             ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
1062             *out = (ast_expression*)ifthen;
1063             return true;
1064         }
1065         parseerror(parser, "Unexpected keyword");
1066         return false;
1067     }
1068     else if (parser->tok == '{')
1069     {
1070         ast_block *inner;
1071         inner = parser_parse_block(parser);
1072         if (!inner)
1073             return false;
1074         *out = (ast_expression*)inner;
1075         return true;
1076     }
1077     else
1078     {
1079         ast_expression *exp = parser_expression(parser);
1080         if (!exp)
1081             return false;
1082         *out = exp;
1083         return true;
1084     }
1085 }
1086
1087 static void parser_pop_local(parser_t *parser)
1088 {
1089     parser->locals_count--;
1090     mem_d(parser->locals[parser->locals_count].name);
1091 }
1092
1093 static ast_block* parser_parse_block(parser_t *parser)
1094 {
1095     size_t oldblocklocal;
1096     ast_block *block = NULL;
1097
1098     oldblocklocal = parser->blocklocal;
1099     parser->blocklocal = parser->locals_count;
1100
1101     if (!parser_next(parser)) { /* skip the '{' */
1102         parseerror(parser, "expected function body");
1103         goto cleanup;
1104     }
1105
1106     block = ast_block_new(parser_ctx(parser));
1107
1108     while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1109     {
1110         ast_expression *expr;
1111         if (parser->tok == '}')
1112             break;
1113
1114         if (!parser_parse_statement(parser, block, &expr)) {
1115             ast_block_delete(block);
1116             block = NULL;
1117             goto cleanup;
1118         }
1119         if (!expr)
1120             continue;
1121         if (!ast_block_exprs_add(block, expr)) {
1122             ast_delete(expr);
1123             ast_block_delete(block);
1124             block = NULL;
1125             goto cleanup;
1126         }
1127     }
1128
1129     if (parser->tok != '}') {
1130         ast_block_delete(block);
1131         block = NULL;
1132     } else {
1133         (void)parser_next(parser);
1134     }
1135
1136 cleanup:
1137     while (parser->locals_count > parser->blocklocal)
1138         parser_pop_local(parser);
1139     parser->blocklocal = oldblocklocal;
1140     /* unroll the local vector */
1141     return block;
1142 }
1143
1144 static ast_expression* parser_parse_statement_or_block(parser_t *parser)
1145 {
1146     ast_expression *expr;
1147     if (parser->tok == '{')
1148         return (ast_expression*)parser_parse_block(parser);
1149     if (!parser_parse_statement(parser, NULL, &expr))
1150         return NULL;
1151     return expr;
1152 }
1153
1154 static bool parser_variable(parser_t *parser, ast_block *localblock)
1155 {
1156     bool          isfunc = false;
1157     ast_function *func = NULL;
1158     lex_ctx       ctx;
1159     ast_value    *var;
1160     varentry_t    varent;
1161
1162     int basetype = parser_token(parser)->constval.t;
1163
1164     while (true)
1165     {
1166         if (!parser_next(parser)) { /* skip basetype or comma */
1167             parseerror(parser, "expected variable declaration");
1168             return false;
1169         }
1170
1171         isfunc = false;
1172         func = NULL;
1173         ctx = parser_ctx(parser);
1174         var = parser_parse_type(parser, basetype, &isfunc);
1175
1176         if (!var)
1177             return false;
1178
1179         if (parser->tok != TOKEN_IDENT) {
1180             parseerror(parser, "expected variable name\n");
1181             return false;
1182         }
1183
1184         if (!localblock && parser_find_global(parser, parser_tokval(parser))) {
1185             ast_value_delete(var);
1186             parseerror(parser, "global already exists: %s\n", parser_tokval(parser));
1187             return false;
1188         }
1189
1190         if (localblock && parser_find_local(parser, parser_tokval(parser), parser->blocklocal)) {
1191             ast_value_delete(var);
1192             parseerror(parser, "local variable already exists: %s\n", parser_tokval(parser));
1193             return false;
1194         }
1195
1196         if (!ast_value_set_name(var, parser_tokval(parser))) {
1197             parseerror(parser, "failed to set variable name\n");
1198             ast_value_delete(var);
1199             return false;
1200         }
1201
1202         if (isfunc) {
1203             /* a function was defined */
1204             ast_value *fval;
1205
1206             /* turn var into a value of TYPE_FUNCTION, with the old var
1207              * as return type
1208              */
1209             fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1210             func = ast_function_new(ctx, var->name, fval);
1211             if (!fval || !func) {
1212                 ast_value_delete(var);
1213                 if (fval) ast_value_delete(fval);
1214                 if (func) ast_function_delete(func);
1215                 return false;
1216             }
1217
1218             fval->expression.next = (ast_expression*)var;
1219             MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1220
1221             if (!parser_t_functions_add(parser, func)) {
1222                 ast_value_delete(var);
1223                 if (fval) ast_value_delete(fval);
1224                 if (func) ast_function_delete(func);
1225                 return false;
1226             }
1227
1228             var = fval;
1229         }
1230
1231         varent.name = util_strdup(var->name);
1232         varent.var = (ast_expression*)var;
1233         if (var->expression.vtype == TYPE_VECTOR)
1234         {
1235             size_t len = strlen(varent.name);
1236             varentry_t vx, vy, vz;
1237             vx.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 0);
1238             vy.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 1);
1239             vz.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 2);
1240             vx.name = mem_a(len+3);
1241             vy.name = mem_a(len+3);
1242             vz.name = mem_a(len+3);
1243             strcpy(vx.name, varent.name);
1244             strcpy(vy.name, varent.name);
1245             strcpy(vz.name, varent.name);
1246             vx.name[len] = vy.name[len] = vz.name[len] = '_';
1247             vx.name[len+1] = 'x';
1248             vy.name[len+1] = 'y';
1249             vz.name[len+1] = 'z';
1250             vx.name[len+2] = vy.name[len+2] = vz.name[len+2] = 0;
1251
1252             if (!localblock) {
1253                 (void)!parser_t_globals_add(parser, varent);
1254                 (void)!parser_t_globals_add(parser, vx);
1255                 (void)!parser_t_globals_add(parser, vy);
1256                 (void)!parser_t_globals_add(parser, vz);
1257             } else {
1258                 (void)!parser_t_locals_add(parser, varent);
1259                 (void)!parser_t_locals_add(parser, vx);
1260                 (void)!parser_t_locals_add(parser, vy);
1261                 (void)!parser_t_locals_add(parser, vz);
1262             }
1263         }
1264         else
1265         {
1266             if ( (!localblock && !parser_t_globals_add(parser, varent)) ||
1267                  ( localblock && !parser_t_locals_add(parser, varent)) )
1268             {
1269                 ast_value_delete(var);
1270                 return false;
1271             }
1272         }
1273         if (localblock && !ast_block_locals_add(localblock, var))
1274         {
1275             parser_pop_local(parser);
1276             ast_value_delete(var);
1277             return false;
1278         }
1279
1280         if (!parser_next(parser)) {
1281             ast_value_delete(var);
1282             return false;
1283         }
1284
1285         if (parser->tok == ';') {
1286             if (!parser_next(parser))
1287                 return parser->tok == TOKEN_EOF;
1288             return true;
1289         }
1290
1291         if (parser->tok == ',') {
1292             /* another var */
1293             continue;
1294         }
1295
1296         if (parser->tok != '=') {
1297             parseerror(parser, "expected '=' or ';'");
1298             return false;
1299         }
1300
1301         if (!parser_next(parser))
1302             return false;
1303
1304         if (parser->tok == '#') {
1305             if (localblock) {
1306                 parseerror(parser, "cannot declare builtins within functions");
1307                 return false;
1308             }
1309             if (!isfunc || !func) {
1310                 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
1311                 return false;
1312             }
1313             if (!parser_next(parser)) {
1314                 parseerror(parser, "expected builtin number");
1315                 return false;
1316             }
1317             if (parser->tok != TOKEN_INTCONST) {
1318                 parseerror(parser, "builtin number must be an integer constant");
1319                 return false;
1320             }
1321             if (parser_token(parser)->constval.i <= 0) {
1322                 parseerror(parser, "builtin number must be positive integer greater than zero");
1323                 return false;
1324             }
1325
1326             func->builtin = -parser_token(parser)->constval.i;
1327         } else if (parser->tok == '{') {
1328             /* function body */
1329             ast_block *block;
1330             ast_function *old = parser->function;
1331
1332             if (localblock) {
1333                 parseerror(parser, "cannot declare functions within functions");
1334                 return false;
1335             }
1336
1337             parser->function = func;
1338             block = parser_parse_block(parser);
1339             parser->function = old;
1340
1341             if (!block)
1342                 return false;
1343
1344             if (!ast_function_blocks_add(func, block)) {
1345                 ast_block_delete(block);
1346                 return false;
1347             }
1348             return true;
1349         } else {
1350             parseerror(parser, "TODO, const assignment");
1351         }
1352
1353         if (!parser_next(parser))
1354             return false;
1355
1356         if (parser->tok == ',') {
1357             /* another */
1358             continue;
1359         }
1360
1361         if (parser->tok != ';') {
1362             parseerror(parser, "expected semicolon");
1363             return false;
1364         }
1365
1366         (void)parser_next(parser);
1367
1368         return true;
1369     }
1370 }
1371
1372 static bool parser_do(parser_t *parser)
1373 {
1374     if (parser->tok == TOKEN_TYPENAME)
1375     {
1376         return parser_variable(parser, NULL);
1377     }
1378     else if (parser->tok == TOKEN_KEYWORD)
1379     {
1380         /* handle 'var' and 'const' */
1381         return false;
1382     }
1383     else if (parser->tok == '.')
1384     {
1385         ast_value *var;
1386         ast_value *fld;
1387         bool       isfunc = false;
1388         int        basetype;
1389         lex_ctx    ctx = parser_ctx(parser);
1390         varentry_t varent;
1391
1392         /* entity-member declaration */
1393         if (!parser_next(parser) || parser->tok != TOKEN_TYPENAME) {
1394             parseerror(parser, "expected member variable definition");
1395             return false;
1396         }
1397
1398         /* remember the base/return type */
1399         basetype = parser_token(parser)->constval.t;
1400
1401         /* parse into the declaration */
1402         if (!parser_next(parser)) {
1403             parseerror(parser, "expected field def");
1404             return false;
1405         }
1406
1407         /* parse the field type fully */
1408         var = parser_parse_type(parser, basetype, &isfunc);
1409         if (!var)
1410             return false;
1411
1412         /* now the field name */
1413         if (parser->tok != TOKEN_IDENT) {
1414             parseerror(parser, "expected field name");
1415             ast_delete(var);
1416             return false;
1417         }
1418
1419         /* check for an existing field
1420          * in original qc we also have to check for an existing
1421          * global named like the field
1422          */
1423         if (opts_standard == COMPILER_QCC) {
1424             if (parser_find_global(parser, parser_tokval(parser))) {
1425                 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
1426                 ast_delete(var);
1427                 return false;
1428             }
1429         }
1430         if (parser_find_field(parser, parser_tokval(parser))) {
1431             parseerror(parser, "field %s already exists", parser_tokval(parser));
1432             ast_delete(var);
1433             return false;
1434         }
1435
1436         /* if it was a function, turn it into a function */
1437         if (isfunc) {
1438             ast_value *fval;
1439             /* turn var into a value of TYPE_FUNCTION, with the old var
1440              * as return type
1441              */
1442             fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1443             if (!fval) {
1444                 ast_value_delete(var);
1445                 ast_value_delete(fval);
1446                 return false;
1447             }
1448
1449             fval->expression.next = (ast_expression*)var;
1450             MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1451
1452             var = fval;
1453         }
1454
1455         /* turn it into a field */
1456         fld = ast_value_new(ctx, parser_tokval(parser), TYPE_FIELD);
1457         fld->expression.next = (ast_expression*)var;
1458
1459         varent.var = (ast_expression*)fld;
1460         if (var->expression.vtype == TYPE_VECTOR)
1461         {
1462             /* create _x, _y and _z fields as well */
1463             parseerror(parser, "TODO: vector field members (_x,_y,_z)");
1464             ast_delete(fld);
1465             return false;
1466         }
1467
1468         varent.name = util_strdup(fld->name);
1469         (void)!parser_t_fields_add(parser, varent);
1470
1471         /* end with a semicolon */
1472         if (!parser_next(parser) || parser->tok != ';') {
1473             parseerror(parser, "semicolon expected");
1474             return false;
1475         }
1476
1477         /* skip the semicolon */
1478         if (!parser_next(parser))
1479             return parser->tok == TOKEN_EOF;
1480
1481         return true;
1482     }
1483     else
1484     {
1485         parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
1486         return false;
1487     }
1488     return true;
1489 }
1490
1491 static parser_t *parser;
1492
1493 bool parser_init()
1494 {
1495     parser = (parser_t*)mem_a(sizeof(parser_t));
1496     if (!parser)
1497         return false;
1498
1499     memset(parser, 0, sizeof(parser));
1500
1501     MEM_VECTOR_INIT(parser, globals);
1502     MEM_VECTOR_INIT(parser, locals);
1503     return true;
1504 }
1505
1506 bool parser_compile(const char *filename)
1507 {
1508     parser->lex = lex_open(filename);
1509     if (!parser->lex) {
1510         printf("failed to open file \"%s\"\n", filename);
1511         return false;
1512     }
1513
1514     /* initial lexer/parser state */
1515     parser->lex->flags.noops = true;
1516
1517     if (parser_next(parser))
1518     {
1519         while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1520         {
1521             if (!parser_do(parser)) {
1522                 if (parser->tok == TOKEN_EOF)
1523                     parseerror(parser, "unexpected eof");
1524                 else
1525                     parseerror(parser, "parse error\n");
1526                 lex_close(parser->lex);
1527                 mem_d(parser);
1528                 return false;
1529             }
1530         }
1531     }
1532
1533     lex_close(parser->lex);
1534
1535     return !parser->errors;
1536 }
1537
1538 void parser_cleanup()
1539 {
1540     size_t i;
1541     for (i = 0; i < parser->functions_count; ++i) {
1542         ast_delete(parser->functions[i]);
1543     }
1544     for (i = 0; i < parser->imm_vector_count; ++i) {
1545         ast_delete(parser->imm_vector[i]);
1546     }
1547     for (i = 0; i < parser->imm_string_count; ++i) {
1548         ast_delete(parser->imm_string[i]);
1549     }
1550     for (i = 0; i < parser->imm_float_count; ++i) {
1551         ast_delete(parser->imm_float[i]);
1552     }
1553     for (i = 0; i < parser->globals_count; ++i) {
1554         ast_delete(parser->globals[i].var);
1555         mem_d(parser->globals[i].name);
1556     }
1557     MEM_VECTOR_CLEAR(parser, globals);
1558
1559     mem_d(parser);
1560 }
1561
1562 bool parser_finish(const char *output)
1563 {
1564     size_t i;
1565     ir_builder *ir;
1566
1567     if (!parser->errors)
1568     {
1569         ir = ir_builder_new("gmqcc_out");
1570         if (!ir) {
1571             printf("failed to allocate builder\n");
1572             return false;
1573         }
1574
1575         for (i = 0; i < parser->imm_float_count; ++i) {
1576             if (!ast_global_codegen(parser->imm_float[i], ir)) {
1577                 printf("failed to generate global %s\n", parser->imm_float[i]->name);
1578                 ir_builder_delete(ir);
1579                 return false;
1580             }
1581         }
1582         for (i = 0; i < parser->imm_string_count; ++i) {
1583             if (!ast_global_codegen(parser->imm_string[i], ir)) {
1584                 printf("failed to generate global %s\n", parser->imm_string[i]->name);
1585                 ir_builder_delete(ir);
1586                 return false;
1587             }
1588         }
1589         for (i = 0; i < parser->imm_vector_count; ++i) {
1590             if (!ast_global_codegen(parser->imm_vector[i], ir)) {
1591                 printf("failed to generate global %s\n", parser->imm_vector[i]->name);
1592                 ir_builder_delete(ir);
1593                 return false;
1594             }
1595         }
1596         for (i = 0; i < parser->fields_count; ++i) {
1597             ast_value *field;
1598             bool isconst;
1599             if (!ast_istype(parser->fields[i].var, ast_value))
1600                 continue;
1601             field = (ast_value*)parser->fields[i].var;
1602             isconst = field->isconst;
1603             field->isconst = false;
1604             if (!ast_global_codegen((ast_value*)field, ir)) {
1605                 printf("failed to generate field %s\n", field->name);
1606                 ir_builder_delete(ir);
1607                 return false;
1608             }
1609             if (isconst) {
1610                 ir_value *ifld;
1611                 ast_expression *subtype;
1612                 field->isconst = true;
1613                 subtype = field->expression.next;
1614                 ifld = ir_builder_create_field(ir, field->name, subtype->expression.vtype);
1615                 if (subtype->expression.vtype == TYPE_FIELD)
1616                     ifld->fieldtype = subtype->expression.next->expression.vtype;
1617                 else if (subtype->expression.vtype == TYPE_FUNCTION)
1618                     ifld->outtype = subtype->expression.next->expression.vtype;
1619                 (void)!ir_value_set_field(field->ir_v, ifld);
1620             }
1621         }
1622         for (i = 0; i < parser->globals_count; ++i) {
1623             if (!ast_istype(parser->globals[i].var, ast_value))
1624                 continue;
1625             if (!ast_global_codegen((ast_value*)(parser->globals[i].var), ir)) {
1626                 printf("failed to generate global %s\n", parser->globals[i].name);
1627                 ir_builder_delete(ir);
1628                 return false;
1629             }
1630         }
1631         for (i = 0; i < parser->functions_count; ++i) {
1632             if (!ast_function_codegen(parser->functions[i], ir)) {
1633                 printf("failed to generate function %s\n", parser->functions[i]->name);
1634                 ir_builder_delete(ir);
1635                 return false;
1636             }
1637             if (!ir_function_finalize(parser->functions[i]->ir_func)) {
1638                 printf("failed to finalize function %s\n", parser->functions[i]->name);
1639                 ir_builder_delete(ir);
1640                 return false;
1641             }
1642         }
1643
1644         ir_builder_dump(ir, printf);
1645
1646         if (!ir_builder_generate(ir, output)) {
1647             printf("*** failed to generate output file\n");
1648             ir_builder_delete(ir);
1649             return false;
1650         }
1651
1652         ir_builder_delete(ir);
1653         return true;
1654     }
1655
1656     printf("*** there were compile errors\n");
1657     return false;
1658 }