midi_compare.py
python
| 1 | """muse compare — semantic comparison between two MIDI snapshots. |
| 2 | |
| 3 | Diffs two commits (or a commit and the working tree) on multiple musical |
| 4 | dimensions: note count, harmonic content, rhythmic feel, pitch range, and |
| 5 | density. Where ``muse diff`` shows note-level insertions/deletions, this |
| 6 | command shows the *musical meaning* of what changed. |
| 7 | |
| 8 | Usage:: |
| 9 | |
| 10 | muse compare tracks/melody.mid HEAD~1 HEAD |
| 11 | muse compare tracks/piano.mid HEAD~3 HEAD~1 |
| 12 | muse compare tracks/bass.mid HEAD --working-tree |
| 13 | muse compare tracks/chords.mid HEAD~1 HEAD --json |
| 14 | |
| 15 | Output:: |
| 16 | |
| 17 | Semantic comparison: tracks/melody.mid |
| 18 | A: HEAD~1 (cb4afaed) B: HEAD (9f3a12e7) |
| 19 | |
| 20 | Dimension A B Δ |
| 21 | ────────────────────────────────────────────────────────── |
| 22 | Notes 48 56 +8 |
| 23 | Bars 16 16 0 |
| 24 | Key G major G major = |
| 25 | Density avg 3.0/beat 3.5/beat +0.5 |
| 26 | Swing ratio 1.00 1.38 +0.38 (swing added) |
| 27 | Syncopation 0.12 0.31 +0.19 (more syncopated) |
| 28 | Quantisation 0.98 0.84 -0.14 (more human) |
| 29 | """ |
| 30 | |
| 31 | from __future__ import annotations |
| 32 | |
| 33 | import argparse |
| 34 | import json |
| 35 | import logging |
| 36 | import pathlib |
| 37 | import sys |
| 38 | |
| 39 | from muse.core.errors import ExitCode |
| 40 | from muse.core.repo import require_repo |
| 41 | from muse.core.store import read_current_branch, resolve_commit_ref |
| 42 | from muse.plugins.midi._analysis import analyze_rhythm, analyze_density |
| 43 | from muse.plugins.midi._query import ( |
| 44 | NoteInfo, |
| 45 | key_signature_guess, |
| 46 | load_track, |
| 47 | load_track_from_workdir, |
| 48 | ) |
| 49 | |
| 50 | logger = logging.getLogger(__name__) |
| 51 | |
| 52 | |
| 53 | def _read_repo_id(root: pathlib.Path) -> str: |
| 54 | import json as _json |
| 55 | |
| 56 | return str(_json.loads((root / ".muse" / "repo.json").read_text())["repo_id"]) |
| 57 | |
| 58 | |
| 59 | def _read_branch(root: pathlib.Path) -> str: |
| 60 | return read_current_branch(root) |
| 61 | |
| 62 | |
| 63 | def _load( |
| 64 | root: pathlib.Path, |
| 65 | track: str, |
| 66 | ref: str, |
| 67 | repo_id: str, |
| 68 | branch: str, |
| 69 | ) -> tuple[list[NoteInfo], int, str]: |
| 70 | commit = resolve_commit_ref(root, repo_id, branch, ref) |
| 71 | if commit is None: |
| 72 | print(f"❌ Commit '{ref}' not found.", file=sys.stderr) |
| 73 | raise SystemExit(ExitCode.USER_ERROR) |
| 74 | result = load_track(root, commit.commit_id, track) |
| 75 | if result is None: |
| 76 | print(f"❌ Track '{track}' not found in commit '{ref}'.", file=sys.stderr) |
| 77 | raise SystemExit(ExitCode.USER_ERROR) |
| 78 | return result[0], result[1], commit.commit_id[:8] |
| 79 | |
| 80 | |
| 81 | def register(subparsers: "argparse._SubParsersAction[argparse.ArgumentParser]") -> None: |
| 82 | """Register the compare subcommand.""" |
| 83 | parser = subparsers.add_parser("compare", help="Compare two MIDI snapshots across musical dimensions.", description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter) |
| 84 | parser.add_argument("track", metavar="TRACK", help="Workspace-relative path to a .mid file.") |
| 85 | parser.add_argument("ref_a", metavar="REF_A", help="First commit reference (older).") |
| 86 | parser.add_argument("ref_b", nargs="?", metavar="REF_B", default=None, help="Second commit reference. Omit to compare REF_A against the working tree.") |
| 87 | parser.add_argument("--json", action="store_true", dest="as_json", help="Emit results as JSON.") |
| 88 | parser.set_defaults(func=run) |
| 89 | |
| 90 | |
| 91 | def run(args: argparse.Namespace) -> None: |
| 92 | """Compare two MIDI snapshots across musical dimensions. |
| 93 | |
| 94 | ``muse compare`` goes beyond raw note diffs — it shows how key, density, |
| 95 | swing, syncopation, and quantisation changed between two points in history. |
| 96 | |
| 97 | For agents: use this after a merge to verify that the merged result |
| 98 | preserves the intended musical character of both parent branches. |
| 99 | """ |
| 100 | track: str = args.track |
| 101 | ref_a: str = args.ref_a |
| 102 | ref_b: str | None = args.ref_b |
| 103 | as_json: bool = args.as_json |
| 104 | |
| 105 | root = require_repo() |
| 106 | repo_id = _read_repo_id(root) |
| 107 | branch = _read_branch(root) |
| 108 | |
| 109 | notes_a, _tpb_a, label_a = _load(root, track, ref_a, repo_id, branch) |
| 110 | |
| 111 | if ref_b is not None: |
| 112 | notes_b, _tpb_b, label_b = _load(root, track, ref_b, repo_id, branch) |
| 113 | else: |
| 114 | raw_b = load_track_from_workdir(root, track) |
| 115 | if raw_b is None: |
| 116 | print(f"❌ Track '{track}' not found in working tree.", file=sys.stderr) |
| 117 | raise SystemExit(ExitCode.USER_ERROR) |
| 118 | notes_b, _tpb_b = raw_b |
| 119 | label_b = "working tree" |
| 120 | |
| 121 | rh_a = analyze_rhythm(notes_a) |
| 122 | rh_b = analyze_rhythm(notes_b) |
| 123 | dens_a = analyze_density(notes_a) |
| 124 | dens_b = analyze_density(notes_b) |
| 125 | avg_dens_a = sum(d["notes_per_beat"] for d in dens_a) / max(len(dens_a), 1) |
| 126 | avg_dens_b = sum(d["notes_per_beat"] for d in dens_b) / max(len(dens_b), 1) |
| 127 | key_a = key_signature_guess(notes_a) |
| 128 | key_b = key_signature_guess(notes_b) |
| 129 | |
| 130 | if as_json: |
| 131 | print(json.dumps({ |
| 132 | "track": track, |
| 133 | "a": {"ref": ref_a, "sha": label_a, "rhythm": rh_a, "key": key_a, "density_avg": round(avg_dens_a, 2)}, |
| 134 | "b": {"ref": ref_b or "working tree", "sha": label_b, "rhythm": rh_b, "key": key_b, "density_avg": round(avg_dens_b, 2)}, |
| 135 | }, indent=2)) |
| 136 | return |
| 137 | |
| 138 | print(f"\nSemantic comparison: {track}") |
| 139 | print(f"A: {ref_a} ({label_a}) B: {ref_b or 'working tree'} ({label_b})\n") |
| 140 | print(f" {'Dimension':<22} {'A':>16} {'B':>16} {'Δ':<30}") |
| 141 | print(" " + "─" * 90) |
| 142 | |
| 143 | def row(dim: str, va: str, vb: str, delta: str) -> None: |
| 144 | print(f" {dim:<22} {va:>16} {vb:>16} {delta:<30}") |
| 145 | |
| 146 | row("Notes", str(rh_a["total_notes"]), str(rh_b["total_notes"]), |
| 147 | f"{rh_b['total_notes'] - rh_a['total_notes']:+d}") |
| 148 | row("Bars", str(rh_a["bars"]), str(rh_b["bars"]), |
| 149 | f"{rh_b['bars'] - rh_a['bars']:+d}") |
| 150 | row("Key", key_a, key_b, "=" if key_a == key_b else f"{key_a} → {key_b}") |
| 151 | row("Density avg", f"{avg_dens_a:.2f}/beat", f"{avg_dens_b:.2f}/beat", |
| 152 | f"{avg_dens_b - avg_dens_a:+.2f}") |
| 153 | row("Swing ratio", f"{rh_a['swing_ratio']:.3f}", f"{rh_b['swing_ratio']:.3f}", |
| 154 | f"{rh_b['swing_ratio'] - rh_a['swing_ratio']:+.3f}") |
| 155 | row("Syncopation", f"{rh_a['syncopation_score']:.3f}", f"{rh_b['syncopation_score']:.3f}", |
| 156 | f"{rh_b['syncopation_score'] - rh_a['syncopation_score']:+.3f}") |
| 157 | row("Quantisation", f"{rh_a['quantization_score']:.3f}", f"{rh_b['quantization_score']:.3f}", |
| 158 | f"{rh_b['quantization_score'] - rh_a['quantization_score']:+.3f}") |
| 159 | row("Subdivision", rh_a["dominant_subdivision"], rh_b["dominant_subdivision"], |
| 160 | "=" if rh_a["dominant_subdivision"] == rh_b["dominant_subdivision"] else "changed") |