units_dialog.cpp

Go to the documentation of this file.

/*
    Copyright (C) 2024
    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 "formatter.hpp"
#include "game_board.hpp"
#include "gettext.hpp"
#include "gui/dialogs/edit_text.hpp"
#include "gui/dialogs/message.hpp"
#include "gui/widgets/listbox.hpp"
#include "gui/widgets/button.hpp"
#include "gui/widgets/image.hpp"
#include "gui/widgets/label.hpp"
#include "gui/widgets/menu_button.hpp"
#include "gui/widgets/styled_widget.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;
}
 
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_variations_(false)
    , 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));
 
    connect_signal_mouse_left_click(
        find_widget<button>("show_help"),
        std::bind(&units_dialog::show_help, this));
 
    keyboard_capture(&filter);
    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<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;
        widget_item column;
        formatter filter_fmt;
        // generate tooltip for ith row
        if (tooltip_gen_) {
            column["tooltip"] = tooltip_gen_(i);
        }
 
        for (const auto& [id, gen] : column_generators_) {
            column["use_markup"] = "true";
            // generate label for ith row and column with 'id'
            column["label"] = gen(i);
            if (id != "unit_image") {
                filter_fmt << column["label"];
            }
            row_data.emplace(id, column);
        }
 
        filter_options_.push_back(filter_fmt.str());
        list.add_row(row_data);
    }
 
    const auto [sorter_id, order] = sort_last.value_or(sort_default);
    list.set_active_sorter(sorter_id, order, true);
}
 
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_);
        std::ostringstream filter_text;
        filter_text << selected_unit.type_name() << " " << name << " " << std::to_string(selected_unit.level());
        for(const std::string& trait : selected_unit.trait_names()) {
            filter_text << " " << trait;
        }
        filter_options_.insert(filter_options_.begin() + selected_index_, filter_text.str());
 
        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 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 is loyal and 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 dialog list
    list.remove_row(selected_index_);
    list_item_clicked();
 
    // Remove the entry from the filter list
    filter_options_.erase(filter_options_.begin() + selected_index_);
    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::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();
    }
 
    if(get_retval() == retval::OK) {
        selected_index_ = list.get_selected_row();
    }
}
 
void units_dialog::filter_text_changed(const std::string& text)
{
    auto& list = find_widget<listbox>("main_list");
    const std::size_t shown = list.filter_rows_by(
        [this, match = translation::make_ci_matcher(text)](std::size_t row) { return match(filter_options_[row]); });
 
    // Disable rename and dismiss buttons if no units are shown
    find_widget<button>("rename").set_active(shown > 0);
    find_widget<button>("dismiss").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 = [&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, &dlg, &types_list](std::size_t index) -> const auto& {
        const unit_type* ut = types_list[index];
 
        dlg->get_toggle().set_members_enabled(
            [ut](const unit_race::GENDER& gender) { return ut->has_gender_variation(gender); });
 
        populate_variations(*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,
    const team& team)
{
    auto dlg = std::make_unique<units_dialog>();
    auto set_column = dlg->make_column_builder(recruit_list);
 
    set_column("unit_image", [&team](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)";
        return image_string;
    }, sort_type::none);
 
    set_column("unit_details", [](const auto& recruit) {
        return recruit->type_name() + unit_helper::format_cost_string(recruit->cost());
    }, 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->on_modified([&recruit_list](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);
 
    // Rename functionality
    button& rename = dlg->find_widget<button>("rename");
    connect_signal_mouse_left_click(rename, std::bind([&]() {
        dlg->rename_unit(unit_list);
    }));
 
    auto set_column = dlg->make_column_builder(unit_list);
 
    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->on_modified([&unit_list, &rename](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 = 0;
    if(resources::controller && resources::controller->get_whiteboard()) {
        wb::future_map future; // So gold takes into account planned spending
        wb_gold = resources::controller->get_whiteboard()->get_spent_gold_for(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);
 
    // Rename functionality
    button& rename = dlg->find_widget<button>("rename");
    connect_signal_mouse_left_click(rename, std::bind([&]() {
        dlg->rename_unit(recall_list);
    }));
 
    // Dismiss functionality
    button& dismiss = dlg->find_widget<button>("dismiss");
    connect_signal_mouse_left_click(dismiss, std::bind([&]() {
        dlg->dismiss_unit(recall_list, team);
    }));
 
    auto set_column = dlg->make_column_builder(recall_list);
 
    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));
            details << unit_helper::format_cost_string(unit->recall_cost(), team.recall_cost());
            return details.str();
        },
        [](const auto& unit) {
            return unit->type_name().str();
        });
 
    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_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",
        [](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) {
            if(unit->can_advance()) {
                return markup::span_color(unit->xp_color(), unit->experience(), "/", unit->max_experience());
            } else {
                return markup::span_color(unit->xp_color(), font::unicode_en_dash);
            }
        },
        [](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_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_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->on_modified([&recall_list, &rename](std::size_t index) -> const auto& {
        const auto& unit = recall_list[index];
        rename.set_active(!unit->unrenamable());
        return *unit;
    });
 
    return dlg;
}
 
 
 
} // namespace dialogs