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         case opid1('&'):
 507             if (NotSameType(TYPE_FLOAT)) {
 508                 parseerror(parser, "type error: cannot perform bit operations on types %s and %s",
 509                            type_name[exprs[0]->expression.vtype],
 510                            type_name[exprs[1]->expression.vtype]);
 511                 return false;
 512             }
 513             out = (ast_expression*)ast_binary_new(ctx,
 514                                                   (op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
 515                                                   exprs[0], exprs[1]);
 516             break;
 517
 518         case opid1('>'):
 519             generated_op += 1; /* INSTR_GT */
 520         case opid1('<'):
 521             generated_op += 1; /* INSTR_LT */
 522         case opid2('>', '='):
 523             generated_op += 1; /* INSTR_GE */
 524         case opid2('<', '='):
 525             generated_op += INSTR_LE;
 526             if (NotSameType(TYPE_FLOAT)) {
 527                 parseerror(parser, "type error: cannot compare types %s and %s",
 528                            type_name[exprs[0]->expression.vtype],
 529                            type_name[exprs[1]->expression.vtype]);
 530                 return false;
 531             }
 532             out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
 533             break;
 534         case opid2('!', '='):
 535             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 536                 parseerror(parser, "type error: cannot compare types %s and %s",
 537                            type_name[exprs[0]->expression.vtype],
 538                            type_name[exprs[1]->expression.vtype]);
 539                 return false;
 540             }
 541             out = (ast_expression*)ast_binary_new(ctx, type_ne_op[exprs[0]->expression.vtype], exprs[0], exprs[1]);
 542             break;
 543         case opid2('=', '='):
 544             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 545                 parseerror(parser, "type error: cannot compare types %s and %s",
 546                            type_name[exprs[0]->expression.vtype],
 547                            type_name[exprs[1]->expression.vtype]);
 548                 return false;
 549             }
 550             out = (ast_expression*)ast_binary_new(ctx, type_eq_op[exprs[0]->expression.vtype], exprs[0], exprs[1]);
 551             break;
 552
 553
 554         case opid1('='):
 555             if (ast_istype(exprs[0], ast_entfield))
 556                 assignop = type_storep_instr[exprs[0]->expression.vtype];
 557             else
 558                 assignop = type_store_instr[exprs[0]->expression.vtype];
 559             out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]);
 560             break;
 561     }
 562 #undef NotSameType
 563
 564     if (!out) {
 565         parseerror(parser, "failed to apply operand %s", op->op);
 566         return false;
 567     }
 568
 569     DEBUGSHUNTDO(printf("applied %s\n", op->op));
 570     sy->out[sy->out_count++] = syexp(ctx, out);
 571     return true;
 572 }
 573
 574 static bool parser_close_call(parser_t *parser, shunt *sy)
 575 {
 576     /* was a function call */
 577     ast_expression *fun;
 578     ast_call       *call;
 579
 580     size_t          fid;
 581     size_t          paramcount;
 582
 583     sy->ops_count--;
 584     fid = sy->ops[sy->ops_count].off;
 585
 586     /* out[fid] is the function
 587      * everything above is parameters...
 588      * 0 params = nothing
 589      * 1 params = ast_expression
 590      * more = ast_block
 591      */
 592
 593     if (sy->out_count < 1 || sy->out_count <= fid) {
 594         parseerror(parser, "internal error: function call needs function and parameter list...");
 595         return false;
 596     }
 597
 598     fun = sy->out[fid].out;
 599
 600     call = ast_call_new(sy->ops[sy->ops_count].ctx, fun);
 601     if (!call) {
 602         parseerror(parser, "out of memory");
 603         return false;
 604     }
 605
 606     if (fid+1 == sy->out_count) {
 607         /* no arguments */
 608         paramcount = 0;
 609     } else if (fid+2 == sy->out_count) {
 610         ast_block *params;
 611         sy->out_count--;
 612         params = sy->out[sy->out_count].block;
 613         if (!params) {
 614             /* 1 param */
 615             paramcount = 1;
 616             if (!ast_call_params_add(call, sy->out[sy->out_count].out)) {
 617                 ast_delete(sy->out[sy->out_count].out);
 618                 parseerror(parser, "out of memory");
 619                 return false;
 620             }
 621         } else {
 622             paramcount = params->exprs_count;
 623             MEM_VECTOR_MOVE(params, exprs, call, params);
 624             ast_delete(params);
 625         }
 626     } else {
 627         parseerror(parser, "invalid function call");
 628         return false;
 629     }
 630
 631     /* overwrite fid, the function, with a call */
 632     sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
 633
 634     if (fun->expression.vtype != TYPE_FUNCTION) {
 635         parseerror(parser, "not a function");
 636         return false;
 637     }
 638
 639     if (!fun->expression.next) {
 640         parseerror(parser, "could not determine function return type");
 641         return false;
 642     } else {
 643         if (fun->expression.params_count != paramcount) {
 644             parseerror(parser, "expected %i parameters, got %i", (int)fun->expression.params_count, paramcount);
 645             return false;
 646         }
 647     }
 648
 649     return true;
 650 }
 651
 652 static bool parser_close_paren(parser_t *parser, shunt *sy, bool functions_only)
 653 {
 654     if (!sy->ops_count) {
 655         parseerror(parser, "unmatched closing paren");
 656         return false;
 657     }
 658     if (sy->ops[sy->ops_count-1].paren == 1) {
 659         parseerror(parser, "empty parenthesis expression");
 660         return false;
 661     }
 662     while (sy->ops_count) {
 663         if (sy->ops[sy->ops_count-1].paren == 'f') {
 664             if (!parser_close_call(parser, sy))
 665                 return false;
 666             break;
 667         }
 668         if (sy->ops[sy->ops_count-1].paren == 1) {
 669             sy->ops_count--;
 670             return !functions_only;
 671         }
 672         if (!parser_sy_pop(parser, sy))
 673             return false;
 674     }
 675     return true;
 676 }
 677
 678 static ast_expression* parser_expression(parser_t *parser)
 679 {
 680     ast_expression *expr = NULL;
 681     shunt sy;
 682     bool wantop = false;
 683     bool gotmemberof = false;
 684
 685     /* count the parens because an if starts with one, so the
 686      * end of a condition is an unmatched closing paren
 687      */
 688     int parens = 0;
 689
 690     MEM_VECTOR_INIT(&sy, out);
 691     MEM_VECTOR_INIT(&sy, ops);
 692
 693     while (true)
 694     {
 695         if (gotmemberof)
 696             gotmemberof = false;
 697         else
 698             parser->memberof = 0;
 699         if (!wantop)
 700         {
 701             bool nextwant = true;
 702             if (parser->tok == TOKEN_IDENT)
 703             {
 704                 /* variable */
 705                 ast_expression *var;
 706                 if (opts_standard == COMPILER_GMQCC)
 707                 {
 708                     if (parser->memberof == TYPE_ENTITY)
 709                         var = parser_find_field(parser, parser_tokval(parser));
 710                     else if (parser->memberof == TYPE_VECTOR)
 711                     {
 712                         parseerror(parser, "TODO: implement effective vector member access");
 713                         goto onerr;
 714                     }
 715                     else if (parser->memberof) {
 716                         parseerror(parser, "namespace for member not found");
 717                         goto onerr;
 718                     }
 719                     else
 720                         var = parser_find_var(parser, parser_tokval(parser));
 721                 } else {
 722                     var = parser_find_var(parser, parser_tokval(parser));
 723                     if (!var)
 724                         var = parser_find_field(parser, parser_tokval(parser));
 725                 }
 726                 if (!var) {
 727                     parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
 728                     goto onerr;
 729                 }
 730                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), var))) {
 731                     parseerror(parser, "out of memory");
 732                     goto onerr;
 733                 }
 734                 DEBUGSHUNTDO(printf("push %s\n", parser_tokval(parser)));
 735             }
 736             else if (parser->tok == TOKEN_FLOATCONST) {
 737                 ast_value *val = parser_const_float(parser, (parser_token(parser)->constval.f));
 738                 if (!val)
 739                     return false;
 740                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 741                     parseerror(parser, "out of memory");
 742                     goto onerr;
 743                 }
 744                 DEBUGSHUNTDO(printf("push %g\n", parser_token(parser)->constval.f));
 745             }
 746             else if (parser->tok == TOKEN_INTCONST) {
 747                 ast_value *val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
 748                 if (!val)
 749                     return false;
 750                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 751                     parseerror(parser, "out of memory");
 752                     goto onerr;
 753                 }
 754                 DEBUGSHUNTDO(printf("push %i\n", parser_token(parser)->constval.i));
 755             }
 756             else if (parser->tok == TOKEN_STRINGCONST) {
 757                 ast_value *val = parser_const_string(parser, parser_tokval(parser));
 758                 if (!val)
 759                     return false;
 760                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 761                     parseerror(parser, "out of memory");
 762                     goto onerr;
 763                 }
 764                 DEBUGSHUNTDO(printf("push string\n"));
 765             }
 766             else if (parser->tok == TOKEN_VECTORCONST) {
 767                 ast_value *val = parser_const_vector(parser, parser_token(parser)->constval.v);
 768                 if (!val)
 769                     return false;
 770                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 771                     parseerror(parser, "out of memory");
 772                     goto onerr;
 773                 }
 774                 DEBUGSHUNTDO(printf("push '%g %g %g'\n",
 775                                     parser_token(parser)->constval.v.x,
 776                                     parser_token(parser)->constval.v.y,
 777                                     parser_token(parser)->constval.v.z));
 778             }
 779             else if (parser->tok == '(') {
 780                 ++parens;
 781                 nextwant = false; /* not expecting an operator next */
 782                 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1, 0))) {
 783                     parseerror(parser, "out of memory");
 784                     goto onerr;
 785                 }
 786                 DEBUGSHUNTDO(printf("push (\n"));
 787             }
 788             else if (parser->tok == ')') {
 789                 DEBUGSHUNTDO(printf("do[nop] )\n"));
 790                 --parens;
 791                 if (parens < 0)
 792                     break;
 793                 /* allowed for function calls */
 794                 if (!parser_close_paren(parser, &sy, true))
 795                     goto onerr;
 796             }
 797             else {
 798                 /* TODO: prefix operators */
 799                 parseerror(parser, "expected statement");
 800                 goto onerr;
 801             }
 802             wantop = nextwant;
 803             parser->lex->flags.noops = !wantop;
 804         } else {
 805             bool nextwant = false;
 806             if (parser->tok == '(') {
 807                 DEBUGSHUNTDO(printf("push (\n"));
 808                 ++parens;
 809                 /* we expected an operator, this is the function-call operator */
 810                 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 'f', sy.out_count-1))) {
 811                     parseerror(parser, "out of memory");
 812                     goto onerr;
 813                 }
 814             }
 815             else if (parser->tok == ')') {
 816                 DEBUGSHUNTDO(printf("do[op] )\n"));
 817                 --parens;
 818                 if (parens < 0)
 819                     break;
 820                 /* we do expect an operator next */
 821                 /* closing an opening paren */
 822                 if (!parser_close_paren(parser, &sy, false))
 823                     goto onerr;
 824                 nextwant = true;
 825             }
 826             else if (parser->tok != TOKEN_OPERATOR) {
 827                 parseerror(parser, "expected operator or end of statement");
 828                 goto onerr;
 829             }
 830             else {
 831                 /* classify the operator */
 832                 /* TODO: suffix operators */
 833                 const oper_info *op;
 834                 const oper_info *olast = NULL;
 835                 size_t o;
 836                 for (o = 0; o < operator_count; ++o) {
 837                     if (!(operators[o].flags & OP_PREFIX) &&
 838                         !(operators[o].flags & OP_SUFFIX) && /* remove this */
 839                         !strcmp(parser_tokval(parser), operators[o].op))
 840                     {
 841                         break;
 842                     }
 843                 }
 844                 if (o == operator_count) {
 845                     /* no operator found... must be the end of the statement */
 846                     break;
 847                 }
 848                 /* found an operator */
 849                 op = &operators[o];
 850                 if (op->id == opid1('.')) {
 851                     /* for gmqcc standard: open up the namespace of the previous type */
 852                     ast_expression *prevex = sy.out[sy.out_count-1].out;
 853                     if (!prevex) {
 854                         parseerror(parser, "unexpected member operator");
 855                         goto onerr;
 856                     }
 857                     if (prevex->expression.vtype == TYPE_ENTITY)
 858                         parser->memberof = TYPE_ENTITY;
 859                     else if (prevex->expression.vtype == TYPE_VECTOR)
 860                         parser->memberof = TYPE_VECTOR;
 861                     else {
 862                         parseerror(parser, "type error: type has no members");
 863                         goto onerr;
 864                     }
 865                     gotmemberof = true;
 866                 }
 867
 868                 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 869                     olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 870
 871                 while (olast && (
 872                         (op->prec < olast->prec) ||
 873                         (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
 874                 {
 875                     if (!parser_sy_pop(parser, &sy))
 876                         goto onerr;
 877                     if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 878                         olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 879                     else
 880                         olast = NULL;
 881                 }
 882
 883                 DEBUGSHUNTDO(printf("push operator %s\n", op->op));
 884                 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
 885                     goto onerr;
 886             }
 887             wantop = nextwant;
 888             parser->lex->flags.noops = !wantop;
 889         }
 890         if (!parser_next(parser)) {
 891             goto onerr;
 892         }
 893         if (parser->tok == ';') {
 894             break;
 895         }
 896     }
 897     if (parens >= 0 && !parser_next(parser)) {
 898         parseerror(parser, "Unexpected end of file");
 899         goto onerr;
 900     }
 901
 902     while (sy.ops_count) {
 903         if (!parser_sy_pop(parser, &sy))
 904             goto onerr;
 905     }
 906
 907     parser->lex->flags.noops = true;
 908     if (!sy.out_count) {
 909         parseerror(parser, "empty expression");
 910         expr = NULL;
 911     } else
 912         expr = sy.out[0].out;
 913     MEM_VECTOR_CLEAR(&sy, out);
 914     MEM_VECTOR_CLEAR(&sy, ops);
 915     DEBUGSHUNTDO(printf("shut done\n"));
 916     return expr;
 917
 918 onerr:
 919     parser->lex->flags.noops = true;
 920     MEM_VECTOR_CLEAR(&sy, out);
 921     MEM_VECTOR_CLEAR(&sy, ops);
 922     return NULL;
 923 }
 924
 925 static bool parser_variable(parser_t *parser, ast_block *localblock);
 926 static ast_block* parser_parse_block(parser_t *parser);
 927 static ast_expression* parser_parse_statement_or_block(parser_t *parser);
 928 static bool parser_parse_statement(parser_t *parser, ast_block *block, ast_expression **out)
 929 {
 930     if (parser->tok == TOKEN_TYPENAME)
 931     {
 932         /* local variable */
 933         if (!block) {
 934             parseerror(parser, "cannot declare a variable from here");
 935             return false;
 936         }
 937         if (!parser_variable(parser, block))
 938             return false;
 939         *out = NULL;
 940         return true;
 941     }
 942     else if (parser->tok == TOKEN_KEYWORD)
 943     {
 944         if (!strcmp(parser_tokval(parser), "return"))
 945         {
 946             ast_expression *exp = NULL;
 947             ast_return     *ret = NULL;
 948             ast_value      *expected = parser->function->vtype;
 949
 950             if (!parser_next(parser)) {
 951                 parseerror(parser, "expected return expression");
 952                 return false;
 953             }
 954
 955             if (parser->tok != ';') {
 956                 exp = parser_expression(parser);
 957                 if (!exp)
 958                     return false;
 959
 960                 if (exp->expression.vtype != expected->expression.next->expression.vtype) {
 961                     parseerror(parser, "return with invalid expression");
 962                 }
 963
 964                 ret = ast_return_new(exp->expression.node.context, exp);
 965                 if (!ret) {
 966                     ast_delete(exp);
 967                     return false;
 968                 }
 969
 970                 *out = (ast_expression*)ret;
 971             } else if (!parser_next(parser)) {
 972                 parseerror(parser, "expected semicolon");
 973                 if (expected->expression.next->expression.vtype != TYPE_VOID) {
 974                     parseerror(parser, "return without value");
 975                 }
 976             }
 977             return true;
 978         }
 979         else if (!strcmp(parser_tokval(parser), "if"))
 980         {
 981             ast_ifthen *ifthen;
 982             ast_expression *cond, *ontrue, *onfalse = NULL;
 983
 984             lex_ctx ctx = parser_ctx(parser);
 985
 986             /* skip the 'if' and check for opening paren */
 987             if (!parser_next(parser) || parser->tok != '(') {
 988                 parseerror(parser, "expected 'if' condition in parenthesis");
 989                 return false;
 990             }
 991             /* parse into the expression */
 992             if (!parser_next(parser)) {
 993                 parseerror(parser, "expected 'if' condition after opening paren");
 994                 return false;
 995             }
 996             /* parse the condition */
 997             cond = parser_expression(parser);
 998             if (!cond)
 999                 return false;
1000             /* closing paren */
1001             if (parser->tok != ')') {
1002                 parseerror(parser, "expected closing paren after 'if' condition");
1003                 ast_delete(cond);
1004                 return false;
1005             }
1006             /* parse into the 'then' branch */
1007             if (!parser_next(parser)) {
1008                 parseerror(parser, "expected statement for on-true branch of 'if'");
1009                 ast_delete(cond);
1010                 return false;
1011             }
1012             ontrue = parser_parse_statement_or_block(parser);
1013             if (!ontrue) {
1014                 ast_delete(cond);
1015                 return false;
1016             }
1017             /* check for an else */
1018             if (!strcmp(parser_tokval(parser), "else")) {
1019                 /* parse into the 'else' branch */
1020                 if (!parser_next(parser)) {
1021                     parseerror(parser, "expected on-false branch after 'else'");
1022                     ast_delete(ontrue);
1023                     ast_delete(cond);
1024                     return false;
1025                 }
1026                 onfalse = parser_parse_statement_or_block(parser);
1027                 if (!onfalse) {
1028                     ast_delete(ontrue);
1029                     ast_delete(cond);
1030                     return false;
1031                 }
1032             }
1033
1034             ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
1035             *out = (ast_expression*)ifthen;
1036             return true;
1037         }
1038         parseerror(parser, "Unexpected keyword");
1039         return false;
1040     }
1041     else if (parser->tok == '{')
1042     {
1043         ast_block *inner;
1044         inner = parser_parse_block(parser);
1045         if (!inner)
1046             return false;
1047         *out = (ast_expression*)inner;
1048         return true;
1049     }
1050     else
1051     {
1052         ast_expression *exp = parser_expression(parser);
1053         if (!exp)
1054             return false;
1055         *out = exp;
1056         return true;
1057     }
1058 }
1059
1060 static void parser_pop_local(parser_t *parser)
1061 {
1062     parser->locals_count--;
1063     mem_d(parser->locals[parser->locals_count].name);
1064 }
1065
1066 static ast_block* parser_parse_block(parser_t *parser)
1067 {
1068     size_t oldblocklocal;
1069     ast_block *block = NULL;
1070
1071     oldblocklocal = parser->blocklocal;
1072     parser->blocklocal = parser->locals_count;
1073
1074     if (!parser_next(parser)) { /* skip the '{' */
1075         parseerror(parser, "expected function body");
1076         goto cleanup;
1077     }
1078
1079     block = ast_block_new(parser_ctx(parser));
1080
1081     while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1082     {
1083         ast_expression *expr;
1084         if (parser->tok == '}')
1085             break;
1086
1087         if (!parser_parse_statement(parser, block, &expr)) {
1088             ast_block_delete(block);
1089             block = NULL;
1090             goto cleanup;
1091         }
1092         if (!expr)
1093             continue;
1094         if (!ast_block_exprs_add(block, expr)) {
1095             ast_delete(expr);
1096             ast_block_delete(block);
1097             block = NULL;
1098             goto cleanup;
1099         }
1100     }
1101
1102     if (parser->tok != '}') {
1103         ast_block_delete(block);
1104         block = NULL;
1105     } else {
1106         (void)parser_next(parser);
1107     }
1108
1109 cleanup:
1110     while (parser->locals_count > parser->blocklocal)
1111         parser_pop_local(parser);
1112     parser->blocklocal = oldblocklocal;
1113     /* unroll the local vector */
1114     return block;
1115 }
1116
1117 static ast_expression* parser_parse_statement_or_block(parser_t *parser)
1118 {
1119     ast_expression *expr;
1120     if (parser->tok == '{')
1121         return (ast_expression*)parser_parse_block(parser);
1122     if (!parser_parse_statement(parser, NULL, &expr))
1123         return NULL;
1124     return expr;
1125 }
1126
1127 static bool parser_variable(parser_t *parser, ast_block *localblock)
1128 {
1129     bool          isfunc = false;
1130     ast_function *func = NULL;
1131     lex_ctx       ctx;
1132     ast_value    *var;
1133     varentry_t    varent;
1134
1135     int basetype = parser_token(parser)->constval.t;
1136
1137     while (true)
1138     {
1139         if (!parser_next(parser)) { /* skip basetype or comma */
1140             parseerror(parser, "expected variable declaration");
1141             return false;
1142         }
1143
1144         isfunc = false;
1145         func = NULL;
1146         ctx = parser_ctx(parser);
1147         var = parser_parse_type(parser, basetype, &isfunc);
1148
1149         if (!var)
1150             return false;
1151
1152         if (parser->tok != TOKEN_IDENT) {
1153             parseerror(parser, "expected variable name\n");
1154             return false;
1155         }
1156
1157         if (!localblock && parser_find_global(parser, parser_tokval(parser))) {
1158             ast_value_delete(var);
1159             parseerror(parser, "global already exists: %s\n", parser_tokval(parser));
1160             return false;
1161         }
1162
1163         if (localblock && parser_find_local(parser, parser_tokval(parser), parser->blocklocal)) {
1164             ast_value_delete(var);
1165             parseerror(parser, "local variable already exists: %s\n", parser_tokval(parser));
1166             return false;
1167         }
1168
1169         if (!ast_value_set_name(var, parser_tokval(parser))) {
1170             parseerror(parser, "failed to set variable name\n");
1171             ast_value_delete(var);
1172             return false;
1173         }
1174
1175         if (isfunc) {
1176             /* a function was defined */
1177             ast_value *fval;
1178
1179             /* turn var into a value of TYPE_FUNCTION, with the old var
1180              * as return type
1181              */
1182             fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1183             func = ast_function_new(ctx, var->name, fval);
1184             if (!fval || !func) {
1185                 ast_value_delete(var);
1186                 if (fval) ast_value_delete(fval);
1187                 if (func) ast_function_delete(func);
1188                 return false;
1189             }
1190
1191             fval->expression.next = (ast_expression*)var;
1192             MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1193
1194             if (!parser_t_functions_add(parser, func)) {
1195                 ast_value_delete(var);
1196                 if (fval) ast_value_delete(fval);
1197                 if (func) ast_function_delete(func);
1198                 return false;
1199             }
1200
1201             var = fval;
1202         }
1203
1204         varent.name = util_strdup(var->name);
1205         varent.var = (ast_expression*)var;
1206         if (var->expression.vtype == TYPE_VECTOR)
1207         {
1208             size_t len = strlen(varent.name);
1209             varentry_t vx, vy, vz;
1210             vx.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 0);
1211             vy.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 1);
1212             vz.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 2);
1213             vx.name = mem_a(len+3);
1214             vy.name = mem_a(len+3);
1215             vz.name = mem_a(len+3);
1216             strcpy(vx.name, varent.name);
1217             strcpy(vy.name, varent.name);
1218             strcpy(vz.name, varent.name);
1219             vx.name[len] = vy.name[len] = vz.name[len] = '_';
1220             vx.name[len+1] = 'x';
1221             vy.name[len+1] = 'y';
1222             vz.name[len+1] = 'z';
1223             vx.name[len+2] = vy.name[len+2] = vz.name[len+2] = 0;
1224
1225             if (!localblock) {
1226                 (void)!parser_t_globals_add(parser, varent);
1227                 (void)!parser_t_globals_add(parser, vx);
1228                 (void)!parser_t_globals_add(parser, vy);
1229                 (void)!parser_t_globals_add(parser, vz);
1230             } else {
1231                 (void)!parser_t_locals_add(parser, varent);
1232                 (void)!parser_t_locals_add(parser, vx);
1233                 (void)!parser_t_locals_add(parser, vy);
1234                 (void)!parser_t_locals_add(parser, vz);
1235             }
1236         }
1237         else
1238         {
1239             if ( (!localblock && !parser_t_globals_add(parser, varent)) ||
1240                  ( localblock && !parser_t_locals_add(parser, varent)) )
1241             {
1242                 ast_value_delete(var);
1243                 return false;
1244             }
1245         }
1246         if (localblock && !ast_block_locals_add(localblock, var))
1247         {
1248             parser_pop_local(parser);
1249             ast_value_delete(var);
1250             return false;
1251         }
1252
1253         if (!parser_next(parser)) {
1254             ast_value_delete(var);
1255             return false;
1256         }
1257
1258         if (parser->tok == ';') {
1259             if (!parser_next(parser))
1260                 return parser->tok == TOKEN_EOF;
1261             return true;
1262         }
1263
1264         if (parser->tok == ',') {
1265             /* another var */
1266             continue;
1267         }
1268
1269         if (parser->tok != '=') {
1270             parseerror(parser, "expected '=' or ';'");
1271             return false;
1272         }
1273
1274         if (!parser_next(parser))
1275             return false;
1276
1277         if (parser->tok == '#') {
1278             if (localblock) {
1279                 parseerror(parser, "cannot declare builtins within functions");
1280                 return false;
1281             }
1282             if (!isfunc || !func) {
1283                 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
1284                 return false;
1285             }
1286             if (!parser_next(parser)) {
1287                 parseerror(parser, "expected builtin number");
1288                 return false;
1289             }
1290             if (parser->tok != TOKEN_INTCONST) {
1291                 parseerror(parser, "builtin number must be an integer constant");
1292                 return false;
1293             }
1294             if (parser_token(parser)->constval.i <= 0) {
1295                 parseerror(parser, "builtin number must be positive integer greater than zero");
1296                 return false;
1297             }
1298
1299             func->builtin = -parser_token(parser)->constval.i;
1300         } else if (parser->tok == '{') {
1301             /* function body */
1302             ast_block *block;
1303             ast_function *old = parser->function;
1304
1305             if (localblock) {
1306                 parseerror(parser, "cannot declare functions within functions");
1307                 return false;
1308             }
1309
1310             parser->function = func;
1311             block = parser_parse_block(parser);
1312             parser->function = old;
1313
1314             if (!block)
1315                 return false;
1316
1317             if (!ast_function_blocks_add(func, block)) {
1318                 ast_block_delete(block);
1319                 return false;
1320             }
1321             return true;
1322         } else {
1323             parseerror(parser, "TODO, const assignment");
1324         }
1325
1326         if (!parser_next(parser))
1327             return false;
1328
1329         if (parser->tok == ',') {
1330             /* another */
1331             continue;
1332         }
1333
1334         if (parser->tok != ';') {
1335             parseerror(parser, "expected semicolon");
1336             return false;
1337         }
1338
1339         (void)parser_next(parser);
1340
1341         return true;
1342     }
1343 }
1344
1345 static bool parser_do(parser_t *parser)
1346 {
1347     if (parser->tok == TOKEN_TYPENAME)
1348     {
1349         return parser_variable(parser, NULL);
1350     }
1351     else if (parser->tok == TOKEN_KEYWORD)
1352     {
1353         /* handle 'var' and 'const' */
1354         return false;
1355     }
1356     else if (parser->tok == '.')
1357     {
1358         ast_value *var;
1359         ast_value *fld;
1360         bool       isfunc = false;
1361         int        basetype;
1362         lex_ctx    ctx = parser_ctx(parser);
1363         varentry_t varent;
1364
1365         /* entity-member declaration */
1366         if (!parser_next(parser) || parser->tok != TOKEN_TYPENAME) {
1367             parseerror(parser, "expected member variable definition");
1368             return false;
1369         }
1370
1371         /* remember the base/return type */
1372         basetype = parser_token(parser)->constval.t;
1373
1374         /* parse into the declaration */
1375         if (!parser_next(parser)) {
1376             parseerror(parser, "expected field def");
1377             return false;
1378         }
1379
1380         /* parse the field type fully */
1381         var = parser_parse_type(parser, basetype, &isfunc);
1382         if (!var)
1383             return false;
1384
1385         /* now the field name */
1386         if (parser->tok != TOKEN_IDENT) {
1387             parseerror(parser, "expected field name");
1388             ast_delete(var);
1389             return false;
1390         }
1391
1392         /* check for an existing field
1393          * in original qc we also have to check for an existing
1394          * global named like the field
1395          */
1396         if (opts_standard == COMPILER_QCC) {
1397             if (parser_find_global(parser, parser_tokval(parser))) {
1398                 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
1399                 ast_delete(var);
1400                 return false;
1401             }
1402         }
1403         if (parser_find_field(parser, parser_tokval(parser))) {
1404             parseerror(parser, "field %s already exists", parser_tokval(parser));
1405             ast_delete(var);
1406             return false;
1407         }
1408
1409         /* if it was a function, turn it into a function */
1410         if (isfunc) {
1411             ast_value *fval;
1412             /* turn var into a value of TYPE_FUNCTION, with the old var
1413              * as return type
1414              */
1415             fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1416             if (!fval) {
1417                 ast_value_delete(var);
1418                 ast_value_delete(fval);
1419                 return false;
1420             }
1421
1422             fval->expression.next = (ast_expression*)var;
1423             MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1424
1425             var = fval;
1426         }
1427
1428         /* turn it into a field */
1429         fld = ast_value_new(ctx, parser_tokval(parser), TYPE_FIELD);
1430         fld->expression.next = (ast_expression*)var;
1431
1432         varent.var = (ast_expression*)fld;
1433         if (var->expression.vtype == TYPE_VECTOR)
1434         {
1435             /* create _x, _y and _z fields as well */
1436             parseerror(parser, "TODO: vector field members (_x,_y,_z)");
1437             ast_delete(fld);
1438             return false;
1439         }
1440
1441         varent.name = util_strdup(fld->name);
1442         (void)!parser_t_fields_add(parser, varent);
1443
1444         /* end with a semicolon */
1445         if (!parser_next(parser) || parser->tok != ';') {
1446             parseerror(parser, "semicolon expected");
1447             return false;
1448         }
1449
1450         /* skip the semicolon */
1451         if (!parser_next(parser))
1452             return parser->tok == TOKEN_EOF;
1453
1454         return true;
1455     }
1456     else
1457     {
1458         parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
1459         return false;
1460     }
1461     return true;
1462 }
1463
1464 static parser_t *parser;
1465
1466 bool parser_init()
1467 {
1468     parser = (parser_t*)mem_a(sizeof(parser_t));
1469     if (!parser)
1470         return false;
1471
1472     memset(parser, 0, sizeof(parser));
1473
1474     MEM_VECTOR_INIT(parser, globals);
1475     MEM_VECTOR_INIT(parser, locals);
1476     return true;
1477 }
1478
1479 bool parser_compile(const char *filename)
1480 {
1481     parser->lex = lex_open(filename);
1482     if (!parser->lex) {
1483         printf("failed to open file \"%s\"\n", filename);
1484         return false;
1485     }
1486
1487     /* initial lexer/parser state */
1488     parser->lex->flags.noops = true;
1489
1490     if (parser_next(parser))
1491     {
1492         while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1493         {
1494             if (!parser_do(parser)) {
1495                 if (parser->tok == TOKEN_EOF)
1496                     parseerror(parser, "unexpected eof");
1497                 else
1498                     parseerror(parser, "parse error\n");
1499                 lex_close(parser->lex);
1500                 mem_d(parser);
1501                 return false;
1502             }
1503         }
1504     }
1505
1506     lex_close(parser->lex);
1507
1508     return !parser->errors;
1509 }
1510
1511 void parser_cleanup()
1512 {
1513     size_t i;
1514     for (i = 0; i < parser->functions_count; ++i) {
1515         ast_delete(parser->functions[i]);
1516     }
1517     for (i = 0; i < parser->imm_vector_count; ++i) {
1518         ast_delete(parser->imm_vector[i]);
1519     }
1520     for (i = 0; i < parser->imm_string_count; ++i) {
1521         ast_delete(parser->imm_string[i]);
1522     }
1523     for (i = 0; i < parser->imm_float_count; ++i) {
1524         ast_delete(parser->imm_float[i]);
1525     }
1526     for (i = 0; i < parser->globals_count; ++i) {
1527         ast_delete(parser->globals[i].var);
1528         mem_d(parser->globals[i].name);
1529     }
1530     MEM_VECTOR_CLEAR(parser, globals);
1531
1532     mem_d(parser);
1533 }
1534
1535 bool parser_finish(const char *output)
1536 {
1537     size_t i;
1538     ir_builder *ir;
1539
1540     if (!parser->errors)
1541     {
1542         ir = ir_builder_new("gmqcc_out");
1543         if (!ir) {
1544             printf("failed to allocate builder\n");
1545             return false;
1546         }
1547
1548         for (i = 0; i < parser->imm_float_count; ++i) {
1549             if (!ast_global_codegen(parser->imm_float[i], ir)) {
1550                 printf("failed to generate global %s\n", parser->imm_float[i]->name);
1551                 ir_builder_delete(ir);
1552                 return false;
1553             }
1554         }
1555         for (i = 0; i < parser->imm_string_count; ++i) {
1556             if (!ast_global_codegen(parser->imm_string[i], ir)) {
1557                 printf("failed to generate global %s\n", parser->imm_string[i]->name);
1558                 ir_builder_delete(ir);
1559                 return false;
1560             }
1561         }
1562         for (i = 0; i < parser->imm_vector_count; ++i) {
1563             if (!ast_global_codegen(parser->imm_vector[i], ir)) {
1564                 printf("failed to generate global %s\n", parser->imm_vector[i]->name);
1565                 ir_builder_delete(ir);
1566                 return false;
1567             }
1568         }
1569         for (i = 0; i < parser->fields_count; ++i) {
1570             ast_value *field;
1571             bool isconst;
1572             if (!ast_istype(parser->fields[i].var, ast_value))
1573                 continue;
1574             field = (ast_value*)parser->fields[i].var;
1575             isconst = field->isconst;
1576             field->isconst = false;
1577             if (!ast_global_codegen((ast_value*)field, ir)) {
1578                 printf("failed to generate field %s\n", field->name);
1579                 ir_builder_delete(ir);
1580                 return false;
1581             }
1582             if (isconst) {
1583                 ir_value *ifld;
1584                 ast_expression *subtype;
1585                 field->isconst = true;
1586                 subtype = field->expression.next;
1587                 ifld = ir_builder_create_field(ir, field->name, subtype->expression.vtype);
1588                 if (subtype->expression.vtype == TYPE_FIELD)
1589                     ifld->fieldtype = subtype->expression.next->expression.vtype;
1590                 else if (subtype->expression.vtype == TYPE_FUNCTION)
1591                     ifld->outtype = subtype->expression.next->expression.vtype;
1592                 (void)!ir_value_set_field(field->ir_v, ifld);
1593             }
1594         }
1595         for (i = 0; i < parser->globals_count; ++i) {
1596             if (!ast_istype(parser->globals[i].var, ast_value))
1597                 continue;
1598             if (!ast_global_codegen((ast_value*)(parser->globals[i].var), ir)) {
1599                 printf("failed to generate global %s\n", parser->globals[i].name);
1600                 ir_builder_delete(ir);
1601                 return false;
1602             }
1603         }
1604         for (i = 0; i < parser->functions_count; ++i) {
1605             if (!ast_function_codegen(parser->functions[i], ir)) {
1606                 printf("failed to generate function %s\n", parser->functions[i]->name);
1607                 ir_builder_delete(ir);
1608                 return false;
1609             }
1610             if (!ir_function_finalize(parser->functions[i]->ir_func)) {
1611                 printf("failed to finalize function %s\n", parser->functions[i]->name);
1612                 ir_builder_delete(ir);
1613                 return false;
1614             }
1615         }
1616
1617         ir_builder_dump(ir, printf);
1618
1619         if (!ir_builder_generate(ir, output)) {
1620             printf("*** failed to generate output file\n");
1621             ir_builder_delete(ir);
1622             return false;
1623         }
1624
1625         ir_builder_delete(ir);
1626         return true;
1627     }
1628
1629     printf("*** there were compile errors\n");
1630     return false;
1631 }