class SubWindow(QMdiSubWindow):
def __init__(self, empty: bool = False):
QMdiSubWindow.__init__(self)
self.setWindowFlag(Qt.WindowMaximizeButtonHint, False)
if not empty:
self.setLayout(QHBoxLayout())
self.main_frame = EasyFrame()
self.main_frame.setMouseTracking(True)
super().layout().addWidget(self.main_frame)
self.main_frame.setLayout(QGridLayout())
self.layout = self.new_layout
self._icon = QIcon(image_file_path + "rs_icon.png")
self.setWindowIcon(self._icon)
def new_layout(self) -> QLayout:
"""
Returns a QGridLayout.
:return QLayout: The grid layout.
"""
return self.main_frame.layout()
class CamView(AbstractView):
def __init__(self,
name: str = "CAM_NONE",
log_handlers: [StreamHandler] = None):
self._logger = getLogger(__name__)
if log_handlers:
for h in log_handlers:
self._logger.addHandler(h)
self._logger.debug("Initializing")
super().__init__(name)
""" Min size for cam window """
self._subwindow_height = 222
self._subwindow_width = 518
self._initialization_bar_frame = EasyFrame()
self._initialization_bar_frame.setMouseTracking(True)
self._initialization_bar_frame.setMaximumHeight(70)
self._initialization_bar_layout = QVBoxLayout(
self._initialization_bar_frame)
self._initialization_bar_label = QLabel(self._initialization_bar_frame)
self._initialization_bar_label.setMouseTracking(True)
self._initialization_bar = QProgressBar(self._initialization_bar_frame)
self._initialization_bar.setMouseTracking(True)
self._initialization_bar.setMaximumHeight(15)
self._initialization_bar.setTextVisible(True)
self._initialization_bar.setAlignment(Qt.AlignHCenter)
self._initialization_bar_layout.addWidget(
self._initialization_bar_label)
self._initialization_bar_layout.addWidget(self._initialization_bar)
self._cam_settings_frame = EasyFrame()
self._cam_settings_layout = QGridLayout(self._cam_settings_frame)
self._resolution_selector_label = QLabel(self._cam_settings_frame)
self._resolution_selector_label.setAlignment(Qt.AlignLeft)
self._resolution_selector = QComboBox(self._cam_settings_frame)
self._resolution_selector.setMaximumHeight(22)
self._cam_settings_layout.addWidget(self._resolution_selector_label, 0,
0)
self._cam_settings_layout.addWidget(self._resolution_selector, 0, 1)
self._fps_selector_label = QLabel(self._cam_settings_frame)
self._fps_selector_label.setAlignment(Qt.AlignLeft)
self._fps_selector = QComboBox(self._cam_settings_frame)
self._fps_selector.setMaximumHeight(22)
self._cam_settings_layout.addWidget(self._fps_selector_label, 1, 0)
self._cam_settings_layout.addWidget(self._fps_selector, 1, 1)
self._show_feed_checkbox_label = QLabel(self._cam_settings_frame)
self._show_feed_checkbox_label.setAlignment(Qt.AlignLeft)
self._show_feed_checkbox = QCheckBox()
self._show_feed_checkbox.setChecked(True)
self._show_feed_checkbox.setLayoutDirection(Qt.RightToLeft)
self._cam_settings_layout.addWidget(self._show_feed_checkbox_label, 2,
0)
self._cam_settings_layout.addWidget(self._show_feed_checkbox, 2, 1)
self._use_cam_checkbox_label = QLabel(self._cam_settings_frame)
self._use_cam_checkbox_label.setAlignment(Qt.AlignLeft)
self._use_cam_checkbox = QCheckBox()
self._use_cam_checkbox.setChecked(True)
self._use_cam_checkbox.setLayoutDirection(Qt.RightToLeft)
self._cam_settings_layout.addWidget(self._use_cam_checkbox_label, 3, 0)
self._cam_settings_layout.addWidget(self._use_cam_checkbox, 3, 1)
self._use_overlay_checkbox_label = QLabel(self._cam_settings_frame)
self._use_overlay_checkbox_label.setAlignment(Qt.AlignLeft)
self._use_overlay_checkbox = QCheckBox()
self._use_overlay_checkbox.setChecked(True)
self._use_overlay_checkbox.setLayoutDirection(Qt.RightToLeft)
self._cam_settings_layout.addWidget(self._use_overlay_checkbox_label,
4, 0)
self._cam_settings_layout.addWidget(self._use_overlay_checkbox, 4, 1)
self._image_display_frame = EasyFrame()
self._image_display_layout = QVBoxLayout(self._image_display_frame)
self._image_display_label = QLabel(self._image_display_frame)
self._image_display_label.setAlignment(Qt.AlignHCenter)
self._image_display = QLabel(self._image_display_frame)
self._image_display.setAlignment(Qt.AlignCenter)
self._image_display.setMouseTracking(True)
self._image_display_layout.addWidget(self._image_display_label)
self._image_display_layout.addWidget(self._image_display)
spacer = QSpacerItem(1, 1, vData=QSizePolicy.Expanding)
self._dev_sets_frame = EasyFrame()
self._dev_sets_layout = QVBoxLayout(self._dev_sets_frame)
self.config_button = ClickAnimationButton()
self.config_button.clicked.connect(self._config_button_handler)
self.layout().addWidget(self.config_button, 0, 0,
Qt.AlignTop | Qt.AlignRight)
self.config_button.setFixedSize(30, 25)
self.config_button.setStyleSheet(
"background-color: rgba(200, 200, 200, 50%)")
self._dev_sets_layout.addWidget(self._cam_settings_frame)
self._dev_sets_layout.addItem(spacer)
self.layout().addWidget(self._image_display, 0, 0)
self.layout().addWidget(self._initialization_bar_frame, 0, 0)
self._config_items = [
self._resolution_selector,
self._fps_selector,
self._use_cam_checkbox,
self._show_feed_checkbox,
self._use_overlay_checkbox,
]
config_win_w = 350
config_win_h = len(self._config_items) * 30
self.config_button.raise_()
self._config_win = ConfigPopUp()
self._config_win.setLayout(self._dev_sets_layout)
self._config_win.setFixedSize(config_win_w, config_win_h)
self._rec_label = QLabel()
self._rec_label.setStyleSheet(
"background-color: rgba(0, 0, 0, 0%); color: red; font: 20px")
self._rec_label.hide()
self.layout().addWidget(self._rec_label, 0, 0,
Qt.AlignBottom | Qt.AlignRight)
self.layout().setMargin(0)
self._window_changing = False
self._showing_images = False
self._hidden = False
w = 320
self._aspect_ratio = 9 / 16
self.resize(w, self.heightForWidth(w))
self._strings = dict()
self._lang_enum = LangEnum.ENG
self.setMinimumSize(self._subwindow_width, self._subwindow_height)
self.old_size = QSize(self.width(), self.height())
self._logger.debug("Initialized")
def set_show_feed_button_handler(self, func) -> None:
"""
Add handler for show camera selector.
:param func: The handler.
:return None:
"""
self._logger.debug("running")
self._show_feed_checkbox.toggled.connect(func)
self._logger.debug("done")
def set_resolution_selector_handler(self, func) -> None:
"""
Add handler for resolution selector.
:param func: The handler.
:return None:
"""
self._logger.debug("running")
self._resolution_selector.activated.connect(func)
self._logger.debug("done")
def set_fps_selector_handler(self, func) -> None:
"""
Add handler for resolution selector.
:param func: The handler.
:return None:
"""
self._logger.debug("running")
self._fps_selector.activated.connect(func)
self._logger.debug("done")
def set_use_cam_button_handler(self, func) -> None:
"""
Add handler for use camera selector.
:param func: The handler.
:return None:
"""
self._logger.debug("running")
self._use_cam_checkbox.toggled.connect(func)
self._logger.debug("done")
def set_use_overlay_button_handler(self, func) -> None:
"""
Add handler for use camera selector.
:param func: The handler.
:return None:
"""
self._logger.debug("running")
self._use_overlay_checkbox.toggled.connect(func)
self._logger.debug("done")
def _config_button_handler(self) -> None:
"""
Show config pop up.
:return None:
"""
self._logger.debug("running")
self.config_button.setStyleSheet(
"background-color: rgba(200, 200, 200, 50%)")
self._config_win.exec_()
self._logger.debug("done")
@property
def language(self) -> LangEnum:
"""
:return: The current lang enum being used.
"""
return self._lang_enum
@language.setter
def language(self, lang: LangEnum) -> None:
"""
Set the language for this view object.
:param lang: The language to use.
:return None:
"""
self._logger.debug("running")
self._lang_enum = lang
self._strings = strings[lang]
self._set_texts()
self._set_tooltips()
self._logger.debug("done")
@property
def resolution_list(self) -> list:
"""
Get list of resolutions.
:return list: The list of resolutions.
"""
ret = list()
return ret
@resolution_list.setter
def resolution_list(self, res_list: list) -> None:
"""
Set list of resolutions available to res_list.
:param res_list: The list of available resolutions.
:return None:
"""
self._logger.debug("running")
self._resolution_selector.clear()
for item in res_list:
self._resolution_selector.addItem(str(item))
self._logger.debug("done")
@property
def resolution(self) -> str:
"""
Get the current resolution selection.
:return str: The current resolution.
"""
return self._resolution_selector.currentText()
@resolution.setter
def resolution(self, res: str) -> None:
"""
Set the current resolution selection.
:param res: The resolution to set to.
:return None:
"""
self._resolution_selector.setCurrentIndex(
self._resolution_selector.findText(res))
@property
def fps_list(self) -> list:
"""
Get list of fps options.
:return list: The list of fps options.
"""
ret = list()
return ret
@fps_list.setter
def fps_list(self, fps_list: list) -> None:
"""
Set list of available fps to fps_list.
:param fps_list:
:return None:
"""
self._logger.debug("running")
self._fps_selector.clear()
for item in fps_list:
self._fps_selector.addItem(str(item))
self._logger.debug("done")
@property
def fps(self) -> str:
"""
Get the current fps selection.
:return str: The current fps selection.
"""
return self._fps_selector.currentText()
@fps.setter
def fps(self, fps: str) -> None:
"""
Set the current fps selection.
:param fps: The fps to set to.
:return None:
"""
self._logger.debug("running")
self._fps_selector.setCurrentIndex(self._fps_selector.findText(fps))
self._logger.debug("done")
@property
def use_feed(self) -> bool:
"""
Get the current use_cam setting.
:return bool: User selection for using cam.
"""
return self._show_feed_checkbox.isChecked()
@use_feed.setter
def use_feed(self, useable: bool) -> None:
"""
Set use_cam setting.
:param useable: The setting to set to.
:return None:
"""
self._logger.debug("running")
self._show_feed_checkbox.setChecked(useable)
self._logger.debug("Done")
@property
def use_cam(self) -> bool:
"""
Get the current use_cam setting.
:return bool: User selection for using cam.
"""
return self._use_cam_checkbox.isChecked()
@use_cam.setter
def use_cam(self, useable: bool) -> None:
"""
Set use_cam setting.
:param useable: The setting to set to.
:return None:
"""
self._logger.debug("running")
self._use_cam_checkbox.setChecked(useable)
self._logger.debug("Done")
@property
def use_overlay(self) -> bool:
"""
Get the current use_overlay setting.
:return bool: User selection for using overlay.
"""
return self._use_overlay_checkbox.isChecked()
@use_overlay.setter
def use_overlay(self, useable: bool) -> None:
"""
Set use_overlay setting.
:param useable: The setting to set to.
:return None:
"""
self._logger.debug("running")
self._use_overlay_checkbox.setChecked(useable)
self._logger.debug("Done")
def heightForWidth(self, w: int) -> int:
return int(w * self._aspect_ratio)
def widthForHeight(self, h: int) -> int:
return int(h / self._aspect_ratio)
def hideEvent(self, hideEvent: QHideEvent):
"""
Track minimize event.
:param hideEvent:
:return None:
"""
self._hidden = True
def showEvent(self, showEvent: QShowEvent):
"""
Track restore event.
:param showEvent:
:return None:
"""
self._hidden = False
def update_image(self, image: QPixmap = None, msg: str = None) -> None:
"""
Update image viewer with new image.
:param image: The new image to show.
:param msg: The text to show if no image.
:return None:
"""
self._logger.debug("running")
if not self._window_changing:
if image is not None:
temp_image_w = image.scaledToWidth(self.width() - 15)
temp_image_h = image.scaledToHeight(self.height() - 35)
if temp_image_w.height() > self.height() - 35:
self._image_display.setPixmap(temp_image_h)
else:
self._image_display.setPixmap(temp_image_w)
elif msg is not None:
self._image_display.setText(msg)
self._logger.debug("done")
def show_images(self) -> None:
"""
Show image display and hide initialization bar.
:return None:
"""
self._logger.debug("running")
geo = self.geometry()
self._showing_images = True
self._initialization_bar_frame.hide()
self._image_display.show()
self.config_button.show()
self.setStyleSheet("background-color: black")
self.setGeometry(geo)
self._logger.debug("done")
def show_initialization(self) -> None:
"""
Show initialization bar and hide image display.
:return None:
"""
self._logger.debug("running")
self._showing_images = False
self._image_display.hide()
self.config_button.hide()
self._initialization_bar_frame.show()
self._logger.debug("done")
def update_init_bar(self, progress: int) -> None:
"""
set progress bar value to progress.
:param progress: The value to set progress bar to.
:return None:
"""
self._logger.debug("running")
if progress > 100:
progress = 100
elif progress < 0:
progress = 0
self._initialization_bar.setValue(progress)
self._logger.debug("done")
def set_config_active(self, is_active: bool) -> None:
"""
Set whether this camera config options are usable.
:param is_active: Usable.
:return None:
"""
self._logger.debug("running")
if self._showing_images:
if is_active:
self._rec_label.hide()
elif self.use_cam:
self._rec_label.show()
for item in self._config_items:
item.setEnabled(is_active)
self._logger.debug("done")
def _set_texts(self) -> None:
"""
Set the texts in this view object.
:return None:
"""
self._logger.debug("running")
self._initialization_bar_label.setText(
self._strings[StringsEnum.INITIALIZATION_BAR_LABEL])
self._initialization_bar.setValue(0)
self._image_display_label.setText(
self._strings[StringsEnum.IMAGE_DISPLAY_LABEL])
self._image_display.setText(self._strings[StringsEnum.IMAGE_DISPLAY])
self._show_feed_checkbox_label.setText(
self._strings[StringsEnum.SHOW_FEED_CHECKBOX_LABEL])
self._use_cam_checkbox_label.setText(
self._strings[StringsEnum.USE_CAM_CHECKBOX_LABEL])
self._use_overlay_checkbox_label.setText(
self._strings[StringsEnum.USE_OVERLAY_CHECKBOX_LABEL])
self._resolution_selector_label.setText(
self._strings[StringsEnum.RESOLUTION_SELECTOR_LABEL])
self._fps_selector_label.setText(
self._strings[StringsEnum.FPS_SELECTOR_LABEL])
self._config_win.setWindowTitle(
self.get_name() + " " +
self._strings[StringsEnum.CONFIG_TAB_LABEL])
self.config_button.setText("...")
self._rec_label.setText("rec ●")
self._logger.debug("done")
def _set_tooltips(self) -> None:
"""
Set the tooltips in this view object.
:return None:
"""
self._logger.debug("running")
self._resolution_selector_label.setToolTip(
self._strings[StringsEnum.RESOLUTION_SELECTOR_TOOLTIP])
self._resolution_selector.setToolTip(
self._strings[StringsEnum.RESOLUTION_SELECTOR_TOOLTIP])
self._fps_selector_label.setToolTip(
self._strings[StringsEnum.FPS_SELECTOR_TOOLTIP])
self._fps_selector.setToolTip(
self._strings[StringsEnum.FPS_SELECTOR_TOOLTIP])
self._show_feed_checkbox_label.setToolTip(
self._strings[StringsEnum.SHOW_FEED_CHECKBOX_TOOLTIP])
self._show_feed_checkbox.setToolTip(
self._strings[StringsEnum.SHOW_FEED_CHECKBOX_TOOLTIP])
self._use_cam_checkbox_label.setToolTip(
self._strings[StringsEnum.USE_CAM_CHECKBOX_TOOLTIP])
self._use_cam_checkbox.setToolTip(
self._strings[StringsEnum.USE_CAM_CHECKBOX_TOOLTIP])
self._use_overlay_checkbox_label.setToolTip(
self._strings[StringsEnum.USE_OVERLAY_TOOLTIP])
self._use_overlay_checkbox.setToolTip(
self._strings[StringsEnum.USE_OVERLAY_TOOLTIP])
self._image_display.setToolTip(
self._strings[StringsEnum.IMAGE_DISPLAY_TOOLTIP])
self._logger.debug("done")