@polygonjs/polygonjs/src/engine/poly/registers/functions/FunctionsRegister.ts

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

import {JsLinesCollectionController} from '../../../nodes/js/code/utils/JsLinesCollectionController';
import {BaseNodeType} from '../../../nodes/_Base';
import {NamedFunctionMap} from './All';
import {BaseNamedFunctionRegister} from './_BaseRegister';

export class NamedFunctionRegister extends BaseNamedFunctionRegister {
	getFunction<K extends keyof NamedFunctionMap>(
		functionName: K,
		node: BaseNodeType,
		shadersCollectionController?: JsLinesCollectionController
	): NamedFunctionMap[K] {
		const funcClass = this._functionByName.get(functionName);
		if (!funcClass) {
			console.error(`function not registered:'${functionName}'`);
		}
		const func = new (funcClass as any)(node, shadersCollectionController);
		return func;
		// return func as unknown as NamedFunctionMap[K];
	}
}