]> git.immae.eu Git - perso/Immae/Projets/Python/MusicSampler.git/blobdiff - helpers/action.py
Cleanup actions and subscribe to music events for loading
[perso/Immae/Projets/Python/MusicSampler.git] / helpers / action.py
index 247e0a9c50ff765cdf73d33490a5895930575a53..a6c48e98d5140ca3460fe7103472f8cffbd9968a 100644 (file)
@@ -1,11 +1,13 @@
 import threading
 import time
 
-from . import debug_print
+from transitions.extensions import HierarchicalMachine as Machine
+from . import debug_print, error_print
 
 class Action:
-    action_types = [
+    ACTION_TYPES = [
         'command',
+        'interrupt_wait',
         'pause',
         'play',
         'seek',
@@ -16,40 +18,106 @@ class Action:
         'wait',
     ]
 
+    STATES = [
+        'initial',
+        'loading',
+        'failed',
+        {
+            'name': 'loaded',
+            'children': ['running']
+        }
+    ]
+
+    TRANSITIONS = [
+        {
+            'trigger': 'load',
+            'source': 'initial',
+            'dest': 'loading'
+        },
+        {
+            'trigger': 'fail',
+            'source': 'loading',
+            'dest': 'failed'
+        },
+        {
+            'trigger': 'success',
+            'source': 'loading',
+            'dest': 'loaded'
+        },
+        {
+            'trigger': 'run',
+            'source': 'loaded',
+            'dest': 'loaded_running',
+            'after': 'finish_action'
+        },
+        {
+            'trigger': 'interrupt',
+            'source': 'loaded_running',
+            'dest': 'loaded',
+            'before': 'trigger_interrupt'
+        },
+        {
+            'trigger': 'finish_action',
+            'source': 'loaded_running',
+            'dest': 'loaded'
+        }
+    ]
+
     def __init__(self, action, key, **kwargs):
-        if action in self.action_types:
-            self.action = action
-        else:
-            raise Exception("Unknown action {}".format(action))
+        Machine(model=self, states=self.STATES,
+                transitions=self.TRANSITIONS, initial='initial',
+                ignore_invalid_triggers=True, queued=True)
 
+        self.action = action
         self.key = key
         self.mapping = key.parent
         self.arguments = kwargs
         self.sleep_event = None
+        self.waiting_music = None
+        self.load()
 
     def ready(self):
-        if 'music' in self.arguments:
-            return self.arguments['music'].check_is_loaded()
+        return self.is_loaded(allow_substates=True)
+
+    def callback_loaded(self, success):
+        if success:
+            self.success()
         else:
-            return True
+            self.fail()
 
-    def run(self):
+    # Machine states / events
+    def on_enter_loading(self):
+        if self.action in self.ACTION_TYPES:
+            if 'music' in self.arguments:
+                self.arguments['music'].subscribe_loaded(self.callback_loaded)
+            else:
+                self.success()
+        else:
+            error_print("Unknown action {}".format(self.action))
+            self.fail()
+
+
+    def on_enter_loaded_running(self):
         debug_print(self.description())
         getattr(self, self.action)(**self.arguments)
 
-    def description(self):
-        return getattr(self, self.action + "_print")(**self.arguments)
-
-    def interrupt(self):
+    def trigger_interrupt(self):
         if getattr(self, self.action + "_interrupt", None):
             return getattr(self, self.action + "_interrupt")(**self.arguments)
 
+    # Helpers
     def music_list(self, music):
         if music is not None:
             return [music]
         else:
             return self.mapping.open_files.values()
 
+    def description(self):
+        if getattr(self, self.action + "_print", None):
+            return getattr(self, self.action + "_print")(**self.arguments)
+        else:
+            return "unknown action {}".format(self.action)
+
     # Actions
     def command(self, command="", **kwargs):
         # FIXME: todo
@@ -70,14 +138,14 @@ class Action:
             loop=0, **kwargs):
         for music in self.music_list(music):
             if restart_if_running:
-                if music.is_not_stopped():
+                if music.is_in_use():
                     music.stop()
                 music.play(
                         volume=volume,
                         fade_in=fade_in,
                         start_at=start_at,
                         loop=loop)
-            elif not music.is_not_stopped():
+            elif not music.is_in_use():
                 music.play(
                         volume=volume,
                         fade_in=fade_in,
@@ -88,16 +156,26 @@ class Action:
         for music in self.music_list(music):
             music.seek(value=value, delta=delta)
 
-    def stop(self, music=None, fade_out=0, wait=False, **kwargs):
+    def interrupt_wait(self, wait_id=None):
+        self.mapping.interrupt_wait(wait_id)
+
+    def stop(self, music=None, fade_out=0, wait=False,
+            set_wait_id=None, **kwargs):
         previous = None
         for music in self.music_list(music):
             if music.is_loaded_paused() or music.is_loaded_playing():
                 if previous is not None:
                     previous.stop(fade_out=fade_out)
                 previous = music
+            else:
+                music.stop(fade_out=fade_out)
 
         if previous is not None:
-            previous.stop(fade_out=fade_out, wait=wait)
+            self.waiting_music = previous
+            previous.stop(
+                    fade_out=fade_out,
+                    wait=wait,
+                    set_wait_id=set_wait_id)
 
     def stop_all_actions(self, **kwargs):
         self.mapping.stop_all_running()
@@ -108,7 +186,10 @@ class Action:
         else:
             self.mapping.set_master_volume(value, delta=delta, fade=fade)
 
-    def wait(self, duration=0, music=None, **kwargs):
+    def wait(self, duration=0, music=None, set_wait_id=None, **kwargs):
+        if set_wait_id is not None:
+            self.mapping.add_wait_id(set_wait_id, self)
+
         self.sleep_event = threading.Event()
 
         if music is not None:
@@ -121,6 +202,9 @@ class Action:
     def command_print(self, command="", **kwargs):
         return "running command {}".format(command)
 
+    def interrupt_wait_print(self, wait_id=None, **kwargs):
+        return "interrupt wait with id {}".format(wait_id)
+
     def pause_print(self, music=None, **kwargs):
         if music is not None:
             return "pausing « {} »".format(music.name)
@@ -159,7 +243,9 @@ class Action:
 
         return message
 
-    def stop_print(self, music=None, fade_out=0, wait=False, **kwargs):
+    def stop_print(self, music=None, fade_out=0, wait=False,
+            set_wait_id=None, **kwargs):
+
         message = "stopping "
         if music is not None:
             message += "music « {} »".format(music.name)
@@ -169,7 +255,11 @@ class Action:
         if fade_out > 0:
             message += " with {}s fadeout".format(fade_out)
             if wait:
-                message += " (waiting the end of fadeout)"
+                if set_wait_id is not None:
+                    message += " (waiting the end of fadeout, with id {})"\
+                            .format(set_wait_id)
+                else:
+                    message += " (waiting the end of fadeout)"
 
         return message
 
@@ -215,22 +305,31 @@ class Action:
 
         return message
 
-    def wait_print(self, duration=0, music=None, **kwargs):
+    def wait_print(self, duration=0, music=None, set_wait_id=None, **kwargs):
+        message = ""
         if music is None:
-            return "waiting {}s" \
+            message += "waiting {}s" \
                     .format(duration)
         elif duration == 0:
-            return "waiting the end of « {} »" \
+            message += "waiting the end of « {} »" \
                     .format(music.name)
         else:
-            return "waiting the end of « {} » + {}s" \
+            message += "waiting the end of « {} » + {}s" \
                     .format(music.name, duration)
 
+        if set_wait_id is not None:
+            message += " (setting id = {})".format(set_wait_id)
+
+        return message
 
-    # Interruptions
+    # Interruptions (only for non-"atomic" actions)
     def wait_interrupt(self, duration=0, music=None, **kwargs):
         if self.sleep_event is not None:
             self.sleep_event.set()
         if music is not None:
             music.wait_event.set()
 
+    def stop_interrupt(self, music=None, fade_out=0, wait=False,
+            set_wait_id=None, **kwargs):
+        if self.waiting_music is not None:
+            self.waiting_music.wait_event.set()