6e9abce5 | /* |
969039eb | * Copyright (c) 2002 Michael Niedermayer <michaelni@gmx.at> |
6e9abce5 | * Copyright (c) 2013 Paul B Mahol * * 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 */ /** * @file * (de)interleave fields filter */ #include "libavutil/opt.h" |
63a99622 | #include "libavutil/imgutils.h" |
6e9abce5 | #include "libavutil/pixdesc.h" #include "avfilter.h" #include "internal.h" enum FilterMode { MODE_NONE, MODE_INTERLEAVE, MODE_DEINTERLEAVE }; |
ed93ed5e | typedef struct IlContext { |
6e9abce5 | const AVClass *class; |
ee17295d | int luma_mode, chroma_mode, alpha_mode; ///<FilterMode |
6e9abce5 | int luma_swap, chroma_swap, alpha_swap; int nb_planes; |
63a99622 | int linesize[4], chroma_height; int has_alpha; |
6e9abce5 | } IlContext; #define OFFSET(x) offsetof(IlContext, x) #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM static const AVOption il_options[] = { {"luma_mode", "select luma mode", OFFSET(luma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "luma_mode"}, {"l", "select luma mode", OFFSET(luma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "luma_mode"}, {"none", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_NONE}, 0, 0, FLAGS, "luma_mode"}, {"interleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "luma_mode"}, {"i", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "luma_mode"}, {"deinterleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "luma_mode"}, {"d", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "luma_mode"}, {"chroma_mode", "select chroma mode", OFFSET(chroma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "chroma_mode"}, {"c", "select chroma mode", OFFSET(chroma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "chroma_mode"}, {"none", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_NONE}, 0, 0, FLAGS, "chroma_mode"}, {"interleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "chroma_mode"}, {"i", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "chroma_mode"}, {"deinterleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "chroma_mode"}, {"d", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "chroma_mode"}, {"alpha_mode", "select alpha mode", OFFSET(alpha_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "alpha_mode"}, {"a", "select alpha mode", OFFSET(alpha_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "alpha_mode"}, {"none", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_NONE}, 0, 0, FLAGS, "alpha_mode"}, {"interleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "alpha_mode"}, {"i", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "alpha_mode"}, {"deinterleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "alpha_mode"}, {"d", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "alpha_mode"}, |
a62bf26c | {"luma_swap", "swap luma fields", OFFSET(luma_swap), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS}, {"ls", "swap luma fields", OFFSET(luma_swap), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS}, {"chroma_swap", "swap chroma fields", OFFSET(chroma_swap), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS}, {"cs", "swap chroma fields", OFFSET(chroma_swap), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS}, {"alpha_swap", "swap alpha fields", OFFSET(alpha_swap), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS}, {"as", "swap alpha fields", OFFSET(alpha_swap), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS}, |
6e9abce5 | {NULL} }; AVFILTER_DEFINE_CLASS(il); |
0b189654 | static int query_formats(AVFilterContext *ctx) { AVFilterFormats *formats = NULL; |
6aaac24d | int fmt, ret; |
0b189654 | |
c0f8801e | for (fmt = 0; av_pix_fmt_desc_get(fmt); fmt++) { |
0b189654 | const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(fmt); |
6aaac24d | if (!(desc->flags & AV_PIX_FMT_FLAG_PAL) && !(desc->flags & AV_PIX_FMT_FLAG_HWACCEL) && (ret = ff_add_format(&formats, fmt)) < 0) return ret; |
0b189654 | } |
a0854c08 | return ff_set_common_formats(ctx, formats); |
0b189654 | } |
6e9abce5 | static int config_input(AVFilterLink *inlink) { |
abcf8610 | IlContext *s = inlink->dst->priv; |
6e9abce5 | const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format); |
2516b393 | int ret; |
6e9abce5 | |
abcf8610 | s->nb_planes = av_pix_fmt_count_planes(inlink->format); |
6e9abce5 | |
abcf8610 | s->has_alpha = !!(desc->flags & AV_PIX_FMT_FLAG_ALPHA); if ((ret = av_image_fill_linesizes(s->linesize, inlink->format, inlink->w)) < 0) |
63a99622 | return ret; |
21f94684 | s->chroma_height = AV_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h); |
6e9abce5 | return 0; } static void interleave(uint8_t *dst, uint8_t *src, int w, int h, int dst_linesize, int src_linesize, enum FilterMode mode, int swap) { const int a = swap; const int b = 1 - a; const int m = h >> 1; int y; switch (mode) { case MODE_DEINTERLEAVE: for (y = 0; y < m; y++) { memcpy(dst + dst_linesize * y , src + src_linesize * (y * 2 + a), w); memcpy(dst + dst_linesize * (y + m), src + src_linesize * (y * 2 + b), w); } break; case MODE_NONE: for (y = 0; y < m; y++) { memcpy(dst + dst_linesize * y * 2 , src + src_linesize * (y * 2 + a), w); memcpy(dst + dst_linesize * (y * 2 + 1), src + src_linesize * (y * 2 + b), w); } break; case MODE_INTERLEAVE: for (y = 0; y < m; y++) { memcpy(dst + dst_linesize * (y * 2 + a), src + src_linesize * y , w); memcpy(dst + dst_linesize * (y * 2 + b), src + src_linesize * (y + m), w); } break; } } |
a05a44e2 | static int filter_frame(AVFilterLink *inlink, AVFrame *inpicref) |
6e9abce5 | { |
abcf8610 | IlContext *s = inlink->dst->priv; |
6e9abce5 | AVFilterLink *outlink = inlink->dst->outputs[0]; |
a05a44e2 | AVFrame *out; |
295ce83e | int comp; |
6e9abce5 | |
a05a44e2 | out = ff_get_video_buffer(outlink, outlink->w, outlink->h); |
6e9abce5 | if (!out) { |
a05a44e2 | av_frame_free(&inpicref); |
6e9abce5 | return AVERROR(ENOMEM); } |
a05a44e2 | av_frame_copy_props(out, inpicref); |
6e9abce5 | interleave(out->data[0], inpicref->data[0], |
abcf8610 | s->linesize[0], inlink->h, |
6e9abce5 | out->linesize[0], inpicref->linesize[0], |
abcf8610 | s->luma_mode, s->luma_swap); |
6e9abce5 | |
abcf8610 | for (comp = 1; comp < (s->nb_planes - s->has_alpha); comp++) { |
63a99622 | interleave(out->data[comp], inpicref->data[comp], |
abcf8610 | s->linesize[comp], s->chroma_height, |
63a99622 | out->linesize[comp], inpicref->linesize[comp], |
abcf8610 | s->chroma_mode, s->chroma_swap); |
6e9abce5 | } |
63a99622 | |
abcf8610 | if (s->has_alpha) { comp = s->nb_planes - 1; |
6e9abce5 | interleave(out->data[comp], inpicref->data[comp], |
abcf8610 | s->linesize[comp], inlink->h, |
6e9abce5 | out->linesize[comp], inpicref->linesize[comp], |
abcf8610 | s->alpha_mode, s->alpha_swap); |
6e9abce5 | } |
a05a44e2 | av_frame_free(&inpicref); |
295ce83e | return ff_filter_frame(outlink, out); |
6e9abce5 | } static const AVFilterPad inputs[] = { { |
b211607b | .name = "default", .type = AVMEDIA_TYPE_VIDEO, .filter_frame = filter_frame, .config_props = config_input, |
6e9abce5 | }, { NULL } }; static const AVFilterPad outputs[] = { { |
b211607b | .name = "default", .type = AVMEDIA_TYPE_VIDEO, |
6e9abce5 | }, { NULL } }; |
325f6e0a | AVFilter ff_vf_il = { |
6e9abce5 | .name = "il", .description = NULL_IF_CONFIG_SMALL("Deinterleave or interleave fields."), .priv_size = sizeof(IlContext), .query_formats = query_formats, .inputs = inputs, .outputs = outputs, .priv_class = &il_class, |
35542388 | .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, |
6e9abce5 | }; |