mirror of https://github.com/mpv-player/mpv
3964 lines
121 KiB
C
3964 lines
121 KiB
C
/*
|
|
* native Matroska demuxer
|
|
* Written by Aurelien Jacobs <aurel@gnuage.org>
|
|
* Based on the one written by Ronald Bultje for gstreamer
|
|
* and on demux_mkv.cpp from Moritz Bunkus.
|
|
* Licence: GPL
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <ctype.h>
|
|
#include <inttypes.h>
|
|
|
|
#include "stream.h"
|
|
#include "demuxer.h"
|
|
#include "stheader.h"
|
|
#include "ebml.h"
|
|
#include "matroska.h"
|
|
|
|
#include "mp_msg.h"
|
|
#include "help_mp.h"
|
|
|
|
#include "subreader.h"
|
|
#include "libvo/sub.h"
|
|
|
|
#include "libass/ass.h"
|
|
#include "libass/ass_mp.h"
|
|
|
|
#ifdef USE_QTX_CODECS
|
|
#include "qtx/qtxsdk/components.h"
|
|
#endif
|
|
|
|
#ifdef HAVE_ZLIB
|
|
#include <zlib.h>
|
|
#endif
|
|
|
|
#ifdef USE_LIBLZO
|
|
#include <lzo1x.h>
|
|
#else
|
|
#include "libmpcodecs/native/minilzo.h"
|
|
#endif
|
|
|
|
#if !defined(MIN)
|
|
#define MIN(a, b) ((a)<(b)?(a):(b))
|
|
#endif
|
|
#if !defined(MAX)
|
|
#define MAX(a, b) ((a)>(b)?(a):(b))
|
|
#endif
|
|
|
|
static unsigned char sipr_swaps[38][2]={
|
|
{0,63},{1,22},{2,44},{3,90},{5,81},{7,31},{8,86},{9,58},{10,36},{12,68},
|
|
{13,39},{14,73},{15,53},{16,69},{17,57},{19,88},{20,34},{21,71},{24,46},
|
|
{25,94},{26,54},{28,75},{29,50},{32,70},{33,92},{35,74},{38,85},{40,56},
|
|
{42,87},{43,65},{45,59},{48,79},{49,93},{51,89},{55,95},{61,76},{67,83},
|
|
{77,80} };
|
|
|
|
// Map flavour to bytes per second
|
|
#define SIPR_FLAVORS 4
|
|
#define ATRC_FLAVORS 8
|
|
#define COOK_FLAVORS 34
|
|
static int sipr_fl2bps[SIPR_FLAVORS] = {813, 1062, 625, 2000};
|
|
static int atrc_fl2bps[ATRC_FLAVORS] = {8269, 11714, 13092, 16538, 18260, 22050, 33075, 44100};
|
|
static int cook_fl2bps[COOK_FLAVORS] = {1000, 1378, 2024, 2584, 4005, 5513, 8010, 4005, 750, 2498,
|
|
4048, 5513, 8010, 11973, 8010, 2584, 4005, 2067, 2584, 2584,
|
|
4005, 4005, 5513, 5513, 8010, 12059, 1550, 8010, 12059, 5513,
|
|
12016, 16408, 22911, 33506};
|
|
|
|
typedef struct
|
|
{
|
|
uint32_t order, type, scope;
|
|
uint32_t comp_algo;
|
|
uint8_t *comp_settings;
|
|
int comp_settings_len;
|
|
} mkv_content_encoding_t;
|
|
|
|
typedef struct mkv_track
|
|
{
|
|
int tnum;
|
|
char *name;
|
|
|
|
char *codec_id;
|
|
int ms_compat;
|
|
char *language;
|
|
|
|
int type;
|
|
|
|
uint32_t v_width, v_height, v_dwidth, v_dheight;
|
|
float v_frate;
|
|
|
|
uint32_t a_formattag;
|
|
uint32_t a_channels, a_bps;
|
|
float a_sfreq;
|
|
|
|
float default_duration;
|
|
|
|
int default_track;
|
|
|
|
void *private_data;
|
|
unsigned int private_size;
|
|
|
|
/* stuff for realmedia */
|
|
int realmedia;
|
|
int rv_kf_base, rv_kf_pts;
|
|
float rv_pts; /* previous video timestamp */
|
|
float ra_pts; /* previous audio timestamp */
|
|
|
|
/** realaudio descrambling */
|
|
int sub_packet_size; ///< sub packet size, per stream
|
|
int sub_packet_h; ///< number of coded frames per block
|
|
int coded_framesize; ///< coded frame size, per stream
|
|
int audiopk_size; ///< audio packet size
|
|
unsigned char *audio_buf; ///< place to store reordered audio data
|
|
float *audio_timestamp; ///< timestamp for each audio packet
|
|
int sub_packet_cnt; ///< number of subpacket already received
|
|
int audio_filepos; ///< file position of first audio packet in block
|
|
|
|
/* stuff for quicktime */
|
|
int fix_i_bps;
|
|
float qt_last_a_pts;
|
|
|
|
int subtitle_type;
|
|
|
|
/* The timecodes of video frames might have to be reordered if they're
|
|
in display order (the timecodes, not the frames themselves!). In this
|
|
case demux packets have to be cached with the help of these variables. */
|
|
int reorder_timecodes;
|
|
demux_packet_t **cached_dps;
|
|
int num_cached_dps, num_allocated_dps;
|
|
float max_pts;
|
|
|
|
/* generic content encoding support */
|
|
mkv_content_encoding_t *encodings;
|
|
int num_encodings;
|
|
|
|
/* For VobSubs and SSA/ASS */
|
|
sh_sub_t sh_sub;
|
|
} mkv_track_t;
|
|
|
|
typedef struct mkv_index
|
|
{
|
|
int tnum;
|
|
uint64_t timecode, filepos;
|
|
} mkv_index_t;
|
|
|
|
typedef struct mkv_attachment
|
|
{
|
|
char* name;
|
|
char* mime;
|
|
uint64_t uid;
|
|
void* data;
|
|
unsigned int data_size;
|
|
} mkv_attachment_t;
|
|
|
|
typedef struct mkv_demuxer
|
|
{
|
|
off_t segment_start;
|
|
|
|
float duration, last_pts;
|
|
uint64_t last_filepos;
|
|
|
|
mkv_track_t **tracks;
|
|
int num_tracks;
|
|
|
|
uint64_t tc_scale, cluster_tc, first_tc;
|
|
int has_first_tc;
|
|
|
|
uint64_t clear_subs_at[SUB_MAX_TEXT];
|
|
subtitle subs;
|
|
|
|
uint64_t cluster_size;
|
|
uint64_t blockgroup_size;
|
|
|
|
mkv_index_t *indexes;
|
|
int num_indexes;
|
|
|
|
off_t *parsed_cues;
|
|
int parsed_cues_num;
|
|
off_t *parsed_seekhead;
|
|
int parsed_seekhead_num;
|
|
|
|
uint64_t *cluster_positions;
|
|
int num_cluster_pos;
|
|
|
|
int64_t skip_to_timecode;
|
|
int v_skip_to_keyframe, a_skip_to_keyframe;
|
|
|
|
int64_t stop_timecode;
|
|
|
|
int last_aid;
|
|
int audio_tracks[MAX_A_STREAMS];
|
|
|
|
mkv_attachment_t *attachments;
|
|
int num_attachments;
|
|
} mkv_demuxer_t;
|
|
|
|
|
|
typedef struct
|
|
{
|
|
uint32_t chunks; /* number of chunks */
|
|
uint32_t timestamp; /* timestamp from packet header */
|
|
uint32_t len; /* length of actual data */
|
|
uint32_t chunktab; /* offset to chunk offset array */
|
|
} dp_hdr_t;
|
|
|
|
typedef struct __attribute__((__packed__))
|
|
{
|
|
uint32_t size;
|
|
uint32_t fourcc1;
|
|
uint32_t fourcc2;
|
|
uint16_t width;
|
|
uint16_t height;
|
|
uint16_t bpp;
|
|
uint32_t unknown1;
|
|
uint32_t fps;
|
|
uint32_t type1;
|
|
uint32_t type2;
|
|
} real_video_props_t;
|
|
|
|
typedef struct __attribute__((__packed__))
|
|
{
|
|
uint32_t fourcc1; /* '.', 'r', 'a', 0xfd */
|
|
uint16_t version1; /* 4 or 5 */
|
|
uint16_t unknown1; /* 00 000 */
|
|
uint32_t fourcc2; /* .ra4 or .ra5 */
|
|
uint32_t unknown2; /* ??? */
|
|
uint16_t version2; /* 4 or 5 */
|
|
uint32_t header_size; /* == 0x4e */
|
|
uint16_t flavor; /* codec flavor id */
|
|
uint32_t coded_frame_size; /* coded frame size */
|
|
uint32_t unknown3; /* big number */
|
|
uint32_t unknown4; /* bigger number */
|
|
uint32_t unknown5; /* yet another number */
|
|
uint16_t sub_packet_h;
|
|
uint16_t frame_size;
|
|
uint16_t sub_packet_size;
|
|
uint16_t unknown6; /* 00 00 */
|
|
uint16_t sample_rate;
|
|
uint16_t unknown8; /* 0 */
|
|
uint16_t sample_size;
|
|
uint16_t channels;
|
|
} real_audio_v4_props_t;
|
|
|
|
typedef struct __attribute__((__packed__))
|
|
{
|
|
uint32_t fourcc1; /* '.', 'r', 'a', 0xfd */
|
|
uint16_t version1; /* 4 or 5 */
|
|
uint16_t unknown1; /* 00 000 */
|
|
uint32_t fourcc2; /* .ra4 or .ra5 */
|
|
uint32_t unknown2; /* ??? */
|
|
uint16_t version2; /* 4 or 5 */
|
|
uint32_t header_size; /* == 0x4e */
|
|
uint16_t flavor; /* codec flavor id */
|
|
uint32_t coded_frame_size; /* coded frame size */
|
|
uint32_t unknown3; /* big number */
|
|
uint32_t unknown4; /* bigger number */
|
|
uint32_t unknown5; /* yet another number */
|
|
uint16_t sub_packet_h;
|
|
uint16_t frame_size;
|
|
uint16_t sub_packet_size;
|
|
uint16_t unknown6; /* 00 00 */
|
|
uint8_t unknown7[6]; /* 0, srate, 0 */
|
|
uint16_t sample_rate;
|
|
uint16_t unknown8; /* 0 */
|
|
uint16_t sample_size;
|
|
uint16_t channels;
|
|
uint32_t genr; /* "genr" */
|
|
uint32_t fourcc3; /* fourcc */
|
|
} real_audio_v5_props_t;
|
|
|
|
|
|
/* for e.g. "-slang ger" */
|
|
extern char *dvdsub_lang;
|
|
extern char *audio_lang;
|
|
extern int dvdsub_id;
|
|
|
|
/**
|
|
* \brief ensures there is space for at least one additional element
|
|
* \param array array to grow
|
|
* \param nelem current number of elements in array
|
|
* \param elsize size of one array element
|
|
*/
|
|
static void grow_array(void **array, int nelem, size_t elsize) {
|
|
if (!(nelem & 31))
|
|
*array = realloc(*array, (nelem + 32) * elsize);
|
|
}
|
|
|
|
static mkv_track_t *
|
|
demux_mkv_find_track_by_num (mkv_demuxer_t *d, int n, int type)
|
|
{
|
|
int i, id;
|
|
|
|
for (i=0, id=0; i < d->num_tracks; i++)
|
|
if (d->tracks[i] != NULL && d->tracks[i]->type == type)
|
|
if (id++ == n)
|
|
return d->tracks[i];
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static mkv_track_t *
|
|
demux_mkv_find_track_by_language (mkv_demuxer_t *d, char *language, int type)
|
|
{
|
|
int i, len;
|
|
|
|
language += strspn(language,",");
|
|
while((len = strcspn(language,",")) > 0)
|
|
{
|
|
for (i=0; i < d->num_tracks; i++)
|
|
if (d->tracks[i] != NULL && d->tracks[i]->language != NULL &&
|
|
d->tracks[i]->type == type &&
|
|
!strncmp(d->tracks[i]->language, language, len))
|
|
return d->tracks[i];
|
|
language += len;
|
|
language += strspn(language,",");
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static void
|
|
add_cluster_position (mkv_demuxer_t *mkv_d, uint64_t position)
|
|
{
|
|
int i = mkv_d->num_cluster_pos;
|
|
|
|
while (i--)
|
|
if (mkv_d->cluster_positions[i] == position)
|
|
return;
|
|
|
|
grow_array(&mkv_d->cluster_positions, mkv_d->num_cluster_pos,
|
|
sizeof(uint64_t));
|
|
mkv_d->cluster_positions[mkv_d->num_cluster_pos++] = position;
|
|
}
|
|
|
|
|
|
#define AAC_SYNC_EXTENSION_TYPE 0x02b7
|
|
static int
|
|
aac_get_sample_rate_index (uint32_t sample_rate)
|
|
{
|
|
if (92017 <= sample_rate)
|
|
return 0;
|
|
else if (75132 <= sample_rate)
|
|
return 1;
|
|
else if (55426 <= sample_rate)
|
|
return 2;
|
|
else if (46009 <= sample_rate)
|
|
return 3;
|
|
else if (37566 <= sample_rate)
|
|
return 4;
|
|
else if (27713 <= sample_rate)
|
|
return 5;
|
|
else if (23004 <= sample_rate)
|
|
return 6;
|
|
else if (18783 <= sample_rate)
|
|
return 7;
|
|
else if (13856 <= sample_rate)
|
|
return 8;
|
|
else if (11502 <= sample_rate)
|
|
return 9;
|
|
else if (9391 <= sample_rate)
|
|
return 10;
|
|
else
|
|
return 11;
|
|
}
|
|
|
|
|
|
static int
|
|
vobsub_parse_size (mkv_track_t *t, const char *start)
|
|
{
|
|
if (sscanf(&start[6], "%dx%d", &t->sh_sub.width, &t->sh_sub.height) == 2)
|
|
{
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] VobSub size: %ux%u\n",
|
|
t->sh_sub.width, t->sh_sub.height);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
vobsub_parse_palette (mkv_track_t *t, const char *start)
|
|
{
|
|
int i, r, g, b, y, u, v, tmp;
|
|
|
|
start += 8;
|
|
while (isspace(*start))
|
|
start++;
|
|
for (i = 0; i < 16; i++)
|
|
{
|
|
if (sscanf(start, "%06x", &tmp) != 1)
|
|
break;
|
|
r = tmp >> 16 & 0xff;
|
|
g = tmp >> 8 & 0xff;
|
|
b = tmp & 0xff;
|
|
y = MIN(MAX((int)(0.1494 * r + 0.6061 * g + 0.2445 * b), 0),
|
|
0xff);
|
|
u = MIN(MAX((int)(0.6066 * r - 0.4322 * g - 0.1744 * b) + 128,
|
|
0), 0xff);
|
|
v = MIN(MAX((int)(-0.08435 * r - 0.3422 * g + 0.4266 * b) +
|
|
128, 0), 0xff);
|
|
t->sh_sub.palette[i] = y << 16 | u << 8 | v;
|
|
start += 6;
|
|
while ((*start == ',') || isspace(*start))
|
|
start++;
|
|
}
|
|
if (i == 16)
|
|
{
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] VobSub palette: %06x,%06x,"
|
|
"%06x,%06x,%06x,%06x,%06x,%06x,%06x,%06x,%06x,%06x,%06x,"
|
|
"%06x,%06x,%06x\n", t->sh_sub.palette[0],
|
|
t->sh_sub.palette[1], t->sh_sub.palette[2],
|
|
t->sh_sub.palette[3], t->sh_sub.palette[4],
|
|
t->sh_sub.palette[5], t->sh_sub.palette[6],
|
|
t->sh_sub.palette[7], t->sh_sub.palette[8],
|
|
t->sh_sub.palette[9], t->sh_sub.palette[10],
|
|
t->sh_sub.palette[11], t->sh_sub.palette[12],
|
|
t->sh_sub.palette[13], t->sh_sub.palette[14],
|
|
t->sh_sub.palette[15]);
|
|
t->sh_sub.has_palette = 1;
|
|
return 2;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
vobsub_parse_custom_colors (mkv_track_t *t, const char *start)
|
|
{
|
|
int use_custom_colors, i;
|
|
|
|
use_custom_colors = 0;
|
|
start += 14;
|
|
while (isspace(*start))
|
|
start++;
|
|
if (!strncasecmp(start, "ON", 2) || (*start == '1'))
|
|
use_custom_colors = 1;
|
|
else if (!strncasecmp(start, "OFF", 3) || (*start == '0'))
|
|
use_custom_colors = 0;
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] VobSub custom colors: %s\n",
|
|
use_custom_colors ? "ON" : "OFF");
|
|
if ((start = strstr(start, "colors:")) != NULL)
|
|
{
|
|
start += 7;
|
|
while (isspace(*start))
|
|
start++;
|
|
for (i = 0; i < 4; i++)
|
|
{
|
|
if (sscanf(start, "%06x", &t->sh_sub.colors[i]) != 1)
|
|
break;
|
|
start += 6;
|
|
while ((*start == ',') || isspace(*start))
|
|
start++;
|
|
}
|
|
if (i == 4)
|
|
{
|
|
t->sh_sub.custom_colors = 4;
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] VobSub colors: %06x,"
|
|
"%06x,%06x,%06x\n", t->sh_sub.colors[0],
|
|
t->sh_sub.colors[1], t->sh_sub.colors[2],
|
|
t->sh_sub.colors[3]);
|
|
}
|
|
}
|
|
if (!use_custom_colors)
|
|
t->sh_sub.custom_colors = 0;
|
|
return 4;
|
|
}
|
|
|
|
static int
|
|
vobsub_parse_forced_subs (mkv_track_t *t, const char *start)
|
|
{
|
|
start += 12;
|
|
while (isspace(*start))
|
|
start++;
|
|
if (!strncasecmp(start, "on", 2) || (*start == '1'))
|
|
t->sh_sub.forced_subs_only = 1;
|
|
else if (!strncasecmp(start, "off", 3) || (*start == '0'))
|
|
t->sh_sub.forced_subs_only = 0;
|
|
else
|
|
return 0;
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] VobSub forced subs: %d\n",
|
|
t->sh_sub.forced_subs_only);
|
|
return 8;
|
|
}
|
|
|
|
/** \brief Free cached demux packets
|
|
*
|
|
* Reordering the timecodes requires caching of demux packets. This function
|
|
* frees all these cached packets and the memory for the cached pointers
|
|
* itself.
|
|
*
|
|
* \param demuxer The demuxer for which the cache is to be freed.
|
|
*/
|
|
static void
|
|
free_cached_dps (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
mkv_track_t *track;
|
|
int i, k;
|
|
|
|
for (k = 0; k < mkv_d->num_tracks; k++)
|
|
{
|
|
track = mkv_d->tracks[k];
|
|
for (i = 0; i < track->num_cached_dps; i++)
|
|
free_demux_packet (track->cached_dps[i]);
|
|
free(track->cached_dps);
|
|
track->cached_dps = NULL;
|
|
track->num_cached_dps = 0;
|
|
track->num_allocated_dps = 0;
|
|
track->max_pts = 0;
|
|
}
|
|
}
|
|
|
|
static int
|
|
demux_mkv_parse_idx (mkv_track_t *t)
|
|
{
|
|
int things_found, last;
|
|
char *buf, *pos, *start;
|
|
|
|
if ((t->private_data == NULL) || (t->private_size == 0))
|
|
return 0;
|
|
|
|
things_found = 0;
|
|
buf = malloc(t->private_size + 1);
|
|
if (buf == NULL)
|
|
return 0;
|
|
memcpy(buf, t->private_data, t->private_size);
|
|
buf[t->private_size] = 0;
|
|
t->sh_sub.type = 'v';
|
|
t->sh_sub.has_palette = 0;
|
|
|
|
pos = buf;
|
|
start = buf;
|
|
last = 0;
|
|
do
|
|
{
|
|
if ((*pos == 0) || (*pos == '\r') || (*pos == '\n'))
|
|
{
|
|
if (*pos == 0)
|
|
last = 1;
|
|
*pos = 0;
|
|
|
|
if (!strncasecmp(start, "size: ", 6))
|
|
things_found |= vobsub_parse_size(t, start);
|
|
else if (!strncasecmp(start, "palette:", 8))
|
|
things_found |= vobsub_parse_palette(t, start);
|
|
else if (!strncasecmp(start, "custom colors:", 14))
|
|
things_found |= vobsub_parse_custom_colors(t, start);
|
|
else if (!strncasecmp(start, "forced subs:", 12))
|
|
things_found |= vobsub_parse_forced_subs(t, start);
|
|
|
|
if (last)
|
|
break;
|
|
do
|
|
{
|
|
pos++;
|
|
}
|
|
while ((*pos == '\r') || (*pos == '\n'));
|
|
start = pos;
|
|
}
|
|
else
|
|
pos++;
|
|
}
|
|
while (!last && (*start != 0));
|
|
|
|
free(buf);
|
|
|
|
return (things_found & 3) == 3;
|
|
}
|
|
|
|
|
|
static int
|
|
demux_mkv_decode (mkv_track_t *track, uint8_t *src, uint8_t **dest,
|
|
uint32_t *size, uint32_t type)
|
|
{
|
|
int i, result;
|
|
int modified = 0;
|
|
|
|
*dest = src;
|
|
if (track->num_encodings <= 0)
|
|
return 0;
|
|
|
|
for (i=0; i<track->num_encodings; i++)
|
|
{
|
|
if (!(track->encodings[i].scope & type))
|
|
continue;
|
|
|
|
#ifdef HAVE_ZLIB
|
|
if (track->encodings[i].comp_algo == 0)
|
|
{
|
|
/* zlib encoded track */
|
|
z_stream zstream;
|
|
|
|
zstream.zalloc = (alloc_func) 0;
|
|
zstream.zfree = (free_func) 0;
|
|
zstream.opaque = (voidpf) 0;
|
|
if (inflateInit (&zstream) != Z_OK)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_ZlibInitializationFailed);
|
|
return modified;
|
|
}
|
|
zstream.next_in = (Bytef *) src;
|
|
zstream.avail_in = *size;
|
|
|
|
modified = 1;
|
|
*dest = malloc (*size);
|
|
zstream.avail_out = *size;
|
|
do {
|
|
*size += 4000;
|
|
*dest = realloc (*dest, *size);
|
|
zstream.next_out = (Bytef *) (*dest + zstream.total_out);
|
|
result = inflate (&zstream, Z_NO_FLUSH);
|
|
if (result != Z_OK && result != Z_STREAM_END)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_ZlibDecompressionFailed);
|
|
free(*dest);
|
|
*dest = NULL;
|
|
inflateEnd (&zstream);
|
|
return modified;
|
|
}
|
|
zstream.avail_out += 4000;
|
|
} while (zstream.avail_out == 4000 &&
|
|
zstream.avail_in != 0 && result != Z_STREAM_END);
|
|
|
|
*size = zstream.total_out;
|
|
inflateEnd (&zstream);
|
|
}
|
|
#endif
|
|
if (track->encodings[i].comp_algo == 2)
|
|
{
|
|
/* lzo encoded track */
|
|
int dstlen = *size * 3;
|
|
|
|
if (lzo_init () != LZO_E_OK)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_LzoInitializationFailed);
|
|
return modified;
|
|
}
|
|
|
|
*dest = malloc (dstlen);
|
|
while (1)
|
|
{
|
|
result = lzo1x_decompress_safe (src, *size, *dest, &dstlen,
|
|
NULL);
|
|
if (result == LZO_E_OK)
|
|
break;
|
|
if (result != LZO_E_OUTPUT_OVERRUN)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_LzoDecompressionFailed);
|
|
return modified;
|
|
}
|
|
mp_msg (MSGT_DEMUX, MSGL_DBG2,
|
|
"[mkv] lzo decompression buffer too small.\n");
|
|
dstlen *= 2;
|
|
*dest = realloc (*dest, dstlen);
|
|
}
|
|
*size = dstlen;
|
|
}
|
|
}
|
|
|
|
return modified;
|
|
}
|
|
|
|
|
|
static int
|
|
demux_mkv_read_info (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t length, l;
|
|
int il;
|
|
uint64_t tc_scale = 1000000;
|
|
long double duration = 0.;
|
|
|
|
length = ebml_read_length (s, NULL);
|
|
while (length > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_TIMECODESCALE:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 1;
|
|
tc_scale = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + timecode scale: %"PRIu64"\n",
|
|
tc_scale);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_DURATION:
|
|
{
|
|
long double num = ebml_read_float (s, &l);
|
|
if (num == EBML_FLOAT_INVALID)
|
|
return 1;
|
|
duration = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + duration: %.3Lfs\n",
|
|
duration * tc_scale / 1000000000.0);
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
length -= l + il;
|
|
}
|
|
mkv_d->tc_scale = tc_scale;
|
|
mkv_d->duration = duration * tc_scale / 1000000000.0;
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* \brief free array of kv_content_encoding_t
|
|
* \param encodings pointer to array
|
|
* \param numencodings number of encodings in array
|
|
*/
|
|
static void
|
|
demux_mkv_free_encodings(mkv_content_encoding_t *encodings, int numencodings)
|
|
{
|
|
while (numencodings-- > 0)
|
|
free(encodings[numencodings].comp_settings);
|
|
free(encodings);
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_trackencodings (demuxer_t *demuxer, mkv_track_t *track)
|
|
{
|
|
stream_t *s = demuxer->stream;
|
|
mkv_content_encoding_t *ce, e;
|
|
uint64_t len, length, l;
|
|
int il, n;
|
|
|
|
ce = malloc (sizeof (*ce));
|
|
n = 0;
|
|
|
|
len = length = ebml_read_length (s, &il);
|
|
len += il;
|
|
while (length > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_CONTENTENCODING:
|
|
{
|
|
uint64_t len;
|
|
int i;
|
|
|
|
memset (&e, 0, sizeof (e));
|
|
e.scope = 1;
|
|
|
|
len = ebml_read_length (s, &i);
|
|
l = len + i;
|
|
|
|
while (len > 0)
|
|
{
|
|
uint64_t num, l;
|
|
int il;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_CONTENTENCODINGORDER:
|
|
num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
goto err_out;
|
|
e.order = num;
|
|
break;
|
|
|
|
case MATROSKA_ID_CONTENTENCODINGSCOPE:
|
|
num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
goto err_out;
|
|
e.scope = num;
|
|
break;
|
|
|
|
case MATROSKA_ID_CONTENTENCODINGTYPE:
|
|
num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
goto err_out;
|
|
e.type = num;
|
|
break;
|
|
|
|
case MATROSKA_ID_CONTENTCOMPRESSION:
|
|
{
|
|
uint64_t le;
|
|
|
|
le = ebml_read_length (s, &i);
|
|
l = le + i;
|
|
|
|
while (le > 0)
|
|
{
|
|
uint64_t l;
|
|
int il;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_CONTENTCOMPALGO:
|
|
num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
goto err_out;
|
|
e.comp_algo = num;
|
|
break;
|
|
|
|
case MATROSKA_ID_CONTENTCOMPSETTINGS:
|
|
l = ebml_read_length (s, &i);
|
|
e.comp_settings = malloc (l);
|
|
stream_read (s, e.comp_settings, l);
|
|
e.comp_settings_len = l;
|
|
l += i;
|
|
break;
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
le -= l + il;
|
|
}
|
|
|
|
if (e.type == 1)
|
|
{
|
|
mp_msg(MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_TrackEncrypted, track->tnum);
|
|
}
|
|
else if (e.type != 0)
|
|
{
|
|
mp_msg(MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_UnknownContentEncoding, track->tnum);
|
|
}
|
|
|
|
if (e.comp_algo != 0 && e.comp_algo != 2)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_UnknownCompression,
|
|
track->tnum, e.comp_algo);
|
|
}
|
|
#ifndef HAVE_ZLIB
|
|
else if (e.comp_algo == 0)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_ZlibCompressionUnsupported,
|
|
track->tnum);
|
|
}
|
|
#endif
|
|
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
len -= l + il;
|
|
}
|
|
for (i=0; i<n; i++)
|
|
if (e.order <= ce[i].order)
|
|
break;
|
|
ce = realloc (ce, (n+1) *sizeof (*ce));
|
|
memmove (ce+i+1, ce+i, (n-i) * sizeof (*ce));
|
|
memcpy (ce+i, &e, sizeof (e));
|
|
n++;
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
|
|
length -= l + il;
|
|
}
|
|
|
|
track->encodings = ce;
|
|
track->num_encodings = n;
|
|
return len;
|
|
|
|
err_out:
|
|
demux_mkv_free_encodings(ce, n);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_trackaudio (demuxer_t *demuxer, mkv_track_t *track)
|
|
{
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t len, length, l;
|
|
int il;
|
|
|
|
track->a_sfreq = 8000.0;
|
|
track->a_channels = 1;
|
|
|
|
len = length = ebml_read_length (s, &il);
|
|
len += il;
|
|
while (length > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_AUDIOSAMPLINGFREQ:
|
|
{
|
|
long double num = ebml_read_float (s, &l);
|
|
if (num == EBML_FLOAT_INVALID)
|
|
return 0;
|
|
track->a_sfreq = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Sampling frequency: %f\n",
|
|
track->a_sfreq);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_AUDIOBITDEPTH:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 0;
|
|
track->a_bps = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Bit depth: %u\n",
|
|
track->a_bps);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_AUDIOCHANNELS:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 0;
|
|
track->a_channels = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Channels: %u\n",
|
|
track->a_channels);
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
length -= l + il;
|
|
}
|
|
return len;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_trackvideo (demuxer_t *demuxer, mkv_track_t *track)
|
|
{
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t len, length, l;
|
|
int il;
|
|
|
|
len = length = ebml_read_length (s, &il);
|
|
len += il;
|
|
while (length > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_VIDEOFRAMERATE:
|
|
{
|
|
long double num = ebml_read_float (s, &l);
|
|
if (num == EBML_FLOAT_INVALID)
|
|
return 0;
|
|
track->v_frate = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Frame rate: %f\n",
|
|
track->v_frate);
|
|
if (track->v_frate > 0)
|
|
track->default_duration = 1 / track->v_frate;
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_VIDEODISPLAYWIDTH:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 0;
|
|
track->v_dwidth = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Display width: %u\n",
|
|
track->v_dwidth);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_VIDEODISPLAYHEIGHT:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 0;
|
|
track->v_dheight = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Display height: %u\n",
|
|
track->v_dheight);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_VIDEOPIXELWIDTH:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 0;
|
|
track->v_width = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Pixel width: %u\n",
|
|
track->v_width);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_VIDEOPIXELHEIGHT:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 0;
|
|
track->v_height = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Pixel height: %u\n",
|
|
track->v_height);
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
length -= l + il;
|
|
}
|
|
return len;
|
|
}
|
|
|
|
/**
|
|
* \brief free any data associated with given track
|
|
* \param track track of which to free data
|
|
*/
|
|
static void
|
|
demux_mkv_free_trackentry(mkv_track_t *track) {
|
|
if (track->name)
|
|
free (track->name);
|
|
if (track->codec_id)
|
|
free (track->codec_id);
|
|
if (track->language)
|
|
free (track->language);
|
|
if (track->private_data)
|
|
free (track->private_data);
|
|
if (track->audio_buf)
|
|
free (track->audio_buf);
|
|
if (track->audio_timestamp)
|
|
free (track->audio_timestamp);
|
|
#ifdef USE_ASS
|
|
if (track->sh_sub.ass_track)
|
|
ass_free_track (track->sh_sub.ass_track);
|
|
#endif
|
|
demux_mkv_free_encodings(track->encodings, track->num_encodings);
|
|
free(track);
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_trackentry (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
mkv_track_t *track;
|
|
uint64_t len, length, l;
|
|
int il;
|
|
|
|
track = calloc (1, sizeof (*track));
|
|
/* set default values */
|
|
track->default_track = 1;
|
|
track->name = 0;
|
|
track->language = strdup("eng");
|
|
|
|
len = length = ebml_read_length (s, &il);
|
|
len += il;
|
|
while (length > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_TRACKNUMBER:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
goto err_out;
|
|
track->tnum = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Track number: %u\n",
|
|
track->tnum);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_TRACKNAME:
|
|
{
|
|
track->name = ebml_read_utf8 (s, &l);
|
|
if (track->name == NULL)
|
|
goto err_out;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Name: %s\n",
|
|
track->name);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_TRACKTYPE:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 0;
|
|
track->type = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Track type: ");
|
|
switch (track->type)
|
|
{
|
|
case MATROSKA_TRACK_AUDIO:
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "Audio\n");
|
|
break;
|
|
case MATROSKA_TRACK_VIDEO:
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "Video\n");
|
|
break;
|
|
case MATROSKA_TRACK_SUBTITLE:
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "Subtitle\n");
|
|
break;
|
|
default:
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "unknown\n");
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_TRACKAUDIO:
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Audio track\n");
|
|
l = demux_mkv_read_trackaudio (demuxer, track);
|
|
if (l == 0)
|
|
goto err_out;
|
|
break;
|
|
|
|
case MATROSKA_ID_TRACKVIDEO:
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Video track\n");
|
|
l = demux_mkv_read_trackvideo (demuxer, track);
|
|
if (l == 0)
|
|
goto err_out;
|
|
break;
|
|
|
|
case MATROSKA_ID_CODECID:
|
|
track->codec_id = ebml_read_ascii (s, &l);
|
|
if (track->codec_id == NULL)
|
|
goto err_out;
|
|
if (!strcmp (track->codec_id, MKV_V_MSCOMP) ||
|
|
!strcmp (track->codec_id, MKV_A_ACM))
|
|
track->ms_compat = 1;
|
|
else if (!strcmp (track->codec_id, MKV_S_VOBSUB))
|
|
track->subtitle_type = MATROSKA_SUBTYPE_VOBSUB;
|
|
else if (!strcmp (track->codec_id, MKV_S_TEXTSSA)
|
|
|| !strcmp (track->codec_id, MKV_S_TEXTASS)
|
|
|| !strcmp (track->codec_id, MKV_S_SSA)
|
|
|| !strcmp (track->codec_id, MKV_S_ASS))
|
|
{
|
|
track->subtitle_type = MATROSKA_SUBTYPE_SSA;
|
|
}
|
|
else if (!strcmp (track->codec_id, MKV_S_TEXTASCII))
|
|
track->subtitle_type = MATROSKA_SUBTYPE_TEXT;
|
|
if (!strcmp (track->codec_id, MKV_S_TEXTUTF8))
|
|
{
|
|
track->subtitle_type = MATROSKA_SUBTYPE_TEXT;
|
|
}
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Codec ID: %s\n",
|
|
track->codec_id);
|
|
break;
|
|
|
|
case MATROSKA_ID_CODECPRIVATE:
|
|
{
|
|
int x;
|
|
uint64_t num = ebml_read_length (s, &x);
|
|
// audit: cheap guard against overflows later..
|
|
if (num > SIZE_MAX - 1000) return 0;
|
|
l = x + num;
|
|
track->private_data = malloc (num);
|
|
if (stream_read(s, track->private_data, num) != (int) num)
|
|
goto err_out;
|
|
track->private_size = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + CodecPrivate, length "
|
|
"%u\n", track->private_size);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_TRACKLANGUAGE:
|
|
free(track->language);
|
|
track->language = ebml_read_utf8 (s, &l);
|
|
if (track->language == NULL)
|
|
goto err_out;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Language: %s\n",
|
|
track->language);
|
|
break;
|
|
|
|
case MATROSKA_ID_TRACKFLAGDEFAULT:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
goto err_out;
|
|
track->default_track = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Default flag: %u\n",
|
|
track->default_track);
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_TRACKDEFAULTDURATION:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
goto err_out;
|
|
if (num == 0)
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Default duration: 0");
|
|
else
|
|
{
|
|
track->v_frate = 1000000000.0 / num;
|
|
track->default_duration = num / 1000000000.0;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + Default duration: "
|
|
"%.3fms ( = %.3f fps)\n",num/1000000.0,track->v_frate);
|
|
}
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_TRACKENCODINGS:
|
|
l = demux_mkv_read_trackencodings (demuxer, track);
|
|
if (l == 0)
|
|
goto err_out;
|
|
break;
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
length -= l + il;
|
|
}
|
|
|
|
mkv_d->tracks[mkv_d->num_tracks++] = track;
|
|
return len;
|
|
|
|
err_out:
|
|
demux_mkv_free_trackentry(track);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_tracks (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t length, l;
|
|
int il;
|
|
|
|
mkv_d->tracks = malloc (sizeof (*mkv_d->tracks));
|
|
mkv_d->num_tracks = 0;
|
|
|
|
length = ebml_read_length (s, NULL);
|
|
while (length > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_TRACKENTRY:
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + a track...\n");
|
|
mkv_d->tracks = realloc (mkv_d->tracks,
|
|
(mkv_d->num_tracks+1)
|
|
*sizeof (*mkv_d->tracks));
|
|
l = demux_mkv_read_trackentry (demuxer);
|
|
if (l == 0)
|
|
return 1;
|
|
break;
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
length -= l + il;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
extern int index_mode;
|
|
|
|
static int
|
|
demux_mkv_read_cues (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t length, l, time, track, pos;
|
|
off_t off;
|
|
int i, il;
|
|
|
|
if (index_mode == 0) {
|
|
ebml_read_skip (s, NULL);
|
|
return 0;
|
|
}
|
|
off = stream_tell (s);
|
|
for (i=0; i<mkv_d->parsed_cues_num; i++)
|
|
if (mkv_d->parsed_cues[i] == off)
|
|
{
|
|
ebml_read_skip (s, NULL);
|
|
return 0;
|
|
}
|
|
mkv_d->parsed_cues = realloc (mkv_d->parsed_cues,
|
|
(mkv_d->parsed_cues_num+1)
|
|
* sizeof (off_t));
|
|
mkv_d->parsed_cues[mkv_d->parsed_cues_num++] = off;
|
|
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] /---- [ parsing cues ] -----------\n");
|
|
length = ebml_read_length (s, NULL);
|
|
|
|
while (length > 0)
|
|
{
|
|
time = track = pos = EBML_UINT_INVALID;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_POINTENTRY:
|
|
{
|
|
uint64_t len;
|
|
|
|
len = ebml_read_length (s, &i);
|
|
l = len + i;
|
|
|
|
while (len > 0)
|
|
{
|
|
uint64_t l;
|
|
int il;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_CUETIME:
|
|
time = ebml_read_uint (s, &l);
|
|
break;
|
|
|
|
case MATROSKA_ID_CUETRACKPOSITION:
|
|
{
|
|
uint64_t le;
|
|
|
|
le = ebml_read_length (s, &i);
|
|
l = le + i;
|
|
|
|
while (le > 0)
|
|
{
|
|
uint64_t l;
|
|
int il;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_CUETRACK:
|
|
track = ebml_read_uint (s, &l);
|
|
break;
|
|
|
|
case MATROSKA_ID_CUECLUSTERPOSITION:
|
|
pos = ebml_read_uint (s, &l);
|
|
break;
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
le -= l + il;
|
|
}
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
len -= l + il;
|
|
}
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
|
|
length -= l + il;
|
|
|
|
if (time != EBML_UINT_INVALID && track != EBML_UINT_INVALID
|
|
&& pos != EBML_UINT_INVALID)
|
|
{
|
|
grow_array(&mkv_d->indexes, mkv_d->num_indexes, sizeof(mkv_index_t));
|
|
mkv_d->indexes[mkv_d->num_indexes].tnum = track;
|
|
mkv_d->indexes[mkv_d->num_indexes].timecode = time;
|
|
mkv_d->indexes[mkv_d->num_indexes].filepos =mkv_d->segment_start+pos;
|
|
mp_msg (MSGT_DEMUX, MSGL_DBG2, "[mkv] |+ found cue point "
|
|
"for track %"PRIu64": timecode %"PRIu64", filepos: %"PRIu64"\n",
|
|
track, time, mkv_d->segment_start + pos);
|
|
mkv_d->num_indexes++;
|
|
}
|
|
}
|
|
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] \\---- [ parsing cues ] -----------\n");
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_chapters (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t length, l;
|
|
int il;
|
|
|
|
if (demuxer->chapters)
|
|
{
|
|
ebml_read_skip (s, NULL);
|
|
return 0;
|
|
}
|
|
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] /---- [ parsing chapters ] ---------\n");
|
|
length = ebml_read_length (s, NULL);
|
|
|
|
while (length > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_EDITIONENTRY:
|
|
{
|
|
uint64_t len;
|
|
int i;
|
|
|
|
len = ebml_read_length (s, &i);
|
|
l = len + i;
|
|
|
|
while (len > 0)
|
|
{
|
|
uint64_t l;
|
|
int il;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_CHAPTERATOM:
|
|
{
|
|
uint64_t len, start=0, end=0;
|
|
char* name = 0;
|
|
int i;
|
|
int cid;
|
|
|
|
len = ebml_read_length (s, &i);
|
|
l = len + i;
|
|
|
|
while (len > 0)
|
|
{
|
|
uint64_t l;
|
|
int il;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_CHAPTERTIMESTART:
|
|
start = ebml_read_uint (s, &l) / 1000000;
|
|
break;
|
|
|
|
case MATROSKA_ID_CHAPTERTIMEEND:
|
|
end = ebml_read_uint (s, &l) / 1000000;
|
|
break;
|
|
|
|
case MATROSKA_ID_CHAPTERDISPLAY:
|
|
{
|
|
uint64_t len;
|
|
int i;
|
|
|
|
len = ebml_read_length (s, &i);
|
|
l = len + i;
|
|
while (len > 0)
|
|
{
|
|
uint64_t l;
|
|
int il;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_CHAPSTRING:
|
|
name = ebml_read_utf8 (s, &l);
|
|
break;
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
len -= l + il;
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
len -= l + il;
|
|
}
|
|
|
|
if (!name)
|
|
name = strdup("(unnamed)");
|
|
|
|
cid = demuxer_add_chapter(demuxer, name, start, end);
|
|
|
|
mp_msg(MSGT_DEMUX, MSGL_V,
|
|
"[mkv] Chapter %u from %02d:%02d:%02d."
|
|
"%03d to %02d:%02d:%02d.%03d, %s\n",
|
|
cid,
|
|
(int) (start / 60 / 60 / 1000),
|
|
(int) ((start / 60 / 1000) % 60),
|
|
(int) ((start / 1000) % 60),
|
|
(int) (start % 1000),
|
|
(int) (end / 60 / 60 / 1000),
|
|
(int) ((end / 60 / 1000) % 60),
|
|
(int) ((end / 1000) % 60),
|
|
(int) (end % 1000), name);
|
|
|
|
free(name);
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
len -= l + il;
|
|
}
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
|
|
length -= l + il;
|
|
}
|
|
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] \\---- [ parsing chapters ] ---------\n");
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_tags (demuxer_t *demuxer)
|
|
{
|
|
ebml_read_skip (demuxer->stream, NULL);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_attachments (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t length, l;
|
|
int il;
|
|
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] /---- [ parsing attachments ] ---------\n");
|
|
length = ebml_read_length (s, NULL);
|
|
|
|
while (length > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_ATTACHEDFILE:
|
|
{
|
|
uint64_t len;
|
|
int i;
|
|
char* name = NULL;
|
|
char* mime = NULL;
|
|
uint64_t uid = 0;
|
|
char* data = NULL;
|
|
int data_size = 0;
|
|
|
|
len = ebml_read_length (s, &i);
|
|
l = len + i;
|
|
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + an attachment...\n");
|
|
|
|
grow_array(&mkv_d->attachments, mkv_d->num_attachments,
|
|
sizeof(*mkv_d->attachments));
|
|
|
|
while (len > 0)
|
|
{
|
|
uint64_t l;
|
|
int il;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_FILENAME:
|
|
name = ebml_read_utf8 (s, &l);
|
|
if (name == NULL)
|
|
return 0;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + FileName: %s\n",
|
|
name);
|
|
break;
|
|
|
|
case MATROSKA_ID_FILEMIMETYPE:
|
|
mime = ebml_read_ascii (s, &l);
|
|
if (mime == NULL)
|
|
return 0;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + FileMimeType: %s\n",
|
|
mime);
|
|
break;
|
|
|
|
case MATROSKA_ID_FILEUID:
|
|
uid = ebml_read_uint (s, &l);
|
|
break;
|
|
|
|
case MATROSKA_ID_FILEDATA:
|
|
{
|
|
int x;
|
|
uint64_t num = ebml_read_length (s, &x);
|
|
l = x + num;
|
|
free(data);
|
|
data = malloc (num);
|
|
if (stream_read(s, data, num) != (int) num)
|
|
{
|
|
free(data);
|
|
return 0;
|
|
}
|
|
data_size = num;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] | + FileData, length "
|
|
"%u\n", data_size);
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
len -= l + il;
|
|
}
|
|
|
|
mkv_d->attachments[mkv_d->num_attachments].name = name;
|
|
mkv_d->attachments[mkv_d->num_attachments].mime = mime;
|
|
mkv_d->attachments[mkv_d->num_attachments].uid = uid;
|
|
mkv_d->attachments[mkv_d->num_attachments].data = data;
|
|
mkv_d->attachments[mkv_d->num_attachments].data_size = data_size;
|
|
mkv_d->num_attachments ++;
|
|
mp_msg(MSGT_DEMUX, MSGL_V,
|
|
"[mkv] Attachment: %s, %s, %u bytes\n",
|
|
name, mime, data_size);
|
|
#ifdef USE_ASS
|
|
if (extract_embedded_fonts && name && data && data_size &&
|
|
mime && (strcmp(mime, "application/x-truetype-font") == 0 ||
|
|
strcmp(mime, "application/x-font") == 0))
|
|
ass_add_font(ass_library, name, data, data_size);
|
|
#endif
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
length -= l + il;
|
|
}
|
|
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] \\---- [ parsing attachments ] ---------\n");
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_seekhead (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t length, l, seek_pos, saved_pos, num;
|
|
uint32_t seek_id;
|
|
int i, il, res = 0;
|
|
off_t off;
|
|
|
|
off = stream_tell (s);
|
|
for (i=0; i<mkv_d->parsed_seekhead_num; i++)
|
|
if (mkv_d->parsed_seekhead[i] == off)
|
|
{
|
|
ebml_read_skip (s, NULL);
|
|
return 0;
|
|
}
|
|
mkv_d->parsed_seekhead = realloc (mkv_d->parsed_seekhead,
|
|
(mkv_d->parsed_seekhead_num+1)
|
|
* sizeof (off_t));
|
|
mkv_d->parsed_seekhead[mkv_d->parsed_seekhead_num++] = off;
|
|
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] /---- [ parsing seek head ] ---------\n");
|
|
length = ebml_read_length (s, NULL);
|
|
/* off now holds the position of the next element after the seek head. */
|
|
off = stream_tell (s) + length;
|
|
while (length > 0 && !res)
|
|
{
|
|
|
|
seek_id = 0;
|
|
seek_pos = EBML_UINT_INVALID;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_SEEKENTRY:
|
|
{
|
|
uint64_t len;
|
|
|
|
len = ebml_read_length (s, &i);
|
|
l = len + i;
|
|
|
|
while (len > 0)
|
|
{
|
|
uint64_t l;
|
|
int il;
|
|
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_SEEKID:
|
|
num = ebml_read_uint (s, &l);
|
|
if (num != EBML_UINT_INVALID)
|
|
seek_id = num;
|
|
break;
|
|
|
|
case MATROSKA_ID_SEEKPOSITION:
|
|
seek_pos = ebml_read_uint (s, &l);
|
|
break;
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
len -= l + il;
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
length -= l + il;
|
|
|
|
if (seek_id == 0 || seek_id == MATROSKA_ID_CLUSTER
|
|
|| seek_pos == EBML_UINT_INVALID ||
|
|
((mkv_d->segment_start + seek_pos) >= (uint64_t)demuxer->movi_end))
|
|
continue;
|
|
|
|
saved_pos = stream_tell (s);
|
|
if (!stream_seek (s, mkv_d->segment_start + seek_pos))
|
|
res = 1;
|
|
else
|
|
{
|
|
if (ebml_read_id (s, &il) != seek_id)
|
|
res = 1;
|
|
else
|
|
switch (seek_id)
|
|
{
|
|
case MATROSKA_ID_CUES:
|
|
if (demux_mkv_read_cues (demuxer))
|
|
res = 1;
|
|
break;
|
|
|
|
case MATROSKA_ID_TAGS:
|
|
if (demux_mkv_read_tags (demuxer))
|
|
res = 1;
|
|
break;
|
|
|
|
case MATROSKA_ID_SEEKHEAD:
|
|
if (demux_mkv_read_seekhead (demuxer))
|
|
res = 1;
|
|
break;
|
|
|
|
case MATROSKA_ID_CHAPTERS:
|
|
if (demux_mkv_read_chapters (demuxer))
|
|
res = 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
stream_seek (s, saved_pos);
|
|
}
|
|
if (res)
|
|
{
|
|
/* If there was an error then try to skip this seek head. */
|
|
if (stream_seek (s, off))
|
|
res = 0;
|
|
}
|
|
else
|
|
if (length > 0)
|
|
stream_seek (s, stream_tell (s) + length);
|
|
mp_msg(MSGT_DEMUX, MSGL_V, "[mkv] \\---- [ parsing seek head ] ---------\n");
|
|
return res;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_open_video (demuxer_t *demuxer, mkv_track_t *track, int vid);
|
|
static int
|
|
demux_mkv_open_audio (demuxer_t *demuxer, mkv_track_t *track, int aid);
|
|
static int
|
|
demux_mkv_open_sub (demuxer_t *demuxer, mkv_track_t *track, int sid);
|
|
|
|
static void
|
|
display_create_tracks (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *)demuxer->priv;
|
|
int i, vid=0, aid=0, sid=0;
|
|
|
|
for (i=0; i<mkv_d->num_tracks; i++)
|
|
{
|
|
char *type = "unknown", str[32];
|
|
*str = '\0';
|
|
switch (mkv_d->tracks[i]->type)
|
|
{
|
|
case MATROSKA_TRACK_VIDEO:
|
|
type = "video";
|
|
demux_mkv_open_video(demuxer, mkv_d->tracks[i], vid);
|
|
if (mkv_d->tracks[i]->name)
|
|
mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VID_%d_NAME=%s\n", vid, mkv_d->tracks[i]->name);
|
|
sprintf (str, "-vid %u", vid++);
|
|
break;
|
|
case MATROSKA_TRACK_AUDIO:
|
|
type = "audio";
|
|
demux_mkv_open_audio(demuxer, mkv_d->tracks[i], aid);
|
|
if (mkv_d->tracks[i]->name)
|
|
mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_AID_%d_NAME=%s\n", aid, mkv_d->tracks[i]->name);
|
|
mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_AID_%d_LANG=%s\n", aid, mkv_d->tracks[i]->language);
|
|
sprintf (str, "-aid %u, -alang %.5s",aid++,mkv_d->tracks[i]->language);
|
|
break;
|
|
case MATROSKA_TRACK_SUBTITLE:
|
|
type = "subtitles";
|
|
demux_mkv_open_sub(demuxer, mkv_d->tracks[i], sid);
|
|
if (mkv_d->tracks[i]->name)
|
|
mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_SID_%d_NAME=%s\n", sid, mkv_d->tracks[i]->name);
|
|
mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_SID_%d_LANG=%s\n", sid, mkv_d->tracks[i]->language);
|
|
sprintf (str, "-sid %u, -slang %.5s",sid++,mkv_d->tracks[i]->language);
|
|
break;
|
|
}
|
|
if (mkv_d->tracks[i]->name)
|
|
mp_msg(MSGT_DEMUX, MSGL_INFO, MSGTR_MPDEMUX_MKV_TrackIDName,
|
|
mkv_d->tracks[i]->tnum, type, mkv_d->tracks[i]->codec_id, mkv_d->tracks[i]->name, str);
|
|
else
|
|
mp_msg(MSGT_DEMUX, MSGL_INFO, MSGTR_MPDEMUX_MKV_TrackID,
|
|
mkv_d->tracks[i]->tnum, type, mkv_d->tracks[i]->codec_id, str);
|
|
}
|
|
}
|
|
|
|
static int
|
|
demux_mkv_open_video (demuxer_t *demuxer, mkv_track_t *track, int vid)
|
|
{
|
|
BITMAPINFOHEADER *bih;
|
|
void *ImageDesc = NULL;
|
|
sh_video_t *sh_v;
|
|
|
|
if (track->ms_compat) /* MS compatibility mode */
|
|
{
|
|
BITMAPINFOHEADER *src;
|
|
|
|
if (track->private_data == NULL
|
|
|| track->private_size < sizeof (BITMAPINFOHEADER))
|
|
return 1;
|
|
|
|
src = (BITMAPINFOHEADER *) track->private_data;
|
|
bih = calloc (1, track->private_size);
|
|
bih->biSize = le2me_32 (src->biSize);
|
|
bih->biWidth = le2me_32 (src->biWidth);
|
|
bih->biHeight = le2me_32 (src->biHeight);
|
|
bih->biPlanes = le2me_16 (src->biPlanes);
|
|
bih->biBitCount = le2me_16 (src->biBitCount);
|
|
bih->biCompression = le2me_32 (src->biCompression);
|
|
bih->biSizeImage = le2me_32 (src->biSizeImage);
|
|
bih->biXPelsPerMeter = le2me_32 (src->biXPelsPerMeter);
|
|
bih->biYPelsPerMeter = le2me_32 (src->biYPelsPerMeter);
|
|
bih->biClrUsed = le2me_32 (src->biClrUsed);
|
|
bih->biClrImportant = le2me_32 (src->biClrImportant);
|
|
memcpy((char *) bih + sizeof (BITMAPINFOHEADER),
|
|
(char *) src + sizeof (BITMAPINFOHEADER),
|
|
track->private_size - sizeof (BITMAPINFOHEADER));
|
|
|
|
if (track->v_width == 0)
|
|
track->v_width = bih->biWidth;
|
|
if (track->v_height == 0)
|
|
track->v_height = bih->biHeight;
|
|
}
|
|
else
|
|
{
|
|
bih = calloc (1, sizeof (BITMAPINFOHEADER));
|
|
bih->biSize = sizeof (BITMAPINFOHEADER);
|
|
bih->biWidth = track->v_width;
|
|
bih->biHeight = track->v_height;
|
|
bih->biBitCount = 24;
|
|
bih->biSizeImage = bih->biWidth * bih->biHeight * bih->biBitCount/8;
|
|
|
|
if (track->private_size >= sizeof (real_video_props_t)
|
|
&& (!strcmp (track->codec_id, MKV_V_REALV10)
|
|
|| !strcmp (track->codec_id, MKV_V_REALV20)
|
|
|| !strcmp (track->codec_id, MKV_V_REALV30)
|
|
|| !strcmp (track->codec_id, MKV_V_REALV40)))
|
|
{
|
|
unsigned char *dst, *src;
|
|
real_video_props_t *rvp;
|
|
uint32_t type2;
|
|
unsigned int cnt;
|
|
|
|
rvp = (real_video_props_t *) track->private_data;
|
|
src = (unsigned char *) (rvp + 1);
|
|
|
|
cnt = track->private_size - sizeof (real_video_props_t);
|
|
bih = realloc(bih, sizeof (BITMAPINFOHEADER)+8+cnt);
|
|
bih->biSize = 48+cnt;
|
|
bih->biPlanes = 1;
|
|
type2 = be2me_32 (rvp->type2);
|
|
if (type2 == 0x10003000 || type2 == 0x10003001)
|
|
bih->biCompression=mmioFOURCC('R','V','1','3');
|
|
else
|
|
bih->biCompression=mmioFOURCC('R','V',track->codec_id[9],'0');
|
|
dst = (unsigned char *) (bih + 1);
|
|
((unsigned int *) dst)[0] = rvp->type1;
|
|
((unsigned int *) dst)[1] = rvp->type2;
|
|
stream_read(demuxer->stream, dst+8, cnt);
|
|
track->realmedia = 1;
|
|
|
|
#ifdef USE_QTX_CODECS
|
|
}
|
|
else if (track->private_size >= sizeof (ImageDescription)
|
|
&& !strcmp(track->codec_id, MKV_V_QUICKTIME))
|
|
{
|
|
ImageDescriptionPtr idesc;
|
|
|
|
idesc = (ImageDescriptionPtr) track->private_data;
|
|
idesc->idSize = be2me_32 (idesc->idSize);
|
|
idesc->cType = be2me_32 (idesc->cType);
|
|
idesc->version = be2me_16 (idesc->version);
|
|
idesc->revisionLevel = be2me_16 (idesc->revisionLevel);
|
|
idesc->vendor = be2me_32 (idesc->vendor);
|
|
idesc->temporalQuality = be2me_32 (idesc->temporalQuality);
|
|
idesc->spatialQuality = be2me_32 (idesc->spatialQuality);
|
|
idesc->width = be2me_16 (idesc->width);
|
|
idesc->height = be2me_16 (idesc->height);
|
|
idesc->hRes = be2me_32 (idesc->hRes);
|
|
idesc->vRes = be2me_32 (idesc->vRes);
|
|
idesc->dataSize = be2me_32 (idesc->dataSize);
|
|
idesc->frameCount = be2me_16 (idesc->frameCount);
|
|
idesc->depth = be2me_16 (idesc->depth);
|
|
idesc->clutID = be2me_16 (idesc->clutID);
|
|
bih->biPlanes = 1;
|
|
bih->biCompression = idesc->cType;
|
|
ImageDesc = idesc;
|
|
#endif /* USE_QTX_CODECS */
|
|
|
|
}
|
|
else if (!strcmp(track->codec_id, MKV_V_MPEG1))
|
|
{
|
|
bih->biCompression = mmioFOURCC('m', 'p', 'g', '1');
|
|
track->reorder_timecodes = !correct_pts;
|
|
}
|
|
else if (!strcmp(track->codec_id, MKV_V_MPEG2))
|
|
{
|
|
bih->biCompression = mmioFOURCC('m', 'p', 'g', '2');
|
|
track->reorder_timecodes = !correct_pts;
|
|
}
|
|
else if (!strcmp(track->codec_id, MKV_V_MPEG4_SP) ||
|
|
!strcmp(track->codec_id, MKV_V_MPEG4_ASP) ||
|
|
!strcmp(track->codec_id, MKV_V_MPEG4_AP))
|
|
{
|
|
bih->biCompression = mmioFOURCC('m', 'p', '4', 'v');
|
|
if (track->private_data && (track->private_size > 0))
|
|
{
|
|
bih->biSize += track->private_size;
|
|
bih = realloc (bih, bih->biSize);
|
|
memcpy (bih + 1, track->private_data, track->private_size);
|
|
}
|
|
track->reorder_timecodes = !correct_pts;
|
|
}
|
|
else if (!strcmp(track->codec_id, MKV_V_MPEG4_AVC))
|
|
{
|
|
bih->biCompression = mmioFOURCC('a', 'v', 'c', '1');
|
|
if (track->private_data && (track->private_size > 0))
|
|
{
|
|
bih->biSize += track->private_size;
|
|
bih = realloc (bih, bih->biSize);
|
|
memcpy (bih + 1, track->private_data, track->private_size);
|
|
}
|
|
track->reorder_timecodes = !correct_pts;
|
|
}
|
|
else
|
|
{
|
|
mp_msg (MSGT_DEMUX,MSGL_WARN, MSGTR_MPDEMUX_MKV_UnknownCodecID,
|
|
track->codec_id, track->tnum);
|
|
free(bih);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
sh_v = new_sh_video_vid (demuxer, track->tnum, vid);
|
|
sh_v->bih = bih;
|
|
sh_v->format = sh_v->bih->biCompression;
|
|
if (track->v_frate == 0.0)
|
|
track->v_frate = 25.0;
|
|
sh_v->fps = track->v_frate;
|
|
sh_v->frametime = 1 / track->v_frate;
|
|
sh_v->aspect = 0;
|
|
if (!track->realmedia)
|
|
{
|
|
sh_v->disp_w = track->v_width;
|
|
sh_v->disp_h = track->v_height;
|
|
if (track->v_dheight)
|
|
sh_v->aspect = (float)track->v_dwidth / (float)track->v_dheight;
|
|
}
|
|
else
|
|
{
|
|
// vd_realvid.c will set aspect to disp_w/disp_h and rederive
|
|
// disp_w and disp_h from the RealVideo stream contents returned
|
|
// by the Real DLLs. If DisplayWidth/DisplayHeight was not set in
|
|
// the Matroska file then it has already been set to PixelWidth/Height
|
|
// by check_track_information.
|
|
sh_v->disp_w = track->v_dwidth;
|
|
sh_v->disp_h = track->v_dheight;
|
|
}
|
|
sh_v->ImageDesc = ImageDesc;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] Aspect: %f\n", sh_v->aspect);
|
|
|
|
sh_v->ds = demuxer->video;
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_open_audio (demuxer_t *demuxer, mkv_track_t *track, int aid)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
sh_audio_t *sh_a = new_sh_audio_aid(demuxer, track->tnum, aid);
|
|
demux_packet_t *dp;
|
|
if(!sh_a) return 1;
|
|
mkv_d->audio_tracks[mkv_d->last_aid] = track->tnum;
|
|
|
|
sh_a->ds = demuxer->audio;
|
|
sh_a->wf = malloc (sizeof (WAVEFORMATEX));
|
|
if (track->ms_compat && (track->private_size >= sizeof(WAVEFORMATEX)))
|
|
{
|
|
WAVEFORMATEX *wf = (WAVEFORMATEX *)track->private_data;
|
|
sh_a->wf = realloc(sh_a->wf, track->private_size);
|
|
sh_a->wf->wFormatTag = le2me_16 (wf->wFormatTag);
|
|
sh_a->wf->nChannels = le2me_16 (wf->nChannels);
|
|
sh_a->wf->nSamplesPerSec = le2me_32 (wf->nSamplesPerSec);
|
|
sh_a->wf->nAvgBytesPerSec = le2me_32 (wf->nAvgBytesPerSec);
|
|
sh_a->wf->nBlockAlign = le2me_16 (wf->nBlockAlign);
|
|
sh_a->wf->wBitsPerSample = le2me_16 (wf->wBitsPerSample);
|
|
sh_a->wf->cbSize = track->private_size - sizeof(WAVEFORMATEX);
|
|
memcpy(sh_a->wf + 1, wf + 1, track->private_size - sizeof(WAVEFORMATEX));
|
|
if (track->a_sfreq == 0.0)
|
|
track->a_sfreq = sh_a->wf->nSamplesPerSec;
|
|
if (track->a_channels == 0)
|
|
track->a_channels = sh_a->wf->nChannels;
|
|
if (track->a_bps == 0)
|
|
track->a_bps = sh_a->wf->wBitsPerSample;
|
|
track->a_formattag = sh_a->wf->wFormatTag;
|
|
}
|
|
else
|
|
{
|
|
memset(sh_a->wf, 0, sizeof (WAVEFORMATEX));
|
|
if (!strcmp(track->codec_id, MKV_A_MP3) ||
|
|
!strcmp(track->codec_id, MKV_A_MP2))
|
|
track->a_formattag = 0x0055;
|
|
else if (!strncmp(track->codec_id, MKV_A_AC3, strlen(MKV_A_AC3)))
|
|
track->a_formattag = 0x2000;
|
|
else if (!strcmp(track->codec_id, MKV_A_DTS))
|
|
track->a_formattag = 0x2001;
|
|
else if (!strcmp(track->codec_id, MKV_A_PCM) ||
|
|
!strcmp(track->codec_id, MKV_A_PCM_BE))
|
|
track->a_formattag = 0x0001;
|
|
else if (!strcmp(track->codec_id, MKV_A_AAC_2MAIN) ||
|
|
!strncmp(track->codec_id, MKV_A_AAC_2LC,
|
|
strlen(MKV_A_AAC_2LC)) ||
|
|
!strcmp(track->codec_id, MKV_A_AAC_2SSR) ||
|
|
!strcmp(track->codec_id, MKV_A_AAC_4MAIN) ||
|
|
!strncmp(track->codec_id, MKV_A_AAC_4LC,
|
|
strlen(MKV_A_AAC_4LC)) ||
|
|
!strcmp(track->codec_id, MKV_A_AAC_4SSR) ||
|
|
!strcmp(track->codec_id, MKV_A_AAC_4LTP) ||
|
|
!strcmp(track->codec_id, MKV_A_AAC))
|
|
track->a_formattag = mmioFOURCC('M', 'P', '4', 'A');
|
|
else if (!strcmp(track->codec_id, MKV_A_VORBIS))
|
|
{
|
|
if (track->private_data == NULL)
|
|
return 1;
|
|
track->a_formattag = mmioFOURCC('v', 'r', 'b', 's');
|
|
}
|
|
else if (!strcmp(track->codec_id, MKV_A_QDMC))
|
|
track->a_formattag = mmioFOURCC('Q', 'D', 'M', 'C');
|
|
else if (!strcmp(track->codec_id, MKV_A_QDMC2))
|
|
track->a_formattag = mmioFOURCC('Q', 'D', 'M', '2');
|
|
else if (!strcmp(track->codec_id, MKV_A_FLAC))
|
|
{
|
|
if (track->private_data == NULL || track->private_size == 0)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_FlacTrackDoesNotContainValidHeaders);
|
|
return 1;
|
|
}
|
|
track->a_formattag = mmioFOURCC ('f', 'L', 'a', 'C');
|
|
}
|
|
else if (track->private_size >= sizeof (real_audio_v4_props_t))
|
|
{
|
|
if (!strcmp(track->codec_id, MKV_A_REAL28))
|
|
track->a_formattag = mmioFOURCC('2', '8', '_', '8');
|
|
else if (!strcmp(track->codec_id, MKV_A_REALATRC))
|
|
track->a_formattag = mmioFOURCC('a', 't', 'r', 'c');
|
|
else if (!strcmp(track->codec_id, MKV_A_REALCOOK))
|
|
track->a_formattag = mmioFOURCC('c', 'o', 'o', 'k');
|
|
else if (!strcmp(track->codec_id, MKV_A_REALDNET))
|
|
track->a_formattag = mmioFOURCC('d', 'n', 'e', 't');
|
|
else if (!strcmp(track->codec_id, MKV_A_REALSIPR))
|
|
track->a_formattag = mmioFOURCC('s', 'i', 'p', 'r');
|
|
}
|
|
else
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN, MSGTR_MPDEMUX_MKV_UnknownAudioCodec,
|
|
track->codec_id, track->tnum);
|
|
free_sh_audio(demuxer, track->tnum);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
sh_a->format = track->a_formattag;
|
|
sh_a->wf->wFormatTag = track->a_formattag;
|
|
sh_a->channels = track->a_channels;
|
|
sh_a->wf->nChannels = track->a_channels;
|
|
sh_a->samplerate = (uint32_t) track->a_sfreq;
|
|
sh_a->wf->nSamplesPerSec = (uint32_t) track->a_sfreq;
|
|
if (track->a_bps == 0)
|
|
{
|
|
sh_a->samplesize = 2;
|
|
sh_a->wf->wBitsPerSample = 16;
|
|
}
|
|
else
|
|
{
|
|
sh_a->samplesize = track->a_bps / 8;
|
|
sh_a->wf->wBitsPerSample = track->a_bps;
|
|
}
|
|
if (track->a_formattag == 0x0055) /* MP3 || MP2 */
|
|
{
|
|
sh_a->wf->nAvgBytesPerSec = 16000;
|
|
sh_a->wf->nBlockAlign = 1152;
|
|
}
|
|
else if ((track->a_formattag == 0x2000) || /* AC3 */
|
|
(track->a_formattag == 0x2001)) /* DTS */
|
|
{
|
|
sh_a->wf->nAvgBytesPerSec = 16000;
|
|
sh_a->wf->nBlockAlign = 1536;
|
|
}
|
|
else if (track->a_formattag == 0x0001) /* PCM || PCM_BE */
|
|
{
|
|
sh_a->wf->nAvgBytesPerSec = sh_a->channels * sh_a->samplerate*2;
|
|
sh_a->wf->nBlockAlign = sh_a->wf->nAvgBytesPerSec;
|
|
if (!strcmp(track->codec_id, MKV_A_PCM_BE))
|
|
sh_a->format = mmioFOURCC('t', 'w', 'o', 's');
|
|
}
|
|
else if (!strcmp(track->codec_id, MKV_A_QDMC) ||
|
|
!strcmp(track->codec_id, MKV_A_QDMC2))
|
|
{
|
|
sh_a->wf->nAvgBytesPerSec = 16000;
|
|
sh_a->wf->nBlockAlign = 1486;
|
|
track->fix_i_bps = 1;
|
|
track->qt_last_a_pts = 0.0;
|
|
if (track->private_data != NULL)
|
|
{
|
|
sh_a->codecdata=malloc(track->private_size);
|
|
memcpy (sh_a->codecdata, track->private_data,
|
|
track->private_size);
|
|
sh_a->codecdata_len = track->private_size;
|
|
}
|
|
}
|
|
else if (track->a_formattag == mmioFOURCC('M', 'P', '4', 'A'))
|
|
{
|
|
int profile, srate_idx;
|
|
|
|
sh_a->wf->nAvgBytesPerSec = 16000;
|
|
sh_a->wf->nBlockAlign = 1024;
|
|
|
|
if (!strcmp (track->codec_id, MKV_A_AAC) &&
|
|
(NULL != track->private_data))
|
|
{
|
|
sh_a->codecdata=malloc(track->private_size);
|
|
memcpy (sh_a->codecdata, track->private_data,
|
|
track->private_size);
|
|
sh_a->codecdata_len = track->private_size;
|
|
return 0;
|
|
}
|
|
|
|
/* Recreate the 'private data' */
|
|
/* which faad2 uses in its initialization */
|
|
srate_idx = aac_get_sample_rate_index (sh_a->samplerate);
|
|
if (!strncmp (&track->codec_id[12], "MAIN", 4))
|
|
profile = 0;
|
|
else if (!strncmp (&track->codec_id[12], "LC", 2))
|
|
profile = 1;
|
|
else if (!strncmp (&track->codec_id[12], "SSR", 3))
|
|
profile = 2;
|
|
else
|
|
profile = 3;
|
|
sh_a->codecdata = malloc (5);
|
|
sh_a->codecdata[0] = ((profile+1) << 3) | ((srate_idx&0xE) >> 1);
|
|
sh_a->codecdata[1] = ((srate_idx&0x1)<<7)|(track->a_channels<<3);
|
|
|
|
if (strstr(track->codec_id, "SBR") != NULL)
|
|
{
|
|
/* HE-AAC (aka SBR AAC) */
|
|
sh_a->codecdata_len = 5;
|
|
|
|
sh_a->samplerate *= 2;
|
|
sh_a->wf->nSamplesPerSec *= 2;
|
|
srate_idx = aac_get_sample_rate_index(sh_a->samplerate);
|
|
sh_a->codecdata[2] = AAC_SYNC_EXTENSION_TYPE >> 3;
|
|
sh_a->codecdata[3] = ((AAC_SYNC_EXTENSION_TYPE&0x07)<<5) | 5;
|
|
sh_a->codecdata[4] = (1 << 7) | (srate_idx << 3);
|
|
track->default_duration = 1024.0 / (sh_a->samplerate / 2);
|
|
}
|
|
else
|
|
{
|
|
sh_a->codecdata_len = 2;
|
|
track->default_duration = 1024.0 / (float)sh_a->samplerate;
|
|
}
|
|
}
|
|
else if (track->a_formattag == mmioFOURCC('v', 'r', 'b', 's')) /* VORBIS */
|
|
{
|
|
sh_a->wf->cbSize = track->private_size;
|
|
sh_a->wf = realloc(sh_a->wf, sizeof(WAVEFORMATEX) + sh_a->wf->cbSize);
|
|
memcpy((unsigned char *) (sh_a->wf+1), track->private_data, sh_a->wf->cbSize);
|
|
}
|
|
else if (track->private_size >= sizeof(real_audio_v4_props_t)
|
|
&& !strncmp (track->codec_id, MKV_A_REALATRC, 7))
|
|
{
|
|
/* Common initialization for all RealAudio codecs */
|
|
real_audio_v4_props_t *ra4p;
|
|
real_audio_v5_props_t *ra5p;
|
|
unsigned char *src;
|
|
int codecdata_length, version;
|
|
|
|
ra4p = (real_audio_v4_props_t *) track->private_data;
|
|
ra5p = (real_audio_v5_props_t *) track->private_data;
|
|
|
|
sh_a->wf->nAvgBytesPerSec = 0; /* FIXME !? */
|
|
sh_a->wf->nBlockAlign = be2me_16 (ra4p->frame_size);
|
|
|
|
version = be2me_16 (ra4p->version1);
|
|
|
|
if (version == 4)
|
|
{
|
|
src = (unsigned char *) (ra4p + 1);
|
|
src += src[0] + 1;
|
|
src += src[0] + 1;
|
|
track->sub_packet_size = be2me_16 (ra4p->sub_packet_size);
|
|
track->sub_packet_h = be2me_16 (ra4p->sub_packet_h);
|
|
track->coded_framesize = be2me_32 (ra4p->coded_frame_size);
|
|
track->audiopk_size = be2me_16 (ra4p->frame_size);
|
|
}
|
|
else
|
|
{
|
|
src = (unsigned char *) (ra5p + 1);
|
|
track->sub_packet_size = be2me_16 (ra5p->sub_packet_size);
|
|
track->sub_packet_h = be2me_16 (ra5p->sub_packet_h);
|
|
track->coded_framesize = be2me_32 (ra5p->coded_frame_size);
|
|
track->audiopk_size = be2me_16 (ra5p->frame_size);
|
|
}
|
|
|
|
src += 3;
|
|
if (version == 5)
|
|
src++;
|
|
codecdata_length = be2me_32 (*(uint32_t *)src);
|
|
src += 4;
|
|
sh_a->wf->cbSize = codecdata_length;
|
|
sh_a->wf = realloc (sh_a->wf,
|
|
sizeof (WAVEFORMATEX) +
|
|
sh_a->wf->cbSize);
|
|
memcpy(((char *)(sh_a->wf + 1)), src, codecdata_length);
|
|
|
|
switch (track->a_formattag) {
|
|
case mmioFOURCC('a', 't', 'r', 'c'):
|
|
sh_a->wf->nAvgBytesPerSec = atrc_fl2bps[be2me_16 (ra4p->flavor)];
|
|
sh_a->wf->nBlockAlign = track->sub_packet_size;
|
|
track->audio_buf = malloc(track->sub_packet_h * track->audiopk_size);
|
|
track->audio_timestamp = malloc(track->sub_packet_h * sizeof(float));
|
|
break;
|
|
case mmioFOURCC('c', 'o', 'o', 'k'):
|
|
sh_a->wf->nAvgBytesPerSec = cook_fl2bps[be2me_16 (ra4p->flavor)];
|
|
sh_a->wf->nBlockAlign = track->sub_packet_size;
|
|
track->audio_buf = malloc(track->sub_packet_h * track->audiopk_size);
|
|
track->audio_timestamp = malloc(track->sub_packet_h * sizeof(float));
|
|
break;
|
|
case mmioFOURCC('s', 'i', 'p', 'r'):
|
|
sh_a->wf->nAvgBytesPerSec = sipr_fl2bps[be2me_16 (ra4p->flavor)];
|
|
sh_a->wf->nBlockAlign = track->coded_framesize;
|
|
track->audio_buf = malloc(track->sub_packet_h * track->audiopk_size);
|
|
track->audio_timestamp = malloc(track->sub_packet_h * sizeof(float));
|
|
break;
|
|
case mmioFOURCC('2', '8', '_', '8'):
|
|
sh_a->wf->nAvgBytesPerSec = 3600;
|
|
sh_a->wf->nBlockAlign = track->coded_framesize;
|
|
track->audio_buf = malloc(track->sub_packet_h * track->audiopk_size);
|
|
track->audio_timestamp = malloc(track->sub_packet_h * sizeof(float));
|
|
break;
|
|
}
|
|
|
|
track->realmedia = 1;
|
|
}
|
|
else if (!strcmp(track->codec_id, MKV_A_FLAC) ||
|
|
(track->a_formattag == 0xf1ac))
|
|
{
|
|
unsigned char *ptr;
|
|
int size;
|
|
free(sh_a->wf);
|
|
sh_a->wf = NULL;
|
|
|
|
if (track->a_formattag == mmioFOURCC('f', 'L', 'a', 'C'))
|
|
{
|
|
ptr = (unsigned char *)track->private_data;
|
|
size = track->private_size;
|
|
}
|
|
else
|
|
{
|
|
sh_a->format = mmioFOURCC('f', 'L', 'a', 'C');
|
|
ptr = (unsigned char *) track->private_data
|
|
+ sizeof (WAVEFORMATEX);
|
|
size = track->private_size - sizeof (WAVEFORMATEX);
|
|
}
|
|
if (size < 4 || ptr[0] != 'f' || ptr[1] != 'L' ||
|
|
ptr[2] != 'a' || ptr[3] != 'C')
|
|
{
|
|
dp = new_demux_packet (4);
|
|
memcpy (dp->buffer, "fLaC", 4);
|
|
}
|
|
else
|
|
{
|
|
dp = new_demux_packet (size);
|
|
memcpy (dp->buffer, ptr, size);
|
|
}
|
|
dp->pts = 0;
|
|
dp->flags = 0;
|
|
ds_add_packet (demuxer->audio, dp);
|
|
}
|
|
else if (!track->ms_compat || (track->private_size < sizeof(WAVEFORMATEX)))
|
|
{
|
|
free_sh_audio(demuxer, track->tnum);
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/** \brief Parse the private data for VobSub subtitle tracks.
|
|
|
|
This function tries to parse the private data for all VobSub tracks.
|
|
The private data contains the normal text from the original .idx file.
|
|
Things like the palette, subtitle dimensions and custom colors are
|
|
stored here.
|
|
|
|
\param demuxer The generic demuxer.
|
|
*/
|
|
static void
|
|
demux_mkv_parse_vobsub_data (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
mkv_track_t *track;
|
|
int i, m, size;
|
|
uint8_t *buffer;
|
|
|
|
for (i = 0; i < mkv_d->num_tracks; i++)
|
|
{
|
|
track = mkv_d->tracks[i];
|
|
if ((track->type != MATROSKA_TRACK_SUBTITLE) ||
|
|
(track->subtitle_type != MATROSKA_SUBTYPE_VOBSUB))
|
|
continue;
|
|
|
|
size = track->private_size;
|
|
m = demux_mkv_decode (track,track->private_data,&buffer,&size,2);
|
|
if (buffer && m)
|
|
{
|
|
free (track->private_data);
|
|
track->private_data = buffer;
|
|
track->private_size = size;
|
|
}
|
|
if (!demux_mkv_parse_idx (track))
|
|
{
|
|
free (track->private_data);
|
|
track->private_data = NULL;
|
|
track->private_size = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
/** \brief Parse the private data for SSA/ASS subtitle tracks.
|
|
|
|
This function tries to parse the private data for all SSA/ASS tracks.
|
|
The private data contains the normal text from the original script,
|
|
from the start to the beginning of 'Events' section, including '[Events]' line.
|
|
|
|
\param demuxer The generic demuxer.
|
|
*/
|
|
#ifdef USE_ASS
|
|
static void
|
|
demux_mkv_parse_ass_data (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
mkv_track_t *track;
|
|
int i, m, size;
|
|
uint8_t *buffer;
|
|
|
|
for (i = 0; i < mkv_d->num_tracks; i++)
|
|
{
|
|
track = mkv_d->tracks[i];
|
|
if (track->type != MATROSKA_TRACK_SUBTITLE)
|
|
continue;
|
|
|
|
track->sh_sub.type = 'a';
|
|
|
|
if (track->subtitle_type == MATROSKA_SUBTYPE_SSA)
|
|
{
|
|
track->sh_sub.ass_track = ass_new_track(ass_library);
|
|
size = track->private_size;
|
|
m = demux_mkv_decode (track,track->private_data,&buffer,&size,2);
|
|
if (buffer && m)
|
|
{
|
|
free (track->private_data);
|
|
track->private_data = buffer;
|
|
track->private_size = size;
|
|
}
|
|
ass_process_codec_private(track->sh_sub.ass_track, track->private_data, track->private_size);
|
|
}
|
|
else
|
|
{
|
|
track->sh_sub.ass_track = ass_default_track(ass_library);
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
static int
|
|
demux_mkv_open_sub (demuxer_t *demuxer, mkv_track_t *track, int sid)
|
|
{
|
|
if (track->subtitle_type != MATROSKA_SUBTYPE_UNKNOWN)
|
|
{
|
|
sh_sub_t *sh = new_sh_sub_sid(demuxer, track->tnum, sid);
|
|
if ((track->subtitle_type == MATROSKA_SUBTYPE_VOBSUB) ||
|
|
(track->subtitle_type == MATROSKA_SUBTYPE_SSA))
|
|
{
|
|
if (track->private_data != NULL)
|
|
{
|
|
if (sh)
|
|
memcpy(sh, &track->sh_sub, sizeof(sh_sub_t));
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_ERR, MSGTR_MPDEMUX_MKV_SubtitleTypeNotSupported,
|
|
track->codec_id);
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void demux_mkv_seek (demuxer_t *demuxer, float rel_seek_secs, float audio_delay, int flags);
|
|
|
|
/** \brief Given a matroska track number and type, find the id that mplayer would ask for.
|
|
* \param d The demuxer for which the subtitle id should be returned.
|
|
* \param num The matroska track number we are looking up.
|
|
* \param type The track type.
|
|
*/
|
|
static int demux_mkv_reverse_id(mkv_demuxer_t *d, int num, int type)
|
|
{
|
|
int i, id;
|
|
|
|
for (i=0, id=0; i < d->num_tracks; i++)
|
|
if (d->tracks[i] != NULL && d->tracks[i]->type == type) {
|
|
if (d->tracks[i]->tnum == num)
|
|
return id;
|
|
id++;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_open (demuxer_t *demuxer)
|
|
{
|
|
stream_t *s = demuxer->stream;
|
|
mkv_demuxer_t *mkv_d;
|
|
mkv_track_t *track;
|
|
int i, version, cont = 0;
|
|
char *str;
|
|
|
|
#ifdef USE_ICONV
|
|
subcp_open(NULL);
|
|
#endif
|
|
|
|
stream_seek(s, s->start_pos);
|
|
str = ebml_read_header (s, &version);
|
|
if (str == NULL || strcmp (str, "matroska") || version > 2)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_DBG2, "[mkv] no head found\n");
|
|
return 0;
|
|
}
|
|
free (str);
|
|
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] Found the head...\n");
|
|
|
|
if (ebml_read_id (s, NULL) != MATROSKA_ID_SEGMENT)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] but no segment :(\n");
|
|
return 0;
|
|
}
|
|
ebml_read_length (s, NULL); /* return bytes number until EOF */
|
|
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] + a segment...\n");
|
|
|
|
mkv_d = calloc (1, sizeof (mkv_demuxer_t));
|
|
demuxer->priv = mkv_d;
|
|
mkv_d->tc_scale = 1000000;
|
|
mkv_d->segment_start = stream_tell (s);
|
|
mkv_d->parsed_cues = malloc (sizeof (off_t));
|
|
mkv_d->parsed_seekhead = malloc (sizeof (off_t));
|
|
|
|
for (i=0; i < SUB_MAX_TEXT; i++)
|
|
mkv_d->subs.text[i] = malloc (256);
|
|
|
|
while (!cont)
|
|
{
|
|
switch (ebml_read_id (s, NULL))
|
|
{
|
|
case MATROSKA_ID_INFO:
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] |+ segment information...\n");
|
|
cont = demux_mkv_read_info (demuxer);
|
|
break;
|
|
|
|
case MATROSKA_ID_TRACKS:
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] |+ segment tracks...\n");
|
|
cont = demux_mkv_read_tracks (demuxer);
|
|
break;
|
|
|
|
case MATROSKA_ID_CUES:
|
|
cont = demux_mkv_read_cues (demuxer);
|
|
break;
|
|
|
|
case MATROSKA_ID_TAGS:
|
|
cont = demux_mkv_read_tags (demuxer);
|
|
break;
|
|
|
|
case MATROSKA_ID_SEEKHEAD:
|
|
cont = demux_mkv_read_seekhead (demuxer);
|
|
break;
|
|
|
|
case MATROSKA_ID_CHAPTERS:
|
|
cont = demux_mkv_read_chapters (demuxer);
|
|
break;
|
|
|
|
case MATROSKA_ID_ATTACHMENTS:
|
|
cont = demux_mkv_read_attachments (demuxer);
|
|
break;
|
|
|
|
case MATROSKA_ID_CLUSTER:
|
|
{
|
|
int p, l;
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] |+ found cluster, headers are "
|
|
"parsed completely :)\n");
|
|
/* get the first cluster timecode */
|
|
p = stream_tell(s);
|
|
l = ebml_read_length (s, NULL);
|
|
while (ebml_read_id (s, NULL) != MATROSKA_ID_CLUSTERTIMECODE)
|
|
{
|
|
ebml_read_skip (s, NULL);
|
|
if (stream_tell (s) >= p + l)
|
|
break;
|
|
}
|
|
if (stream_tell (s) < p + l)
|
|
{
|
|
uint64_t num = ebml_read_uint (s, NULL);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 0;
|
|
mkv_d->first_tc = num * mkv_d->tc_scale / 1000000.0;
|
|
mkv_d->has_first_tc = 1;
|
|
}
|
|
stream_seek (s, p - 4);
|
|
cont = 1;
|
|
break;
|
|
}
|
|
|
|
default:
|
|
cont = 1;
|
|
case EBML_ID_VOID:
|
|
ebml_read_skip (s, NULL);
|
|
break;
|
|
}
|
|
}
|
|
|
|
display_create_tracks (demuxer);
|
|
|
|
/* select video track */
|
|
track = NULL;
|
|
if (demuxer->video->id == -1) /* automatically select a video track */
|
|
{
|
|
/* search for a video track that has the 'default' flag set */
|
|
for (i=0; i<mkv_d->num_tracks; i++)
|
|
if (mkv_d->tracks[i]->type == MATROSKA_TRACK_VIDEO
|
|
&& mkv_d->tracks[i]->default_track)
|
|
{
|
|
track = mkv_d->tracks[i];
|
|
break;
|
|
}
|
|
|
|
if (track == NULL)
|
|
/* no track has the 'default' flag set */
|
|
/* let's take the first video track */
|
|
for (i=0; i<mkv_d->num_tracks; i++)
|
|
if (mkv_d->tracks[i]->type == MATROSKA_TRACK_VIDEO)
|
|
{
|
|
track = mkv_d->tracks[i];
|
|
break;
|
|
}
|
|
}
|
|
else if (demuxer->video->id != -2) /* -2 = no video at all */
|
|
track = demux_mkv_find_track_by_num (mkv_d, demuxer->video->id,
|
|
MATROSKA_TRACK_VIDEO);
|
|
|
|
if (track && demuxer->v_streams[track->tnum])
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_INFO,
|
|
MSGTR_MPDEMUX_MKV_WillPlayVideoTrack, track->tnum);
|
|
demuxer->video->id = track->tnum;
|
|
demuxer->video->sh = demuxer->v_streams[track->tnum];
|
|
}
|
|
else
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_INFO, MSGTR_MPDEMUX_MKV_NoVideoTrackFound);
|
|
demuxer->video->id = -2;
|
|
}
|
|
|
|
/* select audio track */
|
|
track = NULL;
|
|
if (demuxer->audio->id == -1) /* automatically select an audio track */
|
|
{
|
|
/* check if the user specified an audio language */
|
|
if (audio_lang != NULL)
|
|
track = demux_mkv_find_track_by_language(mkv_d, audio_lang,
|
|
MATROSKA_TRACK_AUDIO);
|
|
if (track == NULL)
|
|
/* no audio language specified, or language not found */
|
|
/* search for an audio track that has the 'default' flag set */
|
|
for (i=0; i < mkv_d->num_tracks; i++)
|
|
if (mkv_d->tracks[i]->type == MATROSKA_TRACK_AUDIO
|
|
&& mkv_d->tracks[i]->default_track)
|
|
{
|
|
track = mkv_d->tracks[i];
|
|
break;
|
|
}
|
|
|
|
if (track == NULL)
|
|
/* no track has the 'default' flag set */
|
|
/* let's take the first audio track */
|
|
for (i=0; i < mkv_d->num_tracks; i++)
|
|
if (mkv_d->tracks[i]->type == MATROSKA_TRACK_AUDIO)
|
|
{
|
|
track = mkv_d->tracks[i];
|
|
break;
|
|
}
|
|
}
|
|
else if (demuxer->audio->id != -2) /* -2 = no audio at all */
|
|
track = demux_mkv_find_track_by_num (mkv_d, demuxer->audio->id,
|
|
MATROSKA_TRACK_AUDIO);
|
|
else
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_INFO, MSGTR_MPDEMUX_MKV_NoAudioTrackFound);
|
|
demuxer->audio->id = -2;
|
|
}
|
|
|
|
|
|
if(demuxer->audio->id != -2)
|
|
for (i=0; i < mkv_d->num_tracks; i++)
|
|
{
|
|
if(mkv_d->tracks[i]->type != MATROSKA_TRACK_AUDIO)
|
|
continue;
|
|
if(demuxer->a_streams[track->tnum])
|
|
{
|
|
if(track && mkv_d->tracks[i] == track)
|
|
{
|
|
demuxer->audio->id = track->tnum;
|
|
demuxer->audio->sh = demuxer->a_streams[track->tnum];
|
|
}
|
|
mkv_d->last_aid++;
|
|
if(mkv_d->last_aid == MAX_A_STREAMS)
|
|
break;
|
|
}
|
|
}
|
|
|
|
demux_mkv_parse_vobsub_data (demuxer);
|
|
#ifdef USE_ASS
|
|
if (ass_enabled)
|
|
demux_mkv_parse_ass_data (demuxer);
|
|
#endif
|
|
/* DO NOT automatically select a subtitle track and behave like DVD */
|
|
/* playback: only show subtitles if the user explicitely wants them. */
|
|
track = NULL;
|
|
if (demuxer->sub->id >= 0)
|
|
track = demux_mkv_find_track_by_num (mkv_d, demuxer->sub->id,
|
|
MATROSKA_TRACK_SUBTITLE);
|
|
else if (dvdsub_lang != NULL)
|
|
track = demux_mkv_find_track_by_language (mkv_d, dvdsub_lang,
|
|
MATROSKA_TRACK_SUBTITLE);
|
|
|
|
if (track)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_INFO,
|
|
MSGTR_MPDEMUX_MKV_WillDisplaySubtitleTrack, track->tnum);
|
|
dvdsub_id = demux_mkv_reverse_id(mkv_d, track->tnum, MATROSKA_TRACK_SUBTITLE);
|
|
demuxer->sub->id = track->tnum;
|
|
}
|
|
else
|
|
demuxer->sub->id = -2;
|
|
|
|
if (demuxer->chapters)
|
|
{
|
|
for (i=0; i < (int)demuxer->num_chapters; i++)
|
|
{
|
|
demuxer->chapters[i].start -= mkv_d->first_tc;
|
|
demuxer->chapters[i].end -= mkv_d->first_tc;
|
|
}
|
|
if (dvd_last_chapter > 0 && dvd_last_chapter <= demuxer->num_chapters)
|
|
{
|
|
if (demuxer->chapters[dvd_last_chapter-1].end != 0)
|
|
mkv_d->stop_timecode = demuxer->chapters[dvd_last_chapter-1].end;
|
|
else if (dvd_last_chapter + 1 <= demuxer->num_chapters)
|
|
mkv_d->stop_timecode = demuxer->chapters[dvd_last_chapter].start;
|
|
}
|
|
}
|
|
|
|
if (s->end_pos == 0 || (mkv_d->indexes == NULL && index_mode < 0))
|
|
demuxer->seekable = 0;
|
|
else
|
|
{
|
|
demuxer->movi_start = s->start_pos;
|
|
demuxer->movi_end = s->end_pos;
|
|
demuxer->seekable = 1;
|
|
if (demuxer->chapters && dvd_chapter>1 && dvd_chapter<=demuxer->num_chapters)
|
|
{
|
|
if (!mkv_d->has_first_tc)
|
|
{
|
|
mkv_d->first_tc = 0;
|
|
mkv_d->has_first_tc = 1;
|
|
}
|
|
demux_mkv_seek (demuxer,
|
|
demuxer->chapters[dvd_chapter-1].start/1000.0, 0.0, 1);
|
|
}
|
|
}
|
|
|
|
return DEMUXER_TYPE_MATROSKA;
|
|
}
|
|
|
|
static void
|
|
demux_close_mkv (demuxer_t *demuxer)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
|
|
if (mkv_d)
|
|
{
|
|
int i;
|
|
#ifdef USE_ICONV
|
|
subcp_close();
|
|
#endif
|
|
free_cached_dps (demuxer);
|
|
if (mkv_d->tracks)
|
|
{
|
|
for (i=0; i<mkv_d->num_tracks; i++)
|
|
demux_mkv_free_trackentry(mkv_d->tracks[i]);
|
|
for (i=0; i < SUB_MAX_TEXT; i++)
|
|
if (mkv_d->subs.text[i])
|
|
free (mkv_d->subs.text[i]);
|
|
free (mkv_d->tracks);
|
|
}
|
|
if (mkv_d->indexes)
|
|
free (mkv_d->indexes);
|
|
if (mkv_d->cluster_positions)
|
|
free (mkv_d->cluster_positions);
|
|
if (mkv_d->parsed_cues)
|
|
free (mkv_d->parsed_cues);
|
|
if (mkv_d->parsed_seekhead)
|
|
free (mkv_d->parsed_seekhead);
|
|
if (mkv_d->attachments) {
|
|
for (i = 0; i < mkv_d->num_attachments; ++i) {
|
|
if (mkv_d->attachments[i].name)
|
|
free (mkv_d->attachments[i].name);
|
|
if (mkv_d->attachments[i].mime)
|
|
free (mkv_d->attachments[i].mime);
|
|
if (mkv_d->attachments[i].data)
|
|
free (mkv_d->attachments[i].data);
|
|
}
|
|
free (mkv_d->attachments);
|
|
}
|
|
free (mkv_d);
|
|
}
|
|
}
|
|
|
|
static int
|
|
demux_mkv_read_block_lacing (uint8_t *buffer, uint64_t *size,
|
|
uint8_t *laces, uint32_t **all_lace_sizes)
|
|
{
|
|
uint32_t total = 0, *lace_size;
|
|
uint8_t flags;
|
|
int i;
|
|
|
|
*all_lace_sizes = NULL;
|
|
lace_size = NULL;
|
|
/* lacing flags */
|
|
flags = *buffer++;
|
|
(*size)--;
|
|
|
|
switch ((flags & 0x06) >> 1)
|
|
{
|
|
case 0: /* no lacing */
|
|
*laces = 1;
|
|
lace_size = calloc(*laces, sizeof(uint32_t));
|
|
lace_size[0] = *size;
|
|
break;
|
|
|
|
case 1: /* xiph lacing */
|
|
case 2: /* fixed-size lacing */
|
|
case 3: /* EBML lacing */
|
|
*laces = *buffer++;
|
|
(*size)--;
|
|
(*laces)++;
|
|
lace_size = calloc(*laces, sizeof(uint32_t));
|
|
|
|
switch ((flags & 0x06) >> 1)
|
|
{
|
|
case 1: /* xiph lacing */
|
|
for (i=0; i < *laces-1; i++)
|
|
{
|
|
lace_size[i] = 0;
|
|
do
|
|
{
|
|
lace_size[i] += *buffer;
|
|
(*size)--;
|
|
} while (*buffer++ == 0xFF);
|
|
total += lace_size[i];
|
|
}
|
|
lace_size[i] = *size - total;
|
|
break;
|
|
|
|
case 2: /* fixed-size lacing */
|
|
for (i=0; i < *laces; i++)
|
|
lace_size[i] = *size / *laces;
|
|
break;
|
|
|
|
case 3: /* EBML lacing */
|
|
{
|
|
int l;
|
|
uint64_t num = ebml_read_vlen_uint (buffer, &l);
|
|
if (num == EBML_UINT_INVALID) {
|
|
free(lace_size);
|
|
return 1;
|
|
}
|
|
buffer += l;
|
|
*size -= l;
|
|
|
|
total = lace_size[0] = num;
|
|
for (i=1; i < *laces-1; i++)
|
|
{
|
|
int64_t snum;
|
|
snum = ebml_read_vlen_int (buffer, &l);
|
|
if (snum == EBML_INT_INVALID) {
|
|
free(lace_size);
|
|
return 1;
|
|
}
|
|
buffer += l;
|
|
*size -= l;
|
|
lace_size[i] = lace_size[i-1] + snum;
|
|
total += lace_size[i];
|
|
}
|
|
lace_size[i] = *size - total;
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
*all_lace_sizes = lace_size;
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
clear_subtitles(demuxer_t *demuxer, uint64_t timecode, int clear_all);
|
|
|
|
static void
|
|
handle_subtitles(demuxer_t *demuxer, mkv_track_t *track, char *block,
|
|
int64_t size, uint64_t block_duration, uint64_t timecode)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
char *ptr1, *ptr2;
|
|
int state, i;
|
|
|
|
if (block_duration == 0)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_NoBlockDurationForSubtitleTrackFound);
|
|
return;
|
|
}
|
|
|
|
#ifdef USE_ASS
|
|
if (ass_enabled && track->subtitle_type == MATROSKA_SUBTYPE_SSA) {
|
|
ass_process_chunk(track->sh_sub.ass_track, block, size, (long long)timecode, (long long)block_duration);
|
|
return;
|
|
}
|
|
clear_subtitles(demuxer, timecode, 1);
|
|
#endif
|
|
|
|
ptr1 = block;
|
|
while (ptr1 - block <= size && (*ptr1 == '\n' || *ptr1 == '\r'))
|
|
ptr1++;
|
|
ptr2 = block + size - 1;
|
|
while (ptr2 >= block && (*ptr2 == '\n' || *ptr2 == '\r'))
|
|
{
|
|
*ptr2 = 0;
|
|
ptr2--;
|
|
}
|
|
|
|
if (mkv_d->subs.lines > SUB_MAX_TEXT - 2)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_TooManySublines);
|
|
return;
|
|
}
|
|
ptr2 = mkv_d->subs.text[mkv_d->subs.lines];
|
|
state = 0;
|
|
|
|
if (track->subtitle_type == MATROSKA_SUBTYPE_SSA)
|
|
{
|
|
/* Find text section. */
|
|
for (i=0; i < 8 && *ptr1 != '\0'; ptr1++)
|
|
if (*ptr1 == ',')
|
|
i++;
|
|
if (*ptr1 == '\0') /* Broken line? */
|
|
return;
|
|
|
|
/* Load text. */
|
|
while (ptr1 - block < size)
|
|
{
|
|
if (*ptr1 == '{')
|
|
state = 1;
|
|
else if (*ptr1 == '}' && state == 1)
|
|
state = 2;
|
|
|
|
if (state == 0)
|
|
{
|
|
*ptr2++ = *ptr1;
|
|
if (ptr2 - mkv_d->subs.text[mkv_d->subs.lines] >= 255)
|
|
break;
|
|
}
|
|
ptr1++;
|
|
|
|
/* Newline */
|
|
while (ptr1+1-block < size && *ptr1 == '\\' && (*(ptr1+1)|0x20) == 'n')
|
|
{
|
|
mkv_d->clear_subs_at[mkv_d->subs.lines++]
|
|
= timecode + block_duration;
|
|
*ptr2 = '\0';
|
|
if (mkv_d->subs.lines >= SUB_MAX_TEXT)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_TooManySublinesSkippingAfterFirst,
|
|
SUB_MAX_TEXT);
|
|
mkv_d->subs.lines--;
|
|
ptr1=block+size;
|
|
break;
|
|
}
|
|
ptr2 = mkv_d->subs.text[mkv_d->subs.lines];
|
|
ptr1 += 2;
|
|
}
|
|
|
|
if (state == 2)
|
|
state = 0;
|
|
}
|
|
*ptr2 = '\0';
|
|
}
|
|
else
|
|
{
|
|
while (ptr1 - block < size)
|
|
{
|
|
if (*ptr1 == '\n' || *ptr1 == '\r')
|
|
{
|
|
if (state == 0) /* normal char --> newline */
|
|
{
|
|
*ptr2 = '\0';
|
|
mkv_d->clear_subs_at[mkv_d->subs.lines++]
|
|
= timecode + block_duration;
|
|
if (mkv_d->subs.lines >= SUB_MAX_TEXT)
|
|
{
|
|
mp_msg (MSGT_DEMUX, MSGL_WARN,
|
|
MSGTR_MPDEMUX_MKV_TooManySublinesSkippingAfterFirst,
|
|
SUB_MAX_TEXT);
|
|
mkv_d->subs.lines--;
|
|
ptr1=block+size;
|
|
break;
|
|
}
|
|
ptr2 = mkv_d->subs.text[mkv_d->subs.lines];
|
|
state = 1;
|
|
}
|
|
}
|
|
else if (*ptr1 == '<') /* skip HTML tags */
|
|
state = 2;
|
|
else if (*ptr1 == '>')
|
|
state = 0;
|
|
else if (state != 2) /* normal character */
|
|
{
|
|
state = 0;
|
|
if ((ptr2 - mkv_d->subs.text[mkv_d->subs.lines]) < 255)
|
|
*ptr2++ = *ptr1;
|
|
}
|
|
ptr1++;
|
|
}
|
|
*ptr2 = '\0';
|
|
}
|
|
mkv_d->clear_subs_at[mkv_d->subs.lines++] = timecode + block_duration;
|
|
|
|
sub_utf8 = 1;
|
|
#ifdef USE_ASS
|
|
if (ass_enabled) {
|
|
mkv_d->subs.start = timecode / 10;
|
|
mkv_d->subs.end = (timecode + block_duration) / 10;
|
|
ass_process_subtitle(track->sh_sub.ass_track, &mkv_d->subs);
|
|
} else
|
|
#endif
|
|
vo_sub = &mkv_d->subs;
|
|
vo_osd_changed (OSDTYPE_SUBTITLE);
|
|
}
|
|
|
|
static void
|
|
clear_subtitles(demuxer_t *demuxer, uint64_t timecode, int clear_all)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
int i, lines_cut = 0;
|
|
char *tmp;
|
|
|
|
/* Clear all? */
|
|
if (clear_all)
|
|
{
|
|
lines_cut = mkv_d->subs.lines;
|
|
mkv_d->subs.lines = 0;
|
|
#ifdef USE_ASS
|
|
if (!ass_enabled)
|
|
#endif
|
|
if (lines_cut)
|
|
{
|
|
vo_sub = &mkv_d->subs;
|
|
vo_osd_changed (OSDTYPE_SUBTITLE);
|
|
}
|
|
return;
|
|
}
|
|
|
|
/* Clear the subtitles if they're obsolete now. */
|
|
for (i=0; i < mkv_d->subs.lines; i++)
|
|
{
|
|
if (mkv_d->clear_subs_at[i] <= timecode)
|
|
{
|
|
tmp = mkv_d->subs.text[i];
|
|
memmove (mkv_d->subs.text+i, mkv_d->subs.text+i+1,
|
|
(mkv_d->subs.lines-i-1) * sizeof (*mkv_d->subs.text));
|
|
memmove (mkv_d->clear_subs_at+i, mkv_d->clear_subs_at+i+1,
|
|
(mkv_d->subs.lines-i-1) * sizeof (*mkv_d->clear_subs_at));
|
|
mkv_d->subs.text[--mkv_d->subs.lines] = tmp;
|
|
i--;
|
|
lines_cut = 1;
|
|
}
|
|
}
|
|
#ifdef USE_ASS
|
|
if (!ass_enabled)
|
|
#endif
|
|
if (lines_cut)
|
|
{
|
|
vo_sub = &mkv_d->subs;
|
|
vo_osd_changed (OSDTYPE_SUBTITLE);
|
|
}
|
|
}
|
|
|
|
// Taken from demux_real.c. Thanks to the original developpers :)
|
|
#define SKIP_BITS(n) buffer <<= n
|
|
#define SHOW_BITS(n) ((buffer) >> (32 - (n)))
|
|
|
|
static float real_fix_timestamp(mkv_track_t *track, unsigned char *s,
|
|
int timestamp) {
|
|
float v_pts;
|
|
uint32_t buffer = (s[0] << 24) + (s[1] << 16) + (s[2] << 8) + s[3];
|
|
int kf = timestamp;
|
|
int pict_type;
|
|
int orig_kf;
|
|
|
|
if (!strcmp(track->codec_id, MKV_V_REALV30) ||
|
|
!strcmp(track->codec_id, MKV_V_REALV40)) {
|
|
|
|
if (!strcmp(track->codec_id, MKV_V_REALV30)) {
|
|
SKIP_BITS(3);
|
|
pict_type = SHOW_BITS(2);
|
|
SKIP_BITS(2 + 7);
|
|
}else{
|
|
SKIP_BITS(1);
|
|
pict_type = SHOW_BITS(2);
|
|
SKIP_BITS(2 + 7 + 3);
|
|
}
|
|
kf = SHOW_BITS(13); // kf= 2*SHOW_BITS(12);
|
|
orig_kf = kf;
|
|
if (pict_type <= 1) {
|
|
// I frame, sync timestamps:
|
|
track->rv_kf_base = timestamp - kf;
|
|
mp_msg(MSGT_DEMUX, MSGL_DBG2, "\nTS: base=%08X\n", track->rv_kf_base);
|
|
kf = timestamp;
|
|
} else {
|
|
// P/B frame, merge timestamps:
|
|
int tmp = timestamp - track->rv_kf_base;
|
|
kf |= tmp & (~0x1fff); // combine with packet timestamp
|
|
if (kf < (tmp - 4096)) // workaround wrap-around problems
|
|
kf += 8192;
|
|
else if (kf > (tmp + 4096))
|
|
kf -= 8192;
|
|
kf += track->rv_kf_base;
|
|
}
|
|
if (pict_type != 3) { // P || I frame -> swap timestamps
|
|
int tmp = kf;
|
|
kf = track->rv_kf_pts;
|
|
track->rv_kf_pts = tmp;
|
|
}
|
|
mp_msg(MSGT_DEMUX, MSGL_DBG2, "\nTS: %08X -> %08X (%04X) %d %02X %02X %02X "
|
|
"%02X %5d\n", timestamp, kf, orig_kf, pict_type, s[0], s[1], s[2],
|
|
s[3], kf - (int)(1000.0 * track->rv_pts));
|
|
}
|
|
v_pts = kf * 0.001f;
|
|
track->rv_pts = v_pts;
|
|
|
|
return v_pts;
|
|
}
|
|
|
|
static void
|
|
handle_realvideo (demuxer_t *demuxer, mkv_track_t *track, uint8_t *buffer,
|
|
uint32_t size, int block_bref)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
demux_packet_t *dp;
|
|
dp_hdr_t *hdr;
|
|
uint8_t chunks;
|
|
int isize;
|
|
#ifdef WORDS_BIGENDIAN
|
|
uint8_t *p;
|
|
int i;
|
|
#endif
|
|
|
|
chunks = *buffer++;
|
|
isize = --size - (chunks+1)*8;
|
|
dp = new_demux_packet (sizeof (*hdr) + size);
|
|
memcpy (dp->buffer + sizeof(*hdr), buffer + (chunks+1)*8, isize);
|
|
#ifdef WORDS_BIGENDIAN
|
|
p = (uint8_t *)(dp->buffer + sizeof(*hdr) + isize);
|
|
for (i = 0; i<(chunks+1)*8; i+=4) {
|
|
p[i] = *((uint8_t *)buffer+i+3);
|
|
p[i+1] = *((uint8_t *)buffer+i+2);
|
|
p[i+2] = *((uint8_t *)buffer+i+1);
|
|
p[i+3] = *((uint8_t *)buffer+i);
|
|
}
|
|
#else
|
|
memcpy (dp->buffer + sizeof(*hdr) + isize, buffer, (chunks+1)*8);
|
|
#endif
|
|
|
|
hdr = (dp_hdr_t *) dp->buffer;
|
|
hdr->len = isize;
|
|
hdr->chunks = chunks;
|
|
hdr->timestamp = mkv_d->last_pts * 1000;
|
|
hdr->chunktab = sizeof(*hdr) + isize;
|
|
|
|
dp->len = sizeof(*hdr) + size;
|
|
if (mkv_d->v_skip_to_keyframe)
|
|
{
|
|
dp->pts = mkv_d->last_pts;
|
|
track->rv_kf_base = 0;
|
|
track->rv_kf_pts = hdr->timestamp;
|
|
}
|
|
else
|
|
dp->pts = real_fix_timestamp (track, dp->buffer + sizeof(*hdr),
|
|
hdr->timestamp);
|
|
dp->pos = demuxer->filepos;
|
|
dp->flags = block_bref ? 0 : 0x10;
|
|
|
|
ds_add_packet(demuxer->video, dp);
|
|
}
|
|
|
|
static void
|
|
handle_realaudio (demuxer_t *demuxer, mkv_track_t *track, uint8_t *buffer,
|
|
uint32_t size, int block_bref)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
int sps = track->sub_packet_size;
|
|
int sph = track->sub_packet_h;
|
|
int cfs = track->coded_framesize;
|
|
int w = track->audiopk_size;
|
|
int spc = track->sub_packet_cnt;
|
|
demux_packet_t *dp;
|
|
int x;
|
|
|
|
if ((track->a_formattag == mmioFOURCC('2', '8', '_', '8')) ||
|
|
(track->a_formattag == mmioFOURCC('c', 'o', 'o', 'k')) ||
|
|
(track->a_formattag == mmioFOURCC('a', 't', 'r', 'c')) ||
|
|
(track->a_formattag == mmioFOURCC('s', 'i', 'p', 'r')))
|
|
{
|
|
// if(!block_bref)
|
|
// spc = track->sub_packet_cnt = 0;
|
|
switch (track->a_formattag) {
|
|
case mmioFOURCC('2', '8', '_', '8'):
|
|
for (x = 0; x < sph / 2; x++)
|
|
memcpy(track->audio_buf + x * 2 * w + spc * cfs, buffer + cfs * x, cfs);
|
|
break;
|
|
case mmioFOURCC('c', 'o', 'o', 'k'):
|
|
case mmioFOURCC('a', 't', 'r', 'c'):
|
|
for (x = 0; x < w / sps; x++)
|
|
memcpy(track->audio_buf + sps * (sph * x + ((sph + 1) / 2) * (spc & 1) + (spc >> 1)), buffer + sps * x, sps);
|
|
break;
|
|
case mmioFOURCC('s', 'i', 'p', 'r'):
|
|
memcpy(track->audio_buf + spc * w, buffer, w);
|
|
if (spc == sph - 1)
|
|
{
|
|
int n;
|
|
int bs = sph * w * 2 / 96; // nibbles per subpacket
|
|
// Perform reordering
|
|
for(n=0; n < 38; n++)
|
|
{
|
|
int j;
|
|
int i = bs * sipr_swaps[n][0];
|
|
int o = bs * sipr_swaps[n][1];
|
|
// swap nibbles of block 'i' with 'o' TODO: optimize
|
|
for(j = 0;j < bs; j++)
|
|
{
|
|
int x = (i & 1) ? (track->audio_buf[i >> 1] >> 4) : (track->audio_buf[i >> 1] & 0x0F);
|
|
int y = (o & 1) ? (track->audio_buf[o >> 1] >> 4) : (track->audio_buf[o >> 1] & 0x0F);
|
|
if(o & 1)
|
|
track->audio_buf[o >> 1] = (track->audio_buf[o >> 1] & 0x0F) | (x << 4);
|
|
else
|
|
track->audio_buf[o >> 1] = (track->audio_buf[o >> 1] & 0xF0) | x;
|
|
if(i & 1)
|
|
track->audio_buf[i >> 1] = (track->audio_buf[i >> 1] & 0x0F) | (y << 4);
|
|
else
|
|
track->audio_buf[i >> 1] = (track->audio_buf[i >> 1] & 0xF0) | y;
|
|
++i; ++o;
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
track->audio_timestamp[track->sub_packet_cnt] = (track->ra_pts == mkv_d->last_pts) ? 0 : (mkv_d->last_pts);
|
|
track->ra_pts = mkv_d->last_pts;
|
|
if (track->sub_packet_cnt == 0)
|
|
track->audio_filepos = demuxer->filepos;
|
|
if (++(track->sub_packet_cnt) == sph)
|
|
{
|
|
int apk_usize = ((WAVEFORMATEX*)((sh_audio_t*)demuxer->audio->sh)->wf)->nBlockAlign;
|
|
track->sub_packet_cnt = 0;
|
|
// Release all the audio packets
|
|
for (x = 0; x < sph*w/apk_usize; x++)
|
|
{
|
|
dp = new_demux_packet(apk_usize);
|
|
memcpy(dp->buffer, track->audio_buf + x * apk_usize, apk_usize);
|
|
/* Put timestamp only on packets that correspond to original audio packets in file */
|
|
dp->pts = (x * apk_usize % w) ? 0 : track->audio_timestamp[x * apk_usize / w];
|
|
dp->pos = track->audio_filepos; // all equal
|
|
dp->flags = x ? 0 : 0x10; // Mark first packet as keyframe
|
|
ds_add_packet(demuxer->audio, dp);
|
|
}
|
|
}
|
|
} else { // Not a codec that require reordering
|
|
dp = new_demux_packet (size);
|
|
memcpy(dp->buffer, buffer, size);
|
|
if (track->ra_pts == mkv_d->last_pts && !mkv_d->a_skip_to_keyframe)
|
|
dp->pts = 0;
|
|
else
|
|
dp->pts = mkv_d->last_pts;
|
|
track->ra_pts = mkv_d->last_pts;
|
|
|
|
dp->pos = demuxer->filepos;
|
|
dp->flags = block_bref ? 0 : 0x10;
|
|
ds_add_packet (demuxer->audio, dp);
|
|
}
|
|
}
|
|
|
|
/** Reorder timecodes and add cached demux packets to the queues.
|
|
*
|
|
* Timecode reordering is needed if a video track contains B frames that
|
|
* are timestamped in display order (e.g. MPEG-1, MPEG-2 or "native" MPEG-4).
|
|
* MPlayer doesn't like timestamps in display order. This function adjusts
|
|
* the timestamp of cached frames (which are exactly one I/P frame followed
|
|
* by one or more B frames) so that they are in coding order again.
|
|
*
|
|
* Example: The track with 25 FPS contains four frames with the timecodes
|
|
* I at 0ms, P at 120ms, B at 40ms and B at 80ms. As soon as the next I
|
|
* or P frame arrives these timecodes can be changed to I at 0ms, P at 40ms,
|
|
* B at 80ms and B at 120ms.
|
|
*
|
|
* This works for simple H.264 B-frame pyramids, but not for arbitrary orders.
|
|
*
|
|
* \param demuxer The Matroska demuxer struct for this instance.
|
|
* \param track The track structure whose cache should be handled.
|
|
*/
|
|
static void
|
|
flush_cached_dps (demuxer_t *demuxer, mkv_track_t *track)
|
|
{
|
|
int i, ok;
|
|
|
|
if (track->num_cached_dps == 0)
|
|
return;
|
|
|
|
do {
|
|
ok = 1;
|
|
for (i = 1; i < track->num_cached_dps; i++)
|
|
if (track->cached_dps[i - 1]->pts > track->cached_dps[i]->pts) {
|
|
float tmp_pts = track->cached_dps[i - 1]->pts;
|
|
track->cached_dps[i - 1]->pts = track->cached_dps[i]->pts;
|
|
track->cached_dps[i]->pts = tmp_pts;
|
|
ok = 0;
|
|
}
|
|
} while (!ok);
|
|
|
|
for (i = 0; i < track->num_cached_dps; i++)
|
|
ds_add_packet (demuxer->video, track->cached_dps[i]);
|
|
track->num_cached_dps = 0;
|
|
}
|
|
|
|
/** Cache video frames if timecodes have to be reordered.
|
|
*
|
|
* Timecode reordering is needed if a video track contains B frames that
|
|
* are timestamped in display order (e.g. MPEG-1, MPEG-2 or "native" MPEG-4).
|
|
* This function takes in a Matroska block read from the file, allocates a
|
|
* demux packet for it, fills in its values, allocates space for storing
|
|
* pointers to the cached demux packets and adds the packet to it. If
|
|
* the packet contains an I or a P frame then ::flush_cached_dps is called
|
|
* in order to send the old cached frames downstream.
|
|
*
|
|
* \param demuxer The Matroska demuxer struct for this instance.
|
|
* \param track The packet is meant for this track.
|
|
* \param buffer The actual frame contents.
|
|
* \param size The frame size in bytes.
|
|
* \param block_bref A relative timecode (backward reference). If it is \c 0
|
|
* then the frame is an I frame.
|
|
* \param block_fref A relative timecode (forward reference). If it is \c 0
|
|
* then the frame is either an I frame or a P frame depending on the value
|
|
* of \a block_bref. Otherwise it's a B frame.
|
|
*/
|
|
static void
|
|
handle_video_bframes (demuxer_t *demuxer, mkv_track_t *track, uint8_t *buffer,
|
|
uint32_t size, int block_bref, int block_fref)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
demux_packet_t *dp;
|
|
|
|
dp = new_demux_packet (size);
|
|
memcpy(dp->buffer, buffer, size);
|
|
dp->pos = demuxer->filepos;
|
|
dp->pts = mkv_d->last_pts;
|
|
if ((track->num_cached_dps > 0) && (dp->pts < track->max_pts))
|
|
block_fref = 1;
|
|
if (block_fref == 0) /* I or P frame */
|
|
flush_cached_dps (demuxer, track);
|
|
if (block_bref != 0) /* I frame, don't cache it */
|
|
dp->flags = 0x10;
|
|
if ((track->num_cached_dps + 1) > track->num_allocated_dps)
|
|
{
|
|
track->cached_dps = (demux_packet_t **)
|
|
realloc(track->cached_dps, (track->num_cached_dps + 10) *
|
|
sizeof(demux_packet_t *));
|
|
track->num_allocated_dps += 10;
|
|
}
|
|
track->cached_dps[track->num_cached_dps] = dp;
|
|
track->num_cached_dps++;
|
|
if (dp->pts > track->max_pts)
|
|
track->max_pts = dp->pts;
|
|
}
|
|
|
|
static int
|
|
handle_block (demuxer_t *demuxer, uint8_t *block, uint64_t length,
|
|
uint64_t block_duration, int64_t block_bref, int64_t block_fref, uint8_t simpleblock)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
mkv_track_t *track = NULL;
|
|
demux_stream_t *ds = NULL;
|
|
uint64_t old_length;
|
|
int64_t tc;
|
|
uint32_t *lace_size;
|
|
uint8_t laces, flags;
|
|
int i, num, tmp, use_this_block = 1;
|
|
float current_pts;
|
|
int16_t time;
|
|
|
|
/* first byte(s): track num */
|
|
num = ebml_read_vlen_uint (block, &tmp);
|
|
block += tmp;
|
|
/* time (relative to cluster time) */
|
|
time = block[0] << 8 | block[1];
|
|
block += 2;
|
|
length -= tmp + 2;
|
|
old_length = length;
|
|
flags = block[0];
|
|
if (demux_mkv_read_block_lacing (block, &length, &laces, &lace_size))
|
|
return 0;
|
|
block += old_length - length;
|
|
|
|
tc = ((time*mkv_d->tc_scale+mkv_d->cluster_tc) /1000000.0 - mkv_d->first_tc);
|
|
if (tc < 0)
|
|
tc = 0;
|
|
if (mkv_d->stop_timecode > 0 && tc > mkv_d->stop_timecode) {
|
|
free(lace_size);
|
|
return -1;
|
|
}
|
|
current_pts = tc / 1000.0;
|
|
|
|
clear_subtitles(demuxer, tc, 0);
|
|
|
|
for (i=0; i<mkv_d->num_tracks; i++)
|
|
if (mkv_d->tracks[i]->tnum == num) {
|
|
track = mkv_d->tracks[i];
|
|
break;
|
|
}
|
|
if (track == NULL)
|
|
{
|
|
free(lace_size);
|
|
return 1;
|
|
}
|
|
if (num == demuxer->audio->id)
|
|
{
|
|
ds = demuxer->audio;
|
|
|
|
if (mkv_d->a_skip_to_keyframe)
|
|
{
|
|
if (simpleblock)
|
|
{
|
|
if (!(flags&0x80)) /*current frame isn't a keyframe*/
|
|
use_this_block = 0;
|
|
}
|
|
else if (block_bref != 0)
|
|
use_this_block = 0;
|
|
}
|
|
else if (mkv_d->v_skip_to_keyframe)
|
|
use_this_block = 0;
|
|
|
|
if (track->fix_i_bps && use_this_block)
|
|
{
|
|
sh_audio_t *sh = (sh_audio_t *) ds->sh;
|
|
|
|
if (block_duration != 0)
|
|
{
|
|
sh->i_bps = length * 1000 / block_duration;
|
|
track->fix_i_bps = 0;
|
|
}
|
|
else if (track->qt_last_a_pts == 0.0)
|
|
track->qt_last_a_pts = current_pts;
|
|
else if(track->qt_last_a_pts != current_pts)
|
|
{
|
|
sh->i_bps = length / (current_pts - track->qt_last_a_pts);
|
|
track->fix_i_bps = 0;
|
|
}
|
|
}
|
|
}
|
|
else if (tc < mkv_d->skip_to_timecode)
|
|
use_this_block = 0;
|
|
else if (num == demuxer->video->id)
|
|
{
|
|
ds = demuxer->video;
|
|
if (mkv_d->v_skip_to_keyframe)
|
|
{
|
|
if (simpleblock)
|
|
{
|
|
if (!(flags&0x80)) /*current frame isn't a keyframe*/
|
|
use_this_block = 0;
|
|
}
|
|
else if (block_bref != 0 || block_fref != 0)
|
|
use_this_block = 0;
|
|
}
|
|
}
|
|
else if (num == demuxer->sub->id)
|
|
{
|
|
ds = demuxer->sub;
|
|
if (track->subtitle_type != MATROSKA_SUBTYPE_VOBSUB)
|
|
{
|
|
if (!mkv_d->v_skip_to_keyframe)
|
|
handle_subtitles (demuxer, track, block, length,
|
|
block_duration, tc);
|
|
use_this_block = 0;
|
|
}
|
|
}
|
|
else
|
|
use_this_block = 0;
|
|
|
|
if (use_this_block)
|
|
{
|
|
mkv_d->last_pts = current_pts;
|
|
mkv_d->last_filepos = demuxer->filepos;
|
|
|
|
for (i=0; i < laces; i++)
|
|
{
|
|
if (ds == demuxer->video && track->realmedia)
|
|
handle_realvideo (demuxer, track, block, lace_size[i], block_bref);
|
|
else if (ds == demuxer->audio && track->realmedia)
|
|
handle_realaudio (demuxer, track, block, lace_size[i], block_bref);
|
|
else if (ds == demuxer->video && track->reorder_timecodes)
|
|
handle_video_bframes (demuxer, track, block, lace_size[i],
|
|
block_bref, block_fref);
|
|
else
|
|
{
|
|
int modified, size = lace_size[i];
|
|
demux_packet_t *dp;
|
|
uint8_t *buffer;
|
|
modified = demux_mkv_decode (track, block, &buffer, &size, 1);
|
|
if (buffer)
|
|
{
|
|
dp = new_demux_packet (size);
|
|
memcpy (dp->buffer, buffer, size);
|
|
if (modified)
|
|
free (buffer);
|
|
dp->flags = (block_bref == 0 && block_fref == 0) ? 0x10 : 0;
|
|
/* If default_duration is 0, assume no pts value is known
|
|
* for packets after the first one (rather than all pts
|
|
* values being the same) */
|
|
if (i == 0 || track->default_duration)
|
|
dp->pts = mkv_d->last_pts + i * track->default_duration;
|
|
ds_add_packet (ds, dp);
|
|
}
|
|
}
|
|
block += lace_size[i];
|
|
}
|
|
|
|
if (ds == demuxer->video)
|
|
{
|
|
mkv_d->v_skip_to_keyframe = 0;
|
|
mkv_d->skip_to_timecode = 0;
|
|
}
|
|
else if (ds == demuxer->audio)
|
|
mkv_d->a_skip_to_keyframe = 0;
|
|
|
|
free(lace_size);
|
|
return 1;
|
|
}
|
|
|
|
free(lace_size);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
demux_mkv_fill_buffer (demuxer_t *demuxer, demux_stream_t *ds)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t l;
|
|
int il, tmp;
|
|
|
|
while (1)
|
|
{
|
|
while (mkv_d->cluster_size > 0)
|
|
{
|
|
uint64_t block_duration = 0, block_length = 0;
|
|
int64_t block_bref = 0, block_fref = 0;
|
|
uint8_t *block = NULL;
|
|
|
|
while (mkv_d->blockgroup_size > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_BLOCKDURATION:
|
|
{
|
|
block_duration = ebml_read_uint (s, &l);
|
|
if (block_duration == EBML_UINT_INVALID)
|
|
return 0;
|
|
block_duration *= mkv_d->tc_scale / 1000000.0;
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_BLOCK:
|
|
block_length = ebml_read_length (s, &tmp);
|
|
free(block);
|
|
block = malloc (block_length);
|
|
demuxer->filepos = stream_tell (s);
|
|
if (stream_read (s,block,block_length) != (int) block_length)
|
|
{
|
|
free(block);
|
|
return 0;
|
|
}
|
|
l = tmp + block_length;
|
|
break;
|
|
|
|
case MATROSKA_ID_REFERENCEBLOCK:
|
|
{
|
|
int64_t num = ebml_read_int (s, &l);
|
|
if (num == EBML_INT_INVALID)
|
|
return 0;
|
|
if (num <= 0)
|
|
block_bref = num;
|
|
else
|
|
block_fref = num;
|
|
break;
|
|
}
|
|
|
|
case EBML_ID_INVALID:
|
|
return 0;
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
mkv_d->blockgroup_size -= l + il;
|
|
mkv_d->cluster_size -= l + il;
|
|
}
|
|
|
|
if (block)
|
|
{
|
|
int res = handle_block (demuxer, block, block_length,
|
|
block_duration, block_bref, block_fref, 0);
|
|
free (block);
|
|
if (res < 0)
|
|
return 0;
|
|
if (res)
|
|
return 1;
|
|
}
|
|
|
|
if (mkv_d->cluster_size > 0)
|
|
{
|
|
switch (ebml_read_id (s, &il))
|
|
{
|
|
case MATROSKA_ID_CLUSTERTIMECODE:
|
|
{
|
|
uint64_t num = ebml_read_uint (s, &l);
|
|
if (num == EBML_UINT_INVALID)
|
|
return 0;
|
|
if (!mkv_d->has_first_tc)
|
|
{
|
|
mkv_d->first_tc = num * mkv_d->tc_scale / 1000000.0;
|
|
mkv_d->has_first_tc = 1;
|
|
}
|
|
mkv_d->cluster_tc = num * mkv_d->tc_scale;
|
|
break;
|
|
}
|
|
|
|
case MATROSKA_ID_BLOCKGROUP:
|
|
mkv_d->blockgroup_size = ebml_read_length (s, &tmp);
|
|
l = tmp;
|
|
break;
|
|
|
|
case MATROSKA_ID_SIMPLEBLOCK:
|
|
{
|
|
int res;
|
|
block_length = ebml_read_length (s, &tmp);
|
|
block = malloc (block_length);
|
|
demuxer->filepos = stream_tell (s);
|
|
if (stream_read (s,block,block_length) != (int) block_length)
|
|
{
|
|
free(block);
|
|
return 0;
|
|
}
|
|
l = tmp + block_length;
|
|
res = handle_block (demuxer, block, block_length,
|
|
block_duration, block_bref, block_fref, 1);
|
|
free (block);
|
|
mkv_d->cluster_size -= l + il;
|
|
if (res < 0)
|
|
return 0;
|
|
else if (res)
|
|
return 1;
|
|
else mkv_d->cluster_size += l + il;
|
|
break;
|
|
}
|
|
case EBML_ID_INVALID:
|
|
return 0;
|
|
|
|
default:
|
|
ebml_read_skip (s, &l);
|
|
break;
|
|
}
|
|
mkv_d->cluster_size -= l + il;
|
|
}
|
|
}
|
|
|
|
if (ebml_read_id (s, &il) != MATROSKA_ID_CLUSTER)
|
|
return 0;
|
|
add_cluster_position(mkv_d, stream_tell(s)-il);
|
|
mkv_d->cluster_size = ebml_read_length (s, NULL);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
demux_mkv_seek (demuxer_t *demuxer, float rel_seek_secs, float audio_delay, int flags)
|
|
{
|
|
free_cached_dps (demuxer);
|
|
if (!(flags & 2)) /* time in secs */
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
int64_t target_timecode = 0, diff, min_diff=0xFFFFFFFL;
|
|
int i;
|
|
|
|
if (!(flags & 1)) /* relative seek */
|
|
target_timecode = (int64_t) (mkv_d->last_pts * 1000.0);
|
|
target_timecode += (int64_t)(rel_seek_secs * 1000.0);
|
|
if (target_timecode < 0)
|
|
target_timecode = 0;
|
|
|
|
if (mkv_d->indexes == NULL) /* no index was found */
|
|
{
|
|
uint64_t target_filepos, cluster_pos, max_pos;
|
|
|
|
target_filepos = (uint64_t) (target_timecode * mkv_d->last_filepos
|
|
/ (mkv_d->last_pts * 1000.0));
|
|
|
|
max_pos = mkv_d->cluster_positions[mkv_d->num_cluster_pos-1];
|
|
if (target_filepos > max_pos)
|
|
{
|
|
if ((off_t) max_pos > stream_tell (s))
|
|
stream_seek (s, max_pos);
|
|
else
|
|
stream_seek (s, stream_tell (s) + mkv_d->cluster_size);
|
|
/* parse all the clusters upto target_filepos */
|
|
while (!s->eof && stream_tell(s) < (off_t) target_filepos)
|
|
{
|
|
switch (ebml_read_id (s, &i))
|
|
{
|
|
case MATROSKA_ID_CLUSTER:
|
|
add_cluster_position(mkv_d, (uint64_t) stream_tell(s)-i);
|
|
break;
|
|
|
|
case MATROSKA_ID_CUES:
|
|
demux_mkv_read_cues (demuxer);
|
|
break;
|
|
}
|
|
ebml_read_skip (s, NULL);
|
|
}
|
|
if (s->eof)
|
|
stream_reset(s);
|
|
}
|
|
|
|
if (mkv_d->indexes == NULL)
|
|
{
|
|
cluster_pos = mkv_d->cluster_positions[0];
|
|
/* Let's find the nearest cluster */
|
|
for (i=0; i < mkv_d->num_cluster_pos; i++)
|
|
{
|
|
diff = mkv_d->cluster_positions[i] - target_filepos;
|
|
if (rel_seek_secs < 0 && diff < 0 && -diff < min_diff)
|
|
{
|
|
cluster_pos = mkv_d->cluster_positions[i];
|
|
min_diff = -diff;
|
|
}
|
|
else if (rel_seek_secs > 0
|
|
&& (diff < 0 ? -1 * diff : diff) < min_diff)
|
|
{
|
|
cluster_pos = mkv_d->cluster_positions[i];
|
|
min_diff = diff < 0 ? -1 * diff : diff;
|
|
}
|
|
}
|
|
mkv_d->cluster_size = mkv_d->blockgroup_size = 0;
|
|
stream_seek (s, cluster_pos);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
mkv_index_t *index = NULL;
|
|
int seek_id = (demuxer->video->id < 0) ? demuxer->audio->id : demuxer->video->id;
|
|
|
|
/* let's find the entry in the indexes with the smallest */
|
|
/* difference to the wanted timecode. */
|
|
for (i=0; i < mkv_d->num_indexes; i++)
|
|
if (mkv_d->indexes[i].tnum == seek_id)
|
|
{
|
|
diff = target_timecode + mkv_d->first_tc -
|
|
(int64_t) mkv_d->indexes[i].timecode * mkv_d->tc_scale / 1000000.0;
|
|
|
|
if ((flags & 1 || target_timecode <= mkv_d->last_pts*1000)
|
|
&& diff >= 0 && diff < min_diff)
|
|
{
|
|
min_diff = diff;
|
|
index = mkv_d->indexes + i;
|
|
}
|
|
else if (target_timecode > mkv_d->last_pts*1000
|
|
&& diff < 0 && -diff < min_diff)
|
|
{
|
|
min_diff = -diff;
|
|
index = mkv_d->indexes + i;
|
|
}
|
|
}
|
|
|
|
if (index) /* We've found an entry. */
|
|
{
|
|
mkv_d->cluster_size = mkv_d->blockgroup_size = 0;
|
|
stream_seek (s, index->filepos);
|
|
}
|
|
}
|
|
|
|
if (demuxer->video->id >= 0)
|
|
mkv_d->v_skip_to_keyframe = 1;
|
|
if (rel_seek_secs > 0.0)
|
|
mkv_d->skip_to_timecode = target_timecode;
|
|
mkv_d->a_skip_to_keyframe = 1;
|
|
|
|
/* Clear subtitles. */
|
|
if (target_timecode <= mkv_d->last_pts * 1000)
|
|
clear_subtitles(demuxer, 0, 1);
|
|
|
|
demux_mkv_fill_buffer(demuxer, NULL);
|
|
}
|
|
else if ((demuxer->movi_end <= 0) || !(flags & 1))
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] seek unsupported flags\n");
|
|
else
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
stream_t *s = demuxer->stream;
|
|
uint64_t target_filepos;
|
|
mkv_index_t *index = NULL;
|
|
int i;
|
|
|
|
if (mkv_d->indexes == NULL) /* no index was found */
|
|
{ /* I'm lazy... */
|
|
mp_msg (MSGT_DEMUX, MSGL_V, "[mkv] seek unsupported flags\n");
|
|
return;
|
|
}
|
|
|
|
target_filepos = (uint64_t)(demuxer->movi_end * rel_seek_secs);
|
|
for (i=0; i < mkv_d->num_indexes; i++)
|
|
if (mkv_d->indexes[i].tnum == demuxer->video->id)
|
|
if ((index == NULL) ||
|
|
((mkv_d->indexes[i].filepos >= target_filepos) &&
|
|
((index->filepos < target_filepos) ||
|
|
(mkv_d->indexes[i].filepos < index->filepos))))
|
|
index = &mkv_d->indexes[i];
|
|
|
|
if (!index)
|
|
return;
|
|
|
|
mkv_d->cluster_size = mkv_d->blockgroup_size = 0;
|
|
stream_seek (s, index->filepos);
|
|
|
|
if (demuxer->video->id >= 0)
|
|
mkv_d->v_skip_to_keyframe = 1;
|
|
mkv_d->skip_to_timecode = index->timecode;
|
|
mkv_d->a_skip_to_keyframe = 1;
|
|
|
|
/* Clear subtitles. */
|
|
if (index->timecode <= mkv_d->last_pts * 1000)
|
|
clear_subtitles(demuxer, 0, 1);
|
|
|
|
demux_mkv_fill_buffer(demuxer, NULL);
|
|
}
|
|
}
|
|
|
|
static int
|
|
demux_mkv_control (demuxer_t *demuxer, int cmd, void *arg)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
|
|
switch (cmd)
|
|
{
|
|
case DEMUXER_CTRL_GET_TIME_LENGTH:
|
|
if (mkv_d->duration == 0)
|
|
return DEMUXER_CTRL_DONTKNOW;
|
|
|
|
*((double *)arg) = (double)mkv_d->duration;
|
|
return DEMUXER_CTRL_OK;
|
|
|
|
case DEMUXER_CTRL_GET_PERCENT_POS:
|
|
if (mkv_d->duration == 0)
|
|
{
|
|
return DEMUXER_CTRL_DONTKNOW;
|
|
}
|
|
|
|
*((int *) arg) = (int) (100 * mkv_d->last_pts / mkv_d->duration);
|
|
return DEMUXER_CTRL_OK;
|
|
|
|
case DEMUXER_CTRL_SWITCH_AUDIO:
|
|
if (demuxer->audio && demuxer->audio->sh) {
|
|
sh_audio_t *sh = demuxer->a_streams[demuxer->audio->id];
|
|
int aid = *(int*)arg;
|
|
if (aid < 0)
|
|
aid = (sh->aid + 1) % mkv_d->last_aid;
|
|
if (aid != sh->aid) {
|
|
mkv_track_t *track = demux_mkv_find_track_by_num (mkv_d, aid, MATROSKA_TRACK_AUDIO);
|
|
if (track) {
|
|
demuxer->audio->id = track->tnum;
|
|
sh = demuxer->a_streams[demuxer->audio->id];
|
|
ds_free_packs(demuxer->audio);
|
|
}
|
|
}
|
|
*(int*)arg = sh->aid;
|
|
} else
|
|
*(int*)arg = -2;
|
|
return DEMUXER_CTRL_OK;
|
|
|
|
default:
|
|
return DEMUXER_CTRL_NOTIMPL;
|
|
}
|
|
}
|
|
|
|
/** \brief Change the current subtitle track and return its ID.
|
|
|
|
Changes the current subtitle track. If the new subtitle track is a
|
|
VobSub track then the SPU decoder will be re-initialized.
|
|
|
|
\param demuxer The demuxer whose subtitle track will be changed.
|
|
\param new_num The number of the new subtitle track. The number must be
|
|
between 0 and demux_mkv_num_subs - 1.
|
|
|
|
\returns The Matroska track number of the newly selected track.
|
|
*/
|
|
int
|
|
demux_mkv_change_subs (demuxer_t *demuxer, int new_num)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
mkv_track_t *track;
|
|
int i, num;
|
|
|
|
num = 0;
|
|
track = NULL;
|
|
for (i = 0; i < mkv_d->num_tracks; i++)
|
|
{
|
|
if ((mkv_d->tracks[i]->type == MATROSKA_TRACK_SUBTITLE) &&
|
|
(mkv_d->tracks[i]->subtitle_type != MATROSKA_SUBTYPE_UNKNOWN))
|
|
num++;
|
|
if (num == (new_num + 1))
|
|
{
|
|
track = mkv_d->tracks[i];
|
|
break;
|
|
}
|
|
}
|
|
if (track == NULL)
|
|
return -1;
|
|
|
|
if (demuxer->sub->sh == NULL)
|
|
demuxer->sub->sh = malloc(sizeof(sh_sub_t));
|
|
if (demuxer->sub->sh != NULL)
|
|
memcpy(demuxer->sub->sh, &track->sh_sub, sizeof(sh_sub_t));
|
|
|
|
return track->tnum;
|
|
}
|
|
|
|
/** \brief Get the language code for a subtitle track.
|
|
|
|
Retrieves the language code for a subtitle track if it is known.
|
|
If the language code is "und" then do not copy it ("und" = "undefined").
|
|
|
|
\param demuxer The demuxer to work on
|
|
\param track_num The n'th subtitle track to get the language from
|
|
\param lang Store the language here
|
|
\param maxlen The maximum number of characters to copy into lang
|
|
*/
|
|
void
|
|
demux_mkv_get_sub_lang(demuxer_t *demuxer, int track_num, char *lang,
|
|
int maxlen)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
mkv_track_t *track = demux_mkv_find_track_by_num (mkv_d, track_num, MATROSKA_TRACK_SUBTITLE);
|
|
if (track->language && strcmp(track->language, "und"))
|
|
strlcpy(lang, track->language, maxlen);
|
|
}
|
|
|
|
/** \brief Get the language code for an audio track.
|
|
|
|
Retrieves the language code for an audio track if it is known.
|
|
If the language code is "und" then do not copy it ("und" = "undefined").
|
|
|
|
\param demuxer The demuxer to work on
|
|
\param track_num The n'th audio track to get the language from
|
|
\param lang Store the language here
|
|
\param maxlen The maximum number of characters to copy into lang
|
|
*/
|
|
void
|
|
demux_mkv_get_audio_lang(demuxer_t *demuxer, int track_num, char *lang,
|
|
int maxlen)
|
|
{
|
|
mkv_demuxer_t *mkv_d = (mkv_demuxer_t *) demuxer->priv;
|
|
mkv_track_t *track = demux_mkv_find_track_by_num (mkv_d, track_num, MATROSKA_TRACK_AUDIO);
|
|
if (track->language && strcmp(track->language, "und"))
|
|
strlcpy(lang, track->language, maxlen);
|
|
}
|
|
|
|
|
|
demuxer_desc_t demuxer_desc_matroska = {
|
|
"Matroska demuxer",
|
|
"mkv",
|
|
"Matroska",
|
|
"Aurelien Jacobs",
|
|
"based on gstreamer demuxer by Ronald Bultje and demux_mkv.cpp by Moritz Bunkus",
|
|
DEMUXER_TYPE_MATROSKA,
|
|
1, // safe autodetect
|
|
demux_mkv_open,
|
|
demux_mkv_fill_buffer,
|
|
NULL,
|
|
demux_close_mkv,
|
|
demux_mkv_seek,
|
|
demux_mkv_control
|
|
};
|