2 * Electronic Arts TQI Video Decoder
3 * Copyright (c) 2007-2009 Peter Ross <pross@xvid.org>
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
24 * Electronic Arts TQI Video Decoder
25 * by Peter Ross <pross@xvid.org>
27 * Technical details here:
28 * http://wiki.multimedia.cx/index.php?title=Electronic_Arts_TQI
32 #include "bitstream.h"
34 #include "aandcttab.h"
36 #include "mpegvideo.h"
38 typedef struct TqiContext
{
41 uint8_t *bitstream_buf
;
42 unsigned int bitstream_buf_size
;
45 static av_cold
int tqi_decode_init(AVCodecContext
*avctx
)
47 TqiContext
*t
= avctx
->priv_data
;
48 MpegEncContext
*s
= &t
->s
;
50 if(avctx
->idct_algo
==FF_IDCT_AUTO
)
51 avctx
->idct_algo
=FF_IDCT_EA
;
52 dsputil_init(&s
->dsp
, avctx
);
53 ff_init_scantable(s
->dsp
.idct_permutation
, &s
->intra_scantable
, ff_zigzag_direct
);
55 avctx
->time_base
= (AVRational
){1, 15};
56 avctx
->pix_fmt
= PIX_FMT_YUV420P
;
57 ff_mpeg12_init_vlcs();
61 static void tqi_decode_mb(MpegEncContext
*s
, DCTELEM (*block
)[64])
64 s
->dsp
.clear_blocks(block
[0]);
66 ff_mpeg1_decode_block_intra(s
, block
[n
], n
);
69 static inline void tqi_idct_put(TqiContext
*t
, DCTELEM (*block
)[64])
71 MpegEncContext
*s
= &t
->s
;
72 int linesize
= t
->frame
.linesize
[0];
73 uint8_t *dest_y
= t
->frame
.data
[0] + (s
->mb_y
* 16* linesize
) + s
->mb_x
* 16;
74 uint8_t *dest_cb
= t
->frame
.data
[1] + (s
->mb_y
* 8 * t
->frame
.linesize
[1]) + s
->mb_x
* 8;
75 uint8_t *dest_cr
= t
->frame
.data
[2] + (s
->mb_y
* 8 * t
->frame
.linesize
[2]) + s
->mb_x
* 8;
77 s
->dsp
.idct_put(dest_y
, linesize
, block
[0]);
78 s
->dsp
.idct_put(dest_y
+ 8, linesize
, block
[1]);
79 s
->dsp
.idct_put(dest_y
+ 8*linesize
, linesize
, block
[2]);
80 s
->dsp
.idct_put(dest_y
+ 8*linesize
+ 8, linesize
, block
[3]);
81 if(!(s
->avctx
->flags
&CODEC_FLAG_GRAY
)) {
82 s
->dsp
.idct_put(dest_cb
, t
->frame
.linesize
[1], block
[4]);
83 s
->dsp
.idct_put(dest_cr
, t
->frame
.linesize
[2], block
[5]);
87 static void tqi_calculate_qtable(MpegEncContext
*s
, int quant
)
89 const int qscale
= (215 - 2*quant
)*5;
91 if (s
->avctx
->idct_algo
==FF_IDCT_EA
) {
92 s
->intra_matrix
[0] = (ff_inv_aanscales
[0]*ff_mpeg1_default_intra_matrix
[0])>>11;
94 s
->intra_matrix
[i
] = (ff_inv_aanscales
[i
]*ff_mpeg1_default_intra_matrix
[i
]*qscale
+ 32)>>14;
96 s
->intra_matrix
[0] = ff_mpeg1_default_intra_matrix
[0];
98 s
->intra_matrix
[i
] = (ff_mpeg1_default_intra_matrix
[i
]*qscale
+ 32)>>3;
102 static int tqi_decode_frame(AVCodecContext
*avctx
,
103 void *data
, int *data_size
,
104 const uint8_t *buf
, int buf_size
)
106 const uint8_t *buf_end
= buf
+buf_size
;
107 TqiContext
*t
= avctx
->priv_data
;
108 MpegEncContext
*s
= &t
->s
;
109 DECLARE_ALIGNED_16(DCTELEM
, block
[6][64]);
111 s
->width
= AV_RL16(&buf
[0]);
112 s
->height
= AV_RL16(&buf
[2]);
113 tqi_calculate_qtable(s
, buf
[4]);
116 if (t
->frame
.data
[0])
117 avctx
->release_buffer(avctx
, &t
->frame
);
119 if (s
->avctx
->width
!=s
->width
|| s
->avctx
->height
!=s
->height
)
120 avcodec_set_dimensions(s
->avctx
, s
->width
, s
->height
);
122 if(avctx
->get_buffer(avctx
, &t
->frame
) < 0) {
123 av_log(avctx
, AV_LOG_ERROR
, "get_buffer() failed\n");
127 t
->bitstream_buf
= av_fast_realloc(t
->bitstream_buf
, &t
->bitstream_buf_size
, (buf_end
-buf
) + FF_INPUT_BUFFER_PADDING_SIZE
);
128 if (!t
->bitstream_buf
)
130 s
->dsp
.bswap_buf((uint32_t*)t
->bitstream_buf
, (const uint32_t*)buf
, (buf_end
-buf
)/4);
131 init_get_bits(&s
->gb
, t
->bitstream_buf
, 8*(buf_end
-buf
));
133 s
->last_dc
[0] = s
->last_dc
[1] = s
->last_dc
[2] = 0;
134 for (s
->mb_y
=0; s
->mb_y
<(avctx
->height
+15)/16; s
->mb_y
++)
135 for (s
->mb_x
=0; s
->mb_x
<(avctx
->width
+15)/16; s
->mb_x
++)
137 tqi_decode_mb(s
, block
);
138 tqi_idct_put(t
, block
);
141 *data_size
= sizeof(AVFrame
);
142 *(AVFrame
*)data
= t
->frame
;
146 static av_cold
int tqi_decode_end(AVCodecContext
*avctx
)
148 TqiContext
*t
= avctx
->priv_data
;
150 avctx
->release_buffer(avctx
, &t
->frame
);
151 av_free(t
->bitstream_buf
);
155 AVCodec eatqi_decoder
= {
165 .long_name
= NULL_IF_CONFIG_SMALL("Electronic Arts TQI Video"),