/src/aom/av1/common/obmc.h
Line | Count | Source |
1 | | /* |
2 | | * Copyright (c) 2017, Alliance for Open Media. All rights reserved. |
3 | | * |
4 | | * This source code is subject to the terms of the BSD 2 Clause License and |
5 | | * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
6 | | * was not distributed with this source code in the LICENSE file, you can |
7 | | * obtain it at www.aomedia.org/license/software. If the Alliance for Open |
8 | | * Media Patent License 1.0 was not distributed with this source code in the |
9 | | * PATENTS file, you can obtain it at www.aomedia.org/license/patent. |
10 | | */ |
11 | | |
12 | | #ifndef AOM_AV1_COMMON_OBMC_H_ |
13 | | #define AOM_AV1_COMMON_OBMC_H_ |
14 | | |
15 | | typedef void (*overlappable_nb_visitor_t)(MACROBLOCKD *xd, int rel_mi_row, |
16 | | int rel_mi_col, uint8_t op_mi_size, |
17 | | int dir, MB_MODE_INFO *nb_mi, |
18 | | void *fun_ctxt, const int num_planes); |
19 | | |
20 | | static inline void foreach_overlappable_nb_above(const AV1_COMMON *cm, |
21 | | MACROBLOCKD *xd, int nb_max, |
22 | | overlappable_nb_visitor_t fun, |
23 | 3.46M | void *fun_ctxt) { |
24 | 3.46M | if (!xd->up_available) return; |
25 | | |
26 | 3.24M | const int num_planes = av1_num_planes(cm); |
27 | 3.24M | int nb_count = 0; |
28 | 3.24M | const int mi_col = xd->mi_col; |
29 | | // prev_row_mi points into the mi array, starting at the beginning of the |
30 | | // previous row. |
31 | 3.24M | MB_MODE_INFO **prev_row_mi = xd->mi - mi_col - 1 * xd->mi_stride; |
32 | 3.24M | const int end_col = AOMMIN(mi_col + xd->width, cm->mi_params.mi_cols); |
33 | 3.24M | uint8_t mi_step; |
34 | 7.03M | for (int above_mi_col = mi_col; above_mi_col < end_col && nb_count < nb_max; |
35 | 3.79M | above_mi_col += mi_step) { |
36 | 3.79M | MB_MODE_INFO **above_mi = prev_row_mi + above_mi_col; |
37 | 3.79M | mi_step = |
38 | 3.79M | AOMMIN(mi_size_wide[above_mi[0]->bsize], mi_size_wide[BLOCK_64X64]); |
39 | | // If we're considering a block with width 4, it should be treated as |
40 | | // half of a pair of blocks with chroma information in the second. Move |
41 | | // above_mi_col back to the start of the pair if needed, set above_mbmi |
42 | | // to point at the block with chroma information, and set mi_step to 2 to |
43 | | // step over the entire pair at the end of the iteration. |
44 | 3.79M | if (mi_step == 1) { |
45 | 246k | above_mi_col &= ~1; |
46 | 246k | above_mi = prev_row_mi + above_mi_col + 1; |
47 | 246k | mi_step = 2; |
48 | 246k | } |
49 | 3.79M | if (is_neighbor_overlappable(*above_mi)) { |
50 | 3.31M | ++nb_count; |
51 | 3.31M | fun(xd, 0, above_mi_col - mi_col, AOMMIN(xd->width, mi_step), 0, |
52 | 3.31M | *above_mi, fun_ctxt, num_planes); |
53 | 3.31M | } |
54 | 3.79M | } |
55 | 3.24M | } decodeframe.c:foreach_overlappable_nb_above Line | Count | Source | 23 | 441k | void *fun_ctxt) { | 24 | 441k | if (!xd->up_available) return; | 25 | | | 26 | 441k | const int num_planes = av1_num_planes(cm); | 27 | 441k | int nb_count = 0; | 28 | 441k | const int mi_col = xd->mi_col; | 29 | | // prev_row_mi points into the mi array, starting at the beginning of the | 30 | | // previous row. | 31 | 441k | MB_MODE_INFO **prev_row_mi = xd->mi - mi_col - 1 * xd->mi_stride; | 32 | 441k | const int end_col = AOMMIN(mi_col + xd->width, cm->mi_params.mi_cols); | 33 | 441k | uint8_t mi_step; | 34 | 936k | for (int above_mi_col = mi_col; above_mi_col < end_col && nb_count < nb_max; | 35 | 495k | above_mi_col += mi_step) { | 36 | 495k | MB_MODE_INFO **above_mi = prev_row_mi + above_mi_col; | 37 | 495k | mi_step = | 38 | 495k | AOMMIN(mi_size_wide[above_mi[0]->bsize], mi_size_wide[BLOCK_64X64]); | 39 | | // If we're considering a block with width 4, it should be treated as | 40 | | // half of a pair of blocks with chroma information in the second. Move | 41 | | // above_mi_col back to the start of the pair if needed, set above_mbmi | 42 | | // to point at the block with chroma information, and set mi_step to 2 to | 43 | | // step over the entire pair at the end of the iteration. | 44 | 495k | if (mi_step == 1) { | 45 | 35.0k | above_mi_col &= ~1; | 46 | 35.0k | above_mi = prev_row_mi + above_mi_col + 1; | 47 | 35.0k | mi_step = 2; | 48 | 35.0k | } | 49 | 495k | if (is_neighbor_overlappable(*above_mi)) { | 50 | 436k | ++nb_count; | 51 | 436k | fun(xd, 0, above_mi_col - mi_col, AOMMIN(xd->width, mi_step), 0, | 52 | 436k | *above_mi, fun_ctxt, num_planes); | 53 | 436k | } | 54 | 495k | } | 55 | 441k | } |
reconinter.c:foreach_overlappable_nb_above Line | Count | Source | 23 | 3.02M | void *fun_ctxt) { | 24 | 3.02M | if (!xd->up_available) return; | 25 | | | 26 | 2.80M | const int num_planes = av1_num_planes(cm); | 27 | 2.80M | int nb_count = 0; | 28 | 2.80M | const int mi_col = xd->mi_col; | 29 | | // prev_row_mi points into the mi array, starting at the beginning of the | 30 | | // previous row. | 31 | 2.80M | MB_MODE_INFO **prev_row_mi = xd->mi - mi_col - 1 * xd->mi_stride; | 32 | 2.80M | const int end_col = AOMMIN(mi_col + xd->width, cm->mi_params.mi_cols); | 33 | 2.80M | uint8_t mi_step; | 34 | 6.10M | for (int above_mi_col = mi_col; above_mi_col < end_col && nb_count < nb_max; | 35 | 3.29M | above_mi_col += mi_step) { | 36 | 3.29M | MB_MODE_INFO **above_mi = prev_row_mi + above_mi_col; | 37 | 3.29M | mi_step = | 38 | 3.29M | AOMMIN(mi_size_wide[above_mi[0]->bsize], mi_size_wide[BLOCK_64X64]); | 39 | | // If we're considering a block with width 4, it should be treated as | 40 | | // half of a pair of blocks with chroma information in the second. Move | 41 | | // above_mi_col back to the start of the pair if needed, set above_mbmi | 42 | | // to point at the block with chroma information, and set mi_step to 2 to | 43 | | // step over the entire pair at the end of the iteration. | 44 | 3.29M | if (mi_step == 1) { | 45 | 211k | above_mi_col &= ~1; | 46 | 211k | above_mi = prev_row_mi + above_mi_col + 1; | 47 | 211k | mi_step = 2; | 48 | 211k | } | 49 | 3.29M | if (is_neighbor_overlappable(*above_mi)) { | 50 | 2.88M | ++nb_count; | 51 | 2.88M | fun(xd, 0, above_mi_col - mi_col, AOMMIN(xd->width, mi_step), 0, | 52 | 2.88M | *above_mi, fun_ctxt, num_planes); | 53 | 2.88M | } | 54 | 3.29M | } | 55 | 2.80M | } |
Unexecuted instantiation: rdopt.c:foreach_overlappable_nb_above Unexecuted instantiation: reconinter_enc.c:foreach_overlappable_nb_above Unexecuted instantiation: reconinter_enc_sse2.c:foreach_overlappable_nb_above |
56 | | |
57 | | static inline void foreach_overlappable_nb_left(const AV1_COMMON *cm, |
58 | | MACROBLOCKD *xd, int nb_max, |
59 | | overlappable_nb_visitor_t fun, |
60 | 1.38M | void *fun_ctxt) { |
61 | 1.38M | if (!xd->left_available) return; |
62 | | |
63 | 1.31M | const int num_planes = av1_num_planes(cm); |
64 | 1.31M | int nb_count = 0; |
65 | | // prev_col_mi points into the mi array, starting at the top of the |
66 | | // previous column |
67 | 1.31M | const int mi_row = xd->mi_row; |
68 | 1.31M | MB_MODE_INFO **prev_col_mi = xd->mi - 1 - mi_row * xd->mi_stride; |
69 | 1.31M | const int end_row = AOMMIN(mi_row + xd->height, cm->mi_params.mi_rows); |
70 | 1.31M | uint8_t mi_step; |
71 | 2.86M | for (int left_mi_row = mi_row; left_mi_row < end_row && nb_count < nb_max; |
72 | 1.54M | left_mi_row += mi_step) { |
73 | 1.54M | MB_MODE_INFO **left_mi = prev_col_mi + left_mi_row * xd->mi_stride; |
74 | 1.54M | mi_step = |
75 | 1.54M | AOMMIN(mi_size_high[left_mi[0]->bsize], mi_size_high[BLOCK_64X64]); |
76 | 1.54M | if (mi_step == 1) { |
77 | 131k | left_mi_row &= ~1; |
78 | 131k | left_mi = prev_col_mi + (left_mi_row + 1) * xd->mi_stride; |
79 | 131k | mi_step = 2; |
80 | 131k | } |
81 | 1.54M | if (is_neighbor_overlappable(*left_mi)) { |
82 | 1.33M | ++nb_count; |
83 | 1.33M | fun(xd, left_mi_row - mi_row, 0, AOMMIN(xd->height, mi_step), 1, *left_mi, |
84 | 1.33M | fun_ctxt, num_planes); |
85 | 1.33M | } |
86 | 1.54M | } |
87 | 1.31M | } decodeframe.c:foreach_overlappable_nb_left Line | Count | Source | 60 | 444k | void *fun_ctxt) { | 61 | 444k | if (!xd->left_available) return; | 62 | | | 63 | 444k | const int num_planes = av1_num_planes(cm); | 64 | 444k | int nb_count = 0; | 65 | | // prev_col_mi points into the mi array, starting at the top of the | 66 | | // previous column | 67 | 444k | const int mi_row = xd->mi_row; | 68 | 444k | MB_MODE_INFO **prev_col_mi = xd->mi - 1 - mi_row * xd->mi_stride; | 69 | 444k | const int end_row = AOMMIN(mi_row + xd->height, cm->mi_params.mi_rows); | 70 | 444k | uint8_t mi_step; | 71 | 932k | for (int left_mi_row = mi_row; left_mi_row < end_row && nb_count < nb_max; | 72 | 488k | left_mi_row += mi_step) { | 73 | 488k | MB_MODE_INFO **left_mi = prev_col_mi + left_mi_row * xd->mi_stride; | 74 | 488k | mi_step = | 75 | 488k | AOMMIN(mi_size_high[left_mi[0]->bsize], mi_size_high[BLOCK_64X64]); | 76 | 488k | if (mi_step == 1) { | 77 | 46.5k | left_mi_row &= ~1; | 78 | 46.5k | left_mi = prev_col_mi + (left_mi_row + 1) * xd->mi_stride; | 79 | 46.5k | mi_step = 2; | 80 | 46.5k | } | 81 | 488k | if (is_neighbor_overlappable(*left_mi)) { | 82 | 430k | ++nb_count; | 83 | 430k | fun(xd, left_mi_row - mi_row, 0, AOMMIN(xd->height, mi_step), 1, *left_mi, | 84 | 430k | fun_ctxt, num_planes); | 85 | 430k | } | 86 | 488k | } | 87 | 444k | } |
reconinter.c:foreach_overlappable_nb_left Line | Count | Source | 60 | 936k | void *fun_ctxt) { | 61 | 936k | if (!xd->left_available) return; | 62 | | | 63 | 874k | const int num_planes = av1_num_planes(cm); | 64 | 874k | int nb_count = 0; | 65 | | // prev_col_mi points into the mi array, starting at the top of the | 66 | | // previous column | 67 | 874k | const int mi_row = xd->mi_row; | 68 | 874k | MB_MODE_INFO **prev_col_mi = xd->mi - 1 - mi_row * xd->mi_stride; | 69 | 874k | const int end_row = AOMMIN(mi_row + xd->height, cm->mi_params.mi_rows); | 70 | 874k | uint8_t mi_step; | 71 | 1.93M | for (int left_mi_row = mi_row; left_mi_row < end_row && nb_count < nb_max; | 72 | 1.05M | left_mi_row += mi_step) { | 73 | 1.05M | MB_MODE_INFO **left_mi = prev_col_mi + left_mi_row * xd->mi_stride; | 74 | 1.05M | mi_step = | 75 | 1.05M | AOMMIN(mi_size_high[left_mi[0]->bsize], mi_size_high[BLOCK_64X64]); | 76 | 1.05M | if (mi_step == 1) { | 77 | 85.1k | left_mi_row &= ~1; | 78 | 85.1k | left_mi = prev_col_mi + (left_mi_row + 1) * xd->mi_stride; | 79 | 85.1k | mi_step = 2; | 80 | 85.1k | } | 81 | 1.05M | if (is_neighbor_overlappable(*left_mi)) { | 82 | 903k | ++nb_count; | 83 | 903k | fun(xd, left_mi_row - mi_row, 0, AOMMIN(xd->height, mi_step), 1, *left_mi, | 84 | 903k | fun_ctxt, num_planes); | 85 | 903k | } | 86 | 1.05M | } | 87 | 874k | } |
Unexecuted instantiation: rdopt.c:foreach_overlappable_nb_left Unexecuted instantiation: reconinter_enc.c:foreach_overlappable_nb_left Unexecuted instantiation: reconinter_enc_sse2.c:foreach_overlappable_nb_left |
88 | | |
89 | | #endif // AOM_AV1_COMMON_OBMC_H_ |