diff options
Diffstat (limited to 'silx/gui/plot3d/SceneWindow.py')
-rw-r--r-- | silx/gui/plot3d/SceneWindow.py | 192 |
1 files changed, 192 insertions, 0 deletions
diff --git a/silx/gui/plot3d/SceneWindow.py b/silx/gui/plot3d/SceneWindow.py new file mode 100644 index 0000000..5121a17 --- /dev/null +++ b/silx/gui/plot3d/SceneWindow.py @@ -0,0 +1,192 @@ +# coding: utf-8 +# /*########################################################################## +# +# Copyright (c) 2017-2018 European Synchrotron Radiation Facility +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the "Software"), to deal +# in the Software without restriction, including without limitation the rights +# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +# copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in +# all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +# THE SOFTWARE. +# +# ###########################################################################*/ +"""This module provides a QMainWindow with a 3D SceneWidget and toolbars. +""" + +from __future__ import absolute_import + +__authors__ = ["T. Vincent"] +__license__ = "MIT" +__date__ = "29/11/2017" + + +from ...gui import qt, icons + +from .actions.mode import InteractiveModeAction +from .SceneWidget import SceneWidget +from .tools import OutputToolBar, InteractiveModeToolBar, ViewpointToolBar +from .tools.GroupPropertiesWidget import GroupPropertiesWidget + +from .ParamTreeView import ParamTreeView + +# Imported here for convenience +from . import items # noqa + + +__all__ = ['items', 'SceneWidget', 'SceneWindow'] + + +class _PanPlaneAction(InteractiveModeAction): + """QAction to set plane pan interaction on a Plot3DWidget + + :param parent: See :class:`QAction` + :param ~silx.gui.plot3d.Plot3DWidget.Plot3DWidget plot3d: + Plot3DWidget the action is associated with + """ + def __init__(self, parent, plot3d=None): + super(_PanPlaneAction, self).__init__( + parent, 'panSelectedPlane', plot3d) + self.setIcon(icons.getQIcon('3d-plane-pan')) + self.setText('Pan plane') + self.setCheckable(True) + self.setToolTip( + 'Pan selected plane. Press <b>Ctrl</b> to rotate the scene.') + + def _planeChanged(self, event): + """Handle plane updates""" + if event in (items.ItemChangedType.VISIBLE, + items.ItemChangedType.POSITION): + plane = self.sender() + + isPlaneInteractive = \ + plane._getPlane().plane.isPlane and plane.isVisible() + + if isPlaneInteractive != self.isEnabled(): + self.setEnabled(isPlaneInteractive) + mode = 'panSelectedPlane' if isPlaneInteractive else 'rotate' + self.getPlot3DWidget().setInteractiveMode(mode) + + def _selectionChanged(self, current, previous): + """Handle selected object change""" + if isinstance(previous, items.PlaneMixIn): + previous.sigItemChanged.disconnect(self._planeChanged) + + if isinstance(current, items.PlaneMixIn): + current.sigItemChanged.connect(self._planeChanged) + self.setEnabled(True) + self.getPlot3DWidget().setInteractiveMode('panSelectedPlane') + else: + self.setEnabled(False) + + def setPlot3DWidget(self, widget): + previous = self.getPlot3DWidget() + if isinstance(previous, SceneWidget): + previous.selection().sigCurrentChanged.disconnect( + self._selectionChanged) + self._selectionChanged( + None, previous.selection().getCurrentItem()) + + super(_PanPlaneAction, self).setPlot3DWidget(widget) + + if isinstance(widget, SceneWidget): + self._selectionChanged(widget.selection().getCurrentItem(), None) + widget.selection().sigCurrentChanged.connect( + self._selectionChanged) + + +class SceneWindow(qt.QMainWindow): + """OpenGL 3D scene widget with toolbars.""" + + def __init__(self, parent=None): + super(SceneWindow, self).__init__(parent) + if parent is not None: + # behave as a widget + self.setWindowFlags(qt.Qt.Widget) + + self._sceneWidget = SceneWidget() + self.setCentralWidget(self._sceneWidget) + + self._interactiveModeToolBar = InteractiveModeToolBar(parent=self) + panPlaneAction = _PanPlaneAction(self, plot3d=self._sceneWidget) + self._interactiveModeToolBar.addAction(panPlaneAction) + + self._viewpointToolBar = ViewpointToolBar(parent=self) + self._outputToolBar = OutputToolBar(parent=self) + + for toolbar in (self._interactiveModeToolBar, + self._viewpointToolBar, + self._outputToolBar): + toolbar.setPlot3DWidget(self._sceneWidget) + self.addToolBar(toolbar) + self.addActions(toolbar.actions()) + + self._paramTreeView = ParamTreeView() + self._paramTreeView.setModel(self._sceneWidget.model()) + + selectionModel = self._paramTreeView.selectionModel() + self._sceneWidget.selection()._setSyncSelectionModel( + selectionModel) + + paramDock = qt.QDockWidget() + paramDock.setWindowTitle('Object parameters') + paramDock.setWidget(self._paramTreeView) + self.addDockWidget(qt.Qt.RightDockWidgetArea, paramDock) + + self._sceneGroupResetWidget = GroupPropertiesWidget() + self._sceneGroupResetWidget.setGroup( + self._sceneWidget.getSceneGroup()) + + resetDock = qt.QDockWidget() + resetDock.setWindowTitle('Global parameters') + resetDock.setWidget(self._sceneGroupResetWidget) + self.addDockWidget(qt.Qt.RightDockWidgetArea, resetDock) + self.tabifyDockWidget(paramDock, resetDock) + + paramDock.raise_() + + def getSceneWidget(self): + """Returns the SceneWidget of this window. + + :rtype: ~silx.gui.plot3d.SceneWidget.SceneWidget + """ + return self._sceneWidget + + def getParamTreeView(self): + """Returns the :class:`ParamTreeView` of this window. + + :rtype: ParamTreeView + """ + return self._paramTreeView + + def getInteractiveModeToolBar(self): + """Returns the interactive mode toolbar. + + :rtype: InteractiveModeToolBar + """ + return self._interactiveModeToolBar + + def getViewpointToolBar(self): + """Returns the viewpoint toolbar. + + :rtype: ViewpointToolBar + """ + return self._viewpointToolBar + + def getOutputToolBar(self): + """Returns the output toolbar. + + :rtype: OutputToolBar + """ + return self._outputToolBar |