openscad/dotSCAD
1
0
Fork 0
mirror of https://github.com/JustinSDK/dotSCAD.git synced 2025-08-17 20:11:50 +02:00
Code Issues Releases Wiki Activity

add wfc (under working)

Browse Source
This commit is contained in:
Justin Lin
2021-03-12 09:27:43 +08:00
parent 93e7efd697
commit 39c79cd4d8
1 changed files with 106 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

106
src/experimental/tiles_wfc.scad Normal file
View File

@@ -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"]);