WebGLProgram.js 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790
  1. /**
  2. * @author mrdoob / http://mrdoob.com/
  3. */
  4. import { WebGLUniforms } from './WebGLUniforms.js';
  5. import { WebGLShader } from './WebGLShader.js';
  6. import { ShaderChunk } from '../shaders/ShaderChunk.js';
  7. import { NoToneMapping, AddOperation, MixOperation, MultiplyOperation, EquirectangularRefractionMapping, CubeRefractionMapping, SphericalReflectionMapping, EquirectangularReflectionMapping, CubeUVRefractionMapping, CubeUVReflectionMapping, CubeReflectionMapping, PCFSoftShadowMap, PCFShadowMap, ACESFilmicToneMapping, CineonToneMapping, Uncharted2ToneMapping, ReinhardToneMapping, LinearToneMapping, GammaEncoding, RGBDEncoding, RGBM16Encoding, RGBM7Encoding, RGBEEncoding, sRGBEncoding, LinearEncoding } from '../../constants.js';
  8. var programIdCount = 0;
  9. function addLineNumbers( string ) {
  10. var lines = string.split( '\n' );
  11. for ( var i = 0; i < lines.length; i ++ ) {
  12. lines[ i ] = ( i + 1 ) + ': ' + lines[ i ];
  13. }
  14. return lines.join( '\n' );
  15. }
  16. function getEncodingComponents( encoding ) {
  17. switch ( encoding ) {
  18. case LinearEncoding:
  19. return [ 'Linear', '( value )' ];
  20. case sRGBEncoding:
  21. return [ 'sRGB', '( value )' ];
  22. case RGBEEncoding:
  23. return [ 'RGBE', '( value )' ];
  24. case RGBM7Encoding:
  25. return [ 'RGBM', '( value, 7.0 )' ];
  26. case RGBM16Encoding:
  27. return [ 'RGBM', '( value, 16.0 )' ];
  28. case RGBDEncoding:
  29. return [ 'RGBD', '( value, 256.0 )' ];
  30. case GammaEncoding:
  31. return [ 'Gamma', '( value, float( GAMMA_FACTOR ) )' ];
  32. default:
  33. throw new Error( 'unsupported encoding: ' + encoding );
  34. }
  35. }
  36. function getShaderErrors( gl, shader, type ) {
  37. var status = gl.getShaderParameter( shader, gl.COMPILE_STATUS );
  38. var log = gl.getShaderInfoLog( shader ).trim();
  39. if ( status && log === '' ) return '';
  40. // --enable-privileged-webgl-extension
  41. // console.log( '**' + type + '**', gl.getExtension( 'WEBGL_debug_shaders' ).getTranslatedShaderSource( shader ) );
  42. var source = gl.getShaderSource( shader );
  43. return 'THREE.WebGLShader: gl.getShaderInfoLog() ' + type + '\n' + log + addLineNumbers( source );
  44. }
  45. function getTexelDecodingFunction( functionName, encoding ) {
  46. var components = getEncodingComponents( encoding );
  47. return 'vec4 ' + functionName + '( vec4 value ) { return ' + components[ 0 ] + 'ToLinear' + components[ 1 ] + '; }';
  48. }
  49. function getTexelEncodingFunction( functionName, encoding ) {
  50. var components = getEncodingComponents( encoding );
  51. return 'vec4 ' + functionName + '( vec4 value ) { return LinearTo' + components[ 0 ] + components[ 1 ] + '; }';
  52. }
  53. function getToneMappingFunction( functionName, toneMapping ) {
  54. var toneMappingName;
  55. switch ( toneMapping ) {
  56. case LinearToneMapping:
  57. toneMappingName = 'Linear';
  58. break;
  59. case ReinhardToneMapping:
  60. toneMappingName = 'Reinhard';
  61. break;
  62. case Uncharted2ToneMapping:
  63. toneMappingName = 'Uncharted2';
  64. break;
  65. case CineonToneMapping:
  66. toneMappingName = 'OptimizedCineon';
  67. break;
  68. case ACESFilmicToneMapping:
  69. toneMappingName = 'ACESFilmic';
  70. break;
  71. default:
  72. throw new Error( 'unsupported toneMapping: ' + toneMapping );
  73. }
  74. return 'vec3 ' + functionName + '( vec3 color ) { return ' + toneMappingName + 'ToneMapping( color ); }';
  75. }
  76. function generateExtensions( extensions, parameters, rendererExtensions ) {
  77. extensions = extensions || {};
  78. var chunks = [
  79. ( extensions.derivatives || parameters.envMapCubeUV || parameters.bumpMap || ( parameters.normalMap && ! parameters.objectSpaceNormalMap ) || parameters.flatShading ) ? '#extension GL_OES_standard_derivatives : enable' : '',
  80. ( extensions.fragDepth || parameters.logarithmicDepthBuffer ) && rendererExtensions.get( 'EXT_frag_depth' ) ? '#extension GL_EXT_frag_depth : enable' : '',
  81. ( extensions.drawBuffers ) && rendererExtensions.get( 'WEBGL_draw_buffers' ) ? '#extension GL_EXT_draw_buffers : require' : '',
  82. ( extensions.shaderTextureLOD || parameters.envMap ) && rendererExtensions.get( 'EXT_shader_texture_lod' ) ? '#extension GL_EXT_shader_texture_lod : enable' : ''
  83. ];
  84. return chunks.filter( filterEmptyLine ).join( '\n' );
  85. }
  86. function generateDefines( defines ) {
  87. var chunks = [];
  88. for ( var name in defines ) {
  89. var value = defines[ name ];
  90. if ( value === false ) continue;
  91. chunks.push( '#define ' + name + ' ' + value );
  92. }
  93. return chunks.join( '\n' );
  94. }
  95. function fetchAttributeLocations( gl, program ) {
  96. var attributes = {};
  97. var n = gl.getProgramParameter( program, gl.ACTIVE_ATTRIBUTES );
  98. for ( var i = 0; i < n; i ++ ) {
  99. var info = gl.getActiveAttrib( program, i );
  100. var name = info.name;
  101. // console.log( 'THREE.WebGLProgram: ACTIVE VERTEX ATTRIBUTE:', name, i );
  102. attributes[ name ] = gl.getAttribLocation( program, name );
  103. }
  104. return attributes;
  105. }
  106. function filterEmptyLine( string ) {
  107. return string !== '';
  108. }
  109. function replaceLightNums( string, parameters ) {
  110. return string
  111. .replace( /NUM_DIR_LIGHTS/g, parameters.numDirLights )
  112. .replace( /NUM_SPOT_LIGHTS/g, parameters.numSpotLights )
  113. .replace( /NUM_RECT_AREA_LIGHTS/g, parameters.numRectAreaLights )
  114. .replace( /NUM_POINT_LIGHTS/g, parameters.numPointLights )
  115. .replace( /NUM_HEMI_LIGHTS/g, parameters.numHemiLights );
  116. }
  117. function replaceClippingPlaneNums( string, parameters ) {
  118. return string
  119. .replace( /NUM_CLIPPING_PLANES/g, parameters.numClippingPlanes )
  120. .replace( /UNION_CLIPPING_PLANES/g, ( parameters.numClippingPlanes - parameters.numClipIntersection ) );
  121. }
  122. function parseIncludes( string ) {
  123. var pattern = /^[ \t]*#include +<([\w\d./]+)>/gm;
  124. function replace( match, include ) {
  125. var replace = ShaderChunk[ include ];
  126. if ( replace === undefined ) {
  127. throw new Error( 'Can not resolve #include <' + include + '>' );
  128. }
  129. return parseIncludes( replace );
  130. }
  131. return string.replace( pattern, replace );
  132. }
  133. function unrollLoops( string ) {
  134. var pattern = /#pragma unroll_loop[\s]+?for \( int i \= (\d+)\; i < (\d+)\; i \+\+ \) \{([\s\S]+?)(?=\})\}/g;
  135. function replace( match, start, end, snippet ) {
  136. var unroll = '';
  137. for ( var i = parseInt( start ); i < parseInt( end ); i ++ ) {
  138. unroll += snippet.replace( /\[ i \]/g, '[ ' + i + ' ]' );
  139. }
  140. return unroll;
  141. }
  142. return string.replace( pattern, replace );
  143. }
  144. function WebGLProgram( renderer, extensions, code, material, shader, parameters, capabilities ) {
  145. var gl = renderer.getContext();
  146. var defines = material.defines;
  147. var vertexShader = shader.vertexShader;
  148. var fragmentShader = shader.fragmentShader;
  149. var shadowMapTypeDefine = 'SHADOWMAP_TYPE_BASIC';
  150. if ( parameters.shadowMapType === PCFShadowMap ) {
  151. shadowMapTypeDefine = 'SHADOWMAP_TYPE_PCF';
  152. } else if ( parameters.shadowMapType === PCFSoftShadowMap ) {
  153. shadowMapTypeDefine = 'SHADOWMAP_TYPE_PCF_SOFT';
  154. }
  155. var envMapTypeDefine = 'ENVMAP_TYPE_CUBE';
  156. var envMapModeDefine = 'ENVMAP_MODE_REFLECTION';
  157. var envMapBlendingDefine = 'ENVMAP_BLENDING_MULTIPLY';
  158. if ( parameters.envMap ) {
  159. switch ( material.envMap.mapping ) {
  160. case CubeReflectionMapping:
  161. case CubeRefractionMapping:
  162. envMapTypeDefine = 'ENVMAP_TYPE_CUBE';
  163. break;
  164. case CubeUVReflectionMapping:
  165. case CubeUVRefractionMapping:
  166. envMapTypeDefine = 'ENVMAP_TYPE_CUBE_UV';
  167. break;
  168. case EquirectangularReflectionMapping:
  169. case EquirectangularRefractionMapping:
  170. envMapTypeDefine = 'ENVMAP_TYPE_EQUIREC';
  171. break;
  172. case SphericalReflectionMapping:
  173. envMapTypeDefine = 'ENVMAP_TYPE_SPHERE';
  174. break;
  175. }
  176. switch ( material.envMap.mapping ) {
  177. case CubeRefractionMapping:
  178. case EquirectangularRefractionMapping:
  179. envMapModeDefine = 'ENVMAP_MODE_REFRACTION';
  180. break;
  181. }
  182. switch ( material.combine ) {
  183. case MultiplyOperation:
  184. envMapBlendingDefine = 'ENVMAP_BLENDING_MULTIPLY';
  185. break;
  186. case MixOperation:
  187. envMapBlendingDefine = 'ENVMAP_BLENDING_MIX';
  188. break;
  189. case AddOperation:
  190. envMapBlendingDefine = 'ENVMAP_BLENDING_ADD';
  191. break;
  192. }
  193. }
  194. var gammaFactorDefine = ( renderer.gammaFactor > 0 ) ? renderer.gammaFactor : 1.0;
  195. // console.log( 'building new program ' );
  196. //
  197. var customExtensions = capabilities.isWebGL2 ? '' : generateExtensions( material.extensions, parameters, extensions );
  198. var customDefines = generateDefines( defines );
  199. //
  200. var program = gl.createProgram();
  201. var prefixVertex, prefixFragment;
  202. if ( material.isRawShaderMaterial ) {
  203. prefixVertex = [
  204. customDefines
  205. ].filter( filterEmptyLine ).join( '\n' );
  206. if ( prefixVertex.length > 0 ) {
  207. prefixVertex += '\n';
  208. }
  209. prefixFragment = [
  210. customExtensions,
  211. customDefines
  212. ].filter( filterEmptyLine ).join( '\n' );
  213. if ( prefixFragment.length > 0 ) {
  214. prefixFragment += '\n';
  215. }
  216. } else {
  217. var numMultiviewViews = renderer.multiview.getNumViews();
  218. prefixVertex = [
  219. 'precision ' + parameters.precision + ' float;',
  220. 'precision ' + parameters.precision + ' int;',
  221. '#define SHADER_NAME ' + shader.name,
  222. customDefines,
  223. parameters.supportsVertexTextures ? '#define VERTEX_TEXTURES' : '',
  224. '#define GAMMA_FACTOR ' + gammaFactorDefine,
  225. '#define MAX_BONES ' + parameters.maxBones,
  226. ( parameters.useFog && parameters.fog ) ? '#define USE_FOG' : '',
  227. ( parameters.useFog && parameters.fogExp ) ? '#define FOG_EXP2' : '',
  228. parameters.map ? '#define USE_MAP' : '',
  229. parameters.envMap ? '#define USE_ENVMAP' : '',
  230. parameters.envMap ? '#define ' + envMapModeDefine : '',
  231. parameters.lightMap ? '#define USE_LIGHTMAP' : '',
  232. parameters.aoMap ? '#define USE_AOMAP' : '',
  233. parameters.emissiveMap ? '#define USE_EMISSIVEMAP' : '',
  234. parameters.bumpMap ? '#define USE_BUMPMAP' : '',
  235. parameters.normalMap ? '#define USE_NORMALMAP' : '',
  236. ( parameters.normalMap && parameters.objectSpaceNormalMap ) ? '#define OBJECTSPACE_NORMALMAP' : '',
  237. parameters.displacementMap && parameters.supportsVertexTextures ? '#define USE_DISPLACEMENTMAP' : '',
  238. parameters.specularMap ? '#define USE_SPECULARMAP' : '',
  239. parameters.roughnessMap ? '#define USE_ROUGHNESSMAP' : '',
  240. parameters.metalnessMap ? '#define USE_METALNESSMAP' : '',
  241. parameters.alphaMap ? '#define USE_ALPHAMAP' : '',
  242. parameters.vertexTangents ? '#define USE_TANGENT' : '',
  243. parameters.vertexColors ? '#define USE_COLOR' : '',
  244. parameters.flatShading ? '#define FLAT_SHADED' : '',
  245. parameters.skinning ? '#define USE_SKINNING' : '',
  246. parameters.useVertexTexture ? '#define BONE_TEXTURE' : '',
  247. parameters.morphTargets ? '#define USE_MORPHTARGETS' : '',
  248. parameters.morphNormals && parameters.flatShading === false ? '#define USE_MORPHNORMALS' : '',
  249. parameters.doubleSided ? '#define DOUBLE_SIDED' : '',
  250. parameters.flipSided ? '#define FLIP_SIDED' : '',
  251. parameters.shadowMapEnabled ? '#define USE_SHADOWMAP' : '',
  252. parameters.shadowMapEnabled ? '#define ' + shadowMapTypeDefine : '',
  253. parameters.sizeAttenuation ? '#define USE_SIZEATTENUATION' : '',
  254. parameters.logarithmicDepthBuffer ? '#define USE_LOGDEPTHBUF' : '',
  255. parameters.logarithmicDepthBuffer && ( capabilities.isWebGL2 || extensions.get( 'EXT_frag_depth' ) ) ? '#define USE_LOGDEPTHBUF_EXT' : '',
  256. 'uniform mat4 modelMatrix;',
  257. 'uniform vec3 cameraPosition;',
  258. material.supportsMultiview && renderer.multiview.isEnabled() ? [
  259. 'uniform mat4 modelViewMatrices[' + numMultiviewViews + '];',
  260. 'uniform mat3 normalMatrices[' + numMultiviewViews + '];',
  261. 'uniform mat4 viewMatrices[' + numMultiviewViews + '];',
  262. 'uniform mat4 projectionMatrices[' + numMultiviewViews + '];',
  263. '#define modelViewMatrix modelViewMatrices[VIEW_ID]',
  264. '#define normalMatrix normalMatrices[VIEW_ID]',
  265. '#define viewMatrix viewMatrices[VIEW_ID]',
  266. '#define projectionMatrix projectionMatrices[VIEW_ID]'
  267. ].join( '\n' ) : [
  268. 'uniform mat4 modelViewMatrix;',
  269. 'uniform mat4 projectionMatrix;',
  270. 'uniform mat4 viewMatrix;',
  271. 'uniform mat3 normalMatrix;',
  272. ].join( '\n' ),
  273. 'attribute vec3 position;',
  274. 'attribute vec3 normal;',
  275. 'attribute vec2 uv;',
  276. '#ifdef USE_TANGENT',
  277. ' attribute vec4 tangent;',
  278. '#endif',
  279. '#ifdef USE_COLOR',
  280. ' attribute vec3 color;',
  281. '#endif',
  282. '#ifdef USE_MORPHTARGETS',
  283. ' attribute vec3 morphTarget0;',
  284. ' attribute vec3 morphTarget1;',
  285. ' attribute vec3 morphTarget2;',
  286. ' attribute vec3 morphTarget3;',
  287. ' #ifdef USE_MORPHNORMALS',
  288. ' attribute vec3 morphNormal0;',
  289. ' attribute vec3 morphNormal1;',
  290. ' attribute vec3 morphNormal2;',
  291. ' attribute vec3 morphNormal3;',
  292. ' #else',
  293. ' attribute vec3 morphTarget4;',
  294. ' attribute vec3 morphTarget5;',
  295. ' attribute vec3 morphTarget6;',
  296. ' attribute vec3 morphTarget7;',
  297. ' #endif',
  298. '#endif',
  299. '#ifdef USE_SKINNING',
  300. ' attribute vec4 skinIndex;',
  301. ' attribute vec4 skinWeight;',
  302. '#endif',
  303. '\n'
  304. ].filter( filterEmptyLine ).join( '\n' );
  305. prefixFragment = [
  306. customExtensions,
  307. 'precision ' + parameters.precision + ' float;',
  308. 'precision ' + parameters.precision + ' int;',
  309. '#define SHADER_NAME ' + shader.name,
  310. customDefines,
  311. parameters.alphaTest ? '#define ALPHATEST ' + parameters.alphaTest + ( parameters.alphaTest % 1 ? '' : '.0' ) : '', // add '.0' if integer
  312. '#define GAMMA_FACTOR ' + gammaFactorDefine,
  313. ( parameters.useFog && parameters.fog ) ? '#define USE_FOG' : '',
  314. ( parameters.useFog && parameters.fogExp ) ? '#define FOG_EXP2' : '',
  315. parameters.map ? '#define USE_MAP' : '',
  316. parameters.matcap ? '#define USE_MATCAP' : '',
  317. parameters.envMap ? '#define USE_ENVMAP' : '',
  318. parameters.envMap ? '#define ' + envMapTypeDefine : '',
  319. parameters.envMap ? '#define ' + envMapModeDefine : '',
  320. parameters.envMap ? '#define ' + envMapBlendingDefine : '',
  321. parameters.lightMap ? '#define USE_LIGHTMAP' : '',
  322. parameters.aoMap ? '#define USE_AOMAP' : '',
  323. parameters.emissiveMap ? '#define USE_EMISSIVEMAP' : '',
  324. parameters.bumpMap ? '#define USE_BUMPMAP' : '',
  325. parameters.normalMap ? '#define USE_NORMALMAP' : '',
  326. ( parameters.normalMap && parameters.objectSpaceNormalMap ) ? '#define OBJECTSPACE_NORMALMAP' : '',
  327. parameters.specularMap ? '#define USE_SPECULARMAP' : '',
  328. parameters.roughnessMap ? '#define USE_ROUGHNESSMAP' : '',
  329. parameters.metalnessMap ? '#define USE_METALNESSMAP' : '',
  330. parameters.alphaMap ? '#define USE_ALPHAMAP' : '',
  331. parameters.vertexTangents ? '#define USE_TANGENT' : '',
  332. parameters.vertexColors ? '#define USE_COLOR' : '',
  333. parameters.gradientMap ? '#define USE_GRADIENTMAP' : '',
  334. parameters.flatShading ? '#define FLAT_SHADED' : '',
  335. parameters.doubleSided ? '#define DOUBLE_SIDED' : '',
  336. parameters.flipSided ? '#define FLIP_SIDED' : '',
  337. parameters.shadowMapEnabled ? '#define USE_SHADOWMAP' : '',
  338. parameters.shadowMapEnabled ? '#define ' + shadowMapTypeDefine : '',
  339. parameters.premultipliedAlpha ? '#define PREMULTIPLIED_ALPHA' : '',
  340. parameters.physicallyCorrectLights ? '#define PHYSICALLY_CORRECT_LIGHTS' : '',
  341. parameters.logarithmicDepthBuffer ? '#define USE_LOGDEPTHBUF' : '',
  342. parameters.logarithmicDepthBuffer && ( capabilities.isWebGL2 || extensions.get( 'EXT_frag_depth' ) ) ? '#define USE_LOGDEPTHBUF_EXT' : '',
  343. parameters.envMap && ( capabilities.isWebGL2 || extensions.get( 'EXT_shader_texture_lod' ) ) ? '#define TEXTURE_LOD_EXT' : '',
  344. 'uniform vec3 cameraPosition;',
  345. material.supportsMultiview && renderer.multiview.isEnabled() ? [
  346. 'uniform mat4 viewMatrices[' + numMultiviewViews + '];',
  347. '#define viewMatrix viewMatrices[VIEW_ID]'
  348. ].join( '\n' ) : 'uniform mat4 viewMatrix;',
  349. ( parameters.toneMapping !== NoToneMapping ) ? '#define TONE_MAPPING' : '',
  350. ( parameters.toneMapping !== NoToneMapping ) ? ShaderChunk[ 'tonemapping_pars_fragment' ] : '', // this code is required here because it is used by the toneMapping() function defined below
  351. ( parameters.toneMapping !== NoToneMapping ) ? getToneMappingFunction( 'toneMapping', parameters.toneMapping ) : '',
  352. parameters.dithering ? '#define DITHERING' : '',
  353. ( parameters.outputEncoding || parameters.mapEncoding || parameters.matcapEncoding || parameters.envMapEncoding || parameters.emissiveMapEncoding ) ?
  354. ShaderChunk[ 'encodings_pars_fragment' ] : '', // this code is required here because it is used by the various encoding/decoding function defined below
  355. parameters.mapEncoding ? getTexelDecodingFunction( 'mapTexelToLinear', parameters.mapEncoding ) : '',
  356. parameters.matcapEncoding ? getTexelDecodingFunction( 'matcapTexelToLinear', parameters.matcapEncoding ) : '',
  357. parameters.envMapEncoding ? getTexelDecodingFunction( 'envMapTexelToLinear', parameters.envMapEncoding ) : '',
  358. parameters.emissiveMapEncoding ? getTexelDecodingFunction( 'emissiveMapTexelToLinear', parameters.emissiveMapEncoding ) : '',
  359. parameters.outputEncoding ? getTexelEncodingFunction( 'linearToOutputTexel', parameters.outputEncoding ) : '',
  360. parameters.depthPacking ? '#define DEPTH_PACKING ' + material.depthPacking : '',
  361. '\n'
  362. ].filter( filterEmptyLine ).join( '\n' );
  363. }
  364. vertexShader = parseIncludes( vertexShader );
  365. vertexShader = replaceLightNums( vertexShader, parameters );
  366. vertexShader = replaceClippingPlaneNums( vertexShader, parameters );
  367. fragmentShader = parseIncludes( fragmentShader );
  368. fragmentShader = replaceLightNums( fragmentShader, parameters );
  369. fragmentShader = replaceClippingPlaneNums( fragmentShader, parameters );
  370. vertexShader = unrollLoops( vertexShader );
  371. fragmentShader = unrollLoops( fragmentShader );
  372. if ( capabilities.isWebGL2 && ! material.isRawShaderMaterial ) {
  373. var isGLSL3ShaderMaterial = false;
  374. var versionRegex = /^\s*#version\s+300\s+es\s*\n/;
  375. if ( material.isShaderMaterial &&
  376. vertexShader.match( versionRegex ) !== null &&
  377. fragmentShader.match( versionRegex ) !== null ) {
  378. isGLSL3ShaderMaterial = true;
  379. vertexShader = vertexShader.replace( versionRegex, '' );
  380. fragmentShader = fragmentShader.replace( versionRegex, '' );
  381. }
  382. // GLSL 3.0 conversion
  383. prefixVertex = [
  384. '#version 300 es\n',
  385. material.supportsMultiview && renderer.multiview.isEnabled() ? [
  386. '#extension GL_OVR_multiview2 : require',
  387. 'layout(num_views = 2) in;',
  388. '#define VIEW_ID gl_ViewID_OVR'
  389. ].join( '\n' ) : '',
  390. '#define attribute in',
  391. '#define varying out',
  392. '#define texture2D texture'
  393. ].join( '\n' ) + '\n' + prefixVertex;
  394. prefixFragment = [
  395. '#version 300 es\n',
  396. material.supportsMultiview && renderer.multiview.isEnabled() ? [
  397. '#extension GL_OVR_multiview2 : require',
  398. '#define VIEW_ID gl_ViewID_OVR'
  399. ].join( '\n' ) : '',
  400. '#define varying in',
  401. isGLSL3ShaderMaterial ? '' : 'out highp vec4 pc_fragColor;',
  402. isGLSL3ShaderMaterial ? '' : '#define gl_FragColor pc_fragColor',
  403. '#define gl_FragDepthEXT gl_FragDepth',
  404. '#define texture2D texture',
  405. '#define textureCube texture',
  406. '#define texture2DProj textureProj',
  407. '#define texture2DLodEXT textureLod',
  408. '#define texture2DProjLodEXT textureProjLod',
  409. '#define textureCubeLodEXT textureLod',
  410. '#define texture2DGradEXT textureGrad',
  411. '#define texture2DProjGradEXT textureProjGrad',
  412. '#define textureCubeGradEXT textureGrad'
  413. ].join( '\n' ) + '\n' + prefixFragment;
  414. }
  415. var vertexGlsl = prefixVertex + vertexShader;
  416. var fragmentGlsl = prefixFragment + fragmentShader;
  417. var glVertexShader = WebGLShader( gl, gl.VERTEX_SHADER, vertexGlsl, renderer.debug.checkShaderErrors );
  418. var glFragmentShader = WebGLShader( gl, gl.FRAGMENT_SHADER, fragmentGlsl, renderer.debug.checkShaderErrors );
  419. gl.attachShader( program, glVertexShader );
  420. gl.attachShader( program, glFragmentShader );
  421. // Force a particular attribute to index 0.
  422. if ( material.index0AttributeName !== undefined ) {
  423. gl.bindAttribLocation( program, 0, material.index0AttributeName );
  424. } else if ( parameters.morphTargets === true ) {
  425. // programs with morphTargets displace position out of attribute 0
  426. gl.bindAttribLocation( program, 0, 'position' );
  427. }
  428. gl.linkProgram( program );
  429. // check for link errors
  430. if ( renderer.debug.checkShaderErrors ) {
  431. var programLog = gl.getProgramInfoLog( program ).trim();
  432. var vertexLog = gl.getShaderInfoLog( glVertexShader ).trim();
  433. var fragmentLog = gl.getShaderInfoLog( glFragmentShader ).trim();
  434. var runnable = true;
  435. var haveDiagnostics = true;
  436. if ( gl.getProgramParameter( program, gl.LINK_STATUS ) === false ) {
  437. runnable = false;
  438. var vertexErrors = getShaderErrors( gl, glVertexShader, 'vertex' );
  439. var fragmentErrors = getShaderErrors( gl, glFragmentShader, 'fragment' );
  440. console.error( 'THREE.WebGLProgram: shader error: ', gl.getError(), 'gl.VALIDATE_STATUS', gl.getProgramParameter( program, gl.VALIDATE_STATUS ), 'gl.getProgramInfoLog', programLog, vertexErrors, fragmentErrors );
  441. } else if ( programLog !== '' ) {
  442. console.warn( 'THREE.WebGLProgram: gl.getProgramInfoLog()', programLog );
  443. } else if ( vertexLog === '' || fragmentLog === '' ) {
  444. haveDiagnostics = false;
  445. }
  446. if ( haveDiagnostics ) {
  447. this.diagnostics = {
  448. runnable: runnable,
  449. material: material,
  450. programLog: programLog,
  451. vertexShader: {
  452. log: vertexLog,
  453. prefix: prefixVertex
  454. },
  455. fragmentShader: {
  456. log: fragmentLog,
  457. prefix: prefixFragment
  458. }
  459. };
  460. }
  461. }
  462. // clean up
  463. gl.deleteShader( glVertexShader );
  464. gl.deleteShader( glFragmentShader );
  465. // set up caching for uniform locations
  466. var cachedUniforms;
  467. this.getUniforms = function () {
  468. if ( cachedUniforms === undefined ) {
  469. cachedUniforms = new WebGLUniforms( gl, program );
  470. }
  471. return cachedUniforms;
  472. };
  473. // set up caching for attribute locations
  474. var cachedAttributes;
  475. this.getAttributes = function () {
  476. if ( cachedAttributes === undefined ) {
  477. cachedAttributes = fetchAttributeLocations( gl, program );
  478. }
  479. return cachedAttributes;
  480. };
  481. // free resource
  482. this.destroy = function () {
  483. gl.deleteProgram( program );
  484. this.program = undefined;
  485. };
  486. //
  487. this.name = shader.name;
  488. this.id = programIdCount ++;
  489. this.code = code;
  490. this.usedTimes = 1;
  491. this.program = program;
  492. this.vertexShader = glVertexShader;
  493. this.fragmentShader = glFragmentShader;
  494. return this;
  495. }
  496. export { WebGLProgram };