99 lines · 3.3 KB · Text
RawDownload
/* * 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);}

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t