WebGLNodes.js 781 B

12345678910111213141516171819202122232425262728293031323334
  1. import { WebGLNodeBuilder } from './WebGLNodeBuilder.js';
  2. import NodeFrame from '../../nodes/core/NodeFrame.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. builders.set( this, new WebGLNodeBuilder( object, renderer, parameters ).build() );
  8. };
  9. Material.prototype.onBeforeRender = function ( renderer, scene, camera, geometry, object ) {
  10. const nodeBuilder = builders.get( this );
  11. if ( nodeBuilder !== undefined ) {
  12. nodeFrame.material = this;
  13. nodeFrame.camera = camera;
  14. nodeFrame.object = object;
  15. nodeFrame.renderer = renderer;
  16. for ( const node of nodeBuilder.updateNodes ) {
  17. nodeFrame.updateNode( node );
  18. }
  19. }
  20. };