# HG changeset patch
# User Amine Sehili <amine.sehili@gmail.com>
# Date 1548184952 -3600
# Node ID 87db8004ca238a28c733d79f894dfc9c2af299e3
# Parent  7771f11179fbba852a21efab4e42378b7819624c
Add function to load compressed audio formats with pydub

diff -r 7771f11179fb -r 87db8004ca23 auditok/io.py
--- a/auditok/io.py	Sun Jan 20 17:37:19 2019 +0100
+++ b/auditok/io.py	Tue Jan 22 20:22:32 2019 +0100
@@ -713,6 +713,39 @@
     )
 
 
+def _load_with_pydub(filename, audio_format, use_channel=0):
+    """Open compressed audio file using pydub. If a video file
+    is passed, its audio track(s) are extracted and loaded.
+    This function should not be called directely, use :func:`from_file`
+    instead.
+
+    :Parameters:
+
+    `filename`:
+        path to audio file.
+    `audio_format`:
+        string, audio file format (e.g. raw, webm, wav, ogg)
+    """
+    func_dict = {
+        "mp3": AudioSegment.from_mp3,
+        "ogg": AudioSegment.from_ogg,
+        "flv": AudioSegment.from_flv,
+    }
+    open_function = func_dict.get(audio_format, AudioSegment.from_file)
+    segment = open_function(filename)
+    data = segment._data
+    if segment.channels > 1:
+        data = _extract_selected_channel(
+            data, segment.channels, segment.sample_width, use_channel
+        )
+    return BufferAudioSource(
+        data_buffer=data,
+        sampling_rate=segment.frame_rate,
+        sample_width=segment.sample_width,
+        channels=1,
+    )
+
+
 def from_file(filename):
     """
     Create an `AudioSource` object using the audio file specified by `filename`.