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.
79 lines
4.1 KiB
79 lines
4.1 KiB
import { COMBO_KEY } from '../constants';
|
|
import { isCollapsed } from '../utils/collapsibility';
|
|
import { getSubgraphRelatedEdges } from '../utils/edge';
|
|
import { idOf } from '../utils/id';
|
|
import { BaseTransform } from './base-transform';
|
|
import { reassignTo } from './utils';
|
|
/**
|
|
* <zh/> 处理组合的收起和展开
|
|
*
|
|
* <en/> Process the collapse and expand of combos
|
|
*/
|
|
export class CollapseExpandCombo extends BaseTransform {
|
|
beforeDraw(input, context) {
|
|
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) => {
|
|
var _a;
|
|
const id = idOf(edge);
|
|
const edgeElement = (_a = this.context.element) === null || _a === void 0 ? void 0 : _a.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) => {
|
|
var _a;
|
|
const id = idOf(descendant);
|
|
const elementType = model.getElementType(id);
|
|
const element = (_a = this.context.element) === null || _a === void 0 ? void 0 : _a.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]);
|
|
});
|
|
}
|
|
}
|
|
return input;
|
|
}
|
|
}
|
|
//# sourceMappingURL=collapse-expand-combo.js.map
|