options.cpp

   1 #include <cstring>              // strrchr
   2 #include <iomanip>
   3 #include <iostream>
   4 #include <sstream>
   5 #include <unistd.h>             // getopt
   6 #include <xbt/log.h>
   7 #include "options.h"
   8 #include "misc.h"
   9
  10 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simu);
  11
  12 namespace opt {
  13
  14     const char* program_name;
  15
  16     const char* platform_file;
  17     const char* application_file;
  18
  19     int help_requested = 0;
  20     bool version_requested = false;
  21
  22     unsigned log_rate = 1;
  23
  24     unsigned maxiter = 4;
  25     bool exit_on_close = false;
  26
  27     bool bookkeeping = false;
  28
  29     cost_func comp_cost("1e9, 0"); // fixme: find better defaults
  30
  31 } // namespace opt
  32
  33 namespace {
  34
  35     const char* on_off(bool b)
  36     {
  37         return b ? "on" : "off";
  38     }
  39
  40 }
  41
  42 int opt::parse_args(int* argc, char* argv[])
  43 {
  44     int result = 1;
  45
  46     char* tmp = strrchr(argv[0], '/');
  47     opt::program_name = (tmp ? tmp + 1 : argv[0]);
  48
  49     int c;
  50     opterr = 0;
  51     while ((c = getopt(*argc, argv, "bc:ehi:l:V")) != -1) {
  52         switch (c) {
  53         case 'b':
  54             opt::bookkeeping = true;
  55             break;
  56         case 'e':
  57             opt::exit_on_close = true;
  58             break;
  59         case 'h':
  60             opt::help_requested++;
  61             break;
  62         case 'c':
  63             opt::comp_cost = cost_func(optarg);
  64             break;
  65         case 'i':
  66             std::istringstream(optarg) >> opt::maxiter;
  67             break;
  68         case 'l':
  69             std::istringstream(optarg) >> opt::log_rate;
  70             break;
  71         case 'V':
  72             opt::version_requested = true;
  73             break;
  74         case '?':
  75             ERROR1("invalid option -- '%c'", optopt);
  76             result = 0;
  77             break;
  78         }
  79     }
  80     if (opt::version_requested || opt::help_requested)
  81         return 1;
  82
  83     int rem_args = *argc - optind;
  84     switch (rem_args) {
  85     case 0:
  86         ERROR0("missing parameter -- <plaform_file>");
  87     case 1:
  88         ERROR0("missing parameter -- <application_file>");
  89         result = 0;
  90         break;
  91
  92     default:
  93         opt::platform_file = argv[optind];
  94         opt::application_file = argv[optind + 1];
  95         if (rem_args == 2)
  96             break;
  97         for (int i = optind + 2 ; i < *argc ; ++i)
  98             ERROR1("unused parameter -- \"%s\"", argv[i]);
  99         result = 0;
 100         break;
 101     }
 102
 103     return result;
 104 }
 105
 106 void opt::print()
 107 {
 108     INFO0(",----[ Simulation parameters ]");
 109     INFO1("| platform_file.......: \"%s\"", opt::platform_file);
 110     INFO1("| application_file....: \"%s\"", opt::application_file);
 111     INFO1("| log rate............: %u",     opt::log_rate);
 112     INFO1("| maxiter.............: %u",     opt::maxiter);
 113     INFO1("| exit on close.......: %s",     on_off(opt::exit_on_close));
 114     INFO1("| bookkeeping.........: %s",     on_off(opt::bookkeeping));
 115     INFO1("| comp. cost factors..: [%s]",   opt::comp_cost.to_string().c_str());
 116     INFO0("`----");
 117 }
 118
 119 void opt::usage()
 120 {
 121     const int indent1 = 6;
 122     const int indent2 = 12;
 123
 124 #define oo(opt, arg) std::setw(indent1) << (opt) << " "               \
 125     << std::setw(indent2) << std::left << (arg) << std::right
 126 #define o(opt) oo(opt, "")
 127
 128     std::clog << "Usage: " << opt::program_name
 129               << " [options] <platform_file> <application_file>\n";
 130
 131     std::clog << o("-h")
 132               << "print help and exit (use -hh or -hhh for extended help)\n";
 133     if (opt::help_requested < 1)
 134         return;
 135
 136     std::clog << o("-V") << "print version and exit\n";
 137
 138     std::clog << o("-b") << "activate bookkeeping\n";
 139     std::clog << oo("-c", "[fn,...]f0")
 140               << "polynomial factors for computation cost ("
 141               << opt::comp_cost.to_string() << ")\n";
 142     std::clog << o("-e") << "exit on close reception\n";
 143     std::clog << oo("-i", "value")
 144               << "maximum number of iterations, 0 for infinity ("
 145               << opt::maxiter << ")\n";
 146     std::clog << oo("-l", "value")
 147               << "print current load every \"value\" iterations, 0 to disable ("
 148               << opt::log_rate << ")\n";
 149
 150 #undef o
 151 #undef oo
 152 }
 153
 154 // Local variables:
 155 // mode: c++
 156 // End: