three 基础库
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

108 lines
1.6 KiB

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 };