UNPKG

@visx/vendor

Version:

vendored packages for visx

126 lines (119 loc) 5.69 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.default = bin; var _array = require("./array.js"); var _bisect = _interopRequireDefault(require("./bisect.js")); var _constant = _interopRequireDefault(require("./constant.js")); var _extent = _interopRequireDefault(require("./extent.js")); var _identity = _interopRequireDefault(require("./identity.js")); var _nice = _interopRequireDefault(require("./nice.js")); var _ticks = _interopRequireWildcard(require("./ticks.js")); var _sturges = _interopRequireDefault(require("./threshold/sturges.js")); function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); } function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function bin() { var value = _identity.default, domain = _extent.default, threshold = _sturges.default; function histogram(data) { if (!Array.isArray(data)) data = Array.from(data); var i, n = data.length, x, step, values = new Array(n); for (i = 0; i < n; ++i) { values[i] = value(data[i], i, data); } var xz = domain(values), x0 = xz[0], x1 = xz[1], tz = threshold(values, x0, x1); // Convert number of thresholds into uniform thresholds, and nice the // default domain accordingly. if (!Array.isArray(tz)) { const max = x1, tn = +tz; if (domain === _extent.default) [x0, x1] = (0, _nice.default)(x0, x1, tn); tz = (0, _ticks.default)(x0, x1, tn); // If the domain is aligned with the first tick (which it will by // default), then we can use quantization rather than bisection to bin // values, which is substantially faster. if (tz[0] <= x0) step = (0, _ticks.tickIncrement)(x0, x1, tn); // If the last threshold is coincident with the domain’s upper bound, the // last bin will be zero-width. If the default domain is used, and this // last threshold is coincident with the maximum input value, we can // extend the niced upper bound by one tick to ensure uniform bin widths; // otherwise, we simply remove the last threshold. Note that we don’t // coerce values or the domain to numbers, and thus must be careful to // compare order (>=) rather than strict equality (===)! if (tz[tz.length - 1] >= x1) { if (max >= x1 && domain === _extent.default) { const step = (0, _ticks.tickIncrement)(x0, x1, tn); if (isFinite(step)) { if (step > 0) { x1 = (Math.floor(x1 / step) + 1) * step; } else if (step < 0) { x1 = (Math.ceil(x1 * -step) + 1) / -step; } } } else { tz.pop(); } } } // Remove any thresholds outside the domain. // Be careful not to mutate an array owned by the user! var m = tz.length, a = 0, b = m; while (tz[a] <= x0) ++a; while (tz[b - 1] > x1) --b; if (a || b < m) tz = tz.slice(a, b), m = b - a; var bins = new Array(m + 1), bin; // Initialize bins. for (i = 0; i <= m; ++i) { bin = bins[i] = []; bin.x0 = i > 0 ? tz[i - 1] : x0; bin.x1 = i < m ? tz[i] : x1; } // Assign data to bins by value, ignoring any outside the domain. if (isFinite(step)) { if (step > 0) { for (i = 0; i < n; ++i) { if ((x = values[i]) != null && x0 <= x && x <= x1) { bins[Math.min(m, Math.floor((x - x0) / step))].push(data[i]); } } } else if (step < 0) { for (i = 0; i < n; ++i) { if ((x = values[i]) != null && x0 <= x && x <= x1) { const j = Math.floor((x0 - x) * step); bins[Math.min(m, j + (tz[j] <= x))].push(data[i]); // handle off-by-one due to rounding } } } } else { for (i = 0; i < n; ++i) { if ((x = values[i]) != null && x0 <= x && x <= x1) { bins[(0, _bisect.default)(tz, x, 0, m)].push(data[i]); } } } return bins; } histogram.value = function (_) { return arguments.length ? (value = typeof _ === "function" ? _ : (0, _constant.default)(_), histogram) : value; }; histogram.domain = function (_) { return arguments.length ? (domain = typeof _ === "function" ? _ : (0, _constant.default)([_[0], _[1]]), histogram) : domain; }; histogram.thresholds = function (_) { return arguments.length ? (threshold = typeof _ === "function" ? _ : (0, _constant.default)(Array.isArray(_) ? _array.slice.call(_) : _), histogram) : threshold; }; return histogram; }