X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=cyclegraph.h;h=716e811d64ea6ea4153572d6d9c485762b0988ef;hb=5f84c7161ef4e7fd10af8f1936e1d978731d2e5a;hp=e2b586403994b5573c863d4825857e7945d28c8d;hpb=63b2c687570085f2a87b6a659d26608228af1ee0;p=c11tester.git

diff --git a/cyclegraph.h b/cyclegraph.h
index e2b58640..716e811d 100644
--- a/cyclegraph.h
+++ b/cyclegraph.h
@@ -23,11 +23,12 @@ class CycleGraph {
 public:
 	CycleGraph();
 	~CycleGraph();
-	void addEdge(const ModelAction *from, const ModelAction *to);
-	void addEdge(const ModelAction *from, const ModelAction *to, bool forceedge);
-	void addRMWEdge(const ModelAction *from, const ModelAction *rmw);
+	void addEdges(SnapList<ModelAction *> * edgeset, ModelAction *to);
+	void addEdge(ModelAction *from, ModelAction *to);
+	void addEdge(ModelAction *from, ModelAction *to, bool forceedge);
+	void addRMWEdge(ModelAction *from, ModelAction *rmw);
 	bool checkReachable(const ModelAction *from, const ModelAction *to) const;
-
+	void freeAction(const ModelAction * act);
 #if SUPPORT_MOD_ORDER_DUMP
 	void dumpNodes(FILE *file) const;
 	void dumpGraphToFile(const char *filename) const;
@@ -40,7 +41,7 @@ public:
 private:
 	void addNodeEdge(CycleNode *fromnode, CycleNode *tonode, bool forceedge);
 	void putNode(const ModelAction *act, CycleNode *node);
-	CycleNode * getNode(const ModelAction *act);
+	CycleNode * getNode(ModelAction *act);
 
 	/** @brief A table for mapping ModelActions to CycleNodes */
 	HashTable<const ModelAction *, CycleNode *, uintptr_t, 4> actionToNode;
@@ -58,24 +59,31 @@ private:
  */
 class CycleNode {
 public:
-	CycleNode(const ModelAction *act);
+	CycleNode(ModelAction *act);
 	void addEdge(CycleNode *node);
 	CycleNode * getEdge(unsigned int i) const;
 	unsigned int getNumEdges() const;
-	CycleNode * removeEdge();
+	CycleNode * getInEdge(unsigned int i) const;
+	unsigned int getNumInEdges() const;
 	bool setRMW(CycleNode *);
 	CycleNode * getRMW() const;
 	void clearRMW() { hasRMW = NULL; }
-	const ModelAction * getAction() const { return action; }
+	ModelAction * getAction() const { return action; }
+	void removeInEdge(CycleNode *src);
+	void removeEdge(CycleNode *dst);
+	~CycleNode();
 
 	SNAPSHOTALLOC
 private:
 	/** @brief The ModelAction that this node represents */
-	const ModelAction *action;
+	ModelAction *action;
 
 	/** @brief The edges leading out from this node */
 	SnapVector<CycleNode *> edges;
 
+	/** @brief The edges leading in from this node */
+	SnapVector<CycleNode *> inedges;
+
 	/** Pointer to a RMW node that reads from this node, or NULL, if none
 	 * exists */
 	CycleNode *hasRMW;