diff --git a/src/Python/IO/CSVReadEdit.py b/src/Python/IO/CSVReadEdit.py index 907a130edafae3f3cd0315133d3b3a8ec67ac2eb..82af7c63c7fa79bcd88753888c1de53b579c55b8 100755 --- a/src/Python/IO/CSVReadEdit.py +++ b/src/Python/IO/CSVReadEdit.py @@ -339,74 +339,6 @@ def get_diverging_lut1(start: str, mid: str, end: str, table_size: int = 256): return lut -from vtkmodules.vtkIOImage import ( - vtkBMPWriter, - vtkJPEGWriter, - vtkPNGWriter, - vtkPNMWriter, - vtkPostScriptWriter, - vtkTIFFWriter -) -from vtkmodules.vtkRenderingCore import vtkWindowToImageFilter - - -def write_image(file_name, ren_win, rgba=True): - """ - Write the render window view to an image file. - - Image types supported are: - BMP, JPEG, PNM, PNG, PostScript, TIFF. - The default parameters are used for all writers, change as needed. - :param file_name: The file name, if no extension then PNG is assumed. - :param ren_win: The render window. - :param rgba: Used to set the buffer type. - :return: - """ - - if file_name: - valid_suffixes = ['.bmp', '.jpg', '.png', '.pnm', '.ps', '.tiff'] - # Select the writer to use. - parent = Path(file_name).resolve().parent - path = Path(parent) / file_name - if path.suffix: - ext = path.suffix.lower() - else: - ext = '.png' - path = Path(str(path)).with_suffix(ext) - if path.suffix not in valid_suffixes: - print(f'No writer for this file suffix: {ext}') - return - if ext == '.bmp': - writer = vtkBMPWriter() - elif ext == '.jpg': - writer = vtkJPEGWriter() - elif ext == '.pnm': - writer = vtkPNMWriter() - elif ext == '.ps': - if rgba: - rgba = False - writer = vtkPostScriptWriter() - elif ext == '.tiff': - writer = vtkTIFFWriter() - else: - writer = vtkPNGWriter() - - windowto_image_filter = vtkWindowToImageFilter() - windowto_image_filter.SetInput(ren_win) - windowto_image_filter.SetScale(1) # image quality - if rgba: - windowto_image_filter.SetInputBufferTypeToRGBA() - else: - windowto_image_filter.SetInputBufferTypeToRGB() - # Read from the front buffer. - windowto_image_filter.ReadFrontBufferOff() - windowto_image_filter.Update() - - writer.SetFileName(path) - writer.SetInputConnection(windowto_image_filter.GetOutputPort()) - writer.Write() - else: - raise RuntimeError('Need a filename.') if __name__ == '__main__': main()