feat: add dsmath (#3)

KyberNetwork · Oct 16, 2023 · c085a8d · c085a8d
1 parent e3446d2
commit c085a8d
Showing 1 changed file with 35 additions and 0 deletions.
diff --git a/dsmath/dsmath.go b/dsmath/dsmath.go
@@ -0,0 +1,35 @@
+package dsmath
+
+import (
+	"math/big"
+
+	"github.com/KyberNetwork/blockchain-toolkit/integer"
+)
+
+var (
+	WAD *big.Int = integer.TenPow(18)
+)
+
+// WMul
+// ((x * y) + (WAD / 2)) / WAD;
+func WMul(x *big.Int, y *big.Int) *big.Int {
+	return new(big.Int).Div(
+		new(big.Int).Add(
+			new(big.Int).Mul(x, y),
+			new(big.Int).Div(WAD, integer.Two()),
+		),
+		WAD,
+	)
+}
+
+// WDiv
+// ((x * WAD) + (y / 2)) / y
+func WDiv(x *big.Int, y *big.Int) *big.Int {
+	return new(big.Int).Div(
+		new(big.Int).Add(
+			new(big.Int).Mul(x, WAD),
+			new(big.Int).Div(y, integer.Two()),
+		),
+		y,
+	)
+}