From 4476303fae9dfe44b1b0aee35a1978f9c885ff1d Mon Sep 17 00:00:00 2001 From: "Claudius \"keldu\" Holeksa" Date: Sat, 4 Jul 2026 23:31:09 +0200 Subject: Removing vtk --- vtk_to_png.py | 75 ----------------------------------------------------------- 1 file changed, 75 deletions(-) delete mode 100644 vtk_to_png.py diff --git a/vtk_to_png.py b/vtk_to_png.py deleted file mode 100644 index 9c3f4f3..0000000 --- a/vtk_to_png.py +++ /dev/null @@ -1,75 +0,0 @@ -import vtk -import glob -import os - -vtk_files = sorted(glob.glob("*.vtk")) -output_dir = "frames" -os.makedirs(output_dir, exist_ok=True) - -renderer = vtk.vtkRenderer() -render_window = vtk.vtkRenderWindow() -render_window.SetOffScreenRendering(1) -render_window.SetSize(1024, 512) -render_window.AddRenderer(renderer) - -renderer.SetBackground(1, 1, 1) - -camera_initialized = False - -for i, vtk_file in enumerate(vtk_files): - reader = vtk.vtkStructuredPointsReader() - reader.SetFileName(vtk_file) - reader.Update() - - image = reader.GetOutput() - - mag = vtk.vtkImageMagnitude(); - mag.SetInputData(image); - mag.Update(); - image = mag.GetOutput(); - scalar_name = "Velocity magnitude"; - - scalars = image.GetPointData().GetScalars() - if scalars is None: - raise RuntimeError("No scalar data found") - - vmin, vmax = scalars.GetRange() - - # Image slice mapper (2D!) - slice_mapper = vtk.vtkImageSliceMapper() - slice_mapper.SetInputData(image) - - # Pick the only slice (or middle if dimension >1) - extent = image.GetExtent() - slice_mapper.SetSliceNumber(extent[4]) # Z slice - - slice_actor = vtk.vtkImageSlice() - slice_actor.SetMapper(slice_mapper) - - # Grayscale mapping - prop = slice_actor.GetProperty() - prop.SetColorWindow(vmax - vmin) - prop.SetColorLevel(0.5 * (vmax + vmin)) - - renderer.RemoveAllViewProps() - renderer.AddViewProp(slice_actor) - - if not camera_initialized: - renderer.ResetCamera() - camera_initialized = True - - render_window.Render() - - # Capture to PNG - w2i = vtk.vtkWindowToImageFilter() - w2i.SetInput(render_window) - w2i.ReadFrontBufferOff() - w2i.Update() - - writer = vtk.vtkPNGWriter() - writer.SetFileName(f"{output_dir}/frame_{i:04d}.png") - writer.SetInputConnection(w2i.GetOutputPort()) - writer.Write() - - print(f"Saved frame {i} range=({vmin},{vmax})") - -- cgit v1.2.3