9 #define TRACE_VAR_LOOKUPS 0
10 #define TRACE_NODE_RETURNS 0
13 extern tSpiderFunction *gpExports_First;
16 void Object_Dereference(tSpiderValue *Object);
17 void Object_Reference(tSpiderValue *Object);
18 tSpiderValue *SpiderScript_CreateInteger(uint64_t Value);
19 tSpiderValue *SpiderScript_CreateReal(double Value);
20 tSpiderValue *SpiderScript_CreateString(int Length, const char *Data);
21 tSpiderValue *SpiderScript_CastValueTo(int Type, tSpiderValue *Source);
22 int SpiderScript_IsValueTrue(tSpiderValue *Value);
23 void SpiderScript_FreeValue(tSpiderValue *Value);
24 char *SpiderScript_DumpValue(tSpiderValue *Value);
26 tSpiderValue *AST_ExecuteNode(tAST_BlockState *Block, tAST_Node *Node);
27 tSpiderValue *AST_ExecuteNode_BinOp(tAST_BlockState *Block, tAST_Node *Node, int Operation, tSpiderValue *Left, tSpiderValue *Right);
28 tSpiderValue *AST_ExecuteNode_UniOp(tAST_BlockState *Block, tAST_Node *Node, int Operation, tSpiderValue *Value);
30 tAST_Variable *Variable_Define(tAST_BlockState *Block, int Type, const char *Name, tSpiderValue *Value);
31 int Variable_SetValue(tAST_BlockState *Block, tAST_Node *VarNode, tSpiderValue *Value);
32 tSpiderValue *Variable_GetValue(tAST_BlockState *Block, tAST_Node *VarNode);
33 void Variable_Destroy(tAST_Variable *Variable);
35 void AST_RuntimeMessage(tAST_Node *Node, const char *Type, const char *Format, ...);
36 void AST_RuntimeError(tAST_Node *Node, const char *Format, ...);
39 int giNextBlockIdent = 1;
43 * \brief Dereference a created object
45 void Object_Dereference(tSpiderValue *Object)
48 if(Object == ERRPTR) return ;
49 Object->ReferenceCount --;
50 // printf("%p Dereferenced (%i)\n", Object, Object->ReferenceCount);
51 if( Object->ReferenceCount == 0 ) {
52 switch( (enum eSpiderScript_DataTypes) Object->Type )
54 case SS_DATATYPE_OBJECT:
55 Object->Object->Type->Destructor( Object->Object );
57 case SS_DATATYPE_OPAQUE:
58 Object->Opaque.Destroy( Object->Opaque.Data );
67 void Object_Reference(tSpiderValue *Object)
70 Object->ReferenceCount ++;
71 // printf("%p Referenced (%i)\n", Object, Object->ReferenceCount);
75 * \brief Allocate and initialise a SpiderScript object
77 tSpiderObject *SpiderScript_AllocateObject(tSpiderObjectDef *Class, int ExtraBytes)
79 int size = sizeof(tSpiderObject) + Class->NAttributes * sizeof(tSpiderValue*) + ExtraBytes;
80 tSpiderObject *ret = malloc(size);
83 ret->ReferenceCount = 1;
84 ret->OpaqueData = &ret->Attributes[ Class->NAttributes ];
85 memset( ret->Attributes, 0, Class->NAttributes * sizeof(tSpiderValue*) );
91 * \brief Create an integer object
93 tSpiderValue *SpiderScript_CreateInteger(uint64_t Value)
95 tSpiderValue *ret = malloc( sizeof(tSpiderValue) );
96 ret->Type = SS_DATATYPE_INTEGER;
97 ret->ReferenceCount = 1;
103 * \brief Create an real number object
105 tSpiderValue *SpiderScript_CreateReal(double Value)
107 tSpiderValue *ret = malloc( sizeof(tSpiderValue) );
108 ret->Type = SS_DATATYPE_REAL;
109 ret->ReferenceCount = 1;
115 * \brief Create an string object
117 tSpiderValue *SpiderScript_CreateString(int Length, const char *Data)
119 tSpiderValue *ret = malloc( sizeof(tSpiderValue) + Length + 1 );
120 ret->Type = SS_DATATYPE_STRING;
121 ret->ReferenceCount = 1;
122 ret->String.Length = Length;
124 memcpy(ret->String.Data, Data, Length);
126 memset(ret->String.Data, 0, Length);
127 ret->String.Data[Length] = '\0';
132 * \brief Concatenate two strings
134 tSpiderValue *Object_StringConcat(const tSpiderValue *Str1, const tSpiderValue *Str2)
138 if(Str1) newLen += Str1->String.Length;
139 if(Str2) newLen += Str2->String.Length;
140 ret = malloc( sizeof(tSpiderValue) + newLen + 1 );
141 ret->Type = SS_DATATYPE_STRING;
142 ret->ReferenceCount = 1;
143 ret->String.Length = newLen;
145 memcpy(ret->String.Data, Str1->String.Data, Str1->String.Length);
148 memcpy(ret->String.Data+Str1->String.Length, Str2->String.Data, Str2->String.Length);
150 memcpy(ret->String.Data, Str2->String.Data, Str2->String.Length);
152 ret->String.Data[ newLen ] = '\0';
157 * \brief Cast one object to another
158 * \brief Type Destination type
159 * \brief Source Input data
161 tSpiderValue *SpiderScript_CastValueTo(int Type, tSpiderValue *Source)
163 tSpiderValue *ret = ERRPTR;
170 case SS_DATATYPE_INTEGER: return SpiderScript_CreateInteger(0);
171 case SS_DATATYPE_REAL: return SpiderScript_CreateReal(0);
172 case SS_DATATYPE_STRING: return SpiderScript_CreateString(4, "null");
177 // Check if anything needs to be done
178 if( Source->Type == Type ) {
179 Object_Reference(Source);
185 printf("Casting %i ", Source->Type);
188 case SS_DATATYPE_INTEGER: printf("0x%lx", Source->Integer); break;
189 case SS_DATATYPE_STRING: printf("\"%s\"", Source->String.Data); break;
190 case SS_DATATYPE_REAL: printf("%f", Source->Real); break;
193 printf(" to %i\n", Type);
198 if( Source->Type == SS_DATATYPE_OBJECT )
200 const char *name = NULL;
203 case SS_DATATYPE_INTEGER: name = "cast Integer"; break;
204 case SS_DATATYPE_REAL: name = "cast Real"; break;
205 case SS_DATATYPE_STRING: name = "cast String"; break;
206 case SS_DATATYPE_ARRAY: name = "cast Array"; break;
208 AST_RuntimeError(NULL, "Invalid cast to %i from Object", Type);
213 ret = Object_ExecuteMethod(Left->Object, fcnname, Right);
216 // Fall through and try casting (which will usually fail)
221 switch( (enum eSpiderScript_DataTypes)Type )
223 case SS_DATATYPE_UNDEF:
224 case SS_DATATYPE_ARRAY:
225 case SS_DATATYPE_OPAQUE:
226 case SS_DATATYPE_OBJECT:
227 AST_RuntimeError(NULL, "Invalid cast to %i", Type);
230 case SS_DATATYPE_INTEGER:
231 ret = malloc(sizeof(tSpiderValue));
232 ret->Type = SS_DATATYPE_INTEGER;
233 ret->ReferenceCount = 1;
236 case SS_DATATYPE_INTEGER: break; // Handled above
237 case SS_DATATYPE_STRING: ret->Integer = atoi(Source->String.Data); break;
238 case SS_DATATYPE_REAL: ret->Integer = Source->Real; break;
240 AST_RuntimeError(NULL, "Invalid cast from %i to Integer", Source->Type);
247 case SS_DATATYPE_REAL:
248 ret = malloc(sizeof(tSpiderValue));
249 ret->Type = SS_DATATYPE_REAL;
250 ret->ReferenceCount = 1;
253 case SS_DATATYPE_STRING: ret->Real = atof(Source->String.Data); break;
254 case SS_DATATYPE_INTEGER: ret->Real = Source->Integer; break;
256 AST_RuntimeError(NULL, "Invalid cast from %i to Real", Source->Type);
263 case SS_DATATYPE_STRING:
266 case SS_DATATYPE_INTEGER: len = snprintf(NULL, 0, "%li", Source->Integer); break;
267 case SS_DATATYPE_REAL: len = snprintf(NULL, 0, "%g", Source->Real); break;
270 ret = malloc(sizeof(tSpiderValue) + len + 1);
271 ret->Type = SS_DATATYPE_STRING;
272 ret->ReferenceCount = 1;
273 ret->String.Length = len;
276 case SS_DATATYPE_INTEGER: sprintf(ret->String.Data, "%li", Source->Integer); break;
277 case SS_DATATYPE_REAL:
278 sprintf(ret->String.Data, "%g", Source->Real); break;
280 AST_RuntimeError(NULL, "Invalid cast from %i to String", Source->Type);
288 AST_RuntimeError(NULL, "BUG - BUG REPORT: Unimplemented cast target %i", Type);
297 * \brief Condenses a value down to a boolean
299 int SpiderScript_IsValueTrue(tSpiderValue *Value)
301 if( Value == ERRPTR ) return 0;
302 if( Value == NULL ) return 0;
304 switch( (enum eSpiderScript_DataTypes)Value->Type )
306 case SS_DATATYPE_UNDEF:
309 case SS_DATATYPE_INTEGER:
310 return !!Value->Integer;
312 case SS_DATATYPE_REAL:
313 return (-.5f < Value->Real && Value->Real < 0.5f);
315 case SS_DATATYPE_STRING:
316 return Value->String.Length > 0;
318 case SS_DATATYPE_OBJECT:
319 return Value->Object != NULL;
321 case SS_DATATYPE_OPAQUE:
322 return Value->Opaque.Data != NULL;
324 case SS_DATATYPE_ARRAY:
325 return Value->Array.Length > 0;
327 AST_RuntimeError(NULL, "Unknown type %i in SpiderScript_IsValueTrue", Value->Type);
334 * \brief Free a value
335 * \note Just calls Object_Dereference
337 void SpiderScript_FreeValue(tSpiderValue *Value)
339 Object_Dereference(Value);
343 * \brief Dump a value into a string
344 * \return Heap string
346 char *SpiderScript_DumpValue(tSpiderValue *Value)
349 if( Value == ERRPTR )
350 return strdup("ERRPTR");
352 return strdup("null");
354 switch( (enum eSpiderScript_DataTypes)Value->Type )
356 case SS_DATATYPE_UNDEF: return strdup("undefined");
358 case SS_DATATYPE_INTEGER:
359 ret = malloc( sizeof(Value->Integer)*2 + 3 );
360 sprintf(ret, "0x%lx", Value->Integer);
363 case SS_DATATYPE_REAL:
364 ret = malloc( sprintf(NULL, "%f", Value->Real) + 1 );
365 sprintf(ret, "%f", Value->Real);
368 case SS_DATATYPE_STRING:
369 ret = malloc( Value->String.Length + 3 );
371 strcpy(ret+1, Value->String.Data);
372 ret[Value->String.Length+1] = '"';
373 ret[Value->String.Length+2] = '\0';
376 case SS_DATATYPE_OBJECT:
377 ret = malloc( sprintf(NULL, "{%s *%p}", Value->Object->Type->Name, Value->Object) + 1 );
378 sprintf(ret, "{%s *%p}", Value->Object->Type->Name, Value->Object);
381 case SS_DATATYPE_OPAQUE:
382 ret = malloc( sprintf(NULL, "*%p", Value->Opaque.Data) + 1 );
383 sprintf(ret, "*%p", Value->Opaque.Data);
386 case SS_DATATYPE_ARRAY:
387 return strdup("Array");
390 AST_RuntimeError(NULL, "Unknown type %i in Object_Dump", Value->Type);
397 * \brief Execute a script function
398 * \param Script Script context to execute in
399 * \param Namespace Namespace to search for the function
400 * \param Function Function name to execute
401 * \param NArguments Number of arguments to pass
402 * \param Arguments Arguments passed
404 tSpiderValue *SpiderScript_ExecuteFunction(tSpiderScript *Script,
405 tSpiderNamespace *Namespace, const char *Function,
406 int NArguments, tSpiderValue **Arguments)
408 int bFound = 0; // Used to keep nesting levels down
409 tSpiderValue *ret = ERRPTR;
410 tSpiderFunction *fcn;
412 // First: Find the function in the script
414 tAST_Function *astFcn;
415 for( astFcn = Script->Script->Functions; astFcn; astFcn = astFcn->Next )
417 if( strcmp(astFcn->Name, Function) == 0 )
427 // Build a block State
431 bs.BaseNamespace = &Script->Variant->RootNamespace;
432 bs.CurNamespace = NULL;
434 bs.Ident = giNextBlockIdent ++;
437 for( arg = astFcn->Arguments; arg; arg = arg->NextSibling, i++ )
439 if( i >= NArguments ) break; // TODO: Return gracefully
442 arg->DefVar.DataType, arg->DefVar.Name,
447 ret = AST_ExecuteNode(&bs, astFcn->Code);
450 Object_Dereference(ret); // Dereference output of last block statement
451 ret = bs.RetVal; // Set to return value of block
457 tAST_Variable *nextVar = bs.FirstVar->Next;
458 Variable_Destroy( bs.FirstVar );
459 bs.FirstVar = nextVar;
464 // Didn't find it in script?
467 fcn = NULL; // Just to allow the below code to be neat
469 // Second: Scan current namespace
470 if( !fcn && Namespace )
472 for( fcn = Namespace->Functions; fcn; fcn = fcn->Next )
474 if( strcmp( fcn->Name, Function ) == 0 )
479 // Third: Search the variant's global exports
482 for( fcn = Script->Variant->Functions; fcn; fcn = fcn->Next )
484 if( strcmp( fcn->Name, Function ) == 0 )
489 // Fourth: Search language exports
492 for( fcn = gpExports_First; fcn; fcn = fcn->Next )
494 if( strcmp( fcn->Name, Function ) == 0 )
502 // TODO: Type Checking
503 ret = fcn->Handler( Script, NArguments, Arguments );
511 fprintf(stderr, "Undefined reference to function '%s' (ns='%s')\n",
512 Function, Namespace->Name);
520 * \brief Execute an object method function
521 * \param Script Script context to execute in
522 * \param Object Object in which to find the method
523 * \param MethodName Name of method to call
524 * \param NArguments Number of arguments to pass
525 * \param Arguments Arguments passed
527 tSpiderValue *SpiderScript_ExecuteMethod(tSpiderScript *Script,
528 tSpiderObject *Object, const char *MethodName,
529 int NArguments, tSpiderValue **Arguments)
531 tSpiderFunction *fcn;
533 tSpiderValue *newargs[NArguments+1];
536 // TODO: Support program defined objects
538 // Search for the function
539 for( fcn = Object->Type->Methods; fcn; fcn = fcn->Next )
541 if( strcmp(fcn->Name, MethodName) == 0 )
547 AST_RuntimeError(NULL, "Class '%s' does not have a method '%s'",
548 Object->Type->Name, MethodName);
552 // Create the "this" argument
553 this.Type = SS_DATATYPE_OBJECT;
554 this.ReferenceCount = 1;
555 this.Object = Object;
557 memcpy(&newargs[1], Arguments, NArguments*sizeof(tSpiderValue*));
559 // Check the type of the arguments
560 for( i = 0; fcn->ArgTypes[i]; i ++ )
562 if( i >= NArguments ) {
563 for( ; fcn->ArgTypes[i]; i ++ ) ;
564 AST_RuntimeError(NULL, "Argument count mismatch (%i passed, %i expected)",
568 if( Arguments[i] && Arguments[i]->Type != fcn->ArgTypes[i] )
570 AST_RuntimeError(NULL, "Argument type mismatch (%i, expected %i)",
571 Arguments[i]->Type, fcn->ArgTypes[i]);
577 return fcn->Handler(Script, NArguments+1, newargs);
581 * \brief Execute a script function
582 * \param Script Script context to execute in
583 * \param Function Function name to execute
584 * \param NArguments Number of arguments to pass
585 * \param Arguments Arguments passed
587 tSpiderValue *SpiderScript_CreateObject(tSpiderScript *Script,
588 tSpiderNamespace *Namespace, const char *ClassName,
589 int NArguments, tSpiderValue **Arguments)
591 int bFound = 0; // Used to keep nesting levels down
592 tSpiderValue *ret = ERRPTR;
593 tSpiderObjectDef *class;
595 // First: Find the function in the script
596 // TODO: Implement script-defined classes
599 tAST_Function *astClass;
600 for( astClass = Script->Script->Classes; astClass; astClass = astClass->Next )
602 if( strcmp(astClass->Name, ClassName) == 0 )
612 // Build a block State
616 bs.BaseNamespace = &Script->Variant->RootNamespace;
617 bs.CurNamespace = NULL;
619 bs.Ident = giNextBlockIdent ++;
621 for( arg = astFcn->Arguments; arg; arg = arg->NextSibling, i++ )
623 if( i >= NArguments ) break; // TODO: Return gracefully
626 arg->DefVar.DataType, arg->DefVar.Name,
631 ret = AST_ExecuteNode(&bs, astFcn->Code);
634 Object_Dereference(ret); // Dereference output of last block statement
635 ret = bs.RetVal; // Set to return value of block
641 tAST_Variable *nextVar = bs.FirstVar->Next;
642 Variable_Destroy( bs.FirstVar );
643 bs.FirstVar = nextVar;
649 // Didn't find it in script?
652 class = NULL; // Just to allow the below code to be neat
655 // Namespace = &Script->Variant->RootNamespace;
657 // Second: Scan current namespace
658 if( !class && Namespace )
660 for( class = Namespace->Classes; class; class = class->Next )
662 if( strcmp( class->Name, ClassName ) == 0 )
668 // Third: Search the variant's global exports
671 for( class = Script->Variant->Classes; class; class = fcn->Next )
673 if( strcmp( class->Name, Function ) == 0 )
680 // Fourth: Search language exports
683 for( class = gpExports_First; class; class = fcn->Next )
685 if( strcmp( class->Name, ClassName ) == 0 )
695 // TODO: Type Checking
698 obj = class->Constructor( NArguments, Arguments );
699 if( obj == NULL || obj == ERRPTR )
702 // Creatue return object
703 ret = malloc( sizeof(tSpiderValue) );
704 ret->Type = SS_DATATYPE_OBJECT;
705 ret->ReferenceCount = 1;
714 fprintf(stderr, "Undefined reference to class '%s'\n", ClassName);
723 * \brief Execute an AST node and return its value
724 * \param Block Execution context
725 * \param Node Node to execute
727 tSpiderValue *AST_ExecuteNode(tAST_BlockState *Block, tAST_Node *Node)
730 tSpiderValue *ret = NULL, *tmpobj;
731 tSpiderValue *op1, *op2; // Binary operations
732 int cmp; // Used in comparisons
745 tAST_BlockState blockInfo;
746 blockInfo.Parent = Block;
747 blockInfo.Script = Block->Script;
748 blockInfo.FirstVar = NULL;
749 blockInfo.RetVal = NULL;
750 blockInfo.BaseNamespace = Block->BaseNamespace;
751 blockInfo.CurNamespace = NULL;
752 blockInfo.BreakTarget = NULL;
753 blockInfo.Ident = giNextBlockIdent ++;
755 // Loop over all nodes, or until the return value is set
756 for(node = Node->Block.FirstChild;
757 node && !blockInfo.RetVal && !blockInfo.BreakTarget;
758 node = node->NextSibling )
760 ret = AST_ExecuteNode(&blockInfo, node);
761 if(ret == ERRPTR) break; // Error check
762 if(ret != NULL) Object_Dereference(ret); // Free unused value
764 // Clean up variables
765 while(blockInfo.FirstVar)
767 tAST_Variable *nextVar = blockInfo.FirstVar->Next;
768 Variable_Destroy( blockInfo.FirstVar );
769 blockInfo.FirstVar = nextVar;
771 // Clear ret if not an error
772 if(ret != ERRPTR) ret = NULL;
774 // Set parent's return value if needed
775 if( blockInfo.RetVal )
776 Block->RetVal = blockInfo.RetVal;
777 if( blockInfo.BreakTarget ) {
778 Block->BreakTarget = blockInfo.BreakTarget;
779 Block->BreakType = blockInfo.BreakType;
782 // TODO: Unset break if break type deontes a block break
788 case NODETYPE_ASSIGN:
789 // TODO: Support assigning to object attributes
790 if( Node->Assign.Dest->Type != NODETYPE_VARIABLE ) {
791 AST_RuntimeError(Node, "LVALUE of assignment is not a variable");
794 ret = AST_ExecuteNode(Block, Node->Assign.Value);
795 if(ret == ERRPTR) return ERRPTR;
797 // Perform assignment operation
798 if( Node->Assign.Operation != NODETYPE_NOP )
800 tSpiderValue *varVal = Variable_GetValue(Block, Node->Assign.Dest);
802 value = AST_ExecuteNode_BinOp(Block, Node, Node->Assign.Operation, varVal, ret);
803 if(value == ERRPTR) return ERRPTR;
804 if(ret) Object_Dereference(ret);
805 if(varVal) Object_Dereference(varVal);
809 // Set the variable value
810 if( Variable_SetValue( Block, Node->Assign.Dest, ret ) ) {
811 Object_Dereference( ret );
816 // Post increment/decrement
817 case NODETYPE_POSTINC:
818 case NODETYPE_POSTDEC:
820 tSpiderValue *varVal, *value;
821 static tSpiderValue one = {
822 .Type = SS_DATATYPE_INTEGER,
827 // TODO: Support assigning to object attributes
828 if( Node->UniOp.Value->Type != NODETYPE_VARIABLE ) {
829 AST_RuntimeError(Node, "LVALUE of assignment is not a variable");
833 // Get values (current variable contents and a static one)
834 varVal = Variable_GetValue(Block, Node->UniOp.Value);
836 if( Node->Type == NODETYPE_POSTDEC )
837 value = AST_ExecuteNode_BinOp(Block, Node, NODETYPE_SUBTRACT, varVal, &one);
839 value = AST_ExecuteNode_BinOp(Block, Node, NODETYPE_ADD, varVal, &one);
840 if( value == ERRPTR )
845 if( Variable_SetValue( Block, Node->UniOp.Value, value ) ) {
846 Object_Dereference( ret );
849 Object_Dereference( value );
854 case NODETYPE_METHODCALL:
855 case NODETYPE_FUNCTIONCALL:
856 case NODETYPE_CREATEOBJECT:
857 // Logical block (used to allocate `params`)
859 tSpiderNamespace *ns = Block->CurNamespace;
860 tSpiderValue *params[Node->FunctionCall.NumArgs];
862 for(node = Node->FunctionCall.FirstArg; node; node = node->NextSibling)
864 params[i] = AST_ExecuteNode(Block, node);
865 if( params[i] == ERRPTR ) {
866 while(i--) Object_Dereference(params[i]);
873 if( !ns ) ns = Block->BaseNamespace;
876 if( Node->Type == NODETYPE_CREATEOBJECT )
878 ret = SpiderScript_CreateObject(Block->Script,
880 Node->FunctionCall.Name,
881 Node->FunctionCall.NumArgs, params
884 else if( Node->Type == NODETYPE_METHODCALL )
886 tSpiderValue *obj = AST_ExecuteNode(Block, Node->FunctionCall.Object);
887 if( !obj || obj == ERRPTR || obj->Type != SS_DATATYPE_OBJECT ) {
888 AST_RuntimeError(Node->FunctionCall.Object,
889 "Type Mismatch - Required SS_DATATYPE_OBJECT for method call");
890 while(i--) Object_Dereference(params[i]);
894 ret = SpiderScript_ExecuteMethod(Block->Script,
895 obj->Object, Node->FunctionCall.Name,
896 Node->FunctionCall.NumArgs, params
898 Object_Dereference(obj);
902 ret = SpiderScript_ExecuteFunction(Block->Script,
903 ns, Node->FunctionCall.Name,
904 Node->FunctionCall.NumArgs, params
909 // Dereference parameters
910 while(i--) Object_Dereference(params[i]);
918 ret = AST_ExecuteNode(Block, Node->If.Condition);
919 if( ret == ERRPTR ) break;
920 if( SpiderScript_IsValueTrue(ret) ) {
921 tmpobj = AST_ExecuteNode(Block, Node->If.True);
924 tmpobj = AST_ExecuteNode(Block, Node->If.False);
926 Object_Dereference(ret);
927 if( tmpobj == ERRPTR ) return ERRPTR;
928 Object_Dereference(tmpobj);
935 ret = AST_ExecuteNode(Block, Node->For.Init);
936 if(ret == ERRPTR) break;
938 // Check initial condition
939 if( !Node->For.bCheckAfter )
941 Object_Dereference(ret);
943 ret = AST_ExecuteNode(Block, Node->For.Condition);
944 if(ret == ERRPTR) return ERRPTR;
945 if(!SpiderScript_IsValueTrue(ret)) {
946 Object_Dereference(ret);
955 Object_Dereference(ret);
958 ret = AST_ExecuteNode(Block, Node->For.Code);
959 if(ret == ERRPTR) return ERRPTR;
960 Object_Dereference(ret);
962 if(Block->BreakTarget)
964 if( Block->BreakTarget[0] == '\0' || strcmp(Block->BreakTarget, Node->For.Tag) == 0 )
967 free((void*)Block->BreakTarget); Block->BreakTarget = NULL;
968 if( Block->BreakType == NODETYPE_CONTINUE ) {
969 // Continue, just keep going
975 break; // Break out of this loop
979 ret = AST_ExecuteNode(Block, Node->For.Increment);
980 if(ret == ERRPTR) return ERRPTR;
981 Object_Dereference(ret);
984 ret = AST_ExecuteNode(Block, Node->For.Condition);
985 if(ret == ERRPTR) return ERRPTR;
986 if(!SpiderScript_IsValueTrue(ret)) break;
988 Object_Dereference(ret);
993 case NODETYPE_RETURN:
994 ret = AST_ExecuteNode(Block, Node->UniOp.Value);
995 if(ret == ERRPTR) break;
996 Block->RetVal = ret; // Return value set
997 ret = NULL; // the `return` statement does not return a value
1000 case NODETYPE_BREAK:
1001 case NODETYPE_CONTINUE:
1002 Block->BreakTarget = strdup(Node->Variable.Name);
1003 Block->BreakType = Node->Type;
1006 // Define a variable
1007 case NODETYPE_DEFVAR:
1008 if( Node->DefVar.InitialValue ) {
1009 tmpobj = AST_ExecuteNode(Block, Node->DefVar.InitialValue);
1010 if(tmpobj == ERRPTR) return ERRPTR;
1016 if( Variable_Define(Block, Node->DefVar.DataType, Node->DefVar.Name, tmpobj) == ERRPTR )
1018 Object_Dereference(tmpobj);
1022 case NODETYPE_SCOPE:
1024 tSpiderNamespace *ns;
1026 // Set current namespace if unset
1027 if( !Block->CurNamespace )
1028 Block->CurNamespace = Block->BaseNamespace;
1030 // Empty string means use the root namespace
1031 if( Node->Scope.Name[0] == '\0' )
1033 ns = &Block->Script->Variant->RootNamespace;
1037 // Otherwise scan the current namespace for the element
1038 for( ns = Block->CurNamespace->FirstChild; ns; ns = ns->Next )
1040 if( strcmp(ns->Name, Node->Scope.Name) == 0 )
1045 AST_RuntimeError(Node, "Unknown namespace '%s'", Node->Scope.Name);
1049 Block->CurNamespace = ns;
1051 ret = AST_ExecuteNode(Block, Node->Scope.Element);
1056 case NODETYPE_VARIABLE:
1057 ret = Variable_GetValue( Block, Node );
1060 // Element of an Object
1061 case NODETYPE_ELEMENT:
1062 tmpobj = AST_ExecuteNode( Block, Node->Scope.Element );
1063 if(tmpobj == ERRPTR) return ERRPTR;
1064 if( !tmpobj || tmpobj->Type != SS_DATATYPE_OBJECT )
1066 AST_RuntimeError(Node->Scope.Element, "Unable to dereference a non-object");
1071 for( i = 0; i < tmpobj->Object->Type->NAttributes; i ++ )
1073 if( strcmp(Node->Scope.Name, tmpobj->Object->Type->AttributeDefs[i].Name) == 0 )
1075 ret = tmpobj->Object->Attributes[i];
1076 Object_Reference(ret);
1080 if( i == tmpobj->Object->Type->NAttributes )
1082 AST_RuntimeError(Node->Scope.Element, "Unknown attribute '%s' of class '%s'",
1083 Node->Scope.Name, tmpobj->Object->Type->Name);
1088 // Cast a value to another
1091 tmpobj = AST_ExecuteNode(Block, Node->Cast.Value);
1092 if(tmpobj == ERRPTR) return ERRPTR;
1093 ret = SpiderScript_CastValueTo( Node->Cast.DataType, tmpobj );
1094 Object_Dereference(tmpobj);
1098 // Index into an array
1099 case NODETYPE_INDEX:
1100 op1 = AST_ExecuteNode(Block, Node->BinOp.Left); // Array
1101 if(op1 == ERRPTR) return ERRPTR;
1102 op2 = AST_ExecuteNode(Block, Node->BinOp.Right); // Offset
1104 Object_Dereference(op1);
1108 if( !op1 || op1->Type != SS_DATATYPE_ARRAY )
1110 // TODO: Implement "operator []" on objects
1111 AST_RuntimeError(Node, "Indexing non-array");
1116 if( (!op2 || op2->Type != SS_DATATYPE_INTEGER) && !Block->Script->Variant->bImplicitCasts ) {
1117 AST_RuntimeError(Node, "Array index is not an integer");
1122 if( !op2 || op2->Type != SS_DATATYPE_INTEGER )
1124 tmpobj = SpiderScript_CastValueTo(SS_DATATYPE_INTEGER, op2);
1125 Object_Dereference(op2);
1129 if( op2->Integer >= op1->Array.Length ) {
1130 AST_RuntimeError(Node, "Array index out of bounds %i >= %i",
1131 op2->Integer, op1->Array.Length);
1136 ret = op1->Array.Items[ op2->Integer ];
1137 Object_Reference(ret);
1139 Object_Dereference(op1);
1140 Object_Dereference(op2);
1143 // TODO: Implement runtime constants
1144 case NODETYPE_CONSTANT:
1145 // TODO: Scan namespace for function
1146 AST_RuntimeError(Node, "TODO - Runtime Constants");
1151 case NODETYPE_STRING:
1152 case NODETYPE_INTEGER:
1154 ret = &Node->Constant;
1155 Object_Reference(ret);
1158 // --- Operations ---
1159 // Boolean Operations
1160 case NODETYPE_LOGICALNOT: // Logical NOT (!)
1161 op1 = AST_ExecuteNode(Block, Node->UniOp.Value);
1162 if(op1 == ERRPTR) return ERRPTR;
1163 ret = SpiderScript_CreateInteger( !SpiderScript_IsValueTrue(op1) );
1164 Object_Dereference(op1);
1166 case NODETYPE_LOGICALAND: // Logical AND (&&)
1167 case NODETYPE_LOGICALOR: // Logical OR (||)
1168 case NODETYPE_LOGICALXOR: // Logical XOR (^^)
1169 op1 = AST_ExecuteNode(Block, Node->BinOp.Left);
1170 if(op1 == ERRPTR) return ERRPTR;
1171 op2 = AST_ExecuteNode(Block, Node->BinOp.Right);
1173 Object_Dereference(op1);
1177 switch( Node->Type )
1179 case NODETYPE_LOGICALAND:
1180 ret = SpiderScript_CreateInteger( SpiderScript_IsValueTrue(op1) && SpiderScript_IsValueTrue(op2) );
1182 case NODETYPE_LOGICALOR:
1183 ret = SpiderScript_CreateInteger( SpiderScript_IsValueTrue(op1) || SpiderScript_IsValueTrue(op2) );
1185 case NODETYPE_LOGICALXOR:
1186 ret = SpiderScript_CreateInteger( SpiderScript_IsValueTrue(op1) ^ SpiderScript_IsValueTrue(op2) );
1191 // Free intermediate objects
1192 Object_Dereference(op1);
1193 Object_Dereference(op2);
1197 case NODETYPE_EQUALS:
1198 case NODETYPE_LESSTHAN:
1199 case NODETYPE_GREATERTHAN:
1200 case NODETYPE_LESSTHANEQUAL:
1201 case NODETYPE_GREATERTHANEQUAL:
1202 op1 = AST_ExecuteNode(Block, Node->BinOp.Left);
1203 if(op1 == ERRPTR) return ERRPTR;
1204 op2 = AST_ExecuteNode(Block, Node->BinOp.Right);
1206 Object_Dereference(op1);
1211 if( !op1 || !op2 ) {
1212 AST_RuntimeError(Node, "NULL Comparison (%p and %p)", op1, op2);
1213 if(op1) Object_Dereference(op1);
1214 if(op2) Object_Dereference(op2);
1215 ret = SpiderScript_CreateInteger( !op1 && !op2 );
1220 if( op1->Type != op2->Type ) {
1221 // If dynamically typed, convert op2 to op1's type
1222 if(Block->Script->Variant->bImplicitCasts)
1225 op2 = SpiderScript_CastValueTo(op1->Type, op2);
1226 Object_Dereference(tmpobj);
1228 Object_Dereference(op1);
1232 // If statically typed, this should never happen, but catch it anyway
1234 AST_RuntimeError(Node, "Statically typed implicit cast %i <op> %i",
1235 op1->Type, op2->Type);
1243 // - String Compare (does a strcmp, well memcmp)
1244 case SS_DATATYPE_STRING:
1245 // Call memcmp to do most of the work
1247 op1->String.Data, op2->String.Data,
1248 (op1->String.Length < op2->String.Length) ? op1->String.Length : op2->String.Length
1250 // Handle reaching the end of the string
1252 if( op1->String.Length == op2->String.Length )
1254 else if( op1->String.Length < op2->String.Length )
1261 // - Integer Comparisons
1262 case SS_DATATYPE_INTEGER:
1263 if( op1->Integer == op2->Integer )
1265 else if( op1->Integer < op2->Integer )
1270 // - Real Number Comparisons
1271 case SS_DATATYPE_REAL:
1272 cmp = (op1->Real - op2->Real) / op2->Real * 10000; // < 0.1% difference is equality
1275 AST_RuntimeError(Node, "TODO - Comparison of type %i", op1->Type);
1280 // Free intermediate objects
1281 Object_Dereference(op1);
1282 Object_Dereference(op2);
1291 case NODETYPE_EQUALS: ret = SpiderScript_CreateInteger(cmp == 0); break;
1292 case NODETYPE_LESSTHAN: ret = SpiderScript_CreateInteger(cmp < 0); break;
1293 case NODETYPE_GREATERTHAN: ret = SpiderScript_CreateInteger(cmp > 0); break;
1294 case NODETYPE_LESSTHANEQUAL: ret = SpiderScript_CreateInteger(cmp <= 0); break;
1295 case NODETYPE_GREATERTHANEQUAL: ret = SpiderScript_CreateInteger(cmp >= 0); break;
1297 AST_RuntimeError(Node, "Exec,CmpOp unknown op %i", Node->Type);
1303 // General Unary Operations
1304 case NODETYPE_BWNOT: // Bitwise NOT (~)
1305 case NODETYPE_NEGATE: // Negation (-)
1306 op1 = AST_ExecuteNode(Block, Node->UniOp.Value);
1307 if(op1 == ERRPTR) return ERRPTR;
1308 ret = AST_ExecuteNode_UniOp(Block, Node, Node->Type, op1);
1309 Object_Dereference(op1);
1312 // General Binary Operations
1314 case NODETYPE_SUBTRACT:
1315 case NODETYPE_MULTIPLY:
1316 case NODETYPE_DIVIDE:
1317 case NODETYPE_MODULO:
1318 case NODETYPE_BWAND:
1320 case NODETYPE_BWXOR:
1321 case NODETYPE_BITSHIFTLEFT:
1322 case NODETYPE_BITSHIFTRIGHT:
1323 case NODETYPE_BITROTATELEFT:
1325 op1 = AST_ExecuteNode(Block, Node->BinOp.Left);
1326 if(op1 == ERRPTR) return ERRPTR;
1327 op2 = AST_ExecuteNode(Block, Node->BinOp.Right);
1329 Object_Dereference(op1);
1333 ret = AST_ExecuteNode_BinOp(Block, Node, Node->Type, op1, op2);
1335 // Free intermediate objects
1336 Object_Dereference(op1);
1337 Object_Dereference(op2);
1342 // AST_RuntimeError(Node, "BUG - SpiderScript AST_ExecuteNode Unimplemented %i", Node->Type);
1346 // Reset namespace when no longer needed
1347 if( Node->Type != NODETYPE_SCOPE )
1348 Block->CurNamespace = NULL;
1350 #if TRACE_NODE_RETURNS
1351 if(ret && ret != ERRPTR) {
1352 AST_RuntimeError(Node, "Ret type of %p %i is %i", Node, Node->Type, ret->Type);
1355 AST_RuntimeError(Node, "Ret type of %p %i is %p", Node, Node->Type, ret);
1362 tSpiderValue *AST_ExecuteNode_UniOp(tAST_BlockState *Block, tAST_Node *Node, int Operation, tSpiderValue *Value)
1366 if( Value->Type == SS_DATATYPE_OBJECT )
1368 const char *fcnname;
1371 case NODETYPE_NEGATE: fcnname = "-ve"; break;
1372 case NODETYPE_BWNOT: fcnname = "~"; break;
1373 default: fcnname = NULL; break;
1378 ret = Object_ExecuteMethod(Value->Object, fcnname, );
1381 // Fall through and try casting (which will usually fail)
1387 // Integer Operations
1388 case SS_DATATYPE_INTEGER:
1391 case NODETYPE_NEGATE: ret = SpiderScript_CreateInteger( -Value->Integer ); break;
1392 case NODETYPE_BWNOT: ret = SpiderScript_CreateInteger( ~Value->Integer ); break;
1394 AST_RuntimeError(Node, "SpiderScript internal error: Exec,UniOP,Integer unknown op %i", Operation);
1399 // Real number Operations
1400 case SS_DATATYPE_REAL:
1403 case NODETYPE_NEGATE: ret = SpiderScript_CreateInteger( -Value->Real ); break;
1405 AST_RuntimeError(Node, "SpiderScript internal error: Exec,UniOP,Real unknown op %i", Operation);
1412 AST_RuntimeError(NULL, "Invalid operation (%i) on type (%i)", Operation, Value->Type);
1420 tSpiderValue *AST_ExecuteNode_BinOp(tAST_BlockState *Block, tAST_Node *Node, int Operation, tSpiderValue *Left, tSpiderValue *Right)
1422 tSpiderValue *preCastValue = Right;
1426 if( Left && Right && Left->Type != Right->Type )
1430 // - Operator overload functions
1431 if( Left->Type == SS_DATATYPE_OBJECT )
1433 const char *fcnname;
1436 case NODETYPE_ADD: fcnname = "+"; break;
1437 case NODETYPE_SUBTRACT: fcnname = "-"; break;
1438 case NODETYPE_MULTIPLY: fcnname = "*"; break;
1439 case NODETYPE_DIVIDE: fcnname = "/"; break;
1440 case NODETYPE_MODULO: fcnname = "%"; break;
1441 case NODETYPE_BWAND: fcnname = "&"; break;
1442 case NODETYPE_BWOR: fcnname = "|"; break;
1443 case NODETYPE_BWXOR: fcnname = "^"; break;
1444 case NODETYPE_BITSHIFTLEFT: fcnname = "<<"; break;
1445 case NODETYPE_BITSHIFTRIGHT:fcnname = ">>"; break;
1446 case NODETYPE_BITROTATELEFT:fcnname = "<<<"; break;
1447 default: fcnname = NULL; break;
1452 ret = Object_ExecuteMethod(Left->Object, fcnname, Right);
1455 // Fall through and try casting (which will usually fail)
1460 // If implicit casts are allowed, convert Right to Left's type
1461 if(Block->Script->Variant->bImplicitCasts)
1463 Right = SpiderScript_CastValueTo(Left->Type, Right);
1467 // If statically typed, this should never happen, but catch it anyway
1469 AST_RuntimeError(Node, "Implicit cast not allowed (from %i to %i)", Right->Type, Left->Type);
1475 if( Left == NULL || Right == NULL ) {
1476 if(Right && Right != preCastValue) free(Right);
1483 // String Concatenation
1484 case SS_DATATYPE_STRING:
1487 case NODETYPE_ADD: // Concatenate
1488 ret = Object_StringConcat(Left, Right);
1491 AST_RuntimeError(Node, "SpiderScript internal error: Exec,BinOP,String unknown op %i", Operation);
1496 // Integer Operations
1497 case SS_DATATYPE_INTEGER:
1500 case NODETYPE_ADD: ret = SpiderScript_CreateInteger( Left->Integer + Right->Integer ); break;
1501 case NODETYPE_SUBTRACT: ret = SpiderScript_CreateInteger( Left->Integer - Right->Integer ); break;
1502 case NODETYPE_MULTIPLY: ret = SpiderScript_CreateInteger( Left->Integer * Right->Integer ); break;
1503 case NODETYPE_DIVIDE: ret = SpiderScript_CreateInteger( Left->Integer / Right->Integer ); break;
1504 case NODETYPE_MODULO: ret = SpiderScript_CreateInteger( Left->Integer % Right->Integer ); break;
1505 case NODETYPE_BWAND: ret = SpiderScript_CreateInteger( Left->Integer & Right->Integer ); break;
1506 case NODETYPE_BWOR: ret = SpiderScript_CreateInteger( Left->Integer | Right->Integer ); break;
1507 case NODETYPE_BWXOR: ret = SpiderScript_CreateInteger( Left->Integer ^ Right->Integer ); break;
1508 case NODETYPE_BITSHIFTLEFT: ret = SpiderScript_CreateInteger( Left->Integer << Right->Integer ); break;
1509 case NODETYPE_BITSHIFTRIGHT:ret = SpiderScript_CreateInteger( Left->Integer >> Right->Integer ); break;
1510 case NODETYPE_BITROTATELEFT:
1511 ret = SpiderScript_CreateInteger( (Left->Integer << Right->Integer) | (Left->Integer >> (64-Right->Integer)) );
1514 AST_RuntimeError(Node, "SpiderScript internal error: Exec,BinOP,Integer unknown op %i", Operation);
1521 case SS_DATATYPE_REAL:
1524 case NODETYPE_ADD: ret = SpiderScript_CreateReal( Left->Real + Right->Real ); break;
1525 case NODETYPE_SUBTRACT: ret = SpiderScript_CreateReal( Left->Real - Right->Real ); break;
1526 case NODETYPE_MULTIPLY: ret = SpiderScript_CreateReal( Left->Real * Right->Real ); break;
1527 case NODETYPE_DIVIDE: ret = SpiderScript_CreateReal( Left->Real / Right->Real ); break;
1529 AST_RuntimeError(Node, "SpiderScript internal error: Exec,BinOP,Real unknown op %i", Operation);
1536 AST_RuntimeError(Node, "BUG - Invalid operation (%i) on type (%i)", Operation, Left->Type);
1541 if(Right && Right != preCastValue) free(Right);
1547 * \brief Define a variable
1548 * \param Block Current block state
1549 * \param Type Type of the variable
1550 * \param Name Name of the variable
1551 * \return Boolean Failure
1553 tAST_Variable *Variable_Define(tAST_BlockState *Block, int Type, const char *Name, tSpiderValue *Value)
1555 tAST_Variable *var, *prev = NULL;
1557 for( var = Block->FirstVar; var; prev = var, var = var->Next )
1559 if( strcmp(var->Name, Name) == 0 ) {
1560 AST_RuntimeError(NULL, "Redefinition of variable '%s'", Name);
1565 var = malloc( sizeof(tAST_Variable) + strlen(Name) + 1 );
1568 var->Object = Value;
1569 if(Value) Object_Reference(Value);
1570 strcpy(var->Name, Name);
1572 if(prev) prev->Next = var;
1573 else Block->FirstVar = var;
1575 //printf("Defined variable %s (%i)\n", Name, Type);
1580 tAST_Variable *Variable_Lookup(tAST_BlockState *Block, tAST_Node *VarNode, int CreateType)
1582 tAST_Variable *var = NULL;
1585 if( VarNode->BlockState == Block && VarNode->BlockIdent == Block->Ident ) {
1586 var = VarNode->ValueCache;
1587 #if TRACE_VAR_LOOKUPS
1588 AST_RuntimeMessage(VarNode, "debug", "Fast var fetch on '%s' %p (%p:%i)",
1589 VarNode->Variable.Name, var,
1590 VarNode->BlockState, VarNode->BlockIdent
1596 tAST_BlockState *bs;
1597 for( bs = Block; bs; bs = bs->Parent )
1599 for( var = bs->FirstVar; var; var = var->Next )
1601 if( strcmp(var->Name, VarNode->Variable.Name) == 0 )
1609 if( Block->Script->Variant->bDyamicTyped && CreateType != SS_DATATYPE_UNDEF ) {
1611 var = Variable_Define(Block, CreateType, VarNode->Variable.Name, NULL);
1615 AST_RuntimeError(VarNode, "Variable '%s' is undefined", VarNode->Variable.Name);
1620 #if TRACE_VAR_LOOKUPS
1621 AST_RuntimeMessage(VarNode, "debug", "Saved variable lookup of '%s' %p (%p:%i)",
1622 VarNode->Variable.Name, var,
1623 Block, Block->Ident);
1626 VarNode->ValueCache = var;
1627 VarNode->BlockState = Block;
1628 VarNode->BlockIdent = Block->Ident;
1635 * \brief Set the value of a variable
1636 * \return Boolean Failure
1638 int Variable_SetValue(tAST_BlockState *Block, tAST_Node *VarNode, tSpiderValue *Value)
1642 var = Variable_Lookup(Block, VarNode, (Value ? Value->Type : SS_DATATYPE_UNDEF));
1644 if( !var ) return -1;
1646 if( !Block->Script->Variant->bDyamicTyped && (Value && var->Type != Value->Type) )
1648 AST_RuntimeError(VarNode, "Type mismatch assigning to '%s'",
1649 VarNode->Variable.Name);
1653 // printf("Assign %p to '%s'\n", Value, var->Name);
1654 Object_Reference(Value);
1655 Object_Dereference(var->Object);
1656 var->Object = Value;
1661 * \brief Get the value of a variable
1663 tSpiderValue *Variable_GetValue(tAST_BlockState *Block, tAST_Node *VarNode)
1665 tAST_Variable *var = Variable_Lookup(Block, VarNode, 0);
1667 if( !var ) return ERRPTR;
1669 Object_Reference(var->Object);
1674 * \brief Destorys a variable
1676 void Variable_Destroy(tAST_Variable *Variable)
1678 // printf("Variable_Destroy: (%p'%s')\n", Variable, Variable->Name);
1679 Object_Dereference(Variable->Object);
1683 void AST_RuntimeMessage(tAST_Node *Node, const char *Type, const char *Format, ...)
1688 fprintf(stderr, "%s:%i: ", Node->File, Node->Line);
1690 fprintf(stderr, "%s: ", Type);
1691 va_start(args, Format);
1692 vfprintf(stderr, Format, args);
1694 fprintf(stderr, "\n");
1696 void AST_RuntimeError(tAST_Node *Node, const char *Format, ...)
1701 fprintf(stderr, "%s:%i: ", Node->File, Node->Line);
1703 fprintf(stderr, "error: ");
1704 va_start(args, Format);
1705 vfprintf(stderr, Format, args);
1707 fprintf(stderr, "\n");