X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavfilter%2Fvf_deshake.c;h=4eae9885c6cdf6f5eb5ddf9a4c9049fc308a5cbc;hb=ba7d16a30353a7ff2833fd2ac2fbd3cd977d278f;hp=79fcc200e1b8a2a8a8f32001ce86b99dceaa6def;hpb=74b8fa103d6345c0859434447b0a66aaf5608bd0;p=ffmpeg diff --git a/libavfilter/vf_deshake.c b/libavfilter/vf_deshake.c index 79fcc200e1b..4eae9885c6c 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 = AV_CEIL_RSHIFT(link->w, desc->log2_chroma_w); + const int chroma_height = AV_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