Directory: | ../../../ffmpeg/ |
---|---|
File: | src/libavcodec/roqvideodec.c |
Date: | 2022-07-04 19:11:22 |
Exec | Total | Coverage | |
---|---|---|---|
Lines: | 129 | 137 | 94.2% |
Branches: | 61 | 74 | 82.4% |
Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * Copyright (C) 2003 The FFmpeg project | ||
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 | /** | ||
22 | * @file | ||
23 | * id RoQ Video Decoder by Dr. Tim Ferguson | ||
24 | * For more information about the id RoQ format, visit: | ||
25 | * http://www.csse.monash.edu.au/~timf/ | ||
26 | */ | ||
27 | |||
28 | #include "libavutil/avassert.h" | ||
29 | #include "libavutil/imgutils.h" | ||
30 | |||
31 | #include "avcodec.h" | ||
32 | #include "bytestream.h" | ||
33 | #include "codec_internal.h" | ||
34 | #include "internal.h" | ||
35 | #include "roqvideo.h" | ||
36 | |||
37 | 225 | static void roqvideo_decode_frame(RoqContext *ri, GetByteContext *gb) | |
38 | { | ||
39 | 225 | unsigned int chunk_id = 0, chunk_arg = 0; | |
40 | 225 | unsigned long chunk_size = 0; | |
41 | 225 | int i, j, k, nv1, nv2, vqflg = 0, vqflg_pos = -1; | |
42 | int vqid, xpos, ypos, xp, yp, x, y, mx, my; | ||
43 | roq_qcell *qcell; | ||
44 | int64_t chunk_start; | ||
45 | |||
46 |
1/2✓ Branch 1 taken 428 times.
✗ Branch 2 not taken.
|
428 | while (bytestream2_get_bytes_left(gb) >= 8) { |
47 | 428 | chunk_id = bytestream2_get_le16(gb); | |
48 | 428 | chunk_size = bytestream2_get_le32(gb); | |
49 | 428 | chunk_arg = bytestream2_get_le16(gb); | |
50 | |||
51 |
2/2✓ Branch 0 taken 225 times.
✓ Branch 1 taken 203 times.
|
428 | if(chunk_id == RoQ_QUAD_VQ) |
52 | 225 | break; | |
53 |
1/2✓ Branch 0 taken 203 times.
✗ Branch 1 not taken.
|
203 | if(chunk_id == RoQ_QUAD_CODEBOOK) { |
54 |
2/2✓ Branch 0 taken 67 times.
✓ Branch 1 taken 136 times.
|
203 | if((nv1 = chunk_arg >> 8) == 0) |
55 | 67 | nv1 = 256; | |
56 |
4/4✓ Branch 0 taken 47 times.
✓ Branch 1 taken 156 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 46 times.
|
203 | if((nv2 = chunk_arg & 0xff) == 0 && nv1 * 6 < chunk_size) |
57 | 1 | nv2 = 256; | |
58 |
2/2✓ Branch 0 taken 50899 times.
✓ Branch 1 taken 203 times.
|
51102 | for(i = 0; i < nv1; i++) { |
59 | 50899 | ri->cb2x2[i].y[0] = bytestream2_get_byte(gb); | |
60 | 50899 | ri->cb2x2[i].y[1] = bytestream2_get_byte(gb); | |
61 | 50899 | ri->cb2x2[i].y[2] = bytestream2_get_byte(gb); | |
62 | 50899 | ri->cb2x2[i].y[3] = bytestream2_get_byte(gb); | |
63 | 50899 | ri->cb2x2[i].u = bytestream2_get_byte(gb); | |
64 | 50899 | ri->cb2x2[i].v = bytestream2_get_byte(gb); | |
65 | } | ||
66 |
2/2✓ Branch 0 taken 17731 times.
✓ Branch 1 taken 203 times.
|
17934 | for(i = 0; i < nv2; i++) |
67 |
2/2✓ Branch 0 taken 70924 times.
✓ Branch 1 taken 17731 times.
|
88655 | for(j = 0; j < 4; j++) |
68 | 70924 | ri->cb4x4[i].idx[j] = bytestream2_get_byte(gb); | |
69 | } | ||
70 | } | ||
71 | |||
72 | 225 | chunk_start = bytestream2_tell(gb); | |
73 | 225 | xpos = ypos = 0; | |
74 | |||
75 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 225 times.
|
225 | if (chunk_size > bytestream2_get_bytes_left(gb)) { |
76 | ✗ | av_log(ri->avctx, AV_LOG_ERROR, "Chunk does not fit in input buffer\n"); | |
77 | ✗ | chunk_size = bytestream2_get_bytes_left(gb); | |
78 | } | ||
79 | |||
80 |
1/2✓ Branch 1 taken 113460 times.
✗ Branch 2 not taken.
|
113460 | while (bytestream2_tell(gb) < chunk_start + chunk_size) { |
81 |
2/2✓ Branch 0 taken 226916 times.
✓ Branch 1 taken 113426 times.
|
340342 | for (yp = ypos; yp < ypos + 16; yp += 8) |
82 |
2/2✓ Branch 0 taken 453825 times.
✓ Branch 1 taken 226882 times.
|
680707 | for (xp = xpos; xp < xpos + 16; xp += 8) { |
83 |
2/2✓ Branch 1 taken 4 times.
✓ Branch 2 taken 453821 times.
|
453825 | if (bytestream2_tell(gb) >= chunk_start + chunk_size) { |
84 | 4 | av_log(ri->avctx, AV_LOG_VERBOSE, "Chunk is too short\n"); | |
85 | 4 | return; | |
86 | } | ||
87 |
2/2✓ Branch 0 taken 56771 times.
✓ Branch 1 taken 397050 times.
|
453821 | if (vqflg_pos < 0) { |
88 | 56771 | vqflg = bytestream2_get_le16(gb); | |
89 | 56771 | vqflg_pos = 7; | |
90 | } | ||
91 | 453821 | vqid = (vqflg >> (vqflg_pos * 2)) & 0x3; | |
92 | 453821 | vqflg_pos--; | |
93 | |||
94 |
4/5✓ Branch 0 taken 121738 times.
✓ Branch 1 taken 9277 times.
✓ Branch 2 taken 7001 times.
✓ Branch 3 taken 315805 times.
✗ Branch 4 not taken.
|
453821 | switch(vqid) { |
95 | 121738 | case RoQ_ID_MOT: | |
96 | 121738 | break; | |
97 | 9277 | case RoQ_ID_FCC: { | |
98 | 9277 | int byte = bytestream2_get_byte(gb); | |
99 | 9277 | mx = 8 - (byte >> 4) - ((signed char) (chunk_arg >> 8)); | |
100 | 9277 | my = 8 - (byte & 0xf) - ((signed char) chunk_arg); | |
101 | 9277 | ff_apply_motion_8x8(ri, xp, yp, mx, my); | |
102 | 9277 | break; | |
103 | } | ||
104 | 7001 | case RoQ_ID_SLD: | |
105 | 7001 | qcell = ri->cb4x4 + bytestream2_get_byte(gb); | |
106 | 7001 | ff_apply_vector_4x4(ri, xp, yp, ri->cb2x2 + qcell->idx[0]); | |
107 | 7001 | ff_apply_vector_4x4(ri, xp + 4, yp, ri->cb2x2 + qcell->idx[1]); | |
108 | 7001 | ff_apply_vector_4x4(ri, xp, yp + 4, ri->cb2x2 + qcell->idx[2]); | |
109 | 7001 | ff_apply_vector_4x4(ri, xp + 4, yp + 4, ri->cb2x2 + qcell->idx[3]); | |
110 | 7001 | break; | |
111 | 315805 | case RoQ_ID_CCC: | |
112 |
2/2✓ Branch 0 taken 1263209 times.
✓ Branch 1 taken 315775 times.
|
1578984 | for (k = 0; k < 4; k++) { |
113 | 1263209 | x = xp; y = yp; | |
114 |
2/2✓ Branch 0 taken 631599 times.
✓ Branch 1 taken 631610 times.
|
1263209 | if(k & 0x01) x += 4; |
115 |
2/2✓ Branch 0 taken 631599 times.
✓ Branch 1 taken 631610 times.
|
1263209 | if(k & 0x02) y += 4; |
116 | |||
117 |
2/2✓ Branch 1 taken 30 times.
✓ Branch 2 taken 1263179 times.
|
1263209 | if (bytestream2_tell(gb) >= chunk_start + chunk_size) { |
118 | 30 | av_log(ri->avctx, AV_LOG_VERBOSE, "Chunk is too short\n"); | |
119 | 30 | return; | |
120 | } | ||
121 |
2/2✓ Branch 0 taken 157911 times.
✓ Branch 1 taken 1105268 times.
|
1263179 | if (vqflg_pos < 0) { |
122 | 157911 | vqflg = bytestream2_get_le16(gb); | |
123 | 157911 | vqflg_pos = 7; | |
124 | } | ||
125 | 1263179 | vqid = (vqflg >> (vqflg_pos * 2)) & 0x3; | |
126 |
4/5✓ Branch 0 taken 259115 times.
✓ Branch 1 taken 521989 times.
✓ Branch 2 taken 116817 times.
✓ Branch 3 taken 365258 times.
✗ Branch 4 not taken.
|
1263179 | vqflg_pos--; |
127 | switch(vqid) { | ||
128 | 259115 | case RoQ_ID_MOT: | |
129 | 259115 | break; | |
130 | 521989 | case RoQ_ID_FCC: { | |
131 | 521989 | int byte = bytestream2_get_byte(gb); | |
132 | 521989 | mx = 8 - (byte >> 4) - ((signed char) (chunk_arg >> 8)); | |
133 | 521989 | my = 8 - (byte & 0xf) - ((signed char) chunk_arg); | |
134 | 521989 | ff_apply_motion_4x4(ri, x, y, mx, my); | |
135 | 521989 | break; | |
136 | } | ||
137 | 116817 | case RoQ_ID_SLD: | |
138 | 116817 | qcell = ri->cb4x4 + bytestream2_get_byte(gb); | |
139 | 116817 | ff_apply_vector_2x2(ri, x, y, ri->cb2x2 + qcell->idx[0]); | |
140 | 116817 | ff_apply_vector_2x2(ri, x + 2, y, ri->cb2x2 + qcell->idx[1]); | |
141 | 116817 | ff_apply_vector_2x2(ri, x, y + 2, ri->cb2x2 + qcell->idx[2]); | |
142 | 116817 | ff_apply_vector_2x2(ri, x + 2, y + 2, ri->cb2x2 + qcell->idx[3]); | |
143 | 116817 | break; | |
144 | 365258 | case RoQ_ID_CCC: | |
145 | 365258 | ff_apply_vector_2x2(ri, x, y, ri->cb2x2 + bytestream2_get_byte(gb)); | |
146 | 365258 | ff_apply_vector_2x2(ri, x + 2, y, ri->cb2x2 + bytestream2_get_byte(gb)); | |
147 | 365258 | ff_apply_vector_2x2(ri, x, y + 2, ri->cb2x2 + bytestream2_get_byte(gb)); | |
148 | 365258 | ff_apply_vector_2x2(ri, x + 2, y + 2, ri->cb2x2 + bytestream2_get_byte(gb)); | |
149 | 365258 | break; | |
150 | } | ||
151 | } | ||
152 | 315775 | break; | |
153 | 453791 | default: | |
154 | av_assert2(0); | ||
155 | } | ||
156 | } | ||
157 | |||
158 | 113426 | xpos += 16; | |
159 |
2/2✓ Branch 0 taken 3596 times.
✓ Branch 1 taken 109830 times.
|
113426 | if (xpos >= ri->width) { |
160 | 3596 | xpos -= ri->width; | |
161 | 3596 | ypos += 16; | |
162 | } | ||
163 |
2/2✓ Branch 0 taken 191 times.
✓ Branch 1 taken 113235 times.
|
113426 | if(ypos >= ri->height) |
164 | 191 | break; | |
165 | } | ||
166 | } | ||
167 | |||
168 | |||
169 | 10 | static av_cold int roq_decode_init(AVCodecContext *avctx) | |
170 | { | ||
171 | 10 | RoqContext *s = avctx->priv_data; | |
172 | |||
173 | 10 | s->avctx = avctx; | |
174 | |||
175 |
2/4✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 10 times.
|
10 | if (avctx->width % 16 || avctx->height % 16) { |
176 | ✗ | avpriv_request_sample(avctx, "Dimensions not being a multiple of 16"); | |
177 | ✗ | return AVERROR_PATCHWELCOME; | |
178 | } | ||
179 | |||
180 | 10 | s->width = avctx->width; | |
181 | 10 | s->height = avctx->height; | |
182 | |||
183 | 10 | s->last_frame = av_frame_alloc(); | |
184 | 10 | s->current_frame = av_frame_alloc(); | |
185 |
2/4✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 10 times.
|
10 | if (!s->current_frame || !s->last_frame) |
186 | ✗ | return AVERROR(ENOMEM); | |
187 | |||
188 | 10 | avctx->pix_fmt = AV_PIX_FMT_YUVJ444P; | |
189 | 10 | avctx->color_range = AVCOL_RANGE_JPEG; | |
190 | |||
191 | 10 | return 0; | |
192 | } | ||
193 | |||
194 | 225 | static int roq_decode_frame(AVCodecContext *avctx, AVFrame *rframe, | |
195 | int *got_frame, AVPacket *avpkt) | ||
196 | { | ||
197 | 225 | const uint8_t *buf = avpkt->data; | |
198 | 225 | int buf_size = avpkt->size; | |
199 | 225 | RoqContext *s = avctx->priv_data; | |
200 |
4/4✓ Branch 0 taken 8 times.
✓ Branch 1 taken 217 times.
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 4 times.
|
225 | int copy = !s->current_frame->data[0] && s->last_frame->data[0]; |
201 | GetByteContext gb; | ||
202 | int ret; | ||
203 | |||
204 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 225 times.
|
225 | if ((ret = ff_reget_buffer(avctx, s->current_frame, 0)) < 0) |
205 | ✗ | return ret; | |
206 | |||
207 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 221 times.
|
225 | if (copy) { |
208 | 4 | ret = av_frame_copy(s->current_frame, s->last_frame); | |
209 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (ret < 0) |
210 | ✗ | return ret; | |
211 | } | ||
212 | |||
213 | 225 | bytestream2_init(&gb, buf, buf_size); | |
214 | 225 | roqvideo_decode_frame(s, &gb); | |
215 | |||
216 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 225 times.
|
225 | if ((ret = av_frame_ref(rframe, s->current_frame)) < 0) |
217 | ✗ | return ret; | |
218 | 225 | *got_frame = 1; | |
219 | |||
220 | /* shuffle frames */ | ||
221 | 225 | FFSWAP(AVFrame *, s->current_frame, s->last_frame); | |
222 | |||
223 | 225 | return buf_size; | |
224 | } | ||
225 | |||
226 | 10 | static av_cold int roq_decode_end(AVCodecContext *avctx) | |
227 | { | ||
228 | 10 | RoqContext *s = avctx->priv_data; | |
229 | |||
230 | 10 | av_frame_free(&s->current_frame); | |
231 | 10 | av_frame_free(&s->last_frame); | |
232 | |||
233 | 10 | return 0; | |
234 | } | ||
235 | |||
236 | const FFCodec ff_roq_decoder = { | ||
237 | .p.name = "roqvideo", | ||
238 | .p.long_name = NULL_IF_CONFIG_SMALL("id RoQ video"), | ||
239 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
240 | .p.id = AV_CODEC_ID_ROQ, | ||
241 | .priv_data_size = sizeof(RoqContext), | ||
242 | .init = roq_decode_init, | ||
243 | .close = roq_decode_end, | ||
244 | FF_CODEC_DECODE_CB(roq_decode_frame), | ||
245 | .p.capabilities = AV_CODEC_CAP_DR1, | ||
246 | .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | FF_CODEC_CAP_INIT_CLEANUP, | ||
247 | }; | ||
248 |