@antv/s2
Version:
effective spreadsheet render core lib
60 lines • 2.38 kB
JavaScript
import { filter, isEmpty, map, max, repeat, zip } from 'lodash';
import { CellType, NODE_ID_SEPARATOR } from '../../../common';
import { getAllLevels, getHeaderList, getHeaderMeasureFieldNames, } from '../method';
import { matrixHtmlTransformer, matrixPlainTextTransformer } from './common';
/**
* 过滤出 intersection cell 中所有叶子节点的 cellMeta
* @param interactedCells
* @param maxLevel
* @returns {CellMeta[]}
*/
function getLastLevelCells(interactedCells, maxLevel) {
return filter(interactedCells, (cell) => {
const meta = cell.getMeta();
const isLastLevel = meta.level === maxLevel;
const isLastTotal = meta.isTotals && isEmpty(meta.children);
return isLastLevel || isLastTotal;
});
}
/**
* 获取表头圈选后的 header cells 值矩阵
* @param lastLevelCells
* @param maxLevel
* @param allLevel
* @returns {SimpleData[][]}
*/
function getHeaderCellMatrix(lastLevelCells, maxLevel, allLevel) {
return map(lastLevelCells, (cell) => {
const meta = cell.getMeta();
const { id, value, isTotals, level, spreadsheet } = meta;
let cellId = id;
// 为总计小计补齐高度
if (isTotals && level !== maxLevel) {
cellId = id + NODE_ID_SEPARATOR + repeat(value, maxLevel - level);
}
return getHeaderMeasureFieldNames(getHeaderList(cellId, allLevel.size), spreadsheet);
});
}
/**
* 获取表头圈选后的 header cells 值矩阵
* @param {RowCell[] | ColCell[]} interactedCells
* @return {CopyableList}
*/
export function getBrushHeaderCopyable(interactedCells) {
var _a;
// 获取圈选的层级有哪些
const allLevels = getAllLevels(interactedCells);
const maxLevel = (_a = max(Array.from(allLevels))) !== null && _a !== void 0 ? _a : 0;
// 获取最后一层的 cell
const lastLevelCells = getLastLevelCells(interactedCells, maxLevel);
// 拼接选中行列头的内容矩阵
const isCol = interactedCells[0].cellType === CellType.COL_CELL;
const headerCellMatrix = getHeaderCellMatrix(lastLevelCells, maxLevel, allLevels);
// 如果是列头，需要转置
const cellMatrix = isCol ? zip(...headerCellMatrix) : headerCellMatrix;
return [
matrixPlainTextTransformer(cellMatrix),
matrixHtmlTransformer(cellMatrix),
];
}
//# sourceMappingURL=pivot-header-copy.js.map