Commit d5be317c authored by Måns Rullgård's avatar Måns Rullgård Committed by Michael Niedermayer

H.264 and Vorbis support in matroska patch by (Måns Rullgård |mru inprovide com)

Originally committed as revision 4014 to svn://svn.ffmpeg.org/ffmpeg/trunk
parent 26165f99
......@@ -186,6 +186,7 @@ typedef enum {
#define MATROSKA_CODEC_ID_VIDEO_MPEG4_SP "V_MPEG4/ISO/SP"
#define MATROSKA_CODEC_ID_VIDEO_MPEG4_ASP "V_MPEG4/ISO/ASP"
#define MATROSKA_CODEC_ID_VIDEO_MPEG4_AP "V_MPEG4/ISO/AP"
#define MATROSKA_CODEC_ID_VIDEO_MPEG4_AVC "V_MPEG4/ISO/AVC"
#define MATROSKA_CODEC_ID_VIDEO_MSMPEG4V3 "V_MPEG4/MS/V3"
#define MATROSKA_CODEC_ID_VIDEO_MPEG1 "V_MPEG1"
#define MATROSKA_CODEC_ID_VIDEO_MPEG2 "V_MPEG2"
......@@ -2151,6 +2152,8 @@ matroska_read_header (AVFormatContext *s,
enum CodecID codec_id;
MatroskaTrack *track;
AVStream *st;
void *extradata = NULL;
int extradata_size = 0;
for (i = 0; i < matroska->num_tracks; i++) {
track = matroska->tracks[i];
......@@ -2183,6 +2186,9 @@ matroska_read_header (AVFormatContext *s,
!strcmp(track->codec_id,
MATROSKA_CODEC_ID_VIDEO_MPEG4_AP))
codec_id = CODEC_ID_MPEG4;
else if (!strcmp(track->codec_id,
MATROSKA_CODEC_ID_VIDEO_MPEG4_AVC))
codec_id = CODEC_ID_H264;
/* else if (!strcmp(track->codec_id, */
/* MATROSKA_CODEC_ID_VIDEO_UNCOMPRESSED)) */
/* codec_id = CODEC_ID_???; */
......@@ -2236,12 +2242,50 @@ matroska_read_header (AVFormatContext *s,
/* MATROSKA_CODEC_ID_AUDIO_DTS)) */
/* codec_id = CODEC_ID_DTS; */
else if (!strcmp(track->codec_id,
MATROSKA_CODEC_ID_AUDIO_VORBIS))
MATROSKA_CODEC_ID_AUDIO_VORBIS)) {
u_char *p = track->codec_priv, *cdp;
int cps = track->codec_priv_size;
int nf, s[3], cds;
int i;
nf = *p++;
cps--;
if(nf != 2)
continue;
for(i = 0; i < 2; i++){
int xv;
s[i] = 0;
do {
xv = *p++;
s[i] += xv;
cps--;
} while(xv == 255);
}
s[2] = cps - s[0] - s[1];
cds = cps + 6;
extradata = cdp = av_malloc(cds);
if(extradata == NULL)
return AVERROR_NOMEM;
extradata_size = cds;
for(i = 0; i < 3; i++){
*cdp++ = s[i] >> 8;
*cdp++ = s[i] & 0xff;
memcpy(cdp, p, s[i]);
cdp += s[i];
p += s[i];
cps -= s[i];
}
codec_id = CODEC_ID_VORBIS;
else if (!strcmp(track->codec_id,
MATROSKA_CODEC_ID_AUDIO_MPEG2) ||
!strcmp(track->codec_id,
MATROSKA_CODEC_ID_AUDIO_MPEG4))
} else if (!strcmp(track->codec_id,
MATROSKA_CODEC_ID_AUDIO_MPEG2) ||
!strcmp(track->codec_id,
MATROSKA_CODEC_ID_AUDIO_MPEG4))
codec_id = CODEC_ID_AAC;
else
codec_id = CODEC_ID_NONE;
......@@ -2262,6 +2306,18 @@ matroska_read_header (AVFormatContext *s,
st->codec.codec_id = codec_id;
if(extradata){
st->codec.extradata = extradata;
st->codec.extradata_size = extradata_size;
} else if(track->codec_priv && track->codec_priv_size > 0){
st->codec.extradata = av_malloc(track->codec_priv_size);
if(st->codec.extradata == NULL)
return AVERROR_NOMEM;
st->codec.extradata_size = track->codec_priv_size;
memcpy(st->codec.extradata, track->codec_priv,
track->codec_priv_size);
}
if (track->type == MATROSKA_TRACK_TYPE_VIDEO) {
MatroskaVideoTrack *videotrack = (MatroskaVideoTrack *)track;
......@@ -2672,6 +2728,10 @@ matroska_read_close (AVFormatContext *s)
av_free(track);
}
for (n = 0; n < s->nb_streams; n++) {
av_free(s->streams[n]->codec.extradata);
}
memset(matroska, 0, sizeof(MatroskaDemuxContext));
return 0;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment