musicpom/utils/audio_visualizer.py
billypom on debian 695c1884bd music pom time
2024-01-17 23:36:00 -05:00

20 lines
581 B
Python

from PyQt5 import QtWidgets, QtGui, QtCore
import numpy as np
from .fft_analyser import FFTAnalyser
class AudioVisualizer(QtWidgets.QWidget):
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)