summaryrefslogtreecommitdiff
path: root/modules/visual_script/visual_script_editor.h
blob: 483ae1644cd4d70429184ea3f45caa42b899b901 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
#ifndef VisualSCRIPT_EDITOR_H
#define VisualSCRIPT_EDITOR_H

#include "tools/editor/plugins/script_editor_plugin.h"
#include "visual_script.h"
#include "tools/editor/property_editor.h"
#include "scene/gui/graph_edit.h"
#include "tools/editor/create_dialog.h"
#include "tools/editor/property_selector.h"
class VisualScriptEditorSignalEdit;
class VisualScriptEditorVariableEdit;


#ifdef TOOLS_ENABLED


class VisualScriptEditor : public ScriptEditorBase {
	OBJ_TYPE(VisualScriptEditor,ScriptEditorBase)

	enum {
		TYPE_SEQUENCE=1000,
		INDEX_BASE_SEQUENCE=1024


	};

	enum {
		EDIT_DELETE_NODES,
		EDIT_TOGGLE_BREAKPOINT,
		EDIT_FIND_NODE_TYPE,
		EDIT_COPY_NODES,
		EDIT_CUT_NODES,
		EDIT_PASTE_NODES,
	};

	enum PortAction {

		CREATE_CALL,
		CREATE_SET,
		CREATE_GET,
		CREATE_COND,
		CREATE_SEQUENCE,
		CREATE_SWITCH,
		CREATE_ITERATOR,
		CREATE_WHILE,
		CREATE_RETURN,
	};

	MenuButton *edit_menu;

	Ref<VisualScript> script;

	Button *base_type_select;

	HSplitContainer *main_hsplit;
	VSplitContainer *left_vsplit;

	GraphEdit *graph;

	LineEdit *node_filter;
	TextureFrame *node_filter_icon;

	VisualScriptEditorSignalEdit *signal_editor;

	AcceptDialog *edit_signal_dialog;
	PropertyEditor *edit_signal_edit;

	PropertySelector *method_select;
	PropertySelector *new_connect_node_select;

	VisualScriptEditorVariableEdit *variable_editor;

	AcceptDialog *edit_variable_dialog;
	PropertyEditor *edit_variable_edit;

	CustomPropertyEditor *default_value_edit;

	UndoRedo *undo_redo;

	Tree *members;
	Tree *nodes;

	Label *hint_text;
	Timer *hint_text_timer;

	Label *select_func_text;

	bool updating_graph;

	void _show_hint(const String& p_hint);
	void _hide_timer();

	CreateDialog *select_base_type;

	struct VirtualInMenu {
		String name;
		Variant::Type ret;
		bool ret_variant;
		Vector< Pair<Variant::Type,String> > args;
	};

	Map<int,VirtualInMenu> virtuals_in_menu;

	PopupMenu *new_function_menu;


	StringName edited_func;

	void _update_graph_connections();
	void _update_graph(int p_only_id=-1);

	bool updating_members;

	void _update_members();

	StringName selected;

	String _validate_name(const String& p_name) const;

	struct Clipboard {

		Map<int,Ref<VisualScriptNode> > nodes;
		Map<int,Vector2 > nodes_positions;

		Set<VisualScript::SequenceConnection> sequence_connections;
		Set<VisualScript::DataConnection> data_connections;
	};

	static Clipboard *clipboard;

	PopupMenu *port_action_popup;

	PortAction port_action;
	int port_action_node;
	int port_action_output;
	Vector2 port_action_pos;
	int port_action_new_node;
	void _port_action_menu(int p_option);
	void _selected_connect_node_method_or_setget(const String& p_text);


	int error_line;

	void _node_selected(Node* p_node);
	void _center_on_node(int p_id);

	void _node_filter_changed(const String& p_text);
	void _change_base_type_callback();
	void _change_base_type();
	void _member_selected();
	void _member_edited();
	void _override_pressed(int p_id);

	void _begin_node_move();
	void _end_node_move();
	void _move_node(String func,int p_id,const Vector2& p_to);

	void _node_moved(Vector2 p_from,Vector2 p_to, int p_id);
	void _remove_node(int p_id);
	void _graph_connected(const String& p_from,int p_from_slot,const String& p_to,int p_to_slot);
	void _graph_disconnected(const String& p_from,int p_from_slot,const String& p_to,int p_to_slot);
	void _graph_connect_to_empty(const String& p_from,int p_from_slot,const Vector2& p_release_pos);

	void _node_ports_changed(const String& p_func,int p_id);
	void _available_node_doubleclicked();

	void _update_available_nodes();

	void _member_button(Object *p_item, int p_column, int p_button);

	void _expression_text_changed(const String& p_text,int p_id);


	String revert_on_drag;

	void _input(const InputEvent& p_event);
	void _on_nodes_delete();
	void _on_nodes_duplicate();

	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);


	int editing_id;
	int editing_input;

	void _default_value_changed();
	void _default_value_edited(Node * p_button,int p_id,int p_input_port);

	void _menu_option(int p_what);

	void _graph_ofs_changed(const Vector2& p_ofs);
	void _comment_node_resized(const Vector2& p_new_size,int p_node);

	int selecting_method_id;
	void _selected_method(const String& p_method);

	void _draw_color_over_button(Object* obj,Color p_color);
	void _button_resource_previewed(const String& p_path,const Ref<Texture>& p_preview,Variant p_ud);

	VisualScriptNode::TypeGuess  _guess_output_type(int p_port_action_node,int p_port_action_output,Set<int> &visited_nodes);


protected:

	void _notification(int p_what);
	static void _bind_methods();
public:

	virtual void apply_code();
	virtual Ref<Script> get_edited_script() const;
	virtual Vector<String> get_functions();
	virtual void set_edited_script(const Ref<Script>& p_script);
	virtual void reload_text();
	virtual String get_name();
	virtual Ref<Texture> get_icon();
	virtual bool is_unsaved();
	virtual Variant get_edit_state();
	virtual void set_edit_state(const Variant& p_state);
	virtual void goto_line(int p_line,bool p_with_error=false);
	virtual void trim_trailing_whitespace();
	virtual void ensure_focus();
	virtual void tag_saved_version();
	virtual void reload(bool p_soft);
	virtual void get_breakpoints(List<int> *p_breakpoints);
	virtual bool goto_method(const String& p_method);
	virtual void add_callback(const String& p_function,StringArray p_args);
	virtual void update_settings();
	virtual void set_debugger_active(bool p_active);
	virtual void set_tooltip_request_func(String p_method,Object* p_obj);
	virtual Control *get_edit_menu();
	virtual bool can_lose_focus_on_node_selection() { return false; }

	static void register_editor();

	static void free_clipboard();

	VisualScriptEditor();
	~VisualScriptEditor();
};
#endif

#endif // VisualSCRIPT_EDITOR_H