WebGLPrograms.js 21 KB

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