#include <boost/functional/hash.hpp>
#include "src/storage/expressions/SimpleValuation.h"
#include "src/exceptions/ExceptionMacros.h"
#include "src/exceptions/InvalidArgumentException.h"
#include "src/exceptions/InvalidAccessException.h"

namespace storm {
    namespace expressions {
        bool SimpleValuation::operator==(SimpleValuation const& other) const {
            return this->identifierToValueMap == other.identifierToValueMap;
        }
        
        void SimpleValuation::addBooleanIdentifier(std::string const& name, bool initialValue) {
            LOG_THROW(this->identifierToValueMap.find(name) == this->identifierToValueMap.end(), storm::exceptions::InvalidArgumentException, "Identifier '" << name << "' already registered.");
            this->identifierToValueMap.emplace(name, initialValue);
        }
        
        void SimpleValuation::addIntegerIdentifier(std::string const& name, int_fast64_t initialValue) {
            LOG_THROW(this->identifierToValueMap.find(name) == this->identifierToValueMap.end(), storm::exceptions::InvalidArgumentException, "Identifier '" << name << "' already registered.");
            this->identifierToValueMap.emplace(name, initialValue);
        }
        
        void SimpleValuation::addDoubleIdentifier(std::string const& name, double initialValue) {
            LOG_THROW(this->identifierToValueMap.find(name) == this->identifierToValueMap.end(), storm::exceptions::InvalidArgumentException, "Identifier '" << name << "' already registered.");
            this->identifierToValueMap.emplace(name, initialValue);
        }
        
        void SimpleValuation::setBooleanValue(std::string const& name, bool value) {
            this->identifierToValueMap[name] = value;
        }
        
        void SimpleValuation::setIntegerValue(std::string const& name, int_fast64_t value) {
            this->identifierToValueMap[name] = value;
        }
        
        void SimpleValuation::setDoubleValue(std::string const& name, double value) {
            this->identifierToValueMap[name] = value;
        }
        
        bool SimpleValuation::getBooleanValue(std::string const& name) const {
            auto nameValuePair = this->identifierToValueMap.find(name);
            LOG_THROW(nameValuePair != this->identifierToValueMap.end(), storm::exceptions::InvalidAccessException, "Access to unkown identifier '" << name << "'.");
            return boost::get<bool>(nameValuePair->second);
        }
        
        int_fast64_t SimpleValuation::getIntegerValue(std::string const& name) const {
            auto nameValuePair = this->identifierToValueMap.find(name);
            LOG_THROW(nameValuePair != this->identifierToValueMap.end(), storm::exceptions::InvalidAccessException, "Access to unkown identifier '" << name << "'.");
            return boost::get<int_fast64_t>(nameValuePair->second);
        }
        
        double SimpleValuation::getDoubleValue(std::string const& name) const {
            auto nameValuePair = this->identifierToValueMap.find(name);
            LOG_THROW(nameValuePair != this->identifierToValueMap.end(), storm::exceptions::InvalidAccessException, "Access to unkown identifier '" << name << "'.");
            return boost::get<double>(nameValuePair->second);
        }
        
        std::ostream& operator<<(std::ostream& stream, SimpleValuation const& valuation) {
            stream << "valuation { ";
            for (auto const& nameValuePair : valuation.identifierToValueMap) {
                stream << nameValuePair.first << ": " << nameValuePair.second << std::endl;
            }
            stream << "}";
            
            return stream;
        }
        
        std::size_t SimpleValuationPointerHash::operator()(SimpleValuation* valuation) const {
            size_t seed = 0;
            for (auto const& nameValuePair : valuation->identifierToValueMap) {
                boost::hash_combine(seed, nameValuePair.first);
                boost::hash_combine(seed, nameValuePair.second);
            }
            return seed;
        }
        
        bool SimpleValuationPointerCompare::operator()(SimpleValuation* valuation1, SimpleValuation* valuation2) const {
            return *valuation1 == *valuation2;
        }
        
        bool SimpleValuationPointerLess::operator()(SimpleValuation* valuation1, SimpleValuation* valuation2) const {
            return valuation1->identifierToValueMap < valuation2->identifierToValueMap;
        }
    }
}