-- ToME - Tales of Maj'Eyal
-- Copyright (C) 2009, 2010, 2011, 2012 Nicolas Casalini
--
-- 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 3 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; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program.  If not, see <http://www.gnu.org/licenses/>.
--
-- Nicolas Casalini "DarkGod"
-- darkgod@te4.org

require "engine.class"
local Base = require "engine.ui.Base"
local Focusable = require "engine.ui.Focusable"
local Slider = require "engine.ui.Slider"

--- A talent trees display
module(..., package.seeall, class.inherit(Base, Focusable))

function _M:init(t)
	self.tiles = assert(t.tiles, "no Tiles class")
	self.tree = t.tree or {}
	self.w = assert(t.width, "no width")
	self.h = assert(t.height, "no height")
	self.tooltip = assert(t.tooltip, "no tooltip")
	self.on_use = assert(t.on_use, "no on_use")
	self.on_expand = t.on_expand
	self.scrollbar = t.scrollbar
	self.no_cross = t.no_cross
	self.dont_select_top = t.dont_select_top

	self.icon_size = 48
	self.frame_size = 50
	self.icon_offset = 1
	self.frame_offset = 5

	self.shadow = 0.7

	self.frame_sel = self:makeFrame("ui/selector-sel", self.frame_size, self.frame_size)
	self.frame_usel = self:makeFrame("ui/selector", self.frame_size, self.frame_size)
	self.talent_frame = self:makeFrame("ui/icon-frame/frame", self.frame_size, self.frame_size)
	self.plus = _M:getUITexture("ui/plus.png")
	self.minus = _M:getUITexture("ui/minus.png")

	Base.init(self, t)
end

function _M:onUse(item, inc)
	self.last_scroll = nil
	self.on_use(item, inc)
end

function _M:onExpand(item, inc)
	self.last_scroll = nil
	item.shown = not item.shown
	if self.on_expand then self.on_expand(item) end
end

function _M:updateTooltip()
	if not self.last_mz then
		game.tooltip_x = nil
		return
	end
	local mz = self.last_mz
	local str = self.tooltip(mz.item)
	game:tooltipDisplayAtMap(mz.tx or (self.last_display_x + mz.x2), mz.ty or (self.last_display_y + mz.y1), str)
end

function _M:doScroll(v)
	self.scroll = util.bound(self.scroll + v, 1, self.max_display)
end

function _M:moveSel(i, j)
	local match = nil

	if i == 0 then
		local t = self.tree[self.sel_i]
		if t.nodes then
			self.sel_j = util.bound(self.sel_j + j, 1, #t.nodes)
			match = t.nodes[self.sel_j]
		end
	elseif i == 1 then
		local t = self.tree[self.sel_i]
		if t.shown and self.sel_j == 0 and t.nodes and #t.nodes > 0 then
			self.sel_j = 1
			match = t.nodes[1]
		else
			self.sel_i = util.bound(self.sel_i + i, 1, #self.tree)
			self.sel_j = 0
			local t = self.tree[self.sel_i]
			match = t
		end
	elseif i == -1 then
		local t = self.tree[self.sel_i]
		if t.shown and self.sel_j > 0 and t.nodes and #t.nodes > 0 then
			self.sel_j = 0
			match = t
		else
			self.sel_i = util.bound(self.sel_i + i, 1, #self.tree)
			local t = self.tree[self.sel_i]
			if t.shown and t.nodes and #t.nodes > 0 then
				self.sel_j = 1
				match = t.nodes[1]
			else
				self.sel_j = 0
				match = t
			end
		end
	end

	self.last_scroll = nil
	self.scroll = util.scroll(self.sel_i, self.scroll, self.max_display)
	self:display(self.last_display_bx, self.last_display_by, 0, self.last_display_x, self.last_display_y)

	for i = 1, #self.mousezones do
		local mz = self.mousezones[i]
		if mz.item == match then self.last_mz = mz break end
	end

	self:display(self.last_display_bx, self.last_display_by, 0, self.last_display_x, self.last_display_y)

	if not self.last_mz then return end
	local str, fx, fy = self.tooltip(self.last_mz.item)
	self.last_mz.tx, self.last_mz.ty = fx or (self.last_display_x + self.last_mz.x2), fy or (self.last_display_y + self.last_mz.y1)
	game:tooltipDisplayAtMap(self.last_mz.tx, self.last_mz.ty, str)
end

function _M:generate()
	self.mouse:reset()
	self.key:reset()

	self.scroll = 1

	-- Draw the scrollbar
	if self.scrollbar then
		self.scrollbar = Slider.new{size=self.h, max=#self.tree}
	end

	self.sel_i = 1
	self.sel_j = 1

	self.mousezones = {}

	self:redrawAllItems()

	-- Add UI controls
	self.mouse:registerZone(0, 0, self.w, self.h, function(button, x, y, xrel, yrel, bx, by, event)
		if event == "button" and button == "wheelup" then self:doScroll(-1)
		elseif event == "button" and button == "wheeldown" then self:doScroll(1)
		end

		local done = false
		for i = 1, #self.mousezones do
			local mz = self.mousezones[i]
			if x >= mz.x1 and x <= mz.x2 and y >= mz.y1 and y <= mz.y2 then
				if not self.last_mz or mz.item ~= self.last_mz.item then
					local str, fx, fy = self.tooltip(mz.item)
					mz.tx, mz.ty = fx or (self.last_display_x + mz.x2), fy or (self.last_display_y + mz.y1)
					game:tooltipDisplayAtMap(mz.tx, mz.ty, str)
				end

				if event == "button" and (button == "left" or button == "right") then
					if mz.item.type then
						if x - mz.x1 >= self.plus.w then self:onUse(mz.item, button == "left")
						else self:onExpand(mz.item, button == "left") end
					else
						self:onUse(mz.item, button == "left")
					end
				end

				self.last_mz = mz
				self.sel_i = mz.i
				self.sel_j = mz.j
				done = true
				break
			end
		end
		if not done then game.tooltip_x = nil self.last_mz = nil end
	end)
	self.key:addBinds{
		ACCEPT = function() self:onUse(self.last_mz.item, true) end,
		MOVE_UP = function() self:moveSel(-1, 0) end,
		MOVE_DOWN = function() self:moveSel(1, 0) end,
		MOVE_LEFT = function() self:moveSel(0, -1) end,
		MOVE_RIGHT = function() self:moveSel(0, 1) end,
	}
	self.key:addCommands{
		[{"_RETURN","ctrl"}] = function() self:onUse(self.last_mz.item, false) end,
		[{"_UP","ctrl"}] = function() self.key:triggerVirtual("MOVE_UP") end,
		[{"_DOWN","ctrl"}] = function() self.key:triggerVirtual("MOVE_DOWN") end,
		[{"_LEFT","ctrl"}] = function() self.key:triggerVirtual("MOVE_LEFT") end,
		[{"_RIGHT","ctrl"}] = function() self.key:triggerVirtual("MOVE_RIGHT") end,
		_HOME = function() self.sel_i = 1 self:moveSel(-1, 0) end,
		_END = function() self.sel_i = #self.tree self:moveSel(1, 0)  end,
		_PAGEUP = function() self:doScroll(-self.max_display) end,
		_PAGEDOWN = function() self:doScroll(self.max_display) end,
	}
end

function _M:drawItem(item)
	if item.stat then
		local str = item:status():toString()
		local d = self.font:draw(str, self.font:size(str), 255, 255, 255, true)[1]
		item.text_status = d
	elseif item.type_stat then
		local str = item.name:toString()
		self.font:setStyle("bold")
		local d = self.font:draw(str, self.font:size(str), 255, 255, 255, true)[1]
		self.font:setStyle("normal")
		item.text_status = d
	elseif item.talent then
		local str = item:status():toString()
		local d = self.font:draw(str, self.font:size(str), 255, 255, 255, true)[1]
		item.text_status = d
	elseif item.type then
		local str = item:rawname():toString()
		local c = item:color()
		self.font:setStyle("bold")
		local d = self.font:draw(str, self.font:size(str), c[1], c[2], c[3], true)[1]
		self.font:setStyle("normal")
		item.text_status = d
	end
end

function _M:redrawAllItems()
	for i = 1, #self.tree do
		local tree = self.tree[i]
		self:drawItem(tree)
		for j = 1, #tree.nodes do
			local tal = tree.nodes[j]
			self:drawItem(tal)
		end
	end
end

function _M:display(x, y, nb_keyframes, screen_x, screen_y)
	self.last_display_bx = x
	self.last_display_by = y
	self.last_display_x = screen_x
	self.last_display_y = screen_y

	local mz = {}
	if self.last_scroll ~= self.scroll then self.mousezones = mz end

	local dx, dy = 0, 0

	if self.last_mz then
		self:drawFrame(self.focused and self.frame_sel or self.frame_usel, x+self.last_mz.x1-2, y+self.last_mz.y1-2, 1, 1, 1, 1, self.last_mz.x2-self.last_mz.x1+4, self.last_mz.y2-self.last_mz.y1+4)
	end

	self.max_display = 1
	for i = self.scroll, #self.tree do
		local tree = self.tree[i]

		if tree.text_status then
			local key = tree.text_status
			local cross = not tree.shown and self.plus or self.minus

			mz[#mz+1] = {i=i,j=1, item=tree, x1=dx, y1=dy, x2=dx+cross.w + 3 + key.w, y2=dy+cross.h}

			if not self.no_cross then
				cross.t:toScreenFull(dx+x, dy+y + (-cross.h + key.h) / 2, cross.w, cross.h, cross.tw, cross.th)
				dx = dx + cross.w + 3
			end

			if self.shadow then key._tex:toScreenFull(dx+x + 2, dy+y + 2, key.w, key.h, key._tex_w, key._tex_h, 0, 0, 0, self.shadow) end
			key._tex:toScreenFull(dx+x, dy+y, key.w, key.h, key._tex_w, key._tex_h)
			dy = dy + key.h + 4
		end

		local addh = 0
		if tree.shown then for j = 1, #tree.nodes do
			local tal = tree.nodes[j]

			tal.entity:toScreen(self.tiles, dx+x + self.icon_offset, dy+y + self.icon_offset, self.icon_size, self.icon_size)

			local rgb = tal:color()
			self:drawFrame(self.talent_frame, dx+x, dy+y, rgb[1]/255, rgb[2]/255, rgb[3]/255, 1)

			if tal.text_status then
				local key = tal.text_status
				if self.shadow then key._tex:toScreenFull(dx+x + (self.frame_size - key.w)/2 + 2, dy+y + self.frame_size + 4, key.w, key.h, key._tex_w, key._tex_h, 0, 0, 0, self.shadow) end
				key._tex:toScreenFull(dx+x + (self.frame_size - key.w)/2, dy+y + self.frame_size + 2, key.w, key.h, key._tex_w, key._tex_h)
				addh = key.h
			end

			mz[#mz+1] = {i=i, j=j, item=tal, x1=dx, y1=dy, x2=dx+self.frame_size, y2=dy+self.frame_size+addh}

			dx = dx + self.frame_size + self.frame_offset
			addh = addh + self.frame_size
		end end
		self.max_display = i - self.scroll + 1
		dx = 0
		dy = dy + addh + 12
		if dy + self.frame_size >= self.h then break end
	end

	if self.focused and self.scrollbar then
		self.scrollbar.pos = self.scroll
		self.scrollbar:display(x + self.w - self.scrollbar.w, y)
	end

	self.last_scroll = self.scroll
end