forked from 0ad/0ad
124 lines
2.8 KiB
C++
124 lines
2.8 KiB
C++
#include "stdafx.h"
|
|
#include "rmgen.h"
|
|
#include "map.h"
|
|
#include "output.h"
|
|
|
|
using namespace std;
|
|
|
|
JSRuntime *rt = 0;
|
|
JSContext *cx = 0;
|
|
JSObject *global = 0;
|
|
|
|
Map* theMap = 0;
|
|
|
|
JSFunctionSpec globalFunctions[] = {
|
|
/* name native args */
|
|
{"print", print, 1},
|
|
{"init", init, 3},
|
|
{0}
|
|
};
|
|
|
|
void ErrorReporter(JSContext *cx, const char *message, JSErrorReport *report) {
|
|
cerr << "Error at " << report->filename << ":" << report->lineno << ":\n\t"
|
|
<< message << endl;
|
|
Shutdown(1);
|
|
}
|
|
|
|
void InitJS() {
|
|
rt = JS_NewRuntime(8L * 1024L * 1024L);
|
|
cx = JS_NewContext(rt, 8192);
|
|
|
|
JS_SetErrorReporter(cx, ErrorReporter);
|
|
|
|
static JSClass globalClass = {
|
|
"global",0,
|
|
JS_PropertyStub,JS_PropertyStub,JS_PropertyStub,JS_PropertyStub,
|
|
JS_EnumerateStub,JS_ResolveStub,JS_ConvertStub,JS_FinalizeStub
|
|
};
|
|
global = JS_NewObject(cx, &globalClass, NULL, NULL);
|
|
JS_InitStandardClasses(cx, global);
|
|
|
|
JS_DefineFunctions(cx, global, globalFunctions);
|
|
}
|
|
|
|
void Shutdown(int status) {
|
|
JS_DestroyContext(cx);
|
|
JS_DestroyRuntime(rt);
|
|
JS_ShutDown();
|
|
system("pause");
|
|
exit(status);
|
|
}
|
|
|
|
char* ValToString(jsval val) {
|
|
return JS_GetStringBytes(JS_ValueToString(cx, val));
|
|
}
|
|
|
|
JSBool print(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)
|
|
{
|
|
if(argc != 1) {
|
|
JS_ReportError(cx, "print: expected 1 argument but got %d", argc);
|
|
return JS_FALSE;
|
|
}
|
|
|
|
cout << JS_GetStringBytes(JS_ValueToString(cx, argv[0]));
|
|
return JS_TRUE;
|
|
}
|
|
|
|
JSBool init(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)
|
|
{
|
|
if(argc != 2) {
|
|
JS_ReportError(cx, "init: expected 2 arguments but got %d", argc);
|
|
}
|
|
if(!JSVAL_IS_INT(argv[0])) {
|
|
JS_ReportError(cx, "init: first argument must be an integer");
|
|
}
|
|
if(!JSVAL_IS_STRING(argv[1])) {
|
|
JS_ReportError(cx, "init: second argument must be a string");
|
|
}
|
|
if(theMap != 0) {
|
|
JS_ReportError(cx, "init: cannot be called twice");
|
|
}
|
|
|
|
int size = JSVAL_TO_INT(argv[0]);
|
|
char* baseTerrain = JS_GetStringBytes(JSVAL_TO_STRING(argv[1]));
|
|
|
|
theMap = new Map(size, baseTerrain);
|
|
return JS_TRUE;
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
InitJS();
|
|
|
|
if(argc!=3) {
|
|
cerr << "Usage: rmgen <script> <output name without extension>" << endl;
|
|
Shutdown(1);
|
|
}
|
|
|
|
FILE* scriptFile = fopen(argv[1], "r");
|
|
if(!scriptFile) {
|
|
cerr << "Cannot open " << scriptFile << endl;
|
|
Shutdown(1);
|
|
}
|
|
string script;
|
|
char buf[1025];
|
|
while(fgets(buf, 1024, scriptFile)) {
|
|
script += buf;
|
|
}
|
|
|
|
jsval rval;
|
|
JS_EvaluateScript(cx, global, script.c_str(), script.length(),
|
|
argv[1], 1, &rval);
|
|
|
|
if(!theMap) {
|
|
cerr << "Error:\n\tScript never called init!" << endl;
|
|
Shutdown(1);
|
|
}
|
|
|
|
string outputName = argv[2];
|
|
OutputMap(theMap, outputName);
|
|
|
|
Shutdown(0);
|
|
}
|
|
|