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