コード例 #1
0
        def end_pick_helper(picker, event_id):
            # Merge the selection into a single mesh
            picked = self_().picked_cells
            if isinstance(picked, pyvista.MultiBlock):
                if picked.n_blocks > 0:
                    picked = picked.combine()
                else:
                    picked = pyvista.UnstructuredGrid()
            # Check if valid
            is_valid_selection = picked.n_cells > 0

            if show and is_valid_selection:
                # Use try in case selection is empty
                self_().add_mesh(picked,
                                 name='_cell_picking_selection',
                                 style=style,
                                 color=color,
                                 line_width=line_width,
                                 pickable=False,
                                 reset_camera=False,
                                 **kwargs)

                # render here prior to running the callback
                self_().render()
            elif not is_valid_selection:
                self.remove_actor('_cell_picking_selection')
                self_().picked_cells = None

            if callback is not None:
                try_callback(callback, self_().picked_cells)

            # TODO: Deactivate selection tool
            return
コード例 #2
0
ファイル: picking.py プロジェクト: huangynj/pyvista
        def _the_callback(mesh, idx):
            if mesh is None:
                return
            point = mesh.points[idx]
            if self._last_picked_idx is None:
                self.picked_geodesic = pyvista.PolyData(point)
            else:
                surface = mesh.extract_surface().triangulate()
                locator = vtk.vtkPointLocator()
                locator.SetDataSet(surface)
                locator.BuildLocator()
                start_idx = locator.FindClosestPoint(
                    mesh.points[self._last_picked_idx])
                end_idx = locator.FindClosestPoint(point)
                self.picked_geodesic = self.picked_geodesic + surface.geodesic(
                    start_idx, end_idx)
            self._last_picked_idx = idx

            if show_path:
                self.add_mesh(self.picked_geodesic,
                              color=color,
                              name='_picked_path',
                              line_width=line_width,
                              point_size=point_size,
                              reset_camera=False,
                              **kwargs)
            if hasattr(callback, '__call__'):
                try_callback(callback, self.picked_geodesic)
            return
コード例 #3
0
 def _the_callback(widget, event_id):
     polyline = pyvista.PolyData()
     widget.GetPolyData(polyline)
     ribbon.shallow_copy(polyline.ribbon(normal=(0, 0, 1), angle=90.0))
     if hasattr(callback, '__call__'):
         try_callback(callback, polyline)
     return
コード例 #4
0
        def _the_callback(mesh, idx):
            if mesh is None:
                return
            point = mesh.points[idx]
            if self._last_picked_idx is None:
                self.picked_geodesic = pyvista.PolyData(point)
                self.picked_geodesic['vtkOriginalPointIds'] = [idx]
            else:
                surface = mesh.extract_surface().triangulate()
                locator = _vtk.vtkPointLocator()
                locator.SetDataSet(surface)
                locator.BuildLocator()
                start_idx = locator.FindClosestPoint(mesh.points[self._last_picked_idx])
                end_idx = locator.FindClosestPoint(point)
                self.picked_geodesic += surface.geodesic(start_idx, end_idx,
                                                         keep_order=keep_order)
                if keep_order:
                    # it makes sense to remove adjacent duplicate points
                    self.picked_geodesic.clean(inplace=True, lines_to_points=False,
                                               polys_to_lines=False,
                                               strips_to_polys=False)
            self._last_picked_idx = idx

            if show_path:
                self.add_mesh(self.picked_geodesic, color=color, name='_picked_path',
                              line_width=line_width, point_size=point_size,
                              reset_camera=False, **kwargs)
            if callable(callback):
                try_callback(callback, self.picked_geodesic)
コード例 #5
0
ファイル: widgets.py プロジェクト: huangynj/pyvista
 def _the_callback(widget, event):
     value = widget.GetRepresentation().GetValue()
     if hasattr(callback, '__call__'):
         if pass_widget:
             try_callback(callback, value, widget)
         else:
             try_callback(callback, value)
     return
コード例 #6
0
 def _the_callback(plane_widget, event_id):
     the_plane = vtk.vtkPlane()
     plane_widget.GetPlane(the_plane)
     normal = the_plane.GetNormal()
     origin = the_plane.GetOrigin()
     if hasattr(callback, '__call__'):
         try_callback(callback, normal, origin)
     return
コード例 #7
0
 def _the_callback(widget, event_id):
     point = widget.GetCenter()
     index = widget.WIDGET_INDEX
     if hasattr(callback, '__call__'):
         if num > 1:
             try_callback(callback, point, index)
         else:
             try_callback(callback, point)
     return
コード例 #8
0
 def _the_callback(box_widget, event_id):
     the_box = pyvista.PolyData()
     box_widget.GetPolyData(the_box)
     planes = vtk.vtkPlanes()
     box_widget.GetPlanes(planes)
     if hasattr(callback, '__call__'):
         if use_planes:
             try_callback(callback, planes)
         else:
             try_callback(callback, the_box)
     return
コード例 #9
0
ファイル: widgets.py プロジェクト: huangynj/pyvista
 def _the_callback(widget, event_id):
     point = widget.GetCenter()
     index = widget.WIDGET_INDEX
     if hasattr(callback, '__call__'):
         if num > 1:
             args = [point, index]
         else:
             args = [point]
         if pass_widget:
             args.append(widget)
         try_callback(callback, *args)
     return
コード例 #10
0
 def _the_callback(widget, event_id):
     pointa = widget.GetPoint1()
     pointb = widget.GetPoint2()
     if hasattr(callback, '__call__'):
         if use_vertices:
             try_callback(callback, pointa, pointb)
         else:
             the_line = pyvista.Line(pointa,
                                     pointb,
                                     resolution=resolution)
             try_callback(callback, the_line)
     return
コード例 #11
0
ファイル: picking.py プロジェクト: jackhuu/pyvista
        def _the_callback(path):
            if path.n_points < 2:
                self.remove_actor(name)
                return
            self.picked_horizon = path.ribbon(normal=normal, width=width)

            if show_horizon:
                self.add_mesh(self.picked_horizon, name=name, color=color,
                              opacity=opacity, pickable=False,
                              reset_camera=False)

            if hasattr(callback, '__call__'):
                try_callback(callback, path)
コード例 #12
0
 def _the_callback(mesh, idx):
     if mesh is None:
         return
     the_ids.append(idx)
     the_points.append(mesh.points[idx])
     self.picked_path = pyvista.PolyData(np.array(the_points))
     self.picked_path.lines = make_line_cells(len(the_points))
     if show_path:
         self.add_mesh(self.picked_path, color=color, name='_picked_path',
                       line_width=line_width, point_size=point_size,
                       reset_camera=False, **kwargs)
     if callable(callback):
         try_callback(callback, self.picked_path)
コード例 #13
0
ファイル: picking.py プロジェクト: jackhuu/pyvista
 def _end_pick_event(picker, event):
     self.picked_point = np.array(picker.GetPickPosition())
     self.picked_mesh = picker.GetDataSet()
     self.picked_point_id = picker.GetPointId()
     if show_point:
         self.add_mesh(self.picked_point, color=color,
                       point_size=point_size, name='_picked_point',
                       pickable=False, reset_camera=False, **kwargs)
     if hasattr(callback, '__call__'):
         if use_mesh:
             try_callback(callback, self.picked_mesh, self.picked_point_id)
         else:
             try_callback(callback, self.picked_point)
コード例 #14
0
ファイル: widgets.py プロジェクト: keltonhalbert/pyvista
 def _the_callback(widget, event_id):
     pointa = widget.GetPoint1()
     pointb = widget.GetPoint2()
     if hasattr(callback, '__call__'):
         if use_vertices:
             args = [pointa, pointb]
         else:
             the_line = pyvista.Line(pointa, pointb, resolution=resolution)
             args = [the_line]
         if pass_widget:
             args.append(widget)
         try_callback(callback, *args)
     return
コード例 #15
0
ファイル: widgets.py プロジェクト: huangynj/pyvista
 def _the_callback(box_widget, event_id):
     the_box = pyvista.PolyData()
     box_widget.GetPolyData(the_box)
     planes = vtk.vtkPlanes()
     box_widget.GetPlanes(planes)
     if hasattr(callback, '__call__'):
         if use_planes:
             args = [planes]
         else:
             args = [the_box]
         if pass_widget:
             args.append(box_widget)
         try_callback(callback, *args)
     return
コード例 #16
0
ファイル: picking.py プロジェクト: peggypan0411/pyvista
        def end_pick_helper(picker, event_id):
            # Merge the selection into a single mesh
            picked = self_().picked_cells
            if isinstance(picked, pyvista.MultiBlock):
                if picked.n_blocks > 0:
                    picked = picked.combine()
                else:
                    picked = pyvista.UnstructuredGrid()
            # Check if valid
            is_valid_selection = picked.n_cells > 0

            if show and is_valid_selection:
                # Select the renderer where the mesh is added.
                active_renderer_index = self_()._active_renderer_index
                for index in range(len(self.renderers)):
                    renderer = self.renderers[index]
                    for actor in renderer._actors.values():
                        mapper = actor.GetMapper()
                        if isinstance(mapper, vtk.vtkDataSetMapper
                                      ) and mapper.GetInput() == mesh:
                            loc = self_().index_to_loc(index)
                            self_().subplot(*loc)
                            break

                # Use try in case selection is empty
                self_().add_mesh(picked,
                                 name='_cell_picking_selection',
                                 style=style,
                                 color=color,
                                 line_width=line_width,
                                 pickable=False,
                                 reset_camera=False,
                                 **kwargs)

                # Reset to the active renderer.
                loc = self_().index_to_loc(active_renderer_index)
                self_().subplot(*loc)

                # render here prior to running the callback
                self_().render()
            elif not is_valid_selection:
                self.remove_actor('_cell_picking_selection')
                self_().picked_cells = None

            if callback is not None:
                try_callback(callback, self_().picked_cells)

            # TODO: Deactivate selection tool
            return
コード例 #17
0
ファイル: picking.py プロジェクト: huangynj/pyvista
        def end_pick_helper(picker, event_id):
            if show:
                # Use try in case selection is empty
                try:
                    self.add_mesh(self.picked_cells,
                                  name='_cell_picking_selection',
                                  style=style,
                                  color=color,
                                  line_width=line_width,
                                  pickable=False,
                                  reset_camera=False,
                                  **kwargs)
                except RuntimeError:
                    pass

            if callback is not None and self.picked_cells.n_cells > 0:
                try_callback(callback, self.picked_cells)

            # TODO: Deactivate selection tool
            return
コード例 #18
0
 def _the_callback(slider, event):
     value = slider.GetRepresentation().GetValue()
     if hasattr(callback, '__call__'):
         try_callback(callback, value)
     return
コード例 #19
0
 def _the_callback(*args):
     click_point = self.pick_mouse_position()
     self.fly_to(click_point)
     if callable(callback):
         try_callback(callback, click_point)
コード例 #20
0
ファイル: picking.py プロジェクト: jackhuu/pyvista
 def _the_callback(*args):
     click_point = self.pick_mouse_position()
     self.fly_to(click_point)
     if hasattr(callback, '__call__'):
         try_callback(callback, click_point)