FFmpeg
eatqi.c
Go to the documentation of this file.
1 /*
2  * Electronic Arts TQI Video Decoder
3  * Copyright (c) 2007-2009 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 TQI Video Decoder
25  * @author Peter Ross <pross@xvid.org>
26  * @see http://wiki.multimedia.cx/index.php?title=Electronic_Arts_TQI
27  */
28 
29 #include "avcodec.h"
30 #include "blockdsp.h"
31 #include "bswapdsp.h"
32 #include "get_bits.h"
33 #include "aandcttab.h"
34 #include "eaidct.h"
35 #include "idctdsp.h"
36 #include "internal.h"
37 #include "mpeg12.h"
38 
39 typedef struct TqiContext {
46 
48  unsigned int bitstream_buf_size;
49 
50  int mb_x, mb_y;
51  uint16_t intra_matrix[64];
52  int last_dc[3];
53 
54  DECLARE_ALIGNED(32, int16_t, block)[6][64];
55 } TqiContext;
56 
58 {
59  TqiContext *t = avctx->priv_data;
60 
61  ff_blockdsp_init(&t->bdsp, avctx);
63  ff_idctdsp_init(&t->idsp, avctx);
66 
67  avctx->framerate = (AVRational){ 15, 1 };
68  avctx->pix_fmt = AV_PIX_FMT_YUV420P;
70  return 0;
71 }
72 
73 static int tqi_decode_mb(TqiContext *t, int16_t (*block)[64])
74 {
75  int n;
76 
77  t->bdsp.clear_blocks(block[0]);
78  for (n = 0; n < 6; n++) {
80  t->intra_matrix,
82  t->last_dc, block[n], n, 1);
83  if (ret < 0) {
84  av_log(t->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n",
85  t->mb_x, t->mb_y);
86  return ret;
87  }
88  }
89 
90  return 0;
91 }
92 
94  int16_t (*block)[64])
95 {
96  TqiContext *t = avctx->priv_data;
97  ptrdiff_t linesize = frame->linesize[0];
98  uint8_t *dest_y = frame->data[0] + t->mb_y * 16 * linesize + t->mb_x * 16;
99  uint8_t *dest_cb = frame->data[1] + t->mb_y * 8 * frame->linesize[1] + t->mb_x * 8;
100  uint8_t *dest_cr = frame->data[2] + t->mb_y * 8 * frame->linesize[2] + t->mb_x * 8;
101 
102  ff_ea_idct_put_c(dest_y , linesize, block[0]);
103  ff_ea_idct_put_c(dest_y + 8, linesize, block[1]);
104  ff_ea_idct_put_c(dest_y + 8*linesize , linesize, block[2]);
105  ff_ea_idct_put_c(dest_y + 8*linesize + 8, linesize, block[3]);
106 
107  if (!(avctx->flags & AV_CODEC_FLAG_GRAY)) {
108  ff_ea_idct_put_c(dest_cb, frame->linesize[1], block[4]);
109  ff_ea_idct_put_c(dest_cr, frame->linesize[2], block[5]);
110  }
111 }
112 
114 {
115  const int64_t qscale = (215 - 2*quant)*5;
116  int i;
117 
119  for(i=1; i<64; i++)
120  t->intra_matrix[i] = (ff_inv_aanscales[i] * ff_mpeg1_default_intra_matrix[i] * qscale + 32) >> 14;
121 }
122 
124  void *data, int *got_frame,
125  AVPacket *avpkt)
126 {
127  const uint8_t *buf = avpkt->data;
128  int buf_size = avpkt->size;
129  const uint8_t *buf_end = buf+buf_size;
130  TqiContext *t = avctx->priv_data;
131  AVFrame *frame = data;
132  int ret, w, h;
133 
134  t->avctx = avctx;
135 
136  w = AV_RL16(&buf[0]);
137  h = AV_RL16(&buf[2]);
138  tqi_calculate_qtable(t, buf[4]);
139  buf += 8;
140 
141  ret = ff_set_dimensions(avctx, w, h);
142  if (ret < 0)
143  return ret;
144 
145  if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
146  return ret;
147 
149  buf_end - buf);
150  if (!t->bitstream_buf)
151  return AVERROR(ENOMEM);
152  t->bsdsp.bswap_buf(t->bitstream_buf, (const uint32_t *) buf,
153  (buf_end - buf) / 4);
154  init_get_bits(&t->gb, t->bitstream_buf, 8 * (buf_end - buf));
155 
156  t->last_dc[0] =
157  t->last_dc[1] =
158  t->last_dc[2] = 0;
159  for (t->mb_y = 0; t->mb_y < (h + 15) / 16; t->mb_y++) {
160  for (t->mb_x = 0; t->mb_x < (w + 15) / 16; t->mb_x++) {
161  if (tqi_decode_mb(t, t->block) < 0)
162  goto end;
163  tqi_idct_put(avctx, frame, t->block);
164  }
165  }
166  end:
167 
168  *got_frame = 1;
169  return buf_size;
170 }
171 
173 {
174  TqiContext *t = avctx->priv_data;
175  av_freep(&t->bitstream_buf);
176  return 0;
177 }
178 
180  .name = "eatqi",
181  .long_name = NULL_IF_CONFIG_SMALL("Electronic Arts TQI Video"),
182  .type = AVMEDIA_TYPE_VIDEO,
183  .id = AV_CODEC_ID_TQI,
184  .priv_data_size = sizeof(TqiContext),
186  .close = tqi_decode_end,
188  .capabilities = AV_CODEC_CAP_DR1,
189 };
AVRational framerate
Definition: avcodec.h:3101
This structure describes decoded (raw) audio or video data.
Definition: frame.h:268
AVCodec ff_eatqi_decoder
Definition: eatqi.c:179
ptrdiff_t const GLvoid * data
Definition: opengl_enc.c:100
static av_cold int tqi_decode_init(AVCodecContext *avctx)
Definition: eatqi.c:57
av_cold void ff_mpeg12_init_vlcs(void)
Definition: mpeg12.c:137
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:104
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
AVCodecContext * avctx
Definition: eatqi.c:40
Scantable.
Definition: idctdsp.h:31
int size
Definition: avcodec.h:1478
av_cold void ff_blockdsp_init(BlockDSPContext *c, AVCodecContext *avctx)
Definition: blockdsp.c:60
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1775
static void tqi_idct_put(AVCodecContext *avctx, AVFrame *frame, int16_t(*block)[64])
Definition: eatqi.c:93
void av_fast_padded_malloc(void *ptr, unsigned int *size, size_t min_size)
Same behaviour av_fast_malloc but the buffer has additional AV_INPUT_BUFFER_PADDING_SIZE at the end w...
Definition: utils.c:70
uint8_t permutated[64]
Definition: idctdsp.h:33
static int tqi_decode_mb(TqiContext *t, int16_t(*block)[64])
Definition: eatqi.c:73
AVCodec.
Definition: avcodec.h:3477
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_RL16
Definition: bytestream.h:87
static void decode(AVCodecContext *dec_ctx, AVPacket *pkt, AVFrame *frame, FILE *outfile)
Definition: decode_audio.c:42
uint8_t
static void tqi_calculate_qtable(TqiContext *t, int quant)
Definition: eatqi.c:113
#define av_cold
Definition: attributes.h:82
void(* bswap_buf)(uint32_t *dst, const uint32_t *src, int w)
Definition: bswapdsp.h:25
static av_cold int end(AVCodecContext *avctx)
Definition: avrndec.c:90
static av_cold int tqi_decode_end(AVCodecContext *avctx)
Definition: eatqi.c:172
#define DECLARE_ALIGNED(n, t, v)
Declare a variable that is aligned in memory.
Definition: mem.h:112
uint8_t * data
Definition: avcodec.h:1477
bitstream reader API header.
void ff_ea_idct_put_c(uint8_t *dest, ptrdiff_t linesize, int16_t *block)
Definition: eaidct.c:80
#define AV_CODEC_FLAG_GRAY
Only decode/encode grayscale.
Definition: avcodec.h:883
#define av_log(a,...)
void * bitstream_buf
Definition: eatqi.c:47
BswapDSPContext bsdsp
Definition: eatqi.c:43
BlockDSPContext bdsp
Definition: eatqi.c:42
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:260
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
int last_dc[3]
Definition: eatqi.c:52
void(* clear_blocks)(int16_t *blocks)
Definition: blockdsp.h:37
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
int flags
AV_CODEC_FLAG_*.
Definition: avcodec.h:1645
const char * name
Name of the codec implementation.
Definition: avcodec.h:3484
int mb_x
Definition: eatqi.c:50
av_cold void ff_init_scantable_permutation(uint8_t *idct_permutation, enum idct_permutation_type perm_type)
Definition: idctdsp.c:50
ScanTable intra_scantable
Definition: eatqi.c:45
uint8_t w
Definition: llviddspenc.c:38
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several the filter must be ready for frames arriving randomly on any input any filter with several inputs will most likely require some kind of queuing mechanism It is perfectly acceptable to have a limited queue and to drop frames when the inputs are too unbalanced request_frame For filters that do not use the this method is called when a frame is wanted on an output For a it should directly call filter_frame on the corresponding output For a if there are queued frames already one of these frames should be pushed If the filter should request a frame on one of its repeatedly until at least one frame has been pushed Return or at least make progress towards producing a frame
int n
Definition: avisynth_c.h:760
uint8_t idct_permutation[64]
IDCT input permutation.
Definition: idctdsp.h:96
GetBitContext gb
Definition: eatqi.c:41
Libavcodec external API header.
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:299
const uint16_t ff_inv_aanscales[64]
Definition: aandcttab.c:38
main external API structure.
Definition: avcodec.h:1565
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: decode.c:1964
void * buf
Definition: avisynth_c.h:766
Rational number (pair of numerator and denominator).
Definition: rational.h:58
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:659
const uint16_t ff_mpeg1_default_intra_matrix[256]
Definition: mpeg12data.c:30
const uint8_t ff_zigzag_direct[64]
Definition: mathtables.c:98
IDCTDSPContext idsp
Definition: eatqi.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:282
int mb_y
Definition: eatqi.c:50
int ff_mpeg1_decode_block_intra(GetBitContext *gb, const uint16_t *quant_matrix, uint8_t *const scantable, int last_dc[3], int16_t *block, int index, int qscale)
Definition: mpeg12.c:240
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
common internal api header.
uint16_t intra_matrix[64]
Definition: eatqi.c:51
unsigned int bitstream_buf_size
Definition: eatqi.c:48
av_cold void ff_bswapdsp_init(BswapDSPContext *c)
Definition: bswapdsp.c:49
void * priv_data
Definition: avcodec.h:1592
av_cold void ff_init_scantable(uint8_t *permutation, ScanTable *st, const uint8_t *src_scantable)
Definition: idctdsp.c:29
av_cold void ff_idctdsp_init(IDCTDSPContext *c, AVCodecContext *avctx)
Definition: idctdsp.c:238
static int tqi_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: eatqi.c:123
#define av_freep(p)
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later.That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another.Frame references ownership and permissions
This structure stores compressed data.
Definition: avcodec.h:1454
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:981
int16_t block[6][64]
Definition: eatqi.c:54