Suppress messages from VTK

Author: eigenvivek

diffdrr/visualization.py

@@ -124,9 +124,12 @@ def make_fig(ground_truth):
 
 # %% ../notebooks/api/04_visualization.ipynb 9
 import pyvista
+import vtk
 
 from .drr import DRR
 
+vtk.vtkLogger.SetStderrVerbosity(vtk.vtkLogger.ConvertToVerbosity(-1))
+
 # %% ../notebooks/api/04_visualization.ipynb 10
 def drr_to_mesh(
     drr: DRR,

notebooks/api/04_visualization.ipynb

@@ -213,8 +213,11 @@
    "source": [
     "#| export\n",
     "import pyvista\n",
+    "import vtk\n",
     "\n",
-    "from diffdrr.drr import DRR"
+    "from diffdrr.drr import DRR\n",
+    "\n",
+    "vtk.vtkLogger.SetStderrVerbosity(vtk.vtkLogger.ConvertToVerbosity(-1))"
    ]
   },
   {
@@ -257,7 +260,7 @@
     "            method=\"marching_cubes\",\n",
     "            progress_bar=verbose,\n",
     "        )\n",
-    "    \n",
+    "\n",
     "        # Process the mesh\n",
     "        mesh.extract_largest(inplace=True, progress_bar=verbose)\n",
     "        mesh.point_data.clear()\n",
@@ -275,18 +278,24 @@
     "        mesh.decimate_pro(0.25, inplace=True, progress_bar=verbose)\n",
     "        mesh.fill_holes(100, inplace=True, progress_bar=verbose)\n",
     "        mesh.clean(inplace=True, progress_bar=verbose)\n",
-    "    \n",
+    "\n",
     "    elif method == \"surface_nets\":\n",
-    "        grid.point_data[\"values\"] = drr.volume.cpu().numpy().flatten(order=\"F\") > threshold\n",
+    "        grid.point_data[\"values\"] = (\n",
+    "            drr.volume.cpu().numpy().flatten(order=\"F\") > threshold\n",
+    "        )\n",
     "        try:\n",
     "            mesh = grid.contour_labeled(smoothing=True, progress_bar=verbose)\n",
     "        except AttributeError as e:\n",
-    "            raise AttributeError(f\"{e}, ensure you are using pyvista>=0.43 and vtk>=9.3\")\n",
+    "            raise AttributeError(\n",
+    "                f\"{e}, ensure you are using pyvista>=0.43 and vtk>=9.3\"\n",
+    "            )\n",
     "        mesh.clear_cell_data()\n",
     "\n",
     "    else:\n",
-    "        raise ValueError(f\"method must be `marching_cubes` or `surface_nets`, not {method}\")\n",
-    "        \n",
+    "        raise ValueError(\n",
+    "            f\"method must be `marching_cubes` or `surface_nets`, not {method}\"\n",
+    "        )\n",
+    "\n",
     "    return mesh"
    ]
   },
@@ -298,7 +307,9 @@
    "outputs": [],
    "source": [
     "#| export\n",
-    "def img_to_mesh(drr: DRR, rotations, translations, parameterization, convention=None, **kwargs):\n",
+    "def img_to_mesh(\n",
+    "    drr: DRR, rotations, translations, parameterization, convention=None, **kwargs\n",
+    "):\n",
     "    \"\"\"\n",
     "    For a given pose (not batched), turn the camera and detector into a mesh.\n",
     "    Additionally, render the DRR for the pose. Convert into a texture that\n",