godot/scene/gui/graph_edit.h

129 lines
2.7 KiB
C++

#ifndef GRAPH_EDIT_H
#define GRAPH_EDIT_H
#include "scene/gui/graph_node.h"
#include "scene/gui/scroll_bar.h"
#include "scene/gui/slider.h"
#include "scene/gui/tool_button.h"
#include "texture_frame.h"
class GraphEdit;
class GraphEditFilter : public Control {
OBJ_TYPE(GraphEditFilter,Control);
friend class GraphEdit;
GraphEdit *ge;
virtual bool has_point(const Point2& p_point) const;
public:
GraphEditFilter(GraphEdit *p_edit);
};
class GraphEdit : public Control {
OBJ_TYPE(GraphEdit,Control);
public:
struct Connection {
StringName from;
StringName to;
int from_port;
int to_port;
};
private:
ToolButton *zoom_minus;
ToolButton *zoom_reset;
ToolButton *zoom_plus;
void _zoom_minus();
void _zoom_reset();
void _zoom_plus();
HScrollBar* h_scroll;
VScrollBar* v_scroll;
bool connecting;
String connecting_from;
bool connecting_out;
int connecting_index;
int connecting_type;
Color connecting_color;
bool connecting_target;
Vector2 connecting_to;
String connecting_target_to;
int connecting_target_index;
bool dragging;
bool just_selected;
Vector2 drag_accum;
float zoom;
bool box_selecting;
bool box_selection_mode_aditive;
Point2 box_selecting_from;
Point2 box_selecting_to;
Rect2 box_selecting_rect;
List<GraphNode*> previus_selected;
bool right_disconnects;
bool updating;
List<Connection> connections;
void _draw_cos_line(const Vector2& p_from, const Vector2& p_to,const Color& p_color);
void _graph_node_raised(Node* p_gn);
void _graph_node_moved(Node *p_gn);
void _update_scroll();
void _scroll_moved(double);
void _input_event(const InputEvent& p_ev);
GraphEditFilter *top_layer;
void _top_layer_input(const InputEvent& p_ev);
void _top_layer_draw();
void _update_scroll_offset();
Array _get_connection_list() const;
friend class GraphEditFilter;
bool _filter_input(const Point2& p_point);
protected:
static void _bind_methods();
virtual void add_child_notify(Node *p_child);
virtual void remove_child_notify(Node *p_child);
void _notification(int p_what);
virtual bool clips_input() const;
public:
Error connect_node(const StringName& p_from, int p_from_port,const StringName& p_to,int p_to_port);
bool is_node_connected(const StringName& p_from, int p_from_port,const StringName& p_to,int p_to_port);
void disconnect_node(const StringName& p_from, int p_from_port,const StringName& p_to,int p_to_port);
void clear_connections();
void set_zoom(float p_zoom);
float get_zoom() const;
GraphEditFilter *get_top_layer() const { return top_layer; }
void get_connection_list(List<Connection> *r_connections) const;
void set_right_disconnects(bool p_enable);
bool is_right_disconnects_enabled() const;
Vector2 get_scroll_ofs() const;
GraphEdit();
};
#endif // GRAPHEdit_H