gui_units_window.hpp

Go to the documentation of this file.

#pragma once
 
#include "gui_element_types.hpp"
 
namespace ui {
 
template<typename T>
struct military_unit_info : public std::variant<T, dcon::province_land_construction_id, dcon::province_naval_construction_id> { };
 
template<typename T>
class military_unit_name_text : public simple_text_element_base {
public:
    void on_update(sys::state& state) noexcept override {
        auto content = retrieve<military_unit_info<T>>(state, parent);
 
        if(std::holds_alternative<dcon::province_land_construction_id>(content)) {
            auto c = std::get<dcon::province_land_construction_id>(content);
            auto utid = state.world.province_land_construction_get_type(c);
            auto unitname = utid ? state.military_definitions.unit_base_definitions[utid].name : dcon::text_key{};
            std::string res = text::produce_simple_string(state, unitname);
 
            set_text(state, res);
        }
        else if(std::holds_alternative<dcon::province_naval_construction_id>(content)) {
            auto c = std::get<dcon::province_naval_construction_id>(content);
            auto utid = state.world.province_naval_construction_get_type(c);
            auto unitname = utid ? state.military_definitions.unit_base_definitions[utid].name : dcon::text_key{};
            std::string res = text::produce_simple_string(state, unitname);
 
            set_text(state, res);
        }
        else if(std::holds_alternative<T>(content)) {
            auto fat_id = dcon::fatten(state.world, std::get<T>(content));
            auto unit_name = std::string{ state.to_string_view(fat_id.get_name()) };
            set_text(state, std::string{ state.to_string_view(fat_id.get_name()) });
        }
    }
};
 
template<typename T>
class military_unit_building_progress_bar : public progress_bar {
public:
    tooltip_behavior has_tooltip(sys::state& state) noexcept override {
        return tooltip_behavior::variable_tooltip;
    }
 
    void update_tooltip(sys::state& state, int32_t x, int32_t y, text::columnar_layout& contents) noexcept override {
        auto container = retrieve<military_unit_info<T>>(state, parent);
        float admin_eff = state.world.nation_get_administrative_efficiency(state.local_player_nation);
        float admin_cost_factor = 2.0f - admin_eff;
 
        if(std::holds_alternative<dcon::province_land_construction_id>(container)) {
            auto c = std::get<dcon::province_land_construction_id>(container);
 
            auto& goods = state.military_definitions.unit_base_definitions[state.world.province_land_construction_get_type(c)].build_cost;
            auto& cgoods = state.world.province_land_construction_get_purchased_goods(c);
 
            float total = 0.0f;
            float purchased = 0.0f;
 
            for(uint32_t i = 0; i < economy::commodity_set::set_size; ++i) {
                if(goods.commodity_type[i]) {
                    auto box = text::open_layout_box(contents, 0);
                    text::add_to_layout_box(state, contents, box, state.world.commodity_get_name(goods.commodity_type[i]));
                    text::add_to_layout_box(state, contents, box, std::string_view{ ": " });
                    text::add_to_layout_box(state, contents, box, text::fp_one_place{ cgoods.commodity_amounts[i] });
                    text::add_to_layout_box(state, contents, box, std::string_view{ " / " });
                    text::add_to_layout_box(state, contents, box, text::fp_one_place{ goods.commodity_amounts[i] * admin_cost_factor });
                    text::close_layout_box(contents, box);
                }
            }
        } else if(std::holds_alternative<dcon::province_naval_construction_id>(container)) {
            auto c = std::get<dcon::province_naval_construction_id>(container);
 
            auto& goods = state.military_definitions.unit_base_definitions[state.world.province_naval_construction_get_type(c)].build_cost;
            auto& cgoods = state.world.province_naval_construction_get_purchased_goods(c);
 
            float total = 0.0f;
            float purchased = 0.0f;
 
            for(uint32_t i = 0; i < economy::commodity_set::set_size; ++i) {
                if(goods.commodity_type[i]) {
                    auto box = text::open_layout_box(contents, 0);
                    text::add_to_layout_box(state, contents, box, state.world.commodity_get_name(goods.commodity_type[i]));
                    text::add_to_layout_box(state, contents, box, std::string_view{ ": " });
                    text::add_to_layout_box(state, contents, box, text::fp_one_place{ cgoods.commodity_amounts[i] });
                    text::add_to_layout_box(state, contents, box, std::string_view{ " / " });
                    text::add_to_layout_box(state, contents, box, text::fp_one_place{ goods.commodity_amounts[i] * admin_cost_factor });
                    text::close_layout_box(contents, box);
                }
            }
        }
 
    }
};
 
class military_unit_morale_progress_bar : public vertical_progress_bar {
public:
    tooltip_behavior has_tooltip(sys::state& state) noexcept override {
        return tooltip_behavior::variable_tooltip;
    }
 
    void update_tooltip(sys::state& state, int32_t x, int32_t y, text::columnar_layout& contents) noexcept override {
        auto box = text::open_layout_box(contents, 0);
        text::localised_single_sub_box(state, contents, box, std::string_view("military_morale_tooltip"), text::variable_type::value,
                uint8_t(progress * 100));
        text::close_layout_box(contents, box);
    }
};
 
class military_unit_strength_progress_bar : public vertical_progress_bar {
public:
    tooltip_behavior has_tooltip(sys::state& state) noexcept override {
        return tooltip_behavior::variable_tooltip;
    }
 
    void update_tooltip(sys::state& state, int32_t x, int32_t y, text::columnar_layout& contents) noexcept override {
        auto box = text::open_layout_box(contents, 0);
        // This should change to use "military_shipstrength_tooltip" instead for naval units
        text::localised_single_sub_box(state, contents, box, std::string_view("military_strength_tooltip2"),
                text::variable_type::percent, uint8_t(progress * 100));
        text::close_layout_box(contents, box);
    }
};
 
template<typename T>
class leader_in_army_img : public image_element_base {
    bool visible = true;
    dcon::gfx_object_id default_img;
 
    void on_update(sys::state& state) noexcept override {
        if(!default_img)
            default_img = base_data.data.image.gfx_object;
 
        auto container = retrieve<military_unit_info<T>>(state, parent);
        if(!std::holds_alternative<T>(container)) {
            visible = false;
            return;
        }
 
        auto unit = std::get<T>(container);
        dcon::leader_id lid;
        if constexpr(std::is_same_v<T, dcon::army_id>) {
            lid = state.world.army_get_general_from_army_leadership(unit);
        } else {
            lid = state.world.navy_get_admiral_from_navy_leadership(unit);
        }
 
        auto pculture = state.world.nation_get_primary_culture(state.local_player_nation);
        auto ltype = pculture.get_group_from_culture_group_membership().get_leader();
 
        if(ltype && lid) {
            auto admiral = state.world.leader_get_is_admiral(lid);
            if(admiral) {
                auto arange = ltype.get_admirals();
                if(arange.size() > 0) {
                    auto rval = rng::get_random(state, uint32_t(state.world.leader_get_since(lid).value), uint32_t(lid.value));
                    auto in_range = rng::reduce(uint32_t(rval), arange.size());
                    base_data.data.image.gfx_object = arange[in_range];
                }
            } else {
                auto grange = ltype.get_generals();
                if(grange.size() > 0) {
                    auto rval = rng::get_random(state, uint32_t(state.world.leader_get_since(lid).value), uint32_t(lid.value));
                    auto in_range = rng::reduce(uint32_t(rval), grange.size());
                    base_data.data.image.gfx_object = grange[in_range];
                }
            }
        } else {
            base_data.data.image.gfx_object = default_img;
        }
 
    }
 
    void render(sys::state& state, int32_t x, int32_t y) noexcept override {
        if(visible)
            image_element_base::render(state, x, y);
    }
 
    tooltip_behavior has_tooltip(sys::state& state) noexcept override {
        return visible ? tooltip_behavior::variable_tooltip : tooltip_behavior::no_tooltip;
    }
 
    void update_tooltip(sys::state& state, int32_t x, int32_t y, text::columnar_layout& contents) noexcept override {
        auto container = retrieve<military_unit_info<T>>(state, parent);
        if(!std::holds_alternative<T>(container)) {
            return;
        }
 
        auto unit = std::get<T>(container);
        dcon::leader_id lid;
        if constexpr(std::is_same_v<T, dcon::army_id>) {
            lid = state.world.army_get_general_from_army_leadership(unit);
        } else {
            lid = state.world.navy_get_admiral_from_navy_leadership(unit);
        }
        if(lid)
            display_leader_full(state, lid, contents, 0);
    }
};
 
template<typename T>
class cancel_unit_construction_button { };
 
template<>
class cancel_unit_construction_button<dcon::army_id> : public button_element_base {
public:
    void on_update(sys::state& state) noexcept override {
        auto info = retrieve<military_unit_info<dcon::army_id>>(state, parent);
        if(std::holds_alternative<dcon::province_land_construction_id>(info)) {
            auto pc = std::get<dcon::province_land_construction_id>(info);
            auto pop = state.world.province_land_construction_get_pop(pc);
            disabled = !command::can_cancel_land_unit_construction(state, state.local_player_nation, state.world.pop_get_province_from_pop_location(pop), state.world.pop_get_culture(pop), state.world.province_land_construction_get_type(pc));
        }
    }
    void button_action(sys::state& state) noexcept override {
        auto info = retrieve<military_unit_info<dcon::army_id>>(state, parent);
        if(std::holds_alternative<dcon::province_land_construction_id>(info)) {
            auto pc = std::get<dcon::province_land_construction_id>(info);
            auto pop = state.world.province_land_construction_get_pop(pc);
            command::cancel_land_unit_construction(state, state.local_player_nation, state.world.pop_get_province_from_pop_location(pop), state.world.pop_get_culture(pop), state.world.province_land_construction_get_type(pc));
        }
    }
};
 
template<>
class cancel_unit_construction_button<dcon::navy_id> : public button_element_base {
public:
    void on_update(sys::state& state) noexcept override {
        auto info = retrieve<military_unit_info<dcon::navy_id>>(state, parent);
        if(std::holds_alternative<dcon::province_naval_construction_id>(info)) {
            auto pc = std::get<dcon::province_naval_construction_id>(info);
            auto loc = state.world.province_naval_construction_get_province(pc);
            auto type = state.world.province_naval_construction_get_type(pc);
            disabled = !command::can_cancel_naval_unit_construction(state, state.local_player_nation, loc, type);
        }
    }
    void button_action(sys::state& state) noexcept override {
        auto info = retrieve<military_unit_info<dcon::navy_id>>(state, parent);
        if(std::holds_alternative<dcon::province_naval_construction_id>(info)) {
            auto pc = std::get<dcon::province_naval_construction_id>(info);
            auto loc = state.world.province_naval_construction_get_province(pc);
            auto type = state.world.province_naval_construction_get_type(pc);
            command::cancel_naval_unit_construction(state, state.local_player_nation, loc, type);
        }
    }
};
 
template<typename T>
class mil_goto_background_button : public button_element_base {
public:
    void on_update(sys::state& state) noexcept override {
        auto container = retrieve<military_unit_info<T>>(state, parent);
        interactable = std::holds_alternative<T>(container);
    }
    void render(sys::state& state, int32_t x, int32_t y) noexcept override {
        button_element_base::render(state, x, y);
    }
    void button_action(sys::state& state) noexcept override {
        auto container = retrieve<military_unit_info<T>>(state, parent);
        if(std::holds_alternative<T>(container)) {
            T id = std::get<T>(container);
            state.selected_armies.clear();
            state.selected_navies.clear();
 
            if constexpr(std::is_same_v<T, dcon::army_id>) {
                state.map_state.center_map_on_province(state, state.world.army_get_location_from_army_location(id));
            } else {
                state.map_state.center_map_on_province(state, state.world.navy_get_location_from_navy_location(id));
            }
            state.ui_state.military_subwindow->set_visible(state, false);
            state.select(id);
        }
    }
};
 
template<typename T>
class military_unit_entry : public listbox_row_element_base<military_unit_info<T>> {
    simple_text_element_base* unit_name = nullptr;
    image_element_base* unit_icon = nullptr;
    image_element_base* leader_icon = nullptr;
    button_element_base* cancel_button = nullptr;
    simple_text_element_base* eta_date_text = nullptr;
    simple_text_element_base* location_text = nullptr;
    military_unit_building_progress_bar<T>* unit_building_progress = nullptr;
    simple_text_element_base* unit_regiments_text = nullptr;
    simple_text_element_base* unit_men_text = nullptr;
    military_unit_morale_progress_bar* unit_morale_progress = nullptr;
    military_unit_strength_progress_bar* unit_strength_progress = nullptr;
    image_element_base* unit_moving_icon = nullptr;
    image_element_base* unit_digin_icon = nullptr;
    image_element_base* unit_combat_icon = nullptr;
public:
    std::unique_ptr<element_base> make_child(sys::state& state, std::string_view name, dcon::gui_def_id id) noexcept override {
        auto name2 = name;
        if(name == "military_unit_entry_bg") {
            return make_element_by_type<mil_goto_background_button<T>>(state, id);
        } else if(name == "unit_progress") {
            auto ptr = make_element_by_type<military_unit_building_progress_bar<T>>(state, id);
            unit_building_progress = ptr.get();
            return ptr;
        } else if(name == "leader") {
            auto ptr = make_element_by_type<leader_in_army_img<T>>(state, id);
            leader_icon = ptr.get();
            return ptr;
        } else if(name == "unit_strip") {
            auto ptr = make_element_by_type<image_element_base>(state, id);
            unit_icon = ptr.get();
            return ptr;
        } else if(name == "name") {
            auto ptr = make_element_by_type<military_unit_name_text<T>>(state, id);
            unit_name = ptr.get();
            return ptr;
        } else if(name == "location") {
            auto ptr = make_element_by_type<generic_name_text<dcon::province_id>>(state, id);
            location_text = ptr.get();
            return ptr;
        }
        else if(name == "unit_eta") {
            auto ptr = make_element_by_type<simple_text_element_base>(state, id);
            eta_date_text = ptr.get();
            return ptr;
        } else if(name == "military_cancel_unit") {
            auto ptr = make_element_by_type<cancel_unit_construction_button<T>>(state, id);
            cancel_button = ptr.get();
            return ptr;
        } else if(name == "regiments") {
            auto ptr = make_element_by_type<simple_text_element_base>(state, id);
            unit_regiments_text = ptr.get();
            return ptr;
        } else if(name == "men") {
            auto ptr = make_element_by_type<simple_text_element_base>(state, id);
            unit_men_text = ptr.get();
            return ptr;
        } else if(name == "morale_progress") {
            auto ptr = make_element_by_type<military_unit_morale_progress_bar>(state, id);
            unit_morale_progress = ptr.get();
            return ptr;
        } else if(name == "strength_progress") {
            auto ptr = make_element_by_type<military_unit_strength_progress_bar>(state, id);
            unit_strength_progress = ptr.get();
            return ptr;
        } else if(name == "moving") {
            auto ptr = make_element_by_type<image_element_base>(state, id);
            unit_moving_icon = ptr.get();
            return ptr;
        } else if(name == "digin") {
            auto ptr = make_element_by_type<image_element_base>(state, id);
            unit_digin_icon = ptr.get();
            return ptr;
        } else if(name == "combat") {
            auto ptr = make_element_by_type<image_element_base>(state, id);
            unit_combat_icon = ptr.get();
            return ptr;
        } else {
            return nullptr;
        }
    }
 
    void on_update(sys::state& state) noexcept override {
        auto const& content = listbox_row_element_base<military_unit_info<T>>::content;
 
        bool is_building = !std::holds_alternative<T>(content);
        bool is_moving = false;
        bool is_digin = false;
        bool is_combat = false;
 
        if(is_building) {
            if(std::holds_alternative<dcon::province_land_construction_id>(content)) {
                auto c = std::get<dcon::province_land_construction_id>(content);
                if(unit_icon) unit_icon->frame = state.military_definitions.unit_base_definitions[state.world.province_land_construction_get_type(c)].icon - 1;
                if(unit_building_progress) unit_building_progress->progress = economy::unit_construction_progress(state, c);
            } else if(std::holds_alternative<dcon::province_naval_construction_id>(content)) {
                auto c = std::get<dcon::province_naval_construction_id>(content);
                if(unit_icon) unit_icon->frame = state.military_definitions.unit_base_definitions[state.world.province_naval_construction_get_type(c)].icon - 1;
                if(unit_building_progress) unit_building_progress->progress = economy::unit_construction_progress(state, c);
            }
        } else {
            auto regiments = 0;
            auto strength = 0.f;
            auto full_strength = 0.f;
            // Armies
            if constexpr(std::is_same_v<T, dcon::army_id>) {
                state.world.army_for_each_army_membership_as_army(std::get<dcon::army_id>(content), [&](dcon::army_membership_id amid) {
                    auto rid = state.world.army_membership_get_regiment(amid);
                    full_strength += 1.f * state.defines.pop_size_per_regiment;
                    strength += state.world.regiment_get_strength(rid) * state.defines.pop_size_per_regiment;
                    ++regiments;
                });
            }
            // Navies
            if constexpr(std::is_same_v<T, dcon::navy_id>) {
                state.world.navy_for_each_navy_membership_as_navy(std::get<dcon::navy_id>(content), [&](dcon::navy_membership_id nmid) {
                    auto sid = state.world.navy_membership_get_ship(nmid);
                    full_strength += 1.f;
                    strength += state.world.ship_get_strength(sid);
                    ++regiments;
                });
            }
            if(unit_strength_progress) unit_strength_progress->progress = (full_strength != 0.0f) ? strength / full_strength : 0.f;
            if(unit_men_text) unit_men_text->set_text(state, text::prettify(int32_t(strength)));
            if(unit_regiments_text) unit_regiments_text->set_text(state, std::to_string(regiments));
        }
 
        if(unit_icon) unit_icon->set_visible(state, is_building);
        if(cancel_button) cancel_button->set_visible(state, is_building);
        if(eta_date_text) eta_date_text->set_visible(state, is_building);
        if(location_text) location_text->set_visible(state, is_building);
 
        if(unit_building_progress) unit_building_progress->set_visible(state, is_building);
 
        if(unit_name) unit_name->set_visible(state, true);
        if(leader_icon) leader_icon->set_visible(state, !is_building);
        if(unit_regiments_text) unit_regiments_text->set_visible(state, !is_building);
        if(unit_men_text) unit_men_text->set_visible(state, !is_building);
        if(unit_morale_progress) unit_morale_progress->set_visible(state, !is_building);
        if(unit_strength_progress) unit_strength_progress->set_visible(state, !is_building);
        if(unit_moving_icon) unit_moving_icon->set_visible(state, !is_building && is_moving);
        if(unit_digin_icon) unit_digin_icon->set_visible(state, !is_building && is_digin);
        if(unit_combat_icon) unit_combat_icon->set_visible(state, !is_building && is_combat);
    }
 
 
    message_result get(sys::state& state, Cyto::Any& payload) noexcept override {
        auto const& content = listbox_row_element_base<military_unit_info<T>>::content;
        if(payload.holds_type<dcon::province_id>()) {
            dcon::province_id p{};
            if(std::holds_alternative<dcon::province_land_construction_id>(content)) {
                auto c = std::get<dcon::province_land_construction_id>(content);
                p = state.world.pop_location_get_province(
                        state.world.pop_get_pop_location_as_pop(state.world.province_land_construction_get_pop(c)));
            } else if(std::holds_alternative<dcon::province_naval_construction_id>(content)) {
                auto c = std::get<dcon::province_naval_construction_id>(content);
                p = state.world.province_naval_construction_get_province(c);
            }
            payload.emplace<dcon::province_id>(p);
            return message_result::consumed;
        }
        return listbox_row_element_base<military_unit_info<T>>::get(state, payload);
    }
};
 
template<typename T>
class military_units_listbox : public listbox_element_base<military_unit_entry<T>, military_unit_info<T>> {
protected:
    std::string_view get_row_element_name() override {
        return "unit_entry";
    }
 
public:
    void on_update(sys::state& state) noexcept override {
        auto& row_contents = listbox_element_base<military_unit_entry<T>, military_unit_info<T>>::row_contents;
        row_contents.clear();
        if(listbox_element_base<military_unit_entry<T>, military_unit_info<T>>::parent) {
            Cyto::Any payload = dcon::nation_id{};
            listbox_element_base<military_unit_entry<T>, military_unit_info<T>>::parent->impl_get(state, payload);
            dcon::nation_id n = Cyto::any_cast<dcon::nation_id>(payload);
            // Armies
            if constexpr(std::is_same_v<T, dcon::army_id>) {
                state.world.nation_for_each_army_control_as_controller(n, [&](dcon::army_control_id acid) {
                    auto aid = state.world.army_control_get_army(acid);
                    row_contents.push_back(military_unit_info<T>{aid});
                });
                state.world.nation_for_each_province_land_construction_as_nation(n,
                        [&](dcon::province_land_construction_id p) { row_contents.push_back(military_unit_info<T>{p}); });
            }
            // Navies
            if constexpr(std::is_same_v<T, dcon::navy_id>) {
                state.world.nation_for_each_navy_control_as_controller(n, [&](dcon::navy_control_id ncid) {
                    auto nid = state.world.navy_control_get_navy(ncid);
                    row_contents.push_back(military_unit_info<T>{nid});
                });
                state.world.nation_for_each_province_naval_construction_as_nation(n,
                        [&](dcon::province_naval_construction_id p) { row_contents.push_back(military_unit_info<T>{p}); });
            }
        }
        listbox_element_base<military_unit_entry<T>, military_unit_info<T>>::update(state);
    }
};
 
template<class T>
class build_unit_button : public button_element_base {
public:
    bool disarmed = false;
    bool no_possible_units = false;
    void button_action(sys::state& state) noexcept override {
        state.ui_state.unit_window_army->set_visible(state, false);
        state.ui_state.unit_window_navy->set_visible(state, false);
        state.ui_state.root->move_child_to_front(state.ui_state.build_unit_window);
        Cyto::Any payload = T{};
        state.ui_state.build_unit_window->impl_set(state, payload);
        state.ui_state.build_unit_window->set_visible(state, true);
    }
    tooltip_behavior has_tooltip(sys::state& state) noexcept override {
        return tooltip_behavior::variable_tooltip;
    }
    void update_tooltip(sys::state& state, int32_t x, int32_t y, text::columnar_layout& contents) noexcept override {
        auto box = text::open_layout_box(contents, 0);
        if(disarmed) {
            text::localised_format_box(state, contents, box, std::string_view("cantbuild_forcedisarm"));
        } else if(no_possible_units) {
            text::localised_format_box(state, contents, box, std::string_view("alice_no_possible_units"));
        } else {
            if constexpr(std::is_same_v<T, dcon::army_id>) {
                text::localised_format_box(state, contents, box, std::string_view("military_build_army_tooltip"));
            } else if constexpr(std::is_same_v<T, dcon::navy_id>) {
                text::localised_format_box(state, contents, box, std::string_view("military_build_navy_tooltip"));
            }
        }
        text::close_layout_box(contents, box);
    }
    void on_update(sys::state& state) noexcept override {
        if(state.world.nation_get_disarmed_until(state.local_player_nation) && state.current_date < state.world.nation_get_disarmed_until(state.local_player_nation)) {
            disabled = true;
            disarmed = true;
        } else {
            dcon::unit_type_id utid = dcon::unit_type_id{ 0 };
            disarmed = false;
            uint32_t count = 0;
            uint8_t unit_def_count = 0;
            for(auto const& ubd : state.military_definitions.unit_base_definitions) {
                if(state.military_definitions.unit_base_definitions[dcon::unit_type_id{ unit_def_count }].is_land) {
                    if(!state.military_definitions.unit_base_definitions[dcon::unit_type_id{ unit_def_count }].primary_culture) {
                        if(state.world.nation_get_active_unit(state.local_player_nation, dcon::unit_type_id{ unit_def_count }) || state.military_definitions.unit_base_definitions[dcon::unit_type_id{ unit_def_count }].active) {
                            break;
                        }
                    }
                }
                unit_def_count++;
            }
            utid = dcon::unit_type_id{ unit_def_count };
            if constexpr(std::is_same_v<T, dcon::army_id>) {
                for(auto ucon : state.world.nation_get_province_land_construction(state.local_player_nation)) {
                    count++;
                    if(count) {
                        disabled = false;
                        no_possible_units = false;
                        return;
                    }
                }
                for(auto po : state.world.nation_get_province_ownership_as_nation(state.local_player_nation)) {
                    auto p = po.get_province();
                    state.world.for_each_culture([&](dcon::culture_id c) {
                        if(command::can_start_land_unit_construction(state, state.local_player_nation, p, c, utid)) {
                            count++;
                        }
                    });
                    if(count) {
                        disabled = false;
                        no_possible_units = false;
                        return;
                    }
                }
                disabled = true;
                no_possible_units = true;
            } else {
                disarmed = false;
                utid = dcon::unit_type_id{ 0 };
                count = 0;
                unit_def_count = 0;
                for(auto const& ubd : state.military_definitions.unit_base_definitions) {
                    if(!state.military_definitions.unit_base_definitions[dcon::unit_type_id{ unit_def_count }].is_land) {
                        if(state.world.nation_get_active_unit(state.local_player_nation, dcon::unit_type_id{ unit_def_count }) || state.military_definitions.unit_base_definitions[dcon::unit_type_id{ unit_def_count }].active) {
                            utid = dcon::unit_type_id{ unit_def_count };
                            for(auto po : state.world.nation_get_province_ownership_as_nation(state.local_player_nation)) {
                                auto p = po.get_province();
                                if(command::can_start_naval_unit_construction(state, state.local_player_nation, p, utid)) {
                                    disabled = false;
                                    no_possible_units = false;
                                    return;
                                }
                            }
                        }
                    }
                    unit_def_count++;
                }
                disabled = true;
                no_possible_units = true;
            }
        }
    }
};
 
class military_armies_text : public simple_text_element_base {
public:
    tooltip_behavior has_tooltip(sys::state& state) noexcept override {
        return tooltip_behavior::variable_tooltip;
    }
    void on_update(sys::state& state) noexcept override {
        auto rng = state.world.nation_get_army_control(state.local_player_nation);
        set_text(state, text::prettify(int64_t(rng.end() - rng.begin())));
    }
 
    void update_tooltip(sys::state& state, int32_t x, int32_t y, text::columnar_layout& contents) noexcept override {
        auto box = text::open_layout_box(contents, 0);
        auto rng = state.world.nation_get_army_control(state.local_player_nation);
        text::localised_single_sub_box(state, contents, box, std::string_view("military_army_count_tooltip"),
                text::variable_type::value, int64_t(rng.end() - rng.begin()));
        text::close_layout_box(contents, box);
    }
};
 
class military_navies_text : public simple_text_element_base {
public:
    tooltip_behavior has_tooltip(sys::state& state) noexcept override {
        return tooltip_behavior::variable_tooltip;
    }
    void on_update(sys::state& state) noexcept override {
        auto rng = state.world.nation_get_navy_control(state.local_player_nation);
        set_text(state, text::prettify(int64_t(rng.end() - rng.begin())));
    }
    void update_tooltip(sys::state& state, int32_t x, int32_t y, text::columnar_layout& contents) noexcept override {
        auto box = text::open_layout_box(contents, 0);
        auto rng = state.world.nation_get_navy_control(state.local_player_nation);
        text::localised_single_sub_box(state, contents, box, std::string_view("military_navy_count_tooltip"),
                text::variable_type::value, int64_t(rng.end() - rng.begin()));
        text::close_layout_box(contents, box);
    }
};
 
class military_armies_construction_text : public simple_text_element_base {
public:
    void on_update(sys::state& state) noexcept override {
        auto cstr_range = state.world.nation_get_province_land_construction(state.local_player_nation);
        set_text(state, std::to_string(cstr_range.end() - cstr_range.begin()));
    }
};
 
class military_navies_construction_text : public simple_text_element_base {
public:
    void on_update(sys::state& state) noexcept override {
        auto cstr_range = state.world.nation_get_province_naval_construction(state.local_player_nation);
        set_text(state, std::to_string(cstr_range.end() - cstr_range.begin()));
    }
};
 
class military_units_sortby_name : public button_element_base {
public:
    tooltip_behavior has_tooltip(sys::state& state) noexcept override {
        return tooltip_behavior::variable_tooltip;
    }
 
    void update_tooltip(sys::state& state, int32_t x, int32_t y, text::columnar_layout& contents) noexcept override {
        auto box = text::open_layout_box(contents, 0);
        text::localised_format_box(state, contents, box, std::string_view("military_sort_by_name_tooltip"));
        text::close_layout_box(contents, box);
    }
};
 
class military_units_sortby_strength : public button_element_base {
public:
    tooltip_behavior has_tooltip(sys::state& state) noexcept override {
        return tooltip_behavior::variable_tooltip;
    }
 
    void update_tooltip(sys::state& state, int32_t x, int32_t y, text::columnar_layout& contents) noexcept override {
        auto box = text::open_layout_box(contents, 0);
        text::localised_format_box(state, contents, box, std::string_view("military_sort_by_strength_tooltip"));
        text::close_layout_box(contents, box);
    }
};
 
template<class T>
class military_units_window : public window_element_base {
private:
    image_element_base* cdts_icon = nullptr;
public:
    std::unique_ptr<element_base> make_child(sys::state& state, std::string_view name, dcon::gui_def_id id) noexcept override {
        if(name == "current_count") {
            if constexpr(std::is_same_v<T, dcon::army_id>) {
                return make_element_by_type<military_armies_text>(state, id);
            } else {
                return make_element_by_type<military_navies_text>(state, id);
            }
        } else if(name == "under_construction") {
            if constexpr(std::is_same_v<T, dcon::army_id>) {
                return make_element_by_type<military_armies_construction_text>(state, id);
            } else {
                return make_element_by_type<military_navies_construction_text>(state, id);
            }
        } else if(name == "cut_down_to_size") {
            auto ptr = make_element_by_type<image_element_base>(state, id);
            ptr->set_visible(state, false);
            cdts_icon = ptr.get();
            return ptr;
        } else if(name == "sort_name") {
            return make_element_by_type<military_units_sortby_name>(state, id);
        } else if(name == "sort_strength") {
            return make_element_by_type<military_units_sortby_strength>(state, id);
        } else if(name == "build_new") {
            auto ptr = make_element_by_type<build_unit_button<T>>(state, id);
            if constexpr(std::is_same_v<T, dcon::army_id>) {
                ptr->set_button_text(state, text::produce_simple_string(state, "military_build_army_label"));
            } else {
                ptr->set_button_text(state, text::produce_simple_string(state, "military_build_navy_label"));
            }
            return ptr;
        } else if(name == "unit_listbox") {
            return make_element_by_type<military_units_listbox<T>>(state, id);
        } else {
            return nullptr;
        }
    }
};
 
} // namespace ui