forked from Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
268 lines
7.9 KiB
268 lines
7.9 KiB
// SPDX-License-Identifier: GPL-2.0 |
|
/* |
|
* V4L2 H264 helpers. |
|
* |
|
* Copyright (C) 2019 Collabora, Ltd. |
|
* |
|
* Author: Boris Brezillon <[email protected]> |
|
*/ |
|
|
|
#include <linux/module.h> |
|
#include <linux/sort.h> |
|
|
|
#include <media/v4l2-h264.h> |
|
|
|
/** |
|
* v4l2_h264_init_reflist_builder() - Initialize a P/B0/B1 reference list |
|
* builder |
|
* |
|
* @b: the builder context to initialize |
|
* @dec_params: decode parameters control |
|
* @sps: SPS control |
|
* @dpb: DPB to use when creating the reference list |
|
*/ |
|
void |
|
v4l2_h264_init_reflist_builder(struct v4l2_h264_reflist_builder *b, |
|
const struct v4l2_ctrl_h264_decode_params *dec_params, |
|
const struct v4l2_ctrl_h264_sps *sps, |
|
const struct v4l2_h264_dpb_entry dpb[V4L2_H264_NUM_DPB_ENTRIES]) |
|
{ |
|
int cur_frame_num, max_frame_num; |
|
unsigned int i; |
|
|
|
max_frame_num = 1 << (sps->log2_max_frame_num_minus4 + 4); |
|
cur_frame_num = dec_params->frame_num; |
|
|
|
memset(b, 0, sizeof(*b)); |
|
if (!(dec_params->flags & V4L2_H264_DECODE_PARAM_FLAG_FIELD_PIC)) |
|
b->cur_pic_order_count = min(dec_params->bottom_field_order_cnt, |
|
dec_params->top_field_order_cnt); |
|
else if (dec_params->flags & V4L2_H264_DECODE_PARAM_FLAG_BOTTOM_FIELD) |
|
b->cur_pic_order_count = dec_params->bottom_field_order_cnt; |
|
else |
|
b->cur_pic_order_count = dec_params->top_field_order_cnt; |
|
|
|
for (i = 0; i < V4L2_H264_NUM_DPB_ENTRIES; i++) { |
|
u32 pic_order_count; |
|
|
|
if (!(dpb[i].flags & V4L2_H264_DPB_ENTRY_FLAG_ACTIVE)) |
|
continue; |
|
|
|
b->refs[i].pic_num = dpb[i].pic_num; |
|
if (dpb[i].flags & V4L2_H264_DPB_ENTRY_FLAG_LONG_TERM) |
|
b->refs[i].longterm = true; |
|
|
|
/* |
|
* Handle frame_num wraparound as described in section |
|
* '8.2.4.1 Decoding process for picture numbers' of the spec. |
|
* TODO: This logic will have to be adjusted when we start |
|
* supporting interlaced content. |
|
*/ |
|
if (dpb[i].frame_num > cur_frame_num) |
|
b->refs[i].frame_num = (int)dpb[i].frame_num - |
|
max_frame_num; |
|
else |
|
b->refs[i].frame_num = dpb[i].frame_num; |
|
|
|
if (dpb[i].fields == V4L2_H264_FRAME_REF) |
|
pic_order_count = min(dpb[i].top_field_order_cnt, |
|
dpb[i].bottom_field_order_cnt); |
|
else if (dpb[i].fields & V4L2_H264_BOTTOM_FIELD_REF) |
|
pic_order_count = dpb[i].bottom_field_order_cnt; |
|
else |
|
pic_order_count = dpb[i].top_field_order_cnt; |
|
|
|
b->refs[i].pic_order_count = pic_order_count; |
|
b->unordered_reflist[b->num_valid] = i; |
|
b->num_valid++; |
|
} |
|
|
|
for (i = b->num_valid; i < ARRAY_SIZE(b->unordered_reflist); i++) |
|
b->unordered_reflist[i] = i; |
|
} |
|
EXPORT_SYMBOL_GPL(v4l2_h264_init_reflist_builder); |
|
|
|
static int v4l2_h264_p_ref_list_cmp(const void *ptra, const void *ptrb, |
|
const void *data) |
|
{ |
|
const struct v4l2_h264_reflist_builder *builder = data; |
|
u8 idxa, idxb; |
|
|
|
idxa = *((u8 *)ptra); |
|
idxb = *((u8 *)ptrb); |
|
|
|
if (WARN_ON(idxa >= V4L2_H264_NUM_DPB_ENTRIES || |
|
idxb >= V4L2_H264_NUM_DPB_ENTRIES)) |
|
return 1; |
|
|
|
if (builder->refs[idxa].longterm != builder->refs[idxb].longterm) { |
|
/* Short term pics first. */ |
|
if (!builder->refs[idxa].longterm) |
|
return -1; |
|
else |
|
return 1; |
|
} |
|
|
|
/* |
|
* Short term pics in descending pic num order, long term ones in |
|
* ascending order. |
|
*/ |
|
if (!builder->refs[idxa].longterm) |
|
return builder->refs[idxb].frame_num < |
|
builder->refs[idxa].frame_num ? |
|
-1 : 1; |
|
|
|
return builder->refs[idxa].pic_num < builder->refs[idxb].pic_num ? |
|
-1 : 1; |
|
} |
|
|
|
static int v4l2_h264_b0_ref_list_cmp(const void *ptra, const void *ptrb, |
|
const void *data) |
|
{ |
|
const struct v4l2_h264_reflist_builder *builder = data; |
|
s32 poca, pocb; |
|
u8 idxa, idxb; |
|
|
|
idxa = *((u8 *)ptra); |
|
idxb = *((u8 *)ptrb); |
|
|
|
if (WARN_ON(idxa >= V4L2_H264_NUM_DPB_ENTRIES || |
|
idxb >= V4L2_H264_NUM_DPB_ENTRIES)) |
|
return 1; |
|
|
|
if (builder->refs[idxa].longterm != builder->refs[idxb].longterm) { |
|
/* Short term pics first. */ |
|
if (!builder->refs[idxa].longterm) |
|
return -1; |
|
else |
|
return 1; |
|
} |
|
|
|
/* Long term pics in ascending pic num order. */ |
|
if (builder->refs[idxa].longterm) |
|
return builder->refs[idxa].pic_num < |
|
builder->refs[idxb].pic_num ? |
|
-1 : 1; |
|
|
|
poca = builder->refs[idxa].pic_order_count; |
|
pocb = builder->refs[idxb].pic_order_count; |
|
|
|
/* |
|
* Short term pics with POC < cur POC first in POC descending order |
|
* followed by short term pics with POC > cur POC in POC ascending |
|
* order. |
|
*/ |
|
if ((poca < builder->cur_pic_order_count) != |
|
(pocb < builder->cur_pic_order_count)) |
|
return poca < pocb ? -1 : 1; |
|
else if (poca < builder->cur_pic_order_count) |
|
return pocb < poca ? -1 : 1; |
|
|
|
return poca < pocb ? -1 : 1; |
|
} |
|
|
|
static int v4l2_h264_b1_ref_list_cmp(const void *ptra, const void *ptrb, |
|
const void *data) |
|
{ |
|
const struct v4l2_h264_reflist_builder *builder = data; |
|
s32 poca, pocb; |
|
u8 idxa, idxb; |
|
|
|
idxa = *((u8 *)ptra); |
|
idxb = *((u8 *)ptrb); |
|
|
|
if (WARN_ON(idxa >= V4L2_H264_NUM_DPB_ENTRIES || |
|
idxb >= V4L2_H264_NUM_DPB_ENTRIES)) |
|
return 1; |
|
|
|
if (builder->refs[idxa].longterm != builder->refs[idxb].longterm) { |
|
/* Short term pics first. */ |
|
if (!builder->refs[idxa].longterm) |
|
return -1; |
|
else |
|
return 1; |
|
} |
|
|
|
/* Long term pics in ascending pic num order. */ |
|
if (builder->refs[idxa].longterm) |
|
return builder->refs[idxa].pic_num < |
|
builder->refs[idxb].pic_num ? |
|
-1 : 1; |
|
|
|
poca = builder->refs[idxa].pic_order_count; |
|
pocb = builder->refs[idxb].pic_order_count; |
|
|
|
/* |
|
* Short term pics with POC > cur POC first in POC ascending order |
|
* followed by short term pics with POC < cur POC in POC descending |
|
* order. |
|
*/ |
|
if ((poca < builder->cur_pic_order_count) != |
|
(pocb < builder->cur_pic_order_count)) |
|
return pocb < poca ? -1 : 1; |
|
else if (poca < builder->cur_pic_order_count) |
|
return pocb < poca ? -1 : 1; |
|
|
|
return poca < pocb ? -1 : 1; |
|
} |
|
|
|
/** |
|
* v4l2_h264_build_p_ref_list() - Build the P reference list |
|
* |
|
* @builder: reference list builder context |
|
* @reflist: 16-bytes array used to store the P reference list. Each entry |
|
* is an index in the DPB |
|
* |
|
* This functions builds the P reference lists. This procedure is describe in |
|
* section '8.2.4 Decoding process for reference picture lists construction' |
|
* of the H264 spec. This function can be used by H264 decoder drivers that |
|
* need to pass a P reference list to the hardware. |
|
*/ |
|
void |
|
v4l2_h264_build_p_ref_list(const struct v4l2_h264_reflist_builder *builder, |
|
u8 *reflist) |
|
{ |
|
memcpy(reflist, builder->unordered_reflist, |
|
sizeof(builder->unordered_reflist[0]) * builder->num_valid); |
|
sort_r(reflist, builder->num_valid, sizeof(*reflist), |
|
v4l2_h264_p_ref_list_cmp, NULL, builder); |
|
} |
|
EXPORT_SYMBOL_GPL(v4l2_h264_build_p_ref_list); |
|
|
|
/** |
|
* v4l2_h264_build_b_ref_lists() - Build the B0/B1 reference lists |
|
* |
|
* @builder: reference list builder context |
|
* @b0_reflist: 16-bytes array used to store the B0 reference list. Each entry |
|
* is an index in the DPB |
|
* @b1_reflist: 16-bytes array used to store the B1 reference list. Each entry |
|
* is an index in the DPB |
|
* |
|
* This functions builds the B0/B1 reference lists. This procedure is described |
|
* in section '8.2.4 Decoding process for reference picture lists construction' |
|
* of the H264 spec. This function can be used by H264 decoder drivers that |
|
* need to pass B0/B1 reference lists to the hardware. |
|
*/ |
|
void |
|
v4l2_h264_build_b_ref_lists(const struct v4l2_h264_reflist_builder *builder, |
|
u8 *b0_reflist, u8 *b1_reflist) |
|
{ |
|
memcpy(b0_reflist, builder->unordered_reflist, |
|
sizeof(builder->unordered_reflist[0]) * builder->num_valid); |
|
sort_r(b0_reflist, builder->num_valid, sizeof(*b0_reflist), |
|
v4l2_h264_b0_ref_list_cmp, NULL, builder); |
|
|
|
memcpy(b1_reflist, builder->unordered_reflist, |
|
sizeof(builder->unordered_reflist[0]) * builder->num_valid); |
|
sort_r(b1_reflist, builder->num_valid, sizeof(*b1_reflist), |
|
v4l2_h264_b1_ref_list_cmp, NULL, builder); |
|
|
|
if (builder->num_valid > 1 && |
|
!memcmp(b1_reflist, b0_reflist, builder->num_valid)) |
|
swap(b1_reflist[0], b1_reflist[1]); |
|
} |
|
EXPORT_SYMBOL_GPL(v4l2_h264_build_b_ref_lists); |
|
|
|
MODULE_LICENSE("GPL"); |
|
MODULE_DESCRIPTION("V4L2 H264 Helpers"); |
|
MODULE_AUTHOR("Boris Brezillon <[email protected]>");
|
|
|