Enable Nx2N partitions.

This commit is contained in:
Arttu Ylä-Outinen 2015-11-03 11:30:54 +02:00
parent 4402e251ae
commit 0f531362bf

View file

@ -541,17 +541,20 @@ static double search_cu(encoder_state_t * const state, int x, int y, int depth,
} }
if (depth < MAX_DEPTH) { if (depth < MAX_DEPTH) {
// Try SMP partitioning. // Try SMP partitioning.
mode_cost = kvz_search_cu_smp(state, static const part_mode_t smp_modes[] = {SIZE_2NxN, SIZE_Nx2N};
x, y, for (int i = 0; i < 2; ++i) {
depth, mode_cost = kvz_search_cu_smp(state,
SIZE_2NxN, x, y,
&work_tree[depth + 1]); depth,
if (mode_cost < cost) { smp_modes[i],
cost = mode_cost; &work_tree[depth + 1]);
// TODO: only copy inter prediction info, not pixels // TODO: take cost of coding part mode into account
work_tree_copy_up(x, y, depth, work_tree); if (mode_cost < cost) {
cost = mode_cost;
// TODO: only copy inter prediction info, not pixels
work_tree_copy_up(x, y, depth, work_tree);
}
} }
} }
} }