1/* 2 * V210 decoder 3 * 4 * Copyright (C) 2009 Michael Niedermayer <michaelni@gmx.at> 5 * Copyright (c) 2009 Baptiste Coudurier <baptiste dot coudurier at gmail dot com> 6 * 7 * This file is part of FFmpeg. 8 * 9 * FFmpeg is free software; you can redistribute it and/or 10 * modify it under the terms of the GNU Lesser General Public 11 * License as published by the Free Software Foundation; either 12 * version 2.1 of the License, or (at your option) any later version. 13 * 14 * FFmpeg is distributed in the hope that it will be useful, 15 * but WITHOUT ANY WARRANTY; without even the implied warranty of 16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 17 * Lesser General Public License for more details. 18 * 19 * You should have received a copy of the GNU Lesser General Public 20 * License along with FFmpeg; if not, write to the Free Software 21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 22 */ 23 24#include "avcodec.h" 25#include "internal.h" 26#include "v210dec.h" 27#include "libavutil/bswap.h" 28#include "libavutil/internal.h" 29#include "libavutil/mem.h" 30 31#define READ_PIXELS(a, b, c) \ 32 do { \ 33 val = av_le2ne32(*src++); \ 34 *a++ = val & 0x3FF; \ 35 *b++ = (val >> 10) & 0x3FF; \ 36 *c++ = (val >> 20) & 0x3FF; \ 37 } while (0) 38 39static void v210_planar_unpack_c(const uint32_t *src, uint16_t *y, uint16_t *u, uint16_t *v, int width) 40{ 41 uint32_t val; 42 int i; 43 44 for( i = 0; i < width-5; i += 6 ){ 45 READ_PIXELS(u, y, v); 46 READ_PIXELS(y, u, y); 47 READ_PIXELS(v, y, u); 48 READ_PIXELS(y, v, y); 49 } 50} 51 52static av_cold int decode_init(AVCodecContext *avctx) 53{ 54 V210DecContext *s = avctx->priv_data; 55 56 if (avctx->width & 1) { 57 av_log(avctx, AV_LOG_ERROR, "v210 needs even width\n"); 58 return AVERROR_INVALIDDATA; 59 } 60 avctx->pix_fmt = AV_PIX_FMT_YUV422P10; 61 avctx->bits_per_raw_sample = 10; 62 63 s->unpack_frame = v210_planar_unpack_c; 64 65 if (HAVE_MMX) 66 v210_x86_init(s); 67 68 return 0; 69} 70 71static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, 72 AVPacket *avpkt) 73{ 74 V210DecContext *s = avctx->priv_data; 75 76 int h, w, ret, stride, aligned_input; 77 AVFrame *pic = data; 78 const uint8_t *psrc = avpkt->data; 79 uint16_t *y, *u, *v; 80 81 if (s->custom_stride ) 82 stride = s->custom_stride; 83 else { 84 int aligned_width = ((avctx->width + 47) / 48) * 48; 85 stride = aligned_width * 8 / 3; 86 } 87 88 if (avpkt->size < stride * avctx->height) { 89 if ((((avctx->width + 23) / 24) * 24 * 8) / 3 * avctx->height == avpkt->size) { 90 stride = avpkt->size / avctx->height; 91 if (!s->stride_warning_shown) 92 av_log(avctx, AV_LOG_WARNING, "Broken v210 with too small padding (64 byte) detected\n"); 93 s->stride_warning_shown = 1; 94 } else { 95 av_log(avctx, AV_LOG_ERROR, "packet too small\n"); 96 return AVERROR_INVALIDDATA; 97 } 98 } 99 100 aligned_input = !((uintptr_t)psrc & 0xf) && !(stride & 0xf); 101 if (aligned_input != s->aligned_input) { 102 s->aligned_input = aligned_input; 103 if (HAVE_MMX) 104 v210_x86_init(s); 105 } 106 107 if ((ret = ff_get_buffer(avctx, pic, 0)) < 0) 108 return ret; 109 110 y = (uint16_t*)pic->data[0]; 111 u = (uint16_t*)pic->data[1]; 112 v = (uint16_t*)pic->data[2]; 113 pic->pict_type = AV_PICTURE_TYPE_I; 114 pic->key_frame = 1; 115 116 for (h = 0; h < avctx->height; h++) { 117 const uint32_t *src = (const uint32_t*)psrc; 118 uint32_t val; 119 120 w = (avctx->width / 6) * 6; 121 s->unpack_frame(src, y, u, v, w); 122 123 y += w; 124 u += w >> 1; 125 v += w >> 1; 126 src += (w << 1) / 3; 127 128 if (w < avctx->width - 1) { 129 READ_PIXELS(u, y, v); 130 131 val = av_le2ne32(*src++); 132 *y++ = val & 0x3FF; 133 if (w < avctx->width - 3) { 134 *u++ = (val >> 10) & 0x3FF; 135 *y++ = (val >> 20) & 0x3FF; 136 137 val = av_le2ne32(*src++); 138 *v++ = val & 0x3FF; 139 *y++ = (val >> 10) & 0x3FF; 140 } 141 } 142 143 psrc += stride; 144 y += pic->linesize[0] / 2 - avctx->width; 145 u += pic->linesize[1] / 2 - avctx->width / 2; 146 v += pic->linesize[2] / 2 - avctx->width / 2; 147 } 148 149 if (avctx->field_order > AV_FIELD_PROGRESSIVE) { 150 /* we have interlaced material flagged in container */ 151 pic->interlaced_frame = 1; 152 if (avctx->field_order == AV_FIELD_TT || avctx->field_order == AV_FIELD_TB) 153 pic->top_field_first = 1; 154 } 155 156 *got_frame = 1; 157 158 return avpkt->size; 159} 160 161#define V210DEC_FLAGS AV_OPT_FLAG_DECODING_PARAM | AV_OPT_FLAG_VIDEO_PARAM 162static const AVOption v210dec_options[] = { 163 {"custom_stride", "Custom V210 stride", offsetof(V210DecContext, custom_stride), FF_OPT_TYPE_INT, 164 {.i64 = 0}, INT_MIN, INT_MAX, V210DEC_FLAGS}, 165 {NULL} 166}; 167 168static const AVClass v210dec_class = { 169 "V210 Decoder", 170 av_default_item_name, 171 v210dec_options, 172 LIBAVUTIL_VERSION_INT, 173}; 174 175AVCodec ff_v210_decoder = { 176 .name = "v210", 177 .long_name = NULL_IF_CONFIG_SMALL("Uncompressed 4:2:2 10-bit"), 178 .type = AVMEDIA_TYPE_VIDEO, 179 .id = AV_CODEC_ID_V210, 180 .priv_data_size = sizeof(V210DecContext), 181 .init = decode_init, 182 .decode = decode_frame, 183 .capabilities = CODEC_CAP_DR1, 184 .priv_class = &v210dec_class, 185}; 186