summaryrefslogtreecommitdiff
path: root/builtin/game/statbars.lua
blob: 1b83dd4ac6a66dea3d4af56211da72080dcedd1c (plain)
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
-- cache setting
local enable_damage = core.settings:get_bool("enable_damage")

local health_bar_definition =
{
	hud_elem_type = "statbar",
	position = { x=0.5, y=1 },
	text = "heart.png",
	number = core.PLAYER_MAX_HP_DEFAULT,
	direction = 0,
	size = { x=24, y=24 },
	offset = { x=(-10*24)-25, y=-(48+24+16)},
}

local breath_bar_definition =
{
	hud_elem_type = "statbar",
	position = { x=0.5, y=1 },
	text = "bubble.png",
	number = 20,
	direction = 0,
	size = { x=24, y=24 },
	offset = {x=25,y=-(48+24+16)},
}

local hud_ids = {}

local function initialize_builtin_statbars(player)

	if not player:is_player() then
		return
	end

	local name = player:get_player_name()

	if name == "" then
		return
	end

	if not hud_ids[name] then
		hud_ids[name] = {}
		-- flags are not transmitted to client on connect, we need to make sure
		-- our current flags are transmitted by sending them actively
		player:hud_set_flags(player:hud_get_flags())
	end
	local hud = hud_ids[name]

	if player:hud_get_flags().healthbar and enable_damage then
 		if hud.id_healthbar == nil then
 			local hp = player:get_hp()
 			local max_display_hp = math.max(core.PLAYER_MAX_HP_DEFAULT,
 				math.max(player:get_properties().hp_max, hp))
 			-- Limit width of health bar: Scale to the default maximal HP
			health_bar_definition.number =
				hp / max_display_hp * core.PLAYER_MAX_HP_DEFAULT
			hud.id_healthbar  = player:hud_add(health_bar_definition)
		end
	else
		if hud.id_healthbar ~= nil then
			player:hud_remove(hud.id_healthbar)
			hud.id_healthbar = nil
		end
	end

	if player:get_breath() < core.PLAYER_MAX_BREATH then
		if player:hud_get_flags().breathbar and enable_damage then
			if hud.id_breathbar == nil then
				hud.id_breathbar = player:hud_add(breath_bar_definition)
			end
		else
			if hud.id_breathbar ~= nil then
				player:hud_remove(hud.id_breathbar)
				hud.id_breathbar = nil
			end
		end
	elseif hud.id_breathbar ~= nil then
		player:hud_remove(hud.id_breathbar)
		hud.id_breathbar = nil
	end
end

local function cleanup_builtin_statbars(player)

	if not player:is_player() then
		return
	end

	local name = player:get_player_name()

	if name == "" then
		return
	end

	hud_ids[name] = nil
end

local function player_event_handler(player,eventname)
	assert(player:is_player())

	local name = player:get_player_name()

	if name == "" then
		return
	end

	if eventname == "health_changed" then
		initialize_builtin_statbars(player)

		if hud_ids[name].id_healthbar ~= nil then
 			local hp = player:get_hp()
 			local max_display_hp = math.max(core.PLAYER_MAX_HP_DEFAULT,
 				math.max(player:get_properties().hp_max, hp))
 			-- Limit width of health bar: Scale to the default maximal HP
			local hp_count = hp / max_display_hp * core.PLAYER_MAX_HP_DEFAULT
			player:hud_change(hud_ids[name].id_healthbar, "number", hp_count)
			return true
		end
	end

	if eventname == "breath_changed" then
		initialize_builtin_statbars(player)

		if hud_ids[name].id_breathbar ~= nil then
			player:hud_change(hud_ids[name].id_breathbar, "number", player:get_breath() * 2)
			return true
		end
	end

	if eventname == "hud_changed" then
		initialize_builtin_statbars(player)
		return true
	end

	return false
end

function core.hud_replace_builtin(name, definition)

	if definition == nil or
		type(definition) ~= "table" or
		definition.hud_elem_type ~= "statbar" then
		return false
	end

	if name == "health" then
		health_bar_definition = definition

		for name,ids in pairs(hud_ids) do
			local player = core.get_player_by_name(name)
			if  player and hud_ids[name].id_healthbar then
				player:hud_remove(hud_ids[name].id_healthbar)
				initialize_builtin_statbars(player)
			end
		end
		return true
	end

	if name == "breath" then
		breath_bar_definition = definition

		for name,ids in pairs(hud_ids) do
			local player = core.get_player_by_name(name)
			if  player and hud_ids[name].id_breathbar then
				player:hud_remove(hud_ids[name].id_breathbar)
				initialize_builtin_statbars(player)
			end
		end
		return true
	end

	return false
end

core.register_on_joinplayer(initialize_builtin_statbars)
core.register_on_leaveplayer(cleanup_builtin_statbars)
core.register_playerevent(player_event_handler)