Method getRGB

tensorboard/components/tensor_widget/colormap.ts:160–197

(value: number)

export class JetColorMap extends ColorMap {
  getRGB(value: number): [number, number, number] {
    if (isNaN(value)) {
      // NaN.
      return [MAX_RGB * 0.25, MAX_RGB * 0.25, MAX_RGB * 0.25];
    } else if (!isFinite(value)) {
      if (value < 0) {
        // -Infinity.
        return [MAX_RGB * 0.5, MAX_RGB * 0.5, MAX_RGB * 0.5];
      } else {
        // +Infinity.
        return [MAX_RGB * 0.75, MAX_RGB * 0.75, MAX_RGB * 0.75];
      }
    }

    let relR = 0;
    let relG = 0;
    let relB = 0;
    const lim0 = 0.35;
    const lim1 = 0.65;

    let relValue =
      this.config.min === this.config.max
        ? 0.5
        : (value - this.config.min) / (this.config.max - this.config.min);
    relValue = Math.max(Math.min(relValue, 1), 0);
    if (relValue <= lim0) {
      relG = relValue / lim0;
      relB = 1;
    } else if (relValue > lim0 && relValue <= lim1) {
      relR = (relValue - lim0) / (lim1 - lim0);
      relG = 1;
      relB = (lim1 - relValue) / (lim1 - lim0);
    } else if (relValue > lim1) {
      relR = 1;
      relG = (1 - relValue) / (1 - lim1);
    }
    return [relR * MAX_RGB, relG * MAX_RGB, relB * MAX_RGB];
  }
}