From 41a1de80bd3f4a0013218050f7dc5e04a3bf7297 Mon Sep 17 00:00:00 2001
From: OceanS2000 <30361859+OceanS2000@users.noreply.github.com>
Date: Wed, 15 Feb 2023 23:50:49 +0800
Subject: [PATCH] [utils] Utility functions from rocket-chip

---
 tilelink/src/utils/OH1.scala     | 19 +++++++++++++++++
 tilelink/src/utils/package.scala | 35 ++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)
 create mode 100644 tilelink/src/utils/OH1.scala
 create mode 100644 tilelink/src/utils/package.scala

diff --git a/tilelink/src/utils/OH1.scala b/tilelink/src/utils/OH1.scala
new file mode 100644
index 0000000..7c33b47
--- /dev/null
+++ b/tilelink/src/utils/OH1.scala
@@ -0,0 +1,19 @@
+// SPDX-License-Identifier: Apache-2.0
+// Copyright 2016-2017 SiFive, Inc. See LICENSE.SiFive for details
+
+package utils
+
+import chisel3._
+import chisel3.util.{Cat, OHToUInt}
+
+object OH1 {
+  def OH1ToOH(x: UInt): UInt =
+    ((x << 1).asUInt | 1.U) & (~Cat(0.U(1.W), x)).asUInt
+
+  def OH1ToUInt(x: UInt): UInt = OHToUInt(OH1ToOH(x))
+
+  def UIntToOH1(x: UInt, width: Int): UInt =
+    (~((-1).S(width.W).asUInt << x)(width - 1, 0)).asUInt
+
+  def UIntToOH1(x: UInt): UInt = UIntToOH1(x, (1 << x.getWidth) - 1)
+}
diff --git a/tilelink/src/utils/package.scala b/tilelink/src/utils/package.scala
new file mode 100644
index 0000000..b5833d7
--- /dev/null
+++ b/tilelink/src/utils/package.scala
@@ -0,0 +1,35 @@
+// SPDX-License-Identifier: Apache-2.0
+// Copyright 2016-2017 SiFive, Inc. See LICENSE.SiFive for details
+
+import chisel3._
+
+import scala.annotation.tailrec
+import scala.math.min
+
+package object utils {
+  // Fill 1s from low bits to high bits
+  def leftOR(x: UInt): UInt = leftOR(x, x.getWidth, x.getWidth)
+
+  def leftOR(x: UInt, width: Integer, cap: Integer = 999999): UInt = {
+    val stop = min(width, cap)
+
+    @tailrec
+    def helper(s: Int, x: UInt): UInt =
+      if (s >= stop) x else helper(s + s, x | (x << s)(width - 1, 0))
+
+    helper(1, x)(width - 1, 0)
+  }
+
+  // Fill 1s form high bits to low bits
+  def rightOR(x: UInt): UInt = rightOR(x, x.getWidth, x.getWidth)
+
+  def rightOR(x: UInt, width: Integer, cap: Integer = 999999): UInt = {
+    val stop = min(width, cap)
+
+    @tailrec
+    def helper(s: Int, x: UInt): UInt =
+      if (s >= stop) x else helper(s + s, x | (x >> s)(width - 1, 0))
+
+    helper(1, x)(width - 1, 0)
+  }
+}