mirror of
https://github.com/ossrs/srs.git
synced 2025-03-09 15:49:59 +00:00
289 lines
7.4 KiB
C
289 lines
7.4 KiB
C
/*
|
|
* This file is part of FFmpeg.
|
|
*
|
|
* FFmpeg is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with FFmpeg; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include <fcntl.h>
|
|
#include <sys/mman.h>
|
|
#include <unistd.h>
|
|
|
|
#include <drm.h>
|
|
#include <xf86drm.h>
|
|
|
|
#include "avassert.h"
|
|
#include "hwcontext.h"
|
|
#include "hwcontext_drm.h"
|
|
#include "hwcontext_internal.h"
|
|
#include "imgutils.h"
|
|
|
|
|
|
static void drm_device_free(AVHWDeviceContext *hwdev)
|
|
{
|
|
AVDRMDeviceContext *hwctx = hwdev->hwctx;
|
|
|
|
close(hwctx->fd);
|
|
}
|
|
|
|
static int drm_device_create(AVHWDeviceContext *hwdev, const char *device,
|
|
AVDictionary *opts, int flags)
|
|
{
|
|
AVDRMDeviceContext *hwctx = hwdev->hwctx;
|
|
drmVersionPtr version;
|
|
|
|
hwctx->fd = open(device, O_RDWR);
|
|
if (hwctx->fd < 0)
|
|
return AVERROR(errno);
|
|
|
|
version = drmGetVersion(hwctx->fd);
|
|
if (!version) {
|
|
av_log(hwdev, AV_LOG_ERROR, "Failed to get version information "
|
|
"from %s: probably not a DRM device?\n", device);
|
|
close(hwctx->fd);
|
|
return AVERROR(EINVAL);
|
|
}
|
|
|
|
av_log(hwdev, AV_LOG_VERBOSE, "Opened DRM device %s: driver %s "
|
|
"version %d.%d.%d.\n", device, version->name,
|
|
version->version_major, version->version_minor,
|
|
version->version_patchlevel);
|
|
|
|
drmFreeVersion(version);
|
|
|
|
hwdev->free = &drm_device_free;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int drm_get_buffer(AVHWFramesContext *hwfc, AVFrame *frame)
|
|
{
|
|
frame->buf[0] = av_buffer_pool_get(hwfc->pool);
|
|
if (!frame->buf[0])
|
|
return AVERROR(ENOMEM);
|
|
|
|
frame->data[0] = (uint8_t*)frame->buf[0]->data;
|
|
|
|
frame->format = AV_PIX_FMT_DRM_PRIME;
|
|
frame->width = hwfc->width;
|
|
frame->height = hwfc->height;
|
|
|
|
return 0;
|
|
}
|
|
|
|
typedef struct DRMMapping {
|
|
// Address and length of each mmap()ed region.
|
|
int nb_regions;
|
|
void *address[AV_DRM_MAX_PLANES];
|
|
size_t length[AV_DRM_MAX_PLANES];
|
|
} DRMMapping;
|
|
|
|
static void drm_unmap_frame(AVHWFramesContext *hwfc,
|
|
HWMapDescriptor *hwmap)
|
|
{
|
|
DRMMapping *map = hwmap->priv;
|
|
int i;
|
|
|
|
for (i = 0; i < map->nb_regions; i++)
|
|
munmap(map->address[i], map->length[i]);
|
|
|
|
av_free(map);
|
|
}
|
|
|
|
static int drm_map_frame(AVHWFramesContext *hwfc,
|
|
AVFrame *dst, const AVFrame *src, int flags)
|
|
{
|
|
const AVDRMFrameDescriptor *desc = (AVDRMFrameDescriptor*)src->data[0];
|
|
DRMMapping *map;
|
|
int err, i, p, plane;
|
|
int mmap_prot;
|
|
void *addr;
|
|
|
|
map = av_mallocz(sizeof(*map));
|
|
if (!map)
|
|
return AVERROR(ENOMEM);
|
|
|
|
mmap_prot = 0;
|
|
if (flags & AV_HWFRAME_MAP_READ)
|
|
mmap_prot |= PROT_READ;
|
|
if (flags & AV_HWFRAME_MAP_WRITE)
|
|
mmap_prot |= PROT_WRITE;
|
|
|
|
av_assert0(desc->nb_objects <= AV_DRM_MAX_PLANES);
|
|
for (i = 0; i < desc->nb_objects; i++) {
|
|
addr = mmap(NULL, desc->objects[i].size, mmap_prot, MAP_SHARED,
|
|
desc->objects[i].fd, 0);
|
|
if (addr == MAP_FAILED) {
|
|
err = AVERROR(errno);
|
|
av_log(hwfc, AV_LOG_ERROR, "Failed to map DRM object %d to "
|
|
"memory: %d.\n", desc->objects[i].fd, errno);
|
|
goto fail;
|
|
}
|
|
|
|
map->address[i] = addr;
|
|
map->length[i] = desc->objects[i].size;
|
|
}
|
|
map->nb_regions = i;
|
|
|
|
plane = 0;
|
|
for (i = 0; i < desc->nb_layers; i++) {
|
|
const AVDRMLayerDescriptor *layer = &desc->layers[i];
|
|
for (p = 0; p < layer->nb_planes; p++) {
|
|
dst->data[plane] =
|
|
(uint8_t*)map->address[layer->planes[p].object_index] +
|
|
layer->planes[p].offset;
|
|
dst->linesize[plane] = layer->planes[p].pitch;
|
|
++plane;
|
|
}
|
|
}
|
|
av_assert0(plane <= AV_DRM_MAX_PLANES);
|
|
|
|
dst->width = src->width;
|
|
dst->height = src->height;
|
|
|
|
err = ff_hwframe_map_create(src->hw_frames_ctx, dst, src,
|
|
&drm_unmap_frame, map);
|
|
if (err < 0)
|
|
goto fail;
|
|
|
|
return 0;
|
|
|
|
fail:
|
|
for (i = 0; i < desc->nb_objects; i++) {
|
|
if (map->address[i])
|
|
munmap(map->address[i], map->length[i]);
|
|
}
|
|
av_free(map);
|
|
return err;
|
|
}
|
|
|
|
static int drm_transfer_get_formats(AVHWFramesContext *ctx,
|
|
enum AVHWFrameTransferDirection dir,
|
|
enum AVPixelFormat **formats)
|
|
{
|
|
enum AVPixelFormat *pix_fmts;
|
|
|
|
pix_fmts = av_malloc_array(2, sizeof(*pix_fmts));
|
|
if (!pix_fmts)
|
|
return AVERROR(ENOMEM);
|
|
|
|
pix_fmts[0] = ctx->sw_format;
|
|
pix_fmts[1] = AV_PIX_FMT_NONE;
|
|
|
|
*formats = pix_fmts;
|
|
return 0;
|
|
}
|
|
|
|
static int drm_transfer_data_from(AVHWFramesContext *hwfc,
|
|
AVFrame *dst, const AVFrame *src)
|
|
{
|
|
AVFrame *map;
|
|
int err;
|
|
|
|
if (dst->width > hwfc->width || dst->height > hwfc->height)
|
|
return AVERROR(EINVAL);
|
|
|
|
map = av_frame_alloc();
|
|
if (!map)
|
|
return AVERROR(ENOMEM);
|
|
map->format = dst->format;
|
|
|
|
err = drm_map_frame(hwfc, map, src, AV_HWFRAME_MAP_READ);
|
|
if (err)
|
|
goto fail;
|
|
|
|
map->width = dst->width;
|
|
map->height = dst->height;
|
|
|
|
err = av_frame_copy(dst, map);
|
|
if (err)
|
|
goto fail;
|
|
|
|
err = 0;
|
|
fail:
|
|
av_frame_free(&map);
|
|
return err;
|
|
}
|
|
|
|
static int drm_transfer_data_to(AVHWFramesContext *hwfc,
|
|
AVFrame *dst, const AVFrame *src)
|
|
{
|
|
AVFrame *map;
|
|
int err;
|
|
|
|
if (src->width > hwfc->width || src->height > hwfc->height)
|
|
return AVERROR(EINVAL);
|
|
|
|
map = av_frame_alloc();
|
|
if (!map)
|
|
return AVERROR(ENOMEM);
|
|
map->format = src->format;
|
|
|
|
err = drm_map_frame(hwfc, map, dst, AV_HWFRAME_MAP_WRITE |
|
|
AV_HWFRAME_MAP_OVERWRITE);
|
|
if (err)
|
|
goto fail;
|
|
|
|
map->width = src->width;
|
|
map->height = src->height;
|
|
|
|
err = av_frame_copy(map, src);
|
|
if (err)
|
|
goto fail;
|
|
|
|
err = 0;
|
|
fail:
|
|
av_frame_free(&map);
|
|
return err;
|
|
}
|
|
|
|
static int drm_map_from(AVHWFramesContext *hwfc, AVFrame *dst,
|
|
const AVFrame *src, int flags)
|
|
{
|
|
int err;
|
|
|
|
if (hwfc->sw_format != dst->format)
|
|
return AVERROR(ENOSYS);
|
|
|
|
err = drm_map_frame(hwfc, dst, src, flags);
|
|
if (err)
|
|
return err;
|
|
|
|
err = av_frame_copy_props(dst, src);
|
|
if (err)
|
|
return err;
|
|
|
|
return 0;
|
|
}
|
|
|
|
const HWContextType ff_hwcontext_type_drm = {
|
|
.type = AV_HWDEVICE_TYPE_DRM,
|
|
.name = "DRM",
|
|
|
|
.device_hwctx_size = sizeof(AVDRMDeviceContext),
|
|
|
|
.device_create = &drm_device_create,
|
|
|
|
.frames_get_buffer = &drm_get_buffer,
|
|
|
|
.transfer_get_formats = &drm_transfer_get_formats,
|
|
.transfer_data_to = &drm_transfer_data_to,
|
|
.transfer_data_from = &drm_transfer_data_from,
|
|
.map_from = &drm_map_from,
|
|
|
|
.pix_fmts = (const enum AVPixelFormat[]) {
|
|
AV_PIX_FMT_DRM_PRIME,
|
|
AV_PIX_FMT_NONE
|
|
},
|
|
};
|