/* * 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 */