Class: BipartiteGraph::CompleteGraph
- Inherits:
-
SimpleDelegator
- Object
- SimpleDelegator
- BipartiteGraph::CompleteGraph
- Defined in:
- lib/bipartite_graph/complete_graph.rb
Defined Under Namespace
Classes: FakeNode
Instance Method Summary collapse
-
#initialize(graph) ⇒ CompleteGraph
constructor
A new instance of CompleteGraph.
Constructor Details
#initialize(graph) ⇒ CompleteGraph
Returns a new instance of CompleteGraph.
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
# File 'lib/bipartite_graph/complete_graph.rb', line 7 def initialize(graph) @original_graph = graph @complete_graph = Graph.new sinks = graph.sinks.dup sink_count = sinks.length sources = graph.sources.dup source_count = sources.length if sink_count > source_count (sink_count - source_count).times { sources << FakeNode.new } elsif source_count > sink_count (source_count - sink_count).times { sinks << FakeNode.new } end sources.each do |source| sinks.each do |sink| edge = graph.edge_between(source, sink) weight = edge ? edge.weight : 0 @complete_graph.add_edge(source, sink, weight) end end super(@complete_graph) end |