2018-08-29 22:38:13 +02:00
/**************************************************************************/
/* visual_shader_editor_plugin.h */
/**************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* https://godotengine.org */
/**************************************************************************/
/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
/* */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the */
/* "Software"), to deal in the Software without restriction, including */
/* without limitation the rights to use, copy, modify, merge, publish, */
/* distribute, sublicense, and/or sell copies of the Software, and to */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions: */
/* */
/* The above copyright notice and this permission notice shall be */
/* included in all copies or substantial portions of the Software. */
/* */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
/**************************************************************************/
2018-07-14 23:15:42 +02:00
# ifndef VISUAL_SHADER_EDITOR_PLUGIN_H
# define VISUAL_SHADER_EDITOR_PLUGIN_H
2022-11-11 20:12:48 +01:00
# include "editor/editor_properties.h"
2024-04-21 10:11:35 +02:00
# include "editor/plugins/editor_plugin.h"
2022-07-31 20:14:15 +02:00
# include "editor/plugins/editor_resource_conversion_plugin.h"
2024-01-19 11:35:46 +01:00
# include "scene/gui/graph_edit.h"
2022-11-19 12:45:49 +01:00
# include "scene/resources/syntax_highlighter.h"
2018-07-14 23:15:42 +02:00
# include "scene/resources/visual_shader.h"
2022-11-19 12:45:49 +01:00
class CodeEdit ;
2023-08-09 18:31:15 +02:00
class ColorPicker ;
2022-08-01 20:01:48 +02:00
class CurveEditor ;
2023-08-09 18:31:15 +02:00
class GraphElement ;
2024-02-07 02:37:26 +01:00
class GraphFrame ;
2022-11-19 12:45:49 +01:00
class MenuButton ;
class PopupPanel ;
class RichTextLabel ;
class Tree ;
2022-08-01 20:01:48 +02:00
2022-07-16 08:59:20 +02:00
class VisualShaderEditor ;
2021-06-04 18:03:15 +02:00
class VisualShaderNodePlugin : public RefCounted {
GDCLASS ( VisualShaderNodePlugin , RefCounted ) ;
2019-03-19 19:35:57 +01:00
2022-07-16 08:59:20 +02:00
protected :
VisualShaderEditor * vseditor = nullptr ;
2018-07-14 23:15:42 +02:00
protected :
static void _bind_methods ( ) ;
2022-05-03 01:43:50 +02:00
GDVIRTUAL2RC ( Object * , _create_editor , Ref < Resource > , Ref < VisualShaderNode > )
2021-08-22 03:52:44 +02:00
2018-07-14 23:15:42 +02:00
public :
2022-07-16 08:59:20 +02:00
void set_editor ( VisualShaderEditor * p_editor ) ;
2019-08-05 16:13:02 +02:00
virtual Control * create_editor ( const Ref < Resource > & p_parent_resource , const Ref < VisualShaderNode > & p_node ) ;
2018-07-14 23:15:42 +02:00
} ;
2021-06-04 18:03:15 +02:00
class VisualShaderGraphPlugin : public RefCounted {
GDCLASS ( VisualShaderGraphPlugin , RefCounted ) ;
2020-09-09 16:40:27 +02:00
private :
2022-07-16 08:59:20 +02:00
VisualShaderEditor * editor = nullptr ;
2020-09-10 07:31:07 +02:00
struct InputPort {
2020-11-24 10:12:55 +01:00
Button * default_input_button = nullptr ;
2020-09-10 07:31:07 +02:00
} ;
2020-09-09 16:40:27 +02:00
struct Port {
2020-11-24 10:12:55 +01:00
TextureButton * preview_button = nullptr ;
2020-09-09 16:40:27 +02:00
} ;
struct Link {
2020-11-24 10:12:55 +01:00
VisualShader : : Type type = VisualShader : : Type : : TYPE_MAX ;
VisualShaderNode * visual_node = nullptr ;
2023-08-09 18:31:15 +02:00
GraphElement * graph_element = nullptr ;
2021-04-05 13:16:16 +02:00
bool preview_visible = false ;
2020-11-24 10:12:55 +01:00
int preview_pos = 0 ;
2022-05-13 15:04:37 +02:00
HashMap < int , InputPort > input_ports ;
HashMap < int , Port > output_ports ;
2020-11-24 10:12:55 +01:00
VBoxContainer * preview_box = nullptr ;
2022-08-27 11:22:43 +02:00
LineEdit * parameter_name = nullptr ;
2020-11-24 10:12:55 +01:00
CodeEdit * expression_edit = nullptr ;
2021-07-04 10:56:21 +02:00
CurveEditor * curve_editors [ 3 ] = { nullptr , nullptr , nullptr } ;
2020-09-09 16:40:27 +02:00
} ;
Ref < VisualShader > visual_shader ;
2022-05-13 15:04:37 +02:00
HashMap < int , Link > links ;
2020-09-10 07:31:07 +02:00
List < VisualShader : : Connection > connections ;
2020-09-09 16:40:27 +02:00
2022-04-12 19:09:29 +02:00
Color vector_expanded_color [ 4 ] ;
2020-12-20 16:45:53 +01:00
2024-02-09 15:02:58 +01:00
// Visual shader specific theme for using MSDF fonts (on GraphNodes) which reduce aliasing at higher zoom levels.
Ref < Theme > vs_msdf_fonts_theme ;
2020-09-09 16:40:27 +02:00
protected :
static void _bind_methods ( ) ;
public :
2022-07-16 08:59:20 +02:00
void set_editor ( VisualShaderEditor * p_editor ) ;
2020-09-09 16:40:27 +02:00
void register_shader ( VisualShader * p_visual_shader ) ;
2022-04-07 12:23:40 +02:00
void set_connections ( const List < VisualShader : : Connection > & p_connections ) ;
2023-08-09 18:31:15 +02:00
void register_link ( VisualShader : : Type p_type , int p_id , VisualShaderNode * p_visual_node , GraphElement * p_graph_element ) ;
2020-09-09 16:40:27 +02:00
void register_output_port ( int p_id , int p_port , TextureButton * p_button ) ;
2022-08-27 11:22:43 +02:00
void register_parameter_name ( int p_id , LineEdit * p_parameter_name ) ;
2020-10-01 10:10:51 +02:00
void register_default_input_button ( int p_node_id , int p_port_id , Button * p_button ) ;
2020-10-15 13:47:27 +02:00
void register_expression_edit ( int p_node_id , CodeEdit * p_expression_edit ) ;
2021-07-04 10:56:21 +02:00
void register_curve_editor ( int p_node_id , int p_index , CurveEditor * p_curve_editor ) ;
2020-09-09 16:40:27 +02:00
void clear_links ( ) ;
void set_shader_type ( VisualShader : : Type p_type ) ;
bool is_preview_visible ( int p_id ) const ;
2020-09-21 13:32:59 +02:00
void update_node ( VisualShader : : Type p_type , int p_id ) ;
void update_node_deferred ( VisualShader : : Type p_type , int p_node_id ) ;
2024-02-07 02:37:26 +01:00
void add_node ( VisualShader : : Type p_type , int p_id , bool p_just_update , bool p_update_frames ) ;
2023-01-12 12:19:55 +01:00
void remove_node ( VisualShader : : Type p_type , int p_id , bool p_just_update ) ;
2020-09-10 07:31:07 +02:00
void connect_nodes ( VisualShader : : Type p_type , int p_from_node , int p_from_port , int p_to_node , int p_to_port ) ;
void disconnect_nodes ( VisualShader : : Type p_type , int p_from_node , int p_from_port , int p_to_node , int p_to_port ) ;
2023-02-03 13:09:51 +01:00
void show_port_preview ( VisualShader : : Type p_type , int p_node_id , int p_port_id , bool p_is_valid ) ;
2024-02-07 02:37:26 +01:00
void update_frames ( VisualShader : : Type p_type , int p_node ) ;
2020-09-10 07:31:07 +02:00
void set_node_position ( VisualShader : : Type p_type , int p_id , const Vector2 & p_position ) ;
void refresh_node_ports ( VisualShader : : Type p_type , int p_node ) ;
2024-02-15 17:25:58 +01:00
void set_input_port_default_value ( VisualShader : : Type p_type , int p_node_id , int p_port_id , const Variant & p_value ) ;
2022-08-27 11:22:43 +02:00
void update_parameter_refs ( ) ;
void set_parameter_name ( VisualShader : : Type p_type , int p_node_id , const String & p_name ) ;
2020-10-04 11:11:58 +02:00
void update_curve ( int p_node_id ) ;
2021-07-14 00:06:44 +02:00
void update_curve_xyz ( int p_node_id ) ;
2020-10-15 13:47:27 +02:00
void set_expression ( VisualShader : : Type p_type , int p_node_id , const String & p_expression ) ;
2024-02-07 02:37:26 +01:00
void attach_node_to_frame ( VisualShader : : Type p_type , int p_node_id , int p_frame_id ) ;
void detach_node_from_frame ( VisualShader : : Type p_type , int p_node_id ) ;
void set_frame_color_enabled ( VisualShader : : Type p_type , int p_node_id , bool p_enable ) ;
void set_frame_color ( VisualShader : : Type p_type , int p_node_id , const Color & p_color ) ;
void set_frame_autoshrink_enabled ( VisualShader : : Type p_type , int p_node_id , bool p_enable ) ;
2020-10-01 10:10:51 +02:00
int get_constant_index ( float p_constant ) const ;
2022-12-07 19:50:28 +01:00
Ref < Script > get_node_script ( int p_node_id ) const ;
2020-12-20 16:45:53 +01:00
void update_theme ( ) ;
2023-02-03 13:09:51 +01:00
bool is_node_has_parameter_instances_relatively ( VisualShader : : Type p_type , int p_node ) const ;
2020-09-10 07:31:07 +02:00
VisualShader : : Type get_shader_type ( ) const ;
2020-09-09 16:40:27 +02:00
VisualShaderGraphPlugin ( ) ;
~ VisualShaderGraphPlugin ( ) ;
} ;
2022-08-01 20:01:48 +02:00
class VisualShaderEditedProperty : public RefCounted {
GDCLASS ( VisualShaderEditedProperty , RefCounted ) ;
private :
Variant edited_property ;
protected :
static void _bind_methods ( ) ;
public :
2024-02-15 17:25:58 +01:00
void set_edited_property ( const Variant & p_variant ) ;
2022-08-01 20:01:48 +02:00
Variant get_edited_property ( ) const ;
VisualShaderEditedProperty ( ) { }
} ;
2018-07-14 23:15:42 +02:00
class VisualShaderEditor : public VBoxContainer {
GDCLASS ( VisualShaderEditor , VBoxContainer ) ;
2020-09-09 16:40:27 +02:00
friend class VisualShaderGraphPlugin ;
2018-07-14 23:15:42 +02:00
2022-08-01 20:01:48 +02:00
PopupPanel * property_editor_popup = nullptr ;
EditorProperty * property_editor = nullptr ;
2022-01-07 08:59:15 +01:00
int editing_node = - 1 ;
int editing_port = - 1 ;
2022-08-01 20:01:48 +02:00
Ref < VisualShaderEditedProperty > edited_property_holder ;
2018-07-14 23:15:42 +02:00
Ref < VisualShader > visual_shader ;
2022-01-07 08:59:15 +01:00
GraphEdit * graph = nullptr ;
Button * add_node = nullptr ;
2022-11-16 20:05:09 +01:00
MenuButton * varying_button = nullptr ;
2022-01-07 08:59:15 +01:00
Button * preview_shader = nullptr ;
2018-07-14 23:15:42 +02:00
2020-09-07 09:28:13 +02:00
OptionButton * edit_type = nullptr ;
2022-01-07 08:59:15 +01:00
OptionButton * edit_type_standard = nullptr ;
OptionButton * edit_type_particles = nullptr ;
OptionButton * edit_type_sky = nullptr ;
OptionButton * edit_type_fog = nullptr ;
CheckBox * custom_mode_box = nullptr ;
2020-09-15 09:57:40 +02:00
bool custom_mode_enabled = false ;
2018-07-14 23:15:42 +02:00
2022-01-07 08:59:15 +01:00
bool pending_update_preview = false ;
bool shader_error = false ;
Window * preview_window = nullptr ;
VBoxContainer * preview_vbox = nullptr ;
CodeEdit * preview_text = nullptr ;
Ref < CodeHighlighter > syntax_highlighter = nullptr ;
PanelContainer * error_panel = nullptr ;
Label * error_label = nullptr ;
2019-08-18 11:29:22 +02:00
2023-01-23 10:01:28 +01:00
bool pending_custom_scripts_to_delete = false ;
List < Ref < Script > > custom_scripts_to_delete ;
2022-12-07 19:50:28 +01:00
bool _block_update_options_menu = false ;
bool _block_rebuild_shader = false ;
2019-02-22 18:38:58 +01:00
Point2 saved_node_pos ;
2022-01-07 08:59:15 +01:00
bool saved_node_pos_dirty = false ;
2019-02-22 18:38:58 +01:00
2022-01-07 08:59:15 +01:00
ConfirmationDialog * members_dialog = nullptr ;
2021-07-13 07:52:52 +02:00
VisualShaderNode : : PortType members_input_port_type = VisualShaderNode : : PORT_TYPE_MAX ;
VisualShaderNode : : PortType members_output_port_type = VisualShaderNode : : PORT_TYPE_MAX ;
2022-01-07 08:59:15 +01:00
PopupMenu * popup_menu = nullptr ;
2024-01-19 11:35:46 +01:00
PopupMenu * connection_popup_menu = nullptr ;
2021-07-03 10:11:10 +02:00
PopupMenu * constants_submenu = nullptr ;
2022-01-07 08:59:15 +01:00
MenuButton * tools = nullptr ;
2019-02-22 18:38:58 +01:00
2022-01-09 15:02:13 +01:00
ConfirmationDialog * add_varying_dialog = nullptr ;
OptionButton * varying_type = nullptr ;
LineEdit * varying_name = nullptr ;
OptionButton * varying_mode = nullptr ;
Label * varying_error_label = nullptr ;
ConfirmationDialog * remove_varying_dialog = nullptr ;
Tree * varyings = nullptr ;
2024-02-07 02:37:26 +01:00
PopupPanel * frame_title_change_popup = nullptr ;
LineEdit * frame_title_change_edit = nullptr ;
2021-02-21 09:33:55 +01:00
2024-02-07 02:37:26 +01:00
PopupPanel * frame_tint_color_pick_popup = nullptr ;
ColorPicker * frame_tint_color_picker = nullptr ;
2021-02-21 09:33:55 +01:00
2021-01-10 11:33:14 +01:00
bool preview_first = true ;
bool preview_showed = false ;
2021-04-17 19:16:03 +02:00
enum ShaderModeFlags {
MODE_FLAGS_SPATIAL_CANVASITEM = 1 ,
MODE_FLAGS_SKY = 2 ,
2021-10-29 15:42:33 +02:00
MODE_FLAGS_PARTICLES = 4 ,
MODE_FLAGS_FOG = 8 ,
2021-04-17 19:16:03 +02:00
} ;
int mode = MODE_FLAGS_SPATIAL_CANVASITEM ;
2020-09-06 09:43:14 +02:00
enum TypeFlags {
TYPE_FLAGS_VERTEX = 1 ,
TYPE_FLAGS_FRAGMENT = 2 ,
TYPE_FLAGS_LIGHT = 4 ,
2020-09-07 09:28:13 +02:00
} ;
enum ParticlesTypeFlags {
TYPE_FLAGS_EMIT = 1 ,
TYPE_FLAGS_PROCESS = 2 ,
2020-09-15 09:57:40 +02:00
TYPE_FLAGS_COLLIDE = 4 ,
TYPE_FLAGS_EMIT_CUSTOM = 8 ,
TYPE_FLAGS_PROCESS_CUSTOM = 16 ,
2020-09-06 09:43:14 +02:00
} ;
2019-08-18 11:29:22 +02:00
2021-04-17 19:16:03 +02:00
enum SkyTypeFlags {
TYPE_FLAGS_SKY = 1 ,
} ;
2021-10-29 15:42:33 +02:00
enum FogTypeFlags {
TYPE_FLAGS_FOG = 1 ,
} ;
2019-02-22 18:38:58 +01:00
enum ToolsMenuOptions {
EXPAND_ALL ,
COLLAPSE_ALL
} ;
2023-11-17 20:44:38 +01:00
# ifdef MINGW_ENABLED
# undef DELETE
# endif
2020-02-27 10:10:09 +01:00
enum NodeMenuOptions {
ADD ,
SEPARATOR , // ignore
2021-10-30 15:12:14 +02:00
CUT ,
2020-02-27 10:10:09 +01:00
COPY ,
2020-02-27 11:55:41 +01:00
PASTE ,
2020-02-27 10:10:09 +01:00
DELETE ,
DUPLICATE ,
2021-10-30 15:12:14 +02:00
CLEAR_COPY_BUFFER ,
2020-12-30 09:45:31 +01:00
SEPARATOR2 , // ignore
2021-07-03 10:11:10 +02:00
FLOAT_CONSTANTS ,
2022-08-27 11:22:43 +02:00
CONVERT_CONSTANTS_TO_PARAMETERS ,
CONVERT_PARAMETERS_TO_CONSTANTS ,
2024-02-07 02:37:26 +01:00
UNLINK_FROM_PARENT_FRAME ,
2021-04-04 16:50:19 +02:00
SEPARATOR3 , // ignore
2024-02-07 02:37:26 +01:00
SET_FRAME_TITLE ,
ENABLE_FRAME_COLOR ,
SET_FRAME_COLOR ,
ENABLE_FRAME_AUTOSHRINK ,
2020-02-27 10:10:09 +01:00
} ;
2024-01-19 11:35:46 +01:00
enum ConnectionMenuOptions {
INSERT_NEW_NODE ,
DISCONNECT ,
} ;
2022-01-09 15:02:13 +01:00
enum class VaryingMenuOptions {
ADD ,
REMOVE ,
} ;
2022-01-07 08:59:15 +01:00
Tree * members = nullptr ;
AcceptDialog * alert = nullptr ;
LineEdit * node_filter = nullptr ;
RichTextLabel * node_desc = nullptr ;
Label * highend_label = nullptr ;
2019-02-22 18:38:58 +01:00
void _tools_menu_option ( int p_idx ) ;
2021-07-13 07:52:52 +02:00
void _show_members_dialog ( bool at_mouse_pos , VisualShaderNode : : PortType p_input_port_type = VisualShaderNode : : PORT_TYPE_MAX , VisualShaderNode : : PortType p_output_port_type = VisualShaderNode : : PORT_TYPE_MAX ) ;
2018-07-14 23:15:42 +02:00
2022-01-09 15:02:13 +01:00
void _varying_menu_id_pressed ( int p_idx ) ;
void _show_add_varying_dialog ( ) ;
void _show_remove_varying_dialog ( ) ;
2022-02-07 20:58:37 +01:00
void _update_nodes ( ) ;
2018-07-14 23:15:42 +02:00
void _update_graph ( ) ;
struct AddOption {
String name ;
String category ;
String type ;
2019-02-22 18:38:58 +01:00
String description ;
2022-01-28 12:36:35 +01:00
Vector < Variant > ops ;
2018-07-14 23:15:42 +02:00
Ref < Script > script ;
2020-11-24 10:12:55 +01:00
int mode = 0 ;
int return_type = 0 ;
int func = 0 ;
bool highend = false ;
bool is_custom = false ;
2023-01-04 12:30:20 +01:00
bool is_native = false ;
2020-11-24 10:12:55 +01:00
int temp_idx = 0 ;
2019-02-22 18:38:58 +01:00
2022-09-07 14:00:49 +02:00
AddOption ( const String & p_name = String ( ) , const String & p_category = String ( ) , const String & p_type = String ( ) , const String & p_description = String ( ) , const Vector < Variant > & p_ops = Vector < Variant > ( ) , int p_return_type = - 1 , int p_mode = - 1 , int p_func = - 1 , bool p_highend = false ) {
2019-02-22 18:38:58 +01:00
name = p_name ;
type = p_type ;
2022-09-07 14:00:49 +02:00
category = p_category ;
2019-02-22 18:38:58 +01:00
description = p_description ;
2022-01-28 12:36:35 +01:00
ops = p_ops ;
2019-02-22 18:38:58 +01:00
return_type = p_return_type ;
mode = p_mode ;
2019-05-01 16:04:39 +02:00
func = p_func ;
2019-07-12 16:36:23 +02:00
highend = p_highend ;
2018-07-14 23:15:42 +02:00
}
} ;
2020-02-14 13:11:55 +01:00
struct _OptionComparator {
_FORCE_INLINE_ bool operator ( ) ( const AddOption & a , const AddOption & b ) const {
return a . category . count ( " / " ) > b . category . count ( " / " ) | | ( a . category + " / " + a . name ) . naturalnocasecmp_to ( b . category + " / " + b . name ) < 0 ;
}
} ;
2018-07-14 23:15:42 +02:00
Vector < AddOption > add_options ;
2020-09-29 14:27:36 +02:00
int cubemap_node_option_idx ;
int texture2d_node_option_idx ;
int texture2d_array_node_option_idx ;
int texture3d_node_option_idx ;
2019-10-02 22:26:56 +02:00
int custom_node_option_idx ;
2020-10-04 11:11:58 +02:00
int curve_node_option_idx ;
2021-07-14 00:06:44 +02:00
int curve_xyz_node_option_idx ;
2019-05-12 14:09:39 +02:00
List < String > keyword_list ;
2018-07-14 23:15:42 +02:00
2022-08-27 11:22:43 +02:00
List < VisualShaderNodeParameterRef > uniform_refs ;
2020-09-21 13:32:59 +02:00
2023-08-09 18:31:15 +02:00
void _draw_color_over_button ( Object * p_obj , Color p_color ) ;
2018-07-14 23:15:42 +02:00
2022-01-28 12:36:35 +01:00
void _setup_node ( VisualShaderNode * p_node , const Vector < Variant > & p_ops ) ;
2024-02-15 17:25:58 +01:00
void _add_node ( int p_idx , const Vector < Variant > & p_ops , const String & p_resource_path = " " , int p_node_idx = - 1 ) ;
2022-01-09 15:02:13 +01:00
void _add_varying ( const String & p_name , VisualShader : : VaryingMode p_mode , VisualShader : : VaryingType p_type ) ;
void _remove_varying ( const String & p_name ) ;
2018-07-14 23:15:42 +02:00
void _update_options_menu ( ) ;
2020-09-12 10:01:48 +02:00
void _set_mode ( int p_which ) ;
2018-07-14 23:15:42 +02:00
2019-08-18 18:27:08 +02:00
void _show_preview_text ( ) ;
2021-01-10 11:33:14 +01:00
void _preview_close_requested ( ) ;
void _preview_size_changed ( ) ;
2019-08-18 11:29:22 +02:00
void _update_preview ( ) ;
2023-02-03 13:09:51 +01:00
void _update_next_previews ( int p_node_id ) ;
void _get_next_nodes_recursively ( VisualShader : : Type p_type , int p_node_id , LocalVector < int > & r_nodes ) const ;
2019-08-25 12:06:16 +02:00
String _get_description ( int p_idx ) ;
2019-08-18 11:29:22 +02:00
2024-02-07 02:37:26 +01:00
Vector < int > nodes_link_to_frame_buffer ; // Contains the nodes that are requested to be linked to a frame. This is used to perform one Undo/Redo operation for dragging nodes.
int frame_node_id_to_link_to = - 1 ;
2020-09-27 13:05:19 +02:00
struct DragOp {
2020-11-24 10:12:55 +01:00
VisualShader : : Type type = VisualShader : : Type : : TYPE_MAX ;
int node = 0 ;
2020-09-27 13:05:19 +02:00
Vector2 from ;
Vector2 to ;
} ;
List < DragOp > drag_buffer ;
2024-02-07 02:37:26 +01:00
2020-09-27 13:05:19 +02:00
bool drag_dirty = false ;
2018-07-14 23:15:42 +02:00
void _node_dragged ( const Vector2 & p_from , const Vector2 & p_to , int p_node ) ;
2020-09-27 13:05:19 +02:00
void _nodes_dragged ( ) ;
2022-01-07 08:59:15 +01:00
bool updating = false ;
2018-07-14 23:15:42 +02:00
void _connection_request ( const String & p_from , int p_from_index , const String & p_to , int p_to_index ) ;
void _disconnection_request ( const String & p_from , int p_from_index , const String & p_to , int p_to_index ) ;
void _scroll_changed ( const Vector2 & p_scroll ) ;
void _node_selected ( Object * p_node ) ;
2020-10-15 09:18:37 +02:00
void _delete_nodes ( int p_type , const List < int > & p_nodes ) ;
2023-09-26 16:41:39 +02:00
void _delete_node_request ( int p_type , int p_node ) ;
void _delete_nodes_request ( const TypedArray < StringName > & p_nodes ) ;
2018-07-14 23:15:42 +02:00
void _node_changed ( int p_id ) ;
2024-02-07 02:37:26 +01:00
void _nodes_linked_to_frame_request ( const TypedArray < StringName > & p_nodes , const StringName & p_frame ) ;
void _frame_rect_changed ( const GraphFrame * p_frame , const Rect2 & p_new_rect ) ;
2018-07-14 23:15:42 +02:00
void _edit_port_default_input ( Object * p_button , int p_node , int p_port ) ;
2022-08-01 20:01:48 +02:00
void _port_edited ( const StringName & p_property , const Variant & p_value , const String & p_field , bool p_changing ) ;
2018-07-14 23:15:42 +02:00
2022-01-07 08:59:15 +01:00
int to_node = - 1 ;
int to_slot = - 1 ;
int from_node = - 1 ;
int from_slot = - 1 ;
2019-06-26 20:50:38 +02:00
2024-01-19 11:35:46 +01:00
Ref < GraphEdit : : Connection > clicked_connection ;
bool connection_node_insert_requested = false ;
2022-05-19 17:00:06 +02:00
HashSet < int > selected_constants ;
2022-08-27 11:22:43 +02:00
HashSet < int > selected_parameters ;
2024-02-07 02:37:26 +01:00
int selected_frame = - 1 ;
2021-07-03 10:11:10 +02:00
int selected_float_constant = - 1 ;
2020-12-30 09:45:31 +01:00
2022-08-27 11:22:43 +02:00
void _convert_constants_to_parameters ( bool p_vice_versa ) ;
2024-02-07 02:37:26 +01:00
void _detach_nodes_from_frame_request ( ) ;
void _detach_nodes_from_frame ( int p_type , const List < int > & p_nodes ) ;
2020-12-30 09:45:31 +01:00
void _replace_node ( VisualShader : : Type p_type_id , int p_node_id , const StringName & p_from , const StringName & p_to ) ;
2024-02-15 17:25:58 +01:00
void _update_constant ( VisualShader : : Type p_type_id , int p_node_id , const Variant & p_var , int p_preview_port ) ;
void _update_parameter ( VisualShader : : Type p_type_id , int p_node_id , const Variant & p_var , int p_preview_port ) ;
2020-12-30 09:45:31 +01:00
2024-02-07 02:37:26 +01:00
void _unlink_node_from_parent_frame ( int p_node_id ) ;
2018-07-14 23:15:42 +02:00
void _connection_to_empty ( const String & p_from , int p_from_slot , const Vector2 & p_release_position ) ;
2019-06-26 20:50:38 +02:00
void _connection_from_empty ( const String & p_to , int p_to_slot , const Vector2 & p_release_position ) ;
2024-01-19 11:35:46 +01:00
bool _check_node_drop_on_connection ( const Vector2 & p_position , Ref < GraphEdit : : Connection > * r_closest_connection , int * r_node_id = nullptr , int * r_to_port = nullptr ) ;
void _handle_node_drop_on_connection ( ) ;
2018-07-14 23:15:42 +02:00
2024-02-07 02:37:26 +01:00
void _frame_title_popup_show ( const Point2 & p_position , int p_node_id ) ;
void _frame_title_popup_hide ( ) ;
void _frame_title_popup_focus_out ( ) ;
void _frame_title_text_changed ( const String & p_new_text ) ;
void _frame_title_text_submitted ( const String & p_new_text ) ;
void _frame_color_enabled_changed ( int p_node_id ) ;
void _frame_color_popup_show ( const Point2 & p_position , int p_node_id ) ;
void _frame_color_popup_hide ( ) ;
void _frame_color_confirm ( ) ;
void _frame_color_changed ( const Color & p_color ) ;
2021-02-21 09:33:55 +01:00
2024-02-07 02:37:26 +01:00
void _frame_autoshrink_enabled_changed ( int p_node_id ) ;
2021-02-21 09:33:55 +01:00
2022-08-27 11:22:43 +02:00
void _parameter_line_edit_changed ( const String & p_text , int p_node_id ) ;
2023-08-09 18:31:15 +02:00
void _parameter_line_edit_focus_out ( Object * p_line_edit , int p_node_id ) ;
2018-07-14 23:15:42 +02:00
2023-08-09 18:31:15 +02:00
void _port_name_focus_out ( Object * p_line_edit , int p_node_id , int p_port_id , bool p_output ) ;
2019-05-12 14:09:39 +02:00
2021-10-30 15:12:14 +02:00
struct CopyItem {
int id ;
Ref < VisualShaderNode > node ;
Vector2 position ;
Vector2 size ;
String group_inputs ;
String group_outputs ;
String expression ;
2022-02-07 20:58:37 +01:00
bool disabled = false ;
2021-10-30 15:12:14 +02:00
} ;
void _dup_copy_nodes ( int p_type , List < CopyItem > & r_nodes , List < VisualShader : : Connection > & r_connections ) ;
void _dup_paste_nodes ( int p_type , List < CopyItem > & r_items , const List < VisualShader : : Connection > & p_connections , const Vector2 & p_offset , bool p_duplicate ) ;
2019-07-12 19:36:33 +02:00
2018-07-14 23:15:42 +02:00
void _duplicate_nodes ( ) ;
2022-07-16 08:59:20 +02:00
static Vector2 selection_center ;
static List < CopyItem > copy_items_buffer ;
static List < VisualShader : : Connection > copy_connections_buffer ;
2019-07-12 19:36:33 +02:00
2021-10-30 15:12:14 +02:00
void _clear_copy_buffer ( ) ;
void _copy_nodes ( bool p_cut ) ;
2020-02-27 10:10:09 +01:00
void _paste_nodes ( bool p_use_custom_position = false , const Vector2 & p_custom_position = Vector2 ( ) ) ;
2019-07-12 19:36:33 +02:00
2020-03-17 07:33:00 +01:00
Vector < Ref < VisualShaderNodePlugin > > plugins ;
2020-09-09 16:40:27 +02:00
Ref < VisualShaderGraphPlugin > graph_plugin ;
2018-07-14 23:15:42 +02:00
void _mode_selected ( int p_id ) ;
2020-09-15 09:57:40 +02:00
void _custom_mode_toggled ( bool p_enabled ) ;
2018-07-14 23:15:42 +02:00
2024-02-15 17:25:58 +01:00
void _input_select_item ( Ref < VisualShaderNodeInput > p_input , const String & p_name ) ;
void _parameter_ref_select_item ( Ref < VisualShaderNodeParameterRef > p_parameter_ref , const String & p_name ) ;
void _varying_select_item ( Ref < VisualShaderNodeVarying > p_varying , const String & p_name ) ;
2018-07-14 23:15:42 +02:00
2021-07-03 10:11:10 +02:00
void _float_constant_selected ( int p_which ) ;
2020-10-01 10:10:51 +02:00
2020-09-07 09:28:13 +02:00
VisualShader : : Type get_current_shader_type ( ) const ;
2019-06-26 15:08:25 +02:00
void _add_input_port ( int p_node , int p_port , int p_port_type , const String & p_name ) ;
2019-05-12 14:09:39 +02:00
void _remove_input_port ( int p_node , int p_port ) ;
void _change_input_port_type ( int p_type , int p_node , int p_port ) ;
2020-12-02 10:49:46 +01:00
void _change_input_port_name ( const String & p_text , Object * p_line_edit , int p_node , int p_port ) ;
2019-05-12 14:09:39 +02:00
2019-06-26 15:08:25 +02:00
void _add_output_port ( int p_node , int p_port , int p_port_type , const String & p_name ) ;
2019-05-12 14:09:39 +02:00
void _remove_output_port ( int p_node , int p_port ) ;
void _change_output_port_type ( int p_type , int p_node , int p_port ) ;
2020-12-02 10:49:46 +01:00
void _change_output_port_name ( const String & p_text , Object * p_line_edit , int p_node , int p_port ) ;
2020-12-20 16:45:53 +01:00
void _expand_output_port ( int p_node , int p_port , bool p_expand ) ;
2019-05-12 14:09:39 +02:00
2023-08-09 18:31:15 +02:00
void _expression_focus_out ( Object * p_code_edit , int p_node ) ;
2019-05-12 14:09:39 +02:00
void _set_node_size ( int p_type , int p_node , const Size2 & p_size ) ;
void _node_resized ( const Vector2 & p_new_size , int p_type , int p_node ) ;
2018-07-14 23:15:42 +02:00
void _preview_select_port ( int p_node , int p_port ) ;
2019-07-10 11:54:12 +02:00
void _graph_gui_input ( const Ref < InputEvent > & p_event ) ;
2018-07-14 23:15:42 +02:00
2019-02-22 18:38:58 +01:00
void _member_filter_changed ( const String & p_text ) ;
2019-04-14 23:23:22 +02:00
void _sbox_input ( const Ref < InputEvent > & p_ie ) ;
2019-02-22 18:38:58 +01:00
void _member_selected ( ) ;
void _member_unselected ( ) ;
void _member_create ( ) ;
2019-06-26 20:50:38 +02:00
void _member_cancel ( ) ;
2019-02-22 18:38:58 +01:00
2022-01-09 15:02:13 +01:00
void _varying_create ( ) ;
2023-12-13 17:06:14 +01:00
void _varying_name_changed ( const String & p_name ) ;
2022-01-09 15:02:13 +01:00
void _varying_deleted ( ) ;
void _varying_selected ( ) ;
void _varying_unselected ( ) ;
void _update_varying_tree ( ) ;
2023-09-15 12:27:12 +02:00
void _set_custom_node_option ( int p_index , int p_node , int p_op ) ;
2020-02-27 10:10:09 +01:00
Vector2 menu_point ;
void _node_menu_id_pressed ( int p_idx ) ;
2024-01-19 11:35:46 +01:00
void _connection_menu_id_pressed ( int p_idx ) ;
2020-02-27 10:10:09 +01:00
2019-02-22 18:38:58 +01:00
Variant get_drag_data_fw ( const Point2 & p_point , Control * p_from ) ;
bool can_drop_data_fw ( const Point2 & p_point , const Variant & p_data , Control * p_from ) const ;
void drop_data_fw ( const Point2 & p_point , const Variant & p_data , Control * p_from ) ;
2019-06-26 15:08:25 +02:00
bool _is_available ( int p_mode ) ;
2022-08-27 11:22:43 +02:00
void _update_parameters ( bool p_update_refs ) ;
void _update_parameter_refs ( HashSet < String > & p_names ) ;
2022-01-09 15:02:13 +01:00
void _update_varyings ( ) ;
2019-02-22 18:38:58 +01:00
2022-12-07 19:50:28 +01:00
void _update_options_menu_deferred ( ) ;
void _rebuild_shader_deferred ( ) ;
2021-01-10 11:33:14 +01:00
void _visibility_changed ( ) ;
2023-01-23 10:01:28 +01:00
void _get_current_mode_limits ( int & r_begin_type , int & r_end_type ) const ;
void _update_custom_script ( const Ref < Script > & p_script ) ;
void _script_created ( const Ref < Script > & p_script ) ;
void _resource_saved ( const Ref < Resource > & p_resource ) ;
void _resource_removed ( const Ref < Resource > & p_resource ) ;
void _resources_removed ( ) ;
2018-07-14 23:15:42 +02:00
protected :
void _notification ( int p_what ) ;
static void _bind_methods ( ) ;
public :
void add_plugin ( const Ref < VisualShaderNodePlugin > & p_plugin ) ;
void remove_plugin ( const Ref < VisualShaderNodePlugin > & p_plugin ) ;
2020-09-10 07:31:07 +02:00
VisualShaderGraphPlugin * get_graph_plugin ( ) { return graph_plugin . ptr ( ) ; }
2018-07-14 23:15:42 +02:00
2019-08-09 21:51:48 +02:00
void clear_custom_types ( ) ;
2023-01-04 12:30:20 +01:00
void add_custom_type ( const String & p_name , const String & p_type , const Ref < Script > & p_script , const String & p_description , int p_return_icon_type , const String & p_category , bool p_highend ) ;
2018-07-14 23:15:42 +02:00
2022-12-07 19:50:28 +01:00
Dictionary get_custom_node_data ( Ref < VisualShaderNodeCustom > & p_custom_node ) ;
2023-01-04 12:30:20 +01:00
void update_custom_type ( const Ref < Resource > & p_resource ) ;
2022-12-07 19:50:28 +01:00
2020-07-10 12:34:39 +02:00
virtual Size2 get_minimum_size ( ) const override ;
2024-04-08 15:31:27 +02:00
2018-07-14 23:15:42 +02:00
void edit ( VisualShader * p_visual_shader ) ;
2024-04-08 15:31:27 +02:00
Ref < VisualShader > get_visual_shader ( ) const { return visual_shader ; }
2018-07-14 23:15:42 +02:00
VisualShaderEditor ( ) ;
} ;
class VisualShaderNodePluginDefault : public VisualShaderNodePlugin {
2019-03-19 19:35:57 +01:00
GDCLASS ( VisualShaderNodePluginDefault , VisualShaderNodePlugin ) ;
2018-07-14 23:15:42 +02:00
public :
2020-07-10 12:34:39 +02:00
virtual Control * create_editor ( const Ref < Resource > & p_parent_resource , const Ref < VisualShaderNode > & p_node ) override ;
2018-07-14 23:15:42 +02:00
} ;
2022-08-15 21:39:53 +02:00
class EditorPropertyVisualShaderMode : public EditorProperty {
GDCLASS ( EditorPropertyVisualShaderMode , EditorProperty ) ;
2022-01-07 08:59:15 +01:00
OptionButton * options = nullptr ;
2018-07-14 23:15:42 +02:00
void _option_selected ( int p_which ) ;
protected :
static void _bind_methods ( ) ;
public :
void setup ( const Vector < String > & p_options ) ;
2020-07-10 12:34:39 +02:00
virtual void update_property ( ) override ;
2018-07-14 23:15:42 +02:00
void set_option_button_clip ( bool p_enable ) ;
2022-08-15 21:39:53 +02:00
EditorPropertyVisualShaderMode ( ) ;
2018-07-14 23:15:42 +02:00
} ;
2022-08-15 21:39:53 +02:00
class EditorInspectorVisualShaderModePlugin : public EditorInspectorPlugin {
GDCLASS ( EditorInspectorVisualShaderModePlugin , EditorInspectorPlugin ) ;
2018-07-14 23:15:42 +02:00
public :
2020-07-10 12:34:39 +02:00
virtual bool can_handle ( Object * p_object ) override ;
2023-01-31 19:08:46 +01:00
virtual bool parse_property ( Object * p_object , const Variant : : Type p_type , const String & p_path , const PropertyHint p_hint , const String & p_hint_text , const BitField < PropertyUsageFlags > p_usage , const bool p_wide = false ) override ;
2018-07-14 23:15:42 +02:00
} ;
class VisualShaderNodePortPreview : public Control {
2019-03-19 19:35:57 +01:00
GDCLASS ( VisualShaderNodePortPreview , Control ) ;
2018-07-14 23:15:42 +02:00
Ref < VisualShader > shader ;
2020-11-24 10:12:55 +01:00
VisualShader : : Type type = VisualShader : : Type : : TYPE_MAX ;
int node = 0 ;
int port = 0 ;
2023-02-03 13:09:51 +01:00
bool is_valid = false ;
2018-07-14 23:15:42 +02:00
void _shader_changed ( ) ; //must regen
protected :
void _notification ( int p_what ) ;
static void _bind_methods ( ) ;
public :
2020-07-10 12:34:39 +02:00
virtual Size2 get_minimum_size ( ) const override ;
2023-02-03 13:09:51 +01:00
void setup ( const Ref < VisualShader > & p_shader , VisualShader : : Type p_type , int p_node , int p_port , bool p_is_valid ) ;
2018-07-14 23:15:42 +02:00
} ;
2019-06-28 13:51:51 +02:00
class VisualShaderConversionPlugin : public EditorResourceConversionPlugin {
GDCLASS ( VisualShaderConversionPlugin , EditorResourceConversionPlugin ) ;
public :
2020-07-10 12:34:39 +02:00
virtual String converts_to ( ) const override ;
virtual bool handles ( const Ref < Resource > & p_resource ) const override ;
virtual Ref < Resource > convert ( const Ref < Resource > & p_resource ) const override ;
2019-06-28 13:51:51 +02:00
} ;
2018-07-14 23:15:42 +02:00
# endif // VISUAL_SHADER_EDITOR_PLUGIN_H