RenderObjects.js 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. import ChainMap from './ChainMap.js';
  2. import RenderObject from './RenderObject.js';
  3. class RenderObjects {
  4. constructor( renderer, nodes, geometries, pipelines, bindings, info ) {
  5. this.renderer = renderer;
  6. this.nodes = nodes;
  7. this.geometries = geometries;
  8. this.pipelines = pipelines;
  9. this.bindings = bindings;
  10. this.info = info;
  11. this.chainMaps = {};
  12. }
  13. get( object, material, scene, camera, lightsNode, renderContext, passId ) {
  14. const chainMap = this.getChainMap( passId );
  15. const chainArray = [ object, material, renderContext, lightsNode ];
  16. let renderObject = chainMap.get( chainArray );
  17. if ( renderObject === undefined ) {
  18. renderObject = this.createRenderObject( this.nodes, this.geometries, this.renderer, object, material, scene, camera, lightsNode, renderContext, passId );
  19. chainMap.set( chainArray, renderObject );
  20. } else {
  21. if ( renderObject.version !== material.version || renderObject.needsUpdate ) {
  22. if ( renderObject.initialCacheKey !== renderObject.getCacheKey() ) {
  23. renderObject.dispose();
  24. renderObject = this.get( object, material, scene, camera, lightsNode, renderContext, passId );
  25. } else {
  26. renderObject.version = material.version;
  27. }
  28. }
  29. }
  30. return renderObject;
  31. }
  32. getChainMap( passId = 'default' ) {
  33. return this.chainMaps[ passId ] || ( this.chainMaps[ passId ] = new ChainMap() );
  34. }
  35. dispose() {
  36. this.chainMaps = {};
  37. }
  38. createRenderObject( nodes, geometries, renderer, object, material, scene, camera, lightsNode, renderContext, passId ) {
  39. const chainMap = this.getChainMap( passId );
  40. const renderObject = new RenderObject( nodes, geometries, renderer, object, material, scene, camera, lightsNode, renderContext );
  41. renderObject.onDispose = () => {
  42. this.pipelines.delete( renderObject );
  43. this.bindings.delete( renderObject );
  44. this.nodes.delete( renderObject );
  45. chainMap.delete( renderObject.getChainArray() );
  46. };
  47. return renderObject;
  48. }
  49. }
  50. export default RenderObjects;