/* GStreamer
 *  Copyright (C) 2021 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 "gsth265bitwriter.h"
#include <gst/codecparsers/nalutils.h>
#include <gst/base/gstbitwriter.h>
#include <math.h>

#ifndef GST_DISABLE_GST_DEBUG
#define GST_CAT_DEFAULT gst_h265_debug_category_get()
static GstDebugCategory *
gst_h265_debug_category_get (void)
{
  static gsize cat_gonce = 0;

  if (g_once_init_enter (&cat_gonce)) {
    GstDebugCategory *cat = NULL;

    GST_DEBUG_CATEGORY_INIT (cat, "bitwriter_h265", 0,
        "h265 bitwriter library");

    g_once_init_leave (&cat_gonce, (gsize) cat);
  }

  return (GstDebugCategory *) cat_gonce;
}
#endif /* GST_DISABLE_GST_DEBUG */

/********************************  Utils ********************************/
#define SIGNED(val)    (2 * ABS(val) - ((val) > 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;
}

#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;                                                               \
  }

#define WRITE_UE_UNCHECK(bw, val)  _bs_write_ue (bw, val)

#ifdef WRITE_UE
#undef WRITE_UE
#endif
#define WRITE_UE(bw, val)                                                     \
  if (!(have_space = WRITE_UE_UNCHECK (bw, val)))                             \
    goto error;                                                               \

#define WRITE_UE_MAX(bw, val, max)                                            \
  if ((guint32) val > (max) || !(have_space = WRITE_UE_UNCHECK (bw, val)))    \
    goto error;

#define WRITE_SE(bw, val) WRITE_UE (bw, SIGNED (val))

#define WRITE_SE_RANGE(bw, val, min, max)                                     \
  if (val > max || val < min ||                                               \
      !(have_space = WRITE_UE_UNCHECK (bw, SIGNED (val))))                    \
    goto error;

#define WRITE_BYTES_UNCHECK(bw, ptr, nbytes)                                  \
  gst_bit_writer_put_bytes(bw, ptr, nbytes)

#ifdef WRITE_BYTES
#undef WRITE_BYTES
#endif
#define WRITE_BYTES(bw, ptr, nbytes)                                          \
  if (!(have_space = WRITE_BYTES_UNCHECK (bw, ptr, nbytes)))                  \
    goto error;

/*****************************  End of Utils ****************************/

#define EXTENDED_SAR 255

/**** Default scaling_lists according to Table 7-5 and 7-6 *****/
/* Table 7-5 */
static const guint8 default_scaling_list0[16] = {
  16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16,
  16, 16, 16, 16
};

/*  Combined the values in Table  7-6 to make the calculation easier
 *  Default scaling list of 8x8 and 16x16 matrices for matrixId = 0, 1 and 2
 *  Default scaling list of 32x32 matrix for matrixId = 0
 */
static const guint8 default_scaling_list1[64] = {
  16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 17, 16,
  17, 16, 17, 18, 17, 18, 18, 17, 18, 21, 19, 20,
  21, 20, 19, 21, 24, 22, 22, 24, 24, 22, 22, 24,
  25, 25, 27, 30, 27, 25, 25, 29, 31, 35, 35, 31,
  29, 36, 41, 44, 41, 36, 47, 54, 54, 47, 65, 70,
  65, 88, 88, 115
};

/*  Combined the values in Table 7-6 to make the calculation easier
 *  Default scaling list of 8x8 and 16x16 matrices for matrixId = 3, 4 and 5
 *  Default scaling list of 32x32 matrix for matrixId = 1
 */
static const guint8 default_scaling_list2[64] = {
  16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 17, 17,
  17, 17, 17, 18, 18, 18, 18, 18, 18, 20, 20, 20,
  20, 20, 20, 20, 24, 24, 24, 24, 24, 24, 24, 24,
  25, 25, 25, 25, 25, 25, 25, 28, 28, 28, 28, 28,
  28, 33, 33, 33, 33, 33, 41, 41, 41, 41, 54, 54,
  54, 71, 71, 91
};

static gboolean
_h265_bit_writer_profile_tier_level (const GstH265ProfileTierLevel * ptl,
    guint8 maxNumSubLayersMinus1, GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  guint i, j;

  GST_DEBUG ("writing profile_tier_level");

  WRITE_BITS (bw, ptl->profile_space, 2);
  WRITE_BITS (bw, ptl->tier_flag, 1);
  WRITE_BITS (bw, ptl->profile_idc, 5);

  for (j = 0; j < 32; j++)
    WRITE_BITS (bw, ptl->profile_compatibility_flag[j], 1);

  WRITE_BITS (bw, ptl->progressive_source_flag, 1);
  WRITE_BITS (bw, ptl->interlaced_source_flag, 1);
  WRITE_BITS (bw, ptl->non_packed_constraint_flag, 1);
  WRITE_BITS (bw, ptl->frame_only_constraint_flag, 1);

  if (ptl->profile_idc == 4 || ptl->profile_compatibility_flag[4] ||
      ptl->profile_idc == 5 || ptl->profile_compatibility_flag[5] ||
      ptl->profile_idc == 6 || ptl->profile_compatibility_flag[6] ||
      ptl->profile_idc == 7 || ptl->profile_compatibility_flag[7] ||
      ptl->profile_idc == 8 || ptl->profile_compatibility_flag[8] ||
      ptl->profile_idc == 9 || ptl->profile_compatibility_flag[9] ||
      ptl->profile_idc == 10 || ptl->profile_compatibility_flag[10] ||
      ptl->profile_idc == 11 || ptl->profile_compatibility_flag[11]) {
    WRITE_BITS (bw, ptl->max_12bit_constraint_flag, 1);
    WRITE_BITS (bw, ptl->max_10bit_constraint_flag, 1);
    WRITE_BITS (bw, ptl->max_8bit_constraint_flag, 1);
    WRITE_BITS (bw, ptl->max_422chroma_constraint_flag, 1);
    WRITE_BITS (bw, ptl->max_420chroma_constraint_flag, 1);
    WRITE_BITS (bw, ptl->max_monochrome_constraint_flag, 1);
    WRITE_BITS (bw, ptl->intra_constraint_flag, 1);
    WRITE_BITS (bw, ptl->one_picture_only_constraint_flag, 1);
    WRITE_BITS (bw, ptl->lower_bit_rate_constraint_flag, 1);

    if (ptl->profile_idc == 5 || ptl->profile_compatibility_flag[5] ||
        ptl->profile_idc == 9 || ptl->profile_compatibility_flag[9] ||
        ptl->profile_idc == 10 || ptl->profile_compatibility_flag[10] ||
        ptl->profile_idc == 11 || ptl->profile_compatibility_flag[11]) {
      WRITE_BITS (bw, ptl->max_14bit_constraint_flag, 1);
      /* general_reserved_zero_33bits */
      WRITE_BITS (bw, 0, 32);
      WRITE_BITS (bw, 0, 1);
    } else {
      /* general_reserved_zero_34bits */
      WRITE_BITS (bw, 0, 32);
      WRITE_BITS (bw, 0, 2);
    }
  } else if (ptl->profile_idc == 2 || ptl->profile_compatibility_flag[2]) {
    /* general_reserved_zero_7bits */
    WRITE_BITS (bw, 0, 7);
    WRITE_BITS (bw, ptl->one_picture_only_constraint_flag, 1);
    /* general_reserved_zero_35bits */
    WRITE_BITS (bw, 0, 32);
    WRITE_BITS (bw, 0, 3);
  } else {
    /* general_reserved_zero_43bits */
    WRITE_BITS (bw, 0, 32);
    WRITE_BITS (bw, 0, 11);
  }

  /* general_inbld_flag, just set to 0 */
  WRITE_BITS (bw, 0, 1);

  WRITE_BITS (bw, ptl->level_idc, 8);

  for (j = 0; j < maxNumSubLayersMinus1; j++) {
    if (ptl->sub_layer_profile_present_flag[j]) {
      GST_WARNING ("sub layer profile does not supported now");
      goto error;
    }
    WRITE_BITS (bw, ptl->sub_layer_profile_present_flag[j], 1);

    if (ptl->sub_layer_level_present_flag[j]) {
      GST_WARNING ("sub layer level does not supported now");
      goto error;
    }
    WRITE_BITS (bw, ptl->sub_layer_level_present_flag[j], 1);
  }

  if (maxNumSubLayersMinus1 > 0) {
    for (i = maxNumSubLayersMinus1; i < 8; i++)
      /* reserved_zero_2bits */
      WRITE_BITS (bw, 0, 2);
  }

  /* TODO: Add sub layers profiles. */

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("error to write profile_tier_level");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_sub_layer_hrd_parameters (const GstH265SubLayerHRDParams *
    sub_hrd, guint8 CpbCnt, guint8 sub_pic_hrd_params_present_flag,
    GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  guint i;

  GST_DEBUG ("writing \"subLayer HRD Parameters\"");

  for (i = 0; i <= CpbCnt; i++) {
    WRITE_UE_MAX (bw, sub_hrd->bit_rate_value_minus1[i], G_MAXUINT32 - 1);
    WRITE_UE_MAX (bw, sub_hrd->cpb_size_value_minus1[i], G_MAXUINT32 - 1);

    if (sub_pic_hrd_params_present_flag) {
      WRITE_UE_MAX (bw, sub_hrd->cpb_size_du_value_minus1[i], G_MAXUINT32 - 1);
      WRITE_UE_MAX (bw, sub_hrd->bit_rate_du_value_minus1[i], G_MAXUINT32 - 1);
    }

    WRITE_BITS (bw, sub_hrd->cbr_flag[i], 1);
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("error to write \"subLayer HRD Parameters \"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_hrd_parameters (const GstH265HRDParams * hrd,
    guint8 commonInfPresentFlag, guint8 maxNumSubLayersMinus1,
    GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  guint i;

  GST_DEBUG ("writing \"HRD Parameters\"");

  if (commonInfPresentFlag) {
    WRITE_BITS (bw, hrd->nal_hrd_parameters_present_flag, 1);
    WRITE_BITS (bw, hrd->vcl_hrd_parameters_present_flag, 1);

    if (hrd->nal_hrd_parameters_present_flag
        || hrd->vcl_hrd_parameters_present_flag) {
      WRITE_BITS (bw, hrd->sub_pic_hrd_params_present_flag, 1);

      if (hrd->sub_pic_hrd_params_present_flag) {
        WRITE_BITS (bw, hrd->tick_divisor_minus2, 8);
        WRITE_BITS (bw, hrd->du_cpb_removal_delay_increment_length_minus1, 5);
        WRITE_BITS (bw, hrd->sub_pic_cpb_params_in_pic_timing_sei_flag, 1);
        WRITE_BITS (bw, hrd->dpb_output_delay_du_length_minus1, 5);
      }

      WRITE_BITS (bw, hrd->bit_rate_scale, 4);
      WRITE_BITS (bw, hrd->cpb_size_scale, 4);

      if (hrd->sub_pic_hrd_params_present_flag)
        WRITE_BITS (bw, hrd->cpb_size_du_scale, 4);

      WRITE_BITS (bw, hrd->initial_cpb_removal_delay_length_minus1, 5);
      WRITE_BITS (bw, hrd->au_cpb_removal_delay_length_minus1, 5);
      WRITE_BITS (bw, hrd->dpb_output_delay_length_minus1, 5);
    }
  }

  for (i = 0; i <= maxNumSubLayersMinus1; i++) {
    WRITE_BITS (bw, hrd->fixed_pic_rate_general_flag[i], 1);

    if (!hrd->fixed_pic_rate_general_flag[i]) {
      WRITE_BITS (bw, hrd->fixed_pic_rate_within_cvs_flag[i], 1);
    } else {
      if (hrd->fixed_pic_rate_within_cvs_flag[i] == 0)
        goto error;
    }

    if (hrd->fixed_pic_rate_within_cvs_flag[i]) {
      WRITE_UE_MAX (bw, hrd->elemental_duration_in_tc_minus1[i], 2047);
    } else {
      WRITE_BITS (bw, hrd->low_delay_hrd_flag[i], 1);
    }

    if (!hrd->low_delay_hrd_flag[i])
      WRITE_UE_MAX (bw, hrd->cpb_cnt_minus1[i], 31);

    if (hrd->nal_hrd_parameters_present_flag)
      if (!_h265_bit_writer_sub_layer_hrd_parameters
          (&hrd->sublayer_hrd_params[i], hrd->cpb_cnt_minus1[i],
              hrd->sub_pic_hrd_params_present_flag, bw, &have_space))
        goto error;

    /* TODO: need to separate nal and vcl from hrd_parameters. */
    if (hrd->vcl_hrd_parameters_present_flag)
      if (!_h265_bit_writer_sub_layer_hrd_parameters
          (&hrd->sublayer_hrd_params[i], hrd->cpb_cnt_minus1[i],
              hrd->sub_pic_hrd_params_present_flag, bw, &have_space))
        goto error;
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("error to write \"HRD Parameters\"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_vps (const GstH265VPS * vps, GstBitWriter * bw,
    gboolean * space)
{
  gboolean have_space = TRUE;
  guint i, j;

  GST_DEBUG ("writing VPS");

  WRITE_BITS (bw, vps->id, 4);

  WRITE_BITS (bw, vps->base_layer_internal_flag, 1);
  WRITE_BITS (bw, vps->base_layer_available_flag, 1);

  WRITE_BITS (bw, vps->max_layers_minus1, 6);
  WRITE_BITS (bw, vps->max_sub_layers_minus1, 3);
  WRITE_BITS (bw, vps->temporal_id_nesting_flag, 1);

  /* reserved_0xffff_16bits */
  WRITE_BITS (bw, 0xffff, 16);

  if (!_h265_bit_writer_profile_tier_level (&vps->profile_tier_level,
          vps->max_sub_layers_minus1, bw, &have_space))
    goto error;

  WRITE_BITS (bw, vps->sub_layer_ordering_info_present_flag, 1);

  for (i = (vps->sub_layer_ordering_info_present_flag ? 0 :
          vps->max_sub_layers_minus1); i <= vps->max_sub_layers_minus1; i++) {
    WRITE_UE (bw, vps->max_dec_pic_buffering_minus1[i]);
    WRITE_UE_MAX (bw, vps->max_num_reorder_pics[i],
        vps->max_dec_pic_buffering_minus1[i]);
    WRITE_UE_MAX (bw, vps->max_latency_increase_plus1[i], G_MAXUINT32 - 1);
  }

  /* max_layer_id should be <63, but only support 1 layer now. */
  if (vps->max_layer_id > 1) {
    GST_WARNING ("multi layers are not supported now");
    goto error;
  }

  WRITE_BITS (bw, vps->max_layer_id, 6);

  if (vps->num_layer_sets_minus1 >= 1) {
    GST_WARNING ("layer set is not supported now");
    goto error;
  }
  WRITE_UE_MAX (bw, vps->num_layer_sets_minus1, 1023);

  /* TODO: support multi-layer. */
  for (i = 1; i <= vps->num_layer_sets_minus1; i++) {
    for (j = 0; j <= vps->max_layer_id; j++) {
      /* layer_id_included_flag[i][j] */
      WRITE_BITS (bw, 0, 1);
    }
  }

  WRITE_BITS (bw, vps->timing_info_present_flag, 1);
  if (vps->timing_info_present_flag) {
    WRITE_BITS (bw, vps->num_units_in_tick, 32);
    WRITE_BITS (bw, vps->time_scale, 32);
    WRITE_BITS (bw, vps->poc_proportional_to_timing_flag, 1);

    if (vps->poc_proportional_to_timing_flag)
      WRITE_UE_MAX (bw, vps->num_ticks_poc_diff_one_minus1, G_MAXUINT32 - 1);

    /* TODO: VPS can have multiple hrd parameters, and therefore hrd_params
     * should be an array (like Garray). Just support 1 hdr parameter now.
     */
    if (vps->num_hrd_parameters > 1) {
      GST_WARNING ("HRD parameters > 1 is not supported now");
      goto error;
    }
    WRITE_UE_MAX (bw, vps->num_hrd_parameters, vps->num_layer_sets_minus1 + 1);

    if (vps->num_hrd_parameters) {
      WRITE_UE_MAX (bw, vps->hrd_layer_set_idx, vps->num_layer_sets_minus1);

      if (!_h265_bit_writer_hrd_parameters (&vps->hrd_params,
              vps->cprms_present_flag, vps->max_sub_layers_minus1,
              bw, &have_space))
        goto error;
    }

  }

  if (vps->vps_extension) {
    GST_WARNING ("vps extension is not supported now");
    goto error;
  }
  WRITE_BITS (bw, 0, 1);

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("failed to write VPS");

  *space = have_space;
  return FALSE;
}

/**
 * gst_h265_bit_writer_vps:
 * @vps: the vps of #GstH265VPS to write
 * @start_code: whether adding the nal start code
 * @data: (out): the bit stream generated by the sps
 * @size: (inout): the size in bytes of the input and output
 *
 * Generating the according h265 bit stream by providing the vps.
 *
 * Returns: a #GstH265BitWriterResult
 *
 * Since: 1.22
 **/
GstH265BitWriterResult
gst_h265_bit_writer_vps (const GstH265VPS * vps, gboolean start_code,
    guint8 * data, guint * size)
{
  gboolean have_space = TRUE;
  GstBitWriter bw;

  g_return_val_if_fail (vps != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);

  gst_bit_writer_init_with_data (&bw, data, *size, FALSE);

  if (start_code)
    WRITE_BITS (&bw, 0x00000001, 32);

  /* NAL unit header */
  /* forbidden_zero_bit */
  WRITE_BITS (&bw, 0, 1);
  /* nal_unit_type */
  WRITE_BITS (&bw, GST_H265_NAL_VPS, 6);
  /* nuh_layer_id, only support 0 now */
  WRITE_BITS (&bw, 0, 6);
  /* nuh_temporal_id_plus1, only support 1 now */
  WRITE_BITS (&bw, 1, 3);

  if (!_h265_bit_writer_vps (vps, &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;
  }

  *size = gst_bit_writer_get_size (&bw) / 8;
  gst_bit_writer_reset (&bw);

  return GST_H265_BIT_WRITER_OK;

error:
  gst_bit_writer_reset (&bw);
  *size = 0;

  if (!have_space)
    return GST_H265_BIT_WRITER_NO_MORE_SPACE;
  return GST_H265_BIT_WRITER_INVALID_DATA;
}

static gboolean
_get_scaling_list_params (const GstH265ScalingList * dest_scaling_list,
    guint8 sizeId, guint8 matrixId, const guint8 ** sl, guint8 * size,
    gint16 * scaling_list_dc_coef_minus8)
{
  switch (sizeId) {
    case GST_H265_QUANT_MATIX_4X4:
      *sl = dest_scaling_list->scaling_lists_4x4[matrixId];
      if (size)
        *size = 16;
      break;
    case GST_H265_QUANT_MATIX_8X8:
      *sl = dest_scaling_list->scaling_lists_8x8[matrixId];
      if (size)
        *size = 64;
      break;
    case GST_H265_QUANT_MATIX_16X16:
      *sl = dest_scaling_list->scaling_lists_16x16[matrixId];
      if (size)
        *size = 64;
      if (scaling_list_dc_coef_minus8)
        *scaling_list_dc_coef_minus8 =
            dest_scaling_list->scaling_list_dc_coef_minus8_16x16[matrixId];
      break;
    case GST_H265_QUANT_MATIX_32X32:
      *sl = dest_scaling_list->scaling_lists_32x32[matrixId];
      if (size)
        *size = 64;
      if (scaling_list_dc_coef_minus8)
        *scaling_list_dc_coef_minus8 =
            dest_scaling_list->scaling_list_dc_coef_minus8_32x32[matrixId];
      break;
    default:
      g_assert_not_reached ();
      return FALSE;
  }

  return TRUE;
}

static const guint8 *
_get_default_scaling_lists (GstH265QuantMatrixSize sizeId, guint8 matrixId)
{
  const guint8 *sl;

  switch (sizeId) {
    case GST_H265_QUANT_MATIX_4X4:
      sl = default_scaling_list0;
      break;

    case GST_H265_QUANT_MATIX_8X8:
    case GST_H265_QUANT_MATIX_16X16:
      if (matrixId <= 2) {
        sl = default_scaling_list1;
      } else {
        sl = default_scaling_list2;
      }
      break;

    case GST_H265_QUANT_MATIX_32X32:
      if (matrixId == 0) {
        sl = default_scaling_list1;
      } else {
        sl = default_scaling_list2;
      }
      break;

    default:
      g_assert_not_reached ();
      return NULL;
  }

  return sl;
}

static gboolean
_compare_scaling_list_matrix (GstH265QuantMatrixSize sizeId,
    const guint8 * sl0, const guint8 * sl1,
    gint16 dc_coef_minus8_0, gint16 dc_coef_minus8_1)
{
  guint size = sizeId == GST_H265_QUANT_MATIX_4X4 ? 16 : 64;

  if (memcmp (sl0, sl1, size * sizeof (guint8)))
    return FALSE;

  if (sizeId <= GST_H265_QUANT_MATIX_8X8)
    return TRUE;

  return dc_coef_minus8_0 == dc_coef_minus8_1;
}

static gboolean
_h265_bit_writer_scaling_lists (const GstH265ScalingList * src_scaling_list,
    GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  GstH265QuantMatrixSize sizeId;
  guint8 matrixId;
  guint8 scaling_list_pred_mode_flag = 0;
  guint8 size, i, j;

  GST_DEBUG ("writing scaling lists");

  for (sizeId = 0; sizeId <= GST_H265_QUANT_MATIX_32X32; sizeId++) {
    for (matrixId = 0;
        matrixId < ((sizeId == GST_H265_QUANT_MATIX_32X32) ? 2 : 6);
        matrixId++) {
      gint16 scaling_list_dc_coef_minus8 = 8;
      const guint8 *sl;
      const guint8 *default_sl;
      guint8 nextCoef;
      guint8 scaling_list_pred_matrix_id_delta;

      if (!_get_scaling_list_params (src_scaling_list, sizeId, matrixId,
              &sl, &size, &scaling_list_dc_coef_minus8))
        goto error;

      /* Check whether it is the default matrix. */
      default_sl = _get_default_scaling_lists (sizeId, matrixId);
      if (_compare_scaling_list_matrix (sizeId, sl, default_sl,
              scaling_list_dc_coef_minus8, 8)) {
        scaling_list_pred_mode_flag = 0;
        WRITE_BITS (bw, scaling_list_pred_mode_flag, 1);
        scaling_list_pred_matrix_id_delta = 0;
        WRITE_UE_MAX (bw, scaling_list_pred_matrix_id_delta, matrixId);
        continue;
      }

      /* If some previous matrix is the same, just ref it. */
      scaling_list_pred_matrix_id_delta = 0;
      for (j = 0; j < matrixId; j++) {
        gboolean ret;
        guint8 size2;
        const guint8 *sl2;
        gint16 scaling_list_dc_coef_minus8_2 = 8;

        ret = _get_scaling_list_params (src_scaling_list, sizeId, j,
            &sl2, &size2, &scaling_list_dc_coef_minus8_2);
        g_assert (ret);
        g_assert (size == size2);

        if (_compare_scaling_list_matrix (sizeId, sl, sl2,
                scaling_list_dc_coef_minus8, scaling_list_dc_coef_minus8_2)) {
          scaling_list_pred_matrix_id_delta = matrixId - j;
          break;
        }
      }

      if (scaling_list_pred_matrix_id_delta > 0) {
        scaling_list_pred_mode_flag = 0;
        WRITE_BITS (bw, scaling_list_pred_mode_flag, 1);
        WRITE_UE_MAX (bw, scaling_list_pred_matrix_id_delta, matrixId);
        continue;
      }

      /* Just explicitly signal all matrix coef. */
      scaling_list_pred_mode_flag = 1;
      WRITE_BITS (bw, scaling_list_pred_mode_flag, 1);

      nextCoef = 8;

      if (sizeId > 1) {
        WRITE_SE_RANGE (bw, scaling_list_dc_coef_minus8, -7, 247);
        nextCoef = scaling_list_dc_coef_minus8 + 8;
      }

      for (i = 0; i < size; i++) {
        gint coef_val = sl[i] - nextCoef;
        nextCoef = sl[i];

        if (coef_val > 127) {
          coef_val = coef_val - 256;
        }
        if (coef_val < -128) {
          coef_val = coef_val + 256;
        }

        WRITE_SE_RANGE (bw, coef_val, -128, 127);
      }
    }
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("error to write scaling lists");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_short_term_ref_pic_set (const GstH265ShortTermRefPicSet *
    stRPS, guint8 stRpsIdx, const GstH265SPS * sps,
    GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  gint32 prev;
  gint i = 0;

  GST_DEBUG ("writing \"ShortTermRefPicSetParameter\"");

  if (stRPS->inter_ref_pic_set_prediction_flag) {
    /* TODO */
    GST_WARNING ("inter_ref_pic_set_prediction_flag mode not supported");
    goto error;
  }

  if (stRpsIdx != 0)
    WRITE_BITS (bw, stRPS->inter_ref_pic_set_prediction_flag, 1);

  if (stRPS->NumNegativePics + stRPS->NumPositivePics != stRPS->NumDeltaPocs)
    goto error;

  /* 7-49 */
  WRITE_UE_MAX (bw, stRPS->NumNegativePics,
      sps->max_dec_pic_buffering_minus1[sps->max_sub_layers_minus1]);
  /* 7-50 */
  WRITE_UE_MAX (bw, stRPS->NumPositivePics,
      (sps->max_dec_pic_buffering_minus1[sps->max_sub_layers_minus1] -
          stRPS->NumNegativePics));

  prev = 0;
  for (i = 0; i < stRPS->NumNegativePics; i++) {
    WRITE_UE_MAX (bw, prev - stRPS->DeltaPocS0[i] - 1, 32767);
    prev = stRPS->DeltaPocS0[i];
    /* 7-51 */
    WRITE_BITS (bw, stRPS->UsedByCurrPicS0[i], 1);
  }

  prev = 0;
  for (i = 0; i < stRPS->NumPositivePics; i++) {
    WRITE_UE_MAX (bw, stRPS->DeltaPocS1[i] - prev - 1, 32767);
    prev = stRPS->DeltaPocS1[i];
    /* 7-52 */
    WRITE_BITS (bw, stRPS->UsedByCurrPicS1[i], 1);
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("error to write \"ShortTermRefPicSet Parameters\"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_vui_parameters (const GstH265SPS * sps,
    GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  const GstH265VUIParams *vui = &sps->vui_params;

  GST_DEBUG ("writing \"VUI Parameters\"");

  WRITE_BITS (bw, vui->aspect_ratio_info_present_flag, 1);
  if (vui->aspect_ratio_info_present_flag) {
    WRITE_BITS (bw, vui->aspect_ratio_idc, 8);
    if (vui->aspect_ratio_idc == EXTENDED_SAR) {
      WRITE_BITS (bw, vui->sar_width, 16);
      WRITE_BITS (bw, vui->sar_height, 16);
    }
  }

  WRITE_BITS (bw, vui->overscan_info_present_flag, 1);
  if (vui->overscan_info_present_flag)
    WRITE_BITS (bw, vui->overscan_appropriate_flag, 1);

  WRITE_BITS (bw, vui->video_signal_type_present_flag, 1);
  if (vui->video_signal_type_present_flag) {
    WRITE_BITS (bw, vui->video_format, 3);
    WRITE_BITS (bw, vui->video_full_range_flag, 1);
    WRITE_BITS (bw, vui->colour_description_present_flag, 1);
    if (vui->colour_description_present_flag) {
      WRITE_BITS (bw, vui->colour_primaries, 8);
      WRITE_BITS (bw, vui->transfer_characteristics, 8);
      WRITE_BITS (bw, vui->matrix_coefficients, 8);
    }
  }

  WRITE_BITS (bw, vui->chroma_loc_info_present_flag, 1);
  if (vui->chroma_loc_info_present_flag) {
    WRITE_UE_MAX (bw, vui->chroma_sample_loc_type_top_field, 5);
    WRITE_UE_MAX (bw, vui->chroma_sample_loc_type_bottom_field, 5);
  }

  WRITE_BITS (bw, vui->neutral_chroma_indication_flag, 1);
  WRITE_BITS (bw, vui->field_seq_flag, 1);
  WRITE_BITS (bw, vui->frame_field_info_present_flag, 1);

  WRITE_BITS (bw, vui->default_display_window_flag, 1);
  if (vui->default_display_window_flag) {
    WRITE_UE (bw, vui->def_disp_win_left_offset);
    WRITE_UE (bw, vui->def_disp_win_right_offset);
    WRITE_UE (bw, vui->def_disp_win_top_offset);
    WRITE_UE (bw, vui->def_disp_win_bottom_offset);
  }

  WRITE_BITS (bw, vui->timing_info_present_flag, 1);
  if (vui->timing_info_present_flag) {
    if (vui->num_units_in_tick == 0)
      GST_WARNING ("num_units_in_tick = 0 (incompliant to H.265 E.2.1).");
    WRITE_BITS (bw, vui->num_units_in_tick, 32);

    if (vui->time_scale == 0)
      GST_WARNING ("time_scale = 0 (incompliant to H.265 E.2.1).");
    WRITE_BITS (bw, vui->time_scale, 32);

    WRITE_BITS (bw, vui->poc_proportional_to_timing_flag, 1);
    if (vui->poc_proportional_to_timing_flag)
      WRITE_UE_MAX (bw, vui->num_ticks_poc_diff_one_minus1, G_MAXUINT32 - 1);

    WRITE_BITS (bw, vui->hrd_parameters_present_flag, 1);
    if (vui->hrd_parameters_present_flag)
      if (!_h265_bit_writer_hrd_parameters (&vui->hrd_params, 1,
              sps->max_sub_layers_minus1, bw, &have_space))
        goto error;
  }

  WRITE_BITS (bw, vui->bitstream_restriction_flag, 1);
  if (vui->bitstream_restriction_flag) {
    WRITE_BITS (bw, vui->tiles_fixed_structure_flag, 1);
    WRITE_BITS (bw, vui->motion_vectors_over_pic_boundaries_flag, 1);
    WRITE_BITS (bw, vui->restricted_ref_pic_lists_flag, 1);
    WRITE_UE_MAX (bw, vui->min_spatial_segmentation_idc, 4096);
    WRITE_UE_MAX (bw, vui->max_bytes_per_pic_denom, 16);
    WRITE_UE_MAX (bw, vui->max_bits_per_min_cu_denom, 16);
    WRITE_UE_MAX (bw, vui->log2_max_mv_length_horizontal, 16);
    WRITE_UE_MAX (bw, vui->log2_max_mv_length_vertical, 15);
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("error to write \"VUI Parameters\"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_sps (const GstH265SPS * sps,
    GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  guint i;

  GST_DEBUG ("writing SPS");

  WRITE_BITS (bw, sps->vps->id, 4);

  WRITE_BITS (bw, sps->max_sub_layers_minus1, 3);
  WRITE_BITS (bw, sps->temporal_id_nesting_flag, 1);

  if (!_h265_bit_writer_profile_tier_level (&sps->profile_tier_level,
          sps->max_sub_layers_minus1, bw, &have_space))
    goto error;

  WRITE_UE_MAX (bw, sps->id, GST_H265_MAX_SPS_COUNT - 1);

  WRITE_UE_MAX (bw, sps->chroma_format_idc, 3);
  if (sps->chroma_format_idc == 3)
    WRITE_BITS (bw, sps->separate_colour_plane_flag, 1);

  if (sps->pic_width_in_luma_samples < 1)
    goto error;
  WRITE_UE_MAX (bw, sps->pic_width_in_luma_samples, 16888);

  if (sps->pic_height_in_luma_samples < 1)
    goto error;
  WRITE_UE_MAX (bw, sps->pic_height_in_luma_samples, 16888);

  WRITE_BITS (bw, sps->conformance_window_flag, 1);
  if (sps->conformance_window_flag) {
    WRITE_UE (bw, sps->conf_win_left_offset);
    WRITE_UE (bw, sps->conf_win_right_offset);
    WRITE_UE (bw, sps->conf_win_top_offset);
    WRITE_UE (bw, sps->conf_win_bottom_offset);
  }

  WRITE_UE_MAX (bw, sps->bit_depth_luma_minus8, 6);
  WRITE_UE_MAX (bw, sps->bit_depth_chroma_minus8, 6);
  WRITE_UE_MAX (bw, sps->log2_max_pic_order_cnt_lsb_minus4, 12);

  WRITE_BITS (bw, sps->sub_layer_ordering_info_present_flag, 1);
  for (i = (sps->sub_layer_ordering_info_present_flag ? 0 :
          sps->max_sub_layers_minus1); i <= sps->max_sub_layers_minus1; i++) {
    WRITE_UE_MAX (bw, sps->max_dec_pic_buffering_minus1[i], 16);
    WRITE_UE_MAX (bw, sps->max_num_reorder_pics[i],
        sps->max_dec_pic_buffering_minus1[i]);
    WRITE_UE (bw, sps->max_latency_increase_plus1[i]);
  }

  /* The limits are calculted based on the profile_tier_level constraint
   * in Annex-A: CtbLog2SizeY = 4 to 6 */
  WRITE_UE_MAX (bw, sps->log2_min_luma_coding_block_size_minus3, 3);
  WRITE_UE_MAX (bw, sps->log2_diff_max_min_luma_coding_block_size, 6);
  WRITE_UE_MAX (bw, sps->log2_min_transform_block_size_minus2, 3);
  WRITE_UE_MAX (bw, sps->log2_diff_max_min_transform_block_size, 3);
  WRITE_UE_MAX (bw, sps->max_transform_hierarchy_depth_inter, 4);
  WRITE_UE_MAX (bw, sps->max_transform_hierarchy_depth_intra, 4);

  WRITE_BITS (bw, sps->scaling_list_enabled_flag, 1);
  if (sps->scaling_list_enabled_flag) {
    WRITE_BITS (bw, sps->scaling_list_data_present_flag, 1);

    if (sps->scaling_list_data_present_flag)
      if (!_h265_bit_writer_scaling_lists (&sps->scaling_list, bw, &have_space))
        goto error;
  }

  WRITE_BITS (bw, sps->amp_enabled_flag, 1);
  WRITE_BITS (bw, sps->sample_adaptive_offset_enabled_flag, 1);
  WRITE_BITS (bw, sps->pcm_enabled_flag, 1);

  if (sps->pcm_enabled_flag) {
    WRITE_BITS (bw, sps->pcm_sample_bit_depth_luma_minus1, 4);
    WRITE_BITS (bw, sps->pcm_sample_bit_depth_chroma_minus1, 4);
    WRITE_UE_MAX (bw, sps->log2_min_pcm_luma_coding_block_size_minus3, 2);
    WRITE_UE_MAX (bw, sps->log2_diff_max_min_pcm_luma_coding_block_size, 2);
    WRITE_BITS (bw, sps->pcm_loop_filter_disabled_flag, 1);
  }

  WRITE_UE_MAX (bw, sps->num_short_term_ref_pic_sets, 64);
  for (i = 0; i < sps->num_short_term_ref_pic_sets; i++) {
    if (!_h265_bit_writer_short_term_ref_pic_set
        (&sps->short_term_ref_pic_set[i], i, sps, bw, &have_space))
      goto error;
  }

  WRITE_BITS (bw, sps->long_term_ref_pics_present_flag, 1);
  if (sps->long_term_ref_pics_present_flag) {
    WRITE_UE_MAX (bw, sps->num_long_term_ref_pics_sps, 32);
    for (i = 0; i < sps->num_long_term_ref_pics_sps; i++) {
      WRITE_BITS (bw, sps->lt_ref_pic_poc_lsb_sps[i],
          sps->log2_max_pic_order_cnt_lsb_minus4 + 4);
      WRITE_BITS (bw, sps->used_by_curr_pic_lt_sps_flag[i], 1);
    }
  }

  WRITE_BITS (bw, sps->temporal_mvp_enabled_flag, 1);
  WRITE_BITS (bw, sps->strong_intra_smoothing_enabled_flag, 1);
  WRITE_BITS (bw, sps->vui_parameters_present_flag, 1);

  if (sps->vui_parameters_present_flag) {
    if (!_h265_bit_writer_vui_parameters (sps, bw, &have_space))
      goto error;
  }

  WRITE_BITS (bw, sps->sps_extension_flag, 1);

  if (sps->sps_extension_flag) {
    WRITE_BITS (bw, sps->sps_range_extension_flag, 1);
    WRITE_BITS (bw, sps->sps_multilayer_extension_flag, 1);
    WRITE_BITS (bw, sps->sps_3d_extension_flag, 1);
    WRITE_BITS (bw, sps->sps_scc_extension_flag, 1);
    WRITE_BITS (bw, sps->sps_extension_4bits, 4);
  }

  if (sps->sps_range_extension_flag) {
    WRITE_BITS (bw,
        sps->sps_extension_params.transform_skip_rotation_enabled_flag, 1);
    WRITE_BITS (bw,
        sps->sps_extension_params.transform_skip_context_enabled_flag, 1);
    WRITE_BITS (bw, sps->sps_extension_params.implicit_rdpcm_enabled_flag, 1);
    WRITE_BITS (bw, sps->sps_extension_params.explicit_rdpcm_enabled_flag, 1);
    WRITE_BITS (bw,
        sps->sps_extension_params.extended_precision_processing_flag, 1);
    WRITE_BITS (bw, sps->sps_extension_params.intra_smoothing_disabled_flag, 1);
    WRITE_BITS (bw,
        sps->sps_extension_params.high_precision_offsets_enabled_flag, 1);
    WRITE_BITS (bw,
        sps->sps_extension_params.persistent_rice_adaptation_enabled_flag, 1);
    WRITE_BITS (bw,
        sps->sps_extension_params.cabac_bypass_alignment_enabled_flag, 1);
  }

  if (sps->sps_multilayer_extension_flag) {
    GST_WARNING ("do not support multilayer extension");
    goto error;
  }
  if (sps->sps_3d_extension_flag) {
    GST_WARNING ("do not support 3d extension");
    goto error;
  }

  if (sps->sps_scc_extension_flag) {
    const GstH265SPSSccExtensionParams *scc_params =
        &sps->sps_scc_extension_params;

    WRITE_BITS (bw, scc_params->sps_curr_pic_ref_enabled_flag, 1);
    WRITE_BITS (bw, scc_params->palette_mode_enabled_flag, 1);
    if (scc_params->palette_mode_enabled_flag) {
      WRITE_UE_MAX (bw, scc_params->palette_max_size, 64);
      WRITE_UE_MAX (bw, scc_params->delta_palette_max_predictor_size,
          128 - scc_params->palette_max_size);

      WRITE_BITS (bw,
          scc_params->sps_palette_predictor_initializers_present_flag, 1);
      if (scc_params->sps_palette_predictor_initializers_present_flag) {
        guint comp;
        WRITE_UE_MAX (bw,
            scc_params->sps_num_palette_predictor_initializer_minus1,
            scc_params->palette_max_size +
            scc_params->delta_palette_max_predictor_size - 1);

        for (comp = 0; comp < (sps->chroma_format_idc == 0 ? 1 : 3); comp++) {
          guint num_bits;
          guint num =
              scc_params->sps_num_palette_predictor_initializer_minus1 + 1;

          num_bits = (comp == 0 ? sps->bit_depth_luma_minus8 + 8 :
              sps->bit_depth_chroma_minus8 + 8);
          for (i = 0; i < num; i++)
            WRITE_BITS (bw,
                scc_params->sps_palette_predictor_initializer[comp][i],
                num_bits);
        }
      }
    }

    WRITE_BITS (bw, scc_params->motion_vector_resolution_control_idc, 2);
    WRITE_BITS (bw, scc_params->intra_boundary_filtering_disabled_flag, 1);
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("failed to write SPS");

  *space = have_space;
  return FALSE;
}

/**
 * gst_h265_bit_writer_sps:
 * @sps: the sps of #GstH265SPS to write
 * @start_code: whether adding the nal start code
 * @data: (out): the bit stream generated by the sps
 * @size: (inout): the size in bytes of the input and output
 *
 * Generating the according h265 bit stream by providing the sps.
 *
 * Returns: a #GstH265BitWriterResult
 *
 * Since: 1.22
 **/
GstH265BitWriterResult
gst_h265_bit_writer_sps (const GstH265SPS * sps, gboolean start_code,
    guint8 * data, guint * size)
{
  gboolean have_space = TRUE;
  GstBitWriter bw;

  g_return_val_if_fail (sps != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (sps->vps != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);

  gst_bit_writer_init_with_data (&bw, data, *size, FALSE);

  if (start_code)
    WRITE_BITS (&bw, 0x00000001, 32);

  /* NAL unit header */
  /* forbidden_zero_bit */
  WRITE_BITS (&bw, 0, 1);
  /* nal_unit_type */
  WRITE_BITS (&bw, GST_H265_NAL_SPS, 6);
  /* nuh_layer_id, only support 0 now */
  WRITE_BITS (&bw, 0, 6);
  /* nuh_temporal_id_plus1, only support 1 now */
  WRITE_BITS (&bw, 1, 3);

  if (!_h265_bit_writer_sps (sps, &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;
  }

  *size = gst_bit_writer_get_size (&bw) / 8;
  gst_bit_writer_reset (&bw);

  return GST_H265_BIT_WRITER_OK;

error:
  gst_bit_writer_reset (&bw);
  *size = 0;

  if (!have_space)
    return GST_H265_BIT_WRITER_NO_MORE_SPACE;
  return GST_H265_BIT_WRITER_INVALID_DATA;
}

static gboolean
_h265_bit_writer_pps (const GstH265PPS * pps, GstBitWriter * bw,
    gboolean * space)
{
  gboolean have_space = TRUE;

  GST_DEBUG ("writing PPS");

  WRITE_UE_MAX (bw, pps->id, GST_H265_MAX_PPS_COUNT - 1);
  WRITE_UE_MAX (bw, pps->sps->id, GST_H265_MAX_SPS_COUNT - 1);

  WRITE_BITS (bw, pps->dependent_slice_segments_enabled_flag, 1);
  WRITE_BITS (bw, pps->output_flag_present_flag, 1);
  WRITE_BITS (bw, pps->num_extra_slice_header_bits, 3);
  WRITE_BITS (bw, pps->sign_data_hiding_enabled_flag, 1);
  WRITE_BITS (bw, pps->cabac_init_present_flag, 1);

  WRITE_UE_MAX (bw, pps->num_ref_idx_l0_default_active_minus1, 14);
  WRITE_UE_MAX (bw, pps->num_ref_idx_l1_default_active_minus1, 14);
  WRITE_SE_RANGE (bw, pps->init_qp_minus26,
      -(26 + 6 * pps->sps->bit_depth_luma_minus8), 25);

  WRITE_BITS (bw, pps->constrained_intra_pred_flag, 1);
  WRITE_BITS (bw, pps->transform_skip_enabled_flag, 1);

  WRITE_BITS (bw, pps->cu_qp_delta_enabled_flag, 1);
  if (pps->cu_qp_delta_enabled_flag)
    WRITE_UE_MAX (bw, pps->diff_cu_qp_delta_depth,
        pps->sps->log2_diff_max_min_luma_coding_block_size);

  WRITE_SE_RANGE (bw, pps->cb_qp_offset, -12, 12);
  WRITE_SE_RANGE (bw, pps->cr_qp_offset, -12, 12);

  WRITE_BITS (bw, pps->slice_chroma_qp_offsets_present_flag, 1);
  WRITE_BITS (bw, pps->weighted_pred_flag, 1);
  WRITE_BITS (bw, pps->weighted_bipred_flag, 1);
  WRITE_BITS (bw, pps->transquant_bypass_enabled_flag, 1);
  WRITE_BITS (bw, pps->tiles_enabled_flag, 1);
  WRITE_BITS (bw, pps->entropy_coding_sync_enabled_flag, 1);

  if (pps->tiles_enabled_flag) {
    if (pps->num_tile_columns_minus1 + 1 >
        G_N_ELEMENTS (pps->column_width_minus1)) {
      GST_WARNING ("Invalid \"num_tile_columns_minus1\" %d",
          pps->num_tile_columns_minus1);
      goto error;
    }

    if (pps->num_tile_rows_minus1 + 1 > G_N_ELEMENTS (pps->row_height_minus1)) {
      GST_WARNING ("Invalid \"num_tile_rows_minus1\" %d",
          pps->num_tile_rows_minus1);
      goto error;
    }

    WRITE_UE_MAX (bw, pps->num_tile_columns_minus1, pps->PicWidthInCtbsY - 1);
    WRITE_UE_MAX (bw, pps->num_tile_rows_minus1, pps->PicHeightInCtbsY - 1);

    WRITE_BITS (bw, pps->uniform_spacing_flag, 1);

    /* 6.5.1, 6-4, 6-5, 7.4.3.3.1 */
    if (!pps->uniform_spacing_flag) {
      guint i;

      for (i = 0; i < pps->num_tile_columns_minus1; i++)
        WRITE_UE (bw, pps->column_width_minus1[i]);

      for (i = 0; i < pps->num_tile_rows_minus1; i++)
        WRITE_UE (bw, pps->row_height_minus1[i]);
    }
    WRITE_BITS (bw, pps->loop_filter_across_tiles_enabled_flag, 1);
  }

  WRITE_BITS (bw, pps->loop_filter_across_slices_enabled_flag, 1);

  WRITE_BITS (bw, pps->deblocking_filter_control_present_flag, 1);
  if (pps->deblocking_filter_control_present_flag) {
    WRITE_BITS (bw, pps->deblocking_filter_override_enabled_flag, 1);

    WRITE_BITS (bw, pps->deblocking_filter_disabled_flag, 1);
    if (!pps->deblocking_filter_disabled_flag) {
      WRITE_SE_RANGE (bw, pps->beta_offset_div2, -6, 6);
      WRITE_SE_RANGE (bw, pps->tc_offset_div2, -6, +6);
    }
  }

  WRITE_BITS (bw, pps->scaling_list_data_present_flag, 1);
  if (pps->scaling_list_data_present_flag)
    if (!_h265_bit_writer_scaling_lists (&pps->scaling_list, bw, &have_space))
      goto error;

  WRITE_BITS (bw, pps->lists_modification_present_flag, 1);
  WRITE_UE_MAX (bw, pps->log2_parallel_merge_level_minus2, 4);

  /* TODO: slice_segment_header */
  if (pps->slice_segment_header_extension_present_flag) {
    GST_WARNING
        ("slice_segment_header_extension_present_flag is not supported");
    goto error;
  }
  WRITE_BITS (bw, pps->slice_segment_header_extension_present_flag, 1);

  WRITE_BITS (bw, pps->pps_extension_flag, 1);

  if (pps->pps_extension_flag) {
    WRITE_BITS (bw, pps->pps_range_extension_flag, 1);
    WRITE_BITS (bw, pps->pps_multilayer_extension_flag, 1);
    WRITE_BITS (bw, pps->pps_3d_extension_flag, 1);
    WRITE_BITS (bw, pps->pps_scc_extension_flag, 1);
    WRITE_BITS (bw, pps->pps_extension_4bits, 4);
  }

  if (pps->pps_range_extension_flag) {
    guint i;
    guint32 MaxBitDepthY, MaxBitDepthC;

    if (pps->transform_skip_enabled_flag)
      WRITE_UE (bw,
          pps->pps_extension_params.log2_max_transform_skip_block_size_minus2);

    WRITE_BITS (bw,
        pps->pps_extension_params.cross_component_prediction_enabled_flag, 1);
    WRITE_BITS (bw,
        pps->pps_extension_params.chroma_qp_offset_list_enabled_flag, 1);

    if (pps->pps_extension_params.chroma_qp_offset_list_enabled_flag) {
      WRITE_UE_MAX (bw,
          pps->pps_extension_params.diff_cu_chroma_qp_offset_depth,
          pps->sps->log2_diff_max_min_luma_coding_block_size);

      WRITE_UE_MAX (bw,
          pps->pps_extension_params.chroma_qp_offset_list_len_minus1, 5);
      for (i = 0;
          i <= pps->pps_extension_params.chroma_qp_offset_list_len_minus1;
          i++) {
        WRITE_SE_RANGE (bw, pps->pps_extension_params.cb_qp_offset_list[i],
            -12, 12);
        WRITE_SE_RANGE (bw, pps->pps_extension_params.cr_qp_offset_list[i],
            -12, 12);
      }
    }

    MaxBitDepthY = pps->sps->bit_depth_luma_minus8 > 2 ?
        pps->sps->bit_depth_luma_minus8 - 2 : 0;
    MaxBitDepthC = pps->sps->bit_depth_chroma_minus8 > 2 ?
        pps->sps->bit_depth_chroma_minus8 - 2 : 0;
    WRITE_UE_MAX (bw, pps->pps_extension_params.log2_sao_offset_scale_luma,
        MaxBitDepthY);
    WRITE_UE_MAX (bw, pps->pps_extension_params.log2_sao_offset_scale_chroma,
        MaxBitDepthC);
  }

  if (pps->pps_multilayer_extension_flag) {
    GST_WARNING ("do not support multilayer extension");
    goto error;
  }

  if (pps->pps_3d_extension_flag) {
    GST_WARNING ("do not support 3d extension");
    goto error;
  }

  if (pps->pps_scc_extension_flag) {
    const GstH265PPSSccExtensionParams *pps_scc =
        &pps->pps_scc_extension_params;

    WRITE_BITS (bw, pps_scc->pps_curr_pic_ref_enabled_flag, 1);
    WRITE_BITS (bw,
        pps_scc->residual_adaptive_colour_transform_enabled_flag, 1);
    if (pps_scc->residual_adaptive_colour_transform_enabled_flag) {
      WRITE_BITS (bw, pps_scc->pps_slice_act_qp_offsets_present_flag, 1);
      WRITE_SE_RANGE (bw, (gint8) pps_scc->pps_act_y_qp_offset_plus5, -7, 17);
      WRITE_SE_RANGE (bw, (gint8) pps_scc->pps_act_cb_qp_offset_plus5, -7, 17);
      WRITE_SE_RANGE (bw, (gint8) pps_scc->pps_act_cr_qp_offset_plus3, -9, 15);
    }

    WRITE_BITS (bw,
        pps_scc->pps_palette_predictor_initializers_present_flag, 1);
    if (pps_scc->pps_palette_predictor_initializers_present_flag) {
      guint i;

      WRITE_UE_MAX (bw,
          pps_scc->pps_num_palette_predictor_initializer,
          pps->sps->sps_scc_extension_params.palette_max_size +
          pps->sps->sps_scc_extension_params.delta_palette_max_predictor_size);
      if (pps_scc->pps_num_palette_predictor_initializer > 0) {
        guint comp;

        WRITE_BITS (bw, pps_scc->monochrome_palette_flag, 1);
        /* It is a requirement of bitstream conformance that the value of
           luma_bit_depth_entry_minus8 shall be equal to the value of
           bit_depth_luma_minus8 */
        WRITE_UE_MAX (bw, pps_scc->luma_bit_depth_entry_minus8,
            pps->sps->bit_depth_luma_minus8);
        if (!pps_scc->monochrome_palette_flag) {
          /* It is a requirement of bitstream conformance that the value
             of chroma_bit_depth_entry_minus8 shall be equal to the value
             of bit_depth_chroma_minus8. */
          WRITE_UE_MAX (bw, pps_scc->chroma_bit_depth_entry_minus8,
              pps->sps->bit_depth_chroma_minus8);
        }

        for (comp = 0; comp < (pps_scc->monochrome_palette_flag ? 1 : 3);
            comp++) {
          guint num_bits;
          guint num = pps_scc->pps_num_palette_predictor_initializer;

          num_bits = (comp == 0 ?
              pps_scc->luma_bit_depth_entry_minus8 + 8 :
              pps_scc->chroma_bit_depth_entry_minus8 + 8);
          for (i = 0; i < num; i++)
            WRITE_BITS (bw,
                pps_scc->pps_palette_predictor_initializer[comp][i], num_bits);
        }
      }
    }
  }

  return TRUE;

error:
  GST_WARNING ("failed to write PPS");
  return FALSE;
}

/**
 * gst_h265_bit_writer_pps:
 * @pps: the pps of #GstH265PPS to write
 * @start_code: whether adding the nal start code
 * @data: (out): the bit stream generated by the pps
 * @size: (inout): the size in bytes of the input and output
 *
 * Generating the according h265 bit stream by providing the pps.
 *
 * Returns: a #GstH265BitWriterResult
 *
 * Since: 1.22
 **/
GstH265BitWriterResult
gst_h265_bit_writer_pps (const GstH265PPS * pps, gboolean start_code,
    guint8 * data, guint * size)
{
  gboolean have_space = TRUE;
  GstBitWriter bw;

  g_return_val_if_fail (pps != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (pps->sps != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);

  gst_bit_writer_init_with_data (&bw, data, *size, FALSE);

  if (start_code)
    WRITE_BITS (&bw, 0x00000001, 32);

  /* NAL unit header */
  /* forbidden_zero_bit */
  WRITE_BITS (&bw, 0, 1);
  /* nal_unit_type */
  WRITE_BITS (&bw, GST_H265_NAL_PPS, 6);
  /* nuh_layer_id, only support 0 now */
  WRITE_BITS (&bw, 0, 6);
  /* nuh_temporal_id_plus1, only support 1 now */
  WRITE_BITS (&bw, 1, 3);

  if (!_h265_bit_writer_pps (pps, &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;
  }

  *size = gst_bit_writer_get_size (&bw) / 8;
  gst_bit_writer_reset (&bw);

  return GST_H265_BIT_WRITER_OK;

error:
  gst_bit_writer_reset (&bw);
  *size = 0;

  if (!have_space)
    return GST_H265_BIT_WRITER_NO_MORE_SPACE;
  return GST_H265_BIT_WRITER_INVALID_DATA;
}

static gboolean
_h265_slice_bit_writer_ref_pic_list_modification (const GstH265SliceHdr *
    slice, gint NumPocTotalCurr, GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  guint i;
  const GstH265RefPicListModification *rpl_mod =
      &slice->ref_pic_list_modification;
  const guint n = gst_util_ceil_log2 (NumPocTotalCurr);

  WRITE_BITS (bw, rpl_mod->ref_pic_list_modification_flag_l0, 1);

  if (rpl_mod->ref_pic_list_modification_flag_l0) {
    for (i = 0; i <= slice->num_ref_idx_l0_active_minus1; i++) {
      /* 7.4.7.2 list_entry_l0 */
      WRITE_BITS (bw, rpl_mod->list_entry_l0[i], n);
    }
  }

  if (GST_H265_IS_B_SLICE (slice)) {
    WRITE_BITS (bw, rpl_mod->ref_pic_list_modification_flag_l1, 1);

    if (rpl_mod->ref_pic_list_modification_flag_l1)
      for (i = 0; i <= slice->num_ref_idx_l1_active_minus1; i++) {
        WRITE_BITS (bw, rpl_mod->list_entry_l1[i], n);
      }
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("error to write \"Reference picture list modifications\"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_slice_bit_writer_pred_weight_table (const GstH265SliceHdr * slice,
    GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  const GstH265PredWeightTable *p;
  const GstH265PPS *pps = slice->pps;
  const GstH265SPS *sps = pps->sps;
  gint i, j;

  GST_DEBUG ("writing \"Prediction weight table\"");

  p = &slice->pred_weight_table;

  WRITE_UE_MAX (bw, p->luma_log2_weight_denom, 7);

  if (sps->chroma_format_idc != 0) {
    WRITE_SE_RANGE (bw, p->delta_chroma_log2_weight_denom,
        (0 - p->luma_log2_weight_denom), (7 - p->luma_log2_weight_denom));
  }

  for (i = 0; i <= slice->num_ref_idx_l0_active_minus1; i++)
    WRITE_BITS (bw, p->luma_weight_l0_flag[i], 1);

  if (sps->chroma_format_idc != 0)
    for (i = 0; i <= slice->num_ref_idx_l0_active_minus1; i++)
      WRITE_BITS (bw, p->chroma_weight_l0_flag[i], 1);

  for (i = 0; i <= slice->num_ref_idx_l0_active_minus1; i++) {
    if (p->luma_weight_l0_flag[i]) {
      WRITE_SE_RANGE (bw, p->delta_luma_weight_l0[i], -128, 127);
      WRITE_SE_RANGE (bw, p->luma_offset_l0[i], -128, 127);
    }
    if (p->chroma_weight_l0_flag[i])
      for (j = 0; j < 2; j++) {
        WRITE_SE_RANGE (bw, p->delta_chroma_weight_l0[i][j], -128, 127);
        WRITE_SE_RANGE (bw, p->delta_chroma_offset_l0[i][j], -512, 511);
      }
  }

  if (GST_H265_IS_B_SLICE (slice)) {
    for (i = 0; i <= slice->num_ref_idx_l1_active_minus1; i++)
      WRITE_BITS (bw, p->luma_weight_l1_flag[i], 1);

    if (sps->chroma_format_idc != 0)
      for (i = 0; i <= slice->num_ref_idx_l1_active_minus1; i++)
        WRITE_BITS (bw, p->chroma_weight_l1_flag[i], 1);

    for (i = 0; i <= slice->num_ref_idx_l1_active_minus1; i++) {
      if (p->luma_weight_l1_flag[i]) {
        WRITE_SE_RANGE (bw, p->delta_luma_weight_l1[i], -128, 127);
        WRITE_SE_RANGE (bw, p->luma_offset_l1[i], -128, 127);
      }
      if (p->chroma_weight_l1_flag[i])
        for (j = 0; j < 2; j++) {
          WRITE_SE_RANGE (bw, p->delta_chroma_weight_l1[i][j], -128, 127);
          WRITE_SE_RANGE (bw, p->delta_chroma_offset_l1[i][j], -512, 511);
        }
    }
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("error to write \"Prediction weight table\"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_slice_header (const GstH265SliceHdr * slice,
    guint32 nal_type, GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  guint i;
  const GstH265SPS *sps = slice->pps->sps;
  const GstH265PPSSccExtensionParams *pps_scc_extension_params =
      &slice->pps->pps_scc_extension_params;
  const GstH265PPSExtensionParams *pps_extension_params =
      &slice->pps->pps_extension_params;

  GST_DEBUG ("writing slice header");

  WRITE_BITS (bw, slice->first_slice_segment_in_pic_flag, 1);

  if (GST_H265_IS_NAL_TYPE_IRAP (nal_type))
    WRITE_BITS (bw, slice->no_output_of_prior_pics_flag, 1);

  WRITE_UE_MAX (bw, slice->pps->id, GST_H265_MAX_PPS_COUNT - 1);

  if (!slice->first_slice_segment_in_pic_flag) {
    guint32 PicSizeInCtbsY;
    guint32 PicWidthInCtbsY;
    guint32 PicHeightInCtbsY;
    guint32 CtbSizeY, MinCbLog2SizeY, CtbLog2SizeY;
    guint n;

    /* We can not directly use slice->pps->PicWidthInCtbsY/PicHeightInCtbsY,
       they are calculated value when parsing but may not have value here. */
    MinCbLog2SizeY = sps->log2_min_luma_coding_block_size_minus3 + 3;
    CtbLog2SizeY =
        MinCbLog2SizeY + sps->log2_diff_max_min_luma_coding_block_size;
    CtbSizeY = 1 << CtbLog2SizeY;
    PicHeightInCtbsY =
        ceil ((gdouble) sps->pic_height_in_luma_samples / (gdouble) CtbSizeY);
    PicWidthInCtbsY =
        ceil ((gdouble) sps->pic_width_in_luma_samples / (gdouble) CtbSizeY);
    PicSizeInCtbsY = PicWidthInCtbsY * PicHeightInCtbsY;

    n = gst_util_ceil_log2 (PicSizeInCtbsY);

    if (slice->pps->dependent_slice_segments_enabled_flag)
      WRITE_BITS (bw, slice->dependent_slice_segment_flag, 1);
    /* 7.4.7.1 slice_segment_address parsing */
    WRITE_BITS (bw, slice->segment_address, n);
  }

  if (!slice->dependent_slice_segment_flag) {
    for (i = 0; i < slice->pps->num_extra_slice_header_bits; i++) {
      /* slice_reserved_flag */
      WRITE_BITS (bw, 0, 1);
    }

    WRITE_UE_MAX (bw, slice->type, 63);

    if (slice->pps->output_flag_present_flag)
      WRITE_BITS (bw, slice->pic_output_flag, 1);

    if (sps->separate_colour_plane_flag == 1)
      WRITE_BITS (bw, slice->colour_plane_id, 2);

    if (!GST_H265_IS_NAL_TYPE_IDR (nal_type)) {
      WRITE_BITS (bw, slice->pic_order_cnt_lsb,
          (sps->log2_max_pic_order_cnt_lsb_minus4 + 4));

      WRITE_BITS (bw, slice->short_term_ref_pic_set_sps_flag, 1);
      if (!slice->short_term_ref_pic_set_sps_flag) {
        if (!_h265_bit_writer_short_term_ref_pic_set
            (&slice->short_term_ref_pic_sets, sps->num_short_term_ref_pic_sets,
                slice->pps->sps, bw, &have_space))
          goto error;
      } else if (sps->num_short_term_ref_pic_sets > 1) {
        const guint n = gst_util_ceil_log2 (sps->num_short_term_ref_pic_sets);

        if (slice->short_term_ref_pic_set_idx >
            sps->num_short_term_ref_pic_sets - 1)
          goto error;

        /*  7.4.7.1 short_term_ref_pic_set_idx */
        WRITE_BITS (bw, slice->short_term_ref_pic_set_idx, n);
      }

      if (sps->long_term_ref_pics_present_flag) {
        guint32 limit;

        if (sps->num_long_term_ref_pics_sps > 0)
          WRITE_UE_MAX (bw, slice->num_long_term_sps,
              sps->num_long_term_ref_pics_sps);

        WRITE_UE_MAX (bw, slice->num_long_term_pics, 16);
        limit = slice->num_long_term_sps + slice->num_long_term_pics;
        for (i = 0; i < limit; i++) {
          if (i < slice->num_long_term_sps) {
            if (sps->num_long_term_ref_pics_sps > 1) {
              /* 7.4.7.1 lt_idx_sps */
              const guint n =
                  gst_util_ceil_log2 (sps->num_long_term_ref_pics_sps);
              WRITE_BITS (bw, slice->lt_idx_sps[i], n);
            }
          } else {
            WRITE_BITS (bw, slice->poc_lsb_lt[i],
                (sps->log2_max_pic_order_cnt_lsb_minus4 + 4));
            WRITE_BITS (bw, slice->used_by_curr_pic_lt_flag[i], 1);
          }

          WRITE_BITS (bw, slice->delta_poc_msb_present_flag[i], 1);
          if (slice->delta_poc_msb_present_flag[i])
            WRITE_UE (bw, slice->delta_poc_msb_cycle_lt[i]);
        }
      }

      if (sps->temporal_mvp_enabled_flag)
        WRITE_BITS (bw, slice->temporal_mvp_enabled_flag, 1);
    }

    if (sps->sample_adaptive_offset_enabled_flag) {
      gboolean ChromaArrayType =
          sps->separate_colour_plane_flag == 0 ? sps->chroma_format_idc : 0;

      WRITE_BITS (bw, slice->sao_luma_flag, 1);
      if (ChromaArrayType)
        WRITE_BITS (bw, slice->sao_chroma_flag, 1);
    }

    if (GST_H265_IS_B_SLICE (slice) || GST_H265_IS_P_SLICE (slice)) {
      WRITE_BITS (bw, slice->num_ref_idx_active_override_flag, 1);

      if (slice->num_ref_idx_active_override_flag) {
        WRITE_UE_MAX (bw, slice->num_ref_idx_l0_active_minus1, 14);
        if (GST_H265_IS_B_SLICE (slice))
          WRITE_UE_MAX (bw, slice->num_ref_idx_l1_active_minus1, 14);
      }

      if (slice->pps->lists_modification_present_flag
          && slice->NumPocTotalCurr > 1) {
        if (!_h265_slice_bit_writer_ref_pic_list_modification (slice,
                slice->NumPocTotalCurr, bw, &have_space))
          goto error;
      }

      if (GST_H265_IS_B_SLICE (slice))
        WRITE_BITS (bw, slice->mvd_l1_zero_flag, 1);

      if (slice->pps->cabac_init_present_flag)
        WRITE_BITS (bw, slice->cabac_init_flag, 1);

      if (slice->temporal_mvp_enabled_flag) {
        if (GST_H265_IS_B_SLICE (slice))
          WRITE_BITS (bw, slice->collocated_from_l0_flag, 1);

        if ((slice->collocated_from_l0_flag
                && slice->num_ref_idx_l0_active_minus1 > 0)
            || (!slice->collocated_from_l0_flag
                && slice->num_ref_idx_l1_active_minus1 > 0)) {
          if ((GST_H265_IS_P_SLICE (slice))
              || ((GST_H265_IS_B_SLICE (slice))
                  && (slice->collocated_from_l0_flag))) {
            WRITE_UE_MAX (bw, slice->collocated_ref_idx,
                slice->num_ref_idx_l0_active_minus1);
          } else if ((GST_H265_IS_B_SLICE (slice))
              && (!slice->collocated_from_l0_flag)) {
            WRITE_UE_MAX (bw, slice->collocated_ref_idx,
                slice->num_ref_idx_l1_active_minus1);
          }
        }
      }

      if ((slice->pps->weighted_pred_flag && GST_H265_IS_P_SLICE (slice)) ||
          (slice->pps->weighted_bipred_flag && GST_H265_IS_B_SLICE (slice)))
        if (!_h265_slice_bit_writer_pred_weight_table (slice, bw, &have_space))
          goto error;

      WRITE_UE_MAX (bw, slice->five_minus_max_num_merge_cand, 4);

      if (sps->sps_scc_extension_params.motion_vector_resolution_control_idc
          == 2)
        WRITE_BITS (bw, slice->use_integer_mv_flag, 1);
    }

    WRITE_SE_RANGE (bw, slice->qp_delta, -87, 77);
    if (slice->pps->slice_chroma_qp_offsets_present_flag) {
      WRITE_SE_RANGE (bw, slice->cb_qp_offset, -12, 12);
      WRITE_SE_RANGE (bw, slice->cr_qp_offset, -12, 12);
    }

    if (pps_scc_extension_params->pps_slice_act_qp_offsets_present_flag) {
      WRITE_SE_RANGE (bw, slice->slice_act_y_qp_offset, -12, 12);
      WRITE_SE_RANGE (bw, slice->slice_act_cb_qp_offset, -12, 12);
      WRITE_SE_RANGE (bw, slice->slice_act_cr_qp_offset, -12, 12);
    }

    if (pps_extension_params->chroma_qp_offset_list_enabled_flag)
      WRITE_BITS (bw, slice->cu_chroma_qp_offset_enabled_flag, 1);

    if (slice->pps->deblocking_filter_override_enabled_flag)
      WRITE_BITS (bw, slice->deblocking_filter_override_flag, 1);

    if (slice->deblocking_filter_override_flag) {
      WRITE_BITS (bw, slice->deblocking_filter_disabled_flag, 1);

      if (!slice->deblocking_filter_disabled_flag) {
        WRITE_SE_RANGE (bw, slice->beta_offset_div2, -6, 6);
        WRITE_SE_RANGE (bw, slice->tc_offset_div2, -6, 6);
      }
    }

    if (slice->pps->loop_filter_across_slices_enabled_flag &&
        (slice->sao_luma_flag || slice->sao_chroma_flag ||
            !slice->deblocking_filter_disabled_flag))
      WRITE_BITS (bw, slice->loop_filter_across_slices_enabled_flag, 1);
  }

  if (slice->pps->tiles_enabled_flag
      || slice->pps->entropy_coding_sync_enabled_flag) {
    guint32 offset_max;

    if (!slice->pps->tiles_enabled_flag
        && slice->pps->entropy_coding_sync_enabled_flag) {
      offset_max = slice->pps->PicHeightInCtbsY - 1;
    } else if (slice->pps->tiles_enabled_flag
        && !slice->pps->entropy_coding_sync_enabled_flag) {
      offset_max =
          (slice->pps->num_tile_columns_minus1 +
          1) * (slice->pps->num_tile_rows_minus1 + 1) - 1;
    } else {
      offset_max =
          (slice->pps->num_tile_columns_minus1 +
          1) * slice->pps->PicHeightInCtbsY - 1;
    }

    WRITE_UE_MAX (bw, slice->num_entry_point_offsets, offset_max);
    if (slice->num_entry_point_offsets > 0) {
      WRITE_UE_MAX (bw, slice->offset_len_minus1, 31);
      for (i = 0; i < slice->num_entry_point_offsets; i++)
        WRITE_BITS (bw, slice->entry_point_offset_minus1[i],
            (slice->offset_len_minus1 + 1));
    }
  }

  /* TODO */
  if (slice->pps->slice_segment_header_extension_present_flag) {
    GST_WARNING
        ("slice_segment_header_extension_present_flag is not supported");
    goto error;
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("error to write slice header");

  *space = have_space;
  return FALSE;
}

/**
 * gst_h265_bit_writer_slice_hdr:
 * @slice: the slice header of #GstH265SliceHdr to write
 * @start_code: whether adding the nal start code
 * @nal_type: the slice's nal type of #GstH265NalUnitType
 * @data: (out): the bit stream generated by the slice header
 * @size: (inout): the size in bytes of the input and output
 *
 * Generating the according h265 bit stream by providing the slice header.
 *
 * Returns: a #GstH265BitWriterResult
 *
 * Since: 1.22
 **/
GstH265BitWriterResult
gst_h265_bit_writer_slice_hdr (const GstH265SliceHdr * slice,
    gboolean start_code, guint32 nal_type, guint8 * data, guint * size)
{
  gboolean have_space = TRUE;
  GstBitWriter bw;

  g_return_val_if_fail (slice != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (slice->pps != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (slice->pps->sps != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (nal_type <= GST_H265_NAL_SLICE_CRA_NUT,
      GST_H265_BIT_WRITER_ERROR);

  gst_bit_writer_init_with_data (&bw, data, *size, FALSE);

  if (start_code)
    WRITE_BITS (&bw, 0x00000001, 32);

  /* NAL unit header */
  /* forbidden_zero_bit */
  WRITE_BITS (&bw, 0, 1);
  /* nal_unit_type */
  WRITE_BITS (&bw, nal_type, 6);
  /* nuh_layer_id, only support 0 now */
  WRITE_BITS (&bw, 0, 6);
  /* nuh_temporal_id_plus1, only support 1 now */
  WRITE_BITS (&bw, 1, 3);

  if (!_h265_bit_writer_slice_header (slice, nal_type, &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;
  }

  *size = gst_bit_writer_get_size (&bw) / 8;
  gst_bit_writer_reset (&bw);
  return GST_H265_BIT_WRITER_OK;

error:
  gst_bit_writer_reset (&bw);
  *size = 0;

  if (!have_space)
    return GST_H265_BIT_WRITER_NO_MORE_SPACE;
  return GST_H265_BIT_WRITER_INVALID_DATA;
}


static gboolean
_h265_bit_writer_sei_registered_user_data (const GstH265RegisteredUserData *
    rud, GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;

  GST_DEBUG ("Writing \"Registered user data\"");

  WRITE_BITS (bw, rud->country_code, 8);
  if (rud->country_code == 0xff)
    WRITE_BITS (bw, rud->country_code_extension, 8);

  WRITE_BYTES (bw, rud->data, rud->size);

  *space = TRUE;
  return TRUE;

error:
  GST_DEBUG ("Failed to write \"Registered user data\"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_sei_time_code (const GstH265TimeCode * tc,
    GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;
  gint i;

  GST_DEBUG ("Wrtiting \"Time code\"");

  WRITE_BITS (bw, tc->num_clock_ts, 2);

  for (i = 0; i < tc->num_clock_ts; i++) {
    WRITE_BITS (bw, tc->clock_timestamp_flag[i], 1);
    if (tc->clock_timestamp_flag[i]) {
      WRITE_BITS (bw, tc->units_field_based_flag[i], 1);
      WRITE_BITS (bw, tc->counting_type[i], 5);
      WRITE_BITS (bw, tc->full_timestamp_flag[i], 1);
      WRITE_BITS (bw, tc->discontinuity_flag[i], 1);
      WRITE_BITS (bw, tc->cnt_dropped_flag[i], 1);
      WRITE_BITS (bw, tc->n_frames[i], 9);

      if (tc->full_timestamp_flag[i]) {
        WRITE_BITS (bw, tc->seconds_value[i], 6);
        WRITE_BITS (bw, tc->minutes_value[i], 6);
        WRITE_BITS (bw, tc->hours_value[i], 5);
      } else {
        WRITE_BITS (bw, tc->seconds_flag[i], 1);
        if (tc->seconds_flag[i]) {
          WRITE_BITS (bw, tc->seconds_value[i], 6);
          WRITE_BITS (bw, tc->minutes_flag[i], 1);
          if (tc->minutes_flag[i]) {
            WRITE_BITS (bw, tc->minutes_value[i], 6);
            WRITE_BITS (bw, tc->hours_flag[i], 1);
            if (tc->hours_flag[i]) {
              WRITE_BITS (bw, tc->hours_value[i], 5);
            }
          }
        }
      }
    }

    WRITE_BITS (bw, tc->time_offset_length[i], 5);

    if (tc->time_offset_length[i] > 0)
      WRITE_BITS (bw, tc->time_offset_value[i], tc->time_offset_length[i]);
  }

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("Failed to write \"Time code\"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_sei_mastering_display_colour_volume (const
    GstH265MasteringDisplayColourVolume * mdcv, GstBitWriter * bw,
    gboolean * space)
{
  gboolean have_space = TRUE;
  gint i;

  GST_DEBUG ("Wrtiting \"Mastering display colour volume\"");

  for (i = 0; i < 3; i++) {
    WRITE_BITS (bw, mdcv->display_primaries_x[i], 16);
    WRITE_BITS (bw, mdcv->display_primaries_y[i], 16);
  }

  WRITE_BITS (bw, mdcv->white_point_x, 16);
  WRITE_BITS (bw, mdcv->white_point_y, 16);
  WRITE_BITS (bw, mdcv->max_display_mastering_luminance, 32);
  WRITE_BITS (bw, mdcv->min_display_mastering_luminance, 32);

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("Failed to write \"Mastering display colour volume\"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_sei_content_light_level_info (const
    GstH265ContentLightLevel * cll, GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;

  GST_DEBUG ("Writing \"Content light level\"");

  WRITE_BITS (bw, cll->max_content_light_level, 16);
  WRITE_BITS (bw, cll->max_pic_average_light_level, 16);

  *space = TRUE;
  return TRUE;

error:
  GST_WARNING ("Failed to write \"Content light level\"");

  *space = have_space;
  return FALSE;
}

static gboolean
_h265_bit_writer_sei_message (const GstH265SEIMessage * msg,
    GstBitWriter * bw, gboolean * space)
{
  gboolean have_space = TRUE;

  GST_DEBUG ("writing SEI message");

  switch (msg->payloadType) {
    case GST_H265_SEI_REGISTERED_USER_DATA:
      if (!_h265_bit_writer_sei_registered_user_data
          (&msg->payload.registered_user_data, bw, &have_space))
        goto error;
      break;
    case GST_H265_SEI_TIME_CODE:
      if (!_h265_bit_writer_sei_time_code
          (&msg->payload.time_code, bw, &have_space))
        goto error;
      break;
    case GST_H265_SEI_MASTERING_DISPLAY_COLOUR_VOLUME:
      if (!_h265_bit_writer_sei_mastering_display_colour_volume
          (&msg->payload.mastering_display_colour_volume, bw, &have_space))
        goto error;
      break;
    case GST_H265_SEI_CONTENT_LIGHT_LEVEL:
      if (!_h265_bit_writer_sei_content_light_level_info
          (&msg->payload.content_light_level, bw, &have_space))
        goto error;
      break;
    default:
      break;
  }

  /* Add trailings. */
  WRITE_BITS (bw, 1, 1);
  gst_bit_writer_align_bytes_unchecked (bw, 0);

  *space = TRUE;

  return TRUE;

error:
  GST_WARNING ("error to write SEI message");

  *space = have_space;
  return FALSE;
}

/**
 * gst_h265_bit_writer_sei:
 * @sei_messages: An array of #GstH265SEIMessage to write
 * @start_code: whether adding the nal start code
 * @data: (out): the bit stream generated by the sei messages
 * @size: (inout): the size in bytes of the input and output
 *
 * Generating the according h265 bit stream by providing sei messages.
 *
 * Returns: a #GstH265BitWriterResult
 *
 * Since: 1.22
 **/
GstH265BitWriterResult
gst_h265_bit_writer_sei (GArray * sei_messages,
    GstH265NalUnitType nal_type, gboolean start_code, guint8 * data,
    guint * size)
{
  gboolean have_space = TRUE;
  GstBitWriter bw;
  GstH265SEIMessage *sei;
  gboolean have_written_data = FALSE;
  guint i;

  g_return_val_if_fail (sei_messages != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (nal_type == GST_H265_NAL_PREFIX_SEI
      || nal_type == GST_H265_NAL_SUFFIX_SEI, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);

  if (nal_type == GST_H265_NAL_PREFIX_SEI) {
    GST_WARNING ("prefix sei is not supported");
    return GST_H265_BIT_WRITER_ERROR;
  }

  gst_bit_writer_init_with_data (&bw, data, *size, FALSE);

  if (start_code)
    WRITE_BITS (&bw, 0x00000001, 32);

  /* NAL unit header */
  /* forbidden_zero_bit */
  WRITE_BITS (&bw, 0, 1);
  /* nal_unit_type */
  WRITE_BITS (&bw, nal_type, 6);
  /* nuh_layer_id, only support 0 now */
  WRITE_BITS (&bw, 0, 6);
  /* nuh_temporal_id_plus1, only support 1 now */
  WRITE_BITS (&bw, 1, 3);

  for (i = 0; i < sei_messages->len; i++) {
    guint32 payload_size_data;
    guint32 payload_type_data;

    gst_bit_writer_init (&bw);

    sei = &g_array_index (sei_messages, GstH265SEIMessage, i);
    if (!_h265_bit_writer_sei_message (sei, &bw, &have_space))
      goto error;

    if (gst_bit_writer_get_size (&bw) == 0) {
      GST_FIXME ("Unsupported SEI type %d", sei->payloadType);
      continue;
    }

    have_written_data = TRUE;

    g_assert (gst_bit_writer_get_size (&bw) % 8 == 0);
    payload_size_data = (gst_bit_writer_get_size (&bw) + 7) / 8;
    payload_type_data = sei->payloadType;

    /* write payload type bytes */
    while (payload_type_data >= 0xff) {
      WRITE_BITS (&bw, 0xff, 8);
      payload_type_data -= -0xff;
    }
    WRITE_BITS (&bw, payload_type_data, 8);

    /* write payload size bytes */
    while (payload_size_data >= 0xff) {
      WRITE_BITS (&bw, 0xff, 8);
      payload_size_data -= -0xff;
    }
    WRITE_BITS (&bw, payload_size_data, 8);

    if (gst_bit_writer_get_size (&bw) / 8)
      WRITE_BYTES (&bw, gst_bit_writer_get_data (&bw),
          gst_bit_writer_get_size (&bw) / 8);

    gst_bit_writer_reset (&bw);
  }

  if (!have_written_data) {
    GST_WARNING ("No written sei data");
    goto error;
  }

  /* Add trailings. */
  WRITE_BITS (&bw, 1, 1);
  if (!gst_bit_writer_align_bytes (&bw, 0)) {
    have_space = FALSE;
    goto error;
  }

  *size = gst_bit_writer_get_size (&bw) / 8;
  gst_bit_writer_reset (&bw);
  return GST_H265_BIT_WRITER_OK;

error:
  gst_bit_writer_reset (&bw);
  *size = 0;

  if (!have_space)
    return GST_H265_BIT_WRITER_NO_MORE_SPACE;
  return GST_H265_BIT_WRITER_INVALID_DATA;
}

/**
 * gst_h265_bit_writer_aud:
 * @pic_type: indicate the possible slice types list just
 *   as the H265 spec Table 7-2 defines
 * @start_code: whether adding the nal start code
 * @data: (out): the bit stream generated by the aud
 * @size: (inout): the size in bytes of the input and output
 *
 * Generating the according h265 bit stream of an aud.
 *
 * Returns: a #GstH265BitWriterResult
 *
 * Since: 1.22
 **/
GstH265BitWriterResult
gst_h265_bit_writer_aud (guint8 pic_type, gboolean start_code,
    guint8 * data, guint * size)
{
  gboolean have_space = TRUE;
  GstBitWriter bw;

  g_return_val_if_fail (pic_type <= 2, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);

  gst_bit_writer_init_with_data (&bw, data, *size, FALSE);

  if (start_code)
    WRITE_BITS (&bw, 0x00000001, 32);

  /* NAL unit header */
  /* forbidden_zero_bit */
  WRITE_BITS (&bw, 0, 1);
  /* nal_unit_type */
  WRITE_BITS (&bw, GST_H265_NAL_AUD, 6);
  /* nuh_layer_id, only support 0 now */
  WRITE_BITS (&bw, 0, 6);
  /* nuh_temporal_id_plus1, only support 1 now */
  WRITE_BITS (&bw, 1, 3);

  WRITE_BITS (&bw, pic_type, 3);

  /* Add trailings. */
  WRITE_BITS (&bw, 1, 1);
  if (!gst_bit_writer_align_bytes (&bw, 0)) {
    have_space = FALSE;
    goto error;
  }

  *size = gst_bit_writer_get_size (&bw) / 8;
  gst_bit_writer_reset (&bw);

  return GST_H265_BIT_WRITER_OK;

error:
  gst_bit_writer_reset (&bw);
  *size = 0;

  if (!have_space)
    return GST_H265_BIT_WRITER_NO_MORE_SPACE;
  return GST_H265_BIT_WRITER_INVALID_DATA;
}

/**
 * gst_h265_bit_writer_convert_to_nal:
 * @nal_prefix_size: the size in bytes for the prefix of a nal, may
 *   be 2, 3 or 4
 * @packetized: whether to write the bit stream in packetized format,
 *   which does not have the start code but has a @nal_prefix_size bytes'
 *   size prepending to the real nal data
 * @has_startcode: whether the input already has a start code
 * @add_trailings: whether to add rbsp trailing bits to make the output
 *   aligned to byte
 * @raw_data: the input bit stream
 * @raw_size: the size in bits of the input bit stream
 * @nal_data: (out): the output bit stream converted to a real nal
 * @nal_size: (inout): the size in bytes of the output
 *
 * Converting a bit stream into a real nal packet. If the bit stream already
 * has a start code, it will be replaced by the new one specified by the
 * @nal_prefix_size and @packetized. It is assured that the output aligns to
 * the byte and the all the emulations are inserted.
 *
 * Returns: a #GstH265BitWriterResult
 *
 * Since: 1.22
 **/
GstH265BitWriterResult
gst_h265_bit_writer_convert_to_nal (guint nal_prefix_size,
    gboolean packetized, gboolean has_startcode, gboolean add_trailings,
    const guint8 * raw_data, gsize raw_size, guint8 * nal_data,
    guint * nal_size)
{
  NalWriter nw;
  guint8 *data;
  guint32 size = 0;
  gboolean need_more_space = FALSE;

  g_return_val_if_fail (
      (packetized && nal_prefix_size > 1 && nal_prefix_size < 5) ||
      (!packetized && (nal_prefix_size == 3 || nal_prefix_size == 4)),
      GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (raw_data != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (raw_size > 0, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (nal_data != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (nal_size != NULL, GST_H265_BIT_WRITER_ERROR);
  g_return_val_if_fail (*nal_size > 0, GST_H265_BIT_WRITER_ERROR);

  if (has_startcode) {
    /* Skip the start code, the NalWriter will add it automatically. */
    if (raw_size >= 4 && raw_data[0] == 0
        && raw_data[1] == 0 && raw_data[2] == 0 && raw_data[3] == 0x01) {
      raw_data += 4;
      raw_size -= 4 * 8;
    } else if (raw_size >= 3 && raw_data[0] == 0 && raw_data[1] == 0
        && raw_data[2] == 0x01) {
      raw_data += 3;
      raw_size -= 3 * 8;
    } else {
      /* Fail to find the start code. */
      g_return_val_if_reached (GST_H265_BIT_WRITER_ERROR);
    }
  }

  /* If no RBSP trailing needed, it must align to byte. We assume
     that the rbsp trailing bits are already added. */
  if (!add_trailings)
    g_return_val_if_fail (raw_size % 8 == 0, GST_H265_BIT_WRITER_ERROR);

  nal_writer_init (&nw, nal_prefix_size, packetized);

  if (!nal_writer_put_bytes (&nw, raw_data, raw_size / 8))
    goto error;

  if (raw_size % 8) {
    guint8 data = *(raw_data + raw_size / 8);

    if (!nal_writer_put_bits_uint8 (&nw,
            data >> (8 - raw_size % 8), raw_size % 8))
      goto error;
  }

  if (add_trailings) {
    if (!nal_writer_do_rbsp_trailing_bits (&nw))
      goto error;
  }

  data = nal_writer_reset_and_get_data (&nw, &size);
  if (!data)
    goto error;

  if (size > *nal_size) {
    need_more_space = TRUE;
    g_free (data);
    goto error;
  }

  memcpy (nal_data, data, size);
  *nal_size = size;
  g_free (data);
  nal_writer_reset (&nw);
  return GST_H265_BIT_WRITER_OK;

error:
  nal_writer_reset (&nw);
  *nal_size = 0;

  GST_WARNING ("Failed to convert nal data");

  if (need_more_space)
    return GST_H265_BIT_WRITER_NO_MORE_SPACE;
  return GST_H265_BIT_WRITER_INVALID_DATA;
}
