Differential D9861 Diff 32001 source/blender/nodes/geometry/nodes/node_geo_attribute_color_ramp.cc

Changeset View

Standalone View

source/blender/nodes/geometry/nodes/node_geo_attribute_color_ramp.cc

This file was added.

				/*
				* 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.
				*/

				#include "node_geometry_util.hh"

				#include "BKE_colorband.h"

				static bNodeSocketTemplate geo_node_attribute_color_ramp_in[] = {
				{SOCK_GEOMETRY, N_("Geometry")},
				{SOCK_STRING, N_("Attribute")},
				{SOCK_STRING, N_("Result")},
				{-1, ""},
				};

				static bNodeSocketTemplate geo_node_attribute_color_ramp_out[] = {
				{SOCK_GEOMETRY, N_("Geometry")},
				{-1, ""},
				};

				namespace blender::nodes {

				static void execute_on_component(const GeoNodeExecParams &params, GeometryComponent &component)
				{
				const bNode &bnode = params.node();
				NodeAttributeColorRamp node_storage = (NodeAttributeColorRamp )bnode.storage;

				const std::string result_name = params.get_input<std::string>("Result");
				const AttributeDomain result_domain = ATTR_DOMAIN_POINT;
				const CustomDataType result_type = CD_PROP_COLOR;

				WriteAttributePtr attribute_result = component.attribute_try_ensure_for_write(
				result_name, result_domain, result_type);
				if (!attribute_result) {
				return;
				}

				Color4fWriteAttribute attribute_out = std::move(attribute_result);

				const std::string input_name = params.get_input<std::string>("Attribute");
				FloatReadAttribute attribute_in = component.attribute_get_for_read<float>(
				input_name, result_domain, 0.0f);

				Span<float> data_in = attribute_in.get_span();
				MutableSpan<Color4f> data_out = attribute_out.get_span();

				ColorBand *color_ramp = &node_storage->color_ramp;
				for (const int i : data_in.index_range()) {
				BKE_colorband_evaluate(color_ramp, data_in[i], data_out[i]);
				}

				attribute_out.apply_span();
				}

				static void geo_node_attribute_color_ramp_exec(GeoNodeExecParams params)
				{
				GeometrySet geometry_set = params.extract_input<GeometrySet>("Geometry");

				if (geometry_set.has<MeshComponent>()) {
				execute_on_component(params, geometry_set.get_component_for_write<MeshComponent>());
				}
				if (geometry_set.has<PointCloudComponent>()) {
				execute_on_component(params, geometry_set.get_component_for_write<PointCloudComponent>());
				}

				params.set_output("Geometry", std::move(geometry_set));
				}

				static void geo_node_attribute_color_ramp_init(bNodeTree UNUSED(ntree), bNode node)
				{
				NodeAttributeColorRamp node_storage = (NodeAttributeColorRamp )MEM_callocN(
				sizeof(NodeAttributeColorRamp), __func__);
				BKE_colorband_init(&node_storage->color_ramp, true);
				node->storage = node_storage;
				}

				} // namespace blender::nodes

				void register_node_type_geo_attribute_color_ramp()
				{
				static bNodeType ntype;

				geo_node_type_base(
				&ntype, GEO_NODE_ATTRIBUTE_COLOR_RAMP, "Attribute Color Ramp", NODE_CLASS_ATTRIBUTE, 0);
				node_type_socket_templates(
				&ntype, geo_node_attribute_color_ramp_in, geo_node_attribute_color_ramp_out);
				node_type_storage(
				&ntype, "NodeAttributeColorRamp", node_free_standard_storage, node_copy_standard_storage);
				node_type_init(&ntype, blender::nodes::geo_node_attribute_color_ramp_init);
				node_type_size_preset(&ntype, NODE_SIZE_LARGE);
				ntype.geometry_node_execute = blender::nodes::geo_node_attribute_color_ramp_exec;
				nodeRegisterType(&ntype);
				}