Function main

ci/tools/generate_audio_fixtures.py:451–547

()

def main() -> None:
    parser = argparse.ArgumentParser(
        description="Generate audio fixtures for VocalDetector tests"
    )
    parser.add_argument(
        "--output",
        type=Path,
        default=Path("tests/data/audio"),
        help="Output directory (default: tests/data/audio)",
    )
    parser.add_argument(
        "--synth-only",
        action="store_true",
        help="Use synthetic generation only (no downloads)",
    )
    parser.add_argument(
        "--cache-dir",
        type=Path,
        default=Path(tempfile.gettempdir()) / "fastled_audio_cache",
        help="Cache directory for downloads",
    )
    args = parser.parse_args()

    ffmpeg = find_ffmpeg()
    print(f"Using ffmpeg: {ffmpeg}")

    # Create output directories
    stems_dir = args.output / "stems"
    mixes_dir = args.output / "mixes"
    stems_dir.mkdir(parents=True, exist_ok=True)
    mixes_dir.mkdir(parents=True, exist_ok=True)
    args.cache_dir.mkdir(parents=True, exist_ok=True)

    # --- Generate stems ---
    print("\n=== Generating stems ===")
    all_stems = ["voice_male", "voice_female", "guitar_acoustic", "drums_jazz"]

    for stem_name in all_stems:
        output_path = stems_dir / f"{stem_name}.ogg"
        if output_path.exists():
            print(
                f"  {stem_name}: already exists ({output_path.stat().st_size // 1024}KB)"
            )
            continue

        downloaded = False
        if not args.synth_only and stem_name in SOURCES:
            src = SOURCES[stem_name]
            downloaded = download_and_trim(
                ffmpeg, src["url"], src["start"], output_path, args.cache_dir
            )

        if not downloaded:
            print(f"  {stem_name}: using synthetic fallback")
            generate_synthetic_stem(ffmpeg, stem_name, output_path)

    # --- Generate mixes ---
    print("\n=== Generating mixes ===")