Merge pull request #1844 from YosysHQ/dave/gen-source-loc
[yosys.git] / kernel / hashlib.h
index 94b573e47f47f5a2a1c7f204037759fd08724613..e7cb312ed87f63f4adf944c05bce7a4d4fd42d4e 100644 (file)
@@ -1,5 +1,5 @@
 // This is free and unencumbered software released into the public domain.
-// 
+//
 // Anyone is free to copy, modify, publish, use, compile, sell, or
 // distribute this software, either in source code form or as a compiled
 // binary, for any purpose, commercial or non-commercial, and by any
@@ -10,6 +10,7 @@
 // -------------------------------------------------------
 
 #ifndef HASHLIB_H
+#define HASHLIB_H
 
 #include <stdexcept>
 #include <algorithm>
@@ -30,7 +31,7 @@ inline unsigned int mkhash(unsigned int a, unsigned int b) {
 const unsigned int mkhash_init = 5381;
 
 // The ADD version of DJB2
-// (usunsigned int mkhashe this version for cache locality in b)
+// (use this version for cache locality in b)
 inline unsigned int mkhash_add(unsigned int a, unsigned int b) {
        return ((a << 5) + a) + b;
 }
@@ -58,16 +59,25 @@ template<typename T> struct hash_ops {
        }
 };
 
-template<> struct hash_ops<int> {
+struct hash_int_ops {
        template<typename T>
        static inline bool cmp(T a, T b) {
                return a == b;
        }
-       template<typename T>
-       static inline unsigned int hash(T a) {
+};
+
+template<> struct hash_ops<int32_t> : hash_int_ops
+{
+       static inline unsigned int hash(int32_t a) {
                return a;
        }
 };
+template<> struct hash_ops<int64_t> : hash_int_ops
+{
+       static inline unsigned int hash(int64_t a) {
+               return mkhash((unsigned int)(a), (unsigned int)(a >> 32));
+       }
+};
 
 template<> struct hash_ops<std::string> {
        static inline bool cmp(const std::string &a, const std::string &b) {
@@ -86,9 +96,22 @@ template<typename P, typename Q> struct hash_ops<std::pair<P, Q>> {
                return a == b;
        }
        static inline unsigned int hash(std::pair<P, Q> a) {
-               hash_ops<P> p_ops;
-               hash_ops<Q> q_ops;
-               return mkhash(p_ops.hash(a.first), q_ops.hash(a.second));
+               return mkhash(hash_ops<P>::hash(a.first), hash_ops<Q>::hash(a.second));
+       }
+};
+
+template<typename... T> struct hash_ops<std::tuple<T...>> {
+       static inline bool cmp(std::tuple<T...> a, std::tuple<T...> b) {
+               return a == b;
+       }
+       template<size_t I = 0>
+       static inline typename std::enable_if<I == sizeof...(T), unsigned int>::type hash(std::tuple<T...>) {
+               return mkhash_init;
+       }
+       template<size_t I = 0>
+       static inline typename std::enable_if<I != sizeof...(T), unsigned int>::type hash(std::tuple<T...> a) {
+               typedef hash_ops<typename std::tuple_element<I, std::tuple<T...>>::type> element_ops_t;
+               return mkhash(hash<I+1>(a), element_ops_t::hash(std::get<I>(a)));
        }
 };
 
@@ -97,10 +120,9 @@ template<typename T> struct hash_ops<std::vector<T>> {
                return a == b;
        }
        static inline unsigned int hash(std::vector<T> a) {
-               hash_ops<T> t_ops;
                unsigned int h = mkhash_init;
                for (auto k : a)
-                       h = mkhash(h, t_ops.hash(k));
+                       h = mkhash(h, hash_ops<T>::hash(k));
                return h;
        }
 };
@@ -115,8 +137,8 @@ struct hash_cstr_ops {
        static inline unsigned int hash(const char *a) {
                unsigned int hash = mkhash_init;
                while (*a)
-                        hash = mkhash(hash, *(a++));
-                return hash;
+                       hash = mkhash(hash, *(a++));
+               return hash;
        }
 };
 
@@ -125,7 +147,7 @@ struct hash_ptr_ops {
                return a == b;
        }
        static inline unsigned int hash(const void *a) {
-               return (unsigned long)a;
+               return (uintptr_t)a;
        }
 };
 
@@ -135,10 +157,15 @@ struct hash_obj_ops {
        }
        template<typename T>
        static inline unsigned int hash(const T *a) {
-               return a->hash();
+               return a ? a->hash() : 0;
        }
 };
 
+template<typename T>
+inline unsigned int mkhash(const T &v) {
+       return hash_ops<T>().hash(v);
+}
+
 inline int hashtable_size(int min_size)
 {
        static std::vector<int> zero_and_some_primes = {
@@ -167,6 +194,7 @@ inline int hashtable_size(int min_size)
 template<typename K, typename T, typename OPS = hash_ops<K>> class dict;
 template<typename K, int offset = 0, typename OPS = hash_ops<K>> class idict;
 template<typename K, typename OPS = hash_ops<K>> class pool;
+template<typename K, typename OPS = hash_ops<K>> class mfp;
 
 template<typename K, typename T, typename OPS>
 class dict
@@ -204,7 +232,7 @@ class dict
        void do_rehash()
        {
                hashtable.clear();
-               hashtable.resize(hashtable_size(entries.size() * hashtable_size_factor), -1);
+               hashtable.resize(hashtable_size(entries.capacity() * hashtable_size_factor), -1);
 
                for (int i = 0; i < int(entries.size()); i++) {
                        do_assert(-1 <= entries[i].next && entries[i].next < int(entries.size()));
@@ -477,6 +505,15 @@ public:
                return entries[i].udata.second;
        }
 
+       T at(const K &key, const T &defval) const
+       {
+               int hash = do_hash(key);
+               int i = do_lookup(key, hash);
+               if (i < 0)
+                       return defval;
+               return entries[i].udata.second;
+       }
+
        T& operator[](const K &key)
        {
                int hash = do_hash(key);
@@ -514,14 +551,17 @@ public:
                return !operator==(other);
        }
 
+       void reserve(size_t n) { entries.reserve(n); }
        size_t size() const { return entries.size(); }
        bool empty() const { return entries.empty(); }
        void clear() { hashtable.clear(); entries.clear(); }
 
        iterator begin() { return iterator(this, int(entries.size())-1); }
+       iterator element(int n) { return iterator(this, int(entries.size())-1-n); }
        iterator end() { return iterator(nullptr, -1); }
 
        const_iterator begin() const { return const_iterator(this, int(entries.size())-1); }
+       const_iterator element(int n) const { return const_iterator(this, int(entries.size())-1-n); }
        const_iterator end() const { return const_iterator(nullptr, -1); }
 };
 
@@ -563,7 +603,7 @@ protected:
        void do_rehash()
        {
                hashtable.clear();
-               hashtable.resize(hashtable_size(entries.size() * hashtable_size_factor), -1);
+               hashtable.resize(hashtable_size(entries.capacity() * hashtable_size_factor), -1);
 
                for (int i = 0; i < int(entries.size()); i++) {
                        do_assert(-1 <= entries[i].next && entries[i].next < int(entries.size()));
@@ -803,6 +843,14 @@ public:
                do_rehash();
        }
 
+       K pop()
+       {
+               iterator it = begin();
+               K ret = *it;
+               erase(it);
+               return ret;
+       }
+
        void swap(pool &other)
        {
                hashtable.swap(other.hashtable);
@@ -822,14 +870,24 @@ public:
                return !operator==(other);
        }
 
+       bool hash() const {
+               unsigned int hashval = mkhash_init;
+               for (auto &it : entries)
+                       hashval ^= ops.hash(it.udata);
+               return hashval;
+       }
+
+       void reserve(size_t n) { entries.reserve(n); }
        size_t size() const { return entries.size(); }
        bool empty() const { return entries.empty(); }
        void clear() { hashtable.clear(); entries.clear(); }
 
        iterator begin() { return iterator(this, int(entries.size())-1); }
+       iterator element(int n) { return iterator(this, int(entries.size())-1-n); }
        iterator end() { return iterator(nullptr, -1); }
 
        const_iterator begin() const { return const_iterator(this, int(entries.size())-1); }
+       const_iterator element(int n) const { return const_iterator(this, int(entries.size())-1-n); }
        const_iterator end() const { return const_iterator(nullptr, -1); }
 };
 
@@ -859,6 +917,15 @@ public:
                return i + offset;
        }
 
+       int at(const K &key, int defval) const
+       {
+               int hash = database.do_hash(key);
+               int i = database.do_lookup(key, hash);
+               if (i < 0)
+                       return defval;
+               return i + offset;
+       }
+
        int count(const K &key) const
        {
                int hash = database.do_hash(key);
@@ -878,7 +945,118 @@ public:
                return database.entries.at(index - offset).udata;
        }
 
+       void swap(idict &other)
+       {
+               database.swap(other.database);
+       }
+
+       void reserve(size_t n) { database.reserve(n); }
+       size_t size() const { return database.size(); }
+       bool empty() const { return database.empty(); }
+       void clear() { database.clear(); }
+
+       const_iterator begin() const { return database.begin(); }
+       const_iterator element(int n) const { return database.element(n); }
+       const_iterator end() const { return database.end(); }
+};
+
+template<typename K, typename OPS>
+class mfp
+{
+       mutable idict<K, 0, OPS> database;
+       mutable std::vector<int> parents;
+
+public:
+       typedef typename idict<K, 0, OPS>::const_iterator const_iterator;
+
+       int operator()(const K &key) const
+       {
+               int i = database(key);
+               parents.resize(database.size(), -1);
+               return i;
+       }
+
+       const K &operator[](int index) const
+       {
+               return database[index];
+       }
+
+       int ifind(int i) const
+       {
+               int p = i, k = i;
+
+               while (parents[p] != -1)
+                       p = parents[p];
+
+               while (k != p) {
+                       int next_k = parents[k];
+                       parents[k] = p;
+                       k = next_k;
+               }
+
+               return p;
+       }
+
+       void imerge(int i, int j)
+       {
+               i = ifind(i);
+               j = ifind(j);
+
+               if (i != j)
+                       parents[i] = j;
+       }
+
+       void ipromote(int i)
+       {
+               int k = i;
+
+               while (k != -1) {
+                       int next_k = parents[k];
+                       parents[k] = i;
+                       k = next_k;
+               }
+
+               parents[i] = -1;
+       }
+
+       int lookup(const K &a) const
+       {
+               return ifind((*this)(a));
+       }
+
+       const K &find(const K &a) const
+       {
+               int i = database.at(a, -1);
+               if (i < 0)
+                       return a;
+               return (*this)[ifind(i)];
+       }
+
+       void merge(const K &a, const K &b)
+       {
+               imerge((*this)(a), (*this)(b));
+       }
+
+       void promote(const K &a)
+       {
+               int i = database.at(a, -1);
+               if (i >= 0)
+                       ipromote(i);
+       }
+
+       void swap(mfp &other)
+       {
+               database.swap(other.database);
+               parents.swap(other.parents);
+       }
+
+       void reserve(size_t n) { database.reserve(n); }
+       size_t size() const { return database.size(); }
+       bool empty() const { return database.empty(); }
+       void clear() { database.clear(); parents.clear(); }
+
        const_iterator begin() const { return database.begin(); }
+       const_iterator element(int n) const { return database.element(n); }
        const_iterator end() const { return database.end(); }
 };