From 3f70c9b2c4383ab30472bd18348cfaf457d44e02 Mon Sep 17 00:00:00 2001
From: xiangqian <qian.xiang@njtf.cn>
Date: Mon, 2 Feb 2026 11:18:08 +0800
Subject: [PATCH 1/2] [BUG]fix bugs in make_sdf function

---
 ppsci/geometry/mesh.py | 56 +++++++++++++++++++-----------------------
 1 file changed, 25 insertions(+), 31 deletions(-)

diff --git a/ppsci/geometry/mesh.py b/ppsci/geometry/mesh.py
index 2a5b65cb6..f76298fcc 100644
--- a/ppsci/geometry/mesh.py
+++ b/ppsci/geometry/mesh.py
@@ -1360,44 +1360,38 @@ def sample_in_triangle(v0, v1, v2, n, random="pseudo", criteria=None):
 
 
 def make_sdf(vectors: np.ndarray):
+    verts = vectors.reshape(-1, 3)
+    v_min = verts.min(axis=0)
+    v_max = verts.max(axis=0)
+    max_dis = np.max(v_max - v_min)
+    norm_mesh = (verts - v_min) / max_dis
+
+    mesh_vertices = [tuple(v) for v in norm_mesh.tolist()]
+    mesh_indices = np.arange(len(mesh_vertices), dtype=np.int32)
+
     def sdf_func(points: np.ndarray, compute_sdf_derivatives=False):
-        points = points.copy()
-        x_min, y_min, z_min = np.min(points, axis=0)
-        x_max, y_max, z_max = np.max(points, axis=0)
-        max_dis = max(max((x_max - x_min), (y_max - y_min)), (z_max - z_min))
-        store_triangles = vectors.copy()
-        store_triangles[:, :, 0] -= x_min
-        store_triangles[:, :, 1] -= y_min
-        store_triangles[:, :, 2] -= z_min
-        store_triangles *= 1 / max_dis
-        store_triangles = store_triangles.reshape([-1, 3])
-        points[:, 0] -= x_min
-        points[:, 1] -= y_min
-        points[:, 2] -= z_min
-        points *= 1 / max_dis
-        points = points.astype(np.float64).ravel()
+        pts = (points - v_min) / max_dis
 
-        # compute sdf values
-        sdf = sdf_module.signed_distance_field(
-            store_triangles,
-            np.arange((store_triangles.shape[0])),
-            points,
+        input_points = [tuple(p) for p in pts.tolist()]
+
+        ret = sdf_module.signed_distance_field(
+            mesh_vertices,
+            mesh_indices,
+            input_points,
             include_hit_points=compute_sdf_derivatives,
         )
+
         if compute_sdf_derivatives:
-            sdf, sdf_derives = sdf
+            sdf, grad = ret
+        else:
+            sdf = ret
+            grad = None
 
-        sdf = sdf.numpy()
-        sdf = np.expand_dims(max_dis * sdf, axis=1)
+        sdf = sdf.numpy().reshape(-1, 1) * max_dis
 
-        if compute_sdf_derivatives:
-            sdf_derives = sdf_derives.numpy().reshape(-1)
-            sdf_derives = -(sdf_derives - points)
-            sdf_derives = np.reshape(sdf_derives, (sdf_derives.shape[0] // 3, 3))
-            sdf_derives = sdf_derives / np.linalg.norm(
-                sdf_derives, axis=1, keepdims=True
-            )
-            return sdf, sdf_derives
+        if grad is not None:
+            grad = grad.numpy().reshape(-1, 3)
+            return sdf, grad
 
         return sdf
 

From c9a56fc59dab9b7207c90ea48360c433e34c5ee5 Mon Sep 17 00:00:00 2001
From: xiangqian <qian.xiang@njtf.cn>
Date: Wed, 4 Feb 2026 10:58:58 +0800
Subject: [PATCH 2/2] [BUG]fix bugs in make_sdf function

---
 ppsci/geometry/mesh.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/ppsci/geometry/mesh.py b/ppsci/geometry/mesh.py
index f76298fcc..2376cc4ee 100644
--- a/ppsci/geometry/mesh.py
+++ b/ppsci/geometry/mesh.py
@@ -1382,15 +1382,19 @@ def sdf_func(points: np.ndarray, compute_sdf_derivatives=False):
         )
 
         if compute_sdf_derivatives:
-            sdf, grad = ret
+            sdf, hit_points = ret
         else:
             sdf = ret
-            grad = None
+            hit_points = None
 
         sdf = sdf.numpy().reshape(-1, 1) * max_dis
 
-        if grad is not None:
-            grad = grad.numpy().reshape(-1, 3)
+        if hit_points is not None:
+            hit_points = hit_points.numpy().reshape(-1, 3)
+            grad_vec = pts - hit_points
+            norms = np.linalg.norm(grad_vec, axis=1, keepdims=True)
+            norms[norms < 1e-6] = 1.0
+            grad = grad_vec / norms
             return sdf, grad
 
         return sdf