SQInteger sqstd_register_stringlib( HSQUIRRELVM v ) {
sq_pushstring( v, _SC( "regexp" ), -1 );
sq_newclass( v, SQFalse );
SQInteger i = 0;
while ( rexobj_funcs[i].name != 0 ) {
SQRegFunction &f = rexobj_funcs[i];
sq_pushstring( v, f.name, -1 );
sq_newclosure( v, f.f, 0 );
sq_setparamscheck( v, f.nparamscheck, f.typemask );
sq_setnativeclosurename( v, -1, f.name );
sq_createslot( v, -3 );
i++;
}
sq_createslot( v, -3 );
i = 0;
while ( stringlib_funcs[i].name != 0 ) {
sq_pushstring( v, stringlib_funcs[i].name, -1 );
sq_newclosure( v, stringlib_funcs[i].f, 0 );
sq_setparamscheck( v, stringlib_funcs[i].nparamscheck, stringlib_funcs[i].typemask );
sq_setnativeclosurename( v, -1, stringlib_funcs[i].name );
sq_createslot( v, -3 );
i++;
}
return 1;
}
SQInteger sqstd_register_stringlib(HSQUIRRELVM v)
{
sq_pushstring(v,_SC("regexp"),-1);
sq_newclass(v,SQFalse,"regexp");
SQInteger i = 0;
while(rexobj_funcs[i].name != 0) {
SQRegFunction &f = rexobj_funcs[i];
sq_pushstring(v,f.name,-1);
sq_newclosure(v,f.f,0);
sq_setparamscheck(v,f.nparamscheck,f.typemask);
sq_setnativeclosureinfo(v,-1,f.name,f.file);
if (f.help)
{
sq_pushstring(v,f.help,-1);
sq_sethelp(v, -2);
}
sq_createslot(v,-3);
i++;
}
sq_createslot(v,-3);
i = 0;
while(stringlib_funcs[i].name!=0)
{
sq_pushstring(v,stringlib_funcs[i].name,-1);
sq_newclosure(v,stringlib_funcs[i].f,0);
sq_setparamscheck(v,stringlib_funcs[i].nparamscheck,stringlib_funcs[i].typemask);
sq_setnativeclosureinfo(v,-1,stringlib_funcs[i].name,stringlib_funcs[i].file);
if (stringlib_funcs[i].help)
{
sq_pushstring(v,stringlib_funcs[i].help,-1);
sq_sethelp(v, -2);
}
sq_createslot(v,-3);
i++;
}
sq_getdefaultdelegate(v, OT_STRING);
i = 0;
while(stringextdeleg_funcs[i].name!=0)
{
sq_pushstring(v,stringextdeleg_funcs[i].name,-1);
sq_newclosure(v,stringextdeleg_funcs[i].f,0);
sq_setparamscheck(v,stringextdeleg_funcs[i].nparamscheck,stringextdeleg_funcs[i].typemask);
sq_setnativeclosureinfo(v,-1,stringextdeleg_funcs[i].name,stringextdeleg_funcs[i].file);
if (stringextdeleg_funcs[i].help)
{
sq_pushstring(v,stringextdeleg_funcs[i].help,-1);
sq_sethelp(v,-2);
}
sq_createslot(v,-3);
i++;
}
sq_pop(v, 1);
return 1;
}
bool SQDbgServer::Init()
{
//creates an environment table for the debugger
sq_newtable(_v);
sq_getstackobj(_v,-1,&_debugroot);
sq_addref(_v,&_debugroot);
//creates a emptyslot to store the watches
sq_pushstring(_v,_SC("watches"),-1);
sq_pushnull(_v);
sq_createslot(_v,-3);
sq_pushstring(_v,_SC("beginelement"),-1);
sq_pushuserpointer(_v,this);
sq_newclosure(_v,beginelement,1);
sq_setparamscheck(_v,2,_SC(".s"));
sq_createslot(_v,-3);
sq_pushstring(_v,_SC("endelement"),-1);
sq_pushuserpointer(_v,this);
sq_newclosure(_v,endelement,1);
sq_setparamscheck(_v,2,_SC(".s"));
sq_createslot(_v,-3);
sq_pushstring(_v,_SC("attribute"),-1);
sq_pushuserpointer(_v,this);
sq_newclosure(_v,attribute,1);
sq_setparamscheck(_v,3,_SC(".ss"));
sq_createslot(_v,-3);
sq_pop(_v,1);
//stores debug hook and error handler in the registry
sq_pushregistrytable(_v);
sq_pushstring(_v,SQDBG_DEBUG_HOOK,-1);
sq_pushuserpointer(_v,this);
sq_newclosure(_v,debug_hook,1);
sq_createslot(_v,-3);
sq_pushstring(_v,SQDBG_ERROR_HANDLER,-1);
sq_pushuserpointer(_v,this);
sq_newclosure(_v,error_handler,1);
sq_createslot(_v,-3);
sq_pop(_v,1);
//sets the error handlers
SetErrorHandlers();
return true;
}
BOOL SbuCreateStaticNamespace(HSQUIRRELVM v,ScriptNamespaceDecl *sn)
{
int n = 0;
sq_pushroottable(v);
sq_pushstring(v,sn->name,-1);
sq_newtable(v);
const ScriptClassMemberDecl *members = sn->members;
const ScriptClassMemberDecl *m = NULL;
while(members[n].name) {
m = &members[n];
sq_pushstring(v,m->name,-1);
sq_newclosure(v,m->func,0);
sq_setparamscheck(v,m->params,m->typemask);
sq_setnativeclosurename(v,-1,m->name);
sq_createslot(v,-3);
n++;
}
const ScriptConstantDecl *consts = sn->constants;
const ScriptConstantDecl *c = NULL;
n = 0;
while(consts[n].name) {
c = &consts[n];
sq_pushstring(v,c->name,-1);
switch(c->type) {
case OT_STRING: sq_pushstring(v,c->val.s,-1);break;
case OT_INTEGER: sq_pushinteger(v,c->val.i);break;
case OT_FLOAT: sq_pushfloat(v,c->val.f);break;
}
sq_createslot(v,-3);
n++;
}
if(sn->delegate) {
const ScriptClassMemberDecl *members = sn->delegate;
const ScriptClassMemberDecl *m = NULL;
sq_newtable(v);
while(members[n].name) {
m = &members[n];
sq_pushstring(v,m->name,-1);
sq_newclosure(v,m->func,0);
sq_setparamscheck(v,m->params,m->typemask);
sq_setnativeclosurename(v,-1,m->name);
sq_createslot(v,-3);
n++;
}
sq_setdelegate(v,-2);
}
sq_createslot(v,-3);
sq_pop(v,1);
return TRUE;
}
SQRESULT declare_stream(HSQUIRRELVM v,const SQChar* name,SQUserPointer typetag,const SQChar* reg_name,const SQRegFunction *methods,const SQRegFunction *globals)
{
if(sq_gettype(v,-1) != OT_TABLE)
return sq_throwerror(v,_SC("table expected"));
SQInteger top = sq_gettop(v);
//create delegate
init_streamclass(v);
sq_pushregistrytable(v);
sq_pushstring(v,reg_name,-1);
sq_pushstring(v,_SC("std_stream"),-1);
if(SQ_SUCCEEDED(sq_get(v,-3))) {
sq_newclass(v,SQTrue);
sq_settypetag(v,-1,typetag);
SQInteger i = 0;
while(methods[i].name != 0) {
const SQRegFunction &f = methods[i];
sq_pushstring(v,f.name,-1);
sq_newclosure(v,f.f,0);
sq_setparamscheck(v,f.nparamscheck,f.typemask);
sq_setnativeclosurename(v,-1,f.name);
sq_newslot(v,-3,SQFalse);
i++;
}
sq_newslot(v,-3,SQFalse);
sq_pop(v,1);
i = 0;
while(globals[i].name!=0)
{
const SQRegFunction &f = globals[i];
sq_pushstring(v,f.name,-1);
sq_newclosure(v,f.f,0);
sq_setparamscheck(v,f.nparamscheck,f.typemask);
sq_setnativeclosurename(v,-1,f.name);
sq_newslot(v,-3,SQFalse);
i++;
}
//register the class in the target table
sq_pushstring(v,name,-1);
sq_pushregistrytable(v);
sq_pushstring(v,reg_name,-1);
sq_get(v,-2);
sq_remove(v,-2);
sq_newslot(v,-3,SQFalse);
sq_settop(v,top);
return SQ_OK;
}
sq_settop(v,top);
return SQ_ERROR;
}
void CSquirrel::RegisterFunction(const char * szFunctionName, SQFUNCTION pfnFunction, int iParameterCount, const char * szFunctionTemplate, bool bPushRootTable)
{
// Push the root table onto the stack if needed
if(bPushRootTable)
sq_pushroottable(m_pVM);
// Push the function name onto the stack
sq_pushstring(m_pVM, szFunctionName, -1);
// Create a new function
sq_newclosure(m_pVM, pfnFunction, 0);
// Set the function parameter template and count
if(iParameterCount != -1)
{
char szTemp[128];
if(szFunctionTemplate)
sprintf(szTemp, ".%s", szFunctionTemplate);
sq_setparamscheck(m_pVM, (iParameterCount + 1), (szFunctionTemplate) ? szTemp : NULL);
}
// Create a new slot
sq_createslot(m_pVM, -3);
// Pop the root table from the stack if needed
if(bPushRootTable)
sq_pop(m_pVM, 1);
}
void sq_base_register(HSQUIRRELVM v)
{
SQInteger i=0;
sq_pushroottable(v);
while(base_funcs[i].name!=0) {
sq_pushstring(v,base_funcs[i].name,-1);
sq_newclosure(v,base_funcs[i].f,0);
sq_setnativeclosurename(v,-1,base_funcs[i].name);
sq_setparamscheck(v,base_funcs[i].nparamscheck,base_funcs[i].typemask);
sq_createslot(v,-3);
i++;
}
sq_pushstring(v,_SC("_version_"),-1);
sq_pushstring(v,SQUIRREL_VERSION,-1);
sq_createslot(v,-3);
sq_pushstring(v,_SC("_charsize_"),-1);
sq_pushinteger(v,sizeof(SQChar));
sq_createslot(v,-3);
sq_pushstring(v,_SC("_intsize_"),-1);
sq_pushinteger(v,sizeof(SQInteger));
sq_createslot(v,-3);
sq_pushstring(v,_SC("_floatsize_"),-1);
sq_pushinteger(v,sizeof(SQFloat));
sq_createslot(v,-3);
sq_pop(v,1);
}
void Sq_RegisterFunc(HSQUIRRELVM v,SQFUNCTION f,const char *fname, int ParamNum, const char *params) {
sq_pushstring(v,fname,-1);
sq_newclosure(v,f,0);
sq_newslot(v,-3,SQFalse);
if(params)
sq_setparamscheck(v, ParamNum, _SC(params));
}
BOOL SbuCreateClass(HSQUIRRELVM v,SquirrelClassDecl *cd)
{
int n = 0;
int oldtop = sq_gettop(v);
sq_pushroottable(v);
sq_pushstring(v,cd->name,-1);
if(cd->base) {
sq_pushstring(v,cd->base,-1);
if(SQ_FAILED(sq_get(v,-3))) {
sq_settop(v,oldtop);
return FALSE;
}
}
if(SQ_FAILED(sq_newclass(v,cd->base?1:0))) {
sq_settop(v,oldtop);
return FALSE;
}
sq_settypetag(v,-1,(SQUserPointer)cd);
const ScriptClassMemberDecl *members = cd->members;
const ScriptClassMemberDecl *m = NULL;
while(members[n].name) {
m = &members[n];
sq_pushstring(v,m->name,-1);
sq_newclosure(v,m->func,0);
sq_setparamscheck(v,m->params,m->typemask);
sq_setnativeclosurename(v,-1,m->name);
sq_createslot(v,-3);
n++;
}
sq_createslot(v,-3);
sq_pop(v,1);
return TRUE;
}
void init_streamclass(HSQUIRRELVM v)
{
sq_pushregistrytable(v);
sq_pushstring(v,_SC("std_stream"),-1);
if(SQ_FAILED(sq_get(v,-2))) {
sq_pushstring(v,_SC("std_stream"),-1);
sq_newclass(v,SQFalse);
sq_settypetag(v,-1,(SQUserPointer)SQSTD_STREAM_TYPE_TAG);
SQInteger i = 0;
while(_stream_methods[i].name != 0) {
const SQRegFunction &f = _stream_methods[i];
sq_pushstring(v,f.name,-1);
sq_newclosure(v,f.f,0);
sq_setparamscheck(v,f.nparamscheck,f.typemask);
sq_newslot(v,-3,SQFalse);
i++;
}
sq_newslot(v,-3,SQFalse);
sq_pushroottable(v);
sq_pushstring(v,_SC("stream"),-1);
sq_pushstring(v,_SC("std_stream"),-1);
sq_get(v,-4);
sq_newslot(v,-3,SQFalse);
sq_pop(v,1);
}
else {
sq_pop(v,1); //result
}
sq_pop(v,1);
}
void RegisterFunction(HSQUIRRELVM pVM, char * szFunc, SQFUNCTION func, int params, const char * szTemplate)
{
sq_pushroottable(pVM);
sq_pushstring(pVM, szFunc, -1);
sq_newclosure(pVM, func, 0);
if(params != -1)
{
sq_setparamscheck(pVM, params, szTemplate);
}
sq_createslot(pVM, -3);
}
SQInteger regGlobals(HSQUIRRELVM v) {
SQInteger i=0;
SQRegFunction* pfpps = global_funcs;
while(pfpps[i].name!=0) {
sq_pushstring(v,pfpps[i].name,-1);
sq_newclosure(v,pfpps[i].f,0);
sq_setparamscheck(v,pfpps[i].nparamscheck,pfpps[i].typemask);
sq_setnativeclosurename(v,-1,pfpps[i].name);
sq_newslot(v,-3,SQFalse);
i++;
}
return 1;
}
void Squirrel::AddMethod(const char *method_name, SQFUNCTION proc, uint nparam, const char *params, void *userdata, int size)
{
sq_pushstring(this->vm, OTTD2SQ(method_name), -1);
if (size != 0) {
void *ptr = sq_newuserdata(vm, size);
memcpy(ptr, userdata, size);
}
sq_newclosure(this->vm, proc, size != 0 ? 1 : 0);
if (nparam != 0) sq_setparamscheck(this->vm, nparam, OTTD2SQ(params));
sq_setnativeclosurename(this->vm, -1, OTTD2SQ(method_name));
sq_newslot(this->vm, -3, SQFalse);
}
SQInteger sqstd_register_systemlib(HSQUIRRELVM v)
{
SQInteger i=0;
while(systemlib_funcs[i].name!=0)
{
sq_pushstring(v,systemlib_funcs[i].name,-1);
sq_newclosure(v,systemlib_funcs[i].f,0);
sq_setparamscheck(v,systemlib_funcs[i].nparamscheck,systemlib_funcs[i].typemask);
sq_setnativeclosurename(v,-1,systemlib_funcs[i].name);
sq_newslot(v,-3,SQFalse);
i++;
}
return 1;
}
void sqstd_seterrorhandlers(HSQUIRRELVM v)
{
sq_setcompilererrorhandler(v,_sqstd_compiler_error);
sq_newclosure(v,_sqstd_aux_printerror,0);
sq_seterrorhandler(v);
// additional functions
sq_pushroottable(v);
sq_pushstring(v, _SC("printCallStack"), -1);
sq_newclosure(v, printCallStack, 0);
sq_createslot(v, -3);
sq_pushstring(v, _SC("notifyAllExceptions"), -1);
sq_newclosure(v, notifyAllExceptions, 0);
sq_setparamscheck(v, 2, _SC(".n|b"));
sq_createslot(v, -3);
sq_pop(v,1);
}
SQRESULT sqstd_register_mathlib(HSQUIRRELVM v)
{
SQInteger i=0;
while(mathlib_funcs[i].name!=0) {
sq_pushstring(v,mathlib_funcs[i].name,-1);
sq_newclosure(v,mathlib_funcs[i].f,0);
sq_setparamscheck(v,mathlib_funcs[i].nparamscheck,mathlib_funcs[i].typemask);
sq_setnativeclosurename(v,-1,mathlib_funcs[i].name);
sq_createslot(v,-3);
i++;
}
sq_pushstring(v,_SC("RAND_MAX"),-1);
sq_pushinteger(v,RAND_MAX);
sq_createslot(v,-3);
sq_pushstring(v,_SC("PI"),-1);
sq_pushfloat(v,(SQFloat)M_PI);
sq_createslot(v,-3);
return SQ_OK;
}
void CResourceScript::RegisterFunction ( CString strFunction, SQFUNCTION sqFunction, int iParams, CString strTemplate )
{
// Push function name.
sq_pushstring ( m_pVM, strFunction.Get(), -1 );
// Push function C++ handle,
sq_newclosure ( m_pVM, sqFunction, 0 );
// Make params / check.
if ( iParams != -1 )
{
CString strParams;
strParams.Format( ".%s", strTemplate.Get() );
sq_setparamscheck ( m_pVM, (iParams+1), strParams.Get() );
}
// Create slot for function,
sq_createslot ( m_pVM, -3 );
}
SQRESULT sqstd_register_mathlib(HSQUIRRELVM v)
{
SQInteger i=0;
while(mathlib_funcs[i].name!=0) {
sq_pushstring(v,mathlib_funcs[i].name,-1);
sq_newclosure(v,mathlib_funcs[i].f,0);
sq_setparamscheck(v,mathlib_funcs[i].nparamscheck,mathlib_funcs[i].typemask);
sq_setnativeclosurename(v,-1,mathlib_funcs[i].name);
sq_createslot(v,-3);
i++;
}
#ifdef EXPORT_DEFAULT_SQUIRREL_FUNCTIONS
sq_pushstring(v,"RAND_MAX",-1);
sq_pushinteger(v,RAND_MAX);
sq_createslot(v,-3);
#endif /* EXPORT_DEFAULT_SQUIRREL_FUNCTIONS */
sq_pushstring(v,"PI",-1);
sq_pushfloat(v,(SQFloat)M_PI);
sq_createslot(v,-3);
return SQ_OK;
}
void CSquirrel::RegisterFunction(String strFunctionName, SQFUNCTION pfnFunction, int iParameterCount, String strFunctionTemplate)
{
// Push the function name onto the stack
sq_pushstring(m_pVM, strFunctionName.Get(), -1);
// Create a new function
sq_newclosure(m_pVM, pfnFunction, 0);
// Set the function parameter template and count
if(iParameterCount != -1)
{
String strTypeMask;
if(strFunctionTemplate.IsNotEmpty())
strTypeMask.Format(".%s", strFunctionTemplate.Get());
sq_setparamscheck(m_pVM, (iParameterCount + 1), strTypeMask.Get());
}
// Create a new slot
sq_createslot(m_pVM, -3);
}
void CSquirrelVM::RegisterFunction(const char* szFunctionName, scriptFunction pfnFunction, int iParameterCount, const char* szFunctionTemplate, bool bPushRootTable)
{
// Push the function name onto the stack
sq_pushstring(m_pVM, szFunctionName, -1);
// Create a new function
sq_newclosure(m_pVM, (SQFUNCTION)pfnFunction, 0);
// Set the function parameter template and count
if(iParameterCount != -1)
{
CString strTypeMask;
// TODO: Change parameter
CString strFunctionTemplate = szFunctionTemplate;
if(strFunctionTemplate.IsNotEmpty())
strTypeMask.Format(".%s", strFunctionTemplate.Get());
sq_setparamscheck(m_pVM, (iParameterCount + 1), strTypeMask.Get());
}
sq_setnativeclosurename(m_pVM, -1, szFunctionName);
// Create a new slot
sq_createslot(m_pVM, -3);
}
void Interactive(HSQUIRRELVM v)
{
#define MAXINPUT 1024
SQChar buffer[MAXINPUT];
SQInteger blocks =0;
SQInteger string=0;
SQInteger retval=0;
SQInteger done=0;
PrintVersionInfos();
sq_pushroottable(v);
sq_pushstring(v,_SC("quit"),-1);
sq_pushuserpointer(v,&done);
sq_newclosure(v,quit,1);
sq_setparamscheck(v,1,NULL);
sq_newslot(v,-3,SQFalse);
sq_pop(v,1);
while (!done)
{
SQInteger i = 0;
scprintf(_SC("\nsq>"));
for(;;) {
int c;
if(done)return;
c = getchar();
if (c == _SC('\n')) {
if (i>0 && buffer[i-1] == _SC('\\'))
{
buffer[i-1] = _SC('\n');
}
else if(blocks==0)break;
buffer[i++] = _SC('\n');
}
else if (c==_SC('}')) {blocks--; buffer[i++] = (SQChar)c;}
else if(c==_SC('{') && !string){
blocks++;
buffer[i++] = (SQChar)c;
}
else if(c==_SC('"') || c==_SC('\'')){
string=!string;
buffer[i++] = (SQChar)c;
}
else if (i >= MAXINPUT-1) {
scfprintf(stderr, _SC("sq : input line too long\n"));
break;
}
else{
buffer[i++] = (SQChar)c;
}
}
buffer[i] = _SC('\0');
if(buffer[0]==_SC('=')){
scsprintf(sq_getscratchpad(v,MAXINPUT),_SC("return (%s)"),&buffer[1]);
memcpy(buffer,sq_getscratchpad(v,-1),(scstrlen(sq_getscratchpad(v,-1))+1)*sizeof(SQChar));
retval=1;
}
i=scstrlen(buffer);
if(i>0){
SQInteger oldtop=sq_gettop(v);
if(SQ_SUCCEEDED(sq_compilebuffer(v,buffer,i,_SC("interactive console"),SQTrue))){
sq_pushroottable(v);
if(SQ_SUCCEEDED(sq_call(v,1,retval,SQTrue)) && retval){
scprintf(_SC("\n"));
sq_pushroottable(v);
sq_pushstring(v,_SC("print"),-1);
sq_get(v,-2);
sq_pushroottable(v);
sq_push(v,-4);
sq_call(v,2,SQFalse,SQTrue);
retval=0;
scprintf(_SC("\n"));
}
}
sq_settop(v,oldtop);
}
}
}
/**
* クラスの登録
* @param HSQUIRRELVM v
*/
SQRESULT
TJSObject::createTJSClass(HSQUIRRELVM v)
{
SQInteger top = sq_gettop(v);
if (top < 2) {
return sq_throwerror(v, _SC("invalid param"));
}
// クラスを生成
sq_pushobject(v, SQClassType<TJSObject>::ClassObject());
sq_newclass(v, true); // 継承する
// メンバ登録
const tjs_char *tjsClassName = NULL;
tTJSVariant tjsClassObj;
for (SQInteger i=top; i>1; i--) {
if ((tjsClassName = sqobject::getString(v,i))) {
TVPExecuteExpression(tjsClassName, &tjsClassObj);
if (tjsClassObj.Type() == tvtObject &&
TJS_SUCCEEDED(tjsClassObj.AsObjectClosureNoAddRef().IsInstanceOf(0,NULL,NULL,L"Class",NULL))) {
MemberRegister *r = new MemberRegister(v, tjsClassObj);
tTJSVariantClosure closure(r);
tjsClassObj.AsObjectClosureNoAddRef().EnumMembers(TJS_IGNOREPROP, &closure, NULL);
r->Release();
}
}
}
if (tjsClassName) {
// コンストラクタ登録
sq_pushstring(v, _SC("constructor"), -1);
sq_pushvariant(v, tjsClassObj);
sq_newclosure(v, tjsConstructor, 1);
sq_createslot(v, -3);
// クラス属性に tjsクラスを登録
sq_pushnull(v);
sq_newtable(v);
if (SQ_SUCCEEDED(sq_setattributes(v,-3))) {
sq_pop(v,1);
sq_pushnull(v);
if (SQ_SUCCEEDED(sq_getattributes(v, -2))) {
sq_pushstring(v, tjsClassAttrName, -1);
sq_pushvariant(v, tjsClassObj);
if (SQ_SUCCEEDED(sq_createslot(v, -3))) {
sq_pop(v,1);
} else {
sq_pop(v,2);
}
} else {
// XXX
sq_pop(v,1);
}
} else {
// XXX
sq_pop(v,2);
}
// TJS機能メソッドを登録
sq_pushstring(v, _SC("tjsIsValid"), -1);
sq_newclosure(v, TJSObject::tjsIsValid, 0);
sq_createslot(v, -3);
sq_pushstring(v, _SC("tjsOverride"), -1);
sq_newclosure(v, TJSObject::tjsOverride, 0);
sq_setparamscheck(v, -2, _SC(".sc"));
sq_createslot(v, -3);
}
return 1;
}
