src/gallium/drivers/radeon/radeon_vce.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2013 Advanced Micro Devices, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28 /*
  29  * Authors:
  30  *      Christian König <christian.koenig@amd.com>
  31  *
  32  */
  33
  34 #include <stdio.h>
  35
  36 #include "pipe/p_video_codec.h"
  37
  38 #include "util/u_video.h"
  39 #include "util/u_memory.h"
  40
  41 #include "vl/vl_video_buffer.h"
  42
  43 #include "../../winsys/radeon/drm/radeon_winsys.h"
  44 #include "r600_pipe_common.h"
  45 #include "radeon_video.h"
  46 #include "radeon_vce.h"
  47
  48 #define CPB_SIZE (40 * 1024 * 1024)
  49
  50 /**
  51  * flush commands to the hardware
  52  */
  53 static void flush(struct rvce_encoder *enc)
  54 {
  55         enc->ws->cs_flush(enc->cs, RADEON_FLUSH_ASYNC, 0);
  56 }
  57
  58 #if 0
  59 static void dump_feedback(struct rvce_encoder *enc, struct rvid_buffer *fb)
  60 {
  61         uint32_t *ptr = enc->ws->buffer_map(fb->cs_handle, enc->cs, PIPE_TRANSFER_READ_WRITE);
  62         unsigned i = 0;
  63         fprintf(stderr, "\n");
  64         fprintf(stderr, "encStatus:\t\t\t%08x\n", ptr[i++]);
  65         fprintf(stderr, "encHasBitstream:\t\t%08x\n", ptr[i++]);
  66         fprintf(stderr, "encHasAudioBitstream:\t\t%08x\n", ptr[i++]);
  67         fprintf(stderr, "encBitstreamOffset:\t\t%08x\n", ptr[i++]);
  68         fprintf(stderr, "encBitstreamSize:\t\t%08x\n", ptr[i++]);
  69         fprintf(stderr, "encAudioBitstreamOffset:\t%08x\n", ptr[i++]);
  70         fprintf(stderr, "encAudioBitstreamSize:\t\t%08x\n", ptr[i++]);
  71         fprintf(stderr, "encExtrabytes:\t\t\t%08x\n", ptr[i++]);
  72         fprintf(stderr, "encAudioExtrabytes:\t\t%08x\n", ptr[i++]);
  73         fprintf(stderr, "videoTimeStamp:\t\t\t%08x\n", ptr[i++]);
  74         fprintf(stderr, "audioTimeStamp:\t\t\t%08x\n", ptr[i++]);
  75         fprintf(stderr, "videoOutputType:\t\t%08x\n", ptr[i++]);
  76         fprintf(stderr, "attributeFlags:\t\t\t%08x\n", ptr[i++]);
  77         fprintf(stderr, "seiPrivatePackageOffset:\t%08x\n", ptr[i++]);
  78         fprintf(stderr, "seiPrivatePackageSize:\t\t%08x\n", ptr[i++]);
  79         fprintf(stderr, "\n");
  80         enc->ws->buffer_unmap(fb->cs_handle);
  81 }
  82 #endif
  83
  84 /**
  85  * destroy this video encoder
  86  */
  87 static void rvce_destroy(struct pipe_video_codec *encoder)
  88 {
  89         struct rvce_encoder *enc = (struct rvce_encoder*)encoder;
  90         if (enc->stream_handle) {
  91                 struct rvid_buffer fb;
  92                 rvid_create_buffer(enc->ws, &fb, 512, RADEON_DOMAIN_GTT);
  93                 enc->fb = &fb;
  94                 enc->session(enc);
  95                 enc->feedback(enc);
  96                 enc->destroy(enc);
  97                 flush(enc);
  98                 rvid_destroy_buffer(&fb);
  99         }
 100         rvid_destroy_buffer(&enc->cpb);
 101         FREE(enc);
 102 }
 103
 104 static void rvce_begin_frame(struct pipe_video_codec *encoder,
 105                              struct pipe_video_buffer *source,
 106                              struct pipe_picture_desc *picture)
 107 {
 108         struct rvce_encoder *enc = (struct rvce_encoder*)encoder;
 109         struct vl_video_buffer *vid_buf = (struct vl_video_buffer *)source;
 110         struct pipe_h264_enc_picture_desc *pic = (struct pipe_h264_enc_picture_desc *)picture;
 111
 112         bool need_rate_control =
 113                 enc->pic.rate_ctrl.rate_ctrl_method != pic->rate_ctrl.rate_ctrl_method ||
 114                 enc->pic.quant_i_frames != pic->quant_i_frames ||
 115                 enc->pic.quant_p_frames != pic->quant_p_frames ||
 116                 enc->pic.quant_b_frames != pic->quant_b_frames;
 117
 118         enc->pic = *pic;
 119
 120         enc->get_buffer(vid_buf->resources[0], &enc->handle, &enc->luma);
 121         enc->get_buffer(vid_buf->resources[1], NULL, &enc->chroma);
 122
 123         if (!enc->stream_handle) {
 124                 struct rvid_buffer fb;
 125                 enc->stream_handle = rvid_alloc_stream_handle();
 126                 rvid_create_buffer(enc->ws, &fb, 512, RADEON_DOMAIN_GTT);
 127                 enc->fb = &fb;
 128                 enc->session(enc);
 129                 enc->create(enc);
 130                 enc->rate_control(enc);
 131                 need_rate_control = false;
 132                 enc->config_extension(enc);
 133                 enc->motion_estimation(enc);
 134                 enc->rdo(enc);
 135                 enc->pic_control(enc);
 136                 enc->feedback(enc);
 137                 flush(enc);
 138                 //dump_feedback(enc, &fb);
 139                 rvid_destroy_buffer(&fb);
 140         }
 141
 142         enc->session(enc);
 143
 144         if (need_rate_control)
 145                 enc->rate_control(enc);
 146 }
 147
 148 static void rvce_encode_bitstream(struct pipe_video_codec *encoder,
 149                                   struct pipe_video_buffer *source,
 150                                   struct pipe_resource *destination,
 151                                   void **fb)
 152 {
 153         struct rvce_encoder *enc = (struct rvce_encoder*)encoder;
 154         enc->get_buffer(destination, &enc->bs_handle, NULL);
 155         enc->bs_size = destination->width0;
 156
 157         *fb = enc->fb = CALLOC_STRUCT(rvid_buffer);
 158         if (!rvid_create_buffer(enc->ws, enc->fb, 512, RADEON_DOMAIN_GTT)) {
 159                 RVID_ERR("Can't create feedback buffer.\n");
 160                 return;
 161         }
 162         enc->encode(enc);
 163         enc->feedback(enc);
 164 }
 165
 166 static void rvce_end_frame(struct pipe_video_codec *encoder,
 167                            struct pipe_video_buffer *source,
 168                            struct pipe_picture_desc *picture)
 169 {
 170         struct rvce_encoder *enc = (struct rvce_encoder*)encoder;
 171         flush(enc);
 172 }
 173
 174 static void rvce_get_feedback(struct pipe_video_codec *encoder,
 175                               void *feedback, unsigned *size)
 176 {
 177         struct rvce_encoder *enc = (struct rvce_encoder*)encoder;
 178         struct rvid_buffer *fb = feedback;
 179
 180         if (size) {
 181                 uint32_t *ptr = enc->ws->buffer_map(fb->cs_handle, enc->cs, PIPE_TRANSFER_READ_WRITE);
 182
 183                 if (ptr[1]) {
 184                         *size = ptr[4] - ptr[9];
 185                 } else {
 186                         *size = 0;
 187                 }
 188
 189                 enc->ws->buffer_unmap(fb->cs_handle);
 190         }
 191         //dump_feedback(enc, fb);
 192         rvid_destroy_buffer(fb);
 193         FREE(fb);
 194 }
 195
 196 /**
 197  * flush any outstanding command buffers to the hardware
 198  */
 199 static void rvce_flush(struct pipe_video_codec *encoder)
 200 {
 201 }
 202
 203 static void rvce_cs_flush(void *ctx, unsigned flags)
 204 {
 205         // just ignored
 206 }
 207
 208 struct pipe_video_codec *rvce_create_encoder(struct pipe_context *context,
 209                                              const struct pipe_video_codec *templ,
 210                                              struct radeon_winsys* ws,
 211                                              rvce_get_buffer get_buffer)
 212 {
 213         struct r600_common_screen *rscreen = (struct r600_common_screen *)context->screen;
 214         struct rvce_encoder *enc;
 215
 216         if (!rscreen->info.vce_fw_version) {
 217                 RVID_ERR("Kernel doesn't supports VCE!\n");
 218                 return NULL;
 219
 220         } else if (!rvce_is_fw_version_supported(rscreen)) {
 221                 RVID_ERR("Unsupported VCE fw version loaded!\n");
 222                 return NULL;
 223         }
 224
 225         enc = CALLOC_STRUCT(rvce_encoder);
 226         if (!enc)
 227                 return NULL;
 228
 229         enc->base = *templ;
 230         enc->base.context = context;
 231
 232         enc->base.destroy = rvce_destroy;
 233         enc->base.begin_frame = rvce_begin_frame;
 234         enc->base.encode_bitstream = rvce_encode_bitstream;
 235         enc->base.end_frame = rvce_end_frame;
 236         enc->base.flush = rvce_flush;
 237         enc->base.get_feedback = rvce_get_feedback;
 238         enc->get_buffer = get_buffer;
 239
 240         enc->ws = ws;
 241         enc->cs = ws->cs_create(ws, RING_VCE, NULL);
 242         if (!enc->cs) {
 243                 RVID_ERR("Can't get command submission context.\n");
 244                 goto error;
 245         }
 246
 247         enc->ws->cs_set_flush_callback(enc->cs, rvce_cs_flush, enc);
 248
 249         if (!rvid_create_buffer(enc->ws, &enc->cpb, CPB_SIZE, RADEON_DOMAIN_VRAM)) {
 250                 RVID_ERR("Can't create CPB buffer.\n");
 251                 goto error;
 252         }
 253
 254         radeon_vce_40_2_2_init(enc);
 255
 256         return &enc->base;
 257
 258 error:
 259         if (enc->cs)
 260                 enc->ws->cs_destroy(enc->cs);
 261
 262         rvid_destroy_buffer(&enc->cpb);
 263
 264         FREE(enc);
 265         return NULL;
 266 }
 267
 268 /**
 269  * check if kernel has the right fw version loaded
 270  */
 271 bool rvce_is_fw_version_supported(struct r600_common_screen *rscreen)
 272 {
 273         return rscreen->info.vce_fw_version == ((40 << 24) | (2 << 16) | (2 << 8));
 274 }