File size: 3,278 Bytes
bf237c2 |
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 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
import * as THREE from 'three';
import { version } from '../helpers/constants.js';
class SpotLightMaterial extends THREE.ShaderMaterial {
constructor() {
super({
uniforms: {
depth: {
value: null
},
opacity: {
value: 1
},
attenuation: {
value: 2.5
},
anglePower: {
value: 12
},
spotPosition: {
value: new THREE.Vector3(0, 0, 0)
},
lightColor: {
value: new THREE.Color('white')
},
cameraNear: {
value: 0
},
cameraFar: {
value: 1
},
resolution: {
value: new THREE.Vector2(0, 0)
}
},
transparent: true,
depthWrite: false,
vertexShader: /* glsl */`
varying vec3 vNormal;
varying float vViewZ;
varying float vIntensity;
uniform vec3 spotPosition;
uniform float attenuation;
#include <common>
#include <logdepthbuf_pars_vertex>
void main() {
// compute intensity
vNormal = normalize(normalMatrix * normal);
vec4 worldPosition = modelMatrix * vec4(position, 1);
vec4 viewPosition = viewMatrix * worldPosition;
vViewZ = viewPosition.z;
vIntensity = 1.0 - saturate(distance(worldPosition.xyz, spotPosition) / attenuation);
gl_Position = projectionMatrix * viewPosition;
#include <logdepthbuf_vertex>
}
`,
fragmentShader: /* glsl */`
varying vec3 vNormal;
varying float vViewZ;
varying float vIntensity;
uniform vec3 lightColor;
uniform float anglePower;
uniform sampler2D depth;
uniform vec2 resolution;
uniform float cameraNear;
uniform float cameraFar;
uniform float opacity;
#include <packing>
#include <logdepthbuf_pars_fragment>
float readDepth(sampler2D depthSampler, vec2 uv) {
float fragCoordZ = texture(depthSampler, uv).r;
// https://github.com/mrdoob/three.js/issues/23072
#ifdef USE_LOGDEPTHBUF
float viewZ = 1.0 - exp2(fragCoordZ * log(cameraFar + 1.0) / log(2.0));
#else
float viewZ = perspectiveDepthToViewZ(fragCoordZ, cameraNear, cameraFar);
#endif
return viewZ;
}
void main() {
#include <logdepthbuf_fragment>
vec3 normal = vec3(vNormal.x, vNormal.y, abs(vNormal.z));
float angleIntensity = pow(dot(normal, vec3(0, 0, 1)), anglePower);
float intensity = vIntensity * angleIntensity;
// fades when z is close to sampled depth, meaning the cone is intersecting existing geometry
bool isSoft = resolution[0] > 0.0 && resolution[1] > 0.0;
if (isSoft) {
vec2 uv = gl_FragCoord.xy / resolution;
intensity *= smoothstep(0.0, 1.0, vViewZ - readDepth(depth, uv));
}
gl_FragColor = vec4(lightColor, intensity * opacity);
#include <tonemapping_fragment>
#include <${version >= 154 ? 'colorspace_fragment' : 'encodings_fragment'}>
}
`
});
}
}
export { SpotLightMaterial };
|