@polygonjs/polygonjs
Version:
node-based WebGL 3D engine https://polygonjs.com
57 lines (56 loc) • 1.77 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 { WFCBuilder } from "../../../core/wfc/WFCBuilder";
import { registerWFCBuilder } from "../../../core/wfc/WFCRegister";
class WFCBuilderSopParamsConfig extends NodeParamsConfig {
constructor() {
super(...arguments);
/** @param tileHeight */
this.tileHeight = ParamConfig.FLOAT(1, {
range: [0, 2],
rangeLocked: [true, false]
});
}
}
const ParamsConfig = new WFCBuilderSopParamsConfig();
export class WFCBuilderSopNode extends TypedSopNode {
constructor() {
super(...arguments);
this.paramsConfig = ParamsConfig;
}
static type() {
return SopType.WFC_BUILDER;
}
initializeNode() {
this.io.inputs.setCount(2);
this.io.inputs.initInputsClonedState(InputCloneMode.NEVER);
}
cook(inputCoreGroups) {
const coreGroup0 = inputCoreGroups[0];
const coreGroup1 = inputCoreGroups[1];
const quadObjects = coreGroup0.quadObjects();
if (quadObjects == null || quadObjects.length == 0) {
this.states.error.set("no quad objects found");
return;
}
const tileAndRuleObjects = coreGroup1.threejsObjects();
const newObjects = [];
for (const quadObject of quadObjects) {
const builder = new WFCBuilder({
node: this,
quadObject,
tileAndRuleObjects,
tileHeight: this.pv.tileHeight
});
const group = builder.createObjects();
if (group) {
registerWFCBuilder(builder, group);
newObjects.push(group);
}
}
this.setObjects(newObjects);
}
}