[Function] Add automatic Table Structure Reordering Function. (#176)

* fix bug in PPOCRLabel.py

* code style check

* code style check

* code style check

* code style check

* Add resort function

* fix code style

* fix code style

* modify resort function

* add keys info

* modify resortBoxPosion to resortBoxPosition

---------

Co-authored-by: guoshengjian <guoshengjian@MacBook-Pro.local>

Author: Sam-gsj

PPOCRLabel.py

@@ -349,6 +349,8 @@ def get_str(str_id):
         self.SaveButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
         self.DelButton = QToolButton()
         self.DelButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
+        self.ResortButton = QToolButton()
+        self.ResortButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
 
         leftTopToolBox = QGridLayout()
         leftTopToolBox.addWidget(self.newButton, 0, 0, 1, 1)
@@ -438,6 +440,7 @@ def get_str(str_id):
         leftbtmtoolbox = QHBoxLayout()
         leftbtmtoolbox.addWidget(self.SaveButton)
         leftbtmtoolbox.addWidget(self.DelButton)
+        leftbtmtoolbox.addWidget(self.ResortButton)
         leftbtmtoolboxcontainer = QWidget()
         leftbtmtoolboxcontainer.setLayout(leftbtmtoolbox)
         listLayout.addWidget(leftbtmtoolboxcontainer)
@@ -897,6 +900,14 @@ def get_str(str_id):
             get_str("expandBoxDetail"),
             enabled=False,
         )
+        resort = action(
+            get_str("resortposition"),
+            self.resortBoxPosition,
+            "Ctrl+B",
+            "resort",
+            get_str("resortpositiondetail"),
+            enabled=True,
+        )
 
         self.editButton.setDefaultAction(edit)
         self.newButton.setDefaultAction(create)
@@ -906,6 +917,7 @@ def get_str(str_id):
         self.AutoRecognition.setDefaultAction(AutoRec)
         self.reRecogButton.setDefaultAction(reRec)
         self.tableRecButton.setDefaultAction(tableRec)
+        self.ResortButton.setDefaultAction(resort)
         # self.preButton.setDefaultAction(openPrevImg)
         # self.nextButton.setDefaultAction(openNextImg)
 
@@ -995,6 +1007,7 @@ def get_str(str_id):
             lock=lock,
             exportJSON=exportJSON,
             expand=expand,
+            resort=resort,
             fileMenuActions=(
                 opendir,
                 open_dataset_dir,
@@ -1012,6 +1025,7 @@ def get_str(str_id):
                 delete,
                 singleRere,
                 cellreRec,
+                resort,
                 None,
                 undo,
                 undoLastPoint,
@@ -3665,6 +3679,83 @@ def expandSelectedShape(self):
             self.updateBoxlist()
             self.setDirty()
 
+    def sort_rectangles(self, rectangles, row_height_threshold=0.5):
+        if not rectangles:
+            return []
+
+        def get_top_left(rect):
+            xs = [p[0] for p in rect]
+            ys = [p[1] for p in rect]
+            return (min(xs), min(ys))
+
+        avg_height = sum(
+            [max(p[1] for p in rect) - min(p[1] for p in rect) for rect in rectangles]
+        ) / len(rectangles)
+        threshold = avg_height * row_height_threshold
+        indexed_rects = [(i, get_top_left(rect)) for i, rect in enumerate(rectangles)]
+        indexed_rects.sort(key=lambda x: x[1][1])
+        rows = []
+        current_row = []
+        last_y = indexed_rects[0][1][1]
+        for item in indexed_rects:
+            i, (x, y) = item
+            if abs(y - last_y) <= threshold:
+                current_row.append(item)
+            else:
+                rows.append(current_row)
+                current_row = [item]
+            last_y = y
+        if current_row:
+            rows.append(current_row)
+        sorted_rects = []
+        for row in rows:
+            row.sort(key=lambda x: x[1][0])
+            sorted_rects.extend([rectangles[i] for i, _ in row])
+        return sorted_rects
+
+    def resortBoxPosition(self):
+        # get original elements
+        items = []
+        for i in range(self.BoxList.count()):
+            item = self.BoxList.item(i)
+            items.append({"text": item.text(), "object": item})
+        # get coordinate points
+        rectangles = []
+        for item in items:
+            text = item["text"]
+            try:
+                rect = ast.literal_eval(text)  # 转为列表
+                rectangles.append(rect)
+            except (ValueError, SyntaxError) as e:
+                logger.error(f"Error parsing text: {text}")
+                continue
+        # start resort
+        sorted_rectangles = self.sort_rectangles(rectangles, row_height_threshold=0.5)
+        # old_idx <--> new_idx
+        index_map = []
+        for sorted_rect in sorted_rectangles:
+            for old_idx, rect in enumerate(rectangles):
+                if rect == sorted_rect:
+                    index_map.append(old_idx)
+                    break
+        # resort BoxList labelList canvas.shapes
+        items = [self.BoxList.takeItem(0) for _ in range(self.BoxList.count())]
+        items_label = [
+            self.labelList.takeItem(0) for _ in range(self.labelList.count())
+        ]
+        shapes = self.canvas.shapes
+        self.canvas.shapes = []
+        for new_idx in range(len(index_map)):
+            old_idx = index_map[new_idx]
+            self.BoxList.insertItem(new_idx, items[old_idx])
+            self.labelList.insertItem(new_idx, items_label[old_idx])
+            self.canvas.shapes.insert(new_idx, shapes[old_idx])
+        QMessageBox.information(
+            self,
+            "Information",
+            "resort success!",
+        )
+
 
 def inverted(color):
     return QColor(*[255 - v for v in color.getRgb()])