UNPKG

@polygonjs/polygonjs

Version:

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

51 lines (50 loc) 1.95 kB
"use strict"; import { BaseSDFJsNode } from "./_BaseSDF"; import { NodeParamsConfig, ParamConfig } from "../utils/params/ParamsConfig"; import { JsConnectionPointType, JsConnectionPoint } from "../utils/io/connections/Js"; import { isBooleanTrue } from "../../../core/Type"; import { Vector3 } from "three"; import { Poly } from "../../Poly"; const OUTPUT_NAME = "p"; class SDFElongateJsParamsConfig extends NodeParamsConfig { constructor() { super(...arguments); this.position = ParamConfig.VECTOR3([0, 0, 0], { hidden: true }); this.center = ParamConfig.VECTOR3([0, 0, 0]); this.mult = ParamConfig.VECTOR3([0, 0, 0]); this.fast = ParamConfig.BOOLEAN(1); } } const ParamsConfig = new SDFElongateJsParamsConfig(); export class SDFElongateJsNode extends BaseSDFJsNode { constructor() { super(...arguments); this.paramsConfig = ParamsConfig; } static type() { return "SDFElongate"; } initializeNode() { super.initializeNode(); this.io.connection_points.spare_params.setInputlessParamNames(["fast"]); this.io.outputs.setNamedOutputConnectionPoints([ new JsConnectionPoint(OUTPUT_NAME, JsConnectionPointType.VECTOR3) ]); } setLines(linesController) { const functionName = isBooleanTrue(this.pv.fast) ? "SDFElongateFast" : "SDFElongateSlow"; const position = this.position(linesController); const center = this.variableForInputParam(linesController, this.p.center); const mult = this.variableForInputParam(linesController, this.p.mult); const out = this.jsVarName(OUTPUT_NAME); const tmpVarName = linesController.addVariable(this, new Vector3()); const func = Poly.namedFunctionsRegister.getFunction(functionName, this, linesController); linesController.addBodyOrComputed(this, [ { dataType: JsConnectionPointType.VECTOR3, varName: out, value: func.asString(position, center, mult, tmpVarName) } ]); } }