test_resolution_compression.py

import subprocess
import time
import os

def run_test():
    # 1. Generate Test Image
    print("Step 1: Generating Test Image...")
    subprocess.run(["python3", "gen_test.py"], check=True)
    
    presets = ['ntsc', '720p']
    compression_levels = [str(x) for x in range(0, 101, 10)]
    
    results = []
    
    print(f"Testing presets: {presets}")
    print(f"Testing compression levels: {compression_levels}")
    
    for preset in presets:
        for comp in compression_levels:
            output_image = f"res_comp_{preset}_q{comp}.png"
            print(f"\n--- Testing Preset: {preset} | Compression: {comp} ---")
            start_time = time.time()
            
            try:
                # Run sstv_filter
                cmd = [
                    "python3", "sstv_filter.py", 
                    "test_pattern.png", 
                    output_image, 
                    "--preset", preset,
                    "--compression", comp
                ]
                
                subprocess.run(cmd, check=True)
                elapsed = time.time() - start_time
                print(f"Success! Time: {elapsed:.2f}s")
                results.append((preset, comp, "Success", f"{elapsed:.2f}s", output_image))
                
            except subprocess.CalledProcessError as e:
                print(f"Failed: {e}")
                results.append((preset, comp, "Failed", "N/A", "N/A"))

    # Generate Report
    with open("res_comp_report.md", "w") as f:
        f.write("# SSTV Resolution vs Compression Report\n\n")
        f.write("| Resolution | Quality (0-100) | Status | Time | Result |\n")
        f.write("|------------|-----------------|--------|------|--------|\n")
        
        for preset, comp, status, elapsed, img in results:
            if status == "Success":
                 f.write(f"| {preset} | {comp} | {status} | {elapsed} | ![{preset}_q{comp}]({img}) |\n")
            else:
                 f.write(f"| {preset} | {comp} | {status} | {elapsed} | N/A |\n")
                 
    print("\nTest Complete. Output saved to res_comp_report.md")

if __name__ == "__main__":
    run_test()