3d-tiles-renderer
Version:
https://github.com/AnalyticalGraphicsInc/3d-tiles/tree/master/specification
144 lines (95 loc) • 2.91 kB
JavaScript
const OVERLAY_PARAMS = Symbol( 'OVERLAY_PARAMS' );
// before compile can be used to chain shader adjustments. Returns the added uniforms used for fading.
export function wrapOverlaysMaterial( material, previousOnBeforeCompile ) {
// if the material has already been wrapped then return the params
if ( material[ OVERLAY_PARAMS ] ) {
return material[ OVERLAY_PARAMS ];
}
const params = {
layerMaps: { value: [] },
layerColor: { value: [] },
};
material[ OVERLAY_PARAMS ] = params;
material.defines = {
...( material.defines || {} ),
LAYER_COUNT: 0,
};
material.onBeforeCompile = shader => {
if ( previousOnBeforeCompile ) {
previousOnBeforeCompile( shader );
}
shader.uniforms = {
...shader.uniforms,
...params,
};
shader.vertexShader = shader
.vertexShader
.replace( /void main\(\s*\)\s*{/, value => /* glsl */`
for ( int i = 0; i < 10; i ++ ) {
attribute vec3 layer_uv_UNROLLED_LOOP_INDEX;
varying vec3 v_layer_uv_UNROLLED_LOOP_INDEX;
}
${ value }
for ( int i = 0; i < 10; i ++ ) {
v_layer_uv_UNROLLED_LOOP_INDEX = layer_uv_UNROLLED_LOOP_INDEX;
}
` );
shader.fragmentShader = shader
.fragmentShader
.replace( /void main\(/, value => /* glsl */`
struct LayerTint {
vec3 color;
float opacity;
};
uniform sampler2D layerMaps[ LAYER_COUNT ];
uniform LayerTint layerColor[ LAYER_COUNT ];
for ( int i = 0; i < 10; i ++ ) {
varying vec3 v_layer_uv_UNROLLED_LOOP_INDEX;
}
${ value }
` )
.replace( /
${ value }
{
vec4 tint;
vec3 layerUV;
float layerOpacity;
float wOpacity;
float wDelta;
for ( int i = 0; i < 10; i ++ ) {
layerUV = v_layer_uv_UNROLLED_LOOP_INDEX;
tint = texture( layerMaps[ i ], layerUV.xy );
// discard texture outside 0, 1 on w
wDelta = fwidth( layerUV.z );
wOpacity = smoothstep( - wDelta, 0.0, layerUV.z ) * smoothstep( 1.0 + wDelta, 1.0, layerUV.z );
// apply tint & opacity
tint.rgb *= layerColor[ i ].color;
tint.rgba *= layerColor[ i ].opacity * wOpacity;
// premultiplied alpha equation
diffuseColor = tint + diffuseColor * ( 1.0 - tint.a );
}
}
` );
};
return params;
}