panel-scroll-group.ts

import type { IGroup } from '@antv/g-canvas';
import { updateMergedCells } from '../utils/interaction/merge-cell';
import type { GridInfo } from '../common/interface';
import type { MergedCell } from './../cell/merged-cell';
import { KEY_GROUP_MERGED_CELLS } from './../common/constant/basic';
import { GridGroup } from './grid-group';

export class PanelScrollGroup extends GridGroup {
  protected mergedCellsGroup: IGroup;

  constructor(cfg) {
    super(cfg);
    this.initMergedCellsGroup();
  }

  protected initMergedCellsGroup() {
    if (this.mergedCellsGroup && this.findById(KEY_GROUP_MERGED_CELLS)) {
      return;
    }

    this.mergedCellsGroup = this.addGroup({
      id: KEY_GROUP_MERGED_CELLS,
    });
  }

  updateMergedCells() {
    this.initMergedCellsGroup();
    updateMergedCells(this.s2, this.mergedCellsGroup);
    this.mergedCellsGroup.toFront();
  }

  addMergeCell(mergeCell: MergedCell) {
    this.mergedCellsGroup?.add(mergeCell);
  }

  update(gridInfo: GridInfo) {
    this.updateGrid(gridInfo);
    this.updateMergedCells();
  }
}