@antv/scale/src/scales/threshold.ts

Toolkit for mapping abstract data into visual representation.

import { Base } from './base';
import { Domain, ThresholdOptions, Range } from '../types';
import { bisect, isValid } from '../utils';

/**
 * 将连续的定义域分段，每一段所有的值对应离散的值域中一个值
 */
export class Threshold<O extends ThresholdOptions = ThresholdOptions> extends Base<O> {
  /** threshold 的数量 */
  protected n: number;

  protected thresholds: number[];

  protected getDefaultOptions() {
    return {
      domain: [0.5],
      range: [0, 1],
    } as O;
  }

  constructor(options?: ThresholdOptions) {
    super(options as O);
  }

  /**
   * 二分查找到输入值在哪一段，返回对应的值域中的值
   */
  public map(x: Domain<ThresholdOptions>) {
    if (!isValid(x)) return this.options.unknown;
    const index = bisect(this.thresholds, x, 0, this.n);
    return this.options.range[index];
  }

  /**
   * 在值域中找到对应的值，并返回在定义域中属于哪一段
   */
  public invert(y: Range<ThresholdOptions>) {
    const { range } = this.options;
    const index = range.indexOf(y);
    const domain = this.thresholds;
    return [domain[index - 1], domain[index]];
  }

  public clone() {
    return new Threshold<O>(this.options);
  }

  protected rescale() {
    const { domain, range } = this.options;
    this.n = Math.min(domain.length, range.length - 1);
    this.thresholds = domain;
  }
}