graph_node.h 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194
  1. /**************************************************************************/
  2. /* graph_node.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 "scene/gui/graph_element.h"
  32. class HBoxContainer;
  33. class GraphNode : public GraphElement {
  34. GDCLASS(GraphNode, GraphElement);
  35. friend class GraphEdit;
  36. struct Slot {
  37. bool enable_left = false;
  38. int type_left = 0;
  39. Color color_left = Color(1, 1, 1, 1);
  40. Ref<Texture2D> custom_port_icon_left;
  41. bool enable_right = false;
  42. int type_right = 0;
  43. Color color_right = Color(1, 1, 1, 1);
  44. Ref<Texture2D> custom_port_icon_right;
  45. bool draw_stylebox = true;
  46. };
  47. struct PortCache {
  48. Vector2 pos;
  49. int slot_index;
  50. int type = 0;
  51. Color color;
  52. };
  53. struct _MinSizeCache {
  54. int min_size = 0;
  55. bool will_stretch = false;
  56. int final_size = 0;
  57. };
  58. enum CustomAccessibilityAction {
  59. ACTION_CONNECT_INPUT,
  60. ACTION_CONNECT_OUTPUT,
  61. ACTION_FOLLOW_INPUT,
  62. ACTION_FOLLOW_OUTPUT,
  63. };
  64. void _accessibility_action_slot(const Variant &p_data);
  65. HBoxContainer *titlebar_hbox = nullptr;
  66. Label *title_label = nullptr;
  67. String title;
  68. Vector<PortCache> left_port_cache;
  69. Vector<PortCache> right_port_cache;
  70. HashMap<int, Slot> slot_table;
  71. Vector<int> slot_y_cache;
  72. Control::FocusMode slots_focus_mode = Control::FOCUS_ACCESSIBILITY;
  73. int slot_count = 0;
  74. int selected_slot = -1;
  75. struct ThemeCache {
  76. Ref<StyleBox> panel;
  77. Ref<StyleBox> panel_selected;
  78. Ref<StyleBox> panel_focus;
  79. Ref<StyleBox> titlebar;
  80. Ref<StyleBox> titlebar_selected;
  81. Ref<StyleBox> slot;
  82. Ref<StyleBox> slot_selected;
  83. int separation = 0;
  84. int port_h_offset = 0;
  85. Ref<Texture2D> port;
  86. Ref<Texture2D> resizer;
  87. Color resizer_color;
  88. } theme_cache;
  89. bool port_pos_dirty = true;
  90. bool ignore_invalid_connection_type = false;
  91. void _port_pos_update();
  92. protected:
  93. void _notification(int p_what);
  94. static void _bind_methods();
  95. virtual void _resort() override;
  96. virtual void draw_port(int p_slot_index, Point2i p_pos, bool p_left, const Color &p_color);
  97. GDVIRTUAL4(_draw_port, int, Point2i, bool, const Color &);
  98. bool _set(const StringName &p_name, const Variant &p_value);
  99. bool _get(const StringName &p_name, Variant &r_ret) const;
  100. void _get_property_list(List<PropertyInfo> *p_list) const;
  101. public:
  102. virtual String get_accessibility_container_name(const Node *p_node) const override;
  103. virtual void gui_input(const Ref<InputEvent> &p_event) override;
  104. void set_title(const String &p_title);
  105. String get_title() const;
  106. HBoxContainer *get_titlebar_hbox();
  107. void set_slot(int p_slot_index, bool p_enable_left, int p_type_left, const Color &p_color_left, bool p_enable_right, int p_type_right, const Color &p_color_right, const Ref<Texture2D> &p_custom_left = Ref<Texture2D>(), const Ref<Texture2D> &p_custom_right = Ref<Texture2D>(), bool p_draw_stylebox = true);
  108. void clear_slot(int p_slot_index);
  109. void clear_all_slots();
  110. bool is_slot_enabled_left(int p_slot_index) const;
  111. void set_slot_enabled_left(int p_slot_index, bool p_enable);
  112. void set_slot_type_left(int p_slot_index, int p_type);
  113. int get_slot_type_left(int p_slot_index) const;
  114. void set_slot_color_left(int p_slot_index, const Color &p_color);
  115. Color get_slot_color_left(int p_slot_index) const;
  116. void set_slot_custom_icon_left(int p_slot_index, const Ref<Texture2D> &p_custom_icon);
  117. Ref<Texture2D> get_slot_custom_icon_left(int p_slot_index) const;
  118. bool is_slot_enabled_right(int p_slot_index) const;
  119. void set_slot_enabled_right(int p_slot_index, bool p_enable);
  120. void set_slot_type_right(int p_slot_index, int p_type);
  121. int get_slot_type_right(int p_slot_index) const;
  122. void set_slot_color_right(int p_slot_index, const Color &p_color);
  123. Color get_slot_color_right(int p_slot_index) const;
  124. void set_slot_custom_icon_right(int p_slot_index, const Ref<Texture2D> &p_custom_icon);
  125. Ref<Texture2D> get_slot_custom_icon_right(int p_slot_index) const;
  126. bool is_slot_draw_stylebox(int p_slot_index) const;
  127. void set_slot_draw_stylebox(int p_slot_index, bool p_enable);
  128. void set_ignore_invalid_connection_type(bool p_ignore);
  129. bool is_ignoring_valid_connection_type() const;
  130. int get_input_port_count();
  131. Vector2 get_input_port_position(int p_port_idx);
  132. int get_input_port_type(int p_port_idx);
  133. Color get_input_port_color(int p_port_idx);
  134. int get_input_port_slot(int p_port_idx);
  135. int get_output_port_count();
  136. Vector2 get_output_port_position(int p_port_idx);
  137. int get_output_port_type(int p_port_idx);
  138. Color get_output_port_color(int p_port_idx);
  139. int get_output_port_slot(int p_port_idx);
  140. void set_slots_focus_mode(Control::FocusMode p_focus_mode);
  141. Control::FocusMode get_slots_focus_mode() const;
  142. virtual Size2 get_minimum_size() const override;
  143. virtual CursorShape get_cursor_shape(const Point2 &p_pos = Point2i()) const override;
  144. virtual Vector<int> get_allowed_size_flags_horizontal() const override;
  145. virtual Vector<int> get_allowed_size_flags_vertical() const override;
  146. GraphNode();
  147. };