From 7defd8041e6050a9adf7927b43fd249d75d6389f Mon Sep 17 00:00:00 2001
From: Sebastiano Tronto <sebastiano@tronto.net>
Date: Tue, 4 Mar 2025 08:24:39 +0100
Subject: Begin work for coordinate solver

---
 src/solvers/coord/gendata_coord.h | 134 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 134 insertions(+)
 create mode 100644 src/solvers/coord/gendata_coord.h

(limited to 'src/solvers/coord/gendata_coord.h')

diff --git a/src/solvers/coord/gendata_coord.h b/src/solvers/coord/gendata_coord.h
new file mode 100644
index 0000000..b3932cd
--- /dev/null
+++ b/src/solvers/coord/gendata_coord.h
@@ -0,0 +1,134 @@
+STATIC size_t gendata_coordinate(const coord_t *, void *);
+STATIC size_t gendata_coordinate_name(const char *, void *);
+STATIC tableinfo_t genptable_coordinate(const coord_t *, const void *, uint8_t *);
+STATIC uint8_t get_coord_pval(const coord_t *, const uint8_t *, uint64_t);
+STATIC void set_coord_pval(const coord_t *, uint8_t *, uint64_t, uint8_t);
+
+STATIC size_t
+gendata_coordinate_name(const char *name, void *buf)
+{
+	int i;
+
+	for (i = 0; all_coordinates[i] != NULL; i++)
+		if (strcmp(all_coordinates[i]->name, name) == 0)
+			return gendata_coordinate(all_coordinates[i], buf);
+
+	LOG("Cannot generate data for coordinate '%s': not found\n", name);
+	return 0;
+}
+
+STATIC size_t
+gendata_coordinate(const coord_t *coord, void *buf)
+{
+	uint64_t coord_dsize, tablesize, ninfo;
+	void *pruningbuf, *coord_data;
+	uint8_t *table;
+	tableinfo_t coord_data_info, pruning_info;
+
+	coord_data = buf == NULL ? NULL : ((uint8_t *)buf) + INFOSIZE;
+	coord_dsize = coord->gendata(coord_data);
+	ninfo = coord_dsize == 0 ? 1 : 2;
+	tablesize = DIV_ROUND_UP(coord->max, 2);
+
+	if (buf == NULL)
+		goto gendata_coordinate_return_size;
+
+	if (ninfo == 2) {
+		coord_data_info = (tableinfo_t) {
+			.solver = "coord helper table for ",
+			.type = TABLETYPE_SPECIAL,
+			.infosize = INFOSIZE,
+			.fullsize = INFOSIZE + coord_dsize,
+			.hash = 0, /* TODO */
+			.next = INFOSIZE + coord_dsize,
+
+			/* Unknown / non-applicable values */
+			.entries = 0,
+			.classes = 0,
+			.bits = 0,
+			.base = 0,
+			.maxvalue = 0,
+		};
+
+		append_coord_name(coord, coord_data_info.solver);
+
+		writetableinfo(&coord_data_info, INFOSIZE + coord_dsize, buf);
+
+		pruningbuf = ((uint8_t *)buf) + INFOSIZE + coord_dsize;
+	} else {
+		pruningbuf = buf;
+	}
+
+	table = ((uint8_t *)pruningbuf) + INFOSIZE;
+	pruning_info = genptable_coordinate(coord, coord_data, table);
+	writetableinfo(&pruning_info, INFOSIZE + tablesize, pruningbuf);
+
+gendata_coordinate_return_size:
+	return ninfo * INFOSIZE + coord_dsize + tablesize;
+}
+
+STATIC tableinfo_t
+genptable_coordinate(const coord_t *coord, const void *data, uint8_t *table)
+{
+	uint64_t tablesize, i, j, d, tot;
+	tableinfo_t info;
+	uint8_t m;
+	cube_t c, cc;
+
+	tablesize = DIV_ROUND_UP(coord->max, 2);
+
+	memset(table, 0xFF, tablesize);
+
+	info = (tableinfo_t) {
+		.solver = "coordinate solver for ",
+		.type = TABLETYPE_PRUNING,
+		.infosize = INFOSIZE,
+		.fullsize = INFOSIZE + tablesize,
+		.hash = 0, /* TODO */
+		.entries = coord->max,
+		.classes = 0,
+		.bits = 4,
+		.base = 0,
+		.maxvalue = 0,
+		.next = 0
+	};
+
+	memset(info.distribution, 0, INFO_DISTRIBUTION_LEN * sizeof(uint64_t));
+	append_coord_name(coord, info.solver);
+
+	i = coord->coord(SOLVED_CUBE, data);
+	set_coord_pval(coord, table, i, 0);
+	info.distribution[0] = 1;
+	for (d = 1, tot = 1; tot < coord->max; d++) {
+		for (i = 0; i < coord->max; i++) {
+			if (get_coord_pval(coord, table, i) == d-1) {
+				c = coord->cube(i, data);
+				for (m = 0; m < 18; m++) {
+					cc = move(c, m);
+					j = coord->coord(cc, data);
+					if (get_coord_pval(coord, table, j) > d) {
+						set_coord_pval(coord, table, j, d);
+						tot++;
+						info.distribution[d]++;
+					}
+				}
+			}
+		}
+	}
+	info.maxvalue = d-1;
+
+	return info;
+}
+
+STATIC uint8_t
+get_coord_pval(const coord_t *coord, const uint8_t *table, uint64_t i)
+{
+	return (table[COORD_INDEX(i)] & COORD_MASK(i)) >> COORD_SHIFT(i);
+}
+
+STATIC void
+set_coord_pval(const coord_t *coord, uint8_t *table, uint64_t i, uint8_t val)
+{
+	table[COORD_INDEX(i)] = (table[COORD_INDEX(i)] & (~COORD_MASK(i)))
+	    | (val << COORD_SHIFT(i));
+}
-- 
cgit v1.3