misc_exporter_exr.html 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgl - exporter - exr</title>
  5. <meta charset="utf-8">
  6. <meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
  7. <link type="text/css" rel="stylesheet" href="main.css">
  8. </head>
  9. <body>
  10. <div id="info">
  11. <a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> webgl - exporter - exr
  12. </div>
  13. <script type="importmap">
  14. {
  15. "imports": {
  16. "three": "../build/three.module.js",
  17. "three/addons/": "./jsm/"
  18. }
  19. }
  20. </script>
  21. <script type="module">
  22. import * as THREE from 'three';
  23. import { OrbitControls } from 'three/addons/controls/OrbitControls.js';
  24. import { EXRExporter, ZIP_COMPRESSION, ZIPS_COMPRESSION, NO_COMPRESSION } from 'three/addons/exporters/EXRExporter.js';
  25. import { RGBELoader } from 'three/addons/loaders/RGBELoader.js';
  26. import { GUI } from 'three/addons/libs/lil-gui.module.min.js';
  27. let scene, camera, renderer, exporter, mesh, controls, renderTarget, dataTexture;
  28. const params = {
  29. target: 'pmrem',
  30. type: 'HalfFloatType',
  31. compression: 'ZIP',
  32. export: exportFile
  33. };
  34. init();
  35. animate();
  36. function init() {
  37. renderer = new THREE.WebGLRenderer( { antialias: true } );
  38. renderer.setPixelRatio( window.devicePixelRatio );
  39. renderer.setSize( window.innerWidth, window.innerHeight );
  40. document.body.appendChild( renderer.domElement );
  41. //
  42. camera = new THREE.PerspectiveCamera( 50, window.innerWidth / window.innerHeight, 0.1, 100 );
  43. camera.position.set( 10, 0, 0 );
  44. scene = new THREE.Scene();
  45. exporter = new EXRExporter();
  46. const rgbeloader = new RGBELoader();
  47. //
  48. const pmremGenerator = new THREE.PMREMGenerator( renderer );
  49. pmremGenerator.compileEquirectangularShader();
  50. rgbeloader.load( 'textures/equirectangular/san_giuseppe_bridge_2k.hdr', function ( texture ) {
  51. texture.mapping = THREE.EquirectangularReflectionMapping;
  52. renderTarget = pmremGenerator.fromEquirectangular( texture );
  53. scene.background = renderTarget.texture;
  54. } );
  55. createDataTexture();
  56. //
  57. controls = new OrbitControls( camera, renderer.domElement );
  58. controls.enableDamping = true;
  59. controls.rotateSpeed = - 0.25; // negative, to track mouse pointer
  60. //
  61. window.addEventListener( 'resize', onWindowResize );
  62. const gui = new GUI();
  63. const input = gui.addFolder( 'Input' );
  64. input.add( params, 'target' ).options( [ 'pmrem', 'data-texture' ] ).onChange( swapScene );
  65. const options = gui.addFolder( 'Output Options' );
  66. options.add( params, 'type' ).options( [ 'FloatType', 'HalfFloatType' ] );
  67. options.add( params, 'compression' ).options( [ 'ZIP', 'ZIPS', 'NONE' ] );
  68. gui.add( params, 'export' ).name( 'Export EXR' );
  69. gui.open();
  70. }
  71. function onWindowResize() {
  72. camera.aspect = window.innerWidth / window.innerHeight;
  73. camera.updateProjectionMatrix();
  74. renderer.setSize( window.innerWidth, window.innerHeight );
  75. }
  76. function animate() {
  77. requestAnimationFrame( animate );
  78. controls.update();
  79. renderer.render( scene, camera );
  80. }
  81. function createDataTexture() {
  82. const normal = new THREE.Vector3();
  83. const coord = new THREE.Vector2();
  84. const size = 800, radius = 320, factor = Math.PI * 0.5 / radius;
  85. const data = new Float32Array( 4 * size * size );
  86. for ( let i = 0; i < size; i ++ ) {
  87. for ( let j = 0; j < size; j ++ ) {
  88. const idx = i * size * 4 + j * 4;
  89. coord.set( j, i ).subScalar( size / 2 );
  90. if ( coord.length() < radius )
  91. normal.set(
  92. Math.sin( coord.x * factor ),
  93. Math.sin( coord.y * factor ),
  94. Math.cos( coord.x * factor )
  95. );
  96. else
  97. normal.set( 0, 0, 1 );
  98. data[ idx + 0 ] = .5 + .5 * normal.x;
  99. data[ idx + 1 ] = .5 + .5 * normal.y;
  100. data[ idx + 2 ] = .5 + .5 * normal.z;
  101. data[ idx + 3 ] = 1.;
  102. }
  103. }
  104. dataTexture = new THREE.DataTexture( data, size, size, THREE.RGBAFormat, THREE.FloatType );
  105. dataTexture.needsUpdate = true;
  106. const material = new THREE.MeshBasicMaterial( { map: dataTexture } );
  107. const quad = new THREE.PlaneGeometry( 50, 50 );
  108. mesh = new THREE.Mesh( quad, material );
  109. mesh.visible = false;
  110. scene.add( mesh );
  111. }
  112. function swapScene() {
  113. if ( params.target == 'pmrem' ) {
  114. camera.position.set( 10, 0, 0 );
  115. controls.enabled = true;
  116. scene.background = renderTarget.texture;
  117. mesh.visible = false;
  118. } else {
  119. camera.position.set( 0, 0, 70 );
  120. controls.enabled = false;
  121. scene.background = new THREE.Color( 0, 0, 0 );
  122. mesh.visible = true;
  123. }
  124. }
  125. function exportFile() {
  126. let result, exportType, exportCompression;
  127. if ( params.type == 'HalfFloatType' )
  128. exportType = THREE.HalfFloatType;
  129. else
  130. exportType = THREE.FloatType;
  131. if ( params.compression == 'ZIP' )
  132. exportCompression = ZIP_COMPRESSION;
  133. else if ( params.compression == 'ZIPS' )
  134. exportCompression = ZIPS_COMPRESSION;
  135. else
  136. exportCompression = NO_COMPRESSION;
  137. if ( params.target == 'pmrem' )
  138. result = exporter.parse( renderer, renderTarget, { type: exportType, compression: exportCompression } );
  139. else
  140. result = exporter.parse( dataTexture, { type: exportType, compression: exportCompression } );
  141. saveArrayBuffer( result, params.target + '.exr' );
  142. }
  143. function saveArrayBuffer( buffer, filename ) {
  144. const blob = new Blob( [ buffer ], { type: 'image/x-exr' } );
  145. const link = document.createElement( 'a' );
  146. link.href = URL.createObjectURL( blob );
  147. link.download = filename;
  148. link.click();
  149. }
  150. </script>
  151. </body>
  152. </html>