@visactor/vchart-extension 1.13.21-alpha.9 → 1.13.22-alpha.0

package/build/index.js

@@ -4463,6 +4463,15 @@
 	    return type === 'line' || type === 'area';
 	};
 
+	const traverseTree = (subTree, callback) => {
+	    subTree.forEach(node => {
+	        callback(node);
+	        if (node.children && node.children.length) {
+	            traverseTree(node.children, callback);
+	        }
+	    });
+	};
+
 	class CompareSankeyChart extends vchart.SankeyChart {
 	    constructor() {
 	        super(...arguments);
@@ -4675,6 +4684,33 @@
 	    getSubNodeDatum(filter) {
 	        return this._nodeMark.getProductElements().filter((n) => filter(n.data[0]));
 	    }
+	    getNodeDatumInTree(filter) {
+	        const result = [];
+	        const nodes = this._rawData.latestData[0].nodes;
+	        nodes.forEach((node) => {
+	            traverseTree([node], (node) => {
+	                if (filter(node)) {
+	                    result.push(node);
+	                }
+	            });
+	        });
+	        return result;
+	    }
+	    getNodeDatumInSubTree(filter) {
+	        const list = [];
+	        const subNodes = this._rawData.latestData[0].subNode;
+	        subNodes.forEach((subNode) => {
+	            const { nodes } = subNode, result = __rest(subNode, ["nodes"]);
+	            result.datum = [];
+	            nodes.forEach((node) => {
+	                if (filter(node)) {
+	                    result.datum.push(node);
+	                }
+	            });
+	            list.push(result);
+	        });
+	        return list;
+	    }
 	}
 	CompareSankeySeries.type = 'compareSankey';
 	const registerCompareSankeyChart = (option) => {