options.cpp

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