57 lines
1.4 KiB
JavaScript
57 lines
1.4 KiB
JavaScript
import {ascending, bisect, quantileSorted as threshold} from "d3-array";
|
|
import {initRange} from "./init.js";
|
|
|
|
export default function quantile() {
|
|
var domain = [],
|
|
range = [],
|
|
thresholds = [],
|
|
unknown;
|
|
|
|
function rescale() {
|
|
var i = 0, n = Math.max(1, range.length);
|
|
thresholds = new Array(n - 1);
|
|
while (++i < n) thresholds[i - 1] = threshold(domain, i / n);
|
|
return scale;
|
|
}
|
|
|
|
function scale(x) {
|
|
return x == null || isNaN(x = +x) ? unknown : range[bisect(thresholds, x)];
|
|
}
|
|
|
|
scale.invertExtent = function(y) {
|
|
var i = range.indexOf(y);
|
|
return i < 0 ? [NaN, NaN] : [
|
|
i > 0 ? thresholds[i - 1] : domain[0],
|
|
i < thresholds.length ? thresholds[i] : domain[domain.length - 1]
|
|
];
|
|
};
|
|
|
|
scale.domain = function(_) {
|
|
if (!arguments.length) return domain.slice();
|
|
domain = [];
|
|
for (let d of _) if (d != null && !isNaN(d = +d)) domain.push(d);
|
|
domain.sort(ascending);
|
|
return rescale();
|
|
};
|
|
|
|
scale.range = function(_) {
|
|
return arguments.length ? (range = Array.from(_), rescale()) : range.slice();
|
|
};
|
|
|
|
scale.unknown = function(_) {
|
|
return arguments.length ? (unknown = _, scale) : unknown;
|
|
};
|
|
|
|
scale.quantiles = function() {
|
|
return thresholds.slice();
|
|
};
|
|
|
|
scale.copy = function() {
|
|
return quantile()
|
|
.domain(domain)
|
|
.range(range)
|
|
.unknown(unknown);
|
|
};
|
|
|
|
return initRange.apply(scale, arguments);
|
|
}
|