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 "libavutil/mem_internal.h"
30 
31 #include "avcodec.h"
32 #include "blockdsp.h"
33 #include "bswapdsp.h"
34 #include "get_bits.h"
35 #include "aandcttab.h"
36 #include "eaidct.h"
37 #include "idctdsp.h"
38 #include "internal.h"
39 #include "mpeg12.h"
40 
41 typedef struct TqiContext {
48 
50  unsigned int bitstream_buf_size;
51 
52  int mb_x, mb_y;
53  uint16_t intra_matrix[64];
54  int last_dc[3];
55 
56  DECLARE_ALIGNED(32, int16_t, block)[6][64];
57 } TqiContext;
58 
60 {
61  TqiContext *t = avctx->priv_data;
62 
63  ff_blockdsp_init(&t->bdsp, avctx);
65  ff_idctdsp_init(&t->idsp, avctx);
68 
69  avctx->framerate = (AVRational){ 15, 1 };
70  avctx->pix_fmt = AV_PIX_FMT_YUV420P;
72  return 0;
73 }
74 
75 static int tqi_decode_mb(TqiContext *t, int16_t (*block)[64])
76 {
77  int n;
78 
79  t->bdsp.clear_blocks(block[0]);
80  for (n = 0; n < 6; n++) {
82  t->intra_matrix,
84  t->last_dc, block[n], n, 1);
85  if (ret < 0) {
86  av_log(t->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n",
87  t->mb_x, t->mb_y);
88  return ret;
89  }
90  }
91 
92  return 0;
93 }
94 
95 static inline void tqi_idct_put(AVCodecContext *avctx, AVFrame *frame,
96  int16_t (*block)[64])
97 {
98  TqiContext *t = avctx->priv_data;
99  ptrdiff_t linesize = frame->linesize[0];
100  uint8_t *dest_y = frame->data[0] + t->mb_y * 16 * linesize + t->mb_x * 16;
101  uint8_t *dest_cb = frame->data[1] + t->mb_y * 8 * frame->linesize[1] + t->mb_x * 8;
102  uint8_t *dest_cr = frame->data[2] + t->mb_y * 8 * frame->linesize[2] + t->mb_x * 8;
103 
104  ff_ea_idct_put_c(dest_y , linesize, block[0]);
105  ff_ea_idct_put_c(dest_y + 8, linesize, block[1]);
106  ff_ea_idct_put_c(dest_y + 8*linesize , linesize, block[2]);
107  ff_ea_idct_put_c(dest_y + 8*linesize + 8, linesize, block[3]);
108 
109  if (!(avctx->flags & AV_CODEC_FLAG_GRAY)) {
110  ff_ea_idct_put_c(dest_cb, frame->linesize[1], block[4]);
111  ff_ea_idct_put_c(dest_cr, frame->linesize[2], block[5]);
112  }
113 }
114 
116 {
117  const int64_t qscale = (215 - 2*quant)*5;
118  int i;
119 
121  for(i=1; i<64; i++)
122  t->intra_matrix[i] = (ff_inv_aanscales[i] * ff_mpeg1_default_intra_matrix[i] * qscale + 32) >> 14;
123 }
124 
126  void *data, int *got_frame,
127  AVPacket *avpkt)
128 {
129  const uint8_t *buf = avpkt->data;
130  int buf_size = avpkt->size;
131  const uint8_t *buf_end = buf+buf_size;
132  TqiContext *t = avctx->priv_data;
133  AVFrame *frame = data;
134  int ret, w, h;
135 
136  if (buf_size < 12)
137  return AVERROR_INVALIDDATA;
138 
139  t->avctx = avctx;
140 
141  w = AV_RL16(&buf[0]);
142  h = AV_RL16(&buf[2]);
143  tqi_calculate_qtable(t, buf[4]);
144  buf += 8;
145 
146  ret = ff_set_dimensions(avctx, w, h);
147  if (ret < 0)
148  return ret;
149 
150  if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
151  return ret;
152 
154  buf_end - buf);
155  if (!t->bitstream_buf)
156  return AVERROR(ENOMEM);
157  t->bsdsp.bswap_buf(t->bitstream_buf, (const uint32_t *) buf,
158  (buf_end - buf) / 4);
159  init_get_bits(&t->gb, t->bitstream_buf, 8 * (buf_end - buf));
160 
161  t->last_dc[0] =
162  t->last_dc[1] =
163  t->last_dc[2] = 0;
164  for (t->mb_y = 0; t->mb_y < (h + 15) / 16; t->mb_y++) {
165  for (t->mb_x = 0; t->mb_x < (w + 15) / 16; t->mb_x++) {
166  if (tqi_decode_mb(t, t->block) < 0)
167  goto end;
168  tqi_idct_put(avctx, frame, t->block);
169  }
170  }
171  end:
172 
173  *got_frame = 1;
174  return buf_size;
175 }
176 
178 {
179  TqiContext *t = avctx->priv_data;
180  av_freep(&t->bitstream_buf);
181  return 0;
182 }
183 
185  .name = "eatqi",
186  .long_name = NULL_IF_CONFIG_SMALL("Electronic Arts TQI Video"),
187  .type = AVMEDIA_TYPE_VIDEO,
188  .id = AV_CODEC_ID_TQI,
189  .priv_data_size = sizeof(TqiContext),
191  .close = tqi_decode_end,
193  .capabilities = AV_CODEC_CAP_DR1,
194  .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
195 };
AVCodec
AVCodec.
Definition: codec.h:202
bswapdsp.h
FF_CODEC_CAP_INIT_THREADSAFE
#define FF_CODEC_CAP_INIT_THREADSAFE
The codec does not modify any global variables in the init function, allowing to call the init functi...
Definition: internal.h:42
ff_init_scantable
av_cold void ff_init_scantable(uint8_t *permutation, ScanTable *st, const uint8_t *src_scantable)
Definition: idctdsp.c:29
ff_eatqi_decoder
const AVCodec ff_eatqi_decoder
Definition: eatqi.c:184
TqiContext::idsp
IDCTDSPContext idsp
Definition: eatqi.c:46
blockdsp.h
AVERROR
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
AV_CODEC_ID_TQI
@ AV_CODEC_ID_TQI
Definition: codec_id.h:172
mem_internal.h
BlockDSPContext::clear_blocks
void(* clear_blocks)(int16_t *blocks)
Definition: blockdsp.h:36
tqi_decode_mb
static int tqi_decode_mb(TqiContext *t, int16_t(*block)[64])
Definition: eatqi.c:75
tqi_decode_end
static av_cold int tqi_decode_end(AVCodecContext *avctx)
Definition: eatqi.c:177
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:317
w
uint8_t w
Definition: llviddspenc.c:38
internal.h
AVPacket::data
uint8_t * data
Definition: packet.h:373
data
const char data[16]
Definition: mxf.c:143
TqiContext::last_dc
int last_dc[3]
Definition: eatqi.c:54
BlockDSPContext
Definition: blockdsp.h:34
init_get_bits
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:660
tqi_decode_frame
static int tqi_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: eatqi.c:125
init
static int init
Definition: av_tx.c:47
AVCodecContext::framerate
AVRational framerate
Definition: avcodec.h:1710
ff_mpeg1_decode_block_intra
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:239
GetBitContext
Definition: get_bits.h:62
AVCodecContext::flags
int flags
AV_CODEC_FLAG_*.
Definition: avcodec.h:463
ff_ea_idct_put_c
void ff_ea_idct_put_c(uint8_t *dest, ptrdiff_t linesize, int16_t *block)
Definition: eaidct.c:80
TqiContext::gb
GetBitContext gb
Definition: eatqi.c:43
TqiContext::bitstream_buf
void * bitstream_buf
Definition: eatqi.c:49
quant
static int quant(float coef, const float Q, const float rounding)
Quantize one coefficient.
Definition: aacenc_utils.h:59
mpeg12.h
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
av_cold
#define av_cold
Definition: attributes.h:90
TqiContext::intra_scantable
ScanTable intra_scantable
Definition: eatqi.c:47
TqiContext::bitstream_buf_size
unsigned int bitstream_buf_size
Definition: eatqi.c:50
decode
static void decode(AVCodecContext *dec_ctx, AVPacket *pkt, AVFrame *frame, FILE *outfile)
Definition: decode_audio.c:71
tqi_calculate_qtable
static void tqi_calculate_qtable(TqiContext *t, int quant)
Definition: eatqi.c:115
BswapDSPContext::bswap_buf
void(* bswap_buf)(uint32_t *dst, const uint32_t *src, int w)
Definition: bswapdsp.h:25
get_bits.h
AV_RL16
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_RL16
Definition: bytestream.h:94
TqiContext::bdsp
BlockDSPContext bdsp
Definition: eatqi.c:44
TqiContext::mb_x
int mb_x
Definition: eatqi.c:52
TqiContext::bsdsp
BswapDSPContext bsdsp
Definition: eatqi.c:45
AV_PIX_FMT_YUV420P
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
ff_bswapdsp_init
av_cold void ff_bswapdsp_init(BswapDSPContext *c)
Definition: bswapdsp.c:49
TqiContext
Definition: eatqi.c:41
aandcttab.h
AVRational
Rational number (pair of numerator and denominator).
Definition: rational.h:58
TqiContext::block
int16_t block[6][64]
Definition: eatqi.c:56
tqi_decode_init
static av_cold int tqi_decode_init(AVCodecContext *avctx)
Definition: eatqi.c:59
ff_get_buffer
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: decode.c:1652
AV_CODEC_CAP_DR1
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() or get_encode_buffer() for allocating buffers and supports custom allocators.
Definition: codec.h:52
AV_CODEC_FLAG_GRAY
#define AV_CODEC_FLAG_GRAY
Only decode/encode grayscale.
Definition: avcodec.h:243
AVPacket::size
int size
Definition: packet.h:374
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
FF_IDCT_PERM_NONE
@ FF_IDCT_PERM_NONE
Definition: idctdsp.h:38
ff_mpeg1_default_intra_matrix
const uint16_t ff_mpeg1_default_intra_matrix[256]
Definition: mpeg12data.c:30
ff_init_scantable_permutation
av_cold void ff_init_scantable_permutation(uint8_t *idct_permutation, enum idct_permutation_type perm_type)
Definition: idctdsp.c:50
DECLARE_ALIGNED
#define DECLARE_ALIGNED(n, t, v)
Definition: mem.h:116
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:271
ff_mpeg12_init_vlcs
av_cold void ff_mpeg12_init_vlcs(void)
Definition: mpeg12.c:165
av_fast_padded_malloc
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:50
ff_idctdsp_init
av_cold void ff_idctdsp_init(IDCTDSPContext *c, AVCodecContext *avctx)
Definition: idctdsp.c:238
AVCodec::name
const char * name
Name of the codec implementation.
Definition: codec.h:209
AVCodecContext::pix_fmt
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:593
idctdsp.h
avcodec.h
ff_zigzag_direct
const uint8_t ff_zigzag_direct[64]
Definition: mathtables.c:98
ret
ret
Definition: filter_design.txt:187
frame
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
Definition: filter_design.txt:264
ff_inv_aanscales
const uint16_t ff_inv_aanscales[64]
Definition: aandcttab.c:38
TqiContext::avctx
AVCodecContext * avctx
Definition: eatqi.c:42
IDCTDSPContext
Definition: idctdsp.h:53
AVCodecContext
main external API structure.
Definition: avcodec.h:383
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
ff_set_dimensions
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:86
IDCTDSPContext::idct_permutation
uint8_t idct_permutation[64]
IDCT input permutation.
Definition: idctdsp.h:96
ScanTable
Scantable.
Definition: idctdsp.h:31
AVPacket
This structure stores compressed data.
Definition: packet.h:350
AVCodecContext::priv_data
void * priv_data
Definition: avcodec.h:410
ScanTable::permutated
uint8_t permutated[64]
Definition: idctdsp.h:33
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:35
TqiContext::intra_matrix
uint16_t intra_matrix[64]
Definition: eatqi.c:53
eaidct.h
block
The exact code depends on how similar the blocks are and how related they are to the block
Definition: filter_design.txt:207
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:28
AVERROR_INVALIDDATA
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:61
BswapDSPContext
Definition: bswapdsp.h:24
h
h
Definition: vp9dsp_template.c:2038
ff_blockdsp_init
av_cold void ff_blockdsp_init(BlockDSPContext *c, AVCodecContext *avctx)
Definition: blockdsp.c:59
tqi_idct_put
static void tqi_idct_put(AVCodecContext *avctx, AVFrame *frame, int16_t(*block)[64])
Definition: eatqi.c:95
TqiContext::mb_y
int mb_y
Definition: eatqi.c:52