src/gallium/state_trackers/va/context.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2010 Thomas Balling Sørensen & Orasanu Lucian.
   4  * Copyright 2014 Advanced Micro Devices, Inc.
   5  * All Rights Reserved.
   6  *
   7  * Permission is hereby granted, free of charge, to any person obtaining a
   8  * copy of this software and associated documentation files (the
   9  * "Software"), to deal in the Software without restriction, including
  10  * without limitation the rights to use, copy, modify, merge, publish,
  11  * distribute, sub license, and/or sell copies of the Software, and to
  12  * permit persons to whom the Software is furnished to do so, subject to
  13  * the following conditions:
  14  *
  15  * The above copyright notice and this permission notice (including the
  16  * next paragraph) shall be included in all copies or substantial portions
  17  * of the Software.
  18  *
  19  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  20  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  21  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  22  * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR
  23  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  24  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  25  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  26  *
  27  **************************************************************************/
  28
  29 #include "pipe/p_screen.h"
  30 #include "pipe/p_video_codec.h"
  31 #include "util/u_memory.h"
  32 #include "util/u_handle_table.h"
  33 #include "util/u_video.h"
  34 #include "vl/vl_deint_filter.h"
  35 #include "vl/vl_winsys.h"
  36
  37 #include "va_private.h"
  38
  39 #include <va/va_drmcommon.h>
  40
  41 static struct VADriverVTable vtable =
  42 {
  43    &vlVaTerminate,
  44    &vlVaQueryConfigProfiles,
  45    &vlVaQueryConfigEntrypoints,
  46    &vlVaGetConfigAttributes,
  47    &vlVaCreateConfig,
  48    &vlVaDestroyConfig,
  49    &vlVaQueryConfigAttributes,
  50    &vlVaCreateSurfaces,
  51    &vlVaDestroySurfaces,
  52    &vlVaCreateContext,
  53    &vlVaDestroyContext,
  54    &vlVaCreateBuffer,
  55    &vlVaBufferSetNumElements,
  56    &vlVaMapBuffer,
  57    &vlVaUnmapBuffer,
  58    &vlVaDestroyBuffer,
  59    &vlVaBeginPicture,
  60    &vlVaRenderPicture,
  61    &vlVaEndPicture,
  62    &vlVaSyncSurface,
  63    &vlVaQuerySurfaceStatus,
  64    &vlVaQuerySurfaceError,
  65    &vlVaPutSurface,
  66    &vlVaQueryImageFormats,
  67    &vlVaCreateImage,
  68    &vlVaDeriveImage,
  69    &vlVaDestroyImage,
  70    &vlVaSetImagePalette,
  71    &vlVaGetImage,
  72    &vlVaPutImage,
  73    &vlVaQuerySubpictureFormats,
  74    &vlVaCreateSubpicture,
  75    &vlVaDestroySubpicture,
  76    &vlVaSubpictureImage,
  77    &vlVaSetSubpictureChromakey,
  78    &vlVaSetSubpictureGlobalAlpha,
  79    &vlVaAssociateSubpicture,
  80    &vlVaDeassociateSubpicture,
  81    &vlVaQueryDisplayAttributes,
  82    &vlVaGetDisplayAttributes,
  83    &vlVaSetDisplayAttributes,
  84    &vlVaBufferInfo,
  85    &vlVaLockSurface,
  86    &vlVaUnlockSurface,
  87    NULL, /* DEPRECATED VaGetSurfaceAttributes */
  88    &vlVaCreateSurfaces2,
  89    &vlVaQuerySurfaceAttributes,
  90    &vlVaAcquireBufferHandle,
  91    &vlVaReleaseBufferHandle
  92 };
  93
  94 static struct VADriverVTableVPP vtable_vpp =
  95 {
  96    1,
  97    &vlVaQueryVideoProcFilters,
  98    &vlVaQueryVideoProcFilterCaps,
  99    &vlVaQueryVideoProcPipelineCaps
 100 };
 101
 102 PUBLIC VAStatus
 103 VA_DRIVER_INIT_FUNC(VADriverContextP ctx)
 104 {
 105    vlVaDriver *drv;
 106    struct drm_state *drm_info;
 107
 108    if (!ctx)
 109       return VA_STATUS_ERROR_INVALID_CONTEXT;
 110
 111    drv = CALLOC(1, sizeof(vlVaDriver));
 112    if (!drv)
 113       return VA_STATUS_ERROR_ALLOCATION_FAILED;
 114
 115    switch (ctx->display_type) {
 116    case VA_DISPLAY_ANDROID:
 117       FREE(drv);
 118       return VA_STATUS_ERROR_UNIMPLEMENTED;
 119    case VA_DISPLAY_GLX:
 120    case VA_DISPLAY_X11:
 121 #if defined(HAVE_DRI3)
 122       drv->vscreen = vl_dri3_screen_create(ctx->native_dpy, ctx->x11_screen);
 123 #endif
 124       if (!drv->vscreen)
 125          drv->vscreen = vl_dri2_screen_create(ctx->native_dpy, ctx->x11_screen);
 126       if (!drv->vscreen)
 127          goto error_screen;
 128       break;
 129    case VA_DISPLAY_WAYLAND:
 130    case VA_DISPLAY_DRM:
 131    case VA_DISPLAY_DRM_RENDERNODES: {
 132       drm_info = (struct drm_state *) ctx->drm_state;
 133
 134       if (!drm_info || drm_info->fd < 0) {
 135          FREE(drv);
 136          return VA_STATUS_ERROR_INVALID_PARAMETER;
 137       }
 138
 139       drv->vscreen = vl_drm_screen_create(drm_info->fd);
 140       if (!drv->vscreen)
 141          goto error_screen;
 142       }
 143       break;
 144    default:
 145       FREE(drv);
 146       return VA_STATUS_ERROR_INVALID_DISPLAY;
 147    }
 148
 149    drv->pipe = drv->vscreen->pscreen->context_create(drv->vscreen->pscreen,
 150                                                      drv->vscreen, 0);
 151    if (!drv->pipe)
 152       goto error_pipe;
 153
 154    drv->htab = handle_table_create();
 155    if (!drv->htab)
 156       goto error_htab;
 157
 158    vl_compositor_init(&drv->compositor, drv->pipe);
 159    vl_compositor_init_state(&drv->cstate, drv->pipe);
 160
 161    vl_csc_get_matrix(VL_CSC_COLOR_STANDARD_BT_601, NULL, true, &drv->csc);
 162    vl_compositor_set_csc_matrix(&drv->cstate, (const vl_csc_matrix *)&drv->csc, 1.0f, 0.0f);
 163    pipe_mutex_init(drv->mutex);
 164
 165    ctx->pDriverData = (void *)drv;
 166    ctx->version_major = 0;
 167    ctx->version_minor = 1;
 168    *ctx->vtable = vtable;
 169    *ctx->vtable_vpp = vtable_vpp;
 170    ctx->max_profiles = PIPE_VIDEO_PROFILE_MPEG4_AVC_HIGH - PIPE_VIDEO_PROFILE_UNKNOWN;
 171    ctx->max_entrypoints = 1;
 172    ctx->max_attributes = 1;
 173    ctx->max_image_formats = VL_VA_MAX_IMAGE_FORMATS;
 174    ctx->max_subpic_formats = 1;
 175    ctx->max_display_attributes = 1;
 176    ctx->str_vendor = "mesa gallium vaapi";
 177
 178    return VA_STATUS_SUCCESS;
 179
 180 error_htab:
 181    drv->pipe->destroy(drv->pipe);
 182
 183 error_pipe:
 184    drv->vscreen->destroy(drv->vscreen);
 185
 186 error_screen:
 187    FREE(drv);
 188    return VA_STATUS_ERROR_ALLOCATION_FAILED;
 189 }
 190
 191 VAStatus
 192 vlVaCreateContext(VADriverContextP ctx, VAConfigID config_id, int picture_width,
 193                   int picture_height, int flag, VASurfaceID *render_targets,
 194                   int num_render_targets, VAContextID *context_id)
 195 {
 196    vlVaDriver *drv;
 197    vlVaContext *context;
 198    vlVaConfig *config;
 199    int is_vpp;
 200
 201    if (!ctx)
 202       return VA_STATUS_ERROR_INVALID_CONTEXT;
 203
 204    drv = VL_VA_DRIVER(ctx);
 205    pipe_mutex_lock(drv->mutex);
 206    config = handle_table_get(drv->htab, config_id);
 207    pipe_mutex_unlock(drv->mutex);
 208
 209    is_vpp = config->profile == PIPE_VIDEO_PROFILE_UNKNOWN && !picture_width &&
 210             !picture_height && !flag && !render_targets && !num_render_targets;
 211
 212    if (!(picture_width && picture_height) && !is_vpp)
 213       return VA_STATUS_ERROR_INVALID_IMAGE_FORMAT;
 214
 215    context = CALLOC(1, sizeof(vlVaContext));
 216    if (!context)
 217       return VA_STATUS_ERROR_ALLOCATION_FAILED;
 218
 219    if (is_vpp) {
 220       context->decoder = NULL;
 221       if (!drv->compositor.upload) {
 222          FREE(context);
 223          return VA_STATUS_ERROR_INVALID_CONTEXT;
 224       }
 225    } else {
 226       context->templat.profile = config->profile;
 227       context->templat.entrypoint = config->entrypoint;
 228       context->templat.chroma_format = PIPE_VIDEO_CHROMA_FORMAT_420;
 229       context->templat.width = picture_width;
 230       context->templat.height = picture_height;
 231       context->templat.expect_chunked_decode = true;
 232
 233       switch (u_reduce_video_profile(context->templat.profile)) {
 234       case PIPE_VIDEO_FORMAT_MPEG12:
 235       case PIPE_VIDEO_FORMAT_VC1:
 236       case PIPE_VIDEO_FORMAT_MPEG4:
 237          context->templat.max_references = 2;
 238          break;
 239
 240       case PIPE_VIDEO_FORMAT_MPEG4_AVC:
 241          context->templat.max_references = 0;
 242          if (config->entrypoint != PIPE_VIDEO_ENTRYPOINT_ENCODE) {
 243             context->desc.h264.pps = CALLOC_STRUCT(pipe_h264_pps);
 244             if (!context->desc.h264.pps) {
 245                FREE(context);
 246                return VA_STATUS_ERROR_ALLOCATION_FAILED;
 247             }
 248             context->desc.h264.pps->sps = CALLOC_STRUCT(pipe_h264_sps);
 249             if (!context->desc.h264.pps->sps) {
 250                FREE(context->desc.h264.pps);
 251                FREE(context);
 252                return VA_STATUS_ERROR_ALLOCATION_FAILED;
 253             }
 254          }
 255          break;
 256
 257      case PIPE_VIDEO_FORMAT_HEVC:
 258          context->templat.max_references = num_render_targets;
 259          context->desc.h265.pps = CALLOC_STRUCT(pipe_h265_pps);
 260          if (!context->desc.h265.pps) {
 261             FREE(context);
 262             return VA_STATUS_ERROR_ALLOCATION_FAILED;
 263          }
 264          context->desc.h265.pps->sps = CALLOC_STRUCT(pipe_h265_sps);
 265          if (!context->desc.h265.pps->sps) {
 266             FREE(context->desc.h265.pps);
 267             FREE(context);
 268             return VA_STATUS_ERROR_ALLOCATION_FAILED;
 269          }
 270          break;
 271
 272       default:
 273          break;
 274       }
 275    }
 276
 277    context->desc.base.profile = config->profile;
 278    context->desc.base.entry_point = config->entrypoint;
 279    if (config->entrypoint == PIPE_VIDEO_ENTRYPOINT_ENCODE)
 280       context->desc.h264enc.rate_ctrl.rate_ctrl_method = config->rc;
 281
 282    pipe_mutex_lock(drv->mutex);
 283    *context_id = handle_table_add(drv->htab, context);
 284    pipe_mutex_unlock(drv->mutex);
 285
 286    return VA_STATUS_SUCCESS;
 287 }
 288
 289 VAStatus
 290 vlVaDestroyContext(VADriverContextP ctx, VAContextID context_id)
 291 {
 292    vlVaDriver *drv;
 293    vlVaContext *context;
 294
 295    if (!ctx)
 296       return VA_STATUS_ERROR_INVALID_CONTEXT;
 297
 298    drv = VL_VA_DRIVER(ctx);
 299    pipe_mutex_lock(drv->mutex);
 300    context = handle_table_get(drv->htab, context_id);
 301    if (!context) {
 302       pipe_mutex_unlock(drv->mutex);
 303       return VA_STATUS_ERROR_INVALID_CONTEXT;
 304    }
 305
 306    if (context->decoder) {
 307       if (context->desc.base.entry_point != PIPE_VIDEO_ENTRYPOINT_ENCODE) {
 308          if (u_reduce_video_profile(context->decoder->profile) ==
 309                PIPE_VIDEO_FORMAT_MPEG4_AVC) {
 310             FREE(context->desc.h264.pps->sps);
 311             FREE(context->desc.h264.pps);
 312          }
 313          if (u_reduce_video_profile(context->decoder->profile) ==
 314                PIPE_VIDEO_FORMAT_HEVC) {
 315             FREE(context->desc.h265.pps->sps);
 316             FREE(context->desc.h265.pps);
 317          }
 318       }
 319       context->decoder->destroy(context->decoder);
 320    }
 321    if (context->deint) {
 322       vl_deint_filter_cleanup(context->deint);
 323       FREE(context->deint);
 324    }
 325    FREE(context);
 326    handle_table_remove(drv->htab, context_id);
 327    pipe_mutex_unlock(drv->mutex);
 328
 329    return VA_STATUS_SUCCESS;
 330 }
 331
 332 VAStatus
 333 vlVaTerminate(VADriverContextP ctx)
 334 {
 335    vlVaDriver *drv;
 336
 337    if (!ctx)
 338       return VA_STATUS_ERROR_INVALID_CONTEXT;
 339
 340    drv = ctx->pDriverData;
 341    vl_compositor_cleanup_state(&drv->cstate);
 342    vl_compositor_cleanup(&drv->compositor);
 343    drv->pipe->destroy(drv->pipe);
 344    drv->vscreen->destroy(drv->vscreen);
 345    handle_table_destroy(drv->htab);
 346    pipe_mutex_destroy(drv->mutex);
 347    FREE(drv);
 348
 349    return VA_STATUS_SUCCESS;
 350 }