misc_exporter_exr.html 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220
  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. function init() {
  36. renderer = new THREE.WebGLRenderer( { antialias: true } );
  37. renderer.setPixelRatio( window.devicePixelRatio );
  38. renderer.setSize( window.innerWidth, window.innerHeight );
  39. renderer.setAnimationLoop( animate );
  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. controls.update();
  78. renderer.render( scene, camera );
  79. }
  80. function createDataTexture() {
  81. const normal = new THREE.Vector3();
  82. const coord = new THREE.Vector2();
  83. const size = 800, radius = 320, factor = Math.PI * 0.5 / radius;
  84. const data = new Float32Array( 4 * size * size );
  85. for ( let i = 0; i < size; i ++ ) {
  86. for ( let j = 0; j < size; j ++ ) {
  87. const idx = i * size * 4 + j * 4;
  88. coord.set( j, i ).subScalar( size / 2 );
  89. if ( coord.length() < radius )
  90. normal.set(
  91. Math.sin( coord.x * factor ),
  92. Math.sin( coord.y * factor ),
  93. Math.cos( coord.x * factor )
  94. );
  95. else
  96. normal.set( 0, 0, 1 );
  97. data[ idx + 0 ] = .5 + .5 * normal.x;
  98. data[ idx + 1 ] = .5 + .5 * normal.y;
  99. data[ idx + 2 ] = .5 + .5 * normal.z;
  100. data[ idx + 3 ] = 1.;
  101. }
  102. }
  103. dataTexture = new THREE.DataTexture( data, size, size, THREE.RGBAFormat, THREE.FloatType );
  104. dataTexture.needsUpdate = true;
  105. const material = new THREE.MeshBasicMaterial( { map: dataTexture } );
  106. const quad = new THREE.PlaneGeometry( 50, 50 );
  107. mesh = new THREE.Mesh( quad, material );
  108. mesh.visible = false;
  109. scene.add( mesh );
  110. }
  111. function swapScene() {
  112. if ( params.target == 'pmrem' ) {
  113. camera.position.set( 10, 0, 0 );
  114. controls.enabled = true;
  115. scene.background = renderTarget.texture;
  116. mesh.visible = false;
  117. } else {
  118. camera.position.set( 0, 0, 70 );
  119. controls.enabled = false;
  120. scene.background = new THREE.Color( 0, 0, 0 );
  121. mesh.visible = true;
  122. }
  123. }
  124. async function exportFile() {
  125. let result, exportType, exportCompression;
  126. if ( params.type == 'HalfFloatType' )
  127. exportType = THREE.HalfFloatType;
  128. else
  129. exportType = THREE.FloatType;
  130. if ( params.compression == 'ZIP' )
  131. exportCompression = ZIP_COMPRESSION;
  132. else if ( params.compression == 'ZIPS' )
  133. exportCompression = ZIPS_COMPRESSION;
  134. else
  135. exportCompression = NO_COMPRESSION;
  136. if ( params.target == 'pmrem' )
  137. result = await exporter.parse( renderer, renderTarget, { type: exportType, compression: exportCompression } );
  138. else
  139. result = await exporter.parse( dataTexture, { type: exportType, compression: exportCompression } );
  140. saveArrayBuffer( result, params.target + '.exr' );
  141. }
  142. function saveArrayBuffer( buffer, filename ) {
  143. const blob = new Blob( [ buffer ], { type: 'image/x-exr' } );
  144. const link = document.createElement( 'a' );
  145. link.href = URL.createObjectURL( blob );
  146. link.download = filename;
  147. link.click();
  148. }
  149. </script>
  150. </body>
  151. </html>