2 see copyright notice in squirrel.h
\r
4 #include "sqpcheader.h"
\r
7 #include "sqopcodes.h"
\r
8 #include "sqstring.h"
\r
9 #include "sqfuncproto.h"
\r
10 #include "sqcompiler.h"
\r
11 #include "sqfuncstate.h"
\r
12 #include "sqlexer.h"
\r
15 #define DEREF_NO_DEREF -1
\r
16 #define DEREF_FIELD -2
\r
22 _deref = DEREF_NO_DEREF;
\r
24 _class_or_delete = false;
\r
27 bool _class_or_delete;
\r
33 typedef sqvector<ExpState> ExpStateVec;
\r
35 #define _exst (_expstates.top())
\r
37 #define BEGIN_BREAKBLE_BLOCK() SQInteger __nbreaks__=_fs->_unresolvedbreaks.size(); \
\r
38 SQInteger __ncontinues__=_fs->_unresolvedcontinues.size(); \
\r
39 _fs->_breaktargets.push_back(0);_fs->_continuetargets.push_back(0);
\r
41 #define END_BREAKBLE_BLOCK(continue_target) {__nbreaks__=_fs->_unresolvedbreaks.size()-__nbreaks__; \
\r
42 __ncontinues__=_fs->_unresolvedcontinues.size()-__ncontinues__; \
\r
43 if(__ncontinues__>0)ResolveContinues(_fs,__ncontinues__,continue_target); \
\r
44 if(__nbreaks__>0)ResolveBreaks(_fs,__nbreaks__); \
\r
45 _fs->_breaktargets.pop_back();_fs->_continuetargets.pop_back();}
\r
50 SQCompiler(SQVM *v, SQLEXREADFUNC rg, SQUserPointer up, const SQChar* sourcename, bool raiseerror, bool lineinfo)
\r
53 _lex.Init(_ss(v), rg, up,ThrowError,this);
\r
54 _sourcename = SQString::Create(_ss(v), sourcename);
\r
55 _lineinfo = lineinfo;_raiseerror = raiseerror;
\r
56 compilererror = NULL;
\r
58 static void ThrowError(void *ud, const SQChar *s) {
\r
59 SQCompiler *c = (SQCompiler *)ud;
\r
62 void Error(const SQChar *s, ...)
\r
64 static SQChar temp[256];
\r
67 scvsprintf(temp, s, vl);
\r
69 compilererror = temp;
\r
70 longjmp(_errorjmp,1);
\r
72 void Lex(){ _token = _lex.Lex();}
\r
73 void PushExpState(){ _expstates.push_back(ExpState()); }
\r
74 bool IsDerefToken(SQInteger tok)
\r
77 case _SC('='): case _SC('('): case TK_NEWSLOT:
\r
78 case TK_MODEQ: case TK_MULEQ: case TK_DIVEQ: case TK_MINUSEQ: case TK_PLUSEQ: case TK_PLUSPLUS: case TK_MINUSMINUS: return true;
\r
82 ExpState PopExpState()
\r
84 ExpState ret = _expstates.top();
\r
85 _expstates.pop_back();
\r
88 SQObject Expect(SQInteger tok)
\r
92 if(_token == TK_CONSTRUCTOR && tok == TK_IDENTIFIER) {
\r
93 //ret = SQString::Create(_ss(_vm),_SC("constructor"));
\r
97 const SQChar *etypename;
\r
101 case TK_IDENTIFIER:
\r
102 etypename = _SC("IDENTIFIER");
\r
104 case TK_STRING_LITERAL:
\r
105 etypename = _SC("STRING_LITERAL");
\r
108 etypename = _SC("INTEGER");
\r
111 etypename = _SC("FLOAT");
\r
114 etypename = _lex.Tok2Str(tok);
\r
116 Error(_SC("expected '%s'"), etypename);
\r
118 Error(_SC("expected '%c'"), tok);
\r
124 case TK_IDENTIFIER:
\r
125 ret = _fs->CreateString(_lex._svalue);
\r
127 case TK_STRING_LITERAL:
\r
128 ret = _fs->CreateString(_lex._svalue,_lex._longstr.size()-1);
\r
131 ret = SQObjectPtr(_lex._nvalue);
\r
134 ret = SQObjectPtr(_lex._fvalue);
\r
140 bool IsEndOfStatement() { return ((_lex._prevtoken == _SC('\n')) || (_token == SQUIRREL_EOB) || (_token == _SC('}')) || (_token == _SC(';'))); }
\r
141 void OptionalSemicolon()
\r
143 if(_token == _SC(';')) { Lex(); return; }
\r
144 if(!IsEndOfStatement()) {
\r
145 Error(_SC("end of statement expected (; or lf)"));
\r
148 void MoveIfCurrentTargetIsLocal() {
\r
149 SQInteger trg = _fs->TopTarget();
\r
150 if(_fs->IsLocal(trg)) {
\r
151 trg = _fs->PopTarget(); //no pops the target and move it
\r
152 _fs->AddInstruction(_OP_MOVE, _fs->PushTarget(), trg);
\r
155 bool Compile(SQObjectPtr &o)
\r
160 SQFuncState funcstate(_ss(_vm), NULL,ThrowError,this);
\r
161 funcstate._name = SQString::Create(_ss(_vm), _SC("main"));
\r
163 _fs->AddParameter(_fs->CreateString(_SC("this")));
\r
164 _fs->_sourcename = _sourcename;
\r
165 SQInteger stacksize = _fs->GetStackSize();
\r
166 if(setjmp(_errorjmp) == 0) {
\r
170 if(_lex._prevtoken != _SC('}')) OptionalSemicolon();
\r
172 CleanStack(stacksize);
\r
173 _fs->AddLineInfos(_lex._currentline, _lineinfo, true);
\r
174 _fs->AddInstruction(_OP_RETURN, 0xFF);
\r
175 _fs->SetStackSize(0);
\r
176 o =_fs->BuildProto();
\r
178 _fs->Dump(_funcproto(o));
\r
182 if(_raiseerror && _ss(_vm)->_compilererrorhandler) {
\r
183 _ss(_vm)->_compilererrorhandler(_vm, compilererror, type(_sourcename) == OT_STRING?_stringval(_sourcename):_SC("unknown"),
\r
184 _lex._currentline, _lex._currentcolumn);
\r
186 _vm->_lasterror = SQString::Create(_ss(_vm), compilererror, -1);
\r
193 while(_token != _SC('}') && _token != TK_DEFAULT && _token != TK_CASE) {
\r
195 if(_lex._prevtoken != _SC('}') && _lex._prevtoken != _SC(';')) OptionalSemicolon();
\r
200 _fs->AddLineInfos(_lex._currentline, _lineinfo);
\r
202 case _SC(';'): Lex(); break;
\r
203 case TK_IF: IfStatement(); break;
\r
204 case TK_WHILE: WhileStatement(); break;
\r
205 case TK_DO: DoWhileStatement(); break;
\r
206 case TK_FOR: ForStatement(); break;
\r
207 case TK_FOREACH: ForEachStatement(); break;
\r
208 case TK_SWITCH: SwitchStatement(); break;
\r
209 case TK_LOCAL: LocalDeclStatement(); break;
\r
213 if(_token == TK_RETURN) {
\r
219 _fs->_bgenerator = true;
\r
222 if(!IsEndOfStatement()) {
\r
223 SQInteger retexp = _fs->GetCurrentPos()+1;
\r
225 if(op == _OP_RETURN && _fs->_traps > 0)
\r
226 _fs->AddInstruction(_OP_POPTRAP, _fs->_traps, 0);
\r
227 _fs->_returnexp = retexp;
\r
228 _fs->AddInstruction(op, 1, _fs->PopTarget());
\r
231 if(op == _OP_RETURN && _fs->_traps > 0)
\r
232 _fs->AddInstruction(_OP_POPTRAP, _fs->_traps ,0);
\r
233 _fs->_returnexp = -1;
\r
234 _fs->AddInstruction(op, 0xFF);
\r
238 if(_fs->_breaktargets.size() <= 0)Error(_SC("'break' has to be in a loop block"));
\r
239 if(_fs->_breaktargets.top() > 0){
\r
240 _fs->AddInstruction(_OP_POPTRAP, _fs->_breaktargets.top(), 0);
\r
242 _fs->AddInstruction(_OP_JMP, 0, -1234);
\r
243 _fs->_unresolvedbreaks.push_back(_fs->GetCurrentPos());
\r
247 if(_fs->_continuetargets.size() <= 0)Error(_SC("'continue' has to be in a loop block"));
\r
248 if(_fs->_continuetargets.top() > 0) {
\r
249 _fs->AddInstruction(_OP_POPTRAP, _fs->_continuetargets.top(), 0);
\r
251 _fs->AddInstruction(_OP_JMP, 0, -1234);
\r
252 _fs->_unresolvedcontinues.push_back(_fs->GetCurrentPos());
\r
256 FunctionStatement();
\r
262 SQInteger stacksize = _fs->GetStackSize();
\r
266 _fs->SetStackSize(stacksize);
\r
270 TryCatchStatement();
\r
275 _fs->AddInstruction(_OP_THROW, _fs->PopTarget());
\r
284 void EmitDerefOp(SQOpcode op)
\r
286 SQInteger val = _fs->PopTarget();
\r
287 SQInteger key = _fs->PopTarget();
\r
288 SQInteger src = _fs->PopTarget();
\r
289 _fs->AddInstruction(op,_fs->PushTarget(),src,key,val);
\r
291 void Emit2ArgsOP(SQOpcode op, SQInteger p3 = 0)
\r
293 SQInteger p2 = _fs->PopTarget(); //src in OP_GET
\r
294 SQInteger p1 = _fs->PopTarget(); //key in OP_GET
\r
295 _fs->AddInstruction(op,_fs->PushTarget(), p1, p2, p3);
\r
297 void EmitCompoundArith(SQInteger tok,bool deref)
\r
301 case TK_MINUSEQ: oper = '-'; break;
\r
302 case TK_PLUSEQ: oper = '+'; break;
\r
303 case TK_MULEQ: oper = '*'; break;
\r
304 case TK_DIVEQ: oper = '/'; break;
\r
305 case TK_MODEQ: oper = '%'; break;
\r
306 default: assert(0); break;
\r
309 SQInteger val = _fs->PopTarget();
\r
310 SQInteger key = _fs->PopTarget();
\r
311 SQInteger src = _fs->PopTarget();
\r
312 //mixes dest obj and source val in the arg1(hack?)
\r
313 _fs->AddInstruction(_OP_COMPARITH,_fs->PushTarget(),(src<<16)|val,key,oper);
\r
316 Emit2ArgsOP(_OP_COMPARITHL, oper);
\r
321 for(Expression();_token == ',';_fs->PopTarget(), Lex(), CommaExpr());
\r
323 ExpState Expression(bool funcarg = false)
\r
326 _exst._class_or_delete = false;
\r
327 _exst._funcarg = funcarg;
\r
338 SQInteger op = _token;
\r
339 SQInteger ds = _exst._deref;
\r
340 bool freevar = _exst._freevar;
\r
341 if(ds == DEREF_NO_DEREF) Error(_SC("can't assign expression"));
\r
342 Lex(); Expression();
\r
346 if(freevar) Error(_SC("free variables cannot be modified"));
\r
347 if(ds == DEREF_FIELD)
\r
348 EmitDerefOp(_OP_NEWSLOT);
\r
349 else //if _derefstate != DEREF_NO_DEREF && DEREF_FIELD so is the index of a local
\r
350 Error(_SC("can't 'create' a local slot"));
\r
352 case _SC('='): //ASSIGN
\r
353 if(freevar) Error(_SC("free variables cannot be modified"));
\r
354 if(ds == DEREF_FIELD)
\r
355 EmitDerefOp(_OP_SET);
\r
356 else {//if _derefstate != DEREF_NO_DEREF && DEREF_FIELD so is the index of a local
\r
357 SQInteger p2 = _fs->PopTarget(); //src in OP_GET
\r
358 SQInteger p1 = _fs->TopTarget(); //key in OP_GET
\r
359 _fs->AddInstruction(_OP_MOVE, p1, p2);
\r
367 EmitCompoundArith(op,ds == DEREF_FIELD);
\r
374 _fs->AddInstruction(_OP_JZ, _fs->PopTarget());
\r
375 SQInteger jzpos = _fs->GetCurrentPos();
\r
376 SQInteger trg = _fs->PushTarget();
\r
378 SQInteger first_exp = _fs->PopTarget();
\r
379 if(trg != first_exp) _fs->AddInstruction(_OP_MOVE, trg, first_exp);
\r
380 SQInteger endfirstexp = _fs->GetCurrentPos();
\r
381 _fs->AddInstruction(_OP_JMP, 0, 0);
\r
383 SQInteger jmppos = _fs->GetCurrentPos();
\r
385 SQInteger second_exp = _fs->PopTarget();
\r
386 if(trg != second_exp) _fs->AddInstruction(_OP_MOVE, trg, second_exp);
\r
387 _fs->SetIntructionParam(jmppos, 1, _fs->GetCurrentPos() - jmppos);
\r
388 _fs->SetIntructionParam(jzpos, 1, endfirstexp - jzpos + 1);
\r
393 return PopExpState();
\r
395 void BIN_EXP(SQOpcode op, void (SQCompiler::*f)(void),SQInteger op3 = 0)
\r
397 Lex(); (this->*f)();
\r
398 SQInteger op1 = _fs->PopTarget();SQInteger op2 = _fs->PopTarget();
\r
399 _fs->AddInstruction(op, _fs->PushTarget(), op1, op2, op3);
\r
401 void LogicalOrExp()
\r
404 for(;;) if(_token == TK_OR) {
\r
405 SQInteger first_exp = _fs->PopTarget();
\r
406 SQInteger trg = _fs->PushTarget();
\r
407 _fs->AddInstruction(_OP_OR, trg, 0, first_exp, 0);
\r
408 SQInteger jpos = _fs->GetCurrentPos();
\r
409 if(trg != first_exp) _fs->AddInstruction(_OP_MOVE, trg, first_exp);
\r
410 Lex(); LogicalOrExp();
\r
412 SQInteger second_exp = _fs->PopTarget();
\r
413 if(trg != second_exp) _fs->AddInstruction(_OP_MOVE, trg, second_exp);
\r
415 _fs->SetIntructionParam(jpos, 1, (_fs->GetCurrentPos() - jpos));
\r
419 void LogicalAndExp()
\r
422 for(;;) switch(_token) {
\r
424 SQInteger first_exp = _fs->PopTarget();
\r
425 SQInteger trg = _fs->PushTarget();
\r
426 _fs->AddInstruction(_OP_AND, trg, 0, first_exp, 0);
\r
427 SQInteger jpos = _fs->GetCurrentPos();
\r
428 if(trg != first_exp) _fs->AddInstruction(_OP_MOVE, trg, first_exp);
\r
429 Lex(); LogicalAndExp();
\r
431 SQInteger second_exp = _fs->PopTarget();
\r
432 if(trg != second_exp) _fs->AddInstruction(_OP_MOVE, trg, second_exp);
\r
434 _fs->SetIntructionParam(jpos, 1, (_fs->GetCurrentPos() - jpos));
\r
437 case TK_IN: BIN_EXP(_OP_EXISTS, &SQCompiler::BitwiseOrExp); break;
\r
438 case TK_INSTANCEOF: BIN_EXP(_OP_INSTANCEOF, &SQCompiler::BitwiseOrExp); break;
\r
443 void BitwiseOrExp()
\r
446 for(;;) if(_token == _SC('|'))
\r
447 {BIN_EXP(_OP_BITW, &SQCompiler::BitwiseXorExp,BW_OR);
\r
450 void BitwiseXorExp()
\r
453 for(;;) if(_token == _SC('^'))
\r
454 {BIN_EXP(_OP_BITW, &SQCompiler::BitwiseAndExp,BW_XOR);
\r
457 void BitwiseAndExp()
\r
460 for(;;) if(_token == _SC('&'))
\r
461 {BIN_EXP(_OP_BITW, &SQCompiler::CompExp,BW_AND);
\r
467 for(;;) switch(_token) {
\r
468 case TK_EQ: BIN_EXP(_OP_EQ, &SQCompiler::ShiftExp); break;
\r
469 case _SC('>'): BIN_EXP(_OP_CMP, &SQCompiler::ShiftExp,CMP_G); break;
\r
470 case _SC('<'): BIN_EXP(_OP_CMP, &SQCompiler::ShiftExp,CMP_L); break;
\r
471 case TK_GE: BIN_EXP(_OP_CMP, &SQCompiler::ShiftExp,CMP_GE); break;
\r
472 case TK_LE: BIN_EXP(_OP_CMP, &SQCompiler::ShiftExp,CMP_LE); break;
\r
473 case TK_NE: BIN_EXP(_OP_NE, &SQCompiler::ShiftExp); break;
\r
480 for(;;) switch(_token) {
\r
481 case TK_USHIFTR: BIN_EXP(_OP_BITW, &SQCompiler::PlusExp,BW_USHIFTR); break;
\r
482 case TK_SHIFTL: BIN_EXP(_OP_BITW, &SQCompiler::PlusExp,BW_SHIFTL); break;
\r
483 case TK_SHIFTR: BIN_EXP(_OP_BITW, &SQCompiler::PlusExp,BW_SHIFTR); break;
\r
490 for(;;) switch(_token) {
\r
491 case _SC('+'): case _SC('-'):
\r
492 BIN_EXP(_OP_ARITH, &SQCompiler::MultExp,_token); break;
\r
500 for(;;) switch(_token) {
\r
501 case _SC('*'): case _SC('/'): case _SC('%'):
\r
502 BIN_EXP(_OP_ARITH, &SQCompiler::PrefixedExpr,_token); break;
\r
506 //if 'pos' != -1 the previous variable is a local variable
\r
507 void PrefixedExpr()
\r
509 SQInteger pos = Factor();
\r
515 if(_token == TK_PARENT) {
\r
518 Error(_SC("parent cannot be set"));
\r
519 SQInteger src = _fs->PopTarget();
\r
520 _fs->AddInstruction(_OP_GETPARENT, _fs->PushTarget(), src);
\r
523 _fs->AddInstruction(_OP_LOAD, _fs->PushTarget(), _fs->GetConstant(Expect(TK_IDENTIFIER)));
\r
524 if(NeedGet()) Emit2ArgsOP(_OP_GET);
\r
526 _exst._deref = DEREF_FIELD;
\r
527 _exst._freevar = false;
\r
531 if(_lex._prevtoken == _SC('\n')) Error(_SC("cannot brake deref/or comma needed after [exp]=exp slot declaration"));
\r
532 Lex(); Expression(); Expect(_SC(']'));
\r
534 if(NeedGet()) Emit2ArgsOP(_OP_GET);
\r
535 _exst._deref = DEREF_FIELD;
\r
536 _exst._freevar = false;
\r
538 case TK_MINUSMINUS:
\r
540 if(_exst._deref != DEREF_NO_DEREF && !IsEndOfStatement()) {
\r
541 SQInteger tok = _token; Lex();
\r
543 Emit2ArgsOP(_OP_PINC,tok == TK_MINUSMINUS?-1:1);
\r
544 else {//if _derefstate != DEREF_NO_DEREF && DEREF_FIELD so is the index of a local
\r
545 SQInteger src = _fs->PopTarget();
\r
546 _fs->AddInstruction(_OP_PINCL, _fs->PushTarget(), src, 0, tok == TK_MINUSMINUS?-1:1);
\r
554 if(_exst._deref != DEREF_NO_DEREF) {
\r
556 SQInteger key = _fs->PopTarget(); //key
\r
557 SQInteger table = _fs->PopTarget(); //table etc...
\r
558 SQInteger closure = _fs->PushTarget();
\r
559 SQInteger ttarget = _fs->PushTarget();
\r
560 _fs->AddInstruction(_OP_PREPCALL, closure, key, table, ttarget);
\r
563 _fs->AddInstruction(_OP_MOVE, _fs->PushTarget(), 0);
\r
567 _fs->AddInstruction(_OP_MOVE, _fs->PushTarget(), 0);
\r
568 _exst._deref = DEREF_NO_DEREF;
\r
570 FunctionCallArgs();
\r
581 case TK_STRING_LITERAL: {
\r
582 //SQObjectPtr id(SQString::Create(_ss(_vm), _lex._svalue,_lex._longstr.size()-1));
\r
583 _fs->AddInstruction(_OP_LOAD, _fs->PushTarget(), _fs->GetConstant(_fs->CreateString(_lex._svalue,_lex._longstr.size()-1)));
\r
587 case TK_VARGC: Lex(); _fs->AddInstruction(_OP_VARGC, _fs->PushTarget()); break;
\r
588 case TK_VARGV: { Lex();
\r
592 SQInteger src = _fs->PopTarget();
\r
593 _fs->AddInstruction(_OP_GETVARGV, _fs->PushTarget(), src);
\r
596 case TK_IDENTIFIER:
\r
597 case TK_CONSTRUCTOR:
\r
599 _exst._freevar = false;
\r
602 case TK_IDENTIFIER: id = _fs->CreateString(_lex._svalue); break;
\r
603 case TK_THIS: id = _fs->CreateString(_SC("this")); break;
\r
604 case TK_CONSTRUCTOR: id = _fs->CreateString(_SC("constructor")); break;
\r
606 SQInteger pos = -1;
\r
608 if((pos = _fs->GetLocalVariable(id)) == -1) {
\r
609 //checks if is a free variable
\r
610 if((pos = _fs->GetOuterVariable(id)) != -1) {
\r
611 _exst._deref = _fs->PushTarget();
\r
612 _fs->AddInstruction(_OP_LOADFREEVAR, _exst._deref ,pos);
\r
613 _exst._freevar = true;
\r
615 _fs->PushTarget(0);
\r
616 _fs->AddInstruction(_OP_LOAD, _fs->PushTarget(), _fs->GetConstant(id));
\r
617 if(NeedGet()) Emit2ArgsOP(_OP_GET);
\r
618 _exst._deref = DEREF_FIELD;
\r
622 _fs->PushTarget(pos);
\r
623 _exst._deref = pos;
\r
625 return _exst._deref;
\r
628 case TK_PARENT: Lex();_fs->AddInstruction(_OP_GETPARENT, _fs->PushTarget(), 0); break;
\r
629 case TK_DOUBLE_COLON: // "::"
\r
630 _fs->AddInstruction(_OP_LOADROOTTABLE, _fs->PushTarget());
\r
631 _exst._deref = DEREF_FIELD;
\r
632 _token = _SC('.'); //hack
\r
636 _fs->AddInstruction(_OP_LOADNULLS, _fs->PushTarget(),1);
\r
640 _fs->AddInstruction(_OP_LOAD, _fs->PushTarget(), _fs->GetNumericConstant(_lex._nvalue));
\r
644 _fs->AddInstruction(_OP_LOAD, _fs->PushTarget(), _fs->GetNumericConstant(_lex._fvalue));
\r
647 case TK_TRUE: case TK_FALSE:
\r
648 _fs->AddInstruction(_OP_LOADBOOL, _fs->PushTarget(),_token == TK_TRUE?1:0);
\r
652 _fs->AddInstruction(_OP_NEWARRAY, _fs->PushTarget());
\r
653 SQInteger apos = _fs->GetCurrentPos(),key = 0;
\r
655 while(_token != _SC(']')) {
\r
657 if(_token == _SC(',')) Lex();
\r
658 SQInteger val = _fs->PopTarget();
\r
659 SQInteger array = _fs->TopTarget();
\r
660 _fs->AddInstruction(_OP_APPENDARRAY, array, val);
\r
663 _fs->SetIntructionParam(apos, 1, key);
\r
668 _fs->AddInstruction(_OP_NEWTABLE, _fs->PushTarget());
\r
669 Lex();ParseTableOrClass(_SC(','));
\r
672 case TK_FUNCTION: FunctionExp(_token);break;
\r
673 case TK_CLASS: Lex(); ClassExp();break;
\r
674 case _SC('-'): UnaryOP(_OP_NEG); break;
\r
675 case _SC('!'): UnaryOP(_OP_NOT); break;
\r
676 case _SC('~'): UnaryOP(_OP_BWNOT); break;
\r
677 case TK_TYPEOF : UnaryOP(_OP_TYPEOF); break;
\r
678 case TK_RESUME : UnaryOP(_OP_RESUME); break;
\r
679 case TK_CLONE : UnaryOP(_OP_CLONE); break;
\r
680 case TK_MINUSMINUS :
\r
681 case TK_PLUSPLUS :PrefixIncDec(_token); break;
\r
682 case TK_DELETE : DeleteExpr(); break;
\r
683 case TK_DELEGATE : DelegateExpr(); break;
\r
684 case _SC('('): Lex(); CommaExpr(); Expect(_SC(')'));
\r
686 default: Error(_SC("expression expected"));
\r
690 void UnaryOP(SQOpcode op)
\r
692 Lex(); PrefixedExpr();
\r
693 SQInteger src = _fs->PopTarget();
\r
694 _fs->AddInstruction(op, _fs->PushTarget(), src);
\r
699 case _SC('='): case _SC('('): case TK_NEWSLOT: case TK_PLUSPLUS: case TK_MINUSMINUS:
\r
700 case TK_PLUSEQ: case TK_MINUSEQ: case TK_MULEQ: case TK_DIVEQ: case TK_MODEQ:
\r
703 return (!_exst._class_or_delete) || (_exst._class_or_delete && (_token == _SC('.') || _token == _SC('[')));
\r
706 void FunctionCallArgs()
\r
708 SQInteger nargs = 1;//this
\r
709 while(_token != _SC(')')) {
\r
711 MoveIfCurrentTargetIsLocal();
\r
713 if(_token == _SC(',')){
\r
715 if(_token == ')') Error(_SC("expression expected, found ')'"));
\r
719 for(SQInteger i = 0; i < (nargs - 1); i++) _fs->PopTarget();
\r
720 SQInteger stackbase = _fs->PopTarget();
\r
721 SQInteger closure = _fs->PopTarget();
\r
722 _fs->AddInstruction(_OP_CALL, _fs->PushTarget(), closure, stackbase, nargs);
\r
724 void ParseTableOrClass(SQInteger separator,SQInteger terminator = '}')
\r
726 SQInteger tpos = _fs->GetCurrentPos(),nkeys = 0;
\r
728 while(_token != terminator) {
\r
729 bool hasattrs = false;
\r
730 //check if is an attribute
\r
731 if(separator == ';' && _token == TK_ATTR_OPEN) {
\r
732 _fs->AddInstruction(_OP_NEWTABLE, _fs->PushTarget()); Lex();
\r
733 ParseTableOrClass(',',TK_ATTR_CLOSE);
\r
738 case TK_CONSTRUCTOR:{
\r
739 SQInteger tk = _token;
\r
741 SQObject id = tk == TK_FUNCTION ? Expect(TK_IDENTIFIER) : _fs->CreateString(_SC("constructor"));
\r
743 _fs->AddInstruction(_OP_LOAD, _fs->PushTarget(), _fs->GetConstant(id));
\r
744 CreateFunction(id);
\r
745 _fs->AddInstruction(_OP_CLOSURE, _fs->PushTarget(), _fs->_functions.size() - 1, 0);
\r
749 Lex(); CommaExpr(); Expect(_SC(']'));
\r
750 Expect(_SC('=')); Expression();
\r
753 _fs->AddInstruction(_OP_LOAD, _fs->PushTarget(), _fs->GetConstant(Expect(TK_IDENTIFIER)));
\r
754 Expect(_SC('=')); Expression();
\r
757 if(_token == separator) Lex();//optional comma/semicolon
\r
759 SQInteger val = _fs->PopTarget();
\r
760 SQInteger key = _fs->PopTarget();
\r
761 SQInteger attrs = hasattrs ? _fs->PopTarget():-1;
\r
762 assert(hasattrs && attrs == key-1 || !hasattrs);
\r
763 SQInteger table = _fs->TopTarget(); //<<BECAUSE OF THIS NO COMMON EMIT FUNC IS POSSIBLE
\r
764 _fs->AddInstruction(hasattrs?_OP_NEWSLOTA:_OP_NEWSLOT, _fs->PushTarget(), table, key, val);
\r
767 if(separator == _SC(',')) //hack recognizes a table from the separator
\r
768 _fs->SetIntructionParam(tpos, 1, nkeys);
\r
771 void LocalDeclStatement()
\r
775 Lex(); varname = Expect(TK_IDENTIFIER);
\r
776 if(_token == _SC('=')) {
\r
777 Lex(); Expression();
\r
778 SQInteger src = _fs->PopTarget();
\r
779 SQInteger dest = _fs->PushTarget();
\r
780 if(dest != src) _fs->AddInstruction(_OP_MOVE, dest, src);
\r
783 _fs->AddInstruction(_OP_LOADNULLS, _fs->PushTarget(),1);
\r
786 _fs->PushLocalVariable(varname);
\r
788 } while(_token == _SC(','));
\r
793 bool haselse = false;
\r
794 Lex(); Expect(_SC('(')); CommaExpr(); Expect(_SC(')'));
\r
795 _fs->AddInstruction(_OP_JZ, _fs->PopTarget());
\r
796 SQInteger jnepos = _fs->GetCurrentPos();
\r
797 SQInteger stacksize = _fs->GetStackSize();
\r
801 if(_token != _SC('}') && _token != TK_ELSE) OptionalSemicolon();
\r
803 CleanStack(stacksize);
\r
804 SQInteger endifblock = _fs->GetCurrentPos();
\r
805 if(_token == TK_ELSE){
\r
807 stacksize = _fs->GetStackSize();
\r
808 _fs->AddInstruction(_OP_JMP);
\r
809 jmppos = _fs->GetCurrentPos();
\r
811 Statement(); OptionalSemicolon();
\r
812 CleanStack(stacksize);
\r
813 _fs->SetIntructionParam(jmppos, 1, _fs->GetCurrentPos() - jmppos);
\r
815 _fs->SetIntructionParam(jnepos, 1, endifblock - jnepos + (haselse?1:0));
\r
817 void WhileStatement()
\r
819 SQInteger jzpos, jmppos;
\r
820 SQInteger stacksize = _fs->GetStackSize();
\r
821 jmppos = _fs->GetCurrentPos();
\r
822 Lex(); Expect(_SC('(')); CommaExpr(); Expect(_SC(')'));
\r
824 BEGIN_BREAKBLE_BLOCK();
\r
825 _fs->AddInstruction(_OP_JZ, _fs->PopTarget());
\r
826 jzpos = _fs->GetCurrentPos();
\r
827 stacksize = _fs->GetStackSize();
\r
831 CleanStack(stacksize);
\r
832 _fs->AddInstruction(_OP_JMP, 0, jmppos - _fs->GetCurrentPos() - 1);
\r
833 _fs->SetIntructionParam(jzpos, 1, _fs->GetCurrentPos() - jzpos);
\r
835 END_BREAKBLE_BLOCK(jmppos);
\r
837 void DoWhileStatement()
\r
840 SQInteger jzpos = _fs->GetCurrentPos();
\r
841 SQInteger stacksize = _fs->GetStackSize();
\r
842 BEGIN_BREAKBLE_BLOCK()
\r
844 CleanStack(stacksize);
\r
846 SQInteger continuetrg = _fs->GetCurrentPos();
\r
847 Expect(_SC('(')); CommaExpr(); Expect(_SC(')'));
\r
848 _fs->AddInstruction(_OP_JNZ, _fs->PopTarget(), jzpos - _fs->GetCurrentPos() - 1);
\r
849 END_BREAKBLE_BLOCK(continuetrg);
\r
851 void ForStatement()
\r
854 SQInteger stacksize = _fs->GetStackSize();
\r
856 if(_token == TK_LOCAL) LocalDeclStatement();
\r
857 else if(_token != _SC(';')){
\r
863 SQInteger jmppos = _fs->GetCurrentPos();
\r
864 SQInteger jzpos = -1;
\r
865 if(_token != _SC(';')) { CommaExpr(); _fs->AddInstruction(_OP_JZ, _fs->PopTarget()); jzpos = _fs->GetCurrentPos(); }
\r
868 SQInteger expstart = _fs->GetCurrentPos() + 1;
\r
869 if(_token != _SC(')')) {
\r
875 SQInteger expend = _fs->GetCurrentPos();
\r
876 SQInteger expsize = (expend - expstart) + 1;
\r
877 SQInstructionVec exp;
\r
879 for(SQInteger i = 0; i < expsize; i++)
\r
880 exp.push_back(_fs->GetInstruction(expstart + i));
\r
881 _fs->PopInstructions(expsize);
\r
883 BEGIN_BREAKBLE_BLOCK()
\r
885 SQInteger continuetrg = _fs->GetCurrentPos();
\r
887 for(SQInteger i = 0; i < expsize; i++)
\r
888 _fs->AddInstruction(exp[i]);
\r
890 _fs->AddInstruction(_OP_JMP, 0, jmppos - _fs->GetCurrentPos() - 1, 0);
\r
891 if(jzpos> 0) _fs->SetIntructionParam(jzpos, 1, _fs->GetCurrentPos() - jzpos);
\r
892 CleanStack(stacksize);
\r
894 END_BREAKBLE_BLOCK(continuetrg);
\r
896 void ForEachStatement()
\r
898 SQObject idxname, valname;
\r
899 Lex(); Expect(_SC('(')); valname = Expect(TK_IDENTIFIER);
\r
900 if(_token == _SC(',')) {
\r
902 Lex(); valname = Expect(TK_IDENTIFIER);
\r
905 idxname = _fs->CreateString(_SC("@INDEX@"));
\r
909 //save the stack size
\r
910 SQInteger stacksize = _fs->GetStackSize();
\r
911 //put the table in the stack(evaluate the table expression)
\r
912 Expression(); Expect(_SC(')'));
\r
913 SQInteger container = _fs->TopTarget();
\r
914 //push the index local var
\r
915 SQInteger indexpos = _fs->PushLocalVariable(idxname);
\r
916 _fs->AddInstruction(_OP_LOADNULLS, indexpos,1);
\r
917 //push the value local var
\r
918 SQInteger valuepos = _fs->PushLocalVariable(valname);
\r
919 _fs->AddInstruction(_OP_LOADNULLS, valuepos,1);
\r
920 //push reference index
\r
921 SQInteger itrpos = _fs->PushLocalVariable(_fs->CreateString(_SC("@ITERATOR@"))); //use invalid id to make it inaccessible
\r
922 _fs->AddInstruction(_OP_LOADNULLS, itrpos,1);
\r
923 SQInteger jmppos = _fs->GetCurrentPos();
\r
924 _fs->AddInstruction(_OP_FOREACH, container, 0, indexpos);
\r
925 SQInteger foreachpos = _fs->GetCurrentPos();
\r
926 //generate the statement code
\r
927 BEGIN_BREAKBLE_BLOCK()
\r
929 _fs->AddInstruction(_OP_JMP, 0, jmppos - _fs->GetCurrentPos() - 1);
\r
930 _fs->SetIntructionParam(foreachpos, 1, _fs->GetCurrentPos() - foreachpos);
\r
931 //restore the local variable stack(remove index,val and ref idx)
\r
932 CleanStack(stacksize);
\r
933 END_BREAKBLE_BLOCK(foreachpos - 1);
\r
935 void SwitchStatement()
\r
937 Lex(); Expect(_SC('(')); CommaExpr(); Expect(_SC(')'));
\r
939 SQInteger expr = _fs->TopTarget();
\r
940 bool bfirst = true;
\r
941 SQInteger tonextcondjmp = -1;
\r
942 SQInteger skipcondjmp = -1;
\r
943 SQInteger __nbreaks__ = _fs->_unresolvedbreaks.size();
\r
944 _fs->_breaktargets.push_back(0);
\r
945 while(_token == TK_CASE) {
\r
947 _fs->AddInstruction(_OP_JMP, 0, 0);
\r
948 skipcondjmp = _fs->GetCurrentPos();
\r
949 _fs->SetIntructionParam(tonextcondjmp, 1, _fs->GetCurrentPos() - tonextcondjmp);
\r
952 Lex(); Expression(); Expect(_SC(':'));
\r
953 SQInteger trg = _fs->PopTarget();
\r
954 _fs->AddInstruction(_OP_EQ, trg, trg, expr);
\r
955 _fs->AddInstruction(_OP_JZ, trg, 0);
\r
957 if(skipcondjmp != -1) {
\r
958 _fs->SetIntructionParam(skipcondjmp, 1, (_fs->GetCurrentPos() - skipcondjmp));
\r
960 tonextcondjmp = _fs->GetCurrentPos();
\r
961 SQInteger stacksize = _fs->GetStackSize();
\r
963 _fs->SetStackSize(stacksize);
\r
966 if(tonextcondjmp != -1)
\r
967 _fs->SetIntructionParam(tonextcondjmp, 1, _fs->GetCurrentPos() - tonextcondjmp);
\r
968 if(_token == TK_DEFAULT) {
\r
969 Lex(); Expect(_SC(':'));
\r
970 SQInteger stacksize = _fs->GetStackSize();
\r
972 _fs->SetStackSize(stacksize);
\r
976 __nbreaks__ = _fs->_unresolvedbreaks.size() - __nbreaks__;
\r
977 if(__nbreaks__ > 0)ResolveBreaks(_fs, __nbreaks__);
\r
978 _fs->_breaktargets.pop_back();
\r
981 void FunctionStatement()
\r
984 Lex(); id = Expect(TK_IDENTIFIER);
\r
985 _fs->PushTarget(0);
\r
986 _fs->AddInstruction(_OP_LOAD, _fs->PushTarget(), _fs->GetConstant(id));
\r
987 if(_token == TK_DOUBLE_COLON) Emit2ArgsOP(_OP_GET);
\r
989 while(_token == TK_DOUBLE_COLON) {
\r
991 id = Expect(TK_IDENTIFIER);
\r
992 _fs->AddInstruction(_OP_LOAD, _fs->PushTarget(), _fs->GetConstant(id));
\r
993 if(_token == TK_DOUBLE_COLON) Emit2ArgsOP(_OP_GET);
\r
996 CreateFunction(id);
\r
997 _fs->AddInstruction(_OP_CLOSURE, _fs->PushTarget(), _fs->_functions.size() - 1, 0);
\r
998 EmitDerefOp(_OP_NEWSLOT);
\r
1001 void ClassStatement()
\r
1004 Lex(); PushExpState();
\r
1005 _exst._class_or_delete = true;
\r
1006 _exst._funcarg = false;
\r
1008 es = PopExpState();
\r
1009 if(es._deref == DEREF_NO_DEREF) Error(_SC("invalid class name"));
\r
1010 if(es._deref == DEREF_FIELD) {
\r
1012 EmitDerefOp(_OP_NEWSLOT);
\r
1015 else Error(_SC("cannot create a class in a local with the syntax(class <local>)"));
\r
1017 void TryCatchStatement()
\r
1021 _fs->AddInstruction(_OP_PUSHTRAP,0,0);
\r
1023 if(_fs->_breaktargets.size()) _fs->_breaktargets.top()++;
\r
1024 if(_fs->_continuetargets.size()) _fs->_continuetargets.top()++;
\r
1025 SQInteger trappos = _fs->GetCurrentPos();
\r
1028 _fs->AddInstruction(_OP_POPTRAP, 1, 0);
\r
1029 if(_fs->_breaktargets.size()) _fs->_breaktargets.top()--;
\r
1030 if(_fs->_continuetargets.size()) _fs->_continuetargets.top()--;
\r
1031 _fs->AddInstruction(_OP_JMP, 0, 0);
\r
1032 SQInteger jmppos = _fs->GetCurrentPos();
\r
1033 _fs->SetIntructionParam(trappos, 1, (_fs->GetCurrentPos() - trappos));
\r
1034 Expect(TK_CATCH); Expect(_SC('(')); exid = Expect(TK_IDENTIFIER); Expect(_SC(')'));
\r
1035 SQInteger stacksize = _fs->GetStackSize();
\r
1036 SQInteger ex_target = _fs->PushLocalVariable(exid);
\r
1037 _fs->SetIntructionParam(trappos, 0, ex_target);
\r
1039 _fs->SetIntructionParams(jmppos, 0, (_fs->GetCurrentPos() - jmppos), 0);
\r
1040 CleanStack(stacksize);
\r
1042 void FunctionExp(SQInteger ftype)
\r
1044 Lex(); Expect(_SC('('));
\r
1045 CreateFunction(_null_);
\r
1046 _fs->AddInstruction(_OP_CLOSURE, _fs->PushTarget(), _fs->_functions.size() - 1, ftype == TK_FUNCTION?0:1);
\r
1050 SQInteger base = -1;
\r
1051 SQInteger attrs = -1;
\r
1052 if(_token == TK_EXTENDS) {
\r
1053 Lex(); Expression();
\r
1054 base = _fs->TopTarget();
\r
1056 if(_token == TK_ATTR_OPEN) {
\r
1058 _fs->AddInstruction(_OP_NEWTABLE, _fs->PushTarget());
\r
1059 ParseTableOrClass(_SC(','),TK_ATTR_CLOSE);
\r
1060 attrs = _fs->TopTarget();
\r
1063 if(attrs != -1) _fs->PopTarget();
\r
1064 if(base != -1) _fs->PopTarget();
\r
1065 _fs->AddInstruction(_OP_CLASS, _fs->PushTarget(), base, attrs);
\r
1066 ParseTableOrClass(_SC(';'));
\r
1068 void DelegateExpr()
\r
1070 Lex(); CommaExpr();
\r
1073 SQInteger table = _fs->PopTarget(), delegate = _fs->PopTarget();
\r
1074 _fs->AddInstruction(_OP_DELEGATE, _fs->PushTarget(), table, delegate);
\r
1079 Lex(); PushExpState();
\r
1080 _exst._class_or_delete = true;
\r
1081 _exst._funcarg = false;
\r
1083 es = PopExpState();
\r
1084 if(es._deref == DEREF_NO_DEREF) Error(_SC("can't delete an expression"));
\r
1085 if(es._deref == DEREF_FIELD) Emit2ArgsOP(_OP_DELETE);
\r
1086 else Error(_SC("cannot delete a local"));
\r
1088 void PrefixIncDec(SQInteger token)
\r
1091 Lex(); PushExpState();
\r
1092 _exst._class_or_delete = true;
\r
1093 _exst._funcarg = false;
\r
1095 es = PopExpState();
\r
1096 if(es._deref == DEREF_FIELD) Emit2ArgsOP(_OP_INC,token == TK_MINUSMINUS?-1:1);
\r
1098 SQInteger src = _fs->PopTarget();
\r
1099 _fs->AddInstruction(_OP_INCL, _fs->PushTarget(), src, 0, token == TK_MINUSMINUS?-1:1);
\r
1102 void CreateFunction(SQObject &name)
\r
1105 SQFuncState *funcstate = _fs->PushChildState(_ss(_vm));
\r
1106 funcstate->_name = name;
\r
1107 SQObject paramname;
\r
1108 funcstate->AddParameter(_fs->CreateString(_SC("this")));
\r
1109 funcstate->_sourcename = _sourcename;
\r
1110 while(_token!=_SC(')')) {
\r
1111 if(_token == TK_VARPARAMS) {
\r
1112 funcstate->_varparams = true;
\r
1114 if(_token != _SC(')')) Error(_SC("expected ')'"));
\r
1118 paramname = Expect(TK_IDENTIFIER);
\r
1119 funcstate->AddParameter(paramname);
\r
1120 if(_token == _SC(',')) Lex();
\r
1121 else if(_token != _SC(')')) Error(_SC("expected ')' or ','"));
\r
1126 if(_token == _SC(':')) {
\r
1127 Lex(); Expect(_SC('('));
\r
1128 while(_token != _SC(')')) {
\r
1129 paramname = Expect(TK_IDENTIFIER);
\r
1130 //outers are treated as implicit local variables
\r
1131 funcstate->AddOuterValue(paramname);
\r
1132 if(_token == _SC(',')) Lex();
\r
1133 else if(_token != _SC(')')) Error(_SC("expected ')' or ','"));
\r
1138 SQFuncState *currchunk = _fs;
\r
1141 funcstate->AddLineInfos(_lex._prevtoken == _SC('\n')?_lex._lasttokenline:_lex._currentline, _lineinfo, true);
\r
1142 funcstate->AddInstruction(_OP_RETURN, -1);
\r
1143 funcstate->SetStackSize(0);
\r
1144 //_fs->->_stacksize = _fs->_stacksize;
\r
1145 SQFunctionProto *func = funcstate->BuildProto();
\r
1146 #ifdef _DEBUG_DUMP
\r
1147 funcstate->Dump(func);
\r
1150 _fs->_functions.push_back(func);
\r
1151 _fs->PopChildState();
\r
1153 void CleanStack(SQInteger stacksize)
\r
1155 if(_fs->GetStackSize() != stacksize)
\r
1156 _fs->SetStackSize(stacksize);
\r
1158 void ResolveBreaks(SQFuncState *funcstate, SQInteger ntoresolve)
\r
1160 while(ntoresolve > 0) {
\r
1161 SQInteger pos = funcstate->_unresolvedbreaks.back();
\r
1162 funcstate->_unresolvedbreaks.pop_back();
\r
1163 //set the jmp instruction
\r
1164 funcstate->SetIntructionParams(pos, 0, funcstate->GetCurrentPos() - pos, 0);
\r
1168 void ResolveContinues(SQFuncState *funcstate, SQInteger ntoresolve, SQInteger targetpos)
\r
1170 while(ntoresolve > 0) {
\r
1171 SQInteger pos = funcstate->_unresolvedcontinues.back();
\r
1172 funcstate->_unresolvedcontinues.pop_back();
\r
1173 //set the jmp instruction
\r
1174 funcstate->SetIntructionParams(pos, 0, targetpos - pos, 0);
\r
1181 SQObjectPtr _sourcename;
\r
1185 SQInteger _debugline;
\r
1186 SQInteger _debugop;
\r
1187 ExpStateVec _expstates;
\r
1188 SQChar *compilererror;
\r
1189 jmp_buf _errorjmp;
\r
1193 bool Compile(SQVM *vm,SQLEXREADFUNC rg, SQUserPointer up, const SQChar *sourcename, SQObjectPtr &out, bool raiseerror, bool lineinfo)
\r
1195 SQCompiler p(vm, rg, up, sourcename, raiseerror, lineinfo);
\r
1196 return p.Compile(out);
\r