X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=helpers%2Faction.py;h=4b5a71d3806275b5bf16198be985fda052f7b61f;hb=201d8411d7e1f70f13ced3fc797a696e6593f3a0;hp=69ae96fb8581b5f0bc0006d7841394e37ec87500;hpb=29597680758e4924aa71fc021465189e153f2016;p=perso%2FImmae%2FProjets%2FPython%2FMusicSampler.git diff --git a/helpers/action.py b/helpers/action.py index 69ae96f..4b5a71d 100644 --- a/helpers/action.py +++ b/helpers/action.py @@ -1,152 +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', - 'unpause', - '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'].check_is_loaded() - else: - return True - - def run(self): - print(self.description()) - getattr(self, self.action)(**self.arguments) - - def description(self): - return getattr(self, self.action + "_print")(**self.arguments) - - def command(self, command = "", **kwargs): - pass + def is_loaded_or_failed(self): + return self.is_loaded(allow_substates=True) or self.is_failed() - def music_list(self, music): - if music is not None: - return [music] + def callback_music_loaded(self, success): + if success: + self.success() else: - return self.key.parent.open_files.values() - - def pause(self, music = None, **kwargs): - for music in self.music_list(music): - if music.is_loaded_playing(): - music.pause() - - def unpause(self, music = None, **kwargs): - for music in self.music_list(music): - if music.is_loaded_paused(): - music.unpause() - - def play(self, music = None, fade_in = 0, start_at = 0, - restart_if_running = False, volume = 100, **kwargs): - if music is not None: - if restart_if_running: - if music.is_not_stopped(): - music.stop() - music.play(volume = volume, fade_in = fade_in, start_at = start_at) + 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: - if not music.is_not_stopped(): - music.play(volume = volume, fade_in = fade_in, start_at = start_at) - - def stop(self, music = None, fade_out = 0, **kwargs): - for music in self.music_list(music): - if music.is_loaded_paused() or music.is_loaded_playing(): - music.stop(fade_out = fade_out) - - def stop_all_actions(self, **kwargs): - self.key.parent.stop_all_running() - - def volume(self, music = None, value = 100, **kwargs): - if music is not None: - music.set_volume(value) - else: - # FIXME: todo - pass - - def wait(self, duration = 0, music = None, **kwargs): - # FIXME: Make it stoppable - # http://stackoverflow.com/questions/29082268/python-time-sleep-vs-event-wait - if music is None: - time.sleep(duration) + self.success() else: - # TODO - music.wait_end() - - 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.name) - else: - return "pausing all musics" - - def unpause_print(self, music = None, **kwargs): - if music is not None: - return "unpausing « {} »".format(music.name) - else: - return "unpausing 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 += "« {} »".format(music.name) - 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): + 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: - if fade_out == 0: - return "stopping music « {} »".format(music.name) - else: - return "stopping music « {} » with {}s fadeout".format(music.name, fade_out) + return [music] 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, **kwargs): - return "stopping all actions" + return self.mapping.open_files.values() - def volume_print(self, music = None, value = 100, **kwargs): - if music is not None: - return "setting volume of « {} » to {}%".format(music.name, value) + def description(self): + if hasattr(actions, self.action): + return getattr(actions, self.action)\ + .description(self, **self.arguments) else: - return "setting volume to {}%".format(value) - - def wait_print(self, duration, **kwargs): - return "waiting {}s".format(duration) - - + return "unknown action {}".format(self.action)