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