SpiderScript - Moved header to directory, ready to remove from tree

[tpg/acess2.git] / Usermode / Libraries / libspiderscript.so_src / exec_ast.c

diff --git a/Usermode/Libraries/libspiderscript.so_src/exec_ast.c b/Usermode/Libraries/libspiderscript.so_src/exec_ast.c
index c26d887..a9970ee 100644 (file)
--- a/Usermode/Libraries/libspiderscript.so_src/exec_ast.c
+++ b/Usermode/Libraries/libspiderscript.so_src/exec_ast.c
@@ -21,6 +21,7 @@
 tSpiderValue   *AST_ExecuteNode(tAST_BlockState *Block, tAST_Node *Node);
 tSpiderValue   *AST_ExecuteNode_BinOp(tSpiderScript *Script, tAST_Node *Node, int Operation, tSpiderValue *Left, tSpiderValue *Right);
 tSpiderValue   *AST_ExecuteNode_UniOp(tSpiderScript *Script, tAST_Node *Node, int Operation, tSpiderValue *Value);
+tSpiderValue   *AST_ExecuteNode_Index(tSpiderScript *Script, tAST_Node *Node, tSpiderValue *Array, int Index, tSpiderValue *SaveValue);
 // - Variables
 tAST_Variable *Variable_Define(tAST_BlockState *Block, int Type, const char *Name, tSpiderValue *Value);
  int   Variable_SetValue(tAST_BlockState *Block, tAST_Node *VarNode, tSpiderValue *Value);
@@ -446,28 +447,8 @@ tSpiderValue *AST_ExecuteNode(tAST_BlockState *Block, tAST_Node *Node)
        case NODETYPE_ELEMENT:
                tmpobj = AST_ExecuteNode( Block, Node->Scope.Element );
                if(tmpobj == ERRPTR)    return ERRPTR;
-               if( !tmpobj || tmpobj->Type != SS_DATATYPE_OBJECT )
-               {
-                       AST_RuntimeError(Node->Scope.Element, "Unable to dereference a non-object");
-                       ret = ERRPTR;
-                       break ;
-               }
-               
-               for( i = 0; i < tmpobj->Object->Type->NAttributes; i ++ )
-               {
-                       if( strcmp(Node->Scope.Name, tmpobj->Object->Type->AttributeDefs[i].Name) == 0 )
-                       {
-                               ret = tmpobj->Object->Attributes[i];
-                               SpiderScript_ReferenceValue(ret);
-                               break;
-                       }
-               }
-               if( i == tmpobj->Object->Type->NAttributes )
-               {
-                       AST_RuntimeError(Node->Scope.Element, "Unknown attribute '%s' of class '%s'",
-                               Node->Scope.Name, tmpobj->Object->Type->Name);
-                       ret = ERRPTR;
-               }
+
+               ret = AST_ExecuteNode_Element(Block->Script, Node, tmpobj, Node->Scope.Name, ERRPTR);
                break;
 
        // Cast a value to another
@@ -512,27 +493,8 @@ tSpiderValue *AST_ExecuteNode(tAST_BlockState *Block, tAST_Node *Node)
                        break;
                }
 
-               switch( op1->Type )
-               {
-               case SS_DATATYPE_ARRAY:
-                       if( op2->Integer >= op1->Array.Length ) {
-                               AST_RuntimeError(Node, "Array index out of bounds %i >= %i",
-                                       op2->Integer, op1->Array.Length);
-                               ret = ERRPTR;
-                               break;
-                       }
-                       
-                       ret = op1->Array.Items[ op2->Integer ];
-                       SpiderScript_ReferenceValue(ret);
-                       break;
-               
-               default:
-                       // TODO: Implement "operator []" on objects
-                       AST_RuntimeError(Node, "Indexing non-array");
-                       ret = ERRPTR;
-                       break;
-               }
-               
+               ret = AST_ExecuteNode_Index(Block->Script, Node, op1, op2->Integer, ERRPTR);
+
                SpiderScript_DereferenceValue(op1);
                SpiderScript_DereferenceValue(op2);
                break;
@@ -950,6 +912,99 @@ tSpiderValue *AST_ExecuteNode_BinOp(tSpiderScript *Script, tAST_Node *Node, int
        return ret;
 }
 
+tSpiderValue *AST_ExecuteNode_Index(tSpiderScript *Script, tAST_Node *Node,
+       tSpiderValue *Array, int Index, tSpiderValue *SaveValue)
+{
+       // Quick sanity check
+       if( !Array )
+       {
+               AST_RuntimeError(Node, "Indexing NULL, not a good idea");
+               return ERRPTR;
+       }
+
+       // Array?
+       if( SS_GETARRAYDEPTH(Array->Type) )
+       {
+               if( Index < 0 || Index >= Array->Array.Length ) {
+                       AST_RuntimeError(Node, "Array index out of bounds %i not in (0, %i]",
+                               Index, Array->Array.Length);
+                       return ERRPTR;
+               }
+               
+               if( SaveValue != ERRPTR )
+               {
+                       if( SaveValue && SaveValue->Type != SS_DOWNARRAY(Array->Type) ) {
+                               // TODO: Implicit casting
+                               AST_RuntimeError(Node, "Type mismatch assiging to array element");
+                               return ERRPTR;
+                       }
+                       SpiderScript_DereferenceValue( Array->Array.Items[Index] );
+                       Array->Array.Items[Index] = SaveValue;
+                       SpiderScript_ReferenceValue( Array->Array.Items[Index] );
+                       return NULL;
+               }
+               else
+               {
+                       SpiderScript_ReferenceValue( Array->Array.Items[Index] );
+                       return Array->Array.Items[Index];
+               }
+       }
+       else
+       {
+               AST_RuntimeError(Node, "TODO - Implement indexing on non-arrays (type = %x)",
+                       Array->Type);
+               return ERRPTR;
+       }
+}
+
+/**
+ * \brief Get/Set the value of an element/attribute of a class
+ * \param Script       Executing script
+ * \param Node Current execution node (only used for AST_RuntimeError)
+ * \param Object       Object value
+ * \param ElementName  Name of the attribute to be accessed
+ * \param SaveValue    Value to set the element to (if ERRPTR, element value is returned)
+ */
+tSpiderValue *AST_ExecuteNode_Element(tSpiderScript *Script, tAST_Node *Node,
+       tSpiderValue *Object, const char *ElementName, tSpiderValue *SaveValue)
+{
+        int    i;
+       tSpiderValue    *ret;   
+
+       if( !Object ) {
+               AST_RuntimeError(Node, "Tried to access an element of NULL");
+               return ERRPTR;
+       }
+       
+       switch( Object->Type )
+       {
+       case SS_DATATYPE_OBJECT: {
+               tSpiderObjectDef        *class = Object->Object->Type;
+               for( i = 0; i < class->NAttributes; i ++ )
+               {
+                       if( strcmp(ElementName, class->AttributeDefs[i].Name) == 0 )
+                       {
+                               if( SaveValue != ERRPTR ) {
+                                       Object->Object->Attributes[i] = SaveValue;
+                                       SpiderScript_ReferenceValue(SaveValue);
+                                       return NULL;
+                               }
+                               else {
+                                       ret = Object->Object->Attributes[i];
+                                       SpiderScript_ReferenceValue(ret);
+                                       return ret;
+                               }
+                       }
+               }
+               AST_RuntimeError(Node, "Unknown attribute '%s' of class '%s'",
+                       ElementName, class->Name);
+               return ERRPTR; }
+       default:
+               AST_RuntimeError(Node, "Unable to get element of type %i", Object->Type);
+               return ERRPTR;
+       }
+}
+
 #if USE_AST_EXEC
 /**
  * \brief Define a variable