WebGLProgram.js 31 KB

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