47e8edcae60bd83f73476d0d75f46e51327e13a7
2 * Bitmap Brothers JV video decoder
3 * Copyright (c) 2011 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 Street, Fifth Floor, Boston, MA 02110-1301 USA
24 * Bitmap Brothers JV video decoder
25 * @author Peter Ross <pross@xvid.org>
28 #include "libavutil/intreadwrite.h"
35 typedef struct JvContext
{
38 uint32_t palette
[AVPALETTE_COUNT
];
39 int palette_has_changed
;
42 static av_cold
int decode_init(AVCodecContext
*avctx
)
44 JvContext
*s
= avctx
->priv_data
;
46 s
->frame
= av_frame_alloc();
48 return AVERROR(ENOMEM
);
50 avctx
->pix_fmt
= AV_PIX_FMT_PAL8
;
51 ff_blockdsp_init(&s
->bdsp
, avctx
);
58 static inline void decode2x2(GetBitContext
*gb
, uint8_t *dst
, int linesize
)
62 switch (get_bits(gb
, 2)) {
64 v
[0] = get_bits(gb
, 8);
65 for (j
= 0; j
< 2; j
++)
66 memset(dst
+ j
* linesize
, v
[0], 2);
69 v
[0] = get_bits(gb
, 8);
70 v
[1] = get_bits(gb
, 8);
71 for (j
= 0; j
< 2; j
++)
72 for (i
= 0; i
< 2; i
++)
73 dst
[j
* linesize
+ i
] = v
[get_bits1(gb
)];
76 for (j
= 0; j
< 2; j
++)
77 for (i
= 0; i
< 2; i
++)
78 dst
[j
* linesize
+ i
] = get_bits(gb
, 8);
85 static inline void decode4x4(GetBitContext
*gb
, uint8_t *dst
, int linesize
)
89 switch (get_bits(gb
, 2)) {
91 v
[0] = get_bits(gb
, 8);
92 for (j
= 0; j
< 4; j
++)
93 memset(dst
+ j
* linesize
, v
[0], 4);
96 v
[0] = get_bits(gb
, 8);
97 v
[1] = get_bits(gb
, 8);
98 for (j
= 2; j
>= 0; j
-= 2) {
99 for (i
= 0; i
< 4; i
++)
100 dst
[j
* linesize
+ i
] = v
[get_bits1(gb
)];
101 for (i
= 0; i
< 4; i
++)
102 dst
[(j
+ 1) * linesize
+ i
] = v
[get_bits1(gb
)];
106 for (j
= 0; j
< 4; j
+= 2)
107 for (i
= 0; i
< 4; i
+= 2)
108 decode2x2(gb
, dst
+ j
* linesize
+ i
, linesize
);
115 static inline void decode8x8(GetBitContext
*gb
, uint8_t *dst
, int linesize
,
116 BlockDSPContext
*bdsp
)
120 switch (get_bits(gb
, 2)) {
122 v
[0] = get_bits(gb
, 8);
123 bdsp
->fill_block_tab
[1](dst
, v
[0], linesize
, 8);
126 v
[0] = get_bits(gb
, 8);
127 v
[1] = get_bits(gb
, 8);
128 for (j
= 7; j
>= 0; j
--)
129 for (i
= 0; i
< 8; i
++)
130 dst
[j
* linesize
+ i
] = v
[get_bits1(gb
)];
133 for (j
= 0; j
< 8; j
+= 4)
134 for (i
= 0; i
< 8; i
+= 4)
135 decode4x4(gb
, dst
+ j
* linesize
+ i
, linesize
);
139 static int decode_frame(AVCodecContext
*avctx
, void *data
, int *got_frame
,
142 JvContext
*s
= avctx
->priv_data
;
143 const uint8_t *buf
= avpkt
->data
;
144 const uint8_t *buf_end
= buf
+ avpkt
->size
;
145 int video_size
, video_type
, i
, j
, ret
;
148 return AVERROR_INVALIDDATA
;
150 video_size
= AV_RL32(buf
);
155 if (video_size
< 0 || video_size
> avpkt
->size
- 5) {
156 av_log(avctx
, AV_LOG_ERROR
, "video size %d invalid\n", video_size
);
157 return AVERROR_INVALIDDATA
;
159 if ((ret
= ff_reget_buffer(avctx
, s
->frame
)) < 0)
162 if (video_type
== 0 || video_type
== 1) {
164 init_get_bits(&gb
, buf
, 8 * video_size
);
166 for (j
= 0; j
< avctx
->height
; j
+= 8)
167 for (i
= 0; i
< avctx
->width
; i
+= 8)
169 s
->frame
->data
[0] + j
* s
->frame
->linesize
[0] + i
,
170 s
->frame
->linesize
[0], &s
->bdsp
);
173 } else if (video_type
== 2) {
175 for (j
= 0; j
< avctx
->height
; j
++)
176 memset(s
->frame
->data
[0] + j
* s
->frame
->linesize
[0],
179 av_log(avctx
, AV_LOG_WARNING
,
180 "unsupported frame type %i\n", video_type
);
181 return AVERROR_INVALIDDATA
;
185 if (buf_end
- buf
>= AVPALETTE_COUNT
* 3) {
186 for (i
= 0; i
< AVPALETTE_COUNT
; i
++) {
187 uint32_t pal
= AV_RB24(buf
);
188 s
->palette
[i
] = 0xFFU
<< 24 | pal
<< 2 | ((pal
>> 4) & 0x30303);
191 s
->palette_has_changed
= 1;
195 s
->frame
->key_frame
= 1;
196 s
->frame
->pict_type
= AV_PICTURE_TYPE_I
;
197 s
->frame
->palette_has_changed
= s
->palette_has_changed
;
198 s
->palette_has_changed
= 0;
199 memcpy(s
->frame
->data
[1], s
->palette
, AVPALETTE_SIZE
);
201 if ((ret
= av_frame_ref(data
, s
->frame
)) < 0)
209 static av_cold
int decode_close(AVCodecContext
*avctx
)
211 JvContext
*s
= avctx
->priv_data
;
213 av_frame_free(&s
->frame
);
218 AVCodec ff_jv_decoder
= {
220 .long_name
= NULL_IF_CONFIG_SMALL("Bitmap Brothers JV video"),
221 .type
= AVMEDIA_TYPE_VIDEO
,
222 .id
= AV_CODEC_ID_JV
,
223 .priv_data_size
= sizeof(JvContext
),
225 .close
= decode_close
,
226 .decode
= decode_frame
,
227 .capabilities
= CODEC_CAP_DR1
,