@milaboratories/miplots4 1.0.176 → 1.0.178

package/dist/scatterplot/linearRegression.js

@@ -1,125 +1,105 @@
-import nn from "@stdlib/stats-base-dists-f-cdf";
-import tn from "../node_modules/.pnpm/d3-array@3.2.4/node_modules/d3-array/src/extent.js";
-import en from "../node_modules/.pnpm/d3-shape@3.2.0/node_modules/d3-shape/src/area.js";
-import g from "../node_modules/.pnpm/d3-array@3.2.4/node_modules/d3-array/src/sum.js";
-import w from "../node_modules/.pnpm/d3-array@3.2.4/node_modules/d3-array/src/mean.js";
-const I = 0.05;
-function on(n) {
-  const e = 3.3871328727963665, o = 133.14166789178438, a = 1971.5909503065513, s = 13731.69376550946, m = 45921.95393154987, f = 67265.7709270087, u = 33430.57558358813, i = 2509.0809287301227, r = 42.31333070160091, c = 687.1870074920579, l = 5394.196021424751, x = 21213.794301586597, h = 39307.89580009271, N = 28729.085735721943, v = 5226.495278852854, X = 1.4234371107496835, p = 4.630337846156546, b = 5.769497221460691, q = 3.6478483247632045, y = 1.2704582524523684, Y = 0.2417807251774506, D = 0.022723844989269184, R = 7745450142783414e-19, _ = 2.053191626637759, F = 1.6763848301838038, S = 0.6897673349851, L = 0.14810397642748008, P = 0.015198666563616457, $ = 5475938084995345e-19, z = 10507500716444169e-25, A = 6.657904643501103, B = 5.463784911164114, O = 1.7848265399172913, T = 0.29656057182850487, j = 0.026532189526576124, k = 0.0012426609473880784, G = 27115555687434876e-21, H = 20103343992922881e-23, J = 0.599832206555888, Q = 0.1369298809227358, U = 0.014875361290850615, W = 7868691311456133e-19, Z = 18463183175100548e-21, K = 1421511758316446e-22, d = 20442631033899397e-31, E = n - 0.5;
-  let t, M;
-  return Math.abs(E) <= 0.425 ? (t = 0.180625 - E * E, M = E * (((((((i * t + u) * t + f) * t + m) * t + s) * t + a) * t + o) * t + e) / (((((((v * t + N) * t + h) * t + x) * t + l) * t + c) * t + r) * t + 1), M) : (t = n, t = Math.sqrt(-Math.log(t)), t <= 5 ? (t += -1.6, M = (((((((R * t + D) * t + Y) * t + y) * t + q) * t + b) * t + p) * t + X) / (((((((z * t + $) * t + P) * t + L) * t + S) * t + F) * t + _) * t + 1)) : (t += -5, M = (((((((H * t + G) * t + k) * t + j) * t + T) * t + O) * t + B) * t + A) / (((((((d * t + K) * t + Z) * t + W) * t + U) * t + Q) * t + J) * t + 1)), M = -M, M);
+import { area as e } from "d3-shape";
+import { extent as t, mean as n, sum as r } from "d3-array";
+import i from "@stdlib/stats-base-dists-f-cdf";
+var a = .05;
+function o(e) {
+	if (e < 0 || e > 1) throw Error("P p < 0 || p > 1 in linear regression calculation");
+	if (e == 0) return -Infinity;
+	if (e == 1) return Infinity;
+	let t = e - .5, n, r;
+	return Math.abs(t) <= .425 ? (n = .180625 - t * t, r = t * (((((((2509.0809287301227 * n + 33430.57558358813) * n + 67265.7709270087) * n + 45921.95393154987) * n + 13731.69376550946) * n + 1971.5909503065513) * n + 133.14166789178438) * n + 3.3871328727963665) / (((((((5226.495278852854 * n + 28729.085735721943) * n + 39307.89580009271) * n + 21213.794301586597) * n + 5394.196021424751) * n + 687.1870074920579) * n + 42.31333070160091) * n + 1), r) : (n = t > 0 ? 1 - e : e, n = Math.sqrt(-Math.log(n)), n <= 5 ? (n += -1.6, r = (((((((.0007745450142783414 * n + .022723844989269184) * n + .2417807251774506) * n + 1.2704582524523684) * n + 3.6478483247632045) * n + 5.769497221460691) * n + 4.630337846156546) * n + 1.4234371107496835) / (((((((1.0507500716444169e-9 * n + .0005475938084995345) * n + .015198666563616457) * n + .14810397642748008) * n + .6897673349851) * n + 1.6763848301838038) * n + 2.053191626637759) * n + 1)) : (n += -5, r = (((((((2.0103343992922881e-7 * n + 27115555687434876e-21) * n + .0012426609473880784) * n + .026532189526576124) * n + .29656057182850487) * n + 1.7848265399172913) * n + 5.463784911164114) * n + 6.657904643501103) / (((((((20442631033899397e-31 * n + 1.421511758316446e-7) * n + 18463183175100548e-21) * n + .0007868691311456133) * n + .014875361290850615) * n + .1369298809227358) * n + .599832206555888) * n + 1)), t < 0 && (r = -r), r);
 }
-function V(n, e) {
-  const { sin: o, cos: a, sqrt: s, pow: m, exp: f, PI: u } = Math;
-  let i, r;
-  if (e == 1) return a(n * u / 2) / o(n * u / 2);
-  if (e == 2) return s(2 / (n * (2 - n)) - 2);
-  const c = 1 / (e - 0.5), l = 48 / (c * c);
-  let x = ((20700 * c / l - 98) * c - 16) * c + 96.36;
-  const h = ((94.5 / (l + x) - 3) / l + 1) * s(c * u * 0.5) * e;
-  return i = h * n, r = m(i, 2 / e), r > 0.05 + c ? (i = on(n / 2), r = i * i, e < 5 && (x = x + 0.3 * (e - 4.5) * (i + 0.6)), x = (((0.05 * h * i - 5) * i - 7) * i - 2) * i + l + x, r = (((((0.4 * r + 6.3) * r + 36) * r + 94.5) / x - r - 3) / l + 1) * i, r = c * r * r, r > 2e-3 ? r = f(r) - 1 : r = 0.5 * r * r + r) : r = ((1 / (((e + 6) / (e * r) - 0.089 * h - 0.822) * (e + 2) * 3) + 0.5 / (e + 4)) * r - 1) * (e + 1) / (e + 2) + 1 / r, s(e * r);
+function s(e, t) {
+	let { sin: n, cos: r, sqrt: i, pow: a, exp: s, PI: c } = Math, l, u;
+	if (t == 1) return r(e * c / 2) / n(e * c / 2);
+	if (t == 2) return i(2 / (e * (2 - e)) - 2);
+	let d = 1 / (t - .5), f = 48 / (d * d), p = ((20700 * d / f - 98) * d - 16) * d + 96.36, m = ((94.5 / (f + p) - 3) / f + 1) * i(d * c * .5) * t;
+	return l = m * e, u = a(l, 2 / t), u > .05 + d ? (l = o(e / 2), u = l * l, t < 5 && (p += .3 * (t - 4.5) * (l + .6)), p = (((.05 * m * l - 5) * l - 7) * l - 2) * l + f + p, u = (((((.4 * u + 6.3) * u + 36) * u + 94.5) / p - u - 3) / f + 1) * l, u = d * u * u, u = u > .002 ? s(u) - 1 : .5 * u * u + u) : u = ((1 / (((t + 6) / (t * u) - .089 * m - .822) * (t + 2) * 3) + .5 / (t + 4)) * u - 1) * (t + 1) / (t + 2) + 1 / u, i(t * u);
 }
-function rn(n) {
-  let e = n.length, o = 0, a = 0, s = 0, m = 0;
-  for (let i = 0; i < n.length; ++i) {
-    const r = n[i].x, c = n[i].y;
-    if (!Number.isFinite(r) || !Number.isFinite(c)) {
-      --e;
-      continue;
-    }
-    o += r, a += c, s += r * c, m += r * r;
-  }
-  const f = (e * s - o * a) / (e * m - o * o), u = (a - f * o) / e;
-  return [f, u];
+function c(e) {
+	let t = e.length, n = 0, r = 0, i = 0, a = 0;
+	for (let o = 0; o < e.length; ++o) {
+		let s = e[o].x, c = e[o].y;
+		if (!Number.isFinite(s) || !Number.isFinite(c)) {
+			--t;
+			continue;
+		}
+		n += s, r += c, i += s * c, a += s * s;
+	}
+	let o = (t * i - n * r) / (t * a - n * n);
+	return [o, (r - o * n) / t];
 }
-function cn(n, e, o) {
-  const a = (l) => l * e + o, s = n.length, m = g(n, (l) => l.x) / n.length, f = V(I, s - 2), u = Math.sqrt(
-    g(n, (l) => (l.y - a(l.x)) ** 2) / (s - 2) / g(n, (l) => (l.x - m) ** 2)
-  ), i = u * f, c = u * Math.sqrt(g(n, (l) => l.x ** 2) / s) * f;
-  return { slope: e, intercept: o, slopeError: i, interceptError: c };
+function l(e, t, n) {
+	let i = (e) => e * t + n, o = e.length, c = r(e, (e) => e.x) / e.length, l = s(a, o - 2), u = Math.sqrt(r(e, (e) => (e.y - i(e.x)) ** 2) / (o - 2) / r(e, (e) => (e.x - c) ** 2));
+	return {
+		slope: t,
+		intercept: n,
+		slopeError: u * l,
+		interceptError: u * Math.sqrt(r(e, (e) => e.x ** 2) / o) * l
+	};
 }
-function sn(n, e) {
-  const o = g(n, (u) => u.x) / n.length;
-  let a = 0, s = 0;
-  for (let u = 0; u < n.length; ++u)
-    a += Math.pow(n[u].x - o, 2), s += Math.pow(n[u].y - e(n[u].x), 2);
-  const m = Math.sqrt(s / (n.length - 2)), f = V(I, n.length - 2);
-  return function(u) {
-    const i = e(u), r = m * Math.sqrt(1 / n.length + Math.pow(u - o, 2) / a), c = i - f * r, l = i + f * r;
-    return { x: u, y: i, left: isNaN(c) ? i : c, right: isNaN(l) ? i : l };
-  };
+function u(e, t) {
+	let n = r(e, (e) => e.x) / e.length, i = 0, o = 0;
+	for (let r = 0; r < e.length; ++r) i += (e[r].x - n) ** 2, o += (e[r].y - t(e[r].x)) ** 2;
+	let c = Math.sqrt(o / (e.length - 2)), l = s(a, e.length - 2);
+	return function(r) {
+		let a = t(r), o = c * Math.sqrt(1 / e.length + (r - n) ** 2 / i), s = a - l * o, u = a + l * o;
+		return {
+			x: r,
+			y: a,
+			left: isNaN(s) ? a : s,
+			right: isNaN(u) ? a : u
+		};
+	};
 }
-function un(n) {
-  const e = n.length;
-  return Math.pow(
-    (e * g(n, (o) => o.x * o.y) - g(n, (o) => o.x) * g(n, (o) => o.y)) / (Math.sqrt(
-      e * g(n, (o) => o.x * o.x) - Math.pow(
-        g(n, (o) => o.x),
-        2
-      )
-    ) * Math.sqrt(
-      e * g(n, (o) => o.y * o.y) - Math.pow(
-        g(n, (o) => o.y),
-        2
-      )
-    )),
-    2
-  );
+function d(e) {
+	let t = e.length;
+	return ((t * r(e, (e) => e.x * e.y) - r(e, (e) => e.x) * r(e, (e) => e.y)) / (Math.sqrt(t * r(e, (e) => e.x * e.x) - r(e, (e) => e.x) ** 2) * Math.sqrt(t * r(e, (e) => e.y * e.y) - r(e, (e) => e.y) ** 2))) ** 2;
 }
-function ln(n, e) {
-  const o = w(n, e);
-  return w(n.map((a) => (e(a) - o) ** 2));
+function f(e, t) {
+	let r = n(e, t);
+	return n(e.map((e) => (t(e) - r) ** 2));
 }
-function C(n, e) {
-  return Math.sqrt(ln(n, e));
+function p(e, t) {
+	return Math.sqrt(f(e, t));
 }
-function an(n, e, o) {
-  const a = w(n, e), s = w(n, o), m = n.length;
-  return g(n, (f) => (f.x - a) * (f.y - s)) / m;
+function m(e, t, i) {
+	let a = n(e, t), o = n(e, i), s = e.length;
+	return r(e, (e) => (e.x - a) * (e.y - o)) / s;
 }
-function fn(n) {
-  const e = an(
-    n,
-    (s) => s.x,
-    (s) => s.y
-  ), o = C(n, (s) => s.x), a = C(n, (s) => s.y);
-  return e / (o * a);
+function h(e) {
+	return m(e, (e) => e.x, (e) => e.y) / (p(e, (e) => e.x) * p(e, (e) => e.y));
 }
-function mn(n, e) {
-  const o = n.length, a = 1, s = 1, m = o - a - 1, f = w(n, (h) => h.y), u = g(n, (h) => (e(h.x) - f) ** 2), i = g(n, (h) => (e(h.x) - h.y) ** 2), r = u / s, c = i / m, l = r / c;
-  return 1 - nn(l, s, m);
+function g(e, t) {
+	let a = e.length - 1 - 1, o = n(e, (e) => e.y), s = r(e, (e) => (t(e.x) - o) ** 2), c = r(e, (e) => (t(e.x) - e.y) ** 2);
+	return 1 - i(s / 1 / (c / a), 1, a);
 }
-function Mn(n, e, o, a, s, m) {
-  return m ? o.reduce((f, u) => {
-    const i = e[u].dots;
-    return f[u] = a.map((r) => {
-      const c = i.filter(
-        (p) => s.every((b, q) => String(n.getColumnValue(b.value, p.idx)) === r[q])
-      );
-      if (c.length === 0) return null;
-      const [l = 0, x = 0] = tn(c, (p) => p.x), [h, N] = rn(c), v = (p) => h * p + N, X = sn(c, v);
-      return {
-        idx: c[0].idx,
-        predict: v,
-        getInterval: X,
-        getArea: (p, b, q) => en().x((y) => p(y.x)).y0((y) => b(y.left)).y1((y) => b(y.right))(q.map(X)) ?? "",
-        color: m.color,
-        bounded: m.bounded,
-        showStats: !0,
-        xBounds: [l, x],
-        stats: {
-          r2: un(c),
-          r: fn(c),
-          pValue: mn(c, v),
-          coefficients: cn(c, h, N)
-        }
-      };
-    }).filter((r) => r !== null), f;
-  }, {}) : null;
+function _(n, r, i, a, o, s) {
+	return s ? i.reduce((i, f) => {
+		let p = r[f].dots;
+		return i[f] = a.map((r) => {
+			let i = p.filter((e) => o.every((t, i) => String(n.getColumnValue(t.value, e.idx)) === r[i]));
+			if (i.length === 0) return null;
+			let [a = 0, f = 0] = t(i, (e) => e.x), [m, _] = c(i), v = (e) => m * e + _, y = u(i, v);
+			return {
+				idx: i[0].idx,
+				predict: v,
+				getInterval: y,
+				getArea: (t, n, r) => e().x((e) => t(e.x)).y0((e) => n(e.left)).y1((e) => n(e.right))(r.map(y)) ?? "",
+				color: s.color,
+				bounded: s.bounded,
+				showStats: !0,
+				xBounds: [a, f],
+				stats: {
+					r2: d(i),
+					r: h(i),
+					pValue: g(i, v),
+					coefficients: l(i, m, _)
+				}
+			};
+		}).filter((e) => e !== null), i;
+	}, {}) : null;
 }
-export {
-  sn as confidenceInterval,
-  Mn as getRegressionData,
-  rn as linearRegression
-};
-//# sourceMappingURL=linearRegression.js.map
+export { _ as getRegressionData };
+
+//# sourceMappingURL=linearRegression.js.map

package/dist/scatterplot/linearRegression.js.map

@@ -1 +1 @@
-{"version":3,"file":"linearRegression.js","sources":["../../src/scatterplot/linearRegression.ts"],"sourcesContent":["/* eslint-disable no-loss-of-precision */\n// @ts-ignore\nimport fCDF from '@stdlib/stats-base-dists-f-cdf';\nimport { extent, mean, sum } from 'd3-array';\nimport type { ScaleLinear } from 'd3-scale';\nimport { area } from 'd3-shape';\nimport type { DataFrame } from '../DataFrame';\nimport type { ColumnName, ContinuousAesFromColumn, InheritAesScatterplot } from '../types';\nimport type { Dot, GroupedDots } from './dots';\nimport type { ScatterplotSettingsImpl } from './ScatterplotSettingsImpl';\n\ntype NumericDot = {x: number, y:number};\nconst CONFIDENCE_LEVEL = 0.05;\n\n/*\n Returns the normal deviate Z corresponding to a given lower tail area of P; Z is accurate to about 1 part in 10**16.\n\n Wichura, M. J. (1988) Algorithm AS 241: The percentage points of the normal distribution. Applied Statistics, 37, 477–484.\n\n extension of:\n Beasley, J. D./ Springer, S. G. (1977), The percentage points of the NormalDistribution, Applied Statistics. 26, 118–121.\n\n Excel: STANDNORMINV(); R: qnorm()\n */\n\nfunction normdev(p: number) {\n    if (p < 0 || p > 1) throw Error('P p < 0 || p > 1 in linear regression calculation');\n    if (p == 0) return -Infinity;\n    if (p == 1) return Infinity;\n\n    const a0 = 3.387132872796366608,\n        a1 = 1.3314166789178437745e2,\n        a2 = 1.9715909503065514427e3,\n        a3 = 1.3731693765509461125e4,\n        a4 = 4.5921953931549871457e4,\n        a5 = 6.7265770927008700853e4,\n        a6 = 3.3430575583588128105e4,\n        a7 = 2.5090809287301226727e3,\n        b1 = 4.2313330701600911252e1,\n        b2 = 6.871870074920579083e2,\n        b3 = 5.3941960214247511077e3,\n        b4 = 2.1213794301586595867e4,\n        b5 = 3.930789580009271061e4,\n        b6 = 2.8729085735721942674e4,\n        b7 = 5.226495278852854561e3,\n        c0 = 1.42343711074968357734,\n        c1 = 4.6303378461565452959,\n        c2 = 5.7694972214606914055,\n        c3 = 3.64784832476320460504,\n        c4 = 1.27045825245236838258,\n        c5 = 2.4178072517745061177e-1,\n        c6 = 2.27238449892691845833e-2,\n        c7 = 7.7454501427834140764e-4,\n        d1 = 2.05319162663775882187,\n        d2 = 1.6763848301838038494,\n        d3 = 6.8976733498510000455e-1,\n        d4 = 1.4810397642748007459e-1,\n        d5 = 1.51986665636164571966e-2,\n        d6 = 5.475938084995344946e-4,\n        d7 = 1.05075007164441684324e-9,\n        e0 = 6.6579046435011037772,\n        e1 = 5.4637849111641143699,\n        e2 = 1.7848265399172913358,\n        e3 = 2.9656057182850489123e-1,\n        e4 = 2.6532189526576123093e-2,\n        e5 = 1.2426609473880784386e-3,\n        e6 = 2.71155556874348757815e-5,\n        e7 = 2.01033439929228813265e-7,\n        f1 = 5.9983220655588793769e-1,\n        f2 = 1.3692988092273580531e-1,\n        f3 = 1.48753612908506148525e-2,\n        f4 = 7.868691311456132591e-4,\n        f5 = 1.8463183175100546818e-5,\n        f6 = 1.4215117583164458887e-7,\n        f7 = 2.04426310338993978564e-15;\n\n    const q = p - 0.5;\n    let r, z;\n\n    // p close to 0.5\n    if (Math.abs(q) <= 0.425) {\n        r = 0.180625 - q * q;\n        z =\n            (q * (((((((a7 * r + a6) * r + a5) * r + a4) * r + a3) * r + a2) * r + a1) * r + a0)) /\n            (((((((b7 * r + b6) * r + b5) * r + b4) * r + b3) * r + b2) * r + b1) * r + 1);\n        return z;\n    }\n\n    if (q > 0) r = 1 - p;\n    else r = p;\n    r = Math.sqrt(-Math.log(r));\n\n    // p neither close to 0.5 nor 0 or 1\n    if (r <= 5) {\n        r += -1.6;\n        z =\n            (((((((c7 * r + c6) * r + c5) * r + c4) * r + c3) * r + c2) * r + c1) * r + c0) /\n            (((((((d7 * r + d6) * r + d5) * r + d4) * r + d3) * r + d2) * r + d1) * r + 1);\n    }\n    // p near 0 or 1\n    else {\n        r += -5;\n        z =\n            (((((((e7 * r + e6) * r + e5) * r + e4) * r + e3) * r + e2) * r + e1) * r + e0) /\n            (((((((f7 * r + f6) * r + f5) * r + f4) * r + f3) * r + f2) * r + f1) * r + 1);\n    }\n\n    if (q < 0.0) z = -z;\n    return z;\n}\n\n/*\nHill's approximated inverse t-distribution calculates t given df and two-tail probability:\nHill, G. W. (1970), Algorithm 396: Student's t-quantiles. Communications of the ACM, 13(10), 619–620.\n\nResult should be \"correct to at least 6 significant digits even for the analytic continuation through noninteger values of n > 5\". For higher precision (used in R, ...) see:\nHill, G. W. (1981) Remark on Algorithm 396, ACM Transactions on Mathematical Software, 7, 250–1.\n\nExcel: TINV(); R: qt()\n*/\nfunction inverseT(p: number, df: number) {\n    const {sin, cos, sqrt, pow, exp, PI} = Math;\n    let x: number, y: number;\n\n    if (df == 1) return cos((p * PI) / 2) / sin((p * PI) / 2);\n    if (df == 2) return sqrt(2 / (p * (2 - p)) - 2);\n\n    const a = 1 / (df - 0.5);\n    const b = 48 / (a * a);\n    let c = (((20700 * a) / b - 98) * a - 16) * a + 96.36;\n    const d = ((94.5 / (b + c) - 3) / b + 1) * sqrt(a * PI * 0.5) * df;\n    x = d * p;\n    y = pow(x, 2 / df);\n\n    if (y > 0.05 + a) {\n        // The procedure normdev(p) is assumed to return a negative normal\n        // deviate at the lower tail probability level p, e.g. -2.32 for p = 0.01.\n        x = normdev(p / 2);\n        y = x * x;\n        if (df < 5) c = c + 0.3 * (df - 4.5) * (x + 0.6);\n        c = (((0.05 * d * x - 5) * x - 7) * x - 2) * x + b + c;\n        y = (((((0.4 * y + 6.3) * y + 36) * y + 94.5) / c - y - 3) / b + 1) * x;\n        y = a * y * y;\n        if (y > 0.002) y = exp(y) - 1;\n        else y = 0.5 * y * y + y;\n    } else {\n        y =\n            (((1 / (((df + 6) / (df * y) - 0.089 * d - 0.822) * (df + 2) * 3) + 0.5 / (df + 4)) * y - 1) * (df + 1)) /\n                (df + 2) +\n            1 / y;\n    }\n\n    return sqrt(df * y);\n}\n\nexport function linearRegression(data: NumericDot[]): [number, number] {\n    let n = data.length,\n        sumX = 0,\n        sumY = 0,\n        sumXY = 0,\n        sumX2 = 0;\n    for (let i = 0; i < data.length; ++i) {\n        const xi = data[i].x;\n        const yi = data[i].y;\n        if (!Number.isFinite(xi) || !Number.isFinite(yi)) {\n            --n;\n            continue;\n        }\n        sumX += xi;\n        sumY += yi;\n        sumXY += xi * yi;\n        sumX2 += xi * xi;\n    }\n    const slope = (n * sumXY - sumX * sumY) / (n * sumX2 - sumX * sumX);\n    const intercept = (sumY - slope * sumX) / n;\n\n    return [slope, intercept];\n}\n\nfunction getCoefficientsInterval(dots: NumericDot[], slope: number, intercept: number) {\n    const predict = (x: number) => x * slope + intercept;\n    const n = dots.length;\n    const meanX = sum(dots, d => d.x) / dots.length;\n    const t = inverseT(CONFIDENCE_LEVEL, n - 2);\n\n    const se_b = Math.sqrt(\n        sum(dots, dot => (dot.y - predict(dot.x)) ** 2) / (n - 2) / sum(dots, dot => (dot.x - meanX) ** 2)\n    );\n    const slopeError = se_b * t;\n\n    const se_a = se_b * Math.sqrt(sum(dots, dot => dot.x ** 2) / n);\n    const interceptError = se_a * t;\n\n    return {slope, intercept, slopeError, interceptError};\n}\n\nexport type PointIntervalInfo = {\n    x: number; y: number, left: number; right: number\n}\n// https://stats.stackexchange.com/questions/101318/understanding-shape-and-calculation-of-confidence-bands-in-linear-regression\nexport function confidenceInterval(\n    data: NumericDot[],\n    predict: (x: number) => number\n): (x: number) => PointIntervalInfo {\n    const mean = sum(data, d => d.x) / data.length;\n    let a = 0,\n        b = 0;\n    for (let i = 0; i < data.length; ++i) {\n        a += Math.pow(data[i].x - mean, 2);\n        b += Math.pow(data[i].y - predict(data[i].x), 2);\n    }\n    const sy = Math.sqrt(b / (data.length - 2));\n    const t = inverseT(CONFIDENCE_LEVEL, data.length - 2);\n    return function (x: number) {\n        const Y = predict(x);\n        const se = sy * Math.sqrt(1 / data.length + Math.pow(x - mean, 2) / a);\n        const left = Y - t * se;\n        const right = Y + t * se;\n        return {x, y: Y, left: isNaN(left) ? Y : left, right: isNaN(right) ? Y : right};\n    };\n}\n\nfunction getR2(dots: NumericDot[]) {\n    const n = dots.length;\n    return Math.pow(\n        (n * sum(dots, (d: NumericDot) => d.x * d.y) - sum(dots, (d: NumericDot) => d.x) * sum(dots, (d: NumericDot) => d.y)) /\n            (Math.sqrt(\n                n * sum(dots, d => d.x * d.x) -\n                    Math.pow(\n                        sum(dots, d => d.x),\n                        2\n                    )\n            ) *\n                Math.sqrt(\n                    n * sum(dots, d => d.y * d.y) -\n                        Math.pow(\n                            sum(dots, d => d.y),\n                            2\n                        )\n                )),\n        2\n    );\n}\n\nfunction getVariance(arr: NumericDot[], getter: (d: NumericDot) => number) {\n    const vMean = mean(arr, getter) as number;\n\n    return mean(arr.map(v => (getter(v) - vMean) ** 2)) as number;\n}\n\nfunction getStandardDeviation(arr: NumericDot[], getter: (d: NumericDot) => number) {\n    return Math.sqrt(getVariance(arr, getter));\n}\n\nfunction getCovariance(arr: NumericDot[], getterX: (d: NumericDot) => number, getterY: (d: NumericDot) => number) {\n    const meanX = mean(arr, getterX) as number;\n    const meanY = mean(arr, getterY) as number;\n    const n = arr.length;\n\n    return sum(arr, d => (d.x - meanX) * (d.y - meanY)) / n;\n}\n\nfunction getR(dots: NumericDot[]) {\n    const cov = getCovariance(\n        dots,\n        d => d.x,\n        d => d.y\n    );\n    const sigmaX = getStandardDeviation(dots, d => d.x);\n    const sigmaY = getStandardDeviation(dots, d => d.y);\n\n    return cov / (sigmaX * sigmaY);\n}\n\nfunction getPValue(dots: NumericDot[], predict: (x: number) => number) {\n    const n = dots.length;\n    const p = 1; // number of coefficients\n    const dfm = 1; // degrees of freedom model\n    const dfe = n - p - 1; // degrees of freedom error\n    const meanY = mean(dots, d => d.y) as number;\n    const ssr = sum(dots, d => (predict(d.x) - meanY) ** 2); // sum of squares regression\n    const sse = sum(dots, d => (predict(d.x) - d.y) ** 2); // sum of squares errors\n\n    const msr = ssr / dfm;\n    const mse = sse / dfe;\n\n    const f = msr / mse;\n\n    const pval = fCDF(f, dfm, dfe);\n\n    return 1.0 - pval; // \"greater\" p-value\n}\n\ntype TrendInfo = {\n    idx: number;\n    predict: (x: number) => number;\n    getInterval: (x: number) => PointIntervalInfo;\n    getArea: (x: ScaleLinear<number, number>, y: ScaleLinear<number, number>, range: number[]) => string;\n    color: string | InheritAesScatterplot | ContinuousAesFromColumn<string>;\n    bounded: boolean;\n    showStats: boolean;\n    xBounds: [number, number];\n    stats: {\n        r2?: number;\n        r?: number;\n        pValue?: number;\n        coefficients?: {slope: number, intercept: number, slopeError:number, interceptError: number};\n    };\n};\nexport type TrendsData = Record<string, TrendInfo[]>;\n\nexport function getRegressionData(\n    dataFrame: DataFrame,\n    dotsByFacets: GroupedDots,\n    facetKeys: string[],\n    groupingKeysValues: string[][],\n    grouping: ColumnName[],\n    trend: ScatterplotSettingsImpl['trend']\n): TrendsData | null {\n    if (!trend) {\n        return null;\n    }\n    return facetKeys.reduce((res: TrendsData, facetKey) => {\n        const facetDots = dotsByFacets[facetKey].dots;\n        \n        res[facetKey] = groupingKeysValues\n            .map(values => {\n                const dots = facetDots.filter(dot =>\n                    grouping.every((column, idx) => String(dataFrame.getColumnValue(column.value, dot.idx)) === values[idx])\n                ) as NumericDot[];\n\n                if (dots.length === 0) return null;\n\n                const [minX = 0, maxX = 0] = extent(dots, (dot: NumericDot) => dot.x);\n                const [slope, intercept] = linearRegression(dots);\n                const predict = (x: number) => slope * x + intercept;\n                const getInterval = confidenceInterval(dots, predict);\n\n                return {\n                    idx: (dots[0] as Dot).idx,\n                    predict,\n                    getInterval,\n                    getArea: (scaleX: ScaleLinear<number, number>, scaleY: ScaleLinear<number, number>, range: number[]) =>\n                        area<PointIntervalInfo>()\n                            .x(d => scaleX(d.x))\n                            .y0(d => scaleY(d.left))\n                            .y1(d => scaleY(d.right))(range.map(getInterval)) ?? '',\n                    color: trend.color,\n                    bounded: trend.bounded,\n                    showStats: true,\n                    xBounds: [minX, maxX],\n                    stats: {\n                        r2: getR2(dots),\n                        r: getR(dots),\n                        pValue: getPValue(dots, predict),\n                        coefficients: getCoefficientsInterval(dots, slope, intercept),\n                    },\n                };\n            })\n            .filter((item) => item !== null) as TrendInfo[];\n        \n        return res;\n    }, {});\n}\n"],"names":["CONFIDENCE_LEVEL","normdev","p","a0","a1","a2","a3","a4","a5","a6","a7","b1","b2","b3","b4","b5","b6","b7","c0","c1","c2","c3","c4","c5","c6","c7","d1","d2","d3","d4","d5","d6","d7","e0","e1","e2","e3","e4","e5","e6","e7","f1","f2","f3","f4","f5","f6","f7","q","r","z","inverseT","df","sin","cos","sqrt","pow","exp","PI","x","y","a","b","c","d","linearRegression","data","n","sumX","sumY","sumXY","sumX2","xi","yi","slope","intercept","getCoefficientsInterval","dots","predict","meanX","sum","t","se_b","dot","slopeError","interceptError","confidenceInterval","mean","i","sy","Y","se","left","right","getR2","getVariance","arr","getter","vMean","v","getStandardDeviation","getCovariance","getterX","getterY","meanY","getR","cov","sigmaX","sigmaY","getPValue","dfm","dfe","ssr","sse","msr","mse","f","fCDF","getRegressionData","dataFrame","dotsByFacets","facetKeys","groupingKeysValues","grouping","trend","res","facetKey","facetDots","values","column","idx","minX","maxX","extent","getInterval","scaleX","scaleY","range","area","item"],"mappings":";;;;;AAYA,MAAMA,IAAmB;AAazB,SAASC,GAAQC,GAAW;AAKxB,QAAMC,IAAK,oBACPC,IAAK,oBACLC,IAAK,oBACLC,IAAK,mBACLC,IAAK,mBACLC,IAAK,kBACLC,IAAK,mBACLC,IAAK,oBACLC,IAAK,mBACLC,IAAK,mBACLC,IAAK,mBACLC,IAAK,oBACLC,IAAK,mBACLC,IAAK,oBACLC,IAAK,mBACLC,IAAK,oBACLC,IAAK,mBACLC,IAAK,mBACLC,IAAK,oBACLC,IAAK,oBACLC,IAAK,oBACLC,IAAK,sBACLC,IAAK,sBACLC,IAAK,mBACLC,IAAK,oBACLC,IAAK,iBACLC,IAAK,qBACLC,IAAK,sBACLC,IAAK,sBACLC,IAAK,uBACLC,IAAK,mBACLC,IAAK,mBACLC,IAAK,oBACLC,IAAK,qBACLC,IAAK,sBACLC,IAAK,uBACLC,IAAK,uBACLC,IAAK,uBACLC,IAAK,mBACLC,IAAK,oBACLC,IAAK,sBACLC,IAAK,sBACLC,IAAK,uBACLC,IAAK,sBACLC,IAAK,uBAEHC,IAAI9C,IAAI;AACd,MAAI+C,GAAGC;AAGP,SAAI,KAAK,IAAIF,CAAC,KAAK,SACfC,IAAI,WAAWD,IAAIA,GACnBE,IACKF,WAAWtC,IAAKuC,IAAIxC,KAAMwC,IAAIzC,KAAMyC,IAAI1C,KAAM0C,IAAI3C,KAAM2C,IAAI5C,KAAM4C,IAAI7C,KAAM6C,IAAI9C,YAC1Ec,IAAKgC,IAAIjC,KAAMiC,IAAIlC,KAAMkC,IAAInC,KAAMmC,IAAIpC,KAAMoC,IAAIrC,KAAMqC,IAAItC,KAAMsC,IAAI,IACzEC,MAIND,IAAI/C,GACT+C,IAAI,KAAK,KAAK,CAAC,KAAK,IAAIA,CAAC,CAAC,GAGtBA,KAAK,KACLA,KAAK,MACLC,WACWzB,IAAKwB,IAAIzB,KAAMyB,IAAI1B,KAAM0B,IAAI3B,KAAM2B,IAAI5B,KAAM4B,IAAI7B,KAAM6B,IAAI9B,KAAM8B,IAAI/B,YACrEc,IAAKiB,IAAIlB,KAAMkB,IAAInB,KAAMmB,IAAIpB,KAAMoB,IAAIrB,KAAMqB,IAAItB,KAAMsB,IAAIvB,KAAMuB,IAAI,OAIhFA,KAAK,IACLC,WACWV,IAAKS,IAAIV,KAAMU,IAAIX,KAAMW,IAAIZ,KAAMY,IAAIb,KAAMa,IAAId,KAAMc,IAAIf,KAAMe,IAAIhB,YACrEc,IAAKE,IAAIH,KAAMG,IAAIJ,KAAMI,IAAIL,KAAMK,IAAIN,KAAMM,IAAIP,KAAMO,IAAIR,KAAMQ,IAAI,KAGvEC,IAAI,CAACA,GACXA;AACX;AAWA,SAASC,EAASjD,GAAWkD,GAAY;AACrC,QAAM,EAAC,KAAAC,GAAK,KAAAC,GAAK,MAAAC,GAAM,KAAAC,GAAK,KAAAC,GAAK,IAAAC,MAAM;AACvC,MAAIC,GAAWC;AAEf,MAAIR,KAAM,EAAG,QAAOE,EAAKpD,IAAIwD,IAAM,CAAC,IAAIL,EAAKnD,IAAIwD,IAAM,CAAC;AACxD,MAAIN,KAAM,EAAG,QAAOG,EAAK,KAAKrD,KAAK,IAAIA,MAAM,CAAC;AAE9C,QAAM2D,IAAI,KAAKT,IAAK,MACdU,IAAI,MAAMD,IAAIA;AACpB,MAAIE,MAAO,QAAQF,IAAKC,IAAI,MAAMD,IAAI,MAAMA,IAAI;AAChD,QAAMG,MAAM,QAAQF,IAAIC,KAAK,KAAKD,IAAI,KAAKP,EAAKM,IAAIH,IAAK,GAAG,IAAIN;AAChE,SAAAO,IAAIK,IAAI9D,GACR0D,IAAIJ,EAAIG,GAAG,IAAIP,CAAE,GAEbQ,IAAI,OAAOC,KAGXF,IAAI1D,GAAQC,IAAI,CAAC,GACjB0D,IAAID,IAAIA,GACJP,IAAK,MAAGW,IAAIA,IAAI,OAAOX,IAAK,QAAQO,IAAI,OAC5CI,OAAO,OAAOC,IAAIL,IAAI,KAAKA,IAAI,KAAKA,IAAI,KAAKA,IAAIG,IAAIC,GACrDH,SAAS,MAAMA,IAAI,OAAOA,IAAI,MAAMA,IAAI,QAAQG,IAAIH,IAAI,KAAKE,IAAI,KAAKH,GACtEC,IAAIC,IAAID,IAAIA,GACRA,IAAI,OAAOA,IAAIH,EAAIG,CAAC,IAAI,IACvBA,IAAI,MAAMA,IAAIA,IAAIA,KAEvBA,MACO,OAAOR,IAAK,MAAMA,IAAKQ,KAAK,QAAQI,IAAI,UAAUZ,IAAK,KAAK,KAAK,OAAOA,IAAK,MAAMQ,IAAI,MAAMR,IAAK,MAChGA,IAAK,KACV,IAAIQ,GAGLL,EAAKH,IAAKQ,CAAC;AACtB;AAEO,SAASK,GAAiBC,GAAsC;AACnE,MAAIC,IAAID,EAAK,QACTE,IAAO,GACPC,IAAO,GACPC,IAAQ,GACRC,IAAQ;AACZ,WAAS,IAAI,GAAG,IAAIL,EAAK,QAAQ,EAAE,GAAG;AAClC,UAAMM,IAAKN,EAAK,CAAC,EAAE,GACbO,IAAKP,EAAK,CAAC,EAAE;AACnB,QAAI,CAAC,OAAO,SAASM,CAAE,KAAK,CAAC,OAAO,SAASC,CAAE,GAAG;AAC9C,QAAEN;AACF;AAAA,IACJ;AACA,IAAAC,KAAQI,GACRH,KAAQI,GACRH,KAASE,IAAKC,GACdF,KAASC,IAAKA;AAAA,EAClB;AACA,QAAME,KAASP,IAAIG,IAAQF,IAAOC,MAASF,IAAII,IAAQH,IAAOA,IACxDO,KAAaN,IAAOK,IAAQN,KAAQD;AAE1C,SAAO,CAACO,GAAOC,CAAS;AAC5B;AAEA,SAASC,GAAwBC,GAAoBH,GAAeC,GAAmB;AACnF,QAAMG,IAAU,CAACnB,MAAcA,IAAIe,IAAQC,GACrCR,IAAIU,EAAK,QACTE,IAAQC,EAAIH,GAAM,OAAKb,EAAE,CAAC,IAAIa,EAAK,QACnCI,IAAI9B,EAASnD,GAAkBmE,IAAI,CAAC,GAEpCe,IAAO,KAAK;AAAA,IACdF,EAAIH,GAAM,CAAAM,OAAQA,EAAI,IAAIL,EAAQK,EAAI,CAAC,MAAM,CAAC,KAAKhB,IAAI,KAAKa,EAAIH,GAAM,QAAQM,EAAI,IAAIJ,MAAU,CAAC;AAAA,EAAA,GAE/FK,IAAaF,IAAOD,GAGpBI,IADOH,IAAO,KAAK,KAAKF,EAAIH,GAAM,CAAAM,MAAOA,EAAI,KAAK,CAAC,IAAIhB,CAAC,IAChCc;AAE9B,SAAO,EAAC,OAAAP,GAAO,WAAAC,GAAW,YAAAS,GAAY,gBAAAC,EAAA;AAC1C;AAMO,SAASC,GACZpB,GACAY,GACgC;AAChC,QAAMS,IAAOP,EAAId,GAAM,OAAKF,EAAE,CAAC,IAAIE,EAAK;AACxC,MAAI,IAAI,GACJJ,IAAI;AACR,WAAS0B,IAAI,GAAGA,IAAItB,EAAK,QAAQ,EAAEsB;AAC/B,SAAK,KAAK,IAAItB,EAAKsB,CAAC,EAAE,IAAID,GAAM,CAAC,GACjCzB,KAAK,KAAK,IAAII,EAAKsB,CAAC,EAAE,IAAIV,EAAQZ,EAAKsB,CAAC,EAAE,CAAC,GAAG,CAAC;AAEnD,QAAMC,IAAK,KAAK,KAAK3B,KAAKI,EAAK,SAAS,EAAE,GACpCe,IAAI9B,EAASnD,GAAkBkE,EAAK,SAAS,CAAC;AACpD,SAAO,SAAUP,GAAW;AACxB,UAAM+B,IAAIZ,EAAQnB,CAAC,GACbgC,IAAKF,IAAK,KAAK,KAAK,IAAIvB,EAAK,SAAS,KAAK,IAAIP,IAAI4B,GAAM,CAAC,IAAI,CAAC,GAC/DK,IAAOF,IAAIT,IAAIU,GACfE,IAAQH,IAAIT,IAAIU;AACtB,WAAO,EAAC,GAAAhC,GAAG,GAAG+B,GAAG,MAAM,MAAME,CAAI,IAAIF,IAAIE,GAAM,OAAO,MAAMC,CAAK,IAAIH,IAAIG,EAAA;AAAA,EAC7E;AACJ;AAEA,SAASC,GAAMjB,GAAoB;AAC/B,QAAMV,IAAIU,EAAK;AACf,SAAO,KAAK;AAAA,KACPV,IAAIa,EAAIH,GAAM,CAACb,MAAkBA,EAAE,IAAIA,EAAE,CAAC,IAAIgB,EAAIH,GAAM,CAACb,MAAkBA,EAAE,CAAC,IAAIgB,EAAIH,GAAM,CAACb,MAAkBA,EAAE,CAAC,MAC9G,KAAK;AAAA,MACFG,IAAIa,EAAIH,GAAM,CAAAb,MAAKA,EAAE,IAAIA,EAAE,CAAC,IACxB,KAAK;AAAA,QACDgB,EAAIH,GAAM,CAAAb,MAAKA,EAAE,CAAC;AAAA,QAClB;AAAA,MAAA;AAAA,IACJ,IAEJ,KAAK;AAAA,MACDG,IAAIa,EAAIH,GAAM,CAAAb,MAAKA,EAAE,IAAIA,EAAE,CAAC,IACxB,KAAK;AAAA,QACDgB,EAAIH,GAAM,CAAAb,MAAKA,EAAE,CAAC;AAAA,QAClB;AAAA,MAAA;AAAA,IACJ;AAAA,IAEhB;AAAA,EAAA;AAER;AAEA,SAAS+B,GAAYC,GAAmBC,GAAmC;AACvE,QAAMC,IAAQX,EAAKS,GAAKC,CAAM;AAE9B,SAAOV,EAAKS,EAAI,IAAI,CAAAG,OAAMF,EAAOE,CAAC,IAAID,MAAU,CAAC,CAAC;AACtD;AAEA,SAASE,EAAqBJ,GAAmBC,GAAmC;AAChF,SAAO,KAAK,KAAKF,GAAYC,GAAKC,CAAM,CAAC;AAC7C;AAEA,SAASI,GAAcL,GAAmBM,GAAoCC,GAAoC;AAC9G,QAAMxB,IAAQQ,EAAKS,GAAKM,CAAO,GACzBE,IAAQjB,EAAKS,GAAKO,CAAO,GACzBpC,IAAI6B,EAAI;AAEd,SAAOhB,EAAIgB,GAAK,CAAAhC,OAAMA,EAAE,IAAIe,MAAUf,EAAE,IAAIwC,EAAM,IAAIrC;AAC1D;AAEA,SAASsC,GAAK5B,GAAoB;AAC9B,QAAM6B,IAAML;AAAA,IACRxB;AAAA,IACA,OAAKb,EAAE;AAAA,IACP,OAAKA,EAAE;AAAA,EAAA,GAEL2C,IAASP,EAAqBvB,GAAM,CAAAb,MAAKA,EAAE,CAAC,GAC5C4C,IAASR,EAAqBvB,GAAM,CAAAb,MAAKA,EAAE,CAAC;AAElD,SAAO0C,KAAOC,IAASC;AAC3B;AAEA,SAASC,GAAUhC,GAAoBC,GAAgC;AACnE,QAAMX,IAAIU,EAAK,QACT3E,IAAI,GACJ4G,IAAM,GACNC,IAAM5C,IAAIjE,IAAI,GACdsG,IAAQjB,EAAKV,GAAM,CAAAb,MAAKA,EAAE,CAAC,GAC3BgD,IAAMhC,EAAIH,GAAM,CAAAb,OAAMc,EAAQd,EAAE,CAAC,IAAIwC,MAAU,CAAC,GAChDS,IAAMjC,EAAIH,GAAM,CAAAb,OAAMc,EAAQd,EAAE,CAAC,IAAIA,EAAE,MAAM,CAAC,GAE9CkD,IAAMF,IAAMF,GACZK,IAAMF,IAAMF,GAEZK,IAAIF,IAAMC;AAIhB,SAAO,IAFME,GAAKD,GAAGN,GAAKC,CAAG;AAGjC;AAoBO,SAASO,GACZC,GACAC,GACAC,GACAC,GACAC,GACAC,GACiB;AACjB,SAAKA,IAGEH,EAAU,OAAO,CAACI,GAAiBC,MAAa;AACnD,UAAMC,IAAYP,EAAaM,CAAQ,EAAE;AAEzC,WAAAD,EAAIC,CAAQ,IAAIJ,EACX,IAAI,CAAAM,MAAU;AACX,YAAMnD,IAAOkD,EAAU;AAAA,QAAO,OAC1BJ,EAAS,MAAM,CAACM,GAAQC,MAAQ,OAAOX,EAAU,eAAeU,EAAO,OAAO9C,EAAI,GAAG,CAAC,MAAM6C,EAAOE,CAAG,CAAC;AAAA,MAAA;AAG3G,UAAIrD,EAAK,WAAW,EAAG,QAAO;AAE9B,YAAM,CAACsD,IAAO,GAAGC,IAAO,CAAC,IAAIC,GAAOxD,GAAM,CAACM,MAAoBA,EAAI,CAAC,GAC9D,CAACT,GAAOC,CAAS,IAAIV,GAAiBY,CAAI,GAC1CC,IAAU,CAACnB,MAAce,IAAQf,IAAIgB,GACrC2D,IAAchD,GAAmBT,GAAMC,CAAO;AAEpD,aAAO;AAAA,QACH,KAAMD,EAAK,CAAC,EAAU;AAAA,QACtB,SAAAC;AAAA,QACA,aAAAwD;AAAA,QACA,SAAS,CAACC,GAAqCC,GAAqCC,MAChFC,GAAA,EACK,EAAE,CAAA1E,MAAKuE,EAAOvE,EAAE,CAAC,CAAC,EAClB,GAAG,CAAAA,MAAKwE,EAAOxE,EAAE,IAAI,CAAC,EACtB,GAAG,OAAKwE,EAAOxE,EAAE,KAAK,CAAC,EAAEyE,EAAM,IAAIH,CAAW,CAAC,KAAK;AAAA,QAC7D,OAAOV,EAAM;AAAA,QACb,SAASA,EAAM;AAAA,QACf,WAAW;AAAA,QACX,SAAS,CAACO,GAAMC,CAAI;AAAA,QACpB,OAAO;AAAA,UACH,IAAItC,GAAMjB,CAAI;AAAA,UACd,GAAG4B,GAAK5B,CAAI;AAAA,UACZ,QAAQgC,GAAUhC,GAAMC,CAAO;AAAA,UAC/B,cAAcF,GAAwBC,GAAMH,GAAOC,CAAS;AAAA,QAAA;AAAA,MAChE;AAAA,IAER,CAAC,EACA,OAAO,CAACgE,MAASA,MAAS,IAAI,GAE5Bd;AAAA,EACX,GAAG,CAAA,CAAE,IA1CM;AA2Cf;"}
+{"version":3,"file":"linearRegression.js","names":[],"sources":["../../src/scatterplot/linearRegression.ts"],"sourcesContent":["/* eslint-disable no-loss-of-precision */\n// @ts-ignore\nimport fCDF from '@stdlib/stats-base-dists-f-cdf';\nimport { extent, mean, sum } from 'd3-array';\nimport type { ScaleLinear } from 'd3-scale';\nimport { area } from 'd3-shape';\nimport type { DataFrame } from '../DataFrame';\nimport type { ColumnName, ContinuousAesFromColumn, InheritAesScatterplot } from '../types';\nimport type { Dot, GroupedDots } from './dots';\nimport type { ScatterplotSettingsImpl } from './ScatterplotSettingsImpl';\n\ntype NumericDot = {x: number, y:number};\nconst CONFIDENCE_LEVEL = 0.05;\n\n/*\n Returns the normal deviate Z corresponding to a given lower tail area of P; Z is accurate to about 1 part in 10**16.\n\n Wichura, M. J. (1988) Algorithm AS 241: The percentage points of the normal distribution. Applied Statistics, 37, 477–484.\n\n extension of:\n Beasley, J. D./ Springer, S. G. (1977), The percentage points of the NormalDistribution, Applied Statistics. 26, 118–121.\n\n Excel: STANDNORMINV(); R: qnorm()\n */\n\nfunction normdev(p: number) {\n    if (p < 0 || p > 1) throw Error('P p < 0 || p > 1 in linear regression calculation');\n    if (p == 0) return -Infinity;\n    if (p == 1) return Infinity;\n\n    const a0 = 3.387132872796366608,\n        a1 = 1.3314166789178437745e2,\n        a2 = 1.9715909503065514427e3,\n        a3 = 1.3731693765509461125e4,\n        a4 = 4.5921953931549871457e4,\n        a5 = 6.7265770927008700853e4,\n        a6 = 3.3430575583588128105e4,\n        a7 = 2.5090809287301226727e3,\n        b1 = 4.2313330701600911252e1,\n        b2 = 6.871870074920579083e2,\n        b3 = 5.3941960214247511077e3,\n        b4 = 2.1213794301586595867e4,\n        b5 = 3.930789580009271061e4,\n        b6 = 2.8729085735721942674e4,\n        b7 = 5.226495278852854561e3,\n        c0 = 1.42343711074968357734,\n        c1 = 4.6303378461565452959,\n        c2 = 5.7694972214606914055,\n        c3 = 3.64784832476320460504,\n        c4 = 1.27045825245236838258,\n        c5 = 2.4178072517745061177e-1,\n        c6 = 2.27238449892691845833e-2,\n        c7 = 7.7454501427834140764e-4,\n        d1 = 2.05319162663775882187,\n        d2 = 1.6763848301838038494,\n        d3 = 6.8976733498510000455e-1,\n        d4 = 1.4810397642748007459e-1,\n        d5 = 1.51986665636164571966e-2,\n        d6 = 5.475938084995344946e-4,\n        d7 = 1.05075007164441684324e-9,\n        e0 = 6.6579046435011037772,\n        e1 = 5.4637849111641143699,\n        e2 = 1.7848265399172913358,\n        e3 = 2.9656057182850489123e-1,\n        e4 = 2.6532189526576123093e-2,\n        e5 = 1.2426609473880784386e-3,\n        e6 = 2.71155556874348757815e-5,\n        e7 = 2.01033439929228813265e-7,\n        f1 = 5.9983220655588793769e-1,\n        f2 = 1.3692988092273580531e-1,\n        f3 = 1.48753612908506148525e-2,\n        f4 = 7.868691311456132591e-4,\n        f5 = 1.8463183175100546818e-5,\n        f6 = 1.4215117583164458887e-7,\n        f7 = 2.04426310338993978564e-15;\n\n    const q = p - 0.5;\n    let r, z;\n\n    // p close to 0.5\n    if (Math.abs(q) <= 0.425) {\n        r = 0.180625 - q * q;\n        z =\n            (q * (((((((a7 * r + a6) * r + a5) * r + a4) * r + a3) * r + a2) * r + a1) * r + a0)) /\n            (((((((b7 * r + b6) * r + b5) * r + b4) * r + b3) * r + b2) * r + b1) * r + 1);\n        return z;\n    }\n\n    if (q > 0) r = 1 - p;\n    else r = p;\n    r = Math.sqrt(-Math.log(r));\n\n    // p neither close to 0.5 nor 0 or 1\n    if (r <= 5) {\n        r += -1.6;\n        z =\n            (((((((c7 * r + c6) * r + c5) * r + c4) * r + c3) * r + c2) * r + c1) * r + c0) /\n            (((((((d7 * r + d6) * r + d5) * r + d4) * r + d3) * r + d2) * r + d1) * r + 1);\n    }\n    // p near 0 or 1\n    else {\n        r += -5;\n        z =\n            (((((((e7 * r + e6) * r + e5) * r + e4) * r + e3) * r + e2) * r + e1) * r + e0) /\n            (((((((f7 * r + f6) * r + f5) * r + f4) * r + f3) * r + f2) * r + f1) * r + 1);\n    }\n\n    if (q < 0.0) z = -z;\n    return z;\n}\n\n/*\nHill's approximated inverse t-distribution calculates t given df and two-tail probability:\nHill, G. W. (1970), Algorithm 396: Student's t-quantiles. Communications of the ACM, 13(10), 619–620.\n\nResult should be \"correct to at least 6 significant digits even for the analytic continuation through noninteger values of n > 5\". For higher precision (used in R, ...) see:\nHill, G. W. (1981) Remark on Algorithm 396, ACM Transactions on Mathematical Software, 7, 250–1.\n\nExcel: TINV(); R: qt()\n*/\nfunction inverseT(p: number, df: number) {\n    const {sin, cos, sqrt, pow, exp, PI} = Math;\n    let x: number, y: number;\n\n    if (df == 1) return cos((p * PI) / 2) / sin((p * PI) / 2);\n    if (df == 2) return sqrt(2 / (p * (2 - p)) - 2);\n\n    const a = 1 / (df - 0.5);\n    const b = 48 / (a * a);\n    let c = (((20700 * a) / b - 98) * a - 16) * a + 96.36;\n    const d = ((94.5 / (b + c) - 3) / b + 1) * sqrt(a * PI * 0.5) * df;\n    x = d * p;\n    y = pow(x, 2 / df);\n\n    if (y > 0.05 + a) {\n        // The procedure normdev(p) is assumed to return a negative normal\n        // deviate at the lower tail probability level p, e.g. -2.32 for p = 0.01.\n        x = normdev(p / 2);\n        y = x * x;\n        if (df < 5) c = c + 0.3 * (df - 4.5) * (x + 0.6);\n        c = (((0.05 * d * x - 5) * x - 7) * x - 2) * x + b + c;\n        y = (((((0.4 * y + 6.3) * y + 36) * y + 94.5) / c - y - 3) / b + 1) * x;\n        y = a * y * y;\n        if (y > 0.002) y = exp(y) - 1;\n        else y = 0.5 * y * y + y;\n    } else {\n        y =\n            (((1 / (((df + 6) / (df * y) - 0.089 * d - 0.822) * (df + 2) * 3) + 0.5 / (df + 4)) * y - 1) * (df + 1)) /\n                (df + 2) +\n            1 / y;\n    }\n\n    return sqrt(df * y);\n}\n\nexport function linearRegression(data: NumericDot[]): [number, number] {\n    let n = data.length,\n        sumX = 0,\n        sumY = 0,\n        sumXY = 0,\n        sumX2 = 0;\n    for (let i = 0; i < data.length; ++i) {\n        const xi = data[i].x;\n        const yi = data[i].y;\n        if (!Number.isFinite(xi) || !Number.isFinite(yi)) {\n            --n;\n            continue;\n        }\n        sumX += xi;\n        sumY += yi;\n        sumXY += xi * yi;\n        sumX2 += xi * xi;\n    }\n    const slope = (n * sumXY - sumX * sumY) / (n * sumX2 - sumX * sumX);\n    const intercept = (sumY - slope * sumX) / n;\n\n    return [slope, intercept];\n}\n\nfunction getCoefficientsInterval(dots: NumericDot[], slope: number, intercept: number) {\n    const predict = (x: number) => x * slope + intercept;\n    const n = dots.length;\n    const meanX = sum(dots, d => d.x) / dots.length;\n    const t = inverseT(CONFIDENCE_LEVEL, n - 2);\n\n    const se_b = Math.sqrt(\n        sum(dots, dot => (dot.y - predict(dot.x)) ** 2) / (n - 2) / sum(dots, dot => (dot.x - meanX) ** 2)\n    );\n    const slopeError = se_b * t;\n\n    const se_a = se_b * Math.sqrt(sum(dots, dot => dot.x ** 2) / n);\n    const interceptError = se_a * t;\n\n    return {slope, intercept, slopeError, interceptError};\n}\n\nexport type PointIntervalInfo = {\n    x: number; y: number, left: number; right: number\n}\n// https://stats.stackexchange.com/questions/101318/understanding-shape-and-calculation-of-confidence-bands-in-linear-regression\nexport function confidenceInterval(\n    data: NumericDot[],\n    predict: (x: number) => number\n): (x: number) => PointIntervalInfo {\n    const mean = sum(data, d => d.x) / data.length;\n    let a = 0,\n        b = 0;\n    for (let i = 0; i < data.length; ++i) {\n        a += Math.pow(data[i].x - mean, 2);\n        b += Math.pow(data[i].y - predict(data[i].x), 2);\n    }\n    const sy = Math.sqrt(b / (data.length - 2));\n    const t = inverseT(CONFIDENCE_LEVEL, data.length - 2);\n    return function (x: number) {\n        const Y = predict(x);\n        const se = sy * Math.sqrt(1 / data.length + Math.pow(x - mean, 2) / a);\n        const left = Y - t * se;\n        const right = Y + t * se;\n        return {x, y: Y, left: isNaN(left) ? Y : left, right: isNaN(right) ? Y : right};\n    };\n}\n\nfunction getR2(dots: NumericDot[]) {\n    const n = dots.length;\n    return Math.pow(\n        (n * sum(dots, (d: NumericDot) => d.x * d.y) - sum(dots, (d: NumericDot) => d.x) * sum(dots, (d: NumericDot) => d.y)) /\n            (Math.sqrt(\n                n * sum(dots, d => d.x * d.x) -\n                    Math.pow(\n                        sum(dots, d => d.x),\n                        2\n                    )\n            ) *\n                Math.sqrt(\n                    n * sum(dots, d => d.y * d.y) -\n                        Math.pow(\n                            sum(dots, d => d.y),\n                            2\n                        )\n                )),\n        2\n    );\n}\n\nfunction getVariance(arr: NumericDot[], getter: (d: NumericDot) => number) {\n    const vMean = mean(arr, getter) as number;\n\n    return mean(arr.map(v => (getter(v) - vMean) ** 2)) as number;\n}\n\nfunction getStandardDeviation(arr: NumericDot[], getter: (d: NumericDot) => number) {\n    return Math.sqrt(getVariance(arr, getter));\n}\n\nfunction getCovariance(arr: NumericDot[], getterX: (d: NumericDot) => number, getterY: (d: NumericDot) => number) {\n    const meanX = mean(arr, getterX) as number;\n    const meanY = mean(arr, getterY) as number;\n    const n = arr.length;\n\n    return sum(arr, d => (d.x - meanX) * (d.y - meanY)) / n;\n}\n\nfunction getR(dots: NumericDot[]) {\n    const cov = getCovariance(\n        dots,\n        d => d.x,\n        d => d.y\n    );\n    const sigmaX = getStandardDeviation(dots, d => d.x);\n    const sigmaY = getStandardDeviation(dots, d => d.y);\n\n    return cov / (sigmaX * sigmaY);\n}\n\nfunction getPValue(dots: NumericDot[], predict: (x: number) => number) {\n    const n = dots.length;\n    const p = 1; // number of coefficients\n    const dfm = 1; // degrees of freedom model\n    const dfe = n - p - 1; // degrees of freedom error\n    const meanY = mean(dots, d => d.y) as number;\n    const ssr = sum(dots, d => (predict(d.x) - meanY) ** 2); // sum of squares regression\n    const sse = sum(dots, d => (predict(d.x) - d.y) ** 2); // sum of squares errors\n\n    const msr = ssr / dfm;\n    const mse = sse / dfe;\n\n    const f = msr / mse;\n\n    const pval = fCDF(f, dfm, dfe);\n\n    return 1.0 - pval; // \"greater\" p-value\n}\n\ntype TrendInfo = {\n    idx: number;\n    predict: (x: number) => number;\n    getInterval: (x: number) => PointIntervalInfo;\n    getArea: (x: ScaleLinear<number, number>, y: ScaleLinear<number, number>, range: number[]) => string;\n    color: string | InheritAesScatterplot | ContinuousAesFromColumn<string>;\n    bounded: boolean;\n    showStats: boolean;\n    xBounds: [number, number];\n    stats: {\n        r2?: number;\n        r?: number;\n        pValue?: number;\n        coefficients?: {slope: number, intercept: number, slopeError:number, interceptError: number};\n    };\n};\nexport type TrendsData = Record<string, TrendInfo[]>;\n\nexport function getRegressionData(\n    dataFrame: DataFrame,\n    dotsByFacets: GroupedDots,\n    facetKeys: string[],\n    groupingKeysValues: string[][],\n    grouping: ColumnName[],\n    trend: ScatterplotSettingsImpl['trend']\n): TrendsData | null {\n    if (!trend) {\n        return null;\n    }\n    return facetKeys.reduce((res: TrendsData, facetKey) => {\n        const facetDots = dotsByFacets[facetKey].dots;\n        \n        res[facetKey] = groupingKeysValues\n            .map(values => {\n                const dots = facetDots.filter(dot =>\n                    grouping.every((column, idx) => String(dataFrame.getColumnValue(column.value, dot.idx)) === values[idx])\n                ) as NumericDot[];\n\n                if (dots.length === 0) return null;\n\n                const [minX = 0, maxX = 0] = extent(dots, (dot: NumericDot) => dot.x);\n                const [slope, intercept] = linearRegression(dots);\n                const predict = (x: number) => slope * x + intercept;\n                const getInterval = confidenceInterval(dots, predict);\n\n                return {\n                    idx: (dots[0] as Dot).idx,\n                    predict,\n                    getInterval,\n                    getArea: (scaleX: ScaleLinear<number, number>, scaleY: ScaleLinear<number, number>, range: number[]) =>\n                        area<PointIntervalInfo>()\n                            .x(d => scaleX(d.x))\n                            .y0(d => scaleY(d.left))\n                            .y1(d => scaleY(d.right))(range.map(getInterval)) ?? '',\n                    color: trend.color,\n                    bounded: trend.bounded,\n                    showStats: true,\n                    xBounds: [minX, maxX],\n                    stats: {\n                        r2: getR2(dots),\n                        r: getR(dots),\n                        pValue: getPValue(dots, predict),\n                        coefficients: getCoefficientsInterval(dots, slope, intercept),\n                    },\n                };\n            })\n            .filter((item) => item !== null) as TrendInfo[];\n        \n        return res;\n    }, {});\n}\n"],"mappings":";;;AAYA,IAAM,IAAmB;AAazB,SAAS,EAAQ,GAAW;AACxB,KAAI,IAAI,KAAK,IAAI,EAAG,OAAM,MAAM,oDAAoD;AACpF,KAAI,KAAK,EAAG,QAAO;AACnB,KAAI,KAAK,EAAG,QAAO;CAEnB,IA8CM,IAAI,IAAI,IACV,GAAG;AA+BP,QA5BI,KAAK,IAAI,EAAE,IAAI,QACf,IAAI,UAAW,IAAI,GACnB,IACK,WAAW,qBAAK,IAAI,qBAAM,IAAI,oBAAM,IAAI,qBAAM,IAAI,qBAAM,IAAI,sBAAM,IAAI,sBAAM,IAAI,6BAC1E,oBAAK,IAAI,sBAAM,IAAI,qBAAM,IAAI,sBAAM,IAAI,qBAAM,IAAI,qBAAM,IAAI,qBAAM,IAAI,IACzE,MAGX,AACK,IADD,IAAI,IAAO,IAAI,IACV,GACT,IAAI,KAAK,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,EAGvB,KAAK,KACL,KAAK,MACL,WACW,uBAAK,IAAI,uBAAM,IAAI,qBAAM,IAAI,sBAAM,IAAI,sBAAM,IAAI,qBAAM,IAAI,qBAAM,IAAI,6BACrE,wBAAK,IAAI,wBAAM,IAAI,uBAAM,IAAI,sBAAM,IAAI,kBAAM,IAAI,sBAAM,IAAI,qBAAM,IAAI,OAIhF,KAAK,IACL,WACW,wBAAK,IAAI,yBAAM,IAAI,wBAAM,IAAI,uBAAM,IAAI,sBAAM,IAAI,sBAAM,IAAI,qBAAM,IAAI,4BACrE,wBAAK,IAAI,wBAAM,IAAI,yBAAM,IAAI,wBAAM,IAAI,uBAAM,IAAI,qBAAM,IAAI,oBAAM,IAAI,KAGhF,IAAI,MAAK,IAAI,CAAC,IACX;;AAYX,SAAS,EAAS,GAAW,GAAY;CACrC,IAAM,EAAC,QAAK,QAAK,SAAM,QAAK,QAAK,UAAM,MACnC,GAAW;AAEf,KAAI,KAAM,EAAG,QAAO,EAAK,IAAI,IAAM,EAAE,GAAG,EAAK,IAAI,IAAM,EAAE;AACzD,KAAI,KAAM,EAAG,QAAO,EAAK,KAAK,KAAK,IAAI,MAAM,EAAE;CAE/C,IAAM,IAAI,KAAK,IAAK,KACd,IAAI,MAAM,IAAI,IAChB,MAAO,QAAQ,IAAK,IAAI,MAAM,IAAI,MAAM,IAAI,OAC1C,MAAM,QAAQ,IAAI,KAAK,KAAK,IAAI,KAAK,EAAK,IAAI,IAAK,GAAI,GAAG;AAsBhE,QArBA,IAAI,IAAI,GACR,IAAI,EAAI,GAAG,IAAI,EAAG,EAEd,IAAI,MAAO,KAGX,IAAI,EAAQ,IAAI,EAAE,EAClB,IAAI,IAAI,GACJ,IAAK,MAAG,KAAQ,MAAO,IAAK,QAAQ,IAAI,MAC5C,OAAO,MAAO,IAAI,IAAI,KAAK,IAAI,KAAK,IAAI,KAAK,IAAI,IAAI,GACrD,SAAS,KAAM,IAAI,OAAO,IAAI,MAAM,IAAI,QAAQ,IAAI,IAAI,KAAK,IAAI,KAAK,GACtE,IAAI,IAAI,IAAI,GACZ,AACK,IADD,IAAI,OAAW,EAAI,EAAE,GAAG,IACnB,KAAM,IAAI,IAAI,KAEvB,MACO,OAAO,IAAK,MAAM,IAAK,KAAK,OAAQ,IAAI,SAAU,IAAK,KAAK,KAAK,MAAO,IAAK,MAAM,IAAI,MAAM,IAAK,MAChG,IAAK,KACV,IAAI,GAGL,EAAK,IAAK,EAAE;;AAGvB,SAAgB,EAAiB,GAAsC;CACnE,IAAI,IAAI,EAAK,QACT,IAAO,GACP,IAAO,GACP,IAAQ,GACR,IAAQ;AACZ,MAAK,IAAI,IAAI,GAAG,IAAI,EAAK,QAAQ,EAAE,GAAG;EAClC,IAAM,IAAK,EAAK,GAAG,GACb,IAAK,EAAK,GAAG;AACnB,MAAI,CAAC,OAAO,SAAS,EAAG,IAAI,CAAC,OAAO,SAAS,EAAG,EAAE;AAC9C,KAAE;AACF;;AAKJ,EAHA,KAAQ,GACR,KAAQ,GACR,KAAS,IAAK,GACd,KAAS,IAAK;;CAElB,IAAM,KAAS,IAAI,IAAQ,IAAO,MAAS,IAAI,IAAQ,IAAO;AAG9D,QAAO,CAAC,IAFW,IAAO,IAAQ,KAAQ,EAEjB;;AAG7B,SAAS,EAAwB,GAAoB,GAAe,GAAmB;CACnF,IAAM,KAAW,MAAc,IAAI,IAAQ,GACrC,IAAI,EAAK,QACT,IAAQ,EAAI,IAAM,MAAK,EAAE,EAAE,GAAG,EAAK,QACnC,IAAI,EAAS,GAAkB,IAAI,EAAE,EAErC,IAAO,KAAK,KACd,EAAI,IAAM,OAAQ,EAAI,IAAI,EAAQ,EAAI,EAAE,KAAK,EAAE,IAAI,IAAI,KAAK,EAAI,IAAM,OAAQ,EAAI,IAAI,MAAU,EAAE,CACrG;AAMD,QAAO;EAAC;EAAO;EAAW,YALP,IAAO;EAKY,gBAHzB,IAAO,KAAK,KAAK,EAAI,IAAM,MAAO,EAAI,KAAK,EAAE,GAAG,EAAE,GACjC;EAEuB;;AAOzD,SAAgB,EACZ,GACA,GACgC;CAChC,IAAM,IAAO,EAAI,IAAM,MAAK,EAAE,EAAE,GAAG,EAAK,QACpC,IAAI,GACJ,IAAI;AACR,MAAK,IAAI,IAAI,GAAG,IAAI,EAAK,QAAQ,EAAE,EAE/B,CADA,MAAc,EAAK,GAAG,IAAI,MAAM,GAChC,MAAc,EAAK,GAAG,IAAI,EAAQ,EAAK,GAAG,EAAE,KAAE;CAElD,IAAM,IAAK,KAAK,KAAK,KAAK,EAAK,SAAS,GAAG,EACrC,IAAI,EAAS,GAAkB,EAAK,SAAS,EAAE;AACrD,QAAO,SAAU,GAAW;EACxB,IAAM,IAAI,EAAQ,EAAE,EACd,IAAK,IAAK,KAAK,KAAK,IAAI,EAAK,UAAkB,IAAI,MAAM,IAAK,EAAE,EAChE,IAAO,IAAI,IAAI,GACf,IAAQ,IAAI,IAAI;AACtB,SAAO;GAAC;GAAG,GAAG;GAAG,MAAM,MAAM,EAAK,GAAG,IAAI;GAAM,OAAO,MAAM,EAAM,GAAG,IAAI;GAAM;;;AAIvF,SAAS,EAAM,GAAoB;CAC/B,IAAM,IAAI,EAAK;AACf,UACK,IAAI,EAAI,IAAO,MAAkB,EAAE,IAAI,EAAE,EAAE,GAAG,EAAI,IAAO,MAAkB,EAAE,EAAE,GAAG,EAAI,IAAO,MAAkB,EAAE,EAAE,KAC/G,KAAK,KACF,IAAI,EAAI,IAAM,MAAK,EAAE,IAAI,EAAE,EAAE,GAErB,EAAI,IAAM,MAAK,EAAE,EAAE,IACnB,EAEX,GACG,KAAK,KACD,IAAI,EAAI,IAAM,MAAK,EAAE,IAAI,EAAE,EAAE,GAErB,EAAI,IAAM,MAAK,EAAE,EAAE,IACnB,EAEX,MACT;;AAIR,SAAS,EAAY,GAAmB,GAAmC;CACvE,IAAM,IAAQ,EAAK,GAAK,EAAO;AAE/B,QAAO,EAAK,EAAI,KAAI,OAAM,EAAO,EAAE,GAAG,MAAU,EAAE,CAAC;;AAGvD,SAAS,EAAqB,GAAmB,GAAmC;AAChF,QAAO,KAAK,KAAK,EAAY,GAAK,EAAO,CAAC;;AAG9C,SAAS,EAAc,GAAmB,GAAoC,GAAoC;CAC9G,IAAM,IAAQ,EAAK,GAAK,EAAQ,EAC1B,IAAQ,EAAK,GAAK,EAAQ,EAC1B,IAAI,EAAI;AAEd,QAAO,EAAI,IAAK,OAAM,EAAE,IAAI,MAAU,EAAE,IAAI,GAAO,GAAG;;AAG1D,SAAS,EAAK,GAAoB;AAS9B,QARY,EACR,IACA,MAAK,EAAE,IACP,MAAK,EAAE,EACV,IACc,EAAqB,IAAM,MAAK,EAAE,EAAE,GACpC,EAAqB,IAAM,MAAK,EAAE,EAAE;;AAKvD,SAAS,EAAU,GAAoB,GAAgC;CACnE,IAGM,IAHI,EAAK,SAGC,IAAI,GACd,IAAQ,EAAK,IAAM,MAAK,EAAE,EAAE,EAC5B,IAAM,EAAI,IAAM,OAAM,EAAQ,EAAE,EAAE,GAAG,MAAU,EAAE,EACjD,IAAM,EAAI,IAAM,OAAM,EAAQ,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE;AASrD,QAAO,IAFM,EALD,IAAM,KACN,IAAM,IAIG,GAAK,EAAI;;AAuBlC,SAAgB,EACZ,GACA,GACA,GACA,GACA,GACA,GACiB;AAIjB,QAHK,IAGE,EAAU,QAAQ,GAAiB,MAAa;EACnD,IAAM,IAAY,EAAa,GAAU;AAsCzC,SApCA,EAAI,KAAY,EACX,KAAI,MAAU;GACX,IAAM,IAAO,EAAU,QAAO,MAC1B,EAAS,OAAO,GAAQ,MAAQ,OAAO,EAAU,eAAe,EAAO,OAAO,EAAI,IAAI,CAAC,KAAK,EAAO,GAAK,CAC3G;AAED,OAAI,EAAK,WAAW,EAAG,QAAO;GAE9B,IAAM,CAAC,IAAO,GAAG,IAAO,KAAK,EAAO,IAAO,MAAoB,EAAI,EAAE,EAC/D,CAAC,GAAO,KAAa,EAAiB,EAAK,EAC3C,KAAW,MAAc,IAAQ,IAAI,GACrC,IAAc,EAAmB,GAAM,EAAQ;AAErD,UAAO;IACH,KAAM,EAAK,GAAW;IACtB;IACA;IACA,UAAU,GAAqC,GAAqC,MAChF,GAAyB,CACpB,GAAE,MAAK,EAAO,EAAE,EAAE,CAAC,CACnB,IAAG,MAAK,EAAO,EAAE,KAAK,CAAC,CACvB,IAAG,MAAK,EAAO,EAAE,MAAM,CAAC,CAAC,EAAM,IAAI,EAAY,CAAC,IAAI;IAC7D,OAAO,EAAM;IACb,SAAS,EAAM;IACf,WAAW;IACX,SAAS,CAAC,GAAM,EAAK;IACrB,OAAO;KACH,IAAI,EAAM,EAAK;KACf,GAAG,EAAK,EAAK;KACb,QAAQ,EAAU,GAAM,EAAQ;KAChC,cAAc,EAAwB,GAAM,GAAO,EAAU;KAChE;IACJ;IACH,CACD,QAAQ,MAAS,MAAS,KAAK,EAE7B;IACR,EAAE,CAAC,GA1CK"}

package/dist/scatterplot/utils/createAesGetter.js

@@ -1,41 +1,34 @@
-import { isContinuousAes as m } from "../../types/common.js";
-import "../../types/discrete.js";
-import "../../types/scatterplot.js";
-import "../../types/heatmap.js";
-import "../../types/dendro.js";
-import "../../types/histogram.js";
-import "../../types/bubble.js";
-import { getContinuousColorScale as a } from "../../utils/getContinuousColorScale.js";
-import { DEFAULT_DOT_AES as g, DEFAULT_COMMON_AES as s } from "../constants.js";
-import { sqrt as f } from "../../node_modules/.pnpm/d3-scale@4.0.2/node_modules/d3-scale/src/pow.js";
-function C(o) {
-  return typeof o == "object" && o !== null && "type" in o && "value" in o && o.type === "grouping";
+import { DEFAULT_COMMON_AES as e, DEFAULT_DOT_AES as t } from "../constants.js";
+import { isContinuousAes as n } from "../../types/common.js";
+import "../../types/index.js";
+import { getContinuousColorScale as r } from "../../utils/getContinuousColorScale.js";
+import { scaleSqrt as i } from "d3-scale";
+function a(e) {
+	return typeof e == "object" && !!e && "type" in e && "value" in e && e.type === "grouping";
 }
-function T(o, p, n, t) {
-  if (m(n) && t === "dotSize") {
-    const { domain: r, range: u } = n, i = f().domain(r).range(u);
-    return function(e) {
-      const c = i(Number(o.getColumnValue(n.columnName.value, e)));
-      return Math.max(1, c);
-    };
-  }
-  if (m(n) && (t === "dotFill" || t === "lineColor")) {
-    const { domain: r, range: u, type: i = "linear" } = n, e = a(u, r, i);
-    return function(c) {
-      const l = o.getColumnValue(n.columnName.value, c);
-      return l === null ? g.color : e(Number(l));
-    };
-  }
-  if (C(n)) {
-    const r = n.value, u = p[r].aesMap;
-    return function(i) {
-      const e = o.getColumnValueCategory(r, i);
-      return u(e, t) ?? s[t];
-    };
-  }
-  return () => n;
+function o(o, s, c, l) {
+	if (n(c) && l === "dotSize") {
+		let { domain: e, range: t } = c, n = i().domain(e).range(t);
+		return function(e) {
+			let t = n(Number(o.getColumnValue(c.columnName.value, e)));
+			return Math.max(1, t);
+		};
+	}
+	if (n(c) && (l === "dotFill" || l === "lineColor")) {
+		let { domain: e, range: n, type: i = "linear" } = c, a = r(n, e, i);
+		return function(e) {
+			let n = o.getColumnValue(c.columnName.value, e);
+			return n === null ? t.color : a(Number(n));
+		};
+	}
+	if (a(c)) {
+		let t = c.value, n = s[t].aesMap;
+		return function(r) {
+			return n(o.getColumnValueCategory(t, r), l) ?? e[l];
+		};
+	}
+	return () => c;
 }
-export {
-  T as createAesGetter
-};
-//# sourceMappingURL=createAesGetter.js.map
+export { o as createAesGetter };
+
+//# sourceMappingURL=createAesGetter.js.map

package/dist/scatterplot/utils/createAesGetter.js.map

@@ -1 +1 @@
-{"version":3,"file":"createAesGetter.js","sources":["../../../src/scatterplot/utils/createAesGetter.ts"],"sourcesContent":["import type { ScalePower } from 'd3-scale';\nimport { scaleSqrt } from 'd3-scale';\nimport type { DataFrame } from '../../DataFrame';\nimport type {\n    AesItem,\n    ContinuousAesFromColumn,\n    InheritAesScatterplot,\n    LineShape,\n    PointShape,\n    ScatterplotLegendInfo\n} from '../../types';\nimport { isContinuousAes } from '../../types';\nimport { getContinuousColorScale } from '../../utils/getContinuousColorScale';\nimport { DEFAULT_COMMON_AES, DEFAULT_DOT_AES } from '../constants';\n\nfunction isInheritedAes(v: InheritAesScatterplot | unknown): v is InheritAesScatterplot {\n    return typeof v === 'object' && v !== null && 'type' in v && 'value' in v && v.type === 'grouping';\n}\nexport function createAesGetter<ValueType extends string | number | PointShape | LineShape>(\n    dataFrame: DataFrame,\n    legendInfo: ScatterplotLegendInfo,\n    aesItem: ValueType | ContinuousAesFromColumn<ValueType> | InheritAesScatterplot,\n    field: keyof AesItem\n) {\n    if (isContinuousAes<ValueType>(aesItem) && field === 'dotSize') {\n        const {domain, range} = aesItem;\n        const scale: ScalePower<number, number> = scaleSqrt<number, number>()\n            .domain(domain)\n            .range(range as number[]);\n        return function(rowIndex: number) {\n            const v = scale(Number(dataFrame.getColumnValue(aesItem.columnName.value, rowIndex)));\n            return Math.max(1, v as number) as ValueType;\n        };\n    }\n    if (isContinuousAes<ValueType>(aesItem) && (field === 'dotFill' || field === 'lineColor')) {\n        const {domain, range, type = 'linear'} = aesItem;\n        const scale = getContinuousColorScale(range as string[], domain, type);\n        return function(rowIndex: number) {\n            const v = dataFrame.getColumnValue(aesItem.columnName.value, rowIndex);\n            if (v === null) {\n                return DEFAULT_DOT_AES.color as ValueType;\n            }\n            return scale(Number(v)) as ValueType;\n        };\n    }\n    if (isInheritedAes(aesItem)) {\n        const columnId = aesItem.value;\n        const mapping = legendInfo[columnId].aesMap;\n        return function(rowIndex: number) {\n            const category = dataFrame.getColumnValueCategory(columnId, rowIndex);\n            return (mapping(category, field) ?? DEFAULT_COMMON_AES[field]) as ValueType;\n        };\n    }\n    return () => aesItem as ValueType;\n}\n"],"names":["isInheritedAes","v","createAesGetter","dataFrame","legendInfo","aesItem","field","isContinuousAes","domain","range","scale","scaleSqrt","rowIndex","type","getContinuousColorScale","DEFAULT_DOT_AES","columnId","mapping","category","DEFAULT_COMMON_AES"],"mappings":";;;;;;;;;;AAeA,SAASA,EAAeC,GAAgE;AACpF,SAAO,OAAOA,KAAM,YAAYA,MAAM,QAAQ,UAAUA,KAAK,WAAWA,KAAKA,EAAE,SAAS;AAC5F;AACO,SAASC,EACZC,GACAC,GACAC,GACAC,GACF;AACE,MAAIC,EAA2BF,CAAO,KAAKC,MAAU,WAAW;AAC5D,UAAM,EAAC,QAAAE,GAAQ,OAAAC,EAAA,IAASJ,GAClBK,IAAoCC,IACrC,OAAOH,CAAM,EACb,MAAMC,CAAiB;AAC5B,WAAO,SAASG,GAAkB;AAC9B,YAAMX,IAAIS,EAAM,OAAOP,EAAU,eAAeE,EAAQ,WAAW,OAAOO,CAAQ,CAAC,CAAC;AACpF,aAAO,KAAK,IAAI,GAAGX,CAAW;AAAA,IAClC;AAAA,EACJ;AACA,MAAIM,EAA2BF,CAAO,MAAMC,MAAU,aAAaA,MAAU,cAAc;AACvF,UAAM,EAAC,QAAAE,GAAQ,OAAAC,GAAO,MAAAI,IAAO,aAAYR,GACnCK,IAAQI,EAAwBL,GAAmBD,GAAQK,CAAI;AACrE,WAAO,SAASD,GAAkB;AAC9B,YAAMX,IAAIE,EAAU,eAAeE,EAAQ,WAAW,OAAOO,CAAQ;AACrE,aAAIX,MAAM,OACCc,EAAgB,QAEpBL,EAAM,OAAOT,CAAC,CAAC;AAAA,IAC1B;AAAA,EACJ;AACA,MAAID,EAAeK,CAAO,GAAG;AACzB,UAAMW,IAAWX,EAAQ,OACnBY,IAAUb,EAAWY,CAAQ,EAAE;AACrC,WAAO,SAASJ,GAAkB;AAC9B,YAAMM,IAAWf,EAAU,uBAAuBa,GAAUJ,CAAQ;AACpE,aAAQK,EAAQC,GAAUZ,CAAK,KAAKa,EAAmBb,CAAK;AAAA,IAChE;AAAA,EACJ;AACA,SAAO,MAAMD;AACjB;"}
+{"version":3,"file":"createAesGetter.js","names":[],"sources":["../../../src/scatterplot/utils/createAesGetter.ts"],"sourcesContent":["import type { ScalePower } from 'd3-scale';\nimport { scaleSqrt } from 'd3-scale';\nimport type { DataFrame } from '../../DataFrame';\nimport type {\n    AesItem,\n    ContinuousAesFromColumn,\n    InheritAesScatterplot,\n    LineShape,\n    PointShape,\n    ScatterplotLegendInfo\n} from '../../types';\nimport { isContinuousAes } from '../../types';\nimport { getContinuousColorScale } from '../../utils/getContinuousColorScale';\nimport { DEFAULT_COMMON_AES, DEFAULT_DOT_AES } from '../constants';\n\nfunction isInheritedAes(v: InheritAesScatterplot | unknown): v is InheritAesScatterplot {\n    return typeof v === 'object' && v !== null && 'type' in v && 'value' in v && v.type === 'grouping';\n}\nexport function createAesGetter<ValueType extends string | number | PointShape | LineShape>(\n    dataFrame: DataFrame,\n    legendInfo: ScatterplotLegendInfo,\n    aesItem: ValueType | ContinuousAesFromColumn<ValueType> | InheritAesScatterplot,\n    field: keyof AesItem\n) {\n    if (isContinuousAes<ValueType>(aesItem) && field === 'dotSize') {\n        const {domain, range} = aesItem;\n        const scale: ScalePower<number, number> = scaleSqrt<number, number>()\n            .domain(domain)\n            .range(range as number[]);\n        return function(rowIndex: number) {\n            const v = scale(Number(dataFrame.getColumnValue(aesItem.columnName.value, rowIndex)));\n            return Math.max(1, v as number) as ValueType;\n        };\n    }\n    if (isContinuousAes<ValueType>(aesItem) && (field === 'dotFill' || field === 'lineColor')) {\n        const {domain, range, type = 'linear'} = aesItem;\n        const scale = getContinuousColorScale(range as string[], domain, type);\n        return function(rowIndex: number) {\n            const v = dataFrame.getColumnValue(aesItem.columnName.value, rowIndex);\n            if (v === null) {\n                return DEFAULT_DOT_AES.color as ValueType;\n            }\n            return scale(Number(v)) as ValueType;\n        };\n    }\n    if (isInheritedAes(aesItem)) {\n        const columnId = aesItem.value;\n        const mapping = legendInfo[columnId].aesMap;\n        return function(rowIndex: number) {\n            const category = dataFrame.getColumnValueCategory(columnId, rowIndex);\n            return (mapping(category, field) ?? DEFAULT_COMMON_AES[field]) as ValueType;\n        };\n    }\n    return () => aesItem as ValueType;\n}\n"],"mappings":";;;;;AAeA,SAAS,EAAe,GAAgE;AACpF,QAAO,OAAO,KAAM,cAAY,KAAc,UAAU,KAAK,WAAW,KAAK,EAAE,SAAS;;AAE5F,SAAgB,EACZ,GACA,GACA,GACA,GACF;AACE,KAAI,EAA2B,EAAQ,IAAI,MAAU,WAAW;EAC5D,IAAM,EAAC,WAAQ,aAAS,GAClB,IAAoC,GAA2B,CAChE,OAAO,EAAO,CACd,MAAM,EAAkB;AAC7B,SAAO,SAAS,GAAkB;GAC9B,IAAM,IAAI,EAAM,OAAO,EAAU,eAAe,EAAQ,WAAW,OAAO,EAAS,CAAC,CAAC;AACrF,UAAO,KAAK,IAAI,GAAG,EAAY;;;AAGvC,KAAI,EAA2B,EAAQ,KAAK,MAAU,aAAa,MAAU,cAAc;EACvF,IAAM,EAAC,WAAQ,UAAO,UAAO,aAAY,GACnC,IAAQ,EAAwB,GAAmB,GAAQ,EAAK;AACtE,SAAO,SAAS,GAAkB;GAC9B,IAAM,IAAI,EAAU,eAAe,EAAQ,WAAW,OAAO,EAAS;AAItE,UAHI,MAAM,OACC,EAAgB,QAEpB,EAAM,OAAO,EAAE,CAAC;;;AAG/B,KAAI,EAAe,EAAQ,EAAE;EACzB,IAAM,IAAW,EAAQ,OACnB,IAAU,EAAW,GAAU;AACrC,SAAO,SAAS,GAAkB;AAE9B,UAAQ,EADS,EAAU,uBAAuB,GAAU,EAAS,EAC3C,EAAM,IAAI,EAAmB;;;AAG/D,cAAa"}

package/dist/scatterplot/utils/createLegendInfo.js

@@ -1,62 +1,51 @@
-import { MAX_SHOWN_UNIQUE_VALUES_IN_LABELS as L } from "../../constants.js";
-const I = /* @__PURE__ */ (() => {
-  const n = /* @__PURE__ */ new Map();
-  return {
-    clear: () => n.clear(),
-    getIndex: (a, o) => {
-      let e = n.get(a.value);
-      e === void 0 && (e = /* @__PURE__ */ new Map(), n.set(a.value, e));
-      let r = e.get(o);
-      return r === void 0 && (r = e.size, e.set(o, r)), r;
-    }
-  };
+var e = (() => {
+	let e = /* @__PURE__ */ new Map();
+	return {
+		clear: () => e.clear(),
+		getIndex: (t, n) => {
+			let r = e.get(t.value);
+			r === void 0 && (r = /* @__PURE__ */ new Map(), e.set(t.value, r));
+			let i = r.get(n);
+			return i === void 0 && (i = r.size, r.set(n, i)), i;
+		}
+	};
 })();
-function v(n, h, d) {
-  I.clear();
-  const a = d && Object.keys(d).length > 0, o = h.dotFill, e = o && o.length > 0, r = h.lineColor, p = r && r.length > 0;
-  return (i, g) => {
-    var f;
-    const u = a ? (f = d[i]) == null ? void 0 : f[g] : void 0;
-    return g === "dotFill" && e ? u ?? o[I.getIndex(n, i) % o.length] : g === "lineColor" && p ? u ?? r[I.getIndex(n, i) % r.length] : u;
-  };
+function t(t, n, r) {
+	e.clear();
+	let i = r && Object.keys(r).length > 0, a = n.dotFill, o = a && a.length > 0, s = n.lineColor, c = s && s.length > 0;
+	return (n, l) => {
+		let u = i ? r[n]?.[l] : void 0;
+		return l === "dotFill" && o ? u ?? a[e.getIndex(t, n) % a.length] : l === "lineColor" && c ? u ?? s[e.getIndex(t, n) % s.length] : u;
+	};
 }
-function A(n) {
-  return typeof n == "object" && "type" in n && n.type === "grouping";
+function n(e) {
+	return typeof e == "object" && "type" in e && e.type === "grouping";
 }
-function F(n, h, d) {
-  return h.reduce((a, o) => {
-    const e = o.columnName, r = e.nullValueLabel ?? "NA", p = o.order, x = (s) => e.valueLabels ? n.getColumnValueCategory(e.valueLabels, s) : void 0, i = /* @__PURE__ */ new Set(), g = {};
-    d.forEach((s) => {
-      s.aes && Object.entries(s.aes).forEach(([c, t]) => {
-        A(t) && t.value === e.value && (i.add(c), g[c] = t.palette);
-      });
-    });
-    let u = {}, f = [];
-    if (i.has("dotFill") || i.has("dotShape") || i.has("lineColor")) {
-      const s = n.getColumnCategories(e.value, !1);
-      if (s.length < L) {
-        u = s.reduce((t, l) => {
-          const C = n.getColumnCategoryRowIndex(e.value, l);
-          return t[l] = x(C) ?? l, t[l] === "null" && (t[l] = r), t;
-        }, u);
-        const c = s.reduce((t, l) => (t[l] = (p == null ? void 0 : p.indexOf(l)) ?? -1, t), {});
-        f = s.sort((t, l) => {
-          var C;
-          return c[t] !== c[l] ? c[t] - c[l] : ((C = u[t]) == null ? void 0 : C.localeCompare(u[l] ?? "", "en", { numeric: !0 })) ?? 0;
-        });
-      } else
-        f = s;
-    }
-    return a[e.value] = {
-      values: f,
-      usedAes: [...i],
-      aesMap: v(e, g, o.inheritedAes),
-      labels: u
-    }, a;
-  }, {});
+function r(e, r, i) {
+	return r.reduce((r, a) => {
+		let o = a.columnName, s = o.nullValueLabel ?? "NA", c = a.order, l = (t) => o.valueLabels ? e.getColumnValueCategory(o.valueLabels, t) : void 0, u = /* @__PURE__ */ new Set(), d = {};
+		i.forEach((e) => {
+			e.aes && Object.entries(e.aes).forEach(([e, t]) => {
+				n(t) && t.value === o.value && (u.add(e), d[e] = t.palette);
+			});
+		});
+		let f = {}, p = [];
+		if (u.has("dotFill") || u.has("dotShape") || u.has("lineColor")) {
+			let t = e.getColumnCategories(o.value, !1);
+			if (t.length < 100) {
+				f = t.reduce((t, n) => (t[n] = l(e.getColumnCategoryRowIndex(o.value, n)) ?? n, t[n] === "null" && (t[n] = s), t), f);
+				let n = t.reduce((e, t) => (e[t] = c?.indexOf(t) ?? -1, e), {});
+				p = t.sort((e, t) => n[e] === n[t] ? f[e]?.localeCompare(f[t] ?? "", "en", { numeric: !0 }) ?? 0 : n[e] - n[t]);
+			} else p = t;
+		}
+		return r[o.value] = {
+			values: p,
+			usedAes: [...u],
+			aesMap: t(o, d, a.inheritedAes),
+			labels: f
+		}, r;
+	}, {});
 }
-export {
-  v as addPalettesToAesMapping,
-  F as createLegendInfo
-};
-//# sourceMappingURL=createLegendInfo.js.map
+export { t as addPalettesToAesMapping, r as createLegendInfo };
+
+//# sourceMappingURL=createLegendInfo.js.map

package/dist/scatterplot/utils/createLegendInfo.js.map

@@ -1 +1 @@
-{"version":3,"file":"createLegendInfo.js","sources":["../../../src/scatterplot/utils/createLegendInfo.ts"],"sourcesContent":["import { MAX_SHOWN_UNIQUE_VALUES_IN_LABELS } from '../../constants';\nimport type { DataFrame } from '../../DataFrame';\nimport type {AesRecord} from '../../types';\nimport type {\n    AesItem,\n    Category,\n    ColumnName,\n    ContinuousAesFromColumn,\n    InheritAesScatterplot,\n    ScatterplotLayer,\n    ScatterplotLegendInfo\n} from '../../types';\nimport type { ScatterplotSettingsImpl } from '../ScatterplotSettingsImpl';\n\nconst categoryIndexer = (() => {\n    const map = new Map<string, Map<Category, number>>();\n    const clear = () => map.clear();\n    const getIndex = (columnName: ColumnName, category: Category): number => {\n        let categoryMap = map.get(columnName.value);\n        if (categoryMap === undefined) {\n            categoryMap = new Map();\n            map.set(columnName.value, categoryMap);\n        }\n        let index = categoryMap.get(category);\n        if (index === undefined) {\n            index = categoryMap.size;\n            categoryMap.set(category, index);\n        }\n        return index;\n    };\n\n    return {\n        clear,\n        getIndex,\n    };\n})();\n\nexport function addPalettesToAesMapping(\n    columnName: ColumnName,\n    usedAesFromPalettes: Partial<Record<keyof AesItem, string[]>>,\n    inheritedAes?: AesRecord\n) {\n    categoryIndexer.clear();\n    const hasInheritedAes = inheritedAes && Object.keys(inheritedAes).length > 0;\n    const dotFill = usedAesFromPalettes['dotFill'];\n    const hasDotFill = dotFill && dotFill.length > 0;\n    const lineColor = usedAesFromPalettes['lineColor'];\n    const hasLineColor = lineColor && lineColor.length > 0;\n    const getAesField = (category: Category, field: keyof AesItem) => {\n        const inheritedResult = hasInheritedAes ? inheritedAes[category]?.[field] : undefined;\n        // Use inherited when defined; otherwise generate from palette so overflow categories get a color\n        if (field === 'dotFill' && hasDotFill) {\n            return inheritedResult ?? dotFill[categoryIndexer.getIndex(columnName, category) % dotFill.length];\n        }\n        if (field === 'lineColor' && hasLineColor) {\n            return inheritedResult ?? lineColor[categoryIndexer.getIndex(columnName, category) % lineColor.length];\n        }\n        return inheritedResult;\n    };\n\n    return getAesField;\n}\n\nfunction isInheritMapping<T extends (string | number)>(value:T | InheritAesScatterplot | ContinuousAesFromColumn<T>):value is InheritAesScatterplot {\n    return typeof value === 'object' && 'type' in value && value.type === 'grouping';\n}\nexport function createLegendInfo(\n    data: DataFrame,\n    grouping: ScatterplotSettingsImpl['grouping'],\n    layers: ScatterplotLayer[],\n): ScatterplotLegendInfo {\n    return grouping.reduce((res: ScatterplotLegendInfo, groupingElement) => {\n        // TODO: remove String conversion\n        const column = groupingElement.columnName;\n        const nullValueLabel = column.nullValueLabel ?? 'NA';\n        const order = groupingElement.order;\n        const getValueLabel = (index: number) => {\n            return column.valueLabels\n            ? data.getColumnValueCategory(column.valueLabels, index)\n            : undefined;\n        };\n\n        const usedAes = new Set<keyof AesItem>();\n        const usedAesFromPalettes:Partial<Record<keyof AesItem, string[]>> = {}; // if there is palette in mapping then\n        layers.forEach(layer => {\n            if (!layer.aes)  return;\n            \n            Object.entries(layer.aes).forEach(([key, value]) => {\n                if (!(isInheritMapping(value) && value.value === column.value)) return;\n                \n                usedAes.add(key as keyof AesItem);\n                usedAesFromPalettes[key as keyof AesItem] = value.palette;\n            });\n        });\n        let labels: Record<Category, string> = {};\n        let values: Category[] = [];\n        if (usedAes.has('dotFill') || usedAes.has('dotShape') || usedAes.has('lineColor')) {\n            const categories = data.getColumnCategories(column.value, false);\n            if (categories.length < MAX_SHOWN_UNIQUE_VALUES_IN_LABELS) {\n                labels = categories.reduce((res, category) => {\n                    const index = data.getColumnCategoryRowIndex(column.value, category);\n                    res[category] = getValueLabel(index) ?? category;\n                    if (res[category] === 'null') {\n                        res[category] = nullValueLabel;\n                    }\n                    return res;\n                }, labels);\n                const sortWeights = categories.reduce((res, v) => {\n                    res[v] = order?.indexOf(v) ?? -1;\n                    return res;\n                }, {} as Record<string, number>);\n                values = categories.sort((a, b) => {\n                    if (sortWeights[a] !== sortWeights[b]) {\n                        return sortWeights[a] - sortWeights[b];\n                    }\n                    return labels[a]?.localeCompare(labels[b] ?? '', 'en', {numeric: true}) ?? 0;\n                });\n            } else {\n                values = categories;\n            }\n        }\n        res[column.value] = {\n            values,\n            usedAes: [...usedAes],\n            aesMap: addPalettesToAesMapping(column, usedAesFromPalettes, groupingElement.inheritedAes),\n            labels,\n        };\n        return res;\n    }, {});\n}"],"names":["categoryIndexer","map","columnName","category","categoryMap","index","addPalettesToAesMapping","usedAesFromPalettes","inheritedAes","hasInheritedAes","dotFill","hasDotFill","lineColor","hasLineColor","field","inheritedResult","_a","isInheritMapping","value","createLegendInfo","data","grouping","layers","res","groupingElement","column","nullValueLabel","order","getValueLabel","usedAes","layer","key","labels","values","categories","MAX_SHOWN_UNIQUE_VALUES_IN_LABELS","sortWeights","v","a","b"],"mappings":";AAcA,MAAMA,IAAmB,uBAAM;AAC3B,QAAMC,wBAAU,IAAA;AAgBhB,SAAO;AAAA,IACH,OAhBU,MAAMA,EAAI,MAAA;AAAA,IAiBpB,UAhBa,CAACC,GAAwBC,MAA+B;AACrE,UAAIC,IAAcH,EAAI,IAAIC,EAAW,KAAK;AAC1C,MAAIE,MAAgB,WAChBA,wBAAkB,IAAA,GAClBH,EAAI,IAAIC,EAAW,OAAOE,CAAW;AAEzC,UAAIC,IAAQD,EAAY,IAAID,CAAQ;AACpC,aAAIE,MAAU,WACVA,IAAQD,EAAY,MACpBA,EAAY,IAAID,GAAUE,CAAK,IAE5BA;AAAA,IACX;AAAA,EAII;AAER,GAAA;AAEO,SAASC,EACZJ,GACAK,GACAC,GACF;AACE,EAAAR,EAAgB,MAAA;AAChB,QAAMS,IAAkBD,KAAgB,OAAO,KAAKA,CAAY,EAAE,SAAS,GACrEE,IAAUH,EAAoB,SAC9BI,IAAaD,KAAWA,EAAQ,SAAS,GACzCE,IAAYL,EAAoB,WAChCM,IAAeD,KAAaA,EAAU,SAAS;AAarD,SAZoB,CAACT,GAAoBW,MAAyB;;AAC9D,UAAMC,IAAkBN,KAAkBO,IAAAR,EAAaL,CAAQ,MAArB,gBAAAa,EAAyBF,KAAS;AAE5E,WAAIA,MAAU,aAAaH,IAChBI,KAAmBL,EAAQV,EAAgB,SAASE,GAAYC,CAAQ,IAAIO,EAAQ,MAAM,IAEjGI,MAAU,eAAeD,IAClBE,KAAmBH,EAAUZ,EAAgB,SAASE,GAAYC,CAAQ,IAAIS,EAAU,MAAM,IAElGG;AAAA,EACX;AAGJ;AAEA,SAASE,EAA8CC,GAA6F;AAChJ,SAAO,OAAOA,KAAU,YAAY,UAAUA,KAASA,EAAM,SAAS;AAC1E;AACO,SAASC,EACZC,GACAC,GACAC,GACqB;AACrB,SAAOD,EAAS,OAAO,CAACE,GAA4BC,MAAoB;AAEpE,UAAMC,IAASD,EAAgB,YACzBE,IAAiBD,EAAO,kBAAkB,MAC1CE,IAAQH,EAAgB,OACxBI,IAAgB,CAACvB,MACZoB,EAAO,cACZL,EAAK,uBAAuBK,EAAO,aAAapB,CAAK,IACrD,QAGAwB,wBAAc,IAAA,GACdtB,IAA+D,CAAA;AACrE,IAAAe,EAAO,QAAQ,CAAAQ,MAAS;AACpB,MAAKA,EAAM,OAEX,OAAO,QAAQA,EAAM,GAAG,EAAE,QAAQ,CAAC,CAACC,GAAKb,CAAK,MAAM;AAChD,QAAMD,EAAiBC,CAAK,KAAKA,EAAM,UAAUO,EAAO,UAExDI,EAAQ,IAAIE,CAAoB,GAChCxB,EAAoBwB,CAAoB,IAAIb,EAAM;AAAA,MACtD,CAAC;AAAA,IACL,CAAC;AACD,QAAIc,IAAmC,CAAA,GACnCC,IAAqB,CAAA;AACzB,QAAIJ,EAAQ,IAAI,SAAS,KAAKA,EAAQ,IAAI,UAAU,KAAKA,EAAQ,IAAI,WAAW,GAAG;AAC/E,YAAMK,IAAad,EAAK,oBAAoBK,EAAO,OAAO,EAAK;AAC/D,UAAIS,EAAW,SAASC,GAAmC;AACvD,QAAAH,IAASE,EAAW,OAAO,CAACX,GAAKpB,MAAa;AAC1C,gBAAME,IAAQe,EAAK,0BAA0BK,EAAO,OAAOtB,CAAQ;AACnEoB,iBAAAA,EAAIpB,CAAQ,IAAIyB,EAAcvB,CAAK,KAAKF,GACpCoB,EAAIpB,CAAQ,MAAM,WAClBoB,EAAIpB,CAAQ,IAAIuB,IAEbH;AAAAA,QACX,GAAGS,CAAM;AACT,cAAMI,IAAcF,EAAW,OAAO,CAACX,GAAKc,OACxCd,EAAIc,CAAC,KAAIV,KAAA,gBAAAA,EAAO,QAAQU,OAAM,IACvBd,IACR,CAAA,CAA4B;AAC/B,QAAAU,IAASC,EAAW,KAAK,CAACI,GAAGC,MAAM;;AAC/B,iBAAIH,EAAYE,CAAC,MAAMF,EAAYG,CAAC,IACzBH,EAAYE,CAAC,IAAIF,EAAYG,CAAC,MAElCvB,IAAAgB,EAAOM,CAAC,MAAR,gBAAAtB,EAAW,cAAcgB,EAAOO,CAAC,KAAK,IAAI,MAAM,EAAC,SAAS,GAAA,OAAU;AAAA,QAC/E,CAAC;AAAA,MACL;AACI,QAAAN,IAASC;AAAA,IAEjB;AACA,WAAAX,EAAIE,EAAO,KAAK,IAAI;AAAA,MAChB,QAAAQ;AAAA,MACA,SAAS,CAAC,GAAGJ,CAAO;AAAA,MACpB,QAAQvB,EAAwBmB,GAAQlB,GAAqBiB,EAAgB,YAAY;AAAA,MACzF,QAAAQ;AAAA,IAAA,GAEGT;AAAA,EACX,GAAG,CAAA,CAAE;AACT;"}
+{"version":3,"file":"createLegendInfo.js","names":[],"sources":["../../../src/scatterplot/utils/createLegendInfo.ts"],"sourcesContent":["import { MAX_SHOWN_UNIQUE_VALUES_IN_LABELS } from '../../constants';\nimport type { DataFrame } from '../../DataFrame';\nimport type {AesRecord} from '../../types';\nimport type {\n    AesItem,\n    Category,\n    ColumnName,\n    ContinuousAesFromColumn,\n    InheritAesScatterplot,\n    ScatterplotLayer,\n    ScatterplotLegendInfo\n} from '../../types';\nimport type { ScatterplotSettingsImpl } from '../ScatterplotSettingsImpl';\n\nconst categoryIndexer = (() => {\n    const map = new Map<string, Map<Category, number>>();\n    const clear = () => map.clear();\n    const getIndex = (columnName: ColumnName, category: Category): number => {\n        let categoryMap = map.get(columnName.value);\n        if (categoryMap === undefined) {\n            categoryMap = new Map();\n            map.set(columnName.value, categoryMap);\n        }\n        let index = categoryMap.get(category);\n        if (index === undefined) {\n            index = categoryMap.size;\n            categoryMap.set(category, index);\n        }\n        return index;\n    };\n\n    return {\n        clear,\n        getIndex,\n    };\n})();\n\nexport function addPalettesToAesMapping(\n    columnName: ColumnName,\n    usedAesFromPalettes: Partial<Record<keyof AesItem, string[]>>,\n    inheritedAes?: AesRecord\n) {\n    categoryIndexer.clear();\n    const hasInheritedAes = inheritedAes && Object.keys(inheritedAes).length > 0;\n    const dotFill = usedAesFromPalettes['dotFill'];\n    const hasDotFill = dotFill && dotFill.length > 0;\n    const lineColor = usedAesFromPalettes['lineColor'];\n    const hasLineColor = lineColor && lineColor.length > 0;\n    const getAesField = (category: Category, field: keyof AesItem) => {\n        const inheritedResult = hasInheritedAes ? inheritedAes[category]?.[field] : undefined;\n        // Use inherited when defined; otherwise generate from palette so overflow categories get a color\n        if (field === 'dotFill' && hasDotFill) {\n            return inheritedResult ?? dotFill[categoryIndexer.getIndex(columnName, category) % dotFill.length];\n        }\n        if (field === 'lineColor' && hasLineColor) {\n            return inheritedResult ?? lineColor[categoryIndexer.getIndex(columnName, category) % lineColor.length];\n        }\n        return inheritedResult;\n    };\n\n    return getAesField;\n}\n\nfunction isInheritMapping<T extends (string | number)>(value:T | InheritAesScatterplot | ContinuousAesFromColumn<T>):value is InheritAesScatterplot {\n    return typeof value === 'object' && 'type' in value && value.type === 'grouping';\n}\nexport function createLegendInfo(\n    data: DataFrame,\n    grouping: ScatterplotSettingsImpl['grouping'],\n    layers: ScatterplotLayer[],\n): ScatterplotLegendInfo {\n    return grouping.reduce((res: ScatterplotLegendInfo, groupingElement) => {\n        // TODO: remove String conversion\n        const column = groupingElement.columnName;\n        const nullValueLabel = column.nullValueLabel ?? 'NA';\n        const order = groupingElement.order;\n        const getValueLabel = (index: number) => {\n            return column.valueLabels\n            ? data.getColumnValueCategory(column.valueLabels, index)\n            : undefined;\n        };\n\n        const usedAes = new Set<keyof AesItem>();\n        const usedAesFromPalettes:Partial<Record<keyof AesItem, string[]>> = {}; // if there is palette in mapping then\n        layers.forEach(layer => {\n            if (!layer.aes)  return;\n            \n            Object.entries(layer.aes).forEach(([key, value]) => {\n                if (!(isInheritMapping(value) && value.value === column.value)) return;\n                \n                usedAes.add(key as keyof AesItem);\n                usedAesFromPalettes[key as keyof AesItem] = value.palette;\n            });\n        });\n        let labels: Record<Category, string> = {};\n        let values: Category[] = [];\n        if (usedAes.has('dotFill') || usedAes.has('dotShape') || usedAes.has('lineColor')) {\n            const categories = data.getColumnCategories(column.value, false);\n            if (categories.length < MAX_SHOWN_UNIQUE_VALUES_IN_LABELS) {\n                labels = categories.reduce((res, category) => {\n                    const index = data.getColumnCategoryRowIndex(column.value, category);\n                    res[category] = getValueLabel(index) ?? category;\n                    if (res[category] === 'null') {\n                        res[category] = nullValueLabel;\n                    }\n                    return res;\n                }, labels);\n                const sortWeights = categories.reduce((res, v) => {\n                    res[v] = order?.indexOf(v) ?? -1;\n                    return res;\n                }, {} as Record<string, number>);\n                values = categories.sort((a, b) => {\n                    if (sortWeights[a] !== sortWeights[b]) {\n                        return sortWeights[a] - sortWeights[b];\n                    }\n                    return labels[a]?.localeCompare(labels[b] ?? '', 'en', {numeric: true}) ?? 0;\n                });\n            } else {\n                values = categories;\n            }\n        }\n        res[column.value] = {\n            values,\n            usedAes: [...usedAes],\n            aesMap: addPalettesToAesMapping(column, usedAesFromPalettes, groupingElement.inheritedAes),\n            labels,\n        };\n        return res;\n    }, {});\n}"],"mappings":"AAcA,IAAM,WAAyB;CAC3B,IAAM,oBAAM,IAAI,KAAoC;AAgBpD,QAAO;EACH,aAhBgB,EAAI,OAAO;EAiB3B,WAhBc,GAAwB,MAA+B;GACrE,IAAI,IAAc,EAAI,IAAI,EAAW,MAAM;AAC3C,GAAI,MAAgB,KAAA,MAChB,oBAAc,IAAI,KAAK,EACvB,EAAI,IAAI,EAAW,OAAO,EAAY;GAE1C,IAAI,IAAQ,EAAY,IAAI,EAAS;AAKrC,UAJI,MAAU,KAAA,MACV,IAAQ,EAAY,MACpB,EAAY,IAAI,GAAU,EAAM,GAE7B;;EAMV;IACD;AAEJ,SAAgB,EACZ,GACA,GACA,GACF;AACE,GAAgB,OAAO;CACvB,IAAM,IAAkB,KAAgB,OAAO,KAAK,EAAa,CAAC,SAAS,GACrE,IAAU,EAAoB,SAC9B,IAAa,KAAW,EAAQ,SAAS,GACzC,IAAY,EAAoB,WAChC,IAAe,KAAa,EAAU,SAAS;AAarD,SAZqB,GAAoB,MAAyB;EAC9D,IAAM,IAAkB,IAAkB,EAAa,KAAY,KAAS,KAAA;AAQ5E,SANI,MAAU,aAAa,IAChB,KAAmB,EAAQ,EAAgB,SAAS,GAAY,EAAS,GAAG,EAAQ,UAE3F,MAAU,eAAe,IAClB,KAAmB,EAAU,EAAgB,SAAS,GAAY,EAAS,GAAG,EAAU,UAE5F;;;AAMf,SAAS,EAA8C,GAA6F;AAChJ,QAAO,OAAO,KAAU,YAAY,UAAU,KAAS,EAAM,SAAS;;AAE1E,SAAgB,EACZ,GACA,GACA,GACqB;AACrB,QAAO,EAAS,QAAQ,GAA4B,MAAoB;EAEpE,IAAM,IAAS,EAAgB,YACzB,IAAiB,EAAO,kBAAkB,MAC1C,IAAQ,EAAgB,OACxB,KAAiB,MACZ,EAAO,cACZ,EAAK,uBAAuB,EAAO,aAAa,EAAM,GACtD,KAAA,GAGA,oBAAU,IAAI,KAAoB,EAClC,IAA+D,EAAE;AACvE,IAAO,SAAQ,MAAS;AACf,KAAM,OAEX,OAAO,QAAQ,EAAM,IAAI,CAAC,SAAS,CAAC,GAAK,OAAW;AAC1C,MAAiB,EAAM,IAAI,EAAM,UAAU,EAAO,UAExD,EAAQ,IAAI,EAAqB,EACjC,EAAoB,KAAwB,EAAM;KACpD;IACJ;EACF,IAAI,IAAmC,EAAE,EACrC,IAAqB,EAAE;AAC3B,MAAI,EAAQ,IAAI,UAAU,IAAI,EAAQ,IAAI,WAAW,IAAI,EAAQ,IAAI,YAAY,EAAE;GAC/E,IAAM,IAAa,EAAK,oBAAoB,EAAO,OAAO,GAAM;AAChE,OAAI,EAAW,SAAA,KAA4C;AACvD,QAAS,EAAW,QAAQ,GAAK,OAE7B,EAAI,KAAY,EADF,EAAK,0BAA0B,EAAO,OAAO,EAAS,CAChC,IAAI,GACpC,EAAI,OAAc,WAClB,EAAI,KAAY,IAEb,IACR,EAAO;IACV,IAAM,IAAc,EAAW,QAAQ,GAAK,OACxC,EAAI,KAAK,GAAO,QAAQ,EAAE,IAAI,IACvB,IACR,EAAE,CAA2B;AAChC,QAAS,EAAW,MAAM,GAAG,MACrB,EAAY,OAAO,EAAY,KAG5B,EAAO,IAAI,cAAc,EAAO,MAAM,IAAI,MAAM,EAAC,SAAS,IAAK,CAAC,IAAI,IAFhE,EAAY,KAAK,EAAY,GAG1C;SAEF,KAAS;;AASjB,SANA,EAAI,EAAO,SAAS;GAChB;GACA,SAAS,CAAC,GAAG,EAAQ;GACrB,QAAQ,EAAwB,GAAQ,GAAqB,EAAgB,aAAa;GAC1F;GACH,EACM;IACR,EAAE,CAAC"}