forked from BradLarson/GPUImage2
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request BradLarson#30 from Hyun-je/master
Add Median Filter
- Loading branch information
Showing
3 changed files
with
126 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,109 @@ | ||
| #include <metal_stdlib> | ||
| #include "TexelSamplingTypes.h" | ||
|
|
||
| using namespace metal; | ||
|
|
||
| #define s2(a, b) temp = a; a = min(a, b); b = max(temp, b); | ||
| #define mn3(a, b, c) s2(a, b); s2(a, c); | ||
| #define mx3(a, b, c) s2(b, c); s2(a, c); | ||
|
|
||
| #define mnmx3(a, b, c) mx3(a, b, c); s2(a, b); // 3 exchanges | ||
| #define mnmx4(a, b, c, d) s2(a, b); s2(c, d); s2(a, c); s2(b, d); // 4 exchanges | ||
| #define mnmx5(a, b, c, d, e) s2(a, b); s2(c, d); mn3(a, c, e); mx3(b, d, e); // 6 exchanges | ||
| #define mnmx6(a, b, c, d, e, f) s2(a, d); s2(b, e); s2(c, f); mn3(a, b, c); mx3(d, e, f); // 7 exchanges | ||
|
|
||
|
|
||
|
|
||
| fragment half4 medianFilter(NearbyTexelVertexIO fragmentInput [[stage_in]], | ||
| texture2d<half> inputTexture [[texture(0)]]) { | ||
|
|
||
| constexpr sampler quadSampler(coord::pixel); | ||
|
|
||
| half3 v[6]; | ||
|
|
||
| v[0] = inputTexture.sample(quadSampler, fragmentInput.bottomLeftTextureCoordinate).rgb; | ||
| v[1] = inputTexture.sample(quadSampler, fragmentInput.topRightTextureCoordinate).rgb; | ||
| v[2] = inputTexture.sample(quadSampler, fragmentInput.topLeftTextureCoordinate).rgb; | ||
| v[3] = inputTexture.sample(quadSampler, fragmentInput.bottomRightTextureCoordinate).rgb; | ||
| v[4] = inputTexture.sample(quadSampler, fragmentInput.leftTextureCoordinate).rgb; | ||
| v[5] = inputTexture.sample(quadSampler, fragmentInput.rightTextureCoordinate).rgb; | ||
|
|
||
|
|
||
| half3 temp; | ||
|
|
||
| mnmx6(v[0], v[1], v[2], v[3], v[4], v[5]); | ||
|
|
||
| v[5] = inputTexture.sample(quadSampler, fragmentInput.bottomTextureCoordinate).rgb; | ||
|
|
||
| mnmx5(v[1], v[2], v[3], v[4], v[5]); | ||
|
|
||
| v[5] = inputTexture.sample(quadSampler, fragmentInput.topTextureCoordinate).rgb; | ||
|
|
||
| mnmx4(v[2], v[3], v[4], v[5]); | ||
|
|
||
| v[5] = inputTexture.sample(quadSampler, fragmentInput.textureCoordinate).rgb; | ||
|
|
||
| mnmx3(v[3], v[4], v[5]); | ||
|
|
||
| return half4(v[4], 1.0); | ||
|
|
||
| } | ||
|
|
||
|
|
||
| /* | ||
| varying vec2 textureCoordinate; | ||
| varying vec2 leftTextureCoordinate; | ||
| varying vec2 rightTextureCoordinate; | ||
| varying vec2 topTextureCoordinate; | ||
| varying vec2 topLeftTextureCoordinate; | ||
| varying vec2 topRightTextureCoordinate; | ||
| varying vec2 bottomTextureCoordinate; | ||
| varying vec2 bottomLeftTextureCoordinate; | ||
| varying vec2 bottomRightTextureCoordinate; | ||
| uniform sampler2D inputImageTexture; | ||
| #define s2(a, b) temp = a; a = min(a, b); b = max(temp, b); | ||
| #define mn3(a, b, c) s2(a, b); s2(a, c); | ||
| #define mx3(a, b, c) s2(b, c); s2(a, c); | ||
| #define mnmx3(a, b, c) mx3(a, b, c); s2(a, b); // 3 exchanges | ||
| #define mnmx4(a, b, c, d) s2(a, b); s2(c, d); s2(a, c); s2(b, d); // 4 exchanges | ||
| #define mnmx5(a, b, c, d, e) s2(a, b); s2(c, d); mn3(a, c, e); mx3(b, d, e); // 6 exchanges | ||
| #define mnmx6(a, b, c, d, e, f) s2(a, d); s2(b, e); s2(c, f); mn3(a, b, c); mx3(d, e, f); // 7 exchanges | ||
| void main() | ||
| { | ||
| vec3 v[6]; | ||
| v[0] = texture2D(inputImageTexture, bottomLeftTextureCoordinate).rgb; | ||
| v[1] = texture2D(inputImageTexture, topRightTextureCoordinate).rgb; | ||
| v[2] = texture2D(inputImageTexture, topLeftTextureCoordinate).rgb; | ||
| v[3] = texture2D(inputImageTexture, bottomRightTextureCoordinate).rgb; | ||
| v[4] = texture2D(inputImageTexture, leftTextureCoordinate).rgb; | ||
| v[5] = texture2D(inputImageTexture, rightTextureCoordinate).rgb; | ||
| // v[6] = texture2D(inputImageTexture, bottomTextureCoordinate).rgb; | ||
| // v[7] = texture2D(inputImageTexture, topTextureCoordinate).rgb; | ||
| vec3 temp; | ||
| mnmx6(v[0], v[1], v[2], v[3], v[4], v[5]); | ||
| v[5] = texture2D(inputImageTexture, bottomTextureCoordinate).rgb; | ||
| mnmx5(v[1], v[2], v[3], v[4], v[5]); | ||
| v[5] = texture2D(inputImageTexture, topTextureCoordinate).rgb; | ||
| mnmx4(v[2], v[3], v[4], v[5]); | ||
| v[5] = texture2D(inputImageTexture, textureCoordinate).rgb; | ||
| mnmx3(v[3], v[4], v[5]); | ||
| gl_FragColor = vec4(v[4], 1.0); | ||
| } | ||
| */ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,5 @@ | ||
| public class MedianFilter: TextureSamplingOperation { | ||
| public init() { | ||
| super.init(fragmentFunctionName:"medianFilter") | ||
| } | ||
| } |