mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-11 10:56:38 +00:00
636690f2aa
This new LCEVC decoder plugin is meant to implement all LCEVC decoder elements. For now, it only implements the LCEVC enhancement decoder (lcevcdec) element. This element essentially enhances raw video frames using the LCEVC metadata attached to input buffers into a higher resolution frame. The element is only meant to be used after any base decoder (eg avdec_h264). Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/7330>
720 lines
21 KiB
C
720 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));
|
|
}
|