annotate eatgv.c @ 8006:c7c1e85d14bc libavcodec

Rename variables to clarify the channel coupling element and corresponding target channel element. Patch by Alex Converse (alex converse gmail com)
author superdump
date Mon, 06 Oct 2008 16:22:11 +0000
parents 212e72dbb838
children 573d2361d2ec
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
7510
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
1 /*
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
2 * Electronic Arts TGV Video Decoder
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
3 * Copyright (c) 2007-2008 Peter Ross
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
4 *
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
5 * This file is part of FFmpeg.
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
6 *
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
7 * FFmpeg is free software; you can redistribute it and/or
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
8 * modify it under the terms of the GNU Lesser General Public
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
9 * License as published by the Free Software Foundation; either
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
10 * version 2.1 of the License, or (at your option) any later version.
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
11 *
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
12 * FFmpeg is distributed in the hope that it will be useful,
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
15 * Lesser General Public License for more details.
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
16 *
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
17 * You should have received a copy of the GNU Lesser General Public
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
18 * License along with FFmpeg; if not, write to the Free Software
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
20 */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
21
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
22 /**
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
23 * @file eatgv.c
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
24 * Electronic Arts TGV Video Decoder
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
25 * by Peter Ross (suxen_drol at hotmail dot com)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
26 *
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
27 * Technical details here:
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
28 * http://wiki.multimedia.cx/index.php?title=Electronic_Arts_TGV
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
29 */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
30
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
31 #include "avcodec.h"
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
32 #define ALT_BITSTREAM_READER_LE
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
33 #include "bitstream.h"
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
34 #include <libavutil/lzo.h>
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
35
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
36 #define EA_PREAMBLE_SIZE 8
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
37 #define kVGT_TAG MKTAG('k', 'V', 'G', 'T')
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
38
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
39 typedef struct TgvContext {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
40 AVCodecContext *avctx;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
41 AVFrame frame;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
42 AVFrame last_frame;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
43 int width,height;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
44 unsigned int palette[AVPALETTE_COUNT];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
45
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
46 int (*mv_codebook)[2];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
47 unsigned char (*block_codebook)[16];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
48 int num_mvs; ///< current length of mv_codebook
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
49 int num_blocks_packed; ///< current length of block_codebook
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
50 } TgvContext;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
51
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
52 static av_cold int tgv_decode_init(AVCodecContext *avctx){
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
53 TgvContext *s = avctx->priv_data;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
54 s->avctx = avctx;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
55 avctx->time_base = (AVRational){1, 15};
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
56 avctx->pix_fmt = PIX_FMT_PAL8;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
57 return 0;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
58 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
59
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
60 /**
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
61 * Unpack buffer
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
62 * @return 0 on success, -1 on critical buffer underflow
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
63 */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
64 static int unpack(const uint8_t *src, const uint8_t *src_end, unsigned char *dst, int width, int height) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
65 unsigned char *dst_end = dst + width*height;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
66 int size,size1,size2,offset,run;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
67 unsigned char *dst_start = dst;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
68
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
69 if (src[0] & 0x01)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
70 src += 5;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
71 else
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
72 src += 2;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
73
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
74 if (src+3>src_end)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
75 return -1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
76 size = AV_RB24(src);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
77 src += 3;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
78
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
79 while(size>0 && src<src_end) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
80
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
81 /* determine size1 and size2 */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
82 size1 = (src[0] & 3);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
83 if ( src[0] & 0x80 ) { // 1
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
84 if (src[0] & 0x40 ) { // 11
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
85 if ( src[0] & 0x20 ) { // 111
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
86 if ( src[0] < 0xFC ) // !(111111)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
87 size1 = (((src[0] & 31) + 1) << 2);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
88 src++;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
89 size2 = 0;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
90 } else { // 110
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
91 offset = ((src[0] & 0x10) << 12) + AV_RB16(&src[1]) + 1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
92 size2 = ((src[0] & 0xC) << 6) + src[3] + 5;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
93 src += 4;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
94 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
95 } else { // 10
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
96 size1 = ( ( src[1] & 0xC0) >> 6 );
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
97 offset = (AV_RB16(&src[1]) & 0x3FFF) + 1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
98 size2 = (src[0] & 0x3F) + 4;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
99 src += 3;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
100 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
101 } else { // 0
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
102 offset = ((src[0] & 0x60) << 3) + src[1] + 1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
103 size2 = ((src[0] & 0x1C) >> 2) + 3;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
104 src += 2;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
105 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
106
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
107
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
108 /* fetch strip from src */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
109 if (size1>src_end-src)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
110 break;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
111
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
112 if (size1>0) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
113 size -= size1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
114 run = FFMIN(size1, dst_end-dst);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
115 memcpy(dst, src, run);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
116 dst += run;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
117 src += run;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
118 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
119
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
120 if (size2>0) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
121 if (dst-dst_start<offset)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
122 return 0;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
123 size -= size2;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
124 run = FFMIN(size2, dst_end-dst);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
125 av_memcpy_backptr(dst, offset, run);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
126 dst += run;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
127 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
128 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
129
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
130 return 0;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
131 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
132
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
133 /**
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
134 * Decode inter-frame
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
135 * @return 0 on success, -1 on critical buffer underflow
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
136 */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
137 static int tgv_decode_inter(TgvContext * s, const uint8_t *buf, const uint8_t *buf_end){
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
138 unsigned char *frame0_end = s->last_frame.data[0] + s->avctx->width*s->last_frame.linesize[0];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
139 int num_mvs;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
140 int num_blocks_raw;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
141 int num_blocks_packed;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
142 int vector_bits;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
143 int i,j,x,y;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
144 GetBitContext gb;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
145 int mvbits;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
146 const unsigned char *blocks_raw;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
147
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
148 if(buf+12>buf_end)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
149 return -1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
150
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
151 num_mvs = AV_RL16(&buf[0]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
152 num_blocks_raw = AV_RL16(&buf[2]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
153 num_blocks_packed = AV_RL16(&buf[4]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
154 vector_bits = AV_RL16(&buf[6]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
155 buf += 12;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
156
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
157 /* allocate codebook buffers as neccessary */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
158 if (num_mvs > s->num_mvs) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
159 s->mv_codebook = av_realloc(s->mv_codebook, num_mvs*2*sizeof(int));
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
160 s->num_mvs = num_mvs;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
161 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
162
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
163 if (num_blocks_packed > s->num_blocks_packed) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
164 s->block_codebook = av_realloc(s->block_codebook, num_blocks_packed*16*sizeof(unsigned char));
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
165 s->num_blocks_packed = num_blocks_packed;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
166 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
167
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
168 /* read motion vectors */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
169 mvbits = (num_mvs*2*10+31) & ~31;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
170
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
171 if (buf+(mvbits>>3)+16*num_blocks_raw+8*num_blocks_packed>buf_end)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
172 return -1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
173
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
174 init_get_bits(&gb, buf, mvbits);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
175 for (i=0; i<num_mvs; i++) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
176 s->mv_codebook[i][0] = get_sbits(&gb, 10);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
177 s->mv_codebook[i][1] = get_sbits(&gb, 10);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
178 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
179 buf += mvbits>>3;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
180
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
181 /* note ptr to uncompressed blocks */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
182 blocks_raw = buf;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
183 buf += num_blocks_raw*16;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
184
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
185 /* read compressed blocks */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
186 init_get_bits(&gb, buf, (buf_end-buf)<<3);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
187 for (i=0; i<num_blocks_packed; i++) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
188 int tmp[4];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
189 for(j=0; j<4; j++)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
190 tmp[j] = get_bits(&gb, 8);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
191 for(j=0; j<16; j++)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
192 s->block_codebook[i][15-j] = tmp[get_bits(&gb, 2)];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
193 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
194
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
195 /* read vectors and build frame */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
196 for(y=0; y<s->avctx->height/4; y++)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
197 for(x=0; x<s->avctx->width/4; x++) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
198 unsigned int vector = get_bits(&gb, vector_bits);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
199 const unsigned char *src;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
200 int src_stride;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
201
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
202 if (vector < num_mvs) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
203 src = s->last_frame.data[0] +
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
204 (y*4 + s->mv_codebook[vector][1])*s->last_frame.linesize[0] +
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
205 x*4 + s->mv_codebook[vector][0];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
206 src_stride = s->last_frame.linesize[0];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
207 if (src+3*src_stride+3>=frame0_end)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
208 continue;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
209 }else{
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
210 int offset = vector - num_mvs;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
211 if (offset<num_blocks_raw)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
212 src = blocks_raw + 16*offset;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
213 else if (offset-num_blocks_raw<num_blocks_packed)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
214 src = s->block_codebook[offset-num_blocks_raw];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
215 else
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
216 continue;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
217 src_stride = 4;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
218 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
219
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
220 for(j=0; j<4; j++)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
221 for(i=0; i<4; i++)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
222 s->frame.data[0][ (y*4+j)*s->frame.linesize[0] + (x*4+i) ] =
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
223 src[j*src_stride + i];
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
224 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
225
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
226 return 0;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
227 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
228
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
229 /** release AVFrame buffers if allocated */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
230 static void cond_release_buffer(AVFrame *pic)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
231 {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
232 if (pic->data[0]) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
233 av_freep(&pic->data[0]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
234 av_free(pic->data[1]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
235 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
236 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
237
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
238 static int tgv_decode_frame(AVCodecContext *avctx,
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
239 void *data, int *data_size,
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
240 const uint8_t *buf, int buf_size)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
241 {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
242 TgvContext *s = avctx->priv_data;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
243 const uint8_t *buf_end = buf + buf_size;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
244 int chunk_type;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
245
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
246 chunk_type = AV_RL32(&buf[0]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
247 buf += EA_PREAMBLE_SIZE;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
248
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
249 if (chunk_type==kVGT_TAG) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
250 int pal_count, i;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
251 if(buf+12>buf_end) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
252 av_log(avctx, AV_LOG_WARNING, "truncated header\n");
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
253 return -1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
254 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
255
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
256 s->width = AV_RL16(&buf[0]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
257 s->height = AV_RL16(&buf[2]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
258 if (s->avctx->width!=s->width || s->avctx->height!=s->height) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
259 avcodec_set_dimensions(s->avctx, s->width, s->height);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
260 cond_release_buffer(&s->frame);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
261 cond_release_buffer(&s->last_frame);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
262 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
263
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
264 pal_count = AV_RL16(&buf[6]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
265 buf += 12;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
266 for(i=0; i<pal_count && i<AVPALETTE_COUNT && buf+2<buf_end; i++) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
267 s->palette[i] = AV_RB24(buf);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
268 buf += 3;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
269 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
270 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
271
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
272 if (avcodec_check_dimensions(avctx, s->width, s->height))
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
273 return -1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
274
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
275 /* shuffle */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
276 FFSWAP(AVFrame, s->frame, s->last_frame);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
277 if (!s->frame.data[0]) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
278 s->frame.reference = 1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
279 s->frame.buffer_hints = FF_BUFFER_HINTS_VALID;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
280 s->frame.linesize[0] = s->width;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
281
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
282 /* allocate additional 12 bytes to accomodate av_memcpy_backptr() OUTBUF_PADDED optimisation */
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
283 s->frame.data[0] = av_malloc(s->width*s->height + 12);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
284 if (!s->frame.data[0])
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
285 return AVERROR_NOMEM;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
286 s->frame.data[1] = av_malloc(AVPALETTE_SIZE);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
287 if (!s->frame.data[1]) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
288 av_freep(&s->frame.data[0]);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
289 return AVERROR_NOMEM;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
290 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
291 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
292 memcpy(s->frame.data[1], s->palette, AVPALETTE_SIZE);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
293
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
294 if(chunk_type==kVGT_TAG) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
295 s->frame.key_frame = 1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
296 s->frame.pict_type = FF_I_TYPE;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
297 if (unpack(buf, buf_end, s->frame.data[0], s->avctx->width, s->avctx->height)<0) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
298 av_log(avctx, AV_LOG_WARNING, "truncated intra frame\n");
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
299 return -1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
300 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
301 }else{
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
302 if (!s->last_frame.data[0]) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
303 av_log(avctx, AV_LOG_WARNING, "inter frame without corresponding intra frame\n");
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
304 return buf_size;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
305 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
306 s->frame.key_frame = 0;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
307 s->frame.pict_type = FF_P_TYPE;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
308 if (tgv_decode_inter(s, buf, buf_end)<0) {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
309 av_log(avctx, AV_LOG_WARNING, "truncated inter frame\n");
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
310 return -1;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
311 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
312 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
313
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
314 *data_size = sizeof(AVFrame);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
315 *(AVFrame*)data = s->frame;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
316
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
317 return buf_size;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
318 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
319
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
320 static av_cold int tgv_decode_end(AVCodecContext *avctx)
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
321 {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
322 TgvContext *s = avctx->priv_data;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
323 cond_release_buffer(&s->frame);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
324 cond_release_buffer(&s->last_frame);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
325 av_free(s->mv_codebook);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
326 av_free(s->block_codebook);
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
327 return 0;
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
328 }
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
329
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
330 AVCodec eatgv_decoder = {
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
331 "eatgv",
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
332 CODEC_TYPE_VIDEO,
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
333 CODEC_ID_TGV,
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
334 sizeof(TgvContext),
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
335 tgv_decode_init,
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
336 NULL,
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
337 tgv_decode_end,
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
338 tgv_decode_frame,
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
339 .long_name = NULL_IF_CONFIG_SMALL("Electronic Arts TGV Video"),
212e72dbb838 Electronic Arts TGV decoder
pross
parents:
diff changeset
340 };