more tests

Author: ChrLackner

ngsolve_webgpu/cf.py

@@ -572,16 +572,24 @@ def __init__(
     ):
         # calling super-super class to not create points and vectors
         BaseVectorRenderer.__init__(self)
+        self.scale_with_vector_length = False
         self.cf = cf
         self.mesh = mesh
         # this somehow segfaults in pyodide?
         self.grid_size = grid_size
         self.size = size
 
+    def get_bounding_box(self):
+        bb = self.mesh.ngmesh.bounding_box
+        pmin = [bb[0][0], bb[0][1], bb[0][2]]
+        pmax = [bb[1][0], bb[1][1], bb[1][2]]
+        return (pmin, pmax)
+
     def redraw(self, timestamp=None):
         super().redraw(timestamp=timestamp, cf=self.cf, mesh=self.mesh, grid_size=self.grid_size)
 
     def update(self, options: RenderOptions):
+        self.options = options
         bb = self.mesh.ngmesh.bounding_box
         self.bounding_box = np.array(
             [[bb[0][0], bb[0][1], bb[0][2]], [bb[1][0], bb[1][1], bb[1][2]]]

ngsolve_webgpu/jupyter.py

@@ -1,3 +1,4 @@
+import numpy as np
 import ngsolve as ngs
 import webgpu.jupyter as wj
 from webgpu.clipping import Clipping
@@ -204,16 +205,21 @@ def Draw(
         contact_renderer = ContactPairs(mesh, contact, **kwargs)
         render_objects.append(contact_renderer)
 
-    scene = wj.Draw(render_objects, width, height)
+    boxes = [o.get_bounding_box() for o in render_objects]
+    boxes = [b for b in boxes if b is not None]
+    if boxes:
+        pmin = np.array(boxes[0][0])
+        pmax = np.array(boxes[0][1])
+        for b in boxes[1:]:
+            pmin = np.minimum(pmin, np.array(b[0]))
+            pmax = np.maximum(pmax, np.array(b[1]))
+        center = 0.5 * (pmin + pmax)
+        if _clip is not None and isinstance(_clip, dict):
+            clipping.center = [_clip.get("x", center[0]), _clip.get("y", center[1]), _clip.get("z", center[2])]
+        else:
+            clipping.center = list(center)
 
-    center = 0.5 * (scene.bounding_box[0] + scene.bounding_box[1])
-    if _clip is not None and isinstance(_clip, dict):
-        clipping.center = [_clip.get("x", center[0]), _clip.get("y", center[1]), _clip.get("z", center[2])]
-    else:
-        clipping.center = center
-    import time
-    clipping.update(time.time())
-    scene.render()
+    scene = wj.Draw(render_objects, width, height)
     if dim == 3:
         clipping.add_options_to_gui(scene.gui)
     for r in render_objects:

tests/Dockerfile

@@ -1,7 +1,7 @@
 ARG BASE_IMAGE=webgpu-base
 FROM ${BASE_IMAGE}
 
-RUN pip install --no-cache-dir --break-system-packages ngsolve ipython
+RUN pip install --no-cache-dir --break-system-packages ngsolve ipython webgpu
 
 WORKDIR /app
 COPY pyproject.toml .