material-maker/addons/material_maker/library.gd
Hugo Locurcio 93c84650cd
Add return type hints to all scripts
This also simplifies some functions.

This partially addresses #30.
2019-10-20 16:35:45 +02:00

134 lines
4.0 KiB
GDScript

tool
extends VBoxContainer
var libraries = []
onready var tree : Tree = $Tree
onready var filter_line_edit : LineEdit = $HBoxContainer/Filter
func _ready() -> void:
tree.set_column_expand(0, true)
tree.set_column_expand(1, false)
tree.set_column_min_width(1, 32)
var lib_path = OS.get_executable_path().get_base_dir()+"/library/base.json"
if !add_library(lib_path):
add_library("res://addons/material_maker/library/base.json")
add_library("user://library/user.json")
update_tree()
func _unhandled_input(event : InputEvent) -> void:
if event is InputEventKey and event.pressed and event.command and event.scancode == KEY_F:
filter_line_edit.grab_focus()
filter_line_edit.select_all()
func get_selected_item_name() -> String:
var tree_item : TreeItem = tree.get_selected()
var rv = ""
while tree_item != null and tree_item != tree.get_root():
if rv == "":
rv = tree_item.get_text(0)
else:
rv = tree_item.get_text(0)+"/"+rv
tree_item = tree_item.get_parent()
return rv
func add_library(file_name : String, filter : String = "") -> bool:
var root = tree.get_root()
var file = File.new()
if file.open(file_name, File.READ) != OK:
return false
var lib = parse_json(file.get_as_text())
file.close()
if lib != null and lib.has("lib"):
for m in lib.lib:
m.library = file_name
libraries.push_back(lib.lib)
return true
return false
func update_tree(filter : String = "") -> void:
filter = filter.to_lower()
tree.clear()
var root = tree.create_item()
for l in libraries:
for m in l:
if filter == "" or m.tree_item.to_lower().find(filter) != -1:
add_item(m, m.tree_item, get_preview_texture(m), null, filter != "")
func get_preview_texture(data : Dictionary) -> ImageTexture:
if data.has("icon") and data.has("library"):
var image_path = data.library.left(data.library.rfind("."))+"/"+data.icon+".png"
var t : ImageTexture
if image_path.left(6) == "res://":
return load(image_path) as ImageTexture
else:
t = ImageTexture.new()
var image : Image = Image.new()
image.load(image_path)
t.create_from_image(image)
return t
return null
func add_item(item, item_name, item_icon = null, item_parent = null, force_expand = false) -> TreeItem:
if item_parent == null:
item.tree_item = item_name
item_parent = tree.get_root()
var slash_position = item_name.find("/")
if slash_position == -1:
var new_item : TreeItem = null
var c = item_parent.get_children()
while c != null:
if c.get_text(0) == item_name:
new_item = c
break
c = c.get_next()
if new_item == null:
new_item = tree.create_item(item_parent)
new_item.set_text(0, item_name)
new_item.collapsed = !force_expand
if item_icon != null:
new_item.set_icon(1, item_icon)
new_item.set_icon_max_width(1, 32)
if item.has("type") || item.has("nodes"):
new_item.set_metadata(0, item)
if item.has("collapsed"):
new_item.collapsed = item.collapsed and !force_expand
return new_item
else:
var prefix = item_name.left(slash_position)
var suffix = item_name.right(slash_position+1)
var new_parent = null
var c = item_parent.get_children()
while c != null:
if c.get_text(0) == prefix:
new_parent = c
break
c = c.get_next()
if new_parent == null:
new_parent = tree.create_item(item_parent)
new_parent.collapsed = !force_expand
new_parent.set_text(0, prefix)
return add_item(item, suffix, item_icon, new_parent, force_expand)
func serialize_library(array, library_name = null, item = null) -> void:
if item == null:
item = tree.get_root()
item = item.get_children()
while item != null:
var m = item.get_metadata(0)
if m != null and (library_name == null or (m.has("library") and m.library == library_name)):
array.append(m)
serialize_library(array, library_name, item)
item = item.get_next()
func save_library(library_name, item = null) -> void:
var array = []
serialize_library(array, library_name)
var file = File.new()
if file.open(library_name, File.WRITE) == OK:
file.store_string(to_json({lib=array}))
file.close()
func _on_Filter_text_changed(filter) -> void:
update_tree(filter)