Add Sound Module

2023-12-19 10:01:09 -05:00
parent 11726900f7
commit bb5287743c
7 changed files with 177 additions and 99 deletions
--- a/lib/sounds.py
+++ b/lib/sounds.py
@@ -0,0 +1,108 @@
+from PyQt6.QtCore import QObject, Qt, QUrl, pyqtSlot
+from PyQt6.QtMultimedia import (
+    QAudioDevice,
+    QAudioOutput,
+    QMediaDevices,
+    QMediaPlayer,
+    QSoundEffect,
+)
+
+# from PyQt6.QtWidgets import QWidget
+
+
+class SoundOff(QObject):
+    def __init__(self):
+        super().__init__()
+        #
+        # Setup devices
+        #
+        self.virtualDevice = None
+        dev = None
+        for output in QMediaDevices.audioOutputs():
+            if output.id().data().decode("utf-8") == "virt-input":
+                self.virtualDevice = output
+            if output.isDefault():
+                self.localDevice = output
+
+        self.alertEffect = QSoundEffect()
+        self.alertEffect.setSource(QUrl("qrc:/beep.wav"))
+        self.alertEffect.setAudioDevice(self.localDevice)
+        self.alertEffect.setVolume(0.25)
+        self.alertEffect.setLoopCount(1)
+
+        self.localPlayer = QMediaPlayer()
+        self.localPlayer.setObjectName("localPlayer")
+        self.localOutput = QAudioOutput()
+        self.localOutput.setDevice(self.localDevice)
+        self.localPlayer.setAudioOutput(self.localOutput)
+        if self.virtualDevice:
+            self.virtualPlayer = QMediaPlayer()
+            self.virtualPlayer.setObjectName("virtualPlayer")
+            self.virtualOutput = QAudioOutput()
+            self.virtualOutput.setDevice(self.virtualDevice)
+            self.virtualPlayer.setAudioOutput(self.virtualOutput)
+        #
+        # Connections
+        #
+        self.localPlayer.errorOccurred.connect(self.mediaError)
+        self.localPlayer.mediaStatusChanged.connect(self.mediaStatus)
+        self.localPlayer.playbackStateChanged.connect(self.playbackState)
+        if self.virtualDevice:
+            self.virtualPlayer.errorOccurred.connect(self.mediaError)
+            self.virtualPlayer.mediaStatusChanged.connect(self.mediaStatus)
+            self.virtualPlayer.playbackStateChanged.connect(self.playbackState)
+
+    @pyqtSlot()
+    def alert(self):
+        self.alertEffect.play()
+        return
+
+    @pyqtSlot(QMediaPlayer.Error, str)
+    def mediaError(self, error, string):
+        print(error)
+        print(str)
+        return
+
+    @pyqtSlot(QMediaPlayer.MediaStatus)
+    def mediaStatus(self, status):
+        if status == QMediaPlayer.MediaStatus.LoadedMedia:
+            self.sender().play()
+        return
+
+    @pyqtSlot(QMediaPlayer.PlaybackState)
+    def playbackState(self, state):
+        return
+
+    #
+    # Communications slots
+    #
+    @pyqtSlot()
+    def soundAlert(self):
+        self.alertEffect.play()
+        return
+
+    @pyqtSlot(str)
+    def playSound(self, url):
+        src = QUrl(url)
+        if not self.localPlayer.audioOutput():
+            self.localPlayer.setAudioOutput(self.localOutput)
+        self.localPlayer.setSource(src)
+        self.localPlayer.setPosition(0)
+        if (
+            self.localPlayer.mediaStatus()
+            == QMediaPlayer.MediaStatus.LoadedMedia
+        ):
+            self.localPlayer.play()
+        if not self.virtualDevice:
+            return
+        return
+        self.virtualPlayer.setSource(src)
+        self.virtualPlayer.setPosition(0)
+        if not self.virtualPlayer.audioOutput():
+            self.virtualPlayer.setAudioOutput(self.virtualOutput)
+        if (
+            self.virtualPlayer.mediaStatus()
+            == QMediaPlayer.MediaStatus.LoadedMedia
+        ):
+            self.virtualPlayer.play()
+        return