molstar
Version:
A comprehensive macromolecular library.
134 lines (133 loc) • 5.27 kB
JavaScript
;
/**
* Copyright (c) 2018-2025 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author Alexander Rose <alexander.rose@weirdbyte.de>
* @author Lukáš Polák <admin@lukaspolak.cz>
*/
Object.defineProperty(exports, "__esModule", { value: true });
exports.ColorScale = exports.DefaultColorScaleProps = void 0;
const color_1 = require("./color");
const lists_1 = require("./lists");
const mol_util_1 = require("../../mol-util");
const legend_1 = require("../legend");
const int_1 = require("../../mol-data/int");
const interpolate_1 = require("../../mol-math/interpolate");
exports.DefaultColorScaleProps = {
domain: [0, 1],
reverse: false,
listOrName: 'red-yellow-blue',
minLabel: '',
maxLabel: '',
};
var ColorScale;
(function (ColorScale) {
function create(props) {
return createColorScaleByType(props, 'continuous');
}
ColorScale.create = create;
function createDiscrete(props) {
return createColorScaleByType(props, 'discrete');
}
ColorScale.createDiscrete = createDiscrete;
function createColorScaleByType(props, type) {
const { domain, reverse, listOrName } = { ...exports.DefaultColorScaleProps, ...props };
const list = typeof listOrName === 'string' ? (0, lists_1.getColorListFromName)(listOrName).list : listOrName;
const colors = reverse ? list.slice().reverse() : list;
let diff = 0, min = 0, max = 0;
function setDomain(_min, _max) {
min = _min;
max = _max;
diff = (max - min) || 1;
}
setDomain(domain[0], domain[1]);
const minLabel = (0, mol_util_1.defaults)(props.minLabel, min.toString());
const maxLabel = (0, mol_util_1.defaults)(props.maxLabel, max.toString());
let color;
const hasOffsets = colors.every(c => Array.isArray(c));
if (hasOffsets) {
const sorted = [...colors];
sorted.sort((a, b) => a[1] - b[1]);
const src = sorted.map(c => c[0]);
const off = int_1.SortedArray.ofSortedArray(sorted.map(c => c[1]));
const max = src.length - 1;
switch (type) {
case 'continuous':
color = (value) => valueToColorWithOffsets(value, src, off, min, max, diff);
break;
case 'discrete':
color = (value) => valueToDiscreteColorWithOffsets(value, src, off, min, max, diff);
break;
}
}
else {
switch (type) {
case 'continuous':
color = (value) => valueToColor(value, colors, min, max, diff);
break;
case 'discrete':
color = (value) => valueToDiscreteColor(value, colors, min, max, diff);
break;
}
}
return {
color,
colorToArray: (value, array, offset) => {
color_1.Color.toArray(color(value), array, offset);
},
normalizedColorToArray: (value, array, offset) => {
color_1.Color.toArrayNormalized(color(value), array, offset);
},
setDomain,
get legend() { return (0, legend_1.ScaleLegend)(minLabel, maxLabel, colors); }
};
}
function valueToColorWithOffsets(value, src, off, min, max, diff) {
const t = (0, interpolate_1.clamp)((value - min) / diff, 0, 1);
const i = int_1.SortedArray.findPredecessorIndex(off, t);
if (i === 0) {
return src[min];
}
else if (i > max) {
return src[max];
}
const o1 = off[i - 1], o2 = off[i];
const t1 = (0, interpolate_1.clamp)((t - o1) / (o2 - o1), 0, 1); // TODO: cache the deltas?
return color_1.Color.interpolate(src[i - 1], src[i], t1);
}
function valueToColor(value, colors, min, max, diff) {
const t = Math.min(colors.length - 1, Math.max(0, ((value - min) / diff) * colors.length - 1));
const tf = Math.floor(t);
const c1 = colors[tf];
const c2 = colors[Math.ceil(t)];
return color_1.Color.interpolate(c1, c2, t - tf);
}
function valueToDiscreteColorWithOffsets(value, src, off, min, max, diff) {
if (src.length === 0) {
return color_1.Color.fromRgb(0, 0, 0);
}
const t = (0, interpolate_1.clamp)((value - min) / diff, 0, 1);
const i = int_1.SortedArray.findPredecessorIndex(off, t);
if (i === 0) {
return src[min];
}
else if (i > max) {
return src[max];
}
return src[i];
}
function valueToDiscreteColor(value, colors, min, max, diff) {
if (colors.length === 0) {
return color_1.Color.fromRgb(0, 0, 0);
}
const intervalSize = diff / colors.length;
if (value <= min) {
return colors[0];
}
else if (value >= max) {
return colors[colors.length - 1];
}
const i = Math.min(colors.length - 1, Math.floor((value - min) / intervalSize));
return colors[i];
}
})(ColorScale || (exports.ColorScale = ColorScale = {}));