1 files changed, 111 insertions, 0 deletions
diff --git a/sd.c b/sd.c
new file mode 100644
index 0000000..21caa57
--- /dev/null
+++ b/sd.c
@@ -0,0 +1,111 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdbool.h>
+#include <string.h>
+#include <time.h>
+#include <unistd.h>
+#include <getopt.h>
+#include <sqlite3.h>
+#include "segmenttree.h"
+
+int main(int argc, char* argv[]) {
+	char *file = "cards";
+	bool verbose = false;
+
+	/* Proccess args */
+	static struct option long_options[] = {
+		{"file",    required_argument, NULL, 'f'},
+		{"verbose", no_argument,       NULL, 'v'}
+	};
+	while (1) {
+		int option_index = 0;
+		int c = getopt_long(argc, argv, "f:v", long_options, &option_index);
+		if (c == -1) break;
+		switch (c) {
+			case 'f': file = strdup(optarg); break;
+			case 'v': verbose = true; break;
+			default: abort();
+		}
+	}
+
+	/* Seed the RNG */
+	srand(time(0));
+
+	/* Connect to db */
+	sqlite3 *db;
+	sqlite3_open(file, &db);
+
+	/* Get number of cards */
+	sqlite3_stmt *stmt;
+	sqlite3_prepare_v3(db, "SELECT COUNT(*) FROM cards", -1, 0, &stmt, NULL);
+	sqlite3_step(stmt);
+	int N = sqlite3_column_int(stmt, 0);
+	sqlite3_finalize(stmt);
+
+	/* Get card weights */
+	sqlite3_prepare_v3(db, "SELECT weight FROM cards", -1, 0, &stmt, NULL);
+	seg = (int*)malloc(4 * N * sizeof(int));
+	build(stmt, 0, N - 1, 1);
+	sqlite3_finalize(stmt);
+
+	if (verbose) {
+		for (int i = 0; i < 4 * N; i++) {
+			printf("%d ", seg[i]);
+		}
+		printf("\n");
+	}
+
+	/* Disable input buffering */
+	system("stty -F /dev/tty cbreak min 1");
+	system("stty -F /dev/tty -echo");
+
+	while (true) {
+		int x = (long long)rand() * rand() % seg[1];
+		int res[2];
+		query(res, x, 0, N-1, 1);
+		int w = res[0], i = res[1];
+
+		if (verbose) {
+			printf("%d %d %d %d\n", seg[1], x, w, i);
+		}
+
+		/* Get card contents from database */
+		sqlite3_prepare_v3(db, "SELECT key, val FROM cards WHERE idx=?", -1, 0, &stmt, NULL);
+		sqlite3_bind_int(stmt, 1, i);
+		sqlite3_step(stmt);
+		char *key, *val;
+		key = strdup(sqlite3_column_text(stmt, 0));
+		val = strdup(sqlite3_column_text(stmt, 1));
+		sqlite3_finalize(stmt);
+
+		printf("> %s\n", key);
+
+		/* Wait for confirmation */
+		char b = getchar();
+		printf("%s\n", val);
+
+		/* Read user input */
+		b = getchar();
+		if (b == 'y') {
+			w >>= 1;
+		}
+		else if (b == 'n') {
+			w <<= 3;
+		}
+		else {
+			break;
+		}
+
+		/* Update segment tree and database */
+		update(i, w, 0, N - 1, 1);
+		sqlite3_prepare_v3(db, "UPDATE cards SET weight=? WHERE idx=?", -1, 0, &stmt, NULL);
+		sqlite3_bind_int(stmt, 1, w);
+		sqlite3_bind_int(stmt, 2, i);
+		sqlite3_step(stmt);
+		sqlite3_finalize(stmt);
+	}
+
+	/* Cleanup */
+	system("stty -F /dev/tty echo");
+	sqlite3_close(db);
+}