Function main

ci/perf/compile_perf.py:659–723

Main entry point.

()

def main() -> None:
    """Main entry point."""
    # Parse arguments
    args = sys.argv[1:]
    output_format = "text"
    save_trace = False
    compiler = "clang++"
    threshold = 50.0
    check_thresholds_only = False

    for arg in args:
        if arg.startswith("--output="):
            output_format = arg.split("=", 1)[1]
        elif arg == "--save-trace":
            save_trace = True
        elif arg.startswith("--compiler="):
            compiler = arg.split("=", 1)[1]
        elif arg.startswith("--threshold="):
            threshold = float(arg.split("=", 1)[1])
        elif arg == "--check-thresholds":
            check_thresholds_only = True
        elif arg in ["--help", "-h"]:
            print(__doc__)
            sys.exit(0)

    # Find project root and source file
    script_dir = Path(__file__).parent
    project_root = script_dir.parent.parent
    source_file = script_dir / "test_compile.cpp"
    thresholds_file = script_dir / "thresholds.json"

    if not source_file.exists():
        print(f"Error: Test file not found at {source_file}")
        sys.exit(1)

    # Store compiler in env for report
    os.environ["COMPILER"] = compiler

    # Compile with trace
    print("Compiling test file with -ftime-trace...", file=sys.stderr)
    trace_file = compile_with_trace(
        source_file, compiler, include_dir=project_root / "src"
    )

    print(f"Trace file generated: {trace_file}", file=sys.stderr)

    # Parse trace
    print("Analyzing trace data...", file=sys.stderr)
    analyzer = CompilePerfAnalyzer(str(trace_file))

    # Check thresholds if requested
    if check_thresholds_only:
        success = check_thresholds(analyzer, thresholds_file)
        sys.exit(0 if success else 1)

    # Generate report
    report = analyzer.generate_report(output_format, threshold)
    print(report)