@polygonjs/polygonjs
Version:
node-based WebGL 3D engine https://polygonjs.com
38 lines (37 loc) • 1.4 kB
JavaScript
;
import { TypedGlNode } from "./_Base";
import { GlConnectionPointType, GlConnectionPoint } from "../utils/io/connections/Gl";
import { ThreeToGl } from "../../../core/ThreeToGl";
import { ParamConfig, NodeParamsConfig } from "../utils/params/ParamsConfig";
import { FunctionGLDefinition } from "./utils/GLDefinition";
import PolarGlslLib from "./gl/polar.glsl";
const OUTPUT_NAME = "polar";
class CartesianToPolarGlParamsConfig extends NodeParamsConfig {
constructor() {
super(...arguments);
this.xyz = ParamConfig.VECTOR3([1, 1, 1]);
}
}
const ParamsConfig = new CartesianToPolarGlParamsConfig();
export class CartesianToPolarGlNode extends TypedGlNode {
constructor() {
super(...arguments);
this.paramsConfig = ParamsConfig;
}
static type() {
return "cartesianToPolar";
}
initializeNode() {
super.initializeNode();
this.io.outputs.setNamedOutputConnectionPoints([
new GlConnectionPoint(OUTPUT_NAME, GlConnectionPointType.VEC3)
]);
}
setLines(shadersCollectionController) {
const out = this.glVarName(OUTPUT_NAME);
const xyz = ThreeToGl.vector3(this.variableForInputParam(this.p.xyz));
const bodyLine = `vec3 ${out} = cartesianToPolar(${xyz})`;
shadersCollectionController.addBodyLines(this, [bodyLine]);
shadersCollectionController.addDefinitions(this, [new FunctionGLDefinition(this, PolarGlslLib)]);
}
}