log.js 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  1. /**
  2. * @fileOverview 使用度量,进行log转换
  3. * @author dxq613@gmail.com
  4. */
  5. const each = require('@antv/util/lib/each');
  6. const Base = require('./base');
  7. const Linear = require('./linear');
  8. // 计算log
  9. function log(a, b) {
  10. if (a === 1) {
  11. return 1;
  12. }
  13. return Math.log(b) / Math.log(a);
  14. }
  15. /**
  16. * 度量的log计算
  17. * @class Scale.Log
  18. */
  19. class Log extends Linear {
  20. _initDefaultCfg() {
  21. super._initDefaultCfg();
  22. this.type = 'log';
  23. /**
  24. * @override
  25. * log 的坐标点的个数控制在10个以下
  26. * @type {Number}
  27. */
  28. this.tickCount = 10;
  29. /**
  30. * 进行log计算的基数
  31. * @type {Number}
  32. */
  33. this.base = 2;
  34. // 最小的tick,仅内部使用
  35. this._minTick = null;
  36. }
  37. /**
  38. * @override
  39. */
  40. calculateTicks() {
  41. const self = this;
  42. const base = self.base;
  43. let minTick;
  44. if (self.min < 0) {
  45. throw new Error('The minimum value must be greater than zero!');
  46. }
  47. const maxTick = log(base, self.max);
  48. if (self.min > 0) {
  49. minTick = Math.floor(log(base, self.min));
  50. } else {
  51. const values = self.values;
  52. let positiveMin = self.max; // 查找大于0的第一个值, 如果都小于0,默认为1
  53. each(values, value => {
  54. if (value > 0 && value < positiveMin) {
  55. positiveMin = value;
  56. }
  57. });
  58. if (positiveMin === self.max) {
  59. positiveMin = self.max / base;
  60. }
  61. if (positiveMin > 1) {
  62. positiveMin = 1;
  63. }
  64. minTick = Math.floor(log(base, positiveMin));
  65. self._minTick = minTick;
  66. self.positiveMin = positiveMin;
  67. }
  68. const count = maxTick - minTick;
  69. const tickCount = self.tickCount;
  70. const avg = Math.ceil(count / tickCount);
  71. const ticks = [];
  72. for (let i = minTick; i < maxTick + avg; i = i + avg) {
  73. ticks.push(Math.pow(base, i));
  74. }
  75. if (self.min === 0) {
  76. ticks.unshift(0);
  77. }
  78. return ticks;
  79. }
  80. // 获取度量计算时,value占的定义域百分比
  81. _getScalePercent(value) {
  82. const max = this.max;
  83. let min = this.min;
  84. if (max === min) {
  85. return 0;
  86. }
  87. // 如果值小于等于0,则按照0处理
  88. if (value <= 0) {
  89. return 0;
  90. }
  91. const base = this.base;
  92. const positiveMin = this.positiveMin;
  93. // 如果min == 0, 则根据比0大的最小值,计算比例关系。这个最小值作为坐标轴上的第二个tick,第一个是0但是不显示
  94. if (positiveMin) {
  95. min = positiveMin * 1 / base;
  96. }
  97. let percent;
  98. // 如果数值小于次小值,那么就计算 value / 次小值 占整体的比例
  99. if (value < positiveMin) {
  100. percent = (value / positiveMin) / (log(base, max) - log(base, min));
  101. } else {
  102. percent = (log(base, value) - log(base, min)) / (log(base, max) - log(base, min));
  103. }
  104. return percent;
  105. }
  106. /**
  107. * @override
  108. */
  109. scale(value) {
  110. const percent = this._getScalePercent(value);
  111. const rangeMin = this.rangeMin();
  112. const rangeMax = this.rangeMax();
  113. return rangeMin + percent * (rangeMax - rangeMin);
  114. }
  115. /**
  116. * @override
  117. */
  118. invert(value) {
  119. const base = this.base;
  120. const max = log(base, this.max);
  121. const rangeMin = this.rangeMin();
  122. const range = (this.rangeMax() - rangeMin);
  123. let min;
  124. const positiveMin = this.positiveMin;
  125. if (positiveMin) {
  126. if (value === 0) {
  127. return 0;
  128. }
  129. min = log(base, positiveMin / base);
  130. const appendPercent = (1 / (max - min)) * range; // 0 到 positiveMin的占比
  131. if (value < appendPercent) { // 落到 0 - positiveMin 之间
  132. return value / appendPercent * positiveMin;
  133. }
  134. } else {
  135. min = log(base, this.min);
  136. }
  137. const percent = (value - rangeMin) / (range);
  138. const tmp = percent * (max - min) + min;
  139. return Math.pow(base, tmp);
  140. }
  141. }
  142. Base.Log = Log;
  143. module.exports = Log;