50 lines
No EOL
1.3 KiB
HLSL
50 lines
No EOL
1.3 KiB
HLSL
#include "../sampler.hlsl"
|
|
#include "../math/sum.hlsl"
|
|
|
|
/*
|
|
contributors: nan
|
|
description: |
|
|
morphological dilation operation. Based on:
|
|
https://lettier.github.io/3d-game-shaders-for-beginners/dilation.html
|
|
https://www.shadertoy.com/view/WsyXWc
|
|
use: dilation(<SAMPLER_TYPE> texture, <float2> st, <float2> pixels_scale, <int> passes)
|
|
options:
|
|
- SAMPLER_FNC(TEX, UV)
|
|
- DILATION_TYPE
|
|
- DILATION_SAMPLE_FNC(TEX, UV)
|
|
*/
|
|
|
|
#ifndef DILATION_TYPE
|
|
#define DILATION_TYPE float
|
|
#endif
|
|
|
|
#ifndef DILATION_SAMPLE_FNC
|
|
#define DILATION_SAMPLE_FNC(TEX, UV) SAMPLER_FNC(TEX, UV).x
|
|
#endif
|
|
|
|
#ifndef FNC_DILATE
|
|
#define FNC_DILATE
|
|
|
|
DILATION_TYPE dilation(SAMPLER_TYPE tex,float2 st,float2 pixel, int radius) {
|
|
|
|
float invKR = 1.0 / float(radius);
|
|
DILATION_TYPE acc = DILATION_TYPE(0.0);
|
|
float w = 0.0;
|
|
for(int i = -radius; i <= radius; ++i)
|
|
for(int j = -radius; j <= radius; ++j) {
|
|
float2 rxy = float2(float(i), float(j));
|
|
float2 kst = rxy * invKR;
|
|
float2 texOffset = st + rxy * pixel;
|
|
float kernel = saturate(1.0 - dot(kst, kst));
|
|
DILATION_TYPE t = DILATION_SAMPLE_FNC(tex, texOffset);
|
|
DILATION_TYPE v = t + kernel;
|
|
if (sum(v) > sum(acc)) {
|
|
acc = v;
|
|
w = kernel;
|
|
}
|
|
}
|
|
|
|
return acc - w;
|
|
}
|
|
|
|
#endif |