"""Excel结果导出器""" from pathlib import Path from typing import List, Dict, Any from datetime import datetime from openpyxl import Workbook from openpyxl.styles import Font, Alignment, PatternFill, Border, Side from openpyxl.utils import get_column_letter from measurements.base import MeasurementResult class ExcelExporter: """将测量结果导出为Excel文件""" def __init__(self, output_dir: str): """ 初始化导出器 Args: output_dir: 输出目录路径 """ self.output_dir = Path(output_dir) self.output_dir.mkdir(parents=True, exist_ok=True) def export(self, results: List[MeasurementResult], filename: str = None, task_configs: List[Dict[str, Any]] = None) -> str: """ 导出测量结果到Excel Args: results: 测量结果列表 filename: 输出文件名（不含扩展名），默认使用时间戳 task_configs: 任务配置列表（用于提取描述和参数） Returns: str: 导出的Excel文件路径 """ if filename is None: filename = f"measurement_results_{datetime.now().strftime('%Y%m%d_%H%M%S')}" output_path = self.output_dir / f"{filename}.xlsx" # 创建任务名称到配置的映射 task_config_map = {} if task_configs: for config in task_configs: task_config_map[config.get('name')] = config # 创建工作簿 wb = Workbook() ws = wb.active ws.title = "测量结果" # 写入标题 self._write_header(ws) # 写入数据 row = 2 for result in results: task_config = task_config_map.get(result.task_name, {}) row = self._write_result(ws, result, row, task_config) # 调整列宽 self._adjust_column_widths(ws) # 保存文件 wb.save(output_path) return str(output_path) def _write_header(self, ws): """写入表头""" headers = [ "任务名称", "任务类型", "任务描述", "点位参数", "状态", "平均距离 (mm)", "标准差 (mm)", "最小距离 (mm)", "最大距离 (mm)", "采样点数", "源ROI点数", "目标ROI点数", "源平面内点率", "目标平面内点率", "错误信息" ] # 样式 header_font = Font(bold=True, color="FFFFFF") header_fill = PatternFill(start_color="4472C4", end_color="4472C4", fill_type="solid") header_alignment = Alignment(horizontal="center", vertical="center") border = Border( left=Side(style='thin'), right=Side(style='thin'), top=Side(style='thin'), bottom=Side(style='thin') ) for col, header in enumerate(headers, 1): cell = ws.cell(row=1, column=col) cell.value = header cell.font = header_font cell.fill = header_fill cell.alignment = header_alignment cell.border = border def _format_task_params(self, task_config: Dict[str, Any]) -> str: """ 格式化任务参数为字符串 Args: task_config: 任务配置字典 Returns: str: 格式化后的参数字符串 """ task_type = task_config.get('type', '') params_parts = [] if task_type == 'plane_to_plane': # 平面到平面：显示源ROI和目标ROI source_roi = task_config.get('source_roi', {}) target_roi = task_config.get('target_roi', {}) if source_roi: params_parts.append(f"源ROI: min{source_roi.get('bbox_min', [])} max{source_roi.get('bbox_max', [])}") if target_roi: params_parts.append(f"目标ROI: min{target_roi.get('bbox_min', [])} max{target_roi.get('bbox_max', [])}") elif task_type == 'point_to_plane': mode = task_config.get('mode', '') if mode == 'exact_points': # 精确点模式：显示目标点坐标 target_points = task_config.get('target_points', []) if target_points: params_parts.append(f"目标点({len(target_points)}个): {target_points}") plane = task_config.get('plane', {}) if plane: params_parts.append(f"平面: {plane.get('a', 0)}x+{plane.get('b', 0)}y+{plane.get('c', 0)}z+{plane.get('d', 0)}=0") elif mode == 'bbox_filtered': # 包围盒过滤模式：显示包围盒 bboxes = task_config.get('bboxes', []) if bboxes: bbox_strs = [f"{bbox.get('name', '')}: min{bbox.get('min', [])} max{bbox.get('max', [])}" for bbox in bboxes] params_parts.append(f"包围盒: {'; '.join(bbox_strs)}") filter_info = task_config.get('filter', {}) if filter_info: params_parts.append(f"过滤: {filter_info.get('type', '')} axis={filter_info.get('axis', '')} top_n={filter_info.get('top_n', '')}") elif task_type == 'point_to_point': # 点到点：显示点对 point_pairs = task_config.get('point_pairs', []) if point_pairs: pair_strs = [f"{pair.get('name', '')}: {pair.get('point1', [])} <-> {pair.get('point2', [])}" for pair in point_pairs] params_parts.append(f"点对({len(point_pairs)}对): {'; '.join(pair_strs)}") return '\n'.join(params_parts) if params_parts else "无" def _write_result(self, ws, result: MeasurementResult, row: int, task_config: Dict[str, Any]) -> int: """ 写入单个测量结果 Args: ws: 工作表 result: 测量结果 row: 起始行号 task_config: 任务配置 Returns: int: 下一行号 """ border = Border( left=Side(style='thin'), right=Side(style='thin'), top=Side(style='thin'), bottom=Side(style='thin') ) # 基本信息 ws.cell(row=row, column=1).value = result.task_name ws.cell(row=row, column=2).value = result.task_type ws.cell(row=row, column=3).value = task_config.get('description', '') ws.cell(row=row, column=4).value = self._format_task_params(task_config) ws.cell(row=row, column=5).value = "成功" if result.success else "失败" # 状态颜色 status_cell = ws.cell(row=row, column=5) if result.success: status_cell.fill = PatternFill(start_color="C6EFCE", end_color="C6EFCE", fill_type="solid") status_cell.font = Font(color="006100") else: status_cell.fill = PatternFill(start_color="FFC7CE", end_color="FFC7CE", fill_type="solid") status_cell.font = Font(color="9C0006") # 测量数据 if result.success and result.data: data = result.data ws.cell(row=row, column=6).value = round(data.get('mean_distance', 0), 2) ws.cell(row=row, column=7).value = round(data.get('std_distance', 0), 2) ws.cell(row=row, column=8).value = round(data.get('min_distance', 0), 2) ws.cell(row=row, column=9).value = round(data.get('max_distance', 0), 2) ws.cell(row=row, column=10).value = data.get('sample_count', 0) ws.cell(row=row, column=11).value = data.get('source_roi_points', 0) ws.cell(row=row, column=12).value = data.get('target_roi_points', 0) ws.cell(row=row, column=13).value = f"{data.get('source_inlier_ratio', 0) * 100:.1f}%" ws.cell(row=row, column=14).value = f"{data.get('target_inlier_ratio', 0) * 100:.1f}%" else: # 失败时显示错误信息 ws.cell(row=row, column=15).value = result.error_message # 应用边框和对齐 for col in range(1, 16): cell = ws.cell(row=row, column=col) cell.border = border # 点位参数列使用左对齐，其他居中 if col == 4: cell.alignment = Alignment(horizontal="left", vertical="center", wrap_text=True) else: cell.alignment = Alignment(horizontal="center", vertical="center") return row + 1 def _adjust_column_widths(self, ws): """自动调整列宽""" column_widths = { 1: 25, # 任务名称 2: 18, # 任务类型 3: 30, # 任务描述 4: 60, # 点位参数 5: 10, # 状态 6: 15, # 平均距离 7: 15, # 标准差 8: 15, # 最小距离 9: 15, # 最大距离 10: 12, # 采样点数 11: 12, # 源ROI点数 12: 12, # 目标ROI点数 13: 15, # 源平面内点率 14: 15, # 目标平面内点率 15: 40 # 错误信息 } for col, width in column_widths.items(): ws.column_dimensions[get_column_letter(col)].width = width