@polygonjs/polygonjs/src/engine/nodes/js/code/utils/NodeFinder.ts

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

import {JsType} from './../../../../poly/registers/nodes/types/Js';
import {NodeContext} from './../../../../poly/NodeContext';
import {AssemblerControllerNode} from '../Controller';
import {BaseJsNodeType} from '../../_Base';
import {BaseJsShaderAssembler} from '../assemblers/_Base';

export class JsNodeFinder {
	static findOutputNodes(node: AssemblerControllerNode<BaseJsShaderAssembler>) {
		const output = node.nodesByType(JsType.OUTPUT);
		const outputAmbientLight = node.nodesByType(JsType.OUTPUT_AMBIENT_LIGHT);
		const outputAreaLight = node.nodesByType(JsType.OUTPUT_AREA_LIGHT);
		const outputDirectionalLight = node.nodesByType(JsType.OUTPUT_DIRECTIONAL_LIGHT);
		const outputHemisphereLight = node.nodesByType(JsType.OUTPUT_HEMISPHERE_LIGHT);
		const outputPointLight = node.nodesByType(JsType.OUTPUT_POINT_LIGHT);
		const outputSpotLight = node.nodesByType(JsType.OUTPUT_SPOT_LIGHT);

		return output
			.concat(outputAmbientLight)
			.concat(outputAreaLight)
			.concat(outputDirectionalLight)
			.concat(outputHemisphereLight)
			.concat(outputPointLight)
			.concat(outputSpotLight);
	}
	static findParamGeneratingNodes(node: AssemblerControllerNode<BaseJsShaderAssembler>) {
		const list: BaseJsNodeType[] = [];
		node.childrenController?.traverseChildren(
			(child) => {
				const childJsNode = child as BaseJsNodeType;
				if (childJsNode.paramsGenerating()) {
					list.push(childJsNode);
				}
			},
			(child) => {
				if (!child.childrenController) {
					return child.context() == NodeContext.GL;
				} else {
					return child.context() == NodeContext.GL && child.childrenController.context == NodeContext.GL;
				}
			}
		);

		return list;
	}
	// static findVaryingNodes(node: BaseJsParentNode) {
	// 	const nodes = node.nodesByType(JsType.VARYING_WRITE);
	// 	return nodes;
	// }
	static findAttributeExportNodes(node: AssemblerControllerNode<BaseJsShaderAssembler>) {
		const nodes = node.nodesByType(JsType.ATTRIBUTE);
		return nodes.filter((node) => {
			// do not use attributes that are used as an input, as export
			// return (node.used_output_names().length == 0) &&
			return node.isExporting();
		});
	}
}