Wip++...

[loba.git] / options.cpp

diff --git a/options.cpp b/options.cpp
index 24ae952789781cba83c8517e0d3836d6cbf446fc..4f9c52ca7e398388b14059568ac999499bee0764 100644 (file)
--- a/options.cpp
+++ b/options.cpp
@@ -1,11 +1,14 @@
-#include <cstring>
+#include "options.h"
+
+#include <cstring>              // strrchr
+#include <iomanip>
 #include <iostream>
-#include <unistd.h>
+#include <sstream>
+#include <unistd.h>             // getopt
 #include <xbt/log.h>
-#include "options.h"
 #include "misc.h"
 
-XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simu);
+XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(main);
 
 namespace opt {
 
@@ -17,20 +20,44 @@ namespace opt {
     int help_requested = 0;
     bool version_requested = false;
 
+    unsigned log_rate = 1;
+
+    unsigned maxiter = 4;
+    bool exit_on_close = false;
+
+    bool bookkeeping = false;
+
     cost_func comp_cost("1e9, 0"); // fixme: find better defaults
-    cost_func comm_cost("1e9, 0"); // fixme: find better defaults
+    cost_func comm_cost("1, 0"); // fixme: find better defaults
 
 } // namespace opt
 
+namespace {
+
+    const char* on_off(bool b)
+    {
+        return b ? "on" : "off";
+    }
+
+}
+
 int opt::parse_args(int* argc, char* argv[])
 {
-    char *tmp = strrchr(argv[0], '/');
+    int result = 1;
+
+    char* tmp = strrchr(argv[0], '/');
     opt::program_name = (tmp ? tmp + 1 : argv[0]);
 
     int c;
     opterr = 0;
-    while ((c = getopt(*argc, argv, "hc:C:V")) != -1) {
+    while ((c = getopt(*argc, argv, "bc:C:ehi:l:V")) != -1) {
         switch (c) {
+        case 'b':
+            opt::bookkeeping = true;
+            break;
+        case 'e':
+            opt::exit_on_close = true;
+            break;
         case 'h':
             opt::help_requested++;
             break;
@@ -40,33 +67,45 @@ int opt::parse_args(int* argc, char* argv[])
         case 'C':
             opt::comm_cost = cost_func(optarg);
             break;
+        case 'i':
+            std::istringstream(optarg) >> opt::maxiter;
+            break;
+        case 'l':
+            std::istringstream(optarg) >> opt::log_rate;
+            break;
         case 'V':
             opt::version_requested = true;
             break;
         case '?':
-            WARN1("invalid option -- '%c'", optopt);
+            ERROR1("invalid option -- '%c'", optopt);
+            result = 0;
             break;
         }
     }
     if (opt::version_requested || opt::help_requested)
         return 1;
 
-    switch (*argc - optind) {
+    int rem_args = *argc - optind;
+    switch (rem_args) {
     case 0:
         ERROR0("missing parameter -- <plaform_file>");
     case 1:
         ERROR0("missing parameter -- <application_file>");
-        return 0;
+        result = 0;
+        break;
 
     default:
         opt::platform_file = argv[optind];
         opt::application_file = argv[optind + 1];
+        if (rem_args == 2)
+            break;
         for (int i = optind + 2 ; i < *argc ; ++i)
-            WARN1("unused parameter -- \"%s\"", argv[i]);
+            ERROR1("unused parameter -- \"%s\"", argv[i]);
+        result = 0;
         break;
     }
 
-    return 1;
+    return result;
 }
 
 void opt::print()
@@ -74,12 +113,15 @@ void opt::print()
     INFO0(",----[ Simulation parameters ]");
     INFO1("| platform_file.......: \"%s\"", opt::platform_file);
     INFO1("| application_file....: \"%s\"", opt::application_file);
-    INFO1("| comp. cost factors..: [%s]", opt::comp_cost.to_string().c_str());
-    INFO1("| comm. cost factors..: [%s]", opt::comm_cost.to_string().c_str());
+    INFO1("| log rate............: %u",     opt::log_rate);
+    INFO1("| maxiter.............: %u",     opt::maxiter);
+    INFO1("| exit on close.......: %s",     on_off(opt::exit_on_close));
+    INFO1("| bookkeeping.........: %s",     on_off(opt::bookkeeping));
+    INFO1("| comp. cost factors..: [%s]",   opt::comp_cost.to_string().c_str());
+    INFO1("| comm. cost factors..: [%s]",   opt::comm_cost.to_string().c_str());
     INFO0("`----");
 }
 
-#include <iomanip>
 void opt::usage()
 {
     const int indent1 = 6;
@@ -96,13 +138,23 @@ void opt::usage()
               << "print help and exit (use -hh or -hhh for extended help)\n";
     if (opt::help_requested < 1)
         return;
+
     std::clog << o("-V") << "print version and exit\n";
+
+    std::clog << o("-b") << "activate bookkeeping\n";
     std::clog << oo("-c", "[fn,...]f0")
               << "polynomial factors for computation cost ("
               << opt::comp_cost.to_string() << ")\n";
     std::clog << oo("-C", "[fn,...]f0")
               << "polynomial factors for communication cost ("
               << opt::comm_cost.to_string() << ")\n";
+    std::clog << o("-e") << "exit on close reception\n";
+    std::clog << oo("-i", "value")
+              << "maximum number of iterations, 0 for infinity ("
+              << opt::maxiter << ")\n";
+    std::clog << oo("-l", "value")
+              << "print current load every \"value\" iterations, 0 to disable ("
+              << opt::log_rate << ")\n";
 
 #undef o
 #undef oo