| 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); |
| } |
|
|