Spaces:
Sleeping
Sleeping
File size: 3,489 Bytes
645859a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
import * as THREE from 'three';
import { version } from '../helpers/constants.js';
class ConvolutionMaterial extends THREE.ShaderMaterial {
constructor(texelSize = new THREE.Vector2()) {
super({
uniforms: {
inputBuffer: new THREE.Uniform(null),
depthBuffer: new THREE.Uniform(null),
resolution: new THREE.Uniform(new THREE.Vector2()),
texelSize: new THREE.Uniform(new THREE.Vector2()),
halfTexelSize: new THREE.Uniform(new THREE.Vector2()),
kernel: new THREE.Uniform(0.0),
scale: new THREE.Uniform(1.0),
cameraNear: new THREE.Uniform(0.0),
cameraFar: new THREE.Uniform(1.0),
minDepthThreshold: new THREE.Uniform(0.0),
maxDepthThreshold: new THREE.Uniform(1.0),
depthScale: new THREE.Uniform(0.0),
depthToBlurRatioBias: new THREE.Uniform(0.25)
},
fragmentShader: `#include <common>
#include <dithering_pars_fragment>
uniform sampler2D inputBuffer;
uniform sampler2D depthBuffer;
uniform float cameraNear;
uniform float cameraFar;
uniform float minDepthThreshold;
uniform float maxDepthThreshold;
uniform float depthScale;
uniform float depthToBlurRatioBias;
varying vec2 vUv;
varying vec2 vUv0;
varying vec2 vUv1;
varying vec2 vUv2;
varying vec2 vUv3;
void main() {
float depthFactor = 0.0;
#ifdef USE_DEPTH
vec4 depth = texture2D(depthBuffer, vUv);
depthFactor = smoothstep(minDepthThreshold, maxDepthThreshold, 1.0-(depth.r * depth.a));
depthFactor *= depthScale;
depthFactor = max(0.0, min(1.0, depthFactor + 0.25));
#endif
vec4 sum = texture2D(inputBuffer, mix(vUv0, vUv, depthFactor));
sum += texture2D(inputBuffer, mix(vUv1, vUv, depthFactor));
sum += texture2D(inputBuffer, mix(vUv2, vUv, depthFactor));
sum += texture2D(inputBuffer, mix(vUv3, vUv, depthFactor));
gl_FragColor = sum * 0.25 ;
#include <dithering_fragment>
#include <tonemapping_fragment>
#include <${version >= 154 ? 'colorspace_fragment' : 'encodings_fragment'}>
}`,
vertexShader: `uniform vec2 texelSize;
uniform vec2 halfTexelSize;
uniform float kernel;
uniform float scale;
varying vec2 vUv;
varying vec2 vUv0;
varying vec2 vUv1;
varying vec2 vUv2;
varying vec2 vUv3;
void main() {
vec2 uv = position.xy * 0.5 + 0.5;
vUv = uv;
vec2 dUv = (texelSize * vec2(kernel) + halfTexelSize) * scale;
vUv0 = vec2(uv.x - dUv.x, uv.y + dUv.y);
vUv1 = vec2(uv.x + dUv.x, uv.y + dUv.y);
vUv2 = vec2(uv.x + dUv.x, uv.y - dUv.y);
vUv3 = vec2(uv.x - dUv.x, uv.y - dUv.y);
gl_Position = vec4(position.xy, 1.0, 1.0);
}`,
blending: THREE.NoBlending,
depthWrite: false,
depthTest: false
});
this.toneMapped = false;
this.setTexelSize(texelSize.x, texelSize.y);
this.kernel = new Float32Array([0.0, 1.0, 2.0, 2.0, 3.0]);
}
setTexelSize(x, y) {
this.uniforms.texelSize.value.set(x, y);
this.uniforms.halfTexelSize.value.set(x, y).multiplyScalar(0.5);
}
setResolution(resolution) {
this.uniforms.resolution.value.copy(resolution);
}
}
export { ConvolutionMaterial };
|