123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782 |
- //-----------------------------------------------------------------------------
- // Copyright (c) 2013 GarageGames, LLC
- //
- // Permission is hereby granted, free of charge, to any person obtaining a copy
- // of this software and associated documentation files (the "Software"), to
- // deal in the Software without restriction, including without limitation the
- // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
- // sell copies of the Software, and to permit persons to whom the Software is
- // furnished to do so, subject to the following conditions:
- //
- // The above copyright notice and this permission notice shall be included in
- // all copies or substantial portions of the Software.
- //
- // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
- // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
- // IN THE SOFTWARE.
- //-----------------------------------------------------------------------------
- #include "platform/platform.h"
- #include "console/console.h"
- #include "platform/event.h"
- #include "console/ast.h"
- #include "collection/findIterator.h"
- #include "io/resource/resourceManager.h"
- #include "string/findMatch.h"
- #include "console/consoleInternal.h"
- #include "io/fileStream.h"
- #include "console/compiler.h"
- #include "sim/simBase.h"
- // This enables extra breakpoint lines in the DSO output at the
- // exit points of functions as well as on code outside of functions
- // in the main body of a script file. In practice the only overhead
- // is the one extra NOP per function regardless of size. This #define
- // is here for those that want to disable this.
- #define TORQUE_EXTRA_BREAKLINES
- namespace Compiler
- {
- U32 precompileBlock(StmtNode *block, U32 loopCount)
- {
- U32 sum = 0;
- for(StmtNode *walk = block; walk; walk = walk->getNext())
- sum += walk->precompileStmt(loopCount);
- return sum;
- }
- U32 compileBlock(StmtNode *block, U32 *codeStream, U32 ip, U32 continuePoint, U32 breakPoint)
- {
- for(StmtNode *walk = block; walk; walk = walk->getNext())
- ip = walk->compileStmt(codeStream, ip, continuePoint, breakPoint);
- return ip;
- }
- };
- using namespace Compiler;
- //-----------------------------------------------------------------------------
- void StmtNode::addBreakCount()
- {
- #ifndef TORQUE_EXTRA_BREAKLINES
- if(CodeBlock::smInFunction)
- #endif
- CodeBlock::smBreakLineCount++;
- }
- void StmtNode::addBreakLine(U32 ip)
- {
- #ifndef TORQUE_EXTRA_BREAKLINES
- if(CodeBlock::smInFunction)
- {
- #endif
- U32 line = CodeBlock::smBreakLineCount * 2;
- CodeBlock::smBreakLineCount++;
- if(getBreakCodeBlock()->lineBreakPairs)
- {
- getBreakCodeBlock()->lineBreakPairs[line] = dbgLineNumber;
- getBreakCodeBlock()->lineBreakPairs[line+1] = ip;
- }
- #ifndef TORQUE_EXTRA_BREAKLINES
- }
- #endif
- }
- //------------------------------------------------------------
- StmtNode::StmtNode()
- {
- next = NULL;
- dbgFileName = CodeBlock::smCurrentParser->getCurrentFile();
- dbgLineNumber = CodeBlock::smCurrentParser->getCurrentLine();
- }
- void StmtNode::setPackage(StringTableEntry)
- {
- }
- void StmtNode::append(StmtNode *next)
- {
- StmtNode *walk = this;
- while(walk->next)
- walk = walk->next;
- walk->next = next;
- }
- void FunctionDeclStmtNode::setPackage(StringTableEntry packageName)
- {
- package = packageName;
- }
- //------------------------------------------------------------
- //
- // Console language compilers
- //
- //------------------------------------------------------------
- static U32 conversionOp(TypeReq src, TypeReq dst)
- {
- if(src == TypeReqString)
- {
- switch(dst)
- {
- case TypeReqUInt:
- return OP_STR_TO_UINT;
- case TypeReqFloat:
- return OP_STR_TO_FLT;
- case TypeReqNone:
- return OP_STR_TO_NONE;
- default:
- break;
- }
- }
- else if(src == TypeReqFloat)
- {
- switch(dst)
- {
- case TypeReqUInt:
- return OP_FLT_TO_UINT;
- case TypeReqString:
- return OP_FLT_TO_STR;
- case TypeReqNone:
- return OP_FLT_TO_NONE;
- default:
- break;
- }
- }
- else if(src == TypeReqUInt)
- {
- switch(dst)
- {
- case TypeReqFloat:
- return OP_UINT_TO_FLT;
- case TypeReqString:
- return OP_UINT_TO_STR;
- case TypeReqNone:
- return OP_UINT_TO_NONE;
- default:
- break;
- }
- }
- return OP_INVALID;
- }
- //------------------------------------------------------------
- U32 BreakStmtNode::precompileStmt(U32 loopCount)
- {
- if(loopCount)
- {
- addBreakCount();
- return 2;
- }
- Con::warnf(ConsoleLogEntry::General, "%s (%d): break outside of loop... ignoring.", dbgFileName, dbgLineNumber);
- return 0;
- }
- U32 BreakStmtNode::compileStmt(U32 *codeStream, U32 ip, U32, U32 breakPoint)
- {
- if(breakPoint)
- {
- addBreakLine(ip);
- codeStream[ip++] = OP_JMP;
- codeStream[ip++] = breakPoint;
- }
- return ip;
- }
- //------------------------------------------------------------
- U32 ContinueStmtNode::precompileStmt(U32 loopCount)
- {
- if(loopCount)
- {
- addBreakCount();
- return 2;
- }
- Con::warnf(ConsoleLogEntry::General, "%s (%d): continue outside of loop... ignoring.", dbgFileName, dbgLineNumber);
- return 0;
- }
- U32 ContinueStmtNode::compileStmt(U32 *codeStream, U32 ip, U32 continuePoint, U32)
- {
- if(continuePoint)
- {
- addBreakLine(ip);
- codeStream[ip++] = OP_JMP;
- codeStream[ip++] = continuePoint;
- }
- return ip;
- }
- //------------------------------------------------------------
- U32 ExprNode::precompileStmt(U32)
- {
- addBreakCount();
- return precompile(TypeReqNone);
- }
- U32 ExprNode::compileStmt(U32 *codeStream, U32 ip, U32, U32)
- {
- addBreakLine(ip);
- return compile(codeStream, ip, TypeReqNone);
- }
- //------------------------------------------------------------
- U32 ReturnStmtNode::precompileStmt(U32)
- {
- addBreakCount();
- if(!expr)
- return 1;
- else
- return 1 + expr->precompile(TypeReqString);
- }
- U32 ReturnStmtNode::compileStmt(U32 *codeStream, U32 ip, U32, U32)
- {
- addBreakLine(ip);
- if(!expr)
- codeStream[ip++] = OP_RETURN;
- else
- {
- ip = expr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_RETURN;
- }
- return ip;
- }
- //------------------------------------------------------------
- ExprNode *IfStmtNode::getSwitchOR(ExprNode *left, ExprNode *list, bool string)
- {
- ExprNode *nextExpr = (ExprNode *) list->getNext();
- ExprNode *test;
- if(string)
- test = StreqExprNode::alloc(left, list, true);
- else
- test = IntBinaryExprNode::alloc(opEQ, left, list);
- if(!nextExpr)
- return test;
- return IntBinaryExprNode::alloc(opOR, test, getSwitchOR(left, nextExpr, string));
- }
- void IfStmtNode::propagateSwitchExpr(ExprNode *left, bool string)
- {
- testExpr = getSwitchOR(left, testExpr, string);
- if(propagate && elseBlock)
- ((IfStmtNode *) elseBlock)->propagateSwitchExpr(left, string);
- }
- U32 IfStmtNode::precompileStmt(U32 loopCount)
- {
- U32 exprSize;
- addBreakCount();
- if(testExpr->getPreferredType() == TypeReqUInt)
- {
- exprSize = testExpr->precompile(TypeReqUInt);
- integer = true;
- }
- else
- {
- exprSize = testExpr->precompile(TypeReqFloat);
- integer = false;
- }
- // next is the JMPIFNOT or JMPIFFNOT - size of 2
- U32 ifSize = precompileBlock(ifBlock, loopCount);
- if(!elseBlock)
- endifOffset = ifSize + 2 + exprSize;
- else
- {
- elseOffset = exprSize + 2 + ifSize + 2;
- U32 elseSize = precompileBlock(elseBlock, loopCount);
- endifOffset = elseOffset + elseSize;
- }
- return endifOffset;
- }
- U32 IfStmtNode::compileStmt(U32 *codeStream, U32 ip, U32 continuePoint, U32 breakPoint)
- {
- U32 start = ip;
- addBreakLine(ip);
- ip = testExpr->compile(codeStream, ip, integer ? TypeReqUInt : TypeReqFloat);
- codeStream[ip++] = integer ? OP_JMPIFNOT : OP_JMPIFFNOT;
- if(elseBlock)
- {
- codeStream[ip++] = start + elseOffset;
- ip = compileBlock(ifBlock, codeStream, ip, continuePoint, breakPoint);
- codeStream[ip++] = OP_JMP;
- codeStream[ip++] = start + endifOffset;
- ip = compileBlock(elseBlock, codeStream, ip, continuePoint, breakPoint);
- }
- else
- {
- codeStream[ip++] = start + endifOffset;
- ip = compileBlock(ifBlock, codeStream, ip, continuePoint, breakPoint);
- }
- return ip;
- }
- //------------------------------------------------------------
- U32 LoopStmtNode::precompileStmt(U32 loopCount)
- {
- U32 initSize = 0;
- addBreakCount();
- if(initExpr)
- initSize = initExpr->precompile(TypeReqNone);
- U32 testSize;
- if(testExpr->getPreferredType() == TypeReqUInt)
- {
- integer = true;
- testSize = testExpr->precompile(TypeReqUInt);
- }
- else
- {
- integer = false;
- testSize = testExpr->precompile(TypeReqFloat);
- }
- U32 blockSize = precompileBlock(loopBlock, loopCount + 1);
- U32 endLoopSize = 0;
- if(endLoopExpr)
- endLoopSize = endLoopExpr->precompile(TypeReqNone);
- // if it's a for loop or a while loop it goes:
- // initExpr
- // testExpr
- // OP_JMPIFNOT to break point
- // loopStartPoint:
- // loopBlock
- // continuePoint:
- // endLoopExpr
- // testExpr
- // OP_JMPIF loopStartPoint
- // breakPoint:
- // otherwise if it's a do ... while() it goes:
- // initExpr
- // loopStartPoint:
- // loopBlock
- // continuePoint:
- // endLoopExpr
- // testExpr
- // OP_JMPIF loopStartPoint
- // breakPoint:
- if(!isDoLoop)
- {
- loopBlockStartOffset = initSize + testSize + 2;
- continueOffset = loopBlockStartOffset + blockSize;
- breakOffset = continueOffset + endLoopSize + testSize + 2;
- }
- else
- {
- loopBlockStartOffset = initSize;
- continueOffset = initSize + blockSize;
- breakOffset = continueOffset + endLoopSize + testSize + 2;
- }
- return breakOffset;
- }
- U32 LoopStmtNode::compileStmt(U32 *codeStream, U32 ip, U32, U32)
- {
- addBreakLine(ip);
- U32 start = ip;
- if(initExpr)
- ip = initExpr->compile(codeStream, ip, TypeReqNone);
- if(!isDoLoop)
- {
- ip = testExpr->compile(codeStream, ip, integer ? TypeReqUInt : TypeReqFloat);
- codeStream[ip++] = integer ? OP_JMPIFNOT : OP_JMPIFFNOT;
- codeStream[ip++] = start + breakOffset;
- }
- // Compile internals of loop.
- ip = compileBlock(loopBlock, codeStream, ip, start + continueOffset, start + breakOffset);
- if(endLoopExpr)
- ip = endLoopExpr->compile(codeStream, ip, TypeReqNone);
- ip = testExpr->compile(codeStream, ip, integer ? TypeReqUInt : TypeReqFloat);
- codeStream[ip++] = integer ? OP_JMPIF : OP_JMPIFF;
- codeStream[ip++] = start + loopBlockStartOffset;
- return ip;
- }
- //------------------------------------------------------------
- U32 ConditionalExprNode::precompile(TypeReq type)
- {
- // code is testExpr
- // JMPIFNOT falseStart
- // trueExpr
- // JMP end
- // falseExpr
- U32 exprSize;
- if(testExpr->getPreferredType() == TypeReqUInt)
- {
- exprSize = testExpr->precompile(TypeReqUInt);
- integer = true;
- }
- else
- {
- exprSize = testExpr->precompile(TypeReqFloat);
- integer = false;
- }
- return exprSize +
- trueExpr->precompile(type) +
- falseExpr->precompile(type) + 4;
- }
- U32 ConditionalExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = testExpr->compile(codeStream, ip, integer ? TypeReqUInt : TypeReqFloat);
- codeStream[ip++] = integer ? OP_JMPIFNOT : OP_JMPIFFNOT;
- U32 jumpElseIp = ip++;
- ip = trueExpr->compile(codeStream, ip, type);
- codeStream[ip++] = OP_JMP;
- U32 jumpEndIp = ip++;
- codeStream[jumpElseIp] = ip;
- ip = falseExpr->compile(codeStream, ip, type);
- codeStream[jumpEndIp] = ip;
- return ip;
- }
- TypeReq ConditionalExprNode::getPreferredType()
- {
- return trueExpr->getPreferredType();
- }
- //------------------------------------------------------------
- U32 FloatBinaryExprNode::precompile(TypeReq type)
- {
- U32 addSize = left->precompile(TypeReqFloat) + right->precompile(TypeReqFloat) + 1;
- if(type != TypeReqFloat)
- addSize++;
- return addSize;
- }
- U32 FloatBinaryExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = right->compile(codeStream, ip, TypeReqFloat);
- ip = left->compile(codeStream, ip, TypeReqFloat);
- U32 operand = OP_INVALID;
- switch(op)
- {
- case '+':
- operand = OP_ADD;
- break;
- case '-':
- operand = OP_SUB;
- break;
- case '/':
- operand = OP_DIV;
- break;
- case '*':
- operand = OP_MUL;
- break;
- }
- codeStream[ip++] = operand;
- if(type != TypeReqFloat)
- codeStream[ip++] =conversionOp(TypeReqFloat, type);
- return ip;
- }
- TypeReq FloatBinaryExprNode::getPreferredType()
- {
- return TypeReqFloat;
- }
- //------------------------------------------------------------
- void IntBinaryExprNode::getSubTypeOperand()
- {
- subType = TypeReqUInt;
- switch(op)
- {
- case '^':
- operand = OP_XOR;
- break;
- case '%':
- operand = OP_MOD;
- break;
- case '&':
- operand = OP_BITAND;
- break;
- case '|':
- operand = OP_BITOR;
- break;
- case '<':
- operand = OP_CMPLT;
- subType = TypeReqFloat;
- break;
- case '>':
- operand = OP_CMPGR;
- subType = TypeReqFloat;
- break;
- case opGE:
- operand = OP_CMPGE;
- subType = TypeReqFloat;
- break;
- case opLE:
- operand = OP_CMPLE;
- subType = TypeReqFloat;
- break;
- case opEQ:
- operand = OP_CMPEQ;
- subType = TypeReqFloat;
- break;
- case opNE:
- operand = OP_CMPNE;
- subType = TypeReqFloat;
- break;
- case opOR:
- operand = OP_OR;
- break;
- case opAND:
- operand = OP_AND;
- break;
- case opSHR:
- operand = OP_SHR;
- break;
- case opSHL:
- operand = OP_SHL;
- break;
- }
- }
- U32 IntBinaryExprNode::precompile(TypeReq type)
- {
- getSubTypeOperand();
- U32 addSize = left->precompile(subType) + right->precompile(subType) + 1;
- if(operand == OP_OR || operand == OP_AND)
- addSize++;
- if(type != TypeReqUInt)
- addSize++;
- return addSize;
- }
- U32 IntBinaryExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- if(operand == OP_OR || operand == OP_AND)
- {
- ip = left->compile(codeStream, ip, subType);
- codeStream[ip++] = operand == OP_OR ? OP_JMPIF_NP : OP_JMPIFNOT_NP;
- U32 jmpIp = ip++;
- ip = right->compile(codeStream, ip, subType);
- codeStream[jmpIp] = ip;
- }
- else
- {
- ip = right->compile(codeStream, ip, subType);
- ip = left->compile(codeStream, ip, subType);
- codeStream[ip++] = operand;
- }
- if(type != TypeReqUInt)
- codeStream[ip++] =conversionOp(TypeReqUInt, type);
- return ip;
- }
- TypeReq IntBinaryExprNode::getPreferredType()
- {
- return TypeReqUInt;
- }
- //------------------------------------------------------------
- U32 StreqExprNode::precompile(TypeReq type)
- {
- // eval str left
- // OP_ADVANCE_STR_NUL
- // eval str right
- // OP_COMPARE_STR
- // optional conversion
- U32 addSize = left->precompile(TypeReqString) + right->precompile(TypeReqString) + 2;
- if(!eq)
- addSize ++;
- if(type != TypeReqUInt)
- addSize ++;
- return addSize;
- }
- U32 StreqExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = left->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_ADVANCE_STR_NUL;
- ip = right->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_COMPARE_STR;
- if(!eq)
- codeStream[ip++] = OP_NOT;
- if(type != TypeReqUInt)
- codeStream[ip++] = conversionOp(TypeReqUInt, type);
- return ip;
- }
- TypeReq StreqExprNode::getPreferredType()
- {
- return TypeReqUInt;
- }
- //------------------------------------------------------------
- U32 StrcatExprNode::precompile(TypeReq type)
- {
- U32 addSize = left->precompile(TypeReqString) + right->precompile(TypeReqString) + 2;
- if(appendChar)
- addSize++;
- if(type != TypeReqString)
- addSize ++;
- return addSize;
- }
- U32 StrcatExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = left->compile(codeStream, ip, TypeReqString);
- if(!appendChar)
- codeStream[ip++] = OP_ADVANCE_STR;
- else
- {
- codeStream[ip++] = OP_ADVANCE_STR_APPENDCHAR;
- codeStream[ip++] = appendChar;
- }
- ip = right->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_REWIND_STR;
- if(type == TypeReqUInt)
- codeStream[ip++] = OP_STR_TO_UINT;
- else if(type == TypeReqFloat)
- codeStream[ip++] = OP_STR_TO_FLT;
- return ip;
- }
- TypeReq StrcatExprNode::getPreferredType()
- {
- return TypeReqString;
- }
- //------------------------------------------------------------
- U32 CommaCatExprNode::precompile(TypeReq type)
- {
- U32 addSize = left->precompile(TypeReqString) + right->precompile(TypeReqString) + 2;
- if(type != TypeReqString)
- addSize ++;
- return addSize;
- }
- U32 CommaCatExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = left->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_ADVANCE_STR_COMMA;
- ip = right->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_REWIND_STR;
- // At this point the stack has the concatenated string.
- // But we're paranoid, so accept (but whine) if we get an oddity...
- if(type == TypeReqUInt || type == TypeReqFloat)
- Con::warnf(ConsoleLogEntry::General, "%s (%d): converting comma string to a number... probably wrong.", dbgFileName, dbgLineNumber);
- if(type == TypeReqUInt)
- codeStream[ip++] = OP_STR_TO_UINT;
- else if(type == TypeReqFloat)
- codeStream[ip++] = OP_STR_TO_FLT;
- return ip;
- }
- TypeReq CommaCatExprNode::getPreferredType()
- {
- return TypeReqString;
- }
- //------------------------------------------------------------
- U32 IntUnaryExprNode::precompile(TypeReq type)
- {
- integer = true;
- TypeReq prefType = expr->getPreferredType();
- if(op == '!' && (prefType == TypeReqFloat || prefType == TypeReqString))
- integer = false;
- U32 exprSize = expr->precompile(integer ? TypeReqUInt : TypeReqFloat);
- if(type != TypeReqUInt)
- return exprSize + 2;
- else
- return exprSize + 1;
- }
- U32 IntUnaryExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = expr->compile(codeStream, ip, integer ? TypeReqUInt : TypeReqFloat);
- if(op == '!')
- codeStream[ip++] = integer ? OP_NOT : OP_NOTF;
- else if(op == '~')
- codeStream[ip++] = OP_ONESCOMPLEMENT;
- if(type != TypeReqUInt)
- codeStream[ip++] =conversionOp(TypeReqUInt, type);
- return ip;
- }
- TypeReq IntUnaryExprNode::getPreferredType()
- {
- return TypeReqUInt;
- }
- //------------------------------------------------------------
- U32 FloatUnaryExprNode::precompile(TypeReq type)
- {
- U32 exprSize = expr->precompile(TypeReqFloat);
- if(type != TypeReqFloat)
- return exprSize + 2;
- else
- return exprSize + 1;
- }
- U32 FloatUnaryExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = expr->compile(codeStream, ip, TypeReqFloat);
- codeStream[ip++] = OP_NEG;
- if(type != TypeReqFloat)
- codeStream[ip++] =conversionOp(TypeReqFloat, type);
- return ip;
- }
- TypeReq FloatUnaryExprNode::getPreferredType()
- {
- return TypeReqFloat;
- }
- //------------------------------------------------------------
- U32 VarNode::precompile(TypeReq type)
- {
- // if this has an arrayIndex...
- // OP_LOADIMMED_IDENT
- // varName
- // OP_ADVANCE_STR
- // evaluate arrayIndex TypeReqString
- // OP_REWIND_STR
- // OP_SETCURVAR_ARRAY
- // OP_LOADVAR (type)
- // else
- // OP_SETCURVAR
- // varName
- // OP_LOADVAR (type)
- if(type == TypeReqNone)
- return 0;
- precompileIdent(varName);
- if(arrayIndex)
- return arrayIndex->precompile(TypeReqString) + 7;
- else
- return 4;
- }
- U32 VarNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- if(type == TypeReqNone)
- return ip;
- codeStream[ip++] = arrayIndex ? OP_LOADIMMED_IDENT : OP_SETCURVAR;
- STEtoCode(varName, ip, codeStream);
- ip += 2;
- if(arrayIndex)
- {
- codeStream[ip++] = OP_ADVANCE_STR;
- ip = arrayIndex->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_REWIND_STR;
- codeStream[ip++] = OP_SETCURVAR_ARRAY;
- }
- switch(type)
- {
- case TypeReqUInt:
- codeStream[ip++] = OP_LOADVAR_UINT;
- break;
- case TypeReqFloat:
- codeStream[ip++] = OP_LOADVAR_FLT;
- break;
- case TypeReqString:
- codeStream[ip++] = OP_LOADVAR_STR;
- break;
- default:
- break;
- }
- return ip;
- }
- TypeReq VarNode::getPreferredType()
- {
- return TypeReqNone; // no preferred type
- }
- //------------------------------------------------------------
- U32 IntNode::precompile(TypeReq type)
- {
- if(type == TypeReqNone)
- return 0;
- if(type == TypeReqString)
- index = getCurrentStringTable()->addIntString(value);
- else if(type == TypeReqFloat)
- index = getCurrentFloatTable()->add(value);
- return 2;
- }
- U32 IntNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- switch(type)
- {
- case TypeReqUInt:
- codeStream[ip++] = OP_LOADIMMED_UINT;
- codeStream[ip++] = value;
- break;
- case TypeReqString:
- codeStream[ip++] = OP_LOADIMMED_STR;
- codeStream[ip++] = index;
- break;
- case TypeReqFloat:
- codeStream[ip++] = OP_LOADIMMED_FLT;
- codeStream[ip++] = index;
- break;
- default:
- break;
- }
- return ip;
- }
- TypeReq IntNode::getPreferredType()
- {
- return TypeReqUInt;
- }
- //------------------------------------------------------------
- U32 FloatNode::precompile(TypeReq type)
- {
- if(type == TypeReqNone)
- return 0;
- if(type == TypeReqString)
- index = getCurrentStringTable()->addFloatString(value);
- else if(type == TypeReqFloat)
- index = getCurrentFloatTable()->add(value);
- return 2;
- }
- U32 FloatNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- switch(type)
- {
- case TypeReqUInt:
- codeStream[ip++] = OP_LOADIMMED_UINT;
- codeStream[ip++] = U32(value);
- break;
- case TypeReqString:
- codeStream[ip++] = OP_LOADIMMED_STR;
- codeStream[ip++] = index;
- break;
- case TypeReqFloat:
- codeStream[ip++] = OP_LOADIMMED_FLT;
- codeStream[ip++] = index;
- break;
- default:
- break;
- }
- return ip;
- }
- TypeReq FloatNode::getPreferredType()
- {
- return TypeReqFloat;
- }
- //------------------------------------------------------------
- U32 StrConstNode::precompile(TypeReq type)
- {
- // Early out for documentation block.
- if( doc )
- {
- index = getCurrentStringTable()->add(str, true, tag);
- return 2;
- }
- if(type == TypeReqString)
- {
- index = getCurrentStringTable()->add(str, true, tag);
- return 2;
- }
- else if(type == TypeReqNone)
- {
- return 0;
- }
- fVal = consoleStringToNumber(str, dbgFileName, dbgLineNumber);
- if(type == TypeReqFloat)
- index = getCurrentFloatTable()->add(fVal);
- return 2;
- }
- U32 StrConstNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- // If this is a DOCBLOCK, then process w/ appropriate op...
- if( doc )
- {
- codeStream[ip++] = OP_DOCBLOCK_STR;
- codeStream[ip++] = index;
- return ip;
- }
- // Otherwise, deal with it normally as a string literal case.
- switch(type)
- {
- case TypeReqString:
- codeStream[ip++] = tag ? OP_TAG_TO_STR : OP_LOADIMMED_STR;
- codeStream[ip++] = index;
- break;
- case TypeReqUInt:
- codeStream[ip++] = OP_LOADIMMED_UINT;
- codeStream[ip++] = U32(fVal);
- break;
- case TypeReqFloat:
- codeStream[ip++] = OP_LOADIMMED_FLT;
- codeStream[ip++] = index;
- break;
- default:
- break;
- }
- return ip;
- }
- TypeReq StrConstNode::getPreferredType()
- {
- return TypeReqString;
- }
- //------------------------------------------------------------
- U32 ConstantNode::precompile(TypeReq type)
- {
- if(type == TypeReqString)
- {
- precompileIdent(value);
- return 3;
- }
- else if(type == TypeReqNone)
- return 0;
- fVal = consoleStringToNumber(value, dbgFileName, dbgLineNumber);
- if(type == TypeReqFloat)
- index = getCurrentFloatTable()->add(fVal);
- return 2;
- }
- U32 ConstantNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- switch(type)
- {
- case TypeReqString:
- codeStream[ip++] = OP_LOADIMMED_IDENT;
- STEtoCode(value, ip, codeStream);
- ip += 2;
- break;
- case TypeReqUInt:
- codeStream[ip++] = OP_LOADIMMED_UINT;
- codeStream[ip++] = U32(fVal);
- break;
- case TypeReqFloat:
- codeStream[ip++] = OP_LOADIMMED_FLT;
- codeStream[ip++] = index;
- break;
- default:
- break;
- }
- return ip;
- }
- TypeReq ConstantNode::getPreferredType()
- {
- return TypeReqString;
- }
- //------------------------------------------------------------
- U32 AssignExprNode::precompile(TypeReq type)
- {
- subType = expr->getPreferredType();
- if(subType == TypeReqNone)
- subType = type;
- if(subType == TypeReqNone)
- subType = TypeReqString;
- // if it's an array expr, the formula is:
- // eval expr
- // (push and pop if it's TypeReqString) OP_ADVANCE_STR
- // OP_LOADIMMED_IDENT
- // varName
- // OP_ADVANCE_STR
- // eval array
- // OP_REWIND_STR
- // OP_SETCURVAR_ARRAY_CREATE
- // OP_TERMINATE_REWIND_STR
- // OP_SAVEVAR
- //else
- // eval expr
- // OP_SETCURVAR_CREATE
- // varname
- // OP_SAVEVAR
- U32 addSize = 0;
- if(type != subType)
- addSize = 1;
- U32 retSize = expr->precompile(subType);
- precompileIdent(varName);
- if(arrayIndex)
- {
- if(subType == TypeReqString)
- return arrayIndex->precompile(TypeReqString) + retSize + addSize + 9;
- else
- return arrayIndex->precompile(TypeReqString) + retSize + addSize + 7;
- }
- else
- return retSize + addSize + 4;
- }
- U32 AssignExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = expr->compile(codeStream, ip, subType);
- if(arrayIndex)
- {
- if(subType == TypeReqString)
- codeStream[ip++] = OP_ADVANCE_STR;
- codeStream[ip++] = OP_LOADIMMED_IDENT;
- STEtoCode(varName, ip, codeStream);
- ip += 2;
- codeStream[ip++] = OP_ADVANCE_STR;
- ip = arrayIndex->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_REWIND_STR;
- codeStream[ip++] = OP_SETCURVAR_ARRAY_CREATE;
- if(subType == TypeReqString)
- codeStream[ip++] = OP_TERMINATE_REWIND_STR;
- }
- else
- {
- codeStream[ip++] = OP_SETCURVAR_CREATE;
- STEtoCode(varName, ip, codeStream);
- ip += 2;
- }
- switch(subType)
- {
- case TypeReqString:
- codeStream[ip++] = OP_SAVEVAR_STR;
- break;
- case TypeReqUInt:
- codeStream[ip++] = OP_SAVEVAR_UINT;
- break;
- case TypeReqFloat:
- codeStream[ip++] = OP_SAVEVAR_FLT;
- break;
- default:
- break;
- }
- if(type != subType)
- codeStream[ip++] = conversionOp(subType, type);
- return ip;
- }
- TypeReq AssignExprNode::getPreferredType()
- {
- return expr->getPreferredType();
- }
- //------------------------------------------------------------
- static void getAssignOpTypeOp(S32 op, TypeReq &type, U32 &operand)
- {
- switch(op)
- {
- case '+':
- type = TypeReqFloat;
- operand = OP_ADD;
- break;
- case '-':
- type = TypeReqFloat;
- operand = OP_SUB;
- break;
- case '*':
- type = TypeReqFloat;
- operand = OP_MUL;
- break;
- case '/':
- type = TypeReqFloat;
- operand = OP_DIV;
- break;
- case '%':
- type = TypeReqUInt;
- operand = OP_MOD;
- break;
- case '&':
- type = TypeReqUInt;
- operand = OP_BITAND;
- break;
- case '^':
- type = TypeReqUInt;
- operand = OP_XOR;
- break;
- case '|':
- type = TypeReqUInt;
- operand = OP_BITOR;
- break;
- case opSHL:
- type = TypeReqUInt;
- operand = OP_SHL;
- break;
- case opSHR:
- type = TypeReqUInt;
- operand = OP_SHR;
- break;
- }
- }
- U32 AssignOpExprNode::precompile(TypeReq type)
- {
- // goes like this...
- // eval expr as float or int
- // if there's an arrayIndex
- // OP_LOADIMMED_IDENT
- // varName
- // OP_ADVANCE_STR
- // eval arrayIndex stringwise
- // OP_REWIND_STR
- // OP_SETCURVAR_ARRAY_CREATE
- // else
- // OP_SETCURVAR_CREATE
- // varName
- // OP_LOADVAR_FLT or UINT
- // operand
- // OP_SAVEVAR_FLT or UINT
- // conversion OP if necessary.
- getAssignOpTypeOp(op, subType, operand);
- precompileIdent(varName);
- U32 size = expr->precompile(subType);
- if(type != subType)
- size++;
- if(!arrayIndex)
- return size + 6;
- else
- {
- size += arrayIndex->precompile(TypeReqString);
- return size + 9;
- }
- }
- U32 AssignOpExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = expr->compile(codeStream, ip, subType);
- if(!arrayIndex)
- {
- codeStream[ip++] = OP_SETCURVAR_CREATE;
- STEtoCode(varName, ip, codeStream);
- ip += 2;
- }
- else
- {
- codeStream[ip++] = OP_LOADIMMED_IDENT;
- STEtoCode(varName, ip, codeStream);
- ip += 2;
- codeStream[ip++] = OP_ADVANCE_STR;
- ip = arrayIndex->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_REWIND_STR;
- codeStream[ip++] = OP_SETCURVAR_ARRAY_CREATE;
- }
- codeStream[ip++] = (subType == TypeReqFloat) ? OP_LOADVAR_FLT : OP_LOADVAR_UINT;
- codeStream[ip++] = operand;
- codeStream[ip++] = (subType == TypeReqFloat) ? OP_SAVEVAR_FLT : OP_SAVEVAR_UINT;
- if(subType != type)
- codeStream[ip++] = conversionOp(subType, type);
- return ip;
- }
- TypeReq AssignOpExprNode::getPreferredType()
- {
- getAssignOpTypeOp(op, subType, operand);
- return subType;
- }
- //------------------------------------------------------------
- U32 TTagSetStmtNode::precompileStmt(U32 loopCount)
- {
- return 0;
- }
- U32 TTagSetStmtNode::compileStmt(U32*, U32 ip, U32, U32)
- {
- return ip;
- }
- //------------------------------------------------------------
- U32 TTagDerefNode::precompile(TypeReq)
- {
- return 0;
- }
- U32 TTagDerefNode::compile(U32*, U32 ip, TypeReq)
- {
- return ip;
- }
- TypeReq TTagDerefNode::getPreferredType()
- {
- return TypeReqNone;
- }
- //------------------------------------------------------------
- U32 TTagExprNode::precompile(TypeReq)
- {
- return 0;
- }
- U32 TTagExprNode::compile(U32*, U32 ip, TypeReq)
- {
- return ip;
- }
- TypeReq TTagExprNode::getPreferredType()
- {
- return TypeReqNone;
- }
- //------------------------------------------------------------
- U32 FuncCallExprNode::precompile(TypeReq type)
- {
- // OP_PUSH_FRAME
- // arg OP_PUSH arg OP_PUSH arg OP_PUSH
- // eval all the args, then call the function.
- // OP_CALLFUNC
- // function
- // namespace
- // isDot
- U32 size = 0;
- if(type != TypeReqString)
- size++;
- precompileIdent(funcName);
- precompileIdent(nameSpace);
- for(ExprNode *walk = args; walk; walk = (ExprNode *) walk->getNext())
- size += walk->precompile(TypeReqString) + 1;
- return size + 7;
- }
- U32 FuncCallExprNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- codeStream[ip++] = OP_PUSH_FRAME;
- for(ExprNode *walk = args; walk; walk = (ExprNode *) walk->getNext())
- {
- ip = walk->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_PUSH;
- }
- if(callType == MethodCall || callType == ParentCall)
- codeStream[ip++] = OP_CALLFUNC;
- else
- codeStream[ip++] = OP_CALLFUNC_RESOLVE;
- STEtoCode(funcName, ip, codeStream);
- ip += 2;
- STEtoCode(nameSpace, ip, codeStream);
- ip += 2;
- codeStream[ip++] = callType;
- if(type != TypeReqString)
- codeStream[ip++] = conversionOp(TypeReqString, type);
- return ip;
- }
- TypeReq FuncCallExprNode::getPreferredType()
- {
- return TypeReqString;
- }
- //------------------------------------------------------------
- U32 SlotAccessNode::precompile(TypeReq type)
- {
- if(type == TypeReqNone)
- return 0;
- U32 size = 0;
- precompileIdent(slotName);
- if(arrayExpr)
- {
- // eval array
- // OP_ADVANCE_STR
- // evaluate object expression sub (OP_SETCURFIELD)
- // OP_TERMINATE_REWIND_STR
- // OP_SETCURFIELDARRAY
- // total add of 4 + array precomp
- size += 3 + arrayExpr->precompile(TypeReqString);
- }
- // eval object expression sub + 3 (op_setCurField + OP_SETCUROBJECT)
- size += objectExpr->precompile(TypeReqString) + 4;
- // get field in desired type:
- return size + 1;
- }
- U32 SlotAccessNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- if(type == TypeReqNone)
- return ip;
- if(arrayExpr)
- {
- ip = arrayExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_ADVANCE_STR;
- }
- ip = objectExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_SETCUROBJECT;
-
- codeStream[ip++] = OP_SETCURFIELD;
-
- STEtoCode(slotName, ip, codeStream);
- ip += 2;
- if(arrayExpr)
- {
- codeStream[ip++] = OP_TERMINATE_REWIND_STR;
- codeStream[ip++] = OP_SETCURFIELD_ARRAY;
- }
-
- switch(type)
- {
- case TypeReqUInt:
- codeStream[ip++] = OP_LOADFIELD_UINT;
- break;
- case TypeReqFloat:
- codeStream[ip++] = OP_LOADFIELD_FLT;
- break;
- case TypeReqString:
- codeStream[ip++] = OP_LOADFIELD_STR;
- break;
- default:
- break;
- }
- return ip;
- }
- TypeReq SlotAccessNode::getPreferredType()
- {
- return TypeReqNone;
- }
- //////////////////////////////////////////////////////////////////////////
- U32 InternalSlotAccessNode::precompile(TypeReq type)
- {
- if(type == TypeReqNone)
- return 0;
- U32 size = 3;
-
- // eval object expression sub + 3 (op_setCurField + OP_SETCUROBJECT)
- size += objectExpr->precompile(TypeReqString);
- size += slotExpr->precompile(TypeReqString);
- if(type != TypeReqUInt)
- size++;
- // get field in desired type:
- return size;
- }
- U32 InternalSlotAccessNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- if(type == TypeReqNone)
- return ip;
- ip = objectExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_SETCUROBJECT;
- ip = slotExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_SETCUROBJECT_INTERNAL;
- codeStream[ip++] = recurse;
- if(type != TypeReqUInt)
- codeStream[ip++] = conversionOp(TypeReqUInt, type);
- return ip;
- }
- TypeReq InternalSlotAccessNode::getPreferredType()
- {
- return TypeReqUInt;
- }
- //////////////////////////////////////////////////////////////////////////
- //------------------------------------------------------------
- U32 SlotAssignNode::precompile(TypeReq type)
- {
- // first eval the expression TypeReqString
- // if it's an array:
- // if OP_ADVANCE_STR 1
- // eval array
- // OP_ADVANCE_STR 1
- // evaluate object expr
- // OP_SETCUROBJECT 1
- // OP_SETCURFIELD 1
- // fieldName 1
- // OP_TERMINATE_REWIND_STR 1
- // OP_SETCURFIELDARRAY 1
- // OP_TERMINATE_REWIND_STR 1
- // else
- // OP_ADVANCE_STR
- // evaluate object expr
- // OP_SETCUROBJECT
- // OP_SETCURFIELD
- // fieldName
- // OP_TERMINATE_REWIND_STR
- // OP_SAVEFIELD
- // convert to return type if necessary.
- U32 size = 0;
- if(type != TypeReqString)
- size++;
- precompileIdent(slotName);
- size += valueExpr->precompile(TypeReqString);
- if(objectExpr)
- size += objectExpr->precompile(TypeReqString) + 6;
- else
- size += 6;
- if(arrayExpr)
- size += arrayExpr->precompile(TypeReqString) + 3;
- return size + 1;
- }
- U32 SlotAssignNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = valueExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_ADVANCE_STR;
- if(arrayExpr)
- {
- ip = arrayExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_ADVANCE_STR;
- }
- if(objectExpr)
- {
- ip = objectExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_SETCUROBJECT;
- }
- else
- codeStream[ip++] = OP_SETCUROBJECT_NEW;
- codeStream[ip++] = OP_SETCURFIELD;
- STEtoCode(slotName, ip, codeStream);
- ip += 2;
- if(arrayExpr)
- {
- codeStream[ip++] = OP_TERMINATE_REWIND_STR;
- codeStream[ip++] = OP_SETCURFIELD_ARRAY;
- }
- codeStream[ip++] = OP_TERMINATE_REWIND_STR;
- codeStream[ip++] = OP_SAVEFIELD_STR;
- if(type != TypeReqString)
- codeStream[ip++] = conversionOp(TypeReqString, type);
- return ip;
- }
- TypeReq SlotAssignNode::getPreferredType()
- {
- return TypeReqString;
- }
- //------------------------------------------------------------
- U32 SlotAssignOpNode::precompile(TypeReq type)
- {
- // first eval the expression as its type
- // if it's an array:
- // eval array
- // OP_ADVANCE_STR
- // evaluate object expr
- // OP_SETCUROBJECT
- // OP_SETCURFIELD
- // fieldName
- // OP_TERMINATE_REWIND_STR
- // OP_SETCURFIELDARRAY
- // else
- // evaluate object expr
- // OP_SETCUROBJECT
- // OP_SETCURFIELD
- // fieldName
- // OP_LOADFIELD of appropriate type
- // operand
- // OP_SAVEFIELD of appropriate type
- // convert to return type if necessary.
- getAssignOpTypeOp(op, subType, operand);
- precompileIdent(slotName);
- U32 size = valueExpr->precompile(subType);
- if(type != subType)
- size++;
- if(arrayExpr)
- return size + 10 + arrayExpr->precompile(TypeReqString) + objectExpr->precompile(TypeReqString);
- else
- return size + 7 + objectExpr->precompile(TypeReqString);
- }
- U32 SlotAssignOpNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- ip = valueExpr->compile(codeStream, ip, subType);
- if(arrayExpr)
- {
- ip = arrayExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_ADVANCE_STR;
- }
- ip = objectExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_SETCUROBJECT;
- codeStream[ip++] = OP_SETCURFIELD;
- STEtoCode(slotName, ip, codeStream);
- ip += 2;
- if(arrayExpr)
- {
- codeStream[ip++] = OP_TERMINATE_REWIND_STR;
- codeStream[ip++] = OP_SETCURFIELD_ARRAY;
- }
- codeStream[ip++] = (subType == TypeReqFloat) ? OP_LOADFIELD_FLT : OP_LOADFIELD_UINT;
- codeStream[ip++] = operand;
- codeStream[ip++] = (subType == TypeReqFloat) ? OP_SAVEFIELD_FLT : OP_SAVEFIELD_UINT;
- if(subType != type)
- codeStream[ip++] = conversionOp(subType, type);
- return ip;
- }
- TypeReq SlotAssignOpNode::getPreferredType()
- {
- getAssignOpTypeOp(op, subType, operand);
- return subType;
- }
- //------------------------------------------------------------
- U32 ObjectDeclNode::precompileSubObject(bool)
- {
- // goes
- // OP_PUSHFRAME 1
- // name expr
- // OP_PUSH 1
- // args... PUSH
- // OP_CREATE_OBJECT 1
- // className 1
- // datablock? 1
- // fail point 1
- // for each field, eval
- // OP_ADD_OBJECT (to UINT[0]) 1
- // root? 1
- // add all the sub objects.
- // OP_END_OBJECT 1
- // root? 1
- U32 argSize = 0;
- precompileIdent(parentObject);
- for(ExprNode *exprWalk = argList; exprWalk; exprWalk = (ExprNode *) exprWalk->getNext())
- argSize += exprWalk->precompile(TypeReqString) + 1;
- argSize += classNameExpr->precompile(TypeReqString) + 1;
- U32 nameSize = objectNameExpr->precompile(TypeReqString) + 3; // 2
- U32 slotSize = 0;
- for(SlotAssignNode *slotWalk = slotDecls; slotWalk; slotWalk = (SlotAssignNode *) slotWalk->getNext())
- slotSize += slotWalk->precompile(TypeReqNone);
- // OP_ADD_OBJECT
- U32 subObjSize = 0;
- for(ObjectDeclNode *objectWalk = subObjects; objectWalk; objectWalk = (ObjectDeclNode *) objectWalk->getNext())
- subObjSize += objectWalk->precompileSubObject(false);
- failOffset = 10 + nameSize + argSize + slotSize + subObjSize;
- return failOffset;
- }
- U32 ObjectDeclNode::precompile(TypeReq type)
- {
- // root object decl does:
- // push 0 onto the UINT stack OP_LOADIMMED_UINT
- // precompiles the subObject(true)
- // UINT stack now has object id
- // type conv to type
- U32 ret = 2 + precompileSubObject(true);
- if(type != TypeReqUInt)
- return ret + 1;
- return ret;
- }
- U32 ObjectDeclNode::compileSubObject(U32 *codeStream, U32 ip, bool root)
- {
- U32 start = ip;
- codeStream[ip++] = OP_PUSH_FRAME;
- ip = classNameExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_PUSH;
- ip = objectNameExpr->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_PUSH;
- for(ExprNode *exprWalk = argList; exprWalk; exprWalk = (ExprNode *) exprWalk->getNext())
- {
- ip = exprWalk->compile(codeStream, ip, TypeReqString);
- codeStream[ip++] = OP_PUSH;
- }
- codeStream[ip++] = OP_CREATE_OBJECT;
- STEtoCode(parentObject, ip, codeStream);
- ip += 2;
- codeStream[ip++] = structDecl;
- codeStream[ip++] = isClassNameInternal;
- codeStream[ip++] = isMessage;
- codeStream[ip++] = start + failOffset;
- for(SlotAssignNode *slotWalk = slotDecls; slotWalk; slotWalk = (SlotAssignNode *) slotWalk->getNext())
- ip = slotWalk->compile(codeStream, ip, TypeReqNone);
- codeStream[ip++] = OP_ADD_OBJECT;
- codeStream[ip++] = root;
- for(ObjectDeclNode *objectWalk = subObjects; objectWalk; objectWalk = (ObjectDeclNode *) objectWalk->getNext())
- ip = objectWalk->compileSubObject(codeStream, ip, false);
- codeStream[ip++] = OP_END_OBJECT;
- codeStream[ip++] = root || structDecl;
- return ip;
- }
- U32 ObjectDeclNode::compile(U32 *codeStream, U32 ip, TypeReq type)
- {
- codeStream[ip++] = OP_LOADIMMED_UINT;
- codeStream[ip++] = 0;
- ip = compileSubObject(codeStream, ip, true);
- if(type != TypeReqUInt)
- codeStream[ip++] = conversionOp(TypeReqUInt, type);
- return ip;
- }
- TypeReq ObjectDeclNode::getPreferredType()
- {
- return TypeReqUInt;
- }
- //------------------------------------------------------------
- U32 FunctionDeclStmtNode::precompileStmt(U32)
- {
- // OP_FUNC_DECL
- // func name
- // namespace
- // package
- // func end ip
- // argc
- // ident array[argc]
- // code
- // OP_RETURN
- setCurrentStringTable(&getFunctionStringTable());
- setCurrentFloatTable(&getFunctionFloatTable());
- argc = 0;
- for(VarNode *walk = args; walk; walk = (VarNode *)((StmtNode*)walk)->getNext())
- argc++;
-
- CodeBlock::smInFunction = true;
-
- precompileIdent(fnName);
- precompileIdent(nameSpace);
- precompileIdent(package);
-
- U32 subSize = precompileBlock(stmts, 0);
-
- #ifdef TORQUE_EXTRA_BREAKLINES
- addBreakCount();
- #endif
- CodeBlock::smInFunction = false;
- setCurrentStringTable(&getGlobalStringTable());
- setCurrentFloatTable(&getGlobalFloatTable());
- endOffset = (argc*2) + subSize + 11;
- return endOffset;
- }
- U32 FunctionDeclStmtNode::compileStmt(U32 *codeStream, U32 ip, U32, U32)
- {
- U32 start = ip;
- codeStream[ip++] = OP_FUNC_DECL;
- STEtoCode(fnName, ip, codeStream);
- ip += 2;
- STEtoCode(nameSpace, ip, codeStream);
- ip += 2;
- STEtoCode(package, ip, codeStream);
- ip += 2;
- codeStream[ip++] = bool(stmts != NULL);
- codeStream[ip++] = start + endOffset;
- codeStream[ip++] = argc;
- for(VarNode *walk = args; walk; walk = (VarNode *)((StmtNode*)walk)->getNext())
- {
- STEtoCode(walk->varName, ip, codeStream);
- ip += 2;
- }
- CodeBlock::smInFunction = true;
- ip = compileBlock(stmts, codeStream, ip, 0, 0);
- #ifdef TORQUE_EXTRA_BREAKLINES
- addBreakLine(ip);
- #endif
- CodeBlock::smInFunction = false;
- codeStream[ip++] = OP_RETURN;
- return ip;
- }
|