drmbuffer.cpp

Go to the documentation of this file.

// SPDX-License-Identifier: AGPL-3.0-or-later
 
#include <cstdint>
#include <fcntl.h>
#include <sys/mman.h>
 
extern "C" {
#include <libavutil/frame.h>
#include <libavutil/hwcontext_drm.h>
}
 
#include <drm_fourcc.h>
#include <xf86drm.h>
#include <xf86drmMode.h>
 
#include "drmbuffer.h"
#include "logger.h"
#include "pool.h"
 
/*****************************************************************************
 * Drm Buffer
 ****************************************************************************/
 
cDrmBuffer::cDrmBuffer(void)
{
    m_dirty = false;
    m_dmaBufHandle[0] = 0;
 
    m_numPlanes = 0;
    for (int i = 0; i < 4; i++) {
        m_pPlane[i] = nullptr;
    }
}
 
cDrmBuffer::cDrmBuffer(cDrmBuffer *src)
    : m_width(src->m_width),
      m_height(src->m_height),
      m_pixFmt(src->m_pixFmt),
      m_rectOnScreen(src->GetScreenRect()),
      m_fbId(src->m_fbId),
      m_numPlanes(src->m_numPlanes),
      m_numObjects(src->m_numObjects)
{
    m_dirty = false;
 
    for (int object = 0; object < m_numObjects; object++) {
        m_dmaBufHandle[object] = src->m_dmaBufHandle[object];
    }
 
    for (int i = 0; i < src->m_numPlanes; i++) {
        m_size[i] = src->m_size[i];
        m_pitch[i] = src->m_pitch[i];
        m_planePrimeHandle[i] = src->m_planePrimeHandle[i];
        m_offset[i] = src->m_offset[i];
        m_objIdx[i] = src->m_objIdx[i];
    }
 
    void *src_buffer = NULL;
    void *dst_buffer = NULL;
 
    // planes aren't mmapped, do it (PRIME)
    if (!src->m_pPlane[0]) {
        for (int object = 0; object < m_numObjects; object++) {
            // memcpy mmapped data
            dst_buffer = malloc(src->m_size[object]);
            src_buffer = mmap(NULL, src->m_size[object], PROT_READ, MAP_SHARED, src->m_dmaBufHandle[object], 0);
            if (src_buffer == MAP_FAILED) {
                LOGERROR("drmbuffer: %s (clone): cannot map buffer size %d prime_fd %d (%d): %m",
                    __FUNCTION__, src->m_size[object], src->m_dmaBufHandle[object], errno);
                return;
            }
 
            LOGDEBUG2(L_GRAB, "drmbuffer: %s (clone): Copy %p to %p", __FUNCTION__, src_buffer, dst_buffer);
            memcpy(dst_buffer, src_buffer, src->m_size[object]);
            munmap(src_buffer, src->m_size[object]);
            for (int plane = 0; plane < m_numPlanes; plane++) {
                if (m_objIdx[plane] == object) {
                    m_pPlane[plane] = (uint8_t *)dst_buffer;
                    LOGDEBUG2(L_GRAB, "drmbuffer: %s (clone): plane[%d] gets %p (object %d)", __FUNCTION__, plane, dst_buffer, object);
                }
            }
        }
    } else {
        for (int plane = 0; plane < m_numPlanes; plane++) {
            dst_buffer = malloc(m_size[plane]);
            memcpy(dst_buffer, src->m_pPlane[plane], src->m_size[plane]);
            m_pPlane[plane] = (uint8_t *)dst_buffer;
        }
    }
 
    for (int plane = 0; plane < m_numPlanes; plane++) {
        LOGDEBUG2(L_GRAB, "drmbuffer: %s (clone): Cloned plane %d address %p pitch %d offset %d handle %d size %d",
            __FUNCTION__, plane, m_pPlane[plane], m_pitch[plane], m_offset[plane], m_planePrimeHandle[plane], m_size[plane]);
    }
}
 
#ifdef USE_GLES
cDrmBuffer::cDrmBuffer(int fdDrm, uint32_t width, uint32_t height, uint32_t pixFmt, struct gbm_bo *bo)
    : m_width(width),
      m_height(height),
      m_pixFmt(pixFmt),
      m_drmDeviceFd(fdDrm),
      m_pBo(bo)
{
    m_numPlanes = 0;
    for (int i = 0; i < 4; i++) {
        m_pPlane[i] = nullptr;
        m_planePrimeHandle[i] = 0;
        m_offset[i] = 0;
        m_pitch[i] = 0;
        m_size[i] = 0;
    }
    m_dirty = false;
}
#endif
 
void cDrmBuffer::Destroy(void)
{
    struct drm_mode_destroy_dumb dreq;
    LOGDEBUG2(L_DRM, "drmbuffer: %s: destroy FB %d DMA-BUF handle %d", __FUNCTION__, m_fbId, m_dmaBufHandle[0]);
 
    for (int i = 0; i < m_numPlanes; i++) {
        if (m_pPlane[i]) {
            if (munmap(m_pPlane[i], m_size[i]))
                LOGERROR("drmbuffer: %s: failed unmap FB (%d): %m", __FUNCTION__, errno);
        }
    }
 
    if (drmModeRmFB(m_drmDeviceFd, m_fbId) < 0)
        LOGERROR("drmbuffer: %s: cannot rm FB (%d): %m", __FUNCTION__, errno);
 
    if (m_closeHandleOnDestroy &&m_dmaBufHandle[0] && fcntl(m_dmaBufHandle[0], F_GETFD) != -1) { // the handle can be invalid in reverse trickspeed, because the decoder is rapidly reopened
        if (close(m_dmaBufHandle[0]))
            LOGERROR("drmbuffer: %s: error closing DMA-BUF handle %d (%d): %m", __FUNCTION__, m_dmaBufHandle[0], errno);
    }
 
    for (int i = 0; i < m_numPlanes; i++) {
        if (m_pPlane[i]) {
            memset(&dreq, 0, sizeof(dreq));
            dreq.handle = m_planePrimeHandle[i];
 
            if (drmIoctl(m_drmDeviceFd, DRM_IOCTL_MODE_DESTROY_DUMB, &dreq) < 0)
                LOGERROR("drmbuffer: %s: cannot destroy dumb buffer (%d): %m", __FUNCTION__, errno);
            m_planePrimeHandle[i] = 0;
 
        }
 
        m_pPlane[i] = 0;
        m_size[i] = 0;
        m_pitch[i] = 0;
        m_offset[i] = 0;
        m_objIdx[i] = 0;
    }
 
    for (int i = 0; i < m_numObjects; i++) {
        if (m_objectPrimeHandle[i] && (drmIoctl(m_drmDeviceFd, DRM_IOCTL_GEM_CLOSE, &m_objectPrimeHandle[i]) < 0))
            LOGERROR("drmbuffer: %s: cannot close handle %d FB %d GEM (%d): %m", __FUNCTION__, m_objectPrimeHandle[i], m_fbId, errno);
    }
 
    m_width = 0;
    m_height = 0;
    m_fbId = 0;
    m_dirty = false;
    m_numPlanes = 0;
    m_destroyAfterUse = false;
}
 
static const struct format_info format_info_array[] = {
    { DRM_FORMAT_NV12, "NV12", 2, { { 8, 1, 1 }, { 16, 2, 2 } }, },
    { DRM_FORMAT_YUV420, "YU12", 3, { { 8, 1, 1 }, { 8, 2, 2 }, {8, 2, 2 } }, },
    { DRM_FORMAT_ARGB8888, "AR24", 1, { { 32, 1, 1 } }, },
};
 
#define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
 
const struct format_info *FindFormat(uint32_t format)
{
    for (int i = 0; i < (int)ARRAY_SIZE(format_info_array); i++) {
        if (format == format_info_array[i].format)
            return &format_info_array[i];
    }
    return NULL;
}
 
void cDrmBuffer::Setup(int drmDeviceFd, uint32_t width, uint32_t height, uint32_t pixFmt, AVDRMFrameDescriptor *primedata, bool closeHandleOnDestroy)
{
    uint64_t modifier[4] = { 0, 0, 0, 0 };
    uint32_t mod_flags = 0;
    m_planePrimeHandle[0] = m_planePrimeHandle[1] = m_planePrimeHandle[2] = m_planePrimeHandle[3] = 0;
    m_pitch[0] = m_pitch[1] = m_pitch[2] = m_pitch[3] = 0;
    m_offset[0] = m_offset[1] = m_offset[2] = m_offset[3] = 0;
    m_numObjects = m_numPlanes = 0;
 
    m_width = width;
    m_height = height;
    m_pixFmt = pixFmt;
    m_drmDeviceFd = drmDeviceFd;
    m_closeHandleOnDestroy = closeHandleOnDestroy;
 
    if (primedata) {
        // we have no DRM objects yet, so return
        if (!primedata->nb_objects)
            LOGFATAL("drmbuffer: %s: No primedata objects available!", __FUNCTION__);
 
        AVDRMLayerDescriptor *layer = &primedata->layers[0];
 
        m_pixFmt = layer->format;
        m_numPlanes = layer->nb_planes;
        m_numObjects = primedata->nb_objects;
 
//      LOGDEBUG2(L_DRM, "drmbuffer: %s: PRIMEDATA %d x %d, pix_fmt %4.4s nb_planes %d nb_objects %d", __FUNCTION__,
//          m_width, m_height, (char *)&m_pixFmt, m_numPlanes, m_numObjects);
 
        // create handles for PrimeFDs
        for (int object = 0; object < primedata->nb_objects; object++) {
            if (drmPrimeFDToHandle(drmDeviceFd, primedata->objects[object].fd, &m_objectPrimeHandle[object])) {
                LOGFATAL("drmbuffer: %s: PRIMEDATA Failed to retrieve the Prime Handle %i size %zu (%d): %m", __FUNCTION__,
                    primedata->objects[object].fd,
                    primedata->objects[object].size, errno);
            }
 
            m_dmaBufHandle[object] = primedata->objects[object].fd;
            m_size[object] = primedata->objects[object].size;
//          LOGDEBUG2(L_DRM, "drmbuffer: %s: PRIMEDATA create handle for PrimeFD (%d|%i): PrimeFD %i ToHandle %i size %zu modifier %" PRIx64 "",
//              __FUNCTION__, object, primedata->nb_objects, primedata->objects[object].fd, m_primehandle[object],
//              primedata->objects[object].size, primedata->objects[object].format_modifier);
        }
 
        // fill the planes
        for (int plane = 0; plane < layer->nb_planes; plane++) {
            int object = layer->planes[plane].object_index;
            uint32_t handle = m_objectPrimeHandle[object];
            if (handle) {
                m_planePrimeHandle[plane] = handle;
                m_pitch[plane] = layer->planes[plane].pitch;
                m_offset[plane] = layer->planes[plane].offset;
                m_objIdx[plane] = object;
                if (primedata->objects[object].format_modifier)
                    modifier[plane] = primedata->objects[object].format_modifier;
 
                // LOGDEBUG2(L_DRM, "drmbuffer: %s: PRIMEDATA fill plane %d: handle %d object_index %i pitch %d offset %d size %d modifier %" PRIx64 " (plane not mapped!)",
                //  __FUNCTION__, plane, m_handle[plane], m_objIdx[plane], m_pitch[plane], m_offset[plane], m_size[plane], modifier[plane]);
            }
        }
        if (modifier[0] && modifier[0] != DRM_FORMAT_MOD_INVALID)
            mod_flags = DRM_MODE_FB_MODIFIERS;
    } else {
        const struct format_info *format_info = FindFormat(m_pixFmt);
        if (!format_info)
            LOGFATAL("drmbuffer: %s: No suitable format found!", __FUNCTION__);
 
        m_numPlanes = format_info->num_planes;
 
        // LOGDEBUG2(L_DRM, "drmbuffer: %s:  %d x %d, pix_fmt %4.4s nb_planes %d", __FUNCTION__,
        //  m_width, m_height, (char *)&m_pixFmt, m_numPlanes);
 
        for (int plane = 0; plane < format_info->num_planes; plane++) {
            const struct format_plane_info *plane_info = &format_info->planes[plane];
 
            struct drm_mode_create_dumb creq;
            creq.height = m_height / plane_info->ysub;
            creq.width = m_width / plane_info->xsub;
            creq.bpp = plane_info->bitspp;
            creq.flags = 0;
            creq.handle = 0;
            creq.pitch = 0;
            creq.size = 0;
 
            if (drmIoctl(drmDeviceFd, DRM_IOCTL_MODE_CREATE_DUMB, &creq) < 0)
                LOGFATAL("drmbuffer: %s: cannot create dumb buffer %dx%d@%d (%d): %m", __FUNCTION__,
                    creq.width, creq.height, creq.bpp, errno);
 
            m_planePrimeHandle[plane] = creq.handle;
            m_pitch[plane] = creq.pitch;
            m_size[plane] = creq.size;
 
            struct drm_mode_map_dumb mreq;
            memset(&mreq, 0, sizeof(struct drm_mode_map_dumb));
            mreq.handle = m_planePrimeHandle[plane];
 
            if (drmIoctl(drmDeviceFd, DRM_IOCTL_MODE_MAP_DUMB, &mreq))
                LOGFATAL("drmbuffer: %s: cannot prepare dumb buffer for mapping (%d): %m", __FUNCTION__, errno);
 
            m_pPlane[plane] = (uint8_t *)mmap(0, creq.size, PROT_READ | PROT_WRITE, MAP_SHARED, drmDeviceFd, mreq.offset);
 
            if (m_pPlane[plane] == MAP_FAILED)
                LOGFATAL("drmbuffer: %s: cannot map dumb buffer (%d): %m", __FUNCTION__, errno);
 
            memset(m_pPlane[plane], 0, m_size[plane]);
 
            // LOGDEBUG2(L_DRM, "drmbuffer: %s: fill plane %d: prime handle %d pitch %d offset %d size %d address %p", __FUNCTION__,
            //  plane, m_planePrimeHandle[plane], m_pitch[plane], m_offset[plane], m_size[plane], m_pPlane[plane]);
        }
    }
 
    int ret = -1;
    ret = drmModeAddFB2WithModifiers(drmDeviceFd, m_width, m_height, m_pixFmt, m_planePrimeHandle, m_pitch, m_offset, modifier, &m_fbId, mod_flags);
 
    if (ret) {
        if (mod_flags)
            LOGERROR("drmbuffer: %s: cannot create modifiers framebuffer (%d): %m", __FUNCTION__, errno);
 
        ret = drmModeAddFB2(drmDeviceFd, m_width, m_height, m_pixFmt,
            m_planePrimeHandle, m_pitch, m_offset, &m_fbId, 0);
    }
 
    if (ret)
        LOGFATAL("drmbuffer: %s: cannot create framebuffer (%d): %m", __FUNCTION__, errno);
 
    LOGDEBUG2(L_DRM, "drmbuffer: %s: Added %sFB fb_id %d width %d height %d pix_fmt %4.4s", __FUNCTION__,
        primedata ? "primedata " : "", m_fbId, m_width, m_height, (char *)&m_pixFmt);
 
    m_dirty = true;
}
 
void cDrmBuffer::FillBlack(void)
{
    for (uint32_t i = 0; i < m_width * m_height; ++i) {
        m_pPlane[0][i] = 0x10;
        if (i < m_width * m_height / 2)
            m_pPlane[1][i] = 0x80;
    }
}
 
void cDrmBuffer::PresentationFinished(void)
{
    av_frame_free(&frame);
    m_presentationPending = false;
 
    if (m_destroyAfterUse)
        Destroy();
}
 
/*****************************************************************************
 * Drm Buffer Pool
 ****************************************************************************/
 
cDrmBuffer *cDrmBufferPool::FindByDmaBufHandle(int primeFd)
{
    for (const auto &buf : buffer) {
        if (buf->IsDirty() && buf->DmaBufHandle() == primeFd)
            return buf.get();
    }
 
    return nullptr;
}
 
cDrmBuffer *cDrmBufferPool::FindUninitilized()
{
    int i = 0;
    for (const auto &buf : buffer) {
        if (!buf->IsDirty())
            return buf.get();
 
        i++;
    }
 
    return nullptr;
}
 
cDrmBuffer *cDrmBufferPool::FindNoPresentationPending()
{
    for (const auto &buf : buffer) {
        if (buf->IsDirty() && !buf->IsPresentationPending())
            return buf.get();
    }
 
    return nullptr;
}
 
void cDrmBufferPool::DestroyAllExcept(cDrmBuffer *exceptBuf)
{
    for (const auto &buf : buffer) {
        if (buf.get() != exceptBuf && buf->IsDirty()) {
            av_frame_free(&buf->frame);
            buf->Destroy();
        }
    }
}