feat: rewrite plot_pitch_patterns with compact layout, contour plots, and segmentation

Rewrites pitch pattern visualization to match the IDTAP web app's Analyzer
component. Adds multiple pattern sizes, segmentation modes (section/phrase/
duration), chroma-colored cells with contrast-aware text, and optional
melodic contour curves via Trajectory.compute(). Uses inch-based equal-aspect
layout for compact spreadsheet-style rendering.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: jon-myers

demo_visualizations.py

@@ -9,7 +9,7 @@
 import matplotlib.pyplot as plt
 
 from idtap import SwaraClient, Piece
-from idtap.visualization import plot_pitch_prevalence
+from idtap.visualization import plot_pitch_prevalence, plot_pitch_patterns
 
 OUTPUT_DIR = os.path.join(os.path.dirname(__file__), 'demo_output')
 os.makedirs(OUTPUT_DIR, exist_ok=True)
@@ -189,6 +189,55 @@ def main():
             import traceback
             traceback.print_exc()
 
+    # ------------------------------------------------------------------
+    # Pitch pattern visualizations
+    # ------------------------------------------------------------------
+    pattern_configs = [
+        {
+            'label': '13_patterns_3gram',
+            'desc': 'Patterns / 3-gram / pitchNumber',
+            'kwargs': dict(pattern_size=3, output_type='pitchNumber'),
+        },
+        {
+            'label': '14_patterns_3gram_plot',
+            'desc': 'Patterns / 3-gram / pitchNumber / contour',
+            'kwargs': dict(pattern_size=3, output_type='pitchNumber', plot=True),
+        },
+        {
+            'label': '15_patterns_multi_section',
+            'desc': 'Patterns / [2,3,4]-gram / section segmentation',
+            'kwargs': dict(
+                pattern_sizes=[2, 3, 4], segmentation='section',
+                output_type='pitchNumber', max_patterns=10,
+            ),
+        },
+        {
+            'label': '16_patterns_sargam',
+            'desc': 'Patterns / 3-gram / sargamLetter',
+            'kwargs': dict(pattern_size=3, output_type='sargamLetter'),
+        },
+    ]
+
+    print(f"\nGenerating {len(pattern_configs)} pattern visualizations...\n")
+
+    for cfg in pattern_configs:
+        label = cfg['label']
+        desc = cfg['desc']
+        kwargs = cfg['kwargs']
+
+        print(f"  [{label}] {desc}...", end=' ', flush=True)
+        try:
+            fig = plot_pitch_patterns(piece, title=piece_title, **kwargs)
+            path = os.path.join(OUTPUT_DIR, f'{label}.png')
+            fig.savefig(path, dpi=300, bbox_inches='tight',
+                        pad_inches=0.2, facecolor='white', edgecolor='none')
+            plt.close(fig)
+            print(f"OK → {path}")
+        except Exception as e:
+            print(f"FAILED: {e}")
+            import traceback
+            traceback.print_exc()
+
     print(f"\nAll outputs saved to: {OUTPUT_DIR}/")
 
 

idtap/tests/visualization_test.py

@@ -217,16 +217,31 @@ def test_empty_patterns(self):
         assert fig is not None
         plt.close(fig)
 
-    def test_with_existing_axes(self):
-        fig, ax = plt.subplots()
+    def test_multiple_sizes(self):
         trajs = [
             _fixed_traj('sa', dur=1.0),
             _fixed_traj('re', dur=1.0),
+            _fixed_traj('ga', dur=1.0),
             _fixed_traj('sa', dur=1.0),
             _fixed_traj('re', dur=1.0),
+            _fixed_traj('ga', dur=1.0),
         ]
-        returned_fig = plot_pitch_patterns(trajs, pattern_size=2, ax=ax)
-        assert returned_fig is fig
+        fig = plot_pitch_patterns(trajs, pattern_sizes=[2, 3])
+        assert fig is not None
+        assert hasattr(fig, 'savefig')
+        plt.close(fig)
+
+    def test_with_contour_plot(self):
+        trajs = [
+            _fixed_traj('sa', dur=1.0),
+            _fixed_traj('re', dur=1.0),
+            _fixed_traj('ga', dur=1.0),
+            _fixed_traj('sa', dur=1.0),
+            _fixed_traj('re', dur=1.0),
+            _fixed_traj('ga', dur=1.0),
+        ]
+        fig = plot_pitch_patterns(trajs, pattern_size=2, plot=True)
+        assert fig is not None
         plt.close(fig)
 
     def test_max_patterns_limit(self):