#!/usr/bin/env ruby
+# Copyright (c) 2012-2014. The SimGrid Team.
+# All rights reserved.
+
+# This program is free software; you can redistribute it and/or modify it
+# under the terms of the license (GNU LGPL) which comes with this package.
+
require 'rexml/document'
class HostsExtractor
raise "Cannot run with less than 2 hosts" unless he.hosts.size > 1
output = ARGV.shift
+n = ARGV.shift
+if n == nil or n.to_i < 2
+ n = he.hosts.size - 1
+else
+ n = n.to_i - 1
+end
+puts n
+
dg = DeploymentGenerator.new(output)
dg.write_header
broadcaster = he.hosts.shift
peers = he.hosts
-dg.write_process("Broadcaster", "broadcaster", [broadcaster], [he.hosts.size])
-dg.write_process("Peers", "peer", peers, (1..he.hosts.size))
+dg.write_process("Broadcaster", "broadcaster", [broadcaster], [n])
+dg.write_process("Peers", "peer", peers[0..n-1], (1..n))
dg.write_footer