From 6beefa408540c72dd939d9920dcc04b373124c11 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Sat, 25 Jun 2005 09:04:08 +0000 Subject: [PATCH] *BSD bktr grab support by (Jacob Meuser: jakemsr, jakemsr com) Originally committed as revision 4392 to svn://svn.ffmpeg.org/ffmpeg/trunk --- configure | 14 ++ ffmpeg.c | 4 + libavformat/Makefile | 4 + libavformat/allformats.c | 2 +- libavformat/avformat.h | 2 +- libavformat/grab_bktr.c | 331 +++++++++++++++++++++++++++++++++++++++ 6 files changed, 355 insertions(+), 2 deletions(-) create mode 100644 libavformat/grab_bktr.c diff --git a/configure b/configure index e34d30546f..3ae70513a0 100755 --- a/configure +++ b/configure @@ -57,6 +57,7 @@ echo " --disable-altivec disable AltiVec usage" echo " --disable-audio-oss disable OSS audio support [default=no]" echo " --disable-audio-beos disable BeOS audio support [default=no]" echo " --disable-v4l disable video4linux grabbing [default=no]" +echo " --disable-bktr disable bktr video grabbing [default=no]" echo " --disable-dv1394 disable DV1394 grabbing [default=no]" echo " --disable-network disable network support [default=no]" echo " --disable-zlib disable zlib [default=no]" @@ -153,6 +154,7 @@ case "$cpu" in esac gprof="no" v4l="yes" +bktr="no" audio_oss="yes" audio_beos="no" dv1394="yes" @@ -258,6 +260,7 @@ extralibs="$extralibs -lsocket -lnsl" ;; NetBSD) v4l="no" +bktr="yes" audio_oss="yes" dv1394="no" make="gmake" @@ -269,6 +272,7 @@ esac ;; OpenBSD) v4l="no" +bktr="yes" audio_oss="yes" dv1394="no" make="gmake" @@ -279,6 +283,7 @@ extralibs="$extralibs -lossaudio" ;; FreeBSD) v4l="no" +bktr="yes" audio_oss="yes" dv1394="no" make="gmake" @@ -287,6 +292,7 @@ LDFLAGS="$LDFLAGS -export-dynamic -pthread" ;; BSD/OS) v4l="no" +bktr="yes" audio_oss="yes" dv1394="no" extralibs="-lpoll -lgnugetopt -lm" @@ -431,6 +437,8 @@ for opt do ;; --disable-v4l) v4l="no" ;; + --disable-bktr) bktr="no" + ;; --disable-audio-oss) audio_oss="no" ;; --disable-audio-beos) audio_beos="no" @@ -769,6 +777,7 @@ fi if test "$mingw32" = "yes" ; then v4l="no" + bktr="no" audio_oss="no" dv1394="no" dc1394="no" @@ -1386,6 +1395,11 @@ if test "$v4l" = "yes" ; then echo "CONFIG_VIDEO4LINUX=yes" >> config.mak fi +if test "$bktr" = "yes" ; then + echo "#define CONFIG_BKTR 1" >> $TMPH + echo "CONFIG_BKTR=yes" >> config.mak +fi + if test "$dv1394" = "yes" ; then echo "#define CONFIG_DV1394 1" >> $TMPH echo "CONFIG_DV1394=yes" >> config.mak diff --git a/ffmpeg.c b/ffmpeg.c index a8551323ad..10ecfaccac 100644 --- a/ffmpeg.c +++ b/ffmpeg.c @@ -257,7 +257,11 @@ static int opt_shortest = 0; // static int rate_emu = 0; +#ifdef CONFIG_BKTR +static char *video_grab_format = "bktr"; +#else static char *video_grab_format = "video4linux"; +#endif static char *video_device = NULL; static char *grab_device = NULL; static int video_channel = 0; diff --git a/libavformat/Makefile b/libavformat/Makefile index 795ba022d6..c9a8af5ef2 100644 --- a/libavformat/Makefile +++ b/libavformat/Makefile @@ -40,6 +40,10 @@ ifeq ($(CONFIG_VIDEO4LINUX),yes) OBJS+= grab.o endif +ifeq ($(CONFIG_BKTR),yes) +OBJS+= grab_bktr.o +endif + ifeq ($(CONFIG_DV1394),yes) OBJS+= dv1394.o endif diff --git a/libavformat/allformats.c b/libavformat/allformats.c index d13c90ea58..767d876dac 100644 --- a/libavformat/allformats.c +++ b/libavformat/allformats.c @@ -88,7 +88,7 @@ void av_register_all(void) #endif ffm_init(); -#ifdef CONFIG_VIDEO4LINUX +#if defined(CONFIG_VIDEO4LINUX) || defined(CONFIG_BKTR) video_grab_init(); #endif #if defined(CONFIG_AUDIO_OSS) || defined(CONFIG_AUDIO_BEOS) diff --git a/libavformat/avformat.h b/libavformat/avformat.h index f3f70b5b6e..c48386e7ee 100644 --- a/libavformat/avformat.h +++ b/libavformat/avformat.h @@ -107,7 +107,7 @@ typedef struct AVFormatParameters { enum PixelFormat pix_fmt; struct AVImageFormat *image_format; int channel; /* used to select dv channel */ - const char *device; /* video4linux, audio or DV device */ + const char *device; /* video, audio or DV device */ const char *standard; /* tv standard, NTSC, PAL, SECAM */ int mpeg2ts_raw:1; /* force raw MPEG2 transport stream output, if possible */ int mpeg2ts_compute_pcr:1; /* compute exact PCR for each transport diff --git a/libavformat/grab_bktr.c b/libavformat/grab_bktr.c new file mode 100644 index 0000000000..a8cf7e668c --- /dev/null +++ b/libavformat/grab_bktr.c @@ -0,0 +1,331 @@ +/* + * *BSD video grab interface + * Copyright (c) 2002 Steve O'Hara-Smith + * based on + * Linux video grab interface + * Copyright (c) 2000,2001 Gerard Lantau. + * and + * simple_grab.c Copyright (c) 1999 Roger Hardiman + * + * This library 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 of the License, or (at your option) any later version. + * + * This library 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 this library; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#include "avformat.h" +#if defined(__FreeBSD__) +# if __FreeBSD__ >= 502100 +# include +# include +# else +# include +# include +# endif +#elseif defined(__DragonFly__) +# include +# include +#else +# include +#endif +#include +#include +#include +#include +#include +#include + +typedef struct { + int video_fd; + int tuner_fd; + int width, height; + int frame_rate; + int frame_rate_base; + u_int64_t per_frame; +} VideoData; + + +#define PAL 1 +#define PALBDGHI 1 +#define NTSC 2 +#define NTSCM 2 +#define SECAM 3 +#define PALN 4 +#define PALM 5 +#define NTSCJ 6 + +/* PAL is 768 x 576. NTSC is 640 x 480 */ +#define PAL_HEIGHT 576 +#define SECAM_HEIGHT 576 +#define NTSC_HEIGHT 480 + +#ifndef VIDEO_FORMAT +#define VIDEO_FORMAT NTSC +#endif + +static int bktr_dev[] = { METEOR_DEV0, METEOR_DEV1, METEOR_DEV2, + METEOR_DEV3, METEOR_DEV_SVIDEO }; + +uint8_t *video_buf; +size_t video_buf_size; +u_int64_t last_frame_time; +volatile sig_atomic_t nsignals; + + +static void catchsignal(int signal) +{ + nsignals++; + return; +} + +static int bktr_init(const char *video_device, int width, int height, + int format, int *video_fd, int *tuner_fd, int idev, double frequency) +{ + struct meteor_geomet geo; + int h_max; + long ioctl_frequency; + char *arg; + int c; + struct sigaction act, old; + + if (idev < 0 || idev > 4) + { + arg = getenv ("BKTR_DEV"); + if (arg) + idev = atoi (arg); + if (idev < 0 || idev > 4) + idev = 1; + } + + if (format < 1 || format > 6) + { + arg = getenv ("BKTR_FORMAT"); + if (arg) + format = atoi (arg); + if (format < 1 || format > 6) + format = VIDEO_FORMAT; + } + + if (frequency <= 0) + { + arg = getenv ("BKTR_FREQUENCY"); + if (arg) + frequency = atof (arg); + if (frequency <= 0) + frequency = 0.0; + } + + memset(&act, 0, sizeof(act)); + sigemptyset(&act.sa_mask); + act.sa_handler = catchsignal; + sigaction(SIGUSR1, &act, &old); + + *tuner_fd = open("/dev/tuner0", O_RDONLY); + if (*tuner_fd < 0) + perror("Warning: Tuner not opened, continuing"); + + *video_fd = open(video_device, O_RDONLY); + if (*video_fd < 0) { + perror(video_device); + return -1; + } + + geo.rows = height; + geo.columns = width; + geo.frames = 1; + geo.oformat = METEOR_GEO_YUV_422 | METEOR_GEO_YUV_12; + + switch (format) { + case PAL: h_max = PAL_HEIGHT; c = BT848_IFORM_F_PALBDGHI; break; + case PALN: h_max = PAL_HEIGHT; c = BT848_IFORM_F_PALN; break; + case PALM: h_max = PAL_HEIGHT; c = BT848_IFORM_F_PALM; break; + case SECAM: h_max = SECAM_HEIGHT; c = BT848_IFORM_F_SECAM; break; + case NTSC: h_max = NTSC_HEIGHT; c = BT848_IFORM_F_NTSCM; break; + case NTSCJ: h_max = NTSC_HEIGHT; c = BT848_IFORM_F_NTSCJ; break; + default: h_max = PAL_HEIGHT; c = BT848_IFORM_F_PALBDGHI; break; + } + + if (height <= h_max / 2) + geo.oformat |= METEOR_GEO_EVEN_ONLY; + + if (ioctl(*video_fd, METEORSETGEO, &geo) < 0) { + perror("METEORSETGEO"); + return -1; + } + + if (ioctl(*video_fd, BT848SFMT, &c) < 0) { + perror("BT848SFMT"); + return -1; + } + + c = bktr_dev[idev]; + if (ioctl(*video_fd, METEORSINPUT, &c) < 0) { + perror("METEORSINPUT"); + return -1; + } + + video_buf_size = width * height * 12 / 8; + + video_buf = (uint8_t *)mmap((caddr_t)0, video_buf_size, + PROT_READ, MAP_SHARED, *video_fd, (off_t)0); + if (video_buf == MAP_FAILED) { + perror("mmap"); + return -1; + } + + if (frequency != 0.0) { + ioctl_frequency = (unsigned long)(frequency*16); + if (ioctl(*tuner_fd, TVTUNER_SETFREQ, &ioctl_frequency) < 0) + perror("TVTUNER_SETFREQ"); + } + + c = AUDIO_UNMUTE; + if (ioctl(*tuner_fd, BT848_SAUDIO, &c) < 0) + perror("TVTUNER_SAUDIO"); + + c = METEOR_CAP_CONTINOUS; + ioctl(*video_fd, METEORCAPTUR, &c); + + c = SIGUSR1; + ioctl(*video_fd, METEORSSIGNAL, &c); + + return 0; +} + +static void bktr_getframe(u_int64_t per_frame) +{ + u_int64_t curtime; + + curtime = av_gettime(); + if (!last_frame_time + || ((last_frame_time + per_frame) > curtime)) { + if (!usleep(last_frame_time + per_frame + per_frame / 8 - curtime)) { + if (!nsignals) + av_log(NULL, AV_LOG_INFO, + "SLEPT NO signals - %d microseconds late\n", + (int)(av_gettime() - last_frame_time - per_frame)); + } + } + nsignals = 0; + last_frame_time = curtime; +} + + +/* note: we support only one picture read at a time */ +static int grab_read_packet(AVFormatContext *s1, AVPacket *pkt) +{ + VideoData *s = s1->priv_data; + + if (av_new_packet(pkt, video_buf_size) < 0) + return -EIO; + + bktr_getframe(s->per_frame); + + pkt->pts = av_gettime() & ((1LL << 48) - 1); + memcpy(pkt->data, video_buf, video_buf_size); + + return video_buf_size; +} + +static int grab_read_header(AVFormatContext *s1, AVFormatParameters *ap) +{ + VideoData *s = s1->priv_data; + AVStream *st; + int width, height; + int frame_rate; + int frame_rate_base; + int format = -1; + const char *video_device; + + if (!ap || ap->width <= 0 || ap->height <= 0 || ap->time_base.den <= 0) + return -1; + + width = ap->width; + height = ap->height; + frame_rate = ap->time_base.den; + frame_rate_base = ap->time_base.num; + + video_device = ap->device; + if (!video_device) + video_device = "/dev/bktr0"; + + st = av_new_stream(s1, 0); + if (!st) + return -ENOMEM; + av_set_pts_info(st, 48, 1, 1000000); /* 48 bits pts in use */ + + s->width = width; + s->height = height; + s->frame_rate = frame_rate; + s->frame_rate_base = frame_rate_base; + s->per_frame = ((u_int64_t)1000000 * s->frame_rate_base) / s->frame_rate; + + st->codec.codec_type = CODEC_TYPE_VIDEO; + st->codec.pix_fmt = PIX_FMT_YUV420P; + st->codec.codec_id = CODEC_ID_RAWVIDEO; + st->codec.width = width; + st->codec.height = height; + st->codec.time_base.den = frame_rate; + st->codec.time_base.num = frame_rate_base; + + if (ap->standard) { + if (!strcasecmp(ap->standard, "pal")) + format = PAL; + else if (!strcasecmp(ap->standard, "secam")) + format = SECAM; + else if (!strcasecmp(ap->standard, "ntsc")) + format = NTSC; + } + + if (bktr_init(video_device, width, height, format, + &(s->video_fd), &(s->tuner_fd), -1, 0.0) < 0) + return -EIO; + + nsignals = 0; + last_frame_time = 0; + + return 0; +} + +static int grab_read_close(AVFormatContext *s1) +{ + VideoData *s = s1->priv_data; + int c; + + c = METEOR_CAP_STOP_CONT; + ioctl(s->video_fd, METEORCAPTUR, &c); + close(s->video_fd); + + c = AUDIO_MUTE; + ioctl(s->tuner_fd, BT848_SAUDIO, &c); + close(s->tuner_fd); + + munmap((caddr_t)video_buf, video_buf_size); + + return 0; +} + +AVInputFormat video_grab_device_format = { + "bktr", + "video grab", + sizeof(VideoData), + NULL, + grab_read_header, + grab_read_packet, + grab_read_close, + .flags = AVFMT_NOFILE, +}; + +int video_grab_init(void) +{ + av_register_input_format(&video_grab_device_format); + return 0; +}