Differential D12938 Diff 43876 source/blender/nodes/geometry/nodes/node_geo_viewer.cc

Changeset View

Standalone View

source/blender/nodes/geometry/nodes/node_geo_viewer.cc

	/*			/*
	* This program is free software; you can redistribute it and/or			* This program is free software; you can redistribute it and/or
	* modify it under the terms of the GNU General Public License			* modify it under the terms of the GNU General Public License
	* as published by the Free Software Foundation; either version 2			* as published by the Free Software Foundation; either version 2
	* of the License, or (at your option) any later version.			* of the License, or (at your option) any later version.
	*			*
	* This program is distributed in the hope that it will be useful,			* This program is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of			* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	* GNU General Public License for more details.			* GNU General Public License for more details.
	*			*
	* You should have received a copy of the GNU General Public License			* You should have received a copy of the GNU General Public License
	* along with this program; if not, write to the Free Software Foundation,			* along with this program; if not, write to the Free Software Foundation,
	* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.			* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
	*/			*/

				#include "UI_interface.h"
				#include "UI_resources.h"

	#include "node_geometry_util.hh"			#include "node_geometry_util.hh"

	namespace blender::nodes {			namespace blender::nodes {
	static void geo_node_viewer_declare(NodeDeclarationBuilder &b)			static void geo_node_viewer_declare(NodeDeclarationBuilder &b)
	{			{
	b.add_input<decl::Geometry>("Geometry");			b.add_input<decl::Geometry>("Geometry");
				b.add_input<decl::Float>("Value").supports_field().hide_value();
				b.add_input<decl::Vector>("Value").supports_field().hide_value();
				b.add_input<decl::Color>("Value").supports_field().hide_value();
				b.add_input<decl::Int>("Value").supports_field().hide_value();
				b.add_input<decl::Bool>("Value").supports_field().hide_value();
				}

				static void geo_node_viewer_init(bNodeTree UNUSED(tree), bNode node)
				{
				NodeGeometryViewer data = (NodeGeometryViewer )MEM_callocN(sizeof(NodeGeometryViewer),
				__func__);
				data->data_type = CD_PROP_FLOAT;

				node->storage = data;
	}			}

				static void geo_node_viewer_layout(uiLayout layout, bContext UNUSED(C), PointerRNA *ptr)
				{
				uiItemR(layout, ptr, "data_type", 0, "", ICON_NONE);
				}

				static eNodeSocketDatatype custom_data_type_to_socket_type(const CustomDataType type)
				{
				switch (type) {
				case CD_PROP_FLOAT:
				return SOCK_FLOAT;
				case CD_PROP_INT32:
				return SOCK_INT;
				case CD_PROP_FLOAT3:
				return SOCK_VECTOR;
				case CD_PROP_BOOL:
				return SOCK_BOOLEAN;
				case CD_PROP_COLOR:
				return SOCK_RGBA;
				default:
				BLI_assert_unreachable();
				return SOCK_FLOAT;
				}
				}

				static void geo_node_viewer_update(bNodeTree UNUSED(ntree), bNode node)
				{
				const NodeGeometryViewer &storage = (const NodeGeometryViewer )node->storage;
				const CustomDataType data_type = static_cast<CustomDataType>(storage.data_type);
				const eNodeSocketDatatype socket_type = custom_data_type_to_socket_type(data_type);

				LISTBASE_FOREACH (bNodeSocket *, socket, &node->inputs) {
				if (socket->type == SOCK_GEOMETRY) {
				continue;
				}
				nodeSetSocketAvailability(socket, socket->type == socket_type);
				}
				}

	} // namespace blender::nodes			} // namespace blender::nodes

	void register_node_type_geo_viewer()			void register_node_type_geo_viewer()
	{			{
	static bNodeType ntype;			static bNodeType ntype;

	geo_node_type_base(&ntype, GEO_NODE_VIEWER, "Viewer", NODE_CLASS_OUTPUT, 0);			geo_node_type_base(&ntype, GEO_NODE_VIEWER, "Viewer", NODE_CLASS_OUTPUT, 0);
				node_type_storage(
				&ntype, "NodeGeometryViewer", node_free_standard_storage, node_copy_standard_storage);
				node_type_update(&ntype, blender::nodes::geo_node_viewer_update);
				node_type_init(&ntype, blender::nodes::geo_node_viewer_init);
	ntype.declare = blender::nodes::geo_node_viewer_declare;			ntype.declare = blender::nodes::geo_node_viewer_declare;
				ntype.draw_buttons_ex = blender::nodes::geo_node_viewer_layout;
	nodeRegisterType(&ntype);			nodeRegisterType(&ntype);
	}			}