mirror of
https://github.com/elwrcl/intel-hard-graphics.git
synced 2026-10-31 10:58:03 +03:00
hasvk: Add a slice header parser for H.264 decode.
This is required for long mode, where the driver parses the slice header itself rather than relying on the hardware to do it. Co-authored-by: irql-notlessorequal <[email protected]> [dancingmirrors: Assisted-by: Claude]
This commit is contained in:
dancingmirrorsandirql-notlessorequal authored and elwrcl committed
1 parent
5c6ad1fb4c
commit
9dad5c1a52
3 files changed
+642
No files matched your search
@@ -0,0 +1,542 @@
|
|||||||
|
/*
|
||||||
|
* Copyright © 2026 [email protected]
|
||||||
|
* Copyright © 2026 irql-notlessorequal
|
||||||
|
*
|
||||||
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
||||||
|
* copy of this software and associated documentation files (the "Software"),
|
||||||
|
* to deal in the Software without restriction, including without limitation
|
||||||
|
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
||||||
|
* and/or sell copies of the Software, and to permit persons to whom the
|
||||||
|
* Software is furnished to do so, subject to the following conditions:
|
||||||
|
*
|
||||||
|
* The above copyright notice and this permission notice (including the next
|
||||||
|
* paragraph) shall be included in all copies or substantial portions of the
|
||||||
|
* Software.
|
||||||
|
*
|
||||||
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||||||
|
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||||
|
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
||||||
|
* IN THE SOFTWARE.
|
||||||
|
*/
|
||||||
|
#include <stdbool.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include "util/macros.h"
|
||||||
|
#include "anv_video_h264_slice.h"
|
||||||
|
|
||||||
|
#define VALID_RBSP_OR_BAIL(st) \
|
||||||
|
if (unlikely(st.error)) \
|
||||||
|
return false;
|
||||||
|
|
||||||
|
static inline void
|
||||||
|
anv_h264_rbsp_init(struct anv_h264_rbsp *st,
|
||||||
|
const uint8_t *buf, size_t size)
|
||||||
|
{
|
||||||
|
st->buf = buf;
|
||||||
|
st->cur = buf;
|
||||||
|
st->end = buf + size;
|
||||||
|
st->bit = 7;
|
||||||
|
st->zeros = 0;
|
||||||
|
st->error = false;
|
||||||
|
st->bits_consumed = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline int
|
||||||
|
anv_h264_rbsp_next_byte(struct anv_h264_rbsp *st)
|
||||||
|
{
|
||||||
|
if (st->cur >= st->end) { st->error = true; return -1; }
|
||||||
|
uint8_t c = *st->cur++;
|
||||||
|
if (c == 0) {
|
||||||
|
st->zeros++;
|
||||||
|
} else {
|
||||||
|
st->zeros = 0;
|
||||||
|
}
|
||||||
|
if (st->zeros >= 2 && st->cur < st->end && *st->cur == 0x03) {
|
||||||
|
st->cur++;
|
||||||
|
st->zeros = 0;
|
||||||
|
}
|
||||||
|
return c;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline int
|
||||||
|
anv_h264_rbsp_consume_bit(struct anv_h264_rbsp *st)
|
||||||
|
{
|
||||||
|
if (st->cur >= st->end) { st->error = true; return 0; }
|
||||||
|
int v = !!(*st->cur & (1u << st->bit));
|
||||||
|
st->bits_consumed++;
|
||||||
|
if (st->bit > 0) {
|
||||||
|
st->bit--;
|
||||||
|
} else {
|
||||||
|
if (anv_h264_rbsp_next_byte(st) < 0) return 0;
|
||||||
|
st->bit = 7;
|
||||||
|
}
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline unsigned
|
||||||
|
anv_h264_rbsp_u(struct anv_h264_rbsp *st, int n)
|
||||||
|
{
|
||||||
|
unsigned v = 0;
|
||||||
|
for (int i = 0; i < n && !st->error; i++)
|
||||||
|
v = (v << 1) | anv_h264_rbsp_consume_bit(st);
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline unsigned
|
||||||
|
anv_h264_rbsp_ue(struct anv_h264_rbsp *st)
|
||||||
|
{
|
||||||
|
int leading = 0;
|
||||||
|
while (!st->error && !anv_h264_rbsp_consume_bit(st))
|
||||||
|
leading++;
|
||||||
|
if (leading == 0) return 0;
|
||||||
|
return (1u << leading) - 1 + anv_h264_rbsp_u(st, leading);
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline int
|
||||||
|
anv_h264_rbsp_se(struct anv_h264_rbsp *st)
|
||||||
|
{
|
||||||
|
unsigned v = anv_h264_rbsp_ue(st);
|
||||||
|
return (v & 1) ? (int)((v + 1) >> 1) : -(int)(v >> 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
anv_h264_build_default_ref_list(
|
||||||
|
int slice_type,
|
||||||
|
int32_t curr_poc,
|
||||||
|
uint32_t ref_slot_count,
|
||||||
|
const VkVideoReferenceSlotInfoKHR *ref_slots,
|
||||||
|
uint8_t list0[ANV_H264_MAX_REF_FRAMES],
|
||||||
|
uint8_t list1[ANV_H264_MAX_REF_FRAMES])
|
||||||
|
{
|
||||||
|
memset(list0, ANV_H264_INVALID_SLOT, ANV_H264_MAX_REF_FRAMES);
|
||||||
|
memset(list1, ANV_H264_INVALID_SLOT, ANV_H264_MAX_REF_FRAMES);
|
||||||
|
|
||||||
|
if (slice_type == ANV_H264_SLICE_I || slice_type == ANV_H264_SLICE_SI)
|
||||||
|
return;
|
||||||
|
|
||||||
|
uint8_t st_slots[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int32_t st_poc[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int st_count = 0;
|
||||||
|
|
||||||
|
uint8_t lt_slots[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
uint32_t lt_idx[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int lt_count = 0;
|
||||||
|
|
||||||
|
for (uint32_t i = 0; i < ref_slot_count; i++) {
|
||||||
|
const VkVideoDecodeH264DpbSlotInfoKHR *dpb =
|
||||||
|
vk_find_struct_const(ref_slots[i].pNext,
|
||||||
|
VIDEO_DECODE_H264_DPB_SLOT_INFO_KHR);
|
||||||
|
if (!dpb) continue;
|
||||||
|
const StdVideoDecodeH264ReferenceInfo *ri = dpb->pStdReferenceInfo;
|
||||||
|
if (!ri) continue;
|
||||||
|
int slot = ref_slots[i].slotIndex;
|
||||||
|
if (ri->flags.used_for_long_term_reference) {
|
||||||
|
lt_slots[lt_count] = (uint8_t)slot;
|
||||||
|
lt_idx[lt_count] = ri->FrameNum;
|
||||||
|
lt_count++;
|
||||||
|
} else {
|
||||||
|
st_slots[st_count] = (uint8_t)slot;
|
||||||
|
st_poc[st_count] = ri->PicOrderCnt[0];
|
||||||
|
st_count++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 1; i < st_count; i++) {
|
||||||
|
uint8_t si = st_slots[i]; int32_t pi = st_poc[i];
|
||||||
|
int j = i - 1;
|
||||||
|
while (j >= 0 && st_poc[j] < pi) {
|
||||||
|
st_slots[j+1] = st_slots[j]; st_poc[j+1] = st_poc[j]; j--;
|
||||||
|
}
|
||||||
|
st_slots[j+1] = si; st_poc[j+1] = pi;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 1; i < lt_count; i++) {
|
||||||
|
uint8_t si = lt_slots[i]; uint32_t li = lt_idx[i];
|
||||||
|
int j = i - 1;
|
||||||
|
while (j >= 0 && lt_idx[j] > li) {
|
||||||
|
lt_slots[j+1] = lt_slots[j]; lt_idx[j+1] = lt_idx[j]; j--;
|
||||||
|
}
|
||||||
|
lt_slots[j+1] = si; lt_idx[j+1] = li;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (slice_type == ANV_H264_SLICE_P || slice_type == ANV_H264_SLICE_SP) {
|
||||||
|
int p = 0;
|
||||||
|
for (int i = 0; i < st_count && p < ANV_H264_MAX_REF_FRAMES; i++)
|
||||||
|
list0[p++] = st_slots[i];
|
||||||
|
for (int i = 0; i < lt_count && p < ANV_H264_MAX_REF_FRAMES; i++)
|
||||||
|
list0[p++] = lt_slots[i];
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint8_t past_slots[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int32_t past_poc[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int past_cnt = 0;
|
||||||
|
uint8_t fut_slots[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int32_t fut_poc[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int fut_cnt = 0;
|
||||||
|
|
||||||
|
for (int i = 0; i < st_count; i++) {
|
||||||
|
if (st_poc[i] < curr_poc) {
|
||||||
|
past_slots[past_cnt] = st_slots[i];
|
||||||
|
past_poc[past_cnt++] = st_poc[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for (int i = st_count - 1; i >= 0; i--) {
|
||||||
|
if (st_poc[i] >= curr_poc) {
|
||||||
|
fut_slots[fut_cnt] = st_slots[i];
|
||||||
|
fut_poc[fut_cnt++] = st_poc[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
(void)past_poc; (void)fut_poc;
|
||||||
|
|
||||||
|
int p0 = 0, p1 = 0;
|
||||||
|
for (int i = 0; i < past_cnt && p0 < ANV_H264_MAX_REF_FRAMES; i++)
|
||||||
|
list0[p0++] = past_slots[i];
|
||||||
|
for (int i = 0; i < fut_cnt && p0 < ANV_H264_MAX_REF_FRAMES; i++)
|
||||||
|
list0[p0++] = fut_slots[i];
|
||||||
|
for (int i = 0; i < lt_count && p0 < ANV_H264_MAX_REF_FRAMES; i++)
|
||||||
|
list0[p0++] = lt_slots[i];
|
||||||
|
|
||||||
|
for (int i = 0; i < fut_cnt && p1 < ANV_H264_MAX_REF_FRAMES; i++)
|
||||||
|
list1[p1++] = fut_slots[i];
|
||||||
|
for (int i = 0; i < past_cnt && p1 < ANV_H264_MAX_REF_FRAMES; i++)
|
||||||
|
list1[p1++] = past_slots[i];
|
||||||
|
for (int i = 0; i < lt_count && p1 < ANV_H264_MAX_REF_FRAMES; i++)
|
||||||
|
list1[p1++] = lt_slots[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
anv_h264_apply_ref_pic_list_mod(
|
||||||
|
struct anv_h264_rbsp *st,
|
||||||
|
uint32_t ref_slot_count,
|
||||||
|
const VkVideoReferenceSlotInfoKHR *ref_slots,
|
||||||
|
uint32_t curr_frame_num,
|
||||||
|
uint32_t MaxFrameNum,
|
||||||
|
uint8_t list[ANV_H264_MAX_REF_FRAMES],
|
||||||
|
int active_count)
|
||||||
|
{
|
||||||
|
int mod_flag = (int)anv_h264_rbsp_u(st, 1);
|
||||||
|
if (!mod_flag || st->error) return;
|
||||||
|
|
||||||
|
uint32_t picNumL = curr_frame_num;
|
||||||
|
int refIdxL = 0;
|
||||||
|
|
||||||
|
unsigned idc;
|
||||||
|
do {
|
||||||
|
idc = anv_h264_rbsp_ue(st);
|
||||||
|
if (st->error) return;
|
||||||
|
if (idc == 3) break;
|
||||||
|
if (idc > 3) { st->error = true; return; }
|
||||||
|
|
||||||
|
if (idc == 0 || idc == 1) {
|
||||||
|
unsigned diff = anv_h264_rbsp_ue(st) + 1;
|
||||||
|
if (st->error) return;
|
||||||
|
if (idc == 0)
|
||||||
|
picNumL = (picNumL - diff + MaxFrameNum) % MaxFrameNum;
|
||||||
|
else
|
||||||
|
picNumL = (picNumL + diff) % MaxFrameNum;
|
||||||
|
|
||||||
|
uint8_t found_slot = ANV_H264_INVALID_SLOT;
|
||||||
|
for (uint32_t j = 0; j < ref_slot_count; j++) {
|
||||||
|
const VkVideoDecodeH264DpbSlotInfoKHR *dpb =
|
||||||
|
vk_find_struct_const(ref_slots[j].pNext,
|
||||||
|
VIDEO_DECODE_H264_DPB_SLOT_INFO_KHR);
|
||||||
|
if (!dpb || !dpb->pStdReferenceInfo) continue;
|
||||||
|
if (!dpb->pStdReferenceInfo->flags.used_for_long_term_reference &&
|
||||||
|
dpb->pStdReferenceInfo->FrameNum == picNumL) {
|
||||||
|
found_slot = (uint8_t)ref_slots[j].slotIndex;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (found_slot == ANV_H264_INVALID_SLOT) continue;
|
||||||
|
|
||||||
|
for (int k = active_count - 1; k > refIdxL; k--)
|
||||||
|
list[k] = list[k - 1];
|
||||||
|
list[refIdxL] = found_slot;
|
||||||
|
|
||||||
|
int w = refIdxL + 1;
|
||||||
|
for (int k = refIdxL + 1; k < active_count + 1 && w < ANV_H264_MAX_REF_FRAMES; k++) {
|
||||||
|
if (list[k] != found_slot)
|
||||||
|
list[w++] = list[k];
|
||||||
|
}
|
||||||
|
if (w < ANV_H264_MAX_REF_FRAMES)
|
||||||
|
list[w] = ANV_H264_INVALID_SLOT;
|
||||||
|
|
||||||
|
refIdxL++;
|
||||||
|
} else if (idc == 2) {
|
||||||
|
anv_h264_rbsp_ue(st);
|
||||||
|
if (st->error) return;
|
||||||
|
}
|
||||||
|
} while (idc != 3);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
anv_h264_parse_pred_weight_table(
|
||||||
|
struct anv_h264_rbsp *st,
|
||||||
|
int ChromaArrayType,
|
||||||
|
int num_l0,
|
||||||
|
int num_l1,
|
||||||
|
int slice_type,
|
||||||
|
struct anv_h264_slice_params *out)
|
||||||
|
{
|
||||||
|
out->luma_log2_weight_denom = (int)anv_h264_rbsp_ue(st);
|
||||||
|
out->chroma_log2_weight_denom = 0;
|
||||||
|
if (ChromaArrayType != 0)
|
||||||
|
out->chroma_log2_weight_denom = (int)anv_h264_rbsp_ue(st);
|
||||||
|
if (st->error) return;
|
||||||
|
|
||||||
|
if (out->luma_log2_weight_denom > 7 ||
|
||||||
|
(ChromaArrayType != 0 && out->chroma_log2_weight_denom > 7)) {
|
||||||
|
st->error = true;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int def_luma = 1 << out->luma_log2_weight_denom;
|
||||||
|
int def_chroma = 1 << out->chroma_log2_weight_denom;
|
||||||
|
|
||||||
|
for (int i = 0; i < num_l0; i++) {
|
||||||
|
out->luma_weight_l0[i] = def_luma; out->luma_offset_l0[i] = 0;
|
||||||
|
for (int c = 0; c < 2; c++) {
|
||||||
|
out->chroma_weight_l0[i][c] = def_chroma;
|
||||||
|
out->chroma_offset_l0[i][c] = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for (int i = 0; i < num_l1; i++) {
|
||||||
|
out->luma_weight_l1[i] = def_luma; out->luma_offset_l1[i] = 0;
|
||||||
|
for (int c = 0; c < 2; c++) {
|
||||||
|
out->chroma_weight_l1[i][c] = def_chroma;
|
||||||
|
out->chroma_offset_l1[i][c] = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 0; i < num_l0 && !st->error; i++) {
|
||||||
|
if (anv_h264_rbsp_u(st, 1)) {
|
||||||
|
out->luma_weight_l0[i] = (int16_t)anv_h264_rbsp_se(st);
|
||||||
|
out->luma_offset_l0[i] = (int16_t)anv_h264_rbsp_se(st);
|
||||||
|
if (out->luma_weight_l0[i] != def_luma || out->luma_offset_l0[i])
|
||||||
|
out->has_weight_offsets_l0 = true;
|
||||||
|
}
|
||||||
|
if (ChromaArrayType != 0 && anv_h264_rbsp_u(st, 1)) {
|
||||||
|
for (int c = 0; c < 2; c++) {
|
||||||
|
out->chroma_weight_l0[i][c] = (int16_t)anv_h264_rbsp_se(st);
|
||||||
|
out->chroma_offset_l0[i][c] = (int16_t)anv_h264_rbsp_se(st);
|
||||||
|
if (out->chroma_weight_l0[i][c] != def_chroma ||
|
||||||
|
out->chroma_offset_l0[i][c])
|
||||||
|
out->has_weight_offsets_l0 = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (slice_type != ANV_H264_SLICE_B) return;
|
||||||
|
|
||||||
|
for (int i = 0; i < num_l1 && !st->error; i++) {
|
||||||
|
if (anv_h264_rbsp_u(st, 1)) {
|
||||||
|
out->luma_weight_l1[i] = (int16_t)anv_h264_rbsp_se(st);
|
||||||
|
out->luma_offset_l1[i] = (int16_t)anv_h264_rbsp_se(st);
|
||||||
|
if (out->luma_weight_l1[i] != def_luma || out->luma_offset_l1[i])
|
||||||
|
out->has_weight_offsets_l1 = true;
|
||||||
|
}
|
||||||
|
if (ChromaArrayType != 0 && anv_h264_rbsp_u(st, 1)) {
|
||||||
|
for (int c = 0; c < 2; c++) {
|
||||||
|
out->chroma_weight_l1[i][c] = (int16_t)anv_h264_rbsp_se(st);
|
||||||
|
out->chroma_offset_l1[i][c] = (int16_t)anv_h264_rbsp_se(st);
|
||||||
|
if (out->chroma_weight_l1[i][c] != def_chroma ||
|
||||||
|
out->chroma_offset_l1[i][c])
|
||||||
|
out->has_weight_offsets_l1 = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
anv_h264_consume_dec_ref_pic_marking(struct anv_h264_rbsp *st, bool is_idr)
|
||||||
|
{
|
||||||
|
if (is_idr) {
|
||||||
|
anv_h264_rbsp_u(st, 1);
|
||||||
|
anv_h264_rbsp_u(st, 1);
|
||||||
|
} else {
|
||||||
|
if (anv_h264_rbsp_u(st, 1)) {
|
||||||
|
unsigned op;
|
||||||
|
do {
|
||||||
|
op = anv_h264_rbsp_ue(st);
|
||||||
|
if (st->error) return;
|
||||||
|
if (op > 6) { st->error = true; return; }
|
||||||
|
if (op == 1 || op == 3) anv_h264_rbsp_ue(st);
|
||||||
|
if (op == 2) anv_h264_rbsp_ue(st);
|
||||||
|
if (op == 3 || op == 6) anv_h264_rbsp_ue(st);
|
||||||
|
if (op == 4) anv_h264_rbsp_ue(st);
|
||||||
|
} while (op != 0 && !st->error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool
|
||||||
|
anv_h264_parse_slice_header(
|
||||||
|
const uint8_t *buf,
|
||||||
|
uint32_t slice_nalu_offset,
|
||||||
|
uint32_t slice_nalu_size,
|
||||||
|
const StdVideoH264SequenceParameterSet *sps,
|
||||||
|
const StdVideoH264PictureParameterSet *pps,
|
||||||
|
const VkVideoDecodeH264PictureInfoKHR *pic_info,
|
||||||
|
uint32_t ref_slot_count,
|
||||||
|
const VkVideoReferenceSlotInfoKHR *ref_slots,
|
||||||
|
struct anv_h264_slice_params *out)
|
||||||
|
{
|
||||||
|
memset(out, 0, sizeof(*out));
|
||||||
|
out->slice_type = ANV_H264_SLICE_I;
|
||||||
|
|
||||||
|
for (int i = 0; i < ANV_H264_MAX_REF_FRAMES; i++) {
|
||||||
|
out->luma_weight_l0[i] = 1; out->luma_weight_l1[i] = 1;
|
||||||
|
for (int c = 0; c < 2; c++) {
|
||||||
|
out->chroma_weight_l0[i][c] = 1;
|
||||||
|
out->chroma_weight_l1[i][c] = 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
memset(out->ref_list0, ANV_H264_INVALID_SLOT, sizeof(out->ref_list0));
|
||||||
|
memset(out->ref_list1, ANV_H264_INVALID_SLOT, sizeof(out->ref_list1));
|
||||||
|
|
||||||
|
struct anv_h264_rbsp st;
|
||||||
|
anv_h264_rbsp_init(&st, buf + slice_nalu_offset, slice_nalu_size);
|
||||||
|
|
||||||
|
anv_h264_rbsp_u(&st, 1);
|
||||||
|
int nal_ref_idc = (int)anv_h264_rbsp_u(&st, 2);
|
||||||
|
int nal_unit_type = (int)anv_h264_rbsp_u(&st, 5);
|
||||||
|
|
||||||
|
VALID_RBSP_OR_BAIL(st)
|
||||||
|
|
||||||
|
bool is_idr = (nal_unit_type == 5);
|
||||||
|
|
||||||
|
out->first_mb_in_slice = (int)anv_h264_rbsp_ue(&st);
|
||||||
|
int raw_slice_type = (int)anv_h264_rbsp_ue(&st);
|
||||||
|
|
||||||
|
VALID_RBSP_OR_BAIL(st)
|
||||||
|
|
||||||
|
int slice_type = raw_slice_type;
|
||||||
|
if (slice_type > 4) slice_type -= 5;
|
||||||
|
out->slice_type = slice_type;
|
||||||
|
|
||||||
|
int32_t curr_poc = pic_info->pStdPictureInfo->PicOrderCnt[0];
|
||||||
|
anv_h264_build_default_ref_list(slice_type, curr_poc,
|
||||||
|
ref_slot_count, ref_slots,
|
||||||
|
out->ref_list0, out->ref_list1);
|
||||||
|
|
||||||
|
anv_h264_rbsp_ue(&st);
|
||||||
|
|
||||||
|
if (sps->flags.separate_colour_plane_flag)
|
||||||
|
anv_h264_rbsp_u(&st, 2);
|
||||||
|
|
||||||
|
uint32_t log2_max_fn = sps->log2_max_frame_num_minus4 + 4;
|
||||||
|
uint32_t frame_num = anv_h264_rbsp_u(&st, log2_max_fn);
|
||||||
|
uint32_t MaxFrameNum = 1u << log2_max_fn;
|
||||||
|
|
||||||
|
bool field_pic_flag = false;
|
||||||
|
if (!sps->flags.frame_mbs_only_flag) {
|
||||||
|
field_pic_flag = !!anv_h264_rbsp_u(&st, 1);
|
||||||
|
if (field_pic_flag) anv_h264_rbsp_u(&st, 1);
|
||||||
|
}
|
||||||
|
(void)field_pic_flag;
|
||||||
|
|
||||||
|
if (is_idr) anv_h264_rbsp_ue(&st);
|
||||||
|
|
||||||
|
if (sps->pic_order_cnt_type == 0) {
|
||||||
|
anv_h264_rbsp_u(&st, sps->log2_max_pic_order_cnt_lsb_minus4 + 4);
|
||||||
|
if (pps->flags.bottom_field_pic_order_in_frame_present_flag &&
|
||||||
|
!pic_info->pStdPictureInfo->flags.field_pic_flag)
|
||||||
|
anv_h264_rbsp_se(&st);
|
||||||
|
} else if (sps->pic_order_cnt_type == 1 &&
|
||||||
|
!sps->flags.delta_pic_order_always_zero_flag) {
|
||||||
|
anv_h264_rbsp_se(&st);
|
||||||
|
if (pps->flags.bottom_field_pic_order_in_frame_present_flag &&
|
||||||
|
!pic_info->pStdPictureInfo->flags.field_pic_flag)
|
||||||
|
anv_h264_rbsp_se(&st);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pps->flags.redundant_pic_cnt_present_flag)
|
||||||
|
anv_h264_rbsp_ue(&st);
|
||||||
|
|
||||||
|
if (slice_type == ANV_H264_SLICE_B)
|
||||||
|
out->direct_spatial_mv_pred_flag = (int)anv_h264_rbsp_u(&st, 1);
|
||||||
|
|
||||||
|
out->num_ref_idx_l0_active_minus1 = pps->num_ref_idx_l0_default_active_minus1;
|
||||||
|
out->num_ref_idx_l1_active_minus1 = pps->num_ref_idx_l1_default_active_minus1;
|
||||||
|
if (slice_type == ANV_H264_SLICE_P || slice_type == ANV_H264_SLICE_SP ||
|
||||||
|
slice_type == ANV_H264_SLICE_B) {
|
||||||
|
if (anv_h264_rbsp_u(&st, 1)) {
|
||||||
|
out->num_ref_idx_l0_active_minus1 = (int)anv_h264_rbsp_ue(&st);
|
||||||
|
if (slice_type == ANV_H264_SLICE_B)
|
||||||
|
out->num_ref_idx_l1_active_minus1 = (int)anv_h264_rbsp_ue(&st);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
VALID_RBSP_OR_BAIL(st)
|
||||||
|
|
||||||
|
int active_l0 = out->num_ref_idx_l0_active_minus1 + 1;
|
||||||
|
int active_l1 = out->num_ref_idx_l1_active_minus1 + 1;
|
||||||
|
if (active_l0 > ANV_H264_MAX_REF_FRAMES) active_l0 = ANV_H264_MAX_REF_FRAMES;
|
||||||
|
if (active_l1 > ANV_H264_MAX_REF_FRAMES) active_l1 = ANV_H264_MAX_REF_FRAMES;
|
||||||
|
|
||||||
|
if (slice_type != ANV_H264_SLICE_I && slice_type != ANV_H264_SLICE_SI) {
|
||||||
|
anv_h264_apply_ref_pic_list_mod(&st, ref_slot_count, ref_slots,
|
||||||
|
frame_num, MaxFrameNum,
|
||||||
|
out->ref_list0, active_l0);
|
||||||
|
if (slice_type == ANV_H264_SLICE_B)
|
||||||
|
anv_h264_apply_ref_pic_list_mod(&st, ref_slot_count, ref_slots,
|
||||||
|
frame_num, MaxFrameNum,
|
||||||
|
out->ref_list1, active_l1);
|
||||||
|
}
|
||||||
|
|
||||||
|
VALID_RBSP_OR_BAIL(st)
|
||||||
|
|
||||||
|
bool need_pwt_l0 = (pps->flags.weighted_pred_flag &&
|
||||||
|
(slice_type == ANV_H264_SLICE_P ||
|
||||||
|
slice_type == ANV_H264_SLICE_SP));
|
||||||
|
bool need_pwt_l1 = (pps->weighted_bipred_idc == 1 &&
|
||||||
|
slice_type == ANV_H264_SLICE_B);
|
||||||
|
if (need_pwt_l0 || need_pwt_l1) {
|
||||||
|
anv_h264_parse_pred_weight_table(&st, sps->chroma_format_idc,
|
||||||
|
active_l0, active_l1,
|
||||||
|
slice_type, out);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (nal_ref_idc != 0)
|
||||||
|
anv_h264_consume_dec_ref_pic_marking(&st, is_idr);
|
||||||
|
|
||||||
|
VALID_RBSP_OR_BAIL(st)
|
||||||
|
|
||||||
|
out->cabac_init_idc = 0;
|
||||||
|
if (pps->flags.entropy_coding_mode_flag &&
|
||||||
|
slice_type != ANV_H264_SLICE_I &&
|
||||||
|
slice_type != ANV_H264_SLICE_SI)
|
||||||
|
out->cabac_init_idc = (int)anv_h264_rbsp_ue(&st);
|
||||||
|
|
||||||
|
out->slice_qp_delta = anv_h264_rbsp_se(&st);
|
||||||
|
|
||||||
|
if (slice_type == ANV_H264_SLICE_SP || slice_type == ANV_H264_SLICE_SI) {
|
||||||
|
if (slice_type == ANV_H264_SLICE_SP)
|
||||||
|
anv_h264_rbsp_u(&st, 1);
|
||||||
|
anv_h264_rbsp_se(&st);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pps->flags.deblocking_filter_control_present_flag) {
|
||||||
|
out->disable_deblocking_filter_idc = (int)anv_h264_rbsp_ue(&st);
|
||||||
|
if (out->disable_deblocking_filter_idc != 1) {
|
||||||
|
out->slice_alpha_c0_offset_div2 = anv_h264_rbsp_se(&st);
|
||||||
|
out->slice_beta_offset_div2 = anv_h264_rbsp_se(&st);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
VALID_RBSP_OR_BAIL(st)
|
||||||
|
|
||||||
|
if (pps->flags.entropy_coding_mode_flag) {
|
||||||
|
while ((st.bits_consumed & 7) && !st.error)
|
||||||
|
anv_h264_rbsp_consume_bit(&st);
|
||||||
|
}
|
||||||
|
out->first_mb_byte_offset = (uint32_t)(st.bits_consumed >> 3);
|
||||||
|
out->first_mb_bit_offset = (uint32_t)(st.bits_consumed & 7);
|
||||||
|
|
||||||
|
return !st.error;
|
||||||
|
}
|
||||||
@@ -0,0 +1,99 @@
|
|||||||
|
/*
|
||||||
|
* Copyright © 2026 [email protected]
|
||||||
|
*
|
||||||
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
||||||
|
* copy of this software and associated documentation files (the "Software"),
|
||||||
|
* to deal in the Software without restriction, including without limitation
|
||||||
|
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
||||||
|
* and/or sell copies of the Software, and to permit persons to whom the
|
||||||
|
* Software is furnished to do so, subject to the following conditions:
|
||||||
|
*
|
||||||
|
* The above copyright notice and this permission notice (including the next
|
||||||
|
* paragraph) shall be included in all copies or substantial portions of the
|
||||||
|
* Software.
|
||||||
|
*
|
||||||
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||||||
|
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||||
|
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
||||||
|
* IN THE SOFTWARE.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef ANV_VIDEO_H264_SLICE_H
|
||||||
|
#define ANV_VIDEO_H264_SLICE_H
|
||||||
|
|
||||||
|
#include "vulkan/vulkan.h"
|
||||||
|
#include "vulkan/util/vk_util.h"
|
||||||
|
|
||||||
|
#include "vk_video/vulkan_video_codec_h264std.h"
|
||||||
|
#include "vk_video/vulkan_video_codec_h264std_decode.h"
|
||||||
|
|
||||||
|
#define ANV_H264_MAX_REF_FRAMES 16
|
||||||
|
#define ANV_H264_SLICE_P 0
|
||||||
|
#define ANV_H264_SLICE_B 1
|
||||||
|
#define ANV_H264_SLICE_I 2
|
||||||
|
#define ANV_H264_SLICE_SP 3
|
||||||
|
#define ANV_H264_SLICE_SI 4
|
||||||
|
#define ANV_H264_INVALID_SLOT 0xFF
|
||||||
|
|
||||||
|
struct anv_h264_slice_params {
|
||||||
|
int first_mb_in_slice;
|
||||||
|
int slice_type;
|
||||||
|
int num_ref_idx_l0_active_minus1;
|
||||||
|
int num_ref_idx_l1_active_minus1;
|
||||||
|
int direct_spatial_mv_pred_flag;
|
||||||
|
int cabac_init_idc;
|
||||||
|
int slice_qp_delta;
|
||||||
|
int disable_deblocking_filter_idc;
|
||||||
|
int slice_alpha_c0_offset_div2;
|
||||||
|
int slice_beta_offset_div2;
|
||||||
|
int luma_log2_weight_denom;
|
||||||
|
int chroma_log2_weight_denom;
|
||||||
|
|
||||||
|
uint8_t ref_list0[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
uint8_t ref_list1[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
|
||||||
|
bool has_weight_offsets_l0;
|
||||||
|
bool has_weight_offsets_l1;
|
||||||
|
|
||||||
|
int16_t luma_weight_l0[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int16_t luma_offset_l0[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int16_t chroma_weight_l0[ANV_H264_MAX_REF_FRAMES][2];
|
||||||
|
int16_t chroma_offset_l0[ANV_H264_MAX_REF_FRAMES][2];
|
||||||
|
|
||||||
|
int16_t luma_weight_l1[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int16_t luma_offset_l1[ANV_H264_MAX_REF_FRAMES];
|
||||||
|
int16_t chroma_weight_l1[ANV_H264_MAX_REF_FRAMES][2];
|
||||||
|
int16_t chroma_offset_l1[ANV_H264_MAX_REF_FRAMES][2];
|
||||||
|
|
||||||
|
uint32_t first_mb_byte_offset;
|
||||||
|
uint32_t first_mb_bit_offset;
|
||||||
|
|
||||||
|
bool parse_ok;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct anv_h264_rbsp {
|
||||||
|
const uint8_t *buf;
|
||||||
|
const uint8_t *cur;
|
||||||
|
const uint8_t *end;
|
||||||
|
int bit;
|
||||||
|
int zeros;
|
||||||
|
bool error;
|
||||||
|
int bits_consumed;
|
||||||
|
};
|
||||||
|
|
||||||
|
bool
|
||||||
|
anv_h264_parse_slice_header(
|
||||||
|
const uint8_t *buf,
|
||||||
|
uint32_t slice_nalu_offset,
|
||||||
|
uint32_t slice_nalu_size,
|
||||||
|
const StdVideoH264SequenceParameterSet *sps,
|
||||||
|
const StdVideoH264PictureParameterSet *pps,
|
||||||
|
const VkVideoDecodeH264PictureInfoKHR *pic_info,
|
||||||
|
uint32_t ref_slot_count,
|
||||||
|
const VkVideoReferenceSlotInfoKHR *ref_slots,
|
||||||
|
struct anv_h264_slice_params *out);
|
||||||
|
|
||||||
|
#endif /* ANV_VIDEO_H264_SLICE_H */
|
||||||
@@ -116,6 +116,7 @@ libanv_files = files(
|
|||||||
'anv_util.c',
|
'anv_util.c',
|
||||||
'anv_utrace.c',
|
'anv_utrace.c',
|
||||||
'anv_video.c',
|
'anv_video.c',
|
||||||
|
'anv_video_h264_slice.c',
|
||||||
'anv_wsi.c',
|
'anv_wsi.c',
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user