import { each, isNil } from '@antv/util';

// 求以a为次幂，结果为b的基数，如 x^^a = b;求x
// 虽然数学上 b 不支持负数，但是这里需要支持 负数
export function calBase(a: number, b: number) {
  const e = Math.E;
  let value;
  if (b >= 0) {
    value = Math.pow(e, Math.log(b) / a); // 使用换底公式求底
  } else {
    value = Math.pow(e, Math.log(-b) / a) * -1; // 使用换底公式求底
  }
  return value;
}

export function log(a: number, b: number) {
  if (a === 1) {
    return 1;
  }
  return Math.log(b) / Math.log(a);
}

export function getLogPositiveMin(values, base, max?: number) {
  if (isNil(max)) {
    max = Math.max.apply(null, values);
  }
  let positiveMin = max;
  each(values, (value) => {
    if (value > 0 && value < positiveMin) {
      positiveMin = value;
    }
  });
  if (positiveMin === max) {
    positiveMin = max / base;
  }
  if (positiveMin > 1) {
    positiveMin = 1;
  }
  return positiveMin;
}

function digitLength(num: number) {
  // Get digit length of e
  const eSplit = num.toString().split(/[eE]/);
  const len = (eSplit[0].split('.')[1] || '').length - +(eSplit[1] || 0);
  return len > 0 ? len : 0;
}

/**
 * 高精度加法，解决 0.1 + 0.2 !== 0.3 的经典问题
 *
 * @param num1 加数
 * @param num2 被加数
 * @return {number} 返回值
 */
export function precisionAdd(num1: number, num2: number) {
  const num1Digits = digitLength(num1);
  const num2Digits = digitLength(num2);
  const baseNum = 10 ** Math.max(num1Digits, num2Digits);
  return (num1 * baseNum + num2 * baseNum) / baseNum;
}