X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=cyclegraph.h;h=716e811d64ea6ea4153572d6d9c485762b0988ef;hb=HEAD;hp=bbf7aaf16f71233a19b77d67073909bf9dbf267a;hpb=9077381b71d4e95f5e25204480159decf16281a7;p=c11tester.git

diff --git a/cyclegraph.h b/cyclegraph.h
index bbf7aaf1..716e811d 100644
--- a/cyclegraph.h
+++ b/cyclegraph.h
@@ -23,10 +23,10 @@ class CycleGraph {
 public:
 	CycleGraph();
 	~CycleGraph();
-	void addEdges(SnapList<ModelAction *> * edgeset, const ModelAction *to);
-	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
@@ -41,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;
@@ -59,21 +59,24 @@ 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 * 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;