Commit | Line | Data |
---|---|---|
a8a15e9d MM |
1 | /* |
2 | * Winnov WNV1 codec | |
3 | * Copyright (c) 2005 Konstantin Shishkov | |
4 | * | |
b78e7197 DB |
5 | * This file is part of FFmpeg. |
6 | * | |
7 | * FFmpeg is free software; you can redistribute it and/or | |
a8a15e9d MM |
8 | * modify it under the terms of the GNU Lesser General Public |
9 | * License as published by the Free Software Foundation; either | |
b78e7197 | 10 | * version 2.1 of the License, or (at your option) any later version. |
a8a15e9d | 11 | * |
b78e7197 | 12 | * FFmpeg is distributed in the hope that it will be useful, |
a8a15e9d MM |
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 | |
b78e7197 | 18 | * License along with FFmpeg; if not, write to the Free Software |
5509bffa | 19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
a8a15e9d | 20 | */ |
115329f1 | 21 | |
a8a15e9d | 22 | /** |
bad5537e | 23 | * @file libavcodec/wnv1.c |
a8a15e9d MM |
24 | * Winnov WNV1 codec. |
25 | */ | |
115329f1 | 26 | |
a8a15e9d | 27 | #include "avcodec.h" |
9106a698 | 28 | #include "get_bits.h" |
a8a15e9d MM |
29 | |
30 | ||
31 | typedef struct WNV1Context{ | |
32 | AVCodecContext *avctx; | |
33 | AVFrame pic; | |
34 | ||
35 | int shift; | |
083b80b5 | 36 | GetBitContext gb; |
a8a15e9d MM |
37 | } WNV1Context; |
38 | ||
cf2baeb3 | 39 | static const uint16_t code_tab[16][2]={ |
083b80b5 | 40 | {0x1FD,9}, {0xFD,8}, {0x7D,7}, {0x3D,6}, {0x1D,5}, {0x0D,4}, {0x005,3}, |
115329f1 | 41 | {0x000,1}, |
083b80b5 MN |
42 | {0x004,3}, {0x0C,4}, {0x1C,5}, {0x3C,6}, {0x7C,7}, {0xFC,8}, {0x1FC,9}, {0xFF,8} |
43 | }; | |
44 | ||
45 | #define CODE_VLC_BITS 9 | |
46 | static VLC code_vlc; | |
47 | ||
a8a15e9d MM |
48 | /* returns modified base_value */ |
49 | static inline int wnv1_get_code(WNV1Context *w, int base_value) | |
50 | { | |
083b80b5 | 51 | int v = get_vlc2(&w->gb, code_vlc.table, CODE_VLC_BITS, 1); |
a8a15e9d | 52 | |
083b80b5 MN |
53 | if(v==15) |
54 | return ff_reverse[ get_bits(&w->gb, 8 - w->shift) ]; | |
55 | else | |
56 | return base_value + ((v - 7)<<w->shift); | |
a8a15e9d MM |
57 | } |
58 | ||
115329f1 | 59 | static int decode_frame(AVCodecContext *avctx, |
a8a15e9d | 60 | void *data, int *data_size, |
7a00bbad | 61 | AVPacket *avpkt) |
a8a15e9d | 62 | { |
7a00bbad TB |
63 | const uint8_t *buf = avpkt->data; |
64 | int buf_size = avpkt->size; | |
a8a15e9d MM |
65 | WNV1Context * const l = avctx->priv_data; |
66 | AVFrame * const p= (AVFrame*)&l->pic; | |
67 | unsigned char *Y,*U,*V; | |
68 | int i, j; | |
69 | int prev_y = 0, prev_u = 0, prev_v = 0; | |
b7093975 KS |
70 | uint8_t *rbuf; |
71 | ||
72 | rbuf = av_malloc(buf_size + FF_INPUT_BUFFER_PADDING_SIZE); | |
73 | if(!rbuf){ | |
74 | av_log(avctx, AV_LOG_ERROR, "Cannot allocate temporary buffer\n"); | |
75 | return -1; | |
76 | } | |
a8a15e9d MM |
77 | |
78 | if(p->data[0]) | |
79 | avctx->release_buffer(avctx, p); | |
80 | ||
81 | p->reference = 0; | |
82 | if(avctx->get_buffer(avctx, p) < 0){ | |
83 | av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); | |
b7093975 | 84 | av_free(rbuf); |
a8a15e9d MM |
85 | return -1; |
86 | } | |
87 | p->key_frame = 1; | |
083b80b5 MN |
88 | |
89 | for(i=8; i<buf_size; i++) | |
b7093975 KS |
90 | rbuf[i]= ff_reverse[ buf[i] ]; |
91 | init_get_bits(&l->gb, rbuf+8, (buf_size-8)*8); | |
083b80b5 | 92 | |
a8a15e9d MM |
93 | if (buf[2] >> 4 == 6) |
94 | l->shift = 2; | |
95 | else { | |
96 | l->shift = 8 - (buf[2] >> 4); | |
97 | if (l->shift > 4) { | |
98 | av_log(avctx, AV_LOG_ERROR, "Unknown WNV1 frame header value %i, please upload file for study\n", buf[2] >> 4); | |
99 | l->shift = 4; | |
100 | } | |
101 | if (l->shift < 1) { | |
102 | av_log(avctx, AV_LOG_ERROR, "Unknown WNV1 frame header value %i, please upload file for study\n", buf[2] >> 4); | |
103 | l->shift = 1; | |
104 | } | |
105 | } | |
115329f1 | 106 | |
a8a15e9d MM |
107 | Y = p->data[0]; |
108 | U = p->data[1]; | |
109 | V = p->data[2]; | |
110 | for (j = 0; j < avctx->height; j++) { | |
111 | for (i = 0; i < avctx->width / 2; i++) { | |
112 | Y[i * 2] = wnv1_get_code(l, prev_y); | |
113 | prev_u = U[i] = wnv1_get_code(l, prev_u); | |
114 | prev_y = Y[(i * 2) + 1] = wnv1_get_code(l, Y[i * 2]); | |
115 | prev_v = V[i] = wnv1_get_code(l, prev_v); | |
116 | } | |
117 | Y += p->linesize[0]; | |
118 | U += p->linesize[1]; | |
119 | V += p->linesize[2]; | |
120 | } | |
121 | ||
115329f1 | 122 | |
a8a15e9d MM |
123 | *data_size = sizeof(AVFrame); |
124 | *(AVFrame*)data = l->pic; | |
b7093975 | 125 | av_free(rbuf); |
115329f1 | 126 | |
a8a15e9d MM |
127 | return buf_size; |
128 | } | |
129 | ||
98a6fff9 | 130 | static av_cold int decode_init(AVCodecContext *avctx){ |
a8a15e9d | 131 | WNV1Context * const l = avctx->priv_data; |
bd4110f9 | 132 | static VLC_TYPE code_table[1 << CODE_VLC_BITS][2]; |
a8a15e9d MM |
133 | |
134 | l->avctx = avctx; | |
135 | avctx->pix_fmt = PIX_FMT_YUV422P; | |
136 | ||
bd4110f9 KS |
137 | code_vlc.table = code_table; |
138 | code_vlc.table_allocated = 1 << CODE_VLC_BITS; | |
c994dae1 KS |
139 | init_vlc(&code_vlc, CODE_VLC_BITS, 16, |
140 | &code_tab[0][1], 4, 2, | |
141 | &code_tab[0][0], 4, 2, INIT_VLC_USE_NEW_STATIC); | |
083b80b5 | 142 | |
a8a15e9d MM |
143 | return 0; |
144 | } | |
145 | ||
6d924b5a KS |
146 | static av_cold int decode_end(AVCodecContext *avctx){ |
147 | WNV1Context * const l = avctx->priv_data; | |
148 | AVFrame *pic = &l->pic; | |
149 | ||
150 | if (pic->data[0]) | |
151 | avctx->release_buffer(avctx, pic); | |
6d924b5a KS |
152 | |
153 | return 0; | |
154 | } | |
155 | ||
a8a15e9d MM |
156 | AVCodec wnv1_decoder = { |
157 | "wnv1", | |
158 | CODEC_TYPE_VIDEO, | |
159 | CODEC_ID_WNV1, | |
160 | sizeof(WNV1Context), | |
161 | decode_init, | |
162 | NULL, | |
6d924b5a | 163 | decode_end, |
a8a15e9d MM |
164 | decode_frame, |
165 | CODEC_CAP_DR1, | |
fe4bf374 | 166 | .long_name = NULL_IF_CONFIG_SMALL("Winnov WNV1"), |
a8a15e9d | 167 | }; |