168 lines
3.2 KiB
JavaScript
168 lines
3.2 KiB
JavaScript
|
import Node, { addNodeClass } from '../core/Node.js';
|
||
|
|
import { arrayBufferToBase64, base64ToArrayBuffer } from '../core/NodeUtils.js';
|
||
|
|
import { addNodeElement, nodeProxy, float } from '../shadernode/ShaderNode.js';
|
||
|
|
import { EventDispatcher } from 'three';
|
||
|
|
|
||
|
|
class ScriptableValueNode extends Node {
|
||
|
|
|
||
|
|
constructor( value = null ) {
|
||
|
|
|
||
|
|
super();
|
||
|
|
|
||
|
|
this._value = value;
|
||
|
|
this._cache = null;
|
||
|
|
|
||
|
|
this.inputType = null;
|
||
|
|
this.outpuType = null;
|
||
|
|
|
||
|
|
this.events = new EventDispatcher();
|
||
|
|
|
||
|
|
this.isScriptableValueNode = true;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
get isScriptableOutputNode() {
|
||
|
|
|
||
|
|
return this.outputType !== null;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
set value( val ) {
|
||
|
|
|
||
|
|
if ( this._value === val ) return;
|
||
|
|
|
||
|
|
if ( this._cache && this.inputType === 'URL' && this.value.value instanceof ArrayBuffer ) {
|
||
|
|
|
||
|
|
URL.revokeObjectURL( this._cache );
|
||
|
|
|
||
|
|
this._cache = null;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
this._value = val;
|
||
|
|
|
||
|
|
this.events.dispatchEvent( { type: 'change' } );
|
||
|
|
|
||
|
|
this.refresh();
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
get value() {
|
||
|
|
|
||
|
|
return this._value;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
refresh() {
|
||
|
|
|
||
|
|
this.events.dispatchEvent( { type: 'refresh' } );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
getValue() {
|
||
|
|
|
||
|
|
const value = this.value;
|
||
|
|
|
||
|
|
if ( value && this._cache === null && this.inputType === 'URL' && value.value instanceof ArrayBuffer ) {
|
||
|
|
|
||
|
|
this._cache = URL.createObjectURL( new Blob( [ value.value ] ) );
|
||
|
|
|
||
|
|
} else if ( value && value.value !== null && value.value !== undefined && (
|
||
|
|
( ( this.inputType === 'URL' || this.inputType === 'String' ) && typeof value.value === 'string' ) ||
|
||
|
|
( this.inputType === 'Number' && typeof value.value === 'number' ) ||
|
||
|
|
( this.inputType === 'Vector2' && value.value.isVector2 ) ||
|
||
|
|
( this.inputType === 'Vector3' && value.value.isVector3 ) ||
|
||
|
|
( this.inputType === 'Vector4' && value.value.isVector4 ) ||
|
||
|
|
( this.inputType === 'Color' && value.value.isColor ) ||
|
||
|
|
( this.inputType === 'Matrix3' && value.value.isMatrix3 ) ||
|
||
|
|
( this.inputType === 'Matrix4' && value.value.isMatrix4 )
|
||
|
|
) ) {
|
||
|
|
|
||
|
|
return value.value;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return this._cache || value;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
getNodeType( builder ) {
|
||
|
|
|
||
|
|
return this.value && this.value.isNode ? this.value.getNodeType( builder ) : 'float';
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
setup() {
|
||
|
|
|
||
|
|
return this.value && this.value.isNode ? this.value : float();
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
serialize( data ) {
|
||
|
|
|
||
|
|
super.serialize( data );
|
||
|
|
|
||
|
|
if ( this.value !== null ) {
|
||
|
|
|
||
|
|
if ( this.inputType === 'ArrayBuffer' ) {
|
||
|
|
|
||
|
|
data.value = arrayBufferToBase64( this.value );
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
data.value = this.value ? this.value.toJSON( data.meta ).uuid : null;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
data.value = null;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
data.inputType = this.inputType;
|
||
|
|
data.outputType = this.outputType;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
deserialize( data ) {
|
||
|
|
|
||
|
|
super.deserialize( data );
|
||
|
|
|
||
|
|
let value = null;
|
||
|
|
|
||
|
|
if ( data.value !== null ) {
|
||
|
|
|
||
|
|
if ( data.inputType === 'ArrayBuffer' ) {
|
||
|
|
|
||
|
|
value = base64ToArrayBuffer( data.value );
|
||
|
|
|
||
|
|
} else if ( data.inputType === 'Texture' ) {
|
||
|
|
|
||
|
|
value = data.meta.textures[ data.value ];
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
value = data.meta.nodes[ data.value ] || null;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
this.value = value;
|
||
|
|
|
||
|
|
this.inputType = data.inputType;
|
||
|
|
this.outputType = data.outputType;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
export default ScriptableValueNode;
|
||
|
|
|
||
|
|
export const scriptableValue = nodeProxy( ScriptableValueNode );
|
||
|
|
|
||
|
|
addNodeElement( 'scriptableValue', scriptableValue );
|
||
|
|
|
||
|
|
addNodeClass( 'ScriptableValueNode', ScriptableValueNode );
|