You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

416 lines
14 KiB

2 years ago
<!DOCTYPE html>
<html lang="en">
<head>
<title>Three.js - postprocessing - 3DLUT w/loader</title>
<meta charset="utf-8">
<meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
<style>
html, body {
margin: 0;
height: 100%;
}
#c {
width: 100%;
height: 100%;
display: block;
}
</style>
</head>
<body>
<canvas id="c"></canvas>
</body>
<script type="importmap">
{
"imports": {
"three": "../../build/three.module.js",
"three/addons/": "../../examples/jsm/"
}
}
</script>
<script type="module">
import * as THREE from 'three';
import * as lutParser from './resources/lut-reader.js';
import * as dragAndDrop from './resources/drag-and-drop.js';
import {OrbitControls} from 'three/addons/controls/OrbitControls.js';
import {GLTFLoader} from 'three/addons/loaders/GLTFLoader.js';
import {EffectComposer} from 'three/addons/postprocessing/EffectComposer.js';
import {RenderPass} from 'three/addons/postprocessing/RenderPass.js';
import {ShaderPass} from 'three/addons/postprocessing/ShaderPass.js';
import {GammaCorrectionShader} from 'three/addons/shaders/GammaCorrectionShader.js';
import {GUI} from 'three/addons/libs/lil-gui.module.min.js';
function main() {
const canvas = document.querySelector('#c');
const renderer = new THREE.WebGLRenderer({canvas});
const fov = 45;
const aspect = 2; // the canvas default
const near = 0.1;
const far = 100;
const camera = new THREE.PerspectiveCamera(fov, aspect, near, far);
camera.position.set(0, 10, 20);
const controls = new OrbitControls(camera, canvas);
controls.target.set(0, 5, 0);
controls.update();
const lutTextures = [
{ name: 'identity', size: 2, filter: true , },
{ name: 'identity no filter', size: 2, filter: false, },
{ name: 'custom', url: 'resources/images/lut/3dlut-red-only-s16.png' },
{ name: 'monochrome', url: 'resources/images/lut/monochrome-s8.png' },
{ name: 'sepia', url: 'resources/images/lut/sepia-s8.png' },
{ name: 'saturated', url: 'resources/images/lut/saturated-s8.png', },
{ name: 'posterize', url: 'resources/images/lut/posterize-s8n.png', },
{ name: 'posterize-3-rgb', url: 'resources/images/lut/posterize-3-rgb-s8n.png', },
{ name: 'posterize-3-lab', url: 'resources/images/lut/posterize-3-lab-s8n.png', },
{ name: 'posterize-4-lab', url: 'resources/images/lut/posterize-4-lab-s8n.png', },
{ name: 'posterize-more', url: 'resources/images/lut/posterize-more-s8n.png', },
{ name: 'inverse', url: 'resources/images/lut/inverse-s8.png', },
{ name: 'color negative', url: 'resources/images/lut/color-negative-s8.png', },
{ name: 'high contrast', url: 'resources/images/lut/high-contrast-bw-s8.png', },
{ name: 'funky contrast', url: 'resources/images/lut/funky-contrast-s8.png', },
{ name: 'nightvision', url: 'resources/images/lut/nightvision-s8.png', },
{ name: 'thermal', url: 'resources/images/lut/thermal-s8.png', },
{ name: 'b/w', url: 'resources/images/lut/black-white-s8n.png', },
{ name: 'hue +60', url: 'resources/images/lut/hue-plus-60-s8.png', },
{ name: 'hue +180', url: 'resources/images/lut/hue-plus-180-s8.png', },
{ name: 'hue -60', url: 'resources/images/lut/hue-minus-60-s8.png', },
{ name: 'red to cyan', url: 'resources/images/lut/red-to-cyan-s8.png' },
{ name: 'blues', url: 'resources/images/lut/blues-s8.png' },
{ name: 'infrared', url: 'resources/images/lut/infrared-s8.png' },
{ name: 'radioactive', url: 'resources/images/lut/radioactive-s8.png' },
{ name: 'goolgey', url: 'resources/images/lut/googley-s8.png' },
{ name: 'bgy', url: 'resources/images/lut/bgy-s8.png' },
];
const makeIdentityLutTexture = function() {
const identityLUT = new Uint8Array([
0, 0, 0, 255, // black
255, 0, 0, 255, // red
0, 0, 255, 255, // blue
255, 0, 255, 255, // magenta
0, 255, 0, 255, // green
255, 255, 0, 255, // yellow
0, 255, 255, 255, // cyan
255, 255, 255, 255, // white
]);
return function(filter) {
const texture = new THREE.DataTexture(identityLUT, 4, 2);
texture.minFilter = texture.magFilter = filter ? THREE.LinearFilter : THREE.NearestFilter;
texture.needsUpdate = true;
texture.flipY = false;
return texture;
};
}();
const makeLUTTexture = function() {
const imgLoader = new THREE.ImageLoader();
const ctx = document.createElement('canvas').getContext('2d');
return function(info) {
const lutSize = info.size;
const width = lutSize * lutSize;
const height = lutSize;
const texture = new THREE.DataTexture(new Uint8Array(width * height), width, height);
texture.minFilter = texture.magFilter = info.filter ? THREE.LinearFilter : THREE.NearestFilter;
texture.flipY = false;
if (info.url) {
imgLoader.load(info.url, function(image) {
ctx.canvas.width = width;
ctx.canvas.height = height;
ctx.drawImage(image, 0, 0);
const imageData = ctx.getImageData(0, 0, width, height);
texture.image.data = new Uint8Array(imageData.data.buffer);
texture.image.width = width;
texture.image.height = height;
texture.needsUpdate = true;
});
}
return texture;
};
}();
lutTextures.forEach((info) => {
// if not size set get it from the filename
if (!info.size) {
// assumes filename ends in '-s<num>[n]'
// where <num> is the size of the 3DLUT cube
// and [n] means 'no filtering' or 'nearest'
//
// examples:
// 'foo-s16.png' = size:16, filter: true
// 'bar-s8n.png' = size:8, filter: false
const m = /-s(\d+)(n*)\.[^.]+$/.exec(info.url);
if (m) {
info.size = parseInt(m[1]);
info.filter = info.filter === undefined ? m[2] !== 'n' : info.filter;
}
info.texture = makeLUTTexture(info);
} else {
info.texture = makeIdentityLutTexture(info.filter);
}
});
const lutNameIndexMap = {};
lutTextures.forEach((info, ndx) => {
lutNameIndexMap[info.name] = ndx;
});
const lutSettings = {
lut: lutNameIndexMap.custom,
};
const gui = new GUI({ width: 300 });
gui.add(lutSettings, 'lut', lutNameIndexMap);
const scene = new THREE.Scene();
const sceneBG = new THREE.Scene();
const cameraBG = new THREE.OrthographicCamera(-1, 1, 1, -1, -1, 1);
let bgMesh;
let bgTexture;
{
const loader = new THREE.TextureLoader();
bgTexture = loader.load('resources/images/beach.jpg');
bgTexture.encoding = THREE.sRGBEncoding;
const planeGeo = new THREE.PlaneGeometry(2, 2);
const planeMat = new THREE.MeshBasicMaterial({
map: bgTexture,
depthTest: false,
});
bgMesh = new THREE.Mesh(planeGeo, planeMat);
sceneBG.add(bgMesh);
}
function frameArea(sizeToFitOnScreen, boxSize, boxCenter, camera) {
const halfSizeToFitOnScreen = sizeToFitOnScreen * 0.5;
const halfFovY = THREE.MathUtils.degToRad(camera.fov * .5);
const distance = halfSizeToFitOnScreen / Math.tan(halfFovY);
// compute a unit vector that points in the direction the camera is now
// in the xz plane from the center of the box
const direction = (new THREE.Vector3())
.subVectors(camera.position, boxCenter)
.multiply(new THREE.Vector3(1, 0, 1))
.normalize();
// move the camera to a position distance units way from the center
// in whatever direction the camera was from the center already
camera.position.copy(direction.multiplyScalar(distance).add(boxCenter));
// pick some near and far values for the frustum that
// will contain the box.
camera.near = boxSize / 100;
camera.far = boxSize * 100;
camera.updateProjectionMatrix();
// point the camera to look at the center of the box
camera.lookAt(boxCenter.x, boxCenter.y, boxCenter.z);
}
{
const gltfLoader = new GLTFLoader();
gltfLoader.load('resources/models/3dbustchallange_submission/scene.gltf', (gltf) => {
const root = gltf.scene;
scene.add(root);
// fix materials from r114
root.traverse(({material}) => {
if (material) {
material.depthWrite = true;
}
});
root.updateMatrixWorld();
// compute the box that contains all the stuff
// from root and below
const box = new THREE.Box3().setFromObject(root);
const boxSize = box.getSize(new THREE.Vector3()).length();
const boxCenter = box.getCenter(new THREE.Vector3());
frameArea(boxSize * 0.4, boxSize, boxCenter, camera);
// update the Trackball controls to handle the new size
controls.maxDistance = boxSize * 10;
controls.target.copy(boxCenter);
controls.update();
});
}
const lutShader = {
uniforms: {
tDiffuse: { value: null },
lutMap: { value: null },
lutMapSize: { value: 1, },
},
vertexShader: `
varying vec2 vUv;
void main() {
vUv = uv;
gl_Position = projectionMatrix * modelViewMatrix * vec4( position, 1.0 );
}
`,
fragmentShader: `
#include <common>
#define FILTER_LUT true
uniform sampler2D tDiffuse;
uniform sampler2D lutMap;
uniform float lutMapSize;
varying vec2 vUv;
vec4 sampleAs3DTexture(sampler2D tex, vec3 texCoord, float size) {
float sliceSize = 1.0 / size; // space of 1 slice
float slicePixelSize = sliceSize / size; // space of 1 pixel
float width = size - 1.0;
float sliceInnerSize = slicePixelSize * width; // space of size pixels
float zSlice0 = floor( texCoord.z * width);
float zSlice1 = min( zSlice0 + 1.0, width);
float xOffset = slicePixelSize * 0.5 + texCoord.x * sliceInnerSize;
float yRange = (texCoord.y * width + 0.5) / size;
float s0 = xOffset + (zSlice0 * sliceSize);
#ifdef FILTER_LUT
float s1 = xOffset + (zSlice1 * sliceSize);
vec4 slice0Color = texture2D(tex, vec2(s0, yRange));
vec4 slice1Color = texture2D(tex, vec2(s1, yRange));
float zOffset = mod(texCoord.z * width, 1.0);
return mix(slice0Color, slice1Color, zOffset);
#else
return texture2D(tex, vec2( s0, yRange));
#endif
}
void main() {
vec4 originalColor = texture2D(tDiffuse, vUv);
gl_FragColor = sampleAs3DTexture(lutMap, originalColor.xyz, lutMapSize);
}
`,
};
const lutNearestShader = {
uniforms: {...lutShader.uniforms},
vertexShader: lutShader.vertexShader,
fragmentShader: lutShader.fragmentShader.replace('#define FILTER_LUT', '//'),
};
const effectLUT = new ShaderPass(lutShader);
const effectLUTNearest = new ShaderPass(lutNearestShader);
const renderModel = new RenderPass(scene, camera);
renderModel.clear = false; // so we don't clear out the background
const renderBG = new RenderPass(sceneBG, cameraBG);
const gammaPass = new ShaderPass(GammaCorrectionShader);
const composer = new EffectComposer(renderer);
composer.addPass(renderBG);
composer.addPass(renderModel);
composer.addPass(effectLUT);
composer.addPass(effectLUTNearest);
composer.addPass(gammaPass);
function resizeRendererToDisplaySize(renderer) {
const canvas = renderer.domElement;
const width = canvas.clientWidth * window.devicePixelRatio | 0;
const height = canvas.clientHeight * window.devicePixelRatio | 0;
const needResize = canvas.width !== width || canvas.height !== height;
if (needResize) {
renderer.setSize(width, height, false);
}
return needResize;
}
let then = 0;
function render(now) {
now *= 0.001; // convert to seconds
const delta = now - then;
then = now;
if (resizeRendererToDisplaySize(renderer)) {
const canvas = renderer.domElement;
const canvasAspect = canvas.clientWidth / canvas.clientHeight;
camera.aspect = canvasAspect;
camera.updateProjectionMatrix();
composer.setSize(canvas.width, canvas.height);
// scale the background plane to keep the image's
// aspect correct.
// Note the image may not have loaded yet.
const imageAspect = bgTexture.image ? bgTexture.image.width / bgTexture.image.height : 1;
const aspect = imageAspect / canvasAspect;
bgMesh.scale.x = aspect > 1 ? aspect : 1;
bgMesh.scale.y = aspect > 1 ? 1 : 1 / aspect;
}
const lutInfo = lutTextures[ lutSettings.lut ];
const effect = lutInfo.filter ? effectLUT : effectLUTNearest;
effectLUT.enabled = lutInfo.filter;
effectLUTNearest.enabled = !lutInfo.filter;
const lutTexture = lutInfo.texture;
effect.uniforms.lutMap.value = lutTexture;
effect.uniforms.lutMapSize.value = lutInfo.size;
composer.render(delta);
requestAnimationFrame(render);
}
requestAnimationFrame(render);
dragAndDrop.setup({msg: 'Drop LUT File here'});
dragAndDrop.onDropFile(readLUTFile);
function ext(s) {
const period = s.lastIndexOf('.');
return s.slice(period + 1);
}
function readLUTFile(file) {
const reader = new FileReader();
reader.onload = (e) => {
const type = ext(file.name);
const lut = lutParser.lutTo2D3Drgba8(lutParser.parse(e.target.result, type));
const {size, data, name} = lut;
const texture = new THREE.DataTexture(data, size * size, size);
texture.magFilter = THREE.LinearFilter;
texture.minFilter = THREE.LinearFilter;
texture.needsUpdate = true;
texture.flipY = false;
const lutTexture = {
name: (name && name.toLowerCase().trim() !== 'untitled')
? name
: file.name,
size: size,
filter: true,
texture,
};
lutTextures.push(lutTexture);
lutSettings.lut = lutTextures.length - 1;
};
reader.readAsText(file);
}
}
main();
</script>
</html>