mirror of
https://git.videolan.org/git/ffmpeg.git
synced 2024-08-02 17:29:58 +02:00
7faa8d8b03
Intended to replace existing opencl mode of the unsharp filter. Supports many more pixel formats and works without immediate upload and download of frame data. The options are compatible with the existing filter.
100 lines
3.3 KiB
Common Lisp
100 lines
3.3 KiB
Common Lisp
/*
|
|
* This file is part of FFmpeg.
|
|
*
|
|
* FFmpeg is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with FFmpeg; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
__kernel void unsharp_global(__write_only image2d_t dst,
|
|
__read_only image2d_t src,
|
|
int size_x,
|
|
int size_y,
|
|
float amount,
|
|
__constant float *coef_matrix)
|
|
{
|
|
const sampler_t sampler = (CLK_NORMALIZED_COORDS_FALSE |
|
|
CLK_FILTER_NEAREST);
|
|
int2 loc = (int2)(get_global_id(0), get_global_id(1));
|
|
int2 centre = (int2)(size_x / 2, size_y / 2);
|
|
|
|
float4 val = read_imagef(src, sampler, loc);
|
|
float4 sum = 0.0f;
|
|
int x, y;
|
|
|
|
for (y = 0; y < size_y; y++) {
|
|
for (x = 0; x < size_x; x++) {
|
|
int2 pos = loc + (int2)(x, y) - centre;
|
|
sum += coef_matrix[y * size_x + x] *
|
|
read_imagef(src, sampler, pos);
|
|
}
|
|
}
|
|
|
|
write_imagef(dst, loc, val + (val - sum) * amount);
|
|
}
|
|
|
|
__kernel void unsharp_local(__write_only image2d_t dst,
|
|
__read_only image2d_t src,
|
|
int size_x,
|
|
int size_y,
|
|
float amount,
|
|
__constant float *coef_x,
|
|
__constant float *coef_y)
|
|
{
|
|
const sampler_t sampler = (CLK_NORMALIZED_COORDS_FALSE |
|
|
CLK_ADDRESS_CLAMP_TO_EDGE |
|
|
CLK_FILTER_NEAREST);
|
|
int2 block = (int2)(get_group_id(0), get_group_id(1)) * 16;
|
|
int2 pos = (int2)(get_local_id(0), get_local_id(1));
|
|
|
|
__local float4 tmp[32][32];
|
|
|
|
int rad_x = size_x / 2;
|
|
int rad_y = size_y / 2;
|
|
int x, y;
|
|
|
|
for (y = 0; y <= 1; y++) {
|
|
for (x = 0; x <= 1; x++) {
|
|
tmp[pos.y + 16 * y][pos.x + 16 * x] =
|
|
read_imagef(src, sampler, block + pos + (int2)(16 * x - 8, 16 * y - 8));
|
|
}
|
|
}
|
|
|
|
barrier(CLK_LOCAL_MEM_FENCE);
|
|
|
|
float4 val = tmp[pos.y + 8][pos.x + 8];
|
|
|
|
float4 horiz[2];
|
|
for (y = 0; y <= 1; y++) {
|
|
horiz[y] = 0.0f;
|
|
for (x = 0; x < size_x; x++)
|
|
horiz[y] += coef_x[x] * tmp[pos.y + y * 16][pos.x + 8 + x - rad_x];
|
|
}
|
|
|
|
barrier(CLK_LOCAL_MEM_FENCE);
|
|
|
|
for (y = 0; y <= 1; y++) {
|
|
tmp[pos.y + y * 16][pos.x + 8] = horiz[y];
|
|
}
|
|
|
|
barrier(CLK_LOCAL_MEM_FENCE);
|
|
|
|
float4 sum = 0.0f;
|
|
for (y = 0; y < size_y; y++)
|
|
sum += coef_y[y] * tmp[pos.y + 8 + y - rad_y][pos.x + 8];
|
|
|
|
if (block.x + pos.x < get_image_width(dst) &&
|
|
block.y + pos.y < get_image_height(dst))
|
|
write_imagef(dst, block + pos, val + (val - sum) * amount);
|
|
}
|