rbgraph 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bf0651e43311d3e309fc16dc6a295cbaa6a22d01
-  data.tar.gz: 74de94cc9f37c30a21e0dd71932362502e67fb92
+  metadata.gz: 57cd18654388a00de3e34afe32243bf721820d5d
+  data.tar.gz: 03dbcda9b0895503528e6e56d5ebb6d42b4c5acf
 SHA512:
-  metadata.gz: c10c70096241736953e7dc983f897b6e72749280a79f02830b999b783ba4c46dd6b0b16b46c35ad89ba46b4cbea905bb1735300381861689188b268725a09cef
-  data.tar.gz: 4d7543bae183a0f6b17bf9e06bc86746376c8e80cdd7fa884ff36e413718257835844e1a44230e09b5739b326bd7b471eeb469352ca2252b9c25b0833cffcb2a
+  metadata.gz: aed7cdcf511eb6a0cda1e15732714d37e2357c69a3790d29a5aa18c7b6f81312f5f103bb1cd1eaa9fd02caa737ba437e5f478a38de35f19df53dc8da29eafce3
+  data.tar.gz: 3528826955f305a39c595c97d5ae6b69c40ef3c43f569384549a1e983d00a6d158670952286d51522c4978efa59993f3d90044b0f5fd9930fe730d85ff3e42e4

data/lib/rbgraph/edge.rb

@@ -43,6 +43,10 @@ module Rbgraph
       node == node1 ? node2 : node1
     end
 
+    def different_node(node)
+      ([node1, node2] - [node]).first
+    end
+
     def merge!(edge, &block)
       self.weight += edge.weight unless edge.weight.nil?
       raise "Cannot merging edges of different kind!" if kind != edge.kind

data/lib/rbgraph/node.rb

@@ -69,6 +69,42 @@ module Rbgraph
       edges.select { |eid, edge| edge.in_for?(self) }
     end
 
+    def parent
+      if graph.directed?
+        incoming_edges_arr = incoming_edges.values
+        case incoming_edges_arr.size
+        when 0 then nil
+        when 1 then incoming_edges_arr.first.different_node(self) ||
+            raise("Node #{id} is connected to self!")
+        else
+          raise "Node #{id} has more than 1 incoming edges!"
+        end
+      end
+    end
+
+    def ancestors
+      if graph.directed?
+        nodes = {}
+        up = parent
+        nodes[up.id] = up unless up.nil?
+        while !up.parent.nil?
+          up = up.parent
+          if nodes[up.id].nil?
+            nodes[up.id] = up
+          else
+            raise "Cycle detected while getting ancestors of #{id}!"
+          end
+        end
+        nodes
+      else
+        {}
+      end
+    end
+
+    def root
+      ancestors.values.last
+    end
+
     def out_degree
       outgoing_edges.size
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rbgraph
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - George Lamprianidis