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.
39 lines
945 B
39 lines
945 B
4 months ago
|
"use strict";
|
||
|
|
|
||
|
|
Object.defineProperty(exports, "__esModule", {
|
||
|
|
value: true
|
||
|
|
});
|
||
|
|
exports.default = void 0;
|
||
|
|
var adjMatrix = function adjMatrix(graphData, directed) {
|
||
|
|
var nodes = graphData.nodes,
|
||
|
|
edges = graphData.edges;
|
||
|
|
var matrix = [];
|
||
|
|
// map node with index in data.nodes
|
||
|
|
var nodeMap = {};
|
||
|
|
if (!nodes) {
|
||
|
|
throw new Error("invalid nodes data!");
|
||
|
|
}
|
||
|
|
if (nodes) {
|
||
|
|
nodes.forEach(function (node, i) {
|
||
|
|
nodeMap[node.id] = i;
|
||
|
|
var row = [];
|
||
|
|
matrix.push(row);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
if (edges) {
|
||
|
|
edges.forEach(function (edge) {
|
||
|
|
var source = edge.source,
|
||
|
|
target = edge.target;
|
||
|
|
var sIndex = nodeMap[source];
|
||
|
|
var tIndex = nodeMap[target];
|
||
|
|
if (!sIndex && sIndex !== 0 || !tIndex && tIndex !== 0) return;
|
||
|
|
matrix[sIndex][tIndex] = 1;
|
||
|
|
if (!directed) {
|
||
|
|
matrix[tIndex][sIndex] = 1;
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
return matrix;
|
||
|
|
};
|
||
|
|
var _default = adjMatrix;
|
||
|
|
exports.default = _default;
|