2 * Silicon Graphics Motion Video Compressor 1 & 2 decoder
3 * Copyright (c) 2012 Peter Ross
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 * Silicon Graphics Motion Video Compressor 1 & 2 decoder
27 #include "libavutil/intreadwrite.h"
30 #include "bytestream.h"
33 typedef struct MvcContext
{
38 static av_cold
int mvc_decode_init(AVCodecContext
*avctx
)
40 MvcContext
*s
= avctx
->priv_data
;
41 int width
= avctx
->width
;
42 int height
= avctx
->height
;
45 if (avctx
->codec_id
== AV_CODEC_ID_MVC1
) {
51 if ((ret
= ff_set_dimensions(avctx
, width
, height
)) < 0)
54 avctx
->pix_fmt
= (avctx
->codec_id
== AV_CODEC_ID_MVC1
) ? AV_PIX_FMT_RGB555
56 s
->frame
= av_frame_alloc();
58 return AVERROR(ENOMEM
);
60 s
->vflip
= avctx
->extradata_size
>= 9 &&
61 !memcmp(avctx
->extradata
+ avctx
->extradata_size
- 9, "BottomUp", 9);
65 static int decode_mvc1(AVCodecContext
*avctx
, GetByteContext
*gb
,
66 uint8_t *dst_start
, int width
, int height
, int linesize
)
72 for (y
= 0; y
< height
; y
+= 4) {
73 for (x
= 0; x
< width
; x
+= 4) {
74 if (bytestream2_get_bytes_left(gb
) < 6)
77 mask
= bytestream2_get_be16u(gb
);
78 v
[0] = bytestream2_get_be16u(gb
);
79 v
[1] = bytestream2_get_be16u(gb
);
80 if ((v
[0] & 0x8000)) {
81 if (bytestream2_get_bytes_left(gb
) < 12) {
82 av_log(avctx
, AV_LOG_WARNING
, "buffer overflow\n");
83 return AVERROR_INVALIDDATA
;
85 for (i
= 2; i
< 8; i
++)
86 v
[i
] = bytestream2_get_be16u(gb
);
88 v
[2] = v
[4] = v
[6] = v
[0];
89 v
[3] = v
[5] = v
[7] = v
[1];
92 #define PIX16(target, true, false) \
93 i = (mask & target) ? true : false; \
94 AV_WN16A(dst, v[i] & 0x7FFF); \
97 #define ROW16(row, a1, a0, b1, b0) \
98 dst = dst_start + (y + row) * linesize + x * 2; \
99 PIX16(1 << (row * 4), a1, a0) \
100 PIX16(1 << (row * 4 + 1), a1, a0) \
101 PIX16(1 << (row * 4 + 2), b1, b0) \
102 PIX16(1 << (row * 4 + 3), b1, b0)
104 ROW16(0, 0, 1, 2, 3);
105 ROW16(1, 0, 1, 2, 3);
106 ROW16(2, 4, 5, 6, 7);
107 ROW16(3, 4, 5, 6, 7);
113 static void set_4x4_block(uint8_t *dst
, int linesize
, uint32_t pixel
)
116 for (j
= 0; j
< 4; j
++)
117 for (i
= 0; i
< 4; i
++)
118 AV_WN32A(dst
+ j
* linesize
+ i
* 4, pixel
);
121 #define PIX32(target, true, false) \
122 AV_WN32A(dst, (mask & target) ? v[true] : v[false]); \
125 #define ROW32(row, a1, a0, b1, b0) \
126 dst = dst_start + (y + row) * linesize + x * 4; \
127 PIX32(1 << (row * 4), a1, a0) \
128 PIX32(1 << (row * 4 + 1), a1, a0) \
129 PIX32(1 << (row * 4 + 2), b1, b0) \
130 PIX32(1 << (row * 4 + 3), b1, b0)
133 ROW32(0, 1, 0, 3, 2); \
134 ROW32(1, 1, 0, 3, 2); \
135 ROW32(2, 5, 4, 7, 6); \
136 ROW32(3, 5, 4, 7, 6);
138 static int decode_mvc2(AVCodecContext
*avctx
, GetByteContext
*gb
,
139 uint8_t *dst_start
, int width
, int height
,
140 int linesize
, int vflip
)
143 uint32_t color
[128], v
[8];
144 int w
, h
, nb_colors
, i
, x
, y
, p0
, p1
, mask
;
146 if (bytestream2_get_bytes_left(gb
) < 6)
147 return AVERROR_INVALIDDATA
;
149 w
= bytestream2_get_be16u(gb
);
150 h
= bytestream2_get_be16u(gb
);
151 if ((w
& ~3) != width
|| (h
& ~3) != height
)
152 av_log(avctx
, AV_LOG_WARNING
, "dimension mismatch\n");
154 if (bytestream2_get_byteu(gb
)) {
155 avpriv_request_sample(avctx
, "bitmap feature");
156 return AVERROR_PATCHWELCOME
;
159 nb_colors
= bytestream2_get_byteu(gb
);
160 if (bytestream2_get_bytes_left(gb
) < nb_colors
* 3)
161 return AVERROR_INVALIDDATA
;
162 for (i
= 0; i
< FFMIN(nb_colors
, 128); i
++)
163 color
[i
] = 0xFF000000 | bytestream2_get_be24u(gb
);
165 bytestream2_skip(gb
, (nb_colors
- 128) * 3);
168 dst_start
+= (height
- 1) * linesize
;
169 linesize
= -linesize
;
172 while (bytestream2_get_bytes_left(gb
) >= 1) {
173 p0
= bytestream2_get_byteu(gb
);
177 p0
= (p0
<< 2) | (p0
>> 4);
178 set_4x4_block(dst_start
+ y
* linesize
+ x
* 4, linesize
,
179 0xFF000000 | (p0
<< 16) | (p0
<< 8) | p0
);
183 p0
= (p0
<< 2) | (p0
>> 4);
184 if (bytestream2_get_bytes_left(gb
) < 2)
185 return AVERROR_INVALIDDATA
;
186 g
= bytestream2_get_byteu(gb
);
187 r
= bytestream2_get_byteu(gb
);
188 set_4x4_block(dst_start
+ y
* linesize
+ x
* 4, linesize
,
189 0xFF000000 | (r
<< 16) | (g
<< 8) | p0
);
192 if (bytestream2_get_bytes_left(gb
) < 1)
193 return AVERROR_INVALIDDATA
;
194 p1
= bytestream2_get_byteu(gb
);
196 if ((p0
& 0x7F) == (p1
& 0x7F)) {
197 set_4x4_block(dst_start
+ y
* linesize
+ x
* 4, linesize
,
200 if (bytestream2_get_bytes_left(gb
) < 2)
201 return AVERROR_INVALIDDATA
;
202 v
[0] = v
[2] = v
[4] = v
[6] = color
[p0
& 0x7F];
203 v
[1] = v
[3] = v
[5] = v
[7] = color
[p1
& 0x7F];
204 mask
= bytestream2_get_le16u(gb
);
208 if (bytestream2_get_bytes_left(gb
) < 8)
209 return AVERROR_INVALIDDATA
;
210 v
[0] = color
[p0
& 0x7F];
211 v
[1] = color
[p1
& 0x7F];
212 for (i
= 2; i
< 8; i
++)
213 v
[i
] = color
[bytestream2_get_byteu(gb
) & 0x7F];
214 mask
= bytestream2_get_le16u(gb
);
230 static int mvc_decode_frame(AVCodecContext
*avctx
, void *data
, int *got_frame
,
233 MvcContext
*s
= avctx
->priv_data
;
237 if ((ret
= ff_reget_buffer(avctx
, s
->frame
)) < 0)
240 bytestream2_init(&gb
, avpkt
->data
, avpkt
->size
);
241 if (avctx
->codec_id
== AV_CODEC_ID_MVC1
)
242 ret
= decode_mvc1(avctx
, &gb
, s
->frame
->data
[0],
243 avctx
->width
, avctx
->height
, s
->frame
->linesize
[0]);
245 ret
= decode_mvc2(avctx
, &gb
, s
->frame
->data
[0],
246 avctx
->width
, avctx
->height
, s
->frame
->linesize
[0],
252 if ((ret
= av_frame_ref(data
, s
->frame
)) < 0)
258 static av_cold
int mvc_decode_end(AVCodecContext
*avctx
)
260 MvcContext
*s
= avctx
->priv_data
;
262 av_frame_free(&s
->frame
);
267 #if CONFIG_MVC1_DECODER
268 AVCodec ff_mvc1_decoder
= {
270 .long_name
= NULL_IF_CONFIG_SMALL("Silicon Graphics Motion Video Compressor 1"),
271 .type
= AVMEDIA_TYPE_VIDEO
,
272 .id
= AV_CODEC_ID_MVC1
,
273 .priv_data_size
= sizeof(MvcContext
),
274 .init
= mvc_decode_init
,
275 .close
= mvc_decode_end
,
276 .decode
= mvc_decode_frame
,
277 .capabilities
= CODEC_CAP_DR1
,
281 #if CONFIG_MVC2_DECODER
282 AVCodec ff_mvc2_decoder
= {
284 .long_name
= NULL_IF_CONFIG_SMALL("Silicon Graphics Motion Video Compressor 2"),
285 .type
= AVMEDIA_TYPE_VIDEO
,
286 .id
= AV_CODEC_ID_MVC2
,
287 .priv_data_size
= sizeof(MvcContext
),
288 .init
= mvc_decode_init
,
289 .close
= mvc_decode_end
,
290 .decode
= mvc_decode_frame
,
291 .capabilities
= CODEC_CAP_DR1
,