diff options
Diffstat (limited to 'source/blender/compositor/intern/COM_Node.cc')
| -rw-r--r-- | source/blender/compositor/intern/COM_Node.cc | 210 |
1 files changed, 210 insertions, 0 deletions
diff --git a/source/blender/compositor/intern/COM_Node.cc b/source/blender/compositor/intern/COM_Node.cc new file mode 100644 index 00000000000..897d4e1df02 --- /dev/null +++ b/source/blender/compositor/intern/COM_Node.cc @@ -0,0 +1,210 @@ +/* + * 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. + * + * Copyright 2011, Blender Foundation. + */ + +#include <cstring> + +#include "BKE_node.h" + +#include "RNA_access.h" + +#include "COM_ExecutionSystem.h" +#include "COM_NodeOperation.h" +#include "COM_TranslateOperation.h" + +#include "COM_SocketProxyNode.h" + +#include "COM_defines.h" + +#include "COM_Node.h" /* own include */ + +/************** + **** Node **** + **************/ + +Node::Node(bNode *editorNode, bool create_sockets) + : m_editorNodeTree(nullptr), + m_editorNode(editorNode), + m_inActiveGroup(false), + m_instanceKey(NODE_INSTANCE_KEY_NONE) +{ + if (create_sockets) { + bNodeSocket *input = (bNodeSocket *)editorNode->inputs.first; + while (input != nullptr) { + DataType dt = COM_DT_VALUE; + if (input->type == SOCK_RGBA) { + dt = COM_DT_COLOR; + } + if (input->type == SOCK_VECTOR) { + dt = COM_DT_VECTOR; + } + + this->addInputSocket(dt, input); + input = input->next; + } + bNodeSocket *output = (bNodeSocket *)editorNode->outputs.first; + while (output != nullptr) { + DataType dt = COM_DT_VALUE; + if (output->type == SOCK_RGBA) { + dt = COM_DT_COLOR; + } + if (output->type == SOCK_VECTOR) { + dt = COM_DT_VECTOR; + } + + this->addOutputSocket(dt, output); + output = output->next; + } + } +} + +Node::~Node() +{ + while (!this->m_outputsockets.empty()) { + delete (this->m_outputsockets.back()); + this->m_outputsockets.pop_back(); + } + while (!this->m_inputsockets.empty()) { + delete (this->m_inputsockets.back()); + this->m_inputsockets.pop_back(); + } +} + +void Node::addInputSocket(DataType datatype) +{ + this->addInputSocket(datatype, nullptr); +} + +void Node::addInputSocket(DataType datatype, bNodeSocket *bSocket) +{ + NodeInput *socket = new NodeInput(this, bSocket, datatype); + this->m_inputsockets.push_back(socket); +} + +void Node::addOutputSocket(DataType datatype) +{ + this->addOutputSocket(datatype, nullptr); +} +void Node::addOutputSocket(DataType datatype, bNodeSocket *bSocket) +{ + NodeOutput *socket = new NodeOutput(this, bSocket, datatype); + this->m_outputsockets.push_back(socket); +} + +NodeOutput *Node::getOutputSocket(unsigned int index) const +{ + BLI_assert(index < this->m_outputsockets.size()); + return this->m_outputsockets[index]; +} + +NodeInput *Node::getInputSocket(unsigned int index) const +{ + BLI_assert(index < this->m_inputsockets.size()); + return this->m_inputsockets[index]; +} + +bNodeSocket *Node::getEditorInputSocket(int editorNodeInputSocketIndex) +{ + bNodeSocket *bSock = (bNodeSocket *)this->getbNode()->inputs.first; + int index = 0; + while (bSock != nullptr) { + if (index == editorNodeInputSocketIndex) { + return bSock; + } + index++; + bSock = bSock->next; + } + return nullptr; +} +bNodeSocket *Node::getEditorOutputSocket(int editorNodeOutputSocketIndex) +{ + bNodeSocket *bSock = (bNodeSocket *)this->getbNode()->outputs.first; + int index = 0; + while (bSock != nullptr) { + if (index == editorNodeOutputSocketIndex) { + return bSock; + } + index++; + bSock = bSock->next; + } + return nullptr; +} + +/******************* + **** NodeInput **** + *******************/ + +NodeInput::NodeInput(Node *node, bNodeSocket *b_socket, DataType datatype) + : m_node(node), m_editorSocket(b_socket), m_datatype(datatype), m_link(nullptr) +{ +} + +void NodeInput::setLink(NodeOutput *link) +{ + m_link = link; +} + +float NodeInput::getEditorValueFloat() +{ + PointerRNA ptr; + RNA_pointer_create((ID *)getNode()->getbNodeTree(), &RNA_NodeSocket, getbNodeSocket(), &ptr); + return RNA_float_get(&ptr, "default_value"); +} + +void NodeInput::getEditorValueColor(float *value) +{ + PointerRNA ptr; + RNA_pointer_create((ID *)getNode()->getbNodeTree(), &RNA_NodeSocket, getbNodeSocket(), &ptr); + return RNA_float_get_array(&ptr, "default_value", value); +} + +void NodeInput::getEditorValueVector(float *value) +{ + PointerRNA ptr; + RNA_pointer_create((ID *)getNode()->getbNodeTree(), &RNA_NodeSocket, getbNodeSocket(), &ptr); + return RNA_float_get_array(&ptr, "default_value", value); +} + +/******************** + **** NodeOutput **** + ********************/ + +NodeOutput::NodeOutput(Node *node, bNodeSocket *b_socket, DataType datatype) + : m_node(node), m_editorSocket(b_socket), m_datatype(datatype) +{ +} + +float NodeOutput::getEditorValueFloat() +{ + PointerRNA ptr; + RNA_pointer_create((ID *)getNode()->getbNodeTree(), &RNA_NodeSocket, getbNodeSocket(), &ptr); + return RNA_float_get(&ptr, "default_value"); +} + +void NodeOutput::getEditorValueColor(float *value) +{ + PointerRNA ptr; + RNA_pointer_create((ID *)getNode()->getbNodeTree(), &RNA_NodeSocket, getbNodeSocket(), &ptr); + return RNA_float_get_array(&ptr, "default_value", value); +} + +void NodeOutput::getEditorValueVector(float *value) +{ + PointerRNA ptr; + RNA_pointer_create((ID *)getNode()->getbNodeTree(), &RNA_NodeSocket, getbNodeSocket(), &ptr); + return RNA_float_get_array(&ptr, "default_value", value); +} |