units_dialog.cpp

Go to the documentation of this file.

/*
    Copyright (C) 2024 - 2025
    Part of the Battle for Wesnoth Project https://www.wesnoth.org/
 
    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 2 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.
 
    See the COPYING file for more details.
*/
 
#define GETTEXT_DOMAIN "wesnoth-lib"
 
#include "gui/dialogs/units_dialog.hpp"
 
#include "font/standard_colors.hpp"
#include "formatter.hpp"
#include "game_board.hpp"
#include "gettext.hpp"
#include "gui/dialogs/edit_text.hpp"
#include "gui/dialogs/message.hpp"
#include "gui/dialogs/transient_message.hpp"
#include "gui/widgets/listbox.hpp"
#include "gui/widgets/button.hpp"
#include "gui/widgets/label.hpp"
#include "gui/widgets/menu_button.hpp"
#include "gui/widgets/text_box.hpp"
#include "gui/widgets/toggle_button.hpp"
#include "gui/widgets/unit_preview_pane.hpp"
#include "gui/widgets/window.hpp"
#include "help/help.hpp"
#include "log.hpp"
#include "replay_helper.hpp"
#include "play_controller.hpp"
#include "resources.hpp"
#include "serialization/markup.hpp"
#include "synced_context.hpp"
#include "team.hpp"
#include "units/helper.hpp"
#include "units/unit.hpp"
#include "units/ptr.hpp"
#include "units/types.hpp"
#include "utils/ci_searcher.hpp"
#include "whiteboard/manager.hpp"
 
#include <functional>
#include <string>
 
static lg::log_domain log_display("display");
#define LOG_DP LOG_STREAM(info, log_display)
 
namespace gui2::dialogs
{
 
namespace
{
// Index 2 is by-level
std::pair sort_default{ std::string{"unit_name"}, sort_order::type::ascending };
utils::optional<decltype(sort_default)> sort_last;
 
std::string star(bool starred)
{
    // Filled/unfilled five-pointed stars, used as favorite unit marker
    return starred ? "\u2605" : "\u2606";
}
 
}
 
REGISTER_DIALOG(units_dialog)
 
units_dialog::units_dialog()
    : modal_dialog(window_id())
    , selected_index_(-1)
    , num_rows_(0)
    , ok_label_(_("OK"))
    , cancel_label_(_("Cancel"))
    , show_rename_(false)
    , show_dismiss_(false)
    , show_mark_favorite_(false)
    , show_variations_(false)
    , sort_order_(sort_default)
    , gender_(unit_race::GENDER::MALE)
    , variation_()
    , filter_options_()
    , gender_toggle_()
{
    for(widget* sort_toggle : find_widget<grid>("_header_grid")) {
        // FIXME: only proper (non-spacer) header options are currently given an ID.
        // We want the spacers to remain for sizing, but this is pretty fragile.
        if(!sort_toggle->id().empty()) {
            sort_toggle->set_visible(widget::visibility::invisible);
        }
    }
}
 
namespace {
 
template<typename T>
void dump_recall_list_to_console(const T& units)
{
    log_scope2(log_display, "dump_recall_list_to_console()")
 
    LOG_DP << "size: " << units.size();
 
    std::size_t idx = 0;
    for(const auto& u_ptr : units) {
        LOG_DP << "\tunit[" << (idx++) << "]: " << u_ptr->id() << " name = '" << u_ptr->name() << "'";
    }
}
 
std::string get_title_suffix(int side_num)
{
    if(!resources::gameboard) {
        return "";
    }
 
    unit_map& units = resources::gameboard->units();
 
    int controlled_recruiters = 0;
    for(const auto& team : resources::gameboard->teams()) {
        if(team.is_local_human() && !team.recruits().empty() && units.find_leader(team.side()) !=units.end()) {
            ++controlled_recruiters;
        }
    }
 
    std::stringstream msg;
    if(controlled_recruiters >= 2) {
        unit_map::const_iterator leader = resources::gameboard->units().find_leader(side_num);
        if(leader != resources::gameboard->units().end() && !leader->name().empty()) {
            msg << " (" << leader->name() << ")";
        }
    }
 
    return msg.str();
}
}
 
void units_dialog::pre_show()
{
    text_box& filter = find_widget<text_box>("filter_box");
    filter.on_modified([this](const auto& box) { filter_text_changed(box.text()); });
 
    listbox& list = find_widget<listbox>("main_list");
    connect_signal_notify_modified(list, std::bind(&units_dialog::list_item_clicked, this));
 
    // Prevent close via OK (including keyboard Enter) if no units are available after filter.
    set_exit_hook(window::exit_hook::ok_only, [&list] { return list.get_selected_row() != -1; });
 
    connect_signal_mouse_left_click(
        find_widget<button>("show_help"),
        std::bind(&units_dialog::show_help, this));
 
#ifdef __IPHONEOS__
    // On iOS, opening these dialogs should not immediately summon the software
    // keyboard just because the optional filter box exists.
    keyboard_capture(&list);
#else
    keyboard_capture(&filter);
#endif
    add_to_keyboard_chain(&list);
 
    show_list(list);
 
    find_widget<label>("title").set_label(title_);
    find_widget<button>("ok").set_label(ok_label_);
    find_widget<button>("cancel").set_label(cancel_label_);
    find_widget<button>("dismiss").set_visible(show_dismiss_);
    find_widget<button>("rename").set_visible(show_rename_);
    find_widget<button>("mark_favorite").set_visible(show_mark_favorite_);
    find_widget<grid>("variation_gender_grid").set_visible(show_variations_);
 
    // Gender and variation selectors
    if(show_variations_) {
        connect_signal_notify_modified(
            find_widget<menu_button>("variation_box"), [this](auto&&...) { update_variation(); });
 
        auto& group = get_toggle();
        group.add_member(find_widget<toggle_button>("male_toggle", true, true), unit_race::MALE);
        group.add_member(find_widget<toggle_button>("female_toggle", true, true), unit_race::FEMALE);
 
        group.set_member_states(unit_race::MALE);
        group.on_modified(
            [this](widget&, const unit_race::GENDER& gender) { update_gender(gender); });
    }
 
    list_item_clicked();
}
 
void units_dialog::show_list(listbox& list)
{
    if (num_rows_ == 0) {
        return;
    }
 
    for(std::size_t i = 0; i < num_rows_; i++) {
        widget_data row_data;
 
        // If custom filter text generator exists, use it to generate the filter text
        if(filter_gen_) {
            filter_options_.emplace_back(filter_gen_(i));
        } else {
            filter_options_.emplace_back();
        }
 
        for(const auto& [id, generator] : column_generators_) {
            auto result = std::invoke(generator, i);
 
            // Register labels to be filtered upon provided no filter generator exists
            if(!filter_gen_ && id != "unit_image") {
                filter_options_.back().push_back(result);
            }
 
            row_data.emplace(id, widget_item{{ "use_markup", "true" }, { "label", std::move(result) }});
        }
 
        grid& row_grid = list.add_row(row_data);
 
        // Generate tooltip for ith row
        if(tooltip_gen_) {
            row_grid.find_widget<styled_widget>("row_panel").set_tooltip(tooltip_gen_(i));
        }
    }
 
    const auto [sorter_id, order] = sort_last.value_or(sort_order_);
    list.set_active_sorter(sorter_id, order, true);
    if (is_selected()) { // In this case, is an entry preselected by caller?
        list.select_row(selected_index_);
        fire(event::NOTIFY_MODIFIED, *this, nullptr);
    }
}
 
void units_dialog::rename_unit(std::vector<unit_const_ptr>& unit_list)
{
    listbox& list = find_widget<listbox>("main_list");
 
    selected_index_ = list.get_selected_row();
    if (selected_index_ == -1) {
        return;
    }
 
    unit& selected_unit = const_cast<unit&>(*unit_list[selected_index_]);
 
    std::string name = selected_unit.name();
 
    if(gui2::dialogs::edit_text::execute(_("Rename Unit"), _("Name:"), name)) {
        selected_unit.rename(name);
 
        list.get_row_grid(selected_index_)->find_widget<label>("unit_name").set_label(name);
 
        filter_options_.erase(filter_options_.begin() + selected_index_);
        filter_options_.insert(filter_options_.begin() + selected_index_, filter_gen_(selected_index_));
 
        list_item_clicked();
        invalidate_layout();
    }
}
 
void units_dialog::dismiss_unit(std::vector<unit_const_ptr>& unit_list, const team& team)
{
    LOG_DP << "Recall list units:"; dump_recall_list_to_console(unit_list);
 
    listbox& list = find_widget<listbox>("main_list");
    selected_index_ = list.get_selected_row();
    if (selected_index_ == -1) {
        return;
    }
 
    const unit& u = *unit_list[selected_index_].get();
 
    if(!u.dismissable()) {
        gui2::show_transient_message("", u.block_dismiss_message());
        return;
    }
 
    // If the unit is of level > 1, or is close to advancing, we warn the player about it
    std::stringstream message;
    if(u.loyal()) {
        message << _("This unit requires no upkeep.") << " " << (u.gender() == unit_race::MALE
                 ? _("Do you really want to dismiss him?")
                 : _("Do you really want to dismiss her?"));
 
    } else if(u.level() > 1) {
        message << _("This unit is an experienced one, having advanced levels.") << " " << (u.gender() == unit_race::MALE
                 ? _("Do you really want to dismiss him?")
                 : _("Do you really want to dismiss her?"));
 
    } else if(u.experience() > u.max_experience()/2) {
        message << _("This unit is close to advancing a level.") << " " << (u.gender() == unit_race::MALE
                 ? _("Do you really want to dismiss him?")
                 : _("Do you really want to dismiss her?"));
    }
 
    if(!message.str().empty()) {
        const int res = gui2::show_message(_("Dismiss Unit"), message.str(), message::yes_no_buttons);
        if(res != gui2::retval::OK) {
            return;
        }
    }
 
    unit_list.erase(unit_list.begin() + selected_index_);
 
    // Remove the entry from the filter list
    filter_options_.erase(filter_options_.begin() + selected_index_);
 
    // Remove the entry from the dialog list
    list.remove_row(selected_index_);
 
    // This line can change selected_index_, so the erasing needs to be done before it
    list_item_clicked();
 
    assert(filter_options_.size() == list.get_item_count());
 
    LOG_DP << "Dismissing a unit, side = " << u.side() << ", id = '" << u.id() << "'";
    LOG_DP << "That side's recall list:";
    dump_recall_list_to_console(team.recall_list());
 
    // Find the unit in the recall list.
    unit_const_ptr dismissed_unit = team.recall_list().find_if_matches_id(u.id());
    assert(dismissed_unit);
 
    // Record the dismissal, then delete the unit.
    synced_context::run_and_throw("disband", replay_helper::get_disband(dismissed_unit->id()));
 
    // Close the dialog if all units are dismissed
    if(list.get_item_count() == 0) {
        set_retval(retval::CANCEL);
    }
}
 
void units_dialog::toggle_favorite(std::vector<unit_const_ptr>& unit_list)
{
    listbox& list = find_widget<listbox>("main_list");
 
    selected_index_ = list.get_selected_row();
    if (selected_index_ == -1) {
        return;
    }
 
    unit& selected_unit = const_cast<unit&>(*unit_list[selected_index_]);
    selected_unit.set_favorite(!selected_unit.favorite());
 
    list.get_row_grid(selected_index_)->find_widget<label>("unit_favorite")
        .set_label(star(selected_unit.favorite()));
 
    list_item_clicked();
    invalidate_layout();
}
 
void units_dialog::show_help() const
{
    if (!topic_id_.empty()) {
        help::show_help(topic_id_);
    }
}
 
void units_dialog::list_item_clicked()
{
    selected_index_ = find_widget<listbox>("main_list").get_selected_row();
 
    if (selected_index_ == -1) {
        return;
    }
 
    fire(event::NOTIFY_MODIFIED, *this, nullptr);
}
 
void units_dialog::post_show()
{
    listbox& list = find_widget<listbox>("main_list");
    if(const auto [sorter, order] = list.get_active_sorter(); sorter) {
        sort_last.emplace(sorter->id(), order);
    } else {
        sort_last.reset();
    }
 
    selected_index_ = get_retval() == retval::OK ? list.get_selected_row() : -1;
}
 
void units_dialog::filter_text_changed(const std::string& text)
{
    const std::size_t shown = find_widget<listbox>("main_list")
        .filter_rows_by([this, match = translation::make_ci_matcher(text)](std::size_t row) {
            return match(filter_options_[row]);
        });
 
    // Disable rename, dismiss, and OK buttons if no units are shown
    find_widget<button>("rename").set_active(shown > 0);
    find_widget<button>("dismiss").set_active(shown > 0);
    find_widget<button>("ok").set_active(shown > 0);
}
 
void units_dialog::update_gender(const unit_race::GENDER val)
{
    gender_ = val;
 
    selected_index_ = find_widget<listbox>("main_list").get_selected_row();
    if(selected_index_ == -1) {
        return;
    }
 
    fire(event::NOTIFY_MODIFIED, *this, nullptr);
}
 
void units_dialog::update_variation()
{
    variation_ = find_widget<menu_button>("variation_box").get_value_config()["variation_id"].str();
 
    selected_index_ = find_widget<listbox>("main_list").get_selected_row();
    if(selected_index_ == -1) {
        return;
    }
 
    fire(event::NOTIFY_MODIFIED, *this, nullptr);
}
 
// } -------------------- BUILDERS -------------------- {
std::unique_ptr<units_dialog> units_dialog::build_create_dialog(const std::vector<const unit_type*>& types_list)
{
    auto dlg = std::make_unique<units_dialog>();
 
    const auto type_gen = [](const auto& type) {
        std::string type_name = type->type_name();
        if(type_name != type->id()) {
            type_name += " (" + type->id() + ")";
        }
        return type_name;
    };
 
    const auto race_gen = [](const auto& type) {
        return type->race()->plural_name();
    };
 
    const auto populate_variations = [](units_dialog* dlg, const unit_type& ut) {
        // Populate variations box
        menu_button& var_box = dlg->find_widget<menu_button>("variation_box");
        std::vector<config> var_box_values;
        var_box_values.emplace_back("label", _("unit_variation^Default Variation"), "variation_id", "");
 
        const auto& uvars = ut.variation_types();
 
        var_box.set_active(!uvars.empty());
 
        unsigned n = 0, selection = 0;
 
        for(const auto& [uv_id, uv] : uvars) {
            ++n;
 
            std::string uv_label;
            if(!uv.variation_name().empty()) {
                uv_label = uv.variation_name() + " (" + uv_id + ")";
            } else if(!uv.type_name().empty() && uv.type_name() != ut.type_name()) {
                uv_label = uv.type_name() + " (" + uv_id + ")";
            } else {
                uv_label = uv_id;
            }
 
            var_box_values.emplace_back("label", uv_label, "variation_id", uv_id);
 
            if(uv_id == dlg->variation()) {
                selection = n;
            }
        }
 
        // If we didn't find the variation selection again then the new selected
        // unit type doesn't have that variation id.
        if(!selection) {
            dlg->clear_variation();
        }
 
        var_box.set_values(var_box_values, selection);
    };
 
    dlg->set_title(_("Create Unit"))
        .set_ok_label(_("Create"))
        .set_help_topic("..units")
        .set_row_num(types_list.size())
        .set_show_variations(true);
 
    // Listbox data
    auto set_column = dlg->make_column_builder(types_list);
 
    set_column("unit_name", type_gen, sort_type::generator);
    set_column("unit_details", race_gen, sort_type::generator);
 
    dlg->on_modified([populate_variations, &types_list](units_dialog* dlg, std::size_t index) -> const auto& {
        const unit_type* ut = types_list[index];
 
        if (dlg->is_selected() && (static_cast<int>(index) == dlg->get_selected_index())) {
            dlg->get_toggle().set_member_states(dlg->gender());
        } else {
            dlg->get_toggle().set_members_enabled(
                [ut](const unit_race::GENDER& gender) { return ut->has_gender_variation(gender); });
        }
 
        populate_variations(dlg, *ut);
 
        const auto& g = dlg->gender();
        if(ut->has_gender_variation(g)) {
            ut = &ut->get_gender_unit_type(g);
        }
 
        const auto& var = dlg->variation();
        if(!var.empty()) {
            ut = &ut->get_variation(var);
        }
 
        return *ut;
    });
 
    return dlg;
}
 
std::unique_ptr<units_dialog> units_dialog::build_recruit_dialog(
    const std::vector<const unit_type*>& recruit_list,
    recruit_msgs_map& err_msgs_map,
    const team& team
)
{
    auto dlg = std::make_unique<units_dialog>();
    auto set_column = dlg->make_column_builder(recruit_list);
 
    set_column("unit_image", [&](const auto& recruit) {
        std::string image_string = recruit->icon();
        if (image_string.empty()) {
            image_string = recruit->image();
        }
        image_string += "~RC(" + recruit->flag_rgb() + ">" + team.color() + ")";
        image_string += "~SCALE_INTO(72,72)";
        // Does the unit have error message? If so, grey out image.
        if (!err_msgs_map[recruit].empty()) {
            image_string += "~GS()";
        }
        return image_string;
    }, sort_type::none);
 
    set_column("unit_details", [&](const auto& recruit) {
        // Does the unit have error message? If so, grey out text here.
        bool recruitable = err_msgs_map[recruit].empty();
        return unit_helper::maybe_inactive(recruit->type_name(), recruitable)
            + '\n'
            + unit_helper::format_cost_string(recruit->cost(), recruitable);
    }, sort_type::generator);
 
    dlg->set_title(_("Recruit Unit") + get_title_suffix(team.side()))
        .set_ok_label(_("Recruit"))
        .set_help_topic("recruit_and_recall")
        .set_row_num(recruit_list.size());
 
    dlg->set_tooltip_generator([&](std::size_t index) {
        // Show the error message in case of disabled units, if any.
        return err_msgs_map[recruit_list[index]];
    });
 
    dlg->on_modified([&recruit_list](units_dialog*, std::size_t index) -> const auto& { return *recruit_list[index]; });
 
    return dlg;
}
 
std::unique_ptr<units_dialog> units_dialog::build_unit_list_dialog(std::vector<unit_const_ptr>& unit_list)
{
    auto dlg = std::make_unique<units_dialog>();
    dlg->set_title(_("Unit List"))
        .set_ok_label(_("Scroll To"))
        .set_help_topic("..units")
        .set_row_num(unit_list.size())
        .set_show_rename(true)
        .set_show_favorite(true);
 
    // Rename functionality
    button& rename = dlg->find_widget<button>("rename");
    connect_signal_mouse_left_click(rename, [&](auto&&...) {
        dlg->rename_unit(unit_list);
    });
 
    // Mark favorite functionality
    button& favorite = dlg->find_widget<button>("mark_favorite");
    connect_signal_mouse_left_click(favorite, [&](auto&&...) {
        dlg->toggle_favorite(unit_list);
    });
 
    auto set_column = dlg->make_column_builder(unit_list);
 
    set_column("unit_favorite", [](const auto& unit) {
        return star(unit->favorite());
    }, sort_type::generator);
 
    set_column("unit_name",
        [](const auto& unit) {
            return !unit->name().empty() ? unit->name().str() : font::unicode_en_dash;
        }, sort_type::generator);
 
    set_column("unit_details",
        [](const auto& unit) {
            return unit->type_name().str();
        }, sort_type::generator);
 
    set_column("unit_level",
        [](const auto& unit) {
            return unit_helper::format_level_string(unit->level(), true);
        },
        [](const auto& unit) {
            return std::tuple(unit->level(), -static_cast<int>(unit->experience_to_advance()));
        });
 
    set_column("unit_moves",
        [](const auto& unit) {
            return unit_helper::format_movement_string(unit->movement_left(), unit->total_movement());
        },
        [](const auto& unit) {
            return unit->movement_left();
        });
 
    set_column("unit_hp",
        [](const auto& unit) {
            return markup::span_color(unit->hp_color(), unit->hitpoints(), "/", unit->max_hitpoints());
        },
        [](const auto& unit) {
            return unit->hitpoints();
        });
 
    set_column("unit_xp",
        [](const auto& unit) {
            std::stringstream exp_str;
            if(unit->can_advance()) {
                exp_str << unit->experience() << "/" << unit->max_experience();
            } else {
                exp_str << font::unicode_en_dash;
            }
            return markup::span_color(unit->xp_color(), exp_str.str());
        },
        [](const auto& unit) {
            // this allows 0/35, 0/100 etc to be sorted
            // also sorts 23/35 before 0/35, after which 0/100 comes
            return unit->experience() + unit->max_experience();
        });
 
    set_column("unit_status", [](const auto& unit) {
        // Status
        if(unit->incapacitated()) {
            return "misc/petrified.png";
        }
 
        if(unit->poisoned()) {
            return "misc/poisoned.png";
        }
 
        if(unit->slowed()) {
            return "misc/slowed.png";
        }
 
        if(unit->invisible(unit->get_location(), false)) {
            return "misc/invisible.png";
        }
 
        return "";
    }, sort_type::none);
 
    set_column("unit_traits", [](const auto& unit) {
        return utils::join(unit->trait_names(), ", ");
    }, sort_type::generator);
 
    dlg->set_filter_generator([&unit_list](std::size_t index) {
        const auto& unit = unit_list[index];
        std::vector<std::string> filter_keys;
 
        filter_keys.emplace_back(unit->type_name());
        filter_keys.emplace_back(!unit->name().empty() ? unit->name().str() : font::unicode_en_dash);
        filter_keys.emplace_back(std::to_string(unit->level()));
        filter_keys.emplace_back(unit_type::alignment_description(unit->alignment(), unit->gender()));
 
        if(const auto* race = unit->race()) {
            filter_keys.emplace_back(race->name(unit->gender()));
            filter_keys.emplace_back(race->plural_name());
        }
 
        for(const std::string& trait : unit->trait_names()) {
            filter_keys.emplace_back(trait);
        }
 
        return filter_keys;
    });
 
    dlg->on_modified([&unit_list, &rename](units_dialog*, std::size_t index) -> const auto& {
        auto& unit = unit_list[index];
        rename.set_active(!unit->unrenamable());
        return *unit;
    });
 
    return dlg;
}
 
std::unique_ptr<units_dialog> units_dialog::build_recall_dialog(
    std::vector<unit_const_ptr>& recall_list,
    const team& team)
{
    int wb_gold = unit_helper::planned_gold_spent(team.side());
 
    // Lambda to check if a unit is recallable
    const auto recallable = [wb_gold, &team](const unit& unit) {
        // Note: Our callers apply [filter_recall], but leave it to us to apply cost-based filtering.
        const int recall_cost = unit.recall_cost() > -1 ? unit.recall_cost() : team.recall_cost();
        return recall_cost <= team.gold() - wb_gold;
    };
 
    auto dlg = std::make_unique<units_dialog>();
    dlg->set_title(_("Recall Unit") + get_title_suffix(team.side()))
        .set_ok_label(_("Recall"))
        .set_help_topic("recruit_and_recall")
        .set_row_num(recall_list.size())
        .set_show_rename(true)
        .set_show_dismiss(true)
        .set_show_favorite(true);
 
    // Rename functionality
    button& rename = dlg->find_widget<button>("rename");
    connect_signal_mouse_left_click(rename, [&](auto&&...) {
        dlg->rename_unit(recall_list);
    });
 
    // Dismiss functionality
    button& dismiss = dlg->find_widget<button>("dismiss");
    connect_signal_mouse_left_click(dismiss, [&](auto&&...) {
        dlg->dismiss_unit(recall_list, team);
    });
 
    // Mark favorite functionality
    button& favorite = dlg->find_widget<button>("mark_favorite");
    connect_signal_mouse_left_click(favorite, [&](auto&&...) {
        dlg->toggle_favorite(recall_list);
    });
 
    auto set_column = dlg->make_column_builder(recall_list);
 
    set_column("unit_favorite", [](const auto& unit) {
        return star(unit->favorite());
    }, sort_type::generator);
 
    set_column("unit_image", [recallable](const auto& unit) {
        std::string mods = unit->image_mods();
        if(unit->can_recruit()) { mods += "~BLIT(" + unit::leader_crown() + ")"; }
        for(const std::string& overlay : unit->overlays()) {
            mods += "~BLIT(" + overlay + ")";
        }
        if(!recallable(*unit)) { mods += "~GS()"; }
        mods += "~SCALE_INTO(72,72)";
        return unit->absolute_image() + mods;
    }, sort_type::none);
 
    set_column("unit_name",
        [recallable](const auto& unit) {
            const std::string& name = !unit->name().empty() ? unit->name().str() : font::unicode_en_dash;
            return unit_helper::maybe_inactive(name, recallable(*unit));
        },
        [](const auto& unit) {
            return unit->name().str();
        });
 
    set_column("unit_details",
        [recallable, &team](const auto& unit) {
            std::stringstream details;
            details << unit_helper::maybe_inactive(unit->type_name().str(), recallable(*unit)) << '\n';
            const int recall_cost = unit->recall_cost() == -1 ? team.recall_cost() : unit->recall_cost();
            if (recallable(*unit)) {
                details << unit_helper::format_cost_string(recall_cost, team.recall_cost());
            } else {
                details << unit_helper::format_cost_string(recall_cost, false);
            }
            return details.str();
        },
        [](const auto& unit) {
            return unit->type_name().str();
        });
 
    set_column("unit_moves",
        [recallable](const auto& unit) {
            return unit_helper::format_movement_string(
                unit->movement_left(), unit->total_movement(), recallable(*unit));
        },
        [](const auto& unit) {
            return unit->movement_left();
        });
 
    set_column("unit_level",
        [recallable](const auto& unit) {
            return unit_helper::format_level_string(unit->level(), recallable(*unit));
        },
        [](const auto& unit) {
            return std::tuple(unit->level(), -static_cast<int>(unit->experience_to_advance()));
        });
 
    set_column("unit_hp",
        [recallable](const auto& unit) {
            const color_t& col = recallable(*unit) ? unit->hp_color() : font::GRAY_COLOR;
            return markup::span_color(col, unit->hitpoints(), "/", unit->max_hitpoints());
        },
        [](const auto& unit) {
            return unit->hitpoints();
        });
 
    set_column("unit_xp",
        [recallable](const auto& unit) {
            const color_t& col = recallable(*unit) ? unit->xp_color() : font::GRAY_COLOR;
            if(unit->can_advance()) {
                return markup::span_color(col, unit->experience(), "/", unit->max_experience());
            } else {
                return markup::span_color(col, font::unicode_en_dash);
            }
        },
        [](const auto& unit) {
            return std::tuple(static_cast<int>(unit->experience_to_advance()), unit->max_experience());
        });
 
    set_column("unit_traits",
        [recallable](const auto& unit) {
            std::string traits;
            for(const std::string& trait : unit->trait_names()) {
                traits += (traits.empty() ? "" : "\n") + trait;
            }
            return unit_helper::maybe_inactive((!traits.empty() ? traits : font::unicode_en_dash), recallable(*unit));
        },
        [](const auto& unit) {
            return !unit->trait_names().empty() ? unit->trait_names().front().str() : "";
        });
 
 
    dlg->set_sort_by(std::pair("unit_level", sort_order::type::descending));
 
    dlg->set_tooltip_generator([recallable, wb_gold, &recall_list](std::size_t index) {
        if(recallable(*recall_list[index])) {
            return std::string();
        }
 
        // Just set the tooltip on every single element in this row.
        if(wb_gold > 0) {
            return _("This unit cannot be recalled because you will not have enough gold at this point in your plan.");
        } else {
            return _("This unit cannot be recalled because you do not have enough gold.");
        }
    });
 
    dlg->set_filter_generator([&recall_list](std::size_t index) {
        const auto& unit = recall_list[index];
        std::vector<std::string> filter_keys;
 
        filter_keys.emplace_back(unit->type_name());
        filter_keys.emplace_back(!unit->name().empty() ? unit->name().str() : font::unicode_en_dash);
        filter_keys.emplace_back(std::to_string(unit->level()));
        filter_keys.emplace_back(unit_type::alignment_description(unit->alignment(), unit->gender()));
 
        if(const auto* race = unit->race()) {
            filter_keys.emplace_back(race->name(unit->gender()));
            filter_keys.emplace_back(race->plural_name());
        }
 
        for(const std::string& trait : unit->trait_names()) {
            filter_keys.emplace_back(trait);
        }
 
        return filter_keys;
    });
 
    dlg->on_modified([&recall_list, &rename](units_dialog*, std::size_t index) -> const auto& {
        const auto& unit = recall_list[index];
        rename.set_active(!unit->unrenamable());
        return *unit;
    });
 
    return dlg;
}
 
 
 
} // namespace dialogs