formula_map_literal_node.hpp

// Copyright Take Vos 2020.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)
 
#pragma once
 
#include "formula_node.hpp"
 
namespace hi::inline v1 {
 
struct formula_map_literal_node final : formula_node {
    formula_vector keys;
    formula_vector values;
 
    formula_map_literal_node(parse_location location, formula_vector keys, formula_vector values) :
        formula_node(std::move(location)), keys(std::move(keys)), values(std::move(values))
    {
    }
 
    void post_process(formula_post_process_context &context) override
    {
        for (auto &key : keys) {
            key->post_process(context);
        }
 
        for (auto &value : values) {
            value->post_process(context);
        }
    }
 
    datum evaluate(formula_evaluation_context &context) const override
    {
        hi_assert(keys.size() == values.size());
 
        datum::map_type r;
        for (std::size_t i = 0; i < keys.size(); i++) {
            hilet &key = keys[i];
            hilet &value = values[i];
 
            r[key->evaluate(context)] = value->evaluate(context);
        }
        return datum{std::move(r)};
    }
 
    std::string string() const noexcept override
    {
        hi_assert(keys.size() == values.size());
 
        std::string r = "{";
        for (std::size_t i = 0; i < keys.size(); i++) {
            hilet &key = keys[i];
            hilet &value = values[i];
 
            if (i > 0) {
                r += ", ";
            }
            r += key->string();
            r += ": ";
            r += value->string();
        }
        return r + "}";
    }
};
 
} // namespace hi::inline v1