@antv/s2
Version:
effective spreadsheet render core lib
113 lines • 4.87 kB
JavaScript
import { isEmpty, range } from 'lodash';
import { S2Event } from '../../common/constant';
import { CellType, InteractionBrushSelectionStage, InteractionName, InteractionStateName, } from '../../common/constant/interaction';
import { getDataCellId } from '../../utils';
import { afterSelectDataCells } from '../../utils/interaction/select-event';
import { BaseBrushSelection } from './base-brush-selection';
export class DataCellBrushSelection extends BaseBrushSelection {
constructor() {
super(...arguments);
this.displayedCells = [];
this.brushRangeCells = [];
this.getSelectedCellMetas = (brushRange) => {
const { facet } = this.spreadsheet;
const metas = [];
const rowLeafNodes = facet.getRowLeafNodes();
const colLeafNodes = facet.getColLeafNodes();
const rowIndexRange = range(brushRange.start.rowIndex, brushRange.end.rowIndex + 1);
const colIndexRange = range(brushRange.start.colIndex, brushRange.end.colIndex + 1);
rowIndexRange.forEach((rowIndex) => {
colIndexRange.forEach((colIndex) => {
const colId = String(colLeafNodes[colIndex].id);
const rowId = isEmpty(rowLeafNodes)
? String(rowIndex)
: String(rowLeafNodes[rowIndex].id);
metas.push({
colIndex,
rowIndex,
id: getDataCellId(rowId, colId),
type: CellType.DATA_CELL,
rowId,
colId,
spreadsheet: this.spreadsheet,
});
});
});
return metas;
};
}
bindMouseDown() {
this.spreadsheet.on(S2Event.DATA_CELL_MOUSE_DOWN, (event) => {
if (!this.spreadsheet.interaction.getBrushSelection().dataCell) {
return;
}
super.mouseDown(event);
this.resetScrollDelta();
});
}
bindMouseMove() {
this.spreadsheet.on(S2Event.GLOBAL_MOUSE_MOVE, (event) => {
if (this.brushSelectionStage === InteractionBrushSelectionStage.UN_DRAGGED) {
return;
}
this.setBrushSelectionStage(InteractionBrushSelectionStage.DRAGGED);
const pointInCanvas = this.spreadsheet.interaction.eventController.getViewportPoint(event);
if (this.autoBrushScroll(pointInCanvas)) {
return;
}
this.renderPrepareSelected(pointInCanvas);
});
}
isInBrushRange(meta) {
const { start, end } = this.getBrushRange();
const { rowIndex, colIndex } = meta;
return (rowIndex >= start.rowIndex &&
rowIndex <= end.rowIndex &&
colIndex >= start.colIndex &&
colIndex <= end.colIndex);
}
// 最终刷选的 cell
updateSelectedCells(event) {
const brushRange = this.getBrushRange();
const selectedCellMetas = this.getSelectedCellMetas(brushRange);
this.spreadsheet.interaction.changeState({
cells: selectedCellMetas,
stateName: InteractionStateName.DATA_CELL_BRUSH_SELECTED,
onUpdateCells: afterSelectDataCells,
});
const scrollBrushRangeCells = this.getScrollBrushRangeCells(selectedCellMetas);
this.emitBrushSelectionEvent(S2Event.DATA_CELL_BRUSH_SELECTION, scrollBrushRangeCells, {
event,
targetCell: scrollBrushRangeCells[0],
interactionName: InteractionName.DATA_CELL_BRUSH_SELECTION,
});
}
/**
* @name 获取刷选 (含滚动后不再可视范围内) 的单元格
* @description DataCell 存在滚动刷选, 由于按需加载的特性, 非可视范围内的单元格已被注销
* 如果在可视范围, 直接返回 DataCell, 非可视范围, 由于实例已被销毁, 构造实例后返回
*/
getScrollBrushRangeCells(selectedCellMetas) {
return selectedCellMetas.map((meta) => {
const visibleCell = this.getVisibleBrushRangeCells(meta.id);
if (visibleCell) {
return visibleCell;
}
const viewMeta = this.spreadsheet.facet.getCellMeta(meta.rowIndex, meta.colIndex);
return this.spreadsheet.facet.createDataCell(viewMeta);
});
}
bindMouseUp() {
super.bindMouseUp(true);
}
getPrepareSelectMaskPosition(brushRange) {
const { minX, minY } = this.spreadsheet.facet.panelBBox;
const x = Math.max(brushRange.start.x, minX);
const y = Math.max(brushRange.start.y, minY);
return {
x,
y,
};
}
}
//# sourceMappingURL=data-cell-brush-selection.js.map