access the tracked sets

ianpaulwright · ianpaulwright · commit a31e9fd97bf1 · 2019-11-17T16:37:53.000Z
diff --git a/src/Datalog.h b/src/Datalog.h
@@ -326,7 +326,12 @@ struct State
 
 	template <typename RELATION_TYPE>
 	const typename RELATION_TYPE::Set getSet() const {
-		return convert<RELATION_TYPE>(get<RelationSet<RELATION_TYPE>>(stateRelations).set);
+		return convert<RELATION_TYPE>(getTrackedSet<RELATION_TYPE>());
+	}
+
+	template <typename RELATION_TYPE>
+	const typename RELATION_TYPE::TrackedSet getTrackedSet() const {
+		return get<RelationSet<RELATION_TYPE>>(stateRelations).set;
 	}
 
 	typedef tuple<RelationSize<RELATIONs>...> StateSizesType;
