94 uint8_t *dest_cr,
int mb_x,
int mb_y)
96 int dc, dcu, dcv, y, i;
97 for (i = 0; i < 4; i++) {
98 dc = s->
dc_val[0][mb_x * 2 + (i & 1) + (mb_y * 2 + (i >> 1)) * s->
b8_stride];
103 for (y = 0; y < 8; y++) {
105 for (x = 0; x < 8; x++)
106 dest_y[x + (i & 1) * 8 + (y + (i >> 1) * 8) * s->
linesize] = dc / 8;
119 for (y = 0; y < 8; y++) {
121 for (x = 0; x < 8; x++) {
133 for (y = 1; y < height - 1; y++) {
134 int prev_dc = data[0 + y *
stride];
136 for (x = 1; x < width - 1; x++) {
139 data[x + y *
stride] * 8 -
141 dc = (dc * 10923 + 32768) >> 16;
142 prev_dc = data[x + y *
stride];
143 data[x + y *
stride] = dc;
148 for (x = 1; x < width - 1; x++) {
149 int prev_dc = data[x];
151 for (y = 1; y < height - 1; y++) {
155 data[x + y *
stride] * 8 -
156 data[x + (y + 1) * stride];
157 dc = (dc * 10923 + 32768) >> 16;
158 prev_dc = data[x + y *
stride];
159 data[x + y *
stride] = dc;
170 int h,
int stride,
int is_luma)
174 for (b_y = 0; b_y < h; b_y++) {
175 for (b_x = 0; b_x < w; b_x++) {
176 int color[4] = { 1024, 1024, 1024, 1024 };
177 int distance[4] = { 9999, 9999, 9999, 9999 };
178 int mb_index, error, j;
179 int64_t guess, weight_sum;
180 mb_index = (b_x >> is_luma) + (b_y >> is_luma) * s->
mb_stride;
189 for (j = b_x + 1; j < w; j++) {
190 int mb_index_j = (j >> is_luma) + (b_y >> is_luma) * s->
mb_stride;
193 if (intra_j == 0 || !(error_j & ER_DC_ERROR)) {
194 color[0] = dc[j + b_y *
stride];
195 distance[0] = j - b_x;
201 for (j = b_x - 1; j >= 0; j--) {
202 int mb_index_j = (j >> is_luma) + (b_y >> is_luma) * s->
mb_stride;
205 if (intra_j == 0 || !(error_j & ER_DC_ERROR)) {
206 color[1] = dc[j + b_y *
stride];
207 distance[1] = b_x - j;
213 for (j = b_y + 1; j < h; j++) {
214 int mb_index_j = (b_x >> is_luma) + (j >> is_luma) * s->
mb_stride;
218 if (intra_j == 0 || !(error_j & ER_DC_ERROR)) {
219 color[2] = dc[b_x + j *
stride];
220 distance[2] = j - b_y;
226 for (j = b_y - 1; j >= 0; j--) {
227 int mb_index_j = (b_x >> is_luma) + (j >> is_luma) * s->
mb_stride;
230 if (intra_j == 0 || !(error_j & ER_DC_ERROR)) {
231 color[3] = dc[b_x + j *
stride];
232 distance[3] = b_y - j;
239 for (j = 0; j < 4; j++) {
240 int64_t weight = 256 * 256 * 256 * 16 / distance[j];
241 guess += weight * (int64_t) color[j];
242 weight_sum += weight;
244 guess = (guess + weight_sum / 2) / weight_sum;
245 dc[b_x + b_y *
stride] = guess;
256 int h,
int stride,
int is_luma)
258 int b_x, b_y, mvx_stride, mvy_stride;
261 mvx_stride >>= is_luma;
262 mvy_stride *= mvx_stride;
264 for (b_y = 0; b_y < h; b_y++) {
265 for (b_x = 0; b_x < w - 1; b_x++) {
273 int offset = b_x * 8 + b_y * stride * 8;
276 if (!(left_damage || right_damage))
278 if ((!left_intra) && (!right_intra) &&
279 FFABS(left_mv[0] - right_mv[0]) +
280 FFABS(left_mv[1] + right_mv[1]) < 2)
283 for (y = 0; y < 8; y++) {
286 a = dst[offset + 7 + y *
stride] - dst[offset + 6 + y *
stride];
287 b = dst[offset + 8 + y *
stride] - dst[offset + 7 + y *
stride];
288 c = dst[offset + 9 + y *
stride] - dst[offset + 8 + y *
stride];
298 if (!(left_damage && right_damage))
302 dst[offset + 7 + y *
stride] = cm[dst[offset + 7 + y *
stride] + ((d * 7) >> 4)];
303 dst[offset + 6 + y *
stride] = cm[dst[offset + 6 + y *
stride] + ((d * 5) >> 4)];
304 dst[offset + 5 + y *
stride] = cm[dst[offset + 5 + y *
stride] + ((d * 3) >> 4)];
305 dst[offset + 4 + y *
stride] = cm[dst[offset + 4 + y *
stride] + ((d * 1) >> 4)];
308 dst[offset + 8 + y *
stride] = cm[dst[offset + 8 + y *
stride] - ((d * 7) >> 4)];
309 dst[offset + 9 + y *
stride] = cm[dst[offset + 9 + y *
stride] - ((d * 5) >> 4)];
310 dst[offset + 10+ y *
stride] = cm[dst[offset + 10 + y *
stride] - ((d * 3) >> 4)];
311 dst[offset + 11+ y *
stride] = cm[dst[offset + 11 + y *
stride] - ((d * 1) >> 4)];
326 int b_x, b_y, mvx_stride, mvy_stride;
329 mvx_stride >>= is_luma;
330 mvy_stride *= mvx_stride;
332 for (b_y = 0; b_y < h - 1; b_y++) {
333 for (b_x = 0; b_x < w; b_x++) {
341 int offset = b_x * 8 + b_y * stride * 8;
346 if (!(top_damage || bottom_damage))
349 if ((!top_intra) && (!bottom_intra) &&
350 FFABS(top_mv[0] - bottom_mv[0]) +
351 FFABS(top_mv[1] + bottom_mv[1]) < 2)
354 for (x = 0; x < 8; x++) {
357 a = dst[offset + x + 7 *
stride] - dst[offset + x + 6 *
stride];
358 b = dst[offset + x + 8 *
stride] - dst[offset + x + 7 *
stride];
359 c = dst[offset + x + 9 *
stride] - dst[offset + x + 8 *
stride];
369 if (!(top_damage && bottom_damage))
373 dst[offset + x + 7 *
stride] = cm[dst[offset + x + 7 *
stride] + ((d * 7) >> 4)];
374 dst[offset + x + 6 *
stride] = cm[dst[offset + x + 6 *
stride] + ((d * 5) >> 4)];
375 dst[offset + x + 5 *
stride] = cm[dst[offset + x + 5 *
stride] + ((d * 3) >> 4)];
376 dst[offset + x + 4 *
stride] = cm[dst[offset + x + 4 *
stride] + ((d * 1) >> 4)];
379 dst[offset + x + 8 *
stride] = cm[dst[offset + x + 8 *
stride] - ((d * 7) >> 4)];
380 dst[offset + x + 9 *
stride] = cm[dst[offset + x + 9 *
stride] - ((d * 5) >> 4)];
381 dst[offset + x + 10 *
stride] = cm[dst[offset + x + 10 *
stride] - ((d * 3) >> 4)];
382 dst[offset + x + 11 *
stride] = cm[dst[offset + x + 11 *
stride] - ((d * 1) >> 4)];
394 #define MV_UNCHANGED 1
398 int i,
depth, num_avail;
399 int mb_x, mb_y, mot_step, mot_stride;
404 for (i = 0; i < s->
mb_num; i++) {
420 num_avail <= mb_width / 2) {
421 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
425 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
426 const int mb_xy = mb_x + mb_y * s->
mb_stride;
453 for (depth = 0; ; depth++) {
454 int changed,
pass, none_left;
458 for (pass = 0; (changed || pass < 2) && pass < 10; pass++) {
463 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
467 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
468 const int mb_xy = mb_x + mb_y * s->
mb_stride;
469 int mv_predictor[8][2] = { { 0 } };
473 int best_score = 256 * 256 * 256 * 64;
475 const int mot_index = (mb_x + mb_y * mot_stride) * mot_step;
476 int prev_x, prev_y, prev_ref;
480 if ((mb_x ^ mb_y ^ pass) & 1)
489 if (mb_x > 0 && fixed[mb_xy - 1] ==
MV_FROZEN)
491 if (mb_x + 1 < mb_width && fixed[mb_xy + 1] ==
MV_FROZEN)
493 if (mb_y > 0 && fixed[mb_xy - mb_stride] ==
MV_FROZEN)
495 if (mb_y + 1 < mb_height && fixed[mb_xy + mb_stride] ==
MV_FROZEN)
501 if (mb_x > 0 && fixed[mb_xy - 1 ] ==
MV_CHANGED)
503 if (mb_x + 1 < mb_width && fixed[mb_xy + 1 ] ==
MV_CHANGED)
505 if (mb_y > 0 && fixed[mb_xy - mb_stride] ==
MV_CHANGED)
507 if (mb_y + 1 < mb_height && fixed[mb_xy + mb_stride] ==
MV_CHANGED)
509 if (j == 0 && pass > 1)
514 if (mb_x > 0 && fixed[mb_xy - 1]) {
515 mv_predictor[pred_count][0] =
517 mv_predictor[pred_count][1] =
523 if (mb_x + 1 < mb_width && fixed[mb_xy + 1]) {
524 mv_predictor[pred_count][0] =
526 mv_predictor[pred_count][1] =
532 if (mb_y > 0 && fixed[mb_xy - mb_stride]) {
533 mv_predictor[pred_count][0] =
535 mv_predictor[pred_count][1] =
541 if (mb_y + 1<mb_height && fixed[mb_xy + mb_stride]) {
542 mv_predictor[pred_count][0] =
544 mv_predictor[pred_count][1] =
553 if (pred_count > 1) {
554 int sum_x = 0, sum_y = 0, sum_r = 0;
555 int max_x, max_y, min_x, min_y, max_r, min_r;
557 for (j = 0; j < pred_count; j++) {
558 sum_x += mv_predictor[j][0];
559 sum_y += mv_predictor[j][1];
561 if (j && ref[j] != ref[j - 1])
562 goto skip_mean_and_median;
566 mv_predictor[pred_count][0] = sum_x / j;
567 mv_predictor[pred_count][1] = sum_y / j;
568 ref[pred_count] = sum_r / j;
571 if (pred_count >= 3) {
572 min_y = min_x = min_r = 99999;
573 max_y = max_x = max_r = -99999;
575 min_x = min_y = max_x = max_y = min_r = max_r = 0;
577 for (j = 0; j < pred_count; j++) {
578 max_x =
FFMAX(max_x, mv_predictor[j][0]);
579 max_y =
FFMAX(max_y, mv_predictor[j][1]);
580 max_r =
FFMAX(max_r, ref[j]);
581 min_x =
FFMIN(min_x, mv_predictor[j][0]);
582 min_y =
FFMIN(min_y, mv_predictor[j][1]);
583 min_r =
FFMIN(min_r, ref[j]);
585 mv_predictor[pred_count + 1][0] = sum_x - max_x - min_x;
586 mv_predictor[pred_count + 1][1] = sum_y - max_y - min_y;
587 ref[pred_count + 1] = sum_r - max_r - min_r;
589 if (pred_count == 4) {
590 mv_predictor[pred_count + 1][0] /= 2;
591 mv_predictor[pred_count + 1][1] /= 2;
592 ref[pred_count + 1] /= 2;
597 skip_mean_and_median:
621 mv_predictor[pred_count][0] = prev_x;
622 mv_predictor[pred_count][1] = prev_y;
623 ref[pred_count] = prev_ref;
637 for (j = 0; j < pred_count; j++) {
640 mb_x * 16 + mb_y * 16 * s->
linesize;
643 s->
mv[0][0][0] = mv_predictor[j][0];
645 s->
mv[0][0][1] = mv_predictor[j][1];
653 if (mb_x > 0 && fixed[mb_xy - 1]) {
655 for (k = 0; k < 16; k++)
659 if (mb_x + 1 < mb_width && fixed[mb_xy + 1]) {
661 for (k = 0; k < 16; k++)
665 if (mb_y > 0 && fixed[mb_xy - mb_stride]) {
667 for (k = 0; k < 16; k++)
670 if (mb_y + 1 < mb_height && fixed[mb_xy + mb_stride]) {
672 for (k = 0; k < 16; k++)
677 if (score <= best_score) {
682 score_sum += best_score;
683 s->
mv[0][0][0] = mv_predictor[best_pred][0];
684 s->
mv[0][0][1] = mv_predictor[best_pred][1];
686 for (i = 0; i < mot_step; i++)
687 for (j = 0; j < mot_step; j++) {
695 if (s->
mv[0][0][0] != prev_x || s->
mv[0][0][1] != prev_y) {
709 for (i = 0; i < s->
mb_num; i++) {
720 int is_intra_likely, i, j, undamaged_count, skip_amount, mb_x, mb_y;
726 for (i = 0; i < s->
mb_num; i++) {
740 if (undamaged_count < 5)
749 skip_amount =
FFMAX(undamaged_count / 50, 1);
753 for (mb_y = 0; mb_y < s->
mb_height - 1; mb_y++) {
754 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
756 const int mb_xy = mb_x + mb_y * s->
mb_stride;
764 if ((j % skip_amount) != 0)
769 mb_x * 16 + mb_y * 16 * s->
linesize;
771 mb_x * 16 + mb_y * 16 * s->
linesize;
779 is_intra_likely += s->
dsp.
sad[0](
NULL, last_mb_ptr, mb_ptr,
781 is_intra_likely -= s->
dsp.
sad[0](
NULL, last_mb_ptr,
793 return is_intra_likely > 0;
815 int endx,
int endy,
int status)
817 const int start_i = av_clip(startx + starty * s->
mb_width, 0, s->
mb_num - 1);
818 const int end_i = av_clip(endx + endy * s->
mb_width, 0, s->
mb_num);
826 if (start_i > end_i || start_xy > end_xy) {
828 "internal error, slice end before start\n");
856 (end_xy - start_xy) *
sizeof(uint8_t));
859 for (i = start_xy; i < end_xy; i++)
884 int i, mb_x, mb_y, error, error_type, dc_error, mv_error, ac_error;
886 int threshold_part[4] = { 100, 100, 100 };
912 for (i = 0; i < 2; i++) {
922 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
923 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
933 for (error_type = 1; error_type <= 3; error_type++) {
936 for (i = s->
mb_num - 1; i >= 0; i--) {
940 if (error & (1 << error_type))
942 if (error & (8 << error_type))
957 for (i = s->
mb_num - 1; i >= 0; i--) {
1004 for (error_type = 1; error_type <= 3; error_type++) {
1005 for (i = s->
mb_num - 1; i >= 0; i--) {
1011 if (error & (1 << error_type))
1015 if (distance < threshold_part[error_type - 1])
1018 if (distance < threshold)
1029 for (i = 0; i < s->
mb_num; i++) {
1043 for (i = 0; i < s->
mb_num; i++) {
1052 dc_error = ac_error = mv_error = 0;
1053 for (i = 0; i < s->
mb_num; i++) {
1064 dc_error, ac_error, mv_error);
1069 for (i = 0; i < s->
mb_num; i++) {
1075 if (is_intra_likely)
1083 for (i = 0; i < s->
mb_num; i++) {
1090 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
1094 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
1095 const int mb_xy = mb_x + mb_y * s->
mb_stride;
1114 int mb_index = mb_x * 2 + mb_y * 2 * s->
b8_stride;
1117 for (j = 0; j < 4; j++) {
1137 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
1141 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
1142 int xy = mb_x * 2 + mb_y * 2 * s->
b8_stride;
1143 const int mb_xy = mb_x + mb_y * s->
mb_stride;
1199 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
1200 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
1201 int dc, dcu, dcv, y, n;
1203 uint8_t *dest_y, *dest_cb, *dest_cr;
1204 const int mb_xy = mb_x + mb_y * s->
mb_stride;
1219 for (n = 0; n < 4; n++) {
1221 for (y = 0; y < 8; y++) {
1223 for (x = 0; x < 8; x++)
1224 dc += dest_y[x + (n & 1) * 8 +
1227 dc_ptr[(n & 1) + (n >> 1) * s->
b8_stride] = (dc + 4) >> 3;
1231 for (y = 0; y < 8; y++) {
1233 for (x = 0; x < 8; x++) {
1252 for (mb_y = 0; mb_y < s->
mb_height; mb_y++) {
1253 for (mb_x = 0; mb_x < s->
mb_width; mb_x++) {
1254 uint8_t *dest_y, *dest_cb, *dest_cr;
1255 const int mb_xy = mb_x + mb_y * s->
mb_stride;
1269 put_dc(s, dest_y, dest_cb, dest_cr, mb_x, mb_y);
1293 for (i = 0; i < s->
mb_num; i++) {