ffmpeg: CodecType => AVMediaType
[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/Effect/Video/Deinterlace",
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   GstCaps *src_caps;
116   gboolean ret;
117
118   if (!gst_structure_get_int (structure, "width", &deinterlace->width))
119     return FALSE;
120   if (!gst_structure_get_int (structure, "height", &deinterlace->height))
121     return FALSE;
122
123   ctx = avcodec_alloc_context ();
124   ctx->width = deinterlace->width;
125   ctx->height = deinterlace->height;
126   ctx->pix_fmt = PIX_FMT_NB;
127   gst_ffmpeg_caps_with_codectype (AVMEDIA_TYPE_VIDEO, caps, ctx);
128   if (ctx->pix_fmt == PIX_FMT_NB) {
129     av_free (ctx);
130     return FALSE;
131   }
132
133   deinterlace->pixfmt = ctx->pix_fmt;
134
135   av_free (ctx);
136
137   deinterlace->to_size =
138       avpicture_get_size (deinterlace->pixfmt, deinterlace->width,
139       deinterlace->height);
140
141   src_caps = gst_caps_copy (caps);
142   gst_caps_set_simple (src_caps, "interlaced", G_TYPE_BOOLEAN, FALSE, NULL);
143   ret = gst_pad_set_caps (deinterlace->srcpad, src_caps);
144   gst_caps_unref (src_caps);
145
146   return ret;
147 }
148
149 static void
150 gst_ffmpegdeinterlace_init (GstFFMpegDeinterlace * deinterlace,
151     GstFFMpegDeinterlaceClass * klass)
152 {
153   deinterlace->sinkpad =
154       gst_pad_new_from_static_template (&sink_factory, "sink");
155   gst_pad_set_setcaps_function (deinterlace->sinkpad,
156       gst_ffmpegdeinterlace_sink_setcaps);
157   gst_pad_set_chain_function (deinterlace->sinkpad,
158       gst_ffmpegdeinterlace_chain);
159   gst_element_add_pad (GST_ELEMENT (deinterlace), deinterlace->sinkpad);
160
161   deinterlace->srcpad = gst_pad_new_from_static_template (&src_factory, "src");
162   gst_element_add_pad (GST_ELEMENT (deinterlace), deinterlace->srcpad);
163
164   deinterlace->pixfmt = PIX_FMT_NB;
165 }
166
167 static GstFlowReturn
168 gst_ffmpegdeinterlace_chain (GstPad * pad, GstBuffer * inbuf)
169 {
170   GstFFMpegDeinterlace *deinterlace =
171       GST_FFMPEGDEINTERLACE (gst_pad_get_parent (pad));
172   GstBuffer *outbuf = NULL;
173   GstFlowReturn result;
174
175   result =
176       gst_pad_alloc_buffer (deinterlace->srcpad, GST_BUFFER_OFFSET_NONE,
177       deinterlace->to_size, GST_PAD_CAPS (deinterlace->srcpad), &outbuf);
178   if (result == GST_FLOW_OK) {
179     gst_ffmpeg_avpicture_fill (&deinterlace->from_frame,
180         GST_BUFFER_DATA (inbuf), deinterlace->pixfmt, deinterlace->width,
181         deinterlace->height);
182
183     gst_ffmpeg_avpicture_fill (&deinterlace->to_frame, GST_BUFFER_DATA (outbuf),
184         deinterlace->pixfmt, deinterlace->width, deinterlace->height);
185
186     avpicture_deinterlace (&deinterlace->to_frame, &deinterlace->from_frame,
187         deinterlace->pixfmt, deinterlace->width, deinterlace->height);
188
189     gst_buffer_copy_metadata (outbuf, inbuf, GST_BUFFER_COPY_TIMESTAMPS);
190
191     result = gst_pad_push (deinterlace->srcpad, outbuf);
192   }
193
194   gst_buffer_unref (inbuf);
195
196   return result;
197 }
198
199 gboolean
200 gst_ffmpegdeinterlace_register (GstPlugin * plugin)
201 {
202   return gst_element_register (plugin, "ffdeinterlace",
203       GST_RANK_NONE, GST_TYPE_FFMPEGDEINTERLACE);
204 }