perf: 优化渲染

Author: Littlefean

file_observer.py

@@ -97,16 +97,6 @@ def _entity_folders(self, folder: EntityFolder) -> list[EntityFolder]:
                 res.extend(self._entity_folders(file))
         return res
 
-    def get_entity_folders(self) -> list[EntityFolder]:
-        if self.root_folder is None:
-            return []
-        return self._entity_folders(self.root_folder)
-
-    def get_entity_files(self) -> list[EntityFile]:
-        if self.root_folder is None:
-            return []
-        return self._entity_files(self.root_folder)
-
     def get_entity_by_location(
         self, location_world: NumberVector
     ) -> EntityFile | EntityFolder | None:

main.py

@@ -307,24 +307,8 @@ def paintEvent(self, a0: QPaintEvent | None):
         # 画场景物体
 
         # 先画文件夹
-        for folder_entity in self.file_observer.get_entity_folders():
-            if folder_entity.body_shape.is_collision(self.camera.cover_world_rectangle):
-                # 获得一个世界坐标系的视野矩形，用于排除视野之外的绘制，防止放大了之后会卡
-                paint_folder_rect(
-                    painter,
-                    self.camera,
-                    folder_entity,
-                    folder_entity.deep_level / self.file_observer.folder_max_deep_index,
-                )
-        # 后画文件
-        for file_entity in self.file_observer.get_entity_files():
-            if file_entity.body_shape.is_collision(self.camera.cover_world_rectangle):
-                paint_file_rect(
-                    painter,
-                    self.camera,
-                    file_entity,
-                    file_entity.deep_level / self.file_observer.folder_max_deep_index,
-                )
+        if self.file_observer.root_folder:
+            self.paint_folder_dfs(painter, self.file_observer.root_folder)
         # 绘制选中的区域
         if self.file_observer.dragging_entity:
             paint_selected_rect(
@@ -342,6 +326,34 @@ def paintEvent(self, a0: QPaintEvent | None):
                 f"drag locked: {self.file_observer.is_drag_locked}",
             ],
         )
+    
+    def paint_folder_dfs(self, painter: QPainter, folder_entity: EntityFolder):
+        """
+        递归绘制文件夹，遇到视野之外的直接排除
+        """
+        # 先绘制本体
+        if folder_entity.body_shape.is_collision(self.camera.cover_world_rectangle):
+            paint_folder_rect(
+                painter,
+                self.camera,                
+                folder_entity,
+                folder_entity.deep_level / self.file_observer.folder_max_deep_index,
+            )
+        else: 
+            return
+        # 递归绘制子文件夹
+        for child in folder_entity.children:
+            if isinstance(child, EntityFolder):
+                self.paint_folder_dfs(painter, child)
+            elif isinstance(child, EntityFile):
+                if child.body_shape.is_collision(self.camera.cover_world_rectangle):
+                    paint_file_rect(
+                        painter,
+                        self.camera,
+                        child,
+                        child.deep_level / self.file_observer.folder_max_deep_index,
+                    )
+        pass
 
     def mousePressEvent(self, a0: QMouseEvent | None):
         assert a0 is not None