220 lines
7.6 KiB
C
220 lines
7.6 KiB
C
/*
|
|
* Copyright (C) 2013 Wei Gao <weigao@multicorewareinc.com>
|
|
*
|
|
*
|
|
* 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
|
|
*/
|
|
|
|
#ifndef AVFILTER_DESHAKE_KERNEL_H
|
|
#define AVFILTER_DESHAKE_KERNEL_H
|
|
|
|
#include "libavutil/opencl.h"
|
|
|
|
const char *ff_kernel_deshake_opencl = AV_OPENCL_KERNEL(
|
|
|
|
inline unsigned char pixel(global const unsigned char *src, float x, float y,
|
|
int w, int h,int stride, unsigned char def)
|
|
{
|
|
return (x < 0 || y < 0 || x >= w || y >= h) ? def : src[(int)x + (int)y * stride];
|
|
}
|
|
unsigned char interpolate_nearest(float x, float y, global const unsigned char *src,
|
|
int width, int height, int stride, unsigned char def)
|
|
{
|
|
return pixel(src, (int)(x + 0.5), (int)(y + 0.5), width, height, stride, def);
|
|
}
|
|
|
|
unsigned char interpolate_bilinear(float x, float y, global const unsigned char *src,
|
|
int width, int height, int stride, unsigned char def)
|
|
{
|
|
int x_c, x_f, y_c, y_f;
|
|
int v1, v2, v3, v4;
|
|
|
|
if (x < -1 || x > width || y < -1 || y > height) {
|
|
return def;
|
|
} else {
|
|
x_f = (int)x;
|
|
x_c = x_f + 1;
|
|
|
|
y_f = (int)y;
|
|
y_c = y_f + 1;
|
|
|
|
v1 = pixel(src, x_c, y_c, width, height, stride, def);
|
|
v2 = pixel(src, x_c, y_f, width, height, stride, def);
|
|
v3 = pixel(src, x_f, y_c, width, height, stride, def);
|
|
v4 = pixel(src, x_f, y_f, width, height, stride, def);
|
|
|
|
return (v1*(x - x_f)*(y - y_f) + v2*((x - x_f)*(y_c - y)) +
|
|
v3*(x_c - x)*(y - y_f) + v4*((x_c - x)*(y_c - y)));
|
|
}
|
|
}
|
|
|
|
unsigned char interpolate_biquadratic(float x, float y, global const unsigned char *src,
|
|
int width, int height, int stride, unsigned char def)
|
|
{
|
|
int x_c, x_f, y_c, y_f;
|
|
unsigned char v1, v2, v3, v4;
|
|
float f1, f2, f3, f4;
|
|
|
|
if (x < - 1 || x > width || y < -1 || y > height)
|
|
return def;
|
|
else {
|
|
x_f = (int)x;
|
|
x_c = x_f + 1;
|
|
y_f = (int)y;
|
|
y_c = y_f + 1;
|
|
|
|
v1 = pixel(src, x_c, y_c, width, height, stride, def);
|
|
v2 = pixel(src, x_c, y_f, width, height, stride, def);
|
|
v3 = pixel(src, x_f, y_c, width, height, stride, def);
|
|
v4 = pixel(src, x_f, y_f, width, height, stride, def);
|
|
|
|
f1 = 1 - sqrt((x_c - x) * (y_c - y));
|
|
f2 = 1 - sqrt((x_c - x) * (y - y_f));
|
|
f3 = 1 - sqrt((x - x_f) * (y_c - y));
|
|
f4 = 1 - sqrt((x - x_f) * (y - y_f));
|
|
return (v1 * f1 + v2 * f2 + v3 * f3 + v4 * f4) / (f1 + f2 + f3 + f4);
|
|
}
|
|
}
|
|
|
|
inline const float clipf(float a, float amin, float amax)
|
|
{
|
|
if (a < amin) return amin;
|
|
else if (a > amax) return amax;
|
|
else return a;
|
|
}
|
|
|
|
inline int mirror(int v, int m)
|
|
{
|
|
while ((unsigned)v > (unsigned)m) {
|
|
v = -v;
|
|
if (v < 0)
|
|
v += 2 * m;
|
|
}
|
|
return v;
|
|
}
|
|
|
|
kernel void avfilter_transform(global unsigned char *src,
|
|
global unsigned char *dst,
|
|
global float *matrix,
|
|
global float *matrix2,
|
|
int interpolate,
|
|
int fillmethod,
|
|
int src_stride_lu,
|
|
int dst_stride_lu,
|
|
int src_stride_ch,
|
|
int dst_stride_ch,
|
|
int height,
|
|
int width,
|
|
int ch,
|
|
int cw)
|
|
{
|
|
int global_id = get_global_id(0);
|
|
|
|
global unsigned char *dst_y = dst;
|
|
global unsigned char *dst_u = dst_y + height * dst_stride_lu;
|
|
global unsigned char *dst_v = dst_u + ch * dst_stride_ch;
|
|
|
|
global unsigned char *src_y = src;
|
|
global unsigned char *src_u = src_y + height * src_stride_lu;
|
|
global unsigned char *src_v = src_u + ch * src_stride_ch;
|
|
|
|
global unsigned char *tempdst;
|
|
global unsigned char *tempsrc;
|
|
|
|
int x;
|
|
int y;
|
|
float x_s;
|
|
float y_s;
|
|
int tempsrc_stride;
|
|
int tempdst_stride;
|
|
int temp_height;
|
|
int temp_width;
|
|
int curpos;
|
|
unsigned char def = 0;
|
|
if (global_id < width*height) {
|
|
y = global_id/width;
|
|
x = global_id%width;
|
|
x_s = x * matrix[0] + y * matrix[1] + matrix[2];
|
|
y_s = x * matrix[3] + y * matrix[4] + matrix[5];
|
|
tempdst = dst_y;
|
|
tempsrc = src_y;
|
|
tempsrc_stride = src_stride_lu;
|
|
tempdst_stride = dst_stride_lu;
|
|
temp_height = height;
|
|
temp_width = width;
|
|
} else if ((global_id >= width*height)&&(global_id < width*height + ch*cw)) {
|
|
y = (global_id - width*height)/cw;
|
|
x = (global_id - width*height)%cw;
|
|
x_s = x * matrix2[0] + y * matrix2[1] + matrix2[2];
|
|
y_s = x * matrix2[3] + y * matrix2[4] + matrix2[5];
|
|
tempdst = dst_u;
|
|
tempsrc = src_u;
|
|
tempsrc_stride = src_stride_ch;
|
|
tempdst_stride = dst_stride_ch;
|
|
temp_height = height;
|
|
temp_width = width;
|
|
temp_height = ch;
|
|
temp_width = cw;
|
|
} else {
|
|
y = (global_id - width*height - ch*cw)/cw;
|
|
x = (global_id - width*height - ch*cw)%cw;
|
|
x_s = x * matrix2[0] + y * matrix2[1] + matrix2[2];
|
|
y_s = x * matrix2[3] + y * matrix2[4] + matrix2[5];
|
|
tempdst = dst_v;
|
|
tempsrc = src_v;
|
|
tempsrc_stride = src_stride_ch;
|
|
tempdst_stride = dst_stride_ch;
|
|
temp_height = ch;
|
|
temp_width = cw;
|
|
}
|
|
curpos = y * tempdst_stride + x;
|
|
switch (fillmethod) {
|
|
case 0: //FILL_BLANK
|
|
def = 0;
|
|
break;
|
|
case 1: //FILL_ORIGINAL
|
|
def = tempsrc[y*tempsrc_stride+x];
|
|
break;
|
|
case 2: //FILL_CLAMP
|
|
y_s = clipf(y_s, 0, temp_height - 1);
|
|
x_s = clipf(x_s, 0, temp_width - 1);
|
|
def = tempsrc[(int)y_s * tempsrc_stride + (int)x_s];
|
|
break;
|
|
case 3: //FILL_MIRROR
|
|
y_s = mirror(y_s,temp_height - 1);
|
|
x_s = mirror(x_s,temp_width - 1);
|
|
def = tempsrc[(int)y_s * tempsrc_stride + (int)x_s];
|
|
break;
|
|
}
|
|
switch (interpolate) {
|
|
case 0: //INTERPOLATE_NEAREST
|
|
tempdst[curpos] = interpolate_nearest(x_s, y_s, tempsrc, temp_width, temp_height, tempsrc_stride, def);
|
|
break;
|
|
case 1: //INTERPOLATE_BILINEAR
|
|
tempdst[curpos] = interpolate_bilinear(x_s, y_s, tempsrc, temp_width, temp_height, tempsrc_stride, def);
|
|
break;
|
|
case 2: //INTERPOLATE_BIQUADRATIC
|
|
tempdst[curpos] = interpolate_biquadratic(x_s, y_s, tempsrc, temp_width, temp_height, tempsrc_stride, def);
|
|
break;
|
|
default:
|
|
return;
|
|
}
|
|
}
|
|
);
|
|
|
|
#endif /* AVFILTER_DESHAKE_KERNEL_H */
|