diff options
Diffstat (limited to 'source/blender/editors/space_node/node_intern.hh')
-rw-r--r-- | source/blender/editors/space_node/node_intern.hh | 347 |
1 files changed, 347 insertions, 0 deletions
diff --git a/source/blender/editors/space_node/node_intern.hh b/source/blender/editors/space_node/node_intern.hh new file mode 100644 index 00000000000..cb2980b7db5 --- /dev/null +++ b/source/blender/editors/space_node/node_intern.hh @@ -0,0 +1,347 @@ +/* + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * + * The Original Code is Copyright (C) 2008 Blender Foundation. + * All rights reserved. + */ + +/** \file + * \ingroup spnode + */ + +#pragma once + +#include "BKE_node.h" +#include "UI_interface.h" +#include "UI_view2d.h" + +#include "BLI_vector.hh" +#include "UI_interface.hh" + +#include <stddef.h> /* for size_t */ + +/* internal exports only */ + +struct ARegion; +struct ARegionType; +struct Main; +struct NodeInsertOfsData; +struct View2D; +struct bContext; +struct bNode; +struct bNodeLink; +struct bNodeSocket; +struct wmGizmoGroupType; +struct wmKeyConfig; +struct wmWindow; + +/* temp data to pass on to modal */ +typedef struct bNodeLinkDrag { + struct bNodeLinkDrag *next, *prev; + + /* List of links dragged by the operator. + * NOTE: This is a list of LinkData structs on top of the actual bNodeLinks. + * This way the links can be added to the node tree while being stored in this list. + */ + ListBase links; + bool from_multi_input_socket; + int in_out; + + /** Temporarily stores the last picked link from multi-input socket operator. */ + struct bNodeLink *last_picked_multi_input_socket_link; + + /** Temporarily stores the last hovered socket for multi-input socket operator. + * Store it to recalculate sorting after it is no longer hovered. */ + struct bNode *last_node_hovered_while_dragging_a_link; + + /* Data for edge panning */ + View2DEdgePanData pan_data; +} bNodeLinkDrag; + +typedef struct SpaceNode_Runtime { + float aspect; + + /** Mouse position for drawing socket-less links and adding nodes. */ + float cursor[2]; + + /** For auto compositing. */ + bool recalc; + + /** Temporary data for modal linking operator. */ + struct ListBase linkdrag; + + /* XXX hack for translate_attach op-macros to pass data from transform op to insert_offset op */ + /** Temporary data for node insert offset (in UI called Auto-offset). */ + struct NodeInsertOfsData *iofsd; +} SpaceNode_Runtime; + +/* space_node.c */ + +/* transform between View2Ds in the tree path */ +void space_node_group_offset(struct SpaceNode *snode, float *x, float *y); + +/* node_draw.cc */ +float node_socket_calculate_height(const bNodeSocket *socket); +void node_link_calculate_multi_input_position(const float socket_x, + const float socket_y, + const int index, + const int total_inputs, + float r[2]); + +int node_get_colorid(struct bNode *node); +int node_get_resize_cursor(int directions); +void node_draw_shadow(const struct SpaceNode *snode, + const struct bNode *node, + float radius, + float alpha); +void node_draw_default(const struct bContext *C, + struct ARegion *region, + struct SpaceNode *snode, + struct bNodeTree *ntree, + struct bNode *node, + bNodeInstanceKey key); +void node_draw_sockets(const struct View2D *v2d, + const struct bContext *C, + struct bNodeTree *ntree, + struct bNode *node, + bool draw_outputs, + bool select_all); +void node_update_default(const struct bContext *C, struct bNodeTree *ntree, struct bNode *node); +int node_select_area_default(struct bNode *node, int x, int y); +int node_tweak_area_default(struct bNode *node, int x, int y); +void node_socket_color_get(const struct bContext *C, + struct bNodeTree *ntree, + struct PointerRNA *node_ptr, + struct bNodeSocket *sock, + float r_color[4]); +void node_update_nodetree(const struct bContext *C, struct bNodeTree *ntree); +void node_draw_nodetree(const struct bContext *C, + struct ARegion *region, + struct SpaceNode *snode, + struct bNodeTree *ntree, + bNodeInstanceKey parent_key); +void node_draw_space(const bContext *C, ARegion *region); + +void node_set_cursor(struct wmWindow *win, struct SpaceNode *snode, float cursor[2]); +/* DPI scaled coords */ +void node_to_view(const struct bNode *node, float x, float y, float *rx, float *ry); +void node_to_updated_rect(const struct bNode *node, rctf *r_rect); +void node_from_view(const struct bNode *node, float x, float y, float *rx, float *ry); + +/* node_toolbar.c */ +void node_toolbar_register(struct ARegionType *art); + +/* node_ops.c */ +void node_operatortypes(void); +void node_keymap(struct wmKeyConfig *keyconf); + +/* node_select.c */ +void node_deselect_all(struct SpaceNode *snode); +void node_socket_select(struct bNode *node, struct bNodeSocket *sock); +void node_socket_deselect(struct bNode *node, struct bNodeSocket *sock, const bool deselect_node); +void node_deselect_all_input_sockets(struct SpaceNode *snode, const bool deselect_nodes); +void node_deselect_all_output_sockets(struct SpaceNode *snode, const bool deselect_nodes); +void node_select_single(struct bContext *C, struct bNode *node); + +void NODE_OT_select(struct wmOperatorType *ot); +void NODE_OT_select_all(struct wmOperatorType *ot); +void NODE_OT_select_linked_to(struct wmOperatorType *ot); +void NODE_OT_select_linked_from(struct wmOperatorType *ot); +void NODE_OT_select_box(struct wmOperatorType *ot); +void NODE_OT_select_circle(struct wmOperatorType *ot); +void NODE_OT_select_lasso(struct wmOperatorType *ot); +void NODE_OT_select_grouped(struct wmOperatorType *ot); +void NODE_OT_select_same_type_step(struct wmOperatorType *ot); +void NODE_OT_find_node(struct wmOperatorType *ot); + +/* node_view.c */ +int space_node_view_flag(struct bContext *C, + struct SpaceNode *snode, + ARegion *region, + const int node_flag, + const int smooth_viewtx); + +void NODE_OT_view_all(struct wmOperatorType *ot); +void NODE_OT_view_selected(struct wmOperatorType *ot); +void NODE_OT_geometry_node_view_legacy(struct wmOperatorType *ot); + +void NODE_OT_backimage_move(struct wmOperatorType *ot); +void NODE_OT_backimage_zoom(struct wmOperatorType *ot); +void NODE_OT_backimage_fit(struct wmOperatorType *ot); +void NODE_OT_backimage_sample(struct wmOperatorType *ot); + +/* drawnode.c */ +void nodelink_batch_start(struct SpaceNode *snode); +void nodelink_batch_end(struct SpaceNode *snode); + +void node_draw_link(const struct bContext *C, + struct View2D *v2d, + struct SpaceNode *snode, + struct bNodeLink *link); +void node_draw_link_bezier(const struct bContext *C, + const struct View2D *v2d, + const struct SpaceNode *snode, + const struct bNodeLink *link, + int th_col1, + int th_col2, + int th_col3); +bool node_link_bezier_points(const struct View2D *v2d, + const struct SpaceNode *snode, + const struct bNodeLink *link, + float coord_array[][2], + const int resol); +bool node_link_bezier_handles(const struct View2D *v2d, + const struct SpaceNode *snode, + const struct bNodeLink *link, + float vec[4][2]); +void draw_nodespace_back_pix(const struct bContext *C, + struct ARegion *region, + struct SpaceNode *snode, + bNodeInstanceKey parent_key); + +/* node_add.c */ +bNode *node_add_node( + const struct bContext *C, const char *idname, int type, float locx, float locy); +void NODE_OT_add_reroute(struct wmOperatorType *ot); +void NODE_OT_add_group(struct wmOperatorType *ot); +void NODE_OT_add_object(struct wmOperatorType *ot); +void NODE_OT_add_collection(struct wmOperatorType *ot); +void NODE_OT_add_texture(struct wmOperatorType *ot); +void NODE_OT_add_file(struct wmOperatorType *ot); +void NODE_OT_add_mask(struct wmOperatorType *ot); +void NODE_OT_new_node_tree(struct wmOperatorType *ot); + +/* node_group.c */ +const char *node_group_idname(struct bContext *C); +void NODE_OT_group_make(struct wmOperatorType *ot); +void NODE_OT_group_insert(struct wmOperatorType *ot); +void NODE_OT_group_ungroup(struct wmOperatorType *ot); +void NODE_OT_group_separate(struct wmOperatorType *ot); +void NODE_OT_group_edit(struct wmOperatorType *ot); + +/* node_relationships.c */ +void sort_multi_input_socket_links(struct SpaceNode *snode, + struct bNode *node, + struct bNodeLink *drag_link, + float cursor[2]); +bool node_connected_to_output(struct Main *bmain, struct bNodeTree *ntree, struct bNode *node); + +void NODE_OT_link(struct wmOperatorType *ot); +void NODE_OT_link_make(struct wmOperatorType *ot); +void NODE_OT_links_cut(struct wmOperatorType *ot); +void NODE_OT_links_detach(struct wmOperatorType *ot); +void NODE_OT_links_mute(struct wmOperatorType *ot); + +void NODE_OT_parent_set(struct wmOperatorType *ot); +void NODE_OT_join(struct wmOperatorType *ot); +void NODE_OT_attach(struct wmOperatorType *ot); +void NODE_OT_detach(struct wmOperatorType *ot); + +void NODE_OT_link_viewer(struct wmOperatorType *ot); + +void NODE_OT_insert_offset(struct wmOperatorType *ot); + +/* node_edit.c */ +void snode_notify(struct bContext *C, struct SpaceNode *snode); +void snode_dag_update(struct bContext *C, struct SpaceNode *snode); +void snode_set_context(const struct bContext *C); + +void snode_update(struct SpaceNode *snode, struct bNode *node); +bool composite_node_active(struct bContext *C); +bool composite_node_editable(struct bContext *C); + +bool node_has_hidden_sockets(struct bNode *node); +void node_set_hidden_sockets(struct SpaceNode *snode, bNode *node, int set); +int node_render_changed_exec(bContext *, struct wmOperator *); +int node_find_indicated_socket(struct SpaceNode *snode, + struct bNode **nodep, + struct bNodeSocket **sockp, + const float cursor[2], + int in_out); +float node_link_dim_factor(const struct View2D *v2d, const struct bNodeLink *link); +bool node_link_is_hidden_or_dimmed(const struct View2D *v2d, const struct bNodeLink *link); + +void NODE_OT_duplicate(struct wmOperatorType *ot); +void NODE_OT_delete(struct wmOperatorType *ot); +void NODE_OT_delete_reconnect(struct wmOperatorType *ot); +void NODE_OT_resize(struct wmOperatorType *ot); + +void NODE_OT_mute_toggle(struct wmOperatorType *ot); +void NODE_OT_hide_toggle(struct wmOperatorType *ot); +void NODE_OT_hide_socket_toggle(struct wmOperatorType *ot); +void NODE_OT_preview_toggle(struct wmOperatorType *ot); +void NODE_OT_options_toggle(struct wmOperatorType *ot); +void NODE_OT_node_copy_color(struct wmOperatorType *ot); + +void NODE_OT_read_viewlayers(struct wmOperatorType *ot); +void NODE_OT_render_changed(struct wmOperatorType *ot); + +void NODE_OT_output_file_add_socket(struct wmOperatorType *ot); +void NODE_OT_output_file_remove_active_socket(struct wmOperatorType *ot); +void NODE_OT_output_file_move_active_socket(struct wmOperatorType *ot); + +void NODE_OT_switch_view_update(struct wmOperatorType *ot); + +/* NOTE: clipboard_cut is a simple macro of copy + delete. */ +void NODE_OT_clipboard_copy(struct wmOperatorType *ot); +void NODE_OT_clipboard_paste(struct wmOperatorType *ot); + +void NODE_OT_tree_socket_add(struct wmOperatorType *ot); +void NODE_OT_tree_socket_remove(struct wmOperatorType *ot); +void NODE_OT_tree_socket_change_type(struct wmOperatorType *ot); +void NODE_OT_tree_socket_move(struct wmOperatorType *ot); + +void NODE_OT_shader_script_update(struct wmOperatorType *ot); + +void NODE_OT_viewer_border(struct wmOperatorType *ot); +void NODE_OT_clear_viewer_border(struct wmOperatorType *ot); + +/* node_widgets.c */ +void NODE_GGT_backdrop_transform(struct wmGizmoGroupType *gzgt); +void NODE_GGT_backdrop_crop(struct wmGizmoGroupType *gzgt); +void NODE_GGT_backdrop_sun_beams(struct wmGizmoGroupType *gzgt); +void NODE_GGT_backdrop_corner_pin(struct wmGizmoGroupType *gzgt); + +void NODE_OT_cryptomatte_layer_add(struct wmOperatorType *ot); +void NODE_OT_cryptomatte_layer_remove(struct wmOperatorType *ot); + +/* node_geometry_attribute_search.cc */ +void node_geometry_add_attribute_search_button(const struct bContext *C, + const struct bNodeTree *node_tree, + const struct bNode *node, + struct PointerRNA *socket_ptr, + struct uiLayout *layout); + +extern const char *node_context_dir[]; + +/* XXXXXX */ + +/* Nodes draw without dpi - the view zoom is flexible. */ +#define HIDDEN_RAD (0.75f * U.widget_unit) +#define BASIS_RAD (0.2f * U.widget_unit) +#define NODE_DYS (U.widget_unit / 2) +#define NODE_DY U.widget_unit +#define NODE_SOCKDY (0.1f * U.widget_unit) +#define NODE_WIDTH(node) (node->width * UI_DPI_FAC) +#define NODE_HEIGHT(node) (node->height * UI_DPI_FAC) +#define NODE_MARGIN_X (1.2f * U.widget_unit) +#define NODE_SOCKSIZE (0.25f * U.widget_unit) +#define NODE_MULTI_INPUT_LINK_GAP (0.25f * U.widget_unit) +#define NODE_RESIZE_MARGIN (0.20f * U.widget_unit) +#define NODE_LINK_RESOL 12 + +namespace blender::ed::space_node { +Vector<ui::ContextPathItem> context_path_for_space_node(const bContext &C); +} |