kwin/src/backends/drm/drm_buffer.cpp
2021-12-14 16:31:15 +00:00

153 lines
3.6 KiB
C++

/*
KWin - the KDE window manager
This file is part of the KDE project.
SPDX-FileCopyrightText: 2015 Martin Gräßlin <mgraesslin@kde.org>
SPDX-License-Identifier: GPL-2.0-or-later
*/
#include "drm_buffer.h"
#include "logging.h"
#include "drm_gpu.h"
// system
#include <sys/mman.h>
// c++
#include <cerrno>
// drm
#include <xf86drm.h>
#include <xf86drmMode.h>
#include <drm_fourcc.h>
namespace KWin
{
DrmBuffer::DrmBuffer(DrmGpu *gpu, uint32_t format, uint64_t modifier)
: m_gpu(gpu)
, m_format(format)
, m_modifier(modifier)
{
}
quint32 DrmBuffer::bufferId() const
{
return m_bufferId;
}
const QSize &DrmBuffer::size() const
{
return m_size;
}
DrmGpu *DrmBuffer::gpu() const
{
return m_gpu;
}
uint32_t DrmBuffer::format() const
{
return m_format;
}
uint64_t DrmBuffer::modifier() const
{
return m_modifier;
}
// DrmDumbBuffer
DrmDumbBuffer::DrmDumbBuffer(DrmGpu *gpu, const QSize &size, uint32_t drmFormat)
: DrmBuffer(gpu, drmFormat, DRM_FORMAT_MOD_LINEAR)
{
m_size = size;
drm_mode_create_dumb createArgs;
memset(&createArgs, 0, sizeof createArgs);
createArgs.bpp = 32;
createArgs.width = size.width();
createArgs.height = size.height();
if (drmIoctl(m_gpu->fd(), DRM_IOCTL_MODE_CREATE_DUMB, &createArgs) != 0) {
qCWarning(KWIN_DRM) << "DRM_IOCTL_MODE_CREATE_DUMB failed" << strerror(errno);
return;
}
m_handle = createArgs.handle;
m_bufferSize = createArgs.size;
m_stride = createArgs.pitch;
uint32_t handles[4] = {m_handle, 0, 0, 0};
uint32_t strides[4] = {m_stride, 0, 0, 0};
uint32_t offsets[4] = {0, 0, 0, 0};
if (drmModeAddFB2(m_gpu->fd(), size.width(), size.height(), drmFormat, handles, strides, offsets, &m_bufferId, 0) != 0) {
if (drmModeAddFB(m_gpu->fd(), size.width(), size.height(), 24, 32, m_stride, createArgs.handle, &m_bufferId) != 0) {
qCWarning(KWIN_DRM) << "drmModeAddFB2 and drmModeAddFB both failed!" << strerror(errno);
} else {
qCWarning(KWIN_DRM) << "drmModeAddFB2 failed! Falling back to drmModeAddFB with XRGB8888" << strerror(errno);
m_format = DRM_FORMAT_XRGB8888;
}
}
}
DrmDumbBuffer::~DrmDumbBuffer()
{
if (m_bufferId) {
drmModeRmFB(m_gpu->fd(), m_bufferId);
}
delete m_image;
if (m_memory) {
munmap(m_memory, m_bufferSize);
}
if (m_handle) {
drm_mode_destroy_dumb destroyArgs;
destroyArgs.handle = m_handle;
drmIoctl(m_gpu->fd(), DRM_IOCTL_MODE_DESTROY_DUMB, &destroyArgs);
}
}
bool DrmDumbBuffer::needsModeChange(DrmBuffer *b) const {
if (DrmDumbBuffer *db = dynamic_cast<DrmDumbBuffer*>(b)) {
return m_stride != db->stride();
} else {
return true;
}
}
bool DrmDumbBuffer::map(QImage::Format format)
{
if (!m_handle || !m_bufferId) {
return false;
}
drm_mode_map_dumb mapArgs;
memset(&mapArgs, 0, sizeof mapArgs);
mapArgs.handle = m_handle;
if (drmIoctl(m_gpu->fd(), DRM_IOCTL_MODE_MAP_DUMB, &mapArgs) != 0) {
return false;
}
void *address = mmap(nullptr, m_bufferSize, PROT_WRITE, MAP_SHARED, m_gpu->fd(), mapArgs.offset);
if (address == MAP_FAILED) {
return false;
}
m_memory = address;
m_image = new QImage((uchar*)m_memory, m_size.width(), m_size.height(), m_stride, format);
return !m_image->isNull();
}
quint32 DrmDumbBuffer::handle() const
{
return m_handle;
}
QImage *DrmDumbBuffer::image() const
{
return m_image;
}
void *DrmDumbBuffer::data() const
{
return m_memory;
}
quint32 DrmDumbBuffer::stride() const
{
return m_stride;
}
}