@milaboratories/miplots4 1.0.129 → 1.0.131

package/dist/scatterplot/linearRegression.js.map

@@ -1 +1 @@
-{"version":3,"file":"linearRegression.js","sources":["../../src/scatterplot/linearRegression.ts"],"sourcesContent":["import type {Dot, GroupedDots} from './dots';\nimport type {ScatterplotSettingsImpl} from './ScatterplotSettingsImpl';\nimport type {ContinuousAesFromColumn, InheritAesScatterplot, Row} from '../types';\nimport type {ColumnName} from '../types';\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';\n\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    /* eslint-disable @typescript-eslint/no-loss-of-precision */\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    /* eslint-enable @typescript-eslint/no-loss-of-precision */\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: {x: number; y: number}[]): [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: Dot[], 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: {x: number; y: number}[],\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: Dot[]) {\n    const n = dots.length;\n    return Math.pow(\n        (n * sum(dots, (d: Dot) => d.x * d.y) - sum(dots, (d: Dot) => d.x) * sum(dots, (d: Dot) => 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: Dot[], getter: (d: Dot) => 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: Dot[], getter: (d: Dot) => number) {\n    return Math.sqrt(getVariance(arr, getter));\n}\n\nfunction getCovariance(arr: Dot[], getterX: (d: Dot) => number, getterY: (d: Dot) => 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: Dot[]) {\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: Dot[], 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    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    data: Row;\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\n//dotsByFacets, facetKeys, categoryGroupingCombinations, grouping, trend\nexport function getRegressionData(\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        res[facetKey] = groupingKeysValues.map(values => {\n            const dots = facetDots.filter(dot =>\n                grouping.every((column, idx) => String(dot.data[column.value]) === values[idx])\n            );\n            const [minX = 0, maxX = 0] = extent(dots, (dot: Dot) => 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                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                data: dots[0].data,\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        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","b","c","d","linearRegression","data","n","sumX","sumY","sumXY","sumX2","i","xi","yi","slope","intercept","getCoefficientsInterval","dots","predict","meanX","sum","t","se_b","dot","slopeError","interceptError","confidenceInterval","mean","a","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","dotsByFacets","facetKeys","groupingKeysValues","grouping","trend","res","facetKey","facetDots","values","column","idx","minX","maxX","extent","getInterval","scaleX","scaleY","range","area"],"mappings":";;;;;AASA,MAAMA,IAAmB;AAazB,SAASC,GAAQC,GAAW;AAMxB,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,uBAGHC,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,QAAM,IAAI,KAAKkD,IAAK,MACdS,IAAI,MAAM,IAAI;AACpB,MAAIC,MAAO,QAAQ,IAAKD,IAAI,MAAM,IAAI,MAAM,IAAI;AAChD,QAAME,MAAM,QAAQF,IAAIC,KAAK,KAAKD,IAAI,KAAKN,EAAK,IAAIG,IAAK,GAAG,IAAIN;AAChE,SAAAO,IAAII,IAAI7D,GACR0D,IAAIJ,EAAIG,GAAG,IAAIP,CAAE,GAEbQ,IAAI,OAAO,KAGXD,IAAI1D,GAAQC,IAAI,CAAC,GACjB0D,IAAID,IAAIA,GACJP,IAAK,MAAGU,IAAIA,IAAI,OAAOV,IAAK,QAAQO,IAAI,OAC5CG,OAAO,OAAOC,IAAIJ,IAAI,KAAKA,IAAI,KAAKA,IAAI,KAAKA,IAAIE,IAAIC,GACrDF,SAAS,MAAMA,IAAI,OAAOA,IAAI,MAAMA,IAAI,QAAQE,IAAIF,IAAI,KAAKC,IAAI,KAAKF,GACtEC,IAAI,IAAIA,IAAIA,GACRA,IAAI,OAAOA,IAAIH,EAAIG,CAAC,IAAI,IACvBA,IAAI,MAAMA,IAAIA,IAAIA,KAEvBA,MACO,OAAOR,IAAK,MAAMA,IAAKQ,KAAK,QAAQG,IAAI,UAAUX,IAAK,KAAK,KAAK,OAAOA,IAAK,MAAMQ,IAAI,MAAMR,IAAK,MAChGA,IAAK,KACV,IAAIQ,GAGLL,EAAKH,IAAKQ,CAAC;AACtB;AAEO,SAASI,GAAiBC,GAAkD;AAC/E,MAAIC,IAAID,EAAK,QACTE,IAAO,GACPC,IAAO,GACPC,IAAQ,GACRC,IAAQ;AACZ,WAASC,IAAI,GAAGA,IAAIN,EAAK,QAAQ,EAAEM,GAAG;AAClC,UAAMC,IAAKP,EAAKM,CAAC,EAAE,GACbE,IAAKR,EAAKM,CAAC,EAAE;AACnB,QAAI,CAAC,OAAO,SAASC,CAAE,KAAK,CAAC,OAAO,SAASC,CAAE,GAAG;AAC9C,QAAEP;AACF;AAAA,IACJ;AACA,IAAAC,KAAQK,GACRJ,KAAQK,GACRJ,KAASG,IAAKC,GACdH,KAASE,IAAKA;AAAA,EAClB;AACA,QAAME,KAASR,IAAIG,IAAQF,IAAOC,MAASF,IAAII,IAAQH,IAAOA,IACxDQ,KAAaP,IAAOM,IAAQP,KAAQD;AAE1C,SAAO,CAACQ,GAAOC,CAAS;AAC5B;AAEA,SAASC,GAAwBC,GAAaH,GAAeC,GAAmB;AAC5E,QAAMG,IAAU,CAACnB,MAAcA,IAAIe,IAAQC,GACrCT,IAAIW,EAAK,QACTE,IAAQC,EAAIH,GAAM,OAAKd,EAAE,CAAC,IAAIc,EAAK,QACnCI,IAAI9B,EAASnD,GAAkBkE,IAAI,CAAC,GAEpCgB,IAAO,KAAK;AAAA,IACdF,EAAIH,GAAM,CAAAM,OAAQA,EAAI,IAAIL,EAAQK,EAAI,CAAC,MAAM,CAAC,KAAKjB,IAAI,KAAKc,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,IAAIjB,CAAC,IAChCe;AAE9B,SAAO,EAAC,OAAAP,GAAO,WAAAC,GAAW,YAAAS,GAAY,gBAAAC,EAAA;AAC1C;AAMO,SAASC,GACZrB,GACAa,GACgC;AAChC,QAAMS,IAAOP,EAAIf,GAAM,OAAKF,EAAE,CAAC,IAAIE,EAAK;AACxC,MAAIuB,IAAI,GACJ3B,IAAI;AACR,WAAS,IAAI,GAAG,IAAII,EAAK,QAAQ,EAAE;AAC/B,IAAAuB,KAAK,KAAK,IAAIvB,EAAK,CAAC,EAAE,IAAIsB,GAAM,CAAC,GACjC1B,KAAK,KAAK,IAAII,EAAK,CAAC,EAAE,IAAIa,EAAQb,EAAK,CAAC,EAAE,CAAC,GAAG,CAAC;AAEnD,QAAMwB,IAAK,KAAK,KAAK5B,KAAKI,EAAK,SAAS,EAAE,GACpCgB,IAAI9B,EAASnD,GAAkBiE,EAAK,SAAS,CAAC;AACpD,SAAO,SAAUN,GAAW;AACxB,UAAM+B,IAAIZ,EAAQnB,CAAC,GACbgC,IAAKF,IAAK,KAAK,KAAK,IAAIxB,EAAK,SAAS,KAAK,IAAIN,IAAI4B,GAAM,CAAC,IAAIC,CAAC,GAC/DI,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,GAAa;AACxB,QAAMX,IAAIW,EAAK;AACf,SAAO,KAAK;AAAA,KACPX,IAAIc,EAAIH,GAAM,CAACd,MAAWA,EAAE,IAAIA,EAAE,CAAC,IAAIiB,EAAIH,GAAM,CAACd,MAAWA,EAAE,CAAC,IAAIiB,EAAIH,GAAM,CAACd,MAAWA,EAAE,CAAC,MACzF,KAAK;AAAA,MACFG,IAAIc,EAAIH,GAAM,CAAAd,MAAKA,EAAE,IAAIA,EAAE,CAAC,IACxB,KAAK;AAAA,QACDiB,EAAIH,GAAM,CAAAd,MAAKA,EAAE,CAAC;AAAA,QAClB;AAAA,MAAA;AAAA,IACJ,IAEJ,KAAK;AAAA,MACDG,IAAIc,EAAIH,GAAM,CAAAd,MAAKA,EAAE,IAAIA,EAAE,CAAC,IACxB,KAAK;AAAA,QACDiB,EAAIH,GAAM,CAAAd,MAAKA,EAAE,CAAC;AAAA,QAClB;AAAA,MAAA;AAAA,IACJ;AAAA,IAEhB;AAAA,EAAA;AAER;AAEA,SAASgC,GAAYC,GAAYC,GAA4B;AACzD,QAAMC,IAAQX,EAAKS,GAAKC,CAAM;AAE9B,SAAOV,EAAKS,EAAI,IAAI,CAAAG,OAAMF,EAAOE,CAAC,IAAID,MAAU,CAAC,CAAC;AACtD;AAEA,SAASE,EAAqBJ,GAAYC,GAA4B;AAClE,SAAO,KAAK,KAAKF,GAAYC,GAAKC,CAAM,CAAC;AAC7C;AAEA,SAASI,GAAcL,GAAYM,GAA6BC,GAA6B;AACzF,QAAMxB,IAAQQ,EAAKS,GAAKM,CAAO,GACzBE,IAAQjB,EAAKS,GAAKO,CAAO,GACzBrC,IAAI8B,EAAI;AAEd,SAAOhB,EAAIgB,GAAK,CAAAjC,OAAMA,EAAE,IAAIgB,MAAUhB,EAAE,IAAIyC,EAAM,IAAItC;AAC1D;AAEA,SAASuC,GAAK5B,GAAa;AACvB,QAAM6B,IAAML;AAAA,IACRxB;AAAA,IACA,OAAKd,EAAE;AAAA,IACP,OAAKA,EAAE;AAAA,EAAA,GAEL4C,IAASP,EAAqBvB,GAAM,CAAAd,MAAKA,EAAE,CAAC,GAC5C6C,IAASR,EAAqBvB,GAAM,CAAAd,MAAKA,EAAE,CAAC;AAElD,SAAO2C,KAAOC,IAASC;AAC3B;AAEA,SAASC,GAAUhC,GAAaC,GAAgC;AAC5D,QAAMZ,IAAIW,EAAK,QACT3E,IAAI,GACJ4G,IAAM,GACNC,IAAM7C,IAAIhE,IAAI,GACdsG,IAAQjB,EAAKV,GAAM,CAAAd,MAAKA,EAAE,CAAC,GAC3BiD,IAAMhC,EAAIH,GAAM,CAAAd,OAAMe,EAAQf,EAAE,CAAC,IAAIyC,MAAU,CAAC,GAChDS,IAAMjC,EAAIH,GAAM,CAAAd,OAAMe,EAAQf,EAAE,CAAC,IAAIA,EAAE,MAAM,CAAC,GAE9CmD,IAAMF,IAAMF,GACZK,IAAMF,IAAMF,GAEZK,IAAIF,IAAMC;AAIhB,SAAO,IAFME,GAAKD,GAAGN,GAAKC,CAAG;AAGjC;AAqBO,SAASO,GACZC,GACAC,GACAC,GACAC,GACAC,GACiB;AACjB,SAAKA,IAGEH,EAAU,OAAO,CAACI,GAAiBC,MAAa;AACnD,UAAMC,IAAYP,EAAaM,CAAQ,EAAE;AACzC,WAAAD,EAAIC,CAAQ,IAAIJ,EAAmB,IAAI,CAAAM,MAAU;AAC7C,YAAMlD,IAAOiD,EAAU;AAAA,QAAO,CAAA3C,MAC1BuC,EAAS,MAAM,CAACM,GAAQC,MAAQ,OAAO9C,EAAI,KAAK6C,EAAO,KAAK,CAAC,MAAMD,EAAOE,CAAG,CAAC;AAAA,MAAA,GAE5E,CAACC,IAAO,GAAGC,IAAO,CAAC,IAAIC,GAAOvD,GAAM,CAACM,MAAaA,EAAI,CAAC,GACvD,CAACT,GAAOC,CAAS,IAAIX,GAAiBa,CAAI,GAC1CC,IAAU,CAACnB,MAAce,IAAQf,IAAIgB,GACrC0D,IAAc/C,GAAmBT,GAAMC,CAAO;AAEpD,aAAO;AAAA,QACH,SAAAA;AAAA,QACA,aAAAuD;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,MAAM9C,EAAK,CAAC,EAAE;AAAA,QACd,WAAW;AAAA,QACX,SAAS,CAACqD,GAAMC,CAAI;AAAA,QACpB,OAAO;AAAA,UACH,IAAIrC,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,GACMiD;AAAA,EACX,GAAG,CAAA,CAAE,IAnCM;AAoCf;"}
+{"version":3,"file":"linearRegression.js","sources":["../../src/scatterplot/linearRegression.ts"],"sourcesContent":["import 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\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    /* eslint-disable @typescript-eslint/no-loss-of-precision */\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    /* eslint-enable @typescript-eslint/no-loss-of-precision */\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: {x: number; y: number}[]): [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: Dot[], 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: {x: number; y: number}[],\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: Dot[]) {\n    const n = dots.length;\n    return Math.pow(\n        (n * sum(dots, (d: Dot) => d.x * d.y) - sum(dots, (d: Dot) => d.x) * sum(dots, (d: Dot) => 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: Dot[], getter: (d: Dot) => 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: Dot[], getter: (d: Dot) => number) {\n    return Math.sqrt(getVariance(arr, getter));\n}\n\nfunction getCovariance(arr: Dot[], getterX: (d: Dot) => number, getterY: (d: Dot) => 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: Dot[]) {\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: Dot[], 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                );\n\n                if (dots.length === 0) return null;\n\n                const [minX = 0, maxX = 0] = extent(dots, (dot: Dot) => 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].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":";;;;;AASA,MAAMA,IAAmB;AAazB,SAASC,GAAQC,GAAW;AAMxB,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,uBAGHC,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,GAAkD;AAC/E,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,GAAaH,GAAeC,GAAmB;AAC5E,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,GAAa;AACxB,QAAMV,IAAIU,EAAK;AACf,SAAO,KAAK;AAAA,KACPV,IAAIa,EAAIH,GAAM,CAACb,MAAWA,EAAE,IAAIA,EAAE,CAAC,IAAIgB,EAAIH,GAAM,CAACb,MAAWA,EAAE,CAAC,IAAIgB,EAAIH,GAAM,CAACb,MAAWA,EAAE,CAAC,MACzF,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,GAAYC,GAA4B;AACzD,QAAMC,IAAQX,EAAKS,GAAKC,CAAM;AAE9B,SAAOV,EAAKS,EAAI,IAAI,CAAAG,OAAMF,EAAOE,CAAC,IAAID,MAAU,CAAC,CAAC;AACtD;AAEA,SAASE,EAAqBJ,GAAYC,GAA4B;AAClE,SAAO,KAAK,KAAKF,GAAYC,GAAKC,CAAM,CAAC;AAC7C;AAEA,SAASI,GAAcL,GAAYM,GAA6BC,GAA6B;AACzF,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,GAAa;AACvB,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,GAAaC,GAAgC;AAC5D,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,MAAaA,EAAI,CAAC,GACvD,CAACT,GAAOC,CAAS,IAAIV,GAAiBY,CAAI,GAC1CC,IAAU,CAACnB,MAAce,IAAQf,IAAIgB,GACrC2D,IAAchD,GAAmBT,GAAMC,CAAO;AAEpD,aAAO;AAAA,QACH,KAAKD,EAAK,CAAC,EAAE;AAAA,QACb,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;"}

package/dist/scatterplot/utils/createAesGetter.d.ts

@@ -1,3 +1,4 @@
-import { AesItem, ContinuousAesFromColumn, InheritAesScatterplot, LineShape, PointShape, Row, ScatterplotLegendInfo } from '../../types';
-export declare function createAesGetter<ValueType extends string | number | PointShape | LineShape>(legendInfo: ScatterplotLegendInfo, aesItem: ValueType | ContinuousAesFromColumn<ValueType> | InheritAesScatterplot, field: keyof AesItem): (row: Row) => ValueType;
+import { DataFrame } from '../../DataFrame';
+import { AesItem, ContinuousAesFromColumn, InheritAesScatterplot, LineShape, PointShape, ScatterplotLegendInfo } from '../../types';
+export declare function createAesGetter<ValueType extends string | number | PointShape | LineShape>(dataFrame: DataFrame, legendInfo: ScatterplotLegendInfo, aesItem: ValueType | ContinuousAesFromColumn<ValueType> | InheritAesScatterplot, field: keyof AesItem): (rowIndex: number) => ValueType;
 //# sourceMappingURL=createAesGetter.d.ts.map

package/dist/scatterplot/utils/createAesGetter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"createAesGetter.d.ts","sourceRoot":"","sources":["../../../src/scatterplot/utils/createAesGetter.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACR,OAAO,EACP,uBAAuB,EACvB,qBAAqB,EACrB,SAAS,EACT,UAAU,EAAE,GAAG,EACf,qBAAqB,EACxB,MAAM,aAAa,CAAC;AASrB,wBAAgB,eAAe,CAAC,SAAS,SAAS,MAAM,GAAG,MAAM,GAAG,UAAU,GAAG,SAAS,EACtF,UAAU,EAAE,qBAAqB,EACjC,OAAO,EAAE,SAAS,GAAG,uBAAuB,CAAC,SAAS,CAAC,GAAG,qBAAqB,EAC/E,KAAK,EAAE,MAAM,OAAO,SAOI,GAAG,KAEgB,SAAS,CAsBvD"}
+{"version":3,"file":"createAesGetter.d.ts","sourceRoot":"","sources":["../../../src/scatterplot/utils/createAesGetter.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EACR,OAAO,EACP,uBAAuB,EACvB,qBAAqB,EACrB,SAAS,EACT,UAAU,EACV,qBAAqB,EACxB,MAAM,aAAa,CAAC;AAQrB,wBAAgB,eAAe,CAAC,SAAS,SAAS,MAAM,GAAG,MAAM,GAAG,UAAU,GAAG,SAAS,EACtF,SAAS,EAAE,SAAS,EACpB,UAAU,EAAE,qBAAqB,EACjC,OAAO,EAAE,SAAS,GAAG,uBAAuB,CAAC,SAAS,CAAC,GAAG,qBAAqB,EAC/E,KAAK,EAAE,MAAM,OAAO,cAOU,MAAM,KAEO,SAAS,CAuBvD"}

package/dist/scatterplot/utils/createAesGetter.js

@@ -1,4 +1,3 @@
-import { DEFAULT_DOT_AES as p, DEFAULT_COMMON_AES as a } from "../constants.js";
 import { isContinuousAes as m } from "../../types/common.js";
 import "../../types/discrete.js";
 import "../../types/scatterplot.js";
@@ -6,36 +5,37 @@ import "../../types/heatmap.js";
 import "../../types/dendro.js";
 import "../../types/histogram.js";
 import "../../types/bubble.js";
-import { getContinuousColorScale as f } from "../../utils/getContinuousColorScale.js";
-import { sqrt as s } from "../../node_modules/d3-scale/src/pow.js";
-function A(n) {
-  return typeof n == "object" && n !== null && "type" in n && "value" in n && n.type === "grouping";
+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/d3-scale/src/pow.js";
+function C(o) {
+  return typeof o == "object" && o !== null && "type" in o && "value" in o && o.type === "grouping";
 }
-function D(n, o, e) {
-  if (m(o) && e === "dotSize") {
-    const { domain: t, range: r } = o, u = s().domain(t).range(r);
-    return function(i) {
-      const c = u(Number(i[o.columnName.value]));
+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(o) && (e === "dotFill" || e === "lineColor")) {
-    const { domain: t, range: r, type: u = "linear" } = o, i = f(r, t, u);
+  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 = c[o.columnName.value];
-      return l === null ? p.color : i(Number(l));
+      const l = o.getColumnValue(n.columnName.value, c);
+      return l === null ? g.color : e(Number(l));
     };
   }
-  if (A(o)) {
-    const t = o.value, r = n[t].aesMap;
-    return function(u) {
-      var i;
-      return (i = (r == null ? void 0 : r[String(u[t])]) ?? a) == null ? void 0 : i[e];
+  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 () => o;
+  return () => n;
 }
 export {
-  D as createAesGetter
+  T 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 {DEFAULT_COMMON_AES, DEFAULT_DOT_AES} from '../constants';\nimport type {\n    AesItem,\n    ContinuousAesFromColumn,\n    InheritAesScatterplot,\n    LineShape,\n    PointShape, Row,\n    ScatterplotLegendInfo\n} from '../../types';\nimport {isContinuousAes} from '../../types';\nimport {getContinuousColorScale} from '../../utils/getContinuousColorScale';\nimport type {ScalePower} from 'd3-scale';\nimport {scaleSqrt} from 'd3-scale';\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    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(row:Row) {\n            const v = scale(Number(row[aesItem.columnName.value]));\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(row:Row) {\n            const v = row[aesItem.columnName.value];\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(row:Row) {\n            return (mapping?.[String(row[columnId])] ?? DEFAULT_COMMON_AES)?.[field] as ValueType;\n        };\n    }\n    return () => aesItem as ValueType;\n}\n"],"names":["isInheritedAes","v","createAesGetter","legendInfo","aesItem","field","isContinuousAes","domain","range","scale","scaleSqrt","row","type","getContinuousColorScale","DEFAULT_DOT_AES","columnId","mapping","_a","DEFAULT_COMMON_AES"],"mappings":";;;;;;;;;;AAcA,SAASA,EAAeC,GAAgE;AACpF,SAAO,OAAOA,KAAM,YAAYA,MAAM,QAAQ,UAAUA,KAAK,WAAWA,KAAKA,EAAE,SAAS;AAC5F;AACO,SAASC,EACZC,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,GAAS;AACrB,YAAMV,IAAIQ,EAAM,OAAOE,EAAIP,EAAQ,WAAW,KAAK,CAAC,CAAC;AACrD,aAAO,KAAK,IAAI,GAAGH,CAAW;AAAA,IAClC;AAAA,EACJ;AACA,MAAIK,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,GAAS;AACrB,YAAMV,IAAIU,EAAIP,EAAQ,WAAW,KAAK;AACtC,aAAIH,MAAM,OACCa,EAAgB,QAEpBL,EAAM,OAAOR,CAAC,CAAC;AAAA,IAC1B;AAAA,EACJ;AACA,MAAID,EAAeI,CAAO,GAAG;AACzB,UAAMW,IAAWX,EAAQ,OACnBY,IAAUb,EAAWY,CAAQ,EAAE;AACrC,WAAO,SAASJ,GAAS;;AACrB,cAAQM,KAAAD,KAAA,gBAAAA,EAAU,OAAOL,EAAII,CAAQ,CAAC,OAAMG,MAApC,gBAAAD,EAA0DZ;AAAA,IACtE;AAAA,EACJ;AACA,SAAO,MAAMD;AACjB;"}
+{"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;"}

package/dist/scatterplot/utils/createLegendInfo.d.ts

@@ -1,14 +1,6 @@
 import { DataFrame } from '../../DataFrame';
+import { AesItem, Category, ColumnName, ScatterplotLayer, ScatterplotLegendInfo } from '../../types';
 import { ScatterplotSettingsImpl } from '../ScatterplotSettingsImpl';
-import { AesItem, ColumnName, ScatterplotLayer, ScatterplotLegendInfo } from '../../types';
-export declare function addPalettesToAesMapping(uniqueValues: string[], usedAesFromPalettes: Partial<Record<keyof AesItem, string[]>>, inheritedAes?: ScatterplotSettingsImpl['inheritedAes'][string]): Record<string, {
-    dotFill?: string | undefined;
-    dotShape?: "0" | "1" | "2" | "3" | "4" | "5" | "6" | "7" | "8" | "9" | "10" | "11" | "12" | "13" | "14" | "15" | "16" | "17" | "18" | "20" | "21" | "22" | "23" | "24" | "25" | undefined;
-    dotSize?: number | undefined;
-    lineWidth?: number | undefined;
-    lineShape?: "solid" | "dashed" | "dotted" | "dotdash" | "longdash" | "twodash" | undefined;
-    lineColor?: string | undefined;
-    fillColor?: string | undefined;
-}>;
+export declare function addPalettesToAesMapping(columnName: ColumnName, usedAesFromPalettes: Partial<Record<keyof AesItem, string[]>>, inheritedAes?: ScatterplotSettingsImpl['inheritedAes'][string]): (category: Category, field: keyof AesItem) => string | number | undefined;
 export declare function createLegendInfo(data: DataFrame, grouping: ColumnName[], layers: ScatterplotLayer[], inheritedAes: ScatterplotSettingsImpl['inheritedAes']): ScatterplotLegendInfo;
 //# sourceMappingURL=createLegendInfo.d.ts.map

package/dist/scatterplot/utils/createLegendInfo.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"createLegendInfo.d.ts","sourceRoot":"","sources":["../../../src/scatterplot/utils/createLegendInfo.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,iBAAiB,CAAC;AAC/C,OAAO,KAAK,EAAC,uBAAuB,EAAC,MAAM,4BAA4B,CAAC;AACxE,OAAO,KAAK,EACR,OAAO,EACP,UAAU,EAIV,gBAAgB,EAChB,qBAAqB,EACxB,MAAM,aAAa,CAAC;AAGrB,wBAAgB,uBAAuB,CACnC,YAAY,EAAE,MAAM,EAAE,EACtB,mBAAmB,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,EAC7D,YAAY,CAAC,EAAE,uBAAuB,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC;;;;;;;;GAoBjE;AAKD,wBAAgB,gBAAgB,CAC5B,IAAI,EAAE,SAAS,EACf,QAAQ,EAAE,UAAU,EAAE,EACtB,MAAM,EAAE,gBAAgB,EAAE,EAC1B,YAAY,EAAE,uBAAuB,CAAC,cAAc,CAAC,GACtD,qBAAqB,CA2CvB"}
+{"version":3,"file":"createLegendInfo.d.ts","sourceRoot":"","sources":["../../../src/scatterplot/utils/createLegendInfo.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EACR,OAAO,EACP,QAAQ,EACR,UAAU,EAGV,gBAAgB,EAChB,qBAAqB,EACxB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAyB1E,wBAAgB,uBAAuB,CACnC,UAAU,EAAE,UAAU,EACtB,mBAAmB,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,EAC7D,YAAY,CAAC,EAAE,uBAAuB,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,cAQ/B,QAAQ,SAAS,MAAM,OAAO,iCAchE;AAKD,wBAAgB,gBAAgB,CAC5B,IAAI,EAAE,SAAS,EACf,QAAQ,EAAE,UAAU,EAAE,EACtB,MAAM,EAAE,gBAAgB,EAAE,EAC1B,YAAY,EAAE,uBAAuB,CAAC,cAAc,CAAC,GACtD,qBAAqB,CA4CvB"}

package/dist/scatterplot/utils/createLegendInfo.js

@@ -1,50 +1,57 @@
-import { MAX_SHOWN_UNIQUE_VALUES_IN_LABELS as b } from "../../constants.js";
-import L from "../../node_modules/lodash/lodash.js";
-function C(o, l, p) {
-  const f = p ?? {};
-  return Object.keys(l).length ? o.reduce((r, e, a) => {
-    if (r[e] || (r[e] = {}), l.dotFill) {
-      const t = l.dotFill;
-      r[e].dotFill = t[a % (t == null ? void 0 : t.length)];
+import { MAX_SHOWN_UNIQUE_VALUES_IN_LABELS as v } from "../../constants.js";
+const h = /* @__PURE__ */ (() => {
+  const t = /* @__PURE__ */ new Map();
+  return {
+    clear: () => t.clear(),
+    getIndex: (g, o) => {
+      let e = t.get(g.value);
+      e === void 0 && (e = /* @__PURE__ */ new Map(), t.set(g.value, e));
+      let r = e.get(o);
+      return r === void 0 && (r = e.size, e.set(o, r)), r;
     }
-    if (l.lineColor) {
-      const t = l.lineColor;
-      r[e].lineColor = t[a % (t == null ? void 0 : t.length)];
-    }
-    return r;
-  }, f) : f;
+  };
+})();
+function C(t, c, u) {
+  h.clear();
+  const g = u && Object.keys(u).length > 0, o = c.dotFill, e = o && o.length > 0, r = c.lineColor, d = r && r.length > 0;
+  return (s, a) => {
+    var l;
+    const n = g ? (l = u[s]) == null ? void 0 : l[a] : void 0;
+    return a === "dotFill" && e ? o[h.getIndex(t, s) % o.length] ?? n : a === "lineColor" && d ? r[h.getIndex(t, s) % r.length] ?? n : n;
+  };
 }
-function S(o) {
-  return typeof o == "object" && "type" in o && o.type === "grouping";
+function I(t) {
+  return typeof t == "object" && "type" in t && t.type === "grouping";
 }
-function y(o, l, p, f) {
-  return l.reduce((r, e) => {
-    const a = (i) => String(i[e.value]), t = (i) => e.valueLabels ? String(i[e.valueLabels]) : a(i), d = /* @__PURE__ */ new Set(), h = {};
-    p.forEach((i) => {
-      i.aes && Object.entries(i.aes).forEach(([s, n]) => {
-        S(n) && n.value === e.value && (d.add(s), n.palette && (h[s] = n.palette));
+function L(t, c, u, g) {
+  return c.reduce((o, e) => {
+    const r = (n) => e.valueLabels ? t.getColumnValueCategory(e.valueLabels, n) : void 0, d = /* @__PURE__ */ new Set(), f = {};
+    u.forEach((n) => {
+      n.aes && Object.entries(n.aes).forEach(([l, i]) => {
+        I(i) && i.value === e.value && (d.add(l), f[l] = i.palette);
       });
     });
-    let u = {}, c = [];
-    if (d.has("dotFill")) {
-      const i = o.getColumnCategories(e.value);
-      if (i.length > b)
-        c = i, u = {};
-      else {
-        const s = L.uniqBy(o.rows, (n) => n[e.value]);
-        u = s.reduce((n, g) => (n[a(g)] = t(g), n), {}), c = s.map(a).sort((n, g) => u[n].localeCompare(u[g], "en", { numeric: !0 }));
-      }
+    let s = {}, a = [];
+    if (d.has("dotFill") || d.has("dotShape")) {
+      const n = t.getColumnCategories(e.value);
+      n.length < v ? (s = n.reduce((l, i) => {
+        const p = t.getColumnCategoryRowIndex(e.value, i);
+        return l[i] = r(p) ?? i, l;
+      }, s), a = n.sort((l, i) => {
+        var p;
+        return ((p = s[l]) == null ? void 0 : p.localeCompare(s[i] ?? "", "en", { numeric: !0 })) ?? 0;
+      })) : a = n;
     }
-    return r[e.value] = {
-      values: c,
+    return o[e.value] = {
+      values: a,
       usedAes: [...d],
-      aesMap: C(c, h, f[e.value]),
-      labels: u
-    }, r;
+      aesMap: C(e, f, g[e.value]),
+      labels: s
+    }, o;
   }, {});
 }
 export {
   C as addPalettesToAesMapping,
-  y as createLegendInfo
+  L 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 {ScatterplotSettingsImpl} from '../ScatterplotSettingsImpl';\nimport type {\n    AesItem,\n    ColumnName,\n    ContinuousAesFromColumn,\n    InheritAesScatterplot,\n    Row,\n    ScatterplotLayer,\n    ScatterplotLegendInfo,\n} from '../../types';\nimport lodash from 'lodash';\n\nexport function addPalettesToAesMapping(\n    uniqueValues: string[],\n    usedAesFromPalettes: Partial<Record<keyof AesItem, string[]>>,\n    inheritedAes?: ScatterplotSettingsImpl['inheritedAes'][string],\n) {\n    const result = inheritedAes ?? {};\n    if (!Object.keys(usedAesFromPalettes).length) {\n        return result;\n    }\n    return uniqueValues.reduce((res:Record<string, AesItem>, value, idx) => {\n        if (!res[value]) {\n            res[value] = {};\n        }\n        if (usedAesFromPalettes['dotFill']) {\n            const palette = usedAesFromPalettes['dotFill'];\n            res[value]['dotFill'] = palette[idx % palette?.length];\n        }\n        if (usedAesFromPalettes['lineColor']) {\n            const palette = usedAesFromPalettes['lineColor'];\n            res[value]['lineColor'] = palette[idx % palette?.length];\n        }\n        return res;\n    }, result);\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: ColumnName[],\n    layers: ScatterplotLayer[],\n    inheritedAes: ScatterplotSettingsImpl['inheritedAes']\n): ScatterplotLegendInfo {\n    return grouping.reduce((res: ScatterplotLegendInfo, column) => {\n        const getValue = (row: Row) => String(row[column.value]);\n        const getValueLabel = (row: Row) => (column.valueLabels ? String(row[column.valueLabels]) : getValue(row));\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) {\n                Object.entries(layer.aes).forEach(([key, value]) => {\n                    if (isInheritMapping(value) && value.value === column.value) {\n                        usedAes.add(key as keyof AesItem);\n                        if (value.palette) {\n                            usedAesFromPalettes[key as keyof AesItem] = value.palette;\n                        }\n                    }\n                });\n            }\n        });\n        let labels:Record<string, string> = {};\n        let values:string[] = [];\n        if (usedAes.has('dotFill')) {\n            const uniqueValues = data.getColumnCategories(column.value);\n            if (uniqueValues.length > MAX_SHOWN_UNIQUE_VALUES_IN_LABELS) {\n                values = uniqueValues;\n                labels = {};\n            } else {\n                const uniqRowsByGrouping = lodash.uniqBy(data.rows, row => row[column.value]);\n                labels = uniqRowsByGrouping.reduce((res: Record<string, string>, row) => {\n                    res[getValue(row)] = getValueLabel(row);\n                    return res;\n                }, {});\n                values = uniqRowsByGrouping.map(getValue).sort((a, b) => labels[a].localeCompare(labels[b], 'en', {numeric: true}));\n            }\n        }\n        res[column.value] = {\n            values,\n            usedAes: [...usedAes],\n            aesMap: addPalettesToAesMapping(values, usedAesFromPalettes, inheritedAes[column.value]),\n            labels,\n        };\n        return res;\n    }, {});\n}"],"names":["addPalettesToAesMapping","uniqueValues","usedAesFromPalettes","inheritedAes","result","res","value","idx","palette","isInheritMapping","createLegendInfo","data","grouping","layers","column","getValue","row","getValueLabel","usedAes","layer","key","labels","values","MAX_SHOWN_UNIQUE_VALUES_IN_LABELS","uniqRowsByGrouping","lodash","a","b"],"mappings":";;AAcO,SAASA,EACZC,GACAC,GACAC,GACF;AACE,QAAMC,IAASD,KAAgB,CAAA;AAC/B,SAAK,OAAO,KAAKD,CAAmB,EAAE,SAG/BD,EAAa,OAAO,CAACI,GAA6BC,GAAOC,MAAQ;AAIpE,QAHKF,EAAIC,CAAK,MACVD,EAAIC,CAAK,IAAI,CAAA,IAEbJ,EAAoB,SAAY;AAChC,YAAMM,IAAUN,EAAoB;AACpC,MAAAG,EAAIC,CAAK,EAAE,UAAaE,EAAQD,KAAMC,KAAA,gBAAAA,EAAS,OAAM;AAAA,IACzD;AACA,QAAIN,EAAoB,WAAc;AAClC,YAAMM,IAAUN,EAAoB;AACpC,MAAAG,EAAIC,CAAK,EAAE,YAAeE,EAAQD,KAAMC,KAAA,gBAAAA,EAAS,OAAM;AAAA,IAC3D;AACA,WAAOH;AAAA,EACX,GAAGD,CAAM,IAfEA;AAgBf;AAEA,SAASK,EAA8CH,GAA6F;AAChJ,SAAO,OAAOA,KAAU,YAAY,UAAUA,KAASA,EAAM,SAAS;AAC1E;AACO,SAASI,EACZC,GACAC,GACAC,GACAV,GACqB;AACrB,SAAOS,EAAS,OAAO,CAACP,GAA4BS,MAAW;AAC3D,UAAMC,IAAW,CAACC,MAAa,OAAOA,EAAIF,EAAO,KAAK,CAAC,GACjDG,IAAgB,CAACD,MAAcF,EAAO,cAAc,OAAOE,EAAIF,EAAO,WAAW,CAAC,IAAIC,EAASC,CAAG,GAElGE,wBAAc,IAAA,GACdhB,IAA+D,CAAA;AACrE,IAAAW,EAAO,QAAQ,CAAAM,MAAS;AACpB,MAAIA,EAAM,OACN,OAAO,QAAQA,EAAM,GAAG,EAAE,QAAQ,CAAC,CAACC,GAAKd,CAAK,MAAM;AAChD,QAAIG,EAAiBH,CAAK,KAAKA,EAAM,UAAUQ,EAAO,UAClDI,EAAQ,IAAIE,CAAoB,GAC5Bd,EAAM,YACNJ,EAAoBkB,CAAoB,IAAId,EAAM;AAAA,MAG9D,CAAC;AAAA,IAET,CAAC;AACD,QAAIe,IAAgC,CAAA,GAChCC,IAAkB,CAAA;AACtB,QAAIJ,EAAQ,IAAI,SAAS,GAAG;AACxB,YAAMjB,IAAeU,EAAK,oBAAoBG,EAAO,KAAK;AAC1D,UAAIb,EAAa,SAASsB;AACtB,QAAAD,IAASrB,GACToB,IAAS,CAAA;AAAA,WACN;AACH,cAAMG,IAAqBC,EAAO,OAAOd,EAAK,MAAM,CAAAK,MAAOA,EAAIF,EAAO,KAAK,CAAC;AAC5E,QAAAO,IAASG,EAAmB,OAAO,CAACnB,GAA6BW,OAC7DX,EAAIU,EAASC,CAAG,CAAC,IAAIC,EAAcD,CAAG,GAC/BX,IACR,CAAA,CAAE,GACLiB,IAASE,EAAmB,IAAIT,CAAQ,EAAE,KAAK,CAACW,GAAGC,MAAMN,EAAOK,CAAC,EAAE,cAAcL,EAAOM,CAAC,GAAG,MAAM,EAAC,SAAS,GAAA,CAAK,CAAC;AAAA,MACtH;AAAA,IACJ;AACA,WAAAtB,EAAIS,EAAO,KAAK,IAAI;AAAA,MAChB,QAAAQ;AAAA,MACA,SAAS,CAAC,GAAGJ,CAAO;AAAA,MACpB,QAAQlB,EAAwBsB,GAAQpB,GAAqBC,EAAaW,EAAO,KAAK,CAAC;AAAA,MACvF,QAAAO;AAAA,IAAA,GAEGhB;AAAA,EACX,GAAG,CAAA,CAAE;AACT;"}
+{"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 {\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?: ScatterplotSettingsImpl['inheritedAes'][string],\n) {\n    categoryIndexer.clear();\n    const hasIngeritedAes = 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 = hasIngeritedAes ? inheritedAes[category]?.[field] : undefined;\n        \n        if (field === 'dotFill' && hasDotFill) {\n            return dotFill[categoryIndexer.getIndex(columnName, category) % dotFill.length] ?? inheritedResult;\n        }\n        if (field === 'lineColor' && hasLineColor) {\n            return lineColor[categoryIndexer.getIndex(columnName, category) % lineColor.length] ?? inheritedResult;\n        }\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: ColumnName[],\n    layers: ScatterplotLayer[],\n    inheritedAes: ScatterplotSettingsImpl['inheritedAes']\n): ScatterplotLegendInfo {\n    return grouping.reduce((res: ScatterplotLegendInfo, column) => {\n        // TODO: remove String conversion\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')) {\n            const categories = data.getColumnCategories(column.value);\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                    return res;\n                }, labels);\n                values = categories.sort((a, b) => labels[a]?.localeCompare(labels[b] ?? '', 'en', {numeric: true}) ?? 0);\n            } else {\n                values = categories;\n            }\n        }\n        res[column.value] = {\n            values,\n            usedAes: [...usedAes],\n            aesMap: addPalettesToAesMapping(column, usedAesFromPalettes, inheritedAes[column.value]),\n            labels,\n        };\n        return res;\n    }, {});\n}"],"names":["categoryIndexer","map","columnName","category","categoryMap","index","addPalettesToAesMapping","usedAesFromPalettes","inheritedAes","hasIngeritedAes","dotFill","hasDotFill","lineColor","hasLineColor","field","inheritedResult","_a","isInheritMapping","value","createLegendInfo","data","grouping","layers","res","column","getValueLabel","usedAes","layer","key","labels","values","categories","MAX_SHOWN_UNIQUE_VALUES_IN_LABELS","a","b"],"mappings":";AAaA,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;AAcrD,SAboB,CAACT,GAAoBW,MAAyB;;AAC9D,UAAMC,IAAkBN,KAAkBO,IAAAR,EAAaL,CAAQ,MAArB,gBAAAa,EAAyBF,KAAS;AAE5E,WAAIA,MAAU,aAAaH,IAChBD,EAAQV,EAAgB,SAASE,GAAYC,CAAQ,IAAIO,EAAQ,MAAM,KAAKK,IAEnFD,MAAU,eAAeD,IAClBD,EAAUZ,EAAgB,SAASE,GAAYC,CAAQ,IAAIS,EAAU,MAAM,KAAKG,IAGpFA;AAAA,EACX;AAGJ;AAEA,SAASE,EAA8CC,GAA6F;AAChJ,SAAO,OAAOA,KAAU,YAAY,UAAUA,KAASA,EAAM,SAAS;AAC1E;AACO,SAASC,EACZC,GACAC,GACAC,GACAd,GACqB;AACrB,SAAOa,EAAS,OAAO,CAACE,GAA4BC,MAAW;AAE3D,UAAMC,IAAgB,CAACpB,MACZmB,EAAO,cACZJ,EAAK,uBAAuBI,EAAO,aAAanB,CAAK,IACrD,QAGAqB,wBAAc,IAAA,GACdnB,IAA+D,CAAA;AACrE,IAAAe,EAAO,QAAQ,CAAAK,MAAS;AACpB,MAAKA,EAAM,OAEX,OAAO,QAAQA,EAAM,GAAG,EAAE,QAAQ,CAAC,CAACC,GAAKV,CAAK,MAAM;AAChD,QAAMD,EAAiBC,CAAK,KAAKA,EAAM,UAAUM,EAAO,UAExDE,EAAQ,IAAIE,CAAoB,GAChCrB,EAAoBqB,CAAoB,IAAIV,EAAM;AAAA,MACtD,CAAC;AAAA,IACL,CAAC;AACD,QAAIW,IAAmC,CAAA,GACnCC,IAAqB,CAAA;AACzB,QAAIJ,EAAQ,IAAI,SAAS,KAAKA,EAAQ,IAAI,UAAU,GAAG;AACnD,YAAMK,IAAaX,EAAK,oBAAoBI,EAAO,KAAK;AACxD,MAAIO,EAAW,SAASC,KACpBH,IAASE,EAAW,OAAO,CAACR,GAAKpB,MAAa;AAC1C,cAAME,IAAQe,EAAK,0BAA0BI,EAAO,OAAOrB,CAAQ;AACnEoB,eAAAA,EAAIpB,CAAQ,IAAIsB,EAAcpB,CAAK,KAAKF,GACjCoB;AAAAA,MACX,GAAGM,CAAM,GACTC,IAASC,EAAW,KAAK,CAACE,GAAGC,MAAA;;AAAM,iBAAAlB,IAAAa,EAAOI,CAAC,MAAR,gBAAAjB,EAAW,cAAca,EAAOK,CAAC,KAAK,IAAI,MAAM,EAAC,SAAS,GAAA,OAAU;AAAA,OAAC,KAExGJ,IAASC;AAAA,IAEjB;AACA,WAAAR,EAAIC,EAAO,KAAK,IAAI;AAAA,MAChB,QAAAM;AAAA,MACA,SAAS,CAAC,GAAGJ,CAAO;AAAA,MACpB,QAAQpB,EAAwBkB,GAAQjB,GAAqBC,EAAagB,EAAO,KAAK,CAAC;AAAA,MACvF,QAAAK;AAAA,IAAA,GAEGN;AAAA,EACX,GAAG,CAAA,CAAE;AACT;"}

package/dist/scatterplot/utils/formatColumnValue.d.ts

@@ -0,0 +1,5 @@
+import { DataFrame } from '../../DataFrame';
+import { ColumnName } from '../../types';
+import { numberFormat } from '../../utils/numberFormat';
+export declare function formatColumnValue(data: DataFrame, rowIdx: number, column: ColumnName | null | undefined, numberFormatter?: typeof numberFormat): string | null;
+//# sourceMappingURL=formatColumnValue.d.ts.map

package/dist/scatterplot/utils/formatColumnValue.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"formatColumnValue.d.ts","sourceRoot":"","sources":["../../../src/scatterplot/utils/formatColumnValue.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,iBAAiB,CAAC;AAC/C,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAC,YAAY,EAAC,MAAM,0BAA0B,CAAC;AAEtD,wBAAgB,iBAAiB,CAAC,IAAI,EAAC,SAAS,EAAE,MAAM,EAAC,MAAM,EAAE,MAAM,EAAC,UAAU,GAAG,IAAI,GAAG,SAAS,EAAE,eAAe,sBAAe,iBAUpI"}

package/dist/scatterplot/utils/formatColumnValue.js

@@ -0,0 +1,11 @@
+import { numberFormat as l } from "../../utils/numberFormat.js";
+function m(u, t, r, n = l) {
+  if (!r)
+    return null;
+  const e = u.getColumnValue(r.valueLabels ?? r.value, t);
+  return !(typeof e == "number") || e === null ? e : n(e);
+}
+export {
+  m as formatColumnValue
+};
+//# sourceMappingURL=formatColumnValue.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"formatColumnValue.js","sources":["../../../src/scatterplot/utils/formatColumnValue.ts"],"sourcesContent":["import type {DataFrame} from '../../DataFrame';\nimport type {ColumnName} from '../../types';\nimport {numberFormat} from '../../utils/numberFormat';\n\nexport function formatColumnValue(data:DataFrame, rowIdx:number, column:ColumnName | null | undefined, numberFormatter = numberFormat) {\n    if (!column) {\n        return null;\n    }\n    const value = data.getColumnValue(column.valueLabels ?? column.value, rowIdx);\n    const isNumber = typeof value === 'number';\n    if (!isNumber || value === null) {\n        return value;\n    }\n    return numberFormatter(value);\n}"],"names":["formatColumnValue","data","rowIdx","column","numberFormatter","numberFormat","value"],"mappings":";AAIO,SAASA,EAAkBC,GAAgBC,GAAeC,GAAsCC,IAAkBC,GAAc;AACnI,MAAI,CAACF;AACD,WAAO;AAEX,QAAMG,IAAQL,EAAK,eAAeE,EAAO,eAAeA,EAAO,OAAOD,CAAM;AAE5E,SAAI,EADa,OAAOI,KAAU,aACjBA,MAAU,OAChBA,IAEJF,EAAgBE,CAAK;AAChC;"}

package/dist/scatterplot/utils/getVisibleLabels.d.ts

@@ -2,17 +2,15 @@ import { Dot } from '../dots';
 type XPosition = 'right' | 'middle' | 'left';
 type YPosition = 'top' | 'middle' | 'bottom';
 export type Label = {
+    x: number;
+    y: number;
+    dot: Dot;
     name: string;
     width: number;
     height: number;
-    x: number;
-    y: number;
     padding: number;
     xPosition: XPosition;
     yPosition: YPosition;
-    bbox: LabelBBox;
-    outerBBox: LabelBBox;
-    dot: Dot;
 };
 export type LabelBBox = {
     minX: number;
@@ -20,9 +18,8 @@ export type LabelBBox = {
     minY: number;
     maxY: number;
 };
-export declare function getLabelBBox(label: Label): LabelBBox;
-export declare function getLabelOuterBBox(label: Label, vOffset: number, hOffset: number): LabelBBox;
-export declare function updateLabelsBBoxes(labels: Label[], vOffset?: number, hOffset?: number): void;
-export declare function getFilteredLabels(labels: Label[], width: number, height: number, vOffset?: number, hOffset?: number): Label[];
+export declare function getLabelMinX(label: Label): number;
+export declare function getLabelMinY(label: Label): number;
+export declare function createLabelPositioner(chartW: number, chartH: number, yOffset?: number, xOffset?: number): (x: number, y: number, w: number, h: number, padding: number) => null | [XPosition, YPosition];
 export {};
 //# sourceMappingURL=getVisibleLabels.d.ts.map

package/dist/scatterplot/utils/getVisibleLabels.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"getVisibleLabels.d.ts","sourceRoot":"","sources":["../../../src/scatterplot/utils/getVisibleLabels.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,GAAG,EAAC,MAAM,SAAS,CAAC;AAGjC,KAAK,SAAS,GAAG,OAAO,GAAG,QAAQ,GAAG,MAAM,CAAC;AAC7C,KAAK,SAAS,GAAG,KAAK,GAAG,QAAQ,GAAG,QAAQ,CAAC;AAC7C,MAAM,MAAM,KAAK,GAAG;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;IACV,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,SAAS,CAAC;IACrB,SAAS,EAAE,SAAS,CAAC;IACrB,IAAI,EAAE,SAAS,CAAC;IAChB,SAAS,EAAE,SAAS,CAAC;IACrB,GAAG,EAAE,GAAG,CAAC;CACZ,CAAC;AACF,MAAM,MAAM,SAAS,GAAG;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CAChB,CAAA;AA4BD,wBAAgB,YAAY,CAAC,KAAK,EAAE,KAAK,GAAE,SAAS,CAOnD;AACD,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAE,SAAS,CAO1F;AAQD,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,OAAO,SAAe,EAAE,OAAO,SAAe,QAKjG;AACD,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,SAAe,EAAE,OAAO,SAAe,WA4B/H"}
+{"version":3,"file":"getVisibleLabels.d.ts","sourceRoot":"","sources":["../../../src/scatterplot/utils/getVisibleLabels.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,SAAS,CAAC;AAEnC,KAAK,SAAS,GAAG,OAAO,GAAG,QAAQ,GAAG,MAAM,CAAC;AAC7C,KAAK,SAAS,GAAG,KAAK,GAAG,QAAQ,GAAG,QAAQ,CAAC;AAC7C,MAAM,MAAM,KAAK,GAAG;IAChB,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;IACV,GAAG,EAAE,GAAG,CAAC;IACT,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,SAAS,CAAC;IACrB,SAAS,EAAE,SAAS,CAAC;CACxB,CAAC;AACF,MAAM,MAAM,SAAS,GAAG;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CAChB,CAAA;AAkDD,wBAAgB,YAAY,CAAC,KAAK,EAAE,KAAK,UAExC;AACD,wBAAgB,YAAY,CAAC,KAAK,EAAE,KAAK,UAExC;AAcD,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,SAAe,EAAE,OAAO,SAAe,IAIxG,GAAG,MAAM,EAAE,GAAG,MAAM,EAAE,GAAG,MAAM,EAAE,GAAG,MAAM,EAAE,SAAS,MAAM,KAAG,IAAI,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAYtG"}

package/dist/scatterplot/utils/getVisibleLabels.js

@@ -1,60 +1,50 @@
-import P from "../../node_modules/rbush/index.js";
-function u(t) {
-  const o = t.xPosition !== "middle" && t.yPosition !== "middle", n = t.padding * (o ? 0.7 : 1);
-  return t.xPosition === "right" ? t.x + n : t.xPosition === "middle" ? t.x - t.width / 2 : t.x - t.width - n;
-}
-function m(t) {
-  const o = t.xPosition !== "middle" && t.yPosition !== "middle", n = t.padding * (o ? 0.7 : 1);
-  return t.yPosition === "middle" ? t.y - t.height / 2 : t.yPosition === "top" ? t.y - n - t.height : t.y + n;
-}
-function y(t, o) {
-  return u(t) - o;
-}
-function X(t, o) {
-  return m(t) - o;
-}
-function x(t) {
-  const o = u(t), n = o + t.width, i = m(t), d = i + t.height;
-  return { minX: o, maxX: n, minY: i, maxY: d };
-}
-function g(t, o, n) {
-  const i = y(t, n), d = u(t) + t.width + n, s = X(t, o), e = s + t.height + 2 * o;
-  return { minX: i, maxX: d, minY: s, maxY: e };
-}
-function Y(t, o, n) {
-  return t.minX >= 0 && t.maxX <= o && t.minY >= 0 && t.maxY <= n;
-}
-const c = 2;
-function w(t, o = c, n = c) {
-  t.forEach((i) => {
-    i.bbox = x(i), i.outerBBox = g(i, o, n);
-  });
-}
-function E(t, o, n, i = c, d = c) {
-  const s = [], e = new P(), B = [
-    ["middle", "right"],
-    ["top", "right"],
-    ["bottom", "right"],
-    ["middle", "left"],
-    ["top", "left"],
-    ["bottom", "left"],
-    ["top", "middle"],
-    ["bottom", "middle"]
-  ];
-  for (const h of t)
-    for (const [f, p] of B) {
-      const r = { ...h, yPosition: f, xPosition: p };
-      if (r.bbox = x(r), r.outerBBox = g(r, i, d), Y(r.outerBBox, o, n) && !e.collides(r.outerBBox)) {
-        s.push(r), e.insert(r.outerBBox);
-        break;
-      }
-    }
-  return s;
+import L from "../../node_modules/rbush/index.js";
+function d(n, r, t, o, i) {
+  return o !== "middle" && i !== "middle" && (t *= 0.7), o === "right" ? n + t : o === "middle" ? n - r / 2 : n - r - t;
+}
+function B(n, r, t, o, i) {
+  return o !== "middle" && i !== "middle" && (t *= 0.7), i === "middle" ? n - r / 2 : i === "top" ? n - t - r : n + t;
+}
+function l(n, r, t, o, i, e) {
+  return d(n, r, t, o, i) - e;
+}
+function x(n, r, t, o, i, e) {
+  return B(n, r, t, o, i) - e;
+}
+function p(n, r, t, o, i, e, u, c, s, m = { minX: 0, maxX: 0, minY: 0, maxY: 0 }) {
+  return m.minX = l(n, t, i, e, u, s), m.maxX = d(n, t, i, e, u) + t + s, m.minY = x(r, o, i, e, u, c), m.maxY = m.minY + o + 2 * c, m;
+}
+function C(n, r, t) {
+  return n.minX >= 0 && n.maxX <= r && n.minY >= 0 && n.maxY <= t;
+}
+function E(n) {
+  return d(n.x, n.width, n.padding, n.xPosition, n.yPosition);
+}
+function F(n) {
+  return B(n.y, n.height, n.padding, n.xPosition, n.yPosition);
+}
+const O = [
+  ["middle", "right"],
+  ["top", "right"],
+  ["bottom", "right"],
+  ["middle", "left"],
+  ["top", "left"],
+  ["bottom", "left"],
+  ["top", "middle"],
+  ["bottom", "middle"]
+], Y = 2, M = { minX: 0, maxX: 0, minY: 0, maxY: 0 };
+function _(n, r, t = Y, o = Y) {
+  const i = M, e = new L();
+  return (u, c, s, m, g) => {
+    for (const [f, X] of O)
+      if (p(u, c, s, m, g, X, f, t, o, i), C(i, n, r) && !e.collides(i))
+        return e.insert({ ...i }), [X, f];
+    return null;
+  };
 }
 export {
-  E as getFilteredLabels,
-  x as getLabelBBox,
-  g as getLabelOuterBBox,
-  w as updateLabelsBBoxes
+  _ as createLabelPositioner,
+  E as getLabelMinX,
+  F as getLabelMinY
 };
 //# sourceMappingURL=getVisibleLabels.js.map