2
0

WebGLProgram.js 24 KB

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