Z3 sources

Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>

lib/permutation.h

@@ -0,0 +1,93 @@
+/*++
+Copyright (c) 2011 Microsoft Corporation
+
+Module Name:
+
+    permutation.h
+
+Abstract:
+
+    Simple abstraction for managing permutations.
+
+Author:
+
+    Leonardo de Moura (leonardo) 2011-06-10.
+
+Revision History:
+
+--*/
+#ifndef _PERMUTATION_H_
+#define _PERMUTATION_H_
+
+#include<iostream>
+#include"vector.h"
+
+class permutation {
+    unsigned_vector m_p;
+    unsigned_vector m_inv_p;
+public:
+    permutation(unsigned size = 0);
+    void reset(unsigned size = 0);
+
+    unsigned operator()(unsigned i) const { return m_p[i]; }
+    unsigned inv(unsigned i_prime) const { return m_inv_p[i_prime]; }
+
+    void swap(unsigned i, unsigned j);
+    void move_after(unsigned i, unsigned j);
+    
+    void display(std::ostream & out) const;
+    bool check_invariant() const;
+};
+
+inline std::ostream & operator<<(std::ostream & out, permutation const & p) {
+    p.display(out);
+    return out;
+}
+
+/**
+   \brief Apply permutation p to data.
+   The algorithm does not use any extra memory.
+   
+   Requirement: swap(T, T) must be available.
+
+   This version will perform destructive updates to p.
+   Use apply_permutation if p must not be preserved
+*/
+template<typename T>
+void apply_permutation_core(unsigned sz, T * data, unsigned * p) {
+    int * p1 = reinterpret_cast<int*>(p);
+    for (int i = 0; i < static_cast<int>(sz); i++) {
+        if (p1[i] < 0)
+            continue; // already processed
+        int j = i;
+        while (true) {
+            SASSERT(j >= 0);
+            int p_j = p1[j];
+            SASSERT(p_j >= 0);
+            SASSERT(p_j < static_cast<int>(sz));
+            p1[j]   = - p1[j] - 1; // mark as done
+            if (p_j == i)
+                break; // cycle starting at i is done
+            swap(data[j], data[p_j]);
+            j = p_j;
+        }
+    }
+}
+
+/**
+   \brief Apply permutation p to data.
+   The algorithm does not use any extra memory.
+   
+   Requirement: swap(T, T) must be available.
+*/
+template<typename T>
+void apply_permutation(unsigned sz, T * data, unsigned const * p) {
+    apply_permutation_core(sz, data, const_cast<unsigned*>(p));
+    // restore p
+    int * p1 = reinterpret_cast<int*>(const_cast<unsigned*>(p));
+    for (unsigned i = 0; i < sz; i++) { 
+        p1[i] = - p1[i] - 1;
+    }
+}
+
+#endif