3dapp/libs/three/jsm/nodes/core/FunctionCallNode.js

import { TempNode } from './TempNode.js';

class FunctionCallNode extends TempNode {

	constructor( func, inputs ) {

		super();

		this.setFunction( func, inputs );

	}

	setFunction( func, inputs = [] ) {

		this.value = func;
		this.inputs = inputs;

	}

	getFunction() {

		return this.value;

	}

	getType( builder ) {

		return this.value.getType( builder );

	}

	generate( builder, output ) {

		const type = this.getType( builder ),
			func = this.value;

		let code = func.build( builder, output ) + '( ';
		const params = [];

		for ( let i = 0; i < func.inputs.length; i ++ ) {

			const inpt = func.inputs[ i ],
				param = this.inputs[ i ] || this.inputs[ inpt.name ];

			params.push( param.build( builder, builder.getTypeByFormat( inpt.type ) ) );

		}

		code += params.join( ', ' ) + ' )';

		return builder.format( code, type, output );

	}

	copy( source ) {

		super.copy( source );

		for ( const prop in source.inputs ) {

			this.inputs[ prop ] = source.inputs[ prop ];

		}

		this.value = source.value;

		return this;

	}

	toJSON( meta ) {

		let data = this.getJSONNode( meta );

		if ( ! data ) {

			const func = this.value;

			data = this.createJSONNode( meta );

			data.value = this.value.toJSON( meta ).uuid;

			if ( func.inputs.length ) {

				data.inputs = {};

				for ( let i = 0; i < func.inputs.length; i ++ ) {

					const inpt = func.inputs[ i ],
						node = this.inputs[ i ] || this.inputs[ inpt.name ];

					data.inputs[ inpt.name ] = node.toJSON( meta ).uuid;

				}

			}

		}

		return data;

	}

}

FunctionCallNode.prototype.nodeType = 'FunctionCall';

export { FunctionCallNode };