2013-07-29 07:46:11 +00:00
|
|
|
/*
|
|
|
|
* gstvaapiencoder_h264.c - H.264 encoder
|
|
|
|
*
|
2014-01-22 17:54:14 +00:00
|
|
|
* Copyright (C) 2012-2014 Intel Corporation
|
2014-01-22 17:49:20 +00:00
|
|
|
* Author: Wind Yuan <feng.yuan@intel.com>
|
|
|
|
* Author: Gwenole Beauchesne <gwenole.beauchesne@intel.com>
|
2013-07-29 07:46:11 +00:00
|
|
|
*
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU Lesser General Public License
|
|
|
|
* as published by the Free Software Foundation; either version 2.1
|
|
|
|
* 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
|
|
|
|
* Lesser General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
|
|
* License along with this library; if not, write to the Free
|
|
|
|
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
|
|
|
* Boston, MA 02110-1301 USA
|
|
|
|
*/
|
|
|
|
|
2015-02-24 15:14:33 +00:00
|
|
|
/* GValueArray has deprecated without providing an alternative in glib >= 2.32
|
|
|
|
* See https://bugzilla.gnome.org/show_bug.cgi?id=667228
|
|
|
|
*/
|
|
|
|
#define GLIB_DISABLE_DEPRECATION_WARNINGS
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
#include "sysdeps.h"
|
2014-01-10 13:05:40 +00:00
|
|
|
#include <va/va.h>
|
|
|
|
#include <va/va_enc_h264.h>
|
|
|
|
#include <gst/base/gstbitwriter.h>
|
2014-06-03 12:30:39 +00:00
|
|
|
#include <gst/codecparsers/gsth264parser.h>
|
2013-07-29 07:46:11 +00:00
|
|
|
#include "gstvaapicompat.h"
|
2014-01-21 14:23:01 +00:00
|
|
|
#include "gstvaapiencoder_priv.h"
|
2013-07-29 07:46:11 +00:00
|
|
|
#include "gstvaapiencoder_h264.h"
|
2014-01-21 14:23:01 +00:00
|
|
|
#include "gstvaapiutils_h264.h"
|
2014-01-10 16:02:44 +00:00
|
|
|
#include "gstvaapiutils_h264_priv.h"
|
2013-12-03 15:11:46 +00:00
|
|
|
#include "gstvaapicodedbufferproxy_priv.h"
|
2013-07-29 07:46:11 +00:00
|
|
|
#include "gstvaapisurface.h"
|
|
|
|
|
|
|
|
#define DEBUG 1
|
|
|
|
#include "gstvaapidebug.h"
|
|
|
|
|
2015-02-24 15:14:33 +00:00
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
/* Define the maximum number of views supported */
|
2015-02-24 15:14:33 +00:00
|
|
|
#define MAX_NUM_VIEWS 10
|
|
|
|
|
|
|
|
/* Define the maximum value for view-id */
|
|
|
|
#define MAX_VIEW_ID 1023
|
2014-02-17 03:10:26 +00:00
|
|
|
|
2014-01-21 14:23:01 +00:00
|
|
|
/* Define the maximum IDR period */
|
|
|
|
#define MAX_IDR_PERIOD 512
|
|
|
|
|
2014-01-21 18:04:41 +00:00
|
|
|
/* Default CPB length (in milliseconds) */
|
|
|
|
#define DEFAULT_CPB_LENGTH 1500
|
|
|
|
|
|
|
|
/* Scale factor for CPB size (HRD cpb_size_scale: min = 4) */
|
|
|
|
#define SX_CPB_SIZE 4
|
|
|
|
|
2014-01-22 13:43:24 +00:00
|
|
|
/* Scale factor for bitrate (HRD bit_rate_scale: min = 6) */
|
|
|
|
#define SX_BITRATE 6
|
|
|
|
|
2014-01-06 14:10:36 +00:00
|
|
|
/* Define default rate control mode ("constant-qp") */
|
|
|
|
#define DEFAULT_RATECONTROL GST_VAAPI_RATECONTROL_CQP
|
|
|
|
|
|
|
|
/* Supported set of VA rate controls, within this implementation */
|
|
|
|
#define SUPPORTED_RATECONTROLS \
|
2014-01-23 14:10:11 +00:00
|
|
|
(GST_VAAPI_RATECONTROL_MASK (CQP) | \
|
2014-01-06 14:10:36 +00:00
|
|
|
GST_VAAPI_RATECONTROL_MASK (CBR) | \
|
|
|
|
GST_VAAPI_RATECONTROL_MASK (VBR) | \
|
|
|
|
GST_VAAPI_RATECONTROL_MASK (VBR_CONSTRAINED))
|
|
|
|
|
2014-01-13 09:48:25 +00:00
|
|
|
/* Supported set of tuning options, within this implementation */
|
2014-01-13 10:11:25 +00:00
|
|
|
#define SUPPORTED_TUNE_OPTIONS \
|
|
|
|
(GST_VAAPI_ENCODER_TUNE_MASK (NONE) | \
|
2016-05-13 08:44:57 +00:00
|
|
|
GST_VAAPI_ENCODER_TUNE_MASK (HIGH_COMPRESSION) | \
|
|
|
|
GST_VAAPI_ENCODER_TUNE_MASK (LOW_POWER))
|
2014-01-13 09:48:25 +00:00
|
|
|
|
2014-01-23 14:13:06 +00:00
|
|
|
/* Supported set of VA packed headers, within this implementation */
|
|
|
|
#define SUPPORTED_PACKED_HEADERS \
|
|
|
|
(VA_ENC_PACKED_HEADER_SEQUENCE | \
|
2014-05-06 21:09:45 +00:00
|
|
|
VA_ENC_PACKED_HEADER_PICTURE | \
|
2014-05-06 21:12:39 +00:00
|
|
|
VA_ENC_PACKED_HEADER_SLICE | \
|
2015-07-02 18:00:14 +00:00
|
|
|
VA_ENC_PACKED_HEADER_RAW_DATA | \
|
|
|
|
VA_ENC_PACKED_HEADER_MISC)
|
2014-01-23 14:13:06 +00:00
|
|
|
|
2014-06-03 12:30:39 +00:00
|
|
|
#define GST_H264_NAL_REF_IDC_NONE 0
|
|
|
|
#define GST_H264_NAL_REF_IDC_LOW 1
|
|
|
|
#define GST_H264_NAL_REF_IDC_MEDIUM 2
|
|
|
|
#define GST_H264_NAL_REF_IDC_HIGH 3
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2015-07-02 18:00:14 +00:00
|
|
|
/* only for internal usage, values won't be equal to actual payload type */
|
|
|
|
typedef enum
|
|
|
|
{
|
|
|
|
GST_VAAPI_H264_SEI_UNKNOWN = 0,
|
|
|
|
GST_VAAPI_H264_SEI_BUF_PERIOD = (1 << 0),
|
|
|
|
GST_VAAPI_H264_SEI_PIC_TIMING = (1 << 1)
|
|
|
|
} GstVaapiH264SeiPayloadType;
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
typedef struct
|
|
|
|
{
|
|
|
|
GstVaapiSurfaceProxy *pic;
|
|
|
|
guint poc;
|
|
|
|
guint frame_num;
|
|
|
|
} GstVaapiEncoderH264Ref;
|
|
|
|
|
|
|
|
typedef enum
|
|
|
|
{
|
|
|
|
GST_VAAPI_ENC_H264_REORD_NONE = 0,
|
|
|
|
GST_VAAPI_ENC_H264_REORD_DUMP_FRAMES = 1,
|
|
|
|
GST_VAAPI_ENC_H264_REORD_WAIT_FRAMES = 2
|
|
|
|
} GstVaapiEncH264ReorderState;
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
typedef struct _GstVaapiH264ViewRefPool
|
|
|
|
{
|
|
|
|
GQueue ref_list;
|
|
|
|
guint max_ref_frames;
|
|
|
|
guint max_reflist0_count;
|
|
|
|
guint max_reflist1_count;
|
|
|
|
} GstVaapiH264ViewRefPool;
|
|
|
|
|
|
|
|
typedef struct _GstVaapiH264ViewReorderPool
|
|
|
|
{
|
|
|
|
GQueue reorder_frame_list;
|
|
|
|
guint reorder_state;
|
|
|
|
guint frame_index;
|
2016-02-02 16:59:57 +00:00
|
|
|
guint frame_count; /* monotonically increasing with in every idr period */
|
2014-02-17 03:10:26 +00:00
|
|
|
guint cur_frame_num;
|
|
|
|
guint cur_present_index;
|
|
|
|
} GstVaapiH264ViewReorderPool;
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
static inline gboolean
|
|
|
|
_poc_greater_than (guint poc1, guint poc2, guint max_poc)
|
|
|
|
{
|
|
|
|
return (((poc1 - poc2) & (max_poc - 1)) < max_poc / 2);
|
|
|
|
}
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
/* Get slice_type value for H.264 specification */
|
|
|
|
static guint8
|
|
|
|
h264_get_slice_type (GstVaapiPictureType type)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
|
|
|
switch (type) {
|
|
|
|
case GST_VAAPI_PICTURE_TYPE_I:
|
2014-06-03 12:30:39 +00:00
|
|
|
return GST_H264_I_SLICE;
|
2013-07-29 07:46:11 +00:00
|
|
|
case GST_VAAPI_PICTURE_TYPE_P:
|
2014-06-03 12:30:39 +00:00
|
|
|
return GST_H264_P_SLICE;
|
2013-07-29 07:46:11 +00:00
|
|
|
case GST_VAAPI_PICTURE_TYPE_B:
|
2014-06-03 12:30:39 +00:00
|
|
|
return GST_H264_B_SLICE;
|
2013-07-29 07:46:11 +00:00
|
|
|
default:
|
2014-06-03 12:30:39 +00:00
|
|
|
break;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
/* Get log2_max_frame_num value for H.264 specification */
|
|
|
|
static guint
|
|
|
|
h264_get_log2_max_frame_num (guint num)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
|
|
|
guint ret = 0;
|
|
|
|
|
|
|
|
while (num) {
|
|
|
|
++ret;
|
|
|
|
num >>= 1;
|
|
|
|
}
|
|
|
|
if (ret <= 4)
|
|
|
|
ret = 4;
|
|
|
|
else if (ret > 10)
|
|
|
|
ret = 10;
|
2014-01-21 15:05:22 +00:00
|
|
|
/* must be greater than 4 */
|
2013-07-29 07:46:11 +00:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2014-01-22 10:25:13 +00:00
|
|
|
/* Determines the cpbBrNalFactor based on the supplied profile */
|
|
|
|
static guint
|
|
|
|
h264_get_cpb_nal_factor (GstVaapiProfile profile)
|
|
|
|
{
|
|
|
|
guint f;
|
|
|
|
|
|
|
|
/* Table A-2 */
|
|
|
|
switch (profile) {
|
|
|
|
case GST_VAAPI_PROFILE_H264_HIGH:
|
|
|
|
f = 1500;
|
|
|
|
break;
|
|
|
|
case GST_VAAPI_PROFILE_H264_HIGH10:
|
|
|
|
f = 3600;
|
|
|
|
break;
|
|
|
|
case GST_VAAPI_PROFILE_H264_HIGH_422:
|
|
|
|
case GST_VAAPI_PROFILE_H264_HIGH_444:
|
|
|
|
f = 4800;
|
|
|
|
break;
|
2014-06-26 11:39:52 +00:00
|
|
|
case GST_VAAPI_PROFILE_H264_MULTIVIEW_HIGH:
|
|
|
|
case GST_VAAPI_PROFILE_H264_STEREO_HIGH:
|
|
|
|
f = 1500; /* H.10.2.1 (r) */
|
|
|
|
break;
|
2014-01-22 10:25:13 +00:00
|
|
|
default:
|
|
|
|
f = 1200;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return f;
|
|
|
|
}
|
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
/* ------------------------------------------------------------------------- */
|
|
|
|
/* --- H.264 Bitstream Writer --- */
|
|
|
|
/* ------------------------------------------------------------------------- */
|
|
|
|
|
|
|
|
#define WRITE_UINT32(bs, val, nbits) do { \
|
|
|
|
if (!gst_bit_writer_put_bits_uint32 (bs, val, nbits)) { \
|
|
|
|
GST_WARNING ("failed to write uint32, nbits: %d", nbits); \
|
|
|
|
goto bs_error; \
|
|
|
|
} \
|
|
|
|
} while (0)
|
|
|
|
|
|
|
|
#define WRITE_UE(bs, val) do { \
|
|
|
|
if (!bs_write_ue (bs, val)) { \
|
|
|
|
GST_WARNING ("failed to write ue(v)"); \
|
|
|
|
goto bs_error; \
|
|
|
|
} \
|
|
|
|
} while (0)
|
|
|
|
|
|
|
|
#define WRITE_SE(bs, val) do { \
|
|
|
|
if (!bs_write_se (bs, val)) { \
|
|
|
|
GST_WARNING ("failed to write se(v)"); \
|
|
|
|
goto bs_error; \
|
|
|
|
} \
|
|
|
|
} while (0)
|
|
|
|
|
|
|
|
/* Write an unsigned integer Exp-Golomb-coded syntax element. i.e. ue(v) */
|
|
|
|
static gboolean
|
|
|
|
bs_write_ue (GstBitWriter * bs, guint32 value)
|
|
|
|
{
|
|
|
|
guint32 size_in_bits = 0;
|
|
|
|
guint32 tmp_value = ++value;
|
|
|
|
|
|
|
|
while (tmp_value) {
|
|
|
|
++size_in_bits;
|
|
|
|
tmp_value >>= 1;
|
|
|
|
}
|
|
|
|
if (size_in_bits > 1
|
|
|
|
&& !gst_bit_writer_put_bits_uint32 (bs, 0, size_in_bits - 1))
|
|
|
|
return FALSE;
|
|
|
|
if (!gst_bit_writer_put_bits_uint32 (bs, value, size_in_bits))
|
|
|
|
return FALSE;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Write a signed integer Exp-Golomb-coded syntax element. i.e. se(v) */
|
|
|
|
static gboolean
|
|
|
|
bs_write_se (GstBitWriter * bs, gint32 value)
|
|
|
|
{
|
|
|
|
guint32 new_val;
|
|
|
|
|
|
|
|
if (value <= 0)
|
|
|
|
new_val = -(value << 1);
|
|
|
|
else
|
|
|
|
new_val = (value << 1) - 1;
|
|
|
|
|
|
|
|
if (!bs_write_ue (bs, new_val))
|
|
|
|
return FALSE;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Write the NAL unit header */
|
|
|
|
static gboolean
|
|
|
|
bs_write_nal_header (GstBitWriter * bs, guint32 nal_ref_idc,
|
|
|
|
guint32 nal_unit_type)
|
|
|
|
{
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
|
|
|
WRITE_UINT32 (bs, nal_ref_idc, 2);
|
|
|
|
WRITE_UINT32 (bs, nal_unit_type, 5);
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write NAL unit header");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
/* Write the MVC NAL unit header extension */
|
|
|
|
static gboolean
|
|
|
|
bs_write_nal_header_mvc_extension (GstBitWriter * bs,
|
|
|
|
GstVaapiEncPicture * picture, guint32 view_id)
|
|
|
|
{
|
|
|
|
guint32 svc_extension_flag = 0;
|
|
|
|
guint32 non_idr_flag = 1;
|
|
|
|
guint32 priority_id = 0;
|
|
|
|
guint32 temporal_id = 0;
|
|
|
|
guint32 anchor_pic_flag = 0;
|
|
|
|
guint32 inter_view_flag = 0;
|
|
|
|
|
|
|
|
if (GST_VAAPI_ENC_PICTURE_IS_IDR (picture))
|
|
|
|
non_idr_flag = 0;
|
|
|
|
|
|
|
|
if (picture->type == GST_VAAPI_PICTURE_TYPE_I)
|
|
|
|
anchor_pic_flag = 1;
|
|
|
|
/* svc_extension_flag == 0 for mvc stream */
|
|
|
|
WRITE_UINT32 (bs, svc_extension_flag, 1);
|
|
|
|
|
|
|
|
WRITE_UINT32 (bs, non_idr_flag, 1);
|
|
|
|
WRITE_UINT32 (bs, priority_id, 6);
|
|
|
|
WRITE_UINT32 (bs, view_id, 10);
|
|
|
|
WRITE_UINT32 (bs, temporal_id, 3);
|
|
|
|
WRITE_UINT32 (bs, anchor_pic_flag, 1);
|
|
|
|
WRITE_UINT32 (bs, inter_view_flag, 1);
|
|
|
|
WRITE_UINT32 (bs, 1, 1);
|
|
|
|
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write NAL unit header");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
/* Write the NAL unit trailing bits */
|
|
|
|
static gboolean
|
|
|
|
bs_write_trailing_bits (GstBitWriter * bs)
|
|
|
|
{
|
|
|
|
if (!gst_bit_writer_put_bits_uint32 (bs, 1, 1))
|
|
|
|
goto bs_error;
|
|
|
|
gst_bit_writer_align_bytes_unchecked (bs, 0);
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write NAL unit trailing bits");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Write an SPS NAL unit */
|
|
|
|
static gboolean
|
2014-02-14 07:33:15 +00:00
|
|
|
bs_write_sps_data (GstBitWriter * bs,
|
2014-01-21 18:04:41 +00:00
|
|
|
const VAEncSequenceParameterBufferH264 * seq_param, GstVaapiProfile profile,
|
|
|
|
const VAEncMiscParameterHRD * hrd_params)
|
2014-01-21 14:28:34 +00:00
|
|
|
{
|
|
|
|
guint8 profile_idc;
|
|
|
|
guint32 constraint_set0_flag, constraint_set1_flag;
|
|
|
|
guint32 constraint_set2_flag, constraint_set3_flag;
|
|
|
|
guint32 gaps_in_frame_num_value_allowed_flag = 0; // ??
|
|
|
|
gboolean nal_hrd_parameters_present_flag;
|
|
|
|
|
|
|
|
guint32 b_qpprime_y_zero_transform_bypass = 0;
|
|
|
|
guint32 residual_color_transform_flag = 0;
|
|
|
|
guint32 pic_height_in_map_units =
|
|
|
|
(seq_param->seq_fields.bits.frame_mbs_only_flag ?
|
|
|
|
seq_param->picture_height_in_mbs : seq_param->picture_height_in_mbs / 2);
|
|
|
|
guint32 mb_adaptive_frame_field =
|
|
|
|
!seq_param->seq_fields.bits.frame_mbs_only_flag;
|
|
|
|
guint32 i = 0;
|
|
|
|
|
|
|
|
profile_idc = gst_vaapi_utils_h264_get_profile_idc (profile);
|
|
|
|
constraint_set0_flag = /* A.2.1 (baseline profile constraints) */
|
|
|
|
profile == GST_VAAPI_PROFILE_H264_BASELINE ||
|
|
|
|
profile == GST_VAAPI_PROFILE_H264_CONSTRAINED_BASELINE;
|
|
|
|
constraint_set1_flag = /* A.2.2 (main profile constraints) */
|
|
|
|
profile == GST_VAAPI_PROFILE_H264_MAIN ||
|
|
|
|
profile == GST_VAAPI_PROFILE_H264_CONSTRAINED_BASELINE;
|
|
|
|
constraint_set2_flag = 0;
|
|
|
|
constraint_set3_flag = 0;
|
|
|
|
|
|
|
|
/* profile_idc */
|
|
|
|
WRITE_UINT32 (bs, profile_idc, 8);
|
|
|
|
/* constraint_set0_flag */
|
|
|
|
WRITE_UINT32 (bs, constraint_set0_flag, 1);
|
|
|
|
/* constraint_set1_flag */
|
|
|
|
WRITE_UINT32 (bs, constraint_set1_flag, 1);
|
|
|
|
/* constraint_set2_flag */
|
|
|
|
WRITE_UINT32 (bs, constraint_set2_flag, 1);
|
|
|
|
/* constraint_set3_flag */
|
|
|
|
WRITE_UINT32 (bs, constraint_set3_flag, 1);
|
|
|
|
/* reserved_zero_4bits */
|
|
|
|
WRITE_UINT32 (bs, 0, 4);
|
|
|
|
/* level_idc */
|
|
|
|
WRITE_UINT32 (bs, seq_param->level_idc, 8);
|
|
|
|
/* seq_parameter_set_id */
|
|
|
|
WRITE_UE (bs, seq_param->seq_parameter_set_id);
|
|
|
|
|
2014-02-14 07:33:15 +00:00
|
|
|
if (profile == GST_VAAPI_PROFILE_H264_HIGH ||
|
|
|
|
profile == GST_VAAPI_PROFILE_H264_MULTIVIEW_HIGH ||
|
|
|
|
profile == GST_VAAPI_PROFILE_H264_STEREO_HIGH) {
|
2014-01-21 14:28:34 +00:00
|
|
|
/* for high profile */
|
|
|
|
/* chroma_format_idc = 1, 4:2:0 */
|
|
|
|
WRITE_UE (bs, seq_param->seq_fields.bits.chroma_format_idc);
|
|
|
|
if (3 == seq_param->seq_fields.bits.chroma_format_idc) {
|
|
|
|
WRITE_UINT32 (bs, residual_color_transform_flag, 1);
|
|
|
|
}
|
|
|
|
/* bit_depth_luma_minus8 */
|
|
|
|
WRITE_UE (bs, seq_param->bit_depth_luma_minus8);
|
|
|
|
/* bit_depth_chroma_minus8 */
|
|
|
|
WRITE_UE (bs, seq_param->bit_depth_chroma_minus8);
|
|
|
|
/* b_qpprime_y_zero_transform_bypass */
|
|
|
|
WRITE_UINT32 (bs, b_qpprime_y_zero_transform_bypass, 1);
|
2014-01-21 15:05:22 +00:00
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
/* seq_scaling_matrix_present_flag */
|
2014-01-21 15:05:22 +00:00
|
|
|
g_assert (seq_param->seq_fields.bits.seq_scaling_matrix_present_flag == 0);
|
2014-01-21 14:28:34 +00:00
|
|
|
WRITE_UINT32 (bs,
|
|
|
|
seq_param->seq_fields.bits.seq_scaling_matrix_present_flag, 1);
|
|
|
|
|
|
|
|
#if 0
|
|
|
|
if (seq_param->seq_fields.bits.seq_scaling_matrix_present_flag) {
|
|
|
|
for (i = 0;
|
|
|
|
i < (seq_param->seq_fields.bits.chroma_format_idc != 3 ? 8 : 12);
|
|
|
|
i++) {
|
|
|
|
gst_bit_writer_put_bits_uint8 (bs,
|
|
|
|
seq_param->seq_fields.bits.seq_scaling_list_present_flag, 1);
|
|
|
|
if (seq_param->seq_fields.bits.seq_scaling_list_present_flag) {
|
|
|
|
g_assert (0);
|
|
|
|
/* FIXME, need write scaling list if seq_scaling_matrix_present_flag ==1 */
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
/* log2_max_frame_num_minus4 */
|
|
|
|
WRITE_UE (bs, seq_param->seq_fields.bits.log2_max_frame_num_minus4);
|
|
|
|
/* pic_order_cnt_type */
|
|
|
|
WRITE_UE (bs, seq_param->seq_fields.bits.pic_order_cnt_type);
|
|
|
|
|
|
|
|
if (seq_param->seq_fields.bits.pic_order_cnt_type == 0) {
|
|
|
|
/* log2_max_pic_order_cnt_lsb_minus4 */
|
|
|
|
WRITE_UE (bs, seq_param->seq_fields.bits.log2_max_pic_order_cnt_lsb_minus4);
|
|
|
|
} else if (seq_param->seq_fields.bits.pic_order_cnt_type == 1) {
|
2014-01-21 15:05:22 +00:00
|
|
|
g_assert (0 && "only POC type 0 is supported");
|
2014-01-21 14:28:34 +00:00
|
|
|
WRITE_UINT32 (bs,
|
|
|
|
seq_param->seq_fields.bits.delta_pic_order_always_zero_flag, 1);
|
|
|
|
WRITE_SE (bs, seq_param->offset_for_non_ref_pic);
|
|
|
|
WRITE_SE (bs, seq_param->offset_for_top_to_bottom_field);
|
|
|
|
WRITE_UE (bs, seq_param->num_ref_frames_in_pic_order_cnt_cycle);
|
|
|
|
for (i = 0; i < seq_param->num_ref_frames_in_pic_order_cnt_cycle; i++) {
|
|
|
|
WRITE_SE (bs, seq_param->offset_for_ref_frame[i]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* num_ref_frames */
|
|
|
|
WRITE_UE (bs, seq_param->max_num_ref_frames);
|
|
|
|
/* gaps_in_frame_num_value_allowed_flag */
|
|
|
|
WRITE_UINT32 (bs, gaps_in_frame_num_value_allowed_flag, 1);
|
|
|
|
|
|
|
|
/* pic_width_in_mbs_minus1 */
|
|
|
|
WRITE_UE (bs, seq_param->picture_width_in_mbs - 1);
|
|
|
|
/* pic_height_in_map_units_minus1 */
|
|
|
|
WRITE_UE (bs, pic_height_in_map_units - 1);
|
|
|
|
/* frame_mbs_only_flag */
|
|
|
|
WRITE_UINT32 (bs, seq_param->seq_fields.bits.frame_mbs_only_flag, 1);
|
|
|
|
|
|
|
|
if (!seq_param->seq_fields.bits.frame_mbs_only_flag) { //ONLY mbs
|
2014-01-21 15:05:22 +00:00
|
|
|
g_assert (0 && "only progressive frames encoding is supported");
|
2014-01-21 14:28:34 +00:00
|
|
|
WRITE_UINT32 (bs, mb_adaptive_frame_field, 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* direct_8x8_inference_flag */
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
|
|
|
/* frame_cropping_flag */
|
|
|
|
WRITE_UINT32 (bs, seq_param->frame_cropping_flag, 1);
|
|
|
|
|
|
|
|
if (seq_param->frame_cropping_flag) {
|
|
|
|
/* frame_crop_left_offset */
|
|
|
|
WRITE_UE (bs, seq_param->frame_crop_left_offset);
|
|
|
|
/* frame_crop_right_offset */
|
|
|
|
WRITE_UE (bs, seq_param->frame_crop_right_offset);
|
|
|
|
/* frame_crop_top_offset */
|
|
|
|
WRITE_UE (bs, seq_param->frame_crop_top_offset);
|
|
|
|
/* frame_crop_bottom_offset */
|
|
|
|
WRITE_UE (bs, seq_param->frame_crop_bottom_offset);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* vui_parameters_present_flag */
|
|
|
|
WRITE_UINT32 (bs, seq_param->vui_parameters_present_flag, 1);
|
|
|
|
if (seq_param->vui_parameters_present_flag) {
|
|
|
|
/* aspect_ratio_info_present_flag */
|
|
|
|
WRITE_UINT32 (bs,
|
|
|
|
seq_param->vui_fields.bits.aspect_ratio_info_present_flag, 1);
|
|
|
|
if (seq_param->vui_fields.bits.aspect_ratio_info_present_flag) {
|
|
|
|
WRITE_UINT32 (bs, seq_param->aspect_ratio_idc, 8);
|
|
|
|
if (seq_param->aspect_ratio_idc == 0xFF) {
|
|
|
|
WRITE_UINT32 (bs, seq_param->sar_width, 16);
|
|
|
|
WRITE_UINT32 (bs, seq_param->sar_height, 16);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* overscan_info_present_flag */
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
|
|
|
/* video_signal_type_present_flag */
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
|
|
|
/* chroma_loc_info_present_flag */
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
|
|
|
|
|
|
|
/* timing_info_present_flag */
|
|
|
|
WRITE_UINT32 (bs, seq_param->vui_fields.bits.timing_info_present_flag, 1);
|
|
|
|
if (seq_param->vui_fields.bits.timing_info_present_flag) {
|
|
|
|
WRITE_UINT32 (bs, seq_param->num_units_in_tick, 32);
|
|
|
|
WRITE_UINT32 (bs, seq_param->time_scale, 32);
|
|
|
|
WRITE_UINT32 (bs, 1, 1); /* fixed_frame_rate_flag */
|
|
|
|
}
|
|
|
|
|
|
|
|
/* nal_hrd_parameters_present_flag */
|
2014-01-22 16:07:24 +00:00
|
|
|
nal_hrd_parameters_present_flag = seq_param->bits_per_second > 0;
|
2014-01-21 14:28:34 +00:00
|
|
|
WRITE_UINT32 (bs, nal_hrd_parameters_present_flag, 1);
|
|
|
|
if (nal_hrd_parameters_present_flag) {
|
|
|
|
/* hrd_parameters */
|
|
|
|
/* cpb_cnt_minus1 */
|
|
|
|
WRITE_UE (bs, 0);
|
2014-01-22 13:43:24 +00:00
|
|
|
WRITE_UINT32 (bs, SX_BITRATE - 6, 4); /* bit_rate_scale */
|
2014-01-21 18:04:41 +00:00
|
|
|
WRITE_UINT32 (bs, SX_CPB_SIZE - 4, 4); /* cpb_size_scale */
|
2014-01-21 14:28:34 +00:00
|
|
|
|
|
|
|
for (i = 0; i < 1; ++i) {
|
|
|
|
/* bit_rate_value_minus1[0] */
|
2014-01-22 13:43:24 +00:00
|
|
|
WRITE_UE (bs, (seq_param->bits_per_second >> SX_BITRATE) - 1);
|
2014-01-21 14:28:34 +00:00
|
|
|
/* cpb_size_value_minus1[0] */
|
2014-01-21 18:04:41 +00:00
|
|
|
WRITE_UE (bs, (hrd_params->buffer_size >> SX_CPB_SIZE) - 1);
|
2014-01-21 14:28:34 +00:00
|
|
|
/* cbr_flag[0] */
|
|
|
|
WRITE_UINT32 (bs, 1, 1);
|
|
|
|
}
|
|
|
|
/* initial_cpb_removal_delay_length_minus1 */
|
|
|
|
WRITE_UINT32 (bs, 23, 5);
|
|
|
|
/* cpb_removal_delay_length_minus1 */
|
|
|
|
WRITE_UINT32 (bs, 23, 5);
|
|
|
|
/* dpb_output_delay_length_minus1 */
|
|
|
|
WRITE_UINT32 (bs, 23, 5);
|
|
|
|
/* time_offset_length */
|
|
|
|
WRITE_UINT32 (bs, 23, 5);
|
|
|
|
}
|
2014-01-22 16:07:24 +00:00
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
/* vcl_hrd_parameters_present_flag */
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
2014-01-22 16:07:24 +00:00
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
if (nal_hrd_parameters_present_flag
|
|
|
|
|| 0 /*vcl_hrd_parameters_present_flag */ ) {
|
|
|
|
/* low_delay_hrd_flag */
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
|
|
|
}
|
|
|
|
/* pic_struct_present_flag */
|
2015-07-02 18:00:14 +00:00
|
|
|
WRITE_UINT32 (bs, 1, 1);
|
2014-01-21 14:28:34 +00:00
|
|
|
/* bs_restriction_flag */
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
|
|
|
}
|
2014-02-14 07:33:15 +00:00
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write SPS NAL unit");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static gboolean
|
|
|
|
bs_write_sps (GstBitWriter * bs,
|
|
|
|
const VAEncSequenceParameterBufferH264 * seq_param, GstVaapiProfile profile,
|
|
|
|
const VAEncMiscParameterHRD * hrd_params)
|
|
|
|
{
|
|
|
|
if (!bs_write_sps_data (bs, seq_param, profile, hrd_params))
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
/* rbsp_trailing_bits */
|
|
|
|
bs_write_trailing_bits (bs);
|
|
|
|
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
static gboolean
|
|
|
|
bs_write_subset_sps (GstBitWriter * bs,
|
2014-05-06 21:12:39 +00:00
|
|
|
const VAEncSequenceParameterBufferH264 * seq_param, GstVaapiProfile profile,
|
2016-02-02 16:59:57 +00:00
|
|
|
guint num_views, guint16 * view_ids,
|
|
|
|
const VAEncMiscParameterHRD * hrd_params)
|
2014-02-14 07:33:15 +00:00
|
|
|
{
|
|
|
|
guint32 i, j, k;
|
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
if (!bs_write_sps_data (bs, seq_param, profile, hrd_params))
|
2014-02-14 07:33:15 +00:00
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
if (profile == GST_VAAPI_PROFILE_H264_STEREO_HIGH ||
|
|
|
|
profile == GST_VAAPI_PROFILE_H264_MULTIVIEW_HIGH) {
|
2014-05-06 21:12:39 +00:00
|
|
|
guint32 num_views_minus1, num_level_values_signalled_minus1;
|
2014-02-14 07:33:15 +00:00
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
num_views_minus1 = num_views - 1;
|
|
|
|
g_assert (num_views_minus1 < 1024);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
|
|
|
/* bit equal to one */
|
|
|
|
WRITE_UINT32 (bs, 1, 1);
|
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
WRITE_UE (bs, num_views_minus1);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
|
|
|
for (i = 0; i <= num_views_minus1; i++)
|
2015-02-24 15:14:33 +00:00
|
|
|
WRITE_UE (bs, view_ids[i]);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
|
|
|
for (i = 1; i <= num_views_minus1; i++) {
|
2014-06-26 21:49:34 +00:00
|
|
|
guint32 num_anchor_refs_l0 = 0;
|
2016-02-01 13:02:13 +00:00
|
|
|
guint32 num_anchor_refs_l1 = 0;
|
|
|
|
|
2014-02-14 07:33:15 +00:00
|
|
|
WRITE_UE (bs, num_anchor_refs_l0);
|
|
|
|
for (j = 0; j < num_anchor_refs_l0; j++)
|
2014-05-06 21:12:39 +00:00
|
|
|
WRITE_UE (bs, 0);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
|
|
|
WRITE_UE (bs, num_anchor_refs_l1);
|
|
|
|
for (j = 0; j < num_anchor_refs_l1; j++)
|
2014-05-06 21:12:39 +00:00
|
|
|
WRITE_UE (bs, 0);
|
2014-02-14 07:33:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
for (i = 1; i <= num_views_minus1; i++) {
|
2014-06-26 21:49:34 +00:00
|
|
|
guint32 num_non_anchor_refs_l0 = 0;
|
2016-02-01 13:02:13 +00:00
|
|
|
guint32 num_non_anchor_refs_l1 = 0;
|
|
|
|
|
2014-02-14 07:33:15 +00:00
|
|
|
WRITE_UE (bs, num_non_anchor_refs_l0);
|
|
|
|
for (j = 0; j < num_non_anchor_refs_l0; j++)
|
2014-05-06 21:12:39 +00:00
|
|
|
WRITE_UE (bs, 0);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
|
|
|
WRITE_UE (bs, num_non_anchor_refs_l1);
|
|
|
|
for (j = 0; j < num_non_anchor_refs_l1; j++)
|
2014-05-06 21:12:39 +00:00
|
|
|
WRITE_UE (bs, 0);
|
2014-02-14 07:33:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* num level values signalled minus1 */
|
2014-05-06 21:12:39 +00:00
|
|
|
num_level_values_signalled_minus1 = 0;
|
|
|
|
g_assert (num_level_values_signalled_minus1 < 64);
|
2014-06-25 19:05:52 +00:00
|
|
|
WRITE_UE (bs, num_level_values_signalled_minus1);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
for (i = 0; i <= num_level_values_signalled_minus1; i++) {
|
|
|
|
guint16 num_applicable_ops_minus1 = 0;
|
|
|
|
g_assert (num_applicable_ops_minus1 < 1024);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
WRITE_UINT32 (bs, seq_param->level_idc, 8);
|
|
|
|
WRITE_UE (bs, num_applicable_ops_minus1);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
for (j = 0; j <= num_applicable_ops_minus1; j++) {
|
|
|
|
guint8 temporal_id = 0;
|
|
|
|
guint16 num_target_views_minus1 = 1;
|
2014-02-14 07:33:15 +00:00
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
WRITE_UINT32 (bs, temporal_id, 3);
|
|
|
|
WRITE_UE (bs, num_target_views_minus1);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
for (k = 0; k <= num_target_views_minus1; k++)
|
|
|
|
WRITE_UE (bs, k);
|
2014-02-14 07:33:15 +00:00
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
WRITE_UE (bs, num_views_minus1);
|
2014-02-14 07:33:15 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* mvc_vui_parameters_present_flag */
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* additional_extension2_flag */
|
|
|
|
WRITE_UINT32 (bs, 0, 1);
|
2014-01-21 14:28:34 +00:00
|
|
|
|
|
|
|
/* rbsp_trailing_bits */
|
|
|
|
bs_write_trailing_bits (bs);
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
2014-02-14 07:33:15 +00:00
|
|
|
GST_WARNING ("failed to write subset SPS NAL unit");
|
2014-01-21 14:28:34 +00:00
|
|
|
return FALSE;
|
|
|
|
}
|
2014-02-14 07:33:15 +00:00
|
|
|
return FALSE;
|
2014-01-21 14:28:34 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Write a PPS NAL unit */
|
|
|
|
static gboolean
|
|
|
|
bs_write_pps (GstBitWriter * bs,
|
|
|
|
const VAEncPictureParameterBufferH264 * pic_param, GstVaapiProfile profile)
|
|
|
|
{
|
|
|
|
guint32 num_slice_groups_minus1 = 0;
|
|
|
|
guint32 pic_init_qs_minus26 = 0;
|
|
|
|
guint32 redundant_pic_cnt_present_flag = 0;
|
|
|
|
|
|
|
|
/* pic_parameter_set_id */
|
|
|
|
WRITE_UE (bs, pic_param->pic_parameter_set_id);
|
|
|
|
/* seq_parameter_set_id */
|
|
|
|
WRITE_UE (bs, pic_param->seq_parameter_set_id);
|
|
|
|
/* entropy_coding_mode_flag */
|
|
|
|
WRITE_UINT32 (bs, pic_param->pic_fields.bits.entropy_coding_mode_flag, 1);
|
|
|
|
/* pic_order_present_flag */
|
|
|
|
WRITE_UINT32 (bs, pic_param->pic_fields.bits.pic_order_present_flag, 1);
|
|
|
|
/* slice_groups-1 */
|
|
|
|
WRITE_UE (bs, num_slice_groups_minus1);
|
|
|
|
|
|
|
|
if (num_slice_groups_minus1 > 0) {
|
2014-01-21 15:05:22 +00:00
|
|
|
/*FIXME*/ g_assert (0 && "unsupported arbitrary slice ordering (ASO)");
|
2014-01-21 14:28:34 +00:00
|
|
|
}
|
|
|
|
WRITE_UE (bs, pic_param->num_ref_idx_l0_active_minus1);
|
|
|
|
WRITE_UE (bs, pic_param->num_ref_idx_l1_active_minus1);
|
|
|
|
WRITE_UINT32 (bs, pic_param->pic_fields.bits.weighted_pred_flag, 1);
|
|
|
|
WRITE_UINT32 (bs, pic_param->pic_fields.bits.weighted_bipred_idc, 2);
|
|
|
|
/* pic_init_qp_minus26 */
|
|
|
|
WRITE_SE (bs, pic_param->pic_init_qp - 26);
|
|
|
|
/* pic_init_qs_minus26 */
|
|
|
|
WRITE_SE (bs, pic_init_qs_minus26);
|
|
|
|
/* chroma_qp_index_offset */
|
|
|
|
WRITE_SE (bs, pic_param->chroma_qp_index_offset);
|
|
|
|
|
|
|
|
WRITE_UINT32 (bs,
|
|
|
|
pic_param->pic_fields.bits.deblocking_filter_control_present_flag, 1);
|
|
|
|
WRITE_UINT32 (bs, pic_param->pic_fields.bits.constrained_intra_pred_flag, 1);
|
|
|
|
WRITE_UINT32 (bs, redundant_pic_cnt_present_flag, 1);
|
|
|
|
|
|
|
|
/* more_rbsp_data */
|
2016-07-15 11:41:27 +00:00
|
|
|
if (profile == GST_VAAPI_PROFILE_H264_HIGH
|
|
|
|
|| profile == GST_VAAPI_PROFILE_H264_MULTIVIEW_HIGH
|
|
|
|
|| profile == GST_VAAPI_PROFILE_H264_STEREO_HIGH) {
|
2014-01-21 14:28:34 +00:00
|
|
|
WRITE_UINT32 (bs, pic_param->pic_fields.bits.transform_8x8_mode_flag, 1);
|
|
|
|
WRITE_UINT32 (bs,
|
|
|
|
pic_param->pic_fields.bits.pic_scaling_matrix_present_flag, 1);
|
|
|
|
if (pic_param->pic_fields.bits.pic_scaling_matrix_present_flag) {
|
|
|
|
g_assert (0 && "unsupported scaling lists");
|
|
|
|
/* FIXME */
|
|
|
|
/*
|
|
|
|
for (i = 0; i <
|
|
|
|
(6+(-( (chroma_format_idc ! = 3) ? 2 : 6) * -pic_param->pic_fields.bits.transform_8x8_mode_flag));
|
|
|
|
i++) {
|
|
|
|
gst_bit_writer_put_bits_uint8(bs, pic_param->pic_fields.bits.pic_scaling_list_present_flag, 1);
|
|
|
|
}
|
|
|
|
*/
|
|
|
|
}
|
|
|
|
WRITE_SE (bs, pic_param->second_chroma_qp_index_offset);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* rbsp_trailing_bits */
|
|
|
|
bs_write_trailing_bits (bs);
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write PPS NAL unit");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-21 14:23:01 +00:00
|
|
|
/* ------------------------------------------------------------------------- */
|
|
|
|
/* --- H.264 Encoder --- */
|
|
|
|
/* ------------------------------------------------------------------------- */
|
|
|
|
|
|
|
|
#define GST_VAAPI_ENCODER_H264_CAST(encoder) \
|
|
|
|
((GstVaapiEncoderH264 *)(encoder))
|
|
|
|
|
|
|
|
struct _GstVaapiEncoderH264
|
|
|
|
{
|
|
|
|
GstVaapiEncoder parent_instance;
|
|
|
|
|
|
|
|
GstVaapiProfile profile;
|
|
|
|
GstVaapiLevelH264 level;
|
2016-05-11 09:05:36 +00:00
|
|
|
GstVaapiEntrypoint entrypoint;
|
2014-01-21 14:23:01 +00:00
|
|
|
guint8 profile_idc;
|
|
|
|
guint8 max_profile_idc;
|
|
|
|
guint8 hw_max_profile_idc;
|
|
|
|
guint8 level_idc;
|
|
|
|
guint32 idr_period;
|
|
|
|
guint32 init_qp;
|
|
|
|
guint32 min_qp;
|
|
|
|
guint32 num_slices;
|
|
|
|
guint32 num_bframes;
|
|
|
|
guint32 mb_width;
|
|
|
|
guint32 mb_height;
|
|
|
|
gboolean use_cabac;
|
|
|
|
gboolean use_dct8x8;
|
|
|
|
GstClockTime cts_offset;
|
2014-06-27 16:43:27 +00:00
|
|
|
gboolean config_changed;
|
2014-01-21 14:23:01 +00:00
|
|
|
|
|
|
|
/* frame, poc */
|
|
|
|
guint32 max_frame_num;
|
|
|
|
guint32 log2_max_frame_num;
|
|
|
|
guint32 max_pic_order_cnt;
|
|
|
|
guint32 log2_max_pic_order_cnt;
|
|
|
|
guint32 idr_num;
|
2014-05-06 21:09:45 +00:00
|
|
|
guint8 pic_order_cnt_type;
|
|
|
|
guint8 delta_pic_order_always_zero_flag;
|
2014-01-21 14:23:01 +00:00
|
|
|
|
|
|
|
GstBuffer *sps_data;
|
2014-05-06 21:09:19 +00:00
|
|
|
GstBuffer *subset_sps_data;
|
2014-01-21 14:23:01 +00:00
|
|
|
GstBuffer *pps_data;
|
2014-01-22 13:43:24 +00:00
|
|
|
|
|
|
|
guint bitrate_bits; // bitrate (bits)
|
2014-01-21 18:04:41 +00:00
|
|
|
guint cpb_length; // length of CPB buffer (ms)
|
|
|
|
guint cpb_length_bits; // length of CPB buffer (bits)
|
2014-02-17 03:10:26 +00:00
|
|
|
|
|
|
|
/* MVC */
|
|
|
|
gboolean is_mvc;
|
2016-02-02 16:59:57 +00:00
|
|
|
guint32 view_idx; /* View Order Index (VOIdx) */
|
2014-02-17 03:10:26 +00:00
|
|
|
guint32 num_views;
|
2015-02-24 15:14:33 +00:00
|
|
|
guint16 view_ids[MAX_NUM_VIEWS];
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewRefPool ref_pools[MAX_NUM_VIEWS];
|
|
|
|
GstVaapiH264ViewReorderPool reorder_pools[MAX_NUM_VIEWS];
|
2014-01-21 14:23:01 +00:00
|
|
|
};
|
|
|
|
|
2015-07-02 18:00:14 +00:00
|
|
|
/* Write a SEI buffering period payload */
|
|
|
|
static gboolean
|
|
|
|
bs_write_sei_buf_period (GstBitWriter * bs,
|
|
|
|
GstVaapiEncoderH264 * encoder, GstVaapiEncPicture * picture)
|
|
|
|
{
|
|
|
|
guint initial_cpb_removal_delay = 0;
|
|
|
|
guint initial_cpb_removal_delay_offset = 0;
|
|
|
|
guint8 initial_cpb_removal_delay_length = 24;
|
|
|
|
|
|
|
|
/* sequence_parameter_set_id */
|
|
|
|
WRITE_UE (bs, encoder->view_idx);
|
|
|
|
/* NalHrdBpPresentFlag == TRUE */
|
|
|
|
/* cpb_cnt_minus1 == 0 */
|
|
|
|
|
|
|
|
/* decoding should start when the CPB fullness reaches half of cpb size
|
|
|
|
* initial_cpb_remvoal_delay = (((cpb_length / 2) * 90000) / 1000) */
|
|
|
|
initial_cpb_removal_delay = encoder->cpb_length * 45;
|
|
|
|
|
|
|
|
/* initial_cpb_remvoal_dealy */
|
2016-02-02 16:59:57 +00:00
|
|
|
WRITE_UINT32 (bs, initial_cpb_removal_delay,
|
|
|
|
initial_cpb_removal_delay_length);
|
2015-07-02 18:00:14 +00:00
|
|
|
|
|
|
|
/* initial_cpb_removal_delay_offset */
|
|
|
|
WRITE_UINT32 (bs, initial_cpb_removal_delay_offset,
|
2016-02-02 16:59:57 +00:00
|
|
|
initial_cpb_removal_delay_length);
|
2015-07-02 18:00:14 +00:00
|
|
|
|
|
|
|
/* VclHrdBpPresentFlag == FALSE */
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write Buffering Period SEI message");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Write a SEI picture timing payload */
|
|
|
|
static gboolean
|
|
|
|
bs_write_sei_pic_timing (GstBitWriter * bs,
|
|
|
|
GstVaapiEncoderH264 * encoder, GstVaapiEncPicture * picture)
|
|
|
|
{
|
|
|
|
GstVaapiH264ViewReorderPool *reorder_pool = NULL;
|
|
|
|
guint cpb_removal_delay;
|
|
|
|
guint dpb_output_delay;
|
|
|
|
guint8 cpb_removal_delay_length = 24;
|
|
|
|
guint8 dpb_output_delay_length = 24;
|
|
|
|
guint pic_struct = 0;
|
|
|
|
guint clock_timestamp_flag = 0;
|
|
|
|
|
|
|
|
reorder_pool = &encoder->reorder_pools[encoder->view_idx];
|
|
|
|
if (GST_VAAPI_ENC_PICTURE_IS_IDR (picture))
|
|
|
|
reorder_pool->frame_count = 0;
|
|
|
|
else
|
|
|
|
reorder_pool->frame_count++;
|
|
|
|
|
|
|
|
/* clock-tick = no_units_in_tick/time_scale (C-1)
|
|
|
|
* time_scale = FPS_N * 2 (E.2.1)
|
|
|
|
* num_units_in_tick = FPS_D (E.2.1)
|
|
|
|
* frame_duration = clock-tick * 2
|
|
|
|
* so removal time for one frame is 2 clock-ticks.
|
|
|
|
* but adding a tolerance of one frame duration,
|
|
|
|
* which is 2 more clock-ticks */
|
|
|
|
cpb_removal_delay = (reorder_pool->frame_count * 2 + 2);
|
|
|
|
|
|
|
|
if (picture->type == GST_VAAPI_PICTURE_TYPE_B)
|
|
|
|
dpb_output_delay = 0;
|
|
|
|
else
|
|
|
|
dpb_output_delay = picture->poc - reorder_pool->frame_count * 2;
|
|
|
|
|
|
|
|
/* CpbDpbDelaysPresentFlag == 1 */
|
|
|
|
WRITE_UINT32 (bs, cpb_removal_delay, cpb_removal_delay_length);
|
|
|
|
WRITE_UINT32 (bs, dpb_output_delay, dpb_output_delay_length);
|
|
|
|
|
|
|
|
/* pic_struct_present_flag == 1 */
|
|
|
|
/* pic_struct */
|
|
|
|
WRITE_UINT32 (bs, pic_struct, 4);
|
|
|
|
/* clock_timestamp_flag */
|
|
|
|
WRITE_UINT32 (bs, clock_timestamp_flag, 1);
|
|
|
|
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write Picture Timing SEI message");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-05-06 21:09:45 +00:00
|
|
|
/* Write a Slice NAL unit */
|
|
|
|
static gboolean
|
|
|
|
bs_write_slice (GstBitWriter * bs,
|
|
|
|
const VAEncSliceParameterBufferH264 * slice_param,
|
|
|
|
GstVaapiEncoderH264 * encoder, GstVaapiEncPicture * picture)
|
|
|
|
{
|
|
|
|
const VAEncPictureParameterBufferH264 *const pic_param = picture->param;
|
|
|
|
guint32 field_pic_flag = 0;
|
|
|
|
guint32 ref_pic_list_modification_flag_l0 = 0;
|
|
|
|
guint32 ref_pic_list_modification_flag_l1 = 0;
|
|
|
|
guint32 no_output_of_prior_pics_flag = 0;
|
|
|
|
guint32 long_term_reference_flag = 0;
|
|
|
|
guint32 adaptive_ref_pic_marking_mode_flag = 0;
|
|
|
|
|
|
|
|
/* first_mb_in_slice */
|
|
|
|
WRITE_UE (bs, slice_param->macroblock_address);
|
|
|
|
/* slice_type */
|
|
|
|
WRITE_UE (bs, slice_param->slice_type);
|
|
|
|
/* pic_parameter_set_id */
|
|
|
|
WRITE_UE (bs, slice_param->pic_parameter_set_id);
|
|
|
|
/* frame_num */
|
|
|
|
WRITE_UINT32 (bs, picture->frame_num, encoder->log2_max_frame_num);
|
|
|
|
|
|
|
|
/* XXX: only frames (i.e. non-interlaced) are supported for now */
|
|
|
|
/* frame_mbs_only_flag == 0 */
|
|
|
|
|
|
|
|
/* idr_pic_id */
|
|
|
|
if (GST_VAAPI_ENC_PICTURE_IS_IDR (picture))
|
|
|
|
WRITE_UE (bs, slice_param->idr_pic_id);
|
|
|
|
|
|
|
|
/* XXX: only POC type 0 is supported */
|
|
|
|
if (!encoder->pic_order_cnt_type) {
|
|
|
|
WRITE_UINT32 (bs, slice_param->pic_order_cnt_lsb,
|
|
|
|
encoder->log2_max_pic_order_cnt);
|
|
|
|
/* bottom_field_pic_order_in_frame_present_flag is FALSE */
|
|
|
|
if (pic_param->pic_fields.bits.pic_order_present_flag && !field_pic_flag)
|
|
|
|
WRITE_SE (bs, slice_param->delta_pic_order_cnt_bottom);
|
|
|
|
} else if (encoder->pic_order_cnt_type == 1 &&
|
|
|
|
!encoder->delta_pic_order_always_zero_flag) {
|
|
|
|
WRITE_SE (bs, slice_param->delta_pic_order_cnt[0]);
|
|
|
|
if (pic_param->pic_fields.bits.pic_order_present_flag && !field_pic_flag)
|
|
|
|
WRITE_SE (bs, slice_param->delta_pic_order_cnt[1]);
|
|
|
|
}
|
|
|
|
/* redundant_pic_cnt_present_flag is FALSE, no redundant coded pictures */
|
|
|
|
|
|
|
|
/* only works for B-frames */
|
|
|
|
if (slice_param->slice_type == 1)
|
|
|
|
WRITE_UINT32 (bs, slice_param->direct_spatial_mv_pred_flag, 1);
|
|
|
|
|
|
|
|
/* not supporting SP slices */
|
|
|
|
if (slice_param->slice_type == 0 || slice_param->slice_type == 1) {
|
|
|
|
WRITE_UINT32 (bs, slice_param->num_ref_idx_active_override_flag, 1);
|
|
|
|
if (slice_param->num_ref_idx_active_override_flag) {
|
|
|
|
WRITE_UE (bs, slice_param->num_ref_idx_l0_active_minus1);
|
|
|
|
if (slice_param->slice_type == 1)
|
|
|
|
WRITE_UE (bs, slice_param->num_ref_idx_l1_active_minus1);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
/* XXX: not supporting custom reference picture list modifications */
|
|
|
|
if ((slice_param->slice_type != 2) && (slice_param->slice_type != 4))
|
|
|
|
WRITE_UINT32 (bs, ref_pic_list_modification_flag_l0, 1);
|
|
|
|
if (slice_param->slice_type == 1)
|
|
|
|
WRITE_UINT32 (bs, ref_pic_list_modification_flag_l1, 1);
|
|
|
|
|
|
|
|
/* we have: weighted_pred_flag == FALSE and */
|
|
|
|
/* : weighted_bipred_idc == FALSE */
|
|
|
|
if ((pic_param->pic_fields.bits.weighted_pred_flag &&
|
|
|
|
(slice_param->slice_type == 0)) ||
|
|
|
|
((pic_param->pic_fields.bits.weighted_bipred_idc == 1) &&
|
|
|
|
(slice_param->slice_type == 1))) {
|
|
|
|
/* XXXX: add pred_weight_table() */
|
|
|
|
}
|
|
|
|
|
|
|
|
/* dec_ref_pic_marking() */
|
|
|
|
if (slice_param->slice_type == 0 || slice_param->slice_type == 2) {
|
|
|
|
if (GST_VAAPI_ENC_PICTURE_IS_IDR (picture)) {
|
|
|
|
/* no_output_of_prior_pics_flag = 0 */
|
|
|
|
WRITE_UINT32 (bs, no_output_of_prior_pics_flag, 1);
|
|
|
|
/* long_term_reference_flag = 0 */
|
|
|
|
WRITE_UINT32 (bs, long_term_reference_flag, 1);
|
|
|
|
} else {
|
|
|
|
/* only sliding_window reference picture marking mode is supported */
|
|
|
|
/* adpative_ref_pic_marking_mode_flag = 0 */
|
|
|
|
WRITE_UINT32 (bs, adaptive_ref_pic_marking_mode_flag, 1);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* cabac_init_idc */
|
|
|
|
if (pic_param->pic_fields.bits.entropy_coding_mode_flag &&
|
|
|
|
slice_param->slice_type != 2)
|
|
|
|
WRITE_UE (bs, slice_param->cabac_init_idc);
|
|
|
|
/*slice_qp_delta */
|
|
|
|
WRITE_SE (bs, slice_param->slice_qp_delta);
|
|
|
|
|
|
|
|
/* XXX: only supporting I, P and B type slices */
|
|
|
|
/* no sp_for_switch_flag and no slice_qs_delta */
|
|
|
|
|
|
|
|
if (pic_param->pic_fields.bits.deblocking_filter_control_present_flag) {
|
|
|
|
/* disable_deblocking_filter_idc */
|
|
|
|
WRITE_UE (bs, slice_param->disable_deblocking_filter_idc);
|
|
|
|
if (slice_param->disable_deblocking_filter_idc != 1) {
|
|
|
|
WRITE_SE (bs, slice_param->slice_alpha_c0_offset_div2);
|
|
|
|
WRITE_SE (bs, slice_param->slice_beta_offset_div2);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* XXX: unsupported arbitrary slice ordering (ASO) */
|
|
|
|
/* num_slic_groups_minus1 should be zero */
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write Slice NAL unit");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
static inline void
|
|
|
|
_check_sps_pps_status (GstVaapiEncoderH264 * encoder,
|
|
|
|
const guint8 * nal, guint32 size)
|
|
|
|
{
|
|
|
|
guint8 nal_type;
|
2016-02-03 15:45:18 +00:00
|
|
|
G_GNUC_UNUSED gsize ret; /* FIXME */
|
2014-05-06 21:09:19 +00:00
|
|
|
gboolean has_subset_sps;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
g_assert (size);
|
|
|
|
|
2014-05-06 21:09:19 +00:00
|
|
|
has_subset_sps = !encoder->is_mvc || (encoder->subset_sps_data != NULL);
|
|
|
|
if (encoder->sps_data && encoder->pps_data && has_subset_sps)
|
2013-07-29 07:46:11 +00:00
|
|
|
return;
|
|
|
|
|
|
|
|
nal_type = nal[0] & 0x1F;
|
|
|
|
switch (nal_type) {
|
2014-06-03 12:30:39 +00:00
|
|
|
case GST_H264_NAL_SPS:
|
2013-07-29 07:46:11 +00:00
|
|
|
encoder->sps_data = gst_buffer_new_allocate (NULL, size, NULL);
|
|
|
|
ret = gst_buffer_fill (encoder->sps_data, 0, nal, size);
|
|
|
|
g_assert (ret == size);
|
|
|
|
break;
|
2014-06-03 12:30:39 +00:00
|
|
|
case GST_H264_NAL_SUBSET_SPS:
|
2014-05-06 21:09:19 +00:00
|
|
|
encoder->subset_sps_data = gst_buffer_new_allocate (NULL, size, NULL);
|
|
|
|
ret = gst_buffer_fill (encoder->subset_sps_data, 0, nal, size);
|
|
|
|
g_assert (ret == size);
|
|
|
|
break;
|
2014-06-03 12:30:39 +00:00
|
|
|
case GST_H264_NAL_PPS:
|
2013-07-29 07:46:11 +00:00
|
|
|
encoder->pps_data = gst_buffer_new_allocate (NULL, size, NULL);
|
|
|
|
ret = gst_buffer_fill (encoder->pps_data, 0, nal, size);
|
|
|
|
g_assert (ret == size);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-12 21:24:04 +00:00
|
|
|
/* Determines the largest supported profile by the underlying hardware */
|
|
|
|
static gboolean
|
|
|
|
ensure_hw_profile_limits (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
|
|
|
GstVaapiDisplay *const display = GST_VAAPI_ENCODER_DISPLAY (encoder);
|
|
|
|
GArray *profiles;
|
|
|
|
guint i, profile_idc, max_profile_idc;
|
|
|
|
|
|
|
|
if (encoder->hw_max_profile_idc)
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
profiles = gst_vaapi_display_get_encode_profiles (display);
|
|
|
|
if (!profiles)
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
max_profile_idc = 0;
|
|
|
|
for (i = 0; i < profiles->len; i++) {
|
|
|
|
const GstVaapiProfile profile =
|
|
|
|
g_array_index (profiles, GstVaapiProfile, i);
|
|
|
|
profile_idc = gst_vaapi_utils_h264_get_profile_idc (profile);
|
|
|
|
if (!profile_idc)
|
|
|
|
continue;
|
|
|
|
if (max_profile_idc < profile_idc)
|
|
|
|
max_profile_idc = profile_idc;
|
|
|
|
}
|
|
|
|
g_array_unref (profiles);
|
|
|
|
|
|
|
|
encoder->hw_max_profile_idc = max_profile_idc;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-01-10 16:02:44 +00:00
|
|
|
/* Derives the profile supported by the underlying hardware */
|
|
|
|
static gboolean
|
|
|
|
ensure_hw_profile (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
|
|
|
GstVaapiDisplay *const display = GST_VAAPI_ENCODER_DISPLAY (encoder);
|
2016-05-13 08:44:57 +00:00
|
|
|
GstVaapiEntrypoint entrypoint = encoder->entrypoint;
|
2014-01-10 16:02:44 +00:00
|
|
|
GstVaapiProfile profile, profiles[4];
|
|
|
|
guint i, num_profiles = 0;
|
|
|
|
|
|
|
|
profiles[num_profiles++] = encoder->profile;
|
|
|
|
switch (encoder->profile) {
|
|
|
|
case GST_VAAPI_PROFILE_H264_CONSTRAINED_BASELINE:
|
|
|
|
profiles[num_profiles++] = GST_VAAPI_PROFILE_H264_BASELINE;
|
|
|
|
profiles[num_profiles++] = GST_VAAPI_PROFILE_H264_MAIN;
|
|
|
|
// fall-through
|
|
|
|
case GST_VAAPI_PROFILE_H264_MAIN:
|
|
|
|
profiles[num_profiles++] = GST_VAAPI_PROFILE_H264_HIGH;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
profile = GST_VAAPI_PROFILE_UNKNOWN;
|
|
|
|
for (i = 0; i < num_profiles; i++) {
|
|
|
|
if (gst_vaapi_display_has_encoder (display, profiles[i], entrypoint)) {
|
|
|
|
profile = profiles[i];
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (profile == GST_VAAPI_PROFILE_UNKNOWN)
|
|
|
|
goto error_unsupported_profile;
|
|
|
|
|
|
|
|
GST_VAAPI_ENCODER_CAST (encoder)->profile = profile;
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
error_unsupported_profile:
|
|
|
|
{
|
|
|
|
GST_ERROR ("unsupported HW profile (0x%08x)", encoder->profile);
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-12 21:24:04 +00:00
|
|
|
/* Check target decoder constraints */
|
|
|
|
static gboolean
|
|
|
|
ensure_profile_limits (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
|
|
|
GstVaapiProfile profile;
|
|
|
|
|
|
|
|
if (!encoder->max_profile_idc
|
|
|
|
|| encoder->profile_idc <= encoder->max_profile_idc)
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
GST_WARNING ("lowering coding tools to meet target decoder constraints");
|
|
|
|
|
2014-11-25 10:41:49 +00:00
|
|
|
profile = GST_VAAPI_PROFILE_UNKNOWN;
|
|
|
|
|
2014-01-12 21:24:04 +00:00
|
|
|
/* Try Main profile coding tools */
|
|
|
|
if (encoder->max_profile_idc < 100) {
|
|
|
|
encoder->use_dct8x8 = FALSE;
|
|
|
|
profile = GST_VAAPI_PROFILE_H264_MAIN;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Try Constrained Baseline profile coding tools */
|
|
|
|
if (encoder->max_profile_idc < 77) {
|
|
|
|
encoder->num_bframes = 0;
|
|
|
|
encoder->use_cabac = FALSE;
|
|
|
|
profile = GST_VAAPI_PROFILE_H264_CONSTRAINED_BASELINE;
|
|
|
|
}
|
|
|
|
|
2014-11-25 10:41:49 +00:00
|
|
|
if (profile) {
|
|
|
|
encoder->profile = profile;
|
|
|
|
encoder->profile_idc = encoder->max_profile_idc;
|
|
|
|
}
|
2014-01-12 21:24:04 +00:00
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-01-10 16:02:44 +00:00
|
|
|
/* Derives the minimum profile from the active coding tools */
|
|
|
|
static gboolean
|
|
|
|
ensure_profile (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
|
|
|
GstVaapiProfile profile;
|
|
|
|
|
2014-01-10 17:18:25 +00:00
|
|
|
/* Always start from "constrained-baseline" profile for maximum
|
|
|
|
compatibility */
|
|
|
|
profile = GST_VAAPI_PROFILE_H264_CONSTRAINED_BASELINE;
|
|
|
|
|
|
|
|
/* Main profile coding tools */
|
2013-12-13 09:36:08 +00:00
|
|
|
if (encoder->num_bframes > 0 || encoder->use_cabac)
|
2014-01-10 17:18:25 +00:00
|
|
|
profile = GST_VAAPI_PROFILE_H264_MAIN;
|
2014-01-10 16:02:44 +00:00
|
|
|
|
2013-12-13 09:36:08 +00:00
|
|
|
/* High profile coding tools */
|
|
|
|
if (encoder->use_dct8x8)
|
|
|
|
profile = GST_VAAPI_PROFILE_H264_HIGH;
|
|
|
|
|
2014-02-17 07:51:43 +00:00
|
|
|
/* MVC profiles coding tools */
|
|
|
|
if (encoder->num_views == 2)
|
|
|
|
profile = GST_VAAPI_PROFILE_H264_STEREO_HIGH;
|
|
|
|
else if (encoder->num_views > 2)
|
|
|
|
profile = GST_VAAPI_PROFILE_H264_MULTIVIEW_HIGH;
|
|
|
|
|
2014-01-10 16:02:44 +00:00
|
|
|
encoder->profile = profile;
|
2014-01-10 17:18:25 +00:00
|
|
|
encoder->profile_idc = gst_vaapi_utils_h264_get_profile_idc (profile);
|
2014-01-10 16:02:44 +00:00
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Derives the level from the currently set limits */
|
2014-01-10 17:18:25 +00:00
|
|
|
static gboolean
|
|
|
|
ensure_level (GstVaapiEncoderH264 * encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-01-22 10:25:13 +00:00
|
|
|
const guint cpb_factor = h264_get_cpb_nal_factor (encoder->profile);
|
2014-01-10 17:18:25 +00:00
|
|
|
const GstVaapiH264LevelLimits *limits_table;
|
|
|
|
guint i, num_limits, PicSizeMbs, MaxDpbMbs, MaxMBPS;
|
|
|
|
|
|
|
|
PicSizeMbs = encoder->mb_width * encoder->mb_height;
|
|
|
|
MaxDpbMbs = PicSizeMbs * ((encoder->num_bframes) ? 2 : 1);
|
|
|
|
MaxMBPS = gst_util_uint64_scale_int_ceil (PicSizeMbs,
|
|
|
|
GST_VAAPI_ENCODER_FPS_N (encoder), GST_VAAPI_ENCODER_FPS_D (encoder));
|
|
|
|
|
|
|
|
limits_table = gst_vaapi_utils_h264_get_level_limits_table (&num_limits);
|
|
|
|
for (i = 0; i < num_limits; i++) {
|
|
|
|
const GstVaapiH264LevelLimits *const limits = &limits_table[i];
|
|
|
|
if (PicSizeMbs <= limits->MaxFS &&
|
|
|
|
MaxDpbMbs <= limits->MaxDpbMbs &&
|
2014-01-22 13:43:24 +00:00
|
|
|
MaxMBPS <= limits->MaxMBPS && (!encoder->bitrate_bits
|
2014-01-21 18:04:41 +00:00
|
|
|
|| encoder->bitrate_bits <= (limits->MaxBR * cpb_factor)) &&
|
|
|
|
(!encoder->cpb_length_bits ||
|
|
|
|
encoder->cpb_length_bits <= (limits->MaxCPB * cpb_factor)))
|
2014-01-10 17:18:25 +00:00
|
|
|
break;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
2014-01-10 17:18:25 +00:00
|
|
|
if (i == num_limits)
|
|
|
|
goto error_unsupported_level;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 17:18:25 +00:00
|
|
|
encoder->level = limits_table[i].level;
|
|
|
|
encoder->level_idc = limits_table[i].level_idc;
|
|
|
|
return TRUE;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 17:18:25 +00:00
|
|
|
/* ERRORS */
|
|
|
|
error_unsupported_level:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to find a suitable level matching codec config");
|
|
|
|
return FALSE;
|
|
|
|
}
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-13 10:11:25 +00:00
|
|
|
/* Enable "high-compression" tuning options */
|
|
|
|
static gboolean
|
|
|
|
ensure_tuning_high_compression (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
|
|
|
guint8 profile_idc;
|
|
|
|
|
|
|
|
if (!ensure_hw_profile_limits (encoder))
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
profile_idc = encoder->hw_max_profile_idc;
|
|
|
|
if (encoder->max_profile_idc && encoder->max_profile_idc < profile_idc)
|
|
|
|
profile_idc = encoder->max_profile_idc;
|
|
|
|
|
|
|
|
/* Tuning options to enable Main profile */
|
2014-04-01 11:23:56 +00:00
|
|
|
if (profile_idc >= 77 && profile_idc != 88) {
|
2014-01-13 10:11:25 +00:00
|
|
|
encoder->use_cabac = TRUE;
|
|
|
|
if (!encoder->num_bframes)
|
|
|
|
encoder->num_bframes = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Tuning options to enable High profile */
|
|
|
|
if (profile_idc >= 100) {
|
|
|
|
encoder->use_dct8x8 = TRUE;
|
|
|
|
}
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Ensure tuning options */
|
|
|
|
static gboolean
|
|
|
|
ensure_tuning (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
|
|
|
gboolean success;
|
|
|
|
|
|
|
|
switch (GST_VAAPI_ENCODER_TUNE (encoder)) {
|
|
|
|
case GST_VAAPI_ENCODER_TUNE_HIGH_COMPRESSION:
|
|
|
|
success = ensure_tuning_high_compression (encoder);
|
|
|
|
break;
|
2016-05-13 08:44:57 +00:00
|
|
|
case GST_VAAPI_ENCODER_TUNE_LOW_POWER:
|
|
|
|
/* Set low-power encode entry point. If hardware doesn't have
|
|
|
|
* support, it will fail in ensure_hw_profile() in later stage.
|
|
|
|
* So not duplicating the profile/entrypont query mechanism
|
|
|
|
* here as a part of optimization */
|
|
|
|
encoder->entrypoint = GST_VAAPI_ENTRYPOINT_SLICE_ENCODE_LP;
|
|
|
|
success = TRUE;
|
|
|
|
break;
|
2014-01-13 10:11:25 +00:00
|
|
|
default:
|
|
|
|
success = TRUE;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return success;
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Handle new GOP starts */
|
|
|
|
static void
|
|
|
|
reset_gop_start (GstVaapiEncoderH264 * encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewReorderPool *const reorder_pool =
|
|
|
|
&encoder->reorder_pools[encoder->view_idx];
|
|
|
|
|
|
|
|
reorder_pool->frame_index = 1;
|
|
|
|
reorder_pool->cur_frame_num = 0;
|
|
|
|
reorder_pool->cur_present_index = 0;
|
2013-07-29 07:46:11 +00:00
|
|
|
++encoder->idr_num;
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Marks the supplied picture as a B-frame */
|
2013-07-29 07:46:11 +00:00
|
|
|
static void
|
2014-01-21 15:05:22 +00:00
|
|
|
set_b_frame (GstVaapiEncPicture * pic, GstVaapiEncoderH264 * encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewReorderPool *const reorder_pool =
|
|
|
|
&encoder->reorder_pools[encoder->view_idx];
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
g_assert (pic && encoder);
|
|
|
|
g_return_if_fail (pic->type == GST_VAAPI_PICTURE_TYPE_NONE);
|
|
|
|
pic->type = GST_VAAPI_PICTURE_TYPE_B;
|
2014-02-17 03:10:26 +00:00
|
|
|
pic->frame_num = (reorder_pool->cur_frame_num % encoder->max_frame_num);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Marks the supplied picture as a P-frame */
|
|
|
|
static void
|
|
|
|
set_p_frame (GstVaapiEncPicture * pic, GstVaapiEncoderH264 * encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewReorderPool *const reorder_pool =
|
|
|
|
&encoder->reorder_pools[encoder->view_idx];
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
g_return_if_fail (pic->type == GST_VAAPI_PICTURE_TYPE_NONE);
|
|
|
|
pic->type = GST_VAAPI_PICTURE_TYPE_P;
|
2014-02-17 03:10:26 +00:00
|
|
|
pic->frame_num = (reorder_pool->cur_frame_num % encoder->max_frame_num);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Marks the supplied picture as an I-frame */
|
|
|
|
static void
|
|
|
|
set_i_frame (GstVaapiEncPicture * pic, GstVaapiEncoderH264 * encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewReorderPool *const reorder_pool =
|
|
|
|
&encoder->reorder_pools[encoder->view_idx];
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
g_return_if_fail (pic->type == GST_VAAPI_PICTURE_TYPE_NONE);
|
|
|
|
pic->type = GST_VAAPI_PICTURE_TYPE_I;
|
2014-02-17 03:10:26 +00:00
|
|
|
pic->frame_num = (reorder_pool->cur_frame_num % encoder->max_frame_num);
|
2014-01-14 11:01:11 +00:00
|
|
|
|
|
|
|
g_assert (pic->frame);
|
|
|
|
GST_VIDEO_CODEC_FRAME_SET_SYNC_POINT (pic->frame);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Marks the supplied picture as an IDR frame */
|
|
|
|
static void
|
|
|
|
set_idr_frame (GstVaapiEncPicture * pic, GstVaapiEncoderH264 * encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
|
|
|
g_return_if_fail (pic->type == GST_VAAPI_PICTURE_TYPE_NONE);
|
|
|
|
pic->type = GST_VAAPI_PICTURE_TYPE_I;
|
|
|
|
pic->frame_num = 0;
|
|
|
|
pic->poc = 0;
|
|
|
|
GST_VAAPI_ENC_PICTURE_FLAG_SET (pic, GST_VAAPI_ENC_PICTURE_FLAG_IDR);
|
|
|
|
|
2014-01-14 11:01:11 +00:00
|
|
|
g_assert (pic->frame);
|
|
|
|
GST_VIDEO_CODEC_FRAME_SET_SYNC_POINT (pic->frame);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Marks the supplied picture a a key-frame */
|
|
|
|
static void
|
|
|
|
set_key_frame (GstVaapiEncPicture * picture,
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncoderH264 * encoder, gboolean is_idr)
|
|
|
|
{
|
|
|
|
if (is_idr) {
|
2014-01-21 15:05:22 +00:00
|
|
|
reset_gop_start (encoder);
|
|
|
|
set_idr_frame (picture, encoder);
|
2013-07-29 07:46:11 +00:00
|
|
|
} else
|
2014-01-21 15:05:22 +00:00
|
|
|
set_i_frame (picture, encoder);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Fills in VA HRD parameters */
|
|
|
|
static void
|
|
|
|
fill_hrd_params (GstVaapiEncoderH264 * encoder, VAEncMiscParameterHRD * hrd)
|
|
|
|
{
|
2014-01-22 13:43:24 +00:00
|
|
|
if (encoder->bitrate_bits > 0) {
|
2014-01-21 18:04:41 +00:00
|
|
|
hrd->buffer_size = encoder->cpb_length_bits;
|
2014-01-21 15:05:22 +00:00
|
|
|
hrd->initial_buffer_fullness = hrd->buffer_size / 2;
|
|
|
|
} else {
|
|
|
|
hrd->buffer_size = 0;
|
|
|
|
hrd->initial_buffer_fullness = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Adds the supplied sequence header (SPS) to the list of packed
|
|
|
|
headers to pass down as-is to the encoder */
|
2013-07-29 07:46:11 +00:00
|
|
|
static gboolean
|
2014-01-21 15:05:22 +00:00
|
|
|
add_packed_sequence_header (GstVaapiEncoderH264 * encoder,
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncPicture * picture, GstVaapiEncSequence * sequence)
|
|
|
|
{
|
|
|
|
GstVaapiEncPackedHeader *packed_seq;
|
2014-01-21 14:28:34 +00:00
|
|
|
GstBitWriter bs;
|
2014-01-21 15:05:22 +00:00
|
|
|
VAEncPackedHeaderParameterBuffer packed_seq_param = { 0 };
|
2014-01-10 13:05:40 +00:00
|
|
|
const VAEncSequenceParameterBufferH264 *const seq_param = sequence->param;
|
2014-02-17 07:51:43 +00:00
|
|
|
GstVaapiProfile profile = encoder->profile;
|
|
|
|
|
2014-01-21 18:04:41 +00:00
|
|
|
VAEncMiscParameterHRD hrd_params;
|
2013-07-29 07:46:11 +00:00
|
|
|
guint32 data_bit_size;
|
|
|
|
guint8 *data;
|
|
|
|
|
2014-01-21 18:04:41 +00:00
|
|
|
fill_hrd_params (encoder, &hrd_params);
|
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
gst_bit_writer_init (&bs, 128 * 8);
|
|
|
|
WRITE_UINT32 (&bs, 0x00000001, 32); /* start code */
|
2014-06-03 12:30:39 +00:00
|
|
|
bs_write_nal_header (&bs, GST_H264_NAL_REF_IDC_HIGH, GST_H264_NAL_SPS);
|
2014-02-17 07:51:43 +00:00
|
|
|
|
|
|
|
/* Set High profile for encoding the MVC base view. Otherwise, some
|
|
|
|
traditional decoder cannot recognize MVC profile streams with
|
|
|
|
only the base view in there */
|
|
|
|
if (profile == GST_VAAPI_PROFILE_H264_MULTIVIEW_HIGH ||
|
|
|
|
profile == GST_VAAPI_PROFILE_H264_STEREO_HIGH)
|
|
|
|
profile = GST_VAAPI_PROFILE_H264_HIGH;
|
|
|
|
|
|
|
|
bs_write_sps (&bs, seq_param, profile, &hrd_params);
|
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
g_assert (GST_BIT_WRITER_BIT_SIZE (&bs) % 8 == 0);
|
|
|
|
data_bit_size = GST_BIT_WRITER_BIT_SIZE (&bs);
|
|
|
|
data = GST_BIT_WRITER_DATA (&bs);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
packed_seq_param.type = VAEncPackedHeaderSequence;
|
|
|
|
packed_seq_param.bit_length = data_bit_size;
|
|
|
|
packed_seq_param.has_emulation_bytes = 0;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
packed_seq = gst_vaapi_enc_packed_header_new (GST_VAAPI_ENCODER (encoder),
|
2014-01-21 15:05:22 +00:00
|
|
|
&packed_seq_param, sizeof (packed_seq_param),
|
2013-07-29 07:46:11 +00:00
|
|
|
data, (data_bit_size + 7) / 8);
|
|
|
|
g_assert (packed_seq);
|
|
|
|
|
|
|
|
gst_vaapi_enc_picture_add_packed_header (picture, packed_seq);
|
2014-01-10 13:05:40 +00:00
|
|
|
gst_vaapi_codec_object_replace (&packed_seq, NULL);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/* store sps data */
|
|
|
|
_check_sps_pps_status (encoder, data + 4, data_bit_size / 8 - 4);
|
2014-01-21 14:28:34 +00:00
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
2013-07-29 07:46:11 +00:00
|
|
|
return TRUE;
|
2014-01-21 14:28:34 +00:00
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write SPS NAL unit");
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return FALSE;
|
|
|
|
}
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-02-17 07:51:43 +00:00
|
|
|
static gboolean
|
|
|
|
add_packed_sequence_header_mvc (GstVaapiEncoderH264 * encoder,
|
|
|
|
GstVaapiEncPicture * picture, GstVaapiEncSequence * sequence)
|
|
|
|
{
|
|
|
|
GstVaapiEncPackedHeader *packed_seq;
|
|
|
|
GstBitWriter bs;
|
|
|
|
VAEncPackedHeaderParameterBuffer packed_header_param_buffer = { 0 };
|
2014-05-06 21:12:39 +00:00
|
|
|
const VAEncSequenceParameterBufferH264 *const seq_param = sequence->param;
|
2014-02-17 07:51:43 +00:00
|
|
|
VAEncMiscParameterHRD hrd_params;
|
|
|
|
guint32 data_bit_size;
|
|
|
|
guint8 *data;
|
|
|
|
|
|
|
|
fill_hrd_params (encoder, &hrd_params);
|
|
|
|
|
|
|
|
/* non-base layer, pack one subset sps */
|
|
|
|
gst_bit_writer_init (&bs, 128 * 8);
|
|
|
|
WRITE_UINT32 (&bs, 0x00000001, 32); /* start code */
|
2014-06-03 12:30:39 +00:00
|
|
|
bs_write_nal_header (&bs, GST_H264_NAL_REF_IDC_HIGH, GST_H264_NAL_SUBSET_SPS);
|
2014-02-17 07:51:43 +00:00
|
|
|
|
2016-02-02 16:59:57 +00:00
|
|
|
bs_write_subset_sps (&bs, seq_param, encoder->profile, encoder->num_views,
|
|
|
|
encoder->view_ids, &hrd_params);
|
2014-02-17 07:51:43 +00:00
|
|
|
|
|
|
|
g_assert (GST_BIT_WRITER_BIT_SIZE (&bs) % 8 == 0);
|
|
|
|
data_bit_size = GST_BIT_WRITER_BIT_SIZE (&bs);
|
|
|
|
data = GST_BIT_WRITER_DATA (&bs);
|
|
|
|
|
|
|
|
packed_header_param_buffer.type = VAEncPackedHeaderSequence;
|
|
|
|
packed_header_param_buffer.bit_length = data_bit_size;
|
|
|
|
packed_header_param_buffer.has_emulation_bytes = 0;
|
|
|
|
|
|
|
|
packed_seq = gst_vaapi_enc_packed_header_new (GST_VAAPI_ENCODER (encoder),
|
|
|
|
&packed_header_param_buffer, sizeof (packed_header_param_buffer),
|
|
|
|
data, (data_bit_size + 7) / 8);
|
|
|
|
g_assert (packed_seq);
|
|
|
|
|
|
|
|
gst_vaapi_enc_picture_add_packed_header (picture, packed_seq);
|
|
|
|
gst_vaapi_mini_object_replace ((GstVaapiMiniObject **) & packed_seq, NULL);
|
2014-05-06 21:09:19 +00:00
|
|
|
|
|
|
|
/* store subset sps data */
|
|
|
|
_check_sps_pps_status (encoder, data + 4, data_bit_size / 8 - 4);
|
2014-02-17 07:51:43 +00:00
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write SPS NAL unit");
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Adds the supplied picture header (PPS) to the list of packed
|
|
|
|
headers to pass down as-is to the encoder */
|
2013-07-29 07:46:11 +00:00
|
|
|
static gboolean
|
2014-01-21 15:05:22 +00:00
|
|
|
add_packed_picture_header (GstVaapiEncoderH264 * encoder,
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncPicture * picture)
|
|
|
|
{
|
|
|
|
GstVaapiEncPackedHeader *packed_pic;
|
2014-01-21 14:28:34 +00:00
|
|
|
GstBitWriter bs;
|
2014-01-21 15:05:22 +00:00
|
|
|
VAEncPackedHeaderParameterBuffer packed_pic_param = { 0 };
|
2014-01-10 13:05:40 +00:00
|
|
|
const VAEncPictureParameterBufferH264 *const pic_param = picture->param;
|
2013-07-29 07:46:11 +00:00
|
|
|
guint32 data_bit_size;
|
|
|
|
guint8 *data;
|
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
gst_bit_writer_init (&bs, 128 * 8);
|
|
|
|
WRITE_UINT32 (&bs, 0x00000001, 32); /* start code */
|
2014-06-03 12:30:39 +00:00
|
|
|
bs_write_nal_header (&bs, GST_H264_NAL_REF_IDC_HIGH, GST_H264_NAL_PPS);
|
2014-01-21 14:28:34 +00:00
|
|
|
bs_write_pps (&bs, pic_param, encoder->profile);
|
|
|
|
g_assert (GST_BIT_WRITER_BIT_SIZE (&bs) % 8 == 0);
|
|
|
|
data_bit_size = GST_BIT_WRITER_BIT_SIZE (&bs);
|
|
|
|
data = GST_BIT_WRITER_DATA (&bs);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
packed_pic_param.type = VAEncPackedHeaderPicture;
|
|
|
|
packed_pic_param.bit_length = data_bit_size;
|
|
|
|
packed_pic_param.has_emulation_bytes = 0;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
packed_pic = gst_vaapi_enc_packed_header_new (GST_VAAPI_ENCODER (encoder),
|
2014-01-21 15:05:22 +00:00
|
|
|
&packed_pic_param, sizeof (packed_pic_param),
|
2013-07-29 07:46:11 +00:00
|
|
|
data, (data_bit_size + 7) / 8);
|
|
|
|
g_assert (packed_pic);
|
|
|
|
|
|
|
|
gst_vaapi_enc_picture_add_packed_header (picture, packed_pic);
|
2014-01-10 13:05:40 +00:00
|
|
|
gst_vaapi_codec_object_replace (&packed_pic, NULL);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/* store pps data */
|
|
|
|
_check_sps_pps_status (encoder, data + 4, data_bit_size / 8 - 4);
|
2014-01-21 14:28:34 +00:00
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
2013-07-29 07:46:11 +00:00
|
|
|
return TRUE;
|
2014-01-21 14:28:34 +00:00
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write PPS NAL unit");
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return FALSE;
|
|
|
|
}
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2015-07-02 18:00:14 +00:00
|
|
|
static gboolean
|
|
|
|
add_packed_sei_header (GstVaapiEncoderH264 * encoder,
|
2016-02-02 16:59:57 +00:00
|
|
|
GstVaapiEncPicture * picture, GstVaapiH264SeiPayloadType payloadtype)
|
2015-07-02 18:00:14 +00:00
|
|
|
{
|
|
|
|
GstVaapiEncPackedHeader *packed_sei;
|
|
|
|
GstBitWriter bs, bs_buf_period, bs_pic_timing;
|
|
|
|
VAEncPackedHeaderParameterBuffer packed_sei_param = { 0 };
|
|
|
|
guint32 data_bit_size;
|
|
|
|
guint8 buf_period_payload_size = 0, pic_timing_payload_size = 0;
|
2016-02-01 13:22:10 +00:00
|
|
|
guint8 *data, *buf_period_payload = NULL, *pic_timing_payload = NULL;
|
2015-07-02 18:00:14 +00:00
|
|
|
gboolean need_buf_period, need_pic_timing;
|
|
|
|
|
|
|
|
gst_bit_writer_init (&bs_buf_period, 128 * 8);
|
|
|
|
gst_bit_writer_init (&bs_pic_timing, 128 * 8);
|
|
|
|
gst_bit_writer_init (&bs, 128 * 8);
|
|
|
|
|
|
|
|
need_buf_period = GST_VAAPI_H264_SEI_BUF_PERIOD & payloadtype;
|
|
|
|
need_pic_timing = GST_VAAPI_H264_SEI_PIC_TIMING & payloadtype;
|
|
|
|
|
|
|
|
if (need_buf_period) {
|
|
|
|
/* Write a Buffering Period SEI message */
|
|
|
|
bs_write_sei_buf_period (&bs_buf_period, encoder, picture);
|
|
|
|
/* Write byte alignment bits */
|
|
|
|
if (GST_BIT_WRITER_BIT_SIZE (&bs_buf_period) % 8 != 0)
|
2016-02-02 16:59:57 +00:00
|
|
|
bs_write_trailing_bits (&bs_buf_period);
|
|
|
|
buf_period_payload_size = (GST_BIT_WRITER_BIT_SIZE (&bs_buf_period)) / 8;
|
2015-07-02 18:00:14 +00:00
|
|
|
buf_period_payload = GST_BIT_WRITER_DATA (&bs_buf_period);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (need_pic_timing) {
|
|
|
|
/* Write a Picture Timing SEI message */
|
|
|
|
if (GST_VAAPI_H264_SEI_PIC_TIMING & payloadtype)
|
|
|
|
bs_write_sei_pic_timing (&bs_pic_timing, encoder, picture);
|
|
|
|
/* Write byte alignment bits */
|
|
|
|
if (GST_BIT_WRITER_BIT_SIZE (&bs_pic_timing) % 8 != 0)
|
2016-02-02 16:59:57 +00:00
|
|
|
bs_write_trailing_bits (&bs_pic_timing);
|
|
|
|
pic_timing_payload_size = (GST_BIT_WRITER_BIT_SIZE (&bs_pic_timing)) / 8;
|
2015-07-02 18:00:14 +00:00
|
|
|
pic_timing_payload = GST_BIT_WRITER_DATA (&bs_pic_timing);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Write the SEI message */
|
|
|
|
WRITE_UINT32 (&bs, 0x00000001, 32); /* start code */
|
|
|
|
bs_write_nal_header (&bs, GST_H264_NAL_REF_IDC_NONE, GST_H264_NAL_SEI);
|
|
|
|
|
|
|
|
if (need_buf_period) {
|
|
|
|
WRITE_UINT32 (&bs, GST_H264_SEI_BUF_PERIOD, 8);
|
|
|
|
WRITE_UINT32 (&bs, buf_period_payload_size, 8);
|
|
|
|
/* Add buffering period sei message */
|
|
|
|
gst_bit_writer_put_bytes (&bs, buf_period_payload, buf_period_payload_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (need_pic_timing) {
|
|
|
|
WRITE_UINT32 (&bs, GST_H264_SEI_PIC_TIMING, 8);
|
|
|
|
WRITE_UINT32 (&bs, pic_timing_payload_size, 8);
|
|
|
|
/* Add picture timing sei message */
|
|
|
|
gst_bit_writer_put_bytes (&bs, pic_timing_payload, pic_timing_payload_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* rbsp_trailing_bits */
|
|
|
|
bs_write_trailing_bits (&bs);
|
|
|
|
|
|
|
|
g_assert (GST_BIT_WRITER_BIT_SIZE (&bs) % 8 == 0);
|
|
|
|
data_bit_size = GST_BIT_WRITER_BIT_SIZE (&bs);
|
|
|
|
data = GST_BIT_WRITER_DATA (&bs);
|
|
|
|
|
|
|
|
packed_sei_param.type = VAEncPackedHeaderH264_SEI;
|
|
|
|
packed_sei_param.bit_length = data_bit_size;
|
|
|
|
packed_sei_param.has_emulation_bytes = 0;
|
|
|
|
|
|
|
|
packed_sei = gst_vaapi_enc_packed_header_new (GST_VAAPI_ENCODER (encoder),
|
|
|
|
&packed_sei_param, sizeof (packed_sei_param),
|
|
|
|
data, (data_bit_size + 7) / 8);
|
|
|
|
g_assert (packed_sei);
|
|
|
|
|
|
|
|
gst_vaapi_enc_picture_add_packed_header (picture, packed_sei);
|
|
|
|
gst_vaapi_codec_object_replace (&packed_sei, NULL);
|
|
|
|
|
|
|
|
gst_bit_writer_clear (&bs_buf_period, TRUE);
|
|
|
|
gst_bit_writer_clear (&bs_pic_timing, TRUE);
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write SEI NAL unit");
|
|
|
|
gst_bit_writer_clear (&bs_buf_period, TRUE);
|
|
|
|
gst_bit_writer_clear (&bs_pic_timing, TRUE);
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-05-06 21:09:45 +00:00
|
|
|
static gboolean
|
|
|
|
get_nal_hdr_attributes (GstVaapiEncPicture * picture,
|
|
|
|
guint8 * nal_ref_idc, guint8 * nal_unit_type)
|
|
|
|
{
|
|
|
|
switch (picture->type) {
|
|
|
|
case GST_VAAPI_PICTURE_TYPE_I:
|
2014-06-03 12:30:39 +00:00
|
|
|
*nal_ref_idc = GST_H264_NAL_REF_IDC_HIGH;
|
2014-05-06 21:09:45 +00:00
|
|
|
if (GST_VAAPI_ENC_PICTURE_IS_IDR (picture))
|
2014-06-03 12:30:39 +00:00
|
|
|
*nal_unit_type = GST_H264_NAL_SLICE_IDR;
|
2014-05-06 21:09:45 +00:00
|
|
|
else
|
2014-06-03 12:30:39 +00:00
|
|
|
*nal_unit_type = GST_H264_NAL_SLICE;
|
2014-05-06 21:09:45 +00:00
|
|
|
break;
|
|
|
|
case GST_VAAPI_PICTURE_TYPE_P:
|
2014-06-03 12:30:39 +00:00
|
|
|
*nal_ref_idc = GST_H264_NAL_REF_IDC_MEDIUM;
|
|
|
|
*nal_unit_type = GST_H264_NAL_SLICE;
|
2014-05-06 21:09:45 +00:00
|
|
|
break;
|
|
|
|
case GST_VAAPI_PICTURE_TYPE_B:
|
2014-06-03 12:30:39 +00:00
|
|
|
*nal_ref_idc = GST_H264_NAL_REF_IDC_NONE;
|
|
|
|
*nal_unit_type = GST_H264_NAL_SLICE;
|
2014-05-06 21:09:45 +00:00
|
|
|
break;
|
|
|
|
default:
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
/* Adds the supplied prefix nal header to the list of packed
|
|
|
|
headers to pass down as-is to the encoder */
|
|
|
|
static gboolean
|
|
|
|
add_packed_prefix_nal_header (GstVaapiEncoderH264 * encoder,
|
2014-06-05 12:32:29 +00:00
|
|
|
GstVaapiEncPicture * picture, GstVaapiEncSlice * slice)
|
2014-05-06 21:12:39 +00:00
|
|
|
{
|
|
|
|
GstVaapiEncPackedHeader *packed_prefix_nal;
|
|
|
|
GstBitWriter bs;
|
|
|
|
VAEncPackedHeaderParameterBuffer packed_prefix_nal_param = { 0 };
|
|
|
|
guint32 data_bit_size;
|
|
|
|
guint8 *data;
|
|
|
|
guint8 nal_ref_idc, nal_unit_type;
|
|
|
|
|
|
|
|
gst_bit_writer_init (&bs, 128 * 8);
|
|
|
|
WRITE_UINT32 (&bs, 0x00000001, 32); /* start code */
|
|
|
|
|
|
|
|
if (!get_nal_hdr_attributes (picture, &nal_ref_idc, &nal_unit_type))
|
|
|
|
goto bs_error;
|
2014-06-03 12:30:39 +00:00
|
|
|
nal_unit_type = GST_H264_NAL_PREFIX_UNIT;
|
2014-05-06 21:12:39 +00:00
|
|
|
|
|
|
|
bs_write_nal_header (&bs, nal_ref_idc, nal_unit_type);
|
|
|
|
bs_write_nal_header_mvc_extension (&bs, picture, encoder->view_idx);
|
|
|
|
g_assert (GST_BIT_WRITER_BIT_SIZE (&bs) % 8 == 0);
|
|
|
|
data_bit_size = GST_BIT_WRITER_BIT_SIZE (&bs);
|
|
|
|
data = GST_BIT_WRITER_DATA (&bs);
|
|
|
|
|
|
|
|
packed_prefix_nal_param.type = VAEncPackedHeaderRawData;
|
|
|
|
packed_prefix_nal_param.bit_length = data_bit_size;
|
|
|
|
packed_prefix_nal_param.has_emulation_bytes = 0;
|
|
|
|
|
|
|
|
packed_prefix_nal =
|
|
|
|
gst_vaapi_enc_packed_header_new (GST_VAAPI_ENCODER (encoder),
|
|
|
|
&packed_prefix_nal_param, sizeof (packed_prefix_nal_param), data,
|
|
|
|
(data_bit_size + 7) / 8);
|
|
|
|
g_assert (packed_prefix_nal);
|
|
|
|
|
2014-06-05 12:32:29 +00:00
|
|
|
gst_vaapi_enc_slice_add_packed_header (slice, packed_prefix_nal);
|
2014-05-06 21:12:39 +00:00
|
|
|
gst_vaapi_codec_object_replace (&packed_prefix_nal, NULL);
|
|
|
|
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write Prefix NAL unit header");
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-05-06 21:09:45 +00:00
|
|
|
/* Adds the supplied slice header to the list of packed
|
|
|
|
headers to pass down as-is to the encoder */
|
|
|
|
static gboolean
|
|
|
|
add_packed_slice_header (GstVaapiEncoderH264 * encoder,
|
|
|
|
GstVaapiEncPicture * picture, GstVaapiEncSlice * slice)
|
|
|
|
{
|
|
|
|
GstVaapiEncPackedHeader *packed_slice;
|
|
|
|
GstBitWriter bs;
|
|
|
|
VAEncPackedHeaderParameterBuffer packed_slice_param = { 0 };
|
|
|
|
const VAEncSliceParameterBufferH264 *const slice_param = slice->param;
|
|
|
|
guint32 data_bit_size;
|
|
|
|
guint8 *data;
|
|
|
|
guint8 nal_ref_idc, nal_unit_type;
|
|
|
|
|
|
|
|
gst_bit_writer_init (&bs, 128 * 8);
|
|
|
|
WRITE_UINT32 (&bs, 0x00000001, 32); /* start code */
|
|
|
|
|
|
|
|
if (!get_nal_hdr_attributes (picture, &nal_ref_idc, &nal_unit_type))
|
|
|
|
goto bs_error;
|
2014-06-26 08:39:38 +00:00
|
|
|
/* pack nal_unit_header_mvc_extension() for the non base view */
|
|
|
|
if (encoder->is_mvc && encoder->view_idx) {
|
|
|
|
bs_write_nal_header (&bs, nal_ref_idc, GST_H264_NAL_SLICE_EXT);
|
2016-02-02 16:59:57 +00:00
|
|
|
bs_write_nal_header_mvc_extension (&bs, picture,
|
|
|
|
encoder->view_ids[encoder->view_idx]);
|
2014-06-26 08:39:38 +00:00
|
|
|
} else
|
|
|
|
bs_write_nal_header (&bs, nal_ref_idc, nal_unit_type);
|
2014-05-06 21:09:45 +00:00
|
|
|
|
|
|
|
bs_write_slice (&bs, slice_param, encoder, picture);
|
|
|
|
data_bit_size = GST_BIT_WRITER_BIT_SIZE (&bs);
|
|
|
|
data = GST_BIT_WRITER_DATA (&bs);
|
|
|
|
|
|
|
|
packed_slice_param.type = VAEncPackedHeaderSlice;
|
|
|
|
packed_slice_param.bit_length = data_bit_size;
|
|
|
|
packed_slice_param.has_emulation_bytes = 0;
|
|
|
|
|
|
|
|
packed_slice = gst_vaapi_enc_packed_header_new (GST_VAAPI_ENCODER (encoder),
|
|
|
|
&packed_slice_param, sizeof (packed_slice_param),
|
|
|
|
data, (data_bit_size + 7) / 8);
|
|
|
|
g_assert (packed_slice);
|
|
|
|
|
2014-06-05 12:32:29 +00:00
|
|
|
gst_vaapi_enc_slice_add_packed_header (slice, packed_slice);
|
2014-05-06 21:09:45 +00:00
|
|
|
gst_vaapi_codec_object_replace (&packed_slice, NULL);
|
|
|
|
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_WARNING ("failed to write Slice NAL unit header");
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Reference picture management */
|
2013-07-29 07:46:11 +00:00
|
|
|
static void
|
|
|
|
reference_pic_free (GstVaapiEncoderH264 * encoder, GstVaapiEncoderH264Ref * ref)
|
|
|
|
{
|
|
|
|
if (!ref)
|
|
|
|
return;
|
|
|
|
if (ref->pic)
|
|
|
|
gst_vaapi_encoder_release_surface (GST_VAAPI_ENCODER (encoder), ref->pic);
|
|
|
|
g_slice_free (GstVaapiEncoderH264Ref, ref);
|
|
|
|
}
|
|
|
|
|
|
|
|
static inline GstVaapiEncoderH264Ref *
|
|
|
|
reference_pic_create (GstVaapiEncoderH264 * encoder,
|
|
|
|
GstVaapiEncPicture * picture, GstVaapiSurfaceProxy * surface)
|
|
|
|
{
|
2014-01-10 13:05:40 +00:00
|
|
|
GstVaapiEncoderH264Ref *const ref = g_slice_new0 (GstVaapiEncoderH264Ref);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
ref->pic = surface;
|
|
|
|
ref->frame_num = picture->frame_num;
|
|
|
|
ref->poc = picture->poc;
|
|
|
|
return ref;
|
|
|
|
}
|
|
|
|
|
|
|
|
static gboolean
|
|
|
|
reference_list_update (GstVaapiEncoderH264 * encoder,
|
|
|
|
GstVaapiEncPicture * picture, GstVaapiSurfaceProxy * surface)
|
|
|
|
{
|
|
|
|
GstVaapiEncoderH264Ref *ref;
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewRefPool *const ref_pool =
|
|
|
|
&encoder->ref_pools[encoder->view_idx];
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
if (GST_VAAPI_PICTURE_TYPE_B == picture->type) {
|
|
|
|
gst_vaapi_encoder_release_surface (GST_VAAPI_ENCODER (encoder), surface);
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
if (GST_VAAPI_ENC_PICTURE_IS_IDR (picture)) {
|
2014-02-17 03:10:26 +00:00
|
|
|
while (!g_queue_is_empty (&ref_pool->ref_list))
|
|
|
|
reference_pic_free (encoder, g_queue_pop_head (&ref_pool->ref_list));
|
|
|
|
} else if (g_queue_get_length (&ref_pool->ref_list) >=
|
|
|
|
ref_pool->max_ref_frames) {
|
|
|
|
reference_pic_free (encoder, g_queue_pop_head (&ref_pool->ref_list));
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
ref = reference_pic_create (encoder, picture, surface);
|
2014-02-17 03:10:26 +00:00
|
|
|
g_queue_push_tail (&ref_pool->ref_list, ref);
|
|
|
|
g_assert (g_queue_get_length (&ref_pool->ref_list) <=
|
|
|
|
ref_pool->max_ref_frames);
|
2013-07-29 07:46:11 +00:00
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
static gboolean
|
|
|
|
reference_list_init (GstVaapiEncoderH264 * encoder,
|
|
|
|
GstVaapiEncPicture * picture,
|
|
|
|
GstVaapiEncoderH264Ref ** reflist_0,
|
|
|
|
guint * reflist_0_count,
|
|
|
|
GstVaapiEncoderH264Ref ** reflist_1, guint * reflist_1_count)
|
|
|
|
{
|
|
|
|
GstVaapiEncoderH264Ref *tmp;
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewRefPool *const ref_pool =
|
|
|
|
&encoder->ref_pools[encoder->view_idx];
|
2013-07-29 07:46:11 +00:00
|
|
|
GList *iter, *list_0_start = NULL, *list_1_start = NULL;
|
|
|
|
guint count;
|
|
|
|
|
|
|
|
*reflist_0_count = 0;
|
|
|
|
*reflist_1_count = 0;
|
|
|
|
if (picture->type == GST_VAAPI_PICTURE_TYPE_I)
|
|
|
|
return TRUE;
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
iter = g_queue_peek_tail_link (&ref_pool->ref_list);
|
2013-07-29 07:46:11 +00:00
|
|
|
for (; iter; iter = g_list_previous (iter)) {
|
|
|
|
tmp = (GstVaapiEncoderH264Ref *) iter->data;
|
|
|
|
g_assert (tmp && tmp->poc != picture->poc);
|
2014-04-01 08:26:04 +00:00
|
|
|
if (_poc_greater_than (picture->poc, tmp->poc, encoder->max_pic_order_cnt)) {
|
2013-07-29 07:46:11 +00:00
|
|
|
list_0_start = iter;
|
|
|
|
list_1_start = g_list_next (iter);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* order reflist_0 */
|
|
|
|
g_assert (list_0_start);
|
|
|
|
iter = list_0_start;
|
|
|
|
count = 0;
|
|
|
|
for (; iter; iter = g_list_previous (iter)) {
|
|
|
|
reflist_0[count] = (GstVaapiEncoderH264Ref *) iter->data;
|
|
|
|
++count;
|
|
|
|
}
|
|
|
|
*reflist_0_count = count;
|
|
|
|
|
|
|
|
if (picture->type != GST_VAAPI_PICTURE_TYPE_B)
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* order reflist_1 */
|
|
|
|
count = 0;
|
|
|
|
iter = list_1_start;
|
|
|
|
for (; iter; iter = g_list_next (iter)) {
|
|
|
|
reflist_1[count] = (GstVaapiEncoderH264Ref *) iter->data;
|
|
|
|
++count;
|
|
|
|
}
|
|
|
|
*reflist_1_count = count;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Fills in VA sequence parameter buffer */
|
2013-07-29 07:46:11 +00:00
|
|
|
static gboolean
|
2014-01-21 15:05:22 +00:00
|
|
|
fill_sequence (GstVaapiEncoderH264 * encoder, GstVaapiEncSequence * sequence)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-01-10 13:05:40 +00:00
|
|
|
VAEncSequenceParameterBufferH264 *const seq_param = sequence->param;
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewRefPool *const ref_pool =
|
|
|
|
&encoder->ref_pools[encoder->view_idx];
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
memset (seq_param, 0, sizeof (VAEncSequenceParameterBufferH264));
|
2014-02-17 07:51:43 +00:00
|
|
|
seq_param->seq_parameter_set_id = encoder->view_idx;
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->level_idc = encoder->level_idc;
|
|
|
|
seq_param->intra_period = GST_VAAPI_ENCODER_KEYFRAME_PERIOD (encoder);
|
2014-11-27 09:21:03 +00:00
|
|
|
seq_param->intra_idr_period = GST_VAAPI_ENCODER_KEYFRAME_PERIOD (encoder);
|
2014-01-21 16:35:24 +00:00
|
|
|
seq_param->ip_period = 1 + encoder->num_bframes;
|
2014-11-27 10:11:03 +00:00
|
|
|
seq_param->ip_period = seq_param->intra_period > 1 ?
|
|
|
|
(1 + encoder->num_bframes) : 0;
|
2014-01-22 13:43:24 +00:00
|
|
|
seq_param->bits_per_second = encoder->bitrate_bits;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
seq_param->max_num_ref_frames = ref_pool->max_ref_frames;
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->picture_width_in_mbs = encoder->mb_width;
|
|
|
|
seq_param->picture_height_in_mbs = encoder->mb_height;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/*sequence field values */
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->seq_fields.value = 0;
|
|
|
|
seq_param->seq_fields.bits.chroma_format_idc = 1;
|
|
|
|
seq_param->seq_fields.bits.frame_mbs_only_flag = 1;
|
|
|
|
seq_param->seq_fields.bits.mb_adaptive_frame_field_flag = FALSE;
|
|
|
|
seq_param->seq_fields.bits.seq_scaling_matrix_present_flag = FALSE;
|
2013-07-29 07:46:11 +00:00
|
|
|
/* direct_8x8_inference_flag default false */
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->seq_fields.bits.direct_8x8_inference_flag = FALSE;
|
2013-07-29 07:46:11 +00:00
|
|
|
g_assert (encoder->log2_max_frame_num >= 4);
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->seq_fields.bits.log2_max_frame_num_minus4 =
|
2013-07-29 07:46:11 +00:00
|
|
|
encoder->log2_max_frame_num - 4;
|
|
|
|
/* picture order count */
|
2014-05-06 21:09:45 +00:00
|
|
|
encoder->pic_order_cnt_type = seq_param->seq_fields.bits.pic_order_cnt_type =
|
|
|
|
0;
|
2013-07-29 07:46:11 +00:00
|
|
|
g_assert (encoder->log2_max_pic_order_cnt >= 4);
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->seq_fields.bits.log2_max_pic_order_cnt_lsb_minus4 =
|
2013-07-29 07:46:11 +00:00
|
|
|
encoder->log2_max_pic_order_cnt - 4;
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->bit_depth_luma_minus8 = 0;
|
|
|
|
seq_param->bit_depth_chroma_minus8 = 0;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/* not used if pic_order_cnt_type == 0 */
|
2014-01-10 13:05:40 +00:00
|
|
|
if (seq_param->seq_fields.bits.pic_order_cnt_type == 1) {
|
2014-05-06 21:09:45 +00:00
|
|
|
encoder->delta_pic_order_always_zero_flag =
|
|
|
|
seq_param->seq_fields.bits.delta_pic_order_always_zero_flag = TRUE;
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->num_ref_frames_in_pic_order_cnt_cycle = 0;
|
|
|
|
seq_param->offset_for_non_ref_pic = 0;
|
|
|
|
seq_param->offset_for_top_to_bottom_field = 0;
|
|
|
|
memset (seq_param->offset_for_ref_frame, 0,
|
|
|
|
sizeof (seq_param->offset_for_ref_frame));
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-13 12:41:35 +00:00
|
|
|
/* frame_cropping_flag */
|
|
|
|
if ((GST_VAAPI_ENCODER_WIDTH (encoder) & 15) ||
|
|
|
|
(GST_VAAPI_ENCODER_HEIGHT (encoder) & 15)) {
|
2014-01-15 14:10:48 +00:00
|
|
|
static const guint SubWidthC[] = { 1, 2, 2, 1 };
|
|
|
|
static const guint SubHeightC[] = { 1, 2, 1, 1 };
|
|
|
|
const guint CropUnitX =
|
|
|
|
SubWidthC[seq_param->seq_fields.bits.chroma_format_idc];
|
|
|
|
const guint CropUnitY =
|
|
|
|
SubHeightC[seq_param->seq_fields.bits.chroma_format_idc] *
|
|
|
|
(2 - seq_param->seq_fields.bits.frame_mbs_only_flag);
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->frame_cropping_flag = 1;
|
|
|
|
seq_param->frame_crop_left_offset = 0;
|
2014-01-13 12:41:35 +00:00
|
|
|
seq_param->frame_crop_right_offset =
|
2014-01-15 14:10:48 +00:00
|
|
|
(16 * encoder->mb_width -
|
|
|
|
GST_VAAPI_ENCODER_WIDTH (encoder)) / CropUnitX;
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->frame_crop_top_offset = 0;
|
|
|
|
seq_param->frame_crop_bottom_offset =
|
2014-01-15 14:10:48 +00:00
|
|
|
(16 * encoder->mb_height -
|
|
|
|
GST_VAAPI_ENCODER_HEIGHT (encoder)) / CropUnitY;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-15 14:46:19 +00:00
|
|
|
/* VUI parameters are always set, at least for timing_info (framerate) */
|
|
|
|
seq_param->vui_parameters_present_flag = TRUE;
|
2014-01-10 13:05:40 +00:00
|
|
|
if (seq_param->vui_parameters_present_flag) {
|
2014-11-27 09:13:20 +00:00
|
|
|
seq_param->vui_fields.bits.aspect_ratio_info_present_flag = TRUE;
|
2014-06-27 11:15:13 +00:00
|
|
|
if (seq_param->vui_fields.bits.aspect_ratio_info_present_flag) {
|
|
|
|
const GstVideoInfo *const vip = GST_VAAPI_ENCODER_VIDEO_INFO (encoder);
|
2014-11-27 09:13:20 +00:00
|
|
|
seq_param->aspect_ratio_idc = 0xff;
|
2014-06-27 11:15:13 +00:00
|
|
|
seq_param->sar_width = GST_VIDEO_INFO_PAR_N (vip);
|
|
|
|
seq_param->sar_height = GST_VIDEO_INFO_PAR_D (vip);
|
|
|
|
}
|
2014-01-10 13:05:40 +00:00
|
|
|
seq_param->vui_fields.bits.bitstream_restriction_flag = FALSE;
|
2014-06-27 07:37:38 +00:00
|
|
|
/* if vui_parameters_present_flag is TRUE and sps data belongs to
|
|
|
|
* subset sps, timing_info_preset_flag should be zero (H.7.4.2.1.1) */
|
|
|
|
seq_param->vui_fields.bits.timing_info_present_flag = !encoder->view_idx;
|
2014-01-10 13:05:40 +00:00
|
|
|
if (seq_param->vui_fields.bits.timing_info_present_flag) {
|
|
|
|
seq_param->num_units_in_tick = GST_VAAPI_ENCODER_FPS_D (encoder);
|
|
|
|
seq_param->time_scale = GST_VAAPI_ENCODER_FPS_N (encoder) * 2;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Fills in VA picture parameter buffer */
|
2013-07-29 07:46:11 +00:00
|
|
|
static gboolean
|
2014-01-21 15:05:22 +00:00
|
|
|
fill_picture (GstVaapiEncoderH264 * encoder, GstVaapiEncPicture * picture,
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiCodedBuffer * codedbuf, GstVaapiSurfaceProxy * surface)
|
|
|
|
{
|
2014-01-10 13:05:40 +00:00
|
|
|
VAEncPictureParameterBufferH264 *const pic_param = picture->param;
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewRefPool *const ref_pool =
|
|
|
|
&encoder->ref_pools[encoder->view_idx];
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncoderH264Ref *ref_pic;
|
|
|
|
GList *reflist;
|
|
|
|
guint i;
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
memset (pic_param, 0, sizeof (VAEncPictureParameterBufferH264));
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/* reference list, */
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->CurrPic.picture_id = GST_VAAPI_SURFACE_PROXY_SURFACE_ID (surface);
|
|
|
|
pic_param->CurrPic.TopFieldOrderCnt = picture->poc;
|
2013-07-29 07:46:11 +00:00
|
|
|
i = 0;
|
|
|
|
if (picture->type != GST_VAAPI_PICTURE_TYPE_I) {
|
2014-02-17 03:10:26 +00:00
|
|
|
for (reflist = g_queue_peek_head_link (&ref_pool->ref_list);
|
2013-07-29 07:46:11 +00:00
|
|
|
reflist; reflist = g_list_next (reflist)) {
|
|
|
|
ref_pic = reflist->data;
|
|
|
|
g_assert (ref_pic && ref_pic->pic &&
|
|
|
|
GST_VAAPI_SURFACE_PROXY_SURFACE_ID (ref_pic->pic) != VA_INVALID_ID);
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->ReferenceFrames[i].picture_id =
|
2013-07-29 07:46:11 +00:00
|
|
|
GST_VAAPI_SURFACE_PROXY_SURFACE_ID (ref_pic->pic);
|
|
|
|
++i;
|
|
|
|
}
|
2014-02-17 03:10:26 +00:00
|
|
|
g_assert (i <= 16 && i <= ref_pool->max_ref_frames);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
for (; i < 16; ++i) {
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->ReferenceFrames[i].picture_id = VA_INVALID_ID;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->coded_buf = GST_VAAPI_OBJECT_ID (codedbuf);
|
|
|
|
|
2014-02-17 07:51:43 +00:00
|
|
|
pic_param->pic_parameter_set_id = encoder->view_idx;
|
2015-02-24 15:14:33 +00:00
|
|
|
pic_param->seq_parameter_set_id = encoder->view_idx ? 1 : 0;
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->last_picture = 0; /* means last encoding picture */
|
|
|
|
pic_param->frame_num = picture->frame_num;
|
|
|
|
pic_param->pic_init_qp = encoder->init_qp;
|
|
|
|
pic_param->num_ref_idx_l0_active_minus1 =
|
2014-02-17 03:10:26 +00:00
|
|
|
(ref_pool->max_reflist0_count ? (ref_pool->max_reflist0_count - 1) : 0);
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->num_ref_idx_l1_active_minus1 =
|
2014-02-17 03:10:26 +00:00
|
|
|
(ref_pool->max_reflist1_count ? (ref_pool->max_reflist1_count - 1) : 0);
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->chroma_qp_index_offset = 0;
|
|
|
|
pic_param->second_chroma_qp_index_offset = 0;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/* set picture fields */
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->pic_fields.value = 0;
|
|
|
|
pic_param->pic_fields.bits.idr_pic_flag =
|
|
|
|
GST_VAAPI_ENC_PICTURE_IS_IDR (picture);
|
|
|
|
pic_param->pic_fields.bits.reference_pic_flag =
|
2013-07-29 07:46:11 +00:00
|
|
|
(picture->type != GST_VAAPI_PICTURE_TYPE_B);
|
2013-12-13 09:36:08 +00:00
|
|
|
pic_param->pic_fields.bits.entropy_coding_mode_flag = encoder->use_cabac;
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->pic_fields.bits.weighted_pred_flag = FALSE;
|
|
|
|
pic_param->pic_fields.bits.weighted_bipred_idc = 0;
|
|
|
|
pic_param->pic_fields.bits.constrained_intra_pred_flag = 0;
|
2013-12-13 09:36:08 +00:00
|
|
|
pic_param->pic_fields.bits.transform_8x8_mode_flag = encoder->use_dct8x8;
|
2013-07-29 07:46:11 +00:00
|
|
|
/* enable debloking */
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->pic_fields.bits.deblocking_filter_control_present_flag = TRUE;
|
|
|
|
pic_param->pic_fields.bits.redundant_pic_cnt_present_flag = FALSE;
|
2013-07-29 07:46:11 +00:00
|
|
|
/* bottom_field_pic_order_in_frame_present_flag */
|
2014-01-10 13:05:40 +00:00
|
|
|
pic_param->pic_fields.bits.pic_order_present_flag = FALSE;
|
|
|
|
pic_param->pic_fields.bits.pic_scaling_matrix_present_flag = FALSE;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Adds slice headers to picture */
|
2013-07-29 07:46:11 +00:00
|
|
|
static gboolean
|
2014-01-21 15:05:22 +00:00
|
|
|
add_slice_headers (GstVaapiEncoderH264 * encoder, GstVaapiEncPicture * picture,
|
|
|
|
GstVaapiEncoderH264Ref ** reflist_0, guint reflist_0_count,
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncoderH264Ref ** reflist_1, guint reflist_1_count)
|
|
|
|
{
|
|
|
|
VAEncSliceParameterBufferH264 *slice_param;
|
|
|
|
GstVaapiEncSlice *slice;
|
|
|
|
guint slice_of_mbs, slice_mod_mbs, cur_slice_mbs;
|
2014-01-10 13:05:40 +00:00
|
|
|
guint mb_size;
|
2013-07-29 07:46:11 +00:00
|
|
|
guint last_mb_index;
|
|
|
|
guint i_slice, i_ref;
|
|
|
|
|
|
|
|
g_assert (picture);
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
mb_size = encoder->mb_width * encoder->mb_height;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
g_assert (encoder->num_slices && encoder->num_slices < mb_size);
|
|
|
|
slice_of_mbs = mb_size / encoder->num_slices;
|
|
|
|
slice_mod_mbs = mb_size % encoder->num_slices;
|
2013-07-29 07:46:11 +00:00
|
|
|
last_mb_index = 0;
|
2014-01-10 13:05:40 +00:00
|
|
|
for (i_slice = 0; i_slice < encoder->num_slices; ++i_slice) {
|
2013-07-29 07:46:11 +00:00
|
|
|
cur_slice_mbs = slice_of_mbs;
|
|
|
|
if (slice_mod_mbs) {
|
|
|
|
++cur_slice_mbs;
|
|
|
|
--slice_mod_mbs;
|
|
|
|
}
|
|
|
|
slice = GST_VAAPI_ENC_SLICE_NEW (H264, encoder);
|
|
|
|
g_assert (slice && slice->param_id != VA_INVALID_ID);
|
|
|
|
slice_param = slice->param;
|
|
|
|
|
|
|
|
memset (slice_param, 0, sizeof (VAEncSliceParameterBufferH264));
|
|
|
|
slice_param->macroblock_address = last_mb_index;
|
|
|
|
slice_param->num_macroblocks = cur_slice_mbs;
|
|
|
|
slice_param->macroblock_info = VA_INVALID_ID;
|
2014-01-10 13:05:40 +00:00
|
|
|
slice_param->slice_type = h264_get_slice_type (picture->type);
|
2015-04-03 18:33:44 +00:00
|
|
|
g_assert ((gint8) slice_param->slice_type != -1);
|
2014-02-17 07:51:43 +00:00
|
|
|
slice_param->pic_parameter_set_id = encoder->view_idx;
|
2013-07-29 07:46:11 +00:00
|
|
|
slice_param->idr_pic_id = encoder->idr_num;
|
|
|
|
slice_param->pic_order_cnt_lsb = picture->poc;
|
|
|
|
|
|
|
|
/* not used if pic_order_cnt_type = 0 */
|
|
|
|
slice_param->delta_pic_order_cnt_bottom = 0;
|
2014-01-10 13:05:40 +00:00
|
|
|
memset (slice_param->delta_pic_order_cnt, 0,
|
|
|
|
sizeof (slice_param->delta_pic_order_cnt));
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
/* only works for B frames */
|
2013-07-29 07:46:11 +00:00
|
|
|
slice_param->direct_spatial_mv_pred_flag = FALSE;
|
|
|
|
/* default equal to picture parameters */
|
|
|
|
slice_param->num_ref_idx_active_override_flag = FALSE;
|
|
|
|
if (picture->type != GST_VAAPI_PICTURE_TYPE_I && reflist_0_count > 0)
|
|
|
|
slice_param->num_ref_idx_l0_active_minus1 = reflist_0_count - 1;
|
|
|
|
else
|
|
|
|
slice_param->num_ref_idx_l0_active_minus1 = 0;
|
|
|
|
if (picture->type == GST_VAAPI_PICTURE_TYPE_B && reflist_1_count > 0)
|
|
|
|
slice_param->num_ref_idx_l1_active_minus1 = reflist_1_count - 1;
|
|
|
|
else
|
|
|
|
slice_param->num_ref_idx_l1_active_minus1 = 0;
|
|
|
|
g_assert (slice_param->num_ref_idx_l0_active_minus1 == 0);
|
|
|
|
g_assert (slice_param->num_ref_idx_l1_active_minus1 == 0);
|
|
|
|
|
|
|
|
i_ref = 0;
|
|
|
|
if (picture->type != GST_VAAPI_PICTURE_TYPE_I) {
|
|
|
|
for (; i_ref < reflist_0_count; ++i_ref) {
|
|
|
|
slice_param->RefPicList0[i_ref].picture_id =
|
|
|
|
GST_VAAPI_SURFACE_PROXY_SURFACE_ID (reflist_0[i_ref]->pic);
|
|
|
|
}
|
|
|
|
g_assert (i_ref == 1);
|
|
|
|
}
|
2014-01-10 13:05:40 +00:00
|
|
|
for (; i_ref < G_N_ELEMENTS (slice_param->RefPicList0); ++i_ref) {
|
2013-07-29 07:46:11 +00:00
|
|
|
slice_param->RefPicList0[i_ref].picture_id = VA_INVALID_SURFACE;
|
|
|
|
}
|
|
|
|
|
|
|
|
i_ref = 0;
|
|
|
|
if (picture->type == GST_VAAPI_PICTURE_TYPE_B) {
|
|
|
|
for (; i_ref < reflist_1_count; ++i_ref) {
|
|
|
|
slice_param->RefPicList1[i_ref].picture_id =
|
|
|
|
GST_VAAPI_SURFACE_PROXY_SURFACE_ID (reflist_1[i_ref]->pic);
|
|
|
|
}
|
|
|
|
g_assert (i_ref == 1);
|
|
|
|
}
|
2014-01-10 13:05:40 +00:00
|
|
|
for (; i_ref < G_N_ELEMENTS (slice_param->RefPicList1); ++i_ref) {
|
2013-07-29 07:46:11 +00:00
|
|
|
slice_param->RefPicList1[i_ref].picture_id = VA_INVALID_SURFACE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* not used if pic_param.pic_fields.bits.weighted_pred_flag == FALSE */
|
|
|
|
slice_param->luma_log2_weight_denom = 0;
|
|
|
|
slice_param->chroma_log2_weight_denom = 0;
|
|
|
|
slice_param->luma_weight_l0_flag = FALSE;
|
|
|
|
memset (slice_param->luma_weight_l0, 0,
|
|
|
|
sizeof (slice_param->luma_weight_l0));
|
|
|
|
memset (slice_param->luma_offset_l0, 0,
|
|
|
|
sizeof (slice_param->luma_offset_l0));
|
|
|
|
slice_param->chroma_weight_l0_flag = FALSE;
|
|
|
|
memset (slice_param->chroma_weight_l0, 0,
|
|
|
|
sizeof (slice_param->chroma_weight_l0));
|
|
|
|
memset (slice_param->chroma_offset_l0, 0,
|
|
|
|
sizeof (slice_param->chroma_offset_l0));
|
|
|
|
slice_param->luma_weight_l1_flag = FALSE;
|
|
|
|
memset (slice_param->luma_weight_l1, 0,
|
|
|
|
sizeof (slice_param->luma_weight_l1));
|
|
|
|
memset (slice_param->luma_offset_l1, 0,
|
|
|
|
sizeof (slice_param->luma_offset_l1));
|
|
|
|
slice_param->chroma_weight_l1_flag = FALSE;
|
|
|
|
memset (slice_param->chroma_weight_l1, 0,
|
|
|
|
sizeof (slice_param->chroma_weight_l1));
|
|
|
|
memset (slice_param->chroma_offset_l1, 0,
|
|
|
|
sizeof (slice_param->chroma_offset_l1));
|
|
|
|
|
|
|
|
slice_param->cabac_init_idc = 0;
|
|
|
|
slice_param->slice_qp_delta = encoder->init_qp - encoder->min_qp;
|
|
|
|
if (slice_param->slice_qp_delta > 4)
|
|
|
|
slice_param->slice_qp_delta = 4;
|
|
|
|
slice_param->disable_deblocking_filter_idc = 0;
|
|
|
|
slice_param->slice_alpha_c0_offset_div2 = 2;
|
|
|
|
slice_param->slice_beta_offset_div2 = 2;
|
|
|
|
|
|
|
|
/* set calculation for next slice */
|
|
|
|
last_mb_index += cur_slice_mbs;
|
|
|
|
|
2014-06-26 08:39:38 +00:00
|
|
|
/* add packed Prefix NAL unit before each Coded slice NAL in base view */
|
|
|
|
if (encoder->is_mvc && !encoder->view_idx &&
|
|
|
|
(GST_VAAPI_ENCODER_PACKED_HEADERS (encoder) &
|
|
|
|
VA_ENC_PACKED_HEADER_RAW_DATA)
|
2014-06-05 12:32:29 +00:00
|
|
|
&& !add_packed_prefix_nal_header (encoder, picture, slice))
|
2014-05-06 21:12:39 +00:00
|
|
|
goto error_create_packed_prefix_nal_hdr;
|
2014-05-06 21:09:45 +00:00
|
|
|
if ((GST_VAAPI_ENCODER_PACKED_HEADERS (encoder) &
|
2014-06-05 12:32:29 +00:00
|
|
|
VA_ENC_PACKED_HEADER_SLICE)
|
2014-05-06 21:09:45 +00:00
|
|
|
&& !add_packed_slice_header (encoder, picture, slice))
|
|
|
|
goto error_create_packed_slice_hdr;
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
gst_vaapi_enc_picture_add_slice (picture, slice);
|
2014-01-10 13:05:40 +00:00
|
|
|
gst_vaapi_codec_object_replace (&slice, NULL);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
2014-01-10 13:05:40 +00:00
|
|
|
g_assert (last_mb_index == mb_size);
|
2013-07-29 07:46:11 +00:00
|
|
|
return TRUE;
|
2014-05-06 21:09:45 +00:00
|
|
|
|
|
|
|
error_create_packed_slice_hdr:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to create packed slice header buffer");
|
|
|
|
gst_vaapi_codec_object_replace (&slice, NULL);
|
|
|
|
return FALSE;
|
|
|
|
}
|
2014-05-06 21:12:39 +00:00
|
|
|
error_create_packed_prefix_nal_hdr:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to create packed prefix nal header buffer");
|
|
|
|
gst_vaapi_codec_object_replace (&slice, NULL);
|
|
|
|
return FALSE;
|
|
|
|
}
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Generates and submits SPS header accordingly into the bitstream */
|
2013-07-29 07:46:11 +00:00
|
|
|
static gboolean
|
|
|
|
ensure_sequence (GstVaapiEncoderH264 * encoder, GstVaapiEncPicture * picture)
|
|
|
|
{
|
2014-02-17 07:51:43 +00:00
|
|
|
GstVaapiEncSequence *sequence = NULL;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-06-27 13:38:03 +00:00
|
|
|
/* submit an SPS header before every new I-frame, if codec config changed */
|
|
|
|
if (!encoder->config_changed || picture->type != GST_VAAPI_PICTURE_TYPE_I)
|
2014-01-21 17:01:20 +00:00
|
|
|
return TRUE;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
sequence = GST_VAAPI_ENC_SEQUENCE_NEW (H264, encoder);
|
|
|
|
if (!sequence || !fill_sequence (encoder, sequence))
|
|
|
|
goto error_create_seq_param;
|
|
|
|
|
2014-02-17 07:51:43 +00:00
|
|
|
/* add subset sps for non-base view and sps for base view */
|
|
|
|
if (encoder->is_mvc && encoder->view_idx) {
|
2016-02-02 16:59:57 +00:00
|
|
|
if ((GST_VAAPI_ENCODER_PACKED_HEADERS (encoder) &
|
|
|
|
VA_ENC_PACKED_HEADER_SEQUENCE)
|
2014-02-17 07:51:43 +00:00
|
|
|
&& !add_packed_sequence_header_mvc (encoder, picture, sequence))
|
|
|
|
goto error_create_packed_seq_hdr;
|
|
|
|
} else {
|
2016-02-02 16:59:57 +00:00
|
|
|
if ((GST_VAAPI_ENCODER_PACKED_HEADERS (encoder) &
|
|
|
|
VA_ENC_PACKED_HEADER_SEQUENCE)
|
2014-02-17 07:51:43 +00:00
|
|
|
&& !add_packed_sequence_header (encoder, picture, sequence))
|
|
|
|
goto error_create_packed_seq_hdr;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (sequence) {
|
|
|
|
gst_vaapi_enc_picture_set_sequence (picture, sequence);
|
|
|
|
gst_vaapi_codec_object_replace (&sequence, NULL);
|
|
|
|
}
|
2014-06-27 13:38:03 +00:00
|
|
|
|
|
|
|
if (!encoder->is_mvc || encoder->view_idx > 0)
|
|
|
|
encoder->config_changed = FALSE;
|
2013-07-29 07:46:11 +00:00
|
|
|
return TRUE;
|
|
|
|
|
2014-01-21 17:01:20 +00:00
|
|
|
/* ERRORS */
|
|
|
|
error_create_seq_param:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to create sequence parameter buffer (SPS)");
|
|
|
|
gst_vaapi_codec_object_replace (&sequence, NULL);
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
error_create_packed_seq_hdr:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to create packed sequence header buffer");
|
|
|
|
gst_vaapi_codec_object_replace (&sequence, NULL);
|
|
|
|
return FALSE;
|
|
|
|
}
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Generates additional control parameters */
|
|
|
|
static gboolean
|
|
|
|
ensure_misc_params (GstVaapiEncoderH264 * encoder, GstVaapiEncPicture * picture)
|
|
|
|
{
|
|
|
|
GstVaapiEncMiscParam *misc = NULL;
|
|
|
|
VAEncMiscParameterRateControl *rate_control;
|
|
|
|
|
|
|
|
/* HRD params */
|
|
|
|
misc = GST_VAAPI_ENC_MISC_PARAM_NEW (HRD, encoder);
|
|
|
|
g_assert (misc);
|
|
|
|
if (!misc)
|
|
|
|
return FALSE;
|
|
|
|
fill_hrd_params (encoder, misc->data);
|
|
|
|
gst_vaapi_enc_picture_add_misc_param (picture, misc);
|
|
|
|
gst_vaapi_codec_object_replace (&misc, NULL);
|
|
|
|
|
|
|
|
/* RateControl params */
|
|
|
|
if (GST_VAAPI_ENCODER_RATE_CONTROL (encoder) == GST_VAAPI_RATECONTROL_CBR ||
|
|
|
|
GST_VAAPI_ENCODER_RATE_CONTROL (encoder) == GST_VAAPI_RATECONTROL_VBR) {
|
|
|
|
misc = GST_VAAPI_ENC_MISC_PARAM_NEW (RateControl, encoder);
|
|
|
|
g_assert (misc);
|
|
|
|
if (!misc)
|
|
|
|
return FALSE;
|
|
|
|
rate_control = misc->data;
|
|
|
|
memset (rate_control, 0, sizeof (VAEncMiscParameterRateControl));
|
2014-01-22 13:43:24 +00:00
|
|
|
rate_control->bits_per_second = encoder->bitrate_bits;
|
2014-01-21 15:05:22 +00:00
|
|
|
rate_control->target_percentage = 70;
|
2014-01-21 18:04:41 +00:00
|
|
|
rate_control->window_size = encoder->cpb_length;
|
2014-01-21 15:05:22 +00:00
|
|
|
rate_control->initial_qp = encoder->init_qp;
|
|
|
|
rate_control->min_qp = encoder->min_qp;
|
|
|
|
rate_control->basic_unit_size = 0;
|
|
|
|
gst_vaapi_enc_picture_add_misc_param (picture, misc);
|
|
|
|
gst_vaapi_codec_object_replace (&misc, NULL);
|
2015-07-02 18:00:14 +00:00
|
|
|
|
|
|
|
if (!encoder->view_idx) {
|
|
|
|
if ((GST_VAAPI_ENC_PICTURE_IS_IDR (picture)) &&
|
|
|
|
(GST_VAAPI_ENCODER_PACKED_HEADERS (encoder) &
|
|
|
|
VA_ENC_PACKED_HEADER_MISC) &&
|
2016-02-02 16:59:57 +00:00
|
|
|
!add_packed_sei_header (encoder, picture,
|
2015-07-02 18:00:14 +00:00
|
|
|
GST_VAAPI_H264_SEI_BUF_PERIOD | GST_VAAPI_H264_SEI_PIC_TIMING))
|
|
|
|
goto error_create_packed_sei_hdr;
|
|
|
|
|
|
|
|
else if (!GST_VAAPI_ENC_PICTURE_IS_IDR (picture) &&
|
2016-02-02 16:59:57 +00:00
|
|
|
(GST_VAAPI_ENCODER_PACKED_HEADERS (encoder) &
|
|
|
|
VA_ENC_PACKED_HEADER_MISC) &&
|
|
|
|
!add_packed_sei_header (encoder, picture,
|
|
|
|
GST_VAAPI_H264_SEI_PIC_TIMING))
|
2015-07-02 18:00:14 +00:00
|
|
|
goto error_create_packed_sei_hdr;
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
}
|
|
|
|
return TRUE;
|
2015-07-02 18:00:14 +00:00
|
|
|
|
|
|
|
error_create_packed_sei_hdr:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to create packed SEI header");
|
|
|
|
return FALSE;
|
|
|
|
}
|
2014-01-21 15:05:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Generates and submits PPS header accordingly into the bitstream */
|
2013-07-29 07:46:11 +00:00
|
|
|
static gboolean
|
2013-12-03 15:11:46 +00:00
|
|
|
ensure_picture (GstVaapiEncoderH264 * encoder, GstVaapiEncPicture * picture,
|
|
|
|
GstVaapiCodedBufferProxy * codedbuf_proxy, GstVaapiSurfaceProxy * surface)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2013-12-03 15:11:46 +00:00
|
|
|
GstVaapiCodedBuffer *const codedbuf =
|
2014-01-03 15:57:09 +00:00
|
|
|
GST_VAAPI_CODED_BUFFER_PROXY_BUFFER (codedbuf_proxy);
|
2014-02-17 07:51:43 +00:00
|
|
|
gboolean res = FALSE;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-05-06 21:12:39 +00:00
|
|
|
res = fill_picture (encoder, picture, codedbuf, surface);
|
2014-02-17 07:51:43 +00:00
|
|
|
|
|
|
|
if (!res)
|
2013-07-29 07:46:11 +00:00
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
if (picture->type == GST_VAAPI_PICTURE_TYPE_I &&
|
2016-02-02 16:59:57 +00:00
|
|
|
(GST_VAAPI_ENCODER_PACKED_HEADERS (encoder) &
|
|
|
|
VA_ENC_PACKED_HEADER_PICTURE)
|
2014-01-21 17:35:17 +00:00
|
|
|
&& !add_packed_picture_header (encoder, picture)) {
|
2013-07-29 07:46:11 +00:00
|
|
|
GST_ERROR ("set picture packed header failed");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Generates slice headers */
|
2013-07-29 07:46:11 +00:00
|
|
|
static gboolean
|
|
|
|
ensure_slices (GstVaapiEncoderH264 * encoder, GstVaapiEncPicture * picture)
|
|
|
|
{
|
|
|
|
GstVaapiEncoderH264Ref *reflist_0[16];
|
|
|
|
GstVaapiEncoderH264Ref *reflist_1[16];
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewRefPool *const ref_pool =
|
|
|
|
&encoder->ref_pools[encoder->view_idx];
|
2013-07-29 07:46:11 +00:00
|
|
|
guint reflist_0_count = 0, reflist_1_count = 0;
|
|
|
|
|
|
|
|
g_assert (picture);
|
|
|
|
|
|
|
|
if (picture->type != GST_VAAPI_PICTURE_TYPE_I &&
|
|
|
|
!reference_list_init (encoder, picture,
|
|
|
|
reflist_0, &reflist_0_count, reflist_1, &reflist_1_count)) {
|
|
|
|
GST_ERROR ("reference list reorder failed");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
g_assert (reflist_0_count + reflist_1_count <= ref_pool->max_ref_frames);
|
|
|
|
if (reflist_0_count > ref_pool->max_reflist0_count)
|
|
|
|
reflist_0_count = ref_pool->max_reflist0_count;
|
|
|
|
if (reflist_1_count > ref_pool->max_reflist1_count)
|
|
|
|
reflist_1_count = ref_pool->max_reflist1_count;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
if (!add_slice_headers (encoder, picture,
|
2013-07-29 07:46:11 +00:00
|
|
|
reflist_0, reflist_0_count, reflist_1, reflist_1_count))
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-01-22 13:43:24 +00:00
|
|
|
/* Normalizes bitrate (and CPB size) for HRD conformance */
|
|
|
|
static void
|
|
|
|
ensure_bitrate_hrd (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
|
|
|
GstVaapiEncoder *const base_encoder = GST_VAAPI_ENCODER_CAST (encoder);
|
2014-01-21 18:04:41 +00:00
|
|
|
guint bitrate, cpb_size;
|
2014-01-22 13:43:24 +00:00
|
|
|
|
|
|
|
if (!base_encoder->bitrate) {
|
|
|
|
encoder->bitrate_bits = 0;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Round down bitrate. This is a hard limit mandated by the user */
|
|
|
|
g_assert (SX_BITRATE >= 6);
|
|
|
|
bitrate = (base_encoder->bitrate * 1000) & ~((1U << SX_BITRATE) - 1);
|
2014-06-27 16:43:27 +00:00
|
|
|
if (bitrate != encoder->bitrate_bits) {
|
|
|
|
GST_DEBUG ("HRD bitrate: %u bits/sec", bitrate);
|
|
|
|
encoder->bitrate_bits = bitrate;
|
|
|
|
encoder->config_changed = TRUE;
|
|
|
|
}
|
2014-01-21 18:04:41 +00:00
|
|
|
|
|
|
|
/* Round up CPB size. This is an HRD compliance detail */
|
|
|
|
g_assert (SX_CPB_SIZE >= 4);
|
|
|
|
cpb_size = gst_util_uint64_scale (bitrate, encoder->cpb_length, 1000) &
|
|
|
|
~((1U << SX_CPB_SIZE) - 1);
|
2014-06-27 16:43:27 +00:00
|
|
|
if (cpb_size != encoder->cpb_length_bits) {
|
|
|
|
GST_DEBUG ("HRD CPB size: %u bits", cpb_size);
|
|
|
|
encoder->cpb_length_bits = cpb_size;
|
|
|
|
encoder->config_changed = TRUE;
|
|
|
|
}
|
2014-01-22 13:43:24 +00:00
|
|
|
}
|
|
|
|
|
2014-01-21 15:05:22 +00:00
|
|
|
/* Estimates a good enough bitrate if none was supplied */
|
|
|
|
static void
|
2014-01-10 11:01:51 +00:00
|
|
|
ensure_bitrate (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
|
|
|
GstVaapiEncoder *const base_encoder = GST_VAAPI_ENCODER_CAST (encoder);
|
|
|
|
|
2014-01-13 10:49:14 +00:00
|
|
|
/* Default compression: 48 bits per macroblock in "high-compression" mode */
|
2014-01-10 11:01:51 +00:00
|
|
|
switch (GST_VAAPI_ENCODER_RATE_CONTROL (encoder)) {
|
|
|
|
case GST_VAAPI_RATECONTROL_CBR:
|
|
|
|
case GST_VAAPI_RATECONTROL_VBR:
|
|
|
|
case GST_VAAPI_RATECONTROL_VBR_CONSTRAINED:
|
2014-01-13 10:49:14 +00:00
|
|
|
if (!base_encoder->bitrate) {
|
|
|
|
/* According to the literature and testing, CABAC entropy coding
|
|
|
|
mode could provide for +10% to +18% improvement in general,
|
|
|
|
thus estimating +15% here ; and using adaptive 8x8 transforms
|
|
|
|
in I-frames could bring up to +10% improvement. */
|
|
|
|
guint bits_per_mb = 48;
|
2016-08-10 11:29:45 +00:00
|
|
|
guint64 factor;
|
|
|
|
|
2014-01-13 10:49:14 +00:00
|
|
|
if (!encoder->use_cabac)
|
|
|
|
bits_per_mb += (bits_per_mb * 15) / 100;
|
|
|
|
if (!encoder->use_dct8x8)
|
|
|
|
bits_per_mb += (bits_per_mb * 10) / 100;
|
|
|
|
|
2016-08-10 11:29:45 +00:00
|
|
|
factor = encoder->mb_width * encoder->mb_height * bits_per_mb;
|
2014-01-13 10:49:14 +00:00
|
|
|
base_encoder->bitrate =
|
2016-07-21 15:38:40 +00:00
|
|
|
gst_util_uint64_scale (factor, GST_VAAPI_ENCODER_FPS_N (encoder),
|
|
|
|
GST_VAAPI_ENCODER_FPS_D (encoder)) / 1000;
|
2014-01-13 10:49:14 +00:00
|
|
|
GST_INFO ("target bitrate computed to %u kbps", base_encoder->bitrate);
|
|
|
|
}
|
2014-01-10 11:01:51 +00:00
|
|
|
break;
|
|
|
|
default:
|
|
|
|
base_encoder->bitrate = 0;
|
|
|
|
break;
|
|
|
|
}
|
2014-01-22 13:43:24 +00:00
|
|
|
ensure_bitrate_hrd (encoder);
|
2014-01-21 15:05:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Constructs profile and level information based on user-defined limits */
|
|
|
|
static GstVaapiEncoderStatus
|
|
|
|
ensure_profile_and_level (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
2014-06-27 16:43:27 +00:00
|
|
|
const GstVaapiProfile profile = encoder->profile;
|
|
|
|
const GstVaapiLevelH264 level = encoder->level;
|
|
|
|
|
2016-05-13 08:44:57 +00:00
|
|
|
if (!ensure_tuning (encoder))
|
|
|
|
GST_WARNING ("Failed to set some of the tuning option as expected! ");
|
2014-01-21 15:05:22 +00:00
|
|
|
|
|
|
|
if (!ensure_profile (encoder) || !ensure_profile_limits (encoder))
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_UNSUPPORTED_PROFILE;
|
|
|
|
|
|
|
|
/* Check HW constraints */
|
|
|
|
if (!ensure_hw_profile_limits (encoder))
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_UNSUPPORTED_PROFILE;
|
|
|
|
if (encoder->profile_idc > encoder->hw_max_profile_idc)
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_UNSUPPORTED_PROFILE;
|
2014-01-21 16:04:40 +00:00
|
|
|
|
|
|
|
/* Ensure bitrate if not set already and derive the right level to use */
|
|
|
|
ensure_bitrate (encoder);
|
|
|
|
if (!ensure_level (encoder))
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_OPERATION_FAILED;
|
2014-06-27 16:43:27 +00:00
|
|
|
|
|
|
|
if (encoder->profile != profile || encoder->level != level) {
|
|
|
|
GST_DEBUG ("selected %s profile at level %s",
|
|
|
|
gst_vaapi_utils_h264_get_profile_string (encoder->profile),
|
|
|
|
gst_vaapi_utils_h264_get_level_string (encoder->level));
|
|
|
|
encoder->config_changed = TRUE;
|
|
|
|
}
|
2014-01-21 15:05:22 +00:00
|
|
|
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
2014-01-10 11:01:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
reset_properties (GstVaapiEncoderH264 * encoder)
|
|
|
|
{
|
2014-01-10 12:23:48 +00:00
|
|
|
GstVaapiEncoder *const base_encoder = GST_VAAPI_ENCODER_CAST (encoder);
|
2014-02-17 03:10:26 +00:00
|
|
|
guint mb_size, i;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 12:23:48 +00:00
|
|
|
if (encoder->idr_period < base_encoder->keyframe_period)
|
|
|
|
encoder->idr_period = base_encoder->keyframe_period;
|
2014-01-21 14:23:01 +00:00
|
|
|
if (encoder->idr_period > MAX_IDR_PERIOD)
|
|
|
|
encoder->idr_period = MAX_IDR_PERIOD;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-06 16:46:40 +00:00
|
|
|
if (encoder->min_qp > encoder->init_qp ||
|
|
|
|
(GST_VAAPI_ENCODER_RATE_CONTROL (encoder) == GST_VAAPI_RATECONTROL_CQP &&
|
|
|
|
encoder->min_qp < encoder->init_qp))
|
2013-07-29 07:46:11 +00:00
|
|
|
encoder->min_qp = encoder->init_qp;
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
mb_size = encoder->mb_width * encoder->mb_height;
|
|
|
|
if (encoder->num_slices > (mb_size + 1) / 2)
|
|
|
|
encoder->num_slices = (mb_size + 1) / 2;
|
|
|
|
g_assert (encoder->num_slices);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
if (encoder->num_bframes > (base_encoder->keyframe_period + 1) / 2)
|
|
|
|
encoder->num_bframes = (base_encoder->keyframe_period + 1) / 2;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2016-05-13 08:45:20 +00:00
|
|
|
/* Workaround : vaapi-intel-driver doesn't have support for
|
|
|
|
* B-frame encode when utilizing low-power encode hardware block.
|
|
|
|
* So Disabling b-frame encoding in low-pwer encode.
|
|
|
|
*
|
|
|
|
* Fixme :We should query the VAConfigAttribEncMaxRefFrames
|
|
|
|
* instead of blindly disabling b-frame support and set b/p frame count,
|
|
|
|
* buffer pool size etc based on that.*/
|
|
|
|
if ((encoder->num_bframes > 0)
|
|
|
|
&& (encoder->entrypoint == GST_VAAPI_ENTRYPOINT_SLICE_ENCODE_LP)) {
|
|
|
|
GST_WARNING
|
|
|
|
("Disabling b-frame since the driver doesn't supporting it in low-power encode");
|
|
|
|
encoder->num_bframes = 0;
|
|
|
|
}
|
|
|
|
|
2016-07-05 18:07:15 +00:00
|
|
|
if (encoder->num_bframes > 0 && GST_VAAPI_ENCODER_FPS_N (encoder) > 0)
|
|
|
|
encoder->cts_offset = gst_util_uint64_scale (GST_SECOND,
|
|
|
|
GST_VAAPI_ENCODER_FPS_D (encoder), GST_VAAPI_ENCODER_FPS_N (encoder));
|
2013-07-29 07:46:11 +00:00
|
|
|
else
|
2016-10-08 12:33:59 +00:00
|
|
|
encoder->cts_offset = 0;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/* init max_frame_num, max_poc */
|
2014-01-10 13:05:40 +00:00
|
|
|
encoder->log2_max_frame_num =
|
|
|
|
h264_get_log2_max_frame_num (encoder->idr_period);
|
2013-07-29 07:46:11 +00:00
|
|
|
g_assert (encoder->log2_max_frame_num >= 4);
|
|
|
|
encoder->max_frame_num = (1 << encoder->log2_max_frame_num);
|
|
|
|
encoder->log2_max_pic_order_cnt = encoder->log2_max_frame_num + 1;
|
|
|
|
encoder->max_pic_order_cnt = (1 << encoder->log2_max_pic_order_cnt);
|
|
|
|
encoder->idr_num = 0;
|
2014-02-17 03:10:26 +00:00
|
|
|
|
|
|
|
for (i = 0; i < encoder->num_views; i++) {
|
|
|
|
GstVaapiH264ViewRefPool *const ref_pool = &encoder->ref_pools[i];
|
2016-02-01 13:02:13 +00:00
|
|
|
GstVaapiH264ViewReorderPool *const reorder_pool =
|
|
|
|
&encoder->reorder_pools[i];
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
ref_pool->max_reflist0_count = 1;
|
|
|
|
ref_pool->max_reflist1_count = encoder->num_bframes > 0;
|
|
|
|
ref_pool->max_ref_frames = ref_pool->max_reflist0_count
|
|
|
|
+ ref_pool->max_reflist1_count;
|
|
|
|
|
|
|
|
reorder_pool->frame_index = 0;
|
|
|
|
}
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static GstVaapiEncoderStatus
|
2014-01-10 13:05:40 +00:00
|
|
|
gst_vaapi_encoder_h264_encode (GstVaapiEncoder * base_encoder,
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncPicture * picture, GstVaapiCodedBufferProxy * codedbuf)
|
|
|
|
{
|
2014-01-10 13:05:40 +00:00
|
|
|
GstVaapiEncoderH264 *const encoder =
|
|
|
|
GST_VAAPI_ENCODER_H264_CAST (base_encoder);
|
2013-12-03 16:04:43 +00:00
|
|
|
GstVaapiEncoderStatus ret = GST_VAAPI_ENCODER_STATUS_ERROR_UNKNOWN;
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiSurfaceProxy *reconstruct = NULL;
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
reconstruct = gst_vaapi_encoder_create_surface (base_encoder);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
g_assert (GST_VAAPI_SURFACE_PROXY_SURFACE (reconstruct));
|
|
|
|
|
|
|
|
if (!ensure_sequence (encoder, picture))
|
|
|
|
goto error;
|
2014-01-21 15:05:22 +00:00
|
|
|
if (!ensure_misc_params (encoder, picture))
|
2013-07-29 07:46:11 +00:00
|
|
|
goto error;
|
2014-01-21 15:05:22 +00:00
|
|
|
if (!ensure_picture (encoder, picture, codedbuf, reconstruct))
|
2013-07-29 07:46:11 +00:00
|
|
|
goto error;
|
|
|
|
if (!ensure_slices (encoder, picture))
|
|
|
|
goto error;
|
|
|
|
if (!gst_vaapi_enc_picture_encode (picture))
|
|
|
|
goto error;
|
|
|
|
|
|
|
|
if (!reference_list_update (encoder, picture, reconstruct))
|
|
|
|
goto error;
|
|
|
|
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
|
|
|
error:
|
|
|
|
if (reconstruct)
|
|
|
|
gst_vaapi_encoder_release_surface (GST_VAAPI_ENCODER (encoder),
|
|
|
|
reconstruct);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
static GstVaapiEncoderStatus
|
2014-01-10 13:05:40 +00:00
|
|
|
gst_vaapi_encoder_h264_flush (GstVaapiEncoder * base_encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-01-10 13:05:40 +00:00
|
|
|
GstVaapiEncoderH264 *const encoder =
|
|
|
|
GST_VAAPI_ENCODER_H264_CAST (base_encoder);
|
2014-02-17 03:10:26 +00:00
|
|
|
GstVaapiH264ViewReorderPool *reorder_pool;
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncPicture *pic;
|
2014-02-17 03:10:26 +00:00
|
|
|
guint i;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
for (i = 0; i < encoder->num_views; i++) {
|
|
|
|
reorder_pool = &encoder->reorder_pools[i];
|
|
|
|
reorder_pool->frame_index = 0;
|
|
|
|
reorder_pool->cur_frame_num = 0;
|
|
|
|
reorder_pool->cur_present_index = 0;
|
|
|
|
|
|
|
|
while (!g_queue_is_empty (&reorder_pool->reorder_frame_list)) {
|
|
|
|
pic = (GstVaapiEncPicture *)
|
|
|
|
g_queue_pop_head (&reorder_pool->reorder_frame_list);
|
|
|
|
gst_vaapi_enc_picture_unref (pic);
|
|
|
|
}
|
|
|
|
g_queue_clear (&reorder_pool->reorder_frame_list);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
|
|
|
}
|
|
|
|
|
2014-01-10 13:46:15 +00:00
|
|
|
/* Generate "codec-data" buffer */
|
2013-07-29 07:46:11 +00:00
|
|
|
static GstVaapiEncoderStatus
|
2014-01-10 13:46:15 +00:00
|
|
|
gst_vaapi_encoder_h264_get_codec_data (GstVaapiEncoder * base_encoder,
|
|
|
|
GstBuffer ** out_buffer_ptr)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-01-10 13:46:15 +00:00
|
|
|
GstVaapiEncoderH264 *const encoder =
|
|
|
|
GST_VAAPI_ENCODER_H264_CAST (base_encoder);
|
2013-07-29 07:46:11 +00:00
|
|
|
const guint32 configuration_version = 0x01;
|
2014-01-10 13:46:15 +00:00
|
|
|
const guint32 nal_length_size = 4;
|
|
|
|
guint8 profile_idc, profile_comp, level_idc;
|
2013-07-29 07:46:11 +00:00
|
|
|
GstMapInfo sps_info, pps_info;
|
2014-01-21 14:28:34 +00:00
|
|
|
GstBitWriter bs;
|
2014-01-10 13:46:15 +00:00
|
|
|
GstBuffer *buffer;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
if (!encoder->sps_data || !encoder->pps_data)
|
2013-12-03 16:04:43 +00:00
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_INVALID_HEADER;
|
2014-01-10 13:46:15 +00:00
|
|
|
if (gst_buffer_get_size (encoder->sps_data) < 4)
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_INVALID_HEADER;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
if (!gst_buffer_map (encoder->sps_data, &sps_info, GST_MAP_READ))
|
2014-01-10 13:46:15 +00:00
|
|
|
goto error_map_sps_buffer;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:46:15 +00:00
|
|
|
if (!gst_buffer_map (encoder->pps_data, &pps_info, GST_MAP_READ))
|
|
|
|
goto error_map_pps_buffer;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:46:15 +00:00
|
|
|
/* skip sps_data[0], which is the nal_unit_type */
|
|
|
|
profile_idc = sps_info.data[1];
|
|
|
|
profile_comp = sps_info.data[2];
|
|
|
|
level_idc = sps_info.data[3];
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:46:15 +00:00
|
|
|
/* Header */
|
2014-01-21 14:28:34 +00:00
|
|
|
gst_bit_writer_init (&bs, (sps_info.size + pps_info.size + 64) * 8);
|
|
|
|
WRITE_UINT32 (&bs, configuration_version, 8);
|
|
|
|
WRITE_UINT32 (&bs, profile_idc, 8);
|
|
|
|
WRITE_UINT32 (&bs, profile_comp, 8);
|
|
|
|
WRITE_UINT32 (&bs, level_idc, 8);
|
|
|
|
WRITE_UINT32 (&bs, 0x3f, 6); /* 111111 */
|
|
|
|
WRITE_UINT32 (&bs, nal_length_size - 1, 2);
|
|
|
|
WRITE_UINT32 (&bs, 0x07, 3); /* 111 */
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:46:15 +00:00
|
|
|
/* Write SPS */
|
2014-01-21 14:28:34 +00:00
|
|
|
WRITE_UINT32 (&bs, 1, 5); /* SPS count = 1 */
|
|
|
|
g_assert (GST_BIT_WRITER_BIT_SIZE (&bs) % 8 == 0);
|
|
|
|
WRITE_UINT32 (&bs, sps_info.size, 16);
|
|
|
|
gst_bit_writer_put_bytes (&bs, sps_info.data, sps_info.size);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:46:15 +00:00
|
|
|
/* Write PPS */
|
2014-01-21 14:28:34 +00:00
|
|
|
WRITE_UINT32 (&bs, 1, 8); /* PPS count = 1 */
|
|
|
|
WRITE_UINT32 (&bs, pps_info.size, 16);
|
|
|
|
gst_bit_writer_put_bytes (&bs, pps_info.data, pps_info.size);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
gst_buffer_unmap (encoder->pps_data, &pps_info);
|
|
|
|
gst_buffer_unmap (encoder->sps_data, &sps_info);
|
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
buffer = gst_buffer_new_wrapped (GST_BIT_WRITER_DATA (&bs),
|
|
|
|
GST_BIT_WRITER_BIT_SIZE (&bs) / 8);
|
2014-01-10 13:46:15 +00:00
|
|
|
if (!buffer)
|
|
|
|
goto error_alloc_buffer;
|
|
|
|
*out_buffer_ptr = buffer;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-21 14:28:34 +00:00
|
|
|
gst_bit_writer_clear (&bs, FALSE);
|
2014-01-10 13:46:15 +00:00
|
|
|
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-01-10 13:46:15 +00:00
|
|
|
/* ERRORS */
|
2014-01-21 14:28:34 +00:00
|
|
|
bs_error:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to write codec-data");
|
|
|
|
gst_buffer_unmap (encoder->sps_data, &sps_info);
|
|
|
|
gst_buffer_unmap (encoder->pps_data, &pps_info);
|
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
|
|
|
return FALSE;
|
|
|
|
}
|
2014-01-10 13:46:15 +00:00
|
|
|
error_map_sps_buffer:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to map SPS packed header");
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_ALLOCATION_FAILED;
|
|
|
|
}
|
|
|
|
error_map_pps_buffer:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to map PPS packed header");
|
|
|
|
gst_buffer_unmap (encoder->sps_data, &sps_info);
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_ALLOCATION_FAILED;
|
|
|
|
}
|
|
|
|
error_alloc_buffer:
|
|
|
|
{
|
|
|
|
GST_ERROR ("failed to allocate codec-data buffer");
|
2014-01-21 14:28:34 +00:00
|
|
|
gst_bit_writer_clear (&bs, TRUE);
|
2014-01-10 13:46:15 +00:00
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_ALLOCATION_FAILED;
|
|
|
|
}
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static GstVaapiEncoderStatus
|
2014-01-10 13:05:40 +00:00
|
|
|
gst_vaapi_encoder_h264_reordering (GstVaapiEncoder * base_encoder,
|
2013-12-04 16:55:18 +00:00
|
|
|
GstVideoCodecFrame * frame, GstVaapiEncPicture ** output)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-01-10 13:05:40 +00:00
|
|
|
GstVaapiEncoderH264 *const encoder =
|
|
|
|
GST_VAAPI_ENCODER_H264_CAST (base_encoder);
|
2014-02-17 07:51:43 +00:00
|
|
|
GstVaapiH264ViewReorderPool *reorder_pool = NULL;
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncPicture *picture;
|
|
|
|
gboolean is_idr = FALSE;
|
|
|
|
|
|
|
|
*output = NULL;
|
|
|
|
|
2014-02-17 07:51:43 +00:00
|
|
|
/* encoding views alternatively for MVC */
|
|
|
|
if (encoder->is_mvc) {
|
2015-06-12 15:39:31 +00:00
|
|
|
/* FIXME: Use first-in-bundle flag on buffers to reset view idx? */
|
2014-02-17 07:51:43 +00:00
|
|
|
if (frame)
|
2015-02-24 15:14:33 +00:00
|
|
|
encoder->view_idx = frame->system_frame_number % encoder->num_views;
|
2014-02-17 07:51:43 +00:00
|
|
|
else
|
2015-02-24 15:14:33 +00:00
|
|
|
encoder->view_idx = (encoder->view_idx + 1) % encoder->num_views;
|
2014-02-17 07:51:43 +00:00
|
|
|
}
|
|
|
|
reorder_pool = &encoder->reorder_pools[encoder->view_idx];
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
if (!frame) {
|
2014-02-17 03:10:26 +00:00
|
|
|
if (reorder_pool->reorder_state != GST_VAAPI_ENC_H264_REORD_DUMP_FRAMES)
|
2013-12-03 16:04:43 +00:00
|
|
|
return GST_VAAPI_ENCODER_STATUS_NO_SURFACE;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/* reorder_state = GST_VAAPI_ENC_H264_REORD_DUMP_FRAMES
|
|
|
|
dump B frames from queue, sometime, there may also have P frame or I frame */
|
2014-01-10 13:05:40 +00:00
|
|
|
g_assert (encoder->num_bframes > 0);
|
2014-02-17 03:10:26 +00:00
|
|
|
g_return_val_if_fail (!g_queue_is_empty (&reorder_pool->reorder_frame_list),
|
2013-12-03 16:04:43 +00:00
|
|
|
GST_VAAPI_ENCODER_STATUS_ERROR_UNKNOWN);
|
2014-02-17 03:10:26 +00:00
|
|
|
picture = g_queue_pop_head (&reorder_pool->reorder_frame_list);
|
2013-07-29 07:46:11 +00:00
|
|
|
g_assert (picture);
|
2014-02-17 03:10:26 +00:00
|
|
|
if (g_queue_is_empty (&reorder_pool->reorder_frame_list)) {
|
|
|
|
reorder_pool->reorder_state = GST_VAAPI_ENC_H264_REORD_WAIT_FRAMES;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
goto end;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* new frame coming */
|
2014-05-06 21:12:39 +00:00
|
|
|
picture = GST_VAAPI_ENC_PICTURE_NEW (H264, encoder, frame);
|
2013-07-29 07:46:11 +00:00
|
|
|
if (!picture) {
|
|
|
|
GST_WARNING ("create H264 picture failed, frame timestamp:%"
|
|
|
|
GST_TIME_FORMAT, GST_TIME_ARGS (frame->pts));
|
2013-12-03 16:04:43 +00:00
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_ALLOCATION_FAILED;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
2014-02-17 03:10:26 +00:00
|
|
|
++reorder_pool->cur_present_index;
|
|
|
|
picture->poc = ((reorder_pool->cur_present_index * 2) %
|
2013-07-29 07:46:11 +00:00
|
|
|
encoder->max_pic_order_cnt);
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
is_idr = (reorder_pool->frame_index == 0 ||
|
|
|
|
reorder_pool->frame_index >= encoder->idr_period);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/* check key frames */
|
|
|
|
if (is_idr || GST_VIDEO_CODEC_FRAME_IS_FORCE_KEYFRAME (frame) ||
|
2014-02-17 03:10:26 +00:00
|
|
|
(reorder_pool->frame_index %
|
|
|
|
GST_VAAPI_ENCODER_KEYFRAME_PERIOD (encoder)) == 0) {
|
|
|
|
++reorder_pool->cur_frame_num;
|
|
|
|
++reorder_pool->frame_index;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
/* b frame enabled, check queue of reorder_frame_list */
|
2014-01-10 13:05:40 +00:00
|
|
|
if (encoder->num_bframes
|
2014-02-17 03:10:26 +00:00
|
|
|
&& !g_queue_is_empty (&reorder_pool->reorder_frame_list)) {
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncPicture *p_pic;
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
p_pic = g_queue_pop_tail (&reorder_pool->reorder_frame_list);
|
2014-01-21 15:05:22 +00:00
|
|
|
set_p_frame (p_pic, encoder);
|
2014-02-17 03:10:26 +00:00
|
|
|
g_queue_foreach (&reorder_pool->reorder_frame_list,
|
2014-01-21 15:05:22 +00:00
|
|
|
(GFunc) set_b_frame, encoder);
|
2014-02-17 03:10:26 +00:00
|
|
|
++reorder_pool->cur_frame_num;
|
2014-01-21 15:05:22 +00:00
|
|
|
set_key_frame (picture, encoder, is_idr);
|
2014-02-17 03:10:26 +00:00
|
|
|
g_queue_push_tail (&reorder_pool->reorder_frame_list, picture);
|
2013-07-29 07:46:11 +00:00
|
|
|
picture = p_pic;
|
2014-02-17 03:10:26 +00:00
|
|
|
reorder_pool->reorder_state = GST_VAAPI_ENC_H264_REORD_DUMP_FRAMES;
|
2013-07-29 07:46:11 +00:00
|
|
|
} else { /* no b frames in queue */
|
2014-01-21 15:05:22 +00:00
|
|
|
set_key_frame (picture, encoder, is_idr);
|
2014-02-17 03:10:26 +00:00
|
|
|
g_assert (g_queue_is_empty (&reorder_pool->reorder_frame_list));
|
2014-01-10 13:05:40 +00:00
|
|
|
if (encoder->num_bframes)
|
2014-02-17 03:10:26 +00:00
|
|
|
reorder_pool->reorder_state = GST_VAAPI_ENC_H264_REORD_WAIT_FRAMES;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
goto end;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* new p/b frames coming */
|
2014-02-17 03:10:26 +00:00
|
|
|
++reorder_pool->frame_index;
|
|
|
|
if (reorder_pool->reorder_state == GST_VAAPI_ENC_H264_REORD_WAIT_FRAMES &&
|
|
|
|
g_queue_get_length (&reorder_pool->reorder_frame_list) <
|
2014-01-10 13:05:40 +00:00
|
|
|
encoder->num_bframes) {
|
2014-02-17 03:10:26 +00:00
|
|
|
g_queue_push_tail (&reorder_pool->reorder_frame_list, picture);
|
2013-12-03 16:04:43 +00:00
|
|
|
return GST_VAAPI_ENCODER_STATUS_NO_SURFACE;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
++reorder_pool->cur_frame_num;
|
2014-01-21 15:05:22 +00:00
|
|
|
set_p_frame (picture, encoder);
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
if (reorder_pool->reorder_state == GST_VAAPI_ENC_H264_REORD_WAIT_FRAMES) {
|
|
|
|
g_queue_foreach (&reorder_pool->reorder_frame_list, (GFunc) set_b_frame,
|
2013-07-29 07:46:11 +00:00
|
|
|
encoder);
|
2014-02-17 03:10:26 +00:00
|
|
|
reorder_pool->reorder_state = GST_VAAPI_ENC_H264_REORD_DUMP_FRAMES;
|
|
|
|
g_assert (!g_queue_is_empty (&reorder_pool->reorder_frame_list));
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
end:
|
|
|
|
g_assert (picture);
|
2014-01-14 11:01:11 +00:00
|
|
|
frame = picture->frame;
|
2013-07-29 07:46:11 +00:00
|
|
|
if (GST_CLOCK_TIME_IS_VALID (frame->pts))
|
|
|
|
frame->pts += encoder->cts_offset;
|
|
|
|
*output = picture;
|
|
|
|
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
|
|
|
}
|
|
|
|
|
2014-01-10 16:02:44 +00:00
|
|
|
static GstVaapiEncoderStatus
|
2014-01-10 11:01:51 +00:00
|
|
|
set_context_info (GstVaapiEncoder * base_encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-01-10 13:05:40 +00:00
|
|
|
GstVaapiEncoderH264 *const encoder =
|
|
|
|
GST_VAAPI_ENCODER_H264_CAST (base_encoder);
|
2014-01-10 11:01:51 +00:00
|
|
|
GstVideoInfo *const vip = GST_VAAPI_ENCODER_VIDEO_INFO (encoder);
|
2013-12-04 17:48:35 +00:00
|
|
|
const guint DEFAULT_SURFACES_COUNT = 3;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2013-12-05 17:13:54 +00:00
|
|
|
/* Maximum sizes for common headers (in bits) */
|
2014-01-03 15:57:09 +00:00
|
|
|
enum
|
|
|
|
{
|
2013-12-05 17:13:54 +00:00
|
|
|
MAX_SPS_HDR_SIZE = 16473,
|
|
|
|
MAX_VUI_PARAMS_SIZE = 210,
|
|
|
|
MAX_HRD_PARAMS_SIZE = 4103,
|
|
|
|
MAX_PPS_HDR_SIZE = 101,
|
|
|
|
MAX_SLICE_HDR_SIZE = 397 + 2572 + 6670 + 2402,
|
|
|
|
};
|
|
|
|
|
2014-01-10 16:02:44 +00:00
|
|
|
if (!ensure_hw_profile (encoder))
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_UNSUPPORTED_PROFILE;
|
|
|
|
|
2014-01-10 11:01:51 +00:00
|
|
|
base_encoder->num_ref_frames =
|
2014-02-17 07:51:43 +00:00
|
|
|
((encoder->num_bframes ? 2 : 1) + DEFAULT_SURFACES_COUNT)
|
2014-05-06 21:08:33 +00:00
|
|
|
* encoder->num_views;
|
2013-12-04 18:10:13 +00:00
|
|
|
|
|
|
|
/* Only YUV 4:2:0 formats are supported for now. This means that we
|
|
|
|
have a limit of 3200 bits per macroblock. */
|
|
|
|
/* XXX: check profile and compute RawMbBits */
|
2014-01-10 11:01:51 +00:00
|
|
|
base_encoder->codedbuf_size = (GST_ROUND_UP_16 (vip->width) *
|
|
|
|
GST_ROUND_UP_16 (vip->height) / 256) * 400;
|
2013-12-05 17:13:54 +00:00
|
|
|
|
|
|
|
/* Account for SPS header */
|
|
|
|
/* XXX: exclude scaling lists, MVC/SVC extensions */
|
|
|
|
base_encoder->codedbuf_size += 4 + GST_ROUND_UP_8 (MAX_SPS_HDR_SIZE +
|
|
|
|
MAX_VUI_PARAMS_SIZE + 2 * MAX_HRD_PARAMS_SIZE) / 8;
|
|
|
|
|
|
|
|
/* Account for PPS header */
|
|
|
|
/* XXX: exclude slice groups, scaling lists, MVC/SVC extensions */
|
|
|
|
base_encoder->codedbuf_size += 4 + GST_ROUND_UP_8 (MAX_PPS_HDR_SIZE) / 8;
|
|
|
|
|
2014-01-12 21:14:11 +00:00
|
|
|
/* Account for slice header */
|
|
|
|
base_encoder->codedbuf_size += encoder->num_slices * (4 +
|
2013-12-05 17:13:54 +00:00
|
|
|
GST_ROUND_UP_8 (MAX_SLICE_HDR_SIZE) / 8);
|
2014-01-10 16:02:44 +00:00
|
|
|
|
2016-05-11 09:05:36 +00:00
|
|
|
base_encoder->context_info.entrypoint = encoder->entrypoint;
|
|
|
|
|
2014-01-10 16:02:44 +00:00
|
|
|
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-10 09:54:22 +00:00
|
|
|
static GstVaapiEncoderStatus
|
|
|
|
gst_vaapi_encoder_h264_reconfigure (GstVaapiEncoder * base_encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-01-10 09:54:22 +00:00
|
|
|
GstVaapiEncoderH264 *const encoder =
|
|
|
|
GST_VAAPI_ENCODER_H264_CAST (base_encoder);
|
2015-06-12 15:39:31 +00:00
|
|
|
GstVideoInfo *const vip = GST_VAAPI_ENCODER_VIDEO_INFO (encoder);
|
2014-01-12 21:24:04 +00:00
|
|
|
GstVaapiEncoderStatus status;
|
2014-06-27 16:43:27 +00:00
|
|
|
guint mb_width, mb_height;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-06-27 16:43:27 +00:00
|
|
|
mb_width = (GST_VAAPI_ENCODER_WIDTH (encoder) + 15) / 16;
|
|
|
|
mb_height = (GST_VAAPI_ENCODER_HEIGHT (encoder) + 15) / 16;
|
|
|
|
if (mb_width != encoder->mb_width || mb_height != encoder->mb_height) {
|
|
|
|
GST_DEBUG ("resolution: %dx%d", GST_VAAPI_ENCODER_WIDTH (encoder),
|
|
|
|
GST_VAAPI_ENCODER_HEIGHT (encoder));
|
|
|
|
encoder->mb_width = mb_width;
|
|
|
|
encoder->mb_height = mb_height;
|
|
|
|
encoder->config_changed = TRUE;
|
|
|
|
}
|
2014-01-10 13:05:40 +00:00
|
|
|
|
2015-06-12 15:39:31 +00:00
|
|
|
/* Take number of MVC views from input caps if provided */
|
2016-02-02 16:59:57 +00:00
|
|
|
if (GST_VIDEO_INFO_MULTIVIEW_MODE (vip) ==
|
|
|
|
GST_VIDEO_MULTIVIEW_MODE_FRAME_BY_FRAME
|
|
|
|
|| GST_VIDEO_INFO_MULTIVIEW_MODE (vip) ==
|
|
|
|
GST_VIDEO_MULTIVIEW_MODE_MULTIVIEW_FRAME_BY_FRAME)
|
2015-06-12 15:39:31 +00:00
|
|
|
encoder->num_views = GST_VIDEO_INFO_VIEWS (vip);
|
|
|
|
|
|
|
|
encoder->is_mvc = encoder->num_views > 1;
|
|
|
|
|
2014-01-12 21:24:04 +00:00
|
|
|
status = ensure_profile_and_level (encoder);
|
|
|
|
if (status != GST_VAAPI_ENCODER_STATUS_SUCCESS)
|
|
|
|
return status;
|
|
|
|
|
2014-01-10 11:01:51 +00:00
|
|
|
reset_properties (encoder);
|
2014-01-10 16:02:44 +00:00
|
|
|
return set_context_info (base_encoder);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static gboolean
|
2014-01-10 13:05:40 +00:00
|
|
|
gst_vaapi_encoder_h264_init (GstVaapiEncoder * base_encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2014-01-10 13:05:40 +00:00
|
|
|
GstVaapiEncoderH264 *const encoder =
|
|
|
|
GST_VAAPI_ENCODER_H264_CAST (base_encoder);
|
2014-02-17 03:10:26 +00:00
|
|
|
guint32 i;
|
|
|
|
|
2016-05-13 08:44:57 +00:00
|
|
|
/* Default encoding entrypoint */
|
|
|
|
encoder->entrypoint = GST_VAAPI_ENTRYPOINT_SLICE_ENCODE;
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
/* Multi-view coding information */
|
|
|
|
encoder->is_mvc = FALSE;
|
|
|
|
encoder->num_views = 1;
|
|
|
|
encoder->view_idx = 0;
|
2015-02-24 15:14:33 +00:00
|
|
|
memset (encoder->view_ids, 0, sizeof (encoder->view_ids));
|
2014-02-17 03:10:26 +00:00
|
|
|
|
|
|
|
/* re-ordering list initialize */
|
|
|
|
for (i = 0; i < MAX_NUM_VIEWS; i++) {
|
|
|
|
GstVaapiH264ViewReorderPool *const reorder_pool =
|
|
|
|
&encoder->reorder_pools[i];
|
|
|
|
g_queue_init (&reorder_pool->reorder_frame_list);
|
|
|
|
reorder_pool->reorder_state = GST_VAAPI_ENC_H264_REORD_NONE;
|
|
|
|
reorder_pool->frame_index = 0;
|
|
|
|
reorder_pool->cur_frame_num = 0;
|
|
|
|
reorder_pool->cur_present_index = 0;
|
|
|
|
}
|
2013-07-29 07:46:11 +00:00
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
/* reference list info initialize */
|
|
|
|
for (i = 0; i < MAX_NUM_VIEWS; i++) {
|
|
|
|
GstVaapiH264ViewRefPool *const ref_pool = &encoder->ref_pools[i];
|
|
|
|
g_queue_init (&ref_pool->ref_list);
|
|
|
|
ref_pool->max_ref_frames = 0;
|
|
|
|
ref_pool->max_reflist0_count = 1;
|
|
|
|
ref_pool->max_reflist1_count = 1;
|
|
|
|
}
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
2014-01-10 13:05:40 +00:00
|
|
|
gst_vaapi_encoder_h264_finalize (GstVaapiEncoder * base_encoder)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
|
|
|
/*free private buffers */
|
2014-01-10 13:05:40 +00:00
|
|
|
GstVaapiEncoderH264 *const encoder =
|
|
|
|
GST_VAAPI_ENCODER_H264_CAST (base_encoder);
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncPicture *pic;
|
|
|
|
GstVaapiEncoderH264Ref *ref;
|
2014-02-17 03:10:26 +00:00
|
|
|
guint32 i;
|
2013-07-29 07:46:11 +00:00
|
|
|
|
|
|
|
gst_buffer_replace (&encoder->sps_data, NULL);
|
2014-05-06 21:09:19 +00:00
|
|
|
gst_buffer_replace (&encoder->subset_sps_data, NULL);
|
2013-07-29 07:46:11 +00:00
|
|
|
gst_buffer_replace (&encoder->pps_data, NULL);
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
/* reference list info de-init */
|
|
|
|
for (i = 0; i < MAX_NUM_VIEWS; i++) {
|
|
|
|
GstVaapiH264ViewRefPool *const ref_pool = &encoder->ref_pools[i];
|
|
|
|
while (!g_queue_is_empty (&ref_pool->ref_list)) {
|
|
|
|
ref = (GstVaapiEncoderH264Ref *) g_queue_pop_head (&ref_pool->ref_list);
|
|
|
|
reference_pic_free (encoder, ref);
|
|
|
|
}
|
|
|
|
g_queue_clear (&ref_pool->ref_list);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-02-17 03:10:26 +00:00
|
|
|
/* re-ordering list initialize */
|
|
|
|
for (i = 0; i < MAX_NUM_VIEWS; i++) {
|
|
|
|
GstVaapiH264ViewReorderPool *const reorder_pool =
|
|
|
|
&encoder->reorder_pools[i];
|
|
|
|
while (!g_queue_is_empty (&reorder_pool->reorder_frame_list)) {
|
|
|
|
pic = (GstVaapiEncPicture *)
|
|
|
|
g_queue_pop_head (&reorder_pool->reorder_frame_list);
|
|
|
|
gst_vaapi_enc_picture_unref (pic);
|
|
|
|
}
|
|
|
|
g_queue_clear (&reorder_pool->reorder_frame_list);
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-06 16:46:40 +00:00
|
|
|
static GstVaapiEncoderStatus
|
|
|
|
gst_vaapi_encoder_h264_set_property (GstVaapiEncoder * base_encoder,
|
|
|
|
gint prop_id, const GValue * value)
|
|
|
|
{
|
2014-01-10 13:05:40 +00:00
|
|
|
GstVaapiEncoderH264 *const encoder =
|
|
|
|
GST_VAAPI_ENCODER_H264_CAST (base_encoder);
|
2014-01-06 16:46:40 +00:00
|
|
|
|
|
|
|
switch (prop_id) {
|
|
|
|
case GST_VAAPI_ENCODER_H264_PROP_MAX_BFRAMES:
|
2014-01-10 13:05:40 +00:00
|
|
|
encoder->num_bframes = g_value_get_uint (value);
|
2014-01-06 16:46:40 +00:00
|
|
|
break;
|
|
|
|
case GST_VAAPI_ENCODER_H264_PROP_INIT_QP:
|
|
|
|
encoder->init_qp = g_value_get_uint (value);
|
|
|
|
break;
|
|
|
|
case GST_VAAPI_ENCODER_H264_PROP_MIN_QP:
|
|
|
|
encoder->min_qp = g_value_get_uint (value);
|
|
|
|
break;
|
|
|
|
case GST_VAAPI_ENCODER_H264_PROP_NUM_SLICES:
|
2014-01-10 13:05:40 +00:00
|
|
|
encoder->num_slices = g_value_get_uint (value);
|
2014-01-06 16:46:40 +00:00
|
|
|
break;
|
2013-12-13 09:36:08 +00:00
|
|
|
case GST_VAAPI_ENCODER_H264_PROP_CABAC:
|
|
|
|
encoder->use_cabac = g_value_get_boolean (value);
|
|
|
|
break;
|
|
|
|
case GST_VAAPI_ENCODER_H264_PROP_DCT8X8:
|
|
|
|
encoder->use_dct8x8 = g_value_get_boolean (value);
|
|
|
|
break;
|
2014-01-21 18:04:41 +00:00
|
|
|
case GST_VAAPI_ENCODER_H264_PROP_CPB_LENGTH:
|
|
|
|
encoder->cpb_length = g_value_get_uint (value);
|
|
|
|
break;
|
2014-02-17 07:51:43 +00:00
|
|
|
case GST_VAAPI_ENCODER_H264_PROP_NUM_VIEWS:
|
|
|
|
encoder->num_views = g_value_get_uint (value);
|
|
|
|
break;
|
2016-02-02 16:59:57 +00:00
|
|
|
case GST_VAAPI_ENCODER_H264_PROP_VIEW_IDS:{
|
|
|
|
guint i;
|
|
|
|
GValueArray *view_ids = g_value_get_boxed (value);
|
|
|
|
|
|
|
|
if (view_ids == NULL) {
|
|
|
|
for (i = 0; i < encoder->num_views; i++)
|
|
|
|
encoder->view_ids[i] = i;
|
|
|
|
} else {
|
|
|
|
g_assert (view_ids->n_values <= encoder->num_views);
|
|
|
|
|
|
|
|
for (i = 0; i < encoder->num_views; i++) {
|
|
|
|
GValue *val = g_value_array_get_nth (view_ids, i);
|
|
|
|
encoder->view_ids[i] = g_value_get_uint (val);
|
2015-02-24 15:14:33 +00:00
|
|
|
}
|
2016-02-02 16:59:57 +00:00
|
|
|
}
|
2015-02-24 15:14:33 +00:00
|
|
|
break;
|
|
|
|
}
|
2014-01-06 16:46:40 +00:00
|
|
|
default:
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_ERROR_INVALID_PARAMETER;
|
|
|
|
}
|
|
|
|
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
|
|
|
}
|
|
|
|
|
2014-01-06 14:10:36 +00:00
|
|
|
GST_VAAPI_ENCODER_DEFINE_CLASS_DATA (H264);
|
|
|
|
|
2013-07-29 07:46:11 +00:00
|
|
|
static inline const GstVaapiEncoderClass *
|
2014-01-03 15:57:09 +00:00
|
|
|
gst_vaapi_encoder_h264_class (void)
|
2013-07-29 07:46:11 +00:00
|
|
|
{
|
2013-12-04 10:54:40 +00:00
|
|
|
static const GstVaapiEncoderClass GstVaapiEncoderH264Class = {
|
|
|
|
GST_VAAPI_ENCODER_CLASS_INIT (H264, h264),
|
2014-01-06 16:46:40 +00:00
|
|
|
.set_property = gst_vaapi_encoder_h264_set_property,
|
2013-12-04 10:54:40 +00:00
|
|
|
.get_codec_data = gst_vaapi_encoder_h264_get_codec_data
|
|
|
|
};
|
|
|
|
return &GstVaapiEncoderH264Class;
|
2013-07-29 07:46:11 +00:00
|
|
|
}
|
|
|
|
|
2014-01-10 13:05:40 +00:00
|
|
|
/**
|
|
|
|
* gst_vaapi_encoder_h264_new:
|
|
|
|
* @display: a #GstVaapiDisplay
|
|
|
|
*
|
|
|
|
* Creates a new #GstVaapiEncoder for H.264 encoding. Note that the
|
|
|
|
* only supported output stream format is "byte-stream" format.
|
|
|
|
*
|
|
|
|
* Return value: the newly allocated #GstVaapiEncoder object
|
|
|
|
*/
|
2013-07-29 07:46:11 +00:00
|
|
|
GstVaapiEncoder *
|
|
|
|
gst_vaapi_encoder_h264_new (GstVaapiDisplay * display)
|
|
|
|
{
|
|
|
|
return gst_vaapi_encoder_new (gst_vaapi_encoder_h264_class (), display);
|
|
|
|
}
|
|
|
|
|
2014-01-06 16:46:40 +00:00
|
|
|
/**
|
|
|
|
* gst_vaapi_encoder_h264_get_default_properties:
|
|
|
|
*
|
|
|
|
* Determines the set of common and H.264 specific encoder properties.
|
|
|
|
* The caller owns an extra reference to the resulting array of
|
|
|
|
* #GstVaapiEncoderPropInfo elements, so it shall be released with
|
|
|
|
* g_ptr_array_unref() after usage.
|
|
|
|
*
|
|
|
|
* Return value: the set of encoder properties for #GstVaapiEncoderH264,
|
|
|
|
* or %NULL if an error occurred.
|
|
|
|
*/
|
|
|
|
GPtrArray *
|
|
|
|
gst_vaapi_encoder_h264_get_default_properties (void)
|
|
|
|
{
|
|
|
|
const GstVaapiEncoderClass *const klass = gst_vaapi_encoder_h264_class ();
|
|
|
|
GPtrArray *props;
|
|
|
|
|
|
|
|
props = gst_vaapi_encoder_properties_get_default (klass);
|
|
|
|
if (!props)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* GstVaapiEncoderH264:max-bframes:
|
|
|
|
*
|
|
|
|
* The number of B-frames between I and P.
|
|
|
|
*/
|
|
|
|
GST_VAAPI_ENCODER_PROPERTIES_APPEND (props,
|
|
|
|
GST_VAAPI_ENCODER_H264_PROP_MAX_BFRAMES,
|
|
|
|
g_param_spec_uint ("max-bframes",
|
2016-06-22 21:28:44 +00:00
|
|
|
"Max B-Frames", "Number of B-frames between I and P", 0, 10, 1,
|
2014-01-06 16:46:40 +00:00
|
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
|
|
|
|
/**
|
|
|
|
* GstVaapiEncoderH264:init-qp:
|
|
|
|
*
|
|
|
|
* The initial quantizer value.
|
|
|
|
*/
|
|
|
|
GST_VAAPI_ENCODER_PROPERTIES_APPEND (props,
|
|
|
|
GST_VAAPI_ENCODER_H264_PROP_INIT_QP,
|
|
|
|
g_param_spec_uint ("init-qp",
|
|
|
|
"Initial QP", "Initial quantizer value", 1, 51, 26,
|
|
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
|
|
|
|
/**
|
|
|
|
* GstVaapiEncoderH264:min-qp:
|
|
|
|
*
|
|
|
|
* The minimum quantizer value.
|
|
|
|
*/
|
|
|
|
GST_VAAPI_ENCODER_PROPERTIES_APPEND (props,
|
|
|
|
GST_VAAPI_ENCODER_H264_PROP_MIN_QP,
|
|
|
|
g_param_spec_uint ("min-qp",
|
|
|
|
"Minimum QP", "Minimum quantizer value", 1, 51, 1,
|
|
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
|
|
|
|
/**
|
|
|
|
* GstVaapiEncoderH264:num-slices:
|
|
|
|
*
|
|
|
|
* The number of slices per frame.
|
|
|
|
*/
|
|
|
|
GST_VAAPI_ENCODER_PROPERTIES_APPEND (props,
|
|
|
|
GST_VAAPI_ENCODER_H264_PROP_NUM_SLICES,
|
|
|
|
g_param_spec_uint ("num-slices",
|
|
|
|
"Number of Slices",
|
|
|
|
"Number of slices per frame",
|
|
|
|
1, 200, 1, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
|
2013-12-13 09:36:08 +00:00
|
|
|
/**
|
|
|
|
* GstVaapiEncoderH264:cabac:
|
|
|
|
*
|
|
|
|
* Enable CABAC entropy coding mode for improved compression ratio,
|
|
|
|
* at the expense that the minimum target profile is Main. Default
|
|
|
|
* is CAVLC entropy coding mode.
|
|
|
|
*/
|
|
|
|
GST_VAAPI_ENCODER_PROPERTIES_APPEND (props,
|
|
|
|
GST_VAAPI_ENCODER_H264_PROP_CABAC,
|
|
|
|
g_param_spec_boolean ("cabac",
|
|
|
|
"Enable CABAC",
|
|
|
|
"Enable CABAC entropy coding mode",
|
2016-06-22 21:28:44 +00:00
|
|
|
TRUE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
2013-12-13 09:36:08 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* GstVaapiEncoderH264:dct8x8:
|
|
|
|
*
|
|
|
|
* Enable adaptive use of 8x8 transforms in I-frames. This improves
|
|
|
|
* the compression ratio by the minimum target profile is High.
|
|
|
|
* Default is to use 4x4 DCT only.
|
|
|
|
*/
|
|
|
|
GST_VAAPI_ENCODER_PROPERTIES_APPEND (props,
|
|
|
|
GST_VAAPI_ENCODER_H264_PROP_DCT8X8,
|
|
|
|
g_param_spec_boolean ("dct8x8",
|
|
|
|
"Enable 8x8 DCT",
|
|
|
|
"Enable adaptive use of 8x8 transforms in I-frames",
|
2016-06-22 21:28:44 +00:00
|
|
|
TRUE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
2013-12-13 09:36:08 +00:00
|
|
|
|
2014-01-21 18:04:41 +00:00
|
|
|
/**
|
|
|
|
* GstVaapiEncoderH264:cpb-length:
|
|
|
|
*
|
|
|
|
* The size of the CPB buffer in milliseconds.
|
|
|
|
*/
|
|
|
|
GST_VAAPI_ENCODER_PROPERTIES_APPEND (props,
|
|
|
|
GST_VAAPI_ENCODER_H264_PROP_CPB_LENGTH,
|
|
|
|
g_param_spec_uint ("cpb-length",
|
|
|
|
"CPB Length", "Length of the CPB buffer in milliseconds",
|
|
|
|
1, 10000, DEFAULT_CPB_LENGTH,
|
|
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
|
2014-02-17 07:51:43 +00:00
|
|
|
/**
|
|
|
|
* GstVaapiEncoderH264:num-views:
|
|
|
|
*
|
|
|
|
* The number of views for MVC encoding .
|
|
|
|
*/
|
|
|
|
GST_VAAPI_ENCODER_PROPERTIES_APPEND (props,
|
|
|
|
GST_VAAPI_ENCODER_H264_PROP_NUM_VIEWS,
|
|
|
|
g_param_spec_uint ("num-views",
|
|
|
|
"Number of Views",
|
|
|
|
"Number of Views for MVC encoding",
|
|
|
|
1, MAX_NUM_VIEWS, 1, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
2015-02-24 15:14:33 +00:00
|
|
|
/**
|
|
|
|
* GstVaapiEncoderH264:view-ids:
|
|
|
|
*
|
|
|
|
* The view ids for MVC encoding .
|
|
|
|
*/
|
|
|
|
GST_VAAPI_ENCODER_PROPERTIES_APPEND (props,
|
|
|
|
GST_VAAPI_ENCODER_H264_PROP_VIEW_IDS,
|
|
|
|
g_param_spec_value_array ("view-ids",
|
|
|
|
"View IDs", "Set of View Ids used for MVC encoding",
|
|
|
|
g_param_spec_uint ("view-id-value", "View id value",
|
|
|
|
"view id values used for mvc encoding", 0, MAX_VIEW_ID, 0,
|
|
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS),
|
|
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
2014-02-17 07:51:43 +00:00
|
|
|
|
2014-01-06 16:46:40 +00:00
|
|
|
return props;
|
|
|
|
}
|
2014-01-10 17:18:25 +00:00
|
|
|
|
2014-01-12 21:24:04 +00:00
|
|
|
/**
|
|
|
|
* gst_vaapi_encoder_h264_set_max_profile:
|
|
|
|
* @encoder: a #GstVaapiEncoderH264
|
|
|
|
* @profile: an H.264 #GstVaapiProfile
|
|
|
|
*
|
|
|
|
* Notifies the @encoder to use coding tools from the supplied
|
|
|
|
* @profile at most.
|
|
|
|
*
|
|
|
|
* This means that if the minimal profile derived to
|
|
|
|
* support the specified coding tools is greater than this @profile,
|
|
|
|
* then an error is returned when the @encoder is configured.
|
|
|
|
*
|
|
|
|
* Return value: %TRUE on success
|
|
|
|
*/
|
|
|
|
gboolean
|
|
|
|
gst_vaapi_encoder_h264_set_max_profile (GstVaapiEncoderH264 * encoder,
|
|
|
|
GstVaapiProfile profile)
|
|
|
|
{
|
|
|
|
guint8 profile_idc;
|
|
|
|
|
|
|
|
g_return_val_if_fail (encoder != NULL, FALSE);
|
|
|
|
g_return_val_if_fail (profile != GST_VAAPI_PROFILE_UNKNOWN, FALSE);
|
|
|
|
|
|
|
|
if (gst_vaapi_profile_get_codec (profile) != GST_VAAPI_CODEC_H264)
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
profile_idc = gst_vaapi_utils_h264_get_profile_idc (profile);
|
|
|
|
if (!profile_idc)
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
encoder->max_profile_idc = profile_idc;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2014-01-10 17:18:25 +00:00
|
|
|
/**
|
|
|
|
* gst_vaapi_encoder_h264_get_profile_and_level:
|
|
|
|
* @encoder: a #GstVaapiEncoderH264
|
|
|
|
* @out_profile_ptr: return location for the #GstVaapiProfile
|
|
|
|
* @out_level_ptr: return location for the #GstVaapiLevelH264
|
|
|
|
*
|
|
|
|
* Queries the H.264 @encoder for the active profile and level. That
|
|
|
|
* information is only constructed and valid after the encoder is
|
|
|
|
* configured, i.e. after the gst_vaapi_encoder_set_codec_state()
|
|
|
|
* function is called.
|
|
|
|
*
|
|
|
|
* Return value: %TRUE on success
|
|
|
|
*/
|
|
|
|
gboolean
|
|
|
|
gst_vaapi_encoder_h264_get_profile_and_level (GstVaapiEncoderH264 * encoder,
|
|
|
|
GstVaapiProfile * out_profile_ptr, GstVaapiLevelH264 * out_level_ptr)
|
|
|
|
{
|
|
|
|
g_return_val_if_fail (encoder != NULL, FALSE);
|
|
|
|
|
|
|
|
if (!encoder->profile || !encoder->level)
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
if (out_profile_ptr)
|
|
|
|
*out_profile_ptr = encoder->profile;
|
|
|
|
if (out_level_ptr)
|
|
|
|
*out_level_ptr = encoder->level;
|
|
|
|
return TRUE;
|
|
|
|
}
|