from PyQt5.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QStackedLayout, QFileDialog, QHeaderView from PyQt5.QtCore import Qt from PyQt5.QtWidgets import QTreeWidgetItem as TreeItem from qfluentwidgets import TitleLabel, BodyLabel, SubtitleLabel, StrongBodyLabel, HorizontalSeparator, PushButton, TreeWidget, InfoBar,FluentIcon, Dialog,SubtitleLabel,BodyLabel from PyQt5.QtWidgets import QDialog, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QSpinBox, QPushButton, QTableWidget, QTableWidgetItem, QHeaderView, QCheckBox from qfluentwidgets import CardWidget, LineEdit, SpinBox, CheckBox, TextEdit, PrimaryPushButton, PushButton, InfoBar from qfluentwidgets import HeaderCardWidget from PyQt5.QtWidgets import QScrollArea, QWidget from qfluentwidgets import theme, Theme import os import requests import zipfile import io import re import shutil import yaml import textwrap from jinja2 import Template def preserve_all_user_regions(new_code, old_code): """ 自动保留所有 /* USER XXX BEGIN */ ... /* USER XXX END */ 区域内容。 new_code: 模板生成的新代码 old_code: 旧代码 返回：合并后的代码 """ import re pattern = re.compile( r"/\*\s*(USER [A-Z0-9_ ]+)\s*BEGIN\s*\*/(.*?)/\*\s*\1\s*END\s*\*/", re.DOTALL ) old_regions = {m.group(1): m.group(2) for m in pattern.finditer(old_code or "")} def repl(m): region = m.group(1) old_content = old_regions.get(region) if old_content is not None: return m.group(0).replace(m.group(2), old_content) return m.group(0) return pattern.sub(repl, new_code) def save_with_preserve(path, new_code): """ 写入文件，自动保留所有用户自定义区域 """ if os.path.exists(path): with open(path, "r", encoding="utf-8") as f: old_code = f.read() new_code = preserve_all_user_regions(new_code, old_code) with open(path, "w", encoding="utf-8") as f: f.write(new_code) class IocConfig: def __init__(self, ioc_path): self.ioc_path = ioc_path self.config = {} self._parse() def _parse(self): with open(self.ioc_path, encoding='utf-8') as f: for line in f: line = line.strip() if not line or line.startswith('#'): continue if '=' in line: key, value = line.split('=', 1) self.config[key.strip()] = value.strip() def is_freertos_enabled(self): ip_keys = [k for k in self.config if k.startswith('Mcu.IP')] for k in ip_keys: if self.config[k] == 'FREERTOS': return True for k in self.config: if k.startswith('FREERTOS.'): return True return False class DataInterface(QWidget): def __init__(self, parent=None): super().__init__(parent=parent) self.setObjectName("dataInterface") # 属性初始化 self.project_path = "" self.project_name = "" self.ioc_file = "" self.freertos_enabled = False # 新增属性 # 主布局 self.stacked_layout = QStackedLayout(self) self.setLayout(self.stacked_layout) # --- 页面1：工程路径选择 --- self.select_widget = QWidget() outer_layout = QVBoxLayout(self.select_widget) outer_layout.setContentsMargins(0, 0, 0, 0) outer_layout.addStretch() # 直接用布局和控件，无卡片 content_layout = QVBoxLayout() content_layout.setSpacing(28) content_layout.setContentsMargins(48, 48, 48, 48) # 主标题 title = TitleLabel("MRobot 代码生成") title.setAlignment(Qt.AlignCenter) content_layout.addWidget(title) # 副标题 subtitle = BodyLabel("请选择您的由CUBEMX生成的工程路径（.ico所在的目录），然后开启代码之旅！") subtitle.setAlignment(Qt.AlignCenter) content_layout.addWidget(subtitle) # 简要说明 desc = BodyLabel("支持自动配置和生成任务，自主选择模块代码倒入，自动识别cubemx配置！") desc.setAlignment(Qt.AlignCenter) content_layout.addWidget(desc) content_layout.addSpacing(18) # 选择项目路径按钮 self.choose_btn = PushButton(FluentIcon.FOLDER, "选择项目路径") self.choose_btn.setFixedWidth(200) self.choose_btn.clicked.connect(self.choose_project_folder) content_layout.addWidget(self.choose_btn, alignment=Qt.AlignmentFlag.AlignCenter) # 更新代码库按钮 self.update_template_btn = PushButton(FluentIcon.SYNC, "更新代码库") self.update_template_btn.setFixedWidth(200) self.update_template_btn.clicked.connect(self.update_user_template) content_layout.addWidget(self.update_template_btn, alignment=Qt.AlignmentFlag.AlignCenter) content_layout.addSpacing(10) content_layout.addStretch() outer_layout.addLayout(content_layout) outer_layout.addStretch() self.stacked_layout.addWidget(self.select_widget) # --- 页面2：主配置页面 --- self.config_widget = QWidget() main_layout = QVBoxLayout(self.config_widget) main_layout.setContentsMargins(32, 32, 32, 32) main_layout.setSpacing(18) # 顶部项目信息 info_layout = QHBoxLayout() self.back_btn = PushButton(FluentIcon.SKIP_BACK, "返回") self.back_btn.setFixedWidth(90) self.back_btn.clicked.connect(self.back_to_select) info_layout.addWidget(self.back_btn) # 返回按钮放最左 self.project_name_label = StrongBodyLabel() self.project_path_label = BodyLabel() self.ioc_file_label = BodyLabel() self.freertos_label = BodyLabel() info_layout.addWidget(self.project_name_label) info_layout.addWidget(self.project_path_label) info_layout.addWidget(self.ioc_file_label) info_layout.addWidget(self.freertos_label) info_layout.addStretch() main_layout.addLayout(info_layout) main_layout.addWidget(HorizontalSeparator()) # ======= 新增：左右分栏 ======= content_hbox = QHBoxLayout() content_hbox.setSpacing(24) # 左侧：文件树 left_vbox = QVBoxLayout() left_vbox.addWidget(SubtitleLabel("用户代码模块选择")) left_vbox.addWidget(HorizontalSeparator()) self.file_tree = TreeWidget() self.file_tree.setHeaderLabels(["模块名"]) self.file_tree.setSelectionMode(self.file_tree.ExtendedSelection) self.file_tree.header().setSectionResizeMode(0, QHeaderView.Stretch) self.file_tree.setCheckedColor("#0078d4", "#2d7d9a") self.file_tree.setBorderRadius(8) self.file_tree.setBorderVisible(True) left_vbox.addWidget(self.file_tree, stretch=1) content_hbox.addLayout(left_vbox, 2) # 右侧：操作按钮和说明 right_vbox = QVBoxLayout() right_vbox.setSpacing(18) right_vbox.addWidget(SubtitleLabel("操作区")) right_vbox.addWidget(HorizontalSeparator()) # 操作按钮分组 btn_group = QVBoxLayout() # 自动环境配置按钮 self.env_btn = PushButton("自动环境配置") self.env_btn.setFixedWidth(200) self.env_btn.setToolTip("自动检测并配置常用开发环境（功能开发中）") self.env_btn.clicked.connect(self.auto_env_config) btn_group.addWidget(self.env_btn) # FreeRTOS相关按钮 self.freertos_task_btn = PushButton("自动生成FreeRTOS任务") self.freertos_task_btn.setFixedWidth(200) self.freertos_task_btn.setToolTip("自动在 freertos.c 中插入任务创建代码") self.freertos_task_btn.clicked.connect(self.on_freertos_task_btn_clicked) btn_group.addWidget(self.freertos_task_btn) self.task_code_btn = PushButton("配置并生成任务代码") self.task_code_btn.setFixedWidth(200) self.task_code_btn.setToolTip("配置任务参数并一键生成任务代码文件") self.task_code_btn.clicked.connect(self.on_task_code_btn_clicked) btn_group.addWidget(self.task_code_btn) self.generate_btn = PushButton(FluentIcon.CODE, "生成代码") self.generate_btn.setFixedWidth(200) self.generate_btn.setToolTip("将选中的用户模块代码复制到工程 User 目录") self.generate_btn.clicked.connect(self.generate_code) btn_group.addWidget(self.generate_btn) btn_group.addSpacing(10) right_vbox.addLayout(btn_group) right_vbox.addStretch() content_hbox.addLayout(right_vbox, 1) main_layout.addLayout(content_hbox, stretch=1) self.stacked_layout.addWidget(self.config_widget) self.file_tree.itemChanged.connect(self.on_tree_item_changed) def auto_env_config(self): InfoBar.info( title="敬请期待", content="自动环境配置功能暂未实现，等待后续更新。", parent=self, duration=2000 ) def choose_project_folder(self): folder = QFileDialog.getExistingDirectory(self, "请选择代码项目文件夹") if not folder: return ioc_files = [f for f in os.listdir(folder) if f.endswith('.ioc')] if not ioc_files: InfoBar.warning( title="提示", content="未找到.ioc文件，请确认项目文件夹。", parent=self, duration=2000 ) return self.project_path = folder self.project_name = os.path.basename(folder) self.ioc_file = os.path.join(folder, ioc_files[0]) self.show_config_page() def show_config_page(self): # 更新项目信息 self.project_name_label.setText(f"项目名称: {self.project_name}") self.project_path_label.setText(f"项目路径: {self.project_path}") # self.ioc_file_label.setText(f"IOC 文件: {self.ioc_file}") try: ioc = IocConfig(self.ioc_file) self.freertos_enabled = ioc.is_freertos_enabled() # 记录状态 freertos_status = "已启用" if self.freertos_enabled else "未启用" self.freertos_label.setText(f"FreeRTOS: {freertos_status}") # self.freertos_task_btn.setEnabled(self.freertos_enabled) except Exception as e: self.freertos_label.setText(f"IOC解析失败: {e}") self.freertos_task_btn.hide() self.freertos_enabled = False self.show_user_code_files() self.stacked_layout.setCurrentWidget(self.config_widget) def on_freertos_task_btn_clicked(self): if not self.freertos_enabled: InfoBar.warning( title="未开启 FreeRTOS", content="请先在 CubeMX 中开启 FreeRTOS！", parent=self, duration=2000 ) return self.generate_freertos_task() def on_task_code_btn_clicked(self): if not self.freertos_enabled: InfoBar.warning( title="未开启 FreeRTOS", content="请先在 CubeMX 中开启 FreeRTOS！", parent=self, duration=2000 ) return self.open_task_config_dialog() def back_to_select(self): self.stacked_layout.setCurrentWidget(self.select_widget) def update_user_template(self): url = "http://gitea.qutrobot.top/robofish/MRobot/archive/User_code.zip" local_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../assets/User_code") try: resp = requests.get(url, timeout=30) resp.raise_for_status() z = zipfile.ZipFile(io.BytesIO(resp.content)) if os.path.exists(local_dir): shutil.rmtree(local_dir) for member in z.namelist(): rel_path = os.path.relpath(member, z.namelist()[0]) if rel_path == ".": continue target_path = os.path.join(local_dir, rel_path) if member.endswith('/'): os.makedirs(target_path, exist_ok=True) else: os.makedirs(os.path.dirname(target_path), exist_ok=True) with open(target_path, "wb") as f: f.write(z.read(member)) InfoBar.success( title="更新成功", content="用户模板已更新到最新版本！", parent=self, duration=2000 ) except Exception as e: InfoBar.error( title="更新失败", content=f"用户模板更新失败: {e}", parent=self, duration=3000 ) def show_user_code_files(self): self.file_tree.clear() base_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../assets/User_code") user_dir = os.path.join(self.project_path, "User") sub_dirs = ["bsp", "component", "device", "module"] # 读取所有 describe.csv 和 dependencies.csv describe_map = {} dependencies_map = {} for sub in sub_dirs: dir_path = os.path.join(base_dir, sub) if not os.path.isdir(dir_path): continue # describe desc_path = os.path.join(dir_path, "describe.csv") if os.path.exists(desc_path): with open(desc_path, encoding="utf-8") as f: for line in f: if "," in line: k, v = line.strip().split(",", 1) describe_map[f"{sub}/{k.strip()}"] = v.strip() # dependencies dep_path = os.path.join(dir_path, "dependencies.csv") if os.path.exists(dep_path): with open(dep_path, encoding="utf-8") as f: for line in f: if "," in line: a, b = line.strip().split(",", 1) dependencies_map.setdefault(f"{sub}/{a.strip()}", []).append(b.strip()) self._describe_map = describe_map self._dependencies_map = dependencies_map self.file_tree.setHeaderLabels(["模块名", "描述"]) self.file_tree.setSelectionMode(self.file_tree.ExtendedSelection) self.file_tree.header().setSectionResizeMode(0, QHeaderView.Interactive) self.file_tree.header().setSectionResizeMode(1, QHeaderView.Interactive) self.file_tree.setBorderRadius(8) self.file_tree.setBorderVisible(True) for sub in sub_dirs: dir_path = os.path.join(base_dir, sub) if not os.path.isdir(dir_path): continue group_item = TreeItem([sub, ""]) self.file_tree.addTopLevelItem(group_item) has_file = False for root, _, files in os.walk(dir_path): rel_root = os.path.relpath(root, base_dir) for f in sorted(files): if f.endswith(".c"): mod_name = os.path.splitext(f)[0] rel_c = os.path.join(rel_root, f) key = f"{rel_root}/{mod_name}".replace("\\", "/") desc = describe_map.get(key, "") file_item = TreeItem([mod_name, desc]) file_item.setFlags(file_item.flags() | Qt.ItemIsUserCheckable) file_item.setData(0, Qt.UserRole, rel_c) file_item.setData(0, Qt.UserRole + 1, key) # 存模块key file_item.setToolTip(1, desc) file_item.setTextAlignment(1, Qt.AlignLeft | Qt.AlignVCenter) group_item.addChild(file_item) dst_c = os.path.join(user_dir, rel_c) if os.path.exists(dst_c): file_item.setCheckState(0, Qt.Unchecked) file_item.setText(0, f"{mod_name}（已存在）") file_item.setForeground(0, Qt.gray) else: file_item.setCheckState(0, Qt.Unchecked) group_item.addChild(file_item) has_file = True if not has_file: empty_item = TreeItem(["（无 .c 文件）", ""]) group_item.addChild(empty_item) self.file_tree.expandAll() # 勾选依赖自动勾选 def on_tree_item_changed(self, item, column): if column != 0: return if item.childCount() > 0: return # 只处理叶子 if item.checkState(0) == Qt.Checked: key = item.data(0, Qt.UserRole + 1) deps = self._dependencies_map.get(key, []) if deps: checked = [] root = self.file_tree.invisibleRootItem() for i in range(root.childCount()): group = root.child(i) for j in range(group.childCount()): child = group.child(j) ckey = child.data(0, Qt.UserRole + 1) if ckey in deps and child.checkState(0) != Qt.Checked: child.setCheckState(0, Qt.Checked) checked.append(ckey) if checked: descs = [self._describe_map.get(dep, dep) for dep in checked] InfoBar.info( title="依赖自动勾选", content="已自动勾选依赖模块: " + "，".join(descs), parent=self, duration=2000 ) def get_checked_files(self): files = [] def _traverse(item): for i in range(item.childCount()): child = item.child(i) if child.childCount() == 0 and child.checkState(0) == Qt.Checked: files.append(child.data(0, Qt.UserRole)) _traverse(child) root = self.file_tree.invisibleRootItem() for i in range(root.childCount()): _traverse(root.child(i)) return files def generate_code(self): import shutil base_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../assets/User_code") user_dir = os.path.join(self.project_path, "User") copied = [] files = self.get_checked_files() skipped = [] for rel_c in files: rel_h = rel_c[:-2] + ".h" src_c = os.path.join(base_dir, rel_c) src_h = os.path.join(base_dir, rel_h) dst_c = os.path.join(user_dir, rel_c) dst_h = os.path.join(user_dir, rel_h) # 如果目标文件已存在则跳过 if os.path.exists(dst_c): skipped.append(dst_c) else: os.makedirs(os.path.dirname(dst_c), exist_ok=True) shutil.copy2(src_c, dst_c) copied.append(dst_c) if os.path.exists(src_h): if os.path.exists(dst_h): skipped.append(dst_h) else: os.makedirs(os.path.dirname(dst_h), exist_ok=True) shutil.copy2(src_h, dst_h) copied.append(dst_h) msg = f"已拷贝 {len(copied)} 个文件到 User 目录" if skipped: msg += f"\n{len(skipped)} 个文件已存在，未覆盖" InfoBar.success( title="生成完成", content=msg, parent=self, duration=2000 ) # 生成后刷新文件树，更新标记 self.show_user_code_files() def generate_freertos_task(self): import re freertos_path = os.path.join(self.project_path, "Core", "Src", "freertos.c") if not os.path.exists(freertos_path): InfoBar.error( title="未找到 freertos.c", content="未找到 Core/Src/freertos.c 文件，请确认工程路径。", parent=self, duration=2500 ) return with open(freertos_path, "r", encoding="utf-8") as f: code = f.read() changed = False error_msgs = [] # 1. 添加 #include "task/user_task.h" include_line = '#include "task/user_task.h"' if include_line not in code: # 只插入到 USER CODE BEGIN Includes 区域 include_pattern = r'(\/\* *USER CODE BEGIN Includes *\*\/\s*)' if re.search(include_pattern, code): code = re.sub( include_pattern, r'\1' + include_line + '\n', code ) changed = True else: error_msgs.append("未找到 /* USER CODE BEGIN Includes */ 区域，无法插入 include。") # 2. 在 /* USER CODE BEGIN RTOS_THREADS */ 区域添加 osThreadNew(Task_Init, NULL, &attr_init); rtos_threads_pattern = r'(\/\* *USER CODE BEGIN RTOS_THREADS *\*\/\s*)(.*?)(\/\* *USER CODE END RTOS_THREADS *\*\/)' match = re.search(rtos_threads_pattern, code, re.DOTALL) task_line = ' osThreadNew(Task_Init, NULL, &attr_init); // 创建初始化任务\n' if match: threads_code = match.group(2) if 'Task_Init' not in threads_code: # 保留原有内容，追加新行 new_threads_code = match.group(1) + threads_code + task_line + match.group(3) code = code[:match.start()] + new_threads_code + code[match.end():] changed = True else: error_msgs.append("未找到 /* USER CODE BEGIN RTOS_THREADS */ 区域，无法插入任务创建代码。") # 3. 清空 StartDefaultTask 的 USER CODE 区域，只保留 osThreadTerminate sdt_pattern = r'(\/\* *USER CODE BEGIN StartDefaultTask *\*\/\s*)(.*?)(\/\* *USER CODE END StartDefaultTask *\*\/)' match = re.search(sdt_pattern, code, re.DOTALL) if match: if 'osThreadTerminate(osThreadGetId());' not in match.group(2): new_sdt_code = match.group(1) + ' osThreadTerminate(osThreadGetId());\n' + match.group(3) code = code[:match.start()] + new_sdt_code + code[match.end():] changed = True else: error_msgs.append("未找到 /* USER CODE BEGIN StartDefaultTask */ 区域，无法插入终止代码。") if changed: with open(freertos_path, "w", encoding="utf-8") as f: f.write(code) InfoBar.success( title="生成成功", content="FreeRTOS任务代码已自动生成！", parent=self, duration=2000 ) elif error_msgs: InfoBar.error( title="生成失败", content="\n".join(error_msgs), parent=self, duration=3000 ) else: InfoBar.info( title="无需修改", content="FreeRTOS任务相关代码已存在，无需重复生成。", parent=self, duration=2000 ) def open_task_config_dialog(self): config_path = os.path.join(self.project_path, "User", "task", "config.yaml") dlg = TaskConfigDialog(self, config_path=config_path) if dlg.exec() == QDialog.Accepted: try: tasks = dlg.get_tasks() except Exception as e: InfoBar.error( title="参数错误", content=str(e), parent=self, duration=3000 ) return if not tasks: InfoBar.warning( title="未配置任务", content="请至少添加一个任务！", parent=self, duration=2000 ) return try: self.generate_task_code(tasks) InfoBar.success( title="生成成功", content="任务代码已生成到 User/task 目录！", parent=self, duration=2000 ) except Exception as e: InfoBar.error( title="生成失败", content=f"任务代码生成失败: {e}", parent=self, duration=3000 ) def generate_task_code(self, task_list): base_dir = os.path.dirname(os.path.abspath(__file__)) template_dir = os.path.join(base_dir, "../assets/User_code/task") output_dir = os.path.join(self.project_path, "User", "task") os.makedirs(output_dir, exist_ok=True) user_task_h_tpl = os.path.join(template_dir, "user_task.h.template") user_task_c_tpl = os.path.join(template_dir, "user_task.c.template") init_c_tpl = os.path.join(template_dir, "init.c.template") task_c_tpl = os.path.join(template_dir, "task.c.template") freq_tasks = [t for t in task_list if t.get("freq_control", True)] def render_template(path, context): with open(path, encoding="utf-8") as f: tpl = Template(f.read()) return tpl.render(**context) # ----------- 生成 user_task.h ----------- context_h = { "thread_definitions": "\n".join([f" osThreadId_t {t['name']};" for t in task_list]), "freq_definitions": "\n".join([f" float {t['name']};" for t in freq_tasks]), "stack_definitions": "\n".join([f" UBaseType_t {t['name']};" for t in task_list]), "last_up_time_definitions": "\n".join([f" float {t['name']};" for t in freq_tasks]), "task_frequency_definitions": "\n".join([f"#define {t['name'].upper()}_FREQ ({t['frequency']})" for t in freq_tasks]), "task_init_delay_definitions": "\n".join([f"#define {t['name'].upper()}_INIT_DELAY ({t['delay']})" for t in task_list]), "task_attr_declarations": "\n".join([f"extern const osThreadAttr_t attr_{t['name']};" for t in task_list]), "task_function_declarations": "\n".join([f"void {t['function']}(void *argument);" for t in task_list]), } user_task_h_path = os.path.join(output_dir, "user_task.h") new_user_task_h = render_template(user_task_h_tpl, context_h) save_with_preserve(user_task_h_path, new_user_task_h) # ----------- 生成 user_task.c ----------- context_c = { "task_attr_definitions": "\n".join([ f"const osThreadAttr_t attr_{t['name']} = {{\n" f" .name = \"{t['name']}\",\n" f" .priority = osPriorityNormal,\n" f" .stack_size = {t['stack']} * 4,\n" f"}};" for t in task_list ]) } user_task_c_path = os.path.join(output_dir, "user_task.c") user_task_c = render_template(user_task_c_tpl, context_c) save_with_preserve(user_task_c_path, user_task_c) # ----------- 生成 init.c ----------- thread_creation_code = "\n".join([ f" task_runtime.thread.{t['name']} = osThreadNew({t['function']}, NULL, &attr_{t['name']});" for t in task_list ]) context_init = { "thread_creation_code": thread_creation_code, } init_c_path = os.path.join(output_dir, "init.c") init_c = render_template(init_c_tpl, context_init) save_with_preserve(init_c_path, init_c) # ----------- 生成 task.c ----------- for t in task_list: desc = t.get("description", "") desc_wrapped = "\n ".join(textwrap.wrap(desc, 20)) context_task = { "task_name": t["name"], "task_function": t["function"], "task_frequency": f"{t['name'].upper()}_FREQ" if t.get("freq_control", True) else None, "task_delay": f"{t['name'].upper()}_INIT_DELAY", "task_description": desc_wrapped, "freq_control": t.get("freq_control", True) } with open(task_c_tpl, encoding="utf-8") as f: tpl = Template(f.read()) code = tpl.render(**context_task) task_c_path = os.path.join(output_dir, f"{t['name']}.c") save_with_preserve(task_c_path, code) # ----------- 保存任务配置到 config.yaml ----------- config_yaml_path = os.path.join(output_dir, "config.yaml") with open(config_yaml_path, "w", encoding="utf-8") as f: yaml.safe_dump(task_list, f, allow_unicode=True) class TaskConfigDialog(QDialog): def __init__(self, parent=None, config_path=None): super().__init__(parent) self.setWindowTitle("任务配置") self.resize(900, 520) # 设置背景色跟随主题 if theme() == Theme.DARK: self.setStyleSheet("background-color: #232323;") else: self.setStyleSheet("background-color: #faf9f8;") # 主布局 main_layout = QVBoxLayout(self) main_layout.setContentsMargins(16, 16, 16, 16) main_layout.setSpacing(12) # 顶部横向分栏 self.top_layout = QHBoxLayout() self.top_layout.setSpacing(16) # ----------- 左侧任务按钮区 ----------- self.left_widget = QWidget() self.left_layout = QVBoxLayout(self.left_widget) self.left_layout.setContentsMargins(0, 0, 0, 0) self.left_layout.setSpacing(8) self.add_btn = PrimaryPushButton("添加任务") self.add_btn.clicked.connect(self.add_task) self.left_layout.addWidget(self.add_btn) self.task_btn_area = QScrollArea() self.task_btn_area.setWidgetResizable(True) self.task_btn_area.setFrameShape(QScrollArea.NoFrame) self.task_btn_container = QWidget() self.task_btn_layout = QVBoxLayout(self.task_btn_container) self.task_btn_layout.setContentsMargins(0, 0, 0, 0) self.task_btn_layout.setSpacing(4) self.task_btn_layout.addStretch() self.task_btn_area.setWidget(self.task_btn_container) self.left_layout.addWidget(self.task_btn_area, stretch=1) self.left_widget.setFixedWidth(180) self.top_layout.addWidget(self.left_widget, stretch=0) # ----------- 左侧任务按钮区 END ----------- main_layout.addLayout(self.top_layout, stretch=1) # 下方按钮区 btn_layout = QHBoxLayout() btn_layout.addStretch() self.ok_btn = PrimaryPushButton("生成") self.cancel_btn = PushButton("取消") btn_layout.addWidget(self.ok_btn) btn_layout.addWidget(self.cancel_btn) main_layout.addLayout(btn_layout) self.ok_btn.clicked.connect(self.accept) self.cancel_btn.clicked.connect(self.reject) self.tasks = [] self.current_index = -1 if config_path and os.path.exists(config_path): try: with open(config_path, "r", encoding="utf-8") as f: tasks = yaml.safe_load(f) if tasks: for t in tasks: self.tasks.append(self._make_task_obj(t)) except Exception: pass if not self.tasks: self.tasks.append(self._make_task_obj()) self.current_index = 0 self.refresh_task_btns() self.show_task_form(self.tasks[0]) def refresh_task_btns(self): # 清空旧按钮 while self.task_btn_layout.count(): item = self.task_btn_layout.takeAt(0) w = item.widget() if w: w.deleteLater() # 重新添加按钮 for idx, t in enumerate(self.tasks): btn = PushButton(t["name"]) btn.setCheckable(True) btn.setChecked(idx == self.current_index) btn.clicked.connect(lambda checked, i=idx: self.select_task(i)) self.task_btn_layout.addWidget(btn) # 删除按钮 del_btn = PushButton("删除") del_btn.setFixedWidth(48) del_btn.clicked.connect(lambda _, i=idx: self.delete_task(i)) hbox = QHBoxLayout() hbox.addWidget(btn) hbox.addWidget(del_btn) hbox.setContentsMargins(0, 0, 0, 0) hbox.setSpacing(4) container = QWidget() container.setLayout(hbox) self.task_btn_layout.addWidget(container) self.task_btn_layout.addStretch() def add_task(self): self.save_form() new_idx = len(self.tasks) self.tasks.append(self._make_task_obj({"name": f"Task{new_idx+1}"})) self.current_index = new_idx self.refresh_task_btns() self.show_task_form(self.tasks[self.current_index]) def delete_task(self, idx): if len(self.tasks) <= 1: InfoBar.warning( title="至少保留一个任务", content="至少需要保留一个任务！", parent=self, duration=2000 ) return del self.tasks[idx] if self.current_index >= len(self.tasks): self.current_index = len(self.tasks) - 1 self.refresh_task_btns() self.show_task_form(self.tasks[self.current_index]) def select_task(self, idx): self.save_form() self.current_index = idx self.refresh_task_btns() self.show_task_form(self.tasks[idx]) def show_task_form(self, task): # 先移除旧的 form_widget if hasattr(self, "form_widget") and self.form_widget is not None: self.top_layout.removeWidget(self.form_widget) self.form_widget.deleteLater() self.form_widget = None # 新建 form_widget 和 form_layout self.form_widget = QWidget() self.form_layout = QVBoxLayout(self.form_widget) self.form_layout.setContentsMargins(0, 0, 0, 0) self.form_layout.setSpacing(12) # 添加到右侧 self.top_layout.addWidget(self.form_widget, stretch=1) if not task: label = BodyLabel("未找到任务。") label.setAlignment(Qt.AlignCenter) self.form_layout.addWidget(label) self.form_layout.addStretch() return # 任务名称 row1 = QHBoxLayout() label_name = BodyLabel("任务名称") self.name_edit = LineEdit() self.name_edit.setText(task["name"]) self.name_edit.setPlaceholderText("任务名称") row1.addWidget(label_name) row1.addWidget(self.name_edit) self.form_layout.addLayout(row1) # 频率 row2 = QHBoxLayout() label_freq = BodyLabel("频率") self.freq_spin = SpinBox() self.freq_spin.setRange(1, 10000) self.freq_spin.setSuffix(" Hz") self.freq_spin.setValue(task.get("frequency", 500)) row2.addWidget(label_freq) row2.addWidget(self.freq_spin) self.form_layout.addLayout(row2) # 延迟 row3 = QHBoxLayout() label_delay = BodyLabel("延迟") self.delay_spin = SpinBox() self.delay_spin.setRange(0, 10000) self.delay_spin.setSuffix(" ms") self.delay_spin.setValue(task.get("delay", 0)) row3.addWidget(label_delay) row3.addWidget(self.delay_spin) self.form_layout.addLayout(row3) # 堆栈 row4 = QHBoxLayout() label_stack = BodyLabel("堆栈") self.stack_spin = SpinBox() self.stack_spin.setRange(128, 8192) self.stack_spin.setSingleStep(128) self.stack_spin.setValue(task.get("stack", 256)) row4.addWidget(label_stack) row4.addWidget(self.stack_spin) self.form_layout.addLayout(row4) # 频率控制 row5 = QHBoxLayout() self.freq_ctrl = CheckBox("启用频率控制") self.freq_ctrl.setChecked(task.get("freq_control", True)) row5.addWidget(self.freq_ctrl) self.form_layout.addLayout(row5) # 描述 label_desc = BodyLabel("任务描述") self.desc_edit = TextEdit() self.desc_edit.setText(task.get("description", "")) self.desc_edit.setPlaceholderText("任务描述") self.form_layout.addWidget(label_desc) self.form_layout.addWidget(self.desc_edit) self.form_layout.addStretch() def _make_task_obj(self, task=None): return { "name": task["name"] if task else f"Task1", "frequency": task.get("frequency", 500) if task else 500, "delay": task.get("delay", 0) if task else 0, "stack": task.get("stack", 256) if task else 256, "description": task.get("description", "") if task else "", "freq_control": task.get("freq_control", True) if task else True, } def save_form(self): if self.current_index < 0 or self.current_index >= len(self.tasks): return t = self.tasks[self.current_index] t["name"] = self.name_edit.text().strip() t["frequency"] = self.freq_spin.value() t["delay"] = self.delay_spin.value() t["stack"] = self.stack_spin.value() t["description"] = self.desc_edit.toPlainText().strip() t["freq_control"] = self.freq_ctrl.isChecked() def get_tasks(self): self.save_form() tasks = [] for idx, t in enumerate(self.tasks): name = t["name"].strip() freq = t["frequency"] delay = t["delay"] stack = t["stack"] desc = t["description"].strip() freq_ctrl = t["freq_control"] if stack < 128 or (stack & (stack - 1)) != 0 or stack % 128 != 0: raise ValueError(f"第{idx+1}个任务“{name}”的堆栈大小必须为128、256、512、1024等（128*2^n）") task = { "name": name, "function": f"Task_{name}", "delay": delay, "stack": stack, "description": desc, "freq_control": freq_ctrl } if freq_ctrl: task["frequency"] = freq tasks.append(task) return tasks