Convert Vectors to vtkActor.
(
vecs: List[Vector], size: float = DEFAULT_PT_SIZE, color: str = DEFAULT_PT_COLOR
)
| 124 | |
| 125 | |
| 126 | def _to_vtk_pts( |
| 127 | vecs: List[Vector], size: float = DEFAULT_PT_SIZE, color: str = DEFAULT_PT_COLOR |
| 128 | ) -> vtkActor: |
| 129 | """ |
| 130 | Convert Vectors to vtkActor. |
| 131 | """ |
| 132 | |
| 133 | rv = vtkActor() |
| 134 | |
| 135 | mapper = vtkPolyDataMapper() |
| 136 | points = vtkPoints() |
| 137 | verts = vtkCellArray() |
| 138 | data = vtkPolyData() |
| 139 | |
| 140 | data.SetPoints(points) |
| 141 | data.SetVerts(verts) |
| 142 | |
| 143 | for v in vecs: |
| 144 | ix = points.InsertNextPoint(*v.toTuple()) |
| 145 | verts.InsertNextCell(1) |
| 146 | verts.InsertCellPoint(ix) |
| 147 | |
| 148 | mapper.SetInputData(data) |
| 149 | |
| 150 | rv.SetMapper(mapper) |
| 151 | |
| 152 | rv.GetProperty().SetColor(vtkNamedColors().GetColor3d(color)) |
| 153 | rv.GetProperty().SetPointSize(size) |
| 154 | |
| 155 | return rv |
| 156 | |
| 157 | |
| 158 | def _to_vtk_axs(locs: List[Location], scale: float = 0.1) -> List[vtkProp3D]: |