Extend simulator

1 files changed, 10 insertions, 2 deletions

diff --git a/src/scenarios.cc b/src/scenarios.cc
index 90a7849..0e7c596 100644
--- a/src/scenarios.cc
+++ b/src/scenarios.cc
@@ -17,10 +17,10 @@ using namespace rapidjson;
 
 int main(int argc, char **argv){
     // Setup seed
-    if(argc!=16){
+    if(argc!=20){
         cerr << "Usage: " << argv[0] <<
             " <seed> <simtime> <wakeupevery> <wakeupfor> <n_nodes>" <<
-            " <extended> <hint> <poff> <pon> <prx> <ptx> <datasize> <bitrate> <hintsize> <latency>" <<
+            " <extended> <hint> <poff> <pon> <prx> <ptx> <datasize> <bitrate> <hintsize> <latency> <shutdown_on_rcv> <unschedule_on_rcv> <farhint>" <<
             endl;
         exit(1);
     }
@@ -41,6 +41,10 @@ int main(int argc, char **argv){
     string bitrate(argv[13]);
     unsigned int hintsize=atoi(argv[14]);
     double latency=stod(argv[15]);
+    bool shutdown_on_rcv=!strcmp("true",argv[16]);
+    bool unschedule_on_rcv=!strcmp("true",argv[17]);
+    bool farhint=!strcmp("true",argv[18]);
+    double hintdist=stod(argv[19]);
 
 
     // Setup seed
@@ -56,6 +60,10 @@ int main(int argc, char **argv){
     d.AddMember("latency",latency,d.GetAllocator());
     d.AddMember("extended",extended,d.GetAllocator());
     d.AddMember("hint_size",hintsize,d.GetAllocator());
+    d.AddMember("shutdown_on_rcv",shutdown_on_rcv,d.GetAllocator());
+    d.AddMember("unschedule_on_rcv",unschedule_on_rcv,d.GetAllocator());
+    d.AddMember("farhint",farhint,d.GetAllocator());
+    d.AddMember("hintdist",hintdist,d.GetAllocator());
 
     // Create nodes
     Value nodes(kObjectType);