Method: Network#collect_nodes

Defined in:

lib/NetAnalyzer/network.rb

#collect_nodes(args) ⇒ Object

# File 'lib/NetAnalyzer/network.rb', line 468

def collect_nodes(args)
	nodeIDsA = nil
	nodeIDsB = nil
	if @compute_autorelations
		if args[:layers] == :all
			nodeIDsA = @nodes.keys
		else
			nodeIDsA = []
			args[:layers].each do |layer|
				nodeIDsA.concat(@nodes.select{|id, node| node.type == layer}.keys)
			end
		end
	else
		if args[:layers] != :all
			nodeIDsA = @nodes.select{|id, node| node.type == args[:layers][0]}.keys
			nodeIDsB = @nodes.select{|id, node| node.type == args[:layers][1]}.keys
		end
	end
	return nodeIDsA, nodeIDsB
end