Source code for pygimli.viewer.pv.show3d

+ varying the limits (eg exclude a marker) only has effect on the scalar_bar (color_bar), but not on the display of the vtk
+ there is no good coming back from volumetric slicing atm

from shutil import copyfile
import os
import signal
import sys

import pygimli as pg
from .drawer import drawMesh
from .utils import pgMesh2pvMesh

from PyQt5.QtCore import Qt
from PyQt5.QtGui import QFont, QIcon
from PyQt5.QtWidgets import (
    QMainWindow, QFrame, QVBoxLayout, QComboBox, QPushButton,
    QFileDialog, QSplitter, QLabel, QAction, QDialog, QStatusBar

from .gwidgets import (
    GToolBar, GButton, GLineEdit, GComboBox, GSlider, GDoubleSpinBox, CMAPS

pv = pg.optImport('pyvista', requiredFor="properly visualize 3D data")

__all__ = ['Show3D']

[docs]class Show3D(QMainWindow):
[docs] def __init__(self, application, **kwargs): """ pyGIMLi's GUI for pyvista's QtInteractor class. All kwargs are being forwarded to that. Note ---- Everything pyvista.Plotter would accept: """ super(Show3D, self).__init__(None) # self.tmpMesh = tmpMesh # storage for the minima and maxima = {} self._ignore = ['_Attribute', '_Marker', 'glob_min', 'glob_max'] self._app = application # setup the menubar self.setupMenu() self.setupWidget(**kwargs) # signals signal.signal(signal.SIGINT, self._signalHandler) self.acn_close.triggered.connect(self._signalHandler) self.acn_hkeys.triggered.connect(self.showHotKeys)
def _signalHandler(self, sig, frame=None): """ Stop the GUI on CTRL-C, but not the script it was called from. from: """ sys.stderr.write('\r') self._app.quit()
[docs] def setupMenu(self): """ Create the menubar on top of frame and provide actions. """ bar = self.menuBar() # quit the thing self.acn_close = QAction("&Quit", self) self.acn_close.setShortcut("Q") bar.addAction(self.acn_close) # about the viewer and help ghelp = bar.addMenu("Help") self.acn_hkeys = QAction("Hot Keys") ghelp.addAction(self.acn_hkeys)
[docs] def showHotKeys(self): d = QDialog() textfield = QLabel( "q - Close pyGIMLi 3D Viewer\n" "v - Isometric camera view\n" "w - Switch all datasets to a wireframe representation\n" "s - Switch all datasets to a surface representation\n" "r - Reset the camera to view all datasets\n" "shift+click or middle-click - Pan the rendering scene\n" "left click - Rotate the rendering scene in 3D\n" "ctrl+click - Rotate the rendering scene in 2D (view-plane)\n" "mouse-wheel or right-click - Continuously zoom the rendering scene\n" ) textfield.setFont(QFont('Courier')) lyt = QVBoxLayout() btn_quit = QPushButton('quit') btn_quit.clicked.connect(d.done) lyt.addWidget(textfield) lyt.addWidget(btn_quit) lyt.setContentsMargins(2, 2, 2, 2) d.setLayout(lyt) d.setWindowTitle("Hot Keys") d.exec_()
[docs] def setupWidget(self, **kwargs): # create the frame self.frame = QFrame() # add the pv interactor object self.plotter = pv.QtInteractor(**kwargs) vlayout = QVBoxLayout() vlayout.setContentsMargins(0, 0, 0, 0) self.statusbar = QStatusBar() self.setStatusBar(self.statusbar) self.toolbar = GToolBar() splitter = QSplitter(Qt.Horizontal) splitter.addWidget(self.toolbar) splitter.addWidget(self.plotter.interactor) splitter.setStretchFactor(0, 2) splitter.setStretchFactor(1, 5) vlayout.addWidget(splitter) self.frame.setLayout(vlayout) self.setCentralWidget(self.frame) self.setWindowTitle("pyGIMLi 3D Viewer") # set the icon for the window ipath = os.path.dirname(__file__) icon = os.path.join(ipath, 'favicon.ico') self.setWindowIcon(QIcon(icon))
[docs] def wait(self): """ overload """ self._app.exec() self._app.closeAllWindows()
[docs] def addMesh(self, mesh, data=None, **kwargs): """ Add a mesh to the pyqt frame. Parameters ---------- mesh: pg.Mesh pyGIMLi created mesh. data: iterable Data belonging to the mesh. Note ---- **kwargs label: str A label for the given data. cmap: str The MPL colormap that should be used to display parameters. """ self.mesh = pgMesh2pvMesh(mesh, data, kwargs.pop('label', None)) cMap = kwargs.pop('cmap', 'viridis') if 'alpha' in kwargs: kwargs['opacity'] = kwargs.pop('alpha', 1) self.__kwargs = kwargs _, self._actor = drawMesh( self.plotter, self.mesh, cmap=cMap, returnActor=True, show_Edges=True, **self.__kwargs) # self._actor = self.plotter.add_mesh( # self.mesh, show_edges=True, cmap=cMap, **self.__kwargs) _ = self.plotter.show_bounds(all_edges=True, minor_ticks=True) self.plotter.reset_camera() # set the correctly chosen colormap if cMap.endswith('_r'): cMap = cMap[:-2] self.toolbar.btn_reverse.setChecked(2) # check if given cmap is in list and implement if it isn't if cMap not in CMAPS: self.toolbar.addExtraCMap(cMap) self.toolbar.cbbx_cmap.setCurrentText(cMap) self.allowMeshParameters() self._allowSignals() # set slicers to center after they're enabled _bounds = self.mesh.bounds self.toolbar.slice_x.setMinimum(_bounds[0]) self.toolbar.slice_x.setMaximum(_bounds[1]) self.toolbar.slice_x.setValue(0.5 * (_bounds[0] + _bounds[1])) self.toolbar.slice_y.setMinimum(_bounds[2]) self.toolbar.slice_y.setMaximum(_bounds[3]) self.toolbar.slice_y.setValue(0.5 * (_bounds[2] + _bounds[3])) self.toolbar.slice_z.setMinimum(_bounds[4]) self.toolbar.slice_z.setMaximum(_bounds[5]) self.toolbar.slice_z.setValue(0.5 * (_bounds[4] + _bounds[5])) # show loaded mesh self.plotter.reset_camera()
[docs] def allowMeshParameters(self): """ Make data from the given mesh accessible via GUI. Note ---- This apparently needs to happen when using the gui since on call the cell_arrays will be emptied... """ # enable only when there is something to show self.toolbar.btn_apply.setEnabled(True) self.toolbar.btn_reset.setEnabled(True) self.toolbar.spbx_cmin.setEnabled(True) self.toolbar.spbx_cmax.setEnabled(True) _min = 1e99 _max = -1e99 for label, data in self.mesh.cell_arrays.items(): _mi = min(data) _ma = max(data) _min = _mi if _mi < _min else _mi _max = _ma if _ma > _max else _ma[label] = { 'orig': {'min': _mi, 'max': _ma}, 'user': {'min': _mi, 'max': _ma}, 'data_orig': data, 'data_user': None } for label, data in self.mesh.point_arrays.items(): _mi = min(data) _ma = max(data) _min = _mi if _mi < _min else _mi _max = _ma if _ma > _max else _ma[label] = { 'orig': {'min': _mi, 'max': _ma}, 'user': {'min': _mi, 'max': _ma}, 'data_orig': data, 'data_user': None }['glob_min'] = _min['glob_max'] = _max # supply the combobox with the names to choose from for display self.toolbar.cbbx_params.addItems(self.mesh.array_names) # get the current set parameter curr_param = self.toolbar.cbbx_params.currentText() # set the first cMin/cMax self.toolbar.spbx_cmin.setValue([curr_param]['orig']['min']) self.toolbar.spbx_cmax.setValue([curr_param]['orig']['max']) self.updateParameterView(curr_param)
[docs] def updateParameterView(self, param=None): """ Change the view to given Parameter values. Parameters ---------- param: Current text of the just triggered QComboBox Note ---- May be overloaded. """ # remove the currently displayed mesh self.plotter.remove_actor(self._actor) mesh = self.mesh if param is not None and param not in CMAPS and not isinstance(param, int): # change to the desired parameter distribution self.mesh.set_active_scalars(param) # update the minima and maxima in the limit range # NOTE: if the global limit button is checked, just don't change # the extrema labels to enable the user to set ones own limits. if not self.toolbar.btn_global_limits.isChecked(): _min =[param]['user']['min'] _max =[param]['user']['max'] self.toolbar.spbx_cmin.setRange(_min, _max) self.toolbar.spbx_cmax.setRange(_min, _max) self.toolbar.spbx_cmin.setValue(_min) self.toolbar.spbx_cmax.setValue(_max) cMap = self.toolbar.cbbx_cmap.currentText() if self.toolbar.btn_reverse.isChecked(): cMap += '_r' if self.toolbar.btn_slice_plane.isChecked() and not self.toolbar.btn_slice_volume.isChecked(): x_val = self.toolbar.slice_x.value() y_val = self.toolbar.slice_y.value() z_val = self.toolbar.slice_z.value() # set slicer values into their boxes self.toolbar.la_xval.setText(str(round(x_val, 8))) self.toolbar.la_yval.setText(str(round(y_val, 8))) self.toolbar.la_zval.setText(str(round(z_val, 8))) # get the actual slices mesh = self.mesh.slice_orthogonal(x=x_val, y=y_val, z=z_val) if self.toolbar.chbx_threshold.isChecked(): # get the user defined limits cmin = self.toolbar.spbx_cmin.value() cmax = self.toolbar.spbx_cmax.value() mesh = self.mesh.threshold(value=[cmin, cmax]) # save the camera position # NOTE: this returns [camera position, focal point, and view up] self.camera_pos = self.plotter.camera_position[0] # add the modified one if self.toolbar.btn_slice_volume.isChecked() and not self.toolbar.btn_slice_plane.isChecked(): self._actor = self.plotter.add_mesh_clip_plane( mesh, cmap=cMap, show_edges=True, **self.__kwargs) else: # in case the plane widget was on.. turn it off # self.plotter.disable_plane_widget() self._actor = self.plotter.add_mesh( mesh, cmap=cMap, show_edges=True, **self.__kwargs) # update stuff in the toolbar self.updateScalarBar() # reset the camera position self.plotter.set_position(self.camera_pos)
[docs] def updateScalarBar(self): """ When user set limits are made and finished/accepted the color bar needs to change. """ cmin = float(self.toolbar.spbx_cmin.value()) cmax = float(self.toolbar.spbx_cmax.value()) if cmax >= cmin: # get the active scalar/parameter that is displayed currently param = self.mesh.active_scalars_name # update the user extrema if not self.toolbar.btn_global_limits.isChecked():[param]['user']['min'] = cmin[param]['user']['max'] = cmax # NOTE: has no effect on the displayed vtk # pg._d("RESET SCALAR BAR LIMITS") self.plotter.update_scalar_bar_range([cmin, cmax]) self.plotter.update()
[docs] def toggleBbox(self): """ Toggle the visibility of the axis grid surrounding the model. """ checked = not self.toolbar.btn_bbox.isChecked() if not checked: self.plotter.remove_bounds_axes() self.plotter.remove_bounding_box() else: _ = self.plotter.show_bounds( all_edges=True, minor_ticks=True, show_xaxis=True, show_yaxis=True, show_zaxis=True, show_xlabels=True, show_ylabels=True, show_zlabels=True ) self.plotter.update()
[docs] def takeScreenShot(self): """ Save the scene as image. Todo ---- + might come in handy to open a dialog where one can choose between black/white background and white/black axis grid and so on """ fname = QFileDialog.getSaveFileName( self, 'Open File', None, "Image files (*.jpg *.png)" )[0] if fname: if not len(fname.split('.')) == 2: fname += '.png' self.plotter.screenshot(fname)
[docs] def exportMesh(self): """ Save the displayed data as VTK. """ f = QFileDialog.getSaveFileName( self, 'Export VTK', None, "VTK file (*.vtk)" )[0] if f: f = f + '.vtk' if not f.lower().endswith('.vtk') else f copyfile(self.tmpMesh, f)
[docs] def resetExtrema(self, _btn=False, fromGlobal=False): """ Reset user chosen values to the original ones. Parameters ---------- _btn: bool [False] Catch the default that comes with the button signal. fromGlobal: bool [False] Flag for condition.. is set when resetting user/global limits. """ # get the active scalar/parameter that is displayed currently if fromGlobal is not False: param = fromGlobal else: param = self.mesh.active_scalars_name[param]['user']['min'] =[param]['orig']['min'][param]['user']['max'] =[param]['orig']['max'] if not fromGlobal: # display correctly self.updateParameterView(param)
[docs] def setGlobalLimits(self): """ Manipulate the user limits of the dictionary storing all data. """ if self.toolbar.btn_global_limits.isChecked(): _min =['glob_min'] _max =['glob_max'] self.toolbar.spbx_cmin.setRange(_min, _max) self.toolbar.spbx_cmax.setRange(_min, _max) self.toolbar.spbx_cmin.setValue(_min) self.toolbar.spbx_cmax.setValue(_max) else: for label in if label in self._ignore: continue self.resetExtrema(label) self.updateParameterView()
def _enableSlicers(self): if self.toolbar.btn_slice_plane.isChecked(): self.toolbar.slice_x.setEnabled(True) self.toolbar.slice_y.setEnabled(True) self.toolbar.slice_z.setEnabled(True) else: self.toolbar.slice_x.setEnabled(False) self.toolbar.slice_y.setEnabled(False) self.toolbar.slice_z.setEnabled(False) self.updateParameterView() def _allowSignals(self): # connect signals self.toolbar.cbbx_params.currentTextChanged.connect( self.updateParameterView) self.toolbar.cbbx_cmap.currentTextChanged.connect( self.updateParameterView) self.toolbar.btn_reverse.clicked.connect(self.updateParameterView) self.toolbar.btn_bbox.pressed.connect(self.toggleBbox) self.toolbar.btn_global_limits.clicked.connect(self.setGlobalLimits) self.toolbar.btn_screenshot.clicked.connect(self.takeScreenShot) self.toolbar.btn_exportVTK.clicked.connect(self.exportMesh) self.toolbar.chbx_threshold.clicked.connect(self.updateParameterView) self.toolbar.chbx_threshold.clicked.connect(self._checkStatusThreshold) self.toolbar.btn_apply.clicked.connect(self.updateParameterView) self.toolbar.btn_reset.clicked.connect(self.resetExtrema) self.toolbar.btn_slice_plane.clicked.connect(self._checkStatusPlaneSlice) self.toolbar.btn_slice_volume.clicked.connect(self._checkStatusVolumeSlice) self.toolbar.slice_x.sliderReleased.connect(self.updateParameterView) self.toolbar.slice_y.sliderReleased.connect(self.updateParameterView) self.toolbar.slice_z.sliderReleased.connect(self.updateParameterView) def _checkStatusPlaneSlice(self): if self.toolbar.btn_slice_plane.isChecked(): self.toolbar.btn_slice_volume.setChecked(False) self.toolbar.chbx_threshold.setChecked(False) self._enableSlicers() def _checkStatusVolumeSlice(self): if self.toolbar.btn_slice_volume.isChecked(): self.toolbar.btn_slice_plane.setChecked(False) self.toolbar.chbx_threshold.setChecked(False) self._enableSlicers() def _checkStatusThreshold(self): """ Since its either threshold or slice, just disable the other. """ if self.toolbar.chbx_threshold.isChecked(): self.toolbar.btn_slice_plane.setChecked(False) self.toolbar.btn_slice_volume.setChecked(False)
if __name__ == '__main__': pass