@polygonjs/polygonjs/src/engine/nodes/post/EffectPass.ts

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

/**
 * Adds an effect pass, which can combine multiples passes into 1.
 *
 *
 */
import {TypedPostNode, TypedPostNodeContext} from './_Base';
import {NodeParamsConfig, ParamConfig} from '../utils/params/ParamsConfig';
import {Effect, EffectPass, Pass} from 'postprocessing';
import {rangeWithEnd, arrayCompact} from '../../../core/ArrayUtils';
import {BaseNodeType} from '../_Base';
import {NodeEvent} from '../../poly/NodeEvent';
import {PostType} from '../../poly/registers/nodes/types/Post';

const POSSIBLE_COMBINED_PASSES_COUNT = 5;
export const RANGE = rangeWithEnd(POSSIBLE_COMBINED_PASSES_COUNT);
const DEFAULT_INPUTS_COUNT = POSSIBLE_COMBINED_PASSES_COUNT + 1;

class EffectPassPostParamsConfig extends NodeParamsConfig {
	/** @param number of inputs that this node can merge geometries from */
	inputsCount = ParamConfig.INTEGER(DEFAULT_INPUTS_COUNT, {
		range: [1, 32],
		rangeLocked: [true, false],
		callback: (node: BaseNodeType) => {
			EffectPassPostNode.PARAM_CALLBACK_setInputsCount(node as EffectPassPostNode);
		},
	});
}
const ParamsConfig = new EffectPassPostParamsConfig();
export class EffectPassPostNode extends TypedPostNode<EffectPass, EffectPassPostParamsConfig> {
	override paramsConfig = ParamsConfig;
	static override type() {
		return PostType.EFFECT_PASS;
	}

	override initializeNode() {
		super.initializeNode();
		this.io.inputs.setCount(DEFAULT_INPUTS_COUNT);

		this.params.onParamsCreated('update inputs', () => {
			this._callbackUpdateInputsCount();
		});
	}

	override _setupComposerIfActive(context: TypedPostNodeContext): void {
		const inputIndices = RANGE.map((i) => i + 1);
		const passesToCombine: Pass[] = [];
		arrayCompact(
			inputIndices.map((index) => {
				const input = this.io.inputs.input(index);
				if (input) {
					if (!input.flags.bypass.active()) {
						return input.createPassForContext(context);
					}
				}
			}),
			passesToCombine
		).flat();

		const effectLists: Effect[][] = [];
		arrayCompact(
			passesToCombine.map((pass) => {
				return (pass as any).effects as undefined | Effect[];
			}),
			effectLists
		);
		const effects = effectLists.flat();
		const pass = new EffectPass(context.camera, ...effects);
		context.composerController.addPassByNodeInBuildPassesProcess(this, pass, context.composer);
	}

	override updatePass(pass: EffectPass) {}

	private _callbackUpdateInputsCount() {
		this.io.inputs.setCount(1, this.pv.inputsCount);
		this.emit(NodeEvent.INPUTS_UPDATED);
	}
	static PARAM_CALLBACK_setInputsCount(node: EffectPassPostNode) {
		node._callbackUpdateInputsCount();
	}
}