?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Fix vf_tcdump's compilation
[mplayer/kovensky.git] / libmpdemux / demuxer.c
blobdf1dea62e7330d964215094f348265bb4c996a7d
1 /*
2 * DEMUXER v2.5
3 *
4 * This file is part of MPlayer.
5 *
6 * MPlayer is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * MPlayer is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along
17 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 */
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <unistd.h>
25
26 #include <sys/types.h>
27 #include <sys/stat.h>
28
29 #include "config.h"
30 #include "options.h"
31 #include "talloc.h"
32 #include "mp_msg.h"
33 #include "m_config.h"
34
35 #include "libvo/fastmemcpy.h"
36
37 #include "stream/stream.h"
38 #include "demuxer.h"
39 #include "stheader.h"
40 #include "mf.h"
41
42 #include "libaf/af_format.h"
43 #include "libmpcodecs/dec_teletext.h"
44
45 #include "ass_mp.h"
46
47 #ifdef CONFIG_LIBAVCODEC
48 #include "libavcodec/avcodec.h"
49 #if MP_INPUT_BUFFER_PADDING_SIZE < FF_INPUT_BUFFER_PADDING_SIZE
50 #error MP_INPUT_BUFFER_PADDING_SIZE is too small!
51 #endif
52 #endif
53
54 // This is quite experimental, in particular it will mess up the pts values
55 // in the queue - on the other hand it might fix some issues like generating
56 // broken files with mencoder and stream copy.
57 // Better leave it disabled for now, if we find no use for it this code should
58 // just be removed again.
59 #define PARSE_ON_ADD 0
60
61 static void clear_parser(sh_common_t *sh);
62 void resync_video_stream(sh_video_t *sh_video);
63 void resync_audio_stream(sh_audio_t *sh_audio);
64
65 // Demuxer list
66 extern const demuxer_desc_t demuxer_desc_rawaudio;
67 extern const demuxer_desc_t demuxer_desc_rawvideo;
68 extern const demuxer_desc_t demuxer_desc_tv;
69 extern const demuxer_desc_t demuxer_desc_mf;
70 extern const demuxer_desc_t demuxer_desc_avi;
71 extern const demuxer_desc_t demuxer_desc_y4m;
72 extern const demuxer_desc_t demuxer_desc_asf;
73 extern const demuxer_desc_t demuxer_desc_real;
74 extern const demuxer_desc_t demuxer_desc_smjpeg;
75 extern const demuxer_desc_t demuxer_desc_matroska;
76 extern const demuxer_desc_t demuxer_desc_realaudio;
77 extern const demuxer_desc_t demuxer_desc_vqf;
78 extern const demuxer_desc_t demuxer_desc_mov;
79 extern const demuxer_desc_t demuxer_desc_vivo;
80 extern const demuxer_desc_t demuxer_desc_fli;
81 extern const demuxer_desc_t demuxer_desc_film;
82 extern const demuxer_desc_t demuxer_desc_roq;
83 extern const demuxer_desc_t demuxer_desc_gif;
84 extern const demuxer_desc_t demuxer_desc_ogg;
85 extern const demuxer_desc_t demuxer_desc_avs;
86 extern const demuxer_desc_t demuxer_desc_pva;
87 extern const demuxer_desc_t demuxer_desc_nsv;
88 extern const demuxer_desc_t demuxer_desc_mpeg_ts;
89 extern const demuxer_desc_t demuxer_desc_lmlm4;
90 extern const demuxer_desc_t demuxer_desc_mpeg_ps;
91 extern const demuxer_desc_t demuxer_desc_mpeg_pes;
92 extern const demuxer_desc_t demuxer_desc_mpeg_es;
93 extern const demuxer_desc_t demuxer_desc_mpeg_gxf;
94 extern const demuxer_desc_t demuxer_desc_mpeg4_es;
95 extern const demuxer_desc_t demuxer_desc_h264_es;
96 extern const demuxer_desc_t demuxer_desc_rawdv;
97 extern const demuxer_desc_t demuxer_desc_mpc;
98 extern const demuxer_desc_t demuxer_desc_audio;
99 extern const demuxer_desc_t demuxer_desc_xmms;
100 extern const demuxer_desc_t demuxer_desc_mpeg_ty;
101 extern const demuxer_desc_t demuxer_desc_rtp;
102 extern const demuxer_desc_t demuxer_desc_rtp_nemesi;
103 extern const demuxer_desc_t demuxer_desc_lavf;
104 extern const demuxer_desc_t demuxer_desc_lavf_preferred;
105 extern const demuxer_desc_t demuxer_desc_aac;
106 extern const demuxer_desc_t demuxer_desc_nut;
107 extern const demuxer_desc_t demuxer_desc_mng;
108
109 /* Please do not add any new demuxers here. If you want to implement a new
110 * demuxer, add it to libavformat, except for wrappers around external
111 * libraries and demuxers requiring binary support. */
112
113 const demuxer_desc_t *const demuxer_list[] = {
114 &demuxer_desc_rawaudio,
115 &demuxer_desc_rawvideo,
116 #ifdef CONFIG_TV
117 &demuxer_desc_tv,
118 #endif
119 &demuxer_desc_mf,
120 #ifdef CONFIG_LIBAVFORMAT
121 &demuxer_desc_lavf_preferred,
122 #endif
123 &demuxer_desc_avi,
124 &demuxer_desc_y4m,
125 &demuxer_desc_asf,
126 &demuxer_desc_nsv,
127 &demuxer_desc_real,
128 &demuxer_desc_smjpeg,
129 &demuxer_desc_matroska,
130 &demuxer_desc_realaudio,
131 &demuxer_desc_vqf,
132 &demuxer_desc_mov,
133 &demuxer_desc_vivo,
134 &demuxer_desc_fli,
135 &demuxer_desc_film,
136 &demuxer_desc_roq,
137 #ifdef CONFIG_GIF
138 &demuxer_desc_gif,
139 #endif
140 #ifdef CONFIG_OGGVORBIS
141 &demuxer_desc_ogg,
142 #endif
143 #ifdef CONFIG_WIN32DLL
144 &demuxer_desc_avs,
145 #endif
146 &demuxer_desc_pva,
147 &demuxer_desc_mpeg_ts,
148 &demuxer_desc_lmlm4,
149 &demuxer_desc_mpeg_ps,
150 &demuxer_desc_mpeg_pes,
151 &demuxer_desc_mpeg_es,
152 &demuxer_desc_mpeg_gxf,
153 &demuxer_desc_mpeg4_es,
154 &demuxer_desc_h264_es,
155 &demuxer_desc_audio,
156 &demuxer_desc_mpeg_ty,
157 #ifdef CONFIG_LIVE555
158 &demuxer_desc_rtp,
159 #endif
160 #ifdef CONFIG_LIBNEMESI
161 &demuxer_desc_rtp_nemesi,
162 #endif
163 #ifdef CONFIG_LIBAVFORMAT
164 &demuxer_desc_lavf,
165 #endif
166 #ifdef CONFIG_MUSEPACK
167 &demuxer_desc_mpc,
168 #endif
169 #ifdef CONFIG_LIBDV095
170 &demuxer_desc_rawdv,
171 #endif
172 &demuxer_desc_aac,
173 #ifdef CONFIG_LIBNUT
174 &demuxer_desc_nut,
175 #endif
176 #ifdef CONFIG_XMMS
177 &demuxer_desc_xmms,
178 #endif
179 #ifdef CONFIG_MNG
180 &demuxer_desc_mng,
181 #endif
182 /* Please do not add any new demuxers here. If you want to implement a new
183 * demuxer, add it to libavformat, except for wrappers around external
184 * libraries and demuxers requiring binary support. */
185 NULL
186 };
187
188 void free_demuxer_stream(demux_stream_t *ds)
189 {
190 ds_free_packs(ds);
191 free(ds);
192 }
193
194 demux_stream_t *new_demuxer_stream(struct demuxer *demuxer, int id)
195 {
196 demux_stream_t *ds = malloc(sizeof(demux_stream_t));
197 *ds = (demux_stream_t){
198 .id = id,
199 .demuxer = demuxer,
200 .asf_seq = -1,
201 };
202 return ds;
203 }
204
205
206 /**
207 * Get demuxer description structure for a given demuxer type
208 *
209 * @param file_format type of the demuxer
210 * @return structure for the demuxer, NULL if not found
211 */
212 static const demuxer_desc_t *get_demuxer_desc_from_type(int file_format)
213 {
214 int i;
215
216 for (i = 0; demuxer_list[i]; i++)
217 if (file_format == demuxer_list[i]->type)
218 return demuxer_list[i];
219
220 return NULL;
221 }
222
223
224 demuxer_t *new_demuxer(struct MPOpts *opts, stream_t *stream, int type,
225 int a_id, int v_id, int s_id, char *filename)
226 {
227 struct demuxer *d = talloc_zero(NULL, struct demuxer);
228 d->stream = stream;
229 d->stream_pts = MP_NOPTS_VALUE;
230 d->reference_clock = MP_NOPTS_VALUE;
231 d->movi_start = stream->start_pos;
232 d->movi_end = stream->end_pos;
233 d->seekable = 1;
234 d->synced = 0;
235 d->filepos = -1;
236 d->audio = new_demuxer_stream(d, a_id);
237 d->video = new_demuxer_stream(d, v_id);
238 d->sub = new_demuxer_stream(d, s_id);
239 d->type = type;
240 d->opts = opts;
241 if (type)
242 if (!(d->desc = get_demuxer_desc_from_type(type)))
243 mp_msg(MSGT_DEMUXER, MSGL_ERR,
244 "BUG! Invalid demuxer type in new_demuxer(), "
245 "big troubles ahead.");
246 if (filename) // Filename hack for avs_check_file
247 d->filename = strdup(filename);
248 stream->eof = 0;
249 stream_seek(stream, stream->start_pos);
250 return d;
251 }
252
253 sh_sub_t *new_sh_sub_sid(demuxer_t *demuxer, int id, int sid)
254 {
255 if (id > MAX_S_STREAMS - 1 || id < 0) {
256 mp_msg(MSGT_DEMUXER, MSGL_WARN,
257 "Requested sub stream id overflow (%d > %d)\n", id,
258 MAX_S_STREAMS);
259 return NULL;
260 }
261 if (demuxer->s_streams[id])
262 mp_msg(MSGT_DEMUXER, MSGL_WARN, "Sub stream %i redefined\n", id);
263 else {
264 sh_sub_t *sh = calloc(1, sizeof(sh_sub_t));
265 demuxer->s_streams[id] = sh;
266 sh->sid = sid;
267 sh->opts = demuxer->opts;
268 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_SUBTITLE_ID=%d\n", sid);
269 }
270 if (sid == demuxer->opts->sub_id) {
271 demuxer->sub->id = id;
272 demuxer->sub->sh = demuxer->s_streams[id];
273 }
274 return demuxer->s_streams[id];
275 }
276
277 static void free_sh_sub(sh_sub_t *sh)
278 {
279 mp_msg(MSGT_DEMUXER, MSGL_DBG2, "DEMUXER: freeing sh_sub at %p\n", sh);
280 free(sh->extradata);
281 #ifdef CONFIG_ASS
282 if (sh->ass_track)
283 ass_free_track(sh->ass_track);
284 #endif
285 free(sh->lang);
286 #ifdef CONFIG_LIBAVCODEC
287 clear_parser((sh_common_t *)sh);
288 #endif
289 free(sh);
290 }
291
292 sh_audio_t *new_sh_audio_aid(demuxer_t *demuxer, int id, int aid)
293 {
294 if (id > MAX_A_STREAMS - 1 || id < 0) {
295 mp_msg(MSGT_DEMUXER, MSGL_WARN,
296 "Requested audio stream id overflow (%d > %d)\n", id,
297 MAX_A_STREAMS);
298 return NULL;
299 }
300 if (demuxer->a_streams[id]) {
301 mp_tmsg(MSGT_DEMUXER, MSGL_WARN, "WARNING: Audio stream header %d redefined.\n", id);
302 } else {
303 mp_tmsg(MSGT_DEMUXER, MSGL_V, "==> Found audio stream: %d\n", id);
304 sh_audio_t *sh = calloc(1, sizeof(sh_audio_t));
305 demuxer->a_streams[id] = sh;
306 sh->aid = aid;
307 sh->ds = demuxer->audio;
308 // set some defaults
309 sh->samplesize = 2;
310 sh->sample_format = AF_FORMAT_S16_NE;
311 sh->audio_out_minsize = 8192; /* default size, maybe not enough for Win32/ACM */
312 sh->pts = MP_NOPTS_VALUE;
313 sh->opts = demuxer->opts;
314 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_AUDIO_ID=%d\n", aid);
315 }
316 return demuxer->a_streams[id];
317 }
318
319 void free_sh_audio(demuxer_t *demuxer, int id)
320 {
321 sh_audio_t *sh = demuxer->a_streams[id];
322 demuxer->a_streams[id] = NULL;
323 mp_msg(MSGT_DEMUXER, MSGL_DBG2, "DEMUXER: freeing sh_audio at %p\n", sh);
324 free(sh->wf);
325 free(sh->codecdata);
326 free(sh->lang);
327 #ifdef CONFIG_LIBAVCODEC
328 clear_parser((sh_common_t *)sh);
329 #endif
330 free(sh);
331 }
332
333 sh_video_t *new_sh_video_vid(demuxer_t *demuxer, int id, int vid)
334 {
335 if (id > MAX_V_STREAMS - 1 || id < 0) {
336 mp_msg(MSGT_DEMUXER, MSGL_WARN,
337 "Requested video stream id overflow (%d > %d)\n", id,
338 MAX_V_STREAMS);
339 return NULL;
340 }
341 if (demuxer->v_streams[id])
342 mp_tmsg(MSGT_DEMUXER, MSGL_WARN, "WARNING: Video stream header %d redefined.\n", id);
343 else {
344 mp_tmsg(MSGT_DEMUXER, MSGL_V, "==> Found video stream: %d\n", id);
345 sh_video_t *sh = calloc(1, sizeof *sh);
346 demuxer->v_streams[id] = sh;
347 sh->vid = vid;
348 sh->ds = demuxer->video;
349 sh->opts = demuxer->opts;
350 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VIDEO_ID=%d\n", vid);
351 }
352 return demuxer->v_streams[id];
353 }
354
355 void free_sh_video(sh_video_t *sh)
356 {
357 mp_msg(MSGT_DEMUXER, MSGL_DBG2, "DEMUXER: freeing sh_video at %p\n", sh);
358 free(sh->bih);
359 #ifdef CONFIG_LIBAVCODEC
360 clear_parser((sh_common_t *)sh);
361 #endif
362 free(sh);
363 }
364
365 void free_demuxer(demuxer_t *demuxer)
366 {
367 int i;
368 mp_msg(MSGT_DEMUXER, MSGL_DBG2, "DEMUXER: freeing %s demuxer at %p\n",
369 demuxer->desc->shortdesc, demuxer);
370 if (demuxer->desc->close)
371 demuxer->desc->close(demuxer);
372 // Very ugly hack to make it behave like old implementation
373 if (demuxer->desc->type == DEMUXER_TYPE_DEMUXERS)
374 goto skip_streamfree;
375 // free streams:
376 for (i = 0; i < MAX_A_STREAMS; i++)
377 if (demuxer->a_streams[i])
378 free_sh_audio(demuxer, i);
379 for (i = 0; i < MAX_V_STREAMS; i++)
380 if (demuxer->v_streams[i])
381 free_sh_video(demuxer->v_streams[i]);
382 for (i = 0; i < MAX_S_STREAMS; i++)
383 if (demuxer->s_streams[i])
384 free_sh_sub(demuxer->s_streams[i]);
385 // free demuxers:
386 free_demuxer_stream(demuxer->audio);
387 free_demuxer_stream(demuxer->video);
388 free_demuxer_stream(demuxer->sub);
389 skip_streamfree:
390 if (demuxer->info) {
391 for (i = 0; demuxer->info[i] != NULL; i++)
392 free(demuxer->info[i]);
393 free(demuxer->info);
394 }
395 free(demuxer->filename);
396 if (demuxer->teletext)
397 teletext_control(demuxer->teletext, TV_VBI_CONTROL_STOP, NULL);
398 talloc_free(demuxer);
399 }
400
401
402 static void ds_add_packet_internal(demux_stream_t *ds, demux_packet_t *dp)
403 {
404 // append packet to DS stream:
405 ++ds->packs;
406 ds->bytes += dp->len;
407 if (ds->last) {
408 // next packet in stream
409 ds->last->next = dp;
410 ds->last = dp;
411 } else {
412 // first packet in stream
413 ds->first = ds->last = dp;
414 }
415 mp_dbg(MSGT_DEMUXER, MSGL_DBG2,
416 "DEMUX: Append packet to %s, len=%d pts=%5.3f pos=%u [packs: A=%d V=%d]\n",
417 (ds == ds->demuxer->audio) ? "d_audio" : "d_video", dp->len,
418 dp->pts, (unsigned int) dp->pos, ds->demuxer->audio->packs,
419 ds->demuxer->video->packs);
420 }
421
422 #ifdef CONFIG_LIBAVCODEC
423 static void allocate_parser(AVCodecContext **avctx, AVCodecParserContext **parser, unsigned format)
424 {
425 enum CodecID codec_id = CODEC_ID_NONE;
426 extern int avcodec_initialized;
427 if (!avcodec_initialized) {
428 avcodec_init();
429 avcodec_register_all();
430 avcodec_initialized = 1;
431 }
432 switch (format) {
433 case 0x2000:
434 case 0x332D6361:
435 case 0x332D4341:
436 case MKTAG('d', 'n', 'e', 't'):
437 case MKTAG('s', 'a', 'c', '3'):
438 codec_id = CODEC_ID_AC3;
439 break;
440 case MKTAG('E', 'A', 'C', '3'):
441 codec_id = CODEC_ID_EAC3;
442 break;
443 case 0x2001:
444 case 0x86:
445 codec_id = CODEC_ID_DTS;
446 break;
447 case MKTAG('M', 'L', 'P', ' '):
448 codec_id = CODEC_ID_MLP;
449 break;
450 case 0x55:
451 case 0x5500736d:
452 case MKTAG('.', 'm', 'p', '3'):
453 case MKTAG('M', 'P', 'E', ' '):
454 case MKTAG('L', 'A', 'M', 'E'):
455 codec_id = CODEC_ID_MP3;
456 break;
457 case 0x50:
458 case MKTAG('.', 'm', 'p', '2'):
459 case MKTAG('.', 'm', 'p', '1'):
460 codec_id = CODEC_ID_MP2;
461 break;
462 case MKTAG('T', 'R', 'H', 'D'):
463 codec_id = CODEC_ID_TRUEHD;
464 break;
465 }
466 if (codec_id != CODEC_ID_NONE) {
467 *avctx = avcodec_alloc_context();
468 if (!*avctx)
469 return;
470 *parser = av_parser_init(codec_id);
471 if (!*parser)
472 av_freep(avctx);
473 }
474 }
475
476 static void get_parser(sh_common_t *sh, AVCodecContext **avctx, AVCodecParserContext **parser)
477 {
478 *avctx = NULL;
479 *parser = NULL;
480
481 if (!sh || !sh->needs_parsing)
482 return;
483
484 *avctx = sh->avctx;
485 *parser = sh->parser;
486 if (*parser)
487 return;
488
489 allocate_parser(avctx, parser, sh->format);
490 sh->avctx = *avctx;
491 sh->parser = *parser;
492 }
493
494 int ds_parse(demux_stream_t *ds, uint8_t **buffer, int *len, double pts, off_t pos)
495 {
496 AVCodecContext *avctx;
497 AVCodecParserContext *parser;
498 get_parser(ds->sh, &avctx, &parser);
499 if (!parser)
500 return *len;
501 return av_parser_parse2(parser, avctx, buffer, len, *buffer, *len, pts, pts, pos);
502 }
503
504 static void clear_parser(sh_common_t *sh)
505 {
506 av_parser_close(sh->parser);
507 sh->parser = NULL;
508 av_freep(&sh->avctx);
509 }
510
511 void ds_clear_parser(demux_stream_t *ds)
512 {
513 if (!ds->sh)
514 return;
515 clear_parser(ds->sh);
516 }
517 #endif
518
519 void ds_add_packet(demux_stream_t *ds, demux_packet_t *dp)
520 {
521 #if PARSE_ON_ADD && defined(CONFIG_LIBAVCODEC)
522 int len = dp->len;
523 int pos = 0;
524 while (len > 0) {
525 uint8_t *parsed_start = dp->buffer + pos;
526 int parsed_len = len;
527 int consumed = ds_parse(ds->sh, &parsed_start, &parsed_len, dp->pts, dp->pos);
528 pos += consumed;
529 len -= consumed;
530 if (parsed_start == dp->buffer && parsed_len == dp->len) {
531 ds_add_packet_internal(ds, dp);
532 } else if (parsed_len) {
533 demux_packet_t *dp2 = new_demux_packet(parsed_len);
534 dp2->pos = dp->pos;
535 dp2->pts = dp->pts; // should be parser->pts but that works badly
536 memcpy(dp2->buffer, parsed_start, parsed_len);
537 ds_add_packet_internal(ds, dp2);
538 }
539 }
540 #else
541 ds_add_packet_internal(ds, dp);
542 #endif
543 }
544
545 void ds_read_packet(demux_stream_t *ds, stream_t *stream, int len,
546 double pts, off_t pos, int flags)
547 {
548 demux_packet_t *dp = new_demux_packet(len);
549 len = stream_read(stream, dp->buffer, len);
550 resize_demux_packet(dp, len);
551 dp->pts = pts;
552 dp->pos = pos;
553 dp->flags = flags;
554 // append packet to DS stream:
555 ds_add_packet(ds, dp);
556 }
557
558 // return value:
559 // 0 = EOF or no stream found or invalid type
560 // 1 = successfully read a packet
561
562 int demux_fill_buffer(demuxer_t *demux, demux_stream_t *ds)
563 {
564 // Note: parameter 'ds' can be NULL!
565 return demux->desc->fill_buffer(demux, ds);
566 }
567
568 // return value:
569 // 0 = EOF
570 // 1 = successful
571 #define MAX_ACUMULATED_PACKETS 64
572 int ds_fill_buffer(demux_stream_t *ds)
573 {
574 demuxer_t *demux = ds->demuxer;
575 if (ds->current)
576 free_demux_packet(ds->current);
577 ds->current = NULL;
578 if (mp_msg_test(MSGT_DEMUXER, MSGL_DBG3)) {
579 if (ds == demux->audio)
580 mp_dbg(MSGT_DEMUXER, MSGL_DBG3,
581 "ds_fill_buffer(d_audio) called\n");
582 else if (ds == demux->video)
583 mp_dbg(MSGT_DEMUXER, MSGL_DBG3,
584 "ds_fill_buffer(d_video) called\n");
585 else if (ds == demux->sub)
586 mp_dbg(MSGT_DEMUXER, MSGL_DBG3, "ds_fill_buffer(d_sub) called\n");
587 else
588 mp_dbg(MSGT_DEMUXER, MSGL_DBG3,
589 "ds_fill_buffer(unknown 0x%X) called\n", (unsigned int) ds);
590 }
591 while (1) {
592 if (ds->packs) {
593 demux_packet_t *p = ds->first;
594 #if 0
595 if (demux->reference_clock != MP_NOPTS_VALUE) {
596 if ( p->pts != MP_NOPTS_VALUE
597 && p->pts > demux->reference_clock
598 && ds->packs < MAX_ACUMULATED_PACKETS) {
599 if (demux_fill_buffer(demux, ds))
600 continue;
601 }
602 }
603 #endif
604 // copy useful data:
605 ds->buffer = p->buffer;
606 ds->buffer_pos = 0;
607 ds->buffer_size = p->len;
608 ds->pos = p->pos;
609 ds->dpos += p->len; // !!!
610 ++ds->pack_no;
611 if (p->pts != MP_NOPTS_VALUE) {
612 ds->pts = p->pts;
613 ds->pts_bytes = 0;
614 }
615 ds->pts_bytes += p->len; // !!!
616 if (p->stream_pts != MP_NOPTS_VALUE)
617 demux->stream_pts = p->stream_pts;
618 ds->flags = p->flags;
619 // unlink packet:
620 ds->bytes -= p->len;
621 ds->current = p;
622 ds->first = p->next;
623 if (!ds->first)
624 ds->last = NULL;
625 --ds->packs;
626 /* The code below can set ds->eof to 1 when another stream runs
627 * out of buffer space. That makes sense because in that situation
628 * the calling code should not count on being able to demux more
629 * packets from this stream.
630 * If however the situation improves and we're called again
631 * despite the eof flag then it's better to clear it to avoid
632 * weird behavior. */
633 ds->eof = 0;
634 return 1;
635 }
636
637 #define MaybeNI _("Maybe you are playing a non-interleaved stream/file or the codec failed?\n" \
638 "For AVI files, try to force non-interleaved mode with the -ni option.\n")
639
640 if (demux->audio->packs >= MAX_PACKS
641 || demux->audio->bytes >= MAX_PACK_BYTES) {
642 mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "\nToo many audio packets in the buffer: (%d in %d bytes).\n",
643 demux->audio->packs, demux->audio->bytes);
644 mp_tmsg(MSGT_DEMUXER, MSGL_HINT, MaybeNI);
645 break;
646 }
647 if (demux->video->packs >= MAX_PACKS
648 || demux->video->bytes >= MAX_PACK_BYTES) {
649 mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "\nToo many video packets in the buffer: (%d in %d bytes).\n",
650 demux->video->packs, demux->video->bytes);
651 mp_tmsg(MSGT_DEMUXER, MSGL_HINT, MaybeNI);
652 break;
653 }
654 if (!demux_fill_buffer(demux, ds)) {
655 #if PARSE_ON_ADD && defined(CONFIG_LIBAVCODEC)
656 uint8_t *parsed_start = NULL;
657 int parsed_len = 0;
658 ds_parse(ds->sh, &parsed_start, &parsed_len, MP_NOPTS_VALUE, 0);
659 if (parsed_len) {
660 demux_packet_t *dp2 = new_demux_packet(parsed_len);
661 dp2->pts = MP_NOPTS_VALUE;
662 memcpy(dp2->buffer, parsed_start, parsed_len);
663 ds_add_packet_internal(ds, dp2);
664 continue;
665 }
666 #endif
667 mp_dbg(MSGT_DEMUXER, MSGL_DBG2,
668 "ds_fill_buffer()->demux_fill_buffer() failed\n");
669 break; // EOF
670 }
671 }
672 ds->buffer_pos = ds->buffer_size = 0;
673 ds->buffer = NULL;
674 mp_msg(MSGT_DEMUXER, MSGL_V,
675 "ds_fill_buffer: EOF reached (stream: %s) \n",
676 ds == demux->audio ? "audio" : "video");
677 ds->eof = 1;
678 return 0;
679 }
680
681 int demux_read_data(demux_stream_t *ds, unsigned char *mem, int len)
682 {
683 int x;
684 int bytes = 0;
685 while (len > 0) {
686 x = ds->buffer_size - ds->buffer_pos;
687 if (x == 0) {
688 if (!ds_fill_buffer(ds))
689 return bytes;
690 } else {
691 if (x > len)
692 x = len;
693 if (mem)
694 fast_memcpy(mem + bytes, &ds->buffer[ds->buffer_pos], x);
695 bytes += x;
696 len -= x;
697 ds->buffer_pos += x;
698 }
699 }
700 return bytes;
701 }
702
703 /**
704 * \brief read data until the given 3-byte pattern is encountered, up to maxlen
705 * \param mem memory to read data into, may be NULL to discard data
706 * \param maxlen maximum number of bytes to read
707 * \param read number of bytes actually read
708 * \param pattern pattern to search for (lowest 8 bits are ignored)
709 * \return whether pattern was found
710 */
711 int demux_pattern_3(demux_stream_t *ds, unsigned char *mem, int maxlen,
712 int *read, uint32_t pattern)
713 {
714 register uint32_t head = 0xffffff00;
715 register uint32_t pat = pattern & 0xffffff00;
716 int total_len = 0;
717 do {
718 register unsigned char *ds_buf = &ds->buffer[ds->buffer_size];
719 int len = ds->buffer_size - ds->buffer_pos;
720 register long pos = -len;
721 if (unlikely(pos >= 0)) { // buffer is empty
722 ds_fill_buffer(ds);
723 continue;
724 }
725 do {
726 head |= ds_buf[pos];
727 head <<= 8;
728 } while (++pos && head != pat);
729 len += pos;
730 if (total_len + len > maxlen)
731 len = maxlen - total_len;
732 len = demux_read_data(ds, mem ? &mem[total_len] : NULL, len);
733 total_len += len;
734 } while ((head != pat || total_len < 3) && total_len < maxlen && !ds->eof);
735 if (read)
736 *read = total_len;
737 return total_len >= 3 && head == pat;
738 }
739
740 void ds_free_packs(demux_stream_t *ds)
741 {
742 demux_packet_t *dp = ds->first;
743 while (dp) {
744 demux_packet_t *dn = dp->next;
745 free_demux_packet(dp);
746 dp = dn;
747 }
748 if (ds->asf_packet) {
749 // free unfinished .asf fragments:
750 free(ds->asf_packet->buffer);
751 free(ds->asf_packet);
752 ds->asf_packet = NULL;
753 }
754 ds->first = ds->last = NULL;
755 ds->packs = 0; // !!!!!
756 ds->bytes = 0;
757 if (ds->current)
758 free_demux_packet(ds->current);
759 ds->current = NULL;
760 ds->buffer = NULL;
761 ds->buffer_pos = ds->buffer_size;
762 ds->pts = 0;
763 ds->pts_bytes = 0;
764 }
765
766 int ds_get_packet(demux_stream_t *ds, unsigned char **start)
767 {
768 int len;
769 if (ds->buffer_pos >= ds->buffer_size) {
770 if (!ds_fill_buffer(ds)) {
771 // EOF
772 *start = NULL;
773 return -1;
774 }
775 }
776 len = ds->buffer_size - ds->buffer_pos;
777 *start = &ds->buffer[ds->buffer_pos];
778 ds->buffer_pos += len;
779 return len;
780 }
781
782 int ds_get_packet_pts(demux_stream_t *ds, unsigned char **start, double *pts)
783 {
784 int len;
785 *pts = MP_NOPTS_VALUE;
786 len = ds_get_packet(ds, start);
787 if (len < 0)
788 return len;
789 // Return pts unless this read starts from the middle of a packet
790 if (len == ds->buffer_pos)
791 *pts = ds->current->pts;
792 return len;
793 }
794
795 int ds_get_packet_sub(demux_stream_t *ds, unsigned char **start)
796 {
797 int len;
798 if (ds->buffer_pos >= ds->buffer_size) {
799 *start = NULL;
800 if (!ds->packs)
801 return -1; // no sub
802 if (!ds_fill_buffer(ds))
803 return -1; // EOF
804 }
805 len = ds->buffer_size - ds->buffer_pos;
806 *start = &ds->buffer[ds->buffer_pos];
807 ds->buffer_pos += len;
808 return len;
809 }
810
811 double ds_get_next_pts(demux_stream_t *ds)
812 {
813 demuxer_t *demux = ds->demuxer;
814 while (!ds->first) {
815 if (demux->audio->packs >= MAX_PACKS
816 || demux->audio->bytes >= MAX_PACK_BYTES) {
817 mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "\nToo many audio packets in the buffer: (%d in %d bytes).\n",
818 demux->audio->packs, demux->audio->bytes);
819 mp_tmsg(MSGT_DEMUXER, MSGL_HINT, MaybeNI);
820 return MP_NOPTS_VALUE;
821 }
822 if (demux->video->packs >= MAX_PACKS
823 || demux->video->bytes >= MAX_PACK_BYTES) {
824 mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "\nToo many video packets in the buffer: (%d in %d bytes).\n",
825 demux->video->packs, demux->video->bytes);
826 mp_tmsg(MSGT_DEMUXER, MSGL_HINT, MaybeNI);
827 return MP_NOPTS_VALUE;
828 }
829 if (!demux_fill_buffer(demux, ds))
830 return MP_NOPTS_VALUE;
831 }
832 return ds->first->pts;
833 }
834
835 // ====================================================================
836
837 void demuxer_help(void)
838 {
839 int i;
840
841 mp_msg(MSGT_DEMUXER, MSGL_INFO, "Available demuxers:\n");
842 mp_msg(MSGT_DEMUXER, MSGL_INFO, " demuxer: type info: (comment)\n");
843 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_DEMUXERS\n");
844 for (i = 0; demuxer_list[i]; i++) {
845 if (demuxer_list[i]->type > DEMUXER_TYPE_MAX) // Don't display special demuxers
846 continue;
847 if (demuxer_list[i]->comment && strlen(demuxer_list[i]->comment))
848 mp_msg(MSGT_DEMUXER, MSGL_INFO, "%10s %2d %s (%s)\n",
849 demuxer_list[i]->name, demuxer_list[i]->type,
850 demuxer_list[i]->info, demuxer_list[i]->comment);
851 else
852 mp_msg(MSGT_DEMUXER, MSGL_INFO, "%10s %2d %s\n",
853 demuxer_list[i]->name, demuxer_list[i]->type,
854 demuxer_list[i]->info);
855 }
856 }
857
858
859 /**
860 * Get demuxer type for a given demuxer name
861 *
862 * @param demuxer_name string with demuxer name of demuxer number
863 * @param force will be set if demuxer should be forced.
864 * May be NULL.
865 * @return DEMUXER_TYPE_xxx, -1 if error or not found
866 */
867 int get_demuxer_type_from_name(char *demuxer_name, int *force)
868 {
869 int i;
870 long type_int;
871 char *endptr;
872
873 if (!demuxer_name || !demuxer_name[0])
874 return DEMUXER_TYPE_UNKNOWN;
875 if (force)
876 *force = demuxer_name[0] == '+';
877 if (demuxer_name[0] == '+')
878 demuxer_name = &demuxer_name[1];
879 for (i = 0; demuxer_list[i]; i++) {
880 if (demuxer_list[i]->type > DEMUXER_TYPE_MAX) // Can't select special demuxers from commandline
881 continue;
882 if (strcmp(demuxer_name, demuxer_list[i]->name) == 0)
883 return demuxer_list[i]->type;
884 }
885
886 // No match found, try to parse name as an integer (demuxer number)
887 type_int = strtol(demuxer_name, &endptr, 0);
888 if (*endptr) // Conversion failed
889 return -1;
890 if ((type_int > 0) && (type_int <= DEMUXER_TYPE_MAX))
891 return (int) type_int;
892
893 return -1;
894 }
895
896 int extension_parsing = 1; // 0=off 1=mixed (used only for unstable formats)
897
898 /*
899 NOTE : Several demuxers may be opened at the same time so
900 demuxers should NEVER rely on an external var to enable them
901 self. If a demuxer can't do any auto-detection it should only use
902 file_format. The user can explicitly set file_format with the -demuxer
903 option so there is really no need for another extra var.
904 For convenience an option can be added to set file_format directly
905 to the right type (ex: rawaudio,rawvideo).
906 Also the stream can override the file_format so a demuxer which rely
907 on a special stream type can set file_format at the stream level
908 (ex: tv,mf).
909 */
910
911 static demuxer_t *demux_open_stream(struct MPOpts *opts, stream_t *stream,
912 int file_format, int force, int audio_id,
913 int video_id, int dvdsub_id,
914 char *filename)
915 {
916 demuxer_t *demuxer = NULL;
917
918 sh_video_t *sh_video = NULL;
919
920 const demuxer_desc_t *demuxer_desc;
921 int fformat = 0;
922 int i;
923
924 // If somebody requested a demuxer check it
925 if (file_format) {
926 if ((demuxer_desc = get_demuxer_desc_from_type(file_format))) {
927 demuxer = new_demuxer(opts, stream, demuxer_desc->type, audio_id,
928 video_id, dvdsub_id, filename);
929 if (demuxer_desc->check_file)
930 fformat = demuxer_desc->check_file(demuxer);
931 if (force || !demuxer_desc->check_file)
932 fformat = demuxer_desc->type;
933 if (fformat != 0) {
934 if (fformat == demuxer_desc->type) {
935 demuxer_t *demux2 = demuxer;
936 // Move messages to demuxer detection code?
937 mp_tmsg(MSGT_DEMUXER, MSGL_INFO,
938 "%s file format detected.\n",
939 demuxer_desc->shortdesc);
940 file_format = fformat;
941 if (!demuxer->desc->open
942 || (demux2 = demuxer->desc->open(demuxer))) {
943 demuxer = demux2;
944 goto dmx_open;
945 }
946 } else {
947 // Format changed after check, recurse
948 free_demuxer(demuxer);
949 return demux_open_stream(opts, stream, fformat, force,
950 audio_id, video_id, dvdsub_id,
951 filename);
952 }
953 }
954 // Check failed for forced demuxer, quit
955 free_demuxer(demuxer);
956 return NULL;
957 }
958 }
959 // Test demuxers with safe file checks
960 for (i = 0; (demuxer_desc = demuxer_list[i]); i++) {
961 if (demuxer_desc->safe_check) {
962 demuxer = new_demuxer(opts, stream, demuxer_desc->type, audio_id,
963 video_id, dvdsub_id, filename);
964 if ((fformat = demuxer_desc->check_file(demuxer)) != 0) {
965 if (fformat == demuxer_desc->type) {
966 demuxer_t *demux2 = demuxer;
967 mp_tmsg(MSGT_DEMUXER, MSGL_INFO,
968 "%s file format detected.\n",
969 demuxer_desc->shortdesc);
970 file_format = fformat;
971 if (!demuxer->desc->open
972 || (demux2 = demuxer->desc->open(demuxer))) {
973 demuxer = demux2;
974 goto dmx_open;
975 }
976 } else {
977 if (fformat == DEMUXER_TYPE_PLAYLIST)
978 return demuxer; // handled in mplayer.c
979 // Format changed after check, recurse
980 free_demuxer(demuxer);
981 demuxer = demux_open_stream(opts, stream, fformat, force,
982 audio_id, video_id,
983 dvdsub_id, filename);
984 if (demuxer)
985 return demuxer; // done!
986 file_format = DEMUXER_TYPE_UNKNOWN;
987 }
988 }
989 free_demuxer(demuxer);
990 demuxer = NULL;
991 }
992 }
993
994 // If no forced demuxer perform file extension based detection
995 // Ok. We're over the stable detectable fileformats, the next ones are
996 // a bit fuzzy. So by default (extension_parsing==1) try extension-based
997 // detection first:
998 if (file_format == DEMUXER_TYPE_UNKNOWN && filename
999 && extension_parsing == 1) {
1000 file_format = demuxer_type_by_filename(filename);
1001 if (file_format != DEMUXER_TYPE_UNKNOWN) {
1002 // we like recursion :)
1003 demuxer = demux_open_stream(opts, stream, file_format, force,
1004 audio_id, video_id, dvdsub_id,
1005 filename);
1006 if (demuxer)
1007 return demuxer; // done!
1008 file_format = DEMUXER_TYPE_UNKNOWN; // continue fuzzy guessing...
1009 mp_msg(MSGT_DEMUXER, MSGL_V,
1010 "demuxer: continue fuzzy content-based format guessing...\n");
1011 }
1012 }
1013 // Try detection for all other demuxers
1014 for (i = 0; (demuxer_desc = demuxer_list[i]); i++) {
1015 if (!demuxer_desc->safe_check && demuxer_desc->check_file) {
1016 demuxer = new_demuxer(opts, stream, demuxer_desc->type, audio_id,
1017 video_id, dvdsub_id, filename);
1018 if ((fformat = demuxer_desc->check_file(demuxer)) != 0) {
1019 if (fformat == demuxer_desc->type) {
1020 demuxer_t *demux2 = demuxer;
1021 mp_tmsg(MSGT_DEMUXER, MSGL_INFO,
1022 "%s file format detected.\n",
1023 demuxer_desc->shortdesc);
1024 file_format = fformat;
1025 if (!demuxer->desc->open
1026 || (demux2 = demuxer->desc->open(demuxer))) {
1027 demuxer = demux2;
1028 goto dmx_open;
1029 }
1030 } else {
1031 if (fformat == DEMUXER_TYPE_PLAYLIST)
1032 return demuxer; // handled in mplayer.c
1033 // Format changed after check, recurse
1034 free_demuxer(demuxer);
1035 demuxer = demux_open_stream(opts, stream, fformat, force,
1036 audio_id, video_id,
1037 dvdsub_id, filename);
1038 if (demuxer)
1039 return demuxer; // done!
1040 file_format = DEMUXER_TYPE_UNKNOWN;
1041 }
1042 }
1043 free_demuxer(demuxer);
1044 demuxer = NULL;
1045 }
1046 }
1047
1048 return NULL;
1049 //====== File format recognized, set up these for compatibility: =========
1050 dmx_open:
1051
1052 demuxer->file_format = file_format;
1053
1054 if ((sh_video = demuxer->video->sh) && sh_video->bih) {
1055 int biComp = le2me_32(sh_video->bih->biCompression);
1056 mp_msg(MSGT_DEMUX, MSGL_INFO,
1057 "VIDEO: [%.4s] %dx%d %dbpp %5.3f fps %5.1f kbps (%4.1f kbyte/s)\n",
1058 (char *) &biComp, sh_video->bih->biWidth,
1059 sh_video->bih->biHeight, sh_video->bih->biBitCount,
1060 sh_video->fps, sh_video->i_bps * 0.008f,
1061 sh_video->i_bps / 1024.0f);
1062 }
1063 #ifdef CONFIG_ASS
1064 if (opts->ass_enabled && ass_library) {
1065 for (i = 0; i < MAX_S_STREAMS; ++i) {
1066 sh_sub_t *sh = demuxer->s_streams[i];
1067 if (sh && sh->type == 'a') {
1068 sh->ass_track = ass_new_track(ass_library);
1069 if (sh->ass_track && sh->extradata)
1070 ass_process_codec_private(sh->ass_track, sh->extradata,
1071 sh->extradata_len);
1072 } else if (sh && sh->type != 'v')
1073 sh->ass_track = ass_default_track(ass_library);
1074 }
1075 }
1076 #endif
1077 return demuxer;
1078 }
1079
1080 char *audio_stream = NULL;
1081 char *sub_stream = NULL;
1082 int audio_stream_cache = 0;
1083
1084 char *demuxer_name = NULL; // parameter from -demuxer
1085 char *audio_demuxer_name = NULL; // parameter from -audio-demuxer
1086 char *sub_demuxer_name = NULL; // parameter from -sub-demuxer
1087
1088 extern int hr_mp3_seek;
1089
1090 extern float stream_cache_min_percent;
1091 extern float stream_cache_seek_min_percent;
1092
1093 demuxer_t *demux_open(struct MPOpts *opts, stream_t *vs, int file_format,
1094 int audio_id, int video_id, int dvdsub_id,
1095 char *filename)
1096 {
1097 stream_t *as = NULL, *ss = NULL;
1098 demuxer_t *vd, *ad = NULL, *sd = NULL;
1099 demuxer_t *res;
1100 int afmt = DEMUXER_TYPE_UNKNOWN, sfmt = DEMUXER_TYPE_UNKNOWN;
1101 int demuxer_type;
1102 int audio_demuxer_type = 0, sub_demuxer_type = 0;
1103 int demuxer_force = 0, audio_demuxer_force = 0, sub_demuxer_force = 0;
1104
1105 if ((demuxer_type =
1106 get_demuxer_type_from_name(demuxer_name, &demuxer_force)) < 0) {
1107 mp_msg(MSGT_DEMUXER, MSGL_ERR, "-demuxer %s does not exist.\n",
1108 demuxer_name);
1109 }
1110 if ((audio_demuxer_type =
1111 get_demuxer_type_from_name(audio_demuxer_name,
1112 &audio_demuxer_force)) < 0) {
1113 mp_msg(MSGT_DEMUXER, MSGL_ERR, "-audio-demuxer %s does not exist.\n",
1114 audio_demuxer_name);
1115 }
1116 if ((sub_demuxer_type =
1117 get_demuxer_type_from_name(sub_demuxer_name,
1118 &sub_demuxer_force)) < 0) {
1119 mp_msg(MSGT_DEMUXER, MSGL_ERR, "-sub-demuxer %s does not exist.\n",
1120 sub_demuxer_name);
1121 }
1122
1123 if (audio_stream) {
1124 as = open_stream(audio_stream, 0, &afmt);
1125 if (!as) {
1126 mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "Cannot open audio stream: %s\n",
1127 audio_stream);
1128 return NULL;
1129 }
1130 if (audio_stream_cache) {
1131 if (!stream_enable_cache
1132 (as, audio_stream_cache * 1024,
1133 audio_stream_cache * 1024 * (stream_cache_min_percent /
1134 100.0),
1135 audio_stream_cache * 1024 * (stream_cache_seek_min_percent /
1136 100.0))) {
1137 free_stream(as);
1138 mp_msg(MSGT_DEMUXER, MSGL_ERR,
1139 "Can't enable audio stream cache\n");
1140 return NULL;
1141 }
1142 }
1143 }
1144 if (sub_stream) {
1145 ss = open_stream(sub_stream, 0, &sfmt);
1146 if (!ss) {
1147 mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "Cannot open subtitle stream: %s\n",
1148 sub_stream);
1149 return NULL;
1150 }
1151 }
1152
1153 vd = demux_open_stream(opts, vs, demuxer_type ? demuxer_type : file_format,
1154 demuxer_force, audio_stream ? -2 : audio_id,
1155 video_id, sub_stream ? -2 : dvdsub_id, filename);
1156 if (!vd) {
1157 if (as)
1158 free_stream(as);
1159 if (ss)
1160 free_stream(ss);
1161 return NULL;
1162 }
1163 if (as) {
1164 ad = demux_open_stream(opts, as,
1165 audio_demuxer_type ? audio_demuxer_type : afmt,
1166 audio_demuxer_force, audio_id, -2, -2,
1167 audio_stream);
1168 if (!ad) {
1169 mp_tmsg(MSGT_DEMUXER, MSGL_WARN, "Failed to open audio demuxer: %s\n",
1170 audio_stream);
1171 free_stream(as);
1172 } else if (ad->audio->sh
1173 && ((sh_audio_t *) ad->audio->sh)->format == 0x55) // MP3
1174 hr_mp3_seek = 1; // Enable high res seeking
1175 }
1176 if (ss) {
1177 sd = demux_open_stream(opts, ss,
1178 sub_demuxer_type ? sub_demuxer_type : sfmt,
1179 sub_demuxer_force, -2, -2, dvdsub_id,
1180 sub_stream);
1181 if (!sd) {
1182 mp_tmsg(MSGT_DEMUXER, MSGL_WARN,
1183 "Failed to open subtitle demuxer: %s\n", sub_stream);
1184 free_stream(ss);
1185 }
1186 }
1187
1188 if (ad && sd)
1189 res = new_demuxers_demuxer(vd, ad, sd);
1190 else if (ad)
1191 res = new_demuxers_demuxer(vd, ad, vd);
1192 else if (sd)
1193 res = new_demuxers_demuxer(vd, vd, sd);
1194 else
1195 res = vd;
1196
1197 opts->correct_pts = opts->user_correct_pts;
1198 if (opts->correct_pts < 0)
1199 opts->correct_pts =
1200 demux_control(res, DEMUXER_CTRL_CORRECT_PTS,
1201 NULL) == DEMUXER_CTRL_OK;
1202 return res;
1203 }
1204
1205
1206 void demux_flush(demuxer_t *demuxer)
1207 {
1208 #if PARSE_ON_ADD
1209 ds_clear_parser(demuxer->video);
1210 ds_clear_parser(demuxer->audio);
1211 ds_clear_parser(demuxer->sub);
1212 #endif
1213 ds_free_packs(demuxer->video);
1214 ds_free_packs(demuxer->audio);
1215 ds_free_packs(demuxer->sub);
1216 }
1217
1218 int demux_seek(demuxer_t *demuxer, float rel_seek_secs, float audio_delay,
1219 int flags)
1220 {
1221 demux_stream_t *d_audio = demuxer->audio;
1222 demux_stream_t *d_video = demuxer->video;
1223 sh_audio_t *sh_audio = d_audio->sh;
1224 sh_video_t *sh_video = d_video->sh;
1225 double tmp = 0;
1226 double pts;
1227
1228 if (!demuxer->seekable) {
1229 if (demuxer->file_format == DEMUXER_TYPE_AVI)
1230 mp_tmsg(MSGT_SEEK, MSGL_WARN, "Cannot seek in raw AVI streams. (Index required, try with the -idx switch.)\n");
1231 #ifdef CONFIG_TV
1232 else if (demuxer->file_format == DEMUXER_TYPE_TV)
1233 mp_tmsg(MSGT_SEEK, MSGL_WARN, "TV input is not seekable! (Seeking will probably be for changing channels ;)\n");
1234 #endif
1235 else
1236 mp_tmsg(MSGT_SEEK, MSGL_WARN, "Cannot seek in this file.\n");
1237 return 0;
1238 }
1239 // clear demux buffers:
1240 demux_flush(demuxer);
1241 if (sh_audio)
1242 sh_audio->a_buffer_len = 0;
1243
1244 demuxer->stream->eof = 0;
1245 demuxer->video->eof = 0;
1246 demuxer->audio->eof = 0;
1247
1248 if (sh_video)
1249 sh_video->timer = 0; // !!!!!!
1250
1251 if (flags & SEEK_ABSOLUTE)
1252 pts = 0.0f;
1253 else {
1254 if (demuxer->stream_pts == MP_NOPTS_VALUE)
1255 goto dmx_seek;
1256 pts = demuxer->stream_pts;
1257 }
1258
1259 if (flags & SEEK_FACTOR) {
1260 if (stream_control(demuxer->stream, STREAM_CTRL_GET_TIME_LENGTH, &tmp)
1261 == STREAM_UNSUPPORTED)
1262 goto dmx_seek;
1263 pts += tmp * rel_seek_secs;
1264 } else
1265 pts += rel_seek_secs;
1266
1267 if (stream_control(demuxer->stream, STREAM_CTRL_SEEK_TO_TIME, &pts) !=
1268 STREAM_UNSUPPORTED) {
1269 demux_control(demuxer, DEMUXER_CTRL_RESYNC, NULL);
1270 return 1;
1271 }
1272
1273 dmx_seek:
1274 if (demuxer->desc->seek)
1275 demuxer->desc->seek(demuxer, rel_seek_secs, audio_delay, flags);
1276
1277 if (sh_audio)
1278 resync_audio_stream(sh_audio);
1279
1280 return 1;
1281 }
1282
1283 int demux_info_add(demuxer_t *demuxer, const char *opt, const char *param)
1284 {
1285 char **info = demuxer->info;
1286 int n = 0;
1287
1288
1289 for (n = 0; info && info[2 * n] != NULL; n++) {
1290 if (!strcasecmp(opt, info[2 * n])) {
1291 if (!strcmp(param, info[2 * n + 1])) {
1292 mp_msg(MSGT_DEMUX, MSGL_V, "Demuxer info %s set to unchanged value %s\n", opt, param);
1293 return 0;
1294 }
1295 mp_tmsg(MSGT_DEMUX, MSGL_INFO, "Demuxer info %s changed to %s\n", opt,
1296 param);
1297 free(info[2 * n + 1]);
1298 info[2 * n + 1] = strdup(param);
1299 return 0;
1300 }
1301 }
1302
1303 info = demuxer->info = realloc(info, (2 * (n + 2)) * sizeof(char *));
1304 info[2 * n] = strdup(opt);
1305 info[2 * n + 1] = strdup(param);
1306 memset(&info[2 * (n + 1)], 0, 2 * sizeof(char *));
1307
1308 return 1;
1309 }
1310
1311 int demux_info_print(demuxer_t *demuxer)
1312 {
1313 char **info = demuxer->info;
1314 int n;
1315
1316 if (!info)
1317 return 0;
1318
1319 mp_tmsg(MSGT_DEMUX, MSGL_INFO, "Clip info:\n");
1320 for (n = 0; info[2 * n] != NULL; n++) {
1321 mp_msg(MSGT_DEMUX, MSGL_INFO, " %s: %s\n", info[2 * n],
1322 info[2 * n + 1]);
1323 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CLIP_INFO_NAME%d=%s\n", n,
1324 info[2 * n]);
1325 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CLIP_INFO_VALUE%d=%s\n", n,
1326 info[2 * n + 1]);
1327 }
1328 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CLIP_INFO_N=%d\n", n);
1329
1330 return 0;
1331 }
1332
1333 char *demux_info_get(demuxer_t *demuxer, const char *opt)
1334 {
1335 int i;
1336 char **info = demuxer->info;
1337
1338 for (i = 0; info && info[2 * i] != NULL; i++) {
1339 if (!strcasecmp(opt, info[2 * i]))
1340 return info[2 * i + 1];
1341 }
1342
1343 return NULL;
1344 }
1345
1346 int demux_control(demuxer_t *demuxer, int cmd, void *arg)
1347 {
1348
1349 if (demuxer->desc->control)
1350 return demuxer->desc->control(demuxer, cmd, arg);
1351
1352 return DEMUXER_CTRL_NOTIMPL;
1353 }
1354
1355
1356
1357 double demuxer_get_time_length(demuxer_t *demuxer)
1358 {
1359 double get_time_ans;
1360 sh_video_t *sh_video = demuxer->video->sh;
1361 sh_audio_t *sh_audio = demuxer->audio->sh;
1362 // <= 0 means DEMUXER_CTRL_NOTIMPL or DEMUXER_CTRL_DONTKNOW
1363 if (demux_control
1364 (demuxer, DEMUXER_CTRL_GET_TIME_LENGTH, (void *) &get_time_ans) <= 0) {
1365 if (sh_video && sh_video->i_bps && sh_audio && sh_audio->i_bps)
1366 get_time_ans = (double) (demuxer->movi_end -
1367 demuxer->movi_start) / (sh_video->i_bps +
1368 sh_audio->i_bps);
1369 else if (sh_video && sh_video->i_bps)
1370 get_time_ans = (double) (demuxer->movi_end -
1371 demuxer->movi_start) / sh_video->i_bps;
1372 else if (sh_audio && sh_audio->i_bps)
1373 get_time_ans = (double) (demuxer->movi_end -
1374 demuxer->movi_start) / sh_audio->i_bps;
1375 else
1376 get_time_ans = 0;
1377 }
1378 return get_time_ans;
1379 }
1380
1381 /**
1382 * \brief demuxer_get_current_time() returns the time of the current play in three possible ways:
1383 * either when the stream reader satisfies STREAM_CTRL_GET_CURRENT_TIME (e.g. dvd)
1384 * or using sh_video->pts when the former method fails
1385 * 0 otherwise
1386 * \return the current play time
1387 */
1388 int demuxer_get_current_time(demuxer_t *demuxer)
1389 {
1390 double get_time_ans = 0;
1391 sh_video_t *sh_video = demuxer->video->sh;
1392 if (demuxer->stream_pts != MP_NOPTS_VALUE)
1393 get_time_ans = demuxer->stream_pts;
1394 else if (sh_video)
1395 get_time_ans = sh_video->pts;
1396 return (int) get_time_ans;
1397 }
1398
1399 int demuxer_get_percent_pos(demuxer_t *demuxer)
1400 {
1401 int ans = 0;
1402 int res = demux_control(demuxer, DEMUXER_CTRL_GET_PERCENT_POS, &ans);
1403 int len = (demuxer->movi_end - demuxer->movi_start) / 100;
1404 if (res <= 0) {
1405 off_t pos = demuxer->filepos > 0 ? demuxer->filepos : stream_tell(demuxer->stream);
1406 if (len > 0)
1407 ans = (pos - demuxer->movi_start) / len;
1408 else
1409 ans = 0;
1410 }
1411 if (ans < 0)
1412 ans = 0;
1413 if (ans > 100)
1414 ans = 100;
1415 return ans;
1416 }
1417
1418 int demuxer_switch_audio(demuxer_t *demuxer, int index)
1419 {
1420 int res = demux_control(demuxer, DEMUXER_CTRL_SWITCH_AUDIO, &index);
1421 if (res == DEMUXER_CTRL_NOTIMPL)
1422 index = demuxer->audio->id;
1423 if (demuxer->audio->id >= 0)
1424 demuxer->audio->sh = demuxer->a_streams[demuxer->audio->id];
1425 else
1426 demuxer->audio->sh = NULL;
1427 return index;
1428 }
1429
1430 int demuxer_switch_video(demuxer_t *demuxer, int index)
1431 {
1432 int res = demux_control(demuxer, DEMUXER_CTRL_SWITCH_VIDEO, &index);
1433 if (res == DEMUXER_CTRL_NOTIMPL)
1434 index = demuxer->video->id;
1435 if (demuxer->video->id >= 0)
1436 demuxer->video->sh = demuxer->v_streams[demuxer->video->id];
1437 else
1438 demuxer->video->sh = NULL;
1439 return index;
1440 }
1441
1442 int demuxer_add_attachment(demuxer_t *demuxer, const char *name,
1443 int name_maxlen, const char *type, int type_maxlen,
1444 const void *data, size_t size)
1445 {
1446 if (!(demuxer->num_attachments % 32))
1447 demuxer->attachments = talloc_realloc(demuxer, demuxer->attachments,
1448 struct demux_attachment,
1449 demuxer->num_attachments + 32);
1450
1451 struct demux_attachment *att =
1452 demuxer->attachments + demuxer->num_attachments;
1453 att->name = talloc_strndup(demuxer->attachments, name, name_maxlen);
1454 att->type = talloc_strndup(demuxer->attachments, type, type_maxlen);
1455 att->data = talloc_size(demuxer->attachments, size);
1456 memcpy(att->data, data, size);
1457 att->data_size = size;
1458
1459 return demuxer->num_attachments++;
1460 }
1461
1462 int demuxer_add_chapter(demuxer_t *demuxer, const char *name, int name_maxlen,
1463 uint64_t start, uint64_t end)
1464 {
1465 if (!(demuxer->num_chapters % 32))
1466 demuxer->chapters = talloc_realloc(demuxer, demuxer->chapters,
1467 struct demux_chapter,
1468 demuxer->num_chapters + 32);
1469
1470 demuxer->chapters[demuxer->num_chapters].start = start;
1471 demuxer->chapters[demuxer->num_chapters].end = end;
1472 demuxer->chapters[demuxer->num_chapters].name = name ?
1473 talloc_strndup(demuxer->chapters, name, name_maxlen) :
1474 talloc_strdup(demuxer->chapters, mp_gtext("unknown"));
1475
1476 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CHAPTER_ID=%d\n", demuxer->num_chapters);
1477 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CHAPTER_%d_START=%"PRIu64"\n", demuxer->num_chapters, start);
1478 if (end)
1479 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CHAPTER_%d_END=%"PRIu64"\n", demuxer->num_chapters, end);
1480 if (name)
1481 mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CHAPTER_%d_NAME=%.*s\n", demuxer->num_chapters, name_maxlen, name);
1482
1483 return demuxer->num_chapters++;
1484 }
1485
1486 /**
1487 * \brief demuxer_seek_chapter() seeks to a chapter in two possible ways:
1488 * either using the demuxer->chapters structure set by the demuxer
1489 * or asking help to the stream layer (e.g. dvd)
1490 * \param chapter - chapter number wished - 0-based
1491 * \param seek_pts set by the function to the pts to seek to (if demuxer->chapters is set)
1492 * \param chapter_name name of chapter found (set by this function is param is not null)
1493 * \return -1 on error, current chapter if successful
1494 */
1495
1496 int demuxer_seek_chapter(demuxer_t *demuxer, int chapter, double *seek_pts,
1497 char **chapter_name)
1498 {
1499 int ris;
1500 sh_video_t *sh_video = demuxer->video->sh;
1501 sh_audio_t *sh_audio = demuxer->audio->sh;
1502
1503 if (!demuxer->num_chapters || !demuxer->chapters) {
1504 demux_flush(demuxer);
1505
1506 ris = stream_control(demuxer->stream, STREAM_CTRL_SEEK_TO_CHAPTER,
1507 &chapter);
1508 if (ris != STREAM_UNSUPPORTED)
1509 demux_control(demuxer, DEMUXER_CTRL_RESYNC, NULL);
1510 if (sh_video) {
1511 ds_fill_buffer(demuxer->video);
1512 resync_video_stream(sh_video);
1513 }
1514
1515 if (sh_audio) {
1516 ds_fill_buffer(demuxer->audio);
1517 resync_audio_stream(sh_audio);
1518 }
1519 // exit status may be ok, but main() doesn't have to seek itself
1520 // (because e.g. dvds depend on sectors, not on pts)
1521 *seek_pts = -1.0;
1522
1523 if (chapter_name) {
1524 *chapter_name = NULL;
1525 int num_chapters;
1526 if (stream_control(demuxer->stream, STREAM_CTRL_GET_NUM_CHAPTERS,
1527 &num_chapters) == STREAM_UNSUPPORTED)
1528 num_chapters = 0;
1529 if (num_chapters) {
1530 *chapter_name = talloc_size(NULL, 16);
1531 sprintf(*chapter_name, " of %3d", num_chapters);
1532 }
1533 }
1534
1535 return ris != STREAM_UNSUPPORTED ? chapter : -1;
1536 } else { // chapters structure is set in the demuxer
1537 if (chapter >= demuxer->num_chapters)
1538 return -1;
1539 if (chapter < 0)
1540 chapter = 0;
1541
1542 *seek_pts = demuxer->chapters[chapter].start / 1000.0;
1543
1544 if (chapter_name)
1545 *chapter_name = talloc_strdup(NULL, demuxer->chapters[chapter].name);
1546
1547 return chapter;
1548 }
1549 }
1550
1551 int demuxer_get_current_chapter(demuxer_t *demuxer)
1552 {
1553 int chapter = -1;
1554 if (!demuxer->num_chapters || !demuxer->chapters) {
1555 if (stream_control(demuxer->stream, STREAM_CTRL_GET_CURRENT_CHAPTER,
1556 &chapter) == STREAM_UNSUPPORTED)
1557 chapter = -1;
1558 } else {
1559 sh_video_t *sh_video = demuxer->video->sh;
1560 sh_audio_t *sh_audio = demuxer->audio->sh;
1561 uint64_t now;
1562 now = (sh_video ? sh_video->pts : (sh_audio ? sh_audio->pts : 0))
1563 * 1000 + 0.5;
1564 for (chapter = demuxer->num_chapters - 1; chapter >= 0; --chapter) {
1565 if (demuxer->chapters[chapter].start <= now)
1566 break;
1567 }
1568 }
1569 return chapter;
1570 }
1571
1572 char *demuxer_chapter_name(demuxer_t *demuxer, int chapter)
1573 {
1574 if (demuxer->num_chapters && demuxer->chapters) {
1575 if (chapter >= 0 && chapter < demuxer->num_chapters
1576 && demuxer->chapters[chapter].name)
1577 return strdup(demuxer->chapters[chapter].name);
1578 }
1579 return NULL;
1580 }
1581
1582 char *demuxer_chapter_display_name(demuxer_t *demuxer, int chapter)
1583 {
1584 char *chapter_name = demuxer_chapter_name(demuxer, chapter);
1585 if (chapter_name) {
1586 char *tmp = malloc(strlen(chapter_name) + 14);
1587 snprintf(tmp, 63, "(%d) %s", chapter + 1, chapter_name);
1588 free(chapter_name);
1589 return tmp;
1590 } else {
1591 int chapter_num = demuxer_chapter_count(demuxer);
1592 char tmp[30];
1593 if (chapter_num <= 0)
1594 sprintf(tmp, "(%d)", chapter + 1);
1595 else
1596 sprintf(tmp, "(%d) of %d", chapter + 1, chapter_num);
1597 return strdup(tmp);
1598 }
1599 }
1600
1601 float demuxer_chapter_time(demuxer_t *demuxer, int chapter, float *end)
1602 {
1603 if (demuxer->num_chapters && demuxer->chapters && chapter >= 0
1604 && chapter < demuxer->num_chapters) {
1605 if (end)
1606 *end = demuxer->chapters[chapter].end / 1000.0;
1607 return demuxer->chapters[chapter].start / 1000.0;
1608 }
1609 return -1.0;
1610 }
1611
1612 int demuxer_chapter_count(demuxer_t *demuxer)
1613 {
1614 if (!demuxer->num_chapters || !demuxer->chapters) {
1615 int num_chapters = 0;
1616 if (stream_control(demuxer->stream, STREAM_CTRL_GET_NUM_CHAPTERS,
1617 &num_chapters) == STREAM_UNSUPPORTED)
1618 num_chapters = 0;
1619 return num_chapters;
1620 } else
1621 return demuxer->num_chapters;
1622 }
1623
1624 int demuxer_angles_count(demuxer_t *demuxer)
1625 {
1626 int ris, angles = -1;
1627
1628 ris = stream_control(demuxer->stream, STREAM_CTRL_GET_NUM_ANGLES, &angles);
1629 if (ris == STREAM_UNSUPPORTED)
1630 return -1;
1631 return angles;
1632 }
1633
1634 int demuxer_get_current_angle(demuxer_t *demuxer)
1635 {
1636 int ris, curr_angle = -1;
1637 ris = stream_control(demuxer->stream, STREAM_CTRL_GET_ANGLE, &curr_angle);
1638 if (ris == STREAM_UNSUPPORTED)
1639 return -1;
1640 return curr_angle;
1641 }
1642
1643
1644 int demuxer_set_angle(demuxer_t *demuxer, int angle)
1645 {
1646 int ris, angles = -1;
1647 sh_video_t *sh_video = demuxer->video->sh;
1648 sh_audio_t *sh_audio = demuxer->audio->sh;
1649
1650 angles = demuxer_angles_count(demuxer);
1651 if ((angles < 1) || (angle > angles))
1652 return -1;
1653
1654 demux_flush(demuxer);
1655
1656 ris = stream_control(demuxer->stream, STREAM_CTRL_SET_ANGLE, &angle);
1657 if (ris == STREAM_UNSUPPORTED)
1658 return -1;
1659
1660 demux_control(demuxer, DEMUXER_CTRL_RESYNC, NULL);
1661 if (sh_video) {
1662 ds_fill_buffer(demuxer->video);
1663 resync_video_stream(sh_video);
1664 }
1665
1666 if (sh_audio) {
1667 ds_fill_buffer(demuxer->audio);
1668 resync_audio_stream(sh_audio);
1669 }
1670
1671 return angle;
1672 }
1673
1674 int demuxer_audio_track_by_lang(demuxer_t *d, char *lang)
1675 {
1676 int i, len;
1677 lang += strspn(lang, ",");
1678 while ((len = strcspn(lang, ",")) > 0) {
1679 for (i = 0; i < MAX_A_STREAMS; ++i) {
1680 sh_audio_t *sh = d->a_streams[i];
1681 if (sh && sh->lang && strncmp(sh->lang, lang, len) == 0)
1682 return sh->aid;
1683 }
1684 lang += len;
1685 lang += strspn(lang, ",");
1686 }
1687 return -1;
1688 }
1689
1690 int demuxer_sub_track_by_lang(demuxer_t *d, char *lang)
1691 {
1692 int i, len;
1693 lang += strspn(lang, ",");
1694 while ((len = strcspn(lang, ",")) > 0) {
1695 for (i = 0; i < MAX_S_STREAMS; ++i) {
1696 sh_sub_t *sh = d->s_streams[i];
1697 if (sh && sh->lang && strncmp(sh->lang, lang, len) == 0)
1698 return sh->sid;
1699 }
1700 lang += len;
1701 lang += strspn(lang, ",");
1702 }
1703 return -1;
1704 }
1705
1706 int demuxer_default_audio_track(demuxer_t *d)
1707 {
1708 int i;
1709 for (i = 0; i < MAX_A_STREAMS; ++i) {
1710 sh_audio_t *sh = d->a_streams[i];
1711 if (sh && sh->default_track)
1712 return sh->aid;
1713 }
1714 for (i = 0; i < MAX_A_STREAMS; ++i) {
1715 sh_audio_t *sh = d->a_streams[i];
1716 if (sh)
1717 return sh->aid;
1718 }
1719 return -1;
1720 }
1721
1722 int demuxer_default_sub_track(demuxer_t *d)
1723 {
1724 int i;
1725 for (i = 0; i < MAX_S_STREAMS; ++i) {
1726 sh_sub_t *sh = d->s_streams[i];
1727 if (sh && sh->default_track)
1728 return sh->sid;
1729 }
1730 return -1;
1731 }
Sources used to build Kovensky's MPlayer