react-gear-chart/src/utils/helpers.js


import { sum, values } from 'lodash-es'
import COLORS from './colors'
import catNames from 'cat-names'
import { log } from './logger'
import SID from 'shortid'

export function TODO() { throw Error('TODO') }

export function Random(...args) {

  if (args[0] === undefined) {
    return Math.random()
  }
  if (args.length === 1) {
    return Math.random() * args[0]
  }
  if (args.length > 1) {
    return Math.random() * (args[1] - args[0]) + args[0]
  }

}

export function RandomInt(...args) {
  return Math.round(Random(...args))
}

export function RandomDivide(count) {
  let arr = ArrayGen(count)(_ => Random())
  let _sum = sum(arr)
  return arr.map(_ => _ / _sum)
}

export function ToPrecentage(number, decimal = 2) {
  return (number * 100).toFixed(decimal) + '%'
}

export function GenerateTeethData(count, mode, stripCount, colors = []) {

  const colours = values(COLORS)
  const modes = ['spokerib', 'layer', 'bar']
  mode = modes.find(_ => _ === mode)

  function getRandomStrips(count = 2) {
    let weights
    if (mode === 'spokerib') {
      weights = RandomDivide(count)
    } else if (mode === 'bar') {
      weights = ArrayGen(count)(Random)
    } else { // layer
      weights = RandomDivide(count).map(_ => _ + Random(-.2, .2))
    }
    return ArrayGen(count)((_, i) => {
      let c = colors.length === 0 ? colours : colors
      return {
        color: c[RandomInt(c.length - 1)],
        weight: weights[i]
      }
    })
  }

  let result = ArrayGen(count)(i => ({
      mode: mode || modes[RandomInt(2)],
      label: i + ': ' + catNames.random(),
      id: SID.generate(),
      strips: getRandomStrips(stripCount || RandomInt(1, 4))
    }))
  log(result)
  return result
}

function ArrayGen(times) {
  return function (iteratee) {
    let t = times
    let arr = Array(t)
    for (let i = 0; i < t; i++) {
      arr[i] = iteratee(i, i)
    }
    return arr
  }
}