class ActivityToggleItem(QWidget):
"""Toggle button for Activity Dialog.
A progress indicator is displayed when there are active progress
bars.
"""
def __init__(self, parent=None) -> None:
super().__init__(parent=parent)
self.setLayout(QHBoxLayout())
self._activityBtn = QToolButton()
self._activityBtn.setObjectName("QtActivityButton")
self._activityBtn.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
self._activityBtn.setArrowType(Qt.UpArrow)
self._activityBtn.setIconSize(QSize(11, 11))
self._activityBtn.setText(trans._('activity'))
self._activityBtn.setCheckable(True)
self._inProgressIndicator = QLabel(trans._("in progress..."), self)
sp = self._inProgressIndicator.sizePolicy()
sp.setRetainSizeWhenHidden(True)
self._inProgressIndicator.setSizePolicy(sp)
load_gif = str(Path(napari.resources.__file__).parent / "loading.gif")
mov = QMovie(load_gif)
mov.setScaledSize(QSize(18, 18))
self._inProgressIndicator.setMovie(mov)
self._inProgressIndicator.hide()
self.layout().addWidget(self._inProgressIndicator)
self.layout().addWidget(self._activityBtn)
self.layout().setContentsMargins(0, 0, 0, 0)
class TopToolBarRight(QToolBar):
def __init__(self):
super().__init__()
self.setFloatable(False)
self.setMovable(False)
self.setLayoutDirection(Qt.RightToLeft)
self.hide_button = QToolButton()
self.hide_button.setObjectName("hidebutton")
self.hide_button.setArrowType(Qt.UpArrow)
self.addWidget(self.hide_button)
class DesignTab(QWidget):
def __init__(self):
super(DesignTab, self).__init__()
self.setup_ui()
self.grain_dialog = None
self.btn_new_grain.clicked.connect(self.create_grain_dialog)
def create_grain_dialog(self):
self.grain_dialog = NewGrainDialog()
self.grain_dialog.show()
self.grain_dialog.activateWindow()
self.grain_dialog.raise_()
def setup_ui(self):
sp = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.setSizePolicy(sp)
def setup_grain_design():
# grain table view
master = QHBoxLayout(self)
master.addWidget(QTableView())
# grain design controls
controls = QVBoxLayout(self)
# add a push button
self.btn_new_grain = QPushButton(self.tr("New Grain"))
self.btn_new_grain.setMinimumHeight(50)
controls.addWidget(self.btn_new_grain)
# add a dividing line
line = QFrame()
line.setFrameShape(QFrame.HLine)
line.setFrameShadow(QFrame.Sunken)
controls.addSpacing(5)
controls.addWidget(line)
# rest of the controls buttons
self.btn_edit_grain = QPushButton(self.tr("Edit"))
self.btn_edit_grain.setMinimumHeight(30)
controls.addWidget(self.btn_edit_grain)
self.btn_delete_Grain = QPushButton(self.tr("Delete"))
self.btn_delete_Grain.setMinimumHeight(30)
controls.addWidget(self.btn_delete_Grain)
# move grain up and down
moveup = QHBoxLayout()
self.btn_move_up = QToolButton()
self.btn_move_up.setArrowType(Qt.UpArrow)
moveup.addWidget(self.btn_move_up)
moveup.addWidget(QLabel(self.tr("Move Up")))
controls.addLayout(moveup)
movedown = QHBoxLayout()
self.btn_move_down = QToolButton()
self.btn_move_down.setArrowType(Qt.DownArrow)
movedown.addWidget(self.btn_move_down)
movedown.addWidget(QLabel(self.tr("Move Down")))
controls.addLayout(movedown)
controls.addStretch()
# add info for motor design
fl_propellant_info = QFormLayout()
gb_motor_info = QGroupBox(self.tr("Propellant Info"))
gb_motor_info.setLayout(fl_propellant_info)
self.lbl_num_grains = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Number of Segments:")), self.lbl_num_grains)
self.lbl_motor_dia = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Motor Diameter:")), self.lbl_motor_dia)
self.lbl_motor_len = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Propellant Length:")), self.lbl_motor_len)
self.lbl_prop_mass = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Propellant Mass:")), self.lbl_prop_mass)
self.lbl_volume_loading = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Volume Loading:")), self.lbl_volume_loading)
# set group box's layout
controls.addWidget(gb_motor_info)
# setup master layout
master.addLayout(controls)
self.gb_design = QGroupBox(self.tr("Grain Design"))
self.gb_design.setLayout(master)
def setup_chamber_design():
# master layout
master = QVBoxLayout()
# master group box
self.gb_chamber = QGroupBox(self.tr("Chamber Design"))
# nozzle settings button
self.btn_nozzle_settings = QPushButton(self.tr("Edit Nozzle"))
self.btn_nozzle_settings.setMinimumHeight(50)
master.addWidget(self.btn_nozzle_settings)
# nozzle info pane
fl_nozzle_info = QFormLayout()
gb_nozzle = QGroupBox(self.tr("Nozzle Info"))
gb_nozzle.setLayout(fl_nozzle_info)
self.lbl_nozzle_throat = QLabel()
fl_nozzle_info.addRow(QLabel(self.tr("Throat Diameter:")), self.lbl_nozzle_throat)
self.lbl_nozzle_exit = QLabel()
fl_nozzle_info.addRow(QLabel(self.tr("Exit Diameter:")), self.lbl_nozzle_exit)
self.lbl_nozzle_expansion_ratio = QLabel()
fl_nozzle_info.addRow(QLabel(self.tr("Expansion Ratio:")), self.lbl_nozzle_expansion_ratio)
# add group box to master
master.addWidget(gb_nozzle)
master.addStretch()
# overall motor info pane
fl_motor_info = QFormLayout()
gb_motor = QGroupBox(self.tr("Motor Info"))
gb_motor.setLayout(fl_motor_info)
self.lbl_kn = QLabel()
fl_motor_info.addRow(QLabel(self.tr("Kn:")), self.lbl_kn)
self.lbl_port_throat = QLabel()
fl_motor_info.addRow(QLabel(self.tr("Port/Throat Ratio:")), self.lbl_port_throat)
master.addWidget(gb_motor)
self.gb_chamber.setLayout(master)
def setup_gfx_ui():
# design overview
self.motor_display_view = QGraphicsView()
self.motor_display_scene = QGraphicsScene()
self.motor_display_view.setScene(self.motor_display_scene)
self.motor_display_view.show()
# sliced cross section
self.grain_slice_view = QGraphicsView()
self.grain_slice_scene = QGraphicsScene()
self.grain_slice_view.setScene(self.grain_slice_scene)
self.grain_slice_view.show()
# splitter
self.splt_gfx = QSplitter(Qt.Horizontal)
self.splt_gfx.addWidget(self.motor_display_view)
self.splt_gfx.addWidget(self.grain_slice_view)
self.splt_gfx.setStretchFactor(0, 10)
self.splt_gfx.setStretchFactor(1, 3)
self.splt_gfx.setMinimumHeight(50)
setup_grain_design()
setup_chamber_design()
setup_gfx_ui()
self.splt_grain_design = QSplitter(Qt.Horizontal)
self.splt_grain_design.addWidget(self.gb_design)
self.splt_grain_design.addWidget(self.gb_chamber)
self.splt_grain_design.setStretchFactor(0, 10)
self.splt_grain_design.setStretchFactor(1, 3)
self.splt_main = QSplitter(Qt.Vertical)
self.splt_main.addWidget(self.splt_grain_design)
self.splt_main.addWidget(self.splt_gfx)
self.splt_main.setSizes([300, 150])
layout = QVBoxLayout()
layout.addWidget(self.splt_main)
self.setLayout(layout)
class AxesOrderWidget(FigureWidget):
def __init__(self, ui, parent=None):
super(AxesOrderWidget, self).__init__(ui, parent)
self.setWindowTitle(tr("Axes order"))
self.signal = None
self.create_controls()
self._resize_controls()
def _on_figure_change(self, win):
super(AxesOrderWidget, self)._on_figure_change(win)
self.signal = win2sig(win)
self._update_controls_from_signal(self.signal)
def _update_controls_from_signal(self, signal):
if signal is None:
self.setEnabled(False)
return
self.setEnabled(True)
self.lst_nav.setUpdatesEnabled(False)
self.lst_sig.setUpdatesEnabled(False)
self._updating = True
self.lst_nav.clear()
self.lst_sig.clear()
for ax in signal.signal.axes_manager._get_axes_in_natural_order():
rep = '%s axis, size: %i' % (ax._get_name(), ax.size)
p = self.lst_nav if ax.navigate else self.lst_sig
i = QtWidgets.QListWidgetItem(rep)
i.setData(QtCore.Qt.UserRole, ax)
p.addItem(i)
self._updating = False
self.lst_nav.setUpdatesEnabled(True)
self.lst_sig.setUpdatesEnabled(True)
self._resize_controls()
def _resize_controls(self):
self.lst_nav.setUpdatesEnabled(False)
self.lst_sig.setUpdatesEnabled(False)
max_nav = max(self.lst_nav.sizeHint().height(), 20)
max_sig = max(self.lst_sig.sizeHint().height(), 20)
self.lst_nav.setFixedHeight(max_nav)
self.lst_sig.setFixedHeight(max_sig)
self.setFixedHeight(max_nav + max_sig + 150)
self.lst_nav.setUpdatesEnabled(True)
self.lst_sig.setUpdatesEnabled(True)
def _move_item(self, item):
ax = item.data(QtCore.Qt.UserRole)
ax.navigate = not ax.navigate
self.ui.record_code(
("ax = ui.get_selected_signal().axes_manager[%s]\n"
"ax.navigate = not ax.navigate\n"
"ui.get_selected_wrapper().replot()") % (ax.index_in_array + 3j))
self.signal.replot()
def _list_move(self, item, dst_row, dst):
"""Called when drag and drop moved interal in list.
"""
if self._updating:
return
ax = item.data(QtCore.Qt.UserRole)
space = 1j if ax.navigate else 2j
new_idx = dst_row + space
self.plugin.rollaxis(ax, new_idx)
def _list_insert(self, src_row, dst_row, dst):
"""Called when drag and drop moved between lists.
"""
if self._updating:
return
# Switch space
am = self.signal.signal.axes_manager
old_space = 1j if dst is self.lst_sig else 2j
old_idx = src_row + old_space
ax = am[old_idx]
dst.item(dst_row).setData(QtCore.Qt.UserRole, ax)
ax.navigate = not ax.navigate
self.ui.record_code(
("ax = ui.get_selected_signal().axes_manager[%s]\n"
"ax.navigate = not ax.navigate\n") % (ax.index_in_array + 3j))
space = 1j if ax.navigate else 2j
new_idx = dst_row + space
self.plugin.rollaxis(ax, new_idx)
def _move_down(self):
if self.lst_nav.count() == 0:
# nothing to transfer...
return
i = self.lst_nav.takeItem(self.lst_nav.currentRow())
self.lst_sig.addItem(i)
self._move_item(i)
def _move_up(self):
if self.lst_sig.count() == 0:
# nothing to transfer...
return
# TODO: Track "active" list (only select in last?), and move int/ext as
# needed
i = self.lst_sig.takeItem(self.lst_sig.currentRow())
self.lst_nav.addItem(i)
self._move_item(i)
def _flip_clicked(self):
# Get selected axes
axes = []
for lst in (self.lst_nav, self.lst_sig):
items = lst.selectedItems()
axes.extend([i.data(QtCore.Qt.UserRole) for i in items])
self.plugin.flip_axes(axes)
def create_controls(self):
self.lbl_nav = QLabel(tr("Navigate"), self)
self.lst_nav = AxesListWidget(self)
self.btn_up = QToolButton(self)
self.btn_up.setArrowType(QtCore.Qt.UpArrow)
self.btn_down = QToolButton(self)
self.btn_down.setArrowType(QtCore.Qt.DownArrow)
self.lbl_sig = QLabel(tr("Signal"), self)
self.lst_sig = AxesListWidget(self)
sp = self.lst_sig.sizePolicy()
sp.setVerticalPolicy(QtWidgets.QSizePolicy.Fixed)
self.lst_sig.setSizePolicy(sp)
sp = self.lst_nav.sizePolicy()
sp.setVerticalPolicy(QtWidgets.QSizePolicy.Fixed)
self.lst_nav.setSizePolicy(sp)
self.btn_down.clicked.connect(self._move_down)
self.btn_up.clicked.connect(self._move_up)
self.lst_nav.inserted.connect(self._list_insert)
self.lst_sig.inserted.connect(self._list_insert)
self.lst_nav.moved.connect(self._list_move)
self.lst_sig.moved.connect(self._list_move)
self.btn_flip = QPushButton(tr("Reverse axes"))
self.btn_flip.clicked.connect(self._flip_clicked)
vbox = QVBoxLayout()
vbox.addWidget(self.lbl_nav)
vbox.addWidget(self.lst_nav)
hbox = QHBoxLayout()
hbox.addWidget(self.btn_up)
hbox.addWidget(self.btn_down)
vbox.addLayout(hbox)
vbox.addWidget(self.lbl_sig)
vbox.addWidget(self.lst_sig)
vbox.addWidget(self.btn_flip)
w = QWidget()
w.setLayout(vbox)
self.setWidget(w)
class DesignTab(QWidget):
def __init__(self):
super(DesignTab, self).__init__()
self.setup_ui()
self.grain_dialog = None
self.btn_new_grain.clicked.connect(self.create_grain_dialog)
def create_grain_dialog(self):
self.grain_dialog = NewGrainDialog()
self.grain_dialog.show()
self.grain_dialog.activateWindow()
self.grain_dialog.raise_()
def setup_ui(self):
sp = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.setSizePolicy(sp)
def setup_grain_design():
# grain table view
master = QHBoxLayout(self)
master.addWidget(QTableView())
# grain design controls
controls = QVBoxLayout(self)
# add a push button
self.btn_new_grain = QPushButton(self.tr("New Grain"))
self.btn_new_grain.setMinimumHeight(50)
controls.addWidget(self.btn_new_grain)
# add a dividing line
line = QFrame()
line.setFrameShape(QFrame.HLine)
line.setFrameShadow(QFrame.Sunken)
controls.addSpacing(5)
controls.addWidget(line)
# rest of the controls buttons
self.btn_edit_grain = QPushButton(self.tr("Edit"))
self.btn_edit_grain.setMinimumHeight(30)
controls.addWidget(self.btn_edit_grain)
self.btn_delete_Grain = QPushButton(self.tr("Delete"))
self.btn_delete_Grain.setMinimumHeight(30)
controls.addWidget(self.btn_delete_Grain)
# move grain up and down
moveup = QHBoxLayout()
self.btn_move_up = QToolButton()
self.btn_move_up.setArrowType(Qt.UpArrow)
moveup.addWidget(self.btn_move_up)
moveup.addWidget(QLabel(self.tr("Move Up")))
controls.addLayout(moveup)
movedown = QHBoxLayout()
self.btn_move_down = QToolButton()
self.btn_move_down.setArrowType(Qt.DownArrow)
movedown.addWidget(self.btn_move_down)
movedown.addWidget(QLabel(self.tr("Move Down")))
controls.addLayout(movedown)
controls.addStretch()
# add info for motor design
fl_propellant_info = QFormLayout()
gb_motor_info = QGroupBox(self.tr("Propellant Info"))
gb_motor_info.setLayout(fl_propellant_info)
self.lbl_num_grains = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Number of Segments:")),
self.lbl_num_grains)
self.lbl_motor_dia = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Motor Diameter:")),
self.lbl_motor_dia)
self.lbl_motor_len = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Propellant Length:")),
self.lbl_motor_len)
self.lbl_prop_mass = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Propellant Mass:")),
self.lbl_prop_mass)
self.lbl_volume_loading = QLabel()
fl_propellant_info.addRow(QLabel(self.tr("Volume Loading:")),
self.lbl_volume_loading)
# set group box's layout
controls.addWidget(gb_motor_info)
# setup master layout
master.addLayout(controls)
self.gb_design = QGroupBox(self.tr("Grain Design"))
self.gb_design.setLayout(master)
def setup_chamber_design():
# master layout
master = QVBoxLayout()
# master group box
self.gb_chamber = QGroupBox(self.tr("Chamber Design"))
# nozzle settings button
self.btn_nozzle_settings = QPushButton(self.tr("Edit Nozzle"))
self.btn_nozzle_settings.setMinimumHeight(50)
master.addWidget(self.btn_nozzle_settings)
# nozzle info pane
fl_nozzle_info = QFormLayout()
gb_nozzle = QGroupBox(self.tr("Nozzle Info"))
gb_nozzle.setLayout(fl_nozzle_info)
self.lbl_nozzle_throat = QLabel()
fl_nozzle_info.addRow(QLabel(self.tr("Throat Diameter:")),
self.lbl_nozzle_throat)
self.lbl_nozzle_exit = QLabel()
fl_nozzle_info.addRow(QLabel(self.tr("Exit Diameter:")),
self.lbl_nozzle_exit)
self.lbl_nozzle_expansion_ratio = QLabel()
fl_nozzle_info.addRow(QLabel(self.tr("Expansion Ratio:")),
self.lbl_nozzle_expansion_ratio)
# add group box to master
master.addWidget(gb_nozzle)
master.addStretch()
# overall motor info pane
fl_motor_info = QFormLayout()
gb_motor = QGroupBox(self.tr("Motor Info"))
gb_motor.setLayout(fl_motor_info)
self.lbl_kn = QLabel()
fl_motor_info.addRow(QLabel(self.tr("Kn:")), self.lbl_kn)
self.lbl_port_throat = QLabel()
fl_motor_info.addRow(QLabel(self.tr("Port/Throat Ratio:")),
self.lbl_port_throat)
master.addWidget(gb_motor)
self.gb_chamber.setLayout(master)
def setup_gfx_ui():
# design overview
self.motor_display_view = QGraphicsView()
self.motor_display_scene = QGraphicsScene()
self.motor_display_view.setScene(self.motor_display_scene)
self.motor_display_view.show()
# sliced cross section
self.grain_slice_view = QGraphicsView()
self.grain_slice_scene = QGraphicsScene()
self.grain_slice_view.setScene(self.grain_slice_scene)
self.grain_slice_view.show()
# splitter
self.splt_gfx = QSplitter(Qt.Horizontal)
self.splt_gfx.addWidget(self.motor_display_view)
self.splt_gfx.addWidget(self.grain_slice_view)
self.splt_gfx.setStretchFactor(0, 10)
self.splt_gfx.setStretchFactor(1, 3)
self.splt_gfx.setMinimumHeight(50)
setup_grain_design()
setup_chamber_design()
setup_gfx_ui()
self.splt_grain_design = QSplitter(Qt.Horizontal)
self.splt_grain_design.addWidget(self.gb_design)
self.splt_grain_design.addWidget(self.gb_chamber)
self.splt_grain_design.setStretchFactor(0, 10)
self.splt_grain_design.setStretchFactor(1, 3)
self.splt_main = QSplitter(Qt.Vertical)
self.splt_main.addWidget(self.splt_grain_design)
self.splt_main.addWidget(self.splt_gfx)
self.splt_main.setSizes([300, 150])
layout = QVBoxLayout()
layout.addWidget(self.splt_main)
self.setLayout(layout)
def setup_ui(self):
"""
Initialize the widgets and layouts.
"""
self.setLayout(self.main_layout)
self.pv_layout.addWidget(self.pv_protocol_cmb)
self.pv_layout.addWidget(self.pv_name_line_edt)
self.pv_layout.addWidget(self.pv_connect_push_btn)
self.pv_add_panel.setLayout(self.pv_layout)
QTimer.singleShot(0, self.pv_name_line_edt.setFocus)
self.curve_settings_tab.setLayout(self.curves_tab_layout)
self.chart_settings_tab.setLayout(self.chart_settings_layout)
self.setup_chart_settings_layout()
self.data_settings_tab.setLayout(self.data_tab_layout)
self.setup_data_tab_layout()
self.tab_panel.addTab(self.curve_settings_tab, "Curves")
self.tab_panel.addTab(self.data_settings_tab, "Data")
self.tab_panel.addTab(self.chart_settings_tab, "Graph")
self.crosshair_settings_layout.addWidget(self.enable_crosshair_chk)
self.crosshair_settings_layout.addWidget(self.crosshair_coord_lbl)
self.zoom_x_layout.addWidget(self.zoom_in_x_btn)
self.zoom_x_layout.addWidget(self.zoom_out_x_btn)
self.zoom_y_layout.addWidget(self.zoom_in_y_btn)
self.zoom_y_layout.addWidget(self.zoom_out_y_btn)
self.view_all_reset_chart_layout.addWidget(self.reset_chart_btn)
self.view_all_reset_chart_layout.addWidget(self.view_all_btn)
self.pause_chart_layout.addWidget(self.pause_chart_btn)
self.import_export_data_layout.addWidget(self.import_data_btn)
self.import_export_data_layout.addWidget(self.export_data_btn)
self.chart_control_layout.addLayout(self.zoom_x_layout)
self.chart_control_layout.addLayout(self.zoom_y_layout)
self.chart_control_layout.addLayout(self.view_all_reset_chart_layout)
self.chart_control_layout.addLayout(self.pause_chart_layout)
self.chart_control_layout.addLayout(self.crosshair_settings_layout)
self.chart_control_layout.addLayout(self.import_export_data_layout)
self.chart_control_layout.insertSpacing(5, 30)
self.chart_layout.addWidget(self.chart)
self.chart_layout.addLayout(self.chart_control_layout)
self.chart_panel.setLayout(self.chart_layout)
self.splitter.addWidget(self.chart_panel)
self.splitter.addWidget(self.tab_panel)
self.splitter.setSizes([1, 0])
self.splitter.setHandleWidth(10)
self.splitter.setStretchFactor(0, 0)
self.splitter.setStretchFactor(1, 1)
self.charting_layout.addWidget(self.splitter)
self.body_layout.addWidget(self.pv_add_panel)
self.body_layout.addLayout(self.charting_layout)
self.body_layout.setSpacing(0)
self.body_layout.setContentsMargins(0, 0, 0, 0)
self.main_layout.addLayout(self.body_layout)
self.enable_chart_control_buttons(False)
handle = self.splitter.handle(1)
layout = QVBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
button = QToolButton(handle)
button.setArrowType(Qt.LeftArrow)
button.clicked.connect(lambda: self.handle_splitter_button(True))
layout.addWidget(button)
button = QToolButton(handle)
button.setArrowType(Qt.RightArrow)
button.clicked.connect(lambda: self.handle_splitter_button(False))
layout.addWidget(button)
handle.setLayout(layout)
