WebGLProgram.js 30 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033
  1. import { WebGLUniforms } from './WebGLUniforms.js';
  2. import { WebGLShader } from './WebGLShader.js';
  3. import { ShaderChunk } from '../shaders/ShaderChunk.js';
  4. import { NoToneMapping, AddOperation, MixOperation, MultiplyOperation, CubeRefractionMapping, CubeUVReflectionMapping, CubeReflectionMapping, PCFShadowMap, VSMShadowMap, AgXToneMapping, ACESFilmicToneMapping, NeutralToneMapping, CineonToneMapping, CustomToneMapping, ReinhardToneMapping, LinearToneMapping, GLSL3, LinearTransfer, SRGBTransfer } from '../../constants.js';
  5. import { ColorManagement } from '../../math/ColorManagement.js';
  6. import { Vector3 } from '../../math/Vector3.js';
  7. import { Matrix3 } from '../../math/Matrix3.js';
  8. import { warn, error } from '../../utils.js';
  9. // From https://www.khronos.org/registry/webgl/extensions/KHR_parallel_shader_compile/
  10. const COMPLETION_STATUS_KHR = 0x91B1;
  11. let programIdCount = 0;
  12. function handleSource( string, errorLine ) {
  13. const lines = string.split( '\n' );
  14. const lines2 = [];
  15. const from = Math.max( errorLine - 6, 0 );
  16. const to = Math.min( errorLine + 6, lines.length );
  17. for ( let i = from; i < to; i ++ ) {
  18. const line = i + 1;
  19. lines2.push( `${line === errorLine ? '>' : ' '} ${line}: ${lines[ i ]}` );
  20. }
  21. return lines2.join( '\n' );
  22. }
  23. const _m0 = /*@__PURE__*/ new Matrix3();
  24. function getEncodingComponents( colorSpace ) {
  25. ColorManagement._getMatrix( _m0, ColorManagement.workingColorSpace, colorSpace );
  26. const encodingMatrix = `mat3( ${ _m0.elements.map( ( v ) => v.toFixed( 4 ) ) } )`;
  27. switch ( ColorManagement.getTransfer( colorSpace ) ) {
  28. case LinearTransfer:
  29. return [ encodingMatrix, 'LinearTransferOETF' ];
  30. case SRGBTransfer:
  31. return [ encodingMatrix, 'sRGBTransferOETF' ];
  32. default:
  33. warn( 'WebGLProgram: Unsupported color space: ', colorSpace );
  34. return [ encodingMatrix, 'LinearTransferOETF' ];
  35. }
  36. }
  37. function getShaderErrors( gl, shader, type ) {
  38. const status = gl.getShaderParameter( shader, gl.COMPILE_STATUS );
  39. const shaderInfoLog = gl.getShaderInfoLog( shader ) || '';
  40. const errors = shaderInfoLog.trim();
  41. if ( status && errors === '' ) return '';
  42. const errorMatches = /ERROR: 0:(\d+)/.exec( errors );
  43. if ( errorMatches ) {
  44. // --enable-privileged-webgl-extension
  45. // log( '**' + type + '**', gl.getExtension( 'WEBGL_debug_shaders' ).getTranslatedShaderSource( shader ) );
  46. const errorLine = parseInt( errorMatches[ 1 ] );
  47. return type.toUpperCase() + '\n\n' + errors + '\n\n' + handleSource( gl.getShaderSource( shader ), errorLine );
  48. } else {
  49. return errors;
  50. }
  51. }
  52. function getTexelEncodingFunction( functionName, colorSpace ) {
  53. const components = getEncodingComponents( colorSpace );
  54. return [
  55. `vec4 ${functionName}( vec4 value ) {`,
  56. ` return ${components[ 1 ]}( vec4( value.rgb * ${components[ 0 ]}, value.a ) );`,
  57. '}',
  58. ].join( '\n' );
  59. }
  60. const toneMappingFunctions = {
  61. [ LinearToneMapping ]: 'Linear',
  62. [ ReinhardToneMapping ]: 'Reinhard',
  63. [ CineonToneMapping ]: 'Cineon',
  64. [ ACESFilmicToneMapping ]: 'ACESFilmic',
  65. [ AgXToneMapping ]: 'AgX',
  66. [ NeutralToneMapping ]: 'Neutral',
  67. [ CustomToneMapping ]: 'Custom'
  68. };
  69. function getToneMappingFunction( functionName, toneMapping ) {
  70. const toneMappingName = toneMappingFunctions[ toneMapping ];
  71. if ( toneMappingName === undefined ) {
  72. warn( 'WebGLProgram: Unsupported toneMapping:', toneMapping );
  73. return 'vec3 ' + functionName + '( vec3 color ) { return LinearToneMapping( color ); }';
  74. }
  75. return 'vec3 ' + functionName + '( vec3 color ) { return ' + toneMappingName + 'ToneMapping( color ); }';
  76. }
  77. const _v0 = /*@__PURE__*/ new Vector3();
  78. function getLuminanceFunction() {
  79. ColorManagement.getLuminanceCoefficients( _v0 );
  80. const r = _v0.x.toFixed( 4 );
  81. const g = _v0.y.toFixed( 4 );
  82. const b = _v0.z.toFixed( 4 );
  83. return [
  84. 'float luminance( const in vec3 rgb ) {',
  85. ` const vec3 weights = vec3( ${ r }, ${ g }, ${ b } );`,
  86. ' return dot( weights, rgb );',
  87. '}'
  88. ].join( '\n' );
  89. }
  90. function generateVertexExtensions( parameters ) {
  91. const chunks = [
  92. parameters.extensionClipCullDistance ? '#extension GL_ANGLE_clip_cull_distance : require' : '',
  93. parameters.extensionMultiDraw ? '#extension GL_ANGLE_multi_draw : require' : '',
  94. ];
  95. return chunks.filter( filterEmptyLine ).join( '\n' );
  96. }
  97. function generateDefines( defines ) {
  98. const chunks = [];
  99. for ( const name in defines ) {
  100. const value = defines[ name ];
  101. if ( value === false ) continue;
  102. chunks.push( '#define ' + name + ' ' + value );
  103. }
  104. return chunks.join( '\n' );
  105. }
  106. function fetchAttributeLocations( gl, program ) {
  107. const attributes = {};
  108. const n = gl.getProgramParameter( program, gl.ACTIVE_ATTRIBUTES );
  109. for ( let i = 0; i < n; i ++ ) {
  110. const info = gl.getActiveAttrib( program, i );
  111. const name = info.name;
  112. let locationSize = 1;
  113. if ( info.type === gl.FLOAT_MAT2 ) locationSize = 2;
  114. if ( info.type === gl.FLOAT_MAT3 ) locationSize = 3;
  115. if ( info.type === gl.FLOAT_MAT4 ) locationSize = 4;
  116. // log( 'WebGLProgram: ACTIVE VERTEX ATTRIBUTE:', name, i );
  117. attributes[ name ] = {
  118. type: info.type,
  119. location: gl.getAttribLocation( program, name ),
  120. locationSize: locationSize
  121. };
  122. }
  123. return attributes;
  124. }
  125. function filterEmptyLine( string ) {
  126. return string !== '';
  127. }
  128. function replaceLightNums( string, parameters ) {
  129. const numSpotLightCoords = parameters.numSpotLightShadows + parameters.numSpotLightMaps - parameters.numSpotLightShadowsWithMaps;
  130. return string
  131. .replace( /NUM_DIR_LIGHTS/g, parameters.numDirLights )
  132. .replace( /NUM_SPOT_LIGHTS/g, parameters.numSpotLights )
  133. .replace( /NUM_SPOT_LIGHT_MAPS/g, parameters.numSpotLightMaps )
  134. .replace( /NUM_SPOT_LIGHT_COORDS/g, numSpotLightCoords )
  135. .replace( /NUM_RECT_AREA_LIGHTS/g, parameters.numRectAreaLights )
  136. .replace( /NUM_POINT_LIGHTS/g, parameters.numPointLights )
  137. .replace( /NUM_HEMI_LIGHTS/g, parameters.numHemiLights )
  138. .replace( /NUM_DIR_LIGHT_SHADOWS/g, parameters.numDirLightShadows )
  139. .replace( /NUM_SPOT_LIGHT_SHADOWS_WITH_MAPS/g, parameters.numSpotLightShadowsWithMaps )
  140. .replace( /NUM_SPOT_LIGHT_SHADOWS/g, parameters.numSpotLightShadows )
  141. .replace( /NUM_POINT_LIGHT_SHADOWS/g, parameters.numPointLightShadows );
  142. }
  143. function replaceClippingPlaneNums( string, parameters ) {
  144. return string
  145. .replace( /NUM_CLIPPING_PLANES/g, parameters.numClippingPlanes )
  146. .replace( /UNION_CLIPPING_PLANES/g, ( parameters.numClippingPlanes - parameters.numClipIntersection ) );
  147. }
  148. // Resolve Includes
  149. const includePattern = /^[ \t]*#include +<([\w\d./]+)>/gm;
  150. function resolveIncludes( string ) {
  151. return string.replace( includePattern, includeReplacer );
  152. }
  153. const shaderChunkMap = new Map();
  154. function includeReplacer( match, include ) {
  155. let string = ShaderChunk[ include ];
  156. if ( string === undefined ) {
  157. const newInclude = shaderChunkMap.get( include );
  158. if ( newInclude !== undefined ) {
  159. string = ShaderChunk[ newInclude ];
  160. warn( 'WebGLRenderer: Shader chunk "%s" has been deprecated. Use "%s" instead.', include, newInclude );
  161. } else {
  162. throw new Error( 'THREE.WebGLProgram: Can not resolve #include <' + include + '>' );
  163. }
  164. }
  165. return resolveIncludes( string );
  166. }
  167. // Unroll Loops
  168. const unrollLoopPattern = /#pragma unroll_loop_start\s+for\s*\(\s*int\s+i\s*=\s*(\d+)\s*;\s*i\s*<\s*(\d+)\s*;\s*i\s*\+\+\s*\)\s*{([\s\S]+?)}\s+#pragma unroll_loop_end/g;
  169. function unrollLoops( string ) {
  170. return string.replace( unrollLoopPattern, loopReplacer );
  171. }
  172. function loopReplacer( match, start, end, snippet ) {
  173. let string = '';
  174. for ( let i = parseInt( start ); i < parseInt( end ); i ++ ) {
  175. string += snippet
  176. .replace( /\[\s*i\s*\]/g, '[ ' + i + ' ]' )
  177. .replace( /UNROLLED_LOOP_INDEX/g, i );
  178. }
  179. return string;
  180. }
  181. //
  182. function generatePrecision( parameters ) {
  183. let precisionstring = `precision ${parameters.precision} float;
  184. precision ${parameters.precision} int;
  185. precision ${parameters.precision} sampler2D;
  186. precision ${parameters.precision} samplerCube;
  187. precision ${parameters.precision} sampler3D;
  188. precision ${parameters.precision} sampler2DArray;
  189. precision ${parameters.precision} sampler2DShadow;
  190. precision ${parameters.precision} samplerCubeShadow;
  191. precision ${parameters.precision} sampler2DArrayShadow;
  192. precision ${parameters.precision} isampler2D;
  193. precision ${parameters.precision} isampler3D;
  194. precision ${parameters.precision} isamplerCube;
  195. precision ${parameters.precision} isampler2DArray;
  196. precision ${parameters.precision} usampler2D;
  197. precision ${parameters.precision} usampler3D;
  198. precision ${parameters.precision} usamplerCube;
  199. precision ${parameters.precision} usampler2DArray;
  200. `;
  201. if ( parameters.precision === 'highp' ) {
  202. precisionstring += '\n#define HIGH_PRECISION';
  203. } else if ( parameters.precision === 'mediump' ) {
  204. precisionstring += '\n#define MEDIUM_PRECISION';
  205. } else if ( parameters.precision === 'lowp' ) {
  206. precisionstring += '\n#define LOW_PRECISION';
  207. }
  208. return precisionstring;
  209. }
  210. const shadowMapTypeDefines = {
  211. [ PCFShadowMap ]: 'SHADOWMAP_TYPE_PCF',
  212. [ VSMShadowMap ]: 'SHADOWMAP_TYPE_VSM'
  213. };
  214. function generateShadowMapTypeDefine( parameters ) {
  215. return shadowMapTypeDefines[ parameters.shadowMapType ] || 'SHADOWMAP_TYPE_BASIC';
  216. }
  217. const envMapTypeDefines = {
  218. [ CubeReflectionMapping ]: 'ENVMAP_TYPE_CUBE',
  219. [ CubeRefractionMapping ]: 'ENVMAP_TYPE_CUBE',
  220. [ CubeUVReflectionMapping ]: 'ENVMAP_TYPE_CUBE_UV'
  221. };
  222. function generateEnvMapTypeDefine( parameters ) {
  223. if ( parameters.envMap === false ) return 'ENVMAP_TYPE_CUBE';
  224. return envMapTypeDefines[ parameters.envMapMode ] || 'ENVMAP_TYPE_CUBE';
  225. }
  226. const envMapModeDefines = {
  227. [ CubeRefractionMapping ]: 'ENVMAP_MODE_REFRACTION'
  228. };
  229. function generateEnvMapModeDefine( parameters ) {
  230. if ( parameters.envMap === false ) return 'ENVMAP_MODE_REFLECTION';
  231. return envMapModeDefines[ parameters.envMapMode ] || 'ENVMAP_MODE_REFLECTION';
  232. }
  233. const envMapBlendingDefines = {
  234. [ MultiplyOperation ]: 'ENVMAP_BLENDING_MULTIPLY',
  235. [ MixOperation ]: 'ENVMAP_BLENDING_MIX',
  236. [ AddOperation ]: 'ENVMAP_BLENDING_ADD'
  237. };
  238. function generateEnvMapBlendingDefine( parameters ) {
  239. if ( parameters.envMap === false ) return 'ENVMAP_BLENDING_NONE';
  240. return envMapBlendingDefines[ parameters.combine ] || 'ENVMAP_BLENDING_NONE';
  241. }
  242. function generateCubeUVSize( parameters ) {
  243. const imageHeight = parameters.envMapCubeUVHeight;
  244. if ( imageHeight === null ) return null;
  245. const maxMip = Math.log2( imageHeight ) - 2;
  246. const texelHeight = 1.0 / imageHeight;
  247. const texelWidth = 1.0 / ( 3 * Math.max( Math.pow( 2, maxMip ), 7 * 16 ) );
  248. return { texelWidth, texelHeight, maxMip };
  249. }
  250. function WebGLProgram( renderer, cacheKey, parameters, bindingStates ) {
  251. // TODO Send this event to Three.js DevTools
  252. // log( 'WebGLProgram', cacheKey );
  253. const gl = renderer.getContext();
  254. const defines = parameters.defines;
  255. let vertexShader = parameters.vertexShader;
  256. let fragmentShader = parameters.fragmentShader;
  257. const shadowMapTypeDefine = generateShadowMapTypeDefine( parameters );
  258. const envMapTypeDefine = generateEnvMapTypeDefine( parameters );
  259. const envMapModeDefine = generateEnvMapModeDefine( parameters );
  260. const envMapBlendingDefine = generateEnvMapBlendingDefine( parameters );
  261. const envMapCubeUVSize = generateCubeUVSize( parameters );
  262. const customVertexExtensions = generateVertexExtensions( parameters );
  263. const customDefines = generateDefines( defines );
  264. const program = gl.createProgram();
  265. let prefixVertex, prefixFragment;
  266. let versionString = parameters.glslVersion ? '#version ' + parameters.glslVersion + '\n' : '';
  267. if ( parameters.isRawShaderMaterial ) {
  268. prefixVertex = [
  269. '#define SHADER_TYPE ' + parameters.shaderType,
  270. '#define SHADER_NAME ' + parameters.shaderName,
  271. customDefines
  272. ].filter( filterEmptyLine ).join( '\n' );
  273. if ( prefixVertex.length > 0 ) {
  274. prefixVertex += '\n';
  275. }
  276. prefixFragment = [
  277. '#define SHADER_TYPE ' + parameters.shaderType,
  278. '#define SHADER_NAME ' + parameters.shaderName,
  279. customDefines
  280. ].filter( filterEmptyLine ).join( '\n' );
  281. if ( prefixFragment.length > 0 ) {
  282. prefixFragment += '\n';
  283. }
  284. } else {
  285. prefixVertex = [
  286. generatePrecision( parameters ),
  287. '#define SHADER_TYPE ' + parameters.shaderType,
  288. '#define SHADER_NAME ' + parameters.shaderName,
  289. customDefines,
  290. parameters.extensionClipCullDistance ? '#define USE_CLIP_DISTANCE' : '',
  291. parameters.batching ? '#define USE_BATCHING' : '',
  292. parameters.batchingColor ? '#define USE_BATCHING_COLOR' : '',
  293. parameters.instancing ? '#define USE_INSTANCING' : '',
  294. parameters.instancingColor ? '#define USE_INSTANCING_COLOR' : '',
  295. parameters.instancingMorph ? '#define USE_INSTANCING_MORPH' : '',
  296. parameters.useFog && parameters.fog ? '#define USE_FOG' : '',
  297. parameters.useFog && parameters.fogExp2 ? '#define FOG_EXP2' : '',
  298. parameters.map ? '#define USE_MAP' : '',
  299. parameters.envMap ? '#define USE_ENVMAP' : '',
  300. parameters.envMap ? '#define ' + envMapModeDefine : '',
  301. parameters.lightMap ? '#define USE_LIGHTMAP' : '',
  302. parameters.aoMap ? '#define USE_AOMAP' : '',
  303. parameters.bumpMap ? '#define USE_BUMPMAP' : '',
  304. parameters.normalMap ? '#define USE_NORMALMAP' : '',
  305. parameters.normalMapObjectSpace ? '#define USE_NORMALMAP_OBJECTSPACE' : '',
  306. parameters.normalMapTangentSpace ? '#define USE_NORMALMAP_TANGENTSPACE' : '',
  307. parameters.displacementMap ? '#define USE_DISPLACEMENTMAP' : '',
  308. parameters.emissiveMap ? '#define USE_EMISSIVEMAP' : '',
  309. parameters.anisotropy ? '#define USE_ANISOTROPY' : '',
  310. parameters.anisotropyMap ? '#define USE_ANISOTROPYMAP' : '',
  311. parameters.clearcoatMap ? '#define USE_CLEARCOATMAP' : '',
  312. parameters.clearcoatRoughnessMap ? '#define USE_CLEARCOAT_ROUGHNESSMAP' : '',
  313. parameters.clearcoatNormalMap ? '#define USE_CLEARCOAT_NORMALMAP' : '',
  314. parameters.iridescenceMap ? '#define USE_IRIDESCENCEMAP' : '',
  315. parameters.iridescenceThicknessMap ? '#define USE_IRIDESCENCE_THICKNESSMAP' : '',
  316. parameters.specularMap ? '#define USE_SPECULARMAP' : '',
  317. parameters.specularColorMap ? '#define USE_SPECULAR_COLORMAP' : '',
  318. parameters.specularIntensityMap ? '#define USE_SPECULAR_INTENSITYMAP' : '',
  319. parameters.roughnessMap ? '#define USE_ROUGHNESSMAP' : '',
  320. parameters.metalnessMap ? '#define USE_METALNESSMAP' : '',
  321. parameters.alphaMap ? '#define USE_ALPHAMAP' : '',
  322. parameters.alphaHash ? '#define USE_ALPHAHASH' : '',
  323. parameters.transmission ? '#define USE_TRANSMISSION' : '',
  324. parameters.transmissionMap ? '#define USE_TRANSMISSIONMAP' : '',
  325. parameters.thicknessMap ? '#define USE_THICKNESSMAP' : '',
  326. parameters.sheenColorMap ? '#define USE_SHEEN_COLORMAP' : '',
  327. parameters.sheenRoughnessMap ? '#define USE_SHEEN_ROUGHNESSMAP' : '',
  328. //
  329. parameters.mapUv ? '#define MAP_UV ' + parameters.mapUv : '',
  330. parameters.alphaMapUv ? '#define ALPHAMAP_UV ' + parameters.alphaMapUv : '',
  331. parameters.lightMapUv ? '#define LIGHTMAP_UV ' + parameters.lightMapUv : '',
  332. parameters.aoMapUv ? '#define AOMAP_UV ' + parameters.aoMapUv : '',
  333. parameters.emissiveMapUv ? '#define EMISSIVEMAP_UV ' + parameters.emissiveMapUv : '',
  334. parameters.bumpMapUv ? '#define BUMPMAP_UV ' + parameters.bumpMapUv : '',
  335. parameters.normalMapUv ? '#define NORMALMAP_UV ' + parameters.normalMapUv : '',
  336. parameters.displacementMapUv ? '#define DISPLACEMENTMAP_UV ' + parameters.displacementMapUv : '',
  337. parameters.metalnessMapUv ? '#define METALNESSMAP_UV ' + parameters.metalnessMapUv : '',
  338. parameters.roughnessMapUv ? '#define ROUGHNESSMAP_UV ' + parameters.roughnessMapUv : '',
  339. parameters.anisotropyMapUv ? '#define ANISOTROPYMAP_UV ' + parameters.anisotropyMapUv : '',
  340. parameters.clearcoatMapUv ? '#define CLEARCOATMAP_UV ' + parameters.clearcoatMapUv : '',
  341. parameters.clearcoatNormalMapUv ? '#define CLEARCOAT_NORMALMAP_UV ' + parameters.clearcoatNormalMapUv : '',
  342. parameters.clearcoatRoughnessMapUv ? '#define CLEARCOAT_ROUGHNESSMAP_UV ' + parameters.clearcoatRoughnessMapUv : '',
  343. parameters.iridescenceMapUv ? '#define IRIDESCENCEMAP_UV ' + parameters.iridescenceMapUv : '',
  344. parameters.iridescenceThicknessMapUv ? '#define IRIDESCENCE_THICKNESSMAP_UV ' + parameters.iridescenceThicknessMapUv : '',
  345. parameters.sheenColorMapUv ? '#define SHEEN_COLORMAP_UV ' + parameters.sheenColorMapUv : '',
  346. parameters.sheenRoughnessMapUv ? '#define SHEEN_ROUGHNESSMAP_UV ' + parameters.sheenRoughnessMapUv : '',
  347. parameters.specularMapUv ? '#define SPECULARMAP_UV ' + parameters.specularMapUv : '',
  348. parameters.specularColorMapUv ? '#define SPECULAR_COLORMAP_UV ' + parameters.specularColorMapUv : '',
  349. parameters.specularIntensityMapUv ? '#define SPECULAR_INTENSITYMAP_UV ' + parameters.specularIntensityMapUv : '',
  350. parameters.transmissionMapUv ? '#define TRANSMISSIONMAP_UV ' + parameters.transmissionMapUv : '',
  351. parameters.thicknessMapUv ? '#define THICKNESSMAP_UV ' + parameters.thicknessMapUv : '',
  352. //
  353. parameters.vertexTangents && parameters.flatShading === false ? '#define USE_TANGENT' : '',
  354. parameters.vertexNormals ? '#define HAS_NORMAL' : '',
  355. parameters.vertexColors ? '#define USE_COLOR' : '',
  356. parameters.vertexAlphas ? '#define USE_COLOR_ALPHA' : '',
  357. parameters.vertexUv1s ? '#define USE_UV1' : '',
  358. parameters.vertexUv2s ? '#define USE_UV2' : '',
  359. parameters.vertexUv3s ? '#define USE_UV3' : '',
  360. parameters.pointsUvs ? '#define USE_POINTS_UV' : '',
  361. parameters.flatShading ? '#define FLAT_SHADED' : '',
  362. parameters.skinning ? '#define USE_SKINNING' : '',
  363. parameters.morphTargets ? '#define USE_MORPHTARGETS' : '',
  364. parameters.morphNormals && parameters.flatShading === false ? '#define USE_MORPHNORMALS' : '',
  365. ( parameters.morphColors ) ? '#define USE_MORPHCOLORS' : '',
  366. ( parameters.morphTargetsCount > 0 ) ? '#define MORPHTARGETS_TEXTURE_STRIDE ' + parameters.morphTextureStride : '',
  367. ( parameters.morphTargetsCount > 0 ) ? '#define MORPHTARGETS_COUNT ' + parameters.morphTargetsCount : '',
  368. parameters.doubleSided ? '#define DOUBLE_SIDED' : '',
  369. parameters.flipSided ? '#define FLIP_SIDED' : '',
  370. parameters.shadowMapEnabled ? '#define USE_SHADOWMAP' : '',
  371. parameters.shadowMapEnabled ? '#define ' + shadowMapTypeDefine : '',
  372. parameters.sizeAttenuation ? '#define USE_SIZEATTENUATION' : '',
  373. parameters.numLightProbes > 0 ? '#define USE_LIGHT_PROBES' : '',
  374. parameters.logarithmicDepthBuffer ? '#define USE_LOGARITHMIC_DEPTH_BUFFER' : '',
  375. parameters.reversedDepthBuffer ? '#define USE_REVERSED_DEPTH_BUFFER' : '',
  376. 'uniform mat4 modelMatrix;',
  377. 'uniform mat4 modelViewMatrix;',
  378. 'uniform mat4 projectionMatrix;',
  379. 'uniform mat4 viewMatrix;',
  380. 'uniform mat3 normalMatrix;',
  381. 'uniform vec3 cameraPosition;',
  382. 'uniform bool isOrthographic;',
  383. '#ifdef USE_INSTANCING',
  384. ' attribute mat4 instanceMatrix;',
  385. '#endif',
  386. '#ifdef USE_INSTANCING_COLOR',
  387. ' attribute vec3 instanceColor;',
  388. '#endif',
  389. '#ifdef USE_INSTANCING_MORPH',
  390. ' uniform sampler2D morphTexture;',
  391. '#endif',
  392. 'attribute vec3 position;',
  393. 'attribute vec3 normal;',
  394. 'attribute vec2 uv;',
  395. '#ifdef USE_UV1',
  396. ' attribute vec2 uv1;',
  397. '#endif',
  398. '#ifdef USE_UV2',
  399. ' attribute vec2 uv2;',
  400. '#endif',
  401. '#ifdef USE_UV3',
  402. ' attribute vec2 uv3;',
  403. '#endif',
  404. '#ifdef USE_TANGENT',
  405. ' attribute vec4 tangent;',
  406. '#endif',
  407. '#if defined( USE_COLOR_ALPHA )',
  408. ' attribute vec4 color;',
  409. '#elif defined( USE_COLOR )',
  410. ' attribute vec3 color;',
  411. '#endif',
  412. '#ifdef USE_SKINNING',
  413. ' attribute vec4 skinIndex;',
  414. ' attribute vec4 skinWeight;',
  415. '#endif',
  416. '\n'
  417. ].filter( filterEmptyLine ).join( '\n' );
  418. prefixFragment = [
  419. generatePrecision( parameters ),
  420. '#define SHADER_TYPE ' + parameters.shaderType,
  421. '#define SHADER_NAME ' + parameters.shaderName,
  422. customDefines,
  423. parameters.useFog && parameters.fog ? '#define USE_FOG' : '',
  424. parameters.useFog && parameters.fogExp2 ? '#define FOG_EXP2' : '',
  425. parameters.alphaToCoverage ? '#define ALPHA_TO_COVERAGE' : '',
  426. parameters.map ? '#define USE_MAP' : '',
  427. parameters.matcap ? '#define USE_MATCAP' : '',
  428. parameters.envMap ? '#define USE_ENVMAP' : '',
  429. parameters.envMap ? '#define ' + envMapTypeDefine : '',
  430. parameters.envMap ? '#define ' + envMapModeDefine : '',
  431. parameters.envMap ? '#define ' + envMapBlendingDefine : '',
  432. envMapCubeUVSize ? '#define CUBEUV_TEXEL_WIDTH ' + envMapCubeUVSize.texelWidth : '',
  433. envMapCubeUVSize ? '#define CUBEUV_TEXEL_HEIGHT ' + envMapCubeUVSize.texelHeight : '',
  434. envMapCubeUVSize ? '#define CUBEUV_MAX_MIP ' + envMapCubeUVSize.maxMip + '.0' : '',
  435. parameters.lightMap ? '#define USE_LIGHTMAP' : '',
  436. parameters.aoMap ? '#define USE_AOMAP' : '',
  437. parameters.bumpMap ? '#define USE_BUMPMAP' : '',
  438. parameters.normalMap ? '#define USE_NORMALMAP' : '',
  439. parameters.normalMapObjectSpace ? '#define USE_NORMALMAP_OBJECTSPACE' : '',
  440. parameters.normalMapTangentSpace ? '#define USE_NORMALMAP_TANGENTSPACE' : '',
  441. parameters.packedNormalMap ? '#define USE_PACKED_NORMALMAP' : '',
  442. parameters.emissiveMap ? '#define USE_EMISSIVEMAP' : '',
  443. parameters.anisotropy ? '#define USE_ANISOTROPY' : '',
  444. parameters.anisotropyMap ? '#define USE_ANISOTROPYMAP' : '',
  445. parameters.clearcoat ? '#define USE_CLEARCOAT' : '',
  446. parameters.clearcoatMap ? '#define USE_CLEARCOATMAP' : '',
  447. parameters.clearcoatRoughnessMap ? '#define USE_CLEARCOAT_ROUGHNESSMAP' : '',
  448. parameters.clearcoatNormalMap ? '#define USE_CLEARCOAT_NORMALMAP' : '',
  449. parameters.dispersion ? '#define USE_DISPERSION' : '',
  450. parameters.retroreflective ? '#define USE_RETROREFLECTIVE' : '',
  451. parameters.iridescence ? '#define USE_IRIDESCENCE' : '',
  452. parameters.iridescenceMap ? '#define USE_IRIDESCENCEMAP' : '',
  453. parameters.iridescenceThicknessMap ? '#define USE_IRIDESCENCE_THICKNESSMAP' : '',
  454. parameters.specularMap ? '#define USE_SPECULARMAP' : '',
  455. parameters.specularColorMap ? '#define USE_SPECULAR_COLORMAP' : '',
  456. parameters.specularIntensityMap ? '#define USE_SPECULAR_INTENSITYMAP' : '',
  457. parameters.roughnessMap ? '#define USE_ROUGHNESSMAP' : '',
  458. parameters.metalnessMap ? '#define USE_METALNESSMAP' : '',
  459. parameters.alphaMap ? '#define USE_ALPHAMAP' : '',
  460. parameters.alphaTest ? '#define USE_ALPHATEST' : '',
  461. parameters.alphaHash ? '#define USE_ALPHAHASH' : '',
  462. parameters.sheen ? '#define USE_SHEEN' : '',
  463. parameters.sheenColorMap ? '#define USE_SHEEN_COLORMAP' : '',
  464. parameters.sheenRoughnessMap ? '#define USE_SHEEN_ROUGHNESSMAP' : '',
  465. parameters.transmission ? '#define USE_TRANSMISSION' : '',
  466. parameters.transmissionMap ? '#define USE_TRANSMISSIONMAP' : '',
  467. parameters.thicknessMap ? '#define USE_THICKNESSMAP' : '',
  468. parameters.vertexTangents && parameters.flatShading === false ? '#define USE_TANGENT' : '',
  469. parameters.vertexColors || parameters.instancingColor ? '#define USE_COLOR' : '',
  470. parameters.vertexAlphas || parameters.batchingColor ? '#define USE_COLOR_ALPHA' : '',
  471. parameters.vertexUv1s ? '#define USE_UV1' : '',
  472. parameters.vertexUv2s ? '#define USE_UV2' : '',
  473. parameters.vertexUv3s ? '#define USE_UV3' : '',
  474. parameters.pointsUvs ? '#define USE_POINTS_UV' : '',
  475. parameters.gradientMap ? '#define USE_GRADIENTMAP' : '',
  476. parameters.flatShading ? '#define FLAT_SHADED' : '',
  477. parameters.doubleSided ? '#define DOUBLE_SIDED' : '',
  478. parameters.flipSided ? '#define FLIP_SIDED' : '',
  479. parameters.shadowMapEnabled ? '#define USE_SHADOWMAP' : '',
  480. parameters.shadowMapEnabled ? '#define ' + shadowMapTypeDefine : '',
  481. parameters.premultipliedAlpha ? '#define PREMULTIPLIED_ALPHA' : '',
  482. parameters.numLightProbes > 0 ? '#define USE_LIGHT_PROBES' : '',
  483. parameters.numLightProbeGrids > 0 ? '#define USE_LIGHT_PROBES_GRID' : '',
  484. parameters.decodeVideoTexture ? '#define DECODE_VIDEO_TEXTURE' : '',
  485. parameters.decodeVideoTextureEmissive ? '#define DECODE_VIDEO_TEXTURE_EMISSIVE' : '',
  486. parameters.logarithmicDepthBuffer ? '#define USE_LOGARITHMIC_DEPTH_BUFFER' : '',
  487. parameters.reversedDepthBuffer ? '#define USE_REVERSED_DEPTH_BUFFER' : '',
  488. 'uniform mat4 viewMatrix;',
  489. 'uniform vec3 cameraPosition;',
  490. 'uniform bool isOrthographic;',
  491. ( parameters.toneMapping !== NoToneMapping ) ? '#define TONE_MAPPING' : '',
  492. ( parameters.toneMapping !== NoToneMapping ) ? ShaderChunk[ 'tonemapping_pars_fragment' ] : '', // this code is required here because it is used by the toneMapping() function defined below
  493. ( parameters.toneMapping !== NoToneMapping ) ? getToneMappingFunction( 'toneMapping', parameters.toneMapping ) : '',
  494. parameters.dithering ? '#define DITHERING' : '',
  495. parameters.opaque ? '#define OPAQUE' : '',
  496. ShaderChunk[ 'colorspace_pars_fragment' ], // this code is required here because it is used by the various encoding/decoding function defined below
  497. getTexelEncodingFunction( 'linearToOutputTexel', parameters.outputColorSpace ),
  498. getLuminanceFunction(),
  499. parameters.useDepthPacking ? '#define DEPTH_PACKING ' + parameters.depthPacking : '',
  500. '\n'
  501. ].filter( filterEmptyLine ).join( '\n' );
  502. }
  503. vertexShader = resolveIncludes( vertexShader );
  504. vertexShader = replaceLightNums( vertexShader, parameters );
  505. vertexShader = replaceClippingPlaneNums( vertexShader, parameters );
  506. fragmentShader = resolveIncludes( fragmentShader );
  507. fragmentShader = replaceLightNums( fragmentShader, parameters );
  508. fragmentShader = replaceClippingPlaneNums( fragmentShader, parameters );
  509. vertexShader = unrollLoops( vertexShader );
  510. fragmentShader = unrollLoops( fragmentShader );
  511. if ( parameters.isRawShaderMaterial !== true ) {
  512. // GLSL 3.0 conversion for built-in materials and ShaderMaterial
  513. versionString = '#version 300 es\n';
  514. prefixVertex = [
  515. customVertexExtensions,
  516. '#define attribute in',
  517. '#define varying out',
  518. '#define texture2D texture'
  519. ].join( '\n' ) + '\n' + prefixVertex;
  520. prefixFragment = [
  521. '#define varying in',
  522. ( parameters.glslVersion === GLSL3 ) ? '' : 'layout(location = 0) out highp vec4 pc_fragColor;',
  523. ( parameters.glslVersion === GLSL3 ) ? '' : '#define gl_FragColor pc_fragColor',
  524. '#define gl_FragDepthEXT gl_FragDepth',
  525. '#define texture2D texture',
  526. '#define textureCube texture',
  527. '#define texture2DProj textureProj',
  528. '#define texture2DLodEXT textureLod',
  529. '#define texture2DProjLodEXT textureProjLod',
  530. '#define textureCubeLodEXT textureLod',
  531. '#define texture2DGradEXT textureGrad',
  532. '#define texture2DProjGradEXT textureProjGrad',
  533. '#define textureCubeGradEXT textureGrad'
  534. ].join( '\n' ) + '\n' + prefixFragment;
  535. }
  536. const vertexGlsl = versionString + prefixVertex + vertexShader;
  537. const fragmentGlsl = versionString + prefixFragment + fragmentShader;
  538. // log( '*VERTEX*', vertexGlsl );
  539. // log( '*FRAGMENT*', fragmentGlsl );
  540. const glVertexShader = WebGLShader( gl, gl.VERTEX_SHADER, vertexGlsl );
  541. const glFragmentShader = WebGLShader( gl, gl.FRAGMENT_SHADER, fragmentGlsl );
  542. gl.attachShader( program, glVertexShader );
  543. gl.attachShader( program, glFragmentShader );
  544. // Force a particular attribute to index 0.
  545. if ( parameters.index0AttributeName !== undefined ) {
  546. gl.bindAttribLocation( program, 0, parameters.index0AttributeName );
  547. } else if ( parameters.hasPositionAttribute === true ) {
  548. // below avoids the "Attribute 0 is disabled" performance penalty
  549. gl.bindAttribLocation( program, 0, 'position' );
  550. }
  551. gl.linkProgram( program );
  552. function onFirstUse( self ) {
  553. // check for link errors
  554. if ( renderer.debug.checkShaderErrors ) {
  555. const programInfoLog = gl.getProgramInfoLog( program ) || '';
  556. const vertexShaderInfoLog = gl.getShaderInfoLog( glVertexShader ) || '';
  557. const fragmentShaderInfoLog = gl.getShaderInfoLog( glFragmentShader ) || '';
  558. const programLog = programInfoLog.trim();
  559. const vertexLog = vertexShaderInfoLog.trim();
  560. const fragmentLog = fragmentShaderInfoLog.trim();
  561. let runnable = true;
  562. let haveDiagnostics = true;
  563. if ( gl.getProgramParameter( program, gl.LINK_STATUS ) === false ) {
  564. runnable = false;
  565. if ( typeof renderer.debug.onShaderError === 'function' ) {
  566. renderer.debug.onShaderError( gl, program, glVertexShader, glFragmentShader );
  567. } else {
  568. // default error reporting
  569. const vertexErrors = getShaderErrors( gl, glVertexShader, 'vertex' );
  570. const fragmentErrors = getShaderErrors( gl, glFragmentShader, 'fragment' );
  571. error(
  572. 'WebGLProgram: Shader Error ' + gl.getError() + ' - ' +
  573. 'VALIDATE_STATUS ' + gl.getProgramParameter( program, gl.VALIDATE_STATUS ) + '\n\n' +
  574. 'Material Name: ' + self.name + '\n' +
  575. 'Material Type: ' + self.type + '\n\n' +
  576. 'Program Info Log: ' + programLog + '\n' +
  577. vertexErrors + '\n' +
  578. fragmentErrors
  579. );
  580. }
  581. } else if ( programLog !== '' ) {
  582. warn( 'WebGLProgram: Program Info Log:', programLog );
  583. } else if ( vertexLog === '' || fragmentLog === '' ) {
  584. haveDiagnostics = false;
  585. }
  586. if ( haveDiagnostics ) {
  587. self.diagnostics = {
  588. runnable: runnable,
  589. programLog: programLog,
  590. vertexShader: {
  591. log: vertexLog,
  592. prefix: prefixVertex
  593. },
  594. fragmentShader: {
  595. log: fragmentLog,
  596. prefix: prefixFragment
  597. }
  598. };
  599. }
  600. }
  601. // Clean up
  602. // Crashes in iOS9 and iOS10. #18402
  603. // gl.detachShader( program, glVertexShader );
  604. // gl.detachShader( program, glFragmentShader );
  605. gl.deleteShader( glVertexShader );
  606. gl.deleteShader( glFragmentShader );
  607. cachedUniforms = new WebGLUniforms( gl, program );
  608. cachedAttributes = fetchAttributeLocations( gl, program );
  609. }
  610. // set up caching for uniform locations
  611. let cachedUniforms;
  612. this.getUniforms = function () {
  613. if ( cachedUniforms === undefined ) {
  614. // Populates cachedUniforms and cachedAttributes
  615. onFirstUse( this );
  616. }
  617. return cachedUniforms;
  618. };
  619. // set up caching for attribute locations
  620. let cachedAttributes;
  621. this.getAttributes = function () {
  622. if ( cachedAttributes === undefined ) {
  623. // Populates cachedAttributes and cachedUniforms
  624. onFirstUse( this );
  625. }
  626. return cachedAttributes;
  627. };
  628. // indicate when the program is ready to be used. if the KHR_parallel_shader_compile extension isn't supported,
  629. // flag the program as ready immediately. It may cause a stall when it's first used.
  630. let programReady = ( parameters.rendererExtensionParallelShaderCompile === false );
  631. this.isReady = function () {
  632. if ( programReady === false ) {
  633. programReady = gl.getProgramParameter( program, COMPLETION_STATUS_KHR );
  634. }
  635. return programReady;
  636. };
  637. // free resource
  638. this.destroy = function () {
  639. bindingStates.releaseStatesOfProgram( this );
  640. gl.deleteProgram( program );
  641. this.program = undefined;
  642. };
  643. //
  644. this.type = parameters.shaderType;
  645. this.name = parameters.shaderName;
  646. this.id = programIdCount ++;
  647. this.cacheKey = cacheKey;
  648. this.usedTimes = 1;
  649. this.program = program;
  650. this.vertexShader = glVertexShader;
  651. this.fragmentShader = glFragmentShader;
  652. return this;
  653. }
  654. export { WebGLProgram };
粤ICP备19079148号