polygonjs-engine/dist/src/engine/nodes/gl/Switch.js

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

import {TypedGlNode} from "./_Base";
import {GlConnectionPointType} from "../utils/io/connections/Gl";
import {ThreeToGl as ThreeToGl2} from "../../../core/ThreeToGl";
import {NodeParamsConfig} from "../utils/params/ParamsConfig";
import {CoreMath} from "../../../core/math/_Module";
const MAX_INPUTS_COUNT = 16;
class SwitchParamsConfig extends NodeParamsConfig {
}
const ParamsConfig2 = new SwitchParamsConfig();
const SwitchGlNode2 = class extends TypedGlNode {
  constructor() {
    super(...arguments);
    this.params_config = ParamsConfig2;
  }
  static type() {
    return "switch";
  }
  initializeNode() {
    this.io.connection_points.set_input_name_function(this._gl_input_name.bind(this));
    this.io.connection_points.set_expected_input_types_function(this._expected_input_types.bind(this));
    this.io.connection_points.set_expected_output_types_function(this._expected_output_types.bind(this));
  }
  _gl_input_name(index) {
    if (index == 0) {
      return SwitchGlNode2.INPUT_INDEX;
    } else {
      return `in${index - 1}`;
    }
  }
  _expected_input_types() {
    const second_input_type = this.io.connection_points.input_connection_type(1);
    const type = second_input_type || GlConnectionPointType.FLOAT;
    const current_connections = this.io.connections.input_connections();
    const expected_count = current_connections ? CoreMath.clamp(current_connections.length, 2, MAX_INPUTS_COUNT) : 2;
    const expected_input_types = [GlConnectionPointType.INT];
    for (let i = 0; i < expected_count; i++) {
      expected_input_types.push(type);
    }
    return expected_input_types;
  }
  _expected_output_types() {
    const input_types = this._expected_input_types();
    const type = input_types[1] || GlConnectionPointType.FLOAT;
    return [type];
  }
  set_lines(shaders_collection_controller) {
    const var_type = this.io.outputs.named_output_connection_points[0].type();
    const out = this.gl_var_name(this.io.connection_points.output_name(0));
    const index_point_name = this.io.connection_points.input_name(0);
    const arg_index = ThreeToGl2.int(this.variable_for_input(index_point_name));
    const switch_index_var_name = this.gl_var_name("index");
    const body_lines = [`${var_type} ${out};`, `int ${switch_index_var_name} = ${arg_index}`];
    const lines_count = this._expected_input_types().length - 1;
    for (let i = 0; i < lines_count; i++) {
      const if_else = i == 0 ? "if" : "else if";
      const condition = `${switch_index_var_name} == ${i}`;
      const connection_point_name = this.io.connection_points.input_name(i + 1);
      const assignment = `${out} = ${ThreeToGl2.any(this.variable_for_input(connection_point_name))};`;
      const body_line = `${if_else}(${condition}){${assignment}}`;
      body_lines.push(body_line);
    }
    shaders_collection_controller.add_body_lines(this, body_lines);
  }
};
export let SwitchGlNode = SwitchGlNode2;
SwitchGlNode.INPUT_INDEX = "index";