mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-29 13:11:06 +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>
92 lines
3 KiB
C
92 lines
3 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.
|
|
*/
|
|
|
|
#include "gstlcevcdecutils.h"
|
|
|
|
LCEVC_ColorFormat
|
|
gst_lcevc_dec_utils_get_color_format (GstVideoFormat format)
|
|
{
|
|
switch (format) {
|
|
case GST_VIDEO_FORMAT_I420:
|
|
return LCEVC_I420_8;
|
|
case GST_VIDEO_FORMAT_NV12:
|
|
return LCEVC_NV12_8;
|
|
case GST_VIDEO_FORMAT_NV21:
|
|
return LCEVC_NV21_8;
|
|
case GST_VIDEO_FORMAT_RGB:
|
|
return LCEVC_RGB_8;
|
|
case GST_VIDEO_FORMAT_BGR:
|
|
return LCEVC_BGR_8;
|
|
case GST_VIDEO_FORMAT_RGBA:
|
|
return LCEVC_RGBA_8;
|
|
case GST_VIDEO_FORMAT_BGRA:
|
|
return LCEVC_BGRA_8;
|
|
case GST_VIDEO_FORMAT_ARGB:
|
|
return LCEVC_ARGB_8;
|
|
case GST_VIDEO_FORMAT_ABGR:
|
|
return LCEVC_ABGR_8;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return LCEVC_ColorFormat_Unknown;
|
|
}
|
|
|
|
gboolean
|
|
gst_lcevc_dec_utils_alloc_picture_handle (LCEVC_DecoderHandle decoder_handle,
|
|
GstVideoFrame * frame, LCEVC_PictureHandle * picture_handle)
|
|
{
|
|
LCEVC_PictureDesc picture_desc = { 0, };
|
|
LCEVC_PictureBufferDesc buffer_desc = { 0, };
|
|
LCEVC_PicturePlaneDesc plane_desc[GST_VIDEO_MAX_PLANES] = { 0, };
|
|
LCEVC_ColorFormat fmt;
|
|
guint i;
|
|
|
|
fmt = gst_lcevc_dec_utils_get_color_format (GST_VIDEO_FRAME_FORMAT (frame));
|
|
if (fmt == LCEVC_ColorFormat_Unknown)
|
|
return FALSE;
|
|
|
|
/* Set LCEVC Picture Description */
|
|
if (LCEVC_DefaultPictureDesc (&picture_desc, fmt,
|
|
GST_VIDEO_FRAME_WIDTH (frame), GST_VIDEO_FRAME_HEIGHT (frame))
|
|
!= LCEVC_Success)
|
|
return FALSE;
|
|
|
|
/* Set buffer description */
|
|
buffer_desc.data = GST_VIDEO_FRAME_PLANE_DATA (frame, 0);
|
|
buffer_desc.byteSize = GST_VIDEO_FRAME_SIZE (frame);
|
|
buffer_desc.access = LCEVC_Access_Write;
|
|
|
|
/* Set plane description */
|
|
for (i = 0; i < GST_VIDEO_FRAME_N_PLANES (frame); i++) {
|
|
plane_desc[i].firstSample = GST_VIDEO_FRAME_PLANE_DATA (frame, i);
|
|
plane_desc[i].rowByteStride = GST_VIDEO_FRAME_PLANE_STRIDE (frame, i);
|
|
}
|
|
|
|
/* FIXME: We set the stride on all the array (needed for LCEVCdec 2.0.0) */
|
|
for (; i < GST_VIDEO_MAX_PLANES; i++)
|
|
plane_desc[i].rowByteStride = GST_VIDEO_FRAME_WIDTH (frame);
|
|
|
|
/* Allocate LCEVC Picture */
|
|
if (LCEVC_AllocPictureExternal (decoder_handle, &picture_desc, &buffer_desc,
|
|
plane_desc, picture_handle) != LCEVC_Success)
|
|
return FALSE;
|
|
|
|
return TRUE;
|
|
}
|