yading@10: /* yading@10: * Copyright (c) 2011 Stefano Sabatini yading@10: * Copyright (c) 2009 Giliard B. de Freitas yading@10: * Copyright (C) 2002 Gunnar Monell yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: /** yading@10: * @file yading@10: * Linux framebuffer input device, yading@10: * inspired by code from fbgrab.c by Gunnar Monell. yading@10: * @see http://linux-fbdev.sourceforge.net/ yading@10: */ yading@10: yading@10: /* #define DEBUG */ yading@10: yading@10: #include yading@10: #include yading@10: #include yading@10: #include yading@10: #include yading@10: #include yading@10: yading@10: #include "libavutil/log.h" yading@10: #include "libavutil/mem.h" yading@10: #include "libavutil/opt.h" yading@10: #include "libavutil/time.h" yading@10: #include "libavutil/parseutils.h" yading@10: #include "libavutil/pixdesc.h" yading@10: #include "avdevice.h" yading@10: #include "libavformat/internal.h" yading@10: yading@10: struct rgb_pixfmt_map_entry { yading@10: int bits_per_pixel; yading@10: int red_offset, green_offset, blue_offset, alpha_offset; yading@10: enum AVPixelFormat pixfmt; yading@10: }; yading@10: yading@10: static const struct rgb_pixfmt_map_entry rgb_pixfmt_map[] = { yading@10: // bpp, red_offset, green_offset, blue_offset, alpha_offset, pixfmt yading@10: { 32, 0, 8, 16, 24, AV_PIX_FMT_RGBA }, yading@10: { 32, 16, 8, 0, 24, AV_PIX_FMT_BGRA }, yading@10: { 32, 8, 16, 24, 0, AV_PIX_FMT_ARGB }, yading@10: { 32, 3, 2, 8, 0, AV_PIX_FMT_ABGR }, yading@10: { 24, 0, 8, 16, 0, AV_PIX_FMT_RGB24 }, yading@10: { 24, 16, 8, 0, 0, AV_PIX_FMT_BGR24 }, yading@10: }; yading@10: yading@10: static enum AVPixelFormat get_pixfmt_from_fb_varinfo(struct fb_var_screeninfo *varinfo) yading@10: { yading@10: int i; yading@10: yading@10: for (i = 0; i < FF_ARRAY_ELEMS(rgb_pixfmt_map); i++) { yading@10: const struct rgb_pixfmt_map_entry *entry = &rgb_pixfmt_map[i]; yading@10: if (entry->bits_per_pixel == varinfo->bits_per_pixel && yading@10: entry->red_offset == varinfo->red.offset && yading@10: entry->green_offset == varinfo->green.offset && yading@10: entry->blue_offset == varinfo->blue.offset) yading@10: return entry->pixfmt; yading@10: } yading@10: yading@10: return AV_PIX_FMT_NONE; yading@10: } yading@10: yading@10: typedef struct { yading@10: AVClass *class; ///< class for private options yading@10: int frame_size; ///< size in bytes of a grabbed frame yading@10: AVRational framerate_q; ///< framerate yading@10: char *framerate; ///< framerate string set by a private option yading@10: int64_t time_frame; ///< time for the next frame to output (in 1/1000000 units) yading@10: yading@10: int fd; ///< framebuffer device file descriptor yading@10: int width, height; ///< assumed frame resolution yading@10: int frame_linesize; ///< linesize of the output frame, it is assumed to be constant yading@10: int bytes_per_pixel; yading@10: yading@10: struct fb_var_screeninfo varinfo; ///< variable info; yading@10: struct fb_fix_screeninfo fixinfo; ///< fixed info; yading@10: yading@10: uint8_t *data; ///< framebuffer data yading@10: } FBDevContext; yading@10: yading@10: static av_cold int fbdev_read_header(AVFormatContext *avctx) yading@10: { yading@10: FBDevContext *fbdev = avctx->priv_data; yading@10: AVStream *st = NULL; yading@10: enum AVPixelFormat pix_fmt; yading@10: int ret, flags = O_RDONLY; yading@10: yading@10: ret = av_parse_video_rate(&fbdev->framerate_q, fbdev->framerate); yading@10: if (ret < 0) { yading@10: av_log(avctx, AV_LOG_ERROR, "Could not parse framerate '%s'.\n", fbdev->framerate); yading@10: return ret; yading@10: } yading@10: yading@10: if (!(st = avformat_new_stream(avctx, NULL))) yading@10: return AVERROR(ENOMEM); yading@10: avpriv_set_pts_info(st, 64, 1, 1000000); /* 64 bits pts in microseconds */ yading@10: yading@10: /* NONBLOCK is ignored by the fbdev driver, only set for consistency */ yading@10: if (avctx->flags & AVFMT_FLAG_NONBLOCK) yading@10: flags |= O_NONBLOCK; yading@10: yading@10: if ((fbdev->fd = open(avctx->filename, flags)) == -1) { yading@10: ret = AVERROR(errno); yading@10: av_log(avctx, AV_LOG_ERROR, yading@10: "Could not open framebuffer device '%s': %s\n", yading@10: avctx->filename, strerror(ret)); yading@10: return ret; yading@10: } yading@10: yading@10: if (ioctl(fbdev->fd, FBIOGET_VSCREENINFO, &fbdev->varinfo) < 0) { yading@10: ret = AVERROR(errno); yading@10: av_log(avctx, AV_LOG_ERROR, yading@10: "FBIOGET_VSCREENINFO: %s\n", strerror(errno)); yading@10: goto fail; yading@10: } yading@10: yading@10: if (ioctl(fbdev->fd, FBIOGET_FSCREENINFO, &fbdev->fixinfo) < 0) { yading@10: ret = AVERROR(errno); yading@10: av_log(avctx, AV_LOG_ERROR, yading@10: "FBIOGET_FSCREENINFO: %s\n", strerror(errno)); yading@10: goto fail; yading@10: } yading@10: yading@10: pix_fmt = get_pixfmt_from_fb_varinfo(&fbdev->varinfo); yading@10: if (pix_fmt == AV_PIX_FMT_NONE) { yading@10: ret = AVERROR(EINVAL); yading@10: av_log(avctx, AV_LOG_ERROR, yading@10: "Framebuffer pixel format not supported.\n"); yading@10: goto fail; yading@10: } yading@10: yading@10: fbdev->width = fbdev->varinfo.xres; yading@10: fbdev->height = fbdev->varinfo.yres; yading@10: fbdev->bytes_per_pixel = (fbdev->varinfo.bits_per_pixel + 7) >> 3; yading@10: fbdev->frame_linesize = fbdev->width * fbdev->bytes_per_pixel; yading@10: fbdev->frame_size = fbdev->frame_linesize * fbdev->height; yading@10: fbdev->time_frame = AV_NOPTS_VALUE; yading@10: fbdev->data = mmap(NULL, fbdev->fixinfo.smem_len, PROT_READ, MAP_SHARED, fbdev->fd, 0); yading@10: if (fbdev->data == MAP_FAILED) { yading@10: ret = AVERROR(errno); yading@10: av_log(avctx, AV_LOG_ERROR, "Error in mmap(): %s\n", strerror(errno)); yading@10: goto fail; yading@10: } yading@10: yading@10: st->codec->codec_type = AVMEDIA_TYPE_VIDEO; yading@10: st->codec->codec_id = AV_CODEC_ID_RAWVIDEO; yading@10: st->codec->width = fbdev->width; yading@10: st->codec->height = fbdev->height; yading@10: st->codec->pix_fmt = pix_fmt; yading@10: st->codec->time_base = av_inv_q(fbdev->framerate_q); yading@10: st->codec->bit_rate = yading@10: fbdev->width * fbdev->height * fbdev->bytes_per_pixel * av_q2d(fbdev->framerate_q) * 8; yading@10: yading@10: av_log(avctx, AV_LOG_INFO, yading@10: "w:%d h:%d bpp:%d pixfmt:%s fps:%d/%d bit_rate:%d\n", yading@10: fbdev->width, fbdev->height, fbdev->varinfo.bits_per_pixel, yading@10: av_get_pix_fmt_name(pix_fmt), yading@10: fbdev->framerate_q.num, fbdev->framerate_q.den, yading@10: st->codec->bit_rate); yading@10: return 0; yading@10: yading@10: fail: yading@10: close(fbdev->fd); yading@10: return ret; yading@10: } yading@10: yading@10: static int fbdev_read_packet(AVFormatContext *avctx, AVPacket *pkt) yading@10: { yading@10: FBDevContext *fbdev = avctx->priv_data; yading@10: int64_t curtime, delay; yading@10: struct timespec ts; yading@10: int i, ret; yading@10: uint8_t *pin, *pout; yading@10: yading@10: if (fbdev->time_frame == AV_NOPTS_VALUE) yading@10: fbdev->time_frame = av_gettime(); yading@10: yading@10: /* wait based on the frame rate */ yading@10: while (1) { yading@10: curtime = av_gettime(); yading@10: delay = fbdev->time_frame - curtime; yading@10: av_dlog(avctx, yading@10: "time_frame:%"PRId64" curtime:%"PRId64" delay:%"PRId64"\n", yading@10: fbdev->time_frame, curtime, delay); yading@10: if (delay <= 0) { yading@10: fbdev->time_frame += INT64_C(1000000) / av_q2d(fbdev->framerate_q); yading@10: break; yading@10: } yading@10: if (avctx->flags & AVFMT_FLAG_NONBLOCK) yading@10: return AVERROR(EAGAIN); yading@10: ts.tv_sec = delay / 1000000; yading@10: ts.tv_nsec = (delay % 1000000) * 1000; yading@10: while (nanosleep(&ts, &ts) < 0 && errno == EINTR); yading@10: } yading@10: yading@10: if ((ret = av_new_packet(pkt, fbdev->frame_size)) < 0) yading@10: return ret; yading@10: yading@10: /* refresh fbdev->varinfo, visible data position may change at each call */ yading@10: if (ioctl(fbdev->fd, FBIOGET_VSCREENINFO, &fbdev->varinfo) < 0) yading@10: av_log(avctx, AV_LOG_WARNING, yading@10: "Error refreshing variable info: %s\n", strerror(errno)); yading@10: yading@10: pkt->pts = curtime; yading@10: yading@10: /* compute visible data offset */ yading@10: pin = fbdev->data + fbdev->bytes_per_pixel * fbdev->varinfo.xoffset + yading@10: fbdev->varinfo.yoffset * fbdev->fixinfo.line_length; yading@10: pout = pkt->data; yading@10: yading@10: for (i = 0; i < fbdev->height; i++) { yading@10: memcpy(pout, pin, fbdev->frame_linesize); yading@10: pin += fbdev->fixinfo.line_length; yading@10: pout += fbdev->frame_linesize; yading@10: } yading@10: yading@10: return fbdev->frame_size; yading@10: } yading@10: yading@10: static av_cold int fbdev_read_close(AVFormatContext *avctx) yading@10: { yading@10: FBDevContext *fbdev = avctx->priv_data; yading@10: yading@10: munmap(fbdev->data, fbdev->frame_size); yading@10: close(fbdev->fd); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: #define OFFSET(x) offsetof(FBDevContext, x) yading@10: #define DEC AV_OPT_FLAG_DECODING_PARAM yading@10: static const AVOption options[] = { yading@10: { "framerate","", OFFSET(framerate), AV_OPT_TYPE_STRING, {.str = "25"}, 0, 0, DEC }, yading@10: { NULL }, yading@10: }; yading@10: yading@10: static const AVClass fbdev_class = { yading@10: .class_name = "fbdev indev", yading@10: .item_name = av_default_item_name, yading@10: .option = options, yading@10: .version = LIBAVUTIL_VERSION_INT, yading@10: }; yading@10: yading@10: AVInputFormat ff_fbdev_demuxer = { yading@10: .name = "fbdev", yading@10: .long_name = NULL_IF_CONFIG_SMALL("Linux framebuffer"), yading@10: .priv_data_size = sizeof(FBDevContext), yading@10: .read_header = fbdev_read_header, yading@10: .read_packet = fbdev_read_packet, yading@10: .read_close = fbdev_read_close, yading@10: .flags = AVFMT_NOFILE, yading@10: .priv_class = &fbdev_class, yading@10: };