1 /*
2 * Copyright (c) 2016 Michael Niedermayer
3 *
4 * This file is part of FFmpeg.
5 *
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.
10 *
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.
15 *
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
19 */
20
21 #include "libavutil/intreadwrite.h"
22
23 #include "avcodec.h"
24 #include "codec_internal.h"
25 #include "internal.h"
26
27
m101_decode_init(AVCodecContext * avctx)28 static av_cold int m101_decode_init(AVCodecContext *avctx)
29 {
30 if (avctx->extradata_size < 6*4) {
31 avpriv_request_sample(avctx, "Missing or too small extradata (size %d)", avctx->extradata_size);
32 return AVERROR_INVALIDDATA;
33 }
34
35 if (avctx->extradata[2*4] == 10)
36 avctx->pix_fmt = AV_PIX_FMT_YUV422P10;
37 else if (avctx->extradata[2*4] == 8) {
38 avctx->pix_fmt = AV_PIX_FMT_YUYV422;
39 } else {
40 avpriv_request_sample(avctx, "BPS %d", avctx->extradata[2*4]);
41 return AVERROR_INVALIDDATA;
42 }
43
44 return 0;
45 }
46
m101_decode_frame(AVCodecContext * avctx,AVFrame * frame,int * got_frame,AVPacket * avpkt)47 static int m101_decode_frame(AVCodecContext *avctx, AVFrame *frame,
48 int *got_frame, AVPacket *avpkt)
49 {
50 const uint8_t *buf = avpkt->data;
51 int stride, ret;
52 int x, y;
53 int min_stride = 2 * avctx->width;
54 int bits = avctx->extradata[2*4];
55
56 if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
57 return ret;
58 frame->pict_type = AV_PICTURE_TYPE_I;
59 frame->key_frame = 1;
60
61 stride = AV_RL32(avctx->extradata + 5*4);
62
63 if (avctx->pix_fmt == AV_PIX_FMT_YUV422P10)
64 min_stride = (avctx->width + 15) / 16 * 40;
65
66 if (stride < min_stride || avpkt->size < stride * (uint64_t)avctx->height) {
67 av_log(avctx, AV_LOG_ERROR, "stride (%d) is invalid for packet sized %d\n",
68 stride, avpkt->size);
69 return AVERROR_INVALIDDATA;
70 }
71
72 frame->interlaced_frame = ((avctx->extradata[3*4] & 3) != 3);
73 if (frame->interlaced_frame)
74 frame->top_field_first = avctx->extradata[3*4] & 1;
75
76 for (y = 0; y < avctx->height; y++) {
77 int src_y = y;
78 if (frame->interlaced_frame)
79 src_y = ((y&1)^frame->top_field_first) ? y/2 : (y/2 + avctx->height/2);
80 if (bits == 8) {
81 uint8_t *line = frame->data[0] + y*frame->linesize[0];
82 memcpy(line, buf + src_y*stride, 2*avctx->width);
83 } else {
84 int block;
85 uint16_t *luma = (uint16_t*)&frame->data[0][y*frame->linesize[0]];
86 uint16_t *cb = (uint16_t*)&frame->data[1][y*frame->linesize[1]];
87 uint16_t *cr = (uint16_t*)&frame->data[2][y*frame->linesize[2]];
88 for (block = 0; 16*block < avctx->width; block ++) {
89 const uint8_t *buf_src = buf + src_y*stride + 40*block;
90 for (x = 0; x < 16 && x + 16*block < avctx->width; x++) {
91 int xd = x + 16*block;
92 if (x&1) {
93 luma [xd] = (4*buf_src[2*x + 0]) + ((buf_src[32 + (x>>1)]>>4)&3);
94 } else {
95 luma [xd] = (4*buf_src[2*x + 0]) + (buf_src[32 + (x>>1)] &3);
96 cb[xd>>1] = (4*buf_src[2*x + 1]) + ((buf_src[32 + (x>>1)]>>2)&3);
97 cr[xd>>1] = (4*buf_src[2*x + 3]) + (buf_src[32 + (x>>1)]>>6);
98 }
99 }
100 }
101 }
102 }
103
104 *got_frame = 1;
105 return avpkt->size;
106 }
107
108 const FFCodec ff_m101_decoder = {
109 .p.name = "m101",
110 .p.long_name = NULL_IF_CONFIG_SMALL("Matrox Uncompressed SD"),
111 .p.type = AVMEDIA_TYPE_VIDEO,
112 .p.id = AV_CODEC_ID_M101,
113 .init = m101_decode_init,
114 FF_CODEC_DECODE_CB(m101_decode_frame),
115 .p.capabilities = AV_CODEC_CAP_DR1,
116 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
117 };
118