@uwdata/mosaic-sql 0.17.0 → 0.18.0

package/dist/src/load/sql-from.js

@@ -0,0 +1,29 @@
+import { asLiteral } from '../util/ast.js';
+/**
+ * Create a SQL query that embeds the given data for loading.
+ * @param data The dataset as an array of objects.
+ * @param options Loading options.
+ * @param {string[]|object} [options.columns] The columns to include.
+ *   If not specified, the keys of the first data object are used.
+ * @returns {string} SQL query string to load data.
+ */
+export function sqlFrom(data, { columns = Object.keys(data?.[0] || {}) } = {}) {
+    let keys = [];
+    if (Array.isArray(columns)) {
+        keys = columns;
+        columns = keys.reduce((m, k) => (m[k] = k, m), {});
+    }
+    else if (columns) {
+        keys = Object.keys(columns);
+    }
+    if (!keys.length) {
+        throw new Error('Can not create table from empty column set.');
+    }
+    const subq = [];
+    for (const datum of data) {
+        const sel = keys.map(k => `${asLiteral(datum[k])} AS "${columns[k]}"`);
+        subq.push(`(SELECT ${sel.join(', ')})`);
+    }
+    return subq.join(' UNION ALL ');
+}
+//# sourceMappingURL=sql-from.js.map

package/dist/src/load/sql-from.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql-from.js","sourceRoot":"","sources":["../../../src/load/sql-from.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAM3C;;;;;;;GAOG;AACH,MAAM,UAAU,OAAO,CACrB,IAAiC,EACjC,EACE,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KACpB,EAAE;IAEtB,IAAI,IAAI,GAAa,EAAE,CAAC;IACxB,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,IAAI,GAAG,OAAO,CAAC;QACf,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,EAAE,EAA4B,CAAC,CAAC;IAC/E,CAAC;SAAM,IAAI,OAAO,EAAE,CAAC;QACnB,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;IACjE,CAAC;IACD,MAAM,IAAI,GAAG,EAAE,CAAC;IAChB,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACvE,IAAI,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC1C,CAAC;IACD,OAAO,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;AAClC,CAAC"}

package/dist/src/transforms/bin-1d.d.ts

@@ -0,0 +1,16 @@
+import type { ExprValue } from '../types.js';
+/**
+ * Compute binned values over a one-dimensional extent.
+ * @param x The expression to bin.
+ *  The expression must return numeric values. For example, to bin
+ *  datetime values, the input expression might map them to numeric
+ *  values such as milliseconds since the epoch.
+ * @param lo The low value of the bin extent.
+ * @param hi The high value of the bin extent.
+ * @param bins The integer number of bins to use within the
+ *  defined binning extent.
+ * @param reverse Flag indicating if bins should be
+ *  produced in reverse order from *hi* to *lo* (default `false`).
+ */
+export declare function bin1d(x: ExprValue, lo: number, hi: number, bins: number, reverse?: boolean): import("../index.js").BinaryOpNode;
+//# sourceMappingURL=bin-1d.d.ts.map

package/dist/src/transforms/bin-1d.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-1d.d.ts","sourceRoot":"","sources":["../../../src/transforms/bin-1d.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAI7C;;;;;;;;;;;;GAYG;AACH,wBAAgB,KAAK,CACnB,CAAC,EAAE,SAAS,EACZ,EAAE,EAAE,MAAM,EACV,EAAE,EAAE,MAAM,EACV,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,OAAe,sCAKzB"}

package/dist/src/transforms/bin-1d.js

@@ -0,0 +1,21 @@
+import { float64 } from '../functions/cast.js';
+import { mul, sub } from '../functions/operators.js';
+/**
+ * Compute binned values over a one-dimensional extent.
+ * @param x The expression to bin.
+ *  The expression must return numeric values. For example, to bin
+ *  datetime values, the input expression might map them to numeric
+ *  values such as milliseconds since the epoch.
+ * @param lo The low value of the bin extent.
+ * @param hi The high value of the bin extent.
+ * @param bins The integer number of bins to use within the
+ *  defined binning extent.
+ * @param reverse Flag indicating if bins should be
+ *  produced in reverse order from *hi* to *lo* (default `false`).
+ */
+export function bin1d(x, lo, hi, bins, reverse = false) {
+    const diff = reverse ? sub(hi, float64(x)) : sub(float64(x), lo);
+    const scale = hi === lo ? 0 : bins / (hi - lo);
+    return scale ? mul(diff, float64(scale)) : diff;
+}
+//# sourceMappingURL=bin-1d.js.map

package/dist/src/transforms/bin-1d.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-1d.js","sourceRoot":"","sources":["../../../src/transforms/bin-1d.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AAErD;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,KAAK,CACnB,CAAY,EACZ,EAAU,EACV,EAAU,EACV,IAAY,EACZ,UAAmB,KAAK;IAExB,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACjE,MAAM,KAAK,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/C,OAAO,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AAClD,CAAC"}

package/dist/src/transforms/bin-2d.d.ts

@@ -0,0 +1,19 @@
+import type { SelectQuery } from '../ast/query.js';
+import type { ExprValue } from '../types.js';
+/**
+ * Perform aggregation over a binned 2D domain. This method takes expressions
+ * for the (non-truncated) x and y bin values; these expressions should be
+ * in units of grid indices, but can contain fractional components. The
+ * resulting query performs grouping and aggregation over the binned domain,
+ * and uses a 2D integer bin index of the form (xbin + num_xbins * ybin).
+ * @param q The input query. The FROM and WHERE clauses should
+ *  be added to the query separately, either before or after this method.
+ * @param xp The x bin expression.
+ * @param yp The y bin expression.
+ * @param aggs Named aggregate expressions over bins.
+ * @param {xn The number of bins along the x dimension
+ * @param groupby Group by expressions.
+ * @returns The input query, with binning expressions added.
+ */
+export declare function bin2d(q: SelectQuery, xp: ExprValue, yp: ExprValue, aggs: Record<string, ExprValue>, xn: number, groupby: string[]): SelectQuery;
+//# sourceMappingURL=bin-2d.d.ts.map

package/dist/src/transforms/bin-2d.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-2d.d.ts","sourceRoot":"","sources":["../../../src/transforms/bin-2d.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAK7C;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,KAAK,CACnB,CAAC,EAAE,WAAW,EACd,EAAE,EAAE,SAAS,EACb,EAAE,EAAE,SAAS,EACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,EAC/B,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,MAAM,EAAE,eAQlB"}

package/dist/src/transforms/bin-2d.js

@@ -0,0 +1,27 @@
+import { int32 } from '../functions/cast.js';
+import { floor } from '../functions/numeric.js';
+import { add, mul } from '../functions/operators.js';
+/**
+ * Perform aggregation over a binned 2D domain. This method takes expressions
+ * for the (non-truncated) x and y bin values; these expressions should be
+ * in units of grid indices, but can contain fractional components. The
+ * resulting query performs grouping and aggregation over the binned domain,
+ * and uses a 2D integer bin index of the form (xbin + num_xbins * ybin).
+ * @param q The input query. The FROM and WHERE clauses should
+ *  be added to the query separately, either before or after this method.
+ * @param xp The x bin expression.
+ * @param yp The y bin expression.
+ * @param aggs Named aggregate expressions over bins.
+ * @param {xn The number of bins along the x dimension
+ * @param groupby Group by expressions.
+ * @returns The input query, with binning expressions added.
+ */
+export function bin2d(q, xp, yp, aggs, xn, groupby) {
+    return q
+        .select({
+        index: add(int32(floor(xp)), mul(int32(floor(yp)), xn)),
+        ...aggs
+    })
+        .groupby('index', groupby);
+}
+//# sourceMappingURL=bin-2d.js.map

package/dist/src/transforms/bin-2d.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-2d.js","sourceRoot":"","sources":["../../../src/transforms/bin-2d.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAChD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AAErD;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,KAAK,CACnB,CAAc,EACd,EAAa,EACb,EAAa,EACb,IAA+B,EAC/B,EAAU,EACV,OAAiB;IAEjB,OAAO,CAAC;SACL,MAAM,CAAC;QACN,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACvD,GAAG,IAAI;KACR,CAAC;SACD,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;AAC/B,CAAC"}

package/dist/src/transforms/bin-date.d.ts

@@ -0,0 +1,31 @@
+import type { ExprNode } from '../ast/node.js';
+import type { ExprValue } from '../types.js';
+import { type DateTimeValue, type TimeUnit } from './util/time-interval.js';
+export interface BinDateOptions {
+    /**
+     * A string indicating a time interval unit,
+     * such as 'year', 'day', or 'hour'.
+     */
+    interval?: TimeUnit;
+    /**
+     * The number of time interval steps to take, such as 2 years or 3 months.
+     */
+    step?: number;
+    /**
+     * The number of bin steps (default 0) by which to offset the result.
+     */
+    offset?: number;
+    /**
+     * The desired number of binning steps. This value is a hint,
+     * it does not guarantee an exact number of steps.
+     */
+    steps?: number;
+}
+/**
+ * Return a SQL expression for date/time bins.
+ * @param field The column or expression to bin.
+ * @param extent The min/max extent over which to bin.
+ * @param options Datetime binning options.
+ */
+export declare function binDate(field: ExprValue, extent: [DateTimeValue, DateTimeValue], options?: BinDateOptions): ExprNode;
+//# sourceMappingURL=bin-date.d.ts.map

package/dist/src/transforms/bin-date.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-date.d.ts","sourceRoot":"","sources":["../../../src/transforms/bin-date.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAI7C,OAAO,EAAgB,KAAK,aAAa,EAAE,KAAK,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAE1F,MAAM,WAAW,cAAc;IAC7B;;;OAGG;IACH,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;;;;GAKG;AACH,wBAAgB,OAAO,CACrB,KAAK,EAAE,SAAS,EAChB,MAAM,EAAE,CAAC,aAAa,EAAE,aAAa,CAAC,EACtC,OAAO,GAAE,cAAmB,GAC3B,QAAQ,CASV"}

package/dist/src/transforms/bin-date.js

@@ -0,0 +1,20 @@
+import { dateBin } from '../functions/datetime.js';
+import { interval } from '../functions/interval.js';
+import { add } from '../functions/operators.js';
+import { timeInterval } from './util/time-interval.js';
+/**
+ * Return a SQL expression for date/time bins.
+ * @param field The column or expression to bin.
+ * @param extent The min/max extent over which to bin.
+ * @param options Datetime binning options.
+ */
+export function binDate(field, extent, options = {}) {
+    const { offset = 0 } = options;
+    // use interval if provided, otherwise determine from extent
+    const { unit, step = 1 } = options.interval
+        ? { unit: options.interval, step: options.step }
+        : timeInterval(extent[0], extent[1], options.steps || 40);
+    const bin = dateBin(field, unit, step);
+    return offset ? add(bin, interval(unit, offset * step)) : bin;
+}
+//# sourceMappingURL=bin-date.js.map

package/dist/src/transforms/bin-date.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-date.js","sourceRoot":"","sources":["../../../src/transforms/bin-date.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AAChD,OAAO,EAAE,YAAY,EAAqC,MAAM,yBAAyB,CAAC;AAuB1F;;;;;GAKG;AACH,MAAM,UAAU,OAAO,CACrB,KAAgB,EAChB,MAAsC,EACtC,UAA0B,EAAE;IAE5B,MAAM,EAAE,MAAM,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IAE/B,4DAA4D;IAC5D,MAAM,EAAE,IAAI,EAAE,IAAI,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC,QAAQ;QACzC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE;QAChD,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAC5D,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACvC,OAAO,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;AAChE,CAAC"}

package/dist/src/transforms/bin-histogram.d.ts

@@ -0,0 +1,37 @@
+import type { ExprNode } from '../ast/node.js';
+import type { ExprValue } from '../types.js';
+import { Scale } from './scales.js';
+export interface BinHistogramOptions {
+    /** An exact binning step to use. */
+    step?: number;
+    /**
+     * The desired number of binning steps. This value is a hint,
+     * it does not guarantee an exact number of steps.
+     */
+    steps?: number;
+    /**
+     * A minimum binning step value. No generated
+     * step can be less than this value.
+     */
+    minstep?: number;
+    /**
+     * A boolean flag (default true) indicating if bin extents should be
+     * snapped to "nice" numbers such as multiples of 5 or 10.
+     */
+    nice?: boolean;
+    /**
+     * The number of bin steps (default 0) by which to offset the result.
+     */
+    offset?: number;
+}
+/**
+ * Return a SQL expression for histogram bins.
+ * @param field The column or expression to bin.
+ * @param extent The min/max extent over which to bin.
+ * @param options Binning options.
+ * @param transform Scale transforms to apply to create
+ *  (potentially non-linear) binning intervals.
+ * @returns The resulting SQL expression
+ */
+export declare function binHistogram(field: ExprValue, extent: [number, number], options?: BinHistogramOptions, transform?: Scale<number>): ExprNode;
+//# sourceMappingURL=bin-histogram.d.ts.map

package/dist/src/transforms/bin-histogram.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-histogram.d.ts","sourceRoot":"","sources":["../../../src/transforms/bin-histogram.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAK7C,OAAO,EAAE,KAAK,EAAkB,MAAM,aAAa,CAAC;AAEpD,MAAM,WAAW,mBAAmB;IAClC,oCAAoC;IACpC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;;OAGG;IACH,IAAI,CAAC,EAAE,OAAO,CAAC;IACf;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;;;;;;GAQG;AACH,wBAAgB,YAAY,CAC1B,KAAK,EAAE,SAAS,EAChB,MAAM,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,EACxB,OAAO,GAAE,mBAAwB,EACjC,SAAS,GAAE,KAAK,CAAC,MAAM,CAAuC,GAC7D,QAAQ,CAcV"}

package/dist/src/transforms/bin-histogram.js

@@ -0,0 +1,32 @@
+import { float64 } from '../functions/cast.js';
+import { floor } from '../functions/numeric.js';
+import { add, div, mul, sub } from '../functions/operators.js';
+import { binSpec } from './util/bin-step.js';
+import { scaleTransform } from './scales.js';
+/**
+ * Return a SQL expression for histogram bins.
+ * @param field The column or expression to bin.
+ * @param extent The min/max extent over which to bin.
+ * @param options Binning options.
+ * @param transform Scale transforms to apply to create
+ *  (potentially non-linear) binning intervals.
+ * @returns The resulting SQL expression
+ */
+export function binHistogram(field, extent, options = {}, transform = scaleTransform({ type: 'linear' })) {
+    const [min, max] = extent;
+    const { offset = 0 } = options;
+    const { apply, sqlApply, sqlInvert } = transform;
+    const b = binSpec(apply(min), apply(max), options);
+    const col = sqlApply(field);
+    const alpha = (b.max - b.min) / b.steps;
+    let expr = b.min === 0 ? col : sub(col, b.min);
+    if (alpha !== 1)
+        expr = div(expr, float64(alpha));
+    expr = floor(offset ? add(offset, expr) : expr);
+    if (alpha !== 1)
+        expr = mul(alpha, expr);
+    if (b.min !== 0)
+        expr = add(b.min, expr);
+    return sqlInvert(expr);
+}
+//# sourceMappingURL=bin-histogram.js.map

package/dist/src/transforms/bin-histogram.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-histogram.js","sourceRoot":"","sources":["../../../src/transforms/bin-histogram.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAChD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAS,cAAc,EAAE,MAAM,aAAa,CAAC;AA0BpD;;;;;;;;GAQG;AACH,MAAM,UAAU,YAAY,CAC1B,KAAgB,EAChB,MAAwB,EACxB,UAA+B,EAAE,EACjC,YAA2B,cAAc,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAE;IAE9D,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC;IAC1B,MAAM,EAAE,MAAM,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IAC/B,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC;IACjD,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC5B,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC;IAExC,IAAI,IAAI,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;IAC/C,IAAI,KAAK,KAAK,CAAC;QAAE,IAAI,GAAG,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;IAClD,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAChD,IAAI,KAAK,KAAK,CAAC;QAAE,IAAI,GAAG,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACzC,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC;QAAE,IAAI,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACzC,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC;AACzB,CAAC"}

package/dist/src/transforms/bin-linear-1d.d.ts

@@ -0,0 +1,11 @@
+import type { SelectQuery } from '../ast/query.js';
+import type { ExprValue } from '../types.js';
+/**
+ * Perform linear binning in one dimension.
+ * @param  query The base query to bin.
+ * @param x The expression to bin.
+ * @param weight The expression to weight by.
+ * @param groupby Group by expressions.
+ */
+export declare function binLinear1d(query: SelectQuery, x: ExprValue, weight?: ExprValue | undefined, groupby?: string[]): SelectQuery;
+//# sourceMappingURL=bin-linear-1d.d.ts.map

package/dist/src/transforms/bin-linear-1d.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-linear-1d.d.ts","sourceRoot":"","sources":["../../../src/transforms/bin-linear-1d.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAO7C;;;;;;GAMG;AACH,wBAAgB,WAAW,CACzB,KAAK,EAAE,WAAW,EAClB,CAAC,EAAE,SAAS,EACZ,MAAM,GAAE,SAAS,GAAG,SAAqB,EACzC,OAAO,GAAE,MAAM,EAAO,eAevB"}

package/dist/src/transforms/bin-linear-1d.js

@@ -0,0 +1,25 @@
+import { Query } from '../ast/query.js';
+import { sum } from '../functions/aggregate.js';
+import { int32 } from '../functions/cast.js';
+import { floor } from '../functions/numeric.js';
+import { add, mul, neq, sub } from '../functions/operators.js';
+/**
+ * Perform linear binning in one dimension.
+ * @param  query The base query to bin.
+ * @param x The expression to bin.
+ * @param weight The expression to weight by.
+ * @param groupby Group by expressions.
+ */
+export function binLinear1d(query, x, weight = undefined, groupby = []) {
+    const w = weight
+        ? ((x) => mul(x, weight))
+        : ((x) => x);
+    const p0 = floor(x);
+    const p1 = add(p0, 1);
+    return Query
+        .from(Query.unionAll(query.clone().select({ i: int32(p0), w: w(sub(p1, x)) }), query.clone().select({ i: int32(p1), w: w(sub(x, p0)) })))
+        .select({ index: 'i', density: sum('w') }, groupby)
+        .groupby('index', groupby)
+        .having(neq('density', 0));
+}
+//# sourceMappingURL=bin-linear-1d.js.map

package/dist/src/transforms/bin-linear-1d.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-linear-1d.js","sourceRoot":"","sources":["../../../src/transforms/bin-linear-1d.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AAChD,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAChD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AAE/D;;;;;;GAMG;AACH,MAAM,UAAU,WAAW,CACzB,KAAkB,EAClB,CAAY,EACZ,SAAgC,SAAS,EACzC,UAAoB,EAAE;IAEtB,MAAM,CAAC,GAAG,MAAM;QACd,CAAC,CAAC,CAAC,CAAC,CAAW,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC,CAAC,CAAW,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACzB,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IACpB,MAAM,EAAE,GAAG,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IACtB,OAAO,KAAK;SACT,IAAI,CAAC,KAAK,CAAC,QAAQ,CAClB,KAAK,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EACxD,KAAK,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,CACzD,CAAC;SACD,MAAM,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC;SAClD,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC;SACzB,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;AAC/B,CAAC"}

package/dist/src/transforms/bin-linear-2d.d.ts

@@ -0,0 +1,20 @@
+import type { SelectQuery } from '../ast/query.js';
+import type { ExprValue } from '../types.js';
+/**
+ * Compute densities over a 2D domain using linear binning. The weight of
+ * each data point is linearly distributed over adjacent bins, providing
+ * a better base for subsequent kernel density estimation. This method takes
+ * expressions for the (non-truncated) x and y bin values; these expressions
+ * should be in units of grid indices, but can contain fractional components.
+ * @param q The input query. The FROM and WHERE clauses should
+ *  be added to the query separately, before this method is invoked.
+ * @param xp The x grid bin expression
+ * @param yp The y grid bin expression
+ * @param weight Point weights.
+ * @param xn The number of x grid bins.
+ * @param groupby Group by expressions.
+ * @returns A linear binning query for bin `index` and aggregate `density`
+ *  columns, in addition to any group by expressions.
+ */
+export declare function binLinear2d(q: SelectQuery, xp: ExprValue, yp: ExprValue, weight: ExprValue | undefined, xn: number, groupby?: string[]): SelectQuery;
+//# sourceMappingURL=bin-linear-2d.d.ts.map

package/dist/src/transforms/bin-linear-2d.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-linear-2d.d.ts","sourceRoot":"","sources":["../../../src/transforms/bin-linear-2d.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAQ7C;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,WAAW,CACzB,CAAC,EAAE,WAAW,EACd,EAAE,EAAE,SAAS,EACb,EAAE,EAAE,SAAS,EACb,MAAM,EAAE,SAAS,GAAG,SAAS,EAC7B,EAAE,EAAE,MAAM,EACV,OAAO,GAAE,MAAM,EAAO,eA8BvB"}

package/dist/src/transforms/bin-linear-2d.js

@@ -0,0 +1,49 @@
+import { Query } from '../ast/query.js';
+import { sum } from '../functions/aggregate.js';
+import { int32 } from '../functions/cast.js';
+import { floor } from '../functions/numeric.js';
+import { add, mul, neq, sub } from '../functions/operators.js';
+import { identity } from '../util/identity.js';
+/**
+ * Compute densities over a 2D domain using linear binning. The weight of
+ * each data point is linearly distributed over adjacent bins, providing
+ * a better base for subsequent kernel density estimation. This method takes
+ * expressions for the (non-truncated) x and y bin values; these expressions
+ * should be in units of grid indices, but can contain fractional components.
+ * @param q The input query. The FROM and WHERE clauses should
+ *  be added to the query separately, before this method is invoked.
+ * @param xp The x grid bin expression
+ * @param yp The y grid bin expression
+ * @param weight Point weights.
+ * @param xn The number of x grid bins.
+ * @param groupby Group by expressions.
+ * @returns A linear binning query for bin `index` and aggregate `density`
+ *  columns, in addition to any group by expressions.
+ */
+export function binLinear2d(q, xp, yp, weight, xn, groupby = []) {
+    const w = weight ? (x) => mul(x, weight) : identity;
+    const subq = (i, w) => q.clone().select({ xp, yp, i, w });
+    const index = (x, y) => add(x, mul(y, xn));
+    const xu = int32(floor(xp));
+    const yu = int32(floor(yp));
+    const xv = add(xu, 1);
+    const yv = add(yu, 1);
+    const xpu = sub(xp, xu);
+    const xvp = sub(xv, xp);
+    const ypu = sub(yp, yu);
+    const yvp = sub(yv, yp);
+    return Query
+        .from(Query.unionAll(
+    // grid[xu + yu * xn] += (xv - xp) * (yv - yp) * wi
+    subq(index(xu, yu), w(mul(xvp, yvp))), 
+    // grid[xu + yv * xn] += (xv - xp) * (yp - yu) * wi
+    subq(index(xu, yv), w(mul(xvp, ypu))), 
+    // grid[xv + yu * xn] += (xp - xu) * (yv - yp) * wi
+    subq(index(xv, yu), w(mul(xpu, yvp))), 
+    // grid[xv + yv * xn] += (xp - xu) * (yp - yu) * wi
+    subq(index(xv, yv), w(mul(xpu, ypu)))))
+        .select({ index: 'i', density: sum('w') }, groupby)
+        .groupby('index', groupby)
+        .having(neq('density', 0));
+}
+//# sourceMappingURL=bin-linear-2d.js.map

package/dist/src/transforms/bin-linear-2d.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bin-linear-2d.js","sourceRoot":"","sources":["../../../src/transforms/bin-linear-2d.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AAChD,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAChD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAE/C;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,WAAW,CACzB,CAAc,EACd,EAAa,EACb,EAAa,EACb,MAA6B,EAC7B,EAAU,EACV,UAAoB,EAAE;IAEtB,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,CAAW,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IAE9D,MAAM,IAAI,GAAG,CAAC,CAAY,EAAE,CAAY,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAChF,MAAM,KAAK,GAAG,CAAC,CAAY,EAAE,CAAY,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEjE,MAAM,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5B,MAAM,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5B,MAAM,EAAE,GAAG,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IACtB,MAAM,EAAE,GAAG,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IACtB,MAAM,GAAG,GAAG,GAAG,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACxB,MAAM,GAAG,GAAG,GAAG,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACxB,MAAM,GAAG,GAAG,GAAG,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACxB,MAAM,GAAG,GAAG,GAAG,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IAExB,OAAO,KAAK;SACT,IAAI,CAAC,KAAK,CAAC,QAAQ;IAClB,mDAAmD;IACnD,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IACrC,mDAAmD;IACnD,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IACrC,mDAAmD;IACnD,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IACrC,mDAAmD;IACnD,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CACtC,CAAC;SACD,MAAM,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC;SAClD,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC;SACzB,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;AAC/B,CAAC"}

package/dist/src/transforms/filter-query.d.ts

@@ -0,0 +1,12 @@
+import type { FilterExpr } from '../types.js';
+import { type Query } from '../ast/query.js';
+import { type TableRefNode } from '../ast/table-ref.js';
+/**
+ * Returns a generator function that clones the given query and adds
+ * a WHERE clause for the specified table reference.
+ * @param query The query to clone and extend.
+ * @param tableRef The table to filter.
+ * @returns The generator function.
+ */
+export declare function filterQuery(query: Query, tableRef: TableRefNode): (filter: FilterExpr) => Query;
+//# sourceMappingURL=filter-query.d.ts.map

package/dist/src/transforms/filter-query.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"filter-query.d.ts","sourceRoot":"","sources":["../../../src/transforms/filter-query.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAiB,KAAK,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAC5D,OAAO,EAAc,KAAK,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAIpE;;;;;;GAMG;AACH,wBAAgB,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,YAAY,IACtD,QAAQ,UAAU,WAc3B"}

package/dist/src/transforms/filter-query.js

@@ -0,0 +1,39 @@
+import { isSelectQuery } from '../ast/query.js';
+import { isTableRef } from '../ast/table-ref.js';
+import { deepClone } from '../visit/clone.js';
+import { walk } from '../visit/walk.js';
+/**
+ * Returns a generator function that clones the given query and adds
+ * a WHERE clause for the specified table reference.
+ * @param query The query to clone and extend.
+ * @param tableRef The table to filter.
+ * @returns The generator function.
+ */
+export function filterQuery(query, tableRef) {
+    return (filter) => {
+        const clone = deepClone(query);
+        walk(clone, (node) => {
+            if (isSelectQuery(node) &&
+                node._from.length === 1 &&
+                isTableRef(node._from[0].expr) &&
+                arrayEquals(node._from[0].expr.table, tableRef.table)) {
+                node.where(filter);
+            }
+        });
+        return clone;
+    };
+}
+function arrayEquals(a, b) {
+    if (a === b)
+        return true;
+    if (a == null || b == null)
+        return false;
+    if (a.length !== b.length)
+        return false;
+    for (let i = 0; i < a.length; ++i) {
+        if (a[i] !== b[i])
+            return false;
+    }
+    return true;
+}
+//# sourceMappingURL=filter-query.js.map

package/dist/src/transforms/filter-query.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"filter-query.js","sourceRoot":"","sources":["../../../src/transforms/filter-query.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAc,MAAM,iBAAiB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAqB,MAAM,qBAAqB,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAExC;;;;;;GAMG;AACH,MAAM,UAAU,WAAW,CAAC,KAAY,EAAE,QAAsB;IAC9D,OAAO,CAAC,MAAkB,EAAE,EAAE;QAC5B,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,IAAI,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE;YACnB,IACE,aAAa,CAAC,IAAI,CAAC;gBACnB,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;gBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC9B,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,EACrD,CAAC;gBACD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACrB,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,CAAY,EAAE,CAAY;IAC7C,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACzB,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI;QAAE,OAAO,KAAK,CAAC;IACzC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC;IAExC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;QAClC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC;IAClC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/src/transforms/line-density.d.ts

@@ -0,0 +1,24 @@
+import type { ExprValue } from '../types.js';
+import { type SelectQuery } from '../ast/query.js';
+/**
+ * Compute line segment densities over a gridded 2D domain. The returned
+ * query uses multiple subqueries (CTEs) to identify line segment end point
+ * pairs, perform line rasterization in-database, normalize arc lengths,
+ * and then sum results for all line series to produce a density map.
+ * Based on Moritz and Fisher's work: https://arxiv.org/abs/1808.06019
+ * @param q The base query over the data.
+ * @param x Bin expression for x dimension.
+ *  Provides gridded x coordinates, potentially with a fractional component.
+ * @param y Bin expression for x dimension.
+ *  Provides gridded y coordinates, potentially with a fractional component.
+ * @param z Group by columns that segment data into individual line
+ *  series. An empty array indicates there is only a single line series.
+ * @param xn The number of grid bins for the x dimension.
+ * @param yn The number of grid bins for the y dimension.
+ * @param groupby Additional group by expressions. Separate
+ *  line density maps are created for each of these groups.
+ * @param normalize Flag toggling approximate arc-length normalization to
+ *  improve accuracy and reduce artifacts (default `true`).
+ */
+export declare function lineDensity(q: SelectQuery, x: ExprValue, y: ExprValue, z: string[], xn: number, yn: number, groupby?: string[], normalize?: boolean): SelectQuery;
+//# sourceMappingURL=line-density.d.ts.map

package/dist/src/transforms/line-density.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"line-density.d.ts","sourceRoot":"","sources":["../../../src/transforms/line-density.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAS,KAAK,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAU1D;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAgB,WAAW,CACzB,CAAC,EAAE,WAAW,EACd,CAAC,EAAE,SAAS,EACZ,CAAC,EAAE,SAAS,EACZ,CAAC,EAAE,MAAM,EAAE,EACX,EAAE,EAAE,MAAM,EACV,EAAE,EAAE,MAAM,EACV,OAAO,GAAE,MAAM,EAAO,EACtB,SAAS,GAAE,OAAc,eA+E1B"}

package/dist/src/transforms/line-density.js

@@ -0,0 +1,98 @@
+import { Query } from '../ast/query.js';
+import { count, max, sum } from '../functions/aggregate.js';
+import { int32 } from '../functions/cast.js';
+import { abs, floor, greatest, round, sign } from '../functions/numeric.js';
+import { add, div, gt, isNull, lt, lte, mul, or, sub } from '../functions/operators.js';
+import { asc } from '../functions/order-by.js';
+import { sql } from '../functions/sql-template-tag.js';
+import { lead } from '../functions/window.js';
+import { over } from '../util/ast.js';
+/**
+ * Compute line segment densities over a gridded 2D domain. The returned
+ * query uses multiple subqueries (CTEs) to identify line segment end point
+ * pairs, perform line rasterization in-database, normalize arc lengths,
+ * and then sum results for all line series to produce a density map.
+ * Based on Moritz and Fisher's work: https://arxiv.org/abs/1808.06019
+ * @param q The base query over the data.
+ * @param x Bin expression for x dimension.
+ *  Provides gridded x coordinates, potentially with a fractional component.
+ * @param y Bin expression for x dimension.
+ *  Provides gridded y coordinates, potentially with a fractional component.
+ * @param z Group by columns that segment data into individual line
+ *  series. An empty array indicates there is only a single line series.
+ * @param xn The number of grid bins for the x dimension.
+ * @param yn The number of grid bins for the y dimension.
+ * @param groupby Additional group by expressions. Separate
+ *  line density maps are created for each of these groups.
+ * @param normalize Flag toggling approximate arc-length normalization to
+ *  improve accuracy and reduce artifacts (default `true`).
+ */
+export function lineDensity(q, x, y, z, xn, yn, groupby = [], normalize = true) {
+    // select x, y points binned to the grid
+    q.select({
+        x: int32(floor(x)),
+        y: int32(floor(y))
+    });
+    // select line segment end point pairs
+    // retain only segments within the grid region
+    const groups = groupby.concat(z);
+    const pairs = Query
+        .from(q)
+        .select(groups, {
+        x0: 'x',
+        y0: 'y',
+        dx: sub(lead('x').over('sw'), 'x'),
+        dy: sub(lead('y').over('sw'), 'y')
+    })
+        .window({
+        sw: over().partitionby(groups).orderby(asc('x'))
+    })
+        .qualify([
+        or(lt('x0', xn), lt(add('x0', 'dx'), xn)),
+        or(lt('y0', yn), lt(add('y0', 'dy'), yn)),
+        or(gt('x0', 0), gt(add('x0', 'dx'), 0)),
+        or(gt('y0', 0), gt(add('y0', 'dy'), 0))
+    ]);
+    // create indices to join against for rasterization
+    // generate the maximum number of indices needed
+    const num = Query
+        .select({ x: greatest(max(abs('dx')), max(abs('dy'))) })
+        .from('pairs');
+    const indices = Query.select({
+        i: int32(sql `UNNEST(range((${num})))`)
+    });
+    // rasterize line segments
+    const raster = Query.unionAll(Query
+        .select(groups, {
+        x: add('x0', 'i'),
+        y: add('y0', int32(round(div(mul('i', 'dy'), 'dx'))))
+    })
+        .from('pairs', 'indices')
+        .where([lte(abs('dy'), abs('dx')), lt('i', abs('dx'))]), Query
+        .select(groups, {
+        x: add('x0', int32(round(div(mul(mul(sign('dy'), 'i'), 'dx'), 'dy')))),
+        y: add('y0', mul(sign('dy'), 'i'))
+    })
+        .from('pairs', 'indices')
+        .where([gt(abs('dy'), abs('dx')), lt('i', abs('dy'))]), Query
+        .select(groups, { x: 'x0', y: 'y0' })
+        .from('pairs')
+        .where(isNull('dx')));
+    // filter raster, normalize columns for each series
+    const points = Query
+        .from('raster')
+        .select(groups, 'x', 'y', normalize
+        ? { w: div(1, count().partitionby(['x'].concat(groups))) }
+        : null)
+        .where([lte(0, 'x'), lt('x', xn), lte(0, 'y'), lt('y', yn)]);
+    // sum normalized, rasterized series into output grids
+    return Query
+        .with({ pairs, indices, raster, points })
+        .from('points')
+        .select(groupby, {
+        index: add('x', mul('y', int32(xn))),
+        density: normalize ? sum('w') : count()
+    })
+        .groupby('index', groupby);
+}
+//# sourceMappingURL=line-density.js.map

package/dist/src/transforms/line-density.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"line-density.js","sourceRoot":"","sources":["../../../src/transforms/line-density.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAoB,MAAM,iBAAiB,CAAC;AAC1D,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC7C,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,yBAAyB,CAAC;AAC5E,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,2BAA2B,CAAC;AACxF,OAAO,EAAE,GAAG,EAAE,MAAM,0BAA0B,CAAC;AAC/C,OAAO,EAAE,GAAG,EAAE,MAAM,kCAAkC,CAAC;AACvD,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAEtC;;;;;;;;;;;;;;;;;;;GAmBG;AACH,MAAM,UAAU,WAAW,CACzB,CAAc,EACd,CAAY,EACZ,CAAY,EACZ,CAAW,EACX,EAAU,EACV,EAAU,EACV,UAAoB,EAAE,EACtB,YAAqB,IAAI;IAEzB,wCAAwC;IACxC,CAAC,CAAC,MAAM,CAAC;QACP,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;KACnB,CAAC,CAAC;IAEH,sCAAsC;IACtC,8CAA8C;IAC9C,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACjC,MAAM,KAAK,GAAG,KAAK;SAChB,IAAI,CAAC,CAAC,CAAC;SACP,MAAM,CAAC,MAAM,EAAE;QACd,EAAE,EAAE,GAAG;QACP,EAAE,EAAE,GAAG;QACP,EAAE,EAAE,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC;QAClC,EAAE,EAAE,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC;KACnC,CAAC;SACD,MAAM,CAAC;QACN,EAAE,EAAE,IAAI,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;KACjD,CAAC;SACD,OAAO,CAAC;QACP,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;QACzC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;QACzC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACvC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;KACxC,CAAC,CAAC;IAEL,mDAAmD;IACnD,gDAAgD;IAChD,MAAM,GAAG,GAAG,KAAK;SACd,MAAM,CAAC,EAAE,CAAC,EAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;SACvD,IAAI,CAAC,OAAO,CAAC,CAAC;IACjB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC;QAC3B,CAAC,EAAE,KAAK,CAAC,GAAG,CAAA,iBAAiB,GAAG,KAAK,CAAC;KACvC,CAAC,CAAC;IAEH,0BAA0B;IAC1B,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,CAC3B,KAAK;SACF,MAAM,CAAC,MAAM,EAAE;QACd,CAAC,EAAE,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC;QACjB,CAAC,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;KACtD,CAAC;SACD,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC;SACxB,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EACzD,KAAK;SACF,MAAM,CAAC,MAAM,EAAE;QACd,CAAC,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QACtE,CAAC,EAAE,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC;KACnC,CAAC;SACD,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC;SACxB,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EACxD,KAAK;SACF,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC;SACpC,IAAI,CAAC,OAAO,CAAC;SACb,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CACvB,CAAC;IAEF,mDAAmD;IACnD,MAAM,MAAM,GAAG,KAAK;SACjB,IAAI,CAAC,QAAQ,CAAC;SACd,MAAM,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,EACtB,SAAS;QACP,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE;QAC1D,CAAC,CAAC,IAAI,CACT;SACA,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAE/D,sDAAsD;IACtD,OAAO,KAAK;SACT,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;SACxC,IAAI,CAAC,QAAQ,CAAC;SACd,MAAM,CAAC,OAAO,EAAE;QACf,KAAK,EAAE,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE;KACxC,CAAC;SACD,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;AAC/B,CAAC"}