From b027771da370ba3e0595bbabc10a956bdb3343d8 Mon Sep 17 00:00:00 2001
From: Vick Scarlet <scarlet_vick@outlook.com>
Date: Thu, 27 Aug 2026 18:11:38 +0800
Subject: [PATCH] fix: pickWeight now supports float weights via automatic
 scaling

---
 packages/vitex/index.ts | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/packages/vitex/index.ts b/packages/vitex/index.ts
index 9cbe453..133a037 100644
--- a/packages/vitex/index.ts
+++ b/packages/vitex/index.ts
@@ -11,9 +11,28 @@ export function pick<T>(items: T[], rng?: RNG) {
 export type WeightItem<T> = readonly [T, number]
 export function pickWeight<T>(items: WeightItem<T>[], rng?: RNG) {
     if (items.length === 0) return null
-    const totalWeight = items.reduce((sum, [, weight]) => sum + weight, 0)
+    // Find minimum positive weight to determine scaling factor
+    let minWeight = Infinity
+    for (const [, w] of items) {
+        if (w > 0 && w < minWeight) minWeight = w
+    }
+    // If all weights are integers or no positive weights, use original logic
+    if (minWeight === Infinity || Number.isInteger(minWeight)) {
+        const totalWeight = items.reduce((sum, [, weight]) => sum + weight, 0)
+        let mark = random(totalWeight - 1, 0, rng)
+        for (const [item, weight] of items) {
+            if (mark < weight) return item
+            mark -= weight
+        }
+        return null
+    }
+    // Scale weights to integers based on decimal places of minWeight
+    const decimals = Math.max(0, -Math.floor(Math.log10(minWeight))) + 1
+    const scale = 10 ** decimals
+    const scaledItems: WeightItem<T>[] = items.map(([item, weight]) => [item, Math.round(weight * scale)])
+    const totalWeight = scaledItems.reduce((sum, [, weight]) => sum + weight, 0)
     let mark = random(totalWeight - 1, 0, rng)
-    for (const [item, weight] of items) {
+    for (const [item, weight] of scaledItems) {
         if (mark < weight) return item
         mark -= weight
     }