mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2026-05-21 15:28:18 +02:00
c8e39683da
Vulkan video has parameter overrides, where the driver can override the driver provided parameters with an encoded bitstream where it has made changes. This is the support code to encode the bitstream headers for h265 from parameters (vps/sps/pps). Acked-by: Hyunjun Ko <zzoon@igalia.com> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/25874>
294 lines
10 KiB
C
294 lines
10 KiB
C
/*
|
|
* Copyright © 2021 Red Hat
|
|
*
|
|
* 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 VK_VIDEO_H
|
|
#define VK_VIDEO_H
|
|
|
|
#include "vk_object.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
struct vk_video_session {
|
|
struct vk_object_base base;
|
|
VkVideoCodecOperationFlagsKHR op;
|
|
VkExtent2D max_coded;
|
|
VkFormat picture_format;
|
|
VkFormat ref_format;
|
|
uint32_t max_dpb_slots;
|
|
uint32_t max_active_ref_pics;
|
|
|
|
#ifdef VK_ENABLE_BETA_EXTENSIONS
|
|
struct {
|
|
VkVideoEncodeUsageFlagsKHR video_usage_hints;
|
|
VkVideoEncodeContentFlagsKHR video_content_hints;
|
|
VkVideoEncodeTuningModeKHR tuning_mode;
|
|
} enc_usage;
|
|
#endif
|
|
union {
|
|
struct {
|
|
StdVideoH264ProfileIdc profile_idc;
|
|
} h264;
|
|
struct {
|
|
StdVideoH265ProfileIdc profile_idc;
|
|
} h265;
|
|
};
|
|
};
|
|
|
|
struct vk_video_session_parameters {
|
|
struct vk_object_base base;
|
|
VkVideoCodecOperationFlagsKHR op;
|
|
union {
|
|
struct {
|
|
uint32_t max_std_sps_count;
|
|
uint32_t max_std_pps_count;
|
|
|
|
uint32_t std_sps_count;
|
|
StdVideoH264SequenceParameterSet *std_sps;
|
|
uint32_t std_pps_count;
|
|
StdVideoH264PictureParameterSet *std_pps;
|
|
} h264_dec;
|
|
|
|
struct {
|
|
uint32_t max_std_vps_count;
|
|
uint32_t max_std_sps_count;
|
|
uint32_t max_std_pps_count;
|
|
|
|
uint32_t std_vps_count;
|
|
StdVideoH265VideoParameterSet *std_vps;
|
|
uint32_t std_sps_count;
|
|
StdVideoH265SequenceParameterSet *std_sps;
|
|
uint32_t std_pps_count;
|
|
StdVideoH265PictureParameterSet *std_pps;
|
|
} h265_dec;
|
|
|
|
#ifdef VK_ENABLE_BETA_EXTENSIONS
|
|
struct {
|
|
uint32_t max_std_sps_count;
|
|
uint32_t max_std_pps_count;
|
|
|
|
uint32_t std_sps_count;
|
|
StdVideoH264SequenceParameterSet *std_sps;
|
|
uint32_t std_pps_count;
|
|
StdVideoH264PictureParameterSet *std_pps;
|
|
} h264_enc;
|
|
|
|
struct {
|
|
uint32_t max_std_vps_count;
|
|
uint32_t max_std_sps_count;
|
|
uint32_t max_std_pps_count;
|
|
|
|
uint32_t std_vps_count;
|
|
StdVideoH265VideoParameterSet *std_vps;
|
|
uint32_t std_sps_count;
|
|
StdVideoH265SequenceParameterSet *std_sps;
|
|
uint32_t std_pps_count;
|
|
StdVideoH265PictureParameterSet *std_pps;
|
|
} h265_enc;
|
|
#endif
|
|
};
|
|
};
|
|
|
|
VkResult vk_video_session_init(struct vk_device *device,
|
|
struct vk_video_session *vid,
|
|
const VkVideoSessionCreateInfoKHR *create_info);
|
|
|
|
VkResult vk_video_session_parameters_init(struct vk_device *device,
|
|
struct vk_video_session_parameters *params,
|
|
const struct vk_video_session *vid,
|
|
const struct vk_video_session_parameters *templ,
|
|
const VkVideoSessionParametersCreateInfoKHR *create_info);
|
|
|
|
VkResult vk_video_session_parameters_update(struct vk_video_session_parameters *params,
|
|
const VkVideoSessionParametersUpdateInfoKHR *update);
|
|
|
|
void vk_video_session_parameters_finish(struct vk_device *device,
|
|
struct vk_video_session_parameters *params);
|
|
|
|
void vk_video_derive_h264_scaling_list(const StdVideoH264SequenceParameterSet *sps,
|
|
const StdVideoH264PictureParameterSet *pps,
|
|
StdVideoH264ScalingLists *list);
|
|
|
|
const StdVideoH264SequenceParameterSet *
|
|
vk_video_find_h264_dec_std_sps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
const StdVideoH264PictureParameterSet *
|
|
vk_video_find_h264_dec_std_pps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
const StdVideoH265VideoParameterSet *
|
|
vk_video_find_h265_dec_std_vps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
const StdVideoH265SequenceParameterSet *
|
|
vk_video_find_h265_dec_std_sps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
const StdVideoH265PictureParameterSet *
|
|
vk_video_find_h265_dec_std_pps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
|
|
struct vk_video_h265_slice_params {
|
|
uint32_t slice_size;
|
|
|
|
uint8_t first_slice_segment_in_pic_flag;
|
|
StdVideoH265SliceType slice_type;
|
|
uint8_t dependent_slice_segment;
|
|
uint8_t temporal_mvp_enable;
|
|
uint8_t loop_filter_across_slices_enable;
|
|
int32_t pic_order_cnt_lsb;
|
|
uint8_t sao_luma_flag;
|
|
uint8_t sao_chroma_flag;
|
|
uint8_t collocated_list;
|
|
uint32_t collocated_ref_idx;
|
|
uint8_t mvd_l1_zero_flag;
|
|
|
|
uint8_t num_ref_idx_l0_active;
|
|
uint8_t num_ref_idx_l1_active;
|
|
uint8_t rpl_modification_flag[2];
|
|
uint8_t cabac_init_idc;
|
|
int8_t slice_qp_delta;
|
|
int8_t slice_cb_qp_offset;
|
|
int8_t slice_cr_qp_offset;
|
|
int8_t max_num_merge_cand;
|
|
uint32_t slice_data_bytes_offset;
|
|
uint8_t disable_deblocking_filter_idc;
|
|
int8_t tc_offset_div2;
|
|
int8_t beta_offset_div2;
|
|
uint32_t slice_segment_address;
|
|
|
|
uint8_t luma_log2_weight_denom;
|
|
uint8_t chroma_log2_weight_denom;
|
|
uint8_t luma_weight_l0_flag[16];
|
|
int16_t luma_weight_l0[16];
|
|
int16_t luma_offset_l0[16];
|
|
uint8_t chroma_weight_l0_flag[16];
|
|
int16_t chroma_weight_l0[16][2];
|
|
int16_t chroma_offset_l0[16][2];
|
|
uint8_t luma_weight_l1_flag[16];
|
|
int16_t luma_weight_l1[16];
|
|
int16_t luma_offset_l1[16];
|
|
uint8_t chroma_weight_l1_flag[16];
|
|
int16_t chroma_weight_l1[16][2];
|
|
int16_t chroma_offset_l1[16][2];
|
|
|
|
int8_t delta_luma_weight_l0[16];
|
|
int8_t delta_luma_weight_l1[16];
|
|
int8_t delta_chroma_weight_l0[16][2];
|
|
int8_t delta_chroma_weight_l1[16][2];
|
|
int16_t delta_chroma_offset_l0[16][2];
|
|
int16_t delta_chroma_offset_l1[16][2];
|
|
};
|
|
|
|
void
|
|
vk_video_parse_h265_slice_header(const struct VkVideoDecodeInfoKHR *frame_info,
|
|
const VkVideoDecodeH265PictureInfoKHR *pic_info,
|
|
const StdVideoH265SequenceParameterSet *sps,
|
|
const StdVideoH265PictureParameterSet *pps,
|
|
void *slice_data,
|
|
uint32_t slice_size,
|
|
struct vk_video_h265_slice_params *params);
|
|
|
|
|
|
struct vk_video_h265_reference {
|
|
const VkVideoPictureResourceInfoKHR *pPictureResource;
|
|
StdVideoDecodeH265ReferenceInfoFlags flags;
|
|
uint32_t slot_index;
|
|
int32_t pic_order_cnt;
|
|
};
|
|
|
|
int vk_video_h265_poc_by_slot(const struct VkVideoDecodeInfoKHR *frame_info, int slot);
|
|
|
|
void vk_fill_video_h265_reference_info(const VkVideoDecodeInfoKHR *frame_info,
|
|
const struct VkVideoDecodeH265PictureInfoKHR *pic,
|
|
const struct vk_video_h265_slice_params *slice_params,
|
|
struct vk_video_h265_reference ref_slots[][8]);
|
|
|
|
#define VK_VIDEO_H264_MACROBLOCK_WIDTH 16
|
|
#define VK_VIDEO_H264_MACROBLOCK_HEIGHT 16
|
|
|
|
#define VK_VIDEO_H265_CTU_MAX_WIDTH 64
|
|
#define VK_VIDEO_H265_CTU_MAX_HEIGHT 64
|
|
|
|
void
|
|
vk_video_get_profile_alignments(const VkVideoProfileListInfoKHR *profile_list,
|
|
uint32_t *width_align_out, uint32_t *height_align_out);
|
|
|
|
uint8_t
|
|
vk_video_get_h264_level(StdVideoH264LevelIdc level);
|
|
|
|
#ifdef VK_ENABLE_BETA_EXTENSIONS
|
|
const StdVideoH264SequenceParameterSet *
|
|
vk_video_find_h264_enc_std_sps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
const StdVideoH264PictureParameterSet *
|
|
vk_video_find_h264_enc_std_pps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
|
|
const StdVideoH265VideoParameterSet *
|
|
vk_video_find_h265_enc_std_vps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
const StdVideoH265SequenceParameterSet *
|
|
vk_video_find_h265_enc_std_sps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
const StdVideoH265PictureParameterSet *
|
|
vk_video_find_h265_enc_std_pps(const struct vk_video_session_parameters *params,
|
|
uint32_t id);
|
|
|
|
void
|
|
vk_video_encode_h264_sps(StdVideoH264SequenceParameterSet *sps,
|
|
size_t size_limit,
|
|
size_t *data_size_ptr,
|
|
void *data_ptr);
|
|
|
|
void
|
|
vk_video_encode_h264_pps(StdVideoH264PictureParameterSet *pps,
|
|
bool high_profile,
|
|
size_t size_limit,
|
|
size_t *data_size_ptr,
|
|
void *data_ptr);
|
|
|
|
unsigned
|
|
vk_video_get_h265_nal_unit(StdVideoH265PictureType pic_type, bool irap_pic_flag);
|
|
|
|
void
|
|
vk_video_encode_h265_vps(StdVideoH265VideoParameterSet *vps,
|
|
size_t size_limit,
|
|
size_t *data_size,
|
|
void *data_ptr);
|
|
void
|
|
vk_video_encode_h265_sps(StdVideoH265SequenceParameterSet *sps,
|
|
size_t size_limit,
|
|
size_t* pDataSize,
|
|
void* pData);
|
|
|
|
void
|
|
vk_video_encode_h265_pps(StdVideoH265PictureParameterSet *pps,
|
|
size_t size_limit,
|
|
size_t *data_size,
|
|
void *data_ptr);
|
|
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|