SVQ3: Move svq3 specific fields to their own context.
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
parent
175714c0c3
commit
8811679c8e
@ -594,13 +594,6 @@ typedef struct H264Context{
|
|||||||
// Timestamp stuff
|
// Timestamp stuff
|
||||||
int sei_buffering_period_present; ///< Buffering period SEI flag
|
int sei_buffering_period_present; ///< Buffering period SEI flag
|
||||||
int initial_cpb_removal_delay[32]; ///< Initial timestamps for CPBs
|
int initial_cpb_removal_delay[32]; ///< Initial timestamps for CPBs
|
||||||
|
|
||||||
//SVQ3 specific fields
|
|
||||||
int halfpel_flag;
|
|
||||||
int thirdpel_flag;
|
|
||||||
int unknown_svq3_flag;
|
|
||||||
int next_slice_index;
|
|
||||||
uint32_t svq3_watermark_key;
|
|
||||||
}H264Context;
|
}H264Context;
|
||||||
|
|
||||||
|
|
||||||
|
@ -63,6 +63,15 @@
|
|||||||
* svq3 decoder.
|
* svq3 decoder.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
H264Context h;
|
||||||
|
int halfpel_flag;
|
||||||
|
int thirdpel_flag;
|
||||||
|
int unknown_flag;
|
||||||
|
int next_slice_index;
|
||||||
|
uint32_t watermark_key;
|
||||||
|
} SVQ3Context;
|
||||||
|
|
||||||
#define FULLPEL_MODE 1
|
#define FULLPEL_MODE 1
|
||||||
#define HALFPEL_MODE 2
|
#define HALFPEL_MODE 2
|
||||||
#define THIRDPEL_MODE 3
|
#define THIRDPEL_MODE 3
|
||||||
@ -423,8 +432,9 @@ static inline int svq3_mc_dir(H264Context *h, int size, int mode, int dir,
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int svq3_decode_mb(H264Context *h, unsigned int mb_type)
|
static int svq3_decode_mb(SVQ3Context *svq3, unsigned int mb_type)
|
||||||
{
|
{
|
||||||
|
H264Context *h = &svq3->h;
|
||||||
int i, j, k, m, dir, mode;
|
int i, j, k, m, dir, mode;
|
||||||
int cbp = 0;
|
int cbp = 0;
|
||||||
uint32_t vlc;
|
uint32_t vlc;
|
||||||
@ -456,9 +466,9 @@ static int svq3_decode_mb(H264Context *h, unsigned int mb_type)
|
|||||||
mb_type = MB_TYPE_16x16;
|
mb_type = MB_TYPE_16x16;
|
||||||
}
|
}
|
||||||
} else if (mb_type < 8) { /* INTER */
|
} else if (mb_type < 8) { /* INTER */
|
||||||
if (h->thirdpel_flag && h->halfpel_flag == !get_bits1 (&s->gb)) {
|
if (svq3->thirdpel_flag && svq3->halfpel_flag == !get_bits1 (&s->gb)) {
|
||||||
mode = THIRDPEL_MODE;
|
mode = THIRDPEL_MODE;
|
||||||
} else if (h->halfpel_flag && h->thirdpel_flag == !get_bits1 (&s->gb)) {
|
} else if (svq3->halfpel_flag && svq3->thirdpel_flag == !get_bits1 (&s->gb)) {
|
||||||
mode = HALFPEL_MODE;
|
mode = HALFPEL_MODE;
|
||||||
} else {
|
} else {
|
||||||
mode = FULLPEL_MODE;
|
mode = FULLPEL_MODE;
|
||||||
@ -701,9 +711,11 @@ static int svq3_decode_mb(H264Context *h, unsigned int mb_type)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int svq3_decode_slice_header(H264Context *h)
|
static int svq3_decode_slice_header(AVCodecContext *avctx)
|
||||||
{
|
{
|
||||||
MpegEncContext *const s = (MpegEncContext *) h;
|
SVQ3Context *svq3 = avctx->priv_data;
|
||||||
|
H264Context *h = &svq3->h;
|
||||||
|
MpegEncContext *s = &h->s;
|
||||||
const int mb_xy = h->mb_xy;
|
const int mb_xy = h->mb_xy;
|
||||||
int i, header;
|
int i, header;
|
||||||
|
|
||||||
@ -711,24 +723,24 @@ static int svq3_decode_slice_header(H264Context *h)
|
|||||||
|
|
||||||
if (((header & 0x9F) != 1 && (header & 0x9F) != 2) || (header & 0x60) == 0) {
|
if (((header & 0x9F) != 1 && (header & 0x9F) != 2) || (header & 0x60) == 0) {
|
||||||
/* TODO: what? */
|
/* TODO: what? */
|
||||||
av_log(h->s.avctx, AV_LOG_ERROR, "unsupported slice header (%02X)\n", header);
|
av_log(avctx, AV_LOG_ERROR, "unsupported slice header (%02X)\n", header);
|
||||||
return -1;
|
return -1;
|
||||||
} else {
|
} else {
|
||||||
int length = (header >> 5) & 3;
|
int length = (header >> 5) & 3;
|
||||||
|
|
||||||
h->next_slice_index = get_bits_count(&s->gb) + 8*show_bits(&s->gb, 8*length) + 8*length;
|
svq3->next_slice_index = get_bits_count(&s->gb) + 8*show_bits(&s->gb, 8*length) + 8*length;
|
||||||
|
|
||||||
if (h->next_slice_index > s->gb.size_in_bits) {
|
if (svq3->next_slice_index > s->gb.size_in_bits) {
|
||||||
av_log(h->s.avctx, AV_LOG_ERROR, "slice after bitstream end\n");
|
av_log(avctx, AV_LOG_ERROR, "slice after bitstream end\n");
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
s->gb.size_in_bits = h->next_slice_index - 8*(length - 1);
|
s->gb.size_in_bits = svq3->next_slice_index - 8*(length - 1);
|
||||||
skip_bits(&s->gb, 8);
|
skip_bits(&s->gb, 8);
|
||||||
|
|
||||||
if (h->svq3_watermark_key) {
|
if (svq3->watermark_key) {
|
||||||
uint32_t header = AV_RL32(&s->gb.buffer[(get_bits_count(&s->gb)>>3)+1]);
|
uint32_t header = AV_RL32(&s->gb.buffer[(get_bits_count(&s->gb)>>3)+1]);
|
||||||
AV_WL32(&s->gb.buffer[(get_bits_count(&s->gb)>>3)+1], header ^ h->svq3_watermark_key);
|
AV_WL32(&s->gb.buffer[(get_bits_count(&s->gb)>>3)+1], header ^ svq3->watermark_key);
|
||||||
}
|
}
|
||||||
if (length > 0) {
|
if (length > 0) {
|
||||||
memcpy((uint8_t *) &s->gb.buffer[get_bits_count(&s->gb) >> 3],
|
memcpy((uint8_t *) &s->gb.buffer[get_bits_count(&s->gb) >> 3],
|
||||||
@ -759,7 +771,7 @@ static int svq3_decode_slice_header(H264Context *h)
|
|||||||
/* unknown fields */
|
/* unknown fields */
|
||||||
skip_bits1(&s->gb);
|
skip_bits1(&s->gb);
|
||||||
|
|
||||||
if (h->unknown_svq3_flag) {
|
if (svq3->unknown_flag) {
|
||||||
skip_bits1(&s->gb);
|
skip_bits1(&s->gb);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -788,8 +800,9 @@ static int svq3_decode_slice_header(H264Context *h)
|
|||||||
|
|
||||||
static av_cold int svq3_decode_init(AVCodecContext *avctx)
|
static av_cold int svq3_decode_init(AVCodecContext *avctx)
|
||||||
{
|
{
|
||||||
MpegEncContext *const s = avctx->priv_data;
|
SVQ3Context *svq3 = avctx->priv_data;
|
||||||
H264Context *const h = avctx->priv_data;
|
H264Context *h = &svq3->h;
|
||||||
|
MpegEncContext *s = &h->s;
|
||||||
int m;
|
int m;
|
||||||
unsigned char *extradata;
|
unsigned char *extradata;
|
||||||
unsigned int size;
|
unsigned int size;
|
||||||
@ -806,11 +819,12 @@ static av_cold int svq3_decode_init(AVCodecContext *avctx)
|
|||||||
if (!s->context_initialized) {
|
if (!s->context_initialized) {
|
||||||
s->width = avctx->width;
|
s->width = avctx->width;
|
||||||
s->height = avctx->height;
|
s->height = avctx->height;
|
||||||
h->halfpel_flag = 1;
|
|
||||||
h->thirdpel_flag = 1;
|
|
||||||
h->unknown_svq3_flag = 0;
|
|
||||||
h->chroma_qp[0] = h->chroma_qp[1] = 4;
|
h->chroma_qp[0] = h->chroma_qp[1] = 4;
|
||||||
|
|
||||||
|
svq3->halfpel_flag = 1;
|
||||||
|
svq3->thirdpel_flag = 1;
|
||||||
|
svq3->unknown_flag = 0;
|
||||||
|
|
||||||
if (MPV_common_init(s) < 0)
|
if (MPV_common_init(s) < 0)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
@ -851,8 +865,8 @@ static av_cold int svq3_decode_init(AVCodecContext *avctx)
|
|||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
h->halfpel_flag = get_bits1(&gb);
|
svq3->halfpel_flag = get_bits1(&gb);
|
||||||
h->thirdpel_flag = get_bits1(&gb);
|
svq3->thirdpel_flag = get_bits1(&gb);
|
||||||
|
|
||||||
/* unknown fields */
|
/* unknown fields */
|
||||||
skip_bits1(&gb);
|
skip_bits1(&gb);
|
||||||
@ -869,9 +883,9 @@ static av_cold int svq3_decode_init(AVCodecContext *avctx)
|
|||||||
skip_bits(&gb, 8);
|
skip_bits(&gb, 8);
|
||||||
}
|
}
|
||||||
|
|
||||||
h->unknown_svq3_flag = get_bits1(&gb);
|
svq3->unknown_flag = get_bits1(&gb);
|
||||||
avctx->has_b_frames = !s->low_delay;
|
avctx->has_b_frames = !s->low_delay;
|
||||||
if (h->unknown_svq3_flag) {
|
if (svq3->unknown_flag) {
|
||||||
#if CONFIG_ZLIB
|
#if CONFIG_ZLIB
|
||||||
unsigned watermark_width = svq3_get_ue_golomb(&gb);
|
unsigned watermark_width = svq3_get_ue_golomb(&gb);
|
||||||
unsigned watermark_height = svq3_get_ue_golomb(&gb);
|
unsigned watermark_height = svq3_get_ue_golomb(&gb);
|
||||||
@ -894,9 +908,9 @@ static av_cold int svq3_decode_init(AVCodecContext *avctx)
|
|||||||
av_free(buf);
|
av_free(buf);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
h->svq3_watermark_key = ff_svq1_packet_checksum(buf, buf_len, 0);
|
svq3->watermark_key = ff_svq1_packet_checksum(buf, buf_len, 0);
|
||||||
h->svq3_watermark_key = h->svq3_watermark_key << 16 | h->svq3_watermark_key;
|
svq3->watermark_key = svq3->watermark_key << 16 | svq3->watermark_key;
|
||||||
av_log(avctx, AV_LOG_DEBUG, "watermark key %#x\n", h->svq3_watermark_key);
|
av_log(avctx, AV_LOG_DEBUG, "watermark key %#x\n", svq3->watermark_key);
|
||||||
av_free(buf);
|
av_free(buf);
|
||||||
#else
|
#else
|
||||||
av_log(avctx, AV_LOG_ERROR, "this svq3 file contains watermark which need zlib support compiled in\n");
|
av_log(avctx, AV_LOG_ERROR, "this svq3 file contains watermark which need zlib support compiled in\n");
|
||||||
@ -914,9 +928,10 @@ static int svq3_decode_frame(AVCodecContext *avctx,
|
|||||||
AVPacket *avpkt)
|
AVPacket *avpkt)
|
||||||
{
|
{
|
||||||
const uint8_t *buf = avpkt->data;
|
const uint8_t *buf = avpkt->data;
|
||||||
|
SVQ3Context *svq3 = avctx->priv_data;
|
||||||
|
H264Context *h = &svq3->h;
|
||||||
|
MpegEncContext *s = &h->s;
|
||||||
int buf_size = avpkt->size;
|
int buf_size = avpkt->size;
|
||||||
MpegEncContext *const s = avctx->priv_data;
|
|
||||||
H264Context *const h = avctx->priv_data;
|
|
||||||
int m, mb_type;
|
int m, mb_type;
|
||||||
|
|
||||||
/* special case for last picture */
|
/* special case for last picture */
|
||||||
@ -933,7 +948,7 @@ static int svq3_decode_frame(AVCodecContext *avctx,
|
|||||||
|
|
||||||
s->mb_x = s->mb_y = h->mb_xy = 0;
|
s->mb_x = s->mb_y = h->mb_xy = 0;
|
||||||
|
|
||||||
if (svq3_decode_slice_header(h))
|
if (svq3_decode_slice_header(avctx))
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
s->pict_type = h->slice_type;
|
s->pict_type = h->slice_type;
|
||||||
@ -941,7 +956,7 @@ static int svq3_decode_frame(AVCodecContext *avctx,
|
|||||||
|
|
||||||
if (avctx->debug&FF_DEBUG_PICT_INFO){
|
if (avctx->debug&FF_DEBUG_PICT_INFO){
|
||||||
av_log(h->s.avctx, AV_LOG_DEBUG, "%c hpel:%d, tpel:%d aqp:%d qp:%d, slice_num:%02X\n",
|
av_log(h->s.avctx, AV_LOG_DEBUG, "%c hpel:%d, tpel:%d aqp:%d qp:%d, slice_num:%02X\n",
|
||||||
av_get_picture_type_char(s->pict_type), h->halfpel_flag, h->thirdpel_flag,
|
av_get_picture_type_char(s->pict_type), svq3->halfpel_flag, svq3->thirdpel_flag,
|
||||||
s->adaptive_quant, s->qscale, h->slice_num);
|
s->adaptive_quant, s->qscale, h->slice_num);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1005,10 +1020,10 @@ static int svq3_decode_frame(AVCodecContext *avctx,
|
|||||||
if ( (get_bits_count(&s->gb) + 7) >= s->gb.size_in_bits &&
|
if ( (get_bits_count(&s->gb) + 7) >= s->gb.size_in_bits &&
|
||||||
((get_bits_count(&s->gb) & 7) == 0 || show_bits(&s->gb, (-get_bits_count(&s->gb) & 7)) == 0)) {
|
((get_bits_count(&s->gb) & 7) == 0 || show_bits(&s->gb, (-get_bits_count(&s->gb) & 7)) == 0)) {
|
||||||
|
|
||||||
skip_bits(&s->gb, h->next_slice_index - get_bits_count(&s->gb));
|
skip_bits(&s->gb, svq3->next_slice_index - get_bits_count(&s->gb));
|
||||||
s->gb.size_in_bits = 8*buf_size;
|
s->gb.size_in_bits = 8*buf_size;
|
||||||
|
|
||||||
if (svq3_decode_slice_header(h))
|
if (svq3_decode_slice_header(avctx))
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
/* TODO: support s->mb_skip_run */
|
/* TODO: support s->mb_skip_run */
|
||||||
@ -1021,7 +1036,7 @@ static int svq3_decode_frame(AVCodecContext *avctx,
|
|||||||
} else if (s->pict_type == AV_PICTURE_TYPE_B && mb_type >= 4) {
|
} else if (s->pict_type == AV_PICTURE_TYPE_B && mb_type >= 4) {
|
||||||
mb_type += 4;
|
mb_type += 4;
|
||||||
}
|
}
|
||||||
if (mb_type > 33 || svq3_decode_mb(h, mb_type)) {
|
if (mb_type > 33 || svq3_decode_mb(svq3, mb_type)) {
|
||||||
av_log(h->s.avctx, AV_LOG_ERROR, "error while decoding MB %d %d\n", s->mb_x, s->mb_y);
|
av_log(h->s.avctx, AV_LOG_ERROR, "error while decoding MB %d %d\n", s->mb_x, s->mb_y);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
@ -1055,15 +1070,27 @@ static int svq3_decode_frame(AVCodecContext *avctx,
|
|||||||
return buf_size;
|
return buf_size;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static int svq3_decode_end(AVCodecContext *avctx)
|
||||||
|
{
|
||||||
|
SVQ3Context *svq3 = avctx->priv_data;
|
||||||
|
H264Context *h = &svq3->h;
|
||||||
|
MpegEncContext *s = &h->s;
|
||||||
|
|
||||||
|
ff_h264_free_context(h);
|
||||||
|
|
||||||
|
MPV_common_end(s);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
AVCodec ff_svq3_decoder = {
|
AVCodec ff_svq3_decoder = {
|
||||||
"svq3",
|
"svq3",
|
||||||
AVMEDIA_TYPE_VIDEO,
|
AVMEDIA_TYPE_VIDEO,
|
||||||
CODEC_ID_SVQ3,
|
CODEC_ID_SVQ3,
|
||||||
sizeof(H264Context),
|
sizeof(SVQ3Context),
|
||||||
svq3_decode_init,
|
svq3_decode_init,
|
||||||
NULL,
|
NULL,
|
||||||
ff_h264_decode_end,
|
svq3_decode_end,
|
||||||
svq3_decode_frame,
|
svq3_decode_frame,
|
||||||
CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 | CODEC_CAP_DELAY,
|
CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 | CODEC_CAP_DELAY,
|
||||||
.long_name = NULL_IF_CONFIG_SMALL("Sorenson Vector Quantizer 3 / Sorenson Video 3 / SVQ3"),
|
.long_name = NULL_IF_CONFIG_SMALL("Sorenson Vector Quantizer 3 / Sorenson Video 3 / SVQ3"),
|
||||||
|
Loading…
x
Reference in New Issue
Block a user