#include "config.h"
#endif
#include <string.h>
+#include <math.h>
#include <gst/gst.h>
+#include <gst/video/video.h>
#include "gsty4mencode.h"
-static GstElementDetails lavencode_details = {
- "LavEncode",
+static GstElementDetails y4mencode_details = {
+ "Y4mEncode",
"Codec/Video/Encoder",
"LGPL",
- "Encodes a YUV frame into the lav format (mjpeg_tools)",
+ "Encodes a YUV frame into the yuv4mpeg format (mjpegtools)",
VERSION,
"Wim Taymans <wim.taymans@chello.be>",
"(C) 2001",
ARG_0
};
-GST_PAD_TEMPLATE_FACTORY (lavencode_src_factory,
+GST_PAD_TEMPLATE_FACTORY (y4mencode_src_factory,
"src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_CAPS_NEW (
"test_src",
- "application/x-lav",
- NULL
+ "application/x-yuv4mpeg",
+ "y4mversion", GST_PROPS_INT (1)
)
)
-GST_PAD_TEMPLATE_FACTORY (lavencode_sink_factory,
+GST_PAD_TEMPLATE_FACTORY (y4mencode_sink_factory,
"sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
- GST_CAPS_NEW (
+ gst_caps_new (
"test_src",
- "video/raw",
- "format", GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")),
- "width", GST_PROPS_INT_RANGE (0, G_MAXINT),
- "height", GST_PROPS_INT_RANGE (0, G_MAXINT)
+ "video/x-raw-yuv",
+ GST_VIDEO_YUV_PAD_TEMPLATE_PROPS (
+ GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")))
)
)
-static void gst_lavencode_class_init (GstLavEncodeClass *klass);
-static void gst_lavencode_init (GstLavEncode *filter);
+static void gst_y4mencode_class_init (GstY4mEncodeClass *klass);
+static void gst_y4mencode_init (GstY4mEncode *filter);
-static void gst_lavencode_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec);
-static void gst_lavencode_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec);
+static void gst_y4mencode_set_property (GObject *object,
+ guint prop_id,
+ const GValue *value,
+ GParamSpec *pspec);
+static void gst_y4mencode_get_property (GObject *object,
+ guint prop_id,
+ GValue *value,
+ GParamSpec *pspec);
-static void gst_lavencode_chain (GstPad *pad, GstBuffer *buf);
-static GstElementStateReturn gst_lavencode_change_state (GstElement *element);
+static void gst_y4mencode_chain (GstPad *pad,
+ GstBuffer *buf);
+static GstElementStateReturn
+ gst_y4mencode_change_state (GstElement *element);
static GstElementClass *parent_class = NULL;
/*static guint gst_filter_signals[LAST_SIGNAL] = { 0 }; */
GType
-gst_lavencode_get_type(void) {
- static GType lavencode_type = 0;
+gst_y4mencode_get_type(void) {
+ static GType y4mencode_type = 0;
- if (!lavencode_type) {
- static const GTypeInfo lavencode_info = {
- sizeof(GstLavEncodeClass), NULL,
+ if (!y4mencode_type) {
+ static const GTypeInfo y4mencode_info = {
+ sizeof(GstY4mEncodeClass),
+ NULL,
NULL,
- (GClassInitFunc)gst_lavencode_class_init,
+ (GClassInitFunc)gst_y4mencode_class_init,
NULL,
NULL,
- sizeof(GstLavEncode),
+ sizeof(GstY4mEncode),
0,
- (GInstanceInitFunc)gst_lavencode_init,
+ (GInstanceInitFunc)gst_y4mencode_init,
};
- lavencode_type = g_type_register_static(GST_TYPE_ELEMENT, "GstLavEncode", &lavencode_info, 0);
+ y4mencode_type = g_type_register_static(GST_TYPE_ELEMENT,
+ "GstY4mEncode",
+ &y4mencode_info, 0);
}
- return lavencode_type;
+ return y4mencode_type;
}
static void
-gst_lavencode_class_init (GstLavEncodeClass *klass)
+gst_y4mencode_class_init (GstY4mEncodeClass *klass)
{
GObjectClass *gobject_class;
GstElementClass *gstelement_class;
parent_class = g_type_class_ref(GST_TYPE_ELEMENT);
- gstelement_class->change_state = gst_lavencode_change_state;
+ gstelement_class->change_state = gst_y4mencode_change_state;
- gobject_class->set_property = gst_lavencode_set_property;
- gobject_class->get_property = gst_lavencode_get_property;
+ gobject_class->set_property = gst_y4mencode_set_property;
+ gobject_class->get_property = gst_y4mencode_get_property;
}
static GstPadLinkReturn
-gst_lavencode_sinkconnect (GstPad *pad, GstCaps *caps)
+gst_y4mencode_sinkconnect (GstPad *pad, GstCaps *caps)
{
- GstLavEncode *filter;
-
- filter = GST_LAVENCODE (gst_pad_get_parent (pad));
+ GstY4mEncode *filter;
+ gint idx = -1, i;
+ gfloat fps;
+ float framerates[] = {
+ 00.000,
+ 23.976, 24.000, /* 24fps movie */
+ 25.000, /* PAL */
+ 29.970, 30.000, /* NTSC */
+ 50.000,
+ 59.940, 60.000
+ };
+
+ filter = GST_Y4MENCODE (gst_pad_get_parent (pad));
if (!GST_CAPS_IS_FIXED (caps))
return GST_PAD_LINK_DELAYED;
gst_caps_get_int (caps, "width", &filter->width);
gst_caps_get_int (caps, "height", &filter->height);
+ gst_caps_get_float (caps, "framerate", &fps);
+
+ /* find fps idx */
+ for (i = 1; i < 9; i++) {
+ if (idx == -1) {
+ idx = i;
+ } else {
+ gfloat old_diff = fabs(framerates[idx] - fps),
+ new_diff = fabs(framerates[i] - fps);
+
+ if (new_diff < old_diff) {
+ idx = i;
+ }
+ }
+ }
+ filter->fps_idx = idx;
return GST_PAD_LINK_OK;
}
static void
-gst_lavencode_init (GstLavEncode *filter)
+gst_y4mencode_init (GstY4mEncode *filter)
{
filter->sinkpad = gst_pad_new_from_template(
- GST_PAD_TEMPLATE_GET (lavencode_sink_factory), "sink");
+ GST_PAD_TEMPLATE_GET (y4mencode_sink_factory), "sink");
gst_element_add_pad (GST_ELEMENT (filter), filter->sinkpad);
- gst_pad_set_chain_function (filter->sinkpad, gst_lavencode_chain);
- gst_pad_set_link_function (filter->sinkpad, gst_lavencode_sinkconnect);
+ gst_pad_set_chain_function (filter->sinkpad, gst_y4mencode_chain);
+ gst_pad_set_link_function (filter->sinkpad, gst_y4mencode_sinkconnect);
filter->srcpad = gst_pad_new_from_template(
- GST_PAD_TEMPLATE_GET (lavencode_src_factory), "src");
+ GST_PAD_TEMPLATE_GET (y4mencode_src_factory), "src");
gst_element_add_pad (GST_ELEMENT (filter), filter->srcpad);
filter->init = TRUE;
}
static void
-gst_lavencode_chain (GstPad *pad,GstBuffer *buf)
+gst_y4mencode_chain (GstPad *pad,GstBuffer *buf)
{
- GstLavEncode *filter;
+ GstY4mEncode *filter;
GstBuffer* outbuf;
gchar *header;
gint len;
g_return_if_fail(GST_IS_PAD(pad));
g_return_if_fail(buf != NULL);
- filter = GST_LAVENCODE (gst_pad_get_parent (pad));
+ filter = GST_Y4MENCODE (gst_pad_get_parent (pad));
g_return_if_fail(filter != NULL);
- g_return_if_fail(GST_IS_LAVENCODE(filter));
+ g_return_if_fail(GST_IS_Y4MENCODE(filter));
outbuf = gst_buffer_new ();
GST_BUFFER_DATA (outbuf) = g_malloc (GST_BUFFER_SIZE (buf) + 256);
header = "FRAME\n";
}
- snprintf (GST_BUFFER_DATA (outbuf), 255, header, filter->width, filter->height, 3);
+ snprintf (GST_BUFFER_DATA (outbuf), 255, header,
+ filter->width, filter->height, filter->fps_idx);
len = strlen (GST_BUFFER_DATA (outbuf));
memcpy (GST_BUFFER_DATA (outbuf) + len, GST_BUFFER_DATA (buf), GST_BUFFER_SIZE (buf));
}
static void
-gst_lavencode_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
+gst_y4mencode_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
{
- GstLavEncode *filter;
+ GstY4mEncode *filter;
/* it's not null if we got it, but it might not be ours */
- g_return_if_fail(GST_IS_LAVENCODE(object));
- filter = GST_LAVENCODE(object);
+ g_return_if_fail(GST_IS_Y4MENCODE(object));
+ filter = GST_Y4MENCODE(object);
switch (prop_id) {
default:
}
static void
-gst_lavencode_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
+gst_y4mencode_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
{
- GstLavEncode *filter;
+ GstY4mEncode *filter;
/* it's not null if we got it, but it might not be ours */
- g_return_if_fail(GST_IS_LAVENCODE(object));
- filter = GST_LAVENCODE(object);
+ g_return_if_fail(GST_IS_Y4MENCODE(object));
+ filter = GST_Y4MENCODE(object);
switch (prop_id) {
default:
}
static GstElementStateReturn
-gst_lavencode_change_state (GstElement *element)
+gst_y4mencode_change_state (GstElement *element)
{
- GstLavEncode *filter;
+ GstY4mEncode *filter;
- g_return_val_if_fail (GST_IS_LAVENCODE (element), GST_STATE_FAILURE);
+ g_return_val_if_fail (GST_IS_Y4MENCODE (element), GST_STATE_FAILURE);
- filter = GST_LAVENCODE(element);
+ filter = GST_Y4MENCODE(element);
if (GST_STATE_TRANSITION (element) == GST_STATE_NULL_TO_READY) {
filter->init = TRUE;
{
GstElementFactory *factory;
- factory = gst_element_factory_new("lavenc",GST_TYPE_LAVENCODE,
- &lavencode_details);
+ factory = gst_element_factory_new("y4menc",GST_TYPE_Y4MENCODE,
+ &y4mencode_details);
g_return_val_if_fail(factory != NULL, FALSE);
gst_element_factory_add_pad_template (factory,
- GST_PAD_TEMPLATE_GET (lavencode_src_factory));
+ GST_PAD_TEMPLATE_GET (y4mencode_src_factory));
gst_element_factory_add_pad_template (factory,
- GST_PAD_TEMPLATE_GET (lavencode_sink_factory));
+ GST_PAD_TEMPLATE_GET (y4mencode_sink_factory));
gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));
GstPluginDesc plugin_desc = {
GST_VERSION_MAJOR,
GST_VERSION_MINOR,
- "lavenc",
+ "y4menc",
plugin_init
};
*/
-#ifndef __GST_LAVENCODE_H__
-#define __GST_LAVENCODE_H__
+#ifndef __GST_Y4MENCODE_H__
+#define __GST_Y4MENCODE_H__
#include <config.h>
#endif /* __cplusplus */
-#define GST_TYPE_LAVENCODE \
- (gst_lavencode_get_type())
-#define GST_LAVENCODE(obj) \
- (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_LAVENCODE,GstLavEncode))
-#define GST_LAVENCODE_CLASS(klass) \
- (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_ULAW,GstLavEncode))
-#define GST_IS_LAVENCODE(obj) \
- (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_LAVENCODE))
-#define GST_IS_LAVENCODE_CLASS(obj) \
- (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_LAVENCODE))
+#define GST_TYPE_Y4MENCODE \
+ (gst_y4mencode_get_type())
+#define GST_Y4MENCODE(obj) \
+ (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_Y4MENCODE,GstY4mEncode))
+#define GST_Y4MENCODE_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_ULAW,GstY4mEncode))
+#define GST_IS_Y4MENCODE(obj) \
+ (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_Y4MENCODE))
+#define GST_IS_Y4MENCODE_CLASS(obj) \
+ (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_Y4MENCODE))
-typedef struct _GstLavEncode GstLavEncode;
-typedef struct _GstLavEncodeClass GstLavEncodeClass;
+typedef struct _GstY4mEncode GstY4mEncode;
+typedef struct _GstY4mEncodeClass GstY4mEncodeClass;
-struct _GstLavEncode {
+struct _GstY4mEncode {
GstElement element;
GstPad *sinkpad,*srcpad;
gint width, height;
+ gfloat fps_idx;
gboolean init;
};
-struct _GstLavEncodeClass {
+struct _GstY4mEncodeClass {
GstElementClass parent_class;
};
-GType gst_lavencode_get_type(void);
+GType gst_y4mencode_get_type(void);
#ifdef __cplusplus
}
#endif /* __cplusplus */
-#endif /* __GST_STEREO_H__ */
+#endif /* __GST_Y4MENCODE_H__ */