helpers.c

   1 /*
   2  * Copyright (c) 2014 The Chromium OS Authors. All rights reserved.
   3  * Use of this source code is governed by a BSD-style license that can be
   4  * found in the LICENSE file.
   5  */
   6
   7 #include <assert.h>
   8 #include <stdbool.h>
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include <string.h>
  12 #include <sys/mman.h>
  13 #include <xf86drm.h>
  14
  15 #include "drv_priv.h"
  16 #include "helpers.h"
  17 #include "util.h"
  18
  19 size_t drv_num_planes_from_format(uint32_t format)
  20 {
  21         switch (format) {
  22         case DRV_FORMAT_C8:
  23         case DRV_FORMAT_R8:
  24         case DRV_FORMAT_RG88:
  25         case DRV_FORMAT_GR88:
  26         case DRV_FORMAT_RGB332:
  27         case DRV_FORMAT_BGR233:
  28         case DRV_FORMAT_XRGB4444:
  29         case DRV_FORMAT_XBGR4444:
  30         case DRV_FORMAT_RGBX4444:
  31         case DRV_FORMAT_BGRX4444:
  32         case DRV_FORMAT_ARGB4444:
  33         case DRV_FORMAT_ABGR4444:
  34         case DRV_FORMAT_RGBA4444:
  35         case DRV_FORMAT_BGRA4444:
  36         case DRV_FORMAT_XRGB1555:
  37         case DRV_FORMAT_XBGR1555:
  38         case DRV_FORMAT_RGBX5551:
  39         case DRV_FORMAT_BGRX5551:
  40         case DRV_FORMAT_ARGB1555:
  41         case DRV_FORMAT_ABGR1555:
  42         case DRV_FORMAT_RGBA5551:
  43         case DRV_FORMAT_BGRA5551:
  44         case DRV_FORMAT_RGB565:
  45         case DRV_FORMAT_BGR565:
  46         case DRV_FORMAT_YUYV:
  47         case DRV_FORMAT_YVYU:
  48         case DRV_FORMAT_UYVY:
  49         case DRV_FORMAT_VYUY:
  50         case DRV_FORMAT_RGB888:
  51         case DRV_FORMAT_BGR888:
  52         case DRV_FORMAT_XRGB8888:
  53         case DRV_FORMAT_XBGR8888:
  54         case DRV_FORMAT_RGBX8888:
  55         case DRV_FORMAT_BGRX8888:
  56         case DRV_FORMAT_ARGB8888:
  57         case DRV_FORMAT_ABGR8888:
  58         case DRV_FORMAT_RGBA8888:
  59         case DRV_FORMAT_BGRA8888:
  60         case DRV_FORMAT_XRGB2101010:
  61         case DRV_FORMAT_XBGR2101010:
  62         case DRV_FORMAT_RGBX1010102:
  63         case DRV_FORMAT_BGRX1010102:
  64         case DRV_FORMAT_ARGB2101010:
  65         case DRV_FORMAT_ABGR2101010:
  66         case DRV_FORMAT_RGBA1010102:
  67         case DRV_FORMAT_BGRA1010102:
  68         case DRV_FORMAT_AYUV:
  69                 return 1;
  70         case DRV_FORMAT_NV12:
  71                 return 2;
  72         case DRV_FORMAT_YVU420:
  73                 return 3;
  74         }
  75
  76         fprintf(stderr, "drv: UNKNOWN FORMAT %d\n", format);
  77         return 0;
  78 }
  79
  80 int drv_bpp_from_format(uint32_t format, size_t plane)
  81 {
  82         assert(plane < drv_num_planes_from_format(format));
  83
  84         switch (format) {
  85         case DRV_FORMAT_C8:
  86         case DRV_FORMAT_R8:
  87         case DRV_FORMAT_RGB332:
  88         case DRV_FORMAT_BGR233:
  89                 return 8;
  90
  91         case DRV_FORMAT_NV12:
  92                 return (plane == 0) ? 8 : 4;
  93         case DRV_FORMAT_YVU420:
  94                 return (plane == 0) ? 8 : 2;
  95
  96         case DRV_FORMAT_RG88:
  97         case DRV_FORMAT_GR88:
  98         case DRV_FORMAT_XRGB4444:
  99         case DRV_FORMAT_XBGR4444:
 100         case DRV_FORMAT_RGBX4444:
 101         case DRV_FORMAT_BGRX4444:
 102         case DRV_FORMAT_ARGB4444:
 103         case DRV_FORMAT_ABGR4444:
 104         case DRV_FORMAT_RGBA4444:
 105         case DRV_FORMAT_BGRA4444:
 106         case DRV_FORMAT_XRGB1555:
 107         case DRV_FORMAT_XBGR1555:
 108         case DRV_FORMAT_RGBX5551:
 109         case DRV_FORMAT_BGRX5551:
 110         case DRV_FORMAT_ARGB1555:
 111         case DRV_FORMAT_ABGR1555:
 112         case DRV_FORMAT_RGBA5551:
 113         case DRV_FORMAT_BGRA5551:
 114         case DRV_FORMAT_RGB565:
 115         case DRV_FORMAT_BGR565:
 116         case DRV_FORMAT_YUYV:
 117         case DRV_FORMAT_YVYU:
 118         case DRV_FORMAT_UYVY:
 119         case DRV_FORMAT_VYUY:
 120                 return 16;
 121
 122         case DRV_FORMAT_RGB888:
 123         case DRV_FORMAT_BGR888:
 124                 return 24;
 125
 126         case DRV_FORMAT_XRGB8888:
 127         case DRV_FORMAT_XBGR8888:
 128         case DRV_FORMAT_RGBX8888:
 129         case DRV_FORMAT_BGRX8888:
 130         case DRV_FORMAT_ARGB8888:
 131         case DRV_FORMAT_ABGR8888:
 132         case DRV_FORMAT_RGBA8888:
 133         case DRV_FORMAT_BGRA8888:
 134         case DRV_FORMAT_XRGB2101010:
 135         case DRV_FORMAT_XBGR2101010:
 136         case DRV_FORMAT_RGBX1010102:
 137         case DRV_FORMAT_BGRX1010102:
 138         case DRV_FORMAT_ARGB2101010:
 139         case DRV_FORMAT_ABGR2101010:
 140         case DRV_FORMAT_RGBA1010102:
 141         case DRV_FORMAT_BGRA1010102:
 142         case DRV_FORMAT_AYUV:
 143                 return 32;
 144         }
 145
 146         fprintf(stderr, "drv: UNKNOWN FORMAT %d\n", format);
 147         return 0;
 148 }
 149
 150 /*
 151  * This function returns the stride for a given format, width and plane.
 152  */
 153 int drv_stride_from_format(uint32_t format, uint32_t width, size_t plane)
 154 {
 155         /* Get stride of the first plane */
 156         int stride = width * DIV_ROUND_UP(drv_bpp_from_format(format, 0), 8);
 157
 158         /*
 159          * Only downsample for certain multiplanar formats which are not
 160          * interleaved and have horizontal subsampling.  Only formats supported
 161          * by our drivers are listed here -- add more as needed.
 162          */
 163         if (plane != 0) {
 164                 switch (format) {
 165                 case DRV_FORMAT_YVU420:
 166                         stride = stride / 2;
 167                 }
 168         }
 169
 170         return stride;
 171 }
 172
 173 int drv_dumb_bo_create(struct bo *bo, uint32_t width, uint32_t height,
 174                        uint32_t format, uint32_t flags)
 175 {
 176         struct drm_mode_create_dumb create_dumb;
 177         int ret;
 178
 179         /* Only single-plane formats are supported */
 180         assert(drv_num_planes_from_format(format) == 1);
 181
 182         memset(&create_dumb, 0, sizeof(create_dumb));
 183         create_dumb.height = height;
 184         create_dumb.width = width;
 185         create_dumb.bpp = drv_bpp_from_format(format, 0);
 186         create_dumb.flags = 0;
 187
 188         ret = drmIoctl(bo->drv->fd, DRM_IOCTL_MODE_CREATE_DUMB, &create_dumb);
 189         if (ret) {
 190                 fprintf(stderr, "drv: DRM_IOCTL_MODE_CREATE_DUMB failed\n");
 191                 return ret;
 192         }
 193
 194         bo->handles[0].u32 = create_dumb.handle;
 195         bo->offsets[0] = 0;
 196         bo->sizes[0] = create_dumb.size;
 197         bo->strides[0] = create_dumb.pitch;
 198
 199         return 0;
 200 }
 201
 202 int drv_dumb_bo_destroy(struct bo *bo)
 203 {
 204         struct drm_mode_destroy_dumb destroy_dumb;
 205         int ret;
 206
 207         memset(&destroy_dumb, 0, sizeof(destroy_dumb));
 208         destroy_dumb.handle = bo->handles[0].u32;
 209
 210         ret = drmIoctl(bo->drv->fd, DRM_IOCTL_MODE_DESTROY_DUMB, &destroy_dumb);
 211         if (ret) {
 212                 fprintf(stderr, "drv: DRM_IOCTL_MODE_DESTROY_DUMB failed "
 213                                 "(handle=%x)\n", bo->handles[0].u32);
 214                 return ret;
 215         }
 216
 217         return 0;
 218 }
 219
 220 int drv_gem_bo_destroy(struct bo *bo)
 221 {
 222         struct drm_gem_close gem_close;
 223         int ret, error = 0;
 224         size_t plane, i;
 225
 226         for (plane = 0; plane < bo->num_planes; plane++) {
 227                 for (i = 0; i < plane; i++)
 228                         if (bo->handles[i].u32 == bo->handles[plane].u32)
 229                                 break;
 230                 /* Make sure close hasn't already been called on this handle */
 231                 if (i != plane)
 232                         continue;
 233
 234                 memset(&gem_close, 0, sizeof(gem_close));
 235                 gem_close.handle = bo->handles[plane].u32;
 236
 237                 ret = drmIoctl(bo->drv->fd, DRM_IOCTL_GEM_CLOSE, &gem_close);
 238                 if (ret) {
 239                         fprintf(stderr, "drv: DRM_IOCTL_GEM_CLOSE failed "
 240                                         "(handle=%x) error %d\n",
 241                                         bo->handles[plane].u32, ret);
 242                         error = ret;
 243                 }
 244         }
 245
 246         return error;
 247 }
 248
 249 void *drv_dumb_bo_map(struct bo *bo)
 250 {
 251         int ret;
 252         struct drm_mode_map_dumb map_dumb;
 253
 254         memset(&map_dumb, 0, sizeof(map_dumb));
 255         map_dumb.handle = bo->handles[0].u32;
 256
 257         ret = drmIoctl(bo->drv->fd, DRM_IOCTL_MODE_MAP_DUMB, &map_dumb);
 258         if (ret) {
 259                 fprintf(stderr, "drv: DRM_IOCTL_MODE_MAP_DUMB failed \n");
 260                 return MAP_FAILED;
 261         }
 262
 263         return mmap(0, bo->sizes[0], PROT_READ | PROT_WRITE, MAP_SHARED,
 264                     bo->drv->fd, map_dumb.offset);
 265 }
 266
 267 uintptr_t drv_get_reference_count(struct driver *drv, struct bo *bo,
 268                                   size_t plane)
 269 {
 270         void *count;
 271         uintptr_t num = 0;
 272
 273         if (!drmHashLookup(drv->buffer_table, bo->handles[plane].u32, &count))
 274                 num = (uintptr_t) (count);
 275
 276         return num;
 277 }
 278
 279 void drv_increment_reference_count(struct driver *drv, struct bo *bo,
 280                                    size_t plane)
 281 {
 282         uintptr_t num = drv_get_reference_count(drv, bo, plane);
 283
 284         /* If a value isn't in the table, drmHashDelete is a no-op */
 285         drmHashDelete(drv->buffer_table, bo->handles[plane].u32);
 286         drmHashInsert(drv->buffer_table, bo->handles[plane].u32,
 287                       (void *) (num + 1));
 288 }
 289
 290 void drv_decrement_reference_count(struct driver *drv, struct bo *bo,
 291                                    size_t plane)
 292 {
 293         uintptr_t num = drv_get_reference_count(drv, bo, plane);
 294
 295         drmHashDelete(drv->buffer_table, bo->handles[plane].u32);
 296
 297         if (num > 0)
 298                 drmHashInsert(drv->buffer_table, bo->handles[plane].u32,
 299                               (void *) (num - 1));
 300 }
 301
 302 uint32_t drv_num_buffers_per_bo(struct bo *bo)
 303 {
 304         uint32_t count = 0;
 305         size_t plane, p;
 306
 307         for (plane = 0; plane < bo->num_planes; plane++) {
 308                 for (p = 0; p < plane; p++) {
 309                         if (bo->handles[p].u32 == bo->handles[plane].u32)
 310                                 break;
 311                 }
 312
 313                 if (p == plane)
 314                         count++;
 315         }
 316
 317         return count;
 318 }