"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.getDescendantShapes = getDescendantShapes;
exports.getAncestorShapes = getAncestorShapes;
/**
* 获取图形的所有子元素
*
* Get all the child elements of the shape
* @param shape - 图形元素 | shape
* @returns 子元素数组 | child elements array
*/
function getDescendantShapes(shape) {
const succeeds = [];
// 遍历所有子元素,并将子元素的子元素加入到数组中
const traverse = (shape) => {
if (shape === null || shape === void 0 ? void 0 : shape.children.length) {
shape.children.forEach((child) => {
succeeds.push(child);
traverse(child);
});
}
};
traverse(shape);
return succeeds;
}
/**
* 获取图形的所有祖先元素
*
* Get all the ancestor elements of the shape
* @param shape - 图形元素 | shape
* @returns 祖先元素数组 | ancestor elements array
*/
function getAncestorShapes(shape) {
const ancestors = [];
let currentNode = shape.parentNode;
while (currentNode) {
ancestors.push(currentNode);
currentNode = currentNode.parentNode;
}
return ancestors;
}
//# sourceMappingURL=shape.js.map