Changeset View
Changeset View
Standalone View
Standalone View
source/blender/nodes/geometry/nodes/node_geo_set_material_index.cc
| Show All 30 Lines | static void set_material_index_in_component(GeometryComponent &component, | ||||
| const Field<int> &index_field) | const Field<int> &index_field) | ||||
| { | { | ||||
| GeometryComponentFieldContext field_context{component, ATTR_DOMAIN_FACE}; | GeometryComponentFieldContext field_context{component, ATTR_DOMAIN_FACE}; | ||||
| const int domain_size = component.attribute_domain_size(ATTR_DOMAIN_FACE); | const int domain_size = component.attribute_domain_size(ATTR_DOMAIN_FACE); | ||||
| if (domain_size == 0) { | if (domain_size == 0) { | ||||
| return; | return; | ||||
| } | } | ||||
| fn::FieldEvaluator selection_evaluator{field_context, domain_size}; | |||||
| selection_evaluator.add(selection_field); | |||||
| selection_evaluator.evaluate(); | |||||
| const IndexMask selection = selection_evaluator.get_evaluated_as_mask(0); | |||||
| OutputAttribute_Typed<int> indices = component.attribute_try_get_for_output_only<int>( | OutputAttribute_Typed<int> indices = component.attribute_try_get_for_output_only<int>( | ||||
| "material_index", ATTR_DOMAIN_FACE); | "material_index", ATTR_DOMAIN_FACE); | ||||
| fn::FieldEvaluator material_evaluator{field_context, &selection}; | |||||
| material_evaluator.add_with_destination(index_field, indices.varray()); | fn::FieldEvaluator evaluator{field_context, domain_size}; | ||||
| material_evaluator.evaluate(); | evaluator.set_selection(selection_field); | ||||
| evaluator.add_with_destination(index_field, indices.varray()); | |||||
| evaluator.evaluate(); | |||||
| indices.save(); | indices.save(); | ||||
| } | } | ||||
| static void node_geo_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<int> index_field = params.extract_input<Field<int>>("Material Index"); | Field<int> index_field = params.extract_input<Field<int>>("Material Index"); | ||||
| Show All 24 Lines | |||||