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