hashirama/uvg266
1
0
Fork 0
mirror of https://github.com/ultravideo/uvg266.git synced 2024-11-24 02:24:07 +00:00
Code Issues Actions Projects Releases Wiki Activity

Fix bug in intra mode search.

Browse source 
- The cost of the first mode in the mode list was returned instead of cost of
  the selected mode, as this used to be the best mode when the list was
  sorted. Should only matter when doing inter coding.
- This pretty much affects only --rd=1 in inter frames.
This commit is contained in:
Ari Koivula 2015-04-08 15:34:39 +03:00
parent 73db9fec83
commit 7028846423
1 changed files with 50 additions and 53 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/search.c
View file

@ -1704,19 +1704,19 @@ static INLINE void sort_modes(int8_t *__restrict modes, double *__restrict costs
/** /**
* \brief Select mode with the smallest cost. * \brief Select mode with the smallest cost.
*/ */
static INLINE int8_t select_best_mode(const int8_t *modes, const double *costs, uint8_t length) static INLINE uint8_t select_best_mode_index(const int8_t *modes, const double *costs, uint8_t length)
{ {
double best_mode = modes[0]; uint8_t best_index = 0;
double best_cost = costs[0]; double best_cost = costs[0];
for (uint8_t i = 1; i < length; ++i) { for (uint8_t i = 1; i < length; ++i) {
if (costs[i] < best_cost) { if (costs[i] < best_cost) {
best_cost = costs[i]; best_cost = costs[i];
best_mode = modes[i]; best_index = i;
} }
} }
return best_mode; return best_index;
} }
/** /**
@ -1899,7 +1899,7 @@ static int8_t search_intra_rough(encoder_state_t * const state,
++modes_selected; ++modes_selected;
} }
int8_t best_mode = select_best_mode(modes, costs, modes_selected); int8_t best_mode = modes[select_best_mode_index(modes, costs, modes_selected)];
double best_cost = min_cost; double best_cost = min_cost;
// Skip recursive search if all modes have the same cost. // Skip recursive search if all modes have the same cost.
@ -2149,7 +2149,6 @@ static double search_cu_intra(encoder_state_t * const state,
double costs[35]; double costs[35];
// Find best intra mode for 2Nx2N. // Find best intra mode for 2Nx2N.
{
pixel_t *ref_pixels = &lcu->ref.y[lcu_px.x + lcu_px.y * LCU_WIDTH]; pixel_t *ref_pixels = &lcu->ref.y[lcu_px.x + lcu_px.y * LCU_WIDTH];
unsigned pu_index = PU_INDEX(x_px >> 2, y_px >> 2); unsigned pu_index = PU_INDEX(x_px >> 2, y_px >> 2);
@ -2195,12 +2194,10 @@ static double search_cu_intra(encoder_state_t * const state,
modes, costs, lcu); modes, costs, lcu);
} }
int8_t best_mode = select_best_mode(modes, costs, number_of_modes); uint8_t best_mode_i = select_best_mode_index(modes, costs, number_of_modes);
cur_cu->intra[pu_index].mode = modes[best_mode_i];
cur_cu->intra[pu_index].mode = best_mode; return costs[best_mode_i];
}
return costs[0];
} }
// Return estimate of bits used to code prediction mode of cur_cu. // Return estimate of bits used to code prediction mode of cur_cu.