Platform generation: add a bittorent example using a generated platform

diff --git a/examples/msg/bittorrent/CMakeLists.txt b/examples/msg/bittorrent/CMakeLists.txt
index 2fdb8e0..cada54a 100644 (file)
--- a/examples/msg/bittorrent/CMakeLists.txt
+++ b/examples/msg/bittorrent/CMakeLists.txt
@@ -4,9 +4,12 @@ set(EXECUTABLE_OUTPUT_PATH "${CMAKE_CURRENT_BINARY_DIR}")
 
 add_executable(bittorrent
   "bittorrent.c" "messages.c" "peer.c" "tracker.c" "connection.c")
+add_executable(bittorrent_platfgen
+  "bittorrent_platfgen.c" "messages.c" "peer.c" "tracker.c" "connection.c")
 
 ### Add definitions for compile
 target_link_libraries(bittorrent simgrid )
+target_link_libraries(bittorrent_platfgen simgrid )
 
 set(tesh_files
   ${tesh_files}
@@ -21,6 +24,7 @@ set(xml_files
 set(examples_src
   ${examples_src}
   ${CMAKE_CURRENT_SOURCE_DIR}/bittorrent.c
+  ${CMAKE_CURRENT_SOURCE_DIR}/bittorrent_platfgen.c
   ${CMAKE_CURRENT_SOURCE_DIR}/bittorrent.h
   ${CMAKE_CURRENT_SOURCE_DIR}/connection.c
   ${CMAKE_CURRENT_SOURCE_DIR}/connection.h

diff --git a/examples/msg/bittorrent/bittorrent_platfgen.c b/examples/msg/bittorrent/bittorrent_platfgen.c
new file mode 100644 (file)
index 0000000..ff18530
--- /dev/null
+++ b/examples/msg/bittorrent/bittorrent_platfgen.c
@@ -0,0 +1,144 @@
+/* Copyright (c) 2012. 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. */
+#include "bittorrent.h"
+#include "peer.h"
+#include "tracker.h"
+#include <msg/msg.h>
+#include <simgrid/platf_generator.h>
+/**
+ * Bittorrent example launcher, using a generated platform
+ */
+
+static RngStream rng_stream;
+
+void promoter(context_node_t node);
+void labeler(context_edge_t edge);
+void create_environment(int node_count);
+void dispatch_jobs(double tracker_deadline, double peer_deadline, double seed_percentage);
+
+void promoter(context_node_t node) {
+  s_sg_platf_host_cbarg_t host_parameters;
+
+  if(node->degree == 1) {
+    //We promote only the leaf; as we use a star topology, all the nodes
+    //will be promoted except the first one, which will be a router with
+    //every hosts connected on.
+    host_parameters.id = NULL;
+
+    //Power from 3,000,000 to 10,000,000
+    host_parameters.power_peak = 7000000 * RngStream_RandU01(rng_stream) + 3000000;
+    host_parameters.core_amount = 1;
+    host_parameters.power_scale = 1;
+    host_parameters.power_trace = NULL;
+    host_parameters.initial_state = SURF_RESOURCE_ON;
+    host_parameters.state_trace = NULL;
+    host_parameters.coord = NULL;
+    host_parameters.properties = NULL;
+
+    platf_graph_promote_to_host(node, &host_parameters);
+  }
+}
+
+void labeler(context_edge_t edge) {
+
+  s_sg_platf_link_cbarg_t link_parameters;
+  link_parameters.id = NULL;
+
+  //bandwidth from 3,000,000 to 10,000,000
+  link_parameters.bandwidth = 7000000 * RngStream_RandU01(rng_stream) + 3000000;
+  link_parameters.bandwidth_trace = NULL;
+
+  //Latency from 0ms to 100ms
+  link_parameters.latency = RngStream_RandU01(rng_stream) / 10.0;
+  link_parameters.latency_trace = NULL;
+  link_parameters.state = SURF_RESOURCE_ON;
+  link_parameters.state_trace = NULL;
+  link_parameters.policy = SURF_LINK_SHARED;
+  link_parameters.properties = NULL;
+
+  platf_graph_link_label(edge, &link_parameters);
+}
+
+void create_environment(int node_count) {
+
+  platf_graph_uniform(node_count);
+
+  //every nodes are connected to the first one
+  platf_graph_interconnect_star();
+  //No need to check if the graph is connected, the star topology implies it.
+
+  //register promoter and labeler
+  platf_graph_promoter(promoter);
+  platf_graph_labeler(labeler);
+
+  //promoting and labeling
+  platf_do_promote();
+  platf_do_label();
+
+  //Put the platform into the simulator
+  platf_generate();
+}
+
+void dispatch_jobs(double tracker_deadline, double peer_deadline, double seed_percentage) {
+
+  xbt_dynar_t available_nodes = MSG_hosts_as_dynar();
+  msg_host_t host;
+  unsigned int i;
+
+  char** arguments_tracker;
+  char** arguments_peer;
+
+  unsigned int seed_count = (seed_percentage/100.0) * xbt_dynar_length(available_nodes);
+
+  xbt_dynar_foreach(available_nodes, i, host) {
+    if(i==0) {
+      //The fisrt node is the tracker
+      arguments_tracker = xbt_malloc0(sizeof(char*) * 2);
+      arguments_tracker[0] = bprintf("tracker");
+      arguments_tracker[1] = bprintf("%f", tracker_deadline);
+      MSG_process_create_with_arguments("tracker", tracker, NULL, host, 2, arguments_tracker);
+    } else {
+      //Other nodes are peers
+      int argument_size;
+      arguments_peer = xbt_malloc0(sizeof(char*) * 4);
+      arguments_peer[0] = bprintf("peer");
+      arguments_peer[1] = bprintf("%d", i);
+      arguments_peer[2] = bprintf("%f", peer_deadline);
+
+      //The first peers will be seeders
+      if(seed_count > 0) {
+        seed_count--;
+        arguments_peer[3] = bprintf("1");
+        argument_size = 4;
+      } else {
+        //Other ars leechers
+        arguments_peer[3] = NULL;
+        argument_size = 3;
+      }
+      MSG_process_create_with_arguments("peer", peer, NULL, host, argument_size, arguments_peer);
+    }
+  }
+}
+
+int main(int argc, char *argv[])
+{
+  MSG_init(&argc, argv);
+
+  rng_stream = RngStream_CreateStream(NULL);
+
+  //Maybe these parameters should be set from the command line...
+  //create_environment(<node_count>)
+  create_environment(20);
+
+  //dispatch_jobs(<tracker_deadline>, <peer_deadline>, <seed_percentage>)
+  dispatch_jobs(2000, 2000, 10);
+
+  MSG_main();
+
+  MSG_clean();
+
+  return 0;
+}