gstreamer/gst/y4m/gsty4mencode.c

315 lines
8.2 KiB
C

/* GStreamer
* Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
* Copyright (C) <2006> Mark Nauwelaerts <mnauw@users.sourceforge.net>
*
* 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., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
/**
* SECTION:element-y4menc
*
* <refsect2>
* <para>
* Creates a YU4MPEG2 raw video stream as defined by the mjpegtools project.
* </para>
* <title>Example launch line</title>
* <para>
* (write everything in one line, without the backslash characters)
* <programlisting>
* gst-launch-1.0 videotestsrc num-buffers=250 \
* ! 'video/x-raw,format=(string)I420,width=320,height=240,framerate=(fraction)25/1' \
* ! y4menc ! filesink location=test.yuv
* </programlisting>
* </para>
* </refsect2>
*
*/
/* see mjpegtools/yuv4mpeg.h for yuv4mpeg format */
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <string.h>
#include <gst/gst.h>
#include <gst/video/video.h>
#include "gsty4mencode.h"
/* Filter signals and args */
enum
{
/* FILL ME */
LAST_SIGNAL
};
enum
{
PROP_0
};
static GstStaticPadTemplate y4mencode_src_factory =
GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS ("application/x-yuv4mpeg, " "y4mversion = (int) 2")
);
static GstStaticPadTemplate y4mencode_sink_factory =
GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ IYUV, I420, Y42B, Y41B, Y444 }"))
);
static void gst_y4m_encode_reset (GstY4mEncode * filter);
static GstStateChangeReturn gst_y4m_encode_change_state (GstElement * element,
GstStateChange transition);
static GstFlowReturn
gst_y4m_encode_handle_frame (GstVideoEncoder * encoder,
GstVideoCodecFrame * frame);
static gboolean gst_y4m_encode_set_format (GstVideoEncoder * encoder,
GstVideoCodecState * state);
#define gst_y4m_encode_parent_class parent_class
G_DEFINE_TYPE (GstY4mEncode, gst_y4m_encode, GST_TYPE_VIDEO_ENCODER);
static void
gst_y4m_encode_class_init (GstY4mEncodeClass * klass)
{
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
GstVideoEncoderClass *venc_class = GST_VIDEO_ENCODER_CLASS (klass);
element_class->change_state = GST_DEBUG_FUNCPTR (gst_y4m_encode_change_state);
gst_element_class_add_static_pad_template (element_class,
&y4mencode_src_factory);
gst_element_class_add_static_pad_template (element_class,
&y4mencode_sink_factory);
gst_element_class_set_static_metadata (element_class,
"YUV4MPEG video encoder", "Codec/Encoder/Video",
"Encodes a YUV frame into the yuv4mpeg format (mjpegtools)",
"Wim Taymans <wim.taymans@gmail.com>");
venc_class->set_format = gst_y4m_encode_set_format;
venc_class->handle_frame = gst_y4m_encode_handle_frame;
}
static void
gst_y4m_encode_init (GstY4mEncode * filter)
{
GST_PAD_SET_ACCEPT_TEMPLATE (GST_VIDEO_ENCODER_SINK_PAD (filter));
/* init properties */
gst_y4m_encode_reset (filter);
}
static void
gst_y4m_encode_reset (GstY4mEncode * filter)
{
filter->header = FALSE;
}
static gboolean
gst_y4m_encode_set_format (GstVideoEncoder * encoder,
GstVideoCodecState * state)
{
GstY4mEncode *y4menc;
GstVideoInfo *info;
GstVideoCodecState *output_state;
y4menc = GST_Y4M_ENCODE (encoder);
info = &state->info;
switch (GST_VIDEO_INFO_FORMAT (info)) {
case GST_VIDEO_FORMAT_I420:
y4menc->colorspace = "420";
break;
case GST_VIDEO_FORMAT_Y42B:
y4menc->colorspace = "422";
break;
case GST_VIDEO_FORMAT_Y41B:
y4menc->colorspace = "411";
break;
case GST_VIDEO_FORMAT_Y444:
y4menc->colorspace = "444";
break;
default:
goto invalid_format;
}
y4menc->info = *info;
output_state =
gst_video_encoder_set_output_state (encoder,
gst_static_pad_template_get_caps (&y4mencode_src_factory), state);
gst_video_codec_state_unref (output_state);
return TRUE;
invalid_format:
{
GST_ERROR_OBJECT (y4menc, "Invalid format");
return FALSE;
}
}
static inline GstBuffer *
gst_y4m_encode_get_stream_header (GstY4mEncode * filter, gboolean tff)
{
gpointer header;
GstBuffer *buf;
gchar interlaced;
gsize len;
if (GST_VIDEO_INFO_IS_INTERLACED (&filter->info)) {
if (tff)
interlaced = 't';
else
interlaced = 'b';
} else {
interlaced = 'p';
}
header = g_strdup_printf ("YUV4MPEG2 C%s W%d H%d I%c F%d:%d A%d:%d\n",
filter->colorspace, GST_VIDEO_INFO_WIDTH (&filter->info),
GST_VIDEO_INFO_HEIGHT (&filter->info), interlaced,
GST_VIDEO_INFO_FPS_N (&filter->info),
GST_VIDEO_INFO_FPS_D (&filter->info),
GST_VIDEO_INFO_PAR_N (&filter->info),
GST_VIDEO_INFO_PAR_D (&filter->info));
len = strlen (header);
buf = gst_buffer_new ();
gst_buffer_append_memory (buf,
gst_memory_new_wrapped (0, header, len, 0, len, header, g_free));
return buf;
}
static inline GstBuffer *
gst_y4m_encode_get_frame_header (GstY4mEncode * filter)
{
gpointer header;
GstBuffer *buf;
gsize len;
header = g_strdup_printf ("FRAME\n");
len = strlen (header);
buf = gst_buffer_new ();
gst_buffer_append_memory (buf,
gst_memory_new_wrapped (0, header, len, 0, len, header, g_free));
return buf;
}
static GstFlowReturn
gst_y4m_encode_handle_frame (GstVideoEncoder * encoder,
GstVideoCodecFrame * frame)
{
GstY4mEncode *filter = GST_Y4M_ENCODE (encoder);
GstClockTime timestamp;
/* check we got some decent info from caps */
if (GST_VIDEO_INFO_FORMAT (&filter->info) == GST_VIDEO_FORMAT_UNKNOWN)
goto not_negotiated;
timestamp = GST_BUFFER_TIMESTAMP (frame->input_buffer);
if (G_UNLIKELY (!filter->header)) {
gboolean tff = FALSE;
if (GST_VIDEO_INFO_IS_INTERLACED (&filter->info)) {
tff =
GST_BUFFER_FLAG_IS_SET (frame->input_buffer,
GST_VIDEO_BUFFER_FLAG_TFF);
}
frame->output_buffer = gst_y4m_encode_get_stream_header (filter, tff);
filter->header = TRUE;
frame->output_buffer =
gst_buffer_append (frame->output_buffer,
gst_y4m_encode_get_frame_header (filter));
} else {
frame->output_buffer = gst_y4m_encode_get_frame_header (filter);
}
frame->output_buffer =
gst_buffer_append (frame->output_buffer,
gst_buffer_copy (frame->input_buffer));
/* decorate */
frame->output_buffer = gst_buffer_make_writable (frame->output_buffer);
GST_BUFFER_TIMESTAMP (frame->output_buffer) = timestamp;
return gst_video_encoder_finish_frame (encoder, frame);
not_negotiated:
{
GST_ELEMENT_ERROR (filter, CORE, NEGOTIATION, (NULL),
("format wasn't negotiated"));
return GST_FLOW_NOT_NEGOTIATED;
}
}
static GstStateChangeReturn
gst_y4m_encode_change_state (GstElement * element, GstStateChange transition)
{
GstY4mEncode *filter = GST_Y4M_ENCODE (element);
GstStateChangeReturn ret;
switch (transition) {
case GST_STATE_CHANGE_NULL_TO_READY:
case GST_STATE_CHANGE_READY_TO_PAUSED:
break;
default:
break;
}
ret = GST_CALL_PARENT_WITH_DEFAULT (GST_ELEMENT_CLASS, change_state,
(element, transition), GST_STATE_CHANGE_SUCCESS);
if (ret != GST_STATE_CHANGE_SUCCESS)
return ret;
switch (transition) {
case GST_STATE_CHANGE_PAUSED_TO_READY:
gst_y4m_encode_reset (filter);
break;
default:
break;
}
return GST_STATE_CHANGE_SUCCESS;
}
static gboolean
plugin_init (GstPlugin * plugin)
{
return gst_element_register (plugin, "y4menc", GST_RANK_PRIMARY,
GST_TYPE_Y4M_ENCODE);
}
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
GST_VERSION_MINOR,
y4menc,
"Encodes a YUV frame into the yuv4mpeg format (mjpegtools)",
plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)