ControlHost.lua

-- Path of Building
--
-- Class: Control Host
-- Host for UI controls
--
local t_insert = table.insert
local t_sort = table.sort
local ControlHostClass = newClass("ControlHost", function(self)
	self.controls = { }
end)

function ControlHostClass:SelectControl(newSelControl)
	if self.selControl == newSelControl then
		return
	end
	if self.selControl then
		if self.selControl.selDragActive and self.selControl.dragTargetList then
			return
		end
		self.selControl:SetFocus(false)
	end
	self.selControl = newSelControl
	if self.selControl then
		self.selControl:SetFocus(true)
	end
end

function ControlHostClass:GetMouseOverControl()
	for _, control in pairs(self.controls) do
		if control.IsMouseOver and control:IsMouseOver() then
			return control
		end
	end
end

function ControlHostClass:ProcessControlsInput(inputEvents, viewPort)
	for id, event in ipairs(inputEvents) do
		if event.type == "KeyDown" then
			if self.selControl then
				self:SelectControl(self.selControl:OnKeyDown(event.key, event.doubleClick))
				inputEvents[id] = nil
			end
			if not self.selControl and event.key:match("BUTTON") then
				self:SelectControl()
				if isMouseInRegion(viewPort) then
					local mOverControl = self:GetMouseOverControl()
					if mOverControl and mOverControl.OnKeyDown then
						self:SelectControl(mOverControl:OnKeyDown(event.key, event.doubleClick))
						inputEvents[id] = nil
					end
				end
			end
		elseif event.type == "KeyUp" then
			local selControl = self.selControl

			if selControl then
				if selControl.OnKeyUp then
					self:SelectControl(selControl:OnKeyUp(event.key))
				end
				
				inputEvents[id] = nil
			end

			local mOverControl = self:GetMouseOverControl(viewPort)

			-- Avoid calculating isMouseInRegion as much as possible as it's expensive
			if mOverControl and (not selControl or mOverControl.OnHoverKeyUp) then
				if isMouseInRegion(viewPort) then
					if not selControl and mOverControl.OnKeyUp and mOverControl:OnKeyUp(event.key) then
						inputEvents[id] = nil
					end
	
					if mOverControl.OnHoverKeyUp then
						mOverControl:OnHoverKeyUp(event.key)
					end
				end
			end
		elseif event.type == "Char" then
			if self.selControl then
				if self.selControl.OnChar then
					self:SelectControl(self.selControl:OnChar(event.key))
				end
				inputEvents[id] = nil
			end
		end
	end	
end

function ControlHostClass:DrawControls(viewPort, selControl)
	local negZOrder = nil
	local posZOrder = nil
	for _, control in pairs(self.controls) do
		if control.zOrder and control:IsShown() and control.Draw then
			if control.zOrder < 0 then
				negZOrder = negZOrder or { }
				t_insert(negZOrder, control)
			elseif control.zOrder > 0 then
				posZOrder = posZOrder or { }
				t_insert(posZOrder, control)
			end
		end
	end
	if negZOrder then
		t_sort(negZOrder, function(a, b)
			return a.zOrder < b.zOrder
		end)
		for _, control in ipairs(negZOrder) do
			control:Draw(viewPort, (self.selControl and self.selControl.hasFocus and self.selControl ~= control) or (selControl and selControl.hasFocus and selControl ~= control))
		end
	end
	for _, control in pairs(self.controls) do
		if (not control.zOrder or control.zOrder == 0) and control:IsShown() and control.Draw then
			control:Draw(viewPort, (self.selControl and self.selControl.hasFocus and self.selControl ~= control) or (selControl and selControl.hasFocus and selControl ~= control))
		end
	end
	if posZOrder then
		t_sort(posZOrder, function(a, b)
			return a.zOrder < b.zOrder
		end)
		for _, control in ipairs(posZOrder) do
			control:Draw(viewPort, (self.selControl and self.selControl.hasFocus and self.selControl ~= control) or (selControl and selControl.hasFocus and selControl ~= control))
		end
	end
end