X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/c882e24932c718fad85c22cb546f30e158f1b31d..1350b7802163fda0235f43a9f8a79f09b30c0ed2:/examples/msg/chainsend/generate_deployment_file.rb

diff --git a/examples/msg/chainsend/generate_deployment_file.rb b/examples/msg/chainsend/generate_deployment_file.rb
index fae59a8784..e290ae18a0 100755
--- a/examples/msg/chainsend/generate_deployment_file.rb
+++ b/examples/msg/chainsend/generate_deployment_file.rb
@@ -1,5 +1,11 @@
 #!/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
@@ -84,6 +90,14 @@ he = HostsExtractor.new(xml)
 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
 
@@ -91,7 +105,7 @@ puts he.hosts
 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