FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
mvcdec.c
Go to the documentation of this file.
1 /*
2  * Silicon Graphics Motion Video Compressor 1 & 2 decoder
3  * Copyright (c) 2012 Peter Ross
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 Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 /**
23  * @file
24  * Silicon Graphics Motion Video Compressor 1 & 2 decoder
25  */
26 
27 #include "libavutil/intreadwrite.h"
28 
29 #include "avcodec.h"
30 #include "bytestream.h"
31 #include "internal.h"
32 
33 typedef struct MvcContext {
35  int vflip;
36 } MvcContext;
37 
39 {
40  MvcContext *s = avctx->priv_data;
41  int width = avctx->width;
42  int height = avctx->height;
43  int ret;
44 
45  if (avctx->codec_id == AV_CODEC_ID_MVC1) {
46  width += 3;
47  height += 3;
48  }
49  width &= ~3;
50  height &= ~3;
51  if ((ret = ff_set_dimensions(avctx, width, height)) < 0)
52  return ret;
53 
54  avctx->pix_fmt = (avctx->codec_id == AV_CODEC_ID_MVC1) ? AV_PIX_FMT_RGB555
56  s->frame = av_frame_alloc();
57  if (!s->frame)
58  return AVERROR(ENOMEM);
59 
60  s->vflip = avctx->extradata_size >= 9 &&
61  !memcmp(avctx->extradata + avctx->extradata_size - 9, "BottomUp", 9);
62  return 0;
63 }
64 
65 static int decode_mvc1(AVCodecContext *avctx, GetByteContext *gb,
66  uint8_t *dst_start, int width, int height, int linesize)
67 {
68  uint8_t *dst;
69  uint16_t v[8];
70  int mask, x, y, i;
71 
72  for (y = 0; y < height; y += 4) {
73  for (x = 0; x < width; x += 4) {
74  if (bytestream2_get_bytes_left(gb) < 6)
75  return 0;
76 
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;
84  }
85  for (i = 2; i < 8; i++)
86  v[i] = bytestream2_get_be16u(gb);
87  } else {
88  v[2] = v[4] = v[6] = v[0];
89  v[3] = v[5] = v[7] = v[1];
90  }
91 
92 #define PIX16(target, true, false) \
93  i = (mask & target) ? true : false; \
94  AV_WN16A(dst, v[i] & 0x7FFF); \
95  dst += 2;
96 
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)
103 
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);
108  }
109  }
110  return 0;
111 }
112 
113 static void set_4x4_block(uint8_t *dst, int linesize, uint32_t pixel)
114 {
115  int i, j;
116  for (j = 0; j < 4; j++)
117  for (i = 0; i < 4; i++)
118  AV_WN32A(dst + j * linesize + i * 4, pixel);
119 }
120 
121 #define PIX32(target, true, false) \
122  AV_WN32A(dst, (mask & target) ? v[true] : v[false]); \
123  dst += 4;
124 
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)
131 
132 #define MVC2_BLOCK \
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);
137 
139  uint8_t *dst_start, int width, int height,
140  int linesize, int vflip)
141 {
142  uint8_t *dst;
143  uint32_t color[128], v[8];
144  int w, h, nb_colors, i, x, y, p0, p1, mask;
145 
146  if (bytestream2_get_bytes_left(gb) < 6)
147  return AVERROR_INVALIDDATA;
148 
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");
153 
154  if (bytestream2_get_byteu(gb)) {
155  avpriv_request_sample(avctx, "bitmap feature");
156  return AVERROR_PATCHWELCOME;
157  }
158 
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);
164  if (nb_colors > 128)
165  bytestream2_skip(gb, (nb_colors - 128) * 3);
166 
167  if (vflip) {
168  dst_start += (height - 1) * linesize;
169  linesize = -linesize;
170  }
171  x = y = 0;
172  while (bytestream2_get_bytes_left(gb) >= 1) {
173  p0 = bytestream2_get_byteu(gb);
174  if ((p0 & 0x80)) {
175  if ((p0 & 0x40)) {
176  p0 &= 0x3F;
177  p0 = (p0 << 2) | (p0 >> 4);
178  set_4x4_block(dst_start + y * linesize + x * 4, linesize,
179  0xFF000000 | (p0 << 16) | (p0 << 8) | p0);
180  } else {
181  int g, r;
182  p0 &= 0x3F;
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);
190  }
191  } else {
192  if (bytestream2_get_bytes_left(gb) < 1)
193  return AVERROR_INVALIDDATA;
194  p1 = bytestream2_get_byteu(gb);
195  if ((p1 & 0x80)) {
196  if ((p0 & 0x7F) == (p1 & 0x7F)) {
197  set_4x4_block(dst_start + y * linesize + x * 4, linesize,
198  color[p0 & 0x7F]);
199  } else {
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);
205  MVC2_BLOCK
206  }
207  } else {
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);
215  MVC2_BLOCK
216  }
217  }
218 
219  x += 4;
220  if (x >= width) {
221  y += 4;
222  if (y >= height)
223  break;
224  x = 0;
225  }
226  }
227  return 0;
228 }
229 
230 static int mvc_decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
231  AVPacket *avpkt)
232 {
233  MvcContext *s = avctx->priv_data;
234  GetByteContext gb;
235  int ret;
236 
237  if ((ret = ff_reget_buffer(avctx, s->frame)) < 0)
238  return ret;
239 
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]);
244  else
245  ret = decode_mvc2(avctx, &gb, s->frame->data[0],
246  avctx->width, avctx->height, s->frame->linesize[0],
247  s->vflip);
248  if (ret < 0)
249  return ret;
250 
251  *got_frame = 1;
252  if ((ret = av_frame_ref(data, s->frame)) < 0)
253  return ret;
254 
255  return avpkt->size;
256 }
257 
259 {
260  MvcContext *s = avctx->priv_data;
261 
262  av_frame_free(&s->frame);
263 
264  return 0;
265 }
266 
267 #if CONFIG_MVC1_DECODER
268 AVCodec ff_mvc1_decoder = {
269  .name = "mvc1",
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),
275  .close = mvc_decode_end,
277  .capabilities = AV_CODEC_CAP_DR1,
278 };
279 #endif
280 
281 #if CONFIG_MVC2_DECODER
282 AVCodec ff_mvc2_decoder = {
283  .name = "mvc2",
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),
289  .close = mvc_decode_end,
291  .capabilities = AV_CODEC_CAP_DR1,
292 };
293 #endif
const char * s
Definition: avisynth_c.h:631
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
This structure describes decoded (raw) audio or video data.
Definition: frame.h:180
static void set_4x4_block(uint8_t *dst, int linesize, uint32_t pixel)
Definition: mvcdec.c:113
ptrdiff_t const GLvoid * data
Definition: opengl_enc.c:101
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:182
int ff_set_dimensions(AVCodecContext *s, int width, int height)
Check that the provided frame dimensions are valid and set them on the codec context.
Definition: utils.c:210
const char * g
Definition: vf_curves.c:108
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
int size
Definition: avcodec.h:1575
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1867
static av_always_inline void bytestream2_init(GetByteContext *g, const uint8_t *buf, int buf_size)
Definition: bytestream.h:133
#define MVC2_BLOCK
Definition: mvcdec.c:132
AVCodec.
Definition: avcodec.h:3531
#define AV_WN32A(p, v)
Definition: intreadwrite.h:538
void void avpriv_request_sample(void *avc, const char *msg,...) av_printf_format(2
Log a generic warning message about a missing feature.
uint8_t
#define av_cold
Definition: attributes.h:82
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
Definition: frame.c:140
#define ROW16(row, a1, a0, b1, b0)
int av_frame_ref(AVFrame *dst, const AVFrame *src)
Set up a new reference to the data described by the source frame.
Definition: frame.c:374
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:1754
static int mvc_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: mvcdec.c:230
uint8_t * data
Definition: avcodec.h:1574
#define av_log(a,...)
static const uint16_t mask[17]
Definition: lzw.c:38
#define AVERROR(e)
Definition: error.h:43
static av_always_inline void bytestream2_skip(GetByteContext *g, unsigned int size)
Definition: bytestream.h:164
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:153
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:176
const char * r
Definition: vf_curves.c:107
static av_cold int mvc_decode_init(AVCodecContext *avctx)
Definition: mvcdec.c:38
static av_always_inline unsigned int bytestream2_get_bytes_left(GetByteContext *g)
Definition: bytestream.h:154
const char * name
Name of the codec implementation.
Definition: avcodec.h:3538
int ff_reget_buffer(AVCodecContext *avctx, AVFrame *frame)
Identical in function to av_frame_make_writable(), except it uses ff_get_buffer() to allocate the buf...
Definition: utils.c:977
#define FFMIN(a, b)
Definition: common.h:96
int width
picture width / height.
Definition: avcodec.h:1826
static int decode_mvc2(AVCodecContext *avctx, GetByteContext *gb, uint8_t *dst_start, int width, int height, int linesize, int vflip)
Definition: mvcdec.c:138
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
Definition: error.h:62
static av_cold int mvc_decode_end(AVCodecContext *avctx)
Definition: mvcdec.c:258
Libavcodec external API header.
enum AVCodecID codec_id
Definition: avcodec.h:1656
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:211
main external API structure.
Definition: avcodec.h:1639
#define AV_PIX_FMT_RGB32
Definition: pixfmt.h:315
int extradata_size
Definition: avcodec.h:1755
BYTE int const BYTE int int int height
Definition: avisynth_c.h:676
AVFrame * frame
Definition: mvcdec.c:34
uint8_t pixel
Definition: tiny_ssim.c:42
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:194
static int decode(AVCodecContext *avctx, void *data, int *got_sub, AVPacket *avpkt)
Definition: ccaption_dec.c:722
common internal api header.
int vflip
Definition: mvcdec.c:35
static int decode_mvc1(AVCodecContext *avctx, GetByteContext *gb, uint8_t *dst_start, int width, int height, int linesize)
Definition: mvcdec.c:65
#define AV_PIX_FMT_RGB555
Definition: pixfmt.h:326
void * priv_data
Definition: avcodec.h:1681
This structure stores compressed data.
Definition: avcodec.h:1551
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:950
static int width