#include "constraint.h"
#include "function.h"
#include "table.h"
+#include "csolver.h"
-Element *allocElementSet(Set *s) {
- ElementSet *This = (ElementSet *)ourmalloc(sizeof(ElementSet));
- GETELEMENTTYPE(This) = ELEMSET;
- This->set = s;
- initDefVectorASTNode(GETELEMENTPARENTS(This));
- initElementEncoding(&This->encoding, (Element *) This);
- return &This->base;
-}
-
-Element *allocElementFunction(Function *function, Element **array, uint numArrays, Boolean *overflowstatus) {
- ElementFunction *This = (ElementFunction *) ourmalloc(sizeof(ElementFunction));
- GETELEMENTTYPE(This) = ELEMFUNCRETURN;
- This->function = function;
- ASSERT(GETBOOLEANTYPE(overflowstatus) == BOOLEANVAR);
- This->overflowstatus = overflowstatus;
- initArrayInitElement(&This->inputs, array, numArrays);
- initDefVectorASTNode(GETELEMENTPARENTS(This));
- for (uint i = 0; i < numArrays; i++)
- pushVectorASTNode(GETELEMENTPARENTS(array[i]), (ASTNode *) This);
- initElementEncoding(&This->rangeencoding, (Element *) This);
- initFunctionEncoding(&This->functionencoding, (Element *) This);
- return &This->base;
-}
-
-Element *allocElementConst(uint64_t value, VarType type) {
- ElementConst *This = (ElementConst *)ourmalloc(sizeof(ElementConst));
- GETELEMENTTYPE(This) = ELEMCONST;
- This->value = value;
- This->set = allocSet(type, (uint64_t[]) {value}, 1);
- initDefVectorASTNode(GETELEMENTPARENTS(This));
- initElementEncoding(&This->encoding, (Element *) This);
- return &This->base;
-}
-
-Set *getElementSet(Element *This) {
- switch (GETELEMENTTYPE(This)) {
- case ELEMSET:
- return ((ElementSet *)This)->set;
- case ELEMCONST:
- return ((ElementConst *)This)->set;
- case ELEMFUNCRETURN: {
- Function *func = ((ElementFunction *)This)->function;
- switch (GETFUNCTIONTYPE(func)) {
- case TABLEFUNC:
- return ((FunctionTable *)func)->table->range;
- case OPERATORFUNC:
- return ((FunctionOperator *)func)->range;
- default:
- ASSERT(0);
- }
- }
- default:
- ASSERT(0);
- }
- ASSERT(0);
- return NULL;
-}
-
-void deleteElement(Element *This) {
- switch (GETELEMENTTYPE(This)) {
- case ELEMFUNCRETURN: {
- ElementFunction *ef = (ElementFunction *) This;
- deleteInlineArrayElement(&ef->inputs);
- deleteElementEncoding(&ef->rangeencoding);
- deleteFunctionEncoding(&ef->functionencoding);
- break;
+Element::Element(ASTNodeType _type) :
+ ASTNode(_type),
+ encoding(this) {
+}
+
+ElementSet::ElementSet(Set *s) :
+ Element(ELEMSET),
+ set(s) {
+}
+
+ElementSet::ElementSet(ASTNodeType _type, Set *s) :
+ Element(_type),
+ set(s) {
+}
+
+ElementFunction::ElementFunction(Function *_function, Element **array, uint numArrays, BooleanEdge _overflowstatus) :
+ Element(ELEMFUNCRETURN),
+ inputs(array, numArrays),
+ overflowstatus(_overflowstatus),
+ functionencoding(this),
+ function(_function) {
+}
+
+ElementConst::ElementConst(uint64_t _value, Set *_set) :
+ ElementSet(ELEMCONST, _set),
+ value(_value) {
+}
+
+Element *ElementConst::clone(CSolver *solver, CloneMap *map) {
+ return solver->getElementConst(type, value);
+}
+
+Element *ElementSet::clone(CSolver *solver, CloneMap *map) {
+ Element *e = (Element *) map->get(this);
+ if (e != NULL)
+ return e;
+ e = solver->getElementVar(set->clone(solver, map));
+ map->put(e, e);
+ return e;
+}
+
+Element *ElementFunction::clone(CSolver *solver, CloneMap *map) {
+ Element *array[inputs.getSize()];
+ for (uint i = 0; i < inputs.getSize(); i++) {
+ array[i] = inputs.get(i)->clone(solver, map);
}
- case ELEMSET: {
- ElementSet *es = (ElementSet *) This;
- deleteElementEncoding(&es->encoding);
- break;
+ Element *e = solver->applyFunction(function->clone(solver, map), array, inputs.getSize(), overflowstatus->clone(solver, map));
+ return e;
+}
+
+void ElementFunction::updateParents() {
+ for (uint i = 0; i < inputs.getSize(); i++) inputs.get(i)->parents.push(this);
+}
+
+Set *ElementFunction::getRange() {
+ return function->getRange();
+}
+
+void ElementSet::serialize(Serializer *serializer) {
+ if (serializer->isSerialized(this))
+ return;
+ serializer->addObject(this);
+
+ set->serialize(serializer);
+
+ serializer->mywrite(&type, sizeof(ASTNodeType));
+ ElementSet *This = this;
+ serializer->mywrite(&This, sizeof(ElementSet *));
+ serializer->mywrite(&set, sizeof(Set *));
+}
+
+void ElementSet::print() {
+ model_print("{ElementSet:");
+ set->print();
+ model_print(" %p ", this);
+ getElementEncoding()->print();
+ model_print("}");
+}
+
+void ElementConst::serialize(Serializer *serializer) {
+ if (serializer->isSerialized(this))
+ return;
+ serializer->addObject(this);
+
+ set->serialize(serializer);
+
+ serializer->mywrite(&type, sizeof(ASTNodeType));
+ ElementSet *This = this;
+ serializer->mywrite(&This, sizeof(ElementSet *));
+ VarType type = set->getType();
+ serializer->mywrite(&type, sizeof(VarType));
+ serializer->mywrite(&value, sizeof(uint64_t));
+}
+
+void ElementConst::print() {
+ model_print("{ElementConst: %" PRIu64 "}\n", value);
+}
+
+void ElementFunction::serialize(Serializer *serializer) {
+ if (serializer->isSerialized(this))
+ return;
+ serializer->addObject(this);
+
+ function->serialize(serializer);
+ uint size = inputs.getSize();
+ for (uint i = 0; i < size; i++) {
+ Element *input = inputs.get(i);
+ input->serialize(serializer);
}
- case ELEMCONST: {
- ElementConst *ec = (ElementConst *) This;
- deleteSet(ec->set);//Client did not create, so we free it
- deleteElementEncoding(&ec->encoding);
- break;
+ serializeBooleanEdge(serializer, overflowstatus);
+
+ serializer->mywrite(&type, sizeof(ASTNodeType));
+ ElementFunction *This = this;
+ serializer->mywrite(&This, sizeof(ElementFunction *));
+ serializer->mywrite(&function, sizeof(Function *));
+ serializer->mywrite(&size, sizeof(uint));
+ for (uint i = 0; i < size; i++) {
+ Element *input = inputs.get(i);
+ serializer->mywrite(&input, sizeof(Element *));
}
- default:
- ASSERT(0);
+ Boolean *overflowstat = overflowstatus.getRaw();
+ serializer->mywrite(&overflowstat, sizeof(Boolean *));
+}
+
+void ElementFunction::print() {
+ model_print("{ElementFunction:\n");
+ function->print();
+ model_print("Elements:\n");
+ uint size = inputs.getSize();
+ for (uint i = 0; i < size; i++) {
+ Element *input = inputs.get(i);
+ input->print();
}
- deleteVectorArrayASTNode(GETELEMENTPARENTS(This));
- ourfree(This);
+ model_print("}\n");
}