Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * Duck TrueMotion 2.0 Real Time decoder | ||
3 | * | ||
4 | * This file is part of FFmpeg. | ||
5 | * | ||
6 | * FFmpeg is free software; you can redistribute it and/or | ||
7 | * modify it under the terms of the GNU Lesser General Public | ||
8 | * License as published by the Free Software Foundation; either | ||
9 | * version 2.1 of the License, or (at your option) any later version. | ||
10 | * | ||
11 | * FFmpeg is distributed in the hope that it will be useful, | ||
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
14 | * Lesser General Public License for more details. | ||
15 | * | ||
16 | * You should have received a copy of the GNU Lesser General Public | ||
17 | * License along with FFmpeg; if not, write to the Free Software | ||
18 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
19 | */ | ||
20 | |||
21 | #include "libavutil/internal.h" | ||
22 | #include "libavutil/intreadwrite.h" | ||
23 | |||
24 | #define BITSTREAM_READER_LE | ||
25 | #include "avcodec.h" | ||
26 | #include "codec_internal.h" | ||
27 | #include "decode.h" | ||
28 | #include "get_bits.h" | ||
29 | |||
30 | typedef struct TrueMotion2RTContext { | ||
31 | GetBitContext gb; | ||
32 | int delta_size; | ||
33 | int hscale; | ||
34 | } TrueMotion2RTContext; | ||
35 | |||
36 | static const int16_t delta_tab2[] = { | ||
37 | 5, -7, 36, -36, | ||
38 | }; | ||
39 | |||
40 | static const int16_t delta_tab3[] = { | ||
41 | 2, -3, 8, -8, 18, -18, 36, -36, | ||
42 | }; | ||
43 | |||
44 | static const int16_t delta_tab4[] = { | ||
45 | 1, -1, 2, -3, 8, -8, 18, -18, 36, -36, 54, -54, 96, -96, 144, -144, | ||
46 | }; | ||
47 | |||
48 | static const int16_t *const delta_tabs[] = { | ||
49 | delta_tab2, delta_tab3, delta_tab4, | ||
50 | }; | ||
51 | |||
52 | /* Returns the number of bytes consumed from the bytestream, or | ||
53 | * AVERROR_INVALIDDATA if there was an error while decoding the header. */ | ||
54 | 3 | static int truemotion2rt_decode_header(AVCodecContext *avctx, const AVPacket *avpkt) | |
55 | { | ||
56 | 3 | TrueMotion2RTContext *s = avctx->priv_data; | |
57 | int header_size; | ||
58 | 3 | uint8_t header_buffer[128] = { 0 }; /* logical maximum header size */ | |
59 | 3 | const uint8_t *buf = avpkt->data; | |
60 | 3 | int size = avpkt->size; | |
61 | int width, height; | ||
62 | int ret, i; | ||
63 | |||
64 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (size < 1) { |
65 | ✗ | av_log(avctx, AV_LOG_ERROR, "input packet too small (%d)\n", size); | |
66 | ✗ | return AVERROR_INVALIDDATA; | |
67 | } | ||
68 | |||
69 | 3 | header_size = ((buf[0] >> 5) | (buf[0] << 3)) & 0x7f; | |
70 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (header_size < 10) { |
71 | ✗ | av_log(avctx, AV_LOG_ERROR, "invalid header size (%d)\n", header_size); | |
72 | ✗ | return AVERROR_INVALIDDATA; | |
73 | } | ||
74 | |||
75 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (header_size + 1 > size) { |
76 | ✗ | av_log(avctx, AV_LOG_ERROR, "input packet too small (%d)\n", size); | |
77 | ✗ | return AVERROR_INVALIDDATA; | |
78 | } | ||
79 | |||
80 | /* unscramble the header bytes with a XOR operation */ | ||
81 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 3 times.
|
60 | for (i = 1; i < header_size; i++) |
82 | 57 | header_buffer[i - 1] = buf[i] ^ buf[i + 1]; | |
83 | |||
84 | 3 | s->delta_size = header_buffer[1]; | |
85 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
|
3 | s->hscale = 1 + !!header_buffer[3]; |
86 |
2/4✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 3 times.
|
3 | if (s->delta_size < 2 || s->delta_size > 4) |
87 | ✗ | return AVERROR_INVALIDDATA; | |
88 | |||
89 | 3 | height = AV_RL16(header_buffer + 5); | |
90 | 3 | width = AV_RL16(header_buffer + 7); | |
91 | |||
92 | 3 | ret = ff_set_dimensions(avctx, width, height); | |
93 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (ret < 0) |
94 | ✗ | return ret; | |
95 | |||
96 | 3 | av_log(avctx, AV_LOG_DEBUG, "Header size: %d\n", header_size); | |
97 | 3 | return header_size; | |
98 | } | ||
99 | |||
100 | 3 | static int truemotion2rt_decode_frame(AVCodecContext *avctx, AVFrame *p, | |
101 | int *got_frame, AVPacket *avpkt) | ||
102 | { | ||
103 | 3 | TrueMotion2RTContext *s = avctx->priv_data; | |
104 | 3 | GetBitContext *gb = &s->gb; | |
105 | uint8_t *dst; | ||
106 | int x, y, delta_mode; | ||
107 | int ret; | ||
108 | |||
109 | 3 | ret = truemotion2rt_decode_header(avctx, avpkt); | |
110 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (ret < 0) |
111 | ✗ | return ret; | |
112 | |||
113 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if ((avctx->width + s->hscale - 1)/ s->hscale * avctx->height * s->delta_size > avpkt->size * 8LL * 4) |
114 | ✗ | return AVERROR_INVALIDDATA; | |
115 | |||
116 | 3 | ret = init_get_bits8(gb, avpkt->data + ret, avpkt->size - ret); | |
117 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (ret < 0) |
118 | ✗ | return ret; | |
119 | |||
120 | 3 | ret = ff_get_buffer(avctx, p, 0); | |
121 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (ret < 0) |
122 | ✗ | return ret; | |
123 | |||
124 | 3 | skip_bits(gb, 32); | |
125 | 3 | delta_mode = s->delta_size - 2; | |
126 | 3 | dst = p->data[0]; | |
127 |
2/2✓ Branch 0 taken 660 times.
✓ Branch 1 taken 3 times.
|
663 | for (y = 0; y < avctx->height; y++) { |
128 | 660 | int diff = 0; | |
129 |
2/2✓ Branch 0 taken 92768 times.
✓ Branch 1 taken 660 times.
|
93428 | for (x = 0; x < avctx->width; x += s->hscale) { |
130 | 92768 | diff += delta_tabs[delta_mode][get_bits(gb, s->delta_size)]; | |
131 |
2/2✓ Branch 0 taken 92352 times.
✓ Branch 1 taken 416 times.
|
92768 | dst[x] = av_clip_uint8((y ? dst[x - p->linesize[0]] : 0) + diff); |
132 | } | ||
133 | 660 | dst += p->linesize[0]; | |
134 | } | ||
135 | |||
136 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
|
3 | if (s->hscale > 1) { |
137 | 2 | dst = p->data[0]; | |
138 |
2/2✓ Branch 0 taken 428 times.
✓ Branch 1 taken 2 times.
|
430 | for (y = 0; y < avctx->height; y++) { |
139 |
2/2✓ Branch 0 taken 44512 times.
✓ Branch 1 taken 428 times.
|
44940 | for (x = 1; x < avctx->width; x += s->hscale) |
140 | 44512 | dst[x] = dst[x - 1]; | |
141 | 428 | dst += p->linesize[0]; | |
142 | } | ||
143 | } | ||
144 | |||
145 | 3 | dst = p->data[0]; | |
146 |
2/2✓ Branch 0 taken 660 times.
✓ Branch 1 taken 3 times.
|
663 | for (y = 0; y < avctx->height; y++) { |
147 |
2/2✓ Branch 0 taken 137280 times.
✓ Branch 1 taken 660 times.
|
137940 | for (x = 0; x < avctx->width; x++) |
148 | 137280 | dst[x] = av_clip_uint8(dst[x] + (dst[x] - 128) / 3); | |
149 | 660 | dst += p->linesize[0]; | |
150 | } | ||
151 | |||
152 | 3 | dst = p->data[1]; | |
153 |
2/2✓ Branch 0 taken 165 times.
✓ Branch 1 taken 3 times.
|
168 | for (y = 0; y < avctx->height >> 2; y++) { |
154 | 165 | int diff = 0; | |
155 |
2/2✓ Branch 0 taken 5798 times.
✓ Branch 1 taken 165 times.
|
5963 | for (x = 0; x < avctx->width >> 2; x += s->hscale) { |
156 | 5798 | diff += delta_tabs[delta_mode][get_bits(gb, s->delta_size)]; | |
157 |
2/2✓ Branch 0 taken 5694 times.
✓ Branch 1 taken 104 times.
|
5798 | dst[x] = av_clip_uint8((y ? dst[x - p->linesize[1]] : 128) + diff); |
158 | } | ||
159 | 165 | dst += p->linesize[1]; | |
160 | } | ||
161 | |||
162 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
|
3 | if (s->hscale > 1) { |
163 | 2 | dst = p->data[1]; | |
164 |
2/2✓ Branch 0 taken 107 times.
✓ Branch 1 taken 2 times.
|
109 | for (y = 0; y < avctx->height >> 2; y++) { |
165 |
2/2✓ Branch 0 taken 2782 times.
✓ Branch 1 taken 107 times.
|
2889 | for (x = 1; x < avctx->width >> 2; x += s->hscale) |
166 | 2782 | dst[x] = dst[x - 1]; | |
167 | 107 | dst += p->linesize[1]; | |
168 | } | ||
169 | } | ||
170 | |||
171 | 3 | dst = p->data[1]; | |
172 |
2/2✓ Branch 0 taken 165 times.
✓ Branch 1 taken 3 times.
|
168 | for (y = 0; y < avctx->height >> 2; y++) { |
173 |
2/2✓ Branch 0 taken 8580 times.
✓ Branch 1 taken 165 times.
|
8745 | for (x = 0; x < avctx->width >> 2; x++) |
174 | 8580 | dst[x] += (dst[x] - 128) / 8; | |
175 | 165 | dst += p->linesize[1]; | |
176 | } | ||
177 | |||
178 | 3 | dst = p->data[2]; | |
179 |
2/2✓ Branch 0 taken 165 times.
✓ Branch 1 taken 3 times.
|
168 | for (y = 0; y < avctx->height >> 2; y++) { |
180 | 165 | int diff = 0; | |
181 |
2/2✓ Branch 0 taken 5798 times.
✓ Branch 1 taken 165 times.
|
5963 | for (x = 0; x < avctx->width >> 2; x += s->hscale) { |
182 | 5798 | diff += delta_tabs[delta_mode][get_bits(gb, s->delta_size)]; | |
183 |
2/2✓ Branch 0 taken 5694 times.
✓ Branch 1 taken 104 times.
|
5798 | dst[x] = av_clip_uint8((y ? dst[x - p->linesize[2]] : 128) + diff); |
184 | } | ||
185 | 165 | dst += p->linesize[2]; | |
186 | } | ||
187 | |||
188 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
|
3 | if (s->hscale > 1) { |
189 | 2 | dst = p->data[2]; | |
190 |
2/2✓ Branch 0 taken 107 times.
✓ Branch 1 taken 2 times.
|
109 | for (y = 0; y < avctx->height >> 2; y++) { |
191 |
2/2✓ Branch 0 taken 2782 times.
✓ Branch 1 taken 107 times.
|
2889 | for (x = 1; x < avctx->width >> 2; x += s->hscale) |
192 | 2782 | dst[x] = dst[x - 1]; | |
193 | 107 | dst += p->linesize[2]; | |
194 | } | ||
195 | } | ||
196 | |||
197 | 3 | dst = p->data[2]; | |
198 |
2/2✓ Branch 0 taken 165 times.
✓ Branch 1 taken 3 times.
|
168 | for (y = 0; y < avctx->height >> 2; y++) { |
199 |
2/2✓ Branch 0 taken 8580 times.
✓ Branch 1 taken 165 times.
|
8745 | for (x = 0; x < avctx->width >> 2; x++) |
200 | 8580 | dst[x] += (dst[x] - 128) / 8; | |
201 | 165 | dst += p->linesize[2]; | |
202 | } | ||
203 | |||
204 | 3 | *got_frame = 1; | |
205 | |||
206 | 3 | return avpkt->size; | |
207 | } | ||
208 | |||
209 | 6 | static av_cold int truemotion2rt_decode_init(AVCodecContext *avctx) | |
210 | { | ||
211 | 6 | avctx->pix_fmt = AV_PIX_FMT_YUV410P; | |
212 | 6 | return 0; | |
213 | } | ||
214 | |||
215 | const FFCodec ff_truemotion2rt_decoder = { | ||
216 | .p.name = "truemotion2rt", | ||
217 | CODEC_LONG_NAME("Duck TrueMotion 2.0 Real Time"), | ||
218 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
219 | .p.id = AV_CODEC_ID_TRUEMOTION2RT, | ||
220 | .priv_data_size = sizeof(TrueMotion2RTContext), | ||
221 | .init = truemotion2rt_decode_init, | ||
222 | FF_CODEC_DECODE_CB(truemotion2rt_decode_frame), | ||
223 | .p.capabilities = AV_CODEC_CAP_DR1, | ||
224 | }; | ||
225 |