Libav
Main Page
Related Pages
Modules
Data Structures
Files
Examples
File List
Globals
libavcodec
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 Libav.
6
*
7
* Libav 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
* Libav 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 Libav; if not, write to the Free Software
19
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
20
*/
21
29
#include "
avcodec.h
"
30
#include "
get_bits.h
"
31
#include "
aandcttab.h
"
32
#include "
eaidct.h
"
33
#include "
internal.h
"
34
#include "
mpeg12.h
"
35
#include "
mpegvideo.h
"
36
37
typedef
struct
TqiContext
{
38
MpegEncContext
s
;
39
void
*
bitstream_buf
;
40
unsigned
int
bitstream_buf_size
;
41
DECLARE_ALIGNED
(16, int16_t,
block
)[6][64];
42
}
TqiContext
;
43
44
static
av_cold
int
tqi_decode_init
(
AVCodecContext
*avctx)
45
{
46
TqiContext
*
t
= avctx->
priv_data
;
47
MpegEncContext
*s = &t->
s
;
48
s->
avctx
= avctx;
49
ff_dsputil_init
(&s->
dsp
, avctx);
50
ff_init_scantable_permutation
(s->
dsp
.
idct_permutation
,
FF_NO_IDCT_PERM
);
51
ff_init_scantable
(s->
dsp
.
idct_permutation
, &s->
intra_scantable
,
ff_zigzag_direct
);
52
s->
qscale
= 1;
53
avctx->
time_base
= (
AVRational
){1, 15};
54
avctx->
pix_fmt
=
AV_PIX_FMT_YUV420P
;
55
ff_mpeg12_init_vlcs
();
56
return
0;
57
}
58
59
static
int
tqi_decode_mb
(
MpegEncContext
*s, int16_t (*
block
)[64])
60
{
61
int
n;
62
s->
dsp
.
clear_blocks
(
block
[0]);
63
for
(n=0; n<6; n++)
64
if
(
ff_mpeg1_decode_block_intra
(s,
block
[n], n) < 0)
65
return
-1;
66
67
return
0;
68
}
69
70
static
inline
void
tqi_idct_put
(
TqiContext
*
t
,
AVFrame
*frame, int16_t (*
block
)[64])
71
{
72
MpegEncContext
*s = &t->
s
;
73
int
linesize = frame->
linesize
[0];
74
uint8_t
*dest_y = frame->
data
[0] + (s->
mb_y
* 16* linesize ) + s->
mb_x
* 16;
75
uint8_t
*dest_cb = frame->
data
[1] + (s->
mb_y
* 8 * frame->
linesize
[1]) + s->
mb_x
* 8;
76
uint8_t
*dest_cr = frame->
data
[2] + (s->
mb_y
* 8 * frame->
linesize
[2]) + s->
mb_x
* 8;
77
78
ff_ea_idct_put_c
(dest_y , linesize,
block
[0]);
79
ff_ea_idct_put_c
(dest_y + 8, linesize,
block
[1]);
80
ff_ea_idct_put_c
(dest_y + 8*linesize , linesize,
block
[2]);
81
ff_ea_idct_put_c
(dest_y + 8*linesize + 8, linesize,
block
[3]);
82
if
(!(s->
avctx
->
flags
&
CODEC_FLAG_GRAY
)) {
83
ff_ea_idct_put_c
(dest_cb, frame->
linesize
[1],
block
[4]);
84
ff_ea_idct_put_c
(dest_cr, frame->
linesize
[2],
block
[5]);
85
}
86
}
87
88
static
void
tqi_calculate_qtable
(
MpegEncContext
*s,
int
quant
)
89
{
90
const
int
qscale = (215 - 2*
quant
)*5;
91
int
i;
92
s->
intra_matrix
[0] = (
ff_inv_aanscales
[0]*
ff_mpeg1_default_intra_matrix
[0])>>11;
93
for
(i=1; i<64; i++)
94
s->
intra_matrix
[i] = (
ff_inv_aanscales
[i]*
ff_mpeg1_default_intra_matrix
[i]*qscale + 32)>>14;
95
}
96
97
static
int
tqi_decode_frame
(
AVCodecContext
*avctx,
98
void
*
data
,
int
*got_frame,
99
AVPacket
*avpkt)
100
{
101
const
uint8_t
*buf = avpkt->
data
;
102
int
buf_size = avpkt->
size
;
103
const
uint8_t
*buf_end = buf+buf_size;
104
TqiContext
*t = avctx->
priv_data
;
105
MpegEncContext
*s = &t->
s
;
106
AVFrame
*frame =
data
;
107
int
ret;
108
109
s->
width
=
AV_RL16
(&buf[0]);
110
s->
height
=
AV_RL16
(&buf[2]);
111
tqi_calculate_qtable
(s, buf[4]);
112
buf += 8;
113
114
ret =
ff_set_dimensions
(s->
avctx
, s->
width
, s->
height
);
115
if
(ret < 0)
116
return
ret;
117
118
if
((ret =
ff_get_buffer
(avctx, frame, 0)) < 0) {
119
av_log
(avctx,
AV_LOG_ERROR
,
"get_buffer() failed\n"
);
120
return
ret;
121
}
122
123
av_fast_padded_malloc
(&t->
bitstream_buf
, &t->
bitstream_buf_size
,
124
buf_end - buf);
125
if
(!t->
bitstream_buf
)
126
return
AVERROR
(ENOMEM);
127
s->
dsp
.
bswap_buf
(t->
bitstream_buf
, (
const
uint32_t*)buf, (buf_end-buf)/4);
128
init_get_bits
(&s->
gb
, t->
bitstream_buf
, 8*(buf_end-buf));
129
130
s->
last_dc
[0] = s->
last_dc
[1] = s->
last_dc
[2] = 0;
131
for
(s->
mb_y
=0; s->
mb_y
<(avctx->
height
+15)/16; s->
mb_y
++)
132
for
(s->
mb_x
=0; s->
mb_x
<(avctx->
width
+15)/16; s->
mb_x
++)
133
{
134
if
(
tqi_decode_mb
(s, t->
block
) < 0)
135
break
;
136
tqi_idct_put
(t, frame, t->
block
);
137
}
138
139
*got_frame = 1;
140
return
buf_size;
141
}
142
143
static
av_cold
int
tqi_decode_end
(
AVCodecContext
*avctx)
144
{
145
TqiContext
*t = avctx->
priv_data
;
146
av_free
(t->
bitstream_buf
);
147
return
0;
148
}
149
150
AVCodec
ff_eatqi_decoder
= {
151
.
name
=
"eatqi"
,
152
.long_name =
NULL_IF_CONFIG_SMALL
(
"Electronic Arts TQI Video"
),
153
.type =
AVMEDIA_TYPE_VIDEO
,
154
.id =
AV_CODEC_ID_TQI
,
155
.priv_data_size =
sizeof
(
TqiContext
),
156
.
init
=
tqi_decode_init
,
157
.
close
=
tqi_decode_end
,
158
.
decode
=
tqi_decode_frame
,
159
.capabilities =
CODEC_CAP_DR1
,
160
};
Generated on Thu Sep 30 2021 23:03:10 for Libav by
1.8.1.2