204 lines
6.7 KiB
Python
204 lines
6.7 KiB
Python
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')
|