text_shaper.hpp

// Copyright Take Vos 2021-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 "text_shaper_char.hpp"
#include "text_shaper_line.hpp"
#include "text_cursor.hpp"
#include "text_style.hpp"
#include "../layout/module.hpp"
#include "../font/font.hpp"
#include "../geometry/module.hpp"
#include "../unicode/unicode.hpp"
#include "../macros.hpp"
#include <vector>
#include <tuple>
 
namespace hi::inline v1 {
 
class text_shaper {
public:
    using char_vector = std::vector<text_shaper_char>;
    using char_iterator = char_vector::iterator;
    using char_const_iterator = char_vector::const_iterator;
    using char_reference = char_vector::reference;
    using char_const_reference = char_vector::const_reference;
    using line_vector = std::vector<text_shaper_line>;
    using line_iterator = line_vector::iterator;
    using line_const_iterator = line_vector::const_iterator;
 
    constexpr text_shaper() noexcept = default;
    constexpr text_shaper(text_shaper const&) noexcept = default;
    constexpr text_shaper(text_shaper&&) noexcept = default;
    constexpr text_shaper& operator=(text_shaper const&) noexcept = default;
    constexpr text_shaper& operator=(text_shaper&&) noexcept = default;
 
    [[nodiscard]] text_shaper(
        gstring const& text,
        text_style const& style,
        float dpi_scale,
        hi::alignment alignment,
        bool left_to_right,
        iso_15924 script = iso_15924{"Zyyy"}) noexcept :
        _bidi_context(left_to_right ? unicode_bidi_class::L : unicode_bidi_class::R),
        _dpi_scale(dpi_scale),
        _alignment(alignment),
        _script(script)
    {
        hilet& font = find_font(style->family_id, style->variant);
        _initial_line_metrics = (style->size * dpi_scale) * font.metrics;
 
        _text.reserve(text.size());
        for (hilet& c : text) {
            hilet clean_c = c == '\n' ? grapheme{unicode_PS} : c;
 
            auto& tmp = _text.emplace_back(clean_c, style, dpi_scale);
            tmp.initialize_glyph(font);
        }
 
        _text_direction = unicode_bidi_direction(
            _text.begin(),
            _text.end(),
            [](text_shaper::char_const_reference it) {
                return it.grapheme.starter();
            },
            _bidi_context);
 
        _line_break_opportunities = unicode_line_break(_text.begin(), _text.end(), [](hilet& c) -> decltype(auto) {
            return c.grapheme.starter();
        });
 
        _line_break_widths.reserve(text.size());
        for (hilet& c : _text) {
            _line_break_widths.push_back(is_visible(c.general_category) ? c.width : -c.width);
        }
 
        _word_break_opportunities = unicode_word_break(_text.begin(), _text.end(), [](hilet& c) -> decltype(auto) {
            return c.grapheme.starter();
        });
 
        _sentence_break_opportunities = unicode_sentence_break(_text.begin(), _text.end(), [](hilet& c) -> decltype(auto) {
            return c.grapheme.starter();
        });
 
        resolve_script();
    }
 
    [[nodiscard]] text_shaper(
        std::string_view text,
        text_style const& style,
        float dpi_scale,
        hi::alignment alignment,
        bool left_to_right,
        iso_15924 script = iso_15924{"Zyyy"}) noexcept :
        text_shaper(to_gstring(text), style, dpi_scale, alignment, left_to_right, script)
    {
    }
 
    [[nodiscard]] bool empty() const noexcept
    {
        return _text.empty();
    }
 
    [[nodiscard]] size_t size() const noexcept
    {
        return _text.size();
    }
 
    [[nodiscard]] char_iterator begin() noexcept
    {
        return _text.begin();
    }
 
    [[nodiscard]] char_const_iterator begin() const noexcept
    {
        return _text.begin();
    }
 
    [[nodiscard]] char_const_iterator cbegin() const noexcept
    {
        return _text.cbegin();
    }
 
    [[nodiscard]] char_iterator end() noexcept
    {
        return _text.end();
    }
 
    [[nodiscard]] char_const_iterator end() const noexcept
    {
        return _text.end();
    }
 
    [[nodiscard]] char_const_iterator cend() const noexcept
    {
        return _text.cend();
    }
 
    auto const& lines() const noexcept
    {
        return _lines;
    }
 
    [[nodiscard]] aarectangle
    bounding_rectangle(float maximum_line_width, float line_spacing = 1.0f, float paragraph_spacing = 1.5f) noexcept
    {
        hilet rectangle = aarectangle{
            point2{0.0f, std::numeric_limits<float>::lowest()}, point2{maximum_line_width, std::numeric_limits<float>::max()}};
        constexpr auto baseline = 0.0f;
        constexpr auto sub_pixel_size = extent2{1.0f, 1.0f};
 
        hilet lines = make_lines(rectangle, baseline, sub_pixel_size, line_spacing, paragraph_spacing);
        hi_assert(not lines.empty());
 
        auto max_width = 0.0f;
        for (auto& line : lines) {
            inplace_max(max_width, line.width);
        }
 
        hilet max_y = lines.front().y + std::ceil(lines.front().metrics.ascender);
        hilet min_y = lines.back().y - std::ceil(lines.back().metrics.descender);
        return aarectangle{point2{0.0f, min_y}, point2{std::ceil(max_width), max_y}};
    }
 
    [[nodiscard]] void layout(
        aarectangle rectangle,
        float baseline,
        extent2 sub_pixel_size,
        float line_spacing = 1.0f,
        float paragraph_spacing = 1.5f) noexcept
    {
        _rectangle = rectangle;
        _lines = make_lines(rectangle, baseline, sub_pixel_size, line_spacing, paragraph_spacing);
        hi_assert(not _lines.empty());
        position_glyphs(rectangle, sub_pixel_size);
    }
 
    [[nodiscard]] aarectangle rectangle() const noexcept
    {
        return _rectangle;
    }
 
    [[nodiscard]] unicode_bidi_class text_direction() const noexcept
    {
        return _text_direction;
    }
 
    [[nodiscard]] alignment resolved_alignment() const noexcept
    {
        return resolve(_alignment, _text_direction == unicode_bidi_class::L);
    }
 
    [[nodiscard]] char_const_iterator get_it(size_t index) const noexcept
    {
        if (static_cast<ptrdiff_t>(index) < 0) {
            return begin();
        } else if (index >= size()) {
            return end();
        }
 
        return begin() + index;
    }
 
    [[nodiscard]] char_const_iterator get_it(text_cursor cursor) const noexcept
    {
        return get_it(cursor.index());
    }
 
    [[nodiscard]] char_const_iterator get_it(size_t column_nr, size_t line_nr) const noexcept
    {
        hi_assert(not _lines.empty());
 
        if (static_cast<ptrdiff_t>(line_nr) < 0) {
            return begin();
        } else if (line_nr >= _lines.size()) {
            return end();
        }
 
        hilet left_of_line = static_cast<ptrdiff_t>(column_nr) < 0;
        hilet right_of_line = column_nr >= _lines[line_nr].size();
 
        if (left_of_line or right_of_line) {
            hilet ltr = _lines[line_nr].paragraph_direction == unicode_bidi_class::L;
            hilet go_up = left_of_line == ltr;
            if (go_up) {
                // Go to line above.
                if (static_cast<ptrdiff_t>(--line_nr) < 0) {
                    return begin();
                } else {
                    // Go to end of line above.
                    return _lines[line_nr].paragraph_direction == unicode_bidi_class::L ? _lines[line_nr].back() :
                                                                                          _lines[line_nr].front();
                }
 
            } else {
                // Go to the line below.
                if (++line_nr >= _lines.size()) {
                    return end();
                } else {
                    // Go to begin of line below.
                    return _lines[line_nr].paragraph_direction == unicode_bidi_class::L ? _lines[line_nr].front() :
                                                                                          _lines[line_nr].back();
                }
            }
        }
 
        return _lines[line_nr][column_nr];
    }
 
    [[nodiscard]] char_const_iterator get_it(std::pair<size_t, size_t> column_row) const noexcept
    {
        return get_it(column_row.first, column_row.second);
    }
 
    [[nodiscard]] std::pair<size_t, size_t> get_column_line(text_shaper::char_const_iterator it) const noexcept
    {
        if (it != end()) {
            return {it->column_nr, it->line_nr};
        } else {
            hi_assert(not _lines.empty());
            return {_lines.size() - 1, _lines.back().size()};
        }
    }
 
    [[nodiscard]] std::pair<size_t, size_t> get_column_line(size_t index) const noexcept
    {
        return get_column_line(get_it(index));
    }
 
    [[nodiscard]] std::pair<size_t, size_t> get_column_line(text_cursor cursor) const noexcept
    {
        return get_column_line(cursor.index());
    }
 
    [[nodiscard]] size_t get_index(text_shaper::char_const_iterator it) const noexcept
    {
        return narrow_cast<size_t>(std::distance(begin(), it));
    }
 
    [[nodiscard]] text_cursor get_begin_cursor() const noexcept
    {
        return {};
    }
 
    [[nodiscard]] text_cursor get_end_cursor() const noexcept
    {
        return text_cursor{size() - 1, true}.resize(size());
    }
 
    [[nodiscard]] text_cursor get_before_cursor(size_t index) const noexcept
    {
        return text_cursor{index, false}.resize(size());
    }
 
    [[nodiscard]] text_cursor get_after_cursor(size_t index) const noexcept
    {
        return text_cursor{index, true}.resize(size());
    }
 
    [[nodiscard]] text_cursor get_before_cursor(text_shaper::char_const_iterator it) const noexcept
    {
        return get_before_cursor(get_index(it));
    }
 
    [[nodiscard]] text_cursor get_after_cursor(text_shaper::char_const_iterator it) const noexcept
    {
        return get_after_cursor(get_index(it));
    }
 
    [[nodiscard]] text_cursor get_left_cursor(text_shaper::char_const_iterator it) const noexcept
    {
        if (it != end()) {
            if (it->direction == unicode_bidi_class::L) {
                return get_before_cursor(it);
            } else {
                return get_after_cursor(it);
            }
        } else {
            return get_end_cursor();
        }
    }
 
    [[nodiscard]] text_cursor get_right_cursor(text_shaper::char_const_iterator it) const noexcept
    {
        if (it != end()) {
            if (it->direction == unicode_bidi_class::L) {
                return get_after_cursor(it);
            } else {
                return get_before_cursor(it);
            }
        } else {
            return get_end_cursor();
        }
    }
 
    [[nodiscard]] bool is_on_left(text_cursor cursor) const noexcept
    {
        hilet it = get_it(cursor);
        if (it != end()) {
            return (it->direction == unicode_bidi_class::L) == cursor.before();
        } else {
            hi_assert(begin() == end());
            return true;
        }
    }
 
    [[nodiscard]] bool is_on_right(text_cursor cursor) const noexcept
    {
        hilet it = get_it(cursor);
        if (it != end()) {
            return (it->direction == unicode_bidi_class::L) == cursor.after();
        } else {
            hi_assert(begin() == end());
            return true;
        }
    }
 
    [[nodiscard]] text_cursor get_nearest_cursor(point2 position) const noexcept
    {
        if (_text.empty()) {
            return {};
        }
 
        hilet line_it = std::ranges::min_element(_lines, std::ranges::less{}, [position](hilet& line) {
            return std::abs(line.y - position.y());
        });
 
        if (line_it != _lines.end()) {
            hilet[char_it, after] = line_it->get_nearest(position);
            return {narrow_cast<size_t>(std::distance(_text.begin(), char_it)), after};
        } else {
            return {};
        }
    }
 
    [[nodiscard]] std::pair<text_cursor, text_cursor> select_char(text_cursor cursor) const noexcept
    {
        hilet index = cursor.index();
        return {get_before_cursor(index), get_after_cursor(index)};
    }
 
    [[nodiscard]] std::pair<text_cursor, text_cursor> select_word(text_cursor cursor) const noexcept
    {
        return get_selection_from_break(cursor, _word_break_opportunities);
    }
 
    [[nodiscard]] std::pair<text_cursor, text_cursor> select_sentence(text_cursor cursor) const noexcept
    {
        return get_selection_from_break(cursor, _sentence_break_opportunities);
    }
 
    [[nodiscard]] std::pair<text_cursor, text_cursor> select_paragraph(text_cursor cursor) const noexcept
    {
        hilet first_index = [&]() {
            auto i = cursor.index();
            while (i > 0) {
                if (_text[i - 1].general_category == unicode_general_category::Zp) {
                    return i;
                }
                --i;
            }
            return i;
        }();
        hilet last_index = [&]() {
            auto i = cursor.index();
            while (i < _text.size()) {
                if (_text[i].general_category == unicode_general_category::Zp) {
                    return i;
                }
                ++i;
            }
            return i;
        }();
 
        return {get_before_cursor(first_index), get_after_cursor(last_index)};
    }
 
    [[nodiscard]] std::pair<text_cursor, text_cursor> select_document(text_cursor cursor) const noexcept
    {
        if (_text.empty()) {
            return {{}, {}};
        }
 
        return {{}, get_end_cursor()};
    }
 
    [[nodiscard]] char_const_iterator move_left_char(char_const_iterator it) const noexcept
    {
        hilet[column_nr, line_nr] = get_column_line(it);
        return get_it(column_nr - 1, line_nr);
    }
 
    [[nodiscard]] char_const_iterator move_right_char(char_const_iterator it) const noexcept
    {
        hilet[column_nr, line_nr] = get_column_line(it);
        return get_it(column_nr + 1, line_nr);
    }
 
    [[nodiscard]] text_cursor move_left_char(text_cursor cursor, bool overwrite_mode) const noexcept
    {
        auto it = get_it(cursor);
        if (overwrite_mode) {
            it = move_left_char(it);
            return get_before_cursor(it);
 
        } else {
            if (is_on_left(cursor)) {
                // If the cursor is on the left side of a character, then move one character left.
                it = move_left_char(it);
            }
 
            return get_left_cursor(it);
        }
    }
 
    [[nodiscard]] text_cursor move_right_char(text_cursor cursor, bool overwrite_mode) const noexcept
    {
        auto it = get_it(cursor);
        if (overwrite_mode) {
            it = move_right_char(it);
            return get_before_cursor(it);
 
        } else {
            if (is_on_right(cursor)) {
                // If the cursor is on the left side of a character, then move one character left.
                it = move_right_char(it);
            }
 
            return get_right_cursor(it);
        }
    }
 
    [[nodiscard]] text_cursor move_down_char(text_cursor cursor, float& x) const noexcept
    {
        if (_text.empty()) {
            return {};
        }
 
        auto [column_nr, line_nr] = get_column_line(cursor);
        if (++line_nr == _lines.size()) {
            return get_end_cursor();
        }
 
        if (std::isnan(x)) {
            hilet char_it = get_it(cursor);
            hi_assert(char_it != _text.end());
            x = is_on_left(cursor) ? char_it->rectangle.left() : char_it->rectangle.right();
        }
 
        hilet[new_char_it, after] = _lines[line_nr].get_nearest(point2{x, 0.0f});
        return get_before_cursor(new_char_it);
    }
 
    [[nodiscard]] text_cursor move_up_char(text_cursor cursor, float& x) const noexcept
    {
        if (_text.empty()) {
            return {};
        }
 
        auto [column_nr, line_nr] = get_column_line(cursor);
        if (line_nr-- == 0) {
            return {};
        }
 
        if (std::isnan(x)) {
            auto char_it = get_it(cursor);
            hi_assert(char_it < _text.end());
            x = is_on_left(cursor) ? char_it->rectangle.left() : char_it->rectangle.right();
        }
 
        hilet[new_char_it, after] = _lines[line_nr].get_nearest(point2{x, 0.0f});
        return get_before_cursor(new_char_it);
    }
 
    [[nodiscard]] text_cursor move_left_word(text_cursor cursor, bool overwrite_mode) const noexcept
    {
        cursor = move_left_char(cursor, overwrite_mode).before_neighbor(size());
        auto it = get_it(cursor);
        while (it != end()) {
            if (it->general_category != unicode_general_category::Zs and
                _word_break_opportunities[get_index(it)] != unicode_break_opportunity::no) {
                return get_before_cursor(it);
            }
            it = move_left_char(it);
        }
        return get_end_cursor();
    }
 
    [[nodiscard]] text_cursor move_right_word(text_cursor cursor, bool overwrite_mode) const noexcept
    {
        cursor = move_right_char(cursor, overwrite_mode).before_neighbor(size());
        auto it = get_it(cursor);
        while (it != end()) {
            if (it->general_category != unicode_general_category::Zs and
                _word_break_opportunities[get_index(it)] != unicode_break_opportunity::no) {
                return get_before_cursor(it);
            }
            it = move_right_char(it);
        }
        return get_end_cursor();
    }
 
    [[nodiscard]] text_cursor move_begin_line(text_cursor cursor) const noexcept
    {
        hilet[column_nr, line_nr] = get_column_line(cursor);
        hilet& line = _lines[line_nr];
        return get_before_cursor(line.first);
    }
 
    [[nodiscard]] text_cursor move_end_line(text_cursor cursor) const noexcept
    {
        hilet[column_nr, line_nr] = get_column_line(cursor);
        hilet& line = _lines[line_nr];
 
        auto it = line.last;
        while (it != line.first) {
            --it;
            if (not it->is_trailing_white_space) {
                break;
            }
        }
 
        return get_after_cursor(it);
    }
 
    [[nodiscard]] text_cursor move_begin_sentence(text_cursor cursor) const noexcept
    {
        if (cursor.after()) {
            cursor = {cursor.index(), false};
        } else if (cursor.index() != 0) {
            cursor = {cursor.index() - 1, false};
        }
        hilet[first, last] = select_sentence(cursor);
        return first.before_neighbor(size());
    }
 
    [[nodiscard]] text_cursor move_end_sentence(text_cursor cursor) const noexcept
    {
        if (cursor.before()) {
            cursor = {cursor.index(), true};
        } else if (cursor.index() != _text.size() - 1) {
            cursor = {cursor.index() + 1, true};
        }
        hilet[first, last] = select_sentence(cursor);
        return last.before_neighbor(size());
    }
 
    [[nodiscard]] text_cursor move_begin_paragraph(text_cursor cursor) const noexcept
    {
        if (cursor.after()) {
            cursor = {cursor.index(), false};
        } else if (cursor.index() != 0) {
            cursor = {cursor.index() - 1, false};
        }
        hilet[first, last] = select_paragraph(cursor);
        return first.before_neighbor(size());
    }
 
    [[nodiscard]] text_cursor move_end_paragraph(text_cursor cursor) const noexcept
    {
        if (cursor.before()) {
            cursor = {cursor.index(), true};
        } else if (cursor.index() != _text.size() - 1) {
            cursor = {cursor.index() + 1, true};
        }
        hilet[first, last] = select_paragraph(cursor);
        return last.before_neighbor(size());
    }
 
    [[nodiscard]] text_cursor move_begin_document(text_cursor cursor) const noexcept
    {
        return {};
    }
 
    [[nodiscard]] text_cursor move_end_document(text_cursor cursor) const noexcept
    {
        if (_text.empty()) {
            return {};
        }
 
        return get_end_cursor();
    }
 
private:
    float _dpi_scale;
 
    char_vector _text;
 
    hi::alignment _alignment;
 
    unicode_break_vector _line_break_opportunities;
 
    std::vector<float> _line_break_widths;
 
    unicode_break_vector _word_break_opportunities;
 
    unicode_break_vector _sentence_break_opportunities;
 
    unicode_bidi_context _bidi_context;
 
    unicode_bidi_class _text_direction;
 
    iso_15924 _script;
 
    line_vector _lines;
 
    font_metrics _initial_line_metrics;
 
    aarectangle _rectangle;
 
    static void
    layout_lines_vertical_spacing(text_shaper::line_vector& lines, float line_spacing, float paragraph_spacing) noexcept
    {
        hi_assert(not lines.empty());
 
        auto prev = lines.begin();
        prev->y = 0.0f;
        for (auto it = prev + 1; it != lines.end(); ++it) {
            hilet height =
                prev->metrics.descender + std::max(prev->metrics.line_gap, it->metrics.line_gap) + it->metrics.ascender;
            hilet spacing = prev->last_category == unicode_general_category::Zp ? paragraph_spacing : line_spacing;
            // Lines advance downward on the y-axis.
            it->y = prev->y - spacing * height;
            prev = it;
        }
    }
 
    static void layout_lines_vertical_alignment(
        text_shaper::line_vector& lines,
        vertical_alignment alignment,
        float baseline,
        float min_y,
        float max_y,
        float sub_pixel_height) noexcept
    {
        hi_assert(not lines.empty());
 
        // Calculate the y-adjustment needed to position the base-line of the text to y=0
        auto adjustment = [&]() {
            if (alignment == vertical_alignment::top) {
                return -lines.front().y;
 
            } else if (alignment == vertical_alignment::bottom) {
                return -lines.back().y;
 
            } else {
                hilet mp_index = lines.size() / 2;
                if (lines.size() % 2 == 1) {
                    return -lines[mp_index].y;
 
                } else {
                    return -std::midpoint(lines[mp_index - 1].y, lines[mp_index].y);
                }
            }
        }();
 
        // Add the base-line to the adjustment.
        adjustment += baseline;
 
        // Clamp the adjustment between min_y and max_y.
        // The text may not fit, prioritize to show the top lines.
        if (lines.back().y + adjustment < min_y) {
            adjustment = min_y - lines.back().y;
        }
        if (lines.front().y + adjustment > max_y) {
            adjustment = max_y - lines.front().y;
        }
 
        // Reposition the lines, and round to sub-pixel boundary.
        hilet rcp_sub_pixel_height = 1.0f / sub_pixel_height;
        for (auto& line : lines) {
            line.y = std::round((line.y + adjustment) * rcp_sub_pixel_height) * sub_pixel_height;
        }
    }
 
    static void
    bidi_algorithm(text_shaper::line_vector& lines, text_shaper::char_vector& text, unicode_bidi_context bidi_context) noexcept
    {
        hi_assert(not lines.empty());
 
        // Create a list of all character indices.
        auto char_its = std::vector<text_shaper::char_iterator>{};
        // Make room for implicit line-separators.
        char_its.reserve(text.size() + lines.size());
        for (hilet& line : lines) {
            // Add all the characters of a line.
            for (auto it = line.first; it != line.last; ++it) {
                char_its.push_back(it);
            }
            if (not is_Zp_or_Zl(line.last_category)) {
                // No explicit paragraph-separator or line-separator, at a virtual one.
                char_its.push_back(text.end());
            }
        }
 
        hilet[char_its_last, paragraph_directions] = unicode_bidi(
            char_its.begin(),
            char_its.end(),
            [&](text_shaper::char_const_iterator it) {
                if (it != text.end()) {
                    return it->grapheme.starter();
                } else {
                    return unicode_LS;
                }
            },
            [&](text_shaper::char_iterator it, char32_t code_point) {
                hi_axiom(it != text.end());
                it->replace_glyph(code_point);
            },
            [&](text_shaper::char_iterator it, unicode_bidi_class direction) {
                if (it != text.end()) {
                    it->direction = direction;
                }
            },
            bidi_context);
 
        // The unicode bidi algorithm may have deleted a few characters.
        char_its.erase(char_its_last, char_its.cend());
 
        // Add the paragraph direction for each line.
        auto par_it = paragraph_directions.cbegin();
        for (auto& line : lines) {
            hi_axiom(par_it != paragraph_directions.cend());
            line.paragraph_direction = *par_it;
            if (line.last_category == unicode_general_category::Zp) {
                par_it++;
            }
        }
        hi_assert(par_it <= paragraph_directions.cend());
 
        // Add the character indices for each line in display order.
        auto line_it = lines.begin();
        line_it->columns.clear();
        auto column_nr = 0_uz;
        for (hilet char_it : char_its) {
            if (char_it == text.end()) {
                // Ignore the virtual line separators.
                continue;
            } else if (char_it >= line_it->last) {
                // Skip to the next line.
                hi_axiom(line_it->columns.size() <= narrow_cast<size_t>(std::distance(line_it->first, line_it->last)));
                ++line_it;
                line_it->columns.clear();
                column_nr = 0_uz;
            }
            hi_axiom(line_it != lines.end());
            hi_axiom(char_it >= line_it->first);
            hi_axiom(char_it < line_it->last);
            line_it->columns.push_back(char_it);
 
            // Assign line_nr and column_nr, for quick back referencing.
            char_it->line_nr = line_it->line_nr;
            char_it->column_nr = column_nr++;
        }
 
        // All of the characters in the text must be positioned.
        for (auto& c : text) {
            hi_axiom(c.line_nr != std::numeric_limits<size_t>::max() and c.column_nr != std::numeric_limits<size_t>::max());
        }
    }
 
    [[nodiscard]] static generator<std::pair<std::vector<size_t>, float>>
    get_widths(unicode_break_vector const& opportunities, std::vector<float> const& widths, float dpi_scale) noexcept
    {
        struct entry_type {
            size_t min_height;
            size_t max_height;
            float min_width;
            float max_width;
        };
 
        auto stack = std::vector<entry_type>{};
 
        hilet a4_one_column = 172.0f * 2.83465f * dpi_scale;
        hilet a4_two_column = 88.0f * 2.83465f * dpi_scale;
 
        // Max-width first.
        auto [max_width, max_lines] = detail::unicode_LB_maximum_width(opportunities, widths);
        auto height = max_lines.size();
        co_yield {std::move(max_lines), max_width};
 
        if (max_width >= a4_two_column) {
            // If this is wide text, then only try a few sizes.
            if (max_width > a4_one_column) {
                auto [width, lines] = detail::unicode_LB_width(opportunities, widths, a4_one_column);
                if (std::exchange(height, lines.size()) > lines.size()) {
                    co_yield {std::move(lines), width};
                }
            }
 
            auto [width, lines] = detail::unicode_LB_width(opportunities, widths, a4_two_column);
            if (std::exchange(height, lines.size()) > lines.size()) {
                co_yield {std::move(lines), width};
            }
 
        } else {
            // With small text we try every size that changes the number of lines.
            auto [min_width, min_lines] = detail::unicode_LB_minimum_width(opportunities, widths);
            if (min_lines.size() >= height) {
                // There are no multiple sizes.
                co_return;
            }
 
            stack.emplace_back(min_lines.size(), height, min_width, max_width);
            co_yield {std::move(min_lines), min_width};
 
            do {
                hilet entry = stack.back();
                stack.pop_back();
 
                if (entry.max_height > entry.max_height + 1 and entry.max_width >= entry.min_width + 2.0f) {
                    // There lines between the current two sizes; split in two.
                    hilet half_width = (entry.min_width + entry.max_width) * 0.5f;
 
                    auto [split_width, split_lines] = detail::unicode_LB_width(opportunities, widths, half_width);
                    hilet split_height = split_lines.size();
 
                    if (split_height == entry.min_height) {
                        // We didn't find a proper split, need to try the upper half. Use `half_width` to split right down the
                        // middle.
                        stack.emplace_back(split_height, entry.max_height, half_width, entry.max_width);
 
                    } else if (split_height == entry.max_height) {
                        // We didn't find a proper split, need to try the lower half. Use `half_width` to split right down the
                        // middle.
                        stack.emplace_back(entry.min_height, split_height, entry.min_width, half_width);
 
                    } else {
                        // Split through the middle, use the split_width for faster searching.
                        co_yield {std::move(split_lines), split_width};
                        stack.emplace_back(entry.min_height, split_height, entry.min_width, split_width);
                        stack.emplace_back(split_height, entry.max_height, split_width, entry.max_width);
                    }
                }
            } while (not stack.empty());
        }
    }
 
    [[nodiscard]] line_vector make_lines(
        aarectangle rectangle,
        float baseline,
        extent2 sub_pixel_size,
        float line_spacing,
        float paragraph_spacing) noexcept
    {
        hilet line_sizes = unicode_line_break(_line_break_opportunities, _line_break_widths, rectangle.width());
 
        auto r = text_shaper::line_vector{};
        r.reserve(line_sizes.size());
 
        auto char_it = _text.begin();
        auto width_it = _line_break_widths.begin();
        auto line_nr = 0_uz;
        for (hilet line_size : line_sizes) {
            hi_axiom(line_size > 0);
            hilet char_eol = char_it + line_size;
            hilet width_eol = width_it + line_size;
 
            hilet line_width = detail::unicode_LB_width(width_it, width_eol);
            r.emplace_back(line_nr++, _text.begin(), char_it, char_eol, line_width, _initial_line_metrics);
 
            char_it = char_eol;
            width_it = width_eol;
        }
 
        if (r.empty() or is_Zp_or_Zl(r.back().last_category)) {
            r.emplace_back(line_nr++, _text.begin(), _text.end(), _text.end(), 0.0f, _initial_line_metrics);
            r.back().paragraph_direction = _text_direction;
        }
 
        layout_lines_vertical_spacing(r, line_spacing, paragraph_spacing);
        layout_lines_vertical_alignment(
            r, _alignment.vertical(), baseline, rectangle.bottom(), rectangle.top(), sub_pixel_size.height());
 
        return r;
    }
 
    void position_glyphs(aarectangle rectangle, extent2 sub_pixel_size) noexcept
    {
        hi_assert(not _lines.empty());
 
        // The bidi algorithm will reorder the characters on each line, and mirror the brackets in the text when needed.
        bidi_algorithm(_lines, _text, _bidi_context);
        for (auto& line : _lines) {
            // Position the glyphs on each line. Possibly morph glyphs to handle ligatures and calculate the bounding rectangles.
            line.layout(_alignment.horizontal(), rectangle.left(), rectangle.right(), sub_pixel_size.width());
        }
    }
 
    void resolve_script() noexcept
    {
        // Find the first script in the text if no script is found use the text_shaper's default script.
        auto first_script = _script;
        for (auto& c : _text) {
            hilet script = ucd_get_script(c.grapheme.starter());
            if (script != iso_15924::wildcard() or script == iso_15924::uncoded() or script == iso_15924::common() or
                script == iso_15924::inherited()) {
                first_script = script;
                break;
            }
        }
 
        // Backward pass: fix start of words and open-brackets.
        // After this pass unknown-script is no longer in the text.
        // Close brackets will not be fixed, those will be fixed in the last forward pass.
        auto word_script = iso_15924::common();
        auto previous_script = first_script;
        for (auto i = std::ssize(_text) - 1; i >= 0; --i) {
            auto& c = _text[i];
 
            if (_word_break_opportunities[i + 1] != unicode_break_opportunity::no) {
                word_script = iso_15924::common();
            }
 
            c.script = ucd_get_script(c.grapheme.starter());
            if (c.script == iso_15924::uncoded() or c.script == iso_15924::common()) {
                hilet bracket_type = ucd_get_bidi_paired_bracket_type(c.grapheme.starter());
                // clang-format off
            c.script =
                bracket_type == unicode_bidi_paired_bracket_type::o ? previous_script :
                bracket_type == unicode_bidi_paired_bracket_type::c ? c.script == iso_15924::common() :
                word_script;
                // clang-format on
 
            } else if (c.script != iso_15924::inherited()) {
                previous_script = word_script = c.script;
            }
        }
 
        // Forward pass: fix all common and inherited with previous or first script.
        previous_script = first_script;
        for (auto i = 0_uz; i != _text.size(); ++i) {
            auto& c = _text[i];
 
            if (c.script == iso_15924::common() or c.script == iso_15924::inherited()) {
                c.script = previous_script;
 
            } else {
                previous_script = c.script;
            }
        }
    }
 
    [[nodiscard]] std::pair<text_cursor, text_cursor>
    get_selection_from_break(text_cursor cursor, unicode_break_vector const& break_opportunities) const noexcept
    {
        if (_text.empty()) {
            return {{}, {}};
        }
 
        // In the algorithm below we search before and after the character that the cursor is at.
        // We do not use the before/after differentiation.
 
        hilet first_index = [&]() {
            auto i = cursor.index();
            while (break_opportunities[i] == unicode_break_opportunity::no) {
                --i;
            }
            return i;
        }();
        hilet last_index = [&]() {
            auto i = cursor.index();
            while (break_opportunities[i + 1] == unicode_break_opportunity::no) {
                ++i;
            }
            return i;
        }();
 
        return {get_before_cursor(first_index), get_after_cursor(last_index)};
    }
 
    [[nodiscard]] std::pair<font_metrics, unicode_general_category>
    get_line_metrics(text_shaper::char_const_iterator first, text_shaper::char_const_iterator last) const noexcept
    {
        auto metrics = _initial_line_metrics;
        for (auto it = first; it != last; ++it) {
            // Only calculate line metrics based on visible characters.
            // For example a paragraph separator is seldom available in a font.
            if (is_visible(it->general_category)) {
                inplace_max(metrics, it->font_metrics());
            }
        }
 
        hilet last_category = (first != last) ? (last - 1)->general_category : unicode_general_category::Cn;
        return {metrics, last_category};
    }
 
    [[nodiscard]] float get_text_height(std::vector<size_t> const& lines) const noexcept
    {
        if (lines.empty()) {
            return 0.0f;
        }
 
        auto line_it = lines.cbegin();
        auto char_it_first = _text.begin();
        auto char_it_last = char_it_first + *line_it++;
 
        // Add the x-height of the first line.
        auto [previous_metrics, previous_category] = get_line_metrics(char_it_first, char_it_last);
        auto total_height = previous_metrics.x_height;
 
        for (; line_it != lines.cend(); ++line_it) {
            char_it_first = std::exchange(char_it_last, char_it_last + *line_it);
 
            // Advance to the base-line of the next line.
            auto [current_metrics, current_category] = get_line_metrics(char_it_first, char_it_last);
            hilet line_height = previous_metrics.descender + std::max(previous_metrics.line_gap, current_metrics.line_gap) +
                current_metrics.ascender;
 
            hilet spacing = previous_category == unicode_general_category::Zp ? previous_metrics.paragraph_spacing :
                                                                                previous_metrics.line_spacing;
            total_height += spacing * line_height;
 
            previous_metrics = std::move(current_metrics);
            previous_category = std::move(current_category);
        }
 
        return total_height;
    }
};
 
} // namespace hi::inline v1