expand tabs
[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  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Library General Public
8  * License as published by the Free Software Foundation; either
9  * version 2 of the License, or (at your option) any later version.
10  *
11  * This library 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  * Library General Public License for more details.
15  *
16  * You should have received a copy of the GNU Library General Public
17  * License along with this library; if not, write to the
18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
19  * Boston, MA 02111-1307, USA.
20  */
21
22 #ifdef HAVE_CONFIG_H
23 #include "config.h"
24 #endif
25
26 #include <gst/gst.h>
27 #ifdef HAVE_FFMPEG_UNINSTALLED
28 #include <avcodec.h>
29 #else
30 #include <ffmpeg/avcodec.h>
31 #endif
32
33 #include <gst/video/video.h>
34 #include "gstffmpeg.h"
35 #include "gstffmpegcodecmap.h"
36
37 #define GST_TYPE_FFMPEGDEINTERLACE \
38   (gst_ffmpegdeinterlace_get_type())
39 #define GST_FFMPEGDEINTERLACE(obj) \
40   (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_FFMPEGDEINTERLACE,GstFFMpegDeinterlace))
41 #define GST_FFMPEGDEINTERLACE_CLASS(klass) \
42   (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_FFMPEGDEINTERLACE,GstFFMpegDeinterlace))
43 #define GST_IS_FFMPEGDEINTERLACE(obj) \
44   (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_FFMPEGDEINTERLACE))
45 #define GST_IS_FFMPEGDEINTERLACE_CLASS(obj) \
46   (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_FFMPEGDEINTERLACE))
47
48 typedef struct _GstFFMpegDeinterlace GstFFMpegDeinterlace;
49 typedef struct _GstFFMpegDeinterlaceClass GstFFMpegDeinterlaceClass;
50
51 struct _GstFFMpegDeinterlace
52 {
53   GstElement element;
54
55   GstPad *sinkpad, *srcpad;
56
57   gint width, height;
58
59   enum PixelFormat pixfmt;
60   AVPicture from_frame, to_frame;
61   GstCaps *sinkcaps;
62   
63   guint to_size;
64
65 };
66
67 struct _GstFFMpegDeinterlaceClass
68 {
69   GstElementClass parent_class;
70 };
71
72 /* elementfactory information */
73 static GstElementDetails ffmpegdeinterlace_details = {
74   "FFMPEG Deinterlace element",
75   "Filter/Converter/Video",
76   "Deinterlace video",
77   "Luca Ognibene <luogni@tin.it>",
78 };
79
80 /* Signals and args */
81 enum
82 {
83   /* FILL ME */
84   LAST_SIGNAL
85 };
86
87 enum
88 {
89   ARG_0,
90 };
91
92 static GstStaticPadTemplate gst_ffmpegdeinterlace_src_template =
93 GST_STATIC_PAD_TEMPLATE ("src",
94     GST_PAD_SRC,
95     GST_PAD_ALWAYS,
96     GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420"))
97     );
98
99 static GstStaticPadTemplate gst_ffmpegdeinterlace_sink_template =
100 GST_STATIC_PAD_TEMPLATE ("sink",
101     GST_PAD_SINK,
102     GST_PAD_ALWAYS,
103     GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420"))
104     );
105
106
107 static GType gst_ffmpegdeinterlace_get_type (void);
108
109 static void gst_ffmpegdeinterlace_base_init (GstFFMpegDeinterlaceClass * klass);
110 static void gst_ffmpegdeinterlace_class_init (GstFFMpegDeinterlaceClass * klass);
111 static void gst_ffmpegdeinterlace_init (GstFFMpegDeinterlace * space);
112
113 static void gst_ffmpegdeinterlace_set_property (GObject * object,
114     guint prop_id, const GValue * value, GParamSpec * pspec);
115 static void gst_ffmpegdeinterlace_get_property (GObject * object,
116     guint prop_id, GValue * value, GParamSpec * pspec);
117
118 static GstPadLinkReturn
119 gst_ffmpegdeinterlace_pad_link (GstPad * pad, const GstCaps * caps);
120
121 static void gst_ffmpegdeinterlace_chain (GstPad * pad, GstData * data);
122 static GstStateChangeReturn gst_ffmpegdeinterlace_change_state (GstElement * element,
123     GstStateChange transition);
124
125 static GstElementClass *parent_class = NULL;
126
127 /*static guint gst_ffmpegdeinterlace_signals[LAST_SIGNAL] = { 0 }; */
128
129 static GstCaps *
130 gst_ffmpegdeinterlace_getcaps (GstPad * pad)
131 {
132   GstFFMpegDeinterlace *filter;
133   GstPad *otherpad;
134   
135   filter = GST_FFMPEGDEINTERLACE (gst_pad_get_parent (pad));
136
137   otherpad = (pad == filter->srcpad) ? filter->sinkpad : filter->srcpad;
138
139   otherpad = (pad == filter->srcpad) ? filter->sinkpad : filter->srcpad;
140
141   return gst_pad_get_allowed_caps (otherpad);
142 }
143
144 static GstPadLinkReturn
145 gst_ffmpegdeinterlace_pad_link (GstPad * pad, const GstCaps * caps)
146 {
147   GstStructure *structure;
148   AVCodecContext *ctx;
149   GstFFMpegDeinterlace *deinterlace;
150   GstPad *otherpad;
151   GstPadLinkReturn ret;
152   int height, width;
153
154   deinterlace = GST_FFMPEGDEINTERLACE (gst_pad_get_parent (pad));
155
156   otherpad = (pad == deinterlace->srcpad) ? deinterlace->sinkpad :
157       deinterlace->srcpad;
158
159   structure = gst_caps_get_structure (caps, 0);
160   ret = gst_structure_get_int (structure, "width", &width);
161   ret &= gst_structure_get_int (structure, "height", &height);
162
163   ctx = avcodec_alloc_context ();
164   ctx->width = width;
165   ctx->height = height;
166   ctx->pix_fmt = PIX_FMT_NB;
167   gst_ffmpeg_caps_with_codectype (CODEC_TYPE_VIDEO, caps, ctx);
168   if (ctx->pix_fmt == PIX_FMT_NB) {
169     av_free (ctx);
170
171     /* we disable ourself here */
172     return GST_PAD_LINK_REFUSED;
173   }
174
175   deinterlace->pixfmt = ctx->pix_fmt;
176   av_free (ctx);
177
178   ret = gst_pad_try_set_caps (otherpad, caps);
179   if (GST_PAD_LINK_FAILED (ret)) {
180     return ret;
181   }
182
183   deinterlace->width = width;
184   deinterlace->height = height;
185   deinterlace->to_size = avpicture_get_size (deinterlace->pixfmt, 
186                                              deinterlace->width, deinterlace->height);
187   return GST_PAD_LINK_OK;
188 }
189
190 static GType
191 gst_ffmpegdeinterlace_get_type (void)
192 {
193   static GType ffmpegdeinterlace_type = 0;
194
195   if (!ffmpegdeinterlace_type) {
196     static const GTypeInfo ffmpegdeinterlace_info = {
197       sizeof (GstFFMpegDeinterlaceClass),
198       (GBaseInitFunc) gst_ffmpegdeinterlace_base_init,
199       NULL,
200       (GClassInitFunc) gst_ffmpegdeinterlace_class_init,
201       NULL,
202       NULL,
203       sizeof (GstFFMpegDeinterlace),
204       0,
205       (GInstanceInitFunc) gst_ffmpegdeinterlace_init,
206     };
207
208     ffmpegdeinterlace_type = g_type_register_static (GST_TYPE_ELEMENT,
209         "GstFFMpegDeinterlace", &ffmpegdeinterlace_info, 0);
210   }
211
212   return ffmpegdeinterlace_type;
213 }
214
215 static void
216 gst_ffmpegdeinterlace_base_init (GstFFMpegDeinterlaceClass * klass)
217 {
218   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
219
220   gst_element_class_set_details (element_class, &ffmpegdeinterlace_details);
221
222   gst_element_class_add_pad_template (element_class,
223       gst_static_pad_template_get (&gst_ffmpegdeinterlace_src_template));
224   gst_element_class_add_pad_template (element_class,
225       gst_static_pad_template_get (&gst_ffmpegdeinterlace_sink_template));
226 }
227
228 static void
229 gst_ffmpegdeinterlace_class_init (GstFFMpegDeinterlaceClass * klass)
230 {
231   GObjectClass *gobject_class;
232   GstElementClass *gstelement_class;
233
234   gobject_class = (GObjectClass *) klass;
235   gstelement_class = (GstElementClass *) klass;
236
237   parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
238
239   gobject_class->set_property = gst_ffmpegdeinterlace_set_property;
240   gobject_class->get_property = gst_ffmpegdeinterlace_get_property;
241
242   gstelement_class->change_state = gst_ffmpegdeinterlace_change_state;
243 }
244
245 static void
246 gst_ffmpegdeinterlace_init (GstFFMpegDeinterlace * deinterlace)
247 {
248   deinterlace->sinkpad = gst_pad_new_from_template (gst_static_pad_template_get
249                                               (&gst_ffmpegdeinterlace_sink_template), 
250                                               "sink");
251   gst_pad_set_link_function (deinterlace->sinkpad, gst_ffmpegdeinterlace_pad_link);
252   gst_pad_set_getcaps_function (deinterlace->sinkpad, gst_ffmpegdeinterlace_getcaps);
253   gst_pad_set_chain_function (deinterlace->sinkpad, gst_ffmpegdeinterlace_chain);
254   gst_element_add_pad (GST_ELEMENT (deinterlace), deinterlace->sinkpad);
255
256   deinterlace->srcpad = gst_pad_new_from_template (gst_static_pad_template_get
257                                              (&gst_ffmpegdeinterlace_src_template), 
258                                              "src");
259   gst_element_add_pad (GST_ELEMENT (deinterlace), deinterlace->srcpad);
260   gst_pad_set_link_function (deinterlace->srcpad, gst_ffmpegdeinterlace_pad_link);
261   gst_pad_set_getcaps_function (deinterlace->srcpad, gst_ffmpegdeinterlace_getcaps);
262
263   deinterlace->pixfmt = PIX_FMT_NB;
264 }
265
266 static void
267 gst_ffmpegdeinterlace_chain (GstPad * pad, GstData * data)
268 {
269   GstBuffer *inbuf = GST_BUFFER (data);
270   GstFFMpegDeinterlace *deinterlace;
271   GstBuffer *outbuf = NULL;
272
273   g_return_if_fail (pad != NULL);
274   g_return_if_fail (GST_IS_PAD (pad));
275   g_return_if_fail (inbuf != NULL);
276
277   deinterlace = GST_FFMPEGDEINTERLACE (gst_pad_get_parent (pad));
278
279   g_return_if_fail (deinterlace != NULL);
280   g_return_if_fail (GST_IS_FFMPEGDEINTERLACE (deinterlace));
281
282   if (!GST_PAD_IS_USABLE (deinterlace->srcpad)) {
283     gst_buffer_unref (inbuf);
284     return;
285   }
286
287   outbuf = gst_pad_alloc_buffer_and_set_caps (deinterlace->srcpad, 
288                                  GST_BUFFER_OFFSET_NONE, deinterlace->to_size);
289
290   gst_ffmpeg_avpicture_fill (&deinterlace->from_frame,
291                              GST_BUFFER_DATA (inbuf),
292                              deinterlace->pixfmt, 
293                              deinterlace->width, deinterlace->height);
294
295   gst_ffmpeg_avpicture_fill (&deinterlace->to_frame,
296                              GST_BUFFER_DATA (outbuf),
297                              deinterlace->pixfmt, 
298                              deinterlace->width, deinterlace->height);    
299
300   avpicture_deinterlace (&deinterlace->to_frame, &deinterlace->from_frame,
301                          deinterlace->pixfmt, deinterlace->width, deinterlace->height);
302
303   gst_buffer_stamp (outbuf, (const GstBuffer *) inbuf);
304   
305   gst_buffer_unref (inbuf);
306   gst_pad_push (deinterlace->srcpad, GST_DATA (outbuf));
307   
308 }
309
310 static GstStateChangeReturn
311 gst_ffmpegdeinterlace_change_state (GstElement * element, GstStateChange transition)
312 {
313   GstFFMpegDeinterlace *deinterlace;
314
315   deinterlace = GST_FFMPEGDEINTERLACE (element);
316
317   switch (transition) {
318     case GST_STATE_CHANGE_READY_TO_NULL:
319       break;
320   }
321
322   if (parent_class->change_state)
323     return parent_class->change_state (element, transition);
324
325   return GST_STATE_CHANGE_SUCCESS;
326 }
327
328 static void
329 gst_ffmpegdeinterlace_set_property (GObject * object,
330     guint prop_id, const GValue * value, GParamSpec * pspec)
331 {
332   GstFFMpegDeinterlace *deinterlace;
333
334   /* it's not null if we got it, but it might not be ours */
335   g_return_if_fail (GST_IS_FFMPEGDEINTERLACE (object));
336   deinterlace = GST_FFMPEGDEINTERLACE (object);
337
338   switch (prop_id) {
339     default:
340       break;
341   }
342 }
343
344 static void
345 gst_ffmpegdeinterlace_get_property (GObject * object,
346     guint prop_id, GValue * value, GParamSpec * pspec)
347 {
348   GstFFMpegDeinterlace *deinterlace;
349
350   /* it's not null if we got it, but it might not be ours */
351   g_return_if_fail (GST_IS_FFMPEGDEINTERLACE (object));
352   deinterlace = GST_FFMPEGDEINTERLACE (object);
353
354   switch (prop_id) {
355     default:
356       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
357       break;
358   }
359 }
360
361 gboolean
362 gst_ffmpegdeinterlace_register (GstPlugin * plugin)
363 {
364
365   return gst_element_register (plugin, "ffdeinterlace",
366       GST_RANK_NONE, GST_TYPE_FFMPEGDEINTERLACE);
367 }