| 1 | /*
 | 
|---|
| 2 |  * CommandLineParser.cpp
 | 
|---|
| 3 |  *
 | 
|---|
| 4 |  *  Created on: May 8, 2010
 | 
|---|
| 5 |  *      Author: heber
 | 
|---|
| 6 |  */
 | 
|---|
| 7 | 
 | 
|---|
| 8 | #include "Helpers/MemDebug.hpp"
 | 
|---|
| 9 | 
 | 
|---|
| 10 | #include <boost/program_options.hpp>
 | 
|---|
| 11 | #include <fstream>
 | 
|---|
| 12 | #include <iostream>
 | 
|---|
| 13 | #include <map>
 | 
|---|
| 14 | 
 | 
|---|
| 15 | #include "Patterns/Singleton_impl.hpp"
 | 
|---|
| 16 | #include "CommandLineParser.hpp"
 | 
|---|
| 17 | #include "Helpers/Log.hpp"
 | 
|---|
| 18 | #include "Helpers/Verbose.hpp"
 | 
|---|
| 19 | 
 | 
|---|
| 20 | using namespace std;
 | 
|---|
| 21 | 
 | 
|---|
| 22 | /** Constructor of class CommandLineParser.
 | 
|---|
| 23 |  *
 | 
|---|
| 24 |  */
 | 
|---|
| 25 | CommandLineParser::CommandLineParser() :
 | 
|---|
| 26 |   generic("Generic options"),
 | 
|---|
| 27 |   config("Configuration"),
 | 
|---|
| 28 |   hidden("Hidden options"),
 | 
|---|
| 29 |   visible("Allowed options")
 | 
|---|
| 30 | {}
 | 
|---|
| 31 | 
 | 
|---|
| 32 | /** Destructor of class CommandLineParser.
 | 
|---|
| 33 |  *
 | 
|---|
| 34 |  */
 | 
|---|
| 35 | CommandLineParser::~CommandLineParser()
 | 
|---|
| 36 | {}
 | 
|---|
| 37 | 
 | 
|---|
| 38 | /** Parses the command line arguments.
 | 
|---|
| 39 |  * Calls program_options::store() and program_options::notify()
 | 
|---|
| 40 |  */
 | 
|---|
| 41 | void CommandLineParser::Parse()
 | 
|---|
| 42 | {
 | 
|---|
| 43 |   po::store(po::command_line_parser(argc,argv).options(cmdline_options).run(), vm);
 | 
|---|
| 44 |   ifstream input;
 | 
|---|
| 45 |   input.open("example.cfg");
 | 
|---|
| 46 |   if (!input.fail())
 | 
|---|
| 47 |     po::store(po::parse_config_file(input, config_file_options), vm);
 | 
|---|
| 48 |   input.close();
 | 
|---|
| 49 |   po::notify(vm);
 | 
|---|
| 50 | }
 | 
|---|
| 51 | 
 | 
|---|
| 52 | /** Scan the argument list for -a or --arguments and store their order for later use.
 | 
|---|
| 53 |  * \param &ShortFormToActionMap e.g. gives "help" for "h"
 | 
|---|
| 54 |  */
 | 
|---|
| 55 | void CommandLineParser::scanforSequenceOfArguments(map <std::string, std::string> &ShortFormToActionMap)
 | 
|---|
| 56 | {
 | 
|---|
| 57 |   // go through all arguments
 | 
|---|
| 58 |   for (int i=1;i<argc;i++) {
 | 
|---|
| 59 |     (cout << Verbose(1) << "Checking on " << argv[i] << endl);
 | 
|---|
| 60 |     // check whether they begin with - or -- and check that next letter is not numeric, if so insert
 | 
|---|
| 61 |     if (argv[i][0] == '-') {
 | 
|---|
| 62 |       (cout << Verbose(1) << "Possible argument: " << argv[i] << endl);
 | 
|---|
| 63 |       if (argv[i][1] == '-') {
 | 
|---|
| 64 |         (cout << Verbose(1) << "Putting " << argv[i] << " into the sequence." << endl);
 | 
|---|
| 65 |         SequenceOfActions.push_back(&(argv[i][2]));
 | 
|---|
| 66 |       } else if (((argv[i][1] < '0') || (argv[i][1] > '9')) && ((argv[i][1] != '.'))) {
 | 
|---|
| 67 |         map <std::string, std::string>::iterator iter = ShortFormToActionMap.find(&(argv[i][1]));
 | 
|---|
| 68 |         if (iter != ShortFormToActionMap.end()) {
 | 
|---|
| 69 |           (cout << Verbose(1) << "Putting " << iter->second << " for " << iter->first << " into the sequence." << endl);
 | 
|---|
| 70 |           SequenceOfActions.push_back(iter->second);
 | 
|---|
| 71 |         }
 | 
|---|
| 72 |       }
 | 
|---|
| 73 |     }
 | 
|---|
| 74 |   }
 | 
|---|
| 75 | }
 | 
|---|
| 76 | 
 | 
|---|
| 77 | 
 | 
|---|
| 78 | /** States whether there are command line arguments.
 | 
|---|
| 79 |  * \return true - there are none, false - there is at least one command line argument
 | 
|---|
| 80 |  */
 | 
|---|
| 81 | bool CommandLineParser::isEmpty()
 | 
|---|
| 82 | {
 | 
|---|
| 83 |   return vm.empty();
 | 
|---|
| 84 | }
 | 
|---|
| 85 | 
 | 
|---|
| 86 | /** Sets the options.
 | 
|---|
| 87 |  * \param _argc arg count from main()
 | 
|---|
| 88 |  * \param **_argv argument array from main()
 | 
|---|
| 89 |  */
 | 
|---|
| 90 | void CommandLineParser::setOptions(int _argc, char **_argv)
 | 
|---|
| 91 | {
 | 
|---|
| 92 |   argc = _argc;
 | 
|---|
| 93 |   argv = _argv;
 | 
|---|
| 94 |   cmdline_options.add(generic).add(config).add(hidden);
 | 
|---|
| 95 |   config_file_options.add(config).add(hidden);
 | 
|---|
| 96 |   visible.add(generic).add(config);
 | 
|---|
| 97 | }
 | 
|---|
| 98 | 
 | 
|---|
| 99 | /** Makes the Parser parse the command line options with current known options.
 | 
|---|
| 100 |  * \param _argc arg count from main()
 | 
|---|
| 101 |  * \param **_argv argument array from main()
 | 
|---|
| 102 |  */
 | 
|---|
| 103 | void CommandLineParser::Run(int _argc, char **_argv, map <std::string, std::string> &ShortFormToActionMap)
 | 
|---|
| 104 | {
 | 
|---|
| 105 |   setOptions(_argc,_argv);
 | 
|---|
| 106 |   Parse();
 | 
|---|
| 107 |   scanforSequenceOfArguments(ShortFormToActionMap);
 | 
|---|
| 108 | }
 | 
|---|
| 109 | 
 | 
|---|
| 110 | CONSTRUCT_SINGLETON(CommandLineParser)
 | 
|---|