skeleton_2d_editor_plugin.cpp 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  1. /**************************************************************************/
  2. /* skeleton_2d_editor_plugin.cpp */
  3. /**************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /**************************************************************************/
  8. /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
  9. /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /**************************************************************************/
  30. #include "skeleton_2d_editor_plugin.h"
  31. #include "editor/editor_node.h"
  32. #include "editor/editor_string_names.h"
  33. #include "editor/editor_undo_redo_manager.h"
  34. #include "editor/plugins/canvas_item_editor_plugin.h"
  35. #include "scene/gui/dialogs.h"
  36. #include "scene/gui/menu_button.h"
  37. void Skeleton2DEditor::_node_removed(Node *p_node) {
  38. if (p_node == node) {
  39. node = nullptr;
  40. options->hide();
  41. }
  42. }
  43. void Skeleton2DEditor::edit(Skeleton2D *p_sprite) {
  44. node = p_sprite;
  45. }
  46. void Skeleton2DEditor::_menu_option(int p_option) {
  47. if (!node) {
  48. return;
  49. }
  50. switch (p_option) {
  51. case MENU_OPTION_SET_REST: {
  52. if (node->get_bone_count() == 0) {
  53. err_dialog->set_text(TTR("This skeleton has no bones, create some children Bone2D nodes."));
  54. err_dialog->popup_centered();
  55. return;
  56. }
  57. EditorUndoRedoManager *ur = EditorUndoRedoManager::get_singleton();
  58. ur->create_action(TTR("Set Rest Pose to Bones"));
  59. for (int i = 0; i < node->get_bone_count(); i++) {
  60. Bone2D *bone = node->get_bone(i);
  61. ur->add_do_method(bone, "set_transform", bone->get_rest());
  62. ur->add_undo_method(bone, "set_transform", bone->get_transform());
  63. }
  64. ur->commit_action();
  65. } break;
  66. case MENU_OPTION_MAKE_REST: {
  67. if (node->get_bone_count() == 0) {
  68. err_dialog->set_text(TTR("This skeleton has no bones, create some children Bone2D nodes."));
  69. err_dialog->popup_centered();
  70. return;
  71. }
  72. EditorUndoRedoManager *ur = EditorUndoRedoManager::get_singleton();
  73. ur->create_action(TTR("Create Rest Pose from Bones"));
  74. for (int i = 0; i < node->get_bone_count(); i++) {
  75. Bone2D *bone = node->get_bone(i);
  76. ur->add_do_method(bone, "set_rest", bone->get_transform());
  77. ur->add_undo_method(bone, "set_rest", bone->get_rest());
  78. }
  79. ur->commit_action();
  80. } break;
  81. }
  82. }
  83. Skeleton2DEditor::Skeleton2DEditor() {
  84. options = memnew(MenuButton);
  85. CanvasItemEditor::get_singleton()->add_control_to_menu_panel(options);
  86. options->set_text(TTR("Skeleton2D"));
  87. options->set_button_icon(EditorNode::get_singleton()->get_editor_theme()->get_icon(SNAME("Skeleton2D"), EditorStringName(EditorIcons)));
  88. options->get_popup()->add_item(TTR("Reset to Rest Pose"), MENU_OPTION_SET_REST);
  89. options->get_popup()->add_separator();
  90. // Use the "Overwrite" word to highlight that this is a destructive operation.
  91. options->get_popup()->add_item(TTR("Overwrite Rest Pose"), MENU_OPTION_MAKE_REST);
  92. options->set_switch_on_hover(true);
  93. options->get_popup()->connect(SceneStringName(id_pressed), callable_mp(this, &Skeleton2DEditor::_menu_option));
  94. err_dialog = memnew(AcceptDialog);
  95. add_child(err_dialog);
  96. }
  97. void Skeleton2DEditorPlugin::edit(Object *p_object) {
  98. sprite_editor->edit(Object::cast_to<Skeleton2D>(p_object));
  99. }
  100. bool Skeleton2DEditorPlugin::handles(Object *p_object) const {
  101. return p_object->is_class("Skeleton2D");
  102. }
  103. void Skeleton2DEditorPlugin::make_visible(bool p_visible) {
  104. if (p_visible) {
  105. sprite_editor->options->show();
  106. } else {
  107. sprite_editor->options->hide();
  108. sprite_editor->edit(nullptr);
  109. }
  110. }
  111. Skeleton2DEditorPlugin::Skeleton2DEditorPlugin() {
  112. sprite_editor = memnew(Skeleton2DEditor);
  113. EditorNode::get_singleton()->get_gui_base()->add_child(sprite_editor);
  114. make_visible(false);
  115. //sprite_editor->options->hide();
  116. }
  117. Skeleton2DEditorPlugin::~Skeleton2DEditorPlugin() {
  118. }