#!/usr/bin/env python3 """ 深度检查提示生成器 根据模块名自动生成 Claude Phase 2 深度检查所需的文件列表和检查提示。 用法： python3 deep_check_prompt.py ODFHYS # 生成检查文件列表 python3 deep_check_prompt.py ODFHYS --prompt # 生成完整检查提示 """ import os import re import sys import argparse from typing import List, Dict, Optional # 路径配置 EXTRACTED_DIR = "/home/fmq/program/tlusty/tl208-s54/rust/tlusty/extracted" RUST_BASE_DIR = "/home/fmq/.zeroclaw/workspace/SpectraRust/src" FORTRAN_COMMON_DIR = "/home/fmq/program/tlusty/tl208-s54/tlusty" # 导入 common_db script_dir = os.path.dirname(os.path.abspath(__file__)) if script_dir not in sys.path: sys.path.insert(0, script_dir) from common_db import ( get_includes_for_module, get_commons_for_module, get_vars_for_module, get_rust_structs_for_module, get_mapping, get_structs, get_blocks, ) def find_rust_file(module_name: str) -> Optional[str]: """查找模块的 Rust 文件路径""" rust_name = module_name.lower() math_subdirs = [ 'ali', 'atomic', 'continuum', 'convection', 'eos', 'hydrogen', 'interpolation', 'io', 'odf', 'opacity', 'partition', 'population', 'radiative', 'rates', 'solvers', 'special', 'temperature', 'utils' ] # tlusty/io/ path = os.path.join(RUST_BASE_DIR, 'tlusty', 'io', f"{rust_name}.rs") if os.path.exists(path): return path # tlusty/math/ path = os.path.join(RUST_BASE_DIR, 'tlusty', 'math', f"{rust_name}.rs") if os.path.exists(path): return path # tlusty/math/子目录 for subdir in math_subdirs: path = os.path.join(RUST_BASE_DIR, 'tlusty', 'math', subdir, f"{rust_name}.rs") if os.path.exists(path): return path # tlusty/state/ path = os.path.join(RUST_BASE_DIR, 'tlusty', 'state', f"{rust_name}.rs") if os.path.exists(path): return path return None def find_rust_use_imports(rust_file: str) -> List[str]: """从 Rust 文件中提取 use 引用的 state 文件""" state_files = set() if not os.path.exists(rust_file): return [] with open(rust_file, 'r', encoding='utf-8', errors='ignore') as f: content = f.read() # 匹配 use super::xxx 或 use crate::tlusty::state::xxx patterns = [ r'use\s+super::(\w+)', r'use\s+crate::tlusty::state::(\w+)', r'use\s+super::super::state::(\w+)', ] for pattern in patterns: for m in re.finditer(pattern, content): mod_name = m.group(1) # 查找对应的 .rs 文件 state_file = os.path.join(RUST_BASE_DIR, 'tlusty', 'state', f"{mod_name}.rs") if os.path.exists(state_file): state_files.add(state_file) return sorted(state_files) def generate_file_list(module_name: str) -> Dict[str, str]: """生成深度检查所需的文件列表""" files = {} name_upper = module_name.upper() # 1. Fortran 源文件 fortran_file = os.path.join(EXTRACTED_DIR, f"{module_name.lower()}.f") if os.path.exists(fortran_file): files['fortran_source'] = fortran_file else: files['fortran_source'] = f"(未找到: {fortran_file})" # 2. Rust 源文件 rust_file = find_rust_file(module_name) if rust_file: files['rust_source'] = rust_file else: files['rust_source'] = "(未找到)" # 3. INCLUDE 的 COMMON 定义文件 includes = get_includes_for_module(name_upper) for inc in includes: inc_path = os.path.join(FORTRAN_COMMON_DIR, f"{inc}.FOR") key = f"common_{inc.lower()}" if os.path.exists(inc_path): files[key] = inc_path else: files[key] = f"(未找到: {inc_path})" # 4. Rust state struct 文件（通过 use 导入） if rust_file: state_files = find_rust_use_imports(rust_file) for i, sf in enumerate(state_files): files[f"rust_state_{i}"] = sf return files def generate_prompt(module_name: str) -> str: """生成完整的 Phase 2 检查提示""" files = generate_file_list(module_name) var_map = get_mapping() structs = get_structs() # 获取模块的 COMMON 变量 module_vars = get_vars_for_module(module_name.upper(), var_map) lines = [] lines.append(f"# Phase 2 深度语义检查: {module_name.upper()}") lines.append("") lines.append("## 需要读取的文件") lines.append("") for key, path in files.items(): if not path.startswith("(未找到"): lines.append(f"- `{path}`") else: lines.append(f"- {path}") lines.append("") lines.append("## COMMON 变量映射") lines.append("") lines.append("```") # 按 COMMON 块分组 vars_by_block: Dict[str, List] = {} for vname, var in module_vars.items(): if var.common_block not in vars_by_block: vars_by_block[var.common_block] = [] vars_by_block[var.common_block].append(var) for block_name, vars in sorted(vars_by_block.items()): lines.append(f"COMMON /{block_name}/") for var in sorted(vars, key=lambda v: v.name): dims_str = f"({', '.join(var.dims)})" if var.dims else "" rust_str = f"{var.rust_struct}.{var.rust_field}" if var.rust_field else "(未映射)" lines.append(f" {var.name:20s} {dims_str:20s} → {rust_str}") lines.append("") lines.append("```") lines.append("") lines.append("## 检查清单") lines.append("") lines.append("逐项检查以下内容：") lines.append("") checklist = [ "[ ] COMMON 变量 → 正确的 Rust struct 字段", "[ ] 2D 数组下标顺序（Fortran 列主序 → Rust 行主序）", "[ ] 1-based → 0-based 索引一致性", "[ ] 循环边界转换（DO I=1,N → for i in 0..n）", "[ ] IF 条件完整保留（<= vs <, >= vs >）", "[ ] 所有赋值目标存在（无遗漏）", "[ ] CALL 顺序和数量一致", "[ ] 类型转换正确（INTEGER→i32, REAL*8→f64, LOGICAL→bool）", ] for item in checklist: lines.append(item) lines.append("") lines.append("## 发现问题处理") lines.append("") lines.append("发现 bug → 立即修复 → cargo build 验证 → 继续检查") lines.append("无 bug → 输出 '深度检查通过'") return "\n".join(lines) def main(): parser = argparse.ArgumentParser(description='Phase 2 深度检查提示生成器') parser.add_argument('module', help='模块名') parser.add_argument('--prompt', action='store_true', help='生成完整检查提示') parser.add_argument('--files', action='store_true', help='只列出文件') args = parser.parse_args() if args.prompt: print(generate_prompt(args.module)) elif args.files: files = generate_file_list(args.module) for key, path in files.items(): print(f" {key:20s} {path}") else: # 默认：输出文件列表 files = generate_file_list(args.module) print(f"模块 {args.module.upper()} 深度检查文件列表:") print() for key, path in files.items(): icon = "📄" if not path.startswith("(未找到") else "❓" print(f" {icon} {key:20s} {path}") # 也显示 COMMON 变量数 var_map = get_mapping() module_vars = get_vars_for_module(args.module.upper(), var_map) mapped = sum(1 for v in module_vars.values() if v.rust_field) print(f"\n COMMON 变量: {mapped}/{len(module_vars)} 已映射") # 提示使用 --prompt 获取完整检查提示 print(f"\n 生成完整检查提示: python3 deep_check_prompt.py {args.module} --prompt") if __name__ == "__main__": main()