89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
import os
|
|
import sys
|
|
from pygame import mixer
|
|
from PyQt5.QtCore import QTimer
|
|
|
|
|
|
class AudioPlayer(object):
|
|
"""
|
|
This is the audio player widget. The only public methods are the __init__
|
|
method and set_audio_player, which loads the current file. Everything else
|
|
is managed internally.
|
|
"""
|
|
__time_step = 1000
|
|
def __init__(self, play, pause, stop, volume, audio_progress):
|
|
mixer.init()
|
|
self.__paused = False
|
|
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)
|
|
|
|
def __set_volume(self):
|
|
mixer.music.set_volume(self.__volume.value() / self.__volume.maximum())
|
|
|
|
def __reset_audio_widget(self):
|
|
self.__audio_progress.reset()
|
|
self.__volume.setValue(50)
|
|
self.__enable_buttons(False, False, False)
|
|
self.__paused = False
|
|
if mixer.music.get_busy():
|
|
mixer.music.stop()
|
|
self.__timer.stop()
|
|
|
|
def __update_bar(self):
|
|
pos = mixer.music.get_pos()
|
|
if pos == -1:
|
|
self.__timer.stop()
|
|
self.__audio_progress.reset()
|
|
else:
|
|
self.__audio_progress.setValue(pos)
|
|
|
|
def __set_max_progress_bar(self):
|
|
self.__audio_progress.setMaximum(
|
|
mixer.Sound(self.__audio_file).get_length() * 1000
|
|
)
|
|
|
|
def set_audio_player(self, fname):
|
|
self.__reset_audio_widget()
|
|
full_name = os.path.join('Data', 'Audio', fname + '.ogg')
|
|
if os.path.exists(full_name):
|
|
self.__play.setEnabled(True)
|
|
self.__audio_file = full_name
|
|
self.__set_max_progress_bar()
|
|
mixer.music.load(full_name)
|
|
mixer.music.set_volume(self.__volume.value())
|
|
|
|
def __play_audio(self):
|
|
if not self.__paused:
|
|
mixer.music.play()
|
|
else:
|
|
mixer.music.unpause()
|
|
self.__paused = False
|
|
self.__timer.start(self.__time_step)
|
|
self.__enable_buttons(False, True, True)
|
|
|
|
def __stop_audio(self):
|
|
mixer.music.stop()
|
|
self.__timer.stop()
|
|
self.__enable_buttons(True, False, False)
|
|
|
|
def __pause_audio(self):
|
|
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):
|
|
self.__play.setEnabled(play_en)
|
|
self.__pause.setEnabled(pause_en)
|
|
self.__stop.setEnabled(stop_en)
|