@milaboratories/pl-model-common 1.13.1 → 1.13.2

package/src/drivers/pframe/table_calculate.ts

@@ -1,5 +1,6 @@
 import type { PTableColumnId, PTableColumnSpec } from './table_common';
 import type { PTableVector } from './data_types';
+import type { PObjectId } from '../../pool';
 import { assertNever } from '../../util';
 
 /** Defines a terminal column node in the join request tree */
@@ -11,6 +12,36 @@ export interface ColumnJoinEntry<Col> {
   readonly column: Col;
 }
 
+/**
+ * Axis filter slicing target axis from column axes.
+ * If the axis has parents or is a parent, slicing cannot be applied (an error will be thrown).
+ * */
+export interface ConstantAxisFilter {
+  /** Filter type discriminator */
+  readonly type: 'constant';
+
+  /** Index of axis to slice (zero-based) */
+  readonly axisIndex: number;
+
+  /** Equality filter reference value, see {@link SingleValueEqualPredicate} */
+  readonly constant: string | number;
+}
+
+/** Defines a terminal column node in the join request tree */
+export interface SlicedColumnJoinEntry<Col> {
+  /** Node type discriminator */
+  readonly type: 'slicedColumn';
+
+  /** Local column */
+  readonly column: Col;
+
+  /** New column id */
+  readonly newId: PObjectId;
+
+  /** Non-empty list of axis filters */
+  readonly axisFilters: ConstantAxisFilter[];
+}
+
 /**
  * Defines a join request tree node that will output only records present in
  * all child nodes ({@link entries}).
@@ -68,6 +99,7 @@ export interface OuterJoin<Col> {
  * */
 export type JoinEntry<Col> =
   | ColumnJoinEntry<Col>
+  | SlicedColumnJoinEntry<Col>
   | InnerJoin<Col>
   | FullJoin<Col>
   | OuterJoin<Col>;
@@ -325,6 +357,13 @@ export function mapJoinEntry<C1, C2>(
         type: 'column',
         column: cb(entry.column),
       };
+    case 'slicedColumn':
+      return {
+        type: 'slicedColumn',
+        column: cb(entry.column),
+        newId: entry.newId,
+        axisFilters: entry.axisFilters,
+      };
     case 'inner':
     case 'full':
       return {

package/src/pool/spec.ts

@@ -85,6 +85,9 @@ export function extractAllColumns<D>(entry: JoinEntry<PColumn<D>>): PColumn<D>[]
       case 'column':
         columns.set(entry.column.id, entry.column);
         return;
+      case 'slicedColumn':
+        columns.set(entry.column.id, entry.column);
+        return;
       case 'full':
       case 'inner':
         for (const e of entry.entries) addAllColumns(e);