Merge pull request #69 from djfun/feature-newgui

GUI Redesign with Component System

1 files changed, 203 insertions, 0 deletions

diff --git a/src/components/original.py b/src/components/original.py
new file mode 100644
index 0000000..80228fe
--- /dev/null
+++ b/src/components/original.py
@@ -0,0 +1,203 @@
+import numpy
+from PIL import Image, ImageDraw
+from PyQt5 import QtGui, QtCore, QtWidgets
+from PyQt5.QtGui import QColor
+import os
+import time
+from copy import copy
+
+from ..component import Component
+from ..toolkit.frame import BlankFrame
+
+
+class Component(Component):
+    name = 'Classic Visualizer'
+    version = '1.0.0'
+
+    def names(*args):
+        return ['Original Audio Visualization']
+
+    def properties(self):
+        return ['pcm']
+
+    def widget(self, *args):
+        self.scale = 20
+        self.y = 0
+        super().widget(*args)
+
+        self.page.comboBox_visLayout.addItem("Classic")
+        self.page.comboBox_visLayout.addItem("Split")
+        self.page.comboBox_visLayout.addItem("Bottom")
+        self.page.comboBox_visLayout.addItem("Top")
+        self.page.comboBox_visLayout.setCurrentIndex(0)
+
+        self.page.lineEdit_visColor.setText('255,255,255')
+
+        self.trackWidgets({
+            'visColor': self.page.lineEdit_visColor,
+            'layout': self.page.comboBox_visLayout,
+            'scale': self.page.spinBox_scale,
+            'y': self.page.spinBox_y,
+        }, colorWidgets={
+            'visColor': self.page.pushButton_visColor,
+        }, relativeWidgets=[
+            'y',
+        ])
+
+    def previewRender(self):
+        spectrum = numpy.fromfunction(
+            lambda x: float(self.scale)/2500*(x-128)**2, (255,), dtype="int16")
+        return self.drawBars(
+            self.width, self.height, spectrum, self.visColor, self.layout
+        )
+
+    def preFrameRender(self, **kwargs):
+        super().preFrameRender(**kwargs)
+        self.smoothConstantDown = 0.08
+        self.smoothConstantUp = 0.8
+        self.lastSpectrum = None
+        self.spectrumArray = {}
+
+        for i in range(0, len(self.completeAudioArray), self.sampleSize):
+            if self.canceled:
+                break
+            self.lastSpectrum = self.transformData(
+                i, self.completeAudioArray, self.sampleSize,
+                self.smoothConstantDown, self.smoothConstantUp,
+                self.lastSpectrum)
+            self.spectrumArray[i] = copy(self.lastSpectrum)
+
+            progress = int(100*(i/len(self.completeAudioArray)))
+            if progress >= 100:
+                progress = 100
+            pStr = "Analyzing audio: "+str(progress)+'%'
+            self.progressBarSetText.emit(pStr)
+            self.progressBarUpdate.emit(int(progress))
+
+    def frameRender(self, frameNo):
+        arrayNo = frameNo * self.sampleSize
+        return self.drawBars(
+            self.width, self.height,
+            self.spectrumArray[arrayNo],
+            self.visColor, self.layout)
+
+    def transformData(
+      self, i, completeAudioArray, sampleSize,
+      smoothConstantDown, smoothConstantUp, lastSpectrum):
+        if len(completeAudioArray) < (i + sampleSize):
+            sampleSize = len(completeAudioArray) - i
+
+        window = numpy.hanning(sampleSize)
+        data = completeAudioArray[i:i+sampleSize][::1] * window
+        paddedSampleSize = 2048
+        paddedData = numpy.pad(
+            data, (0, paddedSampleSize - sampleSize), 'constant')
+        spectrum = numpy.fft.fft(paddedData)
+        sample_rate = 44100
+        frequencies = numpy.fft.fftfreq(len(spectrum), 1./sample_rate)
+
+        y = abs(spectrum[0:int(paddedSampleSize/2) - 1])
+
+        # filter the noise away
+        # y[y<80] = 0
+
+        y = self.scale * numpy.log10(y)
+        y[numpy.isinf(y)] = 0
+
+        if lastSpectrum is not None:
+            lastSpectrum[y < lastSpectrum] = \
+                y[y < lastSpectrum] * smoothConstantDown + \
+                lastSpectrum[y < lastSpectrum] * (1 - smoothConstantDown)
+
+            lastSpectrum[y >= lastSpectrum] = \
+                y[y >= lastSpectrum] * smoothConstantUp + \
+                lastSpectrum[y >= lastSpectrum] * (1 - smoothConstantUp)
+        else:
+            lastSpectrum = y
+
+        x = frequencies[0:int(paddedSampleSize/2) - 1]
+
+        return lastSpectrum
+
+    def drawBars(self, width, height, spectrum, color, layout):
+        vH = height-height/8
+        bF = width / 64
+        bH = bF / 2
+        bQ = bF / 4
+        imTop = BlankFrame(width, height)
+        draw = ImageDraw.Draw(imTop)
+        r, g, b = color
+        color2 = (r, g, b, 125)
+
+        bP = height / 1200
+
+        for j in range(0, 63):
+            draw.rectangle((
+                bH + j * bF, vH+bQ, bH + j * bF + bF, vH + bQ -
+                spectrum[j * 4] * bP - bH), fill=color2)
+
+            draw.rectangle((
+                bH + bQ + j * bF, vH, bH + bQ + j * bF + bH, vH -
+                spectrum[j * 4] * bP), fill=color)
+
+        imBottom = imTop.transpose(Image.FLIP_TOP_BOTTOM)
+
+        im = BlankFrame(width, height)
+
+        if layout == 0:  # Classic
+            y = self.y - int(height/100*43)
+            im.paste(imTop, (0, y), mask=imTop)
+            y = self.y + int(height/100*43)
+            im.paste(imBottom, (0, y), mask=imBottom)
+
+        if layout == 1:  # Split
+            y = self.y + int(height/100*10)
+            im.paste(imTop, (0, y), mask=imTop)
+            y = self.y - int(height/100*10)
+            im.paste(imBottom, (0, y), mask=imBottom)
+
+        if layout == 2:  # Bottom
+            y = self.y + int(height/100*10)
+            im.paste(imTop, (0, y), mask=imTop)
+
+        if layout == 3:  # Top
+            y = self.y - int(height/100*10)
+            im.paste(imBottom, (0, y), mask=imBottom)
+
+        return im
+
+    def command(self, arg):
+        if '=' in arg:
+            key, arg = arg.split('=', 1)
+            try:
+                if key == 'color':
+                    self.page.lineEdit_visColor.setText(arg)
+                    return
+                elif key == 'layout':
+                    if arg == 'classic':
+                        self.page.comboBox_visLayout.setCurrentIndex(0)
+                    elif arg == 'split':
+                        self.page.comboBox_visLayout.setCurrentIndex(1)
+                    elif arg == 'bottom':
+                        self.page.comboBox_visLayout.setCurrentIndex(2)
+                    elif arg == 'top':
+                        self.page.comboBox_visLayout.setCurrentIndex(3)
+                    return
+                elif key == 'scale':
+                    arg = int(arg)
+                    self.page.spinBox_scale.setValue(arg)
+                    return
+                elif key == 'y':
+                    arg = int(arg)
+                    self.page.spinBox_y.setValue(arg)
+                    return
+            except ValueError:
+                print('You must enter a number.')
+                quit(1)
+        super().command(arg)
+
+    def commandHelp(self):
+        print('Give a layout name:\n    layout=[classic/split/bottom/top]')
+        print('Specify a color:\n    color=255,255,255')
+        print('Visualizer scale (20 is default):\n    scale=number')
+        print('Y position:\n    y=number')