[GUI] add composition selection for Background image packs in ImagePackWidget

Author: Shengjie Xu

src/preppipe_gui_pyside6/forms/imagepackwidget.ui

@@ -73,7 +73,7 @@
           </sizepolicy>
          </property>
          <property name="title">
-          <string>图片选择</string>
+          <string>差分选择</string>
          </property>
         </widget>
        </item>
@@ -86,7 +86,7 @@
           </sizepolicy>
          </property>
          <property name="title">
-          <string>选区参数</string>
+          <string>选区修改参数</string>
          </property>
         </widget>
        </item>

src/preppipe_gui_pyside6/toolwidgets/imagepack.py

@@ -9,6 +9,7 @@
 from ..forms.generated.ui_imagepackwidget import Ui_ImagePackWidget
 from ..componentwidgets.imageviewerwidget import ImageViewerWidget
 from ..componentwidgets.maskinputwidget import MaskInputWidget
+from ..util.wait import WaitDialog
 
 TR_gui_tool_imagepack = TranslationDomain("gui_tool_imagepack")
 
@@ -68,25 +69,45 @@ def getChildTools(cls, packid : str | None = None, category_kind : ImagePackDesc
   current_pack : ImagePack
 
   # 当前的组合、选区参数
-  current_index : int | list[int] | None = None
+  current_index : int | list[int]
   current_mask_params : list[typing.Any] | None = None
   mask_param_widgets : list[MaskInputWidget]
+  background_radio_buttons : list[QRadioButton] | None = None
+
+  _tr_composition_selection = TR_gui_tool_imagepack.tr("composition_selection",
+    en="Composition Selection",
+    zh_cn="差分选择",
+    zh_hk="差分選擇",
+  )
+  _tr_mask_parameters = TR_gui_tool_imagepack.tr("mask_parameters",
+    en="Customization Parameters",
+    zh_cn="选区修改参数",
+    zh_hk="選區修改參數",
+  )
 
   def __init__(self, parent: QWidget):
     super(ImagePackWidget, self).__init__(parent)
     self.ui = Ui_ImagePackWidget()
     self.ui.setupUi(self)
     self.viewer = ImageViewerWidget(self, context_menu_callback=self.populate_image_rightclick_menu)
     self.ui.viewerLayout.addWidget(self.viewer)
-    self.current_index = None
+    self.bind_text(self.ui.sourceGroupBox.setTitle, self._tr_composition_selection)
+    self.bind_text(self.ui.forkParamGroupBox.setTitle, self._tr_mask_parameters)
+    self.current_index = 0
     self.current_mask_params = None
     self.mask_param_widgets = []
+    self.background_radio_buttons = None
 
   _tr_no_mask = TR_gui_tool_imagepack.tr("no_mask",
-    en="This image pack contains no customizable mask regions.",
+    en="This image pack contains no customizable regions.",
     zh_cn="该图片包没有可修改的选区。",
     zh_hk="該圖片包沒有可修改的選區。",
   )
+  _tr_no_composite = TR_gui_tool_imagepack.tr("no_composite",
+    en="This image pack contains no compositions.",
+    zh_cn="该图片包没有差分组合。",
+    zh_hk="該圖片包沒有差分組合。",
+  )
 
   def setData(self, packid : str | None = None, category_kind : ImagePackDescriptor.ImagePackType | None = None):
     if category_kind is not None:
@@ -119,14 +140,41 @@ def setData(self, packid : str | None = None, category_kind : ImagePackDescripto
         layout.addRow(nameLabel, inputWidget)
         self.add_translatable_widget_child(inputWidget)
         self.mask_param_widgets.append(inputWidget)
-        inputWidget.valueChanged.connect(functools.partial(self.handleMaskParamUpdate, index), Qt.QueuedConnection)
+        inputWidget.valueChanged.connect(functools.partial(self.handleMaskParamUpdate, index))
     else:
       layout = QVBoxLayout()
       label = QLabel(self._tr_no_mask.get())
       self.bind_text(label.setText, self._tr_no_mask)
       label.setWordWrap(True)
       layout.addWidget(label)
       self.ui.forkParamGroupBox.setLayout(layout)
+    if len(self.descriptor.composites_code) > 0:
+      match self.descriptor.packtype:
+        case ImagePackDescriptor.ImagePackType.BACKGROUND:
+          layout = QVBoxLayout()
+          self.background_radio_buttons = []
+          for index, code in enumerate(self.descriptor.composites_code):
+            radio_button = QRadioButton(code)
+            if name := self.descriptor.composites_references.get(code, None):
+              self.bind_text(radio_button.setText, name)
+            self.background_radio_buttons.append(radio_button)
+            if index == 0:
+              radio_button.setChecked(True)
+            radio_button.toggled.connect(self.handleBackgroundCompositionChange)
+            layout.addWidget(radio_button)
+          self.ui.sourceGroupBox.setLayout(layout)
+        case ImagePackDescriptor.ImagePackType.CHARACTER:
+          # TODO
+          pass
+        case _:
+          raise NotImplementedError(f"Pack type {self.descriptor.packtype} is not supported")
+    else:
+      layout = QVBoxLayout()
+      label = QLabel(self._tr_no_composite.get())
+      self.bind_text(label.setText, self._tr_no_composite)
+      label.setWordWrap(True)
+      layout.addWidget(label)
+      self.ui.sourceGroupBox.setLayout(layout)
     # setData() 需要尽快返回，让组件先显示出来，后续再更新内容
     QMetaObject.invokeMethod(self, "updateCurrentImage", Qt.QueuedConnection)
 
@@ -136,7 +184,22 @@ def handleMaskParamUpdate(self, index : int):
       raise RuntimeError("current_mask_params is None")
     widget = self.mask_param_widgets[index]
     self.current_mask_params[index] = widget.getValue()
-    self.updateCurrentPack()
+    WaitDialog.long_running_operation_start()
+    QMetaObject.invokeMethod(self, "updateCurrentPack", Qt.QueuedConnection)
+
+  @Slot()
+  def handleBackgroundCompositionChange(self):
+    if self.background_radio_buttons is None:
+      raise RuntimeError("background_radio_buttons is None")
+    new_index = 0
+    for index, button in enumerate(self.background_radio_buttons):
+      if button.isChecked():
+        new_index = index
+        break
+    if self.current_index == new_index:
+      return
+    self.current_index = new_index
+    QMetaObject.invokeMethod(self, "updateCurrentImage", Qt.QueuedConnection)
 
   @Slot()
   def updateCurrentPack(self):
@@ -151,8 +214,7 @@ def updateCurrentImage(self):
     if isinstance(self.current_index, list):
       img = self.current_pack.get_composed_image_lower(self.current_index)
     else:
-      index = self.current_index if self.current_index is not None else 0
-      img = self.current_pack.get_composed_image(index)
+      img = self.current_pack.get_composed_image(self.current_index)
     self.set_image(img)
 
   def set_image(self, image: PIL.Image.Image | ImageWrapper):

src/preppipe_gui_pyside6/util/wait.py

@@ -0,0 +1,7 @@
+from PySide6.QtCore import *
+
+class WaitDialog:
+  # TODO
+  @staticmethod
+  def long_running_operation_start():
+    QCoreApplication.processEvents()