@luma.gl/effects/dist/passes/postprocessing/image-adjust-filters/denoise.js

Post-processing effects for luma.gl

// luma.gl
// SPDX-License-Identifier: MIT
// Copyright (c) vis.gl contributors
// Do a 9x9 bilateral box filter
const source = /* wgsl */ `\

struct denoiseUniforms {
  strength: f32
};

@group(0), @binding(1) var<uniform> denoise: denoiseUniforms;

fn denoise_sampleColor(source: sampler2D, texSize: vec2<f32>, texCoord: vec2<f32>) -> vec4<f32> {
	let adjustedExponent: f32 = 3. + 200. * pow(1. - denoise.strength, 4.);
	let center: vec4<f32> = sample_texture(BUFFER_source, texCoord);
	var color: vec4<f32> = vec4<f32>(0.);
	var total: f32 = 0.;

	for (var x: f32 = -4.; x <= 4.; x = x + (1.)) {

		for (var y: f32 = -4.; y <= 4.; y = y + (1.)) {
			let offsetColor: vec4<f32> = sample_texture(BUFFER_source, texCoord + vec2<f32>(x, y) / texSize);
			var weight: f32 = 1. - abs(dot(offsetColor.rgb - center.rgb, vec3<f32>(0.25)));
			weight = pow(weight, adjustedExponent);
			color = color + (offsetColor * weight);
			total = total + (weight);
		}

	}

	return color / total;
} 
`;
const fs = /* glsl */ `\
uniform dedenoiseUniforms {
  float strength;
} denoise;

vec4 dedenoise_sampleColor(sampler2D source, vec2 texSize, vec2 texCoord) {
  float adjustedExponent = 3. + 200. * pow(1. - noise.strength, 4.);

  vec4 center = texture(source, texCoord);
  vec4 color = vec4(0.0);
  float total = 0.0;
  for (float x = -4.0; x <= 4.0; x += 1.0) {
    for (float y = -4.0; y <= 4.0; y += 1.0) {
      vec4 offsetColor = texture(source, texCoord + vec2(x, y) / texSize);
      float weight = 1.0 - abs(dot(offsetColor.rgb - center.rgb, vec3(0.25)));
      weight = pow(weight, adjustedExponent);
      color += offsetColor * weight;
      total += weight;
    }
  }

  return color / total;
}
`;
/**
 * Denoise -
 * Smooths over grainy noise in dark images using an 9x9 box filter
 * weighted by color intensity, similar to a bilateral filter.
 */
export const denoise = {
    props: {},
    uniforms: {},
    name: 'denoise',
    uniformTypes: {
        strength: 'f32'
    },
    propTypes: {
        strength: { format: 'f32', value: 0.5, min: 0, max: 1 }
        // strength: {..., adjust: (strength: number): number => 0.53 + 200 * Math.pow(1 - strength, 4) // TODO - JS preprocessing
    },
    source,
    fs,
    passes: [{ sampler: true }, { sampler: true }]
};
//# sourceMappingURL=denoise.js.map