Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * Electronic Arts TQI Video Decoder | ||
3 | * Copyright (c) 2007-2009 Peter Ross <pross@xvid.org> | ||
4 | * | ||
5 | * This file is part of FFmpeg. | ||
6 | * | ||
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. | ||
11 | * | ||
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. | ||
16 | * | ||
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 | ||
20 | */ | ||
21 | |||
22 | /** | ||
23 | * @file | ||
24 | * Electronic Arts TQI Video Decoder | ||
25 | * @author Peter Ross <pross@xvid.org> | ||
26 | * @see http://wiki.multimedia.cx/index.php?title=Electronic_Arts_TQI | ||
27 | */ | ||
28 | |||
29 | #include "libavutil/mem_internal.h" | ||
30 | |||
31 | #include "avcodec.h" | ||
32 | #include "blockdsp.h" | ||
33 | #include "bswapdsp.h" | ||
34 | #include "codec_internal.h" | ||
35 | #include "decode.h" | ||
36 | #include "get_bits.h" | ||
37 | #include "aandcttab.h" | ||
38 | #include "eaidct.h" | ||
39 | #include "mpeg12data.h" | ||
40 | #include "mpeg12dec.h" | ||
41 | |||
42 | typedef struct TqiContext { | ||
43 | AVCodecContext *avctx; | ||
44 | GetBitContext gb; | ||
45 | BlockDSPContext bdsp; | ||
46 | BswapDSPContext bsdsp; | ||
47 | |||
48 | void *bitstream_buf; | ||
49 | unsigned int bitstream_buf_size; | ||
50 | |||
51 | int mb_x, mb_y; | ||
52 | uint16_t intra_matrix[64]; | ||
53 | int last_dc[3]; | ||
54 | |||
55 | DECLARE_ALIGNED(32, int16_t, block)[6][64]; | ||
56 | } TqiContext; | ||
57 | |||
58 | 3 | static av_cold int tqi_decode_init(AVCodecContext *avctx) | |
59 | { | ||
60 | 3 | TqiContext *t = avctx->priv_data; | |
61 | |||
62 | 3 | ff_blockdsp_init(&t->bdsp); | |
63 | 3 | ff_bswapdsp_init(&t->bsdsp); | |
64 | |||
65 | 3 | avctx->framerate = (AVRational){ 15, 1 }; | |
66 | 3 | avctx->pix_fmt = AV_PIX_FMT_YUV420P; | |
67 | 3 | ff_mpeg12_init_vlcs(); | |
68 | 3 | return 0; | |
69 | } | ||
70 | |||
71 | 8400 | static int tqi_decode_mb(TqiContext *t, int16_t (*block)[64]) | |
72 | { | ||
73 | int n; | ||
74 | |||
75 | 8400 | t->bdsp.clear_blocks(block[0]); | |
76 |
2/2✓ Branch 0 taken 50400 times.
✓ Branch 1 taken 8400 times.
|
58800 | for (n = 0; n < 6; n++) { |
77 | 50400 | int ret = ff_mpeg1_decode_block_intra(&t->gb, | |
78 | 50400 | t->intra_matrix, | |
79 | ff_zigzag_direct, | ||
80 | 50400 | t->last_dc, block[n], n, 1); | |
81 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 50400 times.
|
50400 | if (ret < 0) { |
82 | ✗ | av_log(t->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", | |
83 | t->mb_x, t->mb_y); | ||
84 | ✗ | return ret; | |
85 | } | ||
86 | } | ||
87 | |||
88 | 8400 | return 0; | |
89 | } | ||
90 | |||
91 | 8400 | static inline void tqi_idct_put(AVCodecContext *avctx, AVFrame *frame, | |
92 | int16_t (*block)[64]) | ||
93 | { | ||
94 | 8400 | TqiContext *t = avctx->priv_data; | |
95 | 8400 | ptrdiff_t linesize = frame->linesize[0]; | |
96 | 8400 | uint8_t *dest_y = frame->data[0] + t->mb_y * 16 * linesize + t->mb_x * 16; | |
97 | 8400 | uint8_t *dest_cb = frame->data[1] + t->mb_y * 8 * frame->linesize[1] + t->mb_x * 8; | |
98 | 8400 | uint8_t *dest_cr = frame->data[2] + t->mb_y * 8 * frame->linesize[2] + t->mb_x * 8; | |
99 | |||
100 | 8400 | ff_ea_idct_put_c(dest_y , linesize, block[0]); | |
101 | 8400 | ff_ea_idct_put_c(dest_y + 8, linesize, block[1]); | |
102 | 8400 | ff_ea_idct_put_c(dest_y + 8*linesize , linesize, block[2]); | |
103 | 8400 | ff_ea_idct_put_c(dest_y + 8*linesize + 8, linesize, block[3]); | |
104 | |||
105 |
1/2✓ Branch 0 taken 8400 times.
✗ Branch 1 not taken.
|
8400 | if (!(avctx->flags & AV_CODEC_FLAG_GRAY)) { |
106 | 8400 | ff_ea_idct_put_c(dest_cb, frame->linesize[1], block[4]); | |
107 | 8400 | ff_ea_idct_put_c(dest_cr, frame->linesize[2], block[5]); | |
108 | } | ||
109 | 8400 | } | |
110 | |||
111 | 28 | static void tqi_calculate_qtable(TqiContext *t, int quant) | |
112 | { | ||
113 | 28 | const int64_t qscale = (215 - 2*quant)*5; | |
114 | int i; | ||
115 | |||
116 | 28 | t->intra_matrix[0] = (ff_inv_aanscales[0] * ff_mpeg1_default_intra_matrix[0]) >> 11; | |
117 |
2/2✓ Branch 0 taken 1764 times.
✓ Branch 1 taken 28 times.
|
1792 | for(i=1; i<64; i++) |
118 | 1764 | t->intra_matrix[i] = (ff_inv_aanscales[i] * ff_mpeg1_default_intra_matrix[i] * qscale + 32) >> 14; | |
119 | 28 | } | |
120 | |||
121 | 28 | static int tqi_decode_frame(AVCodecContext *avctx, AVFrame *frame, | |
122 | int *got_frame, AVPacket *avpkt) | ||
123 | { | ||
124 | 28 | const uint8_t *buf = avpkt->data; | |
125 | 28 | int buf_size = avpkt->size; | |
126 | 28 | const uint8_t *buf_end = buf+buf_size; | |
127 | 28 | TqiContext *t = avctx->priv_data; | |
128 | int ret, w, h; | ||
129 | |||
130 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
|
28 | if (buf_size < 12) |
131 | ✗ | return AVERROR_INVALIDDATA; | |
132 | |||
133 | 28 | t->avctx = avctx; | |
134 | |||
135 | 28 | w = AV_RL16(&buf[0]); | |
136 | 28 | h = AV_RL16(&buf[2]); | |
137 | 28 | tqi_calculate_qtable(t, buf[4]); | |
138 | 28 | buf += 8; | |
139 | |||
140 | 28 | ret = ff_set_dimensions(avctx, w, h); | |
141 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
|
28 | if (ret < 0) |
142 | ✗ | return ret; | |
143 | |||
144 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 28 times.
|
28 | if ((ret = ff_get_buffer(avctx, frame, 0)) < 0) |
145 | ✗ | return ret; | |
146 | |||
147 | 28 | av_fast_padded_malloc(&t->bitstream_buf, &t->bitstream_buf_size, | |
148 | 28 | buf_end - buf); | |
149 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
|
28 | if (!t->bitstream_buf) |
150 | ✗ | return AVERROR(ENOMEM); | |
151 | 28 | t->bsdsp.bswap_buf(t->bitstream_buf, (const uint32_t *) buf, | |
152 | 28 | (buf_end - buf) / 4); | |
153 | 28 | init_get_bits(&t->gb, t->bitstream_buf, 8 * (buf_end - buf)); | |
154 | |||
155 | 28 | t->last_dc[0] = | |
156 | 28 | t->last_dc[1] = | |
157 | 28 | t->last_dc[2] = 0; | |
158 |
2/2✓ Branch 0 taken 336 times.
✓ Branch 1 taken 28 times.
|
364 | for (t->mb_y = 0; t->mb_y < (h + 15) / 16; t->mb_y++) { |
159 |
2/2✓ Branch 0 taken 8400 times.
✓ Branch 1 taken 336 times.
|
8736 | for (t->mb_x = 0; t->mb_x < (w + 15) / 16; t->mb_x++) { |
160 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 8400 times.
|
8400 | if (tqi_decode_mb(t, t->block) < 0) |
161 | ✗ | goto end; | |
162 | 8400 | tqi_idct_put(avctx, frame, t->block); | |
163 | } | ||
164 | } | ||
165 | 28 | end: | |
166 | |||
167 | 28 | *got_frame = 1; | |
168 | 28 | return buf_size; | |
169 | } | ||
170 | |||
171 | 3 | static av_cold int tqi_decode_end(AVCodecContext *avctx) | |
172 | { | ||
173 | 3 | TqiContext *t = avctx->priv_data; | |
174 | 3 | av_freep(&t->bitstream_buf); | |
175 | 3 | return 0; | |
176 | } | ||
177 | |||
178 | const FFCodec ff_eatqi_decoder = { | ||
179 | .p.name = "eatqi", | ||
180 | CODEC_LONG_NAME("Electronic Arts TQI Video"), | ||
181 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
182 | .p.id = AV_CODEC_ID_TQI, | ||
183 | .priv_data_size = sizeof(TqiContext), | ||
184 | .init = tqi_decode_init, | ||
185 | .close = tqi_decode_end, | ||
186 | FF_CODEC_DECODE_CB(tqi_decode_frame), | ||
187 | .p.capabilities = AV_CODEC_CAP_DR1, | ||
188 | }; | ||
189 |