Differential D13337 Diff 45169 source/blender/nodes/geometry/nodes/node_geo_set_shade_smooth.cc

Changeset View

Standalone View

source/blender/nodes/geometry/nodes/node_geo_set_shade_smooth.cc

Show All 12 Lines
* 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 "node_geometry_util.hh"		#include "node_geometry_util.hh"

namespace blender::nodes::node_geo_set_shade_smooth_cc {		namespace blender::nodes::node_geo_set_shade_smooth_cc {

static void geo_node_set_shade_smooth_declare(NodeDeclarationBuilder &b)		static void node_declare(NodeDeclarationBuilder &b)
{		{
b.add_input<decl::Geometry>(N_("Geometry")).supported_type(GEO_COMPONENT_TYPE_MESH);		b.add_input<decl::Geometry>(N_("Geometry")).supported_type(GEO_COMPONENT_TYPE_MESH);
b.add_input<decl::Bool>(N_("Selection")).default_value(true).hide_value().supports_field();		b.add_input<decl::Bool>(N_("Selection")).default_value(true).hide_value().supports_field();
b.add_input<decl::Bool>(N_("Shade Smooth")).supports_field().default_value(true);		b.add_input<decl::Bool>(N_("Shade Smooth")).supports_field().default_value(true);
b.add_output<decl::Geometry>(N_("Geometry"));		b.add_output<decl::Geometry>(N_("Geometry"));
}		}

static void set_smooth_in_component(GeometryComponent &component,		static void set_smooth_in_component(GeometryComponent &component,
Show All 14 Lines	static void set_smooth_in_component(GeometryComponent &component,
OutputAttribute_Typed<bool> shades = component.attribute_try_get_for_output_only<bool>(		OutputAttribute_Typed<bool> shades = component.attribute_try_get_for_output_only<bool>(
"shade_smooth", ATTR_DOMAIN_FACE);		"shade_smooth", ATTR_DOMAIN_FACE);
fn::FieldEvaluator shade_evaluator{field_context, &selection};		fn::FieldEvaluator shade_evaluator{field_context, &selection};
shade_evaluator.add_with_destination(shade_field, shades.varray());		shade_evaluator.add_with_destination(shade_field, shades.varray());
shade_evaluator.evaluate();		shade_evaluator.evaluate();
shades.save();		shades.save();
}		}

static void geo_node_set_shade_smooth_exec(GeoNodeExecParams params)		static void node_geo_exec(GeoNodeExecParams params)
{		{
GeometrySet geometry_set = params.extract_input<GeometrySet>("Geometry");		GeometrySet geometry_set = params.extract_input<GeometrySet>("Geometry");
Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");		Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");
Field<bool> shade_field = params.extract_input<Field<bool>>("Shade Smooth");		Field<bool> shade_field = params.extract_input<Field<bool>>("Shade Smooth");

geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {		geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {
if (geometry_set.has_mesh()) {		if (geometry_set.has_mesh()) {
set_smooth_in_component(		set_smooth_in_component(
geometry_set.get_component_for_write<MeshComponent>(), selection_field, shade_field);		geometry_set.get_component_for_write<MeshComponent>(), selection_field, shade_field);
}		}
});		});
params.set_output("Geometry", std::move(geometry_set));		params.set_output("Geometry", std::move(geometry_set));
}		}

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

void register_node_type_geo_set_shade_smooth()		void register_node_type_geo_set_shade_smooth()
{		{
namespace file_ns = blender::nodes::node_geo_set_shade_smooth_cc;		namespace file_ns = blender::nodes::node_geo_set_shade_smooth_cc;

static bNodeType ntype;		static bNodeType ntype;

geo_node_type_base(		geo_node_type_base(
&ntype, GEO_NODE_SET_SHADE_SMOOTH, "Set Shade Smooth", NODE_CLASS_GEOMETRY, 0);		&ntype, GEO_NODE_SET_SHADE_SMOOTH, "Set Shade Smooth", NODE_CLASS_GEOMETRY, 0);
ntype.geometry_node_execute = file_ns::geo_node_set_shade_smooth_exec;		ntype.geometry_node_execute = file_ns::node_geo_exec;
ntype.declare = file_ns::geo_node_set_shade_smooth_declare;		ntype.declare = file_ns::node_declare;
nodeRegisterType(&ntype);		nodeRegisterType(&ntype);
}		}