WebGLRenderLists.js 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218
  1. function painterSortStable( a, b ) {
  2. if ( a.groupOrder !== b.groupOrder ) {
  3. return a.groupOrder - b.groupOrder;
  4. } else if ( a.renderOrder !== b.renderOrder ) {
  5. return a.renderOrder - b.renderOrder;
  6. } else if ( a.program !== b.program ) {
  7. return a.program.id - b.program.id;
  8. } else if ( a.material.id !== b.material.id ) {
  9. return a.material.id - b.material.id;
  10. } else if ( a.z !== b.z ) {
  11. return a.z - b.z;
  12. } else {
  13. return a.id - b.id;
  14. }
  15. }
  16. function reversePainterSortStable( a, b ) {
  17. if ( a.groupOrder !== b.groupOrder ) {
  18. return a.groupOrder - b.groupOrder;
  19. } else if ( a.renderOrder !== b.renderOrder ) {
  20. return a.renderOrder - b.renderOrder;
  21. } else if ( a.z !== b.z ) {
  22. return b.z - a.z;
  23. } else {
  24. return a.id - b.id;
  25. }
  26. }
  27. function WebGLRenderList( properties ) {
  28. const renderItems = [];
  29. let renderItemsIndex = 0;
  30. const opaque = [];
  31. const transparent = [];
  32. const defaultProgram = { id: - 1 };
  33. function init() {
  34. renderItemsIndex = 0;
  35. opaque.length = 0;
  36. transparent.length = 0;
  37. }
  38. function getNextRenderItem( object, geometry, material, groupOrder, z, group ) {
  39. let renderItem = renderItems[ renderItemsIndex ];
  40. const materialProperties = properties.get( material );
  41. if ( renderItem === undefined ) {
  42. renderItem = {
  43. id: object.id,
  44. object: object,
  45. geometry: geometry,
  46. material: material,
  47. program: materialProperties.program || defaultProgram,
  48. groupOrder: groupOrder,
  49. renderOrder: object.renderOrder,
  50. z: z,
  51. group: group
  52. };
  53. renderItems[ renderItemsIndex ] = renderItem;
  54. } else {
  55. renderItem.id = object.id;
  56. renderItem.object = object;
  57. renderItem.geometry = geometry;
  58. renderItem.material = material;
  59. renderItem.program = materialProperties.program || defaultProgram;
  60. renderItem.groupOrder = groupOrder;
  61. renderItem.renderOrder = object.renderOrder;
  62. renderItem.z = z;
  63. renderItem.group = group;
  64. }
  65. renderItemsIndex ++;
  66. return renderItem;
  67. }
  68. function push( object, geometry, material, groupOrder, z, group ) {
  69. const renderItem = getNextRenderItem( object, geometry, material, groupOrder, z, group );
  70. ( material.transparent === true ? transparent : opaque ).push( renderItem );
  71. }
  72. function unshift( object, geometry, material, groupOrder, z, group ) {
  73. const renderItem = getNextRenderItem( object, geometry, material, groupOrder, z, group );
  74. ( material.transparent === true ? transparent : opaque ).unshift( renderItem );
  75. }
  76. function sort( customOpaqueSort, customTransparentSort ) {
  77. if ( opaque.length > 1 ) opaque.sort( customOpaqueSort || painterSortStable );
  78. if ( transparent.length > 1 ) transparent.sort( customTransparentSort || reversePainterSortStable );
  79. }
  80. function finish() {
  81. // Clear references from inactive renderItems in the list
  82. for ( let i = renderItemsIndex, il = renderItems.length; i < il; i ++ ) {
  83. const renderItem = renderItems[ i ];
  84. if ( renderItem.id === null ) break;
  85. renderItem.id = null;
  86. renderItem.object = null;
  87. renderItem.geometry = null;
  88. renderItem.material = null;
  89. renderItem.program = null;
  90. renderItem.group = null;
  91. }
  92. }
  93. return {
  94. opaque: opaque,
  95. transparent: transparent,
  96. init: init,
  97. push: push,
  98. unshift: unshift,
  99. finish: finish,
  100. sort: sort
  101. };
  102. }
  103. function WebGLRenderLists( properties ) {
  104. let lists = new WeakMap();
  105. function get( scene, camera ) {
  106. const cameras = lists.get( scene );
  107. let list;
  108. if ( cameras === undefined ) {
  109. list = new WebGLRenderList( properties );
  110. lists.set( scene, new WeakMap() );
  111. lists.get( scene ).set( camera, list );
  112. } else {
  113. list = cameras.get( camera );
  114. if ( list === undefined ) {
  115. list = new WebGLRenderList( properties );
  116. cameras.set( camera, list );
  117. }
  118. }
  119. return list;
  120. }
  121. function dispose() {
  122. lists = new WeakMap();
  123. }
  124. return {
  125. get: get,
  126. dispose: dispose
  127. };
  128. }
  129. export { WebGLRenderLists, WebGLRenderList };