9c4a8d4ca3c48df051ddaca531e6e0c1436f8822
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 if (!avctx
->width
|| !avctx
->height
||
47 (avctx
->width
& 7) || (avctx
->height
& 7)) {
48 av_log(avctx
, AV_LOG_ERROR
, "Invalid video dimensions: %dx%d\n",
49 avctx
->width
, avctx
->height
);
50 return AVERROR(EINVAL
);
53 s
->frame
= av_frame_alloc();
55 return AVERROR(ENOMEM
);
57 avctx
->pix_fmt
= AV_PIX_FMT_PAL8
;
58 ff_blockdsp_init(&s
->bdsp
, avctx
);
65 static inline void decode2x2(GetBitContext
*gb
, uint8_t *dst
, int linesize
)
69 switch (get_bits(gb
, 2)) {
71 v
[0] = get_bits(gb
, 8);
72 for (j
= 0; j
< 2; j
++)
73 memset(dst
+ j
* linesize
, v
[0], 2);
76 v
[0] = get_bits(gb
, 8);
77 v
[1] = get_bits(gb
, 8);
78 for (j
= 0; j
< 2; j
++)
79 for (i
= 0; i
< 2; i
++)
80 dst
[j
* linesize
+ i
] = v
[get_bits1(gb
)];
83 for (j
= 0; j
< 2; j
++)
84 for (i
= 0; i
< 2; i
++)
85 dst
[j
* linesize
+ i
] = get_bits(gb
, 8);
92 static inline void decode4x4(GetBitContext
*gb
, uint8_t *dst
, int linesize
)
96 switch (get_bits(gb
, 2)) {
98 v
[0] = get_bits(gb
, 8);
99 for (j
= 0; j
< 4; j
++)
100 memset(dst
+ j
* linesize
, v
[0], 4);
103 v
[0] = get_bits(gb
, 8);
104 v
[1] = get_bits(gb
, 8);
105 for (j
= 2; j
>= 0; j
-= 2) {
106 for (i
= 0; i
< 4; i
++)
107 dst
[j
* linesize
+ i
] = v
[get_bits1(gb
)];
108 for (i
= 0; i
< 4; i
++)
109 dst
[(j
+ 1) * linesize
+ i
] = v
[get_bits1(gb
)];
113 for (j
= 0; j
< 4; j
+= 2)
114 for (i
= 0; i
< 4; i
+= 2)
115 decode2x2(gb
, dst
+ j
* linesize
+ i
, linesize
);
122 static inline void decode8x8(GetBitContext
*gb
, uint8_t *dst
, int linesize
,
123 BlockDSPContext
*bdsp
)
127 switch (get_bits(gb
, 2)) {
129 v
[0] = get_bits(gb
, 8);
130 bdsp
->fill_block_tab
[1](dst
, v
[0], linesize
, 8);
133 v
[0] = get_bits(gb
, 8);
134 v
[1] = get_bits(gb
, 8);
135 for (j
= 7; j
>= 0; j
--)
136 for (i
= 0; i
< 8; i
++)
137 dst
[j
* linesize
+ i
] = v
[get_bits1(gb
)];
140 for (j
= 0; j
< 8; j
+= 4)
141 for (i
= 0; i
< 8; i
+= 4)
142 decode4x4(gb
, dst
+ j
* linesize
+ i
, linesize
);
146 static int decode_frame(AVCodecContext
*avctx
, void *data
, int *got_frame
,
149 JvContext
*s
= avctx
->priv_data
;
150 const uint8_t *buf
= avpkt
->data
;
151 const uint8_t *buf_end
= buf
+ avpkt
->size
;
152 int video_size
, video_type
, i
, j
, ret
;
155 return AVERROR_INVALIDDATA
;
157 video_size
= AV_RL32(buf
);
162 if (video_size
< 0 || video_size
> avpkt
->size
- 5) {
163 av_log(avctx
, AV_LOG_ERROR
, "video size %d invalid\n", video_size
);
164 return AVERROR_INVALIDDATA
;
166 if ((ret
= ff_reget_buffer(avctx
, s
->frame
)) < 0)
169 if (video_type
== 0 || video_type
== 1) {
171 init_get_bits(&gb
, buf
, 8 * video_size
);
173 for (j
= 0; j
< avctx
->height
; j
+= 8)
174 for (i
= 0; i
< avctx
->width
; i
+= 8)
176 s
->frame
->data
[0] + j
* s
->frame
->linesize
[0] + i
,
177 s
->frame
->linesize
[0], &s
->bdsp
);
180 } else if (video_type
== 2) {
182 for (j
= 0; j
< avctx
->height
; j
++)
183 memset(s
->frame
->data
[0] + j
* s
->frame
->linesize
[0],
186 av_log(avctx
, AV_LOG_WARNING
,
187 "unsupported frame type %i\n", video_type
);
188 return AVERROR_INVALIDDATA
;
192 if (buf_end
- buf
>= AVPALETTE_COUNT
* 3) {
193 for (i
= 0; i
< AVPALETTE_COUNT
; i
++) {
194 uint32_t pal
= AV_RB24(buf
);
195 s
->palette
[i
] = 0xFFU
<< 24 | pal
<< 2 | ((pal
>> 4) & 0x30303);
198 s
->palette_has_changed
= 1;
202 s
->frame
->key_frame
= 1;
203 s
->frame
->pict_type
= AV_PICTURE_TYPE_I
;
204 s
->frame
->palette_has_changed
= s
->palette_has_changed
;
205 s
->palette_has_changed
= 0;
206 memcpy(s
->frame
->data
[1], s
->palette
, AVPALETTE_SIZE
);
208 if ((ret
= av_frame_ref(data
, s
->frame
)) < 0)
216 static av_cold
int decode_close(AVCodecContext
*avctx
)
218 JvContext
*s
= avctx
->priv_data
;
220 av_frame_free(&s
->frame
);
225 AVCodec ff_jv_decoder
= {
227 .long_name
= NULL_IF_CONFIG_SMALL("Bitmap Brothers JV video"),
228 .type
= AVMEDIA_TYPE_VIDEO
,
229 .id
= AV_CODEC_ID_JV
,
230 .priv_data_size
= sizeof(JvContext
),
232 .close
= decode_close
,
233 .decode
= decode_frame
,
234 .capabilities
= CODEC_CAP_DR1
,