-/*\r
- see copyright notice in squirrel.h\r
-*/\r
-#include "sqpcheader.h"\r
-#include "sqcompiler.h"\r
-#include "sqfuncproto.h"\r
-#include "sqstring.h"\r
-#include "sqtable.h"\r
-#include "sqopcodes.h"\r
-#include "sqfuncstate.h"\r
-\r
-#ifdef _DEBUG_DUMP\r
-SQInstructionDesc g_InstrDesc[]={\r
- {_SC("_OP_LINE")},\r
- {_SC("_OP_LOAD")},\r
- {_SC("_OP_LOADINT")},\r
- {_SC("_OP_DLOAD")},\r
- {_SC("_OP_TAILCALL")},\r
- {_SC("_OP_CALL")},\r
- {_SC("_OP_PREPCALL")},\r
- {_SC("_OP_PREPCALLK")},\r
- {_SC("_OP_GETK")},\r
- {_SC("_OP_MOVE")},\r
- {_SC("_OP_NEWSLOT")},\r
- {_SC("_OP_DELETE")},\r
- {_SC("_OP_SET")},\r
- {_SC("_OP_GET")},\r
- {_SC("_OP_EQ")},\r
- {_SC("_OP_NE")},\r
- {_SC("_OP_ARITH")},\r
- {_SC("_OP_BITW")},\r
- {_SC("_OP_RETURN")},\r
- {_SC("_OP_LOADNULLS")},\r
- {_SC("_OP_LOADROOTTABLE")},\r
- {_SC("_OP_LOADBOOL")},\r
- {_SC("_OP_DMOVE")},\r
- {_SC("_OP_JMP")},\r
- {_SC("_OP_JNZ")},\r
- {_SC("_OP_JZ")},\r
- {_SC("_OP_LOADFREEVAR")},\r
- {_SC("_OP_VARGC")},\r
- {_SC("_OP_GETVARGV")},\r
- {_SC("_OP_NEWTABLE")},\r
- {_SC("_OP_NEWARRAY")},\r
- {_SC("_OP_APPENDARRAY")},\r
- {_SC("_OP_GETPARENT")},\r
- {_SC("_OP_COMPARITH")},\r
- {_SC("_OP_COMPARITHL")},\r
- {_SC("_OP_INC")},\r
- {_SC("_OP_INCL")},\r
- {_SC("_OP_PINC")},\r
- {_SC("_OP_PINCL")},\r
- {_SC("_OP_CMP")},\r
- {_SC("_OP_EXISTS")},\r
- {_SC("_OP_INSTANCEOF")},\r
- {_SC("_OP_AND")},\r
- {_SC("_OP_OR")},\r
- {_SC("_OP_NEG")},\r
- {_SC("_OP_NOT")},\r
- {_SC("_OP_BWNOT")},\r
- {_SC("_OP_CLOSURE")},\r
- {_SC("_OP_YIELD")},\r
- {_SC("_OP_RESUME")},\r
- {_SC("_OP_FOREACH")},\r
- {_SC("_OP_DELEGATE")},\r
- {_SC("_OP_CLONE")},\r
- {_SC("_OP_TYPEOF")},\r
- {_SC("_OP_PUSHTRAP")},\r
- {_SC("_OP_POPTRAP")},\r
- {_SC("_OP_THROW")},\r
- {_SC("_OP_CLASS")},\r
- {_SC("_OP_NEWSLOTA")}\r
-};\r
-#endif\r
-void DumpLiteral(SQObjectPtr &o)\r
-{\r
- switch(type(o)){\r
- case OT_STRING: scprintf(_SC("\"%s\""),_stringval(o));break;\r
- case OT_FLOAT: scprintf(_SC("{%f}"),_float(o));break;\r
- case OT_INTEGER: scprintf(_SC("{%d}"),_integer(o));break;\r
- default: assert(0); break; //shut up compiler\r
- }\r
-}\r
-\r
-SQFuncState::SQFuncState(SQSharedState *ss,SQFuncState *parent,CompilerErrorFunc efunc,void *ed)\r
-{\r
- _nliterals = 0;\r
- _literals = SQTable::Create(ss,0);\r
- _strings = SQTable::Create(ss,0);\r
- _sharedstate = ss;\r
- _lastline = 0;\r
- _optimization = true;\r
- _parent = parent;\r
- _stacksize = 0;\r
- _traps = 0;\r
- _returnexp = 0;\r
- _varparams = false;\r
- _errfunc = efunc;\r
- _errtarget = ed;\r
- _bgenerator = false;\r
-\r
-}\r
-\r
-void SQFuncState::Error(const SQChar *err)\r
-{\r
- _errfunc(_errtarget,err);\r
-}\r
-\r
-#ifdef _DEBUG_DUMP\r
-void SQFuncState::Dump(SQFunctionProto *func)\r
-{\r
- SQUnsignedInteger n=0,i;\r
- scprintf(_SC("SQInstruction sizeof %d\n"),sizeof(SQInstruction));\r
- scprintf(_SC("SQObject sizeof %d\n"),sizeof(SQObject));\r
- scprintf(_SC("--------------------------------------------------------------------\n"));\r
- scprintf(_SC("*****FUNCTION [%s]\n"),type(func->_name)==OT_STRING?_stringval(func->_name):_SC("unknown"));\r
- scprintf(_SC("-----LITERALS\n"));\r
- SQObjectPtr refidx,key,val;\r
- SQInteger idx;\r
- SQObjectPtrVec templiterals;\r
- templiterals.resize(_nliterals);\r
- while((idx=_table(_literals)->Next(false,refidx,key,val))!=-1) {\r
- refidx=idx;\r
- templiterals[_integer(val)]=key;\r
- }\r
- for(i=0;i<templiterals.size();i++){\r
- scprintf(_SC("[%d] "),n);\r
- DumpLiteral(templiterals[i]);\r
- scprintf(_SC("\n"));\r
- n++;\r
- }\r
- scprintf(_SC("-----PARAMS\n"));\r
- if(_varparams)\r
- scprintf(_SC("<<VARPARAMS>>\n"));\r
- n=0;\r
- for(i=0;i<_parameters.size();i++){\r
- scprintf(_SC("[%d] "),n);\r
- DumpLiteral(_parameters[i]);\r
- scprintf(_SC("\n"));\r
- n++;\r
- }\r
- scprintf(_SC("-----LOCALS\n"));\r
- for(i=0;i<func->_localvarinfos.size();i++){\r
- SQLocalVarInfo lvi=func->_localvarinfos[i];\r
- scprintf(_SC("[%d] %s \t%d %d\n"),lvi._pos,_stringval(lvi._name),lvi._start_op,lvi._end_op);\r
- n++;\r
- }\r
- scprintf(_SC("-----LINE INFO\n"));\r
- for(i=0;i<_lineinfos.size();i++){\r
- SQLineInfo li=_lineinfos[i];\r
- scprintf(_SC("op [%d] line [%d] \n"),li._op,li._line);\r
- n++;\r
- }\r
- scprintf(_SC("-----dump\n"));\r
- n=0;\r
- for(i=0;i<_instructions.size();i++){\r
- SQInstruction &inst=_instructions[i];\r
- if(inst.op==_OP_LOAD || inst.op==_OP_DLOAD || inst.op==_OP_PREPCALLK || inst.op==_OP_GETK ){\r
- \r
- SQInteger lidx = inst._arg1;\r
- scprintf(_SC("[%03d] %15s %d "),n,g_InstrDesc[inst.op].name,inst._arg0);\r
- if(lidx >= 0xFFFFFFFF)\r
- scprintf(_SC("null"));\r
- else {\r
- SQInteger refidx;\r
- SQObjectPtr val,key,refo;\r
- while(((refidx=_table(_literals)->Next(false,refo,key,val))!= -1) && (_integer(val) != lidx)) {\r
- refo = refidx; \r
- }\r
- DumpLiteral(key);\r
- }\r
- if(inst.op != _OP_DLOAD) {\r
- scprintf(_SC(" %d %d \n"),inst._arg2,inst._arg3);\r
- }\r
- else {\r
- scprintf(_SC(" %d "),inst._arg2);\r
- lidx = inst._arg3;\r
- if(lidx >= 0xFFFFFFFF)\r
- scprintf(_SC("null"));\r
- else {\r
- SQInteger refidx;\r
- SQObjectPtr val,key,refo;\r
- while(((refidx=_table(_literals)->Next(false,refo,key,val))!= -1) && (_integer(val) != lidx)) {\r
- refo = refidx; \r
- }\r
- DumpLiteral(key);\r
- scprintf(_SC("\n"));\r
- }\r
- }\r
- }\r
- else if(inst.op==_OP_ARITH){\r
- scprintf(_SC("[%03d] %15s %d %d %d %c\n"),n,g_InstrDesc[inst.op].name,inst._arg0,inst._arg1,inst._arg2,inst._arg3);\r
- }\r
- else \r
- scprintf(_SC("[%03d] %15s %d %d %d %d\n"),n,g_InstrDesc[inst.op].name,inst._arg0,inst._arg1,inst._arg2,inst._arg3);\r
- n++;\r
- }\r
- scprintf(_SC("-----\n"));\r
- scprintf(_SC("stack size[%d]\n"),func->_stacksize);\r
- scprintf(_SC("--------------------------------------------------------------------\n\n"));\r
-}\r
-#endif\r
-\r
-SQInteger SQFuncState::GetNumericConstant(const SQInteger cons)\r
-{\r
- return GetConstant(SQObjectPtr(cons));\r
-}\r
-\r
-SQInteger SQFuncState::GetNumericConstant(const SQFloat cons)\r
-{\r
- return GetConstant(SQObjectPtr(cons));\r
-}\r
-\r
-SQInteger SQFuncState::GetConstant(const SQObject &cons)\r
-{\r
- SQObjectPtr val;\r
- if(!_table(_literals)->Get(cons,val))\r
- {\r
- val = _nliterals;\r
- _table(_literals)->NewSlot(cons,val);\r
- _nliterals++;\r
- if(_nliterals > MAX_LITERALS) {\r
- val.Null();\r
- Error(_SC("internal compiler error: too many literals"));\r
- }\r
- }\r
- return _integer(val);\r
-}\r
-\r
-void SQFuncState::SetIntructionParams(SQInteger pos,SQInteger arg0,SQInteger arg1,SQInteger arg2,SQInteger arg3)\r
-{\r
- _instructions[pos]._arg0=(unsigned char)*((SQUnsignedInteger *)&arg0);\r
- _instructions[pos]._arg1=(SQInt32)*((SQUnsignedInteger *)&arg1);\r
- _instructions[pos]._arg2=(unsigned char)*((SQUnsignedInteger *)&arg2);\r
- _instructions[pos]._arg3=(unsigned char)*((SQUnsignedInteger *)&arg3);\r
-}\r
-\r
-void SQFuncState::SetIntructionParam(SQInteger pos,SQInteger arg,SQInteger val)\r
-{\r
- switch(arg){\r
- case 0:_instructions[pos]._arg0=(unsigned char)*((SQUnsignedInteger *)&val);break;\r
- case 1:case 4:_instructions[pos]._arg1=(SQInt32)*((SQUnsignedInteger *)&val);break;\r
- case 2:_instructions[pos]._arg2=(unsigned char)*((SQUnsignedInteger *)&val);break;\r
- case 3:_instructions[pos]._arg3=(unsigned char)*((SQUnsignedInteger *)&val);break;\r
- };\r
-}\r
-\r
-SQInteger SQFuncState::AllocStackPos()\r
-{\r
- SQInteger npos=_vlocals.size();\r
- _vlocals.push_back(SQLocalVarInfo());\r
- if(_vlocals.size()>((SQUnsignedInteger)_stacksize)) {\r
- if(_stacksize>MAX_FUNC_STACKSIZE) Error(_SC("internal compiler error: too many locals"));\r
- _stacksize=_vlocals.size();\r
- }\r
- return npos;\r
-}\r
-\r
-SQInteger SQFuncState::PushTarget(SQInteger n)\r
-{\r
- if(n!=-1){\r
- _targetstack.push_back(n);\r
- return n;\r
- }\r
- n=AllocStackPos();\r
- _targetstack.push_back(n);\r
- return n;\r
-}\r
-\r
-SQInteger SQFuncState::GetUpTarget(SQInteger n){\r
- return _targetstack[((_targetstack.size()-1)-n)];\r
-}\r
-\r
-SQInteger SQFuncState::TopTarget(){\r
- return _targetstack.back();\r
-}\r
-SQInteger SQFuncState::PopTarget()\r
-{\r
- SQInteger npos=_targetstack.back();\r
- SQLocalVarInfo t=_vlocals[_targetstack.back()];\r
- if(type(t._name)==OT_NULL){\r
- _vlocals.pop_back();\r
- }\r
- _targetstack.pop_back();\r
- return npos;\r
-}\r
-\r
-SQInteger SQFuncState::GetStackSize()\r
-{\r
- return _vlocals.size();\r
-}\r
-\r
-void SQFuncState::SetStackSize(SQInteger n)\r
-{\r
- SQInteger size=_vlocals.size();\r
- while(size>n){\r
- size--;\r
- SQLocalVarInfo lvi=_vlocals.back();\r
- if(type(lvi._name)!=OT_NULL){\r
- lvi._end_op=GetCurrentPos();\r
- _localvarinfos.push_back(lvi);\r
- }\r
- _vlocals.pop_back();\r
- }\r
-}\r
-\r
-bool SQFuncState::IsLocal(SQUnsignedInteger stkpos)\r
-{\r
- if(stkpos>=_vlocals.size())return false;\r
- else if(type(_vlocals[stkpos]._name)!=OT_NULL)return true;\r
- return false;\r
-}\r
-\r
-SQInteger SQFuncState::PushLocalVariable(const SQObject &name)\r
-{\r
- SQInteger pos=_vlocals.size();\r
- SQLocalVarInfo lvi;\r
- lvi._name=name;\r
- lvi._start_op=GetCurrentPos()+1;\r
- lvi._pos=_vlocals.size();\r
- _vlocals.push_back(lvi);\r
- if(_vlocals.size()>((SQUnsignedInteger)_stacksize))_stacksize=_vlocals.size();\r
- \r
- return pos;\r
-}\r
-\r
-SQInteger SQFuncState::GetLocalVariable(const SQObject &name)\r
-{\r
- SQInteger locals=_vlocals.size();\r
- while(locals>=1){\r
- if(type(_vlocals[locals-1]._name)==OT_STRING && _string(_vlocals[locals-1]._name)==_string(name)){\r
- return locals-1;\r
- }\r
- locals--;\r
- }\r
- return -1;\r
-}\r
-\r
-SQInteger SQFuncState::GetOuterVariable(const SQObject &name)\r
-{\r
- SQInteger outers = _outervalues.size();\r
- for(SQInteger i = 0; i<outers; i++) {\r
- if(_string(_outervalues[i]._name) == _string(name))\r
- return i;\r
- }\r
- return -1;\r
-}\r
-\r
-void SQFuncState::AddOuterValue(const SQObject &name)\r
-{\r
- SQInteger pos=-1;\r
- if(_parent) { \r
- pos = _parent->GetLocalVariable(name);\r
- if(pos == -1) {\r
- pos = _parent->GetOuterVariable(name);\r
- if(pos != -1) {\r
- _outervalues.push_back(SQOuterVar(name,SQObjectPtr(SQInteger(pos)),otOUTER)); //local\r
- return;\r
- }\r
- }\r
- else {\r
- _outervalues.push_back(SQOuterVar(name,SQObjectPtr(SQInteger(pos)),otLOCAL)); //local\r
- return;\r
- }\r
- } \r
- _outervalues.push_back(SQOuterVar(name,name,otSYMBOL)); //global\r
-}\r
-\r
-void SQFuncState::AddParameter(const SQObject &name)\r
-{\r
- PushLocalVariable(name);\r
- _parameters.push_back(name);\r
-}\r
-\r
-void SQFuncState::AddLineInfos(SQInteger line,bool lineop,bool force)\r
-{\r
- if(_lastline!=line || force){\r
- SQLineInfo li;\r
- li._line=line;li._op=(GetCurrentPos()+1);\r
- if(lineop)AddInstruction(_OP_LINE,0,line);\r
- _lineinfos.push_back(li);\r
- _lastline=line;\r
- }\r
-}\r
-\r
-void SQFuncState::AddInstruction(SQInstruction &i)\r
-{\r
- SQInteger size = _instructions.size();\r
- if(size > 0 && _optimization){ //simple optimizer\r
- SQInstruction &pi = _instructions[size-1];//previous instruction\r
- switch(i.op) {\r
- case _OP_RETURN:\r
- if( _parent && i._arg0 != MAX_FUNC_STACKSIZE && pi.op == _OP_CALL && _returnexp < size-1) {\r
- pi.op = _OP_TAILCALL;\r
- }\r
- break;\r
- case _OP_GET:\r
- if( pi.op == _OP_LOAD && pi._arg0 == i._arg2 && (!IsLocal(pi._arg0))){\r
- pi._arg1 = pi._arg1;\r
- pi._arg2 = (unsigned char)i._arg1;\r
- pi.op = _OP_GETK;\r
- pi._arg0 = i._arg0;\r
- \r
- return;\r
- }\r
- break;\r
- case _OP_PREPCALL:\r
- if( pi.op == _OP_LOAD && pi._arg0 == i._arg1 && (!IsLocal(pi._arg0))){\r
- pi.op = _OP_PREPCALLK;\r
- pi._arg0 = i._arg0;\r
- pi._arg1 = pi._arg1;\r
- pi._arg2 = i._arg2;\r
- pi._arg3 = i._arg3;\r
- return;\r
- }\r
- break;\r
- case _OP_APPENDARRAY:\r
- if(pi.op == _OP_LOAD && pi._arg0 == i._arg1 && (!IsLocal(pi._arg0))){\r
- pi.op = _OP_APPENDARRAY;\r
- pi._arg0 = i._arg0;\r
- pi._arg1 = pi._arg1;\r
- pi._arg2 = MAX_FUNC_STACKSIZE;\r
- pi._arg3 = MAX_FUNC_STACKSIZE;\r
- return;\r
- }\r
- break;\r
- case _OP_MOVE: \r
- if((pi.op == _OP_GET || pi.op == _OP_ARITH || pi.op == _OP_BITW) && (pi._arg0 == i._arg1))\r
- {\r
- pi._arg0 = i._arg0;\r
- _optimization = false;\r
- return;\r
- }\r
-\r
- if(pi.op == _OP_MOVE)\r
- {\r
- pi.op = _OP_DMOVE;\r
- pi._arg2 = i._arg0;\r
- pi._arg3 = (unsigned char)i._arg1;\r
- return;\r
- }\r
- break;\r
- case _OP_LOAD:\r
- if(pi.op == _OP_LOAD && i._arg1 < 256) {\r
- pi.op = _OP_DLOAD;\r
- pi._arg2 = i._arg0;\r
- pi._arg3 = (unsigned char)i._arg1;\r
- return;\r
- }\r
- break;\r
- case _OP_EQ:case _OP_NE:\r
- if(pi.op == _OP_LOAD && pi._arg0 == i._arg1 && (!IsLocal(pi._arg0) ))\r
- {\r
- pi.op = i.op;\r
- pi._arg0 = i._arg0;\r
- pi._arg1 = pi._arg1;\r
- pi._arg2 = i._arg2;\r
- pi._arg3 = MAX_FUNC_STACKSIZE;\r
- return;\r
- }\r
- break;\r
- case _OP_LOADNULLS:\r
- if((pi.op == _OP_LOADNULLS && pi._arg0+pi._arg1 == i._arg0)) {\r
- \r
- pi._arg1 = pi._arg1 + 1;\r
- pi.op = _OP_LOADNULLS;\r
- return;\r
- }\r
- break;\r
- case _OP_LINE:\r
- if(pi.op == _OP_LINE) {\r
- _instructions.pop_back();\r
- _lineinfos.pop_back();\r
- }\r
- break;\r
- }\r
- }\r
- _optimization = true;\r
- _instructions.push_back(i);\r
-}\r
-\r
-SQObject SQFuncState::CreateString(const SQChar *s,SQInteger len)\r
-{\r
- SQObjectPtr ns(SQString::Create(_sharedstate,s,len));\r
- _table(_strings)->NewSlot(ns,(SQInteger)1);\r
- return ns;\r
-}\r
-\r
-SQFunctionProto *SQFuncState::BuildProto()\r
-{\r
- SQFunctionProto *f=SQFunctionProto::Create();\r
- f->_literals.resize(_nliterals);\r
- SQObjectPtr refidx,key,val;\r
- SQInteger idx;\r
-\r
- f->_stacksize = _stacksize;\r
- f->_sourcename = _sourcename;\r
- f->_bgenerator = _bgenerator;\r
- f->_name = _name;\r
-\r
- while((idx=_table(_literals)->Next(false,refidx,key,val))!=-1) {\r
- f->_literals[_integer(val)]=key;\r
- refidx=idx;\r
- }\r
-\r
- f->_functions.resize(_functions.size());\r
- f->_functions.copy(_functions);\r
- f->_parameters.resize(_parameters.size());\r
- f->_parameters.copy(_parameters);\r
- f->_outervalues.resize(_outervalues.size());\r
- f->_outervalues.copy(_outervalues);\r
- f->_instructions.resize(_instructions.size());\r
- f->_instructions.copy(_instructions);\r
- f->_localvarinfos.resize(_localvarinfos.size());\r
- f->_localvarinfos.copy(_localvarinfos);\r
- f->_lineinfos.resize(_lineinfos.size());\r
- f->_lineinfos.copy(_lineinfos);\r
- f->_varparams = _varparams;\r
-\r
- return f;\r
-}\r
-\r
-SQFuncState *SQFuncState::PushChildState(SQSharedState *ss)\r
-{\r
- SQFuncState *child = (SQFuncState *)sq_malloc(sizeof(SQFuncState));\r
- new (child) SQFuncState(ss,this,_errfunc,_errtarget);\r
- _childstates.push_back(child);\r
- return child;\r
-}\r
-\r
-void SQFuncState::PopChildState()\r
-{\r
- SQFuncState *child = _childstates.back();\r
- sq_delete(child,SQFuncState);\r
- _childstates.pop_back();\r
-}\r
-\r
-SQFuncState::~SQFuncState()\r
-{\r
- while(_childstates.size() > 0)\r
- {\r
- PopChildState();\r
- }\r
-}\r
+/*
+ see copyright notice in squirrel.h
+*/
+#include "sqpcheader.h"
+#include "sqcompiler.h"
+#include "sqfuncproto.h"
+#include "sqstring.h"
+#include "sqtable.h"
+#include "sqopcodes.h"
+#include "sqfuncstate.h"
+
+#ifdef _DEBUG_DUMP
+SQInstructionDesc g_InstrDesc[]={
+ {_SC("_OP_LINE")},
+ {_SC("_OP_LOAD")},
+ {_SC("_OP_LOADINT")},
+ {_SC("_OP_DLOAD")},
+ {_SC("_OP_TAILCALL")},
+ {_SC("_OP_CALL")},
+ {_SC("_OP_PREPCALL")},
+ {_SC("_OP_PREPCALLK")},
+ {_SC("_OP_GETK")},
+ {_SC("_OP_MOVE")},
+ {_SC("_OP_NEWSLOT")},
+ {_SC("_OP_DELETE")},
+ {_SC("_OP_SET")},
+ {_SC("_OP_GET")},
+ {_SC("_OP_EQ")},
+ {_SC("_OP_NE")},
+ {_SC("_OP_ARITH")},
+ {_SC("_OP_BITW")},
+ {_SC("_OP_RETURN")},
+ {_SC("_OP_LOADNULLS")},
+ {_SC("_OP_LOADROOTTABLE")},
+ {_SC("_OP_LOADBOOL")},
+ {_SC("_OP_DMOVE")},
+ {_SC("_OP_JMP")},
+ {_SC("_OP_JNZ")},
+ {_SC("_OP_JZ")},
+ {_SC("_OP_LOADFREEVAR")},
+ {_SC("_OP_VARGC")},
+ {_SC("_OP_GETVARGV")},
+ {_SC("_OP_NEWTABLE")},
+ {_SC("_OP_NEWARRAY")},
+ {_SC("_OP_APPENDARRAY")},
+ {_SC("_OP_GETPARENT")},
+ {_SC("_OP_COMPARITH")},
+ {_SC("_OP_COMPARITHL")},
+ {_SC("_OP_INC")},
+ {_SC("_OP_INCL")},
+ {_SC("_OP_PINC")},
+ {_SC("_OP_PINCL")},
+ {_SC("_OP_CMP")},
+ {_SC("_OP_EXISTS")},
+ {_SC("_OP_INSTANCEOF")},
+ {_SC("_OP_AND")},
+ {_SC("_OP_OR")},
+ {_SC("_OP_NEG")},
+ {_SC("_OP_NOT")},
+ {_SC("_OP_BWNOT")},
+ {_SC("_OP_CLOSURE")},
+ {_SC("_OP_YIELD")},
+ {_SC("_OP_RESUME")},
+ {_SC("_OP_FOREACH")},
+ {_SC("_OP_DELEGATE")},
+ {_SC("_OP_CLONE")},
+ {_SC("_OP_TYPEOF")},
+ {_SC("_OP_PUSHTRAP")},
+ {_SC("_OP_POPTRAP")},
+ {_SC("_OP_THROW")},
+ {_SC("_OP_CLASS")},
+ {_SC("_OP_NEWSLOTA")}
+};
+#endif
+void DumpLiteral(SQObjectPtr &o)
+{
+ switch(type(o)){
+ case OT_STRING: scprintf(_SC("\"%s\""),_stringval(o));break;
+ case OT_FLOAT: scprintf(_SC("{%f}"),_float(o));break;
+ case OT_INTEGER: scprintf(_SC("{%d}"),_integer(o));break;
+ default: assert(0); break; //shut up compiler
+ }
+}
+
+SQFuncState::SQFuncState(SQSharedState *ss,SQFuncState *parent,CompilerErrorFunc efunc,void *ed)
+{
+ _nliterals = 0;
+ _literals = SQTable::Create(ss,0);
+ _strings = SQTable::Create(ss,0);
+ _sharedstate = ss;
+ _lastline = 0;
+ _optimization = true;
+ _parent = parent;
+ _stacksize = 0;
+ _traps = 0;
+ _returnexp = 0;
+ _varparams = false;
+ _errfunc = efunc;
+ _errtarget = ed;
+ _bgenerator = false;
+
+}
+
+void SQFuncState::Error(const SQChar *err)
+{
+ _errfunc(_errtarget,err);
+}
+
+#ifdef _DEBUG_DUMP
+void SQFuncState::Dump(SQFunctionProto *func)
+{
+ SQUnsignedInteger n=0,i;
+ scprintf(_SC("SQInstruction sizeof %d\n"),sizeof(SQInstruction));
+ scprintf(_SC("SQObject sizeof %d\n"),sizeof(SQObject));
+ scprintf(_SC("--------------------------------------------------------------------\n"));
+ scprintf(_SC("*****FUNCTION [%s]\n"),type(func->_name)==OT_STRING?_stringval(func->_name):_SC("unknown"));
+ scprintf(_SC("-----LITERALS\n"));
+ SQObjectPtr refidx,key,val;
+ SQInteger idx;
+ SQObjectPtrVec templiterals;
+ templiterals.resize(_nliterals);
+ while((idx=_table(_literals)->Next(false,refidx,key,val))!=-1) {
+ refidx=idx;
+ templiterals[_integer(val)]=key;
+ }
+ for(i=0;i<templiterals.size();i++){
+ scprintf(_SC("[%d] "),n);
+ DumpLiteral(templiterals[i]);
+ scprintf(_SC("\n"));
+ n++;
+ }
+ scprintf(_SC("-----PARAMS\n"));
+ if(_varparams)
+ scprintf(_SC("<<VARPARAMS>>\n"));
+ n=0;
+ for(i=0;i<_parameters.size();i++){
+ scprintf(_SC("[%d] "),n);
+ DumpLiteral(_parameters[i]);
+ scprintf(_SC("\n"));
+ n++;
+ }
+ scprintf(_SC("-----LOCALS\n"));
+ for(i=0;i<func->_localvarinfos.size();i++){
+ SQLocalVarInfo lvi=func->_localvarinfos[i];
+ scprintf(_SC("[%d] %s \t%d %d\n"),lvi._pos,_stringval(lvi._name),lvi._start_op,lvi._end_op);
+ n++;
+ }
+ scprintf(_SC("-----LINE INFO\n"));
+ for(i=0;i<_lineinfos.size();i++){
+ SQLineInfo li=_lineinfos[i];
+ scprintf(_SC("op [%d] line [%d] \n"),li._op,li._line);
+ n++;
+ }
+ scprintf(_SC("-----dump\n"));
+ n=0;
+ for(i=0;i<_instructions.size();i++){
+ SQInstruction &inst=_instructions[i];
+ if(inst.op==_OP_LOAD || inst.op==_OP_DLOAD || inst.op==_OP_PREPCALLK || inst.op==_OP_GETK ){
+
+ SQInteger lidx = inst._arg1;
+ scprintf(_SC("[%03d] %15s %d "),n,g_InstrDesc[inst.op].name,inst._arg0);
+ if(lidx >= 0xFFFFFFFF)
+ scprintf(_SC("null"));
+ else {
+ SQInteger refidx;
+ SQObjectPtr val,key,refo;
+ while(((refidx=_table(_literals)->Next(false,refo,key,val))!= -1) && (_integer(val) != lidx)) {
+ refo = refidx;
+ }
+ DumpLiteral(key);
+ }
+ if(inst.op != _OP_DLOAD) {
+ scprintf(_SC(" %d %d \n"),inst._arg2,inst._arg3);
+ }
+ else {
+ scprintf(_SC(" %d "),inst._arg2);
+ lidx = inst._arg3;
+ if(lidx >= 0xFFFFFFFF)
+ scprintf(_SC("null"));
+ else {
+ SQInteger refidx;
+ SQObjectPtr val,key,refo;
+ while(((refidx=_table(_literals)->Next(false,refo,key,val))!= -1) && (_integer(val) != lidx)) {
+ refo = refidx;
+ }
+ DumpLiteral(key);
+ scprintf(_SC("\n"));
+ }
+ }
+ }
+ else if(inst.op==_OP_ARITH){
+ scprintf(_SC("[%03d] %15s %d %d %d %c\n"),n,g_InstrDesc[inst.op].name,inst._arg0,inst._arg1,inst._arg2,inst._arg3);
+ }
+ else
+ scprintf(_SC("[%03d] %15s %d %d %d %d\n"),n,g_InstrDesc[inst.op].name,inst._arg0,inst._arg1,inst._arg2,inst._arg3);
+ n++;
+ }
+ scprintf(_SC("-----\n"));
+ scprintf(_SC("stack size[%d]\n"),func->_stacksize);
+ scprintf(_SC("--------------------------------------------------------------------\n\n"));
+}
+#endif
+
+SQInteger SQFuncState::GetNumericConstant(const SQInteger cons)
+{
+ return GetConstant(SQObjectPtr(cons));
+}
+
+SQInteger SQFuncState::GetNumericConstant(const SQFloat cons)
+{
+ return GetConstant(SQObjectPtr(cons));
+}
+
+SQInteger SQFuncState::GetConstant(const SQObject &cons)
+{
+ SQObjectPtr val;
+ if(!_table(_literals)->Get(cons,val))
+ {
+ val = _nliterals;
+ _table(_literals)->NewSlot(cons,val);
+ _nliterals++;
+ if(_nliterals > MAX_LITERALS) {
+ val.Null();
+ Error(_SC("internal compiler error: too many literals"));
+ }
+ }
+ return _integer(val);
+}
+
+void SQFuncState::SetIntructionParams(SQInteger pos,SQInteger arg0,SQInteger arg1,SQInteger arg2,SQInteger arg3)
+{
+ _instructions[pos]._arg0=(unsigned char)*((SQUnsignedInteger *)&arg0);
+ _instructions[pos]._arg1=(SQInt32)*((SQUnsignedInteger *)&arg1);
+ _instructions[pos]._arg2=(unsigned char)*((SQUnsignedInteger *)&arg2);
+ _instructions[pos]._arg3=(unsigned char)*((SQUnsignedInteger *)&arg3);
+}
+
+void SQFuncState::SetIntructionParam(SQInteger pos,SQInteger arg,SQInteger val)
+{
+ switch(arg){
+ case 0:_instructions[pos]._arg0=(unsigned char)*((SQUnsignedInteger *)&val);break;
+ case 1:case 4:_instructions[pos]._arg1=(SQInt32)*((SQUnsignedInteger *)&val);break;
+ case 2:_instructions[pos]._arg2=(unsigned char)*((SQUnsignedInteger *)&val);break;
+ case 3:_instructions[pos]._arg3=(unsigned char)*((SQUnsignedInteger *)&val);break;
+ };
+}
+
+SQInteger SQFuncState::AllocStackPos()
+{
+ SQInteger npos=_vlocals.size();
+ _vlocals.push_back(SQLocalVarInfo());
+ if(_vlocals.size()>((SQUnsignedInteger)_stacksize)) {
+ if(_stacksize>MAX_FUNC_STACKSIZE) Error(_SC("internal compiler error: too many locals"));
+ _stacksize=_vlocals.size();
+ }
+ return npos;
+}
+
+SQInteger SQFuncState::PushTarget(SQInteger n)
+{
+ if(n!=-1){
+ _targetstack.push_back(n);
+ return n;
+ }
+ n=AllocStackPos();
+ _targetstack.push_back(n);
+ return n;
+}
+
+SQInteger SQFuncState::GetUpTarget(SQInteger n){
+ return _targetstack[((_targetstack.size()-1)-n)];
+}
+
+SQInteger SQFuncState::TopTarget(){
+ return _targetstack.back();
+}
+SQInteger SQFuncState::PopTarget()
+{
+ SQInteger npos=_targetstack.back();
+ SQLocalVarInfo t=_vlocals[_targetstack.back()];
+ if(type(t._name)==OT_NULL){
+ _vlocals.pop_back();
+ }
+ _targetstack.pop_back();
+ return npos;
+}
+
+SQInteger SQFuncState::GetStackSize()
+{
+ return _vlocals.size();
+}
+
+void SQFuncState::SetStackSize(SQInteger n)
+{
+ SQInteger size=_vlocals.size();
+ while(size>n){
+ size--;
+ SQLocalVarInfo lvi=_vlocals.back();
+ if(type(lvi._name)!=OT_NULL){
+ lvi._end_op=GetCurrentPos();
+ _localvarinfos.push_back(lvi);
+ }
+ _vlocals.pop_back();
+ }
+}
+
+bool SQFuncState::IsLocal(SQUnsignedInteger stkpos)
+{
+ if(stkpos>=_vlocals.size())return false;
+ else if(type(_vlocals[stkpos]._name)!=OT_NULL)return true;
+ return false;
+}
+
+SQInteger SQFuncState::PushLocalVariable(const SQObject &name)
+{
+ SQInteger pos=_vlocals.size();
+ SQLocalVarInfo lvi;
+ lvi._name=name;
+ lvi._start_op=GetCurrentPos()+1;
+ lvi._pos=_vlocals.size();
+ _vlocals.push_back(lvi);
+ if(_vlocals.size()>((SQUnsignedInteger)_stacksize))_stacksize=_vlocals.size();
+
+ return pos;
+}
+
+SQInteger SQFuncState::GetLocalVariable(const SQObject &name)
+{
+ SQInteger locals=_vlocals.size();
+ while(locals>=1){
+ if(type(_vlocals[locals-1]._name)==OT_STRING && _string(_vlocals[locals-1]._name)==_string(name)){
+ return locals-1;
+ }
+ locals--;
+ }
+ return -1;
+}
+
+SQInteger SQFuncState::GetOuterVariable(const SQObject &name)
+{
+ SQInteger outers = _outervalues.size();
+ for(SQInteger i = 0; i<outers; i++) {
+ if(_string(_outervalues[i]._name) == _string(name))
+ return i;
+ }
+ return -1;
+}
+
+void SQFuncState::AddOuterValue(const SQObject &name)
+{
+ SQInteger pos=-1;
+ if(_parent) {
+ pos = _parent->GetLocalVariable(name);
+ if(pos == -1) {
+ pos = _parent->GetOuterVariable(name);
+ if(pos != -1) {
+ _outervalues.push_back(SQOuterVar(name,SQObjectPtr(SQInteger(pos)),otOUTER)); //local
+ return;
+ }
+ }
+ else {
+ _outervalues.push_back(SQOuterVar(name,SQObjectPtr(SQInteger(pos)),otLOCAL)); //local
+ return;
+ }
+ }
+ _outervalues.push_back(SQOuterVar(name,name,otSYMBOL)); //global
+}
+
+void SQFuncState::AddParameter(const SQObject &name)
+{
+ PushLocalVariable(name);
+ _parameters.push_back(name);
+}
+
+void SQFuncState::AddLineInfos(SQInteger line,bool lineop,bool force)
+{
+ if(_lastline!=line || force){
+ SQLineInfo li;
+ li._line=line;li._op=(GetCurrentPos()+1);
+ if(lineop)AddInstruction(_OP_LINE,0,line);
+ _lineinfos.push_back(li);
+ _lastline=line;
+ }
+}
+
+void SQFuncState::AddInstruction(SQInstruction &i)
+{
+ SQInteger size = _instructions.size();
+ if(size > 0 && _optimization){ //simple optimizer
+ SQInstruction &pi = _instructions[size-1];//previous instruction
+ switch(i.op) {
+ case _OP_RETURN:
+ if( _parent && i._arg0 != MAX_FUNC_STACKSIZE && pi.op == _OP_CALL && _returnexp < size-1) {
+ pi.op = _OP_TAILCALL;
+ }
+ break;
+ case _OP_GET:
+ if( pi.op == _OP_LOAD && pi._arg0 == i._arg2 && (!IsLocal(pi._arg0))){
+ pi._arg1 = pi._arg1;
+ pi._arg2 = (unsigned char)i._arg1;
+ pi.op = _OP_GETK;
+ pi._arg0 = i._arg0;
+
+ return;
+ }
+ break;
+ case _OP_PREPCALL:
+ if( pi.op == _OP_LOAD && pi._arg0 == i._arg1 && (!IsLocal(pi._arg0))){
+ pi.op = _OP_PREPCALLK;
+ pi._arg0 = i._arg0;
+ pi._arg1 = pi._arg1;
+ pi._arg2 = i._arg2;
+ pi._arg3 = i._arg3;
+ return;
+ }
+ break;
+ case _OP_APPENDARRAY:
+ if(pi.op == _OP_LOAD && pi._arg0 == i._arg1 && (!IsLocal(pi._arg0))){
+ pi.op = _OP_APPENDARRAY;
+ pi._arg0 = i._arg0;
+ pi._arg1 = pi._arg1;
+ pi._arg2 = MAX_FUNC_STACKSIZE;
+ pi._arg3 = MAX_FUNC_STACKSIZE;
+ return;
+ }
+ break;
+ case _OP_MOVE:
+ if((pi.op == _OP_GET || pi.op == _OP_ARITH || pi.op == _OP_BITW) && (pi._arg0 == i._arg1))
+ {
+ pi._arg0 = i._arg0;
+ _optimization = false;
+ return;
+ }
+
+ if(pi.op == _OP_MOVE)
+ {
+ pi.op = _OP_DMOVE;
+ pi._arg2 = i._arg0;
+ pi._arg3 = (unsigned char)i._arg1;
+ return;
+ }
+ break;
+ case _OP_LOAD:
+ if(pi.op == _OP_LOAD && i._arg1 < 256) {
+ pi.op = _OP_DLOAD;
+ pi._arg2 = i._arg0;
+ pi._arg3 = (unsigned char)i._arg1;
+ return;
+ }
+ break;
+ case _OP_EQ:case _OP_NE:
+ if(pi.op == _OP_LOAD && pi._arg0 == i._arg1 && (!IsLocal(pi._arg0) ))
+ {
+ pi.op = i.op;
+ pi._arg0 = i._arg0;
+ pi._arg1 = pi._arg1;
+ pi._arg2 = i._arg2;
+ pi._arg3 = MAX_FUNC_STACKSIZE;
+ return;
+ }
+ break;
+ case _OP_LOADNULLS:
+ if((pi.op == _OP_LOADNULLS && pi._arg0+pi._arg1 == i._arg0)) {
+
+ pi._arg1 = pi._arg1 + 1;
+ pi.op = _OP_LOADNULLS;
+ return;
+ }
+ break;
+ case _OP_LINE:
+ if(pi.op == _OP_LINE) {
+ _instructions.pop_back();
+ _lineinfos.pop_back();
+ }
+ break;
+ }
+ }
+ _optimization = true;
+ _instructions.push_back(i);
+}
+
+SQObject SQFuncState::CreateString(const SQChar *s,SQInteger len)
+{
+ SQObjectPtr ns(SQString::Create(_sharedstate,s,len));
+ _table(_strings)->NewSlot(ns,(SQInteger)1);
+ return ns;
+}
+
+SQFunctionProto *SQFuncState::BuildProto()
+{
+ SQFunctionProto *f=SQFunctionProto::Create();
+ f->_literals.resize(_nliterals);
+ SQObjectPtr refidx,key,val;
+ SQInteger idx;
+
+ f->_stacksize = _stacksize;
+ f->_sourcename = _sourcename;
+ f->_bgenerator = _bgenerator;
+ f->_name = _name;
+
+ while((idx=_table(_literals)->Next(false,refidx,key,val))!=-1) {
+ f->_literals[_integer(val)]=key;
+ refidx=idx;
+ }
+
+ f->_functions.resize(_functions.size());
+ f->_functions.copy(_functions);
+ f->_parameters.resize(_parameters.size());
+ f->_parameters.copy(_parameters);
+ f->_outervalues.resize(_outervalues.size());
+ f->_outervalues.copy(_outervalues);
+ f->_instructions.resize(_instructions.size());
+ f->_instructions.copy(_instructions);
+ f->_localvarinfos.resize(_localvarinfos.size());
+ f->_localvarinfos.copy(_localvarinfos);
+ f->_lineinfos.resize(_lineinfos.size());
+ f->_lineinfos.copy(_lineinfos);
+ f->_varparams = _varparams;
+
+ return f;
+}
+
+SQFuncState *SQFuncState::PushChildState(SQSharedState *ss)
+{
+ SQFuncState *child = (SQFuncState *)sq_malloc(sizeof(SQFuncState));
+ new (child) SQFuncState(ss,this,_errfunc,_errtarget);
+ _childstates.push_back(child);
+ return child;
+}
+
+void SQFuncState::PopChildState()
+{
+ SQFuncState *child = _childstates.back();
+ sq_delete(child,SQFuncState);
+ _childstates.pop_back();
+}
+
+SQFuncState::~SQFuncState()
+{
+ while(_childstates.size() > 0)
+ {
+ PopChildState();
+ }
+}