Changeset View
Changeset View
Standalone View
Standalone View
source/blender/nodes/geometry/nodes/node_geo_sample_nearest.cc
| Show First 20 Lines • Show All 92 Lines • ▼ Show 20 Lines | |||||
| const VArray<float3> &positions, | const VArray<float3> &positions, | ||||
| const IndexMask mask, | const IndexMask mask, | ||||
| const MutableSpan<int> r_corner_indices, | const MutableSpan<int> r_corner_indices, | ||||
| const MutableSpan<float> r_distances_sq, | const MutableSpan<float> r_distances_sq, | ||||
| const MutableSpan<float3> r_positions) | const MutableSpan<float3> r_positions) | ||||
| { | { | ||||
| const Span<float3> vert_positions = mesh.vert_positions(); | const Span<float3> vert_positions = mesh.vert_positions(); | ||||
| const Span<MPoly> polys = mesh.polys(); | const Span<MPoly> polys = mesh.polys(); | ||||
| const Span<MLoop> loops = mesh.loops(); | const Span<int> corner_verts = mesh.corner_verts(); | ||||
| BLI_assert(mesh.totloop > 0); | BLI_assert(mesh.totloop > 0); | ||||
| Array<int> poly_indices(positions.size()); | Array<int> poly_indices(positions.size()); | ||||
| get_closest_mesh_polys(mesh, positions, mask, poly_indices, {}, {}); | get_closest_mesh_polys(mesh, positions, mask, poly_indices, {}, {}); | ||||
| for (const int i : mask) { | for (const int i : mask) { | ||||
| const float3 position = positions[i]; | const float3 position = positions[i]; | ||||
| const int poly_index = poly_indices[i]; | const int poly_index = poly_indices[i]; | ||||
| const MPoly &poly = polys[poly_index]; | const MPoly &poly = polys[poly_index]; | ||||
| /* Find the closest vertex in the polygon. */ | /* Find the closest vertex in the polygon. */ | ||||
| float min_distance_sq = FLT_MAX; | float min_distance_sq = FLT_MAX; | ||||
| int closest_vert_index = 0; | int closest_vert_index = 0; | ||||
| int closest_loop_index = 0; | int closest_loop_index = 0; | ||||
| for (const int loop_index : IndexRange(poly.loopstart, poly.totloop)) { | for (const int loop_index : IndexRange(poly.loopstart, poly.totloop)) { | ||||
| const MLoop &loop = loops[loop_index]; | const int vertex_index = corner_verts[loop_index]; | ||||
| const int vertex_index = loop.v; | |||||
| const float distance_sq = math::distance_squared(position, vert_positions[vertex_index]); | const float distance_sq = math::distance_squared(position, vert_positions[vertex_index]); | ||||
| if (distance_sq < min_distance_sq) { | if (distance_sq < min_distance_sq) { | ||||
| min_distance_sq = distance_sq; | min_distance_sq = distance_sq; | ||||
| closest_loop_index = loop_index; | closest_loop_index = loop_index; | ||||
| closest_vert_index = vertex_index; | closest_vert_index = vertex_index; | ||||
| } | } | ||||
| } | } | ||||
| if (!r_corner_indices.is_empty()) { | if (!r_corner_indices.is_empty()) { | ||||
| ▲ Show 20 Lines • Show All 92 Lines • Show Last 20 Lines | |||||