On 2/13/19 6:51 PM, Michael Tretter wrote:
> The allegro hardware encoder does not write SPS/PPS nal units into the
> encoded video stream. Therefore, we need to write the units in software.
>
> The implementation follows Rec. ITU-T H.264 (04/2017) to allow to
> convert between a C struct and the RBSP representation of the SPS and
> PPS nal units.
>
> The allegro driver writes the nal units into the v4l2 capture buffer in
> front of the actual video data which is written at an offset by the IP
> core. The remaining gap is filled with a filler nal unit.
>
> Signed-off-by: Michael Tretter <m.tretter@xxxxxxxxxxxxxx>
> ---
> v2 -> v3:
> none
>
> v1 -> v2:
> - clean up debug log levels
> - fix missing error handling in allegro_h264_write_sps
> - enable configuration of frame size
> - enable configuration of bit rate and CPB size
> ---
> drivers/staging/media/allegro-dvt/Makefile | 4 +-
> .../staging/media/allegro-dvt/allegro-core.c | 182 ++-
> drivers/staging/media/allegro-dvt/nal-h264.c | 1278 +++++++++++++++++
> drivers/staging/media/allegro-dvt/nal-h264.h | 188 +++
> 4 files changed, 1650 insertions(+), 2 deletions(-)
> create mode 100644 drivers/staging/media/allegro-dvt/nal-h264.c
> create mode 100644 drivers/staging/media/allegro-dvt/nal-h264.h
>
> diff --git a/drivers/staging/media/allegro-dvt/Makefile b/drivers/staging/media/allegro-dvt/Makefile
> index bc30addee47f..eee9713c10e3 100644
> --- a/drivers/staging/media/allegro-dvt/Makefile
> +++ b/drivers/staging/media/allegro-dvt/Makefile
> @@ -1,4 +1,6 @@
> # SPDX-License-Identifier: GPL-2.0
> -allegro-objs := allegro-core.o
> +ccflags-y += -I$(src)
> +
> +allegro-objs := allegro-core.o nal-h264.o
>
> obj-$(CONFIG_VIDEO_ALLEGRO_DVT) += allegro.o
> diff --git a/drivers/staging/media/allegro-dvt/allegro-core.c b/drivers/staging/media/allegro-dvt/allegro-core.c
> index d0e1b2a574a6..8fa53c171110 100644
> --- a/drivers/staging/media/allegro-dvt/allegro-core.c
> +++ b/drivers/staging/media/allegro-dvt/allegro-core.c
> @@ -26,6 +26,8 @@
> #include <media/videobuf2-dma-contig.h>
> #include <media/videobuf2-v4l2.h>
>
> +#include "nal-h264.h"
> +
> /*
> * PG252 June 6, 2018 (H.264/H.265 Video Codec Unit v1.1) Chapter 3
> *
> @@ -1281,6 +1283,131 @@ static int allocate_reference_buffers(struct allegro_channel *channel,
> n, PAGE_ALIGN(size));
> }
>
> +static ssize_t allegro_h264_write_sps(struct allegro_channel *channel,
> + void *dest, size_t n)
> +{
> + struct allegro_dev *dev = channel->dev;
> + struct nal_h264_sps *sps;
> + ssize_t size;
> + unsigned int size_mb = SIZE_MACROBLOCK;
> + /* Calculation of crop units in Rec. ITU-T H.264 (04/2017) p. 76 */
> + unsigned int crop_unit_x = 2;
> + unsigned int crop_unit_y = 2;
> +
> + sps = kzalloc(sizeof(*sps), GFP_KERNEL);
> + if (!sps)
> + return -ENOMEM;
> +
> + sps->profile_idc = nal_h264_profile_from_v4l2(channel->profile);
> + sps->constraint_set0_flag = 0;
> + sps->constraint_set1_flag = 1;
> + sps->constraint_set2_flag = 0;
> + sps->constraint_set3_flag = 0;
> + sps->constraint_set4_flag = 0;
> + sps->constraint_set5_flag = 0;
> + sps->level_idc = nal_h264_level_from_v4l2(channel->level);
> + sps->seq_parameter_set_id = 0;
> + sps->log2_max_frame_num_minus4 = 0;
> + sps->pic_order_cnt_type = 0;
> + sps->log2_max_pic_order_cnt_lsb_minus4 = 6;
> + sps->max_num_ref_frames = 3;
> + sps->gaps_in_frame_num_value_allowed_flag = 0;
> + sps->pic_width_in_mbs_minus1 =
> + DIV_ROUND_UP(channel->width, size_mb) - 1;
> + sps->pic_height_in_map_units_minus1 =
> + DIV_ROUND_UP(channel->height, size_mb) - 1;
> + sps->frame_mbs_only_flag = 1;
> + sps->mb_adaptive_frame_field_flag = 0;
> + sps->direct_8x8_inference_flag = 1;
> + sps->frame_cropping_flag =
> + (channel->width % size_mb) || (channel->height % size_mb);
> + if (sps->frame_cropping_flag) {
> + sps->crop_left = 0;
> + sps->crop_right = (round_up(channel->width, size_mb) - channel->width) / crop_unit_x;
> + sps->crop_top = 0;
> + sps->crop_bottom = (round_up(channel->height, size_mb) - channel->height) / crop_unit_y;
> + }
> + sps->vui_parameters_present_flag = 1;
> + sps->vui.aspect_ratio_info_present_flag = 0;
> + sps->vui.overscan_info_present_flag = 0;
> + sps->vui.video_signal_type_present_flag = 1;
> + sps->vui.video_format = 1;
> + sps->vui.video_full_range_flag = 0;
> + sps->vui.colour_description_present_flag = 1;
> + sps->vui.colour_primaries = 5;
> + sps->vui.transfer_characteristics = 5;
> + sps->vui.matrix_coefficients = 5;
> + sps->vui.chroma_loc_info_present_flag = 1;
> + sps->vui.chroma_sample_loc_type_top_field = 0;
> + sps->vui.chroma_sample_loc_type_bottom_field = 0;
> + sps->vui.timing_info_present_flag = 1;
> + sps->vui.num_units_in_tick = 1;
> + sps->vui.time_scale = 50;
> + sps->vui.fixed_frame_rate_flag = 1;
> + sps->vui.nal_hrd_parameters_present_flag = 0;
> + sps->vui.vcl_hrd_parameters_present_flag = 1;
> + sps->vui.vcl_hrd_parameters.cpb_cnt_minus1 = 0;
> + sps->vui.vcl_hrd_parameters.bit_rate_scale = 0;
> + sps->vui.vcl_hrd_parameters.cpb_size_scale = 1;
> + /* See Rec. ITU-T H.264 (04/2017) p. 410 E-53 */
> + sps->vui.vcl_hrd_parameters.bit_rate_value_minus1[0] =
> + channel->bitrate_peak / (1 << (6 + sps->vui.vcl_hrd_parameters.bit_rate_scale)) - 1;
> + /* See Rec. ITU-T H.264 (04/2017) p. 410 E-54 */
> + sps->vui.vcl_hrd_parameters.cpb_size_value_minus1[0] =
> + (channel->cpb_size * 1000) / (1 << (4 + sps->vui.vcl_hrd_parameters.cpb_size_scale)) - 1;
> + sps->vui.vcl_hrd_parameters.cbr_flag[0] = 1;
> + sps->vui.vcl_hrd_parameters.initial_cpb_removal_delay_length_minus1 = 31;
> + sps->vui.vcl_hrd_parameters.cpb_removal_delay_length_minus1 = 31;
> + sps->vui.vcl_hrd_parameters.dpb_output_delay_length_minus1 = 31;
> + sps->vui.vcl_hrd_parameters.time_offset_length = 0;
> + sps->vui.low_delay_hrd_flag = 0;
> + sps->vui.pic_struct_present_flag = 1;
> + sps->vui.bitstream_restriction_flag = 0;
> +
> + size = nal_h264_write_sps(&dev->plat_dev->dev, dest, n, sps);
> +
> + kfree(sps);
> +
> + return size;
> +}
> +
> +static ssize_t allegro_h264_write_pps(struct allegro_channel *channel,
> + void *dest, size_t n)
> +{
> + struct allegro_dev *dev = channel->dev;
> + struct nal_h264_pps *pps;
> + ssize_t size;
> +
> + pps = kzalloc(sizeof(*pps), GFP_KERNEL);
> + if (!pps)
> + return -ENOMEM;
> +
> + pps->pic_parameter_set_id = 0;
> + pps->seq_parameter_set_id = 0;
> + pps->entropy_coding_mode_flag = 0;
> + pps->bottom_field_pic_order_in_frame_present_flag = 0;
> + pps->num_slice_groups_minus1 = 0;
> + pps->num_ref_idx_l0_default_active_minus1 = 2;
> + pps->num_ref_idx_l1_default_active_minus1 = 2;
> + pps->weighted_pred_flag = 0;
> + pps->weighted_bipred_idc = 0;
> + pps->pic_init_qp_minus26 = 0;
> + pps->pic_init_qs_minus26 = 0;
> + pps->chroma_qp_index_offset = 0;
> + pps->deblocking_filter_control_present_flag = 1;
> + pps->constrained_intra_pred_flag = 0;
> + pps->redundant_pic_cnt_present_flag = 0;
> + pps->transform_8x8_mode_flag = 0;
> + pps->pic_scaling_matrix_present_flag = 0;
> + pps->second_chroma_qp_index_offset = 0;
> +
> + size = nal_h264_write_pps(&dev->plat_dev->dev, dest, n, pps);
> +
> + kfree(pps);
> +
> + return size;
> +}
> +
> static void allegro_finish_frame(struct allegro_channel *channel,
> struct mcu_msg_encode_frame_response *msg)
> {
> @@ -1292,6 +1419,9 @@ static void allegro_finish_frame(struct allegro_channel *channel,
> u32 size;
> } *partition;
> enum vb2_buffer_state state = VB2_BUF_STATE_ERROR;
> + char *curr;
> + ssize_t len;
> + ssize_t free;
>
> src_buf = v4l2_m2m_src_buf_remove(channel->fh.m2m_ctx);
>
> @@ -1340,10 +1470,60 @@ static void allegro_finish_frame(struct allegro_channel *channel,
> * Does the encoder add any data before its configured offset that we
> * need to handle?
> */
> -
> vb2_set_plane_payload(&dst_buf->vb2_buf, 0,
> partition->offset + partition->size);
>
> + curr = vb2_plane_vaddr(&dst_buf->vb2_buf, 0);
> + free = partition->offset;
> + if (msg->is_idr) {
> + len = allegro_h264_write_sps(channel, curr, free);
> + if (len < 0) {
> + v4l2_err(&dev->v4l2_dev,
> + "not enough space for sequence parameter set: %ld left\n",
> + free);
> + goto err;
> + }
> + curr += len;
> + free -= len;
> + v4l2_dbg(1, debug, &dev->v4l2_dev,
> + "channel %d: wrote %ld byte SPS nal unit\n",
> + channel->mcu_channel_id, len);
> + }
> +
> + if (msg->slice_type == AL_ENC_SLICE_TYPE_I) {
> + len = allegro_h264_write_pps(channel, curr, free);
> + if (len < 0) {
> + v4l2_err(&dev->v4l2_dev,
> + "not enough space for picture parameter set: %ld left\n",
> + free);
> + goto err;
> + }
> + curr += len;
> + free -= len;
> + v4l2_dbg(1, debug, &dev->v4l2_dev,
> + "channel %d: wrote %ld byte PPS nal unit\n",
> + channel->mcu_channel_id, len);
> + }
> +
> + len = nal_h264_write_filler(&dev->plat_dev->dev, curr, free);
> + if (len < 0) {
> + v4l2_err(&dev->v4l2_dev,
> + "failed to write %ld filler data\n", free);
> + goto err;
> + }
> + curr += len;
> + free -= len;
> + v4l2_dbg(2, debug, &dev->v4l2_dev,
> + "channel %d: wrote %ld bytes filler nal unit\n",
> + channel->mcu_channel_id, len);
> +
> + if (free != 0) {
> + v4l2_err(&dev->v4l2_dev,
> + "non-VCL NAL units do not fill space until VCL NAL unit: %ld bytes left\n",
> + free);
> + goto err;
> + }
> +
> state = VB2_BUF_STATE_DONE;
>
> dst_buf->vb2_buf.timestamp = src_buf->vb2_buf.timestamp;
> diff --git a/drivers/staging/media/allegro-dvt/nal-h264.c b/drivers/staging/media/allegro-dvt/nal-h264.c
> new file mode 100644
> index 000000000000..83bc98200c1a
> --- /dev/null
> +++ b/drivers/staging/media/allegro-dvt/nal-h264.c
> @@ -0,0 +1,1278 @@
> +// SPDX-License-Identifier: GPL-2.0
> +/*
> + * Copyright (C) 2019 Pengutronix, Michael Tretter <kernel@xxxxxxxxxxxxxx>
> + *
> + * Convert NAL units between raw byte sequence payloads (RBSP) and C structs
> + *
> + * The conversion is defined in "ITU-T Rec. H.264 (04/2017) Advanced video
> + * coding for generic audiovisual services". Decoder drivers may use the
> + * parser to parse RBSP from encoded streams and configure the hardware, if
> + * the hardware is not able to parse RBSP itself. Encoder drivers may use the
> + * generator to generate the RBSP for SPS/PPS nal units and add them to the
> + * encoded stream if the hardware does not generate the units.
> + */
> +
> +#include <linux/kernel.h>
> +#include <linux/types.h>
> +#include <linux/string.h>
> +#include <linux/v4l2-controls.h>
> +
> +#include <linux/device.h>
> +#include <linux/export.h>
> +#include <linux/log2.h>
> +
> +#include <nal-h264.h>
> +
> +struct rbsp {
> + char *buf;
> + int size;
> + int pos;
> + int num_consecutive_zeros;
> +};
> +
> +int nal_h264_profile_from_v4l2(enum v4l2_mpeg_video_h264_profile profile)
> +{
> + switch (profile) {
> + case V4L2_MPEG_VIDEO_H264_PROFILE_BASELINE:
> + return 66;
> + case V4L2_MPEG_VIDEO_H264_PROFILE_MAIN:
> + return 77;
> + case V4L2_MPEG_VIDEO_H264_PROFILE_EXTENDED:
> + return 88;
> + case V4L2_MPEG_VIDEO_H264_PROFILE_HIGH:
> + return 100;
> + default:
> + return -EINVAL;
> + }
> +}
> +
> +int nal_h264_level_from_v4l2(enum v4l2_mpeg_video_h264_level level)
> +{
> + switch (level) {
> + case V4L2_MPEG_VIDEO_H264_LEVEL_1_0:
> + return 10;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_1B:
> + return 9;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_1_1:
> + return 11;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_1_2:
> + return 12;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_1_3:
> + return 13;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_2_0:
> + return 20;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_2_1:
> + return 21;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_2_2:
> + return 22;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_3_0:
> + return 30;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_3_1:
> + return 31;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_3_2:
> + return 32;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_4_0:
> + return 40;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_4_1:
> + return 41;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_4_2:
> + return 42;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_5_0:
> + return 50;
> + case V4L2_MPEG_VIDEO_H264_LEVEL_5_1:
> + return 51;
> + default:
> + return -EINVAL;
> + }
> +}
> +
> +static int rbsp_read_bits(struct rbsp *rbsp, int num, int *val);
> +static int rbsp_write_bits(struct rbsp *rbsp, int num, int val);
> +
> +static int add_emulation_prevention_three_byte(struct rbsp *rbsp)
> +{
> + rbsp->num_consecutive_zeros = 0;
> + /*
> + * We are not actually the emulation_prevention_three_byte, but the 2
> + * one bits of the byte and the 6 zero bits of the next byte.
> + * Therefore, the discarded byte shifted by 6 bits.
> + */
> + rbsp_write_bits(rbsp, 8, (0x3 << 6));
> +
> + return 0;
> +}
> +
> +static int discard_emulation_prevention_three_byte(struct rbsp *rbsp)
> +{
> + unsigned int tmp = 0;
> +
> + rbsp->num_consecutive_zeros = 0;
> + /*
> + * We are not actually discarding the emulation_prevention_three_byte,
> + * but the 2 one bits of the byte and the 6 zero bits of the next
> + * byte. Therefore, the discarded byte shifted by 6 bits.
> + */
> + rbsp_read_bits(rbsp, 8, &tmp);
> + if (tmp != (0x3 << 6))
> + return -EINVAL;
> +
> + return 0;
> +}
> +
> +static inline int rbsp_read_bit(struct rbsp *rbsp)
> +{
> + int shift;
> + int ofs;
> + int bit;
> + int err;
> +
> + if (rbsp->num_consecutive_zeros == 22) {
> + err = discard_emulation_prevention_three_byte(rbsp);
> + if (err)
> + return err;
> + }
> +
> + shift = 7 - (rbsp->pos % 8);
> + ofs = rbsp->pos++ / 8;
> +
> + if (ofs >= rbsp->size)
> + return -EINVAL;
> +
> + bit = (rbsp->buf[ofs] >> shift) & 1;
> +
> + /*
> + * Counting zeros for the emulation_prevention_three_byte only starts
> + * at byte boundaries.
> + */
> + if (bit == 1 ||
> + (rbsp->num_consecutive_zeros < 7 && (rbsp->pos % 8 == 0)))
> + rbsp->num_consecutive_zeros = 0;
> + else
> + rbsp->num_consecutive_zeros++;
> +
> + return bit;
> +}
> +
> +static inline int rbsp_write_bit(struct rbsp *rbsp, int bit)
> +{
> + int shift;
> + int ofs;
> +
> + if (rbsp->num_consecutive_zeros == 22)
> + add_emulation_prevention_three_byte(rbsp);
> +
> + shift = 7 - (rbsp->pos % 8);
> + ofs = rbsp->pos++ / 8;
> +
> + if (ofs >= rbsp->size)
> + return -EINVAL;
> +
> + rbsp->buf[ofs] &= ~(1 << shift);
> + rbsp->buf[ofs] |= bit << shift;
> +
> + /*
> + * Counting zeros for the emulation_prevention_three_byte only starts
> + * at byte boundaries.
> + */
> + if (bit == 1 ||
> + (rbsp->num_consecutive_zeros < 7 && (rbsp->pos % 8 == 0))) {
> + rbsp->num_consecutive_zeros = 0;
> + } else {
> + rbsp->num_consecutive_zeros++;
> + }
> +
> + return 0;
> +}
> +
> +static inline int rbsp_read_bits(struct rbsp *rbsp, int num, int *val)
> +{
> + int i, ret;
> + int tmp = 0;
> +
> + if (num > 32)
> + return -EINVAL;
> +
> + for (i = 0; i < num; i++) {
> + ret = rbsp_read_bit(rbsp);
> + if (ret < 0)
> + return ret;
> + tmp |= ret << (num - i - 1);
> + }
> +
> + if (val)
> + *val = tmp;
> +
> + return 0;
> +}
> +
> +static int rbsp_write_bits(struct rbsp *rbsp, int num, int value)
> +{
> + int ret;
> +
> + while (num--) {
> + ret = rbsp_write_bit(rbsp, (value >> num) & 1);
> + if (ret)
> + return ret;
> + }
> +
> + return 0;
> +}
> +
> +static int rbsp_read_uev(struct rbsp *rbsp, unsigned int *val)
> +{
> + int leading_zero_bits = 0;
> + unsigned int tmp = 0;
> + int ret;
> +
> + while ((ret = rbsp_read_bit(rbsp)) == 0)
> + leading_zero_bits++;
> + if (ret < 0)
> + return ret;
> +
> + if (leading_zero_bits > 0) {
> + ret = rbsp_read_bits(rbsp, leading_zero_bits, &tmp);
> + if (ret)
> + return ret;
> + }
> +
> + if (val)
> + *val = (1 << leading_zero_bits) - 1 + tmp;
> +
> + return 0;
> +}
> +
> +static int rbsp_write_uev(struct rbsp *rbsp, unsigned int value)
> +{
> + int i;
> + int ret;
> + int tmp = value + 1;
> + int leading_zero_bits = fls(tmp) - 1;
> +
> + for (i = 0; i < leading_zero_bits; i++) {
> + ret = rbsp_write_bit(rbsp, 0);
> + if (ret)
> + return ret;
> + }
> +
> + return rbsp_write_bits(rbsp, leading_zero_bits + 1, tmp);
> +}
> +
> +static int rbsp_read_sev(struct rbsp *rbsp, int *val)
> +{
> + unsigned int tmp;
> + int ret;
> +
> + ret = rbsp_read_uev(rbsp, &tmp);
> + if (ret)
> + return ret;
> +
> + if (val) {
> + if (tmp & 1)
> + *val = (tmp + 1) / 2;
> + else
> + *val = -(tmp / 2);
> + }
> +
> + return 0;
> +}
> +
> +static int rbsp_write_sev(struct rbsp *rbsp, int val)
> +{
> + unsigned int tmp;
> +
> + if (val > 0)
> + tmp = (2 * val) | 1;
> + else
> + tmp = -2 * val;
> +
> + return rbsp_write_uev(rbsp, tmp);
> +}
> +
> +#define READ_BIT(field) \
> + do { \
> + int ret = rbsp_read_bit(rbsp); \
> + if (ret < 0) \
> + return ret; \
> + s->field = ret; \
> + } while (0)
> +
> +#define READ_BITS(num, field) \
> + do { \
> + int val; \
> + int ret = rbsp_read_bits(rbsp, (num), &val); \
> + if (ret) \
> + return ret; \
> + s->field = val; \
> + } while (0)
> +
> +#define READ_UEV(field) \
> + do { \
> + int ret = rbsp_read_uev(rbsp, &s->field); \
> + if (ret) \
> + return ret; \
> + } while (0)
> +
> +#define READ_SEV(field) \
> + do { \
> + int ret = rbsp_read_sev(rbsp, &s->field); \
> + if (ret) \
> + return ret; \
> + } while (0)
> +
> +#define WRITE_BIT(field) \
> + do { \
> + int ret = rbsp_write_bit(rbsp, s->field); \
> + if (ret < 0) \
> + return ret; \
> + } while (0)
> +
> +#define WRITE_BITS(num, field) \
> + do { \
> + int ret = rbsp_write_bits(rbsp, (num), s->field); \
> + if (ret) \
> + return ret; \
> + } while (0)
> +
> +#define WRITE_UEV(field) \
> + do { \
> + int ret = rbsp_write_uev(rbsp, s->field); \
> + if (ret) \
> + return ret; \
> + } while (0)
> +
> +#define WRITE_SEV(field) \
> + do { \
> + int ret = rbsp_write_sev(rbsp, s->field); \
> + if (ret) \
> + return ret; \
> + } while (0)
> +
> +#define PRINT_BIT(field) \
> + dev_dbg(dev, "field: %u\n", s->field) \
> +
> +#define PRINT_BITS(num, field) \
> + dev_dbg(dev, "field: %u\n", s->field) \
> +
> +#define PRINT_UEV(field) \
> + dev_dbg(dev, "field: %u\n", s->field) \
> +
> +#define PRINT_SEV(field) \
> + dev_dbg(dev, "field: %d\n", s->field) \
> +
> +static int nal_h264_write_trailing_bits(const struct device *dev,
> + struct rbsp *rbsp)
> +{
> + rbsp_write_bit(rbsp, 1);
> + while (rbsp->pos % 8)
> + rbsp_write_bit(rbsp, 0);
> +
> + return 0;
> +}
> +
> +static int nal_h264_write_hrd_parameters(const struct device *dev,
> + struct rbsp *rbsp,
> + struct nal_h264_hrd_parameters *hrd)
> +{
> + struct nal_h264_hrd_parameters *s = hrd;
> + int i;
> +
> + WRITE_UEV(cpb_cnt_minus1);
> + WRITE_BITS(4, bit_rate_scale);
> + WRITE_BITS(4, cpb_size_scale);
> +
> + for (i = 0; i <= hrd->cpb_cnt_minus1; i++) {
> + WRITE_UEV(bit_rate_value_minus1[i]);
> + WRITE_UEV(cpb_size_value_minus1[i]);
> + WRITE_BIT(cbr_flag[i]);
> + }
> +
> + WRITE_BITS(5, initial_cpb_removal_delay_length_minus1);
> + WRITE_BITS(5, cpb_removal_delay_length_minus1);
> + WRITE_BITS(5, dpb_output_delay_length_minus1);
> + WRITE_BITS(5, time_offset_length);
> +
> + return 0;
> +}
> +
> +static int nal_h264_read_hrd_parameters(const struct device *dev,
> + struct rbsp *rbsp,
> + struct nal_h264_hrd_parameters *hrd)
> +{
> + struct nal_h264_hrd_parameters *s = hrd;
> + unsigned int i;
> +
> + READ_UEV(cpb_cnt_minus1);
> + READ_BITS(4, bit_rate_scale);
> + READ_BITS(4, cpb_size_scale);
> +
> + for (i = 0; i <= hrd->cpb_cnt_minus1; i++) {
> + READ_UEV(bit_rate_value_minus1[i]);
> + READ_UEV(cpb_size_value_minus1[i]);
> + READ_BIT(cbr_flag[i]);
> + }
> +
> + READ_BITS(5, initial_cpb_removal_delay_length_minus1);
> + READ_BITS(5, cpb_removal_delay_length_minus1);
> + READ_BITS(5, dpb_output_delay_length_minus1);
> + READ_BITS(5, time_offset_length);
> +
> + return 0;
> +}
> +
> +static void nal_h264_print_hrd_parameters(const struct device *dev,
> + struct nal_h264_hrd_parameters *hrd)
> +{
> + struct nal_h264_hrd_parameters *s = hrd;
> + unsigned int i;
> +
> + if (!hrd)
> + return;
> +
> + PRINT_UEV(cpb_cnt_minus1);
> + PRINT_BITS(4, bit_rate_scale);
> + PRINT_BITS(4, cpb_size_scale);
> +
> + for (i = 0; i <= s->cpb_cnt_minus1; i++) {
> + PRINT_UEV(bit_rate_value_minus1[i]);
> + PRINT_UEV(cpb_size_value_minus1[i]);
> + PRINT_BIT(cbr_flag[i]);
> + }
> +
> + PRINT_BITS(5, initial_cpb_removal_delay_length_minus1);
> + PRINT_BITS(5, cpb_removal_delay_length_minus1);
> + PRINT_BITS(5, dpb_output_delay_length_minus1);
> + PRINT_BITS(5, time_offset_length);
> +}
> +
> +static int nal_h264_read_vui_parameters(const struct device *dev,
> + struct rbsp *rbsp,
> + struct nal_h264_vui_parameters *vui)
> +{
> + struct nal_h264_vui_parameters *s = vui;
> + int err;
> +
> + READ_BIT(aspect_ratio_info_present_flag);
> + if (vui->aspect_ratio_info_present_flag) {
> + READ_BITS(8, aspect_ratio_idc);
> + if (vui->aspect_ratio_idc == 255) {
> + READ_BITS(16, sar_width);
> + READ_BITS(16, sar_height);
> + }
> + }
> +
> + READ_BIT(overscan_info_present_flag);
> + if (vui->overscan_info_present_flag)
> + READ_BIT(overscan_appropriate_flag);
> +
> + READ_BIT(video_signal_type_present_flag);
> + if (vui->video_signal_type_present_flag) {
> + READ_BITS(3, video_format);
> + READ_BIT(video_full_range_flag);
> + READ_BIT(colour_description_present_flag);
> +
> + if (vui->colour_description_present_flag) {
> + READ_BITS(8, colour_primaries);
> + READ_BITS(8, transfer_characteristics);
> + READ_BITS(8, matrix_coefficients);
> + }
> + }
> +
> + READ_BIT(chroma_loc_info_present_flag);
> + if (vui->chroma_loc_info_present_flag) {
> + READ_UEV(chroma_sample_loc_type_top_field);
> + READ_UEV(chroma_sample_loc_type_bottom_field);
> + }
> +
> + READ_BIT(timing_info_present_flag);
> + if (vui->timing_info_present_flag) {
> + READ_BITS(32, num_units_in_tick);
> + READ_BITS(32, time_scale);
> + READ_BIT(fixed_frame_rate_flag);
> + }
> +
> + READ_BIT(nal_hrd_parameters_present_flag);
> + if (vui->nal_hrd_parameters_present_flag) {
> + err = nal_h264_read_hrd_parameters(dev, rbsp,
> + &vui->nal_hrd_parameters);
> + if (err)
> + return err;
> + }
> +
> + READ_BIT(vcl_hrd_parameters_present_flag);
> + if (vui->vcl_hrd_parameters_present_flag) {
> + err = nal_h264_read_hrd_parameters(dev, rbsp,
> + &vui->vcl_hrd_parameters);
> + if (err)
> + return err;
> + }
> +
> + if (vui->nal_hrd_parameters_present_flag ||
> + vui->vcl_hrd_parameters_present_flag)
> + READ_BIT(low_delay_hrd_flag);
> +
> + READ_BIT(pic_struct_present_flag);
> +
> + READ_BIT(bitstream_restriction_flag);
> + if (vui->bitstream_restriction_flag) {
> + READ_BIT(motion_vectors_over_pic_boundaries_flag);
> + READ_UEV(max_bytes_per_pic_denom);
> + READ_UEV(max_bits_per_mb_denom);
> + READ_UEV(log2_max_mv_length_horizontal);
> + READ_UEV(log21_max_mv_length_vertical);
> + READ_UEV(max_num_reorder_frames);
> + READ_UEV(max_dec_frame_buffering);
> + }
> +
> + return 0;
> +}
> +
> +static ssize_t nal_h264_write_vui_parameters(const struct device *dev,
> + struct rbsp *rbsp,
> + struct nal_h264_vui_parameters *vui)
> +{
> + struct nal_h264_vui_parameters *s = vui;
> + int err;
> +
> + WRITE_BIT(aspect_ratio_info_present_flag);
> + if (vui->aspect_ratio_info_present_flag) {
> + WRITE_BITS(8, aspect_ratio_idc);
> + if (vui->aspect_ratio_idc == 255) {
> + WRITE_BITS(16, sar_width);
> + WRITE_BITS(16, sar_height);
> + }
> + }
> +
> + WRITE_BIT(overscan_info_present_flag);
> + if (vui->overscan_info_present_flag)
> + WRITE_BIT(overscan_appropriate_flag);
> +
> + WRITE_BIT(video_signal_type_present_flag);
> + if (vui->video_signal_type_present_flag) {
> + WRITE_BITS(3, video_format);
> + WRITE_BIT(video_full_range_flag);
> + WRITE_BIT(colour_description_present_flag);
> +
> + if (vui->colour_description_present_flag) {
> + WRITE_BITS(8, colour_primaries);
> + WRITE_BITS(8, transfer_characteristics);
> + WRITE_BITS(8, matrix_coefficients);
> + }
> + }
> +
> + WRITE_BIT(chroma_loc_info_present_flag);
> + if (vui->chroma_loc_info_present_flag) {
> + WRITE_UEV(chroma_sample_loc_type_top_field);
> + WRITE_UEV(chroma_sample_loc_type_bottom_field);
> + }
> +
> + WRITE_BIT(timing_info_present_flag);
> + if (vui->timing_info_present_flag) {
> + WRITE_BITS(32, num_units_in_tick);
> + WRITE_BITS(32, time_scale);
> + WRITE_BIT(fixed_frame_rate_flag);
> + }
> +
> + WRITE_BIT(nal_hrd_parameters_present_flag);
> + if (vui->nal_hrd_parameters_present_flag) {
> + err = nal_h264_write_hrd_parameters(dev, rbsp,
> + &vui->nal_hrd_parameters);
> + if (err)
> + return err;
> + }
> +
> + WRITE_BIT(vcl_hrd_parameters_present_flag);
> + if (vui->vcl_hrd_parameters_present_flag) {
> + err = nal_h264_write_hrd_parameters(dev, rbsp,
> + &vui->vcl_hrd_parameters);
> + if (err)
> + return err;
> + }
> +
> + if (vui->nal_hrd_parameters_present_flag ||
> + vui->vcl_hrd_parameters_present_flag)
> + WRITE_BIT(low_delay_hrd_flag);
> +
> + WRITE_BIT(pic_struct_present_flag);
> +
> + WRITE_BIT(bitstream_restriction_flag);
> + if (vui->bitstream_restriction_flag) {
> + WRITE_BIT(motion_vectors_over_pic_boundaries_flag);
> + WRITE_UEV(max_bytes_per_pic_denom);
> + WRITE_UEV(max_bits_per_mb_denom);
> + WRITE_UEV(log2_max_mv_length_horizontal);
> + WRITE_UEV(log21_max_mv_length_vertical);
> + WRITE_UEV(max_num_reorder_frames);
> + WRITE_UEV(max_dec_frame_buffering);
> + }
> +
> + return 0;
> +}
> +
> +static void nal_h264_print_vui_parameters(const struct device *dev,
> + struct nal_h264_vui_parameters *vui)
> +{
> + struct nal_h264_vui_parameters *s = vui;
> +
> + if (!vui)
> + return;
> +
> + PRINT_BIT(aspect_ratio_info_present_flag);
> + if (vui->aspect_ratio_info_present_flag) {
> + PRINT_BITS(8, aspect_ratio_idc);
> + if (vui->aspect_ratio_idc == 255) {
> + PRINT_BITS(16, sar_width);
> + PRINT_BITS(16, sar_height);
> + }
> + }
> +
> + PRINT_BIT(overscan_info_present_flag);
> + if (vui->overscan_info_present_flag)
> + PRINT_BIT(overscan_appropriate_flag);
> +
> + PRINT_BIT(video_signal_type_present_flag);
> + if (vui->video_signal_type_present_flag) {
> + PRINT_BITS(3, video_format);
> + PRINT_BIT(video_full_range_flag);
> + PRINT_BIT(colour_description_present_flag);
> +
> + if (vui->colour_description_present_flag) {
> + PRINT_BITS(8, colour_primaries);
> + PRINT_BITS(8, transfer_characteristics);
> + PRINT_BITS(8, matrix_coefficients);
> + }
> + }
> +
> + PRINT_BIT(chroma_loc_info_present_flag);
> + if (vui->chroma_loc_info_present_flag) {
> + PRINT_UEV(chroma_sample_loc_type_top_field);
> + PRINT_UEV(chroma_sample_loc_type_bottom_field);
> + }
> +
> + PRINT_BIT(timing_info_present_flag);
> + if (vui->timing_info_present_flag) {
> + PRINT_BITS(32, num_units_in_tick);
> + PRINT_BITS(32, time_scale);
> + PRINT_BIT(fixed_frame_rate_flag);
> + }
> +
> + PRINT_BIT(nal_hrd_parameters_present_flag);
> + if (vui->nal_hrd_parameters_present_flag)
> + nal_h264_print_hrd_parameters(dev, &vui->nal_hrd_parameters);
> +
> + PRINT_BIT(vcl_hrd_parameters_present_flag);
> + if (vui->vcl_hrd_parameters_present_flag)
> + nal_h264_print_hrd_parameters(dev, &vui->vcl_hrd_parameters);
> +
> + if (vui->nal_hrd_parameters_present_flag ||
> + vui->vcl_hrd_parameters_present_flag)
> + PRINT_BIT(low_delay_hrd_flag);
> +
> + PRINT_BIT(pic_struct_present_flag);
> +
> + PRINT_BIT(bitstream_restriction_flag);
> + if (vui->bitstream_restriction_flag) {
> + PRINT_BIT(motion_vectors_over_pic_boundaries_flag);
> + PRINT_UEV(max_bytes_per_pic_denom);
> + PRINT_UEV(max_bits_per_mb_denom);
> + PRINT_UEV(log2_max_mv_length_horizontal);
> + PRINT_UEV(log21_max_mv_length_vertical);
> + PRINT_UEV(max_num_reorder_frames);
> + PRINT_UEV(max_dec_frame_buffering);
> + }
> +}
> +
> +static int nal_h264_rbsp_write_sps(const struct device *dev,
> + struct rbsp *rbsp, struct nal_h264_sps *sps)
> +{
> + struct nal_h264_sps *s = sps;
> + unsigned int i;
> + int err;
> +
> + if (rbsp->size < 3)
> + return -EINVAL;
> +
> + WRITE_BITS(8, profile_idc);
> + WRITE_BIT(constraint_set0_flag);
> + WRITE_BIT(constraint_set1_flag);
> + WRITE_BIT(constraint_set2_flag);
> + WRITE_BIT(constraint_set3_flag);
> + WRITE_BIT(constraint_set4_flag);
> + WRITE_BIT(constraint_set5_flag);
> + WRITE_BITS(2, reserved_zero_2bits);
> + WRITE_BITS(8, level_idc);
> +
> + WRITE_UEV(seq_parameter_set_id);
> +
> + if (sps->profile_idc == 100 || sps->profile_idc == 110 ||
> + sps->profile_idc == 122 || sps->profile_idc == 244 ||
> + sps->profile_idc == 44 || sps->profile_idc == 83 ||
> + sps->profile_idc == 86 || sps->profile_idc == 118 ||
> + sps->profile_idc == 128 || sps->profile_idc == 138 ||
> + sps->profile_idc == 139 || sps->profile_idc == 134 ||
> + sps->profile_idc == 135) {
> + WRITE_UEV(chroma_format_idc);
> +
> + if (sps->chroma_format_idc == 3)
> + WRITE_BIT(separate_colour_plane_flag);
> +
> + WRITE_UEV(bit_depth_luma_minus8);
> + WRITE_UEV(bit_depth_chroma_minus8);
> + WRITE_BIT(qpprime_y_zero_transform_bypass_flag);
> + WRITE_BIT(seq_scaling_matrix_present_flag);
> +
> + if (sps->seq_scaling_matrix_present_flag) {
> + dev_err(dev,
> + "%s: Handling scaling matrix not supported\n",
> + __func__);
> + return -EINVAL;
> + }
> + }
> +
> + WRITE_UEV(log2_max_frame_num_minus4);
> +
> + WRITE_UEV(pic_order_cnt_type);
> + if (sps->pic_order_cnt_type == 0) {
> + WRITE_UEV(log2_max_pic_order_cnt_lsb_minus4);
> + } else if (sps->pic_order_cnt_type == 1) {
> + WRITE_BIT(delta_pic_order_always_zero_flag);
> + WRITE_SEV(offset_for_non_ref_pic);
> + WRITE_SEV(offset_for_top_to_bottom_field);
> +
> + WRITE_UEV(num_ref_frames_in_pic_order_cnt_cycle);
> + for (i = 0; i < sps->num_ref_frames_in_pic_order_cnt_cycle; i++)
> + WRITE_SEV(offset_for_ref_frame[i]);
> + } else {
> + dev_err(dev,
> + "%s: Invalid pic_order_cnt_type %u\n", __func__,
> + sps->pic_order_cnt_type);
> + return -EINVAL;
> + }
> +
> + WRITE_UEV(max_num_ref_frames);
> + WRITE_BIT(gaps_in_frame_num_value_allowed_flag);
> + WRITE_UEV(pic_width_in_mbs_minus1);
> + WRITE_UEV(pic_height_in_map_units_minus1);
> +
> + WRITE_BIT(frame_mbs_only_flag);
> + if (!sps->frame_mbs_only_flag)
> + WRITE_BIT(mb_adaptive_frame_field_flag);
> +
> + WRITE_BIT(direct_8x8_inference_flag);
> +
> + WRITE_BIT(frame_cropping_flag);
> + if (sps->frame_cropping_flag) {
> + WRITE_UEV(crop_left);
> + WRITE_UEV(crop_right);
> + WRITE_UEV(crop_top);
> + WRITE_UEV(crop_bottom);
> + }
> +
> + WRITE_BIT(vui_parameters_present_flag);
> + if (sps->vui_parameters_present_flag) {
> + err = nal_h264_write_vui_parameters(dev, rbsp, &sps->vui);
> + if (err)
> + return err;
> + }
> +
> + return 0;
> +}
> +
> +static int nal_h264_rbsp_read_sps(const struct device *dev,
> + struct rbsp *rbsp, struct nal_h264_sps *sps)
> +{
> + struct nal_h264_sps *s = sps;
> + unsigned int i;
> + int err;
> +
> + if (rbsp->size < 3)
> + return -EINVAL;
> +
> + READ_BITS(8, profile_idc);
> + READ_BIT(constraint_set0_flag);
> + READ_BIT(constraint_set1_flag);
> + READ_BIT(constraint_set2_flag);
> + READ_BIT(constraint_set3_flag);
> + READ_BIT(constraint_set4_flag);
> + READ_BIT(constraint_set5_flag);
> + READ_BITS(2, reserved_zero_2bits);
> + READ_BITS(8, level_idc);
> +
> + READ_UEV(seq_parameter_set_id);
> +
> + if (sps->profile_idc == 100 || sps->profile_idc == 110 ||
> + sps->profile_idc == 122 || sps->profile_idc == 244 ||
> + sps->profile_idc == 44 || sps->profile_idc == 83 ||
> + sps->profile_idc == 86 || sps->profile_idc == 118 ||
> + sps->profile_idc == 128 || sps->profile_idc == 138 ||
> + sps->profile_idc == 139 || sps->profile_idc == 134 ||
> + sps->profile_idc == 135) {
> + READ_UEV(chroma_format_idc);
> +
> + if (sps->chroma_format_idc == 3)
> + READ_BIT(separate_colour_plane_flag);
> +
> + READ_UEV(bit_depth_luma_minus8);
> + READ_UEV(bit_depth_chroma_minus8);
> + READ_BIT(qpprime_y_zero_transform_bypass_flag);
> + READ_BIT(seq_scaling_matrix_present_flag);
> +
> + if (sps->seq_scaling_matrix_present_flag) {
> + dev_err(dev,
> + "%s: Handling scaling matrix not supported\n",
> + __func__);
> + return -EINVAL;
> + }
> + }
> +
> + READ_UEV(log2_max_frame_num_minus4);
> +
> + READ_UEV(pic_order_cnt_type);
> + if (sps->pic_order_cnt_type == 0) {
> + READ_UEV(log2_max_pic_order_cnt_lsb_minus4);
> + } else if (sps->pic_order_cnt_type == 1) {
> + READ_BIT(delta_pic_order_always_zero_flag);
> + READ_SEV(offset_for_non_ref_pic);
> + READ_SEV(offset_for_top_to_bottom_field);
> +
> + READ_UEV(num_ref_frames_in_pic_order_cnt_cycle);
> + for (i = 0; i < sps->num_ref_frames_in_pic_order_cnt_cycle; i++)
> + READ_SEV(offset_for_ref_frame[i]);
> + } else {
> + dev_err(dev,
> + "%s: Invalid pic_order_cnt_type %u\n", __func__,
> + sps->pic_order_cnt_type);
> + return -EINVAL;
> + }
> +
> + READ_UEV(max_num_ref_frames);
> + READ_BIT(gaps_in_frame_num_value_allowed_flag);
> + READ_UEV(pic_width_in_mbs_minus1);
> + READ_UEV(pic_height_in_map_units_minus1);
> +
> + READ_BIT(frame_mbs_only_flag);
> + if (!sps->frame_mbs_only_flag)
> + READ_BIT(mb_adaptive_frame_field_flag);
> +
> + READ_BIT(direct_8x8_inference_flag);
> +
> + READ_BIT(frame_cropping_flag);
> + if (sps->frame_cropping_flag) {
> + READ_UEV(crop_left);
> + READ_UEV(crop_right);
> + READ_UEV(crop_top);
> + READ_UEV(crop_bottom);
> + }
> +
> + READ_BIT(vui_parameters_present_flag);
> + if (sps->vui_parameters_present_flag) {
> + err = nal_h264_read_vui_parameters(dev, rbsp, &sps->vui);
> + if (err)
> + return err;
> + }
> +
> + return 0;
> +}
> +
> +static int nal_h264_rbsp_write_pps(const struct device *dev,
> + struct rbsp *rbsp, struct nal_h264_pps *pps)
> +{
> + struct nal_h264_pps *s = pps;
> + int i;
> +
> + WRITE_UEV(pic_parameter_set_id);
> + WRITE_UEV(seq_parameter_set_id);
> + WRITE_BIT(entropy_coding_mode_flag);
> + WRITE_BIT(bottom_field_pic_order_in_frame_present_flag);
> + WRITE_UEV(num_slice_groups_minus1);
> + if (pps->num_slice_groups_minus1 > 0) {
> + WRITE_UEV(slice_group_map_type);
> + if (pps->slice_group_map_type == 0) {
> + for (i = 0; i < pps->num_slice_groups_minus1; i++)
> + WRITE_UEV(run_length_minus1[i]);
> + } else if (pps->slice_group_map_type == 2) {
> + for (i = 0; i < pps->num_slice_groups_minus1; i++) {
> + WRITE_UEV(top_left[i]);
> + WRITE_UEV(bottom_right[i]);
> + }
> + } else if (pps->slice_group_map_type == 3 ||
> + pps->slice_group_map_type == 4 ||
> + pps->slice_group_map_type == 5) {
> + WRITE_BIT(slice_group_change_direction_flag);
> + WRITE_UEV(slice_group_change_rate_minus1);
> + } else if (pps->slice_group_map_type == 6) {
> + WRITE_UEV(pic_size_in_map_units_minus1);
> + for (i = 0; i < pps->pic_size_in_map_units_minus1; i++)
> + WRITE_BITS(order_base_2
> + (s->num_slice_groups_minus1 + 1),
> + slice_group_id[i]);
> + }
> + }
> + WRITE_UEV(num_ref_idx_l0_default_active_minus1);
> + WRITE_UEV(num_ref_idx_l1_default_active_minus1);
> + WRITE_BIT(weighted_pred_flag);
> + WRITE_BITS(2, weighted_bipred_idc);
> + WRITE_SEV(pic_init_qp_minus26);
> + WRITE_SEV(pic_init_qs_minus26);
> + WRITE_SEV(chroma_qp_index_offset);
> + WRITE_BIT(deblocking_filter_control_present_flag);
> + WRITE_BIT(constrained_intra_pred_flag);
> + WRITE_BIT(redundant_pic_cnt_present_flag);
> + if (/* more_rbsp_data() */ false) {
> + WRITE_BIT(transform_8x8_mode_flag);
> + WRITE_BIT(pic_scaling_matrix_present_flag);
> + if (pps->pic_scaling_matrix_present_flag) {
> + dev_err(dev,
> + "%s: Handling scaling matrix not supported\n",
> + __func__);
> + return -EINVAL;
> + }
> + WRITE_SEV(second_chroma_qp_index_offset);
> + }
> +
> + return 0;
> +}
> +
> +static int nal_h264_rbsp_read_pps(const struct device *dev,
> + struct rbsp *rbsp, struct nal_h264_pps *pps)
> +{
> + struct nal_h264_pps *s = pps;
> + unsigned int i;
> +
> + READ_UEV(pic_parameter_set_id);
> + READ_UEV(seq_parameter_set_id);
> + READ_BIT(entropy_coding_mode_flag);
> + READ_BIT(bottom_field_pic_order_in_frame_present_flag);
> + READ_UEV(num_slice_groups_minus1);
> + if (s->num_slice_groups_minus1 > 0) {
> + READ_UEV(slice_group_map_type);
> + if (pps->slice_group_map_type == 0) {
> + for (i = 0; i < pps->num_slice_groups_minus1; i++)
> + READ_UEV(run_length_minus1[i]);
> + } else if (pps->slice_group_map_type == 2) {
> + for (i = 0; i < pps->num_slice_groups_minus1; i++) {
> + READ_UEV(top_left[i]);
> + READ_UEV(bottom_right[i]);
> + }
> + } else if (s->slice_group_map_type == 3 ||
> + s->slice_group_map_type == 4 ||
> + s->slice_group_map_type == 5) {
> + READ_BIT(slice_group_change_direction_flag);
> + READ_UEV(slice_group_change_rate_minus1);
> + } else if (s->slice_group_map_type == 6) {
> + READ_UEV(pic_size_in_map_units_minus1);
> + for (i = 0; i < s->pic_size_in_map_units_minus1; i++)
> + READ_BITS(order_base_2
> + (s->num_slice_groups_minus1 + 1),
> + slice_group_id[i]);
> + }
> + }
> + READ_UEV(num_ref_idx_l0_default_active_minus1);
> + READ_UEV(num_ref_idx_l1_default_active_minus1);
> + READ_BIT(weighted_pred_flag);
> + READ_BITS(2, weighted_bipred_idc);
> + READ_SEV(pic_init_qp_minus26);
> + READ_SEV(pic_init_qs_minus26);
> + READ_SEV(chroma_qp_index_offset);
> + READ_BIT(deblocking_filter_control_present_flag);
> + READ_BIT(constrained_intra_pred_flag);
> + READ_BIT(redundant_pic_cnt_present_flag);
> + if (/* more_rbsp_data() */ false) {
> + READ_BIT(transform_8x8_mode_flag);
> + READ_BIT(pic_scaling_matrix_present_flag);
> + if (pps->pic_scaling_matrix_present_flag) {
> + dev_err(dev,
> + "%s: Handling scaling matrix not supported\n",
> + __func__);
> + return -EINVAL;
> + }
> + READ_SEV(second_chroma_qp_index_offset);
> + }
> +
> + return 0;
> +}
> +
> +ssize_t nal_h264_write_sps(const struct device *dev,
> + void *dest, size_t n, struct nal_h264_sps *sps)
> +{
> + struct rbsp rbsp;
> + int err;
> + u8 *p = dest;
> +
> + rbsp.buf = p + 5;
> + rbsp.size = n - 5;
> + rbsp.pos = 0;
> +
> + err = nal_h264_rbsp_write_sps(dev, &rbsp, sps);
> + if (err)
> + return err;
> +
> + err = nal_h264_write_trailing_bits(dev, &rbsp);
> + if (err)
> + return err;
> +
> + p[0] = 0x00;
> + p[1] = 0x00;
> + p[2] = 0x00;
> + p[3] = 0x01;
> + p[4] = 0x07;
> +
> + return ((rbsp.pos + 7) / 8) + 5;
> +}
> +EXPORT_SYMBOL_GPL(nal_h264_write_sps);
> +
> +ssize_t nal_h264_read_sps(const struct device *dev,
> + struct nal_h264_sps *sps, void *src, size_t n)
> +{
> + struct rbsp rbsp;
> + int err;
> +
> + rbsp.buf = src;
> + rbsp.size = n;
> + rbsp.pos = 0;
> +
> + rbsp.buf += 5;
> + rbsp.size -= 5;
> +
> + err = nal_h264_rbsp_read_sps(dev, &rbsp, sps);
> + if (err)
> + return err;
> +
> + return ((rbsp.pos + 7) / 8) + 5;
> +}
> +EXPORT_SYMBOL_GPL(nal_h264_read_sps);
> +
> +void nal_h264_print_sps(const struct device *dev, struct nal_h264_sps *sps)
> +{
> + struct nal_h264_sps *s = sps;
> + unsigned int i;
> +
> + if (!sps)
> + return;
> +
> + PRINT_BITS(8, profile_idc);
> + PRINT_BIT(constraint_set0_flag);
> + PRINT_BIT(constraint_set1_flag);
> + PRINT_BIT(constraint_set2_flag);
> + PRINT_BIT(constraint_set3_flag);
> + PRINT_BIT(constraint_set4_flag);
> + PRINT_BIT(constraint_set5_flag);
> + PRINT_BITS(2, reserved_zero_2bits);
> + PRINT_BITS(8, level_idc);
> +
> + PRINT_UEV(seq_parameter_set_id);
> +
> + if (sps->profile_idc == 100 || sps->profile_idc == 110 ||
> + sps->profile_idc == 122 || sps->profile_idc == 244 ||
> + sps->profile_idc == 44 || sps->profile_idc == 83 ||
> + sps->profile_idc == 86 || sps->profile_idc == 118 ||
> + sps->profile_idc == 128 || sps->profile_idc == 138 ||
> + sps->profile_idc == 139 || sps->profile_idc == 134 ||
> + sps->profile_idc == 135) {
> + PRINT_UEV(chroma_format_idc);
> +
> + if (sps->chroma_format_idc == 3)
> + PRINT_BIT(separate_colour_plane_flag);
> +
> + PRINT_UEV(bit_depth_luma_minus8);
> + PRINT_UEV(bit_depth_chroma_minus8);
> + PRINT_BIT(qpprime_y_zero_transform_bypass_flag);
> + PRINT_BIT(seq_scaling_matrix_present_flag);
> +
> + if (sps->seq_scaling_matrix_present_flag)
> + dev_err(dev,
> + "%s: Handling scaling matrix not supported\n",
> + __func__);
> + }
> +
> + PRINT_UEV(log2_max_frame_num_minus4);
> +
> + PRINT_UEV(pic_order_cnt_type);
> + if (sps->pic_order_cnt_type == 0) {
> + PRINT_UEV(log2_max_pic_order_cnt_lsb_minus4);
> + } else if (sps->pic_order_cnt_type == 1) {
> + PRINT_BIT(delta_pic_order_always_zero_flag);
> + PRINT_SEV(offset_for_non_ref_pic);
> + PRINT_SEV(offset_for_top_to_bottom_field);
> +
> + PRINT_UEV(num_ref_frames_in_pic_order_cnt_cycle);
> + for (i = 0; i < sps->num_ref_frames_in_pic_order_cnt_cycle; i++)
> + PRINT_SEV(offset_for_ref_frame[i]);
> + } else {
> + dev_err(dev,
> + "%s: Invalid pic_order_cnt_type %u\n", __func__,
> + sps->pic_order_cnt_type);
> + }
> +
> + PRINT_UEV(max_num_ref_frames);
> + PRINT_BIT(gaps_in_frame_num_value_allowed_flag);
> + PRINT_UEV(pic_width_in_mbs_minus1);
> + PRINT_UEV(pic_height_in_map_units_minus1);
> +
> + PRINT_BIT(frame_mbs_only_flag);
> + if (!sps->frame_mbs_only_flag)
> + PRINT_BIT(mb_adaptive_frame_field_flag);
> +
> + PRINT_BIT(direct_8x8_inference_flag);
> +
> + PRINT_BIT(frame_cropping_flag);
> + if (sps->frame_cropping_flag) {
> + PRINT_UEV(crop_left);
> + PRINT_UEV(crop_right);
> + PRINT_UEV(crop_top);
> + PRINT_UEV(crop_bottom);
> + }
> +
> + PRINT_BIT(vui_parameters_present_flag);
> + if (sps->vui_parameters_present_flag)
> + nal_h264_print_vui_parameters(dev, &sps->vui);
> +}
> +EXPORT_SYMBOL_GPL(nal_h264_print_sps);
> +
> +ssize_t nal_h264_write_pps(const struct device *dev,
> + void *dest, size_t n, struct nal_h264_pps *pps)
> +{
> + struct rbsp rbsp;
> + int err;
> + u8 *p = dest;
> +
> + rbsp.buf = p + 5;
> + rbsp.size = n - 5;
> + rbsp.pos = 0;
> +
> + err = nal_h264_rbsp_write_pps(dev, &rbsp, pps);
> + if (err)
> + return err;
> +
> + err = nal_h264_write_trailing_bits(dev, &rbsp);
> + if (err)
> + return err;
> +
> + p[0] = 0x00;
> + p[1] = 0x00;
> + p[2] = 0x00;
> + p[3] = 0x01;
> + p[4] = 0x08;
> +
> + return ((rbsp.pos + 7) / 8) + 5;
> +}
> +EXPORT_SYMBOL_GPL(nal_h264_write_pps);
> +
> +ssize_t nal_h264_read_pps(const struct device *dev,
> + struct nal_h264_pps *pps, void *src, size_t n)
> +{
> + struct rbsp rbsp;
> + int err;
> +
> + rbsp.buf = src;
> + rbsp.size = n;
> + rbsp.pos = 0;
> +
> + rbsp.buf += 5;
> + rbsp.size -= 5;
> +
> + err = nal_h264_rbsp_read_pps(dev, &rbsp, pps);
> + if (err)
> + return err;
> +
> + return ((rbsp.pos + 7) / 8) + 5;
> +}
> +EXPORT_SYMBOL_GPL(nal_h264_read_pps);
> +
> +void nal_h264_print_pps(const struct device *dev, struct nal_h264_pps *pps)
> +{
> + struct nal_h264_pps *s = pps;
> + unsigned int i;
> +
> + if (!pps)
> + return;
> +
> + PRINT_UEV(pic_parameter_set_id);
> + PRINT_UEV(seq_parameter_set_id);
> + PRINT_BIT(entropy_coding_mode_flag);
> + PRINT_BIT(bottom_field_pic_order_in_frame_present_flag);
> + PRINT_UEV(num_slice_groups_minus1);
> + if (s->num_slice_groups_minus1 > 0) {
> + PRINT_UEV(slice_group_map_type);
> + if (pps->slice_group_map_type == 0) {
> + for (i = 0; i < pps->num_slice_groups_minus1; i++)
> + PRINT_UEV(run_length_minus1[i]);
> + } else if (pps->slice_group_map_type == 2) {
> + for (i = 0; i < pps->num_slice_groups_minus1; i++) {
> + PRINT_UEV(top_left[i]);
> + PRINT_UEV(bottom_right[i]);
> + }
> + } else if (s->slice_group_map_type == 3 ||
> + s->slice_group_map_type == 4 ||
> + s->slice_group_map_type == 5) {
> + PRINT_BIT(slice_group_change_direction_flag);
> + PRINT_UEV(slice_group_change_rate_minus1);
> + } else if (s->slice_group_map_type == 6) {
> + PRINT_UEV(pic_size_in_map_units_minus1);
> + for (i = 0; i < s->pic_size_in_map_units_minus1; i++)
> + PRINT_BITS(order_base_2
> + (s->num_slice_groups_minus1 + 1),
> + slice_group_id[i]);
> + }
> + }
> + PRINT_UEV(num_ref_idx_l0_default_active_minus1);
> + PRINT_UEV(num_ref_idx_l1_default_active_minus1);
> + PRINT_BIT(weighted_pred_flag);
> + PRINT_BITS(2, weighted_bipred_idc);
> + PRINT_SEV(pic_init_qp_minus26);
> + PRINT_SEV(pic_init_qs_minus26);
> + PRINT_SEV(chroma_qp_index_offset);
> + PRINT_BIT(deblocking_filter_control_present_flag);
> + PRINT_BIT(constrained_intra_pred_flag);
> + PRINT_BIT(redundant_pic_cnt_present_flag);
> + if (/* more_rbsp_data() */ false) {
> + PRINT_BIT(transform_8x8_mode_flag);
> + PRINT_BIT(pic_scaling_matrix_present_flag);
> + if (pps->pic_scaling_matrix_present_flag) {
> + dev_err(dev,
> + "%s: Handling scaling matrix not supported\n",
> + __func__);
> + }
> + PRINT_SEV(second_chroma_qp_index_offset);
> + }
> +}
> +EXPORT_SYMBOL_GPL(nal_h264_print_pps);
> +
> +ssize_t nal_h264_read_filler(const struct device *dev, void *src, size_t n)
> +{
> + char *p = src;
> + size_t i = 5;
> +
> + if (p[0] != 0x00 || p[1] != 0x00 || p[2] != 0x00 || p[3] != 0x01)
> + return -EINVAL;
> +
> + if (p[4] != 0x0c)
> + return -EINVAL;
> +
> + while (p[i] == 0xff && i < n)
> + i++;
> +
> + if (p[i] != 0x80)
> + return -EINVAL;
> +
> + return i;
> +}
> +EXPORT_SYMBOL_GPL(nal_h264_read_filler);
> +
> +ssize_t nal_h264_write_filler(const struct device *dev, void *dest, size_t n)
> +{
> + char *p = dest;
> +
> + if (n < 6)
> + return -EINVAL;
> +
> + p[0] = 0x00;
> + p[1] = 0x00;
> + p[2] = 0x00;
> + p[3] = 0x01;
> + p[4] = 0x0c;
> + memset(p + 5, 0xff, n - 6);
> + p[n - 1] = 0x80;
> +
> + return n;
> +}
> +EXPORT_SYMBOL_GPL(nal_h264_write_filler);
> diff --git a/drivers/staging/media/allegro-dvt/nal-h264.h b/drivers/staging/media/allegro-dvt/nal-h264.h
> new file mode 100644
> index 000000000000..bd46e8e42caf
> --- /dev/null
> +++ b/drivers/staging/media/allegro-dvt/nal-h264.h
> @@ -0,0 +1,188 @@
> +/* SPDX-License-Identifier: GPL-2.0 */
> +/*
> + * Copyright (C) 2019 Pengutronix, Michael Tretter <kernel@xxxxxxxxxxxxxx>
> + *
> + * Convert NAL units between raw byte sequence payloads (RBSP) and C structs.
> + */
> +
> +#ifndef __NAL_H264_H__
> +#define __NAL_H264_H__
> +
> +#include <linux/kernel.h>
> +#include <linux/types.h>
> +
> +/* Rec. ITU-T H.264 (04/2017) E.1.2 */
> +struct nal_h264_hrd_parameters {
> + unsigned int cpb_cnt_minus1;
> + unsigned int bit_rate_scale:4;
> + unsigned int cpb_size_scale:4;
> + struct {
> + int bit_rate_value_minus1[16];
> + int cpb_size_value_minus1[16];
> + unsigned int cbr_flag[16]; /* FIXME cbr_flag:1 */
> + };
> + unsigned int initial_cpb_removal_delay_length_minus1:5;
> + unsigned int cpb_removal_delay_length_minus1:5;
> + unsigned int dpb_output_delay_length_minus1:5;
> + unsigned int time_offset_length:5;
> +};
> +
> +/* Rec. ITU-T H.264 (04/2017) E.1.1 */
> +struct nal_h264_vui_parameters {
> + unsigned int aspect_ratio_info_present_flag:1;
> + struct {
> + unsigned int aspect_ratio_idc:8;
> + unsigned int sar_width:16;
> + unsigned int sar_height:16;
> + };
> + unsigned int overscan_info_present_flag:1;
> + unsigned int overscan_appropriate_flag:1;
> + unsigned int video_signal_type_present_flag:1;
> + struct {
> + unsigned int video_format:3;
> + unsigned int video_full_range_flag:1;
> + unsigned int colour_description_present_flag:1;
> + struct {
> + unsigned int colour_primaries:8;
> + unsigned int transfer_characteristics:8;
> + unsigned int matrix_coefficients:8;
> + };
> + };
> + unsigned int chroma_loc_info_present_flag:1;
> + struct {
> + unsigned int chroma_sample_loc_type_top_field;
> + unsigned int chroma_sample_loc_type_bottom_field;
> + };
> + unsigned int timing_info_present_flag:1;
> + struct {
> + unsigned int num_units_in_tick:32;
> + unsigned int time_scale:32;
> + unsigned int fixed_frame_rate_flag:1;
> + };
> + unsigned int nal_hrd_parameters_present_flag:1;
> + struct nal_h264_hrd_parameters nal_hrd_parameters;
> + unsigned int vcl_hrd_parameters_present_flag:1;
> + struct nal_h264_hrd_parameters vcl_hrd_parameters;
> + unsigned int low_delay_hrd_flag:1;
> + unsigned int pic_struct_present_flag:1;
> + unsigned int bitstream_restriction_flag:1;
> + struct {
> + unsigned int motion_vectors_over_pic_boundaries_flag:1;
> + unsigned int max_bytes_per_pic_denom;
> + unsigned int max_bits_per_mb_denom;
> + unsigned int log2_max_mv_length_horizontal;
> + unsigned int log21_max_mv_length_vertical;
> + unsigned int max_num_reorder_frames;
> + unsigned int max_dec_frame_buffering;
> + };
> +};
> +
> +/* Rec. ITU-T H.264 (04/2017) 7.3.2.1.1 Sequence parameter set data syntax */
> +struct nal_h264_sps {
> + unsigned int profile_idc:8;
> + unsigned int constraint_set0_flag:1;
> + unsigned int constraint_set1_flag:1;
> + unsigned int constraint_set2_flag:1;
> + unsigned int constraint_set3_flag:1;
> + unsigned int constraint_set4_flag:1;
> + unsigned int constraint_set5_flag:1;
> + unsigned int reserved_zero_2bits:2;
> + unsigned int level_idc:8;
> + unsigned int seq_parameter_set_id;
> + struct {
> + unsigned int chroma_format_idc;
> + unsigned int separate_colour_plane_flag:1;
> + unsigned int bit_depth_luma_minus8;
> + unsigned int bit_depth_chroma_minus8;
> + unsigned int qpprime_y_zero_transform_bypass_flag:1;
> + unsigned int seq_scaling_matrix_present_flag:1;
> + };
> + unsigned int log2_max_frame_num_minus4;
> + unsigned int pic_order_cnt_type;
> + union {
> + unsigned int log2_max_pic_order_cnt_lsb_minus4;
> + struct {
> + unsigned int delta_pic_order_always_zero_flag:1;
> + int offset_for_non_ref_pic;
> + int offset_for_top_to_bottom_field;
> + unsigned int num_ref_frames_in_pic_order_cnt_cycle;
> + int offset_for_ref_frame[255];
> + };
> + };
> + unsigned int max_num_ref_frames;
> + unsigned int gaps_in_frame_num_value_allowed_flag:1;
> + unsigned int pic_width_in_mbs_minus1;
> + unsigned int pic_height_in_map_units_minus1;
> + unsigned int frame_mbs_only_flag:1;
> + unsigned int mb_adaptive_frame_field_flag:1;
> + unsigned int direct_8x8_inference_flag:1;
> + unsigned int frame_cropping_flag:1;
> + struct {
> + unsigned int crop_left;
> + unsigned int crop_right;
> + unsigned int crop_top;
> + unsigned int crop_bottom;
> + };
> + unsigned int vui_parameters_present_flag:1;
> + struct nal_h264_vui_parameters vui;
> +};
> +
> +/* Rec. ITU-T H.264 (04/2017) 7.3.2.2 Picture parameter set RBSP syntax */
> +struct nal_h264_pps {
> + unsigned int pic_parameter_set_id;
> + unsigned int seq_parameter_set_id;
> + unsigned int entropy_coding_mode_flag:1;
> + unsigned int bottom_field_pic_order_in_frame_present_flag:1;
> + unsigned int num_slice_groups_minus1;
> + unsigned int slice_group_map_type;
> + union {
> + unsigned int run_length_minus1[8];
> + struct {
> + unsigned int top_left[8];
> + unsigned int bottom_right[8];
> + };
> + struct {
> + unsigned int slice_group_change_direction_flag:1;
> + unsigned int slice_group_change_rate_minus1;
> + };
> + struct {
> + unsigned int pic_size_in_map_units_minus1;
> + unsigned int slice_group_id[8];
> + };
> + };
> + unsigned int num_ref_idx_l0_default_active_minus1;
> + unsigned int num_ref_idx_l1_default_active_minus1;
> + unsigned int weighted_pred_flag:1;
> + unsigned int weighted_bipred_idc:2;
> + int pic_init_qp_minus26;
> + int pic_init_qs_minus26;
> + int chroma_qp_index_offset;
> + unsigned int deblocking_filter_control_present_flag:1;
> + unsigned int constrained_intra_pred_flag:1;
> + unsigned int redundant_pic_cnt_present_flag:1;
> + struct {
> + unsigned int transform_8x8_mode_flag:1;
> + unsigned int pic_scaling_matrix_present_flag:1;
> + int second_chroma_qp_index_offset;
> + };
> +};
> +
> +int nal_h264_level_from_v4l2(enum v4l2_mpeg_video_h264_level level);
> +int nal_h264_profile_from_v4l2(enum v4l2_mpeg_video_h264_profile profile);
> +
> +ssize_t nal_h264_write_sps(const struct device *dev,
> + void *dest, size_t n, struct nal_h264_sps *sps);
> +ssize_t nal_h264_read_sps(const struct device *dev,
> + struct nal_h264_sps *sps, void *src, size_t n);
> +void nal_h264_print_sps(const struct device *dev, struct nal_h264_sps *sps);
> +
> +ssize_t nal_h264_write_pps(const struct device *dev,
> + void *dest, size_t n, struct nal_h264_pps *pps);
> +ssize_t nal_h264_read_pps(const struct device *dev,
> + struct nal_h264_pps *pps, void *src, size_t n);
> +void nal_h264_print_pps(const struct device *dev, struct nal_h264_pps *pps);
> +
> +ssize_t nal_h264_write_filler(const struct device *dev, void *dest, size_t n);
> +ssize_t nal_h264_read_filler(const struct device *dev, void *src, size_t n);
Can you document these functions? I think there is a good chance that these
functions can be reused in other drivers in the future, so having some
documentation here would be very useful.
Regards,
Hans
> +
> +#endif /* __NAL_H264_H__ */
>
