X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=helpers%2Faction.py;h=4b5a71d3806275b5bf16198be985fda052f7b61f;hb=6226efcadfe8c3a932c7a948032b433afb0d981d;hp=a68de90e5f1348934804d08c2a194fac14133eb1;hpb=be27763f8be0f647cbe17ecee8c782901ce2cede;p=perso%2FImmae%2FProjets%2FPython%2FMusicSampler.git diff --git a/helpers/action.py b/helpers/action.py index a68de90..4b5a71d 100644 --- a/helpers/action.py +++ b/helpers/action.py @@ -1,119 +1,113 @@ -import pygame -import time +from transitions.extensions import HierarchicalMachine as Machine +from . import debug_print, error_print +from . import actions class Action: - action_types = [ - 'command', - 'pause', - 'play', - 'stop', - 'stop_all_actions', - 'volume', - 'wait', + STATES = [ + 'initial', + 'loading', + 'failed', + { + 'name': 'loaded', + 'children': ['running'] + } + ] + + TRANSITIONS = [ + { + 'trigger': 'load', + 'source': 'initial', + 'dest': 'loading' + }, + { + 'trigger': 'fail', + 'source': 'loading', + 'dest': 'failed', + 'after': 'poll_loaded' + }, + { + 'trigger': 'success', + 'source': 'loading', + 'dest': 'loaded', + 'after': 'poll_loaded' + }, + { + 'trigger': 'run', + 'source': 'loaded', + 'dest': 'loaded_running', + 'after': 'finish_action', + # if a child has no transitions, then it is bubbled to the parent, + # and we don't want that. Not useful in that machine precisely. + 'conditions': ['is_loaded'] + }, + { + '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 - def ready(self): - if 'music' in self.arguments: - return self.arguments['music'].loaded - else: - return True - - def run(self): - print(getattr(self, self.action + "_print")(**self.arguments)) - return getattr(self, self.action)(**self.arguments) - - def command(self, command = "", **kwargs): - pass - - def pause(self, music = None, **kwargs): - if music is not None: - music.pause() - else: - pygame.mixer.pause() - - def play(self, music = None, fade_in = 0, start_at = 0, - restart_if_running = False, volume = 100, **kwargs): - if music is not None: - music.play() - else: - pygame.mixer.unpause() - - def stop(self, music = None, fade_out = 0, **kwargs): - if music is not None: - music.stop() - else: - pygame.mixer.stop() - - def stop_all_actions(self, **kwargs): - self.key.mapping.stop_all_running() - - def volume(self, music = None, value = 100, **kwargs): - pass - - def wait(self, duration = 0, **kwargs): - time.sleep(duration) + def is_loaded_or_failed(self): + return self.is_loaded(allow_substates=True) or self.is_failed() - def command_print(self, command = "", **kwargs): - return "running command {}".format(command) - - def pause_print(self, music = None, **kwargs): - if music is not None: - return "pausing {}".format(music.filename) - else: - return "pausing all musics" - - def play_print(self, music = None, fade_in = 0, start_at = 0, - restart_if_running = False, volume = 100, **kwargs): - message = "starting " - if music is not None: - message += music.filename + def callback_music_loaded(self, success): + if success: + self.success() else: - message += "music" - - if start_at != 0: - message += " at {}s".format(start_at) - - if fade_in != 0: - message += " with {}s fade_in".format(fade_in) - - message += " at volume {}%".format(volume) - - if restart_if_running: - message += " (restarting if already running)" - - return message - - def stop_print(self, music = None, fade_out = 0, **kwargs): - if music is not None: - if fade_out == 0: - return "stopping music {}".format(music.filename) + self.fail() + + # Machine states / events + def on_enter_loading(self): + if hasattr(actions, self.action): + if 'music' in self.arguments: + self.arguments['music'].subscribe_loaded( + self.callback_music_loaded) else: - return "stopping music {} with {}s fadeout".format(music.filename, fade_out) + self.success() else: - if fade_out == 0: - return "stopping all musics" - else: - return "stopping all musics with {}s fadeout".format(fade_out) - - def stop_all_actions_print(self): - return "stopping all actions" - - def volume_print(self, music = None, value = 100, *kwargs): + error_print("Unknown action {}".format(self.action)) + self.fail() + + def on_enter_loaded_running(self, key_start_time): + debug_print(self.description()) + if hasattr(actions, self.action): + getattr(actions, self.action).run(self, + key_start_time=key_start_time, **self.arguments) + + def poll_loaded(self): + self.key.callback_action_ready(self, + self.is_loaded(allow_substates=True)) + + # This one cannot be in the Machine state since it would be queued to run + # *after* the wait is ended... + def interrupt(self): + if getattr(actions, self.action, None) and\ + hasattr(getattr(actions, self.action), 'interrupt'): + return getattr(getattr(actions, self.action), 'interrupt')( + self, **self.arguments) + + # Helpers + def music_list(self, music): if music is not None: - return "setting volume of {} to {}%".format(music.filename, value) + return [music] else: - return "setting volume to {}%".format(value) - - def wait_print(self, duration, **kwargs): - return "waiting {}s".format(duration) - + return self.mapping.open_files.values() + def description(self): + if hasattr(actions, self.action): + return getattr(actions, self.action)\ + .description(self, **self.arguments) + else: + return "unknown action {}".format(self.action)