1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
// See LICENSE for license details.
/** Circuit-land math operations.
*/
package chisel3.util
import chisel3._
object Log2 {
/** Returns the base-2 integer logarithm of the least-significant `width` bits of an UInt.
*
* @note The result is truncated, so e.g. Log2(13.U) === 3.U
*/
def apply(x: Bits, width: Int): UInt = {
if (width < 2) {
0.U
} else if (width == 2) {
x(1)
} else if (width <= divideAndConquerThreshold) {
Mux(x(width-1), UInt((width-1).W), apply(x, width-1))
} else {
val mid = 1 << (log2Ceil(width) - 1)
val hi = x(width-1, mid)
val lo = x(mid-1, 0)
val useHi = hi.orR
Cat(useHi, Mux(useHi, Log2(hi, width - mid), Log2(lo, mid)))
}
}
/** Returns the base-2 integer logarithm of an UInt.
*
* @note The result is truncated, so e.g. Log2(13.U) === 3.U
*/
def apply(x: Bits): UInt = apply(x, x.getWidth)
private def divideAndConquerThreshold = 4
}
|