diff options
Diffstat (limited to 'source/blender/editors/space_node/space_node.c')
-rw-r--r-- | source/blender/editors/space_node/space_node.c | 1558 |
1 files changed, 780 insertions, 778 deletions
diff --git a/source/blender/editors/space_node/space_node.c b/source/blender/editors/space_node/space_node.c index 22564061aed..52dab2a44a8 100644 --- a/source/blender/editors/space_node/space_node.c +++ b/source/blender/editors/space_node/space_node.c @@ -53,1030 +53,1032 @@ #include "WM_api.h" #include "WM_types.h" -#include "node_intern.h" /* own include */ - +#include "node_intern.h" /* own include */ /* ******************** tree path ********************* */ void ED_node_tree_start(SpaceNode *snode, bNodeTree *ntree, ID *id, ID *from) { - bNodeTreePath *path, *path_next; - for (path = snode->treepath.first; path; path = path_next) { - path_next = path->next; - MEM_freeN(path); - } - BLI_listbase_clear(&snode->treepath); + bNodeTreePath *path, *path_next; + for (path = snode->treepath.first; path; path = path_next) { + path_next = path->next; + MEM_freeN(path); + } + BLI_listbase_clear(&snode->treepath); - if (ntree) { - path = MEM_callocN(sizeof(bNodeTreePath), "node tree path"); - path->nodetree = ntree; - path->parent_key = NODE_INSTANCE_KEY_BASE; + if (ntree) { + path = MEM_callocN(sizeof(bNodeTreePath), "node tree path"); + path->nodetree = ntree; + path->parent_key = NODE_INSTANCE_KEY_BASE; - /* copy initial offset from bNodeTree */ - copy_v2_v2(path->view_center, ntree->view_center); + /* copy initial offset from bNodeTree */ + copy_v2_v2(path->view_center, ntree->view_center); - if (id) { - BLI_strncpy(path->node_name, id->name + 2, sizeof(path->node_name)); - } + if (id) { + BLI_strncpy(path->node_name, id->name + 2, sizeof(path->node_name)); + } - BLI_addtail(&snode->treepath, path); + BLI_addtail(&snode->treepath, path); - id_us_ensure_real(&ntree->id); - } + id_us_ensure_real(&ntree->id); + } - /* update current tree */ - snode->nodetree = snode->edittree = ntree; - snode->id = id; - snode->from = from; + /* update current tree */ + snode->nodetree = snode->edittree = ntree; + snode->id = id; + snode->from = from; - ED_node_set_active_viewer_key(snode); + ED_node_set_active_viewer_key(snode); - WM_main_add_notifier(NC_SCENE | ND_NODES, NULL); + WM_main_add_notifier(NC_SCENE | ND_NODES, NULL); } void ED_node_tree_push(SpaceNode *snode, bNodeTree *ntree, bNode *gnode) { - bNodeTreePath *path = MEM_callocN(sizeof(bNodeTreePath), "node tree path"); - bNodeTreePath *prev_path = snode->treepath.last; - path->nodetree = ntree; - if (gnode) { - if (prev_path) { - path->parent_key = BKE_node_instance_key(prev_path->parent_key, prev_path->nodetree, gnode); - } - else { - path->parent_key = NODE_INSTANCE_KEY_BASE; - } + bNodeTreePath *path = MEM_callocN(sizeof(bNodeTreePath), "node tree path"); + bNodeTreePath *prev_path = snode->treepath.last; + path->nodetree = ntree; + if (gnode) { + if (prev_path) { + path->parent_key = BKE_node_instance_key(prev_path->parent_key, prev_path->nodetree, gnode); + } + else { + path->parent_key = NODE_INSTANCE_KEY_BASE; + } - BLI_strncpy(path->node_name, gnode->name, sizeof(path->node_name)); - } - else { - path->parent_key = NODE_INSTANCE_KEY_BASE; - } + BLI_strncpy(path->node_name, gnode->name, sizeof(path->node_name)); + } + else { + path->parent_key = NODE_INSTANCE_KEY_BASE; + } - /* copy initial offset from bNodeTree */ - copy_v2_v2(path->view_center, ntree->view_center); + /* copy initial offset from bNodeTree */ + copy_v2_v2(path->view_center, ntree->view_center); - BLI_addtail(&snode->treepath, path); + BLI_addtail(&snode->treepath, path); - id_us_ensure_real(&ntree->id); + id_us_ensure_real(&ntree->id); - /* update current tree */ - snode->edittree = ntree; + /* update current tree */ + snode->edittree = ntree; - ED_node_set_active_viewer_key(snode); + ED_node_set_active_viewer_key(snode); - WM_main_add_notifier(NC_SCENE | ND_NODES, NULL); + WM_main_add_notifier(NC_SCENE | ND_NODES, NULL); } void ED_node_tree_pop(SpaceNode *snode) { - bNodeTreePath *path = snode->treepath.last; + bNodeTreePath *path = snode->treepath.last; - /* don't remove root */ - if (path == snode->treepath.first) { - return; - } + /* don't remove root */ + if (path == snode->treepath.first) { + return; + } - BLI_remlink(&snode->treepath, path); - MEM_freeN(path); + BLI_remlink(&snode->treepath, path); + MEM_freeN(path); - /* update current tree */ - path = snode->treepath.last; - snode->edittree = path->nodetree; + /* update current tree */ + path = snode->treepath.last; + snode->edittree = path->nodetree; - ED_node_set_active_viewer_key(snode); + ED_node_set_active_viewer_key(snode); - /* listener updates the View2D center from edittree */ - WM_main_add_notifier(NC_SCENE | ND_NODES, NULL); + /* listener updates the View2D center from edittree */ + WM_main_add_notifier(NC_SCENE | ND_NODES, NULL); } int ED_node_tree_depth(SpaceNode *snode) { - return BLI_listbase_count(&snode->treepath); + return BLI_listbase_count(&snode->treepath); } bNodeTree *ED_node_tree_get(SpaceNode *snode, int level) { - bNodeTreePath *path; - int i; - for (path = snode->treepath.last, i = 0; path; path = path->prev, ++i) { - if (i == level) { - return path->nodetree; - } - } - return NULL; + bNodeTreePath *path; + int i; + for (path = snode->treepath.last, i = 0; path; path = path->prev, ++i) { + if (i == level) { + return path->nodetree; + } + } + return NULL; } int ED_node_tree_path_length(SpaceNode *snode) { - bNodeTreePath *path; - int length = 0; - int i; - for (path = snode->treepath.first, i = 0; path; path = path->next, ++i) { - length += strlen(path->node_name); - if (i > 0) { - length += 1; /* for separator char */ - } - } - return length; + bNodeTreePath *path; + int length = 0; + int i; + for (path = snode->treepath.first, i = 0; path; path = path->next, ++i) { + length += strlen(path->node_name); + if (i > 0) { + length += 1; /* for separator char */ + } + } + return length; } void ED_node_tree_path_get(SpaceNode *snode, char *value) { - bNodeTreePath *path; - int i; - - value[0] = '\0'; - for (path = snode->treepath.first, i = 0; path; path = path->next, ++i) { - if (i == 0) { - strcpy(value, path->node_name); - value += strlen(path->node_name); - } - else { - sprintf(value, "/%s", path->node_name); - value += strlen(path->node_name) + 1; - } - } + bNodeTreePath *path; + int i; + + value[0] = '\0'; + for (path = snode->treepath.first, i = 0; path; path = path->next, ++i) { + if (i == 0) { + strcpy(value, path->node_name); + value += strlen(path->node_name); + } + else { + sprintf(value, "/%s", path->node_name); + value += strlen(path->node_name) + 1; + } + } } void ED_node_tree_path_get_fixedbuf(SpaceNode *snode, char *value, int max_length) { - bNodeTreePath *path; - int size, i; - - value[0] = '\0'; - for (path = snode->treepath.first, i = 0; path; path = path->next, ++i) { - if (i == 0) { - size = BLI_strncpy_rlen(value, path->node_name, max_length); - } - else { - size = BLI_snprintf_rlen(value, max_length, "/%s", path->node_name); - } - max_length -= size; - if (max_length <= 0) { - break; - } - value += size; - } + bNodeTreePath *path; + int size, i; + + value[0] = '\0'; + for (path = snode->treepath.first, i = 0; path; path = path->next, ++i) { + if (i == 0) { + size = BLI_strncpy_rlen(value, path->node_name, max_length); + } + else { + size = BLI_snprintf_rlen(value, max_length, "/%s", path->node_name); + } + max_length -= size; + if (max_length <= 0) { + break; + } + value += size; + } } void ED_node_set_active_viewer_key(SpaceNode *snode) { - bNodeTreePath *path = snode->treepath.last; - if (snode->nodetree && path) { - snode->nodetree->active_viewer_key = path->parent_key; - } + bNodeTreePath *path = snode->treepath.last; + if (snode->nodetree && path) { + snode->nodetree->active_viewer_key = path->parent_key; + } } void snode_group_offset(SpaceNode *snode, float *x, float *y) { - bNodeTreePath *path = snode->treepath.last; - - if (path && path->prev) { - float dcenter[2]; - sub_v2_v2v2(dcenter, path->view_center, path->prev->view_center); - *x = dcenter[0]; - *y = dcenter[1]; - } - else { - *x = *y = 0.0f; - } + bNodeTreePath *path = snode->treepath.last; + + if (path && path->prev) { + float dcenter[2]; + sub_v2_v2v2(dcenter, path->view_center, path->prev->view_center); + *x = dcenter[0]; + *y = dcenter[1]; + } + else { + *x = *y = 0.0f; + } } /* ******************** manage regions ********************* */ ARegion *node_has_buttons_region(ScrArea *sa) { - ARegion *ar, *arnew; + ARegion *ar, *arnew; - ar = BKE_area_find_region_type(sa, RGN_TYPE_UI); - if (ar) { - return ar; - } + ar = BKE_area_find_region_type(sa, RGN_TYPE_UI); + if (ar) { + return ar; + } - /* add subdiv level; after header */ - ar = BKE_area_find_region_type(sa, RGN_TYPE_HEADER); + /* add subdiv level; after header */ + ar = BKE_area_find_region_type(sa, RGN_TYPE_HEADER); - /* is error! */ - if (ar == NULL) { - return NULL; - } + /* is error! */ + if (ar == NULL) { + return NULL; + } - arnew = MEM_callocN(sizeof(ARegion), "buttons for node"); + arnew = MEM_callocN(sizeof(ARegion), "buttons for node"); - BLI_insertlinkafter(&sa->regionbase, ar, arnew); - arnew->regiontype = RGN_TYPE_UI; - arnew->alignment = RGN_ALIGN_RIGHT; + BLI_insertlinkafter(&sa->regionbase, ar, arnew); + arnew->regiontype = RGN_TYPE_UI; + arnew->alignment = RGN_ALIGN_RIGHT; - arnew->flag = RGN_FLAG_HIDDEN; + arnew->flag = RGN_FLAG_HIDDEN; - return arnew; + return arnew; } ARegion *node_has_tools_region(ScrArea *sa) { - ARegion *ar, *arnew; + ARegion *ar, *arnew; - ar = BKE_area_find_region_type(sa, RGN_TYPE_TOOLS); - if (ar) { - return ar; - } + ar = BKE_area_find_region_type(sa, RGN_TYPE_TOOLS); + if (ar) { + return ar; + } - /* add subdiv level; after header */ - ar = BKE_area_find_region_type(sa, RGN_TYPE_HEADER); + /* add subdiv level; after header */ + ar = BKE_area_find_region_type(sa, RGN_TYPE_HEADER); - /* is error! */ - if (ar == NULL) { - return NULL; - } + /* is error! */ + if (ar == NULL) { + return NULL; + } - arnew = MEM_callocN(sizeof(ARegion), "node tools"); + arnew = MEM_callocN(sizeof(ARegion), "node tools"); - BLI_insertlinkafter(&sa->regionbase, ar, arnew); - arnew->regiontype = RGN_TYPE_TOOLS; - arnew->alignment = RGN_ALIGN_LEFT; + BLI_insertlinkafter(&sa->regionbase, ar, arnew); + arnew->regiontype = RGN_TYPE_TOOLS; + arnew->alignment = RGN_ALIGN_LEFT; - arnew->flag = RGN_FLAG_HIDDEN; + arnew->flag = RGN_FLAG_HIDDEN; - return arnew; + return arnew; } /* ******************** default callbacks for node space ***************** */ static SpaceLink *node_new(const ScrArea *UNUSED(area), const Scene *UNUSED(scene)) { - ARegion *ar; - SpaceNode *snode; + ARegion *ar; + SpaceNode *snode; - snode = MEM_callocN(sizeof(SpaceNode), "initnode"); - snode->spacetype = SPACE_NODE; + snode = MEM_callocN(sizeof(SpaceNode), "initnode"); + snode->spacetype = SPACE_NODE; - snode->flag = SNODE_SHOW_GPENCIL | SNODE_USE_ALPHA; + snode->flag = SNODE_SHOW_GPENCIL | SNODE_USE_ALPHA; - /* backdrop */ - snode->zoom = 1.0f; + /* backdrop */ + snode->zoom = 1.0f; - /* select the first tree type for valid type */ - NODE_TREE_TYPES_BEGIN (treetype) - { - strcpy(snode->tree_idname, treetype->idname); - break; - } - NODE_TREE_TYPES_END; + /* select the first tree type for valid type */ + NODE_TREE_TYPES_BEGIN (treetype) { + strcpy(snode->tree_idname, treetype->idname); + break; + } + NODE_TREE_TYPES_END; - /* header */ - ar = MEM_callocN(sizeof(ARegion), "header for node"); + /* header */ + ar = MEM_callocN(sizeof(ARegion), "header for node"); - BLI_addtail(&snode->regionbase, ar); - ar->regiontype = RGN_TYPE_HEADER; - ar->alignment = (U.uiflag & USER_HEADER_BOTTOM) ? RGN_ALIGN_BOTTOM : RGN_ALIGN_TOP; + BLI_addtail(&snode->regionbase, ar); + ar->regiontype = RGN_TYPE_HEADER; + ar->alignment = (U.uiflag & USER_HEADER_BOTTOM) ? RGN_ALIGN_BOTTOM : RGN_ALIGN_TOP; - /* buttons/list view */ - ar = MEM_callocN(sizeof(ARegion), "buttons for node"); + /* buttons/list view */ + ar = MEM_callocN(sizeof(ARegion), "buttons for node"); - BLI_addtail(&snode->regionbase, ar); - ar->regiontype = RGN_TYPE_UI; - ar->alignment = RGN_ALIGN_RIGHT; + BLI_addtail(&snode->regionbase, ar); + ar->regiontype = RGN_TYPE_UI; + ar->alignment = RGN_ALIGN_RIGHT; - /* toolbar */ - ar = MEM_callocN(sizeof(ARegion), "node tools"); + /* toolbar */ + ar = MEM_callocN(sizeof(ARegion), "node tools"); - BLI_addtail(&snode->regionbase, ar); - ar->regiontype = RGN_TYPE_TOOLS; - ar->alignment = RGN_ALIGN_LEFT; + BLI_addtail(&snode->regionbase, ar); + ar->regiontype = RGN_TYPE_TOOLS; + ar->alignment = RGN_ALIGN_LEFT; - ar->flag = RGN_FLAG_HIDDEN; + ar->flag = RGN_FLAG_HIDDEN; - /* main region */ - ar = MEM_callocN(sizeof(ARegion), "main region for node"); + /* main region */ + ar = MEM_callocN(sizeof(ARegion), "main region for node"); - BLI_addtail(&snode->regionbase, ar); - ar->regiontype = RGN_TYPE_WINDOW; + BLI_addtail(&snode->regionbase, ar); + ar->regiontype = RGN_TYPE_WINDOW; - ar->v2d.tot.xmin = -12.8f * U.widget_unit; - ar->v2d.tot.ymin = -12.8f * U.widget_unit; - ar->v2d.tot.xmax = 38.4f * U.widget_unit; - ar->v2d.tot.ymax = 38.4f * U.widget_unit; + ar->v2d.tot.xmin = -12.8f * U.widget_unit; + ar->v2d.tot.ymin = -12.8f * U.widget_unit; + ar->v2d.tot.xmax = 38.4f * U.widget_unit; + ar->v2d.tot.ymax = 38.4f * U.widget_unit; - ar->v2d.cur = ar->v2d.tot; + ar->v2d.cur = ar->v2d.tot; - ar->v2d.min[0] = 1.0f; - ar->v2d.min[1] = 1.0f; + ar->v2d.min[0] = 1.0f; + ar->v2d.min[1] = 1.0f; - ar->v2d.max[0] = 32000.0f; - ar->v2d.max[1] = 32000.0f; + ar->v2d.max[0] = 32000.0f; + ar->v2d.max[1] = 32000.0f; - ar->v2d.minzoom = 0.09f; - ar->v2d.maxzoom = 2.31f; + ar->v2d.minzoom = 0.09f; + ar->v2d.maxzoom = 2.31f; - ar->v2d.scroll = (V2D_SCROLL_RIGHT | V2D_SCROLL_BOTTOM); - ar->v2d.keepzoom = V2D_LIMITZOOM | V2D_KEEPASPECT; - ar->v2d.keeptot = 0; + ar->v2d.scroll = (V2D_SCROLL_RIGHT | V2D_SCROLL_BOTTOM); + ar->v2d.keepzoom = V2D_LIMITZOOM | V2D_KEEPASPECT; + ar->v2d.keeptot = 0; - return (SpaceLink *)snode; + return (SpaceLink *)snode; } static void node_free(SpaceLink *sl) { - SpaceNode *snode = (SpaceNode *)sl; - bNodeTreePath *path, *path_next; + SpaceNode *snode = (SpaceNode *)sl; + bNodeTreePath *path, *path_next; - for (path = snode->treepath.first; path; path = path_next) { - path_next = path->next; - MEM_freeN(path); - } + for (path = snode->treepath.first; path; path = path_next) { + path_next = path->next; + MEM_freeN(path); + } } - /* spacetype; init callback */ static void node_init(struct wmWindowManager *UNUSED(wm), ScrArea *UNUSED(sa)) { - } -static void node_area_listener(wmWindow *UNUSED(win), ScrArea *sa, wmNotifier *wmn, Scene *UNUSED(scene)) +static void node_area_listener(wmWindow *UNUSED(win), + ScrArea *sa, + wmNotifier *wmn, + Scene *UNUSED(scene)) { - /* note, ED_area_tag_refresh will re-execute compositor */ - SpaceNode *snode = sa->spacedata.first; - /* shaderfrom is only used for new shading nodes, otherwise all shaders are from objects */ - short shader_type = snode->shaderfrom; - - /* preview renders */ - switch (wmn->category) { - case NC_SCENE: - switch (wmn->data) { - case ND_NODES: - { - ARegion *ar = BKE_area_find_region_type(sa, RGN_TYPE_WINDOW); - bNodeTreePath *path = snode->treepath.last; - /* shift view to node tree center */ - if (ar && path) { - UI_view2d_center_set(&ar->v2d, path->view_center[0], path->view_center[1]); - } - - ED_area_tag_refresh(sa); - break; - } - case ND_FRAME: - ED_area_tag_refresh(sa); - break; - case ND_COMPO_RESULT: - ED_area_tag_redraw(sa); - break; - case ND_TRANSFORM_DONE: - if (ED_node_is_compositor(snode)) { - if (snode->flag & SNODE_AUTO_RENDER) { - snode->recalc = 1; - ED_area_tag_refresh(sa); - } - } - break; - case ND_LAYER_CONTENT: - ED_area_tag_refresh(sa); - break; - } - break; - - /* future: add ID checks? */ - case NC_MATERIAL: - if (ED_node_is_shader(snode)) { - if (wmn->data == ND_SHADING) { - ED_area_tag_refresh(sa); - } - else if (wmn->data == ND_SHADING_DRAW) { - ED_area_tag_refresh(sa); - } - else if (wmn->data == ND_SHADING_LINKS) { - ED_area_tag_refresh(sa); - } - else if (wmn->action == NA_ADDED && snode->edittree) { - nodeSetActiveID(snode->edittree, ID_MA, wmn->reference); - } - - } - break; - case NC_TEXTURE: - if (ED_node_is_shader(snode) || ED_node_is_texture(snode)) { - if (wmn->data == ND_NODES) { - ED_area_tag_refresh(sa); - } - } - break; - case NC_WORLD: - if (ED_node_is_shader(snode) && shader_type == SNODE_SHADER_WORLD) { - ED_area_tag_refresh(sa); - } - break; - case NC_OBJECT: - if (ED_node_is_shader(snode)) { - if (wmn->data == ND_OB_SHADING) { - ED_area_tag_refresh(sa); - } - } - break; - case NC_SPACE: - if (wmn->data == ND_SPACE_NODE) { - ED_area_tag_refresh(sa); - } - else if (wmn->data == ND_SPACE_NODE_VIEW) { - ED_area_tag_redraw(sa); - } - break; - case NC_NODE: - if (wmn->action == NA_EDITED) { - ED_area_tag_refresh(sa); - } - else if (wmn->action == NA_SELECTED) { - ED_area_tag_redraw(sa); - } - break; - case NC_SCREEN: - switch (wmn->data) { - case ND_ANIMPLAY: - ED_area_tag_refresh(sa); - break; - } - break; - case NC_MASK: - if (wmn->action == NA_EDITED) { - if (snode->nodetree && snode->nodetree->type == NTREE_COMPOSIT) { - ED_area_tag_refresh(sa); - } - } - break; - - case NC_IMAGE: - if (wmn->action == NA_EDITED) { - if (ED_node_is_compositor(snode)) { - /* note that nodeUpdateID is already called by BKE_image_signal() on all - * scenes so really this is just to know if the images is used in the compo else - * painting on images could become very slow when the compositor is open. */ - if (nodeUpdateID(snode->nodetree, wmn->reference)) { - ED_area_tag_refresh(sa); - } - } - } - break; - - case NC_MOVIECLIP: - if (wmn->action == NA_EDITED) { - if (ED_node_is_compositor(snode)) { - if (nodeUpdateID(snode->nodetree, wmn->reference)) { - ED_area_tag_refresh(sa); - } - } - } - break; - - case NC_LINESTYLE: - if (ED_node_is_shader(snode) && shader_type == SNODE_SHADER_LINESTYLE) { - ED_area_tag_refresh(sa); - } - break; - case NC_WM: - if (wmn->data == ND_UNDO) { - ED_area_tag_refresh(sa); - } - break; - case NC_GPENCIL: - if (ELEM(wmn->action, NA_EDITED, NA_SELECTED)) { - ED_area_tag_redraw(sa); - } - break; - } + /* note, ED_area_tag_refresh will re-execute compositor */ + SpaceNode *snode = sa->spacedata.first; + /* shaderfrom is only used for new shading nodes, otherwise all shaders are from objects */ + short shader_type = snode->shaderfrom; + + /* preview renders */ + switch (wmn->category) { + case NC_SCENE: + switch (wmn->data) { + case ND_NODES: { + ARegion *ar = BKE_area_find_region_type(sa, RGN_TYPE_WINDOW); + bNodeTreePath *path = snode->treepath.last; + /* shift view to node tree center */ + if (ar && path) { + UI_view2d_center_set(&ar->v2d, path->view_center[0], path->view_center[1]); + } + + ED_area_tag_refresh(sa); + break; + } + case ND_FRAME: + ED_area_tag_refresh(sa); + break; + case ND_COMPO_RESULT: + ED_area_tag_redraw(sa); + break; + case ND_TRANSFORM_DONE: + if (ED_node_is_compositor(snode)) { + if (snode->flag & SNODE_AUTO_RENDER) { + snode->recalc = 1; + ED_area_tag_refresh(sa); + } + } + break; + case ND_LAYER_CONTENT: + ED_area_tag_refresh(sa); + break; + } + break; + + /* future: add ID checks? */ + case NC_MATERIAL: + if (ED_node_is_shader(snode)) { + if (wmn->data == ND_SHADING) { + ED_area_tag_refresh(sa); + } + else if (wmn->data == ND_SHADING_DRAW) { + ED_area_tag_refresh(sa); + } + else if (wmn->data == ND_SHADING_LINKS) { + ED_area_tag_refresh(sa); + } + else if (wmn->action == NA_ADDED && snode->edittree) { + nodeSetActiveID(snode->edittree, ID_MA, wmn->reference); + } + } + break; + case NC_TEXTURE: + if (ED_node_is_shader(snode) || ED_node_is_texture(snode)) { + if (wmn->data == ND_NODES) { + ED_area_tag_refresh(sa); + } + } + break; + case NC_WORLD: + if (ED_node_is_shader(snode) && shader_type == SNODE_SHADER_WORLD) { + ED_area_tag_refresh(sa); + } + break; + case NC_OBJECT: + if (ED_node_is_shader(snode)) { + if (wmn->data == ND_OB_SHADING) { + ED_area_tag_refresh(sa); + } + } + break; + case NC_SPACE: + if (wmn->data == ND_SPACE_NODE) { + ED_area_tag_refresh(sa); + } + else if (wmn->data == ND_SPACE_NODE_VIEW) { + ED_area_tag_redraw(sa); + } + break; + case NC_NODE: + if (wmn->action == NA_EDITED) { + ED_area_tag_refresh(sa); + } + else if (wmn->action == NA_SELECTED) { + ED_area_tag_redraw(sa); + } + break; + case NC_SCREEN: + switch (wmn->data) { + case ND_ANIMPLAY: + ED_area_tag_refresh(sa); + break; + } + break; + case NC_MASK: + if (wmn->action == NA_EDITED) { + if (snode->nodetree && snode->nodetree->type == NTREE_COMPOSIT) { + ED_area_tag_refresh(sa); + } + } + break; + + case NC_IMAGE: + if (wmn->action == NA_EDITED) { + if (ED_node_is_compositor(snode)) { + /* note that nodeUpdateID is already called by BKE_image_signal() on all + * scenes so really this is just to know if the images is used in the compo else + * painting on images could become very slow when the compositor is open. */ + if (nodeUpdateID(snode->nodetree, wmn->reference)) { + ED_area_tag_refresh(sa); + } + } + } + break; + + case NC_MOVIECLIP: + if (wmn->action == NA_EDITED) { + if (ED_node_is_compositor(snode)) { + if (nodeUpdateID(snode->nodetree, wmn->reference)) { + ED_area_tag_refresh(sa); + } + } + } + break; + + case NC_LINESTYLE: + if (ED_node_is_shader(snode) && shader_type == SNODE_SHADER_LINESTYLE) { + ED_area_tag_refresh(sa); + } + break; + case NC_WM: + if (wmn->data == ND_UNDO) { + ED_area_tag_refresh(sa); + } + break; + case NC_GPENCIL: + if (ELEM(wmn->action, NA_EDITED, NA_SELECTED)) { + ED_area_tag_redraw(sa); + } + break; + } } static void node_area_refresh(const struct bContext *C, ScrArea *sa) { - /* default now: refresh node is starting preview */ - SpaceNode *snode = sa->spacedata.first; - - snode_set_context(C); - - if (snode->nodetree) { - if (snode->nodetree->type == NTREE_SHADER) { - if (GS(snode->id->name) == ID_MA) { - Material *ma = (Material *)snode->id; - if (ma->use_nodes) { - ED_preview_shader_job(C, sa, snode->id, NULL, NULL, 100, 100, PR_NODE_RENDER); - } - } - else if (GS(snode->id->name) == ID_LA) { - Light *la = (Light *)snode->id; - if (la->use_nodes) { - ED_preview_shader_job(C, sa, snode->id, NULL, NULL, 100, 100, PR_NODE_RENDER); - } - } - else if (GS(snode->id->name) == ID_WO) { - World *wo = (World *)snode->id; - if (wo->use_nodes) { - ED_preview_shader_job(C, sa, snode->id, NULL, NULL, 100, 100, PR_NODE_RENDER); - } - } - } - else if (snode->nodetree->type == NTREE_COMPOSIT) { - Scene *scene = (Scene *)snode->id; - if (scene->use_nodes) { - /* recalc is set on 3d view changes for auto compo */ - if (snode->recalc) { - snode->recalc = 0; - node_render_changed_exec((struct bContext *)C, NULL); - } - else { - ED_node_composite_job(C, snode->nodetree, scene); - } - } - } - else if (snode->nodetree->type == NTREE_TEXTURE) { - Tex *tex = (Tex *)snode->id; - if (tex->use_nodes) { - ED_preview_shader_job(C, sa, snode->id, NULL, NULL, 100, 100, PR_NODE_RENDER); - } - } - } + /* default now: refresh node is starting preview */ + SpaceNode *snode = sa->spacedata.first; + + snode_set_context(C); + + if (snode->nodetree) { + if (snode->nodetree->type == NTREE_SHADER) { + if (GS(snode->id->name) == ID_MA) { + Material *ma = (Material *)snode->id; + if (ma->use_nodes) { + ED_preview_shader_job(C, sa, snode->id, NULL, NULL, 100, 100, PR_NODE_RENDER); + } + } + else if (GS(snode->id->name) == ID_LA) { + Light *la = (Light *)snode->id; + if (la->use_nodes) { + ED_preview_shader_job(C, sa, snode->id, NULL, NULL, 100, 100, PR_NODE_RENDER); + } + } + else if (GS(snode->id->name) == ID_WO) { + World *wo = (World *)snode->id; + if (wo->use_nodes) { + ED_preview_shader_job(C, sa, snode->id, NULL, NULL, 100, 100, PR_NODE_RENDER); + } + } + } + else if (snode->nodetree->type == NTREE_COMPOSIT) { + Scene *scene = (Scene *)snode->id; + if (scene->use_nodes) { + /* recalc is set on 3d view changes for auto compo */ + if (snode->recalc) { + snode->recalc = 0; + node_render_changed_exec((struct bContext *)C, NULL); + } + else { + ED_node_composite_job(C, snode->nodetree, scene); + } + } + } + else if (snode->nodetree->type == NTREE_TEXTURE) { + Tex *tex = (Tex *)snode->id; + if (tex->use_nodes) { + ED_preview_shader_job(C, sa, snode->id, NULL, NULL, 100, 100, PR_NODE_RENDER); + } + } + } } static SpaceLink *node_duplicate(SpaceLink *sl) { - SpaceNode *snode = (SpaceNode *)sl; - SpaceNode *snoden = MEM_dupallocN(snode); + SpaceNode *snode = (SpaceNode *)sl; + SpaceNode *snoden = MEM_dupallocN(snode); - BLI_duplicatelist(&snoden->treepath, &snode->treepath); + BLI_duplicatelist(&snoden->treepath, &snode->treepath); - /* clear or remove stuff from old */ - BLI_listbase_clear(&snoden->linkdrag); + /* clear or remove stuff from old */ + BLI_listbase_clear(&snoden->linkdrag); - /* Note: no need to set node tree user counts, - * the editor only keeps at least 1 (id_us_ensure_real), - * which is already done by the original SpaceNode. - */ + /* Note: no need to set node tree user counts, + * the editor only keeps at least 1 (id_us_ensure_real), + * which is already done by the original SpaceNode. + */ - return (SpaceLink *)snoden; + return (SpaceLink *)snoden; } - /* add handlers, stuff you only do once or on area/region changes */ static void node_buttons_region_init(wmWindowManager *wm, ARegion *ar) { - wmKeyMap *keymap; + wmKeyMap *keymap; - ED_region_panels_init(wm, ar); + ED_region_panels_init(wm, ar); - keymap = WM_keymap_ensure(wm->defaultconf, "Node Generic", SPACE_NODE, 0); - WM_event_add_keymap_handler(&ar->handlers, keymap); + keymap = WM_keymap_ensure(wm->defaultconf, "Node Generic", SPACE_NODE, 0); + WM_event_add_keymap_handler(&ar->handlers, keymap); } static void node_buttons_region_draw(const bContext *C, ARegion *ar) { - ED_region_panels(C, ar); + ED_region_panels(C, ar); } /* add handlers, stuff you only do once or on area/region changes */ static void node_toolbar_region_init(wmWindowManager *wm, ARegion *ar) { - wmKeyMap *keymap; + wmKeyMap *keymap; - ED_region_panels_init(wm, ar); + ED_region_panels_init(wm, ar); - keymap = WM_keymap_ensure(wm->defaultconf, "Node Generic", SPACE_NODE, 0); - WM_event_add_keymap_handler(&ar->handlers, keymap); + keymap = WM_keymap_ensure(wm->defaultconf, "Node Generic", SPACE_NODE, 0); + WM_event_add_keymap_handler(&ar->handlers, keymap); } static void node_toolbar_region_draw(const bContext *C, ARegion *ar) { - ED_region_panels(C, ar); + ED_region_panels(C, ar); } static void node_cursor(wmWindow *win, ScrArea *sa, ARegion *ar) { - SpaceNode *snode = sa->spacedata.first; - - /* convert mouse coordinates to v2d space */ - UI_view2d_region_to_view(&ar->v2d, win->eventstate->x - ar->winrct.xmin, win->eventstate->y - ar->winrct.ymin, - &snode->cursor[0], &snode->cursor[1]); + SpaceNode *snode = sa->spacedata.first; - /* here snode->cursor is used to detect the node edge for sizing */ - node_set_cursor(win, snode, snode->cursor); + /* convert mouse coordinates to v2d space */ + UI_view2d_region_to_view(&ar->v2d, + win->eventstate->x - ar->winrct.xmin, + win->eventstate->y - ar->winrct.ymin, + &snode->cursor[0], + &snode->cursor[1]); - /* XXX snode->cursor is in placing new nodes space */ - snode->cursor[0] /= UI_DPI_FAC; - snode->cursor[1] /= UI_DPI_FAC; + /* here snode->cursor is used to detect the node edge for sizing */ + node_set_cursor(win, snode, snode->cursor); + /* XXX snode->cursor is in placing new nodes space */ + snode->cursor[0] /= UI_DPI_FAC; + snode->cursor[1] /= UI_DPI_FAC; } /* Initialize main region, setting handlers. */ static void node_main_region_init(wmWindowManager *wm, ARegion *ar) { - wmKeyMap *keymap; - ListBase *lb; + wmKeyMap *keymap; + ListBase *lb; - UI_view2d_region_reinit(&ar->v2d, V2D_COMMONVIEW_CUSTOM, ar->winx, ar->winy); + UI_view2d_region_reinit(&ar->v2d, V2D_COMMONVIEW_CUSTOM, ar->winx, ar->winy); - /* own keymaps */ - keymap = WM_keymap_ensure(wm->defaultconf, "Node Generic", SPACE_NODE, 0); - WM_event_add_keymap_handler(&ar->handlers, keymap); + /* own keymaps */ + keymap = WM_keymap_ensure(wm->defaultconf, "Node Generic", SPACE_NODE, 0); + WM_event_add_keymap_handler(&ar->handlers, keymap); - keymap = WM_keymap_ensure(wm->defaultconf, "Node Editor", SPACE_NODE, 0); - WM_event_add_keymap_handler_bb(&ar->handlers, keymap, &ar->v2d.mask, &ar->winrct); + keymap = WM_keymap_ensure(wm->defaultconf, "Node Editor", SPACE_NODE, 0); + WM_event_add_keymap_handler_bb(&ar->handlers, keymap, &ar->v2d.mask, &ar->winrct); - /* add drop boxes */ - lb = WM_dropboxmap_find("Node Editor", SPACE_NODE, RGN_TYPE_WINDOW); + /* add drop boxes */ + lb = WM_dropboxmap_find("Node Editor", SPACE_NODE, RGN_TYPE_WINDOW); - WM_event_add_dropbox_handler(&ar->handlers, lb); + WM_event_add_dropbox_handler(&ar->handlers, lb); } static void node_main_region_draw(const bContext *C, ARegion *ar) { - drawnodespace(C, ar); + drawnodespace(C, ar); } - /* ************* dropboxes ************* */ -static bool node_ima_drop_poll(bContext *UNUSED(C), wmDrag *drag, const wmEvent *UNUSED(event), const char **UNUSED(tooltip)) +static bool node_ima_drop_poll(bContext *UNUSED(C), + wmDrag *drag, + const wmEvent *UNUSED(event), + const char **UNUSED(tooltip)) { - if (drag->type == WM_DRAG_PATH) { - /* rule might not work? */ - return (ELEM(drag->icon, 0, ICON_FILE_IMAGE, ICON_FILE_MOVIE)); - } - else { - return WM_drag_ID(drag, ID_IM) != NULL; - } + if (drag->type == WM_DRAG_PATH) { + /* rule might not work? */ + return (ELEM(drag->icon, 0, ICON_FILE_IMAGE, ICON_FILE_MOVIE)); + } + else { + return WM_drag_ID(drag, ID_IM) != NULL; + } } -static bool node_mask_drop_poll(bContext *UNUSED(C), wmDrag *drag, const wmEvent *UNUSED(event), const char **UNUSED(tooltip)) +static bool node_mask_drop_poll(bContext *UNUSED(C), + wmDrag *drag, + const wmEvent *UNUSED(event), + const char **UNUSED(tooltip)) { - return WM_drag_ID(drag, ID_MSK) != NULL; + return WM_drag_ID(drag, ID_MSK) != NULL; } static void node_id_drop_copy(wmDrag *drag, wmDropBox *drop) { - ID *id = WM_drag_ID(drag, 0); + ID *id = WM_drag_ID(drag, 0); - RNA_string_set(drop->ptr, "name", id->name + 2); + RNA_string_set(drop->ptr, "name", id->name + 2); } static void node_id_path_drop_copy(wmDrag *drag, wmDropBox *drop) { - ID *id = WM_drag_ID(drag, 0); - - if (id) { - RNA_string_set(drop->ptr, "name", id->name + 2); - RNA_struct_property_unset(drop->ptr, "filepath"); - } - else if (drag->path[0]) { - RNA_string_set(drop->ptr, "filepath", drag->path); - RNA_struct_property_unset(drop->ptr, "name"); - } + ID *id = WM_drag_ID(drag, 0); + + if (id) { + RNA_string_set(drop->ptr, "name", id->name + 2); + RNA_struct_property_unset(drop->ptr, "filepath"); + } + else if (drag->path[0]) { + RNA_string_set(drop->ptr, "filepath", drag->path); + RNA_struct_property_unset(drop->ptr, "name"); + } } /* this region dropbox definition */ static void node_dropboxes(void) { - ListBase *lb = WM_dropboxmap_find("Node Editor", SPACE_NODE, RGN_TYPE_WINDOW); - - WM_dropbox_add(lb, "NODE_OT_add_file", node_ima_drop_poll, node_id_path_drop_copy); - WM_dropbox_add(lb, "NODE_OT_add_mask", node_mask_drop_poll, node_id_drop_copy); + ListBase *lb = WM_dropboxmap_find("Node Editor", SPACE_NODE, RGN_TYPE_WINDOW); + WM_dropbox_add(lb, "NODE_OT_add_file", node_ima_drop_poll, node_id_path_drop_copy); + WM_dropbox_add(lb, "NODE_OT_add_mask", node_mask_drop_poll, node_id_drop_copy); } /* ************* end drop *********** */ - /* add handlers, stuff you only do once or on area/region changes */ static void node_header_region_init(wmWindowManager *UNUSED(wm), ARegion *ar) { - ED_region_header_init(ar); + ED_region_header_init(ar); } static void node_header_region_draw(const bContext *C, ARegion *ar) { - /* find and set the context */ - snode_set_context(C); + /* find and set the context */ + snode_set_context(C); - ED_region_header(C, ar); + ED_region_header(C, ar); } /* used for header + main region */ -static void node_region_listener( - wmWindow *UNUSED(win), ScrArea *UNUSED(sa), ARegion *ar, - wmNotifier *wmn, const Scene *UNUSED(scene)) +static void node_region_listener(wmWindow *UNUSED(win), + ScrArea *UNUSED(sa), + ARegion *ar, + wmNotifier *wmn, + const Scene *UNUSED(scene)) { - wmGizmoMap *gzmap = ar->gizmo_map; - - /* context changes */ - switch (wmn->category) { - case NC_SPACE: - switch (wmn->data) { - case ND_SPACE_NODE: - ED_region_tag_redraw(ar); - break; - case ND_SPACE_NODE_VIEW: - WM_gizmomap_tag_refresh(gzmap); - break; - } - break; - case NC_SCREEN: - if (wmn->data == ND_LAYOUTSET || wmn->action == NA_EDITED) { - WM_gizmomap_tag_refresh(gzmap); - } - switch (wmn->data) { - case ND_ANIMPLAY: - case ND_LAYER: - ED_region_tag_redraw(ar); - break; - } - break; - case NC_WM: - if (wmn->data == ND_JOB) { - ED_region_tag_redraw(ar); - } - break; - case NC_SCENE: - ED_region_tag_redraw(ar); - if (wmn->data == ND_RENDER_RESULT) { - WM_gizmomap_tag_refresh(gzmap); - } - break; - case NC_NODE: - ED_region_tag_redraw(ar); - if (ELEM(wmn->action, NA_EDITED, NA_SELECTED)) { - WM_gizmomap_tag_refresh(gzmap); - } - break; - case NC_MATERIAL: - case NC_TEXTURE: - case NC_WORLD: - case NC_LINESTYLE: - ED_region_tag_redraw(ar); - break; - case NC_OBJECT: - if (wmn->data == ND_OB_SHADING) { - ED_region_tag_redraw(ar); - } - break; - case NC_ID: - if (wmn->action == NA_RENAME) { - ED_region_tag_redraw(ar); - } - break; - case NC_GPENCIL: - if (wmn->action == NA_EDITED) { - ED_region_tag_redraw(ar); - } - else if (wmn->data & ND_GPENCIL_EDITMODE) { - ED_region_tag_redraw(ar); - } - break; - } + wmGizmoMap *gzmap = ar->gizmo_map; + + /* context changes */ + switch (wmn->category) { + case NC_SPACE: + switch (wmn->data) { + case ND_SPACE_NODE: + ED_region_tag_redraw(ar); + break; + case ND_SPACE_NODE_VIEW: + WM_gizmomap_tag_refresh(gzmap); + break; + } + break; + case NC_SCREEN: + if (wmn->data == ND_LAYOUTSET || wmn->action == NA_EDITED) { + WM_gizmomap_tag_refresh(gzmap); + } + switch (wmn->data) { + case ND_ANIMPLAY: + case ND_LAYER: + ED_region_tag_redraw(ar); + break; + } + break; + case NC_WM: + if (wmn->data == ND_JOB) { + ED_region_tag_redraw(ar); + } + break; + case NC_SCENE: + ED_region_tag_redraw(ar); + if (wmn->data == ND_RENDER_RESULT) { + WM_gizmomap_tag_refresh(gzmap); + } + break; + case NC_NODE: + ED_region_tag_redraw(ar); + if (ELEM(wmn->action, NA_EDITED, NA_SELECTED)) { + WM_gizmomap_tag_refresh(gzmap); + } + break; + case NC_MATERIAL: + case NC_TEXTURE: + case NC_WORLD: + case NC_LINESTYLE: + ED_region_tag_redraw(ar); + break; + case NC_OBJECT: + if (wmn->data == ND_OB_SHADING) { + ED_region_tag_redraw(ar); + } + break; + case NC_ID: + if (wmn->action == NA_RENAME) { + ED_region_tag_redraw(ar); + } + break; + case NC_GPENCIL: + if (wmn->action == NA_EDITED) { + ED_region_tag_redraw(ar); + } + else if (wmn->data & ND_GPENCIL_EDITMODE) { + ED_region_tag_redraw(ar); + } + break; + } } const char *node_context_dir[] = { - "selected_nodes", "active_node", "light", "material", "world", NULL -}; + "selected_nodes", "active_node", "light", "material", "world", NULL}; static int node_context(const bContext *C, const char *member, bContextDataResult *result) { - SpaceNode *snode = CTX_wm_space_node(C); - - if (CTX_data_dir(member)) { - CTX_data_dir_set(result, node_context_dir); - return 1; - } - else if (CTX_data_equals(member, "selected_nodes")) { - bNode *node; - - if (snode->edittree) { - for (node = snode->edittree->nodes.last; node; node = node->prev) { - if (node->flag & NODE_SELECT) { - CTX_data_list_add(result, &snode->edittree->id, &RNA_Node, node); - } - } - } - CTX_data_type_set(result, CTX_DATA_TYPE_COLLECTION); - return 1; - } - else if (CTX_data_equals(member, "active_node")) { - if (snode->edittree) { - bNode *node = nodeGetActive(snode->edittree); - CTX_data_pointer_set(result, &snode->edittree->id, &RNA_Node, node); - } - - CTX_data_type_set(result, CTX_DATA_TYPE_POINTER); - return 1; - } - else if (CTX_data_equals(member, "node_previews")) { - if (snode->nodetree) { - CTX_data_pointer_set(result, &snode->nodetree->id, &RNA_NodeInstanceHash, snode->nodetree->previews); - } - - CTX_data_type_set(result, CTX_DATA_TYPE_POINTER); - return 1; - } - else if (CTX_data_equals(member, "material")) { - if (snode->id && GS(snode->id->name) == ID_MA) { - CTX_data_id_pointer_set(result, snode->id); - } - return 1; - } - else if (CTX_data_equals(member, "light")) { - if (snode->id && GS(snode->id->name) == ID_LA) { - CTX_data_id_pointer_set(result, snode->id); - } - return 1; - } - else if (CTX_data_equals(member, "world")) { - if (snode->id && GS(snode->id->name) == ID_WO) { - CTX_data_id_pointer_set(result, snode->id); - } - return 1; - } - - return 0; + SpaceNode *snode = CTX_wm_space_node(C); + + if (CTX_data_dir(member)) { + CTX_data_dir_set(result, node_context_dir); + return 1; + } + else if (CTX_data_equals(member, "selected_nodes")) { + bNode *node; + + if (snode->edittree) { + for (node = snode->edittree->nodes.last; node; node = node->prev) { + if (node->flag & NODE_SELECT) { + CTX_data_list_add(result, &snode->edittree->id, &RNA_Node, node); + } + } + } + CTX_data_type_set(result, CTX_DATA_TYPE_COLLECTION); + return 1; + } + else if (CTX_data_equals(member, "active_node")) { + if (snode->edittree) { + bNode *node = nodeGetActive(snode->edittree); + CTX_data_pointer_set(result, &snode->edittree->id, &RNA_Node, node); + } + + CTX_data_type_set(result, CTX_DATA_TYPE_POINTER); + return 1; + } + else if (CTX_data_equals(member, "node_previews")) { + if (snode->nodetree) { + CTX_data_pointer_set( + result, &snode->nodetree->id, &RNA_NodeInstanceHash, snode->nodetree->previews); + } + + CTX_data_type_set(result, CTX_DATA_TYPE_POINTER); + return 1; + } + else if (CTX_data_equals(member, "material")) { + if (snode->id && GS(snode->id->name) == ID_MA) { + CTX_data_id_pointer_set(result, snode->id); + } + return 1; + } + else if (CTX_data_equals(member, "light")) { + if (snode->id && GS(snode->id->name) == ID_LA) { + CTX_data_id_pointer_set(result, snode->id); + } + return 1; + } + else if (CTX_data_equals(member, "world")) { + if (snode->id && GS(snode->id->name) == ID_WO) { + CTX_data_id_pointer_set(result, snode->id); + } + return 1; + } + + return 0; } static void node_widgets(void) { - /* create the widgetmap for the area here */ - wmGizmoMapType *gzmap_type = WM_gizmomaptype_ensure( - &(const struct wmGizmoMapType_Params){SPACE_NODE, RGN_TYPE_WINDOW}); - WM_gizmogrouptype_append_and_link(gzmap_type, NODE_GGT_backdrop_transform); - WM_gizmogrouptype_append_and_link(gzmap_type, NODE_GGT_backdrop_crop); - WM_gizmogrouptype_append_and_link(gzmap_type, NODE_GGT_backdrop_sun_beams); - WM_gizmogrouptype_append_and_link(gzmap_type, NODE_GGT_backdrop_corner_pin); + /* create the widgetmap for the area here */ + wmGizmoMapType *gzmap_type = WM_gizmomaptype_ensure( + &(const struct wmGizmoMapType_Params){SPACE_NODE, RGN_TYPE_WINDOW}); + WM_gizmogrouptype_append_and_link(gzmap_type, NODE_GGT_backdrop_transform); + WM_gizmogrouptype_append_and_link(gzmap_type, NODE_GGT_backdrop_crop); + WM_gizmogrouptype_append_and_link(gzmap_type, NODE_GGT_backdrop_sun_beams); + WM_gizmogrouptype_append_and_link(gzmap_type, NODE_GGT_backdrop_corner_pin); } static void node_id_remap(ScrArea *UNUSED(sa), SpaceLink *slink, ID *old_id, ID *new_id) { - SpaceNode *snode = (SpaceNode *)slink; - - if (GS(old_id->name) == ID_SCE) { - if (snode->id == old_id) { - /* nasty DNA logic for SpaceNode: - * ideally should be handled by editor code, but would be bad level call - */ - BLI_freelistN(&snode->treepath); - - /* XXX Untested in case new_id != NULL... */ - snode->id = new_id; - snode->from = NULL; - snode->nodetree = NULL; - snode->edittree = NULL; - } - } - else if (GS(old_id->name) == ID_OB) { - if (snode->from == old_id) { - if (new_id == NULL) { - snode->flag &= ~SNODE_PIN; - } - snode->from = new_id; - } - } - else if (GS(old_id->name) == ID_GD) { - if ((ID *)snode->gpd == old_id) { - snode->gpd = (bGPdata *)new_id; - id_us_min(old_id); - id_us_plus(new_id); - } - } - else if (GS(old_id->name) == ID_NT) { - bNodeTreePath *path, *path_next; - - for (path = snode->treepath.first; path; path = path->next) { - if ((ID *)path->nodetree == old_id) { - path->nodetree = (bNodeTree *)new_id; - id_us_min(old_id); - id_us_plus(new_id); - } - if (path == snode->treepath.first) { - /* first nodetree in path is same as snode->nodetree */ - snode->nodetree = path->nodetree; - } - if (path->nodetree == NULL) { - break; - } - } - - /* remaining path entries are invalid, remove */ - for (; path; path = path_next) { - path_next = path->next; - - BLI_remlink(&snode->treepath, path); - MEM_freeN(path); - } - - /* edittree is just the last in the path, - * set this directly since the path may have been shortened above */ - if (snode->treepath.last) { - path = snode->treepath.last; - snode->edittree = path->nodetree; - } - else { - snode->edittree = NULL; - } - } + SpaceNode *snode = (SpaceNode *)slink; + + if (GS(old_id->name) == ID_SCE) { + if (snode->id == old_id) { + /* nasty DNA logic for SpaceNode: + * ideally should be handled by editor code, but would be bad level call + */ + BLI_freelistN(&snode->treepath); + + /* XXX Untested in case new_id != NULL... */ + snode->id = new_id; + snode->from = NULL; + snode->nodetree = NULL; + snode->edittree = NULL; + } + } + else if (GS(old_id->name) == ID_OB) { + if (snode->from == old_id) { + if (new_id == NULL) { + snode->flag &= ~SNODE_PIN; + } + snode->from = new_id; + } + } + else if (GS(old_id->name) == ID_GD) { + if ((ID *)snode->gpd == old_id) { + snode->gpd = (bGPdata *)new_id; + id_us_min(old_id); + id_us_plus(new_id); + } + } + else if (GS(old_id->name) == ID_NT) { + bNodeTreePath *path, *path_next; + + for (path = snode->treepath.first; path; path = path->next) { + if ((ID *)path->nodetree == old_id) { + path->nodetree = (bNodeTree *)new_id; + id_us_min(old_id); + id_us_plus(new_id); + } + if (path == snode->treepath.first) { + /* first nodetree in path is same as snode->nodetree */ + snode->nodetree = path->nodetree; + } + if (path->nodetree == NULL) { + break; + } + } + + /* remaining path entries are invalid, remove */ + for (; path; path = path_next) { + path_next = path->next; + + BLI_remlink(&snode->treepath, path); + MEM_freeN(path); + } + + /* edittree is just the last in the path, + * set this directly since the path may have been shortened above */ + if (snode->treepath.last) { + path = snode->treepath.last; + snode->edittree = path->nodetree; + } + else { + snode->edittree = NULL; + } + } } - static int node_space_subtype_get(ScrArea *sa) { - SpaceNode *snode = sa->spacedata.first; - return rna_node_tree_idname_to_enum(snode->tree_idname); + SpaceNode *snode = sa->spacedata.first; + return rna_node_tree_idname_to_enum(snode->tree_idname); } static void node_space_subtype_set(ScrArea *sa, int value) { - SpaceNode *snode = sa->spacedata.first; - ED_node_set_tree_type(snode, rna_node_tree_type_from_enum(value)); + SpaceNode *snode = sa->spacedata.first; + ED_node_set_tree_type(snode, rna_node_tree_type_from_enum(value)); } -static void node_space_subtype_item_extend( - bContext *C, EnumPropertyItem **item, int *totitem) +static void node_space_subtype_item_extend(bContext *C, EnumPropertyItem **item, int *totitem) { - bool free; - const EnumPropertyItem *item_src = RNA_enum_node_tree_types_itemf_impl(C, &free); - for (const EnumPropertyItem *item_iter = item_src; item_iter->identifier; item_iter++) { - RNA_enum_item_add(item, totitem, item_iter); - } - if (free) { - MEM_freeN((void *)item_src); - } + bool free; + const EnumPropertyItem *item_src = RNA_enum_node_tree_types_itemf_impl(C, &free); + for (const EnumPropertyItem *item_iter = item_src; item_iter->identifier; item_iter++) { + RNA_enum_item_add(item, totitem, item_iter); + } + if (free) { + MEM_freeN((void *)item_src); + } } /* only called once, from space/spacetypes.c */ void ED_spacetype_node(void) { - SpaceType *st = MEM_callocN(sizeof(SpaceType), "spacetype node"); - ARegionType *art; - - st->spaceid = SPACE_NODE; - strncpy(st->name, "Node", BKE_ST_MAXNAME); - - st->new = node_new; - st->free = node_free; - st->init = node_init; - st->duplicate = node_duplicate; - st->operatortypes = node_operatortypes; - st->keymap = node_keymap; - st->listener = node_area_listener; - st->refresh = node_area_refresh; - st->context = node_context; - st->dropboxes = node_dropboxes; - st->gizmos = node_widgets; - st->id_remap = node_id_remap; - st->space_subtype_item_extend = node_space_subtype_item_extend; - st->space_subtype_get = node_space_subtype_get; - st->space_subtype_set = node_space_subtype_set; - - /* regions: main window */ - art = MEM_callocN(sizeof(ARegionType), "spacetype node region"); - art->regionid = RGN_TYPE_WINDOW; - art->init = node_main_region_init; - art->draw = node_main_region_draw; - art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_GIZMO | ED_KEYMAP_TOOL | ED_KEYMAP_VIEW2D | ED_KEYMAP_FRAMES | ED_KEYMAP_GPENCIL; - art->listener = node_region_listener; - art->cursor = node_cursor; - art->event_cursor = true; - - BLI_addhead(&st->regiontypes, art); - - /* regions: header */ - art = MEM_callocN(sizeof(ARegionType), "spacetype node region"); - art->regionid = RGN_TYPE_HEADER; - art->prefsizey = HEADERY; - art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_VIEW2D | ED_KEYMAP_FRAMES | ED_KEYMAP_HEADER; - art->listener = node_region_listener; - art->init = node_header_region_init; - art->draw = node_header_region_draw; - - BLI_addhead(&st->regiontypes, art); - - /* regions: listview/buttons */ - art = MEM_callocN(sizeof(ARegionType), "spacetype node region"); - art->regionid = RGN_TYPE_UI; - art->prefsizex = 180; // XXX - art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_FRAMES; - art->listener = node_region_listener; - art->init = node_buttons_region_init; - art->draw = node_buttons_region_draw; - BLI_addhead(&st->regiontypes, art); - - node_buttons_register(art); - - /* regions: toolbar */ - art = MEM_callocN(sizeof(ARegionType), "spacetype view3d tools region"); - art->regionid = RGN_TYPE_TOOLS; - art->prefsizex = 58; /* XXX */ - art->prefsizey = 50; /* XXX */ - art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_FRAMES; - art->listener = node_region_listener; - art->message_subscribe = ED_region_generic_tools_region_message_subscribe; - art->snap_size = ED_region_generic_tools_region_snap_size; - art->init = node_toolbar_region_init; - art->draw = node_toolbar_region_draw; - BLI_addhead(&st->regiontypes, art); - - node_toolbar_register(art); - - BKE_spacetype_register(st); + SpaceType *st = MEM_callocN(sizeof(SpaceType), "spacetype node"); + ARegionType *art; + + st->spaceid = SPACE_NODE; + strncpy(st->name, "Node", BKE_ST_MAXNAME); + + st->new = node_new; + st->free = node_free; + st->init = node_init; + st->duplicate = node_duplicate; + st->operatortypes = node_operatortypes; + st->keymap = node_keymap; + st->listener = node_area_listener; + st->refresh = node_area_refresh; + st->context = node_context; + st->dropboxes = node_dropboxes; + st->gizmos = node_widgets; + st->id_remap = node_id_remap; + st->space_subtype_item_extend = node_space_subtype_item_extend; + st->space_subtype_get = node_space_subtype_get; + st->space_subtype_set = node_space_subtype_set; + + /* regions: main window */ + art = MEM_callocN(sizeof(ARegionType), "spacetype node region"); + art->regionid = RGN_TYPE_WINDOW; + art->init = node_main_region_init; + art->draw = node_main_region_draw; + art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_GIZMO | ED_KEYMAP_TOOL | ED_KEYMAP_VIEW2D | + ED_KEYMAP_FRAMES | ED_KEYMAP_GPENCIL; + art->listener = node_region_listener; + art->cursor = node_cursor; + art->event_cursor = true; + + BLI_addhead(&st->regiontypes, art); + + /* regions: header */ + art = MEM_callocN(sizeof(ARegionType), "spacetype node region"); + art->regionid = RGN_TYPE_HEADER; + art->prefsizey = HEADERY; + art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_VIEW2D | ED_KEYMAP_FRAMES | ED_KEYMAP_HEADER; + art->listener = node_region_listener; + art->init = node_header_region_init; + art->draw = node_header_region_draw; + + BLI_addhead(&st->regiontypes, art); + + /* regions: listview/buttons */ + art = MEM_callocN(sizeof(ARegionType), "spacetype node region"); + art->regionid = RGN_TYPE_UI; + art->prefsizex = 180; // XXX + art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_FRAMES; + art->listener = node_region_listener; + art->init = node_buttons_region_init; + art->draw = node_buttons_region_draw; + BLI_addhead(&st->regiontypes, art); + + node_buttons_register(art); + + /* regions: toolbar */ + art = MEM_callocN(sizeof(ARegionType), "spacetype view3d tools region"); + art->regionid = RGN_TYPE_TOOLS; + art->prefsizex = 58; /* XXX */ + art->prefsizey = 50; /* XXX */ + art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_FRAMES; + art->listener = node_region_listener; + art->message_subscribe = ED_region_generic_tools_region_message_subscribe; + art->snap_size = ED_region_generic_tools_region_snap_size; + art->init = node_toolbar_region_init; + art->draw = node_toolbar_region_draw; + BLI_addhead(&st->regiontypes, art); + + node_toolbar_register(art); + + BKE_spacetype_register(st); } |