FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
eatgq.c
Go to the documentation of this file.
1 /*
2  * Electronic Arts TGQ Video Decoder
3  * Copyright (c) 2007-2008 Peter Ross <pross@xvid.org>
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 St, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 /**
23  * @file
24  * Electronic Arts TGQ Video Decoder
25  * @author Peter Ross <pross@xvid.org>
26  *
27  * Technical details here:
28  * http://wiki.multimedia.cx/index.php?title=Electronic_Arts_TGQ
29  */
30 
31 #define BITSTREAM_READER_LE
32 #include "aandcttab.h"
33 #include "avcodec.h"
34 #include "bytestream.h"
35 #include "eaidct.h"
36 #include "get_bits.h"
37 #include "idctdsp.h"
38 #include "internal.h"
39 
40 typedef struct TgqContext {
42  int width, height;
44  int qtable[64];
45  DECLARE_ALIGNED(16, int16_t, block)[6][64];
47 } TgqContext;
48 
50 {
51  TgqContext *s = avctx->priv_data;
52  uint8_t idct_permutation[64];
53  s->avctx = avctx;
55  ff_init_scantable(idct_permutation, &s->scantable, ff_zigzag_direct);
56  avctx->framerate = (AVRational){ 15, 1 };
57  avctx->pix_fmt = AV_PIX_FMT_YUV420P;
58  return 0;
59 }
60 
61 static void tgq_decode_block(TgqContext *s, int16_t block[64], GetBitContext *gb)
62 {
64  int i, j, value;
65  block[0] = get_sbits(gb, 8) * s->qtable[0];
66  for (i = 1; i < 64;) {
67  switch (show_bits(gb, 3)) {
68  case 4:
69  block[perm[i++]] = 0;
70  case 0:
71  block[perm[i++]] = 0;
72  skip_bits(gb, 3);
73  break;
74  case 5:
75  case 1:
76  skip_bits(gb, 2);
77  value = get_bits(gb, 6);
78  for (j = 0; j < value; j++)
79  block[perm[i++]] = 0;
80  break;
81  case 6:
82  skip_bits(gb, 3);
83  block[perm[i]] = -s->qtable[perm[i]];
84  i++;
85  break;
86  case 2:
87  skip_bits(gb, 3);
88  block[perm[i]] = s->qtable[perm[i]];
89  i++;
90  break;
91  case 7: // 111b
92  case 3: // 011b
93  skip_bits(gb, 2);
94  if (show_bits(gb, 6) == 0x3F) {
95  skip_bits(gb, 6);
96  block[perm[i]] = get_sbits(gb, 8) * s->qtable[perm[i]];
97  } else {
98  block[perm[i]] = get_sbits(gb, 6) * s->qtable[perm[i]];
99  }
100  i++;
101  break;
102  }
103  }
104  block[0] += 128 << 4;
105 }
106 
107 static void tgq_idct_put_mb(TgqContext *s, int16_t (*block)[64], AVFrame *frame,
108  int mb_x, int mb_y)
109 {
110  int linesize = frame->linesize[0];
111  uint8_t *dest_y = frame->data[0] + (mb_y * 16 * linesize) + mb_x * 16;
112  uint8_t *dest_cb = frame->data[1] + (mb_y * 8 * frame->linesize[1]) + mb_x * 8;
113  uint8_t *dest_cr = frame->data[2] + (mb_y * 8 * frame->linesize[2]) + mb_x * 8;
114 
115  ff_ea_idct_put_c(dest_y , linesize, block[0]);
116  ff_ea_idct_put_c(dest_y + 8, linesize, block[1]);
117  ff_ea_idct_put_c(dest_y + 8 * linesize , linesize, block[2]);
118  ff_ea_idct_put_c(dest_y + 8 * linesize + 8, linesize, block[3]);
119  if (!(s->avctx->flags & AV_CODEC_FLAG_GRAY)) {
120  ff_ea_idct_put_c(dest_cb, frame->linesize[1], block[4]);
121  ff_ea_idct_put_c(dest_cr, frame->linesize[2], block[5]);
122  }
123 }
124 
125 static inline void tgq_dconly(TgqContext *s, unsigned char *dst,
126  int dst_stride, int dc)
127 {
128  int level = av_clip_uint8((dc*s->qtable[0] + 2056) >> 4);
129  int j;
130  for (j = 0; j < 8; j++)
131  memset(dst + j * dst_stride, level, 8);
132 }
133 
135  int mb_x, int mb_y, const int8_t *dc)
136 {
137  int linesize = frame->linesize[0];
138  uint8_t *dest_y = frame->data[0] + (mb_y * 16 * linesize) + mb_x * 16;
139  uint8_t *dest_cb = frame->data[1] + (mb_y * 8 * frame->linesize[1]) + mb_x * 8;
140  uint8_t *dest_cr = frame->data[2] + (mb_y * 8 * frame->linesize[2]) + mb_x * 8;
141  tgq_dconly(s, dest_y, linesize, dc[0]);
142  tgq_dconly(s, dest_y + 8, linesize, dc[1]);
143  tgq_dconly(s, dest_y + 8 * linesize, linesize, dc[2]);
144  tgq_dconly(s, dest_y + 8 * linesize + 8, linesize, dc[3]);
145  if (!(s->avctx->flags & AV_CODEC_FLAG_GRAY)) {
146  tgq_dconly(s, dest_cb, frame->linesize[1], dc[4]);
147  tgq_dconly(s, dest_cr, frame->linesize[2], dc[5]);
148  }
149 }
150 
151 static int tgq_decode_mb(TgqContext *s, AVFrame *frame, int mb_y, int mb_x)
152 {
153  int mode;
154  int i;
155  int8_t dc[6];
156 
157  mode = bytestream2_get_byte(&s->gb);
158  if (mode > 12) {
159  GetBitContext gb;
160  int ret = init_get_bits8(&gb, s->gb.buffer, FFMIN(bytestream2_get_bytes_left(&s->gb), mode));
161  if (ret < 0)
162  return ret;
163 
164  for (i = 0; i < 6; i++)
165  tgq_decode_block(s, s->block[i], &gb);
166  tgq_idct_put_mb(s, s->block, frame, mb_x, mb_y);
167  bytestream2_skip(&s->gb, mode);
168  } else {
169  if (mode == 3) {
170  memset(dc, bytestream2_get_byte(&s->gb), 4);
171  dc[4] = bytestream2_get_byte(&s->gb);
172  dc[5] = bytestream2_get_byte(&s->gb);
173  } else if (mode == 6) {
174  bytestream2_get_buffer(&s->gb, dc, 6);
175  } else if (mode == 12) {
176  for (i = 0; i < 6; i++) {
177  dc[i] = bytestream2_get_byte(&s->gb);
178  bytestream2_skip(&s->gb, 1);
179  }
180  } else {
181  av_log(s->avctx, AV_LOG_ERROR, "unsupported mb mode %i\n", mode);
182  return -1;
183  }
184  tgq_idct_put_mb_dconly(s, frame, mb_x, mb_y, dc);
185  }
186  return 0;
187 }
188 
190 {
191  int i, j;
192  const int a = (14 * (100 - quant)) / 100 + 1;
193  const int b = (11 * (100 - quant)) / 100 + 4;
194  for (j = 0; j < 8; j++)
195  for (i = 0; i < 8; i++)
196  s->qtable[j * 8 + i] = ((a * (j + i) / (7 + 7) + b) *
197  ff_inv_aanscales[j * 8 + i]) >> (14 - 4);
198 }
199 
201  void *data, int *got_frame,
202  AVPacket *avpkt)
203 {
204  const uint8_t *buf = avpkt->data;
205  int buf_size = avpkt->size;
206  TgqContext *s = avctx->priv_data;
207  AVFrame *frame = data;
208  int x, y, ret;
209  int big_endian;
210 
211  if (buf_size < 16) {
212  av_log(avctx, AV_LOG_WARNING, "truncated header\n");
213  return AVERROR_INVALIDDATA;
214  }
215  big_endian = AV_RL32(&buf[4]) > 0x000FFFFF;
216  bytestream2_init(&s->gb, buf + 8, buf_size - 8);
217  if (big_endian) {
218  s->width = bytestream2_get_be16u(&s->gb);
219  s->height = bytestream2_get_be16u(&s->gb);
220  } else {
221  s->width = bytestream2_get_le16u(&s->gb);
222  s->height = bytestream2_get_le16u(&s->gb);
223  }
224 
225  ret = ff_set_dimensions(s->avctx, s->width, s->height);
226  if (ret < 0)
227  return ret;
228 
229  tgq_calculate_qtable(s, bytestream2_get_byteu(&s->gb));
230  bytestream2_skip(&s->gb, 3);
231 
232  if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
233  return ret;
234  frame->key_frame = 1;
235  frame->pict_type = AV_PICTURE_TYPE_I;
236 
237  for (y = 0; y < FFALIGN(avctx->height, 16) >> 4; y++)
238  for (x = 0; x < FFALIGN(avctx->width, 16) >> 4; x++)
239  if (tgq_decode_mb(s, frame, y, x) < 0)
240  return AVERROR_INVALIDDATA;
241 
242  *got_frame = 1;
243 
244  return avpkt->size;
245 }
246 
248  .name = "eatgq",
249  .long_name = NULL_IF_CONFIG_SMALL("Electronic Arts TGQ video"),
250  .type = AVMEDIA_TYPE_VIDEO,
251  .id = AV_CODEC_ID_TGQ,
252  .priv_data_size = sizeof(TgqContext),
255  .capabilities = AV_CODEC_CAP_DR1,
256 };
AVRational framerate
Definition: avcodec.h:3375
const char * s
Definition: avisynth_c.h:768
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
int height
Definition: eatgq.c:42
This structure describes decoded (raw) audio or video data.
Definition: frame.h:184
ptrdiff_t const GLvoid * data
Definition: opengl_enc.c:101
static void tgq_idct_put_mb(TgqContext *s, int16_t(*block)[64], AVFrame *frame, int mb_x, int mb_y)
Definition: eatgq.c:107
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:247
#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
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
Scantable.
Definition: idctdsp.h:29
int size
Definition: avcodec.h:1602
const char * b
Definition: vf_curves.c:113
static void tgq_idct_put_mb_dconly(TgqContext *s, AVFrame *frame, int mb_x, int mb_y, const int8_t *dc)
Definition: eatgq.c:134
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1904
static av_always_inline void bytestream2_init(GetByteContext *g, const uint8_t *buf, int buf_size)
Definition: bytestream.h:133
uint8_t permutated[64]
Definition: idctdsp.h:31
AVCodec.
Definition: avcodec.h:3600
static int get_sbits(GetBitContext *s, int n)
Definition: get_bits.h:232
static av_cold int tgq_decode_init(AVCodecContext *avctx)
Definition: eatgq.c:49
static int16_t block[64]
Definition: dct.c:113
uint8_t
#define av_cold
Definition: attributes.h:82
mode
Definition: f_perms.c:27
static void tgq_decode_block(TgqContext *s, int16_t block[64], GetBitContext *gb)
Definition: eatgq.c:61
static AVFrame * frame
GetByteContext gb
Definition: eatgq.c:46
#define DECLARE_ALIGNED(n, t, v)
Declare a variable that is aligned in memory.
Definition: mem.h:101
uint8_t * data
Definition: avcodec.h:1601
const uint8_t * buffer
Definition: bytestream.h:34
bitstream reader API header.
#define AV_CODEC_FLAG_GRAY
Only decode/encode grayscale.
Definition: avcodec.h:861
#define FFALIGN(x, a)
Definition: macros.h:48
#define av_log(a,...)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
static av_always_inline void bytestream2_skip(GetByteContext *g, unsigned int size)
Definition: bytestream.h:164
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:176
static av_always_inline unsigned int bytestream2_get_buffer(GetByteContext *g, uint8_t *dst, unsigned int size)
Definition: bytestream.h:263
static av_always_inline unsigned int bytestream2_get_bytes_left(GetByteContext *g)
Definition: bytestream.h:154
int flags
AV_CODEC_FLAG_*.
Definition: avcodec.h:1771
const char * name
Name of the codec implementation.
Definition: avcodec.h:3607
static void tgq_dconly(TgqContext *s, unsigned char *dst, int dst_stride, int dc)
Definition: eatgq.c:125
ScanTable scantable
Definition: eatgq.c:43
enum AVPictureType pict_type
Picture type of the frame.
Definition: frame.h:258
#define FFMIN(a, b)
Definition: common.h:96
av_cold void ff_init_scantable_permutation(uint8_t *idct_permutation, enum idct_permutation_type perm_type)
Definition: idctdsp.c:50
int width
picture width / height.
Definition: avcodec.h:1863
GLsizei GLboolean const GLfloat * value
Definition: opengl_enc.c:109
static unsigned int show_bits(GetBitContext *s, int n)
Show 1-25 bits.
Definition: get_bits.h:282
perm
Definition: f_perms.c:74
static int tgq_decode_mb(TgqContext *s, AVFrame *frame, int mb_y, int mb_x)
Definition: eatgq.c:151
AVCodec ff_eatgq_decoder
Definition: eatgq.c:247
Libavcodec external API header.
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:215
const uint16_t ff_inv_aanscales[64]
Definition: aandcttab.c:38
static int init_get_bits8(GetBitContext *s, const uint8_t *buffer, int byte_size)
Initialize GetBitContext.
Definition: get_bits.h:437
main external API structure.
Definition: avcodec.h:1676
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: utils.c:947
void * buf
Definition: avisynth_c.h:690
static void skip_bits(GetBitContext *s, int n)
Definition: get_bits.h:292
Rational number (pair of numerator and denominator).
Definition: rational.h:58
const uint8_t ff_zigzag_direct[64]
Definition: mathtables.c:98
int qtable[64]
Definition: eatgq.c:44
const uint8_t * quant
AAN (Arai, Agui and Nakajima) (I)DCT tables.
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:198
int width
Definition: eatgq.c:42
uint8_t level
Definition: svq3.c:207
void ff_ea_idct_put_c(uint8_t *dest, int linesize, int16_t *block)
Definition: eaidct.c:80
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:62
common internal api header.
void * priv_data
Definition: avcodec.h:1718
av_cold void ff_init_scantable(uint8_t *permutation, ScanTable *st, const uint8_t *src_scantable)
Definition: idctdsp.c:29
int16_t block[6][64]
Definition: eatgq.c:45
int key_frame
1 -> keyframe, 0-> not
Definition: frame.h:253
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(constint16_t *) pi >>8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(constint32_t *) pi >>24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(constfloat *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(constfloat *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(constfloat *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(constdouble *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(constdouble *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(constdouble *) pi *(1U<< 31))))#defineSET_CONV_FUNC_GROUP(ofmt, ifmt) staticvoidset_generic_function(AudioConvert *ac){}voidff_audio_convert_free(AudioConvert **ac){if(!*ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);}AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enumAVSampleFormatout_fmt, enumAVSampleFormatin_fmt, intchannels, intsample_rate, intapply_map){AudioConvert *ac;intin_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) returnNULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method!=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt)>2){ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc){av_free(ac);returnNULL;}returnac;}in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar){ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar?ac->channels:1;}elseif(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;elseac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);returnac;}intff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in){intuse_generic=1;intlen=in->nb_samples;intp;if(ac->dc){av_log(ac->avr, AV_LOG_TRACE,"%dsamples-audio_convert:%sto%s(dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));returnff_convert_dither(ac-> dc
static int decode(AVCodecContext *avctx, AVFrame *frame, int *got_frame, AVPacket *pkt)
Definition: ffmpeg.c:2035
static void tgq_calculate_qtable(TgqContext *s, int quant)
Definition: eatgq.c:189
static int tgq_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: eatgq.c:200
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:87
This structure stores compressed data.
Definition: avcodec.h:1578
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:959
AVCodecContext * avctx
Definition: eatgq.c:41