@elaraai/east 0.0.1-beta.9 → 1.0.0

package/dist/src/expr/array.js

@@ -1,7 +1,8 @@
-import { get_location } from "../location.js";
+import { get_location_id } from "../location.js";
 import { ArrayType, BooleanType, FunctionType, IntegerType, NullType, StringType, isSubtype, isTypeEqual, printType, NeverType, OptionType, isDataType, StructType, VariantType, SetType, FloatType, DictType } from "../types.js";
 import { valueOrExprToAst, valueOrExprToAstTyped } from "./ast.js";
 import { AstSymbol, Expr, FactorySymbol, TypeSymbol } from "./expr.js";
+import { equal, notEqual } from "./block.js";
 import { none, some } from "../containers/variant.js";
 import { CsvSerializeConfigType, csvSerializeOptionsToValue } from "../serialization/csv.js";
 import { BlobType } from "../types.js";
@@ -58,7 +59,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: IntegerType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArraySize",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -83,7 +84,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: IntegerType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArraySize",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -111,7 +112,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: BooleanType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayHas",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], keyAst],
@@ -150,7 +151,7 @@ export class ArrayExpr extends Expr {
             return this[FactorySymbol]({
                 ast_type: "Builtin",
                 type: this.value_type,
-                location: get_location(2),
+                loc_id: get_location_id(),
                 builtin: "ArrayGetOrDefault",
                 type_parameters: [this.value_type],
                 arguments: [this[AstSymbol], keyAst, defaultFnAst],
@@ -160,7 +161,7 @@ export class ArrayExpr extends Expr {
             return this[FactorySymbol]({
                 ast_type: "Builtin",
                 type: this.value_type,
-                location: get_location(2),
+                loc_id: get_location_id(),
                 builtin: "ArrayGet",
                 type_parameters: [this.value_type],
                 arguments: [this[AstSymbol], keyAst],
@@ -200,7 +201,7 @@ export class ArrayExpr extends Expr {
             return this[FactorySymbol]({
                 ast_type: "Builtin",
                 type: this.value_type,
-                location: get_location(2),
+                loc_id: get_location_id(),
                 builtin: "ArrayGetOrDefault",
                 type_parameters: [this.value_type],
                 arguments: [this[AstSymbol], keyAst, defaultFnAst],
@@ -210,7 +211,7 @@ export class ArrayExpr extends Expr {
             return this[FactorySymbol]({
                 ast_type: "Builtin",
                 type: this.value_type,
-                location: get_location(2),
+                loc_id: get_location_id(),
                 builtin: "ArrayGet",
                 type_parameters: [this.value_type],
                 arguments: [this[AstSymbol], keyAst],
@@ -238,7 +239,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: OptionType(this.value_type),
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayTryGet",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], keyAst],
@@ -269,7 +270,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayUpdate",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], keyAst, valueAst],
@@ -282,14 +283,14 @@ export class ArrayExpr extends Expr {
      * @see {@link update} for simply replacing the value.
      */
     merge(key, value, updateFn) {
-        const location = get_location(2);
+        const loc_id = get_location_id();
         const keyAst = valueOrExprToAstTyped(key, IntegerType);
         const value2Type = value[TypeSymbol];
         const updateFnExpr = Expr.from(updateFn, FunctionType([value2Type, this.value_type, IntegerType], this.value_type));
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location,
+            loc_id,
             builtin: "ArrayMerge",
             type_parameters: [this.value_type, value2Type],
             arguments: [this[AstSymbol], keyAst, value[AstSymbol], updateFnExpr[AstSymbol]],
@@ -316,7 +317,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayPushLast",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], valueAst],
@@ -342,7 +343,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this.value_type,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayPopLast",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -369,7 +370,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayPushFirst",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], valueAst],
@@ -395,7 +396,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this.value_type,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayPopFirst",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -422,7 +423,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayAppend",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], arrayExpr[AstSymbol]],
@@ -450,7 +451,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayPrepend",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], arrayExpr[AstSymbol]],
@@ -498,7 +499,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayMergeAll",
             type_parameters: [this.value_type, value2Type],
             arguments: [this[AstSymbol], array[AstSymbol], mergeFnExpr[AstSymbol]],
@@ -523,7 +524,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayClear",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -579,7 +580,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArraySortInPlace",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], byExpr[AstSymbol]],
@@ -634,7 +635,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArraySort",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], byExpr[AstSymbol]],
@@ -661,7 +662,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayReverseInPlace",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -690,7 +691,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayReverse",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -739,7 +740,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: BooleanType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayIsSorted",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], byExpr[AstSymbol]],
@@ -796,7 +797,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: IntegerType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayFindSortedFirst",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], valueAst, byExpr[AstSymbol]],
@@ -841,7 +842,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: IntegerType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayFindSortedLast",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], valueAst, byExpr[AstSymbol]],
@@ -885,7 +886,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: StructType({ start: IntegerType, end: IntegerType }),
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayFindSortedRange",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], valueAst, byExpr[AstSymbol]],
@@ -917,7 +918,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArraySlice",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], startExpr[AstSymbol], endExpr[AstSymbol]],
@@ -946,7 +947,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayConcat",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], otherExpr[AstSymbol]],
@@ -984,8 +985,8 @@ export class ArrayExpr extends Expr {
         const keysExpr = Expr.from(keys, ArrayType(IntegerType));
         let default_function_ast;
         if (onMissing === undefined) {
-            const location = get_location(2);
-            const default_function = Expr.function([IntegerType], this.value_type, ($, key) => $.error(Expr.str `Cannot get key ${key} from array`, location));
+            const loc_id = get_location_id();
+            const default_function = Expr.function([IntegerType], this.value_type, ($, key) => $.error(Expr.str `Cannot get key ${key} from array`, loc_id));
             default_function_ast = Expr.ast(default_function);
         }
         else {
@@ -995,7 +996,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayGetKeys",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], keysExpr[AstSymbol], default_function_ast],
@@ -1030,7 +1031,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayForEach",
             type_parameters: [this.value_type, returnType],
             arguments: [this[AstSymbol], fn[AstSymbol]],
@@ -1061,7 +1062,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayCopy",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -1087,7 +1088,7 @@ export class ArrayExpr extends Expr {
                 return this[FactorySymbol]({
                     ast_type: "Builtin",
                     type: ArrayType(output_type),
-                    location: get_location(2),
+                    loc_id: get_location_id(),
                     builtin: "ArrayMap",
                     type_parameters: [this.value_type, output_type],
                     arguments: [this[AstSymbol], fn[AstSymbol]],
@@ -1159,7 +1160,7 @@ export class ArrayExpr extends Expr {
             }
             return this[FactorySymbol]({
                 ast_type: "Builtin",
-                location: get_location(2),
+                loc_id: get_location_id(),
                 type: this[TypeSymbol],
                 builtin: "ArrayFilter",
                 type_parameters: [this.value_type],
@@ -1183,7 +1184,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: ArrayType(someType),
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayFilterMap",
             type_parameters: [this.value_type, someType],
             arguments: [this[AstSymbol], fnAst],
@@ -1202,7 +1203,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: returnType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayFirstMap",
             type_parameters: [this.value_type, someType],
             arguments: [this[AstSymbol], fnAst],
@@ -1289,7 +1290,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: returnType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayFold",
             type_parameters: [this.value_type, returnType],
             arguments: [this[AstSymbol], initAst, combineAst],
@@ -1302,7 +1303,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: mapType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayMapReduce",
             type_parameters: [this.value_type, mapType],
             arguments: [this[AstSymbol], mapAst, combineAst],
@@ -1592,7 +1593,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: StringType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayStringJoin",
             type_parameters: [],
             arguments: [this[AstSymbol], stringAst],
@@ -1604,7 +1605,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: SetType(keyType),
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayToSet",
             type_parameters: [this.value_type, keyType],
             arguments: [this[AstSymbol], keyFnAst],
@@ -1617,8 +1618,8 @@ export class ArrayExpr extends Expr {
         const valueType = valueFnAst.type.output;
         let onConflictAst;
         if (onConflictFn === undefined) {
-            const location = get_location(2);
-            const onConflictFunction = Expr.function([valueType, valueType, keyType], valueType, ($, existing, value, key) => $.error(Expr.str `Cannot insert duplicate key ${key} into dict`, location));
+            const loc_id = get_location_id();
+            const onConflictFunction = Expr.function([valueType, valueType, keyType], valueType, ($, existing, value, key) => $.error(Expr.str `Cannot insert duplicate key ${key} into dict`, loc_id));
             onConflictAst = Expr.ast(onConflictFunction);
         }
         else {
@@ -1627,7 +1628,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: DictType(keyType, valueType),
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayToDict",
             type_parameters: [this.value_type, keyType, valueType],
             arguments: [this[AstSymbol], keyFnAst, valueFnAst, onConflictAst],
@@ -1643,7 +1644,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: ArrayType(elementType),
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayFlattenToArray",
             type_parameters: [this.value_type, elementType],
             arguments: [this[AstSymbol], fnAst],
@@ -1659,7 +1660,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: SetType(elementType),
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayFlattenToSet",
             type_parameters: [this.value_type, elementType],
             arguments: [this[AstSymbol], fnAst],
@@ -1675,8 +1676,8 @@ export class ArrayExpr extends Expr {
         const valueType = returnType.value;
         let onConflictAst;
         if (onConflictFn === undefined) {
-            const location = get_location(2);
-            const onConflictFunction = Expr.function([valueType, valueType, keyType], valueType, ($, existing, value, key) => $.error(Expr.str `Cannot insert duplicate key ${key} into dict`, location));
+            const loc_id = get_location_id();
+            const onConflictFunction = Expr.function([valueType, valueType, keyType], valueType, ($, existing, value, key) => $.error(Expr.str `Cannot insert duplicate key ${key} into dict`, loc_id));
             onConflictAst = Expr.ast(onConflictFunction);
         }
         else {
@@ -1685,7 +1686,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: DictType(keyType, valueType),
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayFlattenToDict",
             type_parameters: [this.value_type, keyType, valueType],
             arguments: [this[AstSymbol], fnAst, onConflictAst],
@@ -1701,7 +1702,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: DictType(keyType, initType),
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayGroupFold",
             type_parameters: [this.value_type, keyType, initType],
             arguments: [this[AstSymbol], keyFnAst, initFnAst, reduceFnAst],
@@ -1905,11 +1906,60 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: BlobType,
-            location: get_location(2),
+            loc_id: get_location_id(),
             builtin: "ArrayEncodeCsv",
             type_parameters: [this.value_type, CsvSerializeConfigType],
             arguments: [this[AstSymbol], configAst],
         });
     }
+    /**
+     * Checks if this array equals another array (deep comparison).
+     *
+     * @param other - The array to compare against
+     * @returns A BooleanExpr that is true if the arrays are deeply equal
+     *
+     * @example
+     * ```ts
+     * const isEqual = East.function([ArrayType(IntegerType), ArrayType(IntegerType)], BooleanType, ($, a, b) => {
+     *   $.return(a.equals(b));
+     * });
+     * const compiled = East.compile(isEqual.toIR(), []);
+     * compiled([1n, 2n, 3n], [1n, 2n, 3n]);  // true
+     * compiled([1n, 2n], [1n, 2n, 3n]);      // false
+     * ```
+     */
+    equals(other) {
+        return equal(this, other);
+    }
+    /**
+     * Checks if this array does not equal another array.
+     *
+     * @param other - The array to compare against
+     * @returns A BooleanExpr that is true if the arrays are not equal
+     *
+     * @example
+     * ```ts
+     * const isNotEqual = East.function([ArrayType(IntegerType), ArrayType(IntegerType)], BooleanType, ($, a, b) => {
+     *   $.return(a.notEquals(b));
+     * });
+     * const compiled = East.compile(isNotEqual.toIR(), []);
+     * compiled([1n, 2n], [1n, 2n, 3n]);      // true
+     * compiled([1n, 2n, 3n], [1n, 2n, 3n]);  // false
+     * ```
+     */
+    notEquals(other) {
+        return notEqual(this, other);
+    }
+    // ============================================================================
+    // Aliases for comparison operations
+    // ============================================================================
+    /** Alias for {@link equals} */
+    eq = this.equals;
+    /** Alias for {@link equals} */
+    equal = this.equals;
+    /** Alias for {@link notEquals} */
+    ne = this.notEquals;
+    /** Alias for {@link notEquals} */
+    notEqual = this.notEquals;
 }
 //# sourceMappingURL=array.js.map