@uwdata/mosaic-sql 0.17.0 → 0.19.0

package/dist/src/load/create.js

@@ -0,0 +1,14 @@
+export function createTable(name, query, { replace = false, temp = false, view = false } = {}) {
+    return 'CREATE'
+        + (replace ? ' OR REPLACE ' : ' ')
+        + (temp ? 'TEMP ' : '')
+        + (view ? 'VIEW' : 'TABLE')
+        + (replace ? ' ' : ' IF NOT EXISTS ')
+        + name + ' AS ' + query;
+}
+export function createSchema(name, { strict = false } = {}) {
+    return 'CREATE SCHEMA '
+        + (strict ? '' : 'IF NOT EXISTS ')
+        + name;
+}
+//# sourceMappingURL=create.js.map

package/dist/src/load/create.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create.js","sourceRoot":"","sources":["../../../src/load/create.ts"],"names":[],"mappings":"AAMA,MAAM,UAAU,WAAW,CAAC,IAAY,EAAE,KAAa,EAAE,EACvD,OAAO,GAAG,KAAK,EACf,IAAI,GAAG,KAAK,EACZ,IAAI,GAAG,KAAK,KACU,EAAE;IACxB,OAAO,QAAQ;UACX,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC;UAChC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;UACrB,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC;UACzB,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,iBAAiB,CAAC;UACnC,IAAI,GAAG,MAAM,GAAG,KAAK,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,IAAY,EAAE,EACzC,MAAM,GAAG,KAAK,EACf,GAAG,EAAE;IACJ,OAAO,gBAAgB;UACnB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,gBAAgB,CAAC;UAChC,IAAI,CAAC;AACX,CAAC"}

package/dist/src/load/extension.d.ts

@@ -0,0 +1,2 @@
+export declare function loadExtension(name: string): string;
+//# sourceMappingURL=extension.d.ts.map

package/dist/src/load/extension.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"extension.d.ts","sourceRoot":"","sources":["../../../src/load/extension.ts"],"names":[],"mappings":"AAAA,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,UAEzC"}

package/dist/src/load/extension.js

@@ -0,0 +1,4 @@
+export function loadExtension(name) {
+    return `INSTALL ${name}; LOAD ${name}`;
+}
+//# sourceMappingURL=extension.js.map

package/dist/src/load/extension.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extension.js","sourceRoot":"","sources":["../../../src/load/extension.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,aAAa,CAAC,IAAY;IACxC,OAAO,WAAW,IAAI,UAAU,IAAI,EAAE,CAAC;AACzC,CAAC"}

package/dist/src/load/load.d.ts

@@ -0,0 +1,18 @@
+import { CreateTableOptions } from './create.js';
+type Options = {
+    select?: string[];
+} & CreateTableOptions & Record<string, unknown>;
+export declare function load(method: string, tableName: string, fileName: string, options?: Options, defaults?: Record<string, unknown>): string;
+export declare function loadCSV(tableName: string, fileName: string, options?: Options): string;
+export declare function loadJSON(tableName: string, fileName: string, options?: Options): string;
+export declare function loadParquet(tableName: string, fileName: string, options?: Options): string;
+/**
+ * Load geometry data within a spatial file format.
+ * This method requires that the DuckDB spatial extension is loaded.
+ * Supports GeoJSON, TopoJSON, and other common spatial formats.
+ * For TopoJSON, wet the layer option to indicate the feature to extract.
+ */
+export declare function loadSpatial(tableName: string, fileName: string, options?: Options): string;
+export declare function loadObjects(tableName: string, data: (Record<string, unknown>)[], options?: Options): string;
+export {};
+//# sourceMappingURL=load.d.ts.map

package/dist/src/load/load.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"load.d.ts","sourceRoot":"","sources":["../../../src/load/load.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAG9D,KAAK,OAAO,GACR;IAAE,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CAAE,GACtB,kBAAkB,GAClB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAE5B,wBAAgB,IAAI,CAClB,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,OAAY,EACrB,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,UAQvC;AAED,wBAAgB,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,UAE7E;AAED,wBAAgB,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,UAE9E;AAED,wBAAgB,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,UAEjF;AAED;;;;;GAKG;AACH,wBAAgB,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,GAAE,OAAY,UAerF;AAED,wBAAgB,WAAW,CACzB,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,EAAC,OAAO,CAAC,CAAC,EAAE,EAChC,OAAO,GAAE,OAAY,UAQtB"}

package/dist/src/load/load.js

@@ -0,0 +1,80 @@
+import { createTable } from './create.js';
+import { sqlFrom } from './sql-from.js';
+export function load(method, tableName, fileName, options = {}, defaults = {}) {
+    const { select = ['*'], where, view, temp, replace, ...file } = options;
+    const params = parameters({ ...defaults, ...file });
+    const read = `${method}('${fileName}'${params ? ', ' + params : ''})`;
+    const filter = where ? ` WHERE ${where}` : '';
+    const query = `SELECT ${select.join(', ')} FROM ${read}${filter}`;
+    return createTable(tableName, query, { view, temp, replace });
+}
+export function loadCSV(tableName, fileName, options) {
+    return load('read_csv', tableName, fileName, options, { auto_detect: true, sample_size: -1 });
+}
+export function loadJSON(tableName, fileName, options) {
+    return load('read_json', tableName, fileName, options, { auto_detect: true, format: 'auto' });
+}
+export function loadParquet(tableName, fileName, options) {
+    return load('read_parquet', tableName, fileName, options);
+}
+/**
+ * Load geometry data within a spatial file format.
+ * This method requires that the DuckDB spatial extension is loaded.
+ * Supports GeoJSON, TopoJSON, and other common spatial formats.
+ * For TopoJSON, wet the layer option to indicate the feature to extract.
+ */
+export function loadSpatial(tableName, fileName, options = {}) {
+    // nested options map to the open_options argument of st_read
+    const { options: opt, ...rest } = options;
+    if (opt) {
+        // TODO: check correct syntax for open_options
+        const open = Array.isArray(opt) ? opt.join(', ')
+            : typeof opt === 'string' ? opt
+                : Object.entries(opt)
+                    .map(([key, value]) => `${key}=${value}`)
+                    .join(', ');
+        Object.assign(rest, { open_options: open.toUpperCase() });
+    }
+    // TODO: handle box_2d for spatial_filter_box option
+    // TODO: handle wkb_blob for spatial_filter option
+    return load('st_read', tableName, fileName, rest);
+}
+export function loadObjects(tableName, data, options = {}) {
+    const { select = ['*'], ...opt } = options;
+    const values = sqlFrom(data);
+    const query = select.length === 1 && select[0] === '*'
+        ? values
+        : `SELECT ${select} FROM ${values}`;
+    return createTable(tableName, query, opt);
+}
+function parameters(options) {
+    return Object.entries(options)
+        .map(([key, value]) => `${key}=${toDuckDBValue(value)}`)
+        .join(', ');
+}
+function toDuckDBValue(value) {
+    switch (typeof value) {
+        case 'boolean':
+            return String(value);
+        case 'string':
+            return `'${value}'`;
+        case 'undefined':
+        case 'object':
+            if (value == null) {
+                return 'NULL';
+            }
+            else if (Array.isArray(value)) {
+                return '[' + value.map(v => toDuckDBValue(v)).join(', ') + ']';
+            }
+            else {
+                return '{'
+                    + Object.entries(value)
+                        .map(([k, v]) => `'${k}': ${toDuckDBValue(v)}`)
+                        .join(', ')
+                    + '}';
+            }
+        default:
+            return `${value}`;
+    }
+}
+//# sourceMappingURL=load.js.map

package/dist/src/load/load.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../src/load/load.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAsB,MAAM,aAAa,CAAC;AAC9D,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAOxC,MAAM,UAAU,IAAI,CAClB,MAAc,EACd,SAAiB,EACjB,QAAgB,EAChB,UAAmB,EAAE,EACrB,WAAoC,EAAE;IAEtC,MAAM,EAAE,MAAM,GAAG,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;IACxE,MAAM,MAAM,GAAG,UAAU,CAAC,EAAE,GAAG,QAAQ,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;IACpD,MAAM,IAAI,GAAG,GAAG,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC;IACtE,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,UAAU,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAC9C,MAAM,KAAK,GAAG,UAAU,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,IAAI,GAAG,MAAM,EAAE,CAAC;IAClE,OAAO,WAAW,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,UAAU,OAAO,CAAC,SAAiB,EAAE,QAAgB,EAAE,OAAiB;IAC5E,OAAO,IAAI,CAAC,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;AAChG,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,SAAiB,EAAE,QAAgB,EAAE,OAAiB;IAC7E,OAAO,IAAI,CAAC,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;AAChG,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,SAAiB,EAAE,QAAgB,EAAE,OAAiB;IAChF,OAAO,IAAI,CAAC,cAAc,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;AAC5D,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,WAAW,CAAC,SAAiB,EAAE,QAAgB,EAAE,UAAmB,EAAE;IACpF,6DAA6D;IAC7D,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;IAC1C,IAAI,GAAG,EAAE,CAAC;QACR,8CAA8C;QAC9C,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;YAC9C,CAAC,CAAC,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG;gBAC/B,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;qBAChB,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,KAAK,EAAE,CAAC;qBACxC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClB,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IAC5D,CAAC;IACD,oDAAoD;IACpD,kDAAkD;IAClD,OAAO,IAAI,CAAC,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,WAAW,CACzB,SAAiB,EACjB,IAAgC,EAChC,UAAmB,EAAE;IAErB,MAAM,EAAE,MAAM,GAAG,CAAC,GAAG,CAAC,EAAE,GAAG,GAAG,EAAE,GAAG,OAAO,CAAC;IAC3C,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,KAAK,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG;QACpD,CAAC,CAAC,MAAM;QACR,CAAC,CAAC,UAAU,MAAM,SAAS,MAAM,EAAE,CAAC;IACtC,OAAO,WAAW,CAAC,SAAS,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;AAC5C,CAAC;AAED,SAAS,UAAU,CAAC,OAAgC;IAClD,OAAO,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;SAC3B,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;SACvD,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,QAAQ,OAAO,KAAK,EAAE,CAAC;QACrB,KAAK,SAAS;YACZ,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;QACvB,KAAK,QAAQ;YACX,OAAO,IAAI,KAAK,GAAG,CAAC;QACtB,KAAK,WAAW,CAAC;QACjB,KAAK,QAAQ;YACX,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;gBAClB,OAAO,MAAM,CAAC;YAChB,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChC,OAAO,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC;YACjE,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG;sBACN,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC;yBAClB,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;yBAC9C,IAAI,CAAC,IAAI,CAAC;sBACb,GAAG,CAAC;YACV,CAAC;QACH;YACE,OAAO,GAAG,KAAK,EAAE,CAAC;IACtB,CAAC;AACH,CAAC"}

package/dist/src/load/sql-from.d.ts

@@ -0,0 +1,14 @@
+type SQLFromOptions = {
+    columns?: string[] | Record<string, string>;
+} & Record<string, string>;
+/**
+ * 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 declare function sqlFrom(data: (Record<string, unknown>)[], { columns }?: SQLFromOptions): string;
+export {};
+//# sourceMappingURL=sql-from.d.ts.map

package/dist/src/load/sql-from.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql-from.d.ts","sourceRoot":"","sources":["../../../src/load/sql-from.ts"],"names":[],"mappings":"AAEA,KAAK,cAAc,GAAG;IACpB,OAAO,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC7C,GAAG,MAAM,CAAC,MAAM,EAAC,MAAM,CAAC,CAAC;AAE1B;;;;;;;GAOG;AACH,wBAAgB,OAAO,CACrB,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,EAAE,EACjC,EACE,OAAsC,EACvC,GAAE,cAAmB,UAkBvB"}

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"}