artis/src/artis/context/StateValues.hpp

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

/*
 * Copyright (C) 2012-2019 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