FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/rpzaenc.c
Date: 2024-07-24 19:24:46
Exec Total Coverage
Lines: 311 372 83.6%
Functions: 13 17 76.5%
Branches: 120 150 80.0%

Line Branch Exec Source
1 /*
2 * QuickTime RPZA Video Encoder
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 rpzaenc.c
23 * QT RPZA Video Encoder by Todd Kirby <doubleshot@pacbell.net> and David Adler
24 */
25
26 #include "libavutil/avassert.h"
27 #include "libavutil/common.h"
28 #include "libavutil/opt.h"
29
30 #include "avcodec.h"
31 #include "codec_internal.h"
32 #include "encode.h"
33 #include "mathops.h"
34 #include "put_bits.h"
35
36 typedef struct RpzaContext {
37 AVClass *avclass;
38
39 int skip_frame_thresh;
40 int start_one_color_thresh;
41 int continue_one_color_thresh;
42 int sixteen_color_thresh;
43
44 AVFrame *prev_frame; // buffer for previous source frame
45 PutBitContext pb; // buffer for encoded frame data.
46
47 int frame_width; // width in pixels of source frame
48 int frame_height; // height in pixesl of source frame
49
50 int first_frame; // flag set to one when the first frame is being processed
51 // so that comparisons with previous frame data in not attempted
52 } RpzaContext;
53
54 typedef enum channel_offset {
55 RED = 2,
56 GREEN = 1,
57 BLUE = 0,
58 } channel_offset;
59
60 typedef struct rgb {
61 uint8_t r;
62 uint8_t g;
63 uint8_t b;
64 } rgb;
65
66 #define SQR(x) ((x) * (x))
67
68 /* 15 bit components */
69 #define GET_CHAN(color, chan) (((color) >> ((chan) * 5) & 0x1F))
70 #define R(color) GET_CHAN(color, RED)
71 #define G(color) GET_CHAN(color, GREEN)
72 #define B(color) GET_CHAN(color, BLUE)
73
74 typedef struct BlockInfo {
75 int row;
76 int col;
77 int block_width;
78 int block_height;
79 int image_width;
80 int image_height;
81 int block_index;
82 uint16_t start;
83 int rowstride;
84 int prev_rowstride;
85 int blocks_per_row;
86 int total_blocks;
87 } BlockInfo;
88
89 static void get_colors(const uint8_t *min, const uint8_t *max, uint8_t color4[4][3])
90 {
91 uint8_t step;
92
93 color4[0][0] = min[0];
94 color4[0][1] = min[1];
95 color4[0][2] = min[2];
96
97 color4[3][0] = max[0];
98 color4[3][1] = max[1];
99 color4[3][2] = max[2];
100
101 // red components
102 step = (color4[3][0] - color4[0][0] + 1) / 3;
103 color4[1][0] = color4[0][0] + step;
104 color4[2][0] = color4[3][0] - step;
105
106 // green components
107 step = (color4[3][1] - color4[0][1] + 1) / 3;
108 color4[1][1] = color4[0][1] + step;
109 color4[2][1] = color4[3][1] - step;
110
111 // blue components
112 step = (color4[3][2] - color4[0][2] + 1) / 3;
113 color4[1][2] = color4[0][2] + step;
114 color4[2][2] = color4[3][2] - step;
115 }
116
117 /* Fill BlockInfo struct with information about a 4x4 block of the image */
118 3808762 static int get_block_info(BlockInfo *bi, int block, int prev_frame)
119 {
120 3808762 bi->row = block / bi->blocks_per_row;
121 3808762 bi->col = block % bi->blocks_per_row;
122
123 // test for right edge block
124
4/4
✓ Branch 0 taken 45240 times.
✓ Branch 1 taken 3763522 times.
✓ Branch 2 taken 1800 times.
✓ Branch 3 taken 43440 times.
3808762 if (bi->col == bi->blocks_per_row - 1 && (bi->image_width % 4) != 0) {
125 1800 bi->block_width = bi->image_width % 4;
126 } else {
127 3806962 bi->block_width = 4;
128 }
129
130 // test for bottom edge block
131
3/4
✓ Branch 0 taken 1800 times.
✓ Branch 1 taken 3806962 times.
✓ Branch 2 taken 1800 times.
✗ Branch 3 not taken.
3808762 if (bi->row == (bi->image_height / 4) && (bi->image_height % 4) != 0) {
132 1800 bi->block_height = bi->image_height % 4;
133 } else {
134 3806962 bi->block_height = 4;
135 }
136
137
4/4
✓ Branch 0 taken 3807990 times.
✓ Branch 1 taken 772 times.
✓ Branch 2 taken 1903995 times.
✓ Branch 3 taken 1903995 times.
3808762 return block ? (bi->col * 4) + (bi->row * (prev_frame ? bi->prev_rowstride : bi->rowstride) * 4) : 0;
138 }
139
140 98843 static uint16_t rgb24_to_rgb555(const uint8_t *rgb24)
141 {
142 98843 uint16_t rgb555 = 0;
143 uint32_t r, g, b;
144
145 98843 r = rgb24[0];
146 98843 g = rgb24[1];
147 98843 b = rgb24[2];
148
149 98843 rgb555 |= (r << 10);
150 98843 rgb555 |= (g << 5);
151 98843 rgb555 |= (b << 0);
152
153 98843 return rgb555;
154 }
155
156 /*
157 * Returns the total difference between two 24 bit color values
158 */
159 static int diff_colors(const uint8_t *colorA, const uint8_t *colorB)
160 {
161 int tot;
162
163 tot = SQR(colorA[0] - colorB[0]);
164 tot += SQR(colorA[1] - colorB[1]);
165 tot += SQR(colorA[2] - colorB[2]);
166
167 return tot;
168 }
169
170 /*
171 * Returns the maximum channel difference
172 */
173 1275563 static int max_component_diff(const uint16_t *colorA, const uint16_t *colorB)
174 {
175 1275563 int diff, max = 0;
176
177 1275563 diff = FFABS(R(colorA[0]) - R(colorB[0]));
178
2/2
✓ Branch 0 taken 674458 times.
✓ Branch 1 taken 601105 times.
1275563 if (diff > max) {
179 674458 max = diff;
180 }
181 1275563 diff = FFABS(G(colorA[0]) - G(colorB[0]));
182
2/2
✓ Branch 0 taken 454108 times.
✓ Branch 1 taken 821455 times.
1275563 if (diff > max) {
183 454108 max = diff;
184 }
185 1275563 diff = FFABS(B(colorA[0]) - B(colorB[0]));
186
2/2
✓ Branch 0 taken 187367 times.
✓ Branch 1 taken 1088196 times.
1275563 if (diff > max) {
187 187367 max = diff;
188 }
189 1275563 return max;
190 }
191
192 /*
193 * Find the channel that has the largest difference between minimum and maximum
194 * color values. Put the minimum value in min, maximum in max and the channel
195 * in chan.
196 */
197 851101 static void get_max_component_diff(const BlockInfo *bi, const uint16_t *block_ptr,
198 uint8_t *min, uint8_t *max, channel_offset *chan)
199 {
200 int x, y;
201 uint8_t min_r, max_r, min_g, max_g, min_b, max_b;
202 uint8_t r, g, b;
203
204 // fix warning about uninitialized vars
205 851101 min_r = min_g = min_b = UINT8_MAX;
206 851101 max_r = max_g = max_b = 0;
207
208 // loop thru and compare pixels
209
2/2
✓ Branch 0 taken 3403504 times.
✓ Branch 1 taken 851101 times.
4254605 for (y = 0; y < bi->block_height; y++) {
210
2/2
✓ Branch 0 taken 13610616 times.
✓ Branch 1 taken 3403504 times.
17014120 for (x = 0; x < bi->block_width; x++) {
211 // TODO: optimize
212 13610616 min_r = FFMIN(R(block_ptr[x]), min_r);
213 13610616 min_g = FFMIN(G(block_ptr[x]), min_g);
214 13610616 min_b = FFMIN(B(block_ptr[x]), min_b);
215
216 13610616 max_r = FFMAX(R(block_ptr[x]), max_r);
217 13610616 max_g = FFMAX(G(block_ptr[x]), max_g);
218 13610616 max_b = FFMAX(B(block_ptr[x]), max_b);
219 }
220 3403504 block_ptr += bi->rowstride;
221 }
222
223 851101 r = max_r - min_r;
224 851101 g = max_g - min_g;
225 851101 b = max_b - min_b;
226
227
4/4
✓ Branch 0 taken 206436 times.
✓ Branch 1 taken 644665 times.
✓ Branch 2 taken 170355 times.
✓ Branch 3 taken 36081 times.
851101 if (r > g && r > b) {
228 170355 *max = max_r;
229 170355 *min = min_r;
230 170355 *chan = RED;
231
3/4
✓ Branch 0 taken 404520 times.
✓ Branch 1 taken 276226 times.
✓ Branch 2 taken 404520 times.
✗ Branch 3 not taken.
680746 } else if (g > b && g >= r) {
232 404520 *max = max_g;
233 404520 *min = min_g;
234 404520 *chan = GREEN;
235 } else {
236 276226 *max = max_b;
237 276226 *min = min_b;
238 276226 *chan = BLUE;
239 }
240 851101 }
241
242 /*
243 * Compare two 4x4 blocks to determine if the total difference between the
244 * blocks is greater than the thresh parameter. Returns -1 if difference
245 * exceeds threshold or zero otherwise.
246 */
247 934633 static int compare_blocks(const uint16_t *block1, const uint16_t *block2,
248 const BlockInfo *bi, int thresh)
249 {
250 934633 int x, y, diff = 0;
251
2/2
✓ Branch 0 taken 974644 times.
✓ Branch 1 taken 3682 times.
978326 for (y = 0; y < bi->block_height; y++) {
252
2/2
✓ Branch 0 taken 1275563 times.
✓ Branch 1 taken 43693 times.
1319256 for (x = 0; x < bi->block_width; x++) {
253 1275563 diff = max_component_diff(&block1[x], &block2[x]);
254
2/2
✓ Branch 0 taken 930951 times.
✓ Branch 1 taken 344612 times.
1275563 if (diff >= thresh) {
255 930951 return -1;
256 }
257 }
258 43693 block1 += bi->prev_rowstride;
259 43693 block2 += bi->rowstride;
260 }
261 3682 return 0;
262 }
263
264 /*
265 * Determine the fit of one channel to another within a 4x4 block. This
266 * is used to determine the best palette choices for 4-color encoding.
267 */
268 1702202 static int leastsquares(const uint16_t *block_ptr, const BlockInfo *bi,
269 channel_offset xchannel, channel_offset ychannel,
270 int *slope, int *y_intercept, int *correlation_coef)
271 {
272 1702202 int sumx = 0, sumy = 0, sumx2 = 0, sumy2 = 0, sumxy = 0,
273 1702202 sumx_sq = 0, sumy_sq = 0, tmp, tmp2;
274 int i, j, count;
275 uint8_t x, y;
276
277 1702202 count = bi->block_height * bi->block_width;
278
279
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1702202 times.
1702202 if (count < 2)
280 return -1;
281
282
2/2
✓ Branch 0 taken 6807008 times.
✓ Branch 1 taken 1702202 times.
8509210 for (i = 0; i < bi->block_height; i++) {
283
2/2
✓ Branch 0 taken 27221232 times.
✓ Branch 1 taken 6807008 times.
34028240 for (j = 0; j < bi->block_width; j++) {
284 27221232 x = GET_CHAN(block_ptr[j], xchannel);
285 27221232 y = GET_CHAN(block_ptr[j], ychannel);
286 27221232 sumx += x;
287 27221232 sumy += y;
288 27221232 sumx2 += x * x;
289 27221232 sumy2 += y * y;
290 27221232 sumxy += x * y;
291 }
292 6807008 block_ptr += bi->rowstride;
293 }
294
295 1702202 sumx_sq = sumx * sumx;
296 1702202 tmp = (count * sumx2 - sumx_sq);
297
298 // guard against div/0
299
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1702202 times.
1702202 if (tmp == 0)
300 return -2;
301
302 1702202 sumy_sq = sumy * sumy;
303
304 1702202 *slope = (sumx * sumy - sumxy) / tmp;
305 1702202 *y_intercept = (sumy - (*slope) * sumx) / count;
306
307 1702202 tmp2 = count * sumy2 - sumy_sq;
308
2/2
✓ Branch 0 taken 42945 times.
✓ Branch 1 taken 1659257 times.
1702202 if (tmp2 == 0) {
309 42945 *correlation_coef = 0;
310 } else {
311 1659257 *correlation_coef = (count * sumxy - sumx * sumy) /
312 1659257 ff_sqrt((unsigned)tmp * tmp2);
313 }
314
315 1702202 return 0; // success
316 }
317
318 /*
319 * Determine the amount of error in the leastsquares fit.
320 */
321 3030648 static int calc_lsq_max_fit_error(const uint16_t *block_ptr, const BlockInfo *bi,
322 int min, int max, int tmp_min, int tmp_max,
323 channel_offset xchannel, channel_offset ychannel)
324 {
325 int i, j, x, y;
326 int err;
327 3030648 int max_err = 0;
328
329
2/2
✓ Branch 0 taken 12119460 times.
✓ Branch 1 taken 3030648 times.
15150108 for (i = 0; i < bi->block_height; i++) {
330
2/2
✓ Branch 0 taken 48465784 times.
✓ Branch 1 taken 12119460 times.
60585244 for (j = 0; j < bi->block_width; j++) {
331 int x_inc, lin_y, lin_x;
332 48465784 x = GET_CHAN(block_ptr[j], xchannel);
333 48465784 y = GET_CHAN(block_ptr[j], ychannel);
334
335 /* calculate x_inc as the 4-color index (0..3) */
336 48465784 x_inc = (x - min) * 3 / (max - min) + 1;
337
1/2
✓ Branch 0 taken 48465784 times.
✗ Branch 1 not taken.
48465784 x_inc = FFMAX(FFMIN(3, x_inc), 0);
338
339 /* calculate lin_y corresponding to x_inc */
340 48465784 lin_y = tmp_min + (tmp_max - tmp_min) * x_inc / 3 + 1;
341
342 48465784 err = FFABS(lin_y - y);
343
2/2
✓ Branch 0 taken 8561490 times.
✓ Branch 1 taken 39904294 times.
48465784 if (err > max_err)
344 8561490 max_err = err;
345
346 /* calculate lin_x corresponding to x_inc */
347 48465784 lin_x = min + (max - min) * x_inc / 3 + 1;
348
349 48465784 err = FFABS(lin_x - x);
350
2/2
✓ Branch 0 taken 238381 times.
✓ Branch 1 taken 48227403 times.
48465784 if (err > max_err)
351 238381 max_err += err;
352 }
353 12119460 block_ptr += bi->rowstride;
354 }
355
356 3030648 return max_err;
357 }
358
359 /*
360 * Find the closest match to a color within the 4-color palette
361 */
362 static int match_color(const uint16_t *color, uint8_t colors[4][3])
363 {
364 int ret = 0;
365 int smallest_variance = INT_MAX;
366 uint8_t dithered_color[3];
367
368 for (int channel = 0; channel < 3; channel++) {
369 dithered_color[channel] = GET_CHAN(color[0], channel);
370 }
371
372 for (int palette_entry = 0; palette_entry < 4; palette_entry++) {
373 int variance = diff_colors(dithered_color, colors[palette_entry]);
374
375 if (variance < smallest_variance) {
376 smallest_variance = variance;
377 ret = palette_entry;
378 }
379 }
380
381 return ret;
382 }
383
384 /*
385 * Encode a block using the 4-color opcode and palette. return number of
386 * blocks encoded (until we implement multi-block 4 color runs this will
387 * always be 1)
388 */
389 static int encode_four_color_block(const uint8_t *min_color, const uint8_t *max_color,
390 PutBitContext *pb, const uint16_t *block_ptr, const BlockInfo *bi)
391 {
392 const int y_size = FFMIN(4, bi->image_height - bi->row * 4);
393 const int x_size = FFMIN(4, bi->image_width - bi->col * 4);
394 uint8_t color4[4][3];
395 uint16_t rounded_max, rounded_min;
396 int idx;
397
398 // round min and max wider
399 rounded_min = rgb24_to_rgb555(min_color);
400 rounded_max = rgb24_to_rgb555(max_color);
401
402 // put a and b colors
403 // encode 4 colors = first 16 bit color with MSB zeroed and...
404 put_bits(pb, 16, rounded_max & ~0x8000);
405 // ...second 16 bit color with MSB on.
406 put_bits(pb, 16, rounded_min | 0x8000);
407
408 get_colors(min_color, max_color, color4);
409
410 for (int y = 0; y < y_size; y++) {
411 for (int x = 0; x < x_size; x++) {
412 idx = match_color(&block_ptr[x], color4);
413 put_bits(pb, 2, idx);
414 }
415
416 for (int x = x_size; x < 4; x++)
417 put_bits(pb, 2, idx);
418 block_ptr += bi->rowstride;
419 }
420
421 for (int y = y_size; y < 4; y++) {
422 for (int x = 0; x < 4; x++)
423 put_bits(pb, 2, 0);
424 }
425 return 1; // num blocks encoded
426 }
427
428 /*
429 * Copy a 4x4 block from the current frame buffer to the previous frame buffer.
430 */
431 950768 static void update_block_in_prev_frame(const uint16_t *src_pixels,
432 uint16_t *dest_pixels,
433 const BlockInfo *bi, int block_counter)
434 {
435 950768 const int y_size = FFMIN(4, bi->image_height - bi->row * 4);
436 950768 const int x_size = FFMIN(4, bi->image_width - bi->col * 4) * 2;
437
438
2/2
✓ Branch 0 taken 3802172 times.
✓ Branch 1 taken 950768 times.
4752940 for (int y = 0; y < y_size; y++) {
439 3802172 memcpy(dest_pixels, src_pixels, x_size);
440 3802172 dest_pixels += bi->prev_rowstride;
441 3802172 src_pixels += bi->rowstride;
442 }
443 950768 }
444
445 /*
446 * update statistics for the specified block. If first_block,
447 * it initializes the statistics. Otherwise it updates the statistics IF THIS
448 * BLOCK IS SUITABLE TO CONTINUE A 1-COLOR RUN. That is, it checks whether
449 * the range of colors (since the routine was called first_block != 0) are
450 * all close enough intensities to be represented by a single color.
451
452 * The routine returns 0 if this block is too different to be part of
453 * the same run of 1-color blocks. The routine returns 1 if this
454 * block can be part of the same 1-color block run.
455
456 * If the routine returns 1, it also updates its arguments to include
457 * the statistics of this block. Otherwise, the stats are unchanged
458 * and don't include the current block.
459 */
460 1048672 static int update_block_stats(RpzaContext *s, const BlockInfo *bi, const uint16_t *block,
461 uint8_t min_color[3], uint8_t max_color[3],
462 int *total_rgb, int *total_pixels,
463 uint8_t avg_color[3], int first_block)
464 {
465 int x, y;
466 int is_in_range;
467 int total_pixels_blk;
468 int threshold;
469
470 uint8_t min_color_blk[3], max_color_blk[3];
471 int total_rgb_blk[3];
472 uint8_t avg_color_blk[3];
473
474
2/2
✓ Branch 0 taken 949944 times.
✓ Branch 1 taken 98728 times.
1048672 if (first_block) {
475 949944 min_color[0] = UINT8_MAX;
476 949944 min_color[1] = UINT8_MAX;
477 949944 min_color[2] = UINT8_MAX;
478 949944 max_color[0] = 0;
479 949944 max_color[1] = 0;
480 949944 max_color[2] = 0;
481 949944 total_rgb[0] = 0;
482 949944 total_rgb[1] = 0;
483 949944 total_rgb[2] = 0;
484 949944 *total_pixels = 0;
485 949944 threshold = s->start_one_color_thresh;
486 } else {
487 98728 threshold = s->continue_one_color_thresh;
488 }
489
490 /*
491 The *_blk variables will include the current block.
492 Initialize them based on the blocks so far.
493 */
494 1048672 min_color_blk[0] = min_color[0];
495 1048672 min_color_blk[1] = min_color[1];
496 1048672 min_color_blk[2] = min_color[2];
497 1048672 max_color_blk[0] = max_color[0];
498 1048672 max_color_blk[1] = max_color[1];
499 1048672 max_color_blk[2] = max_color[2];
500 1048672 total_rgb_blk[0] = total_rgb[0];
501 1048672 total_rgb_blk[1] = total_rgb[1];
502 1048672 total_rgb_blk[2] = total_rgb[2];
503 1048672 total_pixels_blk = *total_pixels + bi->block_height * bi->block_width;
504
505 /*
506 Update stats for this block's pixels
507 */
508
2/2
✓ Branch 0 taken 4193788 times.
✓ Branch 1 taken 1048672 times.
5242460 for (y = 0; y < bi->block_height; y++) {
509
2/2
✓ Branch 0 taken 16771752 times.
✓ Branch 1 taken 4193788 times.
20965540 for (x = 0; x < bi->block_width; x++) {
510 16771752 total_rgb_blk[0] += R(block[x]);
511 16771752 total_rgb_blk[1] += G(block[x]);
512 16771752 total_rgb_blk[2] += B(block[x]);
513
514 16771752 min_color_blk[0] = FFMIN(R(block[x]), min_color_blk[0]);
515 16771752 min_color_blk[1] = FFMIN(G(block[x]), min_color_blk[1]);
516 16771752 min_color_blk[2] = FFMIN(B(block[x]), min_color_blk[2]);
517
518 16771752 max_color_blk[0] = FFMAX(R(block[x]), max_color_blk[0]);
519 16771752 max_color_blk[1] = FFMAX(G(block[x]), max_color_blk[1]);
520 16771752 max_color_blk[2] = FFMAX(B(block[x]), max_color_blk[2]);
521 }
522 4193788 block += bi->rowstride;
523 }
524
525 /*
526 Calculate average color including current block.
527 */
528 1048672 avg_color_blk[0] = total_rgb_blk[0] / total_pixels_blk;
529 1048672 avg_color_blk[1] = total_rgb_blk[1] / total_pixels_blk;
530 1048672 avg_color_blk[2] = total_rgb_blk[2] / total_pixels_blk;
531
532 /*
533 Are all the pixels within threshold of the average color?
534 */
535 2358988 is_in_range = (max_color_blk[0] - avg_color_blk[0] <= threshold &&
536
2/2
✓ Branch 0 taken 149748 times.
✓ Branch 1 taken 111896 times.
261644 max_color_blk[1] - avg_color_blk[1] <= threshold &&
537
2/2
✓ Branch 0 taken 104996 times.
✓ Branch 1 taken 44752 times.
149748 max_color_blk[2] - avg_color_blk[2] <= threshold &&
538
2/2
✓ Branch 0 taken 103897 times.
✓ Branch 1 taken 1099 times.
104996 avg_color_blk[0] - min_color_blk[0] <= threshold &&
539
4/4
✓ Branch 0 taken 261644 times.
✓ Branch 1 taken 787028 times.
✓ Branch 2 taken 102084 times.
✓ Branch 3 taken 1813 times.
1412400 avg_color_blk[1] - min_color_blk[1] <= threshold &&
540
2/2
✓ Branch 0 taken 99667 times.
✓ Branch 1 taken 2417 times.
102084 avg_color_blk[2] - min_color_blk[2] <= threshold);
541
542
2/2
✓ Branch 0 taken 99667 times.
✓ Branch 1 taken 949005 times.
1048672 if (is_in_range) {
543 /*
544 Set the output variables to include this block.
545 */
546 99667 min_color[0] = min_color_blk[0];
547 99667 min_color[1] = min_color_blk[1];
548 99667 min_color[2] = min_color_blk[2];
549 99667 max_color[0] = max_color_blk[0];
550 99667 max_color[1] = max_color_blk[1];
551 99667 max_color[2] = max_color_blk[2];
552 99667 total_rgb[0] = total_rgb_blk[0];
553 99667 total_rgb[1] = total_rgb_blk[1];
554 99667 total_rgb[2] = total_rgb_blk[2];
555 99667 *total_pixels = total_pixels_blk;
556 99667 avg_color[0] = avg_color_blk[0];
557 99667 avg_color[1] = avg_color_blk[1];
558 99667 avg_color[2] = avg_color_blk[2];
559 }
560
561 1048672 return is_in_range;
562 }
563
564 200 static void rpza_encode_stream(RpzaContext *s, const AVFrame *pict)
565 {
566 BlockInfo bi;
567 200 int block_counter = 0;
568 int n_blocks;
569 int total_blocks;
570 int prev_block_offset;
571 200 int block_offset = 0;
572 200 int pblock_offset = 0;
573 200 uint8_t min = 0, max = 0;
574 channel_offset chan;
575 int i;
576 int tmp_min, tmp_max;
577 int total_rgb[3];
578 uint8_t avg_color[3];
579 int pixel_count;
580 uint8_t min_color[3], max_color[3];
581 int slope, y_intercept, correlation_coef;
582 200 const uint16_t *src_pixels = (const uint16_t *)pict->data[0];
583 200 uint16_t *prev_pixels = (uint16_t *)s->prev_frame->data[0];
584
585 /* Number of 4x4 blocks in frame. */
586 200 total_blocks = ((s->frame_width + 3) / 4) * ((s->frame_height + 3) / 4);
587
588 200 bi.image_width = s->frame_width;
589 200 bi.image_height = s->frame_height;
590 200 bi.rowstride = pict->linesize[0] / 2;
591 200 bi.prev_rowstride = s->prev_frame->linesize[0] / 2;
592
593 200 bi.blocks_per_row = (s->frame_width + 3) / 4;
594
595
2/2
✓ Branch 0 taken 949947 times.
✓ Branch 1 taken 200 times.
950147 while (block_counter < total_blocks) {
596 // SKIP CHECK
597 // make sure we have a valid previous frame and we're not writing
598 // a key frame
599
2/2
✓ Branch 0 taken 930954 times.
✓ Branch 1 taken 18993 times.
949947 if (!s->first_frame) {
600 930954 n_blocks = 0;
601 930954 prev_block_offset = 0;
602
603
2/4
✓ Branch 0 taken 934636 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 934636 times.
✗ Branch 3 not taken.
934636 while (n_blocks < 32 && block_counter + n_blocks < total_blocks) {
604 934636 block_offset = get_block_info(&bi, block_counter + n_blocks, 0);
605 934636 pblock_offset = get_block_info(&bi, block_counter + n_blocks, 1);
606
607 // multi-block opcodes cannot span multiple rows.
608 // If we're starting a new row, break out and write the opcode
609 /* TODO: Should eventually use bi.row here to determine when a
610 row break occurs, but that is currently breaking the
611 quicktime player. This is probably due to a bug in the
612 way I'm calculating the current row.
613 */
614
4/4
✓ Branch 0 taken 3682 times.
✓ Branch 1 taken 930954 times.
✓ Branch 2 taken 3 times.
✓ Branch 3 taken 3679 times.
934636 if (prev_block_offset && block_offset - prev_block_offset > 12) {
615 3 break;
616 }
617
618 934633 prev_block_offset = block_offset;
619
620
2/2
✓ Branch 0 taken 930951 times.
✓ Branch 1 taken 3682 times.
934633 if (compare_blocks(&prev_pixels[pblock_offset],
621 934633 &src_pixels[block_offset], &bi, s->skip_frame_thresh) != 0) {
622 // write out skipable blocks
623
2/2
✓ Branch 0 taken 2314 times.
✓ Branch 1 taken 928637 times.
930951 if (n_blocks) {
624
625 // write skip opcode
626 2314 put_bits(&s->pb, 8, 0x80 | (n_blocks - 1));
627 2314 block_counter += n_blocks;
628
629 2314 goto post_skip;
630 }
631 928637 break;
632 }
633
634 /*
635 * NOTE: we don't update skipped blocks in the previous frame buffer
636 * since skipped needs always to be compared against the first skipped
637 * block to avoid artifacts during gradual fade in/outs.
638 */
639
640 // update_block_in_prev_frame(&src_pixels[block_offset],
641 // &prev_pixels[pblock_offset], &bi, block_counter + n_blocks);
642
643 3682 n_blocks++;
644 }
645
646 // we're either at the end of the frame or we've reached the maximum
647 // of 32 blocks in a run. Write out the run.
648
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 928637 times.
928640 if (n_blocks) {
649 // write skip opcode
650 3 put_bits(&s->pb, 8, 0x80 | (n_blocks - 1));
651 3 block_counter += n_blocks;
652
653 3 continue;
654 }
655
656 } else {
657 18993 block_offset = get_block_info(&bi, block_counter, 0);
658 18993 pblock_offset = get_block_info(&bi, block_counter, 1);
659 }
660 949944 post_skip :
661
662 // ONE COLOR CHECK
663
2/2
✓ Branch 1 taken 98843 times.
✓ Branch 2 taken 851101 times.
949944 if (update_block_stats(s, &bi, &src_pixels[block_offset],
664 min_color, max_color,
665 total_rgb, &pixel_count, avg_color, 1)) {
666 98843 prev_block_offset = block_offset;
667
668 98843 n_blocks = 1;
669
670 /* update this block in the previous frame buffer */
671 98843 update_block_in_prev_frame(&src_pixels[block_offset],
672 98843 &prev_pixels[pblock_offset], &bi, block_counter + n_blocks);
673
674 // check for subsequent blocks with the same color
675
3/4
✓ Branch 0 taken 99667 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 99651 times.
✓ Branch 3 taken 16 times.
99667 while (n_blocks < 32 && block_counter + n_blocks < total_blocks) {
676 99651 block_offset = get_block_info(&bi, block_counter + n_blocks, 0);
677 99651 pblock_offset = get_block_info(&bi, block_counter + n_blocks, 1);
678
679 // multi-block opcodes cannot span multiple rows.
680 // If we've hit end of a row, break out and write the opcode
681
2/2
✓ Branch 0 taken 923 times.
✓ Branch 1 taken 98728 times.
99651 if (block_offset - prev_block_offset > 12) {
682 923 break;
683 }
684
685
2/2
✓ Branch 1 taken 97904 times.
✓ Branch 2 taken 824 times.
98728 if (!update_block_stats(s, &bi, &src_pixels[block_offset],
686 min_color, max_color,
687 total_rgb, &pixel_count, avg_color, 0)) {
688 97904 break;
689 }
690
691 824 prev_block_offset = block_offset;
692
693 /* update this block in the previous frame buffer */
694 824 update_block_in_prev_frame(&src_pixels[block_offset],
695 824 &prev_pixels[pblock_offset], &bi, block_counter + n_blocks);
696
697 824 n_blocks++;
698 }
699
700 // write one color opcode.
701 98843 put_bits(&s->pb, 8, 0xa0 | (n_blocks - 1));
702 // write color to encode.
703 98843 put_bits(&s->pb, 16, rgb24_to_rgb555(avg_color));
704 // skip past the blocks we've just encoded.
705 98843 block_counter += n_blocks;
706 } else { // FOUR COLOR CHECK
707 851101 int err = 0;
708
709 // get max component diff for block
710 851101 get_max_component_diff(&bi, &src_pixels[block_offset], &min, &max, &chan);
711
712 851101 min_color[0] = 0;
713 851101 max_color[0] = 0;
714 851101 min_color[1] = 0;
715 851101 max_color[1] = 0;
716 851101 min_color[2] = 0;
717 851101 max_color[2] = 0;
718
719 // run least squares against other two components
720
2/2
✓ Branch 0 taken 2553303 times.
✓ Branch 1 taken 851101 times.
3404404 for (i = 0; i < 3; i++) {
721
2/2
✓ Branch 0 taken 851101 times.
✓ Branch 1 taken 1702202 times.
2553303 if (i == chan) {
722 851101 min_color[i] = min;
723 851101 max_color[i] = max;
724 851101 continue;
725 }
726
727 1702202 slope = y_intercept = correlation_coef = 0;
728
729
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1702202 times.
1702202 if (leastsquares(&src_pixels[block_offset], &bi, chan, i,
730 &slope, &y_intercept, &correlation_coef)) {
731 min_color[i] = GET_CHAN(src_pixels[block_offset], i);
732 max_color[i] = GET_CHAN(src_pixels[block_offset], i);
733 } else {
734 1702202 tmp_min = 1 + min * slope + y_intercept;
735 1702202 tmp_max = 1 + max * slope + y_intercept;
736
737
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1702202 times.
1702202 av_assert0(tmp_min <= tmp_max);
738 // clamp min and max color values
739 1702202 tmp_min = av_clip_uint8(tmp_min);
740 1702202 tmp_max = av_clip_uint8(tmp_max);
741
742
2/2
✓ Branch 1 taken 1328446 times.
✓ Branch 2 taken 373756 times.
1702202 err = FFMAX(calc_lsq_max_fit_error(&src_pixels[block_offset], &bi,
743 min, max, tmp_min, tmp_max, chan, i), err);
744
745 1702202 min_color[i] = tmp_min;
746 1702202 max_color[i] = tmp_max;
747 }
748 }
749
750
1/2
✓ Branch 0 taken 851101 times.
✗ Branch 1 not taken.
851101 if (err > s->sixteen_color_thresh) { // DO SIXTEEN COLOR BLOCK
751 const uint16_t *row_ptr;
752 int y_size, rgb555;
753
754 851101 block_offset = get_block_info(&bi, block_counter, 0);
755 851101 pblock_offset = get_block_info(&bi, block_counter, 1);
756
757 851101 row_ptr = &src_pixels[block_offset];
758 851101 y_size = FFMIN(4, bi.image_height - bi.row * 4);
759
760
2/2
✓ Branch 0 taken 3403504 times.
✓ Branch 1 taken 851101 times.
4254605 for (int y = 0; y < y_size; y++) {
761
2/2
✓ Branch 0 taken 13614016 times.
✓ Branch 1 taken 3403504 times.
17017520 for (int x = 0; x < 4; x++) {
762 13614016 rgb555 = row_ptr[x] & ~0x8000;
763
764 13614016 put_bits(&s->pb, 16, rgb555);
765 }
766 3403504 row_ptr += bi.rowstride;
767 }
768
769
2/2
✓ Branch 0 taken 900 times.
✓ Branch 1 taken 851101 times.
852001 for (int y = y_size; y < 4; y++) {
770
2/2
✓ Branch 0 taken 3600 times.
✓ Branch 1 taken 900 times.
4500 for (int x = 0; x < 4; x++)
771 3600 put_bits(&s->pb, 16, 0);
772 }
773
774 851101 block_counter++;
775 } else { // FOUR COLOR BLOCK
776 block_counter += encode_four_color_block(min_color, max_color,
777 &s->pb, &src_pixels[block_offset], &bi);
778 }
779
780 /* update this block in the previous frame buffer */
781 851101 update_block_in_prev_frame(&src_pixels[block_offset],
782 851101 &prev_pixels[pblock_offset], &bi, block_counter);
783 }
784 }
785 200 }
786
787 4 static int rpza_encode_init(AVCodecContext *avctx)
788 {
789 4 RpzaContext *s = avctx->priv_data;
790
791 4 s->frame_width = avctx->width;
792 4 s->frame_height = avctx->height;
793
794 4 s->prev_frame = av_frame_alloc();
795
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
4 if (!s->prev_frame)
796 return AVERROR(ENOMEM);
797
798 4 return 0;
799 }
800
801 200 static int rpza_encode_frame(AVCodecContext *avctx, AVPacket *pkt,
802 const AVFrame *pict, int *got_packet)
803 {
804 200 RpzaContext *s = avctx->priv_data;
805 uint8_t *buf;
806 200 int ret = ff_alloc_packet(avctx, pkt, 4LL + 6LL * FFMAX(avctx->height, 4) * FFMAX(avctx->width, 4));
807
808
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 200 times.
200 if (ret < 0)
809 return ret;
810
811 200 init_put_bits(&s->pb, pkt->data, pkt->size);
812
813 // skip 4 byte header, write it later once the size of the chunk is known
814 200 put_bits32(&s->pb, 0x00);
815
816
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 196 times.
200 if (!s->prev_frame->data[0]) {
817 4 s->first_frame = 1;
818 4 s->prev_frame->format = pict->format;
819 4 s->prev_frame->width = pict->width;
820 4 s->prev_frame->height = pict->height;
821 4 ret = av_frame_get_buffer(s->prev_frame, 0);
822
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
4 if (ret < 0)
823 return ret;
824 } else {
825 196 s->first_frame = 0;
826 }
827
828 200 rpza_encode_stream(s, pict);
829
830 200 flush_put_bits(&s->pb);
831
832 200 av_shrink_packet(pkt, put_bytes_output(&s->pb));
833 200 buf = pkt->data;
834
835 // write header opcode
836 200 buf[0] = 0xe1; // chunk opcode
837
838 // write chunk length
839 200 AV_WB24(buf + 1, pkt->size);
840
841 200 *got_packet = 1;
842
843 200 return 0;
844 }
845
846 4 static int rpza_encode_end(AVCodecContext *avctx)
847 {
848 4 RpzaContext *s = (RpzaContext *)avctx->priv_data;
849
850 4 av_frame_free(&s->prev_frame);
851
852 4 return 0;
853 }
854
855 #define OFFSET(x) offsetof(RpzaContext, x)
856 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
857 static const AVOption options[] = {
858 { "skip_frame_thresh", NULL, OFFSET(skip_frame_thresh), AV_OPT_TYPE_INT, {.i64=1}, 0, 24, VE},
859 { "start_one_color_thresh", NULL, OFFSET(start_one_color_thresh), AV_OPT_TYPE_INT, {.i64=1}, 0, 24, VE},
860 { "continue_one_color_thresh", NULL, OFFSET(continue_one_color_thresh), AV_OPT_TYPE_INT, {.i64=0}, 0, 24, VE},
861 { "sixteen_color_thresh", NULL, OFFSET(sixteen_color_thresh), AV_OPT_TYPE_INT, {.i64=1}, 0, 24, VE},
862 { NULL },
863 };
864
865 static const AVClass rpza_class = {
866 .class_name = "rpza",
867 .item_name = av_default_item_name,
868 .option = options,
869 .version = LIBAVUTIL_VERSION_INT,
870 };
871
872 const FFCodec ff_rpza_encoder = {
873 .p.name = "rpza",
874 CODEC_LONG_NAME("QuickTime video (RPZA)"),
875 .p.type = AVMEDIA_TYPE_VIDEO,
876 .p.id = AV_CODEC_ID_RPZA,
877 .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
878 .priv_data_size = sizeof(RpzaContext),
879 .p.priv_class = &rpza_class,
880 .init = rpza_encode_init,
881 FF_CODEC_ENCODE_CB(rpza_encode_frame),
882 .close = rpza_encode_end,
883 .p.pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_RGB555,
884 AV_PIX_FMT_NONE},
885 };
886