clipping vectors on non tets + test

Author: ChrLackner

ngsolve_webgpu/mesh.py

@@ -672,7 +672,7 @@ def update(self, options: RenderOptions):
 
         # indices and colors: one color entry per segment index
         indices = segs["index"].astype(np.uint32)
-        edge_indices = indices - 1
+        edge_indices = indices
         max_index = int(edge_indices.max()) if edge_indices.size > 0 else -1
         # TODO: Colors not yet available in NGSolve
         colors = np.zeros((max_index + 1, 4), dtype=np.float32)

ngsolve_webgpu/vectors.py

@@ -242,7 +242,8 @@ def compute_vectors(self):
         return super().compute_vectors()
 
     def update(self, options):
-        self.n_search_els = self.function_data.mesh_data.ngs_mesh.GetNE(ngs.VOL)
+        self.function_data.update(options)
+        self.n_search_els = self.function_data.mesh_data.num_elements["tets"]
         self.clipping.update(options)
         if not hasattr(self.__clipping, "uniforms"):
             self.__clipping.update(options)

tests/baselines/clipping_vectors_mixed_elements.png

@@ -64,6 +64,35 @@ def test_clipping_vectors_direct(self, webgpu_env):
 
         webgpu_env.assert_matches_baseline(scene, "clipping_vectors_direct.png")
 
+    def test_clipping_vectors_mixed_elements(self, webgpu_env):
+        import ngsolve as ngs
+        import webgpu.jupyter as wj
+        from netgen.occ import Box, Pnt, OCCGeometry
+        from netgen.meshing import BoundaryLayerParameters
+        from ngsolve_webgpu.mesh import MeshData
+        from ngsolve_webgpu.cf import FunctionData
+        from ngsolve_webgpu.vectors import ClippingVectors
+        from webgpu.clipping import Clipping
+
+        webgpu_env.ensure_canvas(600, 600)
+        box = Box(Pnt(0, 0, 0), Pnt(1, 1, 1))
+        geo = OCCGeometry(box)
+        blayer = [
+            BoundaryLayerParameters(boundary=".*", thickness=[0.05], domain=".*", outside=False)]
+        ngmesh = geo.GenerateMesh(maxh=0.3, boundary_layers=blayer)
+        mesh = ngs.Mesh(ngmesh)
+        assert sum(1 for el in mesh.Elements() if el.type == ngs.ET.PRISM) > 0, "Mesh should contain prism elements"
+        mesh_data = MeshData(mesh)
+        cf = ngs.CF((ngs.x, ngs.y, ngs.z))
+        function_data = FunctionData(mesh_data, cf, order=2)
+        clipping = Clipping()
+        clipping.mode = clipping.Mode.PLANE
+        clipping.center = [0.5, 0.5, 0.5]
+        renderer = ClippingVectors(function_data, grid_size=200, clipping=clipping)
+        scene = wj.Draw([renderer], 600, 600)
+
+        webgpu_env.assert_matches_baseline(scene, "clipping_vectors_mixed_elements.png")
+
     def test_fieldlines(self, webgpu_env):
         import ngsolve as ngs
         import webgpu.jupyter as wj