Fixed gendata tool and cleaned up some stuff

1 files changed, 151 insertions, 0 deletions

diff --git a/tools/stats_tables_h48/stats_tables_h48.c b/tools/stats_tables_h48/stats_tables_h48.c
new file mode 100644
index 0000000..7118992
--- /dev/null
+++ b/tools/stats_tables_h48/stats_tables_h48.c
@@ -0,0 +1,151 @@
+#include <pthread.h>
+#include <time.h>
+#include "../timerun.h"
+#include "../../src/cube.h"
+
+#define MAXMOVES 20
+#define NTHREADS 32
+#define NCUBES_PER_THREAD 10000
+#define LOG_EVERY (NCUBES_PER_THREAD / 10)
+
+typedef struct {
+        int n;
+        int thread_id;
+        int64_t v[12][100];
+} thread_arg_t;
+
+const char *filename = "tables/h48h0k4";
+char *buf;
+
+uint64_t rand64(void) {
+        uint64_t i, ret;
+
+        for (i = 0, ret = 0; i < 64; i++)
+                ret |= (uint64_t)(rand() % 2) << i;
+
+        return ret;
+}
+
+void log_stderr(const char *str, ...) {
+        va_list args;
+
+        va_start(args, str);
+        vfprintf(stderr, str, args);
+        va_end(args);
+}
+
+static void *
+run_thread(void *arg)
+{
+        char sols[12], cube[22];
+        int64_t ep, eo, cp, co;
+        int i, j;
+
+        thread_arg_t *a = (thread_arg_t *)arg;
+
+        for (i = 0; i < a->n; i++) {
+                ep = rand64();
+                eo = rand64();
+                cp = rand64();
+                co = rand64();
+                nissy_getcube(ep, eo, cp, co, "fix", cube);
+                nissy_solve(cube, "h48stats", "", "",
+                    0, MAXMOVES, 1, -1, buf, sols);
+                for (j = 0; j < 12; j++)
+                        a->v[j][(int)sols[j]]++;
+                if ((i+1) % LOG_EVERY == 0)
+                        fprintf(stderr, "[thread %d] %d cubes solved...\n",
+                            a->thread_id, i+1);
+        }
+
+        return NULL;
+}
+
+void run(void) {
+        int64_t i, j, k, tot;
+        double avg;
+        pthread_t thread[NTHREADS];
+        thread_arg_t arg[NTHREADS];
+
+        for (i = 0; i < NTHREADS; i++) {
+                arg[i] = (thread_arg_t) {
+                        .thread_id = i,
+                        .n = NCUBES_PER_THREAD,
+                        .v = {{0}}
+                };
+                pthread_create(&thread[i], NULL, run_thread, &arg[i]);
+        }
+
+        for (i = 0; i < NTHREADS; i++)
+                pthread_join(thread[i], NULL);
+
+        for (j = 0; j < 12; j++) {
+                printf("Data for h=%" PRId64 "\n", j);
+                for (k = 0, avg = 0.0; k <= 16; k++) {
+                        for (i = 0, tot = 0; i < NTHREADS; i++)
+                                tot += arg[i].v[j][k];
+                        printf("%" PRId64 "\t%" PRId64 "\n", k, tot);
+                        avg += tot * k;
+                }
+                avg /= (double)(NCUBES_PER_THREAD * NTHREADS);
+                printf("Average: %.4lf\n", avg);
+                printf("\n");
+        }
+}
+
+int getdata(int64_t size) {
+        int64_t s;
+        FILE *f;
+
+        buf = malloc(size);
+
+        if ((f = fopen(filename, "rb")) == NULL) {
+                fprintf(stderr, "Table file not found, generating them."
+                    " This can take a while.\n");
+                s = nissy_gendata("h48stats", "", buf);
+                if (s != size) {
+                        fprintf(stderr, "Error generating table");
+                        if (s != -1)
+                                fprintf(stderr, " (got %" PRId64 " bytes)", s);
+                        fprintf(stderr, "\n");
+                        return 1;
+                }
+                if ((f = fopen(filename, "wb")) == NULL) {
+                        fprintf(stderr, "Could not write tables to file %s"
+                            ", will be regenerated next time.\n", filename);
+                } else {
+                        fwrite(buf, size, 1, f);
+                        fclose(f);
+                }
+        } else {
+                fprintf(stderr, "Reading tables from file %s\n", filename);
+                fread(buf, size, 1, f);
+                fclose(f);
+        }
+
+        return 0;
+}
+
+int main(void) {
+        int64_t size;
+
+        srand(time(NULL));
+
+        nissy_setlogger(log_stderr);
+        size = nissy_datasize("h48stats", "");
+        if (size == -1) {
+                printf("h48 stats: error in datasize\n");
+                return 1;
+        }
+
+        if (getdata(size) != 0) {
+                printf("Error getting table, stopping\n");
+                free(buf);
+                return 1;
+        }
+
+        timerun(run, "h48 table stats");
+
+        free(buf);
+        return 0;
+}