threejs-postprocessing-adobe-lut-to-png-converter.html 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>Three.js - postprocessing - 3DLUT w/loader</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. <style>
  8. body {
  9. min-height: 100vh;
  10. font-family: monospace;
  11. background: #222;
  12. color: white;
  13. }
  14. canvas {
  15. min-width: 512px;
  16. image-rendering: pixelated;
  17. }
  18. @media (max-width: 530px) {
  19. canvas {
  20. min-width: 0px;
  21. width: calc(100vw - 20px);
  22. }
  23. }
  24. </style>
  25. </head>
  26. <body>
  27. <h1>Adobe LUT to PNG converter</h1>
  28. <p>Drag and drop a LUT file here</p>
  29. <div>size:<input id="size" type="number" value="8" min="2" max="64"/></div>
  30. <p id="result"></p>
  31. <p><button type="button">Save...</button></p>
  32. <canvas id="c"></canvas>
  33. </body>
  34. <script src="resources/threejs/r98/three.js"></script>
  35. <script src="resources/threejs/r98/js/shaders/CopyShader.js"></script>
  36. <script src="resources/threejs/r98/js/postprocessing/EffectComposer.js"></script>
  37. <script src="resources/threejs/r98/js/postprocessing/RenderPass.js"></script>
  38. <script src="resources/threejs/r98/js/postprocessing/ShaderPass.js"></script>
  39. <script src="resources/lut-reader.js"></script>
  40. <script src="resources/drag-and-drop.js"></script>
  41. <script>
  42. 'use strict';
  43. /* global THREE, dragAndDrop, lutParser */
  44. function main() {
  45. const canvas = document.querySelector('#c');
  46. const renderer = new THREE.WebGLRenderer({canvas: canvas});
  47. const makeIdentityLutTexture = function() {
  48. const identityLUT = new Uint8Array([
  49. 0, 0, 0, 255, // black
  50. 255, 0, 0, 255, // red
  51. 0, 0, 255, 255, // blue
  52. 255, 0, 255, 255, // magenta
  53. 0, 255, 0, 255, // green
  54. 255, 255, 0, 255, // yellow
  55. 0, 255, 255, 255, // cyan
  56. 255, 255, 255, 255, // white
  57. ]);
  58. return function(filter) {
  59. const texture = new THREE.DataTexture(identityLUT, 4, 2, THREE.RGBAFormat);
  60. texture.minFilter = filter;
  61. texture.magFilter = filter;
  62. texture.needsUpdate = true;
  63. texture.flipY = false;
  64. return texture;
  65. };
  66. }();
  67. const sceneBG = new THREE.Scene();
  68. const cameraBG = new THREE.OrthographicCamera(-1, 1, 1, -1, -1, 1);
  69. const ctx = document.createElement('canvas').getContext('2d');
  70. function drawColorCubeImage(ctx, size) {
  71. const canvas = ctx.canvas;
  72. canvas.width = size * size;
  73. canvas.height = size;
  74. for (let zz = 0; zz < size; ++zz) {
  75. for (let yy = 0; yy < size; ++yy) {
  76. for (let xx = 0; xx < size; ++xx) {
  77. const r = Math.floor(xx / (size - 1) * 255);
  78. const g = Math.floor(yy / (size - 1) * 255);
  79. const b = Math.floor(zz / (size - 1) * 255);
  80. ctx.fillStyle = `rgb(${r},${g},${b})`;
  81. ctx.fillRect(zz * size + xx, yy, 1, 1);
  82. }
  83. }
  84. }
  85. }
  86. const idTexture = new THREE.CanvasTexture(ctx.canvas);
  87. idTexture.magFilter = THREE.NearestFilter;
  88. idTexture.minFilter = THREE.NearestFilter;
  89. {
  90. const planeGeo = new THREE.PlaneBufferGeometry(2, 2);
  91. const planeMat = new THREE.MeshBasicMaterial({
  92. map: idTexture,
  93. depthTest: false,
  94. });
  95. sceneBG.add(new THREE.Mesh(planeGeo, planeMat));
  96. }
  97. const lutShader = {
  98. uniforms: {
  99. tDiffuse: { value: null },
  100. lutMap: { value: null },
  101. lutMapSize: { value: 1, },
  102. },
  103. vertexShader: `
  104. varying vec2 vUv;
  105. void main() {
  106. vUv = uv;
  107. gl_Position = projectionMatrix * modelViewMatrix * vec4( position, 1.0 );
  108. }
  109. `,
  110. fragmentShader: `
  111. #include <common>
  112. #define FILTER_LUT true
  113. uniform sampler2D tDiffuse;
  114. uniform sampler2D lutMap;
  115. uniform float lutMapSize;
  116. varying vec2 vUv;
  117. vec4 sampleAs3DTexture(sampler2D tex, vec3 texCoord, float size) {
  118. float sliceSize = 1.0 / size; // space of 1 slice
  119. float slicePixelSize = sliceSize / size; // space of 1 pixel
  120. float width = size - 1.0;
  121. float sliceInnerSize = slicePixelSize * width; // space of size pixels
  122. float zSlice0 = floor( texCoord.z * width);
  123. float zSlice1 = min( zSlice0 + 1.0, width);
  124. float xOffset = slicePixelSize * 0.5 + texCoord.x * sliceInnerSize;
  125. float yRange = (texCoord.y * width + 0.5) / size;
  126. float s0 = xOffset + (zSlice0 * sliceSize);
  127. #ifdef FILTER_LUT
  128. float s1 = xOffset + (zSlice1 * sliceSize);
  129. vec4 slice0Color = texture2D(tex, vec2(s0, yRange));
  130. vec4 slice1Color = texture2D(tex, vec2(s1, yRange));
  131. float zOffset = mod(texCoord.z * width, 1.0);
  132. return mix(slice0Color, slice1Color, zOffset);
  133. #else
  134. return texture2D(tex, vec2( s0, yRange));
  135. #endif
  136. }
  137. void main() {
  138. vec4 originalColor = texture2D(tDiffuse, vUv);
  139. gl_FragColor = sampleAs3DTexture(lutMap, originalColor.xyz, lutMapSize);
  140. }
  141. `,
  142. };
  143. const effectLUT = new THREE.ShaderPass(lutShader);
  144. effectLUT.renderToScreen = true;
  145. const renderBG = new THREE.RenderPass(sceneBG, cameraBG);
  146. const rtParameters = {
  147. minFilter: THREE.NearestFilter,
  148. magFilter: THREE.NearestFilter,
  149. format: THREE.RGBFormat,
  150. };
  151. const composer = new THREE.EffectComposer(renderer, new THREE.WebGLRenderTarget(1, 1, rtParameters));
  152. composer.addPass(renderBG);
  153. composer.addPass(effectLUT);
  154. let name = 'identity';
  155. const lutTexture = makeIdentityLutTexture(THREE.LinearFilter);
  156. effectLUT.uniforms.lutMap.value = lutTexture;
  157. effectLUT.uniforms.lutMapSize.value = 2;
  158. const sizeElem = document.querySelector('#size');
  159. sizeElem.addEventListener('change', render);
  160. function render() {
  161. const size = parseInt(sizeElem.value);
  162. renderer.setSize(size * size, size, false);
  163. composer.setSize(size * size, size);
  164. drawColorCubeImage(ctx, size);
  165. idTexture.needsUpdate = true;
  166. composer.render(0);
  167. }
  168. render();
  169. dragAndDrop.setup({msg: 'Drop LUT File here'});
  170. dragAndDrop.onDropFile(readLUTFile);
  171. function readLUTFile(file) {
  172. const reader = new FileReader();
  173. reader.onload = (e) => {
  174. const lut = lutParser.lutTo2D3Drgb8(lutParser.parse(e.target.result));
  175. effectLUT.uniforms.lutMapSize.value = lut.size;
  176. lutTexture.image.data = lut.data;
  177. lutTexture.image.width = lut.size * lut.size;
  178. lutTexture.image.height = lut.size;
  179. lutTexture.format = THREE.RGBFormat;
  180. lutTexture.needsUpdate = true;
  181. render();
  182. name = `${file.name || lut.name || 'untitled'}`;
  183. document.querySelector('#result').textContent = `loaded: ${name}`;
  184. };
  185. reader.readAsText(file);
  186. }
  187. const saveData = (function() {
  188. const a = document.createElement('a');
  189. document.body.appendChild(a);
  190. a.style.display = 'none';
  191. return function saveData(blob, fileName) {
  192. const url = window.URL.createObjectURL(blob);
  193. a.href = url;
  194. a.download = fileName;
  195. a.click();
  196. };
  197. }());
  198. document.querySelector('button').addEventListener('click', () => {
  199. renderer.domElement.toBlob((blob) => {
  200. saveData(blob, `${name}-s${renderer.domElement.height}.png`);
  201. });
  202. });
  203. }
  204. main();
  205. </script>
  206. </html>