1 --[[-----------------------------------------------------------------------------
3 Container that uses a tree control to switch between groups.
4 -------------------------------------------------------------------------------]]
5 local Type, Version = "TreeGroup", 34
6 local AceGUI = LibStub and LibStub("AceGUI-3.0", true)
7 if not AceGUI or (AceGUI:GetWidgetVersion(Type) or 0) >= Version then return end
10 local next, pairs, ipairs, assert, type = next, pairs, ipairs, assert, type
11 local math_min, math_max, floor = math.min, math.max, floor
12 local select, tremove, unpack, tconcat = select, table.remove, unpack, table.concat
15 local CreateFrame, UIParent = CreateFrame, UIParent
17 -- Global vars/functions that we don't upvalue since they might get hooked, or upgraded
18 -- List them here for Mikk's FindGlobals script
19 -- GLOBALS: GameTooltip, FONT_COLOR_CODE_CLOSE
21 -- Recycling functions
24 local pool = setmetatable({},{__mode='k'})
42 local DEFAULT_TREE_WIDTH = 175
43 local DEFAULT_TREE_SIZABLE = true
45 --[[-----------------------------------------------------------------------------
47 -------------------------------------------------------------------------------]]
48 local function GetButtonUniqueValue(line)
49 local parent = line.parent
50 if parent and parent.value then
51 return GetButtonUniqueValue(parent).."\001"..line.value
57 local function UpdateButton(button, treeline, selected, canExpand, isExpanded)
58 local self = button.obj
59 local toggle = button.toggle
60 local frame = self.frame
61 local text = treeline.text or ""
62 local icon = treeline.icon
63 local iconCoords = treeline.iconCoords
64 local level = treeline.level
65 local value = treeline.value
66 local uniquevalue = treeline.uniquevalue
67 local disabled = treeline.disabled
69 button.treeline = treeline
71 button.uniquevalue = uniquevalue
73 button:LockHighlight()
74 button.selected = true
76 button:UnlockHighlight()
77 button.selected = false
79 local normalTexture = button:GetNormalTexture()
80 local line = button.line
82 if ( level == 1 ) then
83 button:SetNormalFontObject("GameFontNormal")
84 button:SetHighlightFontObject("GameFontHighlight")
85 button.text:SetPoint("LEFT", (icon and 16 or 0) + 8, 2)
87 button:SetNormalFontObject("GameFontHighlightSmall")
88 button:SetHighlightFontObject("GameFontHighlightSmall")
89 button.text:SetPoint("LEFT", (icon and 16 or 0) + 8 * level, 2)
93 button:EnableMouse(false)
94 button.text:SetText("|cff808080"..text..FONT_COLOR_CODE_CLOSE)
96 button.text:SetText(text)
97 button:EnableMouse(true)
101 button.icon:SetTexture(icon)
102 button.icon:SetPoint("LEFT", 8 * level, (level == 1) and 0 or 1)
104 button.icon:SetTexture(nil)
108 button.icon:SetTexCoord(unpack(iconCoords))
110 button.icon:SetTexCoord(0, 1, 0, 1)
114 if not isExpanded then
115 toggle:SetNormalTexture("Interface\\Buttons\\UI-PlusButton-UP")
116 toggle:SetPushedTexture("Interface\\Buttons\\UI-PlusButton-DOWN")
118 toggle:SetNormalTexture("Interface\\Buttons\\UI-MinusButton-UP")
119 toggle:SetPushedTexture("Interface\\Buttons\\UI-MinusButton-DOWN")
127 local function ShouldDisplayLevel(tree)
129 for k, v in ipairs(tree) do
130 if v.children == nil and v.visible ~= false then
132 elseif v.children then
133 result = result or ShouldDisplayLevel(v.children)
135 if result then return result end
140 local function addLine(self, v, tree, level, parent)
145 line.iconCoords = v.iconCoords
146 line.disabled = v.disabled
150 line.visible = v.visible
151 line.uniquevalue = GetButtonUniqueValue(line)
153 line.hasChildren = true
155 line.hasChildren = nil
157 self.lines[#self.lines+1] = line
161 --fire an update after one frame to catch the treeframes height
162 local function FirstFrameUpdate(frame)
163 local self = frame.obj
164 frame:SetScript("OnUpdate", nil)
168 local function BuildUniqueValue(...)
169 local n = select('#', ...)
173 return (...).."\001"..BuildUniqueValue(select(2,...))
177 --[[-----------------------------------------------------------------------------
179 -------------------------------------------------------------------------------]]
180 local function Expand_OnClick(frame)
181 local button = frame.button
182 local self = button.obj
183 local status = (self.status or self.localstatus).groups
184 status[button.uniquevalue] = not status[button.uniquevalue]
188 local function Button_OnClick(frame)
189 local self = frame.obj
190 self:Fire("OnClick", frame.uniquevalue, frame.selected)
191 if not frame.selected then
192 self:SetSelected(frame.uniquevalue)
193 frame.selected = true
194 frame:LockHighlight()
200 local function Button_OnDoubleClick(button)
201 local self = button.obj
202 local status = self.status or self.localstatus
203 local status = (self.status or self.localstatus).groups
204 status[button.uniquevalue] = not status[button.uniquevalue]
208 local function Button_OnEnter(frame)
209 local self = frame.obj
210 self:Fire("OnButtonEnter", frame.uniquevalue, frame)
212 if self.enabletooltips then
213 GameTooltip:SetOwner(frame, "ANCHOR_NONE")
214 GameTooltip:SetPoint("LEFT",frame,"RIGHT")
215 GameTooltip:SetText(frame.text:GetText() or "", 1, .82, 0, 1)
221 local function Button_OnLeave(frame)
222 local self = frame.obj
223 self:Fire("OnButtonLeave", frame.uniquevalue, frame)
225 if self.enabletooltips then
230 local function OnScrollValueChanged(frame, value)
231 if frame.obj.noupdate then return end
232 local self = frame.obj
233 local status = self.status or self.localstatus
234 status.scrollvalue = value
239 local function Tree_OnSizeChanged(frame)
240 frame.obj:RefreshTree()
243 local function Tree_OnMouseWheel(frame, delta)
244 local self = frame.obj
245 if self.showscroll then
246 local scrollbar = self.scrollbar
247 local min, max = scrollbar:GetMinMaxValues()
248 local value = scrollbar:GetValue()
249 local newvalue = math_min(max,math_max(min,value - delta))
250 if value ~= newvalue then
251 scrollbar:SetValue(newvalue)
256 local function Dragger_OnLeave(frame)
257 frame:SetBackdropColor(1, 1, 1, 0)
260 local function Dragger_OnEnter(frame)
261 frame:SetBackdropColor(1, 1, 1, 0.8)
264 local function Dragger_OnMouseDown(frame)
265 local treeframe = frame:GetParent()
266 treeframe:StartSizing("RIGHT")
269 local function Dragger_OnMouseUp(frame)
270 local treeframe = frame:GetParent()
271 local self = treeframe.obj
272 local frame = treeframe:GetParent()
273 treeframe:StopMovingOrSizing()
274 --treeframe:SetScript("OnUpdate", nil)
275 treeframe:SetUserPlaced(false)
276 --Without this :GetHeight will get stuck on the current height, causing the tree contents to not resize
277 treeframe:SetHeight(0)
278 treeframe:SetPoint("TOPLEFT", frame, "TOPLEFT",0,0)
279 treeframe:SetPoint("BOTTOMLEFT", frame, "BOTTOMLEFT",0,0)
281 local status = self.status or self.localstatus
282 status.treewidth = treeframe:GetWidth()
284 treeframe.obj:Fire("OnTreeResize",treeframe:GetWidth())
285 -- recalculate the content width
286 treeframe.obj:OnWidthSet(status.fullwidth)
287 -- update the layout of the content
288 treeframe.obj:DoLayout()
291 --[[-----------------------------------------------------------------------------
293 -------------------------------------------------------------------------------]]
295 ["OnAcquire"] = function(self)
296 self:SetTreeWidth(DEFAULT_TREE_WIDTH, DEFAULT_TREE_SIZABLE)
297 self:EnableButtonTooltips(true)
300 ["OnRelease"] = function(self)
302 for k, v in pairs(self.localstatus) do
303 if k == "groups" then
304 for k2 in pairs(v) do
308 self.localstatus[k] = nil
311 self.localstatus.scrollvalue = 0
312 self.localstatus.treewidth = DEFAULT_TREE_WIDTH
313 self.localstatus.treesizable = DEFAULT_TREE_SIZABLE
316 ["EnableButtonTooltips"] = function(self, enable)
317 self.enabletooltips = enable
320 ["CreateButton"] = function(self)
321 local num = AceGUI:GetNextWidgetNum("TreeGroupButton")
322 local button = CreateFrame("Button", ("AceGUI30TreeButton%d"):format(num), self.treeframe, "OptionsListButtonTemplate")
325 local icon = button:CreateTexture(nil, "OVERLAY")
330 button:SetScript("OnClick",Button_OnClick)
331 button:SetScript("OnDoubleClick", Button_OnDoubleClick)
332 button:SetScript("OnEnter",Button_OnEnter)
333 button:SetScript("OnLeave",Button_OnLeave)
335 button.toggle.button = button
336 button.toggle:SetScript("OnClick",Expand_OnClick)
341 ["SetStatusTable"] = function(self, status)
342 assert(type(status) == "table")
344 if not status.groups then
347 if not status.scrollvalue then
348 status.scrollvalue = 0
350 if not status.treewidth then
351 status.treewidth = DEFAULT_TREE_WIDTH
353 if status.treesizable == nil then
354 status.treesizable = DEFAULT_TREE_SIZABLE
356 self:SetTreeWidth(status.treewidth,status.treesizable)
360 --sets the tree to be displayed
361 ["SetTree"] = function(self, tree, filter)
364 assert(type(tree) == "table")
370 ["BuildLevel"] = function(self, tree, level, parent)
371 local groups = (self.status or self.localstatus).groups
372 local hasChildren = self.hasChildren
374 for i, v in ipairs(tree) do
376 if not self.filter or ShouldDisplayLevel(v.children) then
377 local line = addLine(self, v, tree, level, parent)
378 if groups[line.uniquevalue] then
379 self:BuildLevel(v.children, level+1, line)
382 elseif v.visible ~= false or not self.filter then
383 addLine(self, v, tree, level, parent)
388 ["RefreshTree"] = function(self,scrollToSelection)
389 local buttons = self.buttons
390 local lines = self.lines
392 for i, v in ipairs(buttons) do
396 local t = tremove(lines)
403 if not self.tree then return end
404 --Build the list of visible entries from the tree and status tables
405 local status = self.status or self.localstatus
406 local groupstatus = status.groups
407 local tree = self.tree
409 local treeframe = self.treeframe
411 status.scrollToSelection = status.scrollToSelection or scrollToSelection -- needs to be cached in case the control hasn't been drawn yet (code bails out below)
413 self:BuildLevel(tree, 1)
415 local numlines = #lines
417 local maxlines = (floor(((self.treeframe:GetHeight()or 0) - 20 ) / 18))
418 if maxlines <= 0 then return end
422 scrollToSelection = status.scrollToSelection
423 status.scrollToSelection = nil
425 if numlines <= maxlines then
426 --the whole tree fits in the frame
427 status.scrollvalue = 0
428 self:ShowScroll(false)
429 first, last = 1, numlines
431 self:ShowScroll(true)
432 --scrolling will be needed
434 self.scrollbar:SetMinMaxValues(0, numlines - maxlines)
435 --check if we are scrolled down too far
436 if numlines - status.scrollvalue < maxlines then
437 status.scrollvalue = numlines - maxlines
440 first, last = status.scrollvalue+1, status.scrollvalue + maxlines
442 if scrollToSelection and status.selected then
444 for i,line in ipairs(lines) do -- find the line number
445 if line.uniquevalue==status.selected then
450 -- selection was deleted or something?
451 elseif show>=first and show<=last then
456 status.scrollvalue = show-1
458 status.scrollvalue = show-maxlines
460 first, last = status.scrollvalue+1, status.scrollvalue + maxlines
463 if self.scrollbar:GetValue() ~= status.scrollvalue then
464 self.scrollbar:SetValue(status.scrollvalue)
469 for i = first, last do
470 local line = lines[i]
471 local button = buttons[buttonnum]
473 button = self:CreateButton()
475 buttons[buttonnum] = button
476 button:SetParent(treeframe)
477 button:SetFrameLevel(treeframe:GetFrameLevel()+1)
478 button:ClearAllPoints()
479 if buttonnum == 1 then
480 if self.showscroll then
481 button:SetPoint("TOPRIGHT", -22, -10)
482 button:SetPoint("TOPLEFT", 0, -10)
484 button:SetPoint("TOPRIGHT", 0, -10)
485 button:SetPoint("TOPLEFT", 0, -10)
488 button:SetPoint("TOPRIGHT", buttons[buttonnum-1], "BOTTOMRIGHT",0,0)
489 button:SetPoint("TOPLEFT", buttons[buttonnum-1], "BOTTOMLEFT",0,0)
493 UpdateButton(button, line, status.selected == line.uniquevalue, line.hasChildren, groupstatus[line.uniquevalue] )
495 buttonnum = buttonnum + 1
500 ["SetSelected"] = function(self, value)
501 local status = self.status or self.localstatus
502 if status.selected ~= value then
503 status.selected = value
504 self:Fire("OnGroupSelected", value)
508 ["Select"] = function(self, uniquevalue, ...)
510 local status = self.status or self.localstatus
511 local groups = status.groups
514 groups[tconcat(path, "\001", 1, i)] = true
516 status.selected = uniquevalue
517 self:RefreshTree(true)
518 self:Fire("OnGroupSelected", uniquevalue)
521 ["SelectByPath"] = function(self, ...)
522 self:Select(BuildUniqueValue(...), ...)
525 ["SelectByValue"] = function(self, uniquevalue)
526 self:Select(uniquevalue, ("\001"):split(uniquevalue))
529 ["ShowScroll"] = function(self, show)
530 self.showscroll = show
532 self.scrollbar:Show()
533 if self.buttons[1] then
534 self.buttons[1]:SetPoint("TOPRIGHT", self.treeframe,"TOPRIGHT",-22,-10)
537 self.scrollbar:Hide()
538 if self.buttons[1] then
539 self.buttons[1]:SetPoint("TOPRIGHT", self.treeframe,"TOPRIGHT",0,-10)
544 ["OnWidthSet"] = function(self, width)
545 local content = self.content
546 local treeframe = self.treeframe
547 local status = self.status or self.localstatus
548 status.fullwidth = width
550 local contentwidth = width - status.treewidth - 20
551 if contentwidth < 0 then
554 content:SetWidth(contentwidth)
555 content.width = contentwidth
557 local maxtreewidth = math_min(400, width - 50)
559 if maxtreewidth > 100 and status.treewidth > maxtreewidth then
560 self:SetTreeWidth(maxtreewidth, status.treesizable)
562 treeframe:SetMaxResize(maxtreewidth, 1600)
565 ["OnHeightSet"] = function(self, height)
566 local content = self.content
567 local contentheight = height - 20
568 if contentheight < 0 then
571 content:SetHeight(contentheight)
572 content.height = contentheight
575 ["SetTreeWidth"] = function(self, treewidth, resizable)
576 if not resizable then
577 if type(treewidth) == 'number' then
579 elseif type(treewidth) == 'boolean' then
580 resizable = treewidth
581 treewidth = DEFAULT_TREE_WIDTH
584 treewidth = DEFAULT_TREE_WIDTH
587 self.treeframe:SetWidth(treewidth)
588 self.dragger:EnableMouse(resizable)
590 local status = self.status or self.localstatus
591 status.treewidth = treewidth
592 status.treesizable = resizable
594 -- recalculate the content width
595 if status.fullwidth then
596 self:OnWidthSet(status.fullwidth)
600 ["GetTreeWidth"] = function(self)
601 local status = self.status or self.localstatus
602 return status.treewidth or DEFAULT_TREE_WIDTH
605 ["LayoutFinished"] = function(self, width, height)
606 if self.noAutoHeight then return end
607 self:SetHeight((height or 0) + 20)
611 --[[-----------------------------------------------------------------------------
613 -------------------------------------------------------------------------------]]
614 local PaneBackdrop = {
615 bgFile = "Interface\\ChatFrame\\ChatFrameBackground",
616 edgeFile = "Interface\\Tooltips\\UI-Tooltip-Border",
617 tile = true, tileSize = 16, edgeSize = 16,
618 insets = { left = 3, right = 3, top = 5, bottom = 3 }
621 local DraggerBackdrop = {
622 bgFile = "Interface\\Tooltips\\UI-Tooltip-Background",
624 tile = true, tileSize = 16, edgeSize = 0,
625 insets = { left = 3, right = 3, top = 7, bottom = 7 }
628 local function Constructor()
629 local num = AceGUI:GetNextWidgetNum(Type)
630 local frame = CreateFrame("Frame", nil, UIParent)
632 local treeframe = CreateFrame("Frame", nil, frame)
633 treeframe:SetPoint("TOPLEFT")
634 treeframe:SetPoint("BOTTOMLEFT")
635 treeframe:SetWidth(DEFAULT_TREE_WIDTH)
636 treeframe:EnableMouseWheel(true)
637 treeframe:SetBackdrop(PaneBackdrop)
638 treeframe:SetBackdropColor(0.1, 0.1, 0.1, 0.5)
639 treeframe:SetBackdropBorderColor(0.4, 0.4, 0.4)
640 treeframe:SetResizable(true)
641 treeframe:SetMinResize(100, 1)
642 treeframe:SetMaxResize(400, 1600)
643 treeframe:SetScript("OnUpdate", FirstFrameUpdate)
644 treeframe:SetScript("OnSizeChanged", Tree_OnSizeChanged)
645 treeframe:SetScript("OnMouseWheel", Tree_OnMouseWheel)
647 local dragger = CreateFrame("Frame", nil, treeframe)
649 dragger:SetPoint("TOP", treeframe, "TOPRIGHT")
650 dragger:SetPoint("BOTTOM", treeframe, "BOTTOMRIGHT")
651 dragger:SetBackdrop(DraggerBackdrop)
652 dragger:SetBackdropColor(1, 1, 1, 0)
653 dragger:SetScript("OnEnter", Dragger_OnEnter)
654 dragger:SetScript("OnLeave", Dragger_OnLeave)
655 dragger:SetScript("OnMouseDown", Dragger_OnMouseDown)
656 dragger:SetScript("OnMouseUp", Dragger_OnMouseUp)
658 local scrollbar = CreateFrame("Slider", ("AceConfigDialogTreeGroup%dScrollBar"):format(num), treeframe, "UIPanelScrollBarTemplate")
659 scrollbar:SetScript("OnValueChanged", nil)
660 scrollbar:SetPoint("TOPRIGHT", -10, -26)
661 scrollbar:SetPoint("BOTTOMRIGHT", -10, 26)
662 scrollbar:SetMinMaxValues(0,0)
663 scrollbar:SetValueStep(1)
664 scrollbar:SetValue(0)
665 scrollbar:SetWidth(16)
666 scrollbar:SetScript("OnValueChanged", OnScrollValueChanged)
668 local scrollbg = scrollbar:CreateTexture(nil, "BACKGROUND")
669 scrollbg:SetAllPoints(scrollbar)
670 scrollbg:SetTexture(0,0,0,0.4)
672 local border = CreateFrame("Frame",nil,frame)
673 border:SetPoint("TOPLEFT", treeframe, "TOPRIGHT")
674 border:SetPoint("BOTTOMRIGHT")
675 border:SetBackdrop(PaneBackdrop)
676 border:SetBackdropColor(0.1, 0.1, 0.1, 0.5)
677 border:SetBackdropBorderColor(0.4, 0.4, 0.4)
680 local content = CreateFrame("Frame", nil, border)
681 content:SetPoint("TOPLEFT", 10, -10)
682 content:SetPoint("BOTTOMRIGHT", -10, 10)
690 localstatus = { groups = {}, scrollvalue = 0 },
692 treeframe = treeframe,
694 scrollbar = scrollbar,
699 for method, func in pairs(methods) do
700 widget[method] = func
702 treeframe.obj, dragger.obj, scrollbar.obj = widget, widget, widget
704 return AceGUI:RegisterAsContainer(widget)
707 AceGUI:RegisterWidgetType(Type, Constructor, Version)