1 files changed, 86 insertions, 0 deletions

diff --git a/tools/expected_distributions.h b/tools/expected_distributions.h
index 535cab2..0f3fec3 100644
--- a/tools/expected_distributions.h
+++ b/tools/expected_distributions.h
@@ -1,3 +1,16 @@
+uint64_t expected_cocsep[21] = {
+        [0] = 1,
+        [1] = 6,
+        [2] = 63,
+        [3] = 468,
+        [4] = 3068,
+        [5] = 15438,
+        [6] = 53814,
+        [7] = 71352,
+        [8] = 8784,
+        [9] = 96
+};
+
 uint64_t expected_h48[12][9][21] = {
         [0] = {
                 [2] = {
@@ -31,3 +44,76 @@ uint64_t expected_h48[12][9][21] = {
                 },
         },
 };
+
+static bool
+distribution_equal(const uint64_t *expected, const uint64_t *actual, int n)
+{
+        bool equal;
+        int i;
+
+        for (i = 0, equal = true; i <= n; i++) {
+                if (expected[i] != actual[i]) {
+                        equal = false;
+                        printf("Wrong value for %d: expected %" PRIu64
+                            ", actual %" PRIu64 "\n",
+                            i, expected[i], actual[i]);
+                }
+        }
+
+        return equal;
+}
+
+static bool
+check_cocsep(const void *data)
+{
+        tableinfo_t info;
+
+        readtableinfo(data, &info);
+        return distribution_equal(
+            expected_cocsep, info.distribution, info.maxvalue);
+}
+
+static bool
+unknown_h48(uint8_t h, uint8_t k)
+{
+        if (k != 2 && k != 4)
+                return true;
+
+        if (k == 4 && h != 0)
+                return true;
+
+        return k == 2 && h > 1;
+}
+
+STATIC bool
+check_distribution(const char *solver, const void *data)
+{
+        tableinfo_t info = {0};
+
+        if (!strcmp(solver, "h48")) {
+                readtableinfo(data, &info);
+                if (!distribution_equal(
+                    expected_cocsep, info.distribution, info.maxvalue)) {
+                        printf("ERROR! cocsep distribution is incorrect\n");
+                        return false;
+                }
+                printf("cocsep distribution is correct\n");
+
+                readtableinfo_n(data, 2, &info);
+                if (unknown_h48(info.h48h, info.bits))
+                        goto check_distribution_unknown;
+
+                if (!distribution_equal(expected_h48[info.h48h][info.bits],
+                    info.distribution, info.maxvalue)) {
+                        printf("ERROR! h48 distribution is incorrect\n");
+                        return false;
+                }
+
+                printf("h48 distribution is correct\n");
+                return true;
+        }
+
+check_distribution_unknown:
+        printf("Distribution unknown, not checked\n");
+        return true;
+}