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