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<type::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>();
}
 
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 1609 → 1619

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