@polygonjs/polygonjs
Version:
node-based WebGL 3D engine https://polygonjs.com
132 lines (131 loc) • 5.39 kB
JavaScript
;
import { TypedJsNode } from "./_Base";
import { NodeParamsConfig, ParamConfig } from "../utils/params/ParamsConfig";
import { JsConnectionPoint, JsConnectionPointType, JS_CONNECTION_POINT_IN_NODE_DEF } from "../utils/io/connections/Js";
import { inputObject3D, setObject3DOutputLine } from "./_BaseObject3D";
import { Poly } from "../../Poly";
const CONNECTION_OPTIONS = JS_CONNECTION_POINT_IN_NODE_DEF;
export const SetGeometryInstanceAttributeInputName = {
// [JsConnectionPointType.TRIGGER]: JsConnectionPointType.TRIGGER,
// [JsConnectionPointType.OBJECT_3D]: JsConnectionPointType.OBJECT_3D,
values: "values",
lerp: "lerp",
attributeNeedsUpdate: "attributeNeedsUpdate"
};
const INPUT_NAMES = [
// SetGeometryInstancePositionsInputName.trigger,
// SetGeometryInstancePositionsInputName.Object3D,
SetGeometryInstanceAttributeInputName.values,
SetGeometryInstanceAttributeInputName.lerp,
SetGeometryInstanceAttributeInputName.attributeNeedsUpdate
];
const DefaultValues = {
[SetGeometryInstanceAttributeInputName.lerp]: 1,
[SetGeometryInstanceAttributeInputName.attributeNeedsUpdate]: true
};
const ALLOWED_TYPES = [
JsConnectionPointType.FLOAT_ARRAY,
JsConnectionPointType.COLOR_ARRAY,
JsConnectionPointType.QUATERNION_ARRAY,
JsConnectionPointType.VECTOR2_ARRAY,
JsConnectionPointType.VECTOR3_ARRAY,
JsConnectionPointType.VECTOR4_ARRAY
];
class SetGeometryInstanceAttributeJsParamsConfig extends NodeParamsConfig {
constructor() {
super(...arguments);
/** @param attribute name */
this.attribName = ParamConfig.STRING("");
}
}
const ParamsConfig = new SetGeometryInstanceAttributeJsParamsConfig();
export class SetGeometryInstanceAttributeJsNode extends TypedJsNode {
constructor() {
super(...arguments);
this.paramsConfig = ParamsConfig;
}
static type() {
return "setGeometryInstanceAttribute";
}
initializeNode() {
super.initializeNode();
this.io.connection_points.spare_params.setInputlessParamNames(["attribName", "size"]);
this.io.inputs.setNamedInputConnectionPoints([
new JsConnectionPoint(JsConnectionPointType.TRIGGER, JsConnectionPointType.TRIGGER, CONNECTION_OPTIONS),
new JsConnectionPoint(JsConnectionPointType.OBJECT_3D, JsConnectionPointType.OBJECT_3D, CONNECTION_OPTIONS)
]);
this.io.connection_points.set_input_name_function((i) => this._expectedInputName(i));
this.io.connection_points.set_expected_input_types_function(() => this._expectedInputTypes());
this.io.connection_points.set_output_name_function(
(i) => [JsConnectionPointType.TRIGGER, JsConnectionPointType.OBJECT_3D][i]
);
this.io.connection_points.set_expected_output_types_function(() => [
JsConnectionPointType.TRIGGER,
JsConnectionPointType.OBJECT_3D
]);
}
_expectedInputTypes() {
const firstInputType = this.io.connection_points.first_input_connection_type();
const type = firstInputType && ALLOWED_TYPES.includes(firstInputType) ? firstInputType : JsConnectionPointType.FLOAT_ARRAY;
return [type, JsConnectionPointType.FLOAT, JsConnectionPointType.BOOLEAN];
}
paramDefaultValue(name) {
return DefaultValues[name];
}
// expectedInputTypes() {
// return [
// JsConnectionPointType.TRIGGER,
// JsConnectionPointType.OBJECT_3D,
// JsConnectionPointType.VECTOR3_ARRAY,
// JsConnectionPointType.FLOAT,
// JsConnectionPointType.BOOLEAN,
// ];
// }
// protected _expectedOutputTypes() {
// return [JsConnectionPointType.TRIGGER];
// }
setLines(linesController) {
setObject3DOutputLine(this, linesController);
}
_expectedInputName(index) {
return INPUT_NAMES[index];
}
setTriggerableLines(shadersCollectionController) {
const object3D = inputObject3D(this, shadersCollectionController);
const attribName = this.pv.attribName;
const values = this.variableForInput(shadersCollectionController, SetGeometryInstanceAttributeInputName.values);
const lerp = this.variableForInput(shadersCollectionController, SetGeometryInstanceAttributeInputName.lerp);
const attributeNeedsUpdate = this.variableForInput(
shadersCollectionController,
SetGeometryInstanceAttributeInputName.attributeNeedsUpdate
);
const functionName = this._functionName();
const func = Poly.namedFunctionsRegister.getFunction(functionName, this, shadersCollectionController);
const bodyLine = func.asString(object3D, attribName, values, lerp, attributeNeedsUpdate);
shadersCollectionController.addTriggerableLines(this, [bodyLine]);
}
_functionName() {
const type = this._expectedInputTypes()[0];
switch (type) {
case JsConnectionPointType.FLOAT_ARRAY: {
return "setGeometryInstanceAttributeFloat";
}
case JsConnectionPointType.COLOR_ARRAY: {
return "setGeometryInstanceAttributeColor";
}
case JsConnectionPointType.QUATERNION_ARRAY: {
return "setGeometryInstanceAttributeQuaternion";
}
case JsConnectionPointType.VECTOR2_ARRAY: {
return "setGeometryInstanceAttributeVector2";
}
case JsConnectionPointType.VECTOR3_ARRAY: {
return "setGeometryInstanceAttributeVector3";
}
case JsConnectionPointType.VECTOR4_ARRAY: {
return "setGeometryInstanceAttributeVector4";
}
}
return "setGeometryInstanceAttributeFloat";
}
}