Implement UpdateDictPattern

supriya-project · Jan 14, 2025 · f0b790c · f0b790c
1 parent f218de9
commit f0b790c
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 5 deletions.
diff --git a/supriya/patterns/__init__.py b/supriya/patterns/__init__.py
@@ -1,4 +1,10 @@
-from .eventpatterns import ChainPattern, EventPattern, MonoEventPattern, UpdatePattern
+from .eventpatterns import (
+    ChainPattern,
+    EventPattern,
+    MonoEventPattern,
+    UpdateDictPattern,
+    UpdatePattern,
+)
 from .events import (
     BusAllocateEvent,
     BusFreeEvent,
@@ -54,4 +60,5 @@
     "SynthAllocateEvent",
     "UnaryOpPattern",
     "UpdatePattern",
+    "UpdateDictPattern",
 ]
diff --git a/supriya/patterns/eventpatterns.py b/supriya/patterns/eventpatterns.py
@@ -7,7 +7,7 @@
 from .patterns import Pattern, SequencePattern
 
 
-class EventPattern(Pattern):
+class EventPattern(Pattern[Event]):
     """
     Akin to SuperCollider's Pbind.
     """
@@ -82,7 +82,7 @@ def is_infinite(self) -> bool:
         return True
 
 
-class UpdatePattern(Pattern):
+class UpdatePattern(Pattern[Event]):
     """
     Akin to SuperCollider's Pbindf.
     """
@@ -129,7 +129,29 @@ def is_infinite(self) -> bool:
         return self._pattern.is_infinite
 
 
-class ChainPattern(Pattern):
+class UpdateDictPattern(Pattern[Event]):
+    def __init__(self, pattern: Pattern[Event], dictionary: Dict[str, Any]) -> None:
+        self._pattern = pattern
+        self._dictionary = dictionary
+
+    def _iterate(
+        self, state: Optional[Dict[str, UUID]] = None
+    ) -> Generator[Event, bool, None]:
+        event_iterator = iter(self._pattern)
+        while True:
+            try:
+                event = next(event_iterator)
+            except StopIteration:
+                return
+            if (yield new(event, **self._dictionary)):
+                return
+
+    @property
+    def is_infinite(self) -> bool:
+        return self._pattern.is_infinite
+
+
+class ChainPattern(Pattern[Event]):
     """
     Akin to SuperCollider's Pchain.
     """

diff --git a/tests/patterns/test_PatternPlayer.py b/tests/patterns/test_PatternPlayer.py
@@ -12,6 +12,7 @@
 from supriya.osc import OscBundle, OscMessage
 from supriya.patterns import (
     BusPattern,
+    Event,
     EventPattern,
     GroupPattern,
     MonoEventPattern,
@@ -25,7 +26,7 @@
     "pattern, until, target_node, expected",
     [
         (
-            SequencePattern(
+            SequencePattern[Event](
                 [
                     EventPattern(frequency=SequencePattern([440, 550, 660])),
                     MonoEventPattern(frequency=SequencePattern([440, 550, 660])),