/**
 * @file artis/context/StateValues.hpp
 * @author See the AUTHORS file
 */

/*
 * Copyright (C) 2012-2017 ULCO http://www.univ-littoral.fr
 *
 * 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 3 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, see <http://www.gnu.org/licenses/>.
 */

#ifndef __ARTIS_KERNEL_STATE_VALUES_HPP
#define __ARTIS_KERNEL_STATE_VALUES_HPP

#include <artis/context/Value.hpp>

#include <map>

#include <boost/serialization/serialization.hpp>
#include <boost/serialization/map.hpp>

namespace artis { namespace context {

class StateValues
{
public:
    StateValues()
    { }

    virtual ~StateValues()
    { }

    void add_external(unsigned int key, const Value& value)
    { _externals[key] = value; }

    void add_internal(unsigned int key, const Value& value)
    { _internals[key] = value; }

    void add_state(unsigned int key, const Value& value)
    { _states[key] = value; }

    const Value& get_external(unsigned int key) const
    {
        std::map < unsigned int, Value >::const_iterator it =
            _externals.find(key);

        if (it != _externals.end()) {
            return it->second;
        } else {
            assert(false);
            return it->second;
        }
    }

    const Value& get_internal(unsigned int key) const
    {
        std::map < unsigned int, Value >::const_iterator it =
            _internals.find(key);

        if (it != _internals.end()) {
            return it->second;
        } else {
            assert(false);
            return it->second;
        }
    }

    const Value& get_state(unsigned int key) const
    {
        std::map < unsigned int, Value >::const_iterator it =
            _states.find(key);

        if (it != _states.end()) {
            return it->second;
        } else {
            assert(false);
            return it->second;
        }
    }

    std::string to_string() const
    {
        std::string str = "externals: [ ";

        for (std::map < unsigned int, Value >::const_iterator it =
                 _externals.begin(); it != _externals.end(); ++it) {
            str += it->second.to_string() + " ";
        }
        str += "]; internals: [ ";
        for (std::map < unsigned int, Value >::const_iterator it =
                 _internals.begin(); it != _internals.end(); ++it) {
            str += it->second.to_string() + " ";
        }
        str += "]; states: [ ";
        for (std::map < unsigned int, Value >::const_iterator it =
                 _states.begin(); it != _states.end(); ++it) {
            str += it->second.to_string() + " ";
        }
        str += "]";
        return str;
    }

private:
    friend class boost::serialization::access;

    template < class Archive >
    void serialize(Archive & ar, const unsigned int version)
    {
        (void) version;

        ar & _externals;
        ar & _internals;
        ar & _states;
    }

    std::map < unsigned int, Value > _externals;
    std::map < unsigned int, Value > _internals;
    std::map < unsigned int, Value > _states;
};

} }

#endif