@polygonjs/polygonjs
Version:
node-based WebGL 3D engine https://polygonjs.com
46 lines (45 loc) • 1.55 kB
JavaScript
;
import { TypedSopNode } from "./_Base";
import { NodeParamsConfig, ParamConfig } from "../utils/params/ParamsConfig";
import { SopType } from "../../poly/registers/nodes/types/Sop";
import { InputCloneMode } from "../../poly/InputCloneMode";
import { CoreWFCTileAttribute } from "../../../core/wfc/WFCAttributes";
import { stringMatchMask } from "../../../core/String";
class WFCRuleTileWeightSopParamsConfig extends NodeParamsConfig {
constructor() {
super(...arguments);
/** @param tile id */
this.tileId = ParamConfig.STRING("*");
/** @param weight */
this.weight = ParamConfig.FLOAT(1, {
range: [0, 1],
rangeLocked: [true, true]
});
}
}
const ParamsConfig = new WFCRuleTileWeightSopParamsConfig();
export class WFCRuleTileWeightSopNode extends TypedSopNode {
constructor() {
super(...arguments);
this.paramsConfig = ParamsConfig;
}
static type() {
return SopType.WFC_RULE_TILE_WEIGHT;
}
initializeNode() {
this.io.inputs.setCount(1);
this.io.inputs.initInputsClonedState(InputCloneMode.FROM_NODE);
}
async cook(inputCoreGroups) {
const coreGroup0 = inputCoreGroups[0];
const inputObjects = coreGroup0.threejsObjects();
const { tileId, weight } = this.pv;
const tileObjects = inputObjects.filter(
(tileObject) => stringMatchMask(CoreWFCTileAttribute.getTileId(tileObject), tileId)
);
for (const tileObject of tileObjects) {
CoreWFCTileAttribute.setWeight(tileObject, weight);
}
this.setObjects(inputObjects);
}
}