util: remove LIST_ADDTAIL macro
[mesa.git] / src / gallium / drivers / radeon / radeon_vce.c
index 012b4f8e13e757d77fa7bce929e8748a0340d781..af46fa776a6ed37d1dfc81a20b08c8bdb84dc094 100644 (file)
  *
  **************************************************************************/
 
-/*
- * Authors:
- *      Christian König <christian.koenig@amd.com>
- *
- */
-
 #include <stdio.h>
 
 #include "pipe/p_video_codec.h"
 
 #include "vl/vl_video_buffer.h"
 
-#include "../../winsys/radeon/drm/radeon_winsys.h"
-#include "r600_pipe_common.h"
+#include "radeonsi/si_pipe.h"
 #include "radeon_video.h"
 #include "radeon_vce.h"
 
+#define FW_40_2_2 ((40 << 24) | (2 << 16) | (2 << 8))
+#define FW_50_0_1 ((50 << 24) | (0 << 16) | (1 << 8))
+#define FW_50_1_2 ((50 << 24) | (1 << 16) | (2 << 8))
+#define FW_50_10_2 ((50 << 24) | (10 << 16) | (2 << 8))
+#define FW_50_17_3 ((50 << 24) | (17 << 16) | (3 << 8))
+#define FW_52_0_3 ((52 << 24) | (0 << 16) | (3 << 8))
+#define FW_52_4_3 ((52 << 24) | (4 << 16) | (3 << 8))
+#define FW_52_8_3 ((52 << 24) | (8 << 16) | (3 << 8))
+#define FW_53 (53 << 24)
+
 /**
  * flush commands to the hardware
  */
 static void flush(struct rvce_encoder *enc)
 {
-       enc->ws->cs_flush(enc->cs, RADEON_FLUSH_ASYNC, 0);
+       enc->ws->cs_flush(enc->cs, PIPE_FLUSH_ASYNC, NULL);
+       enc->task_info_idx = 0;
+       enc->bs_idx = 0;
 }
 
 #if 0
 static void dump_feedback(struct rvce_encoder *enc, struct rvid_buffer *fb)
 {
-       uint32_t *ptr = enc->ws->buffer_map(fb->cs_handle, enc->cs, PIPE_TRANSFER_READ_WRITE);
+       uint32_t *ptr = enc->ws->buffer_map(fb->res->buf, enc->cs, PIPE_TRANSFER_READ_WRITE);
        unsigned i = 0;
        fprintf(stderr, "\n");
        fprintf(stderr, "encStatus:\t\t\t%08x\n", ptr[i++]);
@@ -75,10 +80,161 @@ static void dump_feedback(struct rvce_encoder *enc, struct rvid_buffer *fb)
        fprintf(stderr, "seiPrivatePackageOffset:\t%08x\n", ptr[i++]);
        fprintf(stderr, "seiPrivatePackageSize:\t\t%08x\n", ptr[i++]);
        fprintf(stderr, "\n");
-       enc->ws->buffer_unmap(fb->cs_handle);
+       enc->ws->buffer_unmap(fb->res->buf);
 }
 #endif
 
+/**
+ * reset the CPB handling
+ */
+static void reset_cpb(struct rvce_encoder *enc)
+{
+       unsigned i;
+
+       list_inithead(&enc->cpb_slots);
+       for (i = 0; i < enc->cpb_num; ++i) {
+               struct rvce_cpb_slot *slot = &enc->cpb_array[i];
+               slot->index = i;
+               slot->picture_type = PIPE_H264_ENC_PICTURE_TYPE_SKIP;
+               slot->frame_num = 0;
+               slot->pic_order_cnt = 0;
+               list_addtail(&slot->list, &enc->cpb_slots);
+       }
+}
+
+/**
+ * sort l0 and l1 to the top of the list
+ */
+static void sort_cpb(struct rvce_encoder *enc)
+{
+       struct rvce_cpb_slot *i, *l0 = NULL, *l1 = NULL;
+
+       LIST_FOR_EACH_ENTRY(i, &enc->cpb_slots, list) {
+               if (i->frame_num == enc->pic.ref_idx_l0)
+                       l0 = i;
+
+               if (i->frame_num == enc->pic.ref_idx_l1)
+                       l1 = i;
+
+               if (enc->pic.picture_type == PIPE_H264_ENC_PICTURE_TYPE_P && l0)
+                       break;
+
+               if (enc->pic.picture_type == PIPE_H264_ENC_PICTURE_TYPE_B &&
+                   l0 && l1)
+                       break;
+       }
+
+       if (l1) {
+               LIST_DEL(&l1->list);
+               LIST_ADD(&l1->list, &enc->cpb_slots);
+       }
+
+       if (l0) {
+               LIST_DEL(&l0->list);
+               LIST_ADD(&l0->list, &enc->cpb_slots);
+       }
+}
+
+/**
+ * get number of cpbs based on dpb
+ */
+static unsigned get_cpb_num(struct rvce_encoder *enc)
+{
+       unsigned w = align(enc->base.width, 16) / 16;
+       unsigned h = align(enc->base.height, 16) / 16;
+       unsigned dpb;
+
+       switch (enc->base.level) {
+       case 10:
+               dpb = 396;
+               break;
+       case 11:
+               dpb = 900;
+               break;
+       case 12:
+       case 13:
+       case 20:
+               dpb = 2376;
+               break;
+       case 21:
+               dpb = 4752;
+               break;
+       case 22:
+       case 30:
+               dpb = 8100;
+               break;
+       case 31:
+               dpb = 18000;
+               break;
+       case 32:
+               dpb = 20480;
+               break;
+       case 40:
+       case 41:
+               dpb = 32768;
+               break;
+       case 42:
+               dpb = 34816;
+               break;
+       case 50:
+               dpb = 110400;
+               break;
+       default:
+       case 51:
+       case 52:
+               dpb = 184320;
+               break;
+       }
+
+       return MIN2(dpb / (w * h), 16);
+}
+
+/**
+ * Get the slot for the currently encoded frame
+ */
+struct rvce_cpb_slot *si_current_slot(struct rvce_encoder *enc)
+{
+       return LIST_ENTRY(struct rvce_cpb_slot, enc->cpb_slots.prev, list);
+}
+
+/**
+ * Get the slot for L0
+ */
+struct rvce_cpb_slot *si_l0_slot(struct rvce_encoder *enc)
+{
+       return LIST_ENTRY(struct rvce_cpb_slot, enc->cpb_slots.next, list);
+}
+
+/**
+ * Get the slot for L1
+ */
+struct rvce_cpb_slot *si_l1_slot(struct rvce_encoder *enc)
+{
+       return LIST_ENTRY(struct rvce_cpb_slot, enc->cpb_slots.next->next, list);
+}
+
+/**
+ * Calculate the offsets into the CPB
+ */
+void si_vce_frame_offset(struct rvce_encoder *enc, struct rvce_cpb_slot *slot,
+                        signed *luma_offset, signed *chroma_offset)
+{
+       struct si_screen *sscreen = (struct si_screen *)enc->screen;
+       unsigned pitch, vpitch, fsize;
+
+       if (sscreen->info.chip_class < GFX9) {
+               pitch = align(enc->luma->u.legacy.level[0].nblk_x * enc->luma->bpe, 128);
+               vpitch = align(enc->luma->u.legacy.level[0].nblk_y, 16);
+       } else {
+               pitch = align(enc->luma->u.gfx9.surf_pitch * enc->luma->bpe, 256);
+               vpitch = align(enc->luma->u.gfx9.surf_height, 16);
+       }
+       fsize = pitch * (vpitch + vpitch / 2);
+
+       *luma_offset = slot->index * fsize;
+       *chroma_offset = *luma_offset + pitch * vpitch;
+}
+
 /**
  * destroy this video encoder
  */
@@ -87,16 +243,16 @@ static void rvce_destroy(struct pipe_video_codec *encoder)
        struct rvce_encoder *enc = (struct rvce_encoder*)encoder;
        if (enc->stream_handle) {
                struct rvid_buffer fb;
-               rvid_create_buffer(enc->ws, &fb, 512, RADEON_DOMAIN_GTT);
+               si_vid_create_buffer(enc->screen, &fb, 512, PIPE_USAGE_STAGING);
                enc->fb = &fb;
                enc->session(enc);
-               enc->feedback(enc);
                enc->destroy(enc);
                flush(enc);
-               rvid_destroy_buffer(&fb);
+               si_vid_destroy_buffer(&fb);
        }
-       rvid_destroy_buffer(&enc->cpb);
+       si_vid_destroy_buffer(&enc->cpb);
        enc->ws->cs_destroy(enc->cs);
+       FREE(enc->cpb_array);
        FREE(enc);
 }
 
@@ -115,33 +271,37 @@ static void rvce_begin_frame(struct pipe_video_codec *encoder,
                enc->pic.quant_b_frames != pic->quant_b_frames;
 
        enc->pic = *pic;
+       si_get_pic_param(enc, pic);
 
        enc->get_buffer(vid_buf->resources[0], &enc->handle, &enc->luma);
        enc->get_buffer(vid_buf->resources[1], NULL, &enc->chroma);
+
+       if (pic->picture_type == PIPE_H264_ENC_PICTURE_TYPE_IDR)
+               reset_cpb(enc);
+       else if (pic->picture_type == PIPE_H264_ENC_PICTURE_TYPE_P ||
+                pic->picture_type == PIPE_H264_ENC_PICTURE_TYPE_B)
+               sort_cpb(enc);
        
        if (!enc->stream_handle) {
                struct rvid_buffer fb;
-               enc->stream_handle = rvid_alloc_stream_handle();
-               rvid_create_buffer(enc->ws, &fb, 512, RADEON_DOMAIN_GTT);
+               enc->stream_handle = si_vid_alloc_stream_handle();
+               si_vid_create_buffer(enc->screen, &fb, 512, PIPE_USAGE_STAGING);
                enc->fb = &fb;
                enc->session(enc);
                enc->create(enc);
-               enc->rate_control(enc);
-               need_rate_control = false;
-               enc->config_extension(enc);
-               enc->motion_estimation(enc);
-               enc->rdo(enc);
-               enc->pic_control(enc);
+               enc->config(enc);
                enc->feedback(enc);
                flush(enc);
                //dump_feedback(enc, &fb);
-               rvid_destroy_buffer(&fb);
+               si_vid_destroy_buffer(&fb);
+               need_rate_control = false;
        }
 
-       enc->session(enc);
-
-       if (need_rate_control)
-               enc->rate_control(enc);
+       if (need_rate_control) {
+               enc->session(enc);
+               enc->config(enc);
+               flush(enc);
+       }
 }
 
 static void rvce_encode_bitstream(struct pipe_video_codec *encoder,
@@ -154,10 +314,12 @@ static void rvce_encode_bitstream(struct pipe_video_codec *encoder,
        enc->bs_size = destination->width0;
 
        *fb = enc->fb = CALLOC_STRUCT(rvid_buffer);
-       if (!rvid_create_buffer(enc->ws, enc->fb, 512, RADEON_DOMAIN_GTT)) {
+       if (!si_vid_create_buffer(enc->screen, enc->fb, 512, PIPE_USAGE_STAGING)) {
                RVID_ERR("Can't create feedback buffer.\n");
                return;
        }
+       if (!radeon_emitted(enc->cs, 0))
+               enc->session(enc);
        enc->encode(enc);
        enc->feedback(enc);
 }
@@ -167,7 +329,20 @@ static void rvce_end_frame(struct pipe_video_codec *encoder,
                           struct pipe_picture_desc *picture)
 {
        struct rvce_encoder *enc = (struct rvce_encoder*)encoder;
-       flush(enc);
+       struct rvce_cpb_slot *slot = LIST_ENTRY(
+               struct rvce_cpb_slot, enc->cpb_slots.prev, list);
+
+       if (!enc->dual_inst || enc->bs_idx > 1)
+               flush(enc);
+
+       /* update the CPB backtrack with the just encoded frame */
+       slot->picture_type = enc->pic.picture_type;
+       slot->frame_num = enc->pic.frame_num;
+       slot->pic_order_cnt = enc->pic.pic_order_cnt;
+       if (!enc->pic.not_referenced) {
+               LIST_DEL(&slot->list);
+               LIST_ADD(&slot->list, &enc->cpb_slots);
+       }
 }
 
 static void rvce_get_feedback(struct pipe_video_codec *encoder,
@@ -177,7 +352,9 @@ static void rvce_get_feedback(struct pipe_video_codec *encoder,
        struct rvid_buffer *fb = feedback;
 
        if (size) {
-               uint32_t *ptr = enc->ws->buffer_map(fb->cs_handle, enc->cs, PIPE_TRANSFER_READ_WRITE);
+               uint32_t *ptr = enc->ws->buffer_map(
+                       fb->res->buf, enc->cs,
+                       PIPE_TRANSFER_READ_WRITE | RADEON_TRANSFER_TEMPORARY);
 
                if (ptr[1]) {
                        *size = ptr[4] - ptr[9];
@@ -185,10 +362,10 @@ static void rvce_get_feedback(struct pipe_video_codec *encoder,
                        *size = 0;
                }
 
-               enc->ws->buffer_unmap(fb->cs_handle);
+               enc->ws->buffer_unmap(fb->res->buf);
        }
        //dump_feedback(enc, fb);
-       rvid_destroy_buffer(fb);
+       si_vid_destroy_buffer(fb);
        FREE(fb);
 }
 
@@ -197,29 +374,34 @@ static void rvce_get_feedback(struct pipe_video_codec *encoder,
  */
 static void rvce_flush(struct pipe_video_codec *encoder)
 {
+       struct rvce_encoder *enc = (struct rvce_encoder*)encoder;
+
+       flush(enc);
 }
 
-static void rvce_cs_flush(void *ctx, unsigned flags)
+static void rvce_cs_flush(void *ctx, unsigned flags,
+                         struct pipe_fence_handle **fence)
 {
        // just ignored
 }
 
-struct pipe_video_codec *rvce_create_encoder(struct pipe_context *context,
-                                            const struct pipe_video_codec *templ,
-                                            struct radeon_winsys* ws,
-                                            rvce_get_buffer get_buffer)
+struct pipe_video_codec *si_vce_create_encoder(struct pipe_context *context,
+                                              const struct pipe_video_codec *templ,
+                                              struct radeon_winsys* ws,
+                                              rvce_get_buffer get_buffer)
 {
-       struct r600_common_screen *rscreen = (struct r600_common_screen *)context->screen;
+       struct si_screen *sscreen = (struct si_screen *)context->screen;
+       struct si_context *sctx = (struct si_context*)context;
        struct rvce_encoder *enc;
        struct pipe_video_buffer *tmp_buf, templat = {};
-       struct radeon_surface *tmp_surf;
-       unsigned pitch, vpitch;
+       struct radeon_surf *tmp_surf;
+       unsigned cpb_size;
 
-       if (!rscreen->info.vce_fw_version) {
+       if (!sscreen->info.vce_fw_version) {
                RVID_ERR("Kernel doesn't supports VCE!\n");
                return NULL;
 
-       } else if (!rvce_is_fw_version_supported(rscreen)) {
+       } else if (!si_vce_is_fw_version_supported(sscreen)) {
                RVID_ERR("Unsupported VCE fw version loaded!\n");
                return NULL;
        }
@@ -228,6 +410,23 @@ struct pipe_video_codec *rvce_create_encoder(struct pipe_context *context,
        if (!enc)
                return NULL;
 
+       if (sscreen->info.is_amdgpu)
+               enc->use_vm = true;
+       if ((!sscreen->info.is_amdgpu && sscreen->info.drm_minor >= 42) ||
+            sscreen->info.is_amdgpu)
+               enc->use_vui = true;
+       if (sscreen->info.family >= CHIP_TONGA &&
+           sscreen->info.family != CHIP_STONEY &&
+           sscreen->info.family != CHIP_POLARIS11 &&
+           sscreen->info.family != CHIP_POLARIS12 &&
+           sscreen->info.family != CHIP_VEGAM)
+               enc->dual_pipe = true;
+       /* TODO enable B frame with dual instance */
+       if ((sscreen->info.family >= CHIP_TONGA) &&
+               (templ->max_references == 1) &&
+               (sscreen->info.vce_harvest_config == 0))
+               enc->dual_inst = true;
+
        enc->base = *templ;
        enc->base.context = context;
 
@@ -239,14 +438,14 @@ struct pipe_video_codec *rvce_create_encoder(struct pipe_context *context,
        enc->base.get_feedback = rvce_get_feedback;
        enc->get_buffer = get_buffer;
 
+       enc->screen = context->screen;
        enc->ws = ws;
-       enc->cs = ws->cs_create(ws, RING_VCE, NULL);
+       enc->cs = ws->cs_create(sctx->ctx, RING_VCE, rvce_cs_flush, enc, false);
        if (!enc->cs) {
                RVID_ERR("Can't get command submission context.\n");
                goto error;
        }
 
-       enc->ws->cs_set_flush_callback(enc->cs, rvce_cs_flush, enc);
        templat.buffer_format = PIPE_FORMAT_NV12;
        templat.chroma_format = PIPE_VIDEO_CHROMA_FORMAT_420;
        templat.width = enc->base.width;
@@ -257,18 +456,64 @@ struct pipe_video_codec *rvce_create_encoder(struct pipe_context *context,
                goto error;
        }
 
+       enc->cpb_num = get_cpb_num(enc);
+       if (!enc->cpb_num)
+               goto error;
+
        get_buffer(((struct vl_video_buffer *)tmp_buf)->resources[0], NULL, &tmp_surf);
-       pitch = align(tmp_surf->level[0].pitch_bytes, 128);
-       vpitch = align(tmp_surf->npix_y, 16);
+
+       cpb_size = (sscreen->info.chip_class < GFX9) ?
+               align(tmp_surf->u.legacy.level[0].nblk_x * tmp_surf->bpe, 128) *
+               align(tmp_surf->u.legacy.level[0].nblk_y, 32) :
+
+               align(tmp_surf->u.gfx9.surf_pitch * tmp_surf->bpe, 256) *
+               align(tmp_surf->u.gfx9.surf_height, 32);
+
+       cpb_size = cpb_size * 3 / 2;
+       cpb_size = cpb_size * enc->cpb_num;
+       if (enc->dual_pipe)
+               cpb_size +=  RVCE_MAX_AUX_BUFFER_NUM *
+                       RVCE_MAX_BITSTREAM_OUTPUT_ROW_SIZE * 2;
        tmp_buf->destroy(tmp_buf);
-       if (!rvid_create_buffer(enc->ws, &enc->cpb,
-                       pitch * vpitch * 1.5 * RVCE_NUM_CPB_FRAMES,
-                       RADEON_DOMAIN_VRAM)) {
+       if (!si_vid_create_buffer(enc->screen, &enc->cpb, cpb_size, PIPE_USAGE_DEFAULT)) {
                RVID_ERR("Can't create CPB buffer.\n");
                goto error;
        }
 
-       radeon_vce_40_2_2_init(enc);
+       enc->cpb_array = CALLOC(enc->cpb_num, sizeof(struct rvce_cpb_slot));
+       if (!enc->cpb_array)
+               goto error;
+
+       reset_cpb(enc);
+
+       switch (sscreen->info.vce_fw_version) {
+       case FW_40_2_2:
+               si_vce_40_2_2_init(enc);
+               si_get_pic_param = si_vce_40_2_2_get_param;
+               break;
+
+       case FW_50_0_1:
+       case FW_50_1_2:
+       case FW_50_10_2:
+       case FW_50_17_3:
+               si_vce_50_init(enc);
+               si_get_pic_param = si_vce_50_get_param;
+               break;
+
+       case FW_52_0_3:
+       case FW_52_4_3:
+       case FW_52_8_3:
+               si_vce_52_init(enc);
+               si_get_pic_param = si_vce_52_get_param;
+               break;
+
+       default:
+               if ((sscreen->info.vce_fw_version & (0xff << 24)) >= FW_53) {
+                       si_vce_52_init(enc);
+                       si_get_pic_param = si_vce_52_get_param;
+               } else
+                       goto error;
+       }
 
        return &enc->base;
 
@@ -276,8 +521,9 @@ error:
        if (enc->cs)
                enc->ws->cs_destroy(enc->cs);
 
-       rvid_destroy_buffer(&enc->cpb);
+       si_vid_destroy_buffer(&enc->cpb);
 
+       FREE(enc->cpb_array);
        FREE(enc);
        return NULL;
 }
@@ -285,7 +531,46 @@ error:
 /**
  * check if kernel has the right fw version loaded
  */
-bool rvce_is_fw_version_supported(struct r600_common_screen *rscreen)
+bool si_vce_is_fw_version_supported(struct si_screen *sscreen)
+{
+       switch (sscreen->info.vce_fw_version) {
+       case FW_40_2_2:
+       case FW_50_0_1:
+       case FW_50_1_2:
+       case FW_50_10_2:
+       case FW_50_17_3:
+       case FW_52_0_3:
+       case FW_52_4_3:
+       case FW_52_8_3:
+               return true;
+       default:
+               if ((sscreen->info.vce_fw_version & (0xff << 24)) >= FW_53)
+                       return true;
+               else
+                       return false;
+       }
+}
+
+/**
+ * Add the buffer as relocation to the current command submission
+ */
+void si_vce_add_buffer(struct rvce_encoder *enc, struct pb_buffer *buf,
+                      enum radeon_bo_usage usage, enum radeon_bo_domain domain,
+                      signed offset)
 {
-       return rscreen->info.vce_fw_version == ((40 << 24) | (2 << 16) | (2 << 8));
+       int reloc_idx;
+
+       reloc_idx = enc->ws->cs_add_buffer(enc->cs, buf, usage | RADEON_USAGE_SYNCHRONIZED,
+                                          domain, 0);
+       if (enc->use_vm) {
+               uint64_t addr;
+               addr = enc->ws->buffer_get_virtual_address(buf);
+               addr = addr + offset;
+               RVCE_CS(addr >> 32);
+               RVCE_CS(addr);
+       } else {
+               offset += enc->ws->buffer_get_reloc_offset(buf);
+               RVCE_CS(reloc_idx * 4);
+               RVCE_CS(offset);
+       }
 }