WebGLNodes.js 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. import { WebGLNodeBuilder } from './WebGLNodeBuilder.js';
  2. import { NodeFrame } from '../../../nodes/Nodes.js';
  3. import { Material } from 'three';
  4. const builders = new WeakMap();
  5. export const nodeFrame = new NodeFrame();
  6. Material.prototype.onBuild = function ( object, parameters, renderer ) {
  7. const material = this;
  8. if ( material.isNodeMaterial === true ) {
  9. builders.set( material, new WebGLNodeBuilder( object, renderer, parameters, material ).build() );
  10. }
  11. };
  12. Material.prototype.onBeforeRender = function ( renderer, scene, camera, geometry, object ) {
  13. const nodeBuilder = builders.get( this );
  14. if ( nodeBuilder !== undefined ) {
  15. nodeFrame.material = this;
  16. nodeFrame.camera = camera;
  17. nodeFrame.object = object;
  18. nodeFrame.renderer = renderer;
  19. const updateNodes = nodeBuilder.updateNodes;
  20. if ( updateNodes.length > 0 ) {
  21. // force refresh material uniforms
  22. renderer.state.useProgram( null );
  23. //this.uniformsNeedUpdate = true;
  24. for ( const node of updateNodes ) {
  25. nodeFrame.updateNode( node );
  26. }
  27. }
  28. }
  29. };