From 842e0efe360417556ad0e757d91ef05449bfc8ee Mon Sep 17 00:00:00 2001 From: jussi Date: Wed, 20 Mar 2024 19:18:01 +0200 Subject: Raygui tree view. --- examples/raygui_extensions/tree_view.lua | 414 +++++++++++++++++++++++++++++++ 1 file changed, 414 insertions(+) create mode 100644 examples/raygui_extensions/tree_view.lua (limited to 'examples/raygui_extensions/tree_view.lua') diff --git a/examples/raygui_extensions/tree_view.lua b/examples/raygui_extensions/tree_view.lua new file mode 100644 index 0000000..05e5ac5 --- /dev/null +++ b/examples/raygui_extensions/tree_view.lua @@ -0,0 +1,414 @@ +local TreeView = {} +TreeView.__index = TreeView + +TreeView.SINGLE_SELECT = 0 +TreeView.MULTI_SELECT = 1 +TreeView.RANGE_SELECT = 2 + +TreeView.MOVE_ITEM_NONE = 0 +TreeView.MOVE_ITEM_IN = 1 +TreeView.MOVE_ITEM_UP = 2 +TreeView.MOVE_ITEM_DOWN = 3 + +function TreeView:new( bounds, text, callback, grabCallback, dragCallback, styles, tooltip ) + local object = setmetatable( {}, self ) + object._gui = nil + + object.padding = 4 -- Content edges. + object.spacing = 4 -- Between controls. + object.indentation = 14 -- Indentation for one depth level. + + object.bounds = bounds:clone() + object.text = text + object.scroll = Vec2:new() + object.view = Rect:new() + object.callback = callback + object.grabCallback = grabCallback + object.dragCallback = dragCallback + object.styles = styles + object.tooltip = tooltip + + object.gui = Raygui:new() -- Contains full independent gui system. + object.controls = {} + + -- Set in setSize. + object.framebufferSize = nil + object.framebuffer = nil + object.defaultControlSize = nil + + object.visible = true + object.disabled = false + object.draggable = true + + object.mouseScale = 1 -- Set this if drawing in different size to render texture for example. + object.selectedItems = {} + + object:setSize( Vec2:new( object.bounds.width, object.bounds.height ) ) + + object._forceCheckScroll = false + object._posY = 0 -- In control list update. + object._curDepth = 0 -- Current indentation. + object._curId = 0 -- Running number to give id's for controls. + object._idRange = { 1, 1 } + object._lastActiveItem = nil + object._clickedItem = nil + object._movingItem = object.MOVE_ITEM_NONE + + object:updateMouseOffset() + + return object +end + +function TreeView:getDefaultBounds() + return Rect:new( self.padding, self.padding, self.defaultControlSize.x, self.defaultControlSize.y ) +end + +local function getControlBounds( control ) + return control.viewBounds or control.focusBounds or control.bounds +end + +function TreeView:updateControl( control ) + control.bounds = self:getDefaultBounds() + control._depth = self._curDepth + control._indentation = self.indentation + control._id = self._curId + self._curId = self._curId + 1 + + if control.visible then + control:setPosition( Vec2:new( control.bounds.x, self._posY ) ) + local bounds = getControlBounds( control ) + + bounds.x = bounds.x + self._curDepth * self.indentation + bounds.width = bounds.width - self._curDepth * self.indentation + + if not control._noYAdvance then + self._posY = self._posY + bounds.height + self.spacing + end + self.content = self.content:fit( bounds ) + end + + if 0 < #control.controls then + self._curDepth = self._curDepth + 1 + + for i, groupControl in ipairs( control.controls ) do + groupControl.visible = control.open + groupControl._parent = control + groupControl._childId = i + + if not control.open and 0 < #control.controls then + groupControl.open = false + groupControl.active = false + end + + self:updateControl( groupControl ) + end + + self._curDepth = self._curDepth - 1 + end +end + +function TreeView:updateContent() + self._posY = self.padding + self._curId = 1 + + self.content.width = 0 + self.content.height = 0 + + for i, control in ipairs( self.controls ) do + control._parent = nil -- Sets parent in updateControl if any. + control._childId = i + self:updateControl( control ) + end + self.content.x = 0 + self.content.y = 0 + self.content.height = self.content.height + self.padding + self.content.width = self.content.width + self.padding + self._forceCheckScroll = true +end + +function TreeView:addItem( name, group ) + local control = self.gui:TreeItem( + self:getDefaultBounds(), + name, + { -- Callbacks. + open = function( this ) self:updateContent() end, + toggle = function( this ) self:itemSelect( this ) end, + }, + { -- Styles. + properties = { + { RL.TOGGLE, RL.TEXT_ALIGNMENT, RL.TEXT_ALIGN_LEFT }, + { RL.TOGGLE, RL.BASE_COLOR_NORMAL, RL.ColorToInt( RL.BLANK ) }, + { RL.TOGGLE, RL.BORDER_COLOR_NORMAL, RL.ColorToInt( RL.BLANK ) }, + } + } + ) + if group ~= nil then + table.insert( group.controls, control ) + else + table.insert( self.controls, control ) + end + + self:updateContent() + + return control +end + +function TreeView:checkItem( controls, item, mode ) + for _, control in ipairs( controls ) do + if mode == self.SINGLE_SELECT and control ~= item then + control.active = false + end + if 0 < #control.controls then + self:checkItem( control.controls, item, mode ) + end + if mode == self.RANGE_SELECT then + if self._idRange[1] <= control._id and control._id <= self._idRange[2] then + control.active = true + end + end + if control.active then + table.insert( self.selectedItems, control ) + -- table.insert( self.selectedItems, 1, control ) + end + end +end + +-- Check if trying to move parent to child. +function TreeView:isChild( item, newParent ) + local to = newParent + + while to ~= nil do + to = to._parent + + if item == to then + return true + end + end + + return false +end + +function TreeView:itemSelect( item ) + -- Item move. + + local moveItems = {} + + if self._movingItem ~= self.MOVE_ITEM_NONE then + + for i = #self.selectedItems, 1, -1 do + local moveItem = self.selectedItems[i] + -- print( moveItem.text, "moveItem._id", moveItem._id, item.text, "item._id", item._id ) + + if moveItem ~= item and not self:isChild( moveItem, item ) then + local parentControls = self.controls + + if moveItem._parent ~= nil then + parentControls = moveItem._parent.controls + end + + local pos = #self.selectedItems - i + 1 + + if moveItem._parent == item._parent and moveItem._childId < item._childId then + pos = 1 + end + + table.insert( moveItems, pos, table.remove( parentControls, moveItem._childId ) ) + end + + moveItem.active = false + end + + for _, moveItem in ipairs( moveItems ) do + local newParentControls = self.controls + + if item._parent ~= nil then + newParentControls = item._parent.controls + end + + local offset = 0 + + if moveItem._parent == item._parent and moveItem._childId < item._childId then + offset = -1 + end + + if self._movingItem == self.MOVE_ITEM_IN then + table.insert( item.controls, 1, moveItem ) + elseif self._movingItem == self.MOVE_ITEM_UP then + local pos = RL.Clamp( item._childId + offset, 1, #newParentControls + 1 ) + table.insert( newParentControls, pos, moveItem ) + elseif self._movingItem == self.MOVE_ITEM_DOWN then + local pos = RL.Clamp( item._childId + offset + 1, 1, #newParentControls + 1 ) + table.insert( newParentControls, pos, moveItem ) + end + end + + item.active = false + self._movingItem = self.MOVE_ITEM_NONE + self:updateContent() + + return + end + + -- Item select. + + local mode = self.SINGLE_SELECT + + if RL.IsKeyDown( RL.KEY_LEFT_CONTROL ) or RL.IsKeyDown( RL.KEY_RIGHT_CONTROL ) then + mode = self.MULTI_SELECT + elseif RL.IsKeyDown( RL.KEY_LEFT_SHIFT ) or RL.IsKeyDown( RL.KEY_RIGHT_SHIFT ) then + mode = self.RANGE_SELECT + end + + if self._lastActiveItem ~= nil then + self._idRange = { math.min( self._lastActiveItem._id, item._id ), math.max( self._lastActiveItem._id, item._id ) } + end + self.selectedItems = {} + self:checkItem( self.controls, item, mode ) + + if mode ~= self.RANGE_SELECT then + self._lastActiveItem = item -- Old clicked. + end + + if self.callback ~= nil then + self.callback( self.selectedItems ) + end +end + +function TreeView:update() + local mousePos = Vec2:new( RL.GetMousePosition() ) + local guiMousePos = mousePos + self.gui.mouseOffset + local mouseInView = self.view:checkCollisionPoint( mousePos ) + + if not mouseInView then + self.gui.locked = true + else + self.gui.locked = false + end + + self.gui:update() + + local mouseInClickedItem = false + + if RL.IsMouseButtonPressed( RL.MOUSE_BUTTON_LEFT ) then + if 0 < self.gui.focused and mouseInView and self.gui.controls[ self.gui.focused ].active then + self._clickedItem = self.gui.controls[ self.gui.focused ] + else + self._clickedItem = nil + end + end + + if self._clickedItem ~= nil then + mouseInClickedItem = self._clickedItem.bounds:checkCollisionPoint( guiMousePos ) + end + + RL.BeginTextureMode( self.framebuffer ) + RL.ClearBackground( RL.BLANK ) + RL.rlTranslatef( { self.scroll.x, self.scroll.y, 0 } ) + + self.gui:draw() + + if RL.IsMouseButtonDown( RL.MOUSE_BUTTON_LEFT ) and self._clickedItem ~= nil + and not mouseInClickedItem and 0 < self.gui.focused then + local focusBounds = self.gui.controls[ self.gui.focused ].bounds + local relY = ( guiMousePos.y - focusBounds.y ) / focusBounds.height + + if relY <= 0.35 then + RL.DrawLine( + { focusBounds.x, focusBounds.y }, + { focusBounds.x + focusBounds.width, focusBounds.y }, + 6, RL.RED + ) + self._movingItem = self.MOVE_ITEM_UP + elseif 0.65 <= relY then + RL.DrawLine( + { focusBounds.x, focusBounds.y + focusBounds.height }, + { focusBounds.x + focusBounds.width, focusBounds.y + focusBounds.height }, + 6, RL.RED + ) + self._movingItem = self.MOVE_ITEM_DOWN + else + RL.DrawRectangleLines( focusBounds, RL.RED ) + self._movingItem = self.MOVE_ITEM_IN + end + elseif RL.IsMouseButtonReleased( RL.MOUSE_BUTTON_LEFT ) then + self._clickedItem = nil + self._movingItem = self.MOVE_ITEM_NONE + end + RL.EndTextureMode() + + return self._gui:drag( self ) +end + +function TreeView:draw() + local oldScroll = self.scroll:clone() + local _, scroll, view = RL.GuiScrollPanel( self.bounds, self.text, self.content, self.scroll, self.view ) + self.view:set( view ) + self.scroll:set( scroll ) + + if self.scroll ~= oldScroll or self._forceCheckScroll then + if not self._forceCheckScroll then + self._gui:checkScrolling() + end + self._forceCheckScroll = false + + self:updateMouseOffset() + self.gui.view:set( -self.scroll.x, -self.scroll.y, self.view.width, self.view.height ) + end + + RL.DrawTexturePro( + RL.GetRenderTextureTexture( self.framebuffer ), + { 0, self.framebufferSize.y - self.view.height, self.view.width, -self.view.height }, + { math.floor( self.view.x ), math.floor( self.view.y ), self.view.width, self.view.height }, + { 0, 0 }, + 0.0, + RL.WHITE + ) +end + +function TreeView:updateMouseOffset() + self.gui.mouseOffset = Vec2:new( -self.view.x - self.scroll.x, -self.view.y - self.scroll.y ):scale( self.mouseScale ) +end + +function TreeView:setPosition( pos ) + self.bounds.x = pos.x + self.bounds.y = pos.y + + self:updateMouseOffset() +end + +function TreeView:setSize( size ) + self.bounds.width = size.x + self.bounds.height = size.y + + local scrollBarWidth = RL.GuiGetStyle( RL.LISTVIEW, RL.SCROLLBAR_WIDTH ) + local borderWidth = RL.GuiGetStyle( RL.DEFAULT, RL.BORDER_WIDTH ) + + self.content = Rect:new( + 0, + self.gui.RAYGUI_WINDOWBOX_STATUSBAR_HEIGHT, + self.bounds.width - scrollBarWidth - self.padding * 2 - borderWidth * 2, + self.bounds.height - scrollBarWidth - self.padding * 2 - borderWidth * 2 + ) + self.defaultControlSize = Vec2:new( self.content.width, 22 ) + + local _, _, view = RL.GuiScrollPanel( self.bounds, self.text, self.content, self.scroll, self.view ) + self.view = Rect:new( view ) + + self.gui.view = Rect:new( 0, 0, self.view.width, self.view.height ) + self.framebufferSize = Vec2:new( self.bounds.width, self.bounds.height - self.gui.RAYGUI_WINDOWBOX_STATUSBAR_HEIGHT ) + + if self.framebuffer ~= nil and not RL.IsGCUnloadEnabled() then + RL.UnloadRenderTexture( self.framebuffer ) + end + self.framebuffer = RL.LoadRenderTexture( self.framebufferSize ) + + self:updateContent() +end + +function TreeView:register( gui ) + function gui:TreeView( bounds, text, callback, grabCallback, dragCallback, styles, tooltip ) + return self:addControl( TreeView:new( bounds, text, callback, grabCallback, dragCallback, styles, tooltip ) ) + end +end + +return TreeView -- cgit v1.2.3