python-audio-autotest-3.10 1.5.12rc1__py3-none-any.whl → 1.5.12rc3__py3-none-any.whl

pyaatlibs/logger.py

@@ -12,12 +12,15 @@ except ImportError:
     import Queue as queue
     import StringIO as sio
 
+
 class LoggerThread(threading.Thread):
     MAX_SIZE = 100000
     BUF_SIZE = 10
     LOG_DIR = ROOT_DIR + "{}{}{}".format(SEP, "{}log", SEP)
 
-    def __init__(self, prefix="", logfolder_prefix="", max_size=MAX_SIZE, buf_size=BUF_SIZE, log_dir=LOG_DIR):
+    def __init__(
+        self, prefix="", logfolder_prefix="", max_size=MAX_SIZE, buf_size=BUF_SIZE, log_dir=LOG_DIR
+    ):
         super(LoggerThread, self).__init__()
         self.daemon = True
         self.msg_q = queue.Queue()
@@ -27,7 +30,9 @@ class LoggerThread(threading.Thread):
         self.max_size = max_size
         self.buf_size = buf_size
         self.current_size = 0
-        self.log_dir = log_dir.format("{}-".format(logfolder_prefix) if len(logfolder_prefix) > 0 else logfolder_prefix)
+        self.log_dir = log_dir.format(
+            "{}-".format(logfolder_prefix) if len(logfolder_prefix) > 0 else logfolder_prefix
+        )
         self._to_stdout = False
         self._to_file = False
 
@@ -45,7 +50,9 @@ class LoggerThread(threading.Thread):
     def _update_timestamp(self):
         t = datetime.datetime.now()
         prefix = "{}-".format(self.prefix) if len(self.prefix) > 0 else ""
-        self.filename = "{}{}{:02d}{:02d}_{:02d}{:02d}{:02d}.log.txt".format(prefix, t.year, t.month, t.day, t.hour, t.minute, t.second)
+        self.filename = "{}{}{:02d}{:02d}_{:02d}{:02d}{:02d}.log.txt".format(
+            prefix, t.year, t.month, t.day, t.hour, t.minute, t.second
+        )
         self.log_timestamp = t
 
     def _dump(self):
@@ -61,6 +68,7 @@ class LoggerThread(threading.Thread):
     def wait_for_queue_empty(self):
         while not self.msg_q.empty():
             import time
+
             time.sleep(0.5)
 
     def push(self, msg):
@@ -94,13 +102,13 @@ class LoggerThread(threading.Thread):
 
                 if self._to_stdout:
                     import sys
+
                     sys.stdout.write(logtext)
                     sys.stdout.flush()
 
                 self.current_size += 1
 
-            if self.current_size > 0 and \
-                (self.current_size % self.buf_size == 0 or force_dump):
+            if self.current_size > 0 and (self.current_size % self.buf_size == 0 or force_dump):
                 self._dump()
 
             if self.current_size >= self.max_size:
@@ -112,6 +120,7 @@ class LoggerThread(threading.Thread):
             self._dump()
             self.current_size = 0
 
+
 class Logger(object):
     WORK_THREAD = None
     HAS_BEEN_INIT = False
@@ -131,11 +140,15 @@ class Logger(object):
         BOTH_FILE_AND_STDOUT = STDOUT | FILE
 
     @staticmethod
-    def init(mode=Mode.BOTH_FILE_AND_STDOUT, prefix="", logfolder_prefix="", log_dir=LoggerThread.LOG_DIR):
+    def init(
+        mode=Mode.BOTH_FILE_AND_STDOUT, prefix="", logfolder_prefix="", log_dir=LoggerThread.LOG_DIR
+    ):
         if Logger.HAS_BEEN_INIT:
             return
 
-        Logger.WORK_THREAD = LoggerThread(prefix=prefix, logfolder_prefix=logfolder_prefix, log_dir=log_dir)
+        Logger.WORK_THREAD = LoggerThread(
+            prefix=prefix, logfolder_prefix=logfolder_prefix, log_dir=log_dir
+        )
 
         if mode & Logger.Mode.STDOUT > 0:
             Logger.WORK_THREAD.to_stdout()
@@ -162,7 +175,7 @@ class Logger(object):
     @staticmethod
     def log(tag=None, msg=None, level=Verbosity.NONE):
         if not tag or not msg:
-            raise(ValueError("no tag or msg argument for Logger.log"))
+            raise (ValueError("no tag or msg argument for Logger.log"))
 
         if Logger.VERBOSITY_LEVEL != Logger.Verbosity.NONE and level > Logger.VERBOSITY_LEVEL:
             return

pyaatlibs/signalanalyzer.py

@@ -1,12 +1,14 @@
 import numpy as np
 
+
 def sort_values(signal):
     signal = np.array(signal)
     idices = np.argsort(signal, axis=0)
     return zip(list(idices), list(signal[idices]))
 
+
 def find_peaks(signal):
     peaks = []
     max_idx = np.argmax(signal)
-    peaks.append((max_idx, signal[max_idx]+1e-50))
+    peaks.append((max_idx, signal[max_idx] + 1e-50))
     return peaks

pyaatlibs/signalmatcher.py

@@ -3,6 +3,7 @@ from librosa.core import load as audioload
 from scipy.fftpack import fft
 import matplotlib.pyplot as plt
 
+
 class SignalMatcher(object):
     FRAME_MILLIS_PER_FEATURE = 20
 
@@ -13,33 +14,37 @@ class SignalMatcher(object):
 
     def _gen_feat(self):
         sig_len = len(self.refsig)
-        framesize = int(np.round(self.reffs*SignalMatcher.FRAME_MILLIS_PER_FEATURE/1000.))
-        nfft = int(2**np.ceil(np.log2(framesize)))
-        self.feats = np.zeros([nfft/2, int(np.ceil(sig_len*1./framesize))])
+        framesize = int(np.round(self.reffs * SignalMatcher.FRAME_MILLIS_PER_FEATURE / 1000.0))
+        nfft = int(2 ** np.ceil(np.log2(framesize)))
+        self.feats = np.zeros([nfft / 2, int(np.ceil(sig_len * 1.0 / framesize))])
 
         frame_sig = np.zeros([framesize])
         for frame_idx in range(self.feats.shape[1]):
             frame_sig[:] = 0
             idx_from = frame_idx * framesize
             idx_to = min([idx_from + framesize, sig_len])
-            frame_sig[:idx_to-idx_from] = self.refsig[idx_from:idx_to]
-            self.feats[:, frame_idx] = np.abs(fft(frame_sig, nfft))[:nfft/2]
+            frame_sig[: idx_to - idx_from] = self.refsig[idx_from:idx_to]
+            self.feats[:, frame_idx] = np.abs(fft(frame_sig, nfft))[: nfft / 2]
 
     def visualize_feat(self, outpath):
         feats = np.array(self.feats)
         feats += 1e-32
         feats = 20 * np.log10(feats)
-        plt.imshow(feats, vmax=np.max(feats), vmin=np.max(feats)-40, cmap="gray", origin="lower")
+        plt.imshow(feats, vmax=np.max(feats), vmin=np.max(feats) - 40, cmap="gray", origin="lower")
         plt.colorbar()
 
-        ticks = plt.gca().get_yticks()*1.0/feats.shape[0] * self.reffs/2.0
+        ticks = plt.gca().get_yticks() * 1.0 / feats.shape[0] * self.reffs / 2.0
         ticks = np.array(np.round(ticks), dtype=int)
         plt.gca().set_yticklabels(ticks)
         plt.gca().set_ylabel("frequency (Hz)")
-        plt.gca().set_xlabel("frame index ({} ms/frame)".format(SignalMatcher.FRAME_MILLIS_PER_FEATURE))
+        plt.gca().set_xlabel(
+            "frame index ({} ms/frame)".format(SignalMatcher.FRAME_MILLIS_PER_FEATURE)
+        )
 
         xlim = plt.gca().get_xlim()
-        plt.gcf().set_size_inches([(xlim[1]-xlim[0])*SignalMatcher.FRAME_MILLIS_PER_FEATURE/1000.0, 3])
+        plt.gcf().set_size_inches(
+            [(xlim[1] - xlim[0]) * SignalMatcher.FRAME_MILLIS_PER_FEATURE / 1000.0, 3]
+        )
 
         plt.savefig(outpath, bbox_inches="tight", pad_inches=0, dpi=300)
         plt.gcf().clear()