- 添加拖拽打开文件功能

yatengLG · yatengLG · commit 9cfffceb4ae4 · 2024-08-09T14:37:17.000+08:00
拖拽单个文件夹/单个图片到文件列表，可以直接打开文件夹/图片

- bug修复

  修复了打开空文件夹造成的闪退问题
diff --git a/ISAT/widgets/canvas.py b/ISAT/widgets/canvas.py
@@ -16,6 +16,7 @@ def __init__(self, mainwindow):
         super(AnnotationScene, self).__init__()
         self.mainwindow = mainwindow
         self.image_item: QtWidgets.QGraphicsPixmapItem = None
+        self.mask_item: QtWidgets.QGraphicsPixmapItem = None
         self.image_data = None
         self.current_graph: Polygon = None
         self.current_line: Line = None
@@ -63,6 +64,14 @@ def load_image(self, image_path: str):
         self.setSceneRect(self.image_item.boundingRect())
         self.change_mode_to_view()
 
+    def unload_image(self):
+        self.clear()
+        self.setSceneRect(QtCore.QRectF())
+        self.mainwindow.polygons.clear()
+        self.image_item = None
+        self.mask_item = None
+        self.current_graph = None
+
     def change_mode_to_create(self):
         if self.image_item is None:
             return
@@ -948,14 +957,16 @@ def update_mask(self):
             mask_image = QtGui.QImage(mask_image[:], mask_image.shape[1], mask_image.shape[0], mask_image.shape[1] * 3,
                                       QtGui.QImage.Format_RGB888)
             mask_pixmap = QtGui.QPixmap(mask_image)
-            self.mask_item.setPixmap(mask_pixmap)
+            if self.mask_item is not None:
+                self.mask_item.setPixmap(mask_pixmap)
         else:
             mask_image = np.zeros(self.image_data.shape, dtype=np.uint8)
             mask_image = cv2.addWeighted(self.image_data, 1, mask_image, 0, 0)
             mask_image = QtGui.QImage(mask_image[:], mask_image.shape[1], mask_image.shape[0], mask_image.shape[1] * 3,
                                       QtGui.QImage.Format_RGB888)
             mask_pixmap = QtGui.QPixmap(mask_image)
-            self.mask_item.setPixmap(mask_pixmap)
+            if self.mask_item is not None:
+                self.mask_item.setPixmap(mask_pixmap)
 
     def backspace(self):
         if self.mode == STATUSMode.CREATE:
diff --git a/ISAT/widgets/files_dock_widget.py b/ISAT/widgets/files_dock_widget.py
@@ -14,6 +14,8 @@ def __init__(self, mainwindow):
         self.listWidget.clicked.connect(self.listwidget_doubleclick)
         self.lineEdit_jump.returnPressed.connect(self.mainwindow.jump_to)
 
+        self.setAcceptDrops(True)
+
     def generate_item_and_itemwidget(self, file_name):
         item = QtWidgets.QListWidgetItem()
         item.setSizeHint(QtCore.QSize(200, 30))
@@ -58,3 +60,88 @@ def listwidget_doubleclick(self):
         row = self.listWidget.currentRow()
         self.mainwindow.current_index = row
         self.mainwindow.show_image(row)
+
+    def dragEnterEvent(self, event):
+        if event.mimeData().hasUrls():
+            event.accept()
+        else:
+            event.ignore()
+
+    def dropEvent(self, event):
+        if len(event.mimeData().urls()) != 1:
+            QtWidgets.QMessageBox.warning(self, 'Warning', 'Only support one path or dir.')
+            return
+        # 这里与mainwindow.opend_dir逻辑一致
+        path = event.mimeData().urls()[0].toLocalFile()
+        if os.path.isdir(path):
+            dir = path
+            # 等待sam线程退出，并清空特征缓存
+            if self.mainwindow.use_segment_anything:
+                self.mainwindow.seganythread.wait()
+                self.mainwindow.seganythread.results_dict.clear()
+
+            self.mainwindow.files_list.clear()
+            self.mainwindow.files_dock_widget.listWidget.clear()
+
+            files = []
+            suffixs = tuple(
+                ['{}'.format(fmt.data().decode('ascii').lower()) for fmt in QtGui.QImageReader.supportedImageFormats()])
+            for f in os.listdir(dir):
+                if f.lower().endswith(suffixs):
+                    # f = os.path.join(dir, f)
+                    files.append(f)
+            files = sorted(files)
+            self.mainwindow.files_list = files
+
+            self.mainwindow.files_dock_widget.update_widget()
+
+            self.mainwindow.current_index = 0
+
+            self.mainwindow.image_root = dir
+            self.mainwindow.actionOpen_dir.setStatusTip("Image root: {}".format(self.mainwindow.image_root))
+
+            self.mainwindow.label_root = dir
+            self.mainwindow.actionSave_dir.setStatusTip("Label root: {}".format(self.mainwindow.label_root))
+
+            if os.path.exists(os.path.join(dir, 'isat.yaml')):
+                # load setting yaml
+                self.mainwindow.config_file = os.path.join(dir, 'isat.yaml')
+                self.mainwindow.reload_cfg()
+
+            self.mainwindow.show_image(self.mainwindow.current_index)
+
+        if os.path.isfile(path):
+            # 等待sam线程退出，并清空特征缓存
+            if self.mainwindow.use_segment_anything:
+                self.mainwindow.seganythread.wait()
+                self.mainwindow.seganythread.results_dict.clear()
+
+            self.mainwindow.files_list.clear()
+            self.mainwindow.files_dock_widget.listWidget.clear()
+
+            suffixs = tuple(
+                ['{}'.format(fmt.data().decode('ascii').lower()) for fmt in QtGui.QImageReader.supportedImageFormats()])
+
+            dir, file = os.path.split(path)
+            files = []
+            if path.lower().endswith(suffixs):
+                files = [file]
+
+            self.mainwindow.files_list = files
+
+            self.mainwindow.files_dock_widget.update_widget()
+
+            self.mainwindow.current_index = 0
+
+            self.mainwindow.image_root = dir
+            self.mainwindow.actionOpen_dir.setStatusTip("Image root: {}".format(self.mainwindow.image_root))
+
+            self.mainwindow.label_root = dir
+            self.mainwindow.actionSave_dir.setStatusTip("Label root: {}".format(self.mainwindow.label_root))
+
+            if os.path.exists(os.path.join(dir, 'isat.yaml')):
+                # load setting yaml
+                self.mainwindow.config_file = os.path.join(dir, 'isat.yaml')
+                self.mainwindow.reload_cfg()
+
+            self.mainwindow.show_image(self.mainwindow.current_index)
diff --git a/ISAT/widgets/mainwindow.py b/ISAT/widgets/mainwindow.py
@@ -704,6 +704,9 @@ def update_group_display(self):
 
     def show_image(self, index:int, zoomfit:bool=True):
         self.reset_action()
+        self.scene.cancel_draw()
+        self.scene.unload_image()
+        self.annos_dock_widget.listWidget.clear()
         self.change_bit_map_to_label()
         #
         self.files_dock_widget.label_prev_state.setStyleSheet("background-color: {};".format('#999999'))
@@ -714,21 +717,15 @@ def show_image(self, index:int, zoomfit:bool=True):
         self.load_finished = False
         self.saved = True
         if not -1 < index < len(self.files_list):
-            self.scene.clear()
-            self.scene.setSceneRect(QtCore.QRectF())
             return
         try:
-            self.polygons.clear()
-            self.annos_dock_widget.listWidget.clear()
-            self.scene.cancel_draw()
             file_path = os.path.join(self.image_root, self.files_list[index])
             image_data = Image.open(file_path)
 
             self.png_palette = image_data.getpalette()
             if self.png_palette is not None and file_path.endswith('.png'):
                 self.statusbar.showMessage('This is a label file.')
                 self.can_be_annotated = False
-
             else:
                 self.can_be_annotated = True
 
