|
| 1 | +import os |
| 2 | +from trame.layouts import SinglePage |
| 3 | +from trame.html import vtk, vuetify |
| 4 | + |
| 5 | +from vtkmodules.vtkCommonColor import vtkNamedColors |
| 6 | +from vtkmodules.vtkCommonCore import vtkLookupTable |
| 7 | +from vtkmodules.vtkFiltersCore import ( |
| 8 | + vtkContourFilter, |
| 9 | + vtkGlyph3D, |
| 10 | + vtkMaskPoints, |
| 11 | + vtkThresholdPoints, |
| 12 | +) |
| 13 | +from vtkmodules.vtkFiltersModeling import vtkOutlineFilter |
| 14 | +from vtkmodules.vtkFiltersSources import vtkConeSource |
| 15 | +from vtkmodules.vtkIOLegacy import vtkStructuredPointsReader |
| 16 | +from vtkmodules.vtkRenderingCore import ( |
| 17 | + vtkActor, |
| 18 | + vtkPolyDataMapper, |
| 19 | + vtkRenderer, |
| 20 | + vtkRenderWindow, |
| 21 | + vtkRenderWindowInteractor, |
| 22 | +) |
| 23 | + |
| 24 | +# Required for interacter factory initialization |
| 25 | +from vtkmodules.vtkInteractionStyle import vtkInteractorStyleSwitch # noqa |
| 26 | + |
| 27 | +# Required for remote rendering factory initialization, not necessary for |
| 28 | +# local rendering, but doesn't hurt to include it |
| 29 | +import vtkmodules.vtkRenderingOpenGL2 # noqa |
| 30 | + |
| 31 | +CURRENT_DIRECTORY = os.path.abspath(os.path.dirname(__file__)) |
| 32 | + |
| 33 | +# ----------------------------------------------------------------------------- |
| 34 | +# VTK pipeline |
| 35 | +# ----------------------------------------------------------------------------- |
| 36 | + |
| 37 | +renderer = vtkRenderer() |
| 38 | +renderWindow = vtkRenderWindow() |
| 39 | +renderWindow.AddRenderer(renderer) |
| 40 | + |
| 41 | +renderWindowInteractor = vtkRenderWindowInteractor() |
| 42 | +renderWindowInteractor.SetRenderWindow(renderWindow) |
| 43 | +renderWindowInteractor.GetInteractorStyle().SetCurrentStyleToTrackballCamera() |
| 44 | + |
| 45 | +# Read the data |
| 46 | + |
| 47 | +reader = vtkStructuredPointsReader() |
| 48 | +reader.SetFileName(os.path.join(CURRENT_DIRECTORY, "../data/carotid.vtk")) |
| 49 | + |
| 50 | +# Glyphs |
| 51 | + |
| 52 | +threshold = vtkThresholdPoints() |
| 53 | +threshold.SetInputConnection(reader.GetOutputPort()) |
| 54 | +threshold.ThresholdByUpper(200) |
| 55 | + |
| 56 | +mask = vtkMaskPoints() |
| 57 | +mask.SetInputConnection(threshold.GetOutputPort()) |
| 58 | +mask.SetOnRatio(5) |
| 59 | + |
| 60 | +cone = vtkConeSource() |
| 61 | +cone.SetResolution(11) |
| 62 | +cone.SetHeight(1) |
| 63 | +cone.SetRadius(0.25) |
| 64 | + |
| 65 | +cones = vtkGlyph3D() |
| 66 | +cones.SetInputConnection(mask.GetOutputPort()) |
| 67 | +cones.SetSourceConnection(cone.GetOutputPort()) |
| 68 | +cones.SetScaleFactor(0.4) |
| 69 | +cones.SetScaleModeToScaleByVector() |
| 70 | + |
| 71 | +lut = vtkLookupTable() |
| 72 | +lut.SetHueRange(0.667, 0.0) |
| 73 | +lut.Build() |
| 74 | + |
| 75 | +scalarRange = [0] * 2 |
| 76 | +cones.Update() |
| 77 | +scalarRange[0] = cones.GetOutput().GetPointData().GetScalars().GetRange()[0] |
| 78 | +scalarRange[1] = cones.GetOutput().GetPointData().GetScalars().GetRange()[1] |
| 79 | + |
| 80 | +vectorMapper = vtkPolyDataMapper() |
| 81 | +vectorMapper.SetInputConnection(cones.GetOutputPort()) |
| 82 | +vectorMapper.SetScalarRange(scalarRange[0], scalarRange[1]) |
| 83 | +vectorMapper.SetLookupTable(lut) |
| 84 | + |
| 85 | +vectorActor = vtkActor() |
| 86 | +vectorActor.SetMapper(vectorMapper) |
| 87 | + |
| 88 | +# Contours |
| 89 | + |
| 90 | +iso = vtkContourFilter() |
| 91 | +iso.SetInputConnection(reader.GetOutputPort()) |
| 92 | +iso.SetValue(0, 175) |
| 93 | + |
| 94 | +isoMapper = vtkPolyDataMapper() |
| 95 | +isoMapper.SetInputConnection(iso.GetOutputPort()) |
| 96 | +isoMapper.ScalarVisibilityOff() |
| 97 | + |
| 98 | +isoActor = vtkActor() |
| 99 | +isoActor.SetMapper(isoMapper) |
| 100 | +isoActor.GetProperty().SetRepresentationToWireframe() |
| 101 | +isoActor.GetProperty().SetOpacity(0.25) |
| 102 | + |
| 103 | +# Outline |
| 104 | + |
| 105 | +colors = vtkNamedColors() |
| 106 | + |
| 107 | +outline = vtkOutlineFilter() |
| 108 | +outline.SetInputConnection(reader.GetOutputPort()) |
| 109 | + |
| 110 | +outlineMapper = vtkPolyDataMapper() |
| 111 | +outlineMapper.SetInputConnection(outline.GetOutputPort()) |
| 112 | + |
| 113 | +outlineActor = vtkActor() |
| 114 | +outlineActor.SetMapper(outlineMapper) |
| 115 | +outlineActor.GetProperty().SetColor(colors.GetColor3d("White")) |
| 116 | + |
| 117 | +# Add the actors to the renderer |
| 118 | + |
| 119 | +renderer.AddActor(outlineActor) |
| 120 | +renderer.AddActor(vectorActor) |
| 121 | +renderer.AddActor(isoActor) |
| 122 | +renderer.ResetCamera() |
| 123 | +renderWindow.Render() |
| 124 | + |
| 125 | +# ----------------------------------------------------------------------------- |
| 126 | +# Functions |
| 127 | +# ----------------------------------------------------------------------------- |
| 128 | + |
| 129 | + |
| 130 | +def update_view(**kwargs): |
| 131 | + html_view.update() |
| 132 | + |
| 133 | + |
| 134 | +# ----------------------------------------------------------------------------- |
| 135 | +# GUI |
| 136 | +# ----------------------------------------------------------------------------- |
| 137 | + |
| 138 | +layout = SinglePage("Hello trame", on_ready=update_view) |
| 139 | +layout.title.set_text("Hello trame") |
| 140 | + |
| 141 | +html_view = vtk.VtkLocalView(renderWindow) |
| 142 | + |
| 143 | +layout.content.children += [ |
| 144 | + vuetify.VContainer( |
| 145 | + fluid=True, |
| 146 | + classes="pa-0 fill-height", |
| 147 | + children=[html_view], |
| 148 | + ) |
| 149 | +] |
| 150 | + |
| 151 | +# ----------------------------------------------------------------------------- |
| 152 | +# Main |
| 153 | +# ----------------------------------------------------------------------------- |
| 154 | + |
| 155 | +if __name__ == "__main__": |
| 156 | + layout.start() |
0 commit comments