yading@10: /* yading@10: * Copyright (c) 2002 Michael Niedermayer yading@10: * Copyright (c) 2013 Paul B Mahol yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: /** yading@10: * @file yading@10: * (de)interleave fields filter yading@10: */ yading@10: yading@10: #include "libavutil/opt.h" yading@10: #include "libavutil/imgutils.h" yading@10: #include "libavutil/pixdesc.h" yading@10: #include "avfilter.h" yading@10: #include "internal.h" yading@10: yading@10: enum FilterMode { yading@10: MODE_NONE, yading@10: MODE_INTERLEAVE, yading@10: MODE_DEINTERLEAVE yading@10: }; yading@10: yading@10: typedef struct { yading@10: const AVClass *class; yading@10: enum FilterMode luma_mode, chroma_mode, alpha_mode; yading@10: int luma_swap, chroma_swap, alpha_swap; yading@10: int nb_planes; yading@10: int linesize[4], chroma_height; yading@10: int has_alpha; yading@10: } IlContext; yading@10: yading@10: #define OFFSET(x) offsetof(IlContext, x) yading@10: #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM yading@10: yading@10: static const AVOption il_options[] = { yading@10: {"luma_mode", "select luma mode", OFFSET(luma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "luma_mode"}, yading@10: {"l", "select luma mode", OFFSET(luma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "luma_mode"}, yading@10: {"none", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_NONE}, 0, 0, FLAGS, "luma_mode"}, yading@10: {"interleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "luma_mode"}, yading@10: {"i", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "luma_mode"}, yading@10: {"deinterleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "luma_mode"}, yading@10: {"d", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "luma_mode"}, yading@10: {"chroma_mode", "select chroma mode", OFFSET(chroma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "chroma_mode"}, yading@10: {"c", "select chroma mode", OFFSET(chroma_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "chroma_mode"}, yading@10: {"none", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_NONE}, 0, 0, FLAGS, "chroma_mode"}, yading@10: {"interleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "chroma_mode"}, yading@10: {"i", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "chroma_mode"}, yading@10: {"deinterleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "chroma_mode"}, yading@10: {"d", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "chroma_mode"}, yading@10: {"alpha_mode", "select alpha mode", OFFSET(alpha_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "alpha_mode"}, yading@10: {"a", "select alpha mode", OFFSET(alpha_mode), AV_OPT_TYPE_INT, {.i64=MODE_NONE}, MODE_NONE, MODE_DEINTERLEAVE, FLAGS, "alpha_mode"}, yading@10: {"none", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_NONE}, 0, 0, FLAGS, "alpha_mode"}, yading@10: {"interleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "alpha_mode"}, yading@10: {"i", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_INTERLEAVE}, 0, 0, FLAGS, "alpha_mode"}, yading@10: {"deinterleave", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "alpha_mode"}, yading@10: {"d", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MODE_DEINTERLEAVE}, 0, 0, FLAGS, "alpha_mode"}, yading@10: {"luma_swap", "swap luma fields", OFFSET(luma_swap), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS}, yading@10: {"ls", "swap luma fields", OFFSET(luma_swap), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS}, yading@10: {"chroma_swap", "swap chroma fields", OFFSET(chroma_swap), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS}, yading@10: {"cs", "swap chroma fields", OFFSET(chroma_swap), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS}, yading@10: {"alpha_swap", "swap alpha fields", OFFSET(alpha_swap), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS}, yading@10: {"as", "swap alpha fields", OFFSET(alpha_swap), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS}, yading@10: {NULL} yading@10: }; yading@10: yading@10: AVFILTER_DEFINE_CLASS(il); yading@10: yading@10: static int query_formats(AVFilterContext *ctx) yading@10: { yading@10: AVFilterFormats *formats = NULL; yading@10: int fmt; yading@10: yading@10: for (fmt = 0; fmt < AV_PIX_FMT_NB; fmt++) { yading@10: const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(fmt); yading@10: if (!(desc->flags & PIX_FMT_PAL) && !(desc->flags & PIX_FMT_HWACCEL)) yading@10: ff_add_format(&formats, fmt); yading@10: } yading@10: yading@10: ff_set_common_formats(ctx, formats); yading@10: return 0; yading@10: } yading@10: yading@10: static int config_input(AVFilterLink *inlink) yading@10: { yading@10: IlContext *il = inlink->dst->priv; yading@10: const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format); yading@10: int i, ret; yading@10: yading@10: for (i = 0; i < desc->nb_components; i++) yading@10: il->nb_planes = FFMAX(il->nb_planes, desc->comp[i].plane); yading@10: il->nb_planes++; yading@10: yading@10: il->has_alpha = !!(desc->flags & PIX_FMT_ALPHA); yading@10: if ((ret = av_image_fill_linesizes(il->linesize, inlink->format, inlink->w)) < 0) yading@10: return ret; yading@10: yading@10: il->chroma_height = inlink->h >> desc->log2_chroma_h; yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static void interleave(uint8_t *dst, uint8_t *src, int w, int h, yading@10: int dst_linesize, int src_linesize, yading@10: enum FilterMode mode, int swap) yading@10: { yading@10: const int a = swap; yading@10: const int b = 1 - a; yading@10: const int m = h >> 1; yading@10: int y; yading@10: yading@10: switch (mode) { yading@10: case MODE_DEINTERLEAVE: yading@10: for (y = 0; y < m; y++) { yading@10: memcpy(dst + dst_linesize * y , src + src_linesize * (y * 2 + a), w); yading@10: memcpy(dst + dst_linesize * (y + m), src + src_linesize * (y * 2 + b), w); yading@10: } yading@10: break; yading@10: case MODE_NONE: yading@10: for (y = 0; y < m; y++) { yading@10: memcpy(dst + dst_linesize * y * 2 , src + src_linesize * (y * 2 + a), w); yading@10: memcpy(dst + dst_linesize * (y * 2 + 1), src + src_linesize * (y * 2 + b), w); yading@10: } yading@10: break; yading@10: case MODE_INTERLEAVE: yading@10: for (y = 0; y < m; y++) { yading@10: memcpy(dst + dst_linesize * (y * 2 + a), src + src_linesize * y , w); yading@10: memcpy(dst + dst_linesize * (y * 2 + b), src + src_linesize * (y + m), w); yading@10: } yading@10: break; yading@10: } yading@10: } yading@10: yading@10: static int filter_frame(AVFilterLink *inlink, AVFrame *inpicref) yading@10: { yading@10: IlContext *il = inlink->dst->priv; yading@10: AVFilterLink *outlink = inlink->dst->outputs[0]; yading@10: AVFrame *out; yading@10: int comp; yading@10: yading@10: out = ff_get_video_buffer(outlink, outlink->w, outlink->h); yading@10: if (!out) { yading@10: av_frame_free(&inpicref); yading@10: return AVERROR(ENOMEM); yading@10: } yading@10: av_frame_copy_props(out, inpicref); yading@10: yading@10: interleave(out->data[0], inpicref->data[0], yading@10: il->linesize[0], inlink->h, yading@10: out->linesize[0], inpicref->linesize[0], yading@10: il->luma_mode, il->luma_swap); yading@10: yading@10: for (comp = 1; comp < (il->nb_planes - il->has_alpha); comp++) { yading@10: interleave(out->data[comp], inpicref->data[comp], yading@10: il->linesize[comp], il->chroma_height, yading@10: out->linesize[comp], inpicref->linesize[comp], yading@10: il->chroma_mode, il->chroma_swap); yading@10: } yading@10: yading@10: if (il->has_alpha) { yading@10: comp = il->nb_planes - 1; yading@10: interleave(out->data[comp], inpicref->data[comp], yading@10: il->linesize[comp], inlink->h, yading@10: out->linesize[comp], inpicref->linesize[comp], yading@10: il->alpha_mode, il->alpha_swap); yading@10: } yading@10: yading@10: av_frame_free(&inpicref); yading@10: return ff_filter_frame(outlink, out); yading@10: } yading@10: yading@10: static const AVFilterPad inputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .get_video_buffer = ff_null_get_video_buffer, yading@10: .filter_frame = filter_frame, yading@10: .config_props = config_input, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: static const AVFilterPad outputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: AVFilter avfilter_vf_il = { yading@10: .name = "il", yading@10: .description = NULL_IF_CONFIG_SMALL("Deinterleave or interleave fields."), yading@10: .priv_size = sizeof(IlContext), yading@10: .query_formats = query_formats, yading@10: .inputs = inputs, yading@10: .outputs = outputs, yading@10: .priv_class = &il_class, yading@10: };