Merge branch 'master' of https://github.com/ilyakharlamov/graphlib in…

…to filterNodes
gardner · Jul 10, 2015 · 8905ce8 · 8905ce8
2 parents bdbe857 + 5673047
commit 8905ce8
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/lib/graph.js b/lib/graph.js
@@ -270,6 +270,27 @@ Graph.prototype.neighbors = function(v) {
   }
 };
 
+Graph.prototype.eachNode = function(func) {
+  for (var id in this._nodes) {
+    func(id, this._nodes[id]);
+  }
+};
+
+Graph.prototype.filterNodes = function(filter) {
+  var copy = new this.constructor();
+  copy.graph(this.graph());
+  this.eachNode(function(u, value) {
+    if (filter(u)) {
+      copy.setNode(u, value);
+    }
+  });
+  this.eachEdge(function(id, v, w, value) {
+    if (copy.hasNode(v) && copy.hasNode(w)) {
+      copy.setEdge(v, w, value);
+    }
+  });
+  return copy;
+};
 /* === Edge functions ========== */
 
 Graph.prototype.setDefaultEdgeLabel = function(newDefault) {
@@ -430,6 +451,14 @@ Graph.prototype.nodeEdges = function(v, w) {
   }
 };
 
+Graph.prototype.eachEdge = function(func) {
+  var id, edge;
+  for (id in this._edgeObjs) {
+    edge=this._edgeObjs[id];
+    func(id, edge.v, edge.w, this._edgeLabels[id]);
+  }
+};
+
 function incrementOrInitEntry(map, k) {
   if (_.has(map, k)) {
     map[k]++;