antd
Version:
An enterprise-class UI design language and React components implementation
31 lines • 1.51 kB
JavaScript
import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray";
// Disabled the rule since `fill` is safe here
// but `Array.from` will increase bundle size.
/* eslint-disable unicorn/no-new-array */
import * as React from 'react';
/**
* Auto arrange the items in the masonry layout.
* Always get stable positions by order
* instead of dynamic adjust for next item height.
*/
export default function usePositions(itemHeights, columnCount, verticalGutter) {
// ==================== Auto Order ====================
const [orderItemPositions, orderTotalHeight] = React.useMemo(() => {
const columnHeights = new Array(columnCount).fill(0);
const itemPositions = new Map();
for (let i = 0; i < itemHeights.length; i += 1) {
const [itemKey, itemHeight, itemColumn] = itemHeights[i];
let targetColumnIndex = itemColumn ?? columnHeights.indexOf(Math.min.apply(Math, _toConsumableArray(columnHeights)));
targetColumnIndex = Math.min(targetColumnIndex, columnCount - 1);
const top = columnHeights[targetColumnIndex];
itemPositions.set(itemKey, {
column: targetColumnIndex,
top
});
columnHeights[targetColumnIndex] += itemHeight + verticalGutter;
}
return [itemPositions, Math.max(0, Math.max.apply(Math, _toConsumableArray(columnHeights)) - verticalGutter)];
}, [columnCount, itemHeights, verticalGutter]);
// ====================== Return ======================
return [orderItemPositions, orderTotalHeight];
}