DK/EntityTracker.cpp at master · halbich/DK · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#include "EntityTracker.hpp"
#include "Components.hpp"
#include "Helpers.hpp"

EntityTracker::EntityTracker()
	: curr_tracked_entity_{Component::NO_ENTITY}, entities_{nullptr}
{ /* DUMMY BODY */ }

void EntityTracker::set_tracked_entity(std::size_t id, EntitySystem& ents)
{
	curr_tracked_entity_ = id;
	auto view = window_->getChild("FRAME");
	view->getChild("UPGRADE")->setVisible(false);
	if(id == Component::NO_ENTITY)
		clear();
	else
	{
		view->getChild("ID_VALUE")->setText(std::to_string(id));
		auto health = ents.get_component<HealthComponent>(id);
		if(health)
			update_tracking("HP_VALUE", std::to_string(health->curr_hp) + " / " + std::to_string(health->max_hp));

		auto name = ents.get_component<NameComponent>(id);
		if(name)
			update_tracking("NAME_VALUE", name->name);

		// TODO: MANA

		auto upgrade = ents.get_component<UpgradeComponent>(id);
		if(upgrade)
		{
			update_tracking("EXP_VALUE", std::to_string(upgrade->experience) + " / " + std::to_string(upgrade->exp_needed));
			update_tracking("LVL_VALUE", std::to_string(upgrade->level) + " / " + std::to_string(upgrade->level_cap));
			if(upgrade->experience >= upgrade->exp_needed)
				view->getChild("UPGRADE")->setVisible(true);
		}

		auto faction = ents.get_component<FactionComponent>(id);
		if(faction)
			update_tracking("FACTION_VALUE", FactionHelper::get_faction_name(ents, id));

		auto gold = ents.get_component<GoldComponent>(id);
		if(gold)
			update_tracking("GOLD_VALUE", std::to_string(gold->curr_amount) + " / " + std::to_string(gold->max_amount));
	}
}

std::size_t EntityTracker::get_tracked_entity() const
{
	return curr_tracked_entity_;
}

void EntityTracker::update_tracking(const std::string& label, const std::string& value)
{
	window_->getChild("FRAME/" + label)->setText(value);
}

void EntityTracker::clear()
{
	auto view = window_->getChild("FRAME");
	view->getChild("ID_VALUE")->setText("NONE");
	view->getChild("NAME_VALUE")->setText("UNKNOWN");
	view->getChild("HP_VALUE")->setText("0 / 0");
	view->getChild("MANA_VALUE")->setText("0 / 0");
	view->getChild("EXP_VALUE")->setText("0 / 0");
	view->getChild("LVL_VALUE")->setText("0");
	view->getChild("FACTION_VALUE")->setText("NEUTRAL");
	view->getChild("GOLD_VALUE")->setText("0 / 0");
}

void EntityTracker::init_upgrade_butt(EntitySystem* ents)
{
	if(!window_)
		return;

	window_->getChild("FRAME/UPGRADE")->subscribeEvent(
		CEGUI::PushButton::EventClicked,
		[ents, this](const CEGUI::EventArgs&) -> bool {
			UpgradeHelper::upgrade(*ents, curr_tracked_entity_);
			return true;
		}
	);
}

void EntityTracker::init_()
{
	clear();
}