summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorwm4 <wm4@nowhere>2012-09-07 14:42:25 +0200
committerwm4 <wm4@nowhere>2012-09-18 21:07:30 +0200
commitae9c3d530c55e86eb54d8cf23a23c5ce6301ec6e (patch)
tree20f50ce7713ffa7d671a86fc43b709354ae8991a
parentc323592c3ac8557e09b8235bd43196f968e91168 (diff)
downloadmpv-ae9c3d530c55e86eb54d8cf23a23c5ce6301ec6e.tar.bz2
mpv-ae9c3d530c55e86eb54d8cf23a23c5ce6301ec6e.tar.xz
libmpdemux: add back demux_ts
Someone wanted this. Apparently both libavformat's TS demuxer and demux_ts are crap, and work/fail in different cases. This demuxer has been removed in 1fde09db6f4ce. All code added comes from the revision before that. Some required bits have been added in the commit before this one (re-adding demux_mpg), in particular the changes to video.c. stream_dvb will use this demuxer by default, otherwise demux_lavf is preferred (as it has been before). Some TS related command line options are not re-added. Closed captions might not work.
-rw-r--r--Makefile1
-rw-r--r--libmpdemux/demux_ts.c3533
-rw-r--r--libmpdemux/demux_ts.h24
-rw-r--r--libmpdemux/demuxer.c15
-rw-r--r--libmpdemux/stheader.h2
5 files changed, 3574 insertions, 1 deletions
diff --git a/Makefile b/Makefile
index 34e4bdef8f..863eeb4b85 100644
--- a/Makefile
+++ b/Makefile
@@ -235,6 +235,7 @@ SRCS_COMMON = asxparser.c \
libmpdemux/demux_mf.c \
libmpdemux/demux_mkv.c \
libmpdemux/demux_mpg.c \
+ libmpdemux/demux_ts.c \
libmpdemux/mp3_hdr.c \
libmpdemux/parse_es.c \
libmpdemux/mpeg_hdr.c \
diff --git a/libmpdemux/demux_ts.c b/libmpdemux/demux_ts.c
new file mode 100644
index 0000000000..dbf12b7456
--- /dev/null
+++ b/libmpdemux/demux_ts.c
@@ -0,0 +1,3533 @@
+/*
+ * Demultiplexer for MPEG2 Transport Streams.
+ *
+ * Written by Nico <nsabbi@libero.it>
+ * Kind feedback is appreciated; 'sucks' and alike is not.
+ * Originally based on demux_pva.c written by Matteo Giani and FFmpeg (libavformat) sources
+ *
+ * This file is part of MPlayer.
+ *
+ * MPlayer is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * MPlayer is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with MPlayer; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "config.h"
+#include "mp_msg.h"
+#include "options.h"
+
+#include "libmpcodecs/dec_audio.h"
+#include "stream/stream.h"
+#include "demuxer.h"
+#include "parse_es.h"
+#include "stheader.h"
+#include "ms_hdr.h"
+#include "mpeg_hdr.h"
+#include "demux_ts.h"
+
+#define TS_PH_PACKET_SIZE 192
+#define TS_FEC_PACKET_SIZE 204
+#define TS_PACKET_SIZE 188
+#define NB_PID_MAX 8192
+
+#define MAX_HEADER_SIZE 6 /* enough for PES header + length */
+#define MAX_CHECK_SIZE 65535
+#define NUM_CONSECUTIVE_TS_PACKETS 32
+#define NUM_CONSECUTIVE_AUDIO_PACKETS 348
+#define MAX_A52_FRAME_SIZE 3840
+
+#ifndef SIZE_MAX
+#define SIZE_MAX ((size_t)-1)
+#endif
+
+#define TYPE_AUDIO 1
+#define TYPE_VIDEO 2
+#define TYPE_SUB 3
+
+int ts_prog;
+int ts_keep_broken=0;
+off_t ts_probe = 0;
+int audio_substream_id = -1;
+
+typedef enum
+{
+ UNKNOWN = -1,
+ VIDEO_MPEG1 = 0x10000001,
+ VIDEO_MPEG2 = 0x10000002,
+ VIDEO_MPEG4 = 0x10000004,
+ VIDEO_H264 = 0x10000005,
+ VIDEO_AVC = mmioFOURCC('a', 'v', 'c', '1'),
+ VIDEO_DIRAC = mmioFOURCC('d', 'r', 'a', 'c'),
+ VIDEO_VC1 = mmioFOURCC('W', 'V', 'C', '1'),
+ AUDIO_MP2 = 0x50,
+ AUDIO_A52 = 0x2000,
+ AUDIO_DTS = 0x2001,
+ AUDIO_LPCM_BE = 0x10001,
+ AUDIO_AAC = mmioFOURCC('M', 'P', '4', 'A'),
+ AUDIO_AAC_LATM = mmioFOURCC('M', 'P', '4', 'L'),
+ AUDIO_TRUEHD = mmioFOURCC('T', 'R', 'H', 'D'),
+ AUDIO_S302M = mmioFOURCC('B', 'S', 'S', 'D'),
+ SPU_DVD = 0x3000000,
+ SPU_DVB = 0x3000001,
+ SPU_TELETEXT = 0x3000002,
+ SPU_PGS = 0x3000003,
+ PES_PRIVATE1 = 0xBD00000,
+ SL_PES_STREAM = 0xD000000,
+ SL_SECTION = 0xD100000,
+ MP4_OD = 0xD200000,
+} es_stream_type_t;
+
+typedef struct {
+ uint8_t *buffer;
+ uint16_t buffer_len;
+} ts_section_t;
+
+typedef struct {
+ int size;
+ unsigned char *start;
+ uint16_t payload_size;
+ es_stream_type_t type, subtype;
+ double pts, last_pts;
+ int pid;
+ char lang[4];
+ int last_cc; // last cc code (-1 if first packet)
+ int is_synced;
+ ts_section_t section;
+ uint8_t *extradata;
+ int extradata_alloc, extradata_len;
+ struct {
+ uint8_t au_start, au_end, last_au_end;
+ } sl;
+} ES_stream_t;
+
+typedef struct {
+ void *sh;
+ int id;
+ int type;
+} sh_av_t;
+
+typedef struct MpegTSContext {
+ int packet_size; // raw packet size, including FEC if present e.g. 188 bytes
+ ES_stream_t *pids[NB_PID_MAX];
+ sh_av_t streams[NB_PID_MAX];
+} MpegTSContext;
+
+
+typedef struct {
+ demux_stream_t *ds;
+ demux_packet_t *pack;
+ int offset, buffer_size;
+} av_fifo_t;
+
+#define MAX_EXTRADATA_SIZE 64*1024
+typedef struct {
+ int32_t object_type; //aka codec used
+ int32_t stream_type; //video, audio etc.
+ uint8_t buf[MAX_EXTRADATA_SIZE];
+ uint16_t buf_size;
+ uint8_t szm1;
+} mp4_decoder_config_t;
+
+typedef struct {
+ //flags
+ uint8_t flags;
+ uint8_t au_start;
+ uint8_t au_end;
+ uint8_t random_accesspoint;
+ uint8_t random_accesspoint_only;
+ uint8_t padding;
+ uint8_t use_ts;
+ uint8_t idle;
+ uint8_t duration;
+
+ uint32_t ts_resolution, ocr_resolution;
+ uint8_t ts_len, ocr_len, au_len, instant_bitrate_len, degr_len, au_seqnum_len, packet_seqnum_len;
+ uint32_t timescale;
+ uint16_t au_duration, cts_duration;
+ uint64_t ocr, dts, cts;
+} mp4_sl_config_t;
+
+typedef struct {
+ uint16_t id;
+ uint8_t flags;
+ mp4_decoder_config_t decoder;
+ mp4_sl_config_t sl;
+} mp4_es_descr_t;
+
+typedef struct {
+ uint16_t id;
+ uint8_t flags;
+ mp4_es_descr_t *es;
+ uint16_t es_cnt;
+} mp4_od_t;
+
+typedef struct {
+ uint8_t skip;
+ uint8_t table_id;
+ uint8_t ssi;
+ uint16_t section_length;
+ uint16_t ts_id;
+ uint8_t version_number;
+ uint8_t curr_next;
+ uint8_t section_number;
+ uint8_t last_section_number;
+ struct pat_progs_t {
+ uint16_t id;
+ uint16_t pmt_pid;
+ } *progs;
+ uint16_t progs_cnt;
+ ts_section_t section;
+} pat_t;
+
+typedef struct {
+ uint16_t progid;
+ uint8_t skip;
+ uint8_t table_id;
+ uint8_t ssi;
+ uint16_t section_length;
+ uint8_t version_number;
+ uint8_t curr_next;
+ uint8_t section_number;
+ uint8_t last_section_number;
+ uint16_t PCR_PID;
+ uint16_t prog_descr_length;
+ ts_section_t section;
+ uint16_t es_cnt;
+ struct pmt_es_t {
+ uint16_t pid;
+ uint32_t type; //it's 8 bit long, but cast to the right type as FOURCC
+ uint16_t descr_length;
+ uint8_t format_descriptor[5];
+ uint8_t lang[4];
+ uint16_t mp4_es_id;
+ } *es;
+ mp4_od_t iod, *od;
+ mp4_es_descr_t *mp4es;
+ int od_cnt, mp4es_cnt;
+} pmt_t;
+
+typedef struct {
+ uint64_t size;
+ float duration;
+ double first_pts;
+ double last_pts;
+} TS_stream_info;
+
+typedef struct {
+ MpegTSContext ts;
+ int last_pid;
+ av_fifo_t fifo[3]; //0 for audio, 1 for video, 2 for subs
+ pat_t pat;
+ pmt_t *pmt;
+ uint16_t pmt_cnt;
+ uint32_t prog;
+ uint32_t vbitrate;
+ int keep_broken;
+ int last_aid;
+ int last_vid;
+ int last_sid;
+ char packet[TS_FEC_PACKET_SIZE];
+ TS_stream_info vstr, astr;
+} ts_priv_t;
+
+
+typedef struct {
+ es_stream_type_t type;
+ ts_section_t section;
+} TS_pids_t;
+
+
+static int IS_AUDIO(es_stream_type_t type)
+{
+ switch (type) {
+ case AUDIO_MP2:
+ case AUDIO_A52:
+ case AUDIO_LPCM_BE:
+ case AUDIO_AAC:
+ case AUDIO_AAC_LATM:
+ case AUDIO_DTS:
+ case AUDIO_TRUEHD:
+ case AUDIO_S302M:
+ return 1;
+ }
+ return 0;
+}
+
+static int IS_VIDEO(es_stream_type_t type)
+{
+ switch (type) {
+ case VIDEO_MPEG1:
+ case VIDEO_MPEG2:
+ case VIDEO_MPEG4:
+ case VIDEO_H264:
+ case VIDEO_AVC:
+ case VIDEO_DIRAC:
+ case VIDEO_VC1:
+ return 1;
+ }
+ return 0;
+}
+
+static int IS_SUB(es_stream_type_t type)
+{
+ switch (type) {
+ case SPU_DVD:
+ case SPU_DVB:
+ case SPU_PGS:
+ case SPU_TELETEXT:
+ return 1;
+ }
+ return 0;
+}
+
+static int ts_parse(demuxer_t *demuxer, ES_stream_t *es, unsigned char *packet, int probe);
+
+static uint8_t get_packet_size(const unsigned char *buf, int size)
+{
+ int i;
+
+ if (size < (TS_FEC_PACKET_SIZE * NUM_CONSECUTIVE_TS_PACKETS))
+ return 0;
+
+ for(i=0; i<NUM_CONSECUTIVE_TS_PACKETS; i++)
+ {
+ if (buf[i * TS_PACKET_SIZE] != 0x47)
+ {
+ mp_msg(MSGT_DEMUX, MSGL_DBG2, "GET_PACKET_SIZE, pos %d, char: %2x\n", i, buf[i * TS_PACKET_SIZE]);
+ goto try_fec;
+ }
+ }
+ return TS_PACKET_SIZE;
+
+try_fec:
+ for(i=0; i<NUM_CONSECUTIVE_TS_PACKETS; i++)
+ {
+ if (buf[i * TS_FEC_PACKET_SIZE] != 0x47){
+ mp_msg(MSGT_DEMUX, MSGL_DBG2, "GET_PACKET_SIZE, pos %d, char: %2x\n", i, buf[i * TS_PACKET_SIZE]);
+ goto try_philips;
+ }
+ }
+ return TS_FEC_PACKET_SIZE;
+
+ try_philips:
+ for(i=0; i<NUM_CONSECUTIVE_TS_PACKETS; i++)
+ {
+ if (buf[i * TS_PH_PACKET_SIZE] != 0x47)
+ return 0;
+ }
+ return TS_PH_PACKET_SIZE;
+}
+
+static int parse_avc_sps(uint8_t *buf, int len, int *w, int *h);
+static inline uint8_t *pid_lang_from_pmt(ts_priv_t *priv, int pid);
+
+static void ts_add_stream(demuxer_t * demuxer, ES_stream_t *es)
+{
+ int i;
+ ts_priv_t *priv = (ts_priv_t*) demuxer->priv;
+
+ if(priv->ts.streams[es->pid].sh)
+ return;
+
+ if((IS_AUDIO(es->type) || IS_AUDIO(es->subtype)) && priv->last_aid+1 < MAX_A_STREAMS)
+ {
+ sh_audio_t *sh = new_sh_audio_aid(demuxer, priv->last_aid, es->pid);
+ if(sh)
+ {
+ const char *lang = pid_lang_from_pmt(priv, es->pid);
+ sh->needs_parsing = 1;
+ sh->format = IS_AUDIO(es->type) ? es->type : es->subtype;
+ sh->ds = demuxer->audio;
+
+ priv->ts.streams[es->pid].id = priv->last_aid;
+ priv->ts.streams[es->pid].sh = sh;
+ priv->ts.streams[es->pid].type = TYPE_AUDIO;
+ mp_msg(MSGT_DEMUX, MSGL_V, "\r\nADDED AUDIO PID %d, type: %x stream n. %d\r\n", es->pid, sh->format, priv->last_aid);
+ if (lang && lang[0])
+ mp_msg(MSGT_IDENTIFY, MSGL_V, "ID_AID_%d_LANG=%s\n", es->pid, lang);
+ priv->last_aid++;
+ }
+
+ if(es->extradata && es->extradata_len)
+ {
+ sh->wf = malloc(sizeof(*sh->wf) + es->extradata_len);
+ sh->wf->cbSize = es->extradata_len;
+ memcpy(sh->wf + 1, es->extradata, es->extradata_len);
+ }
+ }
+
+ if((IS_VIDEO(es->type) || IS_VIDEO(es->subtype)) && priv->last_vid+1 < MAX_V_STREAMS)
+ {
+ sh_video_t *sh = new_sh_video_vid(demuxer, priv->last_vid, es->pid);
+ if(sh)
+ {
+ sh->format = IS_VIDEO(es->type) ? es->type : es->subtype;
+ sh->ds = demuxer->video;
+
+ priv->ts.streams[es->pid].id = priv->last_vid;
+ priv->ts.streams[es->pid].sh = sh;
+ priv->ts.streams[es->pid].type = TYPE_VIDEO;
+ mp_msg(MSGT_DEMUX, MSGL_V, "\r\nADDED VIDEO PID %d, type: %x stream n. %d\r\n", es->pid, sh->format, priv->last_vid);
+ priv->last_vid++;
+
+
+ if(sh->format == VIDEO_AVC && es->extradata && es->extradata_len)
+ {
+ int w = 0, h = 0;
+ sh->bih = calloc(1, sizeof(*sh->bih) + es->extradata_len);
+ sh->bih->biSize= sizeof(*sh->bih) + es->extradata_len;
+ sh->bih->biCompression = sh->format;
+ memcpy(sh->bih + 1, es->extradata, es->extradata_len);
+ mp_msg(MSGT_DEMUXER,MSGL_DBG2, "EXTRADATA(%d BYTES): \n", es->extradata_len);
+ for(i = 0;i < es->extradata_len; i++)
+ mp_msg(MSGT_DEMUXER,MSGL_DBG2, "%02x ", (int) es->extradata[i]);
+ mp_msg(MSGT_DEMUXER,MSGL_DBG2,"\n");
+ if(parse_avc_sps(es->extradata, es->extradata_len, &w, &h))
+ {
+ sh->bih->biWidth = w;
+ sh->bih->biHeight = h;
+ }
+ }
+ }
+ }
+
+ if(IS_SUB(es->type) && priv->last_sid+1 < MAX_S_STREAMS)
+ {
+ sh_sub_t *sh = new_sh_sub_sid_lang(demuxer, priv->last_sid, es->pid, pid_lang_from_pmt(priv, es->pid));
+ if (sh) {
+ switch (es->type) {
+ case SPU_DVB:
+ sh->type = 'b'; break;
+ case SPU_DVD:
+ sh->type = 'v'; break;
+ case SPU_PGS:
+ sh->type = 'p'; break;
+ }
+ priv->ts.streams[es->pid].id = priv->last_sid;
+ priv->ts.streams[es->pid].sh = sh;
+ priv->ts.streams[es->pid].type = TYPE_SUB;
+ priv->last_sid++;
+ }
+ }
+}
+
+static int ts_check_file(demuxer_t * demuxer)
+{
+ const int buf_size = (TS_FEC_PACKET_SIZE * NUM_CONSECUTIVE_TS_PACKETS);
+ unsigned char buf[TS_FEC_PACKET_SIZE * NUM_CONSECUTIVE_TS_PACKETS], done = 0, *ptr;
+ uint32_t _read, i, count = 0, is_ts;
+ int cc[NB_PID_MAX], last_cc[NB_PID_MAX], pid, cc_ok, c, good, bad;
+ uint8_t size = 0;
+ off_t pos = 0;
+ off_t init_pos;
+
+ mp_msg(MSGT_DEMUX, MSGL_V, "Checking for MPEG-TS...\n");
+
+ init_pos = stream_tell(demuxer->stream);
+ is_ts = 0;
+ while(! done)
+ {
+ i = 1;
+ c = 0;
+
+ while(((c=stream_read_char(demuxer->stream)) != 0x47)
+ && (c >= 0)
+ && (i < MAX_CHECK_SIZE)
+ && ! demuxer->stream->eof
+ ) i++;
+
+
+ if(c != 0x47)
+ {
+ mp_msg(MSGT_DEMUX, MSGL_V, "THIS DOESN'T LOOK LIKE AN MPEG-TS FILE!\n");
+ is_ts = 0;
+ done = 1;
+ continue;
+ }
+
+ pos = stream_tell(demuxer->stream) - 1;
+ buf[0] = c;
+ _read = stream_read(demuxer->stream, &buf[1], buf_size-1);
+
+ if(_read < buf_size-1)
+ {
+ mp_msg(MSGT_DEMUX, MSGL_V, "COULDN'T READ ENOUGH DATA, EXITING TS_CHECK\n");
+ stream_reset(demuxer->stream);
+ return 0;
+ }
+
+ size = get_packet_size(buf, buf_size);
+ if(size)
+ {
+ done = 1;
+ is_ts = 1;
+ }
+
+ if(pos - init_pos >= MAX_CHECK_SIZE)
+ {
+ done = 1;
+ is_ts = 0;
+ }
+ }
+
+ mp_msg(MSGT_DEMUX, MSGL_V, "TRIED UP TO POSITION %"PRIu64", FOUND %x, packet_size= %d, SEEMS A TS? %d\n", (uint64_t) pos, c, size, is_ts);
+ stream_seek(demuxer->stream, pos);
+
+ if(! is_ts)
+ return 0;
+
+ //LET'S CHECK continuity counters
+ good = bad = 0;
+ for(count = 0; count < NB_PID_MAX; count++)
+ {
+ cc[count] = last_cc[count] = -1;
+ }
+
+ for(count = 0; count < NUM_CONSECUTIVE_TS_PACKETS; count++)
+ {
+ ptr = &(buf[size * count]);
+ pid = ((ptr[1] & 0x1f) << 8) | ptr[2];
+ mp_msg(MSGT_DEMUX, MSGL_DBG2, "BUF: %02x %02x %02x %02x, PID %d, SIZE: %d \n",
+ ptr[0], ptr[1], ptr[2], ptr[3], pid, size);
+
+ if((pid == 8191) || (pid < 16))
+ continue;
+
+ cc[pid] = (ptr[3] & 0xf);
+ cc_ok = (last_cc[pid] < 0) || ((((last_cc[pid] + 1) & 0x0f) == cc[pid]));
+ mp_msg(MSGT_DEMUX, MSGL_DBG2, "PID %d, COMPARE CC %d AND LAST_CC %d\n", pid, cc[pid], last_cc[pid]);
+ if(! cc_ok)
+ //return 0;
+ bad++;
+ else
+ good++;
+
+ last_cc[pid] = cc[pid];
+ }
+
+ mp_msg(MSGT_DEMUX, MSGL_V, "GOOD CC: %d, BAD CC: %d\n", good, bad);
+
+ if(good >= bad)
+ return size;
+ else
+ return 0;
+}
+
+
+static inline int32_t progid_idx_in_pmt(ts_priv_t *priv, uint16_t progid)
+{
+ int x;
+
+ if(priv->pmt == NULL)
+ return -1;
+
+ for(x = 0; x < priv->pmt_cnt; x++)
+ {
+ if(priv->pmt[x].progid == progid)
+ return x;
+ }
+
+ return -1;
+}
+
+
+static inline int32_t progid_for_pid(ts_priv_t *priv, int pid, int32_t req) //finds the first program listing a pid
+{
+ int i, j;
+ pmt_t *pmt;
+
+
+ if(priv->pmt == NULL)
+ return -1;
+
+
+ for(i=0; i < priv->pmt_cnt; i++)
+ {
+ pmt = &(priv->pmt[i]);
+
+ if(pmt->es == NULL)
+ return -1;
+
+ for(j = 0; j < pmt->es_cnt; j++)
+ {
+ if(pmt->es[j].pid == pid)
+ {
+ if((req == 0) || (req == pmt->progid))
+ return pmt->progid;
+ }
+ }
+
+ }
+ return -1;
+}
+
+static inline int32_t prog_pcr_pid(ts_priv_t *priv, int progid)
+{
+ int i;
+
+ if(priv->pmt == NULL)
+ return -1;
+ for(i=0; i < priv->pmt_cnt; i++)
+ {
+ if(priv->pmt[i].progid == progid)
+ return priv->pmt[i].PCR_PID;
+ }
+ return -1;
+}
+
+
+static inline int pid_match_lang(ts_priv_t *priv, uint16_t pid, char *lang)
+{
+ uint16_t i, j;
+ pmt_t *pmt;
+
+ if(priv->pmt == NULL)
+ return -1;
+
+ for(i=0; i < priv->pmt_cnt; i++)
+ {
+ pmt = &(priv->pmt[i]);
+
+ if(pmt->es == NULL)
+ return -1;
+
+ for(j = 0; j < pmt->es_cnt; j++)
+ {
+ if(pmt->es[j].pid != pid)
+ continue;
+
+ mp_msg(MSGT_DEMUXER, MSGL_V, "CMP LANG %s AND %s, pids: %d %d\n",pmt->es[j].lang, lang, pmt->es[j].pid, pid);
+ if(strncmp(pmt->es[j].lang, lang, 3) == 0)
+ {
+ return 1;
+ }
+ }
+
+ }
+
+ return -1;
+}
+
+typedef struct {
+ int32_t atype, vtype, stype; //types
+ int32_t apid, vpid, spid; //stream ids
+ char alang[4]; //languages
+ uint16_t prog;
+ off_t probe;
+} tsdemux_init_t;
+
+//second stage: returns the count of A52 syncwords found
+static int a52_check(char *buf, int len)
+{
+ int cnt, frame_length = 0, ok, srate;
+
+ cnt = ok = 0;
+ if(len < 8)
+ return 0;
+
+ while(cnt < len - 7)
+ {
+ if(buf[cnt] == 0x0B && buf[cnt+1] == 0x77)
+ {
+ frame_length = mp_a52_framesize(&buf[cnt], &srate);
+ if(frame_length>=7 && frame_length<=3840)
+ {
+ cnt += frame_length;
+ ok++;
+ }
+ else
+ cnt++;
+ }
+ else
+ cnt++;
+ }
+
+ mp_msg(MSGT_DEMUXER, MSGL_V, "A52_CHECK(%d input bytes), found %d frame syncwords of %d bytes length\n", len, ok, frame_length);
+ return ok;
+}
+
+
+static off_t ts_detect_streams(demuxer_t *demuxer, tsdemux_init_t *param)
+{
+ int video_found = 0, audio_found = 0, i, num_packets = 0, req_apid, req_vpid, req_spid;
+ int is_audio, is_video, is_sub, has_tables;
+ int32_t p, chosen_pid = 0;
+ off_t pos=0, ret = 0, init_pos, end_pos;
+ ES_stream_t es;
+ unsigned char tmp[TS_FEC_PACKET_SIZE];
+ ts_priv_t *priv = (ts_priv_t*) demuxer->priv;
+ struct {
+ char *buf;
+ int pos;
+ } pes_priv1[8192], *pptr;
+ char *tmpbuf;
+
+ priv->last_pid = 8192; //invalid pid
+
+ req_apid = param->apid;
+ req_vpid = param->vpid;
+ req_spid = param->spid;
+
+ has_tables = 0;
+ memset(pes_priv1, 0, sizeof(pes_priv1));
+ init_pos = stream_tell(demuxer->stream);
+ mp_msg(MSGT_DEMUXER, MSGL_V, "PROBING UP TO %"PRIu64", PROG: %d\n", (uint64_t) param->probe, param->prog);
+ end_pos = init_pos + (param->probe ? param->probe : TS_MAX_PROBE_SIZE);
+ while(1)
+ {
+ pos = stream_tell(demuxer->stream);
+ if(pos > end_pos || demuxer->stream->eof)
+ break;
+
+ if(ts_parse(demuxer, &es, tmp, 1))
+ {
+ //Non PES-aligned A52 audio may escape detection if PMT is not present;
+ //in this case we try to find at least 3 A52 syncwords
+ if((es.type == PES_PRIVATE1) && (! audio_found) && req_apid > -2)
+ {
+ pptr = &pes_priv1[es.pid];
+ if(pptr->pos < 64*1024)
+ {
+ tmpbuf = realloc(pptr->buf, pptr->pos + es.size);
+ if(tmpbuf != NULL)
+ {
+ pptr->buf = tmpbuf;
+ memcpy(&(pptr->buf[ pptr->pos ]), es.start, es.size);
+ pptr->pos += es.size;
+ if(a52_check(pptr->buf, pptr->pos) > 2)
+ {
+ param->atype = AUDIO_A52;
+ param->apid = es.pid;
+ es.type = AUDIO_A52;
+ }
+ }
+ }
+ }
+
+ is_audio = IS_AUDIO(es.type) || ((es.type==SL_PES_STREAM) && IS_AUDIO(es.subtype));
+ is_video = IS_VIDEO(es.type) || ((es.type==SL_PES_STREAM) && IS_VIDEO(es.subtype));
+ is_sub = IS_SUB(es.type);
+
+
+ if((! is_audio) && (! is_video) && (! is_sub))
+ continue;
+ if(is_audio && req_apid==-2)
+ continue;
+
+ if(is_video)
+ {
+ chosen_pid = (req_vpid == es.pid);
+ if((! chosen_pid) && (req_vpid > 0))
+ continue;
+ }
+ else if(is_audio)
+ {
+ if(req_apid > 0)
+ {
+ chosen_pid = (req_apid == es.pid);
+ if(! chosen_pid)
+ continue;
+ }
+ else if(param->alang[0] > 0 && es.lang[0] > 0)
+ {
+ if(pid_match_lang(priv, es.pid, param->alang) == -1)
+ continue;
+
+ chosen_pid = 1;
+ param->apid = req_apid = es.pid;
+ }
+ }
+ else if(is_sub)
+ {
+ chosen_pid = (req_spid == es.pid);
+ if((! chosen_pid) && (req_spid > 0))
+ continue;
+ }
+
+ if(req_apid < 0 && (param->alang[0] == 0) && req_vpid < 0 && req_spid < 0)
+ chosen_pid = 1;
+
+ if((ret == 0) && chosen_pid)
+ {
+ ret = stream_tell(demuxer->stream);
+ }
+
+ p = progid_for_pid(priv, es.pid, param->prog);
+ if(p != -1)
+ {
+ has_tables++;
+ if(!param->prog && chosen_pid)
+ param->prog = p;
+ }
+
+ if((param->prog > 0) && (param->prog != p))
+ {
+ if(audio_found)
+ {
+ if(is_video && (req_vpid == es.pid))
+ {
+ param->vtype = IS_VIDEO(es.type) ? es.type : es.subtype;
+ param->vpid = es.pid;
+ video_found = 1;
+ break;
+ }
+ }
+
+ if(video_found)
+ {
+ if(is_audio && (req_apid == es.pid))
+ {
+ param->atype = IS_AUDIO(es.type) ? es.type : es.subtype;
+ param->apid = es.pid;
+ audio_found = 1;
+ break;
+ }
+ }
+
+
+ continue;
+ }
+
+
+ mp_msg(MSGT_DEMUXER, MSGL_DBG2, "TYPE: %x, PID: %d, PROG FOUND: %d\n", es.type, es.pid, param->prog);
+
+
+ if(is_video)
+ {
+ if((req_vpid == -1) || (req_vpid == es.pid))
+ {
+ param->vtype = IS_VIDEO(es.type) ? es.type : es.subtype;
+ param->vpid = es.pid;
+ video_found = 1;
+ }
+ }
+
+
+ if(((req_vpid == -2) || (num_packets >= NUM_CONSECUTIVE_AUDIO_PACKETS)) && audio_found && !param->probe)
+ {
+ //novideo or we have at least 348 audio packets (64 KB) without video (TS with audio only)
+ param->vtype = 0;
+ break;
+ }
+
+ if(is_sub)
+ {
+ if((req_spid == -1) || (req_spid == es.pid))
+ {
+ param->stype = es.type;
+ param->spid = es.pid;
+ }
+ }
+
+ if(is_audio)
+ {
+ if((req_apid == -1) || (req_apid == es.pid))
+ {
+ param->atype = IS_AUDIO(es.type) ? es.type : es.subtype;
+ param->apid = es.pid;
+ audio_found = 1;
+ }
+ }
+
+ if(audio_found && (param->apid == es.pid) && (! video_found))
+ num_packets++;
+
+ if((has_tables==0) && (video_found && audio_found) && (pos >= 1000000))
+ break;
+ }
+ }
+
+ for(i=0; i<8192; i++)
+ {
+ if(pes_priv1[i].buf != NULL)
+ {
+ free(pes_priv1[i].buf);
+ pes_priv1[i].buf = NULL;
+ pes_priv1[i].pos = 0;
+ }
+ }
+
+ if(video_found)
+ {
+ if(param->vtype == VIDEO_MPEG1)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "VIDEO MPEG1(pid=%d) ", param->vpid);
+ else if(param->vtype == VIDEO_MPEG2)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "VIDEO MPEG2(pid=%d) ", param->vpid);
+ else if(param->vtype == VIDEO_MPEG4)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "VIDEO MPEG4(pid=%d) ", param->vpid);
+ else if(param->vtype == VIDEO_H264)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "VIDEO H264(pid=%d) ", param->vpid);
+ else if(param->vtype == VIDEO_VC1)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "VIDEO VC1(pid=%d) ", param->vpid);
+ else if(param->vtype == VIDEO_AVC)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "VIDEO AVC(NAL-H264, pid=%d) ", param->vpid);
+ }
+ else
+ {
+ param->vtype = UNKNOWN;
+ //WE DIDN'T MATCH ANY VIDEO STREAM
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "NO VIDEO! ");
+ }
+
+ if(param->atype == AUDIO_MP2)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "AUDIO MPA(pid=%d)", param->apid);
+ else if(param->atype == AUDIO_A52)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "AUDIO A52(pid=%d)", param->apid);
+ else if(param->atype == AUDIO_DTS)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "AUDIO DTS(pid=%d)", param->apid);
+ else if(param->atype == AUDIO_LPCM_BE)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "AUDIO LPCM(pid=%d)", param->apid);
+ else if(param->atype == AUDIO_AAC)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "AUDIO AAC(pid=%d)", param->apid);
+ else if(param->atype == AUDIO_AAC_LATM)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "AUDIO AAC LATM(pid=%d)", param->apid);
+ else if(param->atype == AUDIO_TRUEHD)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "AUDIO TRUEHD(pid=%d)", param->apid);
+ else if(param->atype == AUDIO_S302M)
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "AUDIO S302M(pid=%d)", param->apid);
+ else
+ {
+ audio_found = 0;
+ param->atype = UNKNOWN;
+ //WE DIDN'T MATCH ANY AUDIO STREAM, SO WE FORCE THE DEMUXER TO IGNORE AUDIO
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "NO AUDIO! (try increasing -tsprobe)");
+ }
+
+ if(IS_SUB(param->stype))
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, " SUB %s(pid=%d) ", (param->stype==SPU_DVD ? "DVD" : param->stype==SPU_DVB ? "DVB" : "Teletext"), param->spid);
+ else
+ {
+ param->stype = UNKNOWN;
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, " NO SUBS (yet)! ");
+ }
+
+ if(video_found || audio_found)
+ {
+ if(!param->prog)
+ {
+ p = progid_for_pid(priv, video_found ? param->vpid : param->apid, 0);
+ if(p != -1)
+ param->prog = p;
+ }
+
+ if(demuxer->stream->eof && (ret == 0))
+ ret = init_pos;
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, " PROGRAM N. %d\n", param->prog);
+ }
+ else
+ mp_msg(MSGT_DEMUXER, MSGL_INFO, "\n");
+
+
+ for(i=0; i<NB_PID_MAX; i++)
+ {
+ if(priv->ts.pids[i] != NULL)
+ {
+ priv->ts.pids[i]->payload_size = 0;
+ priv->ts.pids[i]->pts = priv->ts.pids[i]->last_pts = 0;
+ priv->ts.pids[i]->last_cc = -1;
+ priv->ts.pids[i]->is_synced = 0;
+ }
+ }
+
+ return ret;
+}
+
+static int parse_avc_sps(uint8_t *buf, int len, int *w, int *h)
+{
+ int sps, sps_len;
+ unsigned char *ptr;
+ mp_mpeg_header_t picture;
+ if(len < 6)
+ return 0;
+ sps = buf[5] & 0x1f;
+ if(!sps)
+ return 0;
+ sps_len = (buf[6] << 8) | buf[7];
+ if(!sps_len || (sps_len > len - 8))
+ return 0;
+ ptr = &(buf[8]);
+ picture.display_picture_width = picture.display_picture_height = 0;
+ h264_parse_sps(&picture, ptr, len - 8);
+ if(!picture.display_picture_width || !picture.display_picture_height)
+ return 0;
+ *w = picture.display_picture_width;
+ *h = picture.display_picture_height;
+ return 1;
+}
+
+static demuxer_t *demux_open_ts(demuxer_t * demuxer)
+{
+ int i;
+ uint8_t packet_size;
+ sh_video_t *sh_video;
+ sh_audio_t *sh_audio;
+ off_t start_pos;
+ tsdemux_init_t params;
+ ts_priv_t * priv = demuxer->priv;
+
+ mp_msg(MSGT_DEMUX, MSGL_V, "DEMUX OPEN, AUDIO_ID: %d, VIDEO_ID: %d, SUBTITLE_ID: %d,\n",
+ demuxer->audio->id, demuxer->video->id, demuxer->sub->id);
+
+
+ demuxer->type= DEMUXER_TYPE_MPEG_TS;
+
+
+ stream_reset(demuxer->stream);
+
+ packet_size = ts_check_file(demuxer);
+ if(!packet_size)
+ return NULL;
+
+ priv = calloc(1, sizeof(ts_priv_t));
+ if(priv == NULL)
+ {
+ mp_msg(MSGT_DEMUX, MSGL_FATAL, "DEMUX_OPEN_TS, couldn't allocate enough memory for ts->priv, exit\n");
+ return NULL;
+ }
+
+ for(i=0; i < NB_PID_MAX; i++)
+ {
+ priv->ts.pids[i] = NULL;
+ priv->ts.streams[i].id = -3;
+ }
+ priv->pat.progs = NULL;
+ priv->pat.progs_cnt = 0;
+ priv->pat.section.buffer = NULL;
+ priv->pat.section.buffer_len = 0;
+
+ priv->pmt = NULL;
+ priv->pmt_cnt = 0;
+
+ priv->keep_broken = ts_keep_broken;
+ priv->ts.packet_size = packet_size;
+
+
+ demuxer->priv = priv;
+ if(demuxer->stream->type != STREAMTYPE_FILE)
+ demuxer->seekable = 1;
+ else
+ demuxer->seekable = 1;
+
+
+ params.atype = params.vtype = params.stype = UNKNOWN;
+ params.apid = demuxer->audio->id;
+ params.vpid = demuxer->video->id;
+ params.spid = demuxer->sub->id;
+ params.prog = ts_prog;
+ params.probe = ts_probe;
+
+ if(demuxer->opts->audio_lang != NULL)
+ {
+ strncpy(params.alang, demuxer->opts->audio_lang[0], 3);
+ params.alang[3] = 0;
+ }
+ else
+ memset(params.alang, 0, 4);
+
+ start_pos = ts_detect_streams(demuxer, &params);
+
+ demuxer->sub->id = params.spid;
+ priv->prog = params.prog;
+
+ if(params.vtype != UNKNOWN)
+ {
+ ts_add_stream(demuxer, priv->ts.pids[params.vpid]);
+ sh_video = priv->ts.streams[params.vpid].sh;
+ demuxer->video->id = priv->ts.streams[params.vpid].id;
+ sh_video->ds = demuxer->video;
+ sh_video->format = params.vtype;
+ demuxer->video->sh = sh_video;
+ }
+
+ if(params.atype != UNKNOWN)
+ {
+ ES_stream_t *es = priv->ts.pids[params.apid];
+
+ if(!IS_AUDIO(es->type) && !IS_AUDIO(es->subtype) && IS_AUDIO(params.atype)) es->subtype = params.atype;
+ ts_add_stream(demuxer, priv->ts.pids[params.apid]);
+ sh_audio = priv->ts.streams[params.apid].sh;
+ demuxer->audio->id = priv->ts.streams[params.apid].id;
+ sh_audio->ds = demuxer->audio;
+ sh_audio->format = params.atype;
+ demuxer->audio->sh = sh_audio;
+ }
+
+
+ mp_msg(MSGT_DEMUXER,MSGL_V, "Opened TS demuxer, audio: %x(pid %d), video: %x(pid %d)...POS=%"PRIu64", PROBE=%"PRIu64"\n", params.atype, demuxer->audio->id, params.vtype, demuxer->video->id, (uint64_t) start_pos, ts_probe);
+
+
+ start_pos = start_pos <= priv->ts.packet_size ?
+ demuxer->stream->start_pos :
+ start_pos - priv->ts.packet_size;
+ demuxer->movi_start = start_pos;
+ demuxer->reference_clock = MP_NOPTS_VALUE;
+ stream_reset(demuxer->stream);
+ stream_seek(demuxer->stream, start_pos); //IF IT'S FROM A PIPE IT WILL FAIL, BUT WHO CARES?
+
+
+ priv->last_pid = 8192; //invalid pid
+
+ for(i = 0; i < 3; i++)
+ {
+ priv->fifo[i].pack = NULL;
+ priv->fifo[i].offset = 0;
+ }
+ priv->fifo[0].ds = demuxer->audio;
+ priv->fifo[1].ds = demuxer->video;
+ priv->fifo[2].ds = demuxer->sub;
+
+ priv->fifo[0].buffer_size = 1536;
+ priv->fifo[1].buffer_size = 32767;
+ priv->fifo[2].buffer_size = 32767;
+
+ priv->pat.section.buffer_len = 0;
+ for(i = 0; i < priv->pmt_cnt; i++)
+ priv->pmt[i].section.buffer_len = 0;
+
+ demuxer->filepos = stream_tell(demuxer->stream);
+ return demuxer;
+}
+
+static void demux_close_ts(demuxer_t * demuxer)
+{
+ uint16_t i;
+ ts_priv_t *priv = (ts_priv_t*) demuxer->priv;
+
+ if(priv)
+ {
+ free(priv->pat.section.buffer);
+ free(priv->pat.progs);
+
+ if(priv->pmt)
+ {
+ for(i = 0; i < priv->pmt_cnt; i++)
+ {
+ free(priv->pmt[i].section.buffer);
+ free(priv->pmt[i].es);
+ }
+ free(priv->pmt);
+ }
+ for (i = 0; i < NB_PID_MAX; i++)
+ {
+ free(priv->ts.pids[i]);
+ priv->ts.pids[i] = NULL;
+ }
+ for (i = 0; i < 3; i++)
+ {
+ if (priv->fifo[i].pack)
+ free_demux_packet(priv->fifo[i].pack);
+ priv->fifo[i].pack = NULL;
+ }
+ free(priv);
+ }
+ demuxer->priv=NULL;