]> git.xonotic.org Git - xonotic/gmqcc.git/blob - parser.c
a2b3f060eed51e5edc7bbd7f1876b2fa7fd6f686
[xonotic/gmqcc.git] / 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_value *imm_float_zero;
24     ast_value *imm_vector_zero;
25
26     ast_function *function;
27     MEM_VECTOR_MAKE(varentry_t, locals);
28     size_t blocklocal;
29
30     size_t errors;
31
32     /* TYPE_FIELD -> parser_find_fields is used instead of find_var
33      * TODO: TYPE_VECTOR -> x, y and z are accepted in the gmqcc standard
34      * anything else: type error
35      */
36     qcint  memberof;
37 } parser_t;
38
39 MEM_VEC_FUNCTIONS(parser_t, varentry_t, globals)
40 MEM_VEC_FUNCTIONS(parser_t, varentry_t, fields)
41 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_float)
42 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_string)
43 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_vector)
44 MEM_VEC_FUNCTIONS(parser_t, varentry_t, locals)
45 MEM_VEC_FUNCTIONS(parser_t, ast_function*, functions)
46
47 static void parser_pop_local(parser_t *parser);
48 static bool parser_variable(parser_t *parser, ast_block *localblock);
49 static ast_block* parser_parse_block(parser_t *parser);
50 static ast_expression* parser_parse_statement_or_block(parser_t *parser);
51 static ast_expression* parser_expression_leave(parser_t *parser, bool stopatcomma);
52 static ast_expression* parser_expression(parser_t *parser, bool stopatcomma);
53
54 void parseerror(parser_t *parser, const char *fmt, ...)
55 {
56         va_list ap;
57
58         parser->errors++;
59
60         va_start(ap, fmt);
61     vprintmsg(LVL_ERROR, parser->lex->tok->ctx.file, parser->lex->tok->ctx.line, "parse error", fmt, ap);
62         va_end(ap);
63 }
64
65 /* returns true if it counts as an error */
66 bool GMQCC_WARN parsewarning(parser_t *parser, int warntype, const char *fmt, ...)
67 {
68         va_list ap;
69         int lvl = LVL_WARNING;
70
71     if (!OPTS_WARN(warntype))
72         return false;
73
74     if (opts_werror) {
75             parser->errors++;
76             lvl = LVL_ERROR;
77         }
78
79         va_start(ap, fmt);
80     vprintmsg(lvl, parser->lex->tok->ctx.file, parser->lex->tok->ctx.line, "warning", fmt, ap);
81         va_end(ap);
82
83         return opts_werror;
84 }
85
86 /**********************************************************************
87  * some maths used for constant folding
88  */
89
90 vector vec3_add(vector a, vector b)
91 {
92     vector out;
93     out.x = a.x + b.x;
94     out.y = a.y + b.y;
95     out.z = a.z + b.z;
96     return out;
97 }
98
99 vector vec3_sub(vector a, vector b)
100 {
101     vector out;
102     out.x = a.x - b.x;
103     out.y = a.y - b.y;
104     out.z = a.z - b.z;
105     return out;
106 }
107
108 qcfloat vec3_mulvv(vector a, vector b)
109 {
110     return (a.x * b.x + a.y * b.y + a.z * b.z);
111 }
112
113 vector vec3_mulvf(vector a, float b)
114 {
115     vector out;
116     out.x = a.x * b;
117     out.y = a.y * b;
118     out.z = a.z * b;
119     return out;
120 }
121
122 /**********************************************************************
123  * parsing
124  */
125
126 bool parser_next(parser_t *parser)
127 {
128     /* lex_do kills the previous token */
129     parser->tok = lex_do(parser->lex);
130     if (parser->tok == TOKEN_EOF || parser->tok >= TOKEN_ERROR)
131         return false;
132     return true;
133 }
134
135 /* lift a token out of the parser so it's not destroyed by parser_next */
136 token *parser_lift(parser_t *parser)
137 {
138     token *tok = parser->lex->tok;
139     parser->lex->tok = NULL;
140     return tok;
141 }
142
143 #define parser_tokval(p) (p->lex->tok->value)
144 #define parser_token(p)  (p->lex->tok)
145 #define parser_ctx(p)    (p->lex->tok->ctx)
146
147 ast_value* parser_const_float(parser_t *parser, double d)
148 {
149     size_t i;
150     ast_value *out;
151     for (i = 0; i < parser->imm_float_count; ++i) {
152         if (parser->imm_float[i]->constval.vfloat == d)
153             return parser->imm_float[i];
154     }
155     out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_FLOAT);
156     out->isconst = true;
157     out->constval.vfloat = d;
158     if (!parser_t_imm_float_add(parser, out)) {
159         ast_value_delete(out);
160         return NULL;
161     }
162     return out;
163 }
164
165 ast_value* parser_const_float_0(parser_t *parser)
166 {
167     if (!parser->imm_float_zero)
168         parser->imm_float_zero = parser_const_float(parser, 0);
169     return parser->imm_float_zero;
170 }
171
172 char *parser_strdup(const char *str)
173 {
174     if (str && !*str) {
175         /* actually dup empty strings */
176         char *out = mem_a(1);
177         *out = 0;
178         return out;
179     }
180     return util_strdup(str);
181 }
182
183 ast_value* parser_const_string(parser_t *parser, const char *str)
184 {
185     size_t i;
186     ast_value *out;
187     for (i = 0; i < parser->imm_string_count; ++i) {
188         if (!strcmp(parser->imm_string[i]->constval.vstring, str))
189             return parser->imm_string[i];
190     }
191     out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_STRING);
192     out->isconst = true;
193     out->constval.vstring = parser_strdup(str);
194     if (!parser_t_imm_string_add(parser, out)) {
195         ast_value_delete(out);
196         return NULL;
197     }
198     return out;
199 }
200
201 ast_value* parser_const_vector(parser_t *parser, vector v)
202 {
203     size_t i;
204     ast_value *out;
205     for (i = 0; i < parser->imm_vector_count; ++i) {
206         if (!memcmp(&parser->imm_vector[i]->constval.vvec, &v, sizeof(v)))
207             return parser->imm_vector[i];
208     }
209     out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_VECTOR);
210     out->isconst = true;
211     out->constval.vvec = v;
212     if (!parser_t_imm_vector_add(parser, out)) {
213         ast_value_delete(out);
214         return NULL;
215     }
216     return out;
217 }
218
219 ast_value* parser_const_vector_f(parser_t *parser, float x, float y, float z)
220 {
221     vector v;
222     v.x = x;
223     v.y = y;
224     v.z = z;
225     return parser_const_vector(parser, v);
226 }
227
228 ast_value* parser_const_vector_0(parser_t *parser)
229 {
230     if (!parser->imm_vector_zero)
231         parser->imm_vector_zero = parser_const_vector_f(parser, 0, 0, 0);
232     return parser->imm_vector_zero;
233 }
234
235 ast_expression* parser_find_field(parser_t *parser, const char *name)
236 {
237     size_t i;
238     for (i = 0; i < parser->fields_count; ++i) {
239         if (!strcmp(parser->fields[i].name, name))
240             return parser->fields[i].var;
241     }
242     return NULL;
243 }
244
245 ast_expression* parser_find_global(parser_t *parser, const char *name)
246 {
247     size_t i;
248     for (i = 0; i < parser->globals_count; ++i) {
249         if (!strcmp(parser->globals[i].name, name))
250             return parser->globals[i].var;
251     }
252     return NULL;
253 }
254
255 ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto)
256 {
257     size_t i;
258     ast_value *fun;
259     for (i = parser->locals_count; i > upto;) {
260         --i;
261         if (!strcmp(parser->locals[i].name, name))
262             return parser->locals[i].var;
263     }
264     if (!parser->function)
265         return NULL;
266     fun = parser->function->vtype;
267     for (i = 0; i < fun->expression.params_count; ++i) {
268         if (!strcmp(fun->expression.params[i]->name, name))
269             return (ast_expression*)(fun->expression.params[i]);
270     }
271     return NULL;
272 }
273
274 ast_expression* parser_find_var(parser_t *parser, const char *name)
275 {
276     ast_expression *v;
277     v         = parser_find_local(parser, name, 0);
278     if (!v) v = parser_find_global(parser, name);
279     return v;
280 }
281
282 typedef struct {
283     MEM_VECTOR_MAKE(ast_value*, p);
284 } paramlist_t;
285 MEM_VEC_FUNCTIONS(paramlist_t, ast_value*, p)
286
287 static ast_value *parser_parse_type(parser_t *parser, int basetype, bool *isfunc)
288 {
289     paramlist_t params;
290     ast_value *var;
291     lex_ctx   ctx = parser_ctx(parser);
292     int vtype = basetype;
293     int temptype;
294     size_t i;
295
296     MEM_VECTOR_INIT(&params, p);
297
298     *isfunc = false;
299
300     if (parser->tok == '(') {
301         *isfunc = true;
302         while (true) {
303             ast_value *param;
304             ast_value *fld;
305             bool isfield = false;
306             bool dummy;
307
308             if (!parser_next(parser))
309                 goto on_error;
310
311             if (parser->tok == ')')
312                 break;
313
314             if (parser->tok == '.') {
315                 isfield = true;
316                 if (!parser_next(parser)) {
317                     parseerror(parser, "expected field parameter type");
318                     goto on_error;
319                 }
320             }
321
322             temptype = parser_token(parser)->constval.t;
323             if (!parser_next(parser))
324                 goto on_error;
325
326             param = parser_parse_type(parser, temptype, &dummy);
327             (void)dummy;
328
329             if (!param)
330                 goto on_error;
331
332             if (parser->tok == TOKEN_IDENT) {
333                 /* named parameter */
334                 if (!ast_value_set_name(param, parser_tokval(parser)))
335                     goto on_error;
336                 if (!parser_next(parser))
337                     goto on_error;
338             }
339
340             if (isfield) {
341                 fld = ast_value_new(ctx, param->name, TYPE_FIELD);
342                 fld->expression.next = (ast_expression*)param;
343                 param = fld;
344             }
345
346             if (!paramlist_t_p_add(&params, param)) {
347                 parseerror(parser, "Out of memory while parsing typename");
348                 goto on_error;
349             }
350
351             if (parser->tok == ',')
352                 continue;
353             if (parser->tok == ')')
354                 break;
355             parseerror(parser, "Unexpected token");
356             goto on_error;
357         }
358         if (!parser_next(parser))
359             goto on_error;
360     }
361
362     var = ast_value_new(ctx, "<unnamed>", vtype);
363     if (!var)
364         goto on_error;
365     MEM_VECTOR_MOVE(&params, p, &var->expression, params);
366     return var;
367 on_error:
368     for (i = 0; i < params.p_count; ++i)
369         ast_value_delete(params.p[i]);
370     MEM_VECTOR_CLEAR(&params, p);
371     return NULL;
372 }
373
374 typedef struct
375 {
376     size_t etype; /* 0 = expression, others are operators */
377     int             paren;
378     size_t          off;
379     ast_expression *out;
380     ast_block      *block; /* for commas and function calls */
381     lex_ctx ctx;
382 } sy_elem;
383 typedef struct
384 {
385     MEM_VECTOR_MAKE(sy_elem, out);
386     MEM_VECTOR_MAKE(sy_elem, ops);
387 } shunt;
388 MEM_VEC_FUNCTIONS(shunt, sy_elem, out)
389 MEM_VEC_FUNCTIONS(shunt, sy_elem, ops)
390
391 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
392     sy_elem e;
393     e.etype = 0;
394     e.out   = v;
395     e.block = NULL;
396     e.ctx   = ctx;
397     e.paren = 0;
398     return e;
399 }
400
401 static sy_elem syblock(lex_ctx ctx, ast_block *v) {
402     sy_elem e;
403     e.etype = 0;
404     e.out   = (ast_expression*)v;
405     e.block = v;
406     e.ctx   = ctx;
407     e.paren = 0;
408     return e;
409 }
410
411 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
412     sy_elem e;
413     e.etype = 1 + (op - operators);
414     e.out   = NULL;
415     e.block = NULL;
416     e.ctx   = ctx;
417     e.paren = 0;
418     return e;
419 }
420
421 static sy_elem syparen(lex_ctx ctx, int p, size_t off) {
422     sy_elem e;
423     e.etype = 0;
424     e.off   = off;
425     e.out   = NULL;
426     e.block = NULL;
427     e.ctx   = ctx;
428     e.paren = p;
429     return e;
430 }
431
432 #ifdef DEBUGSHUNT
433 # define DEBUGSHUNTDO(x) x
434 #else
435 # define DEBUGSHUNTDO(x)
436 #endif
437
438 static bool parser_sy_pop(parser_t *parser, shunt *sy)
439 {
440     const oper_info *op;
441     lex_ctx ctx;
442     ast_expression *out = NULL;
443     ast_expression *exprs[3];
444     ast_block      *blocks[3];
445     ast_value      *asvalue[3];
446     size_t i, assignop;
447     qcint  generated_op = 0;
448
449     if (!sy->ops_count) {
450         parseerror(parser, "internal error: missing operator");
451         return false;
452     }
453
454     if (sy->ops[sy->ops_count-1].paren) {
455         parseerror(parser, "unmatched parenthesis");
456         return false;
457     }
458
459     op = &operators[sy->ops[sy->ops_count-1].etype - 1];
460     ctx = sy->ops[sy->ops_count-1].ctx;
461
462     DEBUGSHUNTDO(printf("apply %s\n", op->op));
463
464     if (sy->out_count < op->operands) {
465         parseerror(parser, "internal error: not enough operands: %i (operator %s (%i))", sy->out_count,
466                    op->op, (int)op->id);
467         return false;
468     }
469
470     sy->ops_count--;
471
472     sy->out_count -= op->operands;
473     for (i = 0; i < op->operands; ++i) {
474         exprs[i]  = sy->out[sy->out_count+i].out;
475         blocks[i] = sy->out[sy->out_count+i].block;
476         asvalue[i] = (ast_value*)exprs[i];
477     }
478
479     if (blocks[0] && !blocks[0]->exprs_count && op->id != opid1(',')) {
480         parseerror(parser, "internal error: operator cannot be applied on empty blocks");
481         return false;
482     }
483
484 #define NotSameType(T) \
485              (exprs[0]->expression.vtype != exprs[1]->expression.vtype || \
486               exprs[0]->expression.vtype != T)
487 #define CanConstFold1(A) \
488              (ast_istype((A), ast_value) && ((ast_value*)(A))->isconst)
489 #define CanConstFold(A, B) \
490              (CanConstFold1(A) && CanConstFold1(B))
491 #define ConstV(i) (asvalue[(i)]->constval.vvec)
492 #define ConstF(i) (asvalue[(i)]->constval.vfloat)
493 #define ConstS(i) (asvalue[(i)]->constval.vstring)
494     switch (op->id)
495     {
496         default:
497             parseerror(parser, "internal error: unhandled operator: %s (%i)", op->op, (int)op->id);
498             return false;
499
500         case opid1('.'):
501             if (exprs[0]->expression.vtype == TYPE_ENTITY) {
502                 if (exprs[1]->expression.vtype != TYPE_FIELD) {
503                     parseerror(parser, "type error: right hand of member-operand should be an entity-field");
504                     return false;
505                 }
506                 out = (ast_expression*)ast_entfield_new(ctx, exprs[0], exprs[1]);
507             }
508             else if (exprs[0]->expression.vtype == TYPE_VECTOR) {
509                 parseerror(parser, "internal error: vector access is not supposed to be handled at this point");
510                 return false;
511             }
512             else {
513                 parseerror(parser, "type error: member-of operator on something that is not an entity or vector");
514                 return false;
515             }
516             break;
517
518         case opid1(','):
519             if (blocks[0]) {
520                 if (!ast_block_exprs_add(blocks[0], exprs[1]))
521                     return false;
522             } else {
523                 blocks[0] = ast_block_new(ctx);
524                 if (!ast_block_exprs_add(blocks[0], exprs[0]) ||
525                     !ast_block_exprs_add(blocks[0], exprs[1]))
526                 {
527                     return false;
528                 }
529             }
530             if (!ast_block_set_type(blocks[0], exprs[1]))
531                 return false;
532
533             sy->out[sy->out_count++] = syblock(ctx, blocks[0]);
534             return true;
535
536         case opid2('-','P'):
537             switch (exprs[0]->expression.vtype) {
538                 case TYPE_FLOAT:
539                     if (CanConstFold1(exprs[0]))
540                         out = (ast_expression*)parser_const_float(parser, -ConstF(0));
541                     else
542                         out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F,
543                                                               (ast_expression*)parser_const_float_0(parser),
544                                                               exprs[0]);
545                     break;
546                 case TYPE_VECTOR:
547                     if (CanConstFold1(exprs[0]))
548                         out = (ast_expression*)parser_const_vector_f(parser,
549                             -ConstV(0).x, -ConstV(0).y, -ConstV(0).z);
550                     else
551                         out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V,
552                                                               (ast_expression*)parser_const_vector_0(parser),
553                                                               exprs[0]);
554                     break;
555                 default:
556                 parseerror(parser, "invalid types used in expression: cannot negate type %s",
557                            type_name[exprs[0]->expression.vtype]);
558                 return false;
559             }
560             break;
561
562         case opid2('!','P'):
563             switch (exprs[0]->expression.vtype) {
564                 case TYPE_FLOAT:
565                     if (CanConstFold1(exprs[0]))
566                         out = (ast_expression*)parser_const_float(parser, !ConstF(0));
567                     else
568                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, exprs[0]);
569                     break;
570                 case TYPE_VECTOR:
571                     if (CanConstFold1(exprs[0]))
572                         out = (ast_expression*)parser_const_float(parser,
573                             (!ConstV(0).x && !ConstV(0).y && !ConstV(0).z));
574                     else
575                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_V, exprs[0]);
576                     break;
577                 case TYPE_STRING:
578                     if (CanConstFold1(exprs[0]))
579                         out = (ast_expression*)parser_const_float(parser, !ConstS(0) || !*ConstS(0));
580                     else
581                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_S, exprs[0]);
582                     break;
583                 /* we don't constant-fold NOT for these types */
584                 case TYPE_ENTITY:
585                     out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_ENT, exprs[0]);
586                     break;
587                 case TYPE_FUNCTION:
588                     out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_FNC, exprs[0]);
589                     break;
590                 default:
591                 parseerror(parser, "invalid types used in expression: cannot logically negate type %s",
592                            type_name[exprs[0]->expression.vtype]);
593                 return false;
594             }
595             break;
596
597         case opid1('+'):
598             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
599                 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
600             {
601                 parseerror(parser, "invalid types used in expression: cannot add type %s and %s",
602                            type_name[exprs[0]->expression.vtype],
603                            type_name[exprs[1]->expression.vtype]);
604                 return false;
605             }
606             switch (exprs[0]->expression.vtype) {
607                 case TYPE_FLOAT:
608                     if (CanConstFold(exprs[0], exprs[1]))
609                     {
610                         out = (ast_expression*)parser_const_float(parser, ConstF(0) + ConstF(1));
611                     }
612                     else
613                         out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
614                     break;
615                 case TYPE_VECTOR:
616                     if (CanConstFold(exprs[0], exprs[1]))
617                         out = (ast_expression*)parser_const_vector(parser, vec3_add(ConstV(0), ConstV(1)));
618                     else
619                         out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
620                     break;
621                 default:
622                     parseerror(parser, "invalid types used in expression: cannot add type %s and %s",
623                                type_name[exprs[0]->expression.vtype],
624                                type_name[exprs[1]->expression.vtype]);
625                     return false;
626             };
627             break;
628         case opid1('-'):
629             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
630                 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
631             {
632                 parseerror(parser, "invalid types used in expression: cannot subtract type %s from %s",
633                            type_name[exprs[1]->expression.vtype],
634                            type_name[exprs[0]->expression.vtype]);
635                 return false;
636             }
637             switch (exprs[0]->expression.vtype) {
638                 case TYPE_FLOAT:
639                     if (CanConstFold(exprs[0], exprs[1]))
640                         out = (ast_expression*)parser_const_float(parser, ConstF(0) - ConstF(1));
641                     else
642                         out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
643                     break;
644                 case TYPE_VECTOR:
645                     if (CanConstFold(exprs[0], exprs[1]))
646                         out = (ast_expression*)parser_const_vector(parser, vec3_sub(ConstV(0), ConstV(1)));
647                     else
648                         out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
649                     break;
650                 default:
651                     parseerror(parser, "invalid types used in expression: cannot subtract type %s from %s",
652                                type_name[exprs[1]->expression.vtype],
653                                type_name[exprs[0]->expression.vtype]);
654                     return false;
655             };
656             break;
657         case opid1('*'):
658             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
659                 exprs[0]->expression.vtype != TYPE_VECTOR &&
660                 exprs[0]->expression.vtype != TYPE_FLOAT &&
661                 exprs[1]->expression.vtype != TYPE_VECTOR &&
662                 exprs[1]->expression.vtype != TYPE_FLOAT)
663             {
664                 parseerror(parser, "invalid types used in expression: cannot multiply types %s and %s",
665                            type_name[exprs[1]->expression.vtype],
666                            type_name[exprs[0]->expression.vtype]);
667                 return false;
668             }
669             switch (exprs[0]->expression.vtype) {
670                 case TYPE_FLOAT:
671                     if (exprs[1]->expression.vtype == TYPE_VECTOR)
672                     {
673                         if (CanConstFold(exprs[0], exprs[1]))
674                             out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(1), ConstF(0)));
675                         else
676                             out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
677                     }
678                     else
679                     {
680                         if (CanConstFold(exprs[0], exprs[1]))
681                             out = (ast_expression*)parser_const_float(parser, ConstF(0) * ConstF(1));
682                         else
683                             out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
684                     }
685                     break;
686                 case TYPE_VECTOR:
687                     if (exprs[1]->expression.vtype == TYPE_FLOAT)
688                     {
689                         if (CanConstFold(exprs[0], exprs[1]))
690                             out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(0), ConstF(1)));
691                         else
692                             out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
693                     }
694                     else
695                     {
696                         if (CanConstFold(exprs[0], exprs[1]))
697                             out = (ast_expression*)parser_const_float(parser, vec3_mulvv(ConstV(0), ConstV(1)));
698                         else
699                             out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
700                     }
701                     break;
702                 default:
703                     parseerror(parser, "invalid types used in expression: cannot multiply types %s and %s",
704                                type_name[exprs[1]->expression.vtype],
705                                type_name[exprs[0]->expression.vtype]);
706                     return false;
707             };
708             break;
709         case opid1('/'):
710             if (NotSameType(TYPE_FLOAT)) {
711                 parseerror(parser, "invalid types used in expression: cannot divide types %s and %s",
712                            type_name[exprs[0]->expression.vtype],
713                            type_name[exprs[1]->expression.vtype]);
714                 return false;
715             }
716             if (CanConstFold(exprs[0], exprs[1]))
717                 out = (ast_expression*)parser_const_float(parser, ConstF(0) / ConstF(1));
718             else
719                 out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
720             break;
721         case opid1('%'):
722         case opid2('%','='):
723             parseerror(parser, "qc does not have a modulo operator");
724             return false;
725         case opid1('|'):
726         case opid1('&'):
727             if (NotSameType(TYPE_FLOAT)) {
728                 parseerror(parser, "invalid types used in expression: cannot perform bit operations between types %s and %s",
729                            type_name[exprs[0]->expression.vtype],
730                            type_name[exprs[1]->expression.vtype]);
731                 return false;
732             }
733             if (CanConstFold(exprs[0], exprs[1]))
734                 out = (ast_expression*)parser_const_float(parser,
735                     (op->id == opid1('|') ? (float)( ((qcint)ConstF(0)) | ((qcint)ConstF(1)) ) :
736                                             (float)( ((qcint)ConstF(0)) & ((qcint)ConstF(1)) ) ));
737             else
738                 out = (ast_expression*)ast_binary_new(ctx,
739                     (op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
740                     exprs[0], exprs[1]);
741             break;
742         case opid1('^'):
743             parseerror(parser, "TODO: bitxor");
744             return false;
745
746         case opid2('<','<'):
747         case opid2('>','>'):
748         case opid3('<','<','='):
749         case opid3('>','>','='):
750             parseerror(parser, "TODO: shifts");
751             return false;
752
753         case opid2('|','|'):
754             generated_op += 1; /* INSTR_OR */
755         case opid2('&','&'):
756             generated_op += INSTR_AND;
757             if (NotSameType(TYPE_FLOAT)) {
758                 parseerror(parser, "invalid types used in expression: cannot perform logical operations between types %s and %s",
759                            type_name[exprs[0]->expression.vtype],
760                            type_name[exprs[1]->expression.vtype]);
761                 parseerror(parser, "TODO: logical ops for arbitrary types using INSTR_NOT");
762                 parseerror(parser, "TODO: optional early out");
763                 return false;
764             }
765             if (opts_standard == COMPILER_GMQCC)
766                 printf("TODO: early out logic\n");
767             if (CanConstFold(exprs[0], exprs[1]))
768                 out = (ast_expression*)parser_const_float(parser,
769                     (generated_op == INSTR_OR ? (ConstF(0) || ConstF(1)) : (ConstF(0) && ConstF(1))));
770             else
771                 out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
772             break;
773
774         case opid1('>'):
775             generated_op += 1; /* INSTR_GT */
776         case opid1('<'):
777             generated_op += 1; /* INSTR_LT */
778         case opid2('>', '='):
779             generated_op += 1; /* INSTR_GE */
780         case opid2('<', '='):
781             generated_op += INSTR_LE;
782             if (NotSameType(TYPE_FLOAT)) {
783                 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
784                            type_name[exprs[0]->expression.vtype],
785                            type_name[exprs[1]->expression.vtype]);
786                 return false;
787             }
788             out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
789             break;
790         case opid2('!', '='):
791             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
792                 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
793                            type_name[exprs[0]->expression.vtype],
794                            type_name[exprs[1]->expression.vtype]);
795                 return false;
796             }
797             out = (ast_expression*)ast_binary_new(ctx, type_ne_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
798             break;
799         case opid2('=', '='):
800             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
801                 parseerror(parser, "invalid types used in expression: cannot perform comparison between types %s and %s",
802                            type_name[exprs[0]->expression.vtype],
803                            type_name[exprs[1]->expression.vtype]);
804                 return false;
805             }
806             out = (ast_expression*)ast_binary_new(ctx, type_eq_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
807             break;
808
809         case opid1('='):
810             if (ast_istype(exprs[0], ast_entfield))
811                 assignop = type_storep_instr[exprs[0]->expression.vtype];
812             else
813                 assignop = type_store_instr[exprs[0]->expression.vtype];
814             out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]);
815             break;
816         case opid2('+','='):
817         case opid2('-','='):
818             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
819                 (exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
820             {
821                 parseerror(parser, "invalid types used in expression: cannot add or subtract type %s and %s",
822                            type_name[exprs[0]->expression.vtype],
823                            type_name[exprs[1]->expression.vtype]);
824                 return false;
825             }
826             if (ast_istype(exprs[0], ast_entfield))
827                 assignop = type_storep_instr[exprs[0]->expression.vtype];
828             else
829                 assignop = type_store_instr[exprs[0]->expression.vtype];
830             switch (exprs[0]->expression.vtype) {
831                 case TYPE_FLOAT:
832                     out = (ast_expression*)ast_binstore_new(ctx, assignop,
833                                                             (op->id == opid2('+','=') ? INSTR_ADD_F : INSTR_SUB_F),
834                                                             exprs[0], exprs[1]);
835                     break;
836                 case TYPE_VECTOR:
837                     out = (ast_expression*)ast_binstore_new(ctx, assignop,
838                                                             (op->id == opid2('+','=') ? INSTR_ADD_V : INSTR_SUB_V),
839                                                             exprs[0], exprs[1]);
840                     break;
841                 default:
842                     parseerror(parser, "invalid types used in expression: cannot add or subtract type %s and %s",
843                                type_name[exprs[0]->expression.vtype],
844                                type_name[exprs[1]->expression.vtype]);
845                     return false;
846             };
847             break;
848     }
849 #undef NotSameType
850
851     if (!out) {
852         parseerror(parser, "failed to apply operand %s", op->op);
853         return false;
854     }
855
856     DEBUGSHUNTDO(printf("applied %s\n", op->op));
857     sy->out[sy->out_count++] = syexp(ctx, out);
858     return true;
859 }
860
861 static bool parser_close_call(parser_t *parser, shunt *sy)
862 {
863     /* was a function call */
864     ast_expression *fun;
865     ast_call       *call;
866
867     size_t          fid;
868     size_t          paramcount;
869
870     sy->ops_count--;
871     fid = sy->ops[sy->ops_count].off;
872
873     /* out[fid] is the function
874      * everything above is parameters...
875      * 0 params = nothing
876      * 1 params = ast_expression
877      * more = ast_block
878      */
879
880     if (sy->out_count < 1 || sy->out_count <= fid) {
881         parseerror(parser, "internal error: function call needs function and parameter list...");
882         return false;
883     }
884
885     fun = sy->out[fid].out;
886
887     call = ast_call_new(sy->ops[sy->ops_count].ctx, fun);
888     if (!call) {
889         parseerror(parser, "out of memory");
890         return false;
891     }
892
893     if (fid+1 == sy->out_count) {
894         /* no arguments */
895         paramcount = 0;
896     } else if (fid+2 == sy->out_count) {
897         ast_block *params;
898         sy->out_count--;
899         params = sy->out[sy->out_count].block;
900         if (!params) {
901             /* 1 param */
902             paramcount = 1;
903             if (!ast_call_params_add(call, sy->out[sy->out_count].out)) {
904                 ast_delete(sy->out[sy->out_count].out);
905                 parseerror(parser, "out of memory");
906                 return false;
907             }
908         } else {
909             paramcount = params->exprs_count;
910             MEM_VECTOR_MOVE(params, exprs, call, params);
911             ast_delete(params);
912         }
913     } else {
914         parseerror(parser, "invalid function call");
915         return false;
916     }
917
918     /* overwrite fid, the function, with a call */
919     sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
920
921     if (fun->expression.vtype != TYPE_FUNCTION) {
922         parseerror(parser, "not a function (%s)", type_name[fun->expression.vtype]);
923         return false;
924     }
925
926     if (!fun->expression.next) {
927         parseerror(parser, "could not determine function return type");
928         return false;
929     } else {
930         if (fun->expression.params_count != paramcount) {
931             parseerror(parser, "expected %i parameters, got %i", (int)fun->expression.params_count, paramcount);
932             return false;
933         }
934     }
935
936     return true;
937 }
938
939 static bool parser_close_paren(parser_t *parser, shunt *sy, bool functions_only)
940 {
941     if (!sy->ops_count) {
942         parseerror(parser, "unmatched closing paren");
943         return false;
944     }
945     if (sy->ops[sy->ops_count-1].paren == 1) {
946         parseerror(parser, "empty parenthesis expression");
947         return false;
948     }
949     while (sy->ops_count) {
950         if (sy->ops[sy->ops_count-1].paren == 'f') {
951             if (!parser_close_call(parser, sy))
952                 return false;
953             break;
954         }
955         if (sy->ops[sy->ops_count-1].paren == 1) {
956             sy->ops_count--;
957             return !functions_only;
958         }
959         if (!parser_sy_pop(parser, sy))
960             return false;
961     }
962     return true;
963 }
964
965 static void parser_reclassify_token(parser_t *parser)
966 {
967     size_t i;
968     for (i = 0; i < operator_count; ++i) {
969         if (!strcmp(parser_tokval(parser), operators[i].op)) {
970             parser->tok = TOKEN_OPERATOR;
971             return;
972         }
973     }
974 }
975
976 static ast_expression* parser_expression_leave(parser_t *parser, bool stopatcomma)
977 {
978     ast_expression *expr = NULL;
979     shunt sy;
980     bool wantop = false;
981     bool gotmemberof = false;
982
983     /* count the parens because an if starts with one, so the
984      * end of a condition is an unmatched closing paren
985      */
986     int parens = 0;
987
988     MEM_VECTOR_INIT(&sy, out);
989     MEM_VECTOR_INIT(&sy, ops);
990
991     parser->lex->flags.noops = false;
992
993     parser_reclassify_token(parser);
994
995     while (true)
996     {
997         if (gotmemberof)
998             gotmemberof = false;
999         else
1000             parser->memberof = 0;
1001
1002         if (parser->tok == TOKEN_IDENT)
1003         {
1004             ast_expression *var;
1005             if (wantop) {
1006                 parseerror(parser, "expected operator or end of statement");
1007                 goto onerr;
1008             }
1009             wantop = true;
1010             /* variable */
1011             if (opts_standard == COMPILER_GMQCC)
1012             {
1013                 if (parser->memberof == TYPE_ENTITY) {
1014                     /* still get vars first since there could be a fieldpointer */
1015                     var = parser_find_var(parser, parser_tokval(parser));
1016                     if (!var)
1017                         var = parser_find_field(parser, parser_tokval(parser));
1018                 }
1019                 else if (parser->memberof == TYPE_VECTOR)
1020                 {
1021                     parseerror(parser, "TODO: implement effective vector member access");
1022                     goto onerr;
1023                 }
1024                 else if (parser->memberof) {
1025                     parseerror(parser, "namespace for member not found");
1026                     goto onerr;
1027                 }
1028                 else
1029                     var = parser_find_var(parser, parser_tokval(parser));
1030             } else {
1031                 var = parser_find_var(parser, parser_tokval(parser));
1032                 if (!var)
1033                     var = parser_find_field(parser, parser_tokval(parser));
1034             }
1035             if (!var) {
1036                 parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
1037                 goto onerr;
1038             }
1039             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), var))) {
1040                 parseerror(parser, "out of memory");
1041                 goto onerr;
1042             }
1043             DEBUGSHUNTDO(printf("push %s\n", parser_tokval(parser)));
1044         }
1045         else if (parser->tok == TOKEN_FLOATCONST) {
1046             ast_value *val;
1047             if (wantop) {
1048                 parseerror(parser, "expected operator or end of statement, got constant");
1049                 goto onerr;
1050             }
1051             wantop = true;
1052             val = parser_const_float(parser, (parser_token(parser)->constval.f));
1053             if (!val)
1054                 return false;
1055             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1056                 parseerror(parser, "out of memory");
1057                 goto onerr;
1058             }
1059             DEBUGSHUNTDO(printf("push %g\n", parser_token(parser)->constval.f));
1060         }
1061         else if (parser->tok == TOKEN_INTCONST) {
1062             ast_value *val;
1063             if (wantop) {
1064                 parseerror(parser, "expected operator or end of statement, got constant");
1065                 goto onerr;
1066             }
1067             wantop = true;
1068             val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
1069             if (!val)
1070                 return false;
1071             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1072                 parseerror(parser, "out of memory");
1073                 goto onerr;
1074             }
1075             DEBUGSHUNTDO(printf("push %i\n", parser_token(parser)->constval.i));
1076         }
1077         else if (parser->tok == TOKEN_STRINGCONST) {
1078             ast_value *val;
1079             if (wantop) {
1080                 parseerror(parser, "expected operator or end of statement, got constant");
1081                 goto onerr;
1082             }
1083             wantop = true;
1084             val = parser_const_string(parser, parser_tokval(parser));
1085             if (!val)
1086                 return false;
1087             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1088                 parseerror(parser, "out of memory");
1089                 goto onerr;
1090             }
1091             DEBUGSHUNTDO(printf("push string\n"));
1092         }
1093         else if (parser->tok == TOKEN_VECTORCONST) {
1094             ast_value *val;
1095             if (wantop) {
1096                 parseerror(parser, "expected operator or end of statement, got constant");
1097                 goto onerr;
1098             }
1099             wantop = true;
1100             val = parser_const_vector(parser, parser_token(parser)->constval.v);
1101             if (!val)
1102                 return false;
1103             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1104                 parseerror(parser, "out of memory");
1105                 goto onerr;
1106             }
1107             DEBUGSHUNTDO(printf("push '%g %g %g'\n",
1108                                 parser_token(parser)->constval.v.x,
1109                                 parser_token(parser)->constval.v.y,
1110                                 parser_token(parser)->constval.v.z));
1111         }
1112         else if (parser->tok == '(') {
1113             parseerror(parser, "internal error: '(' should be classified as operator");
1114             goto onerr;
1115         }
1116         else if (parser->tok == ')') {
1117             if (wantop) {
1118                 DEBUGSHUNTDO(printf("do[op] )\n"));
1119                 --parens;
1120                 if (parens < 0)
1121                     break;
1122                 /* we do expect an operator next */
1123                 /* closing an opening paren */
1124                 if (!parser_close_paren(parser, &sy, false))
1125                     goto onerr;
1126             } else {
1127                 DEBUGSHUNTDO(printf("do[nop] )\n"));
1128                 --parens;
1129                 if (parens < 0)
1130                     break;
1131                 /* allowed for function calls */
1132                 if (!parser_close_paren(parser, &sy, true))
1133                     goto onerr;
1134             }
1135             wantop = true;
1136         }
1137         else if (parser->tok != TOKEN_OPERATOR) {
1138             if (wantop) {
1139                 parseerror(parser, "expected operator or end of statement");
1140                 goto onerr;
1141             }
1142             break;
1143         }
1144         else
1145         {
1146             /* classify the operator */
1147             /* TODO: suffix operators */
1148             const oper_info *op;
1149             const oper_info *olast = NULL;
1150             size_t o;
1151             for (o = 0; o < operator_count; ++o) {
1152                 if ((!(operators[o].flags & OP_PREFIX) == wantop) &&
1153                     !(operators[o].flags & OP_SUFFIX) && /* remove this */
1154                     !strcmp(parser_tokval(parser), operators[o].op))
1155                 {
1156                     break;
1157                 }
1158             }
1159             if (o == operator_count) {
1160                 /* no operator found... must be the end of the statement */
1161                 break;
1162             }
1163             /* found an operator */
1164             op = &operators[o];
1165
1166             /* when declaring variables, a comma starts a new variable */
1167             if (op->id == opid1(',') && !parens && stopatcomma)
1168                 break;
1169
1170             if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
1171                 olast = &operators[sy.ops[sy.ops_count-1].etype-1];
1172
1173             while (olast && (
1174                     (op->prec < olast->prec) ||
1175                     (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
1176             {
1177                 if (!parser_sy_pop(parser, &sy))
1178                     goto onerr;
1179                 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
1180                     olast = &operators[sy.ops[sy.ops_count-1].etype-1];
1181                 else
1182                     olast = NULL;
1183             }
1184
1185             if (op->id == opid1('.') && opts_standard == COMPILER_GMQCC) {
1186                 /* for gmqcc standard: open up the namespace of the previous type */
1187                 ast_expression *prevex = sy.out[sy.out_count-1].out;
1188                 if (!prevex) {
1189                     parseerror(parser, "unexpected member operator");
1190                     goto onerr;
1191                 }
1192                 if (prevex->expression.vtype == TYPE_ENTITY)
1193                     parser->memberof = TYPE_ENTITY;
1194                 else if (prevex->expression.vtype == TYPE_VECTOR)
1195                     parser->memberof = TYPE_VECTOR;
1196                 else {
1197                     parseerror(parser, "type error: type has no members");
1198                     goto onerr;
1199                 }
1200                 gotmemberof = true;
1201             }
1202
1203             if (op->id == opid1('(')) {
1204                 if (wantop) {
1205                     DEBUGSHUNTDO(printf("push (\n"));
1206                     ++parens;
1207                     /* we expected an operator, this is the function-call operator */
1208                     if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 'f', sy.out_count-1))) {
1209                         parseerror(parser, "out of memory");
1210                         goto onerr;
1211                     }
1212                 } else {
1213                     ++parens;
1214                     if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1, 0))) {
1215                         parseerror(parser, "out of memory");
1216                         goto onerr;
1217                     }
1218                     DEBUGSHUNTDO(printf("push (\n"));
1219                 }
1220                 wantop = false;
1221             } else {
1222                 DEBUGSHUNTDO(printf("push operator %s\n", op->op));
1223                 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
1224                     goto onerr;
1225                 wantop = false;
1226             }
1227         }
1228         if (!parser_next(parser)) {
1229             goto onerr;
1230         }
1231         if (parser->tok == ';') {
1232             break;
1233         }
1234     }
1235
1236     while (sy.ops_count) {
1237         if (!parser_sy_pop(parser, &sy))
1238             goto onerr;
1239     }
1240
1241     parser->lex->flags.noops = true;
1242     if (!sy.out_count) {
1243         parseerror(parser, "empty expression");
1244         expr = NULL;
1245     } else
1246         expr = sy.out[0].out;
1247     MEM_VECTOR_CLEAR(&sy, out);
1248     MEM_VECTOR_CLEAR(&sy, ops);
1249     DEBUGSHUNTDO(printf("shunt done\n"));
1250     return expr;
1251
1252 onerr:
1253     parser->lex->flags.noops = true;
1254     MEM_VECTOR_CLEAR(&sy, out);
1255     MEM_VECTOR_CLEAR(&sy, ops);
1256     return NULL;
1257 }
1258
1259 static ast_expression* parser_expression(parser_t *parser, bool stopatcomma)
1260 {
1261     ast_expression *e = parser_expression_leave(parser, stopatcomma);
1262     if (!e)
1263         return NULL;
1264     if (!parser_next(parser)) {
1265         ast_delete(e);
1266         return NULL;
1267     }
1268     return e;
1269 }
1270
1271 static bool parser_parse_if(parser_t *parser, ast_block *block, ast_expression **out)
1272 {
1273     ast_ifthen *ifthen;
1274     ast_expression *cond, *ontrue, *onfalse = NULL;
1275
1276     lex_ctx ctx = parser_ctx(parser);
1277
1278     /* skip the 'if' and check for opening paren */
1279     if (!parser_next(parser) || parser->tok != '(') {
1280         parseerror(parser, "expected 'if' condition in parenthesis");
1281         return false;
1282     }
1283     /* parse into the expression */
1284     if (!parser_next(parser)) {
1285         parseerror(parser, "expected 'if' condition after opening paren");
1286         return false;
1287     }
1288     /* parse the condition */
1289     cond = parser_expression_leave(parser, false);
1290     if (!cond)
1291         return false;
1292     /* closing paren */
1293     if (parser->tok != ')') {
1294         parseerror(parser, "expected closing paren after 'if' condition");
1295         ast_delete(cond);
1296         return false;
1297     }
1298     /* parse into the 'then' branch */
1299     if (!parser_next(parser)) {
1300         parseerror(parser, "expected statement for on-true branch of 'if'");
1301         ast_delete(cond);
1302         return false;
1303     }
1304     ontrue = parser_parse_statement_or_block(parser);
1305     if (!ontrue) {
1306         ast_delete(cond);
1307         return false;
1308     }
1309     /* check for an else */
1310     if (!strcmp(parser_tokval(parser), "else")) {
1311         /* parse into the 'else' branch */
1312         if (!parser_next(parser)) {
1313             parseerror(parser, "expected on-false branch after 'else'");
1314             ast_delete(ontrue);
1315             ast_delete(cond);
1316             return false;
1317         }
1318         onfalse = parser_parse_statement_or_block(parser);
1319         if (!onfalse) {
1320             ast_delete(ontrue);
1321             ast_delete(cond);
1322             return false;
1323         }
1324     }
1325
1326     ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
1327     *out = (ast_expression*)ifthen;
1328     return true;
1329 }
1330
1331 static bool parser_parse_while(parser_t *parser, ast_block *block, ast_expression **out)
1332 {
1333     ast_loop *aloop;
1334     ast_expression *cond, *ontrue;
1335
1336     lex_ctx ctx = parser_ctx(parser);
1337
1338     /* skip the 'while' and check for opening paren */
1339     if (!parser_next(parser) || parser->tok != '(') {
1340         parseerror(parser, "expected 'while' condition in parenthesis");
1341         return false;
1342     }
1343     /* parse into the expression */
1344     if (!parser_next(parser)) {
1345         parseerror(parser, "expected 'while' condition after opening paren");
1346         return false;
1347     }
1348     /* parse the condition */
1349     cond = parser_expression_leave(parser, false);
1350     if (!cond)
1351         return false;
1352     /* closing paren */
1353     if (parser->tok != ')') {
1354         parseerror(parser, "expected closing paren after 'while' condition");
1355         ast_delete(cond);
1356         return false;
1357     }
1358     /* parse into the 'then' branch */
1359     if (!parser_next(parser)) {
1360         parseerror(parser, "expected while-loop body");
1361         ast_delete(cond);
1362         return false;
1363     }
1364     ontrue = parser_parse_statement_or_block(parser);
1365     if (!ontrue) {
1366         ast_delete(cond);
1367         return false;
1368     }
1369
1370     aloop = ast_loop_new(ctx, NULL, cond, NULL, NULL, ontrue);
1371     *out = (ast_expression*)aloop;
1372     return true;
1373 }
1374
1375 static bool parser_parse_dowhile(parser_t *parser, ast_block *block, ast_expression **out)
1376 {
1377     ast_loop *aloop;
1378     ast_expression *cond, *ontrue;
1379
1380     lex_ctx ctx = parser_ctx(parser);
1381
1382     /* skip the 'do' and get the body */
1383     if (!parser_next(parser)) {
1384         parseerror(parser, "expected loop body");
1385         return false;
1386     }
1387     ontrue = parser_parse_statement_or_block(parser);
1388     if (!ontrue)
1389         return false;
1390
1391     /* expect the "while" */
1392     if (parser->tok != TOKEN_KEYWORD ||
1393         strcmp(parser_tokval(parser), "while"))
1394     {
1395         parseerror(parser, "expected 'while' and condition");
1396         ast_delete(ontrue);
1397         return false;
1398     }
1399
1400     /* skip the 'while' and check for opening paren */
1401     if (!parser_next(parser) || parser->tok != '(') {
1402         parseerror(parser, "expected 'while' condition in parenthesis");
1403         ast_delete(ontrue);
1404         return false;
1405     }
1406     /* parse into the expression */
1407     if (!parser_next(parser)) {
1408         parseerror(parser, "expected 'while' condition after opening paren");
1409         ast_delete(ontrue);
1410         return false;
1411     }
1412     /* parse the condition */
1413     cond = parser_expression_leave(parser, false);
1414     if (!cond)
1415         return false;
1416     /* closing paren */
1417     if (parser->tok != ')') {
1418         parseerror(parser, "expected closing paren after 'while' condition");
1419         ast_delete(ontrue);
1420         ast_delete(cond);
1421         return false;
1422     }
1423     /* parse on */
1424     if (!parser_next(parser) || parser->tok != ';') {
1425         parseerror(parser, "expected semicolon after condition");
1426         ast_delete(ontrue);
1427         ast_delete(cond);
1428         return false;
1429     }
1430
1431     if (!parser_next(parser)) {
1432         parseerror(parser, "parse error");
1433         ast_delete(ontrue);
1434         ast_delete(cond);
1435         return false;
1436     }
1437
1438     aloop = ast_loop_new(ctx, NULL, NULL, cond, NULL, ontrue);
1439     *out = (ast_expression*)aloop;
1440     return true;
1441 }
1442
1443 static bool parser_parse_for(parser_t *parser, ast_block *block, ast_expression **out)
1444 {
1445     ast_loop *aloop;
1446     ast_expression *initexpr, *cond, *increment, *ontrue;
1447     size_t oldblocklocal;
1448
1449     lex_ctx ctx = parser_ctx(parser);
1450
1451     oldblocklocal = parser->blocklocal;
1452     parser->blocklocal = parser->locals_count;
1453
1454     initexpr  = NULL;
1455     cond      = NULL;
1456     increment = NULL;
1457     ontrue    = NULL;
1458
1459     /* skip the 'while' and check for opening paren */
1460     if (!parser_next(parser) || parser->tok != '(') {
1461         parseerror(parser, "expected 'for' expressions in parenthesis");
1462         goto onerr;
1463     }
1464     /* parse into the expression */
1465     if (!parser_next(parser)) {
1466         parseerror(parser, "expected 'for' initializer after opening paren");
1467         goto onerr;
1468     }
1469
1470     if (parser->tok == TOKEN_TYPENAME) {
1471         if (opts_standard != COMPILER_GMQCC) {
1472             if (parsewarning(parser, WARN_EXTENSIONS,
1473                              "current standard does not allow variable declarations in for-loop initializers"))
1474                 goto onerr;
1475         }
1476
1477         parseerror(parser, "TODO: assignment of new variables to be non-const");
1478         goto onerr;
1479         if (!parser_variable(parser, block))
1480             goto onerr;
1481     }
1482     else if (parser->tok != ';')
1483     {
1484         initexpr = parser_expression_leave(parser, false);
1485         if (!initexpr)
1486             goto onerr;
1487     }
1488
1489     /* move on to condition */
1490     if (parser->tok != ';') {
1491         parseerror(parser, "expected semicolon after for-loop initializer");
1492         goto onerr;
1493     }
1494     if (!parser_next(parser)) {
1495         parseerror(parser, "expected for-loop condition");
1496         goto onerr;
1497     }
1498
1499     /* parse the condition */
1500     if (parser->tok != ';') {
1501         cond = parser_expression_leave(parser, false);
1502         if (!cond)
1503             goto onerr;
1504     }
1505
1506     /* move on to incrementor */
1507     if (parser->tok != ';') {
1508         parseerror(parser, "expected semicolon after for-loop initializer");
1509         goto onerr;
1510     }
1511     if (!parser_next(parser)) {
1512         parseerror(parser, "expected for-loop condition");
1513         goto onerr;
1514     }
1515
1516     /* parse the incrementor */
1517     if (parser->tok != ')') {
1518         increment = parser_expression_leave(parser, false);
1519         if (!increment)
1520             goto onerr;
1521     }
1522
1523     /* closing paren */
1524     if (parser->tok != ')') {
1525         parseerror(parser, "expected closing paren after 'for-loop' incrementor");
1526         goto onerr;
1527     }
1528     /* parse into the 'then' branch */
1529     if (!parser_next(parser)) {
1530         parseerror(parser, "expected for-loop body");
1531         goto onerr;
1532     }
1533     ontrue = parser_parse_statement_or_block(parser);
1534     if (!ontrue) {
1535         goto onerr;
1536     }
1537
1538     aloop = ast_loop_new(ctx, initexpr, cond, NULL, increment, ontrue);
1539     *out = (ast_expression*)aloop;
1540
1541     while (parser->locals_count > parser->blocklocal)
1542         parser_pop_local(parser);
1543     parser->blocklocal = oldblocklocal;
1544     return true;
1545 onerr:
1546     if (initexpr)  ast_delete(initexpr);
1547     if (cond)      ast_delete(cond);
1548     if (increment) ast_delete(increment);
1549     while (parser->locals_count > parser->blocklocal)
1550         parser_pop_local(parser);
1551     parser->blocklocal = oldblocklocal;
1552     return false;
1553 }
1554
1555 static bool parser_parse_statement(parser_t *parser, ast_block *block, ast_expression **out)
1556 {
1557     if (parser->tok == TOKEN_TYPENAME)
1558     {
1559         /* local variable */
1560         if (!block) {
1561             parseerror(parser, "cannot declare a variable from here");
1562             return false;
1563         }
1564         if (opts_standard == COMPILER_QCC) {
1565             if (parsewarning(parser, WARN_EXTENSIONS, "missing 'local' keyword when declaring a local variable"))
1566                 return false;
1567         }
1568         if (!parser_variable(parser, block))
1569             return false;
1570         *out = NULL;
1571         return true;
1572     }
1573     else if (parser->tok == TOKEN_KEYWORD)
1574     {
1575         if (!strcmp(parser_tokval(parser), "local"))
1576         {
1577             if (!block) {
1578                 parseerror(parser, "cannot declare a local variable here");
1579                 return false;
1580             }
1581             if (!parser_next(parser)) {
1582                 parseerror(parser, "expected variable declaration");
1583                 return false;
1584             }
1585             if (!parser_variable(parser, block))
1586                 return false;
1587             *out = NULL;
1588             return true;
1589         }
1590         else if (!strcmp(parser_tokval(parser), "return"))
1591         {
1592             ast_expression *exp = NULL;
1593             ast_return     *ret = NULL;
1594             ast_value      *expected = parser->function->vtype;
1595
1596             if (!parser_next(parser)) {
1597                 parseerror(parser, "expected return expression");
1598                 return false;
1599             }
1600
1601             if (parser->tok != ';') {
1602                 exp = parser_expression(parser, false);
1603                 if (!exp)
1604                     return false;
1605
1606                 if (exp->expression.vtype != expected->expression.next->expression.vtype) {
1607                     parseerror(parser, "return with invalid expression");
1608                 }
1609
1610                 ret = ast_return_new(exp->expression.node.context, exp);
1611                 if (!ret) {
1612                     ast_delete(exp);
1613                     return false;
1614                 }
1615             } else {
1616                 if (!parser_next(parser))
1617                     parseerror(parser, "parse error");
1618                 if (expected->expression.next->expression.vtype != TYPE_VOID) {
1619                     parseerror(parser, "return without value");
1620                 }
1621                 ret = ast_return_new(parser_ctx(parser), NULL);
1622             }
1623             *out = (ast_expression*)ret;
1624             return true;
1625         }
1626         else if (!strcmp(parser_tokval(parser), "if"))
1627         {
1628             return parser_parse_if(parser, block, out);
1629         }
1630         else if (!strcmp(parser_tokval(parser), "while"))
1631         {
1632             return parser_parse_while(parser, block, out);
1633         }
1634         else if (!strcmp(parser_tokval(parser), "do"))
1635         {
1636             return parser_parse_dowhile(parser, block, out);
1637         }
1638         else if (!strcmp(parser_tokval(parser), "for"))
1639         {
1640             if (opts_standard == COMPILER_QCC) {
1641                 if (parsewarning(parser, WARN_EXTENSIONS, "for loops are not recognized in the original Quake C standard, to enable try an alternate standard --std=?"))
1642                     return false;
1643             }
1644             return parser_parse_for(parser, block, out);
1645         }
1646         parseerror(parser, "Unexpected keyword");
1647         return false;
1648     }
1649     else if (parser->tok == '{')
1650     {
1651         ast_block *inner;
1652         inner = parser_parse_block(parser);
1653         if (!inner)
1654             return false;
1655         *out = (ast_expression*)inner;
1656         return true;
1657     }
1658     else
1659     {
1660         ast_expression *exp = parser_expression(parser, false);
1661         if (!exp)
1662             return false;
1663         *out = exp;
1664         return true;
1665     }
1666 }
1667
1668 static void parser_pop_local(parser_t *parser)
1669 {
1670     parser->locals_count--;
1671     mem_d(parser->locals[parser->locals_count].name);
1672 }
1673
1674 static ast_block* parser_parse_block(parser_t *parser)
1675 {
1676     size_t oldblocklocal;
1677     ast_block *block = NULL;
1678
1679     oldblocklocal = parser->blocklocal;
1680     parser->blocklocal = parser->locals_count;
1681
1682     if (!parser_next(parser)) { /* skip the '{' */
1683         parseerror(parser, "expected function body");
1684         goto cleanup;
1685     }
1686
1687     block = ast_block_new(parser_ctx(parser));
1688
1689     while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1690     {
1691         ast_expression *expr;
1692         if (parser->tok == '}')
1693             break;
1694
1695         if (!parser_parse_statement(parser, block, &expr)) {
1696             parseerror(parser, "parse error");
1697             ast_block_delete(block);
1698             block = NULL;
1699             goto cleanup;
1700         }
1701         if (!expr)
1702             continue;
1703         if (!ast_block_exprs_add(block, expr)) {
1704             ast_delete(expr);
1705             ast_block_delete(block);
1706             block = NULL;
1707             goto cleanup;
1708         }
1709     }
1710
1711     if (parser->tok != '}') {
1712         ast_block_delete(block);
1713         block = NULL;
1714     } else {
1715         (void)parser_next(parser);
1716     }
1717
1718 cleanup:
1719     while (parser->locals_count > parser->blocklocal)
1720         parser_pop_local(parser);
1721     parser->blocklocal = oldblocklocal;
1722     /* unroll the local vector */
1723     return block;
1724 }
1725
1726 static ast_expression* parser_parse_statement_or_block(parser_t *parser)
1727 {
1728     ast_expression *expr = NULL;
1729     if (parser->tok == '{')
1730         return (ast_expression*)parser_parse_block(parser);
1731     if (!parser_parse_statement(parser, NULL, &expr))
1732         return NULL;
1733     return expr;
1734 }
1735
1736 static bool parser_variable(parser_t *parser, ast_block *localblock)
1737 {
1738     bool          isfunc = false;
1739     ast_function *func = NULL;
1740     lex_ctx       ctx;
1741     ast_value    *var;
1742     varentry_t    varent;
1743     ast_expression *olddecl;
1744
1745     bool hadproto;
1746
1747     int basetype = parser_token(parser)->constval.t;
1748
1749     while (true)
1750     {
1751         hadproto = false;
1752
1753         if (!parser_next(parser)) { /* skip basetype or comma */
1754             parseerror(parser, "expected variable declaration");
1755             return false;
1756         }
1757
1758         olddecl = NULL;
1759         isfunc  = false;
1760         func    = NULL;
1761         ctx = parser_ctx(parser);
1762         var = parser_parse_type(parser, basetype, &isfunc);
1763
1764         if (!var)
1765             return false;
1766
1767         if (parser->tok != TOKEN_IDENT) {
1768             parseerror(parser, "expected variable name\n");
1769             return false;
1770         }
1771
1772         if (!isfunc) {
1773             if (!localblock && (olddecl = parser_find_global(parser, parser_tokval(parser)))) {
1774                 ast_value_delete(var);
1775                 parseerror(parser, "global %s already declared here: %s:%i\n",
1776                            parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1777                 return false;
1778             }
1779
1780             if (localblock && parser_find_local(parser, parser_tokval(parser), parser->blocklocal)) {
1781                 ast_value_delete(var);
1782                 parseerror(parser, "local %s already declared here: %s:%i\n",
1783                            parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1784                 return false;
1785             }
1786         }
1787
1788         if (!ast_value_set_name(var, parser_tokval(parser))) {
1789             parseerror(parser, "failed to set variable name\n");
1790             ast_value_delete(var);
1791             return false;
1792         }
1793
1794         if (isfunc) {
1795             /* a function was defined */
1796             ast_value *fval;
1797             ast_value *proto = NULL;
1798
1799             if (!localblock)
1800                 olddecl = parser_find_global(parser, parser_tokval(parser));
1801             else
1802                 olddecl = parser_find_local(parser, parser_tokval(parser), parser->blocklocal);
1803
1804             if (olddecl) {
1805                 /* we had a prototype */
1806                 if (!ast_istype(olddecl, ast_value)) {
1807                     /* theoretically not possible you think?
1808                      * well:
1809                      * vector v;
1810                      * void() v_x = {}
1811                      * got it?
1812                      */
1813                     parseerror(parser, "cannot declare a function with the same name as a vector's member: %s",
1814                                parser_tokval(parser));
1815                     ast_value_delete(var);
1816                     return false;
1817                 }
1818
1819                 proto = (ast_value*)olddecl;
1820             }
1821
1822             /* turn var into a value of TYPE_FUNCTION, with the old var
1823              * as return type
1824              */
1825             fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1826             func = ast_function_new(ctx, var->name, fval);
1827             if (!fval || !func) {
1828                 ast_value_delete(var);
1829                 if (fval) ast_value_delete(fval);
1830                 if (func) ast_function_delete(func);
1831                 return false;
1832             }
1833
1834             fval->expression.next = (ast_expression*)var;
1835             MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1836
1837             /* we compare the type late here, but it's easier than
1838              * messing with the parameter-vector etc. earlier
1839              */
1840             if (proto) {
1841                 if (!ast_compare_type((ast_expression*)proto, (ast_expression*)fval)) {
1842                     parseerror(parser, "conflicting types for `%s`, previous declaration was here: %s:%i",
1843                                proto->name,
1844                                ast_ctx(proto).file, ast_ctx(proto).line);
1845                     ast_function_delete(func);
1846                     ast_value_delete(fval);
1847                     return false;
1848                 }
1849                 ast_function_delete(func);
1850                 ast_value_delete(fval);
1851                 fval = proto;
1852                 func = proto->constval.vfunc;
1853                 hadproto = true;
1854             }
1855             else
1856             {
1857                 if (!parser_t_functions_add(parser, func)) {
1858                     ast_function_delete(func);
1859                     ast_value_delete(fval);
1860                     return false;
1861                 }
1862             }
1863
1864             var = fval;
1865         }
1866
1867         if (!hadproto) {
1868             varent.name = util_strdup(var->name);
1869             varent.var = (ast_expression*)var;
1870             if (var->expression.vtype == TYPE_VECTOR)
1871             {
1872                 size_t len = strlen(varent.name);
1873                 varentry_t vx, vy, vz;
1874                 vx.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 0);
1875                 vy.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 1);
1876                 vz.var = (ast_expression*)ast_member_new(var->expression.node.context, (ast_expression*)var, 2);
1877                 vx.name = (char*)mem_a(len+3);
1878                 vy.name = (char*)mem_a(len+3);
1879                 vz.name = (char*)mem_a(len+3);
1880                 memcpy(vx.name, varent.name, len);
1881                 memcpy(vy.name, varent.name, len);
1882                 memcpy(vz.name, varent.name, len);
1883                 vx.name[len] = vy.name[len] = vz.name[len] = '_';
1884                 vx.name[len+1] = 'x';
1885                 vy.name[len+1] = 'y';
1886                 vz.name[len+1] = 'z';
1887                 vx.name[len+2] = vy.name[len+2] = vz.name[len+2] = 0;
1888
1889                 if (!localblock) {
1890                     (void)!parser_t_globals_add(parser, varent);
1891                     (void)!parser_t_globals_add(parser, vx);
1892                     (void)!parser_t_globals_add(parser, vy);
1893                     (void)!parser_t_globals_add(parser, vz);
1894                 } else {
1895                     (void)!parser_t_locals_add(parser, varent);
1896                     (void)!parser_t_locals_add(parser, vx);
1897                     (void)!parser_t_locals_add(parser, vy);
1898                     (void)!parser_t_locals_add(parser, vz);
1899                     if (!ast_block_locals_add(localblock, var) ||
1900                         !ast_block_collect(localblock, vx.var) ||
1901                         !ast_block_collect(localblock, vy.var) ||
1902                         !ast_block_collect(localblock, vz.var))
1903                     {
1904                         parser_pop_local(parser);
1905                         parser_pop_local(parser);
1906                         parser_pop_local(parser);
1907                         parser_pop_local(parser);
1908                         ast_value_delete(var);
1909                         return false;
1910                     }
1911                 }
1912             }
1913             else
1914             {
1915                 if ( (!localblock && !parser_t_globals_add(parser, varent)) ||
1916                      ( localblock && !parser_t_locals_add(parser, varent)) )
1917                 {
1918                     ast_value_delete(var);
1919                     return false;
1920                 }
1921                 if (localblock && !ast_block_locals_add(localblock, var))
1922                 {
1923                     parser_pop_local(parser);
1924                     ast_value_delete(var);
1925                     return false;
1926                 }
1927             }
1928         }
1929
1930         if (!parser_next(parser)) {
1931             ast_value_delete(var);
1932             return false;
1933         }
1934
1935         if (parser->tok == ';') {
1936             if (!parser_next(parser))
1937                 return parser->tok == TOKEN_EOF;
1938             return true;
1939         }
1940
1941         if (parser->tok == ',') {
1942             /* another var */
1943             continue;
1944         }
1945
1946         if (parser->tok != '=') {
1947             parseerror(parser, "expected '=' or ';'");
1948             return false;
1949         }
1950
1951         if (!parser_next(parser))
1952             return false;
1953
1954         if (parser->tok == '#') {
1955             if (localblock) {
1956                 parseerror(parser, "cannot declare builtins within functions");
1957                 return false;
1958             }
1959             if (!isfunc || !func) {
1960                 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
1961                 return false;
1962             }
1963             if (!parser_next(parser)) {
1964                 parseerror(parser, "expected builtin number");
1965                 return false;
1966             }
1967             if (parser->tok != TOKEN_INTCONST) {
1968                 parseerror(parser, "builtin number must be an integer constant");
1969                 return false;
1970             }
1971             if (parser_token(parser)->constval.i <= 0) {
1972                 parseerror(parser, "builtin number must be positive integer greater than zero");
1973                 return false;
1974             }
1975
1976             func->builtin = -parser_token(parser)->constval.i;
1977
1978             if (!parser_next(parser))
1979                 return false;
1980         } else if (parser->tok == '{') {
1981             /* function body */
1982             ast_block *block;
1983             ast_function *old = parser->function;
1984
1985             if (localblock) {
1986                 parseerror(parser, "cannot declare functions within functions");
1987                 return false;
1988             }
1989
1990             parser->function = func;
1991             block = parser_parse_block(parser);
1992             parser->function = old;
1993
1994             if (!block)
1995                 return false;
1996
1997             if (!ast_function_blocks_add(func, block)) {
1998                 ast_block_delete(block);
1999                 return false;
2000             }
2001
2002             if (parser->tok == ';')
2003                 return parser_next(parser) || parser->tok == TOKEN_EOF;
2004             else if (opts_standard == COMPILER_QCC)
2005                 parseerror(parser, "missing semicolon after function body (mandatory with -std=qcc)");
2006             return true;
2007         } else {
2008             ast_expression *cexp;
2009             ast_value      *cval;
2010
2011             cexp = parser_expression_leave(parser, true);
2012             cval = (ast_value*)cexp;
2013             if (!ast_istype(cval, ast_value) || !cval->isconst)
2014                 parseerror(parser, "cannot initialize a global constant variable with a non-constant expression");
2015             else
2016             {
2017                 var->isconst = true;
2018                 memcpy(&var->constval, &cval->constval, sizeof(var->constval));
2019                 memset(&cval->constval, 0, sizeof(cval->constval));
2020                 ast_unref(cval);
2021             }
2022         }
2023
2024         if (parser->tok == ',') {
2025             /* another */
2026             continue;
2027         }
2028
2029         if (parser->tok != ';') {
2030             parseerror(parser, "missing semicolon");
2031             return false;
2032         }
2033
2034         (void)parser_next(parser);
2035
2036         return true;
2037     }
2038 }
2039
2040 static bool parser_do(parser_t *parser)
2041 {
2042     if (parser->tok == TOKEN_TYPENAME)
2043     {
2044         return parser_variable(parser, NULL);
2045     }
2046     else if (parser->tok == TOKEN_KEYWORD)
2047     {
2048         /* handle 'var' and 'const' */
2049         return false;
2050     }
2051     else if (parser->tok == '.')
2052     {
2053         ast_value *var;
2054         ast_value *typevar;
2055         ast_value *fld;
2056         ast_expression *oldex;
2057         bool       isfunc = false;
2058         int        basetype;
2059         lex_ctx    ctx = parser_ctx(parser);
2060         varentry_t varent;
2061
2062         /* entity-member declaration */
2063         if (!parser_next(parser) || parser->tok != TOKEN_TYPENAME) {
2064             parseerror(parser, "expected member variable definition");
2065             return false;
2066         }
2067
2068         /* remember the base/return type */
2069         basetype = parser_token(parser)->constval.t;
2070
2071         /* parse into the declaration */
2072         if (!parser_next(parser)) {
2073             parseerror(parser, "expected field def");
2074             return false;
2075         }
2076
2077         /* parse the field type fully */
2078         typevar = var = parser_parse_type(parser, basetype, &isfunc);
2079         if (!var)
2080             return false;
2081
2082         while (true) {
2083             var = ast_value_copy(typevar);
2084             /* now the field name */
2085             if (parser->tok != TOKEN_IDENT) {
2086                 parseerror(parser, "expected field name");
2087                 ast_delete(var);
2088                 return false;
2089             }
2090
2091             /* check for an existing field
2092              * in original qc we also have to check for an existing
2093              * global named like the field
2094              */
2095             if (opts_standard == COMPILER_QCC) {
2096                 if (parser_find_global(parser, parser_tokval(parser))) {
2097                     parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
2098                     ast_delete(var);
2099                     return false;
2100                 }
2101             }
2102
2103             if (isfunc) {
2104                 ast_value *fval;
2105                 fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
2106                 if (!fval) {
2107                     ast_value_delete(var);
2108                     return false;
2109                 }
2110                 fval->expression.next = (ast_expression*)var;
2111                 MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
2112                 var = fval;
2113             }
2114
2115             /* turn it into a field */
2116             fld = ast_value_new(ctx, parser_tokval(parser), TYPE_FIELD);
2117             fld->expression.next = (ast_expression*)var;
2118
2119             if ( (oldex = parser_find_field(parser, parser_tokval(parser)))) {
2120                 if (ast_istype(oldex, ast_member)) {
2121                     parseerror(parser, "cannot declare a field with the same name as a vector component, component %s has been declared here: %s:%i",
2122                                parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line);
2123                     ast_delete(fld);
2124                     return false;
2125                 }
2126                 if (!ast_istype(oldex, ast_value)) {
2127                     /* not possible / sanity check */
2128                     parseerror(parser, "internal error: %s is not an ast_value", parser_tokval(parser));
2129                     ast_delete(fld);
2130                     return false;
2131                 }
2132
2133                 if (!ast_compare_type(oldex, (ast_expression*)fld)) {
2134                     parseerror(parser, "field %s has previously been declared with a different type here: %s:%i",
2135                                parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line);
2136                     ast_delete(fld);
2137                     return false;
2138                 } else {
2139                     if (parsewarning(parser, WARN_FIELD_REDECLARED, "field %s has already been declared here: %s:%i",
2140                                      parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line))
2141                     {
2142                         ast_delete(fld);
2143                         return false;
2144                     }
2145                 }
2146
2147                 ast_delete(fld);
2148                 goto nextfield;
2149             }
2150
2151             varent.var = (ast_expression*)fld;
2152             varent.name = util_strdup(fld->name);
2153             (void)!parser_t_fields_add(parser, varent);
2154
2155             if (var->expression.vtype == TYPE_VECTOR)
2156             {
2157                 /* create _x, _y and _z fields as well */
2158                 size_t len;
2159                 varentry_t vx, vy, vz;
2160
2161                 len = strlen(varent.name);
2162                 vx.var = (ast_expression*)ast_member_new(ast_ctx(fld), (ast_expression*)fld, 0);
2163                 vy.var = (ast_expression*)ast_member_new(ast_ctx(fld), (ast_expression*)fld, 1);
2164                 vz.var = (ast_expression*)ast_member_new(ast_ctx(fld), (ast_expression*)fld, 2);
2165                 vx.name = (char*)mem_a(len+3);
2166                 vy.name = (char*)mem_a(len+3);
2167                 vz.name = (char*)mem_a(len+3);
2168                 memcpy(vx.name, varent.name, len);
2169                 memcpy(vy.name, varent.name, len);
2170                 memcpy(vz.name, varent.name, len);
2171                 vx.name[len] = vy.name[len] = vz.name[len] = '_';
2172                 vx.name[len+1] = 'x';
2173                 vy.name[len+1] = 'y';
2174                 vz.name[len+1] = 'z';
2175                 vx.name[len+2] = vy.name[len+2] = vz.name[len+2] = 0;
2176                 (void)!parser_t_fields_add(parser, vx);
2177                 (void)!parser_t_fields_add(parser, vy);
2178                 (void)!parser_t_fields_add(parser, vz);
2179             }
2180
2181 nextfield:
2182             if (!parser_next(parser)) {
2183                 parseerror(parser, "expected semicolon or another field name");
2184                 return false;
2185             }
2186             if (parser->tok == ';')
2187                 break;
2188             if (parser->tok != ',' || !parser_next(parser)) {
2189                 parseerror(parser, "expected semicolon or another field name");
2190                 return false;
2191             }
2192         }
2193         ast_delete(typevar);
2194
2195         /* skip the semicolon */
2196         if (!parser_next(parser))
2197             return parser->tok == TOKEN_EOF;
2198
2199         return true;
2200     }
2201     else if (parser->tok == '$')
2202     {
2203         if (!parser_next(parser)) {
2204             parseerror(parser, "parse error");
2205             return false;
2206         }
2207     }
2208     else
2209     {
2210         parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
2211         return false;
2212     }
2213     return true;
2214 }
2215
2216 static parser_t *parser;
2217
2218 bool parser_init()
2219 {
2220     parser = (parser_t*)mem_a(sizeof(parser_t));
2221     if (!parser)
2222         return false;
2223
2224     memset(parser, 0, sizeof(*parser));
2225     return true;
2226 }
2227
2228 bool parser_compile(const char *filename)
2229 {
2230     parser->lex = lex_open(filename);
2231     if (!parser->lex) {
2232         printf("failed to open file \"%s\"\n", filename);
2233         return false;
2234     }
2235
2236     /* initial lexer/parser state */
2237     parser->lex->flags.noops = true;
2238
2239     if (parser_next(parser))
2240     {
2241         while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
2242         {
2243             if (!parser_do(parser)) {
2244                 if (parser->tok == TOKEN_EOF)
2245                     parseerror(parser, "unexpected eof");
2246                 else if (!parser->errors)
2247                     parseerror(parser, "parse error\n");
2248                 lex_close(parser->lex);
2249                 parser->lex = NULL;
2250                 return false;
2251             }
2252         }
2253     }
2254
2255     lex_close(parser->lex);
2256     parser->lex = NULL;
2257
2258     return !parser->errors;
2259 }
2260
2261 void parser_cleanup()
2262 {
2263     size_t i;
2264     for (i = 0; i < parser->functions_count; ++i) {
2265         ast_delete(parser->functions[i]);
2266     }
2267     for (i = 0; i < parser->imm_vector_count; ++i) {
2268         ast_delete(parser->imm_vector[i]);
2269     }
2270     for (i = 0; i < parser->imm_string_count; ++i) {
2271         ast_delete(parser->imm_string[i]);
2272     }
2273     for (i = 0; i < parser->imm_float_count; ++i) {
2274         ast_delete(parser->imm_float[i]);
2275     }
2276     for (i = 0; i < parser->fields_count; ++i) {
2277         ast_delete(parser->fields[i].var);
2278         mem_d(parser->fields[i].name);
2279     }
2280     for (i = 0; i < parser->globals_count; ++i) {
2281         ast_delete(parser->globals[i].var);
2282         mem_d(parser->globals[i].name);
2283     }
2284     MEM_VECTOR_CLEAR(parser, functions);
2285     MEM_VECTOR_CLEAR(parser, imm_vector);
2286     MEM_VECTOR_CLEAR(parser, imm_string);
2287     MEM_VECTOR_CLEAR(parser, imm_float);
2288     MEM_VECTOR_CLEAR(parser, globals);
2289     MEM_VECTOR_CLEAR(parser, fields);
2290     MEM_VECTOR_CLEAR(parser, locals);
2291
2292     mem_d(parser);
2293 }
2294
2295 bool parser_finish(const char *output)
2296 {
2297     size_t i;
2298     ir_builder *ir;
2299
2300     if (!parser->errors)
2301     {
2302         ir = ir_builder_new("gmqcc_out");
2303         if (!ir) {
2304             printf("failed to allocate builder\n");
2305             return false;
2306         }
2307
2308         for (i = 0; i < parser->imm_float_count; ++i) {
2309             if (!ast_global_codegen(parser->imm_float[i], ir)) {
2310                 printf("failed to generate global %s\n", parser->imm_float[i]->name);
2311                 ir_builder_delete(ir);
2312                 return false;
2313             }
2314         }
2315         for (i = 0; i < parser->imm_string_count; ++i) {
2316             if (!ast_global_codegen(parser->imm_string[i], ir)) {
2317                 printf("failed to generate global %s\n", parser->imm_string[i]->name);
2318                 ir_builder_delete(ir);
2319                 return false;
2320             }
2321         }
2322         for (i = 0; i < parser->imm_vector_count; ++i) {
2323             if (!ast_global_codegen(parser->imm_vector[i], ir)) {
2324                 printf("failed to generate global %s\n", parser->imm_vector[i]->name);
2325                 ir_builder_delete(ir);
2326                 return false;
2327             }
2328         }
2329         for (i = 0; i < parser->fields_count; ++i) {
2330             ast_value *field;
2331             bool isconst;
2332             if (!ast_istype(parser->fields[i].var, ast_value))
2333                 continue;
2334             field = (ast_value*)parser->fields[i].var;
2335             isconst = field->isconst;
2336             field->isconst = false;
2337             if (!ast_global_codegen((ast_value*)field, ir)) {
2338                 printf("failed to generate field %s\n", field->name);
2339                 ir_builder_delete(ir);
2340                 return false;
2341             }
2342             if (isconst) {
2343                 ir_value *ifld;
2344                 ast_expression *subtype;
2345                 field->isconst = true;
2346                 subtype = field->expression.next;
2347                 ifld = ir_builder_create_field(ir, field->name, subtype->expression.vtype);
2348                 if (subtype->expression.vtype == TYPE_FIELD)
2349                     ifld->fieldtype = subtype->expression.next->expression.vtype;
2350                 else if (subtype->expression.vtype == TYPE_FUNCTION)
2351                     ifld->outtype = subtype->expression.next->expression.vtype;
2352                 (void)!ir_value_set_field(field->ir_v, ifld);
2353             }
2354         }
2355         for (i = 0; i < parser->globals_count; ++i) {
2356             if (!ast_istype(parser->globals[i].var, ast_value))
2357                 continue;
2358             if (!ast_global_codegen((ast_value*)(parser->globals[i].var), ir)) {
2359                 printf("failed to generate global %s\n", parser->globals[i].name);
2360                 ir_builder_delete(ir);
2361                 return false;
2362             }
2363         }
2364         for (i = 0; i < parser->functions_count; ++i) {
2365             if (!ast_function_codegen(parser->functions[i], ir)) {
2366                 printf("failed to generate function %s\n", parser->functions[i]->name);
2367                 ir_builder_delete(ir);
2368                 return false;
2369             }
2370             if (!ir_function_finalize(parser->functions[i]->ir_func)) {
2371                 printf("failed to finalize function %s\n", parser->functions[i]->name);
2372                 ir_builder_delete(ir);
2373                 return false;
2374             }
2375         }
2376
2377         if (opts_dump)
2378             ir_builder_dump(ir, printf);
2379
2380         if (!ir_builder_generate(ir, output)) {
2381             printf("*** failed to generate output file\n");
2382             ir_builder_delete(ir);
2383             return false;
2384         }
2385
2386         ir_builder_delete(ir);
2387         return true;
2388     }
2389
2390     printf("*** there were compile errors\n");
2391     return false;
2392 }