fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-01-06 08:50:09 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

radeonsi/vcn: add some av1 encoding function

Browse source 
preparation for enabling av1 encoding in radeonsi,
adding the entropy related functioin.

Reviewed-by: Boyuan Zhang <Boyuan.Zhang@amd.com>
Signed-off-by: Ruijing Dong <ruijing.dong@amd.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/22585>
This commit is contained in:
Ruijing Dong 2023-04-18 22:24:06 -04:00 • committed by Marge Bot
parent 8b3295b9ed
commit 4697b71c4e
2 changed files with 96 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

82
src/gallium/drivers/radeonsi/radeon_vcn_enc.c
View file

@ -894,6 +894,39 @@ void radeon_enc_code_fixed_bits(struct radeon_encoder *enc, unsigned int value,
}
}
void radeon_enc_code_uvlc(struct radeon_encoder *enc, unsigned int value)
{
uint32_t num_bits = 0;
uint64_t value_plus1 = (uint64_t)value + 1;
uint32_t num_leading_zeros = 0;
while ((uint64_t)1 << num_bits <= value_plus1)
num_bits++;
num_leading_zeros = num_bits - 1;
radeon_enc_code_fixed_bits(enc, 0, num_leading_zeros);
radeon_enc_code_fixed_bits(enc, 1, 1);
radeon_enc_code_fixed_bits(enc, (uint32_t)value_plus1, num_leading_zeros);
}
void radeon_enc_code_leb128(uint8_t *buf, uint32_t value,
uint32_t num_bytes)
{
uint8_t leb128_byte = 0;
uint32_t i = 0;
do {
leb128_byte = (value & 0x7f);
value >>= 7;
if (num_bytes > 1)
leb128_byte |= 0x80;
*(buf + i) = leb128_byte;
num_bytes--;
i++;
} while((leb128_byte & 0x80));
}
void radeon_enc_reset(struct radeon_encoder *enc)
{
enc->emulation_prevention = false;
@ -955,3 +988,52 @@ void radeon_enc_code_se(struct radeon_encoder *enc, int value)
radeon_enc_code_ue(enc, v);
}
/* dummy function for re-using the same pipeline */
void radeon_enc_dummy(struct radeon_encoder *enc) {}
/* this function has to be in pair with AV1 header copy instruction type at the end */
static void radeon_enc_av1_bs_copy_end(struct radeon_encoder *enc, uint32_t bits)
{
assert(bits > 0);
/* it must be dword aligned at the end */
*enc->enc_pic.copy_start = ALIGN_TO(bits, 32) * 4 + 12;
*(enc->enc_pic.copy_start + 2) = bits;
}
/* av1 bitstream instruction type */
void radeon_enc_av1_bs_instruction_type(struct radeon_encoder *enc,
uint32_t inst,
uint32_t obu_type)
{
radeon_enc_flush_headers(enc);
if (enc->bits_output)
radeon_enc_av1_bs_copy_end(enc, enc->bits_output);
enc->enc_pic.copy_start = &enc->cs.current.buf[enc->cs.current.cdw++];
RADEON_ENC_CS(inst);
if (inst != RENCODE_HEADER_INSTRUCTION_COPY) {
*enc->enc_pic.copy_start = 8;
if (inst == RENCODE_AV1_BITSTREAM_INSTRUCTION_OBU_START) {
*enc->enc_pic.copy_start += 4;
RADEON_ENC_CS(obu_type);
}
} else
RADEON_ENC_CS(0); /* allocate a dword for number of bits */
radeon_enc_reset(enc);
}
uint32_t radeon_enc_value_bits(uint32_t value)
{
uint32_t i = 1;
while (value > 1) {
i++;
value >>= 1;
}
return i;
}

14
src/gallium/drivers/radeonsi/radeon_vcn_enc.h
View file

@ -854,6 +854,8 @@ struct radeon_encoder {
void radeon_enc_add_buffer(struct radeon_encoder *enc, struct pb_buffer *buf,
unsigned usage, enum radeon_bo_domain domain, signed offset);
void radeon_enc_dummy(struct radeon_encoder *enc);
void radeon_enc_set_emulation_prevention(struct radeon_encoder *enc, bool set);
void radeon_enc_output_one_byte(struct radeon_encoder *enc, unsigned char byte);
@ -873,6 +875,11 @@ void radeon_enc_code_ue(struct radeon_encoder *enc, unsigned int value);
void radeon_enc_code_se(struct radeon_encoder *enc, int value);
void radeon_enc_code_uvlc(struct radeon_encoder *enc, unsigned int value);
void radeon_enc_code_leb128(unsigned char *buf, unsigned int value,
unsigned int num_bytes);
void radeon_enc_1_2_init(struct radeon_encoder *enc);
void radeon_enc_2_0_init(struct radeon_encoder *enc);
@ -881,4 +888,11 @@ void radeon_enc_3_0_init(struct radeon_encoder *enc);
void radeon_enc_4_0_init(struct radeon_encoder *enc);
void radeon_enc_av1_bs_instruction_type(struct radeon_encoder *enc,
unsigned int inst, unsigned int obu_type);
unsigned char *radeon_enc_av1_header_size_offset(struct radeon_encoder *enc);
unsigned int radeon_enc_value_bits(unsigned int value);
#endif // _RADEON_VCN_ENC_H