@polygonjs/polygonjs/dist/src/engine/nodes/gl/Round.js

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

"use strict";
import { BaseNodeGlMathFunctionArg1GlNode } from "./_BaseMathFunction";
import { ThreeToGl } from "../../../core/ThreeToGl";
import { GlConnectionPointComponentsCountMap } from "../utils/io/connections/Gl";
const ALL_COMPONENTS = ["x", "y", "z", "w"];
export class RoundGlNode extends BaseNodeGlMathFunctionArg1GlNode {
  static type() {
    return "round";
  }
  // https://hub.jmonkeyengine.org/t/round-with-glsl/8186/6
  setLines(shadersCollectionController) {
    const namedInputConnectionPoints = this.io.inputs.namedInputConnectionPoints();
    const namedOutputConnectionPoints = this.io.outputs.namedOutputConnectionPoints();
    if (!(namedInputConnectionPoints && namedOutputConnectionPoints)) {
      return;
    }
    const inputConnection = namedInputConnectionPoints[0];
    const value = ThreeToGl.vector2(this.variableForInput(inputConnection.name()));
    const outputConnection = namedOutputConnectionPoints[0];
    const outputType = outputConnection.type();
    const varName = this.glVarName(outputConnection.name());
    const bodyLines = [];
    const linesCount = GlConnectionPointComponentsCountMap[outputType];
    if (linesCount == 1) {
      bodyLines.push(`${outputType} ${varName} = ${this._singleLine(value)}`);
    } else {
      const singleLines = ALL_COMPONENTS.slice(0, linesCount).map((c) => {
        return this._singleLine(`${value}.${c}`);
      });
      bodyLines.push(`${outputType} ${varName} = ${outputType}(${singleLines.join(",")})`);
    }
    shadersCollectionController.addBodyLines(this, bodyLines);
  }
  _singleLine(value) {
    return `sign(${value})*floor(abs(${value})+0.5)`;
  }
}