]> git.xonotic.org Git - xonotic/gmqcc.git/blob - parser.c
Removing old commented-out code
[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                 fld_think     = parser_find_field(parser, "think");
2155                 fld_nextthink = parser_find_field(parser, "nextthink");
2156                 fld_frame     = parser_find_field(parser, "frame");
2157                 if (!fld_think || !fld_nextthink || !fld_frame) {
2158                     parseerror(parser, "cannot use [frame,think] notation without the required fields");
2159                     parseerror(parser, "please declare the following entityfields: `frame`, `think`, `nextthink`");
2160                     return false;
2161                 }
2162                 gbl_time      = parser_find_global(parser, "time");
2163                 gbl_self      = parser_find_global(parser, "self");
2164                 if (!gbl_time || !gbl_self) {
2165                     parseerror(parser, "cannot use [frame,think] notation without the required globals");
2166                     parseerror(parser, "please declare the following globals: `time`, `self`");
2167                     return false;
2168                 }
2169
2170                 if (!parser_next(parser)) {
2171                     return false;
2172                 }
2173
2174                 framenum = parser_expression_leave(parser, true);
2175                 if (!framenum) {
2176                     parseerror(parser, "expected a framenumber constant in[frame,think] notation");
2177                     return false;
2178                 }
2179                 if (!ast_istype(framenum, ast_value) || !( (ast_value*)framenum )->isconst) {
2180                     ast_unref(framenum);
2181                     parseerror(parser, "framenumber in [frame,think] notation must be a constant");
2182                 }
2183
2184                 if (parser->tok != ',') {
2185                     ast_unref(framenum);
2186                     parseerror(parser, "expected comma after frame number in [frame,think] notation");
2187                     parseerror(parser, "Got a %i\n", parser->tok);
2188                     return false;
2189                 }
2190
2191                 if (!parser_next(parser)) {
2192                     ast_unref(framenum);
2193                     return false;
2194                 }
2195
2196                 nextthink = parser_expression_leave(parser, true);
2197                 if (!nextthink) {
2198                     ast_unref(framenum);
2199                     parseerror(parser, "expected a think-function in [frame,think] notation");
2200                     return false;
2201                 }
2202
2203                 if (!ast_istype(nextthink, ast_value) || !( (ast_value*)nextthink )->isconst) {
2204                     ast_unref(nextthink);
2205                     ast_unref(framenum);
2206                     parseerror(parser, "think-function in [frame,think] notation must be a constant");
2207                 }
2208
2209                 if (parser->tok != ']') {
2210                     parseerror(parser, "expected closing `]` for [frame,think] notation");
2211                     ast_unref(nextthink);
2212                     ast_unref(framenum);
2213                     return false;
2214                 }
2215
2216                 if (!parser_next(parser)) {
2217                     ast_unref(nextthink);
2218                     ast_unref(framenum);
2219                     return false;
2220                 }
2221
2222                 if (parser->tok != '{') {
2223                     parseerror(parser, "a function body has to be declared after a [frame,think] declaration");
2224                     ast_unref(nextthink);
2225                     ast_unref(framenum);
2226                     return false;
2227                 }
2228
2229                 has_frame_think = true;
2230             }
2231
2232             block = ast_block_new(parser_ctx(parser));
2233             if (!block) {
2234                 parseerror(parser, "failed to allocate block");
2235                 return false;
2236             }
2237
2238             if (has_frame_think) {
2239                 lex_ctx ctx;
2240                 ast_expression *self_frame;
2241                 ast_expression *self_nextthink;
2242                 ast_expression *self_think;
2243                 ast_expression *time_plus_1;
2244                 ast_store *store_frame;
2245                 ast_store *store_nextthink;
2246                 ast_store *store_think;
2247
2248                 ctx = parser_ctx(parser);
2249                 self_frame     = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_frame);
2250                 self_nextthink = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_nextthink);
2251                 self_think     = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_think);
2252
2253                 time_plus_1    = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F,
2254                                  gbl_time, (ast_expression*)parser_const_float(parser, 0.1));
2255
2256                 store_frame     = ast_store_new(ctx, INSTR_STOREP_F,   self_frame,     framenum);
2257                 store_nextthink = ast_store_new(ctx, INSTR_STOREP_F,   self_nextthink, time_plus_1);
2258                 store_think     = ast_store_new(ctx, INSTR_STOREP_FNC, self_think,     nextthink);
2259
2260                 if (!ast_block_exprs_add(block, (ast_expression*)store_frame)     ||
2261                     !ast_block_exprs_add(block, (ast_expression*)store_nextthink) ||
2262                     !ast_block_exprs_add(block, (ast_expression*)store_think) )
2263                 {
2264                     parseerror(parser, "failed to generate code for [frame,think]");
2265                     ast_block_delete(block);
2266                     return false;
2267                 }
2268             }
2269
2270             for (parami = 0; parami < var->expression.params_count; ++parami) {
2271                 ast_value *param = var->expression.params[parami];
2272                 varentry_t vx, vy, vz;
2273
2274                 if (param->expression.vtype != TYPE_VECTOR &&
2275                     (param->expression.vtype != TYPE_FIELD ||
2276                      param->expression.next->expression.vtype != TYPE_VECTOR))
2277                 {
2278                     continue;
2279                 }
2280
2281                 if (!create_vector_members(parser, param, &vx, &vy, &vz)) {
2282                     ast_block_delete(block);
2283                     return false;
2284                 }
2285
2286                 (void)!parser_t_locals_add(parser, vx);
2287                 (void)!parser_t_locals_add(parser, vy);
2288                 (void)!parser_t_locals_add(parser, vz);
2289                 if (!ast_block_collect(block, vx.var) ||
2290                     !ast_block_collect(block, vy.var) ||
2291                     !ast_block_collect(block, vz.var) )
2292                 {
2293                     parser_pop_local(parser);
2294                     parser_pop_local(parser);
2295                     parser_pop_local(parser);
2296                     ast_block_delete(block);
2297                     return false;
2298                 }
2299             }
2300
2301             parser->function = func;
2302             if (!parser_parse_block_into(parser, block)) {
2303                 ast_block_delete(block);
2304                 parser->function = old;
2305                 return false;
2306             }
2307             parser->function = old;
2308
2309             if (!block)
2310                 return false;
2311
2312             if (!ast_function_blocks_add(func, block)) {
2313                 ast_block_delete(block);
2314                 return false;
2315             }
2316
2317             if (parser->tok == ';')
2318                 return parser_next(parser) || parser->tok == TOKEN_EOF;
2319             else if (opts_standard == COMPILER_QCC)
2320                 parseerror(parser, "missing semicolon after function body (mandatory with -std=qcc)");
2321             return true;
2322         } else {
2323             ast_expression *cexp;
2324             ast_value      *cval;
2325
2326             cexp = parser_expression_leave(parser, true);
2327             cval = (ast_value*)cexp;
2328             if (!ast_istype(cval, ast_value) || !cval->isconst)
2329                 parseerror(parser, "cannot initialize a global constant variable with a non-constant expression");
2330             else
2331             {
2332                 var->isconst = true;
2333                 memcpy(&var->constval, &cval->constval, sizeof(var->constval));
2334                 memset(&cval->constval, 0, sizeof(cval->constval));
2335                 ast_unref(cval);
2336             }
2337         }
2338
2339         if (parser->tok == ',') {
2340             /* another */
2341             continue;
2342         }
2343
2344         if (parser->tok != ';') {
2345             parseerror(parser, "missing semicolon");
2346             return false;
2347         }
2348
2349         (void)parser_next(parser);
2350
2351         return true;
2352     }
2353 }
2354
2355 static bool parser_do(parser_t *parser)
2356 {
2357     if (parser->tok == TOKEN_TYPENAME)
2358     {
2359         return parser_variable(parser, NULL);
2360     }
2361     else if (parser->tok == TOKEN_KEYWORD)
2362     {
2363         /* handle 'var' and 'const' */
2364         return false;
2365     }
2366     else if (parser->tok == '.')
2367     {
2368         ast_value *var;
2369         ast_value *typevar;
2370         ast_value *fld;
2371         ast_expression *oldex;
2372         bool       isfunc = false;
2373         int        basetype;
2374         lex_ctx    ctx = parser_ctx(parser);
2375         varentry_t varent;
2376
2377         /* entity-member declaration */
2378         if (!parser_next(parser) || parser->tok != TOKEN_TYPENAME) {
2379             parseerror(parser, "expected member variable definition");
2380             return false;
2381         }
2382
2383         /* remember the base/return type */
2384         basetype = parser_token(parser)->constval.t;
2385
2386         /* parse into the declaration */
2387         if (!parser_next(parser)) {
2388             parseerror(parser, "expected field def");
2389             return false;
2390         }
2391
2392         /* parse the field type fully */
2393         typevar = var = parser_parse_type(parser, basetype, &isfunc);
2394         if (!var)
2395             return false;
2396
2397         while (true) {
2398             var = ast_value_copy(typevar);
2399             /* now the field name */
2400             if (parser->tok != TOKEN_IDENT) {
2401                 parseerror(parser, "expected field name");
2402                 ast_delete(var);
2403                 return false;
2404             }
2405
2406             /* check for an existing field
2407              * in original qc we also have to check for an existing
2408              * global named like the field
2409              */
2410             if (opts_standard == COMPILER_QCC) {
2411                 if (parser_find_global(parser, parser_tokval(parser))) {
2412                     parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
2413                     ast_delete(var);
2414                     return false;
2415                 }
2416             }
2417
2418             if (isfunc) {
2419                 ast_value *fval;
2420                 fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
2421                 if (!fval) {
2422                     ast_value_delete(var);
2423                     return false;
2424                 }
2425                 fval->expression.next = (ast_expression*)var;
2426                 MEM_VECTOR_MOVE(&var->expression, params, &fval->expression, params);
2427                 var = fval;
2428             }
2429
2430             /* turn it into a field */
2431             fld = ast_value_new(ctx, parser_tokval(parser), TYPE_FIELD);
2432             fld->expression.next = (ast_expression*)var;
2433
2434             if ( (oldex = parser_find_field(parser, parser_tokval(parser)))) {
2435                 if (ast_istype(oldex, ast_member)) {
2436                     parseerror(parser, "cannot declare a field with the same name as a vector component, component %s has been declared here: %s:%i",
2437                                parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line);
2438                     ast_delete(fld);
2439                     return false;
2440                 }
2441                 if (!ast_istype(oldex, ast_value)) {
2442                     /* not possible / sanity check */
2443                     parseerror(parser, "internal error: %s is not an ast_value", parser_tokval(parser));
2444                     ast_delete(fld);
2445                     return false;
2446                 }
2447
2448                 if (!ast_compare_type(oldex, (ast_expression*)fld)) {
2449                     parseerror(parser, "field %s has previously been declared with a different type here: %s:%i",
2450                                parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line);
2451                     ast_delete(fld);
2452                     return false;
2453                 } else {
2454                     if (parsewarning(parser, WARN_FIELD_REDECLARED, "field %s has already been declared here: %s:%i",
2455                                      parser_tokval(parser), ast_ctx(oldex).file, (int)ast_ctx(oldex).line))
2456                     {
2457                         ast_delete(fld);
2458                         return false;
2459                     }
2460                 }
2461
2462                 ast_delete(fld);
2463                 goto nextfield;
2464             }
2465
2466             varent.var = (ast_expression*)fld;
2467             varent.name = util_strdup(fld->name);
2468             (void)!parser_t_fields_add(parser, varent);
2469
2470             if (var->expression.vtype == TYPE_VECTOR)
2471             {
2472                 /* create _x, _y and _z fields as well */
2473                 varentry_t vx, vy, vz;
2474                 if (!create_vector_members(parser, fld, &vx, &vy, &vz)) {
2475                     ast_delete(fld);
2476                     return false;
2477                 }
2478                 (void)!parser_t_fields_add(parser, vx);
2479                 (void)!parser_t_fields_add(parser, vy);
2480                 (void)!parser_t_fields_add(parser, vz);
2481             }
2482
2483 nextfield:
2484             if (!parser_next(parser)) {
2485                 parseerror(parser, "expected semicolon or another field name");
2486                 return false;
2487             }
2488             if (parser->tok == ';')
2489                 break;
2490             if (parser->tok != ',' || !parser_next(parser)) {
2491                 parseerror(parser, "expected semicolon or another field name");
2492                 return false;
2493             }
2494         }
2495         ast_delete(typevar);
2496
2497         /* skip the semicolon */
2498         if (!parser_next(parser))
2499             return parser->tok == TOKEN_EOF;
2500
2501         return true;
2502     }
2503     else if (parser->tok == '$')
2504     {
2505         if (!parser_next(parser)) {
2506             parseerror(parser, "parse error");
2507             return false;
2508         }
2509     }
2510     else
2511     {
2512         parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
2513         return false;
2514     }
2515     return true;
2516 }
2517
2518 static parser_t *parser;
2519
2520 bool parser_init()
2521 {
2522     parser = (parser_t*)mem_a(sizeof(parser_t));
2523     if (!parser)
2524         return false;
2525
2526     memset(parser, 0, sizeof(*parser));
2527     return true;
2528 }
2529
2530 bool parser_compile(const char *filename)
2531 {
2532     parser->lex = lex_open(filename);
2533     if (!parser->lex) {
2534         printf("failed to open file \"%s\"\n", filename);
2535         return false;
2536     }
2537
2538     /* initial lexer/parser state */
2539     parser->lex->flags.noops = true;
2540
2541     if (parser_next(parser))
2542     {
2543         while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
2544         {
2545             if (!parser_do(parser)) {
2546                 if (parser->tok == TOKEN_EOF)
2547                     parseerror(parser, "unexpected eof");
2548                 else if (!parser->errors)
2549                     parseerror(parser, "parse error\n");
2550                 lex_close(parser->lex);
2551                 parser->lex = NULL;
2552                 return false;
2553             }
2554         }
2555     }
2556
2557     lex_close(parser->lex);
2558     parser->lex = NULL;
2559
2560     return !parser->errors;
2561 }
2562
2563 void parser_cleanup()
2564 {
2565     size_t i;
2566     for (i = 0; i < parser->functions_count; ++i) {
2567         ast_delete(parser->functions[i]);
2568     }
2569     for (i = 0; i < parser->imm_vector_count; ++i) {
2570         ast_delete(parser->imm_vector[i]);
2571     }
2572     for (i = 0; i < parser->imm_string_count; ++i) {
2573         ast_delete(parser->imm_string[i]);
2574     }
2575     for (i = 0; i < parser->imm_float_count; ++i) {
2576         ast_delete(parser->imm_float[i]);
2577     }
2578     for (i = 0; i < parser->fields_count; ++i) {
2579         ast_delete(parser->fields[i].var);
2580         mem_d(parser->fields[i].name);
2581     }
2582     for (i = 0; i < parser->globals_count; ++i) {
2583         ast_delete(parser->globals[i].var);
2584         mem_d(parser->globals[i].name);
2585     }
2586     MEM_VECTOR_CLEAR(parser, functions);
2587     MEM_VECTOR_CLEAR(parser, imm_vector);
2588     MEM_VECTOR_CLEAR(parser, imm_string);
2589     MEM_VECTOR_CLEAR(parser, imm_float);
2590     MEM_VECTOR_CLEAR(parser, globals);
2591     MEM_VECTOR_CLEAR(parser, fields);
2592     MEM_VECTOR_CLEAR(parser, locals);
2593
2594     mem_d(parser);
2595 }
2596
2597 bool parser_finish(const char *output)
2598 {
2599     size_t i;
2600     ir_builder *ir;
2601
2602     if (!parser->errors)
2603     {
2604         ir = ir_builder_new("gmqcc_out");
2605         if (!ir) {
2606             printf("failed to allocate builder\n");
2607             return false;
2608         }
2609
2610         for (i = 0; i < parser->imm_float_count; ++i) {
2611             if (!ast_global_codegen(parser->imm_float[i], ir)) {
2612                 printf("failed to generate global %s\n", parser->imm_float[i]->name);
2613                 ir_builder_delete(ir);
2614                 return false;
2615             }
2616         }
2617         for (i = 0; i < parser->imm_string_count; ++i) {
2618             if (!ast_global_codegen(parser->imm_string[i], ir)) {
2619                 printf("failed to generate global %s\n", parser->imm_string[i]->name);
2620                 ir_builder_delete(ir);
2621                 return false;
2622             }
2623         }
2624         for (i = 0; i < parser->imm_vector_count; ++i) {
2625             if (!ast_global_codegen(parser->imm_vector[i], ir)) {
2626                 printf("failed to generate global %s\n", parser->imm_vector[i]->name);
2627                 ir_builder_delete(ir);
2628                 return false;
2629             }
2630         }
2631         for (i = 0; i < parser->fields_count; ++i) {
2632             ast_value *field;
2633             bool isconst;
2634             if (!ast_istype(parser->fields[i].var, ast_value))
2635                 continue;
2636             field = (ast_value*)parser->fields[i].var;
2637             isconst = field->isconst;
2638             field->isconst = false;
2639             if (!ast_global_codegen((ast_value*)field, ir)) {
2640                 printf("failed to generate field %s\n", field->name);
2641                 ir_builder_delete(ir);
2642                 return false;
2643             }
2644             if (isconst) {
2645                 ir_value *ifld;
2646                 ast_expression *subtype;
2647                 field->isconst = true;
2648                 subtype = field->expression.next;
2649                 ifld = ir_builder_create_field(ir, field->name, subtype->expression.vtype);
2650                 if (subtype->expression.vtype == TYPE_FIELD)
2651                     ifld->fieldtype = subtype->expression.next->expression.vtype;
2652                 else if (subtype->expression.vtype == TYPE_FUNCTION)
2653                     ifld->outtype = subtype->expression.next->expression.vtype;
2654                 (void)!ir_value_set_field(field->ir_v, ifld);
2655             }
2656         }
2657         for (i = 0; i < parser->globals_count; ++i) {
2658             if (!ast_istype(parser->globals[i].var, ast_value))
2659                 continue;
2660             if (!ast_global_codegen((ast_value*)(parser->globals[i].var), ir)) {
2661                 printf("failed to generate global %s\n", parser->globals[i].name);
2662                 ir_builder_delete(ir);
2663                 return false;
2664             }
2665         }
2666         for (i = 0; i < parser->functions_count; ++i) {
2667             if (!ast_function_codegen(parser->functions[i], ir)) {
2668                 printf("failed to generate function %s\n", parser->functions[i]->name);
2669                 ir_builder_delete(ir);
2670                 return false;
2671             }
2672             if (!ir_function_finalize(parser->functions[i]->ir_func)) {
2673                 printf("failed to finalize function %s\n", parser->functions[i]->name);
2674                 ir_builder_delete(ir);
2675                 return false;
2676             }
2677         }
2678
2679         if (opts_dump)
2680             ir_builder_dump(ir, printf);
2681
2682         if (!ir_builder_generate(ir, output)) {
2683             printf("*** failed to generate output file\n");
2684             ir_builder_delete(ir);
2685             return false;
2686         }
2687
2688         ir_builder_delete(ir);
2689         return true;
2690     }
2691
2692     printf("*** there were compile errors\n");
2693     return false;
2694 }