Commit | Line | Data |
---|---|---|
6beefa40 MN |
1 | /* |
2 | * *BSD video grab interface | |
3 | * Copyright (c) 2002 Steve O'Hara-Smith | |
4 | * based on | |
5 | * Linux video grab interface | |
6 | * Copyright (c) 2000,2001 Gerard Lantau. | |
7 | * and | |
8 | * simple_grab.c Copyright (c) 1999 Roger Hardiman | |
9 | * | |
b78e7197 DB |
10 | * This file is part of FFmpeg. |
11 | * | |
12 | * FFmpeg is free software; you can redistribute it and/or | |
6beefa40 MN |
13 | * modify it under the terms of the GNU Lesser General Public |
14 | * License as published by the Free Software Foundation; either | |
b78e7197 | 15 | * version 2.1 of the License, or (at your option) any later version. |
6beefa40 | 16 | * |
b78e7197 | 17 | * FFmpeg is distributed in the hope that it will be useful, |
6beefa40 MN |
18 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
19 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
20 | * Lesser General Public License for more details. | |
21 | * | |
22 | * You should have received a copy of the GNU Lesser General Public | |
b78e7197 | 23 | * License along with FFmpeg; if not, write to the Free Software |
5509bffa | 24 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
6beefa40 MN |
25 | */ |
26 | #include "avformat.h" | |
f8aa696f | 27 | #if defined (HAVE_DEV_BKTR_IOCTL_METEOR_H) && defined (HAVE_DEV_BKTR_IOCTL_BT848_H) |
38f0d3ce DB |
28 | # include <dev/bktr/ioctl_meteor.h> |
29 | # include <dev/bktr/ioctl_bt848.h> | |
f8aa696f DB |
30 | #elif defined (HAVE_MACHINE_IOCTL_METEOR_H) && defined (HAVE_MACHINE_IOCTL_BT848_H) |
31 | # include <machine/ioctl_meteor.h> | |
32 | # include <machine/ioctl_bt848.h> | |
33 | #elif defined (HAVE_DEV_VIDEO_METEOR_IOCTL_METEOR_H) && defined (HAVE_DEV_VIDEO_METEOR_IOCTL_BT848_H) | |
6beefa40 MN |
34 | # include <dev/video/meteor/ioctl_meteor.h> |
35 | # include <dev/video/bktr/ioctl_bt848.h> | |
f8aa696f | 36 | #elif HAVE_DEV_IC_BT8XX_H |
6beefa40 MN |
37 | # include <dev/ic/bt8xx.h> |
38 | #endif | |
39 | #include <unistd.h> | |
40 | #include <fcntl.h> | |
41 | #include <sys/ioctl.h> | |
42 | #include <sys/mman.h> | |
43 | #include <sys/time.h> | |
44 | #include <signal.h> | |
45 | ||
46 | typedef struct { | |
47 | int video_fd; | |
48 | int tuner_fd; | |
49 | int width, height; | |
50 | int frame_rate; | |
51 | int frame_rate_base; | |
52 | u_int64_t per_frame; | |
53 | } VideoData; | |
54 | ||
55 | ||
56 | #define PAL 1 | |
57 | #define PALBDGHI 1 | |
58 | #define NTSC 2 | |
59 | #define NTSCM 2 | |
60 | #define SECAM 3 | |
61 | #define PALN 4 | |
62 | #define PALM 5 | |
63 | #define NTSCJ 6 | |
64 | ||
65 | /* PAL is 768 x 576. NTSC is 640 x 480 */ | |
66 | #define PAL_HEIGHT 576 | |
67 | #define SECAM_HEIGHT 576 | |
68 | #define NTSC_HEIGHT 480 | |
69 | ||
70 | #ifndef VIDEO_FORMAT | |
71 | #define VIDEO_FORMAT NTSC | |
72 | #endif | |
73 | ||
74 | static int bktr_dev[] = { METEOR_DEV0, METEOR_DEV1, METEOR_DEV2, | |
75 | METEOR_DEV3, METEOR_DEV_SVIDEO }; | |
76 | ||
77 | uint8_t *video_buf; | |
78 | size_t video_buf_size; | |
79 | u_int64_t last_frame_time; | |
80 | volatile sig_atomic_t nsignals; | |
81 | ||
82 | ||
83 | static void catchsignal(int signal) | |
84 | { | |
85 | nsignals++; | |
86 | return; | |
87 | } | |
88 | ||
89 | static int bktr_init(const char *video_device, int width, int height, | |
90 | int format, int *video_fd, int *tuner_fd, int idev, double frequency) | |
91 | { | |
92 | struct meteor_geomet geo; | |
93 | int h_max; | |
94 | long ioctl_frequency; | |
95 | char *arg; | |
96 | int c; | |
97 | struct sigaction act, old; | |
98 | ||
99 | if (idev < 0 || idev > 4) | |
100 | { | |
101 | arg = getenv ("BKTR_DEV"); | |
102 | if (arg) | |
103 | idev = atoi (arg); | |
104 | if (idev < 0 || idev > 4) | |
105 | idev = 1; | |
106 | } | |
107 | ||
108 | if (format < 1 || format > 6) | |
109 | { | |
110 | arg = getenv ("BKTR_FORMAT"); | |
111 | if (arg) | |
112 | format = atoi (arg); | |
113 | if (format < 1 || format > 6) | |
114 | format = VIDEO_FORMAT; | |
115 | } | |
116 | ||
117 | if (frequency <= 0) | |
118 | { | |
119 | arg = getenv ("BKTR_FREQUENCY"); | |
120 | if (arg) | |
121 | frequency = atof (arg); | |
122 | if (frequency <= 0) | |
123 | frequency = 0.0; | |
124 | } | |
125 | ||
126 | memset(&act, 0, sizeof(act)); | |
127 | sigemptyset(&act.sa_mask); | |
128 | act.sa_handler = catchsignal; | |
129 | sigaction(SIGUSR1, &act, &old); | |
130 | ||
131 | *tuner_fd = open("/dev/tuner0", O_RDONLY); | |
132 | if (*tuner_fd < 0) | |
95206290 | 133 | av_log(NULL, AV_LOG_ERROR, "Warning. Tuner not opened, continuing: %s\n", strerror(errno)); |
6beefa40 MN |
134 | |
135 | *video_fd = open(video_device, O_RDONLY); | |
136 | if (*video_fd < 0) { | |
95206290 | 137 | av_log(NULL, AV_LOG_ERROR, "%s: %s\n", video_device, strerror(errno)); |
6beefa40 MN |
138 | return -1; |
139 | } | |
140 | ||
141 | geo.rows = height; | |
142 | geo.columns = width; | |
143 | geo.frames = 1; | |
144 | geo.oformat = METEOR_GEO_YUV_422 | METEOR_GEO_YUV_12; | |
145 | ||
146 | switch (format) { | |
147 | case PAL: h_max = PAL_HEIGHT; c = BT848_IFORM_F_PALBDGHI; break; | |
148 | case PALN: h_max = PAL_HEIGHT; c = BT848_IFORM_F_PALN; break; | |
149 | case PALM: h_max = PAL_HEIGHT; c = BT848_IFORM_F_PALM; break; | |
150 | case SECAM: h_max = SECAM_HEIGHT; c = BT848_IFORM_F_SECAM; break; | |
151 | case NTSC: h_max = NTSC_HEIGHT; c = BT848_IFORM_F_NTSCM; break; | |
152 | case NTSCJ: h_max = NTSC_HEIGHT; c = BT848_IFORM_F_NTSCJ; break; | |
153 | default: h_max = PAL_HEIGHT; c = BT848_IFORM_F_PALBDGHI; break; | |
154 | } | |
155 | ||
156 | if (height <= h_max / 2) | |
157 | geo.oformat |= METEOR_GEO_EVEN_ONLY; | |
158 | ||
159 | if (ioctl(*video_fd, METEORSETGEO, &geo) < 0) { | |
95206290 | 160 | av_log(NULL, AV_LOG_ERROR, "METEORSETGEO: %s\n", strerror(errno)); |
6beefa40 MN |
161 | return -1; |
162 | } | |
163 | ||
164 | if (ioctl(*video_fd, BT848SFMT, &c) < 0) { | |
95206290 | 165 | av_log(NULL, AV_LOG_ERROR, "BT848SFMT: %s\n", strerror(errno)); |
6beefa40 MN |
166 | return -1; |
167 | } | |
168 | ||
169 | c = bktr_dev[idev]; | |
170 | if (ioctl(*video_fd, METEORSINPUT, &c) < 0) { | |
95206290 | 171 | av_log(NULL, AV_LOG_ERROR, "METEORSINPUT: %s\n", strerror(errno)); |
6beefa40 MN |
172 | return -1; |
173 | } | |
174 | ||
175 | video_buf_size = width * height * 12 / 8; | |
176 | ||
115329f1 | 177 | video_buf = (uint8_t *)mmap((caddr_t)0, video_buf_size, |
6beefa40 MN |
178 | PROT_READ, MAP_SHARED, *video_fd, (off_t)0); |
179 | if (video_buf == MAP_FAILED) { | |
95206290 | 180 | av_log(NULL, AV_LOG_ERROR, "mmap: %s\n", strerror(errno)); |
6beefa40 MN |
181 | return -1; |
182 | } | |
183 | ||
184 | if (frequency != 0.0) { | |
115329f1 | 185 | ioctl_frequency = (unsigned long)(frequency*16); |
6beefa40 | 186 | if (ioctl(*tuner_fd, TVTUNER_SETFREQ, &ioctl_frequency) < 0) |
95206290 | 187 | av_log(NULL, AV_LOG_ERROR, "TVTUNER_SETFREQ: %s\n", strerror(errno)); |
6beefa40 MN |
188 | } |
189 | ||
190 | c = AUDIO_UNMUTE; | |
191 | if (ioctl(*tuner_fd, BT848_SAUDIO, &c) < 0) | |
95206290 | 192 | av_log(NULL, AV_LOG_ERROR, "TVTUNER_SAUDIO: %s\n", strerror(errno)); |
6beefa40 MN |
193 | |
194 | c = METEOR_CAP_CONTINOUS; | |
195 | ioctl(*video_fd, METEORCAPTUR, &c); | |
196 | ||
197 | c = SIGUSR1; | |
198 | ioctl(*video_fd, METEORSSIGNAL, &c); | |
199 | ||
200 | return 0; | |
201 | } | |
202 | ||
203 | static void bktr_getframe(u_int64_t per_frame) | |
204 | { | |
205 | u_int64_t curtime; | |
206 | ||
207 | curtime = av_gettime(); | |
208 | if (!last_frame_time | |
209 | || ((last_frame_time + per_frame) > curtime)) { | |
210 | if (!usleep(last_frame_time + per_frame + per_frame / 8 - curtime)) { | |
211 | if (!nsignals) | |
212 | av_log(NULL, AV_LOG_INFO, | |
213 | "SLEPT NO signals - %d microseconds late\n", | |
214 | (int)(av_gettime() - last_frame_time - per_frame)); | |
215 | } | |
216 | } | |
217 | nsignals = 0; | |
218 | last_frame_time = curtime; | |
219 | } | |
220 | ||
221 | ||
222 | /* note: we support only one picture read at a time */ | |
223 | static int grab_read_packet(AVFormatContext *s1, AVPacket *pkt) | |
224 | { | |
225 | VideoData *s = s1->priv_data; | |
226 | ||
227 | if (av_new_packet(pkt, video_buf_size) < 0) | |
8fa36ae0 | 228 | return AVERROR(EIO); |
6beefa40 MN |
229 | |
230 | bktr_getframe(s->per_frame); | |
231 | ||
0a7b514f | 232 | pkt->pts = av_gettime(); |
6beefa40 MN |
233 | memcpy(pkt->data, video_buf, video_buf_size); |
234 | ||
235 | return video_buf_size; | |
236 | } | |
237 | ||
238 | static int grab_read_header(AVFormatContext *s1, AVFormatParameters *ap) | |
239 | { | |
240 | VideoData *s = s1->priv_data; | |
241 | AVStream *st; | |
242 | int width, height; | |
243 | int frame_rate; | |
244 | int frame_rate_base; | |
245 | int format = -1; | |
6beefa40 | 246 | |
c04c3282 | 247 | if (ap->width <= 0 || ap->height <= 0 || ap->time_base.den <= 0) |
6beefa40 MN |
248 | return -1; |
249 | ||
250 | width = ap->width; | |
251 | height = ap->height; | |
252 | frame_rate = ap->time_base.den; | |
253 | frame_rate_base = ap->time_base.num; | |
254 | ||
6beefa40 MN |
255 | st = av_new_stream(s1, 0); |
256 | if (!st) | |
8fa36ae0 | 257 | return AVERROR(ENOMEM); |
0a7b514f | 258 | av_set_pts_info(st, 64, 1, 1000000); /* 64 bits pts in use */ |
6beefa40 MN |
259 | |
260 | s->width = width; | |
261 | s->height = height; | |
262 | s->frame_rate = frame_rate; | |
263 | s->frame_rate_base = frame_rate_base; | |
264 | s->per_frame = ((u_int64_t)1000000 * s->frame_rate_base) / s->frame_rate; | |
265 | ||
9f747cc3 SH |
266 | st->codec->codec_type = CODEC_TYPE_VIDEO; |
267 | st->codec->pix_fmt = PIX_FMT_YUV420P; | |
268 | st->codec->codec_id = CODEC_ID_RAWVIDEO; | |
269 | st->codec->width = width; | |
270 | st->codec->height = height; | |
271 | st->codec->time_base.den = frame_rate; | |
272 | st->codec->time_base.num = frame_rate_base; | |
6beefa40 MN |
273 | |
274 | if (ap->standard) { | |
275 | if (!strcasecmp(ap->standard, "pal")) | |
276 | format = PAL; | |
277 | else if (!strcasecmp(ap->standard, "secam")) | |
278 | format = SECAM; | |
279 | else if (!strcasecmp(ap->standard, "ntsc")) | |
280 | format = NTSC; | |
281 | } | |
282 | ||
cc58300e | 283 | if (bktr_init(s1->filename, width, height, format, |
6beefa40 | 284 | &(s->video_fd), &(s->tuner_fd), -1, 0.0) < 0) |
8fa36ae0 | 285 | return AVERROR(EIO); |
6beefa40 MN |
286 | |
287 | nsignals = 0; | |
288 | last_frame_time = 0; | |
289 | ||
290 | return 0; | |
291 | } | |
292 | ||
293 | static int grab_read_close(AVFormatContext *s1) | |
294 | { | |
295 | VideoData *s = s1->priv_data; | |
296 | int c; | |
297 | ||
298 | c = METEOR_CAP_STOP_CONT; | |
299 | ioctl(s->video_fd, METEORCAPTUR, &c); | |
300 | close(s->video_fd); | |
301 | ||
302 | c = AUDIO_MUTE; | |
303 | ioctl(s->tuner_fd, BT848_SAUDIO, &c); | |
304 | close(s->tuner_fd); | |
305 | ||
306 | munmap((caddr_t)video_buf, video_buf_size); | |
307 | ||
308 | return 0; | |
309 | } | |
310 | ||
fd484da7 | 311 | AVInputFormat bktr_demuxer = { |
6beefa40 MN |
312 | "bktr", |
313 | "video grab", | |
3d18b282 DB |
314 | sizeof(VideoData), |
315 | NULL, | |
6beefa40 MN |
316 | grab_read_header, |
317 | grab_read_packet, | |
318 | grab_read_close, | |
319 | .flags = AVFMT_NOFILE, | |
320 | }; |