126 lines
3.2 KiB
Text
126 lines
3.2 KiB
Text
%{
|
|
#include <cstdio>
|
|
#include <iostream>
|
|
#include <vector>
|
|
#include <fstream>
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
using namespace std;
|
|
extern "C" int yylex();
|
|
extern "C" int yyparse();
|
|
extern "C" FILE *yyin;
|
|
void yyerror(const char *s);
|
|
vector<string> nameTable;
|
|
int getVarID(string name) {
|
|
auto it=nameTable.end();
|
|
for(auto it=nameTable.begin(); it!=nameTable.end(); it++) {
|
|
if(nameTable[it-nameTable.begin()]==name)
|
|
return it-nameTable.begin();
|
|
}
|
|
nameTable.push_back(name);
|
|
return nameTable.size()-1;
|
|
}
|
|
vector<string> argTable;
|
|
int getArgID(string name) {
|
|
auto it=argTable.end();
|
|
for(auto it=argTable.begin(); it!=argTable.end(); it++) {
|
|
if(argTable[it-argTable.begin()]==name) {
|
|
return (int)(it-argTable.begin())-(int)(argTable.size())-1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
void createArg(string arg) {
|
|
argTable.push_back(arg);
|
|
}
|
|
ofstream out;
|
|
%}
|
|
%union {
|
|
int ival;
|
|
char* sval;
|
|
}
|
|
%token <ival> INT
|
|
%token PLUS
|
|
%token MINUS
|
|
%token MUL
|
|
%token LPARENS
|
|
%token RPARENS
|
|
%token DIV
|
|
%token <sval> VARNAME
|
|
%token EQ
|
|
%token COMMA
|
|
%token FUNCTION
|
|
%token END
|
|
%left PLUS MINUS MUL DIV FUNCTION
|
|
%right EQ
|
|
%%
|
|
program:
|
|
fh function ef program
|
|
| fh function ef
|
|
fh:
|
|
FUNCTION VARNAME LPARENS args RPARENS {out << $2 << ":\n\tbpget\n\tspget\n\tbpset" << endl;}
|
|
| FUNCTION VARNAME LPARENS RPARENS {out << $2 << ":\n\tbpget\n\tspget\n\tbpset" << endl;}
|
|
args:
|
|
VARNAME COMMA args {createArg($1);}
|
|
| VARNAME {createArg($1);}
|
|
ef:
|
|
END {out << "\tbpget\n\tspset\n\tbpset\n\tret"<<endl;argTable.clear();}
|
|
function:
|
|
VARNAME EQ exp function {out << "\t[ci:2] store "<<getVarID($1)*4 << endl;}
|
|
| VARNAME EQ exp {out << "\t[ci:2] store "<<getVarID($1)*4 << endl;}
|
|
| exp function {out << "\tdrop" << endl;}
|
|
| exp {out << "\tdrop" << endl;}
|
|
exp:
|
|
factor
|
|
| exp PLUS exp { out << "\tadd" <<endl;}
|
|
| exp MINUS exp { out << "\tsub" <<endl;}
|
|
factor:
|
|
num
|
|
| factor MUL factor {out << "\tmul" << endl;}
|
|
| factor DIV factor {out << "\tdiv" << endl;}
|
|
num:
|
|
INT { out << "\tpush "<< $1 << endl;}
|
|
| VARNAME LPARENS fargs RPARENS {out << "\tcpget\n\tjmp @" << $1 << "\n\t[ci:2] load " << getVarID($1) << endl;}
|
|
| VARNAME LPARENS RPARENS {out << "\tcpget\n\tjmp @" << $1 << "\n\t[ci:2] load " << getVarID($1) << endl;}
|
|
| VARNAME { char* arg=$1; int id; if((id=getArgID(arg))==0) {out << "\t[ci:2]load "<<getVarID(arg)*4<<endl;} else {out<< "\tget "<< id << endl;}}
|
|
| LPARENS exp RPARENS
|
|
fargs:
|
|
exp COMMA fargs
|
|
| exp
|
|
%%
|
|
int main(int argc, char** argv) {
|
|
int pid;
|
|
mkfifo("tmp.a", 0660);
|
|
if(!(pid=fork())) {
|
|
system("as -o tmp.o crt0.asm tmp.a libmtg.asm");
|
|
return 0;
|
|
}
|
|
out=ofstream("tmp.a",ios::out);
|
|
unlink("tmp.a"); //Unlink when done
|
|
if(argc<=3)
|
|
yyerror("usage: mumei -o outfile infile1 [infile2 [infile3 ...]]");
|
|
if("-o"s!=argv[1])
|
|
yyerror("usage: mumei -o outfile infile1 [infile2 [infile3 ...]]");
|
|
for(int i=3;i<argc;i++) {
|
|
FILE *f=fopen(argv[i],"rb");
|
|
yyin = f;
|
|
do {
|
|
yyparse();
|
|
} while (!feof(yyin));
|
|
fclose(f);
|
|
}
|
|
out.close();
|
|
wait(NULL);
|
|
system(("explink -o "s+argv[2]+" -c tmp.o -C 0").c_str());
|
|
unlink("tmp.o");
|
|
}
|
|
|
|
void yyerror(const char *s) {
|
|
cout << "parse error! Message: " << s << endl;
|
|
out.close();
|
|
fclose(yyin);
|
|
exit(-1);
|
|
}
|