radeonsi/vcn: add some av1 encoding function

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>

diff --git a/src/gallium/drivers/radeonsi/radeon_vcn_enc.c b/src/gallium/drivers/radeonsi/radeon_vcn_enc.c
index 7c02431..baf5c5d 100644 (file)
--- a/src/gallium/drivers/radeonsi/radeon_vcn_enc.c
+++ b/src/gallium/drivers/radeonsi/radeon_vcn_enc.c
@@ -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;
+}

diff --git a/src/gallium/drivers/radeonsi/radeon_vcn_enc.h b/src/gallium/drivers/radeonsi/radeon_vcn_enc.h
index cb79825..c1e1847 100644 (file)
--- a/src/gallium/drivers/radeonsi/radeon_vcn_enc.h
+++ b/src/gallium/drivers/radeonsi/radeon_vcn_enc.h
@@ -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