Rev 1601 | Rev 1619 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | SVN | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 1594 | runge | 1 | /* |
| 2 | * |
||
| 3 | * Copyright (C) 2010 Mattias Runge |
||
| 4 | * |
||
| 5 | * This program is free software; you can redistribute it and/or modify |
||
| 6 | * it under the terms of the GNU General Public License as published by |
||
| 7 | * the Free Software Foundation; either version 2 of the License, or |
||
| 8 | * (at your option) any later version. |
||
| 9 | * |
||
| 10 | * This program is distributed in the hope that it will be useful, |
||
| 11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
||
| 12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||
| 13 | * GNU General Public License for more details. |
||
| 14 | * |
||
| 15 | * You should have received a copy of the GNU General Public License along |
||
| 16 | * with this program; if not, write to the Free Software Foundation, Inc., |
||
| 17 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
||
| 18 | * |
||
| 19 | */ |
||
| 20 | |||
| 21 | #include "Manager.h" |
||
| 22 | |||
| 23 | #include <iostream> |
||
| 24 | #include <fstream> |
||
| 25 | |||
| 26 | #include <boost/filesystem.hpp> |
||
| 27 | |||
| 28 | namespace atom { |
||
| 29 | namespace config { |
||
| 30 | |||
| 1599 | runge | 31 | Manager::Pointer Manager::instance_; |
| 1594 | runge | 32 | |
| 33 | Manager::Manager() : command_line_("Command line options"), configuration_file_("Configuration file options") |
||
| 34 | { |
||
| 35 | std::string default_config_file = "/etc/atom.conf"; |
||
| 36 | |||
| 37 | if (!boost::filesystem::exists(default_config_file)) |
||
| 38 | { |
||
| 39 | default_config_file = "atom.conf"; |
||
| 40 | } |
||
| 41 | |||
| 42 | this->configuration_file_.add_options() |
||
| 43 | ("MonitorPort", boost::program_options::value<int>()->default_value(1201), "TCP port to open for monitor output") |
||
| 44 | ("CommandPort", boost::program_options::value<int>()->default_value(1202), "TCP port to open for command input") |
||
| 45 | ("LogFile", boost::program_options::value<std::string>(), "File to log output to") |
||
| 1599 | runge | 46 | ("LogLevel", boost::program_options::value<int>()->default_value(4), "Level of logging") |
| 1601 | runge | 47 | ("ScriptPath", boost::program_options::value<std::string>(), "Path to where the scripts are") |
| 1594 | runge | 48 | ("ProtocolFile",boost::program_options::value<std::string>(), "File to read the protocol form") |
| 1609 | runge | 49 | ("CanNet", boost::program_options::value<type::StringList>(), "Information on where to locate the CAN networks") |
| 50 | ("Legacy", "Start legacy VM environment"); |
||
| 1594 | runge | 51 | |
| 52 | this->command_line_.add_options() |
||
| 53 | ("version,v", "print version string") |
||
| 54 | ("help,h", "produce help message") |
||
| 55 | ("daemon,d", "start in daemon mode") |
||
| 56 | ("file,f", boost::program_options::value<std::string>()->default_value(default_config_file), "configuration file"); |
||
| 57 | |||
| 58 | this->command_line_.add(this->configuration_file_); |
||
| 59 | } |
||
| 60 | |||
| 61 | Manager::~Manager() |
||
| 62 | { |
||
| 63 | } |
||
| 64 | |||
| 65 | Manager::Pointer Manager::Instance() |
||
| 66 | { |
||
| 67 | return Manager::instance_; |
||
| 68 | } |
||
| 69 | |||
| 1599 | runge | 70 | void Manager::Create() |
| 71 | { |
||
| 72 | Manager::instance_ = Manager::Pointer(new Manager()); |
||
| 73 | } |
||
| 74 | |||
| 1594 | runge | 75 | void Manager::Delete() |
| 76 | { |
||
| 77 | Manager::instance_.reset(); |
||
| 78 | } |
||
| 79 | |||
| 80 | bool Manager::Set(int argument_count, char** argument_vector) |
||
| 81 | { |
||
| 82 | try |
||
| 83 | { |
||
| 84 | boost::program_options::store(boost::program_options::command_line_parser(argument_count, argument_vector).options(this->command_line_).run(), this->variable_map_); |
||
| 85 | } |
||
| 86 | catch (boost::program_options::unknown_option e) |
||
| 87 | { |
||
| 88 | std::cerr << e.what() << std::endl; |
||
| 89 | std::cout << this->command_line_ << std::endl; |
||
| 90 | return false; |
||
| 91 | } |
||
| 92 | catch (boost::program_options::invalid_syntax e) |
||
| 93 | { |
||
| 94 | std::cerr << e.what() << std::endl; |
||
| 95 | std::cout << this->command_line_ << std::endl; |
||
| 96 | return false; |
||
| 97 | } |
||
| 98 | |||
| 99 | std::ifstream ifs(this->GetAsString("file").c_str()); |
||
| 100 | |||
| 101 | if (!ifs.is_open()) |
||
| 102 | { |
||
| 103 | std::cerr << "Error: Could not find " << this->GetAsString("file") << std::endl; |
||
| 104 | return false; |
||
| 105 | } |
||
| 106 | |||
| 107 | try |
||
| 108 | { |
||
| 109 | boost::program_options::store(boost::program_options::parse_config_file(ifs, this->configuration_file_), this->variable_map_); |
||
| 110 | } |
||
| 111 | catch (boost::program_options::unknown_option e) |
||
| 112 | { |
||
| 113 | ifs.close(); |
||
| 114 | std::cerr << e.what() << " found in " << this->GetAsString("file") << std::endl; |
||
| 115 | std::cout << this->configuration_file_ << std::endl; |
||
| 116 | return false; |
||
| 117 | } |
||
| 118 | catch (boost::program_options::invalid_syntax e) |
||
| 119 | { |
||
| 120 | std::cerr << e.what() << " found in " << this->GetAsString("file") << std::endl; |
||
| 121 | std::cout << this->command_line_ << std::endl; |
||
| 122 | return false; |
||
| 123 | } |
||
| 124 | |||
| 125 | ifs.close(); |
||
| 126 | |||
| 127 | if (this->Exist("help")) |
||
| 128 | { |
||
| 129 | std::cout << this->command_line_ << std::endl; |
||
| 130 | return false; |
||
| 131 | } |
||
| 132 | |||
| 133 | return true; |
||
| 134 | } |
||
| 135 | |||
| 136 | bool Manager::Exist(std::string name) |
||
| 137 | { |
||
| 138 | return this->variable_map_.count(name) != 0; |
||
| 139 | } |
||
| 140 | |||
| 141 | int Manager::GetAsInt(std::string name) |
||
| 142 | { |
||
| 143 | return this->variable_map_[name].as<int>(); |
||
| 144 | } |
||
| 145 | |||
| 146 | std::string Manager::GetAsString(std::string name) |
||
| 147 | { |
||
| 148 | return this->variable_map_[name].as<std::string>(); |
||
| 149 | } |
||
| 150 | |||
| 1596 | runge | 151 | type::StringList Manager::GetAsStringVector(std::string name) |
| 1594 | runge | 152 | { |
| 1596 | runge | 153 | return this->variable_map_[name].as<type::StringList>(); |
| 1594 | runge | 154 | } |
| 155 | |||
| 156 | |||
| 157 | }; // namespace config |
||
| 158 | }; // namespace atom |