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")
    ("ProtocolFile",boost::program_options::value<std::string>(),              "File to read the protocol form")
    ("CanNet",      boost::program_options::value<type::StringList>(),         "Information on where to locate the CAN networks");
 
    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>();
}
 
type::StringList Manager::GetAsStringVector(std::string name)
{
    return this->variable_map_[name].as<type::StringList>();
}
 
 
}; // namespace config
}; // namespace atom

Subversion Repositories HomeAutomation

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

Rev 1599		Rev 1601
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	("~~ScriptsPath~~", boost::program_options::value<std::string>(), "Path to where the scripts")	47	("ScriptPath", boost::program_options::value<std::string>(), "Path to where the scripts are")
48	("ProtocolFile",boost::program_options::value<std::string>(), "File to read the protocol form")	48	("ProtocolFile",boost::program_options::value<std::string>(), "File to read the protocol form")
49	("CanNet", boost::program_options::value<type::StringList>(), "Information on where to locate the CAN networks");	49	("CanNet", boost::program_options::value<type::StringList>(), "Information on where to locate the CAN networks");
50		50
51	this->command_line_.add_options()	51	this->command_line_.add_options()
52	("version,v", "print version string")	52	("version,v", "print version string")
53	("help,h", "produce help message")	53	("help,h", "produce help message")
54	("daemon,d", "start in daemon mode")	54	("daemon,d", "start in daemon mode")
55	("file,f", boost::program_options::value<std::string>()->default_value(default_config_file), "configuration file");	55	("file,f", boost::program_options::value<std::string>()->default_value(default_config_file), "configuration file");
56		56
57	this->command_line_.add(this->configuration_file_);	57	this->command_line_.add(this->configuration_file_);
58	}	58	}
59		59
60	Manager::~Manager()	60	Manager::~Manager()
61	{	61	{
62	}	62	}
63		63
64	Manager::Pointer Manager::Instance()	64	Manager::Pointer Manager::Instance()
65	{	65	{
66	return Manager::instance_;	66	return Manager::instance_;
67	}	67	}
68		68
69	void Manager::Create()	69	void Manager::Create()
70	{	70	{
71	Manager::instance_ = Manager::Pointer(new Manager());	71	Manager::instance_ = Manager::Pointer(new Manager());
72	}	72	}
73		73
74	void Manager::Delete()	74	void Manager::Delete()
75	{	75	{
76	Manager::instance_.reset();	76	Manager::instance_.reset();
77	}	77	}
78		78
79	bool Manager::Set(int argument_count, char** argument_vector)	79	bool Manager::Set(int argument_count, char** argument_vector)
80	{	80	{
81	try	81	try
82	{	82	{
83	boost::program_options::store(boost::program_options::command_line_parser(argument_count, argument_vector).options(this->command_line_).run(), this->variable_map_);	83	boost::program_options::store(boost::program_options::command_line_parser(argument_count, argument_vector).options(this->command_line_).run(), this->variable_map_);
84	}	84	}
85	catch (boost::program_options::unknown_option e)	85	catch (boost::program_options::unknown_option e)
86	{	86	{
87	std::cerr << e.what() << std::endl;	87	std::cerr << e.what() << std::endl;
88	std::cout << this->command_line_ << std::endl;	88	std::cout << this->command_line_ << std::endl;
89	return false;	89	return false;
90	}	90	}
91	catch (boost::program_options::invalid_syntax e)	91	catch (boost::program_options::invalid_syntax e)
92	{	92	{
93	std::cerr << e.what() << std::endl;	93	std::cerr << e.what() << std::endl;
94	std::cout << this->command_line_ << std::endl;	94	std::cout << this->command_line_ << std::endl;
95	return false;	95	return false;
96	}	96	}
97		97
98	std::ifstream ifs(this->GetAsString("file").c_str());	98	std::ifstream ifs(this->GetAsString("file").c_str());
99		99
100	if (!ifs.is_open())	100	if (!ifs.is_open())
101	{	101	{
102	std::cerr << "Error: Could not find " << this->GetAsString("file") << std::endl;	102	std::cerr << "Error: Could not find " << this->GetAsString("file") << std::endl;
103	return false;	103	return false;
104	}	104	}
105		105
106	try	106	try
107	{	107	{
108	boost::program_options::store(boost::program_options::parse_config_file(ifs, this->configuration_file_), this->variable_map_);	108	boost::program_options::store(boost::program_options::parse_config_file(ifs, this->configuration_file_), this->variable_map_);
109	}	109	}
110	catch (boost::program_options::unknown_option e)	110	catch (boost::program_options::unknown_option e)
111	{	111	{
112	ifs.close();	112	ifs.close();
113	std::cerr << e.what() << " found in " << this->GetAsString("file") << std::endl;	113	std::cerr << e.what() << " found in " << this->GetAsString("file") << std::endl;
114	std::cout << this->configuration_file_ << std::endl;	114	std::cout << this->configuration_file_ << std::endl;
115	return false;	115	return false;
116	}	116	}
117	catch (boost::program_options::invalid_syntax e)	117	catch (boost::program_options::invalid_syntax e)
118	{	118	{
119	std::cerr << e.what() << " found in " << this->GetAsString("file") << std::endl;	119	std::cerr << e.what() << " found in " << this->GetAsString("file") << std::endl;
120	std::cout << this->command_line_ << std::endl;	120	std::cout << this->command_line_ << std::endl;
121	return false;	121	return false;
122	}	122	}
123		123
124	ifs.close();	124	ifs.close();
125		125
126	if (this->Exist("help"))	126	if (this->Exist("help"))
127	{	127	{
128	std::cout << this->command_line_ << std::endl;	128	std::cout << this->command_line_ << std::endl;
129	return false;	129	return false;
130	}	130	}
131		131
132	return true;	132	return true;
133	}	133	}
134		134
135	bool Manager::Exist(std::string name)	135	bool Manager::Exist(std::string name)
136	{	136	{
137	return this->variable_map_.count(name) != 0;	137	return this->variable_map_.count(name) != 0;
138	}	138	}
139		139
140	int Manager::GetAsInt(std::string name)	140	int Manager::GetAsInt(std::string name)
141	{	141	{
142	return this->variable_map_[name].as<int>();	142	return this->variable_map_[name].as<int>();
143	}	143	}
144		144
145	std::string Manager::GetAsString(std::string name)	145	std::string Manager::GetAsString(std::string name)
146	{	146	{
147	return this->variable_map_[name].as<std::string>();	147	return this->variable_map_[name].as<std::string>();
148	}	148	}
149		149
150	type::StringList Manager::GetAsStringVector(std::string name)	150	type::StringList Manager::GetAsStringVector(std::string name)
151	{	151	{
152	return this->variable_map_[name].as<type::StringList>();	152	return this->variable_map_[name].as<type::StringList>();
153	}	153	}
154		154
155		155
156	}; // namespace config	156	}; // namespace config
157	}; // namespace atom	157	}; // namespace atom