df-script 1.4.0 → 1.5.0

package/dist/columnExpressions/mixins/ArithmeticExpr.js

@@ -2,142 +2,130 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ArithmeticExpr = void 0;
 const ExprBase_1 = require("../ExprBase");
-const utils_1 = require("../../utils");
-const ArithmeticExpr = (Base) => {
-    return class extends Base {
-        abs() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => Math.abs(v)));
-        }
-        acos() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => (v < -1 || v > 1) ? null : Math.acos(v)));
-        }
-        acosh() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => v < 1 ? null : Math.acosh(v)));
-        }
-        add(val) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneBinary)(this, val, (v, r) => v + r));
-        }
-        asin() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => (v < -1 || v > 1) ? null : Math.asin(v)));
-        }
-        asinh() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.asinh));
-        }
-        atan() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.atan));
-        }
-        atanh() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => (v <= -1 || v >= 1) ? null : Math.atanh(v)));
-        }
-        cbrt() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.cbrt));
-        }
-        ceil() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.ceil));
-        }
-        clip(lower = null, upper = null) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => {
-                if (lower !== null && upper !== null) {
-                    return (0, utils_1.clamp)({ val: v, min: lower, max: upper });
-                }
-                else if (lower !== null) {
-                    return Math.max(v, lower);
-                }
-                else if (upper !== null) {
-                    return Math.min(v, upper);
-                }
-                return v;
-            }));
-        }
-        cos() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.cos));
-        }
-        cosh() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.cosh));
-        }
-        degrees() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => v * (180 / Math.PI)));
-        }
-        div(val) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneBinary)(this, val, (v, r) => r === 0 ? null : v / r));
-        }
-        exp() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.exp));
-        }
-        expm1() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.expm1));
-        }
-        floor() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.floor));
-        }
-        floordiv(val) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneBinary)(this, val, (v, r) => r === 0 ? null : Math.floor(v / r)));
-        }
-        hypot(val) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneBinary)(this, val, (v, r) => Math.hypot(v, r)));
-        }
-        log(base = Math.E) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => v <= 0 ? null : (base === Math.E ? Math.log(v) : Math.log(v) / Math.log(base))));
-        }
-        log1p() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => v <= -1 ? null : Math.log1p(v)));
-        }
-        mod(val) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneBinary)(this, val, (v, r) => r === 0 ? null : v % r));
-        }
-        mul(val) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneBinary)(this, val, (v, r) => v * r));
-        }
-        negate() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => -v));
-        }
-        pow(val) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneBinary)(this, val, (v, r) => Math.pow(v, r)));
-        }
-        radians() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => v * (Math.PI / 180)));
-        }
-        rand(seed, { min = 0, max = 1, integer = false } = {}) {
-            return (0, ExprBase_1.derive)(this, (vArray) => {
-                const len = vArray.length;
-                const out = new Float64Array(len);
-                const rnd = seed !== undefined ? (0, utils_1.mulberry32)(seed) : Math.random;
-                const range = max - min;
-                for (let i = 0; i < len; i++) {
-                    const raw = rnd();
-                    out[i] = integer ? Math.floor(raw * (range + 1)) + min : raw * range + min;
-                }
-                return out;
-            });
-        }
-        round(decimals = 0) {
-            const factor = Math.pow(10, decimals);
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => Math.round(v * factor) / factor));
-        }
-        sign() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.sign));
-        }
-        sin() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.sin));
-        }
-        sinh() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.sinh));
-        }
-        sqrt() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => v < 0 ? null : Math.sqrt(v)));
-        }
-        sub(val) {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneBinary)(this, val, (v, r) => v - r));
-        }
-        tan() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.tan));
-        }
-        tanh() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.tanh));
-        }
-        trunc() {
-            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)(Math.trunc));
-        }
-    };
-};
+const utils_1 = require("../utils");
+const utils_2 = require("../../utils");
+class ArithmeticExpr extends ExprBase_1.ExprBase {
+    abs() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => Math.abs(v)));
+    }
+    acos() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => (v < -1 || v > 1) ? null : Math.acos(v)));
+    }
+    acosh() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => v < 1 ? null : Math.acosh(v)));
+    }
+    add(val) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneBinary)(this, val, (v, r) => v + r));
+    }
+    asin() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => (v < -1 || v > 1) ? null : Math.asin(v)));
+    }
+    asinh() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.asinh));
+    }
+    atan() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.atan));
+    }
+    atanh() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => (v <= -1 || v >= 1) ? null : Math.atanh(v)));
+    }
+    cbrt() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.cbrt));
+    }
+    ceil() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.ceil));
+    }
+    clip(lower = null, upper = null) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => (0, utils_2.clamp)(v, { min: lower, max: upper })));
+    }
+    cos() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.cos));
+    }
+    cosh() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.cosh));
+    }
+    degrees() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => v * (180 / Math.PI)));
+    }
+    div(val) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneBinary)(this, val, (v, r) => r === 0 ? null : v / r));
+    }
+    exp() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.exp));
+    }
+    expm1() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.expm1));
+    }
+    floor() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.floor));
+    }
+    floordiv(val) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneBinary)(this, val, (v, r) => r === 0 ? null : Math.floor(v / r)));
+    }
+    hypot(val) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneBinary)(this, val, (v, r) => Math.hypot(v, r)));
+    }
+    log(base = Math.E) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => v <= 0 ? null : (base === Math.E ? Math.log(v) : Math.log(v) / Math.log(base))));
+    }
+    log1p() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => v <= -1 ? null : Math.log1p(v)));
+    }
+    mod(val) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneBinary)(this, val, (v, r) => r === 0 ? null : v % r));
+    }
+    mul(val) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneBinary)(this, val, (v, r) => v * r));
+    }
+    negate() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => -v));
+    }
+    pow(val) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneBinary)(this, val, (v, r) => Math.pow(v, r)));
+    }
+    radians() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => v * (Math.PI / 180)));
+    }
+    rand(seed, { min = 0, max = 1, integer = false } = {}) {
+        return (0, ExprBase_1.derive)(this, (vArray) => {
+            const len = vArray.length;
+            const out = new Float64Array(len);
+            const rnd = seed !== undefined ? (0, utils_2.mulberry32)(seed) : Math.random;
+            const range = max - min;
+            for (let i = 0; i < len; i++) {
+                const raw = rnd();
+                out[i] = integer ? Math.floor(raw * (range + 1)) + min : raw * range + min;
+            }
+            return out;
+        });
+    }
+    round(decimals = 0) {
+        const factor = Math.pow(10, decimals);
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => Math.round(v * factor) / factor));
+    }
+    sign() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.sign));
+    }
+    sin() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.sin));
+    }
+    sinh() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.sinh));
+    }
+    sqrt() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)((v) => v < 0 ? null : Math.sqrt(v)));
+    }
+    sub(val) {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneBinary)(this, val, (v, r) => v - r));
+    }
+    tan() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.tan));
+    }
+    tanh() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.tanh));
+    }
+    trunc() {
+        return (0, ExprBase_1.derive)(this, (0, utils_1.kleeneUnary)(Math.trunc));
+    }
+}
 exports.ArithmeticExpr = ArithmeticExpr;

package/dist/columnExpressions/mixins/BooleanExpr.d.ts

@@ -0,0 +1,40 @@
+import type { ExprConstructor } from "../types";
+export declare const BooleanExpr: <TBase extends ExprConstructor>(Base: TBase) => {
+    new (...args: any[]): {
+        and(other: any): /*elided*/ any;
+        not(): /*elided*/ any;
+        or(other: any): /*elided*/ any;
+        xor(other: any): /*elided*/ any;
+        all(): /*elided*/ any;
+        all_null(): /*elided*/ any;
+        any(): /*elided*/ any;
+        any_null(): /*elided*/ any;
+        ops: import("../..").OpFn[];
+        colName?: string;
+        outputName?: string;
+        isLiteral?: boolean;
+        literalValue?: any;
+        aggFn?: import("../..").AggFn<any> | null;
+        groupingOpsIndex?: number;
+        partitionOpsIndex?: number;
+        partitionBy?: (string | import("../..").IExpr)[] | null;
+        windowOp?: {
+            type: string;
+            [key: string]: any;
+        } | null;
+        isWindow?: boolean;
+        alias(name: string): /*elided*/ any;
+        cast(dataType: import("../..").RegisteredDataType): /*elided*/ any;
+        _resolve(val: any, columns: import("../..").ColumnDict, height: number): import("../..").ColumnData | any;
+        evaluate(columns: import("../..").ColumnDict, height: number): import("../..").ColumnData;
+        evaluatePreGrouping(columns: import("../..").ColumnDict, height: number): import("../..").ColumnData;
+        evaluatePostGrouping(aggregatedArray: any[], columns: import("../..").ColumnDict): import("../..").ColumnData;
+        evaluatePrePartition(columns: import("../..").ColumnDict, height: number): import("../..").ColumnData;
+        evaluatePostPartition(aggregatedArray: any[], columns: import("../..").ColumnDict): import("../..").ColumnData;
+        evaluateWindow?(groupPreValues: any[], partitionIndices: number[], currentIndex: number): any;
+        debug(label?: string): /*elided*/ any;
+        _window(evaluateWindow: (this: import("../..").IExpr, groupPreValues: any[], partitionIndices: number[], currentIndex: number) => any): /*elided*/ any;
+        _rolling(windowSize: number, aggFn: (vals: any[]) => any): /*elided*/ any;
+        _deriveAgg(fn: import("../..").AggFn<any>): /*elided*/ any;
+    };
+} & TBase;

package/dist/columnExpressions/mixins/BooleanExpr.js

@@ -0,0 +1,67 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BooleanExpr = void 0;
+const ExprBase_1 = require("../ExprBase");
+const utils_1 = require("../../utils");
+const BooleanExpr = (Base) => {
+    return class extends Base {
+        and(other) {
+            return (0, ExprBase_1.derive)(this, (vArray, columns) => {
+                const height = vArray.length;
+                const otherVal = this._resolve(other, columns, height);
+                const isOtherArray = (0, utils_1.isArrayOrTypedArray)(otherVal);
+                const result = new Array(height);
+                for (let i = 0; i < height; i++) {
+                    const v = vArray[i];
+                    const w = isOtherArray ? otherVal[i] : otherVal;
+                    if (v === false || w === false)
+                        result[i] = false;
+                    else if (v == null || w == null)
+                        result[i] = null;
+                    else
+                        result[i] = true;
+                }
+                return result;
+            });
+        }
+        not() {
+            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneUnary)((v) => !v));
+        }
+        or(other) {
+            return (0, ExprBase_1.derive)(this, (vArray, columns) => {
+                const height = vArray.length;
+                const otherVal = this._resolve(other, columns, height);
+                const isOtherArray = (0, utils_1.isArrayOrTypedArray)(otherVal);
+                const result = new Array(height);
+                for (let i = 0; i < height; i++) {
+                    const v = vArray[i];
+                    const w = isOtherArray ? otherVal[i] : otherVal;
+                    if (v === true || w === true)
+                        result[i] = true;
+                    else if (v == null || w == null)
+                        result[i] = null;
+                    else
+                        result[i] = false;
+                }
+                return result;
+            });
+        }
+        xor(other) {
+            return (0, ExprBase_1.derive)(this, (0, ExprBase_1.kleeneBinary)(this, other, (v, w) => !!v !== !!w));
+        }
+        // Boolean Aggregations
+        all() {
+            return this._deriveAgg((v) => (0, utils_1.isArrayOfType)(v, (x) => !!x, { mode: "every" }));
+        }
+        all_null() {
+            return this._deriveAgg((v) => (0, utils_1.isArrayOfType)(v, "nullish", { mode: "every" }));
+        }
+        any() {
+            return this._deriveAgg((v) => (0, utils_1.isArrayOfType)(v, (x) => !!x, { mode: "some" }));
+        }
+        any_null() {
+            return this._deriveAgg((v) => (0, utils_1.isArrayOfType)(v, "nullish", { mode: "some" }));
+        }
+    };
+};
+exports.BooleanExpr = BooleanExpr;

package/dist/columnExpressions/mixins/ComparisonExpr.d.ts

@@ -1,60 +1,31 @@
-import type { ExprConstructor } from "../types";
-export declare const ComparisonExpr: <TBase extends ExprConstructor>(Base: TBase) => {
-    new (...args: any[]): {
-        between(lower: any, upper: any, closed?: "both" | "left" | "right" | "none"): /*elided*/ any;
-        eq(val: any): /*elided*/ any;
-        eq_missing(val: any): /*elided*/ any;
-        ge(val: any): /*elided*/ any;
-        gt(val: any): /*elided*/ any;
-        has_nulls(): any;
-        is_close(other: any, options?: {
-            abs_tol?: number;
-            rel_tol?: number;
-            nans_equal?: boolean;
-        }): /*elided*/ any;
-        is_duplicated(): /*elided*/ any;
-        is_empty(options?: {
-            ignoreNulls?: boolean;
-        }): /*elided*/ any;
-        is_finite(): /*elided*/ any;
-        is_first_distinct(): /*elided*/ any;
-        is_in(values: any[] | any): /*elided*/ any;
-        is_infinite(): /*elided*/ any;
-        is_last_distinct(): /*elided*/ any;
-        is_n_distinct(n: number): any;
-        is_nan(): /*elided*/ any;
-        is_not_nan(): /*elided*/ any;
-        is_not_null(): /*elided*/ any;
-        is_null(): /*elided*/ any;
-        is_unique(): /*elided*/ any;
-        le(val: any): /*elided*/ any;
-        lt(val: any): /*elided*/ any;
-        ne(val: any): /*elided*/ any;
-        ne_missing(val: any): /*elided*/ any;
-        not_in(values: any[] | any): /*elided*/ any;
-        ops: import("../..").OpFn[];
-        colName?: string;
-        outputName?: string;
-        isLiteral?: boolean;
-        literalValue?: any;
-        aggFn?: import("../..").AggFn<any> | null;
-        groupingOpsIndex?: number;
-        partitionOpsIndex?: number;
-        partitionBy?: (string | import("../..").IExpr)[] | null;
-        windowOp?: {
-            type: string;
-            [key: string]: any;
-        } | null;
-        isWindow?: boolean;
-        alias(name: string): /*elided*/ any;
-        cast(dataType: any): /*elided*/ any;
-        _resolve(val: any, columns: import("../..").ColumnDict, height: number): import("../..").ColumnData | any;
-        evaluate(columns: import("../..").ColumnDict, height: number): import("../..").ColumnData;
-        evaluatePreGrouping(columns: import("../..").ColumnDict, height: number): import("../..").ColumnData;
-        evaluatePostGrouping(aggregatedArray: any[], columns: import("../..").ColumnDict): import("../..").ColumnData;
-        evaluatePrePartition(columns: import("../..").ColumnDict, height: number): import("../..").ColumnData;
-        evaluatePostPartition(aggregatedArray: any[], columns: import("../..").ColumnDict): import("../..").ColumnData;
-        evaluateWindow?(groupPreValues: any[], partitionIndices: number[], currentIndex: number): any;
-        debug(label?: string): /*elided*/ any;
-    };
-} & TBase;
+import { ExprBase } from "../ExprBase";
+export declare class ComparisonExpr extends ExprBase {
+    between(lower: any, upper: any, closed?: "both" | "left" | "right" | "none"): this;
+    eq(val: any): this;
+    eq_missing(val: any): this;
+    ge(val: any): this;
+    gt(val: any): this;
+    has_nulls(): any;
+    is_close(other: any, { abs_tol, rel_tol, nans_equal }?: {
+        abs_tol?: number;
+        rel_tol?: number;
+        nans_equal?: boolean;
+    }): this;
+    is_duplicated(): this;
+    is_empty({ ignoreNulls }?: {
+        ignoreNulls?: boolean;
+    }): this;
+    is_finite(): this;
+    is_in(values: any[] | any): this;
+    is_infinite(): this;
+    is_nan(): this;
+    is_not_nan(): this;
+    is_not_null(): this;
+    is_null(): this;
+    is_unique(): this;
+    le(val: any): this;
+    lt(val: any): this;
+    ne(val: any): this;
+    ne_missing(val: any): this;
+    not_in(values: any[] | any): this;
+}