Merge branch 'hamed' into brian

[satune.git] / src / AST / element.h

diff --git a/src/AST/element.h b/src/AST/element.h
index b5bc9c1fdfcb7ce5bb21ec1d497a09af33771a50..509ee8c3d34566c21faf9f4e8d78d2fca84f6e4d 100644 (file)
--- a/src/AST/element.h
+++ b/src/AST/element.h
@@ -6,10 +6,10 @@
 #include "astnode.h"
 #include "functionencoding.h"
 #include "elementencoding.h"
+#include "boolean.h"
 
 #define GETELEMENTTYPE(o) GETASTNODETYPE(o)
-#define GETELEMENTPARENTS(o) (&((Element*)o)->parents)
-
+#define GETELEMENTPARENTS(o) (&((Element*)o)->parents)         
 struct Element {
        ASTNode base;
        VectorASTNode parents;
@@ -33,4 +33,22 @@ struct ElementFunction {
 Element * allocElementSet(Set *s);
 Element* allocElementFunction(Function * function, Element ** array, uint numArrays, Boolean * overflowstatus);
 void deleteElement(Element *This);
+Set* getElementSet(Element* This);
+static inline ElementEncoding* getElementEncoding(Element* This){
+       switch(GETELEMENTTYPE(This)){
+               case ELEMSET:
+                       return &((ElementSet*)This)->encoding;
+               case ELEMFUNCRETURN:            
+                       return &((ElementFunction*)This)->domainencoding;
+               default:
+                       ASSERT(0);
+       }
+       return NULL;
+}
+
+
+static inline FunctionEncoding* getElementFunctionEncoding(ElementFunction* func){
+       return &func->functionencoding;
+}
+
 #endif