X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavfilter%2Fvf_deshake.c;h=e7ece443c3f55becaa60f54964efc83b01ce4eb3;hb=25955c927fcf88d1deea16f2d4ff86eb5a78a774;hp=79fcc200e1b8a2a8a8f32001ce86b99dceaa6def;hpb=a240aefc7a2591f7b307e2910df655801ae6ca44;p=ffmpeg diff --git a/libavfilter/vf_deshake.c b/libavfilter/vf_deshake.c index 79fcc200e1b..e7ece443c3f 100644 --- a/libavfilter/vf_deshake.c +++ b/libavfilter/vf_deshake.c @@ -62,9 +62,6 @@ #include "deshake.h" #include "deshake_opencl.h" -#define CHROMA_WIDTH(link) (-((-(link)->w) >> av_pix_fmt_desc_get((link)->format)->log2_chroma_w)) -#define CHROMA_HEIGHT(link) (-((-(link)->h) >> av_pix_fmt_desc_get((link)->format)->log2_chroma_h)) - #define OFFSET(x) offsetof(DeshakeContext, x) #define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM @@ -439,6 +436,9 @@ static int filter_frame(AVFilterLink *link, AVFrame *in) float alpha = 2.0 / deshake->refcount; char tmp[256]; int ret = 0; + const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(link->format); + const int chroma_width = FF_CEIL_RSHIFT(link->w, desc->log2_chroma_w); + const int chroma_height = FF_CEIL_RSHIFT(link->h, desc->log2_chroma_h); out = ff_get_video_buffer(outlink, outlink->w, outlink->h); if (!out) { @@ -527,9 +527,9 @@ static int filter_frame(AVFilterLink *link, AVFrame *in) // Generate a luma transformation matrix avfilter_get_matrix(t.vec.x, t.vec.y, t.angle, 1.0 + t.zoom / 100.0, matrix_y); // Generate a chroma transformation matrix - avfilter_get_matrix(t.vec.x / (link->w / CHROMA_WIDTH(link)), t.vec.y / (link->h / CHROMA_HEIGHT(link)), t.angle, 1.0 + t.zoom / 100.0, matrix_uv); + avfilter_get_matrix(t.vec.x / (link->w / chroma_width), t.vec.y / (link->h / chroma_height), t.angle, 1.0 + t.zoom / 100.0, matrix_uv); // Transform the luma and chroma planes - ret = deshake->transform(link->dst, link->w, link->h, CHROMA_WIDTH(link), CHROMA_HEIGHT(link), + ret = deshake->transform(link->dst, link->w, link->h, chroma_width, chroma_height, matrix_y, matrix_uv, INTERPOLATE_BILINEAR, deshake->edge, in, out); // Cleanup the old reference frame