mirror of https://github.com/mpv-player/mpv
1191 lines
38 KiB
C
1191 lines
38 KiB
C
// MPG/VOB file parser for DEMUXER v2.5 by A'rpi/ESP-team
|
||
|
||
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <unistd.h>
|
||
#include <math.h>
|
||
|
||
#include "config.h"
|
||
#include "mp_msg.h"
|
||
#include "help_mp.h"
|
||
|
||
#include "stream.h"
|
||
#include "demuxer.h"
|
||
#include "parse_es.h"
|
||
#include "stheader.h"
|
||
#include "mp3_hdr.h"
|
||
|
||
//#define MAX_PS_PACKETSIZE 2048
|
||
#define MAX_PS_PACKETSIZE (224*1024)
|
||
|
||
#define UNKNOWN 0
|
||
#define VIDEO_MPEG1 0x10000001
|
||
#define VIDEO_MPEG2 0x10000002
|
||
#define VIDEO_MPEG4 0x10000004
|
||
#define VIDEO_H264 0x10000005
|
||
#define AUDIO_MP2 0x50
|
||
#define AUDIO_A52 0x2000
|
||
#define AUDIO_LPCM_BE 0x10001
|
||
#define AUDIO_AAC mmioFOURCC('M', 'P', '4', 'A')
|
||
|
||
typedef struct mpg_demuxer {
|
||
float last_pts;
|
||
float first_pts; // first pts found in stream
|
||
float first_to_final_pts_len; // difference between final pts and first pts
|
||
int has_valid_timestamps; // !=0 iff time stamps look linear
|
||
// (not necessarily starting with 0)
|
||
unsigned int es_map[0x40]; //es map of stream types (associated to the pes id) from 0xb0 to 0xef
|
||
int num_a_streams;
|
||
int a_stream_ids[MAX_A_STREAMS];
|
||
} mpg_demuxer_t;
|
||
|
||
extern char* dvdsub_lang;
|
||
static int mpeg_pts_error=0;
|
||
off_t ps_probe = 0;
|
||
|
||
static int parse_psm(demuxer_t *demux, int len) {
|
||
unsigned char c, id, type;
|
||
unsigned int plen, prog_len, es_map_len;
|
||
mpg_demuxer_t *priv = (mpg_demuxer_t *) demux->priv;
|
||
|
||
mp_dbg(MSGT_DEMUX,MSGL_V, "PARSE_PSM, len=%d\n", len);
|
||
if(! len || len > 1018)
|
||
return 0;
|
||
|
||
c = stream_read_char(demux->stream);
|
||
if(! (c & 0x80)) {
|
||
stream_skip(demux->stream, len - 1); //not yet valid, discard
|
||
return 0;
|
||
}
|
||
stream_skip(demux->stream, 1);
|
||
prog_len = stream_read_word(demux->stream); //length of program descriptors
|
||
stream_skip(demux->stream, prog_len); //.. that we ignore
|
||
es_map_len = stream_read_word(demux->stream); //length of elementary streams map
|
||
es_map_len = FFMIN(es_map_len, len - prog_len - 8); //sanity check
|
||
while(es_map_len > 0) {
|
||
type = stream_read_char(demux->stream);
|
||
id = stream_read_char(demux->stream);
|
||
if(id >= 0xB0 && id <= 0xEF && priv) {
|
||
int idoffset = id - 0xB0;
|
||
switch(type) {
|
||
case 0x1:
|
||
priv->es_map[idoffset] = VIDEO_MPEG1;
|
||
break;
|
||
case 0x2:
|
||
priv->es_map[idoffset] = VIDEO_MPEG2;
|
||
break;
|
||
case 0x3:
|
||
case 0x4:
|
||
priv->es_map[idoffset] = AUDIO_MP2;
|
||
break;
|
||
case 0x0f:
|
||
case 0x11:
|
||
priv->es_map[idoffset] = AUDIO_AAC;
|
||
break;
|
||
case 0x10:
|
||
priv->es_map[idoffset] = VIDEO_MPEG4;
|
||
break;
|
||
case 0x1b:
|
||
priv->es_map[idoffset] = VIDEO_H264;
|
||
break;
|
||
case 0x81:
|
||
priv->es_map[idoffset] = AUDIO_A52;
|
||
break;
|
||
}
|
||
mp_dbg(MSGT_DEMUX,MSGL_V, "PSM ES, id=0x%x, type=%x, stype: %x\n", id, type, priv->es_map[idoffset]);
|
||
}
|
||
plen = stream_read_word(demux->stream); //length of elementary stream descriptors
|
||
plen = FFMIN(plen, es_map_len); //sanity check
|
||
stream_skip(demux->stream, plen); //skip descriptors for now
|
||
es_map_len -= 4 + plen;
|
||
}
|
||
stream_skip(demux->stream, 4); //skip crc32
|
||
return 1;
|
||
}
|
||
|
||
// 500000 is a wild guess
|
||
#define TIMESTAMP_PROBE_LEN 500000
|
||
|
||
//MAX_PTS_DIFF_FOR_CONSECUTIVE denotes the maximum difference
|
||
//between two pts to consider them consecutive
|
||
//1.0 is a wild guess
|
||
#define MAX_PTS_DIFF_FOR_CONSECUTIVE 1.0
|
||
|
||
//returns the first pts found within TIME_STAMP_PROBE_LEN bytes after stream_pos in demuxer's stream.
|
||
//if no pts is found or an error occurs, -1.0 is returned.
|
||
//Packs are freed.
|
||
static float read_first_mpeg_pts_at_position(demuxer_t* demuxer, off_t stream_pos)
|
||
{
|
||
stream_t *s = demuxer->stream;
|
||
mpg_demuxer_t *mpg_d = demuxer->priv;
|
||
float pts = -1.0; //the pts to return;
|
||
float found_pts1; //the most recently found pts
|
||
float found_pts2; //the pts found before found_pts1
|
||
float found_pts3; //the pts found before found_pts2
|
||
int found = 0;
|
||
|
||
if(!mpg_d || stream_pos < 0)
|
||
return pts;
|
||
|
||
found_pts3 = found_pts2 = found_pts1 = mpg_d->last_pts;
|
||
stream_seek(s, stream_pos);
|
||
|
||
//We look for pts.
|
||
//However, we do not stop at the first found one, as timestamps may reset
|
||
//Therefore, we seek until we found three consecutive
|
||
//pts within MAX_PTS_DIFF_FOR_CONSECUTIVE.
|
||
|
||
while(found<3 && !s->eof
|
||
&& (fabsf(found_pts2-found_pts1) < MAX_PTS_DIFF_FOR_CONSECUTIVE)
|
||
&& (fabsf(found_pts3-found_pts2) < MAX_PTS_DIFF_FOR_CONSECUTIVE)
|
||
&& (stream_tell(s) < stream_pos + TIMESTAMP_PROBE_LEN)
|
||
&& ds_fill_buffer(demuxer->video))
|
||
{
|
||
if(mpg_d->last_pts != found_pts1)
|
||
{
|
||
if(!found)
|
||
found_pts3 = found_pts2 = found_pts1 = mpg_d->last_pts; //the most recently found pts
|
||
else
|
||
{
|
||
found_pts3 = found_pts2;
|
||
found_pts2 = found_pts1;
|
||
found_pts1 = mpg_d->last_pts;
|
||
}
|
||
found++;
|
||
}
|
||
}
|
||
|
||
if(found == 3) pts = found_pts3;
|
||
|
||
//clean up from searching of first pts;
|
||
ds_free_packs(demuxer->audio);
|
||
ds_free_packs(demuxer->video);
|
||
ds_free_packs(demuxer->sub);
|
||
|
||
return pts;
|
||
}
|
||
|
||
/// Open an mpg physical stream
|
||
static demuxer_t* demux_mpg_open(demuxer_t* demuxer) {
|
||
stream_t *s = demuxer->stream;
|
||
mpg_demuxer_t* mpg_d;
|
||
|
||
if (!ds_fill_buffer(demuxer->video)) return 0;
|
||
mpg_d = calloc(1,sizeof(mpg_demuxer_t));
|
||
if(mpg_d)
|
||
{
|
||
demuxer->priv = mpg_d;
|
||
mpg_d->last_pts = -1.0;
|
||
mpg_d->first_pts = -1.0;
|
||
|
||
//if seeking is allowed set has_valid_timestamps if appropriate
|
||
if(demuxer->seekable
|
||
&& demuxer->stream->type == STREAMTYPE_FILE
|
||
&& demuxer->movi_start != demuxer-> movi_end
|
||
)
|
||
{
|
||
//We seek to the beginning of the stream, to somewhere in the
|
||
//middle, and to the end of the stream, while remembering the pts
|
||
//at each of the three positions. With these pts, we check whether
|
||
//or not the pts are "linear enough" to justify seeking by the pts
|
||
//of the stream
|
||
|
||
//The position where the stream is now
|
||
off_t pos = stream_tell(s);
|
||
float first_pts = read_first_mpeg_pts_at_position(demuxer, demuxer->movi_start);
|
||
if(first_pts != -1.0)
|
||
{
|
||
float middle_pts = read_first_mpeg_pts_at_position(demuxer, (demuxer->movi_end - demuxer->movi_start)/2);
|
||
if(middle_pts != -1.0)
|
||
{
|
||
float final_pts = read_first_mpeg_pts_at_position(demuxer, demuxer->movi_end - TIMESTAMP_PROBE_LEN);
|
||
if(final_pts != -1.0)
|
||
{
|
||
// found proper first, middle, and final pts.
|
||
float proportion = (middle_pts-first_pts==0) ? -1 : (final_pts-middle_pts)/(middle_pts-first_pts);
|
||
// if they are linear enough set has_valid_timestamps
|
||
if((0.5 < proportion) && (proportion < 2))
|
||
{
|
||
mpg_d->first_pts = first_pts;
|
||
mpg_d->first_to_final_pts_len = final_pts - first_pts;
|
||
mpg_d->has_valid_timestamps = 1;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
//Cleaning up from seeking in stream
|
||
demuxer->stream->eof=0;
|
||
demuxer->video->eof=0;
|
||
demuxer->audio->eof=0;
|
||
|
||
stream_seek(s,pos);
|
||
ds_fill_buffer(demuxer->video);
|
||
} // if ( demuxer->seekable )
|
||
} // if ( mpg_d )
|
||
return demuxer;
|
||
}
|
||
|
||
static void demux_close_mpg(demuxer_t* demuxer) {
|
||
mpg_demuxer_t* mpg_d = demuxer->priv;
|
||
if (mpg_d) free(mpg_d);
|
||
}
|
||
|
||
|
||
static unsigned long long read_mpeg_timestamp(stream_t *s,int c){
|
||
unsigned int d,e;
|
||
unsigned long long pts;
|
||
d=stream_read_word(s);
|
||
e=stream_read_word(s);
|
||
if( ((c&1)!=1) || ((d&1)!=1) || ((e&1)!=1) ){
|
||
++mpeg_pts_error;
|
||
return 0; // invalid pts
|
||
}
|
||
pts=(((uint64_t)((c>>1)&7))<<30)|((d>>1)<<15)|(e>>1);
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG3," pts {%"PRIu64"}",pts);
|
||
return pts;
|
||
}
|
||
|
||
static void new_audio_stream(demuxer_t *demux, int aid){
|
||
if(!demux->a_streams[aid]){
|
||
mpg_demuxer_t *mpg_d=(mpg_demuxer_t*)demux->priv;
|
||
sh_audio_t* sh_a;
|
||
new_sh_audio(demux,aid);
|
||
sh_a = (sh_audio_t*)demux->a_streams[aid];
|
||
switch(aid & 0xE0){ // 1110 0000 b (high 3 bit: type low 5: id)
|
||
case 0x00: sh_a->format=0x50;break; // mpeg
|
||
case 0xA0: sh_a->format=0x10001;break; // dvd pcm
|
||
case 0x80: if((aid & 0xF8) == 0x88) sh_a->format=0x2001;//dts
|
||
else sh_a->format=0x2000;break; // ac3
|
||
}
|
||
//evo files
|
||
if((aid & 0xC0) == 0xC0) sh_a->format=0x2000;
|
||
if (mpg_d) mpg_d->a_stream_ids[mpg_d->num_a_streams++] = aid;
|
||
}
|
||
if(demux->audio->id==-1) demux->audio->id=aid;
|
||
}
|
||
|
||
static int demux_mpg_read_packet(demuxer_t *demux,int id){
|
||
int d;
|
||
int len;
|
||
int set_pts=0; // !=0 iff pts has been set to a proper value
|
||
unsigned char c=0;
|
||
unsigned long long pts=0;
|
||
unsigned long long dts=0;
|
||
int l;
|
||
int pes_ext2_subid=-1;
|
||
double stream_pts = MP_NOPTS_VALUE;
|
||
demux_stream_t *ds=NULL;
|
||
demux_packet_t* dp;
|
||
mpg_demuxer_t *priv = (mpg_demuxer_t *) demux->priv;
|
||
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG3,"demux_read_packet: %X\n",id);
|
||
|
||
// if(id==0x1F0){
|
||
// demux->synced=0; // force resync after 0x1F0
|
||
// return -1;
|
||
//}
|
||
|
||
// if(id==0x1BA) packet_start_pos=stream_tell(demux->stream);
|
||
if((id<0x1BC || id>=0x1F0) && id != 0x1FD) return -1;
|
||
if(id==0x1BE) return -1; // padding stream
|
||
if(id==0x1BF) return -1; // private2
|
||
|
||
len=stream_read_word(demux->stream);
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG3,"PACKET len=%d",len);
|
||
// if(len==62480){ demux->synced=0;return -1;} /* :) */
|
||
if(len==0 || len>MAX_PS_PACKETSIZE){
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG2,"Invalid PS packet len: %d\n",len);
|
||
return -2; // invalid packet !!!!!!
|
||
}
|
||
|
||
mpeg_pts_error=0;
|
||
|
||
if(id==0x1BC) {
|
||
parse_psm(demux, len);
|
||
return 0;
|
||
}
|
||
|
||
while(len>0){ // Skip stuFFing bytes
|
||
c=stream_read_char(demux->stream);--len;
|
||
if(c!=0xFF)break;
|
||
}
|
||
if((c>>6)==1){ // Read (skip) STD scale & size value
|
||
// printf(" STD_scale=%d",(c>>5)&1);
|
||
d=((c&0x1F)<<8)|stream_read_char(demux->stream);
|
||
len-=2;
|
||
// printf(" STD_size=%d",d);
|
||
c=stream_read_char(demux->stream);
|
||
}
|
||
// Read System-1 stream timestamps:
|
||
if((c>>4)==2){
|
||
pts=read_mpeg_timestamp(demux->stream,c);
|
||
set_pts=1;
|
||
len-=4;
|
||
} else
|
||
if((c>>4)==3){
|
||
pts=read_mpeg_timestamp(demux->stream,c);
|
||
c=stream_read_char(demux->stream);
|
||
if((c>>4)!=1) pts=0; //printf("{ERROR4}");
|
||
else set_pts = 1;
|
||
dts=read_mpeg_timestamp(demux->stream,c);
|
||
len-=4+1+4;
|
||
} else
|
||
if((c>>6)==2){
|
||
int pts_flags;
|
||
int hdrlen;
|
||
int parse_ext2;
|
||
// System-2 (.VOB) stream:
|
||
c=stream_read_char(demux->stream); pts_flags=c>>6;
|
||
parse_ext2 = (id == 0x1FD) && ((c & 0x3F) == 1);
|
||
c=stream_read_char(demux->stream); hdrlen=c;
|
||
len-=2;
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG3," hdrlen=%d (len=%d)",hdrlen,len);
|
||
if(hdrlen>len){ mp_msg(MSGT_DEMUX,MSGL_V,"demux_mpg: invalid header length \n"); return -1;}
|
||
if(pts_flags==2 && hdrlen>=5){
|
||
c=stream_read_char(demux->stream);
|
||
pts=read_mpeg_timestamp(demux->stream,c);
|
||
set_pts=1;
|
||
len-=5;hdrlen-=5;
|
||
} else
|
||
if(pts_flags==3 && hdrlen>=10){
|
||
c=stream_read_char(demux->stream);
|
||
pts=read_mpeg_timestamp(demux->stream,c);
|
||
set_pts=1;
|
||
c=stream_read_char(demux->stream);
|
||
dts=read_mpeg_timestamp(demux->stream,c);
|
||
len-=10;hdrlen-=10;
|
||
}
|
||
len-=hdrlen;
|
||
if(hdrlen>0) {
|
||
if(!parse_ext2)
|
||
stream_skip(demux->stream,hdrlen); // skip header bytes
|
||
else if(hdrlen>=3) {
|
||
c=stream_read_char(demux->stream);
|
||
hdrlen--;
|
||
if(c != 0x1F && c!= 0x0F) { mp_msg(MSGT_DEMUX,MSGL_V,"demux_mpg: invalid pes_extension field \n"); return -1;}
|
||
if(c == 0x1F) {
|
||
//STD
|
||
stream_skip(demux->stream, 2);
|
||
hdrlen-=2;
|
||
}
|
||
c=stream_read_char(demux->stream); //pes_extension2 flag
|
||
hdrlen--;
|
||
if(c!=0x81) { mp_msg(MSGT_DEMUX,MSGL_V,"demux_mpg: unknown pes_extension2 format, len is > 1 \n"); return -1;}
|
||
c=stream_read_char(demux->stream); //pes_extension2 payload === substream id
|
||
if(c<0x55 || c>0x5F) { mp_msg(MSGT_DEMUX,MSGL_V,"demux_mpg: unknown vc1 substream_id: 0x%x \n", c); return -1;}
|
||
pes_ext2_subid=c;
|
||
}
|
||
}
|
||
|
||
if(id==0x1FD && pes_ext2_subid!=-1) {
|
||
//==== EVO VC1 STREAMS ===//
|
||
if(!demux->v_streams[pes_ext2_subid]) new_sh_video(demux,pes_ext2_subid);
|
||
if(demux->video->id==-1) demux->video->id=pes_ext2_subid;
|
||
if(demux->video->id==pes_ext2_subid){
|
||
ds=demux->video;
|
||
if(!ds->sh) ds->sh=demux->v_streams[pes_ext2_subid];
|
||
if(priv && ds->sh) {
|
||
sh_video_t *sh = (sh_video_t *)ds->sh;
|
||
sh->format = mmioFOURCC('W', 'V', 'C', '1');
|
||
}
|
||
}
|
||
}
|
||
//============== DVD Audio sub-stream ======================
|
||
if(id==0x1BD){
|
||
int aid, rawa52 = 0;
|
||
off_t tmppos;
|
||
unsigned int tmp;
|
||
|
||
tmppos = stream_tell(demux->stream);
|
||
tmp = stream_read_word(demux->stream);
|
||
stream_seek(demux->stream, tmppos);
|
||
/// vdr stores A52 without the 4 header bytes, so we have to check this condition first
|
||
if(tmp == 0x0B77) {
|
||
aid = 128;
|
||
rawa52 = 1;
|
||
}
|
||
else {
|
||
aid=stream_read_char(demux->stream);--len;
|
||
if(len<3) return -1; // invalid audio packet
|
||
}
|
||
|
||
// AID:
|
||
// 0x20..0x3F subtitle
|
||
// 0x80..0x9F AC3 audio
|
||
// 0xA0..0xBF PCM audio
|
||
|
||
if((aid & 0xE0) == 0x20){
|
||
// subtitle:
|
||
aid&=0x1F;
|
||
|
||
if(!demux->s_streams[aid]){
|
||
mp_msg(MSGT_DEMUX,MSGL_V,"==> Found subtitle: %d\n",aid);
|
||
new_sh_sub(demux, aid);
|
||
}
|
||
|
||
if(demux->sub->id > -1)
|
||
demux->sub->id &= 0x1F;
|
||
if(!dvdsub_lang && demux->sub->id == -1)
|
||
demux->sub->id = aid;
|
||
if(demux->sub->id==aid){
|
||
ds=demux->sub;
|
||
}
|
||
|
||
} else if((aid & 0xC0) == 0x80 || (aid & 0xE0) == 0x00 || (aid & 0xC0) == 0xC0) {
|
||
|
||
// aid=128+(aid&0x7F);
|
||
// aid=0x80..0xBF
|
||
new_audio_stream(demux, aid);
|
||
if(demux->audio->id==aid){
|
||
int type;
|
||
ds=demux->audio;
|
||
if(!ds->sh) ds->sh=demux->a_streams[aid];
|
||
// READ Packet: Skip additional audio header data:
|
||
if(!rawa52) {
|
||
c=stream_read_char(demux->stream);//num of frames
|
||
type=stream_read_char(demux->stream);//startpos hi
|
||
type=(type<<8)|stream_read_char(demux->stream);//startpos lo
|
||
// printf("\r[%02X][%04X]",c,type);
|
||
len-=3;
|
||
}
|
||
if((aid&0xE0)==0xA0 && len>=3){
|
||
unsigned char* hdr;
|
||
// save audio header as codecdata!
|
||
if(!((sh_audio_t*)(ds->sh))->codecdata_len){
|
||
((sh_audio_t*)(ds->sh))->codecdata=malloc(3);
|
||
((sh_audio_t*)(ds->sh))->codecdata_len=3;
|
||
}
|
||
hdr=((sh_audio_t*)(ds->sh))->codecdata;
|
||
// read LPCM header:
|
||
// emphasis[1], mute[1], rvd[1], frame number[5]:
|
||
hdr[0]=stream_read_char(demux->stream);
|
||
// printf(" [%01X:%02d]",c>>5,c&31);
|
||
// quantization[2],freq[2],rvd[1],channels[3]
|
||
hdr[1]=stream_read_char(demux->stream);
|
||
// printf("[%01X:%01X] ",c>>4,c&15);
|
||
// dynamic range control (0x80=off):
|
||
hdr[2]=stream_read_char(demux->stream);
|
||
// printf("[%02X] ",c);
|
||
len-=3;
|
||
if(len<=0) mp_msg(MSGT_DEMUX,MSGL_V,"End of packet while searching for PCM header\n");
|
||
}
|
||
// printf(" \n");
|
||
} // if(demux->audio->id==aid)
|
||
|
||
} else mp_msg(MSGT_DEMUX,MSGL_V,"Unknown 0x1BD substream: 0x%02X \n",aid);
|
||
|
||
} //if(id==0x1BD)
|
||
|
||
} else {
|
||
if(c!=0x0f){
|
||
mp_msg(MSGT_DEMUX,MSGL_V," {ERROR5,c=%d} \n",c);
|
||
return -1; // invalid packet !!!!!!
|
||
}
|
||
}
|
||
if(mpeg_pts_error) mp_msg(MSGT_DEMUX,MSGL_V," {PTS_err:%d} \n",mpeg_pts_error);
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG3," => len=%d\n",len);
|
||
|
||
// if(len<=0 || len>MAX_PS_PACKETSIZE) return -1; // Invalid packet size
|
||
if(len<=0 || len>MAX_PS_PACKETSIZE){
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG2,"Invalid PS data len: %d\n",len);
|
||
return -1; // invalid packet !!!!!!
|
||
}
|
||
|
||
if(id>=0x1C0 && id<=0x1DF){
|
||
// mpeg audio
|
||
int aid=id-0x1C0;
|
||
new_audio_stream(demux, aid);
|
||
if(demux->audio->id==aid){
|
||
ds=demux->audio;
|
||
if(!ds->sh) ds->sh=demux->a_streams[aid];
|
||
if(priv && ds->sh) {
|
||
sh_audio_t *sh = (sh_audio_t *)ds->sh;
|
||
if(priv->es_map[id - 0x1B0])
|
||
sh->format = priv->es_map[id - 0x1B0];
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG2,"ASSIGNED TO STREAM %d CODEC %x\n", id, priv->es_map[id - 0x1B0]);
|
||
}
|
||
}
|
||
} else
|
||
if(id>=0x1E0 && id<=0x1EF){
|
||
// mpeg video
|
||
int aid=id-0x1E0;
|
||
if(!demux->v_streams[aid]) new_sh_video(demux,aid);
|
||
if(demux->video->id==-1) demux->video->id=aid;
|
||
if(demux->video->id==aid){
|
||
ds=demux->video;
|
||
if(!ds->sh) ds->sh=demux->v_streams[aid];
|
||
if(priv && ds->sh) {
|
||
sh_video_t *sh = (sh_video_t *)ds->sh;
|
||
if(priv->es_map[id - 0x1B0]) {
|
||
sh->format = priv->es_map[id - 0x1B0];
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG2,"ASSIGNED TO STREAM %d CODEC %x\n", id, priv->es_map[id - 0x1B0]);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
if(ds){
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG2,"DEMUX_MPG: Read %d data bytes from packet %04X\n",len,id);
|
||
// printf("packet start = 0x%X \n",stream_tell(demux->stream)-packet_start_pos);
|
||
|
||
dp=new_demux_packet(len);
|
||
if(!dp) {
|
||
mp_dbg(MSGT_DEMUX,MSGL_ERR,"DEMUX_MPG ERROR: couldn't create demux_packet(%d bytes)\n",len);
|
||
stream_skip(demux->stream,len);
|
||
return 0;
|
||
}
|
||
l = stream_read(demux->stream,dp->buffer,len);
|
||
if(l<len)
|
||
resize_demux_packet(dp, l);
|
||
len = l;
|
||
dp->pts=pts/90000.0f;
|
||
dp->pos=demux->filepos;
|
||
/*
|
||
workaround:
|
||
set dp->stream_pts only when feeding the video stream, or strangely interleaved files
|
||
(such as SWIII) will show strange alternations in the stream time, wildly going
|
||
back and forth
|
||
*/
|
||
if(ds == demux->video && stream_control(demux->stream, STREAM_CTRL_GET_CURRENT_TIME,(void *)&stream_pts)!=STREAM_UNSUPORTED)
|
||
dp->stream_pts = stream_pts;
|
||
ds_add_packet(ds,dp);
|
||
if (demux->priv && set_pts) ((mpg_demuxer_t*)demux->priv)->last_pts = pts/90000.0f;
|
||
// if(ds==demux->sub) parse_dvdsub(ds->last->buffer,ds->last->len);
|
||
return 1;
|
||
}
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG2,"DEMUX_MPG: Skipping %d data bytes from packet %04X\n",len,id);
|
||
if(len<=2356) stream_skip(demux->stream,len);
|
||
return 0;
|
||
}
|
||
|
||
static int num_elementary_packets100=0;
|
||
static int num_elementary_packets101=0;
|
||
static int num_elementary_packets12x=0;
|
||
static int num_elementary_packets1B6=0;
|
||
static int num_elementary_packetsPES=0;
|
||
static int num_h264_slice=0; //combined slice
|
||
static int num_h264_dpa=0; //DPA Slice
|
||
static int num_h264_dpb=0; //DPB Slice
|
||
static int num_h264_dpc=0; //DPC Slice
|
||
static int num_h264_idr=0; //IDR Slice
|
||
static int num_h264_sps=0;
|
||
static int num_h264_pps=0;
|
||
|
||
static int num_mp3audio_packets=0;
|
||
|
||
static void clear_stats()
|
||
{
|
||
num_elementary_packets100=0;
|
||
num_elementary_packets101=0;
|
||
num_elementary_packets1B6=0;
|
||
num_elementary_packets12x=0;
|
||
num_elementary_packetsPES=0;
|
||
num_h264_slice=0; //combined slice
|
||
num_h264_dpa=0; //DPA Slice
|
||
num_h264_dpb=0; //DPB Slice
|
||
num_h264_dpc=0; //DPC Slice
|
||
num_h264_idr=0; //IDR Slice
|
||
num_h264_sps=0;
|
||
num_h264_pps=0;
|
||
num_mp3audio_packets=0;
|
||
}
|
||
|
||
//assumes demuxer->synced < 2
|
||
static inline void update_stats(int head)
|
||
{
|
||
if(head==0x1B6) ++num_elementary_packets1B6;
|
||
else if(head==0x100) ++num_elementary_packets100;
|
||
else if(head==0x101) ++num_elementary_packets101;
|
||
else if(head==0x1BD || (0x1C0<=head && head<=0x1EF))
|
||
num_elementary_packetsPES++;
|
||
else if(head>=0x120 && head<=0x12F) ++num_elementary_packets12x;
|
||
else if(head>=0x100 && head<0x1B0)
|
||
{
|
||
if((head&~0x60) == 0x101) ++num_h264_slice;
|
||
else if((head&~0x60) == 0x102) ++num_h264_dpa;
|
||
else if((head&~0x60) == 0x103) ++num_h264_dpb;
|
||
else if((head&~0x60) == 0x104) ++num_h264_dpc;
|
||
else if((head&~0x60) == 0x105 && head != 0x105) ++num_h264_idr;
|
||
else if((head&~0x60) == 0x107 && head != 0x107) ++num_h264_sps;
|
||
else if((head&~0x60) == 0x108 && head != 0x108) ++num_h264_pps;
|
||
}
|
||
}
|
||
|
||
static int demux_mpg_probe(demuxer_t *demuxer) {
|
||
int pes=1;
|
||
int tmp;
|
||
off_t tmppos;
|
||
int file_format = DEMUXER_TYPE_UNKNOWN;
|
||
|
||
tmppos=stream_tell(demuxer->stream);
|
||
tmp=stream_read_dword(demuxer->stream);
|
||
if(tmp==0x1E0 || tmp==0x1C0) {
|
||
tmp=stream_read_word(demuxer->stream);
|
||
if(tmp>1 && tmp<=2048) pes=0; // demuxer->synced=3; // PES...
|
||
}
|
||
stream_seek(demuxer->stream,tmppos);
|
||
|
||
clear_stats();
|
||
|
||
if(demux_mpg_open(demuxer))
|
||
file_format=DEMUXER_TYPE_MPEG_PS;
|
||
else {
|
||
mp_msg(MSGT_DEMUX,MSGL_V,"MPEG packet stats: p100: %d p101: %d p1B6: %d p12x: %d sli: %d a: %d b: %d c: %d idr: %d sps: %d pps: %d PES: %d MP3: %d, synced: %d\n",
|
||
num_elementary_packets100,num_elementary_packets101,
|
||
num_elementary_packets1B6,num_elementary_packets12x,
|
||
num_h264_slice, num_h264_dpa,
|
||
num_h264_dpb, num_h264_dpc=0,
|
||
num_h264_idr, num_h264_sps=0,
|
||
num_h264_pps,
|
||
num_elementary_packetsPES,num_mp3audio_packets, demuxer->synced);
|
||
|
||
//MPEG packet stats: p100: 458 p101: 458 PES: 0 MP3: 1103 (.m2v)
|
||
if(num_mp3audio_packets>50 && num_mp3audio_packets>2*num_elementary_packets100
|
||
&& abs(num_elementary_packets100-num_elementary_packets101)>2)
|
||
return file_format;
|
||
|
||
// some hack to get meaningfull error messages to our unhappy users:
|
||
if(num_elementary_packets100>=2 && num_elementary_packets101>=2 &&
|
||
abs(num_elementary_packets101+8-num_elementary_packets100)<16) {
|
||
if(num_elementary_packetsPES>=4 && num_elementary_packetsPES>=num_elementary_packets100-4) {
|
||
return file_format;
|
||
}
|
||
file_format=DEMUXER_TYPE_MPEG_ES; // <-- hack is here :)
|
||
} else
|
||
// fuzzy mpeg4-es detection. do NOT enable without heavy testing of mpeg formats detection!
|
||
if(num_elementary_packets1B6>3 && num_elementary_packets12x>=1 &&
|
||
num_elementary_packetsPES==0 && num_elementary_packets100<=num_elementary_packets12x &&
|
||
demuxer->synced<2) {
|
||
file_format=DEMUXER_TYPE_MPEG4_ES;
|
||
} else
|
||
// fuzzy h264-es detection. do NOT enable without heavy testing of mpeg formats detection!
|
||
if((num_h264_slice>3 || (num_h264_dpa>3 && num_h264_dpb>3 && num_h264_dpc>3)) &&
|
||
/* FIXME num_h264_sps>=1 && */ num_h264_pps>=1 && num_h264_idr>=1 &&
|
||
num_elementary_packets1B6==0 && num_elementary_packetsPES==0 &&
|
||
demuxer->synced<2) {
|
||
file_format=DEMUXER_TYPE_H264_ES;
|
||
} else
|
||
{
|
||
if(demuxer->synced==2)
|
||
mp_msg(MSGT_DEMUXER,MSGL_ERR,"MPEG: " MSGTR_MissingVideoStreamBug);
|
||
else
|
||
mp_msg(MSGT_DEMUXER,MSGL_V,MSGTR_NotSystemStream);
|
||
}
|
||
}
|
||
//FIXME this shouldn't be necessary
|
||
stream_seek(demuxer->stream,tmppos);
|
||
return file_format;
|
||
}
|
||
|
||
static int demux_mpg_es_fill_buffer(demuxer_t *demux, demux_stream_t *ds){
|
||
// Elementary video stream
|
||
if(demux->stream->eof) return 0;
|
||
demux->filepos=stream_tell(demux->stream);
|
||
ds_read_packet(demux->video,demux->stream,STREAM_BUFFER_SIZE,0,demux->filepos,0);
|
||
return 1;
|
||
}
|
||
|
||
/**
|
||
* \brief discard until 0x100 header and return a filled buffer
|
||
* \param b buffer-end pointer
|
||
* \param pos current pos in stream, negative since b points to end of buffer
|
||
* \param s stream to read from
|
||
* \return new position, differs from original pos when eof hit and thus
|
||
* b was modified to point to the new end of buffer
|
||
*/
|
||
static int find_end(unsigned char **b, int pos, stream_t *s) {
|
||
register int state = 0xffffffff;
|
||
unsigned char *buf = *b;
|
||
int start = pos;
|
||
int read, unused;
|
||
// search already read part
|
||
while (state != 0x100 && pos) {
|
||
state = state << 8 | buf[pos++];
|
||
}
|
||
// continue search in stream
|
||
while (state != 0x100) {
|
||
register int c = stream_read_char(s);
|
||
if (c < 0) break;
|
||
state = state << 8 | c;
|
||
}
|
||
// modify previous header (from 0x1bc or 0x1bf to 0x100)
|
||
buf[start++] = 0;
|
||
// copy remaining buffer part to current pos
|
||
memmove(&buf[start], &buf[pos], -pos);
|
||
unused = start + -pos; // -unused bytes in buffer
|
||
read = stream_read(s, &buf[unused], -unused);
|
||
unused += read;
|
||
// fix buffer so it ends at pos == 0 (eof case)
|
||
*b = &buf[unused];
|
||
start -= unused;
|
||
return start;
|
||
}
|
||
|
||
/**
|
||
* This format usually uses an insane bitrate, which makes this function
|
||
* performance-critical!
|
||
* Be sure to benchmark any changes with different compiler versions.
|
||
*/
|
||
static int demux_mpg_gxf_fill_buffer(demuxer_t *demux, demux_stream_t *ds) {
|
||
demux_packet_t *pack;
|
||
int len;
|
||
demux->filepos = stream_tell(demux->stream);
|
||
pack = new_demux_packet(STREAM_BUFFER_SIZE);
|
||
len = stream_read(demux->stream, pack->buffer, STREAM_BUFFER_SIZE);
|
||
if (len <= 0)
|
||
{
|
||
free_demux_packet(pack);
|
||
return 0;
|
||
}
|
||
{
|
||
register uint32_t state = (uint32_t)demux->priv;
|
||
register int pos = -len;
|
||
unsigned char *buf = &pack->buffer[len];
|
||
do {
|
||
state = state << 8 | buf[pos];
|
||
if (unlikely((state | 3) == 0x1bf))
|
||
pos = find_end(&buf, pos, demux->stream);
|
||
} while (++pos < 0);
|
||
demux->priv = (void *)state;
|
||
len = buf - pack->buffer;
|
||
}
|
||
if (len < STREAM_BUFFER_SIZE)
|
||
resize_demux_packet(pack, len);
|
||
ds_add_packet(ds, pack);
|
||
return 1;
|
||
}
|
||
|
||
int demux_mpg_fill_buffer(demuxer_t *demux, demux_stream_t *ds){
|
||
unsigned int head=0;
|
||
int skipped=0;
|
||
int max_packs=256; // 512kbyte
|
||
int ret=0;
|
||
|
||
// System stream
|
||
do{
|
||
demux->filepos=stream_tell(demux->stream);
|
||
#if 1
|
||
//lame workaround: this is needed to show the progress bar when playing dvdnav://
|
||
//(ths poor guy doesn't know teh length of the stream at startup)
|
||
demux->movi_end = demux->stream->end_pos;
|
||
#endif
|
||
head=stream_read_dword(demux->stream);
|
||
if((head&0xFFFFFF00)!=0x100){
|
||
// sync...
|
||
demux->filepos-=skipped;
|
||
while(1){
|
||
int c=stream_read_char(demux->stream);
|
||
if(c<0) break; //EOF
|
||
head<<=8;
|
||
if(head!=0x100){
|
||
head|=c;
|
||
if(mp_check_mp3_header(head)) ++num_mp3audio_packets;
|
||
++skipped; //++demux->filepos;
|
||
continue;
|
||
}
|
||
head|=c;
|
||
break;
|
||
}
|
||
demux->filepos+=skipped;
|
||
}
|
||
if(stream_eof(demux->stream)) break;
|
||
// sure: head=0x000001XX
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG4,"*** head=0x%X\n",head);
|
||
if(demux->synced==0){
|
||
if(head==0x1BA) demux->synced=1; //else
|
||
// if(head==0x1BD || (head>=0x1C0 && head<=0x1EF)) demux->synced=3; // PES?
|
||
} else
|
||
if(demux->synced==1){
|
||
if(head==0x1BB || head==0x1BD || (head>=0x1C0 && head<=0x1EF)){
|
||
demux->synced=2;
|
||
mp_msg(MSGT_DEMUX,MSGL_V,"system stream synced at 0x%"PRIX64" (%"PRId64")!\n",(int64_t)demux->filepos,(int64_t)demux->filepos);
|
||
num_elementary_packets100=0; // requires for re-sync!
|
||
num_elementary_packets101=0; // requires for re-sync!
|
||
} else demux->synced=0;
|
||
} // else
|
||
if(demux->synced>=2){
|
||
ret=demux_mpg_read_packet(demux,head);
|
||
if(!ret)
|
||
if(--max_packs==0){
|
||
demux->stream->eof=1;
|
||
mp_msg(MSGT_DEMUX,MSGL_ERR,MSGTR_DoesntContainSelectedStream);
|
||
return 0;
|
||
}
|
||
if(demux->synced==3) demux->synced=(ret==1)?2:0; // PES detect
|
||
} else {
|
||
update_stats(head);
|
||
if(head>=0x100 && head<0x1B0)
|
||
mp_msg(MSGT_DEMUX,MSGL_DBG3,"Opps... elementary video packet found: %03X\n",head);
|
||
else if((head>=0x1C0 && head<0x1F0) || head==0x1BD)
|
||
mp_msg(MSGT_DEMUX,MSGL_DBG3,"Opps... PES packet found: %03X\n",head);
|
||
|
||
if(((num_elementary_packets100>50 && num_elementary_packets101>50) ||
|
||
(num_elementary_packetsPES>50)) && skipped>4000000){
|
||
mp_msg(MSGT_DEMUX,MSGL_V,"sync_mpeg_ps: seems to be ES/PES stream...\n");
|
||
demux->stream->eof=1;
|
||
break;
|
||
}
|
||
if(num_mp3audio_packets>100 && num_elementary_packets100<10){
|
||
mp_msg(MSGT_DEMUX,MSGL_V,"sync_mpeg_ps: seems to be MP3 stream...\n");
|
||
demux->stream->eof=1;
|
||
break;
|
||
}
|
||
}
|
||
} while(ret!=1);
|
||
mp_dbg(MSGT_DEMUX,MSGL_DBG2,"demux: %d bad bytes skipped\n",skipped);
|
||
if(demux->stream->eof){
|
||
mp_msg(MSGT_DEMUX,MSGL_V,"MPEG Stream reached EOF\n");
|
||
return 0;
|
||
}
|
||
return 1;
|
||
}
|
||
|
||
extern void skip_audio_frame(sh_audio_t *sh_audio);
|
||
|
||
void demux_seek_mpg(demuxer_t *demuxer,float rel_seek_secs,float audio_delay, int flags){
|
||
demux_stream_t *d_audio=demuxer->audio;
|
||
demux_stream_t *d_video=demuxer->video;
|
||
sh_audio_t *sh_audio=d_audio->sh;
|
||
sh_video_t *sh_video=d_video->sh;
|
||
mpg_demuxer_t *mpg_d=(mpg_demuxer_t*)demuxer->priv;
|
||
int precision = 1;
|
||
float oldpts = 0;
|
||
off_t oldpos = demuxer->filepos;
|
||
float newpts = 0;
|
||
off_t newpos = (flags & 1) ? demuxer->movi_start : oldpos;
|
||
|
||
if(mpg_d)
|
||
oldpts = mpg_d->last_pts;
|
||
newpts = (flags & 1) ? 0.0 : oldpts;
|
||
//================= seek in MPEG ==========================
|
||
//calculate the pts to seek to
|
||
if(flags & 2) {
|
||
if (mpg_d && mpg_d->first_to_final_pts_len > 0.0)
|
||
newpts += mpg_d->first_to_final_pts_len * rel_seek_secs;
|
||
else
|
||
newpts += rel_seek_secs * (demuxer->movi_end - demuxer->movi_start) * oldpts / oldpos;
|
||
} else
|
||
newpts += rel_seek_secs;
|
||
if (newpts < 0) newpts = 0;
|
||
|
||
if(flags&2){
|
||
// float seek 0..1
|
||
newpos+=(demuxer->movi_end-demuxer->movi_start)*rel_seek_secs;
|
||
} else {
|
||
// time seek (secs)
|
||
if (mpg_d && mpg_d->has_valid_timestamps) {
|
||
if (mpg_d->first_to_final_pts_len > 0.0)
|
||
newpos += rel_seek_secs * (demuxer->movi_end - demuxer->movi_start) / mpg_d->first_to_final_pts_len;
|
||
else if (oldpts > 0.0)
|
||
newpos += rel_seek_secs * (oldpos - demuxer->movi_start) / oldpts;
|
||
} else if(!sh_video || !sh_video->i_bps) // unspecified or VBR
|
||
newpos+=2324*75*rel_seek_secs; // 174.3 kbyte/sec
|
||
else
|
||
newpos+=sh_video->i_bps*rel_seek_secs;
|
||
}
|
||
|
||
while (1) {
|
||
if(newpos<demuxer->movi_start){
|
||
if(demuxer->stream->type!=STREAMTYPE_VCD) demuxer->movi_start=0; // for VCD
|
||
if(newpos<demuxer->movi_start) newpos=demuxer->movi_start;
|
||
}
|
||
|
||
#ifdef _LARGEFILE_SOURCE
|
||
newpos&=~((long long)STREAM_BUFFER_SIZE-1); /* sector boundary */
|
||
#else
|
||
newpos&=~(STREAM_BUFFER_SIZE-1); /* sector boundary */
|
||
#endif
|
||
stream_seek(demuxer->stream,newpos);
|
||
|
||
// re-sync video:
|
||
videobuf_code_len=0; // reset ES stream buffer
|
||
|
||
ds_fill_buffer(d_video);
|
||
if(sh_audio){
|
||
ds_fill_buffer(d_audio);
|
||
}
|
||
|
||
while(1){
|
||
int i;
|
||
if(sh_audio && !d_audio->eof && d_video->pts && d_audio->pts){
|
||
float a_pts=d_audio->pts;
|
||
a_pts+=(ds_tell_pts(d_audio)-sh_audio->a_in_buffer_len)/(float)sh_audio->i_bps;
|
||
if(d_video->pts>a_pts){
|
||
skip_audio_frame(sh_audio); // sync audio
|
||
continue;
|
||
}
|
||
}
|
||
if(!sh_video) break;
|
||
i=sync_video_packet(d_video);
|
||
if(sh_video->format == 0x10000004) { //mpeg4
|
||
if(i==0x1B6) { //vop (frame) startcode
|
||
int pos = videobuf_len;
|
||
if(!read_video_packet(d_video)) break; // EOF
|
||
if((videobuffer[pos+4] & 0x3F) == 0) break; //I-frame
|
||
}
|
||
} else if(sh_video->format == 0x10000005){ //h264
|
||
if((i & ~0x60) == 0x105) break;
|
||
} else { //default mpeg1/2
|
||
if(i==0x1B3 || i==0x1B8) break; // found it!
|
||
}
|
||
if(!i || !skip_video_packet(d_video)) break; // EOF?
|
||
}
|
||
if(!mpg_d)
|
||
break;
|
||
if (!precision || abs(newpts - mpg_d->last_pts) < 0.5 || (mpg_d->last_pts == oldpts)) break;
|
||
if ((newpos - oldpos) * (mpg_d->last_pts - oldpts) < 0) { // invalid timestamps
|
||
mpg_d->has_valid_timestamps = 0;
|
||
break;
|
||
}
|
||
precision--;
|
||
//prepare another seek because we are off by more than 0.5s
|
||
if(mpg_d) {
|
||
newpos += (newpts - mpg_d->last_pts) * (newpos - oldpos) / (mpg_d->last_pts - oldpts);
|
||
ds_free_packs(d_audio);
|
||
ds_free_packs(d_video);
|
||
ds_free_packs(demuxer->sub);
|
||
demuxer->stream->eof=0; // clear eof flag
|
||
d_video->eof=0;
|
||
d_audio->eof=0;
|
||
}
|
||
}
|
||
}
|
||
|
||
int demux_mpg_control(demuxer_t *demuxer,int cmd, void *arg){
|
||
mpg_demuxer_t *mpg_d=(mpg_demuxer_t*)demuxer->priv;
|
||
|
||
switch(cmd) {
|
||
case DEMUXER_CTRL_GET_TIME_LENGTH:
|
||
if(stream_control(demuxer->stream, STREAM_CTRL_GET_TIME_LENGTH, arg) != STREAM_UNSUPORTED) {
|
||
mp_msg(MSGT_DEMUXER,MSGL_DBG2,"\r\nDEMUX_MPG_CTRL, (%.3lf)\r\n", *((double*)arg));
|
||
return DEMUXER_CTRL_GUESS;
|
||
}
|
||
if (mpg_d && mpg_d->has_valid_timestamps) {
|
||
*((double *)arg)=(double)mpg_d->first_to_final_pts_len;
|
||
return DEMUXER_CTRL_OK;
|
||
}
|
||
return DEMUXER_CTRL_DONTKNOW;
|
||
|
||
case DEMUXER_CTRL_GET_PERCENT_POS:
|
||
if (mpg_d && mpg_d->has_valid_timestamps && mpg_d->first_to_final_pts_len > 0.0) {
|
||
*((int *)arg)=(int)(100 * (mpg_d->last_pts-mpg_d->first_pts) / mpg_d->first_to_final_pts_len);
|
||
return DEMUXER_CTRL_OK;
|
||
}
|
||
return DEMUXER_CTRL_DONTKNOW;
|
||
|
||
case DEMUXER_CTRL_SWITCH_AUDIO:
|
||
if(! (mpg_d && mpg_d->num_a_streams > 1 && demuxer->audio && demuxer->audio->sh))
|
||
return DEMUXER_CTRL_NOTIMPL;
|
||
else {
|
||
demux_stream_t *d_audio = demuxer->audio;
|
||
sh_audio_t *sh_audio = d_audio->sh;
|
||
sh_audio_t *sh_a = sh_audio;
|
||
int i;
|
||
if(!sh_audio)
|
||
return DEMUXER_CTRL_NOTIMPL;
|
||
if (*((int*)arg) < 0)
|
||
{
|
||
for (i = 0; i < mpg_d->num_a_streams; i++) {
|
||
if (d_audio->id == mpg_d->a_stream_ids[i]) break;
|
||
}
|
||
i = (i+1) % mpg_d->num_a_streams;
|
||
sh_a = (sh_audio_t*)demuxer->a_streams[mpg_d->a_stream_ids[i]];
|
||
}
|
||
else {
|
||
for (i = 0; i < mpg_d->num_a_streams; i++)
|
||
if (*((int*)arg) == mpg_d->a_stream_ids[i]) break;
|
||
if (i < mpg_d->num_a_streams)
|
||
sh_a = (sh_audio_t*)demuxer->a_streams[*((int*)arg)];
|
||
}
|
||
if (i < mpg_d->num_a_streams && d_audio->id != mpg_d->a_stream_ids[i]) {
|
||
d_audio->id = mpg_d->a_stream_ids[i];
|
||
d_audio->sh = sh_a;
|
||
ds_free_packs(d_audio);
|
||
}
|
||
}
|
||
*((int*)arg) = demuxer->audio->id;
|
||
return DEMUXER_CTRL_OK;
|
||
|
||
default:
|
||
return DEMUXER_CTRL_NOTIMPL;
|
||
}
|
||
}
|
||
|
||
|
||
static int demux_mpg_pes_probe(demuxer_t *demuxer) {
|
||
demuxer->synced = 3;
|
||
return (demux_mpg_probe(demuxer) == DEMUXER_TYPE_MPEG_PS) ? DEMUXER_TYPE_MPEG_PES : 0;
|
||
}
|
||
|
||
|
||
static demuxer_t* demux_mpg_es_open(demuxer_t* demuxer)
|
||
{
|
||
sh_video_t *sh_video=NULL;
|
||
|
||
demuxer->audio->sh = NULL; // ES streams has no audio channel
|
||
demuxer->video->sh = new_sh_video(demuxer,0); // create dummy video stream header, id=0
|
||
sh_video=demuxer->video->sh;sh_video->ds=demuxer->video;
|
||
|
||
return demuxer;
|
||
}
|
||
|
||
static demuxer_t *demux_mpg_gxf_open(demuxer_t *demuxer) {
|
||
demuxer->audio->sh = NULL;
|
||
demuxer->video->sh = new_sh_video(demuxer,0);
|
||
((sh_video_t *)demuxer->video->sh)->ds = demuxer->video;
|
||
demuxer->priv = (void *) 0xffffffff;
|
||
return demuxer;
|
||
}
|
||
|
||
static demuxer_t* demux_mpg_ps_open(demuxer_t* demuxer)
|
||
{
|
||
sh_audio_t *sh_audio=NULL;
|
||
sh_video_t *sh_video=NULL;
|
||
|
||
sh_video=demuxer->video->sh;sh_video->ds=demuxer->video;
|
||
|
||
if(demuxer->audio->id!=-2) {
|
||
if(!ds_fill_buffer(demuxer->audio)){
|
||
mp_msg(MSGT_DEMUXER,MSGL_INFO,"MPEG: " MSGTR_MissingAudioStream);
|
||
demuxer->audio->sh=NULL;
|
||
} else {
|
||
sh_audio=demuxer->audio->sh;sh_audio->ds=demuxer->audio;
|
||
}
|
||
}
|
||
|
||
if(!sh_video->format && ps_probe > 0) {
|
||
int head;
|
||
off_t pos = stream_tell(demuxer->stream);
|
||
|
||
clear_stats();
|
||
do {
|
||
head=sync_video_packet(demuxer->video);
|
||
if(!head) break;
|
||
update_stats(head);
|
||
skip_video_packet(demuxer->video);
|
||
} while(stream_tell(demuxer->stream) < pos + ps_probe && !demuxer->stream->eof);
|
||
|
||
ds_free_packs(demuxer->video);
|
||
demuxer->stream->eof=0;
|
||
stream_seek(demuxer->stream, pos);
|
||
mp_msg(MSGT_DEMUX,MSGL_INFO,"MPEG packet stats: p100: %d p101: %d p1B6: %d p12x: %d sli: %d a: %d b: %d c: %d idr: %d sps: %d pps: %d\n",
|
||
num_elementary_packets100, num_elementary_packets101,
|
||
num_elementary_packets1B6, num_elementary_packets12x,
|
||
num_h264_slice, num_h264_dpa, num_h264_dpb, num_h264_dpc,
|
||
num_h264_idr, num_h264_sps, num_h264_pps);
|
||
|
||
if(num_elementary_packets1B6>3 && num_elementary_packets12x>=1 &&
|
||
num_elementary_packets100<=num_elementary_packets12x)
|
||
sh_video->format = 0x10000004;
|
||
else if((num_h264_slice>3 || (num_h264_dpa>3 && num_h264_dpb>3 && num_h264_dpc>3)) &&
|
||
num_h264_sps>=1 && num_h264_pps>=1 && num_h264_idr>=1 &&
|
||
num_elementary_packets1B6==0)
|
||
sh_video->format = 0x10000005;
|
||
else sh_video->format = 0x10000002;
|
||
}
|
||
|
||
return demuxer;
|
||
}
|
||
|
||
|
||
demuxer_desc_t demuxer_desc_mpeg_ps = {
|
||
"MPEG PS demuxer",
|
||
"mpegps",
|
||
"MPEG-PS",
|
||
"Arpi?",
|
||
"Mpeg",
|
||
DEMUXER_TYPE_MPEG_PS,
|
||
0, // unsafe autodetect
|
||
demux_mpg_probe,
|
||
demux_mpg_fill_buffer,
|
||
demux_mpg_ps_open,
|
||
demux_close_mpg,
|
||
demux_seek_mpg,
|
||
demux_mpg_control,
|
||
};
|
||
|
||
|
||
demuxer_desc_t demuxer_desc_mpeg_pes = {
|
||
"MPEG PES demuxer",
|
||
"mpegpes",
|
||
"MPEG-PES",
|
||
"Arpi?",
|
||
"Mpeg",
|
||
DEMUXER_TYPE_MPEG_PES,
|
||
0, // unsafe autodetect
|
||
demux_mpg_pes_probe,
|
||
demux_mpg_fill_buffer,
|
||
demux_mpg_ps_open,
|
||
demux_close_mpg,
|
||
demux_seek_mpg,
|
||
demux_mpg_control,
|
||
};
|
||
|
||
|
||
demuxer_desc_t demuxer_desc_mpeg_gxf = {
|
||
"MPEG ES in GXF demuxer",
|
||
"mpeggxf",
|
||
"MPEG-ES in GXF",
|
||
"Reimar D<>ffinger",
|
||
"Mpeg",
|
||
DEMUXER_TYPE_MPEG_GXF,
|
||
0, // hack autodetection
|
||
NULL,
|
||
demux_mpg_gxf_fill_buffer,
|
||
demux_mpg_gxf_open,
|
||
NULL,
|
||
NULL,
|
||
NULL
|
||
};
|
||
|
||
demuxer_desc_t demuxer_desc_mpeg_es = {
|
||
"MPEG ES demuxer",
|
||
"mpeges",
|
||
"MPEG-ES",
|
||
"Arpi?",
|
||
"Mpeg",
|
||
DEMUXER_TYPE_MPEG_ES,
|
||
0, // hack autodetection
|
||
NULL,
|
||
demux_mpg_es_fill_buffer,
|
||
demux_mpg_es_open,
|
||
demux_close_mpg,
|
||
demux_seek_mpg,
|
||
demux_mpg_control,
|
||
};
|
||
|
||
|
||
demuxer_desc_t demuxer_desc_mpeg4_es = {
|
||
"MPEG4 ES demuxer",
|
||
"mpeg4es",
|
||
"MPEG-ES",
|
||
"Arpi?",
|
||
"Mpeg",
|
||
DEMUXER_TYPE_MPEG4_ES,
|
||
0, // hack autodetection
|
||
NULL,
|
||
demux_mpg_es_fill_buffer,
|
||
demux_mpg_es_open,
|
||
demux_close_mpg,
|
||
demux_seek_mpg,
|
||
demux_mpg_control,
|
||
};
|
||
|
||
|
||
demuxer_desc_t demuxer_desc_h264_es = {
|
||
"H.264 ES demuxer",
|
||
"h264es",
|
||
"H264-ES",
|
||
"Arpi?",
|
||
"Mpeg",
|
||
DEMUXER_TYPE_H264_ES,
|
||
0, // hack autodetection
|
||
NULL,
|
||
demux_mpg_es_fill_buffer,
|
||
demux_mpg_es_open,
|
||
demux_close_mpg,
|
||
demux_seek_mpg,
|
||
demux_mpg_control,
|
||
};
|