unit_recruit.cpp

Go to the documentation of this file.

/*
   Copyright (C) 2016 by the Battle for Wesnoth Project http://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/auxiliary/find_widget.hpp"
#include "gui/dialogs/helper.hpp"
#include "gui/dialogs/unit_recruit.hpp"
#include "gui/widgets/button.hpp"
#include "gui/widgets/image.hpp"
#include "gui/widgets/label.hpp"
#ifdef GUI2_EXPERIMENTAL_LISTBOX
#include "gui/widgets/list.hpp"
#else
#include "gui/widgets/listbox.hpp"
#endif
#include "gui/widgets/unit_preview_pane.hpp"
#include "gui/widgets/settings.hpp"
#include "gui/widgets/window.hpp"
#include "gettext.hpp"
#include "help/help.hpp"
#include "marked-up_text.hpp"
#include "play_controller.hpp"
#include "resources.hpp"
#include "team.hpp"
#include "units/types.hpp"
#include "whiteboard/manager.hpp"

#include "utils/functional.hpp"

namespace gui2
{

REGISTER_DIALOG(unit_recruit)

tunit_recruit::tunit_recruit(std::vector<const unit_type*>& recruit_list, team& team)
    : recruit_list_(recruit_list)
    , team_(team)
    , selected_index_(0)
{
}

static std::string can_afford_unit(const std::string& text, const bool can_afford)
{
    return can_afford ? text : "<span color='#ff0000'>" + text + "</span>";
}

void tunit_recruit::pre_show(twindow& window)
{
    tlistbox& list = find_widget<tlistbox>(&window, "recruit_list", false);

#ifdef GUI2_EXPERIMENTAL_LISTBOX
    connect_signal_notify_modified(*list,
        std::bind(&tunit_recruit::list_item_clicked,
        *this,
        std::ref(window)));
#else
    list.set_callback_value_change(
        dialog_callback<tunit_recruit, &tunit_recruit::list_item_clicked>);
#endif

    connect_signal_mouse_left_click(
        find_widget<tbutton>(&window, "show_help", false),
        std::bind(&tunit_recruit::show_help, this, std::ref(window)));

    for(const auto& recruit : recruit_list_)
    {
        std::map<std::string, string_map> row_data;
        string_map column;

        std::string image_string = recruit->image() + "~RC(" + recruit->flag_rgb() + ">"
            + team::get_side_color_index(team_.side()) + ")";

        int wb_gold = 0;
        if(resources::controller) {
            if(const boost::shared_ptr<wb::manager>& whiteb = resources::controller->get_whiteboard()) {
                wb::future_map future; // So gold takes into account planned spending
                wb_gold = whiteb->get_spent_gold_for(team_.side());
            }
        }

        const bool can_afford = recruit->cost() <= team_.gold() - wb_gold;

        const std::string cost_string = std::to_string(recruit->cost());

        column["label"] = image_string;
        row_data.insert(std::make_pair("unit_image", column));

        column["label"] = can_afford_unit(recruit->type_name(), can_afford);
        column["use_markup"] = "true";
        row_data.insert(std::make_pair("unit_type", column));

        column["label"] = can_afford_unit(cost_string, can_afford);
        column["use_markup"] = "true";
        row_data.insert(std::make_pair("unit_cost", column));

        list.add_row(row_data);
    }

    list_item_clicked(window);
}

void tunit_recruit::list_item_clicked(twindow& window)
{
    const int selected_row
        = find_widget<tlistbox>(&window, "recruit_list", false).get_selected_row();

    if(selected_row == -1) {
        return;
    }

    find_widget<tunit_preview_pane>(&window, "recruit_details", false)
        .set_displayed_type(recruit_list_[selected_row]);
}

void tunit_recruit::show_help(twindow& window)
{
    help::show_help(window.video(), "recruit_and_recall");
}

void tunit_recruit::post_show(twindow& window)
{
    if(get_retval() == twindow::OK) {
        selected_index_ = find_widget<tlistbox>(&window, "recruit_list", false)
            .get_selected_row();
    }
}

} // namespace gui2