animation_tree_editor_plugin.h 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. /**************************************************************************/
  2. /* animation_tree_editor_plugin.h */
  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. #pragma once
  31. #include "editor/plugins/editor_plugin.h"
  32. #include "scene/animation/animation_tree.h"
  33. #include "scene/gui/graph_edit.h"
  34. class Button;
  35. class EditorFileDialog;
  36. class ScrollContainer;
  37. class AnimationTreeNodeEditorPlugin : public VBoxContainer {
  38. GDCLASS(AnimationTreeNodeEditorPlugin, VBoxContainer);
  39. public:
  40. virtual bool can_edit(const Ref<AnimationNode> &p_node) = 0;
  41. virtual void edit(const Ref<AnimationNode> &p_node) = 0;
  42. };
  43. class AnimationTreeEditor : public VBoxContainer {
  44. GDCLASS(AnimationTreeEditor, VBoxContainer);
  45. ScrollContainer *path_edit = nullptr;
  46. HBoxContainer *path_hb = nullptr;
  47. AnimationTree *tree = nullptr;
  48. MarginContainer *editor_base = nullptr;
  49. Vector<String> button_path;
  50. Vector<String> edited_path;
  51. Vector<AnimationTreeNodeEditorPlugin *> editors;
  52. void _update_path();
  53. void _clear_editors();
  54. ObjectID current_root;
  55. void _path_button_pressed(int p_path);
  56. void _animation_list_changed();
  57. static Vector<String> get_animation_list();
  58. protected:
  59. void _notification(int p_what);
  60. void _node_removed(Node *p_node);
  61. static AnimationTreeEditor *singleton;
  62. public:
  63. AnimationTree *get_animation_tree() { return tree; }
  64. void add_plugin(AnimationTreeNodeEditorPlugin *p_editor);
  65. void remove_plugin(AnimationTreeNodeEditorPlugin *p_editor);
  66. String get_base_path();
  67. bool can_edit(const Ref<AnimationNode> &p_node) const;
  68. void edit_path(const Vector<String> &p_path);
  69. Vector<String> get_edited_path() const;
  70. void enter_editor(const String &p_path = "");
  71. static AnimationTreeEditor *get_singleton() { return singleton; }
  72. void edit(AnimationTree *p_tree);
  73. AnimationTreeEditor();
  74. };
  75. class AnimationTreeEditorPlugin : public EditorPlugin {
  76. GDCLASS(AnimationTreeEditorPlugin, EditorPlugin);
  77. AnimationTreeEditor *anim_tree_editor = nullptr;
  78. Button *button = nullptr;
  79. public:
  80. virtual String get_plugin_name() const override { return "AnimationTree"; }
  81. bool has_main_screen() const override { return false; }
  82. virtual void edit(Object *p_object) override;
  83. virtual bool handles(Object *p_object) const override;
  84. virtual void make_visible(bool p_visible) override;
  85. AnimationTreeEditorPlugin();
  86. };