源码

CU/drop.py

@@ -0,0 +1,104 @@
+# File: drop_position_manager.py
+
+import configparser
+import os
+from typing import Optional
+from Model.Position import Real_Position
+
+class DropPositionManager:
+    def __init__(self, config_path="drop.ini"):
+        self.config_path = config_path
+        self.config = configparser.ConfigParser()
+        self._load_config()
+
+    def _load_config(self):
+        if not os.path.exists(self.config_path):
+            raise FileNotFoundError(f"配置文件不存在: {self.config_path}")
+        self.config.read(self.config_path, encoding='utf-8')
+
+    def _save_config(self):
+        with open(self.config_path, 'w', encoding='utf-8') as f:
+            self.config.write(f)
+
+    def get_next_drop_position(self, lineid: int = 1) -> Optional[Real_Position]:
+        print(f"\n🔄 开始获取 lineid={lineid} 的下一个 drop 点...")
+
+        line_section = f"DropLine{lineid}"
+        print(f"🔍 查找路径 section: {line_section}")
+
+        if not self.config.has_section(line_section):
+            print(f"❌ 错误：配置文件中不存在 section [{line_section}]")
+            print(f"📊 所有 sections: {list(self.config.sections())}")
+            return None
+
+        current_index = self.config.getint(line_section, "current_index", fallback=0)
+        print(f"📌 当前索引: {current_index} (fallback=0 如果未设置)")
+
+        points = []
+        found_sections = []
+        matched_sections = []
+
+        for sec in self.config.sections():
+            if sec.startswith("DropPoints"):
+                found_sections.append(sec)
+                try:
+                    file_lineid = self.config.getint(sec, "lineid")
+                    print(f"🔍 {sec}: lineid={file_lineid}")
+                    if file_lineid == lineid:
+                        matched_sections.append(sec)
+                        pos = Real_Position()
+                        pos.X = self.config.getfloat(sec, "x")
+                        pos.Y = self.config.getfloat(sec, "y")
+                        pos.Z = self.config.getfloat(sec, "z")
+                        pos.U = self.config.getfloat(sec, "u")
+                        pos.V = self.config.getfloat(sec, "v")
+                        pos.W = self.config.getfloat(sec, "w")
+                        order = self.config.getint(sec, "order")
+                        points.append((order, pos))
+                        print(f"✅ 匹配到点: {sec}, order={order}, 位置=({pos.X:.3f}, {pos.Y:.3f})")
+                except Exception as e:
+                    print(f"❌ 读取 {sec} 失败: {e}")
+
+        print(f"📁 找到 DropPoints: {found_sections}")
+        print(f"🟢 匹配 lineid={lineid} 的点: {matched_sections}")
+        print(f"📊 共收集 {len(points)} 个有效点")
+
+        if not points:
+            print("❌ 没有可用的 drop 点！请检查：lineid 是否匹配，字段是否正确")
+            return None
+
+        # 按 order 排序
+        points.sort(key=lambda x: x[0])
+        sorted_positions = [pos for _, pos in points]
+        count = len(sorted_positions)
+        actual_index = current_index % count
+
+        selected_pos = sorted_positions[actual_index]
+        print(f"🎯 选择第 {actual_index} 个点: X={selected_pos.X:.3f}, Y={selected_pos.Y:.3f}, Z={selected_pos.Z:.3f}")
+
+        # 索引 +1 并保存回 ini
+        next_index = current_index + 1
+        self.config.set(line_section, "current_index", str(next_index))
+        self._save_config()
+        print(f"💾 已保存 current_index = {next_index} 到 [{line_section}]")
+
+        return selected_pos
+
+
+# ✅ 主函数：用于测试
+if __name__ == "__main__":
+    # 创建管理器，加载 drop.ini
+    manager = DropPositionManager("drop.ini")
+
+    # 获取第一个 drop 点（lineid=1）
+    pos = manager.get_next_drop_position(lineid=1)
+
+    if pos is not None:
+        print(f"\n🎉 成功返回点位: X={pos.X}, Y={pos.Y}, Z={pos.Z}, U={pos.U}, V={pos.V}, W={pos.W}")
+    else:
+        print("\n❌ get_next_drop_position 返回了 None，请根据上面的日志排查")
+        print("💡 常见原因：")
+        print("   1. drop.ini 缺少 current_index")
+        print("   2. lineid 不匹配")
+        print("   3. Real_Position 属性名错误（应为大写 X/Y/Z）")
+        print("   4. 文件路径不对")