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