149 lines
5.0 KiB
Python
149 lines
5.0 KiB
Python
import os
|
|
from pydub import AudioSegment
|
|
from pygame import mixer
|
|
from PyQt5.QtCore import QTimer, pyqtSlot, QObject
|
|
|
|
from constants import Constants
|
|
import qtawesome as qta
|
|
|
|
|
|
class AudioPlayer(QObject):
|
|
"""Subclass QObject. Audio player widget for the audio samples.
|
|
|
|
The only public methods are the __init__
|
|
method, set_audio_player, which loads the current file and refresh_btns_colors.
|
|
Everything else is managed internally."""
|
|
|
|
_TIME_STEP = 500 # Milliseconds.
|
|
|
|
def __init__(self, play,
|
|
pause,
|
|
stop,
|
|
volume,
|
|
audio_progress,
|
|
active_color,
|
|
inactive_color):
|
|
"""Initialize the player."""
|
|
super().__init__()
|
|
self._paused = False
|
|
self._first_call = True
|
|
self._play = play
|
|
self._pause = pause
|
|
self._stop = stop
|
|
self._volume = volume
|
|
self._audio_progress = audio_progress
|
|
self._audio_file = None
|
|
self._timer = QTimer()
|
|
self._timer.timeout.connect(self._update_bar)
|
|
self._play.clicked.connect(self._play_audio)
|
|
self._pause.clicked.connect(self._pause_audio)
|
|
self._stop.clicked.connect(self._stop_audio)
|
|
self._volume.valueChanged.connect(self._set_volume)
|
|
self._play.setIconSize(self._play.size())
|
|
self._pause.setIconSize(self._pause.size())
|
|
self._stop.setIconSize(self._stop.size())
|
|
self.refresh_btns_colors(active_color, inactive_color)
|
|
|
|
def refresh_btns_colors(self, active_color, inactive_color):
|
|
"""Repaint the buttons of the widgetd after the theme has changed."""
|
|
self._play.setIcon(qta.icon('fa5.play-circle',
|
|
color=active_color,
|
|
color_disabled=inactive_color))
|
|
self._pause.setIcon(qta.icon('fa5.pause-circle',
|
|
color=active_color,
|
|
color_disabled=inactive_color))
|
|
self._stop.setIcon(qta.icon('fa5.stop-circle',
|
|
color=active_color,
|
|
color_disabled=inactive_color))
|
|
|
|
@pyqtSlot()
|
|
def _set_volume(self):
|
|
"""Set the volume of the audio samples."""
|
|
if mixer.get_init():
|
|
mixer.music.set_volume(
|
|
self._volume.value() / self._volume.maximum()
|
|
)
|
|
|
|
def _reset_audio_widget(self):
|
|
"""Reset the widget. Stop all playing samples."""
|
|
if mixer.get_init():
|
|
if mixer.music.get_busy():
|
|
mixer.music.stop()
|
|
self._timer.stop()
|
|
mixer.quit()
|
|
self._audio_progress.reset()
|
|
self._enable_buttons(False, False, False)
|
|
self._paused = False
|
|
|
|
@pyqtSlot()
|
|
def _update_bar(self):
|
|
"""Update the progress bar."""
|
|
pos = mixer.music.get_pos()
|
|
if pos == -1:
|
|
self._timer.stop()
|
|
self._audio_progress.reset()
|
|
self._enable_buttons(True, False, False)
|
|
else:
|
|
self._audio_progress.setValue(pos)
|
|
|
|
def _set_max_progress_bar(self):
|
|
"""Set the maximum value of the progress bar."""
|
|
self._audio_progress.setMaximum(
|
|
mixer.Sound(self._audio_file).get_length() * 1000
|
|
)
|
|
|
|
def set_audio_player(self, fname=""):
|
|
"""Set the current audio sample."""
|
|
self._first_call = True
|
|
self._reset_audio_widget()
|
|
full_name = os.path.join(
|
|
Constants.DATA_FOLDER,
|
|
Constants.AUDIO_FOLDER,
|
|
fname + '.ogg'
|
|
)
|
|
if os.path.exists(full_name):
|
|
self._play.setEnabled(True)
|
|
self._audio_file = full_name
|
|
|
|
@pyqtSlot()
|
|
def _play_audio(self):
|
|
"""Play the audio sample."""
|
|
if not self._paused:
|
|
if self._first_call:
|
|
self._first_call = False
|
|
mixer.init(
|
|
frequency=AudioSegment.from_ogg(self._audio_file).frame_rate,
|
|
buffer=2048
|
|
)
|
|
mixer.music.load(self._audio_file)
|
|
self._set_volume()
|
|
self._set_max_progress_bar()
|
|
mixer.music.play()
|
|
else:
|
|
mixer.music.unpause()
|
|
self._paused = False
|
|
self._timer.start(self._TIME_STEP)
|
|
self._enable_buttons(False, True, True)
|
|
|
|
@pyqtSlot()
|
|
def _stop_audio(self):
|
|
"""Stop the audio sample."""
|
|
mixer.music.stop()
|
|
self._audio_progress.reset()
|
|
self._timer.stop()
|
|
self._enable_buttons(True, False, False)
|
|
|
|
@pyqtSlot()
|
|
def _pause_audio(self):
|
|
"""Pause the audio sample."""
|
|
mixer.music.pause()
|
|
self._timer.stop()
|
|
self._paused = True
|
|
self._enable_buttons(True, False, False)
|
|
|
|
def _enable_buttons(self, play_en, pause_en, stop_en):
|
|
"""Set the three buttons status."""
|
|
self._play.setEnabled(play_en)
|
|
self._pause.setEnabled(pause_en)
|
|
self._stop.setEnabled(stop_en)
|