124 lines
4.6 KiB
Python
124 lines
4.6 KiB
Python
|
import sys
|
|||
|
|
import re
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
class CoverageAnalyzer:
|
|||
|
|
def __init__(self, annotated_file_path):
|
|||
|
|
self.file_path = annotated_file_path
|
|||
|
|
self.source_code = self._load_file()
|
|||
|
|
self.module_name = self._extract_module_name()
|
|||
|
|
self.coverage_score = 0.0
|
|||
|
|
self.missing_blocks = []
|
|||
|
|
|
|||
|
|
def _load_file(self):
|
|||
|
|
"""加载文件内容，处理异常"""
|
|||
|
|
if not os.path.exists(self.file_path):
|
|||
|
|
raise FileNotFoundError(f"Cannot find file: {self.file_path}")
|
|||
|
|
with open(self.file_path, 'r', encoding='utf-8') as f:
|
|||
|
|
return f.readlines()
|
|||
|
|
|
|||
|
|
def _extract_module_name(self):
|
|||
|
|
"""通用化步骤1：自动从代码中提取 module name"""
|
|||
|
|
for line in self.source_code:
|
|||
|
|
match = re.search(r'module\s+(\w+)', line)
|
|||
|
|
if match:
|
|||
|
|
return match.group(1)
|
|||
|
|
return "unknown_module"
|
|||
|
|
|
|||
|
|
def analyze(self):
|
|||
|
|
"""核心分析逻辑：计算分数并提取未覆盖块"""
|
|||
|
|
total_lines = 0
|
|||
|
|
covered_lines = 0
|
|||
|
|
current_block = []
|
|||
|
|
recording = False
|
|||
|
|
|
|||
|
|
for i, line in enumerate(self.source_code):
|
|||
|
|
# === [修复点] 安检门：如果行是空的，直接跳过 ===
|
|||
|
|
if not line.strip():
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# 判定行类型
|
|||
|
|
is_missed = line.startswith('%000000') or line.strip().startswith('#')
|
|||
|
|
is_covered = line.startswith('%') and not is_missed
|
|||
|
|
|
|||
|
|
# 统计分数
|
|||
|
|
if is_missed or is_covered:
|
|||
|
|
total_lines += 1
|
|||
|
|
if is_covered:
|
|||
|
|
covered_lines += 1
|
|||
|
|
|
|||
|
|
# === [修复点] 安全切分：防止切割失败 ===
|
|||
|
|
parts = line.split(maxsplit=1)
|
|||
|
|
if len(parts) < 2:
|
|||
|
|
# 如果这一行切不出两部分（比如只有标号没有代码），也跳过
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
clean_code = parts[-1].strip()
|
|||
|
|
# 过滤掉非实质性代码
|
|||
|
|
is_substantive = len(clean_code) > 3 and not clean_code.startswith("//")
|
|||
|
|
|
|||
|
|
if is_missed and is_substantive:
|
|||
|
|
if not recording:
|
|||
|
|
recording = True
|
|||
|
|
# 添加上下文 (尝试找前一行非空行作为上下文比较复杂，这里简化处理)
|
|||
|
|
if i > 0 and self.source_code[i-1].strip():
|
|||
|
|
prev_parts = self.source_code[i-1].split(maxsplit=1)
|
|||
|
|
if len(prev_parts) >= 2:
|
|||
|
|
prev_code = prev_parts[-1].strip()
|
|||
|
|
current_block.append(f"Line {i}: {prev_code} ...")
|
|||
|
|
current_block.append(f"Line {i+1}: {clean_code} <--- MISSING")
|
|||
|
|
else:
|
|||
|
|
if recording:
|
|||
|
|
recording = False
|
|||
|
|
self.missing_blocks.append(current_block)
|
|||
|
|
current_block = []
|
|||
|
|
|
|||
|
|
# 计算最终分数
|
|||
|
|
if total_lines > 0:
|
|||
|
|
self.coverage_score = (covered_lines / total_lines) * 100.0
|
|||
|
|
|
|||
|
|
return self.coverage_score
|
|||
|
|
|
|||
|
|
def generate_prompt(self):
|
|||
|
|
"""通用化步骤2：使用模板生成 Prompt"""
|
|||
|
|
prompt_template = """
|
|||
|
|
You are an expert FPGA verification engineer.
|
|||
|
|
I am verifying a Verilog module named '{module_name}'.
|
|||
|
|
The current testbench achieves only {score:.2f}% coverage.
|
|||
|
|
The following logic blocks are NEVER executed during simulation:
|
|||
|
|
|
|||
|
|
{missing_logic_str}
|
|||
|
|
|
|||
|
|
[Task]
|
|||
|
|
Please write a NEW SystemVerilog task to target these specific missing scenarios.
|
|||
|
|
1. Analyze WHY these lines are not executed (e.g., specific counter value, rare state transition).
|
|||
|
|
2. Generate a task named 'test_coverage_fix' that drives inputs to trigger these conditions.
|
|||
|
|
3. Return ONLY the code block.
|
|||
|
|
"""
|
|||
|
|
blocks_str = ""
|
|||
|
|
for idx, block in enumerate(self.missing_blocks[:4]):
|
|||
|
|
blocks_str += f"--- Missing Scenario {idx+1} ---\n"
|
|||
|
|
blocks_str += "\n".join(block)
|
|||
|
|
blocks_str += "\n\n"
|
|||
|
|
|
|||
|
|
if not blocks_str:
|
|||
|
|
blocks_str = "(No significant missing logic found. Coverage is likely high.)"
|
|||
|
|
|
|||
|
|
return prompt_template.format(
|
|||
|
|
module_name=self.module_name,
|
|||
|
|
score=self.coverage_score,
|
|||
|
|
missing_logic_str=blocks_str
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
if len(sys.argv) < 2:
|
|||
|
|
print("Usage: python3 coverage_agent.py <annotated_file_path>")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
agent = CoverageAnalyzer(sys.argv[1])
|
|||
|
|
score = agent.analyze()
|
|||
|
|
|
|||
|
|
print(f"--- Analysis Report for {agent.module_name} ---")
|
|||
|
|
print(f"Score: {score:.2f}%")
|
|||
|
|
print("-" * 30)
|
|||
|
|
print(agent.generate_prompt())
|