WebGLPrograms.js 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682
  1. import { BackSide, DoubleSide, CubeUVReflectionMapping, ObjectSpaceNormalMap, TangentSpaceNormalMap, NoToneMapping, NormalBlending, SRGBTransfer, RGFormat, RG11_EAC_Format, RED_GREEN_RGTC2_Format } from '../../constants.js';
  2. import { Layers } from '../../core/Layers.js';
  3. import { WebGLProgram } from './WebGLProgram.js';
  4. import { WebGLShaderCache } from './WebGLShaderCache.js';
  5. import { ShaderLib } from '../shaders/ShaderLib.js';
  6. import { UniformsUtils } from '../shaders/UniformsUtils.js';
  7. import { ColorManagement } from '../../math/ColorManagement.js';
  8. import { warn } from '../../utils.js';
  9. function isPackedRGFormat( format ) {
  10. return format === RGFormat || format === RG11_EAC_Format || format === RED_GREEN_RGTC2_Format;
  11. }
  12. function WebGLPrograms( renderer, environments, extensions, capabilities, bindingStates, clipping ) {
  13. const _programLayers = new Layers();
  14. const _customShaders = new WebGLShaderCache();
  15. const _activeChannels = new Set();
  16. const programs = [];
  17. const programsMap = new Map();
  18. const logarithmicDepthBuffer = capabilities.logarithmicDepthBuffer;
  19. let precision = capabilities.precision;
  20. const shaderIDs = {
  21. MeshDepthMaterial: 'depth',
  22. MeshDistanceMaterial: 'distance',
  23. MeshNormalMaterial: 'normal',
  24. MeshBasicMaterial: 'basic',
  25. MeshLambertMaterial: 'lambert',
  26. MeshPhongMaterial: 'phong',
  27. MeshToonMaterial: 'toon',
  28. MeshStandardMaterial: 'physical',
  29. MeshPhysicalMaterial: 'physical',
  30. MeshMatcapMaterial: 'matcap',
  31. LineBasicMaterial: 'basic',
  32. LineDashedMaterial: 'dashed',
  33. PointsMaterial: 'points',
  34. ShadowMaterial: 'shadow',
  35. SpriteMaterial: 'sprite'
  36. };
  37. function getChannel( value ) {
  38. _activeChannels.add( value );
  39. if ( value === 0 ) return 'uv';
  40. return `uv${ value }`;
  41. }
  42. function getParameters( material, lights, shadows, scene, object, lightProbeGrids ) {
  43. const fog = scene.fog;
  44. const geometry = object.geometry;
  45. const environment = ( material.isMeshStandardMaterial || material.isMeshLambertMaterial || material.isMeshPhongMaterial ) ? scene.environment : null;
  46. const usePMREM = material.isMeshStandardMaterial || ( material.isMeshLambertMaterial && ! material.envMap ) || ( material.isMeshPhongMaterial && ! material.envMap );
  47. const envMap = environments.get( material.envMap || environment, usePMREM );
  48. const envMapCubeUVHeight = ( !! envMap ) && ( envMap.mapping === CubeUVReflectionMapping ) ? envMap.image.height : null;
  49. const shaderID = shaderIDs[ material.type ];
  50. // heuristics to create shader parameters according to lights in the scene
  51. // (not to blow over maxLights budget)
  52. if ( material.precision !== null ) {
  53. precision = capabilities.getMaxPrecision( material.precision );
  54. if ( precision !== material.precision ) {
  55. warn( 'WebGLProgram.getParameters:', material.precision, 'not supported, using', precision, 'instead.' );
  56. }
  57. }
  58. //
  59. const morphAttribute = geometry.morphAttributes.position || geometry.morphAttributes.normal || geometry.morphAttributes.color;
  60. const morphTargetsCount = ( morphAttribute !== undefined ) ? morphAttribute.length : 0;
  61. let morphTextureStride = 0;
  62. if ( geometry.morphAttributes.position !== undefined ) morphTextureStride = 1;
  63. if ( geometry.morphAttributes.normal !== undefined ) morphTextureStride = 2;
  64. if ( geometry.morphAttributes.color !== undefined ) morphTextureStride = 3;
  65. //
  66. let vertexShader, fragmentShader;
  67. let customVertexShaderID, customFragmentShaderID;
  68. if ( shaderID ) {
  69. const shader = ShaderLib[ shaderID ];
  70. vertexShader = shader.vertexShader;
  71. fragmentShader = shader.fragmentShader;
  72. } else {
  73. vertexShader = material.vertexShader;
  74. fragmentShader = material.fragmentShader;
  75. const vertexShaderStage = _customShaders.getVertexShaderStage( material );
  76. const fragmentShaderStage = _customShaders.getFragmentShaderStage( material );
  77. _customShaders.update( material, vertexShaderStage, fragmentShaderStage );
  78. customVertexShaderID = vertexShaderStage.id;
  79. customFragmentShaderID = fragmentShaderStage.id;
  80. }
  81. const currentRenderTarget = renderer.getRenderTarget();
  82. const reversedDepthBuffer = renderer.state.buffers.depth.getReversed();
  83. const IS_INSTANCEDMESH = object.isInstancedMesh === true;
  84. const IS_BATCHEDMESH = object.isBatchedMesh === true;
  85. const HAS_MAP = !! material.map;
  86. const HAS_MATCAP = !! material.matcap;
  87. const HAS_ENVMAP = !! envMap;
  88. const HAS_AOMAP = !! material.aoMap;
  89. const HAS_LIGHTMAP = !! material.lightMap;
  90. const HAS_BUMPMAP = !! material.bumpMap && material.wireframe === false;
  91. const HAS_NORMALMAP = !! material.normalMap;
  92. const HAS_DISPLACEMENTMAP = !! material.displacementMap;
  93. const HAS_EMISSIVEMAP = !! material.emissiveMap;
  94. const HAS_METALNESSMAP = !! material.metalnessMap;
  95. const HAS_ROUGHNESSMAP = !! material.roughnessMap;
  96. const HAS_ANISOTROPY = material.anisotropy > 0;
  97. const HAS_CLEARCOAT = material.clearcoat > 0;
  98. const HAS_DISPERSION = material.dispersion > 0;
  99. const HAS_IRIDESCENCE = material.iridescence > 0;
  100. const HAS_SHEEN = material.sheen > 0;
  101. const HAS_TRANSMISSION = material.transmission > 0;
  102. const HAS_ANISOTROPYMAP = HAS_ANISOTROPY && !! material.anisotropyMap;
  103. const HAS_CLEARCOATMAP = HAS_CLEARCOAT && !! material.clearcoatMap;
  104. const HAS_CLEARCOAT_NORMALMAP = HAS_CLEARCOAT && !! material.clearcoatNormalMap;
  105. const HAS_CLEARCOAT_ROUGHNESSMAP = HAS_CLEARCOAT && !! material.clearcoatRoughnessMap;
  106. const HAS_IRIDESCENCEMAP = HAS_IRIDESCENCE && !! material.iridescenceMap;
  107. const HAS_IRIDESCENCE_THICKNESSMAP = HAS_IRIDESCENCE && !! material.iridescenceThicknessMap;
  108. const HAS_SHEEN_COLORMAP = HAS_SHEEN && !! material.sheenColorMap;
  109. const HAS_SHEEN_ROUGHNESSMAP = HAS_SHEEN && !! material.sheenRoughnessMap;
  110. const HAS_SPECULARMAP = !! material.specularMap;
  111. const HAS_SPECULAR_COLORMAP = !! material.specularColorMap;
  112. const HAS_SPECULAR_INTENSITYMAP = !! material.specularIntensityMap;
  113. const HAS_TRANSMISSIONMAP = HAS_TRANSMISSION && !! material.transmissionMap;
  114. const HAS_THICKNESSMAP = HAS_TRANSMISSION && !! material.thicknessMap;
  115. const HAS_GRADIENTMAP = !! material.gradientMap;
  116. const HAS_ALPHAMAP = !! material.alphaMap;
  117. const HAS_ALPHATEST = material.alphaTest > 0;
  118. const HAS_ALPHAHASH = !! material.alphaHash;
  119. const HAS_EXTENSIONS = !! material.extensions;
  120. let toneMapping = NoToneMapping;
  121. if ( material.toneMapped ) {
  122. if ( currentRenderTarget === null || currentRenderTarget.isXRRenderTarget === true ) {
  123. toneMapping = renderer.toneMapping;
  124. }
  125. }
  126. const parameters = {
  127. shaderID: shaderID,
  128. shaderType: material.type,
  129. shaderName: material.name,
  130. vertexShader: vertexShader,
  131. fragmentShader: fragmentShader,
  132. defines: material.defines,
  133. customVertexShaderID: customVertexShaderID,
  134. customFragmentShaderID: customFragmentShaderID,
  135. isRawShaderMaterial: material.isRawShaderMaterial === true,
  136. glslVersion: material.glslVersion,
  137. precision: precision,
  138. batching: IS_BATCHEDMESH,
  139. batchingColor: IS_BATCHEDMESH && object._colorsTexture !== null,
  140. instancing: IS_INSTANCEDMESH,
  141. instancingColor: IS_INSTANCEDMESH && object.instanceColor !== null,
  142. instancingMorph: IS_INSTANCEDMESH && object.morphTexture !== null,
  143. outputColorSpace: ( currentRenderTarget === null ) ? renderer.outputColorSpace : ( currentRenderTarget.isXRRenderTarget === true ? currentRenderTarget.texture.colorSpace : ColorManagement.workingColorSpace ),
  144. alphaToCoverage: !! material.alphaToCoverage,
  145. map: HAS_MAP,
  146. matcap: HAS_MATCAP,
  147. envMap: HAS_ENVMAP,
  148. envMapMode: HAS_ENVMAP && envMap.mapping,
  149. envMapCubeUVHeight: envMapCubeUVHeight,
  150. aoMap: HAS_AOMAP,
  151. lightMap: HAS_LIGHTMAP,
  152. bumpMap: HAS_BUMPMAP,
  153. normalMap: HAS_NORMALMAP,
  154. displacementMap: HAS_DISPLACEMENTMAP,
  155. emissiveMap: HAS_EMISSIVEMAP,
  156. normalMapObjectSpace: HAS_NORMALMAP && material.normalMapType === ObjectSpaceNormalMap,
  157. normalMapTangentSpace: HAS_NORMALMAP && material.normalMapType === TangentSpaceNormalMap,
  158. packedNormalMap: HAS_NORMALMAP && material.normalMapType === TangentSpaceNormalMap && isPackedRGFormat( material.normalMap.format ),
  159. metalnessMap: HAS_METALNESSMAP,
  160. roughnessMap: HAS_ROUGHNESSMAP,
  161. anisotropy: HAS_ANISOTROPY,
  162. anisotropyMap: HAS_ANISOTROPYMAP,
  163. clearcoat: HAS_CLEARCOAT,
  164. clearcoatMap: HAS_CLEARCOATMAP,
  165. clearcoatNormalMap: HAS_CLEARCOAT_NORMALMAP,
  166. clearcoatRoughnessMap: HAS_CLEARCOAT_ROUGHNESSMAP,
  167. dispersion: HAS_DISPERSION,
  168. iridescence: HAS_IRIDESCENCE,
  169. iridescenceMap: HAS_IRIDESCENCEMAP,
  170. iridescenceThicknessMap: HAS_IRIDESCENCE_THICKNESSMAP,
  171. sheen: HAS_SHEEN,
  172. sheenColorMap: HAS_SHEEN_COLORMAP,
  173. sheenRoughnessMap: HAS_SHEEN_ROUGHNESSMAP,
  174. specularMap: HAS_SPECULARMAP,
  175. specularColorMap: HAS_SPECULAR_COLORMAP,
  176. specularIntensityMap: HAS_SPECULAR_INTENSITYMAP,
  177. transmission: HAS_TRANSMISSION,
  178. transmissionMap: HAS_TRANSMISSIONMAP,
  179. thicknessMap: HAS_THICKNESSMAP,
  180. gradientMap: HAS_GRADIENTMAP,
  181. opaque: material.transparent === false && material.blending === NormalBlending && material.alphaToCoverage === false,
  182. alphaMap: HAS_ALPHAMAP,
  183. alphaTest: HAS_ALPHATEST,
  184. alphaHash: HAS_ALPHAHASH,
  185. combine: material.combine,
  186. //
  187. mapUv: HAS_MAP && getChannel( material.map.channel ),
  188. aoMapUv: HAS_AOMAP && getChannel( material.aoMap.channel ),
  189. lightMapUv: HAS_LIGHTMAP && getChannel( material.lightMap.channel ),
  190. bumpMapUv: HAS_BUMPMAP && getChannel( material.bumpMap.channel ),
  191. normalMapUv: HAS_NORMALMAP && getChannel( material.normalMap.channel ),
  192. displacementMapUv: HAS_DISPLACEMENTMAP && getChannel( material.displacementMap.channel ),
  193. emissiveMapUv: HAS_EMISSIVEMAP && getChannel( material.emissiveMap.channel ),
  194. metalnessMapUv: HAS_METALNESSMAP && getChannel( material.metalnessMap.channel ),
  195. roughnessMapUv: HAS_ROUGHNESSMAP && getChannel( material.roughnessMap.channel ),
  196. anisotropyMapUv: HAS_ANISOTROPYMAP && getChannel( material.anisotropyMap.channel ),
  197. clearcoatMapUv: HAS_CLEARCOATMAP && getChannel( material.clearcoatMap.channel ),
  198. clearcoatNormalMapUv: HAS_CLEARCOAT_NORMALMAP && getChannel( material.clearcoatNormalMap.channel ),
  199. clearcoatRoughnessMapUv: HAS_CLEARCOAT_ROUGHNESSMAP && getChannel( material.clearcoatRoughnessMap.channel ),
  200. iridescenceMapUv: HAS_IRIDESCENCEMAP && getChannel( material.iridescenceMap.channel ),
  201. iridescenceThicknessMapUv: HAS_IRIDESCENCE_THICKNESSMAP && getChannel( material.iridescenceThicknessMap.channel ),
  202. sheenColorMapUv: HAS_SHEEN_COLORMAP && getChannel( material.sheenColorMap.channel ),
  203. sheenRoughnessMapUv: HAS_SHEEN_ROUGHNESSMAP && getChannel( material.sheenRoughnessMap.channel ),
  204. specularMapUv: HAS_SPECULARMAP && getChannel( material.specularMap.channel ),
  205. specularColorMapUv: HAS_SPECULAR_COLORMAP && getChannel( material.specularColorMap.channel ),
  206. specularIntensityMapUv: HAS_SPECULAR_INTENSITYMAP && getChannel( material.specularIntensityMap.channel ),
  207. transmissionMapUv: HAS_TRANSMISSIONMAP && getChannel( material.transmissionMap.channel ),
  208. thicknessMapUv: HAS_THICKNESSMAP && getChannel( material.thicknessMap.channel ),
  209. alphaMapUv: HAS_ALPHAMAP && getChannel( material.alphaMap.channel ),
  210. //
  211. vertexTangents: !! geometry.attributes.tangent && ( HAS_NORMALMAP || HAS_ANISOTROPY ),
  212. vertexNormals: !! geometry.attributes.normal,
  213. vertexColors: material.vertexColors,
  214. vertexAlphas: material.vertexColors === true && !! geometry.attributes.color && geometry.attributes.color.itemSize === 4,
  215. pointsUvs: object.isPoints === true && !! geometry.attributes.uv && ( HAS_MAP || HAS_ALPHAMAP ),
  216. fog: !! fog,
  217. useFog: material.fog === true,
  218. fogExp2: ( !! fog && fog.isFogExp2 ),
  219. flatShading: material.wireframe === false && (
  220. material.flatShading === true ||
  221. ( geometry.attributes.normal === undefined && HAS_NORMALMAP === false &&
  222. ( material.isMeshLambertMaterial || material.isMeshPhongMaterial || material.isMeshStandardMaterial || material.isMeshPhysicalMaterial )
  223. )
  224. ),
  225. sizeAttenuation: material.sizeAttenuation === true,
  226. logarithmicDepthBuffer: logarithmicDepthBuffer,
  227. reversedDepthBuffer: reversedDepthBuffer,
  228. skinning: object.isSkinnedMesh === true,
  229. morphTargets: geometry.morphAttributes.position !== undefined,
  230. morphNormals: geometry.morphAttributes.normal !== undefined,
  231. morphColors: geometry.morphAttributes.color !== undefined,
  232. morphTargetsCount: morphTargetsCount,
  233. morphTextureStride: morphTextureStride,
  234. numDirLights: lights.directional.length,
  235. numPointLights: lights.point.length,
  236. numSpotLights: lights.spot.length,
  237. numSpotLightMaps: lights.spotLightMap.length,
  238. numRectAreaLights: lights.rectArea.length,
  239. numHemiLights: lights.hemi.length,
  240. numDirLightShadows: lights.directionalShadowMap.length,
  241. numPointLightShadows: lights.pointShadowMap.length,
  242. numSpotLightShadows: lights.spotShadowMap.length,
  243. numSpotLightShadowsWithMaps: lights.numSpotLightShadowsWithMaps,
  244. numLightProbes: lights.numLightProbes,
  245. numLightProbeGrids: lightProbeGrids.length,
  246. numClippingPlanes: clipping.numPlanes,
  247. numClipIntersection: clipping.numIntersection,
  248. dithering: material.dithering,
  249. shadowMapEnabled: renderer.shadowMap.enabled && shadows.length > 0,
  250. shadowMapType: renderer.shadowMap.type,
  251. toneMapping: toneMapping,
  252. decodeVideoTexture: HAS_MAP && ( material.map.isVideoTexture === true ) && ( ColorManagement.getTransfer( material.map.colorSpace ) === SRGBTransfer ),
  253. decodeVideoTextureEmissive: HAS_EMISSIVEMAP && ( material.emissiveMap.isVideoTexture === true ) && ( ColorManagement.getTransfer( material.emissiveMap.colorSpace ) === SRGBTransfer ),
  254. premultipliedAlpha: material.premultipliedAlpha,
  255. doubleSided: material.side === DoubleSide,
  256. flipSided: material.side === BackSide,
  257. useDepthPacking: material.depthPacking >= 0,
  258. depthPacking: material.depthPacking || 0,
  259. index0AttributeName: material.index0AttributeName,
  260. extensionClipCullDistance: HAS_EXTENSIONS && material.extensions.clipCullDistance === true && extensions.has( 'WEBGL_clip_cull_distance' ),
  261. extensionMultiDraw: ( HAS_EXTENSIONS && material.extensions.multiDraw === true || IS_BATCHEDMESH ) && extensions.has( 'WEBGL_multi_draw' ),
  262. rendererExtensionParallelShaderCompile: extensions.has( 'KHR_parallel_shader_compile' ),
  263. customProgramCacheKey: material.customProgramCacheKey()
  264. };
  265. // the usage of getChannel() determines the active texture channels for this shader
  266. parameters.vertexUv1s = _activeChannels.has( 1 );
  267. parameters.vertexUv2s = _activeChannels.has( 2 );
  268. parameters.vertexUv3s = _activeChannels.has( 3 );
  269. _activeChannels.clear();
  270. return parameters;
  271. }
  272. function getProgramCacheKey( parameters ) {
  273. const array = [];
  274. if ( parameters.shaderID ) {
  275. array.push( parameters.shaderID );
  276. } else {
  277. array.push( parameters.customVertexShaderID );
  278. array.push( parameters.customFragmentShaderID );
  279. }
  280. if ( parameters.defines !== undefined ) {
  281. for ( const name in parameters.defines ) {
  282. array.push( name );
  283. array.push( parameters.defines[ name ] );
  284. }
  285. }
  286. if ( parameters.isRawShaderMaterial === false ) {
  287. getProgramCacheKeyParameters( array, parameters );
  288. getProgramCacheKeyBooleans( array, parameters );
  289. array.push( renderer.outputColorSpace );
  290. }
  291. array.push( parameters.customProgramCacheKey );
  292. return array.join();
  293. }
  294. function getProgramCacheKeyParameters( array, parameters ) {
  295. array.push( parameters.precision );
  296. array.push( parameters.outputColorSpace );
  297. array.push( parameters.envMapMode );
  298. array.push( parameters.envMapCubeUVHeight );
  299. array.push( parameters.mapUv );
  300. array.push( parameters.alphaMapUv );
  301. array.push( parameters.lightMapUv );
  302. array.push( parameters.aoMapUv );
  303. array.push( parameters.bumpMapUv );
  304. array.push( parameters.normalMapUv );
  305. array.push( parameters.displacementMapUv );
  306. array.push( parameters.emissiveMapUv );
  307. array.push( parameters.metalnessMapUv );
  308. array.push( parameters.roughnessMapUv );
  309. array.push( parameters.anisotropyMapUv );
  310. array.push( parameters.clearcoatMapUv );
  311. array.push( parameters.clearcoatNormalMapUv );
  312. array.push( parameters.clearcoatRoughnessMapUv );
  313. array.push( parameters.iridescenceMapUv );
  314. array.push( parameters.iridescenceThicknessMapUv );
  315. array.push( parameters.sheenColorMapUv );
  316. array.push( parameters.sheenRoughnessMapUv );
  317. array.push( parameters.specularMapUv );
  318. array.push( parameters.specularColorMapUv );
  319. array.push( parameters.specularIntensityMapUv );
  320. array.push( parameters.transmissionMapUv );
  321. array.push( parameters.thicknessMapUv );
  322. array.push( parameters.combine );
  323. array.push( parameters.fogExp2 );
  324. array.push( parameters.sizeAttenuation );
  325. array.push( parameters.morphTargetsCount );
  326. array.push( parameters.morphAttributeCount );
  327. array.push( parameters.numDirLights );
  328. array.push( parameters.numPointLights );
  329. array.push( parameters.numSpotLights );
  330. array.push( parameters.numSpotLightMaps );
  331. array.push( parameters.numHemiLights );
  332. array.push( parameters.numRectAreaLights );
  333. array.push( parameters.numDirLightShadows );
  334. array.push( parameters.numPointLightShadows );
  335. array.push( parameters.numSpotLightShadows );
  336. array.push( parameters.numSpotLightShadowsWithMaps );
  337. array.push( parameters.numLightProbes );
  338. array.push( parameters.shadowMapType );
  339. array.push( parameters.toneMapping );
  340. array.push( parameters.numClippingPlanes );
  341. array.push( parameters.numClipIntersection );
  342. array.push( parameters.depthPacking );
  343. }
  344. function getProgramCacheKeyBooleans( array, parameters ) {
  345. _programLayers.disableAll();
  346. if ( parameters.instancing )
  347. _programLayers.enable( 0 );
  348. if ( parameters.instancingColor )
  349. _programLayers.enable( 1 );
  350. if ( parameters.instancingMorph )
  351. _programLayers.enable( 2 );
  352. if ( parameters.matcap )
  353. _programLayers.enable( 3 );
  354. if ( parameters.envMap )
  355. _programLayers.enable( 4 );
  356. if ( parameters.normalMapObjectSpace )
  357. _programLayers.enable( 5 );
  358. if ( parameters.normalMapTangentSpace )
  359. _programLayers.enable( 6 );
  360. if ( parameters.clearcoat )
  361. _programLayers.enable( 7 );
  362. if ( parameters.iridescence )
  363. _programLayers.enable( 8 );
  364. if ( parameters.alphaTest )
  365. _programLayers.enable( 9 );
  366. if ( parameters.vertexColors )
  367. _programLayers.enable( 10 );
  368. if ( parameters.vertexAlphas )
  369. _programLayers.enable( 11 );
  370. if ( parameters.vertexUv1s )
  371. _programLayers.enable( 12 );
  372. if ( parameters.vertexUv2s )
  373. _programLayers.enable( 13 );
  374. if ( parameters.vertexUv3s )
  375. _programLayers.enable( 14 );
  376. if ( parameters.vertexTangents )
  377. _programLayers.enable( 15 );
  378. if ( parameters.anisotropy )
  379. _programLayers.enable( 16 );
  380. if ( parameters.alphaHash )
  381. _programLayers.enable( 17 );
  382. if ( parameters.batching )
  383. _programLayers.enable( 18 );
  384. if ( parameters.dispersion )
  385. _programLayers.enable( 19 );
  386. if ( parameters.batchingColor )
  387. _programLayers.enable( 20 );
  388. if ( parameters.gradientMap )
  389. _programLayers.enable( 21 );
  390. if ( parameters.packedNormalMap )
  391. _programLayers.enable( 22 );
  392. if ( parameters.vertexNormals )
  393. _programLayers.enable( 23 );
  394. array.push( _programLayers.mask );
  395. _programLayers.disableAll();
  396. if ( parameters.fog )
  397. _programLayers.enable( 0 );
  398. if ( parameters.useFog )
  399. _programLayers.enable( 1 );
  400. if ( parameters.flatShading )
  401. _programLayers.enable( 2 );
  402. if ( parameters.logarithmicDepthBuffer )
  403. _programLayers.enable( 3 );
  404. if ( parameters.reversedDepthBuffer )
  405. _programLayers.enable( 4 );
  406. if ( parameters.skinning )
  407. _programLayers.enable( 5 );
  408. if ( parameters.morphTargets )
  409. _programLayers.enable( 6 );
  410. if ( parameters.morphNormals )
  411. _programLayers.enable( 7 );
  412. if ( parameters.morphColors )
  413. _programLayers.enable( 8 );
  414. if ( parameters.premultipliedAlpha )
  415. _programLayers.enable( 9 );
  416. if ( parameters.shadowMapEnabled )
  417. _programLayers.enable( 10 );
  418. if ( parameters.doubleSided )
  419. _programLayers.enable( 11 );
  420. if ( parameters.flipSided )
  421. _programLayers.enable( 12 );
  422. if ( parameters.useDepthPacking )
  423. _programLayers.enable( 13 );
  424. if ( parameters.dithering )
  425. _programLayers.enable( 14 );
  426. if ( parameters.transmission )
  427. _programLayers.enable( 15 );
  428. if ( parameters.sheen )
  429. _programLayers.enable( 16 );
  430. if ( parameters.opaque )
  431. _programLayers.enable( 17 );
  432. if ( parameters.pointsUvs )
  433. _programLayers.enable( 18 );
  434. if ( parameters.decodeVideoTexture )
  435. _programLayers.enable( 19 );
  436. if ( parameters.decodeVideoTextureEmissive )
  437. _programLayers.enable( 20 );
  438. if ( parameters.alphaToCoverage )
  439. _programLayers.enable( 21 );
  440. if ( parameters.numLightProbeGrids > 0 )
  441. _programLayers.enable( 22 );
  442. array.push( _programLayers.mask );
  443. }
  444. function getUniforms( material ) {
  445. const shaderID = shaderIDs[ material.type ];
  446. let uniforms;
  447. if ( shaderID ) {
  448. const shader = ShaderLib[ shaderID ];
  449. uniforms = UniformsUtils.clone( shader.uniforms );
  450. } else {
  451. uniforms = material.uniforms;
  452. }
  453. return uniforms;
  454. }
  455. function acquireProgram( parameters, cacheKey ) {
  456. let program = programsMap.get( cacheKey );
  457. if ( program !== undefined ) {
  458. ++ program.usedTimes;
  459. } else {
  460. program = new WebGLProgram( renderer, cacheKey, parameters, bindingStates );
  461. programs.push( program );
  462. programsMap.set( cacheKey, program );
  463. }
  464. return program;
  465. }
  466. function releaseProgram( program ) {
  467. if ( -- program.usedTimes === 0 ) {
  468. // Remove from unordered set
  469. const i = programs.indexOf( program );
  470. programs[ i ] = programs[ programs.length - 1 ];
  471. programs.pop();
  472. // Remove from map
  473. programsMap.delete( program.cacheKey );
  474. // Free WebGL resources
  475. program.destroy();
  476. }
  477. }
  478. function releaseShaderCache( material ) {
  479. _customShaders.remove( material );
  480. }
  481. function dispose() {
  482. _customShaders.dispose();
  483. }
  484. return {
  485. getParameters: getParameters,
  486. getProgramCacheKey: getProgramCacheKey,
  487. getUniforms: getUniforms,
  488. acquireProgram: acquireProgram,
  489. releaseProgram: releaseProgram,
  490. releaseShaderCache: releaseShaderCache,
  491. // Exposed for resource monitoring & error feedback via renderer.info:
  492. programs: programs,
  493. dispose: dispose
  494. };
  495. }
  496. export { WebGLPrograms };
粤ICP备19079148号