#include "MEM_guardedalloc.h"
#include "DNA_anim_types.h"
#include "DNA_node_types.h"
#include "BLI_blenlib.h"
#include "BLI_easing.h"
#include "BLI_math.h"
#include "BKE_anim_data.h"
#include "BKE_context.h"
#include "BKE_curve.h"
#include "BKE_lib_id.h"
#include "BKE_main.h"
#include "BKE_node.h"
#include "ED_node.h"
#include "ED_render.h"
#include "ED_screen.h"
#include "ED_util.h"
#include "RNA_access.h"
#include "RNA_define.h"
#include "WM_api.h"
#include "WM_types.h"
#include "UI_resources.h"
#include "UI_view2d.h"
#include "BLT_translation.h"
#include "node_intern.h"

Classes
struct	bNodeListItem

struct	NodeInsertOfsData

struct	LinkAndPosition

Macros
#define	NODE_JOIN_DONE 1

#define	NODE_JOIN_IS_DESCENDANT 2

#define	NODE_DETACH_DONE 1

#define	NODE_DETACH_IS_DESCENDANT 2

#define	NODE_INSOFS_ANIM_DURATION 0.25f

Typedefs
typedef struct bNodeListItem	bNodeListItem

typedef struct NodeInsertOfsData	NodeInsertOfsData

typedef struct LinkAndPosition	LinkAndPosition

Functions
static bool	ntree_has_drivers (bNodeTree *ntree)

static bool	ntree_check_nodes_connected_dfs (bNodeTree ntree, bNode from, bNode *to)

static bool	ntree_check_nodes_connected (bNodeTree ntree, bNode from, bNode *to)

static bool	node_group_has_output_dfs (bNode *node)

static bool	node_group_has_output (Main bmain, bNode node)

bool	node_connected_to_output (Main bmain, bNodeTree ntree, bNode *node)

static void	clear_picking_highlight (ListBase *links)

static LinkData *	create_drag_link (Main bmain, SpaceNode snode, bNode node, bNodeSocket sock)

static void	pick_link (const bContext C, wmOperator op, bNodeLinkDrag nldrag, SpaceNode snode, bNode node, bNodeLink link_to_pick)

static void	pick_input_link_by_link_intersect (const bContext C, wmOperator op, bNodeLinkDrag nldrag, const float cursor)

static int	sort_nodes_locx (const void a, const void b)

static bool	socket_is_available (bNodeTree UNUSED(ntree), bNodeSocket sock, const bool allow_used)

static bNodeSocket *	best_socket_output (bNodeTree ntree, bNode node, bNodeSocket *sock_target, const bool allow_multiple)

static bNodeSocket *	best_socket_input (bNodeTree ntree, bNode node, int num, int replace)

static bool	snode_autoconnect_input (SpaceNode snode, bNode node_fr, bNodeSocket sock_fr, bNode node_to, bNodeSocket *sock_to, int replace)

static int	compare_link_by_y_position (const void a, const void b)

void	sort_multi_input_socket_links (SpaceNode snode, bNode node, bNodeLink *drag_link, float cursor[2])

static void	snode_autoconnect (Main bmain, SpaceNode snode, const bool allow_multiple, const bool replace)

static int	node_link_viewer (const bContext C, bNode tonode)

static int	node_active_link_viewer_exec (bContext C, wmOperator UNUSED(op))

void	NODE_OT_link_viewer (wmOperatorType *ot)

static void	node_link_update_header (bContext C, bNodeLinkDrag UNUSED(nldrag))

static int	node_count_links (const bNodeTree ntree, const bNodeSocket socket)

static void	node_remove_extra_links (SpaceNode snode, bNodeLink link)

static void	node_link_exit (bContext C, wmOperator op, bool apply_links)

static void	node_link_find_socket (bContext C, wmOperator op, float cursor[2])

static int	node_link_modal (bContext C, wmOperator op, const wmEvent *event)

static bNodeLinkDrag *	node_link_init (Main bmain, SpaceNode snode, float cursor[2], bool detach)

static int	node_link_invoke (bContext C, wmOperator op, const wmEvent *event)

static void	node_link_cancel (bContext C, wmOperator op)

void	NODE_OT_link (wmOperatorType *ot)

static int	node_make_link_exec (bContext C, wmOperator op)

void	NODE_OT_link_make (wmOperatorType *ot)

static bool	node_links_intersect (bNodeLink *link, const float mcoords[][2], int tot)

static int	cut_links_exec (bContext C, wmOperator op)

void	NODE_OT_links_cut (wmOperatorType *ot)

static int	mute_links_exec (bContext C, wmOperator op)

void	NODE_OT_links_mute (wmOperatorType *ot)

static int	detach_links_exec (bContext C, wmOperator UNUSED(op))

void	NODE_OT_links_detach (wmOperatorType *ot)

static int	node_parent_set_exec (bContext C, wmOperator UNUSED(op))

void	NODE_OT_parent_set (wmOperatorType *ot)

static void	node_join_attach_recursive (bNode node, bNode frame)

static int	node_join_exec (bContext C, wmOperator UNUSED(op))

void	NODE_OT_join (wmOperatorType *ot)

static bNode *	node_find_frame_to_attach (ARegion region, const bNodeTree ntree, const int mouse_xy[2])

static int	node_attach_invoke (bContext C, wmOperator UNUSED(op), const wmEvent *event)

void	NODE_OT_attach (wmOperatorType *ot)

static void	node_detach_recursive (bNode *node)

static int	node_detach_exec (bContext C, wmOperator UNUSED(op))

void	NODE_OT_detach (wmOperatorType *ot)

static bool	ed_node_link_conditions (ScrArea area, bool test, SpaceNode r_snode, bNode *r_select)

void	ED_node_link_intersect_test (ScrArea *area, int test)

static bNodeSocket *	socket_best_match (ListBase *sockets)

static bool	node_parents_offset_flag_enable_cb (bNode parent, void UNUSED(userdata))

static void	node_offset_apply (bNode *node, const float offset_x)

static void	node_parent_offset_apply (NodeInsertOfsData data, bNode parent, const float offset_x)

static bool	node_link_insert_offset_frame_chain_cb (bNode fromnode, bNode tonode, void *userdata, const bool reversed)

static void	node_link_insert_offset_frame_chains (const bNodeTree ntree, const bNode parent, NodeInsertOfsData *data, const bool reversed)

static bool	node_link_insert_offset_chain_cb (bNode fromnode, bNode tonode, void *userdata, const bool reversed)

static void	node_link_insert_offset_ntree (NodeInsertOfsData iofsd, ARegion region, const int mouse_xy[2], const bool right_alignment)

static int	node_insert_offset_modal (bContext C, wmOperator UNUSED(op), const wmEvent *event)

static int	node_insert_offset_invoke (bContext C, wmOperator op, const wmEvent *event)

void	NODE_OT_insert_offset (wmOperatorType *ot)

void	ED_node_link_insert (Main bmain, ScrArea area)

Macro Definition Documentation

◆ NODE_DETACH_DONE

#define NODE_DETACH_DONE 1

Definition at line 1699 of file node_relationships.c.

◆ NODE_DETACH_IS_DESCENDANT

#define NODE_DETACH_IS_DESCENDANT 2

Definition at line 1700 of file node_relationships.c.

◆ NODE_INSOFS_ANIM_DURATION

#define NODE_INSOFS_ANIM_DURATION 0.25f

Definition at line 1948 of file node_relationships.c.

◆ NODE_JOIN_DONE

#define NODE_JOIN_DONE 1

Definition at line 1523 of file node_relationships.c.

◆ NODE_JOIN_IS_DESCENDANT

#define NODE_JOIN_IS_DESCENDANT 2

Definition at line 1524 of file node_relationships.c.

Typedef Documentation

◆ bNodeListItem

typedef struct bNodeListItem bNodeListItem

◆ LinkAndPosition

typedef struct LinkAndPosition LinkAndPosition

◆ NodeInsertOfsData

typedef struct NodeInsertOfsData NodeInsertOfsData

Function Documentation

◆ best_socket_input()

static bNodeSocket* best_socket_input	(	bNodeTree *	ntree,
		bNode *	node,
		int	num,
		int	replace
	)

static

Definition at line 385 of file node_relationships.c.

References Freestyle::a, LISTBASE_FOREACH, max_ii(), node, ntree, NULL, and socket_is_available().

Referenced by snode_autoconnect().

◆ best_socket_output()

static bNodeSocket* best_socket_output	(	bNodeTree *	ntree,
		bNode *	node,
		bNodeSocket *	sock_target,
		const bool	allow_multiple
	)

static

Definition at line 332 of file node_relationships.c.

References LISTBASE_FOREACH, bNodeSocket::name, node, NODE_REROUTE, ntree, NULL, SELECT, socket_is_available(), STREQ, and bNodeSocket::type.

Referenced by snode_autoconnect().

◆ clear_picking_highlight()

static void clear_picking_highlight ( ListBase * links )

static

Definition at line 183 of file node_relationships.c.

References LISTBASE_FOREACH, and NODE_LINK_TEMP_HIGHLIGHT.

Referenced by node_link_cancel(), node_link_modal(), pick_input_link_by_link_intersect(), and pick_link().

◆ compare_link_by_y_position()

static int compare_link_by_y_position	(	const void *	a,
		const void *	b
	)

static

Definition at line 438 of file node_relationships.c.

References Freestyle::a, BLI_assert, LinkAndPosition::link, LinkAndPosition::multi_socket_position, and bNodeLink::tosock.

Referenced by sort_multi_input_socket_links().

◆ create_drag_link()

static LinkData* create_drag_link	(	Main *	bmain,
		SpaceNode *	snode,
		bNode *	node,
		bNodeSocket *	sock
	)

static

Definition at line 190 of file node_relationships.c.

References LinkData::data, SpaceNode::edittree, bNodeLink::flag, bNodeLink::fromnode, bNodeLink::fromsock, bNodeSocket::in_out, MEM_callocN, node, node_connected_to_output(), NODE_LINK_TEST, NODE_LINK_VALID, SOCK_OUT, bNodeLink::tonode, and bNodeLink::tosock.

Referenced by node_link_init(), and pick_link().

◆ cut_links_exec()

static int cut_links_exec	(	bContext *	C,
		wmOperator *	op
	)

static

◆ detach_links_exec()

static int detach_links_exec	(	bContext *	C,
		wmOperator *	UNUSEDop
	)

static

Definition at line 1443 of file node_relationships.c.

References C, CTX_data_main(), CTX_wm_manager(), CTX_wm_space_node(), ED_preview_kill_jobs(), SpaceNode::edittree, LISTBASE_FOREACH, node, nodeInternalRelink(), bNodeTree::nodes, ntree, ntreeUpdateTree(), OPERATOR_FINISHED, SELECT, snode_dag_update(), and snode_notify().

Referenced by NODE_OT_links_detach().

◆ ed_node_link_conditions()

static bool ed_node_link_conditions	(	ScrArea *	area,
		bool	test,
		SpaceNode **	r_snode,
		bNode **	r_select
	)

static

Definition at line 1770 of file node_relationships.c.

References blender::compositor::area(), BLI_listbase_is_empty(), SpaceNode::edittree, ListBase::first, bNodeLink::fromnode, LinkAndPosition::link, bNodeTree::links, LISTBASE_FOREACH, node, nodeLinkIsHidden(), bNodeTree::nodes, NULL, select(), SELECT, SPACE_NODE, and bNodeLink::tonode.

Referenced by ED_node_link_insert(), and ED_node_link_intersect_test().

◆ ED_node_link_insert()

void ED_node_link_insert	(	Main *	bmain,
		ScrArea *	area
	)

◆ ED_node_link_intersect_test()

void ED_node_link_intersect_test	(	ScrArea *	area,
		int	test
	)

Definition at line 1826 of file node_relationships.c.

References blender::compositor::area(), BLI_rctf_isect_segment(), dist_squared_to_line_segment_v2(), ed_node_link_conditions(), SpaceNode::edittree, bNodeLink::flag, LinkAndPosition::link, bNodeTree::links, LISTBASE_FOREACH, min_ff(), node_link_bezier_points(), NODE_LINK_RESOL, NODE_LINKFLAG_HILITE, nodeLinkIsHidden(), NULL, and select().

Referenced by flushTransNodes(), and special_aftertrans_update__node().

◆ mute_links_exec()

static int mute_links_exec	(	bContext *	C,
		wmOperator *	op
	)

static

◆ node_active_link_viewer_exec()

static int node_active_link_viewer_exec	(	bContext *	C,
		wmOperator *	UNUSEDop
	)

static

Definition at line 709 of file node_relationships.c.

References C, CTX_data_main(), CTX_wm_manager(), CTX_wm_space_node(), ED_preview_kill_jobs(), SpaceNode::edittree, node, node_link_viewer(), nodeGetActive(), OPERATOR_CANCELLED, OPERATOR_FINISHED, and snode_notify().

Referenced by NODE_OT_link_viewer().

◆ node_attach_invoke()

static int node_attach_invoke	(	bContext *	C,
		wmOperator *	UNUSEDop,
		const wmEvent *	event
	)

static

Definition at line 1636 of file node_relationships.c.

References C, CTX_wm_region(), CTX_wm_space_node(), ED_node_sort(), SpaceNode::edittree, LISTBASE_FOREACH_BACKWARD, wmEvent::mval, NC_NODE, ND_DISPLAY, node, node_find_frame_to_attach(), NODE_SELECT, nodeAttachNode(), nodeAttachNodeCheck(), nodeDetachNode(), bNodeTree::nodes, ntree, NULL, OPERATOR_FINISHED, bNode::parent, and WM_event_add_notifier().

Referenced by NODE_OT_attach().

◆ node_connected_to_output()

bool node_connected_to_output	(	Main *	bmain,
		bNodeTree *	ntree,
		bNode *	node
	)

Definition at line 130 of file node_relationships.c.

References ELEM, LISTBASE_FOREACH, node, NODE_CUSTOM_GROUP, NODE_DO_OUTPUT, NODE_GROUP, node_group_has_output(), bNodeTree::nodes, ntree, ntree_check_nodes_connected(), ntree_has_drivers(), and NULL.

Referenced by create_drag_link(), cut_links_exec(), ED_node_tag_update_nodetree(), mute_links_exec(), node_delete_exec(), node_duplicate_exec(), node_link_exit(), node_link_init(), and node_mute_exec().

◆ node_count_links()

static int node_count_links	(	const bNodeTree *	ntree,
		const bNodeSocket *	socket
	)

static

Definition at line 754 of file node_relationships.c.

References count, ELEM, bNodeLink::fromsock, LinkAndPosition::link, bNodeTree::links, LISTBASE_FOREACH, ntree, and bNodeLink::tosock.

Referenced by node_remove_extra_links().

◆ node_detach_exec()

static int node_detach_exec	(	bContext *	C,
		wmOperator *	UNUSEDop
	)

static

Definition at line 1728 of file node_relationships.c.

References C, CTX_wm_space_node(), ED_node_sort(), SpaceNode::edittree, LISTBASE_FOREACH, NC_NODE, ND_DISPLAY, node, NODE_DETACH_DONE, node_detach_recursive(), bNodeTree::nodes, ntree, NULL, OPERATOR_FINISHED, and WM_event_add_notifier().

Referenced by NODE_OT_detach().

◆ node_detach_recursive()

static void node_detach_recursive ( bNode * node )

static

Definition at line 1702 of file node_relationships.c.

References node, NODE_DETACH_DONE, NODE_DETACH_IS_DESCENDANT, NODE_SELECT, and nodeDetachNode().

Referenced by node_detach_exec().

◆ node_find_frame_to_attach()

static bNode* node_find_frame_to_attach	(	ARegion *	region,
		const bNodeTree *	ntree,
		const int	mouse_xy[2]
	)

static

Definition at line 1615 of file node_relationships.c.

References BLI_rctf_isect_pt_v(), LISTBASE_FOREACH_BACKWARD, NODE_FRAME, NODE_SELECT, bNodeTree::nodes, ntree, NULL, UI_view2d_region_to_view(), UNPACK2, and ARegion::v2d.

Referenced by node_attach_invoke(), and node_link_insert_offset_ntree().

◆ node_group_has_output()

static bool node_group_has_output	(	Main *	bmain,
		bNode *	node
	)

static

Definition at line 119 of file node_relationships.c.

References BKE_main_id_tag_listbase(), BLI_assert, ELEM, if(), LIB_TAG_DOIT, node, NODE_CUSTOM_GROUP, NODE_GROUP, node_group_has_output_dfs(), Main::nodetrees, ntree, and NULL.

Referenced by node_connected_to_output().

◆ node_group_has_output_dfs()

static bool node_group_has_output_dfs ( bNode * node )

static

Definition at line 98 of file node_relationships.c.

References ListBase::first, bNodeTree::id, if(), LIB_TAG_DOIT, node, NODE_DO_OUTPUT, NODE_GROUP, NODE_GROUP_OUTPUT, bNodeTree::nodes, ntree, NULL, and ID::tag.

Referenced by node_group_has_output().

◆ node_insert_offset_invoke()

static int node_insert_offset_invoke	(	bContext *	C,
		wmOperator *	op,
		const wmEvent *	event
	)

static

◆ node_insert_offset_modal()

static int node_insert_offset_modal	(	bContext *	C,
		wmOperator *	UNUSEDop,
		const wmEvent *	event
	)

static

Modal handler for insert offset animation

Definition at line 2141 of file node_relationships.c.

References NodeInsertOfsData::anim_timer, BLI_easing_cubic_ease_in_out(), C, CLAMP_MAX, CLAMP_MIN, CTX_wm_manager(), CTX_wm_region(), CTX_wm_space_node(), wmEvent::customdata, wmTimer::duration, ED_region_tag_redraw(), SpaceNode::edittree, float(), SpaceNode_Runtime::iofsd, IS_EQF, LISTBASE_FOREACH, MEM_freeN, node, NODE_INSOFS_ANIM_DURATION, bNodeTree::nodes, NULL, OPERATOR_FINISHED, OPERATOR_PASS_THROUGH, OPERATOR_RUNNING_MODAL, SpaceNode::runtime, TIMER, wmEvent::type, UNLIKELY, and WM_event_remove_timer().

Referenced by NODE_OT_insert_offset().

◆ node_join_attach_recursive()

static void node_join_attach_recursive	(	bNode *	node,
		bNode *	frame
	)

static

Definition at line 1526 of file node_relationships.c.

References node, NODE_JOIN_DONE, NODE_JOIN_IS_DESCENDANT, NODE_TEST, nodeAttachNode(), and nodeDetachNode().

Referenced by node_join_exec().

◆ node_join_exec()

static int node_join_exec	(	bContext *	C,
		wmOperator *	UNUSEDop
	)

static

Definition at line 1556 of file node_relationships.c.

References C, CTX_wm_space_node(), ED_node_sort(), SpaceNode::edittree, LISTBASE_FOREACH, NC_NODE, ND_DISPLAY, node, node_add_node(), NODE_FRAME, node_join_attach_recursive(), NODE_JOIN_DONE, NODE_SELECT, NODE_TEST, bNodeTree::nodes, ntree, NULL, OPERATOR_FINISHED, and WM_event_add_notifier().

Referenced by NODE_OT_join().

◆ node_link_cancel()

static void node_link_cancel	(	bContext *	C,
		wmOperator *	op
	)

static

Definition at line 1129 of file node_relationships.c.

References BLI_freelistN(), BLI_remlink(), C, clear_picking_highlight(), CTX_wm_space_node(), wmOperator::customdata, SpaceNode::edittree, SpaceNode_Runtime::linkdrag, bNodeLinkDrag::links, bNodeTree::links, MEM_freeN, and SpaceNode::runtime.

Referenced by NODE_OT_link().

◆ node_link_exit()

static void node_link_exit	(	bContext *	C,
		wmOperator *	op,
		bool	apply_links
	)

static

◆ node_link_find_socket()

static void node_link_find_socket	(	bContext *	C,
		wmOperator *	op,
		float	cursor[2]
	)

static

◆ node_link_init()

static bNodeLinkDrag* node_link_init	(	Main *	bmain,
		SpaceNode *	snode,
		float	cursor[2],
		bool	detach
	)

static

◆ node_link_insert_offset_chain_cb()

static bool node_link_insert_offset_chain_cb	(	bNode *	fromnode,
		bNode *	tonode,
		void *	userdata,
		const bool	reversed
	)

static

Callback that applies NodeInsertOfsData.offset_x to a node or its parent, considering the logic needed for offsetting nodes after link insert

Definition at line 1991 of file node_relationships.c.

References data, bNode::flag, node, node_link_insert_offset_frame_chains(), node_offset_apply(), node_parent_offset_apply(), NODE_TEST, nodeFindRootParent(), nodeIsChildOf(), NULL, and bNode::parent.

Referenced by node_link_insert_offset_ntree().

◆ node_link_insert_offset_frame_chain_cb()

static bool node_link_insert_offset_frame_chain_cb	(	bNode *	fromnode,
		bNode *	tonode,
		void *	userdata,
		const bool	reversed
	)

static

Callback that applies NodeInsertOfsData.offset_x to a node or its parent, similar to node_link_insert_offset_output_chain_cb below, but with slightly different logic

Definition at line 1954 of file node_relationships.c.

References data, node_offset_apply(), and bNode::parent.

Referenced by node_link_insert_offset_frame_chains().