2005-05-11 16:38:34 +00:00
|
|
|
/**
|
2006-02-22 00:18:12 +00:00
|
|
|
Copyright (C) 2005 Matthieu CASTET, Alex Beregszaszi
|
2005-05-11 16:38:34 +00:00
|
|
|
|
|
|
|
Permission is hereby granted, free of charge, to any person
|
|
|
|
obtaining a copy of this software and associated documentation
|
|
|
|
files (the "Software"), to deal in the Software without
|
|
|
|
restriction, including without limitation the rights to use, copy,
|
|
|
|
modify, merge, publish, distribute, sublicense, and/or sell copies
|
|
|
|
of the Software, and to permit persons to whom the Software is
|
|
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
|
|
|
|
The above copyright notice and this permission notice shall be
|
|
|
|
included in all copies or substantial portions of the Software.
|
|
|
|
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
|
|
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
|
|
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
|
|
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
|
|
|
HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
|
|
WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
|
|
DEALINGS IN THE SOFTWARE.
|
|
|
|
**/
|
|
|
|
|
|
|
|
#include <stdlib.h>
|
2008-05-09 11:56:36 +00:00
|
|
|
#include "libavutil/bswap.h"
|
2009-04-13 16:20:26 +00:00
|
|
|
#include "libavcodec/get_bits.h"
|
2005-05-11 16:38:34 +00:00
|
|
|
#include "avformat.h"
|
2011-11-29 18:28:15 +00:00
|
|
|
#include "internal.h"
|
2007-11-07 20:22:32 +00:00
|
|
|
#include "oggdec.h"
|
2005-05-11 16:38:34 +00:00
|
|
|
|
2008-11-06 01:50:56 +00:00
|
|
|
struct theora_params {
|
2005-05-11 16:38:34 +00:00
|
|
|
int gpshift;
|
|
|
|
int gpmask;
|
2008-11-06 01:57:17 +00:00
|
|
|
unsigned version;
|
2008-11-06 01:50:56 +00:00
|
|
|
};
|
2005-05-11 16:38:34 +00:00
|
|
|
|
|
|
|
static int
|
|
|
|
theora_header (AVFormatContext * s, int idx)
|
|
|
|
{
|
2008-11-06 01:50:56 +00:00
|
|
|
struct ogg *ogg = s->priv_data;
|
|
|
|
struct ogg_stream *os = ogg->streams + idx;
|
2005-05-11 16:38:34 +00:00
|
|
|
AVStream *st = s->streams[idx];
|
2008-11-06 01:50:56 +00:00
|
|
|
struct theora_params *thp = os->private;
|
2005-07-17 22:24:36 +00:00
|
|
|
int cds = st->codec->extradata_size + os->psize + 2;
|
2005-05-11 16:38:34 +00:00
|
|
|
uint8_t *cdp;
|
|
|
|
|
2005-05-12 22:33:05 +00:00
|
|
|
if(!(os->buf[os->pstart] & 0x80))
|
2005-05-11 16:38:34 +00:00
|
|
|
return 0;
|
|
|
|
|
|
|
|
if(!thp){
|
2005-12-22 01:10:11 +00:00
|
|
|
thp = av_mallocz(sizeof(*thp));
|
|
|
|
os->private = thp;
|
2005-05-11 16:38:34 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (os->buf[os->pstart] == 0x80) {
|
|
|
|
GetBitContext gb;
|
2007-05-07 15:43:01 +00:00
|
|
|
int width, height;
|
2011-11-24 19:58:22 +00:00
|
|
|
AVRational timebase;
|
2006-02-22 00:18:12 +00:00
|
|
|
|
2005-05-11 16:38:34 +00:00
|
|
|
init_get_bits(&gb, os->buf + os->pstart, os->psize*8);
|
|
|
|
|
2009-05-14 00:02:07 +00:00
|
|
|
skip_bits_long(&gb, 7*8); /* 0x80"theora" */
|
2006-02-22 00:18:12 +00:00
|
|
|
|
2008-11-06 01:57:17 +00:00
|
|
|
thp->version = get_bits_long(&gb, 24);
|
|
|
|
if (thp->version < 0x030100)
|
2006-02-22 00:18:12 +00:00
|
|
|
{
|
|
|
|
av_log(s, AV_LOG_ERROR,
|
2008-11-06 01:57:17 +00:00
|
|
|
"Too old or unsupported Theora (%x)\n", thp->version);
|
2005-05-12 22:33:05 +00:00
|
|
|
return -1;
|
2006-02-22 00:18:12 +00:00
|
|
|
}
|
2005-05-12 22:33:05 +00:00
|
|
|
|
2007-05-07 15:43:01 +00:00
|
|
|
width = get_bits(&gb, 16) << 4;
|
|
|
|
height = get_bits(&gb, 16) << 4;
|
|
|
|
avcodec_set_dimensions(st->codec, width, height);
|
2005-05-11 16:38:34 +00:00
|
|
|
|
2008-11-06 01:57:17 +00:00
|
|
|
if (thp->version >= 0x030400)
|
2007-05-07 15:43:01 +00:00
|
|
|
skip_bits(&gb, 100);
|
|
|
|
|
2008-11-06 01:57:17 +00:00
|
|
|
if (thp->version >= 0x030200) {
|
2008-10-01 14:44:30 +00:00
|
|
|
width = get_bits_long(&gb, 24);
|
|
|
|
height = get_bits_long(&gb, 24);
|
|
|
|
if ( width <= st->codec->width && width > st->codec->width-16
|
|
|
|
&& height <= st->codec->height && height > st->codec->height-16)
|
|
|
|
avcodec_set_dimensions(st->codec, width, height);
|
2007-05-07 15:43:01 +00:00
|
|
|
|
|
|
|
skip_bits(&gb, 16);
|
2008-10-01 14:43:08 +00:00
|
|
|
}
|
2011-11-24 19:58:22 +00:00
|
|
|
timebase.den = get_bits_long(&gb, 32);
|
|
|
|
timebase.num = get_bits_long(&gb, 32);
|
|
|
|
if (!(timebase.num > 0 && timebase.den > 0)) {
|
2009-07-03 11:31:09 +00:00
|
|
|
av_log(s, AV_LOG_WARNING, "Invalid time base in theora stream, assuming 25 FPS\n");
|
2011-11-24 19:58:22 +00:00
|
|
|
timebase.num = 1;
|
|
|
|
timebase.den = 25;
|
2009-07-03 11:31:09 +00:00
|
|
|
}
|
2011-11-24 19:58:22 +00:00
|
|
|
avpriv_set_pts_info(st, 64, timebase.num, timebase.den);
|
2005-12-17 18:14:38 +00:00
|
|
|
|
2008-08-23 23:43:20 +00:00
|
|
|
st->sample_aspect_ratio.num = get_bits_long(&gb, 24);
|
|
|
|
st->sample_aspect_ratio.den = get_bits_long(&gb, 24);
|
2005-05-11 16:38:34 +00:00
|
|
|
|
2008-11-06 01:57:17 +00:00
|
|
|
if (thp->version >= 0x030200)
|
2009-05-14 00:02:07 +00:00
|
|
|
skip_bits_long(&gb, 38);
|
2008-11-06 01:57:17 +00:00
|
|
|
if (thp->version >= 0x304000)
|
2006-02-22 00:18:12 +00:00
|
|
|
skip_bits(&gb, 2);
|
|
|
|
|
2005-05-11 16:38:34 +00:00
|
|
|
thp->gpshift = get_bits(&gb, 5);
|
2005-12-22 01:10:11 +00:00
|
|
|
thp->gpmask = (1 << thp->gpshift) - 1;
|
2005-05-11 16:38:34 +00:00
|
|
|
|
2010-03-30 23:30:55 +00:00
|
|
|
st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
|
2012-08-05 09:11:04 +00:00
|
|
|
st->codec->codec_id = AV_CODEC_ID_THEORA;
|
2010-03-05 00:22:40 +00:00
|
|
|
st->need_parsing = AVSTREAM_PARSE_HEADERS;
|
2005-05-11 16:38:34 +00:00
|
|
|
|
|
|
|
} else if (os->buf[os->pstart] == 0x83) {
|
2010-03-12 05:16:44 +00:00
|
|
|
ff_vorbis_comment (s, &st->metadata, os->buf + os->pstart + 7, os->psize - 8);
|
2005-05-11 16:38:34 +00:00
|
|
|
}
|
|
|
|
|
2009-06-05 23:39:11 +00:00
|
|
|
st->codec->extradata = av_realloc (st->codec->extradata,
|
|
|
|
cds + FF_INPUT_BUFFER_PADDING_SIZE);
|
2005-07-17 22:24:36 +00:00
|
|
|
cdp = st->codec->extradata + st->codec->extradata_size;
|
2005-05-11 16:38:34 +00:00
|
|
|
*cdp++ = os->psize >> 8;
|
|
|
|
*cdp++ = os->psize & 0xff;
|
|
|
|
memcpy (cdp, os->buf + os->pstart, os->psize);
|
2005-07-17 22:24:36 +00:00
|
|
|
st->codec->extradata_size = cds;
|
2005-05-11 16:38:34 +00:00
|
|
|
|
2005-05-12 22:33:05 +00:00
|
|
|
return 1;
|
2005-05-11 16:38:34 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static uint64_t
|
2010-01-11 05:51:09 +00:00
|
|
|
theora_gptopts(AVFormatContext *ctx, int idx, uint64_t gp, int64_t *dts)
|
2005-05-11 16:38:34 +00:00
|
|
|
{
|
2008-11-06 01:50:56 +00:00
|
|
|
struct ogg *ogg = ctx->priv_data;
|
|
|
|
struct ogg_stream *os = ogg->streams + idx;
|
|
|
|
struct theora_params *thp = os->private;
|
2012-03-23 02:43:30 +00:00
|
|
|
uint64_t iframe, pframe;
|
|
|
|
|
|
|
|
if (!thp)
|
|
|
|
return AV_NOPTS_VALUE;
|
|
|
|
|
|
|
|
iframe = gp >> thp->gpshift;
|
|
|
|
pframe = gp & thp->gpmask;
|
2005-05-11 16:38:34 +00:00
|
|
|
|
2008-11-06 01:57:17 +00:00
|
|
|
if (thp->version < 0x030201)
|
|
|
|
iframe++;
|
|
|
|
|
2007-11-11 21:56:18 +00:00
|
|
|
if(!pframe)
|
2010-03-31 12:29:58 +00:00
|
|
|
os->pflags |= AV_PKT_FLAG_KEY;
|
2007-11-11 21:56:18 +00:00
|
|
|
|
2010-01-11 05:51:09 +00:00
|
|
|
if (dts)
|
|
|
|
*dts = iframe + pframe;
|
|
|
|
|
2006-05-12 00:50:43 +00:00
|
|
|
return iframe + pframe;
|
2005-05-11 16:38:34 +00:00
|
|
|
}
|
|
|
|
|
2012-05-04 12:36:17 +00:00
|
|
|
static int theora_packet(AVFormatContext *s, int idx)
|
|
|
|
{
|
|
|
|
struct ogg *ogg = s->priv_data;
|
|
|
|
struct ogg_stream *os = ogg->streams + idx;
|
|
|
|
int duration;
|
|
|
|
|
|
|
|
/* first packet handling
|
|
|
|
here we parse the duration of each packet in the first page and compare
|
|
|
|
the total duration to the page granule to find the encoder delay and
|
|
|
|
set the first timestamp */
|
|
|
|
|
2012-05-04 20:54:35 +00:00
|
|
|
if ((!os->lastpts || os->lastpts == AV_NOPTS_VALUE) && !(os->flags & OGG_FLAG_EOS)) {
|
2012-05-04 12:36:17 +00:00
|
|
|
int seg;
|
2012-05-04 20:54:35 +00:00
|
|
|
|
|
|
|
duration = 1;
|
|
|
|
for (seg = os->segp; seg < os->nsegs; seg++) {
|
|
|
|
if (os->segments[seg] < 255)
|
|
|
|
duration ++;
|
2012-05-04 12:36:17 +00:00
|
|
|
}
|
2012-05-04 20:54:35 +00:00
|
|
|
|
2012-05-04 12:36:17 +00:00
|
|
|
os->lastpts = os->lastdts = theora_gptopts(s, idx, os->granule, NULL) - duration;
|
2012-05-04 20:54:35 +00:00
|
|
|
if(s->streams[idx]->start_time == AV_NOPTS_VALUE) {
|
|
|
|
s->streams[idx]->start_time = os->lastpts;
|
|
|
|
if (s->streams[idx]->duration)
|
|
|
|
s->streams[idx]->duration -= s->streams[idx]->start_time;
|
|
|
|
}
|
2012-05-04 12:36:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* parse packet duration */
|
|
|
|
if (os->psize > 0) {
|
|
|
|
os->pduration = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2008-11-06 01:50:56 +00:00
|
|
|
const struct ogg_codec ff_theora_codec = {
|
2005-05-11 16:38:34 +00:00
|
|
|
.magic = "\200theora",
|
|
|
|
.magicsize = 7,
|
|
|
|
.header = theora_header,
|
2012-05-04 12:36:17 +00:00
|
|
|
.packet = theora_packet,
|
2005-05-11 16:38:34 +00:00
|
|
|
.gptopts = theora_gptopts
|
|
|
|
};
|