scripts/examples: Update ulab fft examples.

This commit is contained in:
iabdalkader 2023-03-07 19:23:47 +01:00
parent 58ee7322fa
commit 488cb5e0a9
4 changed files with 12 additions and 8 deletions

View File

@ -1,6 +1,7 @@
import image, audio, time import image, audio, time
from ulab import numpy as np from ulab import numpy as np
from ulab import scipy as sp from ulab import scipy as sp
from ulab import utils
CHANNELS = 1 CHANNELS = 1
SIZE = 256//(2*CHANNELS) SIZE = 256//(2*CHANNELS)
@ -38,10 +39,10 @@ while (True):
raw_buf = None raw_buf = None
if CHANNELS == 1: if CHANNELS == 1:
fft_buf = sp.signal.spectrogram(pcm_buf) fft_buf = utils.spectrogram(pcm_buf)
l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100) l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100)
else: else:
fft_buf = sp.signal.spectrogram(pcm_buf[0::2]) fft_buf = utils.spectrogram(pcm_buf[0::2])
l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100) l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100)
r_lvl = int((np.mean(abs(pcm_buf[0::2])) / 32768)*100) r_lvl = int((np.mean(abs(pcm_buf[0::2])) / 32768)*100)

View File

@ -1,6 +1,7 @@
import image, audio, time import image, audio, time
from ulab import numpy as np from ulab import numpy as np
from ulab import scipy as sp from ulab import scipy as sp
from ulab import utils
CHANNELS = 1 CHANNELS = 1
FREQUENCY = 32000 FREQUENCY = 32000
@ -41,10 +42,10 @@ while (True):
raw_buf = None raw_buf = None
if CHANNELS == 1: if CHANNELS == 1:
fft_buf = sp.signal.spectrogram(pcm_buf) fft_buf = utils.spectrogram(pcm_buf)
l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100) l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100)
else: else:
fft_buf = sp.signal.spectrogram(pcm_buf[0::2]) fft_buf = utils.spectrogram(pcm_buf[0::2])
l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100) l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100)
r_lvl = int((np.mean(abs(pcm_buf[0::2])) / 32768)*100) r_lvl = int((np.mean(abs(pcm_buf[0::2])) / 32768)*100)

View File

@ -1,6 +1,7 @@
import image, audio, time import image, audio, time
from ulab import numpy as np from ulab import numpy as np
from ulab import scipy as sp from ulab import scipy as sp
from ulab import utils
CHANNELS = 1 CHANNELS = 1
SIZE = 256//(2*CHANNELS) SIZE = 256//(2*CHANNELS)
@ -38,10 +39,10 @@ while (True):
raw_buf = None raw_buf = None
if CHANNELS == 1: if CHANNELS == 1:
fft_buf = sp.signal.spectrogram(pcm_buf) fft_buf = utils.spectrogram(pcm_buf)
l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100) l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100)
else: else:
fft_buf = sp.signal.spectrogram(pcm_buf[0::2]) fft_buf = utils.spectrogram(pcm_buf[0::2])
l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100) l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100)
r_lvl = int((np.mean(abs(pcm_buf[0::2])) / 32768)*100) r_lvl = int((np.mean(abs(pcm_buf[0::2])) / 32768)*100)

View File

@ -1,6 +1,7 @@
import image, audio, time import image, audio, time
from ulab import numpy as np from ulab import numpy as np
from ulab import scipy as sp from ulab import scipy as sp
from ulab import utils
CHANNELS = 2 CHANNELS = 2
SIZE = 512//(2*CHANNELS) SIZE = 512//(2*CHANNELS)
@ -38,10 +39,10 @@ while (True):
raw_buf = None raw_buf = None
if CHANNELS == 1: if CHANNELS == 1:
fft_buf = sp.signal.spectrogram(pcm_buf) fft_buf = utils.spectrogram(pcm_buf)
l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100) l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100)
else: else:
fft_buf = sp.signal.spectrogram(pcm_buf[0::2]) fft_buf = utils.spectrogram(pcm_buf[0::2])
l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100) l_lvl = int((np.mean(abs(pcm_buf[1::2])) / 32768)*100)
r_lvl = int((np.mean(abs(pcm_buf[0::2])) / 32768)*100) r_lvl = int((np.mean(abs(pcm_buf[0::2])) / 32768)*100)