extends Tree

signal group_selected

func update_from_graph_edit(graph_edit) -> void:
	set_column_expand(0, true)
#	columns = 4
#	for i in range(1, 4):
#		set_column_expand(i, false)
#		set_column_min_width(i, 24)
	if graph_edit == null:
		return
	var file_name = "PTex"
	if graph_edit.save_path != null:
		file_name = graph_edit.save_path.get_file()
	fill_tree(graph_edit.top_generator, graph_edit.generator, file_name)

func fill_tree(top : MMGenGraph, selected : MMGenGraph, top_name : String) -> void:
	clear()
	if top == null or selected == null:
		return
	var root : TreeItem = create_item(null)
	root.set_text(0, top_name)
	if top == selected:
		root.set_custom_color(0, Color(0.5, 0.5, 1))
	else:
		root.set_custom_color(0, Color(1, 1, 1))
	root.set_metadata(0, top)
	fill_item(root, top, selected)

#func set_icon(item, generator):
#	var result = generator.render(0, 24, true)
#	while result is GDScriptFunctionState:
#		result = yield(result, "completed")
#	var tex = ImageTexture.new()
#	result.copy_to_texture(tex)
#	result.release()
#	item.set_icon(3, tex)

func fill_item(parent : TreeItem, generator : MMGenGraph, selected : MMGenGraph) -> void:
	for c in generator.get_children():
		if c is MMGenGraph:
			if c.is_template():
				continue
			var item : TreeItem = create_item(parent)
			item.set_text(0, c.get_type_name())
			if c == selected:
				item.set_custom_color(0, Color(0.5, 0.5, 1))
			elif c.is_editable():
				item.set_custom_color(0, Color(1, 1, 1))
			item.set_metadata(0, c)
#			if c.get_output_defs().size() > 0:
#				call_deferred("set_icon", item, c)
			fill_item(item, c, selected)

func _on_Hierarchy_item_double_clicked():
	emit_signal("group_selected", get_selected().get_metadata(0))