Add librespot.audio.decoders

2021-05-20 07:03:32 +09:00
parent 60adbd8beb
commit 8d86accf48
7 changed files with 16 additions and 13 deletions
--- a/librespot/audio/decoders/AudioQuality.py
+++ b/librespot/audio/decoders/AudioQuality.py
@@ -0,0 +1,38 @@
+from __future__ import annotations
+from librespot.proto.Metadata import AudioFile
+import enum
+import typing
+
+
+class AudioQuality(enum.Enum):
+    NORMAL = 0x00
+    HIGH = 0x01
+    VERY_HIGH = 0x02
+
+    @staticmethod
+    def get_quality(audio_format: AudioFile.Format) -> AudioQuality:
+        if audio_format == AudioFile.MP3_96 or \
+                   audio_format == AudioFile.OGG_VORBIS_96 or \
+                   audio_format == AudioFile.AAC_24_NORM:
+            return AudioQuality.NORMAL
+        if audio_format == AudioFile.MP3_160 or \
+                        audio_format == AudioFile.MP3_160_ENC or \
+                        audio_format == AudioFile.OGG_VORBIS_160 or \
+                        audio_format == AudioFile.AAC_24:
+            return AudioQuality.HIGH
+        if audio_format == AudioFile.MP3_320 or \
+                        audio_format == AudioFile.MP3_256 or \
+                        audio_format == AudioFile.OGG_VORBIS_320 or \
+                        audio_format == AudioFile.AAC_48:
+            return AudioQuality.VERY_HIGH
+        raise RuntimeError("Unknown format: {}".format(format))
+
+    def get_matches(self,
+                    files: typing.List[AudioFile]) -> typing.List[AudioFile]:
+        file_list = []
+        for file in files:
+            if hasattr(file, "format") and AudioQuality.get_quality(
+                    file.format) == self:
+                file_list.append(file)
+
+        return file_list