2 see copyright notice in squirrel.h
4 #include "sqpcheader.h"
5 #include "sqcompiler.h"
6 #include "sqfuncproto.h"
10 #include "sqfuncstate.h"
13 SQInstructionDesc g_InstrDesc[]={
16 {_SC("_OP_TAILCALL")},
18 {_SC("_OP_PREPCALL")},
19 {_SC("_OP_PREPCALLK")},
31 {_SC("_OP_LOADNULLS")},
32 {_SC("_OP_LOADROOTTABLE")},
37 {_SC("_OP_LOADFREEVAR")},
39 {_SC("_OP_GETVARGV")},
40 {_SC("_OP_NEWTABLE")},
41 {_SC("_OP_NEWARRAY")},
42 {_SC("_OP_APPENDARRAY")},
43 {_SC("_OP_GETPARENT")},
44 {_SC("_OP_COMPARITH")},
45 {_SC("_OP_COMPARITHL")},
52 {_SC("_OP_INSTANCEOF")},
62 {_SC("_OP_DELEGATE")},
65 {_SC("_OP_PUSHTRAP")},
69 {_SC("_OP_NEWSLOTA")},
73 void DumpLiteral(SQObjectPtr &o)
76 case OT_STRING: scprintf(_SC("\"%s\""),_stringval(o));break;
77 case OT_FLOAT: scprintf(_SC("{%f}"),_float(o));break;
78 case OT_INTEGER: scprintf(_SC("{%d}"),_integer(o));break;
83 SQFuncState::SQFuncState(SQSharedState *ss,SQFunctionProto *func,SQFuncState *parent)
86 _literals = SQTable::Create(ss,0);
99 void SQFuncState::Dump()
102 SQFunctionProto *func=_funcproto(_func);
103 scprintf(_SC("SQInstruction sizeof %d\n"),sizeof(SQInstruction));
104 scprintf(_SC("SQObject sizeof %d\n"),sizeof(SQObject));
105 scprintf(_SC("--------------------------------------------------------------------\n"));
106 scprintf(_SC("*****FUNCTION [%s]\n"),type(func->_name)==OT_STRING?_stringval(func->_name):_SC("unknown"));
107 scprintf(_SC("-----LITERALS\n"));
108 SQObjectPtr refidx,key,val;
110 SQObjectPtrVec templiterals;
111 templiterals.resize(_nliterals);
112 while((idx=_table(_literals)->Next(refidx,key,val))!=-1) {
114 templiterals[_integer(val)]=key;
116 for(i=0;i<templiterals.size();i++){
117 scprintf(_SC("[%d] "),n);
118 DumpLiteral(templiterals[i]);
122 scprintf(_SC("-----PARAMS\n"));
124 scprintf(_SC("<<VARPARAMS>>\n"));
126 for(i=0;i<_parameters.size();i++){
127 scprintf(_SC("[%d] "),n);
128 DumpLiteral(_parameters[i]);
132 scprintf(_SC("-----LOCALS\n"));
133 for(i=0;i<func->_localvarinfos.size();i++){
134 SQLocalVarInfo lvi=func->_localvarinfos[i];
135 scprintf(_SC("[%d] %s \t%d %d\n"),lvi._pos,_stringval(lvi._name),lvi._start_op,lvi._end_op);
138 scprintf(_SC("-----LINE INFO\n"));
139 for(i=0;i<_lineinfos.size();i++){
140 SQLineInfo li=_lineinfos[i];
141 scprintf(_SC("op [%d] line [%d] \n"),li._op,li._line);
144 scprintf(_SC("-----dump\n"));
146 for(i=0;i<_instructions.size();i++){
147 SQInstruction &inst=_instructions[i];
148 if(inst.op==_OP_LOAD || inst.op==_OP_PREPCALLK || inst.op==_OP_GETK ){
150 int lidx = inst._arg1;
151 scprintf(_SC("[%03d] %15s %d "),n,g_InstrDesc[inst.op].name,inst._arg0);
152 if(lidx >= 0xFFFFFFFF)
153 scprintf(_SC("null"));
156 SQObjectPtr val,key,refo;
157 while(((refidx=_table(_literals)->Next(refo,key,val))!= -1) && (_integer(val) != lidx)) {
162 scprintf(_SC(" %d %d \n"),inst._arg2,inst._arg3);
164 else if(inst.op==_OP_ARITH){
165 scprintf(_SC("[%03d] %15s %d %d %d %c\n"),n,g_InstrDesc[inst.op].name,inst._arg0,inst._arg1,inst._arg2,inst._arg3);
168 scprintf(_SC("[%03d] %15s %d %d %d %d\n"),n,g_InstrDesc[inst.op].name,inst._arg0,inst._arg1,inst._arg2,inst._arg3);
171 scprintf(_SC("-----\n"));
172 scprintf(_SC("stack size[%d]\n"),func->_stacksize);
173 scprintf(_SC("--------------------------------------------------------------------\n\n"));
176 int SQFuncState::GetStringConstant(const SQChar *cons)
178 return GetConstant(SQString::Create(_sharedstate,cons));
181 int SQFuncState::GetNumericConstant(const SQInteger cons)
183 return GetConstant(cons);
186 int SQFuncState::GetNumericConstant(const SQFloat cons)
188 return GetConstant(cons);
191 int SQFuncState::GetConstant(SQObjectPtr cons)
195 if(!_table(_literals)->Get(cons,val))
198 _table(_literals)->NewSlot(cons,val);
200 if(_nliterals > MAX_LITERALS) throw ParserException(_SC("internal compiler error: too many literals"));
202 return _integer(val);
205 void SQFuncState::SetIntructionParams(int pos,int arg0,int arg1,int arg2,int arg3)
207 _instructions[pos]._arg0=*((unsigned int *)&arg0);
208 _instructions[pos]._arg1=*((unsigned int *)&arg1);
209 _instructions[pos]._arg2=*((unsigned int *)&arg2);
210 _instructions[pos]._arg3=*((unsigned int *)&arg3);
213 void SQFuncState::SetIntructionParam(int pos,int arg,int val)
216 case 0:_instructions[pos]._arg0=*((unsigned int *)&val);break;
217 case 1:_instructions[pos]._arg1=*((unsigned int *)&val);break;
218 case 2:_instructions[pos]._arg2=*((unsigned int *)&val);break;
219 case 3:_instructions[pos]._arg3=*((unsigned int *)&val);break;
220 case 4:_instructions[pos]._arg1=*((unsigned int *)&val);break;
224 int SQFuncState::AllocStackPos()
226 int npos=_vlocals.size();
227 _vlocals.push_back(SQLocalVarInfo());
228 if(_vlocals.size()>((unsigned int)_stacksize)) {
229 if(_stacksize>MAX_FUNC_STACKSIZE) throw ParserException(_SC("internal compiler error: too many locals"));
230 _stacksize=_vlocals.size();
235 int SQFuncState::PushTarget(int n)
238 _targetstack.push_back(n);
242 _targetstack.push_back(n);
246 int SQFuncState::GetUpTarget(int n){
247 return _targetstack[((_targetstack.size()-1)-n)];
250 int SQFuncState::TopTarget(){
251 return _targetstack.back();
253 int SQFuncState::PopTarget()
255 int npos=_targetstack.back();
256 SQLocalVarInfo t=_vlocals[_targetstack.back()];
257 if(type(t._name)==OT_NULL){
260 _targetstack.pop_back();
264 int SQFuncState::GetStackSize()
266 return _vlocals.size();
269 void SQFuncState::SetStackSize(int n)
271 int size=_vlocals.size();
274 SQLocalVarInfo lvi=_vlocals.back();
275 if(type(lvi._name)!=OT_NULL){
276 lvi._end_op=GetCurrentPos();
277 _localvarinfos.push_back(lvi);
283 bool SQFuncState::IsLocal(unsigned int stkpos)
285 if(stkpos>=_vlocals.size())return false;
286 else if(type(_vlocals[stkpos]._name)!=OT_NULL)return true;
290 int SQFuncState::PushLocalVariable(const SQObjectPtr &name)
292 int pos=_vlocals.size();
295 lvi._start_op=GetCurrentPos()+1;
296 lvi._pos=_vlocals.size();
297 _vlocals.push_back(lvi);
298 if(_vlocals.size()>((unsigned int)_stacksize))_stacksize=_vlocals.size();
303 int SQFuncState::GetLocalVariable(const SQObjectPtr &name)
305 int locals=_vlocals.size();
307 if(type(_vlocals[locals-1]._name)==OT_STRING && _string(_vlocals[locals-1]._name)==_string(name)){
315 int SQFuncState::GetOuterVariable(const SQObjectPtr &name)
317 int outers = _outervalues.size();
318 for(int i = 0; i<outers; i++) {
319 if(_string(_outervalues[i]._name) == _string(name))
325 void SQFuncState::AddOuterValue(const SQObjectPtr &name)
327 //AddParameter(name);
330 pos = _parent->GetLocalVariable(name);
332 pos = _parent->GetOuterVariable(name);
334 _outervalues.push_back(SQOuterVar(name,SQObjectPtr(SQInteger(pos)),otOUTER)); //local
339 _outervalues.push_back(SQOuterVar(name,SQObjectPtr(SQInteger(pos)),otLOCAL)); //local
343 _outervalues.push_back(SQOuterVar(name,name,otSYMBOL)); //global
346 void SQFuncState::AddParameter(const SQObjectPtr &name)
348 PushLocalVariable(name);
349 _parameters.push_back(name);
352 void SQFuncState::AddLineInfos(int line,bool lineop,bool force)
354 if(_lastline!=line || force){
356 li._line=line;li._op=(GetCurrentPos()+1);
357 if(lineop)AddInstruction(_OP_LINE,0,line);
358 _lineinfos.push_back(li);
363 void SQFuncState::AddInstruction(SQInstruction &i)
365 int size = _instructions.size();
366 if(size > 0 && _optimization){ //simple optimizer
367 SQInstruction &pi = _instructions[size-1];//previous instruction
370 if( _parent && i._arg0 != MAX_FUNC_STACKSIZE && pi.op == _OP_CALL && _returnexp < size-1) {
371 pi.op = _OP_TAILCALL;
375 if( pi.op == _OP_LOAD && pi._arg0 == i._arg2 && (!IsLocal(pi._arg0))){
377 pi._arg2 = (unsigned char)i._arg1;
385 if( pi.op == _OP_LOAD && pi._arg0 == i._arg1 && (!IsLocal(pi._arg0))){
386 pi.op = _OP_PREPCALLK;
394 case _OP_APPENDARRAY:
395 if(pi.op == _OP_LOAD && pi._arg0 == i._arg1 && (!IsLocal(pi._arg0))){
396 pi.op = _OP_APPENDARRAY;
399 pi._arg2 = MAX_FUNC_STACKSIZE;
400 pi._arg3 = MAX_FUNC_STACKSIZE;
405 if((pi.op == _OP_GET || pi.op == _OP_ARITH || pi.op == _OP_BITW) && (pi._arg0 == i._arg1))
408 _optimization = false;
412 if(pi.op == _OP_MOVE)
416 pi._arg3 = (unsigned char)i._arg1;
421 case _OP_EQ:case _OP_NE:
422 if(pi.op == _OP_LOAD && pi._arg0 == i._arg1 && (!IsLocal(pi._arg0) ))
428 pi._arg3 = MAX_FUNC_STACKSIZE;
434 if((pi.op == _OP_LOADNULLS && pi._arg0+pi._arg1 == i._arg0)) {
436 pi._arg1 = pi._arg1 + 1;
437 pi.op = _OP_LOADNULLS;
442 if(pi.op == _OP_LINE) {
443 _instructions.pop_back();
444 _lineinfos.pop_back();
449 _optimization = true;
450 _instructions.push_back(i);
453 SQObject SQFuncState::CreateString(const SQChar *s)
455 SQObjectPtr ns(SQString::Create(_sharedstate,s));
456 _stringrefs.push_back(ns);
460 void SQFuncState::Finalize()
462 SQFunctionProto *f=_funcproto(_func);
463 f->_literals.resize(_nliterals);
464 SQObjectPtr refidx,key,val;
466 while((idx=_table(_literals)->Next(refidx,key,val))!=-1) {
467 f->_literals[_integer(val)]=key;
470 f->_functions.resize(_functions.size());
471 f->_functions.copy(_functions);
472 f->_parameters.resize(_parameters.size());
473 f->_parameters.copy(_parameters);
474 f->_outervalues.resize(_outervalues.size());
475 f->_outervalues.copy(_outervalues);
476 f->_instructions.resize(_instructions.size());
477 f->_instructions.copy(_instructions);
478 f->_localvarinfos.resize(_localvarinfos.size());
479 f->_localvarinfos.copy(_localvarinfos);
480 f->_lineinfos.resize(_lineinfos.size());
481 f->_lineinfos.copy(_lineinfos);
482 f->_varparams = _varparams;