[Bf-blender-cvs] [642fa43b136] functions: optimize get position on surface node

[Jacques Lucke]

Commit: 642fa43b136330fcfdf3adc3308122bfab8639ac
Author: Jacques Lucke
Date:   Wed Dec 11 14:10:55 2019 +0100
Branches: functions
https://developer.blender.org/rB642fa43b136330fcfdf3adc3308122bfab8639ac

optimize get position on surface node

===================================================================

M	source/blender/functions/intern/multi_functions/mixed.cc

===================================================================

diff --git a/source/blender/functions/intern/multi_functions/mixed.cc b/source/blender/functions/intern/multi_functions/mixed.cc
index c465eb7af43..0e3f00a4339 100644
--- a/source/blender/functions/intern/multi_functions/mixed.cc
+++ b/source/blender/functions/intern/multi_functions/mixed.cc
@@ -271,7 +271,7 @@ MF_GetPositionOnSurface::MF_GetPositionOnSurface()
 
 void MF_GetPositionOnSurface::call(MFMask mask, MFParams params, MFContext context) const
 {
-  VirtualListRef<SurfaceHook> locations = params.readonly_single_input<SurfaceHook>(
+  VirtualListRef<SurfaceHook> surface_hooks = params.readonly_single_input<SurfaceHook>(
       0, "Surface Hook");
   MutableArrayRef<float3> r_positions = params.uninitialized_single_output<float3>(1, "Position");
 
@@ -283,40 +283,47 @@ void MF_GetPositionOnSurface::call(MFMask mask, MFParams params, MFContext conte
     return;
   }
 
-  for (uint i : mask.indices()) {
-    SurfaceHook location = locations[i];
-    if (location.type() != BKE::SurfaceHookType::MeshObject) {
-      r_positions[i] = fallback;
-      continue;
-    }
+  group_indices_by_same_value(
+      mask.indices(),
+      surface_hooks,
+      [&](SurfaceHook base_hook, ArrayRef<uint> indices_on_same_surface) {
+        if (base_hook.type() != BKE::SurfaceHookType::MeshObject) {
+          r_positions.fill_indices(indices_on_same_surface, fallback);
+          return;
+        }
 
-    Object *object = id_handle_lookup->lookup(location.object_handle());
-    if (object == nullptr) {
-      r_positions[i] = fallback;
-      continue;
-    }
+        Object *object = id_handle_lookup->lookup(base_hook.object_handle());
+        if (object == nullptr) {
+          r_positions.fill_indices(indices_on_same_surface, fallback);
+          return;
+        }
 
-    Mesh *mesh = (Mesh *)object->data;
-    const MLoopTri *triangles = BKE_mesh_runtime_looptri_ensure(mesh);
-    int triangle_amount = BKE_mesh_runtime_looptri_len(mesh);
+        Mesh *mesh = (Mesh *)object->data;
+        const MLoopTri *triangles = BKE_mesh_runtime_looptri_ensure(mesh);
+        int triangle_amount = BKE_mesh_runtime_looptri_len(mesh);
 
-    if (location.triangle_index() >= triangle_amount) {
-      r_positions[i] = fallback;
-      continue;
-    }
+        for (uint i : indices_on_same_surface) {
+          SurfaceHook hook = surface_hooks[i];
 
-    const MLoopTri &triangle = triangles[location.triangle_index()];
-    float3 v1 = mesh->mvert[mesh->mloop[triangle.tri[0]].v].co;
-    float3 v2 = mesh->mvert[mesh->mloop[triangle.tri[1]].v].co;
-    float3 v3 = mesh->mvert[mesh->mloop[triangle.tri[2]].v].co;
+          if (hook.triangle_index() >= triangle_amount) {
+            r_positions[i] = fallback;
+            continue;
+          }
 
-    float3 position;
-    interp_v3_v3v3v3(position, v1, v2, v3, location.bary_coords());
-    float4x4 local_to_world = object->obmat;
-    position = local_to_world.transform_position(position);
+          const MLoopTri &triangle = triangles[hook.triangle_index()];
+          float3 v1 = mesh->mvert[mesh->mloop[triangle.tri[0]].v].co;
+          float3 v2 = mesh->mvert[mesh->mloop[triangle.tri[1]].v].co;
+          float3 v3 = mesh->mvert[mesh->mloop[triangle.tri[2]].v].co;
 
-    r_positions[i] = position;
-  }
+          float3 position;
+          interp_v3_v3v3v3(position, v1, v2, v3, hook.bary_coords());
+          float4x4 local_to_world = object->obmat;
+          position = local_to_world.transform_position(position);
+
+          r_positions[i] = position;
+        }
+      },
+      SurfaceHook::on_same_surface);
 }
 
 MF_GetNormalOnSurface::MF_GetNormalOnSurface()