WebSVN – HomeAutomation – Diff – /trunk/PcSoftware/Atom1.5/src/config/Manager.cpp


/*
 *
 *  Copyright (C) 2010  Mattias Runge
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along
 *  with this program; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */
 
#include "Manager.h"
 
#include <iostream>
#include <fstream>
 
#include <boost/filesystem.hpp>
 
namespace atom {
namespace config {
 
Manager::Pointer Manager::instance_;
 
Manager::Manager() : command_line_("Command line options"), configuration_file_("Configuration file options")
{
    std::string default_config_file = "/etc/atom.conf";
   
    if (!boost::filesystem::exists(default_config_file))
    {
        default_config_file = "atom.conf";
    }
 
    this->configuration_file_.add_options()
    ("MonitorPort", boost::program_options::value<int>()->default_value(1201), "TCP port to open for monitor output")
    ("CommandPort", boost::program_options::value<int>()->default_value(1202), "TCP port to open for command input")
    ("LogFile",     boost::program_options::value<std::string>(),              "File to log output to")
    ("LogLevel",    boost::program_options::value<int>()->default_value(4),    "Level of logging")
    ("ScriptPath",  boost::program_options::value<std::string>(),              "Path to where the scripts are")
    ("StoragePath", boost::program_options::value<std::string>(),              "Path to where the storage files are")
    ("ProtocolFile",boost::program_options::value<std::string>(),              "File to read the protocol form")
    ("CanNet",      boost::program_options::value<common::StringList>(),       "Information on where to locate the CAN networks")
    ("Legacy",                                                                 "Start legacy VM environment");
 
    this->command_line_.add_options()
    ("version,v", "print version string")
    ("help,h",    "produce help message")
    ("daemon,d",  "start in daemon mode")
    ("file,f",    boost::program_options::value<std::string>()->default_value(default_config_file), "configuration file");
   
    this->command_line_.add(this->configuration_file_);
}
 
Manager::~Manager()
{
}
 
Manager::Pointer Manager::Instance()
{
    return Manager::instance_;
}
 
void Manager::Create()
{
    Manager::instance_ = Manager::Pointer(new Manager());
}
 
void Manager::Delete()
{
    Manager::instance_.reset();
}
 
bool Manager::Set(int argument_count, char** argument_vector)
{
    try
    {
        boost::program_options::store(boost::program_options::command_line_parser(argument_count, argument_vector).options(this->command_line_).run(), this->variable_map_);
    }
    catch (boost::program_options::unknown_option e)
    {
        std::cerr << e.what() << std::endl;
        std::cout << this->command_line_ << std::endl;
        return false;
    }
    catch (boost::program_options::invalid_syntax e)
    {
        std::cerr << e.what() << std::endl;
        std::cout << this->command_line_ << std::endl;
        return false;
    }
   
    std::ifstream ifs(this->GetAsString("file").c_str());
       
    if (!ifs.is_open())
    {
        std::cerr << "Error: Could not find " << this->GetAsString("file") << std::endl;
        return false;
    }
   
    try
    {
        boost::program_options::store(boost::program_options::parse_config_file(ifs, this->configuration_file_), this->variable_map_);
    }
    catch (boost::program_options::unknown_option e)
    {
        ifs.close();
        std::cerr << e.what() << " found in " << this->GetAsString("file") << std::endl;
        std::cout << this->configuration_file_ << std::endl;
        return false;
    }
    catch (boost::program_options::invalid_syntax e)
    {
        std::cerr << e.what() << " found in " << this->GetAsString("file") << std::endl;
        std::cout << this->command_line_ << std::endl;
        return false;
    }
       
    ifs.close();
   
    if (this->Exist("help"))
    {
        std::cout << this->command_line_ << std::endl;
        return false;
    }
   
    return true;
}
 
bool Manager::Exist(std::string name)
{
    return this->variable_map_.count(name) != 0;
}
 
int Manager::GetAsInt(std::string name)
{
    return this->variable_map_[name].as<int>();
}
 
std::string Manager::GetAsString(std::string name)
{
    return this->variable_map_[name].as<std::string>();
}
 
common::StringList Manager::GetAsStringVector(std::string name)
{
    return this->variable_map_[name].as<common::StringList>();
}
 
 
}; // namespace config
}; // namespace atom

Subversion Repositories HomeAutomation

(root)/trunk/PcSoftware/Atom1.5/src/config/Manager.cpp – Rev 1642 → 1646

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