hyper-animator-codex 0.7.1 → 0.8.1

package/skills/hyper-animator-codex/vendor/music-beat-detector/beat_detector/analyzer.py

@@ -23,6 +23,7 @@ class MetaInfo:
     sample_rate: int
     bpm: float
     time_signature: str = "4/4"
+    beat_grid: dict = field(default_factory=dict)
 
 
 @dataclass
@@ -116,7 +117,8 @@ def analyze(
         duration_ms=duration_ms,
         sample_rate=sr,
         bpm=beat_result.bpm,
-        time_signature=beat_result.time_signature
+        time_signature=beat_result.time_signature,
+        beat_grid=beat_result.beat_grid
     )
 
     result = AnalysisResult(

package/skills/hyper-animator-codex/vendor/music-beat-detector/beat_detector/beat.py

@@ -30,6 +30,41 @@ class BeatResult:
     bpm: float
     beats: List[Beat]
     time_signature: str = "4/4"
+    beat_grid: dict = None
+
+
+def _fit_beat_grid(beat_times: np.ndarray, fallback_bpm: float) -> tuple[float, dict]:
+    """Fit t_i = phase + i * period and report grid reliability."""
+    if len(beat_times) < 2:
+        period = 60.0 / fallback_bpm if fallback_bpm > 0 else 0.0
+        phase = float(beat_times[0]) if len(beat_times) == 1 else 0.0
+        return fallback_bpm, {
+            "method": "insufficient_beats",
+            "phase_ms": round(phase * 1000, 3),
+            "period_ms": round(period * 1000, 3),
+            "max_residual_ms": None,
+            "rms_residual_ms": None,
+            "reliable": False,
+            "threshold_ms": 15.0,
+        }
+
+    indexes = np.arange(len(beat_times), dtype=float)
+    design = np.vstack([indexes, np.ones_like(indexes)]).T
+    (period, phase), *_ = np.linalg.lstsq(design, beat_times, rcond=None)
+    residuals = beat_times - (phase + indexes * period)
+    max_residual_ms = float(np.max(np.abs(residuals)) * 1000)
+    rms_residual_ms = float(np.sqrt(np.mean(residuals ** 2)) * 1000)
+    fitted_bpm = 60.0 / period if period > 0 else fallback_bpm
+
+    return fitted_bpm, {
+        "method": "least_squares",
+        "phase_ms": round(float(phase) * 1000, 3),
+        "period_ms": round(float(period) * 1000, 3),
+        "max_residual_ms": round(max_residual_ms, 3),
+        "rms_residual_ms": round(rms_residual_ms, 3),
+        "reliable": max_residual_ms <= 15.0,
+        "threshold_ms": 15.0,
+    }
 
 
 def detect_beats(
@@ -52,6 +87,7 @@ def detect_beats(
 
     # 转换为时间（毫秒）
     beat_times = librosa.frames_to_time(beat_frames, sr=sr)
+    bpm, beat_grid = _fit_beat_grid(beat_times, bpm)
     beat_times_ms = [int(t * 1000) for t in beat_times]
 
     # 计算每个beat的能量
@@ -82,7 +118,7 @@ def detect_beats(
     if on_progress:
         on_progress(100.0)
 
-    return BeatResult(bpm=bpm, beats=beats)
+    return BeatResult(bpm=bpm, beats=beats, beat_grid=beat_grid)
 
 
 def _compute_beat_energies(