Skip to content

Commit

Permalink
Replace call to ILog with more intuitive TxSize fns
Browse files Browse the repository at this point in the history
  • Loading branch information
FreezyLemon authored and lu-zero committed Aug 14, 2024
1 parent da229e9 commit 13ee0da
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/transform/inverse.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1649,7 +1649,8 @@ pub(crate) mod rust {

// perform inv txfm on every row
let range = bd + 8;
let txfm_fn = INV_TXFM_FNS[tx_types_1d.1 as usize][ILog::ilog(width) - 3];
let txfm_fn =
INV_TXFM_FNS[tx_types_1d.1 as usize][tx_size.width_log2() - 2];
// 64 point transforms only signal 32 coeffs. We only take chunks of 32
// and skip over the last 32 transforms here.
for (r, buffer_slice) in (0..height.min(32)).zip(buffer.chunks_mut(width))
Expand Down Expand Up @@ -1677,7 +1678,8 @@ pub(crate) mod rust {

// perform inv txfm on every col
let range = cmp::max(bd + 6, 16);
let txfm_fn = INV_TXFM_FNS[tx_types_1d.0 as usize][ILog::ilog(height) - 3];
let txfm_fn =
INV_TXFM_FNS[tx_types_1d.0 as usize][tx_size.height_log2() - 2];
for c in 0..width {
let mut temp_in: [i32; 64] = [0; 64];
let mut temp_out: [i32; 64] = [0; 64];
Expand Down

0 comments on commit 13ee0da

Please sign in to comment.