/* This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ const MersenneTwister = require('./mersennetwister') const {logger} = require('../logging') class random { /** * Must be called before any other methods can be called to initialize MersenneTwister. * @param {number|null|undefined} seed Value to initialize MersenneTwister. */ static init (seed) { if (seed === null || seed === undefined) { random.seed = new Date().getTime() } random.twister = new MersenneTwister() random.twister.seed(random.seed) } /** * Returns an integer in [0, limit). Uniform distribution. * @param limit */ static number (limit) { if (!random.twister) { throw new Error('random.init must be called first.') } if (limit === null || limit === undefined) { limit = 0xffffffff } let x = (0x100000000 / limit) >>> 0 let y = (x * limit) >>> 0 let r do { r = random.twister.int32() } while (y && r >= y) // eslint-disable-line no-unmodified-loop-condition return (r / x) >>> 0 } /** * Returns a float in [0, 1). Uniform distribution. */ static float () { if (!random.twister) { throw new Error('random.init must be called first.') } return random.twister.real2() } /** * Returns an integer in [start, limit]. Uniform distribution. * @param start * @param limit */ static range (start, limit) { if (!random.twister) { throw new Error('random.init must be called first.') } if (isNaN(start) || isNaN(limit)) { logger.traceback() throw new TypeError(`random.range() received non-number type: (${start}, ${limit})`) } return random.number(limit - start + 1) + start } /** * Returns a float in [1, limit]. The logarithm has uniform distribution. * @param {*} limit */ static ludOneTo (limit) { return Math.exp(random.float() * Math.log(limit)) } static item (list) { if (list === undefined || typeof list === 'string' || list.length === undefined) { logger.traceback() throw new TypeError(`random.item() received invalid object: (${list})`) } return list[random.number(list.length)] } /** * Returns a random key of a provided object. * @param {*} obj */ static key (obj) { let list = [] for (let i in obj) { list.push(i) } return random.item(list) } /** * Return a random Boolean value. */ static bool () { return random.item([true, false]) } static pick (obj) { if (typeof obj === 'function') { return obj() } if (Array.isArray(obj)) { return random.pick(random.item(obj)) } return obj } static chance (limit) { if (limit === null || limit === undefined) { limit = 2 } if (isNaN(limit)) { logger.traceback() throw new TypeError(`random.chance() received non-number type: (${limit})`) } return random.number(limit) === 1 } static choose (list, flat) { if (!(Array.isArray(list))) { logger.traceback() throw new TypeError(`random.choose() received non-array type: (${list})`) } let total = 0 for (let i = 0; i < list.length; i++) { total += list[i][0] } let n = random.number(total) for (let i = 0; i < list.length; i++) { if (n < list[i][0]) { if (flat === true) { return list[i][1] } else { return random.pick([list[i][1]]) } } n = n - list[i][0] } if (flat === true) { return list[0][1] } return random.pick([list[0][1]]) } /** * More memory-hungry but hopefully faster than random.choose$flat. * @param {*} wa */ static weighted (wa) { let a = [] for (let i = 0; i < wa.length; ++i) { for (let j = 0; j < wa[i].w; ++j) { a.push(wa[i].v) } } return a } static use (obj) { return random.bool() ? obj : '' } static shuffle (arr) { let i = arr.length while (i--) { let p = random.number(i + 1) let t = arr[i] arr[i] = arr[p] arr[p] = t } } static shuffled (arr) { let newArray = arr.slice() random.shuffle(newArray) return newArray } static subset (list, limit) { if (!(Array.isArray(list))) { logger.traceback() throw new TypeError(`random.subset() received non-array type: (${list})`) } if (typeof limit !== 'number') { limit = random.number(list.length + 1) } let result = [] for (let i = 0; i < limit; i++) { result.push(random.pick(list)) } return result } /** * Removes and returns a random item from an array. * @param {*} arr */ static pop (arr) { let i, obj i = random.number(arr.length) obj = arr[i] arr.splice(i, 1) return obj } static hex (len) { return random.number(Math.pow(2, len * 4)).toString(16) } } module.exports = random