polygonjs-engine/src/engine/nodes/post/DotScreen.ts

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

/**
 * Adds a dot effect
 *
 *
 */
import {TypedPostProcessNode, TypedPostNodeContext, PostParamOptions} from './_Base';
import {DotScreenShader} from '../../../modules/three/examples/jsm/shaders/DotScreenShader';
import {ShaderPass} from '../../../modules/three/examples/jsm/postprocessing/ShaderPass';
import {IUniformN, IUniformV2} from '../utils/code/gl/Uniforms';

interface DotScreenPassWithUniforms extends ShaderPass {
	uniforms: {
		center: IUniformV2;
		angle: IUniformN;
		scale: IUniformN;
	};
}

import {NodeParamsConfig, ParamConfig} from '../utils/params/ParamsConfig';
class DotScreenPostParamsConfig extends NodeParamsConfig {
	center = ParamConfig.VECTOR2([0.5, 0.5], {
		...PostParamOptions,
	});
	angle = ParamConfig.FLOAT('$PI*0.5', {
		range: [0, 10],
		rangeLocked: [false, false],
		...PostParamOptions,
	});
	scale = ParamConfig.FLOAT(1.0, {
		range: [0, 1],
		rangeLocked: [false, false],
		...PostParamOptions,
	});
}
const ParamsConfig = new DotScreenPostParamsConfig();
export class DotScreenPostNode extends TypedPostProcessNode<ShaderPass, DotScreenPostParamsConfig> {
	params_config = ParamsConfig;
	static type() {
		return 'dotScreen';
	}

	protected _create_pass(context: TypedPostNodeContext) {
		const pass = new ShaderPass(DotScreenShader) as DotScreenPassWithUniforms;

		this.update_pass(pass);

		return pass;
	}
	update_pass(pass: DotScreenPassWithUniforms) {
		pass.uniforms.center.value = this.pv.center;
		pass.uniforms.angle.value = this.pv.angle;
		pass.uniforms.scale.value = this.pv.scale;
	}
}