Simplify tx_mask calculation in search_txk_type
Simplified the for loop structure to compute
allowed_tx_mask in search_txk_type.
Change-Id: I7e4c0ff875fc1abec23076b5fcc8bdaeb0067c15
diff --git a/av1/encoder/tx_search.c b/av1/encoder/tx_search.c
index 01f969f..6ffbd59 100644
--- a/av1/encoder/tx_search.c
+++ b/av1/encoder/tx_search.c
@@ -2098,11 +2098,9 @@
max_prob = tx_type_probs[i];
max_idx = i;
}
+ if (tx_type_probs[i] < thresh) prune |= (1 << i);
}
-
- for (i = 0; i < TX_TYPES; i++) {
- if (tx_type_probs[i] < thresh && i != max_idx) prune |= (1 << i);
- }
+ if ((prune >> max_idx) & 0x01) prune &= ~(1 << max_idx);
allowed_tx_mask &= (~prune);
}
for (i = 0; i < TX_TYPES; i++) {
