UNPKG

@polygonjs/polygonjs

Version:

node-based WebGL 3D engine https://polygonjs.com

46 lines (45 loc) 1.55 kB
"use strict"; 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); } }