summaryrefslogtreecommitdiff
path: root/src/main/scala/chisel3/util/CircuitMath.scala
blob: 83e5feb11f449c55e2bb3fab594f21492cdaaa3c (plain)
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
}