libkms++/framebuffer.cpp

   1
   2 #include <cstring>
   3 #include <stdexcept>
   4 #include <sys/mman.h>
   5 #include <xf86drm.h>
   6 #include <xf86drmMode.h>
   7 #include <drm_fourcc.h>
   8 #include <drm.h>
   9 #include <drm_mode.h>
  10
  11 #include "kms++.h"
  12 #include "utils/testpat.h"
  13
  14 #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
  15
  16 namespace kms
  17 {
  18
  19 Framebuffer::Framebuffer(Card &card, uint32_t width, uint32_t height, const char* fourcc)
  20         :DrmObject(card, DRM_MODE_OBJECT_FB)
  21 {
  22         uint32_t a, b, c, d;
  23         a = fourcc[0];
  24         b = fourcc[1];
  25         c = fourcc[2];
  26         d = fourcc[3];
  27
  28         uint32_t code = ((uint32_t)(a) | ((uint32_t)(b) << 8) | ((uint32_t)(c) << 16) | ((uint32_t)(d) << 24));
  29
  30         Create(width, height, code);
  31 }
  32
  33 Framebuffer::~Framebuffer()
  34 {
  35         Destroy();
  36 }
  37
  38 void Framebuffer::print_short() const
  39 {
  40         printf("Framebuffer %d\n", id());
  41 }
  42
  43 struct FormatPlaneInfo
  44 {
  45         uint8_t bitspp; /* bits per (macro) pixel */
  46         uint8_t xsub;
  47         uint8_t ysub;
  48 };
  49
  50 struct FormatInfo
  51 {
  52         uint32_t format;
  53         uint8_t num_planes;
  54         struct FormatPlaneInfo planes[4];
  55 };
  56
  57 static const FormatInfo format_info_array[] = {
  58         /* YUV packed */
  59         { DRM_FORMAT_UYVY, 1, { { 32, 2, 1 } }, },
  60         { DRM_FORMAT_YUYV, 1, { { 32, 2, 1 } }, },
  61         /* YUV semi-planar */
  62         { DRM_FORMAT_NV12, 2, { { 8, 1, 1, }, { 16, 2, 2 } }, },
  63         /* RGB16 */
  64         { DRM_FORMAT_RGB565, 1, { { 16, 1, 1 } }, },
  65         /* RGB32 */
  66         { DRM_FORMAT_XRGB8888, 1, { { 32, 1, 1 } }, },
  67 };
  68
  69 static const FormatInfo& find_format(uint32_t format)
  70 {
  71         for (uint i = 0; i < ARRAY_SIZE(format_info_array); ++i) {
  72                 if (format == format_info_array[i].format)
  73                         return format_info_array[i];
  74         }
  75
  76         throw std::invalid_argument("foo");
  77 }
  78
  79 void Framebuffer::Create(uint32_t width, uint32_t height, uint32_t format)
  80 {
  81         int r;
  82
  83         m_width = width;
  84         m_height = height;
  85         m_format = format;
  86
  87         const FormatInfo& format_info = find_format(format);
  88
  89         m_num_planes = format_info.num_planes;
  90
  91         for (int i = 0; i < format_info.num_planes; ++i) {
  92                 const FormatPlaneInfo& pi = format_info.planes[i];
  93                 FramebufferPlane& plane = m_planes[i];
  94
  95                 /* create dumb buffer */
  96                 struct drm_mode_create_dumb creq = drm_mode_create_dumb();
  97                 creq.width = m_width / pi.xsub;
  98                 creq.height = m_height / pi.ysub;
  99                 creq.bpp = pi.bitspp;
 100                 r = drmIoctl(card().fd(), DRM_IOCTL_MODE_CREATE_DUMB, &creq);
 101                 if (r)
 102                         throw std::invalid_argument("foo");
 103
 104                 plane.handle = creq.handle;
 105                 plane.stride = creq.pitch;
 106                 plane.size = creq.height * creq.pitch;
 107
 108                 /*
 109                 printf("buf %d: %dx%d, bitspp %d, stride %d, size %d\n",
 110                         i, creq.width, creq.height, pi->bitspp, plane->stride, plane->size);
 111                 */
 112
 113                 /* prepare buffer for memory mapping */
 114                 struct drm_mode_map_dumb mreq = drm_mode_map_dumb();
 115                 mreq.handle = plane.handle;
 116                 r = drmIoctl(card().fd(), DRM_IOCTL_MODE_MAP_DUMB, &mreq);
 117                 if (r)
 118                         throw std::invalid_argument("foo");
 119
 120                 /* perform actual memory mapping */
 121                 m_planes[i].map = (uint8_t *)mmap(0, plane.size, PROT_READ | PROT_WRITE, MAP_SHARED,
 122                                                   card().fd(), mreq.offset);
 123                 if (plane.map == MAP_FAILED)
 124                         throw std::invalid_argument("foo");
 125
 126                 /* clear the framebuffer to 0 */
 127                 memset(plane.map, 0, plane.size);
 128         }
 129
 130         /* create framebuffer object for the dumb-buffer */
 131         uint32_t bo_handles[4] = { m_planes[0].handle, m_planes[1].handle };
 132         uint32_t pitches[4] = { m_planes[0].stride, m_planes[1].stride };
 133         uint32_t offsets[4] = { 0 };
 134         uint32_t id;
 135         r = drmModeAddFB2(card().fd(), m_width, m_height, format,
 136                           bo_handles, pitches, offsets, &id, 0);
 137         if (r)
 138                 throw std::invalid_argument("foo");
 139
 140         m_id = id;
 141 }
 142
 143 void Framebuffer::Destroy()
 144 {
 145         /* delete framebuffer */
 146         drmModeRmFB(card().fd(), id());
 147
 148         for (uint i = 0; i < m_num_planes; ++i) {
 149                 FramebufferPlane& plane = m_planes[i];
 150
 151                 /* unmap buffer */
 152                 munmap(plane.map, plane.size);
 153
 154                 /* delete dumb buffer */
 155                 struct drm_mode_destroy_dumb dreq = drm_mode_destroy_dumb();
 156                 dreq.handle = plane.handle;
 157                 drmIoctl(card().fd(), DRM_IOCTL_MODE_DESTROY_DUMB, &dreq);
 158
 159         }
 160 }
 161
 162 void Framebuffer::clear()
 163 {
 164         for (unsigned i = 0; i < m_num_planes; ++i)
 165                 memset(m_planes[i].map, 0, m_planes[i].size);
 166 }
 167 }