#include "mymemory.h"
#include "structs.h"
#include "astnode.h"
+#include "functionencoding.h"
+#include "elementencoding.h"
+#include "boolean.h"
-#define GETELEMENTTYPE(o) GETASTNODETYPE(o)
-#define GETELEMENTPARENTS(o) (&((Element*)o)->parents)
+class Element : public ASTNode {
+public:
+ Element(ASTNodeType type);
+ virtual ~Element() {}
+ Vector<ASTNode *> parents;
+ ElementEncoding encoding;
+ virtual Element *clone(CSolver *solver, CloneMap *map) {ASSERT(0); return NULL;};
-struct Element {
- ASTNode base;
- VectorASTNode parents;
+ CMEMALLOC;
};
-struct ElementSet {
- Element base;
- Set * set;
- ElementEncoding * encoding;
+class ElementConst : public Element {
+public:
+ ElementConst(uint64_t value, VarType type, Set *_set);
+ Set *set;
+ uint64_t value;
+ Element *clone(CSolver *solver, CloneMap *map);
+ CMEMALLOC;
};
-struct ElementFunction {
- Element base;
- Function * function;
- ArrayElement inputs;
- Boolean * overflowstatus;
+class ElementSet : public Element {
+public:
+ ElementSet(Set *s);
+ Set *set;
+ Element *clone(CSolver *solver, CloneMap *map);
+ CMEMALLOC;
};
-Element * allocElementSet(Set *s);
-Element* allocElementFunction(Function * function, Element ** array, uint numArrays, Boolean * overflowstatus);
-void deleteElement(Element *This);
+class ElementFunction : public Element {
+public:
+ ElementFunction(Function *function, Element **array, uint numArrays, BooleanEdge overflowstatus);
+ Function *function;
+ Array<Element *> inputs;
+ BooleanEdge overflowstatus;
+ FunctionEncoding functionencoding;
+ Element *clone(CSolver *solver, CloneMap *map);
+ CMEMALLOC;
+};
+
+Set *getElementSet(Element *This);
+
+static inline ElementEncoding *getElementEncoding(Element *e) {
+ return &e->encoding;
+}
+
+static inline FunctionEncoding *getElementFunctionEncoding(ElementFunction *func) {
+ return &func->functionencoding;
+}
#endif