@polygonjs/polygonjs/src/engine/nodes/js/code/assemblers/_BaseJsPersistedConfig.ts

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

import {PersistedConfigWithShaders} from '../../.././utils/BasePersistedConfig';
import {BaseJsShaderAssembler, BaseFunctionData} from './_Base';
import {
	RegisterableVariable,
	SerializedVariable,
	SerializedVariableType,
	isVariableSerializable,
	serializeVariable,
	deserializeVariable,
} from './_BaseJsPersistedConfigUtils';
// import {SDFBuilderSopNode} from '../../../../sop/SDFBuilder';
import {Poly} from '../../../../Poly';
import {NamedFunctionMap} from '../../../../poly/registers/functions/All';
import {JsParamConfigJSON} from '../utils/JsParamConfig';
import {ParamType} from '../../../../poly/ParamType';
import {AssemblerControllerNode} from '../Controller';

export interface PersistedConfigBaseJsData extends PersistedConfigWithShaders {
	variableNames: string[];
	variables: SerializedVariable<SerializedVariableType>[];
	functionNames: Array<keyof NamedFunctionMap>;
	serializedParamConfigs: JsParamConfigJSON<ParamType>[];
}
export interface SingleBodyPersistedConfigBaseJsData extends PersistedConfigBaseJsData {
	functionBody: string;
}

export function serializedVariablesFromFunctionData(functionData: BaseFunctionData) {
	const {variableNames, variablesByName} = functionData;
	const serializedVariables: SerializedVariable<SerializedVariableType>[] = [];
	for (const variableName of variableNames) {
		const variable = variablesByName[variableName];
		if (variable != null && isVariableSerializable(variable)) {
			const serialized = serializeVariable(variable);
			serializedVariables.push(serialized);
		}
	}
	return serializedVariables;
}
export function variablesByNameFromPersistedConfigData(data: PersistedConfigBaseJsData) {
	const {variableNames, variables} = data;
	const variablesByName: Record<string, RegisterableVariable> = {};
	let i = 0;
	for (const variableName of variableNames) {
		const serialized = variables[i];
		const deserialized = deserializeVariable(serialized);
		variablesByName[variableName] = deserialized;
		i++;
	}
	return variablesByName;
}
export function functionsByNameFromPersistedConfigData(
	data: PersistedConfigBaseJsData,
	node: AssemblerControllerNode<BaseJsShaderAssembler>
) {
	const {functionNames} = data;
	const functionsByName: Record<string, Function> = {};
	// let i = 0;
	for (const functionName of functionNames) {
		const namedFunction = Poly.namedFunctionsRegister.getFunction(functionName, node);
		if (namedFunction) {
			functionsByName[functionName] = namedFunction.func.bind(namedFunction);
		}
		// i++;
	}
	return functionsByName;
}