From f7c6eee791106fead26297e06cc11793acc2bc14 Mon Sep 17 00:00:00 2001
From: Leonardo de Moura <leonardo@microsoft.com>
Date: Mon, 16 Nov 2015 22:07:04 -0800
Subject: [PATCH] feat(library/head_map): add head_map_prio

---
 src/library/head_map.h | 29 ++++++++++++++++++++++++++---
 1 file changed, 26 insertions(+), 3 deletions(-)

diff --git a/src/library/head_map.h b/src/library/head_map.h
index 4a9cc8037..265bf3619 100644
--- a/src/library/head_map.h
+++ b/src/library/head_map.h
@@ -30,10 +30,24 @@ struct head_index {
 
     This index is very naive, but it is effective for many applications.
 */
-template<typename V>
-class head_map {
+template<typename V, typename GetPrio>
+class head_map_prio : private GetPrio {
     rb_map<head_index, list<V>, head_index::cmp> m_map;
+
+    unsigned get_priority(V const & v) const { return GetPrio::operator()(v); }
+
+    list<V> insert_prio(V const & v, list<V> const & vs) {
+        if (!vs)
+            return to_list(v);
+        else if (get_priority(v) >= get_priority(head(vs)))
+            return cons(v, vs);
+        else
+            return cons(head(vs), insert_prio(v, tail(vs)));
+    }
+
 public:
+    head_map_prio() {}
+    head_map_prio(GetPrio const & g):GetPrio(g) {}
     bool empty() const { return m_map.empty(); }
     bool contains(head_index const & h) const { return m_map.contains(h); }
     list<V> const * find(head_index const & h) const { return m_map.find(h); }
@@ -52,7 +66,7 @@ public:
     }
     void insert(head_index const & h, V const & v) {
         if (auto it = m_map.find(h))
-            m_map.insert(h, cons(v, ::lean::filter(*it, [&](V const & v2) { return v != v2; })));
+            m_map.insert(h, insert_prio(v, ::lean::filter(*it, [&](V const & v2) { return v != v2; })));
         else
             m_map.insert(h, to_list(v));
     }
@@ -64,4 +78,13 @@ public:
             });
     }
 };
+
+template<typename V>
+struct constant_priority_fn { unsigned operator()(V const &) const { return 0; } };
+
+template<typename V>
+class head_map : public head_map_prio<V, constant_priority_fn<V>> {
+public:
+    head_map() {}
+};
 }