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

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

import {TypedGlNode} from './_Base';
import {ThreeToGl} from '../../../../src/core/ThreeToGl';

import Rgb2Hsv from './gl/rgb2hsv.glsl';
import {NodeParamsConfig, ParamConfig} from '../utils/params/ParamsConfig';
import {FunctionGLDefinition} from './utils/GLDefinition';
import {GlConnectionPoint, GlConnectionPointType} from '../utils/io/connections/Gl';
import {ShadersCollectionController} from './code/utils/ShadersCollectionController';

const OUTPUT_NAME = 'hsv';
class RgbToHsvGlParamsConfig extends NodeParamsConfig {
	rgb = ParamConfig.VECTOR3([1, 1, 1]);
}
const ParamsConfig = new RgbToHsvGlParamsConfig();
export class RgbToHsvGlNode extends TypedGlNode<RgbToHsvGlParamsConfig> {
	params_config = ParamsConfig;
	static type() {
		return 'rgbToHsv';
	}

	initializeNode() {
		this.io.outputs.setNamedOutputConnectionPoints([
			new GlConnectionPoint(OUTPUT_NAME, GlConnectionPointType.VEC3),
		]);
	}

	set_lines(shaders_collection_controller: ShadersCollectionController) {
		const function_declaration_lines = [];
		const body_lines = [];

		function_declaration_lines.push(new FunctionGLDefinition(this, Rgb2Hsv));

		const rgb = ThreeToGl.vector3(this.variable_for_input('rgb'));

		const hsv = this.gl_var_name('hsv');
		body_lines.push(`vec3 ${hsv} = rgb2hsv(${rgb})`);
		shaders_collection_controller.add_definitions(this, function_declaration_lines);
		shaders_collection_controller.add_body_lines(this, body_lines);
	}
}