polygonjs-engine/src/engine/nodes/gl/Refract.ts

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

import {BaseGlMathFunctionGlNode} from './_BaseMathFunction';
import {GlConnectionPointType} from '../utils/io/connections/Gl';

export class RefractGlNode extends BaseGlMathFunctionGlNode {
	static type() {
		return 'refract';
	}

	initializeNode() {
		super.initializeNode();

		this.io.connection_points.set_input_name_function((index: number) => ['I', 'N', 'eta'][index]);
		this.io.connection_points.set_output_name_function((index: number) => 'refract');
		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_method_name(): string {
		return 'refract';
	}

	protected _expected_input_types() {
		const type = this.io.connection_points.first_input_connection_type() || GlConnectionPointType.VEC3;
		return [type, type, GlConnectionPointType.FLOAT];
	}

	protected _expected_output_types() {
		const type = this._expected_input_types()[0];
		return [type];
	}
}