yading@11: /*
yading@11:  * Copyright (c) 2002 Michael Niedermayer <michaelni@gmx.at>
yading@11:  *
yading@11:  * This file is part of FFmpeg.
yading@11:  *
yading@11:  * FFmpeg is free software; you can redistribute it and/or
yading@11:  * modify it under the terms of the GNU Lesser General Public
yading@11:  * License as published by the Free Software Foundation; either
yading@11:  * version 2.1 of the License, or (at your option) any later version.
yading@11:  *
yading@11:  * FFmpeg is distributed in the hope that it will be useful,
yading@11:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@11:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
yading@11:  * Lesser General Public License for more details.
yading@11:  *
yading@11:  * You should have received a copy of the GNU Lesser General Public
yading@11:  * License along with FFmpeg; if not, write to the Free Software
yading@11:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@11:  */
yading@11: 
yading@11: /**
yading@11:  * @file
yading@11:  * swap UV filter
yading@11:  */
yading@11: 
yading@11: #include "libavutil/pixdesc.h"
yading@11: #include "avfilter.h"
yading@11: #include "formats.h"
yading@11: #include "internal.h"
yading@11: #include "video.h"
yading@11: 
yading@11: static void do_swap(AVFrame *frame)
yading@11: {
yading@11:     FFSWAP(uint8_t*,     frame->data[1],     frame->data[2]);
yading@11:     FFSWAP(int,          frame->linesize[1], frame->linesize[2]);
yading@11:     FFSWAP(uint64_t,     frame->error[1],    frame->error[2]);
yading@11:     FFSWAP(AVBufferRef*, frame->buf[1],      frame->buf[2]);
yading@11: }
yading@11: 
yading@11: static AVFrame *get_video_buffer(AVFilterLink *link, int w, int h)
yading@11: {
yading@11:     AVFrame *picref = ff_default_get_video_buffer(link, w, h);
yading@11:     do_swap(picref);
yading@11:     return picref;
yading@11: }
yading@11: 
yading@11: static int filter_frame(AVFilterLink *link, AVFrame *inpicref)
yading@11: {
yading@11:     do_swap(inpicref);
yading@11:     return ff_filter_frame(link->dst->outputs[0], inpicref);
yading@11: }
yading@11: 
yading@11: static int is_planar_yuv(const AVPixFmtDescriptor *desc)
yading@11: {
yading@11:     int i;
yading@11: 
yading@11:     if (desc->flags & ~(PIX_FMT_BE | PIX_FMT_PLANAR | PIX_FMT_ALPHA) ||
yading@11:         desc->nb_components < 3 ||
yading@11:         (desc->comp[1].depth_minus1 != desc->comp[2].depth_minus1))
yading@11:         return 0;
yading@11:     for (i = 0; i < desc->nb_components; i++) {
yading@11:         if (desc->comp[i].offset_plus1 != 1 ||
yading@11:             desc->comp[i].shift != 0 ||
yading@11:             desc->comp[i].plane != i)
yading@11:             return 0;
yading@11:     }
yading@11: 
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int query_formats(AVFilterContext *ctx)
yading@11: {
yading@11:     AVFilterFormats *formats = NULL;
yading@11:     int fmt;
yading@11: 
yading@11:     for (fmt = 0; fmt < AV_PIX_FMT_NB; fmt++) {
yading@11:         const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(fmt);
yading@11:         if (is_planar_yuv(desc))
yading@11:             ff_add_format(&formats, fmt);
yading@11:     }
yading@11: 
yading@11:     ff_set_common_formats(ctx, formats);
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static const AVFilterPad swapuv_inputs[] = {
yading@11:     {
yading@11:         .name             = "default",
yading@11:         .type             = AVMEDIA_TYPE_VIDEO,
yading@11:         .get_video_buffer = get_video_buffer,
yading@11:         .filter_frame     = filter_frame,
yading@11:     },
yading@11:     { NULL }
yading@11: };
yading@11: 
yading@11: static const AVFilterPad swapuv_outputs[] = {
yading@11:     {
yading@11:         .name = "default",
yading@11:         .type = AVMEDIA_TYPE_VIDEO,
yading@11:     },
yading@11:     { NULL }
yading@11: };
yading@11: 
yading@11: AVFilter avfilter_vf_swapuv = {
yading@11:     .name      = "swapuv",
yading@11:     .description = NULL_IF_CONFIG_SMALL("Swap U and V components."),
yading@11:     .priv_size = 0,
yading@11:     .query_formats = query_formats,
yading@11:     .inputs        = swapuv_inputs,
yading@11:     .outputs       = swapuv_outputs,
yading@11: };