4 * AST to Bytecode Conversion
12 #include "bytecode_gen.h"
13 #include "bytecode_ops.h"
15 #define TRACE_VAR_LOOKUPS 0
16 #define TRACE_NODE_RETURNS 0
17 #define MAX_NAMESPACE_DEPTH 10
20 extern tSpiderFunction *gpExports_First;
23 typedef struct sAST_BlockInfo
25 struct sAST_BlockInfo *Parent;
32 const char *CurNamespaceStack[MAX_NAMESPACE_DEPTH];
37 int AST_ConvertNode(tAST_BlockInfo *Block, tAST_Node *Node, int bKeepValue);
38 int BC_SaveValue(tAST_BlockInfo *Block, tAST_Node *DestNode);
40 int BC_Variable_Define(tAST_BlockInfo *Block, int Type, const char *Name);
41 int BC_Variable_SetValue(tAST_BlockInfo *Block, tAST_Node *VarNode);
42 int BC_Variable_GetValue(tAST_BlockInfo *Block, tAST_Node *VarNode);
44 void AST_RuntimeMessage(tAST_Node *Node, const char *Type, const char *Format, ...);
45 void AST_RuntimeError(tAST_Node *Node, const char *Format, ...);
48 // int giNextBlockIdent = 1;
51 int SpiderScript_BytecodeScript(tSpiderScript *Script)
53 tScript_Function *fcn;
54 for(fcn = Script->Functions; fcn; fcn = fcn->Next)
56 if( Bytecode_ConvertFunction(fcn) == 0 )
63 * \brief Convert a function into bytecode
65 tBC_Function *Bytecode_ConvertFunction(tScript_Function *Fcn)
68 tAST_BlockInfo bi = {0};
70 // TODO: Return BCFcn instead?
71 if(Fcn->BCFcn) return Fcn->BCFcn;
73 ret = Bytecode_CreateFunction(Fcn);
77 if( AST_ConvertNode(&bi, Fcn->ASTFcn, 0) )
79 Bytecode_DeleteFunction(ret);
83 Bytecode_AppendConstInt(ret, 0); // TODO: NULL
84 Bytecode_AppendReturn(ret);
90 // Indepotent operation
91 #define CHECK_IF_NEEDED(b_warn) do { if(!bKeepValue) {\
92 if(b_warn)AST_RuntimeMessage(Node, "Bytecode", "Operation without saving");\
93 Bytecode_AppendDelete(Block->Handle);\
97 * \brief Convert a node into bytecode
98 * \param Block Execution context
99 * \param Node Node to execute
101 int AST_ConvertNode(tAST_BlockInfo *Block, tAST_Node *Node, int bKeepValue)
106 int bAddedValue = 1; // Used to tell if the value needs to be deleted
117 Bytecode_AppendEnterContext(Block->Handle); // Create a new block
119 tAST_BlockInfo blockInfo = {0};
120 blockInfo.Parent = Block;
121 blockInfo.Handle = Block->Handle;
122 // Loop over all nodes, or until the return value is set
123 for(node = Node->Block.FirstChild;
125 node = node->NextSibling )
127 AST_ConvertNode(Block, node, 0);
130 Bytecode_AppendLeaveContext(Block->Handle); // Leave this context
134 case NODETYPE_ASSIGN:
135 // Perform assignment operation
136 if( Node->Assign.Operation != NODETYPE_NOP )
138 ret = AST_ConvertNode(Block, Node->Assign.Dest, 1);
140 ret = AST_ConvertNode(Block, Node->Assign.Value, 1);
142 switch(Node->Assign.Operation)
144 // General Binary Operations
145 case NODETYPE_ADD: op = BC_OP_ADD; break;
146 case NODETYPE_SUBTRACT: op = BC_OP_SUBTRACT; break;
147 case NODETYPE_MULTIPLY: op = BC_OP_MULTIPLY; break;
148 case NODETYPE_DIVIDE: op = BC_OP_DIVIDE; break;
149 case NODETYPE_MODULO: op = BC_OP_MODULO; break;
150 case NODETYPE_BWAND: op = BC_OP_BITAND; break;
151 case NODETYPE_BWOR: op = BC_OP_BITOR; break;
152 case NODETYPE_BWXOR: op = BC_OP_BITXOR; break;
153 case NODETYPE_BITSHIFTLEFT: op = BC_OP_BITSHIFTLEFT; break;
154 case NODETYPE_BITSHIFTRIGHT: op = BC_OP_BITSHIFTRIGHT; break;
155 case NODETYPE_BITROTATELEFT: op = BC_OP_BITROTATELEFT; break;
158 AST_RuntimeError(Node, "Unknown operation in ASSIGN %i", Node->Assign.Operation);
161 // printf("assign, op = %i\n", op);
162 Bytecode_AppendBinOp(Block->Handle, op);
166 ret = AST_ConvertNode(Block, Node->Assign.Value, 1);
171 Bytecode_AppendDuplicate(Block->Handle);
173 ret = BC_SaveValue(Block, Node->Assign.Dest);
176 // Post increment/decrement
177 case NODETYPE_POSTINC:
178 case NODETYPE_POSTDEC:
179 // Save original value if requested
181 ret = BC_Variable_GetValue(Block, Node->UniOp.Value);
185 Bytecode_AppendConstInt(Block->Handle, 1);
187 ret = AST_ConvertNode(Block, Node->UniOp.Value, 1);
190 if( Node->Type == NODETYPE_POSTDEC )
191 Bytecode_AppendBinOp(Block->Handle, BC_OP_SUBTRACT);
193 Bytecode_AppendBinOp(Block->Handle, BC_OP_ADD);
196 ret = BC_SaveValue(Block, Node->UniOp.Value);
200 case NODETYPE_METHODCALL:
201 case NODETYPE_FUNCTIONCALL:
202 case NODETYPE_CREATEOBJECT: {
205 // Put the object earlier on the stack to the arguments (for exec)
206 if( Node->Type == NODETYPE_METHODCALL ) {
207 ret = AST_ConvertNode(Block, Node->FunctionCall.Object, 1);
211 for(node = Node->FunctionCall.FirstArg; node; node = node->NextSibling)
213 ret = AST_ConvertNode(Block, node, 1);
219 if( Node->Type == NODETYPE_METHODCALL )
221 // TODO: Sanity check stack top
222 Bytecode_AppendMethodCall(Block->Handle, Node->FunctionCall.Name, nargs);
228 for( i = 0; i < MAX_NAMESPACE_DEPTH && Block->CurNamespaceStack[i]; i ++ )
229 newnamelen = strlen(Block->CurNamespaceStack[i]) + 1;
230 newnamelen += strlen(Node->FunctionCall.Name) + 1;
231 manglename = alloca(newnamelen);
233 for( i = 0; i < MAX_NAMESPACE_DEPTH && Block->CurNamespaceStack[i]; i ++ ) {
235 strcat(manglename, Block->CurNamespaceStack[i]);
236 pos = strlen(manglename);
237 manglename[pos] = BC_NS_SEPARATOR;
238 manglename[pos+1] = '\0';
240 strcat(manglename, Node->FunctionCall.Name);
242 if( Node->Type == NODETYPE_CREATEOBJECT )
244 // TODO: Sanity check stack top
245 Bytecode_AppendCreateObj(Block->Handle, manglename, nargs);
249 Bytecode_AppendFunctionCall(Block->Handle, manglename, nargs);
252 CHECK_IF_NEEDED(0); // Don't warn
253 // TODO: Implement warn_unused_ret
259 ret = AST_ConvertNode(Block, Node->If.Condition, 1);
262 if_end = Bytecode_AllocateLabel(Block->Handle);
264 if( Node->If.False->Type != NODETYPE_NOP )
266 int if_true = Bytecode_AllocateLabel(Block->Handle);
268 Bytecode_AppendCondJump(Block->Handle, if_true);
271 ret = AST_ConvertNode(Block, Node->If.False, 0);
273 Bytecode_AppendJump(Block->Handle, if_end);
274 Bytecode_SetLabel(Block->Handle, if_true);
278 Bytecode_AppendCondJumpNot(Block->Handle, if_end);
282 ret = AST_ConvertNode(Block, Node->If.True, 0);
286 Bytecode_SetLabel(Block->Handle, if_end);
290 case NODETYPE_LOOP: {
291 int loop_start, loop_end;
292 int saved_break, saved_continue;
293 const char *saved_tag;
296 ret = AST_ConvertNode(Block, Node->For.Init, 0);
299 loop_start = Bytecode_AllocateLabel(Block->Handle);
300 loop_end = Bytecode_AllocateLabel(Block->Handle);
302 saved_break = Block->BreakTarget;
303 saved_continue = Block->ContinueTarget;
304 saved_tag = Block->Tag;
305 Block->BreakTarget = loop_end;
306 Block->ContinueTarget = loop_end;
307 Block->Tag = Node->For.Tag;
309 Bytecode_SetLabel(Block->Handle, loop_start);
311 // Check initial condition
312 if( !Node->For.bCheckAfter )
314 ret = AST_ConvertNode(Block, Node->For.Condition, 1);
316 Bytecode_AppendUniOp(Block->Handle, BC_OP_LOGICNOT);
317 Bytecode_AppendCondJump(Block->Handle, loop_end);
321 ret = AST_ConvertNode(Block, Node->For.Code, 0);
325 ret = AST_ConvertNode(Block, Node->For.Increment, 0);
329 if( Node->For.bCheckAfter )
331 ret = AST_ConvertNode(Block, Node->For.Condition, 1);
333 Bytecode_AppendCondJump(Block->Handle, loop_start);
337 Bytecode_AppendJump(Block->Handle, loop_start);
340 Bytecode_SetLabel(Block->Handle, loop_end);
342 Block->BreakTarget = saved_break;
343 Block->ContinueTarget = saved_continue;
344 Block->Tag = saved_tag;
348 case NODETYPE_RETURN:
349 ret = AST_ConvertNode(Block, Node->UniOp.Value, 1);
351 Bytecode_AppendReturn(Block->Handle);
355 case NODETYPE_CONTINUE: {
356 tAST_BlockInfo *bi = Block;
357 if( Node->Variable.Name[0] ) {
358 while(bi && strcmp(bi->Tag, Node->Variable.Name) == 0) bi = bi->Parent;
361 // TODO: Check if BreakTarget/ContinueTarget are valid
362 if( Node->Type == NODETYPE_BREAK )
363 Bytecode_AppendJump(Block->Handle, bi->BreakTarget);
365 Bytecode_AppendJump(Block->Handle, bi->ContinueTarget);
369 case NODETYPE_DEFVAR:
370 ret = BC_Variable_Define(Block, Node->DefVar.DataType, Node->DefVar.Name);
373 if( Node->DefVar.InitialValue )
375 ret = AST_ConvertNode(Block, Node->DefVar.InitialValue, 1);
377 Bytecode_AppendSaveVar(Block->Handle, Node->DefVar.Name);
383 for( i = 0; i < MAX_NAMESPACE_DEPTH && Block->CurNamespaceStack[i]; i ++ );
384 if( i == MAX_NAMESPACE_DEPTH ) {
385 AST_RuntimeError(Node, "Exceeded max explicit namespace depth (%i)", i);
388 Block->CurNamespaceStack[i] = Node->Scope.Name;
389 ret = AST_ConvertNode(Block, Node->Scope.Element, 2);
390 Block->CurNamespaceStack[i] = NULL;
391 CHECK_IF_NEEDED(0); // No warning?
392 // TODO: Will this collide with _CALLFUNCTION etc?
396 case NODETYPE_VARIABLE:
397 ret = BC_Variable_GetValue( Block, Node );
401 // Element of an Object
402 case NODETYPE_ELEMENT:
403 ret = AST_ConvertNode( Block, Node->Scope.Element, 1 );
406 Bytecode_AppendElement(Block->Handle, Node->Scope.Name);
410 // Cast a value to another
412 ret = AST_ConvertNode(Block, Node->Cast.Value, 1);
414 Bytecode_AppendCast(Block->Handle, Node->Cast.DataType);
418 // Index into an array
420 ret = AST_ConvertNode(Block, Node->BinOp.Left, 1); // Array
422 ret = AST_ConvertNode(Block, Node->BinOp.Right, 1); // Offset
425 Bytecode_AppendIndex(Block->Handle);
429 // TODO: Implement runtime constants
430 case NODETYPE_CONSTANT:
431 // TODO: Scan namespace for constant name
432 AST_RuntimeError(Node, "TODO - Runtime Constants");
437 case NODETYPE_STRING:
438 Bytecode_AppendConstString(Block->Handle, Node->Constant.String.Data, Node->Constant.String.Length);
441 case NODETYPE_INTEGER:
442 Bytecode_AppendConstInt(Block->Handle, Node->Constant.Integer);
446 Bytecode_AppendConstReal(Block->Handle, Node->Constant.Real);
450 // --- Operations ---
451 // Boolean Operations
452 case NODETYPE_LOGICALNOT: // Logical NOT (!)
453 if(!op) op = BC_OP_LOGICNOT;
454 case NODETYPE_BWNOT: // Bitwise NOT (~)
455 if(!op) op = BC_OP_BITNOT;
456 case NODETYPE_NEGATE: // Negation (-)
457 if(!op) op = BC_OP_NEG;
458 ret = AST_ConvertNode(Block, Node->UniOp.Value, 1);
460 Bytecode_AppendUniOp(Block->Handle, op);
465 case NODETYPE_LOGICALAND: if(!op) op = BC_OP_LOGICAND;
466 case NODETYPE_LOGICALOR: if(!op) op = BC_OP_LOGICOR;
467 case NODETYPE_LOGICALXOR: if(!op) op = BC_OP_LOGICXOR;
469 case NODETYPE_EQUALS: if(!op) op = BC_OP_EQUALS;
470 case NODETYPE_NOTEQUALS: if(!op) op = BC_OP_NOTEQUALS;
471 case NODETYPE_LESSTHAN: if(!op) op = BC_OP_LESSTHAN;
472 case NODETYPE_GREATERTHAN: if(!op) op = BC_OP_GREATERTHAN;
473 case NODETYPE_LESSTHANEQUAL: if(!op) op = BC_OP_LESSTHANOREQUAL;
474 case NODETYPE_GREATERTHANEQUAL: if(!op) op = BC_OP_GREATERTHANOREQUAL;
475 // General Binary Operations
476 case NODETYPE_ADD: if(!op) op = BC_OP_ADD;
477 case NODETYPE_SUBTRACT: if(!op) op = BC_OP_SUBTRACT;
478 case NODETYPE_MULTIPLY: if(!op) op = BC_OP_MULTIPLY;
479 case NODETYPE_DIVIDE: if(!op) op = BC_OP_DIVIDE;
480 case NODETYPE_MODULO: if(!op) op = BC_OP_MODULO;
481 case NODETYPE_BWAND: if(!op) op = BC_OP_BITAND;
482 case NODETYPE_BWOR: if(!op) op = BC_OP_BITOR;
483 case NODETYPE_BWXOR: if(!op) op = BC_OP_BITXOR;
484 case NODETYPE_BITSHIFTLEFT: if(!op) op = BC_OP_BITSHIFTLEFT;
485 case NODETYPE_BITSHIFTRIGHT: if(!op) op = BC_OP_BITSHIFTRIGHT;
486 case NODETYPE_BITROTATELEFT: if(!op) op = BC_OP_BITROTATELEFT;
487 ret = AST_ConvertNode(Block, Node->BinOp.Left, 1);
489 ret = AST_ConvertNode(Block, Node->BinOp.Right, 1);
492 Bytecode_AppendBinOp(Block->Handle, op);
498 AST_RuntimeError(Node, "BUG - SpiderScript AST_ConvertNode Unimplemented %i", Node->Type);
502 #if TRACE_NODE_RETURNS
503 if(ret && ret != ERRPTR) {
504 AST_RuntimeError(Node, "Ret type of %p %i is %i", Node, Node->Type, ret->Type);
507 AST_RuntimeError(Node, "Ret type of %p %i is %p", Node, Node->Type, ret);
514 int BC_SaveValue(tAST_BlockInfo *Block, tAST_Node *DestNode)
517 switch(DestNode->Type)
520 case NODETYPE_VARIABLE:
521 ret = BC_Variable_SetValue( Block, DestNode );
525 ret = AST_ConvertNode(Block, DestNode->BinOp.Left, 1); // Array
527 ret = AST_ConvertNode(Block, DestNode->BinOp.Right, 1); // Offset
529 Bytecode_AppendSetIndex( Block->Handle );
532 case NODETYPE_ELEMENT:
533 ret = AST_ConvertNode(Block, DestNode->Scope.Element, 1);
535 Bytecode_AppendSetElement( Block->Handle, DestNode->Scope.Name );
539 // TODO: Support assigning to object attributes
540 AST_RuntimeError(DestNode, "Assignment target is not a LValue");
547 * \brief Define a variable
548 * \param Block Current block state
549 * \param Type Type of the variable
550 * \param Name Name of the variable
551 * \return Boolean Failure
553 int BC_Variable_Define(tAST_BlockInfo *Block, int Type, const char *Name)
556 tAST_Variable *var, *prev = NULL;
558 for( var = Block->FirstVar; var; prev = var, var = var->Next )
560 if( strcmp(var->Name, Name) == 0 ) {
561 AST_RuntimeError(NULL, "Redefinition of variable '%s'", Name);
566 var = malloc( sizeof(tAST_Variable) + strlen(Name) + 1 );
569 strcpy(var->Name, Name);
571 if(prev) prev->Next = var;
572 else Block->FirstVar = var;
576 Bytecode_AppendDefineVar(Block->Handle, Name, Type);
581 tAST_Variable *BC_Variable_Lookup(tAST_BlockInfo *Block, tAST_Node *VarNode, int CreateType)
584 tAST_Variable *var = NULL;
587 if( VarNode->BlockState == Block && VarNode->BlockIdent == Block->Ident ) {
588 var = VarNode->ValueCache;
589 #if TRACE_VAR_LOOKUPS
590 AST_RuntimeMessage(VarNode, "debug", "Fast var fetch on '%s' %p (%p:%i)",
591 VarNode->Variable.Name, var,
592 VarNode->BlockState, VarNode->BlockIdent
599 for( bs = Block; bs; bs = bs->Parent )
601 for( var = bs->FirstVar; var; var = var->Next )
603 if( strcmp(var->Name, VarNode->Variable.Name) == 0 )
611 if( Block->Script->Variant->bDyamicTyped && CreateType != SS_DATATYPE_UNDEF ) {
613 var = BC_Variable_Define(Block, CreateType, VarNode->Variable.Name, NULL);
617 AST_RuntimeError(VarNode, "Variable '%s' is undefined", VarNode->Variable.Name);
622 #if TRACE_VAR_LOOKUPS
623 AST_RuntimeMessage(VarNode, "debug", "Saved variable lookup of '%s' %p (%p:%i)",
624 VarNode->Variable.Name, var,
625 Block, Block->Ident);
628 VarNode->ValueCache = var;
629 VarNode->BlockState = Block;
630 VarNode->BlockIdent = Block->Ident;
640 * \brief Set the value of a variable
641 * \return Boolean Failure
643 int BC_Variable_SetValue(tAST_BlockInfo *Block, tAST_Node *VarNode)
647 var = BC_Variable_Lookup(Block, VarNode, SS_DATATYPE_UNDEF);
652 Bytecode_AppendSaveVar(Block->Handle, VarNode->Variable.Name);
657 * \brief Get the value of a variable
659 int BC_Variable_GetValue(tAST_BlockInfo *Block, tAST_Node *VarNode)
663 var = BC_Variable_Lookup(Block, VarNode, 0);
666 Bytecode_AppendLoadVar(Block->Handle, VarNode->Variable.Name);
671 void AST_RuntimeMessage(tAST_Node *Node, const char *Type, const char *Format, ...)
676 fprintf(stderr, "%s:%i: ", Node->File, Node->Line);
678 fprintf(stderr, "%s: ", Type);
679 va_start(args, Format);
680 vfprintf(stderr, Format, args);
682 fprintf(stderr, "\n");
684 void AST_RuntimeError(tAST_Node *Node, const char *Format, ...)
689 fprintf(stderr, "%s:%i: ", Node->File, Node->Line);
691 fprintf(stderr, "error: ");
692 va_start(args, Format);
693 vfprintf(stderr, Format, args);
695 fprintf(stderr, "\n");