X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/2aaa919aa7451546f9576b3a71fc41f896c24b31..5b3677b425b9cc6949c1573d59ac772540cbf4b2:/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