From: timv <tim.f.vieira@gmail.com>
Date: Fri, 7 Jun 2013 03:37:57 +0000 (-0400)
Subject: ()
X-Git-Url: https://hydra-www.ietfng.org/gitweb/?a=commitdiff_plain;h=ddb5a897d73a1f3bb95f8f6289d3cd1df93a0976;p=dyna2

()
---

diff --git a/src/Dyna/Backend/Python/prioritydict.py b/src/Dyna/Backend/Python/prioritydict.py
new file mode 100644
index 0000000..fe1424a
--- /dev/null
+++ b/src/Dyna/Backend/Python/prioritydict.py
@@ -0,0 +1,75 @@
+from heapq import heapify, heappush, heappop
+
+class prioritydict(dict):
+    """Dictionary that can be used as a priority queue.
+
+    Keys of the dictionary are items to be put into the queue, and values
+    are their respective priorities. All dictionary methods work as expected.
+    The advantage over a standard heapq-based priority queue is
+    that priorities of items can be efficiently updated (amortized O(1))
+    using code as 'thedict[item] = new_priority.'
+
+    The 'smallest' method can be used to return the object with lowest
+    priority, and 'pop_smallest' also removes it.
+
+    The 'sorted_iter' method provides a destructive sorted iterator.
+
+    This implemented is based on:
+
+      Matteo Dell'Amico's implementation
+      http://code.activestate.com/recipes/522995-priority-dict-a-priority-queue-with-updatable-prio/
+
+         which is based on David Eppstein's implementation
+         http://code.activestate.com/recipes/117228/
+
+    """
+
+    def __init__(self, *args, **kwargs):
+        super(prioritydict, self).__init__(*args, **kwargs)
+        self._heap = None
+        self._rebuild_heap()
+
+    def _rebuild_heap(self):
+        self._heap = [(v, k) for k, v in self.iteritems()]
+        heapify(self._heap)
+
+    def smallest(self):
+        """
+        Return the item with the lowest priority.
+
+        Raises IndexError if the object is empty.
+        """
+        heap = self._heap
+        v, k = heap[0]
+        while k not in self or self[k] != v:
+            heappop(heap)
+            v, k = heap[0]
+        return k
+
+    def pop_smallest(self):
+        """
+        Return the item with the lowest priority and remove it.
+
+        Raises IndexError if the object is empty.
+        """
+        heap = self._heap
+        v, k = heappop(heap)
+        while k not in self or self[k] != v:
+            v, k = heappop(heap)
+        del self[k]
+        return k
+
+    def __setitem__(self, key, val):
+        # We are not going to remove the previous value from the heap, since
+        # this would have a cost O(n).
+        super(prioritydict, self).__setitem__(key, val)
+
+        if len(self._heap) < 2 * len(self):
+            heappush(self._heap, (val, key))
+        else:
+            # When the heap grows larger than 2 * len(self), we rebuild it from
+            # scratch to avoid wasting too much memory.
+            self._rebuild_heap()
+
+    setdefault = None
+    update = None