* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
+#include "libavutil/avassert.h"
#include "libavutil/colorspace.h"
#include "libavutil/common.h"
#include "libavutil/opt.h"
enum { Y, U, V, A };
enum { R, G, B };
-enum FillMode { FM_SMEAR, FM_MIRROR, FM_FIXED, FM_NB_MODES };
+enum FillMode { FM_SMEAR, FM_MIRROR, FM_FIXED, FM_REFLECT, FM_WRAP, FM_FADE, FM_NB_MODES };
typedef struct Borders {
int left, right, top, bottom;
AV_PIX_FMT_YUV420P16, AV_PIX_FMT_YUV422P16, AV_PIX_FMT_YUV444P16,
AV_PIX_FMT_YUVA420P9, AV_PIX_FMT_YUVA422P9, AV_PIX_FMT_YUVA444P9,
AV_PIX_FMT_YUVA420P10, AV_PIX_FMT_YUVA422P10, AV_PIX_FMT_YUVA444P10,
+ AV_PIX_FMT_YUVA422P12, AV_PIX_FMT_YUVA444P12,
AV_PIX_FMT_YUVA420P16, AV_PIX_FMT_YUVA422P16, AV_PIX_FMT_YUVA444P16,
AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRP9, AV_PIX_FMT_GBRP10,
AV_PIX_FMT_GBRP12, AV_PIX_FMT_GBRP14, AV_PIX_FMT_GBRP16,
}
}
+static void reflect_borders8(FillBordersContext *s, AVFrame *frame)
+{
+ int p, y, x;
+
+ for (p = 0; p < s->nb_planes; p++) {
+ uint8_t *ptr = frame->data[p];
+ int linesize = frame->linesize[p];
+
+ for (y = s->borders[p].top; y < s->planeheight[p] - s->borders[p].bottom; y++) {
+ for (x = 0; x < s->borders[p].left; x++) {
+ ptr[y * linesize + x] = ptr[y * linesize + s->borders[p].left * 2 - x];
+ }
+
+ for (x = 0; x < s->borders[p].right; x++) {
+ ptr[y * linesize + s->planewidth[p] - s->borders[p].right + x] =
+ ptr[y * linesize + s->planewidth[p] - s->borders[p].right - 2 - x];
+ }
+ }
+
+ for (y = 0; y < s->borders[p].top; y++) {
+ memcpy(ptr + y * linesize,
+ ptr + (s->borders[p].top * 2 - y) * linesize,
+ s->planewidth[p]);
+ }
+
+ for (y = 0; y < s->borders[p].bottom; y++) {
+ memcpy(ptr + (s->planeheight[p] - s->borders[p].bottom + y) * linesize,
+ ptr + (s->planeheight[p] - s->borders[p].bottom - 2 - y) * linesize,
+ s->planewidth[p]);
+ }
+ }
+}
+
+static void reflect_borders16(FillBordersContext *s, AVFrame *frame)
+{
+ int p, y, x;
+
+ for (p = 0; p < s->nb_planes; p++) {
+ uint16_t *ptr = (uint16_t *)frame->data[p];
+ int linesize = frame->linesize[p] / 2;
+
+ for (y = s->borders[p].top; y < s->planeheight[p] - s->borders[p].bottom; y++) {
+ for (x = 0; x < s->borders[p].left; x++) {
+ ptr[y * linesize + x] = ptr[y * linesize + s->borders[p].left * 2 - x];
+ }
+
+ for (x = 0; x < s->borders[p].right; x++) {
+ ptr[y * linesize + s->planewidth[p] - s->borders[p].right + x] =
+ ptr[y * linesize + s->planewidth[p] - s->borders[p].right - 2 - x];
+ }
+ }
+
+ for (y = 0; y < s->borders[p].top; y++) {
+ memcpy(ptr + y * linesize,
+ ptr + (s->borders[p].top * 2 - y) * linesize,
+ s->planewidth[p] * 2);
+ }
+
+ for (y = 0; y < s->borders[p].bottom; y++) {
+ memcpy(ptr + (s->planeheight[p] - s->borders[p].bottom + y) * linesize,
+ ptr + (s->planeheight[p] - s->borders[p].bottom - 2 - y) * linesize,
+ s->planewidth[p] * 2);
+ }
+ }
+}
+
+static void wrap_borders8(FillBordersContext *s, AVFrame *frame)
+{
+ int p, y, x;
+
+ for (p = 0; p < s->nb_planes; p++) {
+ uint8_t *ptr = frame->data[p];
+ int linesize = frame->linesize[p];
+
+ for (y = s->borders[p].top; y < s->planeheight[p] - s->borders[p].bottom; y++) {
+ for (x = 0; x < s->borders[p].left; x++) {
+ ptr[y * linesize + x] = ptr[y * linesize + s->planewidth[p] - s->borders[p].right - s->borders[p].left + x];
+ }
+
+ for (x = 0; x < s->borders[p].right; x++) {
+ ptr[y * linesize + s->planewidth[p] - s->borders[p].right + x] =
+ ptr[y * linesize + s->borders[p].left + x];
+ }
+ }
+
+ for (y = 0; y < s->borders[p].top; y++) {
+ memcpy(ptr + y * linesize,
+ ptr + (s->planeheight[p] - s->borders[p].bottom - s->borders[p].top + y) * linesize,
+ s->planewidth[p]);
+ }
+
+ for (y = 0; y < s->borders[p].bottom; y++) {
+ memcpy(ptr + (s->planeheight[p] - s->borders[p].bottom + y) * linesize,
+ ptr + (s->borders[p].top + y) * linesize,
+ s->planewidth[p]);
+ }
+ }
+}
+
+static void wrap_borders16(FillBordersContext *s, AVFrame *frame)
+{
+ int p, y, x;
+
+ for (p = 0; p < s->nb_planes; p++) {
+ uint16_t *ptr = (uint16_t *)frame->data[p];
+ int linesize = frame->linesize[p] / 2;
+
+ for (y = s->borders[p].top; y < s->planeheight[p] - s->borders[p].bottom; y++) {
+ for (x = 0; x < s->borders[p].left; x++) {
+ ptr[y * linesize + x] = ptr[y * linesize + s->planewidth[p] - s->borders[p].right - s->borders[p].left + x];
+ }
+
+ for (x = 0; x < s->borders[p].right; x++) {
+ ptr[y * linesize + s->planewidth[p] - s->borders[p].right + x] =
+ ptr[y * linesize + s->borders[p].left + x];
+ }
+ }
+
+ for (y = 0; y < s->borders[p].top; y++) {
+ memcpy(ptr + y * linesize,
+ ptr + (s->planeheight[p] - s->borders[p].bottom - s->borders[p].top + y) * linesize,
+ s->planewidth[p] * 2);
+ }
+
+ for (y = 0; y < s->borders[p].bottom; y++) {
+ memcpy(ptr + (s->planeheight[p] - s->borders[p].bottom + y) * linesize,
+ ptr + (s->borders[p].top + y) * linesize,
+ s->planewidth[p] * 2);
+ }
+ }
+}
+
+static int lerp8(int fill, int src, int pos, int size)
+{
+ return av_clip_uint8(((fill * 256 * pos / size) + (src * 256 * (size - pos) / size)) >> 8);
+}
+
+static int lerp16(int fill, int src, int pos, int size, int depth)
+{
+ return av_clip_uintp2_c(((fill * (1LL << depth) * pos / size) + (src * (1LL << depth) * (size - pos) / size)) >> depth, depth);
+}
+
+static void fade_borders8(FillBordersContext *s, AVFrame *frame)
+{
+ int p, y, x;
+
+ for (p = 0; p < s->nb_planes; p++) {
+ uint8_t *ptr = frame->data[p];
+ const uint8_t fill = s->fill[p];
+ const int linesize = frame->linesize[p];
+ const int start_left = s->borders[p].left;
+ const int start_right = s->planewidth[p] - s->borders[p].right;
+ const int start_top = s->borders[p].top;
+ const int start_bottom = s->planeheight[p] - s->borders[p].bottom;
+
+ for (y = 0; y < start_top; y++) {
+ for (x = 0; x < s->planewidth[p]; x++) {
+ int src = ptr[y * linesize + x];
+ ptr[y * linesize + x] = lerp8(fill, src, start_top - y, start_top);
+ }
+ }
+
+ for (y = start_bottom; y < s->planeheight[p]; y++) {
+ for (x = 0; x < s->planewidth[p]; x++) {
+ int src = ptr[y * linesize + x];
+ ptr[y * linesize + x] = lerp8(fill, src, y - start_bottom, s->borders[p].bottom);
+ }
+ }
+
+ for (y = 0; y < s->planeheight[p]; y++) {
+ for (x = 0; x < start_left; x++) {
+ int src = ptr[y * linesize + x];
+ ptr[y * linesize + x] = lerp8(fill, src, start_left - x, start_left);
+ }
+
+ for (x = 0; x < s->borders[p].right; x++) {
+ int src = ptr[y * linesize + start_right + x];
+ ptr[y * linesize + start_right + x] = lerp8(fill, src, x, s->borders[p].right);
+ }
+ }
+ }
+}
+
+static void fade_borders16(FillBordersContext *s, AVFrame *frame)
+{
+ const int depth = s->depth;
+ int p, y, x;
+
+ for (p = 0; p < s->nb_planes; p++) {
+ uint16_t *ptr = (uint16_t *)frame->data[p];
+ const uint16_t fill = s->fill[p] << (depth - 8);
+ const int linesize = frame->linesize[p] / 2;
+ const int start_left = s->borders[p].left;
+ const int start_right = s->planewidth[p] - s->borders[p].right;
+ const int start_top = s->borders[p].top;
+ const int start_bottom = s->planeheight[p] - s->borders[p].bottom;
+
+ for (y = 0; y < start_top; y++) {
+ for (x = 0; x < s->planewidth[p]; x++) {
+ int src = ptr[y * linesize + x];
+ ptr[y * linesize + x] = lerp16(fill, src, start_top - y, start_top, depth);
+ }
+ }
+
+ for (y = start_bottom; y < s->planeheight[p]; y++) {
+ for (x = 0; x < s->planewidth[p]; x++) {
+ int src = ptr[y * linesize + x];
+ ptr[y * linesize + x] = lerp16(fill, src, y - start_bottom, s->borders[p].bottom, depth);
+ }
+ }
+
+ for (y = 0; y < s->planeheight[p]; y++) {
+ for (x = 0; x < start_left; x++) {
+ int src = ptr[y * linesize + x];
+ ptr[y * linesize + x] = lerp16(fill, src, start_left - x, start_left, depth);
+ }
+
+ for (x = 0; x < s->borders[p].right; x++) {
+ int src = ptr[y * linesize + start_right + x];
+ ptr[y * linesize + start_right + x] = lerp16(fill, src, x, s->borders[p].right, depth);
+ }
+ }
+ }
+}
+
static int filter_frame(AVFilterLink *inlink, AVFrame *frame)
{
FillBordersContext *s = inlink->dst->priv;
s->planewidth[1] = s->planewidth[2] = AV_CEIL_RSHIFT(inlink->w, desc->log2_chroma_w);
s->planewidth[0] = s->planewidth[3] = inlink->w;
+ if (inlink->w < s->left + s->right ||
+ inlink->w <= s->left ||
+ inlink->w <= s->right ||
+ inlink->h < s->top + s->bottom ||
+ inlink->h <= s->top ||
+ inlink->h <= s->bottom ||
+ inlink->w < s->left * 2 ||
+ inlink->w < s->right * 2 ||
+ inlink->h < s->top * 2 ||
+ inlink->h < s->bottom * 2) {
+ av_log(ctx, AV_LOG_ERROR, "Borders are bigger than input frame size.\n");
+ return AVERROR(EINVAL);
+ }
+
s->borders[0].left = s->borders[3].left = s->left;
s->borders[0].right = s->borders[3].right = s->right;
s->borders[0].top = s->borders[3].top = s->top;
s->borders[2].top = s->top >> desc->log2_chroma_h;
s->borders[2].bottom = s->bottom >> desc->log2_chroma_h;
- if (inlink->w < s->left + s->right ||
- inlink->w <= s->left ||
- inlink->w <= s->right ||
- inlink->h < s->top + s->bottom ||
- inlink->h <= s->top ||
- inlink->h <= s->bottom ||
- inlink->w < s->left * 2 ||
- inlink->w < s->right * 2 ||
- inlink->h < s->top * 2 ||
- inlink->h < s->bottom * 2) {
- av_log(ctx, AV_LOG_ERROR, "Borders are bigger than input frame size.\n");
- return AVERROR(EINVAL);
- }
-
switch (s->mode) {
case FM_SMEAR: s->fillborders = s->depth <= 8 ? smear_borders8 : smear_borders16; break;
case FM_MIRROR: s->fillborders = s->depth <= 8 ? mirror_borders8 : mirror_borders16; break;
case FM_FIXED: s->fillborders = s->depth <= 8 ? fixed_borders8 : fixed_borders16; break;
+ case FM_REFLECT:s->fillborders = s->depth <= 8 ? reflect_borders8: reflect_borders16;break;
+ case FM_WRAP: s->fillborders = s->depth <= 8 ? wrap_borders8 : wrap_borders16; break;
+ case FM_FADE: s->fillborders = s->depth <= 8 ? fade_borders8 : fade_borders16; break;
+ default: av_assert0(0);
}
s->yuv_color[Y] = RGB_TO_Y_CCIR(s->rgba_color[R], s->rgba_color[G], s->rgba_color[B]);
return 0;
}
+static int process_command(AVFilterContext *ctx, const char *cmd, const char *args,
+ char *res, int res_len, int flags)
+{
+ int ret;
+
+ ret = ff_filter_process_command(ctx, cmd, args, res, res_len, flags);
+ if (ret < 0)
+ return ret;
+
+ return config_input(ctx->inputs[0]);
+}
+
#define OFFSET(x) offsetof(FillBordersContext, x)
-#define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
+#define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
static const AVOption fillborders_options[] = {
{ "left", "set the left fill border", OFFSET(left), AV_OPT_TYPE_INT, {.i64=0}, 0, INT_MAX, FLAGS },
{ "smear", NULL, 0, AV_OPT_TYPE_CONST, {.i64=FM_SMEAR}, 0, 0, FLAGS, "mode" },
{ "mirror", NULL, 0, AV_OPT_TYPE_CONST, {.i64=FM_MIRROR}, 0, 0, FLAGS, "mode" },
{ "fixed", NULL, 0, AV_OPT_TYPE_CONST, {.i64=FM_FIXED}, 0, 0, FLAGS, "mode" },
- { "color", "set the color for the fixed mode", OFFSET(rgba_color), AV_OPT_TYPE_COLOR, {.str = "black"}, .flags = FLAGS },
+ { "reflect",NULL, 0, AV_OPT_TYPE_CONST, {.i64=FM_REFLECT},0, 0, FLAGS, "mode" },
+ { "wrap", NULL, 0, AV_OPT_TYPE_CONST, {.i64=FM_WRAP}, 0, 0, FLAGS, "mode" },
+ { "fade", NULL, 0, AV_OPT_TYPE_CONST, {.i64=FM_FADE}, 0, 0, FLAGS, "mode" },
+ { "color", "set the color for the fixed/fade mode", OFFSET(rgba_color), AV_OPT_TYPE_COLOR, {.str = "black"}, .flags = FLAGS },
{ NULL }
};
{ NULL }
};
-AVFilter ff_vf_fillborders = {
+const AVFilter ff_vf_fillborders = {
.name = "fillborders",
.description = NULL_IF_CONFIG_SMALL("Fill borders of the input video."),
.priv_size = sizeof(FillBordersContext),
.inputs = fillborders_inputs,
.outputs = fillborders_outputs,
.flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC,
+ .process_command = process_command,
};