osrm-backend/Util/BaseConfiguration.h

/*
    open source routing machine
    Copyright (C) Dennis Luxen, 2010

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU AFFERO General Public License as published by
the Free Software Foundation; either version 3 of the License, or
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 Affero General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
or see http://www.gnu.org/licenses/agpl.txt.
 */

#ifndef BASECONFIGURATION_H_
#define BASECONFIGURATION_H_

#include <iostream>
#include <string>
#include <exception>
#include <fstream>

#include "../DataStructures/HashTable.h"

class BaseConfiguration {
public:
    BaseConfiguration(const char * configFile) {
        std::ifstream config( configFile );
        if(!config) {
            std::cerr << "[config] .ini not found" << std::endl;
            return;
        }

        std::string line;
        try {
            if (config.is_open()) {
                while ( config.good() )  {
                    getline (config,line);
                    std::vector<std::string> tokens;
                    Tokenize(line, tokens);
                    if(2 == tokens.size() )
                        parameters.Add(tokens[0], tokens[1]);
                }
                config.close();
            }
        } catch(std::exception& e) {
            ERR("[config] " << configFile << " not found -> Exception: " <<e.what());
            if(config.is_open())
                config.close();
        }
    }

    std::string GetParameter(const char * key){
        return GetParameter(std::string(key));
    }

    std::string GetParameter(std::string key) {
        return parameters.Find(key);
    }

    void SetParameter(const char* key, const char* value) {
        SetParameter(std::string(key), std::string(value));
    }

    void SetParameter(std::string key, std::string value) {
        parameters.Set(key, value);
    }

private:
    void Tokenize(const std::string& str, std::vector<std::string>& tokens,  const std::string& delimiters = "=") {
        std::string::size_type lastPos = str.find_first_not_of(delimiters, 0);
        std::string::size_type pos     = str.find_first_of(delimiters, lastPos);

        while (std::string::npos != pos || std::string::npos != lastPos) {
            std::string temp = str.substr(lastPos, pos - lastPos);
            TrimStringRight(temp);
            TrimStringLeft(temp);
            tokens.push_back( temp );
            lastPos = str.find_first_not_of(delimiters, pos);
            pos = str.find_first_of(delimiters, lastPos);
        }
    }
    void TrimStringRight(std::string& str) {
        std::string::size_type pos = str.find_last_not_of(" ");
        if (pos != std::string::npos)
            str.erase(pos+1);
        else
            str.erase( str.begin() , str.end() );
    }
    void TrimStringLeft(std::string& str) {
        std::string::size_type pos = str.find_first_not_of(" ");
        if (pos != std::string::npos)
            str.erase(0, pos);
        else
            str.erase( str.begin() , str.end() );
    }

    HashTable<std::string, std::string> parameters;
};

#endif /* BASECONFIGURATION_H_ */