src/AST/element.h

   1 #ifndef ELEMENT_H
   2 #define ELEMENT_H
   3 #include "classlist.h"
   4 #include "mymemory.h"
   5 #include "structs.h"
   6 #include "astnode.h"
   7 #include "functionencoding.h"
   8 #include "elementencoding.h"
   9 #include "boolean.h"
  10
  11 #define GETELEMENTTYPE(o) GETASTNODETYPE(o)
  12 #define GETELEMENTPARENTS(o) (&((Element *)o)->parents)
  13 struct Element {
  14         ASTNode base;
  15         VectorASTNode parents;
  16 };
  17
  18 struct ElementConst {
  19         Element base;
  20         Set *set;
  21         uint64_t value;
  22         ElementEncoding encoding;
  23 };
  24
  25 struct ElementSet {
  26         Element base;
  27         Set *set;
  28         ElementEncoding encoding;
  29 };
  30
  31 struct ElementFunction {
  32         Element base;
  33         Function *function;
  34         ArrayElement inputs;
  35         Boolean *overflowstatus;
  36         FunctionEncoding functionencoding;
  37         ElementEncoding rangeencoding;
  38 };
  39
  40 Element *allocElementConst(uint64_t value, VarType type);
  41 Element *allocElementSet(Set *s);
  42 Element *allocElementFunction(Function *function, Element **array, uint numArrays, Boolean *overflowstatus);
  43 void deleteElement(Element *This);
  44 Set *getElementSet(Element *This);
  45
  46 static inline ElementEncoding *getElementEncoding(Element *This) {
  47         switch (GETELEMENTTYPE(This)) {
  48         case ELEMSET:
  49                 return &((ElementSet *)This)->encoding;
  50         case ELEMFUNCRETURN:
  51                 return &((ElementFunction *)This)->rangeencoding;
  52         case ELEMCONST:
  53                 return &((ElementConst *)This)->encoding;
  54         default:
  55                 ASSERT(0);
  56         }
  57         return NULL;
  58 }
  59
  60 static inline FunctionEncoding *getElementFunctionEncoding(ElementFunction *func) {
  61         return &func->functionencoding;
  62 }
  63 #endif