mirror of
https://github.com/goldenfishs/MRobot.git
synced 2025-09-14 12:54:33 +08:00
396 lines
17 KiB
Python
396 lines
17 KiB
Python
from PyQt5.QtWidgets import QWidget, QVBoxLayout, QLineEdit, QCheckBox, QComboBox, QTableWidget, QHeaderView, QMessageBox, QHBoxLayout, QTextEdit
|
||
from qfluentwidgets import TitleLabel, BodyLabel, PushButton, CheckBox, TableWidget, LineEdit, ComboBox, MessageBox, SubtitleLabel, FluentIcon, TextEdit
|
||
from qfluentwidgets import InfoBar
|
||
from PyQt5.QtCore import Qt, pyqtSignal
|
||
from app.tools.code_generator import CodeGenerator
|
||
import os
|
||
import csv
|
||
import shutil
|
||
import re
|
||
|
||
def preserve_all_user_regions(new_code, old_code):
|
||
""" Preserves all user-defined regions in the new code based on the old code.
|
||
This function uses regex to find user-defined regions in the old code and replaces them in the new code.
|
||
Args:
|
||
new_code (str): The new code content.
|
||
old_code (str): The old code content.
|
||
Returns:
|
||
str: The new code with preserved user-defined regions.
|
||
"""
|
||
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)
|
||
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||
with open(path, "w", encoding="utf-8") as f:
|
||
f.write(new_code)
|
||
|
||
def load_descriptions(csv_path):
|
||
"""加载组件描述信息"""
|
||
descriptions = {}
|
||
if os.path.exists(csv_path):
|
||
with open(csv_path, encoding='utf-8') as f:
|
||
reader = csv.reader(f)
|
||
for row in reader:
|
||
if len(row) >= 2:
|
||
key, desc = row[0].strip(), row[1].strip()
|
||
descriptions[key.lower()] = desc
|
||
return descriptions
|
||
|
||
def load_dependencies(csv_path):
|
||
"""加载组件依赖关系"""
|
||
dependencies = {}
|
||
if os.path.exists(csv_path):
|
||
with open(csv_path, encoding='utf-8') as f:
|
||
reader = csv.reader(f)
|
||
for row in reader:
|
||
if len(row) >= 2:
|
||
component = row[0].strip()
|
||
deps = [dep.strip() for dep in row[1:] if dep.strip()]
|
||
dependencies[component] = deps
|
||
return dependencies
|
||
|
||
|
||
def get_component_page(component_name, project_path, component_manager=None):
|
||
"""根据组件名返回对应的页面类，没有特殊类则返回默认ComponentSimple"""
|
||
name_lower = component_name.lower()
|
||
special_classes = {
|
||
"pid": component_pid,
|
||
"filter": component_filter,
|
||
# 以后可以继续添加特殊组件
|
||
}
|
||
if name_lower in special_classes:
|
||
return special_classes[name_lower](project_path, component_manager)
|
||
else:
|
||
template_names = {
|
||
'header': f'{name_lower}.h',
|
||
'source': f'{name_lower}.c'
|
||
}
|
||
return ComponentSimple(project_path, component_name, template_names, component_manager)
|
||
|
||
|
||
|
||
def get_all_dependency_components(dependencies):
|
||
"""获取所有被依赖的组件列表"""
|
||
dependent_components = set()
|
||
for component, deps in dependencies.items():
|
||
for dep_path in deps:
|
||
dep_name = os.path.basename(dep_path)
|
||
dependent_components.add(dep_name.lower())
|
||
return dependent_components
|
||
|
||
|
||
class ComponentSimple(QWidget):
|
||
"""简单组件界面 - 只有开启/关闭功能"""
|
||
def __init__(self, project_path, component_name, template_names, component_manager=None):
|
||
super().__init__()
|
||
self.project_path = project_path
|
||
self.component_name = component_name
|
||
self.template_names = template_names
|
||
self.component_manager = component_manager
|
||
|
||
# 加载描述和依赖信息
|
||
component_dir = os.path.dirname(__file__) + "/../../assets/User_code/component"
|
||
describe_path = os.path.join(component_dir, "describe.csv")
|
||
dependencies_path = os.path.join(component_dir, "dependencies.csv")
|
||
self.descriptions = load_descriptions(describe_path)
|
||
self.dependencies = load_dependencies(dependencies_path)
|
||
|
||
self._init_ui()
|
||
self._load_config()
|
||
|
||
def _init_ui(self):
|
||
layout = QVBoxLayout(self)
|
||
top_layout = QHBoxLayout()
|
||
top_layout.setAlignment(Qt.AlignVCenter)
|
||
self.generate_checkbox = CheckBox(f"启用 {self.component_name}")
|
||
self.generate_checkbox.stateChanged.connect(self._on_checkbox_changed)
|
||
top_layout.addWidget(self.generate_checkbox, alignment=Qt.AlignLeft)
|
||
top_layout.addStretch()
|
||
title = SubtitleLabel(f"{self.component_name} 配置 ")
|
||
title.setAlignment(Qt.AlignHCenter)
|
||
top_layout.addWidget(title, alignment=Qt.AlignHCenter)
|
||
top_layout.addStretch()
|
||
layout.addLayout(top_layout)
|
||
|
||
desc = self.descriptions.get(self.component_name.lower(), "")
|
||
if desc:
|
||
desc_label = BodyLabel(f"功能说明：{desc}")
|
||
desc_label.setWordWrap(True)
|
||
layout.addWidget(desc_label)
|
||
|
||
deps = self.dependencies.get(self.component_name.lower(), [])
|
||
if deps:
|
||
deps_text = f"依赖组件：{', '.join([os.path.basename(dep) for dep in deps])}"
|
||
deps_label = BodyLabel(deps_text)
|
||
deps_label.setWordWrap(True)
|
||
deps_label.setStyleSheet("color: #888888;")
|
||
layout.addWidget(deps_label)
|
||
# 不再自动启用依赖，只做提示
|
||
|
||
layout.addStretch()
|
||
|
||
def _on_checkbox_changed(self, state):
|
||
pass # 不再自动启用依赖
|
||
|
||
def is_need_generate(self):
|
||
return self.generate_checkbox.isChecked()
|
||
|
||
def get_enabled_dependencies(self):
|
||
if not self.is_need_generate():
|
||
return []
|
||
return self.dependencies.get(self.component_name.lower(), [])
|
||
|
||
def _generate_component_code_internal(self):
|
||
if not self.is_need_generate():
|
||
return False
|
||
template_dir = self._get_component_template_dir()
|
||
for key, filename in self.template_names.items():
|
||
template_path = os.path.join(template_dir, filename)
|
||
template_content = CodeGenerator.load_template(template_path)
|
||
if not template_content:
|
||
print(f"模板文件不存在或为空: {template_path}")
|
||
continue
|
||
output_path = os.path.join(self.project_path, f"User/component/{filename}")
|
||
save_with_preserve(output_path, template_content)
|
||
self._save_config()
|
||
return True
|
||
|
||
def _get_component_template_dir(self):
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
while os.path.basename(current_dir) != 'MRobot' and current_dir != '/':
|
||
current_dir = os.path.dirname(current_dir)
|
||
if os.path.basename(current_dir) == 'MRobot':
|
||
return os.path.join(current_dir, "assets/User_code/component")
|
||
else:
|
||
return os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "assets/User_code/component")
|
||
|
||
def _save_config(self):
|
||
config_path = os.path.join(self.project_path, "User/component/component_config.yaml")
|
||
config_data = CodeGenerator.load_config(config_path)
|
||
config_data[self.component_name.lower()] = {
|
||
'enabled': self.is_need_generate(),
|
||
'dependencies': self.dependencies.get(self.component_name.lower(), [])
|
||
}
|
||
CodeGenerator.save_config(config_data, config_path)
|
||
|
||
def _load_config(self):
|
||
config_path = os.path.join(self.project_path, "User/component/component_config.yaml")
|
||
config_data = CodeGenerator.load_config(config_path)
|
||
conf = config_data.get(self.component_name.lower(), {})
|
||
if conf.get('enabled', False):
|
||
self.generate_checkbox.setChecked(True)
|
||
|
||
class ComponentManager:
|
||
"""组件依赖管理器"""
|
||
|
||
def __init__(self):
|
||
self.component_pages = {} # 组件名 -> 页面对象
|
||
|
||
def register_component(self, component_name, page):
|
||
"""注册组件页面"""
|
||
self.component_pages[component_name.lower()] = page
|
||
|
||
def _sync_dependency_states(self):
|
||
"""同步所有依赖状态"""
|
||
# 重新计算所有依赖计数
|
||
new_dependency_count = {}
|
||
|
||
for page_name, page in self.component_pages.items():
|
||
if page.is_need_generate():
|
||
deps = page.get_enabled_dependencies()
|
||
for dep_path in deps:
|
||
dep_name = os.path.basename(dep_path).lower()
|
||
new_dependency_count[dep_name] = new_dependency_count.get(dep_name, 0) + 1
|
||
|
||
# 更新依赖计数
|
||
self.dependency_count = new_dependency_count
|
||
|
||
# 更新所有页面的状态
|
||
for page_name, page in self.component_pages.items():
|
||
if page.is_dependency:
|
||
count = self.dependency_count.get(page_name, 0)
|
||
page.set_dependency_count(count)
|
||
|
||
def enable_dependencies(self, component_name, deps):
|
||
"""启用依赖项"""
|
||
for dep_path in deps:
|
||
dep_name = os.path.basename(dep_path).lower()
|
||
|
||
# 增加依赖计数
|
||
self.dependency_count[dep_name] = self.dependency_count.get(dep_name, 0) + 1
|
||
|
||
# 更新被依赖的组件状态
|
||
if dep_name in self.component_pages:
|
||
page = self.component_pages[dep_name]
|
||
page.set_dependency_count(self.dependency_count[dep_name])
|
||
|
||
def disable_dependencies(self, component_name, deps):
|
||
"""禁用依赖项"""
|
||
for dep_path in deps:
|
||
dep_name = os.path.basename(dep_path).lower()
|
||
|
||
# 减少依赖计数
|
||
if dep_name in self.dependency_count:
|
||
self.dependency_count[dep_name] = max(0, self.dependency_count[dep_name] - 1)
|
||
|
||
# 更新被依赖的组件状态
|
||
if dep_name in self.component_pages:
|
||
page = self.component_pages[dep_name]
|
||
page.set_dependency_count(self.dependency_count[dep_name])
|
||
|
||
# 具体组件类
|
||
class component_pid(ComponentSimple):
|
||
def __init__(self, project_path, component_manager=None):
|
||
super().__init__(
|
||
project_path,
|
||
"PID",
|
||
{'header': 'pid.h', 'source': 'pid.c'},
|
||
component_manager
|
||
)
|
||
|
||
class component_filter(ComponentSimple):
|
||
def __init__(self, project_path, component_manager=None):
|
||
super().__init__(
|
||
project_path,
|
||
"Filter",
|
||
{'header': 'filter.h', 'source': 'filter.c'},
|
||
component_manager
|
||
)
|
||
|
||
# ...existing code... (component 类的 generate_component 方法保持不变)
|
||
|
||
class component(QWidget):
|
||
"""组件管理器"""
|
||
|
||
def __init__(self, project_path):
|
||
super().__init__()
|
||
self.project_path = project_path
|
||
|
||
@staticmethod
|
||
def generate_component(project_path, pages):
|
||
"""生成所有组件代码，处理依赖关系"""
|
||
# 自动添加 component.h
|
||
src_component_h = os.path.join(os.path.dirname(__file__), "../../assets/User_code/component/component.h")
|
||
dst_component_h = os.path.join(project_path, "User/component/component.h")
|
||
os.makedirs(os.path.dirname(dst_component_h), exist_ok=True)
|
||
if os.path.exists(src_component_h):
|
||
shutil.copyfile(src_component_h, dst_component_h)
|
||
|
||
# 收集所有需要生成的组件和它们的依赖
|
||
components_to_generate = set()
|
||
component_pages = {}
|
||
|
||
for page in pages:
|
||
# 检查是否是组件页面（通过类名或者属性判断）
|
||
if hasattr(page, "component_name") and hasattr(page, "is_need_generate"):
|
||
if page.is_need_generate():
|
||
component_name = page.component_name.lower()
|
||
components_to_generate.add(component_name)
|
||
component_pages[component_name] = page
|
||
|
||
# 添加依赖组件，依赖格式是路径形式如 "component/filter"
|
||
deps = page.get_enabled_dependencies()
|
||
for dep_path in deps:
|
||
# 从路径中提取组件名，如 "component/filter" -> "filter"
|
||
dep_name = os.path.basename(dep_path)
|
||
components_to_generate.add(dep_name)
|
||
|
||
# 为没有对应页面但需要生成的依赖组件创建临时页面
|
||
user_code_dir = os.path.join(os.path.dirname(__file__), "../../assets/User_code")
|
||
for comp_name in components_to_generate:
|
||
if comp_name not in component_pages:
|
||
# 创建临时组件页面
|
||
template_names = {'header': f'{comp_name}.h', 'source': f'{comp_name}.c'}
|
||
temp_page = ComponentSimple(project_path, comp_name.upper(), template_names)
|
||
# temp_page.set_forced_enabled(True) # 自动启用依赖组件
|
||
component_pages[comp_name] = temp_page
|
||
|
||
# 如果没有组件需要生成，返回提示信息
|
||
if not components_to_generate:
|
||
return "没有启用的组件需要生成代码。"
|
||
|
||
# 生成代码和依赖文件
|
||
success_count = 0
|
||
fail_count = 0
|
||
fail_list = []
|
||
|
||
# 处理依赖文件的复制
|
||
all_deps = set()
|
||
for page in pages:
|
||
if hasattr(page, "component_name") and hasattr(page, "is_need_generate"):
|
||
if page.is_need_generate():
|
||
deps = page.get_enabled_dependencies()
|
||
all_deps.update(deps)
|
||
|
||
# 复制依赖文件
|
||
for dep_path in all_deps:
|
||
try:
|
||
# dep_path 格式如 "component/filter"
|
||
src_dir = os.path.join(user_code_dir, dep_path)
|
||
if os.path.isdir(src_dir):
|
||
# 如果是目录，复制整个目录
|
||
dst_dir = os.path.join(project_path, "User", dep_path)
|
||
os.makedirs(os.path.dirname(dst_dir), exist_ok=True)
|
||
if os.path.exists(dst_dir):
|
||
shutil.rmtree(dst_dir)
|
||
shutil.copytree(src_dir, dst_dir)
|
||
else:
|
||
# 如果是文件，复制单个文件
|
||
src_file = src_dir
|
||
dst_file = os.path.join(project_path, "User", dep_path)
|
||
os.makedirs(os.path.dirname(dst_file), exist_ok=True)
|
||
if os.path.exists(src_file):
|
||
shutil.copyfile(src_file, dst_file)
|
||
success_count += 1
|
||
print(f"成功复制依赖: {dep_path}")
|
||
except Exception as e:
|
||
fail_count += 1
|
||
fail_list.append(f"{dep_path} (依赖复制异常: {e})")
|
||
print(f"复制依赖失败: {dep_path}, 错误: {e}")
|
||
|
||
# 生成组件代码
|
||
for comp_name in components_to_generate:
|
||
if comp_name in component_pages:
|
||
page = component_pages[comp_name]
|
||
try:
|
||
# 确保调用正确的方法名
|
||
if hasattr(page, '_generate_component_code_internal'):
|
||
result = page._generate_component_code_internal()
|
||
if result:
|
||
success_count += 1
|
||
print(f"成功生成组件: {comp_name}")
|
||
else:
|
||
fail_count += 1
|
||
fail_list.append(f"{comp_name} (生成失败)")
|
||
print(f"生成组件失败: {comp_name}")
|
||
else:
|
||
fail_count += 1
|
||
fail_list.append(f"{comp_name} (缺少生成方法)")
|
||
print(f"组件页面缺少生成方法: {comp_name}")
|
||
except Exception as e:
|
||
fail_count += 1
|
||
fail_list.append(f"{comp_name} (生成异常: {e})")
|
||
print(f"生成组件异常: {comp_name}, 错误: {e}")
|
||
|
||
total_items = len(all_deps) + len(components_to_generate)
|
||
msg = f"组件代码生成完成：总共尝试生成 {total_items} 项，成功 {success_count} 项，失败 {fail_count} 项。"
|
||
if fail_list:
|
||
msg += "\n失败项：\n" + "\n".join(fail_list)
|
||
|
||
return msg |