1 /*
2 * Copyright (c) 2002 Michael Niedermayer <michaelni@gmx.at>
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 /**
22 * @file
23 * swap UV filter
24 */
25
26 #include "libavutil/opt.h"
27 #include "libavutil/pixdesc.h"
28 #include "libavutil/version.h"
29 #include "avfilter.h"
30 #include "formats.h"
31 #include "internal.h"
32 #include "video.h"
33
34 typedef struct SwapUVContext {
35 const AVClass *class;
36 } SwapUVContext;
37
38 static const AVOption swapuv_options[] = {
39 { NULL }
40 };
41
42 AVFILTER_DEFINE_CLASS(swapuv);
43
do_swap(AVFrame * frame)44 static void do_swap(AVFrame *frame)
45 {
46 FFSWAP(uint8_t*, frame->data[1], frame->data[2]);
47 FFSWAP(int, frame->linesize[1], frame->linesize[2]);
48 FFSWAP(AVBufferRef*, frame->buf[1], frame->buf[2]);
49
50 #if FF_API_ERROR_FRAME
51 FF_DISABLE_DEPRECATION_WARNINGS
52 FFSWAP(uint64_t, frame->error[1], frame->error[2]);
53 FF_ENABLE_DEPRECATION_WARNINGS
54 #endif
55 }
56
get_video_buffer(AVFilterLink * link,int w,int h)57 static AVFrame *get_video_buffer(AVFilterLink *link, int w, int h)
58 {
59 AVFrame *picref = ff_default_get_video_buffer(link, w, h);
60 do_swap(picref);
61 return picref;
62 }
63
filter_frame(AVFilterLink * link,AVFrame * inpicref)64 static int filter_frame(AVFilterLink *link, AVFrame *inpicref)
65 {
66 do_swap(inpicref);
67 return ff_filter_frame(link->dst->outputs[0], inpicref);
68 }
69
is_planar_yuv(const AVPixFmtDescriptor * desc)70 static int is_planar_yuv(const AVPixFmtDescriptor *desc)
71 {
72 int i;
73
74 if (desc->flags & ~(AV_PIX_FMT_FLAG_BE | AV_PIX_FMT_FLAG_PLANAR | AV_PIX_FMT_FLAG_ALPHA) ||
75 desc->nb_components < 3 ||
76 (desc->comp[1].depth != desc->comp[2].depth))
77 return 0;
78 for (i = 0; i < desc->nb_components; i++) {
79 if (desc->comp[i].offset != 0 ||
80 desc->comp[i].shift != 0 ||
81 desc->comp[i].plane != i)
82 return 0;
83 }
84
85 return 1;
86 }
87
query_formats(AVFilterContext * ctx)88 static int query_formats(AVFilterContext *ctx)
89 {
90 AVFilterFormats *formats = NULL;
91 int fmt, ret;
92
93 for (fmt = 0; av_pix_fmt_desc_get(fmt); fmt++) {
94 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(fmt);
95 if (is_planar_yuv(desc) && (ret = ff_add_format(&formats, fmt)) < 0)
96 return ret;
97 }
98
99 return ff_set_common_formats(ctx, formats);
100 }
101
102 static const AVFilterPad swapuv_inputs[] = {
103 {
104 .name = "default",
105 .type = AVMEDIA_TYPE_VIDEO,
106 .get_video_buffer = get_video_buffer,
107 .filter_frame = filter_frame,
108 },
109 { NULL }
110 };
111
112 static const AVFilterPad swapuv_outputs[] = {
113 {
114 .name = "default",
115 .type = AVMEDIA_TYPE_VIDEO,
116 },
117 { NULL }
118 };
119
120 AVFilter ff_vf_swapuv = {
121 .name = "swapuv",
122 .description = NULL_IF_CONFIG_SMALL("Swap U and V components."),
123 .query_formats = query_formats,
124 .priv_size = sizeof(SwapUVContext),
125 .priv_class = &swapuv_class,
126 .inputs = swapuv_inputs,
127 .outputs = swapuv_outputs,
128 .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC,
129 };
130