musicpom/components/AudioVisualizer.py

27 lines
698 B
Python

from PyQt5 import QtWidgets
import numpy as np
from utils import FFTAnalyser
class AudioVisualizer(QtWidgets.QWidget):
"""_Audio Visualizer component_
Args:
QtWidgets (_type_): _description_
Returns:
_type_: _description_
"""
def __init__(self, media_player):
super().__init__()
self.media_player = media_player
self.fft_analyser = FFTAnalyser(self.media_player)
self.fft_analyser.calculated_visual.connect(self.set_amplitudes)
self.fft_analyser.start()
self.amps = np.array([])
def get_amplitudes(self):
return self.amps
def set_amplitudes(self, amps):
self.amps = np.array(amps)