extra-rand 0.1.4 → 0.1.5

package/README.md

@@ -38,7 +38,7 @@ function randomByWeight(weights: number[]): number
 The function returns an index of one of weights.
 
 ### Map
-These low-level methods help you to use random number generators other than `Math.random()`.
+These low-level functions help you to use random number generators other than `Math.random()`.
 
 #### mapToRange
 ```ts

package/package.json

@@ -1,12 +1,13 @@
 {
   "name": "extra-rand",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "Yet another random library",
   "keywords": [
     "random"
   ],
   "files": [
-    "lib"
+    "lib",
+    "src"
   ],
   "main": "lib/index.js",
   "types": "lib/index.d.ts",

package/src/index.ts

@@ -0,0 +1,8 @@
+export * from './random'
+export * from './random-int'
+export * from './random-int-inclusive'
+export * from './random-by-weight'
+
+export * from './map-to-range'
+export * from './map-to-int-range'
+export * from './map-to-index-by-weight'

package/src/map-to-index-by-weight.ts

@@ -0,0 +1,36 @@
+import { mapToRange } from './map-to-range'
+import { mapToIntRange } from './map-to-int-range'
+
+export function mapToIndexByWeight(
+  value: number
+, oldMin: number, oldMax: number
+, weights: number[]
+): number {
+  const newMin = 0
+  const newMax = weights.reduce((acc, cur) => acc + Math.max(cur, 0))
+  if (newMax === 0) {
+    // 只有在所有权重都小于等于0的情况下会进入此路径, 这时将所有权重都视为1.
+    const index = mapToIntRange(value, oldMin, oldMax, newMin, weights.length)
+
+    return index === weights.length
+         ? weights.length - 1
+         : index
+  } else {
+    const newValue = mapToRange(value, oldMin, oldMax, newMin, newMax)
+
+    let remains = newMax
+    for (let i = weights.length; i--;) {
+      const weight = weights[i]
+      if (weight <= 0) {
+        continue
+      } else {
+        remains -= weight
+        if (newValue >= remains) {
+          return i
+        }
+      }
+    }
+
+    throw new Error('Impossible route')
+  }
+}

package/src/map-to-int-range.ts

@@ -0,0 +1,9 @@
+import { mapToRange } from './map-to-range'
+
+export function mapToIntRange(
+  value: number
+, oldMin: number, oldMax: number
+, newMin: number, newMax: number
+): number {
+  return Math.floor(mapToRange(value, oldMin, oldMax, newMin, newMax))
+}

package/src/map-to-range.ts

@@ -0,0 +1,9 @@
+export function mapToRange(
+  value: number
+, oldMin: number, oldMax: number
+, newMin: number, newMax: number
+): number {
+  return (value - oldMin) / (oldMax - oldMin)
+       * (newMax - newMin)
+       + newMin
+}

package/src/random-by-weight.ts

@@ -0,0 +1,9 @@
+import { mapToIndexByWeight } from './map-to-index-by-weight'
+
+export function randomByWeight(weights: number[]): number {
+  return mapToIndexByWeight(
+    Math.random()
+  , 0, 1
+  , weights
+  )
+}

package/src/random-int-inclusive.ts

@@ -0,0 +1,9 @@
+import { mapToIntRange } from './map-to-int-range'
+
+export function randomIntInclusive(min: number, max: number): number {
+  return mapToIntRange(
+    Math.random()
+  , 0, 1
+  , Math.ceil(min), Math.floor(max) + 1
+  )
+}

package/src/random-int.ts

@@ -0,0 +1,9 @@
+import { mapToIntRange } from './map-to-int-range'
+
+export function randomInt(min: number, max: number): number {
+  return mapToIntRange(
+    Math.random()
+  , 0, 1
+  , Math.ceil(min), Math.floor(max)
+  )
+}

package/src/random.ts

@@ -0,0 +1,9 @@
+import { mapToRange } from './map-to-range'
+
+export function random(min: number, max: number): number {
+  return mapToRange(
+    Math.random()
+  , 0, 1
+  , min, max
+  )
+}