ArgumentParser.hpp 2.53 KB
Newer Older
wadaniel's avatar
wadaniel committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140
#pragma once

#include <cstring>
#include <map>
#include <vector>
#include <string>

using namespace std;

class Value
{
private:
  string content;

public:

  Value() : content("") {}

  Value(string content_) : content(content_) { }

  double asDouble(double def=0) const
  {
    if (content == "") return def;
    return (double) atof(content.c_str());
  }

  int asInt(int def=0) const
  {
    if (content == "") return def;
    return atoi(content.c_str());
  }

  bool asBool(bool def=false) const
  {
    if (content == "") return def;
    if (content == "0") return false;
    if (content == "false") return false;

    return true;
  }

  string asString(string def="") const
  {
    if (content == "") return def;

    return content;
  }
};

class ArgumentParser
{
private:

  map<string,Value> mapArguments;

  const int iArgC;
  const char** vArgV;
  bool bStrictMode;

public:

  Value operator()(const string arg)
  {
    if (bStrictMode)
    {
      map<string,Value>::const_iterator it = mapArguments.find(arg);

      if (it == mapArguments.end())
      {
        printf("Runtime option NOT SPECIFIED! ABORTING! name: %s\n",arg.data());
        abort();
      }
    }

    return mapArguments[arg];
  }

  ArgumentParser(const int argc, const char ** argv) : mapArguments(), iArgC(argc), vArgV(argv), bStrictMode(false)
  {
    for (int i=1; i<argc; i++)
      if (argv[i][0] == '-')
      {
        string values = "";
        int itemCount = 0;

        for (int j=i+1; j<argc; j++)
          if (argv[j][0] == '-')
            break;
          else
          {
            if (strcmp(values.c_str(), ""))
              values += ' ';

            values += argv[j];
            itemCount++;
          }

        if (itemCount == 0)
          values += '1';
        mapArguments[argv[i]] = Value(values);
        i += itemCount;
      }
  }

  int getargc() const {
    return iArgC;
  }

  const char** getargv() const {
    return vArgV;
  }

  void set_strict_mode()
  {
    bStrictMode = true;
  }

  void unset_strict_mode()
  {
    bStrictMode = false;
  }

  void save_options(string path=".")
  {
    string options;
    for(map<string,Value>::const_iterator it=mapArguments.begin(); it!=mapArguments.end(); it++)
    {
      options+= it->first + " " + it->second.asString() + " ";
    }
    string filepath = (path + "/" + string("argumentparser.log"));
    FILE * f = fopen(filepath.data(), "a");
    if (f == NULL)
    {
      printf("impossible to write %s.\n", filepath.data());
      return;
    }
    fprintf(f, "%s\n", options.data());
    fclose(f);
  }
};