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