iup-stack/iup/srclua5/elem/tree.lua

130 lines
3.8 KiB
Lua
Executable File

------------------------------------------------------------------------------
-- Tree class
------------------------------------------------------------------------------
local ctrl = {
nick = "tree",
parent = iup.WIDGET,
subdir = "elem",
creation = "",
callback = {
selection_cb = "nn",
multiselection_cb = "nn", -- fake definition to be replaced by treefuncs module
multiunselection_cb = "nn", -- fake definition to be replaced by treefuncs module
branchopen_cb = "n",
branchclose_cb = "n",
executeleaf_cb = "n",
executebranch_cb = "n",
noderemoved_cb = "s", -- fake definition to be replaced by treefuncs module
rename_cb = "ns",
showrename_cb = "n",
rightclick_cb = "n",
dragdrop_cb = "nnnn",
togglevalue_cb = "nn",
},
extrafuncs = 1,
}
iup.TREEREFTABLE={}
function iup.TreeSetNodeAttributes(ih, id, attrs)
for attr, val in pairs(attrs) do
ih[attr..id] = val
end
end
ctrl.SetNodeAttributes = iup.TreeSetNodeAttributes
function iup.TreeSetAncestorsAttributes(ih, ini, attrs)
ini = ih["parent"..ini]
local stack = {}
while ini do
table.insert(stack, 1, ini)
ini = ih["parent"..ini]
end
for i = 1, #stack do
iup.TreeSetNodeAttributes(ih, stack[i], attrs)
end
end
ctrl.SetAncestorsAttributes = iup.TreeSetAncestorsAttributes
function iup.TreeSetDescendantsAttributes(ih, ini, attrs)
local id = ini
for i = 1, ih["childcount"..ini] do
id = id+1
iup.TreeSetNodeAttributes(ih, id, attrs)
if ih["kind"..id] == "BRANCH" then
id = iup.TreeSetDescendantsAttributes(ih, id, attrs)
end
end
return id
end
ctrl.SetDescendantsAttributes = iup.TreeSetDescendantsAttributes
function iup.TreeSetAttributeHandle(ih, name, value)
if iup.GetClass(value) == "iupHandle" then value = iup.SetHandleName(value) end
iup.SetAttribute(ih, name, value)
end
-- must be after the branch has nodes
function iup.TreeSetState(ih, tnode, id)
if tnode.state then iup.SetAttribute(ih, "STATE"..id, tnode.state) end
end
function iup.TreeSetNodeAttrib(ih, tnode, id)
if tnode.color then iup.SetAttribute(ih, "COLOR"..id, tnode.color) end
if tnode.titlefont then iup.SetAttribute(ih, "TITLEFONT"..id, tnode.titlefont) end
if tnode.marked then iup.SetAttribute(ih, "MARKED"..id, tnode.marked) end
if tnode.image then iup.TreeSetAttributeHandle(ih, "IMAGE"..id, tnode.image) end
if tnode.imageexpanded then iup.TreeSetAttributeHandle(ih, "IMAGEEXPANDED"..id, tnode.imageexpanded) end
if tnode.userid then iup.TreeSetUserId(ih, id, tnode.userid) end
end
function iup.TreeAddNodesRec(ih, t, id)
if t == nil then return end
local cont = #t
while cont >= 0 do
local tnode = t[cont]
if type(tnode) == "table" then
if tnode.branchname then
iup.SetAttribute(ih, "ADDBRANCH"..id, tnode.branchname)
iup.TreeSetNodeAttrib(ih, tnode, id+1)
iup.TreeAddNodesRec(ih, tnode, id+1)
iup.TreeSetState(ih, tnode, id+1)
elseif tnode.leafname then
iup.SetAttribute(ih, "ADDLEAF"..id, tnode.leafname)
iup.TreeSetNodeAttrib(ih, tnode, id+1)
end
else
if tnode then
iup.SetAttribute(ih, "ADDLEAF"..id, tnode)
end
end
cont = cont - 1
end
end
function iup.TreeAddNodes(ih, t, id)
if (not id) then
id = 0 -- default is the root
if t.branchname then iup.SetAttribute(ih, "TITLE0", t.branchname) end
iup.TreeSetNodeAttrib(ih, t, 0)
end
iup.TreeAddNodesRec(ih, t, id)
if (id == 0) then iup.TreeSetState(ih, t, 0) end
end
ctrl.AddNodes = iup.TreeAddNodes
-- defined in C
ctrl.GetId = iup.TreeGetId
ctrl.GetUserId = iup.TreeGetUserId
ctrl.SetUserId = iup.TreeSetUserId
-- backward compatibility
iup.TreeSetValue = iup.TreeAddNodes
function ctrl.createElement(class, param)
return iup.Tree()
end
iup.RegisterWidget(ctrl)
iup.SetClass(ctrl, "iupWidget")