added mesh surface arrows

2026-06-19 21:22:28 +08:00
parent 2e5c047d1e
commit 258948bcb7
20 changed files with 951 additions and 194 deletions
@@ -28,7 +28,7 @@ class MeshRenderWorker(QThread):
            self._renderer = None
            self._pending = None

-    def submit(self, *, flow, scalar_map, centroid, centroid_found, fnormal) -> None:
+    def submit(self, *, flow, scalar_map, centroid, centroid_found, fnormal, valid_mask=None) -> None:
        with QMutexLocker(self._mutex):
            self._pending = {
                "flow": flow,
@@ -36,6 +36,7 @@ class MeshRenderWorker(QThread):
                "centroid": centroid,
                "centroid_found": centroid_found,
                "fnormal": fnormal,
+                "valid_mask": valid_mask,
            }
            self._wake = True
        if not self.isRunning():
@@ -66,6 +67,7 @@ class MeshRenderWorker(QThread):
                centroid=job["centroid"],
                centroid_found=job["centroid_found"],
                fnormal=job["fnormal"],
+                valid_mask=job["valid_mask"],
            )
            self.rgb_ready.emit(np.ascontiguousarray(rgb))