add wfc (under working)

2025-08-17 04:04:16 +02:00 · 2021-03-12 09:27:43 +08:00
parent 93e7efd697
commit 39c79cd4d8
1 changed files with 106 additions and 0 deletions
--- a/src/experimental/tiles_wfc.scad
+++ b/src/experimental/tiles_wfc.scad
@@ -0,0 +1,106 @@
+use <util/flat.scad>;
+use <util/has.scad>;
+use <util/slice.scad>;
+use <util/every.scad>;
+use <util/map/hashmap.scad>;
+use <util/map/hashmap_len.scad>;
+use <util/map/hashmap_put.scad>;
+use <util/map/hashmap_get.scad>;
+use <util/map/hashmap_del.scad>;
+use <util/map/hashmap_keys.scad>;
+use <util/map/hashmap_values.scad>;
+use <util/map/hashmap_entries.scad>;
+
+sample = [
+    ["S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S"],
+    ["S",  "S", "C0", "CN", "CN", "CN", "CN", "CN", "CN", "CN", "C3",  "S",  "S"],
+    ["S",  "S", "CW",  "L",  "L",  "L",  "L",  "L",  "L",  "L", "CE",  "S",  "S"],
+    ["S",  "S", "CW",  "L",  "L",  "L",  "L",  "L",  "L",  "L", "CE",  "S",  "S"],
+    ["S",  "S", "CW",  "L",  "L",  "L",  "L",  "L",  "L",  "L", "CE",  "S",  "S"],
+    ["S",  "S", "CW",  "L",  "L",  "L",  "L",  "L",  "L",  "L", "CE",  "S",  "S"],
+    ["S",  "S", "CW",  "L",  "L",  "L",  "L",  "L",  "L",  "L", "CE",  "S",  "S"],
+    ["S",  "S", "CW",  "L",  "L",  "L",  "L",  "L",  "L",  "L", "CE",  "S",  "S"],
+    ["S",  "S", "CW",  "L",  "L",  "L",  "L",  "L",  "L",  "L", "CE",  "S",  "S"],    
+    ["S",  "S", "C1", "CS", "CS", "CS", "CS", "CS", "CS", "CS", "C2",  "S",  "S"],
+    ["S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S",  "S"]
+];
+
+function weightsOfTiles(sample) = 
+    let(
+	    symbols = flat(sample),
+		leng = len(symbols),
+		weights = hashmap(number_of_buckets = sqrt(leng))
+	)
+    _weightsOfTiles(weights, symbols, leng);
+
+function _weightsOfTiles(weights, symbols, leng, i = 0) =
+    i == leng ? weights :
+	    let(
+		    tile = symbols[i],
+			w = hashmap_get(weights, tile)
+	    )
+        w == undef ? 
+		    _weightsOfTiles(hashmap_put(weights, tile, 1), symbols, leng, i + 1) :
+			_weightsOfTiles(hashmap_put(weights, tile, w + 1), symbols, leng, i + 1);
+			
+function initialEigenstates(width, height, weights) =
+	let(
+	    keys = hashmap_keys(weights),
+        row = [for(x = [0:width - 1]) keys]
+	)	
+	[for(y = [0:height - 1]) row];
+
+// wave function
+function waveFunction(width, height, weights) = 
+    [width, height, weights, initialEigenstates(width, height, weights)];	
+
+function wf_width(wf) = wf[0];
+function wf_height(wf) = wf[1];
+function wf_weights(wf) = wf[2];
+function wf_eigenstates(wf) = wf[3];
+function wf_eigenstates_at(wf, x, y) = wf_eigenstates(wf)[y][x];
+
+function wf_isAllCollapsed(wf) = every(
+    wf_eigenstates(wf), 
+	function(row) every(row, function(states) len(states) == 1)
+);
+
+function wf_remove(wf, x, y, removedStates) =
+    let(
+	    eigenstates = wf_eigenstates(wf),
+		rowsBeforeY = slice(eigenstates, 0, y),
+		rowY = eigenstates[y],
+		rowsAfterY = slice(eigenstates, y + 1),
+		statesBeforeX = slice(rowY, 0, x),
+		states = rowY[x],
+		statesAfterX = slice(rowY, x + 1),
+		newRowY = concat(
+		    statesBeforeX,
+		    [[for(state = states) if(!has(removedStates, state)) state]],
+			statesAfterX
+		)
+	)
+	[
+	    wf_width(wf),
+		wf_height(wf),
+		wf_weights(wf),
+		concat(
+		    rowsBeforeY,
+			[newRowY],
+			rowsAfterY
+		)
+	];
+
+width = len(sample[0]);
+height = len(sample);
+weights = weightsOfTiles(sample);
+
+wf = waveFunction(width, height, weights);
+eigenstates = initialEigenstates(width, height, weights);
+
+assert(wf_width(wf) == width);
+assert(wf_height(wf) == height);
+assert(wf_eigenstates(wf) == eigenstates);
+assert(wf_isAllCollapsed(wf) == false);
+assert(wf_remove(wf, 0, 0, []) == wf);
+assert(wf_eigenstates_at(wf_remove(wf, 0, 0, ["CE"]), 0, 0) == ["C0", "C1", "CS", "C2", "C3", "S", "CW", "CN", "L"]);