@mui/x-virtualizer 0.2.10 → 0.2.12

package/features/dimensions.d.ts

@@ -1,13 +1,15 @@
 import { Store } from '@mui/x-internals/store';
 import { ColumnWithWidth, DimensionsState, RowId, RowsMetaState, Size } from "../models/index.js";
-import type { BaseState, VirtualizerParams } from "../useVirtualizer.js";
+import type { BaseState, ParamsWithDefaults } from "../useVirtualizer.js";
 export type DimensionsParams = {
   rowHeight: number;
-  columnsTotalWidth: number;
-  leftPinnedWidth: number;
-  rightPinnedWidth: number;
-  topPinnedHeight: number;
-  bottomPinnedHeight: number;
+  columnsTotalWidth?: number;
+  leftPinnedWidth?: number;
+  rightPinnedWidth?: number;
+  topPinnedHeight?: number;
+  bottomPinnedHeight?: number;
+  autoHeight?: boolean;
+  minimalContentHeight?: number | string;
   scrollbarSize?: number;
 };
 export declare const Dimensions: {
@@ -17,8 +19,12 @@ export declare const Dimensions: {
     rootSize: (state: BaseState) => Size;
     dimensions: (state: BaseState) => DimensionsState;
     rowHeight: (state: BaseState) => number;
+    columnsTotalWidth: (state: BaseState) => number;
     contentHeight: (state: BaseState) => number;
+    autoHeight: (state: BaseState) => boolean;
+    minimalContentHeight: (state: BaseState) => string | number | undefined;
     rowsMeta: (state: BaseState) => RowsMetaState;
+    rowPositions: (state: BaseState) => number[];
     columnPositions: (_: any, columns: ColumnWithWidth[]) => number[];
     needsHorizontalScrollbar: (state: BaseState) => boolean;
   };
@@ -32,8 +38,8 @@ export declare namespace Dimensions {
   };
   type API = ReturnType<typeof useDimensions>;
 }
-declare function initializeState(params: VirtualizerParams): Dimensions.State;
-declare function useDimensions(store: Store<BaseState>, params: VirtualizerParams, _api: {}): {
+declare function initializeState(params: ParamsWithDefaults): Dimensions.State;
+declare function useDimensions(store: Store<BaseState>, params: ParamsWithDefaults, _api: {}): {
   updateDimensions: (firstUpdate?: boolean) => void;
   debouncedUpdateDimensions: (((firstUpdate?: boolean) => void) & import("@mui/x-internals/throttle").Cancelable) | undefined;
   rowsMeta: {

package/features/dimensions.js

@@ -38,14 +38,20 @@ const EMPTY_DIMENSIONS = {
   leftPinnedWidth: 0,
   rightPinnedWidth: 0,
   topContainerHeight: 0,
-  bottomContainerHeight: 0
+  bottomContainerHeight: 0,
+  autoHeight: false,
+  minimalContentHeight: undefined
 };
 const selectors = {
   rootSize: state => state.rootSize,
   dimensions: state => state.dimensions,
   rowHeight: state => state.dimensions.rowHeight,
+  columnsTotalWidth: state => state.dimensions.columnsTotalWidth,
   contentHeight: state => state.dimensions.contentSize.height,
+  autoHeight: state => state.dimensions.autoHeight,
+  minimalContentHeight: state => state.dimensions.minimalContentHeight,
   rowsMeta: state => state.rowsMeta,
+  rowPositions: state => state.rowsMeta.positions,
   columnPositions: (0, _store.createSelectorMemoized)((_, columns) => {
     const positions = [];
     let currentPosition = 0;
@@ -63,7 +69,10 @@ const Dimensions = exports.Dimensions = {
   selectors
 };
 function initializeState(params) {
-  const dimensions = (0, _extends2.default)({}, EMPTY_DIMENSIONS, params.dimensions);
+  const dimensions = (0, _extends2.default)({}, EMPTY_DIMENSIONS, params.dimensions, {
+    autoHeight: params.dimensions.autoHeight,
+    minimalContentHeight: params.dimensions.minimalContentHeight
+  });
   const {
     rowCount
   } = params;
@@ -89,7 +98,7 @@ function initializeState(params) {
 function useDimensions(store, params, _api) {
   const isFirstSizing = React.useRef(true);
   const {
-    refs,
+    layout,
     dimensions: {
       rowHeight,
       columnsTotalWidth,
@@ -107,7 +116,7 @@ function useDimensions(store, params, _api) {
     if (isFirstSizing.current) {
       return;
     }
-    const containerNode = refs.container.current;
+    const containerNode = layout.refs.container.current;
     const rootSize = selectors.rootSize(store.state);
     const rowsMeta = selectors.rowsMeta(store.state);
 
@@ -125,7 +134,7 @@ function useDimensions(store, params, _api) {
     let viewportInnerSize;
     let hasScrollX = false;
     let hasScrollY = false;
-    if (params.autoHeight) {
+    if (params.dimensions.autoHeight) {
       hasScrollY = false;
       hasScrollX = Math.round(columnsTotalWidth) > Math.round(rootSize.width);
       viewportOuterSize = {
@@ -192,7 +201,9 @@ function useDimensions(store, params, _api) {
       leftPinnedWidth,
       rightPinnedWidth,
       topContainerHeight,
-      bottomContainerHeight
+      bottomContainerHeight,
+      autoHeight: params.dimensions.autoHeight,
+      minimalContentHeight: params.dimensions.minimalContentHeight
     };
     const prevDimensions = store.state.dimensions;
     if ((0, _isDeepEqual.isDeepEqual)(prevDimensions, newDimensions)) {
@@ -202,7 +213,7 @@ function useDimensions(store, params, _api) {
       dimensions: newDimensions
     });
     onResize?.(newDimensions.root);
-  }, [store, refs.container, params.dimensions.scrollbarSize, params.autoHeight, params.disableHorizontalScroll, params.disableVerticalScroll, onResize, rowHeight, columnsTotalWidth, leftPinnedWidth, rightPinnedWidth, topPinnedHeight, bottomPinnedHeight]);
+  }, [store, layout.refs.container, params.dimensions.scrollbarSize, params.dimensions.autoHeight, params.dimensions.minimalContentHeight, params.disableHorizontalScroll, params.disableVerticalScroll, onResize, rowHeight, columnsTotalWidth, leftPinnedWidth, rightPinnedWidth, topPinnedHeight, bottomPinnedHeight]);
   const {
     resizeThrottleMs
   } = params;
@@ -210,6 +221,14 @@ function useDimensions(store, params, _api) {
   const debouncedUpdateDimensions = React.useMemo(() => resizeThrottleMs > 0 ? (0, _throttle.throttle)(updateDimensionCallback, resizeThrottleMs) : undefined, [resizeThrottleMs, updateDimensionCallback]);
   React.useEffect(() => debouncedUpdateDimensions?.clear, [debouncedUpdateDimensions]);
   (0, _useEnhancedEffect.default)(updateDimensions, [updateDimensions]);
+  (0, _useEnhancedEffect.default)(() => {
+    store.update({
+      dimensions: (0, _extends2.default)({}, store.state.dimensions, {
+        autoHeight: params.dimensions.autoHeight,
+        minimalContentHeight: params.dimensions.minimalContentHeight
+      })
+    });
+  }, [store, params.dimensions.autoHeight, params.dimensions.minimalContentHeight]);
   const rowsMeta = useRowsMeta(store, params, updateDimensions);
   return {
     updateDimensions,

package/features/index.d.ts

@@ -2,4 +2,4 @@ export * from "./colspan.js";
 export * from "./dimensions.js";
 export * from "./keyboard.js";
 export * from "./rowspan.js";
-export * from "./virtualization.js";
+export * from "./virtualization/index.js";

package/features/keyboard.d.ts

@@ -1,5 +1,5 @@
 import { Store } from '@mui/x-internals/store';
-import type { BaseState, VirtualizerParams } from "../useVirtualizer.js";
+import type { BaseState, ParamsWithDefaults } from "../useVirtualizer.js";
 export declare const Keyboard: {
   initialize: typeof initializeState;
   use: typeof useKeyboard;
@@ -9,8 +9,8 @@ export declare namespace Keyboard {
   type State = {};
   type API = ReturnType<typeof useKeyboard>;
 }
-declare function initializeState(_params: VirtualizerParams): Keyboard.State;
-declare function useKeyboard(store: Store<BaseState & Keyboard.State>, params: VirtualizerParams, _api: {}): {
+declare function initializeState(_params: ParamsWithDefaults): Keyboard.State;
+declare function useKeyboard(store: Store<BaseState & Keyboard.State>, params: ParamsWithDefaults, _api: {}): {
   getViewportPageSize: () => number;
 };
 export {};

package/features/rowspan.d.ts

@@ -1,7 +1,7 @@
 import { Store } from '@mui/x-internals/store';
-import type { BaseState, VirtualizerParams } from "../useVirtualizer.js";
+import type { BaseState, ParamsWithDefaults } from "../useVirtualizer.js";
 import type { RowSpanningState } from "../models/rowspan.js";
-import { Virtualization } from "./virtualization.js";
+import { Virtualization } from "./virtualization/index.js";
 export declare const Rowspan: {
   initialize: typeof initializeState;
   use: typeof useRowspan;
@@ -18,8 +18,8 @@ export declare namespace Rowspan {
   };
   type API = ReturnType<typeof useRowspan>;
 }
-declare function initializeState(params: VirtualizerParams): Rowspan.State;
-declare function useRowspan(store: Store<BaseState & Rowspan.State>, _params: VirtualizerParams, _api: Virtualization.API): {
+declare function initializeState(params: ParamsWithDefaults): Rowspan.State;
+declare function useRowspan(store: Store<BaseState & Rowspan.State>, _params: ParamsWithDefaults, _api: Virtualization.API): {
   getHiddenCellsOrigin: () => Record<number, Record<number, number>>;
 };
 export {};

package/features/virtualization/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./layout.js";
+export * from "./virtualization.js";

package/features/virtualization/index.js

@@ -0,0 +1,27 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+var _layout = require("./layout");
+Object.keys(_layout).forEach(function (key) {
+  if (key === "default" || key === "__esModule") return;
+  if (key in exports && exports[key] === _layout[key]) return;
+  Object.defineProperty(exports, key, {
+    enumerable: true,
+    get: function () {
+      return _layout[key];
+    }
+  });
+});
+var _virtualization = require("./virtualization");
+Object.keys(_virtualization).forEach(function (key) {
+  if (key === "default" || key === "__esModule") return;
+  if (key in exports && exports[key] === _virtualization[key]) return;
+  Object.defineProperty(exports, key, {
+    enumerable: true,
+    get: function () {
+      return _virtualization[key];
+    }
+  });
+});

package/features/virtualization/layout.d.ts

@@ -0,0 +1,129 @@
+import * as React from 'react';
+import { Store } from '@mui/x-internals/store';
+import { Dimensions } from "../dimensions.js";
+import { Virtualization, type VirtualizationLayoutParams } from "./virtualization.js";
+import type { BaseState, ParamsWithDefaults } from "../../useVirtualizer.js";
+type RequiredAPI = Dimensions.API;
+type BaseElements = {
+  scroller: React.RefObject<HTMLElement | null>;
+  container: React.RefObject<HTMLElement | null>;
+};
+type AnyElements = BaseElements & Record<string, React.RefObject<HTMLElement | null>>;
+export declare abstract class Layout<E extends AnyElements = AnyElements> {
+  static elements: readonly (keyof AnyElements)[];
+  refs: E;
+  constructor(refs: E);
+  abstract use(store: Store<BaseState>, params: ParamsWithDefaults, api: RequiredAPI, layoutParams: VirtualizationLayoutParams): any;
+  refSetter(name: keyof E): (node: HTMLDivElement | null) => void;
+}
+type DataGridElements = BaseElements & {
+  scrollbarVertical: React.RefObject<HTMLElement | null>;
+  scrollbarHorizontal: React.RefObject<HTMLElement | null>;
+};
+export declare class LayoutDataGrid extends Layout<DataGridElements> {
+  static elements: readonly ["scroller", "container", "content", "positioner", "scrollbarVertical", "scrollbarHorizontal"];
+  use(store: Store<BaseState>, _params: ParamsWithDefaults, _api: RequiredAPI, layoutParams: VirtualizationLayoutParams): void;
+  static selectors: {
+    containerProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      ref: any;
+    };
+    scrollerProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      ref: any;
+      style: {
+        overflowX: string | undefined;
+        overflowY: string | undefined;
+      };
+      role: string;
+      tabIndex: number | undefined;
+    };
+    contentProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      style: React.CSSProperties;
+      role: string;
+    };
+    positionerProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      style: {
+        transform: string;
+      };
+    };
+    scrollbarHorizontalProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      ref: any;
+      scrollPosition: {
+        current: import("../../models/index.js").ScrollPosition;
+      };
+    };
+    scrollbarVerticalProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      ref: any;
+      scrollPosition: {
+        current: import("../../models/index.js").ScrollPosition;
+      };
+    };
+    scrollAreaProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      scrollPosition: {
+        current: import("../../models/index.js").ScrollPosition;
+      };
+    };
+  };
+}
+export declare class LayoutDataGridLegacy extends LayoutDataGrid {
+  use(store: Store<BaseState>, _params: ParamsWithDefaults, _api: RequiredAPI, layoutParams: VirtualizationLayoutParams): {
+    getContainerProps: () => {
+      ref: any;
+    };
+    getScrollerProps: () => {
+      ref: any;
+      style: {
+        overflowX: string | undefined;
+        overflowY: string | undefined;
+      };
+      role: string;
+      tabIndex: number | undefined;
+    };
+    getContentProps: () => {
+      style: React.CSSProperties;
+      role: string;
+    };
+    getPositionerProps: () => {
+      style: {
+        transform: string;
+      };
+    };
+    getScrollbarVerticalProps: () => {
+      ref: any;
+      scrollPosition: {
+        current: import("../../models/index.js").ScrollPosition;
+      };
+    };
+    getScrollbarHorizontalProps: () => {
+      ref: any;
+      scrollPosition: {
+        current: import("../../models/index.js").ScrollPosition;
+      };
+    };
+    getScrollAreaProps: () => {
+      scrollPosition: {
+        current: import("../../models/index.js").ScrollPosition;
+      };
+    };
+  };
+}
+type ListElements = BaseElements;
+export declare class LayoutList extends Layout<ListElements> {
+  static elements: readonly ["scroller", "container", "content", "positioner"];
+  use(store: Store<BaseState>, _params: ParamsWithDefaults, _api: RequiredAPI, layoutParams: VirtualizationLayoutParams): void;
+  static selectors: {
+    containerProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      ref: any;
+      style: React.CSSProperties;
+      role: string;
+      tabIndex: number | undefined;
+    };
+    contentProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      style: React.CSSProperties;
+      role: string;
+    };
+    positionerProps: (args_0: Virtualization.State<Layout<AnyElements>> & Dimensions.State) => {
+      style: React.CSSProperties;
+    };
+  };
+}
+export {};

package/features/virtualization/layout.js

@@ -0,0 +1,163 @@
+"use strict";
+
+var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.LayoutList = exports.LayoutDataGridLegacy = exports.LayoutDataGrid = exports.Layout = void 0;
+var _useForkRef = _interopRequireDefault(require("@mui/utils/useForkRef"));
+var _useEventCallback = _interopRequireDefault(require("@mui/utils/useEventCallback"));
+var platform = _interopRequireWildcard(require("@mui/x-internals/platform"));
+var _store = require("@mui/x-internals/store");
+var _dimensions = require("../../features/dimensions");
+var _virtualization = require("./virtualization");
+/* eslint-disable react-hooks/rules-of-hooks */
+
+class Layout {
+  static elements = ['scroller', 'container'];
+  constructor(refs) {
+    this.refs = refs;
+  }
+  refSetter(name) {
+    return node => {
+      if (node && this.refs[name].current !== node) {
+        this.refs[name].current = node;
+      }
+    };
+  }
+}
+exports.Layout = Layout;
+class LayoutDataGrid extends Layout {
+  static elements = ['scroller', 'container', 'content', 'positioner', 'scrollbarVertical', 'scrollbarHorizontal'];
+  use(store, _params, _api, layoutParams) {
+    const {
+      scrollerRef,
+      containerRef
+    } = layoutParams;
+    const scrollbarVerticalRef = (0, _useEventCallback.default)(this.refSetter('scrollbarVertical'));
+    const scrollbarHorizontalRef = (0, _useEventCallback.default)(this.refSetter('scrollbarHorizontal'));
+    store.state.virtualization.context = {
+      scrollerRef,
+      containerRef,
+      scrollbarVerticalRef,
+      scrollbarHorizontalRef
+    };
+  }
+  static selectors = {
+    containerProps: (0, _store.createSelectorMemoized)(_virtualization.Virtualization.selectors.context, context => ({
+      ref: context.containerRef
+    })),
+    scrollerProps: (0, _store.createSelectorMemoized)(_virtualization.Virtualization.selectors.context, _dimensions.Dimensions.selectors.autoHeight, _dimensions.Dimensions.selectors.needsHorizontalScrollbar, (context, autoHeight, needsHorizontalScrollbar) => ({
+      ref: context.scrollerRef,
+      style: {
+        overflowX: !needsHorizontalScrollbar ? 'hidden' : undefined,
+        overflowY: autoHeight ? 'hidden' : undefined
+      },
+      role: 'presentation',
+      // `tabIndex` shouldn't be used along role=presentation, but it fixes a Firefox bug
+      // https://github.com/mui/mui-x/pull/13891#discussion_r1683416024
+      tabIndex: platform.isFirefox ? -1 : undefined
+    })),
+    contentProps: (0, _store.createSelectorMemoized)(_dimensions.Dimensions.selectors.contentHeight, _dimensions.Dimensions.selectors.minimalContentHeight, _dimensions.Dimensions.selectors.columnsTotalWidth, _dimensions.Dimensions.selectors.needsHorizontalScrollbar, (contentHeight, minimalContentHeight, columnsTotalWidth, needsHorizontalScrollbar) => ({
+      style: {
+        width: needsHorizontalScrollbar ? columnsTotalWidth : 'auto',
+        flexBasis: contentHeight === 0 ? minimalContentHeight : contentHeight,
+        flexShrink: 0
+      },
+      role: 'presentation'
+    })),
+    positionerProps: (0, _store.createSelectorMemoized)(_virtualization.Virtualization.selectors.offsetTop, offsetTop => ({
+      style: {
+        transform: `translate3d(0, ${offsetTop}px, 0)`
+      }
+    })),
+    scrollbarHorizontalProps: (0, _store.createSelectorMemoized)(_virtualization.Virtualization.selectors.context, _virtualization.Virtualization.selectors.scrollPosition, (context, scrollPosition) => ({
+      ref: context.scrollbarHorizontalRef,
+      scrollPosition
+    })),
+    scrollbarVerticalProps: (0, _store.createSelectorMemoized)(_virtualization.Virtualization.selectors.context, _virtualization.Virtualization.selectors.scrollPosition, (context, scrollPosition) => ({
+      ref: context.scrollbarVerticalRef,
+      scrollPosition
+    })),
+    scrollAreaProps: (0, _store.createSelectorMemoized)(_virtualization.Virtualization.selectors.scrollPosition, scrollPosition => ({
+      scrollPosition
+    }))
+  };
+}
+
+// The current virtualizer API is exposed on one of the DataGrid slots, so we need to keep
+// the old API for backward compatibility. This API prevents using fine-grained reactivity
+// as all props are returned in a single object, so everything re-renders on any change.
+//
+// TODO(v9): Remove the legacy API.
+exports.LayoutDataGrid = LayoutDataGrid;
+class LayoutDataGridLegacy extends LayoutDataGrid {
+  use(store, _params, _api, layoutParams) {
+    super.use(store, _params, _api, layoutParams);
+    const containerProps = store.use(LayoutDataGrid.selectors.containerProps);
+    const scrollerProps = store.use(LayoutDataGrid.selectors.scrollerProps);
+    const contentProps = store.use(LayoutDataGrid.selectors.contentProps);
+    const positionerProps = store.use(LayoutDataGrid.selectors.positionerProps);
+    const scrollbarVerticalProps = store.use(LayoutDataGrid.selectors.scrollbarVerticalProps);
+    const scrollbarHorizontalProps = store.use(LayoutDataGrid.selectors.scrollbarHorizontalProps);
+    const scrollAreaProps = store.use(LayoutDataGrid.selectors.scrollAreaProps);
+    return {
+      getContainerProps: () => containerProps,
+      getScrollerProps: () => scrollerProps,
+      getContentProps: () => contentProps,
+      getPositionerProps: () => positionerProps,
+      getScrollbarVerticalProps: () => scrollbarVerticalProps,
+      getScrollbarHorizontalProps: () => scrollbarHorizontalProps,
+      getScrollAreaProps: () => scrollAreaProps
+    };
+  }
+}
+exports.LayoutDataGridLegacy = LayoutDataGridLegacy;
+class LayoutList extends Layout {
+  static elements = ['scroller', 'container', 'content', 'positioner'];
+  use(store, _params, _api, layoutParams) {
+    const {
+      scrollerRef,
+      containerRef
+    } = layoutParams;
+    const mergedRef = (0, _useForkRef.default)(scrollerRef, containerRef);
+    store.state.virtualization.context = {
+      mergedRef
+    };
+  }
+  static selectors = {
+    containerProps: (0, _store.createSelectorMemoized)(_virtualization.Virtualization.selectors.context, _dimensions.Dimensions.selectors.autoHeight, _dimensions.Dimensions.selectors.needsHorizontalScrollbar, (context, autoHeight, needsHorizontalScrollbar) => ({
+      ref: context.mergedRef,
+      style: {
+        overflowX: !needsHorizontalScrollbar ? 'hidden' : undefined,
+        overflowY: autoHeight ? 'hidden' : undefined,
+        position: 'relative'
+      },
+      role: 'presentation',
+      // `tabIndex` shouldn't be used along role=presentation, but it fixes a Firefox bug
+      // https://github.com/mui/mui-x/pull/13891#discussion_r1683416024
+      tabIndex: platform.isFirefox ? -1 : undefined
+    })),
+    contentProps: (0, _store.createSelectorMemoized)(_dimensions.Dimensions.selectors.contentHeight, contentHeight => {
+      return {
+        style: {
+          position: 'absolute',
+          display: 'inline-block',
+          width: '100%',
+          height: contentHeight,
+          top: 0,
+          left: 0,
+          zIndex: -1
+        },
+        role: 'presentation'
+      };
+    }),
+    positionerProps: (0, _store.createSelectorMemoized)(_virtualization.Virtualization.selectors.offsetTop, offsetTop => ({
+      style: {
+        height: offsetTop
+      }
+    }))
+  };
+}
+exports.LayoutList = LayoutList;

package/features/{virtualization.d.ts → virtualization/virtualization.d.ts}

@@ -1,10 +1,11 @@
 import * as React from 'react';
 import type { integer } from '@mui/x-internals/types';
 import { Store } from '@mui/x-internals/store';
-import type { CellColSpanInfo } from "../models/colspan.js";
-import { Dimensions } from "./dimensions.js";
-import type { BaseState, VirtualizerParams } from "../useVirtualizer.js";
-import { PinnedRowPosition, RenderContext, ColumnsRenderContext, ColumnWithWidth, RowId, RowEntry } from "../models/index.js";
+import type { CellColSpanInfo } from "../../models/colspan.js";
+import { Dimensions } from "../dimensions.js";
+import type { BaseState, ParamsWithDefaults } from "../../useVirtualizer.js";
+import type { Layout } from "./layout.js";
+import { RenderContext, ColumnsRenderContext, ColumnWithWidth, RowId, ScrollPosition } from "../../models/index.js";
 export type VirtualizationParams = {
   /** @default false */
   isRtl?: boolean;
@@ -15,11 +16,16 @@ export type VirtualizationParams = {
    * @default 150 */
   columnBufferPx?: number;
 };
-export type VirtualizationState = {
+export type VirtualizationState<K extends string = string> = {
   enabled: boolean;
   enabledForRows: boolean;
   enabledForColumns: boolean;
   renderContext: RenderContext;
+  props: Record<K, Record<string, any>>;
+  context: Record<string, any>;
+  scrollPosition: {
+    current: ScrollPosition;
+  };
 };
 export declare const EMPTY_RENDER_CONTEXT: {
   firstRowIndex: number;
@@ -31,20 +37,31 @@ export declare const Virtualization: {
   initialize: typeof initializeState;
   use: typeof useVirtualization;
   selectors: {
-    store: (state: BaseState) => VirtualizationState;
+    store: (state: BaseState) => VirtualizationState<string>;
     renderContext: (state: BaseState) => RenderContext;
     enabledForRows: (state: BaseState) => boolean;
     enabledForColumns: (state: BaseState) => boolean;
+    offsetTop: (args_0: Virtualization.State<Layout<{
+      scroller: React.RefObject<HTMLElement | null>;
+      container: React.RefObject<HTMLElement | null>;
+    } & Record<string, React.RefObject<HTMLElement | null>>>> & Dimensions.State) => number;
+    context: (state: BaseState) => Record<string, any>;
+    scrollPosition: (state: BaseState) => {
+      current: ScrollPosition;
+    };
   };
 };
 export declare namespace Virtualization {
-  type State = {
-    virtualization: VirtualizationState;
+  type State<L extends Layout> = {
+    virtualization: VirtualizationState<L extends Layout<infer E> ? keyof E : string>;
     getters: ReturnType<typeof useVirtualization>['getters'];
   };
   type API = ReturnType<typeof useVirtualization>;
 }
-declare function initializeState(params: VirtualizerParams): Virtualization.State;
+declare function initializeState(params: ParamsWithDefaults): Virtualization.State<Layout<{
+  scroller: React.RefObject<HTMLElement | null>;
+  container: React.RefObject<HTMLElement | null>;
+} & Record<string, React.RefObject<HTMLElement | null>>>>;
 /** APIs to override for colspan/rowspan */
 type AbstractAPI = {
   getCellColSpanInfo: (rowId: RowId, columnIndex: integer) => CellColSpanInfo;
@@ -52,58 +69,18 @@ type AbstractAPI = {
   getHiddenCellsOrigin: () => Record<RowId, Record<number, number>>;
 };
 type RequiredAPI = Dimensions.API & AbstractAPI;
-declare function useVirtualization(store: Store<BaseState>, params: VirtualizerParams, api: RequiredAPI): {
-  getters: {
-    setPanels: React.Dispatch<React.SetStateAction<Readonly<Map<any, React.ReactNode>>>>;
-    getOffsetTop: () => number;
-    getRows: (rowParams?: {
-      rows?: RowEntry[];
-      position?: PinnedRowPosition;
-      renderContext?: RenderContext;
-    }, unstable_rowTree?: Record<RowId, any>) => React.ReactNode[];
-    rows: RowEntry[];
-    getContainerProps: () => {
-      ref: (node: HTMLDivElement | null) => (() => void | undefined) | undefined;
-    };
-    getScrollerProps: () => {
-      ref: (node: HTMLDivElement | null) => (() => void | undefined) | undefined;
-      style: React.CSSProperties;
-      role: string;
-      tabIndex: number | undefined;
-    };
-    getContentProps: () => {
-      ref: (node: HTMLDivElement | null) => void;
-      style: React.CSSProperties;
-      role: string;
-    };
-    getScrollbarVerticalProps: () => {
-      ref: (node: HTMLDivElement | null) => void;
-      scrollPosition: React.RefObject<{
-        top: number;
-        left: number;
-      }>;
-    };
-    getScrollbarHorizontalProps: () => {
-      ref: (node: HTMLDivElement | null) => void;
-      scrollPosition: React.RefObject<{
-        top: number;
-        left: number;
-      }>;
-    };
-    getScrollAreaProps: () => {
-      scrollPosition: React.RefObject<{
-        top: number;
-        left: number;
-      }>;
-    };
-  };
-  useVirtualization: () => BaseState;
-  setPanels: React.Dispatch<React.SetStateAction<Readonly<Map<any, React.ReactNode>>>>;
-  forceUpdateRenderContext: () => void;
-  scheduleUpdateRenderContext: () => void;
+export type VirtualizationLayoutParams = {
+  containerRef: (node: HTMLDivElement | null) => void;
+  scrollerRef: (node: HTMLDivElement | null) => void;
+};
+declare function useVirtualization(store: Store<BaseState>, params: ParamsWithDefaults, api: RequiredAPI): {
   getCellColSpanInfo: (rowId: RowId, columnIndex: integer) => CellColSpanInfo;
   calculateColSpan: (rowId: RowId, minFirstColumn: integer, maxLastColumn: integer, columns: ColumnWithWidth[]) => void;
   getHiddenCellsOrigin: () => Record<RowId, Record<number, number>>;
+  getters: any;
+  setPanels: React.Dispatch<React.SetStateAction<Readonly<Map<any, React.ReactNode>>>>;
+  forceUpdateRenderContext: () => void;
+  scheduleUpdateRenderContext: () => void;
 };
 export declare function areRenderContextsEqual(context1: RenderContext, context2: RenderContext): boolean;
 export declare function computeOffsetLeft(columnPositions: number[], renderContext: ColumnsRenderContext, pinnedLeftLength: number): number;