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