X-Git-Url: http://plrg.eecs.uci.edu/git/?p=satune.git;a=blobdiff_plain;f=src%2FAST%2Fiterator.h;h=0e7303f7bc374481233f11cad41f2b81571c1ce8;hp=b82ee37ee2c8dc0cf357ef5972b906b485714a20;hb=18aa5690b5afa41ce2498a28adba27b37df7816d;hpb=ee69b0984366edd899227cc651ec272c8f77c66a

diff --git a/src/AST/iterator.h b/src/AST/iterator.h
index b82ee37..0e7303f 100644
--- a/src/AST/iterator.h
+++ b/src/AST/iterator.h
@@ -10,6 +10,7 @@ class BooleanIterator {
 	bool hasNext();
 	Boolean * next();
 	CMEMALLOC;
+
  private:
 	SetIteratorBooleanEdge * solverit;
 	HashsetBoolean discovered;
@@ -20,10 +21,22 @@ class BooleanIterator {
 
 class ElementIterator {
  public:
+	ElementIterator(CSolver *_solver);
+	~ElementIterator();
+	bool hasNext();
+	Element * next();
 	CMEMALLOC;
+
  private:
 	BooleanIterator bit;
-};
+	BooleanPredicate *base;
+	uint baseindex;
 
+	HashsetElement discovered;
+
+	Vector<Element *> element;
+	Vector<uint> index;
+	void updateNext();
+};
 
 #endif