grid_layout.hpp

// Copyright Take Vos 2022.
// 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 "box_constraints.hpp"
#include "box_shape.hpp"
#include "spreadsheet_address.hpp"
#include "../geometry/geometry.hpp"
#include "../utility/utility.hpp"
#include "../macros.hpp"
#include <cstdint>
#include <numeric>
#include <vector>
#include <algorithm>
#include <utility>
#include <cmath>
 
hi_export_module(hikogui.layout.grid_layout);
 
hi_export namespace hi { inline namespace v1 {
namespace detail {
 
template<typename T>
struct grid_layout_cell {
    using value_type = T;
 
    size_t first_column = 0;
    size_t first_row = 0;
    size_t last_column = 0;
    size_t last_row = 0;
    bool beyond_maximum = false;
    value_type value = {};
    box_shape shape = {};
 
    constexpr grid_layout_cell() noexcept = default;
    constexpr grid_layout_cell(grid_layout_cell const&) noexcept = default;
    constexpr grid_layout_cell(grid_layout_cell&&) noexcept = default;
    constexpr grid_layout_cell& operator=(grid_layout_cell const&) noexcept = default;
    constexpr grid_layout_cell& operator=(grid_layout_cell&&) noexcept = default;
 
    template<std::convertible_to<value_type> Value>
    constexpr grid_layout_cell(
        size_t first_column,
        size_t first_row,
        size_t last_column,
        size_t last_row,
        bool beyond_maximum,
        Value&& value) noexcept :
        first_column(first_column),
        first_row(first_row),
        last_column(last_column),
        last_row(last_row),
        beyond_maximum(beyond_maximum),
        value(std::forward<Value>(value))
    {
        hi_assert(first_column < last_column);
        hi_assert(first_row < last_row);
    }
 
    constexpr void set_constraints(box_constraints const& constraints) noexcept
    {
        _constraints = constraints;
    }
 
    template<hi::axis Axis>
    [[nodiscard]] constexpr size_t first() const noexcept
    {
        if constexpr (Axis == axis::x) {
            return first_column;
        } else if constexpr (Axis == axis::y) {
            return first_row;
        } else {
            hi_static_no_default();
        }
    }
 
    template<hi::axis Axis>
    [[nodiscard]] constexpr size_t last() const noexcept
    {
        if constexpr (Axis == axis::x) {
            return last_column;
        } else if constexpr (Axis == axis::y) {
            return last_row;
        } else {
            hi_static_no_default();
        }
    }
 
    template<hi::axis Axis>
    [[nodiscard]] constexpr size_t span() const noexcept
    {
        hi_axiom(first<Axis>() < last<Axis>());
        return last<Axis>() - first<Axis>();
    }
 
    template<hi::axis Axis>
    [[nodiscard]] constexpr auto alignment() const noexcept
    {
        if constexpr (Axis == axis::x) {
            return _constraints.alignment.horizontal();
        } else if constexpr (Axis == axis::y) {
            return _constraints.alignment.vertical();
        } else {
            hi_static_no_default();
        }
    }
 
    template<hi::axis Axis>
    [[nodiscard]] constexpr float minimum() const noexcept
    {
        if constexpr (Axis == axis::x) {
            return _constraints.minimum.width();
        } else if constexpr (Axis == axis::y) {
            return _constraints.minimum.height();
        } else {
            hi_static_no_default();
        }
    }
 
    template<hi::axis Axis>
    [[nodiscard]] constexpr float preferred() const noexcept
    {
        if constexpr (Axis == axis::x) {
            return _constraints.preferred.width();
        } else if constexpr (Axis == axis::y) {
            return _constraints.preferred.height();
        } else {
            hi_static_no_default();
        }
    }
 
    template<hi::axis Axis>
    [[nodiscard]] constexpr float maximum() const noexcept
    {
        if constexpr (Axis == axis::x) {
            return _constraints.maximum.width();
        } else if constexpr (Axis == axis::y) {
            return _constraints.maximum.height();
        } else {
            hi_static_no_default();
        }
    }
 
    template<hi::axis Axis>
    [[nodiscard]] constexpr float margin_before(bool forward) const noexcept
    {
        if constexpr (Axis == axis::x) {
            if (forward) {
                return _constraints.margins.left();
            } else {
                return _constraints.margins.right();
            }
        } else if constexpr (Axis == axis::y) {
            if (forward) {
                return _constraints.margins.bottom();
            } else {
                return _constraints.margins.top();
            }
        } else {
            hi_static_no_default();
        }
    }
 
    template<hi::axis Axis>
    [[nodiscard]] constexpr float margin_after(bool forward) const noexcept
    {
        if constexpr (Axis == axis::x) {
            if (forward) {
                return _constraints.margins.right();
            } else {
                return _constraints.margins.left();
            }
        } else if constexpr (Axis == axis::y) {
            if (forward) {
                return _constraints.margins.top();
            } else {
                return _constraints.margins.bottom();
            }
        } else {
            hi_static_no_default();
        }
    }
 
private:
    box_constraints _constraints;
};
 
template<hi::axis Axis, typename T>
class grid_layout_axis_constraints {
public:
    constexpr static hi::axis axis = Axis;
 
    using value_type = T;
    using alignment_type = std::conditional_t<axis == axis::y, vertical_alignment, horizontal_alignment>;
    using cell_type = grid_layout_cell<value_type>;
    using cell_vector = std::vector<cell_type>;
 
    struct constraint_type {
        float minimum = 0.0f;

        float preferred = 0.0f;

        float maximum = std::numeric_limits<float>::max();

        float margin_before = 0.0f;

        float margin_after = 0.0f;

        alignment_type alignment = alignment_type::none;

        bool beyond_maximum = false;

        float position = 0.0f;

        float extent = 0.0f;

        std::optional<float> guideline = 0.0f;
    };
    using constraint_vector = std::vector<constraint_type>;
    using iterator = constraint_vector::iterator;
    using const_iterator = constraint_vector::const_iterator;
    using reverse_iterator = constraint_vector::reverse_iterator;
    using reference = constraint_vector::reference;
    using const_reference = constraint_vector::const_reference;
 
    constexpr ~grid_layout_axis_constraints() = default;
    constexpr grid_layout_axis_constraints() noexcept = default;
    constexpr grid_layout_axis_constraints(grid_layout_axis_constraints const&) noexcept = default;
    constexpr grid_layout_axis_constraints(grid_layout_axis_constraints&&) noexcept = default;
    constexpr grid_layout_axis_constraints& operator=(grid_layout_axis_constraints const&) noexcept = default;
    constexpr grid_layout_axis_constraints& operator=(grid_layout_axis_constraints&&) noexcept = default;
    [[nodiscard]] constexpr friend bool
    operator==(grid_layout_axis_constraints const&, grid_layout_axis_constraints const&) noexcept = default;

    constexpr grid_layout_axis_constraints(cell_vector const& cells, size_t num, bool forward) noexcept :
        _constraints(num), _forward(forward)
    {
        for (auto const& cell : cells) {
            construct_simple_cell(cell);
        }
        construct_fixup();
 
        for (auto const& cell : cells) {
            construct_span_cell(cell);
        }
        construct_fixup();
    }
 
    [[nodiscard]] constexpr float margin_before() const noexcept
    {
        return empty() ? 0 : _forward ? front().margin_before : back().margin_before;
    }
 
    [[nodiscard]] constexpr float margin_after() const noexcept
    {
        return empty() ? 0 : _forward ? back().margin_after : front().margin_after;
    }
 
    [[nodiscard]] constexpr float padding_before() const noexcept
    {
        return empty() ? 0 : _forward ? front().padding_before : back().padding_before;
    }
 
    [[nodiscard]] constexpr float padding_after() const noexcept
    {
        return empty() ? 0 : _forward ? back().padding_after : front().padding_after;
    }
 
    [[nodiscard]] constexpr std::tuple<float, float, float> update_constraints() const noexcept
    {
        return constraints(begin(), end());
    }

    [[nodiscard]] constexpr std::tuple<float, float, float> constraints(cell_type const& cell) const noexcept
    {
        return constraints(cell.template first<axis>(), cell.template last<axis>());
    }
 
    [[nodiscard]] constexpr float position(cell_type const& cell) const noexcept
    {
        return position(cell.template first<axis>(), cell.template last<axis>());
    }
 
    [[nodiscard]] constexpr float extent(cell_type const& cell) const noexcept
    {
        return extent(cell.template first<axis>(), cell.template last<axis>());
    }
 
    [[nodiscard]] constexpr std::optional<float> guideline(cell_type const& cell) const noexcept
    {
        if (cell.template span<axis>() == 1) {
            return guideline(cell.template first<axis>());
        } else {
            return std::nullopt;
        }
    }

    constexpr void layout(float new_position, float new_extent, std::optional<float> external_guideline, float guideline_width) noexcept
    {
        // Start with the extent of each constraint equal to the preferred extent.
        for (auto& constraint : _constraints) {
            constraint.extent = constraint.preferred;
        }
 
        // If the total extent is too large, shrink the constraints that allow to be shrunk.
        auto [total_extent, count] = layout_shrink(begin(), end());
        while (total_extent > new_extent and count != 0) {
            // The result may shrink slightly too much, which will be fixed by expanding in the next loop.
            std::tie(total_extent, count) = layout_shrink(begin(), end(), total_extent - new_extent, count);
        }
 
        // If the total extent is too small, expand the constraints that allow to be grown.
        std::tie(total_extent, count) = layout_expand(begin(), end());
        while (total_extent < new_extent and count != 0) {
            // The result may expand slightly too much, we don't care.
            std::tie(total_extent, count) = layout_expand(begin(), end(), new_extent - total_extent, count);
        }
 
        // If the total extent is still too small, expand into the cells that are marked beyond_maximum.
        if (total_extent < new_extent) {
            // The result may expand slightly too much, we don't care.
            count = std::count_if(begin(), end(), [](auto const& item) {
                return item.beyond_maximum;
            });
            if (count) {
                auto expand = new_extent - total_extent;
                auto const expand_per = std::ceil(expand / count);
 
                for (auto& constraint : _constraints) {
                    auto const expand_this = std::min(expand_per, expand);
                    if (constraint.beyond_maximum) {
                        constraint.extent += expand_this;
                        expand -= expand_this;
                    }
                }
            }
            total_extent = extent(cbegin(), cend());
        }
 
        // If the total extent is still too small, expand the first constrain above the maximum size.
        if (total_extent < new_extent and not empty()) {
            // The result may expand slightly too much, we don't care.
            front().extent += new_extent - total_extent;
        }
 
        if (_forward) {
            layout_position(begin(), end(), new_position, guideline_width);
        } else {
            layout_position(rbegin(), rend(), new_position, guideline_width);
        }
 
        if (external_guideline and size() == 1) {
            // When there is only 1 cell on this axis, the external guideline is used.
            // XXX If there are more cell, then the external alignment should be taken into account.
            front().guideline = *external_guideline;
        }
    }

    [[nodiscard]] constexpr size_t size() const noexcept
    {
        return _constraints.size();
    }

    [[nodiscard]] constexpr bool empty() const noexcept
    {
        return _constraints.empty();
    }

    [[nodiscard]] constexpr iterator begin() noexcept
    {
        return _constraints.begin();
    }

    [[nodiscard]] constexpr const_iterator begin() const noexcept
    {
        return _constraints.begin();
    }

    [[nodiscard]] constexpr const_iterator cbegin() const noexcept
    {
        return _constraints.cbegin();
    }

    [[nodiscard]] constexpr iterator end() noexcept
    {
        return _constraints.end();
    }

    [[nodiscard]] constexpr const_iterator end() const noexcept
    {
        return _constraints.end();
    }

    [[nodiscard]] constexpr const_iterator cend() const noexcept
    {
        return _constraints.cend();
    }

    [[nodiscard]] constexpr reverse_iterator rbegin() noexcept
    {
        return _constraints.rbegin();
    }

    [[nodiscard]] constexpr reverse_iterator rend() noexcept
    {
        return _constraints.rend();
    }

    [[nodiscard]] constexpr reference operator[](size_t index) noexcept
    {
        hi_axiom(index < size());
        return _constraints[index];
    }

    [[nodiscard]] constexpr const_reference operator[](size_t index) const noexcept
    {
        hi_axiom(index < size());
        return _constraints[index];
    }

    [[nodiscard]] constexpr reference front() noexcept
    {
        hi_axiom(not empty());
        return _constraints.front();
    }

    [[nodiscard]] constexpr const_reference front() const noexcept
    {
        hi_axiom(not empty());
        return _constraints.front();
    }

    [[nodiscard]] constexpr reference back() noexcept
    {
        hi_axiom(not empty());
        return _constraints.back();
    }

    [[nodiscard]] constexpr const_reference back() const noexcept
    {
        hi_axiom(not empty());
        return _constraints.back();
    }
 
private:
    constraint_vector _constraints = {};

    bool _forward = true;

    [[nodiscard]] constexpr std::pair<float, size_t>
    layout_shrink(const_iterator first, const_iterator last, float shrink = 0.0f, size_t count = 1) noexcept
    {
        auto const first_ = begin() + std::distance(cbegin(), first);
        auto const last_ = begin() + std::distance(cbegin(), last);
 
        hi_axiom(shrink >= 0);
 
        auto const shrink_per = std::floor(shrink / count);
 
        auto new_extent = 0.0f;
        auto new_count = 0_uz;
        for (auto it = first_; it != last_; ++it) {
            auto const shrink_this = std::max({shrink_per, shrink, it->extent - it->minimum});
            it->extent -= shrink_this;
            shrink -= shrink_this;
 
            if (it != first_) {
                new_extent += it->margin_before;
            }
            new_extent += it->extent;
 
            if (it->extent > it->minimum) {
                ++new_count;
            }
        }
 
        return {new_extent, new_count};
    }

    [[nodiscard]] constexpr std::pair<float, size_t>
    layout_expand(const_iterator first, const_iterator last, float expand = 0.0f, size_t count = 1) noexcept
    {
        auto const first_ = begin() + std::distance(cbegin(), first);
        auto const last_ = begin() + std::distance(cbegin(), last);
 
        hi_axiom(expand >= 0.0f);
 
        auto const expand_per = std::ceil(expand / count);
        hi_axiom(expand_per >= 0.0f);
 
        auto new_extent = 0.0f;
        auto new_count = 0_uz;
        for (auto it = first_; it != last_; ++it) {
            auto const expand_this = std::min({expand_per, expand, it->maximum - it->extent});
            it->extent += expand_this;
            expand -= expand_this;
 
            if (it != first_) {
                new_extent += it->margin_before;
            }
            new_extent += it->extent;
 
            if (it->extent < it->maximum) {
                ++new_count;
            }
        }
 
        return {new_extent, new_count};
    }
 
    constexpr void layout_position(auto first, auto last, float start_position, float guideline_width) noexcept
    {
        auto position = start_position;
        for (auto it = first; it != last; ++it) {
            it->position = position;
            it->guideline = make_guideline(it->alignment, position, position + it->extent, guideline_width);
 
            position += it->extent;
            position += it->margin_after;
        }
    }

    constexpr void construct_simple_cell(cell_type const& cell) noexcept
    {
        inplace_max(_constraints[cell.template first<axis>()].margin_before, cell.template margin_before<axis>(_forward));
        inplace_max(_constraints[cell.template last<axis>() - 1].margin_after, cell.template margin_after<axis>(_forward));
 
        for (auto i = cell.template first<axis>(); i != cell.template last<axis>(); ++i) {
            _constraints[i].beyond_maximum |= cell.beyond_maximum;
        }
 
        if (cell.template span<axis>() == 1) {
            inplace_max(_constraints[cell.template first<axis>()].alignment, cell.template alignment<axis>());
            inplace_max(_constraints[cell.template first<axis>()].minimum, cell.template minimum<axis>());
            inplace_max(_constraints[cell.template first<axis>()].preferred, cell.template preferred<axis>());
            inplace_min(_constraints[cell.template first<axis>()].maximum, cell.template maximum<axis>());
        }
    }

    constexpr void construct_span_cell(cell_type const& cell) noexcept
    {
        auto num_cells = narrow_cast<float>(cell.template span<axis>());
 
        if (cell.template span<axis>() > 1) {
            auto const[span_minimum, span_preferred, span_maximum] = constraints(cell);
            if (auto const extra = cell.template minimum<axis>() - span_minimum; extra > 0) {
                auto const extra_per_cell = std::floor(extra / num_cells);
                for (auto i = cell.template first<axis>(); i != cell.template last<axis>(); ++i) {
                    _constraints[i].minimum += extra_per_cell;
                }
            }
 
            if (auto const extra = cell.template preferred<axis>() - span_preferred; extra > 0) {
                auto const extra_per_cell = std::floor(extra / num_cells);
                for (auto i = cell.template first<axis>(); i != cell.template last<axis>(); ++i) {
                    _constraints[i].preferred += extra_per_cell;
                }
            }
 
            if (auto const extra = cell.template maximum<axis>() - span_preferred; extra < 0) {
                auto const extra_per_cell = std::ceil(extra / num_cells);
                for (auto i = cell.template first<axis>(); i != cell.template last<axis>(); ++i) {
                    // The maximum could become too low here, fixup() will fix this.
                    _constraints[i].maximum += extra_per_cell;
                }
            }
        }
    }

    constexpr void construct_fixup() noexcept
    {
        for (auto it = begin(); it != end(); ++it) {
            // Fix the margins so that between two constraints they are equal.
            if (it + 1 != end()) {
                it->margin_after = (it + 1)->margin_before = std::max(it->margin_after, (it + 1)->margin_before);
            }
 
            // Fix the constraints so that minimum <= preferred <= maximum.
            inplace_max(it->preferred, it->minimum);
            inplace_max(it->maximum, it->preferred);
        }
    }

    [[nodiscard]] constexpr std::tuple<float, float, float> constraints(const_iterator first, const_iterator last) const noexcept
    {
        auto r_minimum = 0.0f;
        auto r_preferred = 0.0f;
        auto r_maximum = 0.0f;
        auto r_margin = 0.0f;
 
        if (first != last) {
            r_minimum = first->minimum;
            r_preferred = first->preferred;
            r_maximum = first->maximum;
            for (auto it = first + 1; it != last; ++it) {
                r_margin += it->margin_before;
                r_minimum += it->minimum;
                r_preferred += it->preferred;
                r_maximum += it->maximum;
            }
        }
        return {r_minimum + r_margin, r_preferred + r_margin, r_maximum + r_margin};
    }

    [[nodiscard]] constexpr std::tuple<float, float, float> constraints(size_t first, size_t last) const noexcept
    {
        hi_axiom(first <= last);
        hi_axiom(last <= size());
        return constraints(begin() + first, begin() + last);
    }

    [[nodiscard]] constexpr float position(const_iterator first, const_iterator last) const noexcept
    {
        hi_axiom(first != last);
        if (_forward) {
            return first->position;
        } else {
            return (last - 1)->position;
        }
    }

    [[nodiscard]] constexpr float position(size_t first, size_t last) const noexcept
    {
        hi_axiom(first < last);
        hi_axiom(last <= size());
        return position(cbegin() + first, cbegin() + last);
    }

    [[nodiscard]] constexpr float extent(const_iterator first, const_iterator last) const noexcept
    {
        auto r = 0.0f;
        if (first != last) {
            r = first->extent;
            for (auto it = first + 1; it != last; ++it) {
                r += it->margin_before;
                r += it->extent;
            }
        }
        return r;
    }

    [[nodiscard]] constexpr float extent(size_t first, size_t last) const noexcept
    {
        hi_axiom(first <= last);
        hi_axiom(last <= size());
        return extent(cbegin() + first, cbegin() + last);
    }
 
    [[nodiscard]] constexpr std::optional<float> guideline(const_iterator it) const noexcept
    {
        return it->guideline;
    }
 
    [[nodiscard]] constexpr std::optional<float> guideline(size_t i) const noexcept
    {
        return guideline(cbegin() + i);
    }
};
 
} // namespace detail

template<typename T>
class grid_layout {
public:
    using value_type = T;
 
    using cell_type = detail::grid_layout_cell<value_type>;
    using cell_vector = std::vector<cell_type>;
    using iterator = cell_vector::iterator;
    using const_iterator = cell_vector::const_iterator;
    using reference = cell_vector::reference;
    using const_reference = cell_vector::const_reference;
 
    ~grid_layout() = default;
    constexpr grid_layout() noexcept = default;
    constexpr grid_layout(grid_layout const&) noexcept = default;
    constexpr grid_layout(grid_layout&&) noexcept = default;
    constexpr grid_layout& operator=(grid_layout const&) noexcept = default;
    constexpr grid_layout& operator=(grid_layout&&) noexcept = default;
    [[nodiscard]] constexpr friend bool operator==(grid_layout const&, grid_layout const&) noexcept = default;
 
    [[nodiscard]] constexpr bool empty() const noexcept
    {
        return _cells.empty();
    }
 
    [[nodiscard]] constexpr size_t size() const noexcept
    {
        return _cells.size();
    }
 
    [[nodiscard]] constexpr size_t num_columns() const noexcept
    {
        return _num_columns;
    }
 
    [[nodiscard]] constexpr size_t num_rows() const noexcept
    {
        return _num_rows;
    }
 
    [[nodiscard]] constexpr iterator begin() noexcept
    {
        return _cells.begin();
    }
 
    [[nodiscard]] constexpr iterator end() noexcept
    {
        return _cells.end();
    }
 
    [[nodiscard]] constexpr const_iterator begin() const noexcept
    {
        return _cells.begin();
    }
 
    [[nodiscard]] constexpr const_iterator end() const noexcept
    {
        return _cells.end();
    }
 
    [[nodiscard]] constexpr const_iterator cbegin() const noexcept
    {
        return _cells.cbegin();
    }
 
    [[nodiscard]] constexpr const_iterator cend() const noexcept
    {
        return _cells.cend();
    }
 
    [[nodiscard]] constexpr const_reference operator[](size_t i) const noexcept
    {
        return _cells[i];
    }
 
    [[nodiscard]] constexpr reference operator[](size_t i) noexcept
    {
        return _cells[i];
    }

    [[nodiscard]] constexpr bool cell_in_use(size_t first_column, size_t first_row, size_t last_column, size_t last_row) noexcept
    {
        // At least one cell must be in the range.
        hi_axiom(first_column < last_column);
        hi_axiom(first_row < last_row);
 
        for (auto const& cell : _cells) {
            if (first_column >= cell.last_column) {
                continue;
            }
            if (last_column <= cell.first_column) {
                continue;
            }
            if (first_row >= cell.last_row) {
                continue;
            }
            if (last_row <= cell.first_row) {
                continue;
            }
            return true;
        }
        return false;
    }

    template<forward_of<value_type> Value>
    constexpr reference add_cell(
        size_t first_column,
        size_t first_row,
        size_t last_column,
        size_t last_row,
        Value&& value,
        bool beyond_maximum = false) noexcept
    {
        // At least one cell must be in the range.
        hi_assert(first_column < last_column);
        hi_assert(first_row < last_row);
        hi_assert(not cell_in_use(first_column, first_row, last_column, last_row));
        auto& r = _cells.emplace_back(first_column, first_row, last_column, last_row, beyond_maximum, std::forward<Value>(value));
        update_after_insert_or_delete();
        return r;
    }

    template<forward_of<value_type> Value>
    constexpr reference add_cell(size_t column, size_t row, Value&& value, bool beyond_maximum = false) noexcept
    {
        return add_cell(column, row, column + 1, row + 1, std::forward<Value>(value), beyond_maximum);
    }
 
    constexpr void clear() noexcept
    {
        _cells.clear();
        update_after_insert_or_delete();
    }
 
    [[nodiscard]] constexpr box_constraints constraints(bool left_to_right) const noexcept
    {
        // Rows in the grid are laid out from top to bottom which is reverse from the y-axis up.
        _row_constraints = {_cells, num_rows(), false};
        _column_constraints = {_cells, num_columns(), left_to_right};
 
        auto r = box_constraints{};
        std::tie(r.minimum.width(), r.preferred.width(), r.maximum.width()) = _column_constraints.update_constraints();
        r.margins.left() = _column_constraints.margin_before();
        r.margins.right() = _column_constraints.margin_after();
 
        std::tie(r.minimum.height(), r.preferred.height(), r.maximum.height()) = _row_constraints.update_constraints();
        r.margins.bottom() = _row_constraints.margin_after();
        r.margins.top() = _row_constraints.margin_before();
 
        r.alignment = [&] {
            if (num_rows() == 1 and num_columns() == 1) {
                return hi::alignment{_column_constraints.front().alignment, _row_constraints.front().alignment};
            } else if (num_rows() == 1) {
                return hi::alignment{_row_constraints.front().alignment};
            } else if (num_columns() == 1) {
                return hi::alignment{_column_constraints.front().alignment};
            } else {
                return hi::alignment{};
            }
        }();
 
        return r;
    }

    constexpr void set_layout(box_shape const& shape, float baseline_adjustment) noexcept
    {
        // Rows in the grid are laid out from top to bottom which is reverse from the y-axis up.
        _column_constraints.layout(shape.x(), shape.width(), shape.centerline, 0);
        _row_constraints.layout(shape.y(), shape.height(), shape.baseline, baseline_adjustment);
 
        // Assign the shape for each cell.
        for (auto& cell : _cells) {
            cell.shape.rectangle = {
                _column_constraints.position(cell),
                _row_constraints.position(cell),
                _column_constraints.extent(cell),
                _row_constraints.extent(cell)};
            cell.shape.centerline = _column_constraints.guideline(cell);
            cell.shape.baseline = _row_constraints.guideline(cell);
        }
    }
 
private:
    cell_vector _cells = {};
    size_t _num_rows = 0;
    size_t _num_columns = 0;
    mutable detail::grid_layout_axis_constraints<axis::y, value_type> _row_constraints = {};
    mutable detail::grid_layout_axis_constraints<axis::x, value_type> _column_constraints = {};

    constexpr void sort_cells() noexcept
    {
        std::sort(_cells.begin(), _cells.end(), [](cell_type const& lhs, cell_type const& rhs) {
            if (lhs.first_row != rhs.first_row) {
                return lhs.first_row < rhs.first_row;
            } else {
                return lhs.first_column < rhs.first_column;
            }
        });
    }

    constexpr void update_after_insert_or_delete() noexcept
    {
        sort_cells();
 
        _num_rows = 0;
        _num_columns = 0;
        for (auto const& cell : _cells) {
            inplace_max(_num_rows, cell.last_row);
            inplace_max(_num_columns, cell.last_column);
        }
    }
};
 
}} // namespace hi::v1