Usermode/Libraries/libspiderscript.so_src/ast.h

   1 /*
   2  */
   3 #ifndef _AST_H_
   4 #define _AST_H_
   5
   6 #include <spiderscript.h>
   7 #include "tokens.h"
   8
   9 typedef enum eAST_NodeTypes     tAST_NodeType;
  10 typedef struct sAST_Script      tAST_Script;
  11 typedef struct sAST_Function    tAST_Function;
  12 typedef struct sAST_Node        tAST_Node;
  13 typedef struct sAST_BlockState  tAST_BlockState;
  14 typedef struct sAST_Variable    tAST_Variable;
  15
  16 /**
  17  * \brief Node Types
  18  */
  19 enum eAST_NodeTypes
  20 {
  21         NODETYPE_NOP,
  22
  23         NODETYPE_BLOCK, //!< Node Block
  24
  25         NODETYPE_VARIABLE,      //!< Variable
  26         NODETYPE_CONSTANT,      //!< Runtime Constant
  27         NODETYPE_STRING,        //!< String Constant
  28         NODETYPE_INTEGER,       //!< Integer Constant
  29         NODETYPE_REAL,  //!< Real Constant
  30
  31         NODETYPE_DEFVAR,        //!< Define a variable (Variable)
  32         NODETYPE_SCOPE, //!< Dereference a Namespace/Class static
  33         NODETYPE_ELEMENT,       //!< Reference a class attribute
  34         NODETYPE_CAST,  //!< Cast a value to another (Uniop)
  35
  36         NODETYPE_RETURN,        //!< Return from a function (reserved word)
  37         NODETYPE_ASSIGN,        //!< Variable assignment operator
  38         NODETYPE_POSTINC,       //!< Post-increment (i++) - Uniop
  39         NODETYPE_POSTDEC,       //!< Post-decrement (i--) - Uniop
  40         NODETYPE_FUNCTIONCALL,  //!< Call a function
  41         NODETYPE_METHODCALL,    //!< Call a class method
  42         NODETYPE_CREATEOBJECT,  //!< Create an object
  43
  44         NODETYPE_IF,    //!< Conditional
  45         NODETYPE_LOOP,  //!< Looping Construct
  46
  47         NODETYPE_INDEX, //!< Index into an array
  48
  49         NODETYPE_LOGICALNOT,    //!< Logical NOT operator
  50         NODETYPE_LOGICALAND,    //!< Logical AND operator
  51         NODETYPE_LOGICALOR,     //!< Logical OR operator
  52         NODETYPE_LOGICALXOR,    //!< Logical XOR operator
  53
  54         NODETYPE_EQUALS,        //!< Comparison Equals
  55         NODETYPE_LESSTHAN,      //!< Comparison Less Than
  56         NODETYPE_LESSTHANEQUAL, //!< Comparison Less Than or Equal
  57         NODETYPE_GREATERTHAN,   //!< Comparison Greater Than
  58         NODETYPE_GREATERTHANEQUAL,      //!< Comparison Greater Than or Equal
  59
  60         NODETYPE_BWNOT, //!< Bitwise NOT
  61         NODETYPE_BWAND, //!< Bitwise AND
  62         NODETYPE_BWOR,  //!< Bitwise OR
  63         NODETYPE_BWXOR, //!< Bitwise XOR
  64
  65         NODETYPE_BITSHIFTLEFT,  //!< Bitwise Shift Left (Grow)
  66         NODETYPE_BITSHIFTRIGHT, //!< Bitwise Shift Right (Shrink)
  67         NODETYPE_BITROTATELEFT, //!< Bitwise Rotate Left (Grow)
  68
  69         NODETYPE_NEGATE,        //!< Negagte
  70         NODETYPE_ADD,   //!< Add
  71         NODETYPE_SUBTRACT,      //!< Subtract
  72         NODETYPE_MULTIPLY,      //!< Multiply
  73         NODETYPE_DIVIDE,        //!< Divide
  74         NODETYPE_MODULO,        //!< Modulus
  75 };
  76
  77 struct sSpiderScript
  78 {
  79         tSpiderVariant  *Variant;
  80         tAST_Script     *Script;
  81         char    *CurNamespace;  //!< Current namespace prefix (NULL = Root) - No trailing .
  82 };
  83
  84 struct sAST_Script
  85 {
  86         // TODO: Namespaces and Classes
  87         tAST_Function   *Functions;
  88         tAST_Function   *LastFunction;
  89 };
  90
  91 struct sAST_Function
  92 {
  93         tAST_Function   *Next;  //!< Next function in list
  94         tAST_Node       *Code;  //!< Function Code
  95         tAST_Node       *Arguments;     // HACKJOB (Only NODETYPE_DEFVAR is allowed)
  96         tAST_Node       *Arguments_Last;
  97         char    Name[]; //!< Function Name
  98 };
  99
 100 struct sAST_Node
 101 {
 102         tAST_Node       *NextSibling;
 103         tAST_NodeType   Type;
 104
 105         const char      *File;
 106          int    Line;
 107
 108         union
 109         {
 110                 struct {
 111                         tAST_Node       *FirstChild;
 112                         tAST_Node       *LastChild;
 113                 }       Block;
 114
 115                 struct {
 116                          int    Operation;
 117                         tAST_Node       *Dest;
 118                         tAST_Node       *Value;
 119                 }       Assign;
 120
 121                 struct {
 122                         tAST_Node       *Value;
 123                 }       UniOp;
 124
 125                 struct {
 126                         tAST_Node       *Left;
 127                         tAST_Node       *Right;
 128                 }       BinOp;
 129
 130                 struct {
 131                         tAST_Node       *Object;
 132                         tAST_Node       *FirstArg;
 133                         tAST_Node       *LastArg;
 134                          int    NumArgs;
 135                         char    Name[];
 136                 }       FunctionCall;
 137
 138                 struct {
 139                         tAST_Node       *Condition;
 140                         tAST_Node       *True;
 141                         tAST_Node       *False;
 142                 }       If;
 143
 144                 struct {
 145                         tAST_Node       *Init;
 146                          int    bCheckAfter;
 147                         tAST_Node       *Condition;
 148                         tAST_Node       *Increment;
 149                         tAST_Node       *Code;
 150                 }       For;
 151
 152                 /**
 153                  * \note Used for \a NODETYPE_VARIABLE, \a NODETYPE_CONSTANT and
 154                  *       \a NODETYPE_SCOPE
 155                  */
 156                 struct {
 157                         char    _unused;        // Shut GCC up
 158                         char    Name[];
 159                 }       Variable;
 160
 161                 struct {
 162                         tAST_Node       *Element;
 163                         char    Name[];
 164                 }       Scope;  // Used by NODETYPE_SCOPE and NODETYPE_ELEMENT
 165
 166                 struct {
 167                          int    DataType;
 168                         tAST_Node       *LevelSizes;
 169                         tAST_Node       *LevelSizes_Last;
 170                         char    Name[];
 171                 }       DefVar;
 172
 173                 struct {
 174                          int    DataType;
 175                          tAST_Node      *Value;
 176                 }       Cast;
 177
 178                 struct {
 179                          int    Length;
 180                         char    Data[];
 181                 }       String;
 182
 183                 uint64_t        Integer;
 184                 double  Real;
 185         };
 186 };
 187
 188 /**
 189  * \brief Code Block state (stores local variables)
 190  */
 191 struct sAST_BlockState
 192 {
 193         tAST_BlockState *Parent;
 194         tSpiderScript   *Script;        //!< Script
 195         tAST_Variable   *FirstVar;      //!< First variable in the list
 196         tSpiderValue    *RetVal;
 197         tSpiderNamespace        *BaseNamespace; //!< Base namespace (for entire block)
 198         tSpiderNamespace        *CurNamespace;  //!< Currently selected namespace
 199 };
 200
 201 struct sAST_Variable
 202 {
 203         tAST_Variable   *Next;
 204          int    Type;   // Only used for static typing
 205         tSpiderValue    *Object;
 206         char    Name[];
 207 };
 208
 209 // === FUNCTIONS ===
 210 extern tAST_Script      *AST_NewScript(void);
 211 extern size_t   AST_WriteScript(void *Buffer, tAST_Script *Script);
 212 extern size_t   AST_WriteNode(void *Buffer, size_t Offset, tAST_Node *Node);
 213
 214 extern tAST_Function    *AST_AppendFunction(tAST_Script *Script, const char *Name);
 215 extern void     AST_AppendFunctionArg(tAST_Function *Function, tAST_Node *Arg);
 216 extern void     AST_SetFunctionCode(tAST_Function *Function, tAST_Node *Root);
 217
 218 extern tAST_Node        *AST_NewNop(tParser *Parser);
 219
 220 extern tAST_Node        *AST_NewString(tParser *Parser, const char *String, int Length);
 221 extern tAST_Node        *AST_NewInteger(tParser *Parser, int64_t Value);
 222 extern tAST_Node        *AST_NewReal(tParser *Parser, double Value);
 223 extern tAST_Node        *AST_NewVariable(tParser *Parser, const char *Name);
 224 extern tAST_Node        *AST_NewDefineVar(tParser *Parser, int Type, const char *Name);
 225 extern tAST_Node        *AST_NewConstant(tParser *Parser, const char *Name);
 226 extern tAST_Node        *AST_NewClassElement(tParser *Parser, tAST_Node *Object, const char *Name);
 227
 228 extern tAST_Node        *AST_NewFunctionCall(tParser *Parser, const char *Name);
 229 extern tAST_Node        *AST_NewCreateObject(tParser *Parser, const char *Name);
 230 extern tAST_Node        *AST_NewMethodCall(tParser *Parser, tAST_Node *Object, const char *Name);
 231 extern void     AST_AppendFunctionCallArg(tAST_Node *Node, tAST_Node *Arg);
 232
 233 extern tAST_Node        *AST_NewCodeBlock(tParser *Parser);
 234 extern void     AST_AppendNode(tAST_Node *Parent, tAST_Node *Child);
 235
 236 extern tAST_Node        *AST_NewIf(tParser *Parser, tAST_Node *Condition, tAST_Node *True, tAST_Node *False);
 237 extern tAST_Node        *AST_NewLoop(tParser *Parser, tAST_Node *Init, int bPostCheck, tAST_Node *Condition, tAST_Node *Increment, tAST_Node *Code);
 238
 239 extern tAST_Node        *AST_NewAssign(tParser *Parser, int Operation, tAST_Node *Dest, tAST_Node *Value);
 240 extern tAST_Node        *AST_NewCast(tParser *Parser, int Target, tAST_Node *Value);
 241 extern tAST_Node        *AST_NewBinOp(tParser *Parser, int Operation, tAST_Node *Left, tAST_Node *Right);
 242 extern tAST_Node        *AST_NewUniOp(tParser *Parser, int Operation, tAST_Node *Value);
 243 extern tAST_Node        *AST_NewScopeDereference(tParser *Parser, const char *Name, tAST_Node *Child);
 244
 245 extern void     AST_FreeNode(tAST_Node *Node);
 246
 247 // exec_ast.h
 248 extern void     Object_Dereference(tSpiderValue *Object);
 249 extern void     Object_Reference(tSpiderValue *Object);
 250 extern tSpiderValue     *AST_ExecuteNode(tAST_BlockState *Block, tAST_Node *Node);
 251
 252 #endif