2 * Copyright (c) 2003 Michael Niedermayer
4 * This file is part of FFmpeg.
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
26 #include "libavutil/attributes.h"
27 #include "libavutil/mem.h"
35 #include "mpeg12data.h"
38 #define ASV2_LEVEL_VLC_BITS 10
42 static VLC dc_ccp_vlc
;
43 static VLC ac_ccp_vlc
;
44 static VLC asv2_level_vlc
;
46 static av_cold
void init_vlcs(ASV1Context
*a
)
53 INIT_VLC_STATIC(&ccp_vlc
, VLC_BITS
, 17,
54 &ff_asv_ccp_tab
[0][1], 2, 1,
55 &ff_asv_ccp_tab
[0][0], 2, 1, 64);
56 INIT_VLC_STATIC(&dc_ccp_vlc
, VLC_BITS
, 8,
57 &ff_asv_dc_ccp_tab
[0][1], 2, 1,
58 &ff_asv_dc_ccp_tab
[0][0], 2, 1, 64);
59 INIT_VLC_STATIC(&ac_ccp_vlc
, VLC_BITS
, 16,
60 &ff_asv_ac_ccp_tab
[0][1], 2, 1,
61 &ff_asv_ac_ccp_tab
[0][0], 2, 1, 64);
62 INIT_VLC_STATIC(&level_vlc
, VLC_BITS
, 7,
63 &ff_asv_level_tab
[0][1], 2, 1,
64 &ff_asv_level_tab
[0][0], 2, 1, 64);
65 INIT_VLC_STATIC(&asv2_level_vlc
, ASV2_LEVEL_VLC_BITS
, 63,
66 &ff_asv2_level_tab
[0][1], 2, 1,
67 &ff_asv2_level_tab
[0][0], 2, 1, 1024);
71 // FIXME write a reversed bitstream reader to avoid the double reverse
72 static inline int asv2_get_bits(GetBitContext
*gb
, int n
)
74 return ff_reverse
[get_bits(gb
, n
) << (8 - n
)];
77 static inline int asv1_get_level(GetBitContext
*gb
)
79 int code
= get_vlc2(gb
, level_vlc
.table
, VLC_BITS
, 1);
82 return get_sbits(gb
, 8);
87 static inline int asv2_get_level(GetBitContext
*gb
)
89 int code
= get_vlc2(gb
, asv2_level_vlc
.table
, ASV2_LEVEL_VLC_BITS
, 1);
92 return (int8_t) asv2_get_bits(gb
, 8);
97 static inline int asv1_decode_block(ASV1Context
*a
, int16_t block
[64])
101 block
[0] = 8 * get_bits(&a
->gb
, 8);
103 for (i
= 0; i
< 11; i
++) {
104 const int ccp
= get_vlc2(&a
->gb
, ccp_vlc
.table
, VLC_BITS
, 1);
109 if (ccp
< 0 || i
>= 10) {
110 av_log(a
->avctx
, AV_LOG_ERROR
, "coded coeff pattern damaged\n");
111 return AVERROR_INVALIDDATA
;
115 block
[a
->scantable
.permutated
[4 * i
+ 0]] = (asv1_get_level(&a
->gb
) * a
->intra_matrix
[4 * i
+ 0]) >> 4;
117 block
[a
->scantable
.permutated
[4 * i
+ 1]] = (asv1_get_level(&a
->gb
) * a
->intra_matrix
[4 * i
+ 1]) >> 4;
119 block
[a
->scantable
.permutated
[4 * i
+ 2]] = (asv1_get_level(&a
->gb
) * a
->intra_matrix
[4 * i
+ 2]) >> 4;
121 block
[a
->scantable
.permutated
[4 * i
+ 3]] = (asv1_get_level(&a
->gb
) * a
->intra_matrix
[4 * i
+ 3]) >> 4;
128 static inline int asv2_decode_block(ASV1Context
*a
, int16_t block
[64])
132 count
= asv2_get_bits(&a
->gb
, 4);
134 block
[0] = 8 * asv2_get_bits(&a
->gb
, 8);
136 ccp
= get_vlc2(&a
->gb
, dc_ccp_vlc
.table
, VLC_BITS
, 1);
139 block
[a
->scantable
.permutated
[1]] = (asv2_get_level(&a
->gb
) * a
->intra_matrix
[1]) >> 4;
141 block
[a
->scantable
.permutated
[2]] = (asv2_get_level(&a
->gb
) * a
->intra_matrix
[2]) >> 4;
143 block
[a
->scantable
.permutated
[3]] = (asv2_get_level(&a
->gb
) * a
->intra_matrix
[3]) >> 4;
146 for (i
= 1; i
< count
+ 1; i
++) {
147 const int ccp
= get_vlc2(&a
->gb
, ac_ccp_vlc
.table
, VLC_BITS
, 1);
151 block
[a
->scantable
.permutated
[4 * i
+ 0]] = (asv2_get_level(&a
->gb
) * a
->intra_matrix
[4 * i
+ 0]) >> 4;
153 block
[a
->scantable
.permutated
[4 * i
+ 1]] = (asv2_get_level(&a
->gb
) * a
->intra_matrix
[4 * i
+ 1]) >> 4;
155 block
[a
->scantable
.permutated
[4 * i
+ 2]] = (asv2_get_level(&a
->gb
) * a
->intra_matrix
[4 * i
+ 2]) >> 4;
157 block
[a
->scantable
.permutated
[4 * i
+ 3]] = (asv2_get_level(&a
->gb
) * a
->intra_matrix
[4 * i
+ 3]) >> 4;
164 static inline int decode_mb(ASV1Context
*a
, int16_t block
[6][64])
168 a
->bdsp
.clear_blocks(block
[0]);
170 if (a
->avctx
->codec_id
== AV_CODEC_ID_ASV1
) {
171 for (i
= 0; i
< 6; i
++) {
172 if (asv1_decode_block(a
, block
[i
]) < 0)
176 for (i
= 0; i
< 6; i
++) {
177 if (asv2_decode_block(a
, block
[i
]) < 0)
184 static inline void idct_put(ASV1Context
*a
, AVFrame
*frame
, int mb_x
, int mb_y
)
186 int16_t(*block
)[64] = a
->block
;
187 int linesize
= frame
->linesize
[0];
189 uint8_t *dest_y
= frame
->data
[0] + (mb_y
* 16 * linesize
) + mb_x
* 16;
190 uint8_t *dest_cb
= frame
->data
[1] + (mb_y
* 8 * frame
->linesize
[1]) + mb_x
* 8;
191 uint8_t *dest_cr
= frame
->data
[2] + (mb_y
* 8 * frame
->linesize
[2]) + mb_x
* 8;
193 a
->idsp
.idct_put(dest_y
, linesize
, block
[0]);
194 a
->idsp
.idct_put(dest_y
+ 8, linesize
, block
[1]);
195 a
->idsp
.idct_put(dest_y
+ 8 * linesize
, linesize
, block
[2]);
196 a
->idsp
.idct_put(dest_y
+ 8 * linesize
+ 8, linesize
, block
[3]);
198 if (!(a
->avctx
->flags
& CODEC_FLAG_GRAY
)) {
199 a
->idsp
.idct_put(dest_cb
, frame
->linesize
[1], block
[4]);
200 a
->idsp
.idct_put(dest_cr
, frame
->linesize
[2], block
[5]);
204 static int decode_frame(AVCodecContext
*avctx
, void *data
, int *got_frame
,
207 ASV1Context
*const a
= avctx
->priv_data
;
208 const uint8_t *buf
= avpkt
->data
;
209 int buf_size
= avpkt
->size
;
210 AVFrame
*const p
= data
;
213 if ((ret
= ff_get_buffer(avctx
, p
, 0)) < 0)
215 p
->pict_type
= AV_PICTURE_TYPE_I
;
218 av_fast_padded_malloc(&a
->bitstream_buffer
, &a
->bitstream_buffer_size
,
220 if (!a
->bitstream_buffer
)
221 return AVERROR(ENOMEM
);
223 if (avctx
->codec_id
== AV_CODEC_ID_ASV1
) {
224 a
->bbdsp
.bswap_buf((uint32_t *) a
->bitstream_buffer
,
225 (const uint32_t *) buf
, buf_size
/ 4);
228 for (i
= 0; i
< buf_size
; i
++)
229 a
->bitstream_buffer
[i
] = ff_reverse
[buf
[i
]];
232 init_get_bits(&a
->gb
, a
->bitstream_buffer
, buf_size
* 8);
234 for (mb_y
= 0; mb_y
< a
->mb_height2
; mb_y
++) {
235 for (mb_x
= 0; mb_x
< a
->mb_width2
; mb_x
++) {
236 if ((ret
= decode_mb(a
, a
->block
)) < 0)
239 idct_put(a
, p
, mb_x
, mb_y
);
243 if (a
->mb_width2
!= a
->mb_width
) {
245 for (mb_y
= 0; mb_y
< a
->mb_height2
; mb_y
++) {
246 if ((ret
= decode_mb(a
, a
->block
)) < 0)
249 idct_put(a
, p
, mb_x
, mb_y
);
253 if (a
->mb_height2
!= a
->mb_height
) {
254 mb_y
= a
->mb_height2
;
255 for (mb_x
= 0; mb_x
< a
->mb_width
; mb_x
++) {
256 if ((ret
= decode_mb(a
, a
->block
)) < 0)
259 idct_put(a
, p
, mb_x
, mb_y
);
267 return (get_bits_count(&a
->gb
) + 31) / 32 * 4;
270 static av_cold
int decode_init(AVCodecContext
*avctx
)
272 ASV1Context
*const a
= avctx
->priv_data
;
273 const int scale
= avctx
->codec_id
== AV_CODEC_ID_ASV1
? 1 : 2;
276 if (avctx
->extradata_size
< 1) {
277 av_log(avctx
, AV_LOG_WARNING
, "No extradata provided\n");
280 ff_asv_common_init(avctx
);
281 ff_blockdsp_init(&a
->bdsp
, avctx
);
282 ff_idctdsp_init(&a
->idsp
, avctx
);
284 ff_init_scantable(a
->idsp
.idct_permutation
, &a
->scantable
, ff_asv_scantab
);
285 avctx
->pix_fmt
= AV_PIX_FMT_YUV420P
;
287 if (avctx
->extradata_size
< 1 || (a
->inv_qscale
= avctx
->extradata
[0]) == 0) {
288 av_log(avctx
, AV_LOG_ERROR
, "illegal qscale 0\n");
289 if (avctx
->codec_id
== AV_CODEC_ID_ASV1
)
295 for (i
= 0; i
< 64; i
++) {
296 int index
= ff_asv_scantab
[i
];
298 a
->intra_matrix
[i
] = 64 * scale
* ff_mpeg1_default_intra_matrix
[index
] /
305 static av_cold
int decode_end(AVCodecContext
*avctx
)
307 ASV1Context
*const a
= avctx
->priv_data
;
309 av_freep(&a
->bitstream_buffer
);
310 a
->bitstream_buffer_size
= 0;
315 #if CONFIG_ASV1_DECODER
316 AVCodec ff_asv1_decoder
= {
318 .long_name
= NULL_IF_CONFIG_SMALL("ASUS V1"),
319 .type
= AVMEDIA_TYPE_VIDEO
,
320 .id
= AV_CODEC_ID_ASV1
,
321 .priv_data_size
= sizeof(ASV1Context
),
324 .decode
= decode_frame
,
325 .capabilities
= CODEC_CAP_DR1
,
329 #if CONFIG_ASV2_DECODER
330 AVCodec ff_asv2_decoder
= {
332 .long_name
= NULL_IF_CONFIG_SMALL("ASUS V2"),
333 .type
= AVMEDIA_TYPE_VIDEO
,
334 .id
= AV_CODEC_ID_ASV2
,
335 .priv_data_size
= sizeof(ASV1Context
),
338 .decode
= decode_frame
,
339 .capabilities
= CODEC_CAP_DR1
,