aaronrene / gabriel-muse public
midi_compare.py python
155 lines 6.5 KB
Raw
sha256:cf6265cea8c21d9228d90dec13ef6ec2dab5103d466db9cc4590681832de4bf8 docs(KD-STAGING): sync governance after KD-6b DONE Human 12 days ago
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 read_repo_id, 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
54 def _read_branch(root: pathlib.Path) -> str:
55 return read_current_branch(root)
56
57
58 def _load(
59 root: pathlib.Path,
60 track: str,
61 ref: str,
62 repo_id: str,
63 branch: str,
64 ) -> tuple[list[NoteInfo], int, str]:
65 commit = resolve_commit_ref(root, repo_id, branch, ref)
66 if commit is None:
67 print(f"❌ Commit '{ref}' not found.", file=sys.stderr)
68 raise SystemExit(ExitCode.USER_ERROR)
69 result = load_track(root, commit.commit_id, track)
70 if result is None:
71 print(f"❌ Track '{track}' not found in commit '{ref}'.", file=sys.stderr)
72 raise SystemExit(ExitCode.USER_ERROR)
73 return result[0], result[1], commit.commit_id[:8]
74
75
76 def register(subparsers: "argparse._SubParsersAction[argparse.ArgumentParser]") -> None:
77 """Register the compare subcommand."""
78 parser = subparsers.add_parser("compare", help="Compare two MIDI snapshots across musical dimensions.", description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)
79 parser.add_argument("track", metavar="TRACK", help="Workspace-relative path to a .mid file.")
80 parser.add_argument("ref_a", metavar="REF_A", help="First commit reference (older).")
81 parser.add_argument("ref_b", nargs="?", metavar="REF_B", default=None, help="Second commit reference. Omit to compare REF_A against the working tree.")
82 parser.add_argument("--json", action="store_true", dest="as_json", help="Emit results as JSON.")
83 parser.set_defaults(func=run)
84
85
86 def run(args: argparse.Namespace) -> None:
87 """Compare two MIDI snapshots across musical dimensions.
88
89 ``muse compare`` goes beyond raw note diffs — it shows how key, density,
90 swing, syncopation, and quantisation changed between two points in history.
91
92 For agents: use this after a merge to verify that the merged result
93 preserves the intended musical character of both parent branches.
94 """
95 track: str = args.track
96 ref_a: str = args.ref_a
97 ref_b: str | None = args.ref_b
98 as_json: bool = args.as_json
99
100 root = require_repo()
101 repo_id = read_repo_id(root)
102 branch = _read_branch(root)
103
104 notes_a, _tpb_a, label_a = _load(root, track, ref_a, repo_id, branch)
105
106 if ref_b is not None:
107 notes_b, _tpb_b, label_b = _load(root, track, ref_b, repo_id, branch)
108 else:
109 raw_b = load_track_from_workdir(root, track)
110 if raw_b is None:
111 print(f"❌ Track '{track}' not found in working tree.", file=sys.stderr)
112 raise SystemExit(ExitCode.USER_ERROR)
113 notes_b, _tpb_b = raw_b
114 label_b = "working tree"
115
116 rh_a = analyze_rhythm(notes_a)
117 rh_b = analyze_rhythm(notes_b)
118 dens_a = analyze_density(notes_a)
119 dens_b = analyze_density(notes_b)
120 avg_dens_a = sum(d["notes_per_beat"] for d in dens_a) / max(len(dens_a), 1)
121 avg_dens_b = sum(d["notes_per_beat"] for d in dens_b) / max(len(dens_b), 1)
122 key_a = key_signature_guess(notes_a)
123 key_b = key_signature_guess(notes_b)
124
125 if as_json:
126 print(json.dumps({
127 "track": track,
128 "a": {"ref": ref_a, "sha": label_a, "rhythm": rh_a, "key": key_a, "density_avg": round(avg_dens_a, 2)},
129 "b": {"ref": ref_b or "working tree", "sha": label_b, "rhythm": rh_b, "key": key_b, "density_avg": round(avg_dens_b, 2)},
130 }, indent=2))
131 return
132
133 print(f"\nSemantic comparison: {track}")
134 print(f"A: {ref_a} ({label_a}) B: {ref_b or 'working tree'} ({label_b})\n")
135 print(f" {'Dimension':<22} {'A':>16} {'B':>16} {'Δ':<30}")
136 print(" " + "─" * 90)
137
138 def row(dim: str, va: str, vb: str, delta: str) -> None:
139 print(f" {dim:<22} {va:>16} {vb:>16} {delta:<30}")
140
141 row("Notes", str(rh_a["total_notes"]), str(rh_b["total_notes"]),
142 f"{rh_b['total_notes'] - rh_a['total_notes']:+d}")
143 row("Bars", str(rh_a["bars"]), str(rh_b["bars"]),
144 f"{rh_b['bars'] - rh_a['bars']:+d}")
145 row("Key", key_a, key_b, "=" if key_a == key_b else f"{key_a} → {key_b}")
146 row("Density avg", f"{avg_dens_a:.2f}/beat", f"{avg_dens_b:.2f}/beat",
147 f"{avg_dens_b - avg_dens_a:+.2f}")
148 row("Swing ratio", f"{rh_a['swing_ratio']:.3f}", f"{rh_b['swing_ratio']:.3f}",
149 f"{rh_b['swing_ratio'] - rh_a['swing_ratio']:+.3f}")
150 row("Syncopation", f"{rh_a['syncopation_score']:.3f}", f"{rh_b['syncopation_score']:.3f}",
151 f"{rh_b['syncopation_score'] - rh_a['syncopation_score']:+.3f}")
152 row("Quantisation", f"{rh_a['quantization_score']:.3f}", f"{rh_b['quantization_score']:.3f}",
153 f"{rh_b['quantization_score'] - rh_a['quantization_score']:+.3f}")
154 row("Subdivision", rh_a["dominant_subdivision"], rh_b["dominant_subdivision"],
155 "=" if rh_a["dominant_subdivision"] == rh_b["dominant_subdivision"] else "changed")
File History 1 commit
sha256:cf6265cea8c21d9228d90dec13ef6ec2dab5103d466db9cc4590681832de4bf8 docs(KD-STAGING): sync governance after KD-6b DONE Human 12 days ago