projects / gstreamer-omap:gst-ffmpeg.git / blob
? search:
summary | shortlog | log | commit | commitdiff | tree
history | raw | HEAD
build: Add all kinds of compiler warning flags and fix the resulting warnings
[gstreamer-omap:gst-ffmpeg.git] / ext / ffmpeg / gstffmpegdeinterlace.c
1 /* GStreamer
2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
3  * This file:
4  * Copyright (C) 2005 Luca Ognibene <luogni@tin.it>
5  * Copyright (C) 2006 Martin Zlomek <martin.zlomek@itonis.tv>
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Library General Public
9  * License as published by the Free Software Foundation; either
10  * version 2 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Library General Public License for more details.
16  *
17  * You should have received a copy of the GNU Library General Public
18  * License along with this library; if not, write to the
19  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
20  * Boston, MA 02111-1307, USA.
21  */
22
23 #ifdef HAVE_CONFIG_H
24 #  include "config.h"
25 #endif
26
27 #ifdef HAVE_FFMPEG_UNINSTALLED
28 #  include <avcodec.h>
29 #else
30 #  include <libavcodec/avcodec.h>
31 #endif
32
33 #include <gst/gst.h>
34 #include <gst/video/video.h>
35
36 #include "gstffmpeg.h"
37 #include "gstffmpegcodecmap.h"
38 #include "gstffmpegutils.h"
39
40 typedef struct _GstFFMpegDeinterlace
41 {
42   GstElement element;
43
44   GstPad *sinkpad, *srcpad;
45
46   gint width, height;
47   gint to_size;
48
49   enum PixelFormat pixfmt;
50   AVPicture from_frame, to_frame;
51 } GstFFMpegDeinterlace;
52
53 typedef struct _GstFFMpegDeinterlaceClass
54 {
55   GstElementClass parent_class;
56 } GstFFMpegDeinterlaceClass;
57
58 #define GST_TYPE_FFMPEGDEINTERLACE \
59   (gst_ffmpegdeinterlace_get_type())
60 #define GST_FFMPEGDEINTERLACE(obj) \
61   (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_FFMPEGDEINTERLACE,GstFFMpegDeinterlace))
62 #define GST_FFMPEGDEINTERLACE_CLASS(klass) \
63   (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_FFMPEGDEINTERLACE,GstFFMpegDeinterlace))
64 #define GST_IS_FFMPEGDEINTERLACE(obj) \
65   (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_FFMPEGDEINTERLACE))
66 #define GST_IS_FFMPEGDEINTERLACE_CLASS(klass) \
67   (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_FFMPEGDEINTERLACE))
68
69 GType gst_ffmpegdeinterlace_get_type (void);
70
71 static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src",
72     GST_PAD_SRC,
73     GST_PAD_ALWAYS,
74     GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420"))
75     );
76
77 static GstStaticPadTemplate sink_factory = GST_STATIC_PAD_TEMPLATE ("sink",
78     GST_PAD_SINK,
79     GST_PAD_ALWAYS,
80     GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420"))
81     );
82
83 GST_BOILERPLATE (GstFFMpegDeinterlace, gst_ffmpegdeinterlace, GstElement,
84     GST_TYPE_ELEMENT);
85
86 static GstFlowReturn gst_ffmpegdeinterlace_chain (GstPad * pad,
87     GstBuffer * inbuf);
88
89 static void
90 gst_ffmpegdeinterlace_base_init (gpointer g_class)
91 {
92   GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
93
94   gst_element_class_add_pad_template (element_class,
95       gst_static_pad_template_get (&src_factory));
96   gst_element_class_add_pad_template (element_class,
97       gst_static_pad_template_get (&sink_factory));
98   gst_element_class_set_details_simple (element_class,
99       "FFMPEG Deinterlace element", "Filter/Converter/Video",
100       "Deinterlace video", "Luca Ognibene <luogni@tin.it>");
101 }
102
103 static void
104 gst_ffmpegdeinterlace_class_init (GstFFMpegDeinterlaceClass * klass)
105 {
106 }
107
108 static gboolean
109 gst_ffmpegdeinterlace_sink_setcaps (GstPad * pad, GstCaps * caps)
110 {
111   GstFFMpegDeinterlace *deinterlace =
112       GST_FFMPEGDEINTERLACE (gst_pad_get_parent (pad));
113   GstStructure *structure = gst_caps_get_structure (caps, 0);
114   AVCodecContext *ctx;
115
116   if (!gst_structure_get_int (structure, "width", &deinterlace->width))
117     return FALSE;
118   if (!gst_structure_get_int (structure, "height", &deinterlace->height))
119     return FALSE;
120
121   ctx = avcodec_alloc_context ();
122   ctx->width = deinterlace->width;
123   ctx->height = deinterlace->height;
124   ctx->pix_fmt = PIX_FMT_NB;
125   gst_ffmpeg_caps_with_codectype (CODEC_TYPE_VIDEO, caps, ctx);
126   if (ctx->pix_fmt == PIX_FMT_NB) {
127     av_free (ctx);
128     return FALSE;
129   }
130
131   deinterlace->pixfmt = ctx->pix_fmt;
132
133   av_free (ctx);
134
135   deinterlace->to_size =
136       avpicture_get_size (deinterlace->pixfmt, deinterlace->width,
137       deinterlace->height);
138
139   return gst_pad_set_caps (deinterlace->srcpad, caps);
140 }
141
142 static void
143 gst_ffmpegdeinterlace_init (GstFFMpegDeinterlace * deinterlace,
144     GstFFMpegDeinterlaceClass * klass)
145 {
146   deinterlace->sinkpad =
147       gst_pad_new_from_static_template (&sink_factory, "sink");
148   gst_pad_set_setcaps_function (deinterlace->sinkpad,
149       gst_ffmpegdeinterlace_sink_setcaps);
150   gst_pad_set_chain_function (deinterlace->sinkpad,
151       gst_ffmpegdeinterlace_chain);
152   gst_element_add_pad (GST_ELEMENT (deinterlace), deinterlace->sinkpad);
153
154   deinterlace->srcpad = gst_pad_new_from_static_template (&src_factory, "src");
155   gst_element_add_pad (GST_ELEMENT (deinterlace), deinterlace->srcpad);
156
157   deinterlace->pixfmt = PIX_FMT_NB;
158 }
159
160 static GstFlowReturn
161 gst_ffmpegdeinterlace_chain (GstPad * pad, GstBuffer * inbuf)
162 {
163   GstFFMpegDeinterlace *deinterlace =
164       GST_FFMPEGDEINTERLACE (gst_pad_get_parent (pad));
165   GstBuffer *outbuf = NULL;
166   GstFlowReturn result;
167
168   result =
169       gst_pad_alloc_buffer (deinterlace->srcpad, GST_BUFFER_OFFSET_NONE,
170       deinterlace->to_size, GST_PAD_CAPS (deinterlace->srcpad), &outbuf);
171   if (result == GST_FLOW_OK) {
172     gst_ffmpeg_avpicture_fill (&deinterlace->from_frame,
173         GST_BUFFER_DATA (inbuf), deinterlace->pixfmt, deinterlace->width,
174         deinterlace->height);
175
176     gst_ffmpeg_avpicture_fill (&deinterlace->to_frame, GST_BUFFER_DATA (outbuf),
177         deinterlace->pixfmt, deinterlace->width, deinterlace->height);
178
179     avpicture_deinterlace (&deinterlace->to_frame, &deinterlace->from_frame,
180         deinterlace->pixfmt, deinterlace->width, deinterlace->height);
181
182     gst_buffer_copy_metadata (outbuf, inbuf, GST_BUFFER_COPY_TIMESTAMPS);
183
184     result = gst_pad_push (deinterlace->srcpad, outbuf);
185   }
186
187   gst_buffer_unref (inbuf);
188
189   return result;
190 }
191
192 gboolean
193 gst_ffmpegdeinterlace_register (GstPlugin * plugin)
194 {
195   return gst_element_register (plugin, "ffdeinterlace",
196       GST_RANK_NONE, GST_TYPE_FFMPEGDEINTERLACE);
197 }
3737df2e0b4b688075d16193496e847a2a