mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-13 03:46:34 +00:00
721 lines
21 KiB
C
721 lines
21 KiB
C
|
/* GStreamer
|
||
|
* Copyright (C) <2024> V-Nova International Limited
|
||
|
*
|
||
|
* 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.
|
||
|
*/
|
||
|
|
||
|
#ifdef HAVE_CONFIG_H
|
||
|
#include "config.h"
|
||
|
#endif
|
||
|
|
||
|
#include <stdio.h>
|
||
|
#include <stdlib.h>
|
||
|
#include <string.h>
|
||
|
|
||
|
#include <gst/codecparsers/gstlcevcmeta.h>
|
||
|
|
||
|
#include "gstlcevcdecutils.h"
|
||
|
#include "gstlcevcdec.h"
|
||
|
|
||
|
enum
|
||
|
{
|
||
|
PROP_0,
|
||
|
PROP_VERBOSE,
|
||
|
PROP_MAX_WIDTH,
|
||
|
PROP_MAX_HEIGHT,
|
||
|
PROP_MAX_LATENCY
|
||
|
};
|
||
|
|
||
|
#define DEFAULT_MAX_WIDTH 3840
|
||
|
#define DEFAULT_MAX_HEIGHT 2160
|
||
|
#define DEFAULT_MAX_LATENCY 0
|
||
|
|
||
|
#define GST_CAT_DEFAULT gst_lcevc_dec_debug
|
||
|
GST_DEBUG_CATEGORY_STATIC (GST_CAT_DEFAULT);
|
||
|
|
||
|
static GstStaticPadTemplate sink_template_factory =
|
||
|
GST_STATIC_PAD_TEMPLATE ("sink",
|
||
|
GST_PAD_SINK,
|
||
|
GST_PAD_ALWAYS,
|
||
|
GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE
|
||
|
(GST_LCEVC_DEC_UTILS_SUPPORTED_FORMATS))
|
||
|
);
|
||
|
|
||
|
static GstStaticPadTemplate src_template_factory =
|
||
|
GST_STATIC_PAD_TEMPLATE ("src",
|
||
|
GST_PAD_SRC,
|
||
|
GST_PAD_ALWAYS,
|
||
|
GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE
|
||
|
(GST_LCEVC_DEC_UTILS_SUPPORTED_FORMATS))
|
||
|
);
|
||
|
|
||
|
#define DEBUG_INIT \
|
||
|
GST_DEBUG_CATEGORY_INIT (gst_lcevc_dec_debug, \
|
||
|
"lcevcdec", 0, "LCEVC Decoder element");
|
||
|
|
||
|
G_DEFINE_TYPE_WITH_CODE (GstLcevcDec, gst_lcevc_dec, GST_TYPE_VIDEO_DECODER,
|
||
|
DEBUG_INIT);
|
||
|
GST_ELEMENT_REGISTER_DEFINE (lcevcdec, "lcevcdec",
|
||
|
GST_RANK_MARGINAL, GST_TYPE_LCEVC_DEC);
|
||
|
|
||
|
#define GST_LCEVC_DEC_PICTURE_DATA gst_lcevc_dec_picture_data ()
|
||
|
|
||
|
static GQuark
|
||
|
gst_lcevc_dec_picture_data (void)
|
||
|
{
|
||
|
static GQuark quark = 0;
|
||
|
|
||
|
if (quark == 0)
|
||
|
quark = g_quark_from_string ("GstLcevcDecPictureData");
|
||
|
|
||
|
return quark;
|
||
|
}
|
||
|
|
||
|
typedef struct
|
||
|
{
|
||
|
LCEVC_DecoderHandle decoder_handle;
|
||
|
LCEVC_PictureHandle picture_handle;
|
||
|
guint32 width;
|
||
|
guint height;
|
||
|
} PictureData;
|
||
|
|
||
|
static PictureData *
|
||
|
picture_data_new (LCEVC_DecoderHandle decoder_handle, GstVideoFrame * frame)
|
||
|
{
|
||
|
PictureData *ret = g_new0 (PictureData, 1);
|
||
|
|
||
|
ret->decoder_handle = decoder_handle;
|
||
|
ret->width = GST_VIDEO_FRAME_WIDTH (frame);
|
||
|
ret->height = GST_VIDEO_FRAME_HEIGHT (frame);
|
||
|
|
||
|
/* Alloc LCEVC picture handle */
|
||
|
if (!gst_lcevc_dec_utils_alloc_picture_handle (decoder_handle, frame,
|
||
|
&ret->picture_handle)) {
|
||
|
g_free (ret);
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
return ret;
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
picture_data_free (gpointer p)
|
||
|
{
|
||
|
PictureData *data = p;
|
||
|
LCEVC_FreePicture (data->decoder_handle, data->picture_handle);
|
||
|
g_free (data);
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
gst_lcevc_dec_init (GstLcevcDec * lcevc)
|
||
|
{
|
||
|
lcevc->max_width = DEFAULT_MAX_WIDTH;
|
||
|
lcevc->max_height = DEFAULT_MAX_HEIGHT;
|
||
|
lcevc->max_latency = DEFAULT_MAX_LATENCY;
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
gst_lcevc_dec_set_property (GObject * object, guint prop_id,
|
||
|
const GValue * value, GParamSpec * pspec)
|
||
|
{
|
||
|
GstLcevcDec *lcevc = GST_LCEVC_DEC (object);
|
||
|
|
||
|
switch (prop_id) {
|
||
|
case PROP_VERBOSE:
|
||
|
lcevc->verbose = g_value_get_boolean (value);
|
||
|
break;
|
||
|
case PROP_MAX_WIDTH:
|
||
|
lcevc->max_width = g_value_get_int (value);
|
||
|
break;
|
||
|
case PROP_MAX_HEIGHT:
|
||
|
lcevc->max_height = g_value_get_int (value);
|
||
|
break;
|
||
|
case PROP_MAX_LATENCY:
|
||
|
lcevc->max_latency = g_value_get_int (value);
|
||
|
break;
|
||
|
default:
|
||
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
gst_lcevc_dec_get_property (GObject * object, guint prop_id,
|
||
|
GValue * value, GParamSpec * pspec)
|
||
|
{
|
||
|
GstLcevcDec *lcevc = GST_LCEVC_DEC (object);
|
||
|
|
||
|
switch (prop_id) {
|
||
|
case PROP_VERBOSE:
|
||
|
g_value_set_boolean (value, lcevc->verbose);
|
||
|
break;
|
||
|
case PROP_MAX_WIDTH:
|
||
|
g_value_set_int (value, lcevc->max_width);
|
||
|
break;
|
||
|
case PROP_MAX_HEIGHT:
|
||
|
g_value_set_int (value, lcevc->max_height);
|
||
|
break;
|
||
|
case PROP_MAX_LATENCY:
|
||
|
g_value_set_int (value, lcevc->max_latency);
|
||
|
break;
|
||
|
default:
|
||
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
event_callback (LCEVC_DecoderHandle dec, LCEVC_Event event,
|
||
|
LCEVC_PictureHandle pic, const LCEVC_DecodeInformation * info,
|
||
|
const uint8_t * data, uint32_t size, void *user_data)
|
||
|
{
|
||
|
GstLcevcDec *lcevc = user_data;
|
||
|
|
||
|
switch (event) {
|
||
|
case LCEVC_Log:
|
||
|
GST_DEBUG_OBJECT (lcevc, "LCEVC Log");
|
||
|
break;
|
||
|
case LCEVC_Exit:
|
||
|
GST_DEBUG_OBJECT (lcevc, "LCEVC Exit");
|
||
|
break;
|
||
|
case LCEVC_CanSendBase:
|
||
|
GST_DEBUG_OBJECT (lcevc, "LCEVC CanSendBase");
|
||
|
break;
|
||
|
case LCEVC_CanSendEnhancement:
|
||
|
GST_DEBUG_OBJECT (lcevc, "LCEVC CanSendEnhancement");
|
||
|
break;
|
||
|
case LCEVC_CanSendPicture:
|
||
|
GST_DEBUG_OBJECT (lcevc, "LCEVC CanSendPicure");
|
||
|
break;
|
||
|
case LCEVC_CanReceive:
|
||
|
GST_DEBUG_OBJECT (lcevc, "LCEVC CanReceive");
|
||
|
break;
|
||
|
case LCEVC_BasePictureDone:
|
||
|
GST_DEBUG_OBJECT (lcevc, "LCEVC Base Picure Done");
|
||
|
break;
|
||
|
case LCEVC_OutputPictureDone:
|
||
|
GST_DEBUG_OBJECT (lcevc, "LCEVC Output Picure Done");
|
||
|
break;
|
||
|
default:
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
initialize_lcevc_decoder (GstLcevcDec * lcevc)
|
||
|
{
|
||
|
LCEVC_AccelContextHandle accel_context = { 0, };
|
||
|
int32_t events[] = { LCEVC_Log, LCEVC_Exit, LCEVC_CanSendBase,
|
||
|
LCEVC_CanSendEnhancement, LCEVC_CanSendPicture, LCEVC_CanReceive,
|
||
|
LCEVC_BasePictureDone, LCEVC_OutputPictureDone
|
||
|
};
|
||
|
|
||
|
if (LCEVC_CreateDecoder (&lcevc->decoder_handle, accel_context) !=
|
||
|
LCEVC_Success)
|
||
|
return FALSE;
|
||
|
|
||
|
if (lcevc->max_width > 0)
|
||
|
LCEVC_ConfigureDecoderInt (lcevc->decoder_handle, "max_width",
|
||
|
lcevc->max_width);
|
||
|
if (lcevc->max_height > 0)
|
||
|
LCEVC_ConfigureDecoderInt (lcevc->decoder_handle, "max_height",
|
||
|
lcevc->max_height);
|
||
|
if (lcevc->max_latency > 0)
|
||
|
LCEVC_ConfigureDecoderInt (lcevc->decoder_handle, "max_latency",
|
||
|
lcevc->max_latency);
|
||
|
|
||
|
if (lcevc->verbose) {
|
||
|
LCEVC_ConfigureDecoderBool (lcevc->decoder_handle, "log_stdout", TRUE);
|
||
|
LCEVC_ConfigureDecoderInt (lcevc->decoder_handle, "log_level", 2);
|
||
|
}
|
||
|
|
||
|
LCEVC_ConfigureDecoderIntArray (lcevc->decoder_handle, "events", 8, events);
|
||
|
|
||
|
LCEVC_SetDecoderEventCallback (lcevc->decoder_handle, event_callback, lcevc);
|
||
|
|
||
|
if (LCEVC_InitializeDecoder (lcevc->decoder_handle) != LCEVC_Success)
|
||
|
return FALSE;
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
gst_lcevc_dec_start (GstVideoDecoder * decoder)
|
||
|
{
|
||
|
GstLcevcDec *lcevc = GST_LCEVC_DEC (decoder);
|
||
|
|
||
|
/* Initialize LCEVC decoder */
|
||
|
if (!initialize_lcevc_decoder (lcevc)) {
|
||
|
GST_ELEMENT_ERROR (decoder, LIBRARY, INIT, (NULL),
|
||
|
("Couldn't initialize LCEVC decoder"));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
gst_lcevc_dec_stop (GstVideoDecoder * decoder)
|
||
|
{
|
||
|
GstLcevcDec *lcevc = GST_LCEVC_DEC (decoder);
|
||
|
|
||
|
/* Destry LCEVC decoder */
|
||
|
LCEVC_DestroyDecoder (lcevc->decoder_handle);
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
gst_lcevc_dec_decide_allocation (GstVideoDecoder * decoder, GstQuery * query)
|
||
|
{
|
||
|
GstLcevcDec *lcevc = GST_LCEVC_DEC (decoder);
|
||
|
|
||
|
if (!GST_VIDEO_DECODER_CLASS (gst_lcevc_dec_parent_class)->decide_allocation
|
||
|
(decoder, query))
|
||
|
return FALSE;
|
||
|
|
||
|
lcevc->can_crop = gst_query_find_allocation_meta (query,
|
||
|
GST_VIDEO_CROP_META_API_TYPE, NULL);
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
ensure_output_resolution (GstLcevcDec * lcevc, guint32 width, guint32 height,
|
||
|
GstVideoCodecState * state)
|
||
|
{
|
||
|
/* Set output state with input resolution to do passthrough */
|
||
|
if (width != lcevc->out_width || height != lcevc->out_height) {
|
||
|
GstVideoCodecState *s;
|
||
|
|
||
|
s = gst_video_decoder_set_output_state (GST_VIDEO_DECODER (lcevc),
|
||
|
GST_VIDEO_INFO_FORMAT (&lcevc->in_info), width, height, state);
|
||
|
if (!s)
|
||
|
return FALSE;
|
||
|
|
||
|
lcevc->out_width = width;
|
||
|
lcevc->out_height = height;
|
||
|
|
||
|
gst_video_codec_state_unref (s);
|
||
|
}
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
gst_lcevc_dec_set_format (GstVideoDecoder * decoder, GstVideoCodecState * state)
|
||
|
{
|
||
|
GstLcevcDec *lcevc = GST_LCEVC_DEC (decoder);
|
||
|
LCEVC_ColorFormat format;
|
||
|
gint par_n, par_d;
|
||
|
guint32 w, h;
|
||
|
|
||
|
/* Make sure format is supported */
|
||
|
format =
|
||
|
gst_lcevc_dec_utils_get_color_format (GST_VIDEO_INFO_FORMAT
|
||
|
(&state->info));
|
||
|
if (format == LCEVC_ColorFormat_Unknown)
|
||
|
return FALSE;
|
||
|
|
||
|
/* Keep input info */
|
||
|
lcevc->in_info = state->info;
|
||
|
|
||
|
/* Output resultion is always twice as big as input resultion divided by
|
||
|
* pixel aspect ratio */
|
||
|
par_n = GST_VIDEO_INFO_PAR_N (&state->info);
|
||
|
if (par_n == 0)
|
||
|
par_n = 1;
|
||
|
par_d = GST_VIDEO_INFO_PAR_D (&state->info);
|
||
|
if (par_d == 0)
|
||
|
par_d = 1;
|
||
|
w = (GST_VIDEO_INFO_WIDTH (&state->info) * 2) / par_d;
|
||
|
h = (GST_VIDEO_INFO_HEIGHT (&state->info) * 2) / par_n;
|
||
|
|
||
|
/* Set pixel aspect ratio back to 1/1 */
|
||
|
GST_VIDEO_INFO_PAR_N (&state->info) = 1;
|
||
|
GST_VIDEO_INFO_PAR_D (&state->info) = 1;
|
||
|
|
||
|
/* Set output resolution */
|
||
|
if (!ensure_output_resolution (lcevc, w, h, state))
|
||
|
return FALSE;
|
||
|
|
||
|
/* We always work with full RAW video frames */
|
||
|
gst_video_decoder_set_subframe_mode (decoder, FALSE);
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static GstVideoCodecFrame *
|
||
|
find_pending_frame_from_picture_handle (GstLcevcDec * lcevc,
|
||
|
LCEVC_PictureHandle picture_handle)
|
||
|
{
|
||
|
GList *iter;
|
||
|
GList *pending_frames;
|
||
|
GstVideoCodecFrame *found = NULL;
|
||
|
|
||
|
pending_frames = gst_video_decoder_get_frames (GST_VIDEO_DECODER (lcevc));
|
||
|
|
||
|
for (iter = pending_frames; iter; iter = g_list_next (iter)) {
|
||
|
GstVideoCodecFrame *frame = (GstVideoCodecFrame *) iter->data;
|
||
|
PictureData *pd;
|
||
|
|
||
|
pd = gst_mini_object_get_qdata (GST_MINI_OBJECT (frame->output_buffer),
|
||
|
GST_LCEVC_DEC_PICTURE_DATA);
|
||
|
if (pd->picture_handle.hdl == picture_handle.hdl) {
|
||
|
found = gst_video_codec_frame_ref (frame);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
g_list_free_full (pending_frames,
|
||
|
(GDestroyNotify) gst_video_codec_frame_unref);
|
||
|
|
||
|
return found;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
receive_enhanced_picture (GstLcevcDec * lcevc)
|
||
|
{
|
||
|
LCEVC_PictureHandle picture_handle = { 0, };
|
||
|
LCEVC_DecodeInformation decode_info = { 0, };
|
||
|
|
||
|
while (LCEVC_ReceiveDecoderPicture (lcevc->decoder_handle, &picture_handle,
|
||
|
&decode_info) == LCEVC_Success) {
|
||
|
LCEVC_PictureDesc pic_desc = { 0, };
|
||
|
GstVideoCodecFrame *received_frame;
|
||
|
GstVideoCropMeta *cmeta;
|
||
|
|
||
|
if (LCEVC_GetPictureDesc (lcevc->decoder_handle, picture_handle,
|
||
|
&pic_desc) != LCEVC_Success) {
|
||
|
GST_ELEMENT_ERROR (lcevc, STREAM, DECODE, (NULL),
|
||
|
("Could not get desciption of received enhanced picutre"));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
GST_INFO_OBJECT (lcevc,
|
||
|
"Received enhanced picture: ts=%ld e=%d w=%d h=%d t=%d b=%d l=%d r=%d",
|
||
|
decode_info.timestamp, decode_info.enhanced, pic_desc.width,
|
||
|
pic_desc.height, pic_desc.cropTop, pic_desc.cropBottom,
|
||
|
pic_desc.cropLeft, pic_desc.cropRight);
|
||
|
|
||
|
received_frame = find_pending_frame_from_picture_handle (lcevc,
|
||
|
picture_handle);
|
||
|
if (received_frame) {
|
||
|
/* Change output allocation if enhanced picutre resolution changed */
|
||
|
if (!ensure_output_resolution (lcevc, pic_desc.width, pic_desc.height,
|
||
|
NULL)) {
|
||
|
gst_video_codec_frame_unref (received_frame);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* Add crop meta if downstream can crop */
|
||
|
if (lcevc->can_crop) {
|
||
|
cmeta = gst_buffer_add_video_crop_meta (received_frame->output_buffer);
|
||
|
cmeta->x = pic_desc.cropLeft;
|
||
|
cmeta->y = pic_desc.cropTop;
|
||
|
cmeta->width =
|
||
|
pic_desc.width - (pic_desc.cropLeft + pic_desc.cropRight);
|
||
|
cmeta->height =
|
||
|
pic_desc.height - (pic_desc.cropTop + pic_desc.cropBottom);
|
||
|
|
||
|
/* Change output caps if crop values changed */
|
||
|
if (lcevc->out_crop_top != pic_desc.cropTop ||
|
||
|
lcevc->out_crop_bottom != pic_desc.cropBottom ||
|
||
|
lcevc->out_crop_left != pic_desc.cropLeft ||
|
||
|
lcevc->out_crop_right != pic_desc.cropRight) {
|
||
|
GstVideoCodecState *s;
|
||
|
|
||
|
lcevc->out_crop_top = pic_desc.cropTop;
|
||
|
lcevc->out_crop_bottom = pic_desc.cropBottom;
|
||
|
lcevc->out_crop_left = pic_desc.cropLeft;
|
||
|
lcevc->out_crop_right = pic_desc.cropRight;
|
||
|
|
||
|
s = gst_video_decoder_get_output_state (GST_VIDEO_DECODER (lcevc));
|
||
|
if (!s) {
|
||
|
gst_video_codec_frame_unref (received_frame);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
s->caps = gst_video_info_to_caps (&s->info);
|
||
|
gst_caps_set_simple (s->caps,
|
||
|
"width", G_TYPE_INT,
|
||
|
pic_desc.width - (pic_desc.cropLeft + pic_desc.cropRight),
|
||
|
"height", G_TYPE_INT,
|
||
|
pic_desc.height - (pic_desc.cropTop + pic_desc.cropBottom), NULL);
|
||
|
gst_video_decoder_negotiate (GST_VIDEO_DECODER (lcevc));
|
||
|
|
||
|
gst_video_codec_state_unref (s);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Finish frame */
|
||
|
received_frame->output_buffer->pts = decode_info.timestamp;
|
||
|
gst_video_decoder_finish_frame (GST_VIDEO_DECODER (lcevc),
|
||
|
received_frame);
|
||
|
gst_video_codec_frame_unref (received_frame);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
receive_base_picture (GstLcevcDec * lcevc)
|
||
|
{
|
||
|
LCEVC_PictureHandle picture_handle = { 0, };
|
||
|
|
||
|
while (LCEVC_ReceiveDecoderBase (lcevc->decoder_handle, &picture_handle)
|
||
|
== LCEVC_Success) {
|
||
|
GST_DEBUG_OBJECT (lcevc, "Received base picture %ld", picture_handle.hdl);
|
||
|
|
||
|
if (LCEVC_FreePicture (lcevc->decoder_handle, picture_handle)
|
||
|
!= LCEVC_Success) {
|
||
|
GST_ELEMENT_ERROR (lcevc, STREAM, DECODE, (NULL),
|
||
|
("Could not free base picture %ld", picture_handle.hdl));
|
||
|
return FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
send_enhancement_data (GstLcevcDec * lcevc, GstBuffer * input_buffer)
|
||
|
{
|
||
|
gboolean ret = FALSE;
|
||
|
GstLcevcMeta *lcevc_meta;
|
||
|
GstMapInfo enhancement_info;
|
||
|
|
||
|
lcevc_meta = gst_buffer_get_lcevc_meta (input_buffer);
|
||
|
if (!lcevc_meta) {
|
||
|
GST_INFO_OBJECT (lcevc,
|
||
|
"Input buffer %ld enhancement data not found, doing passthrough",
|
||
|
input_buffer->pts);
|
||
|
|
||
|
/* Set output state with input resolution to do passthrough */
|
||
|
return ensure_output_resolution (lcevc,
|
||
|
GST_VIDEO_INFO_WIDTH (&lcevc->in_info),
|
||
|
GST_VIDEO_INFO_HEIGHT (&lcevc->in_info), NULL);
|
||
|
}
|
||
|
|
||
|
if (!gst_buffer_map (lcevc_meta->enhancement_data, &enhancement_info,
|
||
|
GST_MAP_READ)) {
|
||
|
GST_INFO_OBJECT (lcevc, "Could not map input buffer %ld enhancement data",
|
||
|
input_buffer->pts);
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
if (LCEVC_SendDecoderEnhancementData (lcevc->decoder_handle,
|
||
|
input_buffer->pts, TRUE, enhancement_info.data,
|
||
|
enhancement_info.size) != LCEVC_Success) {
|
||
|
GST_INFO_OBJECT (lcevc,
|
||
|
"Could not send input buffer %ld enhancement data with size %ld",
|
||
|
input_buffer->pts, enhancement_info.size);
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
GST_INFO_OBJECT (lcevc,
|
||
|
"Sent input buffer %ld enhancement data with size %lu",
|
||
|
input_buffer->pts, enhancement_info.size);
|
||
|
|
||
|
ret = TRUE;
|
||
|
|
||
|
done:
|
||
|
gst_buffer_unmap (lcevc_meta->enhancement_data, &enhancement_info);
|
||
|
return ret;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
send_base_picture (GstLcevcDec * lcevc, GstBuffer * input_buffer)
|
||
|
{
|
||
|
gboolean ret = FALSE;
|
||
|
LCEVC_PictureHandle picture_handle;
|
||
|
GstVideoFrame frame = { 0, };
|
||
|
|
||
|
if (!gst_video_frame_map (&frame, &lcevc->in_info, input_buffer,
|
||
|
GST_MAP_READ)) {
|
||
|
GST_ELEMENT_ERROR (lcevc, STREAM, DECODE, (NULL),
|
||
|
("Could not map input buffer %ld", input_buffer->pts));
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
if (!gst_lcevc_dec_utils_alloc_picture_handle (lcevc->decoder_handle, &frame,
|
||
|
&picture_handle)) {
|
||
|
GST_ELEMENT_ERROR (lcevc, STREAM, DECODE, (NULL),
|
||
|
("Could not allocate input picture handle %ld", input_buffer->pts));
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
if (LCEVC_SendDecoderBase (lcevc->decoder_handle, input_buffer->pts, TRUE,
|
||
|
picture_handle, 1000000, NULL) != LCEVC_Success) {
|
||
|
GST_ELEMENT_ERROR (lcevc, STREAM, DECODE, (NULL),
|
||
|
("Could not send input buffer %ld base picture", input_buffer->pts));
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
GST_INFO_OBJECT (lcevc, "Sent input buffer %ld base picture",
|
||
|
input_buffer->pts);
|
||
|
|
||
|
ret = TRUE;
|
||
|
|
||
|
done:
|
||
|
gst_video_frame_unmap (&frame);
|
||
|
return ret;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
send_enhanced_picture (GstLcevcDec * lcevc, GstVideoCodecFrame * frame)
|
||
|
{
|
||
|
GstVideoCodecState *s = NULL;
|
||
|
GstVideoFrame map = { 0, };
|
||
|
PictureData *pd;
|
||
|
gboolean res = FALSE;
|
||
|
|
||
|
/* Get output state */
|
||
|
s = gst_video_decoder_get_output_state (GST_VIDEO_DECODER (lcevc));
|
||
|
if (!s) {
|
||
|
GST_ELEMENT_ERROR (lcevc, STREAM, DECODE, (NULL),
|
||
|
("Could not get output state"));
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
/* Map the video frame */
|
||
|
if (!gst_video_frame_map (&map, &s->info, frame->output_buffer,
|
||
|
GST_MAP_WRITE)) {
|
||
|
GST_ELEMENT_ERROR (lcevc, STREAM, DECODE, (NULL),
|
||
|
("Could not map output buffer for writting"));
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
/* Get pic data if any and size didn't change, otherwise create a new one */
|
||
|
pd = gst_mini_object_get_qdata (GST_MINI_OBJECT (frame->output_buffer),
|
||
|
GST_LCEVC_DEC_PICTURE_DATA);
|
||
|
if (!pd || pd->width != lcevc->out_width || pd->height != lcevc->out_height) {
|
||
|
/* Create picture data */
|
||
|
pd = picture_data_new (lcevc->decoder_handle, &map);
|
||
|
if (!pd) {
|
||
|
GST_ELEMENT_ERROR (lcevc, STREAM, DECODE, (NULL),
|
||
|
("Could not create output picture data"));
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
/* Set picture data on buffer */
|
||
|
gst_mini_object_set_qdata (GST_MINI_OBJECT (frame->output_buffer),
|
||
|
GST_LCEVC_DEC_PICTURE_DATA, pd, picture_data_free);
|
||
|
}
|
||
|
|
||
|
/* Send enhanced picture */
|
||
|
if (LCEVC_SendDecoderPicture (lcevc->decoder_handle, pd->picture_handle)
|
||
|
!= LCEVC_Success) {
|
||
|
GST_ELEMENT_ERROR (lcevc, STREAM, DECODE, (NULL),
|
||
|
("Could not send output buffer enhanced picture"));
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
res = TRUE;
|
||
|
|
||
|
done:
|
||
|
gst_video_frame_unmap (&map);
|
||
|
g_clear_pointer (&s, gst_video_codec_state_unref);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
static GstFlowReturn
|
||
|
gst_lcevc_dec_handle_frame (GstVideoDecoder * decoder,
|
||
|
GstVideoCodecFrame * frame)
|
||
|
{
|
||
|
GstLcevcDec *lcevc = GST_LCEVC_DEC (decoder);
|
||
|
|
||
|
GST_DEBUG_OBJECT (decoder, "Handling frame %d with timestamp %ld",
|
||
|
frame->system_frame_number, frame->input_buffer->pts);
|
||
|
|
||
|
if (!send_enhancement_data (lcevc, frame->input_buffer))
|
||
|
goto error;
|
||
|
|
||
|
if (!send_base_picture (lcevc, frame->input_buffer))
|
||
|
goto error;
|
||
|
|
||
|
if (gst_video_decoder_allocate_output_frame (decoder, frame) != GST_FLOW_OK) {
|
||
|
GST_ELEMENT_ERROR (decoder, STREAM, DECODE, (NULL),
|
||
|
("Could not allocate output frame"));
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
if (!send_enhanced_picture (lcevc, frame))
|
||
|
goto error;
|
||
|
|
||
|
if (!receive_enhanced_picture (lcevc))
|
||
|
goto error;
|
||
|
|
||
|
if (!receive_base_picture (lcevc))
|
||
|
goto error;
|
||
|
|
||
|
return GST_FLOW_OK;
|
||
|
|
||
|
error:
|
||
|
gst_video_decoder_drop_frame (decoder, frame);
|
||
|
return GST_FLOW_ERROR;
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
gst_lcevc_dec_class_init (GstLcevcDecClass * klass)
|
||
|
{
|
||
|
GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
|
||
|
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
|
||
|
GstVideoDecoderClass *bt_class = GST_VIDEO_DECODER_CLASS (klass);
|
||
|
|
||
|
gst_element_class_add_static_pad_template (element_class,
|
||
|
&sink_template_factory);
|
||
|
gst_element_class_add_static_pad_template (element_class,
|
||
|
&src_template_factory);
|
||
|
gst_element_class_set_static_metadata (element_class,
|
||
|
"LCEVC Decoder", "Codec/Decoder/Video",
|
||
|
"Enhances video frames using attached LCEVC metadata",
|
||
|
"Julian Bouzas <julian.bouzas@collabora.com>");
|
||
|
|
||
|
gobject_class->set_property = gst_lcevc_dec_set_property;
|
||
|
gobject_class->get_property = gst_lcevc_dec_get_property;
|
||
|
|
||
|
bt_class->start = gst_lcevc_dec_start;
|
||
|
bt_class->stop = gst_lcevc_dec_stop;
|
||
|
bt_class->decide_allocation = gst_lcevc_dec_decide_allocation;
|
||
|
bt_class->set_format = gst_lcevc_dec_set_format;
|
||
|
bt_class->handle_frame = gst_lcevc_dec_handle_frame;
|
||
|
|
||
|
g_object_class_install_property (gobject_class, PROP_VERBOSE,
|
||
|
g_param_spec_boolean ("verbose", "Verbose",
|
||
|
"Output status information of the LCEVC Decoder SDK",
|
||
|
FALSE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
||
|
|
||
|
g_object_class_install_property (gobject_class, PROP_MAX_WIDTH,
|
||
|
g_param_spec_int ("max-width", "Maximum Width",
|
||
|
"The maximum width for the LCEVC decoder (0 = default)",
|
||
|
0, G_MAXINT, DEFAULT_MAX_WIDTH,
|
||
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
||
|
|
||
|
g_object_class_install_property (gobject_class, PROP_MAX_HEIGHT,
|
||
|
g_param_spec_int ("max-height", "Maximum Height",
|
||
|
"The maximum height for the LCEVC decoder (0 = default)",
|
||
|
0, G_MAXINT, DEFAULT_MAX_HEIGHT,
|
||
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
||
|
|
||
|
g_object_class_install_property (gobject_class, PROP_MAX_LATENCY,
|
||
|
g_param_spec_int ("max-latency", "Maximum Latency",
|
||
|
"The maximum latency in frames for the LCEVC decoder (0 = default)",
|
||
|
0, G_MAXINT, DEFAULT_MAX_LATENCY,
|
||
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
||
|
}
|