Commit | Line | Data |
---|---|---|
de6d9b64 FB |
1 | /* |
2 | * Linux audio play and grab interface | |
19720f15 | 3 | * Copyright (c) 2000, 2001 Fabrice Bellard. |
de6d9b64 | 4 | * |
19720f15 FB |
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. | |
de6d9b64 | 9 | * |
19720f15 | 10 | * This library is distributed in the hope that it will be useful, |
de6d9b64 | 11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
19720f15 FB |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
13 | * Lesser General Public License for more details. | |
de6d9b64 | 14 | * |
19720f15 FB |
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 | |
5509bffa | 17 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
de6d9b64 | 18 | */ |
8be1c656 FB |
19 | #include "avformat.h" |
20 | ||
de6d9b64 FB |
21 | #include <stdlib.h> |
22 | #include <stdio.h> | |
23 | #include <string.h> | |
8c802695 JM |
24 | #ifdef __OpenBSD__ |
25 | #include <soundcard.h> | |
26 | #else | |
b13788c5 | 27 | #include <sys/soundcard.h> |
8c802695 | 28 | #endif |
de6d9b64 FB |
29 | #include <unistd.h> |
30 | #include <fcntl.h> | |
31 | #include <sys/ioctl.h> | |
32 | #include <sys/mman.h> | |
de6d9b64 FB |
33 | #include <sys/time.h> |
34 | ||
4972b26f FB |
35 | #define AUDIO_BLOCK_SIZE 4096 |
36 | ||
de6d9b64 FB |
37 | typedef struct { |
38 | int fd; | |
4972b26f | 39 | int sample_rate; |
de6d9b64 | 40 | int channels; |
4972b26f FB |
41 | int frame_size; /* in bytes ! */ |
42 | int codec_id; | |
1de1cce2 | 43 | int flip_left : 1; |
0c1a9eda | 44 | uint8_t buffer[AUDIO_BLOCK_SIZE]; |
4972b26f | 45 | int buffer_ptr; |
de6d9b64 FB |
46 | } AudioData; |
47 | ||
7f172339 | 48 | static int audio_open(AudioData *s, int is_output, const char *audio_device) |
de6d9b64 | 49 | { |
4972b26f | 50 | int audio_fd; |
de6d9b64 | 51 | int tmp, err; |
1de1cce2 | 52 | char *flip = getenv("AUDIO_FLIP_LEFT"); |
de6d9b64 | 53 | |
de6d9b64 | 54 | /* open linux audio device */ |
7f172339 | 55 | if (!audio_device) |
8c802695 | 56 | #ifdef __OpenBSD__ |
bb270c08 | 57 | audio_device = "/dev/sound"; |
8c802695 | 58 | #else |
7f172339 | 59 | audio_device = "/dev/dsp"; |
8c802695 | 60 | #endif |
7f172339 | 61 | |
4972b26f FB |
62 | if (is_output) |
63 | audio_fd = open(audio_device, O_WRONLY); | |
de6d9b64 | 64 | else |
4972b26f | 65 | audio_fd = open(audio_device, O_RDONLY); |
de6d9b64 FB |
66 | if (audio_fd < 0) { |
67 | perror(audio_device); | |
0bd586c5 | 68 | return AVERROR_IO; |
de6d9b64 FB |
69 | } |
70 | ||
1de1cce2 PG |
71 | if (flip && *flip == '1') { |
72 | s->flip_left = 1; | |
73 | } | |
74 | ||
de6d9b64 | 75 | /* non blocking mode */ |
4364a3e0 FB |
76 | if (!is_output) |
77 | fcntl(audio_fd, F_SETFL, O_NONBLOCK); | |
de6d9b64 | 78 | |
4972b26f | 79 | s->frame_size = AUDIO_BLOCK_SIZE; |
de6d9b64 | 80 | #if 0 |
4972b26f FB |
81 | tmp = (NB_FRAGMENTS << 16) | FRAGMENT_BITS; |
82 | err = ioctl(audio_fd, SNDCTL_DSP_SETFRAGMENT, &tmp); | |
de6d9b64 FB |
83 | if (err < 0) { |
84 | perror("SNDCTL_DSP_SETFRAGMENT"); | |
85 | } | |
86 | #endif | |
87 | ||
4972b26f FB |
88 | /* select format : favour native format */ |
89 | err = ioctl(audio_fd, SNDCTL_DSP_GETFMTS, &tmp); | |
115329f1 | 90 | |
4972b26f FB |
91 | #ifdef WORDS_BIGENDIAN |
92 | if (tmp & AFMT_S16_BE) { | |
93 | tmp = AFMT_S16_BE; | |
94 | } else if (tmp & AFMT_S16_LE) { | |
95 | tmp = AFMT_S16_LE; | |
96 | } else { | |
97 | tmp = 0; | |
98 | } | |
99 | #else | |
100 | if (tmp & AFMT_S16_LE) { | |
101 | tmp = AFMT_S16_LE; | |
102 | } else if (tmp & AFMT_S16_BE) { | |
103 | tmp = AFMT_S16_BE; | |
104 | } else { | |
105 | tmp = 0; | |
106 | } | |
107 | #endif | |
108 | ||
109 | switch(tmp) { | |
110 | case AFMT_S16_LE: | |
111 | s->codec_id = CODEC_ID_PCM_S16LE; | |
112 | break; | |
113 | case AFMT_S16_BE: | |
114 | s->codec_id = CODEC_ID_PCM_S16BE; | |
115 | break; | |
116 | default: | |
bc874dae | 117 | av_log(NULL, AV_LOG_ERROR, "Soundcard does not support 16 bit sample format\n"); |
4972b26f | 118 | close(audio_fd); |
0bd586c5 | 119 | return AVERROR_IO; |
4972b26f FB |
120 | } |
121 | err=ioctl(audio_fd, SNDCTL_DSP_SETFMT, &tmp); | |
de6d9b64 FB |
122 | if (err < 0) { |
123 | perror("SNDCTL_DSP_SETFMT"); | |
124 | goto fail; | |
125 | } | |
115329f1 | 126 | |
4972b26f FB |
127 | tmp = (s->channels == 2); |
128 | err = ioctl(audio_fd, SNDCTL_DSP_STEREO, &tmp); | |
de6d9b64 FB |
129 | if (err < 0) { |
130 | perror("SNDCTL_DSP_STEREO"); | |
131 | goto fail; | |
132 | } | |
1de1cce2 PG |
133 | if (tmp) |
134 | s->channels = 2; | |
115329f1 | 135 | |
4972b26f FB |
136 | tmp = s->sample_rate; |
137 | err = ioctl(audio_fd, SNDCTL_DSP_SPEED, &tmp); | |
de6d9b64 FB |
138 | if (err < 0) { |
139 | perror("SNDCTL_DSP_SPEED"); | |
140 | goto fail; | |
141 | } | |
4972b26f | 142 | s->sample_rate = tmp; /* store real sample rate */ |
de6d9b64 FB |
143 | s->fd = audio_fd; |
144 | ||
145 | return 0; | |
146 | fail: | |
147 | close(audio_fd); | |
0bd586c5 | 148 | return AVERROR_IO; |
de6d9b64 FB |
149 | } |
150 | ||
4972b26f | 151 | static int audio_close(AudioData *s) |
de6d9b64 | 152 | { |
de6d9b64 | 153 | close(s->fd); |
4972b26f FB |
154 | return 0; |
155 | } | |
156 | ||
157 | /* sound output support */ | |
158 | static int audio_write_header(AVFormatContext *s1) | |
159 | { | |
c9a65ca8 | 160 | AudioData *s = s1->priv_data; |
4972b26f FB |
161 | AVStream *st; |
162 | int ret; | |
163 | ||
4972b26f | 164 | st = s1->streams[0]; |
01f4895c MN |
165 | s->sample_rate = st->codec->sample_rate; |
166 | s->channels = st->codec->channels; | |
7f172339 | 167 | ret = audio_open(s, 1, NULL); |
4972b26f | 168 | if (ret < 0) { |
0bd586c5 | 169 | return AVERROR_IO; |
4972b26f FB |
170 | } else { |
171 | return 0; | |
172 | } | |
173 | } | |
174 | ||
e928649b | 175 | static int audio_write_packet(AVFormatContext *s1, AVPacket *pkt) |
4972b26f FB |
176 | { |
177 | AudioData *s = s1->priv_data; | |
178 | int len, ret; | |
e928649b MN |
179 | int size= pkt->size; |
180 | uint8_t *buf= pkt->data; | |
4972b26f FB |
181 | |
182 | while (size > 0) { | |
183 | len = AUDIO_BLOCK_SIZE - s->buffer_ptr; | |
184 | if (len > size) | |
185 | len = size; | |
186 | memcpy(s->buffer + s->buffer_ptr, buf, len); | |
187 | s->buffer_ptr += len; | |
188 | if (s->buffer_ptr >= AUDIO_BLOCK_SIZE) { | |
189 | for(;;) { | |
190 | ret = write(s->fd, s->buffer, AUDIO_BLOCK_SIZE); | |
4364a3e0 | 191 | if (ret > 0) |
4972b26f FB |
192 | break; |
193 | if (ret < 0 && (errno != EAGAIN && errno != EINTR)) | |
0bd586c5 | 194 | return AVERROR_IO; |
4972b26f FB |
195 | } |
196 | s->buffer_ptr = 0; | |
197 | } | |
198 | buf += len; | |
199 | size -= len; | |
200 | } | |
201 | return 0; | |
202 | } | |
203 | ||
204 | static int audio_write_trailer(AVFormatContext *s1) | |
205 | { | |
206 | AudioData *s = s1->priv_data; | |
207 | ||
208 | audio_close(s); | |
de6d9b64 FB |
209 | return 0; |
210 | } | |
211 | ||
4972b26f FB |
212 | /* grab support */ |
213 | ||
214 | static int audio_read_header(AVFormatContext *s1, AVFormatParameters *ap) | |
215 | { | |
c9a65ca8 | 216 | AudioData *s = s1->priv_data; |
4972b26f FB |
217 | AVStream *st; |
218 | int ret; | |
219 | ||
220 | if (!ap || ap->sample_rate <= 0 || ap->channels <= 0) | |
221 | return -1; | |
222 | ||
c9a65ca8 | 223 | st = av_new_stream(s1, 0); |
4972b26f | 224 | if (!st) { |
4972b26f FB |
225 | return -ENOMEM; |
226 | } | |
4972b26f FB |
227 | s->sample_rate = ap->sample_rate; |
228 | s->channels = ap->channels; | |
229 | ||
7f172339 | 230 | ret = audio_open(s, 0, ap->device); |
4972b26f | 231 | if (ret < 0) { |
1ea4f593 | 232 | av_free(st); |
0bd586c5 | 233 | return AVERROR_IO; |
4972b26f | 234 | } |
45dd5c69 FB |
235 | |
236 | /* take real parameters */ | |
01f4895c MN |
237 | st->codec->codec_type = CODEC_TYPE_AUDIO; |
238 | st->codec->codec_id = s->codec_id; | |
239 | st->codec->sample_rate = s->sample_rate; | |
240 | st->codec->channels = s->channels; | |
45dd5c69 | 241 | |
0a7b514f | 242 | av_set_pts_info(st, 64, 1, 1000000); /* 64 bits pts in us */ |
45dd5c69 | 243 | return 0; |
4972b26f FB |
244 | } |
245 | ||
246 | static int audio_read_packet(AVFormatContext *s1, AVPacket *pkt) | |
247 | { | |
248 | AudioData *s = s1->priv_data; | |
45dd5c69 FB |
249 | int ret, bdelay; |
250 | int64_t cur_time; | |
251 | struct audio_buf_info abufi; | |
115329f1 | 252 | |
4972b26f | 253 | if (av_new_packet(pkt, s->frame_size) < 0) |
0bd586c5 | 254 | return AVERROR_IO; |
4972b26f | 255 | for(;;) { |
79134973 PG |
256 | struct timeval tv; |
257 | fd_set fds; | |
258 | ||
259 | tv.tv_sec = 0; | |
260 | tv.tv_usec = 30 * 1000; /* 30 msecs -- a bit shorter than 1 frame at 30fps */ | |
261 | ||
262 | FD_ZERO(&fds); | |
263 | FD_SET(s->fd, &fds); | |
264 | ||
265 | /* This will block until data is available or we get a timeout */ | |
266 | (void) select(s->fd + 1, &fds, 0, 0, &tv); | |
267 | ||
4972b26f FB |
268 | ret = read(s->fd, pkt->data, pkt->size); |
269 | if (ret > 0) | |
270 | break; | |
4606ac8d ZK |
271 | if (ret == -1 && (errno == EAGAIN || errno == EINTR)) { |
272 | av_free_packet(pkt); | |
273 | pkt->size = 0; | |
0a7b514f | 274 | pkt->pts = av_gettime(); |
4606ac8d ZK |
275 | return 0; |
276 | } | |
4972b26f FB |
277 | if (!(ret == 0 || (ret == -1 && (errno == EAGAIN || errno == EINTR)))) { |
278 | av_free_packet(pkt); | |
0bd586c5 | 279 | return AVERROR_IO; |
4972b26f FB |
280 | } |
281 | } | |
282 | pkt->size = ret; | |
45dd5c69 FB |
283 | |
284 | /* compute pts of the start of the packet */ | |
285 | cur_time = av_gettime(); | |
286 | bdelay = ret; | |
287 | if (ioctl(s->fd, SNDCTL_DSP_GETISPACE, &abufi) == 0) { | |
288 | bdelay += abufi.bytes; | |
289 | } | |
290 | /* substract time represented by the number of bytes in the audio fifo */ | |
291 | cur_time -= (bdelay * 1000000LL) / (s->sample_rate * s->channels); | |
292 | ||
293 | /* convert to wanted units */ | |
0a7b514f | 294 | pkt->pts = cur_time; |
45dd5c69 | 295 | |
1de1cce2 PG |
296 | if (s->flip_left && s->channels == 2) { |
297 | int i; | |
298 | short *p = (short *) pkt->data; | |
299 | ||
300 | for (i = 0; i < ret; i += 4) { | |
301 | *p = ~*p; | |
302 | p += 2; | |
303 | } | |
304 | } | |
4972b26f FB |
305 | return 0; |
306 | } | |
307 | ||
308 | static int audio_read_close(AVFormatContext *s1) | |
309 | { | |
310 | AudioData *s = s1->priv_data; | |
311 | ||
312 | audio_close(s); | |
4972b26f FB |
313 | return 0; |
314 | } | |
315 | ||
c18a2692 | 316 | static AVInputFormat audio_in_format = { |
c9a65ca8 FB |
317 | "audio_device", |
318 | "audio grab and output", | |
319 | sizeof(AudioData), | |
320 | NULL, | |
321 | audio_read_header, | |
322 | audio_read_packet, | |
323 | audio_read_close, | |
bb76a117 | 324 | .flags = AVFMT_NOFILE, |
c9a65ca8 FB |
325 | }; |
326 | ||
c18a2692 | 327 | static AVOutputFormat audio_out_format = { |
4972b26f FB |
328 | "audio_device", |
329 | "audio grab and output", | |
330 | "", | |
331 | "", | |
c9a65ca8 | 332 | sizeof(AudioData), |
4972b26f FB |
333 | /* XXX: we make the assumption that the soundcard accepts this format */ |
334 | /* XXX: find better solution with "preinit" method, needed also in | |
335 | other formats */ | |
336 | #ifdef WORDS_BIGENDIAN | |
337 | CODEC_ID_PCM_S16BE, | |
338 | #else | |
339 | CODEC_ID_PCM_S16LE, | |
340 | #endif | |
341 | CODEC_ID_NONE, | |
342 | audio_write_header, | |
343 | audio_write_packet, | |
344 | audio_write_trailer, | |
bb76a117 | 345 | .flags = AVFMT_NOFILE, |
de6d9b64 | 346 | }; |
c9a65ca8 FB |
347 | |
348 | int audio_init(void) | |
349 | { | |
350 | av_register_input_format(&audio_in_format); | |
351 | av_register_output_format(&audio_out_format); | |
352 | return 0; | |
353 | } |