Contents

class Link
  attr_accessor :source, :target, :type

  def initialize(arg)
    @args = arg
    @source = @args.delete(:source)
    @target = @args.delete(:target)
    @type = @args.delete(:type) || 'undirected'

    raise Exception.new("source cant be nil") unless @source
    raise Exception.new("target cant be nil") unless @target
    raise Exception.new("source must be of type Node") unless @source.is_a? Node
    raise Exception.new("target must be of type Node") unless @target.is_a? Node

    @args[:weight] ||= 1

    @source.neighbours << @target
    @target.neighbours << @source unless @type == 'directed'
  end

  def method_missing(name, *args)
    super unless args.empty?
    @args[name]
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
rgraph-0.0.15	lib/rgraph/link.rb
rgraph-0.0.14	lib/rgraph/link.rb
rgraph-0.0.13	lib/rgraph/link.rb
rgraph-0.0.12	lib/rgraph/link.rb
rgraph-0.0.11	lib/rgraph/link.rb
rgraph-0.0.10	lib/rgraph/link.rb