mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-11 02:46:33 +00:00
737 lines
21 KiB
C
737 lines
21 KiB
C
/* GStreamer
|
|
* Copyright (C) 2004 Wim Taymans <wim@fluendo.com>
|
|
* Copyright (c) 2012 Collabora Ltd.
|
|
* Author : Edward Hervey <edward@collabora.com>
|
|
* Author : Mark Nauwelaerts <mark.nauwelaerts@collabora.co.uk>
|
|
* Copyright (c) 2013 Sebastian Dröge <slomo@circular-chaos.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., 51 Franklin St, Fifth Floor,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
/**
|
|
* SECTION:element-daalaenc
|
|
* @see_also: daaladec, oggmux
|
|
*
|
|
* This element encodes raw video into a Daala stream.
|
|
* <ulink url="http://www.xiph.org/daala/">Daala</ulink> is a royalty-free
|
|
* video codec maintained by the <ulink url="http://www.xiph.org/">Xiph.org
|
|
* Foundation</ulink>.
|
|
*
|
|
* <refsect2>
|
|
* <title>Example pipeline</title>
|
|
* |[
|
|
* gst-launch -v videotestsrc num-buffers=1000 ! daalaenc ! oggmux ! filesink location=videotestsrc.ogg
|
|
* ]| This example pipeline will encode a test video source to daala muxed in an
|
|
* ogg container. Refer to the daaladec documentation to decode the create
|
|
* stream.
|
|
* </refsect2>
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <string.h>
|
|
#include <stdlib.h> /* free */
|
|
|
|
#include <gst/tag/tag.h>
|
|
#include <gst/video/video.h>
|
|
#include <gst/video/gstvideometa.h>
|
|
|
|
#include "gstdaalaenc.h"
|
|
|
|
#define GST_CAT_DEFAULT daalaenc_debug
|
|
GST_DEBUG_CATEGORY_STATIC (GST_CAT_DEFAULT);
|
|
|
|
enum
|
|
{
|
|
PROP_0,
|
|
PROP_QUANT,
|
|
PROP_KEYFRAME_RATE
|
|
};
|
|
|
|
#define DEFAULT_QUANT 10
|
|
#define DEFAULT_KEYFRAME_RATE 1
|
|
|
|
static GstStaticPadTemplate daala_enc_sink_factory =
|
|
GST_STATIC_PAD_TEMPLATE ("sink",
|
|
GST_PAD_SINK,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ I420, Y444 }"))
|
|
);
|
|
|
|
static GstStaticPadTemplate daala_enc_src_factory =
|
|
GST_STATIC_PAD_TEMPLATE ("src",
|
|
GST_PAD_SRC,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS ("video/x-daala, "
|
|
"framerate = (fraction) [1/MAX, MAX], "
|
|
"width = (int) [ 1, MAX ], " "height = (int) [ 1, MAX ]")
|
|
);
|
|
|
|
#define gst_daala_enc_parent_class parent_class
|
|
G_DEFINE_TYPE (GstDaalaEnc, gst_daala_enc, GST_TYPE_VIDEO_ENCODER);
|
|
|
|
static gboolean daala_enc_start (GstVideoEncoder * enc);
|
|
static gboolean daala_enc_stop (GstVideoEncoder * enc);
|
|
static gboolean daala_enc_flush (GstVideoEncoder * enc);
|
|
static gboolean daala_enc_set_format (GstVideoEncoder * enc,
|
|
GstVideoCodecState * state);
|
|
static GstFlowReturn daala_enc_handle_frame (GstVideoEncoder * enc,
|
|
GstVideoCodecFrame * frame);
|
|
static GstFlowReturn daala_enc_pre_push (GstVideoEncoder * benc,
|
|
GstVideoCodecFrame * frame);
|
|
static GstFlowReturn daala_enc_finish (GstVideoEncoder * enc);
|
|
static gboolean daala_enc_propose_allocation (GstVideoEncoder * encoder,
|
|
GstQuery * query);
|
|
|
|
static GstCaps *daala_enc_getcaps (GstVideoEncoder * encoder, GstCaps * filter);
|
|
static void daala_enc_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec);
|
|
static void daala_enc_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec);
|
|
static void daala_enc_finalize (GObject * object);
|
|
|
|
static void
|
|
gst_daala_enc_class_init (GstDaalaEncClass * klass)
|
|
{
|
|
GObjectClass *gobject_class = (GObjectClass *) klass;
|
|
GstElementClass *element_class = (GstElementClass *) klass;
|
|
GstVideoEncoderClass *gstvideo_encoder_class =
|
|
GST_VIDEO_ENCODER_CLASS (klass);
|
|
|
|
gobject_class->set_property = daala_enc_set_property;
|
|
gobject_class->get_property = daala_enc_get_property;
|
|
gobject_class->finalize = daala_enc_finalize;
|
|
|
|
g_object_class_install_property (gobject_class, PROP_QUANT,
|
|
g_param_spec_int ("quant", "Quant", "Quant",
|
|
0, 511, DEFAULT_QUANT,
|
|
(GParamFlags) G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
g_object_class_install_property (gobject_class, PROP_KEYFRAME_RATE,
|
|
g_param_spec_int ("keyframe-rate", "Keyframe Rate", "Keyframe Rate",
|
|
1, G_MAXINT, DEFAULT_KEYFRAME_RATE,
|
|
(GParamFlags) G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
gst_element_class_add_pad_template (element_class,
|
|
gst_static_pad_template_get (&daala_enc_src_factory));
|
|
gst_element_class_add_pad_template (element_class,
|
|
gst_static_pad_template_get (&daala_enc_sink_factory));
|
|
gst_element_class_set_static_metadata (element_class,
|
|
"Daala video encoder", "Codec/Encoder/Video",
|
|
"Encode raw YUV video to a Daala stream",
|
|
"Sebastian Dröge <slomo@circular-chaos.org>");
|
|
|
|
gstvideo_encoder_class->start = GST_DEBUG_FUNCPTR (daala_enc_start);
|
|
gstvideo_encoder_class->stop = GST_DEBUG_FUNCPTR (daala_enc_stop);
|
|
gstvideo_encoder_class->flush = GST_DEBUG_FUNCPTR (daala_enc_flush);
|
|
gstvideo_encoder_class->set_format = GST_DEBUG_FUNCPTR (daala_enc_set_format);
|
|
gstvideo_encoder_class->handle_frame =
|
|
GST_DEBUG_FUNCPTR (daala_enc_handle_frame);
|
|
gstvideo_encoder_class->pre_push = GST_DEBUG_FUNCPTR (daala_enc_pre_push);
|
|
gstvideo_encoder_class->finish = GST_DEBUG_FUNCPTR (daala_enc_finish);
|
|
gstvideo_encoder_class->getcaps = GST_DEBUG_FUNCPTR (daala_enc_getcaps);
|
|
gstvideo_encoder_class->propose_allocation =
|
|
GST_DEBUG_FUNCPTR (daala_enc_propose_allocation);
|
|
|
|
GST_DEBUG_CATEGORY_INIT (daalaenc_debug, "daalaenc", 0, "Daala encoder");
|
|
}
|
|
|
|
static void
|
|
gst_daala_enc_init (GstDaalaEnc * enc)
|
|
{
|
|
enc->quant = DEFAULT_QUANT;
|
|
enc->keyframe_rate = DEFAULT_KEYFRAME_RATE;
|
|
}
|
|
|
|
static void
|
|
daala_enc_finalize (GObject * object)
|
|
{
|
|
GstDaalaEnc *enc = GST_DAALA_ENC (object);
|
|
|
|
GST_DEBUG_OBJECT (enc, "Finalizing");
|
|
if (enc->encoder)
|
|
daala_encode_free (enc->encoder);
|
|
daala_comment_clear (&enc->comment);
|
|
daala_info_clear (&enc->info);
|
|
|
|
if (enc->input_state)
|
|
gst_video_codec_state_unref (enc->input_state);
|
|
|
|
G_OBJECT_CLASS (parent_class)->finalize (object);
|
|
}
|
|
|
|
static gboolean
|
|
daala_enc_flush (GstVideoEncoder * benc)
|
|
{
|
|
GstDaalaEnc *enc = GST_DAALA_ENC (benc);
|
|
int quant;
|
|
|
|
GST_OBJECT_LOCK (enc);
|
|
quant = enc->quant;
|
|
enc->quant_changed = FALSE;
|
|
enc->info.keyframe_rate = enc->keyframe_rate;
|
|
enc->keyframe_rate_changed = FALSE;
|
|
GST_OBJECT_UNLOCK (enc);
|
|
|
|
if (enc->encoder)
|
|
daala_encode_free (enc->encoder);
|
|
enc->encoder = daala_encode_create (&enc->info);
|
|
|
|
daala_encode_ctl (enc->encoder, OD_SET_QUANT, &quant, sizeof (int));
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean
|
|
daala_enc_start (GstVideoEncoder * benc)
|
|
{
|
|
GstDaalaEnc *enc;
|
|
|
|
GST_DEBUG_OBJECT (benc, "start: init daala");
|
|
enc = GST_DAALA_ENC (benc);
|
|
|
|
enc->packetno = 0;
|
|
enc->initialised = FALSE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean
|
|
daala_enc_stop (GstVideoEncoder * benc)
|
|
{
|
|
GstDaalaEnc *enc;
|
|
|
|
GST_DEBUG_OBJECT (benc, "stop: clearing daala state");
|
|
enc = GST_DAALA_ENC (benc);
|
|
|
|
if (enc->encoder) {
|
|
daala_encode_free (enc->encoder);
|
|
enc->encoder = NULL;
|
|
}
|
|
daala_comment_clear (&enc->comment);
|
|
daala_info_clear (&enc->info);
|
|
|
|
if (enc->input_state)
|
|
gst_video_codec_state_unref (enc->input_state);
|
|
enc->input_state = NULL;
|
|
|
|
enc->initialised = FALSE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static char *
|
|
daala_enc_get_supported_formats (void)
|
|
{
|
|
daala_enc_ctx *encoder;
|
|
daala_info info;
|
|
struct
|
|
{
|
|
GstVideoFormat fmt;
|
|
gint planes;
|
|
gint xdec[3], ydec[3];
|
|
} formats[] = {
|
|
{
|
|
GST_VIDEO_FORMAT_Y444, 3, {
|
|
0, 0, 0}, {
|
|
0, 0, 0}}, {
|
|
GST_VIDEO_FORMAT_I420, 3, {
|
|
0, 1, 1}, {
|
|
0, 1, 1}}
|
|
};
|
|
GString *string = NULL;
|
|
guint i;
|
|
|
|
daala_info_init (&info);
|
|
info.pic_width = 16;
|
|
info.pic_height = 16;
|
|
info.timebase_numerator = 25;
|
|
info.timebase_denominator = 1;
|
|
info.frame_duration = 1;
|
|
for (i = 0; i < G_N_ELEMENTS (formats); i++) {
|
|
gint j;
|
|
|
|
info.nplanes = formats[i].planes;
|
|
for (j = 0; j < formats[i].planes; j++) {
|
|
info.plane_info[j].xdec = formats[i].xdec[j];
|
|
info.plane_info[j].ydec = formats[i].ydec[j];
|
|
}
|
|
|
|
encoder = daala_encode_create (&info);
|
|
if (encoder == NULL)
|
|
continue;
|
|
|
|
GST_LOG ("format %s is supported",
|
|
gst_video_format_to_string (formats[i].fmt));
|
|
daala_encode_free (encoder);
|
|
|
|
if (string == NULL) {
|
|
string = g_string_new (gst_video_format_to_string (formats[i].fmt));
|
|
} else {
|
|
g_string_append (string, ", ");
|
|
g_string_append (string, gst_video_format_to_string (formats[i].fmt));
|
|
}
|
|
}
|
|
daala_info_clear (&info);
|
|
|
|
return string == NULL ? NULL : g_string_free (string, FALSE);
|
|
}
|
|
|
|
static GstCaps *
|
|
daala_enc_getcaps (GstVideoEncoder * encoder, GstCaps * filter)
|
|
{
|
|
GstCaps *caps, *ret;
|
|
char *supported_formats, *caps_string;
|
|
|
|
supported_formats = daala_enc_get_supported_formats ();
|
|
if (!supported_formats) {
|
|
GST_WARNING ("no supported formats found. Encoder disabled?");
|
|
return gst_caps_new_empty ();
|
|
}
|
|
|
|
caps_string = g_strdup_printf ("video/x-raw, "
|
|
"format = (string) { %s }, "
|
|
"framerate = (fraction) [1/MAX, MAX], "
|
|
"width = (int) [ 1, MAX ], " "height = (int) [ 1, MAX ]",
|
|
supported_formats);
|
|
caps = gst_caps_from_string (caps_string);
|
|
g_free (caps_string);
|
|
g_free (supported_formats);
|
|
GST_DEBUG ("Supported caps: %" GST_PTR_FORMAT, caps);
|
|
|
|
ret = gst_video_encoder_proxy_getcaps (encoder, caps, filter);
|
|
gst_caps_unref (caps);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static gboolean
|
|
daala_enc_set_format (GstVideoEncoder * benc, GstVideoCodecState * state)
|
|
{
|
|
GstDaalaEnc *enc = GST_DAALA_ENC (benc);
|
|
GstVideoInfo *info = &state->info;
|
|
|
|
daala_info_clear (&enc->info);
|
|
daala_info_init (&enc->info);
|
|
enc->info.pic_width = GST_VIDEO_INFO_WIDTH (info);
|
|
enc->info.pic_height = GST_VIDEO_INFO_HEIGHT (info);
|
|
switch (GST_VIDEO_INFO_FORMAT (info)) {
|
|
case GST_VIDEO_FORMAT_I420:
|
|
enc->info.nplanes = 3;
|
|
enc->info.plane_info[0].xdec = 0;
|
|
enc->info.plane_info[0].ydec = 0;
|
|
enc->info.plane_info[1].xdec = 1;
|
|
enc->info.plane_info[1].ydec = 1;
|
|
enc->info.plane_info[2].xdec = 1;
|
|
enc->info.plane_info[2].ydec = 1;
|
|
break;
|
|
case GST_VIDEO_FORMAT_Y444:
|
|
enc->info.nplanes = 3;
|
|
enc->info.plane_info[0].xdec = 0;
|
|
enc->info.plane_info[0].ydec = 0;
|
|
enc->info.plane_info[1].xdec = 0;
|
|
enc->info.plane_info[1].ydec = 0;
|
|
enc->info.plane_info[2].xdec = 0;
|
|
enc->info.plane_info[2].ydec = 0;
|
|
break;
|
|
default:
|
|
g_assert_not_reached ();
|
|
}
|
|
|
|
enc->info.timebase_numerator = GST_VIDEO_INFO_FPS_N (info);
|
|
enc->info.timebase_denominator = GST_VIDEO_INFO_FPS_D (info);
|
|
enc->info.frame_duration = 1;
|
|
enc->info.pixel_aspect_numerator = GST_VIDEO_INFO_PAR_N (info);
|
|
enc->info.pixel_aspect_denominator = GST_VIDEO_INFO_PAR_D (info);
|
|
|
|
/* Save input state */
|
|
if (enc->input_state)
|
|
gst_video_codec_state_unref (enc->input_state);
|
|
enc->input_state = gst_video_codec_state_ref (state);
|
|
|
|
daala_enc_flush (benc);
|
|
enc->initialised = TRUE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/* this function does a straight granulepos -> timestamp conversion */
|
|
static GstClockTime
|
|
granulepos_to_timestamp (GstDaalaEnc * enc, ogg_int64_t granulepos)
|
|
{
|
|
guint64 iframe, pframe;
|
|
int shift = enc->info.keyframe_granule_shift;
|
|
|
|
if (granulepos < 0)
|
|
return GST_CLOCK_TIME_NONE;
|
|
|
|
iframe = granulepos >> shift;
|
|
pframe = granulepos - (iframe << shift);
|
|
|
|
/* num and den are 32 bit, so we can safely multiply with GST_SECOND */
|
|
return gst_util_uint64_scale ((guint64) (iframe + pframe),
|
|
GST_SECOND * enc->info.timebase_denominator,
|
|
enc->info.timebase_numerator);
|
|
}
|
|
|
|
static GstFlowReturn
|
|
daala_enc_pre_push (GstVideoEncoder * benc, GstVideoCodecFrame * frame)
|
|
{
|
|
GstDaalaEnc *enc = GST_DAALA_ENC (benc);
|
|
guint64 pfn;
|
|
|
|
/* see ext/ogg/README; OFFSET_END takes "our" granulepos, OFFSET its
|
|
* time representation */
|
|
/* granulepos from sync frame */
|
|
pfn = frame->presentation_frame_number - frame->distance_from_sync;
|
|
/* correct to correspond to linear running time */
|
|
pfn -= enc->pfn_offset;
|
|
pfn += enc->granulepos_offset + 1;
|
|
/* granulepos */
|
|
GST_BUFFER_OFFSET_END (frame->output_buffer) =
|
|
(pfn << enc->info.keyframe_granule_shift) + frame->distance_from_sync;
|
|
GST_BUFFER_OFFSET (frame->output_buffer) = granulepos_to_timestamp (enc,
|
|
GST_BUFFER_OFFSET_END (frame->output_buffer));
|
|
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
static GstFlowReturn
|
|
daala_push_packet (GstDaalaEnc * enc, ogg_packet * packet)
|
|
{
|
|
GstVideoEncoder *benc;
|
|
GstFlowReturn ret;
|
|
GstVideoCodecFrame *frame;
|
|
|
|
benc = GST_VIDEO_ENCODER (enc);
|
|
|
|
frame = gst_video_encoder_get_oldest_frame (benc);
|
|
if (gst_video_encoder_allocate_output_frame (benc, frame,
|
|
packet->bytes) != GST_FLOW_OK) {
|
|
GST_WARNING_OBJECT (enc, "Could not allocate buffer");
|
|
gst_video_codec_frame_unref (frame);
|
|
ret = GST_FLOW_ERROR;
|
|
goto done;
|
|
}
|
|
|
|
if (packet->bytes > 0)
|
|
gst_buffer_fill (frame->output_buffer, 0, packet->packet, packet->bytes);
|
|
|
|
/* the second most significant bit of the first data byte is cleared
|
|
* for keyframes */
|
|
if (packet->bytes > 0 && (packet->packet[0] & 0x40) == 0) {
|
|
GST_VIDEO_CODEC_FRAME_SET_SYNC_POINT (frame);
|
|
} else {
|
|
GST_VIDEO_CODEC_FRAME_UNSET_SYNC_POINT (frame);
|
|
}
|
|
enc->packetno++;
|
|
|
|
ret = gst_video_encoder_finish_frame (benc, frame);
|
|
|
|
done:
|
|
return ret;
|
|
}
|
|
|
|
static GstCaps *
|
|
daala_set_header_on_caps (GstCaps * caps, GList * buffers)
|
|
{
|
|
GstStructure *structure;
|
|
GValue array = { 0 };
|
|
GValue value = { 0 };
|
|
GstBuffer *buffer;
|
|
GList *walk;
|
|
|
|
caps = gst_caps_make_writable (caps);
|
|
structure = gst_caps_get_structure (caps, 0);
|
|
|
|
/* put copies of the buffers in a fixed list */
|
|
g_value_init (&array, GST_TYPE_ARRAY);
|
|
|
|
for (walk = buffers; walk; walk = walk->next) {
|
|
buffer = walk->data;
|
|
g_value_init (&value, GST_TYPE_BUFFER);
|
|
gst_value_set_buffer (&value, buffer);
|
|
gst_value_array_append_value (&array, &value);
|
|
g_value_unset (&value);
|
|
}
|
|
|
|
gst_structure_take_value (structure, "streamheader", &array);
|
|
|
|
return caps;
|
|
}
|
|
|
|
static void
|
|
daala_enc_init_buffer (GstDaalaEnc * enc, od_img * buf, GstVideoFrame * frame)
|
|
{
|
|
guint i;
|
|
|
|
buf->nplanes = 3;
|
|
buf->width = GST_VIDEO_FRAME_WIDTH (frame);
|
|
buf->height = GST_VIDEO_FRAME_HEIGHT (frame);
|
|
|
|
for (i = 0; i < 3; i++) {
|
|
buf->planes[i].data = GST_VIDEO_FRAME_COMP_DATA (frame, i);
|
|
buf->planes[i].xdec = enc->info.plane_info[i].xdec;
|
|
buf->planes[i].ydec = enc->info.plane_info[i].ydec;
|
|
buf->planes[i].xstride = 1;
|
|
buf->planes[i].ystride = GST_VIDEO_FRAME_COMP_STRIDE (frame, i);
|
|
}
|
|
}
|
|
|
|
static void
|
|
daala_enc_reset_ts (GstDaalaEnc * enc, GstClockTime running_time, gint pfn)
|
|
{
|
|
enc->granulepos_offset =
|
|
gst_util_uint64_scale (running_time, enc->input_state->info.fps_n,
|
|
GST_SECOND * enc->input_state->info.fps_d);
|
|
enc->timestamp_offset = running_time;
|
|
enc->pfn_offset = pfn;
|
|
}
|
|
|
|
static GstBuffer *
|
|
daala_enc_buffer_from_header_packet (GstDaalaEnc * enc, ogg_packet * packet)
|
|
{
|
|
GstBuffer *outbuf;
|
|
|
|
outbuf =
|
|
gst_video_encoder_allocate_output_buffer (GST_VIDEO_ENCODER (enc),
|
|
packet->bytes);
|
|
gst_buffer_fill (outbuf, 0, packet->packet, packet->bytes);
|
|
GST_BUFFER_OFFSET (outbuf) = 0;
|
|
GST_BUFFER_OFFSET_END (outbuf) = 0;
|
|
GST_BUFFER_TIMESTAMP (outbuf) = GST_CLOCK_TIME_NONE;
|
|
GST_BUFFER_DURATION (outbuf) = GST_CLOCK_TIME_NONE;
|
|
GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_FLAG_HEADER);
|
|
|
|
GST_DEBUG ("created header packet buffer, %u bytes",
|
|
(guint) gst_buffer_get_size (outbuf));
|
|
return outbuf;
|
|
}
|
|
|
|
static GstFlowReturn
|
|
daala_enc_handle_frame (GstVideoEncoder * benc, GstVideoCodecFrame * frame)
|
|
{
|
|
GstDaalaEnc *enc;
|
|
ogg_packet op;
|
|
GstClockTime timestamp, running_time;
|
|
GstFlowReturn ret;
|
|
gboolean force_keyframe;
|
|
|
|
enc = GST_DAALA_ENC (benc);
|
|
|
|
/* we keep track of two timelines.
|
|
* - The timestamps from the incoming buffers, which we copy to the outgoing
|
|
* encoded buffers as-is. We need to do this as we simply forward the
|
|
* newsegment events.
|
|
* - The running_time of the buffers, which we use to construct the granulepos
|
|
* in the packets.
|
|
*/
|
|
timestamp = frame->pts;
|
|
|
|
/* incoming buffers are clipped, so this should be positive */
|
|
running_time =
|
|
gst_segment_to_running_time (&GST_VIDEO_ENCODER_INPUT_SEGMENT (enc),
|
|
GST_FORMAT_TIME, timestamp);
|
|
g_return_val_if_fail (running_time >= 0 || timestamp < 0, GST_FLOW_ERROR);
|
|
|
|
GST_OBJECT_LOCK (enc);
|
|
if (enc->quant_changed) {
|
|
int quant = enc->quant;
|
|
|
|
daala_encode_ctl (enc->encoder, OD_SET_QUANT, &quant, sizeof (int));
|
|
enc->quant_changed = FALSE;
|
|
}
|
|
|
|
/* see if we need to schedule a keyframe */
|
|
force_keyframe = GST_VIDEO_CODEC_FRAME_IS_FORCE_KEYFRAME (frame);
|
|
GST_OBJECT_UNLOCK (enc);
|
|
|
|
if (enc->packetno == 0) {
|
|
/* no packets written yet, setup headers */
|
|
GstCaps *caps;
|
|
GstBuffer *buf;
|
|
GList *buffers = NULL;
|
|
int result;
|
|
GstVideoCodecState *state;
|
|
|
|
enc->granulepos_offset = 0;
|
|
enc->timestamp_offset = 0;
|
|
|
|
GST_DEBUG_OBJECT (enc, "output headers");
|
|
/* Daala streams begin with three headers; the initial header (with
|
|
most of the codec setup parameters) which is mandated by the Ogg
|
|
bitstream spec. The second header holds any comment fields. The
|
|
third header holds the bitstream codebook. We merely need to
|
|
make the headers, then pass them to libdaala one at a time;
|
|
libdaala handles the additional Ogg bitstream constraints */
|
|
|
|
/* create the remaining daala headers */
|
|
daala_comment_clear (&enc->comment);
|
|
daala_comment_init (&enc->comment);
|
|
|
|
while ((result =
|
|
daala_encode_flush_header (enc->encoder, &enc->comment, &op)) > 0) {
|
|
buf = daala_enc_buffer_from_header_packet (enc, &op);
|
|
buffers = g_list_prepend (buffers, buf);
|
|
}
|
|
if (result < 0) {
|
|
g_list_foreach (buffers, (GFunc) gst_buffer_unref, NULL);
|
|
g_list_free (buffers);
|
|
goto encoder_disabled;
|
|
}
|
|
|
|
buffers = g_list_reverse (buffers);
|
|
|
|
/* mark buffers and put on caps */
|
|
caps = gst_caps_new_empty_simple ("video/x-daala");
|
|
caps = daala_set_header_on_caps (caps, buffers);
|
|
state = gst_video_encoder_set_output_state (benc, caps, enc->input_state);
|
|
|
|
GST_DEBUG ("here are the caps: %" GST_PTR_FORMAT, state->caps);
|
|
|
|
gst_video_codec_state_unref (state);
|
|
|
|
gst_video_encoder_negotiate (GST_VIDEO_ENCODER (enc));
|
|
|
|
gst_video_encoder_set_headers (benc, buffers);
|
|
|
|
daala_enc_reset_ts (enc, running_time, frame->presentation_frame_number);
|
|
}
|
|
|
|
{
|
|
od_img img;
|
|
gint res;
|
|
GstVideoFrame vframe;
|
|
|
|
if (force_keyframe) {
|
|
/* TODO */
|
|
}
|
|
|
|
gst_video_frame_map (&vframe, &enc->input_state->info, frame->input_buffer,
|
|
GST_MAP_READ);
|
|
daala_enc_init_buffer (enc, &img, &vframe);
|
|
|
|
res = daala_encode_img_in (enc->encoder, &img, 1);
|
|
gst_video_frame_unmap (&vframe);
|
|
|
|
/* none of the failure cases can happen here */
|
|
g_assert (res == 0);
|
|
|
|
ret = GST_FLOW_OK;
|
|
while (daala_encode_packet_out (enc->encoder, 0, &op)) {
|
|
ret = daala_push_packet (enc, &op);
|
|
if (ret != GST_FLOW_OK)
|
|
goto beach;
|
|
}
|
|
}
|
|
|
|
beach:
|
|
gst_video_codec_frame_unref (frame);
|
|
return ret;
|
|
|
|
/* ERRORS */
|
|
encoder_disabled:
|
|
{
|
|
gst_video_codec_frame_unref (frame);
|
|
GST_ELEMENT_ERROR (enc, STREAM, ENCODE, (NULL),
|
|
("libdaala has been compiled with the encoder disabled"));
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
}
|
|
|
|
static gboolean
|
|
daala_enc_finish (GstVideoEncoder * benc)
|
|
{
|
|
GstDaalaEnc *enc;
|
|
ogg_packet op;
|
|
|
|
enc = GST_DAALA_ENC (benc);
|
|
|
|
if (enc->initialised) {
|
|
/* push last packet with eos flag, should not be called */
|
|
while (daala_encode_packet_out (enc->encoder, 1, &op)) {
|
|
daala_push_packet (enc, &op);
|
|
}
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean
|
|
daala_enc_propose_allocation (GstVideoEncoder * encoder, GstQuery * query)
|
|
{
|
|
gst_query_add_allocation_meta (query, GST_VIDEO_META_API_TYPE, NULL);
|
|
|
|
return GST_VIDEO_ENCODER_CLASS (parent_class)->propose_allocation (encoder,
|
|
query);
|
|
}
|
|
|
|
static void
|
|
daala_enc_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstDaalaEnc *enc = GST_DAALA_ENC (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_QUANT:
|
|
GST_OBJECT_LOCK (enc);
|
|
enc->quant = g_value_get_int (value);
|
|
enc->quant_changed = TRUE;
|
|
GST_OBJECT_UNLOCK (enc);
|
|
break;
|
|
case PROP_KEYFRAME_RATE:
|
|
GST_OBJECT_LOCK (enc);
|
|
enc->keyframe_rate = g_value_get_int (value);
|
|
enc->keyframe_rate_changed = TRUE;
|
|
GST_OBJECT_UNLOCK (enc);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
daala_enc_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstDaalaEnc *enc = GST_DAALA_ENC (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_QUANT:
|
|
GST_OBJECT_LOCK (enc);
|
|
g_value_set_int (value, enc->quant);
|
|
GST_OBJECT_UNLOCK (enc);
|
|
break;
|
|
case PROP_KEYFRAME_RATE:
|
|
GST_OBJECT_LOCK (enc);
|
|
g_value_set_int (value, enc->keyframe_rate);
|
|
GST_OBJECT_UNLOCK (enc);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
gboolean
|
|
gst_daala_enc_register (GstPlugin * plugin)
|
|
{
|
|
return gst_element_register (plugin, "daalaenc",
|
|
GST_RANK_PRIMARY, GST_TYPE_DAALA_ENC);
|
|
}
|