(self, mimeType)
| 34 | self.windowToImage.SetInput(renderWindow) |
| 35 | |
| 36 | def SetFormat(self, mimeType): |
| 37 | if mimeType == "image/png": |
| 38 | self.writer = vtkPNGWriter() |
| 39 | self.writer.SetInputConnection(self.windowToImage.GetOutputPort()) |
| 40 | elif mimeType == "image/jpg": |
| 41 | self.writer = vtkJPEGWriter() |
| 42 | self.writer.SetInputConnection(self.windowToImage.GetOutputPort()) |
| 43 | |
| 44 | def writeImage(self, path): |
| 45 | if self.writer: |
no test coverage detected