forked from huangxin/ailai
105 lines
4.2 KiB
Python
105 lines
4.2 KiB
Python
# 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. 文件路径不对")
|