From 9b0ce9e798941e700b166da43e100f04f12df05f Mon Sep 17 00:00:00 2001
From: Fabrice Desclaux <fabrice.desclaux@cea.fr>
Date: Fri, 14 Apr 2017 16:59:36 +0200
Subject: Graph: add discard_edge

---
 miasm2/core/graph.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'miasm2/core/graph.py')

diff --git a/miasm2/core/graph.py b/miasm2/core/graph.py
index d97ca8be..ec9eac36 100644
--- a/miasm2/core/graph.py
+++ b/miasm2/core/graph.py
@@ -100,6 +100,11 @@ class DiGraph(object):
         self._nodes_succ[src].remove(dst)
         self._nodes_pred[dst].remove(src)
 
+    def discard_edge(self, src, dst):
+        """Remove edge between @src and @dst if it exits"""
+        if (src, dst) in self._edges:
+            self.del_edge(src, dst)
+
     def predecessors_iter(self, node):
         if not node in self._nodes_pred:
             raise StopIteration
-- 
cgit 1.4.1