1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
%{
#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include "types.h"
#include "qbe.h"
extern int yylex();
extern int yylineno;
extern char *yytext;
void yyerror(const char* s);
exp *exprs[1024];
int exprcount = 0;
function *funcs[1024];
int funccount = 0;
exp *newexpr(char name, char op, int val, bool isvar) {
exp *e = malloc(sizeof(exp));
e->name = name;
e->op = op;
e->var = isvar;
e->v = val;
}
void appendexpr(exp *e) {
exprs[exprcount] = e;
exprcount++;
}
void appendfunction(function *f) {
funcs[funccount] = f;
funccount++;
}
function *copyexprsandclear(char name) {
function *f = malloc(sizeof(function));
f->name = name;
memcpy(f->exprs, exprs, sizeof(exp) * 1024);
f->exprcount = exprcount;
exprcount = 0;
}
%}
%union {
char funcname;
char varname;
int intlit;
exp *expr;
}
%token <intlit> INTLIT
%token <funcname> FUNCTIONNAME
%token <varname> VARNAME
%token FUNCTIONDEC
%token PRINT
%token INPUT
%token RET
%token JMP
%token JNZ
%type <expr> EXPR
%%
PROG : PROG F
| F
;
F : FUNCTIONDEC FUNCTIONNAME '{' EXPRS '}' {appendfunction(copyexprsandclear($2));}
;
EXPRS : EXPRS EXPR {appendexpr($2);}
| EXPR {appendexpr($1);}
;
EXPR : VARNAME {$$ = newexpr($1, 0, 0, false);}
| VARNAME '=' INTLIT {$$ = newexpr($1, '=', $3, false);}
| VARNAME '+' INTLIT {$$ = newexpr($1, '+', $3, false);}
| VARNAME '-' INTLIT {$$ = newexpr($1, '-', $3, false);}
| VARNAME '*' INTLIT {$$ = newexpr($1, '*', $3, false);}
| VARNAME '/' INTLIT {$$ = newexpr($1, '/', $3, false);}
| VARNAME '=' VARNAME {$$ = newexpr($1, '=', $3, true);}
| VARNAME '+' VARNAME {$$ = newexpr($1, '+', $3, true);}
| VARNAME '-' VARNAME {$$ = newexpr($1, '-', $3, true);}
| VARNAME '*' VARNAME {$$ = newexpr($1, '*', $3, true);}
| VARNAME '/' VARNAME {$$ = newexpr($1, '/', $3, true);}
| PRINT VARNAME {$$ = newexpr($2, 'p', 0, false);}
| VARNAME '=' INPUT {$$ = newexpr($1, 'i', 0, false);}
| VARNAME '=' FUNCTIONNAME {$$ = newexpr($1, 'c', $3, false);}
| RET VARNAME {$$ = newexpr($2, 'r', 0, false);}
| VARNAME ':' {$$ = newexpr($1, 'l', 0, false);}
| JMP VARNAME ':' {$$ = newexpr($2, 'j', 0, false);}
| JNZ VARNAME VARNAME ':' {$$ = newexpr($3, 'z', $2, false);}
;
%%
void yyerror(const char *msg) {
fprintf(stderr, "%d:%s\n%s\n", yylineno, msg, yytext);
}
int main() {
yyparse();
compile(stdout, funcs, funccount);
return 0;
}
|
