configurationdatabase.cpp

Go to the documentation of this file.

/***************************************************************************************************
*****           Copyright (C) 2005 John Schneiderman <JohnMS@member.fsf.org>                   *****
*****                                                                                          *****
*****           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 St, Fifth Floor, Boston, MA  02110-1301  USA     *****
***************************************************************************************************/
#include "configurationdatabase.h"

#include <fstream>
using std::ifstream;
using std::endl;

const char ConfigurationDatabase::BEGIN_COMMENT='#';
const char ConfigurationDatabase::BEGIN_HEADER='[';
const char ConfigurationDatabase::END_HEADER=']';

ConfigurationDatabase::ConfigurationDatabase():m_configurations(), m_comments()
{
}

ConfigurationDatabase::ConfigurationDatabase(const string &filename):m_configurations(), m_comments()
{
    ifstream inputFile;

    inputFile.open(filename.c_str());
    if (!inputFile)
        throw "ERROR: Failed to find the configuration file " + filename + ".";
    inputFile >> *this;
    inputFile.close();
}

int ConfigurationDatabase::intValue(const string &header, const string &key) const
{
    int value=0;

    if (m_configurations.find(header) != m_configurations.end())
        if (m_configurations.find(header)->second.find(key) != m_configurations.find(header)->second.end())
            value=atoi(m_configurations.find(header)->second.find(key)->second.c_str());
        else
            throw "ERROR: Failed to find the key " + key + ".";
    else
        throw "ERROR: Failed to find the header " + header + ".";
    return value;
}

double ConfigurationDatabase::doubleValue(const string &header, const string &key) const
{
    double value=0.0;

    if (m_configurations.find(header) != m_configurations.end())
        if (m_configurations.find(header)->second.find(key) != m_configurations.find(header)->second.end())
            value=atof(m_configurations.find(header)->second.find(key)->second.c_str());
        else
            throw "ERROR: Failed to find the key " + key + ".";
    else
        throw "ERROR: Failed to find the header " + header + ".";
    return value;
}

string ConfigurationDatabase::stringValue(const string &header, const string &key) const
{
    string value="";

    if (m_configurations.find(header) != m_configurations.end())
        if (m_configurations.find(header)->second.find(key) != m_configurations.find(header)->second.end())
            value=m_configurations.find(header)->second.find(key)->second.c_str();
        else
            throw "ERROR: Failed to find the key " + key + ".";
    else
        throw "ERROR: Failed to find the header " + header + ".";
    return value;
}

ConfigurationDatabase::Headers ConfigurationDatabase::headers() const
{
    ConfigurationDatabaseType::const_iterator iterHeader=m_configurations.begin();
    Headers headers;

    while (iterHeader != m_configurations.end())
    {
        headers.push_back(iterHeader->first);
        iterHeader++;
    }
    return headers;
}

ConfigurationDatabase::Keys ConfigurationDatabase::keys(const string &header) const
{
    KeyValuePair::const_iterator iterKey=m_configurations.find(header)->second.begin();
    KeyValuePair::const_iterator iterEnd=m_configurations.find(header)->second.end();
    Keys keys;

    while (iterKey != iterEnd)
    {
        keys.push_back(iterKey->first);
        iterKey++;
    }
    return keys;
}

void ConfigurationDatabase::addHeaderKeyValuePair(const string &header, const string &key, const string &value)
{
    m_configurations[header][key]=value;
}

void ConfigurationDatabase::addKeyValuePair(const string &header, const string &key, const string &value)
{
    if (m_configurations.find(header) != m_configurations.end())
        m_configurations[header][key]=value;
    else
        throw "ERROR: Failed to find the header " + header + ".";
}

istream& operator>>(istream &in, ConfigurationDatabase &database)
{
    string testPart="", key="", header="", value="", comment="";
    int lineNumber=-1;

    while (!in.fail())
    {
        lineNumber++;
        testPart="";
        in >> testPart;
        if (testPart[0] == ConfigurationDatabase::BEGIN_COMMENT)
        {
            getline(in, comment); //Get the comment
            database.m_comments[lineNumber]=testPart.substr(1, testPart.length()) + comment; //Add it to our comment list.
        }
        else if (testPart[0] == ConfigurationDatabase::BEGIN_HEADER)
        {
            header=testPart; // testKey is the header
            header.erase(0, 1); //Erase the [
            header.erase(header.length() - 1, 1); //Erase the ]
            continue;
        }
        else if (!testPart.empty())
        {
            key=testPart;
            getline(in, value); //It must be a key value
            if (!value.empty())
            {
                value.erase(0, 1);
                if (header == "")
                    throw "ERROR: Key-value pair, " + key + "-" + value + ", found without a header!";
                database.m_configurations[header][key]=value;
                value="";
                key="";
            }
            else
                throw "ERROR: Attempted to add a key, " + key + ", without a value!";
        }
    }
    return in;
}

ostream& operator<<(ostream &out, const ConfigurationDatabase &database)
{
    ConfigurationDatabase::ConfigurationDatabaseType::const_iterator iterHeader=database.m_configurations.begin();
    ConfigurationDatabase::KeyValuePair::const_iterator iterKey;
    int lineNumber=0;

    while (iterHeader != database.m_configurations.end())
    {
        if (database.m_comments.find(lineNumber) != database.m_comments.end()) //Write out comment
        {
            out << ConfigurationDatabase::BEGIN_COMMENT << database.m_comments.find(lineNumber)->second << endl;
            lineNumber++;
        }
        else //Write out header's key-values
        {
            iterKey=iterHeader->second.begin();
            out << ConfigurationDatabase::BEGIN_HEADER << iterHeader->first << ConfigurationDatabase::END_HEADER << endl;
            lineNumber++;
            while (iterKey != iterHeader->second.end()) //Write out key-value pair
            {
                out << iterKey->first << ' ' << iterKey->second << endl;
                lineNumber++;
                iterKey++;
            }
            iterHeader++;
            out << endl;
        }
    }
    return out;
}

---