towards cloning

[satune.git] / src / AST / mutableset.cc

diff --git a/src/AST/mutableset.cc b/src/AST/mutableset.cc
index 19f6dec8b3878a02126245be10f5a246ad4ec55b..d029f82bbdcdf428ebf5fe1c5f9c1eb342fc8d46 100644 (file)
--- a/src/AST/mutableset.cc
+++ b/src/AST/mutableset.cc
@@ -1,4 +1,5 @@
 #include "mutableset.h"
+#include "csolver.h"
 
 MutableSet::MutableSet(VarType t) : Set(t) {
 }
@@ -6,3 +7,15 @@ MutableSet::MutableSet(VarType t) : Set(t) {
 void MutableSet::addElementMSet(uint64_t element) {
        members->push(element);
 }
+
+Set * MutableSet::clone(CSolver * solver, CloneMap *map) {
+       Set * s=map->set.get(this);
+       if (s != NULL)
+               return s;
+       s=solver->createMutableSet(type);
+       for(uint i=0; i<members->getSize();i++) {
+               solver->addItem((MutableSet *) s, members->get(i));
+       }
+       map->set.put(this, s);
+       return s;
+}