
When nodes are loaded from a .blend file they can potentially have undefined types. This can happen if a type has been deprecated and removed, or if node types were defined in a python script that has not been loaded correctly. Previously all such nodes would automatically be removed from a node tree, assuming that their types were deprecated and no longer in use (more commonly caused by loading new nodes in an older Blender version). Due to the possibility of dynamic registration it is no longer feasible to simply delete such nodes. Display and handling of node trees was simply disabled before this patch, so that a node tree where any node or socket type was undefined would not be displayed at all. To give more information and avoid problems caused by necessary checks for the typeinfo pointer, there is now a 'Undefined' fallback type for trees, nodes and sockets. These types are used as placeholders in case the real type is not registered and can provide useful visual feedback on undefined nodes.
817 lines
20 KiB
C
817 lines
20 KiB
C
/*
|
|
* ***** BEGIN GPL LICENSE BLOCK *****
|
|
*
|
|
* 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.
|
|
*
|
|
*
|
|
* Contributor(s): Blender Foundation
|
|
*
|
|
* ***** END GPL LICENSE BLOCK *****
|
|
*/
|
|
|
|
/** \file blender/editors/space_node/space_node.c
|
|
* \ingroup spnode
|
|
*/
|
|
|
|
|
|
|
|
#include "DNA_lamp_types.h"
|
|
#include "DNA_material_types.h"
|
|
#include "DNA_node_types.h"
|
|
#include "DNA_world_types.h"
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
#include "BLI_blenlib.h"
|
|
#include "BLI_math.h"
|
|
|
|
#include "BKE_context.h"
|
|
#include "BKE_screen.h"
|
|
#include "BKE_node.h"
|
|
|
|
#include "ED_space_api.h"
|
|
#include "ED_node.h"
|
|
#include "ED_render.h"
|
|
#include "ED_screen.h"
|
|
#include "ED_node.h"
|
|
#include "WM_api.h"
|
|
#include "WM_types.h"
|
|
|
|
#include "UI_resources.h"
|
|
#include "UI_view2d.h"
|
|
|
|
#include "RNA_access.h"
|
|
|
|
#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);
|
|
}
|
|
snode->treepath.first = snode->treepath.last = NULL;
|
|
|
|
if (ntree) {
|
|
path = MEM_callocN(sizeof(bNodeTreePath), "node tree path");
|
|
path->nodetree = ntree;
|
|
path->parent_key = NODE_INSTANCE_KEY_BASE;
|
|
if (id)
|
|
BLI_strncpy(path->node_name, id->name + 2, sizeof(path->node_name));
|
|
BLI_addtail(&snode->treepath, path);
|
|
}
|
|
|
|
/* update current tree */
|
|
snode->nodetree = snode->edittree = ntree;
|
|
snode->id = id;
|
|
snode->from = from;
|
|
|
|
/* listener updates the View2D center from edittree */
|
|
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;
|
|
|
|
BLI_strncpy(path->node_name, gnode->name, sizeof(path->node_name));
|
|
}
|
|
else
|
|
path->parent_key = NODE_INSTANCE_KEY_BASE;
|
|
|
|
BLI_addtail(&snode->treepath, path);
|
|
|
|
/* update current tree */
|
|
snode->edittree = ntree;
|
|
|
|
/* listener updates the View2D center from edittree */
|
|
WM_main_add_notifier(NC_SCENE | ND_NODES, NULL);
|
|
}
|
|
|
|
void ED_node_tree_pop(SpaceNode *snode)
|
|
{
|
|
bNodeTreePath *path = snode->treepath.last;
|
|
|
|
/* don't remove root */
|
|
if (path == snode->treepath.first)
|
|
return;
|
|
|
|
BLI_remlink(&snode->treepath, path);
|
|
MEM_freeN(path);
|
|
|
|
/* update current tree */
|
|
path = snode->treepath.last;
|
|
snode->edittree = path->nodetree;
|
|
|
|
/* 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_countlist(&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;
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
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;
|
|
}
|
|
}
|
|
}
|
|
|
|
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) {
|
|
BLI_strncpy(value, path->node_name, max_length);
|
|
size = strlen(path->node_name);
|
|
}
|
|
else {
|
|
BLI_snprintf(value, max_length, "/%s", path->node_name);
|
|
size = strlen(path->node_name) + 1;
|
|
}
|
|
max_length -= size;
|
|
if (max_length <= 0)
|
|
break;
|
|
value += size;
|
|
}
|
|
}
|
|
|
|
void snode_group_offset(SpaceNode *snode, float *x, float *y)
|
|
{
|
|
bNodeTreePath *path = snode->treepath.last;
|
|
float cx, cy;
|
|
|
|
if (path) {
|
|
cx = path->nodetree->view_center[0];
|
|
cy = path->nodetree->view_center[1];
|
|
|
|
if (path->prev) {
|
|
*x = cx - path->prev->nodetree->view_center[0];
|
|
*y = cy - path->prev->nodetree->view_center[1];
|
|
return;
|
|
}
|
|
}
|
|
|
|
*x = *y = 0.0f;
|
|
}
|
|
|
|
/* ******************** manage regions ********************* */
|
|
|
|
ARegion *node_has_buttons_region(ScrArea *sa)
|
|
{
|
|
ARegion *ar, *arnew;
|
|
|
|
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);
|
|
|
|
/* is error! */
|
|
if (ar == NULL) return NULL;
|
|
|
|
arnew = MEM_callocN(sizeof(ARegion), "buttons for node");
|
|
|
|
BLI_insertlinkafter(&sa->regionbase, ar, arnew);
|
|
arnew->regiontype = RGN_TYPE_UI;
|
|
arnew->alignment = RGN_ALIGN_RIGHT;
|
|
|
|
arnew->flag = RGN_FLAG_HIDDEN;
|
|
|
|
return arnew;
|
|
}
|
|
|
|
ARegion *node_has_tools_region(ScrArea *sa)
|
|
{
|
|
ARegion *ar, *arnew;
|
|
|
|
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);
|
|
|
|
/* is error! */
|
|
if (ar == NULL) return NULL;
|
|
|
|
arnew = MEM_callocN(sizeof(ARegion), "node tools");
|
|
|
|
BLI_insertlinkafter(&sa->regionbase, ar, arnew);
|
|
arnew->regiontype = RGN_TYPE_TOOLS;
|
|
arnew->alignment = RGN_ALIGN_LEFT;
|
|
|
|
arnew->flag = RGN_FLAG_HIDDEN;
|
|
|
|
return arnew;
|
|
}
|
|
|
|
/* ******************** default callbacks for node space ***************** */
|
|
|
|
static SpaceLink *node_new(const bContext *UNUSED(C))
|
|
{
|
|
ARegion *ar;
|
|
SpaceNode *snode;
|
|
|
|
snode = MEM_callocN(sizeof(SpaceNode), "initnode");
|
|
snode->spacetype = SPACE_NODE;
|
|
|
|
snode->flag = SNODE_SHOW_GPENCIL | SNODE_USE_ALPHA;
|
|
|
|
/* 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
|
|
|
|
/* header */
|
|
ar = MEM_callocN(sizeof(ARegion), "header for node");
|
|
|
|
BLI_addtail(&snode->regionbase, ar);
|
|
ar->regiontype = RGN_TYPE_HEADER;
|
|
ar->alignment = RGN_ALIGN_BOTTOM;
|
|
|
|
/* 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;
|
|
|
|
/* main area */
|
|
ar = MEM_callocN(sizeof(ARegion), "main area for node");
|
|
|
|
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.cur = ar->v2d.tot;
|
|
|
|
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.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;
|
|
|
|
return (SpaceLink *)snode;
|
|
}
|
|
|
|
static void node_free(SpaceLink *sl)
|
|
{
|
|
SpaceNode *snode = (SpaceNode *)sl;
|
|
bNodeTreePath *path, *path_next;
|
|
|
|
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(ScrArea *sa, wmNotifier *wmn)
|
|
{
|
|
/* note, ED_area_tag_refresh will re-execute compositor */
|
|
SpaceNode *snode = sa->spacedata.first;
|
|
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);
|
|
/* shift view to node tree center */
|
|
if (ar && snode->edittree)
|
|
UI_view2d_setcenter(&ar->v2d, snode->edittree->view_center[0], snode->edittree->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;
|
|
}
|
|
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;
|
|
}
|
|
}
|
|
|
|
static void node_area_refresh(const struct bContext *C, ScrArea *sa)
|
|
{
|
|
/* default now: refresh node is starting preview */
|
|
SpaceNode *snode = sa->spacedata.first;
|
|
|
|
ED_preview_kill_jobs(C);
|
|
|
|
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) {
|
|
Lamp *la = (Lamp *)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);
|
|
|
|
/* clear or remove stuff from old */
|
|
snoden->nodetree = NULL;
|
|
snoden->linkdrag.first = snoden->linkdrag.last = NULL;
|
|
|
|
BLI_duplicatelist(&snoden->treepath, &snode->treepath);
|
|
|
|
return (SpaceLink *)snoden;
|
|
}
|
|
|
|
|
|
/* add handlers, stuff you only do once or on area/region changes */
|
|
static void node_buttons_area_init(wmWindowManager *wm, ARegion *ar)
|
|
{
|
|
wmKeyMap *keymap;
|
|
|
|
ED_region_panels_init(wm, ar);
|
|
|
|
keymap = WM_keymap_find(wm->defaultconf, "Node Generic", SPACE_NODE, 0);
|
|
WM_event_add_keymap_handler(&ar->handlers, keymap);
|
|
}
|
|
|
|
static void node_buttons_area_draw(const bContext *C, ARegion *ar)
|
|
{
|
|
ED_region_panels(C, ar, 1, NULL, -1);
|
|
}
|
|
|
|
/* add handlers, stuff you only do once or on area/region changes */
|
|
static void node_toolbar_area_init(wmWindowManager *wm, ARegion *ar)
|
|
{
|
|
wmKeyMap *keymap;
|
|
|
|
ED_region_panels_init(wm, ar);
|
|
|
|
keymap = WM_keymap_find(wm->defaultconf, "Node Generic", SPACE_NODE, 0);
|
|
WM_event_add_keymap_handler(&ar->handlers, keymap);
|
|
}
|
|
|
|
static void node_toolbar_area_draw(const bContext *C, ARegion *ar)
|
|
{
|
|
ED_region_panels(C, ar, 1, NULL, -1);
|
|
}
|
|
|
|
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]);
|
|
|
|
node_set_cursor(win, snode);
|
|
}
|
|
|
|
/* Initialize main area, setting handlers. */
|
|
static void node_main_area_init(wmWindowManager *wm, ARegion *ar)
|
|
{
|
|
wmKeyMap *keymap;
|
|
ListBase *lb;
|
|
|
|
UI_view2d_region_reinit(&ar->v2d, V2D_COMMONVIEW_CUSTOM, ar->winx, ar->winy);
|
|
|
|
/* own keymaps */
|
|
keymap = WM_keymap_find(wm->defaultconf, "Node Generic", SPACE_NODE, 0);
|
|
WM_event_add_keymap_handler(&ar->handlers, keymap);
|
|
|
|
keymap = WM_keymap_find(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);
|
|
|
|
WM_event_add_dropbox_handler(&ar->handlers, lb);
|
|
}
|
|
|
|
static void node_main_area_draw(const bContext *C, ARegion *ar)
|
|
{
|
|
drawnodespace(C, ar);
|
|
}
|
|
|
|
|
|
/* ************* dropboxes ************* */
|
|
|
|
static int node_drop_poll(bContext *UNUSED(C), wmDrag *drag, const wmEvent *UNUSED(event))
|
|
{
|
|
if (drag->type == WM_DRAG_ID) {
|
|
ID *id = (ID *)drag->poin;
|
|
if (GS(id->name) == ID_IM)
|
|
return 1;
|
|
}
|
|
else if (drag->type == WM_DRAG_PATH) {
|
|
if (ELEM(drag->icon, 0, ICON_FILE_IMAGE)) /* rule might not work? */
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static void node_id_path_drop_copy(wmDrag *drag, wmDropBox *drop)
|
|
{
|
|
ID *id = (ID *)drag->poin;
|
|
|
|
if (id) {
|
|
RNA_string_set(drop->ptr, "name", id->name + 2);
|
|
}
|
|
if (drag->path[0]) {
|
|
RNA_string_set(drop->ptr, "filepath", drag->path);
|
|
}
|
|
}
|
|
|
|
/* 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_drop_poll, node_id_path_drop_copy);
|
|
|
|
}
|
|
|
|
/* ************* end drop *********** */
|
|
|
|
|
|
/* add handlers, stuff you only do once or on area/region changes */
|
|
static void node_header_area_init(wmWindowManager *UNUSED(wm), ARegion *ar)
|
|
{
|
|
ED_region_header_init(ar);
|
|
}
|
|
|
|
static void node_header_area_draw(const bContext *C, ARegion *ar)
|
|
{
|
|
/* find and set the context */
|
|
snode_set_context(C);
|
|
|
|
ED_region_header(C, ar);
|
|
}
|
|
|
|
/* used for header + main area */
|
|
static void node_region_listener(ARegion *ar, wmNotifier *wmn)
|
|
{
|
|
/* context changes */
|
|
switch (wmn->category) {
|
|
case NC_SPACE:
|
|
if (wmn->data == ND_SPACE_NODE)
|
|
ED_region_tag_redraw(ar);
|
|
break;
|
|
case NC_SCREEN:
|
|
switch (wmn->data) {
|
|
case ND_SCREENCAST:
|
|
case ND_ANIMPLAY:
|
|
ED_region_tag_redraw(ar);
|
|
break;
|
|
}
|
|
break;
|
|
case NC_SCENE:
|
|
case NC_MATERIAL:
|
|
case NC_TEXTURE:
|
|
case NC_WORLD:
|
|
case NC_NODE:
|
|
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);
|
|
break;
|
|
}
|
|
}
|
|
|
|
const char *node_context_dir[] = {"selected_nodes", "active_node", 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;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/* 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;
|
|
|
|
/* regions: main window */
|
|
art = MEM_callocN(sizeof(ARegionType), "spacetype node region");
|
|
art->regionid = RGN_TYPE_WINDOW;
|
|
art->init = node_main_area_init;
|
|
art->draw = node_main_area_draw;
|
|
art->listener = node_region_listener;
|
|
art->cursor = node_cursor;
|
|
art->event_cursor = TRUE;
|
|
art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_VIEW2D | ED_KEYMAP_FRAMES | ED_KEYMAP_GPENCIL;
|
|
|
|
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_area_init;
|
|
art->draw = node_header_area_draw;
|
|
|
|
BLI_addhead(&st->regiontypes, art);
|
|
|
|
node_menus_register();
|
|
|
|
/* 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_area_init;
|
|
art->draw = node_buttons_area_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 = 160; /* XXX */
|
|
art->prefsizey = 50; /* XXX */
|
|
art->keymapflag = ED_KEYMAP_UI | ED_KEYMAP_FRAMES;
|
|
art->listener = node_region_listener;
|
|
art->init = node_toolbar_area_init;
|
|
art->draw = node_toolbar_area_draw;
|
|
BLI_addhead(&st->regiontypes, art);
|
|
|
|
node_toolbar_register(art);
|
|
|
|
BKE_spacetype_register(st);
|
|
}
|
|
|