avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
[libav.git] / libavformat / dv1394.c
CommitLineData
8aa3ee32
MK
1/*
2 * Linux DV1394 interface
3 * Copyright (c) 2003 Max Krasnyansky <maxk@qualcomm.com>
4 *
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
9 *
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 */
19
20#include <unistd.h>
21#include <fcntl.h>
6fa5a56c 22#include <errno.h>
8aa3ee32
MK
23#include <sys/ioctl.h>
24#include <sys/mman.h>
25#include <sys/poll.h>
26#include <sys/time.h>
27#include <time.h>
28
29#include "avformat.h"
30
31#undef DV1394_DEBUG
32
33#include "dv1394.h"
34
8aa3ee32
MK
35struct dv1394_data {
36 int fd;
37 int channel;
38 int width, height;
39 int frame_rate;
40 int frame_size;
1501987b 41 int format;
8aa3ee32
MK
42
43 void *ring; /* Ring buffer */
44 int index; /* Current frame index */
45 int avail; /* Number of frames available for reading */
46 int done; /* Number of completed frames */
1501987b
FB
47
48 int stream; /* Current stream. 0 - video, 1 - audio */
0c1a9eda 49 int64_t pts; /* Current timestamp */
1fbe1a61 50 AVStream *vst, *ast;
8aa3ee32
MK
51};
52
1fbe1a61
RS
53/*
54 * The trick here is to kludge around well known problem with kernel Ooopsing
55 * when you try to capture PAL on a device node configure for NTSC. That's
56 * why we have to configure the device node for PAL, and then read only NTSC
57 * amount of data.
58 */
8aa3ee32
MK
59static int dv1394_reset(struct dv1394_data *dv)
60{
61 struct dv1394_init init;
62
1501987b 63 init.channel = dv->channel;
8aa3ee32 64 init.api_version = DV1394_API_VERSION;
1501987b 65 init.n_frames = DV1394_RING_FRAMES;
1fbe1a61 66 init.format = DV1394_PAL;
8aa3ee32
MK
67
68 if (ioctl(dv->fd, DV1394_INIT, &init) < 0)
69 return -1;
70
6fa5a56c
FB
71 dv->avail = dv->done = 0;
72 dv->stream = 0;
8aa3ee32
MK
73 return 0;
74}
75
76static int dv1394_start(struct dv1394_data *dv)
77{
78 /* Tell DV1394 driver to enable receiver */
79 if (ioctl(dv->fd, DV1394_START_RECEIVE, 0) < 0) {
80 perror("Failed to start receiver");
81 return -1;
82 }
83 return 0;
84}
85
86static int dv1394_read_header(AVFormatContext * context, AVFormatParameters * ap)
87{
88 struct dv1394_data *dv = context->priv_data;
7f172339 89 const char *video_device;
8aa3ee32 90
1fbe1a61
RS
91 dv->vst = av_new_stream(context, 0);
92 if (!dv->vst)
8aa3ee32 93 return -ENOMEM;
1fbe1a61
RS
94 dv->ast = av_new_stream(context, 1);
95 if (!dv->ast) {
96 av_free(dv->vst);
1501987b
FB
97 return -ENOMEM;
98 }
8aa3ee32 99
e3ee3283
AB
100 if (ap->standard && !strcasecmp(ap->standard, "pal"))
101 dv->format = DV1394_PAL;
102 else
103 dv->format = DV1394_NTSC;
a5df11ab
FB
104
105 if (ap->channel)
106 dv->channel = ap->channel;
107 else
108 dv->channel = DV1394_DEFAULT_CHANNEL;
8aa3ee32 109
cad6f6cd 110 dv->width = DV1394_WIDTH;
1501987b 111 if (dv->format == DV1394_NTSC) {
cad6f6cd 112 dv->height = DV1394_NTSC_HEIGHT;
1501987b
FB
113 dv->frame_size = DV1394_NTSC_FRAME_SIZE;
114 dv->frame_rate = 30;
115 } else {
cad6f6cd 116 dv->height = DV1394_PAL_HEIGHT;
1501987b
FB
117 dv->frame_size = DV1394_PAL_FRAME_SIZE;
118 dv->frame_rate = 25;
119 }
8aa3ee32
MK
120
121 /* Open and initialize DV1394 device */
7f172339
FB
122 video_device = ap->device;
123 if (!video_device)
124 video_device = "/dev/dv1394/0";
8aa3ee32
MK
125 dv->fd = open(video_device, O_RDONLY);
126 if (dv->fd < 0) {
127 perror("Failed to open DV interface");
128 goto failed;
129 }
130
131 if (dv1394_reset(dv) < 0) {
132 perror("Failed to initialize DV interface");
133 goto failed;
134 }
135
1fbe1a61 136 dv->ring = mmap(NULL, DV1394_PAL_FRAME_SIZE * DV1394_RING_FRAMES,
8aa3ee32 137 PROT_READ, MAP_PRIVATE, dv->fd, 0);
1fbe1a61 138 if (dv->ring == MAP_FAILED) {
8aa3ee32
MK
139 perror("Failed to mmap DV ring buffer");
140 goto failed;
141 }
142
1501987b
FB
143 dv->stream = 0;
144
1fbe1a61
RS
145 dv->vst->codec.codec_type = CODEC_TYPE_VIDEO;
146 dv->vst->codec.codec_id = CODEC_ID_DVVIDEO;
147 dv->vst->codec.width = dv->width;
148 dv->vst->codec.height = dv->height;
149 dv->vst->codec.frame_rate = dv->frame_rate;
150 dv->vst->codec.frame_rate_base = 1;
151 dv->vst->codec.bit_rate = 25000000; /* Consumer DV is 25Mbps */
8aa3ee32 152
1fbe1a61
RS
153 dv->ast->codec.codec_type = CODEC_TYPE_AUDIO;
154 dv->ast->codec.codec_id = CODEC_ID_DVAUDIO;
155 dv->ast->codec.channels = 2;
156 dv->ast->codec.sample_rate= 48000;
8aa3ee32
MK
157
158 av_set_pts_info(context, 48, 1, 1000000);
159
160 if (dv1394_start(dv) < 0)
161 goto failed;
162
163 return 0;
164
165failed:
166 close(dv->fd);
1fbe1a61
RS
167 av_free(dv->vst);
168 av_free(dv->ast);
8aa3ee32
MK
169 return -EIO;
170}
171
6fa5a56c
FB
172static void __destruct_pkt(struct AVPacket *pkt)
173{
174 pkt->data = NULL; pkt->size = 0;
175 return;
176}
177
178static inline int __get_frame(struct dv1394_data *dv, AVPacket *pkt)
8aa3ee32 179{
1fbe1a61 180 char *ptr = dv->ring + (dv->index * DV1394_PAL_FRAME_SIZE);
8aa3ee32 181
1501987b
FB
182 if (dv->stream) {
183 dv->index = (dv->index + 1) % DV1394_RING_FRAMES;
184 dv->done++; dv->avail--;
185 } else {
6fa5a56c 186 dv->pts = av_gettime() & ((1LL << 48) - 1);
1501987b
FB
187 }
188
1fbe1a61
RS
189 dv->format = ((ptr[3] & 0x80) == 0) ? DV1394_NTSC : DV1394_PAL;
190 if (dv->format == DV1394_NTSC) {
191 dv->frame_size = DV1394_NTSC_FRAME_SIZE;
192 dv->vst->codec.height = dv->height = DV1394_NTSC_HEIGHT;
193 dv->vst->codec.frame_rate = dv->frame_rate = 30;
194 } else {
195 dv->frame_size = DV1394_PAL_FRAME_SIZE;
196 dv->vst->codec.height = dv->height = DV1394_PAL_HEIGHT;
197 dv->vst->codec.frame_rate = dv->frame_rate = 25;
198 }
199
6fa5a56c
FB
200 av_init_packet(pkt);
201 pkt->destruct = __destruct_pkt;
202 pkt->data = ptr;
203 pkt->size = dv->frame_size;
204 pkt->pts = dv->pts;
1501987b 205 pkt->stream_index = dv->stream;
e738cee9 206 pkt->flags |= PKT_FLAG_KEY;
8aa3ee32 207
1501987b 208 dv->stream ^= 1;
8aa3ee32
MK
209
210 return dv->frame_size;
211}
212
1501987b 213static int dv1394_read_packet(AVFormatContext *context, AVPacket *pkt)
8aa3ee32
MK
214{
215 struct dv1394_data *dv = context->priv_data;
8aa3ee32
MK
216
217 if (!dv->avail) {
218 struct dv1394_status s;
219 struct pollfd p;
6fa5a56c
FB
220
221 if (dv->done) {
222 /* Request more frames */
223 if (ioctl(dv->fd, DV1394_RECEIVE_FRAMES, dv->done) < 0) {
224 /* This usually means that ring buffer overflowed.
225 * We have to reset :(.
226 */
227
228 fprintf(stderr, "DV1394: Ring buffer overflow. Reseting ..\n");
229
230 dv1394_reset(dv);
231 dv1394_start(dv);
232 }
233 dv->done = 0;
234 }
8aa3ee32
MK
235
236 /* Wait until more frames are available */
6fa5a56c
FB
237restart_poll:
238 p.fd = dv->fd;
239 p.events = POLLIN | POLLERR | POLLHUP;
8aa3ee32 240 if (poll(&p, 1, -1) < 0) {
6fa5a56c
FB
241 if (errno == EAGAIN || errno == EINTR)
242 goto restart_poll;
8aa3ee32
MK
243 perror("Poll failed");
244 return -EIO;
245 }
246
247 if (ioctl(dv->fd, DV1394_GET_STATUS, &s) < 0) {
248 perror("Failed to get status");
249 return -EIO;
250 }
251#ifdef DV1394_DEBUG
252 fprintf(stderr, "DV1394: status\n"
253 "\tactive_frame\t%d\n"
254 "\tfirst_clear_frame\t%d\n"
255 "\tn_clear_frames\t%d\n"
256 "\tdropped_frames\t%d\n",
257 s.active_frame, s.first_clear_frame,
258 s.n_clear_frames, s.dropped_frames);
259#endif
260
261 dv->avail = s.n_clear_frames;
262 dv->index = s.first_clear_frame;
6fa5a56c 263 dv->done = 0;
8aa3ee32
MK
264
265 if (s.dropped_frames) {
266 fprintf(stderr, "DV1394: Frame drop detected (%d). Reseting ..\n",
267 s.dropped_frames);
268
269 dv1394_reset(dv);
270 dv1394_start(dv);
271 }
272 }
273
8aa3ee32
MK
274#ifdef DV1394_DEBUG
275 fprintf(stderr, "index %d, avail %d, done %d\n", dv->index, dv->avail,
276 dv->done);
277#endif
278
6fa5a56c 279 return __get_frame(dv, pkt);
8aa3ee32
MK
280}
281
282static int dv1394_close(AVFormatContext * context)
283{
284 struct dv1394_data *dv = context->priv_data;
285
286 /* Shutdown DV1394 receiver */
287 if (ioctl(dv->fd, DV1394_SHUTDOWN, 0) < 0)
288 perror("Failed to shutdown DV1394");
289
290 /* Unmap ring buffer */
291 if (munmap(dv->ring, DV1394_NTSC_FRAME_SIZE * DV1394_RING_FRAMES) < 0)
292 perror("Failed to munmap DV1394 ring buffer");
293
294 close(dv->fd);
295
296 return 0;
297}
298
299static AVInputFormat dv1394_format = {
300 .name = "dv1394",
301 .long_name = "dv1394 A/V grab",
302 .priv_data_size = sizeof(struct dv1394_data),
303 .read_header = dv1394_read_header,
304 .read_packet = dv1394_read_packet,
305 .read_close = dv1394_close,
306 .flags = AVFMT_NOFILE
307};
308
309int dv1394_init(void)
310{
311 av_register_input_format(&dv1394_format);
312 return 0;
313}