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