feat: Pivot filter UI

plugins/pivot/src/js/src/IrisGridPivotMetricCalculator.ts

@@ -0,0 +1,94 @@
+import {
+  IrisGridMetricCalculator,
+  type IrisGridMetricState,
+} from '@deephaven/iris-grid';
+import type { GridMetrics, ModelIndex, ModelSizeMap } from '@deephaven/grid';
+import { isIrisGridPivotModel } from './IrisGridPivotModel';
+
+export interface PivotGridMetrics extends GridMetrics {
+  // Width of the widest column source header text, including padding
+  sourceTextWidth: number;
+}
+
+class IrisGridPivotMetricCalculator extends IrisGridMetricCalculator {
+  // Gets the text width for a column header group, including padding
+  getColumnHeaderGroupTextWidth(
+    modelColumn: ModelIndex,
+    depth: number,
+    state: IrisGridMetricState,
+    maxColumnWidth: number
+  ): number {
+    return super.getColumnHeaderGroupWidth(
+      modelColumn,
+      depth,
+      state,
+      maxColumnWidth
+    );
+  }
+
+  getColumnHeaderGroupWidth(
+    modelColumn: ModelIndex,
+    depth: number,
+    state: IrisGridMetricState,
+    maxColumnWidth: number
+  ): number {
+    return this.getColumnHeaderGroupTextWidth(
+      modelColumn,
+      depth,
+      state,
+      maxColumnWidth
+    );
+  }
+
+  // TODO: fix performance, cache results?
+  calculateSourceTextWidths(
+    model: unknown,
+    state: IrisGridMetricState
+  ): ModelSizeMap {
+    const sourceTextWidths: ModelSizeMap = new Map();
+    if (!isIrisGridPivotModel(model)) {
+      return sourceTextWidths;
+    }
+
+    const { theme } = state;
+    const { headerHorizontalPadding, maxColumnWidth } = theme;
+
+    // TODO: iterate over parents of column 0 to find source columns
+
+    const sourceIndexes: ModelIndex[] = Object.keys(model.columns)
+      .map(Number)
+      .filter(key => key < 0);
+
+    sourceIndexes.forEach(sourceIndex => {
+      const width = this.getColumnHeaderGroupTextWidth(
+        sourceIndex,
+        0,
+        state,
+        maxColumnWidth
+      );
+      // Extra padding between the text and the sort icon
+      sourceTextWidths.set(sourceIndex, width + headerHorizontalPadding);
+    });
+    return sourceTextWidths;
+  }
+
+  /**
+   * Gets the metrics for the current state. This method has to be called before setColumnSize or resetColumnSize.
+   * @param state The current IrisGridMetricState
+   * @returns The metrics for the current state
+   */
+  getMetrics(state: IrisGridMetricState): PivotGridMetrics {
+    const { model } = state;
+    // Update column widths if columns in the cached model don't match the current model passed in the state
+    const sourceTextWidths = this.calculateSourceTextWidths(model, state);
+
+    const sourceTextWidth = Math.max(...sourceTextWidths.values());
+
+    return {
+      ...super.getMetrics(state),
+      sourceTextWidth,
+    };
+  }
+}
+
+export default IrisGridPivotMetricCalculator;

plugins/pivot/src/js/src/IrisGridPivotModel.ts

@@ -200,12 +200,13 @@ class IrisGridPivotModel<R extends UIPivotRow = UIPivotRow>
   dh: typeof CorePlusDhType;
 
   get filter(): readonly DhType.FilterCondition[] {
-    return EMPTY_ARRAY;
+    return this.pivotTable.filter;
   }
 
-  set filter(_: readonly DhType.FilterCondition[]) {
-    // No-op
-    // TODO: DH-20363: Add support for Pivot filters
+  set filter(filters: DhType.FilterCondition[]) {
+    log.debug2('Setting filter on pivot table', filters);
+    this.pivotTable.applyFilter(filters);
+    this.applyViewport();
   }
 
   hydratePivotSort(
@@ -627,8 +628,7 @@ class IrisGridPivotModel<R extends UIPivotRow = UIPivotRow>
   }
 
   isFilterable(columnIndex: ModelIndex): boolean {
-    // TODO: DH-20363: Add support for Pivot filters
-    return false;
+    return this.columns[columnIndex]?.isFilterable ?? false;
   }
 
   isColumnSortable(columnIndex: ModelIndex): boolean {