281 lines
7.8 KiB
GDScript
281 lines
7.8 KiB
GDScript
tool
|
|
extends EditorPlugin
|
|
|
|
const main_scene_path : String = "res://addons/prop_tool/scenes/main.tscn"
|
|
const temp_scene_path : String = "res://addons/prop_tool/scenes/main_temp.tscn"
|
|
const temp_path : String = "res://addons/prop_tool/scenes/temp/"
|
|
|
|
var buttons_added : bool = false
|
|
var light_button : ToolButton
|
|
var mesh_button : ToolButton
|
|
var prop_button : ToolButton
|
|
var scene_button : ToolButton
|
|
var entity_button : ToolButton
|
|
|
|
var edited_prop : PropData
|
|
|
|
func _enter_tree():
|
|
var dir : Directory = Directory.new()
|
|
dir.copy(main_scene_path, temp_scene_path)
|
|
|
|
light_button = ToolButton.new()
|
|
light_button.text = "Light"
|
|
light_button.connect("pressed", self, "add_light")
|
|
|
|
mesh_button = ToolButton.new()
|
|
mesh_button.text = "Mesh"
|
|
mesh_button.connect("pressed", self, "add_mesh")
|
|
|
|
prop_button = ToolButton.new()
|
|
prop_button.text = "Prop"
|
|
prop_button.connect("pressed", self, "add_prop")
|
|
|
|
scene_button = ToolButton.new()
|
|
scene_button.text = "Scene"
|
|
scene_button.connect("pressed", self, "add_scene")
|
|
|
|
entity_button = ToolButton.new()
|
|
entity_button.text = "Entity"
|
|
entity_button.connect("pressed", self, "add_entity")
|
|
|
|
add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, light_button)
|
|
add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, mesh_button)
|
|
add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, prop_button)
|
|
add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, scene_button)
|
|
add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, entity_button)
|
|
|
|
light_button.hide()
|
|
mesh_button.hide()
|
|
prop_button.hide()
|
|
scene_button.hide()
|
|
entity_button.hide()
|
|
|
|
connect("scene_changed", self, "scene_changed")
|
|
|
|
|
|
func _exit_tree():
|
|
var dir : Directory = Directory.new()
|
|
dir.remove(temp_scene_path)
|
|
|
|
if is_instance_valid(light_button):
|
|
light_button.queue_free()
|
|
|
|
if is_instance_valid(mesh_button):
|
|
mesh_button.queue_free()
|
|
|
|
if is_instance_valid(prop_button):
|
|
prop_button.queue_free()
|
|
|
|
if is_instance_valid(scene_button):
|
|
scene_button.queue_free()
|
|
|
|
if is_instance_valid(entity_button):
|
|
entity_button.queue_free()
|
|
|
|
light_button = null
|
|
mesh_button = null
|
|
prop_button = null
|
|
scene_button = null
|
|
entity_button = null
|
|
|
|
edited_prop = null
|
|
|
|
disconnect("scene_changed", self, "scene_changed")
|
|
|
|
func scene_changed(scene):
|
|
if scene.has_method("set_target_prop"):
|
|
scene.plugin = self
|
|
|
|
if not buttons_added:
|
|
light_button.show()
|
|
mesh_button.show()
|
|
prop_button.show()
|
|
scene_button.show()
|
|
entity_button.show()
|
|
# add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, light_button)
|
|
# add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, mesh_button)
|
|
# add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, prop_button)
|
|
# add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, scene_button)
|
|
# add_control_to_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, entity_button)
|
|
|
|
buttons_added = true
|
|
else:
|
|
if buttons_added:
|
|
light_button.hide()
|
|
mesh_button.hide()
|
|
prop_button.hide()
|
|
scene_button.hide()
|
|
entity_button.hide()
|
|
#
|
|
# remove_control_from_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, light_button)
|
|
# remove_control_from_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, mesh_button)
|
|
# remove_control_from_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, prop_button)
|
|
# remove_control_from_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, scene_button)
|
|
# remove_control_from_container(EditorPlugin.CONTAINER_SPATIAL_EDITOR_MENU, entity_button)
|
|
|
|
buttons_added = false
|
|
|
|
func apply_changes() -> void:
|
|
var scene : Node = get_editor_interface().get_edited_scene_root()
|
|
|
|
if scene is PropTool:
|
|
# if scene.has_method("set_target_prop") and scene.has_method("save"):
|
|
scene.save()
|
|
|
|
func edit(object : Object) -> void:
|
|
var pedited_prop = object as PropData
|
|
|
|
var p : String = create_or_get_scene_path(pedited_prop)
|
|
|
|
get_editor_interface().open_scene_from_path(p)
|
|
|
|
func handles(object : Object) -> bool:
|
|
return object is PropData
|
|
|
|
func add_light():
|
|
var selection : Array = get_editor_interface().get_selection().get_selected_nodes()
|
|
var selected : Node
|
|
|
|
if selection.size() != 1:
|
|
selected = get_editor_interface().get_edited_scene_root()
|
|
else:
|
|
selected = selection[0]
|
|
|
|
var s : Node = selected
|
|
var n = PropToolLight.new()
|
|
|
|
var u : UndoRedo = get_undo_redo()
|
|
u.create_action("Add Light")
|
|
u.add_do_method(s, "add_child", n)
|
|
u.add_do_property(n, "owner", get_editor_interface().get_edited_scene_root())
|
|
u.add_undo_method(s, "remove_child", n)
|
|
u.commit_action()
|
|
|
|
get_editor_interface().get_selection().clear()
|
|
get_editor_interface().get_selection().add_node(n)
|
|
|
|
|
|
func add_mesh():
|
|
var selected : Array = get_editor_interface().get_selection().get_selected_nodes()
|
|
|
|
if selected.size() != 1:
|
|
return
|
|
|
|
var s : Node = selected[0]
|
|
var n = PropToolMesh.new()
|
|
|
|
var u : UndoRedo = get_undo_redo()
|
|
u.create_action("Add Mesh")
|
|
u.add_do_method(s, "add_child", n)
|
|
u.add_do_property(n, "owner", get_editor_interface().get_edited_scene_root())
|
|
u.add_undo_method(s, "remove_child", n)
|
|
u.commit_action()
|
|
|
|
get_editor_interface().get_selection().clear()
|
|
get_editor_interface().get_selection().add_node(n)
|
|
|
|
|
|
func add_prop():
|
|
var selected : Array = get_editor_interface().get_selection().get_selected_nodes()
|
|
|
|
if selected.size() != 1:
|
|
return
|
|
|
|
var s : Node = selected[0]
|
|
var n = PropTool.new()
|
|
|
|
var u : UndoRedo = get_undo_redo()
|
|
u.create_action("Add Prop")
|
|
u.add_do_method(s, "add_child", n)
|
|
u.add_do_property(n, "owner", get_editor_interface().get_edited_scene_root())
|
|
u.add_undo_method(s, "remove_child", n)
|
|
u.commit_action()
|
|
|
|
get_editor_interface().get_selection().clear()
|
|
get_editor_interface().get_selection().add_node(n)
|
|
|
|
func add_scene():
|
|
var selected : Array = get_editor_interface().get_selection().get_selected_nodes()
|
|
|
|
if selected.size() != 1:
|
|
return
|
|
|
|
var s : Node = selected[0]
|
|
var n = PropToolScene.new()
|
|
|
|
var u : UndoRedo = get_undo_redo()
|
|
u.create_action("Add Scene")
|
|
u.add_do_method(s, "add_child", n)
|
|
u.add_do_property(n, "owner", get_editor_interface().get_edited_scene_root())
|
|
u.add_undo_method(s, "remove_child", n)
|
|
u.commit_action()
|
|
|
|
get_editor_interface().get_selection().clear()
|
|
get_editor_interface().get_selection().add_node(n)
|
|
|
|
func add_entity():
|
|
var selected : Array = get_editor_interface().get_selection().get_selected_nodes()
|
|
|
|
if selected.size() != 1:
|
|
return
|
|
|
|
var s : Node = selected[0]
|
|
var n = PropToolEntity.new()
|
|
|
|
var u : UndoRedo = get_undo_redo()
|
|
u.create_action("Add Entity")
|
|
u.add_do_method(s, "add_child", n)
|
|
u.add_do_property(n, "owner", get_editor_interface().get_edited_scene_root())
|
|
u.add_undo_method(s, "remove_child", n)
|
|
u.commit_action()
|
|
|
|
get_editor_interface().get_selection().clear()
|
|
get_editor_interface().get_selection().add_node(n)
|
|
|
|
func create_or_get_scene_path(data: PropData) -> String:
|
|
var path : String = temp_path + data.resource_path.get_file().get_basename() + ".tscn"
|
|
|
|
var dir : Directory = Directory.new()
|
|
|
|
if not dir.file_exists(path):
|
|
create_scene(data)
|
|
|
|
return path
|
|
|
|
func create_or_get_scene(data: PropData) -> PropTool:
|
|
if data == null:
|
|
print("PropTool plugin.gd: Data is null!")
|
|
return null
|
|
|
|
var path : String = temp_path + data.resource_path.get_file().get_basename() + ".tscn"
|
|
|
|
var dir : Directory = Directory.new()
|
|
|
|
var ps : PackedScene
|
|
|
|
if not dir.file_exists(path):
|
|
ps = create_scene(data)
|
|
else:
|
|
ps = ResourceLoader.load(path, "PackedScene")
|
|
|
|
if ps == null:
|
|
return null
|
|
|
|
var pt : PropTool = ps.instance() as PropTool
|
|
pt.plugin = self
|
|
return pt
|
|
|
|
func create_scene(data: PropData) -> PackedScene:
|
|
var pt : PropTool = PropTool.new()
|
|
|
|
pt.plugin = self
|
|
pt.set_target_prop(data)
|
|
|
|
var ps : PackedScene = PackedScene.new()
|
|
ps.pack(pt)
|
|
|
|
var err = ResourceSaver.save(temp_path + data.resource_path.get_file().get_basename() + ".tscn", ps)
|
|
|
|
pt.queue_free()
|
|
return ps
|