Merge
[satune.git] / 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 class Element : public ASTNode {
12 public:
13         Element(ASTNodeType type);
14         virtual ~Element() {}
15         Vector<ASTNode *> parents;
16         ElementEncoding encoding;
17         inline ElementEncoding *getElementEncoding() { return &encoding; }
18         virtual Element *clone(CSolver *solver, CloneMap *map) {ASSERT(0); return NULL;};
19         virtual void serialize(Serializer *serializer) = 0;
20         virtual void print() = 0;
21         virtual void updateParents() {}
22         virtual Set *getRange() = 0;
23         CMEMALLOC;
24 };
25
26 class ElementSet : public Element {
27 public:
28         ElementSet(ASTNodeType type, Set *s);
29         virtual ~ElementSet(){}
30         ElementSet(Set *s);
31         virtual Element *clone(CSolver *solver, CloneMap *map);
32         virtual void serialize(Serializer *serializer);
33         virtual void print();
34         CMEMALLOC;
35         Set *getRange() {return set;}
36 protected:
37         Set *set;
38
39 };
40
41 class ElementConst : public ElementSet {
42 public:
43         ElementConst(uint64_t value, Set *_set);
44         virtual ~ElementConst(){}
45         uint64_t value;
46         virtual void serialize(Serializer *serializer);
47         virtual void print();
48         Element *clone(CSolver *solver, CloneMap *map);
49         CMEMALLOC;
50 };
51
52 class ElementFunction : public Element {
53 public:
54         virtual ~ElementFunction(){}
55         ElementFunction(Function *function, Element **array, uint numArrays, BooleanEdge overflowstatus);
56         Array<Element *> inputs;
57         BooleanEdge overflowstatus;
58         FunctionEncoding functionencoding;
59         Element *clone(CSolver *solver, CloneMap *map);
60         virtual void serialize(Serializer *serializer);
61         virtual void print();
62         Set *getRange();
63         void updateParents();
64         Function *getFunction() {return function;}
65         inline FunctionEncoding *getElementFunctionEncoding() {return &functionencoding;}
66         CMEMALLOC;
67 private:
68         Function *function;
69 };
70
71
72 #endif