mirror of
https://github.com/ultravideo/uvg266.git
synced 2024-11-24 02:24:07 +00:00
Enable Nx2N partitions.
This commit is contained in:
parent
4402e251ae
commit
0f531362bf
23
src/search.c
23
src/search.c
|
@ -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);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue