/new_patching.py
Created Jul 27, 2016
New patching script
| import vtk | |
| import vistrails.core.scripting.api as api | |
| org_vistrails_vistrails_url = api.Package('org.vistrails.vistrails.url') | |
| import locale | |
| import os | |
| from vistrails.packages.vtk import tf_widget | |
| # MODULE 381 org.vistrails.vistrails.vtk:vtkVolumeRayCastCompositeFunction | |
| vtk_volume_ray_cast_composite_function = vtk.vtkVolumeRayCastCompositeFunction() | |
| vtk_volume_ray_cast_composite_function.vtkInstance = vtk_volume_ray_cast_composite_function | |
| # MODULE 376 org.vistrails.vistrails.vtk:vtkCamera | |
| vtk_camera = vtk.vtkCamera() | |
| vtk_camera.vtkInstance = vtk_camera | |
| vtk_camera.SetPosition(745.0, -453.0, 370.0) | |
| vtk_camera.SetViewUp(0.0, 0.0, -1.0) | |
| vtk_camera.SetFocalPoint(135.0, 135.0, 150.0) | |
| # MODULE 410 org.vistrails.vistrails.url:DownloadFile | |
| # FUNCTION url url | |
| url = 'http://www.vistrails.org/download/download.php?type=DATA&id=head.120.vtk' | |
| # DEFAULT insecure insecure | |
| insecure = False | |
| file = org_vistrails_vistrails_url.DownloadFile('file', url=url, insecure=insecure) | |
| # MODULE 404 org.vistrails.vistrails.vtk:vtkStructuredPointsReader | |
| # CONNECTION File file | |
| vtk_structured_points_reader = vtk.vtkStructuredPointsReader() | |
| vtk_structured_points_reader.vtkInstance = vtk_structured_points_reader | |
| vtk_structured_points_reader._previous_locale = locale.setlocale(locale.LC_ALL) | |
| locale.setlocale(locale.LC_ALL, 'C') | |
| file = file.name | |
| vtk_structured_points_reader.SetFileName(file) | |
| # This checks for the presence of file in VTK readers | |
| # Skips the check if it's a vtkImageReader or vtkPLOT3DReader, because | |
| # it has other ways of specifying files, like SetFilePrefix for | |
| # multiple files | |
| if not os.path.isfile(vtk_structured_points_reader.GetFileName()): | |
| raise Exception('File does not exist') | |
| vtk_structured_points_reader.Update() | |
| Output = vtk_structured_points_reader.GetOutput() | |
| GetOutputPort0 = vtk_structured_points_reader.GetOutputPort() | |
| locale.setlocale(locale.LC_ALL, vtk_structured_points_reader._previous_locale) | |
| # MODULE 391 org.vistrails.vistrails.vtk:vtkVolumeRayCastMapper | |
| # CONNECTION VolumeRayCastFunction vtk_volume_ray_cast_composite_function | |
| # CONNECTION AddInputConnection GetOutputPort0 | |
| AddInputConnection = [GetOutputPort0] | |
| VolumeRayCastFunction = [vtk_volume_ray_cast_composite_function] | |
| vtk_volume_ray_cast_mapper = vtk.vtkVolumeRayCastMapper() | |
| vtk_volume_ray_cast_mapper.vtkInstance = vtk_volume_ray_cast_mapper | |
| for AddInputConnectionItem in AddInputConnection: | |
| vtk_volume_ray_cast_mapper.AddInputConnection(AddInputConnectionItem) | |
| for VolumeRayCastFunctionItem in VolumeRayCastFunction: | |
| vtk_volume_ray_cast_mapper.SetVolumeRayCastFunction(VolumeRayCastFunctionItem) | |
| vtk_volume_ray_cast_mapper.Update() | |
| # MODULE 399 org.vistrails.vistrails.vtk:vtkScaledTransferFunction | |
| # FUNCTION TransferFunction TransferFunction_2 | |
| TransferFunction_2 = tf_widget.TransferFunction.parse('<transfer_function max_range="1.0" min_range="0.0"><point opacity="0.0" scalar="0.0"><color B="0.0" G="0.0" R="0.0" /></point><point opacity="0.0" scalar="0.0776021828653"><color B="0.0" G="0.0" R="0.0" /></point><point opacity="0.620071684588" scalar="0.14157014157"><color B="0.462745098039" G="0.537254901961" R="0.890196078431" /></point><point opacity="0.0" scalar="0.20592020592"><color B="0.0" G="0.0" R="0.0" /></point><point opacity="0.0" scalar="0.251911830859"><color B="1.0" G="1.0" R="1.0" /></point><point opacity="0.971655328798" scalar="0.463338831759"><color B="1.0" G="1.0" R="1.0" /></point><point opacity="0.0" scalar="0.683760683761"><color B="0.270588235294" G="0.270588235294" R="0.270588235294" /></point><point opacity="0.0" scalar="1.0"><color B="0.0" G="0.0" R="0.0" /></point></transfer_function>') | |
| # CONNECTION Dataset Output | |
| tf = TransferFunction_2 | |
| tf._min_range, tf._max_range = Output.GetScalarRange() | |
| # MODULE 396 org.vistrails.vistrails.vtk:vtkVolumeProperty | |
| # CONNECTION TransferFunction TransferFunction_2 | |
| vtk_volume_property = vtk.vtkVolumeProperty() | |
| vtk_volume_property.vtkInstance = vtk_volume_property | |
| vtk_volume_property.SetInterpolationTypeToLinear() | |
| vtk_volume_property.ShadeOn() | |
| TransferFunction_2.set_on_vtk_volume_property(vtk_volume_property) | |
| # MODULE 386 org.vistrails.vistrails.vtk:vtkVolume | |
| # CONNECTION Mapper vtk_volume_ray_cast_mapper | |
| # CONNECTION Property vtk_volume_property | |
| Mapper = [vtk_volume_ray_cast_mapper] | |
| Property = [vtk_volume_property] | |
| vtk_volume = vtk.vtkVolume() | |
| vtk_volume.vtkInstance = vtk_volume | |
| for MapperItem in Mapper: | |
| vtk_volume.SetMapper(MapperItem) | |
| for PropertyItem in Property: | |
| vtk_volume.SetProperty(PropertyItem) | |
| vtk_volume.Update() | |
| # MODULE 409 org.vistrails.vistrails.vtk:vtkRenderer | |
| # CONNECTION ActiveCamera vtk_camera | |
| # CONNECTION AddVolume vtk_volume | |
| ActiveCamera = [vtk_camera] | |
| AddVolume = [vtk_volume] | |
| vtk_renderer = vtk.vtkRenderer() | |
| vtk_renderer.vtkInstance = vtk_renderer | |
| for AddVolumeItem in AddVolume: | |
| vtk_renderer.AddVolume(AddVolumeItem) | |
| for ActiveCameraItem in ActiveCamera: | |
| vtk_renderer.SetActiveCamera(ActiveCameraItem) | |
| vtk_renderer.SetBackground((0.0, 0.0, 0.0)) | |
| # MODULE 398 org.vistrails.vistrails.vtk:vtkRendererOutput | |
| # CONNECTION value vtk_renderer | |
| value = [vtk_renderer] | |
| renWin = vtk.vtkRenderWindow() | |
| renWin.AddRenderer(value[0].vtkInstance) | |
| iren = vtk.vtkRenderWindowInteractor() | |
| iren.SetRenderWindow(renWin) | |
| renWin.SetSize(1024, 768) | |
| iren.Initialize() | |
| renWin.Render() | |
| iren.Start() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment