1
0
mirror of https://github.com/mpv-player/mpv synced 2024-12-13 10:26:00 +00:00
mpv/libmpdemux/demux_real.c

403 lines
10 KiB
C
Raw Normal View History

/*
Real parser & demuxer
(C) Alex Beregszaszi <alex@naxine.org>
Based on FFmpeg's libav/rm.c.
*/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include "config.h"
#include "mp_msg.h"
#include "help_mp.h"
#include "stream.h"
#include "demuxer.h"
#include "stheader.h"
#include "bswap.h"
#define MKTAG(a, b, c, d) (a | (b << 8) | (c << 16) | (d << 24))
#define MAX_STREAMS 10
typedef struct {
int num_of_packets;
int last_a_stream;
int a_streams[MAX_STREAMS];
int last_v_stream;
int v_streams[MAX_STREAMS];
} real_priv_t;
static void get_str(int isbyte, demuxer_t *demuxer, char *buf, int buf_size)
{
int len, i;
char *q;
if (isbyte)
len = stream_read_char(demuxer->stream);
else
len = stream_read_word(demuxer->stream);
#if 1
q = buf;
for (i = 0; i < len; i++)
if (i < (buf_size - 1))
*q++ = stream_read_char(demuxer->stream);
*q = 0;
#else
stream_read(demuxer->stream, buf, buf_size);
#endif
}
static void skip_str(int isbyte, demuxer_t *demuxer)
{
int len;
if (isbyte)
len = stream_read_char(demuxer->stream);
else
len = stream_read_word(demuxer->stream);
stream_skip(demuxer->stream, len);
printf("skip_str: %d bytes skipped\n", len);
}
int real_check_file(demuxer_t* demuxer){
real_priv_t *priv;
int c;
mp_msg(MSGT_DEMUX,MSGL_V,"Checking for REAL\n");
c = stream_read_dword_le(demuxer->stream);
if (c == -256)
return 0; /* EOF */
if (c != MKTAG('.', 'R', 'M', 'F'))
return 0; /* bad magic */
priv = malloc(sizeof(real_priv_t));
memset(priv, 0, sizeof(real_priv_t));
demuxer->priv = priv;
return 1;
}
// return value:
// 0 = EOF or no stream found
// 1 = successfully read a packet
int demux_real_fill_buffer(demuxer_t *demuxer)
{
real_priv_t *priv = demuxer->priv;
demux_stream_t *ds = NULL;
int len;
int timestamp;
int stream_id;
int i;
// printf("num_of_packets: %d\n", priv->num_of_packets);
loop:
if (priv->num_of_packets == 0)
return 0; /* EOF */
stream_skip(demuxer->stream, 2);
len = stream_read_word(demuxer->stream);
if (len == -256) /* EOF */
return 0;
if (len < 12)
{
printf("bad packet len (%d)\n", len);
stream_skip(demuxer->stream, len);
goto loop;
// return 0; /* bad packet */
}
stream_id = stream_read_word(demuxer->stream);
timestamp = stream_read_dword(demuxer->stream);
// printf("packet: len: %d, stream_id: %d, timestamp: %d\n",
// len, stream_id, timestamp);
stream_skip(demuxer->stream, 1); /* reserved */
stream_skip(demuxer->stream, 1); /* flags */
priv->num_of_packets--;
len -= 12;
/* check if stream_id is audio stream */
for (i = 0; i < priv->last_a_stream; i++)
{
if (priv->a_streams[i] == stream_id)
{
// printf("packet is audio (id: %d)\n", stream_id);
ds = demuxer->audio; /* FIXME */
break;
}
}
/* check if stream_id is video stream */
for (i = 0; i < priv->last_v_stream; i++)
{
if (priv->v_streams[i] == stream_id)
{
// printf("packet is video (id: %d)\n", stream_id);
ds = demuxer->video; /* FIXME */
break;
}
}
/* id not found */
if (ds == NULL)
{
printf("unknown stream id (%d)\n", stream_id);
stream_skip(demuxer->stream, len);
goto loop;
}
demuxer->filepos = stream_tell(demuxer->stream);
ds_read_packet(ds, demuxer->stream, len, timestamp/90000.0f, demuxer->filepos, 0);
return 1;
}
void demux_open_real(demuxer_t* demuxer)
{
real_priv_t* priv = demuxer->priv;
int num_of_headers;
int i;
stream_skip(demuxer->stream, 4); /* header size */
stream_skip(demuxer->stream, 2);
stream_skip(demuxer->stream, 4);
num_of_headers = stream_read_dword(demuxer->stream);
// stream_skip(demuxer->stream, 4); /* number of headers */
/* parse chunks */
for (i = 1; i < num_of_headers; i++)
{
int chunk, chunk_size;
chunk = stream_read_dword_le(demuxer->stream);
chunk_size = stream_read_dword(demuxer->stream);
stream_skip(demuxer->stream, 2);
if (chunk_size < 10)
goto fail;
printf("Chunk: %.4s (size: %d)\n",
(char *)&chunk, chunk_size);
switch(chunk)
{
case MKTAG('P', 'R', 'O', 'P'):
printf("Properties chunk\n");
stream_skip(demuxer->stream, 4); /* max bitrate */
stream_skip(demuxer->stream, 4); /* avg bitrate */
stream_skip(demuxer->stream, 4); /* max packet size */
stream_skip(demuxer->stream, 4); /* avg packet size */
stream_skip(demuxer->stream, 4); /* nb packets */
stream_skip(demuxer->stream, 4); /* duration */
stream_skip(demuxer->stream, 4); /* preroll */
stream_skip(demuxer->stream, 4); /* index offset */
stream_skip(demuxer->stream, 4); /* data offset */
stream_skip(demuxer->stream, 2); /* nb streams */
stream_skip(demuxer->stream, 2); /* flags */
break;
case MKTAG('C', 'O', 'N', 'T'):
printf("Broadcasting informations (title, author, copyright, comment)\n");
skip_str(0, demuxer); /* title */
skip_str(0, demuxer); /* author */
skip_str(0, demuxer); /* copyright */
skip_str(0, demuxer); /* comment */
break;
case MKTAG('M', 'D', 'P', 'R'):
{
/* new stream */
int stream_id;
int bitrate;
int codec_data_size;
int codec_pos;
int v;
stream_id = stream_read_word(demuxer->stream);
printf("Found new stream (id: %d)\n", stream_id);
stream_skip(demuxer->stream, 4); /* max bitrate */
bitrate = stream_read_dword(demuxer->stream); /* bitrate */
stream_skip(demuxer->stream, 4); /* max packet size */
stream_skip(demuxer->stream, 4); /* avg packet size */
stream_skip(demuxer->stream, 4); /* start time */
stream_skip(demuxer->stream, 4); /* preroll */
stream_skip(demuxer->stream, 4); /* duration */
skip_str(1, demuxer); /* stream description */
skip_str(1, demuxer); /* mimetype */
codec_data_size = stream_read_dword(demuxer->stream);
codec_pos = stream_tell(demuxer->stream);
v = stream_read_dword(demuxer->stream);
if (v == MKTAG(0xfd, 'a', 'r', '.')) /* audio header */
{
sh_audio_t *sh = new_sh_audio(demuxer, stream_id);
printf("Found audio stream!\n");
stream_skip(demuxer->stream, 4); /* version */
stream_skip(demuxer->stream, 4); /* .ra4 */
stream_skip(demuxer->stream, 4);
stream_skip(demuxer->stream, 2);
stream_skip(demuxer->stream, 4); /* header size */
stream_skip(demuxer->stream, 2); /* add codec info */
stream_skip(demuxer->stream, 4); /* coded frame size */
stream_skip(demuxer->stream, 4);
stream_skip(demuxer->stream, 4);
stream_skip(demuxer->stream, 4);
stream_skip(demuxer->stream, 2); /* 1 */
stream_skip(demuxer->stream, 2); /* coded frame size */
stream_skip(demuxer->stream, 4);
sh->samplerate = stream_read_word(demuxer->stream);
stream_skip(demuxer->stream, 4);
sh->channels = stream_read_word(demuxer->stream);
{
char buf[128];
get_str(1, demuxer, buf, sizeof(buf));
get_str(1, demuxer, buf, sizeof(buf));
v = 0; /* not supported audio codec */
if (!strcmp(buf, "dnet"))
{
printf("Found AC3 audio\n");
sh->format = 0x2000; /* ac3 */
v = 1;
}
}
// skip_str(1, demuxer); /* desc */
// skip_str(1, demuxer); /* desc */
/* the audio codec name is stored in desc */
/* available codecs: DNET -> ac3 */
// sh->format = 0x2000; /* ac3 */
if (v)
{
printf("Audio codec: 0x%x\n", sh->format);
/* insert as stream */
demuxer->audio->sh = sh;
sh->ds = demuxer->audio;
demuxer->audio->id = stream_id;
if (priv->last_a_stream+1 < MAX_STREAMS)
{
priv->a_streams[priv->last_a_stream] = stream_id;
priv->last_a_stream++;
}
}
}
else
{
/* video header */
sh_video_t *sh = new_sh_video(demuxer, stream_id);
v = stream_read_dword_le(demuxer->stream);
printf("video: %.4s (%x)\n", (char *)&v, v);
if (v != MKTAG('V', 'I', 'D', 'O'))
{
mp_msg(MSGT_DEMUX, MSGL_ERR, "Unsupported video codec\n");
goto skip_this_chunk;
}
sh->format = stream_read_dword_le(demuxer->stream); /* fourcc */
printf("video fourcc: %.4s (%x)\n", (char *)&sh->format, sh->format);
// emulate BITMAPINFOHEADER:
sh->bih = malloc(sizeof(BITMAPINFOHEADER));
memset(sh->bih, 0, sizeof(BITMAPINFOHEADER));
sh->bih->biSize = 40;
sh->disp_w = sh->bih->biWidth = stream_read_word(demuxer->stream);
sh->disp_h = sh->bih->biHeight = stream_read_word(demuxer->stream);
sh->bih->biPlanes = 1;
sh->bih->biBitCount = 24;
sh->bih->biCompression = sh->format;
sh->bih->biSizeImage= sh->bih->biWidth*sh->bih->biHeight*3;
sh->fps = stream_read_word(demuxer->stream);
sh->frametime = 1.0f/sh->fps;
stream_skip(demuxer->stream, 4);
stream_skip(demuxer->stream, 2);
stream_skip(demuxer->stream, 4);
stream_skip(demuxer->stream, 2);
/* h263 hack */
v = stream_read_dword(demuxer->stream);
switch (v)
{
case 0x10000000:
/* sub id: 0 */
/* codec id: rv10 */
break;
case 0x10003000:
case 0x10003001:
/* sub id: 3 */
/* codec id: rv10 */
break;
case 0x20001000:
/* codec id: rv20 */
break;
default:
/* codec id: none */
}
/* insert as stream */
demuxer->video->sh = sh;
sh->ds = demuxer->video;
demuxer->video->id = stream_id;
if (priv->last_v_stream+1 < MAX_STREAMS)
{
priv->v_streams[priv->last_v_stream] = stream_id;
priv->last_v_stream++;
}
}
skip_this_chunk:
/* skip codec info */
v = stream_tell(demuxer->stream) - codec_pos;
stream_skip(demuxer->stream, codec_data_size - v);
break;
}
case MKTAG('D', 'A', 'T', 'A'):
goto header_end;
default:
printf("Unknown chunk: %x\n", chunk);
stream_skip(demuxer->stream, chunk_size - 10);
break;
}
}
header_end:
priv->num_of_packets = stream_read_dword(demuxer->stream);
// stream_skip(demuxer->stream, 4); /* number of packets */
stream_skip(demuxer->stream, 4); /* next data header */
printf("Packets in file: %d\n", priv->num_of_packets);
/* disable seeking */
demuxer->seekable = 0;
fail:
return;
}
void demux_close_real(demuxer_t *demuxer)
{
real_priv_t* priv = demuxer->priv;
if (priv)
free(priv);
return;
}