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

add wf_entropy

Browse Source
This commit is contained in:
Justin Lin
2021-03-13 18:08:57 +08:00
parent ef50ee9178
commit 95d892b45a
1 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/experimental/tiles_wfc.scad
View File

@@ -99,6 +99,25 @@ function _wf_collapse(wf, x, y, states_weights, leng, threshold, i = 0) =
function _oneStateAt(wf, x, y, state) = _replaceStatesAt(wf, x, y, [state]); function _oneStateAt(wf, x, y, state) = _replaceStatesAt(wf, x, y, [state]);
function wf_entropy(wf, x, y) =
let(
states = wf_eigenstates_at(wf, x, y),
weights = wf_weights(wf),
state_leng = len(states),
sumOfWeights_sumOfWeightLogWeights = _wf_entropy(weights, states, state_leng, 0, 0),
sumOfWeights = sumOfWeights_sumOfWeightLogWeights[0],
sumOfWeightLogWeights = sumOfWeights_sumOfWeightLogWeights[1]
)
ln(sumOfWeights) - (sumOfWeightLogWeights / sumOfWeights);
function _wf_entropy(weights, states, state_leng, sumOfWeights, sumOfWeightLogWeights, i = 0) =
i == state_leng ? [sumOfWeights, sumOfWeightLogWeights] :
let(
opt = states[i],
weight = hashmap_get(weights, opt)
)
_wf_entropy(weights, states, state_leng, sumOfWeights + weight, sumOfWeightLogWeights + weight * ln(weight), i + 1);
function _replaceStatesAt(wf, x, y, states) = function _replaceStatesAt(wf, x, y, states) =
let( let(
eigenstates = wf_eigenstates(wf), eigenstates = wf_eigenstates(wf),
@@ -124,6 +143,8 @@ function _replaceStatesAt(wf, x, y, states) =
) )
]; ];
width = len(sample[0]); width = len(sample[0]);
height = len(sample); height = len(sample);
weights = weightsOfTiles(sample); weights = weightsOfTiles(sample);
@@ -141,4 +162,4 @@ for(y = [0:height - 1]) {
assert(len(wf_eigenstates_at(wf_collapse(wf, x, y), x, y)) == 1); assert(len(wf_eigenstates_at(wf_collapse(wf, x, y), x, y)) == 1);
} }
} }
assert(wf_entropy(wf, 0, 0) == 1.458879520793018);