123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227 |
- import { MathUtils } from '../../../../build/three.module.js';
- function Node( type ) {
- this.uuid = MathUtils.generateUUID();
- this.name = '';
- this.type = type;
- this.userData = {};
- }
- Node.prototype = {
- constructor: Node,
- isNode: true,
- hashProperties: undefined,
- analyze: function ( builder, settings ) {
- settings = settings || {};
- builder.analyzing = true;
- this.build( builder.addFlow( settings.slot, settings.cache, settings.context ), 'v4' );
- builder.clearVertexNodeCode();
- builder.clearFragmentNodeCode();
- builder.removeFlow();
- builder.analyzing = false;
- },
- analyzeAndFlow: function ( builder, output, settings ) {
- settings = settings || {};
- this.analyze( builder, settings );
- return this.flow( builder, output, settings );
- },
- flow: function ( builder, output, settings ) {
- settings = settings || {};
- builder.addFlow( settings.slot, settings.cache, settings.context );
- var flow = {};
- flow.result = this.build( builder, output );
- flow.code = builder.clearNodeCode();
- flow.extra = builder.context.extra;
- builder.removeFlow();
- return flow;
- },
- build: function ( builder, output, uuid ) {
- output = output || this.getType( builder, output );
- var data = builder.getNodeData( uuid || this );
- if ( builder.analyzing ) {
- this.appendDepsNode( builder, data, output );
- }
- if ( builder.nodes.indexOf( this ) === - 1 ) {
- builder.nodes.push( this );
- }
- if ( this.updateFrame !== undefined && builder.updaters.indexOf( this ) === - 1 ) {
- builder.updaters.push( this );
- }
- return this.generate( builder, output, uuid );
- },
- generate: function ( /* builder, output, uuid, type, ns */ ) {
- // This method needs to be implemented in subclasses
- },
- getHash: function () {
- var hash = '{';
- var prop, obj;
- for ( prop in this ) {
- obj = this[ prop ];
- if ( obj instanceof Node ) {
- hash += '"' + prop + '":' + obj.getHash() + ',';
- }
- }
- if ( this.hashProperties ) {
- for ( var i = 0; i < this.hashProperties.length; i ++ ) {
- prop = this.hashProperties[ i ];
- obj = this[ prop ];
- hash += '"' + prop + '":"' + String( obj ) + '",';
- }
- }
- hash += '"id":"' + this.uuid + '"}';
- return hash;
- },
- appendDepsNode: function ( builder, data, output ) {
- data.deps = ( data.deps || 0 ) + 1;
- var outputLen = builder.getTypeLength( output );
- if ( outputLen > ( data.outputMax || 0 ) || this.getType( builder, output ) ) {
- data.outputMax = outputLen;
- data.output = output;
- }
- },
- setName: function ( name ) {
- this.name = name;
- return this;
- },
- getName: function ( /* builder */ ) {
- return this.name;
- },
- getType: function ( builder, output ) {
- return output === 'sampler2D' || output === 'samplerCube' ? output : this.type;
- },
- getJSONNode: function ( meta ) {
- var isRootObject = ( meta === undefined || typeof meta === 'string' );
- if ( ! isRootObject && meta.nodes[ this.uuid ] !== undefined ) {
- return meta.nodes[ this.uuid ];
- }
- },
- copy: function ( source ) {
- if ( source.name !== undefined ) this.name = source.name;
- if ( source.userData !== undefined ) this.userData = JSON.parse( JSON.stringify( source.userData ) );
- return this;
- },
- createJSONNode: function ( meta ) {
- var isRootObject = ( meta === undefined || typeof meta === 'string' );
- var data = {};
- if ( typeof this.nodeType !== 'string' ) throw new Error( 'Node does not allow serialization.' );
- data.uuid = this.uuid;
- data.nodeType = this.nodeType;
- if ( this.name !== '' ) data.name = this.name;
- if ( JSON.stringify( this.userData ) !== '{}' ) data.userData = this.userData;
- if ( ! isRootObject ) {
- meta.nodes[ this.uuid ] = data;
- }
- return data;
- },
- toJSON: function ( meta ) {
- return this.getJSONNode( meta ) || this.createJSONNode( meta );
- }
- };
- export { Node };
|