@polygonjs/polygonjs/src/engine/nodes/gl/NormalToWorld.ts

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

/**
 * normals from local to world
 *
 *
 *
 */

import {TypedGlNode} from './_Base';
import {NodeParamsConfig, ParamConfig} from '../utils/params/ParamsConfig';
import {GlConnectionPointType, GlConnectionPoint} from '../utils/io/connections/Gl';
import {ShadersCollectionController} from './code/utils/ShadersCollectionController';
import {ShaderName} from '../utils/shaders/ShaderName';
import {ThreeToGl} from '../../../core/ThreeToGl';

const OUTPUT_NAME = 'worldNormal';

class NormalToWorldGlParamsConfig extends NodeParamsConfig {
	normal = ParamConfig.VECTOR3([0, 1, 0]);
}
const ParamsConfig = new NormalToWorldGlParamsConfig();
export class NormalToWorldGlNode extends TypedGlNode<NormalToWorldGlParamsConfig> {
	override paramsConfig = ParamsConfig;
	static override type(): Readonly<'normalToWorld'> {
		return 'normalToWorld';
	}

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

	override setLines(linesController: ShadersCollectionController) {
		if (linesController.currentShaderName() == ShaderName.VERTEX) {
			const input = ThreeToGl.vector3(this.variableForInputParam(this.p.normal));
			const outValue = this.glVarName(OUTPUT_NAME);
			const bodyLine = `vec3 ${outValue} = normalize( mat3( modelMatrix[0].xyz, modelMatrix[1].xyz, modelMatrix[2].xyz ) * ${input} )`;
			linesController.addBodyLines(this, [bodyLine], ShaderName.VERTEX);
		}
	}
}