helper.cpp

Go to the documentation of this file.

/*
    Copyright (C) 2003 - 2025
    by David White <dave@whitevine.net>
    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.
*/
 
/**
 * @file
 * Support functions for dealing with units.
 */
 
#include "actions/create.hpp"
#include "formula/string_utils.hpp"
#include "resources.hpp"
#include "units/unit.hpp"
#include "units/helper.hpp"
#include "units/types.hpp"
#include "play_controller.hpp"
#include "serialization/markup.hpp"
#include "utils/general.hpp"
#include "whiteboard/manager.hpp"
 
namespace unit_helper {
 
int number_of_possible_advances(const unit &u)
{
    return u.advances_to().size() + u.get_modification_advances().size();
}
 
bool will_certainly_advance(const unit_map::iterator &u)
{
    if(!u.valid()) {
        return false;
    }
    return u->advances() && number_of_possible_advances(*u) > 0;
}
 
/**
 * Maps resistance <= -60 (resistance value <= -60%) to intense red.
 * Maps resistance >= 60 (resistance value >= 60%) to intense green.
 * Intermediate values are affinely mapped to the red-to-green scale,
 * with 0 (0%) being mapped to yellow.
 * Compare attack_info_percent_color() in reports.cpp.
 */
std::string resistance_color(const int resistance)
{
    // Passing false to select the more saturated red-to-green scale.
    return game_config::red_to_green(50.0 + resistance * 5.0 / 6.0, false).to_hex_string();
}
 
static std::string unit_level_tooltip(const int level, const std::vector<std::string> &adv_to_types, const std::vector<config> &adv_to_mods)
{
    std::ostringstream tooltip;
    tooltip << _("Level: ") << markup::bold(level) << "\n";
    const bool has_advancements = !adv_to_types.empty() || !adv_to_mods.empty();
    if(has_advancements) {
        tooltip << _("Advancements:") << "\n\t";
        if(!adv_to_types.empty())
            tooltip << markup::bold(utils::join(adv_to_types, "\n\t"));
        if(!adv_to_mods.empty()) {
            if(!adv_to_types.empty())
                tooltip << "\n\t";
            std::vector<std::string> descriptions;
            for(const config& adv : adv_to_mods)
                descriptions.push_back(adv["description"].str());
            tooltip << markup::bold(utils::join(descriptions, "\n\t"));
        }
    } else {
        tooltip << _("No advancement");
    }
    return tooltip.str();
}
 
std::string unit_level_tooltip(const unit &u)
{
    return unit_level_tooltip(u.level(), u.advances_to_translated(), u.get_modification_advances());
}
 
std::string unit_level_tooltip(const unit_type &type)
{
    const auto mod_adv_iters = type.modification_advancements();
    const std::vector<config> mod_advancements(mod_adv_iters.begin(), mod_adv_iters.end());
 
    return unit_level_tooltip(type.level(), type.advances_to(), mod_advancements);
}
 
std::string maybe_inactive(const std::string& str, bool active)
{
    return (active ? str : markup::span_color(font::INACTIVE_COLOR, str));
}
 
std::string format_cost_string(int unit_recall_cost, bool active)
{
    std::stringstream str;
    if (active) {
        str << markup::img("themes/gold.png") << ' ' << unit_recall_cost;
    } else {
        str << markup::img("themes/gold.png~GS()") << ' '
            << maybe_inactive(std::to_string(unit_recall_cost), false);
    }
    return str.str();
}
 
std::string format_cost_string(int unit_recall_cost, const int team_recall_cost)
{
    if(unit_recall_cost < 0) {
        unit_recall_cost = team_recall_cost;
    }
 
    std::stringstream str;
    str << markup::img("themes/gold.png") << ' ';
 
    if(unit_recall_cost > team_recall_cost) {
        str << markup::span_color(font::BAD_COLOR, unit_recall_cost);
    } else if(unit_recall_cost < team_recall_cost) {
        str << markup::span_color(font::GOOD_COLOR, unit_recall_cost);
    } else {
        // Default: show cost in white font color.
        // Should handle the unit cost = team cost case.
        str << unit_recall_cost;
    }
 
    return str.str();
}
 
std::string format_level_string(const int level, bool recallable)
{
    if(!recallable) {
        // Same logic as when recallable, but always in inactive_color.
        return markup::span_color(font::INACTIVE_COLOR,
            (level < 2 ? std::to_string(level) : markup::bold(level)));
    } else if(level < 1) {
        return markup::span_color(font::INACTIVE_COLOR, level);
    } else if(level == 1) {
        return std::to_string(level);
    } else if(level == 2) {
        return markup::bold(level);
    } else if(level == 3) {
        return markup::span_color(color_t(0xe2, 0xb7, 0x76), markup::bold(level));
    } else {
        return markup::span_color(color_t(0xdd, 0x66, 0x00), markup::bold(level));
    }
}
 
std::string format_movement_string(const int moves_left, const int moves_max, bool active)
{
    if (!active) {
        return markup::span_color(font::GRAY_COLOR, moves_left, "/", moves_max);
    } else if(moves_left == 0) {
        return markup::span_color(font::BAD_COLOR, moves_left, "/", moves_max);
    } else if(moves_left > moves_max) {
        return markup::span_color(font::YELLOW_COLOR, moves_left, "/", moves_max);
    } else {
        return markup::span_color(font::GOOD_COLOR, moves_left, "/", moves_max);
    }
}
 
// TODO: Return multiple strings here, in case more than one error applies? For
// example, if you start AOI S5 with 0GP and recruit a Mage, two reasons apply,
// leader not on keep (extrarecruit=Mage) and not enough gold.
t_string recruit_message(
    const std::string& type_id,
    map_location& target_hex,
    map_location& recruited_from,
    team& current_team)
{
    const unit_type* u_type = unit_types.find(type_id);
    if(u_type == nullptr) {
        return _("Internal error. Please report this as a bug! Details:\n")
            + "unit_helper::recruit_message: u_type == nullptr for " + type_id;
    }
 
    // search for the unit to be recruited in recruits
    if(!utils::contains(actions::get_recruits(current_team.side(), target_hex), type_id)) {
        return VGETTEXT("You cannot recruit a $unit_type_name at this time.",
                utils::string_map{{ "unit_type_name", u_type->type_name() }});
    }
 
    // TODO take a wb::future_map RAII as units_dialog does
    int wb_gold = 0;
    {
        wb::future_map future;
        wb_gold = (resources::controller->get_whiteboard()
            ? resources::controller->get_whiteboard()->get_spent_gold_for(current_team.side())
            : 0);
    }
    if(u_type->cost() > current_team.gold() - wb_gold)
    {
        if(wb_gold > 0)
            // TRANSLATORS: "plan" refers to Planning Mode
            return _("At this point in your plan, you will not have enough gold to recruit this unit.");
        else
            return _("You do not have enough gold to recruit this unit.");
    }
 
    current_team.last_recruit();
    const events::command_disabler disable_commands;
 
    {
        wb::future_map_if_active future; /* start planned unit map scope if in planning mode */
        std::string msg = actions::find_recruit_location(current_team.side(), target_hex, recruited_from, type_id);
        if(!msg.empty()) {
            return msg;
        }
    } // end planned unit map scope
 
    return {};
}
 
}