You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
3.5 KiB
81 lines
3.5 KiB
import { COMBO_KEY } from '../constants';
|
|
import type { DrawContext } from '../runtime/element';
|
|
import type { ComboData } from '../spec';
|
|
import { isCollapsed } from '../utils/collapsibility';
|
|
import { getSubgraphRelatedEdges } from '../utils/edge';
|
|
import { idOf } from '../utils/id';
|
|
import { BaseTransform } from './base-transform';
|
|
import type { DrawData } from './types';
|
|
import { reassignTo } from './utils';
|
|
|
|
/**
|
|
* <zh/> 处理组合的收起和展开
|
|
*
|
|
* <en/> Process the collapse and expand of combos
|
|
*/
|
|
export class CollapseExpandCombo extends BaseTransform {
|
|
public beforeDraw(input: DrawData, context: DrawContext): DrawData {
|
|
if (context.stage === 'visibility') return input;
|
|
if (!this.context.model.model.hasTreeStructure(COMBO_KEY)) return input;
|
|
|
|
const { model } = this.context;
|
|
const { add, update } = input;
|
|
// combo 添加和更新的顺序为先子后父，因此采用倒序遍历
|
|
// The order of adding and updating combos is first child and then parent, so reverse traversal is used
|
|
const combos = [...input.update.combos.entries(), ...input.add.combos.entries()];
|
|
while (combos.length) {
|
|
const [id, combo] = combos.pop()!;
|
|
|
|
if (isCollapsed(combo)) {
|
|
const descendants = model.getDescendantsData(id);
|
|
const descendantIds = descendants.map(idOf);
|
|
const { internal, external } = getSubgraphRelatedEdges(descendantIds, (id) => model.getRelatedEdgesData(id));
|
|
|
|
// 移除所有后代元素 / Remove all descendant elements
|
|
descendants.forEach((descendant) => {
|
|
const descendantId = idOf(descendant);
|
|
// 不再处理当前 combo 的后代 combo
|
|
// No longer process the descendant combo of the current combo
|
|
const comboIndex = combos.findIndex(([id]) => id === descendantId);
|
|
if (comboIndex !== -1) combos.splice(comboIndex, 1);
|
|
|
|
const elementType = model.getElementType(descendantId);
|
|
reassignTo(input, 'remove', elementType, descendant);
|
|
});
|
|
|
|
// 如果是内部边/节点 销毁
|
|
// If it is an internal edge/node, destroy it
|
|
internal.forEach((edge) => reassignTo(input, 'remove', 'edge', edge));
|
|
|
|
// 如果是外部边，连接到收起对象上
|
|
// If it is an external edge, connect to the collapsed object
|
|
external.forEach((edge) => {
|
|
const id = idOf(edge);
|
|
const edgeElement = this.context.element?.getElement(id);
|
|
if (edgeElement) update.edges.set(id, edge);
|
|
else add.edges.set(id, edge);
|
|
});
|
|
} else {
|
|
const children = model.getChildrenData(id);
|
|
const childrenIds = children.map(idOf);
|
|
const { edges } = getSubgraphRelatedEdges(childrenIds, (id) => model.getRelatedEdgesData(id));
|
|
|
|
[...children, ...edges].forEach((descendant) => {
|
|
const id = idOf(descendant);
|
|
const elementType = model.getElementType(id);
|
|
|
|
const element = this.context.element?.getElement(id);
|
|
// 如果节点不存在，则添加到新增列表，如果存在，添加到更新列表
|
|
// If the node does not exist, add it to the new list, if it exists, add it to the update list
|
|
if (element) reassignTo(input, 'update', elementType, descendant);
|
|
else reassignTo(input, 'add', elementType, descendant);
|
|
|
|
// 继续展开子节点 / Continue to expand child nodes
|
|
if (elementType === 'combo') combos.push([id, descendant as ComboData]);
|
|
});
|
|
}
|
|
}
|
|
|
|
return input;
|
|
}
|
|
}
|
|
|