Wip++...

[loba.git] / options.cpp

diff --git a/options.cpp b/options.cpp
index 4dd53150f784f3f5f680d19b46e7082c80bb8a32..a16023a64d4762b2181ceb03a9432e5fcb987f3c 100644 (file)
--- a/options.cpp
+++ b/options.cpp
@@ -1,6 +1,8 @@
-#include <cstring>
+#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"
@@ -17,6 +19,11 @@ 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
@@ -34,49 +41,66 @@ namespace {
 
 int opt::parse_args(int* argc, char* argv[])
 {
+    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, "bc:hV")) != -1) {
+    while ((c = getopt(*argc, argv, "bc: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;
         case 'c':
             opt::comp_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()
@@ -84,12 +108,14 @@ void opt::print()
     INFO0(",----[ Simulation parameters ]");
     INFO1("| platform_file.......: \"%s\"", opt::platform_file);
     INFO1("| application_file....: \"%s\"", opt::application_file);
+    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());
     INFO0("`----");
 }
 
-#include <iomanip>
 void opt::usage()
 {
     const int indent1 = 6;
@@ -106,11 +132,20 @@ 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 << 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