mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-05 16:06:43 +00:00
2470 lines
72 KiB
C
2470 lines
72 KiB
C
|
/* GStreamer
|
||
|
* Copyright (C) 2022 Intel Corporation
|
||
|
* Author: He Junyan <junyan.he@intel.com>
|
||
|
*
|
||
|
* This library is free software; you can redistribute it and/or
|
||
|
* modify it under the terms of the GNU Library General Public
|
||
|
* License as published by the Free Software Foundation; either
|
||
|
* version 2 of the License, or (at your option) any later version.
|
||
|
*
|
||
|
* This library is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
* Library General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Library General Public
|
||
|
* License along with this library; if not, write to the0
|
||
|
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
||
|
* Boston, MA 02110-1301, USA.
|
||
|
*/
|
||
|
|
||
|
#ifdef HAVE_CONFIG_H
|
||
|
#include "config.h"
|
||
|
#endif
|
||
|
|
||
|
#include "gstav1bitwriter.h"
|
||
|
#include <gst/base/gstbitwriter.h>
|
||
|
|
||
|
#define WRITE_BITS_UNCHECK(bw, val, nbits) \
|
||
|
(nbits <= 8 ? gst_bit_writer_put_bits_uint8 (bw, val, nbits) : \
|
||
|
(nbits <= 16 ? gst_bit_writer_put_bits_uint16 (bw, val, nbits) : \
|
||
|
(nbits <= 32 ? gst_bit_writer_put_bits_uint32 (bw, val, nbits) : \
|
||
|
FALSE)))
|
||
|
|
||
|
#define WRITE_BITS(bw, val, nbits) \
|
||
|
if (!WRITE_BITS_UNCHECK (bw, val, nbits)) { \
|
||
|
g_warning ("Unsupported bit size: %u", nbits); \
|
||
|
have_space = FALSE; \
|
||
|
goto error; \
|
||
|
}
|
||
|
|
||
|
static guint
|
||
|
_av1_uleb_size_in_bytes (guint64 value)
|
||
|
{
|
||
|
guint size = 0;
|
||
|
|
||
|
do {
|
||
|
++size;
|
||
|
} while ((value >>= 7) != 0);
|
||
|
return size;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_encode_uleb (guint64 value, guint available,
|
||
|
guint8 * coded_value, guint coded_size)
|
||
|
{
|
||
|
const guint kMaximumLeb128Size = 8;
|
||
|
const guint leb_size = _av1_uleb_size_in_bytes (value);
|
||
|
const guint64 kMaximumLeb128Value = 0xffffffff /* 4294967295U */ ;
|
||
|
guint i;
|
||
|
|
||
|
g_assert (coded_size >= leb_size);
|
||
|
|
||
|
if (value > kMaximumLeb128Value || coded_size > kMaximumLeb128Size ||
|
||
|
coded_size > available || !coded_value) {
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (leb_size == coded_size) {
|
||
|
for (i = 0; i < leb_size; i++) {
|
||
|
guint8 byte = value & 0x7f;
|
||
|
value >>= 7;
|
||
|
|
||
|
if (value != 0)
|
||
|
byte |= 0x80; // Signal that more bytes follow.
|
||
|
|
||
|
*(coded_value + i) = byte;
|
||
|
}
|
||
|
} else {
|
||
|
memset (coded_value, 0, coded_size);
|
||
|
|
||
|
i = 0;
|
||
|
do {
|
||
|
*(coded_value + i) = value & 0x7fU;
|
||
|
value >>= 7;
|
||
|
i++;
|
||
|
} while (value);
|
||
|
|
||
|
for (i = 0; i < coded_size - 1; i++)
|
||
|
*(coded_value + i) |= 0x80;
|
||
|
}
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
/* 4.10.3
|
||
|
*
|
||
|
* Variable length unsigned n-bit number appearing directly in the
|
||
|
* bitstream */
|
||
|
static gboolean
|
||
|
_av1_write_uvlc (GstBitWriter * bw, guint32 value, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
gint64 shift_val = value;
|
||
|
gint32 leading_zeroes = 1;
|
||
|
|
||
|
value++;
|
||
|
|
||
|
g_assert (shift_val > 0);
|
||
|
|
||
|
while (shift_val >>= 1)
|
||
|
leading_zeroes += 2;
|
||
|
|
||
|
WRITE_BITS (bw, 0, leading_zeroes >> 1);
|
||
|
WRITE_BITS (bw, value, (leading_zeroes + 1) >> 1);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
static gint
|
||
|
_av1_helper_msb (guint n)
|
||
|
{
|
||
|
int log = 0;
|
||
|
guint value = n;
|
||
|
int i;
|
||
|
|
||
|
g_assert (n != 0);
|
||
|
|
||
|
for (i = 4; i >= 0; --i) {
|
||
|
const gint shift = (1 << i);
|
||
|
const guint x = value >> shift;
|
||
|
|
||
|
if (x != 0) {
|
||
|
value = x;
|
||
|
log += shift;
|
||
|
}
|
||
|
}
|
||
|
return log;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_write_uniform (GstBitWriter * bw, guint32 max_value, guint32 value,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
const gint l = max_value ? _av1_helper_msb (max_value) + 1 : 0;
|
||
|
const gint m = (1 << l) - max_value;
|
||
|
|
||
|
if (l == 0)
|
||
|
goto success;
|
||
|
|
||
|
if (value < m) {
|
||
|
WRITE_BITS (bw, value, l - 1);
|
||
|
} else {
|
||
|
WRITE_BITS (bw, m + ((value - m) >> 1), l - 1);
|
||
|
WRITE_BITS (bw, (value - m) & 1, 1);
|
||
|
}
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.13
|
||
|
*
|
||
|
* Delta quantizer */
|
||
|
static gboolean
|
||
|
_av1_write_delta_q (GstBitWriter * bw, gint32 delta_q, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
if (delta_q != 0) {
|
||
|
WRITE_BITS (bw, 1, 1);
|
||
|
WRITE_BITS (bw, delta_q, 7);
|
||
|
} else {
|
||
|
WRITE_BITS (bw, 0, 1);
|
||
|
}
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 4.10.6
|
||
|
*
|
||
|
* su(n) */
|
||
|
static gboolean
|
||
|
_av1_write_su (GstBitWriter * bw, gint32 val, guint n, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
g_assert (n < 31);
|
||
|
WRITE_BITS (bw, val, n + 1);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.16 Tile size calculation
|
||
|
*
|
||
|
* returns the smallest value for k such that blkSize << k is greater
|
||
|
* than or equal to target */
|
||
|
static gint
|
||
|
_av1_helper_tile_log2 (gint blkSize, gint target)
|
||
|
{
|
||
|
gint k;
|
||
|
|
||
|
for (k = 0; (blkSize << k) < target; k++);
|
||
|
|
||
|
return k;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_write_primitive_quniform (GstBitWriter * bw, guint16 n, guint16 v,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
const gint l = _av1_helper_msb (n) + 1;
|
||
|
const gint m = (1 << l) - n;
|
||
|
|
||
|
if (n <= 1)
|
||
|
goto success;
|
||
|
|
||
|
if (v < m) {
|
||
|
WRITE_BITS (bw, v, l - 1);
|
||
|
} else {
|
||
|
WRITE_BITS (bw, m + ((v - m) >> 1), l - 1);
|
||
|
WRITE_BITS (bw, (v - m) & 1, 1);
|
||
|
}
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_write_primitive_subexpfin (GstBitWriter * bw, guint16 n, guint16 k,
|
||
|
guint16 v, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
gint i = 0;
|
||
|
gint mk = 0;
|
||
|
|
||
|
while (1) {
|
||
|
gint b = (i ? k + i - 1 : k);
|
||
|
gint a = (1 << b);
|
||
|
|
||
|
if (n <= mk + 3 * a) {
|
||
|
if (!_av1_write_primitive_quniform (bw, n - mk, v - mk, space))
|
||
|
goto error;
|
||
|
|
||
|
break;
|
||
|
} else {
|
||
|
gint t = (v >= mk + a);
|
||
|
WRITE_BITS (bw, t, 1);
|
||
|
if (t) {
|
||
|
i = i + 1;
|
||
|
mk += a;
|
||
|
} else {
|
||
|
WRITE_BITS (bw, v - mk, b);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* Recenters a non-negative literal v around a reference r */
|
||
|
static guint16
|
||
|
_av1_helper_recenter_nonneg (guint16 r, guint16 v)
|
||
|
{
|
||
|
if (v > (r << 1))
|
||
|
return v;
|
||
|
else if (v >= r)
|
||
|
return ((v - r) << 1);
|
||
|
else
|
||
|
return ((r - v) << 1) - 1;
|
||
|
}
|
||
|
|
||
|
/* Recenters a non-negative literal v in [0, n-1] around
|
||
|
a reference r also in [0, n-1] */
|
||
|
static guint16
|
||
|
_av1_helper_recenter_finite_nonneg (guint16 n, guint16 r, guint16 v)
|
||
|
{
|
||
|
if ((r << 1) <= n) {
|
||
|
return _av1_helper_recenter_nonneg (r, v);
|
||
|
} else {
|
||
|
return _av1_helper_recenter_nonneg (n - 1 - r, n - 1 - v);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_write_primitive_refsubexpfin (GstBitWriter * bw,
|
||
|
guint16 n, guint16 k, guint16 ref, guint16 v, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
if (!_av1_write_primitive_subexpfin (bw, n, k,
|
||
|
_av1_helper_recenter_finite_nonneg (n, ref, v), space))
|
||
|
goto error;
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_write_signed_primitive_refsubexpfin (GstBitWriter * bw,
|
||
|
guint16 n, guint16 k, gint16 ref, gint16 v, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
const guint16 scaled_n = (n << 1) - 1;
|
||
|
ref += n - 1;
|
||
|
v += n - 1;
|
||
|
|
||
|
if (!_av1_write_primitive_refsubexpfin (bw, scaled_n, k, ref, v, space))
|
||
|
goto error;
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_seq_level_idx_is_valid (GstAV1SeqLevels seq_level_idx)
|
||
|
{
|
||
|
return seq_level_idx == GST_AV1_SEQ_LEVEL_MAX
|
||
|
|| (seq_level_idx < GST_AV1_SEQ_LEVELS
|
||
|
/* The following levels are currently undefined. */
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_2_2
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_2_3
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_3_2
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_3_3
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_4_2
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_4_3
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_7_0
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_7_1
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_7_2
|
||
|
&& seq_level_idx != GST_AV1_SEQ_LEVEL_7_3);
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_bit_writer_timing_info (const GstAV1TimingInfo * timing_info,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing timing info");
|
||
|
|
||
|
if (timing_info->num_units_in_display_tick == 0 ||
|
||
|
timing_info->time_scale == 0)
|
||
|
goto error;
|
||
|
|
||
|
WRITE_BITS (bw, timing_info->num_units_in_display_tick, 32);
|
||
|
WRITE_BITS (bw, timing_info->time_scale, 32);
|
||
|
|
||
|
WRITE_BITS (bw, timing_info->equal_picture_interval, 1);
|
||
|
if (timing_info->equal_picture_interval) {
|
||
|
if (timing_info->num_ticks_per_picture_minus_1 == G_MAXUINT)
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_write_uvlc (bw, timing_info->num_ticks_per_picture_minus_1,
|
||
|
&have_space))
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write timing info");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.5.4 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_decoder_model_info (const GstAV1DecoderModelInfo *
|
||
|
decoder_model_info, GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing decoder model info");
|
||
|
|
||
|
WRITE_BITS (bw, decoder_model_info->buffer_delay_length_minus_1, 5);
|
||
|
WRITE_BITS (bw, decoder_model_info->num_units_in_decoding_tick, 32);
|
||
|
WRITE_BITS (bw, decoder_model_info->buffer_removal_time_length_minus_1, 5);
|
||
|
WRITE_BITS (bw, decoder_model_info->frame_presentation_time_length_minus_1,
|
||
|
5);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
error:
|
||
|
GST_WARNING ("failed to write decoder model info");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.5.5 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_operating_parameters_info (const GstAV1SequenceHeaderOBU *
|
||
|
seq_header, const GstAV1OperatingPoint * op_point, GstBitWriter * bw,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
guint32 n = seq_header->decoder_model_info.buffer_delay_length_minus_1 + 1;
|
||
|
|
||
|
GST_DEBUG ("writing operating parameters info");
|
||
|
|
||
|
if (n > 32)
|
||
|
goto error;
|
||
|
|
||
|
WRITE_BITS (bw, op_point->decoder_buffer_delay, n);
|
||
|
WRITE_BITS (bw, op_point->encoder_buffer_delay, n);
|
||
|
WRITE_BITS (bw, op_point->low_delay_mode_flag, 1);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write operating parameters info");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.5.2 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_color_config (const GstAV1SequenceHeaderOBU * seq_header,
|
||
|
const GstAV1ColorConfig * color_config, GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing color config");
|
||
|
|
||
|
WRITE_BITS (bw, color_config->high_bitdepth, 1);
|
||
|
|
||
|
if (seq_header->seq_profile == GST_AV1_PROFILE_2
|
||
|
&& color_config->high_bitdepth) {
|
||
|
WRITE_BITS (bw, color_config->twelve_bit, 1);
|
||
|
} else if (seq_header->seq_profile > GST_AV1_PROFILE_2) {
|
||
|
GST_WARNING ("Unsupported profile/bit-depth combination");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
if (seq_header->seq_profile != GST_AV1_PROFILE_1)
|
||
|
WRITE_BITS (bw, color_config->mono_chrome, 1);
|
||
|
|
||
|
if (seq_header->num_planes != 1 && seq_header->num_planes != 3) {
|
||
|
GST_WARNING ("num_planes is not correct");
|
||
|
goto error;
|
||
|
}
|
||
|
if (!color_config->mono_chrome && seq_header->num_planes != 3) {
|
||
|
GST_WARNING ("num_planes is not correct");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, color_config->color_description_present_flag, 1);
|
||
|
|
||
|
if (color_config->color_description_present_flag) {
|
||
|
WRITE_BITS (bw, color_config->color_primaries, 8);
|
||
|
WRITE_BITS (bw, color_config->transfer_characteristics, 8);
|
||
|
WRITE_BITS (bw, color_config->matrix_coefficients, 8);
|
||
|
}
|
||
|
|
||
|
if (color_config->mono_chrome) {
|
||
|
if (color_config->subsampling_x != 1 || color_config->subsampling_y != 1) {
|
||
|
GST_WARNING ("set subsampling_x or subsampling_y wrong value");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, color_config->color_range, 1);
|
||
|
goto success;
|
||
|
} else if (color_config->color_primaries == GST_AV1_CP_BT_709 &&
|
||
|
color_config->transfer_characteristics == GST_AV1_TC_SRGB &&
|
||
|
color_config->matrix_coefficients == GST_AV1_MC_IDENTITY) {
|
||
|
|
||
|
if (color_config->color_range != 1) {
|
||
|
GST_WARNING ("set color_range wrong value");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
if (color_config->subsampling_x != 0 || color_config->subsampling_y != 0) {
|
||
|
GST_WARNING ("set subsampling_x or subsampling_y wrong value");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
if (!(seq_header->seq_profile == GST_AV1_PROFILE_1 ||
|
||
|
(seq_header->seq_profile == GST_AV1_PROFILE_2
|
||
|
&& seq_header->bit_depth == 12))) {
|
||
|
GST_WARNING ("sRGB colorspace not compatible with specified profile");
|
||
|
goto error;
|
||
|
}
|
||
|
} else {
|
||
|
WRITE_BITS (bw, color_config->color_range, 1);
|
||
|
|
||
|
if (seq_header->seq_profile == GST_AV1_PROFILE_0) {
|
||
|
if (color_config->subsampling_x != 1 || color_config->subsampling_y != 1) {
|
||
|
GST_WARNING ("set subsampling_x or subsampling_y wrong value");
|
||
|
goto error;
|
||
|
}
|
||
|
} else if (seq_header->seq_profile == GST_AV1_PROFILE_1) {
|
||
|
if (color_config->subsampling_x != 0 || color_config->subsampling_y != 0) {
|
||
|
GST_WARNING ("set subsampling_x or subsampling_y wrong value");
|
||
|
goto error;
|
||
|
}
|
||
|
} else if (seq_header->seq_profile == GST_AV1_PROFILE_2) {
|
||
|
if (seq_header->bit_depth == 12) {
|
||
|
WRITE_BITS (bw, color_config->subsampling_x, 1);
|
||
|
|
||
|
if (color_config->subsampling_x) {
|
||
|
/* 422 or 420 */
|
||
|
WRITE_BITS (bw, color_config->subsampling_y, 1);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (color_config->subsampling_x && color_config->subsampling_y)
|
||
|
WRITE_BITS (bw, color_config->chroma_sample_position, 2);
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, color_config->separate_uv_delta_q, 1);
|
||
|
|
||
|
if (!(color_config->subsampling_x == 0 && color_config->subsampling_y == 0) &&
|
||
|
!(color_config->subsampling_x == 1 && color_config->subsampling_y == 1) &&
|
||
|
!(color_config->subsampling_x == 1 && color_config->subsampling_y == 0)) {
|
||
|
GST_WARNING ("Only 4:4:4, 4:2:2 and 4:2:0 are currently supported, "
|
||
|
"%d %d subsampling is not supported.\n",
|
||
|
color_config->subsampling_x, color_config->subsampling_y);
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write color config");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_bit_writer_sequence_header (const GstAV1SequenceHeaderOBU * seq_header,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
gint i;
|
||
|
|
||
|
GST_DEBUG ("writing sequence header");
|
||
|
|
||
|
if (seq_header->seq_profile > GST_AV1_PROFILE_2) {
|
||
|
GST_WARNING ("Unsupported profile %d", seq_header->seq_profile);
|
||
|
goto error;
|
||
|
}
|
||
|
WRITE_BITS (bw, seq_header->seq_profile, 3);
|
||
|
|
||
|
WRITE_BITS (bw, seq_header->still_picture, 1);
|
||
|
|
||
|
if (!seq_header->still_picture && seq_header->reduced_still_picture_header) {
|
||
|
GST_WARNING (" If reduced_still_picture_header is equal to 1, it is a"
|
||
|
" requirement of bitstream conformance that still_picture is equal"
|
||
|
" to 1. ");
|
||
|
goto error;
|
||
|
}
|
||
|
WRITE_BITS (bw, seq_header->reduced_still_picture_header, 1);
|
||
|
|
||
|
if (seq_header->reduced_still_picture_header) {
|
||
|
if (!_av1_seq_level_idx_is_valid
|
||
|
(seq_header->operating_points[0].seq_level_idx)) {
|
||
|
GST_WARNING ("The seq_level_idx is unsupported");
|
||
|
goto error;
|
||
|
}
|
||
|
WRITE_BITS (bw, seq_header->operating_points[0].seq_level_idx, 5);
|
||
|
} else {
|
||
|
WRITE_BITS (bw, seq_header->timing_info_present_flag, 1);
|
||
|
if (seq_header->timing_info_present_flag) {
|
||
|
if (!_av1_bit_writer_timing_info (&seq_header->timing_info, bw,
|
||
|
&have_space))
|
||
|
goto error;
|
||
|
|
||
|
WRITE_BITS (bw, seq_header->decoder_model_info_present_flag, 1);
|
||
|
if (seq_header->decoder_model_info_present_flag) {
|
||
|
if (!_av1_bit_writer_decoder_model_info
|
||
|
(&seq_header->decoder_model_info, bw, &have_space))
|
||
|
goto error;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, seq_header->initial_display_delay_present_flag, 1);
|
||
|
|
||
|
if (seq_header->operating_points_cnt_minus_1 + 1 >
|
||
|
GST_AV1_MAX_OPERATING_POINTS) {
|
||
|
GST_WARNING ("The operating points number %d is too big",
|
||
|
seq_header->operating_points_cnt_minus_1 + 1);
|
||
|
goto error;
|
||
|
}
|
||
|
WRITE_BITS (bw, seq_header->operating_points_cnt_minus_1, 5);
|
||
|
|
||
|
for (i = 0; i < seq_header->operating_points_cnt_minus_1 + 1; i++) {
|
||
|
const GstAV1OperatingPoint *op = &seq_header->operating_points[i];
|
||
|
|
||
|
WRITE_BITS (bw, op->idc, 12);
|
||
|
|
||
|
if (!_av1_seq_level_idx_is_valid (op->seq_level_idx)) {
|
||
|
GST_WARNING ("The seq_level_idx is unsupported");
|
||
|
goto error;
|
||
|
}
|
||
|
WRITE_BITS (bw, op->seq_level_idx, 5);
|
||
|
|
||
|
if (op->seq_level_idx > GST_AV1_SEQ_LEVEL_3_3)
|
||
|
WRITE_BITS (bw, op->seq_tier, 1);
|
||
|
|
||
|
if (seq_header->decoder_model_info_present_flag) {
|
||
|
WRITE_BITS (bw, op->decoder_model_present_for_this_op, 1);
|
||
|
if (op->decoder_model_present_for_this_op) {
|
||
|
if (!_av1_bit_writer_operating_parameters_info (seq_header,
|
||
|
op, bw, &have_space))
|
||
|
goto error;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (seq_header->initial_display_delay_present_flag) {
|
||
|
WRITE_BITS (bw, op->initial_display_delay_present_for_this_op, 1);
|
||
|
|
||
|
if (op->initial_display_delay_present_for_this_op) {
|
||
|
if (op->initial_display_delay_minus_1 + 1 > 10) {
|
||
|
GST_INFO ("AV1 does not support more than 10 decoded frames delay");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, op->initial_display_delay_minus_1, 4);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, seq_header->frame_width_bits_minus_1, 4);
|
||
|
WRITE_BITS (bw, seq_header->frame_height_bits_minus_1, 4);
|
||
|
WRITE_BITS (bw, seq_header->max_frame_width_minus_1,
|
||
|
seq_header->frame_width_bits_minus_1 + 1);
|
||
|
WRITE_BITS (bw, seq_header->max_frame_height_minus_1,
|
||
|
seq_header->frame_height_bits_minus_1 + 1);
|
||
|
|
||
|
if (!seq_header->reduced_still_picture_header)
|
||
|
WRITE_BITS (bw, seq_header->frame_id_numbers_present_flag, 1);
|
||
|
|
||
|
if (seq_header->frame_id_numbers_present_flag) {
|
||
|
if (seq_header->additional_frame_id_length_minus_1 + 1 +
|
||
|
seq_header->delta_frame_id_length_minus_2 + 2 > 16) {
|
||
|
GST_WARNING ("Invalid frame_id_length");
|
||
|
goto error;
|
||
|
}
|
||
|
WRITE_BITS (bw, seq_header->delta_frame_id_length_minus_2, 4);
|
||
|
WRITE_BITS (bw, seq_header->additional_frame_id_length_minus_1, 3);
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, seq_header->use_128x128_superblock, 1);
|
||
|
WRITE_BITS (bw, seq_header->enable_filter_intra, 1);
|
||
|
WRITE_BITS (bw, seq_header->enable_intra_edge_filter, 1);
|
||
|
|
||
|
if (!seq_header->reduced_still_picture_header) {
|
||
|
WRITE_BITS (bw, seq_header->enable_interintra_compound, 1);
|
||
|
WRITE_BITS (bw, seq_header->enable_masked_compound, 1);
|
||
|
WRITE_BITS (bw, seq_header->enable_warped_motion, 1);
|
||
|
WRITE_BITS (bw, seq_header->enable_dual_filter, 1);
|
||
|
WRITE_BITS (bw, seq_header->enable_order_hint, 1);
|
||
|
if (seq_header->enable_order_hint) {
|
||
|
WRITE_BITS (bw, seq_header->enable_jnt_comp, 1);
|
||
|
WRITE_BITS (bw, seq_header->enable_ref_frame_mvs, 1);
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, seq_header->seq_choose_screen_content_tools, 1);
|
||
|
if (!seq_header->seq_choose_screen_content_tools)
|
||
|
WRITE_BITS (bw, seq_header->seq_force_screen_content_tools, 1);
|
||
|
|
||
|
if (seq_header->seq_force_screen_content_tools > 0) {
|
||
|
WRITE_BITS (bw, seq_header->seq_choose_integer_mv, 1);
|
||
|
if (!seq_header->seq_choose_integer_mv)
|
||
|
WRITE_BITS (bw, seq_header->seq_force_integer_mv, 1);
|
||
|
}
|
||
|
|
||
|
if (seq_header->enable_order_hint)
|
||
|
WRITE_BITS (bw, seq_header->order_hint_bits_minus_1, 3);
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, seq_header->enable_superres, 1);
|
||
|
WRITE_BITS (bw, seq_header->enable_cdef, 1);
|
||
|
WRITE_BITS (bw, seq_header->enable_restoration, 1);
|
||
|
|
||
|
if (!_av1_bit_writer_color_config (seq_header, &seq_header->color_config,
|
||
|
bw, &have_space))
|
||
|
goto error;
|
||
|
|
||
|
WRITE_BITS (bw, seq_header->film_grain_params_present, 1);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write sequence header");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* Return the actual size field size. */
|
||
|
static guint
|
||
|
_av1_bit_writer_add_size_field (guint8 * data, guint * size, guint header_size,
|
||
|
guint payload_size, guint size_field_size, gboolean * space)
|
||
|
{
|
||
|
guint size_field_sz;
|
||
|
|
||
|
size_field_sz = _av1_uleb_size_in_bytes (payload_size);
|
||
|
g_assert (size_field_sz > 0);
|
||
|
|
||
|
if (size_field_size > 0) {
|
||
|
if (size_field_sz > size_field_size) {
|
||
|
GST_WARNING ("the fixed size field size is too small");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
size_field_sz = size_field_size;
|
||
|
}
|
||
|
|
||
|
/* Move and write the data size field */
|
||
|
if (header_size + payload_size + size_field_sz > *size) {
|
||
|
*space = FALSE;
|
||
|
goto error;
|
||
|
}
|
||
|
*space = TRUE;
|
||
|
|
||
|
memmove (data + header_size + size_field_sz,
|
||
|
data + header_size, payload_size);
|
||
|
|
||
|
if (!_av1_encode_uleb (payload_size, sizeof (payload_size),
|
||
|
data + header_size, size_field_sz))
|
||
|
goto error;
|
||
|
|
||
|
*size = header_size + payload_size + size_field_sz;
|
||
|
return size_field_sz;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write the size field");
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* gst_av1_bit_writer_sequence_header_obu:
|
||
|
* @seq_hdr: the sequence header of #GstAV1SequenceHeaderOBU to write
|
||
|
* @size_field: whether the header contain size feild
|
||
|
* @data: (out): the bit stream generated by the sequence header
|
||
|
* @size: (inout): the size in bytes of the input and output
|
||
|
*
|
||
|
* Generating the according AV1 bit stream OBU by providing the sequence header.
|
||
|
*
|
||
|
* Returns: a #GstAV1BitWriterResult
|
||
|
*
|
||
|
* Since: 1.22
|
||
|
**/
|
||
|
GstAV1BitWriterResult
|
||
|
gst_av1_bit_writer_sequence_header_obu (const GstAV1SequenceHeaderOBU *
|
||
|
seq_hdr, gboolean size_field, guint8 * data, guint * size)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
GstBitWriter bw;
|
||
|
guint header_size;
|
||
|
guint payload_size;
|
||
|
|
||
|
g_return_val_if_fail (seq_hdr != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (data != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (size != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (*size > 0, GST_AV1_BIT_WRITER_ERROR);
|
||
|
|
||
|
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
|
||
|
|
||
|
/* obu_forbidden_bit */
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
/* obu_type */
|
||
|
WRITE_BITS (&bw, GST_AV1_OBU_SEQUENCE_HEADER, 4);
|
||
|
/* obu_extention_flag */
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
/* obu_has_size_field */
|
||
|
WRITE_BITS (&bw, 1, 1);
|
||
|
/* obu_reserved_1bit */
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
|
||
|
header_size = gst_bit_writer_get_size (&bw);
|
||
|
g_assert (header_size % 8 == 0);
|
||
|
header_size /= 8;
|
||
|
|
||
|
if (!_av1_bit_writer_sequence_header (seq_hdr, &bw, &have_space))
|
||
|
goto error;
|
||
|
|
||
|
/* Add trailings. */
|
||
|
WRITE_BITS (&bw, 1, 1);
|
||
|
if (!gst_bit_writer_align_bytes (&bw, 0)) {
|
||
|
have_space = FALSE;
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
payload_size = gst_bit_writer_get_size (&bw);
|
||
|
g_assert (payload_size % 8 == 0);
|
||
|
payload_size /= 8;
|
||
|
payload_size -= header_size;
|
||
|
|
||
|
gst_bit_writer_reset (&bw);
|
||
|
|
||
|
if (size_field) {
|
||
|
if (!_av1_bit_writer_add_size_field (data, size,
|
||
|
header_size, payload_size, 0, &have_space))
|
||
|
goto error;
|
||
|
} else {
|
||
|
*size = header_size + payload_size;
|
||
|
}
|
||
|
|
||
|
return GST_AV1_BIT_WRITER_OK;
|
||
|
|
||
|
error:
|
||
|
gst_bit_writer_reset (&bw);
|
||
|
*size = 0;
|
||
|
return have_space ? GST_AV1_BIT_WRITER_INVALID_DATA :
|
||
|
GST_AV1_BIT_WRITER_NO_MORE_SPACE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.5 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_superres_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, GstBitWriter * bw,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing superres param");
|
||
|
|
||
|
if (seq_header->enable_superres)
|
||
|
WRITE_BITS (bw, frame_header->use_superres, 1);
|
||
|
|
||
|
if (frame_header->use_superres) {
|
||
|
guint8 coded_denom;
|
||
|
|
||
|
if (frame_header->superres_denom < GST_AV1_SUPERRES_DENOM_MIN)
|
||
|
goto error;
|
||
|
|
||
|
coded_denom = frame_header->superres_denom - GST_AV1_SUPERRES_DENOM_MIN;
|
||
|
if (coded_denom > (1 << GST_AV1_SUPERRES_DENOM_BITS) - 1)
|
||
|
goto error;
|
||
|
|
||
|
WRITE_BITS (bw, coded_denom, GST_AV1_SUPERRES_DENOM_BITS);
|
||
|
}
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write superres param");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.5 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_frame_size (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, GstBitWriter * bw,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing frame size");
|
||
|
|
||
|
if (frame_header->frame_size_override_flag) {
|
||
|
guint16 frame_width_minus_1;
|
||
|
guint16 frame_height_minus_1;
|
||
|
|
||
|
frame_width_minus_1 = frame_header->frame_width - 1;
|
||
|
WRITE_BITS (bw, frame_width_minus_1,
|
||
|
seq_header->frame_width_bits_minus_1 + 1);
|
||
|
frame_height_minus_1 = frame_header->frame_height - 1;
|
||
|
WRITE_BITS (bw, frame_height_minus_1,
|
||
|
seq_header->frame_height_bits_minus_1 + 1);
|
||
|
}
|
||
|
|
||
|
if (!_av1_bit_writer_superres_params (frame_header, seq_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write frame size");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.6 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_render_size (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing render size");
|
||
|
|
||
|
WRITE_BITS (bw, frame_header->render_and_frame_size_different, 1);
|
||
|
|
||
|
if (frame_header->render_and_frame_size_different) {
|
||
|
guint16 render_width_minus_1 = frame_header->render_width - 1;
|
||
|
guint16 render_height_minus_1 = frame_header->render_height - 1;
|
||
|
|
||
|
WRITE_BITS (bw, render_width_minus_1, 16);
|
||
|
WRITE_BITS (bw, render_height_minus_1, 16);
|
||
|
}
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write render size");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.15 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_tile_info (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, GstBitWriter * bw,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
const GstAV1TileInfo *tile_info;
|
||
|
guint32 mi_cols /* MiCols */ ;
|
||
|
guint32 mi_rows /* MiRows */ ;
|
||
|
gint sb_cols /* sbCols */ ;
|
||
|
gint sb_rows /* sbRows */ ;
|
||
|
gint sb_shift /*sbShift */ ;
|
||
|
gint sb_size /* sbSize */ ;
|
||
|
gint max_tile_width_sb /* maxTileWidthSb */ ;
|
||
|
gint max_tile_height_sb /* maxTileHeightSb */ ;
|
||
|
gint max_tile_area_sb /* maxTileAreaSb */ ;
|
||
|
gint min_log2_tile_cols /* minLog2TileCols */ ;
|
||
|
gint max_log2_tile_cols /* maxLog2TileCols */ ;
|
||
|
gint min_log2_tile_rows /* minLog2TileRows */ ;
|
||
|
gint max_log2_tile_rows /* maxLog2TileRows */ ;
|
||
|
gint min_log2_tiles /* minLog2Tiles */ ;
|
||
|
gint size_sb /* sizeSb */ ;
|
||
|
gint widest_tile_sb /* widestTileSb */ ;
|
||
|
|
||
|
tile_info = &frame_header->tile_info;
|
||
|
|
||
|
GST_DEBUG ("writing tile info");
|
||
|
|
||
|
/* User must specify the frame resolution. */
|
||
|
if (frame_header->frame_width == 0 || frame_header->frame_height == 0) {
|
||
|
GST_WARNING ("unknown frame_width or frame_height");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
mi_cols = 2 * ((frame_header->frame_width + 7) >> 3);
|
||
|
mi_rows = 2 * ((frame_header->frame_height + 7) >> 3);
|
||
|
|
||
|
sb_cols = seq_header->use_128x128_superblock ?
|
||
|
((mi_cols + 31) >> 5) : ((mi_cols + 15) >> 4);
|
||
|
sb_rows = seq_header->use_128x128_superblock ?
|
||
|
((mi_rows + 31) >> 5) : ((mi_rows + 15) >> 4);
|
||
|
sb_shift = seq_header->use_128x128_superblock ? 5 : 4;
|
||
|
sb_size = sb_shift + 2;
|
||
|
|
||
|
max_tile_width_sb = GST_AV1_MAX_TILE_WIDTH >> sb_size;
|
||
|
max_tile_area_sb = GST_AV1_MAX_TILE_AREA >> (2 * sb_size);
|
||
|
min_log2_tile_cols = _av1_helper_tile_log2 (max_tile_width_sb, sb_cols);
|
||
|
max_log2_tile_cols = _av1_helper_tile_log2 (1, MIN (sb_cols,
|
||
|
GST_AV1_MAX_TILE_COLS));
|
||
|
max_log2_tile_rows = _av1_helper_tile_log2 (1, MIN (sb_rows,
|
||
|
GST_AV1_MAX_TILE_ROWS));
|
||
|
min_log2_tiles = MAX (min_log2_tile_cols,
|
||
|
_av1_helper_tile_log2 (max_tile_area_sb, sb_rows * sb_cols));
|
||
|
|
||
|
WRITE_BITS (bw, tile_info->uniform_tile_spacing_flag, 1);
|
||
|
if (tile_info->uniform_tile_spacing_flag) {
|
||
|
/* columns */
|
||
|
gint ones = tile_info->tile_cols_log2 - min_log2_tile_cols;
|
||
|
if (ones < 0)
|
||
|
goto error;
|
||
|
|
||
|
while (ones--)
|
||
|
WRITE_BITS (bw, 1, 1);
|
||
|
if (tile_info->tile_cols_log2 < max_log2_tile_cols)
|
||
|
WRITE_BITS (bw, 0, 1);
|
||
|
|
||
|
/* rows */
|
||
|
min_log2_tile_rows = MAX (min_log2_tiles - tile_info->tile_cols_log2, 0);
|
||
|
ones = tile_info->tile_rows_log2 - min_log2_tile_rows;
|
||
|
if (ones < 0)
|
||
|
goto error;
|
||
|
|
||
|
while (ones--)
|
||
|
WRITE_BITS (bw, 1, 1);
|
||
|
if (tile_info->tile_rows_log2 < max_log2_tile_rows)
|
||
|
WRITE_BITS (bw, 0, 1);
|
||
|
} else {
|
||
|
/* Explicit tiles with configurable tile widths and heights */
|
||
|
guint i;
|
||
|
guint width_sb;
|
||
|
guint height_sb;
|
||
|
|
||
|
widest_tile_sb = 0;
|
||
|
|
||
|
/* columns */
|
||
|
width_sb = sb_cols;
|
||
|
for (i = 0; i < tile_info->tile_cols; i++) {
|
||
|
size_sb = tile_info->mi_col_starts[i + 1] - tile_info->mi_col_starts[i];
|
||
|
size_sb = size_sb >> sb_shift;
|
||
|
widest_tile_sb = MAX (size_sb, widest_tile_sb);
|
||
|
if (_av1_write_uniform (bw, MIN (width_sb, max_tile_width_sb),
|
||
|
size_sb - 1, &have_space))
|
||
|
goto error;
|
||
|
|
||
|
width_sb -= size_sb;
|
||
|
}
|
||
|
|
||
|
if (width_sb != 0)
|
||
|
goto error;
|
||
|
|
||
|
// rows
|
||
|
if (min_log2_tiles > 0) {
|
||
|
max_tile_area_sb = (sb_rows * sb_cols) >> (min_log2_tiles + 1);
|
||
|
} else {
|
||
|
max_tile_area_sb = sb_rows * sb_cols;
|
||
|
}
|
||
|
|
||
|
max_tile_height_sb = MAX (max_tile_area_sb / widest_tile_sb, 1);
|
||
|
|
||
|
height_sb = sb_rows;
|
||
|
for (i = 0; i < tile_info->tile_rows; i++) {
|
||
|
size_sb = tile_info->mi_row_starts[i + 1] - tile_info->mi_row_starts[i];
|
||
|
size_sb = size_sb >> sb_shift;
|
||
|
if (_av1_write_uniform (bw, MIN (height_sb, max_tile_height_sb),
|
||
|
size_sb - 1, &have_space))
|
||
|
goto error;
|
||
|
|
||
|
height_sb -= size_sb;
|
||
|
}
|
||
|
|
||
|
if (height_sb != 0)
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
if (tile_info->tile_cols_log2 > 0 || tile_info->tile_rows_log2 > 0) {
|
||
|
WRITE_BITS (bw, tile_info->context_update_tile_id,
|
||
|
tile_info->tile_cols_log2 + tile_info->tile_rows_log2);
|
||
|
|
||
|
WRITE_BITS (bw, tile_info->tile_size_bytes_minus_1, 2);
|
||
|
}
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write tile info");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.12 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_quantization_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, guint * qindex_offset,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
const GstAV1QuantizationParams *quant_params =
|
||
|
&(frame_header->quantization_params);
|
||
|
|
||
|
GST_DEBUG ("writing quantization params");
|
||
|
|
||
|
if (qindex_offset)
|
||
|
*qindex_offset = gst_bit_writer_get_size (bw);
|
||
|
|
||
|
WRITE_BITS (bw, quant_params->base_q_idx, 8);
|
||
|
|
||
|
if (!_av1_write_delta_q (bw,
|
||
|
frame_header->quantization_params.delta_q_y_dc, &have_space))
|
||
|
goto error;
|
||
|
|
||
|
if (seq_header->num_planes > 1) {
|
||
|
if (seq_header->color_config.separate_uv_delta_q)
|
||
|
WRITE_BITS (bw, quant_params->diff_uv_delta, 1);
|
||
|
|
||
|
if (!_av1_write_delta_q (bw,
|
||
|
frame_header->quantization_params.delta_q_u_dc, &have_space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_write_delta_q (bw,
|
||
|
frame_header->quantization_params.delta_q_u_ac, &have_space))
|
||
|
goto error;
|
||
|
|
||
|
if (quant_params->diff_uv_delta) {
|
||
|
if (!_av1_write_delta_q (bw,
|
||
|
frame_header->quantization_params.delta_q_v_dc, &have_space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_write_delta_q (bw,
|
||
|
frame_header->quantization_params.delta_q_v_ac, &have_space))
|
||
|
goto error;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, quant_params->using_qmatrix, 1);
|
||
|
if (quant_params->using_qmatrix) {
|
||
|
WRITE_BITS (bw, quant_params->qm_y, 4);
|
||
|
WRITE_BITS (bw, quant_params->qm_u, 4);
|
||
|
|
||
|
if (seq_header->color_config.separate_uv_delta_q)
|
||
|
WRITE_BITS (bw, quant_params->qm_v, 4);
|
||
|
}
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write quantization params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.14 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_segmentation_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, guint * segmentation_offset,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing segmentation params");
|
||
|
|
||
|
if (segmentation_offset)
|
||
|
*segmentation_offset = gst_bit_writer_get_size (bw);
|
||
|
|
||
|
/* TODO: segmentation support. */
|
||
|
if (frame_header->segmentation_params.segmentation_enabled) {
|
||
|
GST_WARNING ("segmentation is not supported now");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, frame_header->segmentation_params.segmentation_enabled, 1);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write segmentation params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.17 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_delta_q_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, GstBitWriter * bw,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing delta q params");
|
||
|
|
||
|
if (frame_header->quantization_params.base_q_idx > 0)
|
||
|
WRITE_BITS (bw, frame_header->quantization_params.delta_q_present, 1);
|
||
|
|
||
|
if (frame_header->quantization_params.delta_q_present)
|
||
|
WRITE_BITS (bw, frame_header->quantization_params.delta_q_res, 2);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write delta q params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.18 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_delta_lf_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, GstBitWriter * bw,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing delta lf params");
|
||
|
|
||
|
if (frame_header->quantization_params.delta_q_present) {
|
||
|
if (!frame_header->allow_intrabc)
|
||
|
WRITE_BITS (bw, frame_header->loop_filter_params.delta_lf_present, 1);
|
||
|
|
||
|
if (frame_header->loop_filter_params.delta_lf_present) {
|
||
|
WRITE_BITS (bw, frame_header->loop_filter_params.delta_lf_res, 2);
|
||
|
WRITE_BITS (bw, frame_header->loop_filter_params.delta_lf_multi, 1);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write delta lf params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_bit_writer_loop_filter_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, guint * loopfilter_offset,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
const GstAV1LoopFilterParams *lf_params = &frame_header->loop_filter_params;
|
||
|
|
||
|
GST_DEBUG ("writing loop filter params");
|
||
|
|
||
|
if (frame_header->coded_lossless || frame_header->allow_intrabc)
|
||
|
goto success;
|
||
|
|
||
|
if (loopfilter_offset)
|
||
|
*loopfilter_offset = gst_bit_writer_get_size (bw);
|
||
|
|
||
|
WRITE_BITS (bw, lf_params->loop_filter_level[0], 6);
|
||
|
WRITE_BITS (bw, lf_params->loop_filter_level[1], 6);
|
||
|
if (seq_header->num_planes > 1) {
|
||
|
if (lf_params->loop_filter_level[0] || lf_params->loop_filter_level[1]) {
|
||
|
WRITE_BITS (bw, lf_params->loop_filter_level[2], 6);
|
||
|
WRITE_BITS (bw, lf_params->loop_filter_level[3], 6);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, lf_params->loop_filter_sharpness, 3);
|
||
|
|
||
|
WRITE_BITS (bw, lf_params->loop_filter_delta_enabled, 1);
|
||
|
if (lf_params->loop_filter_delta_enabled) {
|
||
|
guint i;
|
||
|
|
||
|
WRITE_BITS (bw, lf_params->loop_filter_delta_update, 1);
|
||
|
|
||
|
if (lf_params->loop_filter_delta_update) {
|
||
|
const gint8 default_loop_filter_ref_deltas[] =
|
||
|
{ 1, 0, 0, 0, -1, 0, -1, -1 };
|
||
|
gboolean update_ref_deltas;
|
||
|
|
||
|
for (i = 0; i < GST_AV1_TOTAL_REFS_PER_FRAME; i++) {
|
||
|
/* If loop_filter_ref_deltas[i] is different from default
|
||
|
value, we update it. */
|
||
|
update_ref_deltas = (lf_params->loop_filter_ref_deltas[i] !=
|
||
|
default_loop_filter_ref_deltas[i]);
|
||
|
|
||
|
WRITE_BITS (bw, update_ref_deltas, 1);
|
||
|
|
||
|
if (update_ref_deltas) {
|
||
|
if (!_av1_write_su (bw, lf_params->loop_filter_ref_deltas[i],
|
||
|
6, space))
|
||
|
goto error;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
for (i = 0; i < 2; i++) {
|
||
|
/* If loop_filter_mode_deltas[i] is different from default
|
||
|
value, we update it. */
|
||
|
update_ref_deltas = (lf_params->loop_filter_mode_deltas[i] != 0);
|
||
|
|
||
|
WRITE_BITS (bw, update_ref_deltas, 1);
|
||
|
|
||
|
if (update_ref_deltas) {
|
||
|
if (!_av1_write_su (bw, lf_params->loop_filter_mode_deltas[i],
|
||
|
6, space))
|
||
|
goto error;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write loop filter params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_bit_writer_cdef_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, guint * cdef_offset,
|
||
|
guint * cdef_size, GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
const GstAV1CDEFParams *cdef_params = &frame_header->cdef_params;
|
||
|
guint cdef_start;
|
||
|
guint i;
|
||
|
|
||
|
GST_DEBUG ("writing cdef params");
|
||
|
|
||
|
if (frame_header->coded_lossless || frame_header->allow_intrabc
|
||
|
|| !seq_header->enable_cdef)
|
||
|
goto success;
|
||
|
|
||
|
cdef_start = gst_bit_writer_get_size (bw);
|
||
|
if (cdef_offset)
|
||
|
*cdef_offset = cdef_start;
|
||
|
|
||
|
WRITE_BITS (bw, cdef_params->cdef_damping - 3, 2);
|
||
|
WRITE_BITS (bw, cdef_params->cdef_bits, 2);
|
||
|
|
||
|
for (i = 0; i < (1 << cdef_params->cdef_bits); i++) {
|
||
|
guint8 cdef_y_sec_strength;
|
||
|
|
||
|
WRITE_BITS (bw, cdef_params->cdef_y_pri_strength[i], 4);
|
||
|
|
||
|
cdef_y_sec_strength = cdef_params->cdef_y_sec_strength[i];
|
||
|
|
||
|
if (cdef_y_sec_strength >= 4) {
|
||
|
GST_WARNING ("cdef_y_sec_strength is not valid");
|
||
|
goto error;
|
||
|
}
|
||
|
WRITE_BITS (bw, cdef_y_sec_strength, 2);
|
||
|
|
||
|
if (seq_header->num_planes > 1) {
|
||
|
guint8 cdef_uv_sec_strength;
|
||
|
|
||
|
WRITE_BITS (bw, cdef_params->cdef_uv_pri_strength[i], 4);
|
||
|
|
||
|
cdef_uv_sec_strength = cdef_params->cdef_uv_sec_strength[i];
|
||
|
|
||
|
if (cdef_uv_sec_strength >= 4) {
|
||
|
GST_WARNING ("cdef_uv_sec_strength is not valid");
|
||
|
goto error;
|
||
|
}
|
||
|
WRITE_BITS (bw, cdef_uv_sec_strength, 2);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (cdef_size)
|
||
|
*cdef_size = gst_bit_writer_get_size (bw) - cdef_start;
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write cdef params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_bit_writer_loop_restoration_params (const GstAV1FrameHeaderOBU *
|
||
|
frame_header, const GstAV1SequenceHeaderOBU * seq_header,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
guint i, j;
|
||
|
guint8 use_chroma_lr = 0 /* useChromaLr */ ;
|
||
|
const GstAV1LoopRestorationParams *lr_params =
|
||
|
&frame_header->loop_restoration_params;
|
||
|
const GstAV1FrameRestorationType remap_lr_type /* Remap_Lr_Type */ [4] = {
|
||
|
GST_AV1_FRAME_RESTORE_NONE,
|
||
|
GST_AV1_FRAME_RESTORE_SWITCHABLE,
|
||
|
GST_AV1_FRAME_RESTORE_WIENER, GST_AV1_FRAME_RESTORE_SGRPROJ
|
||
|
};
|
||
|
|
||
|
GST_DEBUG ("writing loop restoration params");
|
||
|
|
||
|
if (frame_header->all_lossless || frame_header->allow_intrabc
|
||
|
|| !seq_header->enable_restoration)
|
||
|
goto success;
|
||
|
|
||
|
for (i = 0; i < seq_header->num_planes; i++) {
|
||
|
for (j = 0; j < 4; j++) {
|
||
|
if (lr_params->frame_restoration_type[i] == remap_lr_type[j])
|
||
|
break;
|
||
|
}
|
||
|
if (j == 4)
|
||
|
goto error;
|
||
|
|
||
|
if (lr_params->frame_restoration_type[i] != GST_AV1_FRAME_RESTORE_NONE) {
|
||
|
if (!lr_params->uses_lr) {
|
||
|
GST_WARNING ("uses_lr set to wrong value");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
if (i > 1)
|
||
|
use_chroma_lr = 1;
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, j, 2);
|
||
|
}
|
||
|
|
||
|
if (lr_params->uses_lr) {
|
||
|
if (lr_params->lr_unit_shift > 2)
|
||
|
goto error;
|
||
|
|
||
|
if (seq_header->use_128x128_superblock) {
|
||
|
WRITE_BITS (bw, lr_params->lr_unit_shift - 1, 1);
|
||
|
} else {
|
||
|
WRITE_BITS (bw, 1, 1);
|
||
|
|
||
|
if (lr_params->lr_unit_shift > 1)
|
||
|
/* lr_unit_extra_shift */
|
||
|
WRITE_BITS (bw, 1, 1);
|
||
|
}
|
||
|
|
||
|
if (seq_header->color_config.subsampling_x
|
||
|
&& seq_header->color_config.subsampling_y && use_chroma_lr)
|
||
|
WRITE_BITS (bw, lr_params->lr_uv_shift, 1);
|
||
|
}
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write loop restoration params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.22 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_skip_mode_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, GstBitWriter * bw,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing skip mode params");
|
||
|
|
||
|
/* if skipModeAllowed is true */
|
||
|
if (frame_header->skip_mode_frame[0] > 0
|
||
|
|| frame_header->skip_mode_frame[1] > 0)
|
||
|
WRITE_BITS (bw, frame_header->skip_mode_present, 1);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write skip mode params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.24 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_global_motion_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, GstBitWriter * bw,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
gint ref, i;
|
||
|
const GstAV1GlobalMotionParams *gm_params =
|
||
|
&(frame_header->global_motion_params);
|
||
|
gint32 prev_gm_params[GST_AV1_NUM_REF_FRAMES][6] /* PrevGmParams */ ;
|
||
|
|
||
|
GST_DEBUG ("writing global motion params");
|
||
|
|
||
|
if (frame_header->frame_is_intra)
|
||
|
goto success;
|
||
|
|
||
|
if (frame_header->primary_ref_frame != GST_AV1_PRIMARY_REF_NONE) {
|
||
|
memcpy (prev_gm_params, &frame_header->ref_global_motion_params,
|
||
|
sizeof (gint32) * GST_AV1_NUM_REF_FRAMES * 6);
|
||
|
} else {
|
||
|
for (ref = GST_AV1_REF_INTRA_FRAME; ref < GST_AV1_NUM_REF_FRAMES; ref++)
|
||
|
for (i = 0; i < 6; i++)
|
||
|
prev_gm_params[ref][i] =
|
||
|
((i % 3 == 2) ? 1 << GST_AV1_WARPEDMODEL_PREC_BITS : 0);
|
||
|
}
|
||
|
|
||
|
for (ref = GST_AV1_REF_LAST_FRAME; ref <= GST_AV1_REF_ALTREF_FRAME; ref++) {
|
||
|
WRITE_BITS (bw, gm_params->gm_type[ref] != GST_AV1_WARP_MODEL_IDENTITY, 1);
|
||
|
|
||
|
if (gm_params->gm_type[ref] != GST_AV1_WARP_MODEL_IDENTITY) {
|
||
|
WRITE_BITS (bw, gm_params->gm_type[ref] == GST_AV1_WARP_MODEL_ROTZOOM, 1);
|
||
|
|
||
|
if (gm_params->gm_type[ref] != GST_AV1_WARP_MODEL_ROTZOOM)
|
||
|
WRITE_BITS (bw,
|
||
|
gm_params->gm_type[ref] == GST_AV1_WARP_MODEL_TRANSLATION, 1);
|
||
|
}
|
||
|
|
||
|
if (gm_params->gm_type[ref] >= GST_AV1_WARP_MODEL_ROTZOOM) {
|
||
|
if (!_av1_write_signed_primitive_refsubexpfin (bw,
|
||
|
(1 << GST_AV1_GM_ABS_ALPHA_BITS) + 1, 3,
|
||
|
(prev_gm_params[ref][2] >> (GST_AV1_WARPEDMODEL_PREC_BITS -
|
||
|
GST_AV1_GM_ALPHA_PREC_BITS)) -
|
||
|
(1 << GST_AV1_GM_ALPHA_PREC_BITS),
|
||
|
(gm_params->gm_params[ref][2] >> (GST_AV1_WARPEDMODEL_PREC_BITS -
|
||
|
GST_AV1_GM_ALPHA_PREC_BITS)) -
|
||
|
(1 << GST_AV1_GM_ALPHA_PREC_BITS), space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_write_signed_primitive_refsubexpfin (bw,
|
||
|
(1 << GST_AV1_GM_ABS_ALPHA_BITS) + 1, 3,
|
||
|
(prev_gm_params[ref][3] >> (GST_AV1_WARPEDMODEL_PREC_BITS -
|
||
|
GST_AV1_GM_ALPHA_PREC_BITS)),
|
||
|
(gm_params->gm_params[ref][3] >> (GST_AV1_WARPEDMODEL_PREC_BITS -
|
||
|
GST_AV1_GM_ALPHA_PREC_BITS)), space))
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
if (gm_params->gm_type[ref] >= GST_AV1_WARP_MODEL_AFFINE) {
|
||
|
if (!_av1_write_signed_primitive_refsubexpfin (bw,
|
||
|
(1 << GST_AV1_GM_ABS_ALPHA_BITS) + 1, 3,
|
||
|
(prev_gm_params[ref][4] >> (GST_AV1_WARPEDMODEL_PREC_BITS -
|
||
|
GST_AV1_GM_ALPHA_PREC_BITS)),
|
||
|
(gm_params->gm_params[ref][4] >> (GST_AV1_WARPEDMODEL_PREC_BITS -
|
||
|
GST_AV1_GM_ALPHA_PREC_BITS)), space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_write_signed_primitive_refsubexpfin (bw,
|
||
|
(1 << GST_AV1_GM_ABS_ALPHA_BITS) + 1, 3,
|
||
|
(prev_gm_params[ref][5] >> (GST_AV1_WARPEDMODEL_PREC_BITS -
|
||
|
GST_AV1_GM_ALPHA_PREC_BITS)) -
|
||
|
(1 << GST_AV1_GM_ALPHA_PREC_BITS),
|
||
|
(gm_params->gm_params[ref][5] >> (GST_AV1_WARPEDMODEL_PREC_BITS -
|
||
|
GST_AV1_GM_ALPHA_PREC_BITS)) -
|
||
|
(1 << GST_AV1_GM_ALPHA_PREC_BITS), space))
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
if (gm_params->gm_type[ref] >= GST_AV1_WARP_MODEL_TRANSLATION) {
|
||
|
const gint trans_bits =
|
||
|
(gm_params->gm_type[ref] == GST_AV1_WARP_MODEL_TRANSLATION) ?
|
||
|
GST_AV1_GM_ABS_TRANS_ONLY_BITS -
|
||
|
!frame_header->allow_high_precision_mv : GST_AV1_GM_ABS_TRANS_BITS;
|
||
|
const gint trans_prec_diff =
|
||
|
(gm_params->gm_type[ref] == GST_AV1_WARP_MODEL_TRANSLATION) ?
|
||
|
GST_AV1_WARPEDMODEL_PREC_BITS - 3 +
|
||
|
!frame_header->allow_high_precision_mv :
|
||
|
(GST_AV1_WARPEDMODEL_PREC_BITS - GST_AV1_GM_TRANS_PREC_BITS);
|
||
|
|
||
|
if (!_av1_write_signed_primitive_refsubexpfin (bw, (1 << trans_bits) + 1,
|
||
|
3, (prev_gm_params[ref][0] >> trans_prec_diff),
|
||
|
(gm_params->gm_params[ref][0] >> trans_prec_diff), space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_write_signed_primitive_refsubexpfin (bw, (1 << trans_bits) + 1,
|
||
|
3, (prev_gm_params[ref][1] >> trans_prec_diff),
|
||
|
(gm_params->gm_params[ref][1] >> trans_prec_diff), space))
|
||
|
goto error;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write global motion params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.9.30 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_film_grain_params (const GstAV1FrameHeaderOBU * frame_header,
|
||
|
const GstAV1SequenceHeaderOBU * seq_header, GstBitWriter * bw,
|
||
|
gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
guint i;
|
||
|
gint num_pos_chroma /* numPosChroma */ , num_pos_luma /* numPosLuma */ ;
|
||
|
const GstAV1FilmGrainParams *fg_params = &frame_header->film_grain_params;
|
||
|
|
||
|
GST_DEBUG ("writing film grain params");
|
||
|
|
||
|
fg_params = &frame_header->film_grain_params;
|
||
|
if (!seq_header->film_grain_params_present || (!frame_header->show_frame
|
||
|
&& !frame_header->showable_frame))
|
||
|
goto success;
|
||
|
|
||
|
WRITE_BITS (bw, fg_params->apply_grain, 1);
|
||
|
if (!fg_params->apply_grain)
|
||
|
goto success;
|
||
|
|
||
|
WRITE_BITS (bw, fg_params->grain_seed, 16);
|
||
|
|
||
|
if (frame_header->frame_type == GST_AV1_INTER_FRAME)
|
||
|
WRITE_BITS (bw, fg_params->update_grain, 1);
|
||
|
|
||
|
if (!fg_params->update_grain) {
|
||
|
WRITE_BITS (bw, fg_params->film_grain_params_ref_idx, 3);
|
||
|
goto success;
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, fg_params->num_y_points, 4);
|
||
|
|
||
|
for (i = 0; i < fg_params->num_y_points; i++) {
|
||
|
WRITE_BITS (bw, fg_params->point_y_value[i], 8);
|
||
|
WRITE_BITS (bw, fg_params->point_y_scaling[i], 8);
|
||
|
}
|
||
|
|
||
|
if (!seq_header->color_config.mono_chrome)
|
||
|
WRITE_BITS (bw, fg_params->chroma_scaling_from_luma, 1);
|
||
|
|
||
|
if (!(seq_header->color_config.mono_chrome
|
||
|
|| fg_params->chroma_scaling_from_luma
|
||
|
|| (seq_header->color_config.subsampling_x == 1
|
||
|
&& seq_header->color_config.subsampling_y == 1
|
||
|
&& fg_params->num_y_points == 0))) {
|
||
|
WRITE_BITS (bw, fg_params->num_cb_points, 4);
|
||
|
|
||
|
for (i = 0; i < fg_params->num_cb_points; i++) {
|
||
|
WRITE_BITS (bw, fg_params->point_cb_value[i], 8);
|
||
|
WRITE_BITS (bw, fg_params->point_cb_scaling[i], 8);
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, fg_params->num_cr_points, 4);
|
||
|
for (i = 0; i < fg_params->num_cr_points; i++) {
|
||
|
WRITE_BITS (bw, fg_params->point_cr_value[i], 8);
|
||
|
WRITE_BITS (bw, fg_params->point_cr_scaling[i], 8);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, fg_params->grain_scaling_minus_8, 2);
|
||
|
|
||
|
WRITE_BITS (bw, fg_params->ar_coeff_lag, 2);
|
||
|
|
||
|
num_pos_luma = 2 * fg_params->ar_coeff_lag * (fg_params->ar_coeff_lag + 1);
|
||
|
if (fg_params->num_y_points) {
|
||
|
num_pos_chroma = num_pos_luma + 1;
|
||
|
for (i = 0; i < num_pos_luma; i++)
|
||
|
WRITE_BITS (bw, fg_params->ar_coeffs_y_plus_128[i], 8);
|
||
|
} else {
|
||
|
num_pos_chroma = num_pos_luma;
|
||
|
}
|
||
|
|
||
|
if (fg_params->chroma_scaling_from_luma || fg_params->num_cb_points) {
|
||
|
for (i = 0; i < num_pos_chroma; i++)
|
||
|
WRITE_BITS (bw, fg_params->ar_coeffs_cb_plus_128[i], 8);
|
||
|
}
|
||
|
|
||
|
if (fg_params->chroma_scaling_from_luma || fg_params->num_cr_points) {
|
||
|
for (i = 0; i < num_pos_chroma; i++)
|
||
|
WRITE_BITS (bw, fg_params->ar_coeffs_cr_plus_128[i], 8);
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, fg_params->ar_coeff_shift_minus_6, 2);
|
||
|
|
||
|
WRITE_BITS (bw, fg_params->grain_scale_shift, 2);
|
||
|
|
||
|
if (fg_params->num_cb_points) {
|
||
|
WRITE_BITS (bw, fg_params->cb_mult, 8);
|
||
|
WRITE_BITS (bw, fg_params->cb_luma_mult, 8);
|
||
|
WRITE_BITS (bw, fg_params->cb_offset, 9);
|
||
|
}
|
||
|
|
||
|
if (fg_params->num_cr_points) {
|
||
|
WRITE_BITS (bw, fg_params->cr_mult, 8);
|
||
|
WRITE_BITS (bw, fg_params->cr_luma_mult, 8);
|
||
|
WRITE_BITS (bw, fg_params->cr_offset, 9);
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, fg_params->overlap_flag, 1);
|
||
|
WRITE_BITS (bw, fg_params->clip_to_restricted_range, 1);
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write film grain params");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
static gboolean
|
||
|
_av1_bit_writer_uncompressed_frame_header (const GstAV1FrameHeaderOBU *
|
||
|
frame_header, const GstAV1SequenceHeaderOBU * seq_header,
|
||
|
guint8 temporal_id, guint8 spatial_id, guint * qindex_offset,
|
||
|
guint * segmentation_offset, guint * loopfilter_offset,
|
||
|
guint * cdef_offset, guint * cdef_size, GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
guint i;
|
||
|
gint id_len /* idLen */ = 0;
|
||
|
const GstAV1DecoderModelInfo *cmi = &seq_header->decoder_model_info;
|
||
|
|
||
|
GST_DEBUG ("writing frame header");
|
||
|
|
||
|
if (seq_header->frame_id_numbers_present_flag)
|
||
|
id_len = seq_header->additional_frame_id_length_minus_1 + 1 +
|
||
|
seq_header->delta_frame_id_length_minus_2 + 2;
|
||
|
|
||
|
if (!seq_header->reduced_still_picture_header) {
|
||
|
WRITE_BITS (bw, frame_header->show_existing_frame, 1);
|
||
|
if (frame_header->show_existing_frame) {
|
||
|
WRITE_BITS (bw, frame_header->frame_to_show_map_idx, 3);
|
||
|
|
||
|
if (seq_header->decoder_model_info_present_flag
|
||
|
&& !seq_header->timing_info.equal_picture_interval)
|
||
|
WRITE_BITS (bw, frame_header->frame_presentation_time,
|
||
|
cmi->frame_presentation_time_length_minus_1 + 1);
|
||
|
|
||
|
if (seq_header->frame_id_numbers_present_flag) {
|
||
|
if (id_len <= 0)
|
||
|
goto error;
|
||
|
|
||
|
WRITE_BITS (bw, frame_header->display_frame_id, id_len);
|
||
|
}
|
||
|
|
||
|
goto success;
|
||
|
}
|
||
|
|
||
|
if (seq_header->still_picture &&
|
||
|
(frame_header->frame_type != GST_AV1_KEY_FRAME
|
||
|
|| !frame_header->show_frame)) {
|
||
|
GST_INFO ("Still pictures must be coded as shown keyframes");
|
||
|
goto error;
|
||
|
}
|
||
|
WRITE_BITS (bw, frame_header->frame_type, 2);
|
||
|
WRITE_BITS (bw, frame_header->show_frame, 1);
|
||
|
|
||
|
if (frame_header->show_frame
|
||
|
&& seq_header->decoder_model_info_present_flag
|
||
|
&& !seq_header->timing_info.equal_picture_interval)
|
||
|
WRITE_BITS (bw, frame_header->frame_presentation_time,
|
||
|
cmi->frame_presentation_time_length_minus_1 + 1);
|
||
|
|
||
|
if (!frame_header->show_frame)
|
||
|
WRITE_BITS (bw, frame_header->showable_frame, 1);
|
||
|
|
||
|
if (!(frame_header->frame_type == GST_AV1_SWITCH_FRAME
|
||
|
|| (frame_header->frame_type == GST_AV1_KEY_FRAME
|
||
|
&& frame_header->show_frame)))
|
||
|
WRITE_BITS (bw, frame_header->error_resilient_mode, 1);
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, frame_header->disable_cdf_update, 1);
|
||
|
|
||
|
if (seq_header->seq_force_screen_content_tools ==
|
||
|
GST_AV1_SELECT_SCREEN_CONTENT_TOOLS)
|
||
|
WRITE_BITS (bw, frame_header->allow_screen_content_tools, 1);
|
||
|
|
||
|
if (frame_header->allow_screen_content_tools) {
|
||
|
if (seq_header->seq_force_integer_mv == GST_AV1_SELECT_INTEGER_MV)
|
||
|
WRITE_BITS (bw, frame_header->force_integer_mv, 1);
|
||
|
}
|
||
|
|
||
|
if (seq_header->frame_id_numbers_present_flag) {
|
||
|
if (id_len <= 0)
|
||
|
goto error;
|
||
|
|
||
|
WRITE_BITS (bw, frame_header->current_frame_id, id_len);
|
||
|
}
|
||
|
|
||
|
if (frame_header->frame_type != GST_AV1_SWITCH_FRAME
|
||
|
&& !seq_header->reduced_still_picture_header)
|
||
|
WRITE_BITS (bw, frame_header->frame_size_override_flag, 1);
|
||
|
|
||
|
WRITE_BITS (bw, frame_header->order_hint,
|
||
|
seq_header->order_hint_bits_minus_1 + 1);
|
||
|
|
||
|
if (frame_header->frame_is_intra || frame_header->error_resilient_mode) {
|
||
|
if (frame_header->primary_ref_frame != GST_AV1_PRIMARY_REF_NONE) {
|
||
|
GST_WARNING ("primary_ref_frame is not none.");
|
||
|
goto error;
|
||
|
}
|
||
|
} else {
|
||
|
WRITE_BITS (bw, frame_header->primary_ref_frame, 3);
|
||
|
}
|
||
|
|
||
|
if (seq_header->decoder_model_info_present_flag) {
|
||
|
if (frame_header->buffer_removal_time_present_flag) {
|
||
|
guint op_num;
|
||
|
const GstAV1OperatingPoint *operating_points;
|
||
|
|
||
|
for (op_num = 0; op_num <= seq_header->operating_points_cnt_minus_1;
|
||
|
op_num++) {
|
||
|
operating_points = &seq_header->operating_points[op_num];
|
||
|
|
||
|
if (operating_points->decoder_model_present_for_this_op) {
|
||
|
gint op_pt_idc = operating_points->idc;
|
||
|
gint in_temporal_layer = (op_pt_idc >> temporal_id) & 1;
|
||
|
gint in_spatial_layer = (op_pt_idc >> (spatial_id + 8)) & 1;
|
||
|
|
||
|
if (op_pt_idc == 0 || (in_temporal_layer && in_spatial_layer))
|
||
|
WRITE_BITS (bw, frame_header->buffer_removal_time[op_num],
|
||
|
cmi->buffer_removal_time_length_minus_1 + 1);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (frame_header->frame_type == GST_AV1_INTRA_ONLY_FRAME) {
|
||
|
if (frame_header->refresh_frame_flags == 0xFF) {
|
||
|
GST_INFO ("Intra only frames cannot have refresh flags 0xFF");
|
||
|
goto error;
|
||
|
}
|
||
|
}
|
||
|
if (!(frame_header->frame_type == GST_AV1_SWITCH_FRAME ||
|
||
|
(frame_header->frame_type == GST_AV1_KEY_FRAME
|
||
|
&& frame_header->show_frame)))
|
||
|
WRITE_BITS (bw, frame_header->refresh_frame_flags, 8);
|
||
|
|
||
|
if (!frame_header->frame_is_intra
|
||
|
|| frame_header->refresh_frame_flags !=
|
||
|
(1 << GST_AV1_NUM_REF_FRAMES) - 1) {
|
||
|
if (frame_header->error_resilient_mode && seq_header->enable_order_hint) {
|
||
|
for (i = 0; i < GST_AV1_NUM_REF_FRAMES; i++)
|
||
|
WRITE_BITS (bw, frame_header->ref_order_hint[i],
|
||
|
seq_header->order_hint_bits_minus_1 + 1);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (frame_header->frame_is_intra) {
|
||
|
if (!_av1_bit_writer_frame_size (frame_header, seq_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_render_size (frame_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (frame_header->allow_screen_content_tools
|
||
|
&& frame_header->upscaled_width == frame_header->frame_width)
|
||
|
WRITE_BITS (bw, frame_header->allow_intrabc, 1);
|
||
|
} else {
|
||
|
if (seq_header->enable_order_hint) {
|
||
|
WRITE_BITS (bw, frame_header->frame_refs_short_signaling, 1);
|
||
|
|
||
|
if (frame_header->frame_refs_short_signaling) {
|
||
|
WRITE_BITS (bw, frame_header->last_frame_idx, 3);
|
||
|
WRITE_BITS (bw, frame_header->gold_frame_idx, 3);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
for (i = 0; i < GST_AV1_REFS_PER_FRAME; i++) {
|
||
|
if (!frame_header->frame_refs_short_signaling)
|
||
|
WRITE_BITS (bw, frame_header->ref_frame_idx[i], 3);
|
||
|
|
||
|
if (seq_header->frame_id_numbers_present_flag) {
|
||
|
gint32 delta_frame_id /* DeltaFrameId */ ;
|
||
|
|
||
|
if (id_len <= 0)
|
||
|
goto error;
|
||
|
|
||
|
delta_frame_id =
|
||
|
frame_header->current_frame_id - frame_header->expected_frame_id[i];
|
||
|
delta_frame_id = delta_frame_id + (1 << id_len);
|
||
|
delta_frame_id = delta_frame_id % (1 << id_len);
|
||
|
WRITE_BITS (bw, delta_frame_id - 1,
|
||
|
seq_header->delta_frame_id_length_minus_2 + 2);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (frame_header->frame_size_override_flag
|
||
|
&& !frame_header->error_resilient_mode) {
|
||
|
/* 5.9.7 */
|
||
|
/* TODO: reuse reference frame width/height. Just disable now. */
|
||
|
for (i = 0; i < GST_AV1_REFS_PER_FRAME; i++)
|
||
|
WRITE_BITS (bw, 0, 1);
|
||
|
}
|
||
|
|
||
|
if (!_av1_bit_writer_frame_size (frame_header, seq_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_render_size (frame_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!frame_header->force_integer_mv)
|
||
|
WRITE_BITS (bw, frame_header->allow_high_precision_mv, 1);
|
||
|
|
||
|
WRITE_BITS (bw, frame_header->is_filter_switchable, 1);
|
||
|
if (!frame_header->is_filter_switchable)
|
||
|
WRITE_BITS (bw, frame_header->interpolation_filter, 2);
|
||
|
|
||
|
WRITE_BITS (bw, frame_header->is_motion_mode_switchable, 1);
|
||
|
|
||
|
if (!(frame_header->error_resilient_mode
|
||
|
|| !seq_header->enable_ref_frame_mvs))
|
||
|
WRITE_BITS (bw, frame_header->use_ref_frame_mvs, 1);
|
||
|
}
|
||
|
|
||
|
if (!(seq_header->reduced_still_picture_header
|
||
|
|| frame_header->disable_cdf_update))
|
||
|
WRITE_BITS (bw, frame_header->disable_frame_end_update_cdf, 1);
|
||
|
|
||
|
if (!_av1_bit_writer_tile_info (frame_header, seq_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_quantization_params (frame_header, seq_header,
|
||
|
qindex_offset, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_segmentation_params (frame_header, seq_header,
|
||
|
segmentation_offset, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_delta_q_params (frame_header, seq_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_delta_lf_params (frame_header, seq_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_loop_filter_params (frame_header, seq_header,
|
||
|
loopfilter_offset, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_cdef_params (frame_header, seq_header,
|
||
|
cdef_offset, cdef_size, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_loop_restoration_params (frame_header, seq_header,
|
||
|
bw, space))
|
||
|
goto error;
|
||
|
|
||
|
/* 5.9.21 tx_mode() */
|
||
|
if (frame_header->coded_lossless != 1) {
|
||
|
/* Write the tx_mode_select bit. */
|
||
|
if (frame_header->tx_mode == GST_AV1_TX_MODE_SELECT) {
|
||
|
WRITE_BITS (bw, 1, 1);
|
||
|
} else {
|
||
|
WRITE_BITS (bw, 0, 1);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* 5.9.23 inlined frame_reference_mode () */
|
||
|
if (!frame_header->frame_is_intra)
|
||
|
WRITE_BITS (bw, frame_header->reference_select, 1);
|
||
|
|
||
|
if (!_av1_bit_writer_skip_mode_params (frame_header, seq_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!(frame_header->frame_is_intra || frame_header->error_resilient_mode ||
|
||
|
!seq_header->enable_warped_motion))
|
||
|
WRITE_BITS (bw, frame_header->allow_warped_motion, 1);
|
||
|
|
||
|
WRITE_BITS (bw, frame_header->reduced_tx_set, 1);
|
||
|
|
||
|
if (!_av1_bit_writer_global_motion_params (frame_header,
|
||
|
seq_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
if (!_av1_bit_writer_film_grain_params (frame_header, seq_header, bw, space))
|
||
|
goto error;
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write frame header");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* gst_av1_bit_writer_frame_header_obu:
|
||
|
* @frame_hdr: the frame header of #GstAV1FrameHeaderOBU to write
|
||
|
* @seq_hdr: the sequence header of #GstAV1SequenceHeaderOBU to refer
|
||
|
* @temporal_id: specifies the temporal level of the data contained in the OBU.
|
||
|
* @spatial_id: specifies the spatial level of the data contained in the OBU.
|
||
|
* @size_field: whether the OBU header contains the OBU size.
|
||
|
* @data: (out): the bit stream generated by the frame header
|
||
|
* @size: (inout): the size in bytes of the input and output
|
||
|
*
|
||
|
* Generating the according AV1 bit stream OBU by providing the frame header.
|
||
|
*
|
||
|
* Returns: a #GstAV1BitWriterResult
|
||
|
*
|
||
|
* Since: 1.22
|
||
|
**/
|
||
|
GstAV1BitWriterResult
|
||
|
gst_av1_bit_writer_frame_header_obu (const GstAV1FrameHeaderOBU * frame_hdr,
|
||
|
const GstAV1SequenceHeaderOBU * seq_hdr, guint8 temporal_id,
|
||
|
guint8 spatial_id, gboolean size_field, guint8 * data, guint * size)
|
||
|
{
|
||
|
return gst_av1_bit_writer_frame_header_obu_with_offsets (frame_hdr, seq_hdr,
|
||
|
temporal_id, spatial_id, size_field, 0, NULL, NULL, NULL, NULL, NULL,
|
||
|
data, size);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* gst_av1_bit_writer_frame_header_obu_with_offsets:
|
||
|
* @frame_hdr: the frame header of #GstAV1FrameHeaderOBU to write
|
||
|
* @seq_hdr: the sequence header of #GstAV1SequenceHeaderOBU to refer
|
||
|
* @temporal_id: specifies the temporal level of the data contained in the OBU.
|
||
|
* @spatial_id: specifies the spatial level of the data contained in the OBU.
|
||
|
* @size_field: whether the OBU header contains the OBU size.
|
||
|
* @size_field_size: >0 means a fixed OBU header size field.
|
||
|
* @qindex_offset: (out): return the qindex fields offset in bits.
|
||
|
* @segmentation_offset: (out): return the segmentation fields offset in bits.
|
||
|
* @lf_offset: (out): return the loopfilter fields offset in bits.
|
||
|
* @cdef_offset: (out): return the cdef fields offset in bits.
|
||
|
* @cdef_size: (out): return the cdef fields size in bits.
|
||
|
* @data: (out): the bit stream generated by the frame header
|
||
|
* @size: (inout): the size in bytes of the input and output
|
||
|
*
|
||
|
* While Generating the according AV1 bit stream OBU by providing the frame
|
||
|
* header, this function also return bit offsets of qindex, segmentation and
|
||
|
* cdef, etc. These offsets can help to change the content of these fields
|
||
|
* later. This function is useful if the encoder may change the content of
|
||
|
* the frame header after generating it. For example, some HW needs user to
|
||
|
* provide a frame header before the real encoding job, and it will change
|
||
|
* the according fields in the frame header during the real encoding job.
|
||
|
*
|
||
|
* Returns: a #GstAV1BitWriterResult
|
||
|
*
|
||
|
* Since: 1.22
|
||
|
**/
|
||
|
GstAV1BitWriterResult
|
||
|
gst_av1_bit_writer_frame_header_obu_with_offsets (const GstAV1FrameHeaderOBU *
|
||
|
frame_hdr, const GstAV1SequenceHeaderOBU * seq_hdr, guint8 temporal_id,
|
||
|
guint8 spatial_id, gboolean size_field, guint size_field_size,
|
||
|
guint * qindex_offset, guint * segmentation_offset, guint * lf_offset,
|
||
|
guint * cdef_offset, guint * cdef_size, guint8 * data, guint * size)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
GstBitWriter bw;
|
||
|
guint payload_size;
|
||
|
guint header_size;
|
||
|
|
||
|
g_return_val_if_fail (frame_hdr != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (seq_hdr != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (temporal_id < GST_AV1_MAX_NUM_TEMPORAL_LAYERS,
|
||
|
GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (spatial_id < GST_AV1_MAX_NUM_SPATIAL_LAYERS,
|
||
|
GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (data != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (size != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (*size > 0, GST_AV1_BIT_WRITER_ERROR);
|
||
|
|
||
|
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
|
||
|
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
/* obu_type */
|
||
|
WRITE_BITS (&bw, GST_AV1_OBU_FRAME_HEADER, 4);
|
||
|
/* obu_extention_flag */
|
||
|
if (temporal_id > 0 || spatial_id > 0) {
|
||
|
WRITE_BITS (&bw, 1, 1);
|
||
|
} else {
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
}
|
||
|
|
||
|
/* obu_has_size_field */
|
||
|
if (size_field > 0) {
|
||
|
WRITE_BITS (&bw, 1, 1);
|
||
|
} else {
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
}
|
||
|
/* obu_reserved_1bit */
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
|
||
|
header_size = 1;
|
||
|
|
||
|
/* 5.3.3 OBU extension header */
|
||
|
if (temporal_id > 0 || spatial_id > 0) {
|
||
|
WRITE_BITS (&bw, temporal_id, 3);
|
||
|
WRITE_BITS (&bw, spatial_id, 2);
|
||
|
/* obu_extension_header_reserved_3bits */
|
||
|
WRITE_BITS (&bw, 0, 3);
|
||
|
header_size = 2;
|
||
|
}
|
||
|
|
||
|
/* Write size field later */
|
||
|
|
||
|
if (!_av1_bit_writer_uncompressed_frame_header (frame_hdr, seq_hdr,
|
||
|
temporal_id, spatial_id, qindex_offset, segmentation_offset,
|
||
|
lf_offset, cdef_offset, cdef_size, &bw, &have_space))
|
||
|
goto error;
|
||
|
|
||
|
/* Add trailings. */
|
||
|
WRITE_BITS (&bw, 1, 1);
|
||
|
if (!gst_bit_writer_align_bytes (&bw, 0)) {
|
||
|
have_space = FALSE;
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
payload_size = gst_bit_writer_get_size (&bw);
|
||
|
g_assert (payload_size % 8 == 0);
|
||
|
payload_size /= 8;
|
||
|
payload_size -= header_size;
|
||
|
|
||
|
gst_bit_writer_reset (&bw);
|
||
|
|
||
|
if (size_field) {
|
||
|
size_field_size = _av1_bit_writer_add_size_field (data, size,
|
||
|
header_size, payload_size, size_field_size, &have_space);
|
||
|
if (!size_field_size)
|
||
|
goto error;
|
||
|
} else {
|
||
|
*size = header_size + payload_size;
|
||
|
size_field_size = 0;
|
||
|
}
|
||
|
|
||
|
if (qindex_offset)
|
||
|
*qindex_offset = *qindex_offset + size_field_size * 8;
|
||
|
if (segmentation_offset)
|
||
|
*segmentation_offset = *segmentation_offset + size_field_size * 8;
|
||
|
if (lf_offset)
|
||
|
*lf_offset = *lf_offset + size_field_size * 8;
|
||
|
if (cdef_offset)
|
||
|
*cdef_offset = *cdef_offset + size_field_size * 8;
|
||
|
|
||
|
return GST_AV1_BIT_WRITER_OK;
|
||
|
|
||
|
error:
|
||
|
gst_bit_writer_reset (&bw);
|
||
|
*size = 0;
|
||
|
return have_space ? GST_AV1_BIT_WRITER_INVALID_DATA :
|
||
|
GST_AV1_BIT_WRITER_NO_MORE_SPACE;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* gst_av1_bit_writer_temporal_delimiter_obu:
|
||
|
* @size_field: whether the header contain size feild
|
||
|
* @data: (out): the bit stream generated
|
||
|
* @size: (inout): the size in bytes of the input and output
|
||
|
*
|
||
|
* Generating the according temporal delimiter AV1 bit stream OBU.
|
||
|
*
|
||
|
* Returns: a #GstAV1BitWriterResult
|
||
|
*
|
||
|
* Since: 1.22
|
||
|
**/
|
||
|
GstAV1BitWriterResult
|
||
|
gst_av1_bit_writer_temporal_delimiter_obu (gboolean size_field,
|
||
|
guint8 * data, guint * size)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
GstBitWriter bw;
|
||
|
|
||
|
g_return_val_if_fail (data != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (size != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (*size > 0, GST_AV1_BIT_WRITER_ERROR);
|
||
|
|
||
|
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
|
||
|
|
||
|
/* obu_forbidden_bit */
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
/* obu_type */
|
||
|
WRITE_BITS (&bw, GST_AV1_OBU_TEMPORAL_DELIMITER, 4);
|
||
|
/* obu_extention_flag */
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
/* obu_has_size_field */
|
||
|
if (size_field) {
|
||
|
WRITE_BITS (&bw, 1, 1);
|
||
|
} else {
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
}
|
||
|
/* obu_reserved_1bit */
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
|
||
|
/* No need to add trailings */
|
||
|
|
||
|
/* header_size is 1 and payload_size is 0 */
|
||
|
if (size_field) {
|
||
|
if (!_av1_bit_writer_add_size_field (data, size, 1, 0, 0, &have_space))
|
||
|
goto error;
|
||
|
} else {
|
||
|
*size = 1 + 0;
|
||
|
}
|
||
|
|
||
|
return GST_AV1_BIT_WRITER_OK;
|
||
|
|
||
|
error:
|
||
|
gst_bit_writer_reset (&bw);
|
||
|
*size = 0;
|
||
|
return have_space ? GST_AV1_BIT_WRITER_INVALID_DATA :
|
||
|
GST_AV1_BIT_WRITER_NO_MORE_SPACE;
|
||
|
}
|
||
|
|
||
|
/* 5.8.2 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_metadata_itut_t35 (const GstAV1MetadataITUT_T35 * itut_t35,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing metadata itut t35");
|
||
|
|
||
|
WRITE_BITS (bw, itut_t35->itu_t_t35_country_code, 8);
|
||
|
|
||
|
if (itut_t35->itu_t_t35_country_code == 0xFF)
|
||
|
WRITE_BITS (bw, itut_t35->itu_t_t35_country_code_extention_byte, 8);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write metadata itut t35");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.8.3 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_metadata_hdr_cll (const GstAV1MetadataHdrCll * hdr_cll,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
|
||
|
GST_DEBUG ("writing metadata hdr cll");
|
||
|
|
||
|
WRITE_BITS (bw, hdr_cll->max_cll, 16);
|
||
|
WRITE_BITS (bw, hdr_cll->max_fall, 16);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write metadata hdr cll");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.8.4 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_metadata_hdr_mdcv (const GstAV1MetadataHdrMdcv * hdr_mdcv,
|
||
|
GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
gint i;
|
||
|
|
||
|
GST_DEBUG ("writing metadata hdr mdcv");
|
||
|
|
||
|
for (i = 0; i < 3; i++) {
|
||
|
WRITE_BITS (bw, hdr_mdcv->primary_chromaticity_x[i], 16);
|
||
|
WRITE_BITS (bw, hdr_mdcv->primary_chromaticity_y[i], 16);
|
||
|
}
|
||
|
|
||
|
WRITE_BITS (bw, hdr_mdcv->white_point_chromaticity_x, 16);
|
||
|
WRITE_BITS (bw, hdr_mdcv->white_point_chromaticity_y, 16);
|
||
|
|
||
|
WRITE_BITS (bw, hdr_mdcv->luminance_max, 32);
|
||
|
WRITE_BITS (bw, hdr_mdcv->luminance_min, 32);
|
||
|
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write metadata hdr mdcv");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/* 5.8.5 */
|
||
|
static gboolean
|
||
|
_av1_bit_writer_metadata_scalability (const GstAV1MetadataScalability *
|
||
|
scalability, GstBitWriter * bw, gboolean * space)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
guint i, j;
|
||
|
|
||
|
GST_DEBUG ("writing metadata scalability");
|
||
|
|
||
|
WRITE_BITS (bw, scalability->scalability_mode_idc, 8);
|
||
|
|
||
|
if (scalability->scalability_mode_idc != GST_AV1_SCALABILITY_SS)
|
||
|
goto success;
|
||
|
|
||
|
/* 5.8.6 */
|
||
|
WRITE_BITS (bw, scalability->spatial_layers_cnt_minus_1, 2);
|
||
|
WRITE_BITS (bw, scalability->spatial_layer_dimensions_present_flag, 1);
|
||
|
WRITE_BITS (bw, scalability->spatial_layer_description_present_flag, 1);
|
||
|
WRITE_BITS (bw, scalability->temporal_group_description_present_flag, 1);
|
||
|
/* scalability_structure_reserved_3bits */
|
||
|
WRITE_BITS (bw, 0, 3);
|
||
|
|
||
|
if (scalability->spatial_layer_dimensions_present_flag) {
|
||
|
for (i = 0; i <= scalability->spatial_layers_cnt_minus_1; i++) {
|
||
|
WRITE_BITS (bw, scalability->spatial_layer_max_width[i], 16);
|
||
|
WRITE_BITS (bw, scalability->spatial_layer_max_height[i], 16);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (scalability->spatial_layer_description_present_flag) {
|
||
|
for (i = 0; i <= scalability->spatial_layers_cnt_minus_1; i++)
|
||
|
WRITE_BITS (bw, scalability->spatial_layer_ref_id[i], 8);
|
||
|
}
|
||
|
|
||
|
if (scalability->temporal_group_description_present_flag) {
|
||
|
WRITE_BITS (bw, scalability->temporal_group_size, 8);
|
||
|
|
||
|
for (i = 0; i < scalability->temporal_group_size; i++) {
|
||
|
WRITE_BITS (bw, scalability->temporal_group_temporal_id[i], 3);
|
||
|
WRITE_BITS (bw,
|
||
|
scalability->temporal_group_temporal_switching_up_point_flag[i], 1);
|
||
|
WRITE_BITS (bw,
|
||
|
scalability->temporal_group_spatial_switching_up_point_flag[i], 1);
|
||
|
WRITE_BITS (bw, scalability->temporal_group_ref_cnt[i], 3);
|
||
|
for (j = 0; j < scalability->temporal_group_ref_cnt[i]; j++)
|
||
|
WRITE_BITS (bw, scalability->temporal_group_ref_pic_diff[i][j], 8);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
success:
|
||
|
*space = TRUE;
|
||
|
return TRUE;
|
||
|
|
||
|
error:
|
||
|
GST_WARNING ("failed to write metadata scalability");
|
||
|
*space = have_space;
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* gst_av1_bit_writer_metadata_obu:
|
||
|
* @metadata: the meta data of #GstAV1MetadataOBU to write
|
||
|
* @temporal_id: specifies the temporal level of the data contained in the OBU.
|
||
|
* @spatial_id: specifies the spatial level of the data contained in the OBU.
|
||
|
* @size_field: whether the header contain size feild
|
||
|
* @data: (out): the bit stream generated by the meta data
|
||
|
* @size: (inout): the size in bytes of the input and output
|
||
|
*
|
||
|
* Generating the according AV1 bit stream OBU by providing the meta data.
|
||
|
*
|
||
|
* Returns: a #GstAV1BitWriterResult
|
||
|
*
|
||
|
* Since: 1.22
|
||
|
**/
|
||
|
GstAV1BitWriterResult
|
||
|
gst_av1_bit_writer_metadata_obu (const GstAV1MetadataOBU * metadata,
|
||
|
guint8 temporal_id, guint8 spatial_id, gboolean size_field,
|
||
|
guint8 * data, guint * size)
|
||
|
{
|
||
|
gboolean have_space = TRUE;
|
||
|
GstBitWriter bw;
|
||
|
guint header_size;
|
||
|
guint payload_size;
|
||
|
guint8 metadata_type_data[4];
|
||
|
guint metadata_size;
|
||
|
guint i;
|
||
|
|
||
|
g_return_val_if_fail (metadata != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (temporal_id < GST_AV1_MAX_NUM_TEMPORAL_LAYERS,
|
||
|
GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (spatial_id < GST_AV1_MAX_NUM_SPATIAL_LAYERS,
|
||
|
GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (data != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (size != NULL, GST_AV1_BIT_WRITER_ERROR);
|
||
|
g_return_val_if_fail (*size > 0, GST_AV1_BIT_WRITER_ERROR);
|
||
|
|
||
|
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
|
||
|
|
||
|
/* obu_forbidden_bit */
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
/* obu_type */
|
||
|
WRITE_BITS (&bw, GST_AV1_OBU_METADATA, 4);
|
||
|
/* obu_extention_flag */
|
||
|
if (temporal_id > 0 || spatial_id > 0) {
|
||
|
WRITE_BITS (&bw, 1, 1);
|
||
|
} else {
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
}
|
||
|
/* obu_has_size_field */
|
||
|
if (size_field) {
|
||
|
WRITE_BITS (&bw, 1, 1);
|
||
|
} else {
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
}
|
||
|
/* obu_reserved_1bit */
|
||
|
WRITE_BITS (&bw, 0, 1);
|
||
|
|
||
|
header_size = 1;
|
||
|
/* 5.3.3 OBU extension header */
|
||
|
if (temporal_id > 0 || spatial_id > 0) {
|
||
|
WRITE_BITS (&bw, temporal_id, 3);
|
||
|
WRITE_BITS (&bw, spatial_id, 2);
|
||
|
/* obu_extension_header_reserved_3bits */
|
||
|
WRITE_BITS (&bw, 0, 3);
|
||
|
header_size = 2;
|
||
|
}
|
||
|
|
||
|
/* Generate the metadata_type first */
|
||
|
metadata_size = _av1_uleb_size_in_bytes (metadata->metadata_type);
|
||
|
if (metadata_size > 4) {
|
||
|
GST_WARNING ("Invalid metadata_type");
|
||
|
goto error;
|
||
|
}
|
||
|
if (!_av1_encode_uleb (metadata->metadata_type,
|
||
|
sizeof (metadata->metadata_type),
|
||
|
metadata_type_data, metadata_size)) {
|
||
|
GST_WARNING ("Failed to write metadata_type");
|
||
|
goto error;
|
||
|
}
|
||
|
for (i = 0; i < metadata_size; i++)
|
||
|
WRITE_BITS (&bw, metadata_type_data[i], 8);
|
||
|
|
||
|
switch (metadata->metadata_type) {
|
||
|
case GST_AV1_METADATA_TYPE_ITUT_T35:
|
||
|
if (!_av1_bit_writer_metadata_itut_t35 (&metadata->itut_t35,
|
||
|
&bw, &have_space))
|
||
|
goto error;
|
||
|
break;
|
||
|
case GST_AV1_METADATA_TYPE_HDR_CLL:
|
||
|
if (!_av1_bit_writer_metadata_hdr_cll (&metadata->hdr_cll,
|
||
|
&bw, &have_space))
|
||
|
goto error;
|
||
|
break;
|
||
|
case GST_AV1_METADATA_TYPE_HDR_MDCV:
|
||
|
if (!_av1_bit_writer_metadata_hdr_mdcv (&metadata->hdr_mdcv,
|
||
|
&bw, &have_space))
|
||
|
goto error;
|
||
|
break;
|
||
|
case GST_AV1_METADATA_TYPE_SCALABILITY:
|
||
|
if (!_av1_bit_writer_metadata_scalability (&metadata->scalability,
|
||
|
&bw, &have_space))
|
||
|
goto error;
|
||
|
break;
|
||
|
default:
|
||
|
GST_WARNING ("Unsupported metadata_type");
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
/* Add trailings. */
|
||
|
WRITE_BITS (&bw, 1, 1);
|
||
|
if (!gst_bit_writer_align_bytes (&bw, 0)) {
|
||
|
have_space = FALSE;
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
payload_size = gst_bit_writer_get_size (&bw);
|
||
|
g_assert (payload_size % 8 == 0);
|
||
|
payload_size /= 8;
|
||
|
payload_size -= header_size;
|
||
|
|
||
|
gst_bit_writer_reset (&bw);
|
||
|
|
||
|
if (size_field) {
|
||
|
if (!_av1_bit_writer_add_size_field (data, size,
|
||
|
header_size, payload_size, 0, &have_space))
|
||
|
goto error;
|
||
|
} else {
|
||
|
*size = header_size + payload_size;
|
||
|
}
|
||
|
|
||
|
return GST_AV1_BIT_WRITER_OK;
|
||
|
|
||
|
error:
|
||
|
gst_bit_writer_reset (&bw);
|
||
|
*size = 0;
|
||
|
return have_space ? GST_AV1_BIT_WRITER_INVALID_DATA :
|
||
|
GST_AV1_BIT_WRITER_NO_MORE_SPACE;
|
||
|
}
|