Page MenuHome
Differential D12689 Diff 42663 source/blender/nodes/composite/nodes/node_composite_switchview.cc

Changeset View

Standalone View

View Options

source/blender/nodes/composite/nodes/node_composite_switchview.cc

  • This file was moved from source/blender/nodes/composite/nodes/node_composite_switchview.c.
Show All 19 Lines
/** \file /** \file
* \ingroup cmpnodes * \ingroup cmpnodes
*/ */
#include "BKE_context.h" #include "BKE_context.h"
#include "BKE_lib_id.h" #include "BKE_lib_id.h"
#include "../node_composite_util.h" #include "../node_composite_util.hh"
/* **************** SWITCH VIEW ******************** */ /* **************** SWITCH VIEW ******************** */
static bNodeSocketTemplate cmp_node_switch_view_out[] = { static bNodeSocketTemplate cmp_node_switch_view_out[] = {
{SOCK_RGBA, N_("Image"), 0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 1.0f}, {SOCK_RGBA, N_("Image"), 0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 1.0f},
{-1, ""}, {-1, ""},
}; };
static bNodeSocket *ntreeCompositSwitchViewAddSocket(bNodeTree *ntree, static bNodeSocket *ntreeCompositSwitchViewAddSocket(bNodeTree *ntree,
bNode *node, bNode *node,
const char *name) const char *name)
{ {
bNodeSocket *sock = nodeAddStaticSocket(ntree, node, SOCK_IN, SOCK_RGBA, PROP_NONE, NULL, name); bNodeSocket *sock = nodeAddStaticSocket(
ntree, node, SOCK_IN, SOCK_RGBA, PROP_NONE, nullptr, name);
return sock; return sock;
} }
static void cmp_node_switch_view_sanitycheck(bNodeTree *ntree, bNode *node) static void cmp_node_switch_view_sanitycheck(bNodeTree *ntree, bNode *node)
{ {
bNodeSocket *sock;
if (!BLI_listbase_is_empty(&node->inputs)) { if (!BLI_listbase_is_empty(&node->inputs)) {
return; return;
} }
sock = ntreeCompositSwitchViewAddSocket(ntree, node, "No View"); bNodeSocket *sock = ntreeCompositSwitchViewAddSocket(ntree, node, "No View");
sock->flag |= SOCK_HIDDEN; sock->flag |= SOCK_HIDDEN;
} }
static void cmp_node_switch_view_update(bNodeTree *ntree, bNode *node) static void cmp_node_switch_view_update(bNodeTree *ntree, bNode *node)
{ {
bNodeSocket *sock;
SceneRenderView *srv;
Scene *scene = (Scene *)node->id; Scene *scene = (Scene *)node->id;
/* only update when called from the operator button */ /* only update when called from the operator button */
if (node->update != NODE_UPDATE_OPERATOR) { if (node->update != NODE_UPDATE_OPERATOR) {
return; return;
} }
if (scene == NULL) { if (scene == nullptr) {
nodeRemoveAllSockets(ntree, node); nodeRemoveAllSockets(ntree, node);
/* make sure there is always one socket */ /* make sure there is always one socket */
cmp_node_switch_view_sanitycheck(ntree, node); cmp_node_switch_view_sanitycheck(ntree, node);
return; return;
} }
/* remove the views that were removed */ /* remove the views that were removed */
sock = node->inputs.last; bNodeSocket *sock = (bNodeSocket *)node->inputs.last;
while (sock) { while (sock) {
srv = BLI_findstring(&scene->r.views, sock->name, offsetof(SceneRenderView, name)); SceneRenderView *srv = (SceneRenderView *)BLI_findstring(
&scene->r.views, sock->name, offsetof(SceneRenderView, name));
if (srv == NULL) { if (srv == nullptr) {
bNodeSocket *sock_del = sock; bNodeSocket *sock_del = sock;
sock = sock->prev; sock = sock->prev;
nodeRemoveSocket(ntree, node, sock_del); nodeRemoveSocket(ntree, node, sock_del);
} }
else { else {
if (srv->viewflag & SCE_VIEW_DISABLE) { if (srv->viewflag & SCE_VIEW_DISABLE) {
sock->flag |= SOCK_HIDDEN; sock->flag |= SOCK_HIDDEN;
} }
else { else {
sock->flag &= ~SOCK_HIDDEN; sock->flag &= ~SOCK_HIDDEN;
} }
sock = sock->prev; sock = sock->prev;
} }
} }
/* add the new views */ /* add the new views */
for (srv = scene->r.views.first; srv; srv = srv->next) { LISTBASE_FOREACH (SceneRenderView *, srv, &scene->r.views) {
sock = BLI_findstring(&node->inputs, srv->name, offsetof(bNodeSocket, name)); sock = (bNodeSocket *)BLI_findstring(&node->inputs, srv->name, offsetof(bNodeSocket, name));
if (sock == NULL) { if (sock == nullptr) {
sock = ntreeCompositSwitchViewAddSocket(ntree, node, srv->name); sock = ntreeCompositSwitchViewAddSocket(ntree, node, srv->name);
} }
if (srv->viewflag & SCE_VIEW_DISABLE) { if (srv->viewflag & SCE_VIEW_DISABLE) {
sock->flag |= SOCK_HIDDEN; sock->flag |= SOCK_HIDDEN;
} }
else { else {
sock->flag &= ~SOCK_HIDDEN; sock->flag &= ~SOCK_HIDDEN;
} }
} }
/* make sure there is always one socket */ /* make sure there is always one socket */
cmp_node_switch_view_sanitycheck(ntree, node); cmp_node_switch_view_sanitycheck(ntree, node);
} }
static void init_switch_view(const bContext *C, PointerRNA *ptr) static void init_switch_view(const bContext *C, PointerRNA *ptr)
{ {
Scene *scene = CTX_data_scene(C); Scene *scene = CTX_data_scene(C);
bNodeTree *ntree = (bNodeTree *)ptr->owner_id; bNodeTree *ntree = (bNodeTree *)ptr->owner_id;
bNode *node = ptr->data; bNode *node = (bNode *)ptr->data;
SceneRenderView *srv;
bNodeSocket *sock;
int nr;
/* store scene for updates */ /* store scene for updates */
node->id = (ID *)scene; node->id = (ID *)scene;
id_us_plus(node->id); id_us_plus(node->id);
if (scene) { if (scene) {
RenderData *rd = &scene->r; RenderData *rd = &scene->r;
for (nr = 0, srv = rd->views.first; srv; srv = srv->next, nr++) { LISTBASE_FOREACH (SceneRenderView *, srv, &rd->views) {
sock = ntreeCompositSwitchViewAddSocket(ntree, node, srv->name); bNodeSocket *sock = ntreeCompositSwitchViewAddSocket(ntree, node, srv->name);
if (srv->viewflag & SCE_VIEW_DISABLE) { if (srv->viewflag & SCE_VIEW_DISABLE) {
sock->flag |= SOCK_HIDDEN; sock->flag |= SOCK_HIDDEN;
} }
} }
} }
/* make sure there is always one socket */ /* make sure there is always one socket */
cmp_node_switch_view_sanitycheck(ntree, node); cmp_node_switch_view_sanitycheck(ntree, node);
} }
void register_node_type_cmp_switch_view(void) void register_node_type_cmp_switch_view(void)
{ {
static bNodeType ntype; static bNodeType ntype;
cmp_node_type_base(&ntype, CMP_NODE_SWITCH_VIEW, "Switch View", NODE_CLASS_CONVERTER, 0); cmp_node_type_base(&ntype, CMP_NODE_SWITCH_VIEW, "Switch View", NODE_CLASS_CONVERTER, 0);
node_type_socket_templates(&ntype, NULL, cmp_node_switch_view_out); node_type_socket_templates(&ntype, nullptr, cmp_node_switch_view_out);
ntype.initfunc_api = init_switch_view; ntype.initfunc_api = init_switch_view;
node_type_update(&ntype, cmp_node_switch_view_update); node_type_update(&ntype, cmp_node_switch_view_update);
nodeRegisterType(&ntype); nodeRegisterType(&ntype);
} }