@antv/g2plot
Version:
An interactive and responsive charting library
69 lines • 2.58 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.getAllNodes = exports.getField = exports.NODE_ANCESTORS_FIELD = exports.CHILD_NODE_COUNT = exports.NODE_INDEX_FIELD = void 0;
var util_1 = require("@antv/util");
/** export 一些字段常量 */
/** 在同层级，同一父节点下的节点索引顺序 */
exports.NODE_INDEX_FIELD = 'nodeIndex';
/** child 节点数量 */
exports.CHILD_NODE_COUNT = 'childNodeCount';
/** 节点的祖先节点 */
exports.NODE_ANCESTORS_FIELD = 'nodeAncestor';
var INVALID_FIELD_ERR_MSG = 'Invalid field: it must be a string!';
function getField(options, defaultField) {
var field = options.field, fields = options.fields;
if (util_1.isString(field)) {
return field;
}
if (util_1.isArray(field)) {
console.warn(INVALID_FIELD_ERR_MSG);
return field[0];
}
console.warn(INVALID_FIELD_ERR_MSG + " will try to get fields instead.");
if (util_1.isString(fields)) {
return fields;
}
if (util_1.isArray(fields) && fields.length) {
return fields[0];
}
if (defaultField) {
return defaultField;
}
throw new TypeError(INVALID_FIELD_ERR_MSG);
}
exports.getField = getField;
function getAllNodes(root) {
var nodes = [];
if (root && root.each) {
var parent_1;
var index_1;
// d3-hierarchy: Invokes the specified function for node and each descendant in **breadth-first order**
root.each(function (node) {
var _a, _b;
if (node.parent !== parent_1) {
parent_1 = node.parent;
index_1 = 0;
}
else {
index_1 += 1;
}
var ancestors = util_1.filter((((_a = node.ancestors) === null || _a === void 0 ? void 0 : _a.call(node)) || []).map(function (d) { return nodes.find(function (n) { return n.name === d.name; }) || d; }), function (_a) {
var depth = _a.depth;
return depth > 0 && depth < node.depth;
});
node[exports.NODE_ANCESTORS_FIELD] = ancestors;
node[exports.CHILD_NODE_COUNT] = ((_b = node.children) === null || _b === void 0 ? void 0 : _b.length) || 0;
node[exports.NODE_INDEX_FIELD] = index_1;
nodes.push(node);
});
}
else if (root && root.eachNode) {
// @antv/hierarchy
root.eachNode(function (node) {
nodes.push(node);
});
}
return nodes;
}
exports.getAllNodes = getAllNodes;
//# sourceMappingURL=util.js.map