mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-23 10:11:08 +00:00
interlace: new element
This commit is contained in:
parent
135908dfe1
commit
7ce0ca6c89
3 changed files with 476 additions and 0 deletions
|
@ -284,6 +284,7 @@ AG_GST_CHECK_PLUGIN(geometrictransform)
|
|||
AG_GST_CHECK_PLUGIN(h264parse)
|
||||
AG_GST_CHECK_PLUGIN(hdvparse)
|
||||
AG_GST_CHECK_PLUGIN(id3tag)
|
||||
AG_GST_CHECK_PLUGIN(interlace)
|
||||
AG_GST_CHECK_PLUGIN(invtelecine)
|
||||
AG_GST_CHECK_PLUGIN(ivfparse)
|
||||
AG_GST_CHECK_PLUGIN(jpegformat)
|
||||
|
@ -1698,6 +1699,7 @@ gst/geometrictransform/Makefile
|
|||
gst/h264parse/Makefile
|
||||
gst/hdvparse/Makefile
|
||||
gst/id3tag/Makefile
|
||||
gst/interlace/Makefile
|
||||
gst/invtelecine/Makefile
|
||||
gst/ivfparse/Makefile
|
||||
gst/jpegformat/Makefile
|
||||
|
|
17
gst/interlace/Makefile.am
Normal file
17
gst/interlace/Makefile.am
Normal file
|
@ -0,0 +1,17 @@
|
|||
plugin_LTLIBRARIES = libgstinterlace.la
|
||||
|
||||
libgstinterlace_la_SOURCES = \
|
||||
gstinterlace.c
|
||||
|
||||
libgstinterlace_la_CFLAGS = \
|
||||
$(GST_CFLAGS) \
|
||||
$(GST_PLUGINS_BASE_CFLAGS)
|
||||
|
||||
libgstinterlace_la_LIBADD = \
|
||||
$(GST_LIBS) \
|
||||
$(GST_PLUGINS_BASE_LIBS) -lgstvideo-@GST_MAJORMINOR@ \
|
||||
$(LIBM)
|
||||
|
||||
libgstinterlace_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS)
|
||||
libgstinterlace_la_LIBTOOLFLAGS = --tag=disable-static
|
||||
|
457
gst/interlace/gstinterlace.c
Normal file
457
gst/interlace/gstinterlace.c
Normal file
|
@ -0,0 +1,457 @@
|
|||
/* GStreamer
|
||||
* Copyright (C) 2010 David A. Schleef <ds@schleef.org>
|
||||
*
|
||||
* This library is free software; you can redistribute it and/or
|
||||
* modify it under the terms of the GNU Library General Public
|
||||
* License as published by the Free Software Foundation; either
|
||||
* version 2 of the License, or (at your option) any later version.
|
||||
*
|
||||
* This library is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* Library General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Library General Public
|
||||
* License along with this library; if not, write to the
|
||||
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
|
||||
* Boston, MA 02111-1307, USA.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include <gst/gst.h>
|
||||
#include <gst/video/video.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <math.h>
|
||||
|
||||
GST_DEBUG_CATEGORY (gst_interlace_debug);
|
||||
#define GST_CAT_DEFAULT gst_interlace_debug
|
||||
|
||||
#define GST_TYPE_INTERLACE \
|
||||
(gst_interlace_get_type())
|
||||
#define GST_INTERLACE(obj) \
|
||||
(G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_INTERLACE,GstInterlace))
|
||||
#define GST_INTERLACE_DEC_CLASS(klass) \
|
||||
(G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_INTERLACE,GstInterlaceClass))
|
||||
#define GST_IS_GST_INTERLACE(obj) \
|
||||
(G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_INTERLACE))
|
||||
#define GST_IS_GST_INTERLACE_CLASS(obj) \
|
||||
(G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_INTERLACE))
|
||||
|
||||
typedef struct _GstInterlace GstInterlace;
|
||||
typedef struct _GstInterlaceClass GstInterlaceClass;
|
||||
|
||||
struct _GstInterlace
|
||||
{
|
||||
GstElement element;
|
||||
|
||||
GstPad *srcpad;
|
||||
GstPad *sinkpad;
|
||||
GstCaps *srccaps;
|
||||
|
||||
/* properties */
|
||||
gboolean top_field_first;
|
||||
|
||||
/* state */
|
||||
int width;
|
||||
int height;
|
||||
GstVideoFormat format;
|
||||
GstBuffer *stored_frame;
|
||||
|
||||
};
|
||||
|
||||
struct _GstInterlaceClass
|
||||
{
|
||||
GstElementClass element_class;
|
||||
|
||||
};
|
||||
|
||||
enum
|
||||
{
|
||||
ARG_0,
|
||||
PROP_TOP_FIELD_FIRST
|
||||
};
|
||||
|
||||
static GstStaticPadTemplate gst_interlace_src_template =
|
||||
GST_STATIC_PAD_TEMPLATE ("src",
|
||||
GST_PAD_SRC,
|
||||
GST_PAD_ALWAYS,
|
||||
GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("{YUY2,UYVY,I420,YV12}")
|
||||
",interlaced=TRUE")
|
||||
);
|
||||
|
||||
static GstStaticPadTemplate gst_interlace_sink_template =
|
||||
GST_STATIC_PAD_TEMPLATE ("sink",
|
||||
GST_PAD_SINK,
|
||||
GST_PAD_ALWAYS,
|
||||
GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("{YUY2,UYVY,I420,YV12}")
|
||||
",interlaced=FALSE")
|
||||
);
|
||||
|
||||
static void gst_interlace_base_init (gpointer g_class);
|
||||
static void gst_interlace_class_init (GstInterlaceClass * klass);
|
||||
static void gst_interlace_init (GstInterlace * interlace);
|
||||
static GstFlowReturn gst_interlace_chain (GstPad * pad, GstBuffer * buffer);
|
||||
|
||||
static void gst_interlace_set_property (GObject * object,
|
||||
guint prop_id, const GValue * value, GParamSpec * pspec);
|
||||
static void gst_interlace_get_property (GObject * object,
|
||||
guint prop_id, GValue * value, GParamSpec * pspec);
|
||||
|
||||
static gboolean gst_interlace_setcaps (GstPad * pad, GstCaps * caps);
|
||||
static GstCaps *gst_interlace_getcaps (GstPad * pad);
|
||||
static GstStateChangeReturn gst_interlace_change_state (GstElement * element,
|
||||
GstStateChange transition);
|
||||
|
||||
static GstElementClass *parent_class = NULL;
|
||||
|
||||
|
||||
static GType
|
||||
gst_interlace_get_type (void)
|
||||
{
|
||||
static GType interlace_type = 0;
|
||||
|
||||
if (!interlace_type) {
|
||||
static const GTypeInfo interlace_info = {
|
||||
sizeof (GstInterlaceClass),
|
||||
gst_interlace_base_init,
|
||||
NULL,
|
||||
(GClassInitFunc) gst_interlace_class_init,
|
||||
NULL,
|
||||
NULL,
|
||||
sizeof (GstInterlace),
|
||||
0,
|
||||
(GInstanceInitFunc) gst_interlace_init,
|
||||
};
|
||||
|
||||
interlace_type = g_type_register_static (GST_TYPE_ELEMENT,
|
||||
"GstInterlace", &interlace_info, 0);
|
||||
}
|
||||
|
||||
return interlace_type;
|
||||
}
|
||||
|
||||
static void
|
||||
gst_interlace_base_init (gpointer g_class)
|
||||
{
|
||||
|
||||
GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
|
||||
|
||||
gst_element_class_set_details_simple (element_class,
|
||||
"Inverse Telecine filter", "Filter/Video",
|
||||
"Detects and reconstructs progressive content from telecine video",
|
||||
"Entropy Wave <ds@entropywave.com>");
|
||||
|
||||
gst_element_class_add_pad_template (element_class,
|
||||
gst_static_pad_template_get (&gst_interlace_sink_template));
|
||||
gst_element_class_add_pad_template (element_class,
|
||||
gst_static_pad_template_get (&gst_interlace_src_template));
|
||||
}
|
||||
|
||||
static void
|
||||
gst_interlace_class_init (GstInterlaceClass * klass)
|
||||
{
|
||||
GObjectClass *object_class = G_OBJECT_CLASS (klass);
|
||||
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
|
||||
|
||||
parent_class = g_type_class_peek_parent (klass);
|
||||
|
||||
object_class->set_property = gst_interlace_set_property;
|
||||
object_class->get_property = gst_interlace_get_property;
|
||||
|
||||
element_class->change_state = gst_interlace_change_state;
|
||||
|
||||
g_object_class_install_property (object_class, PROP_TOP_FIELD_FIRST,
|
||||
g_param_spec_boolean ("top-field-first", "top field first",
|
||||
"Interlaced stream should be top field first", FALSE,
|
||||
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
||||
|
||||
}
|
||||
|
||||
static void
|
||||
gst_interlace_init (GstInterlace * interlace)
|
||||
{
|
||||
GST_DEBUG ("gst_interlace_init");
|
||||
interlace->sinkpad =
|
||||
gst_pad_new_from_static_template (&gst_interlace_sink_template, "sink");
|
||||
gst_element_add_pad (GST_ELEMENT (interlace), interlace->sinkpad);
|
||||
gst_pad_set_chain_function (interlace->sinkpad, gst_interlace_chain);
|
||||
gst_pad_set_setcaps_function (interlace->sinkpad, gst_interlace_setcaps);
|
||||
gst_pad_set_getcaps_function (interlace->sinkpad, gst_interlace_getcaps);
|
||||
|
||||
interlace->srcpad =
|
||||
gst_pad_new_from_static_template (&gst_interlace_src_template, "src");
|
||||
gst_element_add_pad (GST_ELEMENT (interlace), interlace->srcpad);
|
||||
gst_pad_set_setcaps_function (interlace->srcpad, gst_interlace_setcaps);
|
||||
gst_pad_set_getcaps_function (interlace->srcpad, gst_interlace_getcaps);
|
||||
|
||||
interlace->top_field_first = FALSE;
|
||||
}
|
||||
|
||||
static GstCaps *
|
||||
gst_interlace_getcaps (GstPad * pad)
|
||||
{
|
||||
GstInterlace *interlace;
|
||||
GstPad *otherpad;
|
||||
GstCaps *othercaps;
|
||||
GstCaps *icaps;
|
||||
GstStructure *structure;
|
||||
int i;
|
||||
|
||||
interlace = GST_INTERLACE (gst_pad_get_parent (pad));
|
||||
|
||||
otherpad =
|
||||
(pad == interlace->srcpad) ? interlace->sinkpad : interlace->srcpad;
|
||||
|
||||
othercaps = gst_pad_peer_get_caps (otherpad);
|
||||
if (othercaps == NULL) {
|
||||
icaps = gst_caps_copy (gst_pad_get_pad_template_caps (otherpad));
|
||||
} else {
|
||||
icaps = gst_caps_intersect (othercaps,
|
||||
gst_pad_get_pad_template_caps (otherpad));
|
||||
}
|
||||
|
||||
for (i = 0; i < gst_caps_get_size (icaps); i++) {
|
||||
structure = gst_caps_get_structure (icaps, i);
|
||||
|
||||
if (pad == interlace->srcpad) {
|
||||
gst_structure_set (structure, "interlaced", G_TYPE_BOOLEAN, TRUE, NULL);
|
||||
} else {
|
||||
gst_structure_set (structure, "interlaced", G_TYPE_BOOLEAN, FALSE, NULL);
|
||||
}
|
||||
}
|
||||
|
||||
return icaps;
|
||||
}
|
||||
|
||||
static gboolean
|
||||
gst_interlace_setcaps (GstPad * pad, GstCaps * caps)
|
||||
{
|
||||
GstInterlace *interlace;
|
||||
gboolean ret;
|
||||
int width, height;
|
||||
GstVideoFormat format;
|
||||
gboolean interlaced = TRUE;
|
||||
int fps_n, fps_d;
|
||||
GstPad *otherpad;
|
||||
GstCaps *othercaps;
|
||||
GstStructure *structure;
|
||||
|
||||
interlace = GST_INTERLACE (gst_pad_get_parent (pad));
|
||||
|
||||
otherpad =
|
||||
(pad == interlace->srcpad) ? interlace->sinkpad : interlace->srcpad;
|
||||
|
||||
ret = gst_video_format_parse_caps (caps, &format, &width, &height);
|
||||
gst_video_format_parse_caps_interlaced (caps, &interlaced);
|
||||
ret &= gst_video_parse_caps_framerate (caps, &fps_n, &fps_d);
|
||||
|
||||
if (!ret)
|
||||
goto error;
|
||||
|
||||
othercaps = gst_caps_copy (caps);
|
||||
|
||||
structure = gst_caps_get_structure (othercaps, 0);
|
||||
|
||||
if (pad == interlace->srcpad) {
|
||||
gst_structure_set (structure,
|
||||
"interlaced", G_TYPE_BOOLEAN, FALSE,
|
||||
"framerate", GST_TYPE_FRACTION, fps_n * 2, fps_d, NULL);
|
||||
} else {
|
||||
gst_structure_set (structure,
|
||||
"interlaced", G_TYPE_BOOLEAN, TRUE,
|
||||
"framerate", GST_TYPE_FRACTION, fps_n, fps_d * 2, NULL);
|
||||
}
|
||||
|
||||
ret = gst_pad_set_caps (otherpad, othercaps);
|
||||
if (!ret)
|
||||
goto error;
|
||||
|
||||
interlace->format = format;
|
||||
interlace->width = width;
|
||||
interlace->height = height;
|
||||
|
||||
if (pad == interlace->sinkpad) {
|
||||
interlace->srccaps = gst_caps_ref (othercaps);
|
||||
} else {
|
||||
interlace->srccaps = gst_caps_ref (caps);
|
||||
}
|
||||
|
||||
error:
|
||||
g_object_unref (interlace);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static void
|
||||
copy_field (GstInterlace * interlace, GstBuffer * d, GstBuffer * s,
|
||||
int field_index)
|
||||
{
|
||||
int j;
|
||||
guint8 *dest;
|
||||
guint8 *src;
|
||||
int width = interlace->width;
|
||||
int height = interlace->height;
|
||||
|
||||
if (interlace->format == GST_VIDEO_FORMAT_I420 ||
|
||||
interlace->format == GST_VIDEO_FORMAT_YV12) {
|
||||
/* planar 4:2:0 */
|
||||
for (j = field_index; j < height; j += 2) {
|
||||
dest = GST_BUFFER_DATA (d) + j * width;
|
||||
src = GST_BUFFER_DATA (s) + j * width;
|
||||
memcpy (dest, src, width);
|
||||
}
|
||||
for (j = field_index; j < height / 2; j += 2) {
|
||||
dest = GST_BUFFER_DATA (d) + width * height + j * width / 2;
|
||||
src = GST_BUFFER_DATA (s) + width * height + j * width / 2;
|
||||
memcpy (dest, src, width / 2);
|
||||
}
|
||||
for (j = field_index; j < height / 2; j += 2) {
|
||||
dest =
|
||||
GST_BUFFER_DATA (d) + width * height + width / 2 * height / 2 +
|
||||
j * width / 2;
|
||||
src =
|
||||
GST_BUFFER_DATA (s) + width * height + width / 2 * height / 2 +
|
||||
j * width / 2;
|
||||
memcpy (dest, src, width / 2);
|
||||
}
|
||||
} else {
|
||||
/* packed 4:2:2 */
|
||||
for (j = field_index; j < height; j += 2) {
|
||||
dest = GST_BUFFER_DATA (d) + j * width * 2;
|
||||
src = GST_BUFFER_DATA (s) + j * width * 2;
|
||||
memcpy (dest, src, width * 2);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
static GstFlowReturn
|
||||
gst_interlace_chain (GstPad * pad, GstBuffer * buffer)
|
||||
{
|
||||
GstInterlace *interlace = GST_INTERLACE (gst_pad_get_parent (pad));
|
||||
GstFlowReturn ret;
|
||||
|
||||
GST_DEBUG ("Received buffer at %u:%02u:%02u:%09u",
|
||||
(guint) (GST_BUFFER_TIMESTAMP (buffer) / (GST_SECOND * 60 * 60)),
|
||||
(guint) ((GST_BUFFER_TIMESTAMP (buffer) / (GST_SECOND * 60)) % 60),
|
||||
(guint) ((GST_BUFFER_TIMESTAMP (buffer) / GST_SECOND) % 60),
|
||||
(guint) (GST_BUFFER_TIMESTAMP (buffer) % GST_SECOND));
|
||||
|
||||
GST_DEBUG ("duration %" GST_TIME_FORMAT " flags %04x %s %s %s",
|
||||
GST_TIME_ARGS (GST_BUFFER_DURATION (buffer)),
|
||||
GST_BUFFER_FLAGS (buffer),
|
||||
(GST_BUFFER_FLAGS (buffer) & GST_VIDEO_BUFFER_TFF) ? "tff" : "",
|
||||
(GST_BUFFER_FLAGS (buffer) & GST_VIDEO_BUFFER_RFF) ? "rff" : "",
|
||||
(GST_BUFFER_FLAGS (buffer) & GST_VIDEO_BUFFER_ONEFIELD) ? "onefield" :
|
||||
"");
|
||||
|
||||
if (GST_BUFFER_FLAGS (buffer) & GST_BUFFER_FLAG_DISCONT) {
|
||||
GST_ERROR ("discont");
|
||||
|
||||
}
|
||||
|
||||
if (interlace->stored_frame == NULL) {
|
||||
interlace->stored_frame = buffer;
|
||||
ret = GST_FLOW_OK;
|
||||
} else {
|
||||
GstBuffer *output_buffer;
|
||||
|
||||
output_buffer = gst_buffer_new_and_alloc (GST_BUFFER_SIZE (buffer));
|
||||
|
||||
copy_field (interlace, output_buffer, interlace->stored_frame,
|
||||
!interlace->top_field_first);
|
||||
copy_field (interlace, output_buffer, buffer, interlace->top_field_first);
|
||||
|
||||
GST_BUFFER_TIMESTAMP (output_buffer) =
|
||||
GST_BUFFER_TIMESTAMP (interlace->stored_frame);
|
||||
GST_BUFFER_DURATION (output_buffer) =
|
||||
GST_BUFFER_DURATION (interlace->stored_frame) +
|
||||
GST_BUFFER_DURATION (buffer);
|
||||
gst_buffer_set_caps (output_buffer, interlace->srccaps);
|
||||
|
||||
if (interlace->top_field_first) {
|
||||
GST_BUFFER_FLAG_SET (output_buffer, GST_VIDEO_BUFFER_TFF);
|
||||
}
|
||||
|
||||
ret = gst_pad_push (interlace->srcpad, output_buffer);
|
||||
|
||||
gst_buffer_unref (buffer);
|
||||
gst_buffer_unref (interlace->stored_frame);
|
||||
interlace->stored_frame = NULL;
|
||||
}
|
||||
|
||||
gst_object_unref (interlace);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static void
|
||||
gst_interlace_set_property (GObject * object,
|
||||
guint prop_id, const GValue * value, GParamSpec * pspec)
|
||||
{
|
||||
GstInterlace *interlace = GST_INTERLACE (object);
|
||||
|
||||
switch (prop_id) {
|
||||
case PROP_TOP_FIELD_FIRST:
|
||||
interlace->top_field_first = g_value_get_boolean (value);
|
||||
break;
|
||||
default:
|
||||
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
gst_interlace_get_property (GObject * object,
|
||||
guint prop_id, GValue * value, GParamSpec * pspec)
|
||||
{
|
||||
GstInterlace *interlace = GST_INTERLACE (object);
|
||||
|
||||
switch (prop_id) {
|
||||
case PROP_TOP_FIELD_FIRST:
|
||||
g_value_set_boolean (value, interlace->top_field_first);
|
||||
break;
|
||||
default:
|
||||
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static GstStateChangeReturn
|
||||
gst_interlace_change_state (GstElement * element, GstStateChange transition)
|
||||
{
|
||||
//GstInterlace *interlace = GST_INTERLACE (element);
|
||||
|
||||
switch (transition) {
|
||||
case GST_STATE_CHANGE_PAUSED_TO_READY:
|
||||
//gst_interlace_reset (interlace);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
if (parent_class->change_state)
|
||||
return parent_class->change_state (element, transition);
|
||||
|
||||
return GST_STATE_CHANGE_SUCCESS;
|
||||
}
|
||||
|
||||
static gboolean
|
||||
plugin_init (GstPlugin * plugin)
|
||||
{
|
||||
GST_DEBUG_CATEGORY_INIT (gst_interlace_debug, "interlace", 0,
|
||||
"interlace element");
|
||||
|
||||
return gst_element_register (plugin, "interlace", GST_RANK_NONE,
|
||||
GST_TYPE_INTERLACE);
|
||||
}
|
||||
|
||||
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
|
||||
GST_VERSION_MINOR,
|
||||
"interlace",
|
||||
"Create an interlaced video stream",
|
||||
plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)
|
Loading…
Reference in a new issue