def _setup_actions(self):
"""Setup actions to be used in menues etc."""
self._actions = {}
#exit
self._actions["exit"] = QtGui.QAction("Exit", self)
self._actions["exit"].setShortcut("Ctrl+Q")
self._actions["exit"].triggered.connect(exit)
#save image
self._actions["save image"] = QtGui.QAction("Save image", self)
self._actions["save image"].triggered.connect(self._on_save_image)
def get_matplotlib_widget():
"""Returns four objects: A widget, a figure, a canvas an mpl_toolbar."""
widget = QtGui.QWidget()
fig = Figure((5., 4.))
canvas = FigureCanvas(fig)
canvas.setParent(widget)
mpl_toolbar = NavigationToolbar(canvas, widget)
layout = QtGui.QVBoxLayout()
layout.addWidget(canvas)
layout.addWidget(mpl_toolbar)
widget.setLayout(layout)
return widget, fig, canvas, mpl_toolbar
def _setup_module_select(self):
"""Create a drop down menu for selecting modules. Returns a layout containing it."""
self._module_box = QtGui.QComboBox()
self._module_box.activated.connect(self._select_module)
# self._module_box.activated.connect(self._view_stack.setCurrentIndex)
# self._module_box.activated.connect(self._controll_stack.setCurrentIndex)
label = QtGui.QLabel("Display module: ")
layout = QtGui.QHBoxLayout()
layout.addWidget(label)
layout.addWidget(self._module_box)
layout.addStretch()
return layout
def _setup_gui(self):
"""Setup the entire gui of the program"""
self._view_stack = QtGui.QStackedWidget(self)
self._controll_stack = QtGui.QStackedWidget(self)
layout = QtGui.QVBoxLayout()
layout.addWidget(self._view_stack)
layout.addLayout(self._setup_module_select())
common_controll_group = QtGui.QGroupBox("Common controlls")
common_controll_dummy_layout = QtGui.QHBoxLayout()
common_controll_dummy_layout.addWidget(self._common_controll)
common_controll_group.setLayout(common_controll_dummy_layout)
layout.addWidget(common_controll_group)
module_controll_group = QtGui.QGroupBox("Module controlls")
module_controll_dummy_layout = QtGui.QHBoxLayout()
module_controll_dummy_layout.addWidget(self._controll_stack)
module_controll_group.setLayout(module_controll_dummy_layout)
layout.addWidget(module_controll_group)
size_policy_large = self._view_stack.sizePolicy()
size_policy_large.setVerticalStretch(1)
self._view_stack.setSizePolicy(size_policy_large)
# module_controll_group = QtGui.QGroupBox("Module controlls")
# module_controll_group.setWidget(self._common_controll)
# layout.addWidget(self.module_cocntroll_group)
self._central_widget = QtGui.QWidget()
self._central_widget.setLayout(layout)
self.setCentralWidget(self._central_widget)
def __init__(self, common_controll, viewer, data):
super(Controll, self).__init__()
self._common_controll = common_controll
self._viewer = viewer
self._data = data
self._widget = QtGui.QWidget()
self._active = True
def _setup_dir_chooser(self):
"""Return a directory cooser layout."""
work_dir_label = QtGui.QLabel("Dir:")
file_system_completer = QtGui.QCompleter()
self._file_system_model = QtGui.QFileSystemModel(file_system_completer)
#file_system_model.setRootPath('/')
self._file_system_model.setFilter(QtCore.QDir.Dirs
| QtCore.QDir.Hidden)
file_system_completer.setModel(self._file_system_model)
self._work_dir_edit = QtGui.QLineEdit(os.getcwd())
self._work_dir_edit.setCompleter(file_system_completer)
self._work_dir_edit.editingFinished.connect(self._on_work_dir_changed)
work_dir_layout = QtGui.QHBoxLayout()
work_dir_layout.addWidget(work_dir_label)
work_dir_layout.addWidget(self._work_dir_edit)
return work_dir_layout
def _on_work_dir_changed(self):
"""Handle an attempt to select a new work dir in the
_work_dir_edit text editor"""
new_dir = self._work_dir_edit.text()
self._file_system_model.setRootPath(new_dir)
if not os.path.isdir(new_dir):
palette = self._work_dir_edit.palette()
palette.setColor(QtGui.QPalette.Base, QtGui.QColor(255, 50, 50))
self._work_dir_edit.setPalette(palette)
return
new_dir = self._work_dir_edit.text()
os.chdir(new_dir)
palette = self._work_dir_edit.palette()
palette.setColor(QtGui.QPalette.Base, QtGui.QColor(255, 255, 255))
self._work_dir_edit.setPalette(palette)
self.dirChanged.emit(new_dir)
def _setup_gui(self):
"""Create the gui for the widget."""
view_type_radio_surface = QtGui.QRadioButton("Isosurface plot")
view_type_radio_slice = QtGui.QRadioButton("Slice plot")
view_type_radio_slice.setChecked(True)
view_type_layout = QtGui.QVBoxLayout()
view_type_layout.addWidget(view_type_radio_surface)
view_type_layout.addWidget(view_type_radio_slice)
view_type_radio_surface.clicked.connect(
partial(self.set_view_type, VIEW_TYPE.surface))
view_type_radio_slice.clicked.connect(
partial(self.set_view_type, VIEW_TYPE.slice))
#surface controll setup
self._surface_level_slider = QtGui.QSlider(QtCore.Qt.Horizontal)
self._surface_level_slider.setTracking(True)
self._surface_level_slider.setRange(1, self._slider_length)
def on_slider_changed(self, slider_level):
"""Handles the signal from a surface level slider change."""
# surface_level = slider_level/float(cls._slider_length)
# surface_level = self._slider_length/(float(slider_level))
surface_level = (
(numpy.exp(float(slider_level) / float(self._slider_length)) -
1.) / (numpy.exp(1.) - 1.))
self._viewer.set_surface_level(surface_level)
self._surface_level_slider.valueChanged.connect(
partial(on_slider_changed, self))
self._surface_level_slider.blockSignals(True)
self._surface_level_slider.setSliderPosition(self._slider_length *
INIT_SURFACE_LEVEL)
# slice controll widget setup
log_scale_box = QtGui.QCheckBox()
log_scale_label = QtGui.QLabel('Log Scale')
log_scale_layout = QtGui.QHBoxLayout()
log_scale_layout.addWidget(log_scale_label)
log_scale_layout.addWidget(log_scale_box)
log_scale_layout.addStretch()
log_scale_box.stateChanged.connect(self.set_log_scale)
self._log_scale_widget = QtGui.QWidget()
self._log_scale_widget.setLayout(log_scale_layout)
#self._log_scale_widget.hide()
layout = QtGui.QVBoxLayout()
layout.addLayout(view_type_layout)
layout.addWidget(self._surface_level_slider)
layout.addWidget(self._log_scale_widget)
#layout.addStretch()
self._widget.setLayout(layout)
self._widget.setFixedHeight(120)
def main():
"""Launch program"""
app = QtGui.QApplication(['Controll window'])
#app = QtGui.QApplication.instance()
program = StartMain()
program.show()
program.initialize()
sys.exit(app.exec_())
def _setup_gui(self):
"""Create the gui and set layout."""
self._spin_box = QtGui.QSpinBox()
self._update_spinbox_max(self._max_lim)
#self._spin_box.setRange(self._min_lim, self._max_lim)
self._spin_box.setValue(self._value)
self._max_label = QtGui.QLabel("")
self._update_max_label(self._max_lim)
self._prev_button = QtGui.QPushButton("Previous")
self._next_button = QtGui.QPushButton("Next")
layout = QtGui.QHBoxLayout()
layout.addWidget(self._spin_box)
layout.addWidget(self._max_label)
layout.addWidget(self._prev_button)
layout.addWidget(self._next_button)
self.setLayout(layout)
def _setup_gui(self):
"""Create the gui for the widget."""
layout = QtGui.QVBoxLayout()
self._image_index_chooser = convenient_widgets.IntegerControll(
0, max_lim=self._data.get_number_of_images())
self._image_index_chooser.valueChanged.connect(self.set_image)
self._data.number_of_images_changed.connect(
self._image_index_chooser.set_max)
layout.addWidget(self._image_index_chooser)
log_scale_box = QtGui.QCheckBox()
log_scale_label = QtGui.QLabel("Log Scale")
log_scale_layout = QtGui.QHBoxLayout()
log_scale_layout.addWidget(log_scale_box)
log_scale_layout.addWidget(log_scale_label)
log_scale_layout.addStretch()
log_scale_box.stateChanged.connect(self.set_log_scale)
layout.addLayout(log_scale_layout)
self._widget.setLayout(layout)
def __init__(self, parent=None):
super(MlabWidget, self).__init__(parent)
layout = QtGui.QHBoxLayout(self)
self._vis = MlabVisualization()
self._ui = self._vis.edit_traits(parent=self, kind='subpanel').control
layout.addWidget(self._ui)
self.setLayout(layout)
self._ui.setParent(self)
self.get_scene().background = (1., 1., 1.)
def _setup_gui(self):
"""Create the gui for the widget."""
self._show_average_rotation_radio = QtGui.QRadioButton("Average rot")
self._show_average_rotation_radio.setChecked(True)
self._show_single_rotation_radio = QtGui.QRadioButton("Single rot")
layout = QtGui.QVBoxLayout()
layout.addWidget(self._show_average_rotation_radio)
layout.addWidget(self._show_single_rotation_radio)
self._show_average_rotation_radio.clicked.connect(partial(self.set_rotation_type,
rotation_type=ROTATION_TYPE.average))
self._show_single_rotation_radio.clicked.connect(partial(self.set_rotation_type,
rotation_type=ROTATION_TYPE.single))
self._rotation_image_number_box = QtGui.QSpinBox()
self._rotation_image_number_box.setRange(-1, 10000)
self._single_slice_widget = convenient_widgets.IntegerControll(0)
self._single_slice_widget.valueChanged.connect(self.set_image_number)
layout.addWidget(self._single_slice_widget)
self._widget.setLayout(layout)
self._single_slice_widget.hide()
def __init__(self, parent=None):
super(RotationViewer, self).__init__()
#self._widget = QtGui.QWidget(parent)
self._vtk_widget = QVTKRenderWindowInteractor(self._widget) # _vtk_widget is actually the RenderWindowInteractor
self._vtk_widget.SetInteractorStyle(vtk.vtkInteractorStyleRubberBandPick())
#self._vtk_widget.Initialize()
layout = QtGui.QVBoxLayout()
layout.addWidget(self._vtk_widget)
self._widget.setLayout(layout)
self._renderer = vtk.vtkRenderer()
self._vtk_widget.GetRenderWindow().AddRenderer(self._renderer)
#self._mlab_widget = embedded_mayavi.MlabWidget()
self._setup_done = False
self._points = None
self._default_sphere_n = 10
self._sphere = vtk_tools.SphereMap(self._default_sphere_n)
self._renderer.AddViewProp(self._sphere.get_actor())
self._renderer.SetBackground(1., 1., 1.)
def __init__(self, parent=None):
super(ModelmapViewer, self).__init__(parent)
self._surface_algorithm = None
self._surface_actor = None
self._volume_scalars = None
self._volume = None
self._surface_level = 0.5
self._color = (0.2, 0.8, 0.2)
self._planes = []
self._volume_max = 0.
self._volume_numpy = None
self._vtk_widget = QVTKRenderWindowInteractor(
self._widget) # _vtk_widget is actually the RenderWindowInteractor
self._vtk_widget.SetInteractorStyle(
vtk.vtkInteractorStyleRubberBandPick())
layout = QtGui.QVBoxLayout()
layout.addWidget(self._vtk_widget)
self._widget.setLayout(layout)
#self._vtk_widget.Initialize()
#self._vtk_widget.Start()
self._renderer = vtk.vtkRenderer()
self._vtk_render_window = self._vtk_widget.GetRenderWindow()
self._vtk_render_window.AddRenderer(self._renderer)
#self._renderer = self._vtk_widget.GetRenderWindow().GetRenderer()
self._lut = vtk_tools.get_lookup_table(0.,
1.,
log=False,
colorscale="jet")
self._create_volume_map()
white = (1., 1., 1.)
self._renderer.SetBackground(white)
def on_read_error(self):
"""Handle read error emits"""
palette = self._work_dir_edit.palette()
palette.setColor(QtGui.QPalette.Base, QtGui.QColor(255, 250, 50))
self._work_dir_edit.setPalette(palette)
def __init__(self, parent=None):
super(Viewer, self).__init__(parent)
self._widget = QtGui.QWidget(parent)
def _setup_gui(self):
"""Create the widget gui."""
layout = QtGui.QVBoxLayout()
layout.addLayout(self._setup_dir_chooser())
layout.addWidget(self._setup_iteration_chooser())
self.setLayout(layout)