]> git.xonotic.org Git - xonotic/gmqcc.git/blob - parser.c
Automatic prototyping of frame-functions
[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     if (sy->ops[sy->ops_count-1].paren == 1) {
981         parseerror(parser, "empty parenthesis expression");
982         return false;
983     }
984     while (sy->ops_count) {
985         if (sy->ops[sy->ops_count-1].paren == 'f') {
986             if (!parser_close_call(parser, sy))
987                 return false;
988             break;
989         }
990         if (sy->ops[sy->ops_count-1].paren == 1) {
991             sy->ops_count--;
992             return !functions_only;
993         }
994         if (!parser_sy_pop(parser, sy))
995             return false;
996     }
997     return true;
998 }
999
1000 static void parser_reclassify_token(parser_t *parser)
1001 {
1002     size_t i;
1003     for (i = 0; i < operator_count; ++i) {
1004         if (!strcmp(parser_tokval(parser), operators[i].op)) {
1005             parser->tok = TOKEN_OPERATOR;
1006             return;
1007         }
1008     }
1009 }
1010
1011 static ast_expression* parser_expression_leave(parser_t *parser, bool stopatcomma)
1012 {
1013     ast_expression *expr = NULL;
1014     shunt sy;
1015     bool wantop = false;
1016     bool gotmemberof = false;
1017
1018     /* count the parens because an if starts with one, so the
1019      * end of a condition is an unmatched closing paren
1020      */
1021     int parens = 0;
1022
1023     MEM_VECTOR_INIT(&sy, out);
1024     MEM_VECTOR_INIT(&sy, ops);
1025
1026     parser->lex->flags.noops = false;
1027
1028     parser_reclassify_token(parser);
1029
1030     while (true)
1031     {
1032         if (gotmemberof)
1033             gotmemberof = false;
1034         else
1035             parser->memberof = 0;
1036
1037         if (parser->tok == TOKEN_IDENT)
1038         {
1039             ast_expression *var;
1040             if (wantop) {
1041                 parseerror(parser, "expected operator or end of statement");
1042                 goto onerr;
1043             }
1044             wantop = true;
1045             /* variable */
1046             if (opts_standard == COMPILER_GMQCC)
1047             {
1048                 if (parser->memberof == TYPE_ENTITY) {
1049                     /* still get vars first since there could be a fieldpointer */
1050                     var = parser_find_var(parser, parser_tokval(parser));
1051                     if (!var)
1052                         var = parser_find_field(parser, parser_tokval(parser));
1053                 }
1054                 else if (parser->memberof == TYPE_VECTOR)
1055                 {
1056                     parseerror(parser, "TODO: implement effective vector member access");
1057                     goto onerr;
1058                 }
1059                 else if (parser->memberof) {
1060                     parseerror(parser, "namespace for member not found");
1061                     goto onerr;
1062                 }
1063                 else
1064                     var = parser_find_var(parser, parser_tokval(parser));
1065             } else {
1066                 var = parser_find_var(parser, parser_tokval(parser));
1067                 if (!var)
1068                     var = parser_find_field(parser, parser_tokval(parser));
1069             }
1070             if (!var) {
1071                 parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
1072                 goto onerr;
1073             }
1074             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), var))) {
1075                 parseerror(parser, "out of memory");
1076                 goto onerr;
1077             }
1078             DEBUGSHUNTDO(printf("push %s\n", parser_tokval(parser)));
1079         }
1080         else if (parser->tok == TOKEN_FLOATCONST) {
1081             ast_value *val;
1082             if (wantop) {
1083                 parseerror(parser, "expected operator or end of statement, got constant");
1084                 goto onerr;
1085             }
1086             wantop = true;
1087             val = parser_const_float(parser, (parser_token(parser)->constval.f));
1088             if (!val)
1089                 return false;
1090             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1091                 parseerror(parser, "out of memory");
1092                 goto onerr;
1093             }
1094             DEBUGSHUNTDO(printf("push %g\n", parser_token(parser)->constval.f));
1095         }
1096         else if (parser->tok == TOKEN_INTCONST) {
1097             ast_value *val;
1098             if (wantop) {
1099                 parseerror(parser, "expected operator or end of statement, got constant");
1100                 goto onerr;
1101             }
1102             wantop = true;
1103             val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
1104             if (!val)
1105                 return false;
1106             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1107                 parseerror(parser, "out of memory");
1108                 goto onerr;
1109             }
1110             DEBUGSHUNTDO(printf("push %i\n", parser_token(parser)->constval.i));
1111         }
1112         else if (parser->tok == TOKEN_STRINGCONST) {
1113             ast_value *val;
1114             if (wantop) {
1115                 parseerror(parser, "expected operator or end of statement, got constant");
1116                 goto onerr;
1117             }
1118             wantop = true;
1119             val = parser_const_string(parser, parser_tokval(parser));
1120             if (!val)
1121                 return false;
1122             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1123                 parseerror(parser, "out of memory");
1124                 goto onerr;
1125             }
1126             DEBUGSHUNTDO(printf("push string\n"));
1127         }
1128         else if (parser->tok == TOKEN_VECTORCONST) {
1129             ast_value *val;
1130             if (wantop) {
1131                 parseerror(parser, "expected operator or end of statement, got constant");
1132                 goto onerr;
1133             }
1134             wantop = true;
1135             val = parser_const_vector(parser, parser_token(parser)->constval.v);
1136             if (!val)
1137                 return false;
1138             if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
1139                 parseerror(parser, "out of memory");
1140                 goto onerr;
1141             }
1142             DEBUGSHUNTDO(printf("push '%g %g %g'\n",
1143                                 parser_token(parser)->constval.v.x,
1144                                 parser_token(parser)->constval.v.y,
1145                                 parser_token(parser)->constval.v.z));
1146         }
1147         else if (parser->tok == '(') {
1148             parseerror(parser, "internal error: '(' should be classified as operator");
1149             goto onerr;
1150         }
1151         else if (parser->tok == ')') {
1152             if (wantop) {
1153                 DEBUGSHUNTDO(printf("do[op] )\n"));
1154                 --parens;
1155                 if (parens < 0)
1156                     break;
1157                 /* we do expect an operator next */
1158                 /* closing an opening paren */
1159                 if (!parser_close_paren(parser, &sy, false))
1160                     goto onerr;
1161             } else {
1162                 DEBUGSHUNTDO(printf("do[nop] )\n"));
1163                 --parens;
1164                 if (parens < 0)
1165                     break;
1166                 /* allowed for function calls */
1167                 if (!parser_close_paren(parser, &sy, true))
1168                     goto onerr;
1169             }
1170             wantop = true;
1171         }
1172         else if (parser->tok != TOKEN_OPERATOR) {
1173             if (wantop) {
1174                 parseerror(parser, "expected operator or end of statement");
1175                 goto onerr;
1176             }
1177             break;
1178         }
1179         else
1180         {
1181             /* classify the operator */
1182             /* TODO: suffix operators */
1183             const oper_info *op;
1184             const oper_info *olast = NULL;
1185             size_t o;
1186             for (o = 0; o < operator_count; ++o) {
1187                 if ((!(operators[o].flags & OP_PREFIX) == wantop) &&
1188                     !(operators[o].flags & OP_SUFFIX) && /* remove this */
1189                     !strcmp(parser_tokval(parser), operators[o].op))
1190                 {
1191                     break;
1192                 }
1193             }
1194             if (o == operator_count) {
1195                 /* no operator found... must be the end of the statement */
1196                 break;
1197             }
1198             /* found an operator */
1199             op = &operators[o];
1200
1201             /* when declaring variables, a comma starts a new variable */
1202             if (op->id == opid1(',') && !parens && stopatcomma) {
1203                 /* fixup the token */
1204                 parser->tok = ',';
1205                 break;
1206             }
1207
1208             if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
1209                 olast = &operators[sy.ops[sy.ops_count-1].etype-1];
1210
1211             while (olast && (
1212                     (op->prec < olast->prec) ||
1213                     (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
1214             {
1215                 if (!parser_sy_pop(parser, &sy))
1216                     goto onerr;
1217                 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
1218                     olast = &operators[sy.ops[sy.ops_count-1].etype-1];
1219                 else
1220                     olast = NULL;
1221             }
1222
1223             if (op->id == opid1('.') && opts_standard == COMPILER_GMQCC) {
1224                 /* for gmqcc standard: open up the namespace of the previous type */
1225                 ast_expression *prevex = sy.out[sy.out_count-1].out;
1226                 if (!prevex) {
1227                     parseerror(parser, "unexpected member operator");
1228                     goto onerr;
1229                 }
1230                 if (prevex->expression.vtype == TYPE_ENTITY)
1231                     parser->memberof = TYPE_ENTITY;
1232                 else if (prevex->expression.vtype == TYPE_VECTOR)
1233                     parser->memberof = TYPE_VECTOR;
1234                 else {
1235                     parseerror(parser, "type error: type has no members");
1236                     goto onerr;
1237                 }
1238                 gotmemberof = true;
1239             }
1240
1241             if (op->id == opid1('(')) {
1242                 if (wantop) {
1243                     DEBUGSHUNTDO(printf("push (\n"));
1244                     ++parens;
1245                     /* we expected an operator, this is the function-call operator */
1246                     if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 'f', sy.out_count-1))) {
1247                         parseerror(parser, "out of memory");
1248                         goto onerr;
1249                     }
1250                 } else {
1251                     ++parens;
1252                     if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1, 0))) {
1253                         parseerror(parser, "out of memory");
1254                         goto onerr;
1255                     }
1256                     DEBUGSHUNTDO(printf("push (\n"));
1257                 }
1258                 wantop = false;
1259             } else {
1260                 DEBUGSHUNTDO(printf("push operator %s\n", op->op));
1261                 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
1262                     goto onerr;
1263                 wantop = false;
1264             }
1265         }
1266         if (!parser_next(parser)) {
1267             goto onerr;
1268         }
1269         if (parser->tok == ';' || parser->tok == ']') {
1270             break;
1271         }
1272     }
1273
1274     while (sy.ops_count) {
1275         if (!parser_sy_pop(parser, &sy))
1276             goto onerr;
1277     }
1278
1279     parser->lex->flags.noops = true;
1280     if (!sy.out_count) {
1281         parseerror(parser, "empty expression");
1282         expr = NULL;
1283     } else
1284         expr = sy.out[0].out;
1285     MEM_VECTOR_CLEAR(&sy, out);
1286     MEM_VECTOR_CLEAR(&sy, ops);
1287     DEBUGSHUNTDO(printf("shunt done\n"));
1288     return expr;
1289
1290 onerr:
1291     parser->lex->flags.noops = true;
1292     MEM_VECTOR_CLEAR(&sy, out);
1293     MEM_VECTOR_CLEAR(&sy, ops);
1294     return NULL;
1295 }
1296
1297 static ast_expression* parser_expression(parser_t *parser, bool stopatcomma)
1298 {
1299     ast_expression *e = parser_expression_leave(parser, stopatcomma);
1300     if (!e)
1301         return NULL;
1302     if (!parser_next(parser)) {
1303         ast_delete(e);
1304         return NULL;
1305     }
1306     return e;
1307 }
1308
1309 static bool parser_parse_if(parser_t *parser, ast_block *block, ast_expression **out)
1310 {
1311     ast_ifthen *ifthen;
1312     ast_expression *cond, *ontrue, *onfalse = NULL;
1313
1314     lex_ctx ctx = parser_ctx(parser);
1315
1316     /* skip the 'if' and check for opening paren */
1317     if (!parser_next(parser) || parser->tok != '(') {
1318         parseerror(parser, "expected 'if' condition in parenthesis");
1319         return false;
1320     }
1321     /* parse into the expression */
1322     if (!parser_next(parser)) {
1323         parseerror(parser, "expected 'if' condition after opening paren");
1324         return false;
1325     }
1326     /* parse the condition */
1327     cond = parser_expression_leave(parser, false);
1328     if (!cond)
1329         return false;
1330     /* closing paren */
1331     if (parser->tok != ')') {
1332         parseerror(parser, "expected closing paren after 'if' condition");
1333         ast_delete(cond);
1334         return false;
1335     }
1336     /* parse into the 'then' branch */
1337     if (!parser_next(parser)) {
1338         parseerror(parser, "expected statement for on-true branch of 'if'");
1339         ast_delete(cond);
1340         return false;
1341     }
1342     ontrue = parser_parse_statement_or_block(parser);
1343     if (!ontrue) {
1344         ast_delete(cond);
1345         return false;
1346     }
1347     /* check for an else */
1348     if (!strcmp(parser_tokval(parser), "else")) {
1349         /* parse into the 'else' branch */
1350         if (!parser_next(parser)) {
1351             parseerror(parser, "expected on-false branch after 'else'");
1352             ast_delete(ontrue);
1353             ast_delete(cond);
1354             return false;
1355         }
1356         onfalse = parser_parse_statement_or_block(parser);
1357         if (!onfalse) {
1358             ast_delete(ontrue);
1359             ast_delete(cond);
1360             return false;
1361         }
1362     }
1363
1364     ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
1365     *out = (ast_expression*)ifthen;
1366     return true;
1367 }
1368
1369 static bool parser_parse_while(parser_t *parser, ast_block *block, ast_expression **out)
1370 {
1371     ast_loop *aloop;
1372     ast_expression *cond, *ontrue;
1373
1374     lex_ctx ctx = parser_ctx(parser);
1375
1376     /* skip the 'while' and check for opening paren */
1377     if (!parser_next(parser) || parser->tok != '(') {
1378         parseerror(parser, "expected 'while' condition in parenthesis");
1379         return false;
1380     }
1381     /* parse into the expression */
1382     if (!parser_next(parser)) {
1383         parseerror(parser, "expected 'while' condition after opening paren");
1384         return false;
1385     }
1386     /* parse the condition */
1387     cond = parser_expression_leave(parser, false);
1388     if (!cond)
1389         return false;
1390     /* closing paren */
1391     if (parser->tok != ')') {
1392         parseerror(parser, "expected closing paren after 'while' condition");
1393         ast_delete(cond);
1394         return false;
1395     }
1396     /* parse into the 'then' branch */
1397     if (!parser_next(parser)) {
1398         parseerror(parser, "expected while-loop body");
1399         ast_delete(cond);
1400         return false;
1401     }
1402     ontrue = parser_parse_statement_or_block(parser);
1403     if (!ontrue) {
1404         ast_delete(cond);
1405         return false;
1406     }
1407
1408     aloop = ast_loop_new(ctx, NULL, cond, NULL, NULL, ontrue);
1409     *out = (ast_expression*)aloop;
1410     return true;
1411 }
1412
1413 static bool parser_parse_dowhile(parser_t *parser, ast_block *block, ast_expression **out)
1414 {
1415     ast_loop *aloop;
1416     ast_expression *cond, *ontrue;
1417
1418     lex_ctx ctx = parser_ctx(parser);
1419
1420     /* skip the 'do' and get the body */
1421     if (!parser_next(parser)) {
1422         parseerror(parser, "expected loop body");
1423         return false;
1424     }
1425     ontrue = parser_parse_statement_or_block(parser);
1426     if (!ontrue)
1427         return false;
1428
1429     /* expect the "while" */
1430     if (parser->tok != TOKEN_KEYWORD ||
1431         strcmp(parser_tokval(parser), "while"))
1432     {
1433         parseerror(parser, "expected 'while' and condition");
1434         ast_delete(ontrue);
1435         return false;
1436     }
1437
1438     /* skip the 'while' and check for opening paren */
1439     if (!parser_next(parser) || parser->tok != '(') {
1440         parseerror(parser, "expected 'while' condition in parenthesis");
1441         ast_delete(ontrue);
1442         return false;
1443     }
1444     /* parse into the expression */
1445     if (!parser_next(parser)) {
1446         parseerror(parser, "expected 'while' condition after opening paren");
1447         ast_delete(ontrue);
1448         return false;
1449     }
1450     /* parse the condition */
1451     cond = parser_expression_leave(parser, false);
1452     if (!cond)
1453         return false;
1454     /* closing paren */
1455     if (parser->tok != ')') {
1456         parseerror(parser, "expected closing paren after 'while' condition");
1457         ast_delete(ontrue);
1458         ast_delete(cond);
1459         return false;
1460     }
1461     /* parse on */
1462     if (!parser_next(parser) || parser->tok != ';') {
1463         parseerror(parser, "expected semicolon after condition");
1464         ast_delete(ontrue);
1465         ast_delete(cond);
1466         return false;
1467     }
1468
1469     if (!parser_next(parser)) {
1470         parseerror(parser, "parse error");
1471         ast_delete(ontrue);
1472         ast_delete(cond);
1473         return false;
1474     }
1475
1476     aloop = ast_loop_new(ctx, NULL, NULL, cond, NULL, ontrue);
1477     *out = (ast_expression*)aloop;
1478     return true;
1479 }
1480
1481 static bool parser_parse_for(parser_t *parser, ast_block *block, ast_expression **out)
1482 {
1483     ast_loop *aloop;
1484     ast_expression *initexpr, *cond, *increment, *ontrue;
1485     size_t oldblocklocal;
1486
1487     lex_ctx ctx = parser_ctx(parser);
1488
1489     oldblocklocal = parser->blocklocal;
1490     parser->blocklocal = parser->locals_count;
1491
1492     initexpr  = NULL;
1493     cond      = NULL;
1494     increment = NULL;
1495     ontrue    = NULL;
1496
1497     /* skip the 'while' and check for opening paren */
1498     if (!parser_next(parser) || parser->tok != '(') {
1499         parseerror(parser, "expected 'for' expressions in parenthesis");
1500         goto onerr;
1501     }
1502     /* parse into the expression */
1503     if (!parser_next(parser)) {
1504         parseerror(parser, "expected 'for' initializer after opening paren");
1505         goto onerr;
1506     }
1507
1508     if (parser->tok == TOKEN_TYPENAME) {
1509         if (opts_standard != COMPILER_GMQCC) {
1510             if (parsewarning(parser, WARN_EXTENSIONS,
1511                              "current standard does not allow variable declarations in for-loop initializers"))
1512                 goto onerr;
1513         }
1514
1515         parseerror(parser, "TODO: assignment of new variables to be non-const");
1516         goto onerr;
1517         if (!parser_variable(parser, block))
1518             goto onerr;
1519     }
1520     else if (parser->tok != ';')
1521     {
1522         initexpr = parser_expression_leave(parser, false);
1523         if (!initexpr)
1524             goto onerr;
1525     }
1526
1527     /* move on to condition */
1528     if (parser->tok != ';') {
1529         parseerror(parser, "expected semicolon after for-loop initializer");
1530         goto onerr;
1531     }
1532     if (!parser_next(parser)) {
1533         parseerror(parser, "expected for-loop condition");
1534         goto onerr;
1535     }
1536
1537     /* parse the condition */
1538     if (parser->tok != ';') {
1539         cond = parser_expression_leave(parser, false);
1540         if (!cond)
1541             goto onerr;
1542     }
1543
1544     /* move on to incrementor */
1545     if (parser->tok != ';') {
1546         parseerror(parser, "expected semicolon after for-loop initializer");
1547         goto onerr;
1548     }
1549     if (!parser_next(parser)) {
1550         parseerror(parser, "expected for-loop condition");
1551         goto onerr;
1552     }
1553
1554     /* parse the incrementor */
1555     if (parser->tok != ')') {
1556         increment = parser_expression_leave(parser, false);
1557         if (!increment)
1558             goto onerr;
1559     }
1560
1561     /* closing paren */
1562     if (parser->tok != ')') {
1563         parseerror(parser, "expected closing paren after 'for-loop' incrementor");
1564         goto onerr;
1565     }
1566     /* parse into the 'then' branch */
1567     if (!parser_next(parser)) {
1568         parseerror(parser, "expected for-loop body");
1569         goto onerr;
1570     }
1571     ontrue = parser_parse_statement_or_block(parser);
1572     if (!ontrue) {
1573         goto onerr;
1574     }
1575
1576     aloop = ast_loop_new(ctx, initexpr, cond, NULL, increment, ontrue);
1577     *out = (ast_expression*)aloop;
1578
1579     while (parser->locals_count > parser->blocklocal)
1580         parser_pop_local(parser);
1581     parser->blocklocal = oldblocklocal;
1582     return true;
1583 onerr:
1584     if (initexpr)  ast_delete(initexpr);
1585     if (cond)      ast_delete(cond);
1586     if (increment) ast_delete(increment);
1587     while (parser->locals_count > parser->blocklocal)
1588         parser_pop_local(parser);
1589     parser->blocklocal = oldblocklocal;
1590     return false;
1591 }
1592
1593 static bool parser_parse_statement(parser_t *parser, ast_block *block, ast_expression **out)
1594 {
1595     if (parser->tok == TOKEN_TYPENAME)
1596     {
1597         /* local variable */
1598         if (!block) {
1599             parseerror(parser, "cannot declare a variable from here");
1600             return false;
1601         }
1602         if (opts_standard == COMPILER_QCC) {
1603             if (parsewarning(parser, WARN_EXTENSIONS, "missing 'local' keyword when declaring a local variable"))
1604                 return false;
1605         }
1606         if (!parser_variable(parser, block))
1607             return false;
1608         *out = NULL;
1609         return true;
1610     }
1611     else if (parser->tok == TOKEN_KEYWORD)
1612     {
1613         if (!strcmp(parser_tokval(parser), "local"))
1614         {
1615             if (!block) {
1616                 parseerror(parser, "cannot declare a local variable here");
1617                 return false;
1618             }
1619             if (!parser_next(parser)) {
1620                 parseerror(parser, "expected variable declaration");
1621                 return false;
1622             }
1623             if (!parser_variable(parser, block))
1624                 return false;
1625             *out = NULL;
1626             return true;
1627         }
1628         else if (!strcmp(parser_tokval(parser), "return"))
1629         {
1630             ast_expression *exp = NULL;
1631             ast_return     *ret = NULL;
1632             ast_value      *expected = parser->function->vtype;
1633
1634             if (!parser_next(parser)) {
1635                 parseerror(parser, "expected return expression");
1636                 return false;
1637             }
1638
1639             if (parser->tok != ';') {
1640                 exp = parser_expression(parser, false);
1641                 if (!exp)
1642                     return false;
1643
1644                 if (exp->expression.vtype != expected->expression.next->expression.vtype) {
1645                     parseerror(parser, "return with invalid expression");
1646                 }
1647
1648                 ret = ast_return_new(exp->expression.node.context, exp);
1649                 if (!ret) {
1650                     ast_delete(exp);
1651                     return false;
1652                 }
1653             } else {
1654                 if (!parser_next(parser))
1655                     parseerror(parser, "parse error");
1656                 if (expected->expression.next->expression.vtype != TYPE_VOID) {
1657                     if (opts_standard != COMPILER_GMQCC)
1658                         (void)!parsewarning(parser, WARN_MISSING_RETURN_VALUES, "return without value");
1659                     else
1660                         parseerror(parser, "return without value");
1661                 }
1662                 ret = ast_return_new(parser_ctx(parser), NULL);
1663             }
1664             *out = (ast_expression*)ret;
1665             return true;
1666         }
1667         else if (!strcmp(parser_tokval(parser), "if"))
1668         {
1669             return parser_parse_if(parser, block, out);
1670         }
1671         else if (!strcmp(parser_tokval(parser), "while"))
1672         {
1673             return parser_parse_while(parser, block, out);
1674         }
1675         else if (!strcmp(parser_tokval(parser), "do"))
1676         {
1677             return parser_parse_dowhile(parser, block, out);
1678         }
1679         else if (!strcmp(parser_tokval(parser), "for"))
1680         {
1681             if (opts_standard == COMPILER_QCC) {
1682                 if (parsewarning(parser, WARN_EXTENSIONS, "for loops are not recognized in the original Quake C standard, to enable try an alternate standard --std=?"))
1683                     return false;
1684             }
1685             return parser_parse_for(parser, block, out);
1686         }
1687         parseerror(parser, "Unexpected keyword");
1688         return false;
1689     }
1690     else if (parser->tok == '{')
1691     {
1692         ast_block *inner;
1693         inner = parser_parse_block(parser);
1694         if (!inner)
1695             return false;
1696         *out = (ast_expression*)inner;
1697         return true;
1698     }
1699     else
1700     {
1701         ast_expression *exp = parser_expression(parser, false);
1702         if (!exp)
1703             return false;
1704         *out = exp;
1705         return true;
1706     }
1707 }
1708
1709 static void parser_pop_local(parser_t *parser)
1710 {
1711     parser->locals_count--;
1712     mem_d(parser->locals[parser->locals_count].name);
1713 }
1714
1715 static bool parser_parse_block_into(parser_t *parser, ast_block *block)
1716 {
1717     size_t oldblocklocal;
1718
1719     oldblocklocal = parser->blocklocal;
1720     parser->blocklocal = parser->locals_count;
1721
1722     if (!parser_next(parser)) { /* skip the '{' */
1723         parseerror(parser, "expected function body");
1724         goto cleanup;
1725     }
1726
1727     while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
1728     {
1729         ast_expression *expr;
1730         if (parser->tok == '}')
1731             break;
1732
1733         if (!parser_parse_statement(parser, block, &expr)) {
1734             parseerror(parser, "parse error");
1735             block = NULL;
1736             goto cleanup;
1737         }
1738         if (!expr)
1739             continue;
1740         if (!ast_block_exprs_add(block, expr)) {
1741             ast_delete(expr);
1742             block = NULL;
1743             goto cleanup;
1744         }
1745     }
1746
1747     if (parser->tok != '}') {
1748         block = NULL;
1749     } else {
1750         (void)parser_next(parser);
1751     }
1752
1753 cleanup:
1754     while (parser->locals_count > parser->blocklocal)
1755         parser_pop_local(parser);
1756     parser->blocklocal = oldblocklocal;
1757     return !!block;
1758 }
1759
1760 static ast_block* parser_parse_block(parser_t *parser)
1761 {
1762     ast_block *block;
1763     block = ast_block_new(parser_ctx(parser));
1764     if (!block)
1765         return NULL;
1766     if (!parser_parse_block_into(parser, block)) {
1767         ast_block_delete(block);
1768         return NULL;
1769     }
1770     return block;
1771 }
1772
1773 static ast_expression* parser_parse_statement_or_block(parser_t *parser)
1774 {
1775     ast_expression *expr = NULL;
1776     if (parser->tok == '{')
1777         return (ast_expression*)parser_parse_block(parser);
1778     if (!parser_parse_statement(parser, NULL, &expr))
1779         return NULL;
1780     return expr;
1781 }
1782
1783 static bool create_vector_members(parser_t *parser, ast_value *var,
1784                                   varentry_t *vx, varentry_t *vy, varentry_t *vz)
1785 {
1786     size_t len = strlen(var->name);
1787     vx->var = (ast_expression*)ast_member_new(ast_ctx(var), (ast_expression*)var, 0);
1788     if (!vx->var) {
1789         parseerror(parser, "failed to create vector members (out of memory)");
1790         return false;
1791     }
1792
1793     vy->var = (ast_expression*)ast_member_new(ast_ctx(var), (ast_expression*)var, 1);
1794     if (!vy->var) {
1795         ast_delete(vx->var);
1796         parseerror(parser, "failed to create vector members (out of memory)");
1797         return false;
1798     }
1799
1800     vz->var = (ast_expression*)ast_member_new(ast_ctx(var), (ast_expression*)var, 2);
1801     if (!vz->var) {
1802         ast_delete(vy->var);
1803         ast_delete(vx->var);
1804         parseerror(parser, "failed to create vector members (out of memory)");
1805         return false;
1806     }
1807
1808     if ( !(vx->name = (char*)mem_a(len+3)) ) {
1809         ast_delete(vz->var);
1810         ast_delete(vy->var);
1811         ast_delete(vx->var);
1812         parseerror(parser, "failed to create vector members (out of memory)");
1813         return false;
1814     }
1815     if ( !(vy->name = (char*)mem_a(len+3)) ) {
1816         mem_d(vx->name);
1817         ast_delete(vz->var);
1818         ast_delete(vy->var);
1819         ast_delete(vx->var);
1820         parseerror(parser, "failed to create vector members (out of memory)");
1821         return false;
1822     }
1823     if ( !(vz->name = (char*)mem_a(len+3)) ) {
1824         mem_d(vy->name);
1825         mem_d(vx->name);
1826         ast_delete(vz->var);
1827         ast_delete(vy->var);
1828         ast_delete(vx->var);
1829         parseerror(parser, "failed to create vector members (out of memory)");
1830         return false;
1831     }
1832
1833     memcpy(vx->name, var->name, len);
1834     memcpy(vy->name, var->name, len);
1835     memcpy(vz->name, var->name, len);
1836     vx->name[len] = vy->name[len] = vz->name[len] = '_';
1837     vx->name[len+1] = 'x';
1838     vy->name[len+1] = 'y';
1839     vz->name[len+1] = 'z';
1840     vx->name[len+2] = vy->name[len+2] = vz->name[len+2] = 0;
1841     return true;
1842 }
1843
1844 static bool parser_variable(parser_t *parser, ast_block *localblock)
1845 {
1846     bool          isfunc = false;
1847     ast_function *func = NULL;
1848     lex_ctx       ctx;
1849     ast_value    *var;
1850     varentry_t    varent;
1851     ast_expression *olddecl;
1852
1853     bool hadproto;
1854     bool isparam;
1855
1856     int basetype = parser_token(parser)->constval.t;
1857
1858     while (true)
1859     {
1860         hadproto = false;
1861
1862         if (!parser_next(parser)) { /* skip basetype or comma */
1863             parseerror(parser, "expected variable declaration");
1864             return false;
1865         }
1866
1867         olddecl = NULL;
1868         isfunc  = false;
1869         func    = NULL;
1870         isparam = false;
1871         ctx = parser_ctx(parser);
1872         var = parser_parse_type(parser, basetype, &isfunc);
1873
1874         if (!var)
1875             return false;
1876
1877         if (parser->tok != TOKEN_IDENT) {
1878             parseerror(parser, "expected variable name\n");
1879             return false;
1880         }
1881
1882         if (!isfunc) {
1883             if (!localblock && (olddecl = parser_find_global(parser, parser_tokval(parser)))) {
1884                 ast_value_delete(var);
1885                 parseerror(parser, "global `%s` already declared here: %s:%i",
1886                            parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1887                 return false;
1888             }
1889
1890             if (localblock) {
1891                 olddecl = parser_find_local(parser, parser_tokval(parser), parser->blocklocal, &isparam);
1892                 if (opts_standard == COMPILER_GMQCC)
1893                 {
1894                     if (olddecl)
1895                     {
1896                         if (!isparam) {
1897                             ast_value_delete(var);
1898                             parseerror(parser, "local `%s` already declared here: %s:%i",
1899                                        parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1900                             return false;
1901                         }
1902                     }
1903
1904                     if( (!isparam && olddecl) || (olddecl = parser_find_local(parser, parser_tokval(parser), 0, &isparam)) )
1905                     {
1906                         if (parsewarning(parser, WARN_LOCAL_SHADOWS,
1907                                          "local `%s` is shadowing a parameter", parser_tokval(parser)))
1908                         {
1909                             ast_value_delete(var);
1910                             parseerror(parser, "local `%s` already declared here: %s:%i",
1911                                        parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1912                             return false;
1913                         }
1914                     }
1915                 }
1916                 else
1917                 {
1918                     if (olddecl)
1919                     {
1920                         ast_value_delete(var);
1921                         if (isparam &&
1922                             parsewarning(parser, WARN_LOCAL_SHADOWS,
1923                                          "a parameter is shadowing local `%s`", parser_tokval(parser)))
1924                         {
1925                             return false;
1926                         }
1927                         else if (!isparam)
1928                         {
1929                             parseerror(parser, "local `%s` already declared here: %s:%i",
1930                                        parser_tokval(parser), ast_ctx(olddecl).file, (int)ast_ctx(olddecl).line);
1931                             return false;
1932                         }
1933                         goto nextvar;
1934                     }
1935                 }
1936             }
1937         }
1938
1939         if (!ast_value_set_name(var, parser_tokval(parser))) {
1940             parseerror(parser, "failed to set variable name\n");
1941             ast_value_delete(var);
1942             return false;
1943         }
1944
1945         if (isfunc) {
1946             /* a function was defined */
1947             ast_value *fval;
1948             ast_value *proto = NULL;
1949             bool dummy;
1950
1951             if (!localblock)
1952                 olddecl = parser_find_global(parser, parser_tokval(parser));
1953             else
1954                 olddecl = parser_find_local(parser, parser_tokval(parser), parser->blocklocal, &dummy);
1955
1956             if (olddecl) {
1957                 /* we had a prototype */
1958                 if (!ast_istype(olddecl, ast_value)) {
1959                     /* theoretically not possible you think?
1960                      * well:
1961                      * vector v;
1962                      * void() v_x = {}
1963                      * got it?
1964                      */
1965                     parseerror(parser, "cannot declare a function with the same name as a vector's member: %s",
1966                                parser_tokval(parser));
1967                     ast_value_delete(var);
1968                     return false;
1969                 }
1970
1971                 proto = (ast_value*)olddecl;
1972             }
1973
1974             /* turn var into a value of TYPE_FUNCTION, with the old var
1975              * as return type
1976              */
1977             fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
1978             func = ast_function_new(ctx, var->name, fval);
1979             if (!fval || !func) {
1980                 ast_value_delete(var);
1981                 if (fval) ast_value_delete(fval);
1982                 if (func) ast_function_delete(func);
1983                 return false;
1984             }
1985
1986             fval->expression.next = (ast_expression*)var;
1987             MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
1988
1989             /* we compare the type late here, but it's easier than
1990              * messing with the parameter-vector etc. earlier
1991              */
1992             if (proto) {
1993                 size_t param;
1994                 if (!ast_compare_type((ast_expression*)proto, (ast_expression*)fval)) {
1995                     parseerror(parser, "conflicting types for `%s`, previous declaration was here: %s:%i",
1996                                proto->name,
1997                                ast_ctx(proto).file, ast_ctx(proto).line);
1998                     ast_function_delete(func);
1999                     ast_value_delete(fval);
2000                     return false;
2001                 }
2002                 /* copy over the parameter names */
2003                 for (param = 0; param < fval->expression.params_count; ++param)
2004                     ast_value_set_name(proto->expression.params[param], fval->expression.params[param]->name);
2005
2006                 /* now ditch the rest of the new data */
2007                 ast_function_delete(func);
2008                 ast_value_delete(fval);
2009                 fval = proto;
2010                 func = proto->constval.vfunc;
2011                 hadproto = true;
2012             }
2013             else
2014             {
2015                 if (!parser_t_functions_add(parser, func)) {
2016                     ast_function_delete(func);
2017                     ast_value_delete(fval);
2018                     return false;
2019                 }
2020             }
2021
2022             var = fval;
2023         }
2024
2025         if (!hadproto) {
2026             varent.name = util_strdup(var->name);
2027             varent.var = (ast_expression*)var;
2028             if (var->expression.vtype == TYPE_VECTOR)
2029             {
2030                 varentry_t vx, vy, vz;
2031                 if (!create_vector_members(parser, var, &vx, &vy, &vz)) {
2032                     ast_delete(var);
2033                     return false;
2034                 }
2035
2036                 if (!localblock) {
2037                     (void)!parser_t_globals_add(parser, varent);
2038                     (void)!parser_t_globals_add(parser, vx);
2039                     (void)!parser_t_globals_add(parser, vy);
2040                     (void)!parser_t_globals_add(parser, vz);
2041                 } else {
2042                     (void)!parser_t_locals_add(parser, varent);
2043                     (void)!parser_t_locals_add(parser, vx);
2044                     (void)!parser_t_locals_add(parser, vy);
2045                     (void)!parser_t_locals_add(parser, vz);
2046                     if (!ast_block_locals_add(localblock, var) ||
2047                         !ast_block_collect(localblock, vx.var) ||
2048                         !ast_block_collect(localblock, vy.var) ||
2049                         !ast_block_collect(localblock, vz.var))
2050                     {
2051                         parser_pop_local(parser);
2052                         parser_pop_local(parser);
2053                         parser_pop_local(parser);
2054                         parser_pop_local(parser);
2055                         ast_value_delete(var);
2056                         return false;
2057                     }
2058                 }
2059             }
2060             else
2061             {
2062                 if ( (!localblock && !parser_t_globals_add(parser, varent)) ||
2063                      ( localblock && !parser_t_locals_add(parser, varent)) )
2064                 {
2065                     ast_value_delete(var);
2066                     return false;
2067                 }
2068                 if (localblock && !ast_block_locals_add(localblock, var))
2069                 {
2070                     parser_pop_local(parser);
2071                     ast_value_delete(var);
2072                     return false;
2073                 }
2074             }
2075         }
2076
2077 nextvar:
2078         if (!parser_next(parser)) {
2079             ast_value_delete(var);
2080             return false;
2081         }
2082
2083         if (parser->tok == ';') {
2084             if (!parser_next(parser))
2085                 return parser->tok == TOKEN_EOF;
2086             return true;
2087         }
2088
2089         if (parser->tok == ',') {
2090             /* another var */
2091             continue;
2092         }
2093
2094         if (parser->tok != '=') {
2095             parseerror(parser, "expected '=' or ';'");
2096             return false;
2097         }
2098
2099         if (!parser_next(parser))
2100             return false;
2101
2102         if (parser->tok == '#') {
2103             if (localblock) {
2104                 parseerror(parser, "cannot declare builtins within functions");
2105                 return false;
2106             }
2107             if (!isfunc || !func) {
2108                 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
2109                 return false;
2110             }
2111             if (!parser_next(parser)) {
2112                 parseerror(parser, "expected builtin number");
2113                 return false;
2114             }
2115             if (parser->tok != TOKEN_INTCONST) {
2116                 parseerror(parser, "builtin number must be an integer constant");
2117                 return false;
2118             }
2119             if (parser_token(parser)->constval.i <= 0) {
2120                 parseerror(parser, "builtin number must be positive integer greater than zero");
2121                 return false;
2122             }
2123
2124             func->builtin = -parser_token(parser)->constval.i;
2125
2126             if (!parser_next(parser))
2127                 return false;
2128         } else if (parser->tok == '{' || parser->tok == '[') {
2129             /* function body */
2130             ast_function *old;
2131             ast_block *block;
2132             size_t     parami;
2133
2134             ast_expression *fld_think, *fld_nextthink, *fld_frame;
2135             ast_expression *gbl_time, *gbl_self;
2136             ast_expression *framenum, *nextthink;
2137             bool            has_frame_think;
2138
2139             has_frame_think = false;
2140             old = parser->function;
2141
2142             if (localblock) {
2143                 parseerror(parser, "cannot declare functions within functions");
2144                 return false;
2145             }
2146
2147             if (parser->tok == '[') {
2148                 /* got a frame definition: [ framenum, nextthink ]
2149                  * this translates to:
2150                  * self.frame = framenum;
2151                  * self.nextthink = time + 0.1;
2152                  * self.think = nextthink;
2153                  */
2154                 nextthink = NULL;
2155
2156                 fld_think     = parser_find_field(parser, "think");
2157                 fld_nextthink = parser_find_field(parser, "nextthink");
2158                 fld_frame     = parser_find_field(parser, "frame");
2159                 if (!fld_think || !fld_nextthink || !fld_frame) {
2160                     parseerror(parser, "cannot use [frame,think] notation without the required fields");
2161                     parseerror(parser, "please declare the following entityfields: `frame`, `think`, `nextthink`");
2162                     return false;
2163                 }
2164                 gbl_time      = parser_find_global(parser, "time");
2165                 gbl_self      = parser_find_global(parser, "self");
2166                 if (!gbl_time || !gbl_self) {
2167                     parseerror(parser, "cannot use [frame,think] notation without the required globals");
2168                     parseerror(parser, "please declare the following globals: `time`, `self`");
2169                     return false;
2170                 }
2171
2172                 if (!parser_next(parser)) {
2173                     return false;
2174                 }
2175
2176                 framenum = parser_expression_leave(parser, true);
2177                 if (!framenum) {
2178                     parseerror(parser, "expected a framenumber constant in[frame,think] notation");
2179                     return false;
2180                 }
2181                 if (!ast_istype(framenum, ast_value) || !( (ast_value*)framenum )->isconst) {
2182                     ast_unref(framenum);
2183                     parseerror(parser, "framenumber in [frame,think] notation must be a constant");
2184                 }
2185
2186                 if (parser->tok != ',') {
2187                     ast_unref(framenum);
2188                     parseerror(parser, "expected comma after frame number in [frame,think] notation");
2189                     parseerror(parser, "Got a %i\n", parser->tok);
2190                     return false;
2191                 }
2192
2193                 if (!parser_next(parser)) {
2194                     ast_unref(framenum);
2195                     return false;
2196                 }
2197
2198                 if (parser->tok == TOKEN_IDENT && !parser_find_var(parser, parser_tokval(parser)))
2199                 {
2200                     /* qc allows the use of not-yet-declared functions here
2201                      * - this automatically creates a prototype */
2202                     varentry_t      varent;
2203                     ast_value      *thinkfunc;
2204                     ast_expression *functype = fld_think->expression.next;
2205
2206                     thinkfunc = ast_value_new(parser_ctx(parser), parser_tokval(parser), functype->expression.vtype);
2207                     if (!thinkfunc || !ast_type_adopt(nextthink, functype)) {
2208                         ast_unref(framenum);
2209                         parseerror(parser, "failed to create implicit prototype for `%s`", parser_tokval(parser));
2210                         return false;
2211                     }
2212
2213                     if (!parser_next(parser)) {
2214                         ast_unref(framenum);
2215                         return false;
2216                     }
2217
2218                     varent.var = (ast_expression*)thinkfunc;
2219                     varent.name = util_strdup(thinkfunc->name);
2220                     if (nextthink->expression.vtype == TYPE_FUNCTION)
2221                     {
2222                         ast_function *func;
2223
2224                         func = ast_function_new(parser_ctx(parser), thinkfunc->name, thinkfunc);
2225                         if (!func) {
2226                             ast_delete(nextthink);
2227                             ast_unref(framenum);
2228                             parseerror(parser, "failed to create function for implicit prototype for `%s`",
2229                                        thinkfunc->name);
2230                             return false;
2231                         }
2232                         (void)!parser_t_functions_add(parser, func);
2233                         (void)!parser_t_globals_add(parser, varent);
2234                     }
2235                     else
2236                         (void)!parser_t_globals_add(parser, varent);
2237                     nextthink = (ast_expression*)thinkfunc;
2238
2239                 } else {
2240                     nextthink = parser_expression_leave(parser, true);
2241                     if (!nextthink) {
2242                         ast_unref(framenum);
2243                         parseerror(parser, "expected a think-function in [frame,think] notation");
2244                         return false;
2245                     }
2246                 }
2247
2248                 if (!ast_istype(nextthink, ast_value) || !( (ast_value*)nextthink )->isconst) {
2249                     ast_unref(nextthink);
2250                     ast_unref(framenum);
2251                     parseerror(parser, "think-function in [frame,think] notation must be a constant");
2252                 }
2253
2254                 if (parser->tok != ']') {
2255                     parseerror(parser, "expected closing `]` for [frame,think] notation");
2256                     ast_unref(nextthink);
2257                     ast_unref(framenum);
2258                     return false;
2259                 }
2260
2261                 if (!parser_next(parser)) {
2262                     ast_unref(nextthink);
2263                     ast_unref(framenum);
2264                     return false;
2265                 }
2266
2267                 if (parser->tok != '{') {
2268                     parseerror(parser, "a function body has to be declared after a [frame,think] declaration");
2269                     ast_unref(nextthink);
2270                     ast_unref(framenum);
2271                     return false;
2272                 }
2273
2274                 has_frame_think = true;
2275             }
2276
2277             block = ast_block_new(parser_ctx(parser));
2278             if (!block) {
2279                 parseerror(parser, "failed to allocate block");
2280                 return false;
2281             }
2282
2283             if (has_frame_think) {
2284                 lex_ctx ctx;
2285                 ast_expression *self_frame;
2286                 ast_expression *self_nextthink;
2287                 ast_expression *self_think;
2288                 ast_expression *time_plus_1;
2289                 ast_store *store_frame;
2290                 ast_store *store_nextthink;
2291                 ast_store *store_think;
2292
2293                 ctx = parser_ctx(parser);
2294                 self_frame     = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_frame);
2295                 self_nextthink = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_nextthink);
2296                 self_think     = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_think);
2297
2298                 time_plus_1    = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F,
2299                                  gbl_time, (ast_expression*)parser_const_float(parser, 0.1));
2300
2301                 store_frame     = ast_store_new(ctx, INSTR_STOREP_F,   self_frame,     framenum);
2302                 store_nextthink = ast_store_new(ctx, INSTR_STOREP_F,   self_nextthink, time_plus_1);
2303                 store_think     = ast_store_new(ctx, INSTR_STOREP_FNC, self_think,     nextthink);
2304
2305                 if (!ast_block_exprs_add(block, (ast_expression*)store_frame)     ||
2306                     !ast_block_exprs_add(block, (ast_expression*)store_nextthink) ||
2307                     !ast_block_exprs_add(block, (ast_expression*)store_think) )
2308                 {
2309                     parseerror(parser, "failed to generate code for [frame,think]");
2310                     ast_block_delete(block);
2311                     return false;
2312                 }
2313             }
2314
2315             for (parami = 0; parami < var->expression.params_count; ++parami) {
2316                 ast_value *param = var->expression.params[parami];
2317                 varentry_t vx, vy, vz;
2318
2319                 if (param->expression.vtype != TYPE_VECTOR &&
2320                     (param->expression.vtype != TYPE_FIELD ||
2321                      param->expression.next->expression.vtype != TYPE_VECTOR))
2322                 {
2323                     continue;
2324                 }
2325
2326                 if (!create_vector_members(parser, param, &vx, &vy, &vz)) {
2327                     ast_block_delete(block);
2328                     return false;
2329                 }
2330
2331                 (void)!parser_t_locals_add(parser, vx);
2332                 (void)!parser_t_locals_add(parser, vy);
2333                 (void)!parser_t_locals_add(parser, vz);
2334                 if (!ast_block_collect(block, vx.var) ||
2335                     !ast_block_collect(block, vy.var) ||
2336                     !ast_block_collect(block, vz.var) )
2337                 {
2338                     parser_pop_local(parser);
2339                     parser_pop_local(parser);
2340                     parser_pop_local(parser);
2341                     ast_block_delete(block);
2342                     return false;
2343                 }
2344             }
2345
2346             parser->function = func;
2347             if (!parser_parse_block_into(parser, block)) {
2348                 ast_block_delete(block);
2349                 parser->function = old;
2350                 return false;
2351             }
2352             parser->function = old;
2353
2354             if (!block)
2355                 return false;
2356
2357             if (!ast_function_blocks_add(func, block)) {
2358                 ast_block_delete(block);
2359                 return false;
2360             }
2361
2362             if (parser->tok == ';')
2363                 return parser_next(parser) || parser->tok == TOKEN_EOF;
2364             else if (opts_standard == COMPILER_QCC)
2365                 parseerror(parser, "missing semicolon after function body (mandatory with -std=qcc)");
2366             return true;
2367         } else {
2368             ast_expression *cexp;
2369             ast_value      *cval;
2370
2371             cexp = parser_expression_leave(parser, true);
2372             cval = (ast_value*)cexp;
2373             if (!ast_istype(cval, ast_value) || !cval->isconst)
2374                 parseerror(parser, "cannot initialize a global constant variable with a non-constant expression");
2375             else
2376             {
2377                 var->isconst = true;
2378                 memcpy(&var->constval, &cval->constval, sizeof(var->constval));
2379                 memset(&cval->constval, 0, sizeof(cval->constval));
2380                 ast_unref(cval);
2381             }
2382         }
2383
2384         if (parser->tok == ',') {
2385             /* another */
2386             continue;
2387         }
2388
2389         if (parser->tok != ';') {
2390             parseerror(parser, "missing semicolon");
2391             return false;
2392         }
2393
2394         (void)parser_next(parser);
2395
2396         return true;
2397     }
2398 }
2399
2400 static bool parser_do(parser_t *parser)
2401 {
2402     if (parser->tok == TOKEN_TYPENAME)
2403     {
2404         return parser_variable(parser, NULL);
2405     }
2406     else if (parser->tok == TOKEN_KEYWORD)
2407     {
2408         /* handle 'var' and 'const' */
2409         return false;
2410     }
2411     else if (parser->tok == '.')
2412     {
2413         ast_value *var;
2414         ast_value *typevar;
2415         ast_value *fld;
2416         ast_expression *oldex;
2417         bool       isfunc = false;
2418         int        basetype;
2419         lex_ctx    ctx = parser_ctx(parser);
2420         varentry_t varent;
2421
2422         /* entity-member declaration */
2423         if (!parser_next(parser) || parser->tok != TOKEN_TYPENAME) {
2424             parseerror(parser, "expected member variable definition");
2425             return false;
2426         }
2427
2428         /* remember the base/return type */
2429         basetype = parser_token(parser)->constval.t;
2430
2431         /* parse into the declaration */
2432         if (!parser_next(parser)) {
2433             parseerror(parser, "expected field def");
2434             return false;
2435         }
2436
2437         /* parse the field type fully */
2438         typevar = var = parser_parse_type(parser, basetype, &isfunc);
2439         if (!var)
2440             return false;
2441
2442         while (true) {
2443             var = ast_value_copy(typevar);
2444             /* now the field name */
2445             if (parser->tok != TOKEN_IDENT) {
2446                 parseerror(parser, "expected field name");
2447                 ast_delete(var);
2448                 return false;
2449             }
2450
2451             /* check for an existing field
2452              * in original qc we also have to check for an existing
2453              * global named like the field
2454              */
2455             if (opts_standard == COMPILER_QCC) {
2456                 if (parser_find_global(parser, parser_tokval(parser))) {
2457                     parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
2458                     ast_delete(var);
2459                     return false;
2460                 }
2461             }
2462
2463             if (isfunc) {
2464                 ast_value *fval;
2465                 fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
2466                 if (!fval) {
2467                     ast_value_delete(var);
2468                     return false;
2469                 }
2470                 fval->expression.next = (ast_expression*)var;
2471                 MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
2472                 var = fval;
2473             }
2474
2475             /* turn it into a field */
2476             fld = ast_value_new(ctx, parser_tokval(parser), TYPE_FIELD);
2477             fld->expression.next = (ast_expression*)var;
2478
2479             if ( (oldex = parser_find_field(parser, parser_tokval(parser)))) {
2480                 if (ast_istype(oldex, ast_member)) {
2481                     parseerror(parser, "cannot declare a field with the same name as a vector component, component %s has been declared here: %s:%i",
2482                                parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line);
2483                     ast_delete(fld);
2484                     return false;
2485                 }
2486                 if (!ast_istype(oldex, ast_value)) {
2487                     /* not possible / sanity check */
2488                     parseerror(parser, "internal error: %s is not an ast_value", parser_tokval(parser));
2489                     ast_delete(fld);
2490                     return false;
2491                 }
2492
2493                 if (!ast_compare_type(oldex, (ast_expression*)fld)) {
2494                     parseerror(parser, "field %s has previously been declared with a different type here: %s:%i",
2495                                parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line);
2496                     ast_delete(fld);
2497                     return false;
2498                 } else {
2499                     if (parsewarning(parser, WARN_FIELD_REDECLARED, "field %s has already been declared here: %s:%i",
2500                                      parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line))
2501                     {
2502                         ast_delete(fld);
2503                         return false;
2504                     }
2505                 }
2506
2507                 ast_delete(fld);
2508                 goto nextfield;
2509             }
2510
2511             varent.var = (ast_expression*)fld;
2512             varent.name = util_strdup(fld->name);
2513             (void)!parser_t_fields_add(parser, varent);
2514
2515             if (var->expression.vtype == TYPE_VECTOR)
2516             {
2517                 /* create _x, _y and _z fields as well */
2518                 varentry_t vx, vy, vz;
2519                 if (!create_vector_members(parser, fld, &vx, &vy, &vz)) {
2520                     ast_delete(fld);
2521                     return false;
2522                 }
2523                 (void)!parser_t_fields_add(parser, vx);
2524                 (void)!parser_t_fields_add(parser, vy);
2525                 (void)!parser_t_fields_add(parser, vz);
2526             }
2527
2528 nextfield:
2529             if (!parser_next(parser)) {
2530                 parseerror(parser, "expected semicolon or another field name");
2531                 return false;
2532             }
2533             if (parser->tok == ';')
2534                 break;
2535             if (parser->tok != ',' || !parser_next(parser)) {
2536                 parseerror(parser, "expected semicolon or another field name");
2537                 return false;
2538             }
2539         }
2540         ast_delete(typevar);
2541
2542         /* skip the semicolon */
2543         if (!parser_next(parser))
2544             return parser->tok == TOKEN_EOF;
2545
2546         return true;
2547     }
2548     else if (parser->tok == '$')
2549     {
2550         if (!parser_next(parser)) {
2551             parseerror(parser, "parse error");
2552             return false;
2553         }
2554     }
2555     else
2556     {
2557         parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
2558         return false;
2559     }
2560     return true;
2561 }
2562
2563 static parser_t *parser;
2564
2565 bool parser_init()
2566 {
2567     parser = (parser_t*)mem_a(sizeof(parser_t));
2568     if (!parser)
2569         return false;
2570
2571     memset(parser, 0, sizeof(*parser));
2572     return true;
2573 }
2574
2575 bool parser_compile(const char *filename)
2576 {
2577     parser->lex = lex_open(filename);
2578     if (!parser->lex) {
2579         printf("failed to open file \"%s\"\n", filename);
2580         return false;
2581     }
2582
2583     /* initial lexer/parser state */
2584     parser->lex->flags.noops = true;
2585
2586     if (parser_next(parser))
2587     {
2588         while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
2589         {
2590             if (!parser_do(parser)) {
2591                 if (parser->tok == TOKEN_EOF)
2592                     parseerror(parser, "unexpected eof");
2593                 else if (!parser->errors)
2594                     parseerror(parser, "parse error\n");
2595                 lex_close(parser->lex);
2596                 parser->lex = NULL;
2597                 return false;
2598             }
2599         }
2600     }
2601
2602     lex_close(parser->lex);
2603     parser->lex = NULL;
2604
2605     return !parser->errors;
2606 }
2607
2608 void parser_cleanup()
2609 {
2610     size_t i;
2611     for (i = 0; i < parser->functions_count; ++i) {
2612         ast_delete(parser->functions[i]);
2613     }
2614     for (i = 0; i < parser->imm_vector_count; ++i) {
2615         ast_delete(parser->imm_vector[i]);
2616     }
2617     for (i = 0; i < parser->imm_string_count; ++i) {
2618         ast_delete(parser->imm_string[i]);
2619     }
2620     for (i = 0; i < parser->imm_float_count; ++i) {
2621         ast_delete(parser->imm_float[i]);
2622     }
2623     for (i = 0; i < parser->fields_count; ++i) {
2624         ast_delete(parser->fields[i].var);
2625         mem_d(parser->fields[i].name);
2626     }
2627     for (i = 0; i < parser->globals_count; ++i) {
2628         ast_delete(parser->globals[i].var);
2629         mem_d(parser->globals[i].name);
2630     }
2631     MEM_VECTOR_CLEAR(parser, functions);
2632     MEM_VECTOR_CLEAR(parser, imm_vector);
2633     MEM_VECTOR_CLEAR(parser, imm_string);
2634     MEM_VECTOR_CLEAR(parser, imm_float);
2635     MEM_VECTOR_CLEAR(parser, globals);
2636     MEM_VECTOR_CLEAR(parser, fields);
2637     MEM_VECTOR_CLEAR(parser, locals);
2638
2639     mem_d(parser);
2640 }
2641
2642 bool parser_finish(const char *output)
2643 {
2644     size_t i;
2645     ir_builder *ir;
2646
2647     if (!parser->errors)
2648     {
2649         ir = ir_builder_new("gmqcc_out");
2650         if (!ir) {
2651             printf("failed to allocate builder\n");
2652             return false;
2653         }
2654
2655         for (i = 0; i < parser->imm_float_count; ++i) {
2656             if (!ast_global_codegen(parser->imm_float[i], ir)) {
2657                 printf("failed to generate global %s\n", parser->imm_float[i]->name);
2658                 ir_builder_delete(ir);
2659                 return false;
2660             }
2661         }
2662         for (i = 0; i < parser->imm_string_count; ++i) {
2663             if (!ast_global_codegen(parser->imm_string[i], ir)) {
2664                 printf("failed to generate global %s\n", parser->imm_string[i]->name);
2665                 ir_builder_delete(ir);
2666                 return false;
2667             }
2668         }
2669         for (i = 0; i < parser->imm_vector_count; ++i) {
2670             if (!ast_global_codegen(parser->imm_vector[i], ir)) {
2671                 printf("failed to generate global %s\n", parser->imm_vector[i]->name);
2672                 ir_builder_delete(ir);
2673                 return false;
2674             }
2675         }
2676         for (i = 0; i < parser->fields_count; ++i) {
2677             ast_value *field;
2678             bool isconst;
2679             if (!ast_istype(parser->fields[i].var, ast_value))
2680                 continue;
2681             field = (ast_value*)parser->fields[i].var;
2682             isconst = field->isconst;
2683             field->isconst = false;
2684             if (!ast_global_codegen((ast_value*)field, ir)) {
2685                 printf("failed to generate field %s\n", field->name);
2686                 ir_builder_delete(ir);
2687                 return false;
2688             }
2689             if (isconst) {
2690                 ir_value *ifld;
2691                 ast_expression *subtype;
2692                 field->isconst = true;
2693                 subtype = field->expression.next;
2694                 ifld = ir_builder_create_field(ir, field->name, subtype->expression.vtype);
2695                 if (subtype->expression.vtype == TYPE_FIELD)
2696                     ifld->fieldtype = subtype->expression.next->expression.vtype;
2697                 else if (subtype->expression.vtype == TYPE_FUNCTION)
2698                     ifld->outtype = subtype->expression.next->expression.vtype;
2699                 (void)!ir_value_set_field(field->ir_v, ifld);
2700             }
2701         }
2702         for (i = 0; i < parser->globals_count; ++i) {
2703             if (!ast_istype(parser->globals[i].var, ast_value))
2704                 continue;
2705             if (!ast_global_codegen((ast_value*)(parser->globals[i].var), ir)) {
2706                 printf("failed to generate global %s\n", parser->globals[i].name);
2707                 ir_builder_delete(ir);
2708                 return false;
2709             }
2710         }
2711         for (i = 0; i < parser->functions_count; ++i) {
2712             if (!ast_function_codegen(parser->functions[i], ir)) {
2713                 printf("failed to generate function %s\n", parser->functions[i]->name);
2714                 ir_builder_delete(ir);
2715                 return false;
2716             }
2717             if (!ir_function_finalize(parser->functions[i]->ir_func)) {
2718                 printf("failed to finalize function %s\n", parser->functions[i]->name);
2719                 ir_builder_delete(ir);
2720                 return false;
2721             }
2722         }
2723
2724         if (opts_dump)
2725             ir_builder_dump(ir, printf);
2726
2727         if (!ir_builder_generate(ir, output)) {
2728             printf("*** failed to generate output file\n");
2729             ir_builder_delete(ir);
2730             return false;
2731         }
2732
2733         ir_builder_delete(ir);
2734         return true;
2735     }
2736
2737     printf("*** there were compile errors\n");
2738     return false;
2739 }